如何使用切片作为参数从 Go 调用 Rust 函数



我想从 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

最新更新