feat(benchmark): add log opcode worst case

tests/zkevm/test_worst_opcode.py

@@ -0,0 +1,97 @@
+"""
+abstract: Tests zkEVMs worst-case opcode scenarios.
+    Tests zkEVMs worst-case opcode scenarios.
+
+Tests running worst-case opcodes scenarios for zkEVMs.
+"""
+
+import pytest
+
+from ethereum_test_forks import Fork
+from ethereum_test_tools import (
+    Alloc,
+    Bytecode,
+    Environment,
+    StateTestFiller,
+    Transaction,
+)
+from ethereum_test_tools.vm.opcode import Opcodes as Op
+from ethereum_test_vm.opcode import Opcode
+
+from .helpers import code_loop_precompile_call
+
+
+@pytest.mark.parametrize(
+    "opcode",
+    [
+        pytest.param(Op.LOG0, id="log0"),
+        pytest.param(Op.LOG1, id="log1"),
+        pytest.param(Op.LOG2, id="log2"),
+        pytest.param(Op.LOG3, id="log3"),
+        pytest.param(Op.LOG4, id="log4"),
+    ],
+)
+@pytest.mark.parametrize(
+    "size,non_zero_data",
+    [
+        pytest.param(0, False, id="0_bytes_data"),
+        pytest.param(1024 * 1024, False, id="1_MiB_zeros_data"),  # 1 MiB
+        pytest.param(1024 * 1024, True, id="1_MiB_non_zero_data"),  # 1 MiB
+    ],
+)
+@pytest.mark.parametrize(
+    "zeros_topic", [pytest.param(True, id="zeros_topic"), pytest.param(False, id="non_zero_topic")]
+)
+@pytest.mark.parametrize("fixed_offset", [True, False])
+def test_worst_log_opcodes(
+    state_test: StateTestFiller,
+    pre: Alloc,
+    fork: Fork,
+    opcode: Opcode,
+    zeros_topic: bool,
+    size: int,
+    fixed_offset: bool,
+    non_zero_data: bool,
+):
+    """Test running a block with as many LOG opcodes as possible."""
+    env = Environment()
+    max_code_size = fork.max_code_size()
+
+    calldata = Bytecode()
+
+    # For non-zero data, load  into memory.
+    if non_zero_data:
+        calldata += Op.CODECOPY(dest_offset=0, offset=0, size=Op.CODESIZE)
+
+    # Push the size value onto the stack and access it using the DUP opcode.
+    calldata += Op.PUSH3(size)
+
+    # For non-zeros topic, push a non-zero value for topic.
+    calldata += Op.PUSH0 if zeros_topic else Op.PUSH32(2**256 - 1)
+
+    topic_count = len(opcode.kwargs or []) - 2
+    offset = Op.PUSH0 if fixed_offset else Op.MOD(Op.GAS, 7)
+
+    # Calculate the appropriate DUP opcode based on topic count
+    # 0 topics -> DUP1, 1 topic -> DUP2, N topics -> DUP(N+1)
+    size_op = getattr(Op, f"DUP{topic_count + 2}")
+
+    code_sequence = Op.DUP1 * topic_count + size_op + offset + opcode
+
+    code = code_loop_precompile_call(calldata, code_sequence, fork)
+    assert len(code) <= max_code_size
+
+    code_address = pre.deploy_contract(code=code)
+
+    tx = Transaction(
+        to=code_address,
+        gas_limit=env.gas_limit,
+        sender=pre.fund_eoa(),
+    )
+
+    state_test(
+        env=env,
+        pre=pre,
+        post={},
+        tx=tx,
+    )