X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=examples%2Fmanynodes.c;h=16d25ddcce5d6cb94763394a0efd70b44205173b;hb=64f243ffc7f01bfa3a0980fb0f92106224676f06;hp=3b617e4e5b517d7af0a0c33eeccf5cf249afa130;hpb=31ab43d8a0691e773db6992fa3b52ca24f7f8db4;p=meshlink diff --git a/examples/manynodes.c b/examples/manynodes.c index 3b617e4e..16d25ddc 100644 --- a/examples/manynodes.c +++ b/examples/manynodes.c @@ -13,9 +13,14 @@ #include #include +#include +#include + static int n = 10; static meshlink_handle_t **mesh; +static int nodeindex = 0; + static meshlink_node_t **nodes; static size_t nnodes; @@ -90,7 +95,7 @@ static bool exportmeshgraph(const char* path) return false; } - if(!devtool_export_json_all_edges_state(mesh[0], stream)) + if(!devtool_export_json_all_edges_state(mesh[nodeindex], stream)) { fclose(stream); fprintf(stderr, "could not export graph\n"); @@ -101,6 +106,73 @@ static bool exportmeshgraph(const char* path) return true; } + +void exportmeshgraph_timer(int signum) +{ + struct timeval ts; + gettimeofday(&ts, NULL); + + char name[1024]; + snprintf(name, sizeof(name), "graph_%ld_%03ld.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 immediately... */ + struct itimerval timer; + timer.it_value.tv_sec = 0; + timer.it_value.tv_usec = 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) @@ -114,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; @@ -127,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) { @@ -140,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 { @@ -161,19 +233,32 @@ 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")) { 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, "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); @@ -188,6 +273,8 @@ static void parse_command(char *buf) { "/link Link all nodes together.\n" "/eg Export graph as json file.\n" "/test Test functionality sending some data to all nodes\n" + "/select 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 { @@ -233,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; @@ -245,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; } @@ -256,13 +343,14 @@ 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]; if(argc > 1) n = atoi(argv[1]); if(n < 1) { - fprintf(stderr, "Usage: %s [number of local nodes] [confbase] [prefixnodenames]\n", argv[0]); + fprintf(stderr, "Usage: %s [number of local nodes] [confbase] [prefixnodenames] [graphexport timeout]\n", argv[0]); return 1; } @@ -272,9 +360,12 @@ int main(int argc, char *argv[]) { if(argc > 3) namesprefix = argv[3]; + if(argc > 4) + graphexporttimeout = argv[4]; + 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]; @@ -284,9 +375,7 @@ int main(int argc, char *argv[]) { 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_INFO, log_message); - if(itsnew) - meshlink_add_address(mesh[i], "localhost"); + meshlink_set_log_cb(mesh[i], MESHLINK_WARNING, 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)); @@ -308,11 +397,17 @@ int main(int argc, char *argv[]) { return 1; } + if(graphexporttimeout) + { exportmeshgraph_begin(graphexporttimeout); } + printf("%d nodes started.\nType /help for a list of commands.\n", started); + // handle input while(fgets(buf, sizeof buf, stdin)) parse_input(buf); + exportmeshgraph_end(NULL); + printf("Nodes stopping.\n"); for(int i = 0; i < n; i++)