--- libaitsched/src/aitsched.c	2023/01/18 23:46:18	1.30
+++ libaitsched/src/aitsched.c	2023/08/31 15:31:17	1.34.2.2
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: aitsched.c,v 1.30 2023/01/18 23:46:18 misho Exp $
+* $Id: aitsched.c,v 1.34.2.2 2023/08/31 15:31:17 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -181,7 +181,6 @@ schedRegisterHooks(sched_root_task_t * __restrict root
 	root->root_hooks.hook_add.alarm = sched_hook_alarm;
 	root->root_hooks.hook_add.node = sched_hook_node;
 	root->root_hooks.hook_add.proc = sched_hook_proc;
-	root->root_hooks.hook_add.signal = sched_hook_signal;
 #ifdef EVFILT_USER
 	root->root_hooks.hook_add.user = sched_hook_user;
 #endif
@@ -189,6 +188,7 @@ schedRegisterHooks(sched_root_task_t * __restrict root
 #ifdef HAVE_LIBPTHREAD
 	root->root_hooks.hook_add.thread = sched_hook_thread;
 #endif
+	root->root_hooks.hook_add.signal = sched_hook_signal;
 
 	root->root_hooks.hook_exec.cancel = sched_hook_cancel;
 	root->root_hooks.hook_exec.fetch = sched_hook_fetch;
@@ -211,6 +211,9 @@ schedInit(void ** __restrict data, size_t datlen)
 {
 	sched_root_task_t *root = NULL;
 	int (*func)(sched_root_task_t *);
+#if SUP_ENABLE == KQ_SUPPORT
+	sigset_t ss;
+#endif
 #ifdef HAVE_LIBPTHREAD
 	register int i;
 #endif
@@ -228,6 +231,12 @@ schedInit(void ** __restrict data, size_t datlen)
 		sched_timespecinf(&root->root_poll);
 
 #ifdef HAVE_LIBPTHREAD
+		if (pthread_mutex_init(&root->root_sigmtx, NULL)) {
+			LOGERR;
+			e_free(root);
+			return NULL;
+		}
+
 		for (i = 0; i < taskMAX; i++)
 			if ((errno = pthread_mutex_init(&root->root_mtx[i], NULL))) {
 				LOGERR;
@@ -262,7 +271,16 @@ schedInit(void ** __restrict data, size_t datlen)
 #ifdef HAVE_LIBPTHREAD
 		for (i = 0; i < taskMAX; i++)
 			pthread_mutex_unlock(&root->root_mtx[i]);
+
+#if SUP_ENABLE == KQ_SUPPORT
+		sigfillset(&ss);
+		pthread_sigmask(SIG_BLOCK, &ss, &root->root_oldset);
+#else
+		pthread_sigmask(SIG_BLOCK, NULL, &root->root_oldset);
 #endif
+#else
+		sigprocmask(SIG_BLOCK, NULL, &root->root_oldset);
+#endif
 
 		if (data && *data) {
 			if (datlen) {
@@ -386,7 +404,21 @@ schedEnd(sched_root_task_t ** __restrict root)
 		SCHED_QUNLOCK(*root, i);
 		pthread_mutex_destroy(&(*root)->root_mtx[i]);
 	}
+
+	if ((*root)->root_sigthr) {
+		pthread_cancel((*root)->root_sigthr);
+#ifdef __linux__
+		(*root)->root_sigthr = 0L;
+#else
+		(*root)->root_sigthr = NULL;
 #endif
+		pthread_sigmask(SIG_SETMASK, &(*root)->root_oldset, NULL);
+		pthread_mutex_unlock(&(*root)->root_sigmtx);
+	}
+	pthread_mutex_destroy(&(*root)->root_sigmtx);
+#else
+	sigprocmask(SIG_SETMASK, &(*root)->root_oldset, NULL);
+#endif
 
 	e_free(*root);
 	*root = NULL;
@@ -1109,4 +1141,108 @@ schedResumeby(sched_root_task_t * __restrict root, u_c
 	SCHED_QUNLOCK(root, taskSUSPEND);
 
 	return flg;
+}
+
+static void *
+_sched_sigDisp(void *arg)
+{
+	sched_root_task_t *root = arg;
+	sched_task_t *task, *tmp;
+	int sig, flg;
+	sigset_t ss;
+
+	sigfillset(&ss);
+	while (root->root_sigthr) {
+		if (sigwait(&ss, &sig))
+			continue;
+
+		pthread_mutex_lock(&root->root_sigmtx);
+		if (!sigismember(&root->root_sigset, sig)) {
+			pthread_mutex_unlock(&root->root_sigmtx);
+			continue;
+		}
+
+		flg = 0;
+		TAILQ_FOREACH_SAFE(task, &root->root_signal, task_node, tmp) {
+			if (TASK_VAL(task) == (uintptr_t) sig) {
+				if (!flg) {
+					TASK_RET(task) = 0;
+					TASK_FLAG(task) = 0;
+
+					/* remove signal handle */
+					transit_task2ready(task, &root->root_signal);
+				}
+				flg++;
+			}
+		}
+		if (flg < 2)
+			sigdelset(&root->root_sigset, sig);
+
+		pthread_mutex_unlock(&root->root_sigmtx);
+	}
+
+	return NULL;
+}
+
+/*
+ * schedSignalDispatch() - Activate or Deactivate signal dispatcher
+ *
+ * @root = root task
+ * @on = Activate or =0 deactivate
+ * return: -1 error or 0 ok
+ */
+int
+schedSignalDispatch(sched_root_task_t * __restrict root, int on)
+{
+	sigset_t ss;
+	struct timespec ts = { .tv_sec = 1, .tv_nsec = 0 };
+#ifndef HAVE_LIBPTHREAD
+	sched_SetErr(ENOTSUP, "Library has not support pthreads");
+	return -1;
+#else
+	pthread_attr_t attr;
+#endif
+
+#if SUP_ENABLE == KQ_SUPPORT
+	return 0;
+#endif
+
+	if (!on) {
+		pthread_cancel(root->root_sigthr);
+#ifdef __linux__
+		root->root_sigthr = 0L;
+#else
+		root->root_sigthr = NULL;
+#endif
+		return 0;
+	}
+
+	/* if we have not using polling, then we must turn on */
+	if (sched_timespecisinf(&root->root_poll))
+		schedPolling(root, &ts, NULL);
+
+	pthread_attr_init(&attr);
+	pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
+#ifdef SCHED_RR
+	pthread_attr_setschedpolicy(&attr, SCHED_RR);
+#else
+	pthread_attr_setschedpolicy(&attr, SCHED_OTHER);
+#endif
+
+	sigfillset(&ss);
+	pthread_sigmask(SIG_BLOCK, &ss, NULL);
+
+	if (pthread_create(&root->root_sigthr, &attr, _sched_sigDisp, root)) {
+		sched_SetErr(errno, "pthread_create(SignalDispatch) #%d - %s", 
+				errno, strerror(errno));
+#ifdef __linux__
+		root->root_sigthr = 0L;
+#else
+		root->root_sigthr = NULL;
+#endif
+		pthread_sigmask(SIG_SETMASK, &root->root_oldset, NULL);
+		return -1;
+	}
+
+	return 0;
 }