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