1 use crate::component::InstanceExportLookup;
2 use crate::component::matching::InstanceType;
3 use crate::component::types;
4 use crate::prelude::*;
5 #[cfg(feature = "std")]
6 use crate::runtime::vm::open_file_for_mmap;
7 use crate::runtime::vm::{CompiledModuleId, VMArrayCallFunction, VMFuncRef, VMWasmCallFunction};
8 use crate::{
9     Engine, Module, ResourcesRequired, code::CodeObject, code_memory::CodeMemory,
10     type_registry::TypeCollection,
11 };
12 use crate::{FuncType, ValType};
13 use alloc::sync::Arc;
14 use core::ops::Range;
15 use core::ptr::NonNull;
16 #[cfg(feature = "std")]
17 use std::path::Path;
18 use wasmtime_environ::TypeTrace;
19 use wasmtime_environ::component::{
20     AllCallFunc, CompiledComponentInfo, ComponentArtifacts, ComponentTypes, CoreDef, Export,
21     ExportIndex, GlobalInitializer, InstantiateModule, NameMapNoIntern, OptionsIndex,
22     StaticModuleIndex, TrampolineIndex, TypeComponentIndex, TypeFuncIndex, VMComponentOffsets,
23 };
24 use wasmtime_environ::{FunctionLoc, HostPtr, ObjectKind, PrimaryMap};
25 
26 /// A compiled WebAssembly Component.
27 ///
28 /// This structure represents a compiled component that is ready to be
29 /// instantiated. This owns a region of virtual memory which contains executable
30 /// code compiled from a WebAssembly binary originally. This is the analog of
31 /// [`Module`](crate::Module) in the component embedding API.
32 ///
33 /// A [`Component`] can be turned into an
34 /// [`Instance`](crate::component::Instance) through a
35 /// [`Linker`](crate::component::Linker). [`Component`]s are safe to share
36 /// across threads. The compilation model of a component is the same as that of
37 /// [a module](crate::Module) which is to say:
38 ///
39 /// * Compilation happens synchronously during [`Component::new`].
40 /// * The result of compilation can be saved into storage with
41 ///   [`Component::serialize`].
42 /// * A previously compiled artifact can be parsed with
43 ///   [`Component::deserialize`].
44 /// * No compilation happens at runtime for a component — everything is done
45 ///   by the time [`Component::new`] returns.
46 ///
47 /// ## Components and `Clone`
48 ///
49 /// Using `clone` on a `Component` is a cheap operation. It will not create an
50 /// entirely new component, but rather just a new reference to the existing
51 /// component. In other words it's a shallow copy, not a deep copy.
52 ///
53 /// ## Examples
54 ///
55 /// For example usage see the documentation of [`Module`](crate::Module) as
56 /// [`Component`] has the same high-level API.
57 #[derive(Clone)]
58 pub struct Component {
59     inner: Arc<ComponentInner>,
60 }
61 
62 struct ComponentInner {
63     /// Unique id for this component within this process.
64     ///
65     /// Note that this is repurposing ids for modules intentionally as there
66     /// shouldn't be an issue overlapping them.
67     id: CompiledModuleId,
68 
69     /// The engine that this component belongs to.
70     engine: Engine,
71 
72     /// Component type index
73     ty: TypeComponentIndex,
74 
75     /// Core wasm modules that the component defined internally, indexed by the
76     /// compile-time-assigned `ModuleUpvarIndex`.
77     static_modules: PrimaryMap<StaticModuleIndex, Module>,
78 
79     /// Code-related information such as the compiled artifact, type
80     /// information, etc.
81     ///
82     /// Note that the `Arc` here is used to share this allocation with internal
83     /// modules.
84     code: Arc<CodeObject>,
85 
86     /// Metadata produced during compilation.
87     info: CompiledComponentInfo,
88 
89     /// A cached handle to the `wasmtime::FuncType` for the canonical ABI's
90     /// `realloc`, to avoid the need to look up types in the registry and take
91     /// locks when calling `realloc` via `TypedFunc::call_raw`.
92     realloc_func_type: Arc<FuncType>,
93 }
94 
95 pub(crate) struct AllCallFuncPointers {
96     pub wasm_call: NonNull<VMWasmCallFunction>,
97     pub array_call: NonNull<VMArrayCallFunction>,
98 }
99 
100 impl Component {
101     /// Compiles a new WebAssembly component from the in-memory list of bytes
102     /// provided.
103     ///
104     /// The `bytes` provided can either be the binary or text format of a
105     /// [WebAssembly component]. Note that the text format requires the `wat`
106     /// feature of this crate to be enabled. This API does not support
107     /// streaming compilation.
108     ///
109     /// This function will synchronously validate the entire component,
110     /// including all core modules, and then compile all components, modules,
111     /// etc., found within the provided bytes.
112     ///
113     /// [WebAssembly component]: https://github.com/WebAssembly/component-model/blob/main/design/mvp/Binary.md
114     ///
115     /// # Errors
116     ///
117     /// This function may fail and return an error. Errors may include
118     /// situations such as:
119     ///
120     /// * The binary provided could not be decoded because it's not a valid
121     ///   WebAssembly binary
122     /// * The WebAssembly binary may not validate (e.g. contains type errors)
123     /// * Implementation-specific limits were exceeded with a valid binary (for
124     ///   example too many locals)
125     /// * The wasm binary may use features that are not enabled in the
126     ///   configuration of `engine`
127     /// * If the `wat` feature is enabled and the input is text, then it may be
128     ///   rejected if it fails to parse.
129     ///
130     /// The error returned should contain full information about why compilation
131     /// failed.
132     ///
133     /// # Examples
134     ///
135     /// The `new` function can be invoked with a in-memory array of bytes:
136     ///
137     /// ```no_run
138     /// # use wasmtime::*;
139     /// # use wasmtime::component::Component;
140     /// # fn main() -> anyhow::Result<()> {
141     /// # let engine = Engine::default();
142     /// # let wasm_bytes: Vec<u8> = Vec::new();
143     /// let component = Component::new(&engine, &wasm_bytes)?;
144     /// # Ok(())
145     /// # }
146     /// ```
147     ///
148     /// Or you can also pass in a string to be parsed as the wasm text
149     /// format:
150     ///
151     /// ```
152     /// # use wasmtime::*;
153     /// # use wasmtime::component::Component;
154     /// # fn main() -> anyhow::Result<()> {
155     /// # let engine = Engine::default();
156     /// let component = Component::new(&engine, "(component (core module))")?;
157     /// # Ok(())
158     /// # }
159     #[cfg(any(feature = "cranelift", feature = "winch"))]
160     pub fn new(engine: &Engine, bytes: impl AsRef<[u8]>) -> Result<Component> {
161         crate::CodeBuilder::new(engine)
162             .wasm_binary_or_text(bytes.as_ref(), None)?
163             .compile_component()
164     }
165 
166     /// Compiles a new WebAssembly component from a wasm file on disk pointed
167     /// to by `file`.
168     ///
169     /// This is a convenience function for reading the contents of `file` on
170     /// disk and then calling [`Component::new`].
171     #[cfg(all(feature = "std", any(feature = "cranelift", feature = "winch")))]
172     pub fn from_file(engine: &Engine, file: impl AsRef<Path>) -> Result<Component> {
173         crate::CodeBuilder::new(engine)
174             .wasm_binary_or_text_file(file.as_ref())?
175             .compile_component()
176     }
177 
178     /// Compiles a new WebAssembly component from the in-memory wasm image
179     /// provided.
180     ///
181     /// This function is the same as [`Component::new`] except that it does not
182     /// accept the text format of WebAssembly. Even if the `wat` feature
183     /// is enabled an error will be returned here if `binary` is the text
184     /// format.
185     ///
186     /// For more information on semantics and errors see [`Component::new`].
187     #[cfg(any(feature = "cranelift", feature = "winch"))]
188     pub fn from_binary(engine: &Engine, binary: &[u8]) -> Result<Component> {
189         crate::CodeBuilder::new(engine)
190             .wasm_binary(binary, None)?
191             .compile_component()
192     }
193 
194     /// Same as [`Module::deserialize`], but for components.
195     ///
196     /// Note that the bytes referenced here must contain contents previously
197     /// produced by [`Engine::precompile_component`] or
198     /// [`Component::serialize`].
199     ///
200     /// For more information see the [`Module::deserialize`] method.
201     ///
202     /// # Unsafety
203     ///
204     /// The unsafety of this method is the same as that of the
205     /// [`Module::deserialize`] method.
206     ///
207     /// [`Module::deserialize`]: crate::Module::deserialize
208     pub unsafe fn deserialize(engine: &Engine, bytes: impl AsRef<[u8]>) -> Result<Component> {
209         let code = engine.load_code_bytes(bytes.as_ref(), ObjectKind::Component)?;
210         Component::from_parts(engine, code, None)
211     }
212 
213     /// Same as [`Module::deserialize_raw`], but for components.
214     ///
215     /// See [`Component::deserialize`] for additional information; this method
216     /// works identically except that it will not create a copy of the provided
217     /// memory but will use it directly.
218     ///
219     /// # Unsafety
220     ///
221     /// All of the safety notes from [`Component::deserialize`] apply here as well
222     /// with the additional constraint that the code memory provide by `memory`
223     /// lives for as long as the module and is nevery externally modified for
224     /// the lifetime of the deserialized module.
225     pub unsafe fn deserialize_raw(engine: &Engine, memory: NonNull<[u8]>) -> Result<Component> {
226         // SAFETY: the contract required by `load_code_raw` is the same as this
227         // function.
228         let code = unsafe { engine.load_code_raw(memory, ObjectKind::Component)? };
229         Component::from_parts(engine, code, None)
230     }
231 
232     /// Same as [`Module::deserialize_file`], but for components.
233     ///
234     /// Note that the file referenced here must contain contents previously
235     /// produced by [`Engine::precompile_component`] or
236     /// [`Component::serialize`].
237     ///
238     /// For more information see the [`Module::deserialize_file`] method.
239     ///
240     /// # Unsafety
241     ///
242     /// The unsafety of this method is the same as that of the
243     /// [`Module::deserialize_file`] method.
244     ///
245     /// [`Module::deserialize_file`]: crate::Module::deserialize_file
246     #[cfg(feature = "std")]
247     pub unsafe fn deserialize_file(engine: &Engine, path: impl AsRef<Path>) -> Result<Component> {
248         let file = open_file_for_mmap(path.as_ref())?;
249         let code = engine
250             .load_code_file(file, ObjectKind::Component)
251             .with_context(|| format!("failed to load code for: {}", path.as_ref().display()))?;
252         Component::from_parts(engine, code, None)
253     }
254 
255     /// Returns the type of this component as a [`types::Component`].
256     ///
257     /// This method enables runtime introspection of the type of a component
258     /// before instantiation, if necessary.
259     ///
260     /// ## Component types and Resources
261     ///
262     /// An important point to note here is that the precise type of imports and
263     /// exports of a component change when it is instantiated with respect to
264     /// resources. For example a [`Component`] represents an un-instantiated
265     /// component meaning that its imported resources are represented as abstract
266     /// resource types. These abstract types are not equal to any other
267     /// component's types.
268     ///
269     /// For example:
270     ///
271     /// ```
272     /// # use wasmtime::Engine;
273     /// # use wasmtime::component::Component;
274     /// # use wasmtime::component::types::ComponentItem;
275     /// # fn main() -> wasmtime::Result<()> {
276     /// # let engine = Engine::default();
277     /// let a = Component::new(&engine, r#"
278     ///     (component (import "x" (type (sub resource))))
279     /// "#)?;
280     /// let b = Component::new(&engine, r#"
281     ///     (component (import "x" (type (sub resource))))
282     /// "#)?;
283     ///
284     /// let (_, a_ty) = a.component_type().imports(&engine).next().unwrap();
285     /// let (_, b_ty) = b.component_type().imports(&engine).next().unwrap();
286     ///
287     /// let a_ty = match a_ty {
288     ///     ComponentItem::Resource(ty) => ty,
289     ///     _ => unreachable!(),
290     /// };
291     /// let b_ty = match b_ty {
292     ///     ComponentItem::Resource(ty) => ty,
293     ///     _ => unreachable!(),
294     /// };
295     /// assert!(a_ty != b_ty);
296     /// # Ok(())
297     /// # }
298     /// ```
299     ///
300     /// Additionally, however, these abstract types are "substituted" during
301     /// instantiation meaning that a component type will appear to have changed
302     /// once it is instantiated.
303     ///
304     /// ```
305     /// # use wasmtime::{Engine, Store};
306     /// # use wasmtime::component::{Component, Linker, ResourceType};
307     /// # use wasmtime::component::types::ComponentItem;
308     /// # fn main() -> wasmtime::Result<()> {
309     /// # let engine = Engine::default();
310     /// // Here this component imports a resource and then exports it as-is
311     /// // which means that the export is equal to the import.
312     /// let a = Component::new(&engine, r#"
313     ///     (component
314     ///         (import "x" (type $x (sub resource)))
315     ///         (export "x" (type $x))
316     ///     )
317     /// "#)?;
318     ///
319     /// let (_, import) = a.component_type().imports(&engine).next().unwrap();
320     /// let (_, export) = a.component_type().exports(&engine).next().unwrap();
321     ///
322     /// let import = match import {
323     ///     ComponentItem::Resource(ty) => ty,
324     ///     _ => unreachable!(),
325     /// };
326     /// let export = match export {
327     ///     ComponentItem::Resource(ty) => ty,
328     ///     _ => unreachable!(),
329     /// };
330     /// assert_eq!(import, export);
331     ///
332     /// // However after instantiation the resource type "changes"
333     /// let mut store = Store::new(&engine, ());
334     /// let mut linker = Linker::new(&engine);
335     /// linker.root().resource("x", ResourceType::host::<()>(), |_, _| Ok(()))?;
336     /// let instance = linker.instantiate(&mut store, &a)?;
337     /// let instance_ty = instance.get_resource(&mut store, "x").unwrap();
338     ///
339     /// // Here `instance_ty` is not the same as either `import` or `export`,
340     /// // but it is equal to what we provided as an import.
341     /// assert!(instance_ty != import);
342     /// assert!(instance_ty != export);
343     /// assert!(instance_ty == ResourceType::host::<()>());
344     /// # Ok(())
345     /// # }
346     /// ```
347     ///
348     /// Finally, each instantiation of an exported resource from a component is
349     /// considered "fresh" for all instantiations meaning that different
350     /// instantiations will have different exported resource types:
351     ///
352     /// ```
353     /// # use wasmtime::{Engine, Store};
354     /// # use wasmtime::component::{Component, Linker};
355     /// # fn main() -> wasmtime::Result<()> {
356     /// # let engine = Engine::default();
357     /// let a = Component::new(&engine, r#"
358     ///     (component
359     ///         (type $x (resource (rep i32)))
360     ///         (export "x" (type $x))
361     ///     )
362     /// "#)?;
363     ///
364     /// let mut store = Store::new(&engine, ());
365     /// let linker = Linker::new(&engine);
366     /// let instance1 = linker.instantiate(&mut store, &a)?;
367     /// let instance2 = linker.instantiate(&mut store, &a)?;
368     ///
369     /// let x1 = instance1.get_resource(&mut store, "x").unwrap();
370     /// let x2 = instance2.get_resource(&mut store, "x").unwrap();
371     ///
372     /// // Despite these two resources being the same export of the same
373     /// // component they come from two different instances meaning that their
374     /// // types will be unique.
375     /// assert!(x1 != x2);
376     /// # Ok(())
377     /// # }
378     /// ```
379     pub fn component_type(&self) -> types::Component {
380         self.with_uninstantiated_instance_type(|ty| types::Component::from(self.inner.ty, ty))
381     }
382 
383     fn with_uninstantiated_instance_type<R>(&self, f: impl FnOnce(&InstanceType<'_>) -> R) -> R {
384         let resources = Arc::new(PrimaryMap::new());
385         f(&InstanceType {
386             types: self.types(),
387             resources: &resources,
388         })
389     }
390 
391     /// Final assembly step for a component from its in-memory representation.
392     ///
393     /// If the `artifacts` are specified as `None` here then they will be
394     /// deserialized from `code_memory`.
395     pub(crate) fn from_parts(
396         engine: &Engine,
397         code_memory: Arc<CodeMemory>,
398         artifacts: Option<ComponentArtifacts>,
399     ) -> Result<Component> {
400         let ComponentArtifacts {
401             ty,
402             info,
403             mut types,
404             mut static_modules,
405         } = match artifacts {
406             Some(artifacts) => artifacts,
407             None => postcard::from_bytes(code_memory.wasmtime_info())?,
408         };
409 
410         // Validate that the component can be used with the current instance
411         // allocator.
412         engine.allocator().validate_component(
413             &info.component,
414             &VMComponentOffsets::new(HostPtr, &info.component),
415             &|module_index| &static_modules[module_index].module,
416         )?;
417 
418         // Create a signature registration with the `Engine` for all trampolines
419         // and core wasm types found within this component, both for the
420         // component and for all included core wasm modules.
421         let signatures = engine.register_and_canonicalize_types(
422             types.module_types_mut(),
423             static_modules.iter_mut().map(|(_, m)| &mut m.module),
424         );
425         types.canonicalize_for_runtime_usage(&mut |idx| signatures.shared_type(idx).unwrap());
426 
427         // Assemble the `CodeObject` artifact which is shared by all core wasm
428         // modules as well as the final component.
429         let types = Arc::new(types);
430         let code = Arc::new(CodeObject::new(code_memory, signatures, types.into()));
431 
432         // Convert all information about static core wasm modules into actual
433         // `Module` instances by converting each `CompiledModuleInfo`, the
434         // `types` type information, and the code memory to a runtime object.
435         let static_modules = static_modules
436             .into_iter()
437             .map(|(_, info)| Module::from_parts_raw(engine, code.clone(), info, false))
438             .collect::<Result<_>>()?;
439 
440         let realloc_func_type = Arc::new(FuncType::new(
441             engine,
442             [ValType::I32, ValType::I32, ValType::I32, ValType::I32],
443             [ValType::I32],
444         ));
445 
446         Ok(Component {
447             inner: Arc::new(ComponentInner {
448                 id: CompiledModuleId::new(),
449                 engine: engine.clone(),
450                 ty,
451                 static_modules,
452                 code,
453                 info,
454                 realloc_func_type,
455             }),
456         })
457     }
458 
459     pub(crate) fn ty(&self) -> TypeComponentIndex {
460         self.inner.ty
461     }
462 
463     pub(crate) fn env_component(&self) -> &wasmtime_environ::component::Component {
464         &self.inner.info.component
465     }
466 
467     pub(crate) fn static_module(&self, idx: StaticModuleIndex) -> &Module {
468         &self.inner.static_modules[idx]
469     }
470 
471     #[cfg(feature = "profiling")]
472     pub(crate) fn static_modules(&self) -> impl Iterator<Item = &Module> {
473         self.inner.static_modules.values()
474     }
475 
476     #[inline]
477     pub(crate) fn types(&self) -> &Arc<ComponentTypes> {
478         match self.inner.code.types() {
479             crate::code::Types::Component(types) => types,
480             // The only creator of a `Component` is itself which uses the other
481             // variant, so this shouldn't be possible.
482             crate::code::Types::Module(_) => unreachable!(),
483         }
484     }
485 
486     pub(crate) fn signatures(&self) -> &TypeCollection {
487         self.inner.code.signatures()
488     }
489 
490     pub(crate) fn text(&self) -> &[u8] {
491         self.inner.code.code_memory().text()
492     }
493 
494     pub(crate) fn trampoline_ptrs(&self, index: TrampolineIndex) -> AllCallFuncPointers {
495         let AllCallFunc {
496             wasm_call,
497             array_call,
498         } = &self.inner.info.trampolines[index];
499         AllCallFuncPointers {
500             wasm_call: self.func(wasm_call).cast(),
501             array_call: self.func(array_call).cast(),
502         }
503     }
504 
505     fn func(&self, loc: &FunctionLoc) -> NonNull<u8> {
506         let text = self.text();
507         let trampoline = &text[loc.start as usize..][..loc.length as usize];
508         NonNull::from(trampoline).cast()
509     }
510 
511     pub(crate) fn code_object(&self) -> &Arc<CodeObject> {
512         &self.inner.code
513     }
514 
515     /// Same as [`Module::serialize`], except for a component.
516     ///
517     /// Note that the artifact produced here must be passed to
518     /// [`Component::deserialize`] and is not compatible for use with
519     /// [`Module`].
520     ///
521     /// [`Module::serialize`]: crate::Module::serialize
522     /// [`Module`]: crate::Module
523     pub fn serialize(&self) -> Result<Vec<u8>> {
524         Ok(self.code_object().code_memory().mmap().to_vec())
525     }
526 
527     /// Creates a new `VMFuncRef` with all fields filled out for the destructor
528     /// specified.
529     ///
530     /// The `dtor`'s own `VMFuncRef` won't have `wasm_call` filled out but this
531     /// component may have `resource_drop_wasm_to_native_trampoline` filled out
532     /// if necessary in which case it's filled in here.
533     pub(crate) fn resource_drop_func_ref(&self, dtor: &crate::func::HostFunc) -> VMFuncRef {
534         // Host functions never have their `wasm_call` filled in at this time.
535         assert!(dtor.func_ref().wasm_call.is_none());
536 
537         // Note that if `resource_drop_wasm_to_native_trampoline` is not present
538         // then this can't be called by the component, so it's ok to leave it
539         // blank.
540         let wasm_call = self
541             .inner
542             .info
543             .resource_drop_wasm_to_array_trampoline
544             .as_ref()
545             .map(|i| self.func(i).cast().into());
546         VMFuncRef {
547             wasm_call,
548             ..*dtor.func_ref()
549         }
550     }
551 
552     /// Returns a summary of the resources required to instantiate this
553     /// [`Component`][crate::component::Component].
554     ///
555     /// Note that when a component imports and instantiates another component or
556     /// core module, we cannot determine ahead of time how many resources
557     /// instantiating this component will require, and therefore this method
558     /// will return `None` in these scenarios.
559     ///
560     /// Potential uses of the returned information:
561     ///
562     /// * Determining whether your pooling allocator configuration supports
563     ///   instantiating this component.
564     ///
565     /// * Deciding how many of which `Component` you want to instantiate within
566     ///   a fixed amount of resources, e.g. determining whether to create 5
567     ///   instances of component X or 10 instances of component Y.
568     ///
569     /// # Example
570     ///
571     /// ```
572     /// # fn main() -> wasmtime::Result<()> {
573     /// use wasmtime::{Config, Engine, component::Component};
574     ///
575     /// let mut config = Config::new();
576     /// config.wasm_multi_memory(true);
577     /// config.wasm_component_model(true);
578     /// let engine = Engine::new(&config)?;
579     ///
580     /// let component = Component::new(&engine, &r#"
581     ///     (component
582     ///         ;; Define a core module that uses two memories.
583     ///         (core module $m
584     ///             (memory 1)
585     ///             (memory 6)
586     ///         )
587     ///
588     ///         ;; Instantiate that core module three times.
589     ///         (core instance $i1 (instantiate (module $m)))
590     ///         (core instance $i2 (instantiate (module $m)))
591     ///         (core instance $i3 (instantiate (module $m)))
592     ///     )
593     /// "#)?;
594     ///
595     /// let resources = component.resources_required()
596     ///     .expect("this component does not import any core modules or instances");
597     ///
598     /// // Instantiating the component will require allocating two memories per
599     /// // core instance, and there are three instances, so six total memories.
600     /// assert_eq!(resources.num_memories, 6);
601     /// assert_eq!(resources.max_initial_memory_size, Some(6));
602     ///
603     /// // The component doesn't need any tables.
604     /// assert_eq!(resources.num_tables, 0);
605     /// assert_eq!(resources.max_initial_table_size, None);
606     /// # Ok(()) }
607     /// ```
608     pub fn resources_required(&self) -> Option<ResourcesRequired> {
609         let mut resources = ResourcesRequired {
610             num_memories: 0,
611             max_initial_memory_size: None,
612             num_tables: 0,
613             max_initial_table_size: None,
614         };
615         for init in &self.env_component().initializers {
616             match init {
617                 GlobalInitializer::InstantiateModule(inst) => match inst {
618                     InstantiateModule::Static(index, _) => {
619                         let module = self.static_module(*index);
620                         resources.add(&module.resources_required());
621                     }
622                     InstantiateModule::Import(_, _) => {
623                         // We can't statically determine the resources required
624                         // to instantiate this component.
625                         return None;
626                     }
627                 },
628                 GlobalInitializer::LowerImport { .. }
629                 | GlobalInitializer::ExtractMemory(_)
630                 | GlobalInitializer::ExtractTable(_)
631                 | GlobalInitializer::ExtractRealloc(_)
632                 | GlobalInitializer::ExtractCallback(_)
633                 | GlobalInitializer::ExtractPostReturn(_)
634                 | GlobalInitializer::Resource(_) => {}
635             }
636         }
637         Some(resources)
638     }
639 
640     /// Returns the range, in the host's address space, that this module's
641     /// compiled code resides at.
642     ///
643     /// For more information see
644     /// [`Module::image_range`](crate::Module::image_range).
645     pub fn image_range(&self) -> Range<*const u8> {
646         self.inner.code.code_memory().mmap().image_range()
647     }
648 
649     /// Force initialization of copy-on-write images to happen here-and-now
650     /// instead of when they're requested during first instantiation.
651     ///
652     /// When [copy-on-write memory
653     /// initialization](crate::Config::memory_init_cow) is enabled then Wasmtime
654     /// will lazily create the initialization image for a component. This method
655     /// can be used to explicitly dictate when this initialization happens.
656     ///
657     /// Note that this largely only matters on Linux when memfd is used.
658     /// Otherwise the copy-on-write image typically comes from disk and in that
659     /// situation the creation of the image is trivial as the image is always
660     /// sourced from disk. On Linux, though, when memfd is used a memfd is
661     /// created and the initialization image is written to it.
662     ///
663     /// Also note that this method is not required to be called, it's available
664     /// as a performance optimization if required but is otherwise handled
665     /// automatically.
666     pub fn initialize_copy_on_write_image(&self) -> Result<()> {
667         for (_, module) in self.inner.static_modules.iter() {
668             module.initialize_copy_on_write_image()?;
669         }
670         Ok(())
671     }
672 
673     /// Looks up a specific export of this component by `name` optionally nested
674     /// within the `instance` provided.
675     ///
676     /// See related method [`Self::get_export`] for additional docs and
677     /// examples.
678     ///
679     /// This method is primarily used to acquire a [`ComponentExportIndex`]
680     /// which can be used with [`Instance`](crate::component::Instance) when
681     /// looking up exports. Export lookup with [`ComponentExportIndex`] can
682     /// skip string lookups at runtime and instead use a more efficient
683     /// index-based lookup.
684     ///
685     /// This method only returns the [`ComponentExportIndex`]. If you need the
686     /// corresponding [`types::ComponentItem`], use the related function
687     /// [`Self::get_export`].
688     ///
689     ///
690     /// [`Instance`](crate::component::Instance) has a corresponding method
691     /// [`Instance::get_export_index`](crate::component::Instance::get_export_index).
692     pub fn get_export_index(
693         &self,
694         instance: Option<&ComponentExportIndex>,
695         name: &str,
696     ) -> Option<ComponentExportIndex> {
697         let index = self.lookup_export_index(instance, name)?;
698         Some(ComponentExportIndex {
699             id: self.inner.id,
700             index,
701         })
702     }
703 
704     /// Looks up a specific export of this component by `name` optionally nested
705     /// within the `instance` provided.
706     ///
707     /// This method is primarily used to acquire a [`ComponentExportIndex`]
708     /// which can be used with [`Instance`](crate::component::Instance) when
709     /// looking up exports. Export lookup with [`ComponentExportIndex`] can
710     /// skip string lookups at runtime and instead use a more efficient
711     /// index-based lookup.
712     ///
713     /// This method takes a few arguments:
714     ///
715     /// * `engine` - the engine that was used to compile this component.
716     /// * `instance` - an optional "parent instance" for the export being looked
717     ///   up. If this is `None` then the export is looked up on the root of the
718     ///   component itself, and otherwise the export is looked up on the
719     ///   `instance` specified. Note that `instance` must have come from a
720     ///   previous invocation of this method.
721     /// * `name` - the name of the export that's being looked up.
722     ///
723     /// If the export is located then two values are returned: a
724     /// [`types::ComponentItem`] which enables introspection about the type of
725     /// the export and a [`ComponentExportIndex`]. The index returned notably
726     /// implements the [`InstanceExportLookup`] trait which enables using it
727     /// with [`Instance::get_func`](crate::component::Instance::get_func) for
728     /// example.
729     ///
730     /// The returned [`types::ComponentItem`] is more expensive to calculate
731     /// than the [`ComponentExportIndex`]. If you only consume the
732     /// [`ComponentExportIndex`], use the related method
733     /// [`Self::get_export_index`] instead.
734     ///
735     /// [`Instance`](crate::component::Instance) has a corresponding method
736     /// [`Instance::get_export`](crate::component::Instance::get_export).
737     ///
738     /// # Examples
739     ///
740     /// ```
741     /// use wasmtime::{Engine, Store};
742     /// use wasmtime::component::{Component, Linker};
743     /// use wasmtime::component::types::ComponentItem;
744     ///
745     /// # fn main() -> wasmtime::Result<()> {
746     /// let engine = Engine::default();
747     /// let component = Component::new(
748     ///     &engine,
749     ///     r#"
750     ///         (component
751     ///             (core module $m
752     ///                 (func (export "f"))
753     ///             )
754     ///             (core instance $i (instantiate $m))
755     ///             (func (export "f")
756     ///                 (canon lift (core func $i "f")))
757     ///         )
758     ///     "#,
759     /// )?;
760     ///
761     /// // Perform a lookup of the function "f" before instantiaton.
762     /// let (ty, export) = component.get_export(None, "f").unwrap();
763     /// assert!(matches!(ty, ComponentItem::ComponentFunc(_)));
764     ///
765     /// // After instantiation use `export` to lookup the function in question
766     /// // which notably does not do a string lookup at runtime.
767     /// let mut store = Store::new(&engine, ());
768     /// let instance = Linker::new(&engine).instantiate(&mut store, &component)?;
769     /// let func = instance.get_typed_func::<(), ()>(&mut store, &export)?;
770     /// // ...
771     /// # Ok(())
772     /// # }
773     /// ```
774     pub fn get_export(
775         &self,
776         instance: Option<&ComponentExportIndex>,
777         name: &str,
778     ) -> Option<(types::ComponentItem, ComponentExportIndex)> {
779         let info = self.env_component();
780         let index = self.lookup_export_index(instance, name)?;
781         let item = self.with_uninstantiated_instance_type(|instance| {
782             types::ComponentItem::from_export(
783                 &self.inner.engine,
784                 &info.export_items[index],
785                 instance,
786             )
787         });
788         Some((
789             item,
790             ComponentExportIndex {
791                 id: self.inner.id,
792                 index,
793             },
794         ))
795     }
796 
797     pub(crate) fn lookup_export_index(
798         &self,
799         instance: Option<&ComponentExportIndex>,
800         name: &str,
801     ) -> Option<ExportIndex> {
802         let info = self.env_component();
803         let exports = match instance {
804             Some(idx) => {
805                 if idx.id != self.inner.id {
806                     return None;
807                 }
808                 match &info.export_items[idx.index] {
809                     Export::Instance { exports, .. } => exports,
810                     _ => return None,
811                 }
812             }
813             None => &info.exports,
814         };
815         exports.get(name, &NameMapNoIntern).copied()
816     }
817 
818     pub(crate) fn id(&self) -> CompiledModuleId {
819         self.inner.id
820     }
821 
822     /// Returns the [`Engine`] that this [`Component`] was compiled by.
823     pub fn engine(&self) -> &Engine {
824         &self.inner.engine
825     }
826 
827     pub(crate) fn realloc_func_ty(&self) -> &Arc<FuncType> {
828         &self.inner.realloc_func_type
829     }
830 
831     /// Returns the `Export::LiftedFunction` metadata associated with `export`.
832     ///
833     /// # Panics
834     ///
835     /// Panics if `export` is out of bounds or if it isn't a `LiftedFunction`.
836     pub(crate) fn export_lifted_function(
837         &self,
838         export: ExportIndex,
839     ) -> (TypeFuncIndex, &CoreDef, OptionsIndex) {
840         let component = self.env_component();
841         match &component.export_items[export] {
842             Export::LiftedFunction { ty, func, options } => (*ty, func, *options),
843             _ => unreachable!(),
844         }
845     }
846 }
847 
848 /// A value which represents a known export of a component.
849 ///
850 /// This is the return value of [`Component::get_export`] and implements the
851 /// [`InstanceExportLookup`] trait to work with lookups like
852 /// [`Instance::get_func`](crate::component::Instance::get_func).
853 #[derive(Copy, Clone, Debug, Hash, Eq, PartialEq)]
854 pub struct ComponentExportIndex {
855     pub(crate) id: CompiledModuleId,
856     pub(crate) index: ExportIndex,
857 }
858 
859 impl InstanceExportLookup for ComponentExportIndex {
860     fn lookup(&self, component: &Component) -> Option<ExportIndex> {
861         if component.inner.id == self.id {
862             Some(self.index)
863         } else {
864             None
865         }
866     }
867 }
868 
869 #[cfg(test)]
870 mod tests {
871     use crate::component::Component;
872     use crate::{Config, Engine};
873     use wasmtime_environ::MemoryInitialization;
874 
875     #[test]
876     fn cow_on_by_default() {
877         let mut config = Config::new();
878         config.wasm_component_model(true);
879         let engine = Engine::new(&config).unwrap();
880         let component = Component::new(
881             &engine,
882             r#"
883                 (component
884                     (core module
885                         (memory 1)
886                         (data (i32.const 100) "abcd")
887                     )
888                 )
889             "#,
890         )
891         .unwrap();
892 
893         for (_, module) in component.inner.static_modules.iter() {
894             let init = &module.env_module().memory_initialization;
895             assert!(matches!(init, MemoryInitialization::Static { .. }));
896         }
897     }
898 }
899