elfpak-core 0.5.1

//! Runtime policy: everything that ELF analysis cannot prove.
//!
//! Presets are configuration only. Every file they contribute shows up in the
//! bundle plan with a policy reason attached.

use crate::error::Error;
use serde::{Deserialize, Serialize};
use std::path::{Path, PathBuf};

#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum Preset {
    Minimal,
    Web,
}

impl std::str::FromStr for Preset {
    type Err = String;

    fn from_str(s: &str) -> Result<Preset, String> {
        match s {
            "minimal" => Ok(Preset::Minimal),
            "web" => Ok(Preset::Web),
            other => Err(format!(
                "unknown preset `{other}` (expected minimal or web)"
            )),
        }
    }
}

impl std::fmt::Display for Preset {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.write_str(match self {
            Preset::Minimal => "minimal",
            Preset::Web => "web",
        })
    }
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum RuntimeFeature {
    CaCertificates,
    Tmp,
    PasswdGroup,
    Nsswitch,
    Tzdata,
    LdSoCache,
}

impl RuntimeFeature {
    pub fn as_str(&self) -> &'static str {
        match self {
            RuntimeFeature::CaCertificates => "ca-certificates",
            RuntimeFeature::Tmp => "tmp",
            RuntimeFeature::PasswdGroup => "passwd-group",
            RuntimeFeature::Nsswitch => "nsswitch",
            RuntimeFeature::Tzdata => "tzdata",
            RuntimeFeature::LdSoCache => "ld-so-cache",
        }
    }
}

/// Whether the bundle gets a generated `/etc/ld.so.cache`.
///
/// The loader searches a fixed set of directories plus whatever the objects
/// themselves declare; everything else it knows comes from the cache, and a
/// bundle has no `ldconfig` to build one. [`CachePolicy::Auto`] therefore
/// writes a cache exactly when the plan contains something the loader would
/// otherwise fail to find.
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq)]
pub enum CachePolicy {
    #[default]
    Auto,
    Always,
    Never,
}

impl CachePolicy {
    pub fn as_str(&self) -> &'static str {
        match self {
            CachePolicy::Auto => "auto",
            CachePolicy::Always => "always",
            CachePolicy::Never => "never",
        }
    }

    /// `--ld-so-cache[=BOOL]`: absent leaves the decision to the planner.
    pub fn from_flag(value: Option<bool>) -> CachePolicy {
        match value {
            None => CachePolicy::Auto,
            Some(true) => CachePolicy::Always,
            Some(false) => CachePolicy::Never,
        }
    }

    /// Whether to write a cache, given whether the plan needs one.
    pub fn applies(&self, needed: bool) -> bool {
        match self {
            CachePolicy::Auto => needed,
            CachePolicy::Always => true,
            CachePolicy::Never => false,
        }
    }
}

impl std::fmt::Display for CachePolicy {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.write_str(self.as_str())
    }
}

/// The identity the packaged application is expected to run as.
///
/// The fields are private because the name is rendered verbatim into
/// `/etc/passwd` and `/etc/group`, which are colon- and newline-delimited. A
/// value that reached those files unchecked could declare a second account —
/// including a uid-0 one with a shell — in an image whose whole point is that
/// it contains nothing unaudited. Construct one with [`UserSpec::parse`] or
/// [`UserSpec::new`], both of which enforce that invariant.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct UserSpec {
    uid: u32,
    gid: u32,
    name: String,
    group: String,
}

impl std::fmt::Display for UserSpec {
    /// Canonical `name:uid:gid` form, which [`UserSpec::parse`] round-trips.
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}:{}:{}", self.name, self.uid, self.gid)
    }
}

impl UserSpec {
    /// Name used when only numeric ids were given; a passwd entry needs one.
    const NAME_DEFAULT: &'static str = "app";

    pub fn uid(&self) -> u32 {
        self.uid
    }

    pub fn gid(&self) -> u32 {
        self.gid
    }

    pub fn name(&self) -> &str {
        &self.name
    }

    pub fn group(&self) -> &str {
        &self.group
    }

    /// A checked identity. The name must be a portable account name, and must
    /// not contradict the two accounts every image already has.
    pub fn new(name: &str, uid: u32, gid: u32) -> Result<UserSpec, Error> {
        if !is_safe_account_name(name) {
            return Err(Error::Config {
                message: format!(
                    "invalid account name `{name}` \
                     (1-32 characters of A-Z, a-z, 0-9, `_` or `-`)"
                ),
            });
        }
        check_reserved_account(name, uid, gid)?;
        Ok(UserSpec {
            uid,
            gid,
            name: name.to_string(),
            group: name.to_string(),
        })
    }

    /// Accepts `uid`, `uid:gid` and `name:uid:gid`.
    pub fn parse(value: &str) -> Result<UserSpec, Error> {
        let invalid = || Error::Config {
            message: format!("invalid --user value `{value}` (expected uid[:gid] or name:uid:gid)"),
        };
        let number = |text: &str| text.parse::<u32>().map_err(|_| invalid());

        let parts: Vec<&str> = value.split(':').collect();
        let (name, uid, gid) = match parts.as_slice() {
            [uid] => (None, number(uid)?, number(uid)?),
            [uid, gid] => (None, number(uid)?, number(gid)?),
            [name, uid, gid] => (Some(*name), number(uid)?, number(gid)?),
            _ => return Err(invalid()),
        };
        // A caller that gave only numbers named no account, so naming one is
        // this function's job: `--user 65534:65534` means `nobody`, and calling
        // it `app` would be inventing a second account for those ids.
        let name =
            name.unwrap_or_else(|| reserved_name(uid, gid).unwrap_or(UserSpec::NAME_DEFAULT));
        UserSpec::new(name, uid, gid).map_err(|error| match error {
            // Keep the option in the message; `new` does not know it exists.
            Error::Config { message } => Error::Config {
                message: format!("invalid --user value `{value}`: {message}"),
            },
            other => other,
        })
    }
}

/// Every image already has `root` and `nobody`. A requested account that reuses
/// one of their names or ids without being that account would put two entries
/// with one name, or one id, into `/etc/passwd`; `getpwnam` and `getpwuid` then
/// disagree with the identity the process actually runs as.
fn check_reserved_account(name: &str, uid: u32, gid: u32) -> Result<(), Error> {
    for (account, group, id) in RESERVED_ACCOUNTS {
        let named = name == *account || name == *group;
        if named {
            // Being one of these accounts is fine; redefining it is not.
            if uid == *id && gid == *id {
                return Ok(());
            }
            return Err(Error::Config {
                message: format!("`{name}` is the reserved account {account}:{id}:{id}"),
            });
        }
        // A reserved *uid* under another name would leave the requested
        // account out of `/etc/passwd` entirely, since that file already has an
        // entry for the id. A reserved *gid* is fine: `/etc/group` already has
        // that group, and the passwd entry simply refers to it.
        if uid == *id {
            return Err(Error::Config {
                message: format!(
                    "uid {id} belongs to the reserved account `{account}`, not to `{name}`"
                ),
            });
        }
    }
    Ok(())
}

/// Accounts every generated `/etc/passwd` and `/etc/group` already contains,
/// as `(account, group, id)`.
const RESERVED_ACCOUNTS: &[(&str, &str, u32)] = &[
    ("root", "root", RuntimePolicy::UID_ROOT),
    ("nobody", "nogroup", RuntimePolicy::UID_NOBODY),
];

/// The reserved account a bare `uid[:gid]` names, if it names one.
fn reserved_name(uid: u32, gid: u32) -> Option<&'static str> {
    RESERVED_ACCOUNTS
        .iter()
        .find(|(_, _, id)| uid == *id && gid == *id)
        .map(|(account, _, _)| *account)
}

/// POSIX account names are data embedded in colon-delimited system files.
/// Restrict them to the portable, non-ambiguous subset before rendering.
fn is_safe_account_name(name: &str) -> bool {
    !name.is_empty()
        && name.len() <= 32
        && name
            .bytes()
            .all(|byte| byte.is_ascii_alphanumeric() || matches!(byte, b'_' | b'-'))
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct RuntimePolicy {
    pub ca_certificates: bool,
    pub tmp: bool,
    pub passwd_group: bool,
    pub nsswitch: bool,
    pub tzdata: bool,
    /// Generated `/etc/ld.so.cache`. Not a preset choice: the planner decides
    /// from the closure unless the caller overrides it.
    pub ld_so_cache: CachePolicy,
    pub user: Option<UserSpec>,
    pub includes: Vec<PathBuf>,
}

impl Default for RuntimePolicy {
    fn default() -> RuntimePolicy {
        RuntimePolicy::from_preset(Preset::Minimal)
    }
}

impl RuntimePolicy {
    pub fn from_preset(preset: Preset) -> RuntimePolicy {
        match preset {
            Preset::Minimal => RuntimePolicy {
                ca_certificates: false,
                tmp: false,
                passwd_group: false,
                nsswitch: false,
                tzdata: false,
                ld_so_cache: CachePolicy::Auto,
                user: None,
                includes: Vec::new(),
            },
            // Pragmatic server defaults. Timezone data stays opt-in.
            Preset::Web => RuntimePolicy {
                ca_certificates: true,
                tmp: true,
                passwd_group: true,
                nsswitch: true,
                tzdata: false,
                ld_so_cache: CachePolicy::Auto,
                user: None,
                includes: Vec::new(),
            },
        }
    }

    /// Two reserved accounts every image gets, whatever `--user` says.
    pub(crate) const UID_ROOT: u32 = 0;
    pub(crate) const UID_NOBODY: u32 = 65534;

    /// Locations of a system CA bundle, most common first.
    pub const CA_BUNDLE_CANDIDATES: &'static [&'static str] = &[
        "/etc/ssl/certs/ca-certificates.crt",
        "/etc/pki/tls/certs/ca-bundle.crt",
        "/etc/ssl/ca-bundle.pem",
        "/etc/ssl/cert.pem",
        "/usr/local/share/certs/ca-root-nss.crt",
    ];

    /// NSS modules that older glibc versions `dlopen` at runtime. Since glibc
    /// 2.34 these are built into `libc.so.6`, so they are included only if the
    /// source root actually provides them.
    pub const NSS_MODULES: &'static [&'static str] =
        &["libnss_files.so.2", "libnss_dns.so.2", "libresolv.so.2"];

    /// `/etc/passwd` with root, nobody and, unless it would duplicate one of
    /// them, the requested user.
    pub fn passwd_contents(&self) -> Vec<u8> {
        let mut out = String::from("root:x:0:0:root:/root:/sbin/nologin\n");
        out.push_str("nobody:x:65534:65534:nobody:/nonexistent:/sbin/nologin\n");
        // `UserSpec` refuses any identity that would collide with the two
        // accounts above, so the only way to reach one of their ids here is by
        // asking for that account itself, which is already written.
        if let Some(user) = &self.user
            && user.uid() != Self::UID_ROOT
            && user.uid() != Self::UID_NOBODY
        {
            out.push_str(&format!(
                "{}:x:{}:{}:{}:/nonexistent:/sbin/nologin\n",
                user.name(),
                user.uid(),
                user.gid(),
                user.name()
            ));
        }
        out.into_bytes()
    }

    pub fn group_contents(&self) -> Vec<u8> {
        let mut out = String::from("root:x:0:\n");
        out.push_str("nogroup:x:65534:\n");
        if let Some(user) = &self.user
            && user.gid() != Self::UID_ROOT
            && user.gid() != Self::UID_NOBODY
        {
            out.push_str(&format!("{}:x:{}:\n", user.group(), user.gid()));
        }
        out.into_bytes()
    }

    /// `/etc/nsswitch.conf`. Without one, glibc falls back to a built-in
    /// default that does not include DNS, and the application cannot resolve.
    pub fn nsswitch_contents(&self) -> Vec<u8> {
        let mut out = String::new();
        out.push_str("# generated by elfpak\n");
        out.push_str("passwd:     files\n");
        out.push_str("group:      files\n");
        out.push_str("shadow:     files\n");
        out.push_str("hosts:      files dns\n");
        out.push_str("networks:   files\n");
        out.push_str("protocols:  files\n");
        out.push_str("services:   files\n");
        out.into_bytes()
    }
}

// A preset with no CA bundle candidates or no NSS modules to look for would be
// a policy with nothing to apply.
const _: () = assert!(!RuntimePolicy::CA_BUNDLE_CANDIDATES.is_empty());
const _: () = assert!(!RuntimePolicy::NSS_MODULES.is_empty());
const _: () = assert!(RuntimePolicy::UID_ROOT != RuntimePolicy::UID_NOBODY);

/// Allow-list of shared libraries the build is permitted to depend on.
#[derive(Debug, Clone, Default, PartialEq, Eq)]
pub struct DependencyPolicy {
    /// `None` disables the check entirely; `Some` enforces the list.
    pub allow: Option<Vec<String>>,
}

impl DependencyPolicy {
    pub fn allow_all() -> DependencyPolicy {
        DependencyPolicy { allow: None }
    }

    pub fn allow_list(list: Vec<String>) -> DependencyPolicy {
        DependencyPolicy { allow: Some(list) }
    }

    /// A library is identified by its `DT_SONAME` when present, otherwise by
    /// file name, which is what a user would write on the command line.
    pub fn is_allowed(&self, soname: &str, path: &Path) -> bool {
        let Some(allow) = &self.allow else {
            return true;
        };
        let file_name = path
            .file_name()
            .map(|n| n.to_string_lossy().into_owned())
            .unwrap_or_default();
        allow.iter().any(|a| a == soname || *a == file_name)
    }
}

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

    #[test]
    fn presets_are_explicit() {
        let minimal = RuntimePolicy::from_preset(Preset::Minimal);
        assert!(!minimal.ca_certificates && !minimal.tmp && !minimal.passwd_group);

        let web = RuntimePolicy::from_preset(Preset::Web);
        assert!(web.ca_certificates && web.tmp && web.passwd_group && web.nsswitch);
        // Timezone data stays opt-in even in the web preset.
        assert!(!web.tzdata);
    }

    #[test]
    fn user_spec_accepts_the_documented_forms() {
        assert_eq!(
            UserSpec::parse("65532").unwrap(),
            UserSpec {
                uid: 65532,
                gid: 65532,
                name: "app".into(),
                group: "app".into()
            }
        );
        assert_eq!(UserSpec::parse("65532:1000").unwrap().gid, 1000);
        assert_eq!(UserSpec::parse("svc:1:2").unwrap().name, "svc");
        assert!(UserSpec::parse("nobody").is_err());
        assert!(UserSpec::parse("evil\nroot:1:2").is_err());
        assert!(UserSpec::parse("bad:name:1:2").is_err());
    }

    #[test]
    fn passwd_and_group_include_the_requested_user() {
        let mut policy = RuntimePolicy::from_preset(Preset::Web);
        policy.user = Some(UserSpec::parse("65532:65532").unwrap());
        let passwd = String::from_utf8(policy.passwd_contents()).unwrap();
        assert!(passwd.contains("app:x:65532:65532"));
        assert!(passwd.starts_with("root:x:0:0:"));
        let group = String::from_utf8(policy.group_contents()).unwrap();
        assert!(group.contains("app:x:65532:"));
    }

    #[test]
    fn root_and_nobody_are_not_duplicated() {
        let mut policy = RuntimePolicy::from_preset(Preset::Web);
        policy.user = Some(UserSpec::parse("nobody:65534:65534").unwrap());
        let passwd = String::from_utf8(policy.passwd_contents()).unwrap();
        assert_eq!(passwd.lines().count(), 2);
    }

    /// An identity is rendered into colon- and newline-delimited system files,
    /// so the type refuses anything that could add a line of its own.
    #[test]
    fn an_account_name_cannot_carry_passwd_syntax() {
        assert!(UserSpec::new("svc:x:0:0::/root:/bin/sh\nbackdoor", 1000, 1000).is_err());
        assert!(UserSpec::new("with space", 1000, 1000).is_err());
        assert!(UserSpec::new("", 1000, 1000).is_err());
        assert!(UserSpec::new(&"a".repeat(33), 1000, 1000).is_err());
        assert!(UserSpec::new("svc-1_x", 1000, 1000).is_ok());
    }

    /// Reusing a reserved name or id would put two accounts with one name, or
    /// one id, into the generated files.
    #[test]
    fn reserved_accounts_cannot_be_redefined() {
        // Reusing a reserved name, or a reserved uid under another name.
        assert!(UserSpec::parse("root:1000:1000").is_err());
        assert!(UserSpec::parse("nobody:1000:1000").is_err());
        assert!(UserSpec::parse("nogroup:1000:1000").is_err());
        assert!(UserSpec::parse("app:0:0").is_err());
        assert!(UserSpec::parse("app:65534:65534").is_err());

        // A reserved *group* id under another name is ordinary: the group
        // already exists and the account simply joins it.
        assert_eq!(UserSpec::parse("1000:65534").unwrap().gid(), 65534);
        assert_eq!(UserSpec::parse("app:1000:0").unwrap().gid(), 0);
    }

    /// A bare `uid[:gid]` names no account, so one that matches a reserved id
    /// exactly *is* that account rather than a second one under a made-up name.
    #[test]
    fn numeric_identities_adopt_the_reserved_name_they_match() {
        assert_eq!(UserSpec::parse("65534:65534").unwrap().name(), "nobody");
        assert_eq!(UserSpec::parse("65534").unwrap().name(), "nobody");
        assert_eq!(UserSpec::parse("0").unwrap().name(), "root");
        assert_eq!(UserSpec::parse("65532:65532").unwrap().name(), "app");

        // Being a reserved account adds no second entry to either file.
        let mut policy = RuntimePolicy::from_preset(Preset::Web);
        policy.user = Some(UserSpec::parse("0").unwrap());
        let passwd = String::from_utf8(policy.passwd_contents()).unwrap();
        assert_eq!(passwd.lines().count(), 2);
        assert_eq!(
            String::from_utf8(policy.group_contents())
                .unwrap()
                .lines()
                .count(),
            2
        );
    }

    #[test]
    fn the_cache_is_written_when_the_plan_needs_one() {
        assert!(!CachePolicy::Auto.applies(false));
        assert!(CachePolicy::Auto.applies(true));
        assert!(CachePolicy::Always.applies(false));
        assert!(!CachePolicy::Never.applies(true));

        assert_eq!(CachePolicy::from_flag(None), CachePolicy::Auto);
        assert_eq!(CachePolicy::from_flag(Some(true)), CachePolicy::Always);
        assert_eq!(CachePolicy::from_flag(Some(false)), CachePolicy::Never);
        assert_eq!(RuntimePolicy::default().ld_so_cache, CachePolicy::Auto);
    }

    #[test]
    fn dependency_policy_matches_soname_or_file_name() {
        let policy = DependencyPolicy::allow_list(vec!["libc.so.6".into()]);
        assert!(policy.is_allowed("libc.so.6", Path::new("/lib/libc.so.6")));
        assert!(policy.is_allowed("", Path::new("/lib/libc.so.6")));
        assert!(!policy.is_allowed("libssl.so.3", Path::new("/lib/libssl.so.3")));
        assert!(DependencyPolicy::allow_all().is_allowed("anything.so", Path::new("/x")));
    }
}