manynodes: put a prefix di graph json files because with LXC many machines write...

[meshlink] / examples / manynodes.c

diff --git a/examples/manynodes.c b/examples/manynodes.c
index 99a3e5b2dd806a15d2f32fd5f337b653695691e6..3f59ec38448366da194e442807d83e3ca095ad1c 100644 (file)
--- a/examples/manynodes.c
+++ b/examples/manynodes.c
@@ -18,6 +18,8 @@
 
 static int n = 10;
 static meshlink_handle_t **mesh;
+static char* namesprefix="machine1";
+static int nodeindex = 0;
 
 static meshlink_node_t **nodes;
 static size_t nnodes;
@@ -111,7 +113,7 @@ void exportmeshgraph_timer(int signum)
        gettimeofday(&ts, NULL);
 
        char name[1024];
-       snprintf(name, sizeof(name), "graph_%ld_%03ld.json", ts.tv_sec, ts.tv_usec/1000);
+       snprintf(name, sizeof(name), "%sgraph_%ld_%03ld.json", namesprefix,ts.tv_sec, ts.tv_usec/1000);
 
        exportmeshgraph(name);
 }
@@ -184,7 +186,7 @@ static void parse_command(char *buf) {
                        return;
                }
 
-               invitation = meshlink_invite(mesh[0], arg);
+               invitation = meshlink_invite(mesh[nodeindex], arg);
                if(!invitation) {
                        fprintf(stderr, "Could not invite '%s': %s\n", arg, meshlink_strerror(meshlink_errno));
                        return;
@@ -197,12 +199,12 @@ static void parse_command(char *buf) {
                        fprintf(stderr, "/join requires an argument!\n");
                        return;
                }
-               meshlink_stop(mesh[0]);
-               if(!meshlink_join(mesh[0], arg))
+               meshlink_stop(mesh[nodeindex]);
+               if(!meshlink_join(mesh[nodeindex], arg))
                        fprintf(stderr, "Could not join using invitation: %s\n", meshlink_strerror(meshlink_errno));
                else {
                        fprintf(stderr, "Invitation accepted!\n");
-                       meshlink_start(mesh[0]);
+                       meshlink_start(mesh[nodeindex]);
                }
        } else if(!strcasecmp(buf, "kick")) {
                if(!arg) {
@@ -210,18 +212,18 @@ static void parse_command(char *buf) {
                        return;
                }
 
-               meshlink_node_t *node = meshlink_get_node(mesh[0], arg);
+               meshlink_node_t *node = meshlink_get_node(mesh[nodeindex], arg);
                if(!node) {
                        fprintf(stderr, "Unknown node '%s'\n", arg);
                        return;
                }
 
-               meshlink_blacklist(mesh[0], node);
+               meshlink_blacklist(mesh[nodeindex], node);
 
                printf("Node '%s' blacklisted.\n", arg);
        } else if(!strcasecmp(buf, "who")) {
                if(!arg) {
-                       nodes = meshlink_get_all_nodes(mesh[0], nodes, &nnodes);
+                       nodes = meshlink_get_all_nodes(mesh[nodeindex], nodes, &nnodes);
                        if(!nodes) {
                                fprintf(stderr, "Could not get list of nodes: %s\n", meshlink_strerror(meshlink_errno));
                        } else {
@@ -231,11 +233,11 @@ static void parse_command(char *buf) {
                                printf("\n");
                        }
                } else {
-                       meshlink_node_t *node = meshlink_get_node(mesh[0], arg);
+                       meshlink_node_t *node = meshlink_get_node(mesh[nodeindex], arg);
                        if(!node) {
                                fprintf(stderr, "Unknown node '%s'\n", arg);
                        } else {
-                               printf("Node %s found, pmtu %zd\n", arg, meshlink_get_pmtu(mesh[0], node));
+                               printf("Node %s found, pmtu %zd\n", arg, meshlink_get_pmtu(mesh[nodeindex], node));
                        }
                }
        } else if(!strcasecmp(buf, "link")) {
@@ -248,6 +250,15 @@ static void parse_command(char *buf) {
                exportmeshgraph_end(NULL);
        } else if(!strcasecmp(buf, "test")) {
                testmesh();
+       } else if(!strcasecmp(buf, "select")) {
+               if(!arg) {
+                       fprintf(stderr, "/select requires an argument!\n");
+                       return;
+               }
+               nodeindex = atoi(arg);
+               printf("Index is now %d\n",nodeindex);
+       } else if(!strcasecmp(buf, "stop")) {
+               meshlink_stop(mesh[nodeindex]);
        } else if(!strcasecmp(buf, "quit")) {
                printf("Bye!\n");
                fclose(stdin);
@@ -262,6 +273,8 @@ static void parse_command(char *buf) {
                        "/link                 Link all nodes together.\n"
                        "/eg <path>            Export graph as json file.\n"
                        "/test                 Test functionality sending some data to all nodes\n"
+                       "/select <number>      Select the active node running the user commands\n"
+                       "/stop                 Call meshlink_stop, use /select first to select which node to stop\n"
                        "/quit                 Exit this program.\n"
                        );
        } else {
@@ -307,7 +320,7 @@ static void parse_input(char *buf) {
                if(*msg == ' ')
                        msg++;
 
-               destination = meshlink_get_node(mesh[0], buf);
+               destination = meshlink_get_node(mesh[nodeindex], buf);
                if(!destination) {
                        fprintf(stderr, "Unknown node '%s'\n", buf);
                        return;
@@ -319,7 +332,7 @@ static void parse_input(char *buf) {
                return;
        }
 
-       if(!meshlink_send(mesh[0], destination, msg, strlen(msg) + 1)) {
+       if(!meshlink_send(mesh[nodeindex], destination, msg, strlen(msg) + 1)) {
                fprintf(stderr, "Could not send message to '%s': %s\n", destination->name, meshlink_strerror(meshlink_errno));
                return;
        }
@@ -329,7 +342,6 @@ static void parse_input(char *buf) {
 
 int main(int argc, char *argv[]) {
        const char *basebase = ".manynodes";
-       const char *namesprefix = "machine1";
        const char *graphexporttimeout = NULL;
        char buf[1024];
 
@@ -352,7 +364,7 @@ int main(int argc, char *argv[]) {
 
        mesh = calloc(n, sizeof *mesh);
 
-       meshlink_set_log_cb(NULL, MESHLINK_WARNING, log_message);
+       meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_message);
        mkdir(basebase, 0750);
 
        char filename[PATH_MAX];
@@ -361,8 +373,13 @@ 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", i%_DEV_CLASS_MAX);
-               meshlink_set_log_cb(mesh[i], MESHLINK_WARNING, log_message);
+               if (n/(i+1) > n/4) {
+                       mesh[i] = meshlink_open(filename, nodename, "manynodes", DEV_CLASS_BACKBONE);
+               }
+               else {
+                       mesh[i] = meshlink_open(filename, nodename, "manynodes", DEV_CLASS_PORTABLE);
+               }
+               meshlink_set_log_cb(mesh[i], MESHLINK_DEBUG, log_message);
                if(!mesh[i]) {
                        fprintf(stderr, "errno is: %d\n", meshlink_errno);
                        fprintf(stderr, "Could not open %s: %s\n", filename, meshlink_strerror(meshlink_errno));