1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
use async_fs::File;
use futures::prelude::*;
use hex::FromHex;
use md5::{digest::generic_array::GenericArray, Digest, Md5};
use sha1::Sha1;
use std::{io, path::Path};
use thiserror::Error;
use crate::request::RequestChecksum;
#[derive(Debug, Error)]
pub enum ChecksumError {
#[error("checksum invalid: {0}")]
InvalidInput(String),
#[error("unable to open the file to validate")]
FileOpen(#[source] io::Error),
#[error("file does not match expected size")]
InvalidSize,
#[error("error during read of file")]
FileRead(#[source] io::Error),
#[error("checksum mismatch")]
Mismatch,
}
pub async fn compare_hash(
path: &Path,
expected_size: u64,
expected_hash: &RequestChecksum,
) -> Result<(), ChecksumError> {
let mut file = File::open(path).await.map_err(ChecksumError::FileOpen)?;
let file_size = file.metadata().await.unwrap().len();
if file_size != expected_size {
return Err(ChecksumError::InvalidSize);
}
match expected_hash {
RequestChecksum::Sha1(sum) => {
let expected = <[u8; 20]>::from_hex(&*sum)
.map(GenericArray::from)
.map_err(|_| ChecksumError::InvalidInput(
format!("SHA1 {}", sum)
))?;
let mut buffer = vec![0u8; 8 * 1024];
let mut hasher = Sha1::new();
loop {
match file.read(&mut buffer).await {
Ok(0) => break,
Ok(bytes) => hasher.update(&buffer[..bytes]),
Err(why) => return Err(ChecksumError::FileRead(why)),
}
}
let hash = &*hasher.finalize();
if &*expected == hash {
Ok(())
} else {
Err(ChecksumError::Mismatch)
}
}
RequestChecksum::Md5(sum) => {
let expected = <[u8; 16]>::from_hex(&*sum)
.map(GenericArray::from)
.map_err(|_| ChecksumError::InvalidInput(
format!("MD5 {}", sum)
))?;
let mut buffer = vec![0u8; 8 * 1024];
let mut hasher = Md5::new();
loop {
match file.read(&mut buffer).await {
Ok(0) => break,
Ok(bytes) => hasher.update(&buffer[..bytes]),
Err(why) => return Err(ChecksumError::FileRead(why)),
}
}
let hash = &*hasher.finalize();
if &*expected == hash {
Ok(())
} else {
Err(ChecksumError::Mismatch)
}
}
}
}
