add pid to tokio::net::unix::UCred

tokio-rs · Oct 11, 2020 · b505ddf · b505ddf
1 parent b047f64
commit b505ddf
Showing 1 changed file with 11 additions and 2 deletions.
diff --git 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<pid_t>, // 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<pid_t> {
+        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::<ucred>() {
                 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())
             }