47 lines
1.5 KiB
Rust
47 lines
1.5 KiB
Rust
|
use netstat2::{AddressFamilyFlags, ProtocolFlags, ProtocolSocketInfo};
|
||
|
|
use sysinfo::{System, SystemExt, Pid, ProcessExt};
|
||
|
|
|
||
|
|
use crate::errors::*;
|
||
|
|
|
||
|
|
|
||
|
|
fn get_associated_pids(local_port: u16) -> Result<Vec<u32>, netstat2::error::Error> {
|
||
|
|
let mut it = netstat2::iterate_sockets_info(
|
||
|
|
AddressFamilyFlags::IPV4,
|
||
|
|
ProtocolFlags::TCP
|
||
|
|
)?;
|
||
|
|
|
||
|
|
for (i, item) in it.enumerate() {
|
||
|
|
let sock_info = item?;
|
||
|
|
let proto_info = match sock_info.protocol_socket_info {
|
||
|
|
ProtocolSocketInfo::Tcp(tcp_info) => tcp_info,
|
||
|
|
ProtocolSocketInfo::Udp(_) => {continue;}
|
||
|
|
};
|
||
|
|
|
||
|
|
if proto_info.local_port == local_port
|
||
|
|
&& proto_info.remote_port == 12345
|
||
|
|
&& proto_info.local_addr == std::net::Ipv4Addr::LOCALHOST
|
||
|
|
&& proto_info.remote_addr == std::net::Ipv4Addr::LOCALHOST
|
||
|
|
{
|
||
|
|
println!("PIDs associated with socket: {:?}", &sock_info.associated_pids);
|
||
|
|
println!("Scanned {i} sockets");
|
||
|
|
return Ok(sock_info.associated_pids)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
Ok(vec![])
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
pub fn get_client_info(local_port: u16) -> Result<(), ClientInfoError> {
|
||
|
|
let mut sys = System::new();
|
||
|
|
for p in get_associated_pids(local_port)? {
|
||
|
|
let pid = Pid::from(p as usize);
|
||
|
|
sys.refresh_process(pid);
|
||
|
|
let proc = sys.process(pid)
|
||
|
|
.ok_or(ClientInfoError::PidNotFound)?;
|
||
|
|
let path = proc.exe().to_string_lossy();
|
||
|
|
println!("exe for requesting process: {path}");
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
}
|