/***************************************************************************/ /* Copyright (c) 2024 Microsoft Corporation */ /* Copyright (c) 2026 Eclipse ThreadX contributors */ /* */ /* This program and the accompanying materials are made available under */ /* the terms of the MIT License which is available at */ /* https://opensource.org/licenses/MIT. */ /* */ /* SPDX-License-Identifier: MIT */ /***************************************************************************/ /* This NetX test concentrates on calling nx_tcp_server_socket_accept() for multiple times. * It verifies a bug that TCP server would send SYN+ACK on each accept() call. The ACK number in each packet is different. */ #include "nx_api.h" #include "nx_tcp.h" extern void test_control_return(UINT status); #if !defined(NX_DISABLE_IPV4) #define DEMO_STACK_SIZE 2048 /* Define the ThreadX and NetX object control blocks... */ static TX_THREAD thread_0; static TX_THREAD thread_1; static NX_PACKET_POOL pool_0; static NX_IP ip_0; static NX_IP ip_1; static NX_TCP_SOCKET client_socket; static NX_TCP_SOCKET server_socket; /* Define the counters used in the demo application... */ static ULONG error_counter = 0; static UINT syn_ack_received; static UINT sequence[10]; static UINT ack_number[10]; /* Define thread prototypes. */ static void thread_0_entry(ULONG thread_input); static void thread_1_entry(ULONG thread_input); extern void _nx_ram_network_driver_1500(struct NX_IP_DRIVER_STRUCT *driver_req); static void tcp_packet_receive(NX_IP *ip_ptr, NX_PACKET *packet_ptr); /* Define what the initial system looks like. */ #ifdef CTEST VOID test_application_define(void *first_unused_memory) #else void netx_tcp_duplicate_accept_test_application_define(void *first_unused_memory) #endif { CHAR *pointer; UINT status; /* Setup the working pointer. */ pointer = (CHAR *) first_unused_memory; error_counter = 0; syn_ack_received = 0; /* Create the main thread. */ tx_thread_create(&thread_0, "thread 0", thread_0_entry, 0, pointer, DEMO_STACK_SIZE, 4, 4, TX_NO_TIME_SLICE, TX_AUTO_START); pointer = pointer + DEMO_STACK_SIZE; /* Create the main thread. */ tx_thread_create(&thread_1, "thread 1", thread_1_entry, 0, pointer, DEMO_STACK_SIZE, 3, 3, TX_NO_TIME_SLICE, TX_AUTO_START); pointer = pointer + DEMO_STACK_SIZE; /* Initialize the NetX system. */ nx_system_initialize(); /* Create the packet pool. */ status = nx_packet_pool_create(&pool_0, "NetX Main Packet Pool", 1536, pointer, 10240); pointer = pointer + 10240; if (status) error_counter++; /* Create an IP instance. */ status = nx_ip_create(&ip_0, "NetX IP Instance 0", IP_ADDRESS(1, 2, 3, 4), 0xFFFFFF00UL, &pool_0, _nx_ram_network_driver_1500, pointer, 2048, 1); pointer = pointer + 2048; /* Create another IP instance. */ status += nx_ip_create(&ip_1, "NetX IP Instance 1", IP_ADDRESS(1, 2, 3, 5), 0xFFFFFF00UL, &pool_0, _nx_ram_network_driver_1500, pointer, 2048, 1); pointer = pointer + 2048; if (status) error_counter++; /* Enable ARP and supply ARP cache memory for IP Instance 0. */ status = nx_arp_enable(&ip_0, (void *) pointer, 1024); pointer = pointer + 1024; /* Enable ARP and supply ARP cache memory for IP Instance 1. */ status += nx_arp_enable(&ip_1, (void *) pointer, 1024); pointer = pointer + 1024; /* Check ARP enable status. */ if (status) error_counter++; /* Enable TCP processing for both IP instances. */ status = nx_tcp_enable(&ip_0); status += nx_tcp_enable(&ip_1); /* Check TCP enable status. */ if (status) error_counter++; } /* Define the test threads. */ static void thread_0_entry(ULONG thread_input) { UINT status; ULONG actual_status; /* Print out some test information banners. */ printf("NetX Test: TCP Duplicate Accept Test................................."); /* Ensure the IP instance has been initialized. */ status = nx_ip_status_check(&ip_0, NX_IP_INITIALIZE_DONE, &actual_status, NX_IP_PERIODIC_RATE); if (status) error_counter++; /* Check for earlier error. */ if (error_counter) { printf("ERROR!\n"); test_control_return(1); } /* Create a socket. */ status = nx_tcp_socket_create(&ip_0, &client_socket, "Client Socket", NX_IP_NORMAL, NX_FRAGMENT_OKAY, NX_IP_TIME_TO_LIVE, 65535, NX_NULL, NX_NULL); if (status) error_counter++; /* Bind the socket. */ status = nx_tcp_client_socket_bind(&client_socket, 0x88, NX_WAIT_FOREVER); if (status) error_counter++; /* Filter the TCP packet received by client. */ ip_0.nx_ip_tcp_packet_receive = tcp_packet_receive; /* Attempt to connect the socket. */ nx_tcp_client_socket_connect(&client_socket, IP_ADDRESS(1, 2, 3, 5), 12, NX_NO_WAIT); /* Disconnect this socket. */ status = nx_tcp_socket_disconnect(&client_socket, NX_NO_WAIT); if (status) error_counter++; /* Unbind the socket. */ status = nx_tcp_client_socket_unbind(&client_socket); if (status) error_counter++; /* Delete the socket. */ status = nx_tcp_socket_delete(&client_socket); if (status) error_counter++; } static void thread_1_entry(ULONG thread_input) { UINT status; UINT i; /* Create a socket. */ status = nx_tcp_socket_create(&ip_1, &server_socket, "Server Socket", NX_IP_NORMAL, NX_FRAGMENT_OKAY, NX_IP_TIME_TO_LIVE, 1460, NX_NULL, NX_NULL); if (status) error_counter++; /* Setup this thread to listen. */ status = nx_tcp_server_socket_listen(&ip_1, 12, &server_socket, 5, NX_NULL); if (status) error_counter++; /* Sleep one second. Let client send SYN first. */ tx_thread_sleep(NX_IP_PERIODIC_RATE); /* Accept a client socket connection. */ status = nx_tcp_server_socket_accept(&server_socket, NX_NO_WAIT); if (status != NX_IN_PROGRESS) error_counter++; /* Accept a client socket connection again. */ status = nx_tcp_server_socket_accept(&server_socket, NX_NO_WAIT); if (status != NX_IN_PROGRESS) error_counter++; /* Disconnect the server socket. */ status = nx_tcp_socket_disconnect(&server_socket, NX_NO_WAIT); if (status) error_counter++; /* Unaccept the server socket. */ status = nx_tcp_server_socket_unaccept(&server_socket); if (status) error_counter++; /* Unlisten on the server port 12. */ status = nx_tcp_server_socket_unlisten(&ip_1, 12); if (status) error_counter++; /* Delete the socket. */ status = nx_tcp_socket_delete(&server_socket); if (status) error_counter++; /* Only one SYN+ACK packet is expected. */ if (syn_ack_received != 1) { error_counter++; /* Check whether all sequence and ACK numbers are the same. */ for (i = 1; i < syn_ack_received; i++) { if (sequence[i] != sequence[0]) error_counter++; if (ack_number[i] != ack_number[0]) error_counter++; } } /* Check status. */ if (error_counter) { printf("ERROR!\n"); test_control_return(1); } else { printf("SUCCESS!\n"); test_control_return(0); } } static void tcp_packet_receive(NX_IP *ip_ptr, NX_PACKET *packet_ptr) { NX_TCP_HEADER *tcp_header_ptr; /* Get the TCP header pointer. */ tcp_header_ptr = (NX_TCP_HEADER *) packet_ptr -> nx_packet_prepend_ptr; NX_CHANGE_ULONG_ENDIAN(tcp_header_ptr -> nx_tcp_header_word_3); if((tcp_header_ptr -> nx_tcp_header_word_3 & NX_TCP_SYN_BIT) && (tcp_header_ptr -> nx_tcp_header_word_3 & NX_TCP_ACK_BIT)) { /* Get Sequence and ACK numbers. */ sequence[syn_ack_received] = tcp_header_ptr -> nx_tcp_sequence_number; ack_number[syn_ack_received] = tcp_header_ptr -> nx_tcp_acknowledgment_number; syn_ack_received++; } NX_CHANGE_ULONG_ENDIAN(tcp_header_ptr -> nx_tcp_header_word_3); /* Drop the packet to avoid RST. */ nx_packet_release(packet_ptr); } #else #ifdef CTEST VOID test_application_define(void *first_unused_memory) #else void netx_tcp_duplicate_accept_test_application_define(void *first_unused_memory) #endif { /* Print out test information banner. */ printf("NetX Test: TCP Duplicate Accept Test.................................N/A\n"); test_control_return(3); } #endif