sprawling-runtime 0.0.10

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.
// Copyright (c) 2026 2youg1 and the sprawling contributors

//! How long a caller waits for a child, and how that child's ending is
//! read.
//!
//! Both facts decide what a caller writes down. The window decides
//! whether a command comes back as `Settled` or as `Backgrounded`, and
//! those two answers carry different payloads; the ending decides what
//! the `exec` result says happened. Neither may be spelled twice, so
//! both live here (`crates/runtime/spec/Backlog.lean` §8-28-1).

use std::process::ExitStatus;
use std::time::Duration;

use kernel::AxError;

use super::ceiling::Ceiling;
use super::tail::Tail;
use super::{BacklogId, Body, Claim, Member, Table};

/// The rate a watched command's output is read at, 64 KiB a second:
/// several times what a person reads on a page, and small enough that a
/// flooding child costs each poll a bounded copy (`crates/runtime/spec/Backlog.lean` §8-28-3).
const READ_BYTES_PER_MS: u64 = 64;

/// The short window a caller blocks for, counted in polls.
///
/// Injected rather than read from a constant where the waiting happens.
/// A caller that cannot choose the window has to spend the production
/// one to observe either answer, so the test that proves a command is
/// handed to the background costs ten seconds - and a suite that cannot
/// afford that instead proves nothing about the shape it ships.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct PollBudget {
    polls: u32,
    interval_ms: u64,
}

impl PollBudget {
    /// What a person gets: 500 polls of 20 ms, which is ten seconds.
    pub const DEFAULT: PollBudget = PollBudget {
        polls: 500,
        interval_ms: 20,
    };

    /// A window of this many polls, waiting this long between them.
    #[must_use]
    pub fn new(polls: u32, interval_ms: u64) -> PollBudget {
        PollBudget { polls, interval_ms }
    }

    pub(super) fn polls(self) -> u32 {
        self.polls
    }

    pub(super) fn interval(self) -> Duration {
        Duration::from_millis(self.interval_ms)
    }

    /// How many bytes of a running command's output one poll reads:
    /// proportional to the interval, so a longer interval reads more per
    /// visit and the rate a page is fed stays the same.
    pub(super) fn read_per_poll(self) -> usize {
        usize::try_from(self.interval_ms.saturating_mul(READ_BYTES_PER_MS)).unwrap_or(usize::MAX)
    }
}

impl Default for PollBudget {
    fn default() -> PollBudget {
        PollBudget::DEFAULT
    }
}

/// How a child process stopped.
///
/// Three answers rather than one number: one integer cannot tell "it
/// returned minus one", "a signal stopped it" and "this city never found
/// out" apart, and a program that really does return minus one is
/// ordinary.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Exit {
    /// It ran to the end and returned this code.
    Ended { code: i32 },
    /// A signal stopped it, so there is no code to return. This is what
    /// `halt` leaves behind on a Unix host.
    Signalled,
    /// This city cannot say how it stopped.
    Unknown { why: Unseen },
}

impl Exit {
    /// Reads an ending off the status the operating system reported.
    pub(super) fn of(status: &ExitStatus) -> Exit {
        match status.code() {
            Some(code) => Exit::Ended { code },
            None => Exit::Signalled,
        }
    }

    /// Whether a child has stopped, and how, without waiting for it.
    ///
    /// A host that will not answer is not a child that is still
    /// running: the ending is reported as unread, so the member stops
    /// being polled for ever.
    pub(super) fn polled(child: &mut std::process::Child) -> Option<Exit> {
        match child.try_wait() {
            Ok(Some(status)) => Some(Exit::of(&status)),
            Ok(None) => None,
            Err(_) => Some(Exit::Unknown {
                why: Unseen::WaitRefused,
            }),
        }
    }

    /// What this ending is called wherever it is written down.
    #[must_use]
    pub fn as_str(self) -> &'static str {
        match self {
            Exit::Ended { .. } => "ended",
            Exit::Signalled => "signalled",
            Exit::Unknown { .. } => "unknown",
        }
    }
}

/// Why this city cannot say how a child stopped.
///
/// Two situations, because the caller is owed different things by them:
/// a member that left the table was never waited on at all, while a wait
/// that the host refused happened and gave no answer.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Unseen {
    /// The member was gone from the table before it could be waited on.
    LeftTheTable,
    /// The host refused to report the child's state.
    WaitRefused,
}

impl Unseen {
    /// The sentence a reader of a tool result is given.
    #[must_use]
    pub fn sentence(self) -> &'static str {
        match self {
            Unseen::LeftTheTable => {
                "this command left the table before anything waited on it, so its \
                 ending was never read"
            }
            Unseen::WaitRefused => {
                "this machine refused to report the child's state, so its ending is \
                 not known"
            }
        }
    }
}

impl super::Backlog {
    /// Whether this member has stopped, and how the memory ceiling fared
    /// while it ran. Removing it here is what keeps
    /// [`super::Backlog::harvest`] from reporting a result its own caller is
    /// about to return.
    pub(super) fn settle(&self, id: BacklogId) -> Result<Option<(Exit, Option<Ceiling>)>, AxError> {
        let mut table = self.hold()?;
        let Table { members, jobs, .. } = &mut *table;
        let Some(Member {
            body:
                Body::Command {
                    child,
                    claim,
                    ceiling,
                    ..
                },
            ..
        }) = members.get_mut(&id)
        else {
            return Ok(Some((
                Exit::Unknown {
                    why: Unseen::LeftTheTable,
                },
                None,
            )));
        };
        let stopped = match child.poll() {
            Ok(stopped) => stopped,
            Err(err) => {
                if let Claim::Window(owner) = *claim {
                    *claim = Claim::Run(owner);
                }
                return Err(err);
            }
        };
        let Some(exit) = stopped else {
            return Ok(None);
        };
        let reported = match *claim {
            Claim::Window(owner) | Claim::Run(owner) => jobs.ceiling(owner, *ceiling),
            Claim::Nobody => None,
        };
        members.remove(&id);
        Ok(Some((exit, reported)))
    }

    /// Past the window: the command is owed to its run's harvest, which
    /// reads its output on from where the window's `read` stopped.
    pub(super) fn hand_over(&self, id: BacklogId, read: Tail) -> Result<(), AxError> {
        let mut table = self.hold()?;
        if let Some(Member {
            body: Body::Command { claim, tail, .. },
            ..
        }) = table.members.get_mut(&id)
            && let Claim::Window(owner) = *claim
        {
            *claim = Claim::Run(owner);
            *tail = read;
        }
        Ok(())
    }
}

impl super::Backlog {
    pub(super) fn watch(
        &self,
        id: BacklogId,
        owner: kernel::RunId,
        what: String,
        dir: std::path::PathBuf,
    ) -> Result<super::Started, AxError> {
        let mut tail = Tail::default();
        for _ in 0..self.window.polls() {
            if let Some((exit, ceiling)) = self.settle(id)? {
                let (stdout, stderr) = super::collect(&dir);
                return Ok(super::Started::Settled {
                    exit,
                    stdout,
                    stderr,
                    ceiling,
                });
            }
            if let Some(sink) = &self.sink {
                sink.deliver(tail.take(&dir, (owner, id), self.window.read_per_poll()));
            }
            std::thread::sleep(self.window.interval());
        }
        self.hand_over(id, tail)?;
        Ok(super::Started::Backgrounded { id, what })
    }
}

#[cfg(test)]
#[allow(clippy::unwrap_used, clippy::expect_used, reason = "test code")]
mod tests {
    use super::*;

    /// The three endings are three spellings, and the one that carries a
    /// number is the only one that carries a number.
    #[test]
    fn every_ending_says_which_of_the_three_it_is() {
        assert_eq!(Exit::Ended { code: -1 }.as_str(), "ended");
        assert_eq!(Exit::Signalled.as_str(), "signalled");
        assert_eq!(
            Exit::Unknown {
                why: Unseen::WaitRefused
            }
            .as_str(),
            "unknown"
        );
        assert_ne!(
            Exit::Ended { code: -1 },
            Exit::Signalled,
            "a program that returns minus one has not been signalled"
        );
    }

    /// The window a person gets is the one the SPEC states.
    #[test]
    fn the_default_window_is_ten_seconds_of_short_polls() {
        let budget = PollBudget::DEFAULT;
        assert_eq!(budget.polls(), 500);
        assert_eq!(budget.interval(), Duration::from_millis(20));
    }
}