STM32F7 LWIP socket 创建TCP server
6 [% x. `, r# B/ o) J5 g+ l9 |

1. **阻塞方式：**

复制代码

1. typedef struct {
   
2.     struct sockaddr_in client_info;
   
3.     s32 id;
   6 }; u/ T) N" O, b
4.     u16 port;
   2 I! O0 O) I- t9 y, G% e2 a
5. } client_info_s;
   ! _1 f. Z% u# E2 V+ |" {
6. 
   
7. void client_proc(void* parm) {
   
8.     client_info_s* info = (client_info_s*)parm;
   
9.     u8* rec_data = sys_malloc(1500);
   
10. 
    0 v* M3 A" G$ Y6 `& A) p9 B
11.     WDBG("PORT:%d client[%d] %s:%d\n", info->port, info->id,
    
12.         inet_ntoa(info->client_info.sin_addr), info->client_info.sin_port);
    
13. 
    
14.     while(1) {
    6 I' I) z% S, n' `. `9 h
15.         s32 ret = read(info->id, rec_data, 1500);
    / R1 @+ z" g- s6 U+ X0 r' z
16. 
    # N7 g! e7 R: B8 I
17.         if(ret > 0) {
    ( `: d' o) Z3 o+ q
18.             WDBG("PORT:%d rec[%d]:%d",info->port, info->id, ret);
    
19.             ret = write(info->id, rec_data, ret);
    
20. 
    
21.             if(ret <  0) {
    & ^. s4 J3 T3 p  }; V% x/ r
22.                 WDBG("PORT:%d client[%d] write err:%s:%d",info->port, info->id,
    
23.                     inet_ntoa(info->client_info.sin_addr), info->client_info.sin_port);
    
24.                 sys_free(rec_data);
    
25.                 osThreadExit();
    
26.             }
    
27.         } else if(ret < 0 ) {
    
28.             WDBG("PORT:%d client[%d] read err:%s:%d", info->port,info->id,
    
29.                     inet_ntoa(info->client_info.sin_addr), info->client_info.sin_port);
    
30.             close(info->id);
    
31.             sys_free(rec_data);
    
32.             osThreadExit();
    3 U; I' D) @! L
33.         }            
    ' G4 ~- u# \) k5 ?
34.     }
    
35. }
    
36. 
    
37. typedef struct {
    
38.     u16 port;
    
39.     u8 cnt;
    
40.     u8 rvd;
    2 }. [' p' R! z. l. ?5 ~4 o8 j* K, s
41. } server_parm_s;
    / v  k1 T! m( W- y" {& a* j$ i
42. 
    : v8 j( t; |% r( {, b
43. void create_tcp_task(void* pram) {
    
44.     server_parm_s parm;
    
45.     memcpy(&parm, &pram, 4);
    / u1 B, G# p8 I" W# C7 f4 a6 ?
46.     WDBG("TCP SEVER port:%d cnt:%d", parm.port, parm.cnt);
    
47.     int listenfd, connfd;
    
48.     struct sockaddr_in servaddr;
    9 Q, F- H( A  Z
49.     struct sockaddr client_addr;
    , O1 s9 k! j4 d; v% P! w
50. 
    
51.     if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 )
    
52.     {
    
53.         WDBG("create socket error: %s(errno: %d)",strerror(errno),errno);
    - l4 Z: j9 f" n  Q+ v
54.         while(1);
    
55.     }
    + g3 d$ b( c  O! V: I& l
56.     memset(&servaddr, 0, sizeof(servaddr));
    
57.     servaddr.sin_family = AF_INET;
    * |9 [9 w9 N: h+ O: ]5 w3 B
58.     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);            
    * ]" {6 n! _! n, K6 `4 l
59.     servaddr.sin_port = htons(parm.port);
    
60. 
    7 H! v1 x0 y& o9 P  E, T* ^
61.     if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1)
    
62.     {
    - e: ]) }3 v0 v
63.         WDBG("bind socket error: %s(errno: %d)",strerror(errno),errno);
    
64.         while(1);
    
65.     }
    * x2 i8 F: n2 ~
66. 
    
67.     if( listen(listenfd, parm.cnt) == -1)
    
68.     {
    
69.         WDBG("listen socket error: %s(errno: %d)",strerror(errno),errno);
    ; l- D% a8 W# k" H9 i6 i. X
70.     }
    
71. 
    
72.     while(1)
    
73.     {
    3 u2 \! R4 k7 F4 q
74.         socklen_t socklen = sizeof(client_addr);
    
75.         memset(&client_addr, 0, sizeof(client_addr));
    
76.         if( (connfd = accept(listenfd, (struct sockaddr*)&client_addr, &socklen)) == -1)
    ' G" B$ ?( {0 O
77.         {
    
78.             WDBG("accept socket error: %s(errno: %d)",strerror(errno),errno);
    
79.             continue;
    8 I) k# ?: B# N1 }6 e0 ]2 \, b
80.         }
    
81. 
    
82.         client_info_s* client = sys_malloc(sizeof(client_info_s));
    $ ~3 ~0 x! m2 h1 D7 j9 k& }
83. 
    . E2 R4 t: E& V
84.         if(client) {
    6 e  B) t6 F3 N  {3 `
85.             client->client_info = *(struct sockaddr_in*)&client_addr;
    4 V9 G4 \3 a0 K9 }6 j& c
86.             client->id = connfd;
    
87.             client->port = parm.port;
      G) B5 {7 s4 `  L# n; D
88.             osThreadNew(client_proc,(void*)client, NULL);
    $ E$ z! T9 q$ G0 [
89.         }
    
90. 
    
91.     }
    ' D$ F  d/ Z6 y( e9 C
92. }
    ' I, i8 B* c# v
93. 
    $ x( e! n% j4 N" A  i
94. void create_tcp_server(u16 port, u16 cnt){
    
95.     server_parm_s parm;
    6 X+ H( C. a2 U& E% B; @
96.     parm.port = port;
    8 v) _3 i' N6 Q! e
97.     parm.cnt = cnt;
    ) p  I6 |6 e( x/ i& L( z. `
98.     u32* ptr =(u32*)&parm;
    
99.    osThreadNew(create_tcp_task,(void*)(ptr[0]), NULL);
    ; N, T" f$ K1 Z! g
100. }

复制代码

非阻塞方式：
1 ~- |' y2 l- Q4 X8 N
. y- b: a3 B: ~$ j6 r" H

1. void trace_task(void *argument) {
   
2.     const TickType_t xBlockTime = pdMS_TO_TICKS(50);
   
3.     u8* read_buf = malloc(128);
   
4.     WDBG("TCP SEVER port:%d", 1234);
   , w) N& R; u" K# `
5.     int listenfd = -1;
   
6.     int connfd = -1;
   # I% \1 |" s5 V8 q" P) A
7.     struct sockaddr_in servaddr;
   , w$ o7 u/ g5 P/ T0 z4 R
8.     struct sockaddr_in client_addr;
   ( O. b/ l. P8 s0 T% K9 r4 q: \2 Q0 A# M
9. 
   
10.     if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 )
    $ n+ u3 w- D# s
11.     {
    1 D; h- K" l: z5 Q, D/ R8 E( I
12.         WDBG("create socket error: %s(errno: %d)",strerror(errno),errno);
    
13.         while(1);
    7 v! _# I- I' \7 a, x; N
14.     }
    
15.     memset(&servaddr, 0, sizeof(servaddr));
    
16.     servaddr.sin_family = AF_INET;
    
17.     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);            
    
18.     servaddr.sin_port = htons(1234);
    . L# ?0 h6 ~* X) H
19. 
    
20.     if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1)
    4 U( O7 \0 H; B7 k0 Y/ z( t; N
21.     {
    , d1 b% _: k( l6 Q* v8 B- N( t
22.         WDBG("bind socket error: %s(errno: %d)",strerror(errno),errno);
    
23.         while(1);
    1 Q) u" F# E. d8 P( c+ Y
24.     }
    ; ^9 I* a6 G0 O! f& S' l! m9 ^
25. 
    
26.     if( listen(listenfd, 3) == -1)
    ( ^4 Q% X3 T4 b
27.     {
    ( {% z! g+ D! i! Y
28.         WDBG("listen socket error: %s(errno: %d)",strerror(errno),errno);
    ( z6 P1 n/ A  Z; h4 |) Q5 N
29.     }
    
30. 
    ( V' k5 ?! D/ J( v3 A
31.     {
    8 Z" i' i0 E: y1 X" K$ N9 `3 u
32.         DWORD nMode = 1;
    
33.         ioctlsocket( listenfd, FIONBIO, &nMode );        
    8 s6 X% I; `7 M8 V; Z
34.     }
    
35. 
    
36.     fd_set fdRead;
    
37.     fd_set fd_err;
    8 a0 J, [; Q" d6 D+ p
38.     struct timeval tv;
    $ S5 q" S4 k* k% B" M7 \
39.     socklen_t socklen = sizeof(client_addr);
    6 a3 N; e3 }1 K* Y' }0 m" n
40.     memset(&client_addr, 0, sizeof(client_addr));
    
41. 
    
42.     while(1) {
    
43.         FD_ZERO(&fdRead);
    $ X$ t: u) X, \4 \/ u7 X
44.         FD_ZERO(&fd_err);  
    
45.         FD_SET( listenfd, &fdRead);
    
46.         u8 cnt = 2;
    2 u8 K$ R( k. j+ c) i6 O
47.         if(connfd != -1) {
    . k# S0 g. X+ r% k+ ^1 V  q1 M/ n
48.             cnt++;
    4 \3 T9 h0 ]0 S0 V
49.             FD_SET(connfd, &fdRead);
    
50.             FD_SET(connfd, &fd_err);
    
51.         }
    . p& Q2 H- L! w. |- y/ _/ H
52. 
    
53.         if(connfd != -1) {
    # B1 S' x+ S- a+ h# x& ?8 V# m) j
54.             u32 rec_size = xStreamBufferReceive( g_trace_handle,
    6 t3 a9 x: U( O0 n
55.                     ( void * ) read_buf, 128, xBlockTime);
    
56.             if(rec_size) {
    7 s6 j8 X( x8 R/ ~4 s! j
57.                 s32 ret = send(connfd, read_buf, rec_size, 0);
    ! D8 B7 L+ w2 N7 {7 ?2 h9 L
58. 
    
59.                 if(ret <  0)  {
    
60.                     closesocket(connfd);
    * z5 i8 ]. ?& W
61.                     connfd = -1;
    
62.                     continue;
    
63.                 }
      V0 n% Q6 \8 `
64.             }
    % o1 o/ u" O2 R# f
65. 
    : X% v) x& w) p. `% k& [
66.         }
    
67. 
    
68.         tv.tv_sec  = 0;
    9 Y% q5 B2 X! p8 q! ^
69.         tv.tv_usec = 20000;
    ( n. W9 ?3 d6 |) ?
70.         s32 ret = select(cnt, &fdRead, NULL, &fd_err, &tv);
    
71. 
    
72.         if(ret == 0) {
    
73. 
    9 F# @& q" M7 Y: c
74.             continue;
    
75.         }
    
76. 
    
77.         if(ret > 0) {
    1 ]/ M' n7 j) F& J3 |" h! U
78.             if(connfd != -1 && FD_ISSET(connfd, &fd_err)){//rec.
    0 E& v; [& U, n) ?1 L
79.                 connfd = -1;
    * j1 B: e" H" _
80.                 continue;
    
81.             }
    8 g5 I( H  g. c. V  K0 l4 k
82. 
    * K& N8 C; S; w' q) ^% ~
83.             if(connfd != -1 && FD_ISSET(connfd, &fdRead)){//rec.
    ! g& w& C  f5 j! X2 k% q" C! Z# [5 I
84.                 ret = recv(connfd, read_buf, 128, 0);
    
85.                 if(ret < 0) {
    - G6 L5 T$ H9 O/ g% |4 T( }
86.                     closesocket(connfd);
    
87.                     connfd = -1;
    
88.                     continue;
    + o1 T: W! N. j
89.                 } else if(ret >0) {
    . [" y$ t4 U; ~. g9 r2 {# _, `; t, }
90.                     WDBG("rec size:%d", ret);
    
91.                 }
    
92.             }   
    ; M; W3 @( O7 J) p/ L
93. 
    * q7 Y) d' O5 G/ O3 v1 Z
94.             if(FD_ISSET(listenfd, &fdRead)) {
    
95.                 s32 temp = accept(listenfd, (struct sockaddr*)&client_addr, &socklen);
    
96.                 // WDBG("-client:[%d] %s:%dn", temp, inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
    
97. 
    7 p/ Q8 a# x9 O! h; x9 q4 a
98.                 if(connfd == -1) {
    # Q1 X- H3 D+ ^* h* z
99.                     connfd = temp;
    
100.                 } else {
     7 `7 N' s+ U" S, ^# B8 E) ]
101.                     closesocket(temp);
     
102.                 }
     * O' }/ j- b5 v
103.             }        
     
104.         }
     
105.     }
     
106.     #endif
       o+ e7 `0 D0 }: X. D
107. }
       h! T0 N8 Z/ D$ B1 H2 Q1 b
108. 
     6 h  d! Z" B, ]+ _

复制代码