--- /dev/null
+#ifdef NDEBUG
+#undef NDEBUG
+#endif
+
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/time.h>
+#include <time.h>
+#include <assert.h>
+
+#include "meshlink.h"
+#include "utils.h"
+
+struct sync_flag bar_reachable;
+
+void status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
+ (void)mesh;
+
+ if(reachable && !strcmp(node->name, "bar")) {
+ set_sync_flag(&bar_reachable, true);
+ }
+}
+
+int main() {
+ struct meshlink_node **nodes = NULL;
+ size_t nnodes = 0;
+
+ meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb);
+
+ // Open new meshlink instances.
+
+ assert(meshlink_destroy("get_all_nodes_conf.1"));
+ assert(meshlink_destroy("get_all_nodes_conf.2"));
+ assert(meshlink_destroy("get_all_nodes_conf.3"));
+
+ meshlink_handle_t *mesh[3];
+ mesh[0] = meshlink_open("get_all_nodes_conf.1", "foo", "import-export", DEV_CLASS_BACKBONE);
+ assert(mesh[0]);
+
+ mesh[1] = meshlink_open("get_all_nodes_conf.2", "bar", "import-export", DEV_CLASS_STATIONARY);
+ assert(mesh[1]);
+
+ mesh[2] = meshlink_open("get_all_nodes_conf.3", "baz", "get-all-nodes", DEV_CLASS_STATIONARY);
+ assert(mesh[2]);
+
+ // Check that we only know about ourself.
+
+ nodes = meshlink_get_all_nodes(mesh[0], nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_self(mesh[0]));
+
+ // We should never have been online.
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
+ assert(nnodes == 0);
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_self(mesh[0]));
+
+ // Let nodes know about each other.
+
+ for(int i = 0; i < 3; i++) {
+ meshlink_enable_discovery(mesh[i], false);
+ assert(meshlink_add_address(mesh[i], "localhost"));
+ char *data = meshlink_export(mesh[i]);
+ assert(data);
+
+ for(int j = 0; j < 3; j++) {
+ if(i == j) {
+ continue;
+ }
+
+ assert(meshlink_import(mesh[j], data));
+ }
+ }
+
+ // We should know about all nodes now, and their device class.
+
+ nodes = meshlink_get_all_nodes(mesh[0], nodes, &nnodes);
+ assert(nnodes == 3);
+
+ nodes = meshlink_get_all_nodes_by_dev_class(mesh[0], DEV_CLASS_BACKBONE, nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_self(mesh[0]));
+
+ nodes = meshlink_get_all_nodes_by_dev_class(mesh[0], DEV_CLASS_STATIONARY, nodes, &nnodes);
+ assert(nnodes == 2);
+
+ // But no node should have been online.
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
+ assert(nnodes == 0);
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
+ assert(nnodes == 3);
+
+ // Start foo.
+
+ time_t foo_started = time(NULL);
+ assert(meshlink_start(mesh[0]));
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_self(mesh[0]));
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
+ assert(nnodes == 2);
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], foo_started - 1, -1, nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_self(mesh[0]));
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started - 1, nodes, &nnodes);
+ assert(nnodes == 0);
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started + 1, nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_self(mesh[0]));
+
+ // Start bar and wait for it to connect.
+
+ meshlink_set_node_status_cb(mesh[0], status_cb);
+
+ sleep(2);
+ assert(meshlink_start(mesh[1]));
+ assert(wait_sync_flag(&bar_reachable, 20));
+ time_t bar_started = time(NULL);
+
+ // Validate time ranges.
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
+ assert(nnodes == 2);
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_node(mesh[0], "baz"));
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started + 1, nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_self(mesh[0]));
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_started, bar_started, nodes, &nnodes);
+ assert(nnodes == 2);
+ assert(nodes[0] == meshlink_get_node(mesh[0], "bar"));
+ assert(nodes[1] == meshlink_get_self(mesh[0]));
+
+ // Stop bar.
+
+ meshlink_stop(mesh[1]);
+ sleep(2);
+ time_t bar_stopped = time(NULL);
+
+ // Validate we can see when bar was reachable.
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_stopped, bar_stopped, nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_self(mesh[0]));
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_started, bar_started, nodes, &nnodes);
+ assert(nnodes == 2);
+ assert(nodes[0] == meshlink_get_node(mesh[0], "bar"));
+ assert(nodes[1] == meshlink_get_self(mesh[0]));
+
+ // Close and restart foo, check that it remembers correctly.
+
+ meshlink_close(mesh[0]);
+ sleep(2);
+ time_t foo_stopped = time(NULL);
+ mesh[0] = meshlink_open("get_all_nodes_conf.1", "foo", "import-export", DEV_CLASS_BACKBONE);
+ assert(mesh[0]);
+
+ nodes = meshlink_get_all_nodes(mesh[0], nodes, &nnodes);
+ assert(nnodes == 3);
+
+ nodes = meshlink_get_all_nodes_by_dev_class(mesh[0], DEV_CLASS_BACKBONE, nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_self(mesh[0]));
+
+ nodes = meshlink_get_all_nodes_by_dev_class(mesh[0], DEV_CLASS_STATIONARY, nodes, &nnodes);
+ assert(nnodes == 2);
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_node(mesh[0], "baz"));
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
+ assert(nnodes == 2);
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started - 1, nodes, &nnodes);
+ assert(nnodes == 0);
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started + 1, nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_self(mesh[0]));
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_started, bar_started, nodes, &nnodes);
+ assert(nnodes == 2);
+ assert(nodes[0] == meshlink_get_node(mesh[0], "bar"));
+ assert(nodes[1] == meshlink_get_self(mesh[0]));
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_stopped, bar_stopped, nodes, &nnodes);
+ assert(nnodes == 1);
+ assert(nodes[0] == meshlink_get_self(mesh[0]));
+
+ nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], foo_stopped, -1, nodes, &nnodes);
+ assert(nnodes == 0);
+
+ // Clean up.
+
+ for(int i = 0; i < 3; i++) {
+ meshlink_close(mesh[i]);
+ }
+}
projects / meshlink / blobdiff