147 lines
3.4 KiB
PowerShell
147 lines
3.4 KiB
PowerShell
$ErrorActionPreference = "Stop"
|
|
|
|
$ConfigPath = "C:\MinerOpsAgent\agent-config.json"
|
|
$Config = Get-Content -Raw $ConfigPath | ConvertFrom-Json
|
|
|
|
$HubUrl = ($Config.hub_url + "").TrimEnd("/")
|
|
$AgentId = $Config.agent_id
|
|
$AgentSecret = $Config.agent_secret
|
|
|
|
$Headers = @{
|
|
"X-MinerOps-Agent-Secret" = $AgentSecret
|
|
}
|
|
|
|
function Normalize-MinerOpsIp {
|
|
param([string] $Ip)
|
|
|
|
if (-not $Ip) {
|
|
return ""
|
|
}
|
|
|
|
return ($Ip -replace "/32$", "").Trim()
|
|
}
|
|
|
|
function Invoke-MinerOps4028PoolsSimple {
|
|
param(
|
|
[Parameter(Mandatory=$true)] [string] $Ip,
|
|
[int] $Port = 4028,
|
|
[int] $TimeoutMs = 5000
|
|
)
|
|
|
|
$client = $null
|
|
$stream = $null
|
|
|
|
try {
|
|
$json = '{"command":"pools"}'
|
|
|
|
$client = New-Object System.Net.Sockets.TcpClient
|
|
$client.ReceiveTimeout = $TimeoutMs
|
|
$client.SendTimeout = $TimeoutMs
|
|
$client.Connect($Ip, $Port)
|
|
|
|
$stream = $client.GetStream()
|
|
$stream.ReadTimeout = $TimeoutMs
|
|
$stream.WriteTimeout = $TimeoutMs
|
|
|
|
$bytes = [System.Text.Encoding]::ASCII.GetBytes($json)
|
|
$stream.Write($bytes, 0, $bytes.Length)
|
|
$stream.Flush()
|
|
|
|
Start-Sleep -Milliseconds 500
|
|
|
|
$buffer = New-Object byte[] 131072
|
|
$read = $stream.Read($buffer, 0, $buffer.Length)
|
|
|
|
if ($read -le 0) {
|
|
return $null
|
|
}
|
|
|
|
$raw = [System.Text.Encoding]::ASCII.GetString($buffer, 0, $read).Trim([char]0).Trim()
|
|
|
|
if (-not $raw) {
|
|
return $null
|
|
}
|
|
|
|
$parsed = $raw | ConvertFrom-Json
|
|
|
|
if (-not $parsed.POOLS) {
|
|
return $null
|
|
}
|
|
|
|
return @($parsed.POOLS)
|
|
}
|
|
catch {
|
|
return $null
|
|
}
|
|
finally {
|
|
if ($stream) {
|
|
try { $stream.Close() } catch {}
|
|
}
|
|
|
|
if ($client) {
|
|
try { $client.Close() } catch {}
|
|
}
|
|
}
|
|
}
|
|
|
|
$response = Invoke-RestMethod -Method Get -Uri "$HubUrl/api/agent/known-miners" -Headers $Headers -TimeoutSec 60
|
|
|
|
$knownMiners = @()
|
|
|
|
if ($response.miners) {
|
|
$knownMiners = @($response.miners)
|
|
} elseif ($response.data -and $response.data.miners) {
|
|
$knownMiners = @($response.data.miners)
|
|
} elseif ($response.data -and $response.data.known_miners) {
|
|
$knownMiners = @($response.data.known_miners)
|
|
}
|
|
|
|
Write-Host "Known miners: $($knownMiners.Count)"
|
|
|
|
$poolSnapshots = New-Object System.Collections.Generic.List[object]
|
|
$count = 0
|
|
|
|
foreach ($miner in $knownMiners) {
|
|
$count++
|
|
|
|
$ip = Normalize-MinerOpsIp -Ip $miner.ip_address
|
|
|
|
if (-not $ip) {
|
|
continue
|
|
}
|
|
|
|
if (($count % 25) -eq 0) {
|
|
Write-Host "Checked $count / $($knownMiners.Count). Collected $($poolSnapshots.Count)."
|
|
}
|
|
|
|
$pools = @(Invoke-MinerOps4028PoolsSimple -Ip $ip)
|
|
|
|
if ($pools.Count -gt 0) {
|
|
$poolSnapshots.Add([pscustomobject]@{
|
|
miner_id = $miner.miner_id
|
|
ip_address = $ip
|
|
mac_address = $miner.mac_address
|
|
model = $miner.model
|
|
pools = @($pools)
|
|
})
|
|
}
|
|
}
|
|
|
|
Write-Host "Pool snapshots collected: $($poolSnapshots.Count)"
|
|
|
|
$body = @{
|
|
agent_id = $AgentId
|
|
pool_snapshots = $poolSnapshots
|
|
} | ConvertTo-Json -Depth 20
|
|
|
|
$result = Invoke-RestMethod `
|
|
-Method Post `
|
|
-Uri "$HubUrl/api/agent/pool-snapshots" `
|
|
-Headers $Headers `
|
|
-Body $body `
|
|
-ContentType "application/json" `
|
|
-TimeoutSec 180
|
|
|
|
$result | ConvertTo-Json -Depth 10
|
|
|