Minor bug fixes to wallet extract tool

This commit is contained in:
Hanh 2022-09-26 19:17:59 +08:00
parent 2451d6b29f
commit c6f17da93f
1 changed files with 9 additions and 3 deletions

View File

@ -14,12 +14,17 @@ fn main() -> Result<()> {
let reader = BufReader::new(file); let reader = BufReader::new(file);
let mut started = false; let mut started = false;
let mut key = String::new(); let mut key = String::new();
let mut next_line_is_key = true;
for (i, line) in reader.lines().enumerate() { for (i, line) in reader.lines().enumerate() {
let ln = line?.trim_start().to_string(); let ln = line?.trim_start().to_string();
if !started && ln != "HEADER=END" { continue; } // skip header if ln == "HEADER=END" {
started = true; started = true;
next_line_is_key = true;
continue;
} // skip header
if !started { continue; }
if ln == "DATA=END" { break } // stop at data end if ln == "DATA=END" { break } // stop at data end
if i % 2 == 1 { if next_line_is_key {
let k = hex::decode(ln).unwrap(); let k = hex::decode(ln).unwrap();
let len = k[0] as usize; let len = k[0] as usize;
key = String::from_utf8_lossy(&k[1..=len]).to_string(); // collect key name key = String::from_utf8_lossy(&k[1..=len]).to_string(); // collect key name
@ -32,6 +37,7 @@ fn main() -> Result<()> {
println!("{}", encode_extended_spending_key(MainNetwork.hrp_sapling_extended_spending_key(), &s)); println!("{}", encode_extended_spending_key(MainNetwork.hrp_sapling_extended_spending_key(), &s));
} }
} }
next_line_is_key = !next_line_is_key;
} }
Ok(()) Ok(())