Compare commits

..

4 Commits
v0.1.1 ... main

Author SHA1 Message Date
ad0a24201c
[release] linkspeed v0.1.3 2025-03-29 20:35:03 +01:00
f3f1cb46c4
[deps] cargo update
Updating once_cell v1.21.1 -> v1.21.3
2025-03-29 20:35:02 +01:00
c9180033d1
[release] linkspeed v0.1.2 2025-03-14 01:09:28 +01:00
a03a679f71
[code] better error handling and show available netdevs on some errors 2025-03-14 01:09:03 +01:00
4 changed files with 83 additions and 13 deletions

6
Cargo.lock generated
View File

@ -20,13 +20,13 @@ checksum = "ce23b50ad8242c51a442f3ff322d56b02f08852c77e4c0b4d3fd684abc89c683"
[[package]]
name = "linkspeed"
version = "0.1.1"
version = "0.1.3"
dependencies = [
"eyre",
]
[[package]]
name = "once_cell"
version = "1.21.1"
version = "1.21.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d75b0bedcc4fe52caa0e03d9f1151a323e4aa5e2d78ba3580400cd3c9e2bc4bc"
checksum = "42f5e15c9953c5e4ccceeb2e7382a716482c34515315f7b03532b8b4e8393d2d"

View File

@ -1,12 +1,13 @@
[package]
name = "linkspeed"
version = "0.1.1"
version = "0.1.3"
edition = "2024"
[lints.rust]
unsafe_code = "forbid"
[lints.clippy]
redundant_closure_for_method_calls = "allow"
similar_names = "allow"
pedantic = { level = "warn", priority = -1 }

View File

@ -6,6 +6,8 @@ mod sysfs;
use std::{env, thread::sleep};
use sysfs::NetdevError;
use crate::{cfg::Config, measure::LinkSpeed};
fn main() -> eyre::Result<()> {
@ -19,7 +21,12 @@ fn main() -> eyre::Result<()> {
(Some(_), Some(na)) => na,
(Some(nc), None) => nc,
(None, Some(na)) => na,
(None, None) => eyre::bail!("No network device specified"),
(None, None) => {
eyre::bail!(
"No network device specified. Please provide a network device name as an argument or in the config file.\n\n{}",
NetdevError::available_netdevs_msg()
);
}
};
let link_speed = LinkSpeed::new(&netdev_name)?;

View File

@ -3,24 +3,73 @@ use std::{
io::{Read as _, Seek as _, SeekFrom},
};
use eyre::Context as _;
pub(crate) struct Netdev {
pub struct Netdev {
rx_bytes: File,
tx_bytes: File,
}
#[derive(Debug)]
pub enum NetdevError {
NotFound { name: String },
IoError { name: String, err: std::io::Error },
}
impl NetdevError {
pub fn available_netdevs_msg() -> String {
let list = Netdev::list()
.unwrap_or_default()
.into_iter()
.map(|dev| format!(" {dev}"))
.collect::<Vec<_>>()
.join("\n");
format!("Available network devices:\n{list}")
}
}
impl std::fmt::Display for NetdevError {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
NetdevError::NotFound { name } => write!(
f,
"Netdev '{name}' not found.\n\n{}",
Self::available_netdevs_msg()
),
NetdevError::IoError { name, err } => {
write!(f, "I/O Error occurred for netdev '{name}': {err}")
}
}
}
}
impl Netdev {
const SYS_CLASS_NET: &'static str = "/sys/class/net";
pub fn from_name(name: &str) -> eyre::Result<Self> {
let rx_bytes = Self::open_netdev(name, "rx_bytes")?;
let tx_bytes = Self::open_netdev(name, "tx_bytes")?;
let rx_bytes = Self::open_netdev_stat(name, "rx_bytes")?;
let tx_bytes = Self::open_netdev_stat(name, "tx_bytes")?;
Ok(Self { rx_bytes, tx_bytes })
}
fn open_netdev(dev: &str, stat: &str) -> eyre::Result<File> {
let path = format!("/sys/class/net/{dev}/statistics/{stat}");
File::open(&path).wrap_err(format!("Failed to open netdev '{dev}'"))
fn open_netdev_stat(name: &str, stat: &str) -> eyre::Result<File> {
let sys_class_net = Self::SYS_CLASS_NET;
let path = format!("{sys_class_net}/{name}/statistics/{stat}");
match File::open(&path) {
Ok(file) => Ok(file),
Err(e) => {
if e.kind() == std::io::ErrorKind::NotFound {
Err(eyre::eyre!(NetdevError::NotFound {
name: name.to_string(),
}))
} else {
Err(eyre::eyre!(NetdevError::IoError {
name: name.to_string(),
err: e,
}))
}
}
}
}
fn read_bytes(file: &mut File) -> eyre::Result<u64> {
@ -37,4 +86,17 @@ impl Netdev {
pub fn tx_bytes(&mut self) -> eyre::Result<u64> {
Self::read_bytes(&mut self.tx_bytes)
}
pub fn list() -> eyre::Result<Vec<String>> {
let mut entries = std::fs::read_dir(Self::SYS_CLASS_NET)?
.filter_map(|entry| {
entry
.ok()
.and_then(|entry| entry.file_name().into_string().ok())
})
.collect::<Vec<_>>();
entries.sort();
Ok(entries)
}
}