s5x 0.1.1

SOCKS5 proxy server
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

#![warn(clippy::pedantic)]

mod socks5;
mod prelude;

use prelude::*;
use tokio::net::TcpListener;
use tracing_subscriber::fmt;
use crate::socks5::{config::Config, session::Socks5Session};

#[tokio::main]
async fn main() -> Result<(), AppError> {
    setup_tracing();

    let mut config = Config::new()?;
    config.validate()?;

    let listener = TcpListener::bind(format!("{}:{}", config.host, config.port)).await
        .map_err(|_| AppError::Socks5(format!("port {} is busy or cannot be used", config.port)))?;

    info!(config = ?config, "socks5 started");

    loop {
        let config = config.clone();
        let (stream, client_addr) = listener.accept().await?;
        info!(%client_addr, "new connection");
        tokio::spawn(async move {
            let mut session = Socks5Session::new(config, stream);
            if let Err(error) = session.start().await {
                error!(%error, %client_addr);
            }
            info!(%client_addr, "connection closed");
        });
    }
}

#[cfg(debug_assertions)]
fn setup_tracing() {
    use tracing::Level;

    fmt()
        .with_target(false)
        .with_max_level(Level::TRACE)
        .init();
}

#[cfg(not(debug_assertions))]
fn setup_tracing() {
    use tracing_subscriber::EnvFilter;

    fmt()
        .with_target(false)
        .with_env_filter(
            EnvFilter::try_from_default_env()
                .unwrap_or_else(|_| EnvFilter::new("s5x=info"))
        )
        .init();
}