test_sstore_erc20_generic()

Documentation for tests/benchmark/stateful/bloatnet/test_erc20.py::test_sstore_erc20_generic@5fa5938b.

Generate fixtures for these test cases for Amsterdam with:

fill -v tests/benchmark/stateful/bloatnet/test_erc20.py::test_sstore_erc20_generic --gas-benchmark-values 1

Benchmark SSTORE using ERC20 approve.

Source code in tests/benchmark/stateful/bloatnet/test_erc20.py

@pytest.mark.stub_parametrize("erc20_stub", "test_sstore_erc20_approve_")
def test_sstore_erc20_generic(
    benchmark_test: BenchmarkTestFiller,
    pre: Alloc,
    fork: Fork,
    gas_benchmark_value: int,
    tx_gas_limit: int,
    erc20_stub: str,
) -> None:
    """Benchmark SSTORE using ERC20 approve."""
    sender = pre.fund_eoa()

    threshold = 100_000

    # Stub Account
    erc20_address = pre.deploy_contract(
        code=Bytecode(),
        stub=erc20_stub,
    )

    # MEM[0] = function selector
    # MEM[32] = starting address offset
    setup = Op.MSTORE(
        0,
        APPROVE_SELECTOR,
    ) + Op.MSTORE(
        32,
        Op.SLOAD(0),  # Address Offset
    )

    call_approve = Op.MSTORE(
        64,
        Op.ADD(1, Op.MLOAD(32)),
    ) + Op.POP(
        Op.CALL(
            address=erc20_address,
            args_offset=28,
            args_size=68,
        )
    )

    loop = While(
        body=call_approve + Op.MSTORE(32, Op.ADD(Op.MLOAD(32), 1)),
        condition=Op.GT(Op.GAS, threshold),
    )

    teardown = Op.SSTORE(0, Op.MLOAD(32))

    # Contract Deployment
    code = setup + loop + teardown
    attack_contract_address = pre.deploy_contract(code=code)

    intrinsic_gas = fork.transaction_intrinsic_cost_calculator()()

    # Transaction Loops
    gas_remaining = gas_benchmark_value

    # Collect tx params first, then build Transaction objects
    # so that nonces are allocated contiguously per block.
    tx_gas: list[int] = []
    while gas_remaining > intrinsic_gas:
        gas_available = min(gas_remaining, tx_gas_limit)

        if gas_available < intrinsic_gas:
            break

        tx_gas.append(gas_available)

        gas_remaining -= gas_available

    txs = []
    with TestPhaseManager.execution():
        for gas_available in tx_gas:
            txs.append(
                Transaction(
                    gas_limit=gas_available,
                    to=attack_contract_address,
                    sender=sender,
                )
            )

    blocks = [Block(txs=txs)]

    benchmark_test(
        pre=pre,
        blocks=blocks,
        skip_gas_used_validation=True,
        expected_receipt_status=True,
    )

Parametrized Test Cases

This test generates 1 parametrized test case across 3 forks.