15 #include "../src/meshlink-tiny.h"
18 static struct sync_flag b_responded;
19 static struct sync_flag b_closed;
20 static size_t a_poll_cb_len;
22 static void a_receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) {
26 if(len == 5 && !memcmp(data, "Hello", 5)) {
27 set_sync_flag(&b_responded, true);
29 set_sync_flag(&b_closed, true);
30 set_sync_flag(channel->priv, true);
34 static void b_receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) {
35 // Send one message back, then close the channel.
37 assert(meshlink_channel_send(mesh, channel, data, len) == (ssize_t)len);
40 meshlink_channel_close(mesh, channel);
43 static bool accept_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, uint16_t port, const void *data, size_t len) {
46 meshlink_set_channel_accept_cb(mesh, NULL);
47 meshlink_set_channel_receive_cb(mesh, channel, b_receive_cb);
50 b_receive_cb(mesh, channel, data, len);
56 static void poll_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, size_t len) {
59 meshlink_set_channel_poll_cb(mesh, channel, NULL);
60 set_sync_flag(channel->priv, true);
63 static void poll_cb2(meshlink_handle_t *mesh, meshlink_channel_t *channel, size_t len) {
68 meshlink_set_channel_poll_cb(mesh, channel, NULL);
69 set_sync_flag(channel->priv, true);
73 meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb);
74 init_sync_flag(&b_responded);
75 init_sync_flag(&b_closed);
77 meshlink_handle_t *a, *b;
78 open_meshlink_pair(&a, &b, "channels-udp-cornercases");
82 meshlink_set_channel_accept_cb(b, accept_cb);
84 // Open a channel from a to b before starting the mesh.
86 meshlink_node_t *nb = meshlink_get_node(a, "b");
89 struct sync_flag channel_opened;
90 init_sync_flag(&channel_opened);
92 meshlink_channel_t *channel = meshlink_channel_open_ex(a, nb, 7, a_receive_cb, &channel_opened, 0, MESHLINK_CHANNEL_UDP);
95 meshlink_set_channel_poll_cb(a, channel, poll_cb);
97 // Check that the channel isn't established yet and sending a packet at this point returns 0
98 assert(meshlink_channel_send(a, channel, "test", 4) == 0);
99 assert(wait_sync_flag(&channel_opened, 1) == false);
101 // Start MeshLink and wait for the channel to become connected.
102 start_meshlink_pair(a, b);
104 assert(wait_sync_flag(&channel_opened, 15));
106 // Re-initialize everything
107 meshlink_channel_close(a, channel);
108 close_meshlink_pair(a, b);
109 reset_sync_flag(&channel_opened);
110 reset_sync_flag(&b_responded);
111 reset_sync_flag(&b_closed);
112 open_meshlink_pair(&a, &b, "channels-udp-cornercases");
114 meshlink_set_channel_accept_cb(b, accept_cb);
116 start_meshlink_pair(a, b);
118 // Create a channel to b
119 nb = meshlink_get_node(a, "b");
122 channel = meshlink_channel_open_ex(a, nb, 7, a_receive_cb, &channel_opened, 0, MESHLINK_CHANNEL_UDP);
124 meshlink_set_channel_poll_cb(a, channel, poll_cb);
126 assert(wait_sync_flag(&channel_opened, 15));
128 // Send a message to b
130 struct sync_flag channel_closed;
131 init_sync_flag(&channel_closed);
132 channel->priv = &channel_closed;
134 for(int i = 0; i < 10; i++) {
135 assert(meshlink_channel_send(a, channel, "Hello", 5) == 5);
137 if(wait_sync_flag(&channel_closed, 1)) {
142 assert(wait_sync_flag(&channel_closed, 1));
144 wait_sync_flag(&b_responded, 1);
145 wait_sync_flag(&b_closed, 1);
147 // Try to send data on a closed channel
149 for(int i = 0; i < 10; i++) {
150 if(meshlink_channel_send(a, channel, "Hello", 5) == -1) {
158 // Try to create a second channel
160 struct sync_flag channel_polled;
161 init_sync_flag(&channel_polled);
163 meshlink_channel_t *channel2 = meshlink_channel_open_ex(a, nb, 7, a_receive_cb, &channel_polled, 0, MESHLINK_CHANNEL_UDP);
165 meshlink_set_channel_poll_cb(a, channel2, poll_cb2);
167 assert(wait_sync_flag(&channel_polled, 5));
169 assert(0 == a_poll_cb_len);
171 meshlink_channel_close(a, channel);
172 meshlink_channel_close(a, channel2);
173 close_meshlink_pair(a, b);