daisyway 0.1.3

ETSI-014 adapter for WireGuard (Create WireGuard connections secured by quantum key distribution)
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

use std::{
    future::Future,
    sync::{Arc, Mutex},
};

use anyhow::{Context, Result, ensure};
use log::{error, info};
#[cfg(target_os = "linux")]
use wireguard_uapi::{DeviceInterface, WgSocket};

use super::{OskHandler, SetOskReason};
use crate::internal::{daisyway::crypto::Key, util::base64_to_key};

#[derive(Clone)]
pub struct WireGuardOskHandler {
    pub socket: Arc<Mutex<WgSocket>>,
    pub interface: String,
    pub peer_id: Key,
}

impl WireGuardOskHandler {
    pub fn setup(peer_id: &str, interface: &str) -> Result<Self> {
        let peer_id_u8 = base64_to_key(peer_id.as_bytes()).expect("Invalid peer_id");

        #[cfg(target_os = "linux")]
        let socket = {
            let mut socket =
                WgSocket::connect().context("Failed to connect to WireGuard control socket.")?;

            let device = socket
                .get_device(DeviceInterface::from_name(interface.to_string()))
                .with_context(|| format!("Failed to access WireGuard interface {interface}"))?;
            ensure!(
                device.peers.iter().any(|p| p.public_key == peer_id_u8),
                "Could not find WireGuard peer {peer_id}"
            );

            Arc::new(Mutex::new(socket))
        };

        Ok(Self {
            socket,
            interface: interface.to_owned(),
            peer_id: peer_id_u8.to_owned(),
        })
    }

    async fn set_osk_impl(&self, key: Key, reason: SetOskReason) -> Result<()> {
        use SetOskReason as R;
        match reason {
            R::Fresh => info!(
                "Injecting fresh PSK into WireGuard interface {}",
                self.interface
            ),
            R::Stale => error!(
                "Erasing stale PSK in WireGuard interface {} by overwriting with a random key",
                self.interface
            ),
        };

        let mut set_peer = wireguard_uapi::set::Peer::from_public_key(&self.peer_id);
        set_peer
            .flags
            .push(wireguard_uapi::set::WgPeerF::UpdateOnly);
        set_peer.preshared_key = Some(&key);
        let mut set_dev = wireguard_uapi::set::Device::from_ifname(&self.interface);
        set_dev.peers.push(set_peer);

        self.socket.lock().unwrap().set_device(set_dev)?;

        Ok(())
    }
}

impl OskHandler for WireGuardOskHandler {
    fn set_osk(&self, key: Key, reason: SetOskReason) -> impl Future<Output = Result<()>> {
        self.set_osk_impl(key, reason)
    }
}

impl std::fmt::Debug for WireGuardOskHandler {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("WireGuardOskHandler")
            .field("socket", &"...")
            .field("interface", &self.interface)
            .field("peer_id", &self.peer_id)
            .finish()
    }
}