61 lines
1.8 KiB
Rust
61 lines
1.8 KiB
Rust
use anyhow::Result;
|
|
|
|
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
|
|
pub enum EpiphanyProcessObservation {
|
|
Alive,
|
|
Dead,
|
|
Missing,
|
|
}
|
|
|
|
impl EpiphanyProcessObservation {
|
|
pub fn label(self) -> &'static str {
|
|
match self {
|
|
Self::Alive => "alive",
|
|
Self::Dead => "dead",
|
|
Self::Missing => "missing",
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(windows)]
|
|
pub fn observe_native_process(process_id: u32) -> Result<EpiphanyProcessObservation> {
|
|
use windows_sys::Win32::Foundation::CloseHandle;
|
|
use windows_sys::Win32::System::Threading::{
|
|
GetExitCodeProcess, OpenProcess, PROCESS_QUERY_LIMITED_INFORMATION,
|
|
};
|
|
const STILL_ACTIVE: u32 = 259;
|
|
unsafe {
|
|
let handle = OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, 0, process_id);
|
|
if handle.is_null() {
|
|
return Ok(EpiphanyProcessObservation::Dead);
|
|
}
|
|
let mut exit_code = 0_u32;
|
|
let read = GetExitCodeProcess(handle, &mut exit_code);
|
|
CloseHandle(handle);
|
|
if read == 0 {
|
|
anyhow::bail!("failed to inspect managed service process {process_id}");
|
|
}
|
|
Ok(if exit_code == STILL_ACTIVE {
|
|
EpiphanyProcessObservation::Alive
|
|
} else {
|
|
EpiphanyProcessObservation::Dead
|
|
})
|
|
}
|
|
}
|
|
|
|
#[cfg(unix)]
|
|
pub fn observe_native_process(process_id: u32) -> Result<EpiphanyProcessObservation> {
|
|
let result = unsafe { libc::kill(process_id as i32, 0) };
|
|
if result == 0 {
|
|
Ok(EpiphanyProcessObservation::Alive)
|
|
} else {
|
|
let error = std::io::Error::last_os_error();
|
|
if error.raw_os_error() == Some(libc::ESRCH) {
|
|
Ok(EpiphanyProcessObservation::Dead)
|
|
} else if error.raw_os_error() == Some(libc::EPERM) {
|
|
Ok(EpiphanyProcessObservation::Alive)
|
|
} else {
|
|
Err(error.into())
|
|
}
|
|
}
|
|
}
|