X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=test%2Fchannels.c;h=52d80a9ce1f40579a45e59f4149a890fe40d822f;hb=f73f697142bd5ce32d9b7273ed73ea798594854d;hp=c6d9fa8fd71299eb07d21d7fb263a4979d2ea6d8;hpb=196806f90f40fcc0cf727abd4bed6bc5aefff5ff;p=meshlink diff --git a/test/channels.c b/test/channels.c index c6d9fa8f..52d80a9c 100644 --- a/test/channels.c +++ b/test/channels.c @@ -5,7 +5,6 @@ #include #include "../src/meshlink.h" -#include "../src/node.h" volatile bool bar_reachable = false; volatile bool bar_responded = false; @@ -14,79 +13,114 @@ void log_cb(meshlink_handle_t *mesh, meshlink_log_level_t level, const char *tex static struct timeval tv0; struct timeval tv; - if(tv0.tv_sec == 0) + if(tv0.tv_sec == 0) { gettimeofday(&tv0, NULL); + } + gettimeofday(&tv, NULL); - fprintf(stderr, "%u.%.03u ", (unsigned int)(tv.tv_sec-tv0.tv_sec), (unsigned int)tv.tv_usec/1000); + fprintf(stderr, "%u.%.03u ", (unsigned int)(tv.tv_sec - tv0.tv_sec), (unsigned int)tv.tv_usec / 1000); - if(mesh) + if(mesh) { fprintf(stderr, "(%s) ", mesh->name); + } + fprintf(stderr, "[%d] %s\n", level, text); } void status_cb(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) { - printf("status_cb: %s %sreachable\n", node->name, reachable?"":"un"); - if(!strcmp(node->name, "bar")) + (void)mesh; + + printf("status_cb: %s %sreachable\n", node->name, reachable ? "" : "un"); + + if(!strcmp(node->name, "bar")) { bar_reachable = reachable; + } } void foo_receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) { + (void)mesh; + (void)channel; + printf("foo_receive_cb %zu: ", len); fwrite(data, 1, len, stdout); printf("\n"); - if(len == 5 && !memcmp(data, "Hello", 5)) + + if(len == 5 && !memcmp(data, "Hello", 5)) { bar_responded = true; + } } void bar_receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) { printf("bar_receive_cb %zu: ", len); fwrite(data, 1, len, stdout); + printf("\n"); // Echo the data back. meshlink_channel_send(mesh, channel, data, len); } bool reject_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, uint16_t port, const void *data, size_t len) { + (void)mesh; + (void)channel; + (void)port; + (void)data; + (void)len; + return false; } bool accept_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, uint16_t port, const void *data, size_t len) { printf("accept_cb: (from %s on port %u) ", channel->node->name, (unsigned int)port); + if(data) { fwrite(data, 1, len, stdout); - printf("\n"); } - if(port != 7) + printf("\n"); + + if(port != 7) { return false; + } + meshlink_set_channel_receive_cb(mesh, channel, bar_receive_cb); - if(data) + + if(data) { bar_receive_cb(mesh, channel, data, len); + } + return true; } void poll_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, size_t len) { + (void)len; + meshlink_set_channel_poll_cb(mesh, channel, NULL); - if(meshlink_channel_send(mesh, channel, "Hello", 5) != 5) + + if(meshlink_channel_send(mesh, channel, "Hello", 5) != 5) { fprintf(stderr, "Could not send whole message\n"); + } } -int main(int argc, char *argv[]) { +int main() { meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb); // Open two new meshlink instance. meshlink_handle_t *mesh1 = meshlink_open("channels_conf.1", "foo", "channels", DEV_CLASS_BACKBONE); + if(!mesh1) { fprintf(stderr, "Could not initialize configuration for foo\n"); return 1; } meshlink_handle_t *mesh2 = meshlink_open("channels_conf.2", "bar", "channels", DEV_CLASS_BACKBONE); + if(!mesh2) { fprintf(stderr, "Could not initialize configuration for bar\n"); return 1; } + meshlink_enable_discovery(mesh1, false); + meshlink_enable_discovery(mesh2, false); meshlink_set_log_cb(mesh1, MESHLINK_DEBUG, log_cb); meshlink_set_log_cb(mesh2, MESHLINK_DEBUG, log_cb); @@ -95,6 +129,7 @@ int main(int argc, char *argv[]) { meshlink_add_address(mesh1, "localhost"); char *data = meshlink_export(mesh1); + if(!data) { fprintf(stderr, "Foo could not export its configuration\n"); return 1; @@ -108,6 +143,7 @@ int main(int argc, char *argv[]) { free(data); data = meshlink_export(mesh2); + if(!data) { fprintf(stderr, "Bar could not export its configuration\n"); return 1; @@ -143,8 +179,10 @@ int main(int argc, char *argv[]) { for(int i = 0; i < 20; i++) { sleep(1); - if(bar_reachable) + + if(bar_reachable) { break; + } } if(!bar_reachable) { @@ -155,29 +193,21 @@ int main(int argc, char *argv[]) { // Open a channel from foo to bar. meshlink_node_t *bar = meshlink_get_node(mesh1, "bar"); + if(!bar) { fprintf(stderr, "Foo could not find bar\n"); return 1; } - // XXX not enough to wait for reachable, must wait for SPTPS to complete - for(int i=0; i < 20; i++) { - sleep(1); - if(((node_t *)bar)->status.validkey) - break; - } - if(!((node_t *)bar)->status.validkey) { - fprintf(stderr, "No key exchange after 20 seconds\n"); - return 1; - } - meshlink_channel_t *channel = meshlink_channel_open(mesh1, bar, 7, foo_receive_cb, NULL, 0); meshlink_set_channel_poll_cb(mesh1, channel, poll_cb); - for(int i = 0; i < 5; i++) { + for(int i = 0; i < 20; i++) { sleep(1); - if(bar_responded) + + if(bar_responded) { break; + } } if(!bar_responded) {