Add comprehensive test suite, fix minor issues, rename function and file names to match approved verbs (#708)

2026-08-23 08:02:07 +00:00 · 2026-07-19 22:06:07 +02:00
parent a7292e4f35
commit 9c033dbf98
116 changed files with 4629 additions and 650 deletions
@@ -0,0 +1,125 @@
+<#
+    .SYNOPSIS
+        Normalizes a rooted registry path and returns its hive and subkey components.
+#>
+function Split-RegistryPath {
+    param(
+        [Parameter(Mandatory)]
+        [string]$path
+    )
+
+    $normalizedPath = [string]$path
+    if ([string]::IsNullOrWhiteSpace($normalizedPath)) {
+        return $null
+    }
+
+    $normalizedPath = $normalizedPath.Trim().Replace('/', '\')
+    if ([string]::IsNullOrWhiteSpace($normalizedPath)) {
+        return $null
+    }
+
+    if ($normalizedPath -notmatch '^(?<hive>HKEY_[^\\]+)(?:\\(?<subKey>.*))?$') {
+        return $null
+    }
+
+    $hiveName = [string]$matches.hive
+
+    $normalizedSubKey = if ($null -ne $matches.subKey) {
+        ([string]$matches.subKey).Trim('\\')
+    }
+    else {
+        $null
+    }
+
+    if ($hiveName.Equals('HKEY_USERS', [System.StringComparison]::OrdinalIgnoreCase) -and
+        -not [string]::IsNullOrWhiteSpace($normalizedSubKey) -and
+        -not [string]::IsNullOrWhiteSpace([string]$script:RegistryTargetHiveMountName)) {
+        if ($normalizedSubKey -match '^(?<mount>[^\\]+)(?:\\(?<rest>.*))?$') {
+            $mountName = [string]$matches.mount
+            if ($mountName.Equals('Default', [System.StringComparison]::OrdinalIgnoreCase)) {
+                $remainingSubKey = if ($matches.rest) { [string]$matches.rest } else { '' }
+                $targetMountName = [string]$script:RegistryTargetHiveMountName
+                if ([string]::IsNullOrWhiteSpace($remainingSubKey)) {
+                    $normalizedSubKey = $targetMountName
+                }
+                else {
+                    $normalizedSubKey = "$targetMountName\$remainingSubKey"
+                }
+            }
+        }
+    }
+
+    return [PSCustomObject]@{
+        Hive = $hiveName
+        SubKey = $normalizedSubKey
+    }
+}
+
+<#
+    .SYNOPSIS
+        Returns the .NET registry root key for a supported registry hive name.
+#>
+function Get-RegistryRootKey {
+    param(
+        [Parameter(Mandatory)]
+        [string]$hiveName
+    )
+
+    switch ($hiveName.ToUpperInvariant()) {
+        'HKEY_CURRENT_USER' { return [Microsoft.Win32.Registry]::CurrentUser }
+        'HKEY_LOCAL_MACHINE' { return [Microsoft.Win32.Registry]::LocalMachine }
+        'HKEY_CLASSES_ROOT' { return [Microsoft.Win32.Registry]::ClassesRoot }
+        'HKEY_USERS' { return [Microsoft.Win32.Registry]::Users }
+        'HKEY_CURRENT_CONFIG' { return [Microsoft.Win32.Registry]::CurrentConfig }
+        default { return $null }
+    }
+}
+
+<#
+    .SYNOPSIS
+        Deletes a registry subkey tree and ignores a key that has already disappeared.
+#>
+function Remove-RegistrySubKeyTreeIfExists {
+    param(
+        [Parameter(Mandatory)]
+        $RootKey,
+        [Parameter(Mandatory)]
+        [string]$SubKeyPath
+    )
+
+    try {
+        $RootKey.DeleteSubKeyTree($SubKeyPath, $false)
+    }
+    catch {
+        $failure = $_.Exception
+        while ($failure.InnerException) {
+            $failure = $failure.InnerException
+        }
+        if ($failure -is [System.ArgumentException]) {
+            # The key can disappear between snapshot inspection and deletion.
+            return
+        }
+
+        throw
+    }
+}
+
+<#
+    .SYNOPSIS
+        Returns a feature's registry-file path, using the Sysprep layout when targeting another profile.
+#>
+function Get-RegistryFilePathForFeature {
+    param(
+        [Parameter(Mandatory)]
+        [string]$RegistryKey,
+        [switch]$UseSysprepRegFiles
+    )
+
+    $useSysprepLayout = $UseSysprepRegFiles -or $script:Params.ContainsKey('Sysprep') -or $script:Params.ContainsKey('User')
+    if ($useSysprepLayout) {
+        return Join-Path (Join-Path $script:RegfilesPath 'Sysprep') $RegistryKey
+    }
+
+    return Join-Path $script:RegfilesPath $RegistryKey
+}
+