diff options
| author | Garrett D'Amore <garrett@damore.org> | 2019-12-28 10:55:13 -0800 |
|---|---|---|
| committer | Garrett D'Amore <garrett@damore.org> | 2019-12-28 12:41:31 -0800 |
| commit | 255a85d10f68a898c671f9387da61e0ba62a61c7 (patch) | |
| tree | 17b93ed72f14b023677234a5cb870efe4f9f4f35 /tests/wsstream.c | |
| parent | 50b5f060548987de6f5ac65f39fa0eecb597decb (diff) | |
| download | nng-255a85d10f68a898c671f9387da61e0ba62a61c7.tar.gz nng-255a85d10f68a898c671f9387da61e0ba62a61c7.tar.bz2 nng-255a85d10f68a898c671f9387da61e0ba62a61c7.zip | |
fixes #986 ws_read_finish_str free invalid pointer
Also, this has refactored the websocket stream test to the new
acutest.h, and includes a much deeper test of fragmentation and
reassembly of websocket streams.
Diffstat (limited to 'tests/wsstream.c')
| -rw-r--r-- | tests/wsstream.c | 204 |
1 files changed, 0 insertions, 204 deletions
diff --git a/tests/wsstream.c b/tests/wsstream.c deleted file mode 100644 index c51e2630..00000000 --- a/tests/wsstream.c +++ /dev/null @@ -1,204 +0,0 @@ -// -// Copyright 2019 Staysail Systems, Inc. <info@staysail.tech> -// Copyright 2018 Capitar IT Group BV <info@capitar.com> -// -// 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 <string.h> - -#include <nng/nng.h> - -#include "convey.h" -#include "stubs.h" - -TestMain("Websocket STREAM", { - atexit(nng_fini); - Convey("We can create a dialer and listener", { - nng_stream_dialer * d = NULL; - nng_stream_listener *l = NULL; - Reset({ - nng_stream_listener_free(l); - nng_stream_dialer_free(d); - l = NULL; - d = NULL; - }); - Convey("Listener listens (wildcard)", { - nng_sockaddr sa; - size_t sz; - uint8_t ip[4]; - - So(nng_stream_listener_alloc( - &l, "ws://127.0.0.1:0/test") == 0); - So(nng_stream_listener_listen(l) == 0); - - sz = sizeof(sa); - ip[0] = 127; - ip[1] = 0; - ip[2] = 0; - ip[3] = 1; - So(nng_stream_listener_get( - l, NNG_OPT_LOCADDR, &sa, &sz) == 0); - So(sz == sizeof(sa)); - So(sa.s_in.sa_port != 0); - So(memcmp(&sa.s_in.sa_addr, ip, 4) == 0); - - Convey("We can dial it", { - nng_aio * daio = NULL; - nng_aio * laio = NULL; - nng_aio * maio = NULL; - nng_stream *c1 = NULL; - nng_stream *c2 = NULL; - - char uri[64]; - snprintf(uri, sizeof(uri), - "ws://127.0.0.1:%d/test", - test_htons(sa.s_in.sa_port)); - - So(nng_stream_dialer_alloc(&d, uri) == 0); - So(nng_aio_alloc(&daio, NULL, NULL) == 0); - So(nng_aio_alloc(&laio, NULL, NULL) == 0); - So(nng_aio_alloc(&maio, NULL, NULL) == 0); - - Reset({ - nng_aio_free(daio); - nng_aio_free(laio); - nng_aio_free(maio); - if (c1 != NULL) { - nng_stream_close(c1); - nng_stream_free(c1); - } - if (c2 != NULL) { - nng_stream_close(c2); - nng_stream_free(c2); - } - }); - - nng_stream_dialer_dial(d, daio); - nng_stream_listener_accept(l, laio); - - nng_aio_wait(daio); - So(nng_aio_result(daio) == 0); - nng_aio_wait(laio); - So(nng_aio_result(laio) == 0); - - So(nng_aio_result(daio) == 0); - So(nng_aio_result(laio) == 0); - - c1 = nng_aio_get_output(daio, 0); - c2 = nng_aio_get_output(laio, 0); - So(c1 != NULL); - So(c2 != NULL); - - Convey("They exchange messages", { - nng_aio * aio1; - nng_aio * aio2; - nng_iov iov; - nng_sockaddr sa2; - char buf1[5]; - char buf2[5]; - bool on; - size_t sz; - - So(nng_aio_alloc(&aio1, NULL, NULL) == - 0); - So(nng_aio_alloc(&aio2, NULL, NULL) == - 0); - - Reset({ - nng_aio_free(aio1); - nng_aio_free(aio2); - }); - - on = true; - So(nng_stream_set(c1, - NNG_OPT_TCP_NODELAY, &on, - sizeof(on)) == 0); - So(nng_stream_set(c2, - NNG_OPT_TCP_NODELAY, &on, - sizeof(on)) == 0); - - So(nng_stream_set(c1, - NNG_OPT_TCP_KEEPALIVE, &on, - sizeof(on)) == 0); - - on = false; - sz = sizeof(on); - So(nng_stream_get(c1, - NNG_OPT_TCP_NODELAY, &on, - &sz) == 0); - So(sz == sizeof(on)); - So(on == true); - - on = false; - sz = sizeof(on); - So(nng_stream_get(c1, - NNG_OPT_TCP_KEEPALIVE, &on, - &sz) == 0); - So(sz == sizeof(on)); - So(on == true); - - // 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); - - So(nng_aio_result(aio1) == 0); - So(nng_aio_count(aio1) == 5); - - So(nng_aio_result(aio2) == 0); - So(nng_aio_count(aio2) == 5); - - So(memcmp(buf1, buf2, 5) == 0); - - Convey("Socket name matches", { - sz = sizeof(sa2); - So(nng_stream_get(c2, - NNG_OPT_LOCADDR, &sa2, - &sz) == 0); - So(sz == sizeof(sa2)); - So(sa2.s_in.sa_family == - NNG_AF_INET); - - So(sa2.s_in.sa_addr == - sa.s_in.sa_addr); - So(sa2.s_in.sa_port == - sa.s_in.sa_port); - }); - - Convey("Peer name matches", { - sz = sizeof(sa2); - So(nng_stream_get(c1, - NNG_OPT_REMADDR, &sa2, - &sz) == 0); - So(sz == sizeof(sa2)); - So(sa2.s_in.sa_family == - NNG_AF_INET); - So(sa2.s_in.sa_addr == - sa.s_in.sa_addr); - So(sa2.s_in.sa_port == - sa.s_in.sa_port); - }); - }); - }); - }); - }); -}) |
