Skip to main content

reth_rpc/
trace.rs

1use alloy_consensus::BlockHeader as _;
2use alloy_eips::BlockId;
3use alloy_evm::block::calc::{base_block_reward_pre_merge, block_reward, ommer_reward};
4use alloy_primitives::{
5    map::{HashMap, HashSet},
6    Address, BlockHash, Bytes, B256, U256,
7};
8use alloy_rpc_types_eth::{
9    state::{EvmOverrides, StateOverride},
10    BlockOverrides, Index,
11};
12use alloy_rpc_types_trace::{
13    filter::TraceFilter,
14    opcode::{BlockOpcodeGas, TransactionOpcodeGas},
15    parity::*,
16    tracerequest::TraceCallRequest,
17};
18use async_trait::async_trait;
19use jsonrpsee::core::RpcResult;
20use reth_chainspec::{ChainSpecProvider, EthereumHardforks};
21use reth_evm::ConfigureEvm;
22use reth_primitives_traits::{BlockBody, BlockHeader};
23use reth_rpc_api::TraceApiServer;
24use reth_rpc_convert::RpcTxReq;
25use reth_rpc_eth_api::{
26    helpers::{Call, LoadPendingBlock, LoadTransaction, Trace, TraceExt},
27    FromEthApiError, RpcNodeCore,
28};
29use reth_rpc_eth_types::{error::EthApiError, utils::recover_raw_transaction, EthConfig};
30use reth_storage_api::{BlockNumReader, BlockReader};
31use reth_tasks::pool::BlockingTaskGuard;
32use reth_transaction_pool::{PoolPooledTx, PoolTransaction, TransactionPool};
33use revm::DatabaseCommit;
34use revm_inspectors::{
35    opcode::OpcodeGasInspector,
36    storage::StorageInspector,
37    tracing::{parity::populate_state_diff, TracingInspector, TracingInspectorConfig},
38};
39use serde::{Deserialize, Serialize};
40use std::sync::Arc;
41use tokio::sync::{AcquireError, OwnedSemaphorePermit};
42
43/// `trace` API implementation.
44///
45/// This type provides the functionality for handling `trace` related requests.
46pub struct TraceApi<Eth> {
47    inner: Arc<TraceApiInner<Eth>>,
48}
49
50// === impl TraceApi ===
51
52impl<Eth> TraceApi<Eth> {
53    /// Create a new instance of the [`TraceApi`]
54    pub fn new(
55        eth_api: Eth,
56        blocking_task_guard: BlockingTaskGuard,
57        eth_config: EthConfig,
58    ) -> Self {
59        let inner = Arc::new(TraceApiInner { eth_api, blocking_task_guard, eth_config });
60        Self { inner }
61    }
62
63    /// Acquires a permit to execute a tracing call.
64    async fn acquire_trace_permit(
65        &self,
66    ) -> std::result::Result<OwnedSemaphorePermit, AcquireError> {
67        self.inner.blocking_task_guard.clone().acquire_owned().await
68    }
69
70    /// Access the underlying `Eth` API.
71    pub fn eth_api(&self) -> &Eth {
72        &self.inner.eth_api
73    }
74}
75
76impl<Eth: RpcNodeCore> TraceApi<Eth> {
77    /// Access the underlying provider.
78    pub fn provider(&self) -> &Eth::Provider {
79        self.inner.eth_api.provider()
80    }
81}
82
83// === impl TraceApi === //
84
85impl<Eth> TraceApi<Eth>
86where
87    // tracing methods do _not_ read from mempool, hence no `LoadBlock` trait
88    // bound
89    Eth: Trace + Call + LoadPendingBlock + LoadTransaction + 'static,
90{
91    /// Executes the given call and returns a number of possible traces for it.
92    pub async fn trace_call(
93        &self,
94        trace_request: TraceCallRequest<RpcTxReq<Eth::NetworkTypes>>,
95    ) -> Result<TraceResults, Eth::Error> {
96        let at = trace_request.block_id.unwrap_or_default();
97        let config = TracingInspectorConfig::from_parity_config(&trace_request.trace_types);
98        let overrides =
99            EvmOverrides::new(trace_request.state_overrides, trace_request.block_overrides);
100        let mut inspector = TracingInspector::new(config);
101        let this = self.clone();
102        self.eth_api()
103            .spawn_with_call_at(trace_request.call, at, overrides, move |db, evm_env, tx_env| {
104                let res = this.eth_api().inspect(&mut *db, evm_env, tx_env, &mut inspector)?;
105                let trace_res = inspector
106                    .into_parity_builder()
107                    .into_trace_results_with_state(&res, &trace_request.trace_types, &db)
108                    .map_err(Eth::Error::from_eth_err)?;
109                Ok(trace_res)
110            })
111            .await
112    }
113
114    /// Traces a call to `eth_sendRawTransaction` without making the call, returning the traces.
115    pub async fn trace_raw_transaction(
116        &self,
117        tx: Bytes,
118        trace_types: HashSet<TraceType>,
119        block_id: Option<BlockId>,
120    ) -> Result<TraceResults, Eth::Error> {
121        let tx = recover_raw_transaction::<PoolPooledTx<Eth::Pool>>(&tx)?
122            .map(<Eth::Pool as TransactionPool>::Transaction::pooled_into_consensus);
123
124        let (evm_env, at) = self.eth_api().evm_env_at(block_id.unwrap_or_default()).await?;
125        let tx_env = self.eth_api().evm_config().tx_env(tx);
126
127        let config = TracingInspectorConfig::from_parity_config(&trace_types);
128
129        self.eth_api()
130            .spawn_trace_at_with_state(evm_env, tx_env, config, at, move |inspector, res, db| {
131                inspector
132                    .into_parity_builder()
133                    .into_trace_results_with_state(&res, &trace_types, &db)
134                    .map_err(Eth::Error::from_eth_err)
135            })
136            .await
137    }
138
139    /// Performs multiple call traces on top of the same block. i.e. transaction n will be executed
140    /// on top of a pending block with all n-1 transactions applied (traced) first.
141    ///
142    /// Note: Allows tracing dependent transactions, hence all transactions are traced in sequence
143    pub async fn trace_call_many(
144        &self,
145        calls: Vec<(RpcTxReq<Eth::NetworkTypes>, HashSet<TraceType>)>,
146        block_id: Option<BlockId>,
147    ) -> Result<Vec<TraceResults>, Eth::Error> {
148        let at = block_id.unwrap_or(BlockId::pending());
149        let (evm_env, at) = self.eth_api().evm_env_at(at).await?;
150
151        // execute all transactions on top of each other and record the traces
152        self.eth_api()
153            .spawn_with_state_at_block(at, move |eth_api, mut db| {
154                let mut results = Vec::with_capacity(calls.len());
155                let mut calls = calls.into_iter().peekable();
156
157                while let Some((call, trace_types)) = calls.next() {
158                    let (evm_env, tx_env) = eth_api.prepare_call_env(
159                        evm_env.clone(),
160                        call,
161                        &mut db,
162                        Default::default(),
163                    )?;
164                    let config = TracingInspectorConfig::from_parity_config(&trace_types);
165                    let mut inspector = TracingInspector::new(config);
166                    let res = eth_api.inspect(&mut db, evm_env, tx_env, &mut inspector)?;
167
168                    let trace_res = inspector
169                        .into_parity_builder()
170                        .into_trace_results_with_state(&res, &trace_types, &db)
171                        .map_err(Eth::Error::from_eth_err)?;
172
173                    results.push(trace_res);
174
175                    // need to apply the state changes of this call before executing the
176                    // next call
177                    if calls.peek().is_some() {
178                        db.commit(res.state)
179                    }
180                }
181
182                Ok(results)
183            })
184            .await
185    }
186
187    /// Replays a transaction, returning the traces.
188    pub async fn replay_transaction(
189        &self,
190        hash: B256,
191        trace_types: HashSet<TraceType>,
192    ) -> Result<TraceResults, Eth::Error> {
193        let config = TracingInspectorConfig::from_parity_config(&trace_types);
194        self.eth_api()
195            .spawn_trace_transaction_in_block(hash, config, move |_, inspector, res, db| {
196                let trace_res = inspector
197                    .into_parity_builder()
198                    .into_trace_results_with_state(&res, &trace_types, &db)
199                    .map_err(Eth::Error::from_eth_err)?;
200                Ok(trace_res)
201            })
202            .await
203            .transpose()
204            .ok_or(EthApiError::TransactionNotFound)?
205    }
206
207    /// Returns transaction trace objects at the given index
208    ///
209    /// Note: For compatibility reasons this only supports 1 single index, since this method is
210    /// supposed to return a single trace. See also: <https://github.com/ledgerwatch/erigon/blob/862faf054b8a0fa15962a9c73839b619886101eb/turbo/jsonrpc/trace_filtering.go#L114-L133>
211    ///
212    /// This returns `None` if `indices` is empty
213    pub async fn trace_get(
214        &self,
215        hash: B256,
216        indices: Vec<usize>,
217    ) -> Result<Option<LocalizedTransactionTrace>, Eth::Error> {
218        if indices.len() != 1 {
219            // The OG impl failed if it gets more than a single index
220            return Ok(None)
221        }
222        self.trace_get_index(hash, indices[0]).await
223    }
224
225    /// Returns transaction trace object at the given index.
226    ///
227    /// Returns `None` if the trace object at that index does not exist
228    pub async fn trace_get_index(
229        &self,
230        hash: B256,
231        index: usize,
232    ) -> Result<Option<LocalizedTransactionTrace>, Eth::Error> {
233        Ok(self.trace_transaction(hash).await?.and_then(|traces| traces.into_iter().nth(index)))
234    }
235
236    /// Returns all traces for the given transaction hash
237    pub async fn trace_transaction(
238        &self,
239        hash: B256,
240    ) -> Result<Option<Vec<LocalizedTransactionTrace>>, Eth::Error> {
241        self.eth_api()
242            .spawn_trace_transaction_in_block(
243                hash,
244                TracingInspectorConfig::default_parity(),
245                move |tx_info, inspector, _, _| {
246                    let traces =
247                        inspector.into_parity_builder().into_localized_transaction_traces(tx_info);
248                    Ok(traces)
249                },
250            )
251            .await
252    }
253
254    /// Returns all opcodes with their count and combined gas usage for the given transaction in no
255    /// particular order.
256    pub async fn trace_transaction_opcode_gas(
257        &self,
258        tx_hash: B256,
259    ) -> Result<Option<TransactionOpcodeGas>, Eth::Error> {
260        self.eth_api()
261            .spawn_trace_transaction_in_block_with_inspector(
262                tx_hash,
263                OpcodeGasInspector::default(),
264                move |_tx_info, inspector, _res, _| {
265                    let trace = TransactionOpcodeGas {
266                        transaction_hash: tx_hash,
267                        opcode_gas: inspector.opcode_gas_iter().collect(),
268                    };
269                    Ok(trace)
270                },
271            )
272            .await
273    }
274
275    /// Calculates the base block reward for the given block:
276    ///
277    /// - if Paris hardfork is activated, no block rewards are given
278    /// - if Paris hardfork is not activated, calculate block rewards with block number only
279    fn calculate_base_block_reward<H: BlockHeader>(
280        &self,
281        header: &H,
282    ) -> Result<Option<u128>, Eth::Error> {
283        let chain_spec = self.provider().chain_spec();
284
285        if chain_spec.is_paris_active_at_block(header.number()) {
286            return Ok(None)
287        }
288
289        Ok(Some(base_block_reward_pre_merge(&chain_spec, header.number())))
290    }
291
292    /// Extracts the reward traces for the given block:
293    ///  - block reward
294    ///  - uncle rewards
295    fn extract_reward_traces<H: BlockHeader>(
296        &self,
297        header: &H,
298        ommers: Option<&[H]>,
299        base_block_reward: u128,
300    ) -> Vec<LocalizedTransactionTrace> {
301        let ommers_cnt = ommers.map(|o| o.len()).unwrap_or_default();
302        let mut traces = Vec::with_capacity(ommers_cnt + 1);
303
304        let block_reward = block_reward(base_block_reward, ommers_cnt);
305        traces.push(reward_trace(
306            header,
307            RewardAction {
308                author: header.beneficiary(),
309                reward_type: RewardType::Block,
310                value: U256::from(block_reward),
311            },
312        ));
313
314        let Some(ommers) = ommers else { return traces };
315
316        for uncle in ommers {
317            let uncle_reward = ommer_reward(base_block_reward, header.number(), uncle.number());
318            traces.push(reward_trace(
319                header,
320                RewardAction {
321                    author: uncle.beneficiary(),
322                    reward_type: RewardType::Uncle,
323                    value: U256::from(uncle_reward),
324                },
325            ));
326        }
327        traces
328    }
329}
330
331impl<Eth> TraceApi<Eth>
332where
333    // tracing methods read from mempool, hence `LoadBlock` trait bound via
334    // `TraceExt`
335    Eth: TraceExt + 'static,
336{
337    /// Returns all transaction traces that match the given filter.
338    ///
339    /// This is similar to [`Self::trace_block`] but only returns traces for transactions that match
340    /// the filter.
341    pub async fn trace_filter(
342        &self,
343        filter: TraceFilter,
344    ) -> Result<Vec<LocalizedTransactionTrace>, Eth::Error> {
345        // We'll reuse the matcher across multiple blocks that are traced in parallel
346        let matcher = Arc::new(filter.matcher());
347        let TraceFilter { from_block, to_block, mut after, count, .. } = filter;
348        let start = from_block.unwrap_or(0);
349
350        let latest_block = self.provider().best_block_number().map_err(Eth::Error::from_eth_err)?;
351        if start > latest_block {
352            // can't trace that range
353            return Err(EthApiError::HeaderNotFound(start.into()).into());
354        }
355        let end = to_block.unwrap_or(latest_block);
356        if end > latest_block {
357            return Err(EthApiError::HeaderNotFound(end.into()).into());
358        }
359
360        if start > end {
361            return Err(EthApiError::InvalidParams(
362                "invalid parameters: fromBlock cannot be greater than toBlock".to_string(),
363            )
364            .into())
365        }
366
367        // ensure that the range is not too large, since we need to fetch all blocks in the range
368        let distance = end.saturating_sub(start);
369        if distance > self.inner.eth_config.max_trace_filter_blocks {
370            return Err(EthApiError::InvalidParams(format!(
371                "Block range too large; currently limited to {} blocks",
372                self.inner.eth_config.max_trace_filter_blocks
373            ))
374            .into())
375        }
376
377        let mut all_traces = Vec::new();
378        let mut block_traces = Vec::with_capacity(self.inner.eth_config.max_tracing_requests);
379        for chunk_start in (start..=end).step_by(self.inner.eth_config.max_tracing_requests) {
380            let chunk_end = std::cmp::min(
381                chunk_start + self.inner.eth_config.max_tracing_requests as u64 - 1,
382                end,
383            );
384
385            // fetch all blocks in that chunk
386            let blocks = self
387                .eth_api()
388                .spawn_blocking_io(move |this| {
389                    Ok(this
390                        .provider()
391                        .recovered_block_range(chunk_start..=chunk_end)
392                        .map_err(Eth::Error::from_eth_err)?
393                        .into_iter()
394                        .map(Arc::new)
395                        .collect::<Vec<_>>())
396                })
397                .await?;
398
399            // trace all blocks
400            for block in &blocks {
401                let matcher = matcher.clone();
402                let traces = self.eth_api().trace_block_until(
403                    block.hash().into(),
404                    Some(block.clone()),
405                    None,
406                    TracingInspectorConfig::default_parity(),
407                    move |tx_info, mut ctx| {
408                        let mut traces = ctx
409                            .take_inspector()
410                            .into_parity_builder()
411                            .into_localized_transaction_traces(tx_info);
412                        traces.retain(|trace| matcher.matches(&trace.trace));
413                        Ok(Some(traces))
414                    },
415                );
416                block_traces.push(traces);
417            }
418
419            #[allow(clippy::iter_with_drain)]
420            let block_traces = futures::future::try_join_all(block_traces.drain(..)).await?;
421            all_traces.extend(block_traces.into_iter().flatten().flat_map(|traces| {
422                traces.into_iter().flatten().flat_map(|traces| traces.into_iter())
423            }));
424
425            // add reward traces for all blocks
426            for block in &blocks {
427                if let Some(base_block_reward) = self.calculate_base_block_reward(block.header())? {
428                    all_traces.extend(
429                        self.extract_reward_traces(
430                            block.header(),
431                            block.body().ommers(),
432                            base_block_reward,
433                        )
434                        .into_iter()
435                        .filter(|trace| matcher.matches(&trace.trace)),
436                    );
437                } else {
438                    // no block reward, means we're past the Paris hardfork and don't expect any
439                    // rewards because the blocks in ascending order
440                    break
441                }
442            }
443
444            // Skips the first `after` number of matching traces.
445            if let Some(cutoff) = after.map(|a| a as usize) &&
446                cutoff < all_traces.len()
447            {
448                all_traces.drain(..cutoff);
449                // we removed the first `after` traces
450                after = None;
451            }
452
453            // Return at most `count` of traces
454            if let Some(count) = count {
455                let count = count as usize;
456                if count < all_traces.len() {
457                    all_traces.truncate(count);
458                    return Ok(all_traces)
459                }
460            };
461        }
462
463        // If `after` is greater than or equal to the number of matched traces, it returns an
464        // empty array.
465        if let Some(cutoff) = after.map(|a| a as usize) &&
466            cutoff >= all_traces.len()
467        {
468            return Ok(vec![])
469        }
470
471        Ok(all_traces)
472    }
473
474    /// Returns traces created at given block.
475    pub async fn trace_block(
476        &self,
477        block_id: BlockId,
478    ) -> Result<Option<Vec<LocalizedTransactionTrace>>, Eth::Error> {
479        let Some(block) = self.eth_api().recovered_block(block_id).await? else {
480            return Err(EthApiError::HeaderNotFound(block_id).into());
481        };
482
483        let mut traces = self
484            .eth_api()
485            .trace_block_with(
486                block_id,
487                Some(block.clone()),
488                TracingInspectorConfig::default_parity(),
489                |tx_info, mut ctx| {
490                    let traces = ctx
491                        .take_inspector()
492                        .into_parity_builder()
493                        .into_localized_transaction_traces(tx_info);
494                    Ok(traces)
495                },
496            )
497            .await?
498            .map(|traces| traces.into_iter().flatten().collect::<Vec<_>>());
499
500        if let Some(traces) = traces.as_mut() &&
501            let Some(base_block_reward) = self.calculate_base_block_reward(block.header())?
502        {
503            traces.extend(self.extract_reward_traces(
504                block.header(),
505                block.body().ommers(),
506                base_block_reward,
507            ));
508        }
509
510        Ok(traces)
511    }
512
513    /// Replays all transactions in a block
514    pub async fn replay_block_transactions(
515        &self,
516        block_id: BlockId,
517        trace_types: HashSet<TraceType>,
518    ) -> Result<Option<Vec<TraceResultsWithTransactionHash>>, Eth::Error> {
519        self.eth_api()
520            .trace_block_with(
521                block_id,
522                None,
523                TracingInspectorConfig::from_parity_config(&trace_types),
524                move |tx_info, mut ctx| {
525                    let mut full_trace = ctx
526                        .take_inspector()
527                        .into_parity_builder()
528                        .into_trace_results(&ctx.result, &trace_types);
529
530                    // If statediffs were requested, populate them with the account balance and
531                    // nonce from pre-state
532                    if let Some(ref mut state_diff) = full_trace.state_diff {
533                        populate_state_diff(state_diff, &ctx.db, ctx.state.iter())
534                            .map_err(Eth::Error::from_eth_err)?;
535                    }
536
537                    let trace = TraceResultsWithTransactionHash {
538                        transaction_hash: tx_info.hash.expect("tx hash is set"),
539                        full_trace,
540                    };
541                    Ok(trace)
542                },
543            )
544            .await
545    }
546
547    /// Returns the opcodes of all transactions in the given block.
548    ///
549    /// This is the same as [`Self::trace_transaction_opcode_gas`] but for all transactions in a
550    /// block.
551    pub async fn trace_block_opcode_gas(
552        &self,
553        block_id: BlockId,
554    ) -> Result<Option<BlockOpcodeGas>, Eth::Error> {
555        let Some(block) = self.eth_api().recovered_block(block_id).await? else {
556            return Err(EthApiError::HeaderNotFound(block_id).into());
557        };
558
559        let Some(transactions) = self
560            .eth_api()
561            .trace_block_inspector(
562                block_id,
563                Some(block.clone()),
564                OpcodeGasInspector::default,
565                move |tx_info, ctx| {
566                    let trace = TransactionOpcodeGas {
567                        transaction_hash: tx_info.hash.expect("tx hash is set"),
568                        opcode_gas: ctx.inspector.opcode_gas_iter().collect(),
569                    };
570                    Ok(trace)
571                },
572            )
573            .await?
574        else {
575            return Ok(None);
576        };
577
578        Ok(Some(BlockOpcodeGas {
579            block_hash: block.hash(),
580            block_number: block.number(),
581            transactions,
582        }))
583    }
584
585    /// Returns all storage slots accessed during transaction execution along with their access
586    /// counts.
587    pub async fn trace_block_storage_access(
588        &self,
589        block_id: BlockId,
590    ) -> Result<Option<BlockStorageAccess>, Eth::Error> {
591        let Some(block) = self.eth_api().recovered_block(block_id).await? else {
592            return Err(EthApiError::HeaderNotFound(block_id).into());
593        };
594
595        let Some(transactions) = self
596            .eth_api()
597            .trace_block_inspector(
598                block_id,
599                Some(block.clone()),
600                StorageInspector::default,
601                move |tx_info, ctx| {
602                    let trace = TransactionStorageAccess {
603                        transaction_hash: tx_info.hash.expect("tx hash is set"),
604                        storage_access: ctx.inspector.accessed_slots().clone(),
605                        unique_loads: ctx.inspector.unique_loads(),
606                        warm_loads: ctx.inspector.warm_loads(),
607                    };
608                    Ok(trace)
609                },
610            )
611            .await?
612        else {
613            return Ok(None);
614        };
615
616        Ok(Some(BlockStorageAccess {
617            block_hash: block.hash(),
618            block_number: block.number(),
619            transactions,
620        }))
621    }
622}
623
624#[async_trait]
625impl<Eth> TraceApiServer<RpcTxReq<Eth::NetworkTypes>> for TraceApi<Eth>
626where
627    Eth: TraceExt + 'static,
628{
629    /// Executes the given call and returns a number of possible traces for it.
630    ///
631    /// Handler for `trace_call`
632    async fn trace_call(
633        &self,
634        call: RpcTxReq<Eth::NetworkTypes>,
635        trace_types: HashSet<TraceType>,
636        block_id: Option<BlockId>,
637        state_overrides: Option<StateOverride>,
638        block_overrides: Option<Box<BlockOverrides>>,
639    ) -> RpcResult<TraceResults> {
640        let _permit = self.acquire_trace_permit().await;
641        let request =
642            TraceCallRequest { call, trace_types, block_id, state_overrides, block_overrides };
643        Ok(Self::trace_call(self, request).await.map_err(Into::into)?)
644    }
645
646    /// Handler for `trace_callMany`
647    async fn trace_call_many(
648        &self,
649        calls: Vec<(RpcTxReq<Eth::NetworkTypes>, HashSet<TraceType>)>,
650        block_id: Option<BlockId>,
651    ) -> RpcResult<Vec<TraceResults>> {
652        let _permit = self.acquire_trace_permit().await;
653        Ok(Self::trace_call_many(self, calls, block_id).await.map_err(Into::into)?)
654    }
655
656    /// Handler for `trace_rawTransaction`
657    async fn trace_raw_transaction(
658        &self,
659        data: Bytes,
660        trace_types: HashSet<TraceType>,
661        block_id: Option<BlockId>,
662    ) -> RpcResult<TraceResults> {
663        let _permit = self.acquire_trace_permit().await;
664        Ok(Self::trace_raw_transaction(self, data, trace_types, block_id)
665            .await
666            .map_err(Into::into)?)
667    }
668
669    /// Handler for `trace_replayBlockTransactions`
670    async fn replay_block_transactions(
671        &self,
672        block_id: BlockId,
673        trace_types: HashSet<TraceType>,
674    ) -> RpcResult<Option<Vec<TraceResultsWithTransactionHash>>> {
675        let _permit = self.acquire_trace_permit().await;
676        Ok(Self::replay_block_transactions(self, block_id, trace_types)
677            .await
678            .map_err(Into::into)?)
679    }
680
681    /// Handler for `trace_replayTransaction`
682    async fn replay_transaction(
683        &self,
684        transaction: B256,
685        trace_types: HashSet<TraceType>,
686    ) -> RpcResult<TraceResults> {
687        let _permit = self.acquire_trace_permit().await;
688        Ok(Self::replay_transaction(self, transaction, trace_types).await.map_err(Into::into)?)
689    }
690
691    /// Handler for `trace_block`
692    async fn trace_block(
693        &self,
694        block_id: BlockId,
695    ) -> RpcResult<Option<Vec<LocalizedTransactionTrace>>> {
696        let _permit = self.acquire_trace_permit().await;
697        Ok(Self::trace_block(self, block_id).await.map_err(Into::into)?)
698    }
699
700    /// Handler for `trace_filter`
701    ///
702    /// This is similar to `eth_getLogs` but for traces.
703    ///
704    /// # Limitations
705    /// This currently requires block filter fields, since reth does not have address indices yet.
706    async fn trace_filter(&self, filter: TraceFilter) -> RpcResult<Vec<LocalizedTransactionTrace>> {
707        let _permit = self.inner.blocking_task_guard.clone().acquire_many_owned(2).await;
708        Ok(Self::trace_filter(self, filter).await.map_err(Into::into)?)
709    }
710
711    /// Returns transaction trace at given index.
712    /// Handler for `trace_get`
713    async fn trace_get(
714        &self,
715        hash: B256,
716        indices: Vec<Index>,
717    ) -> RpcResult<Option<LocalizedTransactionTrace>> {
718        let _permit = self.acquire_trace_permit().await;
719        Ok(Self::trace_get(self, hash, indices.into_iter().map(Into::into).collect())
720            .await
721            .map_err(Into::into)?)
722    }
723
724    /// Handler for `trace_transaction`
725    async fn trace_transaction(
726        &self,
727        hash: B256,
728    ) -> RpcResult<Option<Vec<LocalizedTransactionTrace>>> {
729        let _permit = self.acquire_trace_permit().await;
730        Ok(Self::trace_transaction(self, hash).await.map_err(Into::into)?)
731    }
732
733    /// Handler for `trace_transactionOpcodeGas`
734    async fn trace_transaction_opcode_gas(
735        &self,
736        tx_hash: B256,
737    ) -> RpcResult<Option<TransactionOpcodeGas>> {
738        let _permit = self.acquire_trace_permit().await;
739        Ok(Self::trace_transaction_opcode_gas(self, tx_hash).await.map_err(Into::into)?)
740    }
741
742    /// Handler for `trace_blockOpcodeGas`
743    async fn trace_block_opcode_gas(&self, block_id: BlockId) -> RpcResult<Option<BlockOpcodeGas>> {
744        let _permit = self.acquire_trace_permit().await;
745        Ok(Self::trace_block_opcode_gas(self, block_id).await.map_err(Into::into)?)
746    }
747}
748
749impl<Eth> std::fmt::Debug for TraceApi<Eth> {
750    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
751        f.debug_struct("TraceApi").finish_non_exhaustive()
752    }
753}
754impl<Eth> Clone for TraceApi<Eth> {
755    fn clone(&self) -> Self {
756        Self { inner: Arc::clone(&self.inner) }
757    }
758}
759
760struct TraceApiInner<Eth> {
761    /// Access to commonly used code of the `eth` namespace
762    eth_api: Eth,
763    // restrict the number of concurrent calls to `trace_*`
764    blocking_task_guard: BlockingTaskGuard,
765    // eth config settings
766    eth_config: EthConfig,
767}
768
769/// Response type for storage tracing that contains all accessed storage slots
770/// for a transaction.
771#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
772#[serde(rename_all = "camelCase")]
773pub struct TransactionStorageAccess {
774    /// Hash of the transaction
775    pub transaction_hash: B256,
776    /// Tracks storage slots and access counter.
777    pub storage_access: HashMap<Address, HashMap<B256, u64>>,
778    /// Number of unique storage loads
779    pub unique_loads: u64,
780    /// Number of warm storage loads
781    pub warm_loads: u64,
782}
783
784/// Response type for storage tracing that contains all accessed storage slots
785#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
786#[serde(rename_all = "camelCase")]
787pub struct BlockStorageAccess {
788    /// The block hash
789    pub block_hash: BlockHash,
790    /// The block's number
791    pub block_number: u64,
792    /// All executed transactions in the block in the order they were executed
793    pub transactions: Vec<TransactionStorageAccess>,
794}
795
796/// Helper to construct a [`LocalizedTransactionTrace`] that describes a reward to the block
797/// beneficiary.
798fn reward_trace<H: BlockHeader>(header: &H, reward: RewardAction) -> LocalizedTransactionTrace {
799    LocalizedTransactionTrace {
800        block_hash: Some(header.hash_slow()),
801        block_number: Some(header.number()),
802        transaction_hash: None,
803        transaction_position: None,
804        trace: TransactionTrace {
805            trace_address: vec![],
806            subtraces: 0,
807            action: Action::Reward(reward),
808            error: None,
809            result: None,
810        },
811    }
812}