时间戳

首页 > 解决方案 > 为外部 C 库分配不透明缓冲区的正确 Rust 方法是什么？

c - 为外部 C 库分配不透明缓冲区的正确 Rust 方法是什么？

问题描述

我有一个外部库（例如 libcisland.so），其界面如下：

size_t lib_handle_size();
typedef void* handle;
int lib_init(handle h);
int lib_store(handle h, int value);
int lib_restore(handle h, int *pvalue);

该库的用户应执行以下操作：

// allocate some buffer in client address space
handle h = malloc(lib_handle_size());
// pass this buffer to library for initialization
if (lib_init(h)) { /* handle errors */ }
// library initializes this handle by some opaque fashion
// then user uses it
lib_store(h,42);
int r;
lib_restore(h,&r);
// after all work is done, user frees this handle
free(h);

我不知道如何正确地将这个接口包装到 Rust。这就是我最终的结果：

pub struct Island {
    h: Handle,
    v: Vec<u8>,
}

impl Island {
    pub fn new() -> Island {
        let len = unsafe { lib_handle_size() };
        let mut v: Vec<u8> = Vec::with_capacity(len);
        let h: Handle = v.as_mut_ptr();
        Island { v:v, h:h, }
    }

    pub fn store(&mut self, v: i32) {
        unsafe { lib_store(self.h, v); }
    }

    pub fn restore(&mut self) -> i32 {
        let mut v = 0;
        unsafe { lib_restore(self.h, &mut v); }
        v
    }
}

impl Drop for Island {
    fn drop(&mut self) {
        drop(&mut self.v);
    }
}

/// unsafe part
use libc::size_t;
pub type Handle = *mut u8;
#[link(name="cisland")]
extern {
    pub fn lib_handle_size() -> size_t;
    pub fn lib_init(h: Handle) -> i32;
    pub fn lib_store(h: Handle, value: i32) -> i32;
    pub fn lib_restore(h: Handle, pvalue: &mut i32) -> i32;
}

可以Vec(u8)用于此目的吗？这个Drop特征是否正确实施？

标签： crustffi

解决方案

可以为此目的使用 Vec(u8) 吗？

我认为没Vec<u8>问题，但你应该初始化它而不是使用零长度向量，指向未初始化的内存。使用它也会更加健壮，Box<[u8]>因为这将强制它不会被意外重新分配。

这个 Drop 特征是否正确实施？

根本不需要实施Drop。无论如何，每个字段Island都会正确删除。

而不是存储句柄，我每次都会使用一种方法来获取它。那么你的结构就简单多了。

use libc::c_void;

pub struct Island {
    buf: Box<[u8]>,
}

impl Island {
    pub fn new() -> Island {
        let len = unsafe { lib_handle_size() };
        let v: Vec<u8> = vec![0; len];
        Island { buf: v.into_boxed_slice() }
    }

    pub fn store(&mut self, v: i32) {
        unsafe { lib_store(self.handle_mut(), v); }
    }

    pub fn restore(&mut self) -> i32 {
        let mut v = 0;
        unsafe { lib_restore(self.handle_mut(), &mut v); }
        v
    }

    fn handle_mut(&mut self) -> *mut c_void {
        self.buf.as_mut_ptr() as *mut c_void
    }
}

您不需要Drop实现，因为Box它会在超出范围时自动删除（就像 a 一样Vec）。

推荐阅读