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