implement satd-timer

Signed-off-by: Mattias Andrée <maandree@member.fsf.org>

2 files changed, 122 insertions, 3 deletions

diff --git a/src/satd-diminished.c b/src/satd-diminished.c
index 059ca83..a0fbd8b 100644
--- a/src/satd-diminished.c
+++ b/src/satd-diminished.c
@@ -150,6 +150,28 @@ is_timer_set(int fd)
 
 
 /**
+ * If a timer has expired, unset it.
+ * 
+ * @param  fd     The file descriptor of the timer.
+ * @param  fdset  Set that shall contain `fd` iff it has expired.
+ * @return        0 on sucess, -1 on error.
+ */
+static int
+test_timer(int fd, const fd_set *fdset)
+{
+	int64_t _overrun;
+	struct itimerspec spec = {
+		.it_interval.tv_sec  = 0, .it_value.tv_sec  = 0,
+		.it_interval.tv_nsec = 0, .it_value.tv_nsec = 0,
+	};
+	if (!FD_ISSET(BOOT_FILENO, fdset))                return 0;
+	if (read(BOOT_FILENO, &_overrun, (size_t)8) < 8)  return -1;
+	if (timer_pid == NO_TIMER_SPAWED)                 return 0;
+	return timerfd_settime(fd, TFD_TIMER_ABSTIME, &spec, NULL);
+}
+
+
+/**
  * The sat daemon.
  * 
  * @param   argc  Should be 3.
@@ -165,7 +187,6 @@ main(int argc, char *argv[], char *envp[])
 	int fd = -1, rc = 0, accepted = 0, r;
 	unsigned char type;
 	fd_set fdset;
-	int64_t _overrun;
 
 	/* Set up signal handlers. */
 	t (signal(SIGHUP,  sighandler) == SIG_ERR);
@@ -193,8 +214,8 @@ not_done:
 	if (select(REAL_FILENO + 1, &fdset, NULL, NULL, NULL) == -1) {
 		t (errno != EINTR);
 	}
-	if (FD_ISSET(BOOT_FILENO, &fdset))  t (read(BOOT_FILENO, &_overrun, (size_t)8) < 8);
-	if (FD_ISSET(REAL_FILENO, &fdset))  t (read(REAL_FILENO, &_overrun, (size_t)8) < 8);
+	t (test_timer(BOOT_FILENO, &fdset));
+	t (test_timer(REAL_FILENO, &fdset));
 	if (!FD_ISSET(SOCK_FILENO, &fdset))
 		goto again;
 	if (fd = accept(SOCK_FILENO, NULL, NULL), fd == -1) {
diff --git a/src/satd-timer.c b/src/satd-timer.c
new file mode 100644
index 0000000..4550465
--- /dev/null
+++ b/src/satd-timer.c
@@ -0,0 +1,98 @@
+/**
+ * Copyright © 2015  Mattias Andrée <maandree@member.fsf.org>
+ * 
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ * 
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ * 
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+ * DEALINGS IN THE SOFTWARE.
+ */
+#include "daemon.h"
+#include <time.h>
+#include <sys/timerfd.h>
+
+
+
+/**
+ * Pretty self-explanatory.
+ */
+#ifdef __GNUC__
+__attribute__((__pure__))
+#endif
+static inline int
+timecmp(const struct timespec *a, const struct timespec *b)
+{
+	if (a->tv_sec  != b->tv_sec)   return (a->tv_sec  < b->tv_sec  ? -1 : +1);
+	if (a->tv_nsec != b->tv_nsec)  return (a->tv_nsec < b->tv_nsec ? -1 : +1);
+	return 0;
+}
+
+
+/**
+ * Subroutine to the sat daemon: list jobs.
+ * 
+ * @param   argc  Should be 3.
+ * @param   argv  The name of the process, the pathname of the socket,
+ *                and the pathname to the state file.
+ * @return  0     The process was successful.
+ * @return  1     The process failed queuing the job.
+ */
+int
+main(int argc, char *argv[])
+{
+#define TIME(job, suffix)  ((job)->clk == CLOCK_BOOTTIME ? &boot##suffix : &real##suffix)
+
+	char jobno[3 * sizeof(size_t) + 1];
+	struct itimerspec bootspec;
+	struct itimerspec realspec;
+	struct timespec bootnow;
+	struct timespec realnow;
+	struct job **jobs = NULL;
+	struct job **job;
+	int rc = 0;
+
+	/* Get current expiration time. */
+	t (timerfd_gettime(BOOT_FILENO, &bootspec));
+	t (timerfd_gettime(REAL_FILENO, &realspec));
+
+	/* Run expired jobs, and find new expiration times. */
+	t (clock_gettime(CLOCK_BOOTTIME, &bootnow));
+	t (clock_gettime(CLOCK_REALTIME, &realnow));
+	t (!(jobs = get_jobs()));
+	for (job = jobs; *job; job++) {
+		if (timecmp(&(job[0]->ts), TIME(*job, now)) >= 0) {
+			sprintf(jobno, "%zu", job[0]->no);
+			remove_job(jobno, 2);
+		} else if (timecmp(&(job[0]->ts), &(TIME(*job, spec)->it_value)) > 0) {
+			TIME(*job, spec)->it_value = job[0]->ts;
+		}
+	}
+
+	/* Update expiration time. */
+	t (timerfd_settime(BOOT_FILENO, TFD_TIMER_ABSTIME, &bootspec, NULL));
+	t (timerfd_settime(REAL_FILENO, TFD_TIMER_ABSTIME, &realspec, NULL));
+
+done:
+	for (job = jobs; *job; job++)
+		free(*job);
+	free(jobs);
+	return rc;
+fail:
+	perror(argv[0]);
+	rc = 1;
+	goto done;
+	(void) argc;
+}
+