mirror of
https://github.com/valitydev/osquery-1.git
synced 2024-11-07 18:08:53 +00:00
211 lines
8.3 KiB
PowerShell
211 lines
8.3 KiB
PowerShell
# Copyright (c) 2014-present, Facebook, Inc.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the BSD-style license found in the
|
|
# LICENSE file in the root directory of this source tree. An additional grant
|
|
# of patent rights can be found in the PATENTS file in the same directory.
|
|
|
|
# URL of where our pre-compiled third-party dependenices are archived
|
|
$THIRD_PARTY_ARCHIVE_URL = 'https://osquery-packages.s3.amazonaws.com/choco'
|
|
|
|
# Adapted from http://www.jonathanmedd.net/2014/01/testing-for-admin-privileges-in-powershell.html
|
|
function Test-IsAdmin {
|
|
return ([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole(
|
|
[Security.Principal.WindowsBuiltInRole] "Administrator"
|
|
)
|
|
}
|
|
|
|
# Checks if a package is already installed through chocolatey
|
|
# Returns true if:
|
|
# * The package is installed and the user supplies no version
|
|
# * The package is installed and the version matches the user supplied version
|
|
function Check-ChocoPackage-Installed {
|
|
param(
|
|
[string] $packageName = '',
|
|
[string] $packageVersion = ''
|
|
)
|
|
$out = choco list -lr
|
|
|
|
# Parse through the locally installed chocolatey packages and look
|
|
# to see if the supplied package already exists
|
|
ForEach ($pkg in $out) {
|
|
$name, $version = $pkg -split '\|'
|
|
if ($name -eq $packageName) {
|
|
if ($packageVersion -ne "" -and $packageVersion -ne $version) {
|
|
return $false;
|
|
}
|
|
return $true;
|
|
}
|
|
}
|
|
return $false
|
|
}
|
|
|
|
# Attempts to install chocolatey if not already
|
|
function Install-Chocolatey {
|
|
Write-Host " => Attemping to detect presence of chocolatey..." -foregroundcolor DarkYellow
|
|
if ((Get-Command 'choco.exe' -ErrorAction SilentlyContinue) -eq $null) {
|
|
if (Test-Path $env:ALLUSERSPROFILE\chocolatey\bin) {
|
|
Write-Host "[-] WARN: Chocolatey appears to be installed, but was not in path!" -foregroundcolor Yellow
|
|
} else {
|
|
Write-Host " => Did not find. Installing chocolatey..." -foregroundcolor Cyan
|
|
iex ((new-object net.webclient).DownloadString('https://chocolatey.org/install.ps1'))
|
|
}
|
|
Write-Host " => Adding chocolatey to path."
|
|
$env:Path = "$env:Path;$env:ALLUSERSPROFILE\chocolatey\bin"
|
|
} else {
|
|
Write-Host "[*] Chocolatey is already installed." -foregroundcolor Green
|
|
}
|
|
}
|
|
|
|
# Attempts to install a chocolatey package of a specific version if
|
|
# not already there.
|
|
function Install-ChocoPackage {
|
|
param(
|
|
[string] $packageName = '',
|
|
[string] $packageVersion = '',
|
|
[array] $packageOptions = @()
|
|
)
|
|
Write-Host " => Determining whether $packageName is already installed..." -foregroundcolor DarkYellow
|
|
$isInstalled = Check-ChocoPackage-Installed $packageName $packageVersion
|
|
if (-not $isInstalled) {
|
|
Write-Host " => Did not find. Installing $packageName $packageVersion" -foregroundcolor Cyan
|
|
$args = @("install", "-y", "-r", "${packageName}")
|
|
if ($packageVersion -ne '') {
|
|
$args += @("--version", "${packageVersion}")
|
|
}
|
|
if ($packageOptions.count -gt 0) {
|
|
Write-Host "Options: $packageOptions" -foregroundcolor Cyan
|
|
$args += ${packageOptions}
|
|
}
|
|
choco ${args}
|
|
if ($LastExitCode -ne 0) {
|
|
Write-Host "[-] ERROR: $packageName $packageVersion failed to install!" -foregroundcolor Red
|
|
Exit -1
|
|
}
|
|
Write-Host "[+] Done." -foregroundcolor Green
|
|
} else {
|
|
Write-Host "[*] $packageName $packageVersion already installed." -foregroundcolor Green
|
|
}
|
|
}
|
|
|
|
function Install-PipPackages {
|
|
Write-Host " => Attempting to install Python packages" -foregroundcolor DarkYellow
|
|
$env:Path = "$env:Path;$env:HOMEDRIVE\tools\python2;$env:HOMEDRIVE\tools\python2\Scripts"
|
|
if ((Get-Command 'python.exe' -ErrorAction SilentlyContinue) -eq $null) {
|
|
Write-Host "[-] ERROR: failed to find python" -foregroundcolor Red
|
|
Exit -1
|
|
}
|
|
if ((Get-Command 'pip.exe' -ErrorAction SilentlyContinue) -eq $null) {
|
|
Write-Host "[-] ERROR: failed to find pip" -foregroundcolor Red
|
|
Exit -1
|
|
}
|
|
$requirements = Resolve-Path ([System.IO.Path]::Combine($PSScriptRoot, '..', 'requirements.txt'))
|
|
Write-Host " => Upgrading pip..." -foregroundcolor DarkYellow
|
|
python -m pip install --upgrade pip
|
|
if ($LastExitCode -ne 0) {
|
|
Write-Host "[-] ERROR: pip upgrade failed." -foregroundcolor Red
|
|
Exit -1
|
|
}
|
|
Write-Host " => Installing from requirements.txt" -foregroundcolor DarkYellow
|
|
pip install -r $requirements.path
|
|
if ($LastExitCode -ne 0) {
|
|
Write-Host "[-] ERROR: Install packages from requirements failed." -foregroundcolor Red
|
|
Exit -1
|
|
}
|
|
}
|
|
|
|
function Install-ThirdPartyPackages {
|
|
Write-Host " => Retrieving third-party dependencies" -foregroundcolor DarkYellow
|
|
|
|
# XXX: The code below exists because our chocolatey packages are not currently in the chocolatey
|
|
# repository. For now, we will download our packages locally and install from a local source.
|
|
# We also include the official source since thrift-dev depends on the chocolatey thrift package.
|
|
#
|
|
# Once our chocolatey packages are added to the official repository, installing the third-party
|
|
# dependencies will be as easy as Install-ChocoPackage '<package-name>'.
|
|
$packages = @(
|
|
"boost-msvc14.1.59.0",
|
|
"bzip2.1.0.6",
|
|
"doxygen.1.8.11",
|
|
"gflags-dev.2.1.2",
|
|
"glog.0.3.4",
|
|
"openssl.1.0.2",
|
|
"rocksdb.4.4",
|
|
"snappy-msvc.1.1.1.8",
|
|
"thrift-dev.0.9.3",
|
|
"cpp-netlib.0.12.0",
|
|
"linenoise-ng.1.0.0",
|
|
"clang-format.3.9.0"
|
|
)
|
|
$tmpDir = Join-Path $env:TEMP 'osquery-packages'
|
|
Remove-Item $tmpDir -Recurse -ErrorAction Ignore
|
|
mkdir $tmpDir
|
|
Try {
|
|
foreach ($package in $packages) {
|
|
$packageData = $package -split '\.'
|
|
$packageName = $packageData[0]
|
|
$packageVersion = [string]::Join('.', $packageData[1..$packageData.length])
|
|
Write-Host " => Determining whether $packageName is already installed..." -foregroundcolor DarkYellow
|
|
$isInstalled = Check-ChocoPackage-Installed $packageName $packageVersion
|
|
if ($isInstalled) {
|
|
Write-Host "[*] $packageName $packageVersion already installed." -foregroundcolor Green
|
|
continue
|
|
}
|
|
Write-Host " => Did not find. Installing $packageName $packageVersion" -foregroundcolor Cyan
|
|
$downloadUrl = "$THIRD_PARTY_ARCHIVE_URL/$package.nupkg"
|
|
$tmpFilePath = Join-Path $tmpDir "$package.nupkg"
|
|
Write-Host " => Downloading $downloadUrl" -foregroundcolor DarkCyan
|
|
Try {
|
|
(New-Object net.webclient).DownloadFile($downloadUrl, $tmpFilePath)
|
|
} catch [Net.WebException] {
|
|
Write-Host "[-] ERROR: Downloading $package failed. Check connection?" -foregroundcolor Red
|
|
Exit -1
|
|
}
|
|
choco install -y -r $packageName -source "$tmpDir;https://chocolatey.org/api/v2"
|
|
if ($LastExitCode -ne 0) {
|
|
Write-Host "[-] ERROR: Install of $package failed." -foregroundcolor Red
|
|
Exit -1
|
|
}
|
|
Write-Host "[+] DONE" -foregroundcolor Green
|
|
}
|
|
} Finally {
|
|
Remove-Item $tmpDir -Recurse
|
|
}
|
|
}
|
|
|
|
function Update-GitSubmodule {
|
|
if ((Get-Command 'git.exe' -ErrorAction SilentlyContinue) -eq $null) {
|
|
Write-Host "[-] ERROR: Git was not found on the system. Install git." -foregroundcolor Red
|
|
Exit -1
|
|
}
|
|
$thirdPartyPath = Resolve-Path ([System.IO.Path]::Combine($PSScriptRoot, '..', 'third-party'))
|
|
Write-Host " => Updating git submodules in $thirdPartyPath ..." -foregroundcolor Yellow
|
|
pushd $thirdPartyPath
|
|
git submodule update --init
|
|
popd
|
|
Write-Host "[+] Submodules updated!" -foregroundcolor Yellow
|
|
}
|
|
|
|
function Main {
|
|
Write-Host "[+] Provisioning a Win64 build environment for osquery" -foregroundcolor Yellow
|
|
Write-Host " => Verifying script is running with Admin privileges" -foregroundcolor Yellow
|
|
if (-not (Test-IsAdmin)) {
|
|
Write-Host "[-] ERROR: Please run this script with Admin privileges!" -foregroundcolor Red
|
|
Exit -1
|
|
}
|
|
Write-Host "[+] Success!" -foregroundcolor Green
|
|
Install-Chocolatey
|
|
Install-ChocoPackage '7zip.commandline'
|
|
Install-ChocoPackage 'cmake.portable' '3.6.1'
|
|
Install-ChocoPackage 'python2' '2.7.11'
|
|
Install-PipPackages
|
|
Update-GitSubmodule
|
|
$deploymentFile = Resolve-Path ([System.IO.Path]::Combine($PSScriptRoot, 'vsdeploy.xml'))
|
|
$chocoParams = @("--execution-timeout", "7200", "-packageParameters", "--AdminFile ${deploymentFile}")
|
|
Install-ChocoPackage 'visualstudio2015community' '' ${chocoParams}
|
|
Install-ThirdPartyPackages
|
|
Write-Host "[+] Done." -foregroundcolor Yellow
|
|
}
|
|
|
|
$null = Main
|