2024-07-24 18:06:11 -07:00
|
|
|
use std::fs::File;
|
|
|
|
|
use std::io;
|
2023-11-14 16:55:06 -05:00
|
|
|
use std::process::{exit, Command, Stdio};
|
2024-07-25 09:26:54 -07:00
|
|
|
use tokio::io::AsyncReadExt;
|
2023-11-14 16:55:06 -05:00
|
|
|
|
|
|
|
|
/// Performs a double fork with setsid to spawn and detach a command.
|
2024-07-25 09:26:54 -07:00
|
|
|
pub async fn spawn(mut command: Command) -> Option<u32> {
|
2023-11-14 16:55:06 -05:00
|
|
|
command
|
|
|
|
|
.stdin(Stdio::null())
|
|
|
|
|
.stdout(Stdio::null())
|
|
|
|
|
.stderr(Stdio::null());
|
|
|
|
|
|
2024-07-24 18:06:11 -07:00
|
|
|
let Ok((read, write)) = rustix::pipe::pipe_with(rustix::pipe::PipeFlags::CLOEXEC) else {
|
|
|
|
|
return None;
|
|
|
|
|
};
|
2023-11-14 16:55:06 -05:00
|
|
|
|
2024-07-24 18:06:11 -07:00
|
|
|
match unsafe { libc::fork() } {
|
|
|
|
|
// Parent process
|
2024-07-25 09:26:54 -07:00
|
|
|
1.. => {
|
|
|
|
|
drop(write);
|
2024-07-24 18:06:11 -07:00
|
|
|
// Read PID from pipe
|
2024-07-25 09:26:54 -07:00
|
|
|
let mut read = tokio::net::unix::pipe::Receiver::from_owned_fd(read).unwrap();
|
|
|
|
|
read.read_u32().await.ok()
|
2024-07-24 18:06:11 -07:00
|
|
|
}
|
2023-11-14 16:55:06 -05:00
|
|
|
|
2024-07-24 18:06:11 -07:00
|
|
|
// Child process
|
|
|
|
|
0 => {
|
|
|
|
|
let _res = rustix::process::setsid();
|
|
|
|
|
if let Ok(child) = command.spawn() {
|
|
|
|
|
// Write PID to pipe
|
2024-07-25 09:26:54 -07:00
|
|
|
let _ = rustix::io::write(write, &child.id().to_be_bytes());
|
2023-11-14 16:55:06 -05:00
|
|
|
}
|
2024-07-24 18:06:11 -07:00
|
|
|
|
|
|
|
|
exit(0)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
..=-1 => {
|
|
|
|
|
println!(
|
|
|
|
|
"failed to fork and spawn command: {}",
|
|
|
|
|
io::Error::last_os_error()
|
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
None
|
2023-11-14 16:55:06 -05:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|