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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
//
// Copyright (c) 2020-2021 science+computing ag and other contributors
//
// This program and the accompanying materials are made
// available under the terms of the Eclipse Public License 2.0
// which is available at https://www.eclipse.org/legal/epl-2.0/
//
// SPDX-License-Identifier: EPL-2.0
//
use anyhow::anyhow;
use anyhow::Result;
use getset::Getters;
use log::trace;
use serde::Deserialize;
use serde::Serialize;
use url::Url;
use sha2::Digest;
#[derive(Clone, Debug, Serialize, Deserialize, Getters)]
pub struct Source {
#[getset(get = "pub")]
url: Url,
#[getset(get = "pub")]
hash: SourceHash,
#[getset(get = "pub")]
download_manually: bool,
}
impl Source {
#[cfg(test)]
pub fn new(url: Url, hash: SourceHash) -> Self {
Source {
url,
hash,
download_manually: false,
}
}
}
#[derive(Clone, Debug, Serialize, Deserialize, Getters)]
pub struct SourceHash {
#[serde(rename = "type")]
#[getset(get = "pub")]
hashtype: HashType,
#[serde(rename = "hash")]
#[getset(get = "pub")]
value: HashValue,
}
impl SourceHash {
pub fn matches_hash_of(&self, buf: &[u8]) -> Result<()> {
trace!("Hashing buffer with: {:?}", self.hashtype);
let h = self.hashtype.hash_buffer(&buf)?;
trace!("Hashing buffer with: {} finished", self.hashtype);
if h == self.value {
trace!("Hash matches expected hash");
Ok(())
} else {
trace!("Hash mismatch expected hash");
Err(anyhow!(
"Hash mismatch, expected '{}', got '{}'",
self.value,
h
))
}
}
#[cfg(test)]
pub fn new(hashtype: HashType, value: HashValue) -> Self {
SourceHash { hashtype, value }
}
}
#[derive(parse_display::Display, Clone, Debug, Serialize, Deserialize)]
pub enum HashType {
#[serde(rename = "sha1")]
#[display("sha1")]
Sha1,
#[serde(rename = "sha256")]
#[display("sha256")]
Sha256,
#[serde(rename = "sha512")]
#[display("sha512")]
Sha512,
}
impl HashType {
fn hash_buffer(&self, buffer: &[u8]) -> Result<HashValue> {
match self {
HashType::Sha1 => {
trace!("SHA1 hashing buffer");
let mut m = sha1::Sha1::new();
m.update(buffer);
Ok(HashValue(m.digest().to_string()))
}
HashType::Sha256 => {
trace!("SHA256 hashing buffer");
let mut m = sha2::Sha256::new();
m.update(buffer);
Ok(HashValue(String::from_utf8(m.finalize()[..].to_vec())?))
}
HashType::Sha512 => {
trace!("SHA512 hashing buffer");
let mut m = sha2::Sha512::new();
m.update(buffer);
Ok(HashValue(String::from_utf8(m.finalize()[..].to_vec())?))
}
}
}
}
#[derive(parse_display::Display, Serialize, Deserialize, Clone, Debug, Hash, Eq, PartialEq)]
#[serde(transparent)]
#[display("{0}")]
pub struct HashValue(String);
#[cfg(test)]
impl From<String> for HashValue {
fn from(s: String) -> Self {
HashValue(s)
}
}
|