de27fbe20f
* Ensure `store` is in the function names * Don't abort the process on `add_fuel` when fuel isn't configured * Allow learning about failure in both `add_fuel` and `fuel_consumed`
64 lines
1.5 KiB
Rust
64 lines
1.5 KiB
Rust
use crate::{wasm_engine_t, wasmtime_error_t};
|
|
use wasmtime::{InterruptHandle, Store};
|
|
|
|
#[repr(C)]
|
|
#[derive(Clone)]
|
|
pub struct wasm_store_t {
|
|
pub(crate) store: Store,
|
|
}
|
|
|
|
wasmtime_c_api_macros::declare_own!(wasm_store_t);
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn wasm_store_new(engine: &wasm_engine_t) -> Box<wasm_store_t> {
|
|
let engine = &engine.engine;
|
|
Box::new(wasm_store_t {
|
|
store: Store::new(&engine),
|
|
})
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn wasmtime_store_gc(store: &wasm_store_t) {
|
|
store.store.gc();
|
|
}
|
|
|
|
#[repr(C)]
|
|
pub struct wasmtime_interrupt_handle_t {
|
|
handle: InterruptHandle,
|
|
}
|
|
|
|
wasmtime_c_api_macros::declare_own!(wasmtime_interrupt_handle_t);
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn wasmtime_interrupt_handle_new(
|
|
store: &wasm_store_t,
|
|
) -> Option<Box<wasmtime_interrupt_handle_t>> {
|
|
Some(Box::new(wasmtime_interrupt_handle_t {
|
|
handle: store.store.interrupt_handle().ok()?,
|
|
}))
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn wasmtime_interrupt_handle_interrupt(handle: &wasmtime_interrupt_handle_t) {
|
|
handle.handle.interrupt();
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn wasmtime_store_add_fuel(
|
|
store: &wasm_store_t,
|
|
fuel: u64,
|
|
) -> Option<Box<wasmtime_error_t>> {
|
|
crate::handle_result(store.store.add_fuel(fuel), |()| {})
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn wasmtime_store_fuel_consumed(store: &wasm_store_t, fuel: &mut u64) -> bool {
|
|
match store.store.fuel_consumed() {
|
|
Some(amt) => {
|
|
*fuel = amt;
|
|
true
|
|
}
|
|
None => false,
|
|
}
|
|
}
|