STM32F7 LWIP socket 创建TCP server

1. **阻塞方式：**

复制代码

1. typedef struct {
   
2.     struct sockaddr_in client_info;
   
3.     s32 id;
   
4.     u16 port;
   
5. } client_info_s;
   
6. 
   
7. void client_proc(void* parm) {
   
8.     client_info_s* info = (client_info_s*)parm;
   
9.     u8* rec_data = sys_malloc(1500);
   
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);
    
13. 
    
14.     while(1) {
    
15.         s32 ret = read(info->id, rec_data, 1500);
    
16. 
    
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. 
    
21.             if(ret <  0) {
    
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();
    
33.         }            
    
34.     }
    
35. }
    
36. 
    
37. typedef struct {
    
38.     u16 port;
    
39.     u8 cnt;
    
40.     u8 rvd;
    
41. } server_parm_s;
    
42. 
    
43. void create_tcp_task(void* pram) {
    
44.     server_parm_s parm;
    
45.     memcpy(&parm, &pram, 4);
    
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.     {
    
53.         WDBG("create socket error: %s(errno: %d)",strerror(errno),errno);
    
54.         while(1);
    
55.     }
    
56.     memset(&servaddr, 0, sizeof(servaddr));
    
57.     servaddr.sin_family = AF_INET;
    
58.     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);            
    
59.     servaddr.sin_port = htons(parm.port);
    
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.     }
    
66. 
    
67.     if( listen(listenfd, parm.cnt) == -1)
    
68.     {
    
69.         WDBG("listen socket error: %s(errno: %d)",strerror(errno),errno);
    
70.     }
    
71. 
    
72.     while(1)
    
73.     {
    
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)
    
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) {
    
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.         }
    
90. 
    
91.     }
    
92. }
    
93. 
    
94. void create_tcp_server(u16 port, u16 cnt){
    
95.     server_parm_s parm;
    
96.     parm.port = port;
    
97.     parm.cnt = cnt;
    
98.     u32* ptr =(u32*)&parm;
    
99.    osThreadNew(create_tcp_task,(void*)(ptr[0]), NULL);
    
100. }

复制代码

非阻塞方式：

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);
   
5.     int listenfd = -1;
   
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);
    
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);
    
19. 
    
20.     if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1)
    
21.     {
    
22.         WDBG("bind socket error: %s(errno: %d)",strerror(errno),errno);
    
23.         while(1);
    
24.     }
    
25. 
    
26.     if( listen(listenfd, 3) == -1)
    
27.     {
    
28.         WDBG("listen socket error: %s(errno: %d)",strerror(errno),errno);
    
29.     }
    
30. 
    
31.     {
    
32.         DWORD nMode = 1;
    
33.         ioctlsocket( listenfd, FIONBIO, &nMode );        
    
34.     }
    
35. 
    
36.     fd_set fdRead;
    
37.     fd_set fd_err;
    
38.     struct timeval tv;
    
39.     socklen_t socklen = sizeof(client_addr);
    
40.     memset(&client_addr, 0, sizeof(client_addr));
    
41. 
    
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++;
    
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,
    
55.                     ( void * ) read_buf, 128, xBlockTime);
    
56.             if(rec_size) {
    
57.                 s32 ret = send(connfd, read_buf, rec_size, 0);
    
58. 
    
59.                 if(ret <  0)  {
    
60.                     closesocket(connfd);
    
61.                     connfd = -1;
    
62.                     continue;
    
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);
    
71. 
    
72.         if(ret == 0) {
    
73. 
    
74.             continue;
    
75.         }
    
76. 
    
77.         if(ret > 0) {
    
78.             if(connfd != -1 && FD_ISSET(connfd, &fd_err)){//rec.
    
79.                 connfd = -1;
    
80.                 continue;
    
81.             }
    
82. 
    
83.             if(connfd != -1 && FD_ISSET(connfd, &fdRead)){//rec.
    
84.                 ret = recv(connfd, read_buf, 128, 0);
    
85.                 if(ret < 0) {
    
86.                     closesocket(connfd);
    
87.                     connfd = -1;
    
88.                     continue;
    
89.                 } else if(ret >0) {
    
90.                     WDBG("rec size:%d", ret);
    
91.                 }
    
92.             }   
    
93. 
    
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. 
    
98.                 if(connfd == -1) {
    
99.                     connfd = temp;
    
100.                 } else {
     
101.                     closesocket(temp);
     
102.                 }
     
103.             }        
     
104.         }
     
105.     }
     
106.     #endif
     
107. }
     
108. 
     

复制代码