mirror of
https://github.com/microsoft/vcpkg.git
synced 2024-11-28 23:48:59 +08:00
Introduce VcpkgPowershellUtils
This commit is contained in:
Alexander Karatarakis
parent
7e3dcc4f09
commit
2abdcc1eec
185
scripts/VcpkgPowershellUtils.ps1
Normal file
185
scripts/VcpkgPowershellUtils.ps1
Normal file
@ -0,0 +1,185 @@
|
||||
function vcpkgHasModule([Parameter(Mandatory=$true)][string]$moduleName)
|
||||
{
|
||||
return [bool](Get-Module -ListAvailable -Name $moduleName)
|
||||
}
|
||||
|
||||
function vcpkgCreateDirectory([Parameter(Mandatory=$true)][string]$dirPath)
|
||||
{
|
||||
if (!(Test-Path $dirPath))
|
||||
{
|
||||
New-Item -ItemType Directory -Path $dirPath | Out-Null
|
||||
}
|
||||
}
|
||||
|
||||
function vcpkgRemoveDirectory([Parameter(Mandatory=$true)][string]$dirPath)
|
||||
{
|
||||
if (Test-Path $dirPath)
|
||||
{
|
||||
Remove-Item $dirPath -Recurse -Force
|
||||
}
|
||||
}
|
||||
|
||||
function vcpkgRemoveFile([Parameter(Mandatory=$true)][string]$filePath)
|
||||
{
|
||||
if (Test-Path $filePath)
|
||||
{
|
||||
Remove-Item $filePath -Force
|
||||
}
|
||||
}
|
||||
|
||||
function vcpkgHasCommand([Parameter(Mandatory=$true)][string]$commandName)
|
||||
{
|
||||
return [bool](Get-Command -Name $commandName -ErrorAction SilentlyContinue)
|
||||
}
|
||||
|
||||
function vcpkgHasCommandParameter([Parameter(Mandatory=$true)][string]$commandName, [Parameter(Mandatory=$true)][string]$parameterName)
|
||||
{
|
||||
return (Get-Command $commandName).Parameters.Keys -contains $parameterName
|
||||
}
|
||||
|
||||
function vcpkgGetCredentials()
|
||||
{
|
||||
if (vcpkgHasCommandParameter -commandName 'Get-Credential' -parameterName 'Message')
|
||||
{
|
||||
return Get-Credential -Message "Enter credentials for Proxy Authentication"
|
||||
}
|
||||
else
|
||||
{
|
||||
Write-Host "Enter credentials for Proxy Authentication"
|
||||
return Get-Credential
|
||||
}
|
||||
}
|
||||
|
||||
function vcpkgGetSHA256([Parameter(Mandatory=$true)][string]$filePath)
|
||||
{
|
||||
if (vcpkgHasCommand -commandName 'Microsoft.PowerShell.Utility\Get-FileHash')
|
||||
{
|
||||
Write-Verbose("Hashing with Microsoft.PowerShell.Utility\Get-FileHash")
|
||||
$hash = (Microsoft.PowerShell.Utility\Get-FileHash -Path $filePath -Algorithm SHA256).Hash
|
||||
}
|
||||
elseif(vcpkgHasCommand -commandName 'Pscx\Get-Hash')
|
||||
{
|
||||
Write-Verbose("Hashing with Pscx\Get-Hash")
|
||||
$hash = (Pscx\Get-Hash -Path $filePath -Algorithm SHA256).HashString
|
||||
}
|
||||
else
|
||||
{
|
||||
Write-Verbose("Hashing with .NET")
|
||||
$hashAlgorithm = [Security.Cryptography.HashAlgorithm]::Create("SHA256")
|
||||
$fileAsByteArray = [io.File]::ReadAllBytes($filePath)
|
||||
$hashByteArray = $hashAlgorithm.ComputeHash($fileAsByteArray)
|
||||
$hash = -Join ($hashByteArray | ForEach-Object {"{0:x2}" -f $_})
|
||||
}
|
||||
|
||||
return $hash.ToLower()
|
||||
}
|
||||
|
||||
function vcpkgCheckEqualFileHash( [Parameter(Mandatory=$true)][string]$filePath,
|
||||
[Parameter(Mandatory=$true)][string]$expectedHash,
|
||||
[Parameter(Mandatory=$true)][string]$actualHash )
|
||||
{
|
||||
if ($expectedDownloadedFileHash -ne $downloadedFileHash)
|
||||
{
|
||||
Write-Host ("`nFile does not have expected hash:`n" +
|
||||
" File path: [ $filePath ]`n" +
|
||||
" Expected hash: [ $expectedHash ]`n" +
|
||||
" Actual hash: [ $actualHash ]`n")
|
||||
throw "Invalid Hash for file $filePath"
|
||||
}
|
||||
}
|
||||
|
||||
if (vcpkgHasModule -moduleName 'BitsTransfer')
|
||||
{
|
||||
Import-Module BitsTransfer -Verbose:$false
|
||||
}
|
||||
|
||||
function vcpkgDownloadFile( [Parameter(Mandatory=$true)][string]$url,
|
||||
[Parameter(Mandatory=$true)][string]$downloadPath)
|
||||
{
|
||||
if (Test-Path $downloadPath)
|
||||
{
|
||||
return
|
||||
}
|
||||
|
||||
$downloadDir = split-path -parent $downloadPath
|
||||
vcpkgCreateDirectory $downloadDir
|
||||
|
||||
$downloadPartPath = "$downloadPath.part"
|
||||
vcpkgRemoveFile $downloadPartPath
|
||||
|
||||
$wc = New-Object System.Net.WebClient
|
||||
$proxyAuth = !$wc.Proxy.IsBypassed($url)
|
||||
if ($proxyAuth)
|
||||
{
|
||||
$wc.Proxy.Credentials = vcpkgGetCredentials
|
||||
}
|
||||
|
||||
# Some download (e.g. git from github)fail with Start-BitsTransfer
|
||||
if (vcpkgHasCommand -commandName 'Start-BitsTransfer')
|
||||
{
|
||||
try
|
||||
{
|
||||
if ($proxyAuth)
|
||||
{
|
||||
$PSDefaultParameterValues.Add("Start-BitsTransfer:ProxyAuthentication","Basic")
|
||||
$PSDefaultParameterValues.Add("Start-BitsTransfer:ProxyCredential", $wc.Proxy.Credentials)
|
||||
}
|
||||
Start-BitsTransfer -Source $url -Destination $downloadPartPath -ErrorAction Stop
|
||||
Move-Item -Path $downloadPartPath -Destination $downloadPath
|
||||
return
|
||||
}
|
||||
catch [System.Exception]
|
||||
{
|
||||
# If BITS fails for any reason, delete any potentially partially downloaded files and continue
|
||||
vcpkgRemoveFile $downloadPartPath
|
||||
}
|
||||
}
|
||||
|
||||
Write-Verbose("Downloading $Dependency...")
|
||||
$wc.DownloadFile($url, $downloadPartPath)
|
||||
Move-Item -Path $downloadPartPath -Destination $downloadPath
|
||||
}
|
||||
|
||||
function vcpkgExtractFile( [Parameter(Mandatory=$true)][string]$file,
|
||||
[Parameter(Mandatory=$true)][string]$destination)
|
||||
{
|
||||
vcpkgCreateDirectory $destination
|
||||
|
||||
if (vcpkgHasCommand -commandName 'Microsoft.PowerShell.Archive\Expand-Archive')
|
||||
{
|
||||
Write-Verbose("Extracting with Microsoft.PowerShell.Archive\Expand-Archive")
|
||||
Microsoft.PowerShell.Archive\Expand-Archive -path $file -destinationpath $destination
|
||||
}
|
||||
elseif (vcpkgHasCommand -commandName 'Pscx\Expand-Archive')
|
||||
{
|
||||
Write-Verbose("Extracting with Pscx\Expand-Archive")
|
||||
Pscx\Expand-Archive -path $file -OutputPath $destination
|
||||
}
|
||||
else
|
||||
{
|
||||
Write-Verbose("Extracting via shell")
|
||||
$shell = new-object -com shell.application
|
||||
$zip = $shell.NameSpace($file)
|
||||
foreach($item in $zip.items())
|
||||
{
|
||||
# Piping to Out-Null is used to block until finished
|
||||
$shell.Namespace($destination).copyhere($item) | Out-Null
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function vcpkgInvokeCommand()
|
||||
{
|
||||
param ( [Parameter(Mandatory=$true)][string]$executable,
|
||||
[string]$arguments = "",
|
||||
[switch]$wait)
|
||||
|
||||
Write-Verbose "Executing: ${executable} ${arguments}"
|
||||
$process = Start-Process -FilePath $executable -ArgumentList $arguments -PassThru
|
||||
if ($wait)
|
||||
{
|
||||
Wait-Process -InputObject $process
|
||||
$ec = $process.ExitCode
|
||||
Write-Verbose "Execution terminated with exit code $ec."
|
||||
}
|
||||
}
|
||||
@ -3,192 +3,21 @@ param(
|
||||
[string]$Dependency
|
||||
)
|
||||
|
||||
function Test-Command($commandName)
|
||||
{
|
||||
return [bool](Get-Command -Name $commandName -ErrorAction SilentlyContinue)
|
||||
}
|
||||
|
||||
function Test-CommandParameter($commandName, $parameterName)
|
||||
{
|
||||
return (Get-Command $commandName).Parameters.Keys -contains $parameterName
|
||||
}
|
||||
|
||||
function Test-Module($moduleName)
|
||||
{
|
||||
return [bool](Get-Module -ListAvailable -Name $moduleName)
|
||||
}
|
||||
|
||||
function Get-Credential-Backwards-Compatible()
|
||||
{
|
||||
if (Test-CommandParameter -commandName 'Get-Credential' -parameterName 'Message')
|
||||
{
|
||||
return Get-Credential -Message "Enter credentials for Proxy Authentication"
|
||||
}
|
||||
else
|
||||
{
|
||||
Write-Host "Enter credentials for Proxy Authentication"
|
||||
return Get-Credential
|
||||
}
|
||||
}
|
||||
|
||||
function Get-Hash-SHA256()
|
||||
{
|
||||
if (Test-Command -commandName 'Microsoft.PowerShell.Utility\Get-FileHash')
|
||||
{
|
||||
Write-Verbose("Hashing with Microsoft.PowerShell.Utility\Get-FileHash")
|
||||
$downloadedFileHash = (Get-FileHash -Path $downloadPath -Algorithm SHA256).Hash
|
||||
}
|
||||
elseif(Test-Command -commandName 'Pscx\Get-Hash')
|
||||
{
|
||||
Write-Verbose("Hashing with Pscx\Get-Hash")
|
||||
$downloadedFileHash = (Get-Hash -Path $downloadPath -Algorithm SHA256).HashString
|
||||
}
|
||||
else
|
||||
{
|
||||
Write-Verbose("Hashing with .NET")
|
||||
$hashAlgorithm = [Security.Cryptography.HashAlgorithm]::Create("SHA256")
|
||||
$fileAsByteArray = [io.File]::ReadAllBytes($downloadPath)
|
||||
$hashByteArray = $hashAlgorithm.ComputeHash($fileAsByteArray)
|
||||
$downloadedFileHash = -Join ($hashByteArray | ForEach-Object {"{0:x2}" -f $_})
|
||||
}
|
||||
|
||||
return $downloadedFileHash.ToLower()
|
||||
}
|
||||
|
||||
if (Test-Module -moduleName 'BitsTransfer')
|
||||
{
|
||||
Import-Module BitsTransfer -Verbose:$false
|
||||
}
|
||||
$scriptsDir = split-path -parent $MyInvocation.MyCommand.Definition
|
||||
. "$scriptsDir\VcpkgPowershellUtils.ps1"
|
||||
|
||||
Write-Verbose "Fetching dependency: $Dependency"
|
||||
|
||||
$scriptsDir = split-path -parent $MyInvocation.MyCommand.Definition
|
||||
$vcpkgRootDir = & $scriptsDir\findFileRecursivelyUp.ps1 $scriptsDir .vcpkg-root
|
||||
|
||||
$downloadsDir = "$vcpkgRootDir\downloads"
|
||||
|
||||
function SelectProgram([Parameter(Mandatory=$true)][string]$Dependency)
|
||||
{
|
||||
function performDownload( [Parameter(Mandatory=$true)][string]$Dependency,
|
||||
[Parameter(Mandatory=$true)][string]$url,
|
||||
[Parameter(Mandatory=$true)][string]$downloadDir,
|
||||
[Parameter(Mandatory=$true)][string]$downloadPath,
|
||||
[Parameter(Mandatory=$true)][string]$downloadVersion,
|
||||
[Parameter(Mandatory=$true)][string]$requiredVersion)
|
||||
{
|
||||
if (Test-Path $downloadPath)
|
||||
{
|
||||
return
|
||||
}
|
||||
|
||||
if (!(Test-Path $downloadDir))
|
||||
{
|
||||
New-Item -ItemType directory -Path $downloadDir | Out-Null
|
||||
}
|
||||
|
||||
$downloadsTemp = "$downloadDir/temp"
|
||||
if (Test-Path $downloadsTemp) # Delete temp dir if it exists
|
||||
{
|
||||
Remove-Item $downloadsTemp -Recurse -Force
|
||||
}
|
||||
if (!(Test-Path $downloadsTemp)) # Recreate temp dir. It may still be there the dir was in use
|
||||
{
|
||||
New-Item -ItemType directory -Path $downloadsTemp | Out-Null
|
||||
}
|
||||
|
||||
$tempDownloadName = "$downloadsTemp/$Dependency-$downloadVersion.temp"
|
||||
|
||||
$WC = New-Object System.Net.WebClient
|
||||
$ProxyAuth = !$WC.Proxy.IsBypassed($url)
|
||||
|
||||
# git and installerbase fail with Start-BitsTransfer
|
||||
if ((Test-Command -commandName 'Start-BitsTransfer') -and ($Dependency -ne "git")-and ($Dependency -ne "installerbase"))
|
||||
{
|
||||
try
|
||||
{
|
||||
if ($ProxyAuth)
|
||||
{
|
||||
$ProxyCred = Get-Credential-Backwards-Compatible
|
||||
$PSDefaultParameterValues.Add("Start-BitsTransfer:ProxyAuthentication","Basic")
|
||||
$PSDefaultParameterValues.Add("Start-BitsTransfer:ProxyCredential", $ProxyCred)
|
||||
}
|
||||
Start-BitsTransfer -Source $url -Destination $tempDownloadName -ErrorAction Stop
|
||||
Move-Item -Path $tempDownloadName -Destination $downloadPath
|
||||
return
|
||||
}
|
||||
catch [System.Exception]
|
||||
{
|
||||
# If BITS fails for any reason, delete any potentially partially downloaded files and continue
|
||||
if (Test-Path $tempDownloadName)
|
||||
{
|
||||
Remove-Item $tempDownloadName
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if ($ProxyAuth)
|
||||
{
|
||||
$WC.Proxy.Credentials = Get-Credential-Backwards-Compatible
|
||||
}
|
||||
|
||||
Write-Verbose("Downloading $Dependency...")
|
||||
$WC.DownloadFile($url, $tempDownloadName)
|
||||
Move-Item -Path $tempDownloadName -Destination $downloadPath
|
||||
}
|
||||
|
||||
# Enums (without resorting to C#) are only available on powershell 5+.
|
||||
$ExtractionType_NO_EXTRACTION_REQUIRED = 0
|
||||
$ExtractionType_ZIP = 1
|
||||
$ExtractionType_SELF_EXTRACTING_7Z = 2
|
||||
|
||||
|
||||
# Using this to wait for the execution to finish
|
||||
function Invoke-Command()
|
||||
{
|
||||
param ( [Parameter(Mandatory=$true)][string]$program,
|
||||
[string]$argumentString = "",
|
||||
[switch]$waitForExit = $false )
|
||||
|
||||
$psi = new-object "Diagnostics.ProcessStartInfo"
|
||||
$psi.FileName = $program
|
||||
$psi.Arguments = $argumentString
|
||||
$proc = [Diagnostics.Process]::Start($psi)
|
||||
if ( $waitForExit )
|
||||
{
|
||||
$proc.WaitForExit();
|
||||
}
|
||||
}
|
||||
|
||||
function Expand-ZIPFile($file, $destination)
|
||||
{
|
||||
if (!(Test-Path $destination))
|
||||
{
|
||||
New-Item -ItemType Directory -Path $destination | Out-Null
|
||||
}
|
||||
|
||||
if (Test-Command -commandName 'Microsoft.PowerShell.Archive\Expand-Archive')
|
||||
{
|
||||
Write-Verbose("Extracting with Microsoft.PowerShell.Archive\Expand-Archive")
|
||||
Microsoft.PowerShell.Archive\Expand-Archive -path $file -destinationpath $destination
|
||||
}
|
||||
elseif (Test-Command -commandName 'Pscx\Expand-Archive')
|
||||
{
|
||||
Write-Verbose("Extracting with Pscx\Expand-Archive")
|
||||
Pscx\Expand-Archive -path $file -OutputPath $destination
|
||||
}
|
||||
else
|
||||
{
|
||||
Write-Verbose("Extracting via shell")
|
||||
$shell = new-object -com shell.application
|
||||
$zip = $shell.NameSpace($file)
|
||||
foreach($item in $zip.items())
|
||||
{
|
||||
# Piping to Out-Null is used to block until finished
|
||||
$shell.Namespace($destination).copyhere($item) | Out-Null
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if($Dependency -eq "cmake")
|
||||
{
|
||||
$requiredVersion = "3.9.5"
|
||||
@ -249,23 +78,10 @@ function SelectProgram([Parameter(Mandatory=$true)][string]$Dependency)
|
||||
throw "Unknown program requested"
|
||||
}
|
||||
|
||||
$downloadSubdir = Split-path $downloadPath -Parent
|
||||
if (!(Test-Path $downloadSubdir))
|
||||
{
|
||||
New-Item -ItemType Directory -Path $downloadSubdir | Out-Null
|
||||
}
|
||||
vcpkgDownloadFile $url $downloadPath
|
||||
|
||||
performDownload $Dependency $url $downloadsDir $downloadPath $downloadVersion $requiredVersion
|
||||
|
||||
$downloadedFileHash = Get-Hash-SHA256 $downloadPath
|
||||
if ($expectedDownloadedFileHash -ne $downloadedFileHash)
|
||||
{
|
||||
Write-Host ("`nFile does not have expected hash:`n" +
|
||||
" File path: [ $downloadPath ]`n" +
|
||||
" Expected hash: [ $expectedDownloadedFileHash ]`n" +
|
||||
" Actual hash: [ $downloadedFileHash ]`n")
|
||||
throw "Invalid Hash"
|
||||
}
|
||||
$downloadedFileHash = vcpkgGetSHA256 $downloadPath
|
||||
vcpkgCheckEqualFileHash -filePath $downloadPath -expectedHash $expectedDownloadedFileHash -actualHash $downloadedFileHash
|
||||
|
||||
if ($extractionType -eq $ExtractionType_NO_EXTRACTION_REQUIRED)
|
||||
{
|
||||
@ -275,14 +91,14 @@ function SelectProgram([Parameter(Mandatory=$true)][string]$Dependency)
|
||||
{
|
||||
if (-not (Test-Path $executableFromDownload)) # consider renaming the extraction folder to make sure the extraction finished
|
||||
{
|
||||
Expand-ZIPFile -File $downloadPath -Destination $extractionFolder
|
||||
vcpkgExtractFile -File $downloadPath -Destination $extractionFolder
|
||||
}
|
||||
}
|
||||
elseif($extractionType -eq $ExtractionType_SELF_EXTRACTING_7Z)
|
||||
{
|
||||
if (-not (Test-Path $executableFromDownload))
|
||||
{
|
||||
Invoke-Command $downloadPath "-y" -waitForExit:$true
|
||||
vcpkgInvokeCommand $downloadPath "-y" -wait:$true
|
||||
}
|
||||
}
|
||||
else
|
||||
|
||||
Loading…
Reference in New Issue
Block a user