Add flashtestations builder transaction

Author: avalonche

Cargo.lock

@@ -126,6 +126,7 @@ http = "1.0"
 sha3 = "0.10"
 hex = "0.4"
 ureq = "2.10"
+k256 = "0.13.4"
 
 # TODO: change to rev from main once https://github.com/flashbots/rollup-boost/pull/401 is merged
 rollup-boost = { git = "http://github.com/flashbots/rollup-boost", rev = "0c1fb4ce7e00f0afc350f5bf7573b19da6d485ec" }

crates/op-rbuilder/Cargo.toml

@@ -2,7 +2,7 @@ use alloy_consensus::{Eip658Value, Transaction, conditional::BlockConditionalAtt
 use alloy_eips::Typed2718;
 use alloy_evm::Database;
 use alloy_op_evm::block::receipt_builder::OpReceiptBuilder;
-use alloy_primitives::{Bytes, U256};
+use alloy_primitives::{BlockHash, Bytes, U256};
 use alloy_rpc_types_eth::Withdrawals;
 use core::fmt::Debug;
 use op_alloy_consensus::OpDepositReceipt;
@@ -75,41 +75,51 @@ pub struct OpPayloadBuilderCtx<ExtraCtx: Debug + Default = ()> {
 
 impl<ExtraCtx: Debug + Default> OpPayloadBuilderCtx<ExtraCtx> {
     /// Returns the parent block the payload will be build on.
-    pub(super) fn parent(&self) -> &SealedHeader {
+    pub(crate) fn parent(&self) -> &SealedHeader {
         &self.config.parent_header
     }
 
+    /// Returns the parent hash
+    pub(crate) fn parent_hash(&self) -> BlockHash {
+        self.parent().hash()
+    }
+
+    /// Returns the timestamp
+    pub(crate) fn timestamp(&self) -> u64 {
+        self.attributes().timestamp()
+    }
+
     /// Returns the builder attributes.
-    pub(super) const fn attributes(&self) -> &OpPayloadBuilderAttributes<OpTransactionSigned> {
+    pub(crate) const fn attributes(&self) -> &OpPayloadBuilderAttributes<OpTransactionSigned> {
         &self.config.attributes
     }
 
     /// Returns the withdrawals if shanghai is active.
-    pub(super) fn withdrawals(&self) -> Option<&Withdrawals> {
+    pub(crate) fn withdrawals(&self) -> Option<&Withdrawals> {
         self.chain_spec
             .is_shanghai_active_at_timestamp(self.attributes().timestamp())
             .then(|| &self.attributes().payload_attributes.withdrawals)
     }
 
     /// Returns the block gas limit to target.
-    pub(super) fn block_gas_limit(&self) -> u64 {
+    pub(crate) fn block_gas_limit(&self) -> u64 {
         self.attributes()
             .gas_limit
             .unwrap_or(self.evm_env.block_env.gas_limit)
     }
 
     /// Returns the block number for the block.
-    pub(super) fn block_number(&self) -> u64 {
+    pub(crate) fn block_number(&self) -> u64 {
         as_u64_saturated!(self.evm_env.block_env.number)
     }
 
     /// Returns the current base fee
-    pub(super) fn base_fee(&self) -> u64 {
+    pub(crate) fn base_fee(&self) -> u64 {
         self.evm_env.block_env.basefee
     }
 
     /// Returns the current blob gas price.
-    pub(super) fn get_blob_gasprice(&self) -> Option<u64> {
+    pub(crate) fn get_blob_gasprice(&self) -> Option<u64> {
         self.evm_env
             .block_env
             .blob_gasprice()
@@ -119,7 +129,7 @@ impl<ExtraCtx: Debug + Default> OpPayloadBuilderCtx<ExtraCtx> {
     /// Returns the blob fields for the header.
     ///
     /// This will always return `Some(0)` after ecotone.
-    pub(super) fn blob_fields(&self) -> (Option<u64>, Option<u64>) {
+    pub(crate) fn blob_fields(&self) -> (Option<u64>, Option<u64>) {
         // OP doesn't support blobs/EIP-4844.
         // https://specs.optimism.io/protocol/exec-engine.html#ecotone-disable-blob-transactions
         // Need [Some] or [None] based on hardfork to match block hash.
@@ -133,7 +143,7 @@ impl<ExtraCtx: Debug + Default> OpPayloadBuilderCtx<ExtraCtx> {
     /// Returns the extra data for the block.
     ///
     /// After holocene this extracts the extradata from the paylpad
-    pub(super) fn extra_data(&self) -> Result<Bytes, PayloadBuilderError> {
+    pub(crate) fn extra_data(&self) -> Result<Bytes, PayloadBuilderError> {
         if self.is_holocene_active() {
             self.attributes()
                 .get_holocene_extra_data(
@@ -148,47 +158,47 @@ impl<ExtraCtx: Debug + Default> OpPayloadBuilderCtx<ExtraCtx> {
     }
 
     /// Returns the current fee settings for transactions from the mempool
-    pub(super) fn best_transaction_attributes(&self) -> BestTransactionsAttributes {
+    pub(crate) fn best_transaction_attributes(&self) -> BestTransactionsAttributes {
         BestTransactionsAttributes::new(self.base_fee(), self.get_blob_gasprice())
     }
 
     /// Returns the unique id for this payload job.
-    pub(super) fn payload_id(&self) -> PayloadId {
+    pub(crate) fn payload_id(&self) -> PayloadId {
         self.attributes().payload_id()
     }
 
     /// Returns true if regolith is active for the payload.
-    pub(super) fn is_regolith_active(&self) -> bool {
+    pub(crate) fn is_regolith_active(&self) -> bool {
         self.chain_spec
             .is_regolith_active_at_timestamp(self.attributes().timestamp())
     }
 
     /// Returns true if ecotone is active for the payload.
-    pub(super) fn is_ecotone_active(&self) -> bool {
+    pub(crate) fn is_ecotone_active(&self) -> bool {
         self.chain_spec
             .is_ecotone_active_at_timestamp(self.attributes().timestamp())
     }
 
     /// Returns true if canyon is active for the payload.
-    pub(super) fn is_canyon_active(&self) -> bool {
+    pub(crate) fn is_canyon_active(&self) -> bool {
         self.chain_spec
             .is_canyon_active_at_timestamp(self.attributes().timestamp())
     }
 
     /// Returns true if holocene is active for the payload.
-    pub(super) fn is_holocene_active(&self) -> bool {
+    pub(crate) fn is_holocene_active(&self) -> bool {
         self.chain_spec
             .is_holocene_active_at_timestamp(self.attributes().timestamp())
     }
 
     /// Returns true if isthmus is active for the payload.
-    pub(super) fn is_isthmus_active(&self) -> bool {
+    pub(crate) fn is_isthmus_active(&self) -> bool {
         self.chain_spec
             .is_isthmus_active_at_timestamp(self.attributes().timestamp())
     }
 
     /// Returns the chain id
-    pub(super) fn chain_id(&self) -> u64 {
+    pub(crate) fn chain_id(&self) -> u64 {
         self.chain_spec.chain_id()
     }
 }

crates/op-rbuilder/src/builders/context.rs

@@ -14,7 +14,7 @@ use crate::{
         BuilderTransactionCtx, BuilderTransactionError, BuilderTransactions,
         context::OpPayloadBuilderCtx, flashblocks::payload::FlashblocksExtraCtx,
     },
-    flashtestations::service::FlashtestationsBuilderTx,
+    flashtestations::builder_tx::FlashtestationsBuilderTx,
     primitives::reth::ExecutionInfo,
     tx_signer::Signer,
 };

crates/op-rbuilder/src/builders/flashblocks/builder_tx.rs

@@ -14,7 +14,7 @@ use crate::{
         BuilderTransactionCtx, BuilderTransactionError, BuilderTransactions,
         context::OpPayloadBuilderCtx,
     },
-    flashtestations::service::FlashtestationsBuilderTx,
+    flashtestations::builder_tx::FlashtestationsBuilderTx,
     primitives::reth::ExecutionInfo,
     tx_signer::Signer,
 };

crates/op-rbuilder/src/builders/standard/builder_tx.rs

@@ -23,17 +23,24 @@ pub struct FlashtestationsArgs {
     )]
     pub debug: bool,
 
-    // Debug url for attestations
-    #[arg(long = "flashtestations.debug-url", env = "FLASHTESTATIONS_DEBUG_URL")]
-    pub debug_url: Option<String>,
+    // Debug static key for the tee key
+    #[arg(
+        long = "flashtestations.debug-tee-key-seed",
+        env = "FLASHTESTATIONS_DEBUG_TEE_KEY_SEED",
+        default_value = "debug"
+    )]
+    pub debug_tee_key_seed: String,
 
-    /// The rpc url to post the onchain attestation requests to
+    // Remote url for attestations
     #[arg(
-        long = "flashtestations.rpc-url",
-        env = "FLASHTESTATIONS_RPC_URL",
-        default_value = "http://localhost:8545"
+        long = "flashtestations.quote-provider",
+        env = "FLASHTESTATIONS_QUOTE_PROVIDER"
     )]
-    pub rpc_url: String,
+    pub quote_provider: Option<String>,
+
+    /// The rpc url to post the onchain attestation requests to
+    #[arg(long = "flashtestations.rpc-url", env = "FLASHTESTATIONS_RPC_URL")]
+    pub rpc_url: Option<String>,
 
     /// Funding key for the TEE key
     #[arg(

crates/op-rbuilder/src/flashtestations/args.rs

@@ -9,27 +9,27 @@ const DEBUG_QUOTE_SERVICE_URL: &str = "http://ns31695324.ip-141-94-163.eu:10080/
 pub struct AttestationConfig {
     /// If true, uses the debug HTTP service instead of real TDX hardware
     pub debug: bool,
-    /// The URL of the debug HTTP service
-    pub debug_url: Option<String>,
+    /// The URL of the quote provider
+    pub quote_provider: Option<String>,
 }
 
 /// Trait for attestation providers
 pub trait AttestationProvider {
     fn get_attestation(&self, report_data: [u8; 64]) -> eyre::Result<Vec<u8>>;
 }
 
-/// Debug HTTP service attestation provider
-pub struct DebugAttestationProvider {
+/// Remote attestation provider
+pub struct RemoteAttestationProvider {
     service_url: String,
 }
 
-impl DebugAttestationProvider {
+impl RemoteAttestationProvider {
     pub fn new(service_url: String) -> Self {
         Self { service_url }
     }
 }
 
-impl AttestationProvider for DebugAttestationProvider {
+impl AttestationProvider for RemoteAttestationProvider {
     fn get_attestation(&self, report_data: [u8; 64]) -> eyre::Result<Vec<u8>> {
         let report_data_hex = hex::encode(report_data);
         let url = format!("{}/{}", self.service_url, report_data_hex);
@@ -47,19 +47,21 @@ impl AttestationProvider for DebugAttestationProvider {
     }
 }
 
+#[allow(clippy::if_same_then_else)]
 pub fn get_attestation_provider(
     config: AttestationConfig,
 ) -> Box<dyn AttestationProvider + Send + Sync> {
     if config.debug {
-        Box::new(DebugAttestationProvider::new(
+        Box::new(RemoteAttestationProvider::new(
             config
-                .debug_url
+                .quote_provider
                 .unwrap_or(DEBUG_QUOTE_SERVICE_URL.to_string()),
         ))
     } else {
-        // TODO: replace with real attestation provider
-        Box::new(DebugAttestationProvider::new(
-            DEBUG_QUOTE_SERVICE_URL.to_string(),
+        Box::new(RemoteAttestationProvider::new(
+            config
+                .quote_provider
+                .unwrap_or(DEBUG_QUOTE_SERVICE_URL.to_string()), // TODO: remove this once we have a real quote provider
         ))
     }
 }