Merge pull request #44 from conlan0/develop

Add agent shutdown command
amidaware · Feb 22, 2024 · 67abc85 · 67abc85
2 parents a29709c + d3eb6e0
commit 67abc85
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/agent/rpc.go b/agent/rpc.go
@@ -341,6 +341,22 @@ func (a *Agent) RunRPC() {
 				msg.Respond(resp)
 			}()
 
+		case "shutdown":
+			go func() {
+				a.Logger.Debugln("Scheduling immediate shutdown")
+				var resp []byte
+				ret := codec.NewEncoderBytes(&resp, new(codec.MsgpackHandle))
+				ret.Encode("ok")
+				msg.Respond(resp)
+				if runtime.GOOS == "windows" {
+					CMD("shutdown.exe", []string{"/s", "/t", "5", "/f"}, 15, false)
+				} else {
+					opts := a.NewCMDOpts()
+					opts.Command = "shutdown -h now"
+					a.CmdV2(opts)
+				}
+			}()
+
 		case "rebootnow":
 			go func() {
 				a.Logger.Debugln("Scheduling immediate reboot")