]> git.meshlink.io Git - meshlink/blob - test/trio.c
Add unit test cases for encrypted storage key rotation
[meshlink] / test / trio.c
1 #define _GNU_SOURCE
2
3 #include <stdio.h>
4 #include <string.h>
5 #include <unistd.h>
6 #include <errno.h>
7 #include <assert.h>
8 #include <sys/time.h>
9
10 #include "meshlink.h"
11 #include "devtools.h"
12 #include "utils.h"
13
14 static void log_cb(meshlink_handle_t *mesh, meshlink_log_level_t level, const char *text) {
15         (void)mesh;
16
17         static struct timeval tv0;
18         struct timeval tv;
19
20         if(tv0.tv_sec == 0) {
21                 gettimeofday(&tv0, NULL);
22         }
23
24         gettimeofday(&tv, NULL);
25         fprintf(stderr, "%u.%.03u ", (unsigned int)(tv.tv_sec - tv0.tv_sec), (unsigned int)tv.tv_usec / 1000);
26
27         if(mesh) {
28                 fprintf(stderr, "(%s) ", mesh->name);
29         }
30
31         fprintf(stderr, "[%d] %s\n", level, text);
32 }
33
34 static bool received = false;
35
36 static void receive_cb(meshlink_handle_t *mesh, meshlink_node_t *source, const void *data, size_t len) {
37         (void)mesh;
38         (void)source;
39
40         fprintf(stderr, "RECEIVED SOMETHING\n");
41
42         if(len == 5 && !memcmp(data, "Hello", 5)) {
43                 received = true;
44         }
45 }
46
47 int main() {
48         // Create three instances.
49
50         const char *name[3] = {"foo", "bar", "baz"};
51         meshlink_handle_t *mesh[3];
52         char *data[3];
53
54         for(int i = 0; i < 3; i++) {
55                 char *path;
56                 int ret_val;
57                 (void)ret_val;
58                 ret_val = asprintf(&path, "trio_conf.%d", i);
59                 assert(path);
60
61                 mesh[i] = meshlink_open(path, name[i], "trio", DEV_CLASS_BACKBONE);
62                 assert(mesh[i]);
63
64                 meshlink_add_address(mesh[i], "localhost");
65
66                 data[i] = meshlink_export(mesh[i]);
67                 assert(data[i]);
68         }
69
70         // first node knows the two other nodes
71
72         for(int i = 1; i < 3; i++) {
73                 assert(meshlink_import(mesh[i], data[0]));
74                 assert(meshlink_import(mesh[0], data[i]));
75
76                 assert(meshlink_get_node(mesh[i], name[0]));
77                 assert(meshlink_get_node(mesh[0], name[i]));
78         }
79
80         // second and third node should not know each other yet
81
82         assert(!meshlink_get_node(mesh[1], name[2]));
83         assert(!meshlink_get_node(mesh[2], name[1]));
84
85         // start the nodes
86
87         for(int i = 0; i < 3; i++) {
88                 assert(meshlink_start(mesh[i]));
89         }
90
91         // the nodes should now learn about each other
92
93         assert_after(meshlink_get_node(mesh[1], name[2]), 5);
94         assert_after(meshlink_get_node(mesh[2], name[1]), 5);
95
96         // Send a packet, expect it is received
97
98         meshlink_set_receive_cb(mesh[1], receive_cb);
99         assert_after((meshlink_send(mesh[2], meshlink_get_node(mesh[2], name[1]), "Hello", 5), received), 15);
100
101         // Check that the second and third node have autoconnected to each other
102
103         devtool_edge_t *edges = NULL;
104         size_t nedges = 0;
105         assert_after((edges = devtool_get_all_edges(mesh[1], edges, &nedges), nedges == 3), 15);
106
107         // Stop the first node
108
109         meshlink_stop(mesh[0]);
110         sleep(1);
111
112         // Communication should still be possible
113
114         assert_after((meshlink_send(mesh[2], meshlink_get_node(mesh[2], name[1]), "Hello", 5), received), 15);
115
116         // Stop the other nodes
117
118         for(int i = 1; i < 3; i++) {
119                 meshlink_stop(mesh[i]);
120         }
121
122         sleep(1);
123
124         // Start just the other two nodes
125
126         meshlink_set_log_cb(mesh[1], MESHLINK_DEBUG, log_cb);
127
128         for(int i = 1; i < 3; i++) {
129                 assert(meshlink_start(mesh[i]));
130         }
131
132         assert(meshlink_get_node(mesh[1], name[2]));
133         assert(meshlink_get_node(mesh[2], name[1]));
134
135         // Communication should still be possible
136
137         received = false;
138         assert_after((meshlink_send(mesh[2], meshlink_get_node(mesh[2], name[1]), "Hello", 5), received), 25);
139
140         // Clean up.
141
142         for(int i = 0; i < 3; i++) {
143                 meshlink_close(mesh[i]);
144         }
145 }