首页 > 解决方案 > 有没有办法强制在从特定堆栈帧返回后不使用 Rust 原始指针?

问题描述

我正在为(主要是 C 风格的)C++ 插件 SDK 编写 Rust 包装器。插件宿主是一个运行事件循环的图形桌面应用程序。该插件作为该事件循环的一部分定期被调用。每当发生这种情况时,插件都有控制权并可以调用任意主机函数。

我要包装的一个 C 函数返回一个原始指针。在该函数返回后,指针被保证是一个有效的 C 字符串,因此取消引用它是安全的。但是,在插件回调返回后(从而将控制权交还给主机),指针可能会变得陈旧。我如何为此编写一个符合人体工程学的函数包装器,它不会在某些时候导致未定义的行为,例如当消费者尝试在下一个事件循环周期中访问字符串时?

我考虑过以下方法:

1. 返回一个拥有的字符串

我可以立即取消引用指针并将内容复制到拥有的CString

pub fn get_string_from_host() -> CString {
    let ptr: *const c_char = unsafe { ffi.get_string() };
    unsafe { CStr::from_ptr(ptr).to_owned() }
}

这是冒昧的——也许我的包装器的消费者对获取拥有的字符串不感兴趣,因为他们只想进行比较(这甚至是我要说的主要用例)。复制字符串将完全是浪费。

2. 返回原始指针

pub fn get_string_from_host() -> *const c_char {
    unsafe { ffi.get_string() }
}

这只是将问题转移到消费者身上。

3.返回一个CStr引用(不安全的方法)

pub unsafe fn get_string_from_host<'a>() -> &'a CStr {
    let ptr: *const c_char = ffi.get_string();
    CStr::from_ptr(ptr)
}

这是不安全的,因为引用的生命周期不准确。在稍后的时间点访问引用可能会导致未定义的行为。将问题转移给消费者的另一种方式。

4. 关闭而不是返回一些东西

pub fn with_string_from_host<T>(f: impl Fn(&CStr) -> T) -> T {
    let ptr: *const c_char = unsafe { ffi.get_string() };
    f(unsafe { CStr::from_ptr(ptr) })
}

pub fn consuming_function() {
    let length = with_string_from_host(|s| s.to_bytes().len());
}

这有效,但确实需要习惯。


这些解决方案都不是真正令人满意的。

有没有办法确保“立即”使用返回值,这意味着它不会存储在任何地方或永远不会超出调用者的范围?

这听起来像是引用/生命周期的工作,但我不知道有任何生命周期注释,这意味着“仅在当前堆栈帧中有效”。如果有,我会使用它(仅用于说明):

pub fn get_string_from_host() -> &'??? CStr {
    let ptr: *const c_char = unsafe { ffi.get_string() };
    unsafe { CStr::from_ptr(ptr) }
}

pub fn consuming_function() {
    // For example, this shouldn't be possible in this case
    let prolonged: &'static CStr = get_string_from_host();
    // But this should
    let owned = get_string_from_host().to_owned();
}

标签: rustffilifetime

解决方案


您的问题和评论列出了您的选择。它主要归结为满足其他人的期望,即最不意外的规则。这主张返回一个拥有的String. 正如之前所说,所拥有的String涉及一个副本(除非在循环中调用无数次,否则它对性能的影响可以忽略不计)

我强烈建议不要使用原始指针和CStr参考解决方案,它们是脚枪。

就个人而言,我会选择闭包,因为它实现了基本情况:访问字符串的代码的上下文必须移动到字符串所在的位置;我们不能让字符串移动到上下文所在的位置(据我们所知,即使调用者也可能无法控制)。闭包解决方案应该允许您吃蛋糕并吃掉它:类型的闭包impl Fn(&CStr) -> T可以是|s| s.to_owned()with_string_from_host如果需要,可以返回一个副本。


推荐阅读