Add support for more operating systems.

Erk- · Erk- · commit dea95d406708 · 2020-01-12T23:19:57.000+01:00
This commit will add support for operating systems
that have the lsof tool, initially it will only work
on MacOS and FreeBSD, but more operating systems
may be added at a later time.

Signed-off-by: Valdemar Erk &lt;valdemar@erk.io&gt;
diff --git a/src/network/connection.rs b/src/network/connection.rs
@@ -14,7 +14,7 @@ impl Protocol {
     // Currently, linux implementation doesn't use this function.
     // Without this #[cfg] clippy complains about dead code, and CI refuses
     // to pass.
-    #[cfg(target_os = "macos")]
+    #[cfg(any(target_os = "macos", target_os = "freebsd"))]
     pub fn from_str(string: &str) -> Option<Self> {
         match string {
             "TCP" => Some(Protocol::Tcp),
diff --git a/src/os/lsof.rs b/src/os/lsof.rs
diff --git a/src/os/mod.rs b/src/os/mod.rs
@@ -1,10 +1,10 @@
 #[cfg(target_os = "linux")]
 pub(self) mod linux;
 
-#[cfg(target_os = "macos")]
-pub(self) mod macos;
+#[cfg(any(target_os = "macos", target_os = "freebsd"))]
+pub(self) mod lsof;
 
-#[cfg(target_os = "macos")]
+#[cfg(any(target_os = "macos", target_os = "freebsd"))]
 mod lsof_utils;
 
 mod shared;
diff --git a/src/os/shared.rs b/src/os/shared.rs
@@ -13,8 +13,8 @@ use signal_hook::iterator::Signals;
 
 #[cfg(target_os = "linux")]
 use crate::os::linux::get_open_sockets;
-#[cfg(target_os = "macos")]
-use crate::os::macos::get_open_sockets;
+#[cfg(any(target_os = "macos", target_os = "freebsd"))]
+use crate::os::lsof::get_open_sockets;
 use crate::{network::dns, OsInputOutput};
 
 pub type OnSigWinch = dyn Fn(Box<dyn Fn()>) + Send;
@@ -143,7 +143,7 @@ pub fn get_input(
 }
 
 #[inline]
-#[cfg(target_os = "macos")]
+#[cfg(any(target_os = "macos", target_os = "freebsd"))]
 fn eperm_message() -> &'static str {
     "Insufficient permissions to listen on network interface(s). Try running with sudo."
 }
