socks_hub_core/
lib.rs

1cfg_if::cfg_if! {
2    if #[cfg(feature = "acl")] {
3        mod acl;
4        pub use acl::AccessControl;
5    }
6}
7
8//
9// can't use cfg_if here, because it will cause cbindgen to can't generate ffi.h correctly.
10// see this issue: https://github.com/mozilla/cbindgen/issues/935
11//
12// cfg_if::cfg_if! {
13//     if #[cfg(feature = "sockshub")] {
14
15#[cfg(feature = "sockshub")]
16mod config;
17#[cfg(feature = "sockshub")]
18pub use config::{ArgVerbosity, Config};
19
20#[cfg(feature = "sockshub")]
21mod tokiort;
22#[cfg(feature = "sockshub")]
23pub use socks5_impl::protocol::{Address, ProxyParameters, ProxyType, UserKey};
24#[cfg(feature = "sockshub")]
25use tokiort::TokioIo;
26
27#[cfg(feature = "sockshub")]
28mod http2socks;
29#[cfg(feature = "sockshub")]
30mod socks2socks;
31
32#[cfg(feature = "sockshub")]
33mod api;
34#[cfg(feature = "sockshub")]
35mod dump_logger;
36#[cfg(feature = "sockshub")]
37mod ffi;
38
39#[cfg(feature = "sockshub")]
40pub type BoxError = Box<dyn std::error::Error + Send + Sync + 'static>;
41#[cfg(feature = "sockshub")]
42pub type Result<T, E = BoxError> = std::result::Result<T, E>;
43
44#[cfg(feature = "sockshub")]
45pub async fn main_entry<F>(config: &Config, cancel_token: tokio_util::sync::CancellationToken, callback: Option<F>) -> Result<(), BoxError>
46where
47    F: FnOnce(std::net::SocketAddr) + Send + Sync + 'static,
48{
49    if config.remote_server.proxy_type != ProxyType::Socks5 {
50        return Err("remote server must be socks5".into());
51    }
52    if let Some(middle_server) = &config.middle_server {
53        if middle_server.proxy_type != ProxyType::Socks5 {
54            return Err("middle server must be socks5".into());
55        }
56    }
57    match config.listen_proxy_role.proxy_type {
58        ProxyType::Http => http2socks::main_entry(config, cancel_token, callback).await,
59        ProxyType::Socks5 => socks2socks::main_entry(config, cancel_token, callback).await,
60        _ => Err("listen proxy must be http or socks5".into()),
61    }
62}
63
64#[cfg(feature = "sockshub")]
65pub(crate) const CONNECT_TIMEOUT: std::time::Duration = std::time::Duration::from_secs(5);
66
67#[cfg(feature = "sockshub")]
68pub(crate) async fn create_s5_connect(
69    server: ProxyParameters,
70    dur: std::time::Duration,
71    dst: &socks5_impl::protocol::Address,
72    middle_server: Option<ProxyParameters>,
73) -> std::io::Result<tokio::io::BufStream<tokio::net::TcpStream>> {
74    let auth = server.credentials.clone();
75    let mut stream = connect_proxy_stream(server, dur, middle_server).await?;
76    socks5_impl::client::connect(&mut stream, dst, auth)
77        .await
78        .map_err(std_io_error_other)?;
79    Ok(stream)
80}
81
82#[cfg(feature = "sockshub")]
83async fn connect_proxy_stream(
84    server: ProxyParameters,
85    dur: std::time::Duration,
86    middle_server: Option<ProxyParameters>,
87) -> std::io::Result<tokio::io::BufStream<tokio::net::TcpStream>> {
88    let stream = if let Some(middle_server) = middle_server {
89        let middle_addr: std::net::SocketAddr = middle_server.addr.try_into()?;
90        let stream = tokio::time::timeout(dur, tokio::net::TcpStream::connect(middle_addr)).await??;
91        let mut stream = tokio::io::BufStream::new(stream);
92        let middle_auth = middle_server.credentials.clone();
93        socks5_impl::client::connect(&mut stream, server.addr, middle_auth)
94            .await
95            .map_err(std_io_error_other)?;
96        stream
97    } else {
98        let server_addr: std::net::SocketAddr = server.addr.try_into()?;
99        let stream = tokio::time::timeout(dur, tokio::net::TcpStream::connect(server_addr)).await??;
100        tokio::io::BufStream::new(stream)
101    };
102    Ok(stream)
103}
104
105#[cfg(feature = "sockshub")]
106pub(crate) async fn create_s5_udp_client(
107    server: ProxyParameters,
108    dur: std::time::Duration,
109    middle_server: Option<ProxyParameters>,
110) -> std::io::Result<socks5_impl::client::SocksUdpClient> {
111    let stream = connect_proxy_stream(server.clone(), dur, middle_server).await?;
112    let client_addr = if server.addr.is_ipv4() { "0.0.0.0:0" } else { "[::]:0" };
113    let client = tokio::net::UdpSocket::bind(client_addr).await?;
114    let auth = server.credentials.clone();
115    socks5_impl::client::SocksDatagram::udp_associate(stream, client, auth)
116        .await
117        .map_err(std_io_error_other)
118}
119
120#[cfg(feature = "sockshub")]
121pub(crate) fn std_io_error_other<E: Into<BoxError>>(err: E) -> std::io::Error {
122    std::io::Error::other(err)
123}
124
125//     }
126// }
socks_hub_core/lib.rs

socks_hub_core/
lib.rs
