1use crate::io;
2use crate::os::fd::{AsRawFd, FromRawFd, RawFd};
3use crate::sys::cvt;
4use crate::sys::fd::FileDesc;
5use crate::sys::process::ExitStatus;
6use crate::sys_common::{AsInner, FromInner, IntoInner};
7
8#[cfg(test)]
9mod tests;
10
11#[derive(Debug)]
12pub(crate) struct PidFd(FileDesc);
13
14impl PidFd {
15 pub fn kill(&self) -> io::Result<()> {
16 cvt(unsafe {
17 libc::syscall(
18 libc::SYS_pidfd_send_signal,
19 self.0.as_raw_fd(),
20 libc::SIGKILL,
21 crate::ptr::null::<()>(),
22 0,
23 )
24 })
25 .map(drop)
26 }
27
28 pub fn wait(&self) -> io::Result<ExitStatus> {
29 let mut siginfo: libc::siginfo_t = unsafe { crate::mem::zeroed() };
30 cvt(unsafe {
31 libc::waitid(libc::P_PIDFD, self.0.as_raw_fd() as u32, &mut siginfo, libc::WEXITED)
32 })?;
33 Ok(ExitStatus::from_waitid_siginfo(siginfo))
34 }
35
36 pub fn try_wait(&self) -> io::Result<Option<ExitStatus>> {
37 let mut siginfo: libc::siginfo_t = unsafe { crate::mem::zeroed() };
38
39 cvt(unsafe {
40 libc::waitid(
41 libc::P_PIDFD,
42 self.0.as_raw_fd() as u32,
43 &mut siginfo,
44 libc::WEXITED | libc::WNOHANG,
45 )
46 })?;
47 if unsafe { siginfo.si_pid() } == 0 {
48 Ok(None)
49 } else {
50 Ok(Some(ExitStatus::from_waitid_siginfo(siginfo)))
51 }
52 }
53}
54
55impl AsInner<FileDesc> for PidFd {
56 fn as_inner(&self) -> &FileDesc {
57 &self.0
58 }
59}
60
61impl IntoInner<FileDesc> for PidFd {
62 fn into_inner(self) -> FileDesc {
63 self.0
64 }
65}
66
67impl FromInner<FileDesc> for PidFd {
68 fn from_inner(inner: FileDesc) -> Self {
69 Self(inner)
70 }
71}
72
73impl FromRawFd for PidFd {
74 unsafe fn from_raw_fd(fd: RawFd) -> Self {
75 Self(FileDesc::from_raw_fd(fd))
76 }
77}
78