From b505ddf629e988f2e60a8383f373a9bf9e68a3f9 Mon Sep 17 00:00:00 2001 From: Finn Behrens Date: Thu, 25 Jun 2020 20:40:19 +0200 Subject: [PATCH] add pid to tokio::net::unix::UCred --- tokio/src/net/unix/ucred.rs | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/tokio/src/net/unix/ucred.rs b/tokio/src/net/unix/ucred.rs index ef214a702fa..97021452518 100644 --- a/tokio/src/net/unix/ucred.rs +++ b/tokio/src/net/unix/ucred.rs @@ -1,8 +1,10 @@ -use libc::{gid_t, uid_t}; +use libc::{pid_t, gid_t, uid_t}; /// Credentials of a process #[derive(Copy, Clone, Eq, PartialEq, Hash, Debug)] pub struct UCred { + /// PID (process ID) of the process + pid: Option, // TODO: how to get it under macos? /// UID (user ID) of the process uid: uid_t, /// GID (group ID) of the process @@ -19,6 +21,11 @@ impl UCred { pub fn gid(&self) -> gid_t { self.gid } + + /// Gets PID (process ID) of the process. + pub fn pid(&self) -> Option { + self.pid + } } #[cfg(any(target_os = "linux", target_os = "android"))] @@ -75,6 +82,7 @@ pub(crate) mod impl_linux { ); if ret == 0 && ucred_size as usize == mem::size_of::() { Ok(super::UCred { + pid: Some(ucred.pid), uid: ucred.uid, gid: ucred.gid, }) @@ -112,6 +120,7 @@ pub(crate) mod impl_macos { if ret == 0 { Ok(super::UCred { + pid: None, uid: uid.assume_init(), gid: gid.assume_init(), }) @@ -154,7 +163,7 @@ pub(crate) mod impl_solaris { ucred_free(cred); - Ok(super::UCred { uid, gid }) + Ok(super::UCred { pid: None, uid, gid }) } else { Err(io::Error::last_os_error()) }