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.
37 lines
862 B
37 lines
862 B
7 years ago
|
from mythril.laser.ethereum.taint_analysis import *
|
||
|
|
from mythril.laser.ethereum.svm import GlobalState
|
||
|
|
|
||
|
|
|
||
|
|
def test_result_state():
|
||
|
|
# arrange
|
||
|
|
taint_result = TaintResult()
|
||
|
|
record = TaintRecord()
|
||
|
|
state = GlobalState(2, None)
|
||
|
|
state.mstate.stack = [1,2,3]
|
||
|
|
record.add_state(state)
|
||
|
|
record.stack = [False, False, False]
|
||
|
|
# act
|
||
|
|
taint_result.add_records([record])
|
||
|
|
tainted = taint_result.check(state, 2)
|
||
|
|
|
||
|
|
# assert
|
||
|
|
assert tainted is False
|
||
|
|
assert record in taint_result.records
|
||
|
|
|
||
|
|
|
||
|
|
def test_result_no_state():
|
||
|
|
# arrange
|
||
|
|
taint_result = TaintResult()
|
||
|
|
record = TaintRecord()
|
||
|
|
state = GlobalState(2, None)
|
||
|
|
state.mstate.stack = [1,2,3]
|
||
|
|
|
||
|
|
|
||
|
|
# act
|
||
|
|
taint_result.add_records([record])
|
||
|
|
tainted = taint_result.check(state, 2)
|
||
|
|
|
||
|
|
# assert
|
||
|
|
assert tainted is None
|
||
|
|
assert record in taint_result.records
|