fix sourcemap issue (#181)

fuzzland · Aug 28, 2023 · 7703d11 · 7703d11
1 parent 45431ef
commit 7703d11
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/evm/middlewares/coverage.rs b/src/evm/middlewares/coverage.rs
@@ -305,7 +305,11 @@ impl<I, VS, S> Middleware<VS, I, S> for Coverage
             self.sources.insert(address, build_artifact.sources.clone());
 
             let sourcemap = build_artifact.get_sourcemap(
-                Vec::from(host.code.get(&address).unwrap().clone().bytecode())
+                if (host.code.contains_key(&address)) {
+                    Vec::from(host.code.get(&address).unwrap().clone().bytecode())
+                } else {
+                    host.setcode_data.get(&address).unwrap().clone().bytecode.to_vec()
+                }
             );
 
             pcs.iter().for_each(|pc| {