Function handle_connection

crates/openshell-cli/src/edge_tunnel.rs:122–148

Handle a single tunneled connection: open a WebSocket to the edge and bidirectionally copy bytes.

(tcp_stream: TcpStream, config: &TunnelConfig)

/// Handle a single tunneled connection: open a WebSocket to the edge and
/// bidirectionally copy bytes.
async fn handle_connection(tcp_stream: TcpStream, config: &TunnelConfig) -> Result<()> {
    let ws_stream = open_ws(config).await?;
    let (ws_sink, ws_source) = ws_stream.split();
    let (tcp_read, tcp_write) = tokio::io::split(tcp_stream);

    // Two tasks: TCP->WS and WS->TCP. Abort the peer task once either
    // direction finishes so the connection tears down promptly.
    let mut tcp_to_ws = tokio::spawn(copy_tcp_to_ws(tcp_read, ws_sink));
    let mut ws_to_tcp = tokio::spawn(copy_ws_to_tcp(ws_source, tcp_write));

    tokio::select! {
        res = &mut tcp_to_ws => {
            if let Err(e) = res {
                debug!(error = %e, "tcp->ws task panicked");
            }
            ws_to_tcp.abort();
        }
        res = &mut ws_to_tcp => {
            if let Err(e) = res {
                debug!(error = %e, "ws->tcp task panicked");
            }
            tcp_to_ws.abort();
        }
    }

    Ok(())
}

/// Open a WebSocket connection to the edge proxy.
async fn open_ws(config: &TunnelConfig) -> Result<WebSocketStream<MaybeTlsStream<TcpStream>>> {