Add abort scan command

uption · Oct 9, 2023 · 5cce245 · 5cce245
1 parent deb2cd6
commit 5cce245
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 0 deletions.
diff --git a/src/asynchronous.rs b/src/asynchronous.rs
@@ -144,6 +144,12 @@ impl AsyncNlSocket {
         Self::handle_ack_response(recv).await
     }
 
+    pub async fn abort_scan(&mut self, if_index: u32) -> Result<()> {
+        let request = Nl80211Request::abort_scan(if_index);
+        let recv = self.send(request).await?;
+        Self::handle_ack_response(recv).await
+    }
+
     async fn send(
         &self,
         request: Nl80211Request,

diff --git a/src/netlink.rs b/src/netlink.rs
@@ -313,4 +313,33 @@ impl Nl80211Request {
             ),
         }
     }
+
+    pub fn abort_scan(if_index: u32) -> Self {
+        let attrs = {
+            let mut attrs = GenlBuffer::new();
+            let attr_type = AttrTypeBuilder::default()
+                .nla_type(Attribute::Ifindex)
+                .build()
+                .unwrap();
+            attrs.push(
+                NlattrBuilder::default()
+                    .nla_type(attr_type)
+                    .nla_payload(if_index)
+                    .build()
+                    .unwrap(),
+            );
+            attrs
+        };
+        Self {
+            nl_flags: NlmF::REQUEST | NlmF::ACK,
+            nl_payload: NlPayload::Payload(
+                GenlmsghdrBuilder::<Command, Attribute, NoUserHeader>::default()
+                    .cmd(Command::AbortScan)
+                    .version(NL80211_VERSION)
+                    .attrs(attrs)
+                    .build()
+                    .unwrap(),
+            ),
+        }
+    }
 }
diff --git a/src/synchronous.rs b/src/synchronous.rs
@@ -123,12 +123,22 @@ impl NlSocket {
         Ok(responses)
     }
 
+    /// Trigger a new scan.
     pub fn trigger_scan(&mut self, if_index: u32) -> Result<()> {
         let request = Nl80211Request::trigger_scan(if_index);
         let recv = self.send(request)?;
         Self::handle_ack_response(recv)
     }
 
+    /// Stop an ongoing scan.
+    ///
+    /// Returns NlError ENOENT if a scan is not running.
+    pub fn abort_scan(&mut self, if_index: u32) -> Result<()> {
+        let request = Nl80211Request::abort_scan(if_index);
+        let recv = self.send(request)?;
+        Self::handle_ack_response(recv)
+    }
+
     fn send(
         &self,
         request: Nl80211Request,