Function copy_tcp_to_ws

crates/openshell-cli/src/edge_tunnel.rs:186–214

Copy bytes from a local TCP reader into WebSocket binary frames.

(
    mut tcp_read: tokio::io::ReadHalf<TcpStream>,
    mut ws_sink: SplitSink<WebSocketStream<MaybeTlsStream<TcpStream>>, Message>,
)

/// Copy bytes from a local TCP reader into WebSocket binary frames.
async fn copy_tcp_to_ws(
    mut tcp_read: tokio::io::ReadHalf<TcpStream>,
    mut ws_sink: SplitSink<WebSocketStream<MaybeTlsStream<TcpStream>>, Message>,
) {
    let mut buf = vec![0u8; 32 * 1024];
    loop {
        match tcp_read.read(&mut buf).await {
            Ok(0) => {
                // EOF — send a close frame.
                let _ = ws_sink.close().await;
                break;
            }
            Ok(n) => {
                if ws_sink
                    .send(Message::Binary(buf[..n].to_vec().into()))
                    .await
                    .is_err()
                {
                    break;
                }
            }
            Err(e) => {
                debug!(error = %e, "tcp read error");
                let _ = ws_sink.close().await;
                break;
            }
        }
    }
}

/// Copy bytes from WebSocket binary frames into a local TCP writer.
async fn copy_ws_to_tcp(