uloop: add support for interval timers
authorJo-Philipp Wich <jo@mein.io>
Sat, 14 Oct 2023 22:17:36 +0000 (00:17 +0200)
committerFelix Fietkau <nbd@nbd.name>
Thu, 2 Nov 2023 16:49:55 +0000 (17:49 +0100)
So far, the only way to implement periodic interval timers was to use
one-shot uloop_timeout timers which are rearmed within their completion
callback immediately on expiration.

While simple, this approach is not very precise and interval lengths will
slowly drift over time, due to callback execution overhead, scheduling
granularity etc.

In order to make uloop provide stable and precise interval timer
capabilities, this commit introduces a new `uloop_interval` structure
along with the new related `uloop_interval_set()`, `uloop_interval_cancel()`
and `uloop_interval_remaining()` api functions.

Periodic timers are implemented using the timerfd facility an Linux and
kqueue EVFILT_TIMER events on macOS/BSD.

The Lua binding has been updated to include support for the new timer type
as well.

Signed-off-by: Jo-Philipp Wich <jo@mein.io>
examples/uloop-example.lua
lua/uloop.c
uloop-epoll.c
uloop-kqueue.c
uloop.c
uloop.h

index 511b9ea0b3190560a6d181594d4ce10b12c6588d..f3aef60debad4a2a8f19611e001ef9dd9ce1fba8 100755 (executable)
@@ -24,6 +24,23 @@ uloop.timer(function() print("2000 ms timer run"); end, 2000)
 -- timer example 3 (will never run)
 uloop.timer(function() print("3000 ms timer run"); end, 3000):cancel()
 
+-- periodic interval timer
+local intv
+intv = uloop.interval(function()
+       print(string.format("Interval expiration #%d - %dms until next expiration",
+               intv:expirations(), intv:remaining()))
+
+       -- after 5 expirations, lower interval to 500ms
+       if intv:expirations() >= 5 then
+               intv:set(500)
+       end
+
+       -- cancel after 10 expirations
+       if intv:expirations() >= 10 then
+               intv:cancel()
+       end
+end, 1000)
+
 -- process
 function p1(r)
        print("Process 1 completed")
index 7e9ed1072bcff098af732f78ea4d097297737680..45c9bc7ad23ac5945801ff6ca3f08ffa39c6f729 100644 (file)
@@ -41,6 +41,11 @@ struct lua_uloop_process {
        int r;
 };
 
+struct lua_uloop_interval {
+       struct uloop_interval i;
+       int r;
+};
+
 static lua_State *state;
 
 static void *
@@ -382,6 +387,112 @@ static int ul_process(lua_State *L)
        return 1;
 }
 
+static void ul_interval_cb(struct uloop_interval *i)
+{
+       struct lua_uloop_interval *intv = container_of(i, struct lua_uloop_interval, i);
+
+       lua_getglobal(state, "__uloop_cb");
+       lua_rawgeti(state, -1, intv->r);
+       lua_remove(state, -2);
+
+       lua_call(state, 0, 0);
+}
+
+static int ul_interval_set(lua_State *L)
+{
+       struct lua_uloop_interval *intv;
+       double set;
+
+       if (!lua_isnumber(L, -1)) {
+               lua_pushstring(L, "invalid arg list");
+               lua_error(L);
+
+               return 0;
+       }
+
+       set = lua_tointeger(L, -1);
+       intv = lua_touserdata(L, 1);
+       uloop_interval_set(&intv->i, set);
+
+       return 1;
+}
+
+static int ul_interval_expirations(lua_State *L)
+{
+       struct lua_uloop_interval *intv;
+
+       intv = lua_touserdata(L, 1);
+       lua_pushinteger(L, intv->i.expirations);
+       return 1;
+}
+
+static int ul_interval_remaining(lua_State *L)
+{
+       struct lua_uloop_interval *intv;
+
+       intv = lua_touserdata(L, 1);
+       lua_pushnumber(L, uloop_interval_remaining(&intv->i));
+       return 1;
+}
+
+static int ul_interval_free(lua_State *L)
+{
+       struct lua_uloop_interval *intv = lua_touserdata(L, 1);
+
+       uloop_interval_cancel(&intv->i);
+
+       /* obj.__index.__gc = nil , make sure executing only once*/
+       lua_getfield(L, -1, "__index");
+       lua_pushstring(L, "__gc");
+       lua_pushnil(L);
+       lua_settable(L, -3);
+
+       lua_getglobal(state, "__uloop_cb");
+       luaL_unref(state, -1, intv->r);
+
+       return 1;
+}
+
+static const luaL_Reg interval_m[] = {
+       { "set", ul_interval_set },
+       { "expirations", ul_interval_expirations },
+       { "remaining", ul_interval_remaining },
+       { "cancel", ul_interval_free },
+       { NULL, NULL }
+};
+
+static int ul_interval(lua_State *L)
+{
+       struct lua_uloop_interval *intv;
+       unsigned int set = 0;
+       int ref;
+
+       if (lua_isnumber(L, -1)) {
+               set = lua_tointeger(L, -1);
+               lua_pop(L, 1);
+       }
+
+       if (!lua_isfunction(L, -1)) {
+               lua_pushstring(L, "invalid arg list");
+               lua_error(L);
+
+               return 0;
+       }
+
+       lua_getglobal(L, "__uloop_cb");
+       lua_pushvalue(L, -2);
+       ref = luaL_ref(L, -2);
+
+       intv = ul_create_userdata(L, sizeof(*intv), interval_m, ul_interval_free);
+       intv->r = ref;
+       intv->i.cb = ul_interval_cb;
+
+       if (set)
+               uloop_interval_set(&intv->i, set);
+
+       return 1;
+}
+
 static int ul_init(lua_State *L)
 {
        uloop_init();
@@ -410,6 +521,7 @@ static luaL_reg uloop_func[] = {
        {"timer", ul_timer},
        {"process", ul_process},
        {"fd_add", ul_ufd_add},
+       {"interval", ul_interval},
        {"cancel", ul_end},
        {NULL, NULL},
 };
index 70e45e448e738632acd425e3f567a117b9feafbf..cf5733fa973b3ec40723f268c12f0c32bdea4443 100644 (file)
@@ -104,3 +104,83 @@ static int uloop_fetch_events(int timeout)
 
        return nfds;
 }
+
+static void dispatch_timer(struct uloop_fd *u, unsigned int events)
+{
+       if (!(events & ULOOP_READ))
+               return;
+
+       uint64_t fired;
+
+       if (read(u->fd, &fired, sizeof(fired)) != sizeof(fired))
+               return;
+
+       struct uloop_interval *tm = container_of(u, struct uloop_interval, private.ufd);
+
+       tm->expirations += fired;
+       tm->cb(tm);
+}
+
+static int timer_register(struct uloop_interval *tm, unsigned int msecs)
+{
+       if (!tm->private.ufd.registered) {
+               int fd = timerfd_create(CLOCK_MONOTONIC, TFD_CLOEXEC|TFD_NONBLOCK);
+
+               if (fd == -1)
+                       return -1;
+
+               tm->private.ufd.fd = fd;
+               tm->private.ufd.cb = dispatch_timer;
+       }
+
+       struct itimerspec spec = {
+               .it_value = {
+                       .tv_sec = msecs / 1000,
+                       .tv_nsec = (msecs % 1000) * 1000000
+               },
+               .it_interval = {
+                       .tv_sec = msecs / 1000,
+                       .tv_nsec = (msecs % 1000) * 1000000
+               }
+       };
+
+       if (timerfd_settime(tm->private.ufd.fd, 0, &spec, NULL) == -1)
+               goto err;
+
+       if (uloop_fd_add(&tm->private.ufd, ULOOP_READ) == -1)
+               goto err;
+
+       return 0;
+
+err:
+       uloop_fd_delete(&tm->private.ufd);
+       close(tm->private.ufd.fd);
+       memset(&tm->private.ufd, 0, sizeof(tm->private.ufd));
+
+       return -1;
+}
+
+static int timer_remove(struct uloop_interval *tm)
+{
+       int ret = __uloop_fd_delete(&tm->private.ufd);
+
+       if (ret == 0) {
+               close(tm->private.ufd.fd);
+               memset(&tm->private.ufd, 0, sizeof(tm->private.ufd));
+       }
+
+       return ret;
+}
+
+static int64_t timer_next(struct uloop_interval *tm)
+{
+       struct itimerspec spec;
+
+       if (!tm->private.ufd.registered)
+               return -1;
+
+       if (timerfd_gettime(tm->private.ufd.fd, &spec) == -1)
+               return -1;
+
+       return spec.it_value.tv_sec * 1000 + spec.it_value.tv_nsec / 1000000;
+}
index c1275b05a781140a5388e0712b9aba4f1f6819ae..a48cca0725d04e7410c7e0dab74ec50cae177eec 100644 (file)
@@ -103,6 +103,23 @@ static int __uloop_fd_delete(struct uloop_fd *fd)
        return register_poll(fd, 0);
 }
 
+static int64_t get_timestamp_us(void)
+{
+#ifdef CLOCK_MONOTONIC
+       struct timespec ts = { 0, 0 };
+
+       clock_gettime(CLOCK_MONOTONIC, &ts);
+
+       return ts.tv_sec * 1000000 + ts.tv_nsec / 1000;
+#else
+       struct timeval tv = { 0, 0 };
+
+       gettimeofday(&tv, NULL);
+
+       return tv.tv_sec * 1000000 + tv.tv_usec;
+#endif
+}
+
 static int uloop_fetch_events(int timeout)
 {
        struct timespec ts;
@@ -115,6 +132,16 @@ static int uloop_fetch_events(int timeout)
 
        nfds = kevent(poll_fd, NULL, 0, events, ARRAY_SIZE(events), timeout >= 0 ? &ts : NULL);
        for (n = 0; n < nfds; n++) {
+               if (events[n].filter == EVFILT_TIMER) {
+                       struct uloop_interval *tm = events[n].udata;
+
+                       tm->private.time.fired = get_timestamp_us();
+                       tm->expirations += events[n].data;
+                       tm->cb(tm);
+
+                       continue;
+               }
+
                struct uloop_fd_event *cur = &cur_fds[n];
                struct uloop_fd *u = events[n].udata;
                unsigned int ev = 0;
@@ -148,3 +175,35 @@ static int uloop_fetch_events(int timeout)
        }
        return nfds;
 }
+
+static int timer_register(struct uloop_interval *tm, unsigned int msecs)
+{
+       struct kevent ev;
+
+       tm->private.time.msecs = msecs;
+       tm->private.time.fired = get_timestamp_us();
+
+       EV_SET(&ev, (uintptr_t)tm, EVFILT_TIMER, EV_ADD, NOTE_USECONDS, msecs * 1000, tm);
+
+       return kevent(poll_fd, &ev, 1, NULL, 0, NULL);
+}
+
+static int timer_remove(struct uloop_interval *tm)
+{
+       struct kevent ev;
+
+       EV_SET(&ev, (uintptr_t)tm, EVFILT_TIMER, EV_DELETE, 0, 0, NULL);
+
+       return kevent(poll_fd, &ev, 1, NULL, 0, NULL);
+}
+
+static int64_t timer_next(struct uloop_interval *tm)
+{
+       int64_t t1 = tm->private.time.fired;
+       int64_t t2 = get_timestamp_us();
+
+       while (t1 < t2)
+               t1 += tm->private.time.msecs * 1000;
+
+       return (t1 - t2) / 1000;
+}
diff --git a/uloop.c b/uloop.c
index 8fc5aee6743b4a86bbc1277ecdb1f7d90e219326..a3d37124be253713a535216f1c3abd18363d1db8 100644 (file)
--- a/uloop.c
+++ b/uloop.c
@@ -36,6 +36,7 @@
 #endif
 #ifdef USE_EPOLL
 #include <sys/epoll.h>
+#include <sys/timerfd.h>
 #endif
 #include <sys/wait.h>
 
@@ -422,6 +423,21 @@ static void uloop_handle_processes(void)
 
 }
 
+int uloop_interval_set(struct uloop_interval *timer, unsigned int msecs)
+{
+       return timer_register(timer, msecs);
+}
+
+int uloop_interval_cancel(struct uloop_interval *timer)
+{
+       return timer_remove(timer);
+}
+
+int64_t uloop_interval_remaining(struct uloop_interval *timer)
+{
+       return timer_next(timer);
+}
+
 static void uloop_signal_wake(void)
 {
        do {
diff --git a/uloop.h b/uloop.h
index 4fdd43f5084d438daaef17852824db57761ea124..b3f268c069ae10dd39aeb78eb11e1cb27746ef21 100644 (file)
--- a/uloop.h
+++ b/uloop.h
 struct uloop_fd;
 struct uloop_timeout;
 struct uloop_process;
+struct uloop_interval;
 
 typedef void (*uloop_fd_handler)(struct uloop_fd *u, unsigned int events);
 typedef void (*uloop_timeout_handler)(struct uloop_timeout *t);
 typedef void (*uloop_process_handler)(struct uloop_process *c, int ret);
+typedef void (*uloop_interval_handler)(struct uloop_interval *t);
 
 #define ULOOP_READ             (1 << 0)
 #define ULOOP_WRITE            (1 << 1)
@@ -83,6 +85,20 @@ struct uloop_process
        pid_t pid;
 };
 
+struct uloop_interval
+{
+       uloop_interval_handler cb;
+       uint64_t expirations;
+
+       union {
+               struct uloop_fd ufd;
+               struct {
+                       int64_t fired;
+                       unsigned int msecs;
+               } time;
+       } private;
+};
+
 extern bool uloop_cancelled;
 extern bool uloop_handle_sigchld;
 extern uloop_fd_handler uloop_fd_set_cb;
@@ -100,6 +116,10 @@ int64_t uloop_timeout_remaining64(struct uloop_timeout *timeout);
 int uloop_process_add(struct uloop_process *p);
 int uloop_process_delete(struct uloop_process *p);
 
+int uloop_interval_set(struct uloop_interval *timer, unsigned int msecs);
+int uloop_interval_cancel(struct uloop_interval *timer);
+int64_t uloop_interval_remaining(struct uloop_interval *timer);
+
 bool uloop_cancelling(void);
 
 static inline void uloop_end(void)