Add survey dump

Signed-off-by: Fabian Viöl <fv@enqt.de>

Author: Sh3Rm4n

examples/dump_nl80211_survey.rs

@@ -0,0 +1,43 @@
+// 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: dump_nl80211_survey <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()
+        .build()
+        .unwrap();
+    rt.block_on(dump_survey(index))?;
+
+    Ok(())
+}
+
+async fn dump_survey(if_index: u32) -> Result<(), Error> {
+    let (connection, handle, _) = wl_nl80211::new_connection()?;
+    tokio::spawn(connection);
+
+    let mut survey_handle = handle.survey().dump(if_index).execute().await;
+
+    let mut msgs = Vec::new();
+    while let Some(msg) = survey_handle.try_next().await? {
+        msgs.push(msg);
+    }
+    assert!(!msgs.is_empty());
+    for msg in msgs {
+        println!("{:?}", msg);
+    }
+    Ok(())
+}

src/attr.rs

@@ -47,8 +47,8 @@ use crate::{
     Nl80211IfTypeExtCapas, Nl80211IfaceComb, Nl80211IfaceFrameType,
     Nl80211InterfaceType, Nl80211InterfaceTypes, Nl80211MloLink,
     Nl80211ScanFlags, Nl80211SchedScanMatch, Nl80211SchedScanPlan,
-    Nl80211StationInfo, Nl80211TransmitQueueStat, Nl80211VhtCapability,
-    Nl80211WowlanTrigersSupport,
+    Nl80211StationInfo, Nl80211SurveyInfo, Nl80211TransmitQueueStat,
+    Nl80211VhtCapability, Nl80211WowlanTrigersSupport,
 };
 
 const ETH_ALEN: usize = 6;
@@ -204,7 +204,7 @@ const NL80211_ATTR_WIPHY_RTS_THRESHOLD: u16 = 64;
 // const NL80211_ATTR_KEYS:u16 = 81;
 // const NL80211_ATTR_PID:u16 = 82;
 const NL80211_ATTR_4ADDR: u16 = 83;
-// const NL80211_ATTR_SURVEY_INFO:u16 = 84;
+const NL80211_ATTR_SURVEY_INFO: u16 = 84;
 // const NL80211_ATTR_PMKID:u16 = 85;
 const NL80211_ATTR_MAX_NUM_PMKIDS: u16 = 86;
 // const NL80211_ATTR_DURATION:u16 = 87;
@@ -480,6 +480,7 @@ pub enum Nl80211Attr {
     WiphyTxPowerLevel(u32),
     Ssid(String),
     StationInfo(Vec<Nl80211StationInfo>),
+    SurveyInfo(Vec<Nl80211SurveyInfo>),
     TransmitQueueStats(Vec<Nl80211TransmitQueueStat>),
     TransmitQueueLimit(u32),
     TransmitQueueMemoryLimit(u32),
@@ -625,6 +626,7 @@ impl Nla for Nl80211Attr {
             | Self::MaxNumPmkids(_) => 1,
             Self::TransmitQueueStats(nlas) => nlas.as_slice().buffer_len(),
             Self::StationInfo(nlas) => nlas.as_slice().buffer_len(),
+            Self::SurveyInfo(nlas) => nlas.as_slice().buffer_len(),
             Self::MloLinks(links) => links.as_slice().buffer_len(),
             Self::MaxScanIeLen(_) | Self::MaxSchedScanIeLen(_) => 2,
             Self::SupportIbssRsn
@@ -704,6 +706,7 @@ impl Nla for Nl80211Attr {
             Self::WiphyTxPowerLevel(_) => NL80211_ATTR_WIPHY_TX_POWER_LEVEL,
             Self::Ssid(_) => NL80211_ATTR_SSID,
             Self::StationInfo(_) => NL80211_ATTR_STA_INFO,
+            Self::SurveyInfo(_) => NL80211_ATTR_SURVEY_INFO,
             Self::TransmitQueueStats(_) => NL80211_ATTR_TXQ_STATS,
             Self::TransmitQueueLimit(_) => NL80211_ATTR_TXQ_LIMIT,
             Self::TransmitQueueMemoryLimit(_) => NL80211_ATTR_TXQ_MEMORY_LIMIT,
@@ -840,6 +843,7 @@ impl Nla for Nl80211Attr {
             Self::WiphyChannelType(d) => write_u32(buffer, (*d).into()),
             Self::ChannelWidth(d) => write_u32(buffer, (*d).into()),
             Self::StationInfo(nlas) => nlas.as_slice().emit(buffer),
+            Self::SurveyInfo(nlas) => nlas.as_slice().emit(buffer),
             Self::TransmitQueueStats(nlas) => nlas.as_slice().emit(buffer),
             Self::MloLinks(links) => links.as_slice().emit(buffer),
             Self::WiphyRetryShort(d)
@@ -1061,6 +1065,20 @@ impl<'a, T: AsRef<[u8]> + ?Sized> Parseable<NlaBuffer<&'a T>> for Nl80211Attr {
                 }
                 Self::StationInfo(nlas)
             }
+            NL80211_ATTR_SURVEY_INFO => {
+                let err_msg = format!(
+                    "Invalid NL80211_ATTR_SURVEY_INFO value {payload:?}"
+                );
+                let mut nlas = Vec::new();
+                for nla in NlasIterator::new(payload) {
+                    let nla = &nla.context(err_msg.clone())?;
+                    nlas.push(
+                        Nl80211SurveyInfo::parse(nla)
+                            .context(err_msg.clone())?,
+                    );
+                }
+                Self::SurveyInfo(nlas)
+            }
             NL80211_ATTR_TXQ_STATS => {
                 let err_msg = format!(
                     "Invalid NL80211_ATTR_TXQ_STATS value {:?}",

src/handle.rs

@@ -8,7 +8,8 @@ use netlink_packet_utils::DecodeError;
 
 use crate::{
     try_nl80211, Nl80211Error, Nl80211InterfaceHandle, Nl80211Message,
-    Nl80211ScanHandle, Nl80211StationHandle, Nl80211WiphyHandle,
+    Nl80211ScanHandle, Nl80211StationHandle, Nl80211SurveyHandle,
+    Nl80211WiphyHandle,
 };
 
 #[derive(Clone, Debug)]
@@ -41,6 +42,12 @@ impl Nl80211Handle {
         Nl80211ScanHandle::new(self.clone())
     }
 
+    // equivalent to `iw DEVICE survey dump` command
+    #[must_use]
+    pub fn survey(&self) -> Nl80211SurveyHandle {
+        Nl80211SurveyHandle::new(self.clone())
+    }
+
     pub async fn request(
         &mut self,
         message: NetlinkMessage<GenlMessage<Nl80211Message>>,

src/lib.rs

@@ -18,6 +18,7 @@ mod mlo;
 mod scan;
 mod station;
 mod stats;
+mod survey;
 mod wifi4;
 mod wifi5;
 mod wifi6;
@@ -63,6 +64,9 @@ pub use self::station::{
 pub use self::stats::{
     NestedNl80211TidStats, Nl80211TidStats, Nl80211TransmitQueueStat,
 };
+pub use self::survey::{
+    Nl80211SurveyGetRequest, Nl80211SurveyHandle, Nl80211SurveyInfo,
+};
 pub use self::wifi4::{
     Nl80211ElementHtCap, Nl80211HtAMpduPara, Nl80211HtAselCaps,
     Nl80211HtCapabilityMask, Nl80211HtCaps, Nl80211HtExtendedCap,

src/survey/get.rs

@@ -0,0 +1,41 @@
+// SPDX-License-Identifier: MIT
+
+use futures::TryStream;
+use netlink_packet_core::{NLM_F_DUMP, NLM_F_REQUEST};
+use netlink_packet_generic::GenlMessage;
+
+use crate::{
+    nl80211_execute, Nl80211Attr, Nl80211Command, Nl80211Error, Nl80211Handle,
+    Nl80211Message,
+};
+
+pub struct Nl80211SurveyGetRequest {
+    handle: Nl80211Handle,
+    if_index: u32,
+}
+
+impl Nl80211SurveyGetRequest {
+    pub(crate) fn new(handle: Nl80211Handle, if_index: u32) -> Self {
+        Self { handle, if_index }
+    }
+
+    pub async fn execute(
+        self,
+    ) -> impl TryStream<Ok = GenlMessage<Nl80211Message>, Error = Nl80211Error>
+    {
+        let Self {
+            mut handle,
+            if_index,
+        } = self;
+
+        let attributes = vec![Nl80211Attr::IfIndex(if_index)];
+        let nl80211_msg = Nl80211Message {
+            cmd: Nl80211Command::GetSurvey,
+            attributes,
+        };
+
+        let flags = NLM_F_REQUEST | NLM_F_DUMP;
+
+        nl80211_execute(&mut handle, nl80211_msg, flags).await
+    }
+}

src/survey/handle.rs

@@ -0,0 +1,19 @@
+// SPDX-License-Identifier: MIT
+
+use crate::{Nl80211Handle, Nl80211SurveyGetRequest};
+
+pub struct Nl80211SurveyHandle(Nl80211Handle);
+
+impl Nl80211SurveyHandle {
+    #[must_use]
+    pub fn new(handle: Nl80211Handle) -> Self {
+        Self(handle)
+    }
+
+    /// Retrieve the survey info
+    /// (equivalent to `iw dev DEV survey dump`)
+    #[must_use]
+    pub fn dump(&mut self, if_index: u32) -> Nl80211SurveyGetRequest {
+        Nl80211SurveyGetRequest::new(self.0.clone(), if_index)
+    }
+}

src/survey/mod.rs

@@ -0,0 +1,7 @@
+mod get;
+mod handle;
+mod survey_info;
+
+pub use self::get::Nl80211SurveyGetRequest;
+pub use self::handle::Nl80211SurveyHandle;
+pub use self::survey_info::Nl80211SurveyInfo;