#include <strings.h>
#include <sys/types.h>
#include <sys/stat.h>
+
+#if !defined(_WIN32) && !defined(__APPLE__)
#include <linux/limits.h>
+#elif defined(__APPLE__)
+#include <sys/syslimits.h>
+#endif
#include "../src/meshlink.h"
+#include "../src/devtools.h"
+
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <errno.h>
-static int n = 100;
+#include <sys/time.h>
+#include <signal.h>
+#include <assert.h>
+
+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;
static void log_message(meshlink_handle_t *mesh, meshlink_log_level_t level, const char *text) {
- const char *levelstr[] = {"DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"};
- fprintf(stderr, "%s: %s\n", levelstr[level], 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);
}
-static void receive(meshlink_handle_t *mesh, meshlink_node_t *source, const void *data, size_t len) {
- const char *msg = data;
+//Test mesh sending data
+static void testmesh() {
- if(!len || msg[len - 1]) {
- fprintf(stderr, "Received invalid data from %s\n", source->name);
- return;
- }
+ for(int nindex = 0; nindex < n; nindex++) {
- printf("%s says: %s\n", source->name, msg);
-}
+ nodes = meshlink_get_all_nodes(mesh[nindex], nodes, &nnodes);
-static void node_status(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
- if(reachable)
- printf("%s joined.\n", node->name);
- else
- printf("%s left.\n", node->name);
-}
+ if(!nodes) {
+ fprintf(stderr, "Could not get list of nodes: %s\n", meshlink_strerror(meshlink_errno));
+ } else {
+ printf("%zu known nodes:\n", nnodes);
+
+ for(size_t i = 0; i < nnodes; i++) {
+ //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));
+ }
+ }
+
+ }
+ }
+}
// Make all nodes know about each other by importing each others public keys and addresses.
static void linkmesh() {
for(int i = 0; i < n; i++) {
for(int j = i + 1; j < n; j++) {
char *dataj = meshlink_export(mesh[j]);
- meshlink_import(mesh[i], dataj);
- meshlink_import(mesh[j], datai);
+
+ if(!meshlink_import(mesh[i], dataj) || !meshlink_import(mesh[j], datai)) {
+ fprintf(stderr, "Could not exchange keys between %s and %s: %s\n", mesh[i]->name, mesh[j]->name, meshlink_strerror(meshlink_errno));
+ }
+
free(dataj);
}
}
}
+static bool exportmeshgraph(const char *path) {
+ assert(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) {
+ (void)signum;
+
+ struct timeval ts;
+ gettimeofday(&ts, NULL);
+
+ char name[1024];
+ snprintf(name, sizeof(name), "%sgraph_%ld_%03ld.json", namesprefix, ts.tv_sec, ts.tv_usec / 1000);
+
+ exportmeshgraph(name);
+}
+
+#ifndef _WIN32
+static bool exportmeshgraph_started = false;
+
+static bool exportmeshgraph_end(void) {
+ if(!exportmeshgraph_started) {
+ return false;
+ }
+
+ struct itimerval zero_timer;
+
+ 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()) {
+ 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;
+}
+#else
+static bool exportmeshgraph_end(void) {
+ return false;
+}
+
+static bool exportmeshgraph_begin(const char *timeout_str) {
+ return false;
+}
+#endif
+
static void parse_command(char *buf) {
char *arg = strchr(buf, ' ');
- if(arg)
+
+ if(arg) {
*arg++ = 0;
+ }
if(!strcasecmp(buf, "invite")) {
char *invitation;
return;
}
- invitation = meshlink_invite(mesh[0], arg);
+ invitation = meshlink_invite(mesh[nodeindex], NULL, arg);
+
if(!invitation) {
fprintf(stderr, "Could not invite '%s': %s\n", arg, meshlink_strerror(meshlink_errno));
return;
return;
}
- 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
+ } else {
fprintf(stderr, "Invitation accepted!\n");
+ }
+
+ if(!meshlink_start(mesh[nodeindex])) {
+ fprintf(stderr, "Could not restart MeshLink: %s\n", meshlink_strerror(meshlink_errno));
+ exit(1);
+ }
} else if(!strcasecmp(buf, "kick")) {
if(!arg) {
fprintf(stderr, "/kick requires an argument!\n");
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);
+ if(!meshlink_blacklist(mesh[nodeindex], node)) {
+ fprintf(stderr, "Error blacklising '%s': %s", arg, meshlink_strerror(meshlink_errno));
+ return;
+ }
printf("Node '%s' blacklisted.\n", arg);
+ } else if(!strcasecmp(buf, "whitelist")) {
+ if(!arg) {
+ fprintf(stderr, "/whitelist requires an argument!\n");
+ return;
+ }
+
+ meshlink_node_t *node = meshlink_get_node(mesh[nodeindex], arg);
+
+ if(!node) {
+ fprintf(stderr, "Error looking up '%s': %s\n", arg, meshlink_strerror(meshlink_errno));
+ return;
+ }
+
+ if(!meshlink_whitelist(mesh[nodeindex], node)) {
+ fprintf(stderr, "Error whitelising '%s': %s", arg, meshlink_strerror(meshlink_errno));
+ return;
+ }
+
+ printf("Node '%s' whitelisted.\n", arg);
} else if(!strcasecmp(buf, "who")) {
if(!arg) {
- meshlink_node_t *nodes[100];
- size_t n = meshlink_get_all_nodes(mesh[0], nodes, 100);
- if(!n) {
- fprintf(stderr, "No nodes known!\n");
+ 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 {
- printf("Known nodes:");
- for(int i = 0; i < n && i < 100; i++)
+ printf("%zu known nodes:", nnodes);
+
+ for(size_t i = 0; i < nnodes; i++) {
printf(" %s", nodes[i]->name);
- if(n > 100)
- printf(" (and %zu more)", n - 100);
+ }
+
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\n", arg);
+ printf("Node %s found, pmtu %ld\n", arg, (long int)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();
+ } 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);
} else if(!strcasecmp(buf, "help")) {
printf(
- "<name>: <message> Send a message to the given node.\n"
- " Subsequent messages don't need the <name>: prefix.\n"
- "/invite <name> Create an invitation for a new node.\n"
- "/join <invitation> Join an existing mesh using an invitation.\n"
- "/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"
- "/quit Exit this program.\n"
- );
+ "<name>: <message> Send a message to the given node.\n"
+ " Subsequent messages don't need the <name>: prefix.\n"
+ "/invite <name> Create an invitation for a new node.\n"
+ "/join <invitation> Join an existing mesh using an invitation.\n"
+ "/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"
+ "/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 {
fprintf(stderr, "Unknown command '/%s'\n", buf);
}
static meshlink_node_t *destination;
size_t len;
- if(!buf)
+ if(!buf) {
return;
+ }
// Remove newline.
len = strlen(buf);
- if(len && buf[len - 1] == '\n')
+ if(len && buf[len - 1] == '\n') {
buf[--len] = 0;
+ }
- if(len && buf[len - 1] == '\r')
+ if(len && buf[len - 1] == '\r') {
buf[--len] = 0;
+ }
// Ignore empty lines.
- if(!len)
+ if(!len) {
return;
+ }
// Commands start with '/'
- if(*buf == '/')
- return parse_command(buf + 1);
+ if(*buf == '/') {
+ parse_command(buf + 1);
+ return;
+ }
// Lines in the form "name: message..." set the destination node.
if(colon) {
*colon = 0;
msg = colon + 1;
- if(*msg == ' ')
+
+ if(*msg == ' ') {
msg++;
+ }
+
+ destination = meshlink_get_node(mesh[nodeindex], buf);
- destination = meshlink_get_node(mesh[0], buf);
if(!destination) {
fprintf(stderr, "Unknown node '%s'\n", buf);
return;
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;
}
int main(int argc, char *argv[]) {
const char *basebase = ".manynodes";
+ const char *graphexporttimeout = NULL;
char buf[1024];
- if(argc > 1)
+ if(argc > 1) {
n = atoi(argv[1]);
+ }
if(n < 1) {
- fprintf(stderr, "Usage: %s [number of local nodes] [confbase]\n", argv[0]);
+ fprintf(stderr, "Usage: %s [number of local nodes] [confbase] [prefixnodenames] [graphexport timeout]\n", argv[0]);
return 1;
}
- if(argc > 2)
+ if(argc > 2) {
basebase = argv[2];
+ }
+
+ if(argc > 3) {
+ namesprefix = argv[3];
+ }
- mesh = calloc(n, sizeof *mesh);
+ if(argc > 4) {
+ graphexporttimeout = argv[4];
+ }
+ mesh = calloc(n, sizeof(*mesh));
+
+ meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_message);
+#ifndef _WIN32
mkdir(basebase, 0750);
+#else
+ mkdir(basebase);
+#endif
char filename[PATH_MAX];
char nodename[100];
+
for(int i = 0; i < n; i++) {
- snprintf(nodename, sizeof nodename, "node%d", i);
- snprintf(filename, sizeof filename, "%s/%s", basebase, nodename);
- bool itsnew = access(filename, R_OK);
- mesh[i] = meshlink_open(filename, nodename);
- if(itsnew)
- meshlink_add_address(mesh[i], "localhost");
+ snprintf(nodename, sizeof(nodename), "%snode%d", namesprefix, i);
+ snprintf(filename, sizeof(filename), "%s/%s", basebase, nodename);
+
+ 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));
int started = 0;
for(int i = 0; i < n; i++) {
- if(!meshlink_start(mesh[i]))
+ if(!meshlink_start(mesh[i])) {
fprintf(stderr, "Could not start node %d: %s\n", i, meshlink_strerror(meshlink_errno));
- else
+ } else {
started++;
+ }
}
if(!started) {
return 1;
}
+ if(graphexporttimeout) {
+ exportmeshgraph_begin(graphexporttimeout);
+ }
+
printf("%d nodes started.\nType /help for a list of commands.\n", started);
- while(fgets(buf, sizeof buf, stdin))
+ // handle input
+ while(fgets(buf, sizeof(buf), stdin)) {
parse_input(buf);
+ }
+
+ exportmeshgraph_end();
printf("Nodes stopping.\n");
- for(int i = 0; i < n; i++)
+ for(int i = 0; i < n; i++) {
meshlink_close(mesh[i]);
+ }
return 0;
}