anthrove/fuzzysearch
4
0
Fork 0
mirror of https://github.com/Syfaro/fuzzysearch.git synced 2024-09-20 13:59:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle corrupted images.

Browse Source
This commit is contained in:
Syfaro 2020-10-09 19:10:44 -04:00
parent 0b42564c5e
commit 41a993d0b1
1 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/main.rs
View File

@ -122,16 +122,22 @@ async fn process_submission(
.bytes() .bytes()
.await?; .await?;
let num = if let Ok(image) = image::load_from_memory(&data) {
let hasher = img_hash::HasherConfig::with_bytes_type::<[u8; 8]>() let hasher = img_hash::HasherConfig::with_bytes_type::<[u8; 8]>()
.hash_alg(img_hash::HashAlg::Gradient) .hash_alg(img_hash::HashAlg::Gradient)
.hash_size(8, 8) .hash_size(8, 8)
.preproc_dct() .preproc_dct()
.to_hasher(); .to_hasher();
let image = image::load_from_memory(&data)?;
let hash = hasher.hash_image(&image); let hash = hasher.hash_image(&image);
let mut bytes: [u8; 8] = [0; 8]; let mut bytes: [u8; 8] = [0; 8];
bytes.copy_from_slice(hash.as_bytes()); bytes.copy_from_slice(hash.as_bytes());
let num = i64::from_be_bytes(bytes); let num = i64::from_be_bytes(bytes);
Some(num)
} else {
println!("Unable to decode image on submission {}", sub.id);
None
};
let mut hasher = Sha256::new(); let mut hasher = Sha256::new();
hasher.update(&data); hasher.update(&data);