Skip to content

test_extcodehash_in_init_code()

Documentation for tests/constantinople/eip1052_extcodehash/test_extcodehash.py::test_extcodehash_in_init_code@b314d18e.

Generate fixtures for these test cases for Amsterdam with:

fill -v tests/constantinople/eip1052_extcodehash/test_extcodehash.py::test_extcodehash_in_init_code --fork Amsterdam

Test EXTCODEHASH/EXTCODESIZE of an external account during init code.

The init code queries EXTCODEHASH and EXTCODESIZE of a pre-existing contract and stores the results. With a create transaction the checks run in the top-level init code; with CREATE/CREATE2 they run in a contract-initiated creation.

Source code in tests/constantinople/eip1052_extcodehash/test_extcodehash.py
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
@pytest.mark.ported_from(
    [
        "https://github.com/ethereum/tests/blob/v13.3/src/GeneralStateTestsFiller/stExtCodeHash/extCodeHashInInitCodeFiller.json",  # noqa: E501
    ],
)
@pytest.mark.parametrize(
    "create_opcode",
    [pytest.param(None, id="create_tx"), Op.CREATE, Op.CREATE2],
)
def test_extcodehash_in_init_code(
    state_test: StateTestFiller,
    pre: Alloc,
    create_opcode: Opcodes | None,
) -> None:
    """
    Test EXTCODEHASH/EXTCODESIZE of an external account during init code.

    The init code queries EXTCODEHASH and EXTCODESIZE of a pre-existing
    contract and stores the results. With a create transaction the checks
    run in the top-level init code; with CREATE/CREATE2 they run in a
    contract-initiated creation.
    """
    storage = Storage()
    target_code = b"\x11\x22\x33\x44"
    target = pre.deploy_contract(target_code)

    expected_hash = keccak256(target_code)
    expected_size = len(target_code)

    # Init code: execute EXTCODEHASH/EXTCODESIZE checks, then deploy.
    checks = Op.SSTORE(
        storage.store_next(expected_hash),
        Op.EXTCODEHASH(target),
    ) + Op.SSTORE(
        storage.store_next(expected_size),
        Op.EXTCODESIZE(target),
    )
    initcode = checks + Op.RETURN(0, 0)

    if create_opcode is None:
        # Transaction-level creation: init code runs directly.
        sender = pre.fund_eoa()
        tx = Transaction(
            sender=sender,
            to=None,
            data=initcode,
        )
        created = compute_create_address(
            address=sender,
            nonce=0,
        )
    else:
        # Contract-initiated CREATE/CREATE2: copy initcode from calldata.
        factory_code = (
            Op.CALLDATACOPY(0, 0, Op.CALLDATASIZE)
            + create_opcode(
                value=0,
                offset=0,
                size=Op.CALLDATASIZE,
            )
            + Op.STOP
        )
        factory = pre.deploy_contract(factory_code)
        tx = Transaction(
            sender=pre.fund_eoa(),
            to=factory,
            data=initcode,
        )
        created = compute_create_address(
            address=factory,
            nonce=1,
            salt=0,
            initcode=initcode,
            opcode=create_opcode,
        )

    state_test(
        pre=pre,
        post={created: Account(storage=storage)},
        tx=tx,
    )

Parametrized Test Cases

This test generates 3 parametrized test cases across 10 forks.