X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=src%2Fprotocol.c;h=482808a7e71885c6cf37d24ce42444f9d18c8aac;hb=963c5055505f2fc117cd5efa06eaa02c9b2bf85d;hp=fa43453fddde5b8f9d836ad912fbae1ff2bd0bc2;hpb=462ab530e546f5732dfd51134751da6f6910d679;p=meshlink diff --git a/src/protocol.c b/src/protocol.c index fa43453f..482808a7 100644 --- a/src/protocol.c +++ b/src/protocol.c @@ -1,7 +1,6 @@ /* protocol.c -- handle the meta-protocol, basic functions - Copyright (C) 1999-2001 Ivo Timmermans , - 2000,2001 Guus Sliepen + Copyright (C) 2014-2017 Guus Sliepen This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -13,232 +12,247 @@ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. - You should have received a copy of the GNU General Public License - along with this program; if not, write to the Free Software - Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. - - $Id: protocol.c,v 1.30 2002/04/09 15:26:00 zarq Exp $ + You should have received a copy of the GNU General Public License along + with this program; if not, write to the Free Software Foundation, Inc., + 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. */ -#include "config.h" +#include "system.h" -#include +#include "conf.h" +#include "connection.h" +#include "logger.h" +#include "meshlink_internal.h" +#include "meta.h" +#include "protocol.h" +#include "utils.h" +#include "xalloc.h" +#include "submesh.h" -#include -#include -#include -#include -#include -#include +/* Jumptable for the request handlers */ -#include -#include +static bool (*request_handlers[])(meshlink_handle_t *, connection_t *, const char *) = { + id_h, NULL, NULL, NULL /* metakey_h, challenge_h, chal_reply_h */, ack_h, + status_h, error_h, termreq_h, + ping_h, pong_h, + NULL, NULL, //add_subnet_h, del_subnet_h, + add_edge_h, del_edge_h, + key_changed_h, req_key_h, ans_key_h, tcppacket_h, NULL, //control_h, +}; -#include "conf.h" -#include "protocol.h" -#include "meta.h" -#include "connection.h" +/* Request names */ -#include "system.h" +static char (*request_name[]) = { + "ID", "METAKEY", "CHALLENGE", "CHAL_REPLY", "ACK", + "STATUS", "ERROR", "TERMREQ", + "PING", "PONG", + "ADD_SUBNET", "DEL_SUBNET", + "ADD_EDGE", "DEL_EDGE", "KEY_CHANGED", "REQ_KEY", "ANS_KEY", "PACKET", "CONTROL", +}; -avl_tree_t *past_request_tree; +bool check_id(const char *id) { + if(!id || !*id) { + return false; + } -int check_id(char *id) -{ - int i; + for(; *id; id++) + if(!isalnum(*id) && *id != '_' && *id != '-') { + return false; + } - for (i = 0; i < strlen(id); i++) - if(!isalnum(id[i]) && id[i] != '_') - return -1; - - return 0; + return true; } /* Generic request routines - takes care of logging and error detection as well */ -int send_request(connection_t *c, const char *format, ...) -{ - va_list args; - char buffer[MAXBUFSIZE]; - int len, request; - -cp - /* Use vsnprintf instead of vasprintf: faster, no memory - fragmentation, cleanup is automatic, and there is a limit on the - input buffer anyway */ - - va_start(args, format); - len = vsnprintf(buffer, MAXBUFSIZE, format, args); - va_end(args); - - if(len < 0 || len > MAXBUFSIZE-1) - { - syslog(LOG_ERR, _("Output buffer overflow while sending request to %s (%s)"), c->name, c->hostname); - return -1; - } - - if(debug_lvl >= DEBUG_PROTOCOL) - { - sscanf(buffer, "%d", &request); - if(debug_lvl >= DEBUG_META) - syslog(LOG_DEBUG, _("Sending %s to %s (%s): %s"), request_name[request], c->name, c->hostname, buffer); - else - syslog(LOG_DEBUG, _("Sending %s to %s (%s)"), request_name[request], c->name, c->hostname); - } - - buffer[len++] = '\n'; -cp - return send_meta(c, buffer, len); -} +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; -int receive_request(connection_t *c) -{ - int request; -cp - if(sscanf(c->buffer, "%d", &request) == 1) - { - if((request < 0) || (request >= LAST) || (request_handlers[request] == NULL)) - { - if(debug_lvl >= DEBUG_META) - syslog(LOG_DEBUG, _("Unknown request from %s (%s): %s"), - c->name, c->hostname, c->buffer); - else - syslog(LOG_ERR, _("Unknown request from %s (%s)"), - c->name, c->hostname); - - return -1; - } - else - { - if(debug_lvl >= DEBUG_PROTOCOL) - { - if(debug_lvl >= DEBUG_META) - syslog(LOG_DEBUG, _("Got %s from %s (%s): %s"), - request_name[request], c->name, c->hostname, c->buffer); - else - syslog(LOG_DEBUG, _("Got %s from %s (%s)"), - request_name[request], c->name, c->hostname); - } + /* Use vsnprintf instead of vxasprintf: faster, no memory + fragmentation, cleanup is automatic, and there is a limit on the + input buffer anyway */ + + va_start(args, format); + len = vsnprintf(request, MAXBUFSIZE, format, args); + va_end(args); + + if(len < 0 || len > MAXBUFSIZE - 1) { + logger(mesh, MESHLINK_ERROR, "Output buffer overflow while sending request to %s", c->name); + return false; } - if((c->allow_request != ALL) && (c->allow_request != request)) - { - syslog(LOG_ERR, _("Unauthorized request from %s (%s)"), c->name, c->hostname); - return -1; - } - - if(request_handlers[request](c)) - /* Something went wrong. Probably scriptkiddies. Terminate. */ - { - syslog(LOG_ERR, _("Error while processing %s from %s (%s)"), - request_name[request], c->name, c->hostname); - return -1; - } - } - else - { - syslog(LOG_ERR, _("Bogus data received from %s (%s)"), - c->name, c->hostname); - return -1; - } -cp - return 0; + logger(mesh, MESHLINK_DEBUG, "Sending %s to %s: %s", request_name[atoi(request)], c->name, request); + + request[len++] = '\n'; + + if(c == mesh->everyone) { + + if(s) { + broadcast_submesh_meta(mesh, NULL, s, request, len); + } else { + broadcast_meta(mesh, NULL, request, len); + } + + return true; + } else { + return send_meta(mesh, c, request, len); + } } -int past_request_compare(past_request_t *a, past_request_t *b) -{ -cp - return strcmp(a->request, b->request); +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 + int len = strlen(request); + char tmp[len + 1]; + + memcpy(tmp, request, len); + tmp[len] = '\n'; + + if(s) { + broadcast_submesh_meta(mesh, from, s, tmp, sizeof(tmp)); + } else { + broadcast_meta(mesh, from, tmp, sizeof(tmp)); + } } -void free_past_request(past_request_t *r) -{ -cp - if(r->request) - free(r->request); - free(r); -cp +bool receive_request(meshlink_handle_t *mesh, connection_t *c, const char *request) { + assert(request); + + if(c->outgoing && mesh->proxytype == PROXY_HTTP && c->allow_request == ID) { + if(!request[0] || request[0] == '\r') { + return true; + } + + if(!strncasecmp(request, "HTTP/1.1 ", 9)) { + if(!strncmp(request + 9, "200", 3)) { + logger(mesh, MESHLINK_DEBUG, "Proxy request granted"); + return true; + } else { + logger(mesh, MESHLINK_DEBUG, "Proxy request rejected: %s", request + 9); + return false; + } + } + } + + int reqno = atoi(request); + + if(reqno || *request == '0') { + if((reqno < 0) || (reqno >= LAST) || !request_handlers[reqno]) { + logger(mesh, MESHLINK_DEBUG, "Unknown request from %s: %s", c->name, request); + return false; + } else { + logger(mesh, MESHLINK_DEBUG, "Got %s from %s: %s", request_name[reqno], c->name, request); + } + + if((c->allow_request != ALL) && (c->allow_request != reqno)) { + logger(mesh, MESHLINK_ERROR, "Unauthorized request from %s", c->name); + return false; + } + + if(!request_handlers[reqno](mesh, c, request)) { + /* Something went wrong. Probably scriptkiddies. Terminate. */ + + logger(mesh, MESHLINK_ERROR, "Error while processing %s from %s", request_name[reqno], c->name); + return false; + } + } else { + logger(mesh, MESHLINK_ERROR, "Bogus data received from %s", c->name); + return false; + } + + return true; } -void init_requests(void) -{ -cp - past_request_tree = avl_alloc_tree((avl_compare_t)past_request_compare, (avl_action_t)free_past_request); -cp +static int past_request_compare(const past_request_t *a, const past_request_t *b) { + return strcmp(a->request, b->request); } -void exit_requests(void) -{ -cp - avl_delete_tree(past_request_tree); -cp +static void free_past_request(past_request_t *r) { + if(r->request) { + free((void *)r->request); + } + + free(r); } -int seen_request(char *request) -{ - past_request_t p, *new; -cp - p.request = request; - - if(avl_search(past_request_tree, &p)) - { - if(debug_lvl >= DEBUG_SCARY_THINGS) - syslog(LOG_DEBUG, _("Already seen request")); - return 1; - } - else - { - new = (past_request_t *)xmalloc(sizeof(*new)); - new->request = xstrdup(request); - new->firstseen = now; - avl_insert(past_request_tree, new); - return 0; - } -cp +static const int request_timeout = 60; + +static void age_past_requests(event_loop_t *loop, void *data) { + (void)data; + meshlink_handle_t *mesh = loop->data; + int left = 0, deleted = 0; + + for splay_each(past_request_t, p, mesh->past_request_tree) { + if(p->firstseen + request_timeout <= mesh->loop.now.tv_sec) { + splay_delete_node(mesh->past_request_tree, node), deleted++; + } else { + left++; + } + } + + if(left || deleted) { + logger(mesh, MESHLINK_DEBUG, "Aging past requests: deleted %d, left %d", deleted, left); + } + + if(left) { + timeout_set(&mesh->loop, &mesh->past_request_timeout, &(struct timespec) { + 10, prng(mesh, TIMER_FUDGE) + }); + } } -void age_past_requests(void) -{ - avl_node_t *node, *next; - past_request_t *p; - int left = 0, deleted = 0; -cp - for(node = past_request_tree->head; node; node = next) - { - next = node->next; - p = (past_request_t *)node->data; - if(p->firstseen + pingtimeout < now) - avl_delete_node(past_request_tree, node), deleted++; - else - left++; - } - - if(debug_lvl >= DEBUG_SCARY_THINGS && left + deleted) - syslog(LOG_DEBUG, _("Aging past requests: deleted %d, left %d\n"), deleted, left); -cp +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)) { + logger(mesh, MESHLINK_DEBUG, "Already seen request"); + return true; + } else { + 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 timespec) { + 10, prng(mesh, TIMER_FUDGE) + }); + } + + splay_insert(mesh->past_request_tree, new); + return false; + } } -/* Jumptable for the request handlers */ +void init_requests(meshlink_handle_t *mesh) { + assert(!mesh->past_request_tree); -int (*request_handlers[])(connection_t*) = { - id_h, metakey_h, challenge_h, chal_reply_h, ack_h, - status_h, error_h, termreq_h, - ping_h, pong_h, - add_subnet_h, del_subnet_h, - add_edge_h, del_edge_h, - key_changed_h, req_key_h, ans_key_h, - tcppacket_h, -}; + 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 timespec) { + 0, 0 + }); +} -/* Request names */ +void exit_requests(meshlink_handle_t *mesh) { + if(mesh->past_request_tree) { + splay_delete_tree(mesh->past_request_tree); + } -char (*request_name[]) = { - "ID", "METAKEY", "CHALLENGE", "CHAL_REPLY", "ACK", - "STATUS", "ERROR", "TERMREQ", - "PING", "PONG", - "ADD_SUBNET", "DEL_SUBNET", - "ADD_EDGE", "DEL_EDGE", - "KEY_CHANGED", "REQ_KEY", "ANS_KEY", - "PACKET", -}; + mesh->past_request_tree = NULL; + + timeout_del(&mesh->loop, &mesh->past_request_timeout); +}