updated packages
This commit is contained in:
parent
abf1101484
commit
807dffa3be
4 changed files with 72 additions and 9 deletions
|
|
@ -1,13 +1,11 @@
|
|||
use std::{
|
||||
io::{Error, IoSlice},
|
||||
mem,
|
||||
net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddrV4},
|
||||
os::fd::{AsRawFd, FromRawFd, RawFd},
|
||||
};
|
||||
use std::{io::{Error, IoSlice}, mem, net::{Ipv4Addr, SocketAddrV4}, os::fd::{AsRawFd, FromRawFd, RawFd}, ptr};
|
||||
use std::net::{IpAddr, Ipv6Addr, SocketAddrV6};
|
||||
use std::ptr::addr_of;
|
||||
|
||||
use byteorder::{ByteOrder, NetworkEndian};
|
||||
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 tracing::{self, instrument};
|
||||
|
||||
|
|
|
|||
|
|
@ -114,6 +114,25 @@ impl TunInterface {
|
|||
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]
|
||||
#[instrument]
|
||||
pub fn set_broadcast_addr(&self, addr: Ipv4Addr) {
|
||||
|
|
|
|||
|
|
@ -1,11 +1,16 @@
|
|||
use std::{
|
||||
ffi::CStr,
|
||||
io::Error,
|
||||
mem,
|
||||
mem::MaybeUninit,
|
||||
net::{IpAddr, SocketAddr},
|
||||
os::fd::{AsRawFd, FromRawFd, IntoRawFd, RawFd},
|
||||
};
|
||||
|
||||
use tracing::instrument;
|
||||
|
||||
use crate::syscall;
|
||||
|
||||
mod queue;
|
||||
|
||||
#[cfg(target_vendor = "apple")]
|
||||
|
|
@ -60,6 +65,47 @@ impl TunInterface {
|
|||
pub fn set_nonblocking(&mut self, nb: bool) {
|
||||
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]
|
||||
|
|
|
|||
|
|
@ -48,8 +48,8 @@ fn test_set_get_ipv6() {
|
|||
let addr = Ipv6Addr::new(1, 1, 1, 1, 1, 1, 1, 1);
|
||||
tun.set_ipv6_addr(addr)?;
|
||||
|
||||
// let result = tun.ipv6_addr()?;
|
||||
// assert_eq!(addr, result);
|
||||
let result = tun.ipv6_addr()?;
|
||||
assert_eq!(addr, result);
|
||||
}
|
||||
|
||||
#[test]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue