Make Process::wait return Option<ExitStatus>

GuillaumeGomez · Dec 20, 2024 · ee6ea3d · ee6ea3d
1 parent 10e2d8c
commit ee6ea3d
Show file tree

Hide file tree

Showing 7 changed files with 43 additions and 16 deletions.
diff --git a/src/common/system.rs b/src/common/system.rs
@@ -4,6 +4,7 @@ use std::collections::{HashMap, HashSet};
 use std::ffi::{OsStr, OsString};
 use std::fmt;
 use std::path::Path;
+use std::process::ExitStatus;
 use std::str::FromStr;
 
 use crate::common::impl_get_set::impl_get_set;
@@ -1188,7 +1189,8 @@ impl Process {
         self.inner.kill_with(signal)
     }
 
-    /// Wait for process termination.
+    /// Wait for process termination and returns its [`ExitStatus`] if it could be retrieved,
+    /// returns `None` otherwise.
     ///
     /// ```no_run
     /// use sysinfo::{Pid, System};
@@ -1197,11 +1199,11 @@ impl Process {
     ///
     /// if let Some(process) = s.process(Pid::from(1337)) {
     ///     println!("Waiting for pid 1337");
-    ///     process.wait();
-    ///     println!("Pid 1337 exited");
+    ///     let exit_status = process.wait();
+    ///     println!("Pid 1337 exited with: {exit_status:?}");
     /// }
     /// ```
-    pub fn wait(&self) {
+    pub fn wait(&self) -> Option<ExitStatus> {
         self.inner.wait()
     }
 

diff --git a/src/unix/apple/app_store/process.rs b/src/unix/apple/app_store/process.rs
@@ -2,6 +2,7 @@
 
 use std::ffi::{OsStr, OsString};
 use std::path::Path;
+use std::process::ExitStatus;
 
 use crate::{DiskUsage, Gid, Pid, ProcessStatus, Signal, Uid};
 
@@ -88,7 +89,9 @@ impl ProcessInner {
         None
     }
 
-    pub(crate) fn wait(&self) {}
+    pub(crate) fn wait(&self) -> Option<ExitStatus> {
+        None
+    }
 
     pub(crate) fn session_id(&self) -> Option<Pid> {
         None

diff --git a/src/unix/apple/macos/process.rs b/src/unix/apple/macos/process.rs
@@ -3,7 +3,9 @@
 use std::ffi::{OsStr, OsString};
 use std::mem::{self, MaybeUninit};
 use std::os::unix::ffi::{OsStrExt, OsStringExt};
+use std::os::unix::process::ExitStatusExt;
 use std::path::{Path, PathBuf};
+use std::process::ExitStatus;
 
 use libc::{c_int, c_void, kill};
 
@@ -202,7 +204,7 @@ impl ProcessInner {
         self.effective_group_id
     }
 
-    pub(crate) fn wait(&self) {
+    pub(crate) fn wait(&self) -> Option<ExitStatus> {
         let mut status = 0;
         // attempt waiting
         unsafe {
@@ -213,6 +215,7 @@ impl ProcessInner {
                     std::thread::sleep(duration);
                 }
             }
+            Some(ExitStatus::from_raw(status))
         }
     }
 

diff --git a/src/unix/freebsd/process.rs b/src/unix/freebsd/process.rs
@@ -4,7 +4,9 @@ use crate::{DiskUsage, Gid, Pid, Process, ProcessRefreshKind, ProcessStatus, Sig
 
 use std::ffi::{OsStr, OsString};
 use std::fmt;
+use std::os::unix::process::ExitStatusExt;
 use std::path::{Path, PathBuf};
+use std::process::ExitStatus;
 
 use libc::kill;
 
@@ -154,7 +156,7 @@ impl ProcessInner {
         Some(self.effective_group_id)
     }
 
-    pub(crate) fn wait(&self) {
+    pub(crate) fn wait(&self) -> Option<ExitStatus> {
         let mut status = 0;
         // attempt waiting
         unsafe {
@@ -165,6 +167,7 @@ impl ProcessInner {
                     std::thread::sleep(duration);
                 }
             }
+            Some(ExitStatus::from_raw(status))
         }
     }
 

diff --git a/src/unix/linux/process.rs b/src/unix/linux/process.rs
@@ -7,7 +7,9 @@ use std::fmt;
 use std::fs::{self, DirEntry, File};
 use std::io::Read;
 use std::os::unix::ffi::OsStrExt;
+use std::os::unix::process::ExitStatusExt;
 use std::path::{Path, PathBuf};
+use std::process::ExitStatus;
 use std::str::{self, FromStr};
 use std::sync::atomic::{AtomicUsize, Ordering};
 
@@ -251,7 +253,7 @@ impl ProcessInner {
         self.effective_group_id
     }
 
-    pub(crate) fn wait(&self) {
+    pub(crate) fn wait(&self) -> Option<ExitStatus> {
         let mut status = 0;
         // attempt waiting
         unsafe {
@@ -262,6 +264,7 @@ impl ProcessInner {
                     std::thread::sleep(duration);
                 }
             }
+            Some(ExitStatus::from_raw(status))
         }
     }
 

diff --git a/src/unknown/process.rs b/src/unknown/process.rs
@@ -5,6 +5,7 @@ use crate::{DiskUsage, Gid, Pid, ProcessStatus, Signal, Uid};
 use std::ffi::{OsStr, OsString};
 use std::fmt;
 use std::path::Path;
+use std::process::ExitStatus;
 
 impl fmt::Display for ProcessStatus {
     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
@@ -98,7 +99,9 @@ impl ProcessInner {
         None
     }
 
-    pub(crate) fn wait(&self) {}
+    pub(crate) fn wait(&self) -> Option<ExitStatus> {
+        None
+    }
 
     pub(crate) fn session_id(&self) -> Option<Pid> {
         None

diff --git a/src/windows/process.rs b/src/windows/process.rs
@@ -11,9 +11,9 @@ use std::fmt;
 use std::io;
 use std::mem::{size_of, zeroed, MaybeUninit};
 use std::os::windows::ffi::OsStringExt;
-use std::os::windows::process::CommandExt;
+use std::os::windows::process::{CommandExt, ExitStatusExt};
 use std::path::{Path, PathBuf};
-use std::process;
+use std::process::{self, ExitStatus};
 use std::ptr::null_mut;
 use std::str;
 use std::sync::{Arc, OnceLock};
@@ -41,9 +41,9 @@ use windows::Win32::System::ProcessStatus::GetModuleFileNameExW;
 use windows::Win32::System::RemoteDesktop::ProcessIdToSessionId;
 use windows::Win32::System::SystemInformation::OSVERSIONINFOEXW;
 use windows::Win32::System::Threading::{
-    GetProcessIoCounters, GetProcessTimes, GetSystemTimes, OpenProcess, OpenProcessToken,
-    CREATE_NO_WINDOW, IO_COUNTERS, PEB, PROCESS_BASIC_INFORMATION, PROCESS_QUERY_INFORMATION,
-    PROCESS_QUERY_LIMITED_INFORMATION, PROCESS_VM_READ,
+    GetExitCodeProcess, GetProcessIoCounters, GetProcessTimes, GetSystemTimes, OpenProcess,
+    OpenProcessToken, CREATE_NO_WINDOW, IO_COUNTERS, PEB, PROCESS_BASIC_INFORMATION,
+    PROCESS_QUERY_INFORMATION, PROCESS_QUERY_LIMITED_INFORMATION, PROCESS_VM_READ,
 };
 use windows::Win32::UI::Shell::CommandLineToArgvW;
 
@@ -404,18 +404,28 @@ impl ProcessInner {
         None
     }
 
-    pub(crate) fn wait(&self) {
+    pub(crate) fn wait(&self) -> Option<ExitStatus> {
         if let Some(handle) = self.get_handle() {
             while is_proc_running(handle) {
                 if get_start_time(handle) != self.start_time() {
                     // PID owner changed so the previous process was finished!
-                    return;
+                    sysinfo_debug!("PID owner changed so cannot get old process exit status");
+                    return None;
                 }
                 std::thread::sleep(std::time::Duration::from_millis(10));
             }
+            let mut exit_status = 0;
+            match GetExitCodeProcess(handle, &mut exit_status) {
+                Ok(_) => Some(ExitStatus::from_raw(exit_status)),
+                Err(_error) => {
+                    sysinfo_debug!("failed to retrieve process exit status: {_error:?}");
+                    None
+                }
+            }
         } else {
             // In this case, we can't do anything so we just return.
             sysinfo_debug!("can't wait on this process so returning");
+            None
         }
     }