X-Git-Url: http://git.bitcoin.ninja/index.cgi?p=dnsseed-rust;a=blobdiff_plain;f=src%2Fmain.rs;fp=src%2Fmain.rs;h=2dc2f8c327a33fded79f0f54a46de8ee704625fd;hp=9c6e68155f584252805a15240e044ca8b13b06c4;hb=ee912fdc23d6ade355b38629488e42aac540f872;hpb=5767a129c52ab94c8b23291a0ce0dda46927eb95 diff --git a/src/main.rs b/src/main.rs index 9c6e681..2dc2f8c 100644 --- a/src/main.rs +++ b/src/main.rs @@ -469,8 +469,8 @@ fn make_trusted_conn(trusted_sockaddr: SocketAddr, bgp_client: Arc) { } fn main() { - if env::args().len() != 5 { - println!("USAGE: dnsseed-rust datastore localPeerAddress tor_proxy_addr bgp_peer"); + if env::args().len() != 6 { + println!("USAGE: dnsseed-rust datastore localPeerAddress tor_proxy_addr bgp_peer bgp_peer_asn"); return; } @@ -495,13 +495,14 @@ fn main() { unsafe { TOR_PROXY = Some(tor_socks5_sockaddr); } let bgp_sockaddr: SocketAddr = args.next().unwrap().parse().unwrap(); + let bgp_peerasn: u32 = args.next().unwrap().parse().unwrap(); Store::new(path).and_then(move |store| { unsafe { DATA_STORE = Some(Box::new(store)) }; let store = unsafe { DATA_STORE.as_ref().unwrap() }; unsafe { PRINTER = Some(Box::new(Printer::new(store))) }; - let bgp_client = BGPClient::new(bgp_sockaddr, Duration::from_secs(60), unsafe { PRINTER.as_ref().unwrap() }); + let bgp_client = BGPClient::new(bgp_peerasn, bgp_sockaddr, Duration::from_secs(60), unsafe { PRINTER.as_ref().unwrap() }); make_trusted_conn(trusted_sockaddr, Arc::clone(&bgp_client)); reader::read(store, unsafe { PRINTER.as_ref().unwrap() }, bgp_client);