Merge pull request #53 from propeller-heads/fp/fix-vm-indexing

fix(substreams): Fix storage changes indexing for DELEGATECALL
This commit is contained in:
Zizou
2024-07-22 10:25:14 +02:00
committed by GitHub

View File

@@ -10,9 +10,9 @@
/// more [here](https://streamingfastio.medium.com/new-block-model-to-accelerate-chain-integration-9f65126e5425) /// more [here](https://streamingfastio.medium.com/new-block-model-to-accelerate-chain-integration-9f65126e5425)
use std::collections::HashMap; use std::collections::HashMap;
use substreams_ethereum::pb::{ use substreams_ethereum::pb::eth::{
eth, self,
eth::v2::{block::DetailLevel, StorageChange}, v2::{block::DetailLevel, CallType, StorageChange},
}; };
use crate::pb::tycho::evm::v1::{self as tycho}; use crate::pb::tycho::evm::v1::{self as tycho};
@@ -128,11 +128,17 @@ pub fn extract_contract_changes<F: Fn(&[u8]) -> bool>(
let mut balance_changes = Vec::new(); let mut balance_changes = Vec::new();
let mut code_changes = Vec::new(); let mut code_changes = Vec::new();
block_tx let filtered_calls = block_tx.calls.iter().filter(|call| {
.calls let address_included = inclusion_predicate(&call.address);
.iter() let caller_included = inclusion_predicate(&call.caller);
.filter(|call| !call.state_reverted && inclusion_predicate(&call.address)) let is_delegate_or_callcode = call.call_type() == CallType::Delegate ||
.for_each(|call| { call.call_type() == CallType::Callcode;
!call.state_reverted &&
(address_included || (caller_included && is_delegate_or_callcode))
});
filtered_calls.for_each(|call| {
storage_changes.extend(call.storage_changes.iter()); storage_changes.extend(call.storage_changes.iter());
balance_changes.extend(call.balance_changes.iter()); balance_changes.extend(call.balance_changes.iter());
code_changes.extend(call.code_changes.iter()); code_changes.extend(call.code_changes.iter());