Make account code function more similar to stylus

OffchainLabs · Sep 27, 2024 · b5fd6ec · b5fd6ec
1 parent cde7f96
commit b5fd6ec
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/mocks/HostioTest.sol b/src/mocks/HostioTest.sol
@@ -76,7 +76,14 @@ contract HostioTest {
     }
 
     function accountCode(address account) external view returns (bytes memory) {
-        return account.code;
+        uint256 size = 10000;
+        bytes memory code = new bytes(size);
+        assembly {
+            extcodecopy(account, add(code, 32), 0, size)
+            size := extcodesize(account)
+            mstore(code, size)
+        }
+        return code;
     }
 
     function accountCodeSize(address account) external view returns (uint256) {