net: Add socks5+tls support

darkrenaissance · Feb 25, 2025 · 01ba243 · 01ba243
1 parent 315ef49
commit 01ba243
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/net/transport/mod.rs b/src/net/transport/mod.rs
@@ -78,6 +78,9 @@ pub enum DialerVariant {
 
     /// SOCKS5 proxy
     Socks5(socks5::Socks5Dialer),
+
+    /// SOCKS5 proxy with TLS
+    Socks5Tls(socks5::Socks5Dialer),
 }
 
 /// Listener variants
@@ -200,6 +203,14 @@ impl Dialer {
                 Ok(Self { endpoint, variant })
             }
 
+            "socks5+tls" => {
+                // Build a SOCKS5 dialer with TLS encapsulation
+                enforce_hostport!(endpoint);
+                let variant = socks5::Socks5Dialer::new(&endpoint).await?;
+                let variant = DialerVariant::Socks5Tls(variant);
+                Ok(Self { endpoint, variant })
+            }
+
             x => {
                 error!("[P2P] Requested unsupported transport: {}", x);
                 Err(io::Error::from_raw_os_error(libc::ENETUNREACH))
@@ -271,6 +282,13 @@ impl Dialer {
                 let stream = dialer.do_dial().await?;
                 Ok(Box::new(stream))
             }
+
+            DialerVariant::Socks5Tls(dialer) => {
+                let stream = dialer.do_dial().await?;
+                let tlsupgrade = tls::TlsUpgrade::new().await;
+                let stream = tlsupgrade.upgrade_dialer_tls(stream).await?;
+                Ok(Box::new(stream))
+            }
         }
     }