git.meshlink.io Git - meshlink/blob - src/protocol_edge.c

   1 /*
   2     protocol_edge.c -- handle the meta-protocol, edges
   3     Copyright (C) 2014 Guus Sliepen <guus@meshlink.io>
   4
   5     This program is free software; you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation; either version 2 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License along
  16     with this program; if not, write to the Free Software Foundation, Inc.,
  17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 */
  19
  20 #include "system.h"
  21
  22 #include "conf.h"
  23 #include "connection.h"
  24 #include "edge.h"
  25 #include "graph.h"
  26 #include "logger.h"
  27 #include "meshlink_internal.h"
  28 #include "meta.h"
  29 #include "net.h"
  30 #include "netutl.h"
  31 #include "node.h"
  32 #include "protocol.h"
  33 #include "utils.h"
  34 #include "xalloc.h"
  35
  36 extern bool node_write_devclass(meshlink_handle_t *mesh, node_t *n);
  37
  38 bool send_add_edge(meshlink_handle_t *mesh, connection_t *c, const edge_t *e) {
  39         bool x;
  40         char *address, *port;
  41
  42         sockaddr2str(&e->address, &address, &port);
  43
  44         x = send_request(mesh, c, "%d %x %s %d %s %s %s %d %x %d", ADD_EDGE, rand(),
  45                                          e->from->name, e->from->devclass, e->to->name, address, port, e->to->devclass,
  46                                          e->options, e->weight);
  47         free(address);
  48         free(port);
  49
  50         return x;
  51 }
  52
  53 bool add_edge_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
  54         edge_t *e;
  55         node_t *from, *to;
  56         char from_name[MAX_STRING_SIZE];
  57         int from_devclass;
  58         char to_name[MAX_STRING_SIZE];
  59         char to_address[MAX_STRING_SIZE];
  60         char to_port[MAX_STRING_SIZE];
  61         int to_devclass;
  62         sockaddr_t address;
  63         uint32_t options;
  64         int weight;
  65
  66         if(sscanf(request, "%*d %*x "MAX_STRING" %d "MAX_STRING" "MAX_STRING" "MAX_STRING" %d %x %d",
  67                           from_name, &from_devclass, to_name, to_address, to_port, &to_devclass, &options, &weight) != 8) {
  68                 logger(mesh, MESHLINK_ERROR, "Got bad %s from %s (%s)", "ADD_EDGE", c->name,
  69                            c->hostname);
  70                 return false;
  71         }
  72
  73         /* Check if names are valid */
  74
  75         if(!check_id(from_name) || !check_id(to_name)) {
  76                 logger(mesh, MESHLINK_ERROR, "Got bad %s from %s (%s): %s", "ADD_EDGE", c->name,
  77                            c->hostname, "invalid name");
  78                 return false;
  79         }
  80
  81         // Check if devclasses are valid
  82
  83         if(from_devclass < 0 || from_devclass > _DEV_CLASS_MAX) {
  84                 logger(mesh, MESHLINK_ERROR, "Got bad %s from %s (%s): %s", "ADD_EDGE", c->name,
  85                            c->hostname, "from devclass invalid");
  86                 return false;
  87         }
  88
  89         if(to_devclass < 0 || to_devclass > _DEV_CLASS_MAX) {
  90                 logger(mesh, MESHLINK_ERROR, "Got bad %s from %s (%s): %s", "ADD_EDGE", c->name,
  91                            c->hostname, "to devclass invalid");
  92                 return false;
  93         }
  94
  95         if(seen_request(mesh, request))
  96                 return true;
  97
  98         /* Lookup nodes */
  99
 100         from = lookup_node(mesh, from_name);
 101         to = lookup_node(mesh, to_name);
 102
 103         if(!from) {
 104                 from = new_node();
 105                 from->name = xstrdup(from_name);
 106                 node_add(mesh, from);
 107         }
 108
 109         from->devclass = from_devclass;
 110         node_write_devclass(mesh, from);
 111
 112         if(!to) {
 113                 to = new_node();
 114                 to->name = xstrdup(to_name);
 115                 node_add(mesh, to);
 116         }
 117
 118         to->devclass = to_devclass;
 119         node_write_devclass(mesh, to);
 120
 121         /* Convert addresses */
 122
 123         address = str2sockaddr(to_address, to_port);
 124
 125         /* Check if edge already exists */
 126
 127         e = lookup_edge(from, to);
 128
 129         if(e) {
 130                 if(e->weight != weight || e->options != options || sockaddrcmp(&e->address, &address)) {
 131                         if(from == mesh->self) {
 132                                 logger(mesh, MESHLINK_WARNING, "Got %s from %s (%s) for ourself which does not match existing entry",
 133                                                    "ADD_EDGE", c->name, c->hostname);
 134                                 send_add_edge(mesh, c, e);
 135                                 return true;
 136                         } else {
 137                                 logger(mesh, MESHLINK_WARNING, "Got %s from %s (%s) which does not match existing entry",
 138                                                    "ADD_EDGE", c->name, c->hostname);
 139                                 edge_del(mesh, e);
 140                                 graph(mesh);
 141                         }
 142                 } else
 143                         return true;
 144         } else if(from == mesh->self) {
 145                 logger(mesh, MESHLINK_WARNING, "Got %s from %s (%s) for ourself which does not exist",
 146                                    "ADD_EDGE", c->name, c->hostname);
 147                 mesh->contradicting_add_edge++;
 148                 e = new_edge();
 149                 e->from = from;
 150                 e->to = to;
 151                 send_del_edge(mesh, c, e);
 152                 free_edge(e);
 153                 return true;
 154         }
 155
 156         e = new_edge();
 157         e->from = from;
 158         e->to = to;
 159         e->address = address;
 160         e->options = options;
 161         e->weight = weight;
 162         edge_add(mesh, e);
 163
 164         /* Tell the rest about the new edge */
 165
 166         forward_request(mesh, c, request);
 167
 168         /* Run MST before or after we tell the rest? */
 169
 170         graph(mesh);
 171
 172         return true;
 173 }
 174
 175 bool send_del_edge(meshlink_handle_t *mesh, connection_t *c, const edge_t *e) {
 176         return send_request(mesh, c, "%d %x %s %s", DEL_EDGE, rand(),
 177                                                 e->from->name, e->to->name);
 178 }
 179
 180 bool del_edge_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
 181         edge_t *e;
 182         char from_name[MAX_STRING_SIZE];
 183         char to_name[MAX_STRING_SIZE];
 184         node_t *from, *to;
 185
 186         if(sscanf(request, "%*d %*x "MAX_STRING" "MAX_STRING, from_name, to_name) != 2) {
 187                 logger(mesh, MESHLINK_ERROR, "Got bad %s from %s (%s)", "DEL_EDGE", c->name,
 188                            c->hostname);
 189                 return false;
 190         }
 191
 192         /* Check if names are valid */
 193
 194         if(!check_id(from_name) || !check_id(to_name)) {
 195                 logger(mesh, MESHLINK_ERROR, "Got bad %s from %s (%s): %s", "DEL_EDGE", c->name,
 196                            c->hostname, "invalid name");
 197                 return false;
 198         }
 199
 200         if(seen_request(mesh, request))
 201                 return true;
 202
 203         /* Lookup nodes */
 204
 205         from = lookup_node(mesh, from_name);
 206         to = lookup_node(mesh, to_name);
 207
 208         if(!from) {
 209                 logger(mesh, MESHLINK_ERROR, "Got %s from %s (%s) which does not appear in the edge tree",
 210                                    "DEL_EDGE", c->name, c->hostname);
 211                 return true;
 212         }
 213
 214         if(!to) {
 215                 logger(mesh, MESHLINK_ERROR, "Got %s from %s (%s) which does not appear in the edge tree",
 216                                    "DEL_EDGE", c->name, c->hostname);
 217                 return true;
 218         }
 219
 220         /* Check if edge exists */
 221
 222         e = lookup_edge(from, to);
 223
 224         if(!e) {
 225                 logger(mesh, MESHLINK_WARNING, "Got %s from %s (%s) which does not appear in the edge tree",
 226                                    "DEL_EDGE", c->name, c->hostname);
 227                 return true;
 228         }
 229
 230         if(e->from == mesh->self) {
 231                 logger(mesh, MESHLINK_WARNING, "Got %s from %s (%s) for ourself",
 232                                    "DEL_EDGE", c->name, c->hostname);
 233                 mesh->contradicting_del_edge++;
 234                 send_add_edge(mesh, c, e);    /* Send back a correction */
 235                 return true;
 236         }
 237
 238         /* Tell the rest about the deleted edge */
 239
 240         forward_request(mesh, c, request);
 241
 242         /* Delete the edge */
 243
 244         edge_del(mesh, e);
 245
 246         /* Run MST before or after we tell the rest? */
 247
 248         graph(mesh);
 249
 250         /* If the node is not reachable anymore but we remember it had an edge to us, clean it up */
 251
 252         if(!to->status.reachable) {
 253                 e = lookup_edge(to, mesh->self);
 254                 if(e) {
 255                         send_del_edge(mesh, mesh->everyone, e);
 256                         edge_del(mesh, e);
 257                 }
 258         }
 259
 260         return true;
 261 }