Make blocking configurable, defaulting to non-blocking (#23)

Author: dbcfd

src/config.rs

@@ -7,6 +7,7 @@ pub struct Config {
     buffer_size: u32,
     bpf: Option<String>,
     buffer_for: std::time::Duration,
+    blocking: bool,
 }
 
 impl Config {
@@ -55,19 +56,30 @@ impl Config {
         self
     }
 
+    pub fn blocking(&self) -> bool {
+        self.blocking
+    }
+
+    pub fn with_blocking(&mut self, blocking: bool) -> &mut Self {
+        self.blocking = blocking;
+        self
+    }
+
     pub fn new(
         max_packets_read: usize,
         snaplen: u32,
         buffer_size: u32,
         bpf: Option<String>,
         buffer_for: std::time::Duration,
+        blocking: bool,
     ) -> Config {
         Config {
             max_packets_read,
             snaplen,
             buffer_size,
             bpf,
             buffer_for,
+            blocking,
         }
     }
 }
@@ -80,6 +92,7 @@ impl Default for Config {
             buffer_size: 16777216,
             bpf: None,
             buffer_for: std::time::Duration::from_millis(100),
+            blocking: false,
         }
     }
 }

src/stream.rs

@@ -33,7 +33,9 @@ impl PacketStream {
                 .set_promiscuous()?
                 .set_buffer_size(config.buffer_size())?
                 .activate()?;
-            h.set_non_block()?;
+            if !config.blocking() {
+                h.set_non_block()?;
+            }
 
             if let Some(bpf) = config.bpf() {
                 let bpf = handle.compile_bpf(bpf)?;