enforce timeout on initial socks5 proxy connection

bitcoindevkit · Dec 2, 2023 · a009e55 · a009e55
1 parent 5ecb26f
commit a009e55
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/socks/v5.rs b/src/socks/v5.rs
@@ -204,6 +204,9 @@ impl Socks5Stream {
             TcpStream::connect(proxy)?
         };
 
+        socket.set_read_timeout(timeout)?;
+        socket.set_write_timeout(timeout)?;
+
         let target = target.to_target_addr()?;
 
         let packet_len = if auth.is_no_auth() { 3 } else { 4 };