Specify restart options for service manager, to avoid use of manual restart logic

1 files changed, 19 insertions, 0 deletions

diff --git a/src/zenutil/service.cpp b/src/zenutil/service.cpp
index a2ff93efd..e1c5e17fa 100644
--- a/src/zenutil/service.cpp
+++ b/src/zenutil/service.cpp
@@ -449,6 +449,25 @@ InstallService(std::string_view ServiceName, const ServiceSpec& Spec)
 		}
 	}
 
+	// Actions defining what the service manager should do in the event of a zenserver crash.
+	// Attempt an immediate restart twice. If both restarts fail, stop trying.
+	// The attempt count will be reset based on the timeout specified in SERVICE_FAILURE_ACTIONS.
+	// If the service manages to survive for the length of that timeout, the reset attempt count
+	// will be reset and we will try restarting if we fail again.
+	SC_ACTION Actions[] = {{SC_ACTION_RESTART, 0}, {SC_ACTION_RESTART, 0}, {SC_ACTION_NONE, 0}};
+
+	SERVICE_FAILURE_ACTIONS FailureAction = {
+		60,	   // if we haven't failed for one minute, assume the service is healthy and reset the failure count
+		NULL,  // no reboot message - we don't want to reboot the whole system if zen dies
+		NULL,  // no command to run on failure - just attempt restarting the service
+		ZEN_ARRAY_COUNT(Actions),
+		Actions};
+
+	if (!ChangeServiceConfig2(schService, SERVICE_CONFIG_FAILURE_ACTIONS, &FailureAction))
+	{
+		return MakeErrorCodeFromLastError();
+	}
+
 	CloseServiceHandle(schService);
 
 	return {};