]> git.meshlink.io Git - meshlink/blob - test/get-all-nodes.c
Fix spelling errors.
[meshlink] / test / get-all-nodes.c
1 #ifdef NDEBUG
2 #undef NDEBUG
3 #endif
4
5 #include <stdio.h>
6 #include <unistd.h>
7 #include <stdlib.h>
8 #include <string.h>
9 #include <sys/time.h>
10 #include <time.h>
11 #include <assert.h>
12
13 #include "meshlink.h"
14 #include "utils.h"
15
16 struct sync_flag bar_reachable;
17
18 void status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
19         (void)mesh;
20
21         if(reachable && !strcmp(node->name, "bar")) {
22                 set_sync_flag(&bar_reachable, true);
23         }
24 }
25
26 int main() {
27         struct meshlink_node **nodes = NULL;
28         size_t nnodes = 0;
29
30         meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb);
31
32         // Open new meshlink instances.
33
34         assert(meshlink_destroy("get_all_nodes_conf.1"));
35         assert(meshlink_destroy("get_all_nodes_conf.2"));
36         assert(meshlink_destroy("get_all_nodes_conf.3"));
37
38         meshlink_handle_t *mesh[3];
39         mesh[0] = meshlink_open("get_all_nodes_conf.1", "foo", "import-export", DEV_CLASS_BACKBONE);
40         assert(mesh[0]);
41
42         mesh[1] = meshlink_open("get_all_nodes_conf.2", "bar", "import-export", DEV_CLASS_STATIONARY);
43         assert(mesh[1]);
44
45         mesh[2] = meshlink_open("get_all_nodes_conf.3", "baz", "get-all-nodes", DEV_CLASS_STATIONARY);
46         assert(mesh[2]);
47
48         // Check that we only know about ourself.
49
50         nodes = meshlink_get_all_nodes(mesh[0], nodes, &nnodes);
51         assert(nnodes == 1);
52         assert(nodes[0] == meshlink_get_self(mesh[0]));
53
54         // We should never have been online.
55
56         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
57         assert(nnodes == 0);
58
59         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
60         assert(nnodes == 1);
61         assert(nodes[0] == meshlink_get_self(mesh[0]));
62
63         // Let nodes know about each other.
64
65         for(int i = 0; i < 3; i++) {
66                 meshlink_enable_discovery(mesh[i], false);
67                 assert(meshlink_add_address(mesh[i], "localhost"));
68                 char *data = meshlink_export(mesh[i]);
69                 assert(data);
70
71                 for(int j = 0; j < 3; j++) {
72                         if(i == j) {
73                                 continue;
74                         }
75
76                         assert(meshlink_import(mesh[j], data));
77                 }
78         }
79
80         // We should know about all nodes now, and their device class.
81
82         nodes = meshlink_get_all_nodes(mesh[0], nodes, &nnodes);
83         assert(nnodes == 3);
84
85         nodes = meshlink_get_all_nodes_by_dev_class(mesh[0], DEV_CLASS_BACKBONE, nodes, &nnodes);
86         assert(nnodes == 1);
87         assert(nodes[0] == meshlink_get_self(mesh[0]));
88
89         nodes = meshlink_get_all_nodes_by_dev_class(mesh[0], DEV_CLASS_STATIONARY, nodes, &nnodes);
90         assert(nnodes == 2);
91
92         // But no node should have been online.
93
94         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
95         assert(nnodes == 0);
96
97         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
98         assert(nnodes == 3);
99
100         // Start foo.
101
102         time_t foo_started = time(NULL);
103         assert(meshlink_start(mesh[0]));
104
105         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
106         assert(nnodes == 1);
107         assert(nodes[0] == meshlink_get_self(mesh[0]));
108
109         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
110         assert(nnodes == 2);
111
112         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], foo_started - 1, -1, nodes, &nnodes);
113         assert(nnodes == 1);
114         assert(nodes[0] == meshlink_get_self(mesh[0]));
115
116         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started - 1, nodes, &nnodes);
117         assert(nnodes == 0);
118
119         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started + 1, nodes, &nnodes);
120         assert(nnodes == 1);
121         assert(nodes[0] == meshlink_get_self(mesh[0]));
122
123         // Start bar and wait for it to connect.
124
125         meshlink_set_node_status_cb(mesh[0], status_cb);
126
127         sleep(2);
128         assert(meshlink_start(mesh[1]));
129         assert(wait_sync_flag(&bar_reachable, 20));
130         time_t bar_started = time(NULL);
131
132         // Validate time ranges.
133
134         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
135         assert(nnodes == 2);
136
137         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
138         assert(nnodes == 1);
139         assert(nodes[0] == meshlink_get_node(mesh[0], "baz"));
140
141         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started + 1, nodes, &nnodes);
142         assert(nnodes == 1);
143         assert(nodes[0] == meshlink_get_self(mesh[0]));
144
145         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_started, bar_started, nodes, &nnodes);
146         assert(nnodes == 2);
147         assert(nodes[0] == meshlink_get_node(mesh[0], "bar"));
148         assert(nodes[1] == meshlink_get_self(mesh[0]));
149
150         // Stop bar.
151
152         meshlink_stop(mesh[1]);
153         sleep(2);
154         time_t bar_stopped = time(NULL);
155
156         // Validate we can see when bar was reachable.
157
158         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_stopped, bar_stopped, nodes, &nnodes);
159         assert(nnodes == 1);
160         assert(nodes[0] == meshlink_get_self(mesh[0]));
161
162         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_started, bar_started, nodes, &nnodes);
163         assert(nnodes == 2);
164         assert(nodes[0] == meshlink_get_node(mesh[0], "bar"));
165         assert(nodes[1] == meshlink_get_self(mesh[0]));
166
167         // Close and restart foo, check that it remembers correctly.
168
169         meshlink_close(mesh[0]);
170         sleep(2);
171         time_t foo_stopped = time(NULL);
172         mesh[0] = meshlink_open("get_all_nodes_conf.1", "foo", "import-export", DEV_CLASS_BACKBONE);
173         assert(mesh[0]);
174
175         nodes = meshlink_get_all_nodes(mesh[0], nodes, &nnodes);
176         assert(nnodes == 3);
177
178         nodes = meshlink_get_all_nodes_by_dev_class(mesh[0], DEV_CLASS_BACKBONE, nodes, &nnodes);
179         assert(nnodes == 1);
180         assert(nodes[0] == meshlink_get_self(mesh[0]));
181
182         nodes = meshlink_get_all_nodes_by_dev_class(mesh[0], DEV_CLASS_STATIONARY, nodes, &nnodes);
183         assert(nnodes == 2);
184
185         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, 0, nodes, &nnodes);
186         assert(nnodes == 1);
187         assert(nodes[0] == meshlink_get_node(mesh[0], "baz"));
188
189         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 0, -1, nodes, &nnodes);
190         assert(nnodes == 2);
191
192         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started - 1, nodes, &nnodes);
193         assert(nnodes == 0);
194
195         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], 1, foo_started + 1, nodes, &nnodes);
196         assert(nnodes == 1);
197         assert(nodes[0] == meshlink_get_self(mesh[0]));
198
199         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_started, bar_started, nodes, &nnodes);
200         assert(nnodes == 2);
201         assert(nodes[0] == meshlink_get_node(mesh[0], "bar"));
202         assert(nodes[1] == meshlink_get_self(mesh[0]));
203
204         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], bar_stopped, bar_stopped, nodes, &nnodes);
205         assert(nnodes == 1);
206         assert(nodes[0] == meshlink_get_self(mesh[0]));
207
208         nodes = meshlink_get_all_nodes_by_last_reachable(mesh[0], foo_stopped, -1, nodes, &nnodes);
209         assert(nnodes == 0);
210
211         // Clean up.
212
213         for(int i = 0; i < 3; i++) {
214                 meshlink_close(mesh[i]);
215         }
216 }