Set-MSHVMigration.ps1

#Requires -Version 5.0
#Requires -Modules Hyper-V

<#
    .SYNOPSIS
        Enables/disables and configure migration on the Hyper-V host
    
    .DESCRIPTION  
        Use "Win2K12R2 or Win8.x" for execution on Windows Server 2012 R2 or on Windows 8.1,
        when execute on Windows Server 2016 / Windows 10 or newer, use "Newer Systems"

    .NOTES
        This PowerShell script was developed and optimized for ScriptRunner. The use of the scripts requires ScriptRunner. 
        The customer or user is authorized to copy the script from the repository and use them in ScriptRunner. 
        The terms of use for ScriptRunner do not apply to this script. In particular, ScriptRunner Software GmbH assumes no liability for the function, 
        the use and the consequences of the use of this freely available script.
        PowerShell is a product of Microsoft Corporation. ScriptRunner is a product of ScriptRunner Software GmbH.
        © ScriptRunner Software GmbH

    .COMPONENT
        Requires Module Hyper-V

    .LINK
        https://github.com/scriptrunner/ActionPacks/tree/master/Hyper-V/Host
    
    .Parameter VMHostName
        [sr-en] Name of the Hyper-V host

    .Parameter HostName
        [sr-en] Name of the Hyper-V host

    .Parameter AccessAccount
        [sr-en] User account that have permission to perform this action

    .Parameter Action
        [sr-en] Enables/disables the migration

    .Parameter MaxVirtualMachineMigrations
        [sr-en] Maximum number of live migrations that can be performed at the same time on the Hyper-V host

    .Parameter EnableUseAnyNetworkForMigration
        [sr-en] How networks are selected for incoming live migration traffic. 
        If set to $TRUE, any available network on the host can be used for this traffic. 
        If set to $FALSE, incoming live migration traffic is transmitted only on the networks specified in the MigrationNetworks property of the host

    .Parameter VMMigrationAuthenticationType
        [sr-en] Type of authentication to be used for live migrations

    .Parameter VMMigrationPerformanceOption
        [sr-en] Performance option to use for live migration

    .Parameter NetworksToAdd
        [sr-en] String representing one or more IPv4 or IPv6 subnet masks that specifies the networks to be added for virtual machine migration. 
        Use comma to separate the addresses

    .Parameter NetworksToRemove
        [sr-en] String representing one or more IPv4 or IPv6 subnet masks that specifies the networks to be removed from virtual machine migration. 
        Use comma to separate the addresses
#>

param(
    [Parameter(Mandatory = $true,ParameterSetName = "Win2K12R2 or Win8.x")]
    [string]$VMHostName,
    [Parameter(ParameterSetName = "Newer Systems")]
    [string]$HostName,
    [Parameter(ParameterSetName = "Newer Systems")]
    [PSCredential]$AccessAccount,
    [Parameter(ParameterSetName = "Win2K12R2 or Win8.x")]
    [Parameter(ParameterSetName = "Newer Systems")]
    [ValidateSet('Enable','Disable')]
    [string]$Action = "Enable",
    [Parameter(ParameterSetName = "Win2K12R2 or Win8.x")]
    [Parameter(ParameterSetName = "Newer Systems")]
    [int]$MaxVirtualMachineMigrations,
    [Parameter(ParameterSetName = "Win2K12R2 or Win8.x")]
    [Parameter(ParameterSetName = "Newer Systems")]
    [bool]$EnableUseAnyNetworkForMigration,
    [Parameter(ParameterSetName = "Win2K12R2 or Win8.x")]
    [Parameter(ParameterSetName = "Newer Systems")]
    [ValidateSet('CredSSP', 'Kerberos')]
    [string]$VMMigrationAuthenticationType = "CredSSP",
    [Parameter(ParameterSetName = "Win2K12R2 or Win8.x")]
    [Parameter(ParameterSetName = "Newer Systems")]
    [ValidateSet('TCPIP', 'Compression', 'SMB')]
    [string]$VMMigrationPerformanceOption = "TCPIP",
    [Parameter(ParameterSetName = "Win2K12R2 or Win8.x")]
    [Parameter(ParameterSetName = "Newer Systems")]
    [string]$NetworksToAdd,
    [Parameter(ParameterSetName = "Win2K12R2 or Win8.x")]
    [Parameter(ParameterSetName = "Newer Systems")]
    [string]$NetworksToRemove
)

Import-Module Hyper-V

try {
    [string[]]$Script:addAddr = @()
    [string[]]$Script:remAddr = @()
    if($PSCmdlet.ParameterSetName  -eq "Win2K12R2 or Win8.x"){
        $HostName=$VMHostName
    }   
    if([System.String]::IsNullOrWhiteSpace($HostName)){
        $HostName = "."
    }    
    if(-not [System.String]::IsNullOrWhiteSpace($NetworksToAdd)){
        $Script:addAddr = $NetworksToAdd.Split(',')
    }    
    if(-not [System.String]::IsNullOrWhiteSpace($NetworksToRemove)){
        $Script:remAddr = $NetworksToRemove.Split(',')
    }
    [string[]]$Properties = @('Name','VirtualMachineMigrationEnabled','MaximumVirtualMachineMigrations','UseAnyNetworkForMigration','VirtualMachineMigrationAuthenticationType','VirtualMachineMigrationPerformanceOption')
    [hashtable]$cmdArgs = @{'ErrorAction' = 'Stop'}
    if($null -eq $AccessAccount){
        $cmdArgs.Add('ComputerName',$HostName)
    }
    else {
        $Script:Cim = New-CimSession -ComputerName $HostName -Credential $AccessAccount
        $cmdArgs.Add('CimSession',$Script:Cim)
    } 
    
    if($Action -eq 'Enable'){
        Enable-VMMigration @cmdArgs
    }
    else {
        Disable-VMMigration @cmdArgs
    }
    if($PSBoundParameters.ContainsKey('MaxVirtualMachineMigrations') -eq $true ){
        Set-VMHost @cmdArgs -MaximumVirtualMachineMigrations $MaxVirtualMachineMigrations
    }
    if($PSBoundParameters.ContainsKey('EnableUseAnyNetworkForMigration') -eq $true ){
        Set-VMHost @cmdArgs -UseAnyNetworkForMigration $EnableUseAnyNetworkForMigration
    }
    if($PSBoundParameters.ContainsKey('VMMigrationAuthenticationType') -eq $true ){
        Set-VMHost @cmdArgs -VirtualMachineMigrationAuthenticationType $VMMigrationAuthenticationType
    }
    if($PSBoundParameters.ContainsKey('VMMigrationPerformanceOption') -eq $true ){
        Set-VMHost @cmdArgs -VirtualMachineMigrationPerformanceOption $VMMigrationPerformanceOption
    }        
    foreach($mask in $Script:addAddr){
        Add-VMMigrationNetwork @cmdArgs -Subnet $mask.Trim()
    }
    foreach($maskR in $Script:remAddr){
        Remove-VMMigrationNetwork @cmdArgs -Subnet $maskR.Trim()
    }
    $output = Get-VMHost @cmdArgs | Select-Object $Properties
    $output += Get-VMMigrationNetwork @cmdArgs | Select-Object *
            
    if($SRXEnv) {
        $SRXEnv.ResultMessage = $output
    }    
    else {
        Write-Output $output
    }
}
catch {
    throw
}
finally{
    if($null -ne $Script:Cim){
        Remove-CimSession $Script:Cim 
    }
}