Function run_tcp_forwarder

port-forward/src/tcp.rs:13–43

Run a TCP port forwarder: listen on `listen_addr`, forward to `target`.

(
    listen_addr: SocketAddr,
    target: SocketAddr,
    cancel: CancellationToken,
)

/// Run a TCP port forwarder: listen on `listen_addr`, forward to `target`.
pub async fn run_tcp_forwarder(
    listen_addr: SocketAddr,
    target: SocketAddr,
    cancel: CancellationToken,
) {
    let listener = match TcpListener::bind(listen_addr).await {
        Ok(l) => l,
        Err(e) => {
            tracing::error!("tcp bind {listen_addr} failed: {e}");
            return;
        }
    };
    tracing::info!("tcp forwarding {listen_addr} -> {target}");

    loop {
        tokio::select! {
            _ = cancel.cancelled() => break,
            result = listener.accept() => {
                let (client, peer) = match result {
                    Ok(v) => v,
                    Err(e) => {
                        tracing::warn!("tcp accept on {listen_addr}: {e}");
                        continue;
                    }
                };
                let cancel = cancel.child_token();
                tokio::spawn(handle_tcp_connection(client, peer, target, cancel));
            }
        }
    }
}

async fn handle_tcp_connection(
    mut client: TcpStream,