about summary refs log tree commit diff
path: root/src/test/ui/command/command-create-pidfd.rs
diff options
context:
space:
mode:
authorbors <bors@rust-lang.org>2021-08-01 16:45:47 +0000
committerbors <bors@rust-lang.org>2021-08-01 16:45:47 +0000
commit4e21ef2a4eca12180e24a345d66066fc1e4e36da (patch)
treef4a6738e9cc0b047efed979e2f5a9d70b9b851b2 /src/test/ui/command/command-create-pidfd.rs
parent2e9c8705e94826da6aebe46512b4e3bbfc9e008f (diff)
parent4a832d32f232a68acdabfd29e526d2a4b6366a1c (diff)
downloadrust-4e21ef2a4eca12180e24a345d66066fc1e4e36da.tar.gz
rust-4e21ef2a4eca12180e24a345d66066fc1e4e36da.zip
Auto merge of #81825 - voidc:pidfd, r=joshtriplett
Add Linux-specific pidfd process extensions (take 2)

Continuation of #77168.
I addressed the following concerns from the original PR:

- make `CommandExt` and `ChildExt` sealed traits
- wrap file descriptors in `PidFd` struct representing ownership over the fd
- add `take_pidfd` to take the fd out of `Child`
- close fd when dropped

Tracking Issue: #82971
Diffstat (limited to 'src/test/ui/command/command-create-pidfd.rs')
-rw-r--r--src/test/ui/command/command-create-pidfd.rs45
1 files changed, 45 insertions, 0 deletions
diff --git a/src/test/ui/command/command-create-pidfd.rs b/src/test/ui/command/command-create-pidfd.rs
new file mode 100644
index 00000000000..93321ac536a
--- /dev/null
+++ b/src/test/ui/command/command-create-pidfd.rs
@@ -0,0 +1,45 @@
+// run-pass
+// only-linux - pidfds are a linux-specific concept
+
+#![feature(linux_pidfd)]
+#![feature(rustc_private)]
+
+extern crate libc;
+
+use std::io::Error;
+use std::os::linux::process::{ChildExt, CommandExt};
+use std::process::Command;
+
+fn has_clone3() -> bool {
+    let res = unsafe { libc::syscall(libc::SYS_clone3, 0, 0) };
+    let err = (res == -1)
+        .then(|| Error::last_os_error())
+        .expect("probe syscall should not succeed");
+    err.raw_os_error() != Some(libc::ENOSYS)
+}
+
+fn main() {
+    // pidfds require the clone3 syscall
+    if !has_clone3() {
+        return;
+    }
+
+    // We don't assert the precise value, since the standard library
+    // might have opened other file descriptors before our code runs.
+    let _ = Command::new("echo")
+        .create_pidfd(true)
+        .spawn()
+        .unwrap()
+        .pidfd().expect("failed to obtain pidfd");
+
+    let _ = Command::new("echo")
+        .create_pidfd(false)
+        .spawn()
+        .unwrap()
+        .pidfd().expect_err("pidfd should not have been created when create_pid(false) is set");
+
+    let _ = Command::new("echo")
+        .spawn()
+        .unwrap()
+        .pidfd().expect_err("pidfd should not have been created");
+}