Skip to content

Commit

Permalink
process: make Child::kill async (#2823)
Browse files Browse the repository at this point in the history
* This changes the `Child::kill` to be an async method which awaits the
  child after sending a kill signal. This avoids leaving zombie
  processes on Unix platforms if the caller forgets to await the child
  after the kill completes
* A `start_kill` method was also added on `Child` which only sends the
  kill signal to the child process. This allows for kill signals to be
  sent even outside of async contexts.
  • Loading branch information
ipetkov committed Sep 8, 2020
1 parent f4d6ed0 commit 7c254ec
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 15 deletions.
36 changes: 23 additions & 13 deletions tokio/src/process/mod.rs
Expand Up @@ -774,6 +774,23 @@ impl Child {
}
}

/// Attempts to force the child to exit, but does not wait for the request
/// to take effect.
///
/// On Unix platforms, this is the equivalent to sending a SIGKILL. Note
/// that on Unix platforms it is possible for a zombie process to remain
/// after a kill is sent; to avoid this, the caller should ensure that either
/// `child.wait().await` or `child.try_wait()` is invoked successfully.
pub fn start_kill(&mut self) -> io::Result<()> {
match &mut self.child {
FusedChild::Child(child) => child.kill(),
FusedChild::Done(_) => Err(io::Error::new(
io::ErrorKind::InvalidInput,
"invalid argument: can't kill an exited process",
)),
}
}

/// Forces the child to exit.
///
/// This is equivalent to sending a SIGKILL on unix platforms.
Expand All @@ -795,21 +812,14 @@ impl Child {
/// tokio::spawn(async move { send.send(()) });
/// tokio::select! {
/// _ = child.wait() => {}
/// _ = recv => {
/// child.kill().expect("kill failed");
/// // NB: await the child here to avoid a zombie process on Unix platforms
/// child.wait().await.unwrap();
/// }
/// _ = recv => child.kill().await.expect("kill failed"),
/// }
/// }
pub fn kill(&mut self) -> io::Result<()> {
match &mut self.child {
FusedChild::Child(child) => child.kill(),
FusedChild::Done(_) => Err(io::Error::new(
io::ErrorKind::InvalidInput,
"invalid argument: can't kill an exited process",
)),
}
/// ```
pub async fn kill(&mut self) -> io::Result<()> {
self.start_kill()?;
self.wait().await?;
Ok(())
}

/// Waits for the child to exit completely, returning the status that it
Expand Down
3 changes: 1 addition & 2 deletions tokio/tests/process_issue_2174.rs
Expand Up @@ -39,8 +39,7 @@ async fn issue_2174() {
time::delay_for(Duration::from_secs(1)).await;

// Kill the child process.
child.kill().unwrap();
let _ = child.wait().await;
child.kill().await.unwrap();

assert_err!(handle.await);
}

0 comments on commit 7c254ec

Please sign in to comment.