diff options
Diffstat (limited to 'src/transport/ws')
| -rw-r--r-- | src/transport/ws/CMakeLists.txt | 24 | ||||
| -rw-r--r-- | src/transport/ws/README.adoc | 38 | ||||
| -rw-r--r-- | src/transport/ws/websocket.c | 740 | ||||
| -rw-r--r-- | src/transport/ws/ws_test.c | 181 |
4 files changed, 0 insertions, 983 deletions
diff --git a/src/transport/ws/CMakeLists.txt b/src/transport/ws/CMakeLists.txt deleted file mode 100644 index 6e409b43..00000000 --- a/src/transport/ws/CMakeLists.txt +++ /dev/null @@ -1,24 +0,0 @@ -# -# Copyright 2020 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. -# - -# WebSocket transport -nng_directory(ws) - -if (NNG_TRANSPORT_WS OR NNG_TRANSPORT_WSS) - set(WS_ON ON) -endif() - -nng_defines_if(NNG_TRANSPORT_WS NNG_TRANSPORT_WS) -nng_defines_if(NNG_TRANSPORT_WSS NNG_TRANSPORT_WSS) -nng_sources_if(WS_ON websocket.c) -nng_headers_if(WS_ON nng/transport/ws/websocket.h) -nng_test_if(WS_ON ws_test) - - diff --git a/src/transport/ws/README.adoc b/src/transport/ws/README.adoc deleted file mode 100644 index e3101297..00000000 --- a/src/transport/ws/README.adoc +++ /dev/null @@ -1,38 +0,0 @@ -= websocket transport - -This transport provides support for SP over websocket using TCP or TLS. -When using TCP, it is compatible with the libnanomsg legacy transport. -It also is compatible with mangos (both TCP and TLS). - -TLS support requires the mbedTLS library. - -We set the "protocol" such as "pair.sp.nanomsg.org" in the -Sec-WebSocket-Protocol field -- the client sets to the the server's -protocol - i.e. the protocol that the server speaks. For example, -if the the server is a REP, then a REQ client would send "rep.sp.nanomsg.org". - -The server sends the same value (it's own), per the WebSocket specs. (Note -that the client's protocol is never sent, but assumed to be complementary -to the protocol in the Sec-WebSocket-Protocol field.) - -Each SP message is a WebSocket message. - -WebSocket is defined in RFC 6455. - -== Design - -We unfortunately need to implement our own design for this -- the only -reasonable client library would be libcurl, and there is a dearth of -suitable server libraries. Since we don't have to support full HTTP, but -just the initial handshake, this isn't too tragic. - -== Multiple Server Sockets - -In order to support Multiple Server sockets listening on the same port, -the application must be long lived. We will set up a listener on the -configured TCP (or TLS) port, and examine the PATH supplied in the GET. -This will be used to match against the URL requested, and if the URL -matches we will create the appropriate pipe. - -If no server endpoint at that address can be found, we return an -HTTP error, and close the socket. diff --git a/src/transport/ws/websocket.c b/src/transport/ws/websocket.c deleted file mode 100644 index 3f73f47f..00000000 --- a/src/transport/ws/websocket.c +++ /dev/null @@ -1,740 +0,0 @@ -// -// Copyright 2020 Staysail Systems, Inc. <info@staysail.tech> -// Copyright 2018 Capitar IT Group BV <info@capitar.com> -// Copyright 2019 Devolutions <info@devolutions.net> -// -// 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 <stdbool.h> -#include <stdio.h> -#include <string.h> - -#include "core/nng_impl.h" -#include "supplemental/websocket/websocket.h" - -#include <nng/supplemental/tls/tls.h> -#include <nng/transport/ws/websocket.h> - -typedef struct ws_dialer ws_dialer; -typedef struct ws_listener ws_listener; -typedef struct ws_pipe ws_pipe; - -struct ws_dialer { - uint16_t peer; // remote protocol - nni_list aios; - nni_mtx mtx; - nni_aio * connaio; - nng_stream_dialer *dialer; - bool started; -}; - -struct ws_listener { - uint16_t peer; // remote protocol - nni_list aios; - nni_mtx mtx; - nni_aio * accaio; - nng_stream_listener *listener; - bool started; -}; - -struct ws_pipe { - nni_mtx mtx; - bool closed; - uint16_t peer; - nni_aio * user_txaio; - nni_aio * user_rxaio; - nni_aio * txaio; - nni_aio * rxaio; - nng_stream *ws; -}; - -static void -wstran_pipe_send_cb(void *arg) -{ - ws_pipe *p = arg; - nni_aio *taio; - nni_aio *uaio; - - nni_mtx_lock(&p->mtx); - taio = p->txaio; - uaio = p->user_txaio; - p->user_txaio = NULL; - - if (uaio != NULL) { - int rv; - if ((rv = nni_aio_result(taio)) != 0) { - nni_aio_finish_error(uaio, rv); - } else { - nni_aio_finish(uaio, 0, 0); - } - } - nni_mtx_unlock(&p->mtx); -} - -static void -wstran_pipe_recv_cb(void *arg) -{ - ws_pipe *p = arg; - nni_aio *raio = p->rxaio; - nni_aio *uaio; - int rv; - - nni_mtx_lock(&p->mtx); - uaio = p->user_rxaio; - p->user_rxaio = NULL; - if ((rv = nni_aio_result(raio)) != 0) { - if (uaio != NULL) { - nni_aio_finish_error(uaio, rv); - } - } else { - nni_msg *msg = nni_aio_get_msg(raio); - if (uaio != NULL) { - nni_aio_finish_msg(uaio, msg); - } else { - nni_msg_free(msg); - } - } - nni_mtx_unlock(&p->mtx); -} - -static void -wstran_pipe_recv_cancel(nni_aio *aio, void *arg, int rv) -{ - ws_pipe *p = arg; - nni_mtx_lock(&p->mtx); - if (p->user_rxaio != aio) { - nni_mtx_unlock(&p->mtx); - return; - } - p->user_rxaio = NULL; - nni_aio_abort(p->rxaio, rv); - nni_aio_finish_error(aio, rv); - nni_mtx_unlock(&p->mtx); -} - -static void -wstran_pipe_recv(void *arg, nni_aio *aio) -{ - ws_pipe *p = arg; - int rv; - - if (nni_aio_begin(aio) != 0) { - return; - } - nni_mtx_lock(&p->mtx); - if ((rv = nni_aio_schedule(aio, wstran_pipe_recv_cancel, p)) != 0) { - nni_mtx_unlock(&p->mtx); - nni_aio_finish_error(aio, rv); - return; - } - p->user_rxaio = aio; - nng_stream_recv(p->ws, p->rxaio); - nni_mtx_unlock(&p->mtx); -} - -static void -wstran_pipe_send_cancel(nni_aio *aio, void *arg, int rv) -{ - ws_pipe *p = arg; - nni_mtx_lock(&p->mtx); - if (p->user_txaio != aio) { - nni_mtx_unlock(&p->mtx); - return; - } - p->user_txaio = NULL; - nni_aio_abort(p->txaio, rv); - nni_aio_finish_error(aio, rv); - nni_mtx_unlock(&p->mtx); -} - -static void -wstran_pipe_send(void *arg, nni_aio *aio) -{ - ws_pipe *p = arg; - int rv; - - if (nni_aio_begin(aio) != 0) { - return; - } - nni_mtx_lock(&p->mtx); - if ((rv = nni_aio_schedule(aio, wstran_pipe_send_cancel, p)) != 0) { - nni_mtx_unlock(&p->mtx); - nni_aio_finish_error(aio, rv); - return; - } - p->user_txaio = aio; - nni_aio_set_msg(p->txaio, nni_aio_get_msg(aio)); - nni_aio_set_msg(aio, NULL); - - nng_stream_send(p->ws, p->txaio); - nni_mtx_unlock(&p->mtx); -} - -static void -wstran_pipe_stop(void *arg) -{ - ws_pipe *p = arg; - - nni_aio_stop(p->rxaio); - nni_aio_stop(p->txaio); -} - -static int -wstran_pipe_init(void *arg, nni_pipe *pipe) -{ - NNI_ARG_UNUSED(arg); - NNI_ARG_UNUSED(pipe); - return (0); -} - -static void -wstran_pipe_fini(void *arg) -{ - ws_pipe *p = arg; - - nni_aio_free(p->rxaio); - nni_aio_free(p->txaio); - - nng_stream_free(p->ws); - nni_mtx_fini(&p->mtx); - NNI_FREE_STRUCT(p); -} - -static void -wstran_pipe_close(void *arg) -{ - ws_pipe *p = arg; - - nni_aio_close(p->rxaio); - nni_aio_close(p->txaio); - - nni_mtx_lock(&p->mtx); - nng_stream_close(p->ws); - nni_mtx_unlock(&p->mtx); -} - -static int -wstran_pipe_alloc(ws_pipe **pipep, void *ws) -{ - ws_pipe *p; - int rv; - - if ((p = NNI_ALLOC_STRUCT(p)) == NULL) { - return (NNG_ENOMEM); - } - nni_mtx_init(&p->mtx); - - // Initialize AIOs. - if (((rv = nni_aio_alloc(&p->txaio, wstran_pipe_send_cb, p)) != 0) || - ((rv = nni_aio_alloc(&p->rxaio, wstran_pipe_recv_cb, p)) != 0)) { - wstran_pipe_fini(p); - return (rv); - } - p->ws = ws; - - *pipep = p; - return (0); -} - -static uint16_t -wstran_pipe_peer(void *arg) -{ - ws_pipe *p = arg; - - return (p->peer); -} - -static int -ws_listener_bind(void *arg) -{ - ws_listener *l = arg; - int rv; - - if ((rv = nng_stream_listener_listen(l->listener)) == 0) { - l->started = true; - } - return (rv); -} - -static void -ws_listener_cancel(nni_aio *aio, void *arg, int rv) -{ - ws_listener *l = arg; - - nni_mtx_lock(&l->mtx); - if (nni_aio_list_active(aio)) { - nni_aio_list_remove(aio); - nni_aio_finish_error(aio, rv); - } - nni_mtx_unlock(&l->mtx); -} - -static void -wstran_listener_accept(void *arg, nni_aio *aio) -{ - ws_listener *l = arg; - int rv; - - // We already bound, so we just need to look for an available - // pipe (created by the handler), and match it. - // Otherwise we stick the AIO in the accept list. - if (nni_aio_begin(aio) != 0) { - return; - } - nni_mtx_lock(&l->mtx); - if ((rv = nni_aio_schedule(aio, ws_listener_cancel, l)) != 0) { - nni_mtx_unlock(&l->mtx); - nni_aio_finish_error(aio, rv); - return; - } - nni_list_append(&l->aios, aio); - if (aio == nni_list_first(&l->aios)) { - nng_stream_listener_accept(l->listener, l->accaio); - } - nni_mtx_unlock(&l->mtx); -} - -static void -wstran_dialer_cancel(nni_aio *aio, void *arg, int rv) -{ - ws_dialer *d = arg; - - nni_mtx_lock(&d->mtx); - if (nni_aio_list_active(aio)) { - nni_aio_list_remove(aio); - nni_aio_finish_error(aio, rv); - } - nni_mtx_unlock(&d->mtx); -} - -static void -wstran_dialer_connect(void *arg, nni_aio *aio) -{ - ws_dialer *d = arg; - int rv; - - if (nni_aio_begin(aio) != 0) { - return; - } - - nni_mtx_lock(&d->mtx); - if ((rv = nni_aio_schedule(aio, wstran_dialer_cancel, d)) != 0) { - nni_mtx_unlock(&d->mtx); - nni_aio_finish_error(aio, rv); - return; - } - NNI_ASSERT(nni_list_empty(&d->aios)); - d->started = true; - nni_list_append(&d->aios, aio); - nng_stream_dialer_dial(d->dialer, d->connaio); - nni_mtx_unlock(&d->mtx); -} - -static const nni_option ws_pipe_options[] = { - // terminate list - { - .o_name = NULL, - } -}; - -static int -wstran_pipe_getopt( - void *arg, const char *name, void *buf, size_t *szp, nni_type t) -{ - ws_pipe *p = arg; - int rv; - - if ((rv = nni_stream_get(p->ws, name, buf, szp, t)) == NNG_ENOTSUP) { - rv = nni_getopt(ws_pipe_options, name, p, buf, szp, t); - } - return (rv); -} - -static nni_tran_pipe_ops ws_pipe_ops = { - .p_init = wstran_pipe_init, - .p_fini = wstran_pipe_fini, - .p_stop = wstran_pipe_stop, - .p_send = wstran_pipe_send, - .p_recv = wstran_pipe_recv, - .p_close = wstran_pipe_close, - .p_peer = wstran_pipe_peer, - .p_getopt = wstran_pipe_getopt, -}; - -static void -wstran_dialer_fini(void *arg) -{ - ws_dialer *d = arg; - - nni_aio_stop(d->connaio); - nng_stream_dialer_free(d->dialer); - nni_aio_free(d->connaio); - nni_mtx_fini(&d->mtx); - NNI_FREE_STRUCT(d); -} - -static void -wstran_listener_fini(void *arg) -{ - ws_listener *l = arg; - - nni_aio_stop(l->accaio); - nng_stream_listener_free(l->listener); - nni_aio_free(l->accaio); - nni_mtx_fini(&l->mtx); - NNI_FREE_STRUCT(l); -} - -static void -wstran_connect_cb(void *arg) -{ - ws_dialer * d = arg; - ws_pipe * p; - nni_aio * caio = d->connaio; - nni_aio * uaio; - int rv; - nng_stream *ws = NULL; - - nni_mtx_lock(&d->mtx); - if (nni_aio_result(caio) == 0) { - ws = nni_aio_get_output(caio, 0); - } - if ((uaio = nni_list_first(&d->aios)) == NULL) { - // The client stopped caring about this! - nng_stream_free(ws); - nni_mtx_unlock(&d->mtx); - return; - } - nni_aio_list_remove(uaio); - NNI_ASSERT(nni_list_empty(&d->aios)); - if ((rv = nni_aio_result(caio)) != 0) { - nni_aio_finish_error(uaio, rv); - } else if ((rv = wstran_pipe_alloc(&p, ws)) != 0) { - nng_stream_free(ws); - nni_aio_finish_error(uaio, rv); - } else { - p->peer = d->peer; - - nni_aio_set_output(uaio, 0, p); - nni_aio_finish(uaio, 0, 0); - } - nni_mtx_unlock(&d->mtx); -} - -static void -wstran_dialer_close(void *arg) -{ - ws_dialer *d = arg; - - nni_aio_close(d->connaio); - nng_stream_dialer_close(d->dialer); -} - -static void -wstran_listener_close(void *arg) -{ - ws_listener *l = arg; - - nni_aio_close(l->accaio); - nng_stream_listener_close(l->listener); -} - -static void -wstran_accept_cb(void *arg) -{ - ws_listener *l = arg; - nni_aio * aaio = l->accaio; - nni_aio * uaio; - int rv; - - nni_mtx_lock(&l->mtx); - uaio = nni_list_first(&l->aios); - if ((rv = nni_aio_result(aaio)) != 0) { - if (uaio != NULL) { - nni_aio_list_remove(uaio); - nni_aio_finish_error(uaio, rv); - } - } else { - nng_stream *ws = nni_aio_get_output(aaio, 0); - if (uaio != NULL) { - ws_pipe *p; - // Make a pipe - nni_aio_list_remove(uaio); - if ((rv = wstran_pipe_alloc(&p, ws)) != 0) { - nng_stream_close(ws); - nni_aio_finish_error(uaio, rv); - } else { - p->peer = l->peer; - - nni_aio_set_output(uaio, 0, p); - nni_aio_finish(uaio, 0, 0); - } - } - } - if (!nni_list_empty(&l->aios)) { - nng_stream_listener_accept(l->listener, aaio); - } - nni_mtx_unlock(&l->mtx); -} - -static int -wstran_dialer_init(void **dp, nng_url *url, nni_dialer *ndialer) -{ - ws_dialer *d; - nni_sock * s = nni_dialer_sock(ndialer); - int rv; - char name[64]; - - if ((d = NNI_ALLOC_STRUCT(d)) == NULL) { - return (NNG_ENOMEM); - } - nni_mtx_init(&d->mtx); - - nni_aio_list_init(&d->aios); - - d->peer = nni_sock_peer_id(s); - - snprintf( - name, sizeof(name), "%s.sp.nanomsg.org", nni_sock_peer_name(s)); - - if (((rv = nni_ws_dialer_alloc(&d->dialer, url)) != 0) || - ((rv = nni_aio_alloc(&d->connaio, wstran_connect_cb, d)) != 0) || - ((rv = nng_stream_dialer_set_bool( - d->dialer, NNI_OPT_WS_MSGMODE, true)) != 0) || - ((rv = nng_stream_dialer_set_string( - d->dialer, NNG_OPT_WS_PROTOCOL, name)) != 0)) { - wstran_dialer_fini(d); - return (rv); - } - - *dp = d; - return (0); -} - -static int -wstran_listener_init(void **lp, nng_url *url, nni_listener *listener) -{ - ws_listener *l; - int rv; - nni_sock * s = nni_listener_sock(listener); - char name[64]; - - if ((l = NNI_ALLOC_STRUCT(l)) == NULL) { - return (NNG_ENOMEM); - } - nni_mtx_init(&l->mtx); - - nni_aio_list_init(&l->aios); - - l->peer = nni_sock_peer_id(s); - - snprintf( - name, sizeof(name), "%s.sp.nanomsg.org", nni_sock_proto_name(s)); - - if (((rv = nni_ws_listener_alloc(&l->listener, url)) != 0) || - ((rv = nni_aio_alloc(&l->accaio, wstran_accept_cb, l)) != 0) || - ((rv = nng_stream_listener_set_bool( - l->listener, NNI_OPT_WS_MSGMODE, true)) != 0) || - ((rv = nng_stream_listener_set_string( - l->listener, NNG_OPT_WS_PROTOCOL, name)) != 0)) { - wstran_listener_fini(l); - return (rv); - } - *lp = l; - return (0); -} - -static int -wstran_init(void) -{ - return (0); -} - -static void -wstran_fini(void) -{ -} - -static const nni_option wstran_ep_opts[] = { - // terminate list - { - .o_name = NULL, - }, -}; - -static int -wstran_dialer_getopt( - void *arg, const char *name, void *buf, size_t *szp, nni_type t) -{ - ws_dialer *d = arg; - int rv; - - rv = nni_stream_dialer_get(d->dialer, name, buf, szp, t); - if (rv == NNG_ENOTSUP) { - rv = nni_getopt(wstran_ep_opts, name, d, buf, szp, t); - } - return (rv); -} - -static int -wstran_dialer_setopt( - void *arg, const char *name, const void *buf, size_t sz, nni_type t) -{ - ws_dialer *d = arg; - int rv; - - rv = nni_stream_dialer_set(d->dialer, name, buf, sz, t); - if (rv == NNG_ENOTSUP) { - rv = nni_setopt(wstran_ep_opts, name, d, buf, sz, t); - } - return (rv); -} - -static int -wstran_listener_get( - void *arg, const char *name, void *buf, size_t *szp, nni_type t) -{ - ws_listener *l = arg; - int rv; - - rv = nni_stream_listener_get(l->listener, name, buf, szp, t); - if (rv == NNG_ENOTSUP) { - rv = nni_getopt(wstran_ep_opts, name, l, buf, szp, t); - } - return (rv); -} - -static int -wstran_listener_set( - void *arg, const char *name, const void *buf, size_t sz, nni_type t) -{ - ws_listener *l = arg; - int rv; - - rv = nni_stream_listener_set(l->listener, name, buf, sz, t); - if (rv == NNG_ENOTSUP) { - rv = nni_setopt(wstran_ep_opts, name, l, buf, sz, t); - } - return (rv); -} - -static nni_tran_dialer_ops ws_dialer_ops = { - .d_init = wstran_dialer_init, - .d_fini = wstran_dialer_fini, - .d_connect = wstran_dialer_connect, - .d_close = wstran_dialer_close, - .d_setopt = wstran_dialer_setopt, - .d_getopt = wstran_dialer_getopt, -}; - -static nni_tran_listener_ops ws_listener_ops = { - .l_init = wstran_listener_init, - .l_fini = wstran_listener_fini, - .l_bind = ws_listener_bind, - .l_accept = wstran_listener_accept, - .l_close = wstran_listener_close, - .l_setopt = wstran_listener_set, - .l_getopt = wstran_listener_get, -}; - -static nni_tran ws_tran = { - .tran_version = NNI_TRANSPORT_VERSION, - .tran_scheme = "ws", - .tran_dialer = &ws_dialer_ops, - .tran_listener = &ws_listener_ops, - .tran_pipe = &ws_pipe_ops, - .tran_init = wstran_init, - .tran_fini = wstran_fini, -}; - -static nni_tran ws4_tran = { - .tran_version = NNI_TRANSPORT_VERSION, - .tran_scheme = "ws4", - .tran_dialer = &ws_dialer_ops, - .tran_listener = &ws_listener_ops, - .tran_pipe = &ws_pipe_ops, - .tran_init = wstran_init, - .tran_fini = wstran_fini, -}; - -static nni_tran ws6_tran = { - .tran_version = NNI_TRANSPORT_VERSION, - .tran_scheme = "ws6", - .tran_dialer = &ws_dialer_ops, - .tran_listener = &ws_listener_ops, - .tran_pipe = &ws_pipe_ops, - .tran_init = wstran_init, - .tran_fini = wstran_fini, -}; - -int -nng_ws_register(void) -{ - int rv; - if (((rv = nni_tran_register(&ws_tran)) != 0) || - ((rv = nni_tran_register(&ws4_tran)) != 0) || - ((rv = nni_tran_register(&ws6_tran)) != 0)) { - return (rv); - } - - return (0); -} - -#ifdef NNG_TRANSPORT_WSS - -static nni_tran wss_tran = { - .tran_version = NNI_TRANSPORT_VERSION, - .tran_scheme = "wss", - .tran_dialer = &ws_dialer_ops, - .tran_listener = &ws_listener_ops, - .tran_pipe = &ws_pipe_ops, - .tran_init = wstran_init, - .tran_fini = wstran_fini, -}; - -static nni_tran wss4_tran = { - .tran_version = NNI_TRANSPORT_VERSION, - .tran_scheme = "wss4", - .tran_dialer = &ws_dialer_ops, - .tran_listener = &ws_listener_ops, - .tran_pipe = &ws_pipe_ops, - .tran_init = wstran_init, - .tran_fini = wstran_fini, -}; - -static nni_tran wss6_tran = { - .tran_version = NNI_TRANSPORT_VERSION, - .tran_scheme = "wss6", - .tran_dialer = &ws_dialer_ops, - .tran_listener = &ws_listener_ops, - .tran_pipe = &ws_pipe_ops, - .tran_init = wstran_init, - .tran_fini = wstran_fini, -}; - -int -nng_wss_register(void) -{ - int rv; - if (((rv = nni_tran_register(&wss_tran)) != 0) || - ((rv = nni_tran_register(&wss4_tran)) != 0) || - ((rv = nni_tran_register(&wss6_tran)) != 0)) { - return (rv); - } - - return (0); -} - -#else - -int -nng_wss_register(void) -{ - return (0); -} - -#endif // NNG_TRANSPORT_WSS diff --git a/src/transport/ws/ws_test.c b/src/transport/ws/ws_test.c deleted file mode 100644 index 7cbcd9d7..00000000 --- a/src/transport/ws/ws_test.c +++ /dev/null @@ -1,181 +0,0 @@ -// -// Copyright 2020 Staysail Systems, Inc. <info@staysail.tech> -// Copyright 2018 Cody Piersall <cody.piersall@gmail.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 <nuts.h> - -static void -test_ws_url_path_filters(void) -{ - nng_socket s1; - nng_socket s2; - char addr[NNG_MAXADDRLEN]; - - NUTS_OPEN(s1); - NUTS_OPEN(s2); - - nuts_scratch_addr("ws", sizeof(addr), addr); - NUTS_PASS(nng_listen(s1, addr, NULL, 0)); - - // Now try we just remove the last character for now. - // This will make the path different. - addr[strlen(addr) - 1] = '\0'; - NUTS_FAIL(nng_dial(s2, addr, NULL, 0), NNG_ECONNREFUSED); - - NUTS_CLOSE(s1); - NUTS_CLOSE(s2); -} - -static void -test_wild_card_port(void) -{ - nng_socket s1; - nng_socket s2; - nng_socket s3; - nng_socket s4; - nng_socket s5; - nng_socket s6; - - nng_listener l1; - nng_listener l2; - nng_listener l3; - int port1; - int port2; - int port3; - char ws_url[128]; - NUTS_OPEN(s1); - NUTS_OPEN(s2); - NUTS_OPEN(s3); - NUTS_OPEN(s4); - NUTS_OPEN(s5); - NUTS_OPEN(s6); - NUTS_PASS(nng_listen(s1, "ws://127.0.0.1:0/one", &l1, 0)); - NUTS_PASS( - nng_listener_get_int(l1, NNG_OPT_TCP_BOUND_PORT, &port1)); - NUTS_TRUE(port1 != 0); - snprintf(ws_url, sizeof(ws_url), "ws4://127.0.0.1:%d/two", port1); - NUTS_PASS(nng_listen(s2, ws_url, &l2, 0)); - NUTS_PASS( - nng_listener_get_int(l2, NNG_OPT_TCP_BOUND_PORT, &port2)); - NUTS_TRUE(port1 != 0); - NUTS_TRUE(port1 == port2); - // Now try a different wild card port. - NUTS_PASS(nng_listen(s3, "ws4://127.0.0.1:0/three", &l3, 0)); - NUTS_PASS( - nng_listener_get_int(l3, NNG_OPT_TCP_BOUND_PORT, &port3)); - NUTS_TRUE(port3 != 0); - NUTS_TRUE(port3 != port1); - - // Let's make sure can dial to each. - snprintf(ws_url, sizeof(ws_url), "ws://127.0.0.1:%d/one", port1); - NUTS_PASS(nng_dial(s4, ws_url, NULL, 0)); - snprintf(ws_url, sizeof(ws_url), "ws://127.0.0.1:%d/two", port2); - NUTS_PASS(nng_dial(s6, ws_url, NULL, 0)); - snprintf(ws_url, sizeof(ws_url), "ws://127.0.0.1:%d/three", port3); - NUTS_PASS(nng_dial(s6, ws_url, NULL, 0)); - - NUTS_CLOSE(s1); - NUTS_CLOSE(s2); - NUTS_CLOSE(s3); - NUTS_CLOSE(s4); - NUTS_CLOSE(s5); - NUTS_CLOSE(s6); -} - -static void -test_wild_card_host(void) -{ - nng_socket s1; - nng_socket s2; - char addr[NNG_MAXADDRLEN]; - uint16_t port; - - NUTS_OPEN(s1); - NUTS_OPEN(s2); - - port = nuts_next_port(); - - // we use ws4 to ensure 127.0.0.1 binding - snprintf(addr, sizeof(addr), "ws4://*:%u/test", port); - NUTS_PASS(nng_listen(s1, addr, NULL, 0)); - nng_msleep(100); - - snprintf(addr, sizeof(addr), "ws://127.0.0.1:%u/test", port); - NUTS_PASS(nng_dial(s2, addr, NULL, 0)); - - NUTS_CLOSE(s1); - NUTS_CLOSE(s2); -} - -static void -test_empty_host(void) -{ - nng_socket s1; - nng_socket s2; - char addr[NNG_MAXADDRLEN]; - uint16_t port; - - NUTS_OPEN(s1); - NUTS_OPEN(s2); - - port = nuts_next_port(); - - // we use ws4 to ensure 127.0.0.1 binding - snprintf(addr, sizeof(addr), "ws4://:%u/test", port); - NUTS_PASS(nng_listen(s1, addr, NULL, 0)); - nng_msleep(100); - - snprintf(addr, sizeof(addr), "ws://127.0.0.1:%u/test", port); - NUTS_PASS(nng_dial(s2, addr, NULL, 0)); - - NUTS_CLOSE(s1); - NUTS_CLOSE(s2); -} - -void -test_ws_recv_max(void) -{ - char msg[256]; - char buf[256]; - nng_socket s0; - nng_socket s1; - nng_listener l; - size_t sz; - char *addr; - - NUTS_ADDR(addr, "ws"); - NUTS_OPEN(s0); - NUTS_PASS(nng_socket_set_ms(s0, NNG_OPT_RECVTIMEO, 100)); - NUTS_PASS(nng_socket_set_size(s0, NNG_OPT_RECVMAXSZ, 200)); - NUTS_PASS(nng_listener_create(&l, s0, addr)); - NUTS_PASS(nng_socket_get_size(s0, NNG_OPT_RECVMAXSZ, &sz)); - NUTS_TRUE(sz == 200); - NUTS_PASS(nng_listener_set_size(l, NNG_OPT_RECVMAXSZ, 100)); - NUTS_PASS(nng_listener_start(l, 0)); - - NUTS_OPEN(s1); - NUTS_PASS(nng_dial(s1, addr, NULL, 0)); - NUTS_PASS(nng_send(s1, msg, 95, 0)); - NUTS_PASS(nng_socket_set_ms(s1, NNG_OPT_SENDTIMEO, 100)); - NUTS_PASS(nng_recv(s0, buf, &sz, 0)); - NUTS_TRUE(sz == 95); - NUTS_PASS(nng_send(s1, msg, 150, 0)); - NUTS_FAIL(nng_recv(s0, buf, &sz, 0), NNG_ETIMEDOUT); - NUTS_CLOSE(s0); - NUTS_CLOSE(s1); -} - -TEST_LIST = { - { "ws url path filters", test_ws_url_path_filters }, - { "ws wild card port", test_wild_card_port }, - { "ws wild card host", test_wild_card_host }, - { "ws empty host", test_empty_host }, - { "ws recv max", test_ws_recv_max }, - { NULL, NULL }, -};
\ No newline at end of file |