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