Rust TCP服务读取Edge/Chrome浏览器POST上传文件内容截断问题

问题根因 #

你当前的读取逻辑存在根本性错误，完全不符合TCP流和HTTP协议的设计规范：

• TCP是流式传输协议，单次read调用返回的字节数仅取决于当前内核套接字缓冲区中的可用数据量，和请求总长度没有任何关系。你假设「只有最后一次读取返回的字节数才会小于缓冲区长度」的前提完全不成立，Chrome、Edge等浏览器的TCP分片策略和Firefox不同，传输过程中很容易出现某次read返回长度小于1024的情况，你的代码此时直接中断读取，自然会出现请求截断。
• 额外的limita计数限制如果被触发，也会直接中断读取，同样会导致内容截断。
• 你之前尝试的read_to_end无效，是因为默认HTTP会启用keep-alive复用连接，浏览器发完单次请求后不会主动关闭TCP连接的发送端，read_to_end会一直阻塞等待EOF，始终无法返回，最终触发超时主动断开连接，导致浏览器端报错。

解决方案 #

HTTP请求的body长度是通过请求头明确标识的，你必须先解析请求头，再按照头中指定的长度读取body：

1. 先读取所有请求头，直到遇到请求头结束标记\r\n\r\n
2. 从请求头中提取Content-Length字段的值，获取请求body的总长度
3. 按照该长度精确读取对应字节数的内容即可，不需要依赖单次读取的返回长度判断结束

参考修正代码：

use std::io::{BufRead, BufReader, Read};
use std::net::TcpStream;

fn handle_request(mut stream: TcpStream) {
    let mut reader = BufReader::new(&mut stream);
    let mut headers = Vec::new();
    let mut line = Vec::new();
    let mut content_length: usize = 0;

    // 读取所有请求头
    loop {
        line.clear();
        let n = reader.read_until(b'\n', &mut line).unwrap();
        if n == 0 || line == b"\r\n" {
            break;
        }
        headers.push(line.clone());
        // 提取Content-Length
        if line.starts_with(b"Content-Length: ") {
            let val_str = String::from_utf8_lossy(&line[16..line.len()-2]);
            content_length = val_str.parse().unwrap_or(0);
        }
    }

    // 精确读取指定长度的body
    let mut body = vec![0; content_length];
    reader.read_exact(&mut body).unwrap();

    // 后续处理body的逻辑
}

注意事项 #

• 如果要支持Transfer-Encoding: chunked分块传输的请求，还需要额外实现分块编码的解析逻辑
• 生产环境建议直接使用成熟的HTTP框架如axum、actix-web处理HTTP协议解析，不需要手动处理底层字节流的读取逻辑，避免出现协议兼容性问题

内容的提问来源于stack exchange，提问作者navetix