1use wasm_runtime_layer::{
2 ExportType, ExternType, FuncType, GlobalType, ImportType, MemoryType, TableType,
3 backend::{
4 AsContext, AsContextMut, Export, Extern, Imports, Value, WasmEngine, WasmExternRef,
5 WasmFunc, WasmGlobal, WasmInstance, WasmMemory, WasmModule, WasmStore, WasmStoreContext,
6 WasmStoreContextMut, WasmTable,
7 },
8};
9
10use crate::transform::{
11 instcnt::{InstructionCounterInjecter, PerfWitInterfaces},
12 nan::NaNCanonicaliser,
13};
14
15#[derive(Clone)]
16#[repr(transparent)]
17pub struct ReproducibleEngine<E: WasmEngine>(E);
18
19impl<E: WasmEngine> WasmEngine for ReproducibleEngine<E> {
20 type ExternRef = ReproducibleExternRef<E>;
21 type Func = ReproducibleFunc<E>;
22 type Global = ReproducibleGlobal<E>;
23 type Instance = ReproducibleInstance<E>;
24 type Memory = ReproducibleMemory<E>;
25 type Module = ReproducibleModule<E>;
26 type Store<T> = ReproducibleStore<T, E>;
27 type StoreContext<'a, T: 'a> = ReproducibleStoreContext<'a, T, E>;
28 type StoreContextMut<'a, T: 'a> = ReproducibleStoreContextMut<'a, T, E>;
29 type Table = ReproducibleTable<E>;
30}
31
32impl<E: WasmEngine> ReproducibleEngine<E> {
33 pub const fn new(engine: E) -> Self {
34 Self(engine)
35 }
36
37 const fn as_ref(&self) -> &E {
38 &self.0
39 }
40
41 const fn from_ref(engine: &E) -> &Self {
42 #[expect(unsafe_code)]
44 unsafe {
45 &*std::ptr::from_ref(engine).cast()
46 }
47 }
48}
49
50#[derive(Clone)]
51#[repr(transparent)]
52pub struct ReproducibleExternRef<E: WasmEngine>(E::ExternRef);
53
54impl<E: WasmEngine> WasmExternRef<ReproducibleEngine<E>> for ReproducibleExternRef<E> {
55 fn new<T: 'static + Send + Sync>(
56 mut ctx: impl AsContextMut<ReproducibleEngine<E>>,
57 object: T,
58 ) -> Self {
59 Self(<E::ExternRef as WasmExternRef<E>>::new(
60 ctx.as_context_mut().as_inner_context_mut(),
61 object,
62 ))
63 }
64
65 fn downcast<'a, 's: 'a, T: 'static, S: 'a>(
66 &'a self,
67 store: ReproducibleStoreContext<'s, S, E>,
68 ) -> anyhow::Result<&'a T> {
69 WasmExternRef::downcast(&self.0, store.0)
70 }
71}
72
73#[derive(Clone)]
74#[repr(transparent)]
75pub struct ReproducibleFunc<E: WasmEngine>(E::Func);
76
77impl<E: WasmEngine> WasmFunc<ReproducibleEngine<E>> for ReproducibleFunc<E> {
78 fn new<T>(
79 mut ctx: impl AsContextMut<ReproducibleEngine<E>, UserState = T>,
80 ty: FuncType,
81 func: impl 'static
82 + Send
83 + Sync
84 + Fn(
85 ReproducibleStoreContextMut<T, E>,
86 &[Value<ReproducibleEngine<E>>],
87 &mut [Value<ReproducibleEngine<E>>],
88 ) -> anyhow::Result<()>,
89 ) -> Self {
90 Self(<E::Func as WasmFunc<E>>::new(
91 ctx.as_context_mut().as_inner_context_mut(),
92 ty,
93 move |ctx, args, results| {
94 func(
95 ReproducibleStoreContextMut(ctx),
96 from_values(args),
97 from_values_mut(results),
98 )
99 },
100 ))
101 }
102
103 fn ty(&self, ctx: impl AsContext<ReproducibleEngine<E>>) -> FuncType {
104 WasmFunc::ty(&self.0, ctx.as_context().as_inner_context())
105 }
106
107 fn call<T>(
108 &self,
109 mut ctx: impl AsContextMut<ReproducibleEngine<E>>,
110 args: &[Value<ReproducibleEngine<E>>],
111 results: &mut [Value<ReproducibleEngine<E>>],
112 ) -> anyhow::Result<()> {
113 WasmFunc::call::<T>(
114 &self.0,
115 ctx.as_context_mut().as_inner_context_mut(),
116 as_values(args),
117 as_values_mut(results),
118 )
119 }
120}
121
122#[derive(Clone)]
123#[repr(transparent)]
124pub struct ReproducibleGlobal<E: WasmEngine>(E::Global);
125
126impl<E: WasmEngine> WasmGlobal<ReproducibleEngine<E>> for ReproducibleGlobal<E> {
127 fn new(
128 mut ctx: impl AsContextMut<ReproducibleEngine<E>>,
129 value: Value<ReproducibleEngine<E>>,
130 mutable: bool,
131 ) -> Self {
132 Self(<E::Global as WasmGlobal<E>>::new(
133 ctx.as_context_mut().as_inner_context_mut(),
134 into_value(value),
135 mutable,
136 ))
137 }
138
139 fn ty(&self, ctx: impl AsContext<ReproducibleEngine<E>>) -> GlobalType {
140 WasmGlobal::ty(&self.0, ctx.as_context().as_inner_context())
141 }
142
143 fn set(
144 &self,
145 mut ctx: impl AsContextMut<ReproducibleEngine<E>>,
146 new_value: Value<ReproducibleEngine<E>>,
147 ) -> anyhow::Result<()> {
148 WasmGlobal::set(
149 &self.0,
150 ctx.as_context_mut().as_inner_context_mut(),
151 into_value(new_value),
152 )
153 }
154
155 fn get(
156 &self,
157 mut ctx: impl AsContextMut<ReproducibleEngine<E>>,
158 ) -> Value<ReproducibleEngine<E>> {
159 from_value(WasmGlobal::get(
160 &self.0,
161 ctx.as_context_mut().as_inner_context_mut(),
162 ))
163 }
164}
165
166#[derive(Clone)]
167#[repr(transparent)]
168pub struct ReproducibleInstance<E: WasmEngine>(E::Instance);
169
170impl<E: WasmEngine> WasmInstance<ReproducibleEngine<E>> for ReproducibleInstance<E> {
171 fn new(
172 mut store: impl AsContextMut<ReproducibleEngine<E>>,
173 module: &ReproducibleModule<E>,
174 imports: &Imports<ReproducibleEngine<E>>,
175 ) -> anyhow::Result<Self> {
176 let mut new_imports = Imports::new();
177 new_imports.extend(
178 imports
179 .into_iter()
180 .map(|((module, name), value)| ((module, name), into_extern(value))),
181 );
182
183 let PerfWitInterfaces {
184 perf: perf_interface,
185 instruction_counter,
186 } = PerfWitInterfaces::get();
187 new_imports.define(
188 &format!("{perf_interface}"),
189 instruction_counter,
190 Extern::Global(
191 store
192 .as_context_mut()
193 .get_instruction_counter_global()
194 .0
195 .clone(),
196 ),
197 );
198
199 Ok(Self(<E::Instance as WasmInstance<E>>::new(
200 store.as_context_mut().as_inner_context_mut(),
201 &module.0,
202 &new_imports,
203 )?))
204 }
205
206 fn exports(
207 &self,
208 store: impl AsContext<ReproducibleEngine<E>>,
209 ) -> Box<dyn Iterator<Item = Export<ReproducibleEngine<E>>>> {
210 Box::new(
211 WasmInstance::exports(&self.0, store.as_context().as_inner_context()).map(
212 |Export { name, value }| Export {
213 name,
214 value: from_extern(value),
215 },
216 ),
217 )
218 }
219
220 fn get_export(
221 &self,
222 store: impl AsContext<ReproducibleEngine<E>>,
223 name: &str,
224 ) -> Option<Extern<ReproducibleEngine<E>>> {
225 WasmInstance::get_export(&self.0, store.as_context().as_inner_context(), name)
226 .map(from_extern)
227 }
228}
229
230#[derive(Clone)]
231#[repr(transparent)]
232pub struct ReproducibleMemory<E: WasmEngine>(E::Memory);
233
234impl<E: WasmEngine> WasmMemory<ReproducibleEngine<E>> for ReproducibleMemory<E> {
235 fn new(
236 mut ctx: impl AsContextMut<ReproducibleEngine<E>>,
237 ty: MemoryType,
238 ) -> anyhow::Result<Self> {
239 Ok(Self(<E::Memory as WasmMemory<E>>::new(
240 ctx.as_context_mut().as_inner_context_mut(),
241 ty,
242 )?))
243 }
244
245 fn ty(&self, ctx: impl AsContext<ReproducibleEngine<E>>) -> MemoryType {
246 WasmMemory::ty(&self.0, ctx.as_context().as_inner_context())
247 }
248
249 fn grow(
250 &self,
251 mut ctx: impl AsContextMut<ReproducibleEngine<E>>,
252 additional: u32,
253 ) -> anyhow::Result<u32> {
254 WasmMemory::grow(
255 &self.0,
256 ctx.as_context_mut().as_inner_context_mut(),
257 additional,
258 )
259 }
260
261 fn current_pages(&self, ctx: impl AsContext<ReproducibleEngine<E>>) -> u32 {
262 WasmMemory::current_pages(&self.0, ctx.as_context().as_inner_context())
263 }
264
265 fn read(
266 &self,
267 ctx: impl AsContext<ReproducibleEngine<E>>,
268 offset: usize,
269 buffer: &mut [u8],
270 ) -> anyhow::Result<()> {
271 WasmMemory::read(&self.0, ctx.as_context().as_inner_context(), offset, buffer)
272 }
273
274 fn write(
275 &self,
276 mut ctx: impl AsContextMut<ReproducibleEngine<E>>,
277 offset: usize,
278 buffer: &[u8],
279 ) -> anyhow::Result<()> {
280 WasmMemory::write(
281 &self.0,
282 ctx.as_context_mut().as_inner_context_mut(),
283 offset,
284 buffer,
285 )
286 }
287}
288
289pub const DETERMINISTIC_WASM_MODULE_FEATURES: wasmparser::WasmFeaturesInflated =
290 wasmparser::WasmFeaturesInflated {
291 mutable_global: true,
295 saturating_float_to_int: true,
297 sign_extension: true,
299 reference_types: false,
301 multi_value: true,
303 bulk_memory: true,
305 simd: true,
307 relaxed_simd: false,
309 threads: false,
311 shared_everything_threads: false,
313 tail_call: false,
317 floats: false,
319 multi_memory: true,
321 exceptions: false,
323 memory64: false,
326 extended_const: false,
330 component_model: false,
333 function_references: false,
335 memory_control: false,
337 gc: false,
339 custom_page_sizes: false,
343 legacy_exceptions: false,
345 gc_types: false,
348 stack_switching: false,
350 wide_arithmetic: true,
352 cm_values: false,
355 cm_nested_names: false,
358 cm_async: false,
361 cm_async_stackful: false,
364 cm_async_builtins: false,
367 cm_error_context: false,
370 };
371
372#[derive(Clone)]
373#[repr(transparent)]
374pub struct ReproducibleModule<E: WasmEngine>(E::Module);
375
376impl<E: WasmEngine> WasmModule<ReproducibleEngine<E>> for ReproducibleModule<E> {
377 fn new(engine: &ReproducibleEngine<E>, bytes: &[u8]) -> anyhow::Result<Self> {
378 let features = wasmparser::WasmFeatures::from(wasmparser::WasmFeaturesInflated {
379 floats: true,
382 ..DETERMINISTIC_WASM_MODULE_FEATURES
383 });
384
385 wasmparser::Validator::new_with_features(features).validate_all(bytes)?;
386
387 let bytes = InstructionCounterInjecter::apply_to_module(bytes, features)?;
389
390 let bytes = NaNCanonicaliser::apply_to_module(&bytes, features)?;
392
393 Ok(Self(<E::Module as WasmModule<E>>::new(
394 engine.as_ref(),
395 bytes.as_slice(),
396 )?))
397 }
398
399 fn exports(&self) -> Box<dyn '_ + Iterator<Item = ExportType<'_>>> {
400 WasmModule::exports(&self.0)
401 }
402
403 fn get_export(&self, name: &str) -> Option<ExternType> {
404 WasmModule::get_export(&self.0, name)
405 }
406
407 fn imports(&self) -> Box<dyn '_ + Iterator<Item = ImportType<'_>>> {
408 WasmModule::imports(&self.0)
409 }
410}
411
412struct StoreData<T, E: WasmEngine> {
413 data: T,
414 instruction_counter: Option<ReproducibleGlobal<E>>,
415}
416
417#[derive(Clone)]
418#[repr(transparent)]
419pub struct ReproducibleStore<T, E: WasmEngine>(E::Store<StoreData<T, E>>);
420
421impl<T, E: WasmEngine> WasmStore<T, ReproducibleEngine<E>> for ReproducibleStore<T, E> {
422 fn new(engine: &ReproducibleEngine<E>, data: T) -> Self {
423 Self(<E::Store<StoreData<T, E>> as WasmStore<
424 StoreData<T, E>,
425 E,
426 >>::new(
427 engine.as_ref(),
428 StoreData {
429 data,
430 instruction_counter: None,
431 },
432 ))
433 }
434
435 fn engine(&self) -> &ReproducibleEngine<E> {
436 ReproducibleEngine::from_ref(WasmStore::engine(&self.0))
437 }
438
439 fn data(&self) -> &T {
440 &WasmStore::data(&self.0).data
441 }
442
443 fn data_mut(&mut self) -> &mut T {
444 &mut WasmStore::data_mut(&mut self.0).data
445 }
446
447 fn into_data(self) -> T {
448 WasmStore::into_data(self.0).data
449 }
450}
451
452impl<T, E: WasmEngine> AsContext<ReproducibleEngine<E>> for ReproducibleStore<T, E> {
453 type UserState = T;
454
455 fn as_context(&self) -> ReproducibleStoreContext<'_, Self::UserState, E> {
456 ReproducibleStoreContext(AsContext::as_context(&self.0))
457 }
458}
459
460impl<T, E: WasmEngine> AsContextMut<ReproducibleEngine<E>> for ReproducibleStore<T, E> {
461 fn as_context_mut(&mut self) -> ReproducibleStoreContextMut<'_, Self::UserState, E> {
462 ReproducibleStoreContextMut(AsContextMut::as_context_mut(&mut self.0))
463 }
464}
465
466#[repr(transparent)]
467pub struct ReproducibleStoreContext<'a, T: 'a, E: WasmEngine>(E::StoreContext<'a, StoreData<T, E>>);
468
469impl<'a, T: 'a, E: WasmEngine> WasmStoreContext<'a, T, ReproducibleEngine<E>>
470 for ReproducibleStoreContext<'a, T, E>
471{
472 fn engine(&self) -> &ReproducibleEngine<E> {
473 ReproducibleEngine::from_ref(WasmStoreContext::engine(&self.0))
474 }
475
476 fn data(&self) -> &T {
477 &WasmStoreContext::data(&self.0).data
478 }
479}
480
481impl<'a, T: 'a, E: WasmEngine> AsContext<ReproducibleEngine<E>>
482 for ReproducibleStoreContext<'a, T, E>
483{
484 type UserState = T;
485
486 fn as_context(&self) -> ReproducibleStoreContext<'_, Self::UserState, E> {
487 ReproducibleStoreContext(AsContext::as_context(&self.0))
488 }
489}
490
491impl<'a, T: 'a, E: WasmEngine> ReproducibleStoreContext<'a, T, E> {
492 fn as_inner_context(&self) -> E::StoreContext<'_, StoreData<T, E>> {
493 self.0.as_context()
494 }
495}
496
497#[repr(transparent)]
498pub struct ReproducibleStoreContextMut<'a, T: 'a, E: WasmEngine>(
499 E::StoreContextMut<'a, StoreData<T, E>>,
500);
501
502impl<'a, T: 'a, E: WasmEngine> WasmStoreContext<'a, T, ReproducibleEngine<E>>
503 for ReproducibleStoreContextMut<'a, T, E>
504{
505 fn engine(&self) -> &ReproducibleEngine<E> {
506 ReproducibleEngine::from_ref(WasmStoreContext::engine(&self.0))
507 }
508
509 fn data(&self) -> &T {
510 &WasmStoreContext::data(&self.0).data
511 }
512}
513
514impl<'a, T: 'a, E: WasmEngine> WasmStoreContextMut<'a, T, ReproducibleEngine<E>>
515 for ReproducibleStoreContextMut<'a, T, E>
516{
517 fn data_mut(&mut self) -> &mut T {
518 &mut WasmStoreContextMut::data_mut(&mut self.0).data
519 }
520}
521
522impl<'a, T: 'a, E: WasmEngine> AsContext<ReproducibleEngine<E>>
523 for ReproducibleStoreContextMut<'a, T, E>
524{
525 type UserState = T;
526
527 fn as_context(&self) -> ReproducibleStoreContext<'_, Self::UserState, E> {
528 ReproducibleStoreContext(AsContext::as_context(&self.0))
529 }
530}
531
532impl<'a, T: 'a, E: WasmEngine> AsContextMut<ReproducibleEngine<E>>
533 for ReproducibleStoreContextMut<'a, T, E>
534{
535 fn as_context_mut(&mut self) -> ReproducibleStoreContextMut<'_, Self::UserState, E> {
536 ReproducibleStoreContextMut(AsContextMut::as_context_mut(&mut self.0))
537 }
538}
539
540impl<'a, T: 'a, E: WasmEngine> ReproducibleStoreContextMut<'a, T, E> {
541 fn as_inner_context_mut(&mut self) -> E::StoreContextMut<'_, StoreData<T, E>> {
542 self.0.as_context_mut()
543 }
544
545 fn get_instruction_counter_global(&mut self) -> &ReproducibleGlobal<E> {
546 let mut this = self;
547
548 polonius_the_crab::polonius!(|this| -> &'polonius ReproducibleGlobal<E> {
550 let data: &mut StoreData<T, E> = WasmStoreContextMut::data_mut(&mut this.0);
551 if let Some(global) = &data.instruction_counter {
552 polonius_the_crab::polonius_return!(global);
553 }
554 });
555
556 let global = WasmGlobal::new(AsContextMut::as_context_mut(this), Value::I64(0), true);
557
558 let data: &mut StoreData<T, E> = WasmStoreContextMut::data_mut(&mut this.0);
559 data.instruction_counter.insert(global)
560 }
561}
562
563#[derive(Clone)]
564#[repr(transparent)]
565pub struct ReproducibleTable<E: WasmEngine>(E::Table);
566
567impl<E: WasmEngine> WasmTable<ReproducibleEngine<E>> for ReproducibleTable<E> {
568 fn new(
569 mut ctx: impl AsContextMut<ReproducibleEngine<E>>,
570 ty: TableType,
571 init: Value<ReproducibleEngine<E>>,
572 ) -> anyhow::Result<Self> {
573 Ok(Self(<E::Table as WasmTable<E>>::new(
574 ctx.as_context_mut().as_inner_context_mut(),
575 ty,
576 into_value(init),
577 )?))
578 }
579
580 fn ty(&self, ctx: impl AsContext<ReproducibleEngine<E>>) -> TableType {
581 WasmTable::ty(&self.0, ctx.as_context().as_inner_context())
582 }
583
584 fn size(&self, ctx: impl AsContext<ReproducibleEngine<E>>) -> u32 {
585 WasmTable::size(&self.0, ctx.as_context().as_inner_context())
586 }
587
588 fn grow(
589 &self,
590 mut ctx: impl AsContextMut<ReproducibleEngine<E>>,
591 delta: u32,
592 init: Value<ReproducibleEngine<E>>,
593 ) -> anyhow::Result<u32> {
594 WasmTable::grow(
595 &self.0,
596 ctx.as_context_mut().as_inner_context_mut(),
597 delta,
598 into_value(init),
599 )
600 }
601
602 fn get(
603 &self,
604 mut ctx: impl AsContextMut<ReproducibleEngine<E>>,
605 index: u32,
606 ) -> Option<Value<ReproducibleEngine<E>>> {
607 WasmTable::get(&self.0, ctx.as_context_mut().as_inner_context_mut(), index).map(from_value)
608 }
609
610 fn set(
611 &self,
612 mut ctx: impl AsContextMut<ReproducibleEngine<E>>,
613 index: u32,
614 value: Value<ReproducibleEngine<E>>,
615 ) -> anyhow::Result<()> {
616 WasmTable::set(
617 &self.0,
618 ctx.as_context_mut().as_inner_context_mut(),
619 index,
620 into_value(value),
621 )
622 }
623}
624
625const fn as_values<E: WasmEngine>(values: &[Value<ReproducibleEngine<E>>]) -> &[Value<E>] {
626 #[expect(unsafe_code)]
628 unsafe {
629 std::slice::from_raw_parts(values.as_ptr().cast(), values.len())
630 }
631}
632
633const fn as_values_mut<E: WasmEngine>(
634 values: &mut [Value<ReproducibleEngine<E>>],
635) -> &mut [Value<E>] {
636 #[expect(unsafe_code)]
638 unsafe {
639 std::slice::from_raw_parts_mut(values.as_mut_ptr().cast(), values.len())
640 }
641}
642
643const fn from_values<E: WasmEngine>(values: &[Value<E>]) -> &[Value<ReproducibleEngine<E>>] {
644 #[expect(unsafe_code)]
646 unsafe {
647 std::slice::from_raw_parts(values.as_ptr().cast(), values.len())
648 }
649}
650
651const fn from_values_mut<E: WasmEngine>(
652 values: &mut [Value<E>],
653) -> &mut [Value<ReproducibleEngine<E>>] {
654 #[expect(unsafe_code)]
656 unsafe {
657 std::slice::from_raw_parts_mut(values.as_mut_ptr().cast(), values.len())
658 }
659}
660
661fn into_value<E: WasmEngine>(value: Value<ReproducibleEngine<E>>) -> Value<E> {
662 match value {
663 Value::I32(v) => Value::I32(v),
664 Value::I64(v) => Value::I64(v),
665 Value::F32(v) => Value::F32(v),
666 Value::F64(v) => Value::F64(v),
667 Value::FuncRef(v) => Value::FuncRef(v.map(|v| v.0)),
668 Value::ExternRef(v) => Value::ExternRef(v.map(|v| v.0)),
669 }
670}
671
672fn from_value<E: WasmEngine>(value: Value<E>) -> Value<ReproducibleEngine<E>> {
673 match value {
674 Value::I32(v) => Value::I32(v),
675 Value::I64(v) => Value::I64(v),
676 Value::F32(v) => Value::F32(v),
677 Value::F64(v) => Value::F64(v),
678 Value::FuncRef(v) => Value::FuncRef(v.map(ReproducibleFunc)),
679 Value::ExternRef(v) => Value::ExternRef(v.map(ReproducibleExternRef)),
680 }
681}
682
683fn into_extern<E: WasmEngine>(value: Extern<ReproducibleEngine<E>>) -> Extern<E> {
684 match value {
685 Extern::Global(v) => Extern::Global(v.0),
686 Extern::Table(v) => Extern::Table(v.0),
687 Extern::Memory(v) => Extern::Memory(v.0),
688 Extern::Func(v) => Extern::Func(v.0),
689 }
690}
691
692fn from_extern<E: WasmEngine>(value: Extern<E>) -> Extern<ReproducibleEngine<E>> {
693 match value {
694 Extern::Global(v) => Extern::Global(ReproducibleGlobal(v)),
695 Extern::Table(v) => Extern::Table(ReproducibleTable(v)),
696 Extern::Memory(v) => Extern::Memory(ReproducibleMemory(v)),
697 Extern::Func(v) => Extern::Func(ReproducibleFunc(v)),
698 }
699}