parent
c8ad7f0086
commit
c1c31367e0
GPG Key ID:
446580D758689584
|
|
@ -0,0 +1,14 @@
|
|||
{
|
||||
"configurations": [
|
||||
{
|
||||
"type": "cargo",
|
||||
"name": "Debug",
|
||||
"cargoArgs": ["run"],
|
||||
},
|
||||
{
|
||||
"type": "cargo",
|
||||
"name": "Release",
|
||||
"cargoArgs": ["run", "-r"],
|
||||
},
|
||||
]
|
||||
}
|
||||
|
|
@ -2,6 +2,187 @@
|
|||
# It is not intended for manual editing.
|
||||
version = 3
|
||||
|
||||
[[package]]
|
||||
name = "argparse"
|
||||
version = "0.2.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "3f8ebf5827e4ac4fd5946560e6a99776ea73b596d80898f357007317a7141e47"
|
||||
|
||||
[[package]]
|
||||
name = "cfg-if"
|
||||
version = "1.0.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
|
||||
|
||||
[[package]]
|
||||
name = "fuchsia-cprng"
|
||||
version = "0.1.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "a06f77d526c1a601b7c4cdd98f54b5eaabffc14d5f2f0296febdc7f357c6d3ba"
|
||||
|
||||
[[package]]
|
||||
name = "idna"
|
||||
version = "0.1.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "38f09e0f0b1fb55fdee1f17470ad800da77af5186a1a76c026b679358b7e844e"
|
||||
dependencies = [
|
||||
"matches",
|
||||
"unicode-bidi",
|
||||
"unicode-normalization",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "json"
|
||||
version = "0.12.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "078e285eafdfb6c4b434e0d31e8cfcb5115b651496faca5749b88fafd4f23bfd"
|
||||
|
||||
[[package]]
|
||||
name = "libc"
|
||||
version = "0.2.139"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "201de327520df007757c1f0adce6e827fe8562fbc28bfd9c15571c66ca1f5f79"
|
||||
|
||||
[[package]]
|
||||
name = "log"
|
||||
version = "0.3.9"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "e19e8d5c34a3e0e2223db8e060f9e8264aeeb5c5fc64a4ee9965c062211c024b"
|
||||
dependencies = [
|
||||
"log 0.4.17",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "log"
|
||||
version = "0.4.17"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "abb12e687cfb44aa40f41fc3978ef76448f9b6038cad6aef4259d3c095a2382e"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "matches"
|
||||
version = "0.1.10"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2532096657941c2fea9c289d370a250971c689d4f143798ff67113ec042024a5"
|
||||
|
||||
[[package]]
|
||||
name = "rand"
|
||||
version = "0.3.23"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "64ac302d8f83c0c1974bf758f6b041c6c8ada916fbb44a609158ca8b064cc76c"
|
||||
dependencies = [
|
||||
"libc",
|
||||
"rand 0.4.6",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "rand"
|
||||
version = "0.4.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "552840b97013b1a26992c11eac34bdd778e464601a4c2054b5f0bff7c6761293"
|
||||
dependencies = [
|
||||
"fuchsia-cprng",
|
||||
"libc",
|
||||
"rand_core 0.3.1",
|
||||
"rdrand",
|
||||
"winapi",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "rand_core"
|
||||
version = "0.3.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "7a6fdeb83b075e8266dcc8762c22776f6877a63111121f5f8c7411e5be7eed4b"
|
||||
dependencies = [
|
||||
"rand_core 0.4.2",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "rand_core"
|
||||
version = "0.4.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "9c33a3c44ca05fa6f1807d8e6743f3824e8509beca625669633be0acbdf509dc"
|
||||
|
||||
[[package]]
|
||||
name = "rdrand"
|
||||
version = "0.4.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "678054eb77286b51581ba43620cc911abf02758c91f93f479767aed0f90458b2"
|
||||
dependencies = [
|
||||
"rand_core 0.3.1",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "resdns"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"argparse",
|
||||
"json",
|
||||
"resolve",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "resolve"
|
||||
version = "0.2.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "19526b305899bea65f26edda78a64f5313958494321ee0ab66bd94b32958614a"
|
||||
dependencies = [
|
||||
"idna",
|
||||
"libc",
|
||||
"log 0.3.9",
|
||||
"rand 0.3.23",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tinyvec"
|
||||
version = "1.6.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "87cc5ceb3875bb20c2890005a4e226a4651264a5c75edb2421b52861a0a0cb50"
|
||||
dependencies = [
|
||||
"tinyvec_macros",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tinyvec_macros"
|
||||
version = "0.1.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "1f3ccbac311fea05f86f61904b462b55fb3df8837a366dfc601a0161d0532f20"
|
||||
|
||||
[[package]]
|
||||
name = "unicode-bidi"
|
||||
version = "0.3.10"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d54675592c1dbefd78cbd98db9bacd89886e1ca50692a0692baefffdeb92dd58"
|
||||
|
||||
[[package]]
|
||||
name = "unicode-normalization"
|
||||
version = "0.1.22"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5c5713f0fc4b5db668a2ac63cdb7bb4469d8c9fed047b1d0292cc7b0ce2ba921"
|
||||
dependencies = [
|
||||
"tinyvec",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "winapi"
|
||||
version = "0.3.9"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5c839a674fcd7a98952e593242ea400abe93992746761e38641405d28b00f419"
|
||||
dependencies = [
|
||||
"winapi-i686-pc-windows-gnu",
|
||||
"winapi-x86_64-pc-windows-gnu",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "winapi-i686-pc-windows-gnu"
|
||||
version = "0.4.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "ac3b87c63620426dd9b991e5ce0329eff545bccbbb34f3be09ff6fb6ab51b7b6"
|
||||
|
||||
[[package]]
|
||||
name = "winapi-x86_64-pc-windows-gnu"
|
||||
version = "0.4.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f"
|
||||
|
|
|
|||
|
|
@ -6,3 +6,6 @@ edition = "2021"
|
|||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
||||
[dependencies]
|
||||
argparse = "0.2.2"
|
||||
json = "0.12.4"
|
||||
resolve = "0.2.0"
|
||||
|
|
|
|||
171
src/main.rs
171
src/main.rs
|
|
@ -1,5 +1,172 @@
|
|||
use argparse::{ArgumentParser, List, Store};
|
||||
use resolve::{record, DnsConfig, DnsResolver};
|
||||
use std::time::Duration;
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
struct Record {
|
||||
pub host: String,
|
||||
pub addr: String,
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
enum OutputParser {
|
||||
Plain,
|
||||
JSON { pretty: bool },
|
||||
JSONPretty { pretty: bool },
|
||||
CSV,
|
||||
}
|
||||
|
||||
struct Resolver {
|
||||
resolver: DnsResolver,
|
||||
}
|
||||
|
||||
impl Resolver {
|
||||
pub fn init() -> Resolver {
|
||||
let config = DnsConfig {
|
||||
timeout: Duration::from_secs(3),
|
||||
attempts: 2,
|
||||
use_inet6: false,
|
||||
..DnsConfig::load_default().expect("Couldn't get default DnsConfig")
|
||||
};
|
||||
|
||||
Resolver {
|
||||
resolver: DnsResolver::new(config).expect("Couldn't create DNS resolver"),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn resolve(&self, host: &str) -> Vec<Record> {
|
||||
let rec4 = self
|
||||
.resolver
|
||||
.resolve_record::<record::A>(host)
|
||||
.unwrap_or(vec![]);
|
||||
let rec6 = self
|
||||
.resolver
|
||||
.resolve_record::<record::AAAA>(host)
|
||||
.unwrap_or(vec![]);
|
||||
|
||||
let mut recs: Vec<Record> = vec![];
|
||||
for r in rec4 {
|
||||
recs.insert(
|
||||
0,
|
||||
Record {
|
||||
host: host.to_owned(),
|
||||
addr: r.address.to_string(),
|
||||
},
|
||||
)
|
||||
}
|
||||
for r in rec6 {
|
||||
recs.insert(
|
||||
0,
|
||||
Record {
|
||||
host: host.to_owned(),
|
||||
addr: r.address.to_string(),
|
||||
},
|
||||
)
|
||||
}
|
||||
|
||||
recs.sort_unstable_by(|a, b| a.addr.len().cmp(&b.addr.len()));
|
||||
|
||||
recs
|
||||
}
|
||||
}
|
||||
|
||||
fn main() {
|
||||
println!("Hello, world!");
|
||||
}
|
||||
let mut output_flag = String::from("Default");
|
||||
let mut hosts: Vec<String> = vec![];
|
||||
|
||||
// Argument parser
|
||||
{
|
||||
let mut args = ArgumentParser::new();
|
||||
args.set_description("Resolve Host(s) to IP Address(es).");
|
||||
|
||||
// -o, --output, --parser
|
||||
args.refer(&mut output_flag).add_option(
|
||||
&["-o", "--output", "--parser"],
|
||||
Store,
|
||||
"Select how the response is outputted: [Default, Plain, JSON, JSON Pretty, CSV]",
|
||||
);
|
||||
|
||||
// Positional arg: hosts
|
||||
args.refer(&mut hosts)
|
||||
.add_argument("hosts", List, "Hosts to resolve")
|
||||
.required();
|
||||
|
||||
args.parse_args_or_exit();
|
||||
}
|
||||
|
||||
// OutputParser
|
||||
let output = match output_flag.to_lowercase().as_str() {
|
||||
"default" | "plain" | "" => OutputParser::Plain,
|
||||
"json" => OutputParser::JSON { pretty: false },
|
||||
"json-pretty" | "json_pretty" | "json pretty" | "jsonpretty" => {
|
||||
OutputParser::JSONPretty { pretty: true }
|
||||
}
|
||||
"csv" => OutputParser::CSV,
|
||||
_ => {
|
||||
eprintln!("[FATAL] Invalid output parser: \"{output_flag}\"");
|
||||
std::process::exit(1)
|
||||
}
|
||||
};
|
||||
|
||||
// No hosts
|
||||
if hosts.is_empty() {
|
||||
eprintln!("[Fatal] No hosts given");
|
||||
std::process::exit(1)
|
||||
}
|
||||
|
||||
// Resolver
|
||||
let resolver: Resolver = Resolver::init();
|
||||
|
||||
let mut record_pile: Vec<Record> = vec![];
|
||||
for h in hosts {
|
||||
record_pile.append(&mut resolver.resolve(h.as_str()));
|
||||
}
|
||||
|
||||
match output {
|
||||
// Plain / Default output
|
||||
OutputParser::Plain => {
|
||||
for r in record_pile {
|
||||
println!("{}: {}", r.host, r.addr)
|
||||
}
|
||||
}
|
||||
|
||||
// JSON / JSON Pretty output
|
||||
OutputParser::JSON { pretty } | OutputParser::JSONPretty { pretty } => {
|
||||
let mut rj = json::object! { records: {} };
|
||||
|
||||
for r in record_pile {
|
||||
if rj["records"][&r.host].is_null() {
|
||||
rj["records"]
|
||||
.insert(
|
||||
&r.host,
|
||||
json::parse(&format!("[\"{}\"]", r.addr))
|
||||
.expect("Unable to parse JSON"),
|
||||
)
|
||||
.expect("Unable to insert record into JSON");
|
||||
} else {
|
||||
rj["records"][&r.host]
|
||||
.push(r.addr)
|
||||
.expect("Unable to insert record into JSON");
|
||||
}
|
||||
}
|
||||
|
||||
println!(
|
||||
"{}",
|
||||
match pretty {
|
||||
true => rj.pretty(2),
|
||||
false => rj.dump(),
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
// CSV output
|
||||
OutputParser::CSV => {
|
||||
println!("\"Host\", \"Address\"");
|
||||
for r in record_pile {
|
||||
println!("\"{}\", \"{}\"", r.host, r.addr)
|
||||
}
|
||||
}
|
||||
|
||||
// _ => (),
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue