Added CREATE

pending-opcodes
e-ngo 5 years ago
parent 35dc2ef099
commit 5e1fb60670
  1. 39
      mythril/laser/ethereum/instructions.py
  2. 16
      tests/instructions/createopcode_test.py

@ -29,6 +29,8 @@ from mythril.laser.smt import (
) )
from mythril.laser.smt import symbol_factory from mythril.laser.smt import symbol_factory
from mythril.disassembler.disassembly import Disassembly
import mythril.laser.ethereum.util as helper import mythril.laser.ethereum.util as helper
from mythril.laser.ethereum import util from mythril.laser.ethereum import util
from mythril.laser.ethereum.call import get_call_parameters, native_call, get_call_data from mythril.laser.ethereum.call import get_call_parameters, native_call, get_call_data
@ -904,7 +906,8 @@ class Instruction:
state = global_state.mstate state = global_state.mstate
environment = global_state.environment environment = global_state.environment
disassembly = environment.code disassembly = environment.code
state.stack.append(len(disassembly.bytecode) // 2) no_of_bytes = len(disassembly.bytecode) // 2 + 5000
state.stack.append(no_of_bytes)
return [global_state] return [global_state]
@StateTransition(enable_gas=False) @StateTransition(enable_gas=False)
@ -1194,16 +1197,12 @@ class Instruction:
address = Extract(159, 0, stack.pop()) address = Extract(159, 0, stack.pop())
if address.symbolic: if address.symbolic:
log.debug("unsupported symbolic address for EXTCODEHASH") code_hash = symbol_factory.BitVecVal(int(get_code_hash(""), 16), 256)
stack.append(global_state.new_bitvec("extcodehash_" + str(address), 256)) elif address.value not in world_state.accounts:
return [global_state]
if address.value not in world_state.accounts:
code_hash = symbol_factory.BitVecVal(0, 256) code_hash = symbol_factory.BitVecVal(0, 256)
else: else:
code = world_state.accounts_exist_or_load( addr = "0" * (40 - len(hex(address.value))) + hex(address.value)
hex(address.value), self.dynamic_loader code = world_state.accounts_exist_or_load(addr, self.dynamic_loader)
)
code_hash = symbol_factory.BitVecVal(int(get_code_hash(code), 16), 256) code_hash = symbol_factory.BitVecVal(int(get_code_hash(code), 16), 256)
stack.append(code_hash) stack.append(code_hash)
return [global_state] return [global_state]
@ -1615,26 +1614,44 @@ class Instruction:
util.get_concrete_int(mem_offset), util.get_concrete_int(mem_offset),
util.get_concrete_int(mem_offset + mem_size), util.get_concrete_int(mem_offset + mem_size),
) )
except TypeError: except TypeError:
log.debug("Create with symbolic length or offset. Not supported") log.debug("Create with symbolic length or offset. Not supported")
mstate.stack.append(0, 256) mstate.stack.append(0, 256)
return [global_state] return [global_state]
caller = environment.sender call_data = call_data.concrete(None)
code_str = bytes.hex(bytes(call_data))
# for i in call_data.size:
# if isinstance(call_data[0], int):
# code_str += "00" if not call_data[0] else bytes.hex([call_data[i]])
#
code = Disassembly(code_str)
caller = environment.active_account.address
gas_price = environment.gasprice gas_price = environment.gasprice
origin = environment.origin origin = environment.origin
transaction = ContractCreationTransaction( transaction = ContractCreationTransaction(
world_state=world_state, world_state=world_state,
caller=caller, caller=caller,
call_data=call_data, code=code,
gas_price=gas_price, gas_price=gas_price,
gas_limit=mstate.gas_limit,
origin=origin, origin=origin,
call_value=call_value, call_value=call_value,
) )
raise TransactionStartSignal(transaction, self.op_code) raise TransactionStartSignal(transaction, self.op_code)
@StateTransition()
def create_post(self, global_state: GlobalState) -> List[GlobalState]:
global_state.mstate.stack.append(global_state.world_state.accounts[-1].address)
return [global_state]
@StateTransition() @StateTransition()
def create2_(self, global_state: GlobalState) -> List[GlobalState]: def create2_(self, global_state: GlobalState) -> List[GlobalState]:
""" """

@ -0,0 +1,16 @@
from mythril.disassembler.disassembly import Disassembly
from mythril.laser.ethereum.state.environment import Environment
from mythril.laser.ethereum.state.account import Account
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
from mythril.support.support_utils import get_code_hash
from mythril.laser.smt import symbol_factory
def test_create():
# TODO: ...
pass
Loading…
Cancel
Save