2018-08-06 20:51:12 -07:00
|
|
|
#[macro_use]
|
2018-08-04 14:31:12 -07:00
|
|
|
extern crate clap;
|
|
|
|
extern crate serde_json;
|
|
|
|
extern crate solana;
|
|
|
|
|
2018-08-06 16:03:08 -07:00
|
|
|
use clap::{App, Arg, SubCommand};
|
|
|
|
use solana::bank::Bank;
|
2018-08-04 14:31:12 -07:00
|
|
|
use solana::ledger::read_ledger;
|
2018-08-06 17:33:51 -07:00
|
|
|
use solana::logger;
|
2018-08-04 14:31:12 -07:00
|
|
|
use std::io::{stdout, Write};
|
2018-08-06 16:03:08 -07:00
|
|
|
use std::process::exit;
|
2018-08-04 14:31:12 -07:00
|
|
|
|
|
|
|
fn main() {
|
2018-08-06 17:33:51 -07:00
|
|
|
logger::setup();
|
2018-08-06 16:03:08 -07:00
|
|
|
let matches = App::new("ledger-tool")
|
2018-08-06 20:51:12 -07:00
|
|
|
.version(crate_version!())
|
2018-08-04 14:31:12 -07:00
|
|
|
.arg(
|
|
|
|
Arg::with_name("ledger")
|
|
|
|
.short("l")
|
|
|
|
.long("ledger")
|
|
|
|
.value_name("DIR")
|
|
|
|
.takes_value(true)
|
|
|
|
.required(true)
|
|
|
|
.help("use DIR for ledger location"),
|
|
|
|
)
|
2018-08-06 16:03:08 -07:00
|
|
|
.subcommand(SubCommand::with_name("print").about("Print the ledger"))
|
|
|
|
.subcommand(SubCommand::with_name("verify").about("Verify the ledger"))
|
2018-08-04 14:31:12 -07:00
|
|
|
.get_matches();
|
|
|
|
|
|
|
|
let ledger_path = matches.value_of("ledger").unwrap();
|
2018-08-06 16:03:08 -07:00
|
|
|
let entries = match read_ledger(ledger_path) {
|
|
|
|
Ok(entries) => entries,
|
|
|
|
Err(err) => {
|
|
|
|
println!("Failed to open ledger at {}: {}", ledger_path, err);
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
};
|
2018-08-04 14:31:12 -07:00
|
|
|
|
2018-08-06 16:03:08 -07:00
|
|
|
match matches.subcommand() {
|
|
|
|
("print", _) => {
|
|
|
|
stdout().write_all(b"{\"ledger\":[\n").expect("open array");
|
|
|
|
for entry in entries {
|
|
|
|
let entry = entry.unwrap();
|
|
|
|
serde_json::to_writer(stdout(), &entry).expect("serialize");
|
|
|
|
stdout().write_all(b",\n").expect("newline");
|
|
|
|
}
|
|
|
|
stdout().write_all(b"\n]}\n").expect("close array");
|
|
|
|
}
|
|
|
|
("verify", _) => {
|
|
|
|
let entries = entries.map(|entry| entry.unwrap());
|
|
|
|
let bank = Bank::default();
|
|
|
|
let entry_height = bank.process_ledger(entries).expect("process_ledger").0;
|
|
|
|
println!("Ledger is valid. Height: {}", entry_height);
|
|
|
|
}
|
|
|
|
("", _) => {
|
|
|
|
println!("{}", matches.usage());
|
|
|
|
}
|
|
|
|
_ => unreachable!(),
|
|
|
|
};
|
2018-08-04 14:31:12 -07:00
|
|
|
}
|