use std::env; use std::io::{self, ErrorKind}; use std::mem::MaybeUninit; use std::net::{SocketAddr, ToSocketAddrs}; use std::os::fd::{AsRawFd, FromRawFd, OwnedFd}; use std::time::{Duration, Instant}; const ICMP_ECHO_REQUEST: u8 = 8; const ICMP_ECHO_REPLY: u8 = 0; fn checksum(data: &[u8]) -> u16 { let mut sum: u32 = 0; let mut i = 0; while i < data.len() - 1 { sum += u16::from_be_bytes([data[i], data[i + 1]]) as u32; i += 2; } if data.len() % 2 == 1 { sum += (data[data.len() - 1] as u32) << 8; } while sum >> 16 != 0 { sum = (sum & 0xFFFF) + (sum >> 16); } !(sum as u16) } fn build_icmp_packet(seq: u16, id: u16) -> [u8; 64] { let mut packet = [0u8; 64]; packet[0] = ICMP_ECHO_REQUEST; packet[1] = 0; // code // checksum at 2-3, filled later packet[4..6].copy_from_slice(&id.to_be_bytes()); packet[6..8].copy_from_slice(&seq.to_be_bytes()); // payload: fill with pattern for i in 8..64 { packet[i] = i as u8; } let cs = checksum(&packet); packet[2..4].copy_from_slice(&cs.to_be_bytes()); packet } fn create_icmp_socket() -> io::Result { let fd = unsafe { libc::socket(libc::AF_INET, libc::SOCK_DGRAM, libc::IPPROTO_ICMP) }; if fd < 0 { return Err(io::Error::last_os_error()); } Ok(unsafe { OwnedFd::from_raw_fd(fd) }) } fn send_ping(fd: &OwnedFd, addr: &SocketAddr, packet: &[u8]) -> io::Result<()> { let sockaddr = match addr { SocketAddr::V4(a) => libc::sockaddr_in { sin_family: libc::AF_INET as u16, sin_port: 0, sin_addr: libc::in_addr { s_addr: u32::from_ne_bytes(a.ip().octets()), }, sin_zero: [0; 8], sin_len: std::mem::size_of::() as u8, }, _ => return Err(io::Error::new(ErrorKind::Other, "IPv6 not supported")), }; let ret = unsafe { libc::sendto( fd.as_raw_fd(), packet.as_ptr() as *const libc::c_void, packet.len(), 0, &sockaddr as *const _ as *const libc::sockaddr, std::mem::size_of::() as u32, ) }; if ret < 0 { Err(io::Error::last_os_error()) } else { Ok(()) } } fn recv_ping(fd: &OwnedFd, timeout: Duration) -> io::Result<(u16, u16)> { let tv = libc::timeval { tv_sec: timeout.as_secs() as i64, tv_usec: timeout.subsec_micros() as i32, }; unsafe { libc::setsockopt( fd.as_raw_fd(), libc::SOL_SOCKET, libc::SO_RCVTIMEO, &tv as *const _ as *const libc::c_void, std::mem::size_of::() as u32, ); } let mut buf = [MaybeUninit::::uninit(); 1024]; let n = unsafe { libc::recvfrom( fd.as_raw_fd(), buf.as_mut_ptr() as *mut libc::c_void, buf.len(), 0, std::ptr::null_mut(), std::ptr::null_mut(), ) }; if n < 0 { return Err(io::Error::last_os_error()); } let buf: &[u8] = unsafe { std::mem::transmute(&buf[..n as usize]) }; if buf.len() < 8 { return Err(io::Error::new(ErrorKind::Other, "packet too short")); } if buf[0] != ICMP_ECHO_REPLY { return Err(io::Error::new(ErrorKind::Other, "not an echo reply")); } let id = u16::from_be_bytes([buf[4], buf[5]]); let seq = u16::from_be_bytes([buf[6], buf[7]]); Ok((id, seq)) } fn main() -> io::Result<()> { let args: Vec = env::args().collect(); if args.len() < 2 { eprintln!("Usage: piiing "); std::process::exit(1); } let host = &args[1]; let addr: SocketAddr = format!("{}:0", host) .to_socket_addrs()? .find(|a| a.is_ipv4()) .ok_or_else(|| io::Error::new(ErrorKind::Other, "could not resolve host"))?; println!("PIIING {} ({})", host, addr.ip()); let socket = create_icmp_socket()?; let id = std::process::id() as u16; for seq in 0..4 { let packet = build_icmp_packet(seq, id); let start = Instant::now(); send_ping(&socket, &addr, &packet)?; match recv_ping(&socket, Duration::from_secs(2)) { Ok((_rid, rseq)) => { let elapsed = start.elapsed(); println!( "64 bytes from {}: icmp_seq={} time={:.2}ms", addr.ip(), rseq, elapsed.as_secs_f64() * 1000.0 ); } Err(e) => { eprintln!("Request timeout: {}", e); } } std::thread::sleep(Duration::from_secs(1)); } Ok(()) }

Build unprivileged ping tool in Rust