116 lines
3.2 KiB
Rust
116 lines
3.2 KiB
Rust
use std::net::{IpAddr, Ipv4Addr, Ipv6Addr};
|
|
use std::str::FromStr;
|
|
|
|
pub fn extract_ipv4(ip: IpAddr) -> Ipv4Addr {
|
|
match ip {
|
|
IpAddr::V4(i) => i,
|
|
IpAddr::V6(_) => {
|
|
panic!("IPv6 found in IPv4 definition!")
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn extract_ipv6(ip: IpAddr) -> Ipv6Addr {
|
|
match ip {
|
|
IpAddr::V4(_) => {
|
|
panic!("IPv4 found in IPv6 definition!")
|
|
}
|
|
IpAddr::V6(i) => i,
|
|
}
|
|
}
|
|
|
|
pub fn is_ipv4_address_valid<D: AsRef<str>>(ip: D) -> bool {
|
|
Ipv4Addr::from_str(ip.as_ref()).is_ok()
|
|
}
|
|
|
|
pub fn is_ipv6_address_valid<D: AsRef<str>>(ip: D) -> bool {
|
|
Ipv6Addr::from_str(ip.as_ref()).is_ok()
|
|
}
|
|
|
|
pub fn is_ipv4_mask_valid(mask: u8) -> bool {
|
|
mask <= 32
|
|
}
|
|
|
|
pub fn is_ipv6_mask_valid(mask: u8) -> bool {
|
|
mask <= 128
|
|
}
|
|
|
|
pub fn is_mask_valid(ipv: usize, mask: u8) -> bool {
|
|
match ipv {
|
|
4 => is_ipv4_mask_valid(mask),
|
|
6 => is_ipv6_mask_valid(mask),
|
|
_ => panic!("Unsupported IP version"),
|
|
}
|
|
}
|
|
|
|
pub fn is_mac_address_valid<D: AsRef<str>>(mac: D) -> bool {
|
|
lazy_regex::regex!("^([a-fA-F0-9]{2}[:-]){5}[a-fA-F0-9]{2}$").is_match(mac.as_ref())
|
|
}
|
|
|
|
pub fn is_net_interface_name_valid<D: AsRef<str>>(int: D) -> bool {
|
|
lazy_regex::regex!("^[a-zA-Z0-9]+$").is_match(int.as_ref())
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use crate::utils::net_utils::{
|
|
is_ipv4_address_valid, is_ipv6_address_valid, is_mac_address_valid, is_mask_valid,
|
|
is_net_interface_name_valid,
|
|
};
|
|
|
|
#[test]
|
|
fn test_is_mac_address_valid() {
|
|
assert!(is_mac_address_valid("FF:FF:FF:FF:FF:FF"));
|
|
assert!(is_mac_address_valid("02:42:a4:6e:f2:be"));
|
|
|
|
assert!(!is_mac_address_valid("tata"));
|
|
assert!(!is_mac_address_valid("FF:FF:FF:FF:FF:FZ"));
|
|
assert!(!is_mac_address_valid("FF:FF:FF:FF:FF:FF:FF"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_is_ipv4_address_valid() {
|
|
assert!(is_ipv4_address_valid("10.0.0.1"));
|
|
assert!(is_ipv4_address_valid("2.56.58.156"));
|
|
|
|
assert!(!is_ipv4_address_valid("tata"));
|
|
assert!(!is_ipv4_address_valid("1.25.25.288"));
|
|
assert!(!is_ipv4_address_valid("5.5.5.5.5"));
|
|
assert!(!is_ipv4_address_valid("fe80::"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_is_ipv6_address_valid() {
|
|
assert!(is_ipv6_address_valid("fe80::"));
|
|
assert!(is_ipv6_address_valid("fe80:dd::"));
|
|
assert!(is_ipv6_address_valid("00:00:00:00:00::"));
|
|
assert!(is_ipv6_address_valid("0:0:0:0:0:0:0:0"));
|
|
|
|
assert!(!is_ipv6_address_valid("tata"));
|
|
assert!(!is_ipv6_address_valid("2.56.58.156"));
|
|
assert!(!is_ipv6_address_valid("fe::dd::dd"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_is_mask_valid() {
|
|
assert!(is_mask_valid(4, 25));
|
|
assert!(is_mask_valid(4, 32));
|
|
assert!(is_mask_valid(6, 32));
|
|
assert!(is_mask_valid(6, 34));
|
|
|
|
assert!(!is_mask_valid(4, 34));
|
|
assert!(is_mask_valid(6, 69));
|
|
assert!(is_mask_valid(6, 128));
|
|
assert!(!is_mask_valid(6, 129));
|
|
}
|
|
|
|
#[test]
|
|
fn test_is_net_interface_name_valid() {
|
|
assert!(is_net_interface_name_valid("eth0"));
|
|
assert!(is_net_interface_name_valid("enp0s25"));
|
|
|
|
assert!(!is_net_interface_name_valid("enp0s25 "));
|
|
assert!(!is_net_interface_name_valid("@enp0s25 "));
|
|
}
|
|
}
|