use sha2::{Digest, Sha256};
use std::{
collections::BTreeSet,
ffi::OsStr,
fmt::Write as _,
fs::{self, File, OpenOptions},
io::{self, Read as _, Write as _},
path::{Path, PathBuf},
sync::atomic::{AtomicU64, Ordering},
};
static TEMP_FILE_SEQUENCE: AtomicU64 = AtomicU64::new(0);
#[derive(Debug)]
struct AtomicCopyErrorContext {
source_path: PathBuf,
destination_path: PathBuf,
source: io::Error,
}
impl std::fmt::Display for AtomicCopyErrorContext {
fn fmt(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(
formatter,
"failed to atomically copy {} to {}: {}",
self.source_path.display(),
self.destination_path.display(),
self.source
)
}
}
impl std::error::Error for AtomicCopyErrorContext {
fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
Some(&self.source)
}
}
#[derive(Clone, Copy, Debug, Eq, Hash, Ord, PartialEq, PartialOrd)]
pub struct InputDigest([u8; 32]);
impl InputDigest {
#[must_use]
pub const fn as_bytes(&self) -> &[u8; 32] {
&self.0
}
#[must_use]
pub fn to_hex(self) -> String {
let mut hex = String::with_capacity(64);
for byte in self.0 {
write!(hex, "{byte:02x}").expect("writing to a String cannot fail");
}
hex
}
}
impl std::fmt::Display for InputDigest {
fn fmt(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
formatter.write_str(&self.to_hex())
}
}
pub(super) struct InputHasher(Sha256);
impl InputHasher {
pub(super) fn new(domain: &str) -> Self {
let mut hasher = Self(Sha256::new());
hasher.field("domain", domain.as_bytes());
hasher
}
pub(super) fn field(&mut self, label: &str, value: &[u8]) {
self.field_header(
label,
u64::try_from(value.len()).expect("input value length must fit in u64"),
);
self.0.update(value);
}
fn field_header(&mut self, label: &str, value_len: u64) {
self.0.update(
u64::try_from(label.len())
.expect("input label length must fit in u64")
.to_le_bytes(),
);
self.0.update(label.as_bytes());
self.0.update(value_len.to_le_bytes());
}
fn file_field(&mut self, label: &str, path: &Path) -> io::Result<u64> {
let mut file = File::open(path)?;
let expected_len = file.metadata()?.len();
self.field_header(label, expected_len);
let mut actual_len = 0_u64;
let mut buffer = [0_u8; 16 * 1024];
loop {
let read = file.read(&mut buffer)?;
if read == 0 {
break;
}
actual_len = actual_len
.saturating_add(u64::try_from(read).expect("artifact read length must fit in u64"));
self.0.update(&buffer[..read]);
}
if actual_len != expected_len {
return Err(io::Error::new(
io::ErrorKind::InvalidData,
format!(
"file changed size while hashing: expected {expected_len} bytes, read {actual_len}"
),
));
}
Ok(actual_len)
}
pub(super) fn finish(self) -> InputDigest {
InputDigest(self.0.finalize().into())
}
}
pub(super) fn digest_bytes(domain: &str, value: &[u8]) -> InputDigest {
let mut hasher = InputHasher::new(domain);
hasher.field("content", value);
hasher.finish()
}
pub(super) fn digest_file(domain: &str, path: &Path) -> io::Result<(u64, InputDigest)> {
let mut hasher = InputHasher::new(domain);
let bytes = hasher.file_field("content", path)?;
Ok((bytes, hasher.finish()))
}
pub(super) fn digest_labeled_paths(
domain: &str,
paths: &[(PathBuf, PathBuf)],
excluded_roots: &[PathBuf],
) -> io::Result<InputDigest> {
let mut paths = paths.to_vec();
paths.sort_by(|(left, _), (right, _)| {
os_bytes(left.as_os_str()).cmp(&os_bytes(right.as_os_str()))
});
let excluded_roots = excluded_roots
.iter()
.filter_map(|path| path.canonicalize().ok())
.collect::<Vec<_>>();
let mut visited_directories = BTreeSet::new();
let mut hasher = InputHasher::new(domain);
for (label, path) in paths {
hash_path(
&mut hasher,
&label,
&path,
&excluded_roots,
&mut visited_directories,
true,
)?;
}
Ok(hasher.finish())
}
fn hash_path(
hasher: &mut InputHasher,
label: &Path,
path: &Path,
excluded_roots: &[PathBuf],
visited_directories: &mut BTreeSet<PathBuf>,
declared_root: bool,
) -> io::Result<()> {
let canonical = path.canonicalize()?;
if excluded_roots
.iter()
.any(|excluded| canonical.starts_with(excluded))
{
if declared_root {
return Err(io::Error::new(
io::ErrorKind::InvalidInput,
format!(
"declared input is located inside an excluded cache root: {}",
path.display()
),
));
}
return Ok(());
}
let metadata = fs::metadata(path)?;
let label_bytes = os_bytes(label.as_os_str());
if metadata.is_file() {
hasher.field("file-path", &label_bytes);
hasher.file_field("file-content", path)?;
return Ok(());
}
if !metadata.is_dir() {
return Err(io::Error::new(
io::ErrorKind::InvalidInput,
format!(
"watched input is not a regular file or directory: {}",
path.display()
),
));
}
hasher.field("directory", &label_bytes);
if !visited_directories.insert(canonical) {
hasher.field("directory-already-visited", &label_bytes);
return Ok(());
}
let mut entries = fs::read_dir(path)?.collect::<Result<Vec<_>, _>>()?;
entries.sort_by_key(|entry| os_bytes(&entry.file_name()));
for entry in entries {
hash_path(
hasher,
&label.join(entry.file_name()),
&entry.path(),
excluded_roots,
visited_directories,
false,
)?;
}
Ok(())
}
pub(super) fn write_atomic(path: &Path, contents: &[u8]) -> io::Result<()> {
write_file_atomic(path, |file| file.write_all(contents))
}
pub(super) fn copy_file_atomic(source: &Path, destination: &Path) -> io::Result<u64> {
let result = (|| {
let mut source_file = File::open(source)?;
write_file_atomic(destination, |destination_file| {
io::copy(&mut source_file, destination_file)
})
})();
result.map_err(|source_error| {
io::Error::new(
source_error.kind(),
AtomicCopyErrorContext {
source_path: source.to_owned(),
destination_path: destination.to_owned(),
source: source_error,
},
)
})
}
fn write_file_atomic<T>(
path: &Path,
write: impl FnOnce(&mut File) -> io::Result<T>,
) -> io::Result<T> {
let parent = path.parent().ok_or_else(|| {
io::Error::new(
io::ErrorKind::InvalidInput,
format!("atomic output path has no parent: {}", path.display()),
)
})?;
fs::create_dir_all(parent)?;
let file_name = path.file_name().ok_or_else(|| {
io::Error::new(
io::ErrorKind::InvalidInput,
format!("atomic output path has no file name: {}", path.display()),
)
})?;
let sequence = TEMP_FILE_SEQUENCE.fetch_add(1, Ordering::Relaxed);
let mut temp_name = file_name.to_os_string();
temp_name.push(format!(".tmp-{}-{sequence}", std::process::id()));
let temp_path = parent.join(temp_name);
let result = (|| {
let mut file = OpenOptions::new()
.create_new(true)
.write(true)
.open(&temp_path)?;
let value = write(&mut file)?;
file.sync_all()?;
fs::rename(&temp_path, path)?;
Ok(value)
})();
if result.is_err() {
let _ = fs::remove_file(&temp_path);
}
result
}
#[cfg(unix)]
pub(super) fn os_bytes(value: &OsStr) -> Vec<u8> {
use std::os::unix::ffi::OsStrExt as _;
value.as_bytes().to_vec()
}
#[cfg(windows)]
pub(super) fn os_bytes(value: &OsStr) -> Vec<u8> {
use std::os::windows::ffi::OsStrExt as _;
value
.encode_wide()
.flat_map(u16::to_le_bytes)
.collect::<Vec<_>>()
}
#[cfg(not(any(unix, windows)))]
pub(super) fn os_bytes(value: &OsStr) -> Vec<u8> {
value.to_string_lossy().as_bytes().to_vec()
}
#[cfg(test)]
mod tests {
use super::{copy_file_atomic, digest_bytes, digest_file, write_atomic};
use crate::artifacts::test_support::unique_temp_directory;
use std::fs;
#[test]
fn streaming_digest_and_atomic_copy_preserve_exact_bytes() {
let root = unique_temp_directory("streaming-digest");
let source = root.join("source");
let destination = root.join("destination");
let mut contents = vec![0_u8; 192 * 1024];
for (index, byte) in contents.iter_mut().enumerate() {
*byte = u8::try_from(index % 251).expect("test byte must fit");
}
fs::write(&source, &contents).expect("write source");
let (bytes, streamed) = digest_file("streaming-test-v1", &source).expect("digest file");
assert_eq!(bytes, u64::try_from(contents.len()).unwrap());
assert_eq!(streamed, digest_bytes("streaming-test-v1", &contents));
write_atomic(&destination, b"old").expect("write original destination");
assert_eq!(
copy_file_atomic(&source, &destination).expect("copy source atomically"),
bytes
);
assert_eq!(fs::read(&destination).unwrap(), contents);
let missing = root.join("missing");
let error = copy_file_atomic(&missing, &destination).expect_err("missing source must fail");
let message = error.to_string();
assert!(message.contains(&missing.display().to_string()));
assert!(message.contains(&destination.display().to_string()));
fs::remove_dir_all(root).expect("remove streaming-digest test directory");
}
}