连接被对等方重置。东京，选择

Question

我正在写socks5代理服务器。该程序正在异步运行，我尝试使用 tokio::select，但是当我想获取接收到的数据的大小时，程序由于此错误而终止：

thread 'tokio-runtime-worker' panicked at 'called `Result::unwrap()` on an `Err` value: Os { code: 54, kind: ConnectionReset, message: "Connection reset by peer" }'

发生错误的函数：

async fn exchange_loop(mut client: TcpStream, address: SocketAddr, cmd: u8) {
    let mut remote = TcpStream::connect(address).await.unwrap();
    let mut buffer_client: [u8; 4096] = [0; 4096];
    let mut buffer_remote: [u8; 4096] = [0; 4096];

    loop {
        tokio::select! {
            size = client.read(&mut buffer_client) => {
                let size = size.unwrap();
                remote.write(buffer_client.as_ref()).await.unwrap();
                println!("Send from client {} => {} {} KB", client.peer_addr().unwrap(), remote.peer_addr().unwrap(), size as f32 / 1024.);
                if size <= 0 {
                    break;
                };
                buffer_client = [0; 4096];
            }
            size = remote.read(&mut buffer_remote) => {
                let size = size.unwrap();
                client.write(buffer_remote.as_ref()).await.unwrap();
                println!("Send from remote {} => {} {} KB", address, client.peer_addr().unwrap(), size as f32 / 1024.);
                if size <= 0 {
                    break;
                };
                buffer_remote = [0; 4096];
            }
        }
    }
    println!("End connection to {}", address);
}

Answer 1

您需要处理错误，如前面的注释中所指定。您需要能够从客户端所做的任何操作中恢复，例如中断连接。

如何恢复的示例：

if let Ok(actual_size) = size {
    if let Ok(_) = remote.write(buffer_client.as_ref()).await {
        println!("Send from client {} => {} {} KB", client.peer_addr().unwrap(), remote.peer_addr().unwrap(), size as f32 / 1024.);
        if size <= 0 {
            break;
        };

        buffer_client = [0; 4096];
    }
}

我输入 Ok(_) 因为我不知道您对第二次解包有何期望。