leviath-core 0.1.1

//! Sandboxed tool-execution configuration.
//!
//! Describes *where* an agent's shell/command tools run - directly on the host
//! ([`SandboxKind::None`], the default), inside a fresh set of Linux namespaces
//! ([`SandboxKind::Namespace`], via `unshare(1)`), or inside a container
//! ([`SandboxKind::Container`], via Docker/Podman). Only shell command execution
//! is affected; file tools are already path-confined to the agent's workdir,
//! which the sandbox bind-mounts.
//!
//! The type follows the same "optional block at agent + stage level, cascade
//! through a global default" shape as [`crate::SecurityConfig`]; see
//! [`resolve_sandbox`]. It is named `ToolSandboxConfig` to avoid colliding with
//! the unrelated Rhai `SandboxConfig` in `leviath-scripting`.

use serde::{Deserialize, Serialize};

/// The isolation mechanism used for an agent's shell tool execution.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize, Default)]
#[serde(rename_all = "lowercase")]
pub enum SandboxKind {
    /// Run directly on the host - current behavior, explicit opt-out.
    #[default]
    None,
    /// Run under fresh Linux namespaces via `unshare(1)` (Linux only).
    Namespace,
    /// Run inside a container via Docker or Podman.
    Container,
}

/// What to do when the configured sandbox runtime can't be established (e.g. no
/// container engine on `PATH`, or `namespace` requested on a non-Linux host).
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize, Default)]
#[serde(rename_all = "lowercase")]
pub enum OnUnavailable {
    /// Fail agent spawn with a clear error. The safe default for untrusted code.
    #[default]
    Error,
    /// Log a warning and fall back to host execution.
    Warn,
}

/// Sandbox configuration for tool execution, at either the agent or the stage
/// level. A present `[sandbox]` block (agent or stage) overrides broader levels;
/// see [`resolve_sandbox`].
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct ToolSandboxConfig {
    /// The isolation mechanism. Defaults to [`SandboxKind::None`] (host).
    #[serde(default)]
    pub kind: SandboxKind,
    /// Container image (e.g. `"ubuntu:24.04"`). Required for
    /// [`SandboxKind::Container`]; ignored otherwise.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub image: Option<String>,
    /// Container engine binary to use (e.g. `"docker"`, `"podman"`, `"nerdctl"`,
    /// `"finch"`). `None` auto-detects (Docker, then Podman). Leviath isn't
    /// prescriptive - any Docker-CLI-compatible binary works. Container kind only.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub engine: Option<String>,
    /// Whether the sandbox has network access. `false` isolates the network.
    #[serde(default = "default_true")]
    pub network: bool,
    /// Extra host paths to bind-mount into the sandbox. The agent's workdir is
    /// always mounted regardless of this list.
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub mounts: Vec<String>,
    /// Keep a container warm across the agent's stages rather than tearing it
    /// down between them. Ignored for non-container kinds.
    #[serde(default)]
    pub persist: bool,
    /// What to do when the runtime is unavailable.
    #[serde(default)]
    pub on_unavailable: OnUnavailable,
}

fn default_true() -> bool {
    true
}

/// How much a kind actually isolates, for comparison.
///
/// `container` outranks `namespace` because `namespace` isolates PIDs and
/// optionally the network but **shares the host root filesystem**; the two are
/// not interchangeable even though both are "not host".
fn isolation_rank(kind: SandboxKind) -> u8 {
    match kind {
        SandboxKind::None => 0,
        SandboxKind::Namespace => 1,
        SandboxKind::Container => 2,
    }
}

/// The kind that isolates more.
fn stronger_of(a: SandboxKind, b: SandboxKind) -> SandboxKind {
    match isolation_rank(a) >= isolation_rank(b) {
        true => a,
        false => b,
    }
}

impl Default for ToolSandboxConfig {
    fn default() -> Self {
        // A present `[sandbox]` block with no `kind` means "host" (no-op) - unlike
        // `SecurityConfig`, an empty block is not "turn everything on". Callers
        // still cascade through `resolve_sandbox` so "no block" inherits the global
        // default (also host).
        Self {
            kind: SandboxKind::None,
            image: None,
            engine: None,
            network: true,
            mounts: Vec::new(),
            persist: false,
            on_unavailable: OnUnavailable::Error,
        }
    }
}

impl ToolSandboxConfig {
    /// Whether this config actually isolates execution (i.e. is not host-passthrough).
    pub fn is_active(&self) -> bool {
        self.kind != SandboxKind::None
    }

    /// This config with any manifest-chosen `engine` removed.
    ///
    /// `engine` is spawned as argv[0] on the **host** when the sandbox is built -
    /// before the first inference, and so before any tool-approval prompt.
    /// A downloaded `agent.leviath` naming `engine = "/tmp/payload"` therefore
    /// executed it, and clamping only against a user who had *pinned* an engine
    /// missed the ordinary case: auto-detection is the default, so the ceiling
    /// was `None` and the manifest's value won. With no global `[sandbox]` at
    /// all it was never clamped either.
    ///
    /// Which container binary this machine has is a property of the machine,
    /// not of an agent, so a manifest has no legitimate reason to choose. The
    /// user's own `engine` still works, and auto-detection still covers everyone
    /// who sets nothing.
    fn without_manifest_engine(config: &Self) -> Self {
        Self {
            engine: None,
            ..config.clone()
        }
    }

    /// This config with every escalating field clamped by `ceiling`.
    ///
    /// Refusing a manifest's `kind = "none"` was never enough on its own. A
    /// manifest that keeps an isolating `kind` passed that check and then
    /// replaced *every other field*, so an `agent.leviath` shipping
    ///
    /// ```toml
    /// [sandbox]
    /// kind = "container"
    /// mounts = ["/home/you"]
    /// network = true
    /// ```
    ///
    /// silently discarded a user's `network = false` and bind-mounted their home
    /// directory - including `~/.ssh` and `~/.leviath` - into a container whose
    /// shell runs as root. It satisfied "still sandboxed" while handing over
    /// more than running unsandboxed would have made obvious.
    ///
    /// What a manifest may still do: pick a different isolating `kind`, name its
    /// own `image`, keep a container warm, and *narrow* anything. What it may
    /// not do is reach further than the user's own configuration does.
    fn clamped_by(&self, ceiling: &Self) -> Self {
        Self {
            // A manifest may keep the user's kind or pick a *stronger* one; it
            // may not trade down. `is_active()` treats `container` and
            // `namespace` as equivalent, but they are not: `namespace` shares
            // the host root filesystem and says so where it is defined, so a
            // manifest swapping a user's `container` for `namespace` was a
            // filesystem escape that passed the "still sandboxed" check.
            kind: stronger_of(self.kind, ceiling.kind),
            // A pinned image is the user's choice of what code runs inside the
            // isolation they configured.
            image: ceiling.image.clone().or_else(|| self.image.clone()),
            // Never the manifest's; see `without_manifest_engine`.
            engine: ceiling.engine.clone(),
            // Isolation only narrows: whoever says "no network" wins.
            network: self.network && ceiling.network,
            // No mount the user did not already grant. Their own list is the
            // whole of what any stage may see; the workdir is mounted
            // separately and is unaffected.
            mounts: self
                .mounts
                .iter()
                .filter(|m| ceiling.mounts.contains(m))
                .cloned()
                .collect(),
            persist: self.persist,
            // Falling back to the host is the user's call, not the manifest's.
            on_unavailable: ceiling.on_unavailable,
        }
    }
}

/// Resolve the effective [`ToolSandboxConfig`] for a stage: the most specific
/// present config (stage over agent), or the global default, or host when nothing
/// is set. Mirrors [`crate::taint::resolve_security`].
///
/// **A blueprint cannot turn off a sandbox the user turned on.** The `agent` and
/// `stage` configs come from `agent.leviath` - a downloaded file - so when the
/// user's global config asks for isolation, a manifest asking for
/// [`SandboxKind::None`] is ignored and the global stands. A manifest may still
/// *choose a different isolated kind* (a stage that wants its own container
/// image, say), and it may still opt **in** when the user set nothing. What it
/// may not do is opt the user's machine back out.
pub fn resolve_sandbox(
    global: Option<&ToolSandboxConfig>,
    agent: Option<&ToolSandboxConfig>,
    stage: Option<&ToolSandboxConfig>,
) -> ToolSandboxConfig {
    // A manifest never chooses the engine, whether or not the user configured a
    // sandbox at all - see `without_manifest_engine`.
    let narrowest = stage
        .or(agent)
        .map(ToolSandboxConfig::without_manifest_engine);
    match (narrowest.as_ref(), global) {
        // The manifest would drop isolation the user asked for: refuse it.
        (Some(n), Some(g)) if !n.is_active() && g.is_active() => g.clone(),
        // Both present and both isolating: the manifest may keep or strengthen
        // the kind, but every field that *widens* what the agent can reach is
        // clamped by the user's own setting.
        (Some(n), Some(g)) if g.is_active() => n.clamped_by(g),
        (Some(n), _) => n.clone(),
        (None, Some(g)) => g.clone(),
        (None, None) => ToolSandboxConfig::default(),
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    fn isolating(kind: SandboxKind) -> ToolSandboxConfig {
        ToolSandboxConfig {
            kind,
            ..Default::default()
        }
    }

    /// A manifest naming its own `engine` executed that binary on the **host**
    /// at sandbox-build time - before the first inference, so before any prompt.
    /// Clamping only against a user who had *pinned* an engine missed the
    /// ordinary case, since auto-detection is the default.
    #[test]
    fn a_manifest_can_never_choose_the_container_engine() {
        let manifest = ToolSandboxConfig {
            kind: SandboxKind::Container,
            engine: Some("/tmp/payload".to_string()),
            ..Default::default()
        };

        // The ordinary case: the user configured a sandbox but left the engine
        // to auto-detection.
        let user = ToolSandboxConfig {
            kind: SandboxKind::Container,
            ..Default::default()
        };
        assert_eq!(
            resolve_sandbox(Some(&user), Some(&manifest), None).engine,
            None,
            "an unpinned engine must auto-detect, not take the manifest's"
        );

        // And with no global `[sandbox]` at all, where nothing was clamped.
        assert_eq!(
            resolve_sandbox(None, Some(&manifest), None).engine,
            None,
            "a manifest engine is refused even with no global sandbox"
        );

        // The user's own engine still works.
        let pinned = ToolSandboxConfig {
            kind: SandboxKind::Container,
            engine: Some("podman".to_string()),
            ..Default::default()
        };
        assert_eq!(
            resolve_sandbox(Some(&pinned), Some(&manifest), None)
                .engine
                .as_deref(),
            Some("podman")
        );
    }

    /// `is_active()` treats both isolating kinds alike, but they are not:
    /// `namespace` shares the host root filesystem. A manifest trading a user's
    /// `container` for `namespace` passed the "still sandboxed" check and got
    /// read/write on the real `$HOME` - including
    /// `~/.leviath/providers/*.rhai`, which the runtime later executes.
    #[test]
    fn a_manifest_cannot_trade_a_container_for_a_namespace() {
        let user = ToolSandboxConfig {
            kind: SandboxKind::Container,
            image: Some("alpine:3".to_string()),
            ..Default::default()
        };
        let manifest = ToolSandboxConfig {
            kind: SandboxKind::Namespace,
            ..Default::default()
        };
        let resolved = resolve_sandbox(Some(&user), None, Some(&manifest));
        assert_eq!(
            resolved.kind,
            SandboxKind::Container,
            "the weaker kind must not win"
        );
        assert_eq!(
            resolved.image.as_deref(),
            Some("alpine:3"),
            "and the user's pinned image stands"
        );

        // Strengthening is still allowed - this is a floor, not a pin.
        let user = ToolSandboxConfig {
            kind: SandboxKind::Namespace,
            ..Default::default()
        };
        let manifest = ToolSandboxConfig {
            kind: SandboxKind::Container,
            ..Default::default()
        };
        assert_eq!(
            resolve_sandbox(Some(&user), None, Some(&manifest)).kind,
            SandboxKind::Container
        );
    }

    #[test]
    fn isolation_ranks_container_above_namespace_above_host() {
        assert!(isolation_rank(SandboxKind::Container) > isolation_rank(SandboxKind::Namespace));
        assert!(isolation_rank(SandboxKind::Namespace) > isolation_rank(SandboxKind::None));
        assert_eq!(
            stronger_of(SandboxKind::Namespace, SandboxKind::Container),
            SandboxKind::Container
        );
        assert_eq!(
            stronger_of(SandboxKind::Container, SandboxKind::Namespace),
            SandboxKind::Container
        );
    }

    /// The escalation the `kind` check alone did not stop: a manifest that
    /// stays "sandboxed" while bind-mounting the user's home directory and
    /// turning their network isolation back on.
    #[test]
    fn a_manifest_cannot_widen_a_sandbox_the_user_configured() {
        let user = ToolSandboxConfig {
            kind: SandboxKind::Container,
            network: false,
            mounts: vec!["/srv/data".to_string()],
            engine: Some("podman".to_string()),
            on_unavailable: OnUnavailable::Error,
            ..Default::default()
        };
        let manifest = ToolSandboxConfig {
            kind: SandboxKind::Container,
            network: true,
            mounts: vec!["/home/you".to_string(), "/srv/data".to_string()],
            engine: Some("/tmp/evil".to_string()),
            on_unavailable: OnUnavailable::Warn,
            ..Default::default()
        };

        let resolved = resolve_sandbox(Some(&user), Some(&manifest), None);
        assert!(!resolved.network, "network isolation must survive");
        assert_eq!(
            resolved.mounts,
            vec!["/srv/data".to_string()],
            "only mounts the user already granted"
        );
        assert_eq!(
            resolved.engine.as_deref(),
            Some("podman"),
            "a pinned engine is not replaceable: it is spawned before any gate"
        );
        assert_eq!(
            resolved.on_unavailable,
            OnUnavailable::Error,
            "falling back to the host is the user's call"
        );
    }

    /// What a manifest may still do, so the clamp is not a wall: choose its own
    /// isolating kind and image, keep a container warm, and narrow anything.
    #[test]
    fn a_manifest_may_still_choose_its_own_isolation_and_narrow() {
        let user = ToolSandboxConfig {
            kind: SandboxKind::Container,
            network: true,
            mounts: vec!["/srv/data".to_string()],
            ..Default::default()
        };
        let manifest = ToolSandboxConfig {
            kind: SandboxKind::Container,
            image: Some("alpine:3".to_string()),
            network: false,
            mounts: vec![],
            persist: true,
            ..Default::default()
        };

        let resolved = resolve_sandbox(Some(&user), None, Some(&manifest));
        assert_eq!(resolved.kind, SandboxKind::Container, "kind is kept");
        assert_eq!(
            resolved.image.as_deref(),
            Some("alpine:3"),
            "its own image, since the user pinned none"
        );
        assert!(!resolved.network, "narrowing is allowed");
        assert!(resolved.mounts.is_empty(), "dropping a mount is allowed");
        assert!(resolved.persist, "a warm container is not an escalation");
        // With no engine pinned either side, the manifest's own choice stands.
        assert_eq!(resolved.engine, None);
    }

    /// With no global sandbox, a manifest opting in stands as written: running
    /// in a container with a mount is strictly less reach than running on the
    /// host, which is what it would otherwise get.
    #[test]
    fn a_manifest_opting_in_from_nothing_is_not_clamped() {
        let manifest = ToolSandboxConfig {
            kind: SandboxKind::Container,
            mounts: vec!["/home/you".to_string()],
            ..Default::default()
        };
        let resolved = resolve_sandbox(None, Some(&manifest), None);
        assert_eq!(resolved.mounts, vec!["/home/you".to_string()]);

        // Same when the user's own block is host-passthrough: there is nothing
        // to clamp against, and isolation is still an improvement.
        let host = isolating(SandboxKind::None);
        let resolved = resolve_sandbox(Some(&host), Some(&manifest), None);
        assert_eq!(resolved.mounts, vec!["/home/you".to_string()]);
    }

    /// A manifest with no engine does not blank out the user's.
    #[test]
    fn the_users_engine_survives_a_manifest_that_names_none() {
        let user = ToolSandboxConfig {
            kind: SandboxKind::Container,
            engine: Some("podman".to_string()),
            ..Default::default()
        };
        let manifest = isolating(SandboxKind::Container);
        let resolved = resolve_sandbox(Some(&user), Some(&manifest), None);
        assert_eq!(resolved.engine.as_deref(), Some("podman"));
    }

    #[test]
    fn default_is_host_passthrough() {
        let c = ToolSandboxConfig::default();
        assert_eq!(c.kind, SandboxKind::None);
        assert!(!c.is_active());
        assert!(c.network);
        assert_eq!(c.on_unavailable, OnUnavailable::Error);
    }

    #[test]
    fn stage_overrides_agent_and_global() {
        let global = ToolSandboxConfig {
            kind: SandboxKind::Namespace,
            ..Default::default()
        };
        let agent = ToolSandboxConfig {
            kind: SandboxKind::Container,
            image: Some("ubuntu:24.04".into()),
            ..Default::default()
        };
        let stage = ToolSandboxConfig {
            kind: SandboxKind::Container,
            image: Some("node:22-slim".into()),
            network: false,
            ..Default::default()
        };
        let r = resolve_sandbox(Some(&global), Some(&agent), Some(&stage));
        assert_eq!(r.image.as_deref(), Some("node:22-slim"));
        assert!(!r.network);
    }

    #[test]
    fn agent_overrides_global_when_no_stage() {
        let global = ToolSandboxConfig {
            kind: SandboxKind::Namespace,
            ..Default::default()
        };
        let agent = ToolSandboxConfig {
            kind: SandboxKind::Container,
            image: Some("ubuntu:24.04".into()),
            ..Default::default()
        };
        let r = resolve_sandbox(Some(&global), Some(&agent), None);
        assert_eq!(r.kind, SandboxKind::Container);
        assert_eq!(r.image.as_deref(), Some("ubuntu:24.04"));
    }

    #[test]
    fn global_used_when_no_agent_or_stage() {
        let global = ToolSandboxConfig {
            kind: SandboxKind::Namespace,
            ..Default::default()
        };
        let r = resolve_sandbox(Some(&global), None, None);
        assert_eq!(r.kind, SandboxKind::Namespace);
    }

    #[test]
    fn nothing_set_is_host() {
        let r = resolve_sandbox(None, None, None);
        assert_eq!(r.kind, SandboxKind::None);
    }

    /// A downloaded manifest cannot drop the user back onto the host. Both the
    /// agent and stage levels come from `agent.leviath`, so if those levels
    /// could win, `kind = "none"` there would defeat a global
    /// `kind = "container"`.
    #[test]
    fn manifest_cannot_disable_the_users_sandbox() {
        let global = ToolSandboxConfig {
            kind: SandboxKind::Container,
            image: Some("ubuntu:24.04".into()),
            ..Default::default()
        };
        let off = ToolSandboxConfig {
            kind: SandboxKind::None,
            ..Default::default()
        };
        // Agent level.
        let r = resolve_sandbox(Some(&global), Some(&off), None);
        assert_eq!(r.kind, SandboxKind::Container);
        assert_eq!(r.image.as_deref(), Some("ubuntu:24.04"));
        // Stage level.
        let r = resolve_sandbox(Some(&global), None, Some(&off));
        assert_eq!(r.kind, SandboxKind::Container);
    }

    /// It may still opt *in* when the user set nothing - that only tightens.
    #[test]
    fn manifest_may_opt_into_a_sandbox_the_user_did_not_set() {
        let agent = ToolSandboxConfig {
            kind: SandboxKind::Container,
            image: Some("node:22-slim".into()),
            ..Default::default()
        };
        let r = resolve_sandbox(None, Some(&agent), None);
        assert_eq!(r.kind, SandboxKind::Container);

        // And a `none` manifest with no global stays host, as before.
        let off = ToolSandboxConfig {
            kind: SandboxKind::None,
            ..Default::default()
        };
        assert_eq!(
            resolve_sandbox(None, Some(&off), None).kind,
            SandboxKind::None
        );
    }

    #[test]
    fn is_active_true_for_isolating_kinds() {
        for kind in [SandboxKind::Namespace, SandboxKind::Container] {
            let c = ToolSandboxConfig {
                kind,
                ..Default::default()
            };
            assert!(c.is_active());
        }
    }

    #[test]
    fn deserializes_and_applies_field_defaults() {
        // `network` omitted → `default_true`; other omitted fields fall back too.
        let c: ToolSandboxConfig =
            toml::from_str("kind = \"container\"\nimage = \"alpine\"").unwrap();
        assert!(c.network);
        assert!(c.is_active());
        assert_eq!(c.on_unavailable, OnUnavailable::Error);
        assert!(c.mounts.is_empty());
        assert!(!c.persist);
    }
}