ethereum.forks.prague.state_trackerethereum.forks.osaka.state_tracker
State Tracking for Block Execution.
Track state changes on top of a read-only PreState. At block end,
accumulated diffs feed into
PreState.compute_state_root().
.. contents:: Table of Contents :backlinks: none :local:
Introduction
Replace the mutable State class with lightweight state trackers that
record diffs. BlockState accumulates committed transaction
changes across a block. TransactionState tracks in-flight changes
within a single transaction and supports copy-on-write rollback.
BlockState ¶
Accumulate committed transaction-level changes across a block.
Read chain: block writes -> pre_state.
storage_clears records addresses whose pre-existing storage
was wiped earlier in the block, so later reads must not fall back
to pre_state. Contract creation over a storage-only account
and deletion of an empty account that still holds storage both
wipe storage.
| 39 | @final |
|---|
| 40 | @dataclass |
|---|
class BlockState:
pre_state¶
| 54 | pre_state: PreState |
|---|
account_writes¶
| 55 | account_writes: Dict[Address, Optional[Account]] = field( |
|---|---|
| 56 | default_factory=dict |
| 57 | ) |
storage_writes¶
| 58 | storage_writes: Dict[Address, Dict[Bytes32, U256]] = field( |
|---|---|
| 59 | default_factory=dict |
| 60 | ) |
code_writes¶
| 61 | code_writes: Dict[Hash32, Bytes] = field(default_factory=dict) |
|---|
storage_clears¶
| 62 | storage_clears: Set[Address] = field(default_factory=set) |
|---|
TransactionState ¶
Track in-flight state changes within a single transaction.
Read chain: tx writes -> block writes -> pre_state.
| 65 | @final |
|---|
| 66 | @dataclass |
|---|
class TransactionState:
parent¶
| 74 | parent: BlockState |
|---|
account_writes¶
| 75 | account_writes: Dict[Address, Optional[Account]] = field( |
|---|---|
| 76 | default_factory=dict |
| 77 | ) |
storage_writes¶
| 78 | storage_writes: Dict[Address, Dict[Bytes32, U256]] = field( |
|---|---|
| 79 | default_factory=dict |
| 80 | ) |
code_writes¶
| 81 | code_writes: Dict[Hash32, Bytes] = field(default_factory=dict) |
|---|
created_accounts¶
| 82 | created_accounts: Set[Address] = field(default_factory=set) |
|---|
storage_clears¶
| 83 | storage_clears: Set[Address] = field(default_factory=set) |
|---|
transient_storage¶
| 84 | transient_storage: Dict[Tuple[Address, Bytes32], U256] = field( |
|---|---|
| 85 | default_factory=dict |
| 86 | ) |
get_account_optional ¶
Get the Account object at an address. Return None (rather than
EMPTY_ACCOUNT) if there is no account at the address.
Parameters
tx_state : The transaction state. address : Address to look up.
Returns
account : Optional[Account]
Account at address.
def get_account_optional(tx_state: TransactionState, address: Address) -> Optional[Account]:
| 92 | <snip> |
|---|---|
| 109 | if address in tx_state.account_writes: |
| 110 | return tx_state.account_writes[address] |
| 111 | if address in tx_state.parent.account_writes: |
| 112 | return tx_state.parent.account_writes[address] |
| 113 | return tx_state.parent.pre_state.get_account_optional(address) |
get_account ¶
Get the Account object at an address. Return EMPTY_ACCOUNT
if there is no account at the address.
Use get_account_optional() if you care about the difference
between a non-existent account and EMPTY_ACCOUNT.
Parameters
tx_state : The transaction state. address : Address to look up.
Returns
account : Account
Account at address.
def get_account(tx_state: TransactionState, address: Address) -> Account:
| 117 | <snip> |
|---|---|
| 137 | account = get_account_optional(tx_state, address) |
| 138 | if account is None: |
| 139 | return EMPTY_ACCOUNT |
| 140 | else: |
| 141 | return account |
get_code ¶
Get the bytecode for a given code hash.
Read chain: tx code_writes -> block code_writes -> pre_state.
Parameters
tx_state : The transaction state. code_hash : Hash of the code to look up.
Returns
code : Bytes
The bytecode.
def get_code(tx_state: TransactionState, code_hash: Hash32) -> Bytes:
| 145 | <snip> |
|---|---|
| 163 | if code_hash == EMPTY_CODE_HASH: |
| 164 | return b"" |
| 165 | if code_hash in tx_state.code_writes: |
| 166 | return tx_state.code_writes[code_hash] |
| 167 | if code_hash in tx_state.parent.code_writes: |
| 168 | return tx_state.parent.code_writes[code_hash] |
| 169 | return tx_state.parent.pre_state.get_code(code_hash) |
get_storage ¶
Get a value at a storage key on an account. Return U256(0) if
the storage key has not been set previously.
Parameters
tx_state : The transaction state. address : Address of the account. key : Key to look up.
Returns
value : U256
Value at the key.
def get_storage(tx_state: TransactionState, address: Address, key: Bytes32) -> U256:
| 175 | <snip> |
|---|---|
| 194 | if address in tx_state.storage_writes: |
| 195 | if key in tx_state.storage_writes[address]: |
| 196 | return tx_state.storage_writes[address][key] |
| 197 | if address in tx_state.storage_clears: |
| 198 | return U256(0) |
| 199 | if address in tx_state.parent.storage_writes: |
| 200 | if key in tx_state.parent.storage_writes[address]: |
| 201 | return tx_state.parent.storage_writes[address][key] |
| 202 | if address in tx_state.parent.storage_clears: |
| 203 | return U256(0) |
| 204 | return tx_state.parent.pre_state.get_storage(address, key) |
get_storage_original ¶
Get the original value in a storage slot i.e. the value before the
current transaction began. Read from block-level writes, then
pre_state. Return U256(0) for accounts created in the current
transaction.
Parameters
tx_state : The transaction state. address : Address of the account to read the value from. key : Key of the storage slot.
def get_storage_original(tx_state: TransactionState, address: Address, key: Bytes32) -> U256:
| 210 | <snip> |
|---|---|
| 226 | if address in tx_state.created_accounts: |
| 227 | return U256(0) |
| 228 | if address in tx_state.parent.storage_writes: |
| 229 | if key in tx_state.parent.storage_writes[address]: |
| 230 | return tx_state.parent.storage_writes[address][key] |
| 231 | if address in tx_state.parent.storage_clears: |
| 232 | return U256(0) |
| 233 | return tx_state.parent.pre_state.get_storage(address, key) |
get_transient_storage ¶
Get a value at a storage key on an account from transient storage.
Return U256(0) if the storage key has not been set previously.
Parameters
tx_state : The transaction state. address : Address of the account. key : Key to look up.
Returns
value : U256
Value at the key.
def get_transient_storage(tx_state: TransactionState, address: Address, key: Bytes32) -> U256:
| 239 | <snip> |
|---|---|
| 258 | return tx_state.transient_storage.get((address, key), U256(0)) |
account_exists ¶
Check if an account exists in the state trie.
Parameters
tx_state : The transaction state. address : Address of the account that needs to be checked.
Returns
account_exists : bool
True if account exists in the state trie, False otherwise.
def account_exists(tx_state: TransactionState, address: Address) -> bool:
| 262 | <snip> |
|---|---|
| 278 | return get_account_optional(tx_state, address) is not None |
account_deployable ¶
Check if an account's code can be written to.
def account_deployable(tx_state: TransactionState, address: Address) -> bool:
| 282 | <snip> |
|---|---|
| 285 | account = get_account(tx_state, address) |
| 286 | if account.nonce != Uint(0) or account.code_hash != EMPTY_CODE_HASH: |
| 287 | return False |
| 288 | |
| 289 | return True |
account_exists_and_is_empty ¶
Check if an account exists and has zero nonce, empty code and zero balance.
Parameters
tx_state : The transaction state. address : Address of the account that needs to be checked.
Returns
exists_and_is_empty : bool
True if an account exists and has zero nonce, empty code and
zero balance, False otherwise.
def account_exists_and_is_empty(tx_state: TransactionState, address: Address) -> bool:
| 295 | <snip> |
|---|---|
| 313 | account = get_account_optional(tx_state, address) |
| 314 | return ( |
| 315 | account is not None |
| 316 | and account.nonce == Uint(0) |
| 317 | and account.code_hash == EMPTY_CODE_HASH |
| 318 | and account.balance == 0 |
| 319 | ) |
is_account_alive ¶
Check whether an account is both in the state and non-empty.
Parameters
tx_state : The transaction state. address : Address of the account that needs to be checked.
Returns
is_alive : bool
True if the account is alive.
def is_account_alive(tx_state: TransactionState, address: Address) -> bool:
| 323 | <snip> |
|---|---|
| 339 | account = get_account_optional(tx_state, address) |
| 340 | return account is not None and account != EMPTY_ACCOUNT |
set_account ¶
Set the Account object at an address. Setting to None
deletes the account (but not its storage, see
destroy_account()).
Parameters
tx_state : The transaction state. address : Address to set. account : Account to set at address.
def set_account(tx_state: TransactionState, address: Address, account: Optional[Account]) -> None:
| 348 | <snip> |
|---|---|
| 363 | tx_state.account_writes[address] = account |
set_storage ¶
Set a value at a storage key on an account.
Parameters
tx_state : The transaction state. address : Address of the account. key : Key to set. value : Value to set at the key.
def set_storage(tx_state: TransactionState, address: Address, key: Bytes32, value: U256) -> None:
| 372 | <snip> |
|---|---|
| 387 | assert get_account_optional(tx_state, address) is not None |
| 388 | if address not in tx_state.storage_writes: |
| 389 | tx_state.storage_writes[address] = {} |
| 390 | tx_state.storage_writes[address][key] = value |
destroy_account ¶
Completely remove the account at address and all of its storage.
This function is made available exclusively for the SELFDESTRUCT
opcode. It is expected that SELFDESTRUCT will be disabled in a
future hardfork and this function will be removed. Only supports same
transaction destruction.
Parameters
tx_state : The transaction state. address : Address of account to destroy.
def destroy_account(tx_state: TransactionState, address: Address) -> None:
| 394 | <snip> |
|---|---|
| 410 | destroy_storage(tx_state, address) |
| 411 | set_account(tx_state, address, None) |
destroy_storage ¶
Completely remove the storage at address.
Drop this transaction's writes and record the address in
storage_clears so that reads no longer fall back to earlier
block writes or pre_state.
Parameters
tx_state : The transaction state. address : Address of account whose storage is to be deleted.
def destroy_storage(tx_state: TransactionState, address: Address) -> None:
| 415 | <snip> |
|---|---|
| 430 | if address in tx_state.storage_writes: |
| 431 | del tx_state.storage_writes[address] |
| 432 | tx_state.storage_clears.add(address) |
mark_account_created ¶
Mark an account as having been created in the current transaction.
This information is used by get_storage_original() to handle an
obscure edgecase, and to respect the constraints added to
SELFDESTRUCT by EIP-6780.
The marker is not removed even if the account creation reverts.
Since the account cannot have had code prior to its creation and
can't call get_storage_original(), this is harmless.
Parameters
tx_state : The transaction state. address : Address of the account that has been created.
def mark_account_created(tx_state: TransactionState, address: Address) -> None:
| 436 | <snip> |
|---|---|
| 454 | tx_state.created_accounts.add(address) |
set_transient_storage ¶
Set a value at a storage key on an account in transient storage.
Parameters
tx_state : The transaction state. address : Address of the account. key : Key to set. value : Value to set at the key.
modify_state ¶
Modify an Account in the state. If, after modification, the
account exists and has zero nonce, empty code, and zero balance, it
is destroyed.
def modify_state(tx_state: TransactionState, address: Address, f: Callable[[Account], None]) -> None:
| 489 | <snip> |
|---|---|
| 494 | set_account(tx_state, address, modify(get_account(tx_state, address), f)) |
| 495 | if account_exists_and_is_empty(tx_state, address): |
| 496 | destroy_account(tx_state, address) |
move_ether ¶
Move funds between accounts.
Parameters
tx_state : The transaction state. sender_address : Address of the sender. recipient_address : Address of the recipient. amount : The amount to transfer.
def move_ether(tx_state: TransactionState, sender_address: Address, recipient_address: Address, amount: U256) -> None:
| 505 | <snip> |
|---|---|
| 520 | |
| 521 | def reduce_sender_balance(sender: Account) -> None: |
| 522 | if sender.balance < amount: |
| 523 | raise AssertionError |
| 524 | sender.balance -= amount |
| 525 | |
| 526 | def increase_recipient_balance(recipient: Account) -> None: |
| 527 | recipient.balance += amount |
| 528 | |
| 529 | modify_state(tx_state, sender_address, reduce_sender_balance) |
| 530 | modify_state(tx_state, recipient_address, increase_recipient_balance) |
create_ether ¶
Add newly created ether to an account.
Parameters
tx_state : The transaction state. address : Address of the account to which ether is added. amount : The amount of ether to be added to the account of interest.
def create_ether(tx_state: TransactionState, address: Address, amount: U256) -> None:
| 536 | <snip> |
|---|---|
| 549 | |
| 550 | def increase_balance(account: Account) -> None: |
| 551 | account.balance += amount |
| 552 | |
| 553 | modify_state(tx_state, address, increase_balance) |
set_account_balance ¶
Set the balance of an account.
Parameters
tx_state : The transaction state. address : Address of the account whose balance needs to be set. amount : The amount that needs to be set in the balance.
def set_account_balance(tx_state: TransactionState, address: Address, amount: U256) -> None:
| 559 | <snip> |
|---|---|
| 572 | |
| 573 | def set_balance(account: Account) -> None: |
| 574 | account.balance = amount |
| 575 | |
| 576 | modify_state(tx_state, address, set_balance) |
increment_nonce ¶
Increment the nonce of an account.
Parameters
tx_state : The transaction state. address : Address of the account whose nonce needs to be incremented.
def increment_nonce(tx_state: TransactionState, address: Address) -> None:
| 580 | <snip> |
|---|---|
| 591 | |
| 592 | def increase_nonce(sender: Account) -> None: |
| 593 | sender.nonce += Uint(1) |
| 594 | |
| 595 | modify_state(tx_state, address, increase_nonce) |
set_code ¶
Set Account code.
Parameters
tx_state : The transaction state. address : Address of the account whose code needs to be updated. code : The bytecode that needs to be set.
def set_code(tx_state: TransactionState, address: Address, code: Bytes) -> None:
| 601 | <snip> |
|---|---|
| 614 | code_hash = keccak256(code) |
| 615 | if code_hash != EMPTY_CODE_HASH: |
| 616 | tx_state.code_writes[code_hash] = code |
| 617 | |
| 618 | def write_code_hash(sender: Account) -> None: |
| 619 | sender.code_hash = code_hash |
| 620 | |
| 621 | modify_state(tx_state, address, write_code_hash) |
copy_tx_state ¶
Create a snapshot of the transaction state for rollback.
Deep-copy writes and transient storage. The parent reference and
created_accounts are shared (not rolled back).
Parameters
tx_state : The transaction state to snapshot.
Returns
snapshot : TransactionState
A copy of the transaction state.
def copy_tx_state(tx_state: TransactionState) -> TransactionState:
| 628 | <snip> |
|---|---|
| 645 | return TransactionState( |
| 646 | parent=tx_state.parent, |
| 647 | account_writes=dict(tx_state.account_writes), |
| 648 | storage_writes={ |
| 649 | addr: dict(slots) |
| 650 | for addr, slots in tx_state.storage_writes.items() |
| 651 | }, |
| 652 | code_writes=dict(tx_state.code_writes), |
| 653 | created_accounts=tx_state.created_accounts, |
| 654 | storage_clears=set(tx_state.storage_clears), |
| 655 | transient_storage=dict(tx_state.transient_storage), |
| 656 | ) |
restore_tx_state ¶
Restore transaction state from a snapshot (rollback on failure).
Parameters
tx_state : The transaction state to restore. snapshot : The snapshot to restore from.
def restore_tx_state(tx_state: TransactionState, snapshot: TransactionState) -> None:
| 662 | <snip> |
|---|---|
| 673 | tx_state.account_writes = snapshot.account_writes |
| 674 | tx_state.storage_writes = snapshot.storage_writes |
| 675 | tx_state.code_writes = snapshot.code_writes |
| 676 | tx_state.storage_clears = snapshot.storage_clears |
| 677 | tx_state.transient_storage = snapshot.transient_storage |
incorporate_tx_into_block ¶
Merge transaction writes into the block state and clear for reuse.
Parameters
tx_state : The transaction state to commit.
def incorporate_tx_into_block(tx_state: TransactionState) -> None:
| 684 | <snip> |
|---|---|
| 693 | block = tx_state.parent |
| 694 | |
| 695 | for address, account in tx_state.account_writes.items(): |
| 696 | block.account_writes[address] = account |
| 697 | |
| 698 | for address in tx_state.storage_clears: |
| 699 | block.storage_clears.add(address) |
| 700 | block.storage_writes.pop(address, None) |
| 701 | |
| 702 | for address, slots in tx_state.storage_writes.items(): |
| 703 | if address not in block.storage_writes: |
| 704 | block.storage_writes[address] = {} |
| 705 | block.storage_writes[address].update(slots) |
| 706 | |
| 707 | block.code_writes.update(tx_state.code_writes) |
| 708 | |
| 709 | tx_state.account_writes.clear() |
| 710 | tx_state.storage_writes.clear() |
| 711 | tx_state.code_writes.clear() |
| 712 | tx_state.created_accounts.clear() |
| 713 | tx_state.storage_clears.clear() |
| 714 | tx_state.transient_storage.clear() |
extract_block_diff ¶
Extract account, storage, and code diff from the block state.
Parameters
block_state : The block state.
Returns
diff : BlockDiff
Account, storage, and code changes accumulated during block execution.