X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=examples%2Fmanynodes.c;h=3215cfccd1f7fbf97f3d656225e20455036da895;hb=1aa7ef75218e1b45969ee60ce81e0a9b73beb774;hp=96aa5f0f9843db84bb7c4bc7481a285965fa6491;hpb=b70895cf367100792263f04d97a7533b26c5ece3;p=meshlink

diff --git a/examples/manynodes.c b/examples/manynodes.c
index 96aa5f0f..3215cfcc 100644
--- a/examples/manynodes.c
+++ b/examples/manynodes.c
@@ -7,6 +7,14 @@
 #include <linux/limits.h>
 
 #include "../src/meshlink.h"
+#include "../src/devtools.h"
+
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <errno.h>
+
+#include <sys/time.h>
+#include <signal.h>
 
 static int n = 10;
 static meshlink_handle_t **mesh;
@@ -62,6 +70,107 @@ static void linkmesh() {
 	}
 }
 
+static bool exportmeshgraph(const char* path)
+{
+	struct stat ps;
+	int psr = stat(path, &ps);
+
+	if(psr == 0 || errno != ENOENT)
+	{
+		if(psr == -1)
+			{ perror("stat"); }
+		else
+			{ fprintf(stderr, "%s exists already\n", path); }
+
+		return false;
+	}
+
+	FILE* stream = fopen(path, "w");
+
+	if(!stream)
+	{
+		perror("stream");
+		return false;
+	}
+
+	if(!devtool_export_json_all_edges_state(mesh[0], stream))
+	{
+		fclose(stream);
+		fprintf(stderr, "could not export graph\n");
+		return false;
+	}
+
+	fclose(stream);
+	return true;
+}
+
+
+void exportmeshgraph_timer(int signum)
+{
+	struct timeval ts;
+	gettimeofday(&ts, NULL);
+
+	char name[1024];
+	snprintf(name, sizeof(name), "graph_%ld_%ld.json", ts.tv_sec, ts.tv_usec/1000);
+
+	exportmeshgraph(name);
+}
+
+static bool exportmeshgraph_started = false;
+
+static bool exportmeshgraph_end(const char* none)
+{
+	if(!exportmeshgraph_started)
+		{ return false; }
+
+	struct itimerval zero_timer = { 0 };
+	setitimer (ITIMER_REAL, &zero_timer, NULL);
+
+	exportmeshgraph_started = false;
+
+	return true;
+}
+
+static bool exportmeshgraph_begin(const char* timeout_str)
+{
+	if(!timeout_str)
+		return false;
+
+	if(exportmeshgraph_started)
+	{
+		if(!exportmeshgraph_end(NULL))
+			return false;
+	}
+
+	// get timeout
+	int timeout = atoi(timeout_str);
+
+	if(timeout < 100)
+		{ timeout = 100; }
+
+	int timeout_sec = timeout / 1000;
+	int timeout_msec = timeout % 1000;
+
+	/* Install timer_handler as the signal handler for SIGALRM. */
+	signal(SIGALRM, exportmeshgraph_timer);
+
+	/* Configure the timer to expire after X msec... */
+	struct itimerval timer;
+	timer.it_value.tv_sec = timeout_sec;
+	timer.it_value.tv_usec = timeout_msec * 1000;
+
+	/* ... and every X msec after that. */
+	timer.it_interval.tv_sec = timeout_sec;
+	timer.it_interval.tv_usec = timeout_msec * 1000;
+
+	/* Start a real timer. */
+	setitimer (ITIMER_REAL, &timer, NULL);
+
+	exportmeshgraph_started = true;
+
+	return true;
+}
+
 static void parse_command(char *buf) {
 	char *arg = strchr(buf, ' ');
 	if(arg)
@@ -131,6 +240,12 @@ static void parse_command(char *buf) {
 		}
 	} else if(!strcasecmp(buf, "link")) {
 		linkmesh();
+	} else if(!strcasecmp(buf, "eg")) {
+		exportmeshgraph(arg);
+	} else if(!strcasecmp(buf, "egb")) {
+		exportmeshgraph_begin(arg);
+	} else if(!strcasecmp(buf, "ege")) {
+		exportmeshgraph_end(NULL);
 	} else if(!strcasecmp(buf, "test")) {
 		testmesh();
 	} else if(!strcasecmp(buf, "quit")) {
@@ -145,6 +260,7 @@ static void parse_command(char *buf) {
 			"/kick <name>          Blacklist the given node.\n"
 			"/who [<name>]         List all nodes or show information about the given node.\n"
 			"/link                 Link all nodes together.\n"
+			"/eg <path>            Export graph as json file.\n"
 			"/test                 Test functionality sending some data to all nodes\n"
 			"/quit                 Exit this program.\n"
 			);
@@ -232,7 +348,7 @@ int main(int argc, char *argv[]) {
 
 	mesh = calloc(n, sizeof *mesh);
 
-	meshlink_set_log_cb(NULL, MESHLINK_INFO, log_message);
+	meshlink_set_log_cb(NULL, MESHLINK_WARNING, log_message);
 	mkdir(basebase, 0750);
 
 	char filename[PATH_MAX];
@@ -241,8 +357,8 @@ int main(int argc, char *argv[]) {
 		snprintf(nodename, sizeof nodename, "%snode%d", namesprefix,i);
 		snprintf(filename, sizeof filename, "%s/%s", basebase, nodename);
 		bool itsnew = access(filename, R_OK);
-		mesh[i] = meshlink_open(filename, nodename, "manynodes", STATIONARY);
-		meshlink_set_log_cb(mesh[i], MESHLINK_INFO, log_message);
+		mesh[i] = meshlink_open(filename, nodename, "manynodes", i%_DEV_CLASS_MAX);
+		meshlink_set_log_cb(mesh[i], MESHLINK_WARNING, log_message);
 		if(itsnew)
 			meshlink_add_address(mesh[i], "localhost");
 		if(!mesh[i]) {
@@ -268,6 +384,7 @@ int main(int argc, char *argv[]) {
 
 	printf("%d nodes started.\nType /help for a list of commands.\n", started);
 
+	// handle input
 	while(fgets(buf, sizeof buf, stdin))
 		parse_input(buf);