--- libaitsched/src/hooks.c	2012/08/22 13:37:23	1.14.2.2
+++ libaitsched/src/hooks.c	2012/08/22 23:47:22	1.14.2.4
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: hooks.c,v 1.14.2.2 2012/08/22 13:37:23 misho Exp $
+* $Id: hooks.c,v 1.14.2.4 2012/08/22 23:47:22 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,14 @@ sched_hook_cancel(void *task, void *arg __unused)
 #endif
 		case taskTHREAD:
 #ifdef HAVE_LIBPTHREAD
-			pthread_cancel((pthread_t) TASK_VAL(t));
+			/* try to wait for valid thread id */
+			for (i = 0; i < MAX_TASK_MISS && 
+					pthread_kill((pthread_t) TASK_VAL(t), 0); i++)
+				usleep(10);
+			/* try to sure thread is cancelled */
+			for (i = 0; i < MAX_TASK_MISS && 
+					!pthread_kill((pthread_t) TASK_VAL(t), 0); i++)
+				pthread_cancel((pthread_t) TASK_VAL(t));
 			if (TASK_FLAG(t) == PTHREAD_CREATE_JOINABLE)	/* joinable thread */
 				schedTask(TASK_ROOT(t), _sched_threadJoin, TASK_ARG(t),
 						TASK_VAL(t), TASK_DATA(t), TASK_DATLEN(t));
@@ -249,12 +256,12 @@ sched_hook_thread(void *task, void *arg)
 		pthread_sigmask(SIG_SETMASK, &os, NULL);
 		return (void*) -1;
 	}
-	pthread_sigmask(SIG_SETMASK, &os, NULL);
 
 	if (!TASK_ISLOCKED(t))
 		TASK_LOCK(t);
 
 	TASK_VAL(t) = (u_long) tid;
+	pthread_sigmask(SIG_SETMASK, &os, NULL);
 	return NULL;
 }
 #endif