gprobe 0.1.2

A verbose CLI to probe go-ethereum data structures.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

use std::path::Path;

use rusty_leveldb::{LdbIterator, Options, DB};
use spinners::Spinner;

use crate::utils;

/// Walks the tree in a depth-first manner, calling the given function on each node.
pub fn walk(path: impl AsRef<Path>, level: u64, spinner: &mut Option<Spinner>) {
    let db_name = path.as_ref().to_string_lossy();
    tracing::debug!("Walking db \"{}\" tree to a depth of {}", db_name, level);

    // Cleanup the database directory
    if let Err(e) = utils::cleanup_db_dir(&db_name) {
        tracing::error!("Failed to cleanup database directory: {}", e);
        return;
    }

    // Open the leveldb-rs database at the given path
    let open_options = Options {
        create_if_missing: false,
        ..Default::default()
    };
    let mut db = DB::open(path.as_ref(), open_options).unwrap();

    // Create a new db iterator
    let mut iter = match db.new_iter() {
        Ok(iter) => iter,
        Err(e) => {
            tracing::error!("Failed to create leveldb iterator: {}", e);
            return;
        }
    };

    // Stop the spinner so we can print
    if let Some(sp) = spinner {
        sp.stop();
        println!();
    }
    *spinner = None;

    // Warn if db is empty
    if iter.next().is_none() {
        tracing::warn!("Error: Database is empty");
    }
    iter.reset();

    // Walk the database
    while let Some((k, v)) = iter.next() {
        let key = format!("0x{}", hex::encode(&k));
        let value = format!("0x{}", hex::encode(&v));
        println!("Key: {}\nValue: {}\n", key, value);
    }
    println!("Finished walking db \"{}\"", db_name);
}