[examples]更正tcp例程中recv/send返回值判断

2 files changed, 32 insertions, 16 deletions

diff --git a/examples/network/tcpclient.c b/examples/network/tcpclient.c
index 453910ec90..09b94c5216 100644
--- a/examples/network/tcpclient.c
+++ b/examples/network/tcpclient.c
@@ -10,6 +10,7 @@
 
 #include <rtthread.h>
 #include <string.h>
+#include <stdlib.h>
 
 #if !defined(SAL_USING_POSIX)
 #error "Please enable SAL_USING_POSIX!"
@@ -114,15 +115,15 @@ static void tcpclient(void *arg)
         {
             /* Receive failed and close the connection */
             /* 接收失败，关闭这个连接 */
-            LOG_E("Received error, close the socket.");
+            LOG_E("Received error(%d), close the socket.", errno);
             goto __exit;
         }
         else if (bytes_received == 0)
         {
-            /* Print warning message when recv function returns 0 */
-            /* 打印recv函数返回值为0的警告信息 */
-            LOG_W("Received warning, recv function returns 0.");
-            continue;
+            /* Socket has performed an orderly shutdown. */
+            /* 连接已断开 */
+            LOG_E("Socket has performed an orderly shutdown.");
+            goto __exit;
         }
         else
         {
@@ -151,14 +152,19 @@ static void tcpclient(void *arg)
         {
             /* Send failed, close the connection */
             /* 发送失败，关闭这个连接 */
-            LOG_I("send error, close the socket.");
+            LOG_I("send error(%d), close the socket.", errno);
             goto __exit;
         }
         else if (ret == 0)
         {
-            /* Print warning message when send function returns 0 */
-            /* 打印send函数返回值为0的警告信息 */
-            LOG_W("Send warning, send function returns 0.");
+            /* Socket has performed an orderly shutdown. */
+            /* 连接已断开 */
+            LOG_E("Socket has performed an orderly shutdown.");
+            goto __exit;
+        }
+        else if (ret != rt_strlen(send_data))
+        {
+            LOG_W("%d out of %d bytes sent.", ret, rt_strlen(send_data));
         }
     }
 
diff --git a/examples/network/tcpserver.c b/examples/network/tcpserver.c
index 6ec5f0e7a3..4da840d627 100644
--- a/examples/network/tcpserver.c
+++ b/examples/network/tcpserver.c
@@ -10,6 +10,7 @@
 
 #include <rtthread.h>
 #include <string.h>
+#include <stdlib.h>
 
 #if !defined(SAL_USING_POSIX)
 #error "Please enable SAL_USING_POSIX!"
@@ -139,10 +140,12 @@ static void tcpserv(void *arg)
             }
             else if (bytes_received == 0)
             {
-                /* Print warning message when recv function returns 0 */
-                /* 打印recv函数返回值为0的警告信息 */
-                LOG_W("Received warning, recv function returns 0.");
-                continue;
+                /* Socket has performed an orderly shutdown */
+                /* 连接已断开 */
+                LOG_E("Socket has performed an orderly shutdown.");
+                closesocket(connected);
+                connected = -1;
+                break;
             }
             else
             {   /* Receive data successfully and append '\0' at the end of message */
@@ -184,9 +187,16 @@ static void tcpserv(void *arg)
             }
             else if (ret == 0)
             {
-                /* Print warning message when send function returns 0 */
-                /* 打印send函数返回值为0的警告信息 */
-                LOG_W("Send warning, send function returns 0.");
+                /* Socket has performed an orderly shutdown */
+                /* 连接已断开 */
+                LOG_E("Socket has performed an orderly shutdown.");
+                closesocket(connected);
+                connected = -1;
+                break;
+            }
+            else if (ret != rt_strlen(send_data))
+            {
+                LOG_W("%d out of %d bytes sent.", ret, rt_strlen(send_data));
             }
         }
     }