Contract opcodes to compare gas cost for each precompile

Include all test cases and expect precompiles at addresses 01-09
Only include case 0A and 0B to verify no precompile in all forks (except cancun)

Author: reedsa

tests/frontier/precompiles/test_precompiles.py

@@ -0,0 +1,107 @@
+import pytest
+
+from ethereum_test_tools import (
+    Account,
+    Address,
+    Alloc,
+    Environment,
+    StateTestFiller,
+    Transaction,
+)
+from ethereum_test_tools.vm.opcode import Opcodes as Op
+
+
+def precompile_addresses(fork):
+    """
+    Return the precompile addresses for the given fork.
+
+    Args:
+        fork (str): The fork name.
+
+    Returns:
+        list[tuple[str, bool]]: The precompile addresses and whether they exist.
+
+    """
+    return [
+        ("01", True),
+        ("02", True),
+        ("03", True),
+        ("04", True),
+        ("05", True),
+        ("06", True),
+        ("07", True),
+        ("08", True),
+        ("09", True),
+        ("0A", True) if fork == "cancun" else ("0A", False),
+        ("0B", False),
+    ]
+
+
+@pytest.mark.parametrize_by_fork("address,precompile_exists", precompile_addresses)
+def test_precompiles(
+    state_test: StateTestFiller, pre: Alloc, address: str, precompile_exists: bool
+):
+    """Test the MODEXP precompile."""
+    env = Environment()
+
+    address_10000 = 0x10000
+
+    input_buf = 0x1000
+    output_buf = 0x2000
+
+    args_offset = 0x1000
+    args_size = 0x20
+    output_offset = 0x2000
+    output_size = 0x20
+
+    gas_test = 0x00
+    gas_10000 = 0x20
+
+    sender = pre.fund_eoa()
+    account = pre.deploy_contract(
+        Op.MSTORE(args_size, 0x20)
+        + Op.MSTORE(input_buf, 0xFF)
+        + Op.MSTORE(output_buf, 0xFF)
+        + Op.MSTORE8(input_buf, 0xFF)
+        + Op.MSTORE8(output_buf, 0xFF)
+        + Op.MSTORE(gas_test, Op.GAS())
+        # Setup stack to CALL into precompile with the CALLDATA and CALL into it (+ pop value)
+        + Op.CALL(
+            address=Op.CALLDATALOAD(0),
+            args_offset=args_offset,
+            args_size=args_size,
+            output_offset=output_offset,
+            output_size=output_size,
+        )
+        + Op.MSTORE(gas_test, (Op.SUB(Op.GAS(), Op.MLOAD(gas_test))))
+        + Op.MSTORE(gas_10000, Op.GAS())
+        + Op.CALL(
+            address=address_10000,
+            args_offset=args_offset,
+            args_size=args_size,
+            output_offset=output_offset,
+            output_size=output_size,
+        )
+        + Op.MSTORE(gas_10000, (Op.SUB(Op.GAS(), Op.MLOAD(gas_10000))))
+        + Op.JUMPI(0x7D, Op.GT(Op.MLOAD(gas_test), Op.MLOAD(gas_10000)))
+        + Op.JUMPDEST()
+        + Op.SSTORE(0, Op.SUB(Op.MLOAD(gas_test), Op.MLOAD(gas_10000)))
+        + Op.SSTORE(0, Op.LT(Op.SLOAD(gas_test), 0x10))
+        + Op.STOP(),
+    )
+
+    tx = Transaction(
+        ty=0x0,
+        to=account,
+        data=int(address, 16).to_bytes(32),
+        gas_limit=100000,
+        gas_price=10,
+        protected=True,
+        sender=sender,
+    )
+
+    # A high gas cost will result from calling a precompile
+    # Expect 0x00 when a precompile exists at the address, 0x01 otherwise
+    post = {account: Account(storage={0: "0x00" if precompile_exists else "0x01"})}
+
+    state_test(env=env, pre=pre, post=post, tx=tx)