From 0ae5071f9460e0ee0af5afb8bbc3a24b4bb59a5f Mon Sep 17 00:00:00 2001
From: Joachim Wiberg <troglobit@gmail.com>
Date: Thu, 18 Aug 2022 14:05:29 +0200
Subject: [PATCH] API for associtative list of svc_t and pre/post/ready support
 scripts

This support API is for running things like pre:script, post:script and
the upcoming ready:script for services.  A ready:script will be called
when a service notifies Finit of its readyness.

Signed-off-by: Joachim Wiberg <troglobit@gmail.com>
---
 src/service.c | 60 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/src/service.c b/src/service.c
index ce9c2815..3d678d4e 100644
--- a/src/service.c
+++ b/src/service.c
@@ -121,6 +121,66 @@ int service_timeout_cancel(svc_t *svc)
 	return err;
 }
 
+struct assoc {
+	TAILQ_ENTRY(assoc) link;
+
+	pid_t  pid;		/* script pid */
+	svc_t *svc;		/* associated svc_t */
+};
+
+static TAILQ_HEAD(, assoc) svc_assoc_list = TAILQ_HEAD_INITIALIZER(svc_assoc_list);
+
+static void service_script_kill(svc_t *svc)
+{
+	struct assoc *ptr, *next;
+
+	TAILQ_FOREACH_SAFE(ptr, &svc_assoc_list, link, next) {
+		if (ptr->svc != svc || ptr->pid <= 1)
+			continue;
+
+		kill(-ptr->pid, SIGKILL);
+		TAILQ_REMOVE(&svc_assoc_list, ptr, link);
+		free(ptr);
+	}
+}
+
+static int service_script_add(svc_t *svc, pid_t pid)
+{
+	struct assoc *ptr = malloc(sizeof(*ptr));
+
+	if (!ptr) {
+		_pe("Failed starting service script timer");
+		return 1;
+	}
+
+	ptr->svc = svc;
+	ptr->pid = pid;
+	TAILQ_INSERT_TAIL(&svc_assoc_list, ptr, link);
+
+	service_timeout_after(svc, svc->killdelay, service_script_kill);
+
+	return 0;
+}
+
+static int service_script_del(pid_t pid)
+{
+	struct assoc *ptr, *next;
+
+	TAILQ_FOREACH_SAFE(ptr, &svc_assoc_list, link, next) {
+		if (ptr->pid != pid)
+			continue;
+
+		service_timeout_cancel(ptr->svc);
+		kill(-ptr->pid, SIGKILL);
+		TAILQ_REMOVE(&svc_assoc_list, ptr, link);
+		free(ptr);
+
+		return 0;
+	}
+
+	return 1;
+}
+
 /*
  * Redirect stdin to /dev/null => all reads by process = EOF
  * https://www.freedesktop.org/software/systemd/man/systemd.exec.html#Logging%20and%20Standard%20Input/Output