tokio-multi-proxy 0.1.2

A modular async proxy with TLS/mTLS and multiple transport support
Documentation 
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

//! TCP proxy module.

use tokio::{net::{TcpListener, TcpStream}, io::copy_bidirectional};
use anyhow::Result;

/// Start a plain TCP proxy from `bind_addr` to `target_addr`.
///
/// This is raw passthrough — bytes are copied in both directions without encryption.
///
/// ## Example
/// ```no_run
/// async {
///     tokio_multi_proxy::start_tcp("0.0.0.0:8080", "127.0.0.1:9000").await.unwrap();
/// };
/// ```
pub async fn start_tcp(bind_addr: &str, target_addr: &str) -> Result<()> {
    let listener = TcpListener::bind(bind_addr).await?;
    println!("TCP proxy listening on {}", bind_addr);

    loop {
        let (inbound, _) = listener.accept().await?;
        let target = target_addr.to_string();

        tokio::spawn(async move {
            if let Err(e) = handle_connection(inbound, &target).await {
                eprintln!("TCP Proxy error: {:?}", e);
            }
        });
    }
}

async fn handle_connection(mut inbound: TcpStream, target_addr: &str) -> Result<()> {
    let mut outbound = TcpStream::connect(target_addr).await?;
    copy_bidirectional(&mut inbound, &mut outbound).await?;
    Ok(())
}