1 | //! Interface for the `signalfd` syscall. |
2 | //! |
3 | //! # Signal discarding |
4 | //! When a signal can't be delivered to a process (or thread), it will become a pending signal. |
5 | //! Failure to deliver could happen if the signal is blocked by every thread in the process or if |
6 | //! the signal handler is still handling a previous signal. |
7 | //! |
8 | //! If a signal is sent to a process (or thread) that already has a pending signal of the same |
9 | //! type, it will be discarded. This means that if signals of the same type are received faster than |
10 | //! they are processed, some of those signals will be dropped. Because of this limitation, |
11 | //! `signalfd` in itself cannot be used for reliable communication between processes or threads. |
12 | //! |
13 | //! Once the signal is unblocked, or the signal handler is finished, and a signal is still pending |
14 | //! (ie. not consumed from a signalfd) it will be delivered to the signal handler. |
15 | //! |
16 | //! Please note that signal discarding is not specific to `signalfd`, but also happens with regular |
17 | //! signal handlers. |
18 | use crate::errno::Errno; |
19 | pub use crate::sys::signal::{self, SigSet}; |
20 | use crate::Result; |
21 | pub use libc::signalfd_siginfo as siginfo; |
22 | |
23 | use std::mem; |
24 | use std::os::unix::io::{AsFd, AsRawFd, BorrowedFd, FromRawFd, OwnedFd, RawFd}; |
25 | |
26 | libc_bitflags! { |
27 | pub struct SfdFlags: libc::c_int { |
28 | SFD_NONBLOCK; |
29 | SFD_CLOEXEC; |
30 | } |
31 | } |
32 | |
33 | #[deprecated (since = "0.23.0" , note = "use mem::size_of::<siginfo>() instead" )] |
34 | pub const SIGNALFD_SIGINFO_SIZE: usize = mem::size_of::<siginfo>(); |
35 | |
36 | /// Creates a new file descriptor for reading signals. |
37 | /// |
38 | /// **Important:** please read the module level documentation about signal discarding before using |
39 | /// this function! |
40 | /// |
41 | /// The `mask` parameter specifies the set of signals that can be accepted via this file descriptor. |
42 | /// |
43 | /// A signal must be blocked on every thread in a process, otherwise it won't be visible from |
44 | /// signalfd (the default handler will be invoked instead). |
45 | /// |
46 | /// See [the signalfd man page for more information](https://man7.org/linux/man-pages/man2/signalfd.2.html) |
47 | #[deprecated (since = "0.27.0" , note = "Use SignalFd instead" )] |
48 | pub fn signalfd<F: AsFd>( |
49 | fd: Option<F>, |
50 | mask: &SigSet, |
51 | flags: SfdFlags, |
52 | ) -> Result<OwnedFd> { |
53 | _signalfd(fd, mask, flags) |
54 | } |
55 | |
56 | fn _signalfd<F: AsFd>( |
57 | fd: Option<F>, |
58 | mask: &SigSet, |
59 | flags: SfdFlags, |
60 | ) -> Result<OwnedFd> { |
61 | let raw_fd: i32 = fd.map_or(default:-1, |x: F| x.as_fd().as_raw_fd()); |
62 | unsafe { |
63 | Errno::result(libc::signalfd(raw_fd, mask.as_ref(), flags.bits())) |
64 | .map(|raw_fd: i32| FromRawFd::from_raw_fd(raw_fd)) |
65 | } |
66 | } |
67 | |
68 | /// A helper struct for creating, reading and closing a `signalfd` instance. |
69 | /// |
70 | /// **Important:** please read the module level documentation about signal discarding before using |
71 | /// this struct! |
72 | /// |
73 | /// # Examples |
74 | /// |
75 | /// ``` |
76 | /// # use nix::sys::signalfd::*; |
77 | /// // Set the thread to block the SIGUSR1 signal, otherwise the default handler will be used |
78 | /// let mut mask = SigSet::empty(); |
79 | /// mask.add(signal::SIGUSR1); |
80 | /// mask.thread_block().unwrap(); |
81 | /// |
82 | /// // Signals are queued up on the file descriptor |
83 | /// let mut sfd = SignalFd::with_flags(&mask, SfdFlags::SFD_NONBLOCK).unwrap(); |
84 | /// |
85 | /// match sfd.read_signal() { |
86 | /// // we caught a signal |
87 | /// Ok(Some(sig)) => (), |
88 | /// // there were no signals waiting (only happens when the SFD_NONBLOCK flag is set, |
89 | /// // otherwise the read_signal call blocks) |
90 | /// Ok(None) => (), |
91 | /// Err(err) => (), // some error happend |
92 | /// } |
93 | /// ``` |
94 | #[derive (Debug)] |
95 | pub struct SignalFd(OwnedFd); |
96 | |
97 | impl SignalFd { |
98 | pub fn new(mask: &SigSet) -> Result<SignalFd> { |
99 | Self::with_flags(mask, SfdFlags::empty()) |
100 | } |
101 | |
102 | pub fn with_flags(mask: &SigSet, flags: SfdFlags) -> Result<SignalFd> { |
103 | let fd = _signalfd(None::<OwnedFd>, mask, flags)?; |
104 | |
105 | Ok(SignalFd(fd)) |
106 | } |
107 | |
108 | pub fn set_mask(&self, mask: &SigSet) -> Result<()> { |
109 | self.update(mask, SfdFlags::empty()) |
110 | } |
111 | |
112 | pub fn read_signal(&self) -> Result<Option<siginfo>> { |
113 | let mut buffer = mem::MaybeUninit::<siginfo>::uninit(); |
114 | |
115 | let size = mem::size_of_val(&buffer); |
116 | let res = Errno::result(unsafe { |
117 | libc::read(self.0.as_raw_fd(), buffer.as_mut_ptr().cast(), size) |
118 | }) |
119 | .map(|r| r as usize); |
120 | match res { |
121 | Ok(x) if x == size => Ok(Some(unsafe { buffer.assume_init() })), |
122 | Ok(_) => unreachable!("partial read on signalfd" ), |
123 | Err(Errno::EAGAIN) => Ok(None), |
124 | Err(error) => Err(error), |
125 | } |
126 | } |
127 | |
128 | fn update(&self, mask: &SigSet, flags: SfdFlags) -> Result<()> { |
129 | let raw_fd = self.0.as_raw_fd(); |
130 | unsafe { |
131 | Errno::result(libc::signalfd(raw_fd, mask.as_ref(), flags.bits())) |
132 | .map(drop) |
133 | } |
134 | } |
135 | } |
136 | |
137 | impl AsFd for SignalFd { |
138 | fn as_fd(&self) -> BorrowedFd { |
139 | self.0.as_fd() |
140 | } |
141 | } |
142 | impl AsRawFd for SignalFd { |
143 | fn as_raw_fd(&self) -> RawFd { |
144 | self.0.as_raw_fd() |
145 | } |
146 | } |
147 | |
148 | impl Iterator for SignalFd { |
149 | type Item = siginfo; |
150 | |
151 | fn next(&mut self) -> Option<Self::Item> { |
152 | match self.read_signal() { |
153 | Ok(Some(sig: signalfd_siginfo)) => Some(sig), |
154 | Ok(None) | Err(_) => None, |
155 | } |
156 | } |
157 | } |
158 | |