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.
141 lines
5.3 KiB
141 lines
5.3 KiB
import inspect
|
|
|
|
from crytic_compile import CryticCompile
|
|
from crytic_compile.platform.solc_standard_json import SolcStandardJson
|
|
from solc_select import solc_select
|
|
|
|
from slither import Slither
|
|
from slither.detectors import all_detectors
|
|
from slither.detectors.abstract_detector import AbstractDetector
|
|
from slither.slithir.operations import LibraryCall, InternalCall
|
|
|
|
|
|
def _run_all_detectors(slither: Slither) -> None:
|
|
detectors = [getattr(all_detectors, name) for name in dir(all_detectors)]
|
|
detectors = [d for d in detectors if inspect.isclass(d) and issubclass(d, AbstractDetector)]
|
|
|
|
for detector in detectors:
|
|
slither.register_detector(detector)
|
|
|
|
slither.run_detectors()
|
|
|
|
|
|
def test_node() -> None:
|
|
# hardhat must have been installed in tests/test_node_modules
|
|
# For the CI its done through the github action config
|
|
|
|
slither = Slither("./tests/test_node_modules")
|
|
_run_all_detectors(slither)
|
|
|
|
|
|
def test_collision() -> None:
|
|
|
|
standard_json = SolcStandardJson()
|
|
standard_json.add_source_file("./tests/collisions/a.sol")
|
|
standard_json.add_source_file("./tests/collisions/b.sol")
|
|
|
|
compilation = CryticCompile(standard_json)
|
|
slither = Slither(compilation)
|
|
|
|
_run_all_detectors(slither)
|
|
|
|
|
|
def test_cycle() -> None:
|
|
slither = Slither("./tests/test_cyclic_import/a.sol")
|
|
_run_all_detectors(slither)
|
|
|
|
|
|
def test_funcion_id_rec_structure() -> None:
|
|
solc_select.switch_global_version("0.8.0", always_install=True)
|
|
slither = Slither("./tests/function_ids/rec_struct-0.8.sol")
|
|
for compilation_unit in slither.compilation_units:
|
|
for function in compilation_unit.functions:
|
|
assert function.solidity_signature
|
|
|
|
|
|
def test_upgradeable_comments() -> None:
|
|
solc_select.switch_global_version("0.8.10", always_install=True)
|
|
slither = Slither("./tests/custom_comments/upgrade.sol")
|
|
compilation_unit = slither.compilation_units[0]
|
|
proxy = compilation_unit.get_contract_from_name("Proxy")[0]
|
|
|
|
assert proxy.is_upgradeable_proxy
|
|
|
|
v0 = compilation_unit.get_contract_from_name("V0")[0]
|
|
|
|
assert v0.is_upgradeable
|
|
print(v0.upgradeable_version)
|
|
assert v0.upgradeable_version == "version-0"
|
|
|
|
v1 = compilation_unit.get_contract_from_name("V1")[0]
|
|
assert v0.is_upgradeable
|
|
assert v1.upgradeable_version == "version_1"
|
|
|
|
|
|
def test_using_for_top_level_same_name() -> None:
|
|
solc_select.switch_global_version("0.8.15", always_install=True)
|
|
slither = Slither("./tests/ast-parsing/using-for-3-0.8.0.sol")
|
|
contract_c = slither.get_contract_from_name("C")[0]
|
|
libCall = contract_c.get_function_from_full_name("libCall(uint256)")
|
|
for ir in libCall.all_slithir_operations():
|
|
if isinstance(ir, LibraryCall) and ir.destination == "Lib" and ir.function_name == "a":
|
|
return
|
|
assert False
|
|
|
|
|
|
def test_using_for_top_level_implicit_conversion() -> None:
|
|
solc_select.switch_global_version("0.8.15", always_install=True)
|
|
slither = Slither("./tests/ast-parsing/using-for-4-0.8.0.sol")
|
|
contract_c = slither.get_contract_from_name("C")[0]
|
|
libCall = contract_c.get_function_from_full_name("libCall(uint16)")
|
|
for ir in libCall.all_slithir_operations():
|
|
if isinstance(ir, LibraryCall) and ir.destination == "Lib" and ir.function_name == "f":
|
|
return
|
|
assert False
|
|
|
|
|
|
def test_using_for_alias_top_level() -> None:
|
|
solc_select.switch_global_version("0.8.15", always_install=True)
|
|
slither = Slither("./tests/ast-parsing/using-for-alias-top-level-0.8.0.sol")
|
|
contract_c = slither.get_contract_from_name("C")[0]
|
|
libCall = contract_c.get_function_from_full_name("libCall(uint256)")
|
|
ok = False
|
|
for ir in libCall.all_slithir_operations():
|
|
if isinstance(ir, LibraryCall) and ir.destination == "Lib" and ir.function_name == "b":
|
|
ok = True
|
|
if not ok:
|
|
assert False
|
|
topLevelCall = contract_c.get_function_from_full_name("topLevel(uint256)")
|
|
for ir in topLevelCall.all_slithir_operations():
|
|
if isinstance(ir, InternalCall) and ir.function_name == "a":
|
|
return
|
|
assert False
|
|
|
|
|
|
def test_using_for_alias_contract() -> None:
|
|
solc_select.switch_global_version("0.8.15", always_install=True)
|
|
slither = Slither("./tests/ast-parsing/using-for-alias-contract-0.8.0.sol")
|
|
contract_c = slither.get_contract_from_name("C")[0]
|
|
libCall = contract_c.get_function_from_full_name("libCall(uint256)")
|
|
ok = False
|
|
for ir in libCall.all_slithir_operations():
|
|
if isinstance(ir, LibraryCall) and ir.destination == "Lib" and ir.function_name == "b":
|
|
ok = True
|
|
if not ok:
|
|
assert False
|
|
topLevelCall = contract_c.get_function_from_full_name("topLevel(uint256)")
|
|
for ir in topLevelCall.all_slithir_operations():
|
|
if isinstance(ir, InternalCall) and ir.function_name == "a":
|
|
return
|
|
assert False
|
|
|
|
|
|
def test_using_for_in_library() -> None:
|
|
solc_select.switch_global_version("0.8.15", always_install=True)
|
|
slither = Slither("./tests/ast-parsing/using-for-in-library-0.8.0.sol")
|
|
contract_c = slither.get_contract_from_name("A")[0]
|
|
libCall = contract_c.get_function_from_full_name("a(uint256)")
|
|
for ir in libCall.all_slithir_operations():
|
|
if isinstance(ir, LibraryCall) and ir.destination == "B" and ir.function_name == "b":
|
|
return
|
|
assert False
|
|
|