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