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