forked from extern/nushell
1d18f6947e
* Revert "Fix intermittent test crash (#6268)"
This reverts commit 555d9ee763.
* make a working version again
* try second impl
* add
* fmt
* check stdin is atty before acquire stdin
* add libc
* clean comment
* fix typo
42 lines
1.5 KiB
Rust
42 lines
1.5 KiB
Rust
use nix::sys::signal::{self, SigHandler, Signal};
|
|
|
|
/// Blocks the SIGTSTP/SIGTTOU/SIGTTIN/SIGCHLD signals so that the shell never receives
|
|
/// them.
|
|
pub fn block() {
|
|
let mut sigset = signal::SigSet::empty();
|
|
sigset.add(signal::Signal::SIGTSTP);
|
|
sigset.add(signal::Signal::SIGTTOU);
|
|
sigset.add(signal::Signal::SIGTTIN);
|
|
sigset.add(signal::Signal::SIGCHLD);
|
|
if let Err(e) = signal::sigprocmask(signal::SigmaskHow::SIG_BLOCK, Some(&sigset), None) {
|
|
println!("ERROR: Could not block the signals, error message: {e:?}");
|
|
}
|
|
}
|
|
|
|
/// Unblocks the SIGTSTP/SIGTTOU/SIGTTIN/SIGCHLD signals so children processes can be
|
|
/// controlled
|
|
/// by the shell.
|
|
pub fn unblock() {
|
|
let mut sigset = signal::SigSet::empty();
|
|
sigset.add(signal::Signal::SIGTSTP);
|
|
sigset.add(signal::Signal::SIGTTOU);
|
|
sigset.add(signal::Signal::SIGTTIN);
|
|
sigset.add(signal::Signal::SIGCHLD);
|
|
if let Err(e) = signal::sigprocmask(signal::SigmaskHow::SIG_UNBLOCK, Some(&sigset), None) {
|
|
println!("ERROR: Could not unblock the signals, error message: {e:?}");
|
|
}
|
|
}
|
|
|
|
// It's referenced from `set_unique_pid` function in `ion`.
|
|
pub fn set_terminal_leader() {
|
|
let stdin_is_a_tty = atty::is(atty::Stream::Stdin);
|
|
if stdin_is_a_tty {
|
|
// We have make sure that stdin is a tty, it's ok to ignore SIGTTOU.
|
|
unsafe {
|
|
if let Err(e) = signal::signal(Signal::SIGTTOU, SigHandler::SigIgn) {
|
|
println!("WARN: ignore SIGTTOU failed, error message: {e:?}");
|
|
}
|
|
}
|
|
}
|
|
}
|