shellrow
3 years ago
parent
94d26cb15d
commit
b64c013a9b
@ -0,0 +1,26 @@ |
||||
use default_net; |
||||
|
||||
fn main(){ |
||||
let interfaces = default_net::get_interfaces(); |
||||
for interface in interfaces { |
||||
println!("Interface"); |
||||
println!("\tIndex: {}", interface.index); |
||||
println!("\tName: {}", interface.name); |
||||
println!("\tDescription: {:?}", interface.description); |
||||
if let Some(mac_addr) = interface.mac_addr { |
||||
println!("\tMAC: {}", mac_addr); |
||||
}else{ |
||||
println!("\tMAC: (Failed to get mac address)"); |
||||
} |
||||
println!("\tIPv4: {:?}", interface.ipv4); |
||||
println!("\tIPv6: {:?}", interface.ipv6); |
||||
if let Some(gateway) = interface.gateway { |
||||
println!("Gateway"); |
||||
println!("\tMAC: {}", gateway.mac_addr); |
||||
println!("\tIP: {}", gateway.ip_addr); |
||||
}else { |
||||
println!("Gateway: (Not found)"); |
||||
} |
||||
println!(); |
||||
} |
||||
} |
||||
@ -1,6 +1,241 @@ |
||||
use crate::interface::{Interface, MacAddr}; |
||||
use std::net::{IpAddr, Ipv4Addr, Ipv6Addr}; |
||||
use std::net::UdpSocket; |
||||
use std::time::{Duration, Instant}; |
||||
use pnet_packet::Packet; |
||||
use crate::interface::{MacAddr, Interface}; |
||||
use crate::gateway::Gateway; |
||||
|
||||
pub fn get_interfaces() -> Vec<Interface> { |
||||
let interfaces: Vec<Interface> = vec![]; |
||||
return interfaces; |
||||
const TIMEOUT: u64 = 3000; |
||||
|
||||
fn get_default_gateway(interface_index: u32) -> Result<Gateway, String> { |
||||
let interfaces = pnet_datalink::interfaces(); |
||||
let interface = interfaces.into_iter().filter(|interface: &pnet_datalink::NetworkInterface| interface.index == interface_index).next().expect("Failed to get Interface"); |
||||
let config = pnet_datalink::Config { |
||||
write_buffer_size: 4096, |
||||
read_buffer_size: 4096, |
||||
read_timeout: None, |
||||
write_timeout: None, |
||||
channel_type: pnet_datalink::ChannelType::Layer2, |
||||
bpf_fd_attempts: 1000, |
||||
linux_fanout: None, |
||||
promiscuous: false, |
||||
}; |
||||
let (mut _tx, mut rx) = match pnet_datalink::channel(&interface, config) { |
||||
Ok(pnet_datalink::Channel::Ethernet(tx, rx)) => (tx, rx), |
||||
Ok(_) => panic!("Unknown channel type"), |
||||
Err(e) => panic!("Error happened {}", e), |
||||
}; |
||||
match send_udp_packet() { |
||||
Ok(_) => (), |
||||
Err(e) => return Err(format!("Failed to send UDP packet {}", e)), |
||||
} |
||||
receive_packets(&mut rx) |
||||
} |
||||
|
||||
fn send_udp_packet() -> Result<(), String> { |
||||
let buf = [0u8; 0]; |
||||
let socket = match UdpSocket::bind("0.0.0.0:0") { |
||||
Ok(s) => s, |
||||
Err(e) => return Err(format!("Failed to create UDP socket {}", e)), |
||||
}; |
||||
let dst: &str = "1.1.1.1:80"; |
||||
match socket.set_ttl(1) { |
||||
Ok(_) => (), |
||||
Err(e) => return Err(format!("Failed to set TTL {}", e)), |
||||
} |
||||
match socket.send_to(&buf, dst) { |
||||
Ok(_) => (), |
||||
Err(e) => return Err(format!("Failed to send data {}", e)), |
||||
} |
||||
Ok(()) |
||||
} |
||||
|
||||
fn receive_packets(rx: &mut Box<dyn pnet_datalink::DataLinkReceiver>) -> Result<Gateway, String>{ |
||||
let timeout = Duration::from_millis(TIMEOUT); |
||||
let start_time = Instant::now(); |
||||
loop { |
||||
match rx.next() { |
||||
Ok(frame) => { |
||||
let frame = match pnet_packet::ethernet::EthernetPacket::new(frame) { |
||||
Some(f) => f, |
||||
None => return Err(String::from("Failed to read packet")), |
||||
}; |
||||
match frame.get_ethertype() { |
||||
pnet_packet::ethernet::EtherTypes::Ipv4 => { |
||||
if let Some(ip_addr) = ipv4_handler(&frame) { |
||||
let gateway = Gateway { |
||||
mac_addr: MacAddr::new(frame.get_source().octets()), |
||||
ip_addr: ip_addr, |
||||
}; |
||||
return Ok(gateway); |
||||
} |
||||
}, |
||||
pnet_packet::ethernet::EtherTypes::Ipv6 => { |
||||
if let Some(ip_addr) = ipv6_handler(&frame) { |
||||
let gateway = Gateway { |
||||
mac_addr: MacAddr::new(frame.get_source().octets()), |
||||
ip_addr: ip_addr, |
||||
}; |
||||
return Ok(gateway); |
||||
} |
||||
}, |
||||
_ => {} |
||||
} |
||||
}, |
||||
Err(e) => { |
||||
return Err(format!("An error occurred while reading: {}", e)); |
||||
} |
||||
} |
||||
if Instant::now().duration_since(start_time) > timeout { |
||||
return Err(String::from("Recieve timeout")); |
||||
}else{ |
||||
match send_udp_packet() { |
||||
Ok(_) => (), |
||||
Err(e) => return Err(format!("Failed to send UDP packet {}", e)), |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
fn ipv4_handler(ethernet: &pnet_packet::ethernet::EthernetPacket) -> Option<IpAddr> { |
||||
if let Some(packet) = pnet_packet::ipv4::Ipv4Packet::new(ethernet.payload()) { |
||||
match packet.get_next_level_protocol() { |
||||
pnet_packet::ip::IpNextHeaderProtocols::Icmp => { |
||||
return icmp_handler(&packet); |
||||
}, |
||||
_ => { |
||||
None |
||||
} |
||||
} |
||||
}else{ |
||||
None |
||||
} |
||||
} |
||||
|
||||
fn ipv6_handler(ethernet: &pnet_packet::ethernet::EthernetPacket) -> Option<IpAddr> { |
||||
if let Some(packet) = pnet_packet::ipv6::Ipv6Packet::new(ethernet.payload()) { |
||||
match packet.get_next_header() { |
||||
pnet_packet::ip::IpNextHeaderProtocols::Icmpv6 => { |
||||
return icmpv6_handler(&packet); |
||||
}, |
||||
_ => { |
||||
None |
||||
} |
||||
} |
||||
}else{ |
||||
None |
||||
} |
||||
} |
||||
|
||||
fn icmp_handler(ip_packet: &pnet_packet::ipv4::Ipv4Packet) -> Option<IpAddr> { |
||||
if let Some(packet) = pnet_packet::icmp::IcmpPacket::new(ip_packet.payload()) { |
||||
if packet.get_icmp_type() == pnet_packet::icmp::IcmpTypes::TimeExceeded { |
||||
let ipv4_addr = ip_packet.get_source(); |
||||
return Some(IpAddr::V4(ipv4_addr)) |
||||
}else{ |
||||
None |
||||
} |
||||
}else{ |
||||
None |
||||
} |
||||
} |
||||
|
||||
fn icmpv6_handler(ip_packet: &pnet_packet::ipv6::Ipv6Packet) -> Option<IpAddr> { |
||||
if let Some(packet) = pnet_packet::icmpv6::Icmpv6Packet::new(ip_packet.payload()) { |
||||
if packet.get_icmpv6_type() == pnet_packet::icmpv6::Icmpv6Types::TimeExceeded { |
||||
let ipv6_addr = ip_packet.get_source(); |
||||
return Some(IpAddr::V6(ipv6_addr)) |
||||
}else{ |
||||
None |
||||
} |
||||
}else{ |
||||
None |
||||
} |
||||
} |
||||
|
||||
pub fn interfaces() -> Vec<Interface> { |
||||
let mut result: Vec<Interface> = vec![]; |
||||
let local_ip: IpAddr = match super::get_local_ipaddr(){ |
||||
Some(local_ip) => local_ip, |
||||
None => return result, |
||||
}; |
||||
let interfaces = pnet_datalink::interfaces(); |
||||
for iface in interfaces{ |
||||
let mac_addr: Option<MacAddr> = match iface.mac { |
||||
Some(mac_addr) => Some(MacAddr::new(mac_addr.octets())), |
||||
None => None, |
||||
}; |
||||
let mut ipv4_vec: Vec<Ipv4Addr> = vec![]; |
||||
let mut ipv6_vec: Vec<Ipv6Addr> = vec![]; |
||||
let mut ips: Vec<IpAddr> = vec![]; |
||||
for ip in &iface.ips { |
||||
match ip.ip() { |
||||
IpAddr::V4(ipv4_addr) => { |
||||
ipv4_vec.push(ipv4_addr); |
||||
}, |
||||
IpAddr::V6(ipv6_addr) => { |
||||
ipv6_vec.push(ipv6_addr); |
||||
}, |
||||
} |
||||
ips.push(ip.ip()); |
||||
} |
||||
let default_gateway: Option<Gateway> = if ips.contains(&local_ip) { |
||||
match get_default_gateway(iface.index) { |
||||
Ok(default_gateway) => Some(default_gateway), |
||||
Err(_) => None, |
||||
} |
||||
} else{ |
||||
None |
||||
}; |
||||
let desc: Option<String> = if iface.description.is_empty() { |
||||
None |
||||
} else{ |
||||
Some(iface.description) |
||||
}; |
||||
let interface: Interface = Interface{ |
||||
index: iface.index, |
||||
name: iface.name, |
||||
description: desc, |
||||
mac_addr: mac_addr, |
||||
ipv4: ipv4_vec, |
||||
ipv6: ipv6_vec, |
||||
gateway: default_gateway, |
||||
}; |
||||
result.push(interface); |
||||
} |
||||
return result; |
||||
} |
||||
|
||||
/// Get default Interface index
|
||||
pub fn default_interface_index() -> Option<u32> { |
||||
let local_ip: IpAddr = match super::get_local_ipaddr(){ |
||||
Some(local_ip) => local_ip, |
||||
None => return None, |
||||
}; |
||||
let interfaces = pnet_datalink::interfaces(); |
||||
for iface in interfaces { |
||||
for ip in iface.ips { |
||||
if local_ip == ip.ip() { |
||||
return Some(iface.index) |
||||
} |
||||
} |
||||
} |
||||
return None; |
||||
} |
||||
|
||||
/// Get default Interface name
|
||||
pub fn default_interface_name() -> Option<String> { |
||||
let local_ip: IpAddr = match super::get_local_ipaddr(){ |
||||
Some(local_ip) => local_ip, |
||||
None => return None, |
||||
}; |
||||
let interfaces = pnet_datalink::interfaces(); |
||||
for iface in interfaces { |
||||
for ip in iface.ips { |
||||
if local_ip == ip.ip() { |
||||
return Some(iface.name) |
||||
} |
||||
} |
||||
} |
||||
return None; |
||||
} |
||||
|
||||
Loading…
Reference in new issue