maincopy-server 0.1.0

use std::{
    fmt,
    fs::File,
    io::{self, Read},
    path::{Path, PathBuf},
};

use thiserror::Error;
use zeroize::Zeroize as _;

#[derive(Clone, Copy, Debug, Eq, Error, PartialEq)]
pub(crate) enum ProtectedSecretFileError {
    #[error("the secret must be a bounded private regular file owned by this service")]
    Protection,
    #[error("the protected secret file could not be read")]
    Read,
}

/// Open the validated descriptor without following a final symlink or blocking
/// on a special file. Callers must also bound reads against concurrent growth.
#[cfg(unix)]
pub(crate) fn open_protected_secret_file(
    path: &Path,
    max_bytes: u64,
) -> Result<File, ProtectedSecretFileError> {
    use rustix::{
        fs::{Mode, OFlags, open},
        process::geteuid,
    };
    use std::os::unix::fs::MetadataExt as _;
    if !path.is_absolute() {
        return Err(ProtectedSecretFileError::Protection);
    }
    let file = File::from(
        open(
            path,
            OFlags::RDONLY | OFlags::NOFOLLOW | OFlags::NONBLOCK | OFlags::CLOEXEC,
            Mode::empty(),
        )
        .map_err(|_| ProtectedSecretFileError::Read)?,
    );
    let metadata = file
        .metadata()
        .map_err(|_| ProtectedSecretFileError::Read)?;
    if !metadata.is_file()
        || metadata.uid() != geteuid().as_raw()
        || !matches!(metadata.mode() & 0o7777, 0o400 | 0o600)
        || metadata.len() > max_bytes
    {
        return Err(ProtectedSecretFileError::Protection);
    }
    Ok(file)
}

#[cfg(not(unix))]
pub(crate) fn open_protected_secret_file(
    _path: &Path,
    _max_bytes: u64,
) -> Result<File, ProtectedSecretFileError> {
    Err(ProtectedSecretFileError::Protection)
}

const MAX_RESOLVED_SECRET_BYTES: usize = 64 * 1024;
const RESOLVED_SECRET_BUFFER_BYTES: usize = MAX_RESOLVED_SECRET_BYTES + 1;
const RESOLVED_SECRET_TOO_LARGE: &str = "resolved secret exceeds the inclusive 64 KiB hard limit";

macro_rules! redacted_path_type {
    ($(#[$attribute:meta])* $name:ident, $debug:literal, $display:literal) => {
        $(#[$attribute])*
        #[derive(Clone, Eq, PartialEq)]
        pub struct $name(PathBuf);

        impl $name {
            pub fn new(path: PathBuf) -> Option<Self> {
                (!path.as_os_str().is_empty()).then_some(Self(path))
            }

            pub fn path(&self) -> &Path {
                &self.0
            }
        }

        impl fmt::Debug for $name {
            fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
                formatter.write_str($debug)
            }
        }

        impl fmt::Display for $name {
            fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
                formatter.write_str($display)
            }
        }
    };
}

redacted_path_type!(
    /// A redacted reference to a file that contains protected bytes.
    SecretFileReference,
    "SecretFileReference(<redacted>)",
    "<redacted-secret-file-reference>"
);

/// Bytes in one fixed allocation that is wiped before deallocation.
///
/// This type intentionally has no cloning, serialization, dereference, slice
/// conversion, or inner-value extraction API. A consuming callback is the only
/// way to inspect the initialized bytes.
struct ResolvedSecret {
    storage: Box<[u8]>,
    len: usize,
    #[cfg(test)]
    drop_probe: Option<SecretDropProbe>,
}

impl ResolvedSecret {
    /// Reads at most 64 KiB directly into one fixed allocation.
    ///
    /// The extra byte is a sentinel that detects input beyond the inclusive
    /// hard limit. The allocation already belongs to `ResolvedSecret` while
    /// reads occur, so read failures also take the zeroizing drop path.
    fn read_from(reader: &mut impl Read) -> io::Result<Self> {
        Self::empty().fill_from(reader)
    }

    /// Gives one callback a scoped view and wipes the allocation afterwards.
    fn expose_to<Output>(
        self,
        use_secret: impl for<'secret> FnOnce(&'secret [u8]) -> Output,
    ) -> Output {
        use_secret(&self.storage[..self.len])
    }

    fn empty() -> Self {
        Self {
            // This vector contains only zeros. Secret bytes enter storage only
            // after it becomes a fixed-size boxed slice that cannot reallocate.
            storage: vec![0; RESOLVED_SECRET_BUFFER_BYTES].into_boxed_slice(),
            len: 0,
            #[cfg(test)]
            drop_probe: None,
        }
    }

    fn fill_from(mut self, reader: &mut impl Read) -> io::Result<Self> {
        loop {
            match reader.read(&mut self.storage[self.len..]) {
                Ok(0) => return Ok(self),
                Ok(read) => {
                    self.len += read;
                    if self.len > MAX_RESOLVED_SECRET_BYTES {
                        return Err(io::Error::new(
                            io::ErrorKind::InvalidData,
                            RESOLVED_SECRET_TOO_LARGE,
                        ));
                    }
                }
                Err(error) if error.kind() == io::ErrorKind::Interrupted => {}
                Err(error) => return Err(error),
            }
        }
    }

    #[cfg(test)]
    fn read_from_with_probe(
        reader: &mut impl Read,
        drop_probe: SecretDropProbe,
    ) -> io::Result<Self> {
        let mut secret = Self::empty();
        secret.drop_probe = Some(drop_probe);
        secret.fill_from(reader)
    }
}

impl Drop for ResolvedSecret {
    fn drop(&mut self) {
        self.storage.zeroize();
        self.len = 0;

        #[cfg(test)]
        if let Some(probe) = &self.drop_probe {
            probe.store(
                self.storage.iter().all(|byte| *byte == 0),
                std::sync::atomic::Ordering::SeqCst,
            );
        }
    }
}

/// Reads from an already secured source and gives one callback a scoped view.
///
/// The consuming composition boundary must open and validate the credential file
/// before it calls this boundary. This function does not define file-opening
/// policy.
pub(crate) fn with_resolved_secret<Output>(
    reader: &mut impl Read,
    use_secret: impl for<'secret> FnOnce(&'secret [u8]) -> Output,
) -> io::Result<Output> {
    let secret = ResolvedSecret::read_from(reader)?;
    Ok(secret.expose_to(use_secret))
}

redacted_path_type!(
    /// A path whose existence can reveal protected runtime metadata.
    SensitivePath,
    "SensitivePath(<redacted>)",
    "<redacted-sensitive-path>"
);

#[cfg(test)]
type SecretDropProbe = std::sync::Arc<std::sync::atomic::AtomicBool>;

#[cfg(test)]
mod tests {
    use std::{
        borrow::Borrow,
        io::Cursor,
        ops::Deref,
        panic::{AssertUnwindSafe, catch_unwind},
    };

    use serde::{Serialize, de::DeserializeOwned};

    use super::*;

    macro_rules! assert_not_impl {
        ($value:ty: $bound:path) => {
            const _: fn() = || {
                trait AmbiguousIfImpl<Marker> {
                    fn marker() {}
                }

                impl<Value: ?Sized> AmbiguousIfImpl<()> for Value {}
                impl<Value: ?Sized + $bound> AmbiguousIfImpl<u8> for Value {}

                let _ = <$value as AmbiguousIfImpl<_>>::marker;
            };
        };
    }

    assert_not_impl!(ResolvedSecret: Copy);
    assert_not_impl!(ResolvedSecret: Clone);
    assert_not_impl!(ResolvedSecret: Serialize);
    assert_not_impl!(ResolvedSecret: DeserializeOwned);
    assert_not_impl!(ResolvedSecret: Eq);
    assert_not_impl!(ResolvedSecret: PartialEq);
    assert_not_impl!(ResolvedSecret: Deref);
    assert_not_impl!(ResolvedSecret: AsRef<[u8]>);
    assert_not_impl!(ResolvedSecret: Borrow<[u8]>);

    struct FailingReader;

    impl Read for FailingReader {
        fn read(&mut self, _output: &mut [u8]) -> io::Result<usize> {
            Err(io::Error::other("synthetic read failure"))
        }
    }

    fn probed_secret() -> (ResolvedSecret, SecretDropProbe) {
        let probe = SecretDropProbe::default();
        let mut reader = io::repeat(0xa5).take(MAX_RESOLVED_SECRET_BYTES as u64);
        let secret = ResolvedSecret::read_from_with_probe(&mut reader, probe.clone()).unwrap();
        (secret, probe)
    }

    fn assert_zeroized_after_drop(probe: &SecretDropProbe) {
        assert!(probe.load(std::sync::atomic::Ordering::SeqCst));
    }

    #[test]
    fn secret_and_sensitive_paths_are_redacted() {
        let file = SecretFileReference::new(PathBuf::from("/secret/credential.json")).unwrap();
        let cache = SensitivePath::new(PathBuf::from("/secret/private-cache")).unwrap();

        let rendered = format!("{file:?} {file} {cache:?} {cache}");
        for protected in ["/secret", "credential.json", "private-cache"] {
            assert!(!rendered.contains(protected));
        }
    }

    #[test]
    fn resolved_secret_boundary_exposes_only_a_scoped_borrow() {
        let length =
            with_resolved_secret(&mut Cursor::new(b"protected"), |bytes| bytes.len()).unwrap();

        assert_eq!(length, 9);
    }

    #[test]
    fn normal_callback_return_zeroizes_the_complete_allocation() {
        let (secret, probe) = probed_secret();
        assert!(!probe.load(std::sync::atomic::Ordering::SeqCst));

        let length = secret.expose_to(|bytes| bytes.len());

        assert_eq!(length, MAX_RESOLVED_SECRET_BYTES);
        assert_zeroized_after_drop(&probe);
    }

    #[test]
    fn ordinary_drop_zeroizes_the_complete_allocation() {
        let (secret, probe) = probed_secret();

        drop(secret);

        assert_zeroized_after_drop(&probe);
    }

    #[test]
    fn callback_error_zeroizes_the_complete_allocation() {
        let (secret, probe) = probed_secret();

        let result: Result<(), &'static str> = secret.expose_to(|_| Err("synthetic parse error"));

        assert_eq!(result, Err("synthetic parse error"));
        assert_zeroized_after_drop(&probe);
    }

    #[test]
    fn callback_panic_zeroizes_the_complete_allocation_during_unwind() {
        let (secret, probe) = probed_secret();

        let result = catch_unwind(AssertUnwindSafe(|| {
            secret.expose_to::<()>(|_| panic!("synthetic callback panic"));
        }));

        assert!(result.is_err());
        assert_zeroized_after_drop(&probe);
    }

    #[test]
    fn partial_read_error_zeroizes_initialized_bytes_before_deallocation() {
        let probe = SecretDropProbe::default();
        let mut reader = Cursor::new([0xa5; 32]).chain(FailingReader);
        let error = ResolvedSecret::read_from_with_probe(&mut reader, probe.clone())
            .err()
            .unwrap();

        assert_eq!(error.kind(), io::ErrorKind::Other);
        assert_zeroized_after_drop(&probe);
    }

    #[test]
    fn inclusive_hard_limit_is_accepted() {
        let mut reader = io::repeat(0xa5).take(MAX_RESOLVED_SECRET_BYTES as u64);
        let length = ResolvedSecret::read_from(&mut reader)
            .unwrap()
            .expose_to(|bytes| bytes.len());
        assert_eq!(length, MAX_RESOLVED_SECRET_BYTES);
    }

    #[test]
    fn sentinel_rejects_one_byte_over_limit_and_zeroizes_it() {
        let probe = SecretDropProbe::default();
        let mut reader = io::repeat(0xa5).take(RESOLVED_SECRET_BUFFER_BYTES as u64);

        let error = ResolvedSecret::read_from_with_probe(&mut reader, probe.clone())
            .err()
            .unwrap();

        assert_eq!(error.kind(), io::ErrorKind::InvalidData);
        assert_eq!(error.to_string(), RESOLVED_SECRET_TOO_LARGE);
        assert_zeroized_after_drop(&probe);
    }
}