[examples] code cleanup.

1 files changed, 25 insertions, 17 deletions

diff --git a/examples/network/tcpclient.c b/examples/network/tcpclient.c
index 529c612b9a..c7a53c609f 100644
--- a/examples/network/tcpclient.c
+++ b/examples/network/tcpclient.c
@@ -1,16 +1,17 @@
 #include <rtthread.h>
-#include <lwip/netdb.h> /* 为了解析主机名，需要包含netdb.h头文件 */
+
+#include <lwip/netdb.h>   /* 为了解析主机名，需要包含netdb.h头文件 */
 #include <lwip/sockets.h> /* 使用BSD socket，需要包含sockets.h头文件 */
 
-#define BUFSZ	1024
+#define BUFSZ   1024
 
 static const char send_data[] = "This is TCP Client from RT-Thread."; /* 发送用到的数据 */
 void tcpclient(const char* url, int port)
 {
+    int ret;
     char *recv_data;
     struct hostent *host;
     int sock, bytes_received;
-	int ret;
     struct sockaddr_in server_addr;
 
     /* 通过函数入口参数url获得host地址（如果是域名，会做域名解析） */
@@ -61,15 +62,18 @@ void tcpclient(const char* url, int port)
         {
             /* 接收失败，关闭这个连接 */
             lwip_close(sock);
-            rt_kprintf("\nreceived error,close the socket.\r\n");	
+            rt_kprintf("\nreceived error,close the socket.\r\n");
+
             /* 释放接收缓冲 */
             rt_free(recv_data);
             break;
-        }else if (bytes_received == 0)
-		{
-			/* 打印recv函数返回值为0的警告信息 */
-            rt_kprintf("\nReceived warning,recv function return 0.\r\n");						
-		}
+        }
+        else if (bytes_received == 0)
+        {
+            /* 打印recv函数返回值为0的警告信息 */
+            rt_kprintf("\nReceived warning,recv function return 0.\r\n");
+            continue;
+        }
 
         /* 有接收到数据，把末端清零 */
         recv_data[bytes_received] = '\0';
@@ -78,7 +82,8 @@ void tcpclient(const char* url, int port)
         {
             /* 如果是首字母是q或Q，关闭这个连接 */
             lwip_close(sock);
-            rt_kprintf("\n got a 'q' or 'Q',close the socket.\r\n");	
+            rt_kprintf("\n got a 'q' or 'Q',close the socket.\r\n");
+
             /* 释放接收缓冲 */
             rt_free(recv_data);
             break;
@@ -91,17 +96,20 @@ void tcpclient(const char* url, int port)
 
         /* 发送数据到sock连接 */
         ret = send(sock,send_data,strlen(send_data), 0);
-		if (ret < 0)
+        if (ret < 0)
         {
             /* 接收失败，关闭这个连接 */
             lwip_close(sock);
-            rt_kprintf("\nsend error,close the socket.\r\n");		
+            rt_kprintf("\nsend error,close the socket.\r\n");
+
+            rt_free(recv_data);
             break;
-        }else if (ret == 0)
-		{
-			/* 打印send函数返回值为0的警告信息 */
-            rt_kprintf("\n Send warning,send function return 0.\r\n");						
-		}		
+        }
+        else if (ret == 0)
+        {
+            /* 打印send函数返回值为0的警告信息 */
+            rt_kprintf("\n Send warning,send function return 0.\r\n");
+        }
     }
 
     return;