support EXTCODEHASH (#511)

fuzzland · Jun 21, 2024 · 9ab8f5c · 9ab8f5c
1 parent 8683ce2
commit 9ab8f5c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/src/evm/host.rs b/src/evm/host.rs
@@ -1229,11 +1229,11 @@ where
         }
     }
 
-    fn code_hash(&mut self, _address: EVMAddress) -> Option<(B256, bool)> {
-        Some((
-            B256::from_str("0x0000000000000000000000000000000000000000000000000000000000000000").unwrap(),
-            true,
-        ))
+    fn code_hash(&mut self, address: EVMAddress) -> Option<(B256, bool)> {
+        match self.code.get(&address) {
+            Some(code) => Some((code.hash(), true)),
+            None => Some((B256::zero(), true)),
+        }
     }
 
     fn sload(&mut self, address: EVMAddress, index: EVMU256) -> Option<(EVMU256, bool)> {

diff --git a/src/evm/onchain/mod.rs b/src/evm/onchain/mod.rs
@@ -260,8 +260,8 @@ where
             0x46 => {
                 host.env.tx.chain_id = Some(self.endpoint.chain_id as u64);
             }
-            // CALL | CALLCODE | DELEGATECALL | STATICCALL | EXTCODESIZE | EXTCODECOPY
-            0xf1 | 0xf2 | 0xf4 | 0xfa | 0x3b | 0x3c => {
+            // CALL | CALLCODE | DELEGATECALL | STATICCALL | EXTCODESIZE | EXTCODECOPY | EXTCODEHASH
+            0xf1 | 0xf2 | 0xf4 | 0xfa | 0x3b | 0x3c | 0x3f => {
                 let caller = interp.contract.address;
                 let address = match *interp.instruction_pointer {
                     0xf1 | 0xf2 => {
@@ -275,7 +275,7 @@ where
                         interp.stack.peek(1).unwrap()
                     }
                     0xf4 | 0xfa => interp.stack.peek(1).unwrap(),
-                    0x3b | 0x3c => interp.stack.peek(0).unwrap(),
+                    0x3b | 0x3c | 0x3f => interp.stack.peek(0).unwrap(),
                     _ => unreachable!(),
                 };