Rust如何在字段分配前从闭包访问impl字段及WasmTime无self回调取值

解决方案 #

你的两个核心问题可以通过WasmTime提供的Store自定义上下文 + Caller参数 + OnceCell晚初始化容器统一解决：

1. 核心思路 #

• 自定义宿主共享状态全部托管到Store的泛型参数中，用OnceCell封装需要晚初始化的wasmTable字段，满足「闭包创建时未初始化、调用前完成初始化」的语义。
• WasmTime的Host回调支持首位加Caller<'_, T>参数，通过该参数可以直接访问Store中存储的所有宿主状态，不需要额外传入self。

2. 代码修改示例 #

首先引入依赖once_cell = "1.17.0"（单线程场景用unsync版本，多线程场景用sync版本即可：

use once_cell::unsync::OnceCell;
use wasmtime::{Caller, Engine, Linker, Module, Store, Table};
use wasmtime_wasi::WasiCtx;

// 自定义Store上下文，存储所有需要在回调中访问的状态
pub struct HostState {
    wasi: WasiCtx,
    wasm_table: OnceCell<Table>,
}

pub struct EmWasmNode {
    wasm_store: Store<HostState>,
}

impl EmWasmNode {
    // 私有回调函数，不需要self参数
    fn emscripten_set_main_loop(
        mut caller: Caller<'_, HostState>,
        p0: i32,
        p1: i32,
        p2: i32,
    ) {
        println!("emscripten_set_main_loop {} {} {}", p0, p1, p2);
        // 直接从Caller中获取状态
        let table = caller.data().wasm_table.get().unwrap();
        let browser_iteration_func = table
            .get(&mut caller, p0 as u32)
            .unwrap()
            .unwrap_funcref();
        browser_iteration_func.call(&mut caller, &[]).unwrap();
    }

    pub fn new(file_path: &str) -> Result<Self, Box<dyn std::error::Error>> {
        let engine = Engine::default();
        let module = Module::from_file(&engine, file_path)?;
        let mut linker = Linker::new(&engine);
        // 绑定wasi到自定义上下文的wasi字段
        wasmtime_wasi::add_to_linker(&mut linker, |state: &mut HostState| &mut state.wasi)?;

        let wasi = WasiCtxBuilder::new()
            .inherit_stdio()
            .inherit_args()?
            .build();
        // 初始化自定义上下文，wasm_table此时是空的OnceCell
        let host_state = HostState {
            wasi,
            wasm_table: OnceCell::new(),
        };
        let mut store = Store::new(&engine, host_state);

        // 绑定回调，直接传私有函数即可
        linker.func_wrap(
            "env",
            "emscripten_set_main_loop",
            Self::emscripten_set_main_loop
        )?;

        let instance = linker.instantiate(&mut store, &module)?;

        let table = instance
            .get_export(&mut store, "__indirect_function_table")
            .as_ref()
            .and_then(|e| e.table())
            .cloned()
            .unwrap();
        // 回调调用前完成OnceCell赋值
        store.data_mut().wasm_table.set(table).unwrap();

        let start = instance.get_typed_func::<(), (), _>(&mut store, "_start")?;
        start.call(&mut store, ())?;

        Ok(Self {
            wasm_store: store,
        })
    }
}

3. 说明 #

• OnceCell的set操作仅执行一次，后续访问无锁开销，完全匹配你「初始化后不再修改」的场景，也能保证调用时数据一定存在。
• Caller参数是WasmTime Host函数的标准特性，不会破坏回调签名的兼容性，Wasm侧调用时感知不到该参数的存在。
• 所有宿主状态都托管在Store内部，不需要额外维护引用关系，完全符合Rust借用检查规则，不会出现悬垂引用问题。

内容的提问来源于stack exchange，提问作者rygo6