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
634 B
37 lines
634 B
7 years ago
|
from mythril.laser.ethereum.taint_analysis import *
|
||
|
|
|
||
|
|
|
||
|
|
def test_record_tainted_check():
|
||
|
|
# arrange
|
||
|
|
record = TaintRecord()
|
||
|
|
record.stack = [True, False, True]
|
||
|
|
|
||
|
|
# act
|
||
|
|
tainted = record.stack_tainted(2)
|
||
|
|
|
||
|
|
# assert
|
||
|
|
assert tainted is True
|
||
|
|
|
||
|
|
|
||
|
|
def test_record_untainted_check():
|
||
|
|
# arrange
|
||
|
|
record = TaintRecord()
|
||
|
|
record.stack = [True, False, False]
|
||
|
|
|
||
|
|
# act
|
||
|
|
tainted = record.stack_tainted(2)
|
||
|
|
|
||
|
|
# assert
|
||
|
|
assert tainted is False
|
||
|
|
|
||
|
|
|
||
|
|
def test_record_untouched_check():
|
||
|
|
# arrange
|
||
|
|
record = TaintRecord()
|
||
|
|
|
||
|
|
# act
|
||
|
|
tainted = record.stack_tainted(3)
|
||
|
|
|
||
|
|
# assert
|
||
|
|
assert tainted is None
|