1 use crate::prelude::*;
2 use crate::runtime::vm::{
3     ExportFunction, SendSyncPtr, StoreBox, VMArrayCallHostFuncContext, VMContext, VMFuncRef,
4     VMFunctionImport, VMOpaqueContext,
5 };
6 use crate::runtime::Uninhabited;
7 use crate::store::{AutoAssertNoGc, StoreData, StoreOpaque, Stored};
8 use crate::type_registry::RegisteredType;
9 use crate::{
10     AsContext, AsContextMut, CallHook, Engine, Extern, FuncType, Instance, Module, ModuleExport,
11     Ref, StoreContext, StoreContextMut, Val, ValRaw, ValType,
12 };
13 use alloc::sync::Arc;
14 use core::ffi::c_void;
15 use core::future::Future;
16 use core::mem::{self, MaybeUninit};
17 use core::num::NonZeroUsize;
18 use core::pin::Pin;
19 use core::ptr::NonNull;
20 use wasmtime_environ::VMSharedTypeIndex;
21 
22 /// A reference to the abstract `nofunc` heap value.
23 ///
24 /// The are no instances of `(ref nofunc)`: it is an uninhabited type.
25 ///
26 /// There is precisely one instance of `(ref null nofunc)`, aka `nullfuncref`:
27 /// the null reference.
28 ///
29 /// This `NoFunc` Rust type's sole purpose is for use with [`Func::wrap`]- and
30 /// [`Func::typed`]-style APIs for statically typing a function as taking or
31 /// returning a `(ref null nofunc)` (aka `Option<NoFunc>`) which is always
32 /// `None`.
33 ///
34 /// # Example
35 ///
36 /// ```
37 /// # use wasmtime::*;
38 /// # fn _foo() -> Result<()> {
39 /// let mut config = Config::new();
40 /// config.wasm_function_references(true);
41 /// let engine = Engine::new(&config)?;
42 ///
43 /// let module = Module::new(
44 ///     &engine,
45 ///     r#"
46 ///         (module
47 ///             (func (export "f") (param (ref null nofunc))
48 ///                 ;; If the reference is null, return.
49 ///                 local.get 0
50 ///                 ref.is_null nofunc
51 ///                 br_if 0
52 ///
53 ///                 ;; If the reference was not null (which is impossible)
54 ///                 ;; then raise a trap.
55 ///                 unreachable
56 ///             )
57 ///         )
58 ///     "#,
59 /// )?;
60 ///
61 /// let mut store = Store::new(&engine, ());
62 /// let instance = Instance::new(&mut store, &module, &[])?;
63 /// let f = instance.get_func(&mut store, "f").unwrap();
64 ///
65 /// // We can cast a `(ref null nofunc)`-taking function into a typed function that
66 /// // takes an `Option<NoFunc>` via the `Func::typed` method.
67 /// let f = f.typed::<Option<NoFunc>, ()>(&store)?;
68 ///
69 /// // We can call the typed function, passing the null `nofunc` reference.
70 /// let result = f.call(&mut store, NoFunc::null());
71 ///
72 /// // The function should not have trapped, because the reference we gave it was
73 /// // null (as it had to be, since `NoFunc` is uninhabited).
74 /// assert!(result.is_ok());
75 /// # Ok(())
76 /// # }
77 /// ```
78 #[derive(Copy, Clone, Debug, PartialEq, Eq)]
79 pub struct NoFunc {
80     _inner: Uninhabited,
81 }
82 
83 impl NoFunc {
84     /// Get the null `(ref null nofunc)` (aka `nullfuncref`) reference.
85     #[inline]
86     pub fn null() -> Option<NoFunc> {
87         None
88     }
89 
90     /// Get the null `(ref null nofunc)` (aka `nullfuncref`) reference as a
91     /// [`Ref`].
92     #[inline]
93     pub fn null_ref() -> Ref {
94         Ref::Func(None)
95     }
96 
97     /// Get the null `(ref null nofunc)` (aka `nullfuncref`) reference as a
98     /// [`Val`].
99     #[inline]
100     pub fn null_val() -> Val {
101         Val::FuncRef(None)
102     }
103 }
104 
105 /// A WebAssembly function which can be called.
106 ///
107 /// This type typically represents an exported function from a WebAssembly
108 /// module instance. In this case a [`Func`] belongs to an [`Instance`] and is
109 /// loaded from there. A [`Func`] may also represent a host function as well in
110 /// some cases, too.
111 ///
112 /// Functions can be called in a few different ways, either synchronous or async
113 /// and either typed or untyped (more on this below). Note that host functions
114 /// are normally inserted directly into a [`Linker`](crate::Linker) rather than
115 /// using this directly, but both options are available.
116 ///
117 /// # `Func` and `async`
118 ///
119 /// Functions from the perspective of WebAssembly are always synchronous. You
120 /// might have an `async` function in Rust, however, which you'd like to make
121 /// available from WebAssembly. Wasmtime supports asynchronously calling
122 /// WebAssembly through native stack switching. You can get some more
123 /// information about [asynchronous configs](crate::Config::async_support), but
124 /// from the perspective of `Func` it's important to know that whether or not
125 /// your [`Store`](crate::Store) is asynchronous will dictate whether you call
126 /// functions through [`Func::call`] or [`Func::call_async`] (or the typed
127 /// wrappers such as [`TypedFunc::call`] vs [`TypedFunc::call_async`]).
128 ///
129 /// # To `Func::call` or to `Func::typed().call()`
130 ///
131 /// There's a 2x2 matrix of methods to call [`Func`]. Invocations can either be
132 /// asynchronous or synchronous. They can also be statically typed or not.
133 /// Whether or not an invocation is asynchronous is indicated via the method
134 /// being `async` and [`call_async`](Func::call_async) being the entry point.
135 /// Otherwise for statically typed or not your options are:
136 ///
137 /// * Dynamically typed - if you don't statically know the signature of the
138 ///   function that you're calling you'll be using [`Func::call`] or
139 ///   [`Func::call_async`]. These functions take a variable-length slice of
140 ///   "boxed" arguments in their [`Val`] representation. Additionally the
141 ///   results are returned as an owned slice of [`Val`]. These methods are not
142 ///   optimized due to the dynamic type checks that must occur, in addition to
143 ///   some dynamic allocations for where to put all the arguments. While this
144 ///   allows you to call all possible wasm function signatures, if you're
145 ///   looking for a speedier alternative you can also use...
146 ///
147 /// * Statically typed - if you statically know the type signature of the wasm
148 ///   function you're calling, then you'll want to use the [`Func::typed`]
149 ///   method to acquire an instance of [`TypedFunc`]. This structure is static proof
150 ///   that the underlying wasm function has the ascripted type, and type
151 ///   validation is only done once up-front. The [`TypedFunc::call`] and
152 ///   [`TypedFunc::call_async`] methods are much more efficient than [`Func::call`]
153 ///   and [`Func::call_async`] because the type signature is statically known.
154 ///   This eschews runtime checks as much as possible to get into wasm as fast
155 ///   as possible.
156 ///
157 /// # Examples
158 ///
159 /// One way to get a `Func` is from an [`Instance`] after you've instantiated
160 /// it:
161 ///
162 /// ```
163 /// # use wasmtime::*;
164 /// # fn main() -> anyhow::Result<()> {
165 /// let engine = Engine::default();
166 /// let module = Module::new(&engine, r#"(module (func (export "foo")))"#)?;
167 /// let mut store = Store::new(&engine, ());
168 /// let instance = Instance::new(&mut store, &module, &[])?;
169 /// let foo = instance.get_func(&mut store, "foo").expect("export wasn't a function");
170 ///
171 /// // Work with `foo` as a `Func` at this point, such as calling it
172 /// // dynamically...
173 /// match foo.call(&mut store, &[], &mut []) {
174 ///     Ok(()) => { /* ... */ }
175 ///     Err(trap) => {
176 ///         panic!("execution of `foo` resulted in a wasm trap: {}", trap);
177 ///     }
178 /// }
179 /// foo.call(&mut store, &[], &mut [])?;
180 ///
181 /// // ... or we can make a static assertion about its signature and call it.
182 /// // Our first call here can fail if the signatures don't match, and then the
183 /// // second call can fail if the function traps (like the `match` above).
184 /// let foo = foo.typed::<(), ()>(&store)?;
185 /// foo.call(&mut store, ())?;
186 /// # Ok(())
187 /// # }
188 /// ```
189 ///
190 /// You can also use the [`wrap` function](Func::wrap) to create a
191 /// `Func`
192 ///
193 /// ```
194 /// # use wasmtime::*;
195 /// # fn main() -> anyhow::Result<()> {
196 /// let mut store = Store::<()>::default();
197 ///
198 /// // Create a custom `Func` which can execute arbitrary code inside of the
199 /// // closure.
200 /// let add = Func::wrap(&mut store, |a: i32, b: i32| -> i32 { a + b });
201 ///
202 /// // Next we can hook that up to a wasm module which uses it.
203 /// let module = Module::new(
204 ///     store.engine(),
205 ///     r#"
206 ///         (module
207 ///             (import "" "" (func $add (param i32 i32) (result i32)))
208 ///             (func (export "call_add_twice") (result i32)
209 ///                 i32.const 1
210 ///                 i32.const 2
211 ///                 call $add
212 ///                 i32.const 3
213 ///                 i32.const 4
214 ///                 call $add
215 ///                 i32.add))
216 ///     "#,
217 /// )?;
218 /// let instance = Instance::new(&mut store, &module, &[add.into()])?;
219 /// let call_add_twice = instance.get_typed_func::<(), i32>(&mut store, "call_add_twice")?;
220 ///
221 /// assert_eq!(call_add_twice.call(&mut store, ())?, 10);
222 /// # Ok(())
223 /// # }
224 /// ```
225 ///
226 /// Or you could also create an entirely dynamic `Func`!
227 ///
228 /// ```
229 /// # use wasmtime::*;
230 /// # fn main() -> anyhow::Result<()> {
231 /// let mut store = Store::<()>::default();
232 ///
233 /// // Here we need to define the type signature of our `Double` function and
234 /// // then wrap it up in a `Func`
235 /// let double_type = wasmtime::FuncType::new(
236 ///     store.engine(),
237 ///     [wasmtime::ValType::I32].iter().cloned(),
238 ///     [wasmtime::ValType::I32].iter().cloned(),
239 /// );
240 /// let double = Func::new(&mut store, double_type, |_, params, results| {
241 ///     let mut value = params[0].unwrap_i32();
242 ///     value *= 2;
243 ///     results[0] = value.into();
244 ///     Ok(())
245 /// });
246 ///
247 /// let module = Module::new(
248 ///     store.engine(),
249 ///     r#"
250 ///         (module
251 ///             (import "" "" (func $double (param i32) (result i32)))
252 ///             (func $start
253 ///                 i32.const 1
254 ///                 call $double
255 ///                 drop)
256 ///             (start $start))
257 ///     "#,
258 /// )?;
259 /// let instance = Instance::new(&mut store, &module, &[double.into()])?;
260 /// // .. work with `instance` if necessary
261 /// # Ok(())
262 /// # }
263 /// ```
264 #[derive(Copy, Clone, Debug)]
265 #[repr(transparent)] // here for the C API
266 pub struct Func(Stored<FuncData>);
267 
268 pub(crate) struct FuncData {
269     kind: FuncKind,
270 
271     // A pointer to the in-store `VMFuncRef` for this function, if
272     // any.
273     //
274     // When a function is passed to Wasm but doesn't have a Wasm-to-native
275     // trampoline, we have to patch it in. But that requires mutating the
276     // `VMFuncRef`, and this function could be shared across
277     // threads. So we instead copy and pin the `VMFuncRef` into
278     // `StoreOpaque::func_refs`, where we can safely patch the field without
279     // worrying about synchronization and we hold a pointer to it here so we can
280     // reuse it rather than re-copy if it is passed to Wasm again.
281     in_store_func_ref: Option<SendSyncPtr<VMFuncRef>>,
282 
283     // This is somewhat expensive to load from the `Engine` and in most
284     // optimized use cases (e.g. `TypedFunc`) it's not actually needed or it's
285     // only needed rarely. To handle that this is an optionally-contained field
286     // which is lazily loaded into as part of `Func::call`.
287     //
288     // Also note that this is intentionally placed behind a pointer to keep it
289     // small as `FuncData` instances are often inserted into a `Store`.
290     ty: Option<Box<FuncType>>,
291 }
292 
293 /// The three ways that a function can be created and referenced from within a
294 /// store.
295 enum FuncKind {
296     /// A function already owned by the store via some other means. This is
297     /// used, for example, when creating a `Func` from an instance's exported
298     /// function. The instance's `InstanceHandle` is already owned by the store
299     /// and we just have some pointers into that which represent how to call the
300     /// function.
301     StoreOwned { export: ExportFunction },
302 
303     /// A function is shared across possibly other stores, hence the `Arc`. This
304     /// variant happens when a `Linker`-defined function is instantiated within
305     /// a `Store` (e.g. via `Linker::get` or similar APIs). The `Arc` here
306     /// indicates that there's some number of other stores holding this function
307     /// too, so dropping this may not deallocate the underlying
308     /// `InstanceHandle`.
309     SharedHost(Arc<HostFunc>),
310 
311     /// A uniquely-owned host function within a `Store`. This comes about with
312     /// `Func::new` or similar APIs. The `HostFunc` internally owns the
313     /// `InstanceHandle` and that will get dropped when this `HostFunc` itself
314     /// is dropped.
315     ///
316     /// Note that this is intentionally placed behind a `Box` to minimize the
317     /// size of this enum since the most common variant for high-performance
318     /// situations is `SharedHost` and `StoreOwned`, so this ideally isn't
319     /// larger than those two.
320     Host(Box<HostFunc>),
321 
322     /// A reference to a `HostFunc`, but one that's "rooted" in the `Store`
323     /// itself.
324     ///
325     /// This variant is created when an `InstancePre<T>` is instantiated in to a
326     /// `Store<T>`. In that situation the `InstancePre<T>` already has a list of
327     /// host functions that are packaged up in an `Arc`, so the `Arc<[T]>` is
328     /// cloned once into the `Store` to avoid each individual function requiring
329     /// an `Arc::clone`.
330     ///
331     /// The lifetime management of this type is `unsafe` because
332     /// `RootedHostFunc` is a small wrapper around `NonNull<HostFunc>`. To be
333     /// safe this is required that the memory of the host function is pinned
334     /// elsewhere (e.g. the `Arc` in the `Store`).
335     RootedHost(RootedHostFunc),
336 }
337 
338 macro_rules! for_each_function_signature {
339     ($mac:ident) => {
340         $mac!(0);
341         $mac!(1 A1);
342         $mac!(2 A1 A2);
343         $mac!(3 A1 A2 A3);
344         $mac!(4 A1 A2 A3 A4);
345         $mac!(5 A1 A2 A3 A4 A5);
346         $mac!(6 A1 A2 A3 A4 A5 A6);
347         $mac!(7 A1 A2 A3 A4 A5 A6 A7);
348         $mac!(8 A1 A2 A3 A4 A5 A6 A7 A8);
349         $mac!(9 A1 A2 A3 A4 A5 A6 A7 A8 A9);
350         $mac!(10 A1 A2 A3 A4 A5 A6 A7 A8 A9 A10);
351         $mac!(11 A1 A2 A3 A4 A5 A6 A7 A8 A9 A10 A11);
352         $mac!(12 A1 A2 A3 A4 A5 A6 A7 A8 A9 A10 A11 A12);
353         $mac!(13 A1 A2 A3 A4 A5 A6 A7 A8 A9 A10 A11 A12 A13);
354         $mac!(14 A1 A2 A3 A4 A5 A6 A7 A8 A9 A10 A11 A12 A13 A14);
355         $mac!(15 A1 A2 A3 A4 A5 A6 A7 A8 A9 A10 A11 A12 A13 A14 A15);
356         $mac!(16 A1 A2 A3 A4 A5 A6 A7 A8 A9 A10 A11 A12 A13 A14 A15 A16);
357         $mac!(17 A1 A2 A3 A4 A5 A6 A7 A8 A9 A10 A11 A12 A13 A14 A15 A16 A17);
358     };
359 }
360 
361 mod typed;
362 pub use typed::*;
363 
364 impl Func {
365     /// Creates a new `Func` with the given arguments, typically to create a
366     /// host-defined function to pass as an import to a module.
367     ///
368     /// * `store` - the store in which to create this [`Func`], which will own
369     ///   the return value.
370     ///
371     /// * `ty` - the signature of this function, used to indicate what the
372     ///   inputs and outputs are.
373     ///
374     /// * `func` - the native code invoked whenever this `Func` will be called.
375     ///   This closure is provided a [`Caller`] as its first argument to learn
376     ///   information about the caller, and then it's passed a list of
377     ///   parameters as a slice along with a mutable slice of where to write
378     ///   results.
379     ///
380     /// Note that the implementation of `func` must adhere to the `ty` signature
381     /// given, error or traps may occur if it does not respect the `ty`
382     /// signature. For example if the function type declares that it returns one
383     /// i32 but the `func` closures does not write anything into the results
384     /// slice then a trap may be generated.
385     ///
386     /// Additionally note that this is quite a dynamic function since signatures
387     /// are not statically known. For a more performant and ergonomic `Func`
388     /// it's recommended to use [`Func::wrap`] if you can because with
389     /// statically known signatures Wasmtime can optimize the implementation
390     /// much more.
391     ///
392     /// For more information about `Send + Sync + 'static` requirements on the
393     /// `func`, see [`Func::wrap`](#why-send--sync--static).
394     ///
395     /// # Errors
396     ///
397     /// The host-provided function here returns a
398     /// [`Result<()>`](anyhow::Result). If the function returns `Ok(())` then
399     /// that indicates that the host function completed successfully and wrote
400     /// the result into the `&mut [Val]` argument.
401     ///
402     /// If the function returns `Err(e)`, however, then this is equivalent to
403     /// the host function triggering a trap for wasm. WebAssembly execution is
404     /// immediately halted and the original caller of [`Func::call`], for
405     /// example, will receive the error returned here (possibly with
406     /// [`WasmBacktrace`](crate::WasmBacktrace) context information attached).
407     ///
408     /// For more information about errors in Wasmtime see the [`Trap`]
409     /// documentation.
410     ///
411     /// [`Trap`]: crate::Trap
412     ///
413     /// # Panics
414     ///
415     /// Panics if the given function type is not associated with this store's
416     /// engine.
417     pub fn new<T>(
418         store: impl AsContextMut<Data = T>,
419         ty: FuncType,
420         func: impl Fn(Caller<'_, T>, &[Val], &mut [Val]) -> Result<()> + Send + Sync + 'static,
421     ) -> Self {
422         assert!(ty.comes_from_same_engine(store.as_context().engine()));
423         let ty_clone = ty.clone();
424         unsafe {
425             Func::new_unchecked(store, ty, move |caller, values| {
426                 Func::invoke_host_func_for_wasm(caller, &ty_clone, values, &func)
427             })
428         }
429     }
430 
431     /// Creates a new [`Func`] with the given arguments, although has fewer
432     /// runtime checks than [`Func::new`].
433     ///
434     /// This function takes a callback of a different signature than
435     /// [`Func::new`], instead receiving a raw pointer with a list of [`ValRaw`]
436     /// structures. These values have no type information associated with them
437     /// so it's up to the caller to provide a function that will correctly
438     /// interpret the list of values as those coming from the `ty` specified.
439     ///
440     /// If you're calling this from Rust it's recommended to either instead use
441     /// [`Func::new`] or [`Func::wrap`]. The [`Func::wrap`] API, in particular,
442     /// is both safer and faster than this API.
443     ///
444     /// # Errors
445     ///
446     /// See [`Func::new`] for the behavior of returning an error from the host
447     /// function provided here.
448     ///
449     /// # Unsafety
450     ///
451     /// This function is not safe because it's not known at compile time that
452     /// the `func` provided correctly interprets the argument types provided to
453     /// it, or that the results it produces will be of the correct type.
454     ///
455     /// # Panics
456     ///
457     /// Panics if the given function type is not associated with this store's
458     /// engine.
459     pub unsafe fn new_unchecked<T>(
460         mut store: impl AsContextMut<Data = T>,
461         ty: FuncType,
462         func: impl Fn(Caller<'_, T>, &mut [ValRaw]) -> Result<()> + Send + Sync + 'static,
463     ) -> Self {
464         assert!(ty.comes_from_same_engine(store.as_context().engine()));
465         let store = store.as_context_mut().0;
466         let host = HostFunc::new_unchecked(store.engine(), ty, func);
467         host.into_func(store)
468     }
469 
470     /// Creates a new host-defined WebAssembly function which, when called,
471     /// will run the asynchronous computation defined by `func` to completion
472     /// and then return the result to WebAssembly.
473     ///
474     /// This function is the asynchronous analogue of [`Func::new`] and much of
475     /// that documentation applies to this as well. The key difference is that
476     /// `func` returns a future instead of simply a `Result`. Note that the
477     /// returned future can close over any of the arguments, but it cannot close
478     /// over the state of the closure itself. It's recommended to store any
479     /// necessary async state in the `T` of the [`Store<T>`](crate::Store) which
480     /// can be accessed through [`Caller::data`] or [`Caller::data_mut`].
481     ///
482     /// For more information on `Send + Sync + 'static`, see
483     /// [`Func::wrap`](#why-send--sync--static).
484     ///
485     /// # Panics
486     ///
487     /// This function will panic if `store` is not associated with an [async
488     /// config](crate::Config::async_support).
489     ///
490     /// Panics if the given function type is not associated with this store's
491     /// engine.
492     ///
493     /// # Errors
494     ///
495     /// See [`Func::new`] for the behavior of returning an error from the host
496     /// function provided here.
497     ///
498     /// # Examples
499     ///
500     /// ```
501     /// # use wasmtime::*;
502     /// # fn main() -> anyhow::Result<()> {
503     /// // Simulate some application-specific state as well as asynchronous
504     /// // functions to query that state.
505     /// struct MyDatabase {
506     ///     // ...
507     /// }
508     ///
509     /// impl MyDatabase {
510     ///     async fn get_row_count(&self) -> u32 {
511     ///         // ...
512     /// #       100
513     ///     }
514     /// }
515     ///
516     /// let my_database = MyDatabase {
517     ///     // ...
518     /// };
519     ///
520     /// // Using `new_async` we can hook up into calling our async
521     /// // `get_row_count` function.
522     /// let engine = Engine::new(Config::new().async_support(true))?;
523     /// let mut store = Store::new(&engine, MyDatabase {
524     ///     // ...
525     /// });
526     /// let get_row_count_type = wasmtime::FuncType::new(
527     ///     &engine,
528     ///     None,
529     ///     Some(wasmtime::ValType::I32),
530     /// );
531     /// let get = Func::new_async(&mut store, get_row_count_type, |caller, _params, results| {
532     ///     Box::new(async move {
533     ///         let count = caller.data().get_row_count().await;
534     ///         results[0] = Val::I32(count as i32);
535     ///         Ok(())
536     ///     })
537     /// });
538     /// // ...
539     /// # Ok(())
540     /// # }
541     /// ```
542     #[cfg(all(feature = "async", feature = "cranelift"))]
543     pub fn new_async<T, F>(store: impl AsContextMut<Data = T>, ty: FuncType, func: F) -> Func
544     where
545         F: for<'a> Fn(
546                 Caller<'a, T>,
547                 &'a [Val],
548                 &'a mut [Val],
549             ) -> Box<dyn Future<Output = Result<()>> + Send + 'a>
550             + Send
551             + Sync
552             + 'static,
553     {
554         assert!(
555             store.as_context().async_support(),
556             "cannot use `new_async` without enabling async support in the config"
557         );
558         assert!(ty.comes_from_same_engine(store.as_context().engine()));
559         Func::new(store, ty, move |mut caller, params, results| {
560             let async_cx = caller
561                 .store
562                 .as_context_mut()
563                 .0
564                 .async_cx()
565                 .expect("Attempt to spawn new action on dying fiber");
566             let mut future = Pin::from(func(caller, params, results));
567             match unsafe { async_cx.block_on(future.as_mut()) } {
568                 Ok(Ok(())) => Ok(()),
569                 Ok(Err(trap)) | Err(trap) => Err(trap),
570             }
571         })
572     }
573 
574     pub(crate) unsafe fn from_vm_func_ref(
575         store: &mut StoreOpaque,
576         func_ref: NonNull<VMFuncRef>,
577     ) -> Func {
578         debug_assert!(func_ref.as_ref().type_index != VMSharedTypeIndex::default());
579         let export = ExportFunction { func_ref };
580         Func::from_wasmtime_function(export, store)
581     }
582 
583     /// Creates a new `Func` from the given Rust closure.
584     ///
585     /// This function will create a new `Func` which, when called, will
586     /// execute the given Rust closure. Unlike [`Func::new`] the target
587     /// function being called is known statically so the type signature can
588     /// be inferred. Rust types will map to WebAssembly types as follows:
589     ///
590     /// | Rust Argument Type                | WebAssembly Type                          |
591     /// |-----------------------------------|-------------------------------------------|
592     /// | `i32`                             | `i32`                                     |
593     /// | `u32`                             | `i32`                                     |
594     /// | `i64`                             | `i64`                                     |
595     /// | `u64`                             | `i64`                                     |
596     /// | `f32`                             | `f32`                                     |
597     /// | `f64`                             | `f64`                                     |
598     /// | `V128` on x86-64 and aarch64 only | `v128`                                    |
599     /// | `Option<Func>`                    | `funcref` aka `(ref null func)`           |
600     /// | `Func`                            | `(ref func)`                              |
601     /// | `Option<Nofunc>`                  | `nullfuncref` aka `(ref null nofunc)`     |
602     /// | `NoFunc`                          | `(ref nofunc)`                            |
603     /// | `Option<Rooted<ExternRef>>`       | `externref` aka `(ref null extern)`       |
604     /// | `Rooted<ExternRef>`               | `(ref extern)`                            |
605     /// | `Option<NoExtern>`                | `nullexternref` aka `(ref null noextern)` |
606     /// | `NoExtern`                        | `(ref noextern)`                          |
607     /// | `Option<Rooted<AnyRef>>`          | `anyref` aka `(ref null any)`             |
608     /// | `Rooted<AnyRef>`                  | `(ref any)`                               |
609     /// | `Option<Rooted<EqRef>>`           | `eqref` aka `(ref null eq)`               |
610     /// | `Rooted<EqRef>`                   | `(ref eq)`                                |
611     /// | `Option<I31>`                     | `i31ref` aka `(ref null i31)`             |
612     /// | `I31`                             | `(ref i31)`                               |
613     /// | `Option<Rooted<StructRef>>`       | `(ref null struct)`                       |
614     /// | `Rooted<StructRef>`               | `(ref struct)`                            |
615     /// | `Option<Rooted<ArrayRef>>`        | `(ref null array)`                        |
616     /// | `Rooted<ArrayRef>`                | `(ref array)`                             |
617     /// | `Option<NoneRef>`                 | `nullref` aka `(ref null none)`           |
618     /// | `NoneRef`                         | `(ref none)`                              |
619     ///
620     /// Note that anywhere a `Rooted<T>` appears, a `ManuallyRooted<T>` may also
621     /// be used.
622     ///
623     /// Any of the Rust types can be returned from the closure as well, in
624     /// addition to some extra types
625     ///
626     /// | Rust Return Type  | WebAssembly Return Type | Meaning               |
627     /// |-------------------|-------------------------|-----------------------|
628     /// | `()`              | nothing                 | no return value       |
629     /// | `T`               | `T`                     | a single return value |
630     /// | `(T1, T2, ...)`   | `T1 T2 ...`             | multiple returns      |
631     ///
632     /// Note that all return types can also be wrapped in `Result<_>` to
633     /// indicate that the host function can generate a trap as well as possibly
634     /// returning a value.
635     ///
636     /// Finally you can also optionally take [`Caller`] as the first argument of
637     /// your closure. If inserted then you're able to inspect the caller's
638     /// state, for example the [`Memory`](crate::Memory) it has exported so you
639     /// can read what pointers point to.
640     ///
641     /// Note that when using this API, the intention is to create as thin of a
642     /// layer as possible for when WebAssembly calls the function provided. With
643     /// sufficient inlining and optimization the WebAssembly will call straight
644     /// into `func` provided, with no extra fluff entailed.
645     ///
646     /// # Why `Send + Sync + 'static`?
647     ///
648     /// All host functions defined in a [`Store`](crate::Store) (including
649     /// those from [`Func::new`] and other constructors) require that the
650     /// `func` provided is `Send + Sync + 'static`. Additionally host functions
651     /// always are `Fn` as opposed to `FnMut` or `FnOnce`. This can at-a-glance
652     /// feel restrictive since the closure cannot close over as many types as
653     /// before. The reason for this, though, is to ensure that
654     /// [`Store<T>`](crate::Store) can implement both the `Send` and `Sync`
655     /// traits.
656     ///
657     /// Fear not, however, because this isn't as restrictive as it seems! Host
658     /// functions are provided a [`Caller<'_, T>`](crate::Caller) argument which
659     /// allows access to the host-defined data within the
660     /// [`Store`](crate::Store). The `T` type is not required to be any of
661     /// `Send`, `Sync`, or `'static`! This means that you can store whatever
662     /// you'd like in `T` and have it accessible by all host functions.
663     /// Additionally mutable access to `T` is allowed through
664     /// [`Caller::data_mut`].
665     ///
666     /// Most host-defined [`Func`] values provide closures that end up not
667     /// actually closing over any values. These zero-sized types will use the
668     /// context from [`Caller`] for host-defined information.
669     ///
670     /// # Errors
671     ///
672     /// The closure provided here to `wrap` can optionally return a
673     /// [`Result<T>`](anyhow::Result). Returning `Ok(t)` represents the host
674     /// function successfully completing with the `t` result. Returning
675     /// `Err(e)`, however, is equivalent to raising a custom wasm trap.
676     /// Execution of WebAssembly does not resume and the stack is unwound to the
677     /// original caller of the function where the error is returned.
678     ///
679     /// For more information about errors in Wasmtime see the [`Trap`]
680     /// documentation.
681     ///
682     /// [`Trap`]: crate::Trap
683     ///
684     /// # Examples
685     ///
686     /// First up we can see how simple wasm imports can be implemented, such
687     /// as a function that adds its two arguments and returns the result.
688     ///
689     /// ```
690     /// # use wasmtime::*;
691     /// # fn main() -> anyhow::Result<()> {
692     /// # let mut store = Store::<()>::default();
693     /// let add = Func::wrap(&mut store, |a: i32, b: i32| a + b);
694     /// let module = Module::new(
695     ///     store.engine(),
696     ///     r#"
697     ///         (module
698     ///             (import "" "" (func $add (param i32 i32) (result i32)))
699     ///             (func (export "foo") (param i32 i32) (result i32)
700     ///                 local.get 0
701     ///                 local.get 1
702     ///                 call $add))
703     ///     "#,
704     /// )?;
705     /// let instance = Instance::new(&mut store, &module, &[add.into()])?;
706     /// let foo = instance.get_typed_func::<(i32, i32), i32>(&mut store, "foo")?;
707     /// assert_eq!(foo.call(&mut store, (1, 2))?, 3);
708     /// # Ok(())
709     /// # }
710     /// ```
711     ///
712     /// We can also do the same thing, but generate a trap if the addition
713     /// overflows:
714     ///
715     /// ```
716     /// # use wasmtime::*;
717     /// # fn main() -> anyhow::Result<()> {
718     /// # let mut store = Store::<()>::default();
719     /// let add = Func::wrap(&mut store, |a: i32, b: i32| {
720     ///     match a.checked_add(b) {
721     ///         Some(i) => Ok(i),
722     ///         None => anyhow::bail!("overflow"),
723     ///     }
724     /// });
725     /// let module = Module::new(
726     ///     store.engine(),
727     ///     r#"
728     ///         (module
729     ///             (import "" "" (func $add (param i32 i32) (result i32)))
730     ///             (func (export "foo") (param i32 i32) (result i32)
731     ///                 local.get 0
732     ///                 local.get 1
733     ///                 call $add))
734     ///     "#,
735     /// )?;
736     /// let instance = Instance::new(&mut store, &module, &[add.into()])?;
737     /// let foo = instance.get_typed_func::<(i32, i32), i32>(&mut store, "foo")?;
738     /// assert_eq!(foo.call(&mut store, (1, 2))?, 3);
739     /// assert!(foo.call(&mut store, (i32::max_value(), 1)).is_err());
740     /// # Ok(())
741     /// # }
742     /// ```
743     ///
744     /// And don't forget all the wasm types are supported!
745     ///
746     /// ```
747     /// # use wasmtime::*;
748     /// # fn main() -> anyhow::Result<()> {
749     /// # let mut store = Store::<()>::default();
750     /// let debug = Func::wrap(&mut store, |a: i32, b: u32, c: f32, d: i64, e: u64, f: f64| {
751     ///
752     ///     println!("a={}", a);
753     ///     println!("b={}", b);
754     ///     println!("c={}", c);
755     ///     println!("d={}", d);
756     ///     println!("e={}", e);
757     ///     println!("f={}", f);
758     /// });
759     /// let module = Module::new(
760     ///     store.engine(),
761     ///     r#"
762     ///         (module
763     ///             (import "" "" (func $debug (param i32 i32 f32 i64 i64 f64)))
764     ///             (func (export "foo")
765     ///                 i32.const -1
766     ///                 i32.const 1
767     ///                 f32.const 2
768     ///                 i64.const -3
769     ///                 i64.const 3
770     ///                 f64.const 4
771     ///                 call $debug))
772     ///     "#,
773     /// )?;
774     /// let instance = Instance::new(&mut store, &module, &[debug.into()])?;
775     /// let foo = instance.get_typed_func::<(), ()>(&mut store, "foo")?;
776     /// foo.call(&mut store, ())?;
777     /// # Ok(())
778     /// # }
779     /// ```
780     ///
781     /// Finally if you want to get really fancy you can also implement
782     /// imports that read/write wasm module's memory
783     ///
784     /// ```
785     /// use std::str;
786     ///
787     /// # use wasmtime::*;
788     /// # fn main() -> anyhow::Result<()> {
789     /// # let mut store = Store::default();
790     /// let log_str = Func::wrap(&mut store, |mut caller: Caller<'_, ()>, ptr: i32, len: i32| {
791     ///     let mem = match caller.get_export("memory") {
792     ///         Some(Extern::Memory(mem)) => mem,
793     ///         _ => anyhow::bail!("failed to find host memory"),
794     ///     };
795     ///     let data = mem.data(&caller)
796     ///         .get(ptr as u32 as usize..)
797     ///         .and_then(|arr| arr.get(..len as u32 as usize));
798     ///     let string = match data {
799     ///         Some(data) => match str::from_utf8(data) {
800     ///             Ok(s) => s,
801     ///             Err(_) => anyhow::bail!("invalid utf-8"),
802     ///         },
803     ///         None => anyhow::bail!("pointer/length out of bounds"),
804     ///     };
805     ///     assert_eq!(string, "Hello, world!");
806     ///     println!("{}", string);
807     ///     Ok(())
808     /// });
809     /// let module = Module::new(
810     ///     store.engine(),
811     ///     r#"
812     ///         (module
813     ///             (import "" "" (func $log_str (param i32 i32)))
814     ///             (func (export "foo")
815     ///                 i32.const 4   ;; ptr
816     ///                 i32.const 13  ;; len
817     ///                 call $log_str)
818     ///             (memory (export "memory") 1)
819     ///             (data (i32.const 4) "Hello, world!"))
820     ///     "#,
821     /// )?;
822     /// let instance = Instance::new(&mut store, &module, &[log_str.into()])?;
823     /// let foo = instance.get_typed_func::<(), ()>(&mut store, "foo")?;
824     /// foo.call(&mut store, ())?;
825     /// # Ok(())
826     /// # }
827     /// ```
828     pub fn wrap<T, Params, Results>(
829         mut store: impl AsContextMut<Data = T>,
830         func: impl IntoFunc<T, Params, Results>,
831     ) -> Func {
832         let store = store.as_context_mut().0;
833         // part of this unsafety is about matching the `T` to a `Store<T>`,
834         // which is done through the `AsContextMut` bound above.
835         unsafe {
836             let host = HostFunc::wrap(store.engine(), func);
837             host.into_func(store)
838         }
839     }
840 
841     fn wrap_inner<F, T, Params, Results>(mut store: impl AsContextMut<Data = T>, func: F) -> Func
842     where
843         F: Fn(Caller<'_, T>, Params) -> Results + Send + Sync + 'static,
844         Params: WasmTyList,
845         Results: WasmRet,
846     {
847         let store = store.as_context_mut().0;
848         // part of this unsafety is about matching the `T` to a `Store<T>`,
849         // which is done through the `AsContextMut` bound above.
850         unsafe {
851             let host = HostFunc::wrap_inner(store.engine(), func);
852             host.into_func(store)
853         }
854     }
855 
856     /// Same as [`Func::wrap`], except the closure asynchronously produces the
857     /// result and the arguments are passed within a tuple. For more information
858     /// see the [`Func`] documentation.
859     ///
860     /// # Panics
861     ///
862     /// This function will panic if called with a non-asynchronous store.
863     #[cfg(feature = "async")]
864     pub fn wrap_async<T, F, P, R>(store: impl AsContextMut<Data = T>, func: F) -> Func
865     where
866         F: for<'a> Fn(Caller<'a, T>, P) -> Box<dyn Future<Output = R> + Send + 'a>
867             + Send
868             + Sync
869             + 'static,
870         P: WasmTyList,
871         R: WasmRet,
872     {
873         assert!(
874             store.as_context().async_support(),
875             concat!("cannot use `wrap_async` without enabling async support on the config")
876         );
877         Func::wrap_inner(store, move |mut caller: Caller<'_, T>, args| {
878             let async_cx = caller
879                 .store
880                 .as_context_mut()
881                 .0
882                 .async_cx()
883                 .expect("Attempt to start async function on dying fiber");
884             let mut future = Pin::from(func(caller, args));
885 
886             match unsafe { async_cx.block_on(future.as_mut()) } {
887                 Ok(ret) => ret.into_fallible(),
888                 Err(e) => R::fallible_from_error(e),
889             }
890         })
891     }
892 
893     /// Returns the underlying wasm type that this `Func` has.
894     ///
895     /// # Panics
896     ///
897     /// Panics if `store` does not own this function.
898     pub fn ty(&self, store: impl AsContext) -> FuncType {
899         self.load_ty(&store.as_context().0)
900     }
901 
902     /// Forcibly loads the type of this function from the `Engine`.
903     ///
904     /// Note that this is a somewhat expensive method since it requires taking a
905     /// lock as well as cloning a type.
906     pub(crate) fn load_ty(&self, store: &StoreOpaque) -> FuncType {
907         assert!(self.comes_from_same_store(store));
908         FuncType::from_shared_type_index(store.engine(), self.type_index(store.store_data()))
909     }
910 
911     /// Does this function match the given type?
912     ///
913     /// That is, is this function's type a subtype of the given type?
914     ///
915     /// # Panics
916     ///
917     /// Panics if this function is not associated with the given store or if the
918     /// function type is not associated with the store's engine.
919     pub fn matches_ty(&self, store: impl AsContext, func_ty: &FuncType) -> bool {
920         self._matches_ty(store.as_context().0, func_ty)
921     }
922 
923     pub(crate) fn _matches_ty(&self, store: &StoreOpaque, func_ty: &FuncType) -> bool {
924         let actual_ty = self.load_ty(store);
925         actual_ty.matches(func_ty)
926     }
927 
928     pub(crate) fn ensure_matches_ty(&self, store: &StoreOpaque, func_ty: &FuncType) -> Result<()> {
929         if !self.comes_from_same_store(store) {
930             bail!("function used with wrong store");
931         }
932         if self._matches_ty(store, func_ty) {
933             Ok(())
934         } else {
935             let actual_ty = self.load_ty(store);
936             bail!("type mismatch: expected {func_ty}, found {actual_ty}")
937         }
938     }
939 
940     /// Gets a reference to the `FuncType` for this function.
941     ///
942     /// Note that this returns both a reference to the type of this function as
943     /// well as a reference back to the store itself. This enables using the
944     /// `StoreOpaque` while the `FuncType` is also being used (from the
945     /// perspective of the borrow-checker) because otherwise the signature would
946     /// consider `StoreOpaque` borrowed mutable while `FuncType` is in use.
947     fn ty_ref<'a>(&self, store: &'a mut StoreOpaque) -> (&'a FuncType, &'a StoreOpaque) {
948         // If we haven't loaded our type into the store yet then do so lazily at
949         // this time.
950         if store.store_data()[self.0].ty.is_none() {
951             let ty = self.load_ty(store);
952             store.store_data_mut()[self.0].ty = Some(Box::new(ty));
953         }
954 
955         (store.store_data()[self.0].ty.as_ref().unwrap(), store)
956     }
957 
958     pub(crate) fn type_index(&self, data: &StoreData) -> VMSharedTypeIndex {
959         data[self.0].sig_index()
960     }
961 
962     /// Invokes this function with the `params` given and writes returned values
963     /// to `results`.
964     ///
965     /// The `params` here must match the type signature of this `Func`, or an
966     /// error will occur. Additionally `results` must have the same
967     /// length as the number of results for this function. Calling this function
968     /// will synchronously execute the WebAssembly function referenced to get
969     /// the results.
970     ///
971     /// This function will return `Ok(())` if execution completed without a trap
972     /// or error of any kind. In this situation the results will be written to
973     /// the provided `results` array.
974     ///
975     /// # Errors
976     ///
977     /// Any error which occurs throughout the execution of the function will be
978     /// returned as `Err(e)`. The [`Error`](anyhow::Error) type can be inspected
979     /// for the precise error cause such as:
980     ///
981     /// * [`Trap`] - indicates that a wasm trap happened and execution was
982     ///   halted.
983     /// * [`WasmBacktrace`] - optionally included on errors for backtrace
984     ///   information of the trap/error.
985     /// * Other string-based errors to indicate issues such as type errors with
986     ///   `params`.
987     /// * Any host-originating error originally returned from a function defined
988     ///   via [`Func::new`], for example.
989     ///
990     /// Errors typically indicate that execution of WebAssembly was halted
991     /// mid-way and did not complete after the error condition happened.
992     ///
993     /// [`Trap`]: crate::Trap
994     ///
995     /// # Panics
996     ///
997     /// This function will panic if called on a function belonging to an async
998     /// store. Asynchronous stores must always use `call_async`. Also panics if
999     /// `store` does not own this function.
1000     ///
1001     /// [`WasmBacktrace`]: crate::WasmBacktrace
1002     pub fn call(
1003         &self,
1004         mut store: impl AsContextMut,
1005         params: &[Val],
1006         results: &mut [Val],
1007     ) -> Result<()> {
1008         assert!(
1009             !store.as_context().async_support(),
1010             "must use `call_async` when async support is enabled on the config",
1011         );
1012         let mut store = store.as_context_mut();
1013         let need_gc = self.call_impl_check_args(&mut store, params, results)?;
1014         if need_gc {
1015             store.0.gc();
1016         }
1017         unsafe { self.call_impl_do_call(&mut store, params, results) }
1018     }
1019 
1020     /// Invokes this function in an "unchecked" fashion, reading parameters and
1021     /// writing results to `params_and_returns`.
1022     ///
1023     /// This function is the same as [`Func::call`] except that the arguments
1024     /// and results both use a different representation. If possible it's
1025     /// recommended to use [`Func::call`] if safety isn't necessary or to use
1026     /// [`Func::typed`] in conjunction with [`TypedFunc::call`] since that's
1027     /// both safer and faster than this method of invoking a function.
1028     ///
1029     /// Note that if this function takes `externref` arguments then it will
1030     /// **not** automatically GC unlike the [`Func::call`] and
1031     /// [`TypedFunc::call`] functions. This means that if this function is
1032     /// invoked many times with new `ExternRef` values and no other GC happens
1033     /// via any other means then no values will get collected.
1034     ///
1035     /// # Errors
1036     ///
1037     /// For more information about errors see the [`Func::call`] documentation.
1038     ///
1039     /// # Unsafety
1040     ///
1041     /// This function is unsafe because the `params_and_returns` argument is not
1042     /// validated at all. It must uphold invariants such as:
1043     ///
1044     /// * It's a valid pointer to an array
1045     /// * It has enough space to store all parameters
1046     /// * It has enough space to store all results (not at the same time as
1047     ///   parameters)
1048     /// * Parameters are initially written to the array and have the correct
1049     ///   types and such.
1050     /// * Reference types like `externref` and `funcref` are valid at the
1051     ///   time of this call and for the `store` specified.
1052     ///
1053     /// These invariants are all upheld for you with [`Func::call`] and
1054     /// [`TypedFunc::call`].
1055     pub unsafe fn call_unchecked(
1056         &self,
1057         mut store: impl AsContextMut,
1058         params_and_returns: *mut [ValRaw],
1059     ) -> Result<()> {
1060         let mut store = store.as_context_mut();
1061         let data = &store.0.store_data()[self.0];
1062         let func_ref = data.export().func_ref;
1063         Self::call_unchecked_raw(&mut store, func_ref, params_and_returns)
1064     }
1065 
1066     pub(crate) unsafe fn call_unchecked_raw<T>(
1067         store: &mut StoreContextMut<'_, T>,
1068         func_ref: NonNull<VMFuncRef>,
1069         params_and_returns: *mut [ValRaw],
1070     ) -> Result<()> {
1071         invoke_wasm_and_catch_traps(store, |caller| {
1072             func_ref
1073                 .as_ref()
1074                 .array_call(caller.cast::<VMOpaqueContext>(), params_and_returns)
1075         })
1076     }
1077 
1078     /// Converts the raw representation of a `funcref` into an `Option<Func>`
1079     ///
1080     /// This is intended to be used in conjunction with [`Func::new_unchecked`],
1081     /// [`Func::call_unchecked`], and [`ValRaw`] with its `funcref` field.
1082     ///
1083     /// # Unsafety
1084     ///
1085     /// This function is not safe because `raw` is not validated at all. The
1086     /// caller must guarantee that `raw` is owned by the `store` provided and is
1087     /// valid within the `store`.
1088     pub unsafe fn from_raw(mut store: impl AsContextMut, raw: *mut c_void) -> Option<Func> {
1089         Self::_from_raw(store.as_context_mut().0, raw)
1090     }
1091 
1092     pub(crate) unsafe fn _from_raw(store: &mut StoreOpaque, raw: *mut c_void) -> Option<Func> {
1093         Some(Func::from_vm_func_ref(store, NonNull::new(raw.cast())?))
1094     }
1095 
1096     /// Extracts the raw value of this `Func`, which is owned by `store`.
1097     ///
1098     /// This function returns a value that's suitable for writing into the
1099     /// `funcref` field of the [`ValRaw`] structure.
1100     ///
1101     /// # Unsafety
1102     ///
1103     /// The returned value is only valid for as long as the store is alive and
1104     /// this function is properly rooted within it. Additionally this function
1105     /// should not be liberally used since it's a very low-level knob.
1106     pub unsafe fn to_raw(&self, mut store: impl AsContextMut) -> *mut c_void {
1107         self.vm_func_ref(store.as_context_mut().0).as_ptr().cast()
1108     }
1109 
1110     /// Invokes this function with the `params` given, returning the results
1111     /// asynchronously.
1112     ///
1113     /// This function is the same as [`Func::call`] except that it is
1114     /// asynchronous. This is only compatible with stores associated with an
1115     /// [asynchronous config](crate::Config::async_support).
1116     ///
1117     /// It's important to note that the execution of WebAssembly will happen
1118     /// synchronously in the `poll` method of the future returned from this
1119     /// function. Wasmtime does not manage its own thread pool or similar to
1120     /// execute WebAssembly in. Future `poll` methods are generally expected to
1121     /// resolve quickly, so it's recommended that you run or poll this future
1122     /// in a "blocking context".
1123     ///
1124     /// For more information see the documentation on [asynchronous
1125     /// configs](crate::Config::async_support).
1126     ///
1127     /// # Errors
1128     ///
1129     /// For more information on errors see the [`Func::call`] documentation.
1130     ///
1131     /// # Panics
1132     ///
1133     /// Panics if this is called on a function in a synchronous store. This
1134     /// only works with functions defined within an asynchronous store. Also
1135     /// panics if `store` does not own this function.
1136     #[cfg(feature = "async")]
1137     pub async fn call_async<T>(
1138         &self,
1139         mut store: impl AsContextMut<Data = T>,
1140         params: &[Val],
1141         results: &mut [Val],
1142     ) -> Result<()>
1143     where
1144         T: Send,
1145     {
1146         let mut store = store.as_context_mut();
1147         assert!(
1148             store.0.async_support(),
1149             "cannot use `call_async` without enabling async support in the config",
1150         );
1151         let need_gc = self.call_impl_check_args(&mut store, params, results)?;
1152         if need_gc {
1153             store.0.gc_async().await;
1154         }
1155         let result = store
1156             .on_fiber(|store| unsafe { self.call_impl_do_call(store, params, results) })
1157             .await??;
1158         Ok(result)
1159     }
1160 
1161     /// Perform dynamic checks that the arguments given to us match
1162     /// the signature of this function and are appropriate to pass to this
1163     /// function.
1164     ///
1165     /// This involves checking to make sure we have the right number and types
1166     /// of arguments as well as making sure everything is from the same `Store`.
1167     ///
1168     /// This must be called just before `call_impl_do_call`.
1169     ///
1170     /// Returns whether we need to GC before calling `call_impl_do_call`.
1171     fn call_impl_check_args<T>(
1172         &self,
1173         store: &mut StoreContextMut<'_, T>,
1174         params: &[Val],
1175         results: &mut [Val],
1176     ) -> Result<bool> {
1177         let (ty, opaque) = self.ty_ref(store.0);
1178         if ty.params().len() != params.len() {
1179             bail!(
1180                 "expected {} arguments, got {}",
1181                 ty.params().len(),
1182                 params.len()
1183             );
1184         }
1185         if ty.results().len() != results.len() {
1186             bail!(
1187                 "expected {} results, got {}",
1188                 ty.results().len(),
1189                 results.len()
1190             );
1191         }
1192         for (ty, arg) in ty.params().zip(params) {
1193             arg.ensure_matches_ty(opaque, &ty)
1194                 .context("argument type mismatch")?;
1195             if !arg.comes_from_same_store(opaque) {
1196                 bail!("cross-`Store` values are not currently supported");
1197             }
1198         }
1199 
1200         #[cfg(feature = "gc")]
1201         {
1202             // Check whether we need to GC before calling into Wasm.
1203             //
1204             // For example, with the DRC collector, whenever we pass GC refs
1205             // from host code to Wasm code, they go into the
1206             // `VMGcRefActivationsTable`. But the table might be at capacity
1207             // already. If it is at capacity (unlikely) then we need to do a GC
1208             // to free up space.
1209             let num_gc_refs = ty.as_wasm_func_type().non_i31_gc_ref_params_count();
1210             if let Some(num_gc_refs) = NonZeroUsize::new(num_gc_refs) {
1211                 return Ok(opaque
1212                     .gc_store()?
1213                     .gc_heap
1214                     .need_gc_before_entering_wasm(num_gc_refs));
1215             }
1216         }
1217 
1218         Ok(false)
1219     }
1220 
1221     /// Do the actual call into Wasm.
1222     ///
1223     /// # Safety
1224     ///
1225     /// You must have type checked the arguments by calling
1226     /// `call_impl_check_args` immediately before calling this function. It is
1227     /// only safe to call this function if that one did not return an error.
1228     unsafe fn call_impl_do_call<T>(
1229         &self,
1230         store: &mut StoreContextMut<'_, T>,
1231         params: &[Val],
1232         results: &mut [Val],
1233     ) -> Result<()> {
1234         // Store the argument values into `values_vec`.
1235         let (ty, _) = self.ty_ref(store.0);
1236         let values_vec_size = params.len().max(ty.results().len());
1237         let mut values_vec = store.0.take_wasm_val_raw_storage();
1238         debug_assert!(values_vec.is_empty());
1239         values_vec.resize_with(values_vec_size, || ValRaw::v128(0));
1240         for (arg, slot) in params.iter().cloned().zip(&mut values_vec) {
1241             unsafe {
1242                 *slot = arg.to_raw(&mut *store)?;
1243             }
1244         }
1245 
1246         unsafe {
1247             self.call_unchecked(
1248                 &mut *store,
1249                 core::ptr::slice_from_raw_parts_mut(values_vec.as_mut_ptr(), values_vec_size),
1250             )?;
1251         }
1252 
1253         for ((i, slot), val) in results.iter_mut().enumerate().zip(&values_vec) {
1254             let ty = self.ty_ref(store.0).0.results().nth(i).unwrap();
1255             *slot = unsafe { Val::from_raw(&mut *store, *val, ty) };
1256         }
1257         values_vec.truncate(0);
1258         store.0.save_wasm_val_raw_storage(values_vec);
1259         Ok(())
1260     }
1261 
1262     #[inline]
1263     pub(crate) fn vm_func_ref(&self, store: &mut StoreOpaque) -> NonNull<VMFuncRef> {
1264         let func_data = &mut store.store_data_mut()[self.0];
1265         let func_ref = func_data.export().func_ref;
1266         if unsafe { func_ref.as_ref().wasm_call.is_some() } {
1267             return func_ref;
1268         }
1269 
1270         if let Some(in_store) = func_data.in_store_func_ref {
1271             in_store.as_non_null()
1272         } else {
1273             unsafe {
1274                 // Move this uncommon/slow path out of line.
1275                 self.copy_func_ref_into_store_and_fill(store, func_ref)
1276             }
1277         }
1278     }
1279 
1280     unsafe fn copy_func_ref_into_store_and_fill(
1281         &self,
1282         store: &mut StoreOpaque,
1283         func_ref: NonNull<VMFuncRef>,
1284     ) -> NonNull<VMFuncRef> {
1285         let func_ref = store.func_refs().push(func_ref.as_ref().clone());
1286         store.store_data_mut()[self.0].in_store_func_ref = Some(SendSyncPtr::new(func_ref));
1287         store.fill_func_refs();
1288         func_ref
1289     }
1290 
1291     pub(crate) unsafe fn from_wasmtime_function(
1292         export: ExportFunction,
1293         store: &mut StoreOpaque,
1294     ) -> Self {
1295         Func::from_func_kind(FuncKind::StoreOwned { export }, store)
1296     }
1297 
1298     fn from_func_kind(kind: FuncKind, store: &mut StoreOpaque) -> Self {
1299         Func(store.store_data_mut().insert(FuncData {
1300             kind,
1301             in_store_func_ref: None,
1302             ty: None,
1303         }))
1304     }
1305 
1306     pub(crate) fn vmimport(&self, store: &mut StoreOpaque, module: &Module) -> VMFunctionImport {
1307         unsafe {
1308             let f = {
1309                 let func_data = &mut store.store_data_mut()[self.0];
1310                 // If we already patched this `funcref.wasm_call` and saved a
1311                 // copy in the store, use the patched version. Otherwise, use
1312                 // the potentially un-patched version.
1313                 if let Some(func_ref) = func_data.in_store_func_ref {
1314                     func_ref.as_non_null()
1315                 } else {
1316                     func_data.export().func_ref
1317                 }
1318             };
1319             VMFunctionImport {
1320                 wasm_call: if let Some(wasm_call) = f.as_ref().wasm_call {
1321                     wasm_call
1322                 } else {
1323                     // Assert that this is a array-call function, since those
1324                     // are the only ones that could be missing a `wasm_call`
1325                     // trampoline.
1326                     let _ = VMArrayCallHostFuncContext::from_opaque(f.as_ref().vmctx);
1327 
1328                     let sig = self.type_index(store.store_data());
1329                     module.wasm_to_array_trampoline(sig).expect(
1330                         "if the wasm is importing a function of a given type, it must have the \
1331                          type's trampoline",
1332                     )
1333                 },
1334                 array_call: f.as_ref().array_call,
1335                 vmctx: f.as_ref().vmctx,
1336             }
1337         }
1338     }
1339 
1340     pub(crate) fn comes_from_same_store(&self, store: &StoreOpaque) -> bool {
1341         store.store_data().contains(self.0)
1342     }
1343 
1344     fn invoke_host_func_for_wasm<T>(
1345         mut caller: Caller<'_, T>,
1346         ty: &FuncType,
1347         values_vec: &mut [ValRaw],
1348         func: &dyn Fn(Caller<'_, T>, &[Val], &mut [Val]) -> Result<()>,
1349     ) -> Result<()> {
1350         // Translate the raw JIT arguments in `values_vec` into a `Val` which
1351         // we'll be passing as a slice. The storage for our slice-of-`Val` we'll
1352         // be taking from the `Store`. We preserve our slice back into the
1353         // `Store` after the hostcall, ideally amortizing the cost of allocating
1354         // the storage across wasm->host calls.
1355         //
1356         // Note that we have a dynamic guarantee that `values_vec` is the
1357         // appropriate length to both read all arguments from as well as store
1358         // all results into.
1359         let mut val_vec = caller.store.0.take_hostcall_val_storage();
1360         debug_assert!(val_vec.is_empty());
1361         let nparams = ty.params().len();
1362         val_vec.reserve(nparams + ty.results().len());
1363         for (i, ty) in ty.params().enumerate() {
1364             val_vec.push(unsafe { Val::from_raw(&mut caller.store, values_vec[i], ty) })
1365         }
1366 
1367         val_vec.extend((0..ty.results().len()).map(|_| Val::null_func_ref()));
1368         let (params, results) = val_vec.split_at_mut(nparams);
1369         func(caller.sub_caller(), params, results)?;
1370 
1371         // Unlike our arguments we need to dynamically check that the return
1372         // values produced are correct. There could be a bug in `func` that
1373         // produces the wrong number, wrong types, or wrong stores of
1374         // values, and we need to catch that here.
1375         for (i, (ret, ty)) in results.iter().zip(ty.results()).enumerate() {
1376             ret.ensure_matches_ty(caller.store.0, &ty)
1377                 .context("function attempted to return an incompatible value")?;
1378             unsafe {
1379                 values_vec[i] = ret.to_raw(&mut caller.store)?;
1380             }
1381         }
1382 
1383         // Restore our `val_vec` back into the store so it's usable for the next
1384         // hostcall to reuse our own storage.
1385         val_vec.truncate(0);
1386         caller.store.0.save_hostcall_val_storage(val_vec);
1387         Ok(())
1388     }
1389 
1390     /// Attempts to extract a typed object from this `Func` through which the
1391     /// function can be called.
1392     ///
1393     /// This function serves as an alternative to [`Func::call`] and
1394     /// [`Func::call_async`]. This method performs a static type check (using
1395     /// the `Params` and `Results` type parameters on the underlying wasm
1396     /// function. If the type check passes then a `TypedFunc` object is returned,
1397     /// otherwise an error is returned describing the typecheck failure.
1398     ///
1399     /// The purpose of this relative to [`Func::call`] is that it's much more
1400     /// efficient when used to invoke WebAssembly functions. With the types
1401     /// statically known far less setup/teardown is required when invoking
1402     /// WebAssembly. If speed is desired then this function is recommended to be
1403     /// used instead of [`Func::call`] (which is more general, hence its
1404     /// slowdown).
1405     ///
1406     /// The `Params` type parameter is used to describe the parameters of the
1407     /// WebAssembly function. This can either be a single type (like `i32`), or
1408     /// a tuple of types representing the list of parameters (like `(i32, f32,
1409     /// f64)`). Additionally you can use `()` to represent that the function has
1410     /// no parameters.
1411     ///
1412     /// The `Results` type parameter is used to describe the results of the
1413     /// function. This behaves the same way as `Params`, but just for the
1414     /// results of the function.
1415     ///
1416     /// # Translating Between WebAssembly and Rust Types
1417     ///
1418     /// Translation between Rust types and WebAssembly types looks like:
1419     ///
1420     /// | WebAssembly                               | Rust                                  |
1421     /// |-------------------------------------------|---------------------------------------|
1422     /// | `i32`                                     | `i32` or `u32`                        |
1423     /// | `i64`                                     | `i64` or `u64`                        |
1424     /// | `f32`                                     | `f32`                                 |
1425     /// | `f64`                                     | `f64`                                 |
1426     /// | `externref` aka `(ref null extern)`       | `Option<Rooted<ExternRef>>`           |
1427     /// | `(ref extern)`                            | `Rooted<ExternRef>`                   |
1428     /// | `nullexternref` aka `(ref null noextern)` | `Option<NoExtern>`                    |
1429     /// | `(ref noextern)`                          | `NoExtern`                            |
1430     /// | `anyref` aka `(ref null any)`             | `Option<Rooted<AnyRef>>`              |
1431     /// | `(ref any)`                               | `Rooted<AnyRef>`                      |
1432     /// | `eqref` aka `(ref null eq)`               | `Option<Rooted<EqRef>>`               |
1433     /// | `(ref eq)`                                | `Rooted<EqRef>`                       |
1434     /// | `i31ref` aka `(ref null i31)`             | `Option<I31>`                         |
1435     /// | `(ref i31)`                               | `I31`                                 |
1436     /// | `structref` aka `(ref null struct)`       | `Option<Rooted<StructRef>>`           |
1437     /// | `(ref struct)`                            | `Rooted<StructRef>`                   |
1438     /// | `arrayref` aka `(ref null array)`         | `Option<Rooted<ArrayRef>>`            |
1439     /// | `(ref array)`                             | `Rooted<ArrayRef>`                    |
1440     /// | `nullref` aka `(ref null none)`           | `Option<NoneRef>`                     |
1441     /// | `(ref none)`                              | `NoneRef`                             |
1442     /// | `funcref` aka `(ref null func)`           | `Option<Func>`                        |
1443     /// | `(ref func)`                              | `Func`                                |
1444     /// | `(ref null <func type index>)`            | `Option<Func>`                        |
1445     /// | `(ref <func type index>)`                 | `Func`                                |
1446     /// | `nullfuncref` aka `(ref null nofunc)`     | `Option<NoFunc>`                      |
1447     /// | `(ref nofunc)`                            | `NoFunc`                              |
1448     /// | `v128`                                    | `V128` on `x86-64` and `aarch64` only |
1449     ///
1450     /// (Note that this mapping is the same as that of [`Func::wrap`], and that
1451     /// anywhere a `Rooted<T>` appears, a `ManuallyRooted<T>` may also appear).
1452     ///
1453     /// Note that once the [`TypedFunc`] return value is acquired you'll use either
1454     /// [`TypedFunc::call`] or [`TypedFunc::call_async`] as necessary to actually invoke
1455     /// the function. This method does not invoke any WebAssembly code, it
1456     /// simply performs a typecheck before returning the [`TypedFunc`] value.
1457     ///
1458     /// This method also has a convenience wrapper as
1459     /// [`Instance::get_typed_func`](crate::Instance::get_typed_func) to
1460     /// directly get a typed function value from an
1461     /// [`Instance`](crate::Instance).
1462     ///
1463     /// ## Subtyping
1464     ///
1465     /// For result types, you can always use a supertype of the WebAssembly
1466     /// function's actual declared result type. For example, if the WebAssembly
1467     /// function was declared with type `(func (result nullfuncref))` you could
1468     /// successfully call `f.typed::<(), Option<Func>>()` because `Option<Func>`
1469     /// corresponds to `funcref`, which is a supertype of `nullfuncref`.
1470     ///
1471     /// For parameter types, you can always use a subtype of the WebAssembly
1472     /// function's actual declared parameter type. For example, if the
1473     /// WebAssembly function was declared with type `(func (param (ref null
1474     /// func)))` you could successfully call `f.typed::<Func, ()>()` because
1475     /// `Func` corresponds to `(ref func)`, which is a subtype of `(ref null
1476     /// func)`.
1477     ///
1478     /// Additionally, for functions which take a reference to a concrete type as
1479     /// a parameter, you can also use the concrete type's supertype. Consider a
1480     /// WebAssembly function that takes a reference to a function with a
1481     /// concrete type: `(ref null <func type index>)`. In this scenario, there
1482     /// is no static `wasmtime::Foo` Rust type that corresponds to that
1483     /// particular Wasm-defined concrete reference type because Wasm modules are
1484     /// loaded dynamically at runtime. You *could* do `f.typed::<Option<NoFunc>,
1485     /// ()>()`, and while that is correctly typed and valid, it is often overly
1486     /// restrictive. The only value you could call the resulting typed function
1487     /// with is the null function reference, but we'd like to call it with
1488     /// non-null function references that happen to be of the correct
1489     /// type. Therefore, `f.typed<Option<Func>, ()>()` is also allowed in this
1490     /// case, even though `Option<Func>` represents `(ref null func)` which is
1491     /// the supertype, not subtype, of `(ref null <func type index>)`. This does
1492     /// imply some minimal dynamic type checks in this case, but it is supported
1493     /// for better ergonomics, to enable passing non-null references into the
1494     /// function.
1495     ///
1496     /// # Errors
1497     ///
1498     /// This function will return an error if `Params` or `Results` does not
1499     /// match the native type of this WebAssembly function.
1500     ///
1501     /// # Panics
1502     ///
1503     /// This method will panic if `store` does not own this function.
1504     ///
1505     /// # Examples
1506     ///
1507     /// An end-to-end example of calling a function which takes no parameters
1508     /// and has no results:
1509     ///
1510     /// ```
1511     /// # use wasmtime::*;
1512     /// # fn main() -> anyhow::Result<()> {
1513     /// let engine = Engine::default();
1514     /// let mut store = Store::new(&engine, ());
1515     /// let module = Module::new(&engine, r#"(module (func (export "foo")))"#)?;
1516     /// let instance = Instance::new(&mut store, &module, &[])?;
1517     /// let foo = instance.get_func(&mut store, "foo").expect("export wasn't a function");
1518     ///
1519     /// // Note that this call can fail due to the typecheck not passing, but
1520     /// // in our case we statically know the module so we know this should
1521     /// // pass.
1522     /// let typed = foo.typed::<(), ()>(&store)?;
1523     ///
1524     /// // Note that this can fail if the wasm traps at runtime.
1525     /// typed.call(&mut store, ())?;
1526     /// # Ok(())
1527     /// # }
1528     /// ```
1529     ///
1530     /// You can also pass in multiple parameters and get a result back
1531     ///
1532     /// ```
1533     /// # use wasmtime::*;
1534     /// # fn foo(add: &Func, mut store: Store<()>) -> anyhow::Result<()> {
1535     /// let typed = add.typed::<(i32, i64), f32>(&store)?;
1536     /// assert_eq!(typed.call(&mut store, (1, 2))?, 3.0);
1537     /// # Ok(())
1538     /// # }
1539     /// ```
1540     ///
1541     /// and similarly if a function has multiple results you can bind that too
1542     ///
1543     /// ```
1544     /// # use wasmtime::*;
1545     /// # fn foo(add_with_overflow: &Func, mut store: Store<()>) -> anyhow::Result<()> {
1546     /// let typed = add_with_overflow.typed::<(u32, u32), (u32, i32)>(&store)?;
1547     /// let (result, overflow) = typed.call(&mut store, (u32::max_value(), 2))?;
1548     /// assert_eq!(result, 1);
1549     /// assert_eq!(overflow, 1);
1550     /// # Ok(())
1551     /// # }
1552     /// ```
1553     pub fn typed<Params, Results>(
1554         &self,
1555         store: impl AsContext,
1556     ) -> Result<TypedFunc<Params, Results>>
1557     where
1558         Params: WasmParams,
1559         Results: WasmResults,
1560     {
1561         // Type-check that the params/results are all valid
1562         let store = store.as_context().0;
1563         let ty = self.load_ty(store);
1564         Params::typecheck(store.engine(), ty.params(), TypeCheckPosition::Param)
1565             .context("type mismatch with parameters")?;
1566         Results::typecheck(store.engine(), ty.results(), TypeCheckPosition::Result)
1567             .context("type mismatch with results")?;
1568 
1569         // and then we can construct the typed version of this function
1570         // (unsafely), which should be safe since we just did the type check above.
1571         unsafe { Ok(TypedFunc::_new_unchecked(store, *self)) }
1572     }
1573 
1574     /// Get a stable hash key for this function.
1575     ///
1576     /// Even if the same underlying function is added to the `StoreData`
1577     /// multiple times and becomes multiple `wasmtime::Func`s, this hash key
1578     /// will be consistent across all of these functions.
1579     #[allow(dead_code)] // Not used yet, but added for consistency.
1580     pub(crate) fn hash_key(&self, store: &mut StoreOpaque) -> impl core::hash::Hash + Eq {
1581         self.vm_func_ref(store).as_ptr() as usize
1582     }
1583 }
1584 
1585 /// Prepares for entrance into WebAssembly.
1586 ///
1587 /// This function will set up context such that `closure` is allowed to call a
1588 /// raw trampoline or a raw WebAssembly function. This *must* be called to do
1589 /// things like catch traps and set up GC properly.
1590 ///
1591 /// The `closure` provided receives a default "caller" `VMContext` parameter it
1592 /// can pass to the called wasm function, if desired.
1593 pub(crate) fn invoke_wasm_and_catch_traps<T>(
1594     store: &mut StoreContextMut<'_, T>,
1595     closure: impl FnMut(*mut VMContext) -> bool,
1596 ) -> Result<()> {
1597     unsafe {
1598         let exit = enter_wasm(store);
1599 
1600         if let Err(trap) = store.0.call_hook(CallHook::CallingWasm) {
1601             exit_wasm(store, exit);
1602             return Err(trap);
1603         }
1604         let result = crate::runtime::vm::catch_traps(store, closure);
1605         exit_wasm(store, exit);
1606         store.0.call_hook(CallHook::ReturningFromWasm)?;
1607         result.map_err(|t| crate::trap::from_runtime_box(store.0, t))
1608     }
1609 }
1610 
1611 /// This function is called to register state within `Store` whenever
1612 /// WebAssembly is entered within the `Store`.
1613 ///
1614 /// This function sets up various limits such as:
1615 ///
1616 /// * The stack limit. This is what ensures that we limit the stack space
1617 ///   allocated by WebAssembly code and it's relative to the initial stack
1618 ///   pointer that called into wasm.
1619 ///
1620 /// This function may fail if the stack limit can't be set because an
1621 /// interrupt already happened.
1622 fn enter_wasm<T>(store: &mut StoreContextMut<'_, T>) -> Option<usize> {
1623     // If this is a recursive call, e.g. our stack limit is already set, then
1624     // we may be able to skip this function.
1625     //
1626     // For synchronous stores there's nothing else to do because all wasm calls
1627     // happen synchronously and on the same stack. This means that the previous
1628     // stack limit will suffice for the next recursive call.
1629     //
1630     // For asynchronous stores then each call happens on a separate native
1631     // stack. This means that the previous stack limit is no longer relevant
1632     // because we're on a separate stack.
1633     if unsafe { *store.0.runtime_limits().stack_limit.get() } != usize::MAX
1634         && !store.0.async_support()
1635     {
1636         return None;
1637     }
1638 
1639     // Ignore this stack pointer business on miri since we can't execute wasm
1640     // anyway and the concept of a stack pointer on miri is a bit nebulous
1641     // regardless.
1642     if cfg!(miri) {
1643         return None;
1644     }
1645 
1646     let stack_pointer = crate::runtime::vm::get_stack_pointer();
1647 
1648     // Determine the stack pointer where, after which, any wasm code will
1649     // immediately trap. This is checked on the entry to all wasm functions.
1650     //
1651     // Note that this isn't 100% precise. We are requested to give wasm
1652     // `max_wasm_stack` bytes, but what we're actually doing is giving wasm
1653     // probably a little less than `max_wasm_stack` because we're
1654     // calculating the limit relative to this function's approximate stack
1655     // pointer. Wasm will be executed on a frame beneath this one (or next
1656     // to it). In any case it's expected to be at most a few hundred bytes
1657     // of slop one way or another. When wasm is typically given a MB or so
1658     // (a million bytes) the slop shouldn't matter too much.
1659     //
1660     // After we've got the stack limit then we store it into the `stack_limit`
1661     // variable.
1662     let wasm_stack_limit = stack_pointer - store.engine().config().max_wasm_stack;
1663     let prev_stack = unsafe {
1664         mem::replace(
1665             &mut *store.0.runtime_limits().stack_limit.get(),
1666             wasm_stack_limit,
1667         )
1668     };
1669 
1670     Some(prev_stack)
1671 }
1672 
1673 fn exit_wasm<T>(store: &mut StoreContextMut<'_, T>, prev_stack: Option<usize>) {
1674     // If we don't have a previous stack pointer to restore, then there's no
1675     // cleanup we need to perform here.
1676     let prev_stack = match prev_stack {
1677         Some(stack) => stack,
1678         None => return,
1679     };
1680 
1681     unsafe {
1682         *store.0.runtime_limits().stack_limit.get() = prev_stack;
1683     }
1684 }
1685 
1686 /// A trait implemented for types which can be returned from closures passed to
1687 /// [`Func::wrap`] and friends.
1688 ///
1689 /// This trait should not be implemented by user types. This trait may change at
1690 /// any time internally. The types which implement this trait, however, are
1691 /// stable over time.
1692 ///
1693 /// For more information see [`Func::wrap`]
1694 pub unsafe trait WasmRet {
1695     // Same as `WasmTy::compatible_with_store`.
1696     #[doc(hidden)]
1697     fn compatible_with_store(&self, store: &StoreOpaque) -> bool;
1698 
1699     /// Stores this return value into the `ptr` specified using the rooted
1700     /// `store`.
1701     ///
1702     /// Traps are communicated through the `Result<_>` return value.
1703     ///
1704     /// # Unsafety
1705     ///
1706     /// This method is unsafe as `ptr` must have the correct length to store
1707     /// this result. This property is only checked in debug mode, not in release
1708     /// mode.
1709     #[doc(hidden)]
1710     unsafe fn store(
1711         self,
1712         store: &mut AutoAssertNoGc<'_>,
1713         ptr: &mut [MaybeUninit<ValRaw>],
1714     ) -> Result<()>;
1715 
1716     #[doc(hidden)]
1717     fn func_type(engine: &Engine, params: impl Iterator<Item = ValType>) -> FuncType;
1718     #[doc(hidden)]
1719     fn may_gc() -> bool;
1720 
1721     // Utilities used to convert an instance of this type to a `Result`
1722     // explicitly, used when wrapping async functions which always bottom-out
1723     // in a function that returns a trap because futures can be cancelled.
1724     #[doc(hidden)]
1725     type Fallible: WasmRet;
1726     #[doc(hidden)]
1727     fn into_fallible(self) -> Self::Fallible;
1728     #[doc(hidden)]
1729     fn fallible_from_error(error: Error) -> Self::Fallible;
1730 }
1731 
1732 unsafe impl<T> WasmRet for T
1733 where
1734     T: WasmTy,
1735 {
1736     type Fallible = Result<T>;
1737 
1738     fn compatible_with_store(&self, store: &StoreOpaque) -> bool {
1739         <Self as WasmTy>::compatible_with_store(self, store)
1740     }
1741 
1742     unsafe fn store(
1743         self,
1744         store: &mut AutoAssertNoGc<'_>,
1745         ptr: &mut [MaybeUninit<ValRaw>],
1746     ) -> Result<()> {
1747         debug_assert!(ptr.len() > 0);
1748         <Self as WasmTy>::store(self, store, ptr.get_unchecked_mut(0))
1749     }
1750 
1751     fn may_gc() -> bool {
1752         T::may_gc()
1753     }
1754 
1755     fn func_type(engine: &Engine, params: impl Iterator<Item = ValType>) -> FuncType {
1756         FuncType::new(engine, params, Some(<Self as WasmTy>::valtype()))
1757     }
1758 
1759     fn into_fallible(self) -> Result<T> {
1760         Ok(self)
1761     }
1762 
1763     fn fallible_from_error(error: Error) -> Result<T> {
1764         Err(error)
1765     }
1766 }
1767 
1768 unsafe impl<T> WasmRet for Result<T>
1769 where
1770     T: WasmRet,
1771 {
1772     type Fallible = Self;
1773 
1774     fn compatible_with_store(&self, store: &StoreOpaque) -> bool {
1775         match self {
1776             Ok(x) => <T as WasmRet>::compatible_with_store(x, store),
1777             Err(_) => true,
1778         }
1779     }
1780 
1781     unsafe fn store(
1782         self,
1783         store: &mut AutoAssertNoGc<'_>,
1784         ptr: &mut [MaybeUninit<ValRaw>],
1785     ) -> Result<()> {
1786         self.and_then(|val| val.store(store, ptr))
1787     }
1788 
1789     fn may_gc() -> bool {
1790         T::may_gc()
1791     }
1792 
1793     fn func_type(engine: &Engine, params: impl Iterator<Item = ValType>) -> FuncType {
1794         T::func_type(engine, params)
1795     }
1796 
1797     fn into_fallible(self) -> Result<T> {
1798         self
1799     }
1800 
1801     fn fallible_from_error(error: Error) -> Result<T> {
1802         Err(error)
1803     }
1804 }
1805 
1806 macro_rules! impl_wasm_host_results {
1807     ($n:tt $($t:ident)*) => (
1808         #[allow(non_snake_case)]
1809         unsafe impl<$($t),*> WasmRet for ($($t,)*)
1810         where
1811             $($t: WasmTy,)*
1812         {
1813             type Fallible = Result<Self>;
1814 
1815             #[inline]
1816             fn compatible_with_store(&self, _store: &StoreOpaque) -> bool {
1817                 let ($($t,)*) = self;
1818                 $( $t.compatible_with_store(_store) && )* true
1819             }
1820 
1821             #[inline]
1822             unsafe fn store(
1823                 self,
1824                 _store: &mut AutoAssertNoGc<'_>,
1825                 _ptr: &mut [MaybeUninit<ValRaw>],
1826             ) -> Result<()> {
1827                 let ($($t,)*) = self;
1828                 let mut _cur = 0;
1829                 $(
1830                     debug_assert!(_cur < _ptr.len());
1831                     let val = _ptr.get_unchecked_mut(_cur);
1832                     _cur += 1;
1833                     WasmTy::store($t, _store, val)?;
1834                 )*
1835                 Ok(())
1836             }
1837 
1838             #[doc(hidden)]
1839             fn may_gc() -> bool {
1840                 $( $t::may_gc() || )* false
1841             }
1842 
1843             fn func_type(engine: &Engine, params: impl Iterator<Item = ValType>) -> FuncType {
1844                 FuncType::new(
1845                     engine,
1846                     params,
1847                     IntoIterator::into_iter([$($t::valtype(),)*]),
1848                 )
1849             }
1850 
1851             #[inline]
1852             fn into_fallible(self) -> Result<Self> {
1853                 Ok(self)
1854             }
1855 
1856             #[inline]
1857             fn fallible_from_error(error: Error) -> Result<Self> {
1858                 Err(error)
1859             }
1860         }
1861     )
1862 }
1863 
1864 for_each_function_signature!(impl_wasm_host_results);
1865 
1866 /// Internal trait implemented for all arguments that can be passed to
1867 /// [`Func::wrap`] and [`Linker::func_wrap`](crate::Linker::func_wrap).
1868 ///
1869 /// This trait should not be implemented by external users, it's only intended
1870 /// as an implementation detail of this crate.
1871 pub trait IntoFunc<T, Params, Results>: Send + Sync + 'static {
1872     /// Convert this function into a `VM{Array,Native}CallHostFuncContext` and
1873     /// internal `VMFuncRef`.
1874     #[doc(hidden)]
1875     fn into_func(self, engine: &Engine) -> HostContext;
1876 }
1877 
1878 macro_rules! impl_into_func {
1879     ($num:tt $arg:ident) => {
1880         // Implement for functions without a leading `&Caller` parameter,
1881         // delegating to the implementation below which does have the leading
1882         // `Caller` parameter.
1883         #[allow(non_snake_case)]
1884         impl<T, F, $arg, R> IntoFunc<T, $arg, R> for F
1885         where
1886             F: Fn($arg) -> R + Send + Sync + 'static,
1887             $arg: WasmTy,
1888             R: WasmRet,
1889         {
1890             fn into_func(self, engine: &Engine) -> HostContext {
1891                 let f = move |_: Caller<'_, T>, $arg: $arg| {
1892                     self($arg)
1893                 };
1894 
1895                 f.into_func(engine)
1896             }
1897         }
1898 
1899         #[allow(non_snake_case)]
1900         impl<T, F, $arg, R> IntoFunc<T, (Caller<'_, T>, $arg), R> for F
1901         where
1902             F: Fn(Caller<'_, T>, $arg) -> R + Send + Sync + 'static,
1903             $arg: WasmTy,
1904             R: WasmRet,
1905         {
1906             fn into_func(self, engine: &Engine) -> HostContext {
1907                 HostContext::from_closure(engine, move |caller: Caller<'_, T>, ($arg,)| {
1908                     self(caller, $arg)
1909                 })
1910             }
1911         }
1912     };
1913     ($num:tt $($args:ident)*) => {
1914         // Implement for functions without a leading `&Caller` parameter,
1915         // delegating to the implementation below which does have the leading
1916         // `Caller` parameter.
1917         #[allow(non_snake_case)]
1918         impl<T, F, $($args,)* R> IntoFunc<T, ($($args,)*), R> for F
1919         where
1920             F: Fn($($args),*) -> R + Send + Sync + 'static,
1921             $($args: WasmTy,)*
1922             R: WasmRet,
1923         {
1924             fn into_func(self, engine: &Engine) -> HostContext {
1925                 let f = move |_: Caller<'_, T>, $($args:$args),*| {
1926                     self($($args),*)
1927                 };
1928 
1929                 f.into_func(engine)
1930             }
1931         }
1932 
1933         #[allow(non_snake_case)]
1934         impl<T, F, $($args,)* R> IntoFunc<T, (Caller<'_, T>, $($args,)*), R> for F
1935         where
1936             F: Fn(Caller<'_, T>, $($args),*) -> R + Send + Sync + 'static,
1937             $($args: WasmTy,)*
1938             R: WasmRet,
1939         {
1940             fn into_func(self, engine: &Engine) -> HostContext {
1941                 HostContext::from_closure(engine, move |caller: Caller<'_, T>, ( $( $args ),* )| {
1942                     self(caller, $( $args ),* )
1943                 })
1944             }
1945         }
1946     }
1947 }
1948 
1949 for_each_function_signature!(impl_into_func);
1950 
1951 /// Trait implemented for various tuples made up of types which implement
1952 /// [`WasmTy`] that can be passed to [`Func::wrap_inner`] and
1953 /// [`HostContext::from_closure`].
1954 pub unsafe trait WasmTyList {
1955     /// Get the value type that each Type in the list represents.
1956     fn valtypes() -> impl Iterator<Item = ValType>;
1957 
1958     // Load a version of `Self` from the `values` provided.
1959     //
1960     // # Safety
1961     //
1962     // This function is unsafe as it's up to the caller to ensure that `values` are
1963     // valid for this given type.
1964     #[doc(hidden)]
1965     unsafe fn load(store: &mut AutoAssertNoGc<'_>, values: &mut [MaybeUninit<ValRaw>]) -> Self;
1966 
1967     #[doc(hidden)]
1968     fn may_gc() -> bool;
1969 }
1970 
1971 macro_rules! impl_wasm_ty_list {
1972     ($num:tt $($args:ident)*) => (paste::paste!{
1973         #[allow(non_snake_case)]
1974         unsafe impl<$($args),*> WasmTyList for ($($args,)*)
1975         where
1976             $($args: WasmTy,)*
1977         {
1978             fn valtypes() -> impl Iterator<Item = ValType> {
1979                 IntoIterator::into_iter([$($args::valtype(),)*])
1980             }
1981 
1982             unsafe fn load(_store: &mut AutoAssertNoGc<'_>, _values: &mut [MaybeUninit<ValRaw>]) -> Self {
1983                 let mut _cur = 0;
1984                 ($({
1985                     debug_assert!(_cur < _values.len());
1986                     let ptr = _values.get_unchecked(_cur).assume_init_ref();
1987                     _cur += 1;
1988                     $args::load(_store, ptr)
1989                 },)*)
1990             }
1991 
1992             fn may_gc() -> bool {
1993                 $( $args::may_gc() || )* false
1994             }
1995         }
1996     });
1997 }
1998 
1999 for_each_function_signature!(impl_wasm_ty_list);
2000 
2001 /// A structure representing the caller's context when creating a function
2002 /// via [`Func::wrap`].
2003 ///
2004 /// This structure can be taken as the first parameter of a closure passed to
2005 /// [`Func::wrap`] or other constructors, and serves two purposes:
2006 ///
2007 /// * First consumers can use [`Caller<'_, T>`](crate::Caller) to get access to
2008 ///   [`StoreContextMut<'_, T>`](crate::StoreContextMut) and/or get access to
2009 ///   `T` itself. This means that the [`Caller`] type can serve as a proxy to
2010 ///   the original [`Store`](crate::Store) itself and is used to satisfy
2011 ///   [`AsContext`] and [`AsContextMut`] bounds.
2012 ///
2013 /// * Second a [`Caller`] can be used as the name implies, learning about the
2014 ///   caller's context, namely it's exported memory and exported functions. This
2015 ///   allows functions which take pointers as arguments to easily read the
2016 ///   memory the pointers point into, or if a function is expected to call
2017 ///   malloc in the wasm module to reserve space for the output you can do that.
2018 ///
2019 /// Host functions which want access to [`Store`](crate::Store)-level state are
2020 /// recommended to use this type.
2021 pub struct Caller<'a, T> {
2022     pub(crate) store: StoreContextMut<'a, T>,
2023     caller: &'a crate::runtime::vm::Instance,
2024 }
2025 
2026 impl<T> Caller<'_, T> {
2027     unsafe fn with<F, R>(caller: *mut VMContext, f: F) -> R
2028     where
2029         // The closure must be valid for any `Caller` it is given; it doesn't
2030         // get to choose the `Caller`'s lifetime.
2031         F: for<'a> FnOnce(Caller<'a, T>) -> R,
2032         // And the return value must not borrow from the caller/store.
2033         R: 'static,
2034     {
2035         debug_assert!(!caller.is_null());
2036         crate::runtime::vm::InstanceAndStore::from_vmctx(caller, |pair| {
2037             let (instance, mut store) = pair.unpack_context_mut::<T>();
2038 
2039             let (gc_lifo_scope, ret) = {
2040                 let gc_lifo_scope = store.0.gc_roots().enter_lifo_scope();
2041 
2042                 let ret = f(Caller {
2043                     store: store.as_context_mut(),
2044                     caller: &instance,
2045                 });
2046 
2047                 (gc_lifo_scope, ret)
2048             };
2049 
2050             // Safe to recreate a mutable borrow of the store because `ret`
2051             // cannot be borrowing from the store.
2052             store.0.exit_gc_lifo_scope(gc_lifo_scope);
2053 
2054             ret
2055         })
2056     }
2057 
2058     fn sub_caller(&mut self) -> Caller<'_, T> {
2059         Caller {
2060             store: self.store.as_context_mut(),
2061             caller: self.caller,
2062         }
2063     }
2064 
2065     /// Looks up an export from the caller's module by the `name` given.
2066     ///
2067     /// This is a low-level function that's typically used to implement passing
2068     /// of pointers or indices between core Wasm instances, where the callee
2069     /// needs to consult the caller's exports to perform memory management and
2070     /// resolve the references.
2071     ///
2072     /// For comparison, in components, the component model handles translating
2073     /// arguments from one component instance to another and managing memory, so
2074     /// that callees don't need to be aware of their callers, which promotes
2075     /// virtualizability of APIs.
2076     ///
2077     /// # Return
2078     ///
2079     /// If an export with the `name` provided was found, then it is returned as an
2080     /// `Extern`. There are a number of situations, however, where the export may not
2081     /// be available:
2082     ///
2083     /// * The caller instance may not have an export named `name`
2084     /// * There may not be a caller available, for example if `Func` was called
2085     ///   directly from host code.
2086     ///
2087     /// It's recommended to take care when calling this API and gracefully
2088     /// handling a `None` return value.
2089     pub fn get_export(&mut self, name: &str) -> Option<Extern> {
2090         // All instances created have a `host_state` with a pointer pointing
2091         // back to themselves. If this caller doesn't have that `host_state`
2092         // then it probably means it was a host-created object like `Func::new`
2093         // which doesn't have any exports we want to return anyway.
2094         self.caller
2095             .host_state()
2096             .downcast_ref::<Instance>()?
2097             .get_export(&mut self.store, name)
2098     }
2099 
2100     /// Looks up an exported [`Extern`] value by a [`ModuleExport`] value.
2101     ///
2102     /// This is similar to [`Self::get_export`] but uses a [`ModuleExport`] value to avoid
2103     /// string lookups where possible. [`ModuleExport`]s can be obtained by calling
2104     /// [`Module::get_export_index`] on the [`Module`] that an instance was instantiated with.
2105     ///
2106     /// This method will search the module for an export with a matching entity index and return
2107     /// the value, if found.
2108     ///
2109     /// Returns `None` if there was no export with a matching entity index.
2110     /// # Panics
2111     ///
2112     /// Panics if `store` does not own this instance.
2113     ///
2114     /// # Usage
2115     /// ```
2116     /// use std::str;
2117     ///
2118     /// # use wasmtime::*;
2119     /// # fn main() -> anyhow::Result<()> {
2120     /// # let mut store = Store::default();
2121     ///
2122     /// let module = Module::new(
2123     ///     store.engine(),
2124     ///     r#"
2125     ///         (module
2126     ///             (import "" "" (func $log_str (param i32 i32)))
2127     ///             (func (export "foo")
2128     ///                 i32.const 4   ;; ptr
2129     ///                 i32.const 13  ;; len
2130     ///                 call $log_str)
2131     ///             (memory (export "memory") 1)
2132     ///             (data (i32.const 4) "Hello, world!"))
2133     ///     "#,
2134     /// )?;
2135     ///
2136     /// let Some(module_export) = module.get_export_index("memory") else {
2137     ///    anyhow::bail!("failed to find `memory` export in module");
2138     /// };
2139     ///
2140     /// let log_str = Func::wrap(&mut store, move |mut caller: Caller<'_, ()>, ptr: i32, len: i32| {
2141     ///     let mem = match caller.get_module_export(&module_export) {
2142     ///         Some(Extern::Memory(mem)) => mem,
2143     ///         _ => anyhow::bail!("failed to find host memory"),
2144     ///     };
2145     ///     let data = mem.data(&caller)
2146     ///         .get(ptr as u32 as usize..)
2147     ///         .and_then(|arr| arr.get(..len as u32 as usize));
2148     ///     let string = match data {
2149     ///         Some(data) => match str::from_utf8(data) {
2150     ///             Ok(s) => s,
2151     ///             Err(_) => anyhow::bail!("invalid utf-8"),
2152     ///         },
2153     ///         None => anyhow::bail!("pointer/length out of bounds"),
2154     ///     };
2155     ///     assert_eq!(string, "Hello, world!");
2156     ///     println!("{}", string);
2157     ///     Ok(())
2158     /// });
2159     /// let instance = Instance::new(&mut store, &module, &[log_str.into()])?;
2160     /// let foo = instance.get_typed_func::<(), ()>(&mut store, "foo")?;
2161     /// foo.call(&mut store, ())?;
2162     /// # Ok(())
2163     /// # }
2164     /// ```
2165     pub fn get_module_export(&mut self, export: &ModuleExport) -> Option<Extern> {
2166         self.caller
2167             .host_state()
2168             .downcast_ref::<Instance>()?
2169             .get_module_export(&mut self.store, export)
2170     }
2171 
2172     /// Access the underlying data owned by this `Store`.
2173     ///
2174     /// Same as [`Store::data`](crate::Store::data)
2175     pub fn data(&self) -> &T {
2176         self.store.data()
2177     }
2178 
2179     /// Access the underlying data owned by this `Store`.
2180     ///
2181     /// Same as [`Store::data_mut`](crate::Store::data_mut)
2182     pub fn data_mut(&mut self) -> &mut T {
2183         self.store.data_mut()
2184     }
2185 
2186     /// Returns the underlying [`Engine`] this store is connected to.
2187     pub fn engine(&self) -> &Engine {
2188         self.store.engine()
2189     }
2190 
2191     /// Perform garbage collection.
2192     ///
2193     /// Same as [`Store::gc`](crate::Store::gc).
2194     #[cfg(feature = "gc")]
2195     pub fn gc(&mut self) {
2196         self.store.gc()
2197     }
2198 
2199     /// Perform garbage collection asynchronously.
2200     ///
2201     /// Same as [`Store::gc_async`](crate::Store::gc_async).
2202     #[cfg(all(feature = "async", feature = "gc"))]
2203     pub async fn gc_async(&mut self)
2204     where
2205         T: Send,
2206     {
2207         self.store.gc_async().await;
2208     }
2209 
2210     /// Returns the remaining fuel in the store.
2211     ///
2212     /// For more information see [`Store::get_fuel`](crate::Store::get_fuel)
2213     pub fn get_fuel(&self) -> Result<u64> {
2214         self.store.get_fuel()
2215     }
2216 
2217     /// Set the amount of fuel in this store to be consumed when executing wasm code.
2218     ///
2219     /// For more information see [`Store::set_fuel`](crate::Store::set_fuel)
2220     pub fn set_fuel(&mut self, fuel: u64) -> Result<()> {
2221         self.store.set_fuel(fuel)
2222     }
2223 
2224     /// Configures this `Store` to yield while executing futures every N units of fuel.
2225     ///
2226     /// For more information see
2227     /// [`Store::fuel_async_yield_interval`](crate::Store::fuel_async_yield_interval)
2228     pub fn fuel_async_yield_interval(&mut self, interval: Option<u64>) -> Result<()> {
2229         self.store.fuel_async_yield_interval(interval)
2230     }
2231 }
2232 
2233 impl<T> AsContext for Caller<'_, T> {
2234     type Data = T;
2235     fn as_context(&self) -> StoreContext<'_, T> {
2236         self.store.as_context()
2237     }
2238 }
2239 
2240 impl<T> AsContextMut for Caller<'_, T> {
2241     fn as_context_mut(&mut self) -> StoreContextMut<'_, T> {
2242         self.store.as_context_mut()
2243     }
2244 }
2245 
2246 // State stored inside a `VMArrayCallHostFuncContext`.
2247 struct HostFuncState<F> {
2248     // The actual host function.
2249     func: F,
2250 
2251     // NB: We have to keep our `VMSharedTypeIndex` registered in the engine for
2252     // as long as this function exists.
2253     #[allow(dead_code)]
2254     ty: RegisteredType,
2255 }
2256 
2257 #[doc(hidden)]
2258 pub enum HostContext {
2259     Array(StoreBox<VMArrayCallHostFuncContext>),
2260 }
2261 
2262 impl From<StoreBox<VMArrayCallHostFuncContext>> for HostContext {
2263     fn from(ctx: StoreBox<VMArrayCallHostFuncContext>) -> Self {
2264         HostContext::Array(ctx)
2265     }
2266 }
2267 
2268 impl HostContext {
2269     fn from_closure<F, T, P, R>(engine: &Engine, func: F) -> Self
2270     where
2271         F: Fn(Caller<'_, T>, P) -> R + Send + Sync + 'static,
2272         P: WasmTyList,
2273         R: WasmRet,
2274     {
2275         let ty = R::func_type(engine, None::<ValType>.into_iter().chain(P::valtypes()));
2276         let type_index = ty.type_index();
2277 
2278         let array_call = Self::array_call_trampoline::<T, F, P, R>;
2279 
2280         let ctx = unsafe {
2281             VMArrayCallHostFuncContext::new(
2282                 array_call,
2283                 type_index,
2284                 Box::new(HostFuncState {
2285                     func,
2286                     ty: ty.into_registered_type(),
2287                 }),
2288             )
2289         };
2290 
2291         ctx.into()
2292     }
2293 
2294     unsafe extern "C" fn array_call_trampoline<T, F, P, R>(
2295         callee_vmctx: *mut VMOpaqueContext,
2296         caller_vmctx: *mut VMOpaqueContext,
2297         args: *mut ValRaw,
2298         args_len: usize,
2299     ) -> bool
2300     where
2301         F: Fn(Caller<'_, T>, P) -> R + 'static,
2302         P: WasmTyList,
2303         R: WasmRet,
2304     {
2305         // Note that this function is intentionally scoped into a
2306         // separate closure. Handling traps and panics will involve
2307         // longjmp-ing from this function which means we won't run
2308         // destructors. As a result anything requiring a destructor
2309         // should be part of this closure, and the long-jmp-ing
2310         // happens after the closure in handling the result.
2311         let run = move |mut caller: Caller<'_, T>| {
2312             let args =
2313                 core::slice::from_raw_parts_mut(args.cast::<MaybeUninit<ValRaw>>(), args_len);
2314             let vmctx = VMArrayCallHostFuncContext::from_opaque(callee_vmctx);
2315             let state = (*vmctx).host_state();
2316 
2317             // Double-check ourselves in debug mode, but we control
2318             // the `Any` here so an unsafe downcast should also
2319             // work.
2320             debug_assert!(state.is::<HostFuncState<F>>());
2321             let state = &*(state as *const _ as *const HostFuncState<F>);
2322             let func = &state.func;
2323 
2324             let ret = 'ret: {
2325                 if let Err(trap) = caller.store.0.call_hook(CallHook::CallingHost) {
2326                     break 'ret R::fallible_from_error(trap);
2327                 }
2328 
2329                 let mut store = if P::may_gc() {
2330                     AutoAssertNoGc::new(caller.store.0)
2331                 } else {
2332                     unsafe { AutoAssertNoGc::disabled(caller.store.0) }
2333                 };
2334                 let params = P::load(&mut store, args);
2335                 let _ = &mut store;
2336                 drop(store);
2337 
2338                 let r = func(caller.sub_caller(), params);
2339                 if let Err(trap) = caller.store.0.call_hook(CallHook::ReturningFromHost) {
2340                     break 'ret R::fallible_from_error(trap);
2341                 }
2342                 r.into_fallible()
2343             };
2344 
2345             if !ret.compatible_with_store(caller.store.0) {
2346                 bail!("host function attempted to return cross-`Store` value to Wasm")
2347             } else {
2348                 let mut store = if R::may_gc() {
2349                     AutoAssertNoGc::new(caller.store.0)
2350                 } else {
2351                     unsafe { AutoAssertNoGc::disabled(caller.store.0) }
2352                 };
2353                 let ret = ret.store(&mut store, args)?;
2354                 Ok(ret)
2355             }
2356         };
2357 
2358         // With nothing else on the stack move `run` into this
2359         // closure and then run it as part of `Caller::with`.
2360         crate::runtime::vm::catch_unwind_and_record_trap(move || {
2361             let caller_vmctx = VMContext::from_opaque(caller_vmctx);
2362             Caller::with(caller_vmctx, run)
2363         })
2364     }
2365 }
2366 
2367 /// Representation of a host-defined function.
2368 ///
2369 /// This is used for `Func::new` but also for `Linker`-defined functions. For
2370 /// `Func::new` this is stored within a `Store`, and for `Linker`-defined
2371 /// functions they wrap this up in `Arc` to enable shared ownership of this
2372 /// across many stores.
2373 ///
2374 /// Technically this structure needs a `<T>` type parameter to connect to the
2375 /// `Store<T>` itself, but that's an unsafe contract of using this for now
2376 /// rather than part of the struct type (to avoid `Func<T>` in the API).
2377 pub(crate) struct HostFunc {
2378     ctx: HostContext,
2379 
2380     // Stored to unregister this function's signature with the engine when this
2381     // is dropped.
2382     engine: Engine,
2383 }
2384 
2385 impl HostFunc {
2386     /// Analog of [`Func::new`]
2387     ///
2388     /// # Panics
2389     ///
2390     /// Panics if the given function type is not associated with the given
2391     /// engine.
2392     pub fn new<T>(
2393         engine: &Engine,
2394         ty: FuncType,
2395         func: impl Fn(Caller<'_, T>, &[Val], &mut [Val]) -> Result<()> + Send + Sync + 'static,
2396     ) -> Self {
2397         assert!(ty.comes_from_same_engine(engine));
2398         let ty_clone = ty.clone();
2399         unsafe {
2400             HostFunc::new_unchecked(engine, ty, move |caller, values| {
2401                 Func::invoke_host_func_for_wasm(caller, &ty_clone, values, &func)
2402             })
2403         }
2404     }
2405 
2406     /// Analog of [`Func::new_unchecked`]
2407     ///
2408     /// # Panics
2409     ///
2410     /// Panics if the given function type is not associated with the given
2411     /// engine.
2412     pub unsafe fn new_unchecked<T>(
2413         engine: &Engine,
2414         ty: FuncType,
2415         func: impl Fn(Caller<'_, T>, &mut [ValRaw]) -> Result<()> + Send + Sync + 'static,
2416     ) -> Self {
2417         assert!(ty.comes_from_same_engine(engine));
2418         let func = move |caller_vmctx, values: &mut [ValRaw]| {
2419             Caller::<T>::with(caller_vmctx, |mut caller| {
2420                 caller.store.0.call_hook(CallHook::CallingHost)?;
2421                 let result = func(caller.sub_caller(), values)?;
2422                 caller.store.0.call_hook(CallHook::ReturningFromHost)?;
2423                 Ok(result)
2424             })
2425         };
2426         let ctx = crate::trampoline::create_array_call_function(&ty, func)
2427             .expect("failed to create function");
2428         HostFunc::_new(engine, ctx.into())
2429     }
2430 
2431     /// Analog of [`Func::wrap_inner`]
2432     pub fn wrap_inner<F, T, Params, Results>(engine: &Engine, func: F) -> Self
2433     where
2434         F: Fn(Caller<'_, T>, Params) -> Results + Send + Sync + 'static,
2435         Params: WasmTyList,
2436         Results: WasmRet,
2437     {
2438         let ctx = HostContext::from_closure(engine, func);
2439         HostFunc::_new(engine, ctx)
2440     }
2441 
2442     /// Analog of [`Func::wrap`]
2443     pub fn wrap<T, Params, Results>(
2444         engine: &Engine,
2445         func: impl IntoFunc<T, Params, Results>,
2446     ) -> Self {
2447         let ctx = func.into_func(engine);
2448         HostFunc::_new(engine, ctx)
2449     }
2450 
2451     /// Requires that this function's signature is already registered within
2452     /// `Engine`. This happens automatically during the above two constructors.
2453     fn _new(engine: &Engine, ctx: HostContext) -> Self {
2454         HostFunc {
2455             ctx,
2456             engine: engine.clone(),
2457         }
2458     }
2459 
2460     /// Inserts this `HostFunc` into a `Store`, returning the `Func` pointing to
2461     /// it.
2462     ///
2463     /// # Unsafety
2464     ///
2465     /// Can only be inserted into stores with a matching `T` relative to when
2466     /// this `HostFunc` was first created.
2467     pub unsafe fn to_func(self: &Arc<Self>, store: &mut StoreOpaque) -> Func {
2468         self.validate_store(store);
2469         let me = self.clone();
2470         Func::from_func_kind(FuncKind::SharedHost(me), store)
2471     }
2472 
2473     /// Inserts this `HostFunc` into a `Store`, returning the `Func` pointing to
2474     /// it.
2475     ///
2476     /// This function is similar to, but not equivalent, to `HostFunc::to_func`.
2477     /// Notably this function requires that the `Arc<Self>` pointer is otherwise
2478     /// rooted within the `StoreOpaque` via another means. When in doubt use
2479     /// `to_func` above as it's safer.
2480     ///
2481     /// # Unsafety
2482     ///
2483     /// Can only be inserted into stores with a matching `T` relative to when
2484     /// this `HostFunc` was first created.
2485     ///
2486     /// Additionally the `&Arc<Self>` is not cloned in this function. Instead a
2487     /// raw pointer to `Self` is stored within the `Store` for this function.
2488     /// The caller must arrange for the `Arc<Self>` to be "rooted" in the store
2489     /// provided via another means, probably by pushing to
2490     /// `StoreOpaque::rooted_host_funcs`.
2491     ///
2492     /// Similarly, the caller must arrange for `rooted_func_ref` to be rooted in
2493     /// the same store.
2494     pub unsafe fn to_func_store_rooted(
2495         self: &Arc<Self>,
2496         store: &mut StoreOpaque,
2497         rooted_func_ref: Option<NonNull<VMFuncRef>>,
2498     ) -> Func {
2499         self.validate_store(store);
2500 
2501         if rooted_func_ref.is_some() {
2502             debug_assert!(self.func_ref().wasm_call.is_none());
2503             debug_assert!(matches!(self.ctx, HostContext::Array(_)));
2504         }
2505 
2506         Func::from_func_kind(
2507             FuncKind::RootedHost(RootedHostFunc::new(self, rooted_func_ref)),
2508             store,
2509         )
2510     }
2511 
2512     /// Same as [`HostFunc::to_func`], different ownership.
2513     unsafe fn into_func(self, store: &mut StoreOpaque) -> Func {
2514         self.validate_store(store);
2515         Func::from_func_kind(FuncKind::Host(Box::new(self)), store)
2516     }
2517 
2518     fn validate_store(&self, store: &mut StoreOpaque) {
2519         // This assert is required to ensure that we can indeed safely insert
2520         // `self` into the `store` provided, otherwise the type information we
2521         // have listed won't be correct. This is possible to hit with the public
2522         // API of Wasmtime, and should be documented in relevant functions.
2523         assert!(
2524             Engine::same(&self.engine, store.engine()),
2525             "cannot use a store with a different engine than a linker was created with",
2526         );
2527     }
2528 
2529     pub(crate) fn sig_index(&self) -> VMSharedTypeIndex {
2530         self.func_ref().type_index
2531     }
2532 
2533     pub(crate) fn func_ref(&self) -> &VMFuncRef {
2534         match &self.ctx {
2535             HostContext::Array(ctx) => unsafe { (*ctx.get()).func_ref() },
2536         }
2537     }
2538 
2539     pub(crate) fn host_ctx(&self) -> &HostContext {
2540         &self.ctx
2541     }
2542 
2543     fn export_func(&self) -> ExportFunction {
2544         ExportFunction {
2545             func_ref: NonNull::from(self.func_ref()),
2546         }
2547     }
2548 }
2549 
2550 impl FuncData {
2551     #[inline]
2552     fn export(&self) -> ExportFunction {
2553         self.kind.export()
2554     }
2555 
2556     pub(crate) fn sig_index(&self) -> VMSharedTypeIndex {
2557         unsafe { self.export().func_ref.as_ref().type_index }
2558     }
2559 }
2560 
2561 impl FuncKind {
2562     #[inline]
2563     fn export(&self) -> ExportFunction {
2564         match self {
2565             FuncKind::StoreOwned { export, .. } => *export,
2566             FuncKind::SharedHost(host) => host.export_func(),
2567             FuncKind::RootedHost(rooted) => ExportFunction {
2568                 func_ref: NonNull::from(rooted.func_ref()),
2569             },
2570             FuncKind::Host(host) => host.export_func(),
2571         }
2572     }
2573 }
2574 
2575 use self::rooted::*;
2576 
2577 /// An inner module is used here to force unsafe construction of
2578 /// `RootedHostFunc` instead of accidentally safely allowing access to its
2579 /// constructor.
2580 mod rooted {
2581     use super::HostFunc;
2582     use crate::runtime::vm::{SendSyncPtr, VMFuncRef};
2583     use alloc::sync::Arc;
2584     use core::ptr::NonNull;
2585 
2586     /// A variant of a pointer-to-a-host-function used in `FuncKind::RootedHost`
2587     /// above.
2588     ///
2589     /// For more documentation see `FuncKind::RootedHost`, `InstancePre`, and
2590     /// `HostFunc::to_func_store_rooted`.
2591     pub(crate) struct RootedHostFunc {
2592         func: SendSyncPtr<HostFunc>,
2593         func_ref: Option<SendSyncPtr<VMFuncRef>>,
2594     }
2595 
2596     impl RootedHostFunc {
2597         /// Note that this is `unsafe` because this wrapper type allows safe
2598         /// access to the pointer given at any time, including outside the
2599         /// window of validity of `func`, so callers must not use the return
2600         /// value past the lifetime of the provided `func`.
2601         ///
2602         /// Similarly, callers must ensure that the given `func_ref` is valid
2603         /// for the lifetime of the return value.
2604         pub(crate) unsafe fn new(
2605             func: &Arc<HostFunc>,
2606             func_ref: Option<NonNull<VMFuncRef>>,
2607         ) -> RootedHostFunc {
2608             RootedHostFunc {
2609                 func: NonNull::from(&**func).into(),
2610                 func_ref: func_ref.map(|p| p.into()),
2611             }
2612         }
2613 
2614         pub(crate) fn func(&self) -> &HostFunc {
2615             // Safety invariants are upheld by the `RootedHostFunc::new` caller.
2616             unsafe { self.func.as_ref() }
2617         }
2618 
2619         pub(crate) fn func_ref(&self) -> &VMFuncRef {
2620             if let Some(f) = self.func_ref {
2621                 // Safety invariants are upheld by the `RootedHostFunc::new` caller.
2622                 unsafe { f.as_ref() }
2623             } else {
2624                 self.func().func_ref()
2625             }
2626         }
2627     }
2628 }
2629 
2630 #[cfg(test)]
2631 mod tests {
2632     use super::*;
2633     use crate::Store;
2634 
2635     #[test]
2636     fn hash_key_is_stable_across_duplicate_store_data_entries() -> Result<()> {
2637         let mut store = Store::<()>::default();
2638         let module = Module::new(
2639             store.engine(),
2640             r#"
2641                 (module
2642                     (func (export "f")
2643                         nop
2644                     )
2645                 )
2646             "#,
2647         )?;
2648         let instance = Instance::new(&mut store, &module, &[])?;
2649 
2650         // Each time we `get_func`, we call `Func::from_wasmtime` which adds a
2651         // new entry to `StoreData`, so `f1` and `f2` will have different
2652         // indices into `StoreData`.
2653         let f1 = instance.get_func(&mut store, "f").unwrap();
2654         let f2 = instance.get_func(&mut store, "f").unwrap();
2655 
2656         // But their hash keys are the same.
2657         assert!(
2658             f1.hash_key(&mut store.as_context_mut().0)
2659                 == f2.hash_key(&mut store.as_context_mut().0)
2660         );
2661 
2662         // But the hash keys are different from different funcs.
2663         let instance2 = Instance::new(&mut store, &module, &[])?;
2664         let f3 = instance2.get_func(&mut store, "f").unwrap();
2665         assert!(
2666             f1.hash_key(&mut store.as_context_mut().0)
2667                 != f3.hash_key(&mut store.as_context_mut().0)
2668         );
2669 
2670         Ok(())
2671     }
2672 }
2673