From e11c38dd49256584900be3ebaacdcb228a14f87b Mon Sep 17 00:00:00 2001
From: wh1te909 <dcparsi@gmail.com>
Date: Sun, 6 Nov 2022 21:49:26 -0800
Subject: [PATCH] fix race condition in windows agent update
 amidaware/tacticalrmm#1250

---
 agent/agent.go | 2 +-
 agent/rpc.go   | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/agent/agent.go b/agent/agent.go
index d6d7a91..16d1c1c 100644
--- 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,
 		},
 	}
diff --git a/agent/rpc.go b/agent/rpc.go
index ce2a89c..b141be0 100644
--- 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)