pid_open

Author: JonathanWoollett-Light

CHANGELOG.md

@@ -11,6 +11,8 @@ This project adheres to [Semantic Versioning](https://semver.org/).
   (#[1662](https://github.com/nix-rust/nix/pull/1662))
 - Added `pidfd_getfd` on Linux.
   ([#1868](https://github.com/nix-rust/nix/pull/1868))
+- Added `pid_open` on Linux.
+  ([#1868](https://github.com/nix-rust/nix/pull/1868))
 
 ### Changed
 

src/sys/mod.rs

@@ -227,6 +227,6 @@ feature! {
     pub mod timer;
 }
 
-#[cfg(target_os = "linux")]
+#[cfg(all(target_os = "linux", feature = "process"))]
 /// pidfd related functionality
 pub mod pidfd;

src/sys/pidfd.rs

@@ -1,4 +1,5 @@
 use crate::errno::Errno;
+use crate::unistd::Pid;
 use crate::Result;
 use std::convert::TryFrom;
 use std::os::unix::io::RawFd;
@@ -29,3 +30,33 @@ pub fn pidfd_getfd(pidfd: RawFd, targetfd: RawFd) -> Result<RawFd> {
         _ => unreachable!(),
     }
 }
+
+/// Creates a file descriptor that refers to the process whose PID is specified in `pid`.  The file
+/// descriptor is returned as the function result; the close-on-exec flag is set on the file
+/// descriptor.
+///
+/// If `pidfd_nonblock == true` returns a nonblocking file descriptor.  If the process
+/// referred to by the file descriptor has not yet terminated,
+/// then an attempt to wait on the file descriptor using
+/// waitid(2) will immediately return the error EAGAIN rather
+/// than blocking.
+pub fn pid_open(pid_t: Pid, pidfd_nonblock: bool) -> Result<RawFd> {
+    #[allow(clippy::useless_conversion)] // Not useless on all OSes
+    match unsafe {
+        libc::syscall(
+            libc::SYS_pidfd_open,
+            pid_t,
+            if pidfd_nonblock {
+                // MUSL is missing the `PIDFD_NONBLOCK` alias, so the underlying `O_NONBLOCK` needs
+                // to be used for compatibility.
+                libc::PIDFD_NONBLOCK
+            } else {
+                0
+            },
+        )
+    } {
+        -1 => Err(Errno::last()),
+        fd @ 0.. => Ok(RawFd::try_from(fd).unwrap()),
+        _ => unreachable!(),
+    }
+}