如何通过FFI返回大小未知的Rust Vec？

优化后的手动实现 #

你的现有方案思路正确，但可以简化——无需对Vec<T>额外装箱，直接暴露其底层指针、长度和容量更符合FFI惯例，减少内存间接层，同时提升效率：

use std::os::raw::c_uint;
use std::mem::ManuallyDrop;

#[repr(C)]
#[derive(Debug)]
pub struct FfiSlice<T> {
    ptr: *mut T,
    len: c_uint,
    cap: c_uint, // 用于Rust端回收内存，可选但建议保留
}

impl<T> FfiSlice<T> {
    pub fn from_vec(v: Vec<T>) -> Self {
        let mut vec = ManuallyDrop::new(v);
        FfiSlice {
            ptr: vec.as_mut_ptr(),
            len: vec.len() as c_uint,
            cap: vec.capacity() as c_uint,
        }
    }

    // 安全回收Rust分配的内存，避免泄漏
    pub unsafe fn drop(self) {
        let _ = Vec::from_raw_parts(self.ptr, self.len as usize, self.cap as usize);
    }
}

#[no_mangle]
pub extern "C" fn compute_my_vec() -> FfiSlice<u8> {
    // 示例逻辑：生成目标Vec<u8>
    let data = vec![1, 2, 3, 4, 5];
    FfiSlice::from_vec(data)
}

#[no_mangle]
pub extern "C" fn copy_from_my_vec(ffi_slice: FfiSlice<u8>, buffer: *mut u8) {
    let len = ffi_slice.len as usize;
    if !buffer.is_null() && len > 0 {
        unsafe {
            // 用标准库高效复制函数替代手动循环
            std::ptr::copy_nonoverlapping(ffi_slice.ptr, buffer, len);
        }
    }
    // 回收Rust端内存
    unsafe { ffi_slice.drop(); }
}

优化点说明 #

• 直接复用Vec的原始内存结构，避免额外装箱开销
• 使用std::ptr::copy_nonoverlapping提升复制效率
• 通过ManuallyDrop安全转移Vec的内存所有权到FFI层

无需手动实现：常用FFI工具库 #

如果不想自己处理unsafe细节，这些成熟库可以直接用：

• libc：提供标准C类型定义，配合Rust 1.48+的Vec::into_raw_parts，可快速生成C兼容的指针/长度对
• ffi_utils：封装了集合类型的FFI转换逻辑，简化Vec到C结构的转换
• safe-ffi：专注安全FFI交互，提供预封装的FfiVec类型，自动处理内存管理

更简洁的分阶段调用模式 #

如果计算开销不大，可以采用两次调用的模式，无需自定义FFI结构：

1. 第一次调用获取结果长度
2. 第二次调用传入预分配的缓冲区填充数据

use std::os::raw::c_uint;

#[no_mangle]
pub extern "C" fn compute_my_vec_len() -> c_uint {
    // 实际生成数据的逻辑，这里仅作示例
    let data = vec![1,2,3,4,5];
    data.len() as c_uint
}

#[no_mangle]
pub extern "C" fn compute_my_vec_fill(buffer: *mut u8, buffer_len: c_uint) -> c_uint {
    // 注意：如果计算开销大，需避免重复计算，可通过状态缓存或上下文处理
    let data = vec![1,2,3,4,5];
    let actual_len = data.len();
    
    // 缓冲区不足时返回所需长度，提示调用方重新分配
    if buffer.is_null() || buffer_len as usize < actual_len {
        return actual_len as c_uint;
    }
    
    unsafe {
        std::ptr::copy_nonoverlapping(data.as_ptr(), buffer, actual_len);
    }
    actual_len as c_uint
}

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