X-Git-Url: http://git.meshlink.io/?p=meshlink;a=blobdiff_plain;f=src%2Fdevtools.c;h=5a49e3a860296a66ad049bd7d238946424ef1b2d;hp=1cb2d5e88f6e4890d0d3d96761f04fa622cb2e78;hb=963c5055505f2fc117cd5efa06eaa02c9b2bf85d;hpb=196806f90f40fcc0cf727abd4bed6bc5aefff5ff

diff --git a/src/devtools.c b/src/devtools.c
index 1cb2d5e8..5a49e3a8 100644
--- a/src/devtools.c
+++ b/src/devtools.c
@@ -1,106 +1,207 @@
+/*
+    devtools.c -- Debugging and quality control functions.
+    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
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    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.,
+    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+*/
 
 #include "system.h"
+#include <assert.h>
 
 #include "logger.h"
 #include "meshlink_internal.h"
 #include "node.h"
+#include "submesh.h"
 #include "splay_tree.h"
 #include "netutl.h"
 #include "xalloc.h"
 
 #include "devtools.h"
 
-static int node_compare(const void *a, const void *b) {
-	if(a < b)
-		return -1;
+static void trybind_nop_probe(void) {
+	return;
+}
 
-	if(a > b)
-		return 1;
+static void keyrotate_nop_probe(int stage) {
+	(void)stage;
+	return;
+}
 
-	return 0;
+static void inviter_commits_first_nop_probe(bool stage) {
+	(void)stage;
+	return;
 }
 
-static bool fstrwrite(const char* str, FILE* stream) {
-	size_t len = strlen(str);
+void (*devtool_trybind_probe)(void) = trybind_nop_probe;
+void (*devtool_keyrotate_probe)(int stage) = keyrotate_nop_probe;
+void (*devtool_set_inviter_commits_first)(bool inviter_commited_first) = inviter_commits_first_nop_probe;
+
+/* Return an array of edges in the current network graph.
+ * Data captures the current state and will not be updated.
+ * Caller must deallocate data when done.
+ */
+devtool_edge_t *devtool_get_all_edges(meshlink_handle_t *mesh, devtool_edge_t *edges, size_t *nmemb) {
+	if(!mesh || !nmemb || (*nmemb && !edges)) {
+		meshlink_errno = MESHLINK_EINVAL;
+		return NULL;
+	}
 
-	if(fwrite((void*)str, 1, len, stream) != len)
-		return false;
+	pthread_mutex_lock(&mesh->mutex);
 
-	return true;
+	devtool_edge_t *result = NULL;
+	unsigned int result_size = 0;
+
+	result_size = mesh->edges->count / 2;
+
+	// if result is smaller than edges, we have to dealloc all the excess devtool_edge_t
+	if((size_t)result_size > *nmemb) {
+		result = xrealloc(edges, result_size * sizeof(*result));
+	} else {
+		result = edges;
+	}
+
+	if(result) {
+		devtool_edge_t *p = result;
+		unsigned int n = 0;
+
+		for splay_each(edge_t, e, mesh->edges) {
+			// skip edges that do not represent a two-directional connection
+			if(!e->reverse || e->reverse->to != e->from) {
+				continue;
+			}
+
+			// don't count edges twice
+			if(e->to < e->from) {
+				continue;
+			}
+
+			assert(n < result_size);
+
+			p->from = (meshlink_node_t *)e->from;
+			p->to = (meshlink_node_t *)e->to;
+			p->address = e->address.storage;
+			p->weight = e->weight;
+
+			n++;
+			p++;
+		}
+
+		// shrink result to the actual amount of memory used
+		result = xrealloc(result, n * sizeof(*result));
+		*nmemb = n;
+	} else {
+		*nmemb = 0;
+		meshlink_errno = MESHLINK_ENOMEM;
+	}
+
+	pthread_mutex_unlock(&mesh->mutex);
+
+	return result;
 }
 
-static const char* __itoa(int value) {
-	static char buffer[sizeof(int) * 8 + 1];        // not thread safe
+static bool fstrwrite(const char *str, FILE *stream) {
+	assert(stream);
+
+	size_t len = strlen(str);
 
-	if(snprintf(buffer, sizeof(buffer), "%d", value) == -1)
-		return "";
+	if(fwrite((void *)str, 1, len, stream) != len) {
+		return false;
+	}
 
-	return buffer;
+	return true;
 }
 
-bool devtool_export_json_all_edges_state(meshlink_handle_t *mesh, FILE* stream) {
+bool devtool_export_json_all_edges_state(meshlink_handle_t *mesh, FILE *stream) {
+	assert(stream);
+
 	bool result = true;
 
-	pthread_mutex_lock(&(mesh->mesh_mutex));
+	pthread_mutex_lock(&mesh->mutex);
 
 	// export edges and nodes
 	size_t node_count = 0;
 	size_t edge_count = 0;
 
 	meshlink_node_t **nodes = meshlink_get_all_nodes(mesh, NULL, &node_count);
-	meshlink_edge_t **edges = meshlink_get_all_edges_state(mesh, NULL, &edge_count);
+	devtool_edge_t *edges = devtool_get_all_edges(mesh, NULL, &edge_count);
 
-	if((!nodes && node_count != 0) || (!edges && edge_count != 0))
+	if((!nodes && node_count != 0) || (!edges && edge_count != 0)) {
 		goto fail;
+	}
 
 	// export begin
-	if(!fstrwrite("{\n", stream))
+	if(!fstrwrite("{\n", stream)) {
 		goto fail;
+	}
 
 	// export nodes
-	if(!fstrwrite("\t\"nodes\": {\n", stream))
+	if(!fstrwrite("\t\"nodes\": {\n", stream)) {
 		goto fail;
+	}
+
+	char buf[16];
 
 	for(size_t i = 0; i < node_count; ++i) {
-		if(!fstrwrite("\t\t\"", stream) || !fstrwrite(((node_t*)nodes[i])->name, stream) || !fstrwrite("\": {\n", stream))
+		if(!fstrwrite("\t\t\"", stream) || !fstrwrite(((node_t *)nodes[i])->name, stream) || !fstrwrite("\": {\n", stream)) {
 			goto fail;
+		}
 
-		if(!fstrwrite("\t\t\t\"name\": \"", stream) || !fstrwrite(((node_t*)nodes[i])->name, stream) || !fstrwrite("\",\n", stream))
+		if(!fstrwrite("\t\t\t\"name\": \"", stream) || !fstrwrite(((node_t *)nodes[i])->name, stream) || !fstrwrite("\",\n", stream)) {
 			goto fail;
+		}
 
-		if(!fstrwrite("\t\t\t\"options\": ", stream) || !fstrwrite(__itoa(((node_t*)nodes[i])->options), stream) || !fstrwrite(",\n", stream))
-			goto fail;
+		snprintf(buf, sizeof(buf), "%d", ((node_t *)nodes[i])->devclass);
 
-		if(!fstrwrite("\t\t\t\"devclass\": ", stream) || !fstrwrite(__itoa(((node_t*)nodes[i])->devclass), stream) || !fstrwrite("\n", stream))
+		if(!fstrwrite("\t\t\t\"devclass\": ", stream) || !fstrwrite(buf, stream) || !fstrwrite("\n", stream)) {
 			goto fail;
+		}
 
-		if(!fstrwrite((i+1) != node_count ? "\t\t},\n" : "\t\t}\n", stream))
+		if(!fstrwrite((i + 1) != node_count ? "\t\t},\n" : "\t\t}\n", stream)) {
 			goto fail;
+		}
 	}
 
-	if(!fstrwrite("\t},\n", stream))
+	if(!fstrwrite("\t},\n", stream)) {
 		goto fail;
+	}
 
 	// export edges
 
-	if(!fstrwrite("\t\"edges\": {\n", stream))
+	if(!fstrwrite("\t\"edges\": {\n", stream)) {
 		goto fail;
+	}
 
 	for(size_t i = 0; i < edge_count; ++i) {
-		if(!fstrwrite("\t\t\"", stream) || !fstrwrite(edges[i]->from->name, stream) || !fstrwrite("_to_", stream) || !fstrwrite(edges[i]->to->name, stream) || !fstrwrite("\": {\n", stream))
+		if(!fstrwrite("\t\t\"", stream) || !fstrwrite(edges[i].from->name, stream) || !fstrwrite("_to_", stream) || !fstrwrite(edges[i].to->name, stream) || !fstrwrite("\": {\n", stream)) {
 			goto fail;
+		}
 
-		if(!fstrwrite("\t\t\t\"from\": \"", stream) || !fstrwrite(edges[i]->from->name, stream) || !fstrwrite("\",\n", stream))
+		if(!fstrwrite("\t\t\t\"from\": \"", stream) || !fstrwrite(edges[i].from->name, stream) || !fstrwrite("\",\n", stream)) {
 			goto fail;
+		}
 
-		if(!fstrwrite("\t\t\t\"to\": \"", stream) || !fstrwrite(edges[i]->to->name, stream) || !fstrwrite("\",\n", stream))
+		if(!fstrwrite("\t\t\t\"to\": \"", stream) || !fstrwrite(edges[i].to->name, stream) || !fstrwrite("\",\n", stream)) {
 			goto fail;
+		}
 
 		char *host = NULL, *port = NULL, *address = NULL;
-		sockaddr2str((const sockaddr_t *)&(edges[i]->address), &host, &port);
+		sockaddr2str((const sockaddr_t *)&edges[i].address, &host, &port);
 
-		if(host && port)
+		if(host && port) {
 			xasprintf(&address, "{ \"host\": \"%s\", \"port\": %s }", host, port);
+		}
 
 		free(host);
 		free(port);
@@ -112,23 +213,26 @@ bool devtool_export_json_all_edges_state(meshlink_handle_t *mesh, FILE* stream)
 
 		free(address);
 
-		if(!fstrwrite("\t\t\t\"options\": ", stream) || !fstrwrite(__itoa(edges[i]->options), stream) || !fstrwrite(",\n", stream))
-			goto fail;
+		snprintf(buf, sizeof(buf), "%d", edges[i].weight);
 
-		if(!fstrwrite("\t\t\t\"weight\": ", stream) || !fstrwrite(__itoa(edges[i]->weight), stream) || !fstrwrite("\n", stream))
+		if(!fstrwrite("\t\t\t\"weight\": ", stream) || !fstrwrite(buf, stream) || !fstrwrite("\n", stream)) {
 			goto fail;
+		}
 
-		if(!fstrwrite((i+1) != edge_count ? "\t\t},\n" : "\t\t}\n", stream))
+		if(!fstrwrite((i + 1) != edge_count ? "\t\t},\n" : "\t\t}\n", stream)) {
 			goto fail;
+		}
 	}
 
-	if(!fstrwrite("\t}\n", stream))
+	if(!fstrwrite("\t}\n", stream)) {
 		goto fail;
+	}
 
 	// DONE!
 
-	if(!fstrwrite("}", stream))
+	if(!fstrwrite("}", stream)) {
 		goto fail;
+	}
 
 	goto done;
 
@@ -136,17 +240,99 @@ fail:
 	result = false;
 
 done:
+	free(nodes);
+	free(edges);
+
+	pthread_mutex_unlock(&mesh->mutex);
+
+	return result;
+}
 
-	if(nodes)
-		free(nodes);
+void devtool_get_node_status(meshlink_handle_t *mesh, meshlink_node_t *node, devtool_node_status_t *status) {
+	if(!mesh || !node || !status) {
+		meshlink_errno = MESHLINK_EINVAL;
+		return;
+	}
+
+	node_t *internal = (node_t *)node;
+
+	pthread_mutex_lock(&mesh->mutex);
+
+	memcpy(&status->status, &internal->status, sizeof status->status);
+	memcpy(&status->address, &internal->address, sizeof status->address);
+	status->mtu = internal->mtu;
+	status->minmtu = internal->minmtu;
+	status->maxmtu = internal->maxmtu;
+	status->mtuprobes = internal->mtuprobes;
+	status->in_packets = internal->in_packets;
+	status->in_bytes = internal->in_bytes;
+	status->out_packets = internal->out_packets;
+	status->out_bytes = internal->out_bytes;
+
+	// Derive UDP connection status
+	if(internal == mesh->self) {
+		status->udp_status = DEVTOOL_UDP_WORKING;
+	} else if(!internal->status.reachable) {
+		status->udp_status = DEVTOOL_UDP_IMPOSSIBLE;
+	} else if(!internal->status.validkey) {
+		status->udp_status = DEVTOOL_UDP_UNKNOWN;
+	} else if(internal->status.udp_confirmed) {
+		status->udp_status = DEVTOOL_UDP_WORKING;
+	} else if(internal->mtuprobes > 30) {
+		status->udp_status = DEVTOOL_UDP_FAILED;
+	} else if(internal->mtuprobes > 0) {
+		status->udp_status = DEVTOOL_UDP_TRYING;
+	} else {
+		status->udp_status = DEVTOOL_UDP_UNKNOWN;
+	}
+
+	pthread_mutex_unlock(&mesh->mutex);
+}
+
+meshlink_submesh_t **devtool_get_all_submeshes(meshlink_handle_t *mesh, meshlink_submesh_t **submeshes, size_t *nmemb) {
+	if(!mesh || !nmemb || (*nmemb && !submeshes)) {
+		meshlink_errno = MESHLINK_EINVAL;
+		return NULL;
+	}
+
+	meshlink_submesh_t **result;
+
+	//lock mesh->nodes
+	pthread_mutex_lock(&mesh->mutex);
 
-	for(size_t i = 0; edges && i < edge_count; ++i)
-		free(edges[i]);
+	*nmemb = mesh->submeshes->count;
+	result = realloc(submeshes, *nmemb * sizeof(*submeshes));
 
-	if(nodes)
-		free(edges);
+	if(result) {
+		meshlink_submesh_t **p = result;
 
-	pthread_mutex_unlock(&(mesh->mesh_mutex));
+		for list_each(submesh_t, s, mesh->submeshes) {
+			*p++ = (meshlink_submesh_t *)s;
+		}
+	} else {
+		*nmemb = 0;
+		free(submeshes);
+		meshlink_errno = MESHLINK_ENOMEM;
+	}
+
+	pthread_mutex_unlock(&mesh->mutex);
 
 	return result;
 }
+
+meshlink_handle_t *devtool_open_in_netns(const char *confbase, const char *name, const char *appname, dev_class_t devclass, int netns) {
+	meshlink_open_params_t *params = meshlink_open_params_init(confbase, name, appname, devclass);
+	params->netns = dup(netns);
+	meshlink_handle_t *handle;
+
+	if(params->netns == -1) {
+		handle = NULL;
+		meshlink_errno = MESHLINK_EINVAL;
+	} else {
+		handle = meshlink_open_ex(params);
+	}
+
+	meshlink_open_params_free(params);
+
+	return handle;
+}