Create set_ipv6_addr

This adds a new method for setting an ipv6 address on an interface
This commit is contained in:
Conrad Kramer 2023-05-20 05:07:14 -10:00
parent b94356dfb7
commit cc30fcd34c
3 changed files with 40 additions and 6 deletions

View file

@ -4,10 +4,12 @@ use socket2::{Domain, SockAddr, Socket, Type};
use std::fs::OpenOptions;
use std::io::Error;
use std::mem;
use std::net::{Ipv4Addr, SocketAddrV4};
use std::net::{Ipv4Addr, Ipv6Addr, SocketAddrV4, SocketAddrV6};
use std::os::fd::RawFd;
use std::os::unix::io::{AsRawFd, FromRawFd, IntoRawFd};
use libc::in6_ifreq;
use super::{ifname_to_string, string_to_ifname};
mod sys;
@ -51,6 +53,13 @@ impl TunInterface {
iff
}
#[throws]
fn in6_ifreq(&self) -> in6_ifreq {
let mut iff: in6_ifreq = unsafe { mem::zeroed() };
iff.ifr6_ifindex = self.index()?;
iff
}
#[throws]
pub fn index(&self) -> i32 {
let mut iff = self.ifreq()?;
@ -61,10 +70,8 @@ impl TunInterface {
#[throws]
pub fn set_ipv4_addr(&self, addr: Ipv4Addr) {
let addr = SockAddr::from(SocketAddrV4::new(addr, 0));
let mut iff = self.ifreq()?;
iff.ifr_ifru.ifru_addr = unsafe { *addr.as_ptr() };
self.perform(|fd| unsafe { sys::if_set_addr(fd, &iff) })?;
}
@ -93,6 +100,13 @@ impl TunInterface {
Ipv4Addr::from(u32::from_be(addr.sin_addr.s_addr))
}
#[throws]
pub fn set_ipv6_addr(&self, addr: Ipv6Addr) {
let mut iff = self.in6_ifreq()?;
iff.ifr6_addr.s6_addr = addr.octets();
self.perform6(|fd| unsafe { sys::if_set_addr6(fd, &iff) })?;
}
#[throws]
pub fn mtu(&self) -> i32 {
let mut iff = self.ifreq()?;
@ -118,6 +132,12 @@ impl TunInterface {
let socket = Socket::new(Domain::IPV4, Type::DGRAM, None)?;
perform(socket.as_raw_fd())?
}
#[throws]
fn perform6<R>(&self, perform: impl FnOnce(RawFd) -> Result<R, nix::Error>) -> R {
let socket = Socket::new(Domain::IPV6, Type::DGRAM, None)?;
perform(socket.as_raw_fd())?
}
}
#[cfg(test)]