Conrad Irwin
ca4cc4764b
We're seeing panics caused by a buggy implementation of AsyncWrite that is being passed to rustls: https://github.com/rustls/rustls/issues/2316#issuecomment-2662838186 One hypothesis was that we're using (comparatively) non-standard async tools for connecting over websockets; so this attempts to make us be (comparitvely) more standard. Release Notes: - N/A
69 lines
2.0 KiB
Rust
69 lines
2.0 KiB
Rust
//! socks proxy
|
|
use anyhow::{Result, anyhow};
|
|
use http_client::Uri;
|
|
use tokio_socks::tcp::{Socks4Stream, Socks5Stream};
|
|
|
|
pub(crate) async fn connect_socks_proxy_stream(
|
|
proxy: Option<&Uri>,
|
|
rpc_host: (&str, u16),
|
|
) -> Result<Box<dyn AsyncReadWrite>> {
|
|
let stream = match parse_socks_proxy(proxy) {
|
|
Some((socks_proxy, SocksVersion::V4)) => {
|
|
let stream = Socks4Stream::connect_with_socket(
|
|
tokio::net::TcpStream::connect(socks_proxy).await?,
|
|
rpc_host,
|
|
)
|
|
.await
|
|
.map_err(|err| anyhow!("error connecting to socks {}", err))?;
|
|
Box::new(stream) as Box<dyn AsyncReadWrite>
|
|
}
|
|
Some((socks_proxy, SocksVersion::V5)) => Box::new(
|
|
Socks5Stream::connect_with_socket(
|
|
tokio::net::TcpStream::connect(socks_proxy).await?,
|
|
rpc_host,
|
|
)
|
|
.await
|
|
.map_err(|err| anyhow!("error connecting to socks {}", err))?,
|
|
) as Box<dyn AsyncReadWrite>,
|
|
None => {
|
|
Box::new(tokio::net::TcpStream::connect(rpc_host).await?) as Box<dyn AsyncReadWrite>
|
|
}
|
|
};
|
|
Ok(stream)
|
|
}
|
|
|
|
fn parse_socks_proxy(proxy: Option<&Uri>) -> Option<((String, u16), SocksVersion)> {
|
|
let proxy_uri = proxy?;
|
|
let scheme = proxy_uri.scheme_str()?;
|
|
let socks_version = if scheme.starts_with("socks4") {
|
|
// socks4
|
|
SocksVersion::V4
|
|
} else if scheme.starts_with("socks") {
|
|
// socks, socks5
|
|
SocksVersion::V5
|
|
} else {
|
|
return None;
|
|
};
|
|
if let (Some(host), Some(port)) = (proxy_uri.host(), proxy_uri.port_u16()) {
|
|
Some(((host.to_string(), port), socks_version))
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
|
|
// private helper structs and traits
|
|
|
|
enum SocksVersion {
|
|
V4,
|
|
V5,
|
|
}
|
|
|
|
pub(crate) trait AsyncReadWrite:
|
|
tokio::io::AsyncRead + tokio::io::AsyncWrite + Unpin + Send + 'static
|
|
{
|
|
}
|
|
impl<T: tokio::io::AsyncRead + tokio::io::AsyncWrite + Unpin + Send + 'static> AsyncReadWrite
|
|
for T
|
|
{
|
|
}
|