X-Git-Url: https://tinc-vpn.org/git/browse?p=tinc;a=blobdiff_plain;f=src%2Fnet.c;h=2ffe03a7b39a7985fdb01a7ab1108880a62c0184;hp=1de5f1fb711530eff4420f212062ef26424e06bc;hb=6834b882015a9323672e9fce8811aa4283f196f0;hpb=116065afe352221ac6c2c8e34c109252004d6a59

diff --git a/src/net.c b/src/net.c
index 1de5f1fb..2ffe03a7 100644
--- a/src/net.c
+++ b/src/net.c
@@ -1,7 +1,8 @@
 /*
     net.c -- most of the network code
     Copyright (C) 1998-2005 Ivo Timmermans,
-                  2000-2007 Guus Sliepen <guus@tinc-vpn.org>
+                  2000-2010 Guus Sliepen <guus@tinc-vpn.org>
+                  2006      Scott Lamb <slamb@slamb.org>
 
     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,23 +14,18 @@
     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$
+    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 "system.h"
 
-#include <openssl/rand.h>
-
 #include "utils.h"
-#include "avl_tree.h"
+#include "splay_tree.h"
 #include "conf.h"
 #include "connection.h"
 #include "device.h"
-#include "event.h"
 #include "graph.h"
 #include "logger.h"
 #include "meta.h"
@@ -37,27 +33,21 @@
 #include "netutl.h"
 #include "process.h"
 #include "protocol.h"
-#include "route.h"
 #include "subnet.h"
 #include "xalloc.h"
 
-bool do_purge = false;
-volatile bool running = false;
-
-time_t now = 0;
+int contradicting_add_edge = 0;
+int contradicting_del_edge = 0;
 
 /* Purge edges and subnets of unreachable nodes. Use carefully. */
 
-static void purge(void)
-{
-	avl_node_t *nnode, *nnext, *enode, *enext, *snode, *snext;
+void purge(void) {
+	splay_node_t *nnode, *nnext, *enode, *enext, *snode, *snext;
 	node_t *n;
 	edge_t *e;
 	subnet_t *s;
 
-	cp();
-
-	ifdebug(PROTOCOL) logger(LOG_DEBUG, _("Purging unreachable nodes"));
+	ifdebug(PROTOCOL) logger(LOG_DEBUG, "Purging unreachable nodes");
 
 	/* Remove all edges and subnets owned by unreachable nodes. */
 
@@ -66,15 +56,15 @@ static void purge(void)
 		n = nnode->data;
 
 		if(!n->status.reachable) {
-			ifdebug(SCARY_THINGS) logger(LOG_DEBUG, _("Purging node %s (%s)"), n->name,
+			ifdebug(SCARY_THINGS) logger(LOG_DEBUG, "Purging node %s (%s)", n->name,
 					   n->hostname);
 
 			for(snode = n->subnet_tree->head; snode; snode = snext) {
 				snext = snode->next;
 				s = snode->data;
-				if(!tunnelserver)
-					send_del_subnet(broadcast, s);
-				subnet_del(n, s);
+				send_del_subnet(broadcast, s);
+				if(!strictsubnets)
+					subnet_del(n, s);
 			}
 
 			for(enode = n->edge_tree->head; enode; enode = enext) {
@@ -102,60 +92,13 @@ static void purge(void)
 					break;
 			}
 
-			if(!enode)
+			if(!enode && (!strictsubnets || !n->subnet_tree->head))
+				/* in strictsubnets mode do not delete nodes with subnets */
 				node_del(n);
 		}
 	}
 }
 
-/*
-  put all file descriptors in an fd_set array
-  While we're at it, purge stuff that needs to be removed.
-*/
-static int build_fdset(fd_set *readset, fd_set *writeset)
-{
-	avl_node_t *node, *next;
-	connection_t *c;
-	int i, max = 0;
-
-	cp();
-
-	FD_ZERO(readset);
-	FD_ZERO(writeset);
-
-	for(node = connection_tree->head; node; node = next) {
-		next = node->next;
-		c = node->data;
-
-		if(c->status.remove) {
-			connection_del(c);
-			if(!connection_tree->head)
-				purge();
-		} else {
-			FD_SET(c->socket, readset);
-			if(c->outbuflen > 0)
-				FD_SET(c->socket, writeset);
-			if(c->socket > max)
-				max = c->socket;
-		}
-	}
-
-	for(i = 0; i < listen_sockets; i++) {
-		FD_SET(listen_socket[i].tcp, readset);
-		if(listen_socket[i].tcp > max)
-			max = listen_socket[i].tcp;
-		FD_SET(listen_socket[i].udp, readset);
-		if(listen_socket[i].udp > max)
-			max = listen_socket[i].udp;
-	}
-
-	FD_SET(device_fd, readset);
-	if(device_fd > max)
-		max = device_fd;
-	
-	return max;
-}
-
 /*
   Terminate a connection:
   - Close the socket
@@ -163,17 +106,10 @@ static int build_fdset(fd_set *readset, fd_set *writeset)
   - Check if we need to retry making an outgoing connection
   - Deactivate the host
 */
-void terminate_connection(connection_t *c, bool report)
-{
-	cp();
-
-	if(c->status.remove)
-		return;
-
-	ifdebug(CONNECTIONS) logger(LOG_NOTICE, _("Closing connection with %s (%s)"),
+void terminate_connection(connection_t *c, bool report) {
+	ifdebug(CONNECTIONS) logger(LOG_NOTICE, "Closing connection with %s (%s)",
 			   c->name, c->hostname);
 
-	c->status.remove = true;
 	c->status.active = false;
 
 	if(c->node)
@@ -207,16 +143,10 @@ void terminate_connection(connection_t *c, bool report)
 
 	/* Check if this was our outgoing connection */
 
-	if(c->outgoing) {
+	if(c->outgoing)
 		retry_outgoing(c->outgoing);
-		c->outgoing = NULL;
-	}
 
-	free(c->outbuf);
-	c->outbuf = NULL;
-	c->outbuflen = 0;
-	c->outbufsize = 0;
-	c->outbufstart = 0;
+	connection_del(c);
 }
 
 /*
@@ -227,12 +157,11 @@ void terminate_connection(connection_t *c, bool report)
   end does not reply in time, we consider them dead
   and close the connection.
 */
-static void check_dead_connections(void)
-{
-	avl_node_t *node, *next;
+static void timeout_handler(void *arg) {
+	event_t *event = arg;
+	splay_node_t *node, *next;
 	connection_t *c;
-
-	cp();
+	time_t now = time(NULL);
 
 	for(node = connection_tree->head; node; node = next) {
 		next = node->next;
@@ -241,250 +170,219 @@ static void check_dead_connections(void)
 		if(c->last_ping_time + pingtimeout < now) {
 			if(c->status.active) {
 				if(c->status.pinged) {
-					ifdebug(CONNECTIONS) logger(LOG_INFO, _("%s (%s) didn't respond to PING in %ld seconds"),
+					ifdebug(CONNECTIONS) logger(LOG_INFO, "%s (%s) didn't respond to PING in %ld seconds",
 							   c->name, c->hostname, now - c->last_ping_time);
-					c->status.timeout = true;
 					terminate_connection(c, true);
+					continue;
 				} else if(c->last_ping_time + pinginterval < now) {
 					send_ping(c);
 				}
 			} else {
-				if(c->status.remove) {
-					logger(LOG_WARNING, _("Old connection_t for %s (%s) status %04x still lingering, deleting..."),
-						   c->name, c->hostname, c->status.value);
-					connection_del(c);
-					continue;
-				}
-				ifdebug(CONNECTIONS) logger(LOG_WARNING, _("Timeout from %s (%s) during authentication"),
-						   c->name, c->hostname);
 				if(c->status.connecting) {
+					ifdebug(CONNECTIONS)
+						logger(LOG_WARNING, "Timeout while connecting to %s (%s)", c->name, c->hostname);
 					c->status.connecting = false;
 					closesocket(c->socket);
 					do_outgoing_connection(c);
 				} else {
+					ifdebug(CONNECTIONS) logger(LOG_WARNING, "Timeout from %s (%s) during authentication", c->name, c->hostname);
 					terminate_connection(c, false);
+					continue;
 				}
 			}
 		}
-
-		if(c->outbuflen > 0 && c->last_flushed_time + pingtimeout < now) {
-			if(c->status.active) {
-				ifdebug(CONNECTIONS) logger(LOG_INFO,
-						_("%s (%s) could not flush for %ld seconds (%d bytes remaining)"),
-						c->name, c->hostname, now - c->last_flushed_time, c->outbuflen);
-				c->status.timeout = true;
-				terminate_connection(c, true);
-			}
-		}
 	}
-}
 
-/*
-  check all connections to see if anything
-  happened on their sockets
-*/
-static void check_network_activity(fd_set * readset, fd_set * writeset)
-{
-	connection_t *c;
-	avl_node_t *node;
-	int result, i;
-	socklen_t len = sizeof(result);
-	vpn_packet_t packet;
-
-	cp();
-
-	/* check input from kernel */
-	if(FD_ISSET(device_fd, readset)) {
-		if(read_packet(&packet)) {
-			packet.priority = 0;
-			route(myself, &packet);
+	if(contradicting_del_edge && contradicting_add_edge) {
+		logger(LOG_WARNING, "Possible node with same Name as us!");
+
+		if(rand() % 3 == 0) {
+			logger(LOG_ERR, "Shutting down, check configuration of all nodes for duplicate Names!");
+			abort();
+			return;
 		}
+
+		contradicting_add_edge = 0;
+		contradicting_del_edge = 0;
 	}
 
-	/* check meta connections */
-	for(node = connection_tree->head; node; node = node->next) {
-		c = node->data;
+	event->time = now + pingtimeout;
+	event_add(event);
+}
 
-		if(c->status.remove)
-			continue;
+void handle_meta_connection_data(void *data) {
+	connection_t *c = data;
+	int result;
+	socklen_t len = sizeof result;
 
-		if(FD_ISSET(c->socket, readset)) {
-			if(c->status.connecting) {
-				c->status.connecting = false;
-				getsockopt(c->socket, SOL_SOCKET, SO_ERROR, &result, &len);
+	if(c->status.connecting) {
+		c->status.connecting = false;
 
-				if(!result)
-					finish_connecting(c);
-				else {
-					ifdebug(CONNECTIONS) logger(LOG_DEBUG,
-							   _("Error while connecting to %s (%s): %s"),
-							   c->name, c->hostname, strerror(result));
-					closesocket(c->socket);
-					do_outgoing_connection(c);
-					continue;
-				}
-			}
-
-			if(!receive_meta(c)) {
-				terminate_connection(c, c->status.active);
-				continue;
-			}
-		}
+		getsockopt(c->socket, SOL_SOCKET, SO_ERROR, &result, &len);
 
-		if(FD_ISSET(c->socket, writeset)) {
-			if(!flush_meta(c)) {
-				terminate_connection(c, c->status.active);
-				continue;
-			}
+		if(!result) {
+			mutex_lock(&mutex);
+			finish_connecting(c);
+			mutex_unlock(&mutex);
+		} else {
+			ifdebug(CONNECTIONS) logger(LOG_DEBUG,
+					   "Error while connecting to %s (%s): %s",
+					   c->name, c->hostname, sockstrerror(result));
+			closesocket(c->socket);
+			mutex_lock(&mutex);
+			do_outgoing_connection(c);
+			mutex_unlock(&mutex);
+			return;
 		}
 	}
 
-	for(i = 0; i < listen_sockets; i++) {
-		if(FD_ISSET(listen_socket[i].udp, readset))
-			handle_incoming_vpn_data(listen_socket[i].udp);
-
-		if(FD_ISSET(listen_socket[i].tcp, readset))
-			handle_new_meta_connection(listen_socket[i].tcp);
+	while(true) {
+		if (!receive_meta(c)) {
+			terminate_connection(c, c->status.active);
+			break;
+		}
 	}
 }
 
-/*
-  this is where it all happens...
-*/
-int main_loop(void)
-{
-	fd_set readset, writeset;
-	struct timeval tv;
-	int r, maxfd;
-	time_t last_ping_check, last_config_check, last_graph_dump;
-	event_t *event;
-
-	cp();
-
-	last_ping_check = now;
-	last_config_check = now;
-	last_graph_dump = now;
-	
-	srand(now);
-
-	running = true;
-
-	while(running) {
-		now = time(NULL);
+static void sigterm_handler(int signal, short events, void *data) {
+	logger(LOG_NOTICE, "Got %s signal", strsignal(signal));
+	exit(0);
+}
 
-	//	tv.tv_sec = 1 + (rand() & 7);	/* Approx. 5 seconds, randomized to prevent global synchronisation effects */
-		tv.tv_sec = 1;
-		tv.tv_usec = 0;
+static void sighup_handler(int signal, short events, void *data) {
+	logger(LOG_NOTICE, "Got %s signal", strsignal(signal));
+	reload_configuration();
+}
 
-		maxfd = build_fdset(&readset, &writeset);
+int reload_configuration(void) {
+	connection_t *c;
+	splay_node_t *node, *next;
+	char *fname;
+	struct stat s;
+	static time_t last_config_check = 0;
 
-		r = select(maxfd + 1, &readset, &writeset, NULL, &tv);
+	/* Reread our own configuration file */
 
-		if(r < 0) {
-			if(errno != EINTR && errno != EAGAIN) {
-				logger(LOG_ERR, _("Error while waiting for input: %s"),
-					   strerror(errno));
-				cp_trace();
-				dump_connections();
-				return 1;
-			}
+	exit_configuration(&config_tree);
+	init_configuration(&config_tree);
 
-			continue;
-		}
-
-		check_network_activity(&readset, &writeset);
+	if(!read_server_config()) {
+		logger(LOG_ERR, "Unable to reread configuration file, exitting.");
+		abort();
+		return EINVAL;
+	}
 
-		if(do_purge) {
-			purge();
-			do_purge = false;
+	/* Close connections to hosts that have a changed or deleted host config file */
+	
+	for(node = connection_tree->head; node; node = next) {
+		c = node->data;
+		next = node->next;
+		
+		if(c->outgoing) {
+			free(c->outgoing->name);
+			if(c->outgoing->ai)
+				freeaddrinfo(c->outgoing->ai);
+			free(c->outgoing);
+			c->outgoing = NULL;
 		}
+		
+		xasprintf(&fname, "%s/hosts/%s", confbase, c->name);
+		if(stat(fname, &s) || s.st_mtime > last_config_check)
+			terminate_connection(c, c->status.active);
+		free(fname);
+	}
 
-		/* Let's check if everybody is still alive */
-
-		if(last_ping_check + pingtimeout < now) {
-			check_dead_connections();
-			last_ping_check = now;
-
-			if(routing_mode == RMODE_SWITCH)
-				age_subnets();
+	last_config_check = time(NULL);
 
-			age_past_requests();
+	/* If StrictSubnet is set, expire deleted Subnets and read new ones in */
 
-			/* Should we regenerate our key? */
+	if(strictsubnets) {
+		subnet_t *subnet;
 
-			if(keyexpires < now) {
-				ifdebug(STATUS) logger(LOG_INFO, _("Regenerating symmetric key"));
 
-				RAND_pseudo_bytes((unsigned char *)myself->key, myself->keylength);
-				if(myself->cipher)
-					EVP_DecryptInit_ex(&packet_ctx, myself->cipher, NULL, (unsigned char *)myself->key, (unsigned char *)myself->key + myself->cipher->key_len);
-				send_key_changed(broadcast, myself);
-				keyexpires = now + keylifetime;
-			}
+		for(node = subnet_tree->head; node; node = node->next) {
+			subnet = node->data;
+			subnet->expires = 1;
 		}
 
-		if(sigalrm) {
-			logger(LOG_INFO, _("Flushing event queue"));
-			expire_events();
-			sigalrm = false;
-		}
-
-		while((event = get_expired_event())) {
-			event->handler(event->data);
-			free_event(event);
+		load_all_subnets();
+
+		for(node = subnet_tree->head; node; node = next) {
+			next = node->next;
+			subnet = node->data;
+			if(subnet->expires == 1) {
+				send_del_subnet(broadcast, subnet);
+				if(subnet->owner->status.reachable)
+					subnet_update(subnet->owner, subnet, false);
+				subnet_del(subnet->owner, subnet);
+			} else if(subnet->expires == -1) {
+				subnet->expires = 0;
+			} else {
+				send_add_subnet(broadcast, subnet);
+				if(subnet->owner->status.reachable)
+					subnet_update(subnet->owner, subnet, true);
+			}
 		}
+	}
 
-		if(sighup) {
-			connection_t *c;
-			avl_node_t *node;
-			char *fname;
-			struct stat s;
-			
-			sighup = false;
-			
-			/* Reread our own configuration file */
-
-			exit_configuration(&config_tree);
-			init_configuration(&config_tree);
-
-			if(!read_server_config()) {
-				logger(LOG_ERR, _("Unable to reread configuration file, exitting."));
-				return 1;
-			}
+	/* Try to make outgoing connections */
+	
+	try_outgoing_connections();
 
-			/* Close connections to hosts that have a changed or deleted host config file */
-			
-			for(node = connection_tree->head; node; node = node->next) {
-				c = node->data;
-				
-				if(c->outgoing) {
-					free(c->outgoing->name);
-					if(c->outgoing->ai)
-						freeaddrinfo(c->outgoing->ai);
-					free(c->outgoing);
-					c->outgoing = NULL;
-				}
-				
-				asprintf(&fname, "%s/hosts/%s", confbase, c->name);
-				if(stat(fname, &s) || s.st_mtime > last_config_check)
-					terminate_connection(c, c->status.active);
-				free(fname);
-			}
+	return 0;
+}
 
-			last_config_check = now;
+void retry(void) {
+	connection_t *c;
+	splay_node_t *node;
 
-			/* Try to make outgoing connections */
-			
-			try_outgoing_connections();
-		}
+	for(node = connection_tree->head; node; node = node->next) {
+		c = node->data;
 		
-		/* Dump graph if wanted every 60 seconds*/
+		if(c->outgoing && !c->node) {
+			event_del(&c->outgoing->ev);
+			if(c->status.connecting)
+				close(c->socket);
+			c->outgoing->timeout = 0;
+			do_outgoing_connection(c);
+		}
+	}
+}
 
-		if(last_graph_dump + 60 < now) {
-			dump_graph();
-			last_graph_dump = now;
+/*
+  this is where it all happens...
+*/
+int main_loop(void) {
+	struct event timeout_event;
+
+	timeout_event.time = time(NULL) + pingtimeout;
+	timeout_event.handler = timeout_handler;
+	timeout_event.data = &timeout_event;
+
+	event_add(&timeout_event);
+
+#ifdef SIGHUP
+	signal(SIGHUP, sighup_handler);
+#endif
+#ifdef SIGTERM
+	signal(SIGTERM, sigterm_handler);
+#endif
+#ifdef SIGQUIT
+	signal(SIGQUIT, sigterm_handler);
+#endif
+
+	while(true) {
+		mutex_unlock(&mutex);
+		usleep(1000000);
+		mutex_lock(&mutex);
+
+		struct event *event;
+		while((event = get_expired_event())) {
+			event->handler(event->data);
 		}
 	}
 
+	event_del(&timeout_event);
+
 	return 0;
 }