--- libaitsched/src/hooks.c 2012/08/21 11:45:35 1.13.2.2 +++ libaitsched/src/hooks.c 2012/08/23 00:31:41 1.14.2.5 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: hooks.c,v 1.13.2.2 2012/08/21 11:45:35 misho Exp $ +* $Id: hooks.c,v 1.14.2.5 2012/08/23 00:31:41 misho Exp $ * ************************************************************************** The ELWIX and AITNET software is distributed under the following @@ -107,11 +107,11 @@ sched_hook_cancel(void *task, void *arg __unused) sched_task_t *t = task; struct kevent chg[1]; struct timespec timeout = { 0, 0 }; + register int i = 0; #ifdef AIO_SUPPORT struct aiocb *acb; #ifdef EVFILT_LIO struct aiocb **acbs; - register int i; #endif /* EVFILT_LIO */ #endif /* AIO_SUPPORT */ @@ -209,7 +209,16 @@ sched_hook_cancel(void *task, void *arg __unused) #endif case taskTHREAD: #ifdef HAVE_LIBPTHREAD - pthread_cancel((pthread_t) TASK_VAL(t)); + for (i = 0; i < MAX_TASK_MISS; i++) + if (!pthread_cancel((pthread_t) TASK_VAL(t))) { + /* joinable thread */ + if (TASK_FLAG(t) == PTHREAD_CREATE_JOINABLE) + schedTask(TASK_ROOT(t), _sched_threadJoin, + TASK_ARG(t), TASK_VAL(t), + TASK_DATA(t), TASK_DATLEN(t)); + return NULL; + } + return (void*) -1; #endif default: return NULL; @@ -218,6 +227,42 @@ sched_hook_cancel(void *task, void *arg __unused) kevent(TASK_ROOT(t)->root_kq, chg, 1, NULL, 0, &timeout); return NULL; } + +#ifdef HAVE_LIBPTHREAD +/* + * sched_hook_thread() - Default THREAD hook + * + * @task = current task + * @arg = pthread attributes + * return: <0 errors and 0 ok + */ +void * +sched_hook_thread(void *task, void *arg) +{ + sched_task_t *t = task; + pthread_t tid; + sigset_t s, o; + + if (!t || !TASK_ROOT(t)) + return (void*) -1; + + sigfillset(&s); + pthread_sigmask(SIG_BLOCK, &s, &o); + if (pthread_create(&tid, (pthread_attr_t*) arg, + (void *(*)(void*)) TASK_FUNC(t), t)) { + LOGERR; + pthread_sigmask(SIG_SETMASK, &o, NULL); + return (void*) -1; + } + + if (!TASK_ISLOCKED(t)) + TASK_LOCK(t); + + TASK_VAL(t) = (u_long) tid; + pthread_sigmask(SIG_SETMASK, &o, NULL); + return NULL; +} +#endif /* * sched_hook_read() - Default READ hook