--- libaitsched/src/aitsched.c	2023/07/27 20:51:28	1.32
+++ libaitsched/src/aitsched.c	2023/08/03 23:22:33	1.33
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: aitsched.c,v 1.32 2023/07/27 20:51:28 misho Exp $
+* $Id: aitsched.c,v 1.33 2023/08/03 23:22:33 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -211,7 +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
@@ -270,11 +272,19 @@ schedInit(void ** __restrict data, size_t datlen)
 		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
+#if SUP_ENABLE == KQ_SUPPORT
 		sigprocmask(SIG_BLOCK, &ss, &root->root_oldset);
+#else
+		sigprocmask(SIG_BLOCK, NULL, &root->root_oldset);
 #endif
+#endif
 
 		if (data && *data) {
 			if (datlen) {
@@ -1188,6 +1198,7 @@ _sched_sigDisp(void *arg)
 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");
@@ -1222,6 +1233,9 @@ schedSignalDispatch(sched_root_task_t * __restrict roo
 	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));
@@ -1230,6 +1244,7 @@ schedSignalDispatch(sched_root_task_t * __restrict roo
 #else
 		root->root_sigthr = NULL;
 #endif
+		pthread_sigmask(SIG_SETMASK, &root->root_oldset, NULL);
 		return -1;
 	}