--- libaitsched/src/hooks.c	2012/07/24 14:06:11	1.10
+++ libaitsched/src/hooks.c	2012/08/01 14:11:43	1.10.2.2
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: hooks.c,v 1.10 2012/07/24 14:06:11 misho Exp $
+* $Id: hooks.c,v 1.10.2.2 2012/08/01 14:11:43 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -156,6 +156,19 @@ sched_hook_cancel(void *task, void *arg __unused)
 			EV_SET(&chg[0], TASK_VAL(t), EVFILT_SIGNAL, EV_DELETE, 0, 0, (void*) TASK_VAL(t));
 #endif
 			break;
+#ifdef EVFILT_AIO
+		case taskAIO:
+#ifdef __NetBSD__
+			EV_SET(&chg[0], TASK_VAL(t), EVFILT_AIO, EV_DELETE, 0, 0, (intptr_t) TASK_VAL(t));
+#else
+			EV_SET(&chg[0], TASK_VAL(t), EVFILT_AIO, EV_DELETE, 0, 0, (void*) TASK_VAL(t));
+#endif
+			if (TASK_VAL(t)) {
+				free((void*) TASK_VAL(t));
+				TASK_VAL(t) = 0;
+			}
+			break;
+#endif
 #ifdef EVFILT_USER
 		case taskUSER:
 #ifdef __NetBSD__
@@ -435,6 +448,9 @@ sched_hook_fetch(void *root, void *arg __unused)
 	struct kevent evt[1], res[KQ_EVENTS];
 	register int i, flg;
 	int en;
+#ifdef EVFILT_AIO
+	struct aiocb *acb;
+#endif
 
 	if (!r)
 		return NULL;
@@ -752,6 +768,38 @@ sched_hook_fetch(void *root, void *arg __unused)
 				if (flg > 1)
 					evt->flags ^= evt->flags;
 				break;
+#ifdef EVFILT_AIO
+			case EVFILT_AIO:
+				flg = 0;
+				acb = (struct aiocb*) TASK_VAL(task);
+				TAILQ_FOREACH_SAFE(task, &r->root_aio, task_node, tmp) {
+					if (acb != ((struct aiocb*) res[i].ident) || 
+							acb->aio_sigevent.sigev_value.sival_ptr != res[i].udata)
+						continue;
+					else
+						flg++;
+					/* remove user handle */
+#ifdef HAVE_LIBPTHREAD
+					pthread_mutex_lock(&r->root_mtx[taskAIO]);
+#endif
+					TAILQ_REMOVE(&r->root_aio, task, task_node);
+#ifdef HAVE_LIBPTHREAD
+					pthread_mutex_unlock(&r->root_mtx[taskAIO]);
+#endif
+					task->task_type = taskREADY;
+#ifdef HAVE_LIBPTHREAD
+					pthread_mutex_lock(&r->root_mtx[taskREADY]);
+#endif
+					TAILQ_INSERT_TAIL(&r->root_ready, task, task_node);
+#ifdef HAVE_LIBPTHREAD
+					pthread_mutex_unlock(&r->root_mtx[taskREADY]);
+#endif
+				}
+				/* if match at least 2, don't remove resouce of event */
+				if (flg > 1)
+					evt->flags ^= evt->flags;
+				break;
+#endif	/* EVFILT_AIO */
 #ifdef EVFILT_USER
 			case EVFILT_USER:
 				flg = 0;
@@ -784,7 +832,7 @@ sched_hook_fetch(void *root, void *arg __unused)
 				if (flg > 1)
 					evt->flags ^= evt->flags;
 				break;
-#endif
+#endif	/* EVFILT_USER */
 		}
 		if (kevent(r->root_kq, evt, 1, NULL, 0, &now) == -1) {
 			if (r->root_hooks.hook_exec.exception) {