json export of node and edge structure

[meshlink] / examples / manynodes.c

diff --git a/examples/manynodes.c b/examples/manynodes.c
index b435aaf432002334970836b2447a8fc1d878b91b..3b617e4e5b517d7af0a0c33eeccf5cf249afa130 100644 (file)
--- a/examples/manynodes.c
+++ b/examples/manynodes.c
@@ -7,6 +7,11 @@
 #include <linux/limits.h>
 
 #include "../src/meshlink.h"
+#include "../src/devtools.h"
+
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <errno.h>
 
 static int n = 10;
 static meshlink_handle_t **mesh;
@@ -14,6 +19,17 @@ static meshlink_handle_t **mesh;
 static meshlink_node_t **nodes;
 static size_t nnodes;
 
+static void log_message(meshlink_handle_t *mesh, meshlink_log_level_t level, const char *text) {
+       const char *levelstr[] = {
+               [MESHLINK_DEBUG] = "\x1b[34mDEBUG",
+               [MESHLINK_INFO] = "\x1b[32mINFO",
+               [MESHLINK_WARNING] = "\x1b[33mWARNING",
+               [MESHLINK_ERROR] = "\x1b[31mERROR",
+               [MESHLINK_CRITICAL] = "\x1b[31mCRITICAL",
+       };
+       fprintf(stderr, "%s\t%s:\x1b[0m %s\n", mesh ? mesh->name : "global",levelstr[level], text);
+}
+
 //Test mesh sending data
 static void testmesh () {
 
@@ -23,13 +39,13 @@ static void testmesh () {
                        if(!nodes) {
                                fprintf(stderr, "Could not get list of nodes: %s\n", meshlink_strerror(meshlink_errno));
                        } else {
-                               printf("%zu known nodes:", nnodes);
+                               printf("%zu known nodes:\n", nnodes);
                                for(int i = 0; i < nnodes; i++) {
-                                       printf(" %s", nodes[i]->name);
-                                       if(!meshlink_send(mesh[nindex], nodes[i], "magic", strlen("magic") + 1)) {
+                                       //printf(" %s\n", nodes[i]->name);
+                                               if(!meshlink_send(mesh[nindex], nodes[i], "magic", strlen("magic") + 1)) {
                fprintf(stderr, "Could not send message to '%s': %s\n", nodes[i]->name, meshlink_strerror(meshlink_errno));
-       }
-       }
+                                               }
+                               }
 
                        }
 
@@ -51,6 +67,40 @@ 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;
+}
+
 static void parse_command(char *buf) {
        char *arg = strchr(buf, ' ');
        if(arg)
@@ -115,11 +165,13 @@ static void parse_command(char *buf) {
                        if(!node) {
                                fprintf(stderr, "Unknown node '%s'\n", arg);
                        } else {
-                               printf("Node %s found\n", arg);
+                               printf("Node %s found, pmtu %zd\n", arg, meshlink_get_pmtu(mesh[0], node));
                        }
                }
        } else if(!strcasecmp(buf, "link")) {
                linkmesh();
+       } else if(!strcasecmp(buf, "eg")) {
+               exportmeshgraph(arg);
        } else if(!strcasecmp(buf, "test")) {
                testmesh();
        } else if(!strcasecmp(buf, "quit")) {
@@ -134,6 +186,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"
                        );
@@ -221,6 +274,7 @@ int main(int argc, char *argv[]) {
 
        mesh = calloc(n, sizeof *mesh);
 
+       meshlink_set_log_cb(NULL, MESHLINK_INFO, log_message);
        mkdir(basebase, 0750);
 
        char filename[PATH_MAX];
@@ -229,7 +283,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");
+               mesh[i] = meshlink_open(filename, nodename, "manynodes", i%_DEV_CLASS_MAX);
+               meshlink_set_log_cb(mesh[i], MESHLINK_INFO, log_message);
                if(itsnew)
                        meshlink_add_address(mesh[i], "localhost");
                if(!mesh[i]) {