2017-06-18 18:14:46 +08:00
|
|
|
#[macro_use]
|
|
|
|
extern crate log;
|
|
|
|
extern crate env_logger;
|
|
|
|
extern crate getopts;
|
|
|
|
extern crate smoltcp;
|
|
|
|
extern crate byteorder;
|
|
|
|
|
|
|
|
mod utils;
|
|
|
|
|
2017-09-25 08:55:54 +08:00
|
|
|
use std::str::FromStr;
|
2017-11-21 18:52:53 +08:00
|
|
|
use std::collections::BTreeMap;
|
2018-02-06 21:35:49 +08:00
|
|
|
use std::cmp;
|
2017-08-30 03:35:09 +08:00
|
|
|
use std::os::unix::io::AsRawFd;
|
2018-02-06 21:35:49 +08:00
|
|
|
use smoltcp::time::{Duration, Instant};
|
2017-10-02 18:47:51 +08:00
|
|
|
use smoltcp::phy::Device;
|
2017-08-30 03:35:09 +08:00
|
|
|
use smoltcp::phy::wait as phy_wait;
|
2017-10-30 08:19:34 +08:00
|
|
|
use smoltcp::wire::{EthernetAddress, IpAddress, IpCidr,
|
|
|
|
Ipv4Address, Icmpv4Repr, Icmpv4Packet};
|
2017-12-18 21:29:30 +08:00
|
|
|
use smoltcp::iface::{NeighborCache, EthernetInterfaceBuilder};
|
2018-02-22 20:23:35 +08:00
|
|
|
use smoltcp::socket::{SocketSet, IcmpSocket, IcmpSocketBuffer, IcmpPacketMetadata, IcmpEndpoint};
|
2017-06-18 18:14:46 +08:00
|
|
|
use std::collections::HashMap;
|
|
|
|
use byteorder::{ByteOrder, NetworkEndian};
|
|
|
|
|
|
|
|
fn main() {
|
2017-07-23 22:57:04 +08:00
|
|
|
utils::setup_logging("warn");
|
|
|
|
|
|
|
|
let (mut opts, mut free) = utils::create_options();
|
|
|
|
utils::add_tap_options(&mut opts, &mut free);
|
|
|
|
utils::add_middleware_options(&mut opts, &mut free);
|
|
|
|
opts.optopt("c", "count", "Amount of echo request packets to send (default: 4)", "COUNT");
|
|
|
|
opts.optopt("i", "interval",
|
|
|
|
"Interval between successive packets sent (seconds) (default: 1)", "INTERVAL");
|
|
|
|
opts.optopt("", "timeout",
|
|
|
|
"Maximum wait duration for an echo response packet (seconds) (default: 5)",
|
|
|
|
"TIMEOUT");
|
|
|
|
free.push("ADDRESS");
|
|
|
|
|
|
|
|
let mut matches = utils::parse_options(&opts, free);
|
|
|
|
let device = utils::parse_tap_options(&mut matches);
|
2017-08-30 03:35:09 +08:00
|
|
|
let fd = device.as_raw_fd();
|
2017-07-23 22:57:04 +08:00
|
|
|
let device = utils::parse_middleware_options(&mut matches, device, /*loopback=*/false);
|
2017-10-03 15:28:00 +08:00
|
|
|
let device_caps = device.capabilities();
|
2017-07-23 22:57:04 +08:00
|
|
|
let address = Ipv4Address::from_str(&matches.free[0]).expect("invalid address format");
|
|
|
|
let count = matches.opt_str("count").map(|s| usize::from_str(&s).unwrap()).unwrap_or(4);
|
2018-02-06 21:35:49 +08:00
|
|
|
let interval = matches.opt_str("interval")
|
|
|
|
.map(|s| Duration::from_secs(u64::from_str(&s).unwrap()))
|
|
|
|
.unwrap_or(Duration::from_secs(1));
|
|
|
|
let timeout = Duration::from_secs(
|
|
|
|
matches.opt_str("timeout").map(|s| u64::from_str(&s).unwrap()).unwrap_or(5)
|
|
|
|
);
|
2017-06-18 18:14:46 +08:00
|
|
|
|
2017-11-21 18:52:53 +08:00
|
|
|
let neighbor_cache = NeighborCache::new(BTreeMap::new());
|
2017-06-18 18:14:46 +08:00
|
|
|
|
2017-07-23 22:57:04 +08:00
|
|
|
let remote_addr = address;
|
2017-06-18 18:14:46 +08:00
|
|
|
let local_addr = Ipv4Address::new(192, 168, 69, 1);
|
|
|
|
|
2018-03-24 11:21:26 +08:00
|
|
|
let icmp_rx_buffer = IcmpSocketBuffer::new(vec![IcmpPacketMetadata::EMPTY], vec![0; 256]);
|
|
|
|
let icmp_tx_buffer = IcmpSocketBuffer::new(vec![IcmpPacketMetadata::EMPTY], vec![0; 256]);
|
2017-10-30 08:19:34 +08:00
|
|
|
let icmp_socket = IcmpSocket::new(icmp_rx_buffer, icmp_tx_buffer);
|
2017-06-18 18:14:46 +08:00
|
|
|
|
2017-10-03 23:12:31 +08:00
|
|
|
let ethernet_addr = EthernetAddress([0x02, 0x00, 0x00, 0x00, 0x00, 0x02]);
|
|
|
|
let ip_addr = IpCidr::new(IpAddress::from(local_addr), 24);
|
2017-10-03 15:28:00 +08:00
|
|
|
let default_v4_gw = Ipv4Address::new(192, 168, 69, 100);
|
2017-12-18 21:29:30 +08:00
|
|
|
let mut iface = EthernetInterfaceBuilder::new(device)
|
|
|
|
.ethernet_addr(ethernet_addr)
|
|
|
|
.ip_addrs([ip_addr])
|
|
|
|
.ipv4_gateway(default_v4_gw)
|
|
|
|
.neighbor_cache(neighbor_cache)
|
|
|
|
.finalize();
|
2017-06-18 18:14:46 +08:00
|
|
|
|
|
|
|
let mut sockets = SocketSet::new(vec![]);
|
2017-10-30 08:19:34 +08:00
|
|
|
let icmp_handle = sockets.add(icmp_socket);
|
2017-06-18 18:14:46 +08:00
|
|
|
|
2018-02-06 21:35:49 +08:00
|
|
|
let mut send_at = Instant::from_millis(0);
|
2017-06-18 18:14:46 +08:00
|
|
|
let mut seq_no = 0;
|
|
|
|
let mut received = 0;
|
|
|
|
let mut echo_payload = [0xffu8; 40];
|
|
|
|
let mut waiting_queue = HashMap::new();
|
2017-10-30 08:19:34 +08:00
|
|
|
let ident = 0x22b;
|
|
|
|
let endpoint = IpAddress::Ipv4(remote_addr);
|
2017-06-18 18:14:46 +08:00
|
|
|
|
|
|
|
loop {
|
2018-02-06 21:35:49 +08:00
|
|
|
iface.poll(&mut sockets, Instant::now()).unwrap();
|
2017-12-22 20:04:35 +08:00
|
|
|
|
2017-06-18 18:14:46 +08:00
|
|
|
{
|
2018-02-06 21:35:49 +08:00
|
|
|
let timestamp = Instant::now();
|
2017-10-30 08:19:34 +08:00
|
|
|
let mut socket = sockets.get::<IcmpSocket>(icmp_handle);
|
|
|
|
if !socket.is_open() {
|
2018-02-06 21:35:49 +08:00
|
|
|
socket.bind(IcmpEndpoint::Ident(ident)).unwrap();
|
|
|
|
send_at = timestamp;
|
2017-10-30 08:19:34 +08:00
|
|
|
}
|
2017-06-18 18:14:46 +08:00
|
|
|
|
2017-08-30 03:35:09 +08:00
|
|
|
if socket.can_send() && seq_no < count as u16 &&
|
2018-02-06 21:35:49 +08:00
|
|
|
send_at <= timestamp {
|
|
|
|
NetworkEndian::write_i64(&mut echo_payload, timestamp.total_millis());
|
2017-06-18 18:14:46 +08:00
|
|
|
let icmp_repr = Icmpv4Repr::EchoRequest {
|
2017-10-30 08:19:34 +08:00
|
|
|
ident: ident,
|
2017-06-18 18:14:46 +08:00
|
|
|
seq_no,
|
|
|
|
data: &echo_payload,
|
|
|
|
};
|
|
|
|
|
2017-10-30 08:19:34 +08:00
|
|
|
let icmp_payload = socket
|
|
|
|
.send(icmp_repr.buffer_len(), endpoint)
|
2017-06-18 18:14:46 +08:00
|
|
|
.unwrap();
|
|
|
|
|
2017-10-30 08:19:34 +08:00
|
|
|
let mut icmp_packet = Icmpv4Packet::new(icmp_payload);
|
2017-10-03 15:28:00 +08:00
|
|
|
icmp_repr.emit(&mut icmp_packet, &device_caps.checksum);
|
2017-06-18 18:14:46 +08:00
|
|
|
|
|
|
|
waiting_queue.insert(seq_no, timestamp);
|
|
|
|
seq_no += 1;
|
2018-02-06 21:35:49 +08:00
|
|
|
send_at += interval;
|
2017-06-18 18:14:46 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
if socket.can_recv() {
|
2017-10-30 08:19:34 +08:00
|
|
|
let (payload, _) = socket.recv().unwrap();
|
|
|
|
let icmp_packet = Icmpv4Packet::new(&payload);
|
|
|
|
let icmp_repr = Icmpv4Repr::parse(&icmp_packet, &device_caps.checksum).unwrap();
|
|
|
|
|
|
|
|
if let Icmpv4Repr::EchoReply { seq_no, data, .. } = icmp_repr {
|
|
|
|
if let Some(_) = waiting_queue.get(&seq_no) {
|
2018-02-22 20:25:07 +08:00
|
|
|
let packet_timestamp_ms = NetworkEndian::read_i64(data);
|
|
|
|
println!("{} bytes from {}: icmp_seq={}, time={}ms",
|
2017-10-30 08:19:34 +08:00
|
|
|
data.len(), remote_addr, seq_no,
|
2018-02-22 20:25:07 +08:00
|
|
|
timestamp.total_millis() - packet_timestamp_ms);
|
2017-10-30 08:19:34 +08:00
|
|
|
waiting_queue.remove(&seq_no);
|
|
|
|
received += 1;
|
2017-06-18 18:14:46 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
waiting_queue.retain(|seq, from| {
|
2018-02-06 21:35:49 +08:00
|
|
|
if timestamp - *from < timeout {
|
2017-06-18 18:14:46 +08:00
|
|
|
true
|
|
|
|
} else {
|
|
|
|
println!("From {} icmp_seq={} timeout", remote_addr, seq);
|
|
|
|
false
|
|
|
|
}
|
2017-08-30 03:35:09 +08:00
|
|
|
});
|
|
|
|
|
|
|
|
if seq_no == count as u16 && waiting_queue.is_empty() {
|
|
|
|
break
|
|
|
|
}
|
2017-06-18 18:14:46 +08:00
|
|
|
}
|
|
|
|
|
2018-02-06 21:35:49 +08:00
|
|
|
let timestamp = Instant::now();
|
|
|
|
match iface.poll_at(&sockets, timestamp) {
|
|
|
|
Some(poll_at) if timestamp < poll_at => {
|
|
|
|
let resume_at = cmp::min(poll_at, send_at);
|
|
|
|
phy_wait(fd, Some(resume_at - timestamp)).expect("wait error");
|
|
|
|
},
|
|
|
|
Some(_) => (),
|
|
|
|
None => {
|
|
|
|
phy_wait(fd, Some(send_at - timestamp)).expect("wait error");
|
|
|
|
}
|
|
|
|
}
|
2017-06-18 18:14:46 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
println!("--- {} ping statistics ---", remote_addr);
|
|
|
|
println!("{} packets transmitted, {} received, {:.0}% packet loss",
|
|
|
|
seq_no, received, 100.0 * (seq_no - received) as f64 / seq_no as f64);
|
|
|
|
}
|