Rust跨大小Vec转换：如何确保无多余容量避免内存泄漏？

问题背景 #

你定义了C布局的结构体BinaryCandle，并实现了将Vec<u8>转换为Vec<BinaryCandle>的函数：

#[repr(C)]
pub struct BinaryCandle {
    pub open: f64,
    pub high: f64,
    pub low: f64,
    pub close: f64,
    pub volume: f64,
    pub time: i64,
}

转换函数原本依赖shrink_to_fit()尝试将原Vec<u8>的容量收缩至与长度一致，避免后续用from_raw_parts转换时泄漏多余内存：

pub fn owned_vec_from(bytes: Vec<u8>) -> Result<Vec<BinaryCandle>, BinaryError> {
    // Do not deallocate buffer for the original Vec as it has the
    // same memory space as the returned Vec.
    let mut bytes = core::mem::ManuallyDrop::new(bytes);

    // Ensure capacity equals length.
    // If not, we could leak memory, or have extra allocated memory
    // that is not a proper multiple of BinaryCandle.
    bytes.shrink_to_fit();

    let ptr = bytes.as_ptr().cast::<BinaryCandle>();

    Self::check_alignment(ptr)?;
    Self::check_length(&bytes)?;

    // Unchecked division is safe as size is non-zero.
    let len = bytes.len() / Self::size();

    // SAFETY: Pointer is guaranteed to be non-null, and properly aligned.
    // We have asserted the memory range is a valid length/multiple for BinaryCandle.
    // BinaryCandle does not implement Drop, so we do not risk a double free.
    // Original Vec will not be deallocated.
    // By calling shrink_to_fit first, we ensure we are not leaking any memory,
    // as the capacity will equal the length once that function is called.
    unsafe { Ok(Vec::from_raw_parts(ptr.cast_mut(), len, len)) }
}

但shrink_to_fit()仅为提示性操作，标准库不保证完全释放多余容量；into_boxed_slice()内部同样依赖shrink_to_fit()，无法确保最终容量与长度严格一致，因此需要可靠方案解决这个问题。

可靠解决方案 #

方案1：通过Box<[u8]>中转（复用原内存，无泄漏风险） #

Box<[u8]>是固定长度的内存块，不存在多余容量。将原Vec<u8>转换为Box<[u8]>后再转回Vec<u8>，可以确保新Vec的容量与长度完全相等。

修改后的代码如下：

pub fn owned_vec_from(bytes: Vec<u8>) -> Result<Vec<BinaryCandle>, BinaryError> {
    // 转换为Box<[u8]>确保无多余容量，再转回Vec
    let bytes_box = bytes.into_boxed_slice();
    let mut bytes = core::mem::ManuallyDrop::new(Vec::from(bytes_box));

    // 此时bytes的容量已严格等于长度，无需调用shrink_to_fit()

    let ptr = bytes.as_ptr().cast::<BinaryCandle>();

    Self::check_alignment(ptr)?;
    Self::check_length(&bytes)?;

    let len = bytes.len() / Self::size();

    // SAFETY: 指针非空且对齐，内存长度符合BinaryCandle的倍数要求
    // BinaryCandle无Drop实现，原Vec已被ManuallyDrop包裹不会释放
    // 当前Vec容量等于长度，无多余内存泄漏风险
    unsafe { Ok(Vec::from_raw_parts(ptr.cast_mut(), len, len)) }
}

原理说明：

• into_boxed_slice()会将Vec的内存调整为与长度完全匹配的块（即使内部调用shrink_to_fit()，最终Box<[u8]>的内存是精确长度）。
• Vec::from(bytes_box)将固定长度的切片Box转换为Vec时，Vec的容量会被设置为切片的长度，确保cap == len。

方案2：安全内存复制（无unsafe，性能略有开销） #

如果可以接受内存复制的性能开销，完全避免unsafe代码的方案更稳妥：

pub fn owned_vec_from_safe(bytes: Vec<u8>) -> Result<Vec<BinaryCandle>, BinaryError> {
    Self::check_length(&bytes)?;
    let candle_count = bytes.len() / Self::size();
    
    // 预分配刚好容纳所有BinaryCandle的Vec
    let mut candles = Vec::with_capacity(candle_count);
    
    unsafe {
        // 提前设置长度，后续直接复制内存
        candles.set_len(candle_count);
        // 安全复制：原指针已通过check_alignment确保对齐，长度符合要求
        core::ptr::copy_nonoverlapping(
            bytes.as_ptr().cast::<BinaryCandle>(),
            candles.as_mut_ptr(),
            candle_count,
        );
    }
    
    Ok(candles)
}

原理说明：

• 直接创建容量匹配的Vec<BinaryCandle>，通过copy_nonoverlapping将原Vec<u8>的内存复制到新Vec中。
• 原Vec<u8>会被正常Drop，无内存泄漏风险；新Vec的容量与长度严格一致。

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