124 lines
9.3 KiB
PowerShell
124 lines
9.3 KiB
PowerShell
param([Parameter(Mandatory=$true)]$Settings,[string]$ResourceContextPath)
|
|
Import-Module (Join-Path $Settings.UpstreamRoot 'WinImageBuilder.psm1') -Force
|
|
Import-Module (Join-Path $Settings.UpstreamRoot 'Config.psm1') -Force
|
|
& (Get-Module WinImageBuilder) {
|
|
param($ProjectRoot,$AssetsRoot,$ManifestPath,$ResourceContextPath)
|
|
$script:PveProjectRoot=$ProjectRoot;$script:PveAssetsRoot=$AssetsRoot;$script:PveQemuManifest=$ManifestPath
|
|
$script:PveDeferredServicing=$false
|
|
$script:PveServicingIsoPath=$null
|
|
$script:PveResourceContextPath=$ResourceContextPath
|
|
$script:PveProductKeyMode='Legacy'
|
|
. (Join-Path $ProjectRoot 'scripts\Build.Resources.ps1')
|
|
. (Join-Path $ProjectRoot 'scripts\CloudbaseInit.Resources.ps1')
|
|
if (-not (Get-Variable -Name PveOriginalDownloadCloudbaseInit -Scope Script -ErrorAction SilentlyContinue)) { $script:PveOriginalDownloadCloudbaseInit=${function:Download-CloudbaseInit} }
|
|
function script:Download-CloudbaseInit {
|
|
param([string]$resourcesDir,[string]$osArch,[switch]$BetaRelease,[string]$MsiPath,[string]$CloudbaseInitConfigPath,[string]$CloudbaseInitUnattendedConfigPath)
|
|
$cachePath=Get-PveCloudbaseInitCachePath -AssetsRoot $script:PveAssetsRoot -OsArch $osArch -BetaRelease:$BetaRelease
|
|
$templatePath=Get-PveCloudbaseInitCachePath -AssetsRoot $script:PveAssetsRoot
|
|
if (-not $MsiPath -or [IO.Path]::GetFullPath($MsiPath) -in @($cachePath,$templatePath)) {
|
|
$PSBoundParameters['MsiPath']=Execute-Retry {
|
|
Initialize-PveCloudbaseInitInstaller -AssetsRoot $script:PveAssetsRoot -OsArch $osArch -BetaRelease:$BetaRelease
|
|
}
|
|
}
|
|
& $script:PveOriginalDownloadCloudbaseInit @PSBoundParameters
|
|
}
|
|
. (Join-Path $ProjectRoot 'scripts\Unattend.Adapter.ps1')
|
|
if (-not (Get-Variable -Name PveOriginalGenerateUnattend -Scope Script -ErrorAction SilentlyContinue)) { $script:PveOriginalGenerateUnattend=${function:Generate-UnattendXml} }
|
|
function script:Generate-UnattendXml {
|
|
param([string]$inUnattendXmlPath,[string]$outUnattendXmlPath,$image,[string]$productKey,$administratorPassword)
|
|
$setupKey=Get-PveSetupProductKey -Image $image -RequestedProductKey $productKey -Mode $script:PveProductKeyMode
|
|
$automatic=($script:PveProductKeyMode -eq 'Kms' -or (-not $productKey -and -not [string]::IsNullOrWhiteSpace($setupKey)))
|
|
if ($setupKey) { $PSBoundParameters['productKey']=$setupKey }
|
|
else { $PSBoundParameters.Remove('productKey') | Out-Null }
|
|
& $script:PveOriginalGenerateUnattend @PSBoundParameters
|
|
Update-PveUnattendSettings -Path $outUnattendXmlPath -Image $image -AutomaticSetupKey:$automatic
|
|
if ($automatic) { Write-Log 'Public KMS client setup key supplied for the selected edition; build-time automatic activation disabled.' }
|
|
}
|
|
. (Join-Path $ProjectRoot 'scripts\Servicing.Adapter.ps1')
|
|
if (-not (Get-Variable -Name PveOriginalGetPath -Scope Script -ErrorAction SilentlyContinue)) { $script:PveOriginalGetPath = ${function:Get-PathWithoutExtension} }
|
|
function script:Get-PathWithoutExtension {
|
|
param([string]$Path,[int]$Depth=0)
|
|
if ([IO.Path]::GetExtension($Path).ToLowerInvariant() -in @('.qcow2','.vhdx','.vhd','.raw')) {
|
|
return Join-Path ([IO.Path]::GetDirectoryName($Path)) ([IO.Path]::GetFileNameWithoutExtension($Path))
|
|
}
|
|
& $script:PveOriginalGetPath -Path $Path -Depth $Depth
|
|
}
|
|
if (-not (Get-Variable -Name PveOriginalRunSysprep -Scope Script -ErrorAction SilentlyContinue)) {
|
|
$script:PveOriginalRunSysprep = ${function:Run-Sysprep}
|
|
}
|
|
if (-not (Get-Variable -Name PveOriginalCopyUnattendResources -Scope Script -ErrorAction SilentlyContinue)) { $script:PveOriginalCopyUnattendResources=${function:Copy-UnattendResources} }
|
|
function script:Copy-UnattendResources {
|
|
param([string]$resourcesDir,[string]$imageInstallationType,[bool]$InstallMaaSHooks,[string]$VMwareToolsPath)
|
|
& $script:PveOriginalCopyUnattendResources @PSBoundParameters
|
|
Copy-Item -LiteralPath (Join-Path $script:PveProjectRoot 'resources\Builder.Specialize.ps1') -Destination (Join-Path $resourcesDir 'Specialize.ps1') -Force
|
|
}
|
|
if (-not (Get-Variable -Name PveOriginalAddVirtioIso -Scope Script -ErrorAction SilentlyContinue)) { $script:PveOriginalAddVirtioIso=${function:Add-VirtIODriversFromISO} }
|
|
function script:Add-VirtIODriversFromISO {
|
|
param([string]$vhdDriveLetter,$image,[string]$isoPath)
|
|
if ($script:PveResourceContextPath) {
|
|
$directory=Join-Path $script:PveRunLogDirectory ('temporary-virtio-'+[guid]::NewGuid().ToString('N'))
|
|
Register-PveBuildDirectory -ContextPath $script:PveResourceContextPath -Path $directory
|
|
New-Item -ItemType Directory -Path $directory | Out-Null
|
|
$copy=Join-Path $directory 'virtio.iso'
|
|
Copy-Item -LiteralPath $isoPath -Destination $copy
|
|
try { & $script:PveOriginalAddVirtioIso -vhdDriveLetter $vhdDriveLetter -image $image -isoPath $copy }
|
|
finally {
|
|
if (Test-Path -LiteralPath $copy -PathType Leaf) { Assert-PveResourcePath -Path $copy -Root $directory -Descendant | Out-Null; Remove-Item -LiteralPath $copy -Force }
|
|
if ((Test-Path -LiteralPath $directory -PathType Container) -and -not @(Get-ChildItem -LiteralPath $directory -Force).Count) { Remove-Item -LiteralPath $directory }
|
|
}
|
|
} else { & $script:PveOriginalAddVirtioIso @PSBoundParameters }
|
|
}
|
|
function script:Run-Sysprep {
|
|
param([string]$Name,[string]$VhdPath,[uint64]$Memory,[int]$CpuCores,[string]$VMSwitch,[string]$Generation='1',[switch]$DisableSecureBoot)
|
|
if($script:PveDeferredServicing){
|
|
. (Join-Path $script:PveProjectRoot 'scripts\WindowsPE.Servicing.ps1')
|
|
Write-Log 'Starting isolated Windows PE EFI configuration and driver injection.'
|
|
Invoke-WindowsPeServicing -VhdPath $VhdPath -WindowsIsoPath $script:PveServicingIsoPath -LogDirectory (Join-Path $script:PveRunLogDirectory 'windows-pe')
|
|
}
|
|
if ($script:PveResourceContextPath) {
|
|
$Name='PveBuilder-Sysprep-'+[guid]::NewGuid().ToString('N')
|
|
$PSBoundParameters['Name']=$Name
|
|
Register-PveBuildVm -ContextPath $script:PveResourceContextPath -Name $Name -VhdPath $VhdPath
|
|
}
|
|
try { & $script:PveOriginalRunSysprep @PSBoundParameters }
|
|
catch {
|
|
$own=Get-VM -Name $Name -ErrorAction SilentlyContinue
|
|
if($own){$disks=@(Get-VMHardDiskDrive -VM $own);if($disks.Count -eq 1 -and $disks[0].Path -ieq $VhdPath -and $own.State -ne 'Off'){Stop-VM -VM $own -TurnOff -Force}}
|
|
throw
|
|
}
|
|
. (Join-Path $script:PveProjectRoot 'scripts\Build.Validation.ps1')
|
|
$verification=Get-OfflineBuildVerification -VhdPath $VhdPath -Directory (Join-Path $script:PveRunLogDirectory 'verification')
|
|
[IO.File]::WriteAllText((Join-Path $script:PveRunLogDirectory 'verified-guest.json'),($verification|ConvertTo-Json -Depth 6),[Text.UTF8Encoding]::new($false))
|
|
Write-Log 'Guest software, required services, and generalized Sysprep state verified.'
|
|
}
|
|
function script:Wait-ForVMShutdown {
|
|
param([string]$Name)
|
|
$deadline=[DateTime]::UtcNow.AddHours(4)
|
|
$lastMessages=@{}
|
|
if ($script:PveResourceContextPath) {
|
|
$owned=Get-VM -Name $Name -ErrorAction Stop
|
|
$ownedDisks=@(Get-VMHardDiskDrive -VM $owned -ErrorAction Stop)
|
|
if ($ownedDisks.Count -ne 1) { throw 'The build VM must have exactly one owned disk.' }
|
|
Register-PveBuildVm -ContextPath $script:PveResourceContextPath -Name $Name -VhdPath $ownedDisks[0].Path -Id ([string]$owned.Id)
|
|
}
|
|
while ((Get-VM -Name $Name).State -ne 'Off') {
|
|
if([DateTime]::UtcNow -gt $deadline){throw ('Build VM timed out; retained for diagnosis: '+$Name)}
|
|
$messages=$null
|
|
try { $messages=Get-KVPData -VMName $Name } catch { Write-Verbose 'VM runtime logs are not available yet.' }
|
|
if($messages){foreach($key in $messages.Keys){if($lastMessages[$key] -ne $messages[$key]){Write-Log ('{0}: {1}' -f $key,$messages[$key])};if($key -ieq 'ERROR'){throw ('Build guest reported an error; VM retained for diagnosis: '+$Name+'. '+$messages[$key])}};$lastMessages=$messages}
|
|
Start-Sleep -Seconds 3
|
|
}
|
|
}
|
|
function script:Convert-VirtualDisk {
|
|
param([Parameter(Mandatory=$true)][string]$vhdPath,[Parameter(Mandatory=$true)][string]$outPath,[Parameter(Mandatory=$true)][string]$format,[bool]$CompressQcow2)
|
|
if($format.ToLower() -eq 'qcow2' -and $CompressQcow2){
|
|
$result=& (Join-Path $script:PveProjectRoot 'src\Compress-Qcow2Image.ps1') -SourcePath $vhdPath -OutputPath $outPath -ToolManifestPath $script:PveQemuManifest -BackupDirectory (Join-Path $script:PveAssetsRoot 'archive\verified-builds') -ReplaceExisting -ResourceContextPath $script:PveResourceContextPath
|
|
Write-Log ('Verified compressed image: '+$result.OutputPath)
|
|
}else{
|
|
$manifest=Get-Content -Raw -Encoding UTF8 -LiteralPath $script:PveQemuManifest|ConvertFrom-Json
|
|
& $manifest.ExecutablePath convert -m 1 -O $format $vhdPath $outPath
|
|
if($LASTEXITCODE -ne 0){throw 'Disk conversion failed.'}
|
|
}
|
|
}
|
|
} $Settings.ProjectRoot $Settings.AssetsRoot $Settings.QemuManifestPath $ResourceContextPath
|