#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;
[MESHLINK_ERROR] = "\x1b[31mERROR",
[MESHLINK_CRITICAL] = "\x1b[31mCRITICAL",
};
- fprintf(stderr, "%s\t%s:\x1b[0m %s\n", mesh->name,levelstr[level], text);
+ fprintf(stderr, "%s\t%s:\x1b[0m %s\n", mesh ? mesh->name : "global",levelstr[level], text);
}
//Test mesh sending data
} else {
printf("%zu known nodes:\n", nnodes);
for(int i = 0; i < nnodes; i++) {
- printf(" %s\n", 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));
- }
- }
+ }
+ }
}
}
}
+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)
}
} 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")) {
"/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"
);
mesh = calloc(n, sizeof *mesh);
- meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_message);
+ meshlink_set_log_cb(NULL, MESHLINK_WARNING, log_message);
mkdir(basebase, 0750);
char filename[PATH_MAX];
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]) {
printf("%d nodes started.\nType /help for a list of commands.\n", started);
+ // handle input
while(fgets(buf, sizeof buf, stdin))
parse_input(buf);