--- libaitsched/src/hooks.c	2011/08/05 15:52:00	1.1
+++ libaitsched/src/hooks.c	2011/10/04 13:50:05	1.2.2.2
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: hooks.c,v 1.1 2011/08/05 15:52:00 misho Exp $
+* $Id: hooks.c,v 1.2.2.2 2011/10/04 13:50:05 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -119,31 +119,30 @@ sched_hook_cancel(void *task, void *arg __unused)
 	struct sched_IO *io;
 	sched_task_t *t = task;
 	struct kevent chg[1];
-	struct timespec timeout;
+	struct timespec timeout = { 0, 0 };
 
 	if (!t || !t->task_root || !ROOT_DATA(t->task_root) || !ROOT_DATLEN(t->task_root))
 		return (void*) -1;
 	else
 		io = ROOT_DATA(t->task_root);
 
-	timespecclear(&timeout);
 	switch (t->task_type) {
 		case taskREAD:
-			if (FD_ISSET(TASK_FD(t), &io->wfd))
-				EV_SET(&chg[0], TASK_FD(t), EVFILT_WRITE, EV_ADD, 0, 0, &TASK_FD(t));
-			else
-				EV_SET(&chg[0], TASK_FD(t), EVFILT_WRITE, EV_DELETE, 0, 0, &TASK_FD(t));
+#ifdef __NetBSD__
+			EV_SET(&chg[0], TASK_FD(t), EVFILT_READ, EV_DELETE, 0, 0, (intptr_t) TASK_FD(t));
+#else
+			EV_SET(&chg[0], TASK_FD(t), EVFILT_READ, EV_DELETE, 0, 0, (void*) TASK_FD(t));
+#endif
 			kevent(t->task_root->root_kq, chg, 1, NULL, 0, &timeout);
-
 			FD_CLR(TASK_FD(t), &io->rfd);
 			break;
 		case taskWRITE:
-			if (FD_ISSET(TASK_FD(t), &io->rfd))
-				EV_SET(&chg[0], TASK_FD(t), EVFILT_READ, EV_ADD, 0, 0, &TASK_FD(t));
-			else
-				EV_SET(&chg[0], TASK_FD(t), EVFILT_READ, EV_DELETE, 0, 0, &TASK_FD(t));
+#ifdef __NetBSD__
+			EV_SET(&chg[0], TASK_FD(t), EVFILT_WRITE, EV_DELETE, 0, 0, (intptr_t) TASK_FD(t));
+#else
+			EV_SET(&chg[0], TASK_FD(t), EVFILT_WRITE, EV_DELETE, 0, 0, (void*) TASK_FD(t));
+#endif
 			kevent(t->task_root->root_kq, chg, 1, NULL, 0, &timeout);
-
 			FD_CLR(TASK_FD(t), &io->wfd);
 			break;
 		default:
@@ -165,7 +164,7 @@ sched_hook_read(void *task, void *arg __unused)
 	struct sched_IO *io;
 	sched_task_t *t = task;
 	struct kevent chg[1];
-	struct timespec timeout;
+	struct timespec timeout = { 0, 0 };
 
 	if (!t || !t->task_root || !ROOT_DATA(t->task_root) || !ROOT_DATLEN(t->task_root))
 		return (void*) -1;
@@ -177,10 +176,16 @@ sched_hook_read(void *task, void *arg __unused)
 	else
 		FD_SET(TASK_FD(t), &io->rfd);
 
-	timespecclear(&timeout);
-	EV_SET(&chg[0], TASK_FD(t), EVFILT_READ, EV_ADD, 0, 0, &TASK_FD(t));
+#ifdef __NetBSD__
+	EV_SET(&chg[0], TASK_FD(t), EVFILT_READ, EV_ADD, 0, 0, (intptr_t) TASK_FD(t));
+#else
+	EV_SET(&chg[0], TASK_FD(t), EVFILT_READ, EV_ADD, 0, 0, (void*) TASK_FD(t));
+#endif
 	if (kevent(t->task_root->root_kq, chg, 1, NULL, 0, &timeout) == -1) {
-		LOGERR;
+		if (t->task_root->root_hooks.hook_exec.exception)
+			t->task_root->root_hooks.hook_exec.exception(t->task_root, NULL);
+		else
+			LOGERR;
 		return (void*) -1;
 	}
 
@@ -199,7 +204,7 @@ sched_hook_write(void *task, void *arg __unused)
 	struct sched_IO *io;
 	sched_task_t *t = task;
 	struct kevent chg[1];
-	struct timespec timeout;
+	struct timespec timeout = { 0, 0 };
 
 	if (!t || !t->task_root || !ROOT_DATA(t->task_root) || !ROOT_DATLEN(t->task_root))
 		return (void*) -1;
@@ -211,10 +216,16 @@ sched_hook_write(void *task, void *arg __unused)
 	else
 		FD_SET(TASK_FD(t), &io->wfd);
 
-	timespecclear(&timeout);
-	EV_SET(&chg[0], TASK_FD(t), EVFILT_WRITE, EV_ADD, 0, 0, &TASK_FD(t));
+#ifdef __NetBSD__
+	EV_SET(&chg[0], TASK_FD(t), EVFILT_WRITE, EV_ADD, 0, 0, (intptr_t) TASK_FD(t));
+#else
+	EV_SET(&chg[0], TASK_FD(t), EVFILT_WRITE, EV_ADD, 0, 0, (void*) TASK_FD(t));
+#endif
 	if (kevent(t->task_root->root_kq, chg, 1, NULL, 0, &timeout) == -1) {
-		LOGERR;
+		if (t->task_root->root_hooks.hook_exec.exception)
+			t->task_root->root_hooks.hook_exec.exception(t->task_root, NULL);
+		else
+			LOGERR;
 		return (void*) -1;
 	}
 
@@ -303,11 +314,21 @@ retry:
 	} else	/* wait INFTIM */
 		timeout = NULL;
 	if ((en = kevent(r->root_kq, NULL, 0, res, KQ_EVENTS, timeout)) == -1) {
-		LOGERR;
+		if (r->root_hooks.hook_exec.exception) {
+			if (r->root_hooks.hook_exec.exception(r, NULL))
+				return NULL;	/* exit from scheduler */
+		} else
+			LOGERR;
+#ifdef NDEBUG
+		/* kevent no exit by error, if non-debug version */
 		goto retry;
+#else
+		/* diagnostic exit from scheduler if kevent error occur */
+		return NULL;
+#endif
 	}
 
-	timespecclear(&nw);
+	nw.tv_sec = nw.tv_nsec = 0;
 	/* Go and catch the cat into pipes ... */
 	for (i = 0; i < en; i++) {
 		memcpy(evt, &res[i], sizeof evt);
@@ -316,39 +337,63 @@ retry:
 		switch (res[i].filter) {
 			case EVFILT_READ:
 				TAILQ_FOREACH(task, &r->root_read, task_node) {
-					if (TASK_FD(task) != *((int*) res[i].udata))
+					if (TASK_FD(task) != ((int) res[i].udata))
 						continue;
 					/* remove read handle */
 					io = ROOT_DATA(task->task_root);
 					FD_CLR(TASK_FD(task), &io->rfd);
 
 					TAILQ_REMOVE(&r->root_read, task, task_node);
-					task->task_type = taskREADY;
-					TAILQ_INSERT_TAIL(&r->root_ready, task, task_node);
+					if (r->root_hooks.hook_exec.exception && res[i].flags & EV_EOF) {
+ 						if (r->root_hooks.hook_exec.exception(r, (void*) EV_EOF)) {
+							task->task_type = taskUNUSE;
+							TAILQ_INSERT_TAIL(&r->root_unuse, task, task_node);
+						} else {
+							task->task_type = taskREADY;
+							TAILQ_INSERT_TAIL(&r->root_ready, task, task_node);
+						}
+					} else {
+						task->task_type = taskREADY;
+						TAILQ_INSERT_TAIL(&r->root_ready, task, task_node);
+					}
 					break;
 				}
 				break;
 			case EVFILT_WRITE:
 				TAILQ_FOREACH(task, &r->root_write, task_node) {
-					if (TASK_FD(task) != *((int*) res[i].udata))
+					if (TASK_FD(task) != ((int) res[i].udata))
 						continue;
 					/* remove write handle */
 					io = ROOT_DATA(task->task_root);
 					FD_CLR(TASK_FD(task), &io->wfd);
 
 					TAILQ_REMOVE(&r->root_write, task, task_node);
-					task->task_type = taskREADY;
-					TAILQ_INSERT_TAIL(&r->root_ready, task, task_node);
+					if (r->root_hooks.hook_exec.exception && res[i].flags & EV_EOF) {
+ 						if (r->root_hooks.hook_exec.exception(r, (void*) EV_EOF)) {
+							task->task_type = taskUNUSE;
+							TAILQ_INSERT_TAIL(&r->root_unuse, task, task_node);
+						} else {
+							task->task_type = taskREADY;
+							TAILQ_INSERT_TAIL(&r->root_ready, task, task_node);
+						}
+					} else {
+						task->task_type = taskREADY;
+						TAILQ_INSERT_TAIL(&r->root_ready, task, task_node);
+					}
 					break;
 				}
 				break;
 		}
-
-		if (kevent(r->root_kq, evt, 1, NULL, 0, &nw) == -1)
-			LOGERR;
+		if (kevent(r->root_kq, evt, 1, NULL, 0, &nw) == -1) {
+			if (r->root_hooks.hook_exec.exception) {
+				if (r->root_hooks.hook_exec.exception(r, NULL))
+					return NULL;	/* exit from scheduler */
+			} else
+				LOGERR;
+		}
 	}
 
-	/* timer update */
+	/* timer update & put in ready queue */
 	clock_gettime(CLOCK_MONOTONIC, &nw);
 	now.tv_sec = nw.tv_sec;
 	now.tv_usec = nw.tv_nsec / 1000;
@@ -383,3 +428,32 @@ retry:
 	return task;
 }
 
+/*
+ * sched_hook_exception() - Default EXCEPTION hook
+ * @root = root task
+ * @arg = custom handling: if arg == EV_EOF or other value; default: arg == NULL log errno
+ * return: <0 errors and 0 ok
+ */
+void *
+sched_hook_exception(void *root, void *arg)
+{
+	sched_root_task_t *r = root;
+
+	if (!r || !ROOT_DATA(r) || !ROOT_DATLEN(r))
+		return NULL;
+
+	/* custom exception handling ... */
+	if (arg) {
+		if (arg == (void*) EV_EOF)
+			return NULL;
+		return (void*) -1;	/* raise scheduler error!!! */
+	}
+
+	/* if error hook exists */
+	if (r->root_hooks.hook_root.error)
+		return (r->root_hooks.hook_root.error(root, (void*) errno));
+
+	/* default case! */
+	LOGERR;
+	return NULL;
+}