我想从 Go 调用一些用 Rust 编写的外部函数,并引用切片。
我有以下 Rust 代码:
extern crate libc;
#[no_mangle]
pub extern "C" fn callme(data: &mut [libc::c_double]) -> i32 {
data.len() as i32
}
此函数通过以下 C 样式头文件提供给 cgo 编译器:
#IFNDEF BOGUSLIB_H
#DEFINE BOGUSLIB_H
extern int callme(double* data);
#ENDIF
我现在可以从 Go 调用这个函数,使用 Rust crate 编译为 cdylib:
//#cgo CFLAGS: -Ipath/to/libfolder
//#cgo LDFLAGS: -Lpath/to/libfolder -lboguslib
//#include <boguslib.h>
import "C"
import (
"unsafe"
. "fmt"
)
func CallmeExternal() {
data := make([]float64, 1, 1)
data[0] = 1.0
ptr := (*C.double)(unsafe.Pointer(&data[0]))
size := C.callme(ptr)
printf("size %v",size)
}
Go 代码使用不安全的指针技巧来访问后备数组,因为切片定义如下
type Slice struct {
data *byte
uint32 len
uint32 cap
}
当我执行上面的代码时,传递的引用的长度非常大。如何访问实际数据,以及此时返回的内容?
根据 @matthieu-m 提供的 The Rust FFI Omnibus,我已经成功地重写了代码。函数签名必须接受目标语言理解的类型。
Rust 函数签名更改为:
#[no_mangle]
pub extern "C" fn callme(slice: *const libc::c_double, len: libc::size_t) -> libc::c_int {
let data = slice::from_raw_parts(slice, len as usize);
data.len() as i32
}
头文件中的声明如下所示:
// skip include guards
#include <stdio.h>
extern int callme(double* slice, size_t len);
来自Go的电话现在也发生了变化
func CallmeExternal() {
data := make([]float64, 2, 2)
data[0] = 1.0
ptr := (*C.double)(unsafe.Pointer(&data[0]))
len := C.size_t(len(data))
size := C.callme(ptr, len)
printf("size %v",size)
}
这将返回2
。