common/

preserve.rs

use anyhow::{Context, Result};
use serde::{Deserialize, Serialize};
use std::os::unix::fs::MetadataExt;
use std::os::unix::prelude::PermissionsExt;
use tracing::instrument;

pub trait Metadata {
    fn uid(&self) -> u32;
    fn gid(&self) -> u32;
    fn atime(&self) -> i64;
    fn atime_nsec(&self) -> i64;
    fn mtime(&self) -> i64;
    fn mtime_nsec(&self) -> i64;
    fn permissions(&self) -> std::fs::Permissions;
    // ctime cannot be set manually, but we include it for comparison purposes
    // default implementation returns 0 to indicate ctime is not available (e.g., in protocol::Metadata)
    fn ctime(&self) -> i64 {
        0
    }
    fn ctime_nsec(&self) -> i64 {
        0
    }
    // size is not preserved (cannot be set), but included for comparison purposes
    // default implementation returns 0 to indicate size is not available or not applicable
    fn size(&self) -> u64 {
        0
    }
}

impl Metadata for std::fs::Metadata {
    fn uid(&self) -> u32 {
        MetadataExt::uid(self)
    }
    fn gid(&self) -> u32 {
        MetadataExt::gid(self)
    }
    fn atime(&self) -> i64 {
        MetadataExt::atime(self)
    }
    fn atime_nsec(&self) -> i64 {
        MetadataExt::atime_nsec(self)
    }
    fn mtime(&self) -> i64 {
        MetadataExt::mtime(self)
    }
    fn mtime_nsec(&self) -> i64 {
        MetadataExt::mtime_nsec(self)
    }
    fn permissions(&self) -> std::fs::Permissions {
        self.permissions()
    }
    fn ctime(&self) -> i64 {
        MetadataExt::ctime(self)
    }
    fn ctime_nsec(&self) -> i64 {
        MetadataExt::ctime_nsec(self)
    }
    fn size(&self) -> u64 {
        self.len()
    }
}

#[derive(Copy, Clone, Debug, Default, Deserialize, Serialize)]
pub struct UserAndTimeSettings {
    pub uid: bool,
    pub gid: bool,
    pub time: bool,
}

impl UserAndTimeSettings {
    #[must_use]
    pub fn any(&self) -> bool {
        self.uid || self.gid || self.time
    }
}

pub type ModeMask = u32;

#[derive(Copy, Clone, Debug, Deserialize, Serialize)]
pub struct FileSettings {
    pub user_and_time: UserAndTimeSettings,
    pub mode_mask: ModeMask,
}

impl Default for FileSettings {
    fn default() -> Self {
        Self {
            user_and_time: UserAndTimeSettings::default(),
            mode_mask: 0o0777, // remove sticky bit, setuid and setgid to mimic "cp" tool
        }
    }
}

#[derive(Copy, Clone, Debug, Deserialize, Serialize)]
pub struct DirSettings {
    pub user_and_time: UserAndTimeSettings,
    pub mode_mask: ModeMask,
}

impl Default for DirSettings {
    fn default() -> Self {
        Self {
            user_and_time: UserAndTimeSettings::default(),
            mode_mask: 0o0777,
        }
    }
}

#[derive(Copy, Clone, Debug, Default, Deserialize, Serialize)]
pub struct SymlinkSettings {
    pub user_and_time: UserAndTimeSettings,
}

impl SymlinkSettings {
    #[must_use]
    pub fn any(&self) -> bool {
        self.user_and_time.any()
    }
}

#[derive(Copy, Clone, Debug, Default, Deserialize, Serialize)]
pub struct Settings {
    pub file: FileSettings,
    pub dir: DirSettings,
    pub symlink: SymlinkSettings,
}

#[instrument]
async fn set_owner<Meta: Metadata + std::fmt::Debug>(
    settings: &UserAndTimeSettings,
    path: &std::path::Path,
    metadata: &Meta,
) -> Result<()> {
    if !settings.uid && !settings.gid {
        return Ok(());
    }
    let settings = settings.to_owned();
    let dst = path.to_owned();
    let uid = metadata.uid();
    let gid = metadata.gid();
    crate::walk::run_metadata_probed(
        congestion::Side::Destination,
        congestion::MetadataOp::Chmod,
        async {
            tokio::task::spawn_blocking(move || -> Result<()> {
                tracing::debug!("setting uid and gid");
                let uid_val = if settings.uid { Some(uid.into()) } else { None };
                let gid_val = if settings.gid { Some(gid.into()) } else { None };
                nix::unistd::fchownat(
                    nix::fcntl::AT_FDCWD,
                    &dst,
                    uid_val,
                    gid_val,
                    nix::fcntl::AtFlags::AT_SYMLINK_NOFOLLOW,
                )
                .with_context(|| {
                    format!(
                        "cannot set {:?} owner to {:?} and/or group id to {:?}",
                        &dst, &uid_val, &gid_val
                    )
                })?;
                Ok(())
            })
            .await?
        },
    )
    .await
}

#[instrument]
async fn set_time<Meta: Metadata + std::fmt::Debug>(
    settings: &UserAndTimeSettings,
    path: &std::path::Path,
    metadata: &Meta,
) -> Result<()> {
    if !settings.time {
        return Ok(());
    }
    let dst = path.to_owned();
    let atime = metadata.atime();
    let atime_nsec = metadata.atime_nsec();
    let mtime = metadata.mtime();
    let mtime_nsec = metadata.mtime_nsec();
    crate::walk::run_metadata_probed(
        congestion::Side::Destination,
        congestion::MetadataOp::Chmod,
        async {
            tokio::task::spawn_blocking(move || -> Result<()> {
                tracing::debug!("setting timestamps");
                let atime_spec = nix::sys::time::TimeSpec::new(atime, atime_nsec);
                let mtime_spec = nix::sys::time::TimeSpec::new(mtime, mtime_nsec);
                nix::sys::stat::utimensat(
                    nix::fcntl::AT_FDCWD,
                    &dst,
                    &atime_spec,
                    &mtime_spec,
                    nix::sys::stat::UtimensatFlags::NoFollowSymlink,
                )
                .with_context(|| format!("failed setting timestamps for {:?}", &dst))?;
                Ok(())
            })
            .await?
        },
    )
    .await
}

pub async fn set_file_metadata<Meta: Metadata + std::fmt::Debug>(
    settings: &Settings,
    metadata: &Meta,
    path: &std::path::Path,
) -> Result<()> {
    let permissions = if settings.file.mode_mask == 0o7777 {
        // special case for default preserve
        metadata.permissions()
    } else {
        std::fs::Permissions::from_mode(metadata.permissions().mode() & settings.file.mode_mask)
    };
    // ordering: chown → chmod → utimensat
    //
    // chown first because fchownat clears setuid/setgid on regular files;
    // chmod afterwards restores them. utimensat last because both chown and
    // chmod update ctime and may touch mtime, so we set the desired
    // timestamps as the final step.
    //
    // if chown fails (e.g. EPERM when not root), we bail out early rather
    // than applying permissions for an unverified owner — setting setuid on
    // a file whose ownership we couldn't control would be a security risk.
    set_owner(&settings.file.user_and_time, path, metadata).await?;
    let file = crate::walk::run_metadata_probed(
        congestion::Side::Destination,
        congestion::MetadataOp::Stat,
        tokio::fs::File::open(path),
    )
    .await?;
    crate::walk::run_metadata_probed(
        congestion::Side::Destination,
        congestion::MetadataOp::Chmod,
        file.set_permissions(permissions.clone()),
    )
    .await
    .with_context(|| format!("cannot set {:?} permissions to {:?}", &path, &permissions))?;
    drop(file);
    set_time(&settings.file.user_and_time, path, metadata).await?;
    Ok(())
}

pub async fn set_dir_metadata<Meta: Metadata + std::fmt::Debug>(
    settings: &Settings,
    metadata: &Meta,
    path: &std::path::Path,
) -> Result<()> {
    let permissions = if settings.dir.mode_mask == 0o7777 {
        // special case for default preserve
        metadata.permissions()
    } else {
        std::fs::Permissions::from_mode(metadata.permissions().mode() & settings.dir.mode_mask)
    };
    // same ordering as set_file_metadata: chown → chmod → utimensat.
    // see that function for rationale.
    set_owner(&settings.dir.user_and_time, path, metadata).await?;
    crate::walk::run_metadata_probed(
        congestion::Side::Destination,
        congestion::MetadataOp::Chmod,
        tokio::fs::set_permissions(path, permissions.clone()),
    )
    .await
    .with_context(|| format!("cannot set {:?} permissions to {:?}", &path, &permissions))?;
    set_time(&settings.dir.user_and_time, path, metadata).await?;
    Ok(())
}

pub async fn set_symlink_metadata<Meta: Metadata + std::fmt::Debug>(
    settings: &Settings,
    metadata: &Meta,
    path: &std::path::Path,
) -> Result<()> {
    // we don't set permissions for symlinks, only owner and time
    set_owner(&settings.symlink.user_and_time, path, metadata).await?;
    set_time(&settings.symlink.user_and_time, path, metadata).await?;
    Ok(())
}

#[must_use]
pub fn preserve_all() -> Settings {
    let user_and_time = UserAndTimeSettings {
        uid: true,
        gid: true,
        time: true,
    };

    Settings {
        file: FileSettings {
            user_and_time,
            mode_mask: 0o7777,
        },
        dir: DirSettings {
            user_and_time,
            mode_mask: 0o7777,
        },
        symlink: SymlinkSettings { user_and_time },
    }
}

#[must_use]
pub fn preserve_none() -> Settings {
    Settings::default()
}