44 lines
1.1 KiB
Rust
44 lines
1.1 KiB
Rust
use {
|
|
crate::utils::{errorfmt::ErrorFmt, oserror::OsError, trim::AsciiTrim},
|
|
bstr::ByteSlice,
|
|
uapi::{c, OwnedFd},
|
|
};
|
|
|
|
pub struct PidInfo {
|
|
pub _uid: c::uid_t,
|
|
pub pid: c::pid_t,
|
|
pub comm: String,
|
|
}
|
|
|
|
pub fn get_pid_info(uid: c::uid_t, pid: c::pid_t) -> PidInfo {
|
|
let comm = match std::fs::read(format!("/proc/{}/comm", pid)) {
|
|
Ok(name) => name.trim().as_bstr().to_string(),
|
|
Err(e) => {
|
|
log::warn!("Could not read `comm` of pid {}: {}", pid, ErrorFmt(e));
|
|
"Unknown".to_string()
|
|
}
|
|
};
|
|
PidInfo {
|
|
_uid: uid,
|
|
pid,
|
|
comm,
|
|
}
|
|
}
|
|
|
|
pub fn get_socket_creds(socket: &OwnedFd) -> Option<(c::uid_t, c::pid_t)> {
|
|
let mut cred = c::ucred {
|
|
pid: 0,
|
|
uid: 0,
|
|
gid: 0,
|
|
};
|
|
match uapi::getsockopt(socket.raw(), c::SOL_SOCKET, c::SO_PEERCRED, &mut cred) {
|
|
Ok(_) => Some((cred.uid, cred.pid)),
|
|
Err(e) => {
|
|
log::error!(
|
|
"Cannot determine peer credentials of new connection: {:?}",
|
|
OsError::from(e)
|
|
);
|
|
None
|
|
}
|
|
}
|
|
}
|