--- libaitsched/src/hooks.c	2012/08/21 12:54:39	1.14
+++ libaitsched/src/hooks.c	2012/08/23 00:31:41	1.14.2.5
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: hooks.c,v 1.14 2012/08/21 12:54:39 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,9 +209,17 @@ 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
-			TASK_UNLOCK(t);
 		default:
 			return NULL;
 	}
@@ -233,13 +241,17 @@ 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;
 	}
 
@@ -247,6 +259,7 @@ sched_hook_thread(void *task, void *arg)
 		TASK_LOCK(t);
 
 	TASK_VAL(t) = (u_long) tid;
+	pthread_sigmask(SIG_SETMASK, &o, NULL);
 	return NULL;
 }
 #endif