STM32F7 LWIP socket 创建TCP server

1. **阻塞方式：**

复制代码

1. typedef struct {
   + s9 |1 O1 k$ E) Q2 p' m" C( j
2.     struct sockaddr_in client_info;
   ; }# s3 H4 p% J) F& b& E
3.     s32 id;
   ; n% W+ i( l8 z0 c, ~
4.     u16 port;
   
5. } client_info_s;
   5 V3 X, F/ o3 ]
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 \; y' `- H! c
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. 
    7 r/ Z. G+ W/ w+ ~
14.     while(1) {
    
15.         s32 ret = read(info->id, rec_data, 1500);
    
16. 
    + Y! [+ Q1 r2 X6 R' }$ Y
17.         if(ret > 0) {
    ! d: {0 D4 k0 ~; e
18.             WDBG("PORT:%d rec[%d]:%d",info->port, info->id, ret);
    
19.             ret = write(info->id, rec_data, ret);
    
20. 
    : K# q+ ^& s; M* S) F/ K
21.             if(ret <  0) {
    - M# `# ^) T: z" \' z
22.                 WDBG("PORT:%d client[%d] write err:%s:%d",info->port, info->id,
    . W1 d& C& ?9 k/ ~
23.                     inet_ntoa(info->client_info.sin_addr), info->client_info.sin_port);
    
24.                 sys_free(rec_data);
    
25.                 osThreadExit();
    9 u7 I1 [5 |5 c- |) i5 o
26.             }
    
27.         } else if(ret < 0 ) {
    3 g7 _! o+ K4 W3 [- y
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);
    6 Y- x- ~5 c$ r" v8 \
31.             sys_free(rec_data);
    & e0 x  G. Q/ @, w& h9 }: h
32.             osThreadExit();
    
33.         }            
    . a* Z% [/ \* ^& E7 p: Z
34.     }
    / _7 K; i" _! O: C* ]
35. }
    
36. 
    
37. typedef struct {
    
38.     u16 port;
    ( C: R. M+ a: a. X3 X. n. O" d
39.     u8 cnt;
    
40.     u8 rvd;
    
41. } server_parm_s;
    
42. 
    
43. void create_tcp_task(void* pram) {
    
44.     server_parm_s parm;
    6 N0 q) m% Y) p) h" q( H$ Z
45.     memcpy(&parm, &pram, 4);
    * B' j/ Z  h" ]
46.     WDBG("TCP SEVER port:%d cnt:%d", parm.port, parm.cnt);
    , U3 X9 R3 f% i& T9 P, [
47.     int listenfd, connfd;
    - R" ^! w5 m  x
48.     struct sockaddr_in servaddr;
    
49.     struct sockaddr client_addr;
    3 o. S/ ]4 r! l1 x5 w
50. 
    3 R" E) ?% r  E+ A0 r0 g9 E& V0 M
51.     if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 )
    ! `6 A8 r9 d! ~) y2 T1 N6 I$ e
52.     {
    
53.         WDBG("create socket error: %s(errno: %d)",strerror(errno),errno);
    
54.         while(1);
    
55.     }
    
56.     memset(&servaddr, 0, sizeof(servaddr));
    , M* M; o$ o1 u/ M( v
57.     servaddr.sin_family = AF_INET;
    & a: u( B- l! k
58.     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);            
    , G! h: b6 w/ _, G9 M
59.     servaddr.sin_port = htons(parm.port);
    
60. 
    
61.     if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1)
    # }" }$ ^- B+ D7 m' H3 f' M  ]
62.     {
    
63.         WDBG("bind socket error: %s(errno: %d)",strerror(errno),errno);
    $ {' j- e1 u. U" k8 O! \! Z
64.         while(1);
    
65.     }
    
66. 
    + D4 t& ~5 O- s' F! i
67.     if( listen(listenfd, parm.cnt) == -1)
    
68.     {
    
69.         WDBG("listen socket error: %s(errno: %d)",strerror(errno),errno);
    0 W# Z: N* o5 z! z- P9 t4 B
70.     }
    
71. 
    6 s  D& T4 M- I3 ]+ L
72.     while(1)
    
73.     {
    
74.         socklen_t socklen = sizeof(client_addr);
    7 v! L$ v$ ?8 c. t0 u
75.         memset(&client_addr, 0, sizeof(client_addr));
    
76.         if( (connfd = accept(listenfd, (struct sockaddr*)&client_addr, &socklen)) == -1)
    
77.         {
    ; p+ y7 A/ M- a: e
78.             WDBG("accept socket error: %s(errno: %d)",strerror(errno),errno);
    + D1 E  H5 l( v' v+ D1 m  v
79.             continue;
    
80.         }
    
81. 
    + ^  {! b& t. @1 u
82.         client_info_s* client = sys_malloc(sizeof(client_info_s));
    
83. 
    / x. d# `+ m( ^! r8 |4 h! e, B7 A$ `% v
84.         if(client) {
    
85.             client->client_info = *(struct sockaddr_in*)&client_addr;
    3 K$ R: C% |; ?9 f+ q
86.             client->id = connfd;
    ) n: [5 ~7 x8 v2 E5 _
87.             client->port = parm.port;
    0 ?3 Y' b+ _3 d: V4 ?9 e& |' C
88.             osThreadNew(client_proc,(void*)client, NULL);
    6 F; F+ b, e% b0 x) V* m
89.         }
    
90. 
    
91.     }
    
92. }
    
93. 
    
94. void create_tcp_server(u16 port, u16 cnt){
    7 X4 C$ H3 S0 i) x
95.     server_parm_s parm;
    
96.     parm.port = port;
    ' Y! O$ g5 o& h2 W; J, C- E$ k' o
97.     parm.cnt = cnt;
    
98.     u32* ptr =(u32*)&parm;
    : d2 A' D" N- D- N
99.    osThreadNew(create_tcp_task,(void*)(ptr[0]), NULL);
    
100. }

复制代码

非阻塞方式：

2 t* i: g- ~9 H# s( O

1. void trace_task(void *argument) {
     u8 I2 S3 O% ~) ~
2.     const TickType_t xBlockTime = pdMS_TO_TICKS(50);
   
3.     u8* read_buf = malloc(128);
   
4.     WDBG("TCP SEVER port:%d", 1234);
   
5.     int listenfd = -1;
   3 t! _% n2 |' a0 b  H
6.     int connfd = -1;
   , @+ ]/ I7 ?% V& J* H
7.     struct sockaddr_in servaddr;
   
8.     struct sockaddr_in client_addr;
   , n: }9 F% u( u- D2 Q
9. 
   ! C7 y# _! M8 m/ Y& l4 ~! Z, h% T
10.     if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 )
    
11.     {
    + u( f5 B& T1 k. G
12.         WDBG("create socket error: %s(errno: %d)",strerror(errno),errno);
    % D4 c1 g1 M7 h2 X
13.         while(1);
    
14.     }
    9 m$ l: S% F/ \0 j- S- z' h
15.     memset(&servaddr, 0, sizeof(servaddr));
    
16.     servaddr.sin_family = AF_INET;
    9 q+ R% T5 c$ P: q
17.     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);            
    : l) B$ G6 X& W; [- i+ u7 b3 k- l* K
18.     servaddr.sin_port = htons(1234);
    . P, o! u$ f9 |. o
19. 
    
20.     if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1)
    
21.     {
    
22.         WDBG("bind socket error: %s(errno: %d)",strerror(errno),errno);
      B- ~. x3 X" r9 I4 e. N
23.         while(1);
    $ k/ J. l. W9 T  X! U5 N
24.     }
    
25. 
    . S( B+ ^. G6 y: h& Q4 U
26.     if( listen(listenfd, 3) == -1)
    % ?8 h. x. S7 ]" E3 J
27.     {
    * m) `4 x% y* D2 X
28.         WDBG("listen socket error: %s(errno: %d)",strerror(errno),errno);
    2 q& s9 l; Q  t$ s; d6 C
29.     }
    
30. 
    7 Z- w% z: V1 A. w
31.     {
    - D, X8 ~0 @6 I. R8 u1 I- Y
32.         DWORD nMode = 1;
    2 B. a' |8 j+ l- \( n
33.         ioctlsocket( listenfd, FIONBIO, &nMode );        
    
34.     }
    
35. 
    
36.     fd_set fdRead;
    
37.     fd_set fd_err;
    # I5 S9 |; ?" Y: U' t
38.     struct timeval tv;
    5 f. [; T5 @8 g5 w0 Y1 h& U' Y
39.     socklen_t socklen = sizeof(client_addr);
    * |# b# v+ N- Q& D, `' f
40.     memset(&client_addr, 0, sizeof(client_addr));
    ) x, v0 e5 A: Z
41. 
    
42.     while(1) {
    
43.         FD_ZERO(&fdRead);
    : Q) Z) e/ J1 Z  }/ i
44.         FD_ZERO(&fd_err);  
    
45.         FD_SET( listenfd, &fdRead);
    
46.         u8 cnt = 2;
    9 t2 ?) H' v8 R
47.         if(connfd != -1) {
    7 J- K' ~$ x5 N  c4 h; W( A0 x8 b+ ^
48.             cnt++;
    
49.             FD_SET(connfd, &fdRead);
    5 O1 U9 c9 B* d' w! G7 R( I3 S
50.             FD_SET(connfd, &fd_err);
    
51.         }
    
52. 
    ! J/ m, O" w) \; v0 O1 g/ j$ ]6 h
53.         if(connfd != -1) {
    
54.             u32 rec_size = xStreamBufferReceive( g_trace_handle,
    + G0 h) h; k" t/ a( Y$ d3 c4 N- [3 B
55.                     ( void * ) read_buf, 128, xBlockTime);
    ! \, `/ J- y  i3 s
56.             if(rec_size) {
    + z; T5 f4 H- [7 `) w
57.                 s32 ret = send(connfd, read_buf, rec_size, 0);
    9 g3 ]. i2 a! Z8 ^0 K
58. 
    
59.                 if(ret <  0)  {
    
60.                     closesocket(connfd);
    
61.                     connfd = -1;
    
62.                     continue;
    5 {9 n8 a1 i, ~6 B3 v
63.                 }
    
64.             }
    
65. 
    
66.         }
    
67. 
    
68.         tv.tv_sec  = 0;
    
69.         tv.tv_usec = 20000;
    # ^- S. G' A7 t: @, V
70.         s32 ret = select(cnt, &fdRead, NULL, &fd_err, &tv);
    
71. 
    # H! O8 ]# y! E" ^7 K' c  Y
72.         if(ret == 0) {
    
73. 
    # n! O0 H* A$ \
74.             continue;
    8 @% w. P8 r  x/ k' N8 _* k
75.         }
    
76. 
    
77.         if(ret > 0) {
    ! H7 M/ e5 N  @' x( R9 |+ `0 H
78.             if(connfd != -1 && FD_ISSET(connfd, &fd_err)){//rec.
    
79.                 connfd = -1;
    0 H) Q/ ~$ L+ B4 U, M, H& Y
80.                 continue;
    
81.             }
    
82. 
    
83.             if(connfd != -1 && FD_ISSET(connfd, &fdRead)){//rec.
    
84.                 ret = recv(connfd, read_buf, 128, 0);
    ) i  y( Y* Q4 Y/ w" e
85.                 if(ret < 0) {
    
86.                     closesocket(connfd);
    ' Q1 y+ }! b/ T* [7 W; |% |
87.                     connfd = -1;
    7 E. @( T0 b, X3 ^1 {! R
88.                     continue;
    . p# C( }' o$ C
89.                 } else if(ret >0) {
    : H" ^; T+ z) J* b
90.                     WDBG("rec size:%d", ret);
    
91.                 }
    1 b4 h" @+ I8 B4 r1 N- z
92.             }   
    # U: o9 e# v  x
93. 
    
94.             if(FD_ISSET(listenfd, &fdRead)) {
    - H6 {4 C0 y* f  m
95.                 s32 temp = accept(listenfd, (struct sockaddr*)&client_addr, &socklen);
    0 d! C4 H. {$ o1 W  k: r# q5 W, ^4 V5 i
96.                 // WDBG("-client:[%d] %s:%dn", temp, inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
    
97. 
    / [) f  O% b! x: N3 M
98.                 if(connfd == -1) {
    
99.                     connfd = temp;
    1 G+ ^! Q( f. c+ s; L. K" j
100.                 } else {
     
101.                     closesocket(temp);
     8 ^' H+ V7 g: Z5 o9 \" t
102.                 }
     
103.             }        
     
104.         }
     ' N9 c% f" [0 p; ~. Q  B6 s
105.     }
     
106.     #endif
     % g: R) J" f2 ?" i$ q/ Q' N( w
107. }
     
108. 
     

复制代码

  U( s; M1 A8 S, |  e* g  @