Propagate shared function type into the code builder

crates/wasm-smith/src/core.rs

@@ -2258,8 +2258,9 @@ impl Module {
 
         self.code.reserve(self.num_defined_funcs);
         let mut allocs = CodeBuilderAllocations::new(self, self.config.exports.is_some());
-        for (_, ty) in self.funcs[self.funcs.len() - self.num_defined_funcs..].iter() {
-            let body = self.arbitrary_func_body(u, ty, &mut allocs)?;
+        for (idx, ty) in self.funcs[self.funcs.len() - self.num_defined_funcs..].iter() {
+            let shared = self.is_shared_type(*idx);
+            let body = self.arbitrary_func_body(u, ty, &mut allocs, shared)?;
             self.code.push(body);
         }
         allocs.finish(u, self)?;
@@ -2271,9 +2272,10 @@ impl Module {
         u: &mut Unstructured,
         ty: &FuncType,
         allocs: &mut CodeBuilderAllocations,
+        shared: bool,
     ) -> Result<Code> {
         let mut locals = self.arbitrary_locals(u)?;
-        let builder = allocs.builder(ty, &mut locals);
+        let builder = allocs.builder(ty, &mut locals, shared);
         let instructions = if self.config.allow_invalid_funcs && u.arbitrary().unwrap_or(false) {
             Instructions::Arbitrary(arbitrary_vec_u8(u)?)
         } else {

crates/wasm-smith/src/core/code_builder.rs

@@ -668,6 +668,7 @@ pub(crate) struct CodeBuilderAllocations {
 }
 
 pub(crate) struct CodeBuilder<'a> {
+    shared: bool,
     func_ty: &'a FuncType,
     locals: &'a mut Vec<ValType>,
     allocs: &'a mut CodeBuilderAllocations,
@@ -903,6 +904,7 @@ impl CodeBuilderAllocations {
         &'a mut self,
         func_ty: &'a FuncType,
         locals: &'a mut Vec<ValType>,
+        shared: bool,
     ) -> CodeBuilder<'a> {
         self.controls.clear();
         self.controls.push(Control {
@@ -916,6 +918,7 @@ impl CodeBuilderAllocations {
         self.options.clear();
 
         CodeBuilder {
+            shared,
             func_ty,
             locals,
             allocs: self,