wasmtime/runtime/
instance.rs

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