Revert 9693 to prevent CPU hangs (#9893)

# Description

This reverts #9693 as it lead to CPU hangs. (btw, did the revert by hand
as it couldn't be done automatically. Hopefully I didn't miss anything 😅
)

Fixes #9859

cc @IanManske 

# User-Facing Changes
<!-- List of all changes that impact the user experience here. This
helps us keep track of breaking changes. -->

# Tests + Formatting
<!--
Don't forget to add tests that cover your changes.

Make sure you've run and fixed any issues with these commands:

- `cargo fmt --all -- --check` to check standard code formatting (`cargo
fmt --all` applies these changes)
- `cargo clippy --workspace -- -D warnings -D clippy::unwrap_used -A
clippy::needless_collect -A clippy::result_large_err` to check that
you're using the standard code style
- `cargo test --workspace` to check that all tests pass
- `cargo run -- -c "use std testing; testing run-tests --path
crates/nu-std"` to run the tests for the standard library

> **Note**
> from `nushell` you can also use the `toolkit` as follows
> ```bash
> use toolkit.nu # or use an `env_change` hook to activate it
automatically
> toolkit check pr
> ```
-->

# After Submitting
<!-- If your PR had any user-facing changes, update [the
documentation](https://github.com/nushell/nushell.github.io) after the
PR is merged, if necessary. This will help us keep the docs up to date.
-->
This commit is contained in:
JT 2023-08-02 11:24:28 +12:00 committed by GitHub
parent 778a00efa1
commit d25df9c00b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 47 additions and 68 deletions

View File

@ -213,7 +213,7 @@ impl ExternalCommand {
// fails to be run as a normal executable: // fails to be run as a normal executable:
// 1. "shell out" to cmd.exe if the command is a known cmd.exe internal command // 1. "shell out" to cmd.exe if the command is a known cmd.exe internal command
// 2. Otherwise, use `which-rs` to look for batch files etc. then run those in cmd.exe // 2. Otherwise, use `which-rs` to look for batch files etc. then run those in cmd.exe
match fg_process.spawn(engine_state.is_interactive) { match fg_process.spawn() {
Err(err) => { Err(err) => {
// set the default value, maybe we'll override it later // set the default value, maybe we'll override it later
child = Err(err); child = Err(err);
@ -235,7 +235,7 @@ impl ExternalCommand {
cmd, cmd,
engine_state.pipeline_externals_state.clone(), engine_state.pipeline_externals_state.clone(),
); );
child = cmd_process.spawn(engine_state.is_interactive); child = cmd_process.spawn();
} else { } else {
#[cfg(feature = "which-support")] #[cfg(feature = "which-support")]
{ {
@ -269,8 +269,7 @@ impl ExternalCommand {
cmd, cmd,
engine_state.pipeline_externals_state.clone(), engine_state.pipeline_externals_state.clone(),
); );
child = child = cmd_process.spawn();
cmd_process.spawn(engine_state.is_interactive);
} }
} }
} }
@ -287,7 +286,7 @@ impl ExternalCommand {
#[cfg(not(windows))] #[cfg(not(windows))]
{ {
child = fg_process.spawn(engine_state.is_interactive) child = fg_process.spawn()
} }
match child { match child {

View File

@ -28,7 +28,6 @@ pub struct ForegroundProcess {
pub struct ForegroundChild { pub struct ForegroundChild {
inner: Child, inner: Child,
pipeline_state: Arc<(AtomicU32, AtomicU32)>, pipeline_state: Arc<(AtomicU32, AtomicU32)>,
interactive: bool,
} }
impl ForegroundProcess { impl ForegroundProcess {
@ -39,14 +38,14 @@ impl ForegroundProcess {
} }
} }
pub fn spawn(&mut self, interactive: bool) -> std::io::Result<ForegroundChild> { pub fn spawn(&mut self) -> std::io::Result<ForegroundChild> {
let (ref pgrp, ref pcnt) = *self.pipeline_state; let (ref pgrp, ref pcnt) = *self.pipeline_state;
let existing_pgrp = pgrp.load(Ordering::SeqCst); let existing_pgrp = pgrp.load(Ordering::SeqCst);
fg_process_setup::prepare_to_foreground(&mut self.inner, existing_pgrp, interactive); fg_process_setup::prepare_to_foreground(&mut self.inner, existing_pgrp);
self.inner self.inner
.spawn() .spawn()
.map(|child| { .map(|child| {
fg_process_setup::set_foreground(&child, existing_pgrp, interactive); fg_process_setup::set_foreground(&child, existing_pgrp);
let _ = pcnt.fetch_add(1, Ordering::SeqCst); let _ = pcnt.fetch_add(1, Ordering::SeqCst);
if existing_pgrp == 0 { if existing_pgrp == 0 {
pgrp.store(child.id(), Ordering::SeqCst); pgrp.store(child.id(), Ordering::SeqCst);
@ -54,11 +53,10 @@ impl ForegroundProcess {
ForegroundChild { ForegroundChild {
inner: child, inner: child,
pipeline_state: self.pipeline_state.clone(), pipeline_state: self.pipeline_state.clone(),
interactive,
} }
}) })
.map_err(|e| { .map_err(|e| {
fg_process_setup::reset_foreground_id(interactive); fg_process_setup::reset_foreground_id();
e e
}) })
} }
@ -75,7 +73,7 @@ impl Drop for ForegroundChild {
let (ref pgrp, ref pcnt) = *self.pipeline_state; let (ref pgrp, ref pcnt) = *self.pipeline_state;
if pcnt.fetch_sub(1, Ordering::SeqCst) == 1 { if pcnt.fetch_sub(1, Ordering::SeqCst) == 1 {
pgrp.store(0, Ordering::SeqCst); pgrp.store(0, Ordering::SeqCst);
fg_process_setup::reset_foreground_id(self.interactive) fg_process_setup::reset_foreground_id()
} }
} }
} }
@ -103,10 +101,8 @@ mod fg_process_setup {
pub(super) fn prepare_to_foreground( pub(super) fn prepare_to_foreground(
external_command: &mut std::process::Command, external_command: &mut std::process::Command,
existing_pgrp: u32, existing_pgrp: u32,
interactive: bool,
) { ) {
let tty = TtyHandle(unistd::dup(nix::libc::STDIN_FILENO).expect("dup")); let tty = TtyHandle(unistd::dup(nix::libc::STDIN_FILENO).expect("dup"));
let interactive = interactive && std::io::stdin().is_terminal();
unsafe { unsafe {
// Safety: // Safety:
// POSIX only allows async-signal-safe functions to be called. // POSIX only allows async-signal-safe functions to be called.
@ -128,9 +124,7 @@ mod fg_process_setup {
// According to glibc's job control manual: // According to glibc's job control manual:
// https://www.gnu.org/software/libc/manual/html_node/Launching-Jobs.html // https://www.gnu.org/software/libc/manual/html_node/Launching-Jobs.html
// This has to be done *both* in the parent and here in the child due to race conditions. // This has to be done *both* in the parent and here in the child due to race conditions.
if interactive {
set_foreground_pid(unistd::getpid(), existing_pgrp, tty.0); set_foreground_pid(unistd::getpid(), existing_pgrp, tty.0);
}
// Now let the child process have all the signals by resetting with SIG_SETMASK. // Now let the child process have all the signals by resetting with SIG_SETMASK.
let mut sigset = signal::SigSet::empty(); let mut sigset = signal::SigSet::empty();
@ -143,13 +137,9 @@ mod fg_process_setup {
} }
} }
pub(super) fn set_foreground( pub(super) fn set_foreground(process: &std::process::Child, existing_pgrp: u32) {
process: &std::process::Child,
existing_pgrp: u32,
interactive: bool,
) {
// called from the parent shell process - do the stdin tty check here // called from the parent shell process - do the stdin tty check here
if interactive && std::io::stdin().is_terminal() { if std::io::stdin().is_terminal() {
set_foreground_pid( set_foreground_pid(
Pid::from_raw(process.id() as i32), Pid::from_raw(process.id() as i32),
existing_pgrp, existing_pgrp,
@ -173,8 +163,8 @@ mod fg_process_setup {
} }
/// Reset the foreground process group to the shell /// Reset the foreground process group to the shell
pub(super) fn reset_foreground_id(interactive: bool) { pub(super) fn reset_foreground_id() {
if interactive && std::io::stdin().is_terminal() { if std::io::stdin().is_terminal() {
if let Err(e) = nix::unistd::tcsetpgrp(nix::libc::STDIN_FILENO, unistd::getpgrp()) { if let Err(e) = nix::unistd::tcsetpgrp(nix::libc::STDIN_FILENO, unistd::getpgrp()) {
println!("ERROR: reset foreground id failed, tcsetpgrp result: {e:?}"); println!("ERROR: reset foreground id failed, tcsetpgrp result: {e:?}");
} }
@ -184,9 +174,9 @@ mod fg_process_setup {
#[cfg(any(not(target_family = "unix"), target_os = "macos"))] #[cfg(any(not(target_family = "unix"), target_os = "macos"))]
mod fg_process_setup { mod fg_process_setup {
pub(super) fn prepare_to_foreground(_: &mut std::process::Command, _: u32, _: bool) {} pub(super) fn prepare_to_foreground(_: &mut std::process::Command, _: u32) {}
pub(super) fn set_foreground(_: &std::process::Child, _: u32, _: bool) {} pub(super) fn set_foreground(_: &std::process::Child, _: u32) {}
pub(super) fn reset_foreground_id(_: bool) {} pub(super) fn reset_foreground_id() {}
} }

View File

@ -82,10 +82,7 @@ fn main() -> Result<()> {
.unwrap_or_else(|_| std::process::exit(1)); .unwrap_or_else(|_| std::process::exit(1));
// keep this condition in sync with the branches at the end // keep this condition in sync with the branches at the end
engine_state.is_interactive = parsed_nu_cli_args.interactive_shell.is_some() engine_state.is_interactive = parsed_nu_cli_args.interactive_shell.is_some();
|| (parsed_nu_cli_args.testbin.is_none()
&& parsed_nu_cli_args.commands.is_none()
&& script_name.is_empty());
engine_state.is_login = parsed_nu_cli_args.login_shell.is_some(); engine_state.is_login = parsed_nu_cli_args.login_shell.is_some();
@ -147,7 +144,7 @@ fn main() -> Result<()> {
); );
start_time = std::time::Instant::now(); start_time = std::time::Instant::now();
acquire_terminal(engine_state.is_interactive); acquire_terminal(parsed_nu_cli_args.commands.is_none() && script_name.is_empty());
perf( perf(
"acquire_terminal", "acquire_terminal",
start_time, start_time,
@ -293,6 +290,7 @@ fn main() -> Result<()> {
input, input,
) )
} else { } else {
engine_state.is_interactive = true;
run_repl(&mut engine_state, parsed_nu_cli_args, entire_start_time) run_repl(&mut engine_state, parsed_nu_cli_args, entire_start_time)
} }
} }

View File

@ -1,16 +1,13 @@
#[cfg(unix)] #[cfg(unix)]
pub(crate) fn acquire_terminal(interactive: bool) { pub(crate) fn acquire_terminal(interactive: bool) {
use is_terminal::IsTerminal; use is_terminal::IsTerminal;
use nix::{ use nix::sys::signal::{signal, SigHandler, Signal};
errno::Errno,
sys::signal::{signal, SigHandler, Signal},
unistd,
};
if interactive && std::io::stdin().is_terminal() { if !std::io::stdin().is_terminal() {
// see also: https://www.gnu.org/software/libc/manual/html_node/Initializing-the-Shell.html return;
}
take_control(); take_control(interactive);
unsafe { unsafe {
// SIGINT and SIGQUIT have special handling above // SIGINT and SIGQUIT have special handling above
@ -18,22 +15,6 @@ pub(crate) fn acquire_terminal(interactive: bool) {
signal(Signal::SIGTTIN, SigHandler::SigIgn).expect("signal ignore"); signal(Signal::SIGTTIN, SigHandler::SigIgn).expect("signal ignore");
signal(Signal::SIGTTOU, SigHandler::SigIgn).expect("signal ignore"); signal(Signal::SIGTTOU, SigHandler::SigIgn).expect("signal ignore");
} }
// Put ourselves in our own process group and take control of terminal, if not already
let shell_pgid = unistd::getpid();
match unistd::setpgid(shell_pgid, shell_pgid) {
// setpgid returns EPERM if we are the session leader (e.g., as a login shell).
// The other cases that return EPERM cannot happen, since we gave our own pid.
// See: setpgid(2)
// Therefore, it is safe to ignore EPERM.
Ok(()) | Err(Errno::EPERM) => (),
Err(_) => {
eprintln!("ERROR: failed to put nushell in its own process group");
std::process::exit(1);
}
}
let _ = unistd::tcsetpgrp(nix::libc::STDIN_FILENO, shell_pgid);
}
} }
#[cfg(not(unix))] #[cfg(not(unix))]
@ -41,7 +22,7 @@ pub(crate) fn acquire_terminal(_: bool) {}
// Inspired by fish's acquire_tty_or_exit // Inspired by fish's acquire_tty_or_exit
#[cfg(unix)] #[cfg(unix)]
fn take_control() { fn take_control(interactive: bool) {
use nix::{ use nix::{
errno::Errno, errno::Errno,
sys::signal::{self, SaFlags, SigAction, SigHandler, SigSet, Signal}, sys::signal::{self, SaFlags, SigAction, SigHandler, SigSet, Signal},
@ -78,23 +59,32 @@ fn take_control() {
} }
} }
let mut success = false;
for _ in 0..4096 { for _ in 0..4096 {
match unistd::tcgetpgrp(nix::libc::STDIN_FILENO) { match unistd::tcgetpgrp(nix::libc::STDIN_FILENO) {
Ok(owner_pgid) if owner_pgid == shell_pgid => { Ok(owner_pgid) if owner_pgid == shell_pgid => {
// success success = true;
return; break;
} }
Ok(owner_pgid) if owner_pgid == Pid::from_raw(0) => { Ok(owner_pgid) if owner_pgid == Pid::from_raw(0) => {
// Zero basically means something like "not owned" and we can just take it // Zero basically means something like "not owned" and we can just take it
let _ = unistd::tcsetpgrp(nix::libc::STDIN_FILENO, shell_pgid); let _ = unistd::tcsetpgrp(nix::libc::STDIN_FILENO, shell_pgid);
} }
Err(Errno::ENOTTY) => { Err(Errno::ENOTTY) => {
if !interactive {
// that's fine
return;
}
eprintln!("ERROR: no TTY for interactive shell"); eprintln!("ERROR: no TTY for interactive shell");
std::process::exit(1); std::process::exit(1);
} }
_ => { _ => {
// fish also has other heuristics than "too many attempts" for the orphan check, but they're optional // fish also has other heuristics than "too many attempts" for the orphan check, but they're optional
if signal::killpg(shell_pgid, Signal::SIGTTIN).is_err() { if signal::killpg(Pid::from_raw(-shell_pgid.as_raw()), Signal::SIGTTIN).is_err() {
if !interactive {
// that's fine
return;
}
eprintln!("ERROR: failed to SIGTTIN ourselves"); eprintln!("ERROR: failed to SIGTTIN ourselves");
std::process::exit(1); std::process::exit(1);
} }
@ -102,6 +92,8 @@ fn take_control() {
} }
} }
if !success && interactive {
eprintln!("ERROR: failed take control of the terminal, we might be orphaned"); eprintln!("ERROR: failed take control of the terminal, we might be orphaned");
std::process::exit(1); std::process::exit(1);
}
} }