mirror of https://github.com/ConsenSys/mythril
parent
3d1b0e889d
commit
0869cbf394
@ -0,0 +1,57 @@ |
||||
from mythril.laser.smt import symbol_factory |
||||
from mythril.disassembler.disassembly import Disassembly |
||||
from mythril.laser.ethereum.state.environment import Environment |
||||
from mythril.laser.ethereum.state.machine_state import MachineState |
||||
from mythril.laser.ethereum.state.global_state import GlobalState |
||||
from mythril.laser.ethereum.state.world_state import WorldState |
||||
from mythril.laser.ethereum.instructions import Instruction |
||||
from mythril.laser.ethereum.transaction.transaction_models import MessageCallTransaction |
||||
|
||||
|
||||
def test_extcodecopy(): |
||||
# Arrange |
||||
new_world_state = WorldState() |
||||
new_account = new_world_state.create_account(balance=10, address=101) |
||||
new_account.code = Disassembly("60616240") |
||||
ext_account = new_world_state.create_account(balance=1000, address=121) |
||||
ext_account.code = Disassembly("6040404040") |
||||
|
||||
new_environment = Environment(new_account, None, None, None, None, None) |
||||
state = GlobalState( |
||||
new_world_state, new_environment, None, MachineState(gas_limit=8000000) |
||||
) |
||||
state.transaction_stack.append( |
||||
(MessageCallTransaction(world_state=WorldState(), gas_limit=8000000), None) |
||||
) |
||||
|
||||
state.mstate.stack = [3, 0, 0, 121] |
||||
instruction = Instruction("extcodecopy", dynamic_loader=None) |
||||
|
||||
# Act |
||||
new_state = instruction.evaluate(state)[0] |
||||
# Assert |
||||
assert new_state.mstate.memory[0:3] == [96, 64, 64] |
||||
|
||||
|
||||
def test_extcodecopy_fail(): |
||||
# Arrange |
||||
new_world_state = WorldState() |
||||
new_account = new_world_state.create_account(balance=10, address=101) |
||||
new_account.code = Disassembly("60616240") |
||||
new_environment = Environment(new_account, None, None, None, None, None) |
||||
state = GlobalState( |
||||
new_world_state, new_environment, None, MachineState(gas_limit=8000000) |
||||
) |
||||
state.transaction_stack.append( |
||||
(MessageCallTransaction(world_state=WorldState(), gas_limit=8000000), None) |
||||
) |
||||
|
||||
state.mstate.stack = [2, 2, 2, symbol_factory.BitVecSym("FAIL", 256)] |
||||
instruction = Instruction("extcodecopy", dynamic_loader=None) |
||||
|
||||
# Act |
||||
new_state = instruction.evaluate(state)[0] |
||||
|
||||
# Assert |
||||
assert new_state.mstate.stack == [] |
||||
assert new_state.mstate.memory._memory == state.mstate.memory._memory |
Loading…
Reference in new issue