debug_backtraceAt

crates/rpc/rpc-api/src/debug.rs

@@ -123,4 +123,10 @@ pub trait DebugApi {
         opts: Option<GethDebugTracingOptions>,
         state_override: Option<StateOverride>,
     ) -> RpcResult<Vec<GethTrace>>;
+
+    ///Sets the logging backtrace location. When a backtrace location is set and a log message is
+    /// emitted at that location,  the stack of the goroutine executing the log statement will
+    /// be printed to stderr.
+    #[method(name = "backtraceAt")]
+    async fn debug_backtrace_at(&self, location: &str) -> RpcResult<()>;
 }

crates/rpc/rpc/Cargo.toml

@@ -25,6 +25,7 @@ reth-tasks.workspace = true
 reth-metrics.workspace = true
 reth-consensus-common = { path = "../../consensus/common" }
 reth-rpc-types-compat.workspace = true
+lazy_static = "*"
 
 # eth
 revm = { workspace = true, features = [

crates/rpc/rpc/src/debug.rs

@@ -45,6 +45,12 @@ use std::sync::Arc;
 use tokio::sync::{mpsc, AcquireError, OwnedSemaphorePermit};
 use tokio_stream::{wrappers::ReceiverStream, StreamExt};
 
+use lazy_static::lazy_static;
+use std::sync::Mutex;
+lazy_static! {
+    static ref LOCATION: Mutex<String> = Mutex::new(String::new());
+    static ref BACKTRACE_ENABLED: Mutex<bool> = Mutex::new(false);
+}
 /// `debug` API implementation.
 ///
 /// This type provides the functionality for handling `debug` related requests.
@@ -643,6 +649,34 @@ where
         Ok(res.into())
     }
 
+    async fn debug_backtrace_at(&self, location: &str) -> RpcResult<()> {
+        let mut parts: Vec<&str> = location.split(':').collect();
+        if parts.len() != 2 {
+            return Err(internal_rpc_err("Invalid location format".to_string()))
+        }
+        parts[0] = parts[0].trim();
+        parts[1] = parts[1].trim();
+
+        if parts[0].is_empty() || parts[1].is_empty() {
+            return Err(internal_rpc_err("Invalid location format".to_string()))
+        }
+        if !parts[0].ends_with(".rust") {
+            return Err(internal_rpc_err("Invalid location format".to_string()))
+        }
+
+        if parts[1].parse::<i32>().is_err() {
+            return Err(internal_rpc_err("Invalid location format".to_string()))
+        }
+
+        let mut location_lock = LOCATION.lock().unwrap();
+        *location_lock = location.to_string();
+
+        let mut backtrace_enabled_lock = BACKTRACE_ENABLED.lock().unwrap();
+        *backtrace_enabled_lock = !location.is_empty();
+
+        Ok(())
+    }
+
     /// Handler for `debug_getRawBlock`
     async fn raw_block(&self, block_id: BlockId) -> RpcResult<Bytes> {
         let block = self.inner.provider.block_by_id(block_id).to_rpc_result()?;