feat: timeout on endpoint check

Author: alx0x10

src/main.rs

@@ -7,7 +7,7 @@ use std::{
     fmt::{Display, Formatter},
     io,
     net::{SocketAddr, SocketAddrV4, SocketAddrV6, ToSocketAddrs},
-    str::FromStr,
+    str::FromStr, time::Duration,
 };
 use tokio::{
     io::{AsyncRead, AsyncWrite, AsyncWriteExt},
@@ -354,7 +354,12 @@ async fn create_endpoint(
         builder = builder.bind_addr_v6(addr);
     }
     let endpoint = builder.bind().await?;
-    endpoint.online().await;
+
+    let _ = tokio::time::timeout(Duration::from_secs(30), async {
+        if !(common.relay == RelayModeOption::Disabled) {
+            endpoint.online().await;
+        }
+    }).await;
 
     Ok(endpoint)
 }

tests/cli.rs

@@ -19,7 +19,7 @@ fn dumbpipe_bin() -> &'static str {
 ///
 /// This assumes that the header lines are ASCII and can be parsed byte by byte.
 fn read_ascii_lines(mut n: usize, reader: &mut impl Read) -> io::Result<Vec<u8>> {
-    let mut buf = [0u8; 1];
+let mut buf = [0u8; 1];
     let mut res = Vec::new();
     loop {
         if reader.read(&mut buf)? != 1 {