Skip to main content

wasmtime/runtime/
instance.rs

1use crate::linker::{Definition, DefinitionType};
2use crate::prelude::*;
3use crate::runtime::vm::{
4    self, Imports, ModuleRuntimeInfo, UncaughtException, VMFuncRef, VMFunctionImport,
5    VMGlobalImport, VMMemoryImport, VMStore, VMTableImport, VMTagImport,
6};
7use crate::store::{
8    AllocateInstanceKind, Asyncness, InstanceId, StoreInstanceId, StoreOpaque, StoreResourceLimiter,
9};
10use crate::types::matching;
11use crate::{
12    AsContextMut, Engine, Export, Extern, Func, Global, Memory, Module, ModuleExport, SharedMemory,
13    StoreContext, StoreContextMut, Table, Tag, TypedFunc,
14};
15use alloc::sync::Arc;
16use core::ptr::NonNull;
17use wasmtime_environ::{
18    EntityIndex, EntityType, FuncIndex, GlobalIndex, MemoryIndex, TableIndex, TagIndex, 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, PartialEq, Eq)]
35#[repr(C)]
36pub struct Instance {
37    pub(crate) 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.
42const _: () = {
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
50impl 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    /// * The `store` provided requires the use of [`Instance::new_async`]
95    ///   instead, such as if epochs or fuel are configured.
96    ///
97    /// When instantiation fails it's recommended to inspect the return value to
98    /// see why it failed, or bubble it upwards. If you'd like to specifically
99    /// check for trap errors, you can use `error.downcast::<Trap>()`. For more
100    /// about error handling see the [`Trap`] documentation.
101    ///
102    /// [`Trap`]: crate::Trap
103    ///
104    /// # Panics
105    ///
106    /// This function will panic if any [`Extern`] supplied is not owned by
107    /// `store`.
108    ///
109    /// [inst]: https://webassembly.github.io/spec/core/exec/modules.html#exec-instantiation
110    /// [`ExternType`]: crate::ExternType
111    ///
112    /// This function will return an [`OutOfMemory`][crate::OutOfMemory] error when
113    /// memory allocation fails. See the `OutOfMemory` type's documentation for
114    /// details on Wasmtime's out-of-memory handling.
115    pub fn new(
116        mut store: impl AsContextMut,
117        module: &Module,
118        imports: &[Extern],
119    ) -> Result<Instance> {
120        let mut store = store.as_context_mut();
121        store.0.validate_sync_call()?;
122        let imports = Instance::typecheck_externs(store.0, module, imports)?;
123        // Note that the unsafety here should be satisfied by the call to
124        // `typecheck_externs` above which satisfies the condition that all
125        // the imports are valid for this module.
126        vm::assert_ready(unsafe {
127            Instance::new_started(&mut store, module, imports.as_ref(), Asyncness::No)
128        })
129    }
130
131    /// Same as [`Instance::new`], except for usage in [asynchronous stores].
132    ///
133    /// For more details about this function see the documentation on
134    /// [`Instance::new`]. The only difference between these two methods is that
135    /// this one will asynchronously invoke the wasm start function in case it
136    /// calls any imported function which is an asynchronous host function (e.g.
137    /// created with [`Func::new_async`](crate::Func::new_async).
138    ///
139    /// # Panics
140    ///
141    /// This function will panic, like [`Instance::new`], if any [`Extern`]
142    /// specified does not belong to `store`.
143    ///
144    /// # Examples
145    ///
146    /// An example of using this function:
147    ///
148    /// ```
149    /// use wasmtime::{Result, Store, Engine, Module, Instance};
150    ///
151    /// #[tokio::main]
152    /// async fn main() -> Result<()> {
153    ///     let engine = Engine::default();
154    ///
155    ///     // For this example, a module with no imports is being used hence
156    ///     // the empty array to `Instance::new_async`.
157    ///     let module = Module::new(&engine, "(module)")?;
158    ///     let mut store = Store::new(&engine, ());
159    ///     let instance = Instance::new_async(&mut store, &module, &[]).await?;
160    ///
161    ///     // ... use `instance` and exports and such ...
162    ///
163    ///     Ok(())
164    /// }
165    /// ```
166    ///
167    /// Note, though, that the future returned from this function is only
168    /// `Send` if the store's own data is `Send` meaning that this does not
169    /// compile for example:
170    ///
171    /// ```compile_fail
172    /// use wasmtime::{Result, Store, Engine, Module, Instance};
173    /// use std::rc::Rc;
174    ///
175    /// #[tokio::main]
176    /// async fn main() -> Result<()> {
177    ///     let engine = Engine::default();
178    ///
179    ///     let module = Module::new(&engine, "(module)")?;
180    ///
181    ///     // Note that `Rc<()>` is NOT `Send`, which is what many future
182    ///     // runtimes require and below will cause a failure.
183    ///     let mut store = Store::new(&engine, Rc::new(()));
184    ///
185    ///     // Compile failure because `Store<Rc<()>>` is not `Send`
186    ///     assert_send(Instance::new_async(&mut store, &module, &[])).await?;
187    ///
188    ///     Ok(())
189    /// }
190    ///
191    /// fn assert_send<T: Send>(t: T) -> T { t }
192    /// ```
193    ///
194    /// # Errors
195    ///
196    /// This function will return an [`OutOfMemory`][crate::OutOfMemory] error when
197    /// memory allocation fails. See the `OutOfMemory` type's documentation for
198    /// details on Wasmtime's out-of-memory handling.
199    #[cfg(feature = "async")]
200    pub async fn new_async(
201        mut store: impl AsContextMut,
202        module: &Module,
203        imports: &[Extern],
204    ) -> Result<Instance> {
205        let mut store = store.as_context_mut();
206        let imports = Instance::typecheck_externs(store.0, module, imports)?;
207        // See `new` for notes on this unsafety
208        unsafe { Instance::new_started(&mut store, module, imports.as_ref(), Asyncness::Yes).await }
209    }
210
211    fn typecheck_externs(
212        store: &mut StoreOpaque,
213        module: &Module,
214        imports: &[Extern],
215    ) -> Result<OwnedImports> {
216        for import in imports {
217            if !import.comes_from_same_store(store) {
218                bail!("cross-`Store` instantiation is not currently supported");
219            }
220        }
221
222        typecheck(store.engine(), module, imports, |cx, ty, item| {
223            let item = DefinitionType::from(store, item);
224            cx.definition(ty, &item)
225        })?;
226
227        // When pushing functions into `OwnedImports` it's required that their
228        // `wasm_call` fields are all filled out. This `module` is guaranteed
229        // to have any trampolines necessary for functions so register the
230        // module with the store and then attempt to fill out any outstanding
231        // holes.
232        //
233        // Note that under normal operation this shouldn't do much as the list
234        // of funcs-with-holes should generally be empty. As a result the
235        // process of filling this out is not super optimized at this point.
236        let (modules, engine, breakpoints) = store.modules_and_engine_and_breakpoints_mut();
237        modules.register_module(module, engine, breakpoints)?;
238        let (funcrefs, modules) = store.func_refs_and_modules();
239        funcrefs.fill(modules);
240
241        let mut owned_imports = OwnedImports::new(module)?;
242        for import in imports {
243            owned_imports.push(import, store)?;
244        }
245        Ok(owned_imports)
246    }
247
248    /// Internal function to create an instance and run the start function.
249    ///
250    /// This function's unsafety is the same as `Instance::new_raw`.
251    pub(crate) async unsafe fn new_started<T>(
252        store: &mut StoreContextMut<'_, T>,
253        module: &Module,
254        imports: Imports<'_>,
255        asyncness: Asyncness,
256    ) -> Result<Instance> {
257        let (instance, needs_startup) = {
258            let (mut limiter, store) = store.0.resource_limiter_and_store_opaque();
259            // SAFETY: the safety contract of `new_raw` is the same as this
260            // function.
261            unsafe { Instance::new_raw(store, limiter.as_mut(), module, imports).await? }
262        };
263
264        // If this instance requires startup, which is a dynamic decision made
265        // at this point in conjunction with analysis at compile time, the
266        // instance gets started. Note that this isn't just the wasm start
267        // function itself, but it's finalization of initialization of this
268        // instance, for example for complicated global initialization
269        // expressions.
270        if needs_startup {
271            if asyncness == Asyncness::No {
272                instance.start_raw(store, UncaughtException::Propagate)?;
273            } else {
274                #[cfg(feature = "async")]
275                {
276                    store
277                        .on_fiber(|store| instance.start_raw(store, UncaughtException::Propagate))
278                        .await??;
279                }
280                #[cfg(not(feature = "async"))]
281                unreachable!();
282            }
283        }
284        Ok(instance)
285    }
286
287    /// Internal function to create an instance which doesn't have its `start`
288    /// function run yet.
289    ///
290    /// # Unsafety
291    ///
292    /// This method is unsafe because it does not type-check the `imports`
293    /// provided. The `imports` provided must be suitable for the module
294    /// provided as well.
295    pub(crate) async unsafe fn new_raw(
296        store: &mut StoreOpaque,
297        mut limiter: Option<&mut StoreResourceLimiter<'_>>,
298        module: &Module,
299        imports: Imports<'_>,
300    ) -> Result<(Instance, bool)> {
301        if !Engine::same(store.engine(), module.engine()) {
302            bail!("cross-`Engine` instantiation is not currently supported");
303        }
304        store.bump_resource_counts(module)?;
305
306        // Allocate the GC heap, if necessary.
307        if module.env_module().needs_gc_heap {
308            store.ensure_gc_store(limiter.as_deref_mut()).await?;
309        }
310
311        // Register the module just before instantiation to ensure we keep the module
312        // properly referenced while in use by the store.
313        let (modules, engine, breakpoints) = store.modules_and_engine_and_breakpoints_mut();
314        let module_id = modules.register_module(module, engine, breakpoints)?;
315
316        // The first thing we do is issue an instance allocation request
317        // to the instance allocator. This, on success, will give us an
318        // instance handle.
319        //
320        // SAFETY: this module, by construction, was already validated within
321        // the store.
322        let id = unsafe {
323            store
324                .allocate_instance(
325                    limiter.as_deref_mut(),
326                    AllocateInstanceKind::Module(module_id),
327                    &ModuleRuntimeInfo::Module(module.clone()),
328                    imports,
329                )
330                .await?
331        };
332
333        let instance = Instance::from_wasmtime(id, store);
334
335        let needs_startup = instance.id.get_mut(store).needs_startup();
336
337        // At this point the instance is created and stored within the store,
338        // but it's also not quite usable just yet. Initialization hasn't
339        // completed (e.g. active data/element segments) and the `start`
340        // function additionally has not yet been invoked. That's the
341        // responsibility of the caller to handle, however.
342        Ok((instance, needs_startup))
343    }
344
345    pub(crate) fn from_wasmtime(id: InstanceId, store: &mut StoreOpaque) -> Instance {
346        Instance {
347            id: StoreInstanceId::new(store.id(), id),
348        }
349    }
350
351    pub(crate) fn start_raw<T>(
352        &self,
353        store: &mut StoreContextMut<'_, T>,
354        uncaught_exception: UncaughtException,
355    ) -> Result<()> {
356        // If a start function is present, invoke it. Make sure we use all the
357        // trap-handling configuration in `store` as well.
358        let store_id = store.0.id();
359        let (mut instance, registry) = self.id.get_mut_and_module_registry(store.0);
360        // SAFETY: the `store_id` is the id of the store that owns this
361        // instance and any function stored within the instance.
362        let f = unsafe {
363            instance
364                .as_mut()
365                .get_startup_func(registry, store_id)
366                .expect("should have a startup function")
367        };
368        let caller_vmctx = instance.vmctx();
369        unsafe {
370            let funcref = f.vm_func_ref(store.0);
371            super::func::invoke_wasm_and_catch_traps(
372                store,
373                uncaught_exception,
374                |_default_caller, vm| {
375                    VMFuncRef::array_call(funcref, vm, caller_vmctx, NonNull::from(&mut []))
376                },
377            )?;
378        }
379        Ok(())
380    }
381
382    /// Get this instance's module.
383    pub fn module<'a, T: 'static>(&self, store: impl Into<StoreContext<'a, T>>) -> &'a Module {
384        self._module(store.into().0)
385    }
386
387    pub(crate) fn _module<'a>(&self, store: &'a StoreOpaque) -> &'a Module {
388        store.module_for_instance(self.id).unwrap()
389    }
390
391    /// Returns the list of exported items from this [`Instance`].
392    ///
393    /// # Panics
394    ///
395    /// Panics if `store` does not own this instance, or if memory allocation
396    /// fails.
397    pub fn exports<'a, T: 'static>(
398        &'a self,
399        store: impl Into<StoreContextMut<'a, T>>,
400    ) -> impl ExactSizeIterator<Item = Export<'a>> + 'a {
401        let store = store.into().0;
402        let store_id = store.id();
403        let engine = store.engine().clone();
404
405        let (instance, registry) = store.instance_and_module_registry_mut(self.id());
406        let (module, mut instance) = instance.module_and_self();
407        module.exports.iter().map(move |(name, entity)| {
408            // SAFETY: the `store_id` owns this instance and all exports
409            // contained within.
410            let export = unsafe {
411                instance
412                    .as_mut()
413                    .get_export_by_index_mut(registry, store_id, *entity)
414            };
415
416            let ext = Extern::from_wasmtime_export(export, &engine);
417            Export::new(&module.strings[name], ext)
418        })
419    }
420
421    /// Looks up an exported [`Extern`] value by name.
422    ///
423    /// This method will search the module for an export named `name` and return
424    /// the value, if found.
425    ///
426    /// Returns `None` if there was no export named `name`.
427    ///
428    /// # Panics
429    ///
430    /// Panics if `store` does not own this instance.
431    ///
432    /// # Why does `get_export` take a mutable context?
433    ///
434    /// This method requires a mutable context because an instance's exports are
435    /// lazily populated, and we cache them as they are accessed. This makes
436    /// instantiating a module faster, but also means this method requires a
437    /// mutable context.
438    pub fn get_export(&self, mut store: impl AsContextMut, name: &str) -> Option<Extern> {
439        let store = store.as_context_mut().0;
440        let module = store[self.id].env_module();
441        let name = module.strings.get_atom(name)?;
442        let entity = *module.exports.get(&name)?;
443        Some(self._get_export(store, entity))
444    }
445
446    /// Looks up an exported [`Extern`] value by a [`ModuleExport`] value.
447    ///
448    /// This is similar to [`Instance::get_export`] but uses a [`ModuleExport`] value to avoid
449    /// string lookups where possible. [`ModuleExport`]s can be obtained by calling
450    /// [`Module::get_export_index`] on the [`Module`] that this instance was instantiated with.
451    ///
452    /// This method will search the module for an export with a matching entity index and return
453    /// the value, if found.
454    ///
455    /// Returns `None` if there was no export with a matching entity index.
456    ///
457    /// # Panics
458    ///
459    /// Panics if `store` does not own this instance.
460    pub fn get_module_export(
461        &self,
462        mut store: impl AsContextMut,
463        export: &ModuleExport,
464    ) -> Option<Extern> {
465        let store = store.as_context_mut().0;
466
467        // Verify the `ModuleExport` matches the module used in this instance.
468        if self._module(store).id() != export.module {
469            return None;
470        }
471
472        Some(self._get_export(store, export.entity))
473    }
474
475    fn _get_export(&self, store: &mut StoreOpaque, entity: EntityIndex) -> Extern {
476        let id = store.id();
477        // SAFETY: the store `id` owns this instance and all exports contained
478        // within.
479        let export = unsafe {
480            let (instance, registry) = self.id.get_mut_and_module_registry(store);
481            instance.get_export_by_index_mut(registry, id, entity)
482        };
483        Extern::from_wasmtime_export(export, store.engine())
484    }
485
486    /// Looks up an exported [`Func`] value by name.
487    ///
488    /// Returns `None` if there was no export named `name`, or if there was but
489    /// it wasn't a function.
490    ///
491    /// # Panics
492    ///
493    /// Panics if `store` does not own this instance.
494    pub fn get_func(&self, store: impl AsContextMut, name: &str) -> Option<Func> {
495        self.get_export(store, name)?.into_func()
496    }
497
498    /// Looks up an exported [`Func`] value by name and with its type.
499    ///
500    /// This function is a convenience wrapper over [`Instance::get_func`] and
501    /// [`Func::typed`]. For more information see the linked documentation.
502    ///
503    /// Returns an error if `name` isn't a function export or if the export's
504    /// type did not match `Params` or `Results`
505    ///
506    /// # Panics
507    ///
508    /// Panics if `store` does not own this instance.
509    ///
510    /// # Errors
511    ///
512    /// This function will return an [`OutOfMemory`][crate::OutOfMemory] error when
513    /// memory allocation fails. See the `OutOfMemory` type's documentation for
514    /// details on Wasmtime's out-of-memory handling.
515    pub fn get_typed_func<Params, Results>(
516        &self,
517        mut store: impl AsContextMut,
518        name: &str,
519    ) -> Result<TypedFunc<Params, Results>>
520    where
521        Params: crate::WasmParams,
522        Results: crate::WasmResults,
523    {
524        let f = self
525            .get_export(store.as_context_mut(), name)
526            .and_then(|f| f.into_func())
527            .ok_or_else(|| format_err!("failed to find function export `{name}`"))?;
528        Ok(f.typed::<Params, Results>(store)
529            .with_context(|| format!("failed to convert function `{name}` to given type"))?)
530    }
531
532    /// Looks up an exported [`Table`] value by name.
533    ///
534    /// Returns `None` if there was no export named `name`, or if there was but
535    /// it wasn't a table.
536    ///
537    /// # Panics
538    ///
539    /// Panics if `store` does not own this instance.
540    pub fn get_table(&self, store: impl AsContextMut, name: &str) -> Option<Table> {
541        self.get_export(store, name)?.into_table()
542    }
543
544    /// Looks up an exported [`Memory`] value by name.
545    ///
546    /// Returns `None` if there was no export named `name`, or if there was but
547    /// it wasn't a memory.
548    ///
549    /// # Panics
550    ///
551    /// Panics if `store` does not own this instance.
552    pub fn get_memory(&self, store: impl AsContextMut, name: &str) -> Option<Memory> {
553        self.get_export(store, name)?.into_memory()
554    }
555
556    /// Looks up an exported [`SharedMemory`] value by name.
557    ///
558    /// Returns `None` if there was no export named `name`, or if there was but
559    /// it wasn't a shared memory.
560    ///
561    /// # Panics
562    ///
563    /// Panics if `store` does not own this instance.
564    pub fn get_shared_memory(
565        &self,
566        mut store: impl AsContextMut,
567        name: &str,
568    ) -> Option<SharedMemory> {
569        let mut store = store.as_context_mut();
570        self.get_export(&mut store, name)?.into_shared_memory()
571    }
572
573    /// Looks up an exported [`Global`] value by name.
574    ///
575    /// Returns `None` if there was no export named `name`, or if there was but
576    /// it wasn't a global.
577    ///
578    /// # Panics
579    ///
580    /// Panics if `store` does not own this instance.
581    pub fn get_global(&self, store: impl AsContextMut, name: &str) -> Option<Global> {
582        self.get_export(store, name)?.into_global()
583    }
584
585    /// Looks up a tag [`Tag`] by name.
586    ///
587    /// Returns `None` if there was no export named `name`, or if there was but
588    /// it wasn't a tag.
589    ///
590    /// # Panics
591    ///
592    /// Panics if `store` does not own this instance.
593    pub fn get_tag(&self, store: impl AsContextMut, name: &str) -> Option<Tag> {
594        self.get_export(store, name)?.into_tag()
595    }
596
597    #[allow(
598        dead_code,
599        reason = "c-api crate does not yet support exnrefs and causes this method to be dead."
600    )]
601    pub(crate) fn id(&self) -> InstanceId {
602        self.id.instance()
603    }
604
605    /// Return a unique-within-Store index for this `Instance`.
606    ///
607    /// Allows distinguishing instance identities when introspecting
608    /// the `Store`, e.g. via debug APIs.
609    ///
610    /// This index will match the instance's position in the sequence
611    /// returned by `Store::debug_all_instances()`.
612    #[cfg(feature = "debug")]
613    pub fn debug_index_in_store(&self) -> u32 {
614        self.id.instance().as_u32()
615    }
616
617    /// Get all globals within this instance.
618    ///
619    /// Returns both import and defined globals.
620    ///
621    /// Returns both exported and non-exported globals.
622    ///
623    /// Gives access to the full globals space.
624    #[cfg(feature = "coredump")]
625    pub(crate) fn all_globals<'a>(
626        &'a self,
627        store: &'a mut StoreOpaque,
628    ) -> impl ExactSizeIterator<Item = (GlobalIndex, Global)> + 'a {
629        let store_id = store.id();
630        store[self.id].all_globals(store_id)
631    }
632
633    /// Get all memories within this instance.
634    ///
635    /// Returns both import and defined memories.
636    ///
637    /// Returns both exported and non-exported memories.
638    ///
639    /// Gives access to the full memories space.
640    #[cfg(feature = "coredump")]
641    pub(crate) fn all_memories<'a>(
642        &'a self,
643        store: &'a StoreOpaque,
644    ) -> impl ExactSizeIterator<Item = (MemoryIndex, vm::ExportMemory)> + 'a {
645        let store_id = store.id();
646        store[self.id].all_memories(store_id)
647    }
648}
649
650pub(crate) struct OwnedImports {
651    functions: TryPrimaryMap<FuncIndex, VMFunctionImport>,
652    tables: TryPrimaryMap<TableIndex, VMTableImport>,
653    memories: TryPrimaryMap<MemoryIndex, VMMemoryImport>,
654    globals: TryPrimaryMap<GlobalIndex, VMGlobalImport>,
655    tags: TryPrimaryMap<TagIndex, VMTagImport>,
656}
657
658impl OwnedImports {
659    fn new(module: &Module) -> Result<OwnedImports, OutOfMemory> {
660        let mut ret = OwnedImports::empty();
661        ret.reserve(module)?;
662        Ok(ret)
663    }
664
665    pub(crate) fn empty() -> OwnedImports {
666        OwnedImports {
667            functions: TryPrimaryMap::new(),
668            tables: TryPrimaryMap::new(),
669            memories: TryPrimaryMap::new(),
670            globals: TryPrimaryMap::new(),
671            tags: TryPrimaryMap::new(),
672        }
673    }
674
675    pub(crate) fn reserve(&mut self, module: &Module) -> Result<(), OutOfMemory> {
676        let raw = module.compiled_module().module();
677        self.functions.reserve(raw.num_imported_funcs)?;
678        self.tables.reserve(raw.num_imported_tables)?;
679        self.memories.reserve(raw.num_imported_memories)?;
680        self.globals.reserve(raw.num_imported_globals)?;
681        self.tags.reserve(raw.num_imported_tags)?;
682        Ok(())
683    }
684
685    #[cfg(feature = "component-model")]
686    pub(crate) fn clear(&mut self) {
687        self.functions.clear();
688        self.tables.clear();
689        self.memories.clear();
690        self.globals.clear();
691        self.tags.clear();
692    }
693
694    fn push(&mut self, item: &Extern, store: &mut StoreOpaque) -> Result<(), OutOfMemory> {
695        match item {
696            Extern::Func(i) => {
697                self.functions.push(i.vmimport(store))?;
698            }
699            Extern::Global(i) => {
700                self.globals.push(i.vmimport(store))?;
701            }
702            Extern::Table(i) => {
703                self.tables.push(i.vmimport(store))?;
704            }
705            Extern::Memory(i) => {
706                self.memories.push(i.vmimport(store))?;
707            }
708            Extern::SharedMemory(i) => {
709                self.memories.push(i.vmimport(store))?;
710            }
711            Extern::Tag(i) => {
712                self.tags.push(i.vmimport(store))?;
713            }
714        }
715        Ok(())
716    }
717
718    /// Note that this is unsafe as the validity of `item` is not verified and
719    /// it contains a bunch of raw pointers.
720    #[cfg(feature = "component-model")]
721    pub(crate) fn push_export(
722        &mut self,
723        store: &StoreOpaque,
724        item: &crate::runtime::vm::Export,
725    ) -> Result<(), OutOfMemory> {
726        match item {
727            crate::runtime::vm::Export::Function(f) => {
728                self.functions.push(f.vmimport(store))?;
729            }
730            crate::runtime::vm::Export::Global(g) => {
731                self.globals.push(g.vmimport(store))?;
732            }
733            crate::runtime::vm::Export::Table(t) => {
734                self.tables.push(t.vmimport(store))?;
735            }
736            crate::runtime::vm::Export::Memory(m) => {
737                self.memories.push(m.vmimport(store))?;
738            }
739            crate::runtime::vm::Export::SharedMemory(_, vmimport) => {
740                self.memories.push(*vmimport)?;
741            }
742            crate::runtime::vm::Export::Tag(t) => {
743                self.tags.push(t.vmimport(store))?;
744            }
745        }
746        Ok(())
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
774pub 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` to quickly move a strong reference
782    /// to everything internally into a `Store<T>` without having to clone each
783    /// individual item.
784    items: Arc<TryVec<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` for the same reason as `items`.
796    func_refs: Arc<TryVec<VMFuncRef>>,
797
798    /// Whether or not any import in `items` is flagged as needing async.
799    ///
800    /// This is used to update stores during instantiation as to whether they
801    /// require async entrypoints.
802    asyncness: Asyncness,
803
804    _marker: core::marker::PhantomData<fn() -> T>,
805}
806
807/// InstancePre's clone does not require T: Clone
808impl<T> Clone for InstancePre<T> {
809    fn clone(&self) -> Self {
810        Self {
811            module: self.module.clone(),
812            items: self.items.clone(),
813            host_funcs: self.host_funcs,
814            func_refs: self.func_refs.clone(),
815            asyncness: self.asyncness,
816            _marker: self._marker,
817        }
818    }
819}
820
821impl<T: 'static> InstancePre<T> {
822    /// Creates a new `InstancePre` which type-checks the `items` provided and
823    /// on success is ready to instantiate a new instance.
824    ///
825    /// `engine` is the engine that `items` belong to, and this returns an error
826    /// if that is not also `module`'s engine. This also returns an error if an
827    /// individual item within `items` reports an engine of its own that is not
828    /// `engine`, which happens when that item was taken from a store belonging
829    /// to a different engine than the linker it was defined in.
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(
837        engine: &Engine,
838        module: &Module,
839        items: TryVec<Definition>,
840    ) -> Result<InstancePre<T>> {
841        typecheck(engine, module, &items, |cx, ty, item| {
842            cx.definition(ty, &item.ty())
843        })?;
844
845        let mut func_refs = TryVec::with_capacity(items.len())?;
846        let mut host_funcs = 0;
847        let mut asyncness = Asyncness::No;
848        for item in &items {
849            match item {
850                Definition::Extern { .. } => {}
851                Definition::HostFunc(f) => {
852                    host_funcs += 1;
853                    if f.func_ref().wasm_call.is_none() {
854                        func_refs.push(VMFuncRef {
855                            wasm_call: module
856                                .wasm_to_array_trampoline(f.sig_index())
857                                .map(|f| f.into()),
858                            ..*f.func_ref()
859                        })?;
860                    }
861                    asyncness = asyncness | f.asyncness();
862                }
863            }
864        }
865
866        Ok(InstancePre {
867            module: module.clone(),
868            items: try_new::<Arc<_>>(items)?,
869            host_funcs,
870            func_refs: try_new::<Arc<_>>(func_refs)?,
871            asyncness,
872            _marker: core::marker::PhantomData,
873        })
874    }
875
876    /// Returns a reference to the module that this [`InstancePre`] will be
877    /// instantiating.
878    pub fn module(&self) -> &Module {
879        &self.module
880    }
881
882    /// Instantiates this instance, creating a new instance within the provided
883    /// `store`.
884    ///
885    /// This function will run the actual process of instantiation to
886    /// completion. This will use all of the previously-closed-over items as
887    /// imports to instantiate the module that this was originally created with.
888    ///
889    /// For more information about instantiation see [`Instance::new`].
890    ///
891    /// # Panics
892    ///
893    /// Panics if any import closed over by this [`InstancePre`] isn't owned by
894    /// `store`, or if `store` has async support enabled. Additionally this
895    /// function will panic if the `store` provided comes from a different
896    /// [`Engine`] than the [`InstancePre`] originally came from.
897    ///
898    /// # Errors
899    ///
900    /// This function will return an [`OutOfMemory`][crate::OutOfMemory] error when
901    /// memory allocation fails. See the `OutOfMemory` type's documentation for
902    /// details on Wasmtime's out-of-memory handling.
903    pub fn instantiate(&self, mut store: impl AsContextMut<Data = T>) -> Result<Instance> {
904        let mut store = store.as_context_mut();
905        let imports = pre_instantiate_raw(
906            &mut store.0,
907            &self.module,
908            &self.items,
909            self.host_funcs,
910            &self.func_refs,
911            self.asyncness,
912        )?;
913
914        // Note that this is specifically done after `pre_instantiate_raw` to
915        // handle the case that if any imports in this `InstancePre` require
916        // async that it's flagged in the store by that point which will reject
917        // this instantiation to say "use `instantiate_async` instead".
918        store.0.validate_sync_call()?;
919
920        // This unsafety should be handled by the type-checking performed by the
921        // constructor of `InstancePre` to assert that all the imports we're passing
922        // in match the module we're instantiating.
923        vm::assert_ready(unsafe {
924            Instance::new_started(&mut store, &self.module, imports.as_ref(), Asyncness::No)
925        })
926    }
927
928    /// Creates a new instance, running the start function asynchronously
929    /// instead of inline.
930    ///
931    /// For more information about asynchronous instantiation see the
932    /// documentation on [`Instance::new_async`].
933    ///
934    /// # Panics
935    ///
936    /// Panics if any import closed over by this [`InstancePre`] isn't owned by
937    /// `store`, or if `store` does not have async support enabled.
938    ///
939    /// # Errors
940    ///
941    /// This function will return an [`OutOfMemory`][crate::OutOfMemory] error when
942    /// memory allocation fails. See the `OutOfMemory` type's documentation for
943    /// details on Wasmtime's out-of-memory handling.
944    #[cfg(feature = "async")]
945    pub async fn instantiate_async(
946        &self,
947        mut store: impl AsContextMut<Data = T>,
948    ) -> Result<Instance> {
949        let mut store = store.as_context_mut();
950        let imports = pre_instantiate_raw(
951            &mut store.0,
952            &self.module,
953            &self.items,
954            self.host_funcs,
955            &self.func_refs,
956            self.asyncness,
957        )?;
958
959        // This unsafety should be handled by the type-checking performed by the
960        // constructor of `InstancePre` to assert that all the imports we're passing
961        // in match the module we're instantiating.
962        unsafe {
963            Instance::new_started(&mut store, &self.module, imports.as_ref(), Asyncness::Yes).await
964        }
965    }
966}
967
968/// Helper function shared between
969/// `InstancePre::{instantiate,instantiate_async}`
970///
971/// This is an out-of-line function to avoid the generic on `InstancePre` and
972/// get this compiled into the `wasmtime` crate to avoid having it monomorphized
973/// elsewhere.
974fn pre_instantiate_raw(
975    store: &mut StoreOpaque,
976    module: &Module,
977    items: &Arc<TryVec<Definition>>,
978    host_funcs: usize,
979    func_refs: &Arc<TryVec<VMFuncRef>>,
980    asyncness: Asyncness,
981) -> Result<OwnedImports> {
982    // Register this module and use it to fill out any funcref wasm_call holes
983    // we can. For more comments on this see `typecheck_externs`.
984    let (modules, engine, breakpoints) = store.modules_and_engine_and_breakpoints_mut();
985    modules.register_module(module, engine, breakpoints)?;
986    let (funcrefs, modules) = store.func_refs_and_modules();
987    funcrefs.fill(modules);
988
989    if host_funcs > 0 {
990        // Any linker-defined function of the `Definition::HostFunc` variant
991        // will insert a function into the store automatically as part of
992        // instantiation, so reserve space here to make insertion more efficient
993        // as it won't have to realloc during the instantiation.
994        funcrefs.reserve_storage(host_funcs)?;
995
996        // The usage of `to_extern_store_rooted` requires that the items are
997        // rooted via another means, which happens here by cloning the list of
998        // items into the store once. This avoids cloning each individual item
999        // below.
1000        funcrefs.push_instance_pre_definitions(items.clone())?;
1001        funcrefs.push_instance_pre_func_refs(func_refs.clone())?;
1002    }
1003
1004    store.set_async_required(asyncness);
1005
1006    let mut func_refs = func_refs.iter().map(|f| NonNull::from(f));
1007    let mut imports = OwnedImports::new(module)?;
1008    for import in items.iter() {
1009        if !import.comes_from_same_store(store) {
1010            bail!("cross-`Store` instantiation is not currently supported");
1011        }
1012        // This unsafety should be encapsulated in the constructor of
1013        // `InstancePre` where the `T` of the original item should match the
1014        // `T` of the store. Additionally the rooting necessary has happened
1015        // above.
1016        let item = match import {
1017            Definition::Extern { item, .. } => item.clone(),
1018            Definition::HostFunc(func) => unsafe {
1019                func.to_func_store_rooted(
1020                    store,
1021                    if func.func_ref().wasm_call.is_none() {
1022                        Some(func_refs.next().unwrap())
1023                    } else {
1024                        None
1025                    },
1026                )
1027                .into()
1028            },
1029        };
1030        imports.push(&item, store)?;
1031    }
1032
1033    Ok(imports)
1034}
1035
1036/// An item that can be supplied as an import argument during instantiation.
1037///
1038/// # Safety
1039///
1040/// Implementations must return an associated engine if they own a handle to
1041/// one. Failure to do so may allow cross-`Engine` type confusion.
1042///
1043/// (Items that are just identifiers indexing into a store, for example
1044/// `Extern::Global(wasmtime::Global)`, do not have their own handle to an
1045/// engine. Their engine is the engine of the store they belong to, and it is
1046/// the store, not them, that holds an owning handle to the engine.)
1047unsafe trait ImportArg {
1048    fn engine(&self) -> Option<&Engine>;
1049}
1050
1051// SAFETY: `Extern::SharedMemory` is the only variant with an `Engine` handle.
1052unsafe impl ImportArg for Extern {
1053    fn engine(&self) -> Option<&Engine> {
1054        match self {
1055            Extern::SharedMemory(m) => Some(m.engine()),
1056            Extern::Func(_)
1057            | Extern::Global(_)
1058            | Extern::Table(_)
1059            | Extern::Memory(_)
1060            | Extern::Tag(_) => None,
1061        }
1062    }
1063}
1064
1065// SAFETY: `Definition::engine` is complete.
1066unsafe impl ImportArg for Definition {
1067    fn engine(&self) -> Option<&Engine> {
1068        Some(Definition::engine(self))
1069    }
1070}
1071
1072/// Type check the `import_args` against the imports that `module` declares.
1073///
1074/// `engine` is the engine that the `import_args` belong to. It must be the same
1075/// engine as `module`'s: entity types are compared by `VMSharedTypeIndex`, which
1076/// only means anything within the engine that assigned it, so checking one
1077/// engine's items against another engine's module would compare unrelated types
1078/// and consider them equal.
1079fn typecheck<I>(
1080    engine: &Engine,
1081    module: &Module,
1082    import_args: &[I],
1083    check: impl Fn(&matching::MatchCx<'_>, &EntityType, &I) -> Result<()>,
1084) -> Result<()>
1085where
1086    I: ImportArg,
1087{
1088    ensure!(
1089        Engine::same(engine, module.engine()),
1090        "cross-`Engine` instantiation is not currently supported"
1091    );
1092    let env_module = module.compiled_module().module();
1093    let expected_len = env_module.imports().count();
1094    let actual_len = import_args.len();
1095    if expected_len != actual_len {
1096        bail!("expected {expected_len} imports, found {actual_len}");
1097    }
1098    let cx = matching::MatchCx::new(module.engine());
1099    for ((name, field, expected_ty), actual) in env_module.imports().zip(import_args) {
1100        debug_assert!(expected_ty.is_canonicalized_for_runtime_usage());
1101        if let Some(actual_engine) = actual.engine() {
1102            ensure!(
1103                Engine::same(actual_engine, engine),
1104                "cross-`Engine` instantiation is not currently supported: \
1105                 the item provided for `{name}::{field}` belongs to a \
1106                 different engine than the module being instantiated"
1107            );
1108        }
1109        check(&cx, &expected_ty, actual)
1110            .with_context(|| format!("incompatible import type for `{name}::{field}`"))?;
1111    }
1112    Ok(())
1113}