forward 0.1.0-alpha.1

open a TCP port and foward all trafics through a proxy to another server or port
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
44
45
46

use super::Socks5Stream;
use crate::auth::Authentication;
use crate::server::pipe;
use crate::target_addr::ToTargetAddr;
use futures::try_join;
use std::io;
use std::net::Shutdown;
use tokio::net::TcpStream;
use tokio::net::ToSocketAddrs;

pub async fn forward_tcp_to_socks5(
    mut client: TcpStream,
    proxy: impl ToSocketAddrs,
    proxy_auth: &Authentication,
    target: impl ToTargetAddr,
) -> io::Result<()> {
    println!("Accepted connection from {:?}", client.peer_addr().unwrap());

    let (client_read, client_write) = client.split();

    // let mut proxy_stream = Socks5Stream::connect(proxy, target).await.unwrap();
    let mut proxy_stream = Socks5Stream::connect(proxy, target, proxy_auth)
        .await?
        .into_inner();

    let (proxy_read, proxy_write) = proxy_stream.split();

    try_join!(
        pipe(client_read, proxy_write),
        pipe(proxy_read, client_write),
    )?;

    println!("[client] joined task done");

    // TODO: better error
    match client.shutdown(Shutdown::Both) {
        Ok(e) => e,
        Err(error) => eprintln!("[client] client <=> here shutdown with failue: {}", error),
    }
    match proxy_stream.shutdown(Shutdown::Both) {
        Ok(e) => e,
        Err(error) => eprintln!("[client] client <=> here shutdown with failue: {}", error),
    }

    Ok(())
}