commit da313b993aaaa9ecefaa5fb4966d9e86061c2922 Author: Lucas Schumacher Date: Fri Nov 15 14:06:44 2024 -0500 first commit diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ea8c4bf --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +/target diff --git a/Cargo.lock b/Cargo.lock new file mode 100644 index 0000000..c591fcf --- /dev/null +++ b/Cargo.lock @@ -0,0 +1,244 @@ +# This file is automatically @generated by Cargo. +# It is not intended for manual editing. +version = 3 + +[[package]] +name = "anstream" +version = "0.6.18" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8acc5369981196006228e28809f761875c0327210a891e941f4c683b3a99529b" +dependencies = [ + "anstyle", + "anstyle-parse", + "anstyle-query", + "anstyle-wincon", + "colorchoice", + "is_terminal_polyfill", + "utf8parse", +] + +[[package]] +name = "anstyle" +version = "1.0.10" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "55cc3b69f167a1ef2e161439aa98aed94e6028e5f9a59be9a6ffb47aef1651f9" + +[[package]] +name = "anstyle-parse" +version = "0.2.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "3b2d16507662817a6a20a9ea92df6652ee4f94f914589377d69f3b21bc5798a9" +dependencies = [ + "utf8parse", +] + +[[package]] +name = "anstyle-query" +version = "1.1.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "79947af37f4177cfead1110013d678905c37501914fba0efea834c3fe9a8d60c" +dependencies = [ + "windows-sys", +] + +[[package]] +name = "anstyle-wincon" +version = "3.0.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "2109dbce0e72be3ec00bed26e6a7479ca384ad226efdd66db8fa2e3a38c83125" +dependencies = [ + "anstyle", + "windows-sys", +] + +[[package]] +name = "anyhow" +version = "1.0.93" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "4c95c10ba0b00a02636238b814946408b1322d5ac4760326e6fb8ec956d85775" + +[[package]] +name = "clap" +version = "4.5.20" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "b97f376d85a664d5837dbae44bf546e6477a679ff6610010f17276f686d867e8" +dependencies = [ + "clap_builder", + "clap_derive", +] + +[[package]] +name = "clap_builder" +version = "4.5.20" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "19bc80abd44e4bed93ca373a0704ccbd1b710dc5749406201bb018272808dc54" +dependencies = [ + "anstream", + "anstyle", + "clap_lex", + "strsim", +] + +[[package]] +name = "clap_derive" +version = "4.5.18" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "4ac6a0c7b1a9e9a5186361f67dfa1b88213572f427fb9ab038efb2bd8c582dab" +dependencies = [ + "heck", + "proc-macro2", + "quote", + "syn", +] + +[[package]] +name = "clap_lex" +version = "0.7.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1462739cb27611015575c0c11df5df7601141071f07518d56fcc1be504cbec97" + +[[package]] +name = "colorchoice" +version = "1.0.3" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "5b63caa9aa9397e2d9480a9b13673856c78d8ac123288526c37d7839f2a86990" + +[[package]] +name = "heck" +version = "0.5.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea" + +[[package]] +name = "is_terminal_polyfill" +version = "1.70.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7943c866cc5cd64cbc25b2e01621d07fa8eb2a1a23160ee81ce38704e97b8ecf" + +[[package]] +name = "pnet" +version = "0.1.0" +dependencies = [ + "anyhow", + "clap", +] + +[[package]] +name = "proc-macro2" +version = "1.0.89" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "f139b0662de085916d1fb67d2b4169d1addddda1919e696f3252b740b629986e" +dependencies = [ + "unicode-ident", +] + +[[package]] +name = "quote" +version = "1.0.37" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "b5b9d34b8991d19d98081b46eacdd8eb58c6f2b201139f7c5f643cc155a633af" +dependencies = [ + "proc-macro2", +] + +[[package]] +name = "strsim" +version = "0.11.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f" + +[[package]] +name = "syn" +version = "2.0.87" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "25aa4ce346d03a6dcd68dd8b4010bcb74e54e62c90c573f394c46eae99aba32d" +dependencies = [ + "proc-macro2", + "quote", + "unicode-ident", +] + +[[package]] +name = "unicode-ident" +version = "1.0.13" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "e91b56cd4cadaeb79bbf1a5645f6b4f8dc5bde8834ad5894a8db35fda9efa1fe" + +[[package]] +name = "utf8parse" +version = "0.2.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "06abde3611657adf66d383f00b093d7faecc7fa57071cce2578660c9f1010821" + +[[package]] +name = "windows-sys" +version = "0.59.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1e38bc4d79ed67fd075bcc251a1c39b32a1776bbe92e5bef1f0bf1f8c531853b" +dependencies = [ + "windows-targets", +] + +[[package]] +name = "windows-targets" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "9b724f72796e036ab90c1021d4780d4d3d648aca59e491e6b98e725b84e99973" +dependencies = [ + "windows_aarch64_gnullvm", + "windows_aarch64_msvc", + "windows_i686_gnu", + "windows_i686_gnullvm", + "windows_i686_msvc", + "windows_x86_64_gnu", + "windows_x86_64_gnullvm", + "windows_x86_64_msvc", +] + +[[package]] +name = "windows_aarch64_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "32a4622180e7a0ec044bb555404c800bc9fd9ec262ec147edd5989ccd0c02cd3" + +[[package]] +name = "windows_aarch64_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "09ec2a7bb152e2252b53fa7803150007879548bc709c039df7627cabbd05d469" + +[[package]] +name = "windows_i686_gnu" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8e9b5ad5ab802e97eb8e295ac6720e509ee4c243f69d781394014ebfe8bbfa0b" + +[[package]] +name = "windows_i686_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "0eee52d38c090b3caa76c563b86c3a4bd71ef1a819287c19d586d7334ae8ed66" + +[[package]] +name = "windows_i686_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "240948bc05c5e7c6dabba28bf89d89ffce3e303022809e73deaefe4f6ec56c66" + +[[package]] +name = "windows_x86_64_gnu" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "147a5c80aabfbf0c7d901cb5895d1de30ef2907eb21fbbab29ca94c5b08b1a78" + +[[package]] +name = "windows_x86_64_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "24d5b23dc417412679681396f2b49f3de8c1473deb516bd34410872eff51ed0d" + +[[package]] +name = "windows_x86_64_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec" diff --git a/Cargo.toml b/Cargo.toml new file mode 100644 index 0000000..28c69b4 --- /dev/null +++ b/Cargo.toml @@ -0,0 +1,8 @@ +[package] +name = "pnet" +version = "0.1.0" +edition = "2021" + +[dependencies] +anyhow = "1.0.93" +clap = { version = "4.5.20", features = ["derive"] } diff --git a/src/main.rs b/src/main.rs new file mode 100644 index 0000000..1e8632f --- /dev/null +++ b/src/main.rs @@ -0,0 +1,147 @@ +use anyhow::Result; +use clap::Parser; +use std::{ + fmt::Display, + fs, + net::{Ipv4Addr, SocketAddrV4}, + path::Path, + u32, +}; + +/// Simple program to print network socket information on linux systems using /proc/net/ +#[derive(Parser, Debug)] +#[command(version, about, long_about = None)] +struct Args { + /// Print TCP socket info + #[arg(short, long, default_value_t = true)] + tcp: bool, + + /// Print UDP socket info + #[arg(short, long, default_value_t = false)] + udp: bool, +} + +fn read_or_fail + Display>(path: P) -> String { + match fs::read_to_string(&path) { + Ok(s) => return s, + Err(e) => { + panic!("Error accessing {path}: {e}") + } + } +} + +fn parse_sock_addr(raw: &str) -> Result { + let split: Vec<_> = raw.split(":").collect(); + if split.len() != 2 { + anyhow::bail!(invalid_data_error()); + } + let ipv4 = Ipv4Addr::from_bits(u32::from_str_radix(split[0], 16)?.to_be()); + let port = u16::from_str_radix(split[1], 16)?.to_be(); + return Ok(SocketAddrV4::new(ipv4, port)); +} + +#[allow(non_camel_case_types)] +#[repr(u8)] +#[derive(Debug)] +enum TcpStatus { + TCP_ESTABLISHED = 1, + TCP_SYN_SENT, + TCP_SYN_RECV, + TCP_FIN_WAIT1, + TCP_FIN_WAIT2, + TCP_TIME_WAIT, + TCP_CLOSE, + TCP_CLOSE_WAIT, + TCP_LAST_ACK, + TCP_LISTEN, + TCP_CLOSING, +} +impl TryFrom for TcpStatus { + type Error = anyhow::Error; + + fn try_from(value: u8) -> Result { + match value { + x if x == TcpStatus::TCP_ESTABLISHED as u8 => Ok(TcpStatus::TCP_ESTABLISHED), + x if x == TcpStatus::TCP_SYN_SENT as u8 => Ok(TcpStatus::TCP_SYN_SENT), + x if x == TcpStatus::TCP_SYN_RECV as u8 => Ok(TcpStatus::TCP_SYN_RECV), + x if x == TcpStatus::TCP_FIN_WAIT1 as u8 => Ok(TcpStatus::TCP_FIN_WAIT1), + x if x == TcpStatus::TCP_FIN_WAIT2 as u8 => Ok(TcpStatus::TCP_FIN_WAIT2), + x if x == TcpStatus::TCP_TIME_WAIT as u8 => Ok(TcpStatus::TCP_TIME_WAIT), + x if x == TcpStatus::TCP_CLOSE as u8 => Ok(TcpStatus::TCP_CLOSE), + x if x == TcpStatus::TCP_CLOSE_WAIT as u8 => Ok(TcpStatus::TCP_CLOSE_WAIT), + x if x == TcpStatus::TCP_LAST_ACK as u8 => Ok(TcpStatus::TCP_LAST_ACK), + x if x == TcpStatus::TCP_LISTEN as u8 => Ok(TcpStatus::TCP_LISTEN), + x if x == TcpStatus::TCP_CLOSING as u8 => Ok(TcpStatus::TCP_CLOSING), + _ => Err(anyhow::anyhow!(invalid_data_error())), // Return an error on invalid value + } + } +} +impl TryFrom<&str> for TcpStatus { + type Error = anyhow::Error; + + fn try_from(value: &str) -> Result { + let int = u8::from_str_radix(value, 16)?; + return TcpStatus::try_from(int); + } +} + +#[derive(Debug)] +#[allow(dead_code)] +struct TcpInfo<'a> { + sl: &'a str, + local_address: SocketAddrV4, + rem_address: SocketAddrV4, + st: TcpStatus, + queue: &'a str, + tr_tm_when: &'a str, + retrnsmt: &'a str, + uid: &'a str, + timeout: &'a str, + inode: Vec<&'a str>, +} +fn invalid_data_error() -> std::io::Error { + std::io::Error::new(std::io::ErrorKind::InvalidData, "Unknown / Invalid format!") +} +impl<'a> TcpInfo<'a> { + pub fn new(info: Vec<&'a str>) -> Result> { + if info.len() < 12 { + anyhow::bail!(invalid_data_error()); + } + let sl = info[0]; + let local_address = parse_sock_addr(info[1])?; + let rem_address = parse_sock_addr(info[2])?; + let st = TcpStatus::try_from(info[3])?; + let tcp_info: TcpInfo = TcpInfo { + sl, + local_address, + rem_address, + st, + queue: info[4], + tr_tm_when: info[5], + retrnsmt: info[6], + uid: info[7], + timeout: info[8], + inode: info[9..].to_vec(), + }; + return Ok(tcp_info); + } +} + +fn main() { + let args = Args::parse(); + + if args.tcp { + let tcp_info = read_or_fail("/proc/net/tcp"); + for line in tcp_info.split("\n").skip(1) { + let info: Vec<&str> = line.split_whitespace().collect(); + if info.len() == 0 { + continue; + } + let tcp_info = TcpInfo::new(info).unwrap(); + println!( + "{:>5} {:21}\t{:21}\t{:?}", + tcp_info.sl, tcp_info.local_address, tcp_info.rem_address, tcp_info.st + ); + } + } +}