git.meshlink.io Git - meshlink/blob - test/utils.c

   1 #define _GNU_SOURCE
   2
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <pthread.h>
   6 #include <assert.h>
   7 #include <errno.h>
   8
   9 #include "utils.h"
  10
  11 void set_sync_flag(struct sync_flag *s) {
  12         pthread_mutex_lock(&s->mutex);
  13         s->flag = true;
  14         pthread_cond_broadcast(&s->cond);
  15         pthread_mutex_unlock(&s->mutex);
  16 }
  17
  18 bool wait_sync_flag(struct sync_flag *s, int seconds) {
  19         struct timespec timeout;
  20         clock_gettime(CLOCK_REALTIME, &timeout);
  21         timeout.tv_sec += seconds;
  22
  23         while(!s->flag)
  24                 if(!pthread_cond_timedwait(&s->cond, &s->mutex, &timeout) || errno != EINTR) {
  25                         break;
  26                 }
  27
  28         return s->flag;
  29 }
  30
  31 void open_meshlink_pair(meshlink_handle_t **pa, meshlink_handle_t **pb, const char *prefix) {
  32         // Create two new MeshLink instances
  33
  34         *pa = *pb = NULL;
  35
  36         char *a_name, *b_name;
  37
  38         asprintf(&a_name, "%s_conf.1", prefix);
  39         assert(a_name);
  40
  41         asprintf(&b_name, "%s_conf.2", prefix);
  42         assert(b_name);
  43
  44         meshlink_handle_t *a = meshlink_open(a_name, "a", prefix, DEV_CLASS_BACKBONE);
  45         assert(a);
  46
  47         meshlink_handle_t *b = meshlink_open(b_name, "b", prefix, DEV_CLASS_BACKBONE);
  48         assert(b);
  49
  50         meshlink_enable_discovery(a, false);
  51         meshlink_enable_discovery(b, false);
  52
  53         // Import and export both side's data
  54
  55         meshlink_add_address(a, "localhost");
  56
  57         char *data = meshlink_export(a);
  58         assert(data);
  59         assert(meshlink_import(b, data));
  60         free(data);
  61
  62         data = meshlink_export(b);
  63         assert(data);
  64         assert(meshlink_import(a, data));
  65         free(data);
  66
  67         *pa = a;
  68         *pb = b;
  69 }
  70
  71 // Don't poll in the application thread, use a condition variable to signal when the peer is online.
  72 static void pair_status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
  73         set_sync_flag(mesh->priv);
  74 }
  75
  76 void start_meshlink_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
  77         struct sync_flag pair_status = {};
  78
  79         a->priv = &pair_status;
  80         meshlink_set_node_status_cb(a, pair_status_cb);
  81
  82         pthread_mutex_lock(&pair_status.mutex);
  83
  84         meshlink_start(a);
  85         meshlink_start(b);
  86
  87         assert(wait_sync_flag(&pair_status, 5));
  88
  89         pthread_mutex_unlock(&pair_status.mutex);
  90
  91         meshlink_set_node_status_cb(a, NULL);
  92         a->priv = NULL;
  93 }
  94
  95 void stop_meshlink_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
  96         meshlink_stop(a);
  97         meshlink_stop(b);
  98 }
  99
 100 void close_meshlink_pair(meshlink_handle_t *a, meshlink_handle_t *b, const char *prefix) {
 101         meshlink_close(a);
 102         meshlink_close(b);
 103
 104         if(prefix) {
 105                 char *a_name, *b_name;
 106
 107                 asprintf(&a_name, "%s_conf.1", prefix);
 108                 assert(a_name);
 109                 assert(meshlink_destroy(a_name));
 110
 111                 asprintf(&b_name, "%s_conf.2", prefix);
 112                 assert(b_name);
 113                 assert(meshlink_destroy(b_name));
 114         }
 115 }