分享Netxduo 监听多个客户端连接

snibelyh

项目需要用到tcp socket监听多个port，在外网的论坛上找到一个类似的，tcp创建多个socket监听同一个port，分享给有需要的坛友。
源链接：https://renesasrulz.com/synergy/ ... p-client-connection

1. #include "tcp_thread.h"
   
2. 
   
3. #include "rtt/SEGGER_RTT.h"
   
4. 
   
5. #define APP_TCP_SOCKET_NUM      (3)
   
6. #define APP_TCP_PORT            (5000)
   
7. #define APP_TCP_INSTANT_ECHO    (1)
   
8. #define APP_TCP_RTT_LOG         (1)
   
9. 
   
10. static NX_TCP_SOCKET g_tcp_sck[APP_TCP_SOCKET_NUM];
    
11. 
    
12. static ULONG g_not_listening = TX_FALSE;
    
13. 
    
14. static void g_tcp_sck_listen_cb(NX_TCP_SOCKET *, UINT);
    
15. static void g_tcp_sck_receive_cb(NX_TCP_SOCKET *);
    
16. static void g_tcp_sck_disconn_cb(NX_TCP_SOCKET *);
    
17. 
    
18. /* TCP Thread entry function */
    
19. void tcp_thread_entry(void)
    
20. {
    
21.     UINT status;
    
22. 
    
23.     /* Create all sockets */
    
24.     for (INT i = 0; i < APP_TCP_SOCKET_NUM; i++)
    
25.     {
    
26.         status = nx_tcp_socket_create(&g_ip, g_tcp_sck + i, "TCP Socket",
    
27.                                       NX_IP_NORMAL, NX_FRAGMENT_OKAY,
    
28.                                       NX_IP_TIME_TO_LIVE, 512, NX_NULL,
    
29.                                       g_tcp_sck_disconn_cb);
    
30.         if (NX_SUCCESS != status)
    
31.         {
    
32.             __BKPT(0);
    
33.         }
    
34.     }
    
35. 
    
36. #if APP_TCP_RTT_LOG
    
37.     SEGGER_RTT_printf(0, "Created %u TCP socket%s on port %lu\n",
    
38.                       APP_TCP_SOCKET_NUM, APP_TCP_SOCKET_NUM > 1 ? "s" : "",
    
39.                       APP_TCP_PORT);
    
40. #endif
    
41. 
    
42.     /* Start listening on the first socket */
    
43.     status = nx_tcp_server_socket_listen(&g_ip, APP_TCP_PORT, g_tcp_sck, 0,
    
44.                                          g_tcp_sck_listen_cb);
    
45.     if (NX_SUCCESS != status)
    
46.     {
    
47.         __BKPT(0);
    
48.     }
    
49. 
    
50.     while (1)
    
51.     {
    
52. #if APP_TCP_INSTANT_ECHO
    
53.         /* Everything else is handled from TCP callbacks dispatched from
    
54.          * IP instance thread */
    
55.         tx_thread_suspend(tx_thread_identify());
    
56. #else
    
57.         /* Receive pointers to TCP socket and network packet from
    
58.          * TCP callback */
    
59.         ULONG msg[2];
    
60.         status = tx_queue_receive(&g_tcp_q, msg, TX_WAIT_FOREVER);
    
61. 
    
62.         if (TX_SUCCESS == status)
    
63.         {
    
64.             NX_TCP_SOCKET * p_sck    = (NX_TCP_SOCKET *) msg[0];
    
65.             NX_PACKET     * p_packet = (NX_PACKET *)     msg[1];
    
66. 
    
67.             status = nx_tcp_socket_send(p_sck, p_packet, NX_NO_WAIT);
    
68.             if (NX_SUCCESS != status)
    
69.             {
    
70.                 nx_packet_release(p_packet);
    
71.             }
    
72.         }
    
73. 
    
74.         else
    
75.         {
    
76.             /* Unrecoverable error */
    
77.             tx_thread_suspend(tx_thread_identify());
    
78.         }
    
79. #endif
    
80.     }
    
81. }
    
82. 
    
83. 
    
84. static void g_tcp_sck_listen_cb(NX_TCP_SOCKET * p_sck, UINT port)
    
85. {
    
86.     /* Incoming connection, accept and queueing new requests */
    
87.     nx_tcp_server_socket_accept(p_sck, NX_NO_WAIT);
    
88.     nx_tcp_server_socket_unlisten(&g_ip, port);
    
89.     nx_tcp_socket_receive_notify(p_sck, g_tcp_sck_receive_cb);
    
90. 
    
91. #if APP_TCP_RTT_LOG
    
92.     ULONG ip, client_port;
    
93.     nx_tcp_socket_peer_info_get(p_sck, &ip, &client_port);
    
94.     SEGGER_RTT_printf(0, "[TCP %x] Incoming connection from %lu.%lu.%lu.%lu:%lu\n",
    
95.                       p_sck, (ip >> 24) & 0xFF, (ip >> 16) & 0xFF,
    
96.                       (ip >> 8) & 0xFF, (ip) & 0xFF, client_port);
    
97. #endif
    
98. 
    
99.     /* Attempt to find another idle socket to start listening on */
    
100.     ULONG state = 0;
     
101. 
     
102.     for (INT i = 0; i < APP_TCP_SOCKET_NUM; i++)
     
103.     {
     
104.         /* Get socket state value */
     
105.         nx_tcp_socket_info_get(g_tcp_sck + i,
     
106.                                0, 0, 0, 0, 0, 0, 0, &state, 0, 0, 0);
     
107. 
     
108.         /* Start lisnening if socket is idle */
     
109.         if (NX_TCP_CLOSED == state)
     
110.         {
     
111.             nx_tcp_server_socket_listen(&g_ip, port, g_tcp_sck + i, 0,
     
112.                                         g_tcp_sck_listen_cb);
     
113. 
     
114.             break;
     
115.         }
     
116.     }
     
117. 
     
118.     /* Ran out of sockets, set appropriate flag to let next socket to
     
119.      * disconnect know that it should start listening right away. */
     
120.     if (NX_TCP_CLOSED != state)
     
121.     {
     
122.         g_not_listening = TX_TRUE;
     
123.     }
     
124. }
     
125. 
     
126. 
     
127. static void g_tcp_sck_receive_cb(NX_TCP_SOCKET * p_sck)
     
128. {
     
129.     NX_PACKET * p_packet;
     
130. 
     
131.     /* This callback is invoked when data is already received. Retrieving
     
132.      * packet with no suspension. */
     
133.     nx_tcp_socket_receive(p_sck, &p_packet, NX_NO_WAIT);
     
134. 
     
135. #if APP_TCP_RTT_LOG
     
136.     ULONG ip, client_port;
     
137.     nx_tcp_socket_peer_info_get(p_sck, &ip, &client_port);
     
138.     SEGGER_RTT_printf(0, "[TCP %x] Incoming packet (%lu bytes) from %lu.%lu.%lu.%lu:%lu\n",
     
139.                       p_sck, p_packet->nx_packet_length, (ip >> 24) & 0xFF,
     
140.                       (ip >> 16) & 0xFF, (ip >> 8) & 0xFF, (ip) & 0xFF, client_port);
     
141. #endif
     
142. 
     
143. #if APP_TCP_INSTANT_ECHO
     
144.     /* Send packet back on the same TCP socket */
     
145.     nx_tcp_socket_send(p_sck, p_packet, NX_NO_WAIT);
     
146. #else
     
147.     ULONG msg[2] =
     
148.     {
     
149.         (ULONG) p_sck, (ULONG) p_packet
     
150.     };
     
151. 
     
152.     /* Sent TCP socket and packet pointers to user thread */
     
153.     if (TX_SUCCESS != tx_queue_send(&g_tcp_q, msg, TX_NO_WAIT))
     
154.     {
     
155.         nx_packet_release(p_packet);
     
156.     }
     
157. #endif
     
158. }
     
159. 
     
160. 
     
161. static void g_tcp_sck_disconn_cb(NX_TCP_SOCKET * p_sck)
     
162. {
     
163. #if APP_TCP_RTT_LOG
     
164.     SEGGER_RTT_printf(0, "[TCP %x] Client disconnected\n", p_sck);
     
165. #endif
     
166. 
     
167.     nx_tcp_server_socket_unaccept(p_sck);
     
168. 
     
169.     /* If all sockets are busy, start listening again */
     
170.     if (TX_TRUE == g_not_listening)
     
171.     {
     
172.         nx_tcp_server_socket_listen(&g_ip, APP_TCP_PORT, p_sck, 0,
     
173.                                     g_tcp_sck_listen_cb);
     
174. 
     
175.         g_not_listening = TX_FALSE;
     
176.     }
     
177. }
     

复制代码

eric2013

谢谢分享