From 0bd34114fded68765ac665ba2f8b99c8f5289bb3 Mon Sep 17 00:00:00 2001 From: Garrett D'Amore Date: Sat, 30 Nov 2024 23:14:22 -0500 Subject: tests: convert tcpsupp test to NUTS Also renamed this to tcp_stream_test. --- src/platform/CMakeLists.txt | 1 + src/platform/tcp_stream_test.c | 136 +++++++++++++++++++++++++++++++++++++++++ 2 files changed, 137 insertions(+) create mode 100644 src/platform/tcp_stream_test.c (limited to 'src') diff --git a/src/platform/CMakeLists.txt b/src/platform/CMakeLists.txt index 6a49d1fe..ce0cb0b7 100644 --- a/src/platform/CMakeLists.txt +++ b/src/platform/CMakeLists.txt @@ -15,4 +15,5 @@ add_subdirectory(windows) nng_test(platform_test) nng_test(resolver_test) +nng_test(tcp_stream_test) nng_test(udp_test) diff --git a/src/platform/tcp_stream_test.c b/src/platform/tcp_stream_test.c new file mode 100644 index 00000000..c1353e8a --- /dev/null +++ b/src/platform/tcp_stream_test.c @@ -0,0 +1,136 @@ +// +// Copyright 2024 Staysail Systems, Inc. +// Copyright 2018 Capitar IT Group BV +// +// This software is supplied under the terms of the MIT License, a +// copy of which should be located in the distribution where this +// file was obtained (LICENSE.txt). A copy of the license may also be +// found online at https://opensource.org/licenses/MIT. +// + +#include + +#include + +#include + +void +test_tcp_stream(void) +{ + nng_stream_dialer *d = NULL; + nng_stream_listener *l = NULL; + nng_sockaddr sa; + uint8_t ip[4]; + nng_aio *daio = NULL; + nng_aio *laio = NULL; + nng_aio *maio = NULL; + nng_stream *c1 = NULL; + nng_stream *c2 = NULL; + nng_aio *aio1; + nng_aio *aio2; + nng_iov iov; + nng_sockaddr sa2; + char buf1[5]; + char buf2[5]; + bool on; + + NUTS_PASS(nng_aio_alloc(&daio, NULL, NULL)); + NUTS_PASS(nng_aio_alloc(&laio, NULL, NULL)); + NUTS_PASS(nng_aio_alloc(&maio, NULL, NULL)); + NUTS_PASS(nng_aio_alloc(&aio1, NULL, NULL)); + NUTS_PASS(nng_aio_alloc(&aio2, NULL, NULL)); + + NUTS_PASS(nng_stream_listener_alloc(&l, "tcp://127.0.0.1")); + NUTS_PASS(nng_stream_listener_listen(l)); + + ip[0] = 127; + ip[1] = 0; + ip[2] = 0; + ip[3] = 1; + NUTS_PASS(nng_stream_listener_get_addr(l, NNG_OPT_LOCADDR, &sa)); + NUTS_TRUE(sa.s_in.sa_port != 0); + NUTS_TRUE(memcmp(&sa.s_in.sa_addr, ip, 4) == 0); + + char uri[64]; + snprintf(uri, sizeof(uri), "tcp://127.0.0.1:%d", + nuts_be16(sa.s_in.sa_port)); + + NUTS_PASS(nng_stream_dialer_alloc(&d, uri)); + nng_stream_dialer_dial(d, daio); + nng_stream_listener_accept(l, laio); + + nng_aio_wait(daio); + NUTS_PASS(nng_aio_result(daio)); + nng_aio_wait(laio); + NUTS_PASS(nng_aio_result(laio)); + + c1 = nng_aio_get_output(daio, 0); + c2 = nng_aio_get_output(laio, 0); + NUTS_TRUE(c1 != NULL); + NUTS_TRUE(c2 != NULL); + + on = false; + NUTS_PASS(nng_stream_get_bool(c1, NNG_OPT_TCP_NODELAY, &on)); + NUTS_TRUE(on); + + on = false; + NUTS_PASS(nng_stream_get_bool(c1, NNG_OPT_TCP_KEEPALIVE, &on)); + + // This relies on send completing for + // for just 5 bytes, and on recv doing + // the same. Technically this isn't + // guaranteed, but it would be weird + // to split such a small payload. + memcpy(buf1, "TEST", 5); + memset(buf2, 0, 5); + iov.iov_buf = buf1; + iov.iov_len = 5; + + nng_aio_set_iov(aio1, 1, &iov); + + iov.iov_buf = buf2; + iov.iov_len = 5; + nng_aio_set_iov(aio2, 1, &iov); + nng_stream_send(c1, aio1); + nng_stream_recv(c2, aio2); + nng_aio_wait(aio1); + nng_aio_wait(aio2); + + NUTS_PASS(nng_aio_result(aio1)); + NUTS_TRUE(nng_aio_count(aio1) == 5); + + NUTS_PASS(nng_aio_result(aio2)); + NUTS_TRUE(nng_aio_count(aio2) == 5); + + NUTS_TRUE(memcmp(buf1, buf2, 5) == 0); + + NUTS_PASS(nng_stream_get_addr(c2, NNG_OPT_LOCADDR, &sa2)); + NUTS_TRUE(sa2.s_in.sa_family == NNG_AF_INET); + + NUTS_TRUE(sa2.s_in.sa_addr == sa.s_in.sa_addr); + NUTS_TRUE(sa2.s_in.sa_port == sa.s_in.sa_port); + + NUTS_PASS(nng_stream_get_addr(c1, NNG_OPT_REMADDR, &sa2)); + NUTS_TRUE(sa2.s_in.sa_family == NNG_AF_INET); + NUTS_TRUE(sa2.s_in.sa_addr == sa.s_in.sa_addr); + NUTS_TRUE(sa2.s_in.sa_port == sa.s_in.sa_port); + + nng_stream_listener_close(l); + nng_stream_listener_free(l); + nng_stream_dialer_close(d); + nng_stream_dialer_free(d); + nng_aio_free(aio1); + nng_aio_free(aio2); + nng_aio_free(daio); + nng_aio_free(laio); + nng_aio_free(maio); + nng_stream_close(c1); + nng_stream_free(c1); + nng_stream_close(c2); + nng_stream_free(c2); +} + +NUTS_TESTS = { + { "tcp stream", test_tcp_stream }, + { NULL, NULL }, +}; -- cgit v1.2.3-70-g09d2