You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
2.3 KiB
106 lines
2.3 KiB
|
|
#![allow(dead_code)]
|
|
|
|
#[cfg(feature="threads")] use rayon::prelude::*;
|
|
#[allow(unused_imports)]
|
|
use std::{
|
|
path::Path,
|
|
io, fs::{self, OpenOptions,},
|
|
convert::TryInto,
|
|
};
|
|
use smallvec::SmallVec;
|
|
use cfg_if::cfg_if;
|
|
|
|
fn usage() -> !
|
|
{
|
|
eprintln!("fcmprs: Compare files for identity");
|
|
eprintln!("Usage: {} <files...>", std::env::args().next().unwrap());
|
|
|
|
std::process::exit(-1)
|
|
}
|
|
|
|
mod error;
|
|
use error::ResultPrintExt as _;
|
|
|
|
mod map;
|
|
use map::MappedFile as _;
|
|
|
|
use error::UnmatchError;
|
|
|
|
fn main() {
|
|
let (map1, rest) = {
|
|
let mut args = std::env::args().skip(1);
|
|
if let Some(one) = args.next() {
|
|
(one, args)
|
|
} else {
|
|
usage();
|
|
}
|
|
};
|
|
|
|
std::process::exit({
|
|
if let Some(map1) = map::map(&map1).discard_msg(format!("Failed to map file {}", map1)) {
|
|
let slice = map1.as_slice();
|
|
#[cfg(feature="threads")] let map1_sz: u64 = slice.len().try_into().expect("File size could not fit into u64. This should never happen."); // For now, non-threaded mode doesn't use this.
|
|
let mut ok = true;
|
|
let chk: SmallVec<[_; 32]> = rest.filter_map(|filename| {
|
|
let path = Path::new(&filename);
|
|
if path.exists() && path.is_file() {
|
|
map::map(path).discard_msg(format!("Failed to map file {}", filename))
|
|
} else {
|
|
eprintln!("File {} does not exist or is not a normal file", filename);
|
|
ok=false;
|
|
None
|
|
}
|
|
}).collect();
|
|
|
|
if !ok {
|
|
-1
|
|
} else {
|
|
|
|
cfg_if! {
|
|
if #[cfg(feature="threads")] {
|
|
match chk.into_par_iter()
|
|
.map(|map| {
|
|
if let Ok(stat) = map.as_file().metadata() {
|
|
if stat.len() != map1_sz {
|
|
return Err(UnmatchError::Size);
|
|
}
|
|
if !stat.is_file() {
|
|
return Err(UnmatchError::Unknown);
|
|
}
|
|
}
|
|
if slice == map.as_slice() {
|
|
Ok(())
|
|
} else {
|
|
Err(UnmatchError::Data)
|
|
}
|
|
})
|
|
.try_reduce_with(|_, _| Ok(()))
|
|
{
|
|
Some(Ok(_)) => 0,
|
|
Some(Err(UnmatchError::Data)) => 1,
|
|
Some(Err(UnmatchError::Size)) => 2,
|
|
None => usage(),
|
|
_ => -1,
|
|
}
|
|
} else {
|
|
match chk.into_iter()
|
|
.map(|map| {
|
|
slice == map.as_slice()
|
|
})
|
|
.try_fold((false, true), |(_, a), b| if a && b {Ok((true, true))} else {Err(UnmatchError::Data)})
|
|
{
|
|
Ok((true, _)) => 0,
|
|
Ok((false, _)) => usage(),
|
|
Err(_) => 1,
|
|
}
|
|
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
-1
|
|
}
|
|
})
|
|
}
|