mirror of https://github.com/ospab/ostp.git
211 lines
8.5 KiB
PowerShell
211 lines
8.5 KiB
PowerShell
$ErrorActionPreference = "Stop"
|
|
$repo = "ospab/ostp"
|
|
|
|
# 1. Smart & Aggressive Installation Path Resolution
|
|
$InstallDir = "C:\opt\ostp" # Standard default fallback
|
|
|
|
if (Test-Path "config.json") {
|
|
# Config is in active current directory
|
|
$InstallDir = (Get-Item .).FullName
|
|
} elseif (Test-Path "ostp.exe") {
|
|
# Binary is in active current directory
|
|
$InstallDir = (Get-Item .).FullName
|
|
} elseif ($cmd = Get-Command "ostp" -ErrorAction SilentlyContinue) {
|
|
# Binary is registered in system PATH
|
|
$InstallDir = Split-Path $cmd.Path
|
|
} else {
|
|
# Aggressive search in current directory tree (excluding compiler target and Git directories)
|
|
$found = Get-ChildItem -Filter "ostp.exe" -Recurse -File -ErrorAction SilentlyContinue |
|
|
Where-Object { $_.FullName -notlike "*\target\*" -and $_.FullName -notlike "*\.git\*" } |
|
|
Select-Object -First 1
|
|
if ($found) {
|
|
$InstallDir = Split-Path $found.FullName
|
|
} else {
|
|
# Scan parent directory as fallback
|
|
$parentFound = Get-ChildItem -Path .. -Filter "ostp.exe" -File -ErrorAction SilentlyContinue | Select-Object -First 1
|
|
if ($parentFound) {
|
|
$InstallDir = Split-Path $parentFound.FullName
|
|
}
|
|
}
|
|
}
|
|
|
|
Write-Host "========================================================"
|
|
Write-Host " Installing Ospab Stealth Transport Protocol (OSTP)"
|
|
Write-Host "========================================================"
|
|
Write-Host "Target deployment location: $InstallDir"
|
|
|
|
# 2. Check Write Access & Elevation Status
|
|
$isAdmin = ([Security.Principal.WindowsPrincipal] [Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)
|
|
|
|
if (-not (Test-Path $InstallDir)) {
|
|
try {
|
|
New-Item -ItemType Directory -Path $InstallDir -ErrorAction Stop | Out-Null
|
|
} catch {
|
|
Write-Error "Access Denied: Cannot create target directory '$InstallDir'. Run as Administrator."
|
|
exit 1
|
|
}
|
|
} else {
|
|
try {
|
|
$testFile = Join-Path $InstallDir "ostp_write_test_$($PID).tmp"
|
|
"test" | Set-Content $testFile -ErrorAction Stop
|
|
Remove-Item $testFile -Force
|
|
} catch {
|
|
Write-Error "Access Denied: Directory '$InstallDir' is read-only. Run as Administrator to update."
|
|
exit 1
|
|
}
|
|
}
|
|
|
|
# 3. Detect Architecture
|
|
$arch = "amd64"
|
|
if ([System.Environment]::Is64BitOperatingSystem -and ($Env:PROCESSOR_ARCHITECTURE -eq "ARM64" -or $Env:PROCESSOR_ARCHITEW6432 -eq "ARM64")) {
|
|
$arch = "arm64"
|
|
}
|
|
|
|
# 4. Fetch Stable Version Asset (with -UseBasicParsing to prevent Internet Explorer hangs)
|
|
Write-Host "Fetching latest stable version from the repository..."
|
|
try {
|
|
[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
|
|
# Explicitly set -UseBasicParsing to guarantee execution doesn't hang on headless servers
|
|
$api = Invoke-RestMethod -Uri "https://api.github.com/repos/$repo/releases/latest" -UseBasicParsing
|
|
$tag = $api.tag_name
|
|
} catch {
|
|
Write-Host "[Notice] Failed to retrieve tag automatically."
|
|
$tag = Read-Host "Enter release version tag manually (e.g., v0.1.23)"
|
|
if (-not $tag) { exit 1 }
|
|
}
|
|
|
|
$archive = "ostp-windows-$arch.zip"
|
|
$url = "https://github.com/$repo/releases/download/$tag/$archive"
|
|
$zipPath = Join-Path $env:TEMP "ostp_temp_$($PID).zip"
|
|
$extractPath = Join-Path $env:TEMP "ostp_extract_$($PID)"
|
|
|
|
Write-Host "Downloading asset windows-${arch}: $url ..."
|
|
# Explicitly set -UseBasicParsing to prevent any engine initialization stalls
|
|
Invoke-WebRequest -Uri $url -OutFile $zipPath -UseBasicParsing
|
|
|
|
if (-not (Test-Path $zipPath)) {
|
|
Write-Error "Failed to download zip package."
|
|
exit 1
|
|
}
|
|
|
|
# Overwrite and clear file locks
|
|
if (Test-Path $extractPath) { Remove-Item $extractPath -Recurse -Force }
|
|
Expand-Archive -Path $zipPath -DestinationPath $extractPath -Force
|
|
|
|
$extractedFiles = Get-ChildItem -Path $extractPath -File -Recurse
|
|
if ($extractedFiles.Count -gt 0) {
|
|
Write-Host "Stopping any active instances of ostp to unlock binary targets..."
|
|
Stop-Process -Name "ostp", "tun2socks" -Force -ErrorAction SilentlyContinue
|
|
Start-Sleep -Seconds 2
|
|
|
|
foreach ($file in $extractedFiles) {
|
|
$destPath = Join-Path $InstallDir $file.Name
|
|
if (Test-Path $destPath) {
|
|
# Rename the existing executable out of the way to bypass file locks
|
|
$oldPath = $destPath + ".old_$PID"
|
|
Rename-Item -Path $destPath -NewName $oldPath -ErrorAction SilentlyContinue
|
|
}
|
|
Copy-Item -Path $file.FullName -Destination $destPath -Force
|
|
(Get-Item $destPath).LastWriteTime = [DateTime]::Now
|
|
}
|
|
|
|
# Try to clean up any leftover old files
|
|
Get-ChildItem -Path $InstallDir -Filter "*.old_*" | Remove-Item -Force -ErrorAction SilentlyContinue
|
|
|
|
Write-Host "Executables and dependencies successfully deployed to $InstallDir."
|
|
} else {
|
|
Write-Error "No files found in archive package."
|
|
exit 1
|
|
}
|
|
|
|
Remove-Item $zipPath -Force
|
|
Remove-Item $extractPath -Recurse -Force
|
|
|
|
# 5. Smart Auto-Updater Mode
|
|
$configPath = Join-Path $InstallDir "config.json"
|
|
if (Test-Path $configPath) {
|
|
Write-Host "--------------------------------------------------------"
|
|
Write-Host "[Update] Existing configuration detected at $configPath."
|
|
Write-Host "[Update] Binary successfully hot-swapped to version $tag."
|
|
Write-Host "--------------------------------------------------------"
|
|
Write-Host "Update completed successfully!"
|
|
exit 0
|
|
}
|
|
|
|
# 6. Interactive Setup
|
|
Write-Host "--------------------------------------------------------"
|
|
Write-Host "Select configuration mode:"
|
|
Write-Host "1) Configure Server"
|
|
Write-Host "2) Configure Client"
|
|
Write-Host "--------------------------------------------------------"
|
|
$mode = Read-Host "Enter choice [1-2]"
|
|
|
|
Push-Location $InstallDir
|
|
|
|
if ($mode -eq "1") {
|
|
Write-Host "Initializing server configuration..."
|
|
& .\ostp.exe --init server --config config.json
|
|
|
|
$config = Get-Content "config.json" -Raw | ConvertFrom-Json
|
|
$listen = Read-Host "Enter IP and port to accept incoming traffic [default: 0.0.0.0:50000]"
|
|
if ($listen) { $config.listen = $listen }
|
|
|
|
$keyCount = Read-Host "How many access keys to generate? [default: 1]"
|
|
if (-not $keyCount) { $keyCount = 1 }
|
|
|
|
if ([int]$keyCount -gt 1) {
|
|
Write-Host "Generating additional security keys..."
|
|
$keys = & .\ostp.exe -g -c $keyCount
|
|
$config.access_keys = $keys -split "`r`n" | Where-Object { $_ -ne "" }
|
|
}
|
|
|
|
$config | ConvertTo-Json -Depth 10 | Set-Content "config.json"
|
|
Write-Host "Server configuration completed. Config written to $(Join-Path $InstallDir 'config.json')"
|
|
|
|
} elseif ($mode -eq "2") {
|
|
Write-Host "Initializing client configuration..."
|
|
& .\ostp.exe --init client --config config.json
|
|
|
|
$config = Get-Content "config.json" -Raw | ConvertFrom-Json
|
|
$server = Read-Host "Enter remote server address (IP:PORT)"
|
|
if ($server) { $config.server = $server }
|
|
|
|
$key = Read-Host "Enter access key (leave blank to generate automatically)"
|
|
if (-not $key) {
|
|
$key = & .\ostp.exe -g
|
|
Write-Host "Automatically generated client access key: $key"
|
|
}
|
|
$config.access_key = $key.Trim()
|
|
|
|
$socks = Read-Host "Enter SOCKS5 listening address [default: 127.0.0.1:1088]"
|
|
if ($socks) { $config.socks5_bind = $socks }
|
|
|
|
$config | ConvertTo-Json -Depth 10 | Set-Content "config.json"
|
|
Write-Host "Client configuration completed. Config written to $(Join-Path $InstallDir 'config.json')"
|
|
} else {
|
|
Write-Error "Invalid configuration choice."
|
|
Pop-Location
|
|
exit 1
|
|
}
|
|
|
|
Pop-Location
|
|
|
|
# 7. Environment PATH Registration
|
|
Write-Host "--------------------------------------------------------"
|
|
Write-Host "Registering binary route in Environment PATH..."
|
|
$targetScope = if ($isAdmin) { [EnvironmentVariableTarget]::Machine } else { [EnvironmentVariableTarget]::User }
|
|
$sysPath = [Environment]::GetEnvironmentVariable("Path", $targetScope)
|
|
if ($sysPath -notlike "*$InstallDir*") {
|
|
$newPath = "$sysPath;$InstallDir"
|
|
[Environment]::SetEnvironmentVariable("Path", $newPath, $targetScope)
|
|
Write-Host "Environment PATH updated successfully ($($targetScope.ToString()) scope)."
|
|
} else {
|
|
Write-Host "$InstallDir is already registered in PATH."
|
|
}
|
|
|
|
Write-Host "--------------------------------------------------------"
|
|
Write-Host "Deployment completed successfully!"
|
|
Write-Host "Binary can be executed globally by typing: ostp"
|
|
Write-Host "Config location: $(Join-Path $InstallDir 'config.json')"
|
|
Write-Host "--------------------------------------------------------"
|