parent
5453b661be
commit
fdcb241ef3
@ -0,0 +1,172 @@ |
|||||||
|
/* |
||||||
|
* Copyright (c) 2022-2023 Niko Bonnieure, Par le Peuple, NextGraph.org developers |
||||||
|
* All rights reserved. |
||||||
|
* Licensed under the Apache License, Version 2.0 |
||||||
|
* <LICENSE-APACHE2 or http://www.apache.org/licenses/LICENSE-2.0>
|
||||||
|
* or the MIT license <LICENSE-MIT or http://opensource.org/licenses/MIT>,
|
||||||
|
* at your option. All files in the project carrying such |
||||||
|
* notice may not be copied, modified, or distributed except |
||||||
|
* according to those terms. |
||||||
|
*/ |
||||||
|
use p2p_net::utils::{is_ipv4_global, is_ipv4_private, is_ipv6_global, is_ipv6_private}; |
||||||
|
use std::net::{IpAddr, Ipv4Addr, Ipv6Addr}; |
||||||
|
|
||||||
|
#[derive(Clone, Debug, PartialEq, Eq)] |
||||||
|
pub enum InterfaceType { |
||||||
|
Loopback, |
||||||
|
Private, |
||||||
|
Public, |
||||||
|
Invalid, |
||||||
|
} |
||||||
|
|
||||||
|
impl InterfaceType { |
||||||
|
pub fn is_ipv4_valid_for_type(&self, ip: &Ipv4Addr) -> bool { |
||||||
|
match self { |
||||||
|
InterfaceType::Loopback => ip.is_loopback(), |
||||||
|
InterfaceType::Public => is_public_ipv4(ip), |
||||||
|
// we allow to bind to link-local for IPv4
|
||||||
|
InterfaceType::Private => is_ipv4_private(ip), |
||||||
|
_ => false, |
||||||
|
} |
||||||
|
} |
||||||
|
pub fn is_ipv6_valid_for_type(&self, ip: &Ipv6Addr) -> bool { |
||||||
|
match self { |
||||||
|
InterfaceType::Loopback => ip.is_loopback(), |
||||||
|
InterfaceType::Public => is_public_ipv6(ip), |
||||||
|
// we do NOT allow to bind to link-local for IPv6
|
||||||
|
InterfaceType::Private => is_ipv6_private(ip), |
||||||
|
_ => false, |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
pub fn print_ipv4(ip: &default_net::ip::Ipv4Net) -> String { |
||||||
|
format!("{}/{}", ip.addr, ip.prefix_len) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn print_ipv6(ip: &default_net::ip::Ipv6Net) -> String { |
||||||
|
format!("{}/{}", ip.addr, ip.prefix_len) |
||||||
|
} |
||||||
|
|
||||||
|
#[derive(Clone, Debug)] |
||||||
|
pub struct Interface { |
||||||
|
pub if_type: InterfaceType, |
||||||
|
pub name: String, |
||||||
|
pub mac_addr: Option<default_net::interface::MacAddr>, |
||||||
|
/// List of Ipv4Net for the network interface
|
||||||
|
pub ipv4: Vec<default_net::ip::Ipv4Net>, |
||||||
|
/// List of Ipv6Net for the network interface
|
||||||
|
pub ipv6: Vec<default_net::ip::Ipv6Net>, |
||||||
|
} |
||||||
|
|
||||||
|
pub fn find_first(list: &Vec<Interface>, iftype: InterfaceType) -> Option<Interface> { |
||||||
|
for inf in list { |
||||||
|
if inf.if_type == iftype { |
||||||
|
return Some(inf.clone()); |
||||||
|
} |
||||||
|
} |
||||||
|
None |
||||||
|
} |
||||||
|
|
||||||
|
pub fn find_first_or_name( |
||||||
|
list: &Vec<Interface>, |
||||||
|
iftype: InterfaceType, |
||||||
|
name: &String, |
||||||
|
) -> Option<Interface> { |
||||||
|
for inf in list { |
||||||
|
if (name == "default" || *name == inf.name) && inf.if_type == iftype { |
||||||
|
return Some(inf.clone()); |
||||||
|
} |
||||||
|
} |
||||||
|
None |
||||||
|
} |
||||||
|
|
||||||
|
pub fn find_name(list: &Vec<Interface>, name: &String) -> Option<Interface> { |
||||||
|
for inf in list { |
||||||
|
if *name == inf.name { |
||||||
|
return Some(inf.clone()); |
||||||
|
} |
||||||
|
} |
||||||
|
None |
||||||
|
} |
||||||
|
|
||||||
|
pub fn is_public_ipv4(ip: &Ipv4Addr) -> bool { |
||||||
|
// TODO, use core::net::Ipv6Addr.is_global when it will be stable
|
||||||
|
return is_ipv4_global(ip); |
||||||
|
} |
||||||
|
|
||||||
|
pub fn is_public_ipv6(ip: &Ipv6Addr) -> bool { |
||||||
|
// TODO, use core::net::Ipv6Addr.is_global when it will be stable
|
||||||
|
return is_ipv6_global(ip); |
||||||
|
} |
||||||
|
|
||||||
|
pub fn is_public_ip(ip: &IpAddr) -> bool { |
||||||
|
match ip { |
||||||
|
IpAddr::V4(v4) => is_public_ipv4(v4), |
||||||
|
IpAddr::V6(v6) => is_public_ipv6(v6), |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
pub fn is_private_ip(ip: &IpAddr) -> bool { |
||||||
|
match ip { |
||||||
|
IpAddr::V4(v4) => is_ipv4_private(v4), |
||||||
|
IpAddr::V6(v6) => is_ipv6_private(v6), |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
pub fn get_interface() -> Vec<Interface> { |
||||||
|
let mut res: Vec<Interface> = vec![]; |
||||||
|
let interfaces = default_net::get_interfaces(); |
||||||
|
for interface in interfaces { |
||||||
|
if interface.ipv4.len() > 0 { |
||||||
|
let first_v4 = interface.ipv4[0].addr; |
||||||
|
let if_type = if first_v4.is_loopback() { |
||||||
|
InterfaceType::Loopback |
||||||
|
} else if is_ipv4_private(&first_v4) { |
||||||
|
InterfaceType::Private |
||||||
|
} else if is_public_ipv4(&first_v4) { |
||||||
|
InterfaceType::Public |
||||||
|
} else { |
||||||
|
continue; |
||||||
|
}; |
||||||
|
let interf = Interface { |
||||||
|
if_type, |
||||||
|
name: interface.name, |
||||||
|
mac_addr: interface.mac_addr, |
||||||
|
ipv4: interface.ipv4, |
||||||
|
ipv6: interface.ipv6, |
||||||
|
}; |
||||||
|
res.push(interf); |
||||||
|
} |
||||||
|
} |
||||||
|
res |
||||||
|
} |
||||||
|
|
||||||
|
pub fn print_interfaces() { |
||||||
|
let interfaces = get_interface(); |
||||||
|
for interface in interfaces { |
||||||
|
println!("{} \t{:?}", interface.name, interface.if_type); |
||||||
|
|
||||||
|
println!( |
||||||
|
"\tIPv4: {}", |
||||||
|
interface |
||||||
|
.ipv4 |
||||||
|
.iter() |
||||||
|
.map(|ip| print_ipv4(ip)) |
||||||
|
.collect::<Vec<String>>() |
||||||
|
.join(" ") |
||||||
|
); |
||||||
|
println!( |
||||||
|
"\tIPv6: {}", |
||||||
|
interface |
||||||
|
.ipv6 |
||||||
|
.iter() |
||||||
|
.map(|ip| print_ipv6(ip)) |
||||||
|
.collect::<Vec<String>>() |
||||||
|
.join(" ") |
||||||
|
); |
||||||
|
if let Some(mac_addr) = interface.mac_addr { |
||||||
|
println!("\tMAC: {}", mac_addr); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue