1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

use std::{io::Result, os::unix::process::CommandExt, process::Command};

use nix::sys::signal::{sigprocmask, SigSet, SigmaskHow};
use tracing::trace;

use super::{StdCommandWrap, StdCommandWrapper};

/// Wrapper which resets the process signal mask.
///
/// By default a Command on Unix inherits its parent's [signal mask]. However, in some cases this
/// is not what you want. This wrapper resets the command's sigmask by unblocking all signals.
#[derive(Clone, Copy, Debug)]
pub struct ResetSigmask;

impl StdCommandWrapper for ResetSigmask {
	fn pre_spawn(&mut self, command: &mut Command, _core: &StdCommandWrap) -> Result<()> {
		unsafe {
			command.pre_exec(|| {
				let mut oldset = SigSet::empty();
				let newset = SigSet::all();
				trace!(unblocking=?newset, "resetting process sigmask");
				sigprocmask(SigmaskHow::SIG_UNBLOCK, Some(&newset), Some(&mut oldset))?;
				trace!(?oldset, "sigmask reset");
				Ok(())
			});
		}
		Ok(())
	}
}