X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=src%2Fprotocol.c;h=fa1dc13786b4a30cb02cfe079a2a06b37ecf8925;hb=be83b0af60449c7b35d17d97f2e6dc12f611e831;hp=fd180f4faafab84fd4c3bfedf0bf465dbf96b13c;hpb=b1cae6a2011f704dc4d3b02972def561d5c6bcb9;p=meshlink diff --git a/src/protocol.c b/src/protocol.c index fd180f4f..fa1dc137 100644 --- a/src/protocol.c +++ b/src/protocol.c @@ -67,14 +67,14 @@ bool check_id(const char *id) { detection as well */ bool send_request(meshlink_handle_t *mesh, connection_t *c, submesh_t *s, const char *format, ...) { + assert(c); + assert(format); + assert(*format); + va_list args; char request[MAXBUFSIZE]; int len; - if(!c) { - return false; - } - /* Use vsnprintf instead of vxasprintf: faster, no memory fragmentation, cleanup is automatic, and there is a limit on the input buffer anyway */ @@ -107,6 +107,10 @@ bool send_request(meshlink_handle_t *mesh, connection_t *c, submesh_t *s, const } void forward_request(meshlink_handle_t *mesh, connection_t *from, submesh_t *s, const char *request) { + assert(from); + assert(request); + assert(*request); + logger(mesh, MESHLINK_DEBUG, "Forwarding %s from %s: %s", request_name[atoi(request)], from->name, request); // Create a temporary newline-terminated copy of the request @@ -124,6 +128,9 @@ void forward_request(meshlink_handle_t *mesh, connection_t *from, submesh_t *s, } bool receive_request(meshlink_handle_t *mesh, connection_t *c, const char *request) { + assert(request); + assert(*request); + if(c->outgoing && mesh->proxytype == PROXY_HTTP && c->allow_request == ID) { if(!request[0] || request[0] == '\r') { return true; @@ -200,13 +207,17 @@ static void age_past_requests(event_loop_t *loop, void *data) { logger(mesh, MESHLINK_DEBUG, "Aging past requests: deleted %d, left %d", deleted, left); } - if(left) + if(left) { timeout_set(&mesh->loop, &mesh->past_request_timeout, &(struct timeval) { - 10, rand() % 100000 - }); + 10, prng(mesh, TIMER_FUDGE) + }); + } } bool seen_request(meshlink_handle_t *mesh, const char *request) { + assert(request); + assert(*request); + past_request_t *new, p = {.request = request}; if(splay_search(mesh->past_request_tree, &p)) { @@ -216,16 +227,25 @@ bool seen_request(meshlink_handle_t *mesh, const char *request) { new = xmalloc(sizeof(*new)); new->request = xstrdup(request); new->firstseen = mesh->loop.now.tv_sec; + + if(!mesh->past_request_tree->head) { + timeout_set(&mesh->loop, &mesh->past_request_timeout, &(struct timeval) { + 10, prng(mesh, TIMER_FUDGE) + }); + } + splay_insert(mesh->past_request_tree, new); - timeout_add(&mesh->loop, &mesh->past_request_timeout, age_past_requests, NULL, &(struct timeval) { - 10, rand() % 100000 - }); return false; } } void init_requests(meshlink_handle_t *mesh) { + assert(!mesh->past_request_tree); + mesh->past_request_tree = splay_alloc_tree((splay_compare_t) past_request_compare, (splay_action_t) free_past_request); + timeout_add(&mesh->loop, &mesh->past_request_timeout, age_past_requests, NULL, &(struct timeval) { + 0, 0 + }); } void exit_requests(meshlink_handle_t *mesh) {