时间戳

首页 > 解决方案 > Rust TCP如何获取字节数组长度？

tcp - Rust TCP如何获取字节数组长度？

问题描述

我有一个 rust 的 TCP 客户端，它应该与 Java 服务器通信。我掌握了基础知识，可以在它们之间发送字节数组。

但是对于字节数组缓冲区，我需要知道字节数组的长度。但我不知道我应该得到它。目前，我现在只有一个固定大小的缓冲区。

我的 Rust 代码如下所示：

    loop {
        let mut buffer = vec![0; 12]; //fixed buffer length
        let n = stream.read(&mut buffer).await;
        let text = from_utf8(&buffer).unwrap();
        println!("{}", text);
    }

在 Java 中，您可以使用 DataInputStream 直接将缓冲区的大小作为整数发送。有没有办法在生锈中做到这一点？

例如，这就是我在 Java 中的做法：

    public String readMsg(Socket socket) throws IOException {
        DataInputStream in = new DataInputStream(new BufferedInputStream(socket.getInputStream()));
        byte[] bytes = new byte[in.readInt()]; //dynamic buffer length
        in.readFully(bytes);
        return new String(bytes, StandardCharsets.US_ASCII);
    }

标签： tcprust

解决方案

您想知道的是您正在使用的协议的属性。它不是您使用的编程语言的属性。根据您的 Java 代码，您似乎正在使用一种协议，该协议在消息数据（有符号/无符号？）之前发送一个 4 字节长度字段。

如果是这种情况，您可以在 Rust 中以相同的方式处理读取消息：1. 读取 4 个字节以获得长度信息 2. 读取剩余数据 3. 反序列化数据

fn read_message(stream: Read) -> io::Result<String> {
    let mut buffer = [0u8; 4];
    // Read the length information
    stream.read_exact(&mut buffer[..])?;
    // Deserialize the length
    let size = u32::from_be_bytes(buffer);
    // Allocate a buffer for the message
    // Be sure to check against a maximum size before doing this in production
    let mut payload = vec![0; size];
    stream.read_exact(&mut payload[..]).await;
    // Convert the buffer into a string
    let text = String::from_utf8(payload).map_err(/* omitted */)?;
    println!("{}", text);
    Ok(text)
}

这显然只有在您的协议使用带有 4 字节无符号整数前缀的长度前缀消息时才是正确的。这是您需要检查的内容。

推荐阅读