Rust封装C库：如何让借用检查器跟踪Data对buffer的所有权

要让Rust的借用检查器识别Data实例持有buffer的使用权，核心是给Data结构体添加生命周期参数，将其与输入buffer的生命周期绑定，这样借用检查器就能跟踪到Data实例存活期间，原buffer不能被其他可变引用访问。

1. 修改Data结构体，添加生命周期参数 #

将Data定义为带生命周期的结构体，明确它依赖于buffer的生命周期：

#[repr(C)]
pub struct data_type {
    flag: u32,
    buffer: *mut f32,
}

extern "C" {
    // 修正原代码类型错误：data_struct → data_type
    pub fn init_struct(
        s: *mut data_type,
        flag: i32,
        buffer: *mut f32,
    );

    pub fn run(
        s: *const data_type,
        input_data: *const f32,
        output_data: *mut f32,
    );
}

// 添加生命周期 'a，表明 Data 实例存活期不能超过 buffer 的存活期
pub struct Data<'a>(data_type, &'a mut [f32]);

这里额外保留buffer的可变引用在Data中，是为了让借用检查器直接跟踪该引用，避免仅靠原始指针导致的检查失效。

2. 实现Data::new方法，绑定生命周期 #

调整new方法，让返回的Data实例与输入buffer的生命周期绑定：

impl<'a> Data<'a> {
    pub fn new(flag: u32, buffer: &'a mut [f32]) -> Self {
        let mut data = MaybeUninit::<data_type>::uninit();
        unsafe {
            init_struct(data.as_mut_ptr(), flag as i32, buffer.as_mut_ptr());
            // 持有初始化后的C结构体和buffer的可变引用
            Data(data.assume_init(), buffer)
        }
    }

    pub fn run(&self, input: &[f32], output: &mut [f32]) {
        // 确保输入输出长度符合C API要求（示例中为64）
        assert_eq!(input.len(), 64);
        assert_eq!(output.len(), 64);
        
        unsafe {
            run(
                &self.0,
                input.as_ptr(),
                output.as_mut_ptr(),
            );
        }
    }
}

3. 效果验证 #

现在创建Data实例后，原buffer的可变引用会被Data持有，借用检查器会阻止任何对原buffer的新可变引用：

fn main() {
    let mut buffer = vec![0.0f32; 1000];
    {
        let data = Data::new(32, &mut buffer);
        // 以下代码会被借用检查器拒绝：buffer已被Data持有可变引用
        // let mut another_ref = &mut buffer;
        let mut input = [0.0f32; 64];
        let mut output = [0.0f32; 64];
        data.run(&input, &mut output);
    }
    // Data实例销毁后，buffer可重新被借用
    let mut ref_after = &mut buffer;
}

关键说明 #

• 生命周期'a明确了Data实例的存活期不能超过传入的buffer，借用检查器会严格执行这一约束。
• 保留&'a mut [f32]是让Rust直接跟踪引用关系，避免仅靠原始指针绕过借用检查。
• 修正了原代码中data_struct的笔误，确保与C结构体名称一致。

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