From 160ef94e8b6aa580225555a9c38a99af2d4e27d2 Mon Sep 17 00:00:00 2001
From: David Oberhollenzer <david.oberhollenzer@tele2.at>
Date: Mon, 26 Mar 2018 00:34:00 +0200
Subject: Add service respawn limit

Signed-off-by: David Oberhollenzer <david.oberhollenzer@tele2.at>
---
 lib/include/service.h |  2 ++
 lib/src/rdsvc.c       | 28 ++++++++++++++++++++++++++++
 2 files changed, 30 insertions(+)

(limited to 'lib')

diff --git a/lib/include/service.h b/lib/include/service.h
index 28d73f1..76d48ce 100644
--- a/lib/include/service.h
+++ b/lib/include/service.h
@@ -53,6 +53,8 @@ typedef struct service_t {
 	size_t num_exec;	/* number of command lines */
 	char *ctty;		/* controlling tty or log file */
 
+	int rspwn_limit;	/* maximum respawn count */
+
 	char **before;		/* services that must be executed later */
 	size_t num_before;
 	char **after;		/* services that must be executed first */
diff --git a/lib/src/rdsvc.c b/lib/src/rdsvc.c
index 972f7e4..54c6d24 100644
--- a/lib/src/rdsvc.c
+++ b/lib/src/rdsvc.c
@@ -23,6 +23,7 @@
 #include <stdio.h>
 #include <errno.h>
 #include <fcntl.h>
+#include <ctype.h>
 
 #include "service.h"
 #include "util.h"
@@ -132,6 +133,32 @@ static int svc_target(service_t *svc, char *arg,
 	return 0;
 }
 
+static int svc_rspwn_limit(service_t *svc, char *arg,
+			   const char *filename, size_t lineno)
+{
+	const char *ptr;
+
+	svc->rspwn_limit = 0;
+
+	if (!isdigit(*arg))
+		goto fail;
+
+	for (ptr = arg; isdigit(*ptr); ++ptr)
+		svc->rspwn_limit = svc->rspwn_limit * 10 + (*ptr - '0');
+
+	if (*ptr != '\0')
+		goto fail;
+
+	free(arg);
+	return 0;
+fail:
+	fprintf(stderr,
+		"%s: %zu: expected numeric argument for respawn limit\n",
+		filename, lineno);
+	free(arg);
+	return -1;
+}
+
 
 static const struct {
 	const char *key;
@@ -147,6 +174,7 @@ static const struct {
 	{ "tty", svc_tty },
 	{ "before", svc_before },
 	{ "after", svc_after },
+	{ "respawn_limit", svc_rspwn_limit },
 };
 
 
-- 
cgit v1.2.3