nimbus-eth1/nimbus/core/tx_pool/tx_packer.nim

341 lines
11 KiB
Nim

# Nimbus
# Copyright (c) 2018-2024 Status Research & Development GmbH
# Licensed under either of
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or
# http://www.apache.org/licenses/LICENSE-2.0)
# * MIT license ([LICENSE-MIT](LICENSE-MIT) or
# http://opensource.org/licenses/MIT)
# at your option. This file may not be copied, modified, or distributed except
# according to those terms.
## Transaction Pool Tasklets: Packer, VM execute and compact txs
## =============================================================
##
{.push raises: [].}
import
stew/sorted_set,
stew/byteutils,
../../db/ledger,
../../common/common,
../../utils/utils,
../../constants,
../../transaction/call_evm,
../../transaction,
../../evm/state,
../../evm/types,
../executor,
../validate,
../casper,
../eip4844,
../eip6110,
../eip7691,
./tx_desc,
./tx_item,
./tx_tabs
type
TxPacker = ref object
# Packer state
vmState: BaseVMState
numBlobPerBlock: int
# Packer results
blockValue: UInt256
stateRoot: Hash32
receiptsRoot: Hash32
logsBloom: Bloom
packedTxs: seq[TxItemRef]
withdrawalReqs: seq[byte]
consolidationReqs: seq[byte]
depositReqs: seq[byte]
const
receiptsExtensionSize = ##\
## Number of slots to extend the `receipts[]` at the same time.
20
ContinueWithNextAccount = true
StopCollecting = false
# ------------------------------------------------------------------------------
# Private helpers
# ------------------------------------------------------------------------------
proc classifyValidatePacked(vmState: BaseVMState; item: TxItemRef): bool =
## Verify the argument `item` against the accounts database. This function
## is a wrapper around the `verifyTransaction()` call to be used in a similar
## fashion as in `asyncProcessTransactionImpl()`.
let
roDB = vmState.readOnlyLedger
baseFee = vmState.blockCtx.baseFeePerGas.get(0.u256)
fork = vmState.fork
gasLimit = vmState.blockCtx.gasLimit
excessBlobGas = calcExcessBlobGas(vmState.parent, vmState.fork >= FkPrague)
roDB.validateTransaction(
item.tx, item.sender, gasLimit, baseFee, excessBlobGas, fork).isOk
proc classifyPacked(vmState: BaseVMState; moreBurned: GasInt): bool =
## Classifier for *packing* (i.e. adding up `gasUsed` values after executing
## in the VM.) This function checks whether the sum of the arguments
## `gasBurned` and `moreGasBurned` is within acceptable constraints.
let totalGasUsed = vmState.cumulativeGasUsed + moreBurned
totalGasUsed < vmState.blockCtx.gasLimit
proc classifyPackedNext(vmState: BaseVMState): bool =
## Classifier for *packing* (i.e. adding up `gasUsed` values after executing
## in the VM.) This function returns `true` if the packing level is still
## low enough to proceed trying to accumulate more items.
##
## This function is typically called as a follow up after a `false` return of
## `classifyPack()`.
vmState.cumulativeGasUsed < vmState.blockCtx.gasLimit
func baseFee(pst: TxPacker): GasInt =
## Getter, baseFee for the next bock header. This value is auto-generated
## when a new insertion point is set via `head=`.
if pst.vmState.blockCtx.baseFeePerGas.isSome:
pst.vmState.blockCtx.baseFeePerGas.get.truncate(GasInt)
else:
0.GasInt
func feeRecipient(pst: TxPacker): Address =
pst.vmState.com.pos.feeRecipient
# ------------------------------------------------------------------------------
# Private functions
# ------------------------------------------------------------------------------
proc runTx(pst: var TxPacker; item: TxItemRef): GasInt =
## Execute item transaction and update `vmState` book keeping. Returns the
## `gasUsed` after executing the transaction.
let gasUsed = item.tx.txCallEvm(item.sender, pst.vmState, pst.baseFee)
doAssert 0 <= gasUsed
gasUsed
proc runTxCommit(pst: var TxPacker; item: TxItemRef; gasBurned: GasInt) =
## Book keeping after executing argument `item` transaction in the VM. The
## function returns the next number of items `nItems+1`.
let
vmState = pst.vmState
inx = pst.packedTxs.len
gasTip = item.tx.tip(pst.baseFee)
let reward = gasBurned.u256 * gasTip.u256
vmState.ledger.addBalance(pst.feeRecipient, reward)
pst.blockValue += reward
# Update receipts sequence
if vmState.receipts.len <= inx:
vmState.receipts.setLen(inx + receiptsExtensionSize)
# Return remaining gas to the block gas counter so it is
# available for the next transaction.
vmState.gasPool += item.tx.gasLimit - gasBurned
# gasUsed accounting
vmState.cumulativeGasUsed += gasBurned
vmState.receipts[inx] = vmState.makeReceipt(item.tx.txType)
pst.packedTxs.add item
# ------------------------------------------------------------------------------
# Private functions: packer packerVmExec() helpers
# ------------------------------------------------------------------------------
proc vmExecInit(xp: TxPoolRef): Result[TxPacker, string] =
let packer = TxPacker(
vmState: xp.vmState,
numBlobPerBlock: 0,
blockValue: 0.u256,
stateRoot: xp.vmState.parent.stateRoot,
)
# EIP-4788
if xp.nextFork >= FkCancun:
let beaconRoot = xp.vmState.com.pos.parentBeaconBlockRoot
xp.vmState.processBeaconBlockRoot(beaconRoot).isOkOr:
return err(error)
# EIP-2935
if xp.nextFork >= FkPrague:
xp.vmState.processParentBlockHash(xp.vmState.blockCtx.parentHash).isOkOr:
return err(error)
ok(packer)
proc vmExecGrabItem(pst: var TxPacker; item: TxItemRef): bool =
## Greedily collect & compact items as long as the accumulated `gasLimit`
## values are below the maximum block size.
let
vmState = pst.vmState
electra = vmState.fork >= FkPrague
# EIP-4844
let maxBlobsPerBlock = getMaxBlobsPerBlock(electra)
if (pst.numBlobPerBlock + item.tx.versionedHashes.len).uint64 > maxBlobsPerBlock:
return ContinueWithNextAccount
let
blobGasUsed = item.tx.getTotalBlobGas
maxBlobGasPerBlock = getMaxBlobGasPerBlock(electra)
if vmState.blobGasUsed + blobGasUsed > maxBlobGasPerBlock:
return ContinueWithNextAccount
# Verify we have enough gas in gasPool
if vmState.gasPool < item.tx.gasLimit:
# skip this transaction and
# continue with next account
# if we don't have enough gas
return ContinueWithNextAccount
# Validate transaction relative to the current vmState
if not vmState.classifyValidatePacked(item):
return ContinueWithNextAccount
# EIP-1153
vmState.ledger.clearTransientStorage()
# Execute EVM for this transaction
let
accTx = vmState.ledger.beginSavepoint
gasUsed = pst.runTx(item)
# Find out what to do next: accepting this tx or trying the next account
if not vmState.classifyPacked(gasUsed):
vmState.ledger.rollback(accTx)
if vmState.classifyPackedNext():
return ContinueWithNextAccount
return StopCollecting
# Commit ledger changes
vmState.ledger.commit(accTx)
vmState.ledger.persist(clearEmptyAccount = vmState.fork >= FkSpurious)
# Finish book-keeping
pst.runTxCommit(item, gasUsed)
pst.numBlobPerBlock += item.tx.versionedHashes.len
vmState.blobGasUsed += blobGasUsed
vmState.gasPool -= item.tx.gasLimit
ContinueWithNextAccount
proc vmExecCommit(pst: var TxPacker): Result[void, string] =
let
vmState = pst.vmState
ledger = vmState.ledger
# EIP-4895
if vmState.fork >= FkShanghai:
for withdrawal in vmState.com.pos.withdrawals:
ledger.addBalance(withdrawal.address, withdrawal.weiAmount)
# EIP-6110, EIP-7002, EIP-7251
if vmState.fork >= FkPrague:
pst.withdrawalReqs = processDequeueWithdrawalRequests(vmState)
pst.consolidationReqs = processDequeueConsolidationRequests(vmState)
pst.depositReqs = ?parseDepositLogs(vmState.allLogs, vmState.com.depositContractAddress)
# Finish up, then vmState.ledger.stateRoot may be accessed
ledger.persist(clearEmptyAccount = vmState.fork >= FkSpurious)
# Update flexi-array, set proper length
vmState.receipts.setLen(pst.packedTxs.len)
pst.receiptsRoot = vmState.receipts.calcReceiptsRoot
pst.logsBloom = vmState.receipts.createBloom
pst.stateRoot = vmState.ledger.getStateRoot()
ok()
# ------------------------------------------------------------------------------
# Public functions
# ------------------------------------------------------------------------------
proc packerVmExec*(xp: TxPoolRef): Result[TxPacker, string] =
## Execute as much transactions as possible.
let db = xp.vmState.com.db
let dbTx = db.ctx.txFrameBegin()
defer: dbTx.dispose()
var pst = xp.vmExecInit.valueOr:
return err(error)
for item in xp.byPriceAndNonce:
let rc = pst.vmExecGrabItem(item)
if rc == StopCollecting:
break
?pst.vmExecCommit()
ok(pst)
func getExtraData(com: CommonRef): seq[byte] =
if com.extraData.len > 32:
com.extraData.toBytes[0..<32]
else:
com.extraData.toBytes
proc assembleHeader*(pst: TxPacker): Header =
## Generate a new header, a child of the cached `head`
let
vmState = pst.vmState
com = vmState.com
pos = com.pos
result = Header(
parentHash: vmState.blockCtx.parentHash,
ommersHash: EMPTY_UNCLE_HASH,
coinbase: pos.feeRecipient,
stateRoot: pst.stateRoot,
receiptsRoot: pst.receiptsRoot,
logsBloom: pst.logsBloom,
difficulty: UInt256.zero(),
number: vmState.blockNumber,
gasLimit: vmState.blockCtx.gasLimit,
gasUsed: vmState.cumulativeGasUsed,
timestamp: pos.timestamp,
extraData: getExtraData(com),
mixHash: pos.prevRandao,
nonce: default(Bytes8),
baseFeePerGas: vmState.blockCtx.baseFeePerGas,
)
if com.isShanghaiOrLater(pos.timestamp):
result.withdrawalsRoot = Opt.some(calcWithdrawalsRoot(pos.withdrawals))
if com.isCancunOrLater(pos.timestamp):
result.parentBeaconBlockRoot = Opt.some(pos.parentBeaconBlockRoot)
result.blobGasUsed = Opt.some vmState.blobGasUsed
result.excessBlobGas = Opt.some vmState.blockCtx.excessBlobGas
if com.isPragueOrLater(pos.timestamp):
let requestsHash = calcRequestsHash([
(DEPOSIT_REQUEST_TYPE, pst.depositReqs),
(WITHDRAWAL_REQUEST_TYPE, pst.withdrawalReqs),
(CONSOLIDATION_REQUEST_TYPE, pst.consolidationReqs)
])
result.requestsHash = Opt.some(requestsHash)
func blockValue*(pst: TxPacker): UInt256 =
pst.blockValue
func executionRequests*(pst: var TxPacker): seq[seq[byte]] =
template append(dst, reqType, reqData) =
if reqData.len > 0:
reqData.insert(reqType)
dst.add(move(reqData))
result.append(DEPOSIT_REQUEST_TYPE, pst.depositReqs)
result.append(WITHDRAWAL_REQUEST_TYPE, pst.withdrawalReqs)
result.append(CONSOLIDATION_REQUEST_TYPE, pst.consolidationReqs)
iterator packedTxs*(pst: TxPacker): TxItemRef =
for item in pst.packedTxs:
yield item
# ------------------------------------------------------------------------------
# End
# ------------------------------------------------------------------------------