Add flashtestations builder transaction trait to payload builder

Author: avalonche

Cargo.lock

@@ -121,8 +121,7 @@ hex = "0.4"
 ureq = "2.10"
 
 rollup-boost = { git = "http://github.com/flashbots/rollup-boost", branch = "main" }
-tdx = { git = "https://github.com/automata-network/tdx-attestation-sdk.git", optional = true }
-dcap-rs = { git = "https://github.com/automata-network/dcap-rs.git", optional = true }
+tdx = { git = "https://github.com/automata-network/tdx-attestation-sdk.git", rev = "ce02e0009c20bb4fc37c4c95701925fc3f139b41", optional = true }
 
 dashmap = { version = "6.1", optional = true }
 nanoid = { version = "0.4", optional = true }
@@ -188,7 +187,7 @@ testing = [
 
 interop = []
 
-flashtestations = ["dcap-rs", "tdx"]
+flashtestations = ["tdx"]
 
 telemetry = ["reth-tracing-otlp", "opentelemetry"]
 

crates/op-rbuilder/Cargo.toml

@@ -1,12 +1,133 @@
+use alloy_consensus::TxEip1559;
+use alloy_eips::{eip7623::TOTAL_COST_FLOOR_PER_TOKEN, Encodable2718};
+use alloy_primitives::{
+    map::foldhash::{HashSet, HashSetExt},
+    Address, TxKind,
+};
+use core::fmt::Debug;
+use op_alloy_consensus::OpTypedTransaction;
+use reth_evm::{eth::receipt_builder::ReceiptBuilderCtx, ConfigureEvm, Evm};
+use reth_node_api::PayloadBuilderError;
 use reth_optimism_primitives::OpTransactionSigned;
 use reth_primitives::Recovered;
+use reth_provider::ProviderError;
+use reth_revm::State;
+use revm::{context::result::ResultAndState, Database, DatabaseCommit};
+use tracing::{debug, warn};
 
-use crate::tx_signer::Signer;
+use crate::{
+    builders::context::OpPayloadBuilderCtx, primitives::reth::ExecutionInfo, tx_signer::Signer,
+};
 
-pub trait BuilderTx {
-    fn estimated_builder_tx_gas(&self) -> u64;
-    fn estimated_builder_tx_da_size(&self) -> Option<u64>;
-    fn signed_builder_tx(&self) -> Result<Recovered<OpTransactionSigned>, secp256k1::Error>;
+pub struct BuilderTransactionCtx {
+    pub gas_used: u64,
+    pub da_size: u64,
+    pub signed_tx: Recovered<OpTransactionSigned>,
+}
+
+/// Possible error variants during construction of builder txs.
+#[derive(Debug, thiserror::Error)]
+pub enum BuilderTransactionError {
+    /// Thrown when builder account load fails to get builder nonce
+    #[error("failed to load account {0}")]
+    AccountLoadFailed(Address),
+    /// Thrown when signature signing fails
+    #[error("failed to sign transaction: {0}")]
+    SigningError(secp256k1::Error),
+    /// Unrecoverable error during evm execution.
+    #[error("evm execution error {0}")]
+    EvmExecutionError(Box<dyn core::error::Error + Send + Sync>),
+    /// Any other builder transaction errors.
+    #[error(transparent)]
+    Other(Box<dyn core::error::Error + Send + Sync>),
+}
+
+impl From<secp256k1::Error> for BuilderTransactionError {
+    fn from(error: secp256k1::Error) -> Self {
+        BuilderTransactionError::SigningError(error)
+    }
+}
+
+impl From<BuilderTransactionError> for PayloadBuilderError {
+    fn from(error: BuilderTransactionError) -> Self {
+        match error {
+            BuilderTransactionError::EvmExecutionError(e) => {
+                PayloadBuilderError::EvmExecutionError(e)
+            }
+            _ => PayloadBuilderError::Other(Box::new(error)),
+        }
+    }
+}
+
+pub trait BuilderTransactions {
+    fn simulate_builder_txs<DB, Extra: Debug + Default>(
+        &self,
+        info: &mut ExecutionInfo<Extra>,
+        ctx: &OpPayloadBuilderCtx,
+        db: &mut State<DB>,
+    ) -> Result<Vec<BuilderTransactionCtx>, BuilderTransactionError>
+    where
+        DB: Database<Error = ProviderError>;
+
+    fn add_builder_txs<DB, Extra: Debug + Default>(
+        &self,
+        info: &mut ExecutionInfo<Extra>,
+        builder_ctx: &OpPayloadBuilderCtx,
+        db: &mut State<DB>,
+    ) -> Result<(), BuilderTransactionError>
+    where
+        DB: Database<Error = ProviderError>,
+    {
+        {
+            let mut evm = builder_ctx
+                .evm_config
+                .evm_with_env(&mut *db, builder_ctx.evm_env.clone());
+            let mut invalid: HashSet<Address> = HashSet::new();
+            let builder_txs = self.simulate_builder_txs(info, builder_ctx, evm.db_mut())?;
+            for builder_tx in builder_txs {
+                if invalid.contains(&builder_tx.signed_tx.signer()) {
+                    debug!(target: "payload_builder", tx_hash = ?builder_tx.signed_tx.tx_hash(), "builder signer invalid as previous builder tx reverted");
+                    continue;
+                }
+
+                let ResultAndState { result, state } = evm
+                    .transact(&builder_tx.signed_tx)
+                    .map_err(|err| BuilderTransactionError::EvmExecutionError(Box::new(err)))?;
+
+                if !result.is_success() {
+                    warn!(target: "payload_builder", tx_hash = ?builder_tx.signed_tx.tx_hash(), "builder tx reverted");
+                    invalid.insert(builder_tx.signed_tx.signer());
+                    continue;
+                }
+
+                // Add gas used by the transaction to cumulative gas used, before creating the receipt
+                let gas_used = result.gas_used();
+                info.cumulative_gas_used += gas_used;
+
+                let ctx = ReceiptBuilderCtx {
+                    tx: builder_tx.signed_tx.inner(),
+                    evm: &evm,
+                    result,
+                    state: &state,
+                    cumulative_gas_used: info.cumulative_gas_used,
+                };
+                info.receipts.push(builder_ctx.build_receipt(ctx, None));
+
+                // Commit changes
+                evm.db_mut().commit(state);
+
+                // Append sender and transaction to the respective lists
+                info.executed_senders.push(builder_tx.signed_tx.signer());
+                info.executed_transactions
+                    .push(builder_tx.signed_tx.into_inner());
+            }
+
+            // Release the db reference by dropping evm
+            drop(evm);
+
+            Ok(())
+        }
+    }
 }
 
 // Scaffolding for how to construct the end of block builder transaction
@@ -17,16 +138,94 @@ pub struct StandardBuilderTx {
     pub signer: Option<Signer>,
 }
 
-impl BuilderTx for StandardBuilderTx {
-    fn estimated_builder_tx_gas(&self) -> u64 {
-        todo!()
+impl StandardBuilderTx {
+    pub fn new(signer: Option<Signer>) -> Self {
+        Self { signer }
+    }
+
+    fn estimate_builder_tx_gas(&self, input: &[u8]) -> u64 {
+        // Count zero and non-zero bytes
+        let (zero_bytes, nonzero_bytes) = input.iter().fold((0, 0), |(zeros, nonzeros), &byte| {
+            if byte == 0 {
+                (zeros + 1, nonzeros)
+            } else {
+                (zeros, nonzeros + 1)
+            }
+        });
+
+        // Calculate gas cost (4 gas per zero byte, 16 gas per non-zero byte)
+        let zero_cost = zero_bytes * 4;
+        let nonzero_cost = nonzero_bytes * 16;
+
+        // Tx gas should be not less than floor gas https://eips.ethereum.org/EIPS/eip-7623
+        let tokens_in_calldata = zero_bytes + nonzero_bytes * 4;
+        let floor_gas = 21_000 + tokens_in_calldata * TOTAL_COST_FLOOR_PER_TOKEN;
+
+        std::cmp::max(zero_cost + nonzero_cost + 21_000, floor_gas)
     }
 
-    fn estimated_builder_tx_da_size(&self) -> Option<u64> {
-        todo!()
+    fn signed_builder_tx<DB>(
+        &self,
+        ctx: &OpPayloadBuilderCtx,
+        db: &mut State<DB>,
+        signer: Signer,
+        gas_used: u64,
+        message: Vec<u8>,
+    ) -> Result<Recovered<OpTransactionSigned>, BuilderTransactionError>
+    where
+        DB: Database<Error = ProviderError>,
+    {
+        let nonce = db
+            .load_cache_account(signer.address)
+            .map(|acc| acc.account_info().unwrap_or_default().nonce)
+            .map_err(|_| BuilderTransactionError::AccountLoadFailed(signer.address))?;
+
+        // Create the EIP-1559 transaction
+        let tx = OpTypedTransaction::Eip1559(TxEip1559 {
+            chain_id: ctx.chain_id(),
+            nonce,
+            gas_limit: gas_used,
+            max_fee_per_gas: ctx.base_fee().into(),
+            max_priority_fee_per_gas: 0,
+            to: TxKind::Call(Address::ZERO),
+            // Include the message as part of the transaction data
+            input: message.into(),
+            ..Default::default()
+        });
+        // Sign the transaction
+        let builder_tx = signer
+            .sign_tx(tx)
+            .map_err(BuilderTransactionError::SigningError)?;
+
+        Ok(builder_tx)
     }
+}
 
-    fn signed_builder_tx(&self) -> Result<Recovered<OpTransactionSigned>, secp256k1::Error> {
-        todo!()
+impl BuilderTransactions for StandardBuilderTx {
+    fn simulate_builder_txs<DB, Extra: Debug + Default>(
+        &self,
+        _info: &mut ExecutionInfo<Extra>,
+        ctx: &OpPayloadBuilderCtx,
+        db: &mut State<DB>,
+    ) -> Result<Vec<BuilderTransactionCtx>, BuilderTransactionError>
+    where
+        DB: Database<Error = ProviderError>,
+    {
+        match self.signer {
+            Some(signer) => {
+                let message: Vec<u8> = format!("Block Number: {}", ctx.block_number()).into_bytes();
+                let gas_used = self.estimate_builder_tx_gas(&message);
+                let signed_tx = self.signed_builder_tx(ctx, db, signer, gas_used, message)?;
+                let da_size = op_alloy_flz::tx_estimated_size_fjord_bytes(
+                    signed_tx.encoded_2718().as_slice(),
+                );
+                Ok(vec![BuilderTransactionCtx {
+                    gas_used,
+                    da_size,
+                    signed_tx,
+                }])
+            }
+            None => Ok(vec![]),
+        }
     }
 }