Read::read 是否保证追加数据而不覆盖任何现有数据？

Question

我正在开发一个 SMTP 库，该库使用缓冲 reader.

通过网络读取行

我想要一种不错的、安全的方式来从网络读取数据，而不依赖于 Rust 内部机制来确保代码按预期工作。具体来说，我想知道 Read trait guarantees that data read with Read::read 是否附加到作为参数传递的缓冲区而不是完全覆盖缓冲区。

目前，I use a Range 确保现有数据不会在不依赖 Rust 内部结构的情况下被覆盖。

但是，考虑到 Rust used to have 是一种很好的方法来做我想做的事情，我想知道是否可以改进当前代码，也可能删除 unsafe 块。

Answer 1

不，它不保证：

use std::io::prelude::*;
use std::str;

fn main() {
    let mut source1 = "hello, world!".as_bytes();
    let mut source2 = "moo".as_bytes();

    let mut dest = [0; 128];
    source1.read(&mut dest).unwrap();
    source2.read(&mut dest).unwrap();

    let s = str::from_utf8(&dest[..16]).unwrap();

    println!("{:?}", s)
}

这会打印

"moolo, world!\u{0}\u{0}\u{0}"

具体来说，它不能做你想做的，纯粹基于类型签名：

fn read(&mut self, buf: &mut [u8]) -> Result<usize>;

read 方法可以访问的所有内容都是您的可变切片 - 没有地方可以存储像 "how far in the buffer you are" 这样的信息。此外，您不能 "extend" 具有更多元素的可变切片 - 您只能改变切片内的值。

对于您的具体情况，您可能需要查看 BufRead::read_until。这是一个几乎没有测试过的例子：

use std::io::{BufRead,BufReader};
use std::str;

fn main() {
    let source1 = "header 1\r\nheader 2\r\n".as_bytes();
    let mut reader = BufReader::new(source1);

    let mut buf = vec![];
    buf.reserve(128); // Maybe more efficient?

    loop {
        match reader.read_until(b'\n', &mut buf) {
            Ok(0) => break,
            Ok(_) => {},
            Err(_) => panic!("Handle errors"),
        }

        if buf.len() < 2 { continue }

        if buf[buf.len() - 2] == b'\r' {
            {
                let s = str::from_utf8(&buf).unwrap();
                println!("Got a header {:?}", s);
            }
            buf.clear();
        }
    }
}

Read::read 是否保证追加数据而不覆盖任何现有数据？

Does Read::read guarantee to append data and not overwrite any existing one?

rust