X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=src%2Fnet.c;h=3a44ce66b61d151dd42329e1118a2d5c00c54f43;hb=38c25d62c2bc76908bd95fb21c8f5e39ad269884;hp=f6b14704607477544b6a34f54cf2144096fe2548;hpb=6c6535a4161d04accb3a22c51477e9f92ae34086;p=meshlink diff --git a/src/net.c b/src/net.c index f6b14704..3a44ce66 100644 --- a/src/net.c +++ b/src/net.c @@ -29,7 +29,7 @@ #include "conf.h" #include "connection.h" #include "device.h" -#include "event.h" +#include "tevent.h" #include "graph.h" #include "logger.h" #include "meta.h" @@ -109,10 +109,10 @@ static void purge(void) } /* - put all file descriptors in an fd_set array - While we're at it, purge stuff that needs to be removed. + put all file descriptors into events + While we're at it, purge stuf that needs to be removed. */ -static int build_fdset(fd_set *readset, fd_set *writeset) +static int build_fdset(void) { avl_node_t *node, *next; connection_t *c; @@ -120,9 +120,6 @@ static int build_fdset(fd_set *readset, fd_set *writeset) cp(); - FD_ZERO(readset); - FD_ZERO(writeset); - for(node = connection_tree->head; node; node = next) { next = node->next; c = node->data; @@ -132,28 +129,17 @@ static int build_fdset(fd_set *readset, fd_set *writeset) if(!connection_tree->head) purge(); } else { - FD_SET(c->socket, readset); + short events = EV_READ; if(c->outbuflen > 0) - FD_SET(c->socket, writeset); - if(c->socket > max) - max = c->socket; + events |= EV_WRITE; + event_del(&c->ev); + event_set(&c->ev, c->socket, events, + handle_meta_connection_data, c); + if (event_add(&c->ev, NULL) < 0) + return -1; } } - - for(i = 0; i < listen_sockets; i++) { - FD_SET(listen_socket[i].tcp, readset); - if(listen_socket[i].tcp > max) - max = listen_socket[i].tcp; - FD_SET(listen_socket[i].udp, readset); - if(listen_socket[i].udp > max) - max = listen_socket[i].udp; - } - - FD_SET(device_fd, readset); - if(device_fd > max) - max = device_fd; - - return max; + return 0; } /* @@ -279,83 +265,59 @@ static void check_dead_connections(void) } } -/* - check all connections to see if anything - happened on their sockets -*/ -static void check_network_activity(fd_set * readset, fd_set * writeset) +void handle_meta_connection_data(int fd, short events, void *data) { - connection_t *c; - avl_node_t *node; - int result, i; + connection_t *c = data; + int result; socklen_t len = sizeof(result); - vpn_packet_t packet; - cp(); - - /* check input from kernel */ - if(FD_ISSET(device_fd, readset)) { - if(read_packet(&packet)) - route(myself, &packet); - } - - /* check meta connections */ - for(node = connection_tree->head; node; node = node->next) { - c = node->data; - - if(c->status.remove) - continue; - - if(FD_ISSET(c->socket, readset)) { - if(c->status.connecting) { - c->status.connecting = false; - getsockopt(c->socket, SOL_SOCKET, SO_ERROR, &result, &len); - - if(!result) - finish_connecting(c); - else { - ifdebug(CONNECTIONS) logger(LOG_DEBUG, - _("Error while connecting to %s (%s): %s"), - c->name, c->hostname, strerror(result)); - closesocket(c->socket); - do_outgoing_connection(c); - continue; - } - } + if (c->status.remove) + return; - if(!receive_meta(c)) { - terminate_connection(c, c->status.active); - continue; + if (events & EV_READ) { + if(c->status.connecting) { + c->status.connecting = false; + getsockopt(c->socket, SOL_SOCKET, SO_ERROR, &result, &len); + + if(!result) + finish_connecting(c); + else { + ifdebug(CONNECTIONS) logger(LOG_DEBUG, + _("Error while connecting to %s (%s): %s"), + c->name, c->hostname, strerror(result)); + closesocket(c->socket); + do_outgoing_connection(c); + return; } } - if(FD_ISSET(c->socket, writeset)) { - if(!flush_meta(c)) { - terminate_connection(c, c->status.active); - continue; - } + if (!receive_meta(c)) { + terminate_connection(c, c->status.active); + return; } } - for(i = 0; i < listen_sockets; i++) { - if(FD_ISSET(listen_socket[i].udp, readset)) - handle_incoming_vpn_data(listen_socket[i].udp); - - if(FD_ISSET(listen_socket[i].tcp, readset)) - handle_new_meta_connection(listen_socket[i].tcp); + if (events & EV_WRITE) { + if(!flush_meta(c)) { + terminate_connection(c, c->status.active); + } } } +static void dummy(int a, short b, void *c) +{ +} + /* this is where it all happens... */ int main_loop(void) { - fd_set readset, writeset; struct timeval tv; - int r, maxfd; + int r; time_t last_ping_check, last_config_check, last_graph_dump; - event_t *event; + tevent_t *event; + struct event timeout; cp(); @@ -374,23 +336,30 @@ int main_loop(void) tv.tv_sec = 1; tv.tv_usec = 0; - maxfd = build_fdset(&readset, &writeset); - - r = select(maxfd + 1, &readset, &writeset, NULL, &tv); + /* XXX: libevent transition: old timeout code in this loop */ + timeout_set(&timeout, dummy, NULL); + timeout_add(&timeout, &tv); + r = build_fdset(); if(r < 0) { - if(errno != EINTR && errno != EAGAIN) { - logger(LOG_ERR, _("Error while waiting for input: %s"), - strerror(errno)); - cp_trace(); - dump_connections(); - return 1; - } + logger(LOG_ERR, _("Error building fdset: %s"), strerror(errno)); + cp_trace(); + dump_connections(); + return 1; + } - continue; + r = event_loop(EVLOOP_ONCE); + now = time(NULL); + if(r < 0) { + logger(LOG_ERR, _("Error while waiting for input: %s"), + strerror(errno)); + cp_trace(); + dump_connections(); + return 1; } - check_network_activity(&readset, &writeset); + /* XXX: more libevent transition */ + timeout_del(&timeout); if(do_purge) { purge(); @@ -422,14 +391,14 @@ int main_loop(void) } - while((event = get_expired_event())) { + while((event = get_expired_tevent())) { event->handler(event->data); - free_event(event); + free_tevent(event); } if(sigalrm) { logger(LOG_INFO, _("Flushing event queue")); - flush_events(); + flush_tevents(); sigalrm = false; }