+/*
+ 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 "devtools.h"
+/* 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;
+ }
+
+ pthread_mutex_lock(&(mesh->mesh_mutex));
+
+ 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 = realloc(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->options = e->options;
+ p->weight = e->weight;
+
+ n++;
+ p++;
+ }
+
+ // shrink result to the actual amount of memory used
+ result = realloc(result, n * sizeof(*result));
+ *nmemb = n;
+ } else {
+ *nmemb = 0;
+ meshlink_errno = MESHLINK_ENOMEM;
+ }
+
+ pthread_mutex_unlock(&(mesh->mesh_mutex));
+
+ return result;
+}
+
static bool fstrwrite(const char *str, FILE *stream) {
size_t len = strlen(str);
- if(fwrite((void *)str, 1, len, stream) != len)
+ if(fwrite((void *)str, 1, len, stream) != len) {
return false;
+ }
return true;
}
static const char *__itoa(int value) {
static char buffer[sizeof(int) * 8 + 1]; // not thread safe
- if(snprintf(buffer, sizeof(buffer), "%d", value) == -1)
+ if(snprintf(buffer, sizeof(buffer), "%d", value) == -1) {
return "";
+ }
return buffer;
}
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;
+ }
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))
+ if(!fstrwrite("\t\t\t\"options\": ", stream) || !fstrwrite(__itoa(((node_t *)nodes[i])->options), stream) || !fstrwrite(",\n", stream)) {
goto fail;
+ }
- 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(__itoa(((node_t *)nodes[i])->devclass), 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);
free(address);
- if(!fstrwrite("\t\t\t\"options\": ", stream) || !fstrwrite(__itoa(edges[i]->options), stream) || !fstrwrite(",\n", stream))
+ if(!fstrwrite("\t\t\t\"options\": ", stream) || !fstrwrite(__itoa(edges[i].options), stream) || !fstrwrite(",\n", stream)) {
goto fail;
+ }
- if(!fstrwrite("\t\t\t\"weight\": ", stream) || !fstrwrite(__itoa(edges[i]->weight), stream) || !fstrwrite("\n", stream))
+ if(!fstrwrite("\t\t\t\"weight\": ", stream) || !fstrwrite(__itoa(edges[i].weight), 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;
result = false;
done:
+ free(nodes);
+ free(edges);
- if(nodes)
- free(nodes);
+ pthread_mutex_unlock(&(mesh->mesh_mutex));
- for(size_t i = 0; edges && i < edge_count; ++i)
- free(edges[i]);
+ return result;
+}
- if(nodes)
- free(edges);
+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;
+ }
- pthread_mutex_unlock(&(mesh->mesh_mutex));
+ node_t *internal = (node_t *)node;
+
+ pthread_mutex_lock(&mesh->mesh_mutex);
+
+ status->options = internal->options;
+ 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;
+ }
- return result;
+ pthread_mutex_unlock(&mesh->mesh_mutex);
}