为了账号安全,请及时绑定邮箱和手机立即绑定

如何从 Go 中调用 Rust 函数,并使用字符串作为参数?

如何从 Go 中调用 Rust 函数,并使用字符串作为参数?

Go
梦里花落0921 2022-08-01 17:00:49
我一直在尝试将字符串传递给Rust函数(编译为Wasm),但是对于我的理解,现在没有办法直接传递字符串,因为“str”不是“FFI世界”中的类型(至少这是rust编译器所说的):= help: consider using `*const u8` and a length instead所以我所做的是将函数更改为此形式(而不是使用简单的 &str 类型):#[no_mangle]pub extern "C" fn greet(s: *mut u8, len: usize) {    let s = std::str::from_utf8(unsafe { std::slice::from_raw_parts(s, len) }).unwrap();    println!("Hello, {}!", s)}这意味着我需要一个指针和u8中的字符串长度。但是,有人让我注意到WASM模块是沙盒的,因此它们不能像普通应用程序那样使用普通指针。因此,我必须使用这样的函数将内存分配到模块的线性内存中:use std::alloc::{alloc, dealloc, Layout};#[no_mangle]pub unsafe fn my_alloc(len: usize) -> *mut u8 {    let align = std::mem::align_of::<usize>();    let layout = Layout::from_size_align_unchecked(size, align);    alloc(layout)}这是一个 JS 函数的示例,它使用如下的分配函数:function copyMemory(data, instance) {  var ptr = instance.exports.alloc(data.length);  var mem = new Uint8Array(instance.exports.memory.buffer, ptr, data.length);  mem.set(new Uint8Array(data));  return ptr;}我的问题是我不知道如何将这个函数转换为Go,那是因为我被困在“var mem”行,原因如下:我在Go中找不到“instance.exports.memory.buffer”的等效项(实例是“*wasmtime.实例“类型)。我不知道如何做Unit8Buffer在Go中所做的。关于Wasm中记忆的好读:(https://radu-matei.com/blog/practical-guide-to-wasm-memory/#exchanging-strings-between-modules-and-runtimes)
查看完整描述

1 回答

?
UYOU

TA贡献1878条经验 获得超4个赞

我花了一点时间来确切地理解wasmtime Go-package是如何工作的,但最终我解决了我这样做的问题:


func main() {

    dir, err := ioutil.TempDir("", "out")

    check(err)

    defer os.RemoveAll(dir)

    stdoutPath := filepath.Join(dir, "stdout")


    engine := wasmtime.NewEngine()

    store := wasmtime.NewStore(engine)


    linker := wasmtime.NewLinker(store)


    // Configure WASI imports to write stdout into a file.

    wasiConfig := wasmtime.NewWasiConfig()

    wasiConfig.SetStdoutFile(stdoutPath)



    wasi, err := wasmtime.NewWasiInstance(store, wasiConfig, "wasi_snapshot_preview1")

    check(err)


    // Link WASI

    err = linker.DefineWasi(wasi)

    check(err)


    // Create our module

    module, err := wasmtime.NewModuleFromFile(store.Engine, "wasm_file.wasm")

    check(err)

    instance, err := linker.Instantiate(module)

    check(err)



    fn := instance.GetExport("greet").Func()

    memory := instance.GetExport("memory").Memory()

    alloc := instance.GetExport("my_alloc").Func()


    // // string for alloc

    size2 := int32(len([]byte("Elvis")))


    // //Allocate memomory

    ptr2, err := alloc.Call(size2)

    pointer, _ := ptr2.(int32)


    buf := memory.UnsafeData()

    for i, v := range []byte("Elvis") {

        buf[pointer+int32(i)] = v

    }


    // Use string func

    _, err = fn.Call(pointer, size2 )

    check(err)

    

    // Print WASM stdout

    out, err := ioutil.ReadFile(stdoutPath)

    check(err)

    fmt.Print(string(out))

}


查看完整回答
反对 回复 2022-08-01
  • 1 回答
  • 0 关注
  • 177 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信