git.meshlink.io Git - meshlink/blob - test/import-export.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 <assert.h>
  11
  12 #include "meshlink.h"
  13 #include "utils.h"
  14
  15 static struct sync_flag bar_reachable;
  16 static struct sync_flag pmtu_flag;
  17
  18 static 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 static void pmtu_cb(meshlink_handle_t *mesh, meshlink_node_t *node, uint16_t pmtu) {
  27         (void)mesh;
  28
  29         if(pmtu && !strcmp(node->name, "bar")) {
  30                 set_sync_flag(&pmtu_flag, true);
  31         }
  32 }
  33
  34 int main(void) {
  35         init_sync_flag(&bar_reachable);
  36
  37         meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb);
  38
  39         // Open two new meshlink instance.
  40
  41         assert(meshlink_destroy("import_export_conf.1"));
  42         assert(meshlink_destroy("import_export_conf.2"));
  43
  44         meshlink_handle_t *mesh1 = meshlink_open("import_export_conf.1", "foo", "import-export", DEV_CLASS_BACKBONE);
  45         assert(mesh1);
  46
  47         meshlink_handle_t *mesh2 = meshlink_open("import_export_conf.2", "bar", "import-export", DEV_CLASS_BACKBONE);
  48         assert(mesh2);
  49
  50         // Disable local discovery
  51
  52         meshlink_enable_discovery(mesh1, false);
  53         meshlink_enable_discovery(mesh2, false);
  54
  55         // Import and export both side's data
  56
  57         assert(meshlink_set_canonical_address(mesh1, meshlink_get_self(mesh1), "localhost", NULL));
  58         assert(meshlink_set_canonical_address(mesh2, meshlink_get_self(mesh2), "localhost", NULL));
  59
  60         char *data = meshlink_export(mesh1);
  61         assert(data);
  62
  63         assert(meshlink_import(mesh2, data));
  64         free(data);
  65
  66         data = meshlink_export(mesh2);
  67         assert(data);
  68
  69         assert(meshlink_import(mesh1, data));
  70
  71         // Check that importing twice is fine
  72         assert(meshlink_import(mesh1, data));
  73         free(data);
  74
  75         // Check that importing garbage is not fine
  76         assert(!meshlink_import(mesh1, "Garbage\n"));
  77
  78         // Check that foo knows bar, but that it is not reachable.
  79
  80         time_t last_reachable;
  81         time_t last_unreachable;
  82         meshlink_node_t *bar = meshlink_get_node(mesh1, "bar");
  83         assert(bar);
  84         assert(!meshlink_get_node_reachability(mesh1, bar, &last_reachable, &last_unreachable));
  85         assert(!last_reachable);
  86         assert(!last_unreachable);
  87
  88         // Start both instances
  89
  90         meshlink_set_node_status_cb(mesh1, status_cb);
  91         meshlink_set_node_pmtu_cb(mesh1, pmtu_cb);
  92
  93         assert(meshlink_start(mesh1));
  94         assert(meshlink_start(mesh2));
  95
  96         // Wait for the two to connect.
  97
  98         assert(wait_sync_flag(&bar_reachable, 10));
  99
 100         // Wait for UDP communication to become possible.
 101
 102         assert(wait_sync_flag(&pmtu_flag, 10));
 103
 104         // Check that we now have reachability information
 105
 106         assert(meshlink_get_node_reachability(mesh1, bar, &last_reachable, &last_unreachable));
 107         assert(last_reachable);
 108
 109         // Stop the meshes.
 110
 111         meshlink_stop(mesh1);
 112         meshlink_stop(mesh2);
 113
 114         // Check that bar is no longer reachable
 115
 116         assert(!meshlink_get_node_reachability(mesh1, bar, &last_reachable, &last_unreachable));
 117         assert(last_reachable);
 118         assert(last_unreachable);
 119         assert(last_reachable <= last_unreachable);
 120
 121         // Clean up.
 122
 123         meshlink_close(mesh2);
 124         meshlink_close(mesh1);
 125 }