X-Git-Url: http://git.meshlink.io/?p=meshlink;a=blobdiff_plain;f=src%2Fprotocol_auth.c;h=c17d9e8214b9b196c30b4725c2a456e204142a1e;hp=b7f6acdda03315f2ec72aa899983f56f473022dd;hb=8c8b506da9ce578fe6e6e406a91bbd21af13f7a3;hpb=ad87362afd13c4f14ce9a4b2cf0586a753b5a422

diff --git a/src/protocol_auth.c b/src/protocol_auth.c
index b7f6acdd..c17d9e82 100644
--- a/src/protocol_auth.c
+++ b/src/protocol_auth.c
@@ -21,6 +21,7 @@
 
 #include "conf.h"
 #include "connection.h"
+#include "devtools.h"
 #include "ecdsa.h"
 #include "edge.h"
 #include "graph.h"
@@ -30,6 +31,7 @@
 #include "net.h"
 #include "netutl.h"
 #include "node.h"
+#include "packmsg.h"
 #include "prf.h"
 #include "protocol.h"
 #include "sptps.h"
@@ -41,178 +43,40 @@
 
 extern bool node_write_devclass(meshlink_handle_t *mesh, node_t *n);
 
-static bool send_proxyrequest(meshlink_handle_t *mesh, connection_t *c) {
-	switch(mesh->proxytype) {
-	case PROXY_HTTP: {
-		char *host;
-		char *port;
-
-		sockaddr2str(&c->address, &host, &port);
-		send_request(mesh, c, NULL, "CONNECT %s:%s HTTP/1.1\r\n\r", host, port);
-		free(host);
-		free(port);
-		return true;
-	}
-
-	case PROXY_SOCKS4: {
-		if(c->address.sa.sa_family != AF_INET) {
-			logger(mesh, MESHLINK_ERROR, "Cannot connect to an IPv6 host through a SOCKS 4 proxy!");
-			return false;
-		}
-
-		char s4req[9 + (mesh->proxyuser ? strlen(mesh->proxyuser) : 0)];
-		s4req[0] = 4;
-		s4req[1] = 1;
-		memcpy(s4req + 2, &c->address.in.sin_port, 2);
-		memcpy(s4req + 4, &c->address.in.sin_addr, 4);
-
-		if(mesh->proxyuser) {
-			memcpy(s4req + 8, mesh->proxyuser, strlen(mesh->proxyuser));
-		}
-
-		s4req[sizeof(s4req) - 1] = 0;
-		c->tcplen = 8;
-		return send_meta(mesh, c, s4req, sizeof(s4req));
-	}
-
-	case PROXY_SOCKS5: {
-		int len = 3 + 6 + (c->address.sa.sa_family == AF_INET ? 4 : 16);
-		c->tcplen = 2;
-
-		if(mesh->proxypass) {
-			len += 3 + strlen(mesh->proxyuser) + strlen(mesh->proxypass);
-		}
-
-		char s5req[len];
-		int i = 0;
-		s5req[i++] = 5;
-		s5req[i++] = 1;
-
-		if(mesh->proxypass) {
-			s5req[i++] = 2;
-			s5req[i++] = 1;
-			s5req[i++] = strlen(mesh->proxyuser);
-			memcpy(s5req + i, mesh->proxyuser, strlen(mesh->proxyuser));
-			i += strlen(mesh->proxyuser);
-			s5req[i++] = strlen(mesh->proxypass);
-			memcpy(s5req + i, mesh->proxypass, strlen(mesh->proxypass));
-			i += strlen(mesh->proxypass);
-			c->tcplen += 2;
-		} else {
-			s5req[i++] = 0;
-		}
-
-		s5req[i++] = 5;
-		s5req[i++] = 1;
-		s5req[i++] = 0;
-
-		if(c->address.sa.sa_family == AF_INET) {
-			s5req[i++] = 1;
-			memcpy(s5req + i, &c->address.in.sin_addr, 4);
-			i += 4;
-			memcpy(s5req + i, &c->address.in.sin_port, 2);
-			i += 2;
-			c->tcplen += 10;
-		} else if(c->address.sa.sa_family == AF_INET6) {
-			s5req[i++] = 3;
-			memcpy(s5req + i, &c->address.in6.sin6_addr, 16);
-			i += 16;
-			memcpy(s5req + i, &c->address.in6.sin6_port, 2);
-			i += 2;
-			c->tcplen += 22;
-		} else {
-			logger(mesh, MESHLINK_ERROR, "Address family %hx not supported for SOCKS 5 proxies!", c->address.sa.sa_family);
-			return false;
-		}
-
-		if(i > len) {
-			abort();
-		}
-
-		return send_meta(mesh, c, s5req, sizeof(s5req));
-	}
-
-	case PROXY_SOCKS4A:
-		logger(mesh, MESHLINK_ERROR, "Proxy type not implemented yet");
-		return false;
-
-	default:
-		logger(mesh, MESHLINK_ERROR, "Unknown proxy type");
-		return false;
-	}
-}
-
 bool send_id(meshlink_handle_t *mesh, connection_t *c) {
-
-	int minor = mesh->self->connection->protocol_minor;
-
-	if(mesh->proxytype && c->outgoing)
-		if(!send_proxyrequest(mesh, c)) {
-			return false;
-		}
-
-	return send_request(mesh, c, NULL, "%d %s %d.%d %s", ID, mesh->self->connection->name, mesh->self->connection->protocol_major, minor, mesh->appname);
+	return send_request(mesh, c, NULL, "%d %s %d.%d %s", ID, mesh->self->name, PROT_MAJOR, PROT_MINOR, mesh->appname);
 }
 
-static bool finalize_invitation(meshlink_handle_t *mesh, connection_t *c, const void *data, uint16_t len) {
-	(void)len;
-
-	if(strchr(data, '\n')) {
-		logger(mesh, MESHLINK_ERROR, "Received invalid key from invited node %s!\n", c->name);
-		return false;
-	}
-
-	// Create a new host config file
-	char filename[PATH_MAX];
-	snprintf(filename, sizeof(filename), "%s" SLASH "hosts" SLASH "%s", mesh->confbase, c->name);
-
-	if(!access(filename, F_OK)) {
-		logger(mesh, MESHLINK_ERROR, "Host config file for %s already exists!\n", c->name);
-		return false;
-	}
+static bool commit_invitation(meshlink_handle_t *mesh, connection_t *c, const void *data) {
+	// Create a new node
+	node_t *n = new_node();
+	n->name = xstrdup(c->name);
+	n->devclass = DEV_CLASS_UNKNOWN;
+	n->ecdsa = ecdsa_set_public_key(data);
+	n->submesh = c->submesh;
 
-	FILE *f = fopen(filename, "w");
+	// Remember its current address
+	node_add_recent_address(mesh, n, &c->address);
 
-	if(!f) {
-		logger(mesh, MESHLINK_ERROR, "Error trying to create %s: %s\n", filename, strerror(errno));
+	if(!node_write_config(mesh, n) || !config_sync(mesh, "current")) {
+		logger(mesh, MESHLINK_ERROR, "Error writing configuration file for invited node %s!\n", c->name);
+		free_node(n);
 		return false;
-	}
-
-	fprintf(f, "ECDSAPublicKey = %s\n", (const char *)data);
 
-	if(c->submesh) {
-		fprintf(f, "SubMesh = %s\n", c->submesh->name);
 	}
 
-	fclose(f);
+	node_add(mesh, n);
 
 	logger(mesh, MESHLINK_INFO, "Key successfully received from %s", c->name);
 
 	//TODO: callback to application to inform of an accepted invitation
 
-	sptps_send_record(&c->sptps, 2, data, 0);
-
-	load_all_nodes(mesh);
+	sptps_send_record(&c->sptps, 1, "", 0);
 
 	return true;
 }
 
-static bool receive_invitation_sptps(void *handle, uint8_t type, const void *data, uint16_t len) {
-	connection_t *c = handle;
-	meshlink_handle_t *mesh = c->mesh;
-
-	if(type == 128) {
-		return true;
-	}
-
-	if(type == 1 && c->status.invitation_used) {
-		return finalize_invitation(mesh, c, data, len);
-	}
-
-	if(type != 0 || len != 18 || c->status.invitation_used) {
-		return false;
-	}
-
+static bool process_invitation(meshlink_handle_t *mesh, connection_t *c, const void *data) {
 	// Recover the filename from the cookie and the key
 	char *fingerprint = ecdsa_get_base64_public_key(mesh->invitation_key);
 	char hash[64];
@@ -224,118 +88,53 @@ static bool receive_invitation_sptps(void *handle, uint8_t type, const void *dat
 	b64encode_urlsafe(hash, cookie, 18);
 	free(fingerprint);
 
-	char filename[PATH_MAX], usedname[PATH_MAX];
-	snprintf(filename, sizeof(filename), "%s" SLASH "invitations" SLASH "%s", mesh->confbase, cookie);
-	snprintf(usedname, sizeof(usedname), "%s" SLASH "invitations" SLASH "%s.used", mesh->confbase, cookie);
-
-	// Atomically rename the invitation file
-	if(rename(filename, usedname)) {
-		if(errno == ENOENT) {
-			logger(mesh, MESHLINK_ERROR, "Peer %s tried to use non-existing invitation %s\n", c->name, cookie);
-		} else {
-			logger(mesh, MESHLINK_ERROR, "Error trying to rename invitation %s\n", cookie);
-		}
-
-		return false;
-	}
-
-	// Open the renamed file
-	FILE *f = fopen(usedname, "r");
+	config_t config;
 
-	if(!f) {
-		logger(mesh, MESHLINK_ERROR, "Error trying to open invitation %s\n", cookie);
-		unlink(usedname);
-		return false;
-	}
-
-	// Check the timestamp
-	struct stat st;
-
-	if(fstat(fileno(f), &st)) {
-		logger(mesh, MESHLINK_ERROR, "Could not stat invitation file %s\n", usedname);
-		fclose(f);
-		unlink(usedname);
-		return false;
-	}
-
-	if(time(NULL) > (time_t)(st.st_mtime + mesh->invitation_timeout)) {
-		logger(mesh, MESHLINK_ERROR, "Peer %s tried to use an outdated invitation file %s\n", c->name, usedname);
-		fclose(f);
-		unlink(usedname);
+	if(!invitation_read(mesh, "current", cookie, &config, mesh->config_key)) {
+		logger(mesh, MESHLINK_ERROR, "Error while trying to read invitation file\n");
 		return false;
 	}
 
 	// Read the new node's Name from the file
-	char buf[1024];
-	fgets(buf, sizeof(buf), f);
-
-	if(*buf) {
-		buf[strlen(buf) - 1] = 0;
-	}
-
-	len = strcspn(buf, " \t=");
-	char *name = buf + len;
-	name += strspn(name, " \t");
-
-	if(*name == '=') {
-		name++;
-		name += strspn(name, " \t");
-	}
-
-	buf[len] = 0;
-
-	if(!*buf || !*name || strcasecmp(buf, "Name") || !check_id(name)) {
-		logger(mesh, MESHLINK_ERROR, "Invalid invitation file %s\n", cookie);
-		fclose(f);
-		return false;
-	}
-
+	packmsg_input_t in = {config.buf, config.len};
+	packmsg_get_uint32(&in); // skip version
 	free(c->name);
-	c->name = xstrdup(name);
+	c->name = packmsg_get_str_dup(&in);
 
 	// Check if the file contains Sub-Mesh information
-	buf[0] = 0;
-	fgets(buf, sizeof(buf), f);
-
-	if(*buf) {
-		buf[strlen(buf) - 1] = 0;
-	}
-
-	if(!strncmp(buf, "SubMesh", 7)) {
-		len = strcspn(buf, " \t=");
-		char *submesh_name = buf + len;
-		submesh_name += strspn(submesh_name, " \t");
-
-		if(*submesh_name == '=') {
-			submesh_name++;
-			submesh_name += strspn(submesh_name, " \t");
-		}
+	char *submesh_name = packmsg_get_str_dup(&in);
 
+	if(!strcmp(submesh_name, CORE_MESH)) {
+		free(submesh_name);
+		c->submesh = NULL;
+	} else {
 		if(!check_id(submesh_name)) {
 			logger(mesh, MESHLINK_ERROR, "Invalid invitation file %s\n", cookie);
-			fclose(f);
+			free(submesh_name);
 			return false;
 		}
 
-		c->submesh = NULL;
 		c->submesh = lookup_or_create_submesh(mesh, submesh_name);
+		free(submesh_name);
 
 		if(!c->submesh) {
+			logger(mesh, MESHLINK_ERROR, "Unknown submesh in invitation file %s\n", cookie);
 			return false;
 		}
 	}
 
-	// Send the node the contents of the invitation file
-	rewind(f);
-	size_t result;
+	if(mesh->inviter_commits_first && !commit_invitation(mesh, c, (const char *)data + 18)) {
+		return false;
+	}
 
-	while((result = fread(buf, 1, sizeof(buf), f))) {
-		sptps_send_record(&c->sptps, 0, buf, result);
+	if(mesh->inviter_commits_first) {
+		devtool_set_inviter_commits_first(true);
 	}
 
-	sptps_send_record(&c->sptps, 1, buf, 0);
-	fclose(f);
-	unlink(usedname);
+	// Send the node the contents of the invitation file
+	sptps_send_record(&c->sptps, 0, config.buf, config.len);
+
+	config_free(&config);
 
 	c->status.invitation_used = true;
 
@@ -343,7 +142,34 @@ static bool receive_invitation_sptps(void *handle, uint8_t type, const void *dat
 	return true;
 }
 
+static bool receive_invitation_sptps(void *handle, uint8_t type, const void *data, uint16_t len) {
+	connection_t *c = handle;
+	meshlink_handle_t *mesh = c->mesh;
+
+	if(type == SPTPS_HANDSHAKE) {
+		// The peer should send its cookie first.
+		return true;
+	}
+
+	if(mesh->inviter_commits_first) {
+		if(type == 2 && len == 18 + 32 && !c->status.invitation_used) {
+			return process_invitation(mesh, c, data);
+		}
+	} else {
+		if(type == 0 && len == 18 && !c->status.invitation_used) {
+			return process_invitation(mesh, c, data);
+		} else if(type == 1 && len == 32 && c->status.invitation_used) {
+			return commit_invitation(mesh, c, data);
+		}
+	}
+
+	return false;
+}
+
 bool id_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
+	assert(request);
+	assert(*request);
+
 	char name[MAX_STRING_SIZE];
 
 	if(sscanf(request, "%*d " MAX_STRING " %d.%d", name, &c->protocol_major, &c->protocol_minor) < 2) {
@@ -409,37 +235,30 @@ bool id_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
 
 	/* Check if version matches */
 
-	if(c->protocol_major != mesh->self->connection->protocol_major) {
+	if(c->protocol_major != PROT_MAJOR) {
 		logger(mesh, MESHLINK_ERROR, "Peer %s uses incompatible version %d.%d",
 		       c->name, c->protocol_major, c->protocol_minor);
 		return false;
 	}
 
-	if(!c->config_tree) {
-		init_configuration(&c->config_tree);
+	/* Check if we know this node */
 
-		if(!read_host_config(mesh, c->config_tree, c->name)) {
-			logger(mesh, MESHLINK_ERROR, "Peer %s has unknown identity", c->name);
-			return false;
-		}
-	}
-
-	bool blacklisted = false;
-	get_config_bool(lookup_config(c->config_tree, "blacklisted"), &blacklisted);
+	node_t *n = lookup_node(mesh, c->name);
 
-	if(blacklisted) {
-		logger(mesh, MESHLINK_INFO, "Peer %s is blacklisted", c->name);
+	if(!n) {
+		logger(mesh, MESHLINK_ERROR, "Peer %s has unknown identity", c->name);
 		return false;
 	}
 
-	read_ecdsa_public_key(mesh, c);
+	if(n->status.blacklisted) {
+		logger(mesh, MESHLINK_WARNING, "Peer %s is blacklisted", c->name);
+		return false;
+	}
 
-	if(!ecdsa_active(c->ecdsa)) {
+	if(!node_read_public_key(mesh, n)) {
 		logger(mesh, MESHLINK_ERROR, "No key known for peer %s", c->name);
 
-		node_t *n = lookup_node(mesh, c->name);
-
-		if(n && n->status.reachable && !n->status.waitingforkey) {
+		if(n->status.reachable && !n->status.waitingforkey) {
 			logger(mesh, MESHLINK_INFO, "Requesting key from peer %s", c->name);
 			send_req_key(mesh, n);
 		}
@@ -464,18 +283,15 @@ bool id_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
 		snprintf(label, sizeof(label), "%s %s %s", meshlink_tcp_label, c->name, mesh->self->name);
 	}
 
-	return sptps_start(&c->sptps, c, c->outgoing, false, mesh->self->connection->ecdsa, c->ecdsa, label, sizeof(label) - 1, send_meta_sptps, receive_meta_sptps);
+	char buf1[1024], buf2[1024];
+	bin2hex((uint8_t *)mesh->private_key + 64, buf1, 32);
+	bin2hex((uint8_t *)n->ecdsa + 64, buf2, 32);
+	logger(mesh, MESHLINK_DEBUG, "Connection to %s mykey %s hiskey %s", c->name, buf1, buf2);
+	return sptps_start(&c->sptps, c, c->outgoing, false, mesh->private_key, n->ecdsa, label, sizeof(label) - 1, send_meta_sptps, receive_meta_sptps);
 }
 
 bool send_ack(meshlink_handle_t *mesh, connection_t *c) {
-
-	/* Check some options */
-
-	if(mesh->self->options & OPTION_PMTU_DISCOVERY) {
-		c->options |= OPTION_PMTU_DISCOVERY;
-	}
-
-	return send_request(mesh, c, NULL, "%d %s %d %x", ACK, mesh->myport, mesh->devclass, (c->options & 0xffffff) | (PROT_MINOR << 24));
+	return send_request(mesh, c, NULL, "%d %s %d %x", ACK, mesh->myport, mesh->devclass, OPTION_PMTU_DISCOVERY | (PROT_MINOR << 24));
 }
 
 static void send_everything(meshlink_handle_t *mesh, connection_t *c) {
@@ -489,8 +305,10 @@ static void send_everything(meshlink_handle_t *mesh, connection_t *c) {
 }
 
 bool ack_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
+	assert(request);
+	assert(*request);
+
 	char hisport[MAX_STRING_SIZE];
-	char *hisaddress;
 	int devclass;
 	uint32_t options;
 	node_t *n;
@@ -500,7 +318,7 @@ bool ack_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
 		return false;
 	}
 
-	if(devclass < 0 || devclass > _DEV_CLASS_MAX) {
+	if(devclass < 0 || devclass >= DEV_CLASS_COUNT) {
 		logger(mesh, MESHLINK_ERROR, "Got bad %s from %s: %s", "ACK", c->name, "devclass invalid");
 		return false;
 	}
@@ -528,30 +346,26 @@ bool ack_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
 				n->connection->outgoing = NULL;
 			}
 
+			/* Remove the edge before terminating the connection, to prevent a graph update. */
+			edge_del(mesh, n->connection->edge);
+			n->connection->edge = NULL;
+
 			terminate_connection(mesh, n->connection, false);
-			/* Run graph algorithm to keep things in sync */
-			graph(mesh);
 		}
 	}
 
 	n->devclass = devclass;
-	node_write_devclass(mesh, n);
+	n->status.dirty = true;
 
-	n->last_successfull_connection = time(NULL);
+	n->last_successfull_connection = mesh->loop.now.tv_sec;
 
 	n->connection = c;
 	c->node = n;
 
-	if(!(c->options & options & OPTION_PMTU_DISCOVERY)) {
-		c->options &= ~OPTION_PMTU_DISCOVERY;
-		options &= ~OPTION_PMTU_DISCOVERY;
-	}
-
-	c->options |= options;
-
 	/* Activate this connection */
 
 	c->allow_request = ALL;
+	c->last_key_renewal = mesh->loop.now.tv_sec;
 	c->status.active = true;
 
 	logger(mesh, MESHLINK_INFO, "Connection with %s activated", c->name);
@@ -562,17 +376,14 @@ bool ack_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
 
 	/* Create an edge_t for this connection */
 
-	assert(devclass >= 0 && devclass <= _DEV_CLASS_MAX);
+	assert(devclass >= 0 && devclass < DEV_CLASS_COUNT);
 
 	c->edge = new_edge();
 	c->edge->from = mesh->self;
 	c->edge->to = n;
-	sockaddr2str(&c->address, &hisaddress, NULL);
-	c->edge->address = str2sockaddr(hisaddress, hisport);
-	free(hisaddress);
-	c->edge->weight = dev_class_traits[devclass].edge_weight;
+	sockaddrcpy_setport(&c->edge->address, &c->address, atoi(hisport));
+	c->edge->weight = mesh->dev_class_traits[devclass].edge_weight;
 	c->edge->connection = c;
-	c->edge->options = c->options;
 
 	edge_add(mesh, c->edge);