Как мне вызвать функцию Rust из Go со строкой в качестве параметра?
Я пытался передать строку в функцию Rust (скомпилированную в Wasm), однако, насколько я понял, прямо сейчас нет возможности передать строки напрямую, потому что "str" не является типом в "мире FFI" (по крайней мере, так говорит компилятор ржавчины):
= 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, которая использует функцию alloc, подобную этой:
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» по следующим причинам:
- Я не могу найти эквивалент «instance.exports.memory.buffer» в Go (экземпляр — это тип «*wasmtime.Instance»).
- Я не знаю, как сделать то, что делает Unit8Buffer в Go.
Полезно прочитать о памяти в Wasm: (https://radu-matei.com/blog/practical-guide-to-wasm-memory/#exchange-strings-between-modules-and-runtimes)
1 ответ
Мне потребовалось некоторое время, чтобы понять, как именно работает пакет wasmtime Go, но в конце концов я решил свою проблему следующим образом:
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))
}