Skip to main content

wasmtime/runtime/component/
instance.rs

1use crate::component::func::HostFunc;
2use crate::component::matching::InstanceType;
3use crate::component::store::{ComponentInstanceId, StoreComponentInstanceId};
4use crate::component::{
5    Component, ComponentExportIndex, ComponentNamedList, Func, Lift, Lower, ResourceType,
6    TypedFunc, types::ComponentItem,
7};
8use crate::component::{ExportLookup, RuntimeInstance};
9use crate::instance::OwnedImports;
10use crate::linker::DefinitionType;
11use crate::prelude::*;
12use crate::runtime::vm::component::{ComponentInstance, TypedResource, TypedResourceIndex};
13use crate::runtime::vm::{self, VMFuncRef};
14use crate::store::{AsStoreOpaque, Asyncness, StoreOpaque};
15use crate::{AsContext, AsContextMut, Engine, Module, StoreContextMut};
16use alloc::sync::Arc;
17use core::marker;
18use core::pin::Pin;
19use core::ptr::NonNull;
20use wasmtime_environ::{EngineOrModuleTypeIndex, component::*};
21use wasmtime_environ::{EntityIndex, EntityType, PrimaryMap};
22
23/// An instantiated component.
24///
25/// This type represents an instantiated [`Component`](super::Component).
26/// Instances have exports which can be accessed through functions such as
27/// [`Instance::get_func`] or [`Instance::get_export`]. Instances are owned by a
28/// [`Store`](crate::Store) and all methods require a handle to the store.
29///
30/// Component instances are created through
31/// [`Linker::instantiate`](super::Linker::instantiate) and its family of
32/// methods.
33///
34/// This type is similar to the core wasm version
35/// [`wasmtime::Instance`](crate::Instance) except that it represents an
36/// instantiated component instead of an instantiated module.
37#[derive(Copy, Clone, Debug)]
38#[repr(transparent)]
39pub struct Instance {
40    id: StoreComponentInstanceId,
41}
42
43// Double-check that the C representation in `component/instance.h` matches our
44// in-Rust representation here in terms of size/alignment/etc.
45const _: () = {
46    #[repr(C)]
47    struct C(u64, u32);
48    assert!(core::mem::size_of::<C>() == core::mem::size_of::<Instance>());
49    assert!(core::mem::align_of::<C>() == core::mem::align_of::<Instance>());
50    assert!(core::mem::offset_of!(Instance, id) == 0);
51};
52
53impl Instance {
54    /// Creates a raw `Instance` from the internal identifiers within the store.
55    pub(crate) fn from_wasmtime(store: &StoreOpaque, id: ComponentInstanceId) -> Instance {
56        Instance {
57            id: StoreComponentInstanceId::new(store.id(), id),
58        }
59    }
60
61    /// Looks up an exported function by name within this [`Instance`].
62    ///
63    /// The `store` argument provided must be the store that this instance
64    /// lives within and the `name` argument is the lookup key by which to find
65    /// the exported function. If the function is found then `Some` is returned
66    /// and otherwise `None` is returned.
67    ///
68    /// The `name` here can be a string such as `&str` or it can be a
69    /// [`ComponentExportIndex`] which is loaded prior from a [`Component`].
70    ///
71    /// # Panics
72    ///
73    /// Panics if `store` does not own this instance.
74    ///
75    /// # Examples
76    ///
77    /// Looking up a function which is exported from the root of a component:
78    ///
79    /// ```
80    /// use wasmtime::{Engine, Store};
81    /// use wasmtime::component::{Component, Linker};
82    ///
83    /// # fn main() -> wasmtime::Result<()> {
84    /// let engine = Engine::default();
85    /// let component = Component::new(
86    ///     &engine,
87    ///     r#"
88    ///         (component
89    ///             (core module $m
90    ///                 (func (export "f"))
91    ///             )
92    ///             (core instance $i (instantiate $m))
93    ///             (func (export "f")
94    ///                 (canon lift (core func $i "f")))
95    ///         )
96    ///     "#,
97    /// )?;
98    ///
99    /// // Look up the function by name
100    /// let mut store = Store::new(&engine, ());
101    /// let instance = Linker::new(&engine).instantiate(&mut store, &component)?;
102    /// let func = instance.get_func(&mut store, "f").unwrap();
103    ///
104    /// // The function can also be looked up by an index via a precomputed index.
105    /// let export = component.get_export_index(None, "f").unwrap();
106    /// let func = instance.get_func(&mut store, &export).unwrap();
107    /// # Ok(())
108    /// # }
109    /// ```
110    ///
111    /// Looking up a function which is exported from a nested instance:
112    ///
113    /// ```
114    /// use wasmtime::{Engine, Store};
115    /// use wasmtime::component::{Component, Linker};
116    ///
117    /// # fn main() -> wasmtime::Result<()> {
118    /// let engine = Engine::default();
119    /// let component = Component::new(
120    ///     &engine,
121    ///     r#"
122    ///         (component
123    ///             (core module $m
124    ///                 (func (export "f"))
125    ///             )
126    ///             (core instance $i (instantiate $m))
127    ///             (func $f
128    ///                 (canon lift (core func $i "f")))
129    ///
130    ///             (instance $i
131    ///                 (export "f" (func $f)))
132    ///             (export "i" (instance $i))
133    ///         )
134    ///     "#,
135    /// )?;
136    ///
137    /// // First look up the exported instance, then use that to lookup the
138    /// // exported function.
139    /// let instance_index = component.get_export_index(None, "i").unwrap();
140    /// let func_index = component.get_export_index(Some(&instance_index), "f").unwrap();
141    ///
142    /// // Then use `func_index` at runtime.
143    /// let mut store = Store::new(&engine, ());
144    /// let instance = Linker::new(&engine).instantiate(&mut store, &component)?;
145    /// let func = instance.get_func(&mut store, &func_index).unwrap();
146    ///
147    /// // Alternatively the `instance` can be used directly in conjunction with
148    /// // the `get_export_index` method.
149    /// let instance_index = instance.get_export_index(&mut store, None, "i").unwrap();
150    /// let func_index = instance.get_export_index(&mut store, Some(&instance_index), "f").unwrap();
151    /// let func = instance.get_func(&mut store, &func_index).unwrap();
152    /// # Ok(())
153    /// # }
154    /// ```
155    pub fn get_func(&self, mut store: impl AsContextMut, name: impl ExportLookup) -> Option<Func> {
156        let store = store.as_context_mut().0;
157        let instance = self.id.get(store);
158        let component = instance.component();
159
160        // Validate that `name` exists within `self.`
161        let index = name.lookup(component, None)?;
162
163        // Validate that `index` is indeed a lifted function.
164        match &component.env_component().export_items[index] {
165            Export::LiftedFunction { .. } => {}
166            _ => return None,
167        }
168
169        // And package up the indices!
170        Some(Func::from_lifted_func(*self, index))
171    }
172
173    /// Looks up an exported [`Func`] value by name and with its type.
174    ///
175    /// This function is a convenience wrapper over [`Instance::get_func`] and
176    /// [`Func::typed`]. For more information see the linked documentation.
177    ///
178    /// Returns an error if `name` isn't a function export or if the export's
179    /// type did not match `Params` or `Results`
180    ///
181    /// This function will return an [`OutOfMemory`][crate::OutOfMemory] error when
182    /// memory allocation fails. See the `OutOfMemory` type's documentation for
183    /// details on Wasmtime's out-of-memory handling.
184    ///
185    /// # Panics
186    ///
187    /// Panics if `store` does not own this instance.
188    pub fn get_typed_func<Params, Results>(
189        &self,
190        mut store: impl AsContextMut,
191        name: impl ExportLookup,
192    ) -> Result<TypedFunc<Params, Results>>
193    where
194        Params: ComponentNamedList + Lower,
195        Results: ComponentNamedList + Lift,
196    {
197        let f = self
198            .get_func(store.as_context_mut(), name)
199            .ok_or_else(|| format_err!("failed to find function export"))?;
200        Ok(f.typed::<Params, Results>(store)
201            .with_context(|| format!("failed to convert function to given type"))?)
202    }
203
204    /// Looks up an exported module by name within this [`Instance`].
205    ///
206    /// The `store` argument provided must be the store that this instance
207    /// lives within and the `name` argument is the lookup key by which to find
208    /// the exported module. If the module is found then `Some` is returned
209    /// and otherwise `None` is returned.
210    ///
211    /// The `name` here can be a string such as `&str` or it can be a
212    /// [`ComponentExportIndex`] which is loaded prior from a [`Component`].
213    ///
214    /// For some examples see [`Instance::get_func`] for loading values from a
215    /// component.
216    ///
217    /// # Panics
218    ///
219    /// Panics if `store` does not own this instance.
220    pub fn get_module(
221        &self,
222        mut store: impl AsContextMut,
223        name: impl ExportLookup,
224    ) -> Option<Module> {
225        let store = store.as_context_mut().0;
226        let (instance, export) = self.lookup_export(store, name)?;
227        match export {
228            Export::ModuleStatic { index, .. } => {
229                Some(instance.component().static_module(*index).clone())
230            }
231            Export::ModuleImport { import, .. } => match instance.runtime_import(*import) {
232                RuntimeImport::Module(m) => Some(m.clone()),
233                _ => unreachable!(),
234            },
235            _ => None,
236        }
237    }
238
239    /// Looks up an exported resource type by name within this [`Instance`].
240    ///
241    /// The `store` argument provided must be the store that this instance
242    /// lives within and the `name` argument is the lookup key by which to find
243    /// the exported resource. If the resource is found then `Some` is returned
244    /// and otherwise `None` is returned.
245    ///
246    /// The `name` here can be a string such as `&str` or it can be a
247    /// [`ComponentExportIndex`] which is loaded prior from a [`Component`].
248    ///
249    /// For some examples see [`Instance::get_func`] for loading values from a
250    /// component.
251    ///
252    /// # Panics
253    ///
254    /// Panics if `store` does not own this instance.
255    pub fn get_resource(
256        &self,
257        mut store: impl AsContextMut,
258        name: impl ExportLookup,
259    ) -> Option<ResourceType> {
260        let store = store.as_context_mut().0;
261        let (instance, export) = self.lookup_export(store, name)?;
262        match export {
263            Export::Type(TypeDef::Resource(id)) => {
264                Some(InstanceType::new(instance).resource_type(*id))
265            }
266            Export::Type(_)
267            | Export::LiftedFunction { .. }
268            | Export::ModuleStatic { .. }
269            | Export::ModuleImport { .. }
270            | Export::Instance { .. } => None,
271        }
272    }
273
274    /// A methods similar to [`Component::get_export`] except for this
275    /// instance.
276    ///
277    /// This method will lookup the `name` provided within the `instance`
278    /// provided and return a [`ComponentItem`] describing the export,
279    /// and [`ComponentExportIndex`] which can be passed other `get_*`
280    /// functions like [`Instance::get_func`].
281    ///
282    /// The [`ComponentItem`] is more expensive to compute than the
283    /// [`ComponentExportIndex`]. If you are not consuming the
284    /// [`ComponentItem`], use [`Instance::get_export_index`] instead.
285    ///
286    /// # Panics
287    ///
288    /// Panics if `store` does not own this instance.
289    pub fn get_export(
290        &self,
291        mut store: impl AsContextMut,
292        instance: Option<&ComponentExportIndex>,
293        name: &str,
294    ) -> Option<(ComponentItem, ComponentExportIndex)> {
295        self._get_export(store.as_context_mut().0, instance, name)
296    }
297
298    fn _get_export(
299        &self,
300        store: &StoreOpaque,
301        instance: Option<&ComponentExportIndex>,
302        name: &str,
303    ) -> Option<(ComponentItem, ComponentExportIndex)> {
304        let data = self.id().get(store);
305        let component = data.component();
306        let index = component.lookup_export_index(instance, name)?;
307        let item = ComponentItem::from_export(
308            &store.engine(),
309            &component.env_component().export_items[index],
310            &InstanceType::new(data),
311        );
312        Some((
313            item,
314            ComponentExportIndex {
315                id: data.component().id(),
316                index,
317            },
318        ))
319    }
320
321    /// A methods similar to [`Component::get_export_index`] except for this
322    /// instance.
323    ///
324    /// This method will lookup the `name` provided within the `instance`
325    /// provided and return a [`ComponentExportIndex`] which can be passed
326    /// other `get_*` functions like [`Instance::get_func`].
327    ///
328    /// If you need the [`ComponentItem`] corresponding to this export, use
329    /// the [`Instance::get_export`] instead.
330    ///
331    /// # Panics
332    ///
333    /// Panics if `store` does not own this instance.
334    pub fn get_export_index(
335        &self,
336        mut store: impl AsContextMut,
337        instance: Option<&ComponentExportIndex>,
338        name: &str,
339    ) -> Option<ComponentExportIndex> {
340        let data = self.id().get(store.as_context_mut().0);
341        let index = data.component().lookup_export_index(instance, name)?;
342        Some(ComponentExportIndex {
343            id: data.component().id(),
344            index,
345        })
346    }
347
348    fn lookup_export<'a>(
349        &self,
350        store: &'a StoreOpaque,
351        name: impl ExportLookup,
352    ) -> Option<(&'a ComponentInstance, &'a Export)> {
353        let data = self.id().get(store);
354        let index = name.lookup(data.component(), None)?;
355        Some((data, &data.component().env_component().export_items[index]))
356    }
357
358    /// Returns the [`InstancePre`] that was used to create this instance.
359    pub fn instance_pre<T>(&self, store: impl AsContext<Data = T>) -> InstancePre<T> {
360        // This indexing operation asserts the Store owns the Instance.
361        // Therefore, the InstancePre<T> must match the Store<T>.
362        let data = self.id().get(store.as_context().0);
363
364        // SAFETY: calling this method safely here relies on matching the `T`
365        // in `InstancePre<T>` to the store itself, which is happening in the
366        // type signature just above by ensuring the store's data is `T` which
367        // matches the return value.
368        unsafe { data.instance_pre() }
369    }
370
371    pub(crate) fn id(&self) -> StoreComponentInstanceId {
372        self.id
373    }
374
375    /// Implementation of the `resource.new` intrinsic for `i32`
376    /// representations.
377    pub(crate) fn resource_new32(
378        self,
379        store: &mut StoreOpaque,
380        ty: TypeResourceTableIndex,
381        rep: u32,
382    ) -> Result<u32> {
383        store
384            .component_resource_tables(Some(self))
385            .resource_new(TypedResource::Component { ty, rep })
386    }
387
388    /// Implementation of the `resource.rep` intrinsic for `i32`
389    /// representations.
390    pub(crate) fn resource_rep32(
391        self,
392        store: &mut StoreOpaque,
393        ty: TypeResourceTableIndex,
394        index: u32,
395    ) -> Result<u32> {
396        store
397            .component_resource_tables(Some(self))
398            .resource_rep(TypedResourceIndex::Component { ty, index })
399    }
400
401    /// Implementation of the `resource.drop` intrinsic.
402    pub(crate) fn resource_drop(
403        self,
404        store: &mut StoreOpaque,
405        ty: TypeResourceTableIndex,
406        index: u32,
407    ) -> Result<Option<u32>> {
408        store
409            .component_resource_tables(Some(self))
410            .resource_drop(TypedResourceIndex::Component { ty, index })
411    }
412
413    pub(crate) fn resource_transfer_own(
414        self,
415        store: &mut StoreOpaque,
416        index: u32,
417        src: TypeResourceTableIndex,
418        dst: TypeResourceTableIndex,
419    ) -> Result<u32> {
420        let mut tables = store.component_resource_tables(Some(self));
421        let rep = tables.resource_lift_own(TypedResourceIndex::Component { ty: src, index })?;
422        tables.resource_lower_own(TypedResource::Component { ty: dst, rep })
423    }
424
425    pub(crate) fn resource_transfer_borrow(
426        self,
427        store: &mut StoreOpaque,
428        index: u32,
429        src: TypeResourceTableIndex,
430        dst: TypeResourceTableIndex,
431    ) -> Result<u32> {
432        let dst_owns_resource = self.id().get(store).resource_owned_by_own_instance(dst);
433        let mut tables = store.component_resource_tables(Some(self));
434        let rep = tables.resource_lift_borrow(TypedResourceIndex::Component { ty: src, index })?;
435        // Implement `lower_borrow`'s special case here where if a borrow's
436        // resource type is owned by `dst` then the destination receives the
437        // representation directly rather than a handle to the representation.
438        //
439        // This can perhaps become a different libcall in the future to avoid
440        // this check at runtime since we know at compile time whether the
441        // destination type owns the resource, but that's left as a future
442        // refactoring if truly necessary.
443        if dst_owns_resource {
444            return Ok(rep);
445        }
446        tables.resource_lower_borrow(TypedResource::Component { ty: dst, rep })
447    }
448
449    pub(crate) fn lookup_vmdef(&self, store: &mut StoreOpaque, def: &CoreDef) -> vm::Export {
450        lookup_vmdef(store, self.id.instance(), def)
451    }
452
453    pub(crate) fn options<'a>(
454        &self,
455        store: &'a StoreOpaque,
456        options: OptionsIndex,
457    ) -> &'a CanonicalOptions {
458        &self.id.get(store).component().env_component().options[options]
459    }
460
461    fn options_memory_raw(
462        &self,
463        store: &StoreOpaque,
464        options: OptionsIndex,
465    ) -> Option<NonNull<vm::VMMemoryDefinition>> {
466        let instance = self.id.get(store);
467        let options = &instance.component().env_component().options[options];
468        let memory = match options.data_model {
469            CanonicalOptionsDataModel::Gc { .. } => return None,
470            CanonicalOptionsDataModel::LinearMemory(o) => match o.memory {
471                Some(m) => m,
472                None => return None,
473            },
474        };
475
476        Some(instance.runtime_memory(memory))
477    }
478
479    pub(crate) fn options_memory<'a>(
480        &self,
481        store: &'a StoreOpaque,
482        options: OptionsIndex,
483    ) -> &'a [u8] {
484        let memory = match self.options_memory_raw(store, options) {
485            Some(m) => m,
486            None => return &[],
487        };
488        // SAFETY: we're borrowing the entire `StoreOpaque` which owns the
489        // memory allocation to return the result of memory. That means that the
490        // lifetime connection here should be safe and the actual ptr/length are
491        // trusted parts of the runtime here.
492        unsafe {
493            let memory = memory.as_ref();
494            core::slice::from_raw_parts(memory.base.as_ptr(), memory.current_length())
495        }
496    }
497
498    pub(crate) fn options_memory_mut<'a>(
499        &self,
500        store: &'a mut StoreOpaque,
501        options: OptionsIndex,
502    ) -> &'a mut [u8] {
503        let memory = match self.options_memory_raw(store, options) {
504            Some(m) => m,
505            None => return &mut [],
506        };
507        // SAFETY: See `options_memory` comment above, and note that this is
508        // taking `&mut StoreOpaque` to thread the lifetime through instead.
509        unsafe {
510            let memory = memory.as_ref();
511            core::slice::from_raw_parts_mut(memory.base.as_ptr(), memory.current_length())
512        }
513    }
514
515    /// Helper function to simultaneously get a borrow to this instance's
516    /// component as well as the store that this component is contained within.
517    ///
518    /// Note that this function signature is not possible with safe Rust, so
519    /// this is using `unsafe` internally.
520    pub(crate) fn component_and_store_mut<'a, S>(
521        &self,
522        store: &'a mut S,
523    ) -> (&'a Component, &'a mut S)
524    where
525        S: AsStoreOpaque,
526    {
527        let store_opaque = store.as_store_opaque();
528        let instance = self.id.get_mut(store_opaque);
529        let component = instance.component();
530
531        // SAFETY: the goal of this function is to derive a pointer from
532        // `&mut S`, here `&Component`, and then return both so they can both be
533        // used at the same time. In general this is not safe operation since
534        // the original mutable pointer could be mutated or overwritten which
535        // would invalidate the derived pointer.
536        //
537        // In this case though we have a few guarantees which should make this
538        // safe:
539        //
540        // * Embedders never have the ability to overwrite a `StoreOpaque`. For
541        //   example the closest thing of `StoreContextMut` wraps up the
542        //   reference internally so it's inaccessible to the outside world.
543        //   This means that while mutations can still happen it's not possible
544        //   to overwrite a `StoreOpaque` directly.
545        //
546        // * Components are referred to by `vm::ComponentInstance` which holds a
547        //   strong reference. All `ComponentInstance` structures are allocated
548        //   within the store and unconditionally live as long as the entire
549        //   store itself. This means that there's no worry of the rooting
550        //   container going away or otherwise getting deallocated.
551        //
552        // * The `ComponentInstance` container has an invariant that after
553        //   creation the component used to create it cannot be changed. This is
554        //   enforced through `Pin<&mut ComponentInstance>` which disallows
555        //   mutable access to the `component` field, instead only allowing
556        //   read-only access.
557        //
558        // Putting all of this together it's not possible for a component,
559        // within a component instance, within a store, to be deallocated or mutated while
560        // a store is in use. Consequently it should be safe to simultaneously
561        // have a borrow to both at the same time, even if the store has a
562        // mutable borrow itself.
563        unsafe {
564            let component: *const Component = component;
565            (&*component, store)
566        }
567    }
568
569    pub(crate) fn runtime_instance(&self, index: RuntimeComponentInstanceIndex) -> RuntimeInstance {
570        RuntimeInstance {
571            instance: self.id.instance(),
572            index,
573        }
574    }
575}
576
577/// Translates a `CoreDef`, a definition of a core wasm item, to an
578/// [`Export`] which is the runtime core wasm definition.
579pub(crate) fn lookup_vmdef(
580    store: &mut StoreOpaque,
581    id: ComponentInstanceId,
582    def: &CoreDef,
583) -> vm::Export {
584    match def {
585        CoreDef::Export(e) => lookup_vmexport(store, id, e),
586        CoreDef::Trampoline(idx) => {
587            let funcref = store
588                .store_data_mut()
589                .component_instance_mut(id)
590                .trampoline_func_ref(*idx);
591            // SAFETY: the `funcref` is owned by `store` and is valid within
592            // that store, so it's safe to create a `Func`.
593            vm::Export::Function(unsafe { crate::Func::from_vm_func_ref(store.id(), funcref) })
594        }
595        CoreDef::InstanceFlags(idx) => {
596            let id = StoreComponentInstanceId::new(store.id(), id);
597            vm::Export::Global(crate::Global::from_component_flags(id, *idx))
598        }
599        CoreDef::UnsafeIntrinsic(intrinsic) => {
600            let funcref = store
601                .store_data_mut()
602                .component_instance_mut(id)
603                .unsafe_intrinsic_func_ref(*intrinsic);
604            // SAFETY: as above, the `funcref` is owned by `store` and is valid
605            // within that store, so it's safe to create a `Func`.
606            vm::Export::Function(unsafe { crate::Func::from_vm_func_ref(store.id(), funcref) })
607        }
608        CoreDef::TaskMayBlock => vm::Export::Global(crate::Global::from_task_may_block(
609            StoreComponentInstanceId::new(store.id(), id),
610        )),
611    }
612}
613
614/// Translates a `CoreExport<T>`, an export of some core instance within
615/// this component, to the actual runtime definition of that item.
616pub(crate) fn lookup_vmexport<T>(
617    store: &mut StoreOpaque,
618    id: ComponentInstanceId,
619    item: &CoreExport<T>,
620) -> vm::Export
621where
622    T: Copy + Into<EntityIndex>,
623{
624    let store_id = store.id();
625    let id = store
626        .store_data_mut()
627        .component_instance_mut(id)
628        .instance(item.instance);
629    let (instance, registry) = store.instance_and_module_registry_mut(id);
630    let idx = match &item.item {
631        ExportItem::Index(idx) => (*idx).into(),
632
633        // FIXME: ideally at runtime we don't actually do any name lookups
634        // here. This will only happen when the host supplies an imported
635        // module so while the structure can't be known at compile time we
636        // do know at `InstancePre` time, for example, what all the host
637        // imports are. In theory we should be able to, as part of
638        // `InstancePre` construction, perform all name=>index mappings
639        // during that phase so the actual instantiation of an `InstancePre`
640        // skips all string lookups. This should probably only be
641        // investigated if this becomes a performance issue though.
642        ExportItem::Name(name) => {
643            let module = instance.env_module();
644            let name = module.strings.get_atom(name).unwrap();
645            module.exports[&name]
646        }
647    };
648    // SAFETY: the `store_id` owns this instance and all exports contained
649    // within.
650    unsafe { instance.get_export_by_index_mut(registry, store_id, idx) }
651}
652
653struct Instantiator<'a> {
654    component: &'a Component,
655    id: ComponentInstanceId,
656    core_imports: OwnedImports,
657    imports: &'a PrimaryMap<RuntimeImportIndex, RuntimeImport>,
658}
659
660pub(crate) enum RuntimeImport {
661    Func(Arc<HostFunc>),
662    Module(Module),
663    Resource {
664        ty: ResourceType,
665
666        // A strong reference to the host function that represents the
667        // destructor for this resource. At this time all resources here are
668        // host-defined resources. Note that this is itself never read because
669        // the funcref below points to it.
670        //
671        // Also note that the `Arc` here is used to support the same host
672        // function being used across multiple instances simultaneously. Or
673        // otherwise this makes `InstancePre::instantiate` possible to create
674        // separate instances all sharing the same host function.
675        dtor: Arc<crate::func::HostFunc>,
676
677        // A raw function which is filled out (including `wasm_call`) which
678        // points to the internals of the `_dtor` field. This is read and
679        // possibly executed by wasm.
680        dtor_funcref: VMFuncRef,
681    },
682}
683
684pub type ImportedResources = TryPrimaryMap<ResourceIndex, ResourceType>;
685
686impl<'a> Instantiator<'a> {
687    fn new(
688        component: &'a Component,
689        store: &mut StoreOpaque,
690        imports: &'a Arc<PrimaryMap<RuntimeImportIndex, RuntimeImport>>,
691    ) -> Result<Instantiator<'a>> {
692        let env_component = component.env_component();
693        let (modules, engine, breakpoints) = store.modules_and_engine_and_breakpoints_mut();
694        modules.register_component(component, engine, breakpoints)?;
695        let imported_resources: ImportedResources =
696            TryPrimaryMap::with_capacity(env_component.imported_resources.len())?;
697
698        let instance = ComponentInstance::new(
699            store.store_data().components.next_component_instance_id(),
700            component,
701            try_new::<Arc<_>>(imported_resources)?,
702            imports,
703            store.traitobj(),
704        )?;
705        let id = store.store_data_mut().push_component_instance(instance)?;
706
707        Ok(Instantiator {
708            component,
709            imports,
710            core_imports: OwnedImports::empty(),
711            id,
712        })
713    }
714
715    async fn run<T>(
716        &mut self,
717        store: &mut StoreContextMut<'_, T>,
718        asyncness: Asyncness,
719    ) -> Result<()> {
720        let env_component = self.component.env_component();
721
722        // Before all initializers are processed configure all destructors for
723        // host-defined resources. No initializer will correspond to these and
724        // it's required to happen before they're needed, so execute this first.
725        for (idx, import) in env_component.imported_resources.iter() {
726            let (ty, func_ref) = match &self.imports[*import] {
727                RuntimeImport::Resource {
728                    ty, dtor_funcref, ..
729                } => (*ty, NonNull::from(dtor_funcref)),
730                _ => unreachable!(),
731            };
732            let i = self.instance_resource_types_mut(store.0).push(ty)?;
733            assert_eq!(i, idx);
734            self.instance_mut(store.0)
735                .set_resource_destructor(idx, Some(func_ref));
736        }
737
738        // Next configure all `VMFuncRef`s for trampolines that this component
739        // will require. These functions won't actually get used until their
740        // associated state has been initialized through the global initializers
741        // below, but the funcrefs can all be configured here.
742        for (idx, sig) in env_component.trampolines.iter() {
743            let ptrs = self.component.trampoline_ptrs(idx);
744            let signature = match self.component.signatures().shared_type(*sig) {
745                Some(s) => s,
746                None => panic!("found unregistered signature: {sig:?}"),
747            };
748
749            self.instance_mut(store.0).set_trampoline(
750                idx,
751                ptrs.wasm_call,
752                ptrs.array_call,
753                signature,
754            );
755        }
756
757        // Initialize the unsafe intrinsics used by this component, if any.
758        for (i, module_ty) in env_component
759            .unsafe_intrinsics
760            .iter()
761            .enumerate()
762            .filter_map(|(i, ty)| ty.expand().map(|ty| (i, ty)))
763        {
764            let i = u32::try_from(i).unwrap();
765            let intrinsic = UnsafeIntrinsic::from_u32(i);
766            let ptrs = self.component.unsafe_intrinsic_ptrs(intrinsic).expect(
767                "should have intrinsic pointers given that we assigned the intrinsic a type",
768            );
769            let shared_ty = self
770                .component
771                .signatures()
772                .shared_type(module_ty)
773                .expect("should have a shared type");
774            self.instance_mut(store.0).set_intrinsic(
775                intrinsic,
776                ptrs.wasm_call,
777                ptrs.array_call,
778                shared_ty,
779            );
780        }
781
782        for initializer in env_component.initializers.iter() {
783            match initializer {
784                GlobalInitializer::InstantiateModule(m, component_instance) => {
785                    let instance = self.id;
786                    let module;
787                    let imports = match m {
788                        // Since upvars are statically know we know that the
789                        // `args` list is already in the right order.
790                        InstantiateModule::Static(idx, args) => {
791                            module = self.component.static_module(*idx);
792                            self.build_imports(store.0, module, args.iter())?
793                        }
794
795                        // With imports, unlike upvars, we need to do runtime
796                        // lookups with strings to determine the order of the
797                        // imports since it's whatever the actual module
798                        // requires.
799                        //
800                        // FIXME: see the note in `ExportItem::Name` handling
801                        // above for how we ideally shouldn't do string lookup
802                        // here.
803                        InstantiateModule::Import(idx, args) => {
804                            module = match &self.imports[*idx] {
805                                RuntimeImport::Module(m) => m,
806                                _ => unreachable!(),
807                            };
808                            let args = module
809                                .imports()
810                                .map(|import| &args[import.module()][import.name()]);
811                            self.build_imports(store.0, module, args)?
812                        }
813                    };
814
815                    let exit = if let Some(component_instance) = *component_instance {
816                        store.0.enter_guest_sync_call(
817                            None,
818                            false,
819                            RuntimeInstance {
820                                instance,
821                                index: component_instance,
822                            },
823                        )?;
824                        true
825                    } else {
826                        false
827                    };
828
829                    // Note that the unsafety here should be ok because the
830                    // validity of the component means that type-checks have
831                    // already been performed. This means that the unsafety due
832                    // to imports having the wrong type should not happen here.
833                    //
834                    // Also note we are calling new_started_impl because we have
835                    // already checked for asyncness and are running on a fiber
836                    // if required.
837
838                    let i = unsafe {
839                        crate::Instance::new_started(store, module, imports.as_ref(), asyncness)
840                            .await?
841                    };
842
843                    if exit {
844                        store.0.exit_guest_sync_call()?;
845                    }
846
847                    self.instance_mut(store.0).push_instance_id(i.id())?;
848                }
849
850                GlobalInitializer::LowerImport { import, index } => {
851                    let func = match &self.imports[*import] {
852                        RuntimeImport::Func(func) => func,
853                        _ => unreachable!(),
854                    };
855                    self.instance_mut(store.0)
856                        .set_lowering(*index, func.lowering());
857                }
858
859                GlobalInitializer::ExtractTable(table) => self.extract_table(store.0, table),
860
861                GlobalInitializer::ExtractMemory(mem) => self.extract_memory(store.0, mem),
862
863                GlobalInitializer::ExtractRealloc(realloc) => {
864                    self.extract_realloc(store.0, realloc)
865                }
866
867                GlobalInitializer::ExtractCallback(callback) => {
868                    self.extract_callback(store.0, callback)
869                }
870
871                GlobalInitializer::ExtractPostReturn(post_return) => {
872                    self.extract_post_return(store.0, post_return)
873                }
874
875                GlobalInitializer::Resource(r) => self.resource(store.0, r)?,
876            }
877        }
878        Ok(())
879    }
880
881    fn resource(&mut self, store: &mut StoreOpaque, resource: &Resource) -> Result<()> {
882        let dtor = resource
883            .dtor
884            .as_ref()
885            .map(|dtor| lookup_vmdef(store, self.id, dtor));
886        let dtor = dtor.map(|export| match export {
887            crate::runtime::vm::Export::Function(f) => f.vm_func_ref(store),
888            _ => unreachable!(),
889        });
890        let index = self
891            .component
892            .env_component()
893            .resource_index(resource.index);
894        let instance = self.instance(store);
895        let ty = ResourceType::guest(store.id(), instance, resource.index);
896        self.instance_mut(store)
897            .set_resource_destructor(index, dtor);
898        let i = self.instance_resource_types_mut(store).push(ty)?;
899        debug_assert_eq!(i, index);
900        Ok(())
901    }
902
903    fn extract_memory(&mut self, store: &mut StoreOpaque, memory: &ExtractMemory) {
904        let import = match lookup_vmexport(store, self.id, &memory.export) {
905            crate::runtime::vm::Export::Memory(memory) => memory.vmimport(store),
906            crate::runtime::vm::Export::SharedMemory(_, import) => import,
907            _ => unreachable!(),
908        };
909        self.instance_mut(store)
910            .set_runtime_memory(memory.index, import.from.as_non_null());
911    }
912
913    fn extract_realloc(&mut self, store: &mut StoreOpaque, realloc: &ExtractRealloc) {
914        let func_ref = match lookup_vmdef(store, self.id, &realloc.def) {
915            crate::runtime::vm::Export::Function(f) => f.vm_func_ref(store),
916            _ => unreachable!(),
917        };
918        self.instance_mut(store)
919            .set_runtime_realloc(realloc.index, func_ref);
920    }
921
922    fn extract_callback(&mut self, store: &mut StoreOpaque, callback: &ExtractCallback) {
923        let func_ref = match lookup_vmdef(store, self.id, &callback.def) {
924            crate::runtime::vm::Export::Function(f) => f.vm_func_ref(store),
925            _ => unreachable!(),
926        };
927        self.instance_mut(store)
928            .set_runtime_callback(callback.index, func_ref);
929    }
930
931    fn extract_post_return(&mut self, store: &mut StoreOpaque, post_return: &ExtractPostReturn) {
932        let func_ref = match lookup_vmdef(store, self.id, &post_return.def) {
933            crate::runtime::vm::Export::Function(f) => f.vm_func_ref(store),
934            _ => unreachable!(),
935        };
936        self.instance_mut(store)
937            .set_runtime_post_return(post_return.index, func_ref);
938    }
939
940    fn extract_table(&mut self, store: &mut StoreOpaque, table: &ExtractTable) {
941        let export = match lookup_vmexport(store, self.id, &table.export) {
942            crate::runtime::vm::Export::Table(t) => t,
943            _ => unreachable!(),
944        };
945        let import = export.vmimport(store);
946        self.instance_mut(store)
947            .set_runtime_table(table.index, import);
948    }
949
950    fn build_imports<'b>(
951        &mut self,
952        store: &mut StoreOpaque,
953        module: &Module,
954        args: impl Iterator<Item = &'b CoreDef>,
955    ) -> Result<&OwnedImports, OutOfMemory> {
956        self.core_imports.clear();
957        self.core_imports.reserve(module)?;
958        let mut imports = module.compiled_module().module().imports();
959
960        for arg in args {
961            // The general idea of Wasmtime is that at runtime type-checks for
962            // core wasm instantiations internally within a component are
963            // unnecessary and superfluous. Naturally though mistakes may be
964            // made, so double-check this property of wasmtime in debug mode.
965
966            if cfg!(debug_assertions) {
967                let (imp_module, imp_name, expected) = imports.next().unwrap();
968                self.assert_type_matches(store, module, arg, imp_module, imp_name, expected);
969            }
970
971            // The unsafety here should be ok since the `export` is loaded
972            // directly from an instance which should only give us valid export
973            // items.
974            let export = lookup_vmdef(store, self.id, arg);
975            self.core_imports.push_export(store, &export)?;
976        }
977        debug_assert!(imports.next().is_none());
978
979        Ok(&self.core_imports)
980    }
981
982    fn assert_type_matches(
983        &self,
984        store: &mut StoreOpaque,
985        module: &Module,
986        arg: &CoreDef,
987        imp_module: &str,
988        imp_name: &str,
989        expected: EntityType,
990    ) {
991        let export = lookup_vmdef(store, self.id, arg);
992
993        // If this value is a core wasm function then the type check is inlined
994        // here. This can otherwise fail `Extern::from_wasmtime_export` because
995        // there's no guarantee that there exists a trampoline for `f` so this
996        // can't fall through to the case below
997        if let crate::runtime::vm::Export::Function(f) = &export {
998            let expected = match expected.unwrap_func() {
999                EngineOrModuleTypeIndex::Engine(e) => Some(e),
1000                EngineOrModuleTypeIndex::Module(m) => module.signatures().shared_type(m),
1001                EngineOrModuleTypeIndex::RecGroup(_) => unreachable!(),
1002            };
1003            let actual = unsafe { f.vm_func_ref(store).as_ref().type_index };
1004            assert_eq!(
1005                expected,
1006                Some(actual),
1007                "type mismatch for import {imp_module:?} {imp_name:?}!!!\n\n\
1008                 expected {:#?}\n\n\
1009                 found {:#?}",
1010                expected.and_then(|e| store.engine().signatures().borrow(e)),
1011                store.engine().signatures().borrow(actual)
1012            );
1013            return;
1014        }
1015
1016        let val = crate::Extern::from_wasmtime_export(export, store.engine());
1017        let ty = DefinitionType::from(store, &val);
1018        crate::types::matching::MatchCx::new(module.engine())
1019            .definition(&expected, &ty)
1020            .expect("unexpected typecheck failure");
1021    }
1022
1023    /// Convenience helper to return the `&ComponentInstance` that's being
1024    /// instantiated.
1025    fn instance<'b>(&self, store: &'b StoreOpaque) -> &'b ComponentInstance {
1026        store.store_data().component_instance(self.id)
1027    }
1028
1029    /// Same as [`Self::instance`], but for mutability.
1030    fn instance_mut<'b>(&self, store: &'b mut StoreOpaque) -> Pin<&'b mut ComponentInstance> {
1031        store.store_data_mut().component_instance_mut(self.id)
1032    }
1033
1034    // NB: This method is only intended to be called during the instantiation
1035    // process because the `Arc::get_mut` here is fallible and won't generally
1036    // succeed once the instance has been handed to the embedder. Before that
1037    // though it should be guaranteed that the single owning reference currently
1038    // lives within the `ComponentInstance` that's being built.
1039    fn instance_resource_types_mut<'b>(
1040        &self,
1041        store: &'b mut StoreOpaque,
1042    ) -> &'b mut ImportedResources {
1043        Arc::get_mut(self.instance_mut(store).resource_types_mut()).unwrap()
1044    }
1045}
1046
1047/// A "pre-instantiated" [`Instance`] which has all of its arguments already
1048/// supplied and is ready to instantiate.
1049///
1050/// This structure represents an efficient form of instantiation where import
1051/// type-checking and import lookup has all been resolved by the time that this
1052/// type is created. This type is primarily created through the
1053/// [`Linker::instantiate_pre`](crate::component::Linker::instantiate_pre)
1054/// method.
1055pub struct InstancePre<T: 'static> {
1056    component: Component,
1057    imports: Arc<PrimaryMap<RuntimeImportIndex, RuntimeImport>>,
1058    resource_types: Arc<TryPrimaryMap<ResourceIndex, ResourceType>>,
1059    asyncness: Asyncness,
1060    _marker: marker::PhantomData<fn() -> T>,
1061}
1062
1063// `InstancePre`'s clone does not require `T: Clone`
1064impl<T: 'static> Clone for InstancePre<T> {
1065    fn clone(&self) -> Self {
1066        Self {
1067            component: self.component.clone(),
1068            imports: self.imports.clone(),
1069            resource_types: self.resource_types.clone(),
1070            asyncness: self.asyncness,
1071            _marker: self._marker,
1072        }
1073    }
1074}
1075
1076impl<T: 'static> InstancePre<T> {
1077    /// This function is `unsafe` since there's no guarantee that the
1078    /// `RuntimeImport` items provided are guaranteed to work with the `T` of
1079    /// the store.
1080    ///
1081    /// Additionally there is no static guarantee that the `imports` provided
1082    /// satisfy the imports of the `component` provided.
1083    pub(crate) unsafe fn new_unchecked(
1084        component: Component,
1085        imports: Arc<PrimaryMap<RuntimeImportIndex, RuntimeImport>>,
1086        resource_types: Arc<TryPrimaryMap<ResourceIndex, ResourceType>>,
1087    ) -> InstancePre<T> {
1088        let mut asyncness = Asyncness::No;
1089        for (_, import) in imports.iter() {
1090            asyncness = asyncness
1091                | match import {
1092                    RuntimeImport::Func(f) => f.asyncness(),
1093                    RuntimeImport::Module(_) => Asyncness::No,
1094                    RuntimeImport::Resource { dtor, .. } => dtor.asyncness(),
1095                };
1096        }
1097        InstancePre {
1098            component,
1099            imports,
1100            resource_types,
1101            asyncness,
1102            _marker: marker::PhantomData,
1103        }
1104    }
1105
1106    /// Returns the underlying component that will be instantiated.
1107    pub fn component(&self) -> &Component {
1108        &self.component
1109    }
1110
1111    #[doc(hidden)]
1112    /// Returns the type at which the underlying component will be
1113    /// instantiated. This contains the instantiated type information which
1114    /// was determined by the Linker.
1115    pub fn instance_type(&self) -> InstanceType<'_> {
1116        InstanceType {
1117            types: &self.component.types(),
1118            resources: Some(&self.resource_types),
1119        }
1120    }
1121
1122    /// Returns the underlying engine.
1123    pub fn engine(&self) -> &Engine {
1124        self.component.engine()
1125    }
1126
1127    /// Performs the instantiation process into the store specified.
1128    ///
1129    /// # Errors
1130    ///
1131    /// This function will return an [`OutOfMemory`][crate::OutOfMemory] error when
1132    /// memory allocation fails. See the `OutOfMemory` type's documentation for
1133    /// details on Wasmtime's out-of-memory handling.
1134    //
1135    // TODO: needs more docs
1136    pub fn instantiate(&self, mut store: impl AsContextMut<Data = T>) -> Result<Instance> {
1137        let store = store.as_context_mut();
1138
1139        // If this instance requires an async host, set that flag in the store,
1140        // and then afterwards assert nothing else in the store, nor this
1141        // instance, required async.
1142        store.0.set_async_required(self.asyncness);
1143        store.0.validate_sync_call()?;
1144
1145        vm::assert_ready(self._instantiate(store, Asyncness::No))
1146    }
1147    /// Performs the instantiation process into the store specified.
1148    ///
1149    /// Exactly like [`Self::instantiate`] except for use on async stores.
1150    ///
1151    /// # Errors
1152    ///
1153    /// This function will return an [`OutOfMemory`][crate::OutOfMemory] error when
1154    /// memory allocation fails. See the `OutOfMemory` type's documentation for
1155    /// details on Wasmtime's out-of-memory handling.
1156    //
1157    // TODO: needs more docs
1158    #[cfg(feature = "async")]
1159    pub async fn instantiate_async(&self, store: impl AsContextMut<Data = T>) -> Result<Instance> {
1160        self._instantiate(store, Asyncness::Yes).await
1161    }
1162
1163    async fn _instantiate(
1164        &self,
1165        mut store: impl AsContextMut<Data = T>,
1166        asyncness: Asyncness,
1167    ) -> Result<Instance> {
1168        let store = store.as_context_mut();
1169        store.0.set_async_required(self.asyncness);
1170        store
1171            .engine()
1172            .allocator()
1173            .increment_component_instance_count()?;
1174
1175        // Helper structure to pair the above increment with a decrement should
1176        // anything fail below.
1177        let mut decrement = DecrementComponentInstanceCountOnDrop {
1178            store,
1179            enabled: true,
1180        };
1181
1182        let mut instantiator =
1183            Instantiator::new(&self.component, decrement.store.0, &self.imports)?;
1184        instantiator.run(&mut decrement.store, asyncness).await?;
1185
1186        let instance = Instance::from_wasmtime(decrement.store.0, instantiator.id);
1187        decrement.store.0.push_component_instance(instance);
1188
1189        // Everything has passed, don't decrement the instance count and let the
1190        // destructor for the `Store` handle that at this point.
1191        decrement.enabled = false;
1192        return Ok(instance);
1193
1194        struct DecrementComponentInstanceCountOnDrop<'a, T: 'static> {
1195            store: StoreContextMut<'a, T>,
1196            enabled: bool,
1197        }
1198
1199        impl<T> Drop for DecrementComponentInstanceCountOnDrop<'_, T> {
1200            fn drop(&mut self) {
1201                if self.enabled {
1202                    self.store
1203                        .engine()
1204                        .allocator()
1205                        .decrement_component_instance_count();
1206                }
1207            }
1208        }
1209    }
1210}