add address to metric

Author: avalonche

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

@@ -0,0 +1,90 @@
+use alloy_primitives::{hex, keccak256, Bytes};
+use jsonrpsee::proc_macros::rpc;
+use jsonrpsee_core::RpcResult;
+use std::sync::Arc;
+use tokio::sync::RwLock;
+
+use super::{
+    args::FlashtestationsArgs,
+    attestation::{AttestationConfig, get_attestation_provider},
+    service::load_or_generate_tee_key,
+};
+
+/// Admin API for op-rbuilder flashtestations
+#[rpc(server, namespace = "admin")]
+pub trait AdminApi {
+    /// Get the raw attestation quote (cached after first request)
+    #[method(name = "getAttestationQuote")]
+    async fn get_attestation_quote(&self) -> RpcResult<Option<String>>;
+}
+
+/// Admin RPC server implementation
+pub struct AdminRpcServer {
+    flashtestations_args: FlashtestationsArgs,
+    cached_quote: Arc<RwLock<Option<Vec<u8>>>>,
+}
+
+impl AdminRpcServer {
+    pub fn new(flashtestations_args: FlashtestationsArgs) -> Self {
+        Self {
+            flashtestations_args,
+            cached_quote: Arc::new(RwLock::new(None)),
+        }
+    }
+}
+
+#[async_trait::async_trait]
+impl AdminApiServer for AdminRpcServer {
+    async fn get_attestation_quote(&self) -> RpcResult<Option<String>> {
+        // Check if quote is already cached
+        {
+            let cache = self.cached_quote.read().await;
+            if let Some(quote) = cache.as_ref() {
+                return Ok(Some(hex::encode(quote)));
+            }
+        }
+
+        // Load TEE key using same logic as bootstrap
+        let tee_service_signer = match load_or_generate_tee_key(
+            &self.flashtestations_args.flashtestations_key_path,
+            self.flashtestations_args.debug,
+            &self.flashtestations_args.debug_tee_key_seed,
+        ) {
+            Ok(signer) => signer,
+            Err(e) => {
+                tracing::error!(error = %e, "Failed to load TEE key");
+                return Ok(None);
+            }
+        };
+
+        // Quote not cached, fetch it
+        let attestation_provider = get_attestation_provider(AttestationConfig {
+            debug: self.flashtestations_args.debug,
+            quote_provider: self.flashtestations_args.quote_provider.clone(),
+        });
+
+        // Prepare report data same as in bootstrap
+        let mut report_data = [0u8; 64];
+        let tee_address_bytes: [u8; 20] = tee_service_signer.address.into();
+        report_data[0..20].copy_from_slice(&tee_address_bytes);
+
+        // Use empty ext_data same as bootstrap
+        let ext_data = Bytes::from(b"");
+        let ext_data_hash = keccak256(ext_data.as_ref());
+        report_data[20..52].copy_from_slice(ext_data_hash.as_ref());
+
+        // Request attestation
+        match attestation_provider.get_attestation(report_data).await {
+            Ok(quote) => {
+                // Cache the quote for future requests
+                let mut cache = self.cached_quote.write().await;
+                *cache = Some(quote.clone());
+                Ok(Some(hex::encode(quote)))
+            }
+            Err(e) => {
+                tracing::error!(error = %e, "Failed to get attestation quote");
+                Ok(None)
+            }
+        }
+    }
+}

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

@@ -76,7 +76,7 @@ where
     let attestation = attestation_provider.get_attestation(report_data).await?;
 
     // Record TEE metrics (workload ID, MRTD, RTMR0)
-    record_tee_metrics(&attestation, tee_service_signer.address)?;
+    record_tee_metrics(&attestation, &tee_service_signer.address)?;
 
     // Use an external rpc when the builder is not the same as the builder actively building blocks onchain
     let registered = if let Some(rpc_url) = args.rpc_url {

crates/op-rbuilder/src/metrics.rs

@@ -1,3 +1,4 @@
+use alloy_primitives::{Address, hex};
 use metrics::IntoF64;
 use reth_metrics::{
     Metrics,
@@ -216,12 +217,13 @@ pub fn record_tee_metrics(raw_quote: &[u8], tee_address: &Address) -> eyre::Resu
     let mr_td_hex = hex::encode(parsed_quote.mr_td);
     let rt_mr0_hex = hex::encode(parsed_quote.rt_mr0);
 
+    let tee_address_static: &'static str = Box::leak(tee_address.to_string().into_boxed_str());
     let workload_id_static: &'static str = Box::leak(workload_id_hex.into_boxed_str());
     let mr_td_static: &'static str = Box::leak(mr_td_hex.into_boxed_str());
     let rt_mr0_static: &'static str = Box::leak(rt_mr0_hex.into_boxed_str());
 
     // Record TEE address
-    let tee_address_labels: [(&str, &str); 1] = [("tee_address", tee_address.to_string().as_str())];
+    let tee_address_labels: [(&str, &str); 1] = [("tee_address", tee_address_static)];
     gauge!("op_rbuilder_tee_address", &tee_address_labels).set(1);
 
     // Record workload ID