]> git.meshlink.io Git - meshlink/blobdiff - src/protocol_edge.c
Avoid allocating packet buffers unnecessarily.
[meshlink] / src / protocol_edge.c
index 7663a46fefa11133c5854f5e8a3fa5bd56eda95b..c42a10d28701825adef6440f3ccd3b82a4c96e7f 100644 (file)
@@ -1,7 +1,6 @@
 /*
     protocol_edge.c -- handle the meta-protocol, edges
-    Copyright (C) 1999-2002 Ivo Timmermans <ivo@o2w.nl>,
-                  2000-2002 Guus Sliepen <guus@sliepen.eu.org>
+    Copyright (C) 2014 Guus Sliepen <guus@meshlink.io>
 
     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
     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_edge.c,v 1.1.4.10 2002/09/04 16:26:45 guus 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 <stdlib.h>
-#include <string.h>
-#include <syslog.h>
-#include <stdio.h>
-#include <stdarg.h>
-#include <errno.h>
-
-#include <utils.h>
-#include <xalloc.h>
-#include <avl_tree.h>
+#include "system.h"
 
 #include "conf.h"
-#include "net.h"
-#include "netutl.h"
-#include "protocol.h"
-#include "meta.h"
 #include "connection.h"
-#include "node.h"
 #include "edge.h"
 #include "graph.h"
-
-#include "system.h"
-
-int send_add_edge(connection_t *c, edge_t *e)
-{
-  int x;
-  char *address, *port;
-cp
-  sockaddr2str(&e->address, &address, &port);
-  x = send_request(c, "%d %lx %s %s %s %s %lx %d", ADD_EDGE, random(),
-                      e->from->name, e->to->name, address, port,
-                     e->options, e->weight);
-  free(address);
-  free(port);
-cp
-  return x;
+#include "logger.h"
+#include "meshlink_internal.h"
+#include "meta.h"
+#include "net.h"
+#include "netutl.h"
+#include "node.h"
+#include "protocol.h"
+#include "utils.h"
+#include "xalloc.h"
+#include "submesh.h"
+
+bool send_add_edge(meshlink_handle_t *mesh, connection_t *c, const edge_t *e, int contradictions) {
+       bool x;
+       char *address, *port;
+       char *from_submesh, *to_submesh;
+       submesh_t *s = NULL;
+
+       if(c->node && c->node->submesh) {
+               if(!submesh_allows_node(e->from->submesh, c->node)) {
+                       return true;
+               }
+
+               if(!submesh_allows_node(e->to->submesh, c->node)) {
+                       return true;
+               }
+       }
+
+       if(e->from->submesh && e->to->submesh && (e->from->submesh != e->to->submesh)) {
+               return true;
+       }
+
+       sockaddr2str(&e->address, &address, &port);
+
+       if(e->from->submesh) {
+               from_submesh = e->from->submesh->name;
+       } else {
+               from_submesh = CORE_MESH;
+       }
+
+       if(e->to->submesh) {
+               to_submesh = e->to->submesh->name;
+       } else {
+               to_submesh = CORE_MESH;
+       }
+
+       if(e->from->submesh) {
+               s = e->from->submesh;
+       } else {
+               s = e->to->submesh;
+       }
+
+       x = send_request(mesh, c, s, "%d %x %s %d %s %s %s %s %d %s %x %d %d %x", ADD_EDGE, prng(mesh, UINT_MAX),
+                        e->from->name, e->from->devclass, from_submesh, e->to->name, address, port,
+                        e->to->devclass, to_submesh, OPTION_PMTU_DISCOVERY, e->weight, contradictions, e->from->session_id);
+       free(address);
+       free(port);
+
+       return x;
 }
 
-int add_edge_h(connection_t *c)
-{
-  edge_t *e;
-  node_t *from, *to;
-  char from_name[MAX_STRING_SIZE];
-  char to_name[MAX_STRING_SIZE];
-  char to_address[MAX_STRING_SIZE];
-  char to_port[MAX_STRING_SIZE];
-  sockaddr_t address;
-  long int options;
-  int weight;
-cp
-  if(sscanf(c->buffer, "%*d %*x "MAX_STRING" "MAX_STRING" "MAX_STRING" "MAX_STRING" %lx %d",
-            from_name, to_name, to_address, to_port, &options, &weight) != 6)
-    {
-       syslog(LOG_ERR, _("Got bad %s from %s (%s)"), "ADD_EDGE", c->name, c->hostname);
-       return -1;
-    }
-
-  /* Check if names are valid */
-
-  if(check_id(from_name))
-    {
-      syslog(LOG_ERR, _("Got bad %s from %s (%s): %s"), "ADD_EDGE", c->name, c->hostname, _("invalid name"));
-      return -1;
-    }
-
-  if(check_id(to_name))
-    {
-      syslog(LOG_ERR, _("Got bad %s from %s (%s): %s"), "ADD_EDGE", c->name, c->hostname, _("invalid name"));
-      return -1;
-    }
-
-  if(seen_request(c->buffer))
-    return 0;
-
-  /* Lookup nodes */
-
-  from = lookup_node(from_name);
-  
-  if(!from)
-    {
-      from = new_node();
-      from->name = xstrdup(from_name);
-      node_add(from);
-    }
-
-  to = lookup_node(to_name);
-  
-  if(!to)
-    {
-      to = new_node();
-      to->name = xstrdup(to_name);
-      node_add(to);
-    }
-
-  /* Convert addresses */
-  
-  address = str2sockaddr(to_address, to_port);
-
-  /* Check if edge already exists */
-  
-  e = lookup_edge(from, to);
-  
-  if(e)
-  {
-    if(e->weight != weight || e->options != options || sockaddrcmp(&e->address, &address))
-    {
-      if(from == myself)
-      {
-        if(debug_lvl >= DEBUG_PROTOCOL)
-          syslog(LOG_WARNING, _("Got %s from %s (%s) for ourself which does not match existing entry"), "ADD_EDGE", c->name, c->hostname);
-        send_add_edge(c, e);
-        return 0;
-      }
-      else
-      {
-        if(debug_lvl >= DEBUG_PROTOCOL)
-          syslog(LOG_WARNING, _("Got %s from %s (%s) which does not match existing entry"), "ADD_EDGE", c->name, c->hostname);
-        edge_del(e);
-      }
-    }
-    else
-      return 0;
-  }
-  else if(from == myself)
-  {
-    if(debug_lvl >= DEBUG_PROTOCOL)
-      syslog(LOG_WARNING, _("Got %s from %s (%s) for ourself which does not exist"), "ADD_EDGE", c->name, c->hostname);
-    e = new_edge();
-    e->from = from;
-    e->to = to;
-    send_del_edge(c, e);
-    free_edge(e);
-    return 0;
-  }
-
-  e = new_edge();
-  e->from = from;
-  e->to = to;
-  e->address = address;
-  e->options = options;
-  e->weight = weight;
-  edge_add(e);
-
-  /* Tell the rest about the new edge */
-
-  forward_request(c);
-
-  /* Run MST before or after we tell the rest? */
-
-  graph();
-cp
-  return 0;
+bool add_edge_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
+       assert(request);
+       assert(*request);
+
+       edge_t *e;
+       node_t *from, *to;
+       char from_name[MAX_STRING_SIZE];
+       int from_devclass;
+       char from_submesh_name[MAX_STRING_SIZE] = "";
+       char to_name[MAX_STRING_SIZE];
+       char to_address[MAX_STRING_SIZE];
+       char to_port[MAX_STRING_SIZE];
+       int to_devclass;
+       char to_submesh_name[MAX_STRING_SIZE] = "";
+       sockaddr_t address;
+       int weight;
+       int contradictions = 0;
+       uint32_t session_id = 0;
+       submesh_t *s = NULL;
+
+       if(sscanf(request, "%*d %*x "MAX_STRING" %d "MAX_STRING" "MAX_STRING" "MAX_STRING" "MAX_STRING" %d "MAX_STRING" %*x %d %d %x",
+                       from_name, &from_devclass, from_submesh_name, to_name, to_address, to_port, &to_devclass, to_submesh_name,
+                       &weight, &contradictions, &session_id) < 9) {
+               logger(mesh, MESHLINK_ERROR, "Got bad %s from %s", "ADD_EDGE", c->name);
+               return false;
+       }
+
+       // Check if devclasses are valid
+
+       if(from_devclass < 0 || from_devclass >= DEV_CLASS_COUNT) {
+               logger(mesh, MESHLINK_ERROR, "Got bad %s from %s: %s", "ADD_EDGE", c->name, "from devclass invalid");
+               return false;
+       }
+
+       if(to_devclass < 0 || to_devclass >= DEV_CLASS_COUNT) {
+               logger(mesh, MESHLINK_ERROR, "Got bad %s from %s: %s", "ADD_EDGE", c->name, "to devclass invalid");
+               return false;
+       }
+
+       if(0 == strcmp(from_submesh_name, "")) {
+               logger(mesh, MESHLINK_ERROR, "Got bad %s from %s: %s", "ADD_EDGE", c->name, "invalid submesh id");
+               return false;
+       }
+
+       if(0 == strcmp(to_submesh_name, "")) {
+               logger(mesh, MESHLINK_ERROR, "Got bad %s from %s: %s", "ADD_EDGE", c->name, "invalid submesh id");
+               return false;
+       }
+
+       if(seen_request(mesh, request)) {
+               return true;
+       }
+
+       /* Lookup nodes */
+
+       from = lookup_node(mesh, from_name);
+       to = lookup_node(mesh, to_name);
+
+       if(!from) {
+               from = new_node();
+               from->status.dirty = true;
+               from->status.blacklisted = mesh->default_blacklist;
+               from->name = xstrdup(from_name);
+               from->devclass = from_devclass;
+
+               from->submesh = NULL;
+
+               if(0 != strcmp(from_submesh_name, CORE_MESH)) {
+                       if(!(from->submesh = lookup_or_create_submesh(mesh, from_submesh_name))) {
+                               return false;
+                       }
+               }
+
+               node_add(mesh, from);
+       }
+
+       if(contradictions > 50) {
+               handle_duplicate_node(mesh, from);
+       }
+
+       from->devclass = from_devclass;
+
+       if(!from->session_id) {
+               from->session_id = session_id;
+       }
+
+       if(!to) {
+               to = new_node();
+               to->status.dirty = true;
+               to->status.blacklisted = mesh->default_blacklist;
+               to->name = xstrdup(to_name);
+               to->devclass = to_devclass;
+
+               to->submesh = NULL;
+
+               if(0 != strcmp(to_submesh_name, CORE_MESH)) {
+                       if(!(to->submesh = lookup_or_create_submesh(mesh, to_submesh_name))) {
+                               return false;
+
+                       }
+               }
+
+               node_add(mesh, to);
+       }
+
+       to->devclass = to_devclass;
+
+       /* Convert addresses */
+
+       address = str2sockaddr(to_address, to_port);
+
+       /* Check if edge already exists */
+
+       e = lookup_edge(from, to);
+
+       if(e) {
+               if(e->weight != weight || e->session_id != session_id || sockaddrcmp(&e->address, &address)) {
+                       if(from == mesh->self) {
+                               logger(mesh, MESHLINK_DEBUG, "Got %s from %s for ourself which does not match existing entry", "ADD_EDGE", c->name);
+                               send_add_edge(mesh, c, e, 0);
+                               return true;
+                       } else {
+                               logger(mesh, MESHLINK_DEBUG, "Got %s from %s which does not match existing entry", "ADD_EDGE", c->name);
+                               edge_del(mesh, e);
+                       }
+               } else {
+                       return true;
+               }
+       } else if(from == mesh->self) {
+               logger(mesh, MESHLINK_WARNING, "Got %s from %s for ourself which does not exist", "ADD_EDGE", c->name);
+               mesh->contradicting_add_edge++;
+               e = new_edge();
+               e->from = from;
+               e->to = to;
+               e->session_id = session_id;
+               send_del_edge(mesh, c, e, mesh->contradicting_add_edge);
+               free_edge(e);
+               return true;
+       }
+
+       e = new_edge();
+       e->from = from;
+       e->to = to;
+       e->address = address;
+       e->weight = weight;
+       e->session_id = session_id;
+       edge_add(mesh, e);
+
+       /* Run MST before or after we tell the rest? */
+
+       graph(mesh);
+
+       if(e->from->submesh && e->to->submesh && (e->from->submesh != e->to->submesh)) {
+               logger(mesh, MESHLINK_ERROR, "Dropping add edge ( %s to %s )", e->from->submesh->name, e->to->submesh->name);
+               return false;
+       }
+
+       if(e->from->submesh) {
+               s = e->from->submesh;
+       } else {
+               s = e->to->submesh;
+       }
+
+       /* Tell the rest about the new edge */
+
+       forward_request(mesh, c, s, request);
+
+       return true;
 }
 
-int send_del_edge(connection_t *c, edge_t *e)
-{
-cp
-  return send_request(c, "%d %lx %s %s", DEL_EDGE, random(),
-                      e->from->name, e->to->name);
+bool send_del_edge(meshlink_handle_t *mesh, connection_t *c, const edge_t *e, int contradictions) {
+       submesh_t *s = NULL;
+
+       if(c->node && c->node->submesh) {
+               if(!submesh_allows_node(e->from->submesh, c->node)) {
+                       return true;
+               }
+
+               if(!submesh_allows_node(e->to->submesh, c->node)) {
+                       return true;
+               }
+       }
+
+       if(e->from->submesh && e->to->submesh && (e->from->submesh != e->to->submesh)) {
+               return true;
+       }
+
+
+       if(e->from->submesh) {
+               s = e->from->submesh;
+       } else {
+               s = e->to->submesh;
+       }
+
+       return send_request(mesh, c, s, "%d %x %s %s %d %x", DEL_EDGE, prng(mesh, UINT_MAX),
+                           e->from->name, e->to->name, contradictions, e->session_id);
 }
 
-int del_edge_h(connection_t *c)
-{
-  edge_t *e;
-  char from_name[MAX_STRING_SIZE];
-  char to_name[MAX_STRING_SIZE];
-  node_t *from, *to;
-cp
-  if(sscanf(c->buffer, "%*d %*x "MAX_STRING" "MAX_STRING"", from_name, to_name) != 2)
-    {
-      syslog(LOG_ERR, _("Got bad %s from %s (%s)"), "DEL_EDGE",
-             c->name, c->hostname);
-      return -1;
-    }
-
-  /* Check if names are valid */
-
-  if(check_id(from_name))
-    {
-      syslog(LOG_ERR, _("Got bad %s from %s (%s): %s"), "DEL_EDGE", c->name, c->hostname, _("invalid name"));
-      return -1;
-    }
-
-  if(check_id(to_name))
-    {
-      syslog(LOG_ERR, _("Got bad %s from %s (%s): %s"), "DEL_EDGE", c->name, c->hostname, _("invalid name"));
-      return -1;
-    }
-
-  if(seen_request(c->buffer))
-    return 0;
-
-  /* Lookup nodes */
-
-  from = lookup_node(from_name);
-  
-  if(!from)
-    {
-      if(debug_lvl >= DEBUG_PROTOCOL)
-        syslog(LOG_ERR, _("Got %s from %s (%s) which does not appear in the edge tree"), "DEL_EDGE", c->name, c->hostname);
-      return 0;
-    }
-
-  to = lookup_node(to_name);
-  
-  if(!to)
-    {
-      if(debug_lvl >= DEBUG_PROTOCOL)
-        syslog(LOG_ERR, _("Got %s from %s (%s) which does not appear in the edge tree"), "DEL_EDGE", c->name, c->hostname);
-      return 0;
-    }
-
-  /* Check if edge exists */
-  
-  e = lookup_edge(from, to);
-  
-  if(!e)
-  {
-    if(debug_lvl >= DEBUG_PROTOCOL)
-      syslog(LOG_WARNING, _("Got %s from %s (%s) which does not appear in the edge tree"), "DEL_EDGE", c->name, c->hostname);
-    return 0;
-  }
-
-  if(e->from == myself)
-  {
-    if(debug_lvl >= DEBUG_PROTOCOL)
-      syslog(LOG_WARNING, _("Got %s from %s (%s) for ourself"), "DEL_EDGE", c->name, c->hostname);
-    send_add_edge(c, e); /* Send back a correction */
-    return 0;
-  }
-
-  /* Tell the rest about the deleted edge */
-
-  forward_request(c);
-
-  /* Delete the edge */
-  
-  edge_del(e);
-
-  /* Run MST before or after we tell the rest? */
-
-  graph();
-cp
-  return 0;
+bool del_edge_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
+       assert(request);
+       assert(*request);
+
+       edge_t *e;
+       char from_name[MAX_STRING_SIZE];
+       char to_name[MAX_STRING_SIZE];
+       node_t *from, *to;
+       int contradictions = 0;
+       uint32_t session_id = 0;
+       submesh_t *s = NULL;
+
+       if(sscanf(request, "%*d %*x "MAX_STRING" "MAX_STRING" %d %x", from_name, to_name, &contradictions, &session_id) < 2) {
+               logger(mesh, MESHLINK_ERROR, "Got bad %s from %s", "DEL_EDGE", c->name);
+               return false;
+       }
+
+       if(seen_request(mesh, request)) {
+               return true;
+       }
+
+       /* Lookup nodes */
+
+       from = lookup_node(mesh, from_name);
+       to = lookup_node(mesh, to_name);
+
+       if(!from) {
+               logger(mesh, MESHLINK_WARNING, "Got %s from %s which does not appear in the edge tree", "DEL_EDGE", c->name);
+               return true;
+       }
+
+       if(!to) {
+               logger(mesh, MESHLINK_WARNING, "Got %s from %s which does not appear in the edge tree", "DEL_EDGE", c->name);
+               return true;
+       }
+
+       if(contradictions > 50) {
+               handle_duplicate_node(mesh, from);
+       }
+
+       /* Check if edge exists */
+
+       e = lookup_edge(from, to);
+
+       if(!e) {
+               logger(mesh, MESHLINK_WARNING, "Got %s from %s which does not appear in the edge tree", "DEL_EDGE", c->name);
+               return true;
+       }
+
+       if(e->from == mesh->self) {
+               logger(mesh, MESHLINK_WARNING, "Got %s from %s for ourself", "DEL_EDGE", c->name);
+               mesh->contradicting_del_edge++;
+               send_add_edge(mesh, c, e, mesh->contradicting_del_edge);    /* Send back a correction */
+               return true;
+       }
+
+       /* Tell the rest about the deleted edge */
+
+
+       if(!e->from->submesh || !e->to->submesh || (e->from->submesh == e->to->submesh)) {
+               if(e->from->submesh) {
+                       s = e->from->submesh;
+               } else {
+                       s = e->to->submesh;
+               }
+
+               /* Tell the rest about the deleted edge */
+               forward_request(mesh, c, s, request);
+
+       } else {
+               logger(mesh, MESHLINK_ERROR, "Dropping del edge ( %s to %s )", e->from->submesh->name, e->to->submesh->name);
+               return false;
+       }
+
+       /* Delete the edge */
+
+       edge_del(mesh, e);
+
+       /* Run MST before or after we tell the rest? */
+
+       graph(mesh);
+
+       /* If the node is not reachable anymore but we remember it had an edge to us, clean it up */
+
+       if(!to->status.reachable) {
+               e = lookup_edge(to, mesh->self);
+
+               if(e) {
+                       send_del_edge(mesh, mesh->everyone, e, 0);
+                       edge_del(mesh, e);
+               }
+       }
+
+       return true;
 }