Parcourir la source

[telnet] 优化代码

Meco Man il y a 4 ans
Parent
commit
6daa7c0997
1 fichiers modifiés avec 13 ajouts et 13 suppressions
  1. 13 13
      telnet/telnet.c

+ 13 - 13
telnet/telnet.c

@@ -22,7 +22,7 @@
  * 2012-04-01     Bernard      first version
  * 2018-01-25     armink       Fix it on RT-Thread 3.0+
  */
-#include <rtthread.h>
+
 #include <rtdevice.h>
 
 #ifdef PKG_NETUTILS_TELNET
@@ -30,14 +30,14 @@
 #include <sys/socket.h>
 #else
 #include <lwip/sockets.h>
-#endif /* SAL_USING_POSIX */
+#endif /* defined(RT_USING_DFS_NET) || defined(SAL_USING_POSIX) */
 
-#if defined(RT_USING_POSIX)
+#ifdef RT_USING_POSIX
 #include <dfs_posix.h>
 #include <dfs_poll.h>
 #include <libc.h>
 static int dev_old_flag;
-#endif
+#endif /* RT_USING_POSIX */
 
 #include <finsh.h>
 #include <msh.h>
@@ -201,7 +201,6 @@ static void process_rx(struct telnet_session* telnet, rt_uint8_t *data, rt_size_
         data++;
     }
 
-
 #if !defined(RT_USING_POSIX)
     rt_size_t rx_length;
     rt_mutex_take(telnet->rx_ringbuffer_lock, RT_WAITING_FOREVER);
@@ -214,7 +213,7 @@ static void process_rx(struct telnet_session* telnet, rt_uint8_t *data, rt_size_
     {
         telnet->device.rx_indicate(&telnet->device, rx_length);
     }
-#endif
+#endif /* !defined(RT_USING_POSIX) */
 
     return;
 }
@@ -225,7 +224,7 @@ static void client_close(struct telnet_session* telnet)
     /* set console */
     rt_console_set_device(RT_CONSOLE_DEVICE_NAME);
     /* set finsh device */
-#if defined(RT_USING_POSIX)
+#ifdef RT_USING_POSIX
     ioctl(libc_stdio_get_console(), F_SETFL, (void *) dev_old_flag);
     libc_stdio_set_console(RT_CONSOLE_DEVICE_NAME, O_RDWR);
 #else
@@ -321,12 +320,12 @@ static rt_err_t telnet_control(rt_device_t dev, int cmd, void *args)
         telnet_write,
         telnet_control
     };
-#endif
+#endif /* RT_USING_DEVICE_OPS */
+
 /* telnet server thread entry */
 static void telnet_thread(void* parameter)
 {
 #define RECV_BUF_LEN 64
-
     struct sockaddr_in addr;
     socklen_t addr_size;
     rt_uint8_t recv_buf[RECV_BUF_LEN];
@@ -372,7 +371,7 @@ static void telnet_thread(void* parameter)
     telnet->device.read     = telnet_read;
     telnet->device.write    = telnet_write;
     telnet->device.control  = telnet_control;
-#endif
+#endif /* RT_USING_DEVICE_OPS */
 
     /* no private */
     telnet->device.user_data = RT_NULL;
@@ -395,8 +394,9 @@ static void telnet_thread(void* parameter)
         /* process the new connection */
         /* set console */
         rt_console_set_device("telnet");
+
         /* set finsh device */
-#if defined(RT_USING_POSIX)
+#ifdef RT_USING_POSIX
         /* backup flag */
         dev_old_flag = ioctl(libc_stdio_get_console(), F_GETFL, (void *) RT_NULL);
         /* add non-block flag */
@@ -423,8 +423,8 @@ static void telnet_thread(void* parameter)
         finsh_set_echo(0);
         /* output RT-Thread version and shell prompt */
 #ifdef FINSH_USING_MSH
-        msh_exec("version", strlen("version"));
-#endif
+        msh_exec("version", rt_strlen("version"));
+#endif /* FINSH_USING_MSH */
         rt_kprintf(FINSH_PROMPT);
 
         while (1)