#define _GNU_SOURCE
+#ifdef NDEBUG
+#undef NDEBUG
+#endif
+
#include <stdio.h>
+#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <assert.h>
+#include <sys/time.h>
#include "meshlink.h"
+#include "devtools.h"
#include "utils.h"
-static void log_cb(meshlink_handle_t *mesh, meshlink_log_level_t level, const char *text) {
- static struct timeval tv0;
- struct timeval tv;
+static struct sync_flag received;
+static struct sync_flag bar_learned_baz;
+static struct sync_flag baz_learned_bar;
- if(tv0.tv_sec == 0)
- gettimeofday(&tv0, NULL);
- gettimeofday(&tv, NULL);
- fprintf(stderr, "%u.%.03u ", (unsigned int)(tv.tv_sec - tv0.tv_sec), (unsigned int)tv.tv_usec / 1000);
+static void receive_cb(meshlink_handle_t *mesh, meshlink_node_t *source, const void *data, size_t len) {
+ (void)mesh;
+ (void)source;
- if(mesh)
- fprintf(stderr, "(%s) ", mesh->name);
- fprintf(stderr, "[%d] %s\n", level, text);
+ if(len == 5 && !memcmp(data, "Hello", 5)) {
+ set_sync_flag(&received, true);
+ }
}
-static bool received = false;
+static void bar_status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
+ (void)mesh;
+ (void)reachable;
-static void receive_cb(meshlink_handle_t *mesh, meshlink_node_t *source, const void *data, size_t len) {
- fprintf(stderr, "RECEIVED SOMETHING\n");
- if(len == 5 && !memcmp(data, "Hello", 5))
- received = true;
+ if(!strcmp(node->name, "baz")) {
+ set_sync_flag(&bar_learned_baz, true);
+ }
+}
+
+static void baz_status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
+ (void)mesh;
+ (void)reachable;
+
+ if(!strcmp(node->name, "bar")) {
+ set_sync_flag(&baz_learned_bar, true);
+ }
}
-int main(int argc, char *argv[]) {
+int main() {
+ meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb);
+
// Create three instances.
const char *name[3] = {"foo", "bar", "baz"};
char *data[3];
for(int i = 0; i < 3; i++) {
- char *path;
- asprintf(&path, "trio_conf.%d", i);
- assert(path);
+ char *path = NULL;
+ assert(asprintf(&path, "trio_conf.%d", i) != -1 && path);
+ assert(meshlink_destroy(path));
mesh[i] = meshlink_open(path, name[i], "trio", DEV_CLASS_BACKBONE);
assert(mesh[i]);
+ free(path);
+
+ assert(meshlink_set_canonical_address(mesh[i], meshlink_get_self(mesh[i]), "localhost", NULL));
data[i] = meshlink_export(mesh[i]);
assert(data[i]);
// start the nodes
- for(int i = 0; i < 3; i++)
- meshlink_start(mesh[i]);
+ meshlink_set_node_status_cb(mesh[1], bar_status_cb);
+ meshlink_set_node_status_cb(mesh[2], baz_status_cb);
+
+ for(int i = 0; i < 3; i++) {
+ free(data[i]);
+ assert(meshlink_start(mesh[i]));
+ }
// the nodes should now learn about each other
- assert_after(meshlink_get_node(mesh[1], name[2]), 5);
- assert_after(meshlink_get_node(mesh[2], name[1]), 5);
+ assert(wait_sync_flag(&bar_learned_baz, 5));
+ assert(wait_sync_flag(&baz_learned_bar, 5));
- // Send a packet
+ // Send a packet, expect it is received
meshlink_set_receive_cb(mesh[1], receive_cb);
- assert_after((meshlink_send(mesh[2], meshlink_get_node(mesh[2], name[1]), "Hello", 5), received), 15);
+
+ for(int i = 0; i < 15; i++) {
+ assert(meshlink_send(mesh[2], meshlink_get_node(mesh[2], name[1]), "Hello", 5));
+
+ if(wait_sync_flag(&received, 1)) {
+ break;
+ }
+ }
+
+ assert(wait_sync_flag(&received, 15));
+
+ // Check that the second and third node have autoconnected to each other
+
+ devtool_edge_t *edges = NULL;
+ size_t nedges = 0;
+ assert_after((edges = devtool_get_all_edges(mesh[1], edges, &nedges), nedges == 3), 15);
+ free(edges);
// Stop the first node
// Communication should still be possible
- assert_after((meshlink_send(mesh[2], meshlink_get_node(mesh[2], name[1]), "Hello", 5), received), 15);
+ set_sync_flag(&received, false);
+
+ for(int i = 0; i < 15; i++) {
+ assert(meshlink_send(mesh[2], meshlink_get_node(mesh[2], name[1]), "Hello", 5));
+
+ if(wait_sync_flag(&received, 1)) {
+ break;
+ }
+ }
+
+ assert(wait_sync_flag(&received, 15));
// Stop the other nodes
- for(int i = 1; i < 3; i++)
+ for(int i = 1; i < 3; i++) {
meshlink_stop(mesh[i]);
+ }
sleep(1);
// Start just the other two nodes
- meshlink_set_log_cb(mesh[1], MESHLINK_DEBUG, log_cb);
-
- for(int i = 1; i < 3; i++)
- meshlink_start(mesh[i]);
+ for(int i = 1; i < 3; i++) {
+ assert(meshlink_start(mesh[i]));
+ }
assert(meshlink_get_node(mesh[1], name[2]));
assert(meshlink_get_node(mesh[2], name[1]));
// Communication should still be possible
- received = false;
- assert_after((meshlink_send(mesh[2], meshlink_get_node(mesh[2], name[1]), "Hello", 5), received), 25);
+ set_sync_flag(&received, false);
+
+ for(int i = 0; i < 15; i++) {
+ assert(meshlink_send(mesh[2], meshlink_get_node(mesh[2], name[1]), "Hello", 5));
+
+ if(wait_sync_flag(&received, 1)) {
+ break;
+ }
+ }
+
+ assert(wait_sync_flag(&received, 1));
// Clean up.
- for(int i = 0; i < 3; i++)
+ for(int i = 0; i < 3; i++) {
meshlink_close(mesh[i]);
+ }
}
projects / meshlink / blobdiff