/**************************************************************************** * examples/usrsocktest/usrsocktest_block_send.c * Send through the socket in blocking mode * * Copyright (C) 2015, 2017 Haltian Ltd. All rights reserved. * Authors: Roman Saveljev * Jussi Kivilinna * * 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 #include #include #include #include #include #include #include #include #include #include "defines.h" /**************************************************************************** * Pre-processor Definitions ****************************************************************************/ /**************************************************************************** * Private Types ****************************************************************************/ /**************************************************************************** * Private Function Prototypes ****************************************************************************/ /**************************************************************************** * Private Data ****************************************************************************/ static bool started; static int sd; /**************************************************************************** * Public Data ****************************************************************************/ /**************************************************************************** * Private Functions ****************************************************************************/ /**************************************************************************** * Name: ConnectSend * * Description: * Open socket, connect in blocking mode and send * * Input Parameters: * dconf - socket daemon configuration * * Returned Value: * None * * Assumptions/Limitations: * None * ****************************************************************************/ static void ConnectSend(FAR struct usrsocktest_daemon_conf_s *dconf) { ssize_t ret; size_t datalen; const void *data; struct sockaddr_in addr; /* Start test daemon. */ dconf->endpoint_addr = "127.0.0.1"; dconf->endpoint_port = 255; dconf->endpoint_block_connect = true; dconf->endpoint_block_send = true; TEST_ASSERT_EQUAL(OK, usrsocktest_daemon_start(dconf)); 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()); /* Do connect, should succeed (after connect block released). */ inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); addr.sin_family = AF_INET; addr.sin_port = htons(255); TEST_ASSERT_TRUE(usrsocktest_send_delayed_command('E', 100)); 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()); /* Send data to remote, */ data = "abcde"; datalen = strlen("abcde"); TEST_ASSERT_TRUE(usrsocktest_send_delayed_command('W', 100)); ret = sendto(sd, data, datalen, 0, NULL, 0); TEST_ASSERT_EQUAL(datalen, ret); TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); TEST_ASSERT_EQUAL(datalen, usrsocktest_daemon_get_send_bytes()); /* 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); } /**************************************************************************** * Name: NonBlockConnectSend * * Description: * Open socket, connect in non-blocking mode and send * * Input Parameters: * dconf - socket daemon configuration * * Returned Value: * None * * Assumptions/Limitations: * None * ****************************************************************************/ static void NonBlockConnectSend(FAR struct usrsocktest_daemon_conf_s *dconf) { ssize_t ret; size_t datalen; const void *data; struct sockaddr_in addr; /* Start test daemon. */ dconf->endpoint_addr = "127.0.0.1"; dconf->endpoint_port = 255; dconf->endpoint_block_connect = true; dconf->endpoint_block_send = false; TEST_ASSERT_EQUAL(OK, usrsocktest_daemon_start(dconf)); 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()); /* Do connect, should succeed (after connect block released). */ inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); addr.sin_family = AF_INET; addr.sin_port = htons(255); TEST_ASSERT_TRUE(usrsocktest_send_delayed_command('W', 100)); TEST_ASSERT_TRUE(usrsocktest_send_delayed_command('E', 100)); 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()); /* Send data to remote. */ data = "abcde"; datalen = strlen("abcde"); ret = write(sd, data, datalen); TEST_ASSERT_EQUAL(datalen, ret); TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); TEST_ASSERT_EQUAL(datalen, usrsocktest_daemon_get_send_bytes()); /* 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); } /**************************************************************************** * Name: SendTimeout * * Description: * Open socket, connect in blocking mode and send with SO_SNDTIMEO * * Input Parameters: * dconf - socket daemon configuration * * Returned Value: * None * * Assumptions/Limitations: * None * ****************************************************************************/ static void SendTimeout(FAR struct usrsocktest_daemon_conf_s *dconf) { ssize_t ret; size_t datalen; const void *data; struct sockaddr_in addr; struct timeval tv; dconf->endpoint_addr = "127.0.0.1"; dconf->endpoint_port = 255; dconf->endpoint_block_connect = true; dconf->endpoint_block_send = true; TEST_ASSERT_EQUAL(OK, usrsocktest_daemon_start(dconf)); 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()); /* Do connect, should succeed (after connect block released). */ inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr.s_addr); addr.sin_family = AF_INET; addr.sin_port = htons(255); TEST_ASSERT_TRUE(usrsocktest_send_delayed_command('E', 100)); 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()); /* Setup send timeout. */ tv.tv_sec = 0; tv.tv_usec = 100 * 1000; ret = setsockopt(sd, SOL_SOCKET, SO_SNDTIMEO, (FAR const void *)&tv, sizeof(tv)); TEST_ASSERT_EQUAL(0, ret); /* Try send data to remote. */ data = "abcde"; datalen = strlen("abcde"); ret = sendto(sd, data, datalen, 0, NULL, 0); TEST_ASSERT_EQUAL(-1, ret); TEST_ASSERT_EQUAL(EAGAIN, errno); TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_connected_sockets()); TEST_ASSERT_EQUAL(1, usrsocktest_daemon_get_num_active_sockets()); TEST_ASSERT_EQUAL(0, usrsocktest_daemon_get_send_bytes()); /* 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); } /**************************************************************************** * Name: BlockSend test group setup * * Description: * Setup function executed before each testcase in this test group * * Input Parameters: * None * * Returned Value: * None * * Assumptions/Limitations: * None * ****************************************************************************/ TEST_SETUP(BlockSend) { sd = -1; started = false; } /**************************************************************************** * Name: BlockSend test group teardown * * Description: * Setup function executed after each testcase in this test group * * Input Parameters: * None * * Returned Value: * None * * Assumptions/Limitations: * None * ****************************************************************************/ TEST_TEAR_DOWN(BlockSend) { int ret; if (sd >= 0) { ret = close(sd); assert(ret >= 0); } if (started) { ret = usrsocktest_daemon_stop(); assert(ret == OK); } } TEST(BlockSend, ConnectSend) { usrsocktest_daemon_config = usrsocktest_daemon_defconf; ConnectSend(&usrsocktest_daemon_config); } TEST(BlockSend, ConnectSendDelay) { usrsocktest_daemon_config = usrsocktest_daemon_defconf; usrsocktest_daemon_config.delay_all_responses = true; ConnectSend(&usrsocktest_daemon_config); } TEST(BlockSend, NonBlockConnectSend) { usrsocktest_daemon_config = usrsocktest_daemon_defconf; NonBlockConnectSend(&usrsocktest_daemon_config); } TEST(BlockSend, NonBlockConnectSendDelay) { usrsocktest_daemon_config = usrsocktest_daemon_defconf; usrsocktest_daemon_config.delay_all_responses = true; NonBlockConnectSend(&usrsocktest_daemon_config); } TEST(BlockSend, SendTimeout) { usrsocktest_daemon_config = usrsocktest_daemon_defconf; SendTimeout(&usrsocktest_daemon_config); } TEST(BlockSend, SendTimeoutDelay) { usrsocktest_daemon_config = usrsocktest_daemon_defconf; usrsocktest_daemon_config.delay_all_responses = true; SendTimeout(&usrsocktest_daemon_config); } /**************************************************************************** * Public Functions ****************************************************************************/ TEST_GROUP(BlockSend) { RUN_TEST_CASE(BlockSend, ConnectSend); RUN_TEST_CASE(BlockSend, ConnectSendDelay); RUN_TEST_CASE(BlockSend, NonBlockConnectSend); RUN_TEST_CASE(BlockSend, NonBlockConnectSendDelay); RUN_TEST_CASE(BlockSend, SendTimeout); RUN_TEST_CASE(BlockSend, SendTimeoutDelay); }