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