[Feature] Improve subprocesses termination handle

2 files changed, 55 insertions, 2 deletions

diff --git a/src/libserver/worker_util.c b/src/libserver/worker_util.c
index 2edaa2612..38e58aa83 100644
--- a/src/libserver/worker_util.c
+++ b/src/libserver/worker_util.c
@@ -186,6 +186,34 @@ rspamd_worker_ignore_signal (int signo)
 	sigaction (signo, &sig, NULL);
 }
 
+static void
+rspamd_worker_default_signal (int signo)
+{
+	struct sigaction sig;
+
+	sigemptyset (&sig.sa_mask);
+	sigaddset (&sig.sa_mask, signo);
+	sig.sa_handler = SIG_DFL;
+	sig.sa_flags = 0;
+	sigaction (signo, &sig, NULL);
+}
+
+static void
+rspamd_sigh_free (void *p)
+{
+	struct rspamd_worker_signal_handler *sigh = p;
+	struct rspamd_worker_signal_cb *cb, *tmp;
+
+	DL_FOREACH_SAFE (sigh->cb, cb, tmp) {
+		DL_DELETE (sigh->cb, cb);
+		g_free (cb);
+	}
+
+	event_del (&sigh->ev);
+	rspamd_worker_default_signal (sigh->signo);
+	g_free (sigh);
+}
+
 void
 rspamd_worker_set_signal_handler (int signo, struct rspamd_worker *worker,
 		struct event_base *base,
@@ -219,7 +247,7 @@ rspamd_worker_set_signal_handler (int signo, struct rspamd_worker *worker,
 	DL_APPEND (sigh->cb, cb);
 }
 
-static void
+void
 rspamd_worker_init_signals (struct rspamd_worker *worker, struct event_base *base)
 {
 	struct sigaction signals;
@@ -274,7 +302,7 @@ rspamd_prepare_worker (struct rspamd_worker *worker, const char *name,
 
 	worker->srv->pid = getpid ();
 	worker->signal_events = g_hash_table_new_full (g_direct_hash, g_direct_equal,
-			NULL, g_free);
+			NULL, rspamd_sigh_free);
 
 	ev_base = event_init ();
 
@@ -663,6 +691,20 @@ rspamd_worker_block_signals (void)
 }
 
 void
+rspamd_worker_unblock_signals (void)
+{
+	sigset_t set;
+
+	sigemptyset (&set);
+	sigaddset (&set, SIGTERM);
+	sigaddset (&set, SIGINT);
+	sigaddset (&set, SIGHUP);
+	sigaddset (&set, SIGUSR1);
+	sigaddset (&set, SIGUSR2);
+	sigprocmask (SIG_UNBLOCK, &set, NULL);
+}
+
+void
 rspamd_hard_terminate (struct rspamd_main *rspamd_main)
 {
 	GHashTableIter it;
diff --git a/src/libserver/worker_util.h b/src/libserver/worker_util.h
index d8decbac8..3af9c961d 100644
--- a/src/libserver/worker_util.h
+++ b/src/libserver/worker_util.h
@@ -31,6 +31,12 @@ struct rspamd_worker;
 struct rspamd_worker_signal_handler;
 
 /**
+ * Init basic signals for a worker
+ * @param worker
+ * @param base
+ */
+void rspamd_worker_init_signals (struct rspamd_worker *worker, struct event_base *base);
+/**
  * Prepare worker's startup
  * @param worker worker structure
  * @param name name of the worker
@@ -124,6 +130,11 @@ void rspamd_worker_stop_accept (struct rspamd_worker *worker);
 void rspamd_worker_block_signals (void);
 
 /**
+ * Unblock signals
+ */
+void rspamd_worker_unblock_signals (void);
+
+/**
  * Kill rspamd main and all workers
  * @param rspamd_main
  */