def diff_state( expected_state: AccountState, state: BaseState) -> AccountDiff: for account, account_data in sorted(expected_state.items()): expected_balance = account_data['balance'] expected_nonce = account_data['nonce'] expected_code = account_data['code'] actual_nonce = state.get_nonce(account) actual_code = state.get_code(account) actual_balance = state.get_balance(account) if actual_nonce != expected_nonce: yield (account, 'nonce', actual_nonce, expected_nonce) if actual_code != expected_code: yield (account, 'code', actual_code, expected_code) if actual_balance != expected_balance: yield (account, 'balance', actual_balance, expected_balance) for slot, expected_storage_value in sorted(account_data['storage'].items()): actual_storage_value = state.get_storage(account, slot) if actual_storage_value != expected_storage_value: yield ( account, f'storage[{slot}]', actual_storage_value, expected_storage_value, )
def setup_state(desired_state: AccountState, state: BaseState) -> None: for account, account_data in desired_state.items(): for slot, value in account_data['storage'].items(): state.set_storage(account, slot, value) nonce = account_data['nonce'] code = account_data['code'] balance = account_data['balance'] state.set_nonce(account, nonce) state.set_code(account, code) state.set_balance(account, balance) state.persist()
def _get_computation_error(state: BaseState, transaction: SpoofTransaction) -> Optional[VMError]: snapshot = state.snapshot() try: computation = state.execute_transaction(transaction) if computation.is_error: return computation._error else: return None finally: state.revert(snapshot)
def apply_state_dict(state: BaseState, state_dict: AccountState) -> None: for account, account_data in state_dict.items(): state.set_balance(account, account_data["balance"]) state.set_nonce(account, account_data["nonce"]) state.set_code(account, account_data["code"]) for slot, value in account_data["storage"].items(): state.set_storage(account, slot, value)
def validate_frontier_transaction(state: BaseState, transaction: BaseOrSpoofTransaction) -> None: gas_cost = transaction.gas * transaction.gas_price sender_balance = state.get_balance(transaction.sender) if sender_balance < gas_cost: raise ValidationError( "Sender account balance cannot afford txn gas: `{0}`".format( transaction.sender)) total_cost = transaction.value + gas_cost if sender_balance < total_cost: raise ValidationError("Sender account balance cannot afford txn") if state.get_nonce(transaction.sender) != transaction.nonce: raise ValidationError("Invalid transaction nonce")
def make_receipt( base_header: BlockHeader, transaction: BaseTransaction, computation: BaseComputation, state: BaseState) -> Receipt: receipt_without_state_root = make_frontier_receipt(base_header, transaction, computation) return receipt_without_state_root.copy( state_root=state.make_state_root() )