use std::env;
use std::process;
use std::sync::mpsc;
use std::thread;
use std::time::Duration;
use std::net::{IpAddr, ToSocketAddrs};
use ::mping::{ping, PingOption};
fn main() {
let args: Vec<String> = env::args().collect();
if args.len() != 2 {
println!("Usage: {} <ip-address>", args[0]);
process::exit(1);
}
let pid = process::id() as u16;
let target = args[1].clone();
let addr = parse_ip(&target);
let addrs = vec![addr.parse().unwrap()];
let popt = PingOption {
timeout: Duration::from_secs(1),
ttl: 64,
tos: None,
ident: pid,
len: 56,
rate: 100,
rate_for_all: false,
delay: 3,
count: None,
};
let (tx, rx) = mpsc::channel();
thread::spawn(move || match ping(addrs, popt, false, Some(tx)) {
Ok(_) => {}
Err(e) => {
println!("error: {:?}", e);
}
});
for tr in rx {
let total = tr.received + tr.loss;
let loss_rate = if total == 0 {
0.0
} else {
(tr.loss as f64) / (total as f64)
};
if tr.received == 0 {
println!(
"{}: sent:{}, recv:{}, loss rate: {:.2}%, latency: {}ms",
addr,
total,
tr.received,
loss_rate * 100.0,
0
)
} else {
println!(
"{}: sent:{}, recv:{}, loss rate: {:.2}%, latency: {:.2}ms",
addr,
total,
tr.received,
loss_rate * 100.0,
Duration::from_nanos(tr.latency as u64 / (tr.received as u64)).as_secs_f64()
* 1000.0
)
}
}
}
fn parse_ip(s: &str) -> String {
if let Ok(_) = s.parse::<IpAddr>() {
return s.to_string();
} else if let Ok(addrs) = (s, 0).to_socket_addrs() {
for addr in addrs {
if let IpAddr::V4(ipv4) = addr.ip() {
return IpAddr::V4(ipv4).to_string();
}
}
}
s.to_string()
}