Skip to content

Commit

Permalink
new(tests): EOF - EIP-3540 container size
Browse files Browse the repository at this point in the history
  • Loading branch information
pdobacz committed Jun 27, 2024
1 parent f514b68 commit cb472ef
Show file tree
Hide file tree
Showing 3 changed files with 121 additions and 0 deletions.
3 changes: 3 additions & 0 deletions src/ethereum_test_tools/exceptions/evmone_exceptions.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,6 +74,9 @@ class EvmoneExceptionMapper:
EOFException.TOPLEVEL_CONTAINER_TRUNCATED, "err: toplevel_container_truncated"
),
ExceptionMessage(EOFException.ORPHAN_SUBCONTAINER, "err: unreferenced_subcontainer"),
ExceptionMessage(
EOFException.CONTAINER_SIZE_ABOVE_LIMIT, "err: container_size_above_limit"
),
)

def __init__(self) -> None:
Expand Down
4 changes: 4 additions & 0 deletions src/ethereum_test_tools/exceptions/exceptions.py
Original file line number Diff line number Diff line change
Expand Up @@ -692,6 +692,10 @@ class EOFException(ExceptionBase):
"""
EOF container has an unreferenced subcontainer.
'"""
CONTAINER_SIZE_ABOVE_LIMIT = auto()
"""
EOF container is above size limit
"""


"""
Expand Down
114 changes: 114 additions & 0 deletions tests/prague/eip7692_eof_v1/eip3540_eof_v1/test_container_size.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,114 @@
"""
EOF validation tests for EIP-3540 container size
"""

import pytest

from ethereum_test_tools import EOFTestFiller
from ethereum_test_tools import Opcodes as Op
from ethereum_test_tools.eof.v1 import Container, EOFException, Section
from ethereum_test_tools.eof.v1.constants import MAX_INITCODE_SIZE

from .. import EOF_FORK_NAME

REFERENCE_SPEC_GIT_PATH = "EIPS/eip-3540.md"
REFERENCE_SPEC_VERSION = "" # todo

pytestmark = pytest.mark.valid_from(EOF_FORK_NAME)

VALID_CONTAINER = Container(sections=[Section.Code(code=Op.STOP)])


@pytest.mark.parametrize(
"over_limit",
[0, 1, 2, 2**16 - MAX_INITCODE_SIZE],
)
def test_max_size(
eof_test: EOFTestFiller,
over_limit: int,
):
"""
Verify EOF container valid at maximum size, invalid above
"""
minimal_code = bytearray(bytes(VALID_CONTAINER))
# Expand the minimal EOF code by more noop code, reaching the desired target container size.
code = Container(
sections=[
Section.Code(
code=Op.JUMPDEST * (MAX_INITCODE_SIZE - len(minimal_code) + over_limit) + Op.STOP
)
]
)
assert len(code) == MAX_INITCODE_SIZE + over_limit
eof_test(
data=bytes(code),
expect_exception=None if over_limit == 0 else EOFException.CONTAINER_SIZE_ABOVE_LIMIT,
)


@pytest.mark.parametrize(
"size",
[MAX_INITCODE_SIZE + 1, MAX_INITCODE_SIZE * 2],
)
def test_above_max_size_raw(
eof_test: EOFTestFiller,
size: int,
):
"""
Verify EOF container invalid above maximum size, regardless of header contents
"""
code = Op.INVALID * size
eof_test(
data=bytes(code),
expect_exception=EOFException.CONTAINER_SIZE_ABOVE_LIMIT,
)


@pytest.mark.parametrize(
"code",
[
pytest.param(
Container(sections=[Section.Code(code=Op.STOP, custom_size=MAX_INITCODE_SIZE)]),
id="1st_code_section",
),
pytest.param(
Container(
sections=[
Section.Code(code=Op.STOP),
Section.Code(code=Op.STOP, custom_size=MAX_INITCODE_SIZE),
]
),
id="2nd_code_section",
),
pytest.param(
Container(
sections=[
Section.Code(code=Op.STOP),
Section.Container(container=Op.STOP, custom_size=MAX_INITCODE_SIZE),
]
),
id="1st_container_section",
),
pytest.param(
Container(
sections=[
Section.Code(code=Op.STOP),
Section.Container(container=Op.STOP),
Section.Container(container=Op.STOP, custom_size=MAX_INITCODE_SIZE),
]
),
id="2nd_container_section",
),
],
)
def test_section_after_end_of_container(
eof_test: EOFTestFiller,
code: Container,
):
"""
Verify EOF container is invalid if any of sections declares above container size
"""
eof_test(
data=bytes(code),
expect_exception=EOFException.INVALID_SECTION_BODIES_SIZE,
)

0 comments on commit cb472ef

Please sign in to comment.