io: io_time_override to insert fake times.

[ccan] / ccan / io / poll.c

diff --git a/ccan/io/poll.c b/ccan/io/poll.c
index 078de1403ff8a886db059f14b2437799bab7edf3..cddc3cacd266e5b80392d91af367d689da093bd6 100644 (file)
--- a/ccan/io/poll.c
+++ b/ccan/io/poll.c
@@ -8,14 +8,22 @@
 #include <sys/socket.h>
 #include <limits.h>
 #include <errno.h>
-#include <ccan/list/list.h>
 #include <ccan/time/time.h>
 #include <ccan/timer/timer.h>
 
 static size_t num_fds = 0, max_fds = 0, num_waiting = 0;
 static struct pollfd *pollfds = NULL;
 static struct fd **fds = NULL;
-static struct io_conn *closing = NULL, *always = NULL;
+static LIST_HEAD(closing);
+static LIST_HEAD(always);
+static struct timeabs (*nowfn)(void) = time_now;
+
+struct timeabs (*io_time_override(struct timeabs (*now)(void)))(void)
+{
+       struct timeabs (*old)(void) = nowfn;
+       nowfn = now;
+       return old;
+}
 
 static bool add_fd(struct fd *fd, short events)
 {
@@ -78,6 +86,12 @@ static void del_fd(struct fd *fd)
        }
        num_fds--;
        fd->backend_info = -1;
+
+       /* Closing a local socket doesn't wake poll() because other end
+        * has them open.  See 2.6.  When should I use shutdown()?
+        * in http://www.faqs.org/faqs/unix-faq/socket/ */
+       shutdown(fd->fd, SHUT_RDWR);
+
        close(fd->fd);
 }
 
@@ -90,31 +104,21 @@ bool add_listener(struct io_listener *l)
 
 void remove_from_always(struct io_conn *conn)
 {
-       struct io_conn **p = &always;
-
-       while (*p != conn)
-               p = &(*p)->list;
-
-       *p = conn->list;
+       list_del_init(&conn->always);
 }
 
 void backend_new_closing(struct io_conn *conn)
 {
-       /* Already on always list?  Remove it. */
-       if (conn->list)
-               remove_from_always(conn);
-
-       conn->list = closing;
-       closing = conn;
+       /* In case it's on always list, remove it. */
+       list_del_init(&conn->always);
+       list_add_tail(&closing, &conn->closing);
 }
 
 void backend_new_always(struct io_conn *conn)
 {
-       /* May already be in always list (other plan), or closing. */
-       if (!conn->list) {
-               conn->list = always;
-               always = conn;
-       }
+       /* In case it's already in always list. */
+       list_del(&conn->always);
+       list_add_tail(&always, &conn->always);
 }
 
 void backend_new_plan(struct io_conn *conn)
@@ -152,11 +156,11 @@ void backend_wake(const void *wait)
                c = (void *)fds[i];
                if (c->plan[IO_IN].status == IO_WAITING
                    && c->plan[IO_IN].arg.u1.const_vp == wait)
-                       io_do_wakeup(c, &c->plan[IO_IN]);
+                       io_do_wakeup(c, IO_IN);
 
                if (c->plan[IO_OUT].status == IO_WAITING
                    && c->plan[IO_OUT].arg.u1.const_vp == wait)
-                       io_do_wakeup(c, &c->plan[IO_OUT]);
+                       io_do_wakeup(c, IO_OUT);
        }
 }
 
@@ -196,14 +200,12 @@ static void accept_conn(struct io_listener *l)
 static bool close_conns(void)
 {
        bool ret = false;
+       struct io_conn *conn;
 
-       while (closing) {
-               struct io_conn *conn = closing;
-
+       while ((conn = list_pop(&closing, struct io_conn, closing)) != NULL) {
                assert(conn->plan[IO_IN].status == IO_CLOSING);
                assert(conn->plan[IO_OUT].status == IO_CLOSING);
 
-               closing = closing->list;
                del_conn(conn);
                ret = true;
        }
@@ -213,16 +215,14 @@ static bool close_conns(void)
 static bool handle_always(void)
 {
        bool ret = false;
+       struct io_conn *conn;
 
-       while (always) {
-               struct io_conn *conn = always;
-
+       while ((conn = list_pop(&always, struct io_conn, always)) != NULL) {
                assert(conn->plan[IO_IN].status == IO_ALWAYS
                       || conn->plan[IO_OUT].status == IO_ALWAYS);
 
-               /* Remove from list, and mark it so it knows that. */
-               always = always->list;
-               conn->list = NULL;
+               /* Re-initialize, for next time. */
+               list_node_init(&conn->always);
                io_do_always(conn);
                ret = true;
        }
@@ -230,16 +230,16 @@ static bool handle_always(void)
 }
 
 /* This is the main loop. */
-void *io_loop(struct timers *timers, struct list_head *expired)
+void *io_loop(struct timers *timers, struct timer **expired)
 {
        void *ret;
 
        /* if timers is NULL, expired must be.  If not, not. */
        assert(!timers == !expired);
 
-       /* Make sure this is empty if we exit for some other reason. */
+       /* Make sure this is NULL if we exit for some other reason. */
        if (expired)
-               list_head_init(expired);
+               *expired = NULL;
 
        while (!io_loop_return) {
                int i, r, ms_timeout = -1;
@@ -264,11 +264,11 @@ void *io_loop(struct timers *timers, struct list_head *expired)
                if (timers) {
                        struct timeabs now, first;
 
-                       now = time_now();
+                       now = nowfn();
 
                        /* Call functions for expired timers. */
-                       timers_expire(timers, now, expired);
-                       if (!list_empty(expired))
+                       *expired = timers_expire(timers, now);
+                       if (*expired)
                                break;
 
                        /* Now figure out how long to wait for the next one. */