fix race condition in windows agent update amidaware/tacticalrmm#1250

2022-11-06 21:49:26 -08:00 · 2022-11-06 21:49:26 -08:00 · e11c38dd49
commit e11c38dd49
parent eddafb873b
2 changed files with 2 additions and 1 deletions
--- a/agent/agent.go
+++ b/agent/agent.go
@ -154,7 +154,7 @@ func New(logger *logrus.Logger, version string) *Agent {
 		Option: service.KeyValue{
 			"StartType":              "automatic",
 			"OnFailure":              "restart",
-			"OnFailureDelayDuration": "5s",
+			"OnFailureDelayDuration": "12s",
 			"OnFailureResetPeriod":   10,
 		},
 	}
--- a/agent/rpc.go
+++ b/agent/rpc.go
@ -479,6 +479,7 @@ func (a *Agent) RunRPC() {
 					atomic.StoreUint32(&agentUpdateLocker, 0)
 					nc.Flush()
 					nc.Close()
+					a.ControlService(winSvcName, "stop")
 					os.Exit(0)
 				}
 			}(payload)