Squashed 'libs/libzmq/' content from commit 2b2fb9c7

git-subtree-dir: libs/libzmq git-subtree-split: 2b2fb9c7082dbc16c1323b97040a4edcfa2b997b
2022-10-22 14:31:55 +02:00
commit e988fee167
1118 changed files with 149390 additions and 0 deletions
--- a/tests/test_srcfd.cpp
+++ b/tests/test_srcfd.cpp
@@ -0,0 +1,118 @@
+/*
+    Copyright (c) 2007-2017 Contributors as noted in the AUTHORS file
+
+    This file is part of libzmq, the ZeroMQ core engine in C++.
+
+    libzmq is free software; you can redistribute it and/or modify it under
+    the terms of the GNU Lesser General Public License (LGPL) as published
+    by the Free Software Foundation; either version 3 of the License, or
+    (at your option) any later version.
+
+    As a special exception, the Contributors give you permission to link
+    this library with independent modules to produce an executable,
+    regardless of the license terms of these independent modules, and to
+    copy and distribute the resulting executable under terms of your choice,
+    provided that you also meet, for each linked independent module, the
+    terms and conditions of the license of that module. An independent
+    module is a module which is not derived from or based on this library.
+    If you modify this library, you must extend this exception to your
+    version of the library.
+
+    libzmq is distributed in the hope that it will be useful, but WITHOUT
+    ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+    FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
+    License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "testutil.hpp"
+#include "testutil_unity.hpp"
+
+#include <string.h>
+
+SETUP_TEARDOWN_TESTCONTEXT
+
+#define MSG_SIZE 20
+
+#ifdef _WIN32
+#include <winsock2.h>
+#include <ws2tcpip.h>
+#else
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netdb.h>
+#endif
+
+void test_srcfd ()
+{
+    char my_endpoint[MAX_SOCKET_STRING];
+
+    //  Create the infrastructure
+
+    void *rep = test_context_socket (ZMQ_REP);
+    void *req = test_context_socket (ZMQ_REQ);
+
+    bind_loopback_ipv4 (rep, my_endpoint, sizeof (my_endpoint));
+
+    TEST_ASSERT_SUCCESS_ERRNO (zmq_connect (req, my_endpoint));
+
+    char tmp[MSG_SIZE];
+    memset (tmp, 0, MSG_SIZE);
+    zmq_send (req, tmp, MSG_SIZE, 0);
+
+    zmq_msg_t msg;
+    TEST_ASSERT_SUCCESS_ERRNO (zmq_msg_init (&msg));
+
+    zmq_recvmsg (rep, &msg, 0);
+    TEST_ASSERT_EQUAL_UINT (MSG_SIZE, zmq_msg_size (&msg));
+
+    // get the messages source file descriptor
+    int src_fd = zmq_msg_get (&msg, ZMQ_SRCFD);
+    TEST_ASSERT_GREATER_OR_EQUAL (0, src_fd);
+
+    TEST_ASSERT_SUCCESS_ERRNO (zmq_msg_close (&msg));
+
+    // get the remote endpoint
+    struct sockaddr_storage ss;
+#ifdef ZMQ_HAVE_HPUX
+    int addrlen = sizeof ss;
+#else
+    socklen_t addrlen = sizeof ss;
+#endif
+    TEST_ASSERT_SUCCESS_RAW_ERRNO (
+      getpeername (src_fd, (struct sockaddr *) &ss, &addrlen));
+
+    char host[NI_MAXHOST];
+    TEST_ASSERT_SUCCESS_RAW_ERRNO (getnameinfo ((struct sockaddr *) &ss,
+                                                addrlen, host, sizeof host,
+                                                NULL, 0, NI_NUMERICHOST));
+
+    // assert it is localhost which connected
+    TEST_ASSERT_EQUAL_STRING ("127.0.0.1", host);
+
+    test_context_socket_close (rep);
+    test_context_socket_close (req);
+
+    // sleep a bit for the socket to be freed
+    msleep (SETTLE_TIME);
+
+    // getting name from closed socket will fail
+#ifdef ZMQ_HAVE_WINDOWS
+    const int expected_errno = WSAENOTSOCK;
+#else
+    const int expected_errno = EBADF;
+#endif
+    TEST_ASSERT_FAILURE_RAW_ERRNO (
+      expected_errno, getpeername (src_fd, (struct sockaddr *) &ss, &addrlen));
+}
+
+int main ()
+{
+    setup_test_environment ();
+
+    UNITY_BEGIN ();
+    RUN_TEST (test_srcfd);
+    return UNITY_END ();
+}