git.meshlink.io Git - meshlink/blob - test/get-all-nodes.c

   1 #ifdef NDEBUG
   2 #undef NDEBUG
   3 #endif
   4
   5 #include <stdio.h>
   6 #include <unistd.h>
   7 #include <stdlib.h>
   8 #include <string.h>
   9 #include <sys/time.h>
  10 #include <time.h>
  11 #include <assert.h>
  12
  13 #include "meshlink.h"
  14 #include "utils.h"
  15
  16 struct sync_flag bar_reachable;
  17
  18 void status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
  19         (void)mesh;
  20
  21         if(reachable && !strcmp(node->name, "bar")) {
  22                 set_sync_flag(&bar_reachable, true);
  23         }
  24 }
  25
  26 int main() {
  27         struct meshlink_node **nodes = NULL;
  28         size_t nnodes = 0;
  29
  30         meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb);
  31
  32         // Open new meshlink instances.
  33
  34         assert(meshlink_destroy("get_all_nodes_conf.1"));
  35         assert(meshlink_destroy("get_all_nodes_conf.2"));
  36         assert(meshlink_destroy("get_all_nodes_conf.3"));
  37
  38         meshlink_handle_t *mesh[3];
  39         mesh[0] = meshlink_open("get_all_nodes_conf.1", "foo", "import-export", DEV_CLASS_BACKBONE);
  40         assert(mesh[0]);
  41
  42         mesh[1] = meshlink_open("get_all_nodes_conf.2", "bar", "import-export", DEV_CLASS_STATIONARY);
  43         assert(mesh[1]);
  44
  45         mesh[2] = meshlink_open("get_all_nodes_conf.3", "baz", "get-all-nodes", DEV_CLASS_STATIONARY);
  46         assert(mesh[2]);
  47
  48         // Check that we only know about ourself.
  49
  50         nodes = meshlink_get_all_nodes(mesh[0], nodes, &nnodes);
  51         assert(nnodes == 1);
  52         assert(nodes[0] == meshlink_get_self(mesh[0]));
  53
  54         // We should never have been online.
  55
  56         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
  57         assert(nnodes == 0);
  58
  59         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
  60         assert(nnodes == 1);
  61         assert(nodes[0] == meshlink_get_self(mesh[0]));
  62
  63         // Let nodes know about each other.
  64
  65         for(int i = 0; i < 3; i++) {
  66                 meshlink_enable_discovery(mesh[i], false);
  67                 assert(meshlink_set_canonical_address(mesh[i], meshlink_get_self(mesh[i]), "localhost", NULL));
  68                 char *data = meshlink_export(mesh[i]);
  69                 assert(data);
  70
  71                 for(int j = 0; j < 3; j++) {
  72                         if(i == j) {
  73                                 continue;
  74                         }
  75
  76                         assert(meshlink_import(mesh[j], data));
  77                 }
  78         }
  79
  80         // We should know about all nodes now, and their device class.
  81
  82         nodes = meshlink_get_all_nodes(mesh[0], nodes, &nnodes);
  83         assert(nnodes == 3);
  84
  85         nodes = meshlink_get_all_nodes_by_dev_class(mesh[0], DEV_CLASS_BACKBONE, nodes, &nnodes);
  86         assert(nnodes == 1);
  87         assert(nodes[0] == meshlink_get_self(mesh[0]));
  88
  89         nodes = meshlink_get_all_nodes_by_dev_class(mesh[0], DEV_CLASS_STATIONARY, nodes, &nnodes);
  90         assert(nnodes == 2);
  91
  92         // But no node should have been online.
  93
  94         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
  95         assert(nnodes == 0);
  96
  97         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
  98         assert(nnodes == 3);
  99
 100         // Start foo.
 101
 102         time_t foo_started = time(NULL);
 103         assert(meshlink_start(mesh[0]));
 104
 105         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
 106         assert(nnodes == 1);
 107         assert(nodes[0] == meshlink_get_self(mesh[0]));
 108
 109         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
 110         assert(nnodes == 2);
 111
 112         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], foo_started - 1, -1, nodes, &nnodes);
 113         assert(nnodes == 1);
 114         assert(nodes[0] == meshlink_get_self(mesh[0]));
 115
 116         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started - 1, nodes, &nnodes);
 117         assert(nnodes == 0);
 118
 119         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started + 1, nodes, &nnodes);
 120         assert(nnodes == 1);
 121         assert(nodes[0] == meshlink_get_self(mesh[0]));
 122
 123         // Start bar and wait for it to connect.
 124
 125         meshlink_set_node_status_cb(mesh[0], status_cb);
 126
 127         sleep(2);
 128         assert(meshlink_start(mesh[1]));
 129         assert(wait_sync_flag(&bar_reachable, 20));
 130         time_t bar_started = time(NULL);
 131
 132         // Validate time ranges.
 133
 134         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
 135         assert(nnodes == 2);
 136
 137         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
 138         assert(nnodes == 1);
 139         assert(nodes[0] == meshlink_get_node(mesh[0], "baz"));
 140
 141         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started + 1, nodes, &nnodes);
 142         assert(nnodes == 1);
 143         assert(nodes[0] == meshlink_get_self(mesh[0]));
 144
 145         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_started, bar_started, nodes, &nnodes);
 146         assert(nnodes == 2);
 147         assert(nodes[0] == meshlink_get_node(mesh[0], "bar"));
 148         assert(nodes[1] == meshlink_get_self(mesh[0]));
 149
 150         // Stop bar.
 151
 152         meshlink_stop(mesh[1]);
 153         sleep(2);
 154         time_t bar_stopped = time(NULL);
 155
 156         // Validate we can see when bar was reachable.
 157
 158         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_stopped, bar_stopped, nodes, &nnodes);
 159         assert(nnodes == 1);
 160         assert(nodes[0] == meshlink_get_self(mesh[0]));
 161
 162         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_started, bar_started, nodes, &nnodes);
 163         assert(nnodes == 2);
 164         assert(nodes[0] == meshlink_get_node(mesh[0], "bar"));
 165         assert(nodes[1] == meshlink_get_self(mesh[0]));
 166
 167         // Close and restart foo, check that it remembers correctly.
 168
 169         meshlink_close(mesh[0]);
 170         sleep(2);
 171         time_t foo_stopped = time(NULL);
 172         mesh[0] = meshlink_open("get_all_nodes_conf.1", "foo", "import-export", DEV_CLASS_BACKBONE);
 173         assert(mesh[0]);
 174
 175         nodes = meshlink_get_all_nodes(mesh[0], nodes, &nnodes);
 176         assert(nnodes == 3);
 177
 178         nodes = meshlink_get_all_nodes_by_dev_class(mesh[0], DEV_CLASS_BACKBONE, nodes, &nnodes);
 179         assert(nnodes == 1);
 180         assert(nodes[0] == meshlink_get_self(mesh[0]));
 181
 182         nodes = meshlink_get_all_nodes_by_dev_class(mesh[0], DEV_CLASS_STATIONARY, nodes, &nnodes);
 183         assert(nnodes == 2);
 184
 185         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
 186         assert(nnodes == 1);
 187         assert(nodes[0] == meshlink_get_node(mesh[0], "baz"));
 188
 189         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
 190         assert(nnodes == 2);
 191
 192         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started - 1, nodes, &nnodes);
 193         assert(nnodes == 0);
 194
 195         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started + 1, nodes, &nnodes);
 196         assert(nnodes == 1);
 197         assert(nodes[0] == meshlink_get_self(mesh[0]));
 198
 199         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_started, bar_started, nodes, &nnodes);
 200         assert(nnodes == 2);
 201         assert(nodes[0] == meshlink_get_node(mesh[0], "bar"));
 202         assert(nodes[1] == meshlink_get_self(mesh[0]));
 203
 204         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_stopped, bar_stopped, nodes, &nnodes);
 205         assert(nnodes == 1);
 206         assert(nodes[0] == meshlink_get_self(mesh[0]));
 207
 208         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], foo_stopped, -1, nodes, &nnodes);
 209         assert(nnodes == 0);
 210
 211         // Clean up.
 212
 213         for(int i = 0; i < 3; i++) {
 214                 meshlink_close(mesh[i]);
 215         }
 216 }