[PE Helper/PXE Helpers] Add IPv6 support

Author: CodingWonders

Helpers/extps1/PE_Helper/pxehelpers/common/PXEHelpers.Common.ps1

@@ -3,6 +3,12 @@ using namespace System.Collections.Generic
 $global:product = ""
 $global:description = ""
 
+enum IPAddress {
+    IPv4 = 0
+    IPv6 = 1
+    Unknown = 2
+}
+
 function Show-CenteredTextBox {
     param (
         [string]$Text,
@@ -115,3 +121,25 @@ function Disable-Networking {
         Remove-Item -Path "$env:SYSTEMDRIVE\net_initiated" -Force
     }
 }
+
+function Test-IPAddressSyntax {
+    param (
+        [Parameter(Mandatory, Position = 0)] [string]$ipAddr
+    )
+
+    # For some stupid reason, the regex method that works in both a non-WinPE PWSH session and in a .NET application does
+    # NOT work on WinPE. GPT suggested using the IPAddress.AddressFamily property, and it produces viable results under a debugger environment. So
+    # we'll use these here.
+
+    # We'll first check IPv4, then IPv6
+    $parsed = $null
+    if ([System.Net.IPAddress]::TryParse($ipAddr.Trim(), [ref]$parsed)) {
+        if ($parsed.AddressFamily -eq [System.Net.Sockets.AddressFamily]::InterNetwork) {
+            return [IPAddress]::IPv4
+        } elseif ($parsed.AddressFamily -eq [System.Net.Sockets.AddressFamily]::InterNetworkV6) {
+            return [IPAddress]::IPv6
+        }
+    }
+
+    return [IPAddress]::Unknown
+}

Helpers/extps1/PE_Helper/pxehelpers/fog/foghelper.ps1

@@ -14,6 +14,10 @@ class ServerAuthentication {
         $this.serverIP = $ip
         $this.serverPort = $port
         $this.serverUnderlyingFogServerIP = $underlyingIp
+
+        if ((Test-IPAddressSyntax -ipAddr $this.serverIP) -eq [IPAddress]::IPv6) {
+            $this.serverIP = "[$($this.serverIP)]"
+        }
     }
 }
 

Helpers/extps1/PE_Helper/pxehelpers/wds/wdshelper.ps1

@@ -16,6 +16,10 @@ class ServerAuthentication {
         $this.serverPort = $port
         $this.serverUser = $user
         $this.serverPassword = $password
+
+        if ((Test-IPAddressSyntax -ipAddr $this.serverIP) -eq [IPAddress]::IPv6) {
+            $this.serverIP = "[$($this.serverIP)]"
+        }
     }
 }