#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 void nop_probe(void) {
+ return;
+}
+
+static void keyrotate_nop_probe(int stage) {
+ (void)stage;
+ return;
+}
+
+static void inviter_commits_first_nop_probe(bool stage) {
+ (void)stage;
+ return;
+}
+
+static void sptps_renewal_nop_probe(meshlink_node_t *node) {
+ (void)node;
+ return;
+}
+
+void (*devtool_trybind_probe)(void) = 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;
+void (*devtool_adns_resolve_probe)(void) = nop_probe;
+void (*devtool_sptps_renewal_probe)(meshlink_node_t *node) = sptps_renewal_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.
return NULL;
}
- pthread_mutex_lock(&(mesh->mesh_mutex));
+ if(pthread_mutex_lock(&mesh->mutex) != 0) {
+ abort();
+ }
devtool_edge_t *result = NULL;
unsigned int result_size = 0;
// 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));
+ result = xrealloc(edges, result_size * sizeof(*result));
} else {
result = edges;
}
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)) {
+ if(!e->reverse || e->reverse->to != e->from) {
continue;
}
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++;
}
// shrink result to the actual amount of memory used
- result = realloc(result, n * sizeof(*result));
+ result = xrealloc(result, n * sizeof(*result));
*nmemb = n;
} else {
*nmemb = 0;
meshlink_errno = MESHLINK_ENOMEM;
}
- pthread_mutex_unlock(&(mesh->mesh_mutex));
+ pthread_mutex_unlock(&mesh->mutex);
return result;
}
static bool fstrwrite(const char *str, FILE *stream) {
+ assert(stream);
+
size_t len = strlen(str);
if(fwrite((void *)str, 1, len, stream) != len) {
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) {
- return "";
- }
-
- return buffer;
-}
-
bool devtool_export_json_all_edges_state(meshlink_handle_t *mesh, FILE *stream) {
+ assert(stream);
+
bool result = true;
- pthread_mutex_lock(&(mesh->mesh_mutex));
+ if(pthread_mutex_lock(&mesh->mutex) != 0) {
+ abort();
+ }
// export edges and nodes
size_t node_count = 0;
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)) {
goto fail;
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;
}
}
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) {
xasprintf(&address, "{ \"host\": \"%s\", \"port\": %s }", host, port);
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;
}
free(nodes);
free(edges);
- pthread_mutex_unlock(&(mesh->mesh_mutex));
+ pthread_mutex_unlock(&mesh->mutex);
+
+ return result;
+}
+
+static void devtool_get_reset_node_status(meshlink_handle_t *mesh, meshlink_node_t *node, devtool_node_status_t *status, bool reset) {
+ node_t *internal = (node_t *)node;
+
+ if(pthread_mutex_lock(&mesh->mutex) != 0) {
+ abort();
+ }
+
+ if(status) {
+ 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_data = internal->in_data;
+ status->out_data = internal->out_data;
+ status->in_forward = internal->in_forward;
+ status->out_forward = internal->out_forward;
+ status->in_meta = internal->in_meta;
+ status->out_meta = internal->out_meta;
+
+ // 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;
+ }
+ }
+
+ if(reset) {
+ internal->in_data = 0;
+ internal->out_data = 0;
+ internal->in_forward = 0;
+ internal->out_forward = 0;
+ internal->in_meta = 0;
+ internal->out_meta = 0;
+ }
+
+ pthread_mutex_unlock(&mesh->mutex);
+}
+
+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;
+ }
+
+ devtool_get_reset_node_status(mesh, node, status, false);
+}
+
+void devtool_reset_node_counters(meshlink_handle_t *mesh, meshlink_node_t *node, devtool_node_status_t *status) {
+ if(!mesh || !node) {
+ meshlink_errno = MESHLINK_EINVAL;
+ return;
+ }
+
+ devtool_get_reset_node_status(mesh, node, status, true);
+}
+
+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
+ if(pthread_mutex_lock(&mesh->mutex) != 0) {
+ abort();
+ }
+
+ *nmemb = mesh->submeshes->count;
+ result = realloc(submeshes, *nmemb * sizeof(*submeshes));
+
+ if(result) {
+ meshlink_submesh_t **p = result;
+
+ 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;
+}
+
+void devtool_force_sptps_renewal(meshlink_handle_t *mesh, meshlink_node_t *node) {
+ if(!mesh || !node) {
+ meshlink_errno = MESHLINK_EINVAL;
+ return;
+ }
+
+ node_t *n = (node_t *)node;
+ connection_t *c = n->connection;
+
+ n->last_req_key = -3600;
+
+ if(c) {
+ c->last_key_renewal = -3600;
+ }
+}
+
+void devtool_set_meta_status_cb(meshlink_handle_t *mesh, meshlink_node_status_cb_t cb) {
+ if(!mesh) {
+ meshlink_errno = MESHLINK_EINVAL;
+ return;
+ }
+
+ if(pthread_mutex_lock(&mesh->mutex) != 0) {
+ abort();
+ }
+
+ mesh->meta_status_cb = cb;
+ pthread_mutex_unlock(&mesh->mutex);
+}