2 devtools.c -- Debugging and quality control functions.
3 Copyright (C) 2014, 2017 Guus Sliepen <guus@meshlink.io>
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 2 of the License, or
8 (at your option) any later version.
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
15 You should have received a copy of the GNU General Public License along
16 with this program; if not, write to the Free Software Foundation, Inc.,
17 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
24 #include "meshlink_internal.h"
26 #include "splay_tree.h"
32 /* Return an array of edges in the current network graph.
33 * Data captures the current state and will not be updated.
34 * Caller must deallocate data when done.
36 devtool_edge_t *devtool_get_all_edges(meshlink_handle_t *mesh, devtool_edge_t *edges, size_t *nmemb) {
37 if(!mesh || !nmemb || (*nmemb && !edges)) {
38 meshlink_errno = MESHLINK_EINVAL;
42 pthread_mutex_lock(&(mesh->mesh_mutex));
44 devtool_edge_t *result = NULL;
47 result_size = mesh->edges->count / 2;
49 // if result is smaller than edges, we have to dealloc all the excess devtool_edge_t
50 if(result_size > *nmemb)
51 result = realloc(edges, result_size * sizeof(*result));
56 devtool_edge_t *p = result;
59 for splay_each(edge_t, e, mesh->edges) {
60 // skip edges that do not represent a two-directional connection
61 if((!e->reverse) || (e->reverse->to != e->from))
64 // don't count edges twice
68 assert(n < result_size);
70 p->from = (meshlink_node_t *)e->from;
71 p->to = (meshlink_node_t *)e->to;
72 p->address = e->address.storage;
73 p->options = e->options;
74 p->weight = e->weight;
80 // shrink result to the actual amount of memory used
81 result = realloc(result, n * sizeof(*result));
85 meshlink_errno = MESHLINK_ENOMEM;
88 pthread_mutex_unlock(&(mesh->mesh_mutex));
93 static bool fstrwrite(const char *str, FILE *stream) {
94 size_t len = strlen(str);
96 if(fwrite((void *)str, 1, len, stream) != len)
102 static const char *__itoa(int value) {
103 static char buffer[sizeof(int) * 8 + 1]; // not thread safe
105 if(snprintf(buffer, sizeof(buffer), "%d", value) == -1)
111 bool devtool_export_json_all_edges_state(meshlink_handle_t *mesh, FILE *stream) {
114 pthread_mutex_lock(&(mesh->mesh_mutex));
116 // export edges and nodes
117 size_t node_count = 0;
118 size_t edge_count = 0;
120 meshlink_node_t **nodes = meshlink_get_all_nodes(mesh, NULL, &node_count);
121 devtool_edge_t *edges = devtool_get_all_edges(mesh, NULL, &edge_count);
123 if((!nodes && node_count != 0) || (!edges && edge_count != 0))
127 if(!fstrwrite("{\n", stream))
131 if(!fstrwrite("\t\"nodes\": {\n", stream))
134 for(size_t i = 0; i < node_count; ++i) {
135 if(!fstrwrite("\t\t\"", stream) || !fstrwrite(((node_t *)nodes[i])->name, stream) || !fstrwrite("\": {\n", stream))
138 if(!fstrwrite("\t\t\t\"name\": \"", stream) || !fstrwrite(((node_t *)nodes[i])->name, stream) || !fstrwrite("\",\n", stream))
141 if(!fstrwrite("\t\t\t\"options\": ", stream) || !fstrwrite(__itoa(((node_t *)nodes[i])->options), stream) || !fstrwrite(",\n", stream))
144 if(!fstrwrite("\t\t\t\"devclass\": ", stream) || !fstrwrite(__itoa(((node_t *)nodes[i])->devclass), stream) || !fstrwrite("\n", stream))
147 if(!fstrwrite((i + 1) != node_count ? "\t\t},\n" : "\t\t}\n", stream))
151 if(!fstrwrite("\t},\n", stream))
156 if(!fstrwrite("\t\"edges\": {\n", stream))
159 for(size_t i = 0; i < edge_count; ++i) {
160 if(!fstrwrite("\t\t\"", stream) || !fstrwrite(edges[i].from->name, stream) || !fstrwrite("_to_", stream) || !fstrwrite(edges[i].to->name, stream) || !fstrwrite("\": {\n", stream))
163 if(!fstrwrite("\t\t\t\"from\": \"", stream) || !fstrwrite(edges[i].from->name, stream) || !fstrwrite("\",\n", stream))
166 if(!fstrwrite("\t\t\t\"to\": \"", stream) || !fstrwrite(edges[i].to->name, stream) || !fstrwrite("\",\n", stream))
169 char *host = NULL, *port = NULL, *address = NULL;
170 sockaddr2str((const sockaddr_t *) & (edges[i].address), &host, &port);
173 xasprintf(&address, "{ \"host\": \"%s\", \"port\": %s }", host, port);
178 if(!fstrwrite("\t\t\t\"address\": ", stream) || !fstrwrite(address ? address : "null", stream) || !fstrwrite(",\n", stream)) {
185 if(!fstrwrite("\t\t\t\"options\": ", stream) || !fstrwrite(__itoa(edges[i].options), stream) || !fstrwrite(",\n", stream))
188 if(!fstrwrite("\t\t\t\"weight\": ", stream) || !fstrwrite(__itoa(edges[i].weight), stream) || !fstrwrite("\n", stream))
191 if(!fstrwrite((i + 1) != edge_count ? "\t\t},\n" : "\t\t}\n", stream))
195 if(!fstrwrite("\t}\n", stream))
200 if(!fstrwrite("}", stream))
212 pthread_mutex_unlock(&(mesh->mesh_mutex));