2018-07-09 18:04:49 -07:00
|
|
|
extern crate clap;
|
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-07-09 18:04:49 -07:00
|
|
|
use clap::{App, Arg};
|
2018-07-31 22:07:53 -07:00
|
|
|
use solana::client::mk_client;
|
2018-07-11 00:18:48 -07:00
|
|
|
use solana::crdt::{NodeInfo, TestNode};
|
2018-07-31 22:07:53 -07:00
|
|
|
use solana::drone::DRONE_PORT;
|
2018-07-12 16:09:58 -07:00
|
|
|
use solana::fullnode::{Config, FullNode, LedgerFile};
|
2018-07-27 21:37:53 -07:00
|
|
|
use solana::logger;
|
2018-07-17 10:48:46 -07:00
|
|
|
use solana::metrics::set_panic_hook;
|
2018-07-03 21:14:08 -07:00
|
|
|
use solana::service::Service;
|
2018-07-05 12:01:40 -07:00
|
|
|
use solana::signature::{KeyPair, KeyPairUtil};
|
2018-07-31 22:07:53 -07:00
|
|
|
use solana::wallet::request_airdrop;
|
2018-05-23 13:03:19 -07:00
|
|
|
use std::fs::File;
|
2018-07-02 11:20:35 -07:00
|
|
|
use std::net::{IpAddr, Ipv4Addr, SocketAddr};
|
2018-04-19 07:06:19 -07:00
|
|
|
use std::process::exit;
|
2018-05-24 23:18:41 -07:00
|
|
|
//use std::time::Duration;
|
2018-02-28 17:04:35 -08:00
|
|
|
|
2018-07-02 11:23:20 -07:00
|
|
|
fn main() -> () {
|
2018-07-27 21:37:53 -07:00
|
|
|
logger::setup();
|
2018-07-17 11:00:01 -07:00
|
|
|
set_panic_hook("fullnode");
|
2018-07-09 18:04:49 -07:00
|
|
|
let matches = App::new("fullnode")
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("identity")
|
2018-07-12 15:45:41 -07:00
|
|
|
.short("i")
|
|
|
|
.long("identity")
|
2018-07-09 18:04:49 -07:00
|
|
|
.value_name("FILE")
|
|
|
|
.takes_value(true)
|
|
|
|
.help("run with the identity found in FILE"),
|
|
|
|
)
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("testnet")
|
|
|
|
.short("t")
|
|
|
|
.long("testnet")
|
|
|
|
.value_name("HOST:PORT")
|
|
|
|
.takes_value(true)
|
2018-07-12 15:45:41 -07:00
|
|
|
.help("connect to the network at this gossip entry point"),
|
2018-07-09 18:04:49 -07:00
|
|
|
)
|
|
|
|
.arg(
|
2018-07-12 14:29:36 -07:00
|
|
|
Arg::with_name("ledger")
|
|
|
|
.short("L")
|
|
|
|
.long("ledger")
|
2018-07-09 18:04:49 -07:00
|
|
|
.value_name("FILE")
|
|
|
|
.takes_value(true)
|
2018-07-12 14:29:36 -07:00
|
|
|
.help("use FILE as persistent ledger (defaults to stdin/stdout)"),
|
2018-07-09 18:04:49 -07:00
|
|
|
)
|
|
|
|
.get_matches();
|
2018-04-21 06:12:57 -07:00
|
|
|
|
2018-05-31 15:06:04 -07:00
|
|
|
let bind_addr = SocketAddr::new(IpAddr::V4(Ipv4Addr::new(0, 0, 0, 0)), 8000);
|
2018-07-05 12:01:40 -07:00
|
|
|
let mut keypair = KeyPair::new();
|
2018-07-11 00:18:48 -07:00
|
|
|
let mut repl_data = NodeInfo::new_leader_with_pubkey(keypair.pubkey(), &bind_addr);
|
2018-07-12 15:45:41 -07:00
|
|
|
if let Some(i) = matches.value_of("identity") {
|
|
|
|
let path = i.to_string();
|
2018-05-30 07:17:04 -07:00
|
|
|
if let Ok(file) = File::open(path.clone()) {
|
2018-07-05 12:01:40 -07:00
|
|
|
let parse: serde_json::Result<Config> = serde_json::from_reader(file);
|
|
|
|
if let Ok(data) = parse {
|
|
|
|
keypair = data.keypair();
|
2018-07-10 12:02:51 -07:00
|
|
|
repl_data = data.node_info;
|
2018-05-30 07:17:04 -07:00
|
|
|
} 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-07-31 22:07:53 -07:00
|
|
|
|
|
|
|
let leader_pubkey = keypair.pubkey();
|
|
|
|
let repl_clone = repl_data.clone();
|
|
|
|
|
2018-07-12 14:29:36 -07:00
|
|
|
let ledger = if let Some(l) = matches.value_of("ledger") {
|
2018-07-12 16:09:58 -07:00
|
|
|
LedgerFile::Path(l.to_string())
|
2018-07-12 14:29:36 -07:00
|
|
|
} else {
|
2018-07-12 16:09:58 -07:00
|
|
|
LedgerFile::StdInOut
|
2018-07-12 14:29:36 -07:00
|
|
|
};
|
|
|
|
|
2018-07-02 14:46:23 -07:00
|
|
|
let mut node = TestNode::new_with_bind_addr(repl_data, bind_addr);
|
2018-08-01 15:09:47 -07:00
|
|
|
let mut drone_addr = SocketAddr::new(IpAddr::V4(Ipv4Addr::new(127, 0, 0, 0)), DRONE_PORT);
|
2018-07-09 18:04:49 -07:00
|
|
|
let fullnode = if let Some(t) = matches.value_of("testnet") {
|
|
|
|
let testnet_address_string = t.to_string();
|
2018-08-01 15:09:47 -07:00
|
|
|
let testnet_addr: SocketAddr = testnet_address_string.parse().unwrap();
|
|
|
|
drone_addr.set_ip(testnet_addr.ip());
|
2018-07-12 14:29:36 -07:00
|
|
|
|
2018-07-31 22:07:53 -07:00
|
|
|
FullNode::new(node, false, ledger, keypair, Some(testnet_addr))
|
2018-05-23 13:03:19 -07:00
|
|
|
} else {
|
2018-07-11 12:32:54 -07:00
|
|
|
node.data.leader_id = node.data.id;
|
2018-06-14 22:47:08 -07:00
|
|
|
|
2018-07-31 22:07:53 -07:00
|
|
|
FullNode::new(node, true, ledger, keypair, None)
|
2018-05-23 13:03:19 -07:00
|
|
|
};
|
2018-07-31 22:07:53 -07:00
|
|
|
|
|
|
|
let mut client = mk_client(&repl_clone);
|
|
|
|
let previous_balance = client.poll_get_balance(&leader_pubkey).unwrap();
|
|
|
|
eprintln!("balance is {}", previous_balance);
|
|
|
|
|
|
|
|
if previous_balance == 0 {
|
2018-08-01 15:09:47 -07:00
|
|
|
eprintln!("requesting airdrop from {}", drone_addr);
|
2018-07-31 22:07:53 -07:00
|
|
|
request_airdrop(&drone_addr, &leader_pubkey, 50).unwrap_or_else(|_| {
|
|
|
|
panic!(
|
|
|
|
"Airdrop failed, is the drone address correct {:?} drone running?",
|
|
|
|
drone_addr
|
|
|
|
)
|
|
|
|
});
|
|
|
|
|
|
|
|
let balance = client.poll_get_balance(&leader_pubkey).unwrap();
|
|
|
|
eprintln!("new balance is {}", balance);
|
|
|
|
assert!(balance > 0);
|
|
|
|
}
|
|
|
|
|
2018-07-03 21:14:08 -07:00
|
|
|
fullnode.join().expect("join");
|
2018-05-23 13:03:19 -07:00
|
|
|
}
|