rfmon

Author: dbcfd

src/config.rs

@@ -19,6 +19,7 @@ pub struct Config {
     bpf: Option<String>,
     buffer_for: std::time::Duration,
     blocking: bool,
+    rfmon: bool,
 }
 
 impl Config {
@@ -93,6 +94,15 @@ impl Config {
         self.blocking = blocking;
         self
     }
+
+    pub fn rfmon(&self) -> bool {
+        self.rfmon
+    }
+
+    pub fn with_rfmon(&mut self, rfmon: bool) -> &mut Self {
+        self.rfmon = rfmon;
+        self
+    }
 }
 
 impl Default for Config {
@@ -106,6 +116,7 @@ impl Default for Config {
             bpf: None,
             buffer_for: std::time::Duration::from_millis(100),
             blocking: false,
+            rfmon: false,
         }
     }
 }

src/handle.rs

@@ -185,6 +185,14 @@ impl PendingHandle {
         }
     }
 
+    pub fn set_rfmon(self) -> Result<Self, Error> {
+        if 0 != unsafe { pcap_sys::pcap_set_rfmon(self.handle, 1) } {
+            Err(pcap_util::convert_libpcap_error(self.handle))
+        } else {
+            Ok(self)
+        }
+    }
+
     pub fn activate(self) -> Result<Handle, Error> {
         let h = Handle {
             handle: self.handle,
@@ -216,6 +224,9 @@ impl std::convert::TryFrom<&Config> for PendingHandle {
                 .set_snaplen(v.snaplen())?
                 .set_promiscuous()?
                 .set_buffer_size(v.buffer_size())?;
+            if v.rfmon() {
+                pending = pending.set_rfmon()?;
+            }
         }
 
         Ok(pending)