rt-thread TCP客户端自动连接代码

fenglove

各位好！最近要用到以太网通信，系统是rt-thread，根据官方的TCP客户端例程，
修改了几个地方，加入了TCP断开之后的重新连接，测试了一天没有出现问题，欢迎讨论 ，

1. #include <rtthread.h>
   
2. 
   
3. //#include <lwip/netdb.h>   /* 为了解析主机名，需要包含netdb.h头文件 */
   
4. //#include <lwip/sockets.h> /* 使用BSD socket，需要包含sockets.h头文件 */
   
5. 
   
6. #include <sys/socket.h> /* 使用BSD socket，需要包含sockets.h头文件 */
   
7. #include "netdb.h"
   
8. /* 服务器地址或者域名 */
   
9. char  tcp_url[32] =  "192.168.1.17";
   
10. 
    
11. /* 端口号 */
    
12. int tcp_c_port = 12345;
    
13. 
    
14. static const char send_data[] = "This is TCP Client from rt-thread "; /* 发送用到的数据 */
    
15. 
    
16. #define BUFSZ 512
    
17. 
    
18. void tcpclient(void *p)
    
19. {
    
20.     int ret;
    
21.     char *recv_data;
    
22.     struct hostent *host;
    
23.     int sock, bytes_received;
    
24.     struct sockaddr_in server_addr;
    
25.        
    
26.         /*地址和端口号 */
    
27.     char* url =  tcp_url;
    
28.     int port = tcp_c_port;
    
29. 
    
30.     /* 需要等待PHY芯片初始化完成 */
    
31.         rt_thread_delay(100);
    
32.        
    
33.     /* 通过函数入口参数url获得host地址（如果是域名，会做域名解析） */
    
34.     host = gethostbyname(url);
    
35. 
    
36.     /* 分配用于存放接收数据的缓冲 */
    
37.     recv_data = rt_malloc(BUFSZ);
    
38.     if (recv_data == RT_NULL)
    
39.     {
    
40.         rt_kprintf("No memory\n");
    
41.         return;
    
42.     }
    
43.        
    
44.     /* 创建一个socket，类型是SOCKET_STREAM，TCP类型 */
    
45.     if ((sock = socket(AF_INET, SOCK_STREAM, 0)) == -1)
    
46.     {
    
47.         /* 创建socket失败 */
    
48.         rt_kprintf("Socket error\n");
    
49.        
    
50.         /* 释放接收缓冲 */
    
51.         rt_free(recv_data);
    
52.         return;
    
53.     }
    
54. 
    
55.     /* 初始化预连接的服务端地址 */
    
56.     server_addr.sin_family = AF_INET;
    
57.     server_addr.sin_port = htons(port);
    
58.     server_addr.sin_addr = *((struct in_addr *)host->h_addr);
    
59.     rt_memset(&(server_addr.sin_zero), 0, sizeof(server_addr.sin_zero));
    
60. 
    
61.     /* 连接到服务端 最多尝试连接 5次 */
    
62.     if (connect(sock, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) == -1)
    
63.     {
    
64.                
    
65.     }
    
66. 
    
67.     while(1)
    
68.     {
    
69.         /* 从sock连接中接收最大BUFSZ - 1字节数据 */
    
70.         bytes_received = recv(sock, recv_data, BUFSZ - 1, 0);
    
71.         if (bytes_received < 0)
    
72.         {
    
73.             /* TCP重连 */
    
74.             goto again;
    
75.         }
    
76.         else if (bytes_received == 0)
    
77.         {
    
78.                         /* 这种情况就是链路断开了？？ */
    
79.             /* 打印recv函数返回值为0的警告信息 */
    
80.             rt_kprintf("\nReceived warning,recv function return 0.\r\n");
    
81.                        
    
82.             /* TCP重连 */
    
83.             goto again;   
    
84.   
    
85.         }
    
86. 
    
87.         /* 有接收到数据，把末端清零 */
    
88.         recv_data[bytes_received] = '\0';
    
89. 
    
90.         rt_kprintf("\nReceived data\n = %s " , recv_data);
    
91.             
    
92.         /* 发送数据到sock连接 */
    
93.         ret = send(sock,send_data_buffer, 138, 0);
    
94.         if (ret > 0)
    
95.         {
    
96.             /* 数据发送成功 直接退出即可 */
    
97.         }
    
98.         else if (ret < 0)
    
99.         {
    
100.             /* 发送失败，重新连接 */
     
101.             rt_kprintf("\n send error,close the socket.\r\n");
     
102.                        
     
103.             /* 重连*/
     
104.             goto again;   
     
105.         }
     
106.         else if (ret == 0)
     
107.         {
     
108.             /* 打印send函数返回值为0的警告信息 */
     
109.             rt_kprintf("\n Send warning,send function return 0.\r\n");
     
110.                        
     
111.                         /* 重连*/
     
112.                         goto again;   
     
113.         }
     
114.                
     
115.                 /* 客户端和服务器连接正常情况下执行到此处即可 */
     
116.         continue;
     
117. 
     
118.         /* 客户端重连部分 */
     
119.         again:
     
120.         {
     
121.             /* 链接失败 关闭SOCKET */
     
122.             closesocket(sock);
     
123.             
     
124.              rt_thread_delay(1);
     
125.             
     
126.             /* 创建一个socket，类型是SOCKET_STREAM，TCP类型 */
     
127.             if ((sock = socket(AF_INET, SOCK_STREAM, 0)) == -1)
     
128.             {
     
129.                 /* 创建socket失败 */
     
130.                 rt_kprintf("Socket error\n");
     
131.                 return;
     
132.             }
     
133.    
     
134.             if (connect(sock, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) == -1)
     
135.             {
     
136.                         /* 连接失败 */
     
137.                         rt_kprintf("Connect fail!\n");
     
138. 
     
139.                         /* 稍微延时一下 */
     
140.                        
     
141.                 rt_thread_delay(100);
     
142.             
     
143.                 /* TCP重连 */
     
144.                 goto again;   
     
145.             }
     
146.         }
     
147. 
     
148.         #if 0
     
149.         erro:
     
150.         {
     
151.             /* 连接失败 */
     
152.             rt_kprintf("Connect fail!\n");
     
153. 
     
154.             closesocket(sock);
     
155. 
     
156.             /*释放接收缓冲 */
     
157.             rt_free(recv_data);
     
158.             return;
     
159.         }
     
160.         #endif
     
161.     }
     
162. }
     
163. 
     
164. /* 线程栈大小 */
     
165. #define THREAD_STACK_SIZE 512
     
166. 
     
167. /* 优先级 */
     
168. #define THREAD_PRIORITY 22
     
169. 
     
170. /* 线程时间片 */
     
171. #define  THREAD_TIMESLICE 10
     
172. 
     
173. int tcp_client_thread_init(void)
     
174. {
     
175.     rt_thread_t tid;
     
176. 
     
177.     tid = rt_thread_create("tcp_c",
     
178.         tcpclient, RT_NULL,
     
179.         THREAD_STACK_SIZE, THREAD_PRIORITY, THREAD_TIMESLICE);
     
180.     if (tid != RT_NULL)
     
181.         rt_thread_startup(tid);
     
182.    
     
183.     return 0;
     
184. }
     
185. 
     
186. #ifdef RT_USING_FINSH
     
187. 
     
188. #include <finsh.h>
     
189. /* 输出tcpclient函数到finsh shell中 */
     
190. FINSH_FUNCTION_EXPORT(tcp_client_thread_init, startup tcp client);
     
191. 
     
192. #endif

复制代码