mirror of https://github.com/ConsenSys/mythril
blockchainethereumsmart-contractssoliditysecurityprogram-analysissecurity-analysissymbolic-execution
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.
72 lines
2.6 KiB
72 lines
2.6 KiB
1 year ago
|
import pytest
|
||
|
|
from pathlib import Path
|
||
|
|
from mythril.mythril import MythrilDisassembler
|
||
|
|
from mythril.solidity.features import SolidityFeatureExtractor
|
||
|
|
from mythril.solidity.soliditycontract import SolidityContract, SolcAST
|
||
|
|
|
||
|
|
TEST_FILES = Path(__file__).parent / "testdata/input_contracts"
|
||
|
|
solc_binary = MythrilDisassembler._init_solc_binary("v0.5.0")
|
||
|
|
|
||
|
|
test_cases = [
|
||
|
|
("suicide.sol", 1, "kill", "contains_selfdestruct", True),
|
||
|
|
(
|
||
|
|
"SimpleModifier.sol",
|
||
|
|
1,
|
||
|
|
"withdrawfunds",
|
||
|
|
"all_require_vars",
|
||
|
|
set(["msg", "owner"]),
|
||
|
|
),
|
||
|
|
("SimpleModifier.sol", 1, "withdrawfunds", "transfer_vars", set(["owner"])),
|
||
|
|
("rubixi.sol", 18, "init", "contains_selfdestruct", False),
|
||
|
|
("rubixi.sol", 18, "collectAllFees", "has_owner_modifier", True),
|
||
|
|
("rubixi.sol", 18, "collectAllFees", "is_payable", False),
|
||
|
|
(
|
||
|
|
"rubixi.sol",
|
||
|
|
18,
|
||
|
|
"collectAllFees",
|
||
|
|
"all_require_vars",
|
||
|
|
set(["collectedFees", "creator"]),
|
||
|
|
),
|
||
|
|
(
|
||
|
|
"rubixi.sol",
|
||
|
|
18,
|
||
|
|
"collectPercentOfFees",
|
||
|
|
"all_require_vars",
|
||
|
|
set(["collectedFees", "_pcent", "creator"]),
|
||
|
|
),
|
||
|
|
(
|
||
|
|
"rubixi.sol",
|
||
|
|
18,
|
||
|
|
"changeMultiplier",
|
||
|
|
"all_require_vars",
|
||
|
|
set(["_mult", "creator"]),
|
||
|
|
),
|
||
|
|
("rubixi.sol", 18, "", "is_payable", True),
|
||
|
|
("rubixi.sol", 18, "collectAllFees", "transfer_vars", set(["creator"])),
|
||
|
|
("exceptions.sol", 8, "assert3", "contains_assert", True),
|
||
|
|
("exceptions.sol", 8, "requireisfine", "all_require_vars", set(["input"])),
|
||
|
|
("WalletLibrary.sol", 23, "execute", "has_owner_modifier", True),
|
||
|
|
("WalletLibrary.sol", 23, "initWallet", "has_owner_modifier", False),
|
||
|
|
("WalletLibrary.sol", 23, "initWallet", "all_require_vars", set(["m_numOwners"])),
|
||
|
|
("WalletLibrary.sol", 23, "confirm", "all_require_vars", set(["success"])),
|
||
|
|
("kcalls.sol", 3, "callSetN", "contains_call", True),
|
||
|
|
("kcalls.sol", 3, "delegatecallSetN", "contains_delegatecall", True),
|
||
|
|
("kcalls.sol", 3, "callcodeSetN", "contains_staticcall", True),
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"file_name, num_funcs, func_name, field, expected_value", test_cases
|
||
|
|
)
|
||
|
|
def test_feature_selfdestruct(file_name, num_funcs, func_name, field, expected_value):
|
||
|
|
input_file = TEST_FILES / file_name
|
||
|
|
name = file_name.split(".")[0]
|
||
|
|
if name[0].islower():
|
||
|
|
name = name.capitalize()
|
||
|
|
contract = SolidityContract(str(input_file), name=name, solc_binary=solc_binary)
|
||
|
|
ms = contract.solc_json["sources"][str(input_file)]["ast"]
|
||
|
|
sfe = SolidityFeatureExtractor(ms)
|
||
|
|
fe = sfe.extract_features()
|
||
|
|
assert len(fe) == num_funcs
|
||
|
|
assert fe[func_name][field] == expected_value
|