wiretun 0.5.0

WireGuard Library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

use std::mem;
use std::os::fd::{AsRawFd, RawFd};

use libc::{__c_anonymous_ifr_ifru, c_char, ifreq};
use nix::fcntl::{fcntl, FcntlArg, OFlag};
use nix::sys::socket::{socket, AddressFamily, SockFlag, SockType};
use nix::{ioctl_read_bad, ioctl_write_ptr_bad};

use crate::tun::Error;

ioctl_write_ptr_bad!(ioctl_tun_set_iff, 0x400454ca, ifreq);
ioctl_read_bad!(ioctl_tun_get_iff, 0x800454d2, ifreq);
ioctl_write_ptr_bad!(ioctl_set_mtu, 0x8922, ifreq);
ioctl_read_bad!(ioctl_get_mtu, 0x8921, ifreq);

pub fn new_ifreq(name: &str) -> ifreq {
    let mut ifr: ifreq = unsafe { mem::zeroed() };
    let ifr_name: Vec<c_char> = name.as_bytes().iter().map(|c| *c as _).collect();
    ifr.ifr_name[..name.len()].copy_from_slice(&ifr_name);
    ifr
}

pub fn set_nonblocking(fd: RawFd) -> Result<(), Error> {
    let flag = fcntl(fd, FcntlArg::F_GETFL)
        .map(OFlag::from_bits_retain)
        .map_err(Error::Sys)?;
    let flag = OFlag::O_NONBLOCK | flag;
    fcntl(fd, FcntlArg::F_SETFL(flag)).map_err(Error::Sys)?;
    Ok(())
}

pub fn set_mtu(name: &str, mtu: u16) -> Result<(), Error> {
    let fd = socket(
        AddressFamily::Inet,
        SockType::Datagram,
        SockFlag::empty(),
        None,
    )
    .map_err(Error::Sys)?;
    let mut ifr = new_ifreq(name);
    ifr.ifr_ifru = __c_anonymous_ifr_ifru { ifru_mtu: mtu as _ };
    unsafe { ioctl_set_mtu(fd.as_raw_fd(), &ifr) }.map_err(Error::Sys)?;
    Ok(())
}

pub fn get_mtu(name: &str) -> Result<u16, Error> {
    let fd = socket(
        AddressFamily::Inet,
        SockType::Datagram,
        SockFlag::empty(),
        None,
    )
    .map_err(Error::Sys)?;
    let mut ifr = new_ifreq(name);
    unsafe { ioctl_get_mtu(fd.as_raw_fd(), &mut ifr) }.map_err(Error::Sys)?;
    Ok(unsafe { ifr.ifr_ifru.ifru_mtu as _ })
}