X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=src%2Freader.rs;fp=src%2Freader.rs;h=a6a1fb0a0280ddbbe33c02b9df928cbd639de1ff;hb=8808d87877909d7d4f97d7ff320d8d165ece5800;hp=0000000000000000000000000000000000000000;hpb=405c10047bad68f7f2fabcf975ca0574444a7d47;p=dnsseed-rust

diff --git a/src/reader.rs b/src/reader.rs
new file mode 100644
index 0000000..a6a1fb0
--- /dev/null
+++ b/src/reader.rs
@@ -0,0 +1,75 @@
+use std::sync::atomic::Ordering;
+use std::io::BufReader;
+use std::net::SocketAddr;
+use std::time::Instant;
+
+use tokio::prelude::*;
+use tokio::io::{stdin, lines};
+
+use crate::printer::Printer;
+use crate::datastore::{Store, AddressState, U64Setting, RegexSetting};
+
+use crate::{START_SHUTDOWN, scan_node};
+
+use regex::Regex;
+
+pub fn read(store: &'static Store, printer: &'static Printer) {
+	tokio::spawn(lines(BufReader::new(stdin())).for_each(move |line| {
+		macro_rules! err {
+			() => { {
+				printer.add_line(format!("Unparsable input: \"{}\"", line), true);
+				return future::ok(());
+			} }
+		}
+		let mut line_iter = line.split(' ');
+		macro_rules! get_next_chunk {
+			() => { {
+				match line_iter.next() {
+					Some(c) => c,
+					None => err!(),
+				}
+			} }
+		}
+		macro_rules! try_parse_next_chunk {
+			($type: ty) => { {
+				match get_next_chunk!().parse::<$type>() {
+					Ok(res) => res,
+					Err(_) => err!(),
+				}
+			} }
+		}
+		match get_next_chunk!() {
+			"c" => store.set_u64(U64Setting::ConnsPerSec, try_parse_next_chunk!(u64)),
+			"t" => store.set_u64(U64Setting::RunTimeout, try_parse_next_chunk!(u64)),
+			"v" => store.set_u64(U64Setting::MinProtocolVersion, try_parse_next_chunk!(u64)),
+			"w" => store.set_u64(U64Setting::WasGoodTimeout, try_parse_next_chunk!(u64)),
+			"s" => store.set_regex(RegexSetting::SubverRegex, try_parse_next_chunk!(Regex)),
+			"a" => scan_node(Instant::now(), try_parse_next_chunk!(SocketAddr)),
+			"r" => {
+				match try_parse_next_chunk!(u8) {
+					0 => store.set_u64(U64Setting::RescanInterval(AddressState::Untested), try_parse_next_chunk!(u64)),
+					1 => store.set_u64(U64Setting::RescanInterval(AddressState::LowBlockCount), try_parse_next_chunk!(u64)),
+					2 => store.set_u64(U64Setting::RescanInterval(AddressState::HighBlockCount), try_parse_next_chunk!(u64)),
+					3 => store.set_u64(U64Setting::RescanInterval(AddressState::LowVersion), try_parse_next_chunk!(u64)),
+					4 => store.set_u64(U64Setting::RescanInterval(AddressState::BadVersion), try_parse_next_chunk!(u64)),
+					5 => store.set_u64(U64Setting::RescanInterval(AddressState::NotFullNode), try_parse_next_chunk!(u64)),
+					6 => store.set_u64(U64Setting::RescanInterval(AddressState::ProtocolViolation), try_parse_next_chunk!(u64)),
+					7 => store.set_u64(U64Setting::RescanInterval(AddressState::Timeout), try_parse_next_chunk!(u64)),
+					8 => store.set_u64(U64Setting::RescanInterval(AddressState::TimeoutDuringRequest), try_parse_next_chunk!(u64)),
+					9 => store.set_u64(U64Setting::RescanInterval(AddressState::Good), try_parse_next_chunk!(u64)),
+					10 => store.set_u64(U64Setting::RescanInterval(AddressState::WasGood), try_parse_next_chunk!(u64)),
+					_ => err!(),
+				}
+			},
+			"q" => {
+				START_SHUTDOWN.store(true, Ordering::SeqCst);
+				return future::err(std::io::Error::new(std::io::ErrorKind::Other, ""));
+			},
+			_ => err!(),
+		}
+		future::ok(())
+	}).then(|_| {
+		println!("Exiting");
+		future::ok(())
+	}));
+}