use std::{path::Path, sync::atomic::AtomicBool}; use gix_features::progress::Progress; /// pub mod checksum { /// Returned by various methods to verify the checksum of a memory mapped file that might also exist on disk. #[derive(thiserror::Error, Debug)] #[allow(missing_docs)] pub enum Error { #[error("Interrupted by user")] Interrupted, #[error("index checksum mismatch: expected {expected}, got {actual}")] Mismatch { expected: gix_hash::ObjectId, actual: gix_hash::ObjectId, }, } } /// Returns the `index` at which the following `index + 1` value is not an increment over the value at `index`. pub fn fan(data: &[u32]) -> Option { data.windows(2) .enumerate() .find_map(|(win_index, v)| (v[0] > v[1]).then_some(win_index)) } /// Calculate the hash of the given kind by trying to read the file from disk at `data_path` or falling back on the mapped content in `data`. /// `Ok(desired_hash)` or `Err(Some(actual_hash))` is returned if the hash matches or mismatches. /// If the `Err(None)` is returned, the operation was interrupted. pub fn checksum_on_disk_or_mmap( data_path: &Path, data: &[u8], expected: gix_hash::ObjectId, object_hash: gix_hash::Kind, progress: &mut dyn Progress, should_interrupt: &AtomicBool, ) -> Result { let data_len_without_trailer = data.len() - object_hash.len_in_bytes(); let actual = match gix_features::hash::bytes_of_file( data_path, data_len_without_trailer as u64, object_hash, progress, should_interrupt, ) { Ok(id) => id, Err(err) if err.kind() == std::io::ErrorKind::Interrupted => return Err(checksum::Error::Interrupted), Err(_io_err) => { let start = std::time::Instant::now(); let mut hasher = gix_features::hash::hasher(object_hash); hasher.update(&data[..data_len_without_trailer]); progress.inc_by(data_len_without_trailer); progress.show_throughput(start); gix_hash::ObjectId::from(hasher.digest()) } }; if actual == expected { Ok(actual) } else { Err(checksum::Error::Mismatch { actual, expected }) } }