ethereum-execution 2.18.0rc6.dev2__py3-none-any.whl

ethereum/forks/dao_fork/vm/memory.py

@@ -0,0 +1,84 @@
+"""
+Ethereum Virtual Machine (EVM) Memory.
+
+.. contents:: Table of Contents
+    :backlinks: none
+    :local:
+
+Introduction
+------------
+
+EVM memory operations.
+"""
+
+from ethereum_types.bytes import Bytes
+from ethereum_types.numeric import U256, Uint
+
+from ethereum.utils.byte import right_pad_zero_bytes
+
+
+def memory_write(
+    memory: bytearray, start_position: U256, value: Bytes
+) -> None:
+    """
+    Writes to memory.
+
+    Parameters
+    ----------
+    memory :
+        Memory contents of the EVM.
+    start_position :
+        Starting pointer to the memory.
+    value :
+        Data to write to memory.
+
+    """
+    memory[start_position : int(start_position) + len(value)] = value
+
+
+def memory_read_bytes(
+    memory: bytearray, start_position: U256, size: U256
+) -> Bytes:
+    """
+    Read bytes from memory.
+
+    Parameters
+    ----------
+    memory :
+        Memory contents of the EVM.
+    start_position :
+        Starting pointer to the memory.
+    size :
+        Size of the data that needs to be read from `start_position`.
+
+    Returns
+    -------
+    data_bytes :
+        Data read from memory.
+
+    """
+    return Bytes(memory[start_position : Uint(start_position) + Uint(size)])
+
+
+def buffer_read(buffer: Bytes, start_position: U256, size: U256) -> Bytes:
+    """
+    Read bytes from a buffer. Padding with zeros if necessary.
+
+    Parameters
+    ----------
+    buffer :
+        Memory contents of the EVM.
+    start_position :
+        Starting pointer to the memory.
+    size :
+        Size of the data that needs to be read from `start_position`.
+
+    Returns
+    -------
+    data_bytes :
+        Data read from memory.
+
+    """
+    return right_pad_zero_bytes(
+        buffer[start_position : Uint(start_position) + Uint(size)], size
+    )

ethereum/forks/dao_fork/vm/precompiled_contracts/__init__.py

@@ -0,0 +1,27 @@
+"""
+Precompiled Contract Addresses.
+
+.. contents:: Table of Contents
+    :backlinks: none
+    :local:
+
+Introduction
+------------
+
+Addresses of precompiled contracts and mappings to their
+implementations.
+"""
+
+from ...utils.hexadecimal import hex_to_address
+
+__all__ = (
+    "ECRECOVER_ADDRESS",
+    "SHA256_ADDRESS",
+    "RIPEMD160_ADDRESS",
+    "IDENTITY_ADDRESS",
+)
+
+ECRECOVER_ADDRESS = hex_to_address("0x01")
+SHA256_ADDRESS = hex_to_address("0x02")
+RIPEMD160_ADDRESS = hex_to_address("0x03")
+IDENTITY_ADDRESS = hex_to_address("0x04")

ethereum/forks/dao_fork/vm/precompiled_contracts/ecrecover.py

@@ -0,0 +1,64 @@
+"""
+Ethereum Virtual Machine (EVM) ECRECOVER PRECOMPILED CONTRACT.
+
+.. contents:: Table of Contents
+    :backlinks: none
+    :local:
+
+Introduction
+------------
+
+Implementation of the ECRECOVER precompiled contract.
+"""
+
+from ethereum_types.numeric import U256
+
+from ethereum.crypto.elliptic_curve import SECP256K1N, secp256k1_recover
+from ethereum.crypto.hash import Hash32, keccak256
+from ethereum.exceptions import InvalidSignatureError
+from ethereum.utils.byte import left_pad_zero_bytes
+
+from ...vm import Evm
+from ...vm.gas import GAS_ECRECOVER, charge_gas
+from ...vm.memory import buffer_read
+
+
+def ecrecover(evm: Evm) -> None:
+    """
+    Decrypts the address using elliptic curve DSA recovery mechanism and writes
+    the address to output.
+
+    Parameters
+    ----------
+    evm :
+        The current EVM frame.
+
+    """
+    data = evm.message.data
+
+    # GAS
+    charge_gas(evm, GAS_ECRECOVER)
+
+    # OPERATION
+    message_hash_bytes = buffer_read(data, U256(0), U256(32))
+    message_hash = Hash32(message_hash_bytes)
+    v = U256.from_be_bytes(buffer_read(data, U256(32), U256(32)))
+    r = U256.from_be_bytes(buffer_read(data, U256(64), U256(32)))
+    s = U256.from_be_bytes(buffer_read(data, U256(96), U256(32)))
+
+    if v != U256(27) and v != U256(28):
+        return
+    if U256(0) >= r or r >= SECP256K1N:
+        return
+    if U256(0) >= s or s >= SECP256K1N:
+        return
+
+    try:
+        public_key = secp256k1_recover(r, s, v - U256(27), message_hash)
+    except InvalidSignatureError:
+        # unable to extract public key
+        return
+
+    address = keccak256(public_key)[12:32]
+    padded_address = left_pad_zero_bytes(address, 32)
+    evm.output = padded_address

ethereum/forks/dao_fork/vm/precompiled_contracts/identity.py

@@ -0,0 +1,39 @@
+"""
+Ethereum Virtual Machine (EVM) IDENTITY PRECOMPILED CONTRACT.
+
+.. contents:: Table of Contents
+    :backlinks: none
+    :local:
+
+Introduction
+------------
+
+Implementation of the `IDENTITY` precompiled contract.
+"""
+
+from ethereum_types.numeric import Uint
+
+from ethereum.utils.numeric import ceil32
+
+from ...vm import Evm
+from ...vm.gas import GAS_IDENTITY, GAS_IDENTITY_WORD, charge_gas
+
+
+def identity(evm: Evm) -> None:
+    """
+    Writes the message data to output.
+
+    Parameters
+    ----------
+    evm :
+        The current EVM frame.
+
+    """
+    data = evm.message.data
+
+    # GAS
+    word_count = ceil32(Uint(len(data))) // Uint(32)
+    charge_gas(evm, GAS_IDENTITY + GAS_IDENTITY_WORD * word_count)
+
+    # OPERATION
+    evm.output = data

ethereum/forks/dao_fork/vm/precompiled_contracts/mapping.py

@@ -0,0 +1,33 @@
+"""
+Precompiled Contract Addresses.
+
+.. contents:: Table of Contents
+    :backlinks: none
+    :local:
+
+Introduction
+------------
+
+Mapping of precompiled contracts their implementations.
+"""
+
+from typing import Callable, Dict
+
+from ...fork_types import Address
+from . import (
+    ECRECOVER_ADDRESS,
+    IDENTITY_ADDRESS,
+    RIPEMD160_ADDRESS,
+    SHA256_ADDRESS,
+)
+from .ecrecover import ecrecover
+from .identity import identity
+from .ripemd160 import ripemd160
+from .sha256 import sha256
+
+PRE_COMPILED_CONTRACTS: Dict[Address, Callable] = {
+    ECRECOVER_ADDRESS: ecrecover,
+    SHA256_ADDRESS: sha256,
+    RIPEMD160_ADDRESS: ripemd160,
+    IDENTITY_ADDRESS: identity,
+}

ethereum/forks/dao_fork/vm/precompiled_contracts/ripemd160.py

@@ -0,0 +1,44 @@
+"""
+Ethereum Virtual Machine (EVM) RIPEMD160 PRECOMPILED CONTRACT.
+
+.. contents:: Table of Contents
+    :backlinks: none
+    :local:
+
+Introduction
+------------
+
+Implementation of the `RIPEMD160` precompiled contract.
+"""
+
+import hashlib
+
+from ethereum_types.numeric import Uint
+
+from ethereum.utils.byte import left_pad_zero_bytes
+from ethereum.utils.numeric import ceil32
+
+from ...vm import Evm
+from ...vm.gas import GAS_RIPEMD160, GAS_RIPEMD160_WORD, charge_gas
+
+
+def ripemd160(evm: Evm) -> None:
+    """
+    Writes the ripemd160 hash to output.
+
+    Parameters
+    ----------
+    evm :
+        The current EVM frame.
+
+    """
+    data = evm.message.data
+
+    # GAS
+    word_count = ceil32(Uint(len(data))) // Uint(32)
+    charge_gas(evm, GAS_RIPEMD160 + GAS_RIPEMD160_WORD * word_count)
+
+    # OPERATION
+    hash_bytes = hashlib.new("ripemd160", data).digest()
+    padded_hash = left_pad_zero_bytes(hash_bytes, 32)
+    evm.output = padded_hash

ethereum/forks/dao_fork/vm/precompiled_contracts/sha256.py

@@ -0,0 +1,41 @@
+"""
+Ethereum Virtual Machine (EVM) SHA256 PRECOMPILED CONTRACT.
+
+.. contents:: Table of Contents
+    :backlinks: none
+    :local:
+
+Introduction
+------------
+
+Implementation of the `SHA256` precompiled contract.
+"""
+
+import hashlib
+
+from ethereum_types.numeric import Uint
+
+from ethereum.utils.numeric import ceil32
+
+from ...vm import Evm
+from ...vm.gas import GAS_SHA256, GAS_SHA256_WORD, charge_gas
+
+
+def sha256(evm: Evm) -> None:
+    """
+    Writes the sha256 hash to output.
+
+    Parameters
+    ----------
+    evm :
+        The current EVM frame.
+
+    """
+    data = evm.message.data
+
+    # GAS
+    word_count = ceil32(Uint(len(data))) // Uint(32)
+    charge_gas(evm, GAS_SHA256 + GAS_SHA256_WORD * word_count)
+
+    # OPERATION
+    evm.output = hashlib.sha256(data).digest()

ethereum/forks/dao_fork/vm/runtime.py

@@ -0,0 +1,69 @@
+"""
+Ethereum Virtual Machine (EVM) Runtime Operations.
+
+.. contents:: Table of Contents
+    :backlinks: none
+    :local:
+
+Introduction
+------------
+
+Runtime related operations used while executing EVM code.
+"""
+
+from typing import Set
+
+from ethereum_types.bytes import Bytes
+from ethereum_types.numeric import Uint, ulen
+
+from .instructions import Ops
+
+
+def get_valid_jump_destinations(code: Bytes) -> Set[Uint]:
+    """
+    Analyze the evm code to obtain the set of valid jump destinations.
+
+    Valid jump destinations are defined as follows:
+        * The jump destination is less than the length of the code.
+        * The jump destination should have the `JUMPDEST` opcode (0x5B).
+        * The jump destination shouldn't be part of the data corresponding to
+          `PUSH-N` opcodes.
+
+    Note - Jump destinations are 0-indexed.
+
+    Parameters
+    ----------
+    code :
+        The EVM code which is to be executed.
+
+    Returns
+    -------
+    valid_jump_destinations: `Set[Uint]`
+        The set of valid jump destinations in the code.
+
+    """
+    valid_jump_destinations = set()
+    pc = Uint(0)
+
+    while pc < ulen(code):
+        try:
+            current_opcode = Ops(code[pc])
+        except ValueError:
+            # Skip invalid opcodes, as they don't affect the jumpdest
+            # analysis. Nevertheless, such invalid opcodes would be caught
+            # and raised when the interpreter runs.
+            pc += Uint(1)
+            continue
+
+        if current_opcode == Ops.JUMPDEST:
+            valid_jump_destinations.add(pc)
+        elif Ops.PUSH1.value <= current_opcode.value <= Ops.PUSH32.value:
+            # If PUSH-N opcodes are encountered, skip the current opcode along
+            # with the trailing data segment corresponding to the PUSH-N
+            # opcodes.
+            push_data_size = current_opcode.value - Ops.PUSH1.value + 1
+            pc += Uint(push_data_size)
+
+        pc += Uint(1)
+
+    return valid_jump_destinations

ethereum/forks/dao_fork/vm/stack.py

@@ -0,0 +1,58 @@
+"""
+Ethereum Virtual Machine (EVM) Stack.
+
+.. contents:: Table of Contents
+    :backlinks: none
+    :local:
+
+Introduction
+------------
+
+Implementation of the stack operators for the EVM.
+"""
+
+from typing import List
+
+from ethereum_types.numeric import U256
+
+from .exceptions import StackOverflowError, StackUnderflowError
+
+
+def pop(stack: List[U256]) -> U256:
+    """
+    Pops the top item off of `stack`.
+
+    Parameters
+    ----------
+    stack :
+        EVM stack.
+
+    Returns
+    -------
+    value : `U256`
+        The top element on the stack.
+
+    """
+    if len(stack) == 0:
+        raise StackUnderflowError
+
+    return stack.pop()
+
+
+def push(stack: List[U256], value: U256) -> None:
+    """
+    Pushes `value` onto `stack`.
+
+    Parameters
+    ----------
+    stack :
+        EVM stack.
+
+    value :
+        Item to be pushed onto `stack`.
+
+    """
+    if len(stack) == 1024:
+        raise StackOverflowError
+
+    return stack.append(value)

ethereum/forks/frontier/__init__.py

@@ -0,0 +1,7 @@
+"""
+Frontier is the first production-ready iteration of the Ethereum protocol.
+"""
+
+from ethereum.fork_criteria import ByBlockNumber
+
+FORK_CRITERIA = ByBlockNumber(0)