bcf3544924
* Optimize `Func::call` and its C API This commit is an alternative to #3298 which achieves effectively the same goal of optimizing the `Func::call` API as well as its C API sibling of `wasmtime_func_call`. The strategy taken here is different than #3298 though where a new API isn't created, rather a small tweak to an existing API is done. Specifically this commit handles the major sources of slowness with `Func::call` with: * Looking up the type of a function, to typecheck the arguments with and use to guide how the results should be loaded, no longer hits the rwlock in the `Engine` but instead each `Func` contains its own `FuncType`. This can be an unnecessary allocation for funcs not used with `Func::call`, so this is a downside of this implementation relative to #3298. A mitigating factor, though, is that instance exports are loaded lazily into the `Store` and in theory not too many funcs are active in the store as `Func` objects. * Temporary storage is amortized with a long-lived `Vec` in the `Store` rather than allocating a new vector on each call. This is basically the same strategy as #3294 only applied to different types in different places. Specifically `wasmtime::Store` now retains a `Vec<u128>` for `Func::call`, and the C API retains a `Vec<Val>` for calling `Func::call`. * Finally, an API breaking change is made to `Func::call` and its type signature (as well as `Func::call_async`). Instead of returning `Box<[Val]>` as it did before this function now takes a `results: &mut [Val]` parameter. This allows the caller to manage the allocation and we can amortize-remove it in `wasmtime_func_call` by using space after the parameters in the `Vec<Val>` we're passing in. This change is naturally a breaking change and we'll want to consider it carefully, but mitigating factors are that most embeddings are likely using `TypedFunc::call` instead and this signature taking a mutable slice better aligns with `Func::new` which receives a mutable slice for the results. Overall this change, in the benchmark of "call a nop function from the C API" is not quite as good as #3298. It's still a bit slower, on the order of 15ns, because there's lots of capacity checks around vectors and the type checks are slightly less optimized than before. Overall though this is still significantly better than today because allocations and the rwlock to acquire the type information are both avoided. I personally feel that this change is the best to do because it has less of an API impact than #3298. * Rebase issues
587 lines
19 KiB
Rust
587 lines
19 KiB
Rust
use anyhow::Error;
|
|
use wasmtime::*;
|
|
|
|
// Crate a synchronous Func, call it directly:
|
|
#[test]
|
|
fn call_wrapped_func() -> Result<(), Error> {
|
|
let mut store = Store::<State>::default();
|
|
store.call_hook(State::call_hook);
|
|
let f = Func::wrap(
|
|
&mut store,
|
|
|caller: Caller<State>, a: i32, b: i64, c: f32, d: f64| {
|
|
// Calling this func will switch context into wasm, then back to host:
|
|
assert_eq!(caller.data().context, vec![Context::Wasm, Context::Host]);
|
|
|
|
assert_eq!(
|
|
caller.data().calls_into_host,
|
|
caller.data().returns_from_host + 1
|
|
);
|
|
assert_eq!(
|
|
caller.data().calls_into_wasm,
|
|
caller.data().returns_from_wasm + 1
|
|
);
|
|
|
|
assert_eq!(a, 1);
|
|
assert_eq!(b, 2);
|
|
assert_eq!(c, 3.0);
|
|
assert_eq!(d, 4.0);
|
|
},
|
|
);
|
|
|
|
f.call(
|
|
&mut store,
|
|
&[Val::I32(1), Val::I64(2), 3.0f32.into(), 4.0f64.into()],
|
|
&mut [],
|
|
)?;
|
|
|
|
// One switch from vm to host to call f, another in return from f.
|
|
assert_eq!(store.data().calls_into_host, 1);
|
|
assert_eq!(store.data().returns_from_host, 1);
|
|
assert_eq!(store.data().calls_into_wasm, 1);
|
|
assert_eq!(store.data().returns_from_wasm, 1);
|
|
|
|
f.typed::<(i32, i64, f32, f64), (), _>(&store)?
|
|
.call(&mut store, (1, 2, 3.0, 4.0))?;
|
|
|
|
assert_eq!(store.data().calls_into_host, 2);
|
|
assert_eq!(store.data().returns_from_host, 2);
|
|
assert_eq!(store.data().calls_into_wasm, 2);
|
|
assert_eq!(store.data().returns_from_wasm, 2);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
// Create an async Func, call it directly:
|
|
#[tokio::test]
|
|
async fn call_wrapped_async_func() -> Result<(), Error> {
|
|
let mut config = Config::new();
|
|
config.async_support(true);
|
|
let engine = Engine::new(&config)?;
|
|
let mut store = Store::new(&engine, State::default());
|
|
store.call_hook(State::call_hook);
|
|
let f = Func::wrap4_async(
|
|
&mut store,
|
|
|caller: Caller<State>, a: i32, b: i64, c: f32, d: f64| {
|
|
Box::new(async move {
|
|
// Calling this func will switch context into wasm, then back to host:
|
|
assert_eq!(caller.data().context, vec![Context::Wasm, Context::Host]);
|
|
|
|
assert_eq!(
|
|
caller.data().calls_into_host,
|
|
caller.data().returns_from_host + 1
|
|
);
|
|
assert_eq!(
|
|
caller.data().calls_into_wasm,
|
|
caller.data().returns_from_wasm + 1
|
|
);
|
|
|
|
assert_eq!(a, 1);
|
|
assert_eq!(b, 2);
|
|
assert_eq!(c, 3.0);
|
|
assert_eq!(d, 4.0);
|
|
})
|
|
},
|
|
);
|
|
|
|
f.call_async(
|
|
&mut store,
|
|
&[Val::I32(1), Val::I64(2), 3.0f32.into(), 4.0f64.into()],
|
|
&mut [],
|
|
)
|
|
.await?;
|
|
|
|
// One switch from vm to host to call f, another in return from f.
|
|
assert_eq!(store.data().calls_into_host, 1);
|
|
assert_eq!(store.data().returns_from_host, 1);
|
|
assert_eq!(store.data().calls_into_wasm, 1);
|
|
assert_eq!(store.data().returns_from_wasm, 1);
|
|
|
|
f.typed::<(i32, i64, f32, f64), (), _>(&store)?
|
|
.call_async(&mut store, (1, 2, 3.0, 4.0))
|
|
.await?;
|
|
|
|
assert_eq!(store.data().calls_into_host, 2);
|
|
assert_eq!(store.data().returns_from_host, 2);
|
|
assert_eq!(store.data().calls_into_wasm, 2);
|
|
assert_eq!(store.data().returns_from_wasm, 2);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
// Use the Linker to define a sync func, call it through WebAssembly:
|
|
#[test]
|
|
fn call_linked_func() -> Result<(), Error> {
|
|
let engine = Engine::default();
|
|
let mut store = Store::new(&engine, State::default());
|
|
store.call_hook(State::call_hook);
|
|
let mut linker = Linker::new(&engine);
|
|
|
|
linker.func_wrap(
|
|
"host",
|
|
"f",
|
|
|caller: Caller<State>, a: i32, b: i64, c: f32, d: f64| {
|
|
// Calling this func will switch context into wasm, then back to host:
|
|
assert_eq!(caller.data().context, vec![Context::Wasm, Context::Host]);
|
|
|
|
assert_eq!(
|
|
caller.data().calls_into_host,
|
|
caller.data().returns_from_host + 1
|
|
);
|
|
assert_eq!(
|
|
caller.data().calls_into_wasm,
|
|
caller.data().returns_from_wasm + 1
|
|
);
|
|
|
|
assert_eq!(a, 1);
|
|
assert_eq!(b, 2);
|
|
assert_eq!(c, 3.0);
|
|
assert_eq!(d, 4.0);
|
|
},
|
|
)?;
|
|
|
|
let wat = r#"
|
|
(module
|
|
(import "host" "f"
|
|
(func $f (param i32) (param i64) (param f32) (param f64)))
|
|
(func (export "export")
|
|
(call $f (i32.const 1) (i64.const 2) (f32.const 3.0) (f64.const 4.0)))
|
|
)
|
|
"#;
|
|
let module = Module::new(&engine, wat)?;
|
|
|
|
let inst = linker.instantiate(&mut store, &module)?;
|
|
let export = inst
|
|
.get_export(&mut store, "export")
|
|
.expect("get export")
|
|
.into_func()
|
|
.expect("export is func");
|
|
|
|
export.call(&mut store, &[], &mut [])?;
|
|
|
|
// One switch from vm to host to call f, another in return from f.
|
|
assert_eq!(store.data().calls_into_host, 1);
|
|
assert_eq!(store.data().returns_from_host, 1);
|
|
assert_eq!(store.data().calls_into_wasm, 1);
|
|
assert_eq!(store.data().returns_from_wasm, 1);
|
|
|
|
export.typed::<(), (), _>(&store)?.call(&mut store, ())?;
|
|
|
|
assert_eq!(store.data().calls_into_host, 2);
|
|
assert_eq!(store.data().returns_from_host, 2);
|
|
assert_eq!(store.data().calls_into_wasm, 2);
|
|
assert_eq!(store.data().returns_from_wasm, 2);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
// Use the Linker to define an async func, call it through WebAssembly:
|
|
#[tokio::test]
|
|
async fn call_linked_func_async() -> Result<(), Error> {
|
|
let mut config = Config::new();
|
|
config.async_support(true);
|
|
let engine = Engine::new(&config)?;
|
|
let mut store = Store::new(&engine, State::default());
|
|
store.call_hook(State::call_hook);
|
|
|
|
let f = Func::wrap4_async(
|
|
&mut store,
|
|
|caller: Caller<State>, a: i32, b: i64, c: f32, d: f64| {
|
|
Box::new(async move {
|
|
// Calling this func will switch context into wasm, then back to host:
|
|
assert_eq!(caller.data().context, vec![Context::Wasm, Context::Host]);
|
|
|
|
assert_eq!(
|
|
caller.data().calls_into_host,
|
|
caller.data().returns_from_host + 1
|
|
);
|
|
assert_eq!(
|
|
caller.data().calls_into_wasm,
|
|
caller.data().returns_from_wasm + 1
|
|
);
|
|
assert_eq!(a, 1);
|
|
assert_eq!(b, 2);
|
|
assert_eq!(c, 3.0);
|
|
assert_eq!(d, 4.0);
|
|
})
|
|
},
|
|
);
|
|
|
|
let mut linker = Linker::new(&engine);
|
|
|
|
linker.define("host", "f", f)?;
|
|
|
|
let wat = r#"
|
|
(module
|
|
(import "host" "f"
|
|
(func $f (param i32) (param i64) (param f32) (param f64)))
|
|
(func (export "export")
|
|
(call $f (i32.const 1) (i64.const 2) (f32.const 3.0) (f64.const 4.0)))
|
|
)
|
|
"#;
|
|
let module = Module::new(&engine, wat)?;
|
|
|
|
let inst = linker.instantiate_async(&mut store, &module).await?;
|
|
let export = inst
|
|
.get_export(&mut store, "export")
|
|
.expect("get export")
|
|
.into_func()
|
|
.expect("export is func");
|
|
|
|
export.call_async(&mut store, &[], &mut []).await?;
|
|
|
|
// One switch from vm to host to call f, another in return from f.
|
|
assert_eq!(store.data().calls_into_host, 1);
|
|
assert_eq!(store.data().returns_from_host, 1);
|
|
assert_eq!(store.data().calls_into_wasm, 1);
|
|
assert_eq!(store.data().returns_from_wasm, 1);
|
|
|
|
export
|
|
.typed::<(), (), _>(&store)?
|
|
.call_async(&mut store, ())
|
|
.await?;
|
|
|
|
assert_eq!(store.data().calls_into_host, 2);
|
|
assert_eq!(store.data().returns_from_host, 2);
|
|
assert_eq!(store.data().calls_into_wasm, 2);
|
|
assert_eq!(store.data().returns_from_wasm, 2);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn instantiate() -> Result<(), Error> {
|
|
let mut store = Store::<State>::default();
|
|
store.call_hook(State::call_hook);
|
|
|
|
let m = Module::new(store.engine(), "(module)")?;
|
|
Instance::new(&mut store, &m, &[])?;
|
|
assert_eq!(store.data().calls_into_wasm, 0);
|
|
assert_eq!(store.data().calls_into_host, 0);
|
|
|
|
let m = Module::new(store.engine(), "(module (func) (start 0))")?;
|
|
Instance::new(&mut store, &m, &[])?;
|
|
assert_eq!(store.data().calls_into_wasm, 1);
|
|
assert_eq!(store.data().calls_into_host, 0);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn instantiate_async() -> Result<(), Error> {
|
|
let mut config = Config::new();
|
|
config.async_support(true);
|
|
let engine = Engine::new(&config)?;
|
|
let mut store = Store::new(&engine, State::default());
|
|
store.call_hook(State::call_hook);
|
|
|
|
let m = Module::new(store.engine(), "(module)")?;
|
|
Instance::new_async(&mut store, &m, &[]).await?;
|
|
assert_eq!(store.data().calls_into_wasm, 0);
|
|
assert_eq!(store.data().calls_into_host, 0);
|
|
|
|
let m = Module::new(store.engine(), "(module (func) (start 0))")?;
|
|
Instance::new_async(&mut store, &m, &[]).await?;
|
|
assert_eq!(store.data().calls_into_wasm, 1);
|
|
assert_eq!(store.data().calls_into_host, 0);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn recursion() -> Result<(), Error> {
|
|
// Make sure call hook behaves reasonably when called recursively
|
|
|
|
let engine = Engine::default();
|
|
let mut store = Store::new(&engine, State::default());
|
|
store.call_hook(State::call_hook);
|
|
let mut linker = Linker::new(&engine);
|
|
|
|
linker.func_wrap("host", "f", |mut caller: Caller<State>, n: i32| {
|
|
assert_eq!(caller.data().context.last(), Some(&Context::Host));
|
|
|
|
assert_eq!(caller.data().calls_into_host, caller.data().calls_into_wasm);
|
|
|
|
// Recurse
|
|
if n > 0 {
|
|
caller
|
|
.get_export("export")
|
|
.expect("caller exports \"export\"")
|
|
.into_func()
|
|
.expect("export is a func")
|
|
.typed::<i32, (), _>(&caller)
|
|
.expect("export typing")
|
|
.call(&mut caller, n - 1)
|
|
.unwrap()
|
|
}
|
|
})?;
|
|
|
|
let wat = r#"
|
|
(module
|
|
(import "host" "f"
|
|
(func $f (param i32)))
|
|
(func (export "export") (param i32)
|
|
(call $f (local.get 0)))
|
|
)
|
|
"#;
|
|
let module = Module::new(&engine, wat)?;
|
|
|
|
let inst = linker.instantiate(&mut store, &module)?;
|
|
let export = inst
|
|
.get_export(&mut store, "export")
|
|
.expect("get export")
|
|
.into_func()
|
|
.expect("export is func");
|
|
|
|
// Recursion depth:
|
|
let n: usize = 10;
|
|
|
|
export.call(&mut store, &[Val::I32(n as i32)], &mut [])?;
|
|
|
|
// Recurse down to 0: n+1 calls
|
|
assert_eq!(store.data().calls_into_host, n + 1);
|
|
assert_eq!(store.data().returns_from_host, n + 1);
|
|
assert_eq!(store.data().calls_into_wasm, n + 1);
|
|
assert_eq!(store.data().returns_from_wasm, n + 1);
|
|
|
|
export
|
|
.typed::<i32, (), _>(&store)?
|
|
.call(&mut store, n as i32)?;
|
|
|
|
assert_eq!(store.data().calls_into_host, 2 * (n + 1));
|
|
assert_eq!(store.data().returns_from_host, 2 * (n + 1));
|
|
assert_eq!(store.data().calls_into_wasm, 2 * (n + 1));
|
|
assert_eq!(store.data().returns_from_wasm, 2 * (n + 1));
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn trapping() -> Result<(), Error> {
|
|
const TRAP_IN_F: i32 = 0;
|
|
const TRAP_NEXT_CALL_HOST: i32 = 1;
|
|
const TRAP_NEXT_RETURN_HOST: i32 = 2;
|
|
const TRAP_NEXT_CALL_WASM: i32 = 3;
|
|
const TRAP_NEXT_RETURN_WASM: i32 = 4;
|
|
|
|
let engine = Engine::default();
|
|
|
|
let mut linker = Linker::new(&engine);
|
|
|
|
linker.func_wrap(
|
|
"host",
|
|
"f",
|
|
|mut caller: Caller<State>, action: i32, recur: i32| -> Result<(), Trap> {
|
|
assert_eq!(caller.data().context.last(), Some(&Context::Host));
|
|
assert_eq!(caller.data().calls_into_host, caller.data().calls_into_wasm);
|
|
|
|
match action {
|
|
TRAP_IN_F => return Err(Trap::new("trapping in f")),
|
|
TRAP_NEXT_CALL_HOST => caller.data_mut().trap_next_call_host = true,
|
|
TRAP_NEXT_RETURN_HOST => caller.data_mut().trap_next_return_host = true,
|
|
TRAP_NEXT_CALL_WASM => caller.data_mut().trap_next_call_wasm = true,
|
|
TRAP_NEXT_RETURN_WASM => caller.data_mut().trap_next_return_wasm = true,
|
|
_ => {} // Do nothing
|
|
}
|
|
|
|
// recur so that we can trigger a next call.
|
|
// propogate its trap, if it traps!
|
|
if recur > 0 {
|
|
let _ = caller
|
|
.get_export("export")
|
|
.expect("caller exports \"export\"")
|
|
.into_func()
|
|
.expect("export is a func")
|
|
.typed::<(i32, i32), (), _>(&caller)
|
|
.expect("export typing")
|
|
.call(&mut caller, (action, 0))?;
|
|
}
|
|
|
|
Ok(())
|
|
},
|
|
)?;
|
|
|
|
let wat = r#"
|
|
(module
|
|
(import "host" "f"
|
|
(func $f (param i32) (param i32)))
|
|
(func (export "export") (param i32) (param i32)
|
|
(call $f (local.get 0) (local.get 1)))
|
|
)
|
|
"#;
|
|
let module = Module::new(&engine, wat)?;
|
|
|
|
let run = |action: i32, recur: bool| -> (State, Option<Error>) {
|
|
let mut store = Store::new(&engine, State::default());
|
|
store.call_hook(State::call_hook);
|
|
let inst = linker
|
|
.instantiate(&mut store, &module)
|
|
.expect("instantiate");
|
|
let export = inst
|
|
.get_export(&mut store, "export")
|
|
.expect("get export")
|
|
.into_func()
|
|
.expect("export is func");
|
|
|
|
let r = export.call(
|
|
&mut store,
|
|
&[Val::I32(action), Val::I32(if recur { 1 } else { 0 })],
|
|
&mut [],
|
|
);
|
|
(store.into_data(), r.err())
|
|
};
|
|
|
|
let (s, e) = run(TRAP_IN_F, false);
|
|
assert!(e.unwrap().to_string().starts_with("trapping in f"));
|
|
assert_eq!(s.calls_into_host, 1);
|
|
assert_eq!(s.returns_from_host, 1);
|
|
assert_eq!(s.calls_into_wasm, 1);
|
|
assert_eq!(s.returns_from_wasm, 1);
|
|
|
|
// trap in next call to host. No calls after the bit is set, so this trap shouldn't happen
|
|
let (s, e) = run(TRAP_NEXT_CALL_HOST, false);
|
|
assert!(e.is_none());
|
|
assert_eq!(s.calls_into_host, 1);
|
|
assert_eq!(s.returns_from_host, 1);
|
|
assert_eq!(s.calls_into_wasm, 1);
|
|
assert_eq!(s.returns_from_wasm, 1);
|
|
|
|
// trap in next call to host. recur, so the second call into host traps:
|
|
let (s, e) = run(TRAP_NEXT_CALL_HOST, true);
|
|
assert!(e
|
|
.unwrap()
|
|
.to_string()
|
|
.starts_with("call_hook: trapping on CallingHost"));
|
|
assert_eq!(s.calls_into_host, 2);
|
|
assert_eq!(s.returns_from_host, 1);
|
|
assert_eq!(s.calls_into_wasm, 2);
|
|
assert_eq!(s.returns_from_wasm, 2);
|
|
|
|
// trap in the return from host. should trap right away, without recursion
|
|
let (s, e) = run(TRAP_NEXT_RETURN_HOST, false);
|
|
assert!(e
|
|
.unwrap()
|
|
.to_string()
|
|
.starts_with("call_hook: trapping on ReturningFromHost"));
|
|
assert_eq!(s.calls_into_host, 1);
|
|
assert_eq!(s.returns_from_host, 1);
|
|
assert_eq!(s.calls_into_wasm, 1);
|
|
assert_eq!(s.returns_from_wasm, 1);
|
|
|
|
// trap in next call to wasm. No calls after the bit is set, so this trap shouldnt happen:
|
|
let (s, e) = run(TRAP_NEXT_CALL_WASM, false);
|
|
assert!(e.is_none());
|
|
assert_eq!(s.calls_into_host, 1);
|
|
assert_eq!(s.returns_from_host, 1);
|
|
assert_eq!(s.calls_into_wasm, 1);
|
|
assert_eq!(s.returns_from_wasm, 1);
|
|
|
|
// trap in next call to wasm. recur, so the second call into wasm traps:
|
|
let (s, e) = run(TRAP_NEXT_CALL_WASM, true);
|
|
assert!(e
|
|
.unwrap()
|
|
.to_string()
|
|
.starts_with("call_hook: trapping on CallingWasm"));
|
|
assert_eq!(s.calls_into_host, 1);
|
|
assert_eq!(s.returns_from_host, 1);
|
|
assert_eq!(s.calls_into_wasm, 2);
|
|
assert_eq!(s.returns_from_wasm, 1);
|
|
|
|
// trap in the return from wasm. should trap right away, without recursion
|
|
let (s, e) = run(TRAP_NEXT_RETURN_WASM, false);
|
|
assert!(e
|
|
.unwrap()
|
|
.to_string()
|
|
.starts_with("call_hook: trapping on ReturningFromWasm"));
|
|
assert_eq!(s.calls_into_host, 1);
|
|
assert_eq!(s.returns_from_host, 1);
|
|
assert_eq!(s.calls_into_wasm, 1);
|
|
assert_eq!(s.returns_from_wasm, 1);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[derive(Debug, PartialEq, Eq)]
|
|
enum Context {
|
|
Host,
|
|
Wasm,
|
|
}
|
|
|
|
struct State {
|
|
context: Vec<Context>,
|
|
|
|
calls_into_host: usize,
|
|
returns_from_host: usize,
|
|
calls_into_wasm: usize,
|
|
returns_from_wasm: usize,
|
|
|
|
trap_next_call_host: bool,
|
|
trap_next_return_host: bool,
|
|
trap_next_call_wasm: bool,
|
|
trap_next_return_wasm: bool,
|
|
}
|
|
|
|
impl Default for State {
|
|
fn default() -> Self {
|
|
State {
|
|
context: Vec::new(),
|
|
calls_into_host: 0,
|
|
returns_from_host: 0,
|
|
calls_into_wasm: 0,
|
|
returns_from_wasm: 0,
|
|
trap_next_call_host: false,
|
|
trap_next_return_host: false,
|
|
trap_next_call_wasm: false,
|
|
trap_next_return_wasm: false,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl State {
|
|
// This implementation asserts that hooks are always called in a stack-like manner.
|
|
fn call_hook(&mut self, s: CallHook) -> Result<(), Trap> {
|
|
match s {
|
|
CallHook::CallingHost => {
|
|
self.calls_into_host += 1;
|
|
if self.trap_next_call_host {
|
|
return Err(Trap::new("call_hook: trapping on CallingHost"));
|
|
} else {
|
|
self.context.push(Context::Host);
|
|
}
|
|
}
|
|
CallHook::ReturningFromHost => match self.context.pop() {
|
|
Some(Context::Host) => {
|
|
self.returns_from_host += 1;
|
|
if self.trap_next_return_host {
|
|
return Err(Trap::new("call_hook: trapping on ReturningFromHost"));
|
|
}
|
|
}
|
|
c => panic!(
|
|
"illegal context: expected Some(Host), got {:?}. remaining: {:?}",
|
|
c, self.context
|
|
),
|
|
},
|
|
CallHook::CallingWasm => {
|
|
self.calls_into_wasm += 1;
|
|
if self.trap_next_call_wasm {
|
|
return Err(Trap::new("call_hook: trapping on CallingWasm"));
|
|
} else {
|
|
self.context.push(Context::Wasm);
|
|
}
|
|
}
|
|
CallHook::ReturningFromWasm => match self.context.pop() {
|
|
Some(Context::Wasm) => {
|
|
self.returns_from_wasm += 1;
|
|
if self.trap_next_return_wasm {
|
|
return Err(Trap::new("call_hook: trapping on ReturningFromWasm"));
|
|
}
|
|
}
|
|
c => panic!(
|
|
"illegal context: expected Some(Wasm), got {:?}. remaining: {:?}",
|
|
c, self.context
|
|
),
|
|
},
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|