class BaseComputation(Configurable, ABC):
"""
The base class for all execution computations.
.. note::
Each :class:`~eth.vm.computation.BaseComputation` class must be configured with:
``opcodes``: A mapping from the opcode integer value to the logic function for the opcode.
``_precompiles``: A mapping of contract address to the precompile function for execution
of precompiled contracts.
"""
state = None
msg = None
transaction_context = None
_memory = None
_stack = None
_gas_meter = None
code = None
children = None # type: List[BaseComputation]
_output = b''
return_data = b''
_error = None # type: VMError
_log_entries = None # type: List[Tuple[int, bytes, List[int], bytes]]
accounts_to_delete = None # type: Dict[bytes, bytes]
# VM configuration
opcodes = None # type: Dict[int, Opcode]
_precompiles = None # type: Dict[Address, Callable[['BaseComputation'], Any]]
logger = cast(TraceLogger,
logging.getLogger('eth.vm.computation.Computation'))
def __init__(self, state: BaseState, message: Message,
transaction_context: BaseTransactionContext) -> None:
self.state = state
self.msg = message
self.transaction_context = transaction_context
self._memory = Memory()
self._stack = Stack()
self._gas_meter = GasMeter(message.gas)
self.children = []
self.accounts_to_delete = {}
self._log_entries = []
code = message.code
self.code = CodeStream(code)
#
# Convenience
#
@property
def is_origin_computation(self) -> bool:
"""
Return ``True`` if this computation is the outermost computation at ``depth == 0``.
"""
return self.msg.sender == self.transaction_context.origin
@property
def is_success(self) -> bool:
"""
Return ``True`` if the computation did not result in an error.
"""
return self._error is None
@property
def is_error(self) -> bool:
"""
Return ``True`` if the computation resulted in an error.
"""
return not self.is_success
def raise_if_error(self) -> None:
"""
If there was an error during computation, raise it as an exception immediately.
:raise VMError:
"""
if self._error is not None:
raise self._error
@property
def should_burn_gas(self) -> bool:
"""
Return ``True`` if the remaining gas should be burned.
"""
return self.is_error and self._error.burns_gas
@property
def should_return_gas(self) -> bool:
"""
Return ``True`` if the remaining gas should be returned.
"""
return not self.should_burn_gas
@property
def should_erase_return_data(self) -> bool:
"""
Return ``True`` if the return data should be zerod out due to an error.
"""
return self.is_error and self._error.erases_return_data
#
# Execution
#
def prepare_child_message(self, gas: int, to: bytes, value: int,
data: bytes, code: bytes,
**kwargs: Any) -> Message:
"""
Helper method for creating a child computation.
"""
kwargs.setdefault('sender', self.msg.storage_address)
child_message = Message(gas=gas,
to=to,
value=value,
data=data,
code=code,
depth=self.msg.depth + 1,
**kwargs)
return child_message
#
# Memory Management
#
def extend_memory(self, start_position: int, size: int) -> None:
"""
Extend the size of the memory to be at minimum ``start_position + size``
bytes in length. Raise `eth.exceptions.OutOfGas` if there is not enough
gas to pay for extending the memory.
"""
validate_uint256(start_position, title="Memory start position")
validate_uint256(size, title="Memory size")
before_size = ceil32(len(self._memory))
after_size = ceil32(start_position + size)
before_cost = memory_gas_cost(before_size)
after_cost = memory_gas_cost(after_size)
self.logger.trace(
"MEMORY: size (%s -> %s) | cost (%s -> %s)",
before_size,
after_size,
before_cost,
after_cost,
)
if size:
if before_cost < after_cost:
gas_fee = after_cost - before_cost
self._gas_meter.consume_gas(gas_fee,
reason=" ".join((
"Expanding memory",
str(before_size),
"->",
str(after_size),
)))
self._memory.extend(start_position, size)
def memory_write(self, start_position: int, size: int,
value: bytes) -> None:
"""
Write ``value`` to memory at ``start_position``. Require that ``len(value) == size``.
"""
return self._memory.write(start_position, size, value)
def memory_read(self, start_position: int, size: int) -> bytes:
"""
Read and return ``size`` bytes from memory starting at ``start_position``.
"""
return self._memory.read(start_position, size)
def consume_gas(self, amount: int, reason: str) -> None:
"""
Consume ``amount`` of gas from the remaining gas.
Raise `eth.exceptions.OutOfGas` if there is not enough gas remaining.
"""
return self._gas_meter.consume_gas(amount, reason)
def return_gas(self, amount: int) -> None:
"""
Return ``amount`` of gas to the available gas pool.
"""
return self._gas_meter.return_gas(amount)
def refund_gas(self, amount: int) -> None:
"""
Add ``amount`` of gas to the pool of gas marked to be refunded.
"""
return self._gas_meter.refund_gas(amount)
def stack_pop(self, num_items=1, type_hint=None):
"""
Pop and return a number of items equal to ``num_items`` from the stack.
``type_hint`` can be either ``'uint256'`` or ``'bytes'``. The return value
will be an ``int`` or ``bytes`` type depending on the value provided for
the ``type_hint``.
Raise `eth.exceptions.InsufficientStack` if there are not enough items on
the stack.
"""
return self._stack.pop(num_items, type_hint)
def stack_push(self, value):
"""
Push ``value`` onto the stack.
Raise `eth.exceptions.StackDepthLimit` if the stack is full.
"""
return self._stack.push(value)
def stack_swap(self, position):
"""
Swap the item on the top of the stack with the item at ``position``.
"""
return self._stack.swap(position)
def stack_dup(self, position):
"""
Duplicate the stack item at ``position`` and pushes it onto the stack.
"""
return self._stack.dup(position)
#
# Computed properties.
#
@property
def output(self) -> bytes:
"""
Get the return value of the computation.
"""
if self.should_erase_return_data:
return b''
else:
return self._output
@output.setter
def output(self, value: bytes) -> None:
"""
Set the return value of the computation.
"""
validate_is_bytes(value)
self._output = value
#
# Runtime operations
#
def apply_child_computation(self, child_msg: Message) -> 'BaseComputation':
"""
Apply the vm message ``child_msg`` as a child computation.
"""
child_computation = self.generate_child_computation(child_msg)
self.add_child_computation(child_computation)
return child_computation
def generate_child_computation(self,
child_msg: Message) -> 'BaseComputation':
if child_msg.is_create:
child_computation = self.__class__(
self.state,
child_msg,
self.transaction_context,
).apply_create_message()
else:
child_computation = self.__class__(
self.state,
child_msg,
self.transaction_context,
).apply_message()
return child_computation
def add_child_computation(self,
child_computation: 'BaseComputation') -> None:
if child_computation.is_error:
if child_computation.msg.is_create:
self.return_data = child_computation.output
elif child_computation.should_burn_gas:
self.return_data = b''
else:
self.return_data = child_computation.output
else:
if child_computation.msg.is_create:
self.return_data = b''
else:
self.return_data = child_computation.output
self.children.append(child_computation)
def register_account_for_deletion(self, beneficiary: Address) -> None:
validate_canonical_address(beneficiary,
title="Self destruct beneficiary address")
if self.msg.storage_address in self.accounts_to_delete:
raise ValueError(
"Invariant. Should be impossible for an account to be "
"registered for deletion multiple times")
self.accounts_to_delete[self.msg.storage_address] = beneficiary
def add_log_entry(self, account: Address, topics: List[int],
data: bytes) -> None:
validate_canonical_address(account, title="Log entry address")
for topic in topics:
validate_uint256(topic, title="Log entry topic")
validate_is_bytes(data, title="Log entry data")
self._log_entries.append(
(self.transaction_context.get_next_log_counter(), account, topics,
data))
#
# Getters
#
def get_accounts_for_deletion(self) -> Tuple[Tuple[bytes, bytes], ...]:
if self.is_error:
return tuple()
else:
return tuple(
dict(
itertools.chain(
self.accounts_to_delete.items(),
*(child.get_accounts_for_deletion()
for child in self.children))).items())
def _get_log_entries(self) -> List[Tuple[int, bytes, List[int], bytes]]:
"""
Return the log entries for this computation and its children.
They are sorted in the same order they were emitted during the transaction processing, and
include the sequential counter as the first element of the tuple representing every entry.
"""
if self.is_error:
return []
else:
return sorted(
itertools.chain(
self._log_entries,
*(child._get_log_entries() for child in self.children)))
def get_log_entries(self) -> Tuple[Tuple[bytes, List[int], bytes], ...]:
return tuple(log[1:] for log in self._get_log_entries())
def get_gas_refund(self) -> int:
if self.is_error:
return 0
else:
return self._gas_meter.gas_refunded + sum(c.get_gas_refund()
for c in self.children)
def get_gas_used(self) -> int:
if self.should_burn_gas:
return self.msg.gas
else:
return max(
0,
self.msg.gas - self._gas_meter.gas_remaining,
)
def get_gas_remaining(self) -> int:
if self.should_burn_gas:
return 0
else:
return self._gas_meter.gas_remaining
#
# Context Manager API
#
def __enter__(self) -> 'BaseComputation':
self.logger.trace(
("COMPUTATION STARTING: gas: %s | from: %s | to: %s | value: %s "
"| depth %s | static: %s"),
self.msg.gas,
encode_hex(self.msg.sender),
encode_hex(self.msg.to),
self.msg.value,
self.msg.depth,
"y" if self.msg.is_static else "n",
)
return self
def __exit__(self, exc_type: None, exc_value: None,
traceback: None) -> None:
if exc_value and isinstance(exc_value, VMError):
self.logger.trace(
("COMPUTATION ERROR: gas: %s | from: %s | to: %s | value: %s | "
"depth: %s | static: %s | error: %s"),
self.msg.gas,
encode_hex(self.msg.sender),
encode_hex(self.msg.to),
self.msg.value,
self.msg.depth,
"y" if self.msg.is_static else "n",
exc_value,
)
self._error = exc_value
if self.should_burn_gas:
self.consume_gas(
self._gas_meter.gas_remaining,
reason=" ".join((
"Zeroing gas due to VM Exception:",
str(exc_value),
)),
)
# suppress VM exceptions
return True
elif exc_type is None:
self.logger.trace(
("COMPUTATION SUCCESS: from: %s | to: %s | value: %s | "
"depth: %s | static: %s | gas-used: %s | gas-remaining: %s"),
encode_hex(self.msg.sender),
encode_hex(self.msg.to),
self.msg.value,
self.msg.depth,
"y" if self.msg.is_static else "n",
self.msg.gas - self._gas_meter.gas_remaining,
self._gas_meter.gas_remaining,
)
#
# State Transition
#
@abstractmethod
def apply_message(self) -> 'BaseComputation':
"""
Execution of an VM message.
"""
raise NotImplementedError("Must be implemented by subclasses")
@abstractmethod
def apply_create_message(self) -> 'BaseComputation':
"""
Execution of an VM message to create a new contract.
"""
raise NotImplementedError("Must be implemented by subclasses")
@classmethod
def apply_computation(
cls, state: BaseState, message: Message,
transaction_context: BaseTransactionContext) -> 'BaseComputation':
"""
Perform the computation that would be triggered by the VM message.
"""
with cls(state, message, transaction_context) as computation:
# Early exit on pre-compiles
if message.code_address in computation.precompiles:
computation.precompiles[message.code_address](computation)
return computation
for opcode in computation.code:
opcode_fn = computation.get_opcode_fn(opcode)
computation.logger.trace(
"OPCODE: 0x%x (%s) | pc: %s",
opcode,
opcode_fn.mnemonic,
max(0, computation.code.pc - 1),
)
try:
opcode_fn(computation=computation)
except Halt:
break
return computation
#
# Opcode API
#
@property
def precompiles(self) -> Dict[Address, Callable[['BaseComputation'], Any]]:
if self._precompiles is None:
return dict()
else:
return self._precompiles
def get_opcode_fn(self, opcode):
try:
return self.opcodes[opcode]
except KeyError:
return InvalidOpcode(opcode)
class BaseComputation(Configurable, ComputationAPI):
"""
The base class for all execution computations.
.. note::
Each :class:`~eth.vm.computation.BaseComputation` class must be configured with:
``opcodes``: A mapping from the opcode integer value to the logic function for the opcode.
``_precompiles``: A mapping of contract address to the precompile function for execution
of precompiled contracts.
"""
state: StateAPI = None
msg: MessageAPI = None
transaction_context: TransactionContextAPI = None
_memory: MemoryAPI = None
_stack: StackAPI = None
_gas_meter: GasMeterAPI = None
code: CodeStreamAPI = None
children: List[ComputationAPI] = None
_output: bytes = b''
return_data: bytes = b''
_error: VMError = None
# TODO: use a NamedTuple for log entries
_log_entries: List[Tuple[int, Address, Tuple[int, ...], bytes]] = None
accounts_to_delete: Dict[Address, Address] = None
# VM configuration
opcodes: Dict[int, OpcodeAPI] = None
_precompiles: Dict[Address, Callable[[ComputationAPI], ComputationAPI]] = None
logger = get_extended_debug_logger('eth.vm.computation.Computation')
def __init__(self,
state: StateAPI,
message: MessageAPI,
transaction_context: TransactionContextAPI) -> None:
self.state = state
self.msg = message
self.transaction_context = transaction_context
self._memory = Memory()
self._stack = Stack()
self._gas_meter = self.get_gas_meter()
self.children = []
self.accounts_to_delete = {}
self._log_entries = []
code = message.code
self.code = CodeStream(code)
#
# Convenience
#
@property
def is_origin_computation(self) -> bool:
return self.msg.sender == self.transaction_context.origin
#
# Error handling
#
@property
def is_success(self) -> bool:
return self._error is None
@property
def is_error(self) -> bool:
return not self.is_success
@property
def error(self) -> VMError:
if self._error is not None:
return self._error
raise AttributeError("Computation does not have an error")
@error.setter
def error(self, value: VMError) -> None:
if self._error is not None:
raise AttributeError(f"Computation already has an error set: {self._error}")
self._error = value
def raise_if_error(self) -> None:
if self._error is not None:
raise self._error
@property
def should_burn_gas(self) -> bool:
return self.is_error and self._error.burns_gas
@property
def should_return_gas(self) -> bool:
return not self.should_burn_gas
@property
def should_erase_return_data(self) -> bool:
return self.is_error and self._error.erases_return_data
#
# Memory Management
#
def extend_memory(self, start_position: int, size: int) -> None:
validate_uint256(start_position, title="Memory start position")
validate_uint256(size, title="Memory size")
before_size = ceil32(len(self._memory))
after_size = ceil32(start_position + size)
before_cost = memory_gas_cost(before_size)
after_cost = memory_gas_cost(after_size)
if self.logger.show_debug2:
self.logger.debug2(
"MEMORY: size (%s -> %s) | cost (%s -> %s)",
before_size,
after_size,
before_cost,
after_cost,
)
if size:
if before_cost < after_cost:
gas_fee = after_cost - before_cost
self._gas_meter.consume_gas(
gas_fee,
reason=" ".join((
"Expanding memory",
str(before_size),
"->",
str(after_size),
))
)
self._memory.extend(start_position, size)
def memory_write(self, start_position: int, size: int, value: bytes) -> None:
return self._memory.write(start_position, size, value)
def memory_read(self, start_position: int, size: int) -> memoryview:
return self._memory.read(start_position, size)
def memory_read_bytes(self, start_position: int, size: int) -> bytes:
return self._memory.read_bytes(start_position, size)
#
# Gas Consumption
#
def get_gas_meter(self) -> GasMeterAPI:
return GasMeter(self.msg.gas)
def consume_gas(self, amount: int, reason: str) -> None:
return self._gas_meter.consume_gas(amount, reason)
def return_gas(self, amount: int) -> None:
return self._gas_meter.return_gas(amount)
def refund_gas(self, amount: int) -> None:
return self._gas_meter.refund_gas(amount)
def get_gas_refund(self) -> int:
if self.is_error:
return 0
else:
return self._gas_meter.gas_refunded + sum(c.get_gas_refund() for c in self.children)
def get_gas_used(self) -> int:
if self.should_burn_gas:
return self.msg.gas
else:
return max(
0,
self.msg.gas - self._gas_meter.gas_remaining,
)
def get_gas_remaining(self) -> int:
if self.should_burn_gas:
return 0
else:
return self._gas_meter.gas_remaining
#
# Stack management
#
def stack_swap(self, position: int) -> None:
return self._stack.swap(position)
def stack_dup(self, position: int) -> None:
return self._stack.dup(position)
# Stack manipulation is performance-sensitive code.
# Avoid method call overhead by proxying stack method directly to stack object
@cached_property
def stack_pop_ints(self) -> Callable[[int], Tuple[int, ...]]:
return self._stack.pop_ints
@cached_property
def stack_pop_bytes(self) -> Callable[[int], Tuple[bytes, ...]]:
return self._stack.pop_bytes
@cached_property
def stack_pop_any(self) -> Callable[[int], Tuple[Union[int, bytes], ...]]:
return self._stack.pop_any
@cached_property
def stack_pop1_int(self) -> Callable[[], int]:
return self._stack.pop1_int
@cached_property
def stack_pop1_bytes(self) -> Callable[[], bytes]:
return self._stack.pop1_bytes
@cached_property
def stack_pop1_any(self) -> Callable[[], Union[int, bytes]]:
return self._stack.pop1_any
@cached_property
def stack_push_int(self) -> Callable[[int], None]:
return self._stack.push_int
@cached_property
def stack_push_bytes(self) -> Callable[[bytes], None]:
return self._stack.push_bytes
#
# Computation result
#
@property
def output(self) -> bytes:
if self.should_erase_return_data:
return b''
else:
return self._output
@output.setter
def output(self, value: bytes) -> None:
validate_is_bytes(value)
self._output = value
#
# Runtime operations
#
def prepare_child_message(self,
gas: int,
to: Address,
value: int,
data: BytesOrView,
code: bytes,
**kwargs: Any) -> MessageAPI:
kwargs.setdefault('sender', self.msg.storage_address)
child_message = Message(
gas=gas,
to=to,
value=value,
data=data,
code=code,
depth=self.msg.depth + 1,
**kwargs
)
return child_message
def apply_child_computation(self, child_msg: MessageAPI) -> ComputationAPI:
child_computation = self.generate_child_computation(child_msg)
self.add_child_computation(child_computation)
return child_computation
def generate_child_computation(self, child_msg: MessageAPI) -> ComputationAPI:
if child_msg.is_create:
child_computation = self.__class__(
self.state,
child_msg,
self.transaction_context,
).apply_create_message()
else:
child_computation = self.__class__(
self.state,
child_msg,
self.transaction_context,
).apply_message()
return child_computation
def add_child_computation(self, child_computation: ComputationAPI) -> None:
if child_computation.is_error:
if child_computation.msg.is_create:
self.return_data = child_computation.output
elif child_computation.should_burn_gas:
self.return_data = b''
else:
self.return_data = child_computation.output
else:
if child_computation.msg.is_create:
self.return_data = b''
else:
self.return_data = child_computation.output
self.children.append(child_computation)
#
# Account management
#
def register_account_for_deletion(self, beneficiary: Address) -> None:
validate_canonical_address(beneficiary, title="Self destruct beneficiary address")
if self.msg.storage_address in self.accounts_to_delete:
raise ValueError(
"Invariant. Should be impossible for an account to be "
"registered for deletion multiple times"
)
self.accounts_to_delete[self.msg.storage_address] = beneficiary
def get_accounts_for_deletion(self) -> Tuple[Tuple[Address, Address], ...]:
if self.is_error:
return tuple()
else:
return tuple(dict(itertools.chain(
self.accounts_to_delete.items(),
*(child.get_accounts_for_deletion() for child in self.children)
)).items())
#
# EVM logging
#
def add_log_entry(self, account: Address, topics: Tuple[int, ...], data: bytes) -> None:
validate_canonical_address(account, title="Log entry address")
for topic in topics:
validate_uint256(topic, title="Log entry topic")
validate_is_bytes(data, title="Log entry data")
self._log_entries.append(
(self.transaction_context.get_next_log_counter(), account, topics, data))
def get_raw_log_entries(self) -> Tuple[Tuple[int, bytes, Tuple[int, ...], bytes], ...]:
if self.is_error:
return ()
else:
return tuple(sorted(itertools.chain(
self._log_entries,
*(child.get_raw_log_entries() for child in self.children)
)))
def get_log_entries(self) -> Tuple[Tuple[bytes, Tuple[int, ...], bytes], ...]:
return tuple(log[1:] for log in self.get_raw_log_entries())
#
# Context Manager API
#
def __enter__(self) -> ComputationAPI:
if self.logger.show_debug2:
self.logger.debug2(
(
"COMPUTATION STARTING: gas: %s | from: %s | to: %s | value: %s "
"| depth %s | static: %s"
),
self.msg.gas,
encode_hex(self.msg.sender),
encode_hex(self.msg.to),
self.msg.value,
self.msg.depth,
"y" if self.msg.is_static else "n",
)
return self
def __exit__(self,
exc_type: Optional[Type[BaseException]],
exc_value: Optional[BaseException],
traceback: Optional[TracebackType]) -> Union[None, bool]:
if exc_value and isinstance(exc_value, VMError):
if self.logger.show_debug2:
self.logger.debug2(
(
"COMPUTATION ERROR: gas: %s | from: %s | to: %s | value: %s | "
"depth: %s | static: %s | error: %s"
),
self.msg.gas,
encode_hex(self.msg.sender),
encode_hex(self.msg.to),
self.msg.value,
self.msg.depth,
"y" if self.msg.is_static else "n",
exc_value,
)
self._error = exc_value
if self.should_burn_gas:
self.consume_gas(
self._gas_meter.gas_remaining,
reason=" ".join((
"Zeroing gas due to VM Exception:",
str(exc_value),
)),
)
# suppress VM exceptions
return True
elif exc_type is None and self.logger.show_debug2:
self.logger.debug2(
(
"COMPUTATION SUCCESS: from: %s | to: %s | value: %s | "
"depth: %s | static: %s | gas-used: %s | gas-remaining: %s"
),
encode_hex(self.msg.sender),
encode_hex(self.msg.to),
self.msg.value,
self.msg.depth,
"y" if self.msg.is_static else "n",
self.get_gas_used(),
self._gas_meter.gas_remaining,
)
return None
#
# State Transition
#
@abstractmethod
def apply_message(self) -> ComputationAPI:
raise NotImplementedError("Must be implemented by subclasses")
@abstractmethod
def apply_create_message(self) -> ComputationAPI:
raise NotImplementedError("Must be implemented by subclasses")
@classmethod
def apply_computation(cls,
state: StateAPI,
message: MessageAPI,
transaction_context: TransactionContextAPI) -> ComputationAPI:
with cls(state, message, transaction_context) as computation:
# Early exit on pre-compiles
precompile = computation.precompiles.get(message.code_address, NO_RESULT)
if precompile is not NO_RESULT:
precompile(computation)
return computation
show_debug2 = computation.logger.show_debug2
opcode_lookup = computation.opcodes
for opcode in computation.code:
try:
opcode_fn = opcode_lookup[opcode]
except KeyError:
opcode_fn = InvalidOpcode(opcode)
if show_debug2:
computation.logger.debug2(
"OPCODE: 0x%x (%s) | pc: %s",
opcode,
opcode_fn.mnemonic,
max(0, computation.code.program_counter - 1),
)
try:
opcode_fn(computation=computation)
except Halt:
break
return computation
#
# Opcode API
#
@property
def precompiles(self) -> Dict[Address, Callable[[ComputationAPI], Any]]:
if self._precompiles is None:
return dict()
else:
return self._precompiles
def get_opcode_fn(self, opcode: int) -> OpcodeAPI:
try:
return self.opcodes[opcode]
except KeyError:
return InvalidOpcode(opcode)