Update client.rs

re-fix the issue shadowsocks#3
dev4u · Mar 15, 2022 · d748fc8 · d748fc8
1 parent 3c231ab
commit d748fc8
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/src/client.rs b/src/client.rs
@@ -94,12 +94,6 @@ async fn main() -> Result<()> {
 
     while let Ok((inbound, _)) = listener.accept().await {
         info!("connection incoming");
-        let socket = if cfg!(target_os = "windows") {
-            std::net::UdpSocket::bind("0.0.0.0:0").unwrap()
-        } else {
-            std::net::UdpSocket::bind("[::]:0").unwrap()
-        };
-        endpoint.rebind(socket)?;
 
         let remote = Arc::clone(&remote);
         let host = Arc::clone(&host);
@@ -121,7 +115,19 @@ async fn transfer(
     let new_conn = endpoint
         .connect(*remote, &host)?
         .await
-        .map_err(|e| anyhow!("failed to connect: {}", e))?;
+        .map_err(|e| {
+            if e == ConnectionError::TimedOut {
+                let socket = if cfg!(target_os = "windows") {
+                    std::net::UdpSocket::bind("0.0.0.0:0").unwrap()
+                } else {
+                    std::net::UdpSocket::bind("[::]:0").unwrap()
+                };
+                endpoint.rebind(socket)?;
+                Ok(())
+            } else {
+                Err(anyhow!("failed to connect: {:?}", e))
+            }
+        }).unwrap();
 
     let quinn::NewConnection {
         connection: conn, ..