]> git.meshlink.io Git - meshlink/blob - test/utils.c
Reduce how often we have to poll the packet queue.
[meshlink] / 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 }