add retry logic to install-vsb.ps1

Author: justice-adams-apple

.github/workflows/scripts/windows/install-vsb.ps1

@@ -9,35 +9,147 @@
 ## See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
 ##
 ##===----------------------------------------------------------------------===##
-$VSB='https://download.visualstudio.microsoft.com/download/pr/5536698c-711c-4834-876f-2817d31a2ef2/c792bdb0fd46155de19955269cac85d52c4c63c23db2cf43d96b9390146f9390/vs_BuildTools.exe'
-$VSB_SHA256='C792BDB0FD46155DE19955269CAC85D52C4C63C23DB2CF43D96B9390146F9390'
-Set-Variable ErrorActionPreference Stop
-Set-Variable ProgressPreference SilentlyContinue
-Write-Host -NoNewLine ('Downloading {0} ... ' -f ${VSB})
-Invoke-WebRequest -Uri $VSB -OutFile $env:TEMP\vs_buildtools.exe
-Write-Host 'SUCCESS'
-Write-Host -NoNewLine ('Verifying SHA256 ({0}) ... ' -f $VSB_SHA256)
-$Hash = Get-FileHash $env:TEMP\vs_buildtools.exe -Algorithm sha256
-if ($Hash.Hash -eq $VSB_SHA256) {
-    Write-Host 'SUCCESS'
-} else {
-    Write-Host  ('FAILED ({0})' -f $Hash.Hash)
-    exit 1
+
+# Retry configuration
+$MaxRetries = 5
+$RetryDelay = 5
+
+function Invoke-WebRequestWithRetry {
+    param (
+        [string]$Uri,
+        [string]$OutFile,
+        [int]$TimeoutSec = 300
+    )
+
+    $attempt = 1
+
+    while ($attempt -le $MaxRetries) {
+        try {
+            if ($attempt -gt 1) {
+                Write-Host "Retry attempt $attempt of $MaxRetries after ${RetryDelay}s delay..."
+                Start-Sleep -Seconds $RetryDelay
+            }
+
+            Write-Host "Attempt $attempt`: Downloading from $Uri"
+
+            # Clean up any existing partial download
+            if (Test-Path $OutFile) {
+                Remove-Item -Force $OutFile -ErrorAction SilentlyContinue
+            }
+
+            # Get expected file size from HTTP headers
+            $headRequest = Invoke-WebRequest -Uri $Uri -Method Head -UseBasicParsing -TimeoutSec 30
+            $expectedSize = $null
+            if ($headRequest.Headers.ContainsKey('Content-Length')) {
+                $expectedSize = [long]$headRequest.Headers['Content-Length'][0]
+                Write-Host "Expected file size: $([math]::Round($expectedSize / 1MB, 2)) MB"
+            }
+
+            # Download with progress tracking disabled for better performance
+            $webClient = New-Object System.Net.WebClient
+            $webClient.DownloadFile($Uri, $OutFile)
+            $webClient.Dispose()
+
+            # Verify file exists and has content
+            if (-not (Test-Path $OutFile)) {
+                throw "Download completed but file not found at $OutFile"
+            }
+
+            $actualSize = (Get-Item $OutFile).Length
+            Write-Host "Downloaded file size: $([math]::Round($actualSize / 1MB, 2)) MB"
+
+            # Verify file size matches expected size
+            if ($expectedSize -and $actualSize -ne $expectedSize) {
+                throw "File size mismatch. Expected: $expectedSize bytes, Got: $actualSize bytes"
+            }
+
+            # Verify file is not corrupted by checking if it's a valid PE executable
+            $fileBytes = [System.IO.File]::ReadAllBytes($OutFile)
+            if ($fileBytes.Length -lt 2 -or $fileBytes[0] -ne 0x4D -or $fileBytes[1] -ne 0x5A) {
+                throw "Downloaded file is not a valid executable (missing MZ header)"
+            }
+
+            Write-Host "Download completed and verified successfully"
+            return $true
+        }
+        catch {
+            Write-Host "Download failed on attempt $attempt`: $($_.Exception.Message)"
+
+            # Clean up partial download if it exists
+            if (Test-Path $OutFile) {
+                Remove-Item -Force $OutFile -ErrorAction SilentlyContinue
+            }
+
+            if ($attempt -eq $MaxRetries) {
+                Write-Host "Download failed after $MaxRetries attempts"
+                throw
+            }
+        }
+
+        $attempt++
+    }
+
+    return $false
 }
-Write-Host -NoNewLine 'Installing Visual Studio Build Tools ... '
-$Process =
-    Start-Process $env:TEMP\vs_buildtools.exe -Wait -PassThru -NoNewWindow -ArgumentList @(
-        '--quiet',
-        '--wait',
-        '--norestart',
-        '--nocache',
-        '--add', 'Microsoft.VisualStudio.Component.Windows11SDK.22000',
-        '--add', 'Microsoft.VisualStudio.Component.VC.Tools.x86.x64'
+
+function Install-VisualStudioBuildTools {
+    param (
+        [string]$Url,
+        [string]$Sha256
     )
-if ($Process.ExitCode -eq 0 -or $Process.ExitCode -eq 3010) {
-    Write-Host 'SUCCESS'
-} else {
-    Write-Host  ('FAILED ({0})' -f $Process.ExitCode)
-    exit 1
+
+    Set-Variable ErrorActionPreference Stop
+    Set-Variable ProgressPreference SilentlyContinue
+
+    $installerPath = "$env:TEMP\vs_buildtools.exe"
+
+    Write-Host "Downloading Visual Studio Build Tools from $Url"
+
+    try {
+        Invoke-WebRequestWithRetry -Uri $Url -OutFile $installerPath
+        Write-Host 'Download SUCCESS'
+    }
+    catch {
+        Write-Host "Download FAILED: $($_.Exception.Message)"
+        exit 1
+    }
+
+    Write-Host -NoNewLine ('Verifying SHA256 ({0}) ... ' -f $Sha256)
+    $Hash = Get-FileHash $installerPath -Algorithm sha256
+    if ($Hash.Hash -eq $VSB_SHA256) {
+        Write-Host 'SUCCESS'
+    } else {
+        Write-Host  ('FAILED ({0})' -f $Hash.Hash)
+        exit 1
+    }
+
+    Write-Host -NoNewLine 'Installing Visual Studio Build Tools ... '
+    try {
+        $Process = Start-Process $installerPath -Wait -PassThru -NoNewWindow -ArgumentList @(
+            '--quiet',
+            '--wait',
+            '--norestart',
+            '--nocache',
+            '--add', 'Microsoft.VisualStudio.Component.Windows11SDK.22000',
+            '--add', 'Microsoft.VisualStudio.Component.VC.Tools.x86.x64'
+        )
+        if ($Process.ExitCode -eq 0 -or $Process.ExitCode -eq 3010) {
+            Write-Host 'SUCCESS'
+        } else {
+            Write-Host  ('FAILED ({0})' -f $Process.ExitCode)
+            exit 1
+        }
+    }
+    catch {
+        Write-Host "FAILED: $($_.Exception.Message)"
+        Remove-Item -Force $installerPath -ErrorAction SilentlyContinue
+        exit 1
+    }
+
+    Remove-Item -Force $installerPath -ErrorAction SilentlyContinue
 }
-Remove-Item -Force $env:TEMP\vs_buildtools.exe
+
+$VSB = 'https://download.visualstudio.microsoft.com/download/pr/5536698c-711c-4834-876f-2817d31a2ef2/c792bdb0fd46155de19955269cac85d52c4c63c23db2cf43d96b9390146f9390/vs_BuildTools.exe'
+$VSB_SHA256 = 'C792BDB0FD46155DE19955269CAC85D52C4C63C23DB2CF43D96B9390146F9390'
+
+Install-VisualStudioBuildTools -Url $VSB -Sha256 $VSB_SHA256