From 6921fdc6d6134fd7aaf38ffc1686623eca9bbd18 Mon Sep 17 00:00:00 2001 From: Ryan Dwyer Date: Sun, 14 Oct 2018 12:28:38 +1000 Subject: Remove timerfd from loop implementation timerfd doesn't work on the BSDs, so this replaces it with a timespec for the expiry and uses a poll timeout to check the timers when needed. --- common/loop.c | 143 ++++++++++++++++++++++++++++++++++++++++------------------ 1 file changed, 100 insertions(+), 43 deletions(-) (limited to 'common/loop.c') diff --git a/common/loop.c b/common/loop.c index da3c2142..c358e212 100644 --- a/common/loop.c +++ b/common/loop.c @@ -1,18 +1,24 @@ +#include #include #include #include #include #include -#include +#include #include #include "list.h" #include "log.h" #include "loop.h" -struct loop_event { +struct loop_fd_event { void (*callback)(int fd, short mask, void *data); void *data; - bool is_timer; +}; + +struct loop_timer { + void (*callback)(void *data); + void *data; + struct timespec expiry; }; struct loop { @@ -20,7 +26,8 @@ struct loop { int fd_length; int fd_capacity; - list_t *events; // struct loop_event + list_t *fd_events; // struct loop_fd_event + list_t *timers; // struct loop_timer }; struct loop *loop_create(void) { @@ -31,86 +38,136 @@ struct loop *loop_create(void) { } loop->fd_capacity = 10; loop->fds = malloc(sizeof(struct pollfd) * loop->fd_capacity); - loop->events = create_list(); + loop->fd_events = create_list(); + loop->timers = create_list(); return loop; } void loop_destroy(struct loop *loop) { - list_foreach(loop->events, free); - list_free(loop->events); + list_foreach(loop->fd_events, free); + list_foreach(loop->timers, free); + list_free(loop->fd_events); + list_free(loop->timers); free(loop); } void loop_poll(struct loop *loop) { - poll(loop->fds, loop->fd_length, -1); + // Calculate next timer in ms + int ms = INT_MAX; + if (loop->timers->length) { + struct timespec now; + clock_gettime(CLOCK_MONOTONIC, &now); + for (int i = 0; i < loop->timers->length; ++i) { + struct loop_timer *timer = loop->timers->items[i]; + int timer_ms = (timer->expiry.tv_sec - now.tv_sec) * 1000; + timer_ms += (timer->expiry.tv_nsec - now.tv_nsec) / 1000000; + if (timer_ms < ms) { + ms = timer_ms; + } + } + } + poll(loop->fds, loop->fd_length, ms); + + // Dispatch fds for (int i = 0; i < loop->fd_length; ++i) { struct pollfd pfd = loop->fds[i]; - struct loop_event *event = loop->events->items[i]; + struct loop_fd_event *event = loop->fd_events->items[i]; // Always send these events unsigned events = pfd.events | POLLHUP | POLLERR; if (pfd.revents & events) { event->callback(pfd.fd, pfd.revents, event->data); + } + } - if (event->is_timer) { - loop_remove_event(loop, event); + // Dispatch timers + if (loop->timers->length) { + struct timespec now; + clock_gettime(CLOCK_MONOTONIC, &now); + for (int i = 0; i < loop->timers->length; ++i) { + struct loop_timer *timer = loop->timers->items[i]; + bool expired = timer->expiry.tv_sec < now.tv_sec || + (timer->expiry.tv_sec == now.tv_sec && + timer->expiry.tv_nsec < now.tv_nsec); + if (expired) { + timer->callback(timer->data); + loop_remove_timer(loop, timer); --i; } } } } -struct loop_event *loop_add_fd(struct loop *loop, int fd, short mask, +void loop_add_fd(struct loop *loop, int fd, short mask, void (*callback)(int fd, short mask, void *data), void *data) { + struct loop_fd_event *event = calloc(1, sizeof(struct loop_fd_event)); + if (!event) { + wlr_log(WLR_ERROR, "Unable to allocate memory for event"); + return; + } + event->callback = callback; + event->data = data; + list_add(loop->fd_events, event); + struct pollfd pfd = {fd, mask, 0}; if (loop->fd_length == loop->fd_capacity) { loop->fd_capacity += 10; - loop->fds = realloc(loop->fds, sizeof(struct pollfd) * loop->fd_capacity); + loop->fds = realloc(loop->fds, + sizeof(struct pollfd) * loop->fd_capacity); } loop->fds[loop->fd_length++] = pfd; +} - struct loop_event *event = calloc(1, sizeof(struct loop_event)); - event->callback = callback; - event->data = data; +struct loop_timer *loop_add_timer(struct loop *loop, int ms, + void (*callback)(void *data), void *data) { + struct loop_timer *timer = calloc(1, sizeof(struct loop_timer)); + if (!timer) { + wlr_log(WLR_ERROR, "Unable to allocate memory for timer"); + return NULL; + } + timer->callback = callback; + timer->data = data; - list_add(loop->events, event); + clock_gettime(CLOCK_MONOTONIC, &timer->expiry); + timer->expiry.tv_sec += ms / 1000; - return event; -} + long int nsec = (ms % 1000) * 1000000; + if (timer->expiry.tv_nsec + nsec >= 1000000000) { + timer->expiry.tv_sec++; + nsec -= 1000000000; + } + timer->expiry.tv_nsec += nsec; -struct loop_event *loop_add_timer(struct loop *loop, int ms, - void (*callback)(int fd, short mask, void *data), void *data) { - int fd = timerfd_create(CLOCK_MONOTONIC, 0); - struct itimerspec its; - its.it_interval.tv_sec = 0; - its.it_interval.tv_nsec = 0; - its.it_value.tv_sec = ms / 1000; - its.it_value.tv_nsec = (ms % 1000) * 1000000; - timerfd_settime(fd, 0, &its, NULL); - - struct loop_event *event = loop_add_fd(loop, fd, POLLIN, callback, data); - event->is_timer = true; - - return event; -} + list_add(loop->timers, timer); -bool loop_remove_event(struct loop *loop, struct loop_event *event) { - for (int i = 0; i < loop->events->length; ++i) { - if (loop->events->items[i] == event) { - list_del(loop->events, i); + return timer; +} - if (event->is_timer) { - close(loop->fds[i].fd); - } +bool loop_remove_fd(struct loop *loop, int fd) { + for (int i = 0; i < loop->fd_length; ++i) { + if (loop->fds[i].fd == fd) { + free(loop->fd_events->items[i]); + list_del(loop->fd_events, i); loop->fd_length--; - memmove(&loop->fds[i], &loop->fds[i + 1], sizeof(void*) * (loop->fd_length - i)); + memmove(&loop->fds[i], &loop->fds[i + 1], + sizeof(void*) * (loop->fd_length - i)); + + return true; + } + } + return false; +} - free(event); +bool loop_remove_timer(struct loop *loop, struct loop_timer *timer) { + for (int i = 0; i < loop->timers->length; ++i) { + if (loop->timers->items[i] == timer) { + list_del(loop->timers, i); + free(timer); return true; } } -- cgit v1.2.3