如何将一个大的 Vec<i32> 切片为 &[u8]？

Question

我不知道如何将 Vec<i32> 转换为 &[u8] 切片。

fn main() {
    let v: Vec<i32> = vec![1; 100_000_000];
    let v_bytes: &[u8] = /* ... */;
}

我想将一个大文件 Vec<i32> 写入一个文件，以便将来可以读回。

Answer 1

您可以使用 std::slice::from_raw_parts:

let v_bytes: &[u8] = unsafe {
    std::slice::from_raw_parts(
        v.as_ptr() as *const u8,
        v.len() * std::mem::size_of::<i32>(),
    )
};

根据对这个答案的评论，你应该把这段代码包装在一个函数中，并让 return 值借用输入，这样你就可以尽可能地使用借用检查器：

fn as_u8_slice(v: &[i32]) -> &[u8] {
    unsafe {
        std::slice::from_raw_parts(
            v.as_ptr() as *const u8,
            v.len() * std::mem::size_of::<i32>(),
        )
    }
}

Answer 2

从 Rust 1.30 开始，最好的解决方案是使用 slice::align_to:

fn main() {
    let v: Vec<i32> = vec![1; 8];

    let (head, body, tail) = unsafe { v.align_to::<u8>() };
    assert!(head.is_empty());
    assert!(tail.is_empty());

    println!("{:#x?}", body);
}

这可以正确处理第一种类型和第二种类型的对齐方式不匹配的情况。在此示例中，我通过 assert! 语句确保 i32 的对齐方式大于 u8 的对齐方式。

我用和运行得到了硬币的另一面 - 读回向量：

use std::fs::File;
use std::io::{Read, Write};
use std::{mem, slice};

fn as_u8_slice(v: &[i32]) -> &[u8] {
    let element_size = mem::size_of::<i32>();
    unsafe { slice::from_raw_parts(v.as_ptr() as *const u8, v.len() * element_size) }
}

fn from_u8(v: Vec<u8>) -> Vec<i32> {
    let data = v.as_ptr();
    let len = v.len();
    let capacity = v.capacity();
    let element_size = mem::size_of::<i32>();

    // Make sure we have a proper amount of capacity (may be overkill)
    assert_eq!(capacity % element_size, 0);
    // Make sure we are going to read a full chunk of stuff
    assert_eq!(len % element_size, 0);

    unsafe {
        // Don't allow the current vector to be dropped
        // (which would invalidate the memory)
        mem::forget(v);

        Vec::from_raw_parts(
            data as *mut i32,
            len / element_size,
            capacity / element_size,
        )
    }
}

fn do_write(filename: &str, v: &[i32]) {
    let mut f = File::create(filename).unwrap();
    f.write_all(as_u8_slice(v)).unwrap();
}

fn do_read(filename: &str) -> Vec<i32> {
    let mut f = File::open(filename).unwrap();
    let mut bytes = Vec::new();

    f.read_to_end(&mut bytes).unwrap();

    from_u8(bytes)
}

fn main() {
    let v = vec![42; 10];
    do_write("vector.dump", &v);
    let v2 = do_read("vector.dump");

    assert_eq!(v, v2);
    println!("{:?}", v2)
}

如何将一个大的 Vec<i32> 切片为 &[u8]？

How to slice a large Vec<i32> as &[u8]?

rust