| /**************************************************************************** |
| * examples/usrsocktest/usrsocktest_noblock_connect.c |
| * Socket connect tests in non-blocking mode |
| * |
| * Copyright (C) 2015, 2017 Haltian Ltd. All rights reserved. |
| * Authors: Roman Saveljev <roman.saveljev@haltian.com> |
| * Jussi Kivilinna <jussi.kivilinna@haltian.com> |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * |
| * 1. Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in |
| * the documentation and/or other materials provided with the |
| * distribution. |
| * 3. Neither the name NuttX nor the names of its contributors may be |
| * used to endorse or promote products derived from this software |
| * without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS |
| * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE |
| * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, |
| * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, |
| * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS |
| * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED |
| * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
| * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN |
| * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
| * POSSIBILITY OF SUCH DAMAGE. |
| * |
| ****************************************************************************/ |
| |
| /**************************************************************************** |
| * Included Files |
| ****************************************************************************/ |
| |
| #include <sys/socket.h> |
| #include <errno.h> |
| #include <sys/types.h> |
| #include <stdbool.h> |
| #include <assert.h> |
| #include <fcntl.h> |
| #include <netinet/in.h> |
| #include <arpa/inet.h> |
| |
| #include "defines.h" |
| |
| /**************************************************************************** |
| * Pre-processor Definitions |
| ****************************************************************************/ |
| |
| /**************************************************************************** |
| * Private Types |
| ****************************************************************************/ |
| |
| /**************************************************************************** |
| * Private Function Prototypes |
| ****************************************************************************/ |
| |
| /**************************************************************************** |
| * Private Data |
| ****************************************************************************/ |
| |
| static bool started; |
| static int sd; |
| static int sd2; |
| |
| /**************************************************************************** |
| * Public Data |
| ****************************************************************************/ |
| |
| /**************************************************************************** |
| * Private Functions |
| ****************************************************************************/ |
| |
| TEST_SETUP(no_block_connect) |
| { |
| sd = -1; |
| sd2 = -1; |
| started = false; |
| } |
| |
| TEST_TEAR_DOWN(no_block_connect) |
| { |
| int ret; |
| |
| if (sd >= 0) |
| { |
| ret = close(sd); |
| assert(ret == 0); |
| } |
| |
| if (sd2 >= 0) |
| { |
| ret = close(sd2); |
| assert(ret == 0); |
| } |
| |
| if (started) |
| { |
| ret = usrsocktest_daemon_stop(); |
| assert(ret == OK); |
| } |
| } |
| |
| TEST(no_block_connect, instant_connect) |
| { |
| int flags; |
| int ret; |
| struct sockaddr_in addr; |
| |
| usrsocktest_daemon_config = usrsocktest_daemon_defconf; |
| usrsocktest_daemon_config.endpoint_addr = "127.0.0.1"; |
| usrsocktest_daemon_config.endpoint_port = 255; |
| TEST_ASSERT_EQUAL(OK, |
| usrsocktest_daemon_start(&usrsocktest_daemon_config)); |
| started = true; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| |
| /* Open socket */ |
| |
| sd = socket(AF_INET, SOCK_STREAM, 0); |
| TEST_ASSERT_TRUE(sd >= 0); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Make socket non-blocking */ |
| |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(0, flags & O_NONBLOCK); |
| ret = fcntl(sd, F_SETFL, flags | O_NONBLOCK); |
| TEST_ASSERT_EQUAL(0, ret); |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(O_NONBLOCK, flags & O_NONBLOCK); |
| |
| /* Do connect, should succeed instantly. */ |
| |
| inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); |
| addr.sin_family = AF_INET; |
| addr.sin_port = htons(255); |
| ret = connect(sd, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(0, ret); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Close socket */ |
| |
| TEST_ASSERT_TRUE(close(sd) >= 0); |
| sd = -1; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| |
| /* Stopping daemon should succeed. */ |
| |
| TEST_ASSERT_EQUAL(OK, usrsocktest_daemon_stop()); |
| started = false; |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_endp_malloc_cnt); |
| TEST_ASSERT_EQUAL(0, usrsocktest_dcmd_malloc_cnt); |
| } |
| |
| TEST(no_block_connect, delayed_connect) |
| { |
| int flags; |
| int ret; |
| int count; |
| struct sockaddr_in addr; |
| |
| /* Start test daemon. */ |
| |
| usrsocktest_daemon_config = usrsocktest_daemon_defconf; |
| usrsocktest_daemon_config.endpoint_block_connect = true; |
| usrsocktest_daemon_config.endpoint_addr = "127.0.0.1"; |
| usrsocktest_daemon_config.endpoint_port = 255; |
| usrsocktest_daemon_config.delay_all_responses = true; |
| TEST_ASSERT_EQUAL(OK, |
| usrsocktest_daemon_start(&usrsocktest_daemon_config)); |
| started = true; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| |
| /* Open socket */ |
| |
| sd = socket(AF_INET, SOCK_STREAM, 0); |
| TEST_ASSERT_TRUE(sd >= 0); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Make socket non-blocking */ |
| |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(0, flags & O_NONBLOCK); |
| ret = fcntl(sd, F_SETFL, flags | O_NONBLOCK); |
| TEST_ASSERT_EQUAL(0, ret); |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(O_NONBLOCK, flags & O_NONBLOCK); |
| |
| /* Launch connect attempt, daemon delays actual connection until |
| * triggered. |
| */ |
| |
| inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); |
| addr.sin_family = AF_INET; |
| addr.sin_port = htons(255); |
| ret = connect(sd, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EINPROGRESS, errno); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Another connect attempt results EALREADY. */ |
| |
| ret = connect(sd, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EALREADY, errno); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Release delayed connect. */ |
| |
| TEST_ASSERT_TRUE(usrsocktest_daemon_establish_waiting_connections()); |
| for (count = 0; |
| usrsocktest_daemon_get_num_waiting_connect_sockets() > 0; count++) |
| { |
| TEST_ASSERT_TRUE(count <= 5); |
| usleep(10 * 1000); |
| } |
| |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Another connect attempt results EISCONN. */ |
| |
| ret = connect(sd, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EISCONN, errno); |
| |
| /* Close socket. */ |
| |
| TEST_ASSERT_TRUE(close(sd) >= 0); |
| sd = -1; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Stopping daemon should succeed. */ |
| |
| TEST_ASSERT_EQUAL(OK, usrsocktest_daemon_stop()); |
| started = false; |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, |
| usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_endp_malloc_cnt); |
| TEST_ASSERT_EQUAL(0, usrsocktest_dcmd_malloc_cnt); |
| } |
| |
| TEST(no_block_connect, close_not_connected) |
| { |
| int flags; |
| int ret; |
| struct sockaddr_in addr; |
| |
| /* Start test daemon. */ |
| |
| usrsocktest_daemon_config = usrsocktest_daemon_defconf; |
| usrsocktest_daemon_config.endpoint_block_connect = true; |
| usrsocktest_daemon_config.endpoint_addr = "127.0.0.1"; |
| usrsocktest_daemon_config.endpoint_port = 255; |
| usrsocktest_daemon_config.delay_all_responses = true; |
| TEST_ASSERT_EQUAL(OK, |
| usrsocktest_daemon_start(&usrsocktest_daemon_config)); |
| started = true; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| |
| /* Open socket */ |
| |
| sd = socket(AF_INET, SOCK_STREAM, 0); |
| TEST_ASSERT_TRUE(sd >= 0); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Make socket non-blocking */ |
| |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(0, flags & O_NONBLOCK); |
| ret = fcntl(sd, F_SETFL, flags | O_NONBLOCK); |
| TEST_ASSERT_EQUAL(0, ret); |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(O_NONBLOCK, flags & O_NONBLOCK); |
| |
| /* Launch connect attempt, daemon delays actual connection until |
| * triggered. |
| */ |
| |
| inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); |
| addr.sin_family = AF_INET; |
| addr.sin_port = htons(255); |
| ret = connect(sd, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EINPROGRESS, errno); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Close socket. */ |
| |
| TEST_ASSERT_TRUE(close(sd) >= 0); |
| sd = -1; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Stopping daemon should succeed. */ |
| |
| TEST_ASSERT_EQUAL(OK, usrsocktest_daemon_stop()); |
| started = false; |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, |
| usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_endp_malloc_cnt); |
| TEST_ASSERT_EQUAL(0, usrsocktest_dcmd_malloc_cnt); |
| } |
| |
| TEST(no_block_connect, early_drop) |
| { |
| int flags; |
| int ret; |
| struct sockaddr_in addr; |
| |
| /* Start test daemon. */ |
| |
| usrsocktest_daemon_config = usrsocktest_daemon_defconf; |
| usrsocktest_daemon_config.endpoint_block_connect = true; |
| usrsocktest_daemon_config.endpoint_addr = "127.0.0.1"; |
| usrsocktest_daemon_config.endpoint_port = 255; |
| usrsocktest_daemon_config.delay_all_responses = false; |
| TEST_ASSERT_EQUAL(OK, |
| usrsocktest_daemon_start(&usrsocktest_daemon_config)); |
| started = true; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| |
| /* Open socket */ |
| |
| sd = socket(AF_INET, SOCK_STREAM, 0); |
| TEST_ASSERT_TRUE(sd >= 0); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Make socket non-blocking */ |
| |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(0, flags & O_NONBLOCK); |
| ret = fcntl(sd, F_SETFL, flags | O_NONBLOCK); |
| TEST_ASSERT_EQUAL(0, ret); |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(O_NONBLOCK, flags & O_NONBLOCK); |
| |
| /* Launch connect attempt, daemon delays actual connection until |
| * triggered. |
| */ |
| |
| inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); |
| addr.sin_family = AF_INET; |
| addr.sin_port = htons(255); |
| ret = connect(sd, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EINPROGRESS, errno); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Stopping daemon should succeed. */ |
| |
| TEST_ASSERT_EQUAL(OK, usrsocktest_daemon_stop()); |
| started = false; |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, |
| usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_endp_malloc_cnt); |
| TEST_ASSERT_EQUAL(0, usrsocktest_dcmd_malloc_cnt); |
| |
| /* Close socket. */ |
| |
| TEST_ASSERT_TRUE(close(sd) >= 0); |
| sd = -1; |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, |
| usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| } |
| |
| TEST(no_block_connect, multiple) |
| { |
| int flags; |
| int ret; |
| int count; |
| struct sockaddr_in addr; |
| |
| /* Start test daemon. */ |
| |
| usrsocktest_daemon_config = usrsocktest_daemon_defconf; |
| usrsocktest_daemon_config.endpoint_block_connect = true; |
| usrsocktest_daemon_config.endpoint_addr = "127.0.0.1"; |
| usrsocktest_daemon_config.endpoint_port = 255; |
| usrsocktest_daemon_config.delay_all_responses = false; |
| TEST_ASSERT_EQUAL(OK, |
| usrsocktest_daemon_start(&usrsocktest_daemon_config)); |
| started = true; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| |
| /* Open sockets */ |
| |
| sd = socket(AF_INET, SOCK_STREAM, 0); |
| TEST_ASSERT_TRUE(sd >= 0); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| sd2 = socket(AF_INET, SOCK_STREAM, 0); |
| TEST_ASSERT_TRUE(sd2 >= 0); |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Make socket non-blocking */ |
| |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(0, flags & O_NONBLOCK); |
| ret = fcntl(sd, F_SETFL, flags | O_NONBLOCK); |
| TEST_ASSERT_EQUAL(0, ret); |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(O_NONBLOCK, flags & O_NONBLOCK); |
| |
| flags = fcntl(sd2, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(0, flags & O_NONBLOCK); |
| ret = fcntl(sd2, F_SETFL, flags | O_NONBLOCK); |
| TEST_ASSERT_EQUAL(0, ret); |
| flags = fcntl(sd2, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(O_NONBLOCK, flags & O_NONBLOCK); |
| |
| /* Launch connect attempts, daemon delays actual connection until |
| * triggered. |
| */ |
| |
| inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); |
| addr.sin_family = AF_INET; |
| addr.sin_port = htons(255); |
| ret = connect(sd2, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EINPROGRESS, errno); |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Release delayed connections. */ |
| |
| TEST_ASSERT_TRUE(usrsocktest_daemon_establish_waiting_connections()); |
| for (count = 0; |
| usrsocktest_daemon_get_num_waiting_connect_sockets() > 0; count++) |
| { |
| TEST_ASSERT_TRUE(count <= 5); |
| usleep(10 * 1000); |
| } |
| |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); |
| addr.sin_family = AF_INET; |
| addr.sin_port = htons(255); |
| ret = connect(sd2, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EISCONN, errno); |
| |
| inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); |
| addr.sin_family = AF_INET; |
| addr.sin_port = htons(255); |
| ret = connect(sd, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EINPROGRESS, errno); |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Release delayed connections. */ |
| |
| TEST_ASSERT_TRUE(usrsocktest_daemon_establish_waiting_connections()); |
| for (count = 0; usrsocktest_daemon_get_num_waiting_connect_sockets() > 0; |
| count++) |
| { |
| TEST_ASSERT_TRUE(count <= 5); |
| usleep(10 * 1000); |
| } |
| |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Close sockets. */ |
| |
| TEST_ASSERT_TRUE(close(sd2) >= 0); |
| sd2 = -1; |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| TEST_ASSERT_TRUE(close(sd) >= 0); |
| sd = -1; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Open sockets */ |
| |
| sd = socket(AF_INET, SOCK_STREAM, 0); |
| TEST_ASSERT_TRUE(sd >= 0); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| sd2 = socket(AF_INET, SOCK_STREAM, 0); |
| TEST_ASSERT_TRUE(sd2 >= 0); |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Make socket non-blocking */ |
| |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(0, flags & O_NONBLOCK); |
| ret = fcntl(sd, F_SETFL, flags | O_NONBLOCK); |
| TEST_ASSERT_EQUAL(0, ret); |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(O_NONBLOCK, flags & O_NONBLOCK); |
| |
| flags = fcntl(sd2, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(0, flags & O_NONBLOCK); |
| ret = fcntl(sd2, F_SETFL, flags | O_NONBLOCK); |
| TEST_ASSERT_EQUAL(0, ret); |
| flags = fcntl(sd2, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(O_NONBLOCK, flags & O_NONBLOCK); |
| |
| /* Launch connect attempts, daemon delays actual connection until |
| * triggered. |
| */ |
| |
| inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); |
| addr.sin_family = AF_INET; |
| addr.sin_port = htons(255); |
| ret = connect(sd, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EINPROGRESS, errno); |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| ret = connect(sd2, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EINPROGRESS, errno); |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Release delayed connections. */ |
| |
| TEST_ASSERT_TRUE(usrsocktest_daemon_establish_waiting_connections()); |
| for (count = 0; |
| usrsocktest_daemon_get_num_waiting_connect_sockets() > 0; count++) |
| { |
| TEST_ASSERT_TRUE(count <= 5); |
| usleep(10 * 1000); |
| } |
| |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Close sockets. */ |
| |
| TEST_ASSERT_TRUE(close(sd2) >= 0); |
| sd2 = -1; |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| TEST_ASSERT_TRUE(close(sd) >= 0); |
| sd = -1; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Open sockets */ |
| |
| sd = socket(AF_INET, SOCK_STREAM, 0); |
| TEST_ASSERT_TRUE(sd >= 0); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| sd2 = socket(AF_INET, SOCK_STREAM, 0); |
| TEST_ASSERT_TRUE(sd2 >= 0); |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Make socket non-blocking */ |
| |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(0, flags & O_NONBLOCK); |
| ret = fcntl(sd, F_SETFL, flags | O_NONBLOCK); |
| TEST_ASSERT_EQUAL(0, ret); |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(O_NONBLOCK, flags & O_NONBLOCK); |
| |
| flags = fcntl(sd2, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(0, flags & O_NONBLOCK); |
| ret = fcntl(sd2, F_SETFL, flags | O_NONBLOCK); |
| TEST_ASSERT_EQUAL(0, ret); |
| flags = fcntl(sd2, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(O_NONBLOCK, flags & O_NONBLOCK); |
| |
| /* Launch connect attempt, daemon delays actual connection until |
| * triggered. |
| */ |
| |
| inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); |
| addr.sin_family = AF_INET; |
| addr.sin_port = htons(255); |
| ret = connect(sd, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EINPROGRESS, errno); |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Release delayed connections. */ |
| |
| TEST_ASSERT_TRUE(usrsocktest_daemon_establish_waiting_connections()); |
| for (count = 0; |
| usrsocktest_daemon_get_num_waiting_connect_sockets() > 0; count++) |
| { |
| TEST_ASSERT_TRUE(count <= 5); |
| usleep(10 * 1000); |
| } |
| |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Launch another connect attempt, daemon delays actual connection until |
| * triggered. |
| */ |
| |
| ret = connect(sd2, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EINPROGRESS, errno); |
| TEST_ASSERT_EQUAL(2, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Close sockets. */ |
| |
| TEST_ASSERT_TRUE(close(sd) >= 0); |
| sd = -1; |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| TEST_ASSERT_TRUE(close(sd2) >= 0); |
| sd2 = -1; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Stopping daemon should succeed. */ |
| |
| TEST_ASSERT_EQUAL(OK, usrsocktest_daemon_stop()); |
| started = false; |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, |
| usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_endp_malloc_cnt); |
| TEST_ASSERT_EQUAL(0, usrsocktest_dcmd_malloc_cnt); |
| } |
| |
| TEST(no_block_connect, basic_daemon_dup2) |
| { |
| int flags; |
| int ret; |
| int count; |
| struct sockaddr_in addr; |
| |
| /* Start test daemon. */ |
| |
| usrsocktest_daemon_config = usrsocktest_daemon_defconf; |
| usrsocktest_daemon_config.endpoint_block_connect = true; |
| usrsocktest_daemon_config.endpoint_addr = "127.0.0.1"; |
| usrsocktest_daemon_config.endpoint_port = 255; |
| usrsocktest_daemon_config.delay_all_responses = true; |
| TEST_ASSERT_EQUAL(OK, |
| usrsocktest_daemon_start(&usrsocktest_daemon_config)); |
| started = true; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| |
| /* Open socket */ |
| |
| sd = socket(AF_INET, SOCK_STREAM, 0); |
| TEST_ASSERT_TRUE(sd >= 0); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Duplicate socket */ |
| |
| sd2 = dup(sd); |
| TEST_ASSERT_TRUE(sd2 >= 0); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| |
| /* Make socket non-blocking */ |
| |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(0, flags & O_NONBLOCK); |
| ret = fcntl(sd, F_SETFL, flags | O_NONBLOCK); |
| TEST_ASSERT_EQUAL(0, ret); |
| flags = fcntl(sd, F_GETFL, 0); |
| TEST_ASSERT_TRUE(flags >= 0); |
| TEST_ASSERT_EQUAL(O_RDWR, flags & O_RDWR); |
| TEST_ASSERT_EQUAL(O_NONBLOCK, flags & O_NONBLOCK); |
| |
| /* Launch connect attempt, daemon delays actual connection until |
| * triggered. |
| */ |
| |
| inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); |
| addr.sin_family = AF_INET; |
| addr.sin_port = htons(255); |
| ret = connect(sd, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EINPROGRESS, errno); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Another connect attempt results EALREADY. */ |
| |
| ret = connect(sd2, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EALREADY, errno); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Release delayed connect. */ |
| |
| TEST_ASSERT_TRUE(usrsocktest_daemon_establish_waiting_connections()); |
| for (count = 0; |
| usrsocktest_daemon_get_num_waiting_connect_sockets() > 0; count++) |
| { |
| TEST_ASSERT_TRUE(count <= 5); |
| usleep(10 * 1000); |
| } |
| |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Another connect attempt results EISCONN. */ |
| |
| ret = connect(sd, (FAR const struct sockaddr *)&addr, sizeof(addr)); |
| TEST_ASSERT_EQUAL(-1, ret); |
| TEST_ASSERT_EQUAL(EISCONN, errno); |
| |
| /* Close sockets. */ |
| |
| TEST_ASSERT_TRUE(close(sd) >= 0); |
| sd = -1; |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| TEST_ASSERT_TRUE(close(sd2) >= 0); |
| sd2 = -1; |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| |
| /* Stopping daemon should succeed. */ |
| |
| TEST_ASSERT_EQUAL(OK, usrsocktest_daemon_stop()); |
| started = false; |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_active_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, usrsocktest_daemon_get_num_connected_sockets()); |
| TEST_ASSERT_EQUAL(-ENODEV, |
| usrsocktest_daemon_get_num_waiting_connect_sockets()); |
| TEST_ASSERT_EQUAL(0, usrsocktest_endp_malloc_cnt); |
| TEST_ASSERT_EQUAL(0, usrsocktest_dcmd_malloc_cnt); |
| } |
| |
| /**************************************************************************** |
| * Public Functions |
| ****************************************************************************/ |
| |
| TEST_GROUP(no_block_connect) |
| { |
| RUN_TEST_CASE(no_block_connect, instant_connect); |
| RUN_TEST_CASE(no_block_connect, delayed_connect); |
| RUN_TEST_CASE(no_block_connect, close_not_connected); |
| RUN_TEST_CASE(no_block_connect, early_drop); |
| RUN_TEST_CASE(no_block_connect, multiple); |
| RUN_TEST_CASE(no_block_connect, basic_daemon_dup2); |
| } |
