tailscale-cli 1.3.1

//! Handing the CLI something too big or too private for an argument list.
//!
//! An argument list is world-readable on every platform we support: `ps` shows
//! it, `/proc` shows it, and a crash reporter will happily upload it. Tailscale
//! anticipates this and accepts `file:<path>` wherever it accepts a key, so the
//! secret goes into a file only this process can read, for only as long as the
//! call takes. That is [`SecretFile`].
//!
//! [`PrivateFile`] is the other direction: a document the CLI insists on
//! exchanging through a file rather than a stream, in either direction. It
//! protects the *directory* rather than the file, because a file the CLI
//! creates is created with the CLI's idea of a mode, not ours.

use std::io::Write;
use std::path::Path;

use crate::exec::ExecError;

/// A private file holding one secret, removed when this value is dropped.
///
/// Dropping is enough for the ordinary paths, including a timeout: the guard
/// lives in the same scope as the call, so unwinding or an early return takes
/// the file with it. A hard kill of the server leaves it behind, which is why
/// it is created under the system temporary directory with a private mode
/// rather than somewhere durable.
#[derive(Debug)]
pub struct SecretFile {
    file: tempfile::NamedTempFile,
}

impl SecretFile {
    /// Write `secret` to a new private file.
    pub fn new(secret: &str) -> Result<Self, ExecError> {
        let mut builder = tempfile::Builder::new();
        builder.prefix("tailscale-mcp-").suffix(".key");
        #[cfg(unix)]
        {
            use std::os::unix::fs::PermissionsExt as _;
            let _ = std::fs::Permissions::from_mode(0o600);
            builder.permissions(std::fs::Permissions::from_mode(0o600));
        }
        let mut file = builder.tempfile().map_err(ExecError::SecretFile)?;
        file.write_all(secret.as_bytes())
            .and_then(|()| file.flush())
            .map_err(ExecError::SecretFile)?;
        Ok(Self { file })
    }

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

    /// The argument value the CLI expects: a path behind the `file:` scheme.
    pub fn arg(&self) -> String {
        format!("file:{}", self.file.path().display())
    }
}

/// A scratch file inside a directory only this user can enter, removed when
/// this value is dropped along with the directory holding it.
///
/// Used for the two `serve` commands that exchange configuration through a
/// file. The file itself may not exist: `serve get-config` refuses a path that
/// is already taken, so the reserved form hands over a name and lets the client
/// create it.
#[derive(Debug)]
pub struct PrivateFile {
    // Held, not read: dropping the directory is what removes whatever is
    // inside it, the client's file included.
    #[expect(dead_code, reason = "kept alive so that dropping it cleans up")]
    dir: tempfile::TempDir,
    path: std::path::PathBuf,
}

impl PrivateFile {
    /// A name inside a new private directory, with nothing at it yet.
    pub fn reserved(name: &str) -> Result<Self, ExecError> {
        let mut builder = tempfile::Builder::new();
        builder.prefix("tailscale-mcp-");
        #[cfg(unix)]
        {
            use std::os::unix::fs::PermissionsExt as _;
            builder.permissions(std::fs::Permissions::from_mode(0o700));
        }
        let dir = builder.tempdir().map_err(ExecError::SecretFile)?;
        let path = dir.path().join(name);
        Ok(Self { dir, path })
    }

    /// The same, with `contents` already written to it.
    pub fn written(name: &str, contents: &[u8]) -> Result<Self, ExecError> {
        let file = Self::reserved(name)?;
        std::fs::write(&file.path, contents).map_err(ExecError::SecretFile)?;
        Ok(file)
    }

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

    /// The path as the CLI wants it: a plain path, since these commands take a
    /// filename rather than a URL.
    pub fn arg(&self) -> String {
        self.path.display().to_string()
    }

    /// What is at the path now. An error if the client wrote nothing.
    pub fn read(&self) -> Result<Vec<u8>, ExecError> {
        std::fs::read(&self.path).map_err(ExecError::SecretFile)
    }

    /// The directory the file lives in, for the tests that check it is private.
    #[cfg(test)]
    fn directory(&self) -> &Path {
        self.path.parent().unwrap_or(&self.path)
    }
}

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

    #[test]
    fn a_reserved_name_is_free_for_the_client_to_create() {
        let file = PrivateFile::reserved("serve-config.json").expect("a private directory");
        assert!(
            !file.path().exists(),
            "the client refuses a path that is already taken"
        );
        std::fs::write(file.path(), b"{}").expect("the client can create it");
        assert_eq!(file.read().expect("readable"), b"{}");
    }

    #[cfg(unix)]
    #[test]
    fn nobody_else_can_enter_the_directory_the_file_lives_in() {
        use std::os::unix::fs::PermissionsExt as _;
        let file = PrivateFile::written("serve-config.json", b"{}").expect("a private directory");
        let mode = std::fs::metadata(file.directory())
            .expect("metadata")
            .permissions()
            .mode();
        assert_eq!(mode & 0o077, 0, "group and other must have no access");
    }

    #[test]
    fn the_file_and_its_directory_go_away_with_the_call() {
        let (path, dir) = {
            let file = PrivateFile::written("serve-config.json", b"{}").expect("a private file");
            (file.path().to_path_buf(), file.directory().to_path_buf())
        };
        assert!(!path.exists(), "{} outlived its guard", path.display());
        assert!(!dir.exists(), "{} outlived its guard", dir.display());
    }

    #[test]
    fn the_secret_reaches_the_cli_by_reference_not_by_value() {
        let secret = "tskey-auth-example-notreal";
        let file = SecretFile::new(secret).expect("a temporary file");
        let arg = file.arg();
        assert!(arg.starts_with("file:"));
        assert!(
            !arg.contains(secret),
            "the argument must not carry the secret: {arg}"
        );
        assert_eq!(
            std::fs::read_to_string(file.path()).expect("readable"),
            secret
        );
    }

    #[cfg(unix)]
    #[test]
    fn the_file_is_readable_only_by_this_user() {
        use std::os::unix::fs::PermissionsExt as _;
        let file = SecretFile::new("tskey-auth-example-notreal").expect("a temporary file");
        let mode = std::fs::metadata(file.path())
            .expect("metadata")
            .permissions()
            .mode();
        assert_eq!(mode & 0o077, 0, "group and other must have no access");
    }

    #[test]
    fn the_file_goes_away_with_the_call() {
        let path = {
            let file = SecretFile::new("tskey-auth-example-notreal").expect("a temporary file");
            file.path().to_path_buf()
        };
        assert!(!path.exists(), "{} outlived its guard", path.display());
    }
}