如何正确分配DST结构体？修正fat pointer尺寸错误问题

问题核心：直接将分配得到的裸指针强制转换为*mut DST时，Rust会把整个结构体的字节大小当作data字段的长度，而非我们期望的data_size。要解决这个问题，需要手动构造正确的Fat Pointer，让其长度部分对应data字段的实际大小。

解决方案步骤 #

1. 计算结构体前缀大小：即除data字段外所有成员的总大小，这里是4个usize的长度和。
2. 分配正确的内存布局：总大小为前缀大小 + data_size，对齐方式使用结构体自身的对齐要求（更严谨）。
3. 手动构造Fat Pointer：使用std::ptr::from_raw_parts_mut将裸指针和data_size组合成正确的*mut DST，该函数会为DST结构体生成包含正确长度信息的Fat Pointer。

修改后的代码 #

use std::alloc::{Allocator, Global, Layout};
use std::mem;
use std::ptr;

#[repr(C)]
struct DST {
    other_data: usize,
    other_data2: usize,
    other_data3: usize,
    other_data4: usize,
    data: [u8],
}

// 计算结构体前缀（非DST部分）的大小
const DST_PREFIX_SIZE: usize = mem::size_of::<(usize, usize, usize, usize)>();

unsafe fn create(data_size: usize) {
    // 计算总内存大小和正确的布局
    let total_size = DST_PREFIX_SIZE + data_size;
    let layout = Layout::from_size_align(total_size, mem::align_of::<DST>())
        .expect("Invalid layout");

    // 分配内存
    let raw_ptr = Global.allocate_zeroed(layout)
        .expect("Allocation failed")
        .as_ptr();

    // 手动构造正确的*mut DST Fat Pointer
    let fat_ptr: *mut DST = ptr::from_raw_parts_mut(raw_ptr, data_size);

    // 验证结果
    assert_eq!((*fat_ptr).data.len(), data_size); // 现在会通过
}

关键说明 #

• ptr::from_raw_parts_mut是构造自定义DST Fat Pointer的标准方式，第二个参数直接对应结构体中切片字段的长度。
• 使用mem::align_of::<DST>()确保内存对齐符合结构体要求，避免未定义行为。
• 必须保证total_size足够容纳前缀和data字段，否则会触发内存越界。

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