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

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