aboutsummaryrefslogtreecommitdiff
path: root/src/transport/tcp
diff options
context:
space:
mode:
Diffstat (limited to 'src/transport/tcp')
-rw-r--r--src/transport/tcp/CMakeLists.txt17
-rw-r--r--src/transport/tcp/tcp.c1263
-rw-r--r--src/transport/tcp/tcp_test.c297
3 files changed, 0 insertions, 1577 deletions
diff --git a/src/transport/tcp/CMakeLists.txt b/src/transport/tcp/CMakeLists.txt
deleted file mode 100644
index d6022329..00000000
--- a/src/transport/tcp/CMakeLists.txt
+++ /dev/null
@@ -1,17 +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.
-#
-
-# TCP protocol
-nng_directory(tcp)
-
-nng_sources_if(NNG_TRANSPORT_TCP tcp.c)
-nng_headers_if(NNG_TRANSPORT_TCP nng/transport/tcp/tcp.h)
-nng_defines_if(NNG_TRANSPORT_TCP NNG_TRANSPORT_TCP)
-nng_test(tcp_test) \ No newline at end of file
diff --git a/src/transport/tcp/tcp.c b/src/transport/tcp/tcp.c
deleted file mode 100644
index 524c6988..00000000
--- a/src/transport/tcp/tcp.c
+++ /dev/null
@@ -1,1263 +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 <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-
-#include "core/nng_impl.h"
-
-// TCP transport. Platform specific TCP operations must be
-// supplied as well.
-
-typedef struct tcptran_pipe tcptran_pipe;
-typedef struct tcptran_ep tcptran_ep;
-
-// tcp_pipe is one end of a TCP connection.
-struct tcptran_pipe {
- nng_stream * conn;
- nni_pipe * npipe;
- uint16_t peer;
- uint16_t proto;
- size_t rcvmax;
- bool closed;
- nni_list_node node;
- tcptran_ep * ep;
- nni_atomic_flag reaped;
- nni_reap_node reap;
- uint8_t txlen[sizeof(uint64_t)];
- uint8_t rxlen[sizeof(uint64_t)];
- size_t gottxhead;
- size_t gotrxhead;
- size_t wanttxhead;
- size_t wantrxhead;
- nni_list recvq;
- nni_list sendq;
- nni_aio * txaio;
- nni_aio * rxaio;
- nni_aio * negoaio;
- nni_msg * rxmsg;
- nni_mtx mtx;
-};
-
-struct tcptran_ep {
- nni_mtx mtx;
- uint16_t proto;
- size_t rcvmax;
- bool fini;
- bool started;
- bool closed;
- nng_url * url;
- const char * host; // for dialers
- nng_sockaddr src;
- int refcnt; // active pipes
- nni_aio * useraio;
- nni_aio * connaio;
- nni_aio * timeaio;
- nni_list busypipes; // busy pipes -- ones passed to socket
- nni_list waitpipes; // pipes waiting to match to socket
- nni_list negopipes; // pipes busy negotiating
- nni_reap_node reap;
- nng_stream_dialer * dialer;
- nng_stream_listener *listener;
-
-#ifdef NNG_ENABLE_STATS
- nni_stat_item st_rcv_max;
-#endif
-};
-
-static void tcptran_pipe_send_start(tcptran_pipe *);
-static void tcptran_pipe_recv_start(tcptran_pipe *);
-static void tcptran_pipe_send_cb(void *);
-static void tcptran_pipe_recv_cb(void *);
-static void tcptran_pipe_nego_cb(void *);
-static void tcptran_ep_fini(void *);
-static void tcptran_pipe_fini(void *);
-
-static nni_reap_list tcptran_ep_reap_list = {
- .rl_offset = offsetof(tcptran_ep, reap),
- .rl_func = tcptran_ep_fini,
-};
-
-static nni_reap_list tcptran_pipe_reap_list = {
- .rl_offset = offsetof (tcptran_pipe, reap),
- .rl_func = tcptran_pipe_fini,
-};
-
-static int
-tcptran_init(void)
-{
- return (0);
-}
-
-static void
-tcptran_fini(void)
-{
-}
-
-static void
-tcptran_pipe_close(void *arg)
-{
- tcptran_pipe *p = arg;
-
- nni_mtx_lock(&p->mtx);
- p->closed = true;
- nni_mtx_unlock(&p->mtx);
-
- nni_aio_close(p->rxaio);
- nni_aio_close(p->txaio);
- nni_aio_close(p->negoaio);
-
- nng_stream_close(p->conn);
-}
-
-static void
-tcptran_pipe_stop(void *arg)
-{
- tcptran_pipe *p = arg;
-
- nni_aio_stop(p->rxaio);
- nni_aio_stop(p->txaio);
- nni_aio_stop(p->negoaio);
-}
-
-static int
-tcptran_pipe_init(void *arg, nni_pipe *npipe)
-{
- tcptran_pipe *p = arg;
- p->npipe = npipe;
-
- return (0);
-}
-
-static void
-tcptran_pipe_fini(void *arg)
-{
- tcptran_pipe *p = arg;
- tcptran_ep * ep;
-
- tcptran_pipe_stop(p);
- if ((ep = p->ep) != NULL) {
- nni_mtx_lock(&ep->mtx);
- nni_list_node_remove(&p->node);
- ep->refcnt--;
- if (ep->fini && (ep->refcnt == 0)) {
- nni_reap(&tcptran_ep_reap_list, ep);
- }
- nni_mtx_unlock(&ep->mtx);
- }
-
- nni_aio_free(p->rxaio);
- nni_aio_free(p->txaio);
- nni_aio_free(p->negoaio);
- nng_stream_free(p->conn);
- nni_msg_free(p->rxmsg);
- nni_mtx_fini(&p->mtx);
- NNI_FREE_STRUCT(p);
-}
-
-static void
-tcptran_pipe_reap(tcptran_pipe *p)
-{
- if (!nni_atomic_flag_test_and_set(&p->reaped)) {
- if (p->conn != NULL) {
- nng_stream_close(p->conn);
- }
- nni_reap(&tcptran_pipe_reap_list, p);
- }
-}
-
-static int
-tcptran_pipe_alloc(tcptran_pipe **pipep)
-{
- tcptran_pipe *p;
- int rv;
-
- if ((p = NNI_ALLOC_STRUCT(p)) == NULL) {
- return (NNG_ENOMEM);
- }
- nni_mtx_init(&p->mtx);
- if (((rv = nni_aio_alloc(&p->txaio, tcptran_pipe_send_cb, p)) != 0) ||
- ((rv = nni_aio_alloc(&p->rxaio, tcptran_pipe_recv_cb, p)) != 0) ||
- ((rv = nni_aio_alloc(&p->negoaio, tcptran_pipe_nego_cb, p)) !=
- 0)) {
- tcptran_pipe_fini(p);
- return (rv);
- }
- nni_aio_list_init(&p->recvq);
- nni_aio_list_init(&p->sendq);
- nni_atomic_flag_reset(&p->reaped);
-
- *pipep = p;
-
- return (0);
-}
-
-static void
-tcptran_ep_match(tcptran_ep *ep)
-{
- nni_aio * aio;
- tcptran_pipe *p;
-
- if (((aio = ep->useraio) == NULL) ||
- ((p = nni_list_first(&ep->waitpipes)) == NULL)) {
- return;
- }
- nni_list_remove(&ep->waitpipes, p);
- nni_list_append(&ep->busypipes, p);
- ep->useraio = NULL;
- p->rcvmax = ep->rcvmax;
- nni_aio_set_output(aio, 0, p);
- nni_aio_finish(aio, 0, 0);
-}
-
-static void
-tcptran_pipe_nego_cb(void *arg)
-{
- tcptran_pipe *p = arg;
- tcptran_ep * ep = p->ep;
- nni_aio * aio = p->negoaio;
- nni_aio * uaio;
- int rv;
-
- nni_mtx_lock(&ep->mtx);
-
- if ((rv = nni_aio_result(aio)) != 0) {
- goto error;
- }
-
- // We start transmitting before we receive.
- if (p->gottxhead < p->wanttxhead) {
- p->gottxhead += nni_aio_count(aio);
- } else if (p->gotrxhead < p->wantrxhead) {
- p->gotrxhead += nni_aio_count(aio);
- }
-
- if (p->gottxhead < p->wanttxhead) {
- nni_iov iov;
- iov.iov_len = p->wanttxhead - p->gottxhead;
- iov.iov_buf = &p->txlen[p->gottxhead];
- // send it down...
- nni_aio_set_iov(aio, 1, &iov);
- nng_stream_send(p->conn, aio);
- nni_mtx_unlock(&ep->mtx);
- return;
- }
- if (p->gotrxhead < p->wantrxhead) {
- nni_iov iov;
- iov.iov_len = p->wantrxhead - p->gotrxhead;
- iov.iov_buf = &p->rxlen[p->gotrxhead];
- nni_aio_set_iov(aio, 1, &iov);
- nng_stream_recv(p->conn, aio);
- nni_mtx_unlock(&ep->mtx);
- return;
- }
- // We have both sent and received the headers. Lets check the
- // receive side header.
- if ((p->rxlen[0] != 0) || (p->rxlen[1] != 'S') ||
- (p->rxlen[2] != 'P') || (p->rxlen[3] != 0) || (p->rxlen[6] != 0) ||
- (p->rxlen[7] != 0)) {
- rv = NNG_EPROTO;
- goto error;
- }
-
- NNI_GET16(&p->rxlen[4], p->peer);
-
- // We are all ready now. We put this in the wait list, and
- // then try to run the matcher.
- nni_list_remove(&ep->negopipes, p);
- nni_list_append(&ep->waitpipes, p);
-
- tcptran_ep_match(ep);
- nni_mtx_unlock(&ep->mtx);
-
- return;
-
-error:
- nng_stream_close(p->conn);
-
- if ((uaio = ep->useraio) != NULL) {
- ep->useraio = NULL;
- nni_aio_finish_error(uaio, rv);
- }
- nni_mtx_unlock(&ep->mtx);
- tcptran_pipe_reap(p);
-}
-
-static void
-tcptran_pipe_send_cb(void *arg)
-{
- tcptran_pipe *p = arg;
- int rv;
- nni_aio * aio;
- size_t n;
- nni_msg * msg;
- nni_aio * txaio = p->txaio;
-
- nni_mtx_lock(&p->mtx);
- aio = nni_list_first(&p->sendq);
-
- if ((rv = nni_aio_result(txaio)) != 0) {
- nni_pipe_bump_error(p->npipe, rv);
- // Intentionally we do not queue up another transfer.
- // There's an excellent chance that the pipe is no longer
- // usable, with a partial transfer.
- // The protocol should see this error, and close the
- // pipe itself, we hope.
- nni_aio_list_remove(aio);
- nni_mtx_unlock(&p->mtx);
- nni_aio_finish_error(aio, rv);
- return;
- }
-
- n = nni_aio_count(txaio);
- nni_aio_iov_advance(txaio, n);
- if (nni_aio_iov_count(txaio) > 0) {
- nng_stream_send(p->conn, txaio);
- nni_mtx_unlock(&p->mtx);
- return;
- }
-
- nni_aio_list_remove(aio);
- tcptran_pipe_send_start(p);
-
- msg = nni_aio_get_msg(aio);
- n = nni_msg_len(msg);
- nni_pipe_bump_tx(p->npipe, n);
- nni_mtx_unlock(&p->mtx);
-
- nni_aio_set_msg(aio, NULL);
- nni_msg_free(msg);
- nni_aio_finish_sync(aio, 0, n);
-}
-
-static void
-tcptran_pipe_recv_cb(void *arg)
-{
- tcptran_pipe *p = arg;
- nni_aio * aio;
- int rv;
- size_t n;
- nni_msg * msg;
- nni_aio * rxaio = p->rxaio;
-
- nni_mtx_lock(&p->mtx);
- aio = nni_list_first(&p->recvq);
-
- if ((rv = nni_aio_result(rxaio)) != 0) {
- goto recv_error;
- }
-
- n = nni_aio_count(rxaio);
- nni_aio_iov_advance(rxaio, n);
- if (nni_aio_iov_count(rxaio) > 0) {
- nng_stream_recv(p->conn, rxaio);
- nni_mtx_unlock(&p->mtx);
- return;
- }
-
- // If we don't have a message yet, we were reading the TCP message
- // header, which is just the length. This tells us the size of the
- // message to allocate and how much more to expect.
- if (p->rxmsg == NULL) {
- uint64_t len;
- // We should have gotten a message header.
- NNI_GET64(p->rxlen, len);
-
- // Make sure the message payload is not too big. If it is
- // the caller will shut down the pipe.
- if ((len > p->rcvmax) && (p->rcvmax > 0)) {
- rv = NNG_EMSGSIZE;
- goto recv_error;
- }
-
- if ((rv = nni_msg_alloc(&p->rxmsg, (size_t) len)) != 0) {
- goto recv_error;
- }
-
- // Submit the rest of the data for a read -- we want to
- // read the entire message now.
- if (len != 0) {
- nni_iov iov;
- iov.iov_buf = nni_msg_body(p->rxmsg);
- iov.iov_len = (size_t) len;
-
- nni_aio_set_iov(rxaio, 1, &iov);
- nng_stream_recv(p->conn, rxaio);
- nni_mtx_unlock(&p->mtx);
- return;
- }
- }
-
- // We read a message completely. Let the user know the good news.
- nni_aio_list_remove(aio);
- msg = p->rxmsg;
- p->rxmsg = NULL;
- n = nni_msg_len(msg);
-
- nni_pipe_bump_rx(p->npipe, n);
- tcptran_pipe_recv_start(p);
- nni_mtx_unlock(&p->mtx);
-
- nni_aio_set_msg(aio, msg);
- nni_aio_finish_sync(aio, 0, n);
- return;
-
-recv_error:
- nni_aio_list_remove(aio);
- msg = p->rxmsg;
- p->rxmsg = NULL;
- nni_pipe_bump_error(p->npipe, rv);
- // Intentionally, we do not queue up another receive.
- // The protocol should notice this error and close the pipe.
- nni_mtx_unlock(&p->mtx);
-
- nni_msg_free(msg);
- nni_aio_finish_error(aio, rv);
-}
-
-static void
-tcptran_pipe_send_cancel(nni_aio *aio, void *arg, int rv)
-{
- tcptran_pipe *p = arg;
-
- nni_mtx_lock(&p->mtx);
- if (!nni_aio_list_active(aio)) {
- nni_mtx_unlock(&p->mtx);
- return;
- }
- // If this is being sent, then cancel the pending transfer.
- // The callback on the txaio will cause the user aio to
- // be canceled too.
- if (nni_list_first(&p->sendq) == aio) {
- nni_aio_abort(p->txaio, rv);
- nni_mtx_unlock(&p->mtx);
- return;
- }
- nni_aio_list_remove(aio);
- nni_mtx_unlock(&p->mtx);
-
- nni_aio_finish_error(aio, rv);
-}
-
-static void
-tcptran_pipe_send_start(tcptran_pipe *p)
-{
- nni_aio *aio;
- nni_aio *txaio;
- nni_msg *msg;
- int niov;
- nni_iov iov[3];
- uint64_t len;
-
- if (p->closed) {
- while ((aio = nni_list_first(&p->sendq)) != NULL) {
- nni_list_remove(&p->sendq, aio);
- nni_aio_finish_error(aio, NNG_ECLOSED);
- }
- return;
- }
-
- if ((aio = nni_list_first(&p->sendq)) == NULL) {
- return;
- }
-
- // This runs to send the message.
- msg = nni_aio_get_msg(aio);
- len = nni_msg_len(msg) + nni_msg_header_len(msg);
-
- NNI_PUT64(p->txlen, len);
-
- txaio = p->txaio;
- niov = 0;
- iov[0].iov_buf = p->txlen;
- iov[0].iov_len = sizeof(p->txlen);
- niov++;
- if (nni_msg_header_len(msg) > 0) {
- iov[niov].iov_buf = nni_msg_header(msg);
- iov[niov].iov_len = nni_msg_header_len(msg);
- niov++;
- }
- if (nni_msg_len(msg) > 0) {
- iov[niov].iov_buf = nni_msg_body(msg);
- iov[niov].iov_len = nni_msg_len(msg);
- niov++;
- }
- nni_aio_set_iov(txaio, niov, iov);
- nng_stream_send(p->conn, txaio);
-}
-
-static void
-tcptran_pipe_send(void *arg, nni_aio *aio)
-{
- tcptran_pipe *p = arg;
- int rv;
-
- if (nni_aio_begin(aio) != 0) {
- return;
- }
- nni_mtx_lock(&p->mtx);
- if ((rv = nni_aio_schedule(aio, tcptran_pipe_send_cancel, p)) != 0) {
- nni_mtx_unlock(&p->mtx);
- nni_aio_finish_error(aio, rv);
- return;
- }
- nni_list_append(&p->sendq, aio);
- if (nni_list_first(&p->sendq) == aio) {
- tcptran_pipe_send_start(p);
- }
- nni_mtx_unlock(&p->mtx);
-}
-
-static void
-tcptran_pipe_recv_cancel(nni_aio *aio, void *arg, int rv)
-{
- tcptran_pipe *p = arg;
-
- nni_mtx_lock(&p->mtx);
- if (!nni_aio_list_active(aio)) {
- nni_mtx_unlock(&p->mtx);
- return;
- }
- // If receive in progress, then cancel the pending transfer.
- // The callback on the rxaio will cause the user aio to
- // be canceled too.
- if (nni_list_first(&p->recvq) == aio) {
- nni_aio_abort(p->rxaio, rv);
- nni_mtx_unlock(&p->mtx);
- return;
- }
- nni_aio_list_remove(aio);
- nni_mtx_unlock(&p->mtx);
- nni_aio_finish_error(aio, rv);
-}
-
-static void
-tcptran_pipe_recv_start(tcptran_pipe *p)
-{
- nni_aio *rxaio;
- nni_iov iov;
- NNI_ASSERT(p->rxmsg == NULL);
-
- if (p->closed) {
- nni_aio *aio;
- while ((aio = nni_list_first(&p->recvq)) != NULL) {
- nni_list_remove(&p->recvq, aio);
- nni_aio_finish_error(aio, NNG_ECLOSED);
- }
- return;
- }
- if (nni_list_empty(&p->recvq)) {
- return;
- }
-
- // Schedule a read of the header.
- rxaio = p->rxaio;
- iov.iov_buf = p->rxlen;
- iov.iov_len = sizeof(p->rxlen);
- nni_aio_set_iov(rxaio, 1, &iov);
-
- nng_stream_recv(p->conn, rxaio);
-}
-
-static void
-tcptran_pipe_recv(void *arg, nni_aio *aio)
-{
- tcptran_pipe *p = arg;
- int rv;
-
- if (nni_aio_begin(aio) != 0) {
- return;
- }
- nni_mtx_lock(&p->mtx);
- if ((rv = nni_aio_schedule(aio, tcptran_pipe_recv_cancel, p)) != 0) {
- nni_mtx_unlock(&p->mtx);
- nni_aio_finish_error(aio, rv);
- return;
- }
-
- nni_list_append(&p->recvq, aio);
- if (nni_list_first(&p->recvq) == aio) {
- tcptran_pipe_recv_start(p);
- }
- nni_mtx_unlock(&p->mtx);
-}
-
-static uint16_t
-tcptran_pipe_peer(void *arg)
-{
- tcptran_pipe *p = arg;
-
- return (p->peer);
-}
-
-static int
-tcptran_pipe_getopt(
- void *arg, const char *name, void *buf, size_t *szp, nni_type t)
-{
- tcptran_pipe *p = arg;
- return (nni_stream_get(p->conn, name, buf, szp, t));
-}
-
-static void
-tcptran_pipe_start(tcptran_pipe *p, nng_stream *conn, tcptran_ep *ep)
-{
- nni_iov iov;
-
- ep->refcnt++;
-
- p->conn = conn;
- p->ep = ep;
- p->proto = ep->proto;
-
- p->txlen[0] = 0;
- p->txlen[1] = 'S';
- p->txlen[2] = 'P';
- p->txlen[3] = 0;
- NNI_PUT16(&p->txlen[4], p->proto);
- NNI_PUT16(&p->txlen[6], 0);
-
- p->gotrxhead = 0;
- p->gottxhead = 0;
- p->wantrxhead = 8;
- p->wanttxhead = 8;
- iov.iov_len = 8;
- iov.iov_buf = &p->txlen[0];
- nni_aio_set_iov(p->negoaio, 1, &iov);
- nni_list_append(&ep->negopipes, p);
-
- nni_aio_set_timeout(p->negoaio, 10000); // 10 sec timeout to negotiate
- nng_stream_send(p->conn, p->negoaio);
-}
-
-static void
-tcptran_ep_fini(void *arg)
-{
- tcptran_ep *ep = arg;
-
- nni_mtx_lock(&ep->mtx);
- ep->fini = true;
- if (ep->refcnt != 0) {
- nni_mtx_unlock(&ep->mtx);
- return;
- }
- nni_mtx_unlock(&ep->mtx);
- nni_aio_stop(ep->timeaio);
- nni_aio_stop(ep->connaio);
- nng_stream_dialer_free(ep->dialer);
- nng_stream_listener_free(ep->listener);
- nni_aio_free(ep->timeaio);
- nni_aio_free(ep->connaio);
-
- nni_mtx_fini(&ep->mtx);
- NNI_FREE_STRUCT(ep);
-}
-
-static void
-tcptran_ep_close(void *arg)
-{
- tcptran_ep * ep = arg;
- tcptran_pipe *p;
-
- nni_mtx_lock(&ep->mtx);
-
- ep->closed = true;
- nni_aio_close(ep->timeaio);
- if (ep->dialer != NULL) {
- nng_stream_dialer_close(ep->dialer);
- }
- if (ep->listener != NULL) {
- nng_stream_listener_close(ep->listener);
- }
- NNI_LIST_FOREACH (&ep->negopipes, p) {
- tcptran_pipe_close(p);
- }
- NNI_LIST_FOREACH (&ep->waitpipes, p) {
- tcptran_pipe_close(p);
- }
- NNI_LIST_FOREACH (&ep->busypipes, p) {
- tcptran_pipe_close(p);
- }
- if (ep->useraio != NULL) {
- nni_aio_finish_error(ep->useraio, NNG_ECLOSED);
- ep->useraio = NULL;
- }
-
- nni_mtx_unlock(&ep->mtx);
-}
-
-// This parses off the optional source address that this transport uses.
-// The special handling of this URL format is quite honestly an historical
-// mistake, which we would remove if we could.
-static int
-tcptran_url_parse_source(nng_url *url, nng_sockaddr *sa, const nng_url *surl)
-{
- int af;
- char * semi;
- char * src;
- size_t len;
- int rv;
- nni_aio *aio;
-
- // We modify the URL. This relies on the fact that the underlying
- // transport does not free this, so we can just use references.
-
- url->u_scheme = surl->u_scheme;
- url->u_port = surl->u_port;
- url->u_hostname = surl->u_hostname;
-
- if ((semi = strchr(url->u_hostname, ';')) == NULL) {
- memset(sa, 0, sizeof(*sa));
- return (0);
- }
-
- len = (size_t)(semi - url->u_hostname);
- url->u_hostname = semi + 1;
-
- if (strcmp(surl->u_scheme, "tcp") == 0) {
- af = NNG_AF_UNSPEC;
- } else if (strcmp(surl->u_scheme, "tcp4") == 0) {
- af = NNG_AF_INET;
- } else if (strcmp(surl->u_scheme, "tcp6") == 0) {
- af = NNG_AF_INET6;
- } else {
- return (NNG_EADDRINVAL);
- }
-
- if ((src = nni_alloc(len + 1)) == NULL) {
- return (NNG_ENOMEM);
- }
- memcpy(src, surl->u_hostname, len);
- src[len] = '\0';
-
- if ((rv = nni_aio_alloc(&aio, NULL, NULL)) != 0) {
- nni_free(src, len + 1);
- return (rv);
- }
-
- nni_resolv_ip(src, "0", af, true, sa, aio);
- nni_aio_wait(aio);
- nni_aio_free(aio);
- nni_free(src, len + 1);
- return (rv);
-}
-
-static void
-tcptran_timer_cb(void *arg)
-{
- tcptran_ep *ep = arg;
- if (nni_aio_result(ep->timeaio) == 0) {
- nng_stream_listener_accept(ep->listener, ep->connaio);
- }
-}
-
-static void
-tcptran_accept_cb(void *arg)
-{
- tcptran_ep * ep = arg;
- nni_aio * aio = ep->connaio;
- tcptran_pipe *p;
- int rv;
- nng_stream * conn;
-
- nni_mtx_lock(&ep->mtx);
-
- if ((rv = nni_aio_result(aio)) != 0) {
- goto error;
- }
-
- conn = nni_aio_get_output(aio, 0);
- if ((rv = tcptran_pipe_alloc(&p)) != 0) {
- nng_stream_free(conn);
- goto error;
- }
-
- if (ep->closed) {
- tcptran_pipe_fini(p);
- nng_stream_free(conn);
- rv = NNG_ECLOSED;
- goto error;
- }
- tcptran_pipe_start(p, conn, ep);
- nng_stream_listener_accept(ep->listener, ep->connaio);
- nni_mtx_unlock(&ep->mtx);
- return;
-
-error:
- // When an error here occurs, let's send a notice up to the consumer.
- // That way it can be reported properly.
- if ((aio = ep->useraio) != NULL) {
- ep->useraio = NULL;
- nni_aio_finish_error(aio, rv);
- }
- switch (rv) {
-
- case NNG_ENOMEM:
- case NNG_ENOFILES:
- nng_sleep_aio(10, ep->timeaio);
- break;
-
- default:
- if (!ep->closed) {
- nng_stream_listener_accept(ep->listener, ep->connaio);
- }
- break;
- }
- nni_mtx_unlock(&ep->mtx);
-}
-
-static void
-tcptran_dial_cb(void *arg)
-{
- tcptran_ep * ep = arg;
- nni_aio * aio = ep->connaio;
- tcptran_pipe *p;
- int rv;
- nng_stream * conn;
-
- if ((rv = nni_aio_result(aio)) != 0) {
- goto error;
- }
-
- conn = nni_aio_get_output(aio, 0);
- if ((rv = tcptran_pipe_alloc(&p)) != 0) {
- nng_stream_free(conn);
- goto error;
- }
- nni_mtx_lock(&ep->mtx);
- if (ep->closed) {
- tcptran_pipe_fini(p);
- nng_stream_free(conn);
- rv = NNG_ECLOSED;
- nni_mtx_unlock(&ep->mtx);
- goto error;
- } else {
- tcptran_pipe_start(p, conn, ep);
- }
- nni_mtx_unlock(&ep->mtx);
- return;
-
-error:
- // Error connecting. We need to pass this straight back
- // to the user.
- nni_mtx_lock(&ep->mtx);
- if ((aio = ep->useraio) != NULL) {
- ep->useraio = NULL;
- nni_aio_finish_error(aio, rv);
- }
- nni_mtx_unlock(&ep->mtx);
-}
-
-static int
-tcptran_ep_init(tcptran_ep **epp, nng_url *url, nni_sock *sock)
-{
- tcptran_ep *ep;
-
- if ((ep = NNI_ALLOC_STRUCT(ep)) == NULL) {
- return (NNG_ENOMEM);
- }
- nni_mtx_init(&ep->mtx);
- NNI_LIST_INIT(&ep->busypipes, tcptran_pipe, node);
- NNI_LIST_INIT(&ep->waitpipes, tcptran_pipe, node);
- NNI_LIST_INIT(&ep->negopipes, tcptran_pipe, node);
-
- ep->proto = nni_sock_proto_id(sock);
- ep->url = url;
-
-#ifdef NNG_ENABLE_STATS
- static const nni_stat_info rcv_max_info = {
- .si_name = "rcv_max",
- .si_desc = "maximum receive size",
- .si_type = NNG_STAT_LEVEL,
- .si_unit = NNG_UNIT_BYTES,
- .si_atomic = true,
- };
- nni_stat_init(&ep->st_rcv_max, &rcv_max_info);
-#endif
-
- *epp = ep;
- return (0);
-}
-
-static int
-tcptran_dialer_init(void **dp, nng_url *url, nni_dialer *ndialer)
-{
- tcptran_ep * ep;
- int rv;
- nng_sockaddr srcsa;
- nni_sock * sock = nni_dialer_sock(ndialer);
- nng_url myurl;
-
- // Check for invalid URL components.
- if ((strlen(url->u_path) != 0) && (strcmp(url->u_path, "/") != 0)) {
- return (NNG_EADDRINVAL);
- }
- if ((url->u_fragment != NULL) || (url->u_userinfo != NULL) ||
- (url->u_query != NULL) || (strlen(url->u_hostname) == 0) ||
- (strlen(url->u_port) == 0)) {
- return (NNG_EADDRINVAL);
- }
-
- if ((rv = tcptran_url_parse_source(&myurl, &srcsa, url)) != 0) {
- return (rv);
- }
-
- if ((rv = tcptran_ep_init(&ep, url, sock)) != 0) {
- return (rv);
- }
-
- if ((rv != 0) ||
- ((rv = nni_aio_alloc(&ep->connaio, tcptran_dial_cb, ep)) != 0) ||
- ((rv = nng_stream_dialer_alloc_url(&ep->dialer, &myurl)) != 0)) {
- tcptran_ep_fini(ep);
- return (rv);
- }
- if ((srcsa.s_family != NNG_AF_UNSPEC) &&
- ((rv = nni_stream_dialer_set(ep->dialer, NNG_OPT_LOCADDR, &srcsa,
- sizeof(srcsa), NNI_TYPE_SOCKADDR)) != 0)) {
- tcptran_ep_fini(ep);
- return (rv);
- }
-
-#ifdef NNG_ENABLE_STATS
- nni_dialer_add_stat(ndialer, &ep->st_rcv_max);
-#endif
- *dp = ep;
- return (0);
-}
-
-static int
-tcptran_listener_init(void **lp, nng_url *url, nni_listener *nlistener)
-{
- tcptran_ep *ep;
- int rv;
- nni_sock * sock = nni_listener_sock(nlistener);
-
- // Check for invalid URL components.
- if ((strlen(url->u_path) != 0) && (strcmp(url->u_path, "/") != 0)) {
- return (NNG_EADDRINVAL);
- }
- if ((url->u_fragment != NULL) || (url->u_userinfo != NULL) ||
- (url->u_query != NULL)) {
- return (NNG_EADDRINVAL);
- }
-
- if ((rv = tcptran_ep_init(&ep, url, sock)) != 0) {
- return (rv);
- }
-
- if (((rv = nni_aio_alloc(&ep->connaio, tcptran_accept_cb, ep)) != 0) ||
- ((rv = nni_aio_alloc(&ep->timeaio, tcptran_timer_cb, ep)) != 0) ||
- ((rv = nng_stream_listener_alloc_url(&ep->listener, url)) != 0)) {
- tcptran_ep_fini(ep);
- return (rv);
- }
-#ifdef NNG_ENABLE_STATS
- nni_listener_add_stat(nlistener, &ep->st_rcv_max);
-#endif
-
- *lp = ep;
- return (0);
-}
-
-static void
-tcptran_ep_cancel(nni_aio *aio, void *arg, int rv)
-{
- tcptran_ep *ep = arg;
- nni_mtx_lock(&ep->mtx);
- if (ep->useraio == aio) {
- ep->useraio = NULL;
- nni_aio_finish_error(aio, rv);
- }
- nni_mtx_unlock(&ep->mtx);
-}
-
-static void
-tcptran_ep_connect(void *arg, nni_aio *aio)
-{
- tcptran_ep *ep = arg;
- int rv;
-
- if (nni_aio_begin(aio) != 0) {
- return;
- }
- nni_mtx_lock(&ep->mtx);
- if (ep->closed) {
- nni_mtx_unlock(&ep->mtx);
- nni_aio_finish_error(aio, NNG_ECLOSED);
- return;
- }
- if (ep->useraio != NULL) {
- nni_mtx_unlock(&ep->mtx);
- nni_aio_finish_error(aio, NNG_EBUSY);
- return;
- }
- if ((rv = nni_aio_schedule(aio, tcptran_ep_cancel, ep)) != 0) {
- nni_mtx_unlock(&ep->mtx);
- nni_aio_finish_error(aio, rv);
- return;
- }
- ep->useraio = aio;
-
- nng_stream_dialer_dial(ep->dialer, ep->connaio);
- nni_mtx_unlock(&ep->mtx);
-}
-
-static int
-tcptran_ep_get_url(void *arg, void *v, size_t *szp, nni_opt_type t)
-{
- tcptran_ep *ep = arg;
- char * s;
- int rv;
- int port = 0;
-
- if (ep->listener != NULL) {
- (void) nng_stream_listener_get_int(
- ep->listener, NNG_OPT_TCP_BOUND_PORT, &port);
- }
-
- if ((rv = nni_url_asprintf_port(&s, ep->url, port)) == 0) {
- rv = nni_copyout_str(s, v, szp, t);
- nni_strfree(s);
- }
- return (rv);
-}
-
-static int
-tcptran_ep_get_recvmaxsz(void *arg, void *v, size_t *szp, nni_opt_type t)
-{
- tcptran_ep *ep = arg;
- int rv;
-
- nni_mtx_lock(&ep->mtx);
- rv = nni_copyout_size(ep->rcvmax, v, szp, t);
- nni_mtx_unlock(&ep->mtx);
- return (rv);
-}
-
-static int
-tcptran_ep_set_recvmaxsz(void *arg, const void *v, size_t sz, nni_opt_type t)
-{
- tcptran_ep *ep = arg;
- size_t val;
- int rv;
- if ((rv = nni_copyin_size(&val, v, sz, 0, NNI_MAXSZ, t)) == 0) {
- tcptran_pipe *p;
- nni_mtx_lock(&ep->mtx);
- ep->rcvmax = val;
- NNI_LIST_FOREACH (&ep->waitpipes, p) {
- p->rcvmax = val;
- }
- NNI_LIST_FOREACH (&ep->negopipes, p) {
- p->rcvmax = val;
- }
- NNI_LIST_FOREACH (&ep->busypipes, p) {
- p->rcvmax = val;
- }
- nni_mtx_unlock(&ep->mtx);
-#ifdef NNG_ENABLE_STATS
- nni_stat_set_value(&ep->st_rcv_max, val);
-#endif
- }
- return (rv);
-}
-
-static int
-tcptran_ep_bind(void *arg)
-{
- tcptran_ep *ep = arg;
- int rv;
-
- nni_mtx_lock(&ep->mtx);
- rv = nng_stream_listener_listen(ep->listener);
- nni_mtx_unlock(&ep->mtx);
-
- return (rv);
-}
-
-static void
-tcptran_ep_accept(void *arg, nni_aio *aio)
-{
- tcptran_ep *ep = arg;
- int rv;
-
- if (nni_aio_begin(aio) != 0) {
- return;
- }
- nni_mtx_lock(&ep->mtx);
- if (ep->closed) {
- nni_mtx_unlock(&ep->mtx);
- nni_aio_finish_error(aio, NNG_ECLOSED);
- return;
- }
- if (ep->useraio != NULL) {
- nni_mtx_unlock(&ep->mtx);
- nni_aio_finish_error(aio, NNG_EBUSY);
- return;
- }
- if ((rv = nni_aio_schedule(aio, tcptran_ep_cancel, ep)) != 0) {
- nni_mtx_unlock(&ep->mtx);
- nni_aio_finish_error(aio, rv);
- return;
- }
- ep->useraio = aio;
- if (!ep->started) {
- ep->started = true;
- nng_stream_listener_accept(ep->listener, ep->connaio);
- } else {
- tcptran_ep_match(ep);
- }
- nni_mtx_unlock(&ep->mtx);
-}
-
-static nni_tran_pipe_ops tcptran_pipe_ops = {
- .p_init = tcptran_pipe_init,
- .p_fini = tcptran_pipe_fini,
- .p_stop = tcptran_pipe_stop,
- .p_send = tcptran_pipe_send,
- .p_recv = tcptran_pipe_recv,
- .p_close = tcptran_pipe_close,
- .p_peer = tcptran_pipe_peer,
- .p_getopt = tcptran_pipe_getopt,
-};
-
-static const nni_option tcptran_ep_opts[] = {
- {
- .o_name = NNG_OPT_RECVMAXSZ,
- .o_get = tcptran_ep_get_recvmaxsz,
- .o_set = tcptran_ep_set_recvmaxsz,
- },
- {
- .o_name = NNG_OPT_URL,
- .o_get = tcptran_ep_get_url,
- },
- // terminate list
- {
- .o_name = NULL,
- },
-};
-
-static int
-tcptran_dialer_getopt(
- void *arg, const char *name, void *buf, size_t *szp, nni_type t)
-{
- tcptran_ep *ep = arg;
- int rv;
-
- rv = nni_stream_dialer_get(ep->dialer, name, buf, szp, t);
- if (rv == NNG_ENOTSUP) {
- rv = nni_getopt(tcptran_ep_opts, name, ep, buf, szp, t);
- }
- return (rv);
-}
-
-static int
-tcptran_dialer_setopt(
- void *arg, const char *name, const void *buf, size_t sz, nni_type t)
-{
- tcptran_ep *ep = arg;
- int rv;
-
- rv = nni_stream_dialer_set(ep->dialer, name, buf, sz, t);
- if (rv == NNG_ENOTSUP) {
- rv = nni_setopt(tcptran_ep_opts, name, ep, buf, sz, t);
- }
- return (rv);
-}
-
-static int
-tcptran_listener_getopt(
- void *arg, const char *name, void *buf, size_t *szp, nni_type t)
-{
- tcptran_ep *ep = arg;
- int rv;
-
- rv = nni_stream_listener_get(ep->listener, name, buf, szp, t);
- if (rv == NNG_ENOTSUP) {
- rv = nni_getopt(tcptran_ep_opts, name, ep, buf, szp, t);
- }
- return (rv);
-}
-
-static int
-tcptran_listener_setopt(
- void *arg, const char *name, const void *buf, size_t sz, nni_type t)
-{
- tcptran_ep *ep = arg;
- int rv;
-
- rv = nni_stream_listener_set(ep->listener, name, buf, sz, t);
- if (rv == NNG_ENOTSUP) {
- rv = nni_setopt(tcptran_ep_opts, name, ep, buf, sz, t);
- }
- return (rv);
-}
-
-static nni_tran_dialer_ops tcptran_dialer_ops = {
- .d_init = tcptran_dialer_init,
- .d_fini = tcptran_ep_fini,
- .d_connect = tcptran_ep_connect,
- .d_close = tcptran_ep_close,
- .d_getopt = tcptran_dialer_getopt,
- .d_setopt = tcptran_dialer_setopt,
-};
-
-static nni_tran_listener_ops tcptran_listener_ops = {
- .l_init = tcptran_listener_init,
- .l_fini = tcptran_ep_fini,
- .l_bind = tcptran_ep_bind,
- .l_accept = tcptran_ep_accept,
- .l_close = tcptran_ep_close,
- .l_getopt = tcptran_listener_getopt,
- .l_setopt = tcptran_listener_setopt,
-};
-
-static nni_tran tcp_tran = {
- .tran_version = NNI_TRANSPORT_VERSION,
- .tran_scheme = "tcp",
- .tran_dialer = &tcptran_dialer_ops,
- .tran_listener = &tcptran_listener_ops,
- .tran_pipe = &tcptran_pipe_ops,
- .tran_init = tcptran_init,
- .tran_fini = tcptran_fini,
-};
-
-static nni_tran tcp4_tran = {
- .tran_version = NNI_TRANSPORT_VERSION,
- .tran_scheme = "tcp4",
- .tran_dialer = &tcptran_dialer_ops,
- .tran_listener = &tcptran_listener_ops,
- .tran_pipe = &tcptran_pipe_ops,
- .tran_init = tcptran_init,
- .tran_fini = tcptran_fini,
-};
-
-static nni_tran tcp6_tran = {
- .tran_version = NNI_TRANSPORT_VERSION,
- .tran_scheme = "tcp6",
- .tran_dialer = &tcptran_dialer_ops,
- .tran_listener = &tcptran_listener_ops,
- .tran_pipe = &tcptran_pipe_ops,
- .tran_init = tcptran_init,
- .tran_fini = tcptran_fini,
-};
-
-int
-nng_tcp_register(void)
-{
- int rv;
- if (((rv = nni_tran_register(&tcp_tran)) != 0) ||
- ((rv = nni_tran_register(&tcp4_tran)) != 0) ||
- ((rv = nni_tran_register(&tcp6_tran)) != 0)) {
- return (rv);
- }
- return (0);
-}
diff --git a/src/transport/tcp/tcp_test.c b/src/transport/tcp/tcp_test.c
deleted file mode 100644
index d23227d7..00000000
--- a/src/transport/tcp/tcp_test.c
+++ /dev/null
@@ -1,297 +0,0 @@
-//
-// Copyright 2020 Staysail Systems, Inc. <info@staysail.tech>
-// Copyright 2018 Capitar IT Group BV <info@capitar.com>
-// Copyright 2018 Devolutions <info@devolutions.net>
-// 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>
-
-// TCP tests.
-
-static void
-test_tcp_wild_card_connect_fail(void)
-{
- nng_socket s;
- char addr[NNG_MAXADDRLEN];
-
- NUTS_OPEN(s);
- (void) snprintf(addr, sizeof(addr), "tcp://*:%u", nuts_next_port());
- NUTS_FAIL(nng_dial(s, addr, NULL, 0), NNG_EADDRINVAL);
- NUTS_CLOSE(s);
-}
-
-void
-test_tcp_wild_card_bind(void)
-{
- nng_socket s1;
- nng_socket s2;
- char addr[NNG_MAXADDRLEN];
- uint16_t port;
-
- port = nuts_next_port();
-
- NUTS_OPEN(s1);
- NUTS_OPEN(s2);
- (void) snprintf(addr, sizeof(addr), "tcp4://*:%u", port);
- NUTS_PASS(nng_listen(s1, addr, NULL, 0));
- (void) snprintf(addr, sizeof(addr), "tcp://127.0.0.1:%u", port);
- NUTS_PASS(nng_dial(s2, addr, NULL, 0));
- NUTS_CLOSE(s2);
- NUTS_CLOSE(s1);
-}
-
-void
-test_tcp_local_address_connect(void)
-{
-
- nng_socket s1;
- nng_socket s2;
- char addr[NNG_MAXADDRLEN];
- uint16_t port;
-
- NUTS_OPEN(s1);
- NUTS_OPEN(s2);
- port = nuts_next_port();
- (void) snprintf(addr, sizeof(addr), "tcp://127.0.0.1:%u", port);
- NUTS_PASS(nng_listen(s1, addr, NULL, 0));
- (void) snprintf(
- addr, sizeof(addr), "tcp://127.0.0.1;127.0.0.1:%u", port);
- NUTS_PASS(nng_dial(s2, addr, NULL, 0));
- NUTS_CLOSE(s2);
- NUTS_CLOSE(s1);
-}
-
-void
-test_tcp_port_zero_bind(void)
-{
- nng_socket s1;
- nng_socket s2;
- nng_sockaddr sa;
- nng_listener l;
- char * addr;
-
- NUTS_OPEN(s1);
- NUTS_OPEN(s2);
- NUTS_PASS(nng_listen(s1, "tcp://127.0.0.1:0", &l, 0));
- NUTS_PASS(nng_listener_get_string(l, NNG_OPT_URL, &addr));
- NUTS_TRUE(memcmp(addr, "tcp://", 6) == 0);
- NUTS_PASS(nng_listener_get_addr(l, NNG_OPT_LOCADDR, &sa));
- NUTS_TRUE(sa.s_in.sa_family == NNG_AF_INET);
- NUTS_TRUE(sa.s_in.sa_port != 0);
- NUTS_TRUE(sa.s_in.sa_addr = nuts_be32(0x7f000001));
- NUTS_PASS(nng_dial(s2, addr, NULL, 0));
- nng_strfree(addr);
- NUTS_CLOSE(s2);
- NUTS_CLOSE(s1);
-}
-
-void
-test_tcp_bad_local_interface(void)
-{
- nng_socket s1;
-
- NUTS_OPEN(s1);
- NUTS_FAIL(nng_dial(s1, "tcp://bogus1;127.0.0.1:80", NULL, 0),
- NNG_EADDRINVAL);
- NUTS_CLOSE(s1);
-}
-
-void
-test_tcp_non_local_address(void)
-{
- nng_socket s1;
-
- NUTS_OPEN(s1);
- NUTS_FAIL(nng_dial(s1, "tcp://8.8.8.8;127.0.0.1:80", NULL, 0),
- NNG_EADDRINVAL);
- NUTS_CLOSE(s1);
-}
-
-void
-test_tcp_malformed_address(void)
-{
- nng_socket s1;
-
- NUTS_OPEN(s1);
- NUTS_FAIL(
- nng_dial(s1, "tcp://127.0.0.1", NULL, 0), NNG_EADDRINVAL);
- NUTS_FAIL(
- nng_dial(s1, "tcp://127.0.0.1.32", NULL, 0), NNG_EADDRINVAL);
- NUTS_FAIL(
- nng_dial(s1, "tcp://127.0.x.1.32", NULL, 0), NNG_EADDRINVAL);
- NUTS_FAIL(
- nng_listen(s1, "tcp://127.0.0.1.32", NULL, 0), NNG_EADDRINVAL);
- NUTS_FAIL(
- nng_listen(s1, "tcp://127.0.x.1.32", NULL, 0), NNG_EADDRINVAL);
- NUTS_CLOSE(s1);
-}
-
-void
-test_tcp_no_delay_option(void)
-{
- nng_socket s;
- nng_dialer d;
- nng_listener l;
- bool v;
- int x;
- char *addr;
-
- NUTS_ADDR(addr, "tcp");
-
- NUTS_OPEN(s);
-#ifndef NNG_ELIDE_DEPRECATED
- NUTS_PASS(nng_socket_get_bool(s, NNG_OPT_TCP_NODELAY, &v));
- NUTS_TRUE(v);
-#endif
- NUTS_PASS(nng_dialer_create(&d, s, addr));
- NUTS_PASS(nng_dialer_get_bool(d, NNG_OPT_TCP_NODELAY, &v));
- NUTS_TRUE(v);
- NUTS_PASS(nng_dialer_set_bool(d, NNG_OPT_TCP_NODELAY, false));
- NUTS_PASS(nng_dialer_get_bool(d, NNG_OPT_TCP_NODELAY, &v));
- NUTS_TRUE(v == false);
- NUTS_FAIL(
- nng_dialer_get_int(d, NNG_OPT_TCP_NODELAY, &x), NNG_EBADTYPE);
- x = 0;
- NUTS_FAIL(
- nng_dialer_set_int(d, NNG_OPT_TCP_NODELAY, x), NNG_EBADTYPE);
- // This assumes sizeof (bool) != sizeof (int)
- if (sizeof(bool) != sizeof(int)) {
- NUTS_FAIL(
- nng_dialer_set(d, NNG_OPT_TCP_NODELAY, &x, sizeof(x)),
- NNG_EINVAL);
- }
-
- NUTS_PASS(nng_listener_create(&l, s, addr));
- NUTS_PASS(nng_listener_get_bool(l, NNG_OPT_TCP_NODELAY, &v));
- NUTS_TRUE(v == true);
- x = 0;
- NUTS_FAIL(
- nng_listener_set_int(l, NNG_OPT_TCP_NODELAY, x), NNG_EBADTYPE);
- // This assumes sizeof (bool) != sizeof (int)
- NUTS_FAIL(nng_listener_set(l, NNG_OPT_TCP_NODELAY, &x, sizeof(x)),
- NNG_EINVAL);
-
- NUTS_PASS(nng_dialer_close(d));
- NUTS_PASS(nng_listener_close(l));
-
- // Make sure socket wide defaults apply.
-#ifndef NNG_ELIDE_DEPRECATED
- NUTS_PASS(nng_socket_set_bool(s, NNG_OPT_TCP_NODELAY, true));
- v = false;
- NUTS_PASS(nng_socket_get_bool(s, NNG_OPT_TCP_NODELAY, &v));
- NUTS_TRUE(v);
- NUTS_PASS(nng_socket_set_bool(s, NNG_OPT_TCP_NODELAY, false));
- NUTS_PASS(nng_dialer_create(&d, s, addr));
- NUTS_PASS(nng_dialer_get_bool(d, NNG_OPT_TCP_NODELAY, &v));
- NUTS_TRUE(v == false);
-#endif
- NUTS_CLOSE(s);
-}
-
-void
-test_tcp_keep_alive_option(void)
-{
- nng_socket s;
- nng_dialer d;
- nng_listener l;
- bool v;
- int x;
- char *addr;
-
- NUTS_ADDR(addr, "tcp");
- NUTS_OPEN(s);
-#ifndef NNG_ELIDE_DEPRECATED
- NUTS_PASS(nng_socket_get_bool(s, NNG_OPT_TCP_KEEPALIVE, &v));
- NUTS_TRUE(v == false);
-#endif
- NUTS_PASS(nng_dialer_create(&d, s, addr));
- NUTS_PASS(nng_dialer_get_bool(d, NNG_OPT_TCP_KEEPALIVE, &v));
- NUTS_TRUE(v == false);
- NUTS_PASS(nng_dialer_set_bool(d, NNG_OPT_TCP_KEEPALIVE, true));
- NUTS_PASS(nng_dialer_get_bool(d, NNG_OPT_TCP_KEEPALIVE, &v));
- NUTS_TRUE(v);
- NUTS_FAIL(
- nng_dialer_get_int(d, NNG_OPT_TCP_KEEPALIVE, &x), NNG_EBADTYPE);
- x = 1;
- NUTS_FAIL(
- nng_dialer_set_int(d, NNG_OPT_TCP_KEEPALIVE, x), NNG_EBADTYPE);
-
- NUTS_PASS(nng_listener_create(&l, s, addr));
- NUTS_PASS(nng_listener_get_bool(l, NNG_OPT_TCP_KEEPALIVE, &v));
- NUTS_TRUE(v == false);
- x = 1;
- NUTS_FAIL(
- nng_listener_set_int(l, NNG_OPT_TCP_KEEPALIVE, x), NNG_EBADTYPE);
-
- NUTS_PASS(nng_dialer_close(d));
- NUTS_PASS(nng_listener_close(l));
-
- // Make sure socket wide defaults apply.
-#ifndef NNG_ELIDE_DEPRECATED
- NUTS_PASS(nng_socket_set_bool(s, NNG_OPT_TCP_KEEPALIVE, false));
- v = true;
- NUTS_PASS(nng_socket_get_bool(s, NNG_OPT_TCP_KEEPALIVE, &v));
- NUTS_TRUE(v == false);
- NUTS_PASS(nng_socket_set_bool(s, NNG_OPT_TCP_KEEPALIVE, true));
- NUTS_PASS(nng_dialer_create(&d, s, addr));
- NUTS_PASS(nng_dialer_get_bool(d, NNG_OPT_TCP_KEEPALIVE, &v));
- NUTS_TRUE(v);
-#endif
- NUTS_CLOSE(s);
-}
-
-void
-test_tcp_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, "tcp");
-
- 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_PASS(nng_close(s0));
- NUTS_CLOSE(s1);
-}
-
-NUTS_TESTS = {
-
- { "tcp wild card connect fail", test_tcp_wild_card_connect_fail },
- { "tcp wild card bind", test_tcp_wild_card_bind },
- { "tcp port zero bind", test_tcp_port_zero_bind },
- { "tcp local address connect", test_tcp_local_address_connect },
- { "tcp bad local interface", test_tcp_bad_local_interface },
- { "tcp non-local address", test_tcp_non_local_address },
- { "tcp malformed address", test_tcp_malformed_address },
- { "tcp no delay option", test_tcp_no_delay_option },
- { "tcp keep alive option", test_tcp_keep_alive_option },
- { "tcp recv max", test_tcp_recv_max },
- { NULL, NULL },
-}; \ No newline at end of file
generated by cgit v1.2.3-70-g09d2 (git 2.46.0) at 2026-02-11 00:27:46 +0000