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