2018-06-13 18:42:30 -07:00
|
|
|
extern crate atty;
|
2018-03-26 21:07:11 -07:00
|
|
|
extern crate env_logger;
|
2018-04-09 11:28:54 -07:00
|
|
|
extern crate getopts;
|
2018-06-18 15:49:41 -07:00
|
|
|
extern crate log;
|
2018-03-03 23:13:40 -08:00
|
|
|
extern crate serde_json;
|
2018-03-27 15:24:05 -07:00
|
|
|
extern crate solana;
|
2018-02-28 17:04:35 -08:00
|
|
|
|
2018-06-13 15:07:36 -07:00
|
|
|
use atty::{is, Stream};
|
2018-06-13 18:42:30 -07:00
|
|
|
use getopts::Options;
|
2018-05-14 14:33:11 -07:00
|
|
|
use solana::bank::Bank;
|
2018-05-31 15:06:04 -07:00
|
|
|
use solana::crdt::ReplicatedData;
|
2018-04-02 13:51:38 -07:00
|
|
|
use solana::entry::Entry;
|
2018-05-29 12:55:21 -07:00
|
|
|
use solana::payment_plan::PaymentPlan;
|
2018-05-15 10:00:01 -07:00
|
|
|
use solana::server::Server;
|
2018-05-22 20:42:04 -07:00
|
|
|
use solana::transaction::Instruction;
|
2018-04-11 19:24:14 -07:00
|
|
|
use std::env;
|
2018-05-23 13:03:19 -07:00
|
|
|
use std::fs::File;
|
2018-06-19 09:10:51 -07:00
|
|
|
use std::io::{stdin, stdout, BufRead, Write};
|
2018-05-31 15:06:04 -07:00
|
|
|
use std::net::{IpAddr, Ipv4Addr, SocketAddr, UdpSocket};
|
2018-04-19 07:06:19 -07:00
|
|
|
use std::process::exit;
|
2018-05-11 12:06:05 -07:00
|
|
|
use std::sync::atomic::AtomicBool;
|
2018-05-29 20:20:28 -07:00
|
|
|
use std::sync::Arc;
|
2018-05-24 23:18:41 -07:00
|
|
|
//use std::time::Duration;
|
2018-02-28 17:04:35 -08:00
|
|
|
|
2018-04-19 10:16:20 -07:00
|
|
|
fn print_usage(program: &str, opts: Options) {
|
|
|
|
let mut brief = format!("Usage: cat <transaction.log> | {} [options]\n\n", program);
|
|
|
|
brief += " Run a Solana node to handle transactions and\n";
|
|
|
|
brief += " write a new transaction log to stdout.\n";
|
|
|
|
brief += " Takes existing transaction log from stdin.";
|
|
|
|
|
|
|
|
print!("{}", opts.usage(&brief));
|
|
|
|
}
|
|
|
|
|
2018-02-28 17:04:35 -08:00
|
|
|
fn main() {
|
2018-06-14 15:45:14 -07:00
|
|
|
env_logger::init();
|
2018-04-09 11:28:54 -07:00
|
|
|
let mut opts = Options::new();
|
2018-04-19 10:16:20 -07:00
|
|
|
opts.optflag("h", "help", "print help");
|
2018-06-14 22:47:08 -07:00
|
|
|
opts.optopt("l", "", "run with the identity found in FILE", "FILE");
|
2018-06-23 16:08:53 -07:00
|
|
|
opts.optopt(
|
|
|
|
"t",
|
|
|
|
"",
|
2018-06-23 16:44:56 -07:00
|
|
|
"testnet; connect to the network at this gossip entry point",
|
2018-06-23 16:43:26 -07:00
|
|
|
"HOST:PORT",
|
2018-06-23 16:08:53 -07:00
|
|
|
);
|
2018-05-23 13:03:19 -07:00
|
|
|
opts.optopt(
|
2018-06-14 22:47:08 -07:00
|
|
|
"o",
|
2018-05-23 13:03:19 -07:00
|
|
|
"",
|
2018-06-14 22:47:08 -07:00
|
|
|
"output log to FILE, defaults to stdout (ignored by validators)",
|
|
|
|
"FILE",
|
2018-05-23 13:03:19 -07:00
|
|
|
);
|
2018-06-14 22:47:08 -07:00
|
|
|
|
2018-04-09 11:28:54 -07:00
|
|
|
let args: Vec<String> = env::args().collect();
|
|
|
|
let matches = match opts.parse(&args[1..]) {
|
|
|
|
Ok(m) => m,
|
2018-04-19 07:06:19 -07:00
|
|
|
Err(e) => {
|
|
|
|
eprintln!("{}", e);
|
|
|
|
exit(1);
|
|
|
|
}
|
2018-04-09 11:28:54 -07:00
|
|
|
};
|
2018-04-19 10:16:20 -07:00
|
|
|
if matches.opt_present("h") {
|
|
|
|
let program = args[0].clone();
|
|
|
|
print_usage(&program, opts);
|
|
|
|
return;
|
|
|
|
}
|
2018-06-13 15:07:36 -07:00
|
|
|
if is(Stream::Stdin) {
|
2018-04-21 06:12:57 -07:00
|
|
|
eprintln!("nothing found on stdin, expected a log file");
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
|
2018-04-16 13:43:32 -07:00
|
|
|
eprintln!("Initializing...");
|
2018-06-19 09:10:51 -07:00
|
|
|
let stdin = stdin();
|
|
|
|
let mut entries = stdin.lock().lines().map(|line| {
|
|
|
|
let entry: Entry = serde_json::from_str(&line.unwrap()).unwrap_or_else(|e| {
|
2018-04-19 07:55:47 -07:00
|
|
|
eprintln!("failed to parse json: {}", e);
|
|
|
|
exit(1);
|
2018-06-19 09:10:51 -07:00
|
|
|
});
|
|
|
|
entry
|
2018-04-19 07:55:47 -07:00
|
|
|
});
|
2018-05-04 11:11:39 -07:00
|
|
|
eprintln!("done parsing...");
|
|
|
|
|
2018-04-02 13:51:38 -07:00
|
|
|
// The first item in the ledger is required to be an entry with zero num_hashes,
|
|
|
|
// which implies its id can be used as the ledger's seed.
|
2018-06-19 09:10:51 -07:00
|
|
|
let entry0 = entries.next().expect("invalid ledger: empty");
|
2018-04-02 13:51:38 -07:00
|
|
|
|
|
|
|
// The second item in the ledger is a special transaction where the to and from
|
|
|
|
// fields are the same. That entry should be treated as a deposit, not a
|
|
|
|
// transfer to oneself.
|
2018-06-19 09:10:51 -07:00
|
|
|
let entry1 = entries
|
|
|
|
.next()
|
|
|
|
.expect("invalid ledger: need at least 2 entries");
|
2018-05-25 15:05:37 -07:00
|
|
|
let tx = &entry1.transactions[0];
|
|
|
|
let deposit = if let Instruction::NewContract(contract) = &tx.instruction {
|
2018-05-23 22:53:10 -07:00
|
|
|
contract.plan.final_payment()
|
2018-04-02 13:51:38 -07:00
|
|
|
} else {
|
|
|
|
None
|
2018-06-19 09:10:51 -07:00
|
|
|
}.expect("invalid ledger, needs to start with a contract");
|
2018-04-02 13:51:38 -07:00
|
|
|
|
2018-05-14 14:33:11 -07:00
|
|
|
eprintln!("creating bank...");
|
2018-05-04 11:11:39 -07:00
|
|
|
|
2018-06-19 09:10:51 -07:00
|
|
|
let bank = Bank::new_from_deposit(&deposit);
|
2018-05-14 14:33:11 -07:00
|
|
|
bank.register_entry_id(&entry0.id);
|
|
|
|
bank.register_entry_id(&entry1.id);
|
2018-04-02 13:51:38 -07:00
|
|
|
|
2018-06-27 12:35:58 -07:00
|
|
|
// entry_height is the network-wide agreed height of the ledger.
|
|
|
|
// initialize it from the input ledger
|
2018-05-04 11:11:39 -07:00
|
|
|
eprintln!("processing entries...");
|
2018-06-27 12:35:58 -07:00
|
|
|
let entry_height = bank.process_entries(entries).expect("process_entries");
|
|
|
|
eprintln!("processed {} entries...", entry_height);
|
2018-04-02 13:51:38 -07:00
|
|
|
|
2018-05-04 11:11:39 -07:00
|
|
|
eprintln!("creating networking stack...");
|
|
|
|
|
2018-05-31 15:06:04 -07:00
|
|
|
let bind_addr = SocketAddr::new(IpAddr::V4(Ipv4Addr::new(0, 0, 0, 0)), 8000);
|
2018-05-31 14:11:16 -07:00
|
|
|
let mut repl_data = ReplicatedData::new_leader(&bind_addr);
|
2018-05-24 23:18:41 -07:00
|
|
|
if matches.opt_present("l") {
|
|
|
|
let path = matches.opt_str("l").unwrap();
|
2018-05-30 07:17:04 -07:00
|
|
|
if let Ok(file) = File::open(path.clone()) {
|
|
|
|
if let Ok(data) = serde_json::from_reader(file) {
|
|
|
|
repl_data = data;
|
|
|
|
} else {
|
2018-06-16 09:35:30 -07:00
|
|
|
eprintln!("failed to parse {}", path);
|
|
|
|
exit(1);
|
2018-05-30 07:17:04 -07:00
|
|
|
}
|
2018-06-15 15:29:43 -07:00
|
|
|
} else {
|
2018-06-16 09:35:30 -07:00
|
|
|
eprintln!("failed to read {}", path);
|
|
|
|
exit(1);
|
2018-05-24 23:18:41 -07:00
|
|
|
}
|
|
|
|
}
|
2018-06-15 15:29:43 -07:00
|
|
|
|
2018-06-28 16:39:51 -07:00
|
|
|
let mut local_gossip_addr = bind_addr.clone();
|
|
|
|
local_gossip_addr.set_port(repl_data.gossip_addr.port());
|
|
|
|
|
|
|
|
let mut local_replicate_addr = bind_addr.clone();
|
|
|
|
local_replicate_addr.set_port(repl_data.replicate_addr.port());
|
|
|
|
|
|
|
|
let mut local_requests_addr = bind_addr.clone();
|
|
|
|
local_requests_addr.set_port(repl_data.requests_addr.port());
|
|
|
|
|
|
|
|
let mut local_transactions_addr = bind_addr.clone();
|
|
|
|
local_transactions_addr.set_port(repl_data.transactions_addr.port());
|
|
|
|
|
|
|
|
let mut local_repair_addr = bind_addr.clone();
|
|
|
|
local_repair_addr.set_port(repl_data.repair_addr.port());
|
|
|
|
|
2018-06-16 09:35:30 -07:00
|
|
|
let exit = Arc::new(AtomicBool::new(false));
|
2018-06-23 16:08:53 -07:00
|
|
|
let threads = if matches.opt_present("t") {
|
2018-06-23 17:01:38 -07:00
|
|
|
let testnet_address_string = matches.opt_str("t").unwrap();
|
2018-06-15 15:29:43 -07:00
|
|
|
eprintln!(
|
2018-06-28 16:39:51 -07:00
|
|
|
"starting validator... {} (advertising {}) connecting to {}",
|
|
|
|
local_requests_addr, repl_data.requests_addr, testnet_address_string
|
2018-06-15 15:29:43 -07:00
|
|
|
);
|
2018-06-23 17:01:38 -07:00
|
|
|
let testnet_addr = testnet_address_string.parse().unwrap();
|
|
|
|
let newtwork_entry_point = ReplicatedData::new_entry_point(testnet_addr);
|
2018-05-23 13:03:19 -07:00
|
|
|
let s = Server::new_validator(
|
|
|
|
bank,
|
2018-06-27 12:35:58 -07:00
|
|
|
entry_height,
|
2018-05-23 13:03:19 -07:00
|
|
|
repl_data.clone(),
|
2018-06-28 16:39:51 -07:00
|
|
|
UdpSocket::bind(local_requests_addr).unwrap(),
|
2018-05-23 13:03:19 -07:00
|
|
|
UdpSocket::bind("0.0.0.0:0").unwrap(),
|
2018-06-28 16:39:51 -07:00
|
|
|
UdpSocket::bind(local_replicate_addr).unwrap(),
|
|
|
|
UdpSocket::bind(local_gossip_addr).unwrap(),
|
|
|
|
UdpSocket::bind(local_repair_addr).unwrap(),
|
2018-06-23 17:01:38 -07:00
|
|
|
newtwork_entry_point,
|
2018-05-23 13:03:19 -07:00
|
|
|
exit.clone(),
|
|
|
|
);
|
|
|
|
s.thread_hdls
|
|
|
|
} else {
|
2018-06-28 16:39:51 -07:00
|
|
|
eprintln!(
|
|
|
|
"starting leader... {} (advertising {})",
|
|
|
|
local_requests_addr, repl_data.requests_addr
|
|
|
|
);
|
2018-05-23 13:03:19 -07:00
|
|
|
repl_data.current_leader_id = repl_data.id.clone();
|
2018-06-14 22:47:08 -07:00
|
|
|
|
|
|
|
let outfile: Box<Write + Send + 'static> = if matches.opt_present("o") {
|
|
|
|
let path = matches.opt_str("o").unwrap();
|
|
|
|
Box::new(
|
|
|
|
File::create(&path).expect(&format!("unable to open output file \"{}\"", path)),
|
|
|
|
)
|
|
|
|
} else {
|
|
|
|
Box::new(stdout())
|
|
|
|
};
|
|
|
|
|
2018-05-23 13:03:19 -07:00
|
|
|
let server = Server::new_leader(
|
|
|
|
bank,
|
2018-06-27 12:35:58 -07:00
|
|
|
entry_height,
|
2018-05-24 23:18:41 -07:00
|
|
|
//Some(Duration::from_millis(1000)),
|
|
|
|
None,
|
2018-05-23 13:03:19 -07:00
|
|
|
repl_data.clone(),
|
2018-06-28 16:39:51 -07:00
|
|
|
UdpSocket::bind(local_requests_addr).unwrap(),
|
|
|
|
UdpSocket::bind(local_transactions_addr).unwrap(),
|
2018-05-23 13:03:19 -07:00
|
|
|
UdpSocket::bind("0.0.0.0:0").unwrap(),
|
|
|
|
UdpSocket::bind("0.0.0.0:0").unwrap(),
|
2018-06-28 16:39:51 -07:00
|
|
|
UdpSocket::bind(local_gossip_addr).unwrap(),
|
2018-05-23 13:03:19 -07:00
|
|
|
exit.clone(),
|
2018-06-14 22:47:08 -07:00
|
|
|
outfile,
|
2018-05-23 13:03:19 -07:00
|
|
|
);
|
|
|
|
server.thread_hdls
|
|
|
|
};
|
2018-06-28 16:39:51 -07:00
|
|
|
eprintln!(
|
|
|
|
"Ready. Listening on {} (advertising {})",
|
|
|
|
local_transactions_addr, repl_data.transactions_addr
|
|
|
|
);
|
2018-05-23 13:03:19 -07:00
|
|
|
|
|
|
|
for t in threads {
|
|
|
|
t.join().expect("join");
|
|
|
|
}
|
|
|
|
}
|