nmrs 3.0.0

A Rust library for NetworkManager over D-Bus
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131

//! Per-Wi-Fi-device enumeration and control.
//!
//! NetworkManager's global `WirelessEnabled` flag toggles every Wi-Fi radio
//! at once. To disable one specific Wi-Fi device while leaving the others
//! online, we set `Device.Autoconnect = false` and then call
//! `Device.Disconnect()` — the kernel hands the radio back to NM but no
//! connection will be re-activated until autoconnect is re-enabled.

use log::{debug, warn};
use zbus::Connection;

use crate::Result;
use crate::api::models::{ConnectionError, WifiDevice};
use crate::core::connection::{disconnect_wifi_and_wait, get_device_by_interface};
use crate::dbus::{NMAccessPointProxy, NMDeviceProxy, NMProxy, NMWirelessProxy};
use crate::types::constants::device_type;
use crate::util::utils::decode_ssid_or_hidden;

/// Lists every managed Wi-Fi device with current MAC, state, and active SSID.
pub(crate) async fn list_wifi_devices(conn: &Connection) -> Result<Vec<WifiDevice>> {
    let nm = NMProxy::new(conn).await?;
    let paths = nm.get_devices().await?;

    let mut out = Vec::new();
    for p in paths {
        let dev = NMDeviceProxy::builder(conn)
            .path(p.clone())?
            .build()
            .await?;
        if dev.device_type().await? != device_type::WIFI {
            continue;
        }

        let interface = dev.interface().await.unwrap_or_default();
        let hw_address = dev
            .hw_address()
            .await
            .unwrap_or_else(|_| String::from("00:00:00:00:00:00"));
        let permanent_hw_address = dev.perm_hw_address().await.ok();
        let driver = dev.driver().await.ok();
        let state = dev.state().await?.into();
        let managed = dev.managed().await.unwrap_or(false);
        let autoconnect = dev.autoconnect().await.unwrap_or(true);

        let wifi = NMWirelessProxy::builder(conn)
            .path(p.clone())?
            .build()
            .await?;
        let active_ap_path = wifi.active_access_point().await.ok();
        let (is_active, active_ssid) = match active_ap_path {
            Some(ap_path) if ap_path.as_str() != "/" => {
                match NMAccessPointProxy::builder(conn)
                    .path(ap_path)?
                    .build()
                    .await
                {
                    Ok(ap) => match ap.ssid().await {
                        Ok(bytes) => (true, Some(decode_ssid_or_hidden(&bytes).into_owned())),
                        Err(_) => (true, None),
                    },
                    Err(_) => (true, None),
                }
            }
            _ => (false, None),
        };

        out.push(WifiDevice {
            path: p,
            interface,
            hw_address,
            permanent_hw_address,
            driver,
            state,
            managed,
            autoconnect,
            is_active,
            active_ssid,
        });
    }

    Ok(out)
}

/// Disable or re-enable a single Wi-Fi device.
///
/// `enabled = false` clears `Device.Autoconnect` and disconnects the device.
/// `enabled = true` re-enables autoconnect; NM will activate any saved
/// connection on its own.
///
/// This is independent of NetworkManager's global `WirelessEnabled` killswitch
/// (controlled via [`crate::NetworkManager::set_wireless_enabled`]).
pub(crate) async fn set_wifi_enabled_for_interface(
    conn: &Connection,
    interface: &str,
    enabled: bool,
) -> Result<()> {
    let path = match get_device_by_interface(conn, interface).await {
        Ok(p) => p,
        Err(ConnectionError::NotFound) => {
            return Err(ConnectionError::WifiInterfaceNotFound {
                interface: interface.to_string(),
            });
        }
        Err(e) => return Err(e),
    };

    let dev = NMDeviceProxy::builder(conn)
        .path(path.clone())?
        .build()
        .await?;
    if dev.device_type().await? != device_type::WIFI {
        return Err(ConnectionError::NotAWifiDevice {
            interface: interface.to_string(),
        });
    }

    debug!("setting Autoconnect={} for {}", enabled, interface);
    if let Err(e) = dev.set_autoconnect(enabled).await {
        warn!("failed to set autoconnect on {}: {}", interface, e);
        return Err(ConnectionError::DbusOperation {
            context: format!("failed to set Autoconnect on {}", interface),
            source: e,
        });
    }

    if !enabled {
        disconnect_wifi_and_wait(conn, &path, None).await?;
    }

    Ok(())
}