From 5e69d9c205ed456125b67b18e4c5ac9f62a79b02 Mon Sep 17 00:00:00 2001
From: Alexey-Ayupov <116575425+Alexey-Ayupov@users.noreply.github.com>
Date: Mon, 30 Dec 2024 11:11:22 +0100
Subject: [PATCH] [Windows] Add WSL2 to Windows-2025 image (#11242)

---
 .../windows/scripts/build/Configure-System.ps1   |  4 ++--
 images/windows/scripts/build/Install-WSL2.ps1    | 16 ++++++++++++++++
 .../scripts/docs-gen/Generate-SoftwareReport.ps1 |  3 +++
 .../scripts/docs-gen/SoftwareReport.Tools.psm1   |  4 ++++
 .../scripts/tests/WindowsFeatures.Tests.ps1      |  6 ++++++
 images/windows/templates/windows-2025.pkr.hcl    |  3 ++-
 6 files changed, 33 insertions(+), 3 deletions(-)
 create mode 100644 images/windows/scripts/build/Install-WSL2.ps1

diff --git a/images/windows/scripts/build/Configure-System.ps1 b/images/windows/scripts/build/Configure-System.ps1
index 1291b7b47..5e144f2d5 100644
--- a/images/windows/scripts/build/Configure-System.ps1
+++ b/images/windows/scripts/build/Configure-System.ps1
@@ -93,11 +93,11 @@ $servicesToDisable = @(
     'wuauserv'
     'DiagTrack'
     'dmwappushservice'
-    'PcaSvc'
+    $(if(-not (Test-IsWin25)){'PcaSvc'})
     'SysMain'
     'gupdate'
     'gupdatem'
-    'StorSvc'
+    $(if(-not (Test-IsWin25)){'StorSvc'})
 ) | Get-Service -ErrorAction SilentlyContinue
 Stop-Service $servicesToDisable
 $servicesToDisable.WaitForStatus('Stopped', "00:01:00")
diff --git a/images/windows/scripts/build/Install-WSL2.ps1 b/images/windows/scripts/build/Install-WSL2.ps1
new file mode 100644
index 000000000..4134b2afd
--- /dev/null
+++ b/images/windows/scripts/build/Install-WSL2.ps1
@@ -0,0 +1,16 @@
+Write-Host "Install WSL2"
+
+$version = (Get-GithubReleasesByVersion -Repo "microsoft/WSL" -Version "latest").version
+$downloadUrl =  Resolve-GithubReleaseAssetUrl `
+    -Repo "microsoft/WSL" `
+    -Version $version `
+    -UrlMatchPattern "wsl.*.x64.msi"
+
+Install-Binary -Type MSI `
+    -Url $downloadUrl `
+    -ExpectedSHA256Sum "CD3F2A68A1A5836F6A1CC9965A7F5F54DB267CA221EAA87DF29345AB7957AEC4"
+
+Write-Host "Performing wsl --install --no-distribution"
+wsl.exe --install --no-distribution
+
+Invoke-PesterTests -TestFile "WindowsFeatures" -TestName "WSL2"
diff --git a/images/windows/scripts/docs-gen/Generate-SoftwareReport.ps1 b/images/windows/scripts/docs-gen/Generate-SoftwareReport.ps1
index 7077a39f3..3cc3ee470 100644
--- a/images/windows/scripts/docs-gen/Generate-SoftwareReport.ps1
+++ b/images/windows/scripts/docs-gen/Generate-SoftwareReport.ps1
@@ -21,6 +21,9 @@ Import-Module (Join-Path $PSScriptRoot "SoftwareReport.VisualStudio.psm1") -Disa
 $softwareReport = [SoftwareReport]::new($(Build-OSInfoSection))
 $optionalFeatures = $softwareReport.Root.AddHeader("Windows features")
 $optionalFeatures.AddToolVersion("Windows Subsystem for Linux (WSLv1):", "Enabled")
+if (Test-IsWin25) {
+    $optionalFeatures.AddToolVersion("Windows Subsystem for Linux (Default, WSLv2):", $(Get-WSL2Version))
+}
 $installedSoftware = $softwareReport.Root.AddHeader("Installed Software")
 
 # Language and Runtime
diff --git a/images/windows/scripts/docs-gen/SoftwareReport.Tools.psm1 b/images/windows/scripts/docs-gen/SoftwareReport.Tools.psm1
index bd83432c0..c4672f82d 100644
--- a/images/windows/scripts/docs-gen/SoftwareReport.Tools.psm1
+++ b/images/windows/scripts/docs-gen/SoftwareReport.Tools.psm1
@@ -318,3 +318,7 @@ function Get-ImageMagickVersion {
 function Get-MongoshVersion {
     return $(mongosh --version)
 }
+
+function Get-WSL2Version {
+    return $((Get-AppxPackage -Name "MicrosoftCorporationII.WindowsSubsystemForLinux").version)
+}
diff --git a/images/windows/scripts/tests/WindowsFeatures.Tests.ps1 b/images/windows/scripts/tests/WindowsFeatures.Tests.ps1
index 03b56a210..09e682498 100644
--- a/images/windows/scripts/tests/WindowsFeatures.Tests.ps1
+++ b/images/windows/scripts/tests/WindowsFeatures.Tests.ps1
@@ -82,3 +82,9 @@ Describe "Windows Updates" {
         $State | Should -BeIn $expect
     }
 }
+
+Describe "WSL2" {
+    It "WSL status should return zero exit code" {
+        "wsl --status" | Should -ReturnZeroExitCode
+    }
+}
diff --git a/images/windows/templates/windows-2025.pkr.hcl b/images/windows/templates/windows-2025.pkr.hcl
index a4ba4ed9e..f1f3fdc55 100644
--- a/images/windows/templates/windows-2025.pkr.hcl
+++ b/images/windows/templates/windows-2025.pkr.hcl
@@ -243,7 +243,7 @@ build {
     inline = ["if (-not ((net localgroup Administrators) -contains '${var.install_user}')) { exit 1 }"]
   }
 
-  provisioner "powershell" {
+provisioner "powershell" {
     elevated_password = "${var.install_password}"
     elevated_user     = "${var.install_user}"
     inline            = ["bcdedit.exe /set TESTSIGNING ON"]
@@ -256,6 +256,7 @@ build {
       "${path.root}/../scripts/build/Configure-WindowsDefender.ps1",
       "${path.root}/../scripts/build/Configure-PowerShell.ps1",
       "${path.root}/../scripts/build/Install-PowerShellModules.ps1",
+      "${path.root}/../scripts/build/Install-WSL2.ps1",
       "${path.root}/../scripts/build/Install-WindowsFeatures.ps1",
       "${path.root}/../scripts/build/Install-Chocolatey.ps1",
       "${path.root}/../scripts/build/Configure-BaseImage.ps1",