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

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