mirror of https://github.com/crytic/slither
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
2.1 KiB
54 lines
2.1 KiB
from pathlib import Path
|
|
from collections import namedtuple
|
|
from slither import Slither
|
|
from slither.slithir.operations import Operation, NewContract
|
|
from solc_select import solc_select
|
|
|
|
TEST_DATA_DIR = Path(__file__).resolve().parent / "test_data"
|
|
|
|
|
|
def check_num_local_vars_read(function, slithir_op: Operation, num_reads_expected: int):
|
|
for node in function.nodes:
|
|
for operation in node.irs:
|
|
if isinstance(operation, slithir_op):
|
|
assert len(operation.read) == num_reads_expected
|
|
assert len(node.local_variables_read) == num_reads_expected
|
|
|
|
|
|
def check_num_states_vars_read(function, slithir_op: Operation, num_reads_expected: int):
|
|
for node in function.nodes:
|
|
for operation in node.irs:
|
|
if isinstance(operation, slithir_op):
|
|
assert len(operation.read) == num_reads_expected
|
|
assert len(node.state_variables_read) == num_reads_expected
|
|
|
|
|
|
OperationTest = namedtuple("OperationTest", "contract_name slithir_op")
|
|
|
|
OPERATION_TEST = [OperationTest("NewContract", NewContract)]
|
|
|
|
|
|
def test_operation_reads() -> None:
|
|
"""
|
|
Every slithir operation has its own contract and reads all local and state variables in readAllLocalVariables and readAllStateVariables, respectively.
|
|
"""
|
|
solc_select.switch_global_version("0.8.15", always_install=True)
|
|
slither = Slither(Path(TEST_DATA_DIR, "operation_reads.sol").as_posix())
|
|
|
|
for op_test in OPERATION_TEST:
|
|
print(op_test)
|
|
available = slither.get_contract_from_name(op_test.contract_name)
|
|
assert len(available) == 1
|
|
target = available[0]
|
|
|
|
num_state_variables = len(target.state_variables_ordered)
|
|
state_function = target.get_function_from_signature("readAllStateVariables()")
|
|
check_num_states_vars_read(state_function, op_test.slithir_op, num_state_variables)
|
|
|
|
local_function = target.get_function_from_signature("readAllLocalVariables()")
|
|
num_local_vars = len(local_function.local_variables)
|
|
check_num_local_vars_read(local_function, op_test.slithir_op, num_local_vars)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_operation_reads()
|
|
|