]> git.meshlink.io Git - meshlink/blob - test/trio.c
bdb3904e37e4540faf734e4bd5c7c145f3eddd35
[meshlink] / test / trio.c
1 #define _GNU_SOURCE
2
3 #ifdef NDEBUG
4 #undef NDEBUG
5 #endif
6
7 #include <stdio.h>
8 #include <stdlib.h>
9 #include <string.h>
10 #include <unistd.h>
11 #include <errno.h>
12 #include <assert.h>
13 #include <sys/time.h>
14
15 #include "meshlink.h"
16 #include "devtools.h"
17 #include "utils.h"
18
19 static struct sync_flag received;
20 static struct sync_flag bar_learned_baz;
21 static struct sync_flag baz_learned_bar;
22
23 static void receive_cb(meshlink_handle_t *mesh, meshlink_node_t *source, const void *data, size_t len) {
24         (void)mesh;
25         (void)source;
26
27         if(len == 5 && !memcmp(data, "Hello", 5)) {
28                 set_sync_flag(&received, true);
29         }
30 }
31
32 static void bar_status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
33         (void)mesh;
34         (void)reachable;
35
36         if(!strcmp(node->name, "baz")) {
37                 set_sync_flag(&bar_learned_baz, true);
38         }
39 }
40
41 static void baz_status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
42         (void)mesh;
43         (void)reachable;
44
45         if(!strcmp(node->name, "bar")) {
46                 set_sync_flag(&baz_learned_bar, true);
47         }
48 }
49
50 int main(void) {
51         init_sync_flag(&received);
52         init_sync_flag(&bar_learned_baz);
53         init_sync_flag(&baz_learned_bar);
54
55         meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb);
56
57         // Create three instances.
58
59         const char *name[3] = {"foo", "bar", "baz"};
60         meshlink_handle_t *mesh[3];
61         char *data[3];
62
63         for(int i = 0; i < 3; i++) {
64                 char *path = NULL;
65                 assert(asprintf(&path, "trio_conf.%d", i) != -1 && path);
66
67                 assert(meshlink_destroy(path));
68                 mesh[i] = meshlink_open(path, name[i], "trio", DEV_CLASS_BACKBONE);
69                 assert(mesh[i]);
70                 free(path);
71
72                 assert(meshlink_set_canonical_address(mesh[i], meshlink_get_self(mesh[i]), "localhost", NULL));
73
74                 data[i] = meshlink_export(mesh[i]);
75                 assert(data[i]);
76         }
77
78         // first node knows the two other nodes
79
80         for(int i = 1; i < 3; i++) {
81                 assert(meshlink_import(mesh[i], data[0]));
82                 assert(meshlink_import(mesh[0], data[i]));
83
84                 assert(meshlink_get_node(mesh[i], name[0]));
85                 assert(meshlink_get_node(mesh[0], name[i]));
86         }
87
88         // second and third node should not know each other yet
89
90         assert(!meshlink_get_node(mesh[1], name[2]));
91         assert(!meshlink_get_node(mesh[2], name[1]));
92
93         // start the nodes
94
95         meshlink_set_node_status_cb(mesh[1], bar_status_cb);
96         meshlink_set_node_status_cb(mesh[2], baz_status_cb);
97
98         for(int i = 0; i < 3; i++) {
99                 free(data[i]);
100                 assert(meshlink_start(mesh[i]));
101         }
102
103         // the nodes should now learn about each other
104
105         assert(wait_sync_flag(&bar_learned_baz, 5));
106         assert(wait_sync_flag(&baz_learned_bar, 5));
107
108         // Send a packet, expect it is received
109
110         meshlink_set_receive_cb(mesh[1], receive_cb);
111
112         for(int i = 0; i < 15; i++) {
113                 assert(meshlink_send(mesh[2], meshlink_get_node(mesh[2], name[1]), "Hello", 5));
114
115                 if(wait_sync_flag(&received, 1)) {
116                         break;
117                 }
118         }
119
120         assert(wait_sync_flag(&received, 15));
121
122         // Check that the second and third node have autoconnected to each other
123
124         devtool_edge_t *edges = NULL;
125         size_t nedges = 0;
126         assert_after((edges = devtool_get_all_edges(mesh[1], edges, &nedges), nedges == 3), 15);
127         free(edges);
128
129         // Stop the first node
130
131         meshlink_stop(mesh[0]);
132         sleep(1);
133
134         // Communication should still be possible
135
136         set_sync_flag(&received, false);
137
138         for(int i = 0; i < 15; i++) {
139                 assert(meshlink_send(mesh[2], meshlink_get_node(mesh[2], name[1]), "Hello", 5));
140
141                 if(wait_sync_flag(&received, 1)) {
142                         break;
143                 }
144         }
145
146         assert(wait_sync_flag(&received, 15));
147
148         // Stop the other nodes
149
150         for(int i = 1; i < 3; i++) {
151                 meshlink_stop(mesh[i]);
152         }
153
154         sleep(1);
155
156         // Start just the other two nodes
157
158         for(int i = 1; i < 3; i++) {
159                 assert(meshlink_start(mesh[i]));
160         }
161
162         assert(meshlink_get_node(mesh[1], name[2]));
163         assert(meshlink_get_node(mesh[2], name[1]));
164
165         // Communication should still be possible
166
167         set_sync_flag(&received, false);
168
169         for(int i = 0; i < 15; i++) {
170                 assert(meshlink_send(mesh[2], meshlink_get_node(mesh[2], name[1]), "Hello", 5));
171
172                 if(wait_sync_flag(&received, 1)) {
173                         break;
174                 }
175         }
176
177         assert(wait_sync_flag(&received, 1));
178
179         // Clean up.
180
181         for(int i = 0; i < 3; i++) {
182                 meshlink_close(mesh[i]);
183         }
184 }