forked from p15670423/monkey
275 lines
9.9 KiB
PowerShell
275 lines
9.9 KiB
PowerShell
param(
|
|
[Parameter(Mandatory = $false, Position = 0)]
|
|
[String] $monkey_home = (Get-Item -Path ".\").FullName,
|
|
|
|
[Parameter(Mandatory = $false, Position = 1)]
|
|
[System.String]
|
|
$branch = "develop",
|
|
[Parameter(Mandatory = $false, Position = 2)]
|
|
[Bool]
|
|
$agents = $true
|
|
)
|
|
|
|
function Configure-precommit([String] $git_repo_dir)
|
|
{
|
|
Write-Output "Installing pre-commit and setting up pre-commit hook"
|
|
Push-Location $git_repo_dir
|
|
python -m pip install pre-commit
|
|
if ($LastExitCode) {
|
|
exit
|
|
}
|
|
pre-commit install -t pre-commit -t pre-push
|
|
if ($LastExitCode) {
|
|
exit
|
|
}
|
|
Pop-Location
|
|
|
|
# Set env variable to skip Swimm verification during pre-commit, Windows not supported yet
|
|
$skipValue = [System.Environment]::GetEnvironmentVariable('SKIP', [System.EnvironmentVariableTarget]::User)
|
|
if ($skipValue) { # if `SKIP` is not empty
|
|
if (-Not ($skipValue -split ',' -contains 'swimm-verify')) { # if `SKIP` doesn't already have "swimm-verify"
|
|
[System.Environment]::SetEnvironmentVariable('SKIP', $env:SKIP + ',swimm-verify', [System.EnvironmentVariableTarget]::User)
|
|
}
|
|
}
|
|
else {
|
|
[System.Environment]::SetEnvironmentVariable('SKIP', 'swimm-verify', [System.EnvironmentVariableTarget]::User)
|
|
}
|
|
|
|
Write-Output "Pre-commit successfully installed"
|
|
}
|
|
|
|
function Deploy-Windows([String] $monkey_home = (Get-Item -Path ".\").FullName, [String] $branch = "develop")
|
|
{
|
|
Write-Output "Downloading to $monkey_home"
|
|
Write-Output "Branch $branch"
|
|
# Set variables for script execution
|
|
[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
|
|
$webClient = New-Object System.Net.WebClient
|
|
|
|
|
|
# Import the config variables
|
|
$config_filename = New-TemporaryFile
|
|
$config_filename = "$PsScriptRoot\config.ps1"
|
|
$config_url = "https://raw.githubusercontent.com/guardicore/monkey/" + $branch + "/deployment_scripts/config.ps1"
|
|
$webClient.DownloadFile($config_url, $config_filename)
|
|
. ./config.ps1
|
|
"Config variables from config.ps1 imported"
|
|
Remove-Item $config_filename
|
|
|
|
|
|
# If we want monkey in current dir we need to create an empty folder for source files
|
|
if ((Join-Path $monkey_home '') -eq (Join-Path (Get-Item -Path ".\").FullName ''))
|
|
{
|
|
$monkey_home = Join-Path -Path $monkey_home -ChildPath $MONKEY_FOLDER_NAME
|
|
}
|
|
|
|
|
|
# We check if git is installed
|
|
try
|
|
{
|
|
git | Out-Null -ErrorAction Stop
|
|
"Git requirement satisfied"
|
|
}
|
|
catch [System.Management.Automation.CommandNotFoundException]
|
|
{
|
|
"Please install git before running this script or add it to path and restart cmd"
|
|
return
|
|
}
|
|
|
|
# Download the monkey
|
|
$command = "git clone --single-branch --recurse-submodules -b $branch $MONKEY_GIT_URL $monkey_home 2>&1"
|
|
Write-Output $command
|
|
$output = cmd.exe /c $command
|
|
$binDir = (Join-Path -Path $monkey_home -ChildPath $MONKEY_ISLAND_DIR | Join-Path -ChildPath "\bin")
|
|
if ($output -like "*already exists and is not an empty directory.*")
|
|
{
|
|
"Assuming you already have the source directory. If not, make sure to set an empty directory as monkey's home directory."
|
|
}
|
|
elseif ($output -like "fatal:*")
|
|
{
|
|
"Error while cloning monkey from the repository:"
|
|
$output
|
|
return
|
|
}
|
|
else
|
|
{
|
|
"Monkey cloned from the repository"
|
|
# Create bin directory
|
|
New-Item -ItemType directory -path $binDir
|
|
"Bin directory added"
|
|
}
|
|
|
|
# We check if python is installed
|
|
try
|
|
{
|
|
$version = cmd.exe /c '"python" --version 2>&1'
|
|
if ($version -like 'Python 3.*')
|
|
{
|
|
"Python 3.* was found, installing dependencies"
|
|
}
|
|
else
|
|
{
|
|
throw System.Management.Automation.CommandNotFoundException
|
|
}
|
|
}
|
|
catch [System.Management.Automation.CommandNotFoundException]
|
|
{
|
|
"Downloading python 3 ..."
|
|
"Select 'add to PATH' when installing"
|
|
$webClient.DownloadFile($PYTHON_URL, $TEMP_PYTHON_INSTALLER)
|
|
Start-Process -Wait $TEMP_PYTHON_INSTALLER -ErrorAction Stop
|
|
$env:Path = [System.Environment]::GetEnvironmentVariable("Path", "Machine") + ";" + [System.Environment]::GetEnvironmentVariable("Path", "User")
|
|
Remove-Item $TEMP_PYTHON_INSTALLER
|
|
# Check if installed correctly
|
|
$version = cmd.exe /c '"python" --version 2>&1'
|
|
if ($version -like '* is not recognized*')
|
|
{
|
|
"Python is not found in PATH. Add it to PATH and relaunch the script."
|
|
return
|
|
}
|
|
}
|
|
|
|
"Upgrading pip..."
|
|
$output = cmd.exe /c 'python -m pip install --user --upgrade pip 2>&1'
|
|
$output
|
|
if ($output -like '*No module named pip*')
|
|
{
|
|
"Make sure pip module is installed and re-run this script."
|
|
return
|
|
}
|
|
|
|
"Installing pipx"
|
|
pip install --user -U pipx
|
|
pipx ensurepath
|
|
pipx install pipenv
|
|
|
|
"Installing python packages for island"
|
|
Push-Location -Path (Join-Path -Path $monkey_home -ChildPath $MONKEY_ISLAND_DIR) -ErrorAction Stop
|
|
pipenv install --dev
|
|
Pop-Location
|
|
"Installing python packages for monkey"
|
|
Push-Location -Path (Join-Path -Path $monkey_home -ChildPath $MONKEY_DIR) -ErrorAction Stop
|
|
pipenv install --dev
|
|
Pop-Location
|
|
|
|
Configure-precommit($monkey_home)
|
|
|
|
$user_python_dir = cmd.exe /c 'py -m site --user-site'
|
|
$user_python_dir = Join-Path (Split-Path $user_python_dir) -ChildPath "\Scripts"
|
|
if (!($ENV:Path | Select-String -SimpleMatch $user_python_dir))
|
|
{
|
|
"Adding python scripts path to user's env"
|
|
$env:Path += ";" + $user_python_dir
|
|
[Environment]::SetEnvironmentVariable("Path", $env:Path, "User")
|
|
}
|
|
|
|
# Download mongodb
|
|
if (!(Test-Path -Path (Join-Path -Path $binDir -ChildPath "mongodb")))
|
|
{
|
|
"Downloading mongodb ..."
|
|
$webClient.DownloadFile($MONGODB_URL, $TEMP_MONGODB_ZIP)
|
|
"Unzipping mongodb"
|
|
Expand-Archive $TEMP_MONGODB_ZIP -DestinationPath $binDir
|
|
# Get unzipped folder's name
|
|
$mongodb_folder = Get-ChildItem -Path $binDir | Where-Object -FilterScript {
|
|
($_.Name -like "mongodb*")
|
|
} | Select-Object -ExpandProperty Name
|
|
# Move all files from extracted folder to mongodb folder
|
|
New-Item -ItemType directory -Path (Join-Path -Path $binDir -ChildPath "mongodb")
|
|
"Moving extracted files"
|
|
Move-Item -Path (Join-Path -Path $binDir -ChildPath $mongodb_folder | Join-Path -ChildPath "\bin\*") -Destination (Join-Path -Path $binDir -ChildPath "mongodb\")
|
|
"Removing zip file"
|
|
Remove-Item $TEMP_MONGODB_ZIP
|
|
Remove-Item (Join-Path -Path $binDir -ChildPath $mongodb_folder) -Recurse
|
|
}
|
|
|
|
# Download OpenSSL
|
|
"Downloading OpenSSL ..."
|
|
$webClient.DownloadFile($OPEN_SSL_URL, $TEMP_OPEN_SSL_ZIP)
|
|
"Unzipping OpenSSl"
|
|
Expand-Archive $TEMP_OPEN_SSL_ZIP -DestinationPath (Join-Path -Path $binDir -ChildPath "openssl") -ErrorAction SilentlyContinue
|
|
"Removing zip file"
|
|
Remove-Item $TEMP_OPEN_SSL_ZIP
|
|
|
|
# Download and install C++ redistributable
|
|
"Downloading C++ redistributable ..."
|
|
$webClient.DownloadFile($CPP_URL, $TEMP_CPP_INSTALLER)
|
|
Start-Process -Wait $TEMP_CPP_INSTALLER -ErrorAction Stop
|
|
Remove-Item $TEMP_CPP_INSTALLER
|
|
|
|
# Generate ssl certificate
|
|
"Generating ssl certificate"
|
|
Push-Location -Path (Join-Path -Path $monkey_home -ChildPath $MONKEY_ISLAND_DIR)
|
|
. .\windows\create_certificate.bat
|
|
Pop-Location
|
|
|
|
if ($agents)
|
|
{
|
|
# Adding binaries
|
|
"Adding binaries"
|
|
$binaries = (Join-Path -Path $monkey_home -ChildPath $MONKEY_ISLAND_DIR | Join-Path -ChildPath "\cc\binaries")
|
|
New-Item -ItemType directory -path $binaries -ErrorAction SilentlyContinue
|
|
$webClient.DownloadFile($LINUX_64_BINARY_URL, (Join-Path -Path $binaries -ChildPath $LINUX_64_BINARY_PATH))
|
|
$webClient.DownloadFile($WINDOWS_64_BINARY_URL, (Join-Path -Path $binaries -ChildPath $WINDOWS_64_BINARY_PATH))
|
|
}
|
|
|
|
|
|
# Check if NPM installed
|
|
"Installing npm"
|
|
try
|
|
{
|
|
$version = cmd.exe /c '"npm" --version 2>&1'
|
|
if ($version -like "*is not recognized*")
|
|
{
|
|
throw System.Management.Automation.CommandNotFoundException
|
|
}
|
|
else
|
|
{
|
|
"Npm already installed"
|
|
}
|
|
}
|
|
catch [System.Management.Automation.CommandNotFoundException]
|
|
{
|
|
"Downloading npm ..."
|
|
$webClient.DownloadFile($NPM_URL, $TEMP_NPM_INSTALLER)
|
|
Start-Process -Wait $TEMP_NPM_INSTALLER
|
|
$env:Path = [System.Environment]::GetEnvironmentVariable("Path", "Machine")
|
|
Remove-Item $TEMP_NPM_INSTALLER
|
|
}
|
|
|
|
"Updating npm"
|
|
Push-Location -Path (Join-Path -Path $monkey_home -ChildPath $MONKEY_ISLAND_DIR | Join-Path -ChildPath "\cc\ui")
|
|
& npm update
|
|
& npm run dist
|
|
Pop-Location
|
|
|
|
# Create infection_monkey/bin directory if not already present
|
|
$binDir = (Join-Path -Path $monkey_home -ChildPath $MONKEY_DIR | Join-Path -ChildPath "\bin")
|
|
New-Item -ItemType directory -path $binaries -ErrorAction SilentlyContinue
|
|
|
|
# Download upx
|
|
if (!(Test-Path -Path (Join-Path -Path $binDir -ChildPath "upx.exe")))
|
|
{
|
|
"Downloading upx ..."
|
|
$webClient.DownloadFile($UPX_URL, $TEMP_UPX_ZIP)
|
|
"Unzipping upx"
|
|
Expand-Archive $TEMP_UPX_ZIP -DestinationPath $binDir -ErrorAction SilentlyContinue
|
|
Move-Item -Path (Join-Path -Path $binDir -ChildPath $UPX_FOLDER | Join-Path -ChildPath "upx.exe") -Destination $binDir
|
|
# Remove unnecessary files
|
|
Remove-Item -Recurse -Force (Join-Path -Path $binDir -ChildPath $UPX_FOLDER)
|
|
"Removing zip file"
|
|
Remove-Item $TEMP_UPX_ZIP
|
|
}
|
|
|
|
# Get Swimm
|
|
"Downloading Swimm..."
|
|
$swimm_filename = Join-Path -Path $HOME -ChildPath "swimm.exe"
|
|
$webClient.DownloadFile($SWIMM_URL, $swimm_filename)
|
|
Start-Process $swimm_filename
|
|
|
|
|
|
"Script finished"
|
|
|
|
}
|
|
Deploy-Windows -monkey_home $monkey_home -branch $branch
|