jleony/nanoreth
1
0
Fork 0
mirror of https://github.com/hl-archive-node/nanoreth.git synced 2025-12-06 10:59:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: push_trace iteratively (#2979)

Browse Source
This commit is contained in:
Matthias Seitz
2023-06-05 04:26:57 +02:00
committed by GitHub
parent 5e01f9d7b4
commit 2d0e8162b8
1 changed files with 27 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
crates/revm/revm-inspectors/src/tracing/arena.rs
View File

@ -11,35 +11,36 @@ pub struct CallTraceArena {
impl CallTraceArena {
/// Pushes a new trace into the arena, returning the trace ID
pub(crate) fn push_trace(&mut self, entry: usize, new_trace: CallTrace) -> usize {
match new_trace.depth {
// The entry node, just update it
0 => {
self.arena[0].trace = new_trace;
0
}
// We found the parent node, add the new trace as a child
_ if self.arena[entry].trace.depth == new_trace.depth - 1 => {
let id = self.arena.len();
pub(crate) fn push_trace(&mut self, mut entry: usize, new_trace: CallTrace) -> usize {
loop {
match new_trace.depth {
// The entry node, just update it
0 => {
self.arena[0].trace = new_trace;
return 0
}
// We found the parent node, add the new trace as a child
_ if self.arena[entry].trace.depth == new_trace.depth - 1 => {
let id = self.arena.len();
let trace_location = self.arena[entry].children.len();
self.arena[entry].ordering.push(LogCallOrder::Call(trace_location));
let node = CallTraceNode {
parent: Some(entry),
trace: new_trace,
idx: id,
..Default::default()
};
self.arena.push(node);
self.arena[entry].children.push(id);
let trace_location = self.arena[entry].children.len();
self.arena[entry].ordering.push(LogCallOrder::Call(trace_location));
let node = CallTraceNode {
parent: Some(entry),
trace: new_trace,
idx: id,
..Default::default()
};
self.arena.push(node);
self.arena[entry].children.push(id);
id
return id
}
_ => {
// We haven't found the parent node, go deeper
entry = *self.arena[entry].children.last().expect("Disconnected trace");
}
}
// We haven't found the parent node, go deeper
_ => self.push_trace(
*self.arena[entry].children.last().expect("Disconnected trace"),
new_trace,
),
}
}
}