1 use crate::linker::{Definition, DefinitionType};
2 use crate::prelude::*;
3 use crate::runtime::vm::{
4     self, Imports, ModuleRuntimeInfo, VMFuncRef, VMFunctionImport, VMGlobalImport, VMMemoryImport,
5     VMTableImport, VMTagImport,
6 };
7 use crate::store::{AllocateInstanceKind, InstanceId, StoreInstanceId, StoreOpaque};
8 use crate::types::matching;
9 use crate::{
10     AsContextMut, Engine, Export, Extern, Func, Global, Memory, Module, ModuleExport, SharedMemory,
11     StoreContext, StoreContextMut, Table, Tag, TypedFunc,
12 };
13 use alloc::sync::Arc;
14 use core::ptr::NonNull;
15 use wasmparser::WasmFeatures;
16 use wasmtime_environ::{
17     EntityIndex, EntityType, FuncIndex, GlobalIndex, MemoryIndex, PrimaryMap, TableIndex, TagIndex,
18     TypeTrace,
19 };
20 
21 /// An instantiated WebAssembly module.
22 ///
23 /// This type represents the instantiation of a [`Module`]. Once instantiated
24 /// you can access the [`exports`](Instance::exports) which are of type
25 /// [`Extern`] and provide the ability to call functions, set globals, read
26 /// memory, etc. When interacting with any wasm code you'll want to make an
27 /// [`Instance`] to call any code or execute anything.
28 ///
29 /// Instances are owned by a [`Store`](crate::Store) which is passed in at
30 /// creation time. It's recommended to create instances with
31 /// [`Linker::instantiate`](crate::Linker::instantiate) or similar
32 /// [`Linker`](crate::Linker) methods, but a more low-level constructor is also
33 /// available as [`Instance::new`].
34 #[derive(Copy, Clone, Debug)]
35 #[repr(C)]
36 pub struct Instance {
37     id: StoreInstanceId,
38 }
39 
40 // Double-check that the C representation in `instance.h` matches our in-Rust
41 // representation here in terms of size/alignment/etc.
42 const _: () = {
43     #[repr(C)]
44     struct C(u64, usize);
45     assert!(core::mem::size_of::<C>() == core::mem::size_of::<Instance>());
46     assert!(core::mem::align_of::<C>() == core::mem::align_of::<Instance>());
47     assert!(core::mem::offset_of!(Instance, id) == 0);
48 };
49 
50 impl Instance {
51     /// Creates a new [`Instance`] from the previously compiled [`Module`] and
52     /// list of `imports` specified.
53     ///
54     /// This method instantiates the `module` provided with the `imports`,
55     /// following the procedure in the [core specification][inst] to
56     /// instantiate. Instantiation can fail for a number of reasons (many
57     /// specified below), but if successful the `start` function will be
58     /// automatically run (if specified in the `module`) and then the
59     /// [`Instance`] will be returned.
60     ///
61     /// Per the WebAssembly spec, instantiation includes running the module's
62     /// start function, if it has one (not to be confused with the `_start`
63     /// function, which is not run).
64     ///
65     /// Note that this is a low-level function that just performs an
66     /// instantiation. See the [`Linker`](crate::Linker) struct for an API which
67     /// provides a convenient way to link imports and provides automatic Command
68     /// and Reactor behavior.
69     ///
70     /// ## Providing Imports
71     ///
72     /// The entries in the list of `imports` are intended to correspond 1:1
73     /// with the list of imports returned by [`Module::imports`]. Before
74     /// calling [`Instance::new`] you'll want to inspect the return value of
75     /// [`Module::imports`] and, for each import type, create an [`Extern`]
76     /// which corresponds to that type.  These [`Extern`] values are all then
77     /// collected into a list and passed to this function.
78     ///
79     /// Note that this function is intentionally relatively low level. For an
80     /// easier time passing imports by doing name-based resolution it's
81     /// recommended to instead use the [`Linker`](crate::Linker) type.
82     ///
83     /// ## Errors
84     ///
85     /// This function can fail for a number of reasons, including, but not
86     /// limited to:
87     ///
88     /// * The number of `imports` provided doesn't match the number of imports
89     ///   returned by the `module`'s [`Module::imports`] method.
90     /// * The type of any [`Extern`] doesn't match the corresponding
91     ///   [`ExternType`] entry that it maps to.
92     /// * The `start` function in the instance, if present, traps.
93     /// * Module/instance resource limits are exceeded.
94     ///
95     /// When instantiation fails it's recommended to inspect the return value to
96     /// see why it failed, or bubble it upwards. If you'd like to specifically
97     /// check for trap errors, you can use `error.downcast::<Trap>()`. For more
98     /// about error handling see the [`Trap`] documentation.
99     ///
100     /// [`Trap`]: crate::Trap
101     ///
102     /// # Panics
103     ///
104     /// This function will panic if called with a store associated with a
105     /// [`asynchronous config`](crate::Config::async_support). This function
106     /// will also panic if any [`Extern`] supplied is not owned by `store`.
107     ///
108     /// [inst]: https://webassembly.github.io/spec/core/exec/modules.html#exec-instantiation
109     /// [`ExternType`]: crate::ExternType
110     pub fn new(
111         mut store: impl AsContextMut,
112         module: &Module,
113         imports: &[Extern],
114     ) -> Result<Instance> {
115         let mut store = store.as_context_mut();
116         let imports = Instance::typecheck_externs(store.0, module, imports)?;
117         // Note that the unsafety here should be satisfied by the call to
118         // `typecheck_externs` above which satisfies the condition that all
119         // the imports are valid for this module.
120         unsafe { Instance::new_started(&mut store, module, imports.as_ref()) }
121     }
122 
123     /// Same as [`Instance::new`], except for usage in [asynchronous stores].
124     ///
125     /// For more details about this function see the documentation on
126     /// [`Instance::new`]. The only difference between these two methods is that
127     /// this one will asynchronously invoke the wasm start function in case it
128     /// calls any imported function which is an asynchronous host function (e.g.
129     /// created with [`Func::new_async`](crate::Func::new_async).
130     ///
131     /// # Panics
132     ///
133     /// This function will panic if called with a store associated with a
134     /// [`synchronous config`](crate::Config::new). This is only compatible with
135     /// stores associated with an [`asynchronous
136     /// config`](crate::Config::async_support).
137     ///
138     /// This function will also panic, like [`Instance::new`], if any [`Extern`]
139     /// specified does not belong to `store`.
140     ///
141     /// # Examples
142     ///
143     /// An example of using this function:
144     ///
145     /// ```
146     /// use wasmtime::{Result, Store, Engine, Config, Module, Instance};
147     ///
148     /// #[tokio::main]
149     /// async fn main() -> Result<()> {
150     ///     let mut config = Config::new();
151     ///     config.async_support(true);
152     ///     let engine = Engine::new(&config)?;
153     ///
154     ///     // For this example, a module with no imports is being used hence
155     ///     // the empty array to `Instance::new_async`.
156     ///     let module = Module::new(&engine, "(module)")?;
157     ///     let mut store = Store::new(&engine, ());
158     ///     let instance = Instance::new_async(&mut store, &module, &[]).await?;
159     ///
160     ///     // ... use `instance` and exports and such ...
161     ///
162     ///     Ok(())
163     /// }
164     /// ```
165     ///
166     /// Note, though, that the future returned from this function is only
167     /// `Send` if the store's own data is `Send` meaning that this does not
168     /// compile for example:
169     ///
170     /// ```compile_fail
171     /// use wasmtime::{Result, Store, Engine, Config, Module, Instance};
172     /// use std::rc::Rc;
173     ///
174     /// #[tokio::main]
175     /// async fn main() -> Result<()> {
176     ///     let mut config = Config::new();
177     ///     config.async_support(true);
178     ///     let engine = Engine::new(&config)?;
179     ///
180     ///     let module = Module::new(&engine, "(module)")?;
181     ///
182     ///     // Note that `Rc<()>` is NOT `Send`, which is what many future
183     ///     // runtimes require and below will cause a failure.
184     ///     let mut store = Store::new(&engine, Rc::new(()));
185     ///
186     ///     // Compile failure because `Store<Rc<()>>` is not `Send`
187     ///     assert_send(Instance::new_async(&mut store, &module, &[])).await?;
188     ///
189     ///     Ok(())
190     /// }
191     ///
192     /// fn assert_send<T: Send>(t: T) -> T { t }
193     /// ```
194     #[cfg(feature = "async")]
195     pub async fn new_async(
196         mut store: impl AsContextMut,
197         module: &Module,
198         imports: &[Extern],
199     ) -> Result<Instance> {
200         let mut store = store.as_context_mut();
201         let imports = Instance::typecheck_externs(store.0, module, imports)?;
202         // See `new` for notes on this unsafety
203         unsafe { Instance::new_started_async(&mut store, module, imports.as_ref()).await }
204     }
205 
206     fn typecheck_externs(
207         store: &mut StoreOpaque,
208         module: &Module,
209         imports: &[Extern],
210     ) -> Result<OwnedImports> {
211         for import in imports {
212             if !import.comes_from_same_store(store) {
213                 bail!("cross-`Store` instantiation is not currently supported");
214             }
215         }
216 
217         typecheck(module, imports, |cx, ty, item| {
218             let item = DefinitionType::from(store, item);
219             cx.definition(ty, &item)
220         })?;
221 
222         // When pushing functions into `OwnedImports` it's required that their
223         // `wasm_call` fields are all filled out. This `module` is guaranteed
224         // to have any trampolines necessary for functions so register the
225         // module with the store and then attempt to fill out any outstanding
226         // holes.
227         //
228         // Note that under normal operation this shouldn't do much as the list
229         // of funcs-with-holes should generally be empty. As a result the
230         // process of filling this out is not super optimized at this point.
231         store.modules_mut().register_module(module);
232         let (funcrefs, modules) = store.func_refs_and_modules();
233         funcrefs.fill(modules);
234 
235         let mut owned_imports = OwnedImports::new(module);
236         for import in imports {
237             owned_imports.push(import, store);
238         }
239         Ok(owned_imports)
240     }
241 
242     /// Internal function to create an instance and run the start function.
243     ///
244     /// This function's unsafety is the same as `Instance::new_raw`.
245     pub(crate) unsafe fn new_started<T>(
246         store: &mut StoreContextMut<'_, T>,
247         module: &Module,
248         imports: Imports<'_>,
249     ) -> Result<Instance> {
250         assert!(
251             !store.0.async_support(),
252             "must use async instantiation when async support is enabled",
253         );
254 
255         // SAFETY: the safety contract of `new_started_impl` is the same as this
256         // function.
257         unsafe { Self::new_started_impl(store, module, imports) }
258     }
259 
260     /// Internal function to create an instance and run the start function.
261     ///
262     /// ONLY CALL THIS IF YOU HAVE ALREADY CHECKED FOR ASYNCNESS AND HANDLED
263     /// THE FIBER NONSENSE
264     pub(crate) unsafe fn new_started_impl<T>(
265         store: &mut StoreContextMut<'_, T>,
266         module: &Module,
267         imports: Imports<'_>,
268     ) -> Result<Instance> {
269         // SAFETY: the safety contract of `new_raw` is the same as this
270         // function.
271         let (instance, start) = unsafe { Instance::new_raw(store.0, module, imports)? };
272         if let Some(start) = start {
273             instance.start_raw(store, start)?;
274         }
275         Ok(instance)
276     }
277 
278     /// Internal function to create an instance and run the start function.
279     ///
280     /// This function's unsafety is the same as `Instance::new_raw`.
281     #[cfg(feature = "async")]
282     async unsafe fn new_started_async<T>(
283         store: &mut StoreContextMut<'_, T>,
284         module: &Module,
285         imports: Imports<'_>,
286     ) -> Result<Instance> {
287         assert!(
288             store.0.async_support(),
289             "must use sync instantiation when async support is disabled",
290         );
291 
292         store
293             .on_fiber(|store| {
294                 // SAFETY: the unsafe contract of `new_started_impl` is the same
295                 // as this function.
296                 unsafe { Self::new_started_impl(store, module, imports) }
297             })
298             .await?
299     }
300 
301     /// Internal function to create an instance which doesn't have its `start`
302     /// function run yet.
303     ///
304     /// This is not intended to be exposed from Wasmtime, it's intended to
305     /// refactor out common code from `new_started` and `new_started_async`.
306     ///
307     /// Note that this step needs to be run on a fiber in async mode even
308     /// though it doesn't do any blocking work because an async resource
309     /// limiter may need to yield.
310     ///
311     /// # Unsafety
312     ///
313     /// This method is unsafe because it does not type-check the `imports`
314     /// provided. The `imports` provided must be suitable for the module
315     /// provided as well.
316     unsafe fn new_raw(
317         store: &mut StoreOpaque,
318         module: &Module,
319         imports: Imports<'_>,
320     ) -> Result<(Instance, Option<FuncIndex>)> {
321         if !Engine::same(store.engine(), module.engine()) {
322             bail!("cross-`Engine` instantiation is not currently supported");
323         }
324         store.bump_resource_counts(module)?;
325 
326         // Allocate the GC heap, if necessary.
327         if module.env_module().needs_gc_heap {
328             store.ensure_gc_store()?;
329         }
330 
331         let compiled_module = module.compiled_module();
332 
333         // Register the module just before instantiation to ensure we keep the module
334         // properly referenced while in use by the store.
335         let module_id = store.modules_mut().register_module(module);
336 
337         // The first thing we do is issue an instance allocation request
338         // to the instance allocator. This, on success, will give us an
339         // instance handle.
340         //
341         // SAFETY: this module, by construction, was already validated within
342         // the store.
343         let id = unsafe {
344             store.allocate_instance(
345                 AllocateInstanceKind::Module(module_id),
346                 &ModuleRuntimeInfo::Module(module.clone()),
347                 imports,
348             )?
349         };
350 
351         // Additionally, before we start doing fallible instantiation, we
352         // do one more step which is to insert an `InstanceData`
353         // corresponding to this instance. This `InstanceData` can be used
354         // via `Caller::get_export` if our instance's state "leaks" into
355         // other instances, even if we don't return successfully from this
356         // function.
357         //
358         // We don't actually load all exports from the instance at this
359         // time, instead preferring to lazily load them as they're demanded.
360         // For module/instance exports, though, those aren't actually
361         // stored in the instance handle so we need to immediately handle
362         // those here.
363         let instance = Instance::from_wasmtime(id, store);
364 
365         // Now that we've recorded all information we need to about this
366         // instance within a `Store` we can start performing fallible
367         // initialization. Note that we still defer the `start` function to
368         // later since that may need to run asynchronously.
369         //
370         // If this returns an error (or if the start function traps) then
371         // any other initialization which may have succeeded which placed
372         // items from this instance into other instances should be ok when
373         // those items are loaded and run we'll have all the metadata to
374         // look at them.
375         let bulk_memory = store
376             .engine()
377             .features()
378             .contains(WasmFeatures::BULK_MEMORY);
379 
380         vm::initialize_instance(store, id, compiled_module.module(), bulk_memory)?;
381 
382         Ok((instance, compiled_module.module().start_func))
383     }
384 
385     pub(crate) fn from_wasmtime(id: InstanceId, store: &mut StoreOpaque) -> Instance {
386         Instance {
387             id: StoreInstanceId::new(store.id(), id),
388         }
389     }
390 
391     fn start_raw<T>(&self, store: &mut StoreContextMut<'_, T>, start: FuncIndex) -> Result<()> {
392         // If a start function is present, invoke it. Make sure we use all the
393         // trap-handling configuration in `store` as well.
394         let store_id = store.0.id();
395         let mut instance = self.id.get_mut(store.0);
396         // SAFETY: the `store_id` is the id of the store that owns this
397         // instance and any function stored within the instance.
398         let f = unsafe { instance.as_mut().get_exported_func(store_id, start) };
399         let caller_vmctx = instance.vmctx();
400         unsafe {
401             let funcref = f.vm_func_ref(store.0);
402             super::func::invoke_wasm_and_catch_traps(store, |_default_caller, vm| {
403                 VMFuncRef::array_call(funcref, vm, caller_vmctx, NonNull::from(&mut []))
404             })?;
405         }
406         Ok(())
407     }
408 
409     /// Get this instance's module.
410     pub fn module<'a, T: 'static>(&self, store: impl Into<StoreContext<'a, T>>) -> &'a Module {
411         self._module(store.into().0)
412     }
413 
414     fn _module<'a>(&self, store: &'a StoreOpaque) -> &'a Module {
415         store.module_for_instance(self.id).unwrap()
416     }
417 
418     /// Returns the list of exported items from this [`Instance`].
419     ///
420     /// # Panics
421     ///
422     /// Panics if `store` does not own this instance.
423     pub fn exports<'a, T: 'static>(
424         &'a self,
425         store: impl Into<StoreContextMut<'a, T>>,
426     ) -> impl ExactSizeIterator<Item = Export<'a>> + 'a {
427         self._exports(store.into().0)
428     }
429 
430     fn _exports<'a>(
431         &'a self,
432         store: &'a mut StoreOpaque,
433     ) -> impl ExactSizeIterator<Item = Export<'a>> + 'a {
434         let module = store[self.id].env_module().clone();
435         let mut items = Vec::new();
436         for (_name, entity) in module.exports.iter() {
437             items.push(self._get_export(store, *entity));
438         }
439         store[self.id]
440             .env_module()
441             .exports
442             .iter()
443             .zip(items)
444             .map(|((name, _), item)| Export::new(name, item))
445     }
446 
447     /// Looks up an exported [`Extern`] value by name.
448     ///
449     /// This method will search the module for an export named `name` and return
450     /// the value, if found.
451     ///
452     /// Returns `None` if there was no export named `name`.
453     ///
454     /// # Panics
455     ///
456     /// Panics if `store` does not own this instance.
457     ///
458     /// # Why does `get_export` take a mutable context?
459     ///
460     /// This method requires a mutable context because an instance's exports are
461     /// lazily populated, and we cache them as they are accessed. This makes
462     /// instantiating a module faster, but also means this method requires a
463     /// mutable context.
464     pub fn get_export(&self, mut store: impl AsContextMut, name: &str) -> Option<Extern> {
465         let store = store.as_context_mut().0;
466         let entity = *store[self.id].env_module().exports.get(name)?;
467         Some(self._get_export(store, entity))
468     }
469 
470     /// Looks up an exported [`Extern`] value by a [`ModuleExport`] value.
471     ///
472     /// This is similar to [`Instance::get_export`] but uses a [`ModuleExport`] value to avoid
473     /// string lookups where possible. [`ModuleExport`]s can be obtained by calling
474     /// [`Module::get_export_index`] on the [`Module`] that this instance was instantiated with.
475     ///
476     /// This method will search the module for an export with a matching entity index and return
477     /// the value, if found.
478     ///
479     /// Returns `None` if there was no export with a matching entity index.
480     ///
481     /// # Panics
482     ///
483     /// Panics if `store` does not own this instance.
484     pub fn get_module_export(
485         &self,
486         mut store: impl AsContextMut,
487         export: &ModuleExport,
488     ) -> Option<Extern> {
489         let store = store.as_context_mut().0;
490 
491         // Verify the `ModuleExport` matches the module used in this instance.
492         if self._module(store).id() != export.module {
493             return None;
494         }
495 
496         Some(self._get_export(store, export.entity))
497     }
498 
499     fn _get_export(&self, store: &mut StoreOpaque, entity: EntityIndex) -> Extern {
500         let id = store.id();
501         // SAFETY: the store `id` owns this instance and all exports contained
502         // within.
503         let export = unsafe { self.id.get_mut(store).get_export_by_index_mut(id, entity) };
504         unsafe { Extern::from_wasmtime_export(export, store) }
505     }
506 
507     /// Looks up an exported [`Func`] value by name.
508     ///
509     /// Returns `None` if there was no export named `name`, or if there was but
510     /// it wasn't a function.
511     ///
512     /// # Panics
513     ///
514     /// Panics if `store` does not own this instance.
515     pub fn get_func(&self, store: impl AsContextMut, name: &str) -> Option<Func> {
516         self.get_export(store, name)?.into_func()
517     }
518 
519     /// Looks up an exported [`Func`] value by name and with its type.
520     ///
521     /// This function is a convenience wrapper over [`Instance::get_func`] and
522     /// [`Func::typed`]. For more information see the linked documentation.
523     ///
524     /// Returns an error if `name` isn't a function export or if the export's
525     /// type did not match `Params` or `Results`
526     ///
527     /// # Panics
528     ///
529     /// Panics if `store` does not own this instance.
530     pub fn get_typed_func<Params, Results>(
531         &self,
532         mut store: impl AsContextMut,
533         name: &str,
534     ) -> Result<TypedFunc<Params, Results>>
535     where
536         Params: crate::WasmParams,
537         Results: crate::WasmResults,
538     {
539         let f = self
540             .get_export(store.as_context_mut(), name)
541             .and_then(|f| f.into_func())
542             .ok_or_else(|| anyhow!("failed to find function export `{}`", name))?;
543         Ok(f.typed::<Params, Results>(store)
544             .with_context(|| format!("failed to convert function `{name}` to given type"))?)
545     }
546 
547     /// Looks up an exported [`Table`] value by name.
548     ///
549     /// Returns `None` if there was no export named `name`, or if there was but
550     /// it wasn't a table.
551     ///
552     /// # Panics
553     ///
554     /// Panics if `store` does not own this instance.
555     pub fn get_table(&self, store: impl AsContextMut, name: &str) -> Option<Table> {
556         self.get_export(store, name)?.into_table()
557     }
558 
559     /// Looks up an exported [`Memory`] value by name.
560     ///
561     /// Returns `None` if there was no export named `name`, or if there was but
562     /// it wasn't a memory.
563     ///
564     /// # Panics
565     ///
566     /// Panics if `store` does not own this instance.
567     pub fn get_memory(&self, store: impl AsContextMut, name: &str) -> Option<Memory> {
568         self.get_export(store, name)?.into_memory()
569     }
570 
571     /// Looks up an exported [`SharedMemory`] value by name.
572     ///
573     /// Returns `None` if there was no export named `name`, or if there was but
574     /// it wasn't a shared memory.
575     ///
576     /// # Panics
577     ///
578     /// Panics if `store` does not own this instance.
579     pub fn get_shared_memory(
580         &self,
581         mut store: impl AsContextMut,
582         name: &str,
583     ) -> Option<SharedMemory> {
584         let mut store = store.as_context_mut();
585         self.get_export(&mut store, name)?.into_shared_memory()
586     }
587 
588     /// Looks up an exported [`Global`] value by name.
589     ///
590     /// Returns `None` if there was no export named `name`, or if there was but
591     /// it wasn't a global.
592     ///
593     /// # Panics
594     ///
595     /// Panics if `store` does not own this instance.
596     pub fn get_global(&self, store: impl AsContextMut, name: &str) -> Option<Global> {
597         self.get_export(store, name)?.into_global()
598     }
599 
600     /// Looks up a tag [`Tag`] by name.
601     ///
602     /// Returns `None` if there was no export named `name`, or if there was but
603     /// it wasn't a tag.
604     ///
605     /// # Panics
606     ///
607     /// Panics if `store` does not own this instance.
608     pub fn get_tag(&self, store: impl AsContextMut, name: &str) -> Option<Tag> {
609         self.get_export(store, name)?.into_tag()
610     }
611 
612     #[allow(
613         dead_code,
614         reason = "c-api crate does not yet support exnrefs and causes this method to be dead."
615     )]
616     pub(crate) fn id(&self) -> InstanceId {
617         self.id.instance()
618     }
619 
620     /// Get all globals within this instance.
621     ///
622     /// Returns both import and defined globals.
623     ///
624     /// Returns both exported and non-exported globals.
625     ///
626     /// Gives access to the full globals space.
627     #[cfg(feature = "coredump")]
628     pub(crate) fn all_globals<'a>(
629         &'a self,
630         store: &'a mut StoreOpaque,
631     ) -> impl ExactSizeIterator<Item = (GlobalIndex, Global)> + 'a {
632         let store_id = store.id();
633         store[self.id].all_globals(store_id)
634     }
635 
636     /// Get all memories within this instance.
637     ///
638     /// Returns both import and defined memories.
639     ///
640     /// Returns both exported and non-exported memories.
641     ///
642     /// Gives access to the full memories space.
643     #[cfg(feature = "coredump")]
644     pub(crate) fn all_memories<'a>(
645         &'a self,
646         store: &'a StoreOpaque,
647     ) -> impl ExactSizeIterator<Item = (MemoryIndex, Memory)> + 'a {
648         let store_id = store.id();
649         store[self.id].all_memories(store_id)
650     }
651 }
652 
653 pub(crate) struct OwnedImports {
654     functions: PrimaryMap<FuncIndex, VMFunctionImport>,
655     tables: PrimaryMap<TableIndex, VMTableImport>,
656     memories: PrimaryMap<MemoryIndex, VMMemoryImport>,
657     globals: PrimaryMap<GlobalIndex, VMGlobalImport>,
658     tags: PrimaryMap<TagIndex, VMTagImport>,
659 }
660 
661 impl OwnedImports {
662     fn new(module: &Module) -> OwnedImports {
663         let mut ret = OwnedImports::empty();
664         ret.reserve(module);
665         return ret;
666     }
667 
668     pub(crate) fn empty() -> OwnedImports {
669         OwnedImports {
670             functions: PrimaryMap::new(),
671             tables: PrimaryMap::new(),
672             memories: PrimaryMap::new(),
673             globals: PrimaryMap::new(),
674             tags: PrimaryMap::new(),
675         }
676     }
677 
678     pub(crate) fn reserve(&mut self, module: &Module) {
679         let raw = module.compiled_module().module();
680         self.functions.reserve(raw.num_imported_funcs);
681         self.tables.reserve(raw.num_imported_tables);
682         self.memories.reserve(raw.num_imported_memories);
683         self.globals.reserve(raw.num_imported_globals);
684         self.tags.reserve(raw.num_imported_tags);
685     }
686 
687     #[cfg(feature = "component-model")]
688     pub(crate) fn clear(&mut self) {
689         self.functions.clear();
690         self.tables.clear();
691         self.memories.clear();
692         self.globals.clear();
693         self.tags.clear();
694     }
695 
696     fn push(&mut self, item: &Extern, store: &mut StoreOpaque) {
697         match item {
698             Extern::Func(i) => {
699                 self.functions.push(i.vmimport(store));
700             }
701             Extern::Global(i) => {
702                 self.globals.push(i.vmimport(store));
703             }
704             Extern::Table(i) => {
705                 self.tables.push(i.vmimport(store));
706             }
707             Extern::Memory(i) => {
708                 self.memories.push(i.vmimport(store));
709             }
710             Extern::SharedMemory(i) => {
711                 self.memories.push(i.vmimport(store));
712             }
713             Extern::Tag(i) => {
714                 self.tags.push(i.vmimport(store));
715             }
716         }
717     }
718 
719     /// Note that this is unsafe as the validity of `item` is not verified and
720     /// it contains a bunch of raw pointers.
721     #[cfg(feature = "component-model")]
722     pub(crate) fn push_export(&mut self, store: &StoreOpaque, item: &crate::runtime::vm::Export) {
723         match item {
724             crate::runtime::vm::Export::Function(f) => {
725                 // SAFETY: the funcref associated with a `Func` is valid to use
726                 // under the `store` that owns the function.
727                 let f = unsafe { f.vm_func_ref(store).as_ref() };
728                 self.functions.push(VMFunctionImport {
729                     wasm_call: f.wasm_call.unwrap(),
730                     array_call: f.array_call,
731                     vmctx: f.vmctx,
732                 });
733             }
734             crate::runtime::vm::Export::Global(g) => {
735                 self.globals.push(g.vmimport(store));
736             }
737             crate::runtime::vm::Export::Table(t) => {
738                 self.tables.push(t.vmimport(store));
739             }
740             crate::runtime::vm::Export::Memory { memory, .. } => {
741                 self.memories.push(memory.vmimport(store));
742             }
743             crate::runtime::vm::Export::Tag(t) => {
744                 self.tags.push(t.vmimport(store));
745             }
746         }
747     }
748 
749     pub(crate) fn as_ref(&self) -> Imports<'_> {
750         Imports {
751             tables: self.tables.values().as_slice(),
752             globals: self.globals.values().as_slice(),
753             memories: self.memories.values().as_slice(),
754             functions: self.functions.values().as_slice(),
755             tags: self.tags.values().as_slice(),
756         }
757     }
758 }
759 
760 /// An instance, pre-instantiation, that is ready to be instantiated.
761 ///
762 /// This structure represents an instance *just before* it was instantiated,
763 /// after all type-checking and imports have been resolved. The only thing left
764 /// to do for this instance is to actually run the process of instantiation.
765 ///
766 /// Note that an `InstancePre` may not be tied to any particular [`Store`] if
767 /// none of the imports it closed over are tied to any particular [`Store`].
768 ///
769 /// This structure is created through the [`Linker::instantiate_pre`] method,
770 /// which also has some more information and examples.
771 ///
772 /// [`Store`]: crate::Store
773 /// [`Linker::instantiate_pre`]: crate::Linker::instantiate_pre
774 pub struct InstancePre<T> {
775     module: Module,
776 
777     /// The items which this `InstancePre` use to instantiate the `module`
778     /// provided, passed to `Instance::new_started` after inserting them into a
779     /// `Store`.
780     ///
781     /// Note that this is stored as an `Arc<[T]>` to quickly move a strong
782     /// reference to everything internally into a `Store<T>` without having to
783     /// clone each individual item.
784     items: Arc<[Definition]>,
785 
786     /// A count of `Definition::HostFunc` entries in `items` above to
787     /// preallocate space in a `Store` up front for all entries to be inserted.
788     host_funcs: usize,
789 
790     /// The `VMFuncRef`s for the functions in `items` that do not
791     /// have a `wasm_call` trampoline. We pre-allocate and pre-patch these
792     /// `VMFuncRef`s so that we don't have to do it at
793     /// instantiation time.
794     ///
795     /// This is an `Arc<[T]>` for the same reason as `items`.
796     func_refs: Arc<[VMFuncRef]>,
797 
798     _marker: core::marker::PhantomData<fn() -> T>,
799 }
800 
801 /// InstancePre's clone does not require T: Clone
802 impl<T> Clone for InstancePre<T> {
803     fn clone(&self) -> Self {
804         Self {
805             module: self.module.clone(),
806             items: self.items.clone(),
807             host_funcs: self.host_funcs,
808             func_refs: self.func_refs.clone(),
809             _marker: self._marker,
810         }
811     }
812 }
813 
814 impl<T: 'static> InstancePre<T> {
815     /// Creates a new `InstancePre` which type-checks the `items` provided and
816     /// on success is ready to instantiate a new instance.
817     ///
818     /// # Unsafety
819     ///
820     /// This method is unsafe as the `T` of the `InstancePre<T>` is not
821     /// guaranteed to be the same as the `T` within the `Store`, the caller must
822     /// verify that.
823     pub(crate) unsafe fn new(module: &Module, items: Vec<Definition>) -> Result<InstancePre<T>> {
824         typecheck(module, &items, |cx, ty, item| cx.definition(ty, &item.ty()))?;
825 
826         let mut func_refs = vec![];
827         let mut host_funcs = 0;
828         for item in &items {
829             match item {
830                 Definition::Extern(_, _) => {}
831                 Definition::HostFunc(f) => {
832                     host_funcs += 1;
833                     if f.func_ref().wasm_call.is_none() {
834                         // `f` needs its `VMFuncRef::wasm_call` patched with a
835                         // Wasm-to-native trampoline.
836                         debug_assert!(matches!(f.host_ctx(), crate::HostContext::Array(_)));
837                         func_refs.push(VMFuncRef {
838                             wasm_call: module
839                                 .wasm_to_array_trampoline(f.sig_index())
840                                 .map(|f| f.into()),
841                             ..*f.func_ref()
842                         });
843                     }
844                 }
845             }
846         }
847 
848         Ok(InstancePre {
849             module: module.clone(),
850             items: items.into(),
851             host_funcs,
852             func_refs: func_refs.into(),
853             _marker: core::marker::PhantomData,
854         })
855     }
856 
857     /// Returns a reference to the module that this [`InstancePre`] will be
858     /// instantiating.
859     pub fn module(&self) -> &Module {
860         &self.module
861     }
862 
863     /// Instantiates this instance, creating a new instance within the provided
864     /// `store`.
865     ///
866     /// This function will run the actual process of instantiation to
867     /// completion. This will use all of the previously-closed-over items as
868     /// imports to instantiate the module that this was originally created with.
869     ///
870     /// For more information about instantiation see [`Instance::new`].
871     ///
872     /// # Panics
873     ///
874     /// Panics if any import closed over by this [`InstancePre`] isn't owned by
875     /// `store`, or if `store` has async support enabled. Additionally this
876     /// function will panic if the `store` provided comes from a different
877     /// [`Engine`] than the [`InstancePre`] originally came from.
878     pub fn instantiate(&self, mut store: impl AsContextMut<Data = T>) -> Result<Instance> {
879         let mut store = store.as_context_mut();
880         let imports = pre_instantiate_raw(
881             &mut store.0,
882             &self.module,
883             &self.items,
884             self.host_funcs,
885             &self.func_refs,
886         )?;
887 
888         // This unsafety should be handled by the type-checking performed by the
889         // constructor of `InstancePre` to assert that all the imports we're passing
890         // in match the module we're instantiating.
891         unsafe { Instance::new_started(&mut store, &self.module, imports.as_ref()) }
892     }
893 
894     /// Creates a new instance, running the start function asynchronously
895     /// instead of inline.
896     ///
897     /// For more information about asynchronous instantiation see the
898     /// documentation on [`Instance::new_async`].
899     ///
900     /// # Panics
901     ///
902     /// Panics if any import closed over by this [`InstancePre`] isn't owned by
903     /// `store`, or if `store` does not have async support enabled.
904     #[cfg(feature = "async")]
905     pub async fn instantiate_async(
906         &self,
907         mut store: impl AsContextMut<Data = T>,
908     ) -> Result<Instance> {
909         let mut store = store.as_context_mut();
910         let imports = pre_instantiate_raw(
911             &mut store.0,
912             &self.module,
913             &self.items,
914             self.host_funcs,
915             &self.func_refs,
916         )?;
917 
918         // This unsafety should be handled by the type-checking performed by the
919         // constructor of `InstancePre` to assert that all the imports we're passing
920         // in match the module we're instantiating.
921         unsafe { Instance::new_started_async(&mut store, &self.module, imports.as_ref()).await }
922     }
923 }
924 
925 /// Helper function shared between
926 /// `InstancePre::{instantiate,instantiate_async}`
927 ///
928 /// This is an out-of-line function to avoid the generic on `InstancePre` and
929 /// get this compiled into the `wasmtime` crate to avoid having it monomorphized
930 /// elsewhere.
931 fn pre_instantiate_raw(
932     store: &mut StoreOpaque,
933     module: &Module,
934     items: &Arc<[Definition]>,
935     host_funcs: usize,
936     func_refs: &Arc<[VMFuncRef]>,
937 ) -> Result<OwnedImports> {
938     // Register this module and use it to fill out any funcref wasm_call holes
939     // we can. For more comments on this see `typecheck_externs`.
940     store.modules_mut().register_module(module);
941     let (funcrefs, modules) = store.func_refs_and_modules();
942     funcrefs.fill(modules);
943 
944     if host_funcs > 0 {
945         // Any linker-defined function of the `Definition::HostFunc` variant
946         // will insert a function into the store automatically as part of
947         // instantiation, so reserve space here to make insertion more efficient
948         // as it won't have to realloc during the instantiation.
949         funcrefs.reserve_storage(host_funcs);
950 
951         // The usage of `to_extern_store_rooted` requires that the items are
952         // rooted via another means, which happens here by cloning the list of
953         // items into the store once. This avoids cloning each individual item
954         // below.
955         funcrefs.push_instance_pre_definitions(items.clone());
956         funcrefs.push_instance_pre_func_refs(func_refs.clone());
957     }
958 
959     let mut func_refs = func_refs.iter().map(|f| NonNull::from(f));
960     let mut imports = OwnedImports::new(module);
961     for import in items.iter() {
962         if !import.comes_from_same_store(store) {
963             bail!("cross-`Store` instantiation is not currently supported");
964         }
965         // This unsafety should be encapsulated in the constructor of
966         // `InstancePre` where the `T` of the original item should match the
967         // `T` of the store. Additionally the rooting necessary has happened
968         // above.
969         let item = match import {
970             Definition::Extern(e, _) => e.clone(),
971             Definition::HostFunc(func) => unsafe {
972                 func.to_func_store_rooted(
973                     store,
974                     if func.func_ref().wasm_call.is_none() {
975                         Some(func_refs.next().unwrap())
976                     } else {
977                         None
978                     },
979                 )
980                 .into()
981             },
982         };
983         imports.push(&item, store);
984     }
985 
986     Ok(imports)
987 }
988 
989 fn typecheck<I>(
990     module: &Module,
991     import_args: &[I],
992     check: impl Fn(&matching::MatchCx<'_>, &EntityType, &I) -> Result<()>,
993 ) -> Result<()> {
994     let env_module = module.compiled_module().module();
995     let expected_len = env_module.imports().count();
996     let actual_len = import_args.len();
997     if expected_len != actual_len {
998         bail!("expected {expected_len} imports, found {actual_len}");
999     }
1000     let cx = matching::MatchCx::new(module.engine());
1001     for ((name, field, expected_ty), actual) in env_module.imports().zip(import_args) {
1002         debug_assert!(expected_ty.is_canonicalized_for_runtime_usage());
1003         check(&cx, &expected_ty, actual)
1004             .with_context(|| format!("incompatible import type for `{name}::{field}`"))?;
1005     }
1006     Ok(())
1007 }
1008