--- libaitsched/src/aitsched.c 2023/02/23 15:41:45 1.30.2.3 +++ libaitsched/src/aitsched.c 2024/02/26 12:58:55 1.36 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: aitsched.c,v 1.30.2.3 2023/02/23 15:41:45 misho Exp $ +* $Id: aitsched.c,v 1.36 2024/02/26 12:58:55 misho Exp $ * ************************************************************************** The ELWIX and AITNET software is distributed under the following @@ -12,7 +12,7 @@ terms: All of the documentation and software included in the ELWIX and AITNET Releases is copyrighted by ELWIX - Sofia/Bulgaria -Copyright 2004 - 2023 +Copyright 2004 - 2024 by Michael Pounov . All rights reserved. Redistribution and use in source and binary forms, with or without @@ -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) { @@ -394,7 +412,12 @@ schedEnd(sched_root_task_t ** __restrict root) #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); @@ -1123,14 +1146,39 @@ schedResumeby(sched_root_task_t * __restrict root, u_c static void * _sched_sigDisp(void *arg) { - sched_root_task_t *sched = arg; + sched_root_task_t *root = arg; + sched_task_t *task, *tmp; + int sig, flg; sigset_t ss; - int sig; sigfillset(&ss); - while (sched->root_sigthr) { - sigwait(&ss, &sig); + 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; @@ -1147,6 +1195,7 @@ 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; @@ -1154,8 +1203,13 @@ schedSignalDispatch(sched_root_task_t * __restrict roo pthread_attr_t attr; #endif +#if SUP_ENABLE == KQ_SUPPORT + return 0; +#endif + if (!on) { - pthread_cancel(root->root_sigthr); + if (root->root_sigthr) + pthread_cancel(root->root_sigthr); #ifdef __linux__ root->root_sigthr = 0L; #else @@ -1164,6 +1218,14 @@ schedSignalDispatch(sched_root_task_t * __restrict roo return 0; } + /* if we have already started signal thread, then skip routine */ + if (root->root_sigthr) + 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 @@ -1174,9 +1236,16 @@ schedSignalDispatch(sched_root_task_t * __restrict roo 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; }