锐客网

  1. 首页 > 睿知 > it技术 > >

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

Python调用rust

最直接的版本是这样的：
use libc::c_char;
use std::ffi::CString;
use std::mem;
#[no_mangle]
【python|python 调用rust,从Rust函数返回一个字符串到Python】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值将拥有分配,当它被删除时,分配被释放.这正是我们想要的.

    推荐阅读


    上一篇：python调用rust,将字符串列表从Python传递到Rust

    下一篇：python格式化字符串漏洞_Python新型字符串格式漏洞分析及解决方案