Call utcp_timeout() once every iteration of the event loop.

[meshlink] / src / meshlink.c

diff --git a/src/meshlink.c b/src/meshlink.c
index bd74b1d8d358ba93ee6afbbb42c7e1df4b5f9d39..8266633269845631160fbf9ee5641d46e1f7bd5f 100644 (file)
--- a/src/meshlink.c
+++ b/src/meshlink.c
@@ -699,6 +699,20 @@ static bool ecdsa_keygen(meshlink_handle_t *mesh) {
        return true;
 }
 
+static struct timeval idle(event_loop_t *loop, void *data) {
+       meshlink_handle_t *mesh = data;
+       int t, tmin = -1;
+       for splay_each(node_t, n, mesh->nodes) {
+               if(!n->utcp)
+                       continue;
+               t = utcp_timeout(n->utcp);
+               if(t >= 0 && t < tmin)
+                       tmin = t;
+       }
+       struct timeval tv = {.tv_sec = t};
+       return tv;
+}
+
 static bool meshlink_setup(meshlink_handle_t *mesh) {
        if(mkdir(mesh->confbase, 0777) && errno != EEXIST) {
                logger(mesh, MESHLINK_DEBUG, "Could not create directory %s: %s\n", mesh->confbase, strerror(errno));
@@ -845,6 +859,8 @@ meshlink_handle_t *meshlink_open_with_size(const char *confbase, const char *nam
                return NULL;
        }
 
+       idle_set(&mesh->loop, idle, mesh);
+
        logger(NULL, MESHLINK_DEBUG, "meshlink_open returning\n");
        return mesh;
 }
@@ -1961,6 +1977,21 @@ static void channel_receive(meshlink_handle_t *mesh, meshlink_node_t *source, co
        utcp_recv(n->utcp, data, len);
 }
 
+static void channel_poll(struct utcp_connection *connection, size_t len) {
+       meshlink_channel_t *channel = connection->priv;
+       if(!channel)
+               abort();
+       node_t *n = channel->node;
+       meshlink_handle_t *mesh = n->mesh;
+       if(channel->poll_cb)
+               channel->poll_cb(mesh, channel, len);
+}
+
+void meshlink_set_channel_poll_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, meshlink_channel_poll_cb_t cb) {
+       channel->poll_cb = cb;
+       utcp_set_poll_cb(channel->c, cb ? channel_poll : NULL);
+}
+
 void meshlink_set_channel_accept_cb(meshlink_handle_t *mesh, meshlink_channel_accept_cb_t cb) {
        pthread_mutex_lock(&mesh->mesh_mutex);
        mesh->channel_accept_cb = cb;