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.
45 lines
1.6 KiB
45 lines
1.6 KiB
from mythril.analysis.callgraph import generate_graph
|
|
from mythril.mythril import MythrilAnalyzer, MythrilDisassembler
|
|
from mythril.ethereum import util
|
|
from mythril.solidity.soliditycontract import EVMContract
|
|
from tests import (
|
|
BaseTestCase,
|
|
TESTDATA_INPUTS,
|
|
TESTDATA_OUTPUTS_EXPECTED,
|
|
TESTDATA_OUTPUTS_CURRENT,
|
|
)
|
|
import re
|
|
|
|
|
|
class GraphTest(BaseTestCase):
|
|
def test_generate_graph(self):
|
|
for input_file in TESTDATA_INPUTS.iterdir():
|
|
output_expected = TESTDATA_OUTPUTS_EXPECTED / (
|
|
input_file.name + ".graph.html"
|
|
)
|
|
output_current = TESTDATA_OUTPUTS_CURRENT / (
|
|
input_file.name + ".graph.html"
|
|
)
|
|
|
|
contract = EVMContract(input_file.read_text())
|
|
disassembler = MythrilDisassembler()
|
|
disassembler.contracts.append(contract)
|
|
analyzer = MythrilAnalyzer(
|
|
disassembler=disassembler,
|
|
strategy="dfs",
|
|
execution_timeout=5,
|
|
max_depth=30,
|
|
address=(util.get_indexed_address(0)),
|
|
)
|
|
|
|
html = analyzer.graph_html(transaction_count=1)
|
|
output_current.write_text(html)
|
|
|
|
lines_expected = re.findall(
|
|
r"'label': '.*'", str(output_current.read_text())
|
|
)
|
|
lines_found = re.findall(r"'label': '.*'", str(output_current.read_text()))
|
|
if not (lines_expected == lines_found):
|
|
self.found_changed_files(input_file, output_expected, output_current)
|
|
|
|
self.assert_and_show_changed_files()
|
|
|