blob: 31e32668a44a9c84ddaf79fe80ef16ddecf15e4c (
plain)
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);
open_sockets.insert(connection.local_socket, raw_connection.process_name.clone());
connections_vec.push(connection);
}
OpenSockets {
sockets_to_procs: open_sockets,
connections: connections_vec,
}
}
|