STM32F7 LWIP socket 创建TCP server

1. **阻塞方式：**

复制代码

1. typedef struct {
   
2.     struct sockaddr_in client_info;
   
3.     s32 id;
   ! L, ^2 r2 c, g2 q, ~' K& Q
4.     u16 port;
   2 O8 F; c1 H- A. S& d2 {
5. } client_info_s;
   
6. 
   ! c  j- R: H& d/ ]) m4 W5 [
7. void client_proc(void* parm) {
   
8.     client_info_s* info = (client_info_s*)parm;
   6 f; N' |. F# b! ^' \  Q
9.     u8* rec_data = sys_malloc(1500);
   6 e+ h) G  [$ R; u' i1 V
10. 
    ; K8 v/ Y" X$ o4 s: l; j- G: c
11.     WDBG("PORT:%d client[%d] %s:%d\n", info->port, info->id,
    9 S5 D0 X0 @; x
12.         inet_ntoa(info->client_info.sin_addr), info->client_info.sin_port);
    ( [" ^  w. w3 w1 J) D/ k2 k
13. 
    
14.     while(1) {
    % t/ Z( l7 s0 r( k4 B
15.         s32 ret = read(info->id, rec_data, 1500);
    
16. 
    + A; U! X' l3 i  @
17.         if(ret > 0) {
    
18.             WDBG("PORT:%d rec[%d]:%d",info->port, info->id, ret);
    - }8 n& r# g3 ^2 J. H
19.             ret = write(info->id, rec_data, ret);
    ! \" y7 D  y. G9 p* Z+ a1 R
20. 
    ' H$ e( i- Q2 l, r  ^
21.             if(ret <  0) {
    
22.                 WDBG("PORT:%d client[%d] write err:%s:%d",info->port, info->id,
    1 k0 g0 `# T% c2 O" L
23.                     inet_ntoa(info->client_info.sin_addr), info->client_info.sin_port);
    
24.                 sys_free(rec_data);
    
25.                 osThreadExit();
    , x% v( D) y& L
26.             }
    , P2 _0 F$ N" o/ T0 ?5 g
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);
    $ Z3 Q- p: S" c9 T, z8 n6 Q
31.             sys_free(rec_data);
    
32.             osThreadExit();
    " q, ]5 c* D8 i* u( x
33.         }            
    
34.     }
    , W" E2 x4 Y+ A" p( v
35. }
    
36. 
    ( \5 @& B) ^7 o
37. typedef struct {
    
38.     u16 port;
    
39.     u8 cnt;
    " F; l0 _$ m+ T* {' }
40.     u8 rvd;
    " F0 D! b" x( y- k" s; I* W
41. } server_parm_s;
    
42. 
    
43. void create_tcp_task(void* pram) {
    
44.     server_parm_s parm;
    + @$ c) X. i2 y7 L
45.     memcpy(&parm, &pram, 4);
    
46.     WDBG("TCP SEVER port:%d cnt:%d", parm.port, parm.cnt);
    ) }( O* f4 {7 V( b0 |
47.     int listenfd, connfd;
    3 |# z6 T' R3 l( b
48.     struct sockaddr_in servaddr;
    ) n, `7 M/ k  x9 m' h8 H
49.     struct sockaddr client_addr;
    
50. 
    ' j# w( E# z- I. E, g; u
51.     if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 )
    
52.     {
    
53.         WDBG("create socket error: %s(errno: %d)",strerror(errno),errno);
    7 d; X  }" j* c$ o) Y, E
54.         while(1);
    
55.     }
    
56.     memset(&servaddr, 0, sizeof(servaddr));
    * S7 V. S5 M* u5 x1 |
57.     servaddr.sin_family = AF_INET;
      D* w# Z6 W  y( J. G3 o$ J
58.     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);            
    
59.     servaddr.sin_port = htons(parm.port);
    
60. 
    3 @2 w  l6 ?& n) T  I2 {
61.     if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1)
    
62.     {
    ( \- [& c8 ~$ S
63.         WDBG("bind socket error: %s(errno: %d)",strerror(errno),errno);
    
64.         while(1);
    , f& b8 t' x1 C0 Z) a# G
65.     }
    
66. 
    
67.     if( listen(listenfd, parm.cnt) == -1)
    + n# y) s& `6 l3 J, [' k
68.     {
    
69.         WDBG("listen socket error: %s(errno: %d)",strerror(errno),errno);
    
70.     }
    , a& U5 d+ @! \8 G: n  Z- l! o
71. 
    
72.     while(1)
      C2 Q4 g# M( R+ L. A$ A
73.     {
    ) l& h% D- t0 ~4 n" s
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)
    $ e6 r. `; Z2 n1 S/ t
77.         {
    $ R' @3 Q4 w- q5 m2 m
78.             WDBG("accept socket error: %s(errno: %d)",strerror(errno),errno);
    
79.             continue;
    
80.         }
    ( v3 ?. @: T% W) \  U2 m  [
81. 
    / a7 s+ [, g* O0 T' v5 S
82.         client_info_s* client = sys_malloc(sizeof(client_info_s));
    ! p& P) O$ r0 l2 \0 x0 H
83. 
    * i  t" x$ K* n" {4 L+ ^8 l" f
84.         if(client) {
    & S4 Q5 h, q' u9 i1 m- O
85.             client->client_info = *(struct sockaddr_in*)&client_addr;
    
86.             client->id = connfd;
    
87.             client->port = parm.port;
    7 p1 e: s3 H8 z; D- r$ w5 l) H4 D
88.             osThreadNew(client_proc,(void*)client, NULL);
    8 M) A( L. m9 N8 ?3 h4 y, K$ Q
89.         }
    
90. 
    ' ]/ d% p- C  T
91.     }
    ! {! a% B5 f9 q) z4 ~' j
92. }
    
93. 
    
94. void create_tcp_server(u16 port, u16 cnt){
    # c3 F9 z: S. R: p$ e
95.     server_parm_s parm;
    5 L! i: r+ M& m- t1 @' v7 _
96.     parm.port = port;
    . g! @" o# b, b- r5 F1 |- a# X
97.     parm.cnt = cnt;
    
98.     u32* ptr =(u32*)&parm;
    
99.    osThreadNew(create_tcp_task,(void*)(ptr[0]), NULL);
    ' ~, l, Z" e3 D" a
100. }

复制代码

" B) e, T2 Y6 W非阻塞方式：
1 D1 }5 P6 O- X7 |7 }
+ n. I% Q  `: ~& Q+ h8 y

1. void trace_task(void *argument) {
   
2.     const TickType_t xBlockTime = pdMS_TO_TICKS(50);
   ' v1 r5 ?/ J6 g
3.     u8* read_buf = malloc(128);
   ) d! c# a; @; q" ]' y' H6 a
4.     WDBG("TCP SEVER port:%d", 1234);
   4 {0 t' f0 x$ w* f/ N
5.     int listenfd = -1;
   9 b) A, ?* \+ ~3 Q' x1 v, M* c
6.     int connfd = -1;
   / n# D- s4 F, z2 n* F: G
7.     struct sockaddr_in servaddr;
   
8.     struct sockaddr_in client_addr;
   
9. 
   " w6 L6 O% Y* p0 P6 T- _7 b
10.     if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 )
    
11.     {
    
12.         WDBG("create socket error: %s(errno: %d)",strerror(errno),errno);
    # E4 R5 A6 s: }9 r: B6 z
13.         while(1);
    
14.     }
    ) {: S# `& v6 d  N
15.     memset(&servaddr, 0, sizeof(servaddr));
    2 V8 j6 G3 w5 M7 ^) i
16.     servaddr.sin_family = AF_INET;
    
17.     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);            
    
18.     servaddr.sin_port = htons(1234);
    
19. 
      h$ B$ p+ g- ]# e$ M
20.     if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1)
    
21.     {
    
22.         WDBG("bind socket error: %s(errno: %d)",strerror(errno),errno);
    ( }; o) T9 X  o( r; M
23.         while(1);
    ; J9 \* o( M- b. q; \) @
24.     }
    2 W2 W( E% M; E4 Z1 T
25. 
    
26.     if( listen(listenfd, 3) == -1)
    " U- N9 p1 S0 \4 l* P4 o
27.     {
    - ?( J$ Q/ T9 N7 Q% X! v% ?
28.         WDBG("listen socket error: %s(errno: %d)",strerror(errno),errno);
    ! }% i# g: T  K8 N" G8 g3 h
29.     }
    - f2 G: j; f8 Z4 q6 u, b; _
30. 
    3 W9 O1 u+ N. M6 M
31.     {
    0 w4 o  I- U' @+ a2 w1 m( x1 N
32.         DWORD nMode = 1;
    
33.         ioctlsocket( listenfd, FIONBIO, &nMode );        
    
34.     }
    
35. 
    % T' x: e) p6 D
36.     fd_set fdRead;
    & O% K3 q# W' a
37.     fd_set fd_err;
    
38.     struct timeval tv;
    1 X/ U: p% `3 j$ g
39.     socklen_t socklen = sizeof(client_addr);
    
40.     memset(&client_addr, 0, sizeof(client_addr));
    
41. 
    
42.     while(1) {
    
43.         FD_ZERO(&fdRead);
    - L! q" U7 X  f/ b- {) p
44.         FD_ZERO(&fd_err);  
    3 N8 |! v6 Z+ }; s1 b; x
45.         FD_SET( listenfd, &fdRead);
    . Y# K: Z' b# w
46.         u8 cnt = 2;
    5 H, c% O9 i$ r! E/ V1 a1 Q) X  D# m
47.         if(connfd != -1) {
    
48.             cnt++;
    1 c& v* V$ B4 }' ~  c
49.             FD_SET(connfd, &fdRead);
    9 Y" X8 D9 q0 e( `
50.             FD_SET(connfd, &fd_err);
    9 n  Q5 |* E5 ~9 C  P( ^; s- Z9 I7 F
51.         }
    
52. 
    
53.         if(connfd != -1) {
    0 j: D% h/ m7 z, h2 y. _# o
54.             u32 rec_size = xStreamBufferReceive( g_trace_handle,
    
55.                     ( void * ) read_buf, 128, xBlockTime);
    
56.             if(rec_size) {
    7 ^- K- g0 t; k) b
57.                 s32 ret = send(connfd, read_buf, rec_size, 0);
    1 S. F; D6 m8 {
58. 
    ) ?/ t- u7 t, _; U
59.                 if(ret <  0)  {
    ; {4 q0 x' C' u; w' _
60.                     closesocket(connfd);
    7 U3 t0 {, S* g4 @8 P8 m
61.                     connfd = -1;
    
62.                     continue;
    $ j1 b1 n; J. j4 W% G
63.                 }
    
64.             }
    
65. 
    
66.         }
    
67. 
    
68.         tv.tv_sec  = 0;
    
69.         tv.tv_usec = 20000;
    
70.         s32 ret = select(cnt, &fdRead, NULL, &fd_err, &tv);
      Y1 z' B9 F+ ]1 M3 |. I8 Z0 N1 ?
71. 
    
72.         if(ret == 0) {
    
73. 
    ( B: `* {- c  d; l: N1 c
74.             continue;
    ) y- A$ i8 L" o6 E% W( n( ^
75.         }
    4 j/ R; g9 n" r/ s0 f
76. 
    . J% a! P8 K7 q' ?7 n+ g% Q( s
77.         if(ret > 0) {
    
78.             if(connfd != -1 && FD_ISSET(connfd, &fd_err)){//rec.
    7 W: D7 S" l' a& P' Q
79.                 connfd = -1;
    2 N8 b$ k( I  I) C7 w& K
80.                 continue;
    * Y/ H6 H3 z1 m$ r4 e
81.             }
    
82. 
    2 T+ t; {, M9 l# t. _
83.             if(connfd != -1 && FD_ISSET(connfd, &fdRead)){//rec.
    ( M7 A+ r! b2 n
84.                 ret = recv(connfd, read_buf, 128, 0);
    ) L  j, L$ L$ o! ^/ t: @6 |
85.                 if(ret < 0) {
    
86.                     closesocket(connfd);
    
87.                     connfd = -1;
    $ X0 f* d4 x* B! I
88.                     continue;
    6 I$ K; ^3 W5 [* L" r% s
89.                 } else if(ret >0) {
    
90.                     WDBG("rec size:%d", ret);
    6 u! C7 I9 u2 M( u; ?$ d
91.                 }
    
92.             }   
    
93. 
    
94.             if(FD_ISSET(listenfd, &fdRead)) {
    
95.                 s32 temp = accept(listenfd, (struct sockaddr*)&client_addr, &socklen);
    0 q0 i  V$ B* V/ a% I( k9 {
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;
    
100.                 } else {
     9 K2 \, K1 n' g; s7 X- Y
101.                     closesocket(temp);
     - Z; j8 |2 p0 M& ~& c
102.                 }
     
103.             }        
     0 L& V6 x1 ~. o: l, J! c
104.         }
     ) f; Z! ~6 ]0 h( B( W$ X
105.     }
     ) d  a' K" s* D0 c, u( @- g
106.     #endif
     
107. }
     5 h% ]0 a( C* K1 l8 f) o
108. 
     

复制代码

9 q# K" u5 M& w