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(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 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 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 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 break
441 }
442 }
443
444 if let Some(cutoff) = after.map(|a| a as usize) &&
446 cutoff < all_traces.len()
447 {
448 all_traces.drain(..cutoff);
449 after = None;
451 }
452
453 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 eth_api: Eth,
763 blocking_task_guard: BlockingTaskGuard,
765 eth_config: EthConfig,
767}
768
769#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
772#[serde(rename_all = "camelCase")]
773pub struct TransactionStorageAccess {
774 pub transaction_hash: B256,
776 pub storage_access: HashMap<Address, HashMap<B256, u64>>,
778 pub unique_loads: u64,
780 pub warm_loads: u64,
782}
783
784#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
786#[serde(rename_all = "camelCase")]
787pub struct BlockStorageAccess {
788 pub block_hash: BlockHash,
790 pub block_number: u64,
792 pub transactions: Vec<TransactionStorageAccess>,
794}
795
796fn 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}