66 lines · 2.1 KB · Rust
RawDownload
//! client proxy mod http_proxy;mod socks_proxy; use anyhow::{Context as _, Result};use http_client::Url;use http_proxy::{HttpProxyType, connect_http_proxy_stream, parse_http_proxy};use socks_proxy::{SocksVersion, connect_socks_proxy_stream, parse_socks_proxy}; pub(crate) async fn connect_proxy_stream(    proxy: &Url,    rpc_host: (&str, u16),) -> Result<Box<dyn AsyncReadWrite>> {    let Some(((proxy_domain, proxy_port), proxy_type)) = parse_proxy_type(proxy) else {        // If parsing the proxy URL fails, we must avoid falling back to an insecure connection.        // SOCKS proxies are often used in contexts where security and privacy are critical,        // so any fallback could expose users to significant risks.        anyhow::bail!("Parsing proxy url failed");    };     // Connect to proxy and wrap protocol later    let stream = tokio::net::TcpStream::connect((proxy_domain.as_str(), proxy_port))        .await        .context("Failed to connect to proxy")?;     let proxy_stream = match proxy_type {        ProxyType::SocksProxy(proxy) => connect_socks_proxy_stream(stream, proxy, rpc_host).await?,        ProxyType::HttpProxy(proxy) => {            connect_http_proxy_stream(stream, proxy, rpc_host, &proxy_domain).await?        }    };     Ok(proxy_stream)} enum ProxyType<'t> {    SocksProxy(SocksVersion<'t>),    HttpProxy(HttpProxyType<'t>),} fn parse_proxy_type(proxy: &Url) -> Option<((String, u16), ProxyType<'_>)> {    let scheme = proxy.scheme();    let host = proxy.host()?.to_string();    let port = proxy.port_or_known_default()?;    let proxy_type = match scheme {        scheme if scheme.starts_with("socks") => {            Some(ProxyType::SocksProxy(parse_socks_proxy(scheme, proxy)))        }        scheme if scheme.starts_with("http") => {            Some(ProxyType::HttpProxy(parse_http_proxy(scheme, proxy)))        }        _ => None,    }?;     Some(((host, port), proxy_type))} 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{}