一个简单的 TCP 客户端测试程序

murphyzhao

现在都 lwip 210 版本了，无奈，因为客户 SDK 的原因还在捣鼓 lwip151 版本。

将客户 SDK 升级到 RT-Thread 系统的时候，发现低功耗部分与 LWIP 有耦合，不能直接使用 RT-Thread 移植的 LWIP 组件，也就意味着，我不同享受 RT-Thread netdev 组件和 SAL 组件开发的便利。

于是，给自己挖了个坑，把 LWIP151 版本对接到 netdev 和 sal 组件上

为了测试，捣鼓了一个简单的 TCP 客户端程序，后面有时间也把对接 RT-Thread netdev 组件和 SAL 组件的过程也给分享出来。

这里提供的代码基于 BSD-type 风格的 socket 函数接口，你可以在任何支持 BSD socket 的平台上使用这段程序，当然这段程序也可以在包含了 RT-Thread SAL 的使用方法，引入 SAL 相关头文件。

1. #include <stdio.h>
   
2. #include <string.h>
   
3. #include <rtthread.h>
   
4. 
   
5. /* 引入 lwip socket 头文件 */
   
6. #include "lwip/sockets.h"
   
7. #include "lwip/netdb.h"
   
8. #include "lwip/sys.h"
   
9. 
   
10. #define TCP_TEST_HOST    "your host or ip addr" /* 输入你的 TCP server 域名或者 ip 地址 */
    
11. #define TCP_TEST_PORT    (80u)                  /* 输入你的 TCP Server 断口号 */
    
12. #define TEST_BUFSZ       (1024u)
    
13. 
    
14. static const char *send_tcp_req_data = "Hi, I am from tcp client.";
    
15. 
    
16. static void _tcp_test(void)
    
17. {
    
18.     int ret, i;
    
19.     char *recv_data;
    
20.     struct hostent *host;
    
21.     int sock = -1, bytes_received;
    
22.     struct sockaddr_in server_addr;
    
23. 
    
24.     /* 通过函数入口参数url获得host地址（如果是域名，会做域名解析） */
    
25.     host = gethostbyname(TCP_TEST_HOST);
    
26.     if (!host)
    
27.     {
    
28.         rt_kprintf("gethostbyname failed!\r\n");
    
29.         return;
    
30.     }
    
31.     else
    
32.     {
    
33.         rt_kprintf("gethostbyname success!\r\n");
    
34.     }
    
35. 
    
36.     recv_data = calloc(1, TEST_BUFSZ);
    
37.     if (recv_data == RT_NULL)
    
38.     {
    
39.         rt_kprintf("No memory\n");
    
40.         return;
    
41.     }
    
42. 
    
43.     /* 创建一个socket，类型是SOCKET_STREAM，TCP 协议, TLS 类型 */
    
44.     if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0)
    
45.     {
    
46.         rt_kprintf("Socket error\n");
    
47.         goto __exit;
    
48.     }
    
49.     else
    
50.     {
    
51.         rt_kprintf("Socket pass\n");
    
52.     }
    
53. 
    
54.     /* 初始化预连接的服务端地址 */
    
55.     server_addr.sin_family = AF_INET;
    
56.     server_addr.sin_port = htons(TCP_TEST_PORT);
    
57.     server_addr.sin_addr = *((struct in_addr *)host->h_addr);
    
58.     memset(&(server_addr.sin_zero), 0, sizeof(server_addr.sin_zero));
    
59. 
    
60.     rt_kprintf("will connect...\n");
    
61.     if (connect(sock, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) < 0)
    
62.     {
    
63.         rt_kprintf("Connect fail!\n");
    
64.         goto __exit;
    
65.     }
    
66.     else
    
67.     {
    
68.         rt_kprintf("Connect pass!\n");
    
69.     }
    
70. 
    
71.     /* 发送数据到 socket 连接 */
    
72.     ret = send(sock, send_tcp_req_data, strlen(send_tcp_req_data), 0);
    
73.     if (ret <= 0)
    
74.     {
    
75.         rt_kprintf("send error,close the socket.\n");
    
76.         goto __exit;
    
77.     }
    
78.     else
    
79.     {
    
80.         rt_kprintf("send pass!\n");
    
81.     }
    
82. 
    
83.     /* 接收并打印响应的数据，使用加密数据传输 */
    
84.     bytes_received = recv(sock, recv_data, TEST_BUFSZ - 1, 0);
    
85.     if (bytes_received <= 0)
    
86.     {
    
87.         rt_kprintf("received error,close the socket.\n");
    
88.         goto __exit;
    
89.     }
    
90. 
    
91.     rt_kprintf("recv data:\n");
    
92.     for (i = 0; i < bytes_received; i++)
    
93.     {
    
94.         rt_kprintf("%c", recv_data);
    
95.     }
    
96. 
    
97. __exit:
    
98.     if (recv_data)
    
99.         rt_free(recv_data);
    
100. 
     
101.     if (sock >= 0)
     
102.         close(sock);
     
103. }
     
104. 
     
105. void tcp_test(void)
     
106. {
     
107.     rt_thread_t tid;
     
108.     tid = rt_thread_create("tcp", _tcp_test, NULL, 4096, 23, 5);
     
109.     if (tid)
     
110.         rt_thread_startup(tid);
     
111.     else
     
112.     {
     
113.         rt_kprintf("tcp thread create failed!\r\n");
     
114.     }
     
115. }
     
116. 
     
117. #ifdef FINSH_USING_MSH
     
118. #include <finsh.h>
     
119. MSH_CMD_EXPORT(tcp_test, TCP function test);
     
120. #endif /* FINSH_USING_MSH */
     

复制代码

如果你也是使用 RT-Thread，那么直接复制代码加入编译，然后在 MSH 中运行既可以：

1. msh /> tcp_test

复制代码

tcp_client.rar (1.32 KB, 下载次数: 3)

2019-6-27 09:26 上传

点击文件名下载附件

murphyzhao

之前粘贴的代码有问题，这下学会怎么插入代码了。

从文本编辑器复制代码，粘贴到论坛里的代码插入标签，提交之后会插入到当前光标位置。
这个时候你看到的代码格式是乱掉的，没有空格缩进等，不要紧，保存提交帖子即可，这个时候会发现，代码格式显示正常了。