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
43pub struct TraceApi<Eth> {
47 inner: Arc<TraceApiInner<Eth>>,
48}
49
50impl<Eth> TraceApi<Eth> {
53 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 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 pub fn eth_api(&self) -> &Eth {
72 &self.inner.eth_api
73 }
74}
75
76impl<Eth: RpcNodeCore> TraceApi<Eth> {
77 pub fn provider(&self) -> &Eth::Provider {
79 self.inner.eth_api.provider()
80 }
81}
82
83impl<Eth> TraceApi<Eth>
86where
87 Eth: Trace + Call + LoadPendingBlock + LoadTransaction + 'static,
90{
91 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 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 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 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 if calls.peek().is_some() {
178 db.commit(res.state)
179 }
180 }
181
182 Ok(results)
183 })
184 .await
185 }
186
187 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 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 return Ok(None)
221 }
222 self.trace_get_index(hash, indices[0]).await
223 }
224
225 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 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 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 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 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 Eth: TraceExt + 'static,
336{
337 pub async fn trace_filter(
342 &self,
343 filter: TraceFilter,
344 ) -> Result<Vec<LocalizedTransactionTrace>, Eth::Error> {
345 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 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 let distance = end.saturating_sub(start);
369 if distance > self.inner.eth_config.max_trace_filter_blocks {
370 return Err(EthApiError::InvalidParams(
371 "Block range too large; currently limited to 100 blocks".to_string(),
372 )
373 .into())
374 }
375
376 let mut all_traces = Vec::new();
377 let mut block_traces = Vec::with_capacity(self.inner.eth_config.max_tracing_requests);
378 for chunk_start in (start..=end).step_by(self.inner.eth_config.max_tracing_requests) {
379 let chunk_end = std::cmp::min(
380 chunk_start + self.inner.eth_config.max_tracing_requests as u64 - 1,
381 end,
382 );
383
384 let blocks = self
386 .eth_api()
387 .spawn_blocking_io(move |this| {
388 Ok(this
389 .provider()
390 .recovered_block_range(chunk_start..=chunk_end)
391 .map_err(Eth::Error::from_eth_err)?
392 .into_iter()
393 .map(Arc::new)
394 .collect::<Vec<_>>())
395 })
396 .await?;
397
398 for block in &blocks {
400 let matcher = matcher.clone();
401 let traces = self.eth_api().trace_block_until(
402 block.hash().into(),
403 Some(block.clone()),
404 None,
405 TracingInspectorConfig::default_parity(),
406 move |tx_info, mut ctx| {
407 let mut traces = ctx
408 .take_inspector()
409 .into_parity_builder()
410 .into_localized_transaction_traces(tx_info);
411 traces.retain(|trace| matcher.matches(&trace.trace));
412 Ok(Some(traces))
413 },
414 );
415 block_traces.push(traces);
416 }
417
418 #[allow(clippy::iter_with_drain)]
419 let block_traces = futures::future::try_join_all(block_traces.drain(..)).await?;
420 all_traces.extend(block_traces.into_iter().flatten().flat_map(|traces| {
421 traces.into_iter().flatten().flat_map(|traces| traces.into_iter())
422 }));
423
424 for block in &blocks {
426 if let Some(base_block_reward) = self.calculate_base_block_reward(block.header())? {
427 all_traces.extend(
428 self.extract_reward_traces(
429 block.header(),
430 block.body().ommers(),
431 base_block_reward,
432 )
433 .into_iter()
434 .filter(|trace| matcher.matches(&trace.trace)),
435 );
436 } else {
437 break
440 }
441 }
442
443 if let Some(cutoff) = after.map(|a| a as usize) &&
445 cutoff < all_traces.len()
446 {
447 all_traces.drain(..cutoff);
448 after = None;
450 }
451
452 if let Some(count) = count {
454 let count = count as usize;
455 if count < all_traces.len() {
456 all_traces.truncate(count);
457 return Ok(all_traces)
458 }
459 };
460 }
461
462 if let Some(cutoff) = after.map(|a| a as usize) &&
465 cutoff >= all_traces.len()
466 {
467 return Ok(vec![])
468 }
469
470 Ok(all_traces)
471 }
472
473 pub async fn trace_block(
475 &self,
476 block_id: BlockId,
477 ) -> Result<Option<Vec<LocalizedTransactionTrace>>, Eth::Error> {
478 let traces = self.eth_api().trace_block_with(
479 block_id,
480 None,
481 TracingInspectorConfig::default_parity(),
482 |tx_info, mut ctx| {
483 let traces = ctx
484 .take_inspector()
485 .into_parity_builder()
486 .into_localized_transaction_traces(tx_info);
487 Ok(traces)
488 },
489 );
490
491 let block = self.eth_api().recovered_block(block_id);
492 let (maybe_traces, maybe_block) = futures::try_join!(traces, block)?;
493
494 let mut maybe_traces =
495 maybe_traces.map(|traces| traces.into_iter().flatten().collect::<Vec<_>>());
496
497 if let (Some(block), Some(traces)) = (maybe_block, maybe_traces.as_mut()) &&
498 let Some(base_block_reward) = self.calculate_base_block_reward(block.header())?
499 {
500 traces.extend(self.extract_reward_traces(
501 block.header(),
502 block.body().ommers(),
503 base_block_reward,
504 ));
505 }
506
507 Ok(maybe_traces)
508 }
509
510 pub async fn replay_block_transactions(
512 &self,
513 block_id: BlockId,
514 trace_types: HashSet<TraceType>,
515 ) -> Result<Option<Vec<TraceResultsWithTransactionHash>>, Eth::Error> {
516 self.eth_api()
517 .trace_block_with(
518 block_id,
519 None,
520 TracingInspectorConfig::from_parity_config(&trace_types),
521 move |tx_info, mut ctx| {
522 let mut full_trace = ctx
523 .take_inspector()
524 .into_parity_builder()
525 .into_trace_results(&ctx.result, &trace_types);
526
527 if let Some(ref mut state_diff) = full_trace.state_diff {
530 populate_state_diff(state_diff, &ctx.db, ctx.state.iter())
531 .map_err(Eth::Error::from_eth_err)?;
532 }
533
534 let trace = TraceResultsWithTransactionHash {
535 transaction_hash: tx_info.hash.expect("tx hash is set"),
536 full_trace,
537 };
538 Ok(trace)
539 },
540 )
541 .await
542 }
543
544 pub async fn trace_block_opcode_gas(
549 &self,
550 block_id: BlockId,
551 ) -> Result<Option<BlockOpcodeGas>, Eth::Error> {
552 let res = self
553 .eth_api()
554 .trace_block_inspector(
555 block_id,
556 None,
557 OpcodeGasInspector::default,
558 move |tx_info, ctx| {
559 let trace = TransactionOpcodeGas {
560 transaction_hash: tx_info.hash.expect("tx hash is set"),
561 opcode_gas: ctx.inspector.opcode_gas_iter().collect(),
562 };
563 Ok(trace)
564 },
565 )
566 .await?;
567
568 let Some(transactions) = res else { return Ok(None) };
569
570 let Some(block) = self.eth_api().recovered_block(block_id).await? else { return Ok(None) };
571
572 Ok(Some(BlockOpcodeGas {
573 block_hash: block.hash(),
574 block_number: block.number(),
575 transactions,
576 }))
577 }
578
579 pub async fn trace_block_storage_access(
582 &self,
583 block_id: BlockId,
584 ) -> Result<Option<BlockStorageAccess>, Eth::Error> {
585 let res = self
586 .eth_api()
587 .trace_block_inspector(
588 block_id,
589 None,
590 StorageInspector::default,
591 move |tx_info, ctx| {
592 let trace = TransactionStorageAccess {
593 transaction_hash: tx_info.hash.expect("tx hash is set"),
594 storage_access: ctx.inspector.accessed_slots().clone(),
595 unique_loads: ctx.inspector.unique_loads(),
596 warm_loads: ctx.inspector.warm_loads(),
597 };
598 Ok(trace)
599 },
600 )
601 .await?;
602
603 let Some(transactions) = res else { return Ok(None) };
604
605 let Some(block) = self.eth_api().recovered_block(block_id).await? else { return Ok(None) };
606
607 Ok(Some(BlockStorageAccess {
608 block_hash: block.hash(),
609 block_number: block.number(),
610 transactions,
611 }))
612 }
613}
614
615#[async_trait]
616impl<Eth> TraceApiServer<RpcTxReq<Eth::NetworkTypes>> for TraceApi<Eth>
617where
618 Eth: TraceExt + 'static,
619{
620 async fn trace_call(
624 &self,
625 call: RpcTxReq<Eth::NetworkTypes>,
626 trace_types: HashSet<TraceType>,
627 block_id: Option<BlockId>,
628 state_overrides: Option<StateOverride>,
629 block_overrides: Option<Box<BlockOverrides>>,
630 ) -> RpcResult<TraceResults> {
631 let _permit = self.acquire_trace_permit().await;
632 let request =
633 TraceCallRequest { call, trace_types, block_id, state_overrides, block_overrides };
634 Ok(Self::trace_call(self, request).await.map_err(Into::into)?)
635 }
636
637 async fn trace_call_many(
639 &self,
640 calls: Vec<(RpcTxReq<Eth::NetworkTypes>, HashSet<TraceType>)>,
641 block_id: Option<BlockId>,
642 ) -> RpcResult<Vec<TraceResults>> {
643 let _permit = self.acquire_trace_permit().await;
644 Ok(Self::trace_call_many(self, calls, block_id).await.map_err(Into::into)?)
645 }
646
647 async fn trace_raw_transaction(
649 &self,
650 data: Bytes,
651 trace_types: HashSet<TraceType>,
652 block_id: Option<BlockId>,
653 ) -> RpcResult<TraceResults> {
654 let _permit = self.acquire_trace_permit().await;
655 Ok(Self::trace_raw_transaction(self, data, trace_types, block_id)
656 .await
657 .map_err(Into::into)?)
658 }
659
660 async fn replay_block_transactions(
662 &self,
663 block_id: BlockId,
664 trace_types: HashSet<TraceType>,
665 ) -> RpcResult<Option<Vec<TraceResultsWithTransactionHash>>> {
666 let _permit = self.acquire_trace_permit().await;
667 Ok(Self::replay_block_transactions(self, block_id, trace_types)
668 .await
669 .map_err(Into::into)?)
670 }
671
672 async fn replay_transaction(
674 &self,
675 transaction: B256,
676 trace_types: HashSet<TraceType>,
677 ) -> RpcResult<TraceResults> {
678 let _permit = self.acquire_trace_permit().await;
679 Ok(Self::replay_transaction(self, transaction, trace_types).await.map_err(Into::into)?)
680 }
681
682 async fn trace_block(
684 &self,
685 block_id: BlockId,
686 ) -> RpcResult<Option<Vec<LocalizedTransactionTrace>>> {
687 let _permit = self.acquire_trace_permit().await;
688 Ok(Self::trace_block(self, block_id).await.map_err(Into::into)?)
689 }
690
691 async fn trace_filter(&self, filter: TraceFilter) -> RpcResult<Vec<LocalizedTransactionTrace>> {
698 let _permit = self.inner.blocking_task_guard.clone().acquire_many_owned(2).await;
699 Ok(Self::trace_filter(self, filter).await.map_err(Into::into)?)
700 }
701
702 async fn trace_get(
705 &self,
706 hash: B256,
707 indices: Vec<Index>,
708 ) -> RpcResult<Option<LocalizedTransactionTrace>> {
709 let _permit = self.acquire_trace_permit().await;
710 Ok(Self::trace_get(self, hash, indices.into_iter().map(Into::into).collect())
711 .await
712 .map_err(Into::into)?)
713 }
714
715 async fn trace_transaction(
717 &self,
718 hash: B256,
719 ) -> RpcResult<Option<Vec<LocalizedTransactionTrace>>> {
720 let _permit = self.acquire_trace_permit().await;
721 Ok(Self::trace_transaction(self, hash).await.map_err(Into::into)?)
722 }
723
724 async fn trace_transaction_opcode_gas(
726 &self,
727 tx_hash: B256,
728 ) -> RpcResult<Option<TransactionOpcodeGas>> {
729 let _permit = self.acquire_trace_permit().await;
730 Ok(Self::trace_transaction_opcode_gas(self, tx_hash).await.map_err(Into::into)?)
731 }
732
733 async fn trace_block_opcode_gas(&self, block_id: BlockId) -> RpcResult<Option<BlockOpcodeGas>> {
735 let _permit = self.acquire_trace_permit().await;
736 Ok(Self::trace_block_opcode_gas(self, block_id).await.map_err(Into::into)?)
737 }
738}
739
740impl<Eth> std::fmt::Debug for TraceApi<Eth> {
741 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
742 f.debug_struct("TraceApi").finish_non_exhaustive()
743 }
744}
745impl<Eth> Clone for TraceApi<Eth> {
746 fn clone(&self) -> Self {
747 Self { inner: Arc::clone(&self.inner) }
748 }
749}
750
751struct TraceApiInner<Eth> {
752 eth_api: Eth,
754 blocking_task_guard: BlockingTaskGuard,
756 eth_config: EthConfig,
758}
759
760#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
763#[serde(rename_all = "camelCase")]
764pub struct TransactionStorageAccess {
765 pub transaction_hash: B256,
767 pub storage_access: HashMap<Address, HashMap<B256, u64>>,
769 pub unique_loads: u64,
771 pub warm_loads: u64,
773}
774
775#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
777#[serde(rename_all = "camelCase")]
778pub struct BlockStorageAccess {
779 pub block_hash: BlockHash,
781 pub block_number: u64,
783 pub transactions: Vec<TransactionStorageAccess>,
785}
786
787fn reward_trace<H: BlockHeader>(header: &H, reward: RewardAction) -> LocalizedTransactionTrace {
790 LocalizedTransactionTrace {
791 block_hash: Some(header.hash_slow()),
792 block_number: Some(header.number()),
793 transaction_hash: None,
794 transaction_position: None,
795 trace: TransactionTrace {
796 trace_address: vec![],
797 subtraces: 0,
798 action: Action::Reward(reward),
799 error: None,
800 result: None,
801 },
802 }
803}