]> git.meshlink.io Git - meshlink/blob - test/channels.c
Fix compiling with GCC 10.
[meshlink] / test / channels.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 <assert.h>
11
12 #include "utils.h"
13 #include "../src/meshlink.h"
14
15 static struct sync_flag b_responded;
16
17 static void a_receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) {
18         (void)mesh;
19         (void)channel;
20
21         printf("a_receive_cb %zu: ", len);
22         fwrite(data, 1, len, stdout);
23         printf("\n");
24
25         if(len == 5 && !memcmp(data, "Hello", 5)) {
26                 set_sync_flag(&b_responded, true);
27         }
28 }
29
30 static void b_receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) {
31         printf("b_receive_cb %zu: ", len);
32         fwrite(data, 1, len, stdout);
33         printf("\n");
34         // Echo the data back.
35         assert(meshlink_channel_send(mesh, channel, data, len) == (ssize_t)len);
36 }
37
38 static bool reject_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, uint16_t port, const void *data, size_t len) {
39         (void)mesh;
40         (void)channel;
41         (void)port;
42         (void)data;
43         (void)len;
44
45         return false;
46 }
47
48 static bool accept_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, uint16_t port, const void *data, size_t len) {
49         printf("accept_cb: (from %s on port %u) ", channel->node->name, (unsigned int)port);
50
51         if(data) {
52                 fwrite(data, 1, len, stdout);
53         }
54
55         printf("\n");
56
57         if(port != 7) {
58                 return false;
59         }
60
61         meshlink_set_channel_receive_cb(mesh, channel, b_receive_cb);
62
63         if(data) {
64                 b_receive_cb(mesh, channel, data, len);
65         }
66
67         return true;
68 }
69
70 static void poll_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, size_t len) {
71         (void)len;
72
73         meshlink_set_channel_poll_cb(mesh, channel, NULL);
74
75         assert(meshlink_channel_send(mesh, channel, "Hello", 5) == 5);
76 }
77
78 int main() {
79         meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb);
80
81         // Open two new meshlink instance.
82
83         meshlink_handle_t *mesh_a, *mesh_b;
84         open_meshlink_pair(&mesh_a, &mesh_b, "channels");
85
86         // Set the callbacks.
87
88         meshlink_set_channel_accept_cb(mesh_a, reject_cb);
89         meshlink_set_channel_accept_cb(mesh_b, accept_cb);
90
91         // Start both instances
92
93         start_meshlink_pair(mesh_a, mesh_b);
94
95         // Open a channel from a to b.
96
97         meshlink_node_t *b = meshlink_get_node(mesh_a, "b");
98         assert(b);
99
100         meshlink_channel_t *channel = meshlink_channel_open(mesh_a, b, 7, a_receive_cb, NULL, 0);
101         assert(channel);
102
103         meshlink_set_channel_poll_cb(mesh_a, channel, poll_cb);
104         assert(wait_sync_flag(&b_responded, 20));
105
106         meshlink_channel_close(mesh_a, channel);
107
108         // Clean up.
109
110         close_meshlink_pair(mesh_a, mesh_b);
111 }