src/os/lsof.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
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,
    }
}