Merge pull request #751 from matter-labs/vv-evm-simplify-evmCodeHash

[EVM-Equivalence-YUL] Implement evmCodeHash without mapping
matter-labs · Aug 30, 2024 · b022c9b · b022c9b
2 parents 2fe0652 + 944b6c9
commit b022c9b
Showing 1 changed file with 21 additions and 3 deletions.
diff --git a/system-contracts/contracts/ContractDeployer.sol b/system-contracts/contracts/ContractDeployer.sol
@@ -33,7 +33,7 @@ contract ContractDeployer is IContractDeployer, ISystemContract {
         Deployed
     }
 
-    mapping(address => bytes32) public evmCodeHash;
+    uint256 constant EVM_HASHES_PREFIX = 1 << 254;
 
     uint256 public constructorReturnGas;
 
@@ -49,7 +49,7 @@ contract ContractDeployer is IContractDeployer, ISystemContract {
         uint256 bytecodeLen = uint256(bytes32(paddedNewDeployedCode[:32]));
         bytes memory trueBytecode = paddedNewDeployedCode[32:32 + bytecodeLen];
 
-        evmCodeHash[msg.sender] = keccak256(trueBytecode);
+        _setEvmCodeHash(msg.sender, keccak256(trueBytecode));
         constructorReturnGas = constructorGasLeft;
 
         // ToDO: use efficient call
@@ -64,6 +64,10 @@ contract ContractDeployer is IContractDeployer, ISystemContract {
         _;
     }
 
+    function evmCodeHash(address _address) external view returns (bytes32 _hash) {
+        _hash = _getEvmCodeHash(_address);
+    }
+
     /// @notice Returns information about a certain account.
     function getAccountInfo(address _address) external view returns (AccountInfo memory info) {
         return accountInfo[_address];
@@ -512,6 +516,20 @@ contract ContractDeployer is IContractDeployer, ISystemContract {
             }
         }
 
-        require(evmCodeHash[_newAddress] != 0x0, "The code hash must be set after the constructor call");
+        require(_getEvmCodeHash(_newAddress) != 0x0, "The code hash must be set after the constructor call");
+    }
+
+    function _setEvmCodeHash(address _address, bytes32 _hash) internal {
+        assembly {
+            let slot := or(EVM_HASHES_PREFIX, _address)
+            sstore(slot, _hash)
+        }
+    }
+
+    function _getEvmCodeHash(address _address) internal view returns (bytes32 _hash) {
+        assembly {
+            let slot := or(EVM_HASHES_PREFIX, _address)
+            _hash := sload(slot)
+        }
     }
 }