STM32F7 LWIP socket 创建TCP server
1 h+ m3 `- l4 K  n1 L

1. **阻塞方式：**

复制代码

1. typedef struct {
   . X6 w' o% w1 F3 Z, v9 U
2.     struct sockaddr_in client_info;
   0 I" D% C$ u' u5 O
3.     s32 id;
   
4.     u16 port;
   
5. } client_info_s;
   6 Q8 {  [. k3 y+ s! _
6. 
   
7. void client_proc(void* parm) {
   
8.     client_info_s* info = (client_info_s*)parm;
   
9.     u8* rec_data = sys_malloc(1500);
   " }! D7 E* o6 H5 r
10. 
    
11.     WDBG("PORT:%d client[%d] %s:%d\n", info->port, info->id,
      w2 X$ F, c, O$ u0 j3 t6 ^
12.         inet_ntoa(info->client_info.sin_addr), info->client_info.sin_port);
    
13. 
    & h1 v8 o. A3 I4 Z( ^2 {* N
14.     while(1) {
    2 D% a: s' k, O7 j' [
15.         s32 ret = read(info->id, rec_data, 1500);
    
16. 
    # C  L0 {9 I0 a; O# H
17.         if(ret > 0) {
    
18.             WDBG("PORT:%d rec[%d]:%d",info->port, info->id, ret);
    
19.             ret = write(info->id, rec_data, ret);
    
20. 
    4 x" H. p; X4 P. M, T. p
21.             if(ret <  0) {
    0 y& E; S- _0 m! k+ [+ [
22.                 WDBG("PORT:%d client[%d] write err:%s:%d",info->port, info->id,
    0 _% x: V# @4 D/ N2 O/ [& Z1 A6 s
23.                     inet_ntoa(info->client_info.sin_addr), info->client_info.sin_port);
    % q- b, I4 [1 `8 D
24.                 sys_free(rec_data);
    
25.                 osThreadExit();
    1 ]" d" N! z9 v9 O8 J( J& w
26.             }
    
27.         } else if(ret < 0 ) {
    
28.             WDBG("PORT:%d client[%d] read err:%s:%d", info->port,info->id,
    9 j. n2 q8 I8 P% k4 i
29.                     inet_ntoa(info->client_info.sin_addr), info->client_info.sin_port);
    
30.             close(info->id);
    
31.             sys_free(rec_data);
    6 U1 I5 S, F- J  n5 G
32.             osThreadExit();
    . M; W' S+ d1 B9 O, |" O) y
33.         }            
    
34.     }
    
35. }
    
36. 
    ) T7 W1 |% B+ y4 |7 R
37. typedef struct {
    
38.     u16 port;
    : R0 N6 A7 h" k
39.     u8 cnt;
    
40.     u8 rvd;
    5 q# y* D4 o' t5 H7 G
41. } server_parm_s;
    3 O- E5 S$ |1 X/ s3 d  P
42. 
    - I7 }8 Z, V# F, k4 O  a
43. void create_tcp_task(void* pram) {
    
44.     server_parm_s parm;
    
45.     memcpy(&parm, &pram, 4);
    * i# H) R/ m; y7 t" o9 o2 U
46.     WDBG("TCP SEVER port:%d cnt:%d", parm.port, parm.cnt);
    , y) p) ?* v" O1 e: }: O' P: `
47.     int listenfd, connfd;
    / U" N1 R+ X5 m5 z6 \; }& m
48.     struct sockaddr_in servaddr;
    , l( y' @- e# N8 _8 |$ u+ Q
49.     struct sockaddr client_addr;
    ! i' j4 T' {+ E- J
50. 
    $ R# w; K  f  }: j/ R( Z+ L
51.     if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 )
    + |- i4 _$ P" f
52.     {
    % f; B- i; G/ e# \9 W, e
53.         WDBG("create socket error: %s(errno: %d)",strerror(errno),errno);
    
54.         while(1);
    
55.     }
    ' o; @6 Q  b# g2 m% M5 o1 S
56.     memset(&servaddr, 0, sizeof(servaddr));
    
57.     servaddr.sin_family = AF_INET;
    ' A; y7 i. `7 u: t
58.     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);            
    6 [0 C7 R$ R# h5 E; |) P* X
59.     servaddr.sin_port = htons(parm.port);
    6 X7 A1 o9 D! a, ~. Q
60. 
    
61.     if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1)
    
62.     {
    
63.         WDBG("bind socket error: %s(errno: %d)",strerror(errno),errno);
    
64.         while(1);
    
65.     }
    - L& U% x0 N9 B3 H+ W$ @' ^. A! b8 ]% r
66. 
    
67.     if( listen(listenfd, parm.cnt) == -1)
    . ]- T6 b! H4 T% |4 _2 v+ a: C. b
68.     {
    
69.         WDBG("listen socket error: %s(errno: %d)",strerror(errno),errno);
    ) u/ ~! q9 T% m. W$ M
70.     }
    $ Z, C. S7 |' Y+ G! A4 w2 R
71. 
    4 [! i* t7 L, q  {. ~  W
72.     while(1)
    3 m2 [# w& i  w, I
73.     {
    
74.         socklen_t socklen = sizeof(client_addr);
    - y: \) ~" G: k. z
75.         memset(&client_addr, 0, sizeof(client_addr));
    1 S' ~- f3 ^& a/ s
76.         if( (connfd = accept(listenfd, (struct sockaddr*)&client_addr, &socklen)) == -1)
    
77.         {
    1 ?# ]# C7 M, e6 ~2 g% f
78.             WDBG("accept socket error: %s(errno: %d)",strerror(errno),errno);
    0 y3 ^: q/ Y) r; Z' _5 R
79.             continue;
    8 L0 `: K* k4 s$ ~/ h" _. G
80.         }
    6 K& K8 D8 ]* b8 R) |6 W" F0 ?' f# S1 _
81. 
    # w/ i- x; L; H; i: j
82.         client_info_s* client = sys_malloc(sizeof(client_info_s));
    
83. 
    ! c2 d' f6 H: y$ |$ g$ L$ v, a
84.         if(client) {
    ! G6 g1 T& @$ e0 z0 k
85.             client->client_info = *(struct sockaddr_in*)&client_addr;
    
86.             client->id = connfd;
    
87.             client->port = parm.port;
    
88.             osThreadNew(client_proc,(void*)client, NULL);
    
89.         }
    1 ^$ o/ z) h* W
90. 
    
91.     }
    0 ~% }1 d3 ]4 {5 t
92. }
    
93. 
    
94. void create_tcp_server(u16 port, u16 cnt){
    
95.     server_parm_s parm;
    1 h/ L: f, j& F9 V
96.     parm.port = port;
    1 Q/ n, x5 P2 C) Q7 H0 S% A: _8 N
97.     parm.cnt = cnt;
    
98.     u32* ptr =(u32*)&parm;
    
99.    osThreadNew(create_tcp_task,(void*)(ptr[0]), NULL);
    ; _% ?( F2 g& j6 R2 ?: I; S5 m
100. }

复制代码

非阻塞方式：
% R4 G0 w6 l5 c1 L
- n( y0 ?7 h$ U5 L% h, o

1. void trace_task(void *argument) {
   7 O$ i1 K0 l7 |4 |( I) H1 k
2.     const TickType_t xBlockTime = pdMS_TO_TICKS(50);
   
3.     u8* read_buf = malloc(128);
   ' y8 h5 O3 D5 d; u4 v( B- g; B
4.     WDBG("TCP SEVER port:%d", 1234);
   
5.     int listenfd = -1;
   7 n/ l* m4 h2 A2 S* b$ U
6.     int connfd = -1;
   
7.     struct sockaddr_in servaddr;
   9 G; @9 y; Z% ?! d4 M6 A9 [
8.     struct sockaddr_in client_addr;
   4 k7 ^" {" |8 [" k) ~; \, s
9. 
   
10.     if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 )
    
11.     {
    ! j1 R5 E# l+ w6 e1 h" J
12.         WDBG("create socket error: %s(errno: %d)",strerror(errno),errno);
    
13.         while(1);
    ( a  r: j: X6 A1 u7 B2 N5 m5 A8 f5 L
14.     }
    + ?1 n- T6 \. I9 o3 ]
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);
    & O5 z+ A1 G  \; ^3 w
19. 
    
20.     if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1)
    
21.     {
    
22.         WDBG("bind socket error: %s(errno: %d)",strerror(errno),errno);
    3 p% V" L7 d+ \9 U
23.         while(1);
    " ?& m% K4 F7 R2 v( \1 Y
24.     }
    9 x: N/ D; Q" R1 C/ E& T3 v
25. 
    
26.     if( listen(listenfd, 3) == -1)
    
27.     {
    . G/ S' J4 `7 _; F4 A, N2 F
28.         WDBG("listen socket error: %s(errno: %d)",strerror(errno),errno);
    * F2 @* |  L+ I" e" L) o
29.     }
    
30. 
    5 p  w# N9 N( Q' O# c/ Z
31.     {
    + o: }( ~+ `) @9 j$ U
32.         DWORD nMode = 1;
    
33.         ioctlsocket( listenfd, FIONBIO, &nMode );        
    9 x- w# D( B5 e4 H
34.     }
    : d  o# g3 X, B& h! @$ B9 r
35. 
    
36.     fd_set fdRead;
    ) _) p. w3 ^9 A9 G7 v. W
37.     fd_set fd_err;
    
38.     struct timeval tv;
    
39.     socklen_t socklen = sizeof(client_addr);
    . o. `; w7 R5 W
40.     memset(&client_addr, 0, sizeof(client_addr));
    
41. 
    ) c% R- O2 J7 c7 h; E$ ]
42.     while(1) {
    
43.         FD_ZERO(&fdRead);
    
44.         FD_ZERO(&fd_err);  
    
45.         FD_SET( listenfd, &fdRead);
    
46.         u8 cnt = 2;
    
47.         if(connfd != -1) {
    
48.             cnt++;
    6 @+ P) ^+ L/ @3 P" H
49.             FD_SET(connfd, &fdRead);
    
50.             FD_SET(connfd, &fd_err);
    
51.         }
    
52. 
    
53.         if(connfd != -1) {
    
54.             u32 rec_size = xStreamBufferReceive( g_trace_handle,
    ; _+ }, I& q% z# h$ Q/ ^. N: B
55.                     ( void * ) read_buf, 128, xBlockTime);
    6 y. Y6 d; j2 ^7 [4 r# r
56.             if(rec_size) {
    2 t9 ], y4 G0 K! X/ D
57.                 s32 ret = send(connfd, read_buf, rec_size, 0);
    4 |6 v7 i9 \& G0 K  }9 B
58. 
    & T3 y/ e6 y9 o( l5 p2 L# M
59.                 if(ret <  0)  {
    
60.                     closesocket(connfd);
    6 ~1 t* }: S+ w) a; a. k  l
61.                     connfd = -1;
      Z6 c7 V" F$ |' n5 H5 B
62.                     continue;
    
63.                 }
    
64.             }
    % x' {) p3 [; n5 Q
65. 
    6 Q0 B" w; S9 ^$ Q1 }! \* d
66.         }
    
67. 
    
68.         tv.tv_sec  = 0;
    * h, }: P+ g$ @- J9 Y
69.         tv.tv_usec = 20000;
    6 }1 Y. A% S. l, p
70.         s32 ret = select(cnt, &fdRead, NULL, &fd_err, &tv);
    
71. 
    1 B. j' D# }* x9 J( u9 l
72.         if(ret == 0) {
    
73. 
    
74.             continue;
    
75.         }
    ; o! u- h% E3 l
76. 
    
77.         if(ret > 0) {
    
78.             if(connfd != -1 && FD_ISSET(connfd, &fd_err)){//rec.
    ' Q9 N) {0 p& m3 d) ?
79.                 connfd = -1;
    
80.                 continue;
    ( _9 V2 A# S/ S& I
81.             }
    
82. 
    
83.             if(connfd != -1 && FD_ISSET(connfd, &fdRead)){//rec.
    # U2 w; A5 F/ k  ?, q
84.                 ret = recv(connfd, read_buf, 128, 0);
    
85.                 if(ret < 0) {
    7 @8 q9 o, |' h1 e8 \) l
86.                     closesocket(connfd);
    
87.                     connfd = -1;
    
88.                     continue;
    
89.                 } else if(ret >0) {
    
90.                     WDBG("rec size:%d", ret);
    & i% L. f  a" E/ `4 l) Z2 J/ ]. _
91.                 }
    % O& N) R4 a" I' \
92.             }   
    
93. 
    & }1 L4 h  Y) w" z5 P
94.             if(FD_ISSET(listenfd, &fdRead)) {
    
95.                 s32 temp = accept(listenfd, (struct sockaddr*)&client_addr, &socklen);
    7 T+ ^6 ?3 a$ q; S$ T5 ]4 x4 x
96.                 // WDBG("-client:[%d] %s:%dn", temp, inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
    " G$ s6 _1 m1 V; O  R4 E
97. 
      q% F1 e+ }* h" z8 {: z% a) s
98.                 if(connfd == -1) {
    $ p% A+ G) d; A* X
99.                     connfd = temp;
    
100.                 } else {
     ; x% Z/ u5 ?2 R* N
101.                     closesocket(temp);
     4 U+ B# D' K# k' f6 a' R, Z$ V: P. t4 {
102.                 }
     5 J& u3 i" R, _" p2 l. A; I) X! B
103.             }        
     
104.         }
     
105.     }
     
106.     #endif
     
107. }
     
108. 
     + x( A( B9 \. O/ U$ k$ D

复制代码

; j; A( t  _! u7 l1 f/ I