replace sha2 with blake2 for duplicates
This commit is contained in:
parent
552ced69e1
commit
6966b141da
@ -1,13 +1,13 @@
|
||||
[package]
|
||||
name = "du-rust"
|
||||
version = "0.1.0"
|
||||
version = "0.1.1"
|
||||
edition = "2021"
|
||||
authors = ["Sam Hadow"]
|
||||
|
||||
[dependencies]
|
||||
blake2 = "0.10.6"
|
||||
clap = { version = "4.4.6", features = ["derive"] }
|
||||
rayon = "1.8.0"
|
||||
sha2 = "0.10.8"
|
||||
|
||||
[profile.release]
|
||||
strip = true
|
||||
|
2
PKGBUILD
2
PKGBUILD
@ -2,7 +2,7 @@
|
||||
|
||||
_pkgname=disk-usage
|
||||
pkgname=du-rust
|
||||
pkgver=0.1.0
|
||||
pkgver=0.1.1
|
||||
pkgrel=1
|
||||
pkgdesc="Disk usage tool in rust."
|
||||
arch=('x86_64')
|
||||
|
@ -8,7 +8,7 @@ use std::fs;
|
||||
use std::path::{Path, PathBuf};
|
||||
|
||||
// hash
|
||||
use sha2::{Digest, Sha256};
|
||||
use blake2::{Blake2b512, Digest};
|
||||
use std::io::{BufReader, Read};
|
||||
|
||||
// parallelism
|
||||
@ -18,7 +18,7 @@ use std::sync::{Arc, Mutex};
|
||||
impl FileTree {
|
||||
/// Find duplicates in a directory (including sub-directories).
|
||||
///
|
||||
/// If path exist in tree, find duplicates using sha256. If 2 (or more) files have the same hash they're duplicates.
|
||||
/// If path exist in tree, find duplicates using Blake2b512. If 2 (or more) files have the same hash they're duplicates.
|
||||
///
|
||||
/// returns a `Vec` containing a `Vec<PathBuf>` for each group of duplicates.
|
||||
///
|
||||
@ -52,7 +52,7 @@ impl FileTree {
|
||||
// parallelized loop
|
||||
intersection.par_iter().for_each(|item| {
|
||||
if let Ok(file) = fs::File::open(item) {
|
||||
let mut sha256 = Sha256::new();
|
||||
let mut hasher = Blake2b512::new();
|
||||
|
||||
// sizable buffer
|
||||
let mut buffer = [0; 8192];
|
||||
@ -66,17 +66,18 @@ impl FileTree {
|
||||
if count == 0 {
|
||||
break;
|
||||
}
|
||||
sha256.update(&buffer[..count]);
|
||||
hasher.update(&buffer[..count]);
|
||||
}
|
||||
|
||||
let hash = sha256.finalize();
|
||||
let hash = hasher.finalize();
|
||||
let hash_bytes: [u8; 32] = hash.as_slice()[..32].try_into().unwrap();
|
||||
|
||||
// Use a Mutex to update HashMap in parallel
|
||||
let mut locked_hashes = hashes.lock().unwrap();
|
||||
locked_hashes
|
||||
.entry(hash.into())
|
||||
.entry(hash_bytes)
|
||||
.or_default()
|
||||
.push(item.clone());
|
||||
.push(PathBuf::from(item));
|
||||
}
|
||||
});
|
||||
|
||||
@ -100,7 +101,7 @@ impl FileTree {
|
||||
}
|
||||
}
|
||||
|
||||
/// Collision shouldn't happen with sha256, but a method to check if there is a collision in duplicates found
|
||||
/// Collision shouldn't happen with blake2, but a method to check if there is a collision in duplicates found
|
||||
/// # Examples
|
||||
/// ```no_run
|
||||
/// let p = Path::new(".");
|
||||
|
Loading…
x
Reference in New Issue
Block a user