libs/rt-thread
libs
/
rt-thread
4
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fixed bug: close sock when connect fail. 

git-svn-id: https://rt-thread.googlecode.com/svn/trunk@2287 bbd45198-f89e-11dd-88c7-29a3b14d5316
This commit is contained in:
wuyangyong 2012-09-10 14:01:30 +00:00
parent 7ce935b601
commit 8d4d8a8338
1 changed files with 68 additions and 67 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

135
examples/network/tcpclient.c
View File

@ -7,87 +7,88 @@
static const char send_data[] = "This is TCP Client from RT-Thread."; /* 发送用到的数据 */ static const char send_data[] = "This is TCP Client from RT-Thread."; /* 发送用到的数据 */
void tcpclient(const char* url, int port) void tcpclient(const char* url, int port)
{ {
char *recv_data; char *recv_data;
struct hostent *host; struct hostent *host;
int sock, bytes_received; int sock, bytes_received;
struct sockaddr_in server_addr; struct sockaddr_in server_addr;
/* 通过函数入口参数url获得host地址（如果是域名，会做域名解析） */ /* 通过函数入口参数url获得host地址（如果是域名，会做域名解析） */
host = gethostbyname(url); host = gethostbyname(url);
/* 分配用于存放接收数据的缓冲 */ /* 分配用于存放接收数据的缓冲 */
recv_data = rt_malloc(BUFSZ); recv_data = rt_malloc(BUFSZ);
if (recv_data == RT_NULL) if (recv_data == RT_NULL)
{ {
rt_kprintf("No memory\n"); rt_kprintf("No memory\n");
return; return;
} }
/* 创建一个socket，类型是SOCKET_STREAM，TCP类型 */ /* 创建一个socket，类型是SOCKET_STREAM，TCP类型 */
if ((sock = socket(AF_INET, SOCK_STREAM, 0)) == -1) if ((sock = socket(AF_INET, SOCK_STREAM, 0)) == -1)
{ {
/* 创建socket失败 */ /* 创建socket失败 */
rt_kprintf("Socket error\n"); rt_kprintf("Socket error\n");
/* 释放接收缓冲 */ /* 释放接收缓冲 */
rt_free(recv_data); rt_free(recv_data);
return; return;
} }
/* 初始化预连接的服务端地址 */ /* 初始化预连接的服务端地址 */
server_addr.sin_family = AF_INET; server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(port); server_addr.sin_port = htons(port);
server_addr.sin_addr = *((struct in_addr *)host->h_addr); server_addr.sin_addr = *((struct in_addr *)host->h_addr);
rt_memset(&(server_addr.sin_zero), 0, sizeof(server_addr.sin_zero)); rt_memset(&(server_addr.sin_zero), 0, sizeof(server_addr.sin_zero));
/* 连接到服务端 */ /* 连接到服务端 */
if (connect(sock, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) == -1) if (connect(sock, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) == -1)
{ {
/* 连接失败 */ /* 连接失败 */
rt_kprintf("Connect error\n"); rt_kprintf("Connect fail!\n");
lwip_close(sock);
/*释放接收缓冲 */ /*释放接收缓冲 */
rt_free(recv_data); rt_free(recv_data);
return; return;
} }
while(1) while(1)
{ {
/* 从sock连接中接收最大BUFSZ - 1字节数据 */ /* 从sock连接中接收最大BUFSZ - 1字节数据 */
bytes_received = recv(sock, recv_data, BUFSZ - 1, 0); bytes_received = recv(sock, recv_data, BUFSZ - 1, 0);
if (bytes_received <= 0) if (bytes_received <= 0)
{ {
/* 接收失败，关闭这个连接 */ /* 接收失败，关闭这个连接 */
lwip_close(sock); lwip_close(sock);
/* 释放接收缓冲 */ /* 释放接收缓冲 */
rt_free(recv_data); rt_free(recv_data);
break; break;
} }
/* 有接收到数据，把末端清零 */ /* 有接收到数据，把末端清零 */
recv_data[bytes_received] = '\0'; recv_data[bytes_received] = '\0';
if (strcmp(recv_data , "q") == 0 || strcmp(recv_data , "Q") == 0) if (strcmp(recv_data , "q") == 0 || strcmp(recv_data , "Q") == 0)
{ {
/* 如果是首字母是q或Q，关闭这个连接 */ /* 如果是首字母是q或Q，关闭这个连接 */
lwip_close(sock); lwip_close(sock);
/* 释放接收缓冲 */ /* 释放接收缓冲 */
rt_free(recv_data); rt_free(recv_data);
break; break;
} }
else else
{ {
/* 在控制终端显示收到的数据 */ /* 在控制终端显示收到的数据 */
rt_kprintf("\nRecieved data = %s " , recv_data); rt_kprintf("\nRecieved data = %s " , recv_data);
} }
/* 发送数据到sock连接 */ /* 发送数据到sock连接 */
send(sock,send_data,strlen(send_data), 0); send(sock,send_data,strlen(send_data), 0);
} }
return; return;
} }
#ifdef RT_USING_FINSH #ifdef RT_USING_FINSH