From 14fc467036dd90393e6f436f85b57972a86f83bb Mon Sep 17 00:00:00 2001 From: Guus Sliepen Date: Sat, 26 Feb 2022 00:55:15 +0100 Subject: [PATCH] Test multiple UDP and TCP channels to the same port on the same node. This tests that we correctly demultiplex multiple channels with the same destination node and port combination. --- test/Makefile.am | 5 ++ test/channels-mixed.c | 170 ++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 175 insertions(+) create mode 100644 test/channels-mixed.c diff --git a/test/Makefile.am b/test/Makefile.am index 688d8b7d..a0b5dc12 100644 --- a/test/Makefile.am +++ b/test/Makefile.am @@ -11,6 +11,7 @@ TESTS = \ channels-cornercases \ channels-failure \ channels-fork \ + channels-mixed \ channels-no-partial \ channels-udp \ channels-udp-cornercases \ @@ -58,6 +59,7 @@ check_PROGRAMS = \ channels-cornercases \ channels-failure \ channels-fork \ + channels-mixed \ channels-no-partial \ channels-udp \ channels-udp-cornercases \ @@ -122,6 +124,9 @@ channels_failure_LDADD = $(top_builddir)/src/libmeshlink.la channels_fork_SOURCES = channels-fork.c utils.c utils.h channels_fork_LDADD = $(top_builddir)/src/libmeshlink.la +channels_mixed_SOURCES = channels-mixed.c utils.c utils.h +channels_mixed_LDADD = $(top_builddir)/src/libmeshlink.la + channels_cornercases_SOURCES = channels-cornercases.c utils.c utils.h channels_cornercases_LDADD = $(top_builddir)/src/libmeshlink.la diff --git a/test/channels-mixed.c b/test/channels-mixed.c new file mode 100644 index 00000000..32dbdefd --- /dev/null +++ b/test/channels-mixed.c @@ -0,0 +1,170 @@ +#ifdef NDEBUG +#undef NDEBUG +#endif + +#include +#include +#include +#include +#include +#include +#include + +#include "../src/meshlink.h" +#include "utils.h" + +#define SMALL_SIZE 512 +#define SMALL_COUNT 2500 +#define LARGE_SIZE 131072 +#define NCHANNELS 10 +#define PORT 123 + +static struct channel { + meshlink_channel_t *channel; + bool udp; + struct sync_flag open_flag; + struct sync_flag close_flag; +} channels[NCHANNELS]; + +static size_t received[NCHANNELS]; + +static struct sync_flag pmtu_flag; + +static void a_pmtu_cb(meshlink_handle_t *mesh, meshlink_node_t *node, uint16_t pmtu) { + assert(mesh); + + if(pmtu >= SMALL_SIZE && !strcmp(node->name, "b")) { + set_sync_flag(&pmtu_flag, true); + } +} + +static void a_receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) { + struct channel *info = channel->priv; + assert(info); + + if(!data && !len) { + set_sync_flag(&info->close_flag, true); + meshlink_channel_close(mesh, channel); + return; + } +} + +static void aio_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len, void *priv) { + (void)data; + (void)len; + (void)priv; + + meshlink_channel_shutdown(mesh, channel, SHUT_WR); +} + +static void b_receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) { + if(!data && !len) { + meshlink_channel_close(mesh, channel); + return; + } + + assert(data && len); + size_t *rx = channel->priv; + *rx += len; +} + +static bool b_accept_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, uint16_t port, const void *data, size_t len) { + (void)data; + (void)len; + + assert(!data); + assert(!len); + assert(port == PORT); + + static int i; + channel->priv = &received[i++]; + + meshlink_set_channel_receive_cb(mesh, channel, b_receive_cb); + return true; +} + +static void a_poll_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, size_t len) { + assert(len); + + meshlink_set_channel_poll_cb(mesh, channel, NULL); + + struct channel *info = channel->priv; + assert(info); + + set_sync_flag(&info->open_flag, true); +} + +int main(void) { + meshlink_set_log_cb(NULL, MESHLINK_WARNING, log_cb); + + meshlink_handle_t *a, *b; + open_meshlink_pair(&a, &b, "channels-mixed"); + + meshlink_set_channel_accept_cb(b, b_accept_cb); + meshlink_set_node_pmtu_cb(a, a_pmtu_cb); + start_meshlink_pair(a, b); + + // Create a number of TCP and UDP channels + + meshlink_node_t *nb = meshlink_get_node(a, "b"); + + for(int i = 0; i < NCHANNELS; i++) { + channels[i].udp = i % 2 == 1; + init_sync_flag(&channels[i].open_flag); + init_sync_flag(&channels[i].close_flag); + channels[i].channel = meshlink_channel_open_ex(a, nb, PORT, a_receive_cb, &channels[i], 0, channels[i].udp ? MESHLINK_CHANNEL_UDP : MESHLINK_CHANNEL_TCP); + meshlink_set_channel_poll_cb(a, channels[i].channel, a_poll_cb); + assert(channels[i].channel); + } + + // Wait for all channels to connect + + for(int i = 0; i < NCHANNELS; i++) { + assert(wait_sync_flag(&channels[i].open_flag, 10)); + } + + // Wait for PMTU to finish + + assert(wait_sync_flag(&pmtu_flag, 10)); + + // Send data on all channels + + size_t size = SMALL_SIZE * SMALL_COUNT; + char *data = malloc(size); + memset(data, 'U', size); + + for(int i = 0; i < NCHANNELS; i++) { + assert(meshlink_channel_aio_send(a, channels[i].channel, data, size, aio_cb, &channels[i])); + } + + // Wait for the other end to close the channels + + for(int i = 0; i < NCHANNELS; i++) { + assert(wait_sync_flag(&channels[i].close_flag, 10)); + } + + // Check that most of the data has been transmitted + + for(int i = 0; i < NCHANNELS; i++) { + fprintf(stderr, "%d received %zu\n", i, received[i]); + } + + int received_all = 0; + + for(int i = 0; i < NCHANNELS; i++) { + assert(received[i] >= size / 2); + assert(received[i] <= size); + + if(received[i] == size) { + received_all++; + } + } + + assert(received_all >= NCHANNELS / 2); + + // Done + + close_meshlink_pair(a, b); + + return 0; +} -- 2.39.5