wifi: Split out from manager

src/wifi.rs

@@ -0,0 +1,144 @@
+/*
+ * Copyright © 2023 Collabora Ltd.
+ * Copyright © 2024 Valve Software
+ *
+ * SPDX-License-Identifier: MIT
+ */
+
+use anyhow::Result;
+use std::fmt;
+use tokio::fs;
+use tracing::error;
+
+use crate::process::{run_script, SYSTEMCTL_PATH};
+
+const OVERRIDE_CONTENTS: &str = "[Service]
+ExecStart=
+ExecStart=/usr/lib/iwd/iwd -d
+";
+const OVERRIDE_FOLDER: &str = "/etc/systemd/system/iwd.service.d";
+const OVERRIDE_PATH: &str = "/etc/systemd/system/iwd.service.d/override.conf";
+
+// Only use one path for output for now. If needed we can add a timestamp later
+// to have multiple files, etc.
+const OUTPUT_FILE: &str = "/var/log/wifitrace.dat";
+const TRACE_CMD_PATH: &str = "/usr/bin/trace-cmd";
+
+#[derive(PartialEq, Debug, Copy, Clone)]
+#[repr(u32)]
+pub enum WifiDebugMode {
+    Off,
+    On,
+}
+
+#[derive(PartialEq, Debug, Copy, Clone)]
+#[repr(u32)]
+pub enum WifiPowerManagement {
+    UnsupportedFeature = 0,
+    Disabled = 1,
+    Enabled = 2,
+}
+
+impl TryFrom<u32> for WifiDebugMode {
+    type Error = &'static str;
+    fn try_from(v: u32) -> Result<Self, Self::Error> {
+        match v {
+            x if x == WifiDebugMode::Off as u32 => Ok(WifiDebugMode::Off),
+            x if x == WifiDebugMode::On as u32 => Ok(WifiDebugMode::On),
+            _ => Err("No enum match for value {v}"),
+        }
+    }
+}
+
+impl fmt::Display for WifiDebugMode {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        match self {
+            WifiDebugMode::Off => write!(f, "Off"),
+            WifiDebugMode::On => write!(f, "On"),
+        }
+    }
+}
+
+impl TryFrom<u32> for WifiPowerManagement {
+    type Error = &'static str;
+    fn try_from(v: u32) -> Result<Self, Self::Error> {
+        match v {
+            x if x == WifiPowerManagement::UnsupportedFeature as u32 => {
+                Ok(WifiPowerManagement::UnsupportedFeature)
+            }
+            x if x == WifiPowerManagement::Disabled as u32 => Ok(WifiPowerManagement::Disabled),
+            x if x == WifiPowerManagement::Enabled as u32 => Ok(WifiPowerManagement::Enabled),
+            _ => Err("No enum match for value {v}"),
+        }
+    }
+}
+
+impl fmt::Display for WifiPowerManagement {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        match self {
+            WifiPowerManagement::UnsupportedFeature => write!(f, "Unsupported feature"),
+            WifiPowerManagement::Disabled => write!(f, "Disabled"),
+            WifiPowerManagement::Enabled => write!(f, "Enabled"),
+        }
+    }
+}
+
+pub async fn setup_iwd_config(want_override: bool) -> std::io::Result<()> {
+    // Copy override.conf file into place or out of place depending
+    // on install value
+
+    if want_override {
+        // Copy it in
+        // Make sure the folder exists
+        fs::create_dir_all(OVERRIDE_FOLDER).await?;
+        // Then write the contents into the file
+        fs::write(OVERRIDE_PATH, OVERRIDE_CONTENTS).await
+    } else {
+        // Delete it
+        fs::remove_file(OVERRIDE_PATH).await
+    }
+}
+
+pub async fn restart_iwd() -> Result<bool> {
+    // First reload systemd since we modified the config most likely
+    // otherwise we wouldn't be restarting iwd.
+    match run_script("reload systemd", SYSTEMCTL_PATH, &["daemon-reload"]).await {
+        Ok(value) => {
+            if value {
+                // worked, now restart iwd
+                run_script("restart iwd", SYSTEMCTL_PATH, &["restart", "iwd"]).await
+            } else {
+                // reload failed
+                error!("restart_iwd: reload systemd failed with non-zero exit code");
+                Ok(false)
+            }
+        }
+        Err(message) => {
+            error!("restart_iwd: reload systemd got an error: {message}");
+            Err(message)
+        }
+    }
+}
+
+pub async fn stop_tracing() -> Result<bool> {
+    // Stop tracing and extract ring buffer to disk for capture
+    run_script("stop tracing", TRACE_CMD_PATH, &["stop"]).await?;
+    // stop tracing worked
+    run_script(
+        "extract traces",
+        TRACE_CMD_PATH,
+        &["extract", "-o", OUTPUT_FILE],
+    )
+    .await
+}
+
+pub async fn start_tracing(buffer_size: u32) -> Result<bool> {
+    // Start tracing
+    let size_str = format!("{}", buffer_size);
+    run_script(
+        "start tracing",
+        TRACE_CMD_PATH,
+        &["start", "-e", "ath11k_wmi_diag", "-b", &size_str],
+    )
+    .await
+}