testing/nettest: Added two new testcases and a set of public function…

…s for packet capture Add a solinger testcase and a tcpserver error port processing testcase. In addition, a set of public packet capture functions for testing were added. Signed-off-by: zhangshuai39 <[email protected]>
open-vela · Jan 2, 2025 · 8f4a4e8 · 8f4a4e8
1 parent 30b3d77
commit 8f4a4e8
Show file tree

Hide file tree

Showing 11 changed files with 729 additions and 45 deletions.
diff --git a/testing/nettest/CMakeLists.txt b/testing/nettest/CMakeLists.txt
@@ -24,14 +24,25 @@ if(CONFIG_TESTING_NET_TEST)
         ${CMAKE_CURRENT_LIST_DIR}/tcp/test_tcp_common.c
         ${CMAKE_CURRENT_LIST_DIR}/utils/nettest_tcpserver.c)
 
-    if(CONCONFIG_NET_IPv4)
-        list(APPEND SRCS
-             ${CMAKE_CURRENT_LIST_DIR}/tcp/test_tcp_connect_ipv4.c)
-    endif
+    if(CONFIG_NET_IPv4)
+      list(APPEND SRCS
+           ${CMAKE_CURRENT_LIST_DIR}/tcp/test_tcp_connect_ipv4.c)
+    endif()
 
     if(CONFIG_NET_IPv6)
+      list(APPEND SRCS
+           ${CMAKE_CURRENT_LIST_DIR}/tcp/test_tcp_connect_ipv6.c)
+    endif()
+
+    if(CONFIG_NET_PKT)
+      list(APPEND SRCS
+           ${CMAKE_CURRENT_LIST_DIR}/tcp/test_tcp_connect_rst.c
+           ${CMAKE_CURRENT_LIST_DIR}/utils/nettest_netdump.c)
+
+      if(CONFIG_NET_SOLINGER)
         list(APPEND SRCS
-             ${CMAKE_CURRENT_LIST_DIR}/tcp/test_tcp_connect_ipv6.c)
+             ${CMAKE_CURRENT_LIST_DIR}/tcp/test_tcp_connect_solinger.c)
+      endif()
     endif()
 
     nuttx_add_application(

diff --git a/testing/nettest/Makefile b/testing/nettest/Makefile
@@ -36,11 +36,20 @@ PROGNAME += cmocka_net_tcp
 CSRCS    += tcp/test_tcp_common.c utils/nettest_tcpserver.c
 
 ifeq ($(CONFIG_NET_IPv4), y)
-CSRCS    += tcp/test_tcp_connect_ipv4.c
+CSRCS += tcp/test_tcp_connect_ipv4.c
 endif
 
 ifeq ($(CONFIG_NET_IPv6), y)
-CSRCS    += tcp/test_tcp_connect_ipv6.c
+CSRCS += tcp/test_tcp_connect_ipv6.c
+endif
+
+ifeq ($(CONFIG_NET_PKT), y)
+CSRCS += tcp/test_tcp_connect_rst.c utils/nettest_netdump.c
+
+ifeq ($(CONFIG_NET_SOLINGER), y)
+CSRCS += tcp/test_tcp_connect_solinger.c
+endif
+
 endif
 
 endif

diff --git a/testing/nettest/tcp/test_tcp.c b/testing/nettest/tcp/test_tcp.c
@@ -48,6 +48,16 @@ int main(int argc, FAR char *argv[])
                                       test_tcp_connect_ipv6_setup,
                                       test_tcp_common_teardown),
 #endif
+#ifdef CONFIG_NET_PKT
+#ifdef CONFIG_NET_SOLINGER
+      cmocka_unit_test_setup_teardown(test_tcp_connect_solinger,
+                                      test_tcp_connect_solinger_setup,
+                                      test_tcp_common_teardown),
+#endif
+      cmocka_unit_test_setup_teardown(test_tcp_connect_rst,
+                                      test_tcp_connect_rst_setup,
+                                      test_tcp_common_teardown),
+#endif /* CONFIG_NET_PKT */
     };
 
   return cmocka_run_group_tests(tcp_tests, test_tcp_group_setup,

diff --git a/testing/nettest/tcp/test_tcp.h b/testing/nettest/tcp/test_tcp.h
@@ -27,14 +27,19 @@
 
 #include <pthread.h>
 
+#include "utils.h"
+
 /****************************************************************************
  * Public Types
  ****************************************************************************/
 
 struct nettest_tcp_state_s
 {
-  int       client_fd;
-  pthread_t server_tid;
+  int               client_fd;
+  pthread_t         server_tid;
+#ifdef CONFIG_NET_PKT
+  struct net_dump_s dump_state;
+#endif
 };
 
 /****************************************************************************
@@ -59,6 +64,13 @@ int test_tcp_group_teardown(FAR void **state);
 
 int test_tcp_common_teardown(FAR void **state);
 
+/****************************************************************************
+ * Name: test_tcp_common_setup
+ ****************************************************************************/
+
+int test_tcp_common_setup(FAR struct nettest_tcp_state_s *tcp_state,
+                          int family, nettest_filter_t filter);
+
 /****************************************************************************
  * Name: test_tcp_connect_ipv4_setup
  ****************************************************************************/
@@ -86,4 +98,32 @@ int test_tcp_connect_ipv6_setup(FAR void **state);
 
 void test_tcp_connect_ipv6(FAR void **state);
 #endif
+
+/****************************************************************************
+ * Name: test_tcp_connect_solinger_setup
+ ****************************************************************************/
+
+#ifdef CONFIG_NET_PKT
+#ifdef CONFIG_NET_SOLINGER
+int test_tcp_connect_solinger_setup(FAR void **state);
+
+/****************************************************************************
+ * Name: test_tcp_connect_solinger
+ ****************************************************************************/
+
+void test_tcp_connect_solinger(FAR void **state);
+#endif
+
+/****************************************************************************
+ * Name: test_tcp_connect_rst_setup
+ ****************************************************************************/
+
+int test_tcp_connect_rst_setup(FAR void **state);
+
+/****************************************************************************
+ * Name: test_tcp_connect_rst
+ ****************************************************************************/
+
+void test_tcp_connect_rst(FAR void **state);
+#endif /* CONFIG_NET_PKT */
 #endif /* __APPS_TESTING_NETTEST_TCP_TEST_TCP_H */
diff --git a/testing/nettest/tcp/test_tcp_common.c b/testing/nettest/tcp/test_tcp_common.c
@@ -28,11 +28,20 @@
 #include <stddef.h>
 #include <stdint.h>
 #include <stdlib.h>
+#include <sys/time.h>
+
 #include <cmocka.h>
 
 #include "test_tcp.h"
 #include "utils.h"
 
+/****************************************************************************
+ * Pre-processor Definitions
+ ****************************************************************************/
+
+#define NETDEV_NAME          "lo"
+#define TEST_SND_RCV_TIMEOUT 10000
+
 /****************************************************************************
  * Public Functions
  ****************************************************************************/
@@ -85,5 +94,55 @@ int test_tcp_common_teardown(FAR void **state)
       tcp_state->client_fd = -1;
     }
 
+#ifdef CONFIG_NET_PKT
+  nettest_dump_destroy(&tcp_state->dump_state);
+#endif
+
   return 0;
 }
+
+/****************************************************************************
+ * Name: test_tcp_common_setup
+ ****************************************************************************/
+
+int test_tcp_common_setup(FAR struct nettest_tcp_state_s *tcp_state,
+                          int family, nettest_filter_t filter)
+{
+  struct timeval tv;
+  int ret;
+
+  assert(family == AF_INET || family == AF_INET6);
+
+  tcp_state->client_fd = socket(family, SOCK_STREAM, 0);
+  assert_return_code(tcp_state->client_fd, errno);
+
+  tv.tv_sec  = 0;
+  tv.tv_usec = TEST_SND_RCV_TIMEOUT;
+  ret = setsockopt(tcp_state->client_fd, SOL_SOCKET, SO_RCVTIMEO,
+                   &tv, sizeof(tv));
+  if (ret < 0)
+    {
+      goto errout;
+    }
+
+  ret = setsockopt(tcp_state->client_fd, SOL_SOCKET, SO_SNDTIMEO,
+                   &tv, sizeof(tv));
+  if (ret < 0)
+    {
+      goto errout;
+    }
+
+#ifdef CONFIG_NET_PKT
+  if (filter != NULL)
+    {
+      ret = nettest_dump_setup(&tcp_state->dump_state, NETDEV_NAME,
+                               filter, 0);
+    }
+#endif
+
+  return ret;
+
+errout:
+  close(tcp_state->client_fd);
+  return ret;
+}
diff --git a/testing/nettest/tcp/test_tcp_connect_ipv4.c b/testing/nettest/tcp/test_tcp_connect_ipv4.c
@@ -27,9 +27,9 @@
 #include <stdarg.h>
 #include <stddef.h>
 #include <stdint.h>
-#include <cmocka.h>
 #include <sys/socket.h>
-#include <sys/time.h>
+
+#include <cmocka.h>
 
 #include "test_tcp.h"
 #include "utils.h"
@@ -53,23 +53,8 @@
 int test_tcp_connect_ipv4_setup(FAR void **state)
 {
   FAR struct nettest_tcp_state_s *tcp_state = *state;
-  struct timeval tv;
-  int ret;
-
-  tcp_state->client_fd = socket(PF_INET, SOCK_STREAM, 0);
-  assert_true(tcp_state->client_fd > 0);
-
-  tv.tv_sec  = 0;
-  tv.tv_usec = 10000;
-  ret = setsockopt(tcp_state->client_fd, SOL_SOCKET,
-                   SO_RCVTIMEO, &tv, sizeof(tv));
-  assert_return_code(ret, errno);
-
-  ret = setsockopt(tcp_state->client_fd, SOL_SOCKET,
-                   SO_SNDTIMEO, &tv, sizeof(tv));
-  assert_return_code(ret, errno);
 
-  return 0;
+  return test_tcp_common_setup(tcp_state, AF_INET, NULL);
 }
 
 /****************************************************************************

diff --git a/testing/nettest/tcp/test_tcp_connect_ipv6.c b/testing/nettest/tcp/test_tcp_connect_ipv6.c
@@ -27,8 +27,8 @@
 #include <stdarg.h>
 #include <stddef.h>
 #include <stdint.h>
+
 #include <cmocka.h>
-#include <sys/time.h>
 
 #include "test_tcp.h"
 #include "utils.h"
@@ -52,23 +52,8 @@
 int test_tcp_connect_ipv6_setup(FAR void **state)
 {
   FAR struct nettest_tcp_state_s *tcp_state = *state;
-  struct timeval tv;
-  int ret;
-
-  tcp_state->client_fd = socket(PF_INET6, SOCK_STREAM, 0);
-  assert_true(tcp_state->client_fd > 0);
-
-  tv.tv_sec  = 0;
-  tv.tv_usec = 10000;
-  ret = setsockopt(tcp_state->client_fd, SOL_SOCKET,
-                   SO_RCVTIMEO, &tv, sizeof(tv));
-  assert_return_code(ret, errno);
-
-  ret = setsockopt(tcp_state->client_fd, SOL_SOCKET,
-                   SO_SNDTIMEO, &tv, sizeof(tv));
-  assert_return_code(ret, errno);
 
-  return 0;
+  return test_tcp_common_setup(tcp_state, AF_INET6, NULL);
 }
 
 /****************************************************************************

diff --git a/testing/nettest/tcp/test_tcp_connect_rst.c b/testing/nettest/tcp/test_tcp_connect_rst.c
@@ -0,0 +1,92 @@
+/****************************************************************************
+ * apps/testing/nettest/tcp/test_tcp_connect_rst.c
+ *
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.  The
+ * ASF licenses this file to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance with the
+ * License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
+ * License for the specific language governing permissions and limitations
+ * under the License.
+ *
+ ****************************************************************************/
+
+/****************************************************************************
+ * Included Files
+ ****************************************************************************/
+
+#include <setjmp.h>
+#include <stdarg.h>
+#include <stddef.h>
+#include <stdint.h>
+#include <cmocka.h>
+
+#include <nuttx/net/tcp.h>
+
+#include "test_tcp.h"
+#include "utils.h"
+
+/****************************************************************************
+ * Pre-processor Definitions
+ ****************************************************************************/
+
+#define TEST_WRONG_PORT 9999
+
+/****************************************************************************
+ * Private Functions
+ ****************************************************************************/
+
+static bool filter(FAR uint8_t *buf)
+{
+  return nettest_filter_tcp_state(NET_LL_LOOPBACK, TCP_RST, buf);
+}
+
+/****************************************************************************
+ * Public Functions
+ ****************************************************************************/
+
+/****************************************************************************
+ * Name: test_tcp_connect_rst_setup
+ ****************************************************************************/
+
+int test_tcp_connect_rst_setup(FAR void **state)
+{
+  FAR struct nettest_tcp_state_s *tcp_state = *state;
+
+  return test_tcp_common_setup(tcp_state, AF_INET, filter);
+}
+
+/****************************************************************************
+ * Name: test_tcp_connect_rst
+ ****************************************************************************/
+
+void test_tcp_connect_rst(FAR void **state)
+{
+  FAR struct nettest_tcp_state_s *tcp_state = *state;
+  struct sockaddr_in myaddr;
+  int ret;
+
+  assert_false(TEST_WRONG_PORT == CONFIG_TESTING_NET_TCP_PORT);
+
+  myaddr.sin_family      = AF_INET;
+  myaddr.sin_port        = htons(TEST_WRONG_PORT);
+  myaddr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
+
+  ret = connect(tcp_state->client_fd, (FAR struct sockaddr *)&myaddr,
+                sizeof(struct sockaddr_in));
+  assert_true(ret < 0 && errno == ECONNREFUSED);
+
+  close(tcp_state->client_fd);
+  tcp_state->client_fd = -1;
+
+  nettest_netdump_stop(&tcp_state->dump_state);
+
+  assert_int_equal(sq_count(&tcp_state->dump_state.data), 1);
+}