|
|
@ -38,6 +38,7 @@ from mythril.laser.ethereum.evm_exceptions import ( |
|
|
|
InvalidJumpDestination, |
|
|
|
InvalidJumpDestination, |
|
|
|
InvalidInstruction, |
|
|
|
InvalidInstruction, |
|
|
|
OutOfGasException, |
|
|
|
OutOfGasException, |
|
|
|
|
|
|
|
WriteProtection, |
|
|
|
) |
|
|
|
) |
|
|
|
from mythril.laser.ethereum.gas import OPCODE_GAS |
|
|
|
from mythril.laser.ethereum.gas import OPCODE_GAS |
|
|
|
from mythril.laser.ethereum.state.global_state import GlobalState |
|
|
|
from mythril.laser.ethereum.state.global_state import GlobalState |
|
|
@ -88,14 +89,18 @@ class StateTransition(object): |
|
|
|
if `increment_pc=True`. |
|
|
|
if `increment_pc=True`. |
|
|
|
""" |
|
|
|
""" |
|
|
|
|
|
|
|
|
|
|
|
def __init__(self, increment_pc=True, enable_gas=True): |
|
|
|
def __init__( |
|
|
|
|
|
|
|
self, increment_pc=True, enable_gas=True, is_state_mutation_instruction=False |
|
|
|
|
|
|
|
): |
|
|
|
""" |
|
|
|
""" |
|
|
|
|
|
|
|
|
|
|
|
:param increment_pc: |
|
|
|
:param increment_pc: |
|
|
|
:param enable_gas: |
|
|
|
:param enable_gas: |
|
|
|
|
|
|
|
:param is_state_mutation_instruction: The function mutates state |
|
|
|
""" |
|
|
|
""" |
|
|
|
self.increment_pc = increment_pc |
|
|
|
self.increment_pc = increment_pc |
|
|
|
self.enable_gas = enable_gas |
|
|
|
self.enable_gas = enable_gas |
|
|
|
|
|
|
|
self.is_state_mutation_instruction = is_state_mutation_instruction |
|
|
|
|
|
|
|
|
|
|
|
@staticmethod |
|
|
|
@staticmethod |
|
|
|
def call_on_state_copy(func: Callable, func_obj: "Instruction", state: GlobalState): |
|
|
|
def call_on_state_copy(func: Callable, func_obj: "Instruction", state: GlobalState): |
|
|
@ -165,6 +170,13 @@ class StateTransition(object): |
|
|
|
:param global_state: |
|
|
|
:param global_state: |
|
|
|
:return: |
|
|
|
:return: |
|
|
|
""" |
|
|
|
""" |
|
|
|
|
|
|
|
if self.is_state_mutation_instruction and global_state.environment.static: |
|
|
|
|
|
|
|
raise WriteProtection( |
|
|
|
|
|
|
|
"The function {} cannot be executed in a static call".format( |
|
|
|
|
|
|
|
func.__name__[:-1] |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
new_global_states = self.call_on_state_copy(func, func_obj, global_state) |
|
|
|
new_global_states = self.call_on_state_copy(func, func_obj, global_state) |
|
|
|
new_global_states = [ |
|
|
|
new_global_states = [ |
|
|
|
self.accumulate_gas(state) for state in new_global_states |
|
|
|
self.accumulate_gas(state) for state in new_global_states |
|
|
@ -758,34 +770,33 @@ class Instruction: |
|
|
|
""" |
|
|
|
""" |
|
|
|
state = global_state.mstate |
|
|
|
state = global_state.mstate |
|
|
|
environment = global_state.environment |
|
|
|
environment = global_state.environment |
|
|
|
state.stack.append(environment.calldata.calldatasize) |
|
|
|
|
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@StateTransition() |
|
|
|
if isinstance(global_state.current_transaction, ContractCreationTransaction): |
|
|
|
def calldatacopy_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
log.debug("Attempt to use CALLDATASIZE in creation transaction") |
|
|
|
""" |
|
|
|
state.stack.append(0) |
|
|
|
|
|
|
|
else: |
|
|
|
|
|
|
|
state.stack.append(environment.calldata.calldatasize) |
|
|
|
|
|
|
|
|
|
|
|
:param global_state: |
|
|
|
return [global_state] |
|
|
|
:return: |
|
|
|
|
|
|
|
""" |
|
|
|
@staticmethod |
|
|
|
state = global_state.mstate |
|
|
|
def _calldata_copy_helper(global_state, state, mstart, dstart, size): |
|
|
|
environment = global_state.environment |
|
|
|
environment = global_state.environment |
|
|
|
op0, op1, op2 = state.stack.pop(), state.stack.pop(), state.stack.pop() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
try: |
|
|
|
try: |
|
|
|
mstart = util.get_concrete_int(op0) |
|
|
|
mstart = util.get_concrete_int(mstart) |
|
|
|
except TypeError: |
|
|
|
except TypeError: |
|
|
|
log.debug("Unsupported symbolic memory offset in CALLDATACOPY") |
|
|
|
log.debug("Unsupported symbolic memory offset in CALLDATACOPY") |
|
|
|
return [global_state] |
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
try: |
|
|
|
try: |
|
|
|
dstart = util.get_concrete_int(op1) # type: Union[int, BitVec] |
|
|
|
dstart = util.get_concrete_int(dstart) # type: Union[int, BitVec] |
|
|
|
except TypeError: |
|
|
|
except TypeError: |
|
|
|
log.debug("Unsupported symbolic calldata offset in CALLDATACOPY") |
|
|
|
log.debug("Unsupported symbolic calldata offset in CALLDATACOPY") |
|
|
|
dstart = simplify(op1) |
|
|
|
dstart = simplify(dstart) |
|
|
|
|
|
|
|
|
|
|
|
try: |
|
|
|
try: |
|
|
|
size = util.get_concrete_int(op2) # type: Union[int, BitVec] |
|
|
|
size = util.get_concrete_int(size) # type: Union[int, BitVec] |
|
|
|
except TypeError: |
|
|
|
except TypeError: |
|
|
|
log.debug("Unsupported symbolic size in CALLDATACOPY") |
|
|
|
log.debug("Unsupported symbolic size in CALLDATACOPY") |
|
|
|
size = 320 # The excess size will get overwritten |
|
|
|
size = 320 # The excess size will get overwritten |
|
|
@ -839,6 +850,22 @@ class Instruction: |
|
|
|
) |
|
|
|
) |
|
|
|
return [global_state] |
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@StateTransition() |
|
|
|
|
|
|
|
def calldatacopy_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
|
|
|
|
""" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
:param global_state: |
|
|
|
|
|
|
|
:return: |
|
|
|
|
|
|
|
""" |
|
|
|
|
|
|
|
state = global_state.mstate |
|
|
|
|
|
|
|
op0, op1, op2 = state.stack.pop(), state.stack.pop(), state.stack.pop() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if isinstance(global_state.current_transaction, ContractCreationTransaction): |
|
|
|
|
|
|
|
log.debug("Attempt to use CALLDATACOPY in creation transaction") |
|
|
|
|
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return self._calldata_copy_helper(global_state, state, op0, op1, op2) |
|
|
|
|
|
|
|
|
|
|
|
# Environment |
|
|
|
# Environment |
|
|
|
@StateTransition() |
|
|
|
@StateTransition() |
|
|
|
def address_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
def address_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
@ -902,7 +929,17 @@ 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) |
|
|
|
if isinstance(global_state.current_transaction, ContractCreationTransaction): |
|
|
|
|
|
|
|
# Hacky way to ensure constructor arguments work - Pick some reasonably large size. |
|
|
|
|
|
|
|
no_of_bytes = ( |
|
|
|
|
|
|
|
len(disassembly.bytecode) // 2 + 0x200 |
|
|
|
|
|
|
|
) # space for 16 32-byte arguments |
|
|
|
|
|
|
|
global_state.mstate.constraints.append( |
|
|
|
|
|
|
|
global_state.environment.calldata.size == no_of_bytes |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
else: |
|
|
|
|
|
|
|
no_of_bytes = len(disassembly.bytecode) // 2 |
|
|
|
|
|
|
|
state.stack.append(no_of_bytes) |
|
|
|
return [global_state] |
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
@StateTransition(enable_gas=False) |
|
|
|
@StateTransition(enable_gas=False) |
|
|
@ -1026,14 +1063,28 @@ class Instruction: |
|
|
|
global_state.mstate.stack.pop(), |
|
|
|
global_state.mstate.stack.pop(), |
|
|
|
global_state.mstate.stack.pop(), |
|
|
|
global_state.mstate.stack.pop(), |
|
|
|
) |
|
|
|
) |
|
|
|
return self._code_copy_helper( |
|
|
|
|
|
|
|
code=global_state.environment.code.bytecode, |
|
|
|
if ( |
|
|
|
memory_offset=memory_offset, |
|
|
|
isinstance(global_state.current_transaction, ContractCreationTransaction) |
|
|
|
code_offset=code_offset, |
|
|
|
and code_offset >= len(global_state.environment.code.bytecode) // 2 |
|
|
|
size=size, |
|
|
|
): |
|
|
|
op="CODECOPY", |
|
|
|
# Treat creation code after the expected disassembly as calldata. |
|
|
|
global_state=global_state, |
|
|
|
# This is a slightly hacky way to ensure that symbolic constructor |
|
|
|
) |
|
|
|
# arguments work correctly. |
|
|
|
|
|
|
|
offset = code_offset - len(global_state.environment.code.bytecode) // 2 |
|
|
|
|
|
|
|
log.warning("Doing hacky thing offset: {} size: {}".format(offset, size)) |
|
|
|
|
|
|
|
return self._calldata_copy_helper( |
|
|
|
|
|
|
|
global_state, global_state.mstate, memory_offset, offset, size |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
else: |
|
|
|
|
|
|
|
return self._code_copy_helper( |
|
|
|
|
|
|
|
code=global_state.environment.code.bytecode, |
|
|
|
|
|
|
|
memory_offset=memory_offset, |
|
|
|
|
|
|
|
code_offset=code_offset, |
|
|
|
|
|
|
|
size=size, |
|
|
|
|
|
|
|
op="CODECOPY", |
|
|
|
|
|
|
|
global_state=global_state, |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
@StateTransition() |
|
|
|
@StateTransition() |
|
|
|
def extcodesize_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
def extcodesize_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
@ -1396,7 +1447,7 @@ class Instruction: |
|
|
|
state.stack.append(global_state.environment.active_account.storage[index]) |
|
|
|
state.stack.append(global_state.environment.active_account.storage[index]) |
|
|
|
return [global_state] |
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
@StateTransition() |
|
|
|
@StateTransition(is_state_mutation_instruction=True) |
|
|
|
def sstore_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
def sstore_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
""" |
|
|
|
""" |
|
|
|
|
|
|
|
|
|
|
@ -1563,7 +1614,7 @@ class Instruction: |
|
|
|
global_state.mstate.stack.append(global_state.new_bitvec("gas", 256)) |
|
|
|
global_state.mstate.stack.append(global_state.new_bitvec("gas", 256)) |
|
|
|
return [global_state] |
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
@StateTransition() |
|
|
|
@StateTransition(is_state_mutation_instruction=True) |
|
|
|
def log_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
def log_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
""" |
|
|
|
""" |
|
|
|
|
|
|
|
|
|
|
@ -1578,7 +1629,7 @@ class Instruction: |
|
|
|
# Not supported |
|
|
|
# Not supported |
|
|
|
return [global_state] |
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
@StateTransition() |
|
|
|
@StateTransition(is_state_mutation_instruction=True) |
|
|
|
def create_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
def create_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
""" |
|
|
|
""" |
|
|
|
|
|
|
|
|
|
|
@ -1586,13 +1637,14 @@ class Instruction: |
|
|
|
:return: |
|
|
|
:return: |
|
|
|
""" |
|
|
|
""" |
|
|
|
# TODO: implement me |
|
|
|
# TODO: implement me |
|
|
|
|
|
|
|
|
|
|
|
state = global_state.mstate |
|
|
|
state = global_state.mstate |
|
|
|
state.stack.pop(), state.stack.pop(), state.stack.pop() |
|
|
|
state.stack.pop(), state.stack.pop(), state.stack.pop() |
|
|
|
# Not supported |
|
|
|
# Not supported |
|
|
|
state.stack.append(0) |
|
|
|
state.stack.append(0) |
|
|
|
return [global_state] |
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
@StateTransition() |
|
|
|
@StateTransition(is_state_mutation_instruction=True) |
|
|
|
def create2_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
def create2_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
""" |
|
|
|
""" |
|
|
|
|
|
|
|
|
|
|
@ -1628,7 +1680,7 @@ class Instruction: |
|
|
|
return_data = state.memory[offset : offset + length] |
|
|
|
return_data = state.memory[offset : offset + length] |
|
|
|
global_state.current_transaction.end(global_state, return_data) |
|
|
|
global_state.current_transaction.end(global_state, return_data) |
|
|
|
|
|
|
|
|
|
|
|
@StateTransition() |
|
|
|
@StateTransition(is_state_mutation_instruction=True) |
|
|
|
def suicide_(self, global_state: GlobalState): |
|
|
|
def suicide_(self, global_state: GlobalState): |
|
|
|
""" |
|
|
|
""" |
|
|
|
|
|
|
|
|
|
|
@ -1733,6 +1785,21 @@ class Instruction: |
|
|
|
) |
|
|
|
) |
|
|
|
return [global_state] |
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if environment.static: |
|
|
|
|
|
|
|
if isinstance(value, int) and value > 0: |
|
|
|
|
|
|
|
raise WriteProtection( |
|
|
|
|
|
|
|
"Cannot call with non zero value in a static call" |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
if isinstance(value, BitVec): |
|
|
|
|
|
|
|
if value.symbolic: |
|
|
|
|
|
|
|
global_state.mstate.constraints.append( |
|
|
|
|
|
|
|
value == symbol_factory.BitVecVal(0, 256) |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
elif value.value > 0: |
|
|
|
|
|
|
|
raise WriteProtection( |
|
|
|
|
|
|
|
"Cannot call with non zero value in a static call" |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
native_result = native_call( |
|
|
|
native_result = native_call( |
|
|
|
global_state, callee_address, call_data, memory_out_offset, memory_out_size |
|
|
|
global_state, callee_address, call_data, memory_out_offset, memory_out_size |
|
|
|
) |
|
|
|
) |
|
|
@ -1747,8 +1814,9 @@ class Instruction: |
|
|
|
callee_account=callee_account, |
|
|
|
callee_account=callee_account, |
|
|
|
call_data=call_data, |
|
|
|
call_data=call_data, |
|
|
|
call_value=value, |
|
|
|
call_value=value, |
|
|
|
|
|
|
|
static=environment.static, |
|
|
|
) |
|
|
|
) |
|
|
|
raise TransactionStartSignal(transaction, self.op_code) |
|
|
|
raise TransactionStartSignal(transaction, self.op_code, global_state) |
|
|
|
|
|
|
|
|
|
|
|
@StateTransition() |
|
|
|
@StateTransition() |
|
|
|
def call_post(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
def call_post(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
@ -1757,65 +1825,8 @@ class Instruction: |
|
|
|
:param global_state: |
|
|
|
:param global_state: |
|
|
|
:return: |
|
|
|
:return: |
|
|
|
""" |
|
|
|
""" |
|
|
|
instr = global_state.get_current_instruction() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
try: |
|
|
|
return self.post_handler(global_state, function_name="call") |
|
|
|
callee_address, callee_account, call_data, value, gas, memory_out_offset, memory_out_size = get_call_parameters( |
|
|
|
|
|
|
|
global_state, self.dynamic_loader, True |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
except ValueError as e: |
|
|
|
|
|
|
|
log.debug( |
|
|
|
|
|
|
|
"Could not determine required parameters for call, putting fresh symbol on the stack. \n{}".format( |
|
|
|
|
|
|
|
e |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
global_state.mstate.stack.append( |
|
|
|
|
|
|
|
global_state.new_bitvec("retval_" + str(instr["address"]), 256) |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if global_state.last_return_data is None: |
|
|
|
|
|
|
|
# Put return value on stack |
|
|
|
|
|
|
|
return_value = global_state.new_bitvec( |
|
|
|
|
|
|
|
"retval_" + str(instr["address"]), 256 |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
global_state.mstate.stack.append(return_value) |
|
|
|
|
|
|
|
global_state.mstate.constraints.append(return_value == 0) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
try: |
|
|
|
|
|
|
|
memory_out_offset = ( |
|
|
|
|
|
|
|
util.get_concrete_int(memory_out_offset) |
|
|
|
|
|
|
|
if isinstance(memory_out_offset, Expression) |
|
|
|
|
|
|
|
else memory_out_offset |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
memory_out_size = ( |
|
|
|
|
|
|
|
util.get_concrete_int(memory_out_size) |
|
|
|
|
|
|
|
if isinstance(memory_out_size, Expression) |
|
|
|
|
|
|
|
else memory_out_size |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
except TypeError: |
|
|
|
|
|
|
|
global_state.mstate.stack.append( |
|
|
|
|
|
|
|
global_state.new_bitvec("retval_" + str(instr["address"]), 256) |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Copy memory |
|
|
|
|
|
|
|
global_state.mstate.mem_extend( |
|
|
|
|
|
|
|
memory_out_offset, min(memory_out_size, len(global_state.last_return_data)) |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
for i in range(min(memory_out_size, len(global_state.last_return_data))): |
|
|
|
|
|
|
|
global_state.mstate.memory[ |
|
|
|
|
|
|
|
i + memory_out_offset |
|
|
|
|
|
|
|
] = global_state.last_return_data[i] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Put return value on stack |
|
|
|
|
|
|
|
return_value = global_state.new_bitvec("retval_" + str(instr["address"]), 256) |
|
|
|
|
|
|
|
global_state.mstate.stack.append(return_value) |
|
|
|
|
|
|
|
global_state.mstate.constraints.append(return_value == 1) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@StateTransition() |
|
|
|
@StateTransition() |
|
|
|
def callcode_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
def callcode_(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
@ -1831,7 +1842,6 @@ class Instruction: |
|
|
|
callee_address, callee_account, call_data, value, gas, _, _ = get_call_parameters( |
|
|
|
callee_address, callee_account, call_data, value, gas, _, _ = get_call_parameters( |
|
|
|
global_state, self.dynamic_loader, True |
|
|
|
global_state, self.dynamic_loader, True |
|
|
|
) |
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
if callee_account is not None and callee_account.code.bytecode == "": |
|
|
|
if callee_account is not None and callee_account.code.bytecode == "": |
|
|
|
log.debug("The call is related to ether transfer between accounts") |
|
|
|
log.debug("The call is related to ether transfer between accounts") |
|
|
|
sender = global_state.environment.active_account.address |
|
|
|
sender = global_state.environment.active_account.address |
|
|
@ -1864,8 +1874,9 @@ class Instruction: |
|
|
|
callee_account=environment.active_account, |
|
|
|
callee_account=environment.active_account, |
|
|
|
call_data=call_data, |
|
|
|
call_data=call_data, |
|
|
|
call_value=value, |
|
|
|
call_value=value, |
|
|
|
|
|
|
|
static=environment.static, |
|
|
|
) |
|
|
|
) |
|
|
|
raise TransactionStartSignal(transaction, self.op_code) |
|
|
|
raise TransactionStartSignal(transaction, self.op_code, global_state) |
|
|
|
|
|
|
|
|
|
|
|
@StateTransition() |
|
|
|
@StateTransition() |
|
|
|
def callcode_post(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
def callcode_post(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
@ -1949,7 +1960,7 @@ class Instruction: |
|
|
|
|
|
|
|
|
|
|
|
if callee_account is not None and callee_account.code.bytecode == "": |
|
|
|
if callee_account is not None and callee_account.code.bytecode == "": |
|
|
|
log.debug("The call is related to ether transfer between accounts") |
|
|
|
log.debug("The call is related to ether transfer between accounts") |
|
|
|
sender = environment.active_account.address |
|
|
|
sender = global_state.environment.active_account.address |
|
|
|
receiver = callee_account.address |
|
|
|
receiver = callee_account.address |
|
|
|
transfer_ether(global_state, sender, receiver, value) |
|
|
|
transfer_ether(global_state, sender, receiver, value) |
|
|
|
|
|
|
|
|
|
|
@ -1979,8 +1990,9 @@ class Instruction: |
|
|
|
callee_account=environment.active_account, |
|
|
|
callee_account=environment.active_account, |
|
|
|
call_data=call_data, |
|
|
|
call_data=call_data, |
|
|
|
call_value=environment.callvalue, |
|
|
|
call_value=environment.callvalue, |
|
|
|
|
|
|
|
static=environment.static, |
|
|
|
) |
|
|
|
) |
|
|
|
raise TransactionStartSignal(transaction, self.op_code) |
|
|
|
raise TransactionStartSignal(transaction, self.op_code, global_state) |
|
|
|
|
|
|
|
|
|
|
|
@StateTransition() |
|
|
|
@StateTransition() |
|
|
|
def delegatecall_post(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
def delegatecall_post(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
@ -2012,6 +2024,7 @@ class Instruction: |
|
|
|
"retval_" + str(instr["address"]), 256 |
|
|
|
"retval_" + str(instr["address"]), 256 |
|
|
|
) |
|
|
|
) |
|
|
|
global_state.mstate.stack.append(return_value) |
|
|
|
global_state.mstate.stack.append(return_value) |
|
|
|
|
|
|
|
global_state.mstate.constraints.append(return_value == 0) |
|
|
|
return [global_state] |
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
try: |
|
|
|
try: |
|
|
@ -2053,8 +2066,8 @@ class Instruction: |
|
|
|
:param global_state: |
|
|
|
:param global_state: |
|
|
|
:return: |
|
|
|
:return: |
|
|
|
""" |
|
|
|
""" |
|
|
|
# TODO: implement me |
|
|
|
|
|
|
|
instr = global_state.get_current_instruction() |
|
|
|
instr = global_state.get_current_instruction() |
|
|
|
|
|
|
|
environment = global_state.environment |
|
|
|
try: |
|
|
|
try: |
|
|
|
callee_address, callee_account, call_data, value, gas, memory_out_offset, memory_out_size = get_call_parameters( |
|
|
|
callee_address, callee_account, call_data, value, gas, memory_out_offset, memory_out_size = get_call_parameters( |
|
|
|
global_state, self.dynamic_loader |
|
|
|
global_state, self.dynamic_loader |
|
|
@ -2062,7 +2075,7 @@ class Instruction: |
|
|
|
|
|
|
|
|
|
|
|
if callee_account is not None and callee_account.code.bytecode == "": |
|
|
|
if callee_account is not None and callee_account.code.bytecode == "": |
|
|
|
log.debug("The call is related to ether transfer between accounts") |
|
|
|
log.debug("The call is related to ether transfer between accounts") |
|
|
|
sender = global_state.environment.active_account.address |
|
|
|
sender = environment.active_account.address |
|
|
|
receiver = callee_account.address |
|
|
|
receiver = callee_account.address |
|
|
|
transfer_ether(global_state, sender, receiver, value) |
|
|
|
transfer_ether(global_state, sender, receiver, value) |
|
|
|
|
|
|
|
|
|
|
@ -2085,10 +2098,82 @@ class Instruction: |
|
|
|
native_result = native_call( |
|
|
|
native_result = native_call( |
|
|
|
global_state, callee_address, call_data, memory_out_offset, memory_out_size |
|
|
|
global_state, callee_address, call_data, memory_out_offset, memory_out_size |
|
|
|
) |
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
if native_result: |
|
|
|
if native_result: |
|
|
|
return native_result |
|
|
|
return native_result |
|
|
|
|
|
|
|
|
|
|
|
global_state.mstate.stack.append( |
|
|
|
transaction = MessageCallTransaction( |
|
|
|
global_state.new_bitvec("retval_" + str(instr["address"]), 256) |
|
|
|
world_state=global_state.world_state, |
|
|
|
|
|
|
|
gas_price=environment.gasprice, |
|
|
|
|
|
|
|
gas_limit=gas, |
|
|
|
|
|
|
|
origin=environment.origin, |
|
|
|
|
|
|
|
code=callee_account.code, |
|
|
|
|
|
|
|
caller=environment.address, |
|
|
|
|
|
|
|
callee_account=environment.active_account, |
|
|
|
|
|
|
|
call_data=call_data, |
|
|
|
|
|
|
|
call_value=value, |
|
|
|
|
|
|
|
static=True, |
|
|
|
) |
|
|
|
) |
|
|
|
|
|
|
|
raise TransactionStartSignal(transaction, self.op_code, global_state) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def staticcall_post(self, global_state: GlobalState) -> List[GlobalState]: |
|
|
|
|
|
|
|
return self.post_handler(global_state, function_name="staticcall") |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def post_handler(self, global_state, function_name: str): |
|
|
|
|
|
|
|
instr = global_state.get_current_instruction() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
try: |
|
|
|
|
|
|
|
callee_address, callee_account, call_data, value, gas, memory_out_offset, memory_out_size = get_call_parameters( |
|
|
|
|
|
|
|
global_state, self.dynamic_loader, True |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
except ValueError as e: |
|
|
|
|
|
|
|
log.debug( |
|
|
|
|
|
|
|
"Could not determine required parameters for {}, putting fresh symbol on the stack. \n{}".format( |
|
|
|
|
|
|
|
function_name, e |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
global_state.mstate.stack.append( |
|
|
|
|
|
|
|
global_state.new_bitvec("retval_" + str(instr["address"]), 256) |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if global_state.last_return_data is None: |
|
|
|
|
|
|
|
# Put return value on stack |
|
|
|
|
|
|
|
return_value = global_state.new_bitvec( |
|
|
|
|
|
|
|
"retval_" + str(instr["address"]), 256 |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
global_state.mstate.stack.append(return_value) |
|
|
|
|
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
try: |
|
|
|
|
|
|
|
memory_out_offset = ( |
|
|
|
|
|
|
|
util.get_concrete_int(memory_out_offset) |
|
|
|
|
|
|
|
if isinstance(memory_out_offset, Expression) |
|
|
|
|
|
|
|
else memory_out_offset |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
memory_out_size = ( |
|
|
|
|
|
|
|
util.get_concrete_int(memory_out_size) |
|
|
|
|
|
|
|
if isinstance(memory_out_size, Expression) |
|
|
|
|
|
|
|
else memory_out_size |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
except TypeError: |
|
|
|
|
|
|
|
global_state.mstate.stack.append( |
|
|
|
|
|
|
|
global_state.new_bitvec("retval_" + str(instr["address"]), 256) |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
return [global_state] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Copy memory |
|
|
|
|
|
|
|
global_state.mstate.mem_extend( |
|
|
|
|
|
|
|
memory_out_offset, min(memory_out_size, len(global_state.last_return_data)) |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
for i in range(min(memory_out_size, len(global_state.last_return_data))): |
|
|
|
|
|
|
|
global_state.mstate.memory[ |
|
|
|
|
|
|
|
i + memory_out_offset |
|
|
|
|
|
|
|
] = global_state.last_return_data[i] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Put return value on stack |
|
|
|
|
|
|
|
return_value = global_state.new_bitvec("retval_" + str(instr["address"]), 256) |
|
|
|
|
|
|
|
global_state.mstate.stack.append(return_value) |
|
|
|
|
|
|
|
global_state.mstate.constraints.append(return_value == 1) |
|
|
|
|
|
|
|
|
|
|
|
return [global_state] |
|
|
|
return [global_state] |
|
|
|