python 调用rust,从Rust函数返回一个字符串到Python

最直接的版本是这样的：

use libc::c_char;

use std::ffi::CString;

use std::mem;

#[no_mangle]

pub extern fn query() -> *mut c_char {

let s = CString::new("Hello!").unwrap();

s.into_raw()

}

这里我们返回一个指向零终止的字符序列的指针,可以传递给Python的c_char_p.您不能返回只是CString,因为它是不应该被直接用于C代码的Rust结构 – 它包裹Vec< u8>实际上由三个指针大小的整数组成.它直接与C的char *不兼容.我们需要从中获取原始指针. CString::into_raw()方法这样做 – 它通过值消耗CString,“忘记”它所以它的分配不会被销毁,并返回一个* mut c_char指针到数组的开头.

然而,这样的方式,字符串将被泄露,因为我们忘记了它在Rust方面的分配,它永远不会被释放.我不太了解Python的FFI,但解决这个问题的最直接方法是创建两个函数,一个用于生成数据,另一个用于释放它.那么你需要通过调用这个释放函数来释放Python端的数据：

// above function

#[no_mangle]

pub extern fn query() -> *mut c_char { ... }

#[no_mangle]

pub extern fn free_query(c: *mut c_char) {

// convert the pointer back to `CString`

// it will be automatically dropped immediately

unsafe { CString::from_raw(c); }

}

CString::from_raw()方法接受一个* mut c_char指针,并从中创建一个CString实例,计算进程中底层零终止字符串的长度.这个操作意味着所有权转移,所以产生的CString值将拥有分配,当它被删除时,分配被释放.这正是我们想要的.