From 588be32d9817a67933417836c870d30d3aaf3448 Mon Sep 17 00:00:00 2001 From: Matt Corallo Date: Mon, 26 Aug 2019 16:27:19 -0400 Subject: [PATCH] Support scanning by .onion hostname --- src/reader.rs | 70 +++++++++++++++++++++++++++++++++++++++++++++++++-- 1 file changed, 68 insertions(+), 2 deletions(-) diff --git a/src/reader.rs b/src/reader.rs index 08dc5f0..0098ef4 100644 --- a/src/reader.rs +++ b/src/reader.rs @@ -1,7 +1,7 @@ use std::sync::Arc; use std::sync::atomic::Ordering; use std::io::BufReader; -use std::net::{IpAddr, SocketAddr}; +use std::net::{IpAddr, Ipv6Addr, SocketAddr}; use std::time::Instant; use tokio::prelude::*; @@ -15,6 +15,56 @@ use crate::{START_SHUTDOWN, scan_node}; use regex::Regex; +// base32 decoder and tests stolen (transliterated) from Bitcoin Core +// Copyright (c) 2012-2019 The Bitcoin Core developers +// Distributed under the MIT software license, see +// http://www.opensource.org/licenses/mit-license.php. +fn decode_base32(inp: &[u8]) -> Option> { + let decode32_table: [i8; 256] = [ + -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 26, 27, 28, 29, 30, 31, -1, -1, -1, -1, + -1, -1, -1, -1, -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, + 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1, -1, 0, 1, 2, + 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, + 23, 24, 25, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1 + ]; + + let mut ret = Vec::with_capacity((inp.len() * 5) / 8); + + let mut acc: u16 = 0; + let mut bits: u8 = 0; + for i in inp { + if *i == '=' as u8 { break; } + let codepoint = decode32_table[*i as usize]; + if codepoint < 0 { return None; } + acc = ((acc << 5) | codepoint as u16) & ((1 << (8 + 5 - 1)) - 1); + bits += 5; + while bits >= 8 { + bits -= 8; + ret.push((acc >> bits) as u8); + } + } + Some(ret) +} + +#[test] +fn test_decode_base32() { + let tests_in = ["","f","fo","foo","foob","fooba","foobar"]; + let tests_out = ["","my======","mzxq====","mzxw6===","mzxw6yq=","mzxw6ytb","mzxw6ytboi======"]; + for (inp, out) in tests_in.iter().zip(tests_out.iter()) { + assert_eq!(&decode_base32(out.as_bytes()).unwrap()[..], inp.as_bytes()); + } + // My seednode's onion addr: + assert_eq!(decode_base32("nkf5e6b7pl4jfd4a".as_bytes()).unwrap()[..],[0x6a, 0x8b, 0xd2, 0x78, 0x3f, 0x7a, 0xf8, 0x92, 0x8f, 0x80]); +} + pub fn read(store: &'static Store, printer: &'static Printer, bgp_client: Arc) { tokio::spawn(lines(BufReader::new(stdin())).for_each(move |line| { macro_rules! err { @@ -54,7 +104,23 @@ pub fn read(store: &'static Store, printer: &'static Printer, bgp_client: Arc err!(), }); }, - "a" => scan_node(Instant::now(), try_parse_next_chunk!(SocketAddr), true), + "a" => { + let host_port = get_next_chunk!(); + let parsed = if host_port.len() > 23 && &host_port[16..23] == ".onion:" { + let port = match host_port[23..].parse::() { Ok(res) => res, Err(_) => err!(), }; + + let ipv6 = match decode_base32(host_port[0..16].as_bytes()) { Some(res) => res, None => err!(), }; + if ipv6.len() != 10 { err!(); } + let mut octets = [0xFD,0x87,0xD8,0x7E,0xEB,0x43, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]; + octets[6..].copy_from_slice(&ipv6[0..10]); + + SocketAddr::new(IpAddr::V6(Ipv6Addr::from(octets)), port) + } else { + match host_port.parse::() { + Ok(res) => res, Err(_) => err!(), } + }; + scan_node(Instant::now(), parsed, true) + }, "b" => { let ip = try_parse_next_chunk!(IpAddr); printer.add_line(format!("ASN for {} is {}", ip, bgp_client.get_asn(ip)), false); -- 2.39.5