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
38
39
40
41
42
43

//! Unix domain socket proxy module.
//!
//! Listens on a local UDS socket and forwards incoming stream connections
//! to a target TCP address.
//!
//! Useful for local inter-process communication or socket activation scenarios.

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

/// Start a Unix domain socket proxy from `unix_path` to `target_addr`.
///
/// Removes the existing socket file if it exists before binding.
///
/// ## Example
/// ```no_run
/// async {
///     tokio_multi_proxy::start_unix("/tmp/proxy.sock", "127.0.0.1:9000").await.unwrap();
/// };
/// ```
pub async fn start_unix(unix_path: &str, target_addr: &str) -> Result<()> {
    let _ = std::fs::remove_file(unix_path); // Clean up old socket
    let listener = UnixListener::bind(unix_path)?;
    println!("Unix socket proxy listening on {}", unix_path);

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

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

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