tawasuyu/brahman

1//! Bootstrap del entorno kernel para PID 1: monta procfs/sysfs/devtmpfs/cgroup2
2//! y registra al proceso como subreaper para adoptar huérfanos.
3//!
4//! Idempotente: si los puntos de montaje ya existen (initramfs los montó),
5//! el segundo mount falla con EBUSY y simplemente lo ignoramos.
6
7use nix::mount::{mount, MsFlags};
8use tracing::debug;
9
10/// Monta los pseudo-filesystems esenciales. Errores benignos (ya montados)
11/// se ignoran; errores serios se propagan.
12pub fn bootstrap_kernel_surface() -> anyhow::Result<()> {
13    // Cada uno con sus flags estándar — NOSUID/NOEXEC/NODEV donde aplica.
14    mount::<str, str, str, str>(
15        Some("proc"), "/proc", Some("proc"),
16        MsFlags::MS_NOSUID | MsFlags::MS_NOEXEC | MsFlags::MS_NODEV, None,
17    ).ok();
18    mount::<str, str, str, str>(
19        Some("sysfs"), "/sys", Some("sysfs"),
20        MsFlags::MS_NOSUID | MsFlags::MS_NOEXEC | MsFlags::MS_NODEV, None,
21    ).ok();
22    mount::<str, str, str, str>(
23        Some("devtmpfs"), "/dev", Some("devtmpfs"),
24        MsFlags::MS_NOSUID, None,
25    ).ok();
26    mount::<str, str, str, str>(
27        Some("cgroup2"), "/sys/fs/cgroup", Some("cgroup2"),
28        MsFlags::MS_NOSUID | MsFlags::MS_NOEXEC | MsFlags::MS_NODEV, None,
29    ).ok();
30    debug!("kernel surface bootstrap completo");
31    Ok(())
32}
33
34/// PR_SET_CHILD_SUBREAPER: que adoptemos huérfanos del fractal.
35///
36/// En PID 1 esto es redundante (el kernel ya lo hace), pero se deja explícito
37/// para que ente-zero corriendo como sub-init en un container mantenga la
38/// misma semántica.
39pub fn become_child_subreaper() -> anyhow::Result<()> {
40    let r = unsafe { libc::prctl(libc::PR_SET_CHILD_SUBREAPER, 1u64, 0u64, 0u64, 0u64) };
41    if r != 0 {
42        anyhow::bail!(
43            "prctl PR_SET_CHILD_SUBREAPER falló: {}",
44            std::io::Error::last_os_error()
45        );
46    }
47    Ok(())
48}
49
50/// Cosechar zombis hasta vaciar la cola de niños muertos. Devuelve los
51/// PIDs cosechados con su estado, como tuplas.
52pub fn reap_all() -> Vec<ReapedChild> {
53    use nix::errno::Errno;
54    use nix::sys::wait::{waitpid, WaitPidFlag, WaitStatus};
55    let mut out = Vec::new();
56    loop {
57        match waitpid(None, Some(WaitPidFlag::WNOHANG)) {
58            Ok(WaitStatus::Exited(pid, code)) => {
59                out.push(ReapedChild { pid: pid.as_raw(), status: ReapStatus::Exited(code) });
60            }
61            Ok(WaitStatus::Signaled(pid, sig, _core)) => {
62                out.push(ReapedChild { pid: pid.as_raw(), status: ReapStatus::Signaled(sig as i32) });
63            }
64            Ok(WaitStatus::StillAlive) => return out,
65            Err(Errno::ECHILD) => return out,
66            Err(_) => return out,
67            Ok(_) => continue, // Stopped/Continued — irrelevantes
68        }
69    }
70    // unreachable, satisface al borrow checker
71    #[allow(unreachable_code)]
72    out
73}
74
75#[derive(Debug, Clone)]
76pub struct ReapedChild {
77    pub pid: i32,
78    pub status: ReapStatus,
79}
80
81#[derive(Debug, Clone)]
82pub enum ReapStatus {
83    Exited(i32),
84    Signaled(i32),
85}