X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=src%2Fdevtools.c;h=22a0818b559b3e6f291a3e27e36531e4a305071e;hb=HEAD;hp=5a49e3a860296a66ad049bd7d238946424ef1b2d;hpb=18c0a32a4de44adfeee6be42d1c1ae231d093cbf;p=meshlink

diff --git a/src/devtools.c b/src/devtools.c
index 5a49e3a8..12c5432e 100644
--- a/src/devtools.c
+++ b/src/devtools.c
@@ -30,7 +30,7 @@
 
 #include "devtools.h"
 
-static void trybind_nop_probe(void) {
+static void nop_probe(void) {
 	return;
 }
 
@@ -44,9 +44,16 @@ static void inviter_commits_first_nop_probe(bool stage) {
 	return;
 }
 
-void (*devtool_trybind_probe)(void) = trybind_nop_probe;
+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.
@@ -58,7 +65,9 @@ devtool_edge_t *devtool_get_all_edges(meshlink_handle_t *mesh, devtool_edge_t *e
 		return NULL;
 	}
 
-	pthread_mutex_lock(&mesh->mutex);
+	if(pthread_mutex_lock(&mesh->mutex) != 0) {
+		abort();
+	}
 
 	devtool_edge_t *result = NULL;
 	unsigned int result_size = 0;
@@ -128,7 +137,9 @@ bool devtool_export_json_all_edges_state(meshlink_handle_t *mesh, FILE *stream)
 
 	bool result = true;
 
-	pthread_mutex_lock(&mesh->mutex);
+	if(pthread_mutex_lock(&mesh->mutex) != 0) {
+		abort();
+	}
 
 	// export edges and nodes
 	size_t node_count = 0;
@@ -248,45 +259,73 @@ done:
 	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;
 	}
 
-	node_t *internal = (node_t *)node;
+	devtool_get_reset_node_status(mesh, node, status, false);
+}
 
-	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;
+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;
 	}
 
-	pthread_mutex_unlock(&mesh->mutex);
+	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) {
@@ -298,7 +337,9 @@ meshlink_submesh_t **devtool_get_all_submeshes(meshlink_handle_t *mesh, meshlink
 	meshlink_submesh_t **result;
 
 	//lock mesh->nodes
-	pthread_mutex_lock(&mesh->mutex);
+	if(pthread_mutex_lock(&mesh->mutex) != 0) {
+		abort();
+	}
 
 	*nmemb = mesh->submeshes->count;
 	result = realloc(submeshes, *nmemb * sizeof(*submeshes));
@@ -336,3 +377,33 @@ meshlink_handle_t *devtool_open_in_netns(const char *confbase, const char *name,
 
 	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);
+}