updated packages

This commit is contained in:
Jett Chen 2024-05-18 23:03:56 +08:00
parent abf1101484
commit 807dffa3be
4 changed files with 72 additions and 9 deletions

View file

@ -1,13 +1,11 @@
use std::{ use std::{io::{Error, IoSlice}, mem, net::{Ipv4Addr, SocketAddrV4}, os::fd::{AsRawFd, FromRawFd, RawFd}, ptr};
io::{Error, IoSlice}, use std::net::{IpAddr, Ipv6Addr, SocketAddrV6};
mem, use std::ptr::addr_of;
net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddrV4},
os::fd::{AsRawFd, FromRawFd, RawFd},
};
use byteorder::{ByteOrder, NetworkEndian}; use byteorder::{ByteOrder, NetworkEndian};
use fehler::throws; use fehler::throws;
use libc::{c_char, iovec, writev, AF_INET, AF_INET6}; use libc::{c_char, iovec, sockaddr_in6, writev, AF_INET, AF_INET6};
use nix::sys::socket::SockaddrIn6;
use socket2::{Domain, SockAddr, Socket, Type}; use socket2::{Domain, SockAddr, Socket, Type};
use tracing::{self, instrument}; use tracing::{self, instrument};

View file

@ -114,6 +114,25 @@ impl TunInterface {
Ipv4Addr::from(u32::from_be(addr.sin_addr.s_addr)) Ipv4Addr::from(u32::from_be(addr.sin_addr.s_addr))
} }
#[throws]
#[instrument]
pub fn ipv6_addrs(&self) -> Vec<Ipv6Addr> {
let ip_addrs = self.ip_addrs()?;
let mut ipv6_addrs: Vec<Ipv6Addr> = vec![];
for ip_addr in ip_addrs.iter() {
if ip_addr.is_ipv6() {
match ip_addr {
IpAddr::V6(addr) => {
ipv6_addrs.push(*addr);
}
_ => {}
}
}
}
ipv6_addrs
}
#[throws] #[throws]
#[instrument] #[instrument]
pub fn set_broadcast_addr(&self, addr: Ipv4Addr) { pub fn set_broadcast_addr(&self, addr: Ipv4Addr) {

View file

@ -1,11 +1,16 @@
use std::{ use std::{
ffi::CStr,
io::Error, io::Error,
mem,
mem::MaybeUninit, mem::MaybeUninit,
net::{IpAddr, SocketAddr},
os::fd::{AsRawFd, FromRawFd, IntoRawFd, RawFd}, os::fd::{AsRawFd, FromRawFd, IntoRawFd, RawFd},
}; };
use tracing::instrument; use tracing::instrument;
use crate::syscall;
mod queue; mod queue;
#[cfg(target_vendor = "apple")] #[cfg(target_vendor = "apple")]
@ -60,6 +65,47 @@ impl TunInterface {
pub fn set_nonblocking(&mut self, nb: bool) { pub fn set_nonblocking(&mut self, nb: bool) {
self.socket.set_nonblocking(nb)?; self.socket.set_nonblocking(nb)?;
} }
#[throws]
#[instrument]
pub fn ip_addrs(&self) -> Vec<IpAddr> {
let mut result: Vec<IpAddr> = vec![];
let mut addrs: *mut libc::ifaddrs = std::ptr::null_mut();
let if_name = self.name()?;
syscall!(getifaddrs(&mut addrs as *mut _))?;
unsafe {
while !addrs.is_null() {
let addr = &*addrs;
addrs = addr.ifa_next;
let name = CStr::from_ptr(addr.ifa_name).to_str().unwrap();
if if_name != name {
continue;
}
let family = (*addr.ifa_addr).sa_family;
let addr_len = match family as i32 {
libc::AF_INET => mem::size_of::<libc::sockaddr_in>(),
libc::AF_INET6 => mem::size_of::<libc::sockaddr_in6>(),
_ => continue,
};
let (_, sock_addr) = socket2::SockAddr::try_init(|addr_storage, len| {
*len = addr_len as u32;
std::ptr::copy_nonoverlapping(
addr.ifa_addr as *const libc::c_void,
addr_storage as *mut _,
addr_len,
);
Ok(())
})?;
if let Some(socket_addr) = sock_addr.as_socket() {
result.push(socket_addr.ip());
}
}
}
result
}
} }
#[instrument] #[instrument]

View file

@ -48,8 +48,8 @@ fn test_set_get_ipv6() {
let addr = Ipv6Addr::new(1, 1, 1, 1, 1, 1, 1, 1); let addr = Ipv6Addr::new(1, 1, 1, 1, 1, 1, 1, 1);
tun.set_ipv6_addr(addr)?; tun.set_ipv6_addr(addr)?;
// let result = tun.ipv6_addr()?; let result = tun.ipv6_addr()?;
// assert_eq!(addr, result); assert_eq!(addr, result);
} }
#[test] #[test]