diff options
| author | Garrett D'Amore <garrett@damore.org> | 2024-10-22 21:22:20 -0700 |
|---|---|---|
| committer | Garrett D'Amore <garrett@damore.org> | 2024-10-22 21:22:20 -0700 |
| commit | 4de579a2617bfa1ee501a1669f3c351511135b2f (patch) | |
| tree | c15c13b110b6c2fd1702ae6e0b2a23acefb9e15c /src/compat | |
| parent | beca672e36109bf8f49c260257882fc3e9cea034 (diff) | |
| download | nng-4de579a2617bfa1ee501a1669f3c351511135b2f.tar.gz nng-4de579a2617bfa1ee501a1669f3c351511135b2f.tar.bz2 nng-4de579a2617bfa1ee501a1669f3c351511135b2f.zip | |
Remove the libnanomsg compatibility layer
Diffstat (limited to 'src/compat')
| -rw-r--r-- | src/compat/nanomsg/CMakeLists.txt | 15 | ||||
| -rw-r--r-- | src/compat/nanomsg/compat_msg_test.c | 92 | ||||
| -rw-r--r-- | src/compat/nanomsg/compat_tcp_test.c | 233 | ||||
| -rw-r--r-- | src/compat/nanomsg/nn.c | 1336 | ||||
| -rw-r--r-- | src/compat/nanomsg/nuts_compat.h | 73 |
5 files changed, 0 insertions, 1749 deletions
diff --git a/src/compat/nanomsg/CMakeLists.txt b/src/compat/nanomsg/CMakeLists.txt deleted file mode 100644 index 51679b34..00000000 --- a/src/compat/nanomsg/CMakeLists.txt +++ /dev/null @@ -1,15 +0,0 @@ -# -# Copyright 2023 Staysail Systems, Inc. <info@staysail.tech> -# -# 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. -# - -nng_sources(nn.c) - -set(NNG_TEST_PREFIX ${NNG_TEST_PREFIX}.compat.nanomsg) - -nng_test(compat_msg_test) -nng_test(compat_tcp_test) diff --git a/src/compat/nanomsg/compat_msg_test.c b/src/compat/nanomsg/compat_msg_test.c deleted file mode 100644 index d037a0b4..00000000 --- a/src/compat/nanomsg/compat_msg_test.c +++ /dev/null @@ -1,92 +0,0 @@ -// -// Copyright 2023 Staysail Systems, Inc. <info@staysail.tech> -// -// 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 <nng/compat/nanomsg/nn.h> -#include <nng/compat/nanomsg/pair.h> -#include <nng/compat/nanomsg/tcp.h> - -#include "nuts_compat.h" - -#include <nuts.h> - -void -test_msg_alloc(void) -{ - char *msg; - msg = nn_allocmsg(1, 0); - NUTS_TRUE(msg != NULL); - NUTS_NN_PASS(nn_freemsg(msg)); -} - -void -test_msg_zero_length(void) -{ - char *msg; - msg = nn_allocmsg(0, 0); // empty message is invalid - NUTS_TRUE(msg == NULL); - NUTS_TRUE(nn_errno() == EINVAL); -} - -void -test_msg_overflow(void) -{ - char *msg; - msg = nn_allocmsg((size_t)-1, 0); // this will overflow - NUTS_TRUE(msg == NULL); - NUTS_TRUE(nn_errno() == EINVAL); -} - -void -test_msg_bad_type(void) -{ - char *msg; - msg = nn_allocmsg(0, 1); // we only support message type 0 - NUTS_TRUE(msg == NULL); - NUTS_TRUE(nn_errno() == EINVAL); -} - -void -test_msg_realloc(void) -{ - char *msg0; - char *msg1; - char *msg2; - char *msg3; - - msg0 = nn_allocmsg(5, 0); - NUTS_TRUE(msg0 != NULL); - - memcpy(msg0, "this", 5); - - msg1 = nn_reallocmsg(msg0, 65536); - NUTS_TRUE(msg1 != NULL); - NUTS_TRUE(msg1 != msg0); - NUTS_MATCH(msg1, "this"); - - msg1[65000] = 'A'; - - msg2 = nn_reallocmsg(msg1, 5); - NUTS_TRUE(msg2 == msg1); - - // test for overflow - msg3 = nn_reallocmsg(msg2, (size_t)-1); - NUTS_TRUE(msg3 == NULL); - NUTS_TRUE(nn_errno() == EINVAL); - - nn_freemsg(msg2); -} - -TEST_LIST = { - { "alloc msg", test_msg_alloc }, - { "zero length", test_msg_zero_length }, - { "invalid type", test_msg_bad_type }, - { "overflow", test_msg_overflow }, - { "reallocate msg", test_msg_realloc }, - { NULL, NULL }, -}; diff --git a/src/compat/nanomsg/compat_tcp_test.c b/src/compat/nanomsg/compat_tcp_test.c deleted file mode 100644 index a23c878d..00000000 --- a/src/compat/nanomsg/compat_tcp_test.c +++ /dev/null @@ -1,233 +0,0 @@ -/* - Copyright 2020 Staysail Systems, Inc. <info@staysail.tech> - Copyright (c) 2012 Martin Sustrik All rights reserved. - Copyright 2016 Franklin "Snaipe" Mathieu <franklinmathieu@gmail.com> - - Permission is hereby granted, free of charge, to any person obtaining a - copy of this software and associated documentation files (the "Software"), - to deal in the Software without restriction, including without limitation - the rights to use, copy, modify, merge, publish, distribute, sublicense, - and/or sell copies of the Software, and to permit persons to whom - the Software is furnished to do so, subject to the following conditions: - - The above copyright notice and this permission notice shall be included - in all copies or substantial portions of the Software. - - THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR - IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, - FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL - THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER - LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING - FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER - DEALINGS IN THE SOFTWARE. -*/ - -// This test is a pretty much completely rewritten version of the -// legacy nanomsg tcp test. The NNG test infrastructure is a bit more -// robust, and we take advantage of that. - -#include <nng/compat/nanomsg/nn.h> -#include <nng/compat/nanomsg/pair.h> -#include <nng/compat/nanomsg/tcp.h> - -#include "nuts_compat.h" - -#include <nuts.h> - -void -test_bind_and_close(void) -{ - int sb; - char *addr; - - NUTS_ADDR(addr, "tcp"); - - NUTS_TRUE((sb = nn_socket(AF_SP, NN_PAIR)) >= 0); - NUTS_TRUE(nn_bind(sb, addr) >= 0); - NUTS_TRUE(nn_close(sb) == 0); -} - -void -test_connect_and_close(void) -{ - int sc; - char *addr; - - NUTS_ADDR(addr, "tcp"); - NUTS_NN_PASS((sc = nn_socket(AF_SP, NN_PAIR)) >= 0); - NUTS_NN_PASS(nn_connect(sc, addr)); - NUTS_NN_PASS(nn_close(sc)); -} - -void -test_bind_and_connect(void) -{ - int sb, sc, p1, p2; - char *addr; - - NUTS_ADDR(addr, "tcp"); - NUTS_TRUE((sb = nn_socket(AF_SP, NN_PAIR)) >= 0); - NUTS_TRUE((sc = nn_socket(AF_SP, NN_PAIR)) >= 0); - NUTS_TRUE(sb != sc); - - NUTS_NN_MARRY_EX(sb, sc, addr, p1, p2); - - NUTS_NN_PASS(nn_close(sb)); - NUTS_NN_PASS(nn_close(sc)); -} - -void -test_bad_addresses(void) -{ - int s; - NUTS_TRUE((s = nn_socket(AF_SP, NN_PAIR)) >= 0); - - NUTS_NN_FAIL(nn_connect(s, "tcp://*:"), EINVAL); - NUTS_NN_FAIL(nn_connect(s, "tcp://*:1000000"), EINVAL); - NUTS_NN_FAIL(nn_connect(s, "tcp://*:some_port"), EINVAL); - NUTS_NN_FAIL(nn_connect(s, "tcp://127.0.0.1"), EINVAL); - NUTS_NN_FAIL(nn_connect(s, "tcp://:5555"), EINVAL); - NUTS_NN_FAIL(nn_connect(s, "tcp://abc.123.---.#:5555"), EINVAL); - - NUTS_NN_FAIL(nn_bind(s, "tcp://127.0.0.1:1000000"), EINVAL); - NUTS_NN_PASS(nn_close(s)); -} - -void -test_ping_pong(void) -{ - int sb, sc, p1, p2; - char *addr; - - NUTS_ADDR(addr, "tcp"); - NUTS_NN_PASS((sb = nn_socket(AF_SP, NN_PAIR))); - NUTS_NN_PASS((sc = nn_socket(AF_SP, NN_PAIR))); - NUTS_TRUE(sb != sc); - NUTS_NN_MARRY_EX(sc, sb, addr, p1, p2); - NUTS_TRUE(p1 >= 0); - NUTS_TRUE(p2 >= 0); - - /* Ping-pong test. */ - for (int i = 0; i != 100; ++i) { - - char buf[4]; - int n; - NUTS_NN_PASS(nn_send(sc, "ABC", 3, 0)); - NUTS_NN_PASS(n = nn_recv(sb, buf, 4, 0)); - NUTS_TRUE(n == 3); - NUTS_TRUE(memcmp(buf, "ABC", 3) == 0); - - NUTS_NN_PASS(nn_send(sb, "DEF", 3, 0)); - NUTS_NN_PASS(n = nn_recv(sc, buf, 4, 0)); - NUTS_TRUE(n == 3); - NUTS_TRUE(memcmp(buf, "DEF", 3) == 0); - } - - NUTS_NN_PASS(nn_close(sb)); - NUTS_NN_PASS(nn_close(sc)); -} - -void -test_pair_reject(void) -{ - int sb, sc, sd, p1, p2; - char *addr; - - NUTS_ADDR(addr, "tcp"); - - NUTS_TRUE((sb = nn_socket(AF_SP, NN_PAIR)) >= 0); - NUTS_TRUE((sc = nn_socket(AF_SP, NN_PAIR)) >= 0); - NUTS_TRUE((sd = nn_socket(AF_SP, NN_PAIR)) >= 0); - NUTS_TRUE(sb != sc); - - NUTS_NN_MARRY_EX(sc, sb, addr, p1, p2); - - NUTS_TRUE(nn_connect(sd, addr) >= 0); - NUTS_SLEEP(200); - - NUTS_NN_PASS(nn_close(sb)); - NUTS_NN_PASS(nn_close(sc)); - NUTS_NN_PASS(nn_close(sd)); -} - -void -test_addr_in_use(void) -{ - int sb, sc; - char *addr; - - NUTS_ADDR(addr, "tcp"); - NUTS_TRUE((sb = nn_socket(AF_SP, NN_PAIR)) >= 0); - NUTS_TRUE((sc = nn_socket(AF_SP, NN_PAIR)) >= 0); - NUTS_TRUE(sb != sc); - NUTS_NN_PASS(nn_bind(sb, addr)); - NUTS_NN_FAIL(nn_bind(sc, addr), EADDRINUSE); - - NUTS_NN_PASS(nn_close(sb)); - NUTS_NN_PASS(nn_close(sc)); -} - -void -test_max_recv_size(void) -{ - int sb, sc, p1, p2; - int opt; - int n; - size_t sz; - char buf[64]; - char *addr; - - NUTS_ADDR(addr, "tcp"); - - NUTS_TRUE((sb = nn_socket(AF_SP, NN_PAIR)) >= 0); - NUTS_TRUE((sc = nn_socket(AF_SP, NN_PAIR)) >= 0); - NUTS_TRUE(sb != sc); - opt = 100; - sz = sizeof(opt); - NUTS_NN_PASS(nn_setsockopt(sb, NN_SOL_SOCKET, NN_RCVTIMEO, &opt, sz)); - - /* Test that NN_RCVMAXSIZE can be -1, but not lower */ - sz = sizeof(opt); - opt = -1; - NUTS_NN_PASS( - nn_setsockopt(sb, NN_SOL_SOCKET, NN_RCVMAXSIZE, &opt, sz)); - - opt = -2; - NUTS_NN_FAIL( - nn_setsockopt(sb, NN_SOL_SOCKET, NN_RCVMAXSIZE, &opt, sz), EINVAL); - - opt = 4; - NUTS_NN_PASS( - nn_setsockopt(sb, NN_SOL_SOCKET, NN_RCVMAXSIZE, &opt, sz)); - opt = -5; - NUTS_NN_PASS( - nn_getsockopt(sb, NN_SOL_SOCKET, NN_RCVMAXSIZE, &opt, &sz)); - NUTS_TRUE(opt == 4); - NUTS_TRUE(sz == sizeof(opt)); - - NUTS_NN_MARRY_EX(sc, sb, addr, p1, p2); - - NUTS_NN_PASS(nn_send(sc, "ABC", 4, 0)); - NUTS_NN_PASS(nn_send(sc, "012345", 6, 0)); - - NUTS_NN_PASS(n = nn_recv(sb, buf, sizeof(buf), 0)); - NUTS_TRUE(n == 4); - NUTS_TRUE(strcmp(buf, "ABC") == 0); - - NUTS_NN_FAIL(nn_recv(sb, buf, sizeof(buf), 0), ETIMEDOUT); - - NUTS_NN_PASS(nn_close(sb)); - NUTS_NN_PASS(nn_close(sc)); -} - -TEST_LIST = { - { "compat tcp bind and close ", test_bind_and_close }, - { "compat tcp connect and close ", test_connect_and_close }, - { "compat tcp bind and connect ", test_bind_and_connect }, - { "compat tcp invalid addresses", test_bad_addresses }, - { "compat tcp ping pong", test_ping_pong }, - { "compat tcp pair reject", test_pair_reject }, - { "compat tcp addr in use", test_addr_in_use }, - { "compat tcp max recv size", test_max_recv_size }, - { NULL, NULL }, -}; diff --git a/src/compat/nanomsg/nn.c b/src/compat/nanomsg/nn.c deleted file mode 100644 index af4db5d3..00000000 --- a/src/compat/nanomsg/nn.c +++ /dev/null @@ -1,1336 +0,0 @@ -// -// Copyright 2023 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 "nng/compat/nanomsg/nn.h" - -// transports -#include "nng/compat/nanomsg/inproc.h" -#include "nng/compat/nanomsg/ipc.h" -#include "nng/compat/nanomsg/tcp.h" -#include "nng/compat/nanomsg/ws.h" - -// protocols -#include "nng/compat/nanomsg/bus.h" -#include "nng/compat/nanomsg/pair.h" -#include "nng/compat/nanomsg/pipeline.h" -#include "nng/compat/nanomsg/pubsub.h" -#include "nng/compat/nanomsg/reqrep.h" -#include "nng/compat/nanomsg/survey.h" - -// underlying NNG headers -#include "nng/nng.h" -#include "nng/protocol/bus0/bus.h" -#include "nng/protocol/pair0/pair.h" -#include "nng/protocol/pipeline0/pull.h" -#include "nng/protocol/pipeline0/push.h" -#include "nng/protocol/pubsub0/pub.h" -#include "nng/protocol/pubsub0/sub.h" -#include "nng/protocol/reqrep0/rep.h" -#include "nng/protocol/reqrep0/req.h" -#include "nng/protocol/survey0/respond.h" -#include "nng/protocol/survey0/survey.h" - -#include "core/nng_impl.h" - -#include <stdio.h> -#include <string.h> - -// This file supplies the legacy compatibility API. Applications should -// avoid using these if at all possible, and instead use the new style APIs. - -static const struct { - int nerr; - int perr; -} nn_errnos[] = { - // clang-format off - { NNG_EINTR, EINTR }, - { NNG_ENOMEM, ENOMEM }, - { NNG_EINVAL, EINVAL }, - { NNG_EBUSY, EBUSY }, - { NNG_ETIMEDOUT, ETIMEDOUT }, - { NNG_ECONNREFUSED, ECONNREFUSED }, - { NNG_ECLOSED, EBADF }, - { NNG_EAGAIN, EAGAIN }, - { NNG_ENOTSUP, ENOTSUP }, - { NNG_EADDRINUSE, EADDRINUSE }, - { NNG_ESTATE, EFSM }, - { NNG_ENOENT, ENOENT }, - { NNG_EPROTO, EPROTO }, - { NNG_EUNREACHABLE, EHOSTUNREACH }, - { NNG_EADDRINVAL, EINVAL }, - { NNG_EPERM, EACCES }, - { NNG_EMSGSIZE, EMSGSIZE }, - { NNG_ECONNABORTED, ECONNABORTED }, - { NNG_ECONNRESET, ECONNRESET }, - { NNG_ECANCELED, EBADF }, - { NNG_EEXIST, EEXIST }, - { NNG_EWRITEONLY, EACCES }, - { NNG_EREADONLY, EACCES }, - { NNG_ECRYPTO, EACCES }, - { NNG_EPEERAUTH, EACCES }, - { NNG_EBADTYPE, EINVAL }, - { NNG_EAMBIGUOUS, EINVAL }, - { NNG_ENOFILES, EMFILE }, - { NNG_ENOSPC, ENOSPC }, - { 0, 0 }, - // clang-format on -}; - -const char * -nn_strerror(int err) -{ - int i; - static char msgbuf[32]; - - for (i = 0; nn_errnos[i].perr != 0; i++) { - if (nn_errnos[i].perr == err) { - return (nng_strerror(nn_errnos[i].nerr)); - } - } - if (err == EIO) { - return ("Unknown I/O error"); - } - - // Arguably we could use strerror() here, but we should only - // be getting errnos we understand at this point. - (void) snprintf(msgbuf, sizeof(msgbuf), "Unknown error %d", err); - return (msgbuf); -} - -static void -nn_seterror(int err) -{ - int i; - - for (i = 0; nn_errnos[i].nerr != 0; i++) { - if (nn_errnos[i].nerr == err) { - errno = nn_errnos[i].perr; - return; - } - } - // No idea... - errno = EIO; -} - -int -nn_errno(void) -{ - return (errno); -} - -static const struct { - uint16_t p_id; - int (*p_open)(nng_socket *); - int (*p_open_raw)(nng_socket *); -} nn_protocols[] = { -#ifdef NNG_HAVE_BUS0 - { - .p_id = NN_BUS, - .p_open = nng_bus0_open, - .p_open_raw = nng_bus0_open_raw, - }, -#endif -#ifdef NNG_HAVE_PAIR0 - { - .p_id = NN_PAIR, - .p_open = nng_pair0_open, - .p_open_raw = nng_pair0_open_raw, - }, -#endif -#ifdef NNG_HAVE_PULL0 - { - .p_id = NN_PULL, - .p_open = nng_pull0_open, - .p_open_raw = nng_pull0_open_raw, - }, -#endif -#ifdef NNG_HAVE_PUSH0 - { - .p_id = NN_PUSH, - .p_open = nng_push0_open, - .p_open_raw = nng_push0_open_raw, - }, -#endif -#ifdef NNG_HAVE_PUB0 - { - .p_id = NN_PUB, - .p_open = nng_pub0_open, - .p_open_raw = nng_pub0_open_raw, - }, -#endif -#ifdef NNG_HAVE_SUB0 - { - .p_id = NN_SUB, - .p_open = nng_sub0_open, - .p_open_raw = nng_sub0_open_raw, - }, -#endif -#ifdef NNG_HAVE_REQ0 - { - .p_id = NN_REQ, - .p_open = nng_req0_open, - .p_open_raw = nng_req0_open_raw, - }, -#endif -#ifdef NNG_HAVE_REP0 - { - .p_id = NN_REP, - .p_open = nng_rep0_open, - .p_open_raw = nng_rep0_open_raw, - }, -#endif -#ifdef NNG_HAVE_SURVEYOR0 - { - .p_id = NN_SURVEYOR, - .p_open = nng_surveyor0_open, - .p_open_raw = nng_surveyor0_open_raw, - }, -#endif -#ifdef NNG_HAVE_RESPONDENT0 - { - .p_id = NN_RESPONDENT, - .p_open = nng_respondent0_open, - .p_open_raw = nng_respondent0_open_raw, - }, -#endif - { - .p_id = 0, - }, -}; - -int -nn_socket(int domain, int protocol) -{ - nng_socket sock; - int rv; - int i; - - if ((domain != AF_SP) && (domain != AF_SP_RAW)) { - errno = EAFNOSUPPORT; - return (-1); - } - - for (i = 0; nn_protocols[i].p_id != 0; i++) { - if (nn_protocols[i].p_id == protocol) { - break; - } - } - if (nn_protocols[i].p_open == NULL) { - errno = ENOTSUP; - return (-1); - } - - if (domain == AF_SP_RAW) { - rv = nn_protocols[i].p_open_raw(&sock); - } else { - rv = nn_protocols[i].p_open(&sock); - } - if (rv != 0) { - nn_seterror(rv); - return (-1); - } - - // Legacy sockets have Nagle disabled. - (void) nng_socket_set_bool(sock, NNG_OPT_TCP_NODELAY, false); - return ((int) sock.id); -} - -int -nn_close(int s) -{ - int rv; - nng_socket sid; - - sid.id = (uint32_t) s; - - if ((rv = nng_close(sid)) != 0) { - nn_seterror(rv); - return (-1); - } - return (0); -} - -int -nn_bind(int s, const char *addr) -{ - int rv; - nng_listener l; - nng_socket sid; - - sid.id = (uint32_t) s; - if ((rv = nng_listen(sid, addr, &l, 0)) != 0) { - nn_seterror(rv); - return (-1); - } - return ((int) l.id); -} - -int -nn_connect(int s, const char *addr) -{ - int rv; - nng_dialer d; - nng_socket sid; - - sid.id = (uint32_t) s; - if ((rv = nng_dial(sid, addr, &d, NNG_FLAG_NONBLOCK)) != 0) { - nn_seterror(rv); - return (-1); - } - return ((int) d.id); -} - -int -nn_shutdown(int s, int ep) -{ - int rv; - (void) s; // Unused - nng_dialer d; - nng_listener l; - - // Socket is wired into the endpoint... so passing a bad endpoint - // ID can result in affecting the wrong socket. But this requires - // a buggy application, and because we don't recycle endpoints - // until wrap, it's unlikely to actually come up in practice. - // Note that listeners and dialers share the same namespace - // in the core, so we can close either one this way. - - d.id = l.id = (uint32_t) ep; - if (((rv = nng_dialer_close(d)) != 0) && - ((rv = nng_listener_close(l)) != 0)) { - nn_seterror(rv); - return (-1); - } - return (0); -} - -void * -nn_allocmsg(size_t size, int type) -{ - nng_msg *msg; - int rv; - - // Validate type and non-zero size. This also checks for overflow. - if ((type != 0) || (size < 1) || ((size + sizeof(msg) < size))) { - nn_seterror(NNG_EINVAL); - return (NULL); - } - - // So our "messages" from nn are really going to be nng messages - // but to make this work, we use a bit of headroom in the message - // to stash the message header. - if ((rv = nng_msg_alloc(&msg, size)) != 0) { - nn_seterror(rv); - return (NULL); - } - - nng_msg_insert(msg, &msg, sizeof(msg)); - - // We are counting on the implementation of nn_msg_trim to not - // reallocate the message but just to leave the prefix inplace. - (void) nng_msg_trim(msg, sizeof(msg)); - - return (nng_msg_body(msg)); -} - -int -nn_freemsg(void *ptr) -{ - nng_msg *msg; - - msg = *(nng_msg **) (((char *) ptr) - sizeof(msg)); - nng_msg_free(msg); - return (0); -} - -void * -nn_reallocmsg(void *ptr, size_t len) -{ - nng_msg *msg; - int rv; - - if ((len + sizeof(msg)) < len) { - // overflowed! - nn_seterror(NNG_EINVAL); - return (NULL); - } - - // This counts on message bodies being aligned sensibly. - msg = *(nng_msg **) (((char *) ptr) - sizeof(msg)); - - // We need to realloc the requested len, plus size for our header. - if ((rv = nng_msg_realloc(msg, len)) != 0) { - // We don't free the old message. Code is free to cope - // as it sees fit. - nn_seterror(rv); - return (NULL); - } - // Stash the msg header pointer - nng_msg_insert(msg, &msg, sizeof(msg)); - nng_msg_trim(msg, sizeof(msg)); - return (nng_msg_body(msg)); -} - -static int -nn_flags(int flags) -{ - switch (flags) { - case 0: - return (0); - - case NN_DONTWAIT: - return (NNG_FLAG_NONBLOCK); - - default: - nn_seterror(NNG_EINVAL); - return (-1); - } -} - -int -nn_send(int s, const void *buf, size_t len, int flags) -{ - struct nn_iovec iov; - struct nn_msghdr hdr; - - iov.iov_base = (void *) buf; - iov.iov_len = len; - - hdr.msg_iov = &iov; - hdr.msg_iovlen = 1; - hdr.msg_control = NULL; - hdr.msg_controllen = 0; - - return (nn_sendmsg(s, &hdr, flags)); -} - -int -nn_recv(int s, void *buf, size_t len, int flags) -{ - struct nn_iovec iov; - struct nn_msghdr hdr; - - iov.iov_base = buf; - iov.iov_len = len; - - hdr.msg_iov = &iov; - hdr.msg_iovlen = 1; - hdr.msg_control = NULL; - hdr.msg_controllen = 0; - - return (nn_recvmsg(s, &hdr, flags)); -} - -int -nn_recvmsg(int s, struct nn_msghdr *mh, int flags) -{ - int rv; - nng_msg *msg; - size_t len; - int keep = 0; - nng_socket sid; - - if ((flags = nn_flags(flags)) == -1) { - return (-1); - } - if (mh == NULL) { - nn_seterror(NNG_EINVAL); - return (-1); - } - if (mh->msg_iovlen < 0) { - nn_seterror(NNG_EMSGSIZE); - return (-1); - } - - sid.id = (uint32_t) s; - if ((rv = nng_recvmsg(sid, &msg, flags)) != 0) { - nn_seterror(rv); - return (-1); - } - if ((mh->msg_iovlen == 1) && (mh->msg_iov[0].iov_len == NN_MSG)) { - // Receiver wants to have a dynamically allocated message. - // There can only be one of these. - if ((rv = nng_msg_insert(msg, &msg, sizeof(msg))) != 0) { - nng_msg_free(msg); - nn_seterror(rv); - return (-1); - } - nng_msg_trim(msg, sizeof(msg)); - *(void **) (mh->msg_iov[0].iov_base) = nng_msg_body(msg); - len = nng_msg_len(msg); - keep = 1; // Do not discard message! - } else { - // copyout to multiple iovecs. - char *ptr = nng_msg_body(msg); - len = nng_msg_len(msg); - - for (int i = 0; i < mh->msg_iovlen; i++) { - size_t n; - if ((n = mh->msg_iov[i].iov_len) == NN_MSG) { - // This is forbidden! - nn_seterror(NNG_EINVAL); - nng_msg_free(msg); - return (-1); - } - if (n > len) { - n = len; - } - memcpy(mh->msg_iov[i].iov_base, ptr, n); - len -= n; - ptr += n; - } - - // If we copied everything, len will be zero, otherwise, - // it represents the amount of data that we were unable to - // copyout. The caller is responsible for noticing this, - // as there is no API to pass this information out. - len = nng_msg_len(msg); - } - - // If the caller has requested control information (header details), - // we grab it. - if (mh->msg_control != NULL) { - char *cdata; - size_t clen; - size_t tlen; - size_t spsz; - struct nn_cmsghdr *hdr; - - spsz = nng_msg_header_len(msg); - clen = NN_CMSG_SPACE(sizeof(spsz) + spsz); - - if ((tlen = mh->msg_controllen) == NN_MSG) { - // Ideally we'd use the same msg, but we would need - // to set up reference counts on the message, so - // instead we just make a new message. - nng_msg *nmsg; - - rv = nng_msg_alloc(&nmsg, clen + sizeof(nmsg)); - if (rv != 0) { - nng_msg_free(msg); - nn_seterror(rv); - return (-1); - } - memcpy(nng_msg_body(nmsg), &nmsg, sizeof(nmsg)); - nng_msg_trim(nmsg, sizeof(nmsg)); - cdata = nng_msg_body(nmsg); - *(void **) mh->msg_control = cdata; - tlen = clen; - } else { - cdata = mh->msg_control; - memset(cdata, 0, - tlen > sizeof(*hdr) ? sizeof(*hdr) : tlen); - } - - if (clen <= tlen) { - uint8_t *ptr = NN_CMSG_DATA(cdata); - hdr = (void *) cdata; - hdr->cmsg_len = clen; - hdr->cmsg_level = PROTO_SP; - hdr->cmsg_type = SP_HDR; - - memcpy(ptr, &spsz, sizeof(spsz)); - ptr += sizeof(spsz); - memcpy(ptr, nng_msg_header(msg), spsz); - } - } - - if (!keep) { - nng_msg_free(msg); - } - return ((int) len); -} - -int -nn_sendmsg(int s, const struct nn_msghdr *mh, int flags) -{ - nng_msg *msg = NULL; - nng_msg *cmsg = NULL; - nng_socket sid; - char *cdata; - int keep = 0; - size_t sz; - int rv; - - sid.id = (uint32_t) s; - - if ((flags = nn_flags(flags)) == -1) { - return (-1); - } - - if (mh == NULL) { - nn_seterror(NNG_EINVAL); - return (-1); - } - - if (mh->msg_iovlen < 0) { - nn_seterror(NNG_EMSGSIZE); - return (-1); - } - - if ((mh->msg_iovlen == 1) && (mh->msg_iov[0].iov_len == NN_MSG)) { - char *bufp = *(char **) (mh->msg_iov[0].iov_base); - - msg = *(nng_msg **) (bufp - sizeof(msg)); - keep = 1; // keep the message on error - } else { - char *ptr; - int i; - - sz = 0; - // Get the total message size. - for (i = 0; i < mh->msg_iovlen; i++) { - sz += mh->msg_iov[i].iov_len; - } - if ((rv = nng_msg_alloc(&msg, sz)) != 0) { - nn_seterror(rv); - return (-1); - } - // Now copy it out. - ptr = nng_msg_body(msg); - for (i = 0; i < mh->msg_iovlen; i++) { - memcpy(ptr, mh->msg_iov[i].iov_base, - mh->msg_iov[i].iov_len); - ptr += mh->msg_iov[i].iov_len; - } - } - - // Now suck up the control data... - // This POSIX-inspired API is one of the most painful for - // usability we've ever seen. - cmsg = NULL; - if ((cdata = mh->msg_control) != NULL) { - size_t clen; - size_t offs; - size_t spsz; - unsigned char *data; - - if ((clen = mh->msg_controllen) == NN_MSG) { - // Underlying data is a message. This is awkward, - // because we have to copy the data, but we should - // only free this message on success. So we save the - // message now. - cdata = *(void **) cdata; - cmsg = *(nng_msg **) (cdata - sizeof(cmsg)); - clen = nng_msg_len(cmsg); - } else { - clen = mh->msg_controllen; - } - - offs = 0; - while ((offs + sizeof(NN_CMSG_LEN(0))) < clen) { - struct nn_cmsghdr *chdr = (void *) (cdata + offs); - if ((chdr->cmsg_level != PROTO_SP) || - (chdr->cmsg_type != SP_HDR)) { - offs += chdr->cmsg_len; - } - - // SP header in theory. Starts with size, then - // any backtrace details. - if (chdr->cmsg_len < sizeof(size_t)) { - offs += chdr->cmsg_len; - continue; - } - data = NN_CMSG_DATA(chdr); - memcpy(&spsz, data, sizeof(spsz)); - if ((spsz + sizeof(spsz)) > chdr->cmsg_len) { - // Truncated header? Ignore it. - offs += chdr->cmsg_len; - continue; - } - data += sizeof(spsz); - rv = nng_msg_header_append(msg, data, spsz); - if (rv != 0) { - if (!keep) { - nng_msg_free(msg); - } - nn_seterror(rv); - return (-1); - } - - break; - } - } - - sz = nng_msg_len(msg); - if ((rv = nng_sendmsg(sid, msg, flags)) != 0) { - if (!keep) { - nng_msg_free(msg); - } - nn_seterror(rv); - return (-1); - } - - if (cmsg != NULL) { - // We sent successfully, so free up the control message. - nng_msg_free(cmsg); - } - return ((int) sz); -} - -static int -nn_getdomain(nng_socket s, void *valp, size_t *szp) -{ - int i; - bool b; - int rv; - - if ((rv = nng_socket_get_bool(s, NNG_OPT_RAW, &b)) != 0) { - nn_seterror(rv); - return (-1); - } - i = b ? AF_SP_RAW : AF_SP; - memcpy(valp, &i, *szp < sizeof(int) ? *szp : sizeof(int)); - *szp = sizeof(int); - return (0); -} - -#ifndef NNG_PLATFORM_WINDOWS -#define SOCKET int -#endif - -static int -nn_getfd(nng_socket s, void *valp, size_t *szp, const char *opt) -{ - int ifd; - int rv; - SOCKET sfd; - - if ((rv = nng_socket_get_int(s, opt, &ifd)) != 0) { - nn_seterror(rv); - return (-1); - } - sfd = (SOCKET) ifd; - memcpy(valp, &sfd, *szp < sizeof(sfd) ? *szp : sizeof(sfd)); - *szp = sizeof(sfd); - return (0); -} - -static int -nn_getrecvfd(nng_socket s, void *valp, size_t *szp) -{ - return (nn_getfd(s, valp, szp, NNG_OPT_RECVFD)); -} - -static int -nn_getsendfd(nng_socket s, void *valp, size_t *szp) -{ - return (nn_getfd(s, valp, szp, NNG_OPT_SENDFD)); -} - -static int -nn_getzero(nng_socket s, void *valp, size_t *szp) -{ - int zero = 0; - NNI_ARG_UNUSED(s); - memcpy(valp, &zero, *szp < sizeof(zero) ? *szp : sizeof(zero)); - *szp = sizeof(zero); - return (0); -} - -static int -nn_setignore(nng_socket s, const void *valp, size_t sz) -{ - NNI_ARG_UNUSED(valp); - NNI_ARG_UNUSED(s); - if (sz != sizeof(int)) { - nn_seterror(NNG_EINVAL); - return (-1); - } - return (0); -} - -static int -nn_getwsmsgtype(nng_socket s, void *valp, size_t *szp) -{ - int val = NN_WS_MSG_TYPE_BINARY; - NNI_ARG_UNUSED(s); - memcpy(valp, &val, *szp < sizeof(val) ? *szp : sizeof(val)); - *szp = sizeof(val); - return (0); -} - -static int -nn_setwsmsgtype(nng_socket s, const void *valp, size_t sz) -{ - int val; - NNI_ARG_UNUSED(s); - if (sz != sizeof(val)) { - nn_seterror(NNG_EINVAL); - return (-1); - } - memcpy(&val, valp, sizeof(val)); - if (val != NN_WS_MSG_TYPE_BINARY) { - nn_seterror(NNG_EINVAL); - return (-1); - } - return (0); -} - -static int -nn_settcpnodelay(nng_socket s, const void *valp, size_t sz) -{ - bool val; - int ival; - int rv; - - if (sz != sizeof(ival)) { - errno = EINVAL; - return (-1); - } - memcpy(&ival, valp, sizeof(ival)); - switch (ival) { - case 0: - val = false; - break; - case 1: - val = true; - break; - default: - nn_seterror(NNG_EINVAL); - return (-1); - } - - if ((rv = nng_socket_set_bool(s, NNG_OPT_TCP_NODELAY, val)) != 0) { - nn_seterror(rv); - return (-1); - } - return (0); -} - -static int -nn_gettcpnodelay(nng_socket s, void *valp, size_t *szp) -{ - bool val; - int ival; - int rv; - - if ((rv = nng_socket_get_bool(s, NNG_OPT_TCP_NODELAY, &val)) != 0) { - nn_seterror(rv); - return (-1); - } - ival = val ? 1 : 0; - memcpy(valp, &ival, *szp < sizeof(ival) ? *szp : sizeof(ival)); - *szp = sizeof(ival); - return (0); -} - -static int -nn_getrcvbuf(nng_socket s, void *valp, size_t *szp) -{ - int cnt; - int rv; - - if ((rv = nng_socket_get_int(s, NNG_OPT_RECVBUF, &cnt)) != 0) { - nn_seterror(rv); - return (-1); - } - cnt *= 1024; - memcpy(valp, &cnt, *szp < sizeof(cnt) ? *szp : sizeof(cnt)); - *szp = sizeof(cnt); - return (0); -} - -static int -nn_setrcvbuf(nng_socket s, const void *valp, size_t sz) -{ - int cnt; - int rv; - - if (sz != sizeof(cnt)) { - errno = EINVAL; - return (-1); - } - memcpy(&cnt, valp, sizeof(cnt)); - // Round up to a whole number of kilobytes, then divide by kB to - // go from buffer size in bytes to messages. This is a coarse - // estimate, and assumes messages are 1kB on average. - cnt += 1023; - cnt /= 1024; - if ((rv = nng_socket_set_int(s, NNG_OPT_RECVBUF, cnt)) != 0) { - nn_seterror(rv); - return (-1); - } - return (0); -} - -static int -nn_getsndbuf(nng_socket s, void *valp, size_t *szp) -{ - int cnt; - int rv; - - if ((rv = nng_socket_get_int(s, NNG_OPT_SENDBUF, &cnt)) != 0) { - nn_seterror(rv); - return (-1); - } - cnt *= 1024; - memcpy(valp, &cnt, *szp < sizeof(cnt) ? *szp : sizeof(cnt)); - *szp = sizeof(cnt); - return (0); -} - -static int -nn_setsndbuf(nng_socket s, const void *valp, size_t sz) -{ - int cnt; - int rv; - - if (sz != sizeof(cnt)) { - errno = EINVAL; - return (-1); - } - memcpy(&cnt, valp, sizeof(cnt)); - // Round up to a whole number of kilobytes, then divide by kB to - // go from buffer size in bytes to messages. This is a coarse - // estimate, and assumes messages are 1kB on average. - cnt += 1023; - cnt /= 1024; - if ((rv = nng_socket_set_int(s, NNG_OPT_SENDBUF, cnt)) != 0) { - nn_seterror(rv); - return (-1); - } - return (0); -} - -static int -nn_setrcvmaxsz(nng_socket s, const void *valp, size_t sz) -{ - int ival; - size_t val; - int rv; - - if (sz != sizeof(ival)) { - errno = EINVAL; - return (-1); - } - memcpy(&ival, valp, sizeof(ival)); - if (ival == -1) { - val = 0; - } else if (ival >= 0) { - // Note that if the user sets 0, it disables the limit. - // This is a different semantic. - val = (size_t) ival; - } else { - errno = EINVAL; - return (-1); - } - if ((rv = nng_socket_set_size(s, NNG_OPT_RECVMAXSZ, val)) != 0) { - nn_seterror(rv); - return (-1); - } - return (0); -} - -static int -nn_getrcvmaxsz(nng_socket s, void *valp, size_t *szp) -{ - int ival; - int rv; - size_t val; - - if ((rv = nng_socket_get_size(s, NNG_OPT_RECVMAXSZ, &val)) != 0) { - nn_seterror(rv); - return (-1); - } - // Legacy uses -1 to mean unlimited. New code uses 0. Note that - // as a consequence, we can't set a message limit of zero. - // We report any size beyond 2GB as effectively unlimited. - // There is an implicit assumption here that ints are 32-bits, - // but that's generally true of any platform we support. - if ((val == 0) || (val > 0x7FFFFFFF)) { - ival = -1; - } else { - ival = (int) val; - } - memcpy(valp, &ival, *szp < sizeof(ival) ? *szp : sizeof(ival)); - *szp = sizeof(ival); - return (0); -} - -// options which we convert -- most of the array is initialized at run time. -static const struct { - int nnlevel; - int nnopt; - const char *opt; - int (*get)(nng_socket, void *, size_t *); - int (*set)(nng_socket, const void *, size_t); -} options[] = { - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_LINGER, - .get = nn_getzero, - .set = nn_setignore, - }, // review - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_DOMAIN, - .get = nn_getdomain, - .set = NULL, - }, - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_RCVBUF, - .get = nn_getrcvbuf, - .set = nn_setrcvbuf, - }, - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_SNDBUF, - .get = nn_getsndbuf, - .set = nn_setsndbuf, - }, - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_RECONNECT_IVL, - .opt = NNG_OPT_RECONNMINT, - }, - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_RECONNECT_IVL_MAX, - .opt = NNG_OPT_RECONNMAXT, - }, - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_SNDFD, - .opt = NNG_OPT_SENDFD, - .get = nn_getsendfd, - }, - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_RCVFD, - .opt = NNG_OPT_RECVFD, - .get = nn_getrecvfd, - }, - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_RCVMAXSIZE, - .get = nn_getrcvmaxsz, - .set = nn_setrcvmaxsz, - }, - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_MAXTTL, - .opt = NNG_OPT_MAXTTL, - }, - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_RCVTIMEO, - .opt = NNG_OPT_RECVTIMEO, - }, - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_SNDTIMEO, - .opt = NNG_OPT_SENDTIMEO, - }, - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_PROTOCOL, - .opt = NNG_OPT_PROTO, - }, - { - .nnlevel = NN_SOL_SOCKET, - .nnopt = NN_SOCKET_NAME, - .opt = NNG_OPT_SOCKNAME, - }, - { - .nnlevel = NN_REQ, - .nnopt = NN_REQ_RESEND_IVL, - .opt = NNG_OPT_REQ_RESENDTIME, - }, - { - .nnlevel = NN_SUB, - .nnopt = NN_SUB_SUBSCRIBE, - .opt = NNG_OPT_SUB_SUBSCRIBE, - }, - { - .nnlevel = NN_SUB, - .nnopt = NN_SUB_UNSUBSCRIBE, - .opt = NNG_OPT_SUB_UNSUBSCRIBE, - }, - { - .nnlevel = NN_SURVEYOR, - .nnopt = NN_SURVEYOR_DEADLINE, - .opt = NNG_OPT_SURVEYOR_SURVEYTIME, - }, - { - .nnlevel = NN_TCP, - .nnopt = NN_TCP_NODELAY, - .get = nn_gettcpnodelay, - .set = nn_settcpnodelay, - }, - { - .nnlevel = NN_WS, - .nnopt = NN_WS_MSG_TYPE, - .get = nn_getwsmsgtype, - .set = nn_setwsmsgtype, - } - // XXX: IPV4ONLY, SNDPRIO, RCVPRIO -}; - -int -nn_getsockopt(int s, int nnlevel, int nnopt, void *valp, size_t *szp) -{ - const char *name = NULL; - int (*get)(nng_socket, void *, size_t *) = NULL; - int rv; - nng_socket sid; - - sid.id = (uint32_t) s; - - for (unsigned i = 0; i < sizeof(options) / sizeof(options[0]); i++) { - if ((options[i].nnlevel == nnlevel) && - (options[i].nnopt == nnopt)) { - get = options[i].get; - name = options[i].opt; - break; - } - } - - if (get != NULL) { - return (get(sid, valp, szp)); - } - - if (name == NULL) { - errno = ENOPROTOOPT; - return (-1); - } - - if ((rv = nng_socket_get(sid, name, valp, szp)) != 0) { - nn_seterror(rv); - return (-1); - } - - return (0); -} - -int -nn_setsockopt(int s, int nnlevel, int nnopt, const void *valp, size_t sz) -{ - nng_socket sid; - const char *name = NULL; - int (*set)(nng_socket, const void *, size_t) = NULL; - int rv; - - sid.id = (uint32_t) s; - - for (unsigned i = 0; i < sizeof(options) / sizeof(options[0]); i++) { - if ((options[i].nnlevel == nnlevel) && - (options[i].nnopt == nnopt)) { - - set = options[i].set; - name = options[i].opt; - break; - } - } - - if (set != NULL) { - return (set(sid, valp, sz)); - } - - if (name == NULL) { - errno = ENOPROTOOPT; - return (-1); - } - - if ((rv = nng_socket_set(sid, name, valp, sz)) != 0) { - nn_seterror(rv); - return (-1); - } - if ((nnlevel == NN_REQ) && (nnopt == NN_REQ_RESEND_IVL)) { - // Only one context here, so it won't be too bad to tick - // as quickly as this, and it avoids some possible friction - // (e.g. with legacy tests). - (void) nng_socket_set_ms(sid, NNG_OPT_REQ_RESENDTICK, 10); - } - - return (0); -} - -struct nn_cmsghdr * -nn_cmsg_next(struct nn_msghdr *mh, struct nn_cmsghdr *first) -{ - size_t clen; - char *data; - - // We only support SP headers, so there can be at most one header. - if (first != NULL) { - return (NULL); - } - if ((clen = mh->msg_controllen) == NN_MSG) { - nng_msg *msg; - data = *((void **) (mh->msg_control)); - msg = *(nng_msg **) (data - sizeof(msg)); - clen = nng_msg_len(msg); - } else { - data = mh->msg_control; - } - - if (first == NULL) { - first = (void *) data; - } else { - first = first + first->cmsg_len; - } - - if (((char *) first + sizeof(*first)) > (data + clen)) { - return (NULL); - } - return (first); -} - -int -nn_device(int s1, int s2) -{ - int rv; - nng_socket sid1; - nng_socket sid2; - - sid1.id = (uint32_t) s1; - sid2.id = (uint32_t) s2; - - rv = nng_device(sid1, sid2); - // rv must always be nonzero - nn_seterror(rv); - return (-1); -} - -// Windows stuff. -#ifdef NNG_PLATFORM_WINDOWS -#define poll WSAPoll -#ifndef WIN32_LEAN_AND_MEAN -#define WIN32_LEAN_AND_MEAN -#endif -#include <mswsock.h> -#include <windows.h> -#elif defined NNG_PLATFORM_POSIX -#include <poll.h> -#endif - -int -nn_poll(struct nn_pollfd *fds, int nfds, int timeout) -{ -// This function is rather unfortunate. poll() is available -// on POSIX, and on Windows as WSAPoll. On other systems it might -// not exist at all. We could also benefit from using a notification -// that didn't have to access file descriptors... sort of access to -// the pollable element on the socket. We don't have that, so we -// just use poll. This function is definitely suboptimal compared to -// using callbacks. -#if defined(NNG_PLATFORM_WINDOWS) || defined(NNG_PLATFORM_POSIX) - struct pollfd *pfd; - int npfd; - int rv; - - if ((pfd = NNI_ALLOC_STRUCTS(pfd, nfds * 2)) == NULL) { - errno = ENOMEM; - return (-1); - } - - // First prepare the master polling structure. - npfd = 0; - for (int i = 0; i < nfds; i++) { - int fd; - if (fds[i].events & NN_POLLIN) { - nng_socket s; - s.id = fds[i].fd; - if ((rv = nng_socket_get_int( - s, NNG_OPT_RECVFD, &fd)) != 0) { - nn_seterror(rv); - NNI_FREE_STRUCTS(pfd, nfds * 2); - return (-1); - } -#ifdef NNG_PLATFORM_WINDOWS - pfd[npfd].fd = (SOCKET) fd; -#else - pfd[npfd].fd = fd; -#endif - pfd[npfd].events = POLLIN; - npfd++; - } - if (fds[i].events & NN_POLLOUT) { - nng_socket s; - s.id = fds[i].fd; - if ((rv = nng_socket_get_int( - s, NNG_OPT_SENDFD, &fd)) != 0) { - nn_seterror(rv); - NNI_FREE_STRUCTS(pfd, nfds * 2); - return (-1); - } -#ifdef NNG_PLATFORM_WINDOWS - pfd[npfd].fd = (SOCKET) fd; -#else - pfd[npfd].fd = fd; -#endif - pfd[npfd].events = POLLIN; - npfd++; - } - } - - rv = poll(pfd, npfd, timeout); - if (rv < 0) { - int e = errno; - NNI_FREE_STRUCTS(pfd, nfds * 2); - errno = e; - return (-1); - } - - // Now update the nn_poll from the system poll. - npfd = 0; - rv = 0; - for (int i = 0; i < nfds; i++) { - fds[i].revents = 0; - if (fds[i].events & NN_POLLIN) { - if (pfd[npfd].revents & POLLIN) { - fds[i].revents |= NN_POLLIN; - } - npfd++; - } - if (fds[i].events & NN_POLLOUT) { - if (pfd[npfd].revents & POLLIN) { - fds[i].revents |= NN_POLLOUT; - } - npfd++; - } - if (fds[i].revents) { - rv++; - } - } - NNI_FREE_STRUCTS(pfd, nfds * 2); - return (rv); - -#else // NNG_PLATFORM_WINDOWS or NNG_PLATFORM_POSIX - NNI_ARG_UNUSED(fds); - NNI_ARG_UNUSED(nfds); - NNI_ARG_UNUSED(timeout); - errno = ENOTSUP; - return (-1); -#endif -} - -// nn_term is suitable only for shutting down the entire library, -// and is not thread-safe with other functions. -void -nn_term(void) -{ - // This function is relatively toxic, since it can affect - // all sockets in the process, including those - // in use by libraries, etc. Accordingly, do not use this - // in a library -- only e.g. atexit() and similar. - nni_sock_closeall(); -} - -uint64_t -nn_get_statistic(int x, int y) -{ - (void) x; - (void) y; - - return (0); -} diff --git a/src/compat/nanomsg/nuts_compat.h b/src/compat/nanomsg/nuts_compat.h deleted file mode 100644 index d7bd56e7..00000000 --- a/src/compat/nanomsg/nuts_compat.h +++ /dev/null @@ -1,73 +0,0 @@ -// -// Copyright 2021 Staysail Systems, Inc. <info@staysail.tech> -// -// 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. -// - -#ifndef NUTS_COMPAT_H -#define NUTS_COMPAT_H - -#include <stdbool.h> -#include <stdint.h> - -// The following headers are provided for test code convenience. -#include <nng/nng.h> - -#ifdef __cplusplus -extern "C" { -#endif - -// NUTS_NN_PASS tests for NN success. It reports the failure if it did not. -#define NUTS_NN_PASS(cond) \ - do { \ - int result_ = (cond); \ - TEST_CHECK_(result_ >= 0, "%s succeeds", #cond); \ - TEST_MSG("%s: expected success, got %s (%d)", #cond, \ - nn_strerror(errno), errno); \ - } while (0) - -#define NUTS_NN_FAIL(cond, expect) \ - do { \ - int result_ = (cond); \ - int err_ = errno; \ - TEST_CHECK_(result_ < 0, "%s did not succeed", #cond); \ - TEST_CHECK_( \ - err_ == (expect), "%s fails with %s", #cond, #expect); \ - TEST_MSG("%s: expected %s, got %d / %d (%s)", #cond, #expect, \ - result_, (expect), nng_strerror(err_)); \ - } while (0) - -// These macros use some details of the socket and pipe which are not public. -// We do that to facilitate testing. Don't rely on this equivalence in your -// own application code. - -#define NUTS_NN_MARRY(s1, s2) \ - do { \ - nng_socket s1_, s2_; \ - s1_.id = s1; \ - s2_.id = s2; \ - \ - NUTS_MARRY(s1_, s2_); \ - } while (0) - -#define NUTS_NN_MARRY_EX(s1, s2, url, p1, p2) \ - do { \ - nng_socket s1_, s2_; \ - nng_pipe p1_, p2_; \ - s1_.id = s1; \ - s2_.id = s2; \ - NUTS_MARRY_EX(s1_, s2_, url, &p1_, &p2_); \ - (p1) = p1_.id; \ - (p2) = p2_.id; \ - NUTS_TRUE((p1) >= 0); \ - NUTS_TRUE((p2) >= 0); \ - } while (0) - -#ifdef __cplusplus -}; -#endif - -#endif // NUTS_COMPAT |
