X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=examples%2Fmanynodes.c;h=99a3e5b2dd806a15d2f32fd5f337b653695691e6;hb=fabbc781bb11f66383f2ef37ef84ec23094244d6;hp=ecc8b458e3abdb9e775ec80df94b0b2920ec0ae7;hpb=f416117836dc944822cac06efb54377fb21ee66f;p=meshlink diff --git a/examples/manynodes.c b/examples/manynodes.c index ecc8b458..99a3e5b2 100644 --- a/examples/manynodes.c +++ b/examples/manynodes.c @@ -7,6 +7,14 @@ #include #include "../src/meshlink.h" +#include "../src/devtools.h" + +#include +#include +#include + +#include +#include static int n = 10; static meshlink_handle_t **mesh; @@ -14,6 +22,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 +42,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 +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_%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) @@ -120,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")) { @@ -134,6 +260,7 @@ static void parse_command(char *buf) { "/kick Blacklist the given node.\n" "/who [] List all nodes or show information about the given node.\n" "/link Link all nodes together.\n" + "/eg Export graph as json file.\n" "/test Test functionality sending some data to all nodes\n" "/quit Exit this program.\n" ); @@ -203,13 +330,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; } @@ -219,8 +347,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_WARNING, log_message); mkdir(basebase, 0750); char filename[PATH_MAX]; @@ -229,9 +361,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); - if(itsnew) - meshlink_add_address(mesh[i], "localhost"); + mesh[i] = meshlink_open(filename, nodename, "manynodes", i%_DEV_CLASS_MAX); + 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)); @@ -253,11 +384,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++)