ADO OneFuzz pipeline updated with setup script (#420)

This commit is contained in:
Chuck Walbourn 2023-12-06 19:17:02 -08:00 коммит произвёл GitHub
Родитель e8dd6c7559
Коммит 0f8b427bf4
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
3 изменённых файлов: 114 добавлений и 3 удалений

Просмотреть файл

@ -118,6 +118,13 @@ jobs:
Invoke-WebRequest $url -o $target
}
- task: PowerShell@2
displayName: Copy OneFuzz setup script
inputs:
targetType: 'inline'
script: |
Copy-Item -Path .\build\onefuzz-setup.ps1 -Destination .drop/setup.ps1
- task: MSBuild@1
displayName: 'Copy ASan binaries'
inputs:
@ -141,4 +148,3 @@ jobs:
onefuzzDropPAT: $(ONEFUZZ_PAT)
SYSTEM_ACCESSTOKEN: $(System.AccessToken)
onefuzzBugFilingPAT: $(BUGFILING_PAT)

Просмотреть файл

@ -14,7 +14,7 @@
"ProjectName": "Direct3D",
"TargetName": "DirectXTex",
"TargetOptions": [
"-rss_limit_mb=4096"
" -rss_limit_mb=4096"
]
}
],
@ -22,7 +22,8 @@
"fuzzloaders.exe",
"fuzzloaders.pdb",
"clang_rt.asan_dynamic-x86_64.dll",
"msdia140.dll"
"msdia140.dll",
"setup.ps1"
],
"AdoTemplate": {
"Org": "microsoft",

104
build/onefuzz-setup.ps1 Normal file
Просмотреть файл

@ -0,0 +1,104 @@
function Execute-Setup {
# Temporary work-around while OneFuzz does not run script from setup dir
Set-Location -Path $PSScriptRoot
Write-Log "Executing custom setup script in $(pwd)"
# Exclude any uploaded DLL from known DLLs
gci -filter '*.dll' | Exclude-Library
# Update environment values for ASAN_OPTIONS.
# These environment variables affect how the ASan runtime operates.
# Use the 'allocator_may_return_null=1' environment variable if you have compiled
# your fuzzer using clang with the ASan flags
#
# Use 'windows_hook_rtl_allocators=true:allocator_may_return_null=1' if your fuzzer has
# been compiled using MSVC with the ASan flags
#
# $AsanOptions = 'windows_hook_rtl_allocators=true:allocator_may_return_null=1'
# $AsanOptions = 'allocator_may_return_null=1'
$AsanOptions = 'windows_hook_rtl_allocators=true:allocator_may_return_null=1'
# Use the 'machine' scope to make this permanent because the machine will reboot
[Environment]::SetEnvironmentVariable('ASAN_OPTIONS', $AsanOptions, 'Machine')
Write-Log "Set ASAN_OPTIONS to $AsanOptions"
# Done. Useful to know that the script did not prematurely error out
Write-Log 'Setup script finished successfully'
}
# Write log statements into the event log.
function Write-Log {
Param(
[Parameter(Position=0,
Mandatory,
ValueFromPipeline,
ValueFromRemainingArguments)]
[String[]] $Messages,
[Parameter()] [Int] $EventId=0)
Begin {
$EventSource = 'onefuzz-setup.ps1'
$EventLog = 'Application'
if (![System.Diagnostics.EventLog]::SourceExists($EventSource)) {
New-EventLog -LogName $EventLog -Source $EventSource
}
}
Process {
$Messages.ForEach({
Write-EventLog -LogName $EventLog -Source $EventSource -EventId $EventId -EntryType Information -Message $_
})
}
End {}
}
# This function is used to exclude DLL's that the fuzzer is dependent on. The dependent DLL's
# have been built with ASan and copied into the setup directory along with the fuzzer.
function Exclude-Library {
Param(
[Parameter(Position=0,
Mandatory,
ValueFromPipeline,
ValueFromRemainingArguments)]
[String[]] $Libraries)
Begin {
$Path = 'HKLM:System\CurrentControlSet\Control\Session Manager'
$Name = 'ExcludeFromKnownDLLs'
}
Process {
# Get existing excluded libraries
$ExistingProperty = Get-ItemProperty -Path $Path -Name $Name -ErrorAction SilentlyContinue
$ExistingExclusions = @()
# Normalize DLL name to lowercase for comparison
if ($ExistingProperty -ne $null) {
$ExistingExclusions = $ExistingProperty.$Name.ForEach("ToLower")
}
# Normalize DLL name to lowercase for comparison, remove duplicates
$Libs = $Libraries.ForEach("ToLower") | Select-Object -Unique
Write-Log "Excluding libraries $Libs"
# Discard empty strings and libraries already excluded
$Libs = $Libs.Where({$_.Length -gt 0 -and !($ExistingExclusions.Contains($_))})
# If anything remains either add or update registry key
if ($Libs.Length -gt 0) {
if ($ExistingProperty -eq $null) {
# Create registry key to exclude DLLs
New-ItemProperty -Path $Path -Name $Name -PropertyType MultiString -Value $Libs
Write-Log "Created known DLLs exclusions with $Libs"
} else {
# Update registry key to exclude DLLs
Set-ItemProperty -Path $Path -Name $Name -Value ($ExistingProperty.$Name + $Libs)
Write-Log "Updated known DLLs exclusions with $Libs"
}
} else {
# DLLs already excluded
Write-Log "Known DLL exclusions already exist for $Libraries"
}
}
End {}
}
Execute-Setup