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

use super::{link_message, WireGuardDeviceLinkOperation};
use crate::err::{ConnectError, LinkDeviceError};
use neli::consts::NlFamily;
use neli::socket::NlSocket;

pub struct RouteSocket {
    sock: NlSocket,
}

impl RouteSocket {
    pub fn connect() -> Result<Self, ConnectError> {
        let track_seq = true;
        let mut sock = NlSocket::new(NlFamily::Route, track_seq)?;

        // Autoselect a PID
        let pid = None;
        let groups = None;
        sock.bind(pid, groups)?;

        Ok(Self { sock })
    }

    pub fn add_device(&mut self, ifname: &str) -> Result<(), LinkDeviceError> {
        let operation = WireGuardDeviceLinkOperation::Add;
        self.sock.send_nl(link_message(ifname, operation)?)?;
        self.sock.recv_ack()?;
        Ok(())
    }

    pub fn del_device(&mut self, ifname: &str) -> Result<(), LinkDeviceError> {
        let operation = WireGuardDeviceLinkOperation::Delete;
        self.sock.send_nl(link_message(ifname, operation)?)?;
        self.sock.recv_ack()?;
        Ok(())
    }
}