Set-NetIPAddressViaSession.ps1

[CmdletBinding()]
param(
    [Parameter(Mandatory=$true)]
    [System.Management.Automation.Runspaces.PSSession[]]$Session,

    [string]$AdapterName = 'Ethernet',

    [Parameter(Mandatory=$true)]
    [string]$IPAddress,

    [Parameter(Mandatory=$true)]
    [byte]$PrefixLength,
    
    [Parameter(Mandatory=$true)]
    [string]$DefaultGateway,
    
    [string[]]$DnsAddresses = @('8.8.8.8','8.8.4.4'),

    [ValidateSet('Public', 'Private')]
    [string]$NetworkCategory = 'Public'
)

$ErrorActionPreference = 'Stop'

Invoke-Command -Session $Session { 
    Remove-NetRoute -NextHop $using:DefaultGateway -Confirm:$false -ErrorAction SilentlyContinue
    $neta = Get-NetAdapter $using:AdapterName        # Use the exact adapter name for multi-adapter VMs
    $neta | Set-NetConnectionProfile -NetworkCategory $using:NetworkCategory
    $neta | Set-NetIPInterface -Dhcp Disabled
    $neta | Get-NetIPAddress -AddressFamily IPv4 -ErrorAction SilentlyContinue | Remove-NetIPAddress -Confirm:$false 

    # New-NetIPAddress may fail for certain scenarios (e.g. PrefixLength = 32). Using netsh instead.
    $mask = [IPAddress](([UInt32]::MaxValue) -shl (32 - $using:PrefixLength) -shr (32 - $using:PrefixLength))
    netsh interface ipv4 set address name="$($neta.InterfaceAlias)" static $using:IPAddress $mask.IPAddressToString $using:DefaultGateway

    $neta | Set-DnsClientServerAddress -Addresses $using:DnsAddresses
} | Out-Null