1use crate::runtime::vm::VMGcRef;
4use crate::store::{Asyncness, StoreId, StoreResourceLimiter};
5#[cfg(feature = "async")]
6use crate::vm::VMStore;
7use crate::vm::{self, VMExnRef, VMGcHeader};
8use crate::{
9 AsContext, AsContextMut, GcRefImpl, GcRootIndex, HeapType, OwnedRooted, RefType, Rooted, Val,
10 ValRaw, ValType, WasmTy,
11 store::{AutoAssertNoGc, StoreOpaque},
12};
13use crate::{ExnType, FieldType, GcHeapOutOfMemory, StoreContextMut, Tag, prelude::*};
14use alloc::sync::Arc;
15use core::mem;
16use core::mem::MaybeUninit;
17use wasmtime_environ::{GcLayout, GcStructLayout, VMGcKind, VMSharedTypeIndex};
18
19pub struct ExnRefPre {
69 store_id: StoreId,
70 ty: ExnType,
71}
72
73impl ExnRefPre {
74 pub fn new(mut store: impl AsContextMut, ty: ExnType) -> Self {
77 Self::_new(store.as_context_mut().0, ty)
78 }
79
80 pub(crate) fn _new(store: &mut StoreOpaque, ty: ExnType) -> Self {
81 store.insert_gc_host_alloc_type(ty.registered_type().clone());
82 let store_id = store.id();
83 ExnRefPre { store_id, ty }
84 }
85
86 pub(crate) fn layout(&self) -> &GcStructLayout {
87 self.ty
88 .registered_type()
89 .layout()
90 .expect("exn types have a layout")
91 .unwrap_struct()
92 }
93
94 pub(crate) fn type_index(&self) -> VMSharedTypeIndex {
95 self.ty.registered_type().index()
96 }
97}
98
99#[derive(Debug)]
110#[repr(transparent)]
111pub struct ExnRef {
112 pub(super) inner: GcRootIndex,
113}
114
115unsafe impl GcRefImpl for ExnRef {
116 fn transmute_ref(index: &GcRootIndex) -> &Self {
117 let me: &Self = unsafe { mem::transmute(index) };
119
120 assert!(matches!(
122 me,
123 Self {
124 inner: GcRootIndex { .. },
125 }
126 ));
127
128 me
129 }
130}
131
132impl ExnRef {
133 pub fn from_raw(mut store: impl AsContextMut, raw: u32) -> Option<Rooted<Self>> {
164 let mut store = AutoAssertNoGc::new(store.as_context_mut().0);
165 Self::_from_raw(&mut store, raw)
166 }
167
168 pub(crate) fn _from_raw(store: &mut AutoAssertNoGc, raw: u32) -> Option<Rooted<Self>> {
170 let gc_ref = VMGcRef::from_raw_u32(raw)?;
171 let gc_ref = store.clone_gc_ref(&gc_ref);
172 Some(Self::from_cloned_gc_ref(store, gc_ref))
173 }
174
175 pub fn new(
205 mut store: impl AsContextMut,
206 allocator: &ExnRefPre,
207 tag: &Tag,
208 fields: &[Val],
209 ) -> Result<Rooted<ExnRef>> {
210 let (mut limiter, store) = store
211 .as_context_mut()
212 .0
213 .validate_sync_resource_limiter_and_store_opaque()?;
214 vm::assert_ready(Self::_new_async(
215 store,
216 limiter.as_mut(),
217 allocator,
218 tag,
219 fields,
220 Asyncness::No,
221 ))
222 }
223
224 #[cfg(feature = "async")]
249 pub async fn new_async(
250 mut store: impl AsContextMut,
251 allocator: &ExnRefPre,
252 tag: &Tag,
253 fields: &[Val],
254 ) -> Result<Rooted<ExnRef>> {
255 let (mut limiter, store) = store.as_context_mut().0.resource_limiter_and_store_opaque();
256 Self::_new_async(
257 store,
258 limiter.as_mut(),
259 allocator,
260 tag,
261 fields,
262 Asyncness::Yes,
263 )
264 .await
265 }
266
267 pub(crate) async fn _new_async(
268 store: &mut StoreOpaque,
269 limiter: Option<&mut StoreResourceLimiter<'_>>,
270 allocator: &ExnRefPre,
271 tag: &Tag,
272 fields: &[Val],
273 asyncness: Asyncness,
274 ) -> Result<Rooted<ExnRef>> {
275 Self::type_check_tag_and_fields(store, allocator, tag, fields)?;
276 store
277 .retry_after_gc_async(limiter, (), asyncness, |store, ()| {
278 Self::new_unchecked(store, allocator, tag, fields)
279 })
280 .await
281 }
282
283 fn type_check_tag_and_fields(
286 store: &mut StoreOpaque,
287 allocator: &ExnRefPre,
288 tag: &Tag,
289 fields: &[Val],
290 ) -> Result<(), Error> {
291 assert!(
292 tag.comes_from_same_store(store),
293 "tag comes from the wrong store"
294 );
295 ensure!(
296 tag.wasmtime_ty(store).signature.unwrap_engine_type_index()
297 == allocator.ty.tag_type().ty().type_index(),
298 "incorrect signature for tag when creating exception object"
299 );
300 let expected_len = allocator.ty.fields().len();
301 let actual_len = fields.len();
302 ensure!(
303 actual_len == expected_len,
304 "expected {expected_len} fields, got {actual_len}"
305 );
306 for (ty, val) in allocator.ty.fields().zip(fields) {
307 assert!(
308 val.comes_from_same_store(store),
309 "field value comes from the wrong store",
310 );
311 let ty = ty.element_type().unpack();
312 val.ensure_matches_ty(store, ty)
313 .context("field type mismatch")?;
314 }
315 Ok(())
316 }
317
318 fn new_unchecked(
323 store: &mut StoreOpaque,
324 allocator: &ExnRefPre,
325 tag: &Tag,
326 fields: &[Val],
327 ) -> Result<Rooted<ExnRef>> {
328 assert_eq!(
329 store.id(),
330 allocator.store_id,
331 "attempted to use a `ExnRefPre` with the wrong store"
332 );
333
334 let exnref = store
337 .require_gc_store_mut()?
338 .alloc_uninit_exn(allocator.type_index(), &allocator.layout())
339 .context("unrecoverable error when allocating new `exnref`")?
340 .map_err(|n| GcHeapOutOfMemory::new((), n))?;
341
342 let mut store = AutoAssertNoGc::new(store);
347 match (|| {
348 let (instance, index) = tag.to_raw_indices();
349 exnref.initialize_tag(&mut store, instance, index)?;
350 for (index, (ty, val)) in allocator.ty.fields().zip(fields).enumerate() {
351 exnref.initialize_field(
352 &mut store,
353 allocator.layout(),
354 ty.element_type(),
355 index,
356 *val,
357 )?;
358 }
359 Ok(())
360 })() {
361 Ok(()) => Ok(Rooted::new(&mut store, exnref.into())),
362 Err(e) => {
363 store.require_gc_store_mut()?.dealloc_uninit_exn(exnref)?;
364 Err(e)
365 }
366 }
367 }
368
369 pub(crate) fn type_index(&self, store: &StoreOpaque) -> Result<VMSharedTypeIndex> {
370 let gc_ref = self.inner.try_gc_ref(store)?;
371 let header = store.require_gc_store()?.header(gc_ref)?;
372 debug_assert!(header.kind().matches(VMGcKind::ExnRef));
373 Ok(header.ty().expect("exnrefs should have concrete types"))
374 }
375
376 pub(crate) fn from_cloned_gc_ref(
383 store: &mut AutoAssertNoGc<'_>,
384 gc_ref: VMGcRef,
385 ) -> Rooted<Self> {
386 debug_assert!(
387 store
388 .unwrap_gc_store()
389 .kind(&gc_ref)
390 .unwrap()
391 .matches(VMGcKind::ExnRef)
392 );
393 Rooted::new(store, gc_ref)
394 }
395
396 #[inline]
397 pub(crate) fn comes_from_same_store(&self, store: &StoreOpaque) -> bool {
398 self.inner.comes_from_same_store(store)
399 }
400
401 pub fn to_raw(&self, mut store: impl AsContextMut) -> Result<u32> {
414 let mut store = AutoAssertNoGc::new(store.as_context_mut().0);
415 self._to_raw(&mut store)
416 }
417
418 pub(crate) fn _to_raw(&self, store: &mut AutoAssertNoGc<'_>) -> Result<u32> {
419 self.inner.expose_gc_ref_to_wasm(store).map(|r| r.get())
420 }
421
422 pub fn ty(&self, store: impl AsContext) -> Result<ExnType> {
432 self._ty(store.as_context().0)
433 }
434
435 pub(crate) fn _ty(&self, store: &StoreOpaque) -> Result<ExnType> {
436 assert!(self.comes_from_same_store(store));
437 let index = self.type_index(store)?;
438 Ok(ExnType::from_shared_type_index(store.engine(), index))
439 }
440
441 pub fn matches_ty(&self, store: impl AsContext, ty: &HeapType) -> Result<bool> {
453 self._matches_ty(store.as_context().0, ty)
454 }
455
456 pub(crate) fn _matches_ty(&self, store: &StoreOpaque, ty: &HeapType) -> Result<bool> {
457 assert!(self.comes_from_same_store(store));
458 Ok(HeapType::from(self._ty(store)?).matches(ty))
459 }
460
461 pub(crate) fn ensure_matches_ty(&self, store: &StoreOpaque, ty: &HeapType) -> Result<()> {
462 if !self.comes_from_same_store(store) {
463 bail!("function used with wrong store");
464 }
465 if self._matches_ty(store, ty)? {
466 Ok(())
467 } else {
468 let actual_ty = self._ty(store)?;
469 bail!("type mismatch: expected `(ref {ty})`, found `(ref {actual_ty})`")
470 }
471 }
472
473 pub fn fields<'a, T: 'static>(
483 &'a self,
484 store: impl Into<StoreContextMut<'a, T>>,
485 ) -> Result<impl ExactSizeIterator<Item = Val> + 'a> {
486 self._fields(store.into().0)
487 }
488
489 pub(crate) fn _fields<'a>(
490 &'a self,
491 store: &'a mut StoreOpaque,
492 ) -> Result<impl ExactSizeIterator<Item = Val> + 'a> {
493 assert!(self.comes_from_same_store(store));
494 let store = AutoAssertNoGc::new(store);
495
496 let gc_ref = self.inner.try_gc_ref(&store)?;
497 let header = store.require_gc_store()?.header(gc_ref)?;
498 debug_assert!(header.kind().matches(VMGcKind::ExnRef));
499
500 let index = header.ty().expect("exnrefs should have concrete types");
501 let ty = ExnType::from_shared_type_index(store.engine(), index);
502 let len = ty.fields().len();
503
504 return Ok(Fields {
505 exnref: self,
506 store,
507 index: 0,
508 len,
509 });
510
511 struct Fields<'a, 'b> {
512 exnref: &'a ExnRef,
513 store: AutoAssertNoGc<'b>,
514 index: usize,
515 len: usize,
516 }
517
518 impl Iterator for Fields<'_, '_> {
519 type Item = Val;
520
521 #[inline]
522 fn next(&mut self) -> Option<Self::Item> {
523 let i = self.index;
524 debug_assert!(i <= self.len);
525 if i >= self.len {
526 return None;
527 }
528 self.index += 1;
529 self.exnref._field(&mut self.store, i).ok()
530 }
531
532 #[inline]
533 fn size_hint(&self) -> (usize, Option<usize>) {
534 let len = self.len - self.index;
535 (len, Some(len))
536 }
537 }
538
539 impl ExactSizeIterator for Fields<'_, '_> {
540 #[inline]
541 fn len(&self) -> usize {
542 self.len - self.index
543 }
544 }
545 }
546
547 fn header<'a>(&self, store: &'a AutoAssertNoGc<'_>) -> Result<&'a VMGcHeader> {
548 assert!(self.comes_from_same_store(&store));
549 let gc_ref = self.inner.try_gc_ref(store)?;
550 Ok(store.require_gc_store()?.header(gc_ref)?)
551 }
552
553 fn exnref<'a>(&self, store: &'a AutoAssertNoGc<'_>) -> Result<&'a VMExnRef> {
554 assert!(self.comes_from_same_store(&store));
555 let gc_ref = self.inner.try_gc_ref(store)?;
556 debug_assert!(self.header(store)?.kind().matches(VMGcKind::ExnRef));
557 Ok(gc_ref.as_exnref_unchecked())
558 }
559
560 fn layout(&self, store: &AutoAssertNoGc<'_>) -> Result<Arc<GcStructLayout>> {
561 assert!(self.comes_from_same_store(&store));
562 let type_index = self.type_index(store)?;
563 let layout = store
564 .engine()
565 .signatures()
566 .layout(type_index)
567 .expect("exn types should have GC layouts");
568 match layout {
569 GcLayout::Struct(s) => Ok(s),
570 GcLayout::Array(_) => unreachable!(),
571 }
572 }
573
574 fn field_ty(&self, store: &StoreOpaque, field: usize) -> Result<FieldType> {
575 let ty = self._ty(store)?;
576 match ty.field(field) {
577 Some(f) => Ok(f),
578 None => {
579 let len = ty.fields().len();
580 bail!("cannot access field {field}: exn only has {len} fields")
581 }
582 }
583 }
584
585 pub fn field(&self, mut store: impl AsContextMut, index: usize) -> Result<Val> {
596 let mut store = AutoAssertNoGc::new(store.as_context_mut().0);
597 self._field(&mut store, index)
598 }
599
600 pub(crate) fn _field(&self, store: &mut AutoAssertNoGc<'_>, index: usize) -> Result<Val> {
601 assert!(self.comes_from_same_store(store));
602 let exnref = self.exnref(store)?.unchecked_copy();
603 let field_ty = self.field_ty(store, index)?;
604 let layout = self.layout(store)?;
605 exnref.read_field(store, &layout, field_ty.element_type(), index)
606 }
607
608 pub fn tag(&self, mut store: impl AsContextMut) -> Result<Tag> {
618 let mut store = AutoAssertNoGc::new(store.as_context_mut().0);
619 assert!(self.comes_from_same_store(&store));
620 let exnref = self.exnref(&store)?.unchecked_copy();
621 let (instance, index) = exnref.tag(&mut store)?;
622 Ok(Tag::from_raw_indices(&*store, instance, index))
623 }
624}
625
626unsafe impl WasmTy for Rooted<ExnRef> {
627 #[inline]
628 fn valtype() -> ValType {
629 ValType::Ref(RefType::new(false, HeapType::Exn))
630 }
631
632 #[inline]
633 fn compatible_with_store(&self, store: &StoreOpaque) -> bool {
634 self.comes_from_same_store(store)
635 }
636
637 #[inline]
638 fn dynamic_concrete_type_check(
639 &self,
640 _store: &StoreOpaque,
641 _nullable: bool,
642 _ty: &HeapType,
643 ) -> Result<()> {
644 Ok(())
647 }
648
649 fn store(self, store: &mut AutoAssertNoGc<'_>, ptr: &mut MaybeUninit<ValRaw>) -> Result<()> {
650 self.wasm_ty_store(store, ptr, ValRaw::anyref)
651 }
652
653 unsafe fn load(store: &mut AutoAssertNoGc<'_>, ptr: &ValRaw) -> Self {
654 Self::wasm_ty_load(store, ptr.get_anyref(), ExnRef::from_cloned_gc_ref)
655 }
656}
657
658unsafe impl WasmTy for Option<Rooted<ExnRef>> {
659 #[inline]
660 fn valtype() -> ValType {
661 ValType::EXNREF
662 }
663
664 #[inline]
665 fn compatible_with_store(&self, store: &StoreOpaque) -> bool {
666 self.map_or(true, |x| x.comes_from_same_store(store))
667 }
668
669 #[inline]
670 fn dynamic_concrete_type_check(
671 &self,
672 store: &StoreOpaque,
673 nullable: bool,
674 ty: &HeapType,
675 ) -> Result<()> {
676 match self {
677 Some(a) => a.ensure_matches_ty(store, ty),
678 None => {
679 ensure!(
680 nullable,
681 "expected a non-null reference, but found a null reference"
682 );
683 Ok(())
684 }
685 }
686 }
687
688 #[inline]
689 fn is_vmgcref_and_points_to_object(&self) -> bool {
690 self.is_some()
691 }
692
693 fn store(self, store: &mut AutoAssertNoGc<'_>, ptr: &mut MaybeUninit<ValRaw>) -> Result<()> {
694 <Rooted<ExnRef>>::wasm_ty_option_store(self, store, ptr, ValRaw::anyref)
695 }
696
697 unsafe fn load(store: &mut AutoAssertNoGc<'_>, ptr: &ValRaw) -> Self {
698 <Rooted<ExnRef>>::wasm_ty_option_load(store, ptr.get_anyref(), ExnRef::from_cloned_gc_ref)
699 }
700}
701
702unsafe impl WasmTy for OwnedRooted<ExnRef> {
703 #[inline]
704 fn valtype() -> ValType {
705 ValType::Ref(RefType::new(false, HeapType::Exn))
706 }
707
708 #[inline]
709 fn compatible_with_store(&self, store: &StoreOpaque) -> bool {
710 self.comes_from_same_store(store)
711 }
712
713 #[inline]
714 fn dynamic_concrete_type_check(
715 &self,
716 store: &StoreOpaque,
717 _nullable: bool,
718 ty: &HeapType,
719 ) -> Result<()> {
720 self.ensure_matches_ty(store, ty)
721 }
722
723 fn store(self, store: &mut AutoAssertNoGc<'_>, ptr: &mut MaybeUninit<ValRaw>) -> Result<()> {
724 self.wasm_ty_store(store, ptr, ValRaw::anyref)
725 }
726
727 unsafe fn load(store: &mut AutoAssertNoGc<'_>, ptr: &ValRaw) -> Self {
728 Self::wasm_ty_load(store, ptr.get_anyref(), ExnRef::from_cloned_gc_ref)
729 }
730}
731
732unsafe impl WasmTy for Option<OwnedRooted<ExnRef>> {
733 #[inline]
734 fn valtype() -> ValType {
735 ValType::EXNREF
736 }
737
738 #[inline]
739 fn compatible_with_store(&self, store: &StoreOpaque) -> bool {
740 self.as_ref()
741 .map_or(true, |x| x.comes_from_same_store(store))
742 }
743
744 #[inline]
745 fn dynamic_concrete_type_check(
746 &self,
747 store: &StoreOpaque,
748 nullable: bool,
749 ty: &HeapType,
750 ) -> Result<()> {
751 match self {
752 Some(a) => a.ensure_matches_ty(store, ty),
753 None => {
754 ensure!(
755 nullable,
756 "expected a non-null reference, but found a null reference"
757 );
758 Ok(())
759 }
760 }
761 }
762
763 #[inline]
764 fn is_vmgcref_and_points_to_object(&self) -> bool {
765 self.is_some()
766 }
767
768 fn store(self, store: &mut AutoAssertNoGc<'_>, ptr: &mut MaybeUninit<ValRaw>) -> Result<()> {
769 <OwnedRooted<ExnRef>>::wasm_ty_option_store(self, store, ptr, ValRaw::anyref)
770 }
771
772 unsafe fn load(store: &mut AutoAssertNoGc<'_>, ptr: &ValRaw) -> Self {
773 <OwnedRooted<ExnRef>>::wasm_ty_option_load(
774 store,
775 ptr.get_anyref(),
776 ExnRef::from_cloned_gc_ref,
777 )
778 }
779}