wiphy: Add support for setting phy parameters

Author: brandonweeks

examples/nl80211_set_wiphy.rs

@@ -0,0 +1,44 @@
+// SPDX-License-Identifier: MIT
+
+use std::env::args;
+
+use anyhow::{bail, Context, Error};
+use futures::stream::TryStreamExt;
+
+fn main() -> Result<(), Error> {
+    let argv: Vec<_> = args().collect();
+
+    if argv.len() < 2 {
+        eprintln!("Usage: nl80211_set_wiphy <interface index>");
+        bail!("Required arguments not given");
+    }
+
+    let err_msg = format!("Invalid interface index value: {}", argv[1]);
+    let index = argv[1].parse::<u32>().context(err_msg)?;
+
+    let rt = tokio::runtime::Builder::new_current_thread()
+        .enable_io()
+        .enable_time()
+        .build()
+        .unwrap();
+    rt.block_on(set_wiphy_params(index));
+
+    Ok(())
+}
+
+async fn set_wiphy_params(if_index: u32) {
+    let (connection, handle, _) = wl_nl80211::new_connection().unwrap();
+    tokio::spawn(connection);
+
+    let attrs = wl_nl80211::Nl80211Channel::new(if_index)
+        .frequency(5180)
+        .frequency_offset(0)
+        .channel_width(wl_nl80211::Nl80211ChannelWidth::NoHt20)
+        .channel_type(wl_nl80211::Nl80211HtWiphyChannelType::NoHt)
+        .center_frequency(5180)
+        .build();
+
+    let mut channel_handle =
+        handle.wireless_physic().set(attrs).execute().await;
+    channel_handle.try_next().await.unwrap();
+}

src/lib.rs

@@ -83,7 +83,8 @@ pub use self::wifi7::{
 };
 pub use self::wiphy::{
     Nl80211Band, Nl80211BandInfo, Nl80211BandType, Nl80211BandTypes,
-    Nl80211CipherSuit, Nl80211Frequency, Nl80211FrequencyInfo, Nl80211IfMode,
+    Nl80211Channel, Nl80211ChannelSwitchRequest, Nl80211CipherSuit,
+    Nl80211Frequency, Nl80211FrequencyInfo, Nl80211IfMode,
     Nl80211WiphyGetRequest, Nl80211WiphyHandle, Nl80211WowlanTcpTrigerSupport,
     Nl80211WowlanTrigerPatternSupport, Nl80211WowlanTrigersSupport,
 };

src/wiphy/handle.rs

@@ -1,6 +1,10 @@
 // SPDX-License-Identifier: MIT
 
-use crate::{Nl80211Handle, Nl80211WiphyGetRequest};
+use crate::{
+    Nl80211Attr, Nl80211AttrsBuilder, Nl80211ChannelSwitchRequest,
+    Nl80211ChannelWidth, Nl80211Handle, Nl80211HtWiphyChannelType,
+    Nl80211WiphyGetRequest,
+};
 
 #[derive(Debug)]
 pub struct Nl80211WiphyHandle(Nl80211Handle);
@@ -15,4 +19,42 @@ impl Nl80211WiphyHandle {
     pub fn get(&mut self) -> Nl80211WiphyGetRequest {
         Nl80211WiphyGetRequest::new(self.0.clone())
     }
+
+    pub fn set(
+        &mut self,
+        attributes: Vec<Nl80211Attr>,
+    ) -> Nl80211ChannelSwitchRequest {
+        Nl80211ChannelSwitchRequest::new(self.0.clone(), attributes)
+    }
+}
+
+#[derive(Debug)]
+pub struct Nl80211Channel;
+
+impl Nl80211Channel {
+    pub fn new(if_index: u32) -> Nl80211AttrsBuilder<Self> {
+        Nl80211AttrsBuilder::<Self>::new().if_index(if_index)
+    }
+}
+
+impl Nl80211AttrsBuilder<Nl80211Channel> {
+    pub fn frequency(self, value: u32) -> Self {
+        self.replace(Nl80211Attr::WiphyFreq(value))
+    }
+
+    pub fn frequency_offset(self, value: u32) -> Self {
+        self.replace(Nl80211Attr::WiphyFreqOffset(value))
+    }
+
+    pub fn channel_width(self, value: Nl80211ChannelWidth) -> Self {
+        self.replace(Nl80211Attr::ChannelWidth(value))
+    }
+
+    pub fn channel_type(self, value: Nl80211HtWiphyChannelType) -> Self {
+        self.replace(Nl80211Attr::WiphyChannelType(value))
+    }
+
+    pub fn center_frequency(self, value: u32) -> Self {
+        self.replace(Nl80211Attr::CenterFreq1(value))
+    }
 }

src/wiphy/mod.rs

@@ -6,6 +6,7 @@ mod command;
 mod get;
 mod handle;
 mod ifmode;
+mod set;
 mod wowlan;
 
 pub use self::band::{
@@ -14,8 +15,9 @@ pub use self::band::{
 };
 pub use self::cipher::Nl80211CipherSuit;
 pub use self::get::Nl80211WiphyGetRequest;
-pub use self::handle::Nl80211WiphyHandle;
+pub use self::handle::{Nl80211Channel, Nl80211WiphyHandle};
 pub use self::ifmode::Nl80211IfMode;
+pub use self::set::Nl80211ChannelSwitchRequest;
 pub use self::wowlan::{
     Nl80211WowlanTcpTrigerSupport, Nl80211WowlanTrigerPatternSupport,
     Nl80211WowlanTrigersSupport,

src/wiphy/set.rs

@@ -0,0 +1,42 @@
+// SPDX-License-Identifier: MIT
+
+use futures::TryStream;
+use netlink_packet_core::{NLM_F_ACK, NLM_F_REQUEST};
+use netlink_packet_generic::GenlMessage;
+
+use crate::{
+    nl80211_execute, Nl80211Attr, Nl80211Command, Nl80211Error, Nl80211Handle,
+    Nl80211Message,
+};
+
+pub struct Nl80211ChannelSwitchRequest {
+    handle: Nl80211Handle,
+    attributes: Vec<Nl80211Attr>,
+}
+
+impl Nl80211ChannelSwitchRequest {
+    pub(crate) fn new(
+        handle: Nl80211Handle,
+        attributes: Vec<Nl80211Attr>,
+    ) -> Self {
+        Nl80211ChannelSwitchRequest { handle, attributes }
+    }
+
+    pub async fn execute(
+        self,
+    ) -> impl TryStream<Ok = GenlMessage<Nl80211Message>, Error = Nl80211Error>
+    {
+        let Nl80211ChannelSwitchRequest {
+            mut handle,
+            attributes,
+        } = self;
+
+        let nl80211_msg = Nl80211Message {
+            cmd: Nl80211Command::SetWiphy,
+            attributes,
+        };
+        let flags = NLM_F_REQUEST | NLM_F_ACK;
+
+        nl80211_execute(&mut handle, nl80211_msg, flags).await
+    }
+}