X-Git-Url: https://git.xandkar.net/?a=blobdiff_plain;f=x5%2Fkhatus.c;h=0ecac5afd4adb25055c3de5f5ba85528194a911f;hb=55883d9a0ab2ca4956bda685c219557f8bd940f1;hp=18abc19a7a7078fa21e4ac7bdd99333854b89d96;hpb=0741fd047bd22883b0b8a13fbeea3cec6bdcc3f5;p=khatus.git

diff --git a/x5/khatus.c b/x5/khatus.c
index 18abc19..0ecac5a 100644
--- a/x5/khatus.c
+++ b/x5/khatus.c
@@ -316,7 +316,7 @@ opts_parse(Config *cfg, int argc, char *argv[])
 }
 
 void
-fifo_expire_one(Fifo *f, struct timespec t, char *buf)
+fifo_expire(Fifo *f, struct timespec t, char *buf)
 {
 	struct timespec td;
 
@@ -328,15 +328,6 @@ fifo_expire_one(Fifo *f, struct timespec t, char *buf)
 	}
 }
 
-void
-fifo_expire_all(Config *cfg, struct timespec t, char *buf)
-{
-	Fifo *f;
-
-	for (f = cfg->fifos; f; f = f->next)
-		fifo_expire_one(f, t, buf);
-}
-
 void
 fifo_read_error(Fifo *f, char *buf)
 {
@@ -397,7 +388,7 @@ fifo_read_one(Fifo *f, struct timespec t, char *buf)
 }
 
 void
-fifo_read_all(Config *cfg, char *buf)
+fifo_read_all(Config *cfg, struct timespec *ti, char *buf)
 {
 	fd_set fds;
 	int maxfd = -1;
@@ -436,14 +427,23 @@ fifo_read_all(Config *cfg, char *buf)
 		FD_SET(f->fd, &fds);
 	}
 	debug("selecting...\n");
-	ready = select(maxfd + 1, &fds, NULL, NULL, NULL);
+	ready = pselect(maxfd + 1, &fds, NULL, NULL, ti, NULL);
 	debug("ready: %d\n", ready);
-	assert(ready != 0);
-	if (ready < 0)
-		/* TODO: Do we really want to fail here? */
-		fatal("%s", strerror(errno));
 	clock_gettime(CLOCK_MONOTONIC, &t);
-	while (ready) {
+	if (ready == -1) {
+		switch (errno) {
+		case EINTR:
+			error("pselect temp failure: %d, errno: %d, msg: %s\n",
+			    ready, errno, strerror(errno));
+			/* TODO: Reconsider what to do here. */
+			return;
+		default:
+			fatal("pselect failed: %d, errno: %d, msg: %s\n",
+			    ready, errno, strerror(errno));
+		}
+	}
+	/* At-least-once ensures that expiries are still checked on timeouts. */
+	do {
 		for (Fifo *f = cfg->fifos; f; f = f->next) {
 			if (FD_ISSET(f->fd, &fds)) {
 				debug("reading: %s\n", f->name);
@@ -474,9 +474,11 @@ fifo_read_all(Config *cfg, char *buf)
 				default:
 					assert(0);
 				}
+			} else {
+				fifo_expire(f, t, buf);
 			}
 		}
-	}
+	} while (ready);
 	assert(ready == 0);
 }
 
@@ -557,11 +559,7 @@ main(int argc, char *argv[])
 	/* TODO: Handle signals */
 	for (;;) {
 		clock_gettime(CLOCK_MONOTONIC, &t0); // FIXME: check errors
-		/* TODO: Set timeout on fifo_read_all based on diff of last time of
-		 *       fifo_read_all and desired time of next TTL check?
-		 */
-		/* TODO: How long to wait on IO? Max TTL? */
-		fifo_read_all(cfg, buf);
+		fifo_read_all(cfg, &ti, buf);
 		if (cfg->output_to_x_root_window) {
 			if (XStoreName(display, DefaultRootWindow(display), buf) < 0)
 				fatal("XStoreName failed.\n");
@@ -570,14 +568,6 @@ main(int argc, char *argv[])
 			puts(buf);
 			fflush(stdout);
 		}
-
-		/*
-		 * This is a good place for expiry check, since we're about to
-		 * sleep anyway and the time taken by the check will be
-		 * subtracted from the sleep period.
-		 */
-		fifo_expire_all(cfg, t0, buf);
-
 		clock_gettime(CLOCK_MONOTONIC, &t1); // FIXME: check errors
 		timespecsub(&t1, &t0, &td);
 		debug("td {tv_sec = %ld, tv_nsec = %ld}\n", td.tv_sec, td.tv_nsec);