1 files changed, 42 insertions, 0 deletions
diff --git a/src/os/lsof.rs b/src/os/lsof.rs
new file mode 100644
index 0000000..68b4bad
--- /dev/null
+++ b/src/os/lsof.rs
@@ -0,0 +1,42 @@
+use ::std::collections::HashMap;
+
+use crate::network::Connection;
+use crate::OpenSockets;
+
+use super::lsof_utils;
+use std::net::SocketAddr;
+
+#[derive(Debug)]
+struct RawConnection {
+    ip: String,
+    local_port: String,
+    remote_port: String,
+    protocol: String,
+    process_name: String,
+}
+
+pub(crate) fn get_open_sockets() -> OpenSockets {
+    let mut open_sockets = HashMap::new();
+    let mut connections_vec = std::vec::Vec::new();
+
+    let connections = lsof_utils::get_connections();
+
+    for raw_connection in connections {
+        let protocol = raw_connection.get_protocol();
+        let remote_ip = raw_connection.get_remote_ip();
+        let local_ip = raw_connection.get_local_ip();
+        let remote_port = raw_connection.get_remote_port();
+        let local_port = raw_connection.get_local_port();
+
+        let socket_addr = SocketAddr::new(remote_ip, remote_port);
+        let connection = Connection::new(socket_addr, local_ip, local_port, protocol).unwrap();
+
+        open_sockets.insert(connection.local_socket, raw_connection.process_name.clone());
+        connections_vec.push(connection);
+    }
+
+    OpenSockets {
+        sockets_to_procs: open_sockets,
+        connections: connections_vec,
+    }
+}