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