Extract function to remove available windows features

We suspect this sometimes triggers the need for a VM restart so we want to try running it before the final restart to avoid a situation where Dism.exe is unable to run successfully.
cloudfoundry · Sep 12, 2023 · 9787719 · 9787719
1 parent 54292a2
commit 9787719
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 4 deletions.
diff --git a/modules/BOSH.Disk/BOSH.Disk.psd1 b/modules/BOSH.Disk/BOSH.Disk.psd1
@@ -7,7 +7,7 @@ Copyright = '(c) 2017 BOSH'
 Description = 'Commands for Disk Utilities on a BOSH deployed vm'
 PowerShellVersion = '4.0'
 RequiredModules = @('BOSH.Utils')
-FunctionsToExport = @('Compress-Disk','Optimize-Disk')
+FunctionsToExport = @('Compress-Disk','Optimize-Disk', 'Remove-Available-Windows-Features')
 CmdletsToExport = @()
 VariablesToExport = '*'
 AliasesToExport = @()

diff --git a/modules/BOSH.Disk/BOSH.Disk.psm1 b/modules/BOSH.Disk/BOSH.Disk.psm1
@@ -16,11 +16,10 @@ function Compress-Disk {
 function Optimize-Disk {
     Write-Log "Starting to clean disk"
 
-    Get-WindowsFeature |
-    ? { $_.InstallState -eq 'Available' } |
-    Uninstall-WindowsFeature -Remove
+    Remove-Available-Windows-Features
 
     # Cleanup WinSxS folder: https://technet.microsoft.com/en-us/library/dn251565.aspx
+    # /LogLevel default is 3
     Write-Log "Running 'Dism.exe /online /LogLevel:4 /Cleanup-Image /StartComponentCleanup /ResetBase'"
     Dism.exe /online /LogLevel:4 /Cleanup-Image /StartComponentCleanup /ResetBase
     if ($LASTEXITCODE -ne 0) {
@@ -38,6 +37,16 @@ function Optimize-Disk {
     Write-Log "Finished clean disk"
 }
 
+function Remove-Available-Windows-Features {
+    Write-Log "Starting to remove 'Available' Windows Features"
+
+    Get-WindowsFeature |
+            ? { $_.InstallState -eq 'Available' } |
+            Uninstall-WindowsFeature -Remove
+
+    Write-Log "Finished removing 'Available' Windows Features"
+}
+
 function DefragDisk {
     # First - get the volumes via WMI
     $volumes = gwmi win32_volume

diff --git a/stemcell-automation/AutomationHelpers.Tests.ps1 b/stemcell-automation/AutomationHelpers.Tests.ps1
@@ -24,6 +24,7 @@ Describe "Setup" {
         Mock Install-WUCerts { $provisionerCalls.Add("Install-WUCerts") }
 
         Mock InstallCFFeatures { $provisionerCalls.Add("InstallCFFeatures") }
+        Mock Remove-Available-Windows-Features { }
         Mock Create-VersionFile { $provisionerCalls.Add("Create-VersionFile") }
         Mock Restart-Computer { $provisionerCalls.Add("Restart-Computer") }
 

diff --git a/stemcell-automation/AutomationHelpers.ps1 b/stemcell-automation/AutomationHelpers.ps1
@@ -36,6 +36,7 @@ function Setup()
     Install-SecurityPoliciesAndRegistries
     Enable-SSHD
     InstallCFFeatures
+    Remove-Available-Windows-Features
 
     try { Install-WUCerts }
     catch [Exception]