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

   1 #define _GNU_SOURCE
   2
   3 #ifdef NDEBUG
   4 #undef NDEBUG
   5 #endif
   6
   7 #include <stdio.h>
   8 #include <stdlib.h>
   9 #include <pthread.h>
  10 #include <assert.h>
  11 #include <errno.h>
  12 #include <time.h>
  13
  14 #include "full.h"
  15 #include "utils.h"
  16
  17 void init_sync_flag(struct sync_flag *s) {
  18         assert(pthread_mutex_init(&s->mutex, NULL) == 0);
  19         assert(pthread_cond_init(&s->cond, NULL) == 0);
  20         s->flag = false;
  21 }
  22
  23 void set_sync_flag(struct sync_flag *s, bool value) {
  24         assert(pthread_mutex_lock(&s->mutex) == 0);
  25         s->flag = value;
  26         assert(pthread_cond_broadcast(&s->cond) == 0);
  27         assert(pthread_mutex_unlock(&s->mutex) == 0);
  28 }
  29
  30 void reset_sync_flag(struct sync_flag *s) {
  31         assert(pthread_mutex_lock(&s->mutex) == 0);
  32         s->flag = false;
  33         assert(pthread_mutex_unlock(&s->mutex) == 0);
  34 }
  35
  36 bool check_sync_flag(struct sync_flag *s) {
  37         bool flag;
  38         assert(pthread_mutex_lock(&s->mutex) == 0);
  39         flag = s->flag;
  40         assert(pthread_mutex_unlock(&s->mutex) == 0);
  41         return flag;
  42 }
  43
  44 bool wait_sync_flag(struct sync_flag *s, int seconds) {
  45         struct timespec timeout;
  46         clock_gettime(CLOCK_REALTIME, &timeout);
  47         timeout.tv_sec += seconds;
  48
  49         assert(pthread_mutex_lock(&s->mutex) == 0);
  50
  51         while(!s->flag) {
  52                 if(!pthread_cond_timedwait(&s->cond, &s->mutex, &timeout) || errno != EINTR) {
  53                         break;
  54                 }
  55         }
  56
  57         assert(pthread_mutex_unlock(&s->mutex) == 0);
  58
  59         return s->flag;
  60 }
  61
  62 void link_meshlink_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
  63         // Import and export both side's data
  64
  65         assert(meshlink_set_canonical_address(a, meshlink_get_self(a), "localhost", NULL));
  66         assert(meshlink_set_canonical_address(b, meshlink_get_self(b), "localhost", NULL));
  67
  68         char *data = meshlink_export(a);
  69         assert(data);
  70         assert(meshlink_import(b, data));
  71         free(data);
  72
  73         data = meshlink_export(b);
  74         assert(data);
  75         assert(meshlink_import(a, data));
  76         free(data);
  77 }
  78
  79 void open_meshlink_pair(meshlink_handle_t **pa, meshlink_handle_t **pb, const char *prefix) {
  80         // Create two new MeshLink instances
  81
  82         *pa = *pb = NULL;
  83
  84         char *a_name, *b_name;
  85
  86         assert(asprintf(&a_name, "%s_conf.1", prefix) > 0);
  87         assert(a_name);
  88
  89         assert(asprintf(&b_name, "%s_conf.2", prefix) > 0);
  90         assert(b_name);
  91
  92         assert(meshlink_destroy(a_name));
  93         assert(meshlink_destroy(b_name));
  94
  95         meshlink_handle_t *a = meshlink_open(a_name, "a", prefix, DEV_CLASS_BACKBONE);
  96         assert(a);
  97
  98         meshlink_handle_t *b = meshlink_open(b_name, "b", prefix, DEV_CLASS_BACKBONE);
  99         assert(b);
 100
 101         free(a_name);
 102         free(b_name);
 103
 104         link_meshlink_pair(a, b);
 105
 106         *pa = a;
 107         *pb = b;
 108 }
 109
 110 void open_meshlink_pair_ephemeral(meshlink_handle_t **pa, meshlink_handle_t **pb, const char *prefix) {
 111         // Create two new MeshLink instances
 112
 113         *pa = *pb = NULL;
 114
 115         meshlink_handle_t *a = meshlink_open_ephemeral("a", prefix, DEV_CLASS_BACKBONE);
 116         meshlink_handle_t *b = meshlink_open_ephemeral("b", prefix, DEV_CLASS_BACKBONE);
 117
 118         assert(a);
 119         assert(b);
 120
 121         link_meshlink_pair(a, b);
 122
 123         *pa = a;
 124         *pb = b;
 125 }
 126
 127 // Don't poll in the application thread, use a condition variable to signal when the peer is online.
 128 static void pair_status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
 129         (void)node;
 130
 131         if(reachable && meshlink_get_self(mesh) != node) {
 132                 set_sync_flag(mesh->priv, true);
 133         }
 134 }
 135
 136 void start_meshlink_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
 137         struct sync_flag pair_status = {.flag = false};
 138         init_sync_flag(&pair_status);
 139
 140         a->priv = &pair_status;
 141         meshlink_set_node_status_cb(a, pair_status_cb);
 142
 143         assert(meshlink_start(a));
 144         assert(meshlink_start(b));
 145
 146         assert(wait_sync_flag(&pair_status, 5));
 147
 148         meshlink_set_node_status_cb(a, NULL);
 149         a->priv = NULL;
 150 }
 151
 152 void start_full_tiny_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
 153         struct sync_flag pair_status = {.flag = false};
 154         init_sync_flag(&pair_status);
 155
 156         a->priv = &pair_status;
 157         full_meshlink_set_node_status_cb(a, pair_status_cb);
 158
 159         assert(full_meshlink_start(a));
 160         assert(meshlink_start(b));
 161
 162         assert(wait_sync_flag(&pair_status, 5));
 163
 164         full_meshlink_set_node_status_cb(a, NULL);
 165         a->priv = NULL;
 166 }
 167
 168 void stop_meshlink_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
 169         meshlink_stop(a);
 170         meshlink_stop(b);
 171 }
 172
 173 void stop_full_tiny_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
 174         full_meshlink_stop(a);
 175         meshlink_stop(b);
 176 }
 177
 178 void close_meshlink_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
 179         meshlink_close(a);
 180         meshlink_close(b);
 181 }
 182
 183 void close_full_tiny_pair(meshlink_handle_t *a, meshlink_handle_t *b) {
 184         full_meshlink_close(a);
 185         meshlink_close(b);
 186 }
 187
 188 void log_cb(meshlink_handle_t *mesh, meshlink_log_level_t level, const char *text) {
 189         static const char *levelstr[] = {
 190                 [MESHLINK_DEBUG] = "DEBUG",
 191                 [MESHLINK_INFO] = "INFO",
 192                 [MESHLINK_WARNING] = "WARNING",
 193                 [MESHLINK_ERROR] = "ERROR",
 194                 [MESHLINK_CRITICAL] = "CRITICAL",
 195         };
 196
 197         static struct timespec ts0;
 198         struct timespec ts;
 199
 200         clock_gettime(CLOCK_MONOTONIC, &ts);
 201
 202         if(ts0.tv_sec == 0) {
 203                 ts0 = ts;
 204         }
 205
 206         float diff = (ts.tv_sec - ts0.tv_sec) + (ts.tv_nsec - ts0.tv_nsec) * 1e-9;
 207
 208         fprintf(stderr, "%7.3f (%s) [%s] %s\n",
 209                 diff,
 210                 mesh ? mesh->name : "",
 211                 levelstr[level],
 212                 text);
 213 }