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.
47 lines
1.8 KiB
47 lines
1.8 KiB
import unittest
|
|
from mythril.ethereum.ethcontract import ETHContract
|
|
|
|
|
|
class ETHContractTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.code = "0x60606040525b603c5b60006010603e565b9050593681016040523660008237602060003683856040603f5a0204f41560545760206000f35bfe5b50565b005b73c3b2ae46792547a96b9f84405e36d0e07edcd05c5b905600a165627a7a7230582062a884f947232ada573f95940cce9c8bfb7e4e14e21df5af4e884941afb55e590029"
|
|
self.creation_code = "0x60606040525b603c5b60006010603e565b9050593681016040523660008237602060003683856040603f5a0204f41560545760206000f35bfe5b50565b005b73c3b2ae46792547a96b9f84405e36d0e07edcd05c5b905600a165627a7a7230582062a884f947232ada573f95940cce9c8bfb7e4e14e21df5af4e884941afb55e590029"
|
|
|
|
|
|
class Getinstruction_listTestCase(ETHContractTestCase):
|
|
def runTest(self):
|
|
contract = ETHContract(self.code, self.creation_code)
|
|
|
|
disassembly = contract.disassembly
|
|
|
|
self.assertEqual(
|
|
len(disassembly.instruction_list),
|
|
53,
|
|
"Error disassembling code using ETHContract.get_instruction_list()",
|
|
)
|
|
|
|
|
|
class GetEASMTestCase(ETHContractTestCase):
|
|
def runTest(self):
|
|
contract = ETHContract(self.code)
|
|
|
|
instruction_list = contract.get_easm()
|
|
|
|
self.assertTrue(
|
|
"PUSH1 0x60" in instruction_list,
|
|
"Error obtaining EASM code through ETHContract.get_easm()",
|
|
)
|
|
|
|
|
|
class MatchesExpressionTestCase(ETHContractTestCase):
|
|
def runTest(self):
|
|
contract = ETHContract(self.code)
|
|
|
|
self.assertTrue(
|
|
contract.matches_expression("code#PUSH1# or code#PUSH1#"),
|
|
"Unexpected result in expression matching",
|
|
)
|
|
self.assertFalse(
|
|
contract.matches_expression("func#abcdef#"),
|
|
"Unexpected result in expression matching",
|
|
)
|
|
|