STM32F7 LWIP socket 创建TCP server

1. **阻塞方式：**

复制代码

1. typedef struct {
   
2.     struct sockaddr_in client_info;
   
3.     s32 id;
     l8 e0 P1 q1 R: j2 y5 f8 q- V8 {& I. J
4.     u16 port;
   6 V+ d3 s; N, M8 a. C
5. } client_info_s;
   
6. 
   
7. void client_proc(void* parm) {
   7 I' I/ M' x' R6 V
8.     client_info_s* info = (client_info_s*)parm;
   
9.     u8* rec_data = sys_malloc(1500);
   ! G/ j% I; M- a" V1 m" [2 g
10. 
    
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);
    , P4 V9 ^7 N" H
13. 
    % }3 C& M0 W3 s  u9 @5 p. X2 r
14.     while(1) {
    
15.         s32 ret = read(info->id, rec_data, 1500);
    # t6 r+ U( s5 v  n% N  w) \: c
16. 
    
17.         if(ret > 0) {
    5 p. w7 w0 ]& B/ J* U6 d% ^: X
18.             WDBG("PORT:%d rec[%d]:%d",info->port, info->id, ret);
    
19.             ret = write(info->id, rec_data, ret);
    
20. 
    ! Z( W+ a7 f1 L7 L
21.             if(ret <  0) {
    * R. B; t1 w8 S+ h' `
22.                 WDBG("PORT:%d client[%d] write err:%s:%d",info->port, info->id,
    ! N: ?8 K, J( c9 E1 G  E
23.                     inet_ntoa(info->client_info.sin_addr), info->client_info.sin_port);
    4 m- O$ i7 N* f3 e0 x* M
24.                 sys_free(rec_data);
    * r, o2 v# S. Z; b8 |" C# I! p+ i
25.                 osThreadExit();
    
26.             }
    $ a5 h0 j" q0 R+ y. i! q7 V
27.         } else if(ret < 0 ) {
    - j8 L  g1 [: \  L! f+ l
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);
    , u+ [/ K2 S# J+ e  x! @
30.             close(info->id);
    : ?- U; d1 `2 e5 \
31.             sys_free(rec_data);
    
32.             osThreadExit();
    ! ~' [+ Y* d/ }" \6 D
33.         }            
    
34.     }
    4 x8 T. }0 ?3 i; x  C: v2 T) N6 m
35. }
    
36. 
    # e" u) ~7 ~$ x
37. typedef struct {
      \+ x  j" H! E9 v- |5 U2 @( _( r, w: x
38.     u16 port;
    
39.     u8 cnt;
    
40.     u8 rvd;
    ! y" Q& \. U: w7 R3 N
41. } server_parm_s;
    
42. 
    
43. void create_tcp_task(void* pram) {
    
44.     server_parm_s parm;
    
45.     memcpy(&parm, &pram, 4);
    ; P" `& {0 u0 I+ F, Y# r0 v, j, X
46.     WDBG("TCP SEVER port:%d cnt:%d", parm.port, parm.cnt);
    
47.     int listenfd, connfd;
    
48.     struct sockaddr_in servaddr;
    
49.     struct sockaddr client_addr;
    
50. 
    
51.     if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 )
    
52.     {
    # Q, h1 e; _  G3 i. R- ~4 G
53.         WDBG("create socket error: %s(errno: %d)",strerror(errno),errno);
    % d5 o5 Y. r2 p$ y
54.         while(1);
    
55.     }
    
56.     memset(&servaddr, 0, sizeof(servaddr));
    - i. z3 ~9 Y; K2 q4 P: E+ Q
57.     servaddr.sin_family = AF_INET;
    6 {! d+ F# g7 a1 V6 ?. x
58.     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);            
    
59.     servaddr.sin_port = htons(parm.port);
    ! b' J7 {. K1 V
60. 
    ; i* t" H* @9 Z
61.     if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1)
    ! y! G! Y) l" E+ k& C2 [
62.     {
    
63.         WDBG("bind socket error: %s(errno: %d)",strerror(errno),errno);
    
64.         while(1);
    
65.     }
    
66. 
    
67.     if( listen(listenfd, parm.cnt) == -1)
    & F$ t7 c% v8 c% O* L8 s
68.     {
    
69.         WDBG("listen socket error: %s(errno: %d)",strerror(errno),errno);
      }+ t7 A- k6 n
70.     }
    
71. 
    , u9 I. T# U/ E2 A: G
72.     while(1)
    
73.     {
    
74.         socklen_t socklen = sizeof(client_addr);
    ' g# Y9 I5 K! ^2 i; }* f* s5 G
75.         memset(&client_addr, 0, sizeof(client_addr));
    ; V$ ~- X: H. g3 d9 u4 p7 e- o
76.         if( (connfd = accept(listenfd, (struct sockaddr*)&client_addr, &socklen)) == -1)
    / u- g& j' g/ W% b- y
77.         {
    
78.             WDBG("accept socket error: %s(errno: %d)",strerror(errno),errno);
    
79.             continue;
    
80.         }
    
81. 
    
82.         client_info_s* client = sys_malloc(sizeof(client_info_s));
    
83. 
    
84.         if(client) {
    ' }* z" Y8 q% {4 a* n7 z. h
85.             client->client_info = *(struct sockaddr_in*)&client_addr;
    
86.             client->id = connfd;
    ( W6 H* o/ L: c% k5 _1 h5 H9 F8 a
87.             client->port = parm.port;
    
88.             osThreadNew(client_proc,(void*)client, NULL);
    
89.         }
    ! G4 l# E7 Y7 e$ K3 ~! r! q
90. 
    # L3 E9 ?$ \8 O5 u/ g! P
91.     }
    0 y' t- L5 t8 Y: M3 `- W
92. }
    
93. 
    
94. void create_tcp_server(u16 port, u16 cnt){
    2 ~( R* k6 p/ b3 Z7 p
95.     server_parm_s parm;
    " L' {7 {; i  M
96.     parm.port = port;
    
97.     parm.cnt = cnt;
    4 K9 |% {( F  c- F
98.     u32* ptr =(u32*)&parm;
    - S4 c9 C9 {8 U- _- s
99.    osThreadNew(create_tcp_task,(void*)(ptr[0]), NULL);
    - d/ Y7 S4 t  D. @2 h$ a4 L8 c/ I- p
100. }

复制代码

非阻塞方式：
" ]1 J2 e# Q  b! @3 h$ V5 K
7 g0 z" p5 e. K  X0 y5 X

1. void trace_task(void *argument) {
   7 T2 S* Y1 J1 a* t9 S  C- b
2.     const TickType_t xBlockTime = pdMS_TO_TICKS(50);
   
3.     u8* read_buf = malloc(128);
   
4.     WDBG("TCP SEVER port:%d", 1234);
   % _1 S5 c0 @: N
5.     int listenfd = -1;
   5 R9 }; U" x9 }( Y
6.     int connfd = -1;
   
7.     struct sockaddr_in servaddr;
   
8.     struct sockaddr_in client_addr;
   
9. 
   
10.     if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 )
    
11.     {
    
12.         WDBG("create socket error: %s(errno: %d)",strerror(errno),errno);
    
13.         while(1);
    ' |9 V1 v& B% ?$ B
14.     }
    
15.     memset(&servaddr, 0, sizeof(servaddr));
    1 a$ [0 h' |+ C# z/ s
16.     servaddr.sin_family = AF_INET;
    % l2 A/ p. p+ N2 S2 v& u
17.     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);            
    - o3 V% ]( r3 R4 v
18.     servaddr.sin_port = htons(1234);
    
19. 
    
20.     if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1)
    
21.     {
    5 O- F# T3 o; a7 }5 x, y: [
22.         WDBG("bind socket error: %s(errno: %d)",strerror(errno),errno);
    / J0 q6 u- z- l+ c0 X6 Q  i
23.         while(1);
    ( n0 S  H% \: _0 a- u2 T" n
24.     }
    
25. 
    
26.     if( listen(listenfd, 3) == -1)
    
27.     {
    ' K: A# y1 A+ J. Z& K  e& e7 }
28.         WDBG("listen socket error: %s(errno: %d)",strerror(errno),errno);
    1 y- ?9 N( e% V3 f4 U8 q
29.     }
    
30. 
    
31.     {
    + p- x' S9 \# D2 j. q
32.         DWORD nMode = 1;
    0 [4 n/ V1 x! T& I4 ]# J0 Q! g
33.         ioctlsocket( listenfd, FIONBIO, &nMode );        
      s& Z  Q* R9 D
34.     }
    $ c' X% |- s, \7 L
35. 
    
36.     fd_set fdRead;
    ; {2 u7 ~" Z$ b* v, ^# `; A
37.     fd_set fd_err;
    
38.     struct timeval tv;
    - Z. Q( L0 P8 P3 ?* v6 J
39.     socklen_t socklen = sizeof(client_addr);
    
40.     memset(&client_addr, 0, sizeof(client_addr));
    
41. 
    8 H- @% h: [8 d% i: Q; d. [
42.     while(1) {
    9 L  W( ?8 ~0 P( E( ]( b
43.         FD_ZERO(&fdRead);
    . C: E! t5 e+ u8 e3 Z9 \+ W
44.         FD_ZERO(&fd_err);  
    
45.         FD_SET( listenfd, &fdRead);
    
46.         u8 cnt = 2;
    / V) G7 f! C3 }9 _' H
47.         if(connfd != -1) {
    3 B) `/ b6 a+ ]" ?; v; k( y
48.             cnt++;
    : u1 G" f: x8 ]; V! ^
49.             FD_SET(connfd, &fdRead);
    
50.             FD_SET(connfd, &fd_err);
    : L9 c) Q; g8 R5 U
51.         }
    
52. 
    9 B4 E3 I0 o! G9 N# b9 L, [1 V6 M0 i* E5 D, N
53.         if(connfd != -1) {
    
54.             u32 rec_size = xStreamBufferReceive( g_trace_handle,
    ( [. |& Z6 _( I* K5 B" F
55.                     ( void * ) read_buf, 128, xBlockTime);
    " E: r& I# Z% i1 Z! L
56.             if(rec_size) {
    # s' n; ?& m: f1 T& X: t% @7 t
57.                 s32 ret = send(connfd, read_buf, rec_size, 0);
    
58. 
    
59.                 if(ret <  0)  {
    
60.                     closesocket(connfd);
    , P; w9 i2 y/ a( ~9 W
61.                     connfd = -1;
    # Q# c; l* I- |6 m; h& ]) z* h
62.                     continue;
    
63.                 }
    
64.             }
    ; h2 y( {7 M( }  i# B4 Y1 N
65. 
    
66.         }
    
67. 
    7 p7 Y; R$ w  `, |8 k$ J
68.         tv.tv_sec  = 0;
    4 H( f( Q3 g: Z8 U$ g% i; ?
69.         tv.tv_usec = 20000;
    
70.         s32 ret = select(cnt, &fdRead, NULL, &fd_err, &tv);
    
71. 
    
72.         if(ret == 0) {
    
73. 
      @' E% S/ P/ A/ v8 _* y! ?8 M
74.             continue;
    
75.         }
    
76. 
    1 a# G* a1 B, x; t! Y: m/ N* @1 H
77.         if(ret > 0) {
    5 J& b/ B' C6 p6 `
78.             if(connfd != -1 && FD_ISSET(connfd, &fd_err)){//rec.
    # {' |; W* Q  j' Q9 K
79.                 connfd = -1;
    : @# E) g! V' p! o) F7 `4 s1 }; O
80.                 continue;
    
81.             }
    
82. 
    - H8 Z0 ]+ b, ^0 t. ~) u  D1 }
83.             if(connfd != -1 && FD_ISSET(connfd, &fdRead)){//rec.
    2 N3 ]: ], P2 z% H
84.                 ret = recv(connfd, read_buf, 128, 0);
    
85.                 if(ret < 0) {
    8 |" S) z3 I) ~
86.                     closesocket(connfd);
    9 D4 t* R% ]2 X6 w# L) r
87.                     connfd = -1;
    5 B4 G6 `. i) Q$ C: A0 \5 U; w
88.                     continue;
    
89.                 } else if(ret >0) {
    
90.                     WDBG("rec size:%d", ret);
    
91.                 }
    
92.             }   
    
93. 
    
94.             if(FD_ISSET(listenfd, &fdRead)) {
    4 c+ P) h2 ?$ J- U$ k' u
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. 
    
98.                 if(connfd == -1) {
    
99.                     connfd = temp;
    5 k" S* V; T' v
100.                 } else {
     
101.                     closesocket(temp);
     9 s4 ~! P$ h; q# W9 U5 q& q- Y' f
102.                 }
     % i# @' w) n- {2 ?/ f* N
103.             }        
     ! ]' R7 c1 v" W% ]  g- J# m
104.         }
     ' r& K& \; V" B6 s2 f
105.     }
     
106.     #endif
     
107. }
     
108. 
     , k% \1 l8 r4 J9 v, t

复制代码