[examples][network] 更新 UDP client & server

Signed-off-by: MurphyZhao <[email protected]>

1 files changed, 140 insertions, 31 deletions

diff --git a/examples/network/udpserver.c b/examples/network/udpserver.c
index a684e9cd0c..15cc2096d6 100644
--- a/examples/network/udpserver.c
+++ b/examples/network/udpserver.c
@@ -1,10 +1,33 @@
 #include <rtthread.h>
+#include <string.h>
 
+#if !defined(SAL_USING_POSIX)
+#error "Please enable SAL_USING_POSIX!"
+#else
+#include <sys/time.h>
+#include <sys/select.h>
+#endif
 #include <sys/socket.h> /* 使用BSD socket，需要包含socket.h头文件 */
 #include "netdb.h"
 
+#define DEBUG_UDP_SERVER
+
+#define DBG_ENABLE
+#define DBG_SECTION_NAME               "UDP"
+#ifdef DEBUG_UDP_SERVER
+#define DBG_LEVEL                      DBG_LOG
+#else
+#define DBG_LEVEL                      DBG_INFO /* DBG_ERROR */
+#endif
+#define DBG_COLOR
+#include <rtdbg.h>
+
 #define BUFSZ   1024
 
+static int started = 0;
+static int is_running = 0;
+static int port = 5000;
+
 static void udpserv(void *paramemter)
 {
     int sock;
@@ -13,28 +36,27 @@ static void udpserv(void *paramemter)
     socklen_t addr_len;
     struct sockaddr_in server_addr, client_addr;
 
+    struct timeval timeout;
+    fd_set readset;
+
     /* 分配接收用的数据缓冲 */
     recv_data = rt_malloc(BUFSZ);
     if (recv_data == RT_NULL)
     {
-        /* 分配内存失败，返回 */
-        rt_kprintf("No memory\n");
+        LOG_E("No memory");
         return;
     }
 
     /* 创建一个socket，类型是SOCK_DGRAM，UDP类型 */
     if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
     {
-        rt_kprintf("Socket error\n");
-
-        /* 释放接收用的数据缓冲 */
-        rt_free(recv_data);
-        return;
+        LOG_E("Create socket error");
+        goto __exit;
     }
 
     /* 初始化服务端地址 */
     server_addr.sin_family = AF_INET;
-    server_addr.sin_port = htons(5000);
+    server_addr.sin_port = htons(port);
     server_addr.sin_addr.s_addr = INADDR_ANY;
     rt_memset(&(server_addr.sin_zero), 0, sizeof(server_addr.sin_zero));
 
@@ -42,47 +64,134 @@ static void udpserv(void *paramemter)
     if (bind(sock, (struct sockaddr *)&server_addr,
              sizeof(struct sockaddr)) == -1)
     {
-        /* 绑定地址失败 */
-        rt_kprintf("Bind error\n");
-
-        /* 释放接收用的数据缓冲 */
-        rt_free(recv_data);
-        return;
+        LOG_E("Unable to bind");
+        goto __exit;
     }
 
     addr_len = sizeof(struct sockaddr);
-    rt_kprintf("UDPServer Waiting for client on port 5000...\n");
+    LOG_I("UDPServer Waiting for client on port %d...", port);
 
-    while (1)
+    started = 1;
+    is_running = 1;
+
+    timeout.tv_sec = 3;
+    timeout.tv_usec = 0;
+
+    while (is_running)
     {
+        FD_ZERO(&readset);
+        FD_SET(sock, &readset);
+
+        /* Wait for read or write */
+        if (select(sock + 1, &readset, RT_NULL, RT_NULL, &timeout) == 0)
+            continue;
+
         /* 从sock中收取最大BUFSZ - 1字节数据 */
         bytes_read = recvfrom(sock, recv_data, BUFSZ - 1, 0,
                               (struct sockaddr *)&client_addr, &addr_len);
-        /* UDP不同于TCP，它基本不会出现收取的数据失败的情况，除非设置了超时等待 */
+        if (bytes_read < 0)
+        {
+            LOG_E("Received error, close the connect.");
+            goto __exit;
+        }
+        else if (bytes_read == 0)
+        {
+            LOG_W("Received warning, recv function return 0.");
+            continue;
+        }
+        else
+        {
+            recv_data[bytes_read] = '\0'; /* 把末端清零 */
 
-        recv_data[bytes_read] = '\0'; /* 把末端清零 */
+            /* 输出接收的数据 */
+            LOG_D("Received data = %s", recv_data);
 
-        /* 输出接收的数据 */
-        rt_kprintf("\n(%s , %d) said : ", inet_ntoa(client_addr.sin_addr),
-                   ntohs(client_addr.sin_port));
-        rt_kprintf("%s", recv_data);
+            /* 如果接收数据是exit，退出 */
+            if (strcmp(recv_data, "exit") == 0)
+            {
+                goto __exit;
+            }
+        }
+    }
 
-        /* 如果接收数据是exit，退出 */
-        if (strcmp(recv_data, "exit") == 0)
+__exit:
+    if (recv_data)
+    {
+        rt_free(recv_data);
+        recv_data = RT_NULL;
+    }
+    if (sock >= 0)
+    {
+        closesocket(sock);
+        sock = -1;
+    }
+    started = 0;
+    is_running = 0;
+}
+
+static void usage(void)
+{
+    LOG_I("Usage: udpserver -p <port>");
+    LOG_I("       udpserver --stop");
+    LOG_I("       udpserver --help");
+    LOG_I("");
+    LOG_I("Miscellaneous:");
+    LOG_I("  -p           Specify the host port number");
+    LOG_I("  --stop       Stop udpserver program");
+    LOG_I("  --help       Print help information");
+}
+
+static void udpserver_test(int argc, char** argv)
+{
+    rt_thread_t tid;
+
+    if (argc == 1 || argc > 3)
+    {
+        LOG_I("Please check the command you entered!\n");
+        goto __usage;
+    }
+    else
+    {
+        if (rt_strcmp(argv[1], "--help") == 0)
+        {
+            goto __usage;
+        }
+        else if (rt_strcmp(argv[1], "--stop") == 0)
+        {
+            is_running = 0;
+            return;
+        }
+        else if (rt_strcmp(argv[1], "-p") == 0)
         {
-            closesocket(sock);
+            if (started)
+            {
+                LOG_I("The tcpclient has started!");
+                LOG_I("Please stop tcpclient firstly, by: tcpclient --stop");
+                return;
+            }
 
-            /* 释放接收用的数据缓冲 */
-            rt_free(recv_data);
-            break;
+            port = atoi(argv[2]);
+        }
+        else
+        {
+            goto __usage;
         }
     }
 
+    tid = rt_thread_create("udp_serv",
+        udpserv, RT_NULL,
+        2048, RT_THREAD_PRIORITY_MAX/3, 20);
+    if (tid != RT_NULL)
+    {
+        rt_thread_startup(tid);
+    }
     return;
+
+__usage:
+    usage();
 }
 
 #ifdef RT_USING_FINSH
-#include <finsh.h>
-/* 输出udpserv函数到finsh shell中 */
-FINSH_FUNCTION_EXPORT(udpserv, startup udp server);
+MSH_CMD_EXPORT_ALIAS(udpserver_test, udpserver,
+    Start a udp server. Help: udpserver --help);
 #endif