/*
net.c -- most of the network code
- Copyright (C) 2014 Guus Sliepen <guus@meshlink.io>
+ Copyright (C) 2014-2017 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
#include "system.h"
-#include "utils.h"
+#include "autoconnect.h"
#include "conf.h"
#include "connection.h"
#include "graph.h"
#include "net.h"
#include "netutl.h"
#include "protocol.h"
+#include "utils.h"
#include "xalloc.h"
#include <assert.h>
-static const int min(int a, int b) {
+#if !defined(min)
+static inline int min(int a, int b) {
return a < b ? a : b;
}
+#endif
+
+static const int default_timeout = 5;
/*
Terminate a connection:
- Check if we need to retry making an outgoing connection
*/
void terminate_connection(meshlink_handle_t *mesh, connection_t *c, bool report) {
- logger(mesh, MESHLINK_INFO, "Closing connection with %s (%s)", c->name, c->hostname);
+ logger(mesh, MESHLINK_INFO, "Closing connection with %s", c->name);
c->status.active = false;
- if(c->node && c->node->connection == c)
+ if(c->node && c->node->connection == c) {
c->node->connection = NULL;
+ }
if(c->edge) {
- if(report)
- send_del_edge(mesh, mesh->everyone, c->edge);
+ if(report) {
+ send_del_edge(mesh, mesh->everyone, c->edge, 0);
+ }
edge_del(mesh, c->edge);
c->edge = NULL;
/* If the node is not reachable anymore but we remember it had an edge to us, clean it up */
- if(report && !c->node->status.reachable) {
+ if(report && c->node && !c->node->status.reachable) {
edge_t *e;
e = lookup_edge(c->node, mesh->self);
+
if(e) {
- send_del_edge(mesh, mesh->everyone, e);
+ send_del_edge(mesh, mesh->everyone, e, 0);
edge_del(mesh, e);
}
}
/* Check if this was our outgoing connection */
- if(outgoing)
+ if(outgoing) {
do_outgoing_connection(mesh, outgoing);
+ }
#ifndef HAVE_MINGW
/* Clean up dead proxy processes */
while(waitpid(-1, NULL, WNOHANG) > 0);
+
#endif
}
logger(mesh, MESHLINK_DEBUG, "timeout_handler()");
for list_each(connection_t, c, mesh->connections) {
+ int pingtimeout = c->node ? mesh->dev_class_traits[c->node->devclass].pingtimeout : default_timeout;
+
// Also make sure that if outstanding key requests for the UDP counterpart of a connection has timed out, we restart it.
if(c->node) {
- if(c->node->status.waitingforkey && c->node->last_req_key + mesh->pingtimeout <= mesh->loop.now.tv_sec)
+ if(c->node->status.waitingforkey && c->node->last_req_key + pingtimeout <= mesh->loop.now.tv_sec) {
send_req_key(mesh, c->node);
+ }
}
- if(c->last_ping_time + mesh->pingtimeout <= mesh->loop.now.tv_sec) {
+
+ if(c->last_ping_time + pingtimeout <= mesh->loop.now.tv_sec) {
if(c->status.active) {
if(c->status.pinged) {
- logger(mesh, MESHLINK_INFO, "%s (%s) didn't respond to PING in %ld seconds", c->name, c->hostname, (long)mesh->loop.now.tv_sec - c->last_ping_time);
- } else if(c->last_ping_time + mesh->pinginterval <= mesh->loop.now.tv_sec) {
+ logger(mesh, MESHLINK_INFO, "%s didn't respond to PING in %ld seconds", c->name, (long)mesh->loop.now.tv_sec - c->last_ping_time);
+ } else if(c->last_ping_time + mesh->dev_class_traits[c->node->devclass].pinginterval <= mesh->loop.now.tv_sec) {
send_ping(mesh, c);
continue;
} else {
continue;
}
} else {
- if(c->status.connecting)
- logger(mesh, MESHLINK_WARNING, "Timeout while connecting to %s (%s)", c->name, c->hostname);
- else
- logger(mesh, MESHLINK_WARNING, "Timeout from %s (%s) during authentication", c->name, c->hostname);
+ if(c->status.connecting) {
+ logger(mesh, MESHLINK_WARNING, "Timeout while connecting to %s", c->name);
+ } else {
+ logger(mesh, MESHLINK_WARNING, "Timeout from %s during authentication", c->name);
+ }
}
+
terminate_connection(mesh, c, c->status.active);
}
}
- timeout_set(&mesh->loop, data, &(struct timeval){mesh->pingtimeout, rand() % 100000});
-}
-
-// devclass asc, last_connect_try desc
-static int node_compare_devclass_asc_last_connect_try_desc(const void *a, const void *b)
-{
- const node_t *na = a, *nb = b;
-
- if(na->devclass < nb->devclass)
- { return -1; }
-
- if(na->devclass > nb->devclass)
- { return 1; }
-
- if(na->last_connect_try == nb->last_connect_try)
- return 0;
-
- if(nb->last_connect_try == 0 || na->last_connect_try < nb->last_connect_try)
- return -1;
-
- if(na->last_connect_try == 0 || na->last_connect_try > nb->last_connect_try)
- return 1;
-
- return 0;
-}
-
-// last_connect_try desc
-static int node_compare_last_connect_try_desc(const void *a, const void *b)
-{
- const node_t *na = a, *nb = b;
-
- if(na->last_connect_try == nb->last_connect_try)
- return 0;
-
- if(nb->last_connect_try == 0 || na->last_connect_try < nb->last_connect_try)
- return -1;
-
- if(na->last_connect_try == 0 || na->last_connect_try > nb->last_connect_try)
- return 1;
-
- return 0;
-}
-
-// devclass desc
-static int node_compare_devclass_desc(const void *a, const void *b)
-{
- const node_t *na = a, *nb = b;
-
- if(na->devclass < nb->devclass)
- { return -1; }
-
- if(na->devclass > nb->devclass)
- { return 1; }
-
- return 0;
-}
-
-
-/*
-
-
-autoconnect()
-{
- timeout = 5
-
- // find the best one for initial connect
-
- if cur < min
- newcon =
- first from nodes
- where dclass <= my.dclass and !connection and (timestamp - last_retry) > retry_timeout
- order by dclass asc, last_connection desc
- if newcon
- timeout = 0
- goto connect
-
-
- // find better nodes to connect to: in case we have less than min connections within [BACKBONE, i] and there are nodes which we are not connected to within the range
-
- if min <= cur < max
- j = 0
- for i = BACKBONE to my.dclass
- j += count(from connections where node.dclass = i)
- if j < min
- newcon =
- first from nodes
- where dclass = i and !connection and (timestamp - last_retry) > retry_timeout
- order by last_connection desc
- if newcon
- goto connect
- else
- break
-
-
- // heal partitions
-
- if min <= cur < max
- newcon =
- first from nodes
- where dclass <= my.dclass and !reachable and (timestamp - last_retry) > retry_timeout
- order by dclass asc, last_connection desc
- if newcon
- goto connect
-
-
- // connect
-
-connect:
- if newcon
- connect newcon
-
-
- // disconnect outgoing connections in case we have more than min connections within [BACKBONE, i] and there are nodes which we are connected to within the range [i, PORTABLE]
-
- if min < cur <= max
- j = 0
- for i = BACKBONE to my.dclass
- j += count(from connections where node.dclass = i)
- if min < j
- delcon =
- first from nodes
- where dclass >= i and outgoing_connection
- order by dclass desc
- if disconnect
- goto disconnect
- else
- break
-
-
- // disconnect connections in case we have more than enough connections
-
- if max < cur
- delcon =
- first from nodes
- where outgoing_connection
- order by dclass desc
- goto disconnect
-
- // disconnect
-
-disconnect
- if delcon
- disconnect delcon
-
-
- // next iteration
- next (timeout, autoconnect)
-
+ timeout_set(&mesh->loop, data, &(struct timeval) {
+ default_timeout, rand() % 100000
+ });
}
-
- */
-
-
static void periodic_handler(event_loop_t *loop, void *data) {
meshlink_handle_t *mesh = loop->data;
logger(mesh, MESHLINK_WARNING, "Possible node with same Name as us! Sleeping %d seconds.", mesh->sleeptime);
usleep(mesh->sleeptime * 1000000LL);
mesh->sleeptime *= 2;
- if(mesh->sleeptime < 0)
+
+ if(mesh->sleeptime < 0) {
mesh->sleeptime = 3600;
+ }
} else {
mesh->sleeptime /= 2;
- if(mesh->sleeptime < 10)
+
+ if(mesh->sleeptime < 10) {
mesh->sleeptime = 10;
+ }
}
mesh->contradicting_add_edge = 0;
mesh->contradicting_del_edge = 0;
- int timeout = 5;
+ int timeout = default_timeout;
/* Check if we need to make or break connections. */
if(mesh->nodes->count > 1) {
+ unsigned int cur_connects = do_autoconnect(mesh);
- logger(mesh, MESHLINK_INFO, "--- autoconnect begin ---");
-
-
- int retry_timeout = min(mesh->nodes->count * 5, 60);
-
- // connect disconnect nodes
-
- node_t* connect_to = NULL;
- node_t* disconnect_from = NULL;
-
-
- // get cur_connects
-
- int cur_connects = 0;
-
- for list_each(connection_t, c, mesh->connections)
- {
- if(!c->status.remove_unused)
- {
- cur_connects += 1;
- }
+ // reduce timeout if we don't have enough connections + outgoings
+ if (cur_connects + mesh->outgoings->count < 3) {
+ timeout = 1;
}
+ }
- logger(mesh, MESHLINK_INFO, "* cur_connects = %d", cur_connects);
-
-
- // get min_connects and max_connects
-
- assert(mesh->devclass >= 0 && mesh->devclass <= _DEV_CLASS_MAX);
-
- int min_connects = dev_class_traits[mesh->devclass].min_connects;
- int max_connects = dev_class_traits[mesh->devclass].max_connects;
-
- logger(mesh, MESHLINK_INFO, "* min_connects = %d", min_connects);
- logger(mesh, MESHLINK_INFO, "* max_connects = %d", max_connects);
-
-
- // find the best one for initial connect
-
- if(cur_connects < min_connects)
- {
- splay_tree_t *nodes = splay_alloc_tree(node_compare_devclass_asc_last_connect_try_desc, NULL);
-
- for splay_each(node_t, n, mesh->nodes)
- {
- if(n->devclass <= mesh->devclass && !n->connection && (n->last_connect_try == 0 || (time(NULL) - n->last_connect_try) > retry_timeout))
- { splay_insert(nodes, n); }
- }
-
- if(nodes->head)
- {
- logger(mesh, MESHLINK_INFO, "* found best one for initial connect");
-
- //timeout = 0;
- connect_to = (node_t*)nodes->head->data;
- }
+ /* Write dirty config files out to disk */
- splay_free_tree(nodes);
+ for splay_each(node_t, n, mesh->nodes) {
+ if(n->status.dirty) {
+ node_write_config(mesh, n);
+ n->status.dirty = false;
}
+ }
+ timeout_set(&mesh->loop, data, &(struct timeval) {
+ timeout, rand() % 100000
+ });
+}
- // find better nodes to connect to
-
- if(!connect_to && min_connects <= cur_connects && cur_connects < max_connects)
- {
- unsigned int connects = 0;
-
- for(int devclass = 0; devclass <= mesh->devclass; ++devclass)
- {
- for list_each(connection_t, c, mesh->connections)
- {
- if(!c->status.remove_unused && c->node && c->node->devclass == devclass)
- { connects += 1; }
- }
-
- if( connects < min_connects )
- {
- splay_tree_t *nodes = splay_alloc_tree(node_compare_last_connect_try_desc, NULL);
+void handle_meta_connection_data(meshlink_handle_t *mesh, connection_t *c) {
+ if(!receive_meta(mesh, c)) {
+ terminate_connection(mesh, c, c->status.active);
+ return;
+ }
+}
- for splay_each(node_t, n, mesh->nodes)
- {
- if(n->devclass == devclass && !n->connection && (n->last_connect_try == 0 || (time(NULL) - n->last_connect_try) > retry_timeout))
- { splay_insert(nodes, n); }
- }
+void retry(meshlink_handle_t *mesh) {
+ /* Reset the reconnection timers for all outgoing connections */
+ for list_each(outgoing_t, outgoing, mesh->outgoings) {
+ outgoing->timeout = 0;
- if(nodes->head)
- {
- logger(mesh, MESHLINK_INFO, "* found better node");
- connect_to = (node_t*)nodes->head->data;
+ if(outgoing->ev.cb)
+ timeout_set(&mesh->loop, &outgoing->ev, &(struct timeval) {
+ 0, 0
+ });
+ }
- splay_free_tree(nodes);
- break;
- }
+#ifdef HAVE_IFADDRS_H
+ struct ifaddrs *ifa = NULL;
+ getifaddrs(&ifa);
+#endif
- splay_free_tree(nodes);
- }
- else
- { break; }
- }
+ /* For active connections, check if their addresses are still valid.
+ * If yes, reset their ping timers, otherwise terminate them. */
+ for list_each(connection_t, c, mesh->connections) {
+ if(!c->status.active) {
+ continue;
}
-
- // heal partitions
-
- if(!connect_to && min_connects <= cur_connects && cur_connects < max_connects)
- {
- splay_tree_t *nodes = splay_alloc_tree(node_compare_devclass_asc_last_connect_try_desc, NULL);
-
- for splay_each(node_t, n, mesh->nodes)
- {
- if(n->devclass <= mesh->devclass && !n->status.reachable && (n->last_connect_try == 0 || (time(NULL) - n->last_connect_try) > retry_timeout))
- { splay_insert(nodes, n); }
- }
-
- if(nodes->head)
- {
- logger(mesh, MESHLINK_INFO, "* try to heal partition");
- connect_to = (node_t*)nodes->head->data;
- }
-
- splay_free_tree(nodes);
+ if(!c->status.pinged) {
+ c->last_ping_time = 0;
}
+#ifdef HAVE_IFADDRS_H
- // perform connect
-
- if(connect_to && !connect_to->connection)
- {
- /* check if there is already a connection attempt to this node */
- bool found = false;
- for list_each(outgoing_t, outgoing, mesh->outgoings) {
- if(!strcmp(outgoing->name, connect_to->name)) {
- found = true;
- break;
- }
- }
-
- if(!found)
- {
- logger(mesh, MESHLINK_INFO, "Autoconnecting to %s", connect_to->name);
- outgoing_t *outgoing = xzalloc(sizeof(outgoing_t));
- outgoing->mesh = mesh;
- outgoing->name = xstrdup(connect_to->name);
- list_insert_tail(mesh->outgoings, outgoing);
- setup_outgoing_connection(mesh, outgoing);
- }
+ if(!ifa) {
+ continue;
}
+ sockaddr_t sa;
+ socklen_t salen = sizeof(sa);
- // disconnect suboptimal outgoing connections
-
- if(min_connects < cur_connects && cur_connects <= max_connects)
- {
- unsigned int connects = 0;
-
- for(int devclass = 0; devclass <= mesh->devclass; ++devclass)
- {
- for list_each(connection_t, c, mesh->connections)
- {
- if(!c->status.remove_unused && c->node && c->node->devclass == devclass)
- { connects += 1; }
- }
-
- if( min_connects < connects )
- {
- splay_tree_t *nodes = splay_alloc_tree(node_compare_devclass_desc, NULL);
-
- for list_each(connection_t, c, mesh->connections)
- {
- if(!c->status.remove_unused && c->outgoing && c->node && c->node->devclass >= devclass)
- { splay_insert(nodes, c->node); }
- }
-
- if(nodes->head)
- {
- logger(mesh, MESHLINK_INFO, "* disconnect suboptimal outgoing connection");
- disconnect_from = (node_t*)nodes->head->data;
- }
-
- splay_free_tree(nodes);
- break;
- }
- }
+ if(getsockname(c->socket, &sa.sa, &salen)) {
+ continue;
}
+ bool found = false;
- // disconnect connections (too many connections)
-
- if(!disconnect_from && max_connects < cur_connects)
- {
- splay_tree_t *nodes = splay_alloc_tree(node_compare_devclass_desc, NULL);
-
- for list_each(connection_t, c, mesh->connections)
- {
- if(!c->status.remove_unused && c->node)
- { splay_insert(nodes, c->node); }
+ for(struct ifaddrs *ifap = ifa; ifap; ifap = ifap->ifa_next) {
+ if(ifap->ifa_addr && !sockaddrcmp_noport(&sa, (sockaddr_t *)ifap->ifa_addr)) {
+ found = true;
+ break;
}
- if(nodes->head)
- {
- logger(mesh, MESHLINK_INFO, "* disconnect connection (too many connections");
-
- //timeout = 0;
- disconnect_from = (node_t*)nodes->head->data;
- }
-
- splay_free_tree(nodes);
}
-
- // perform disconnect
-
- if(disconnect_from && disconnect_from->connection)
- {
- logger(mesh, MESHLINK_INFO, "Autodisconnecting from %s", disconnect_from->connection->name);
- list_delete(mesh->outgoings, disconnect_from->connection->outgoing);
- disconnect_from->connection->outgoing = NULL;
- terminate_connection(mesh, disconnect_from->connection, disconnect_from->connection->status.active);
+ if(!found) {
+ logger(mesh, MESHLINK_DEBUG, "Local address for connection to %s no longer valid, terminating", c->name);
+ terminate_connection(mesh, c, c->status.active);
}
-
- // done!
-
- logger(mesh, MESHLINK_INFO, "--- autoconnect end ---");
+#endif
}
- timeout_set(&mesh->loop, data, &(struct timeval){timeout, rand() % 100000});
-}
+#ifdef HAVE_IFADDRS_H
-void handle_meta_connection_data(meshlink_handle_t *mesh, connection_t *c) {
- if (!receive_meta(mesh, c)) {
- terminate_connection(mesh, c, c->status.active);
- return;
+ if(ifa) {
+ freeifaddrs(ifa);
}
-}
-void retry(meshlink_handle_t *mesh) {
- /* Reset the reconnection timers for all outgoing connections */
- for list_each(outgoing_t, outgoing, mesh->outgoings) {
- outgoing->timeout = 0;
- if(outgoing->ev.cb)
- timeout_set(&mesh->loop, &outgoing->ev, &(struct timeval){0, 0});
- }
-
- /* Check for outgoing connections that are in progress, and reset their ping timers */
- for list_each(connection_t, c, mesh->connections) {
- if(c->outgoing && !c->node)
- c->last_ping_time = 0;
- }
+#endif
/* Kick the ping timeout handler */
- timeout_set(&mesh->loop, &mesh->pingtimer, &(struct timeval){0, 0});
+ timeout_set(&mesh->loop, &mesh->pingtimer, &(struct timeval) {
+ 0, 0
+ });
}
/*
this is where it all happens...
*/
int main_loop(meshlink_handle_t *mesh) {
- timeout_add(&mesh->loop, &mesh->pingtimer, timeout_handler, &mesh->pingtimer, &(struct timeval){mesh->pingtimeout, rand() % 100000});
- timeout_add(&mesh->loop, &mesh->periodictimer, periodic_handler, &mesh->periodictimer, &(struct timeval){0, 0});
+ timeout_add(&mesh->loop, &mesh->pingtimer, timeout_handler, &mesh->pingtimer, &(struct timeval) {
+ default_timeout, rand() % 100000
+ });
+ timeout_add(&mesh->loop, &mesh->periodictimer, periodic_handler, &mesh->periodictimer, &(struct timeval) {
+ 0, 0
+ });
//Add signal handler
mesh->datafromapp.signum = 0;
- signal_add(&(mesh->loop),&(mesh->datafromapp), (signal_cb_t)meshlink_send_from_queue,mesh, mesh->datafromapp.signum);
+ signal_add(&(mesh->loop), &(mesh->datafromapp), (signal_cb_t)meshlink_send_from_queue, mesh, mesh->datafromapp.signum);
- if(!event_loop_run(&mesh->loop)) {
+ if(!event_loop_run(&(mesh->loop), &(mesh->mesh_mutex))) {
logger(mesh, MESHLINK_ERROR, "Error while waiting for input: %s", strerror(errno));
+ abort();
+ timeout_del(&mesh->loop, &mesh->periodictimer);
+ timeout_del(&mesh->loop, &mesh->pingtimer);
+
return 1;
}