201 lines
4.7 KiB
Rust
201 lines
4.7 KiB
Rust
use log::{debug, warn};
|
|
use sha3::{
|
|
Keccak224, Keccak256, Keccak256Full, Keccak384, Keccak512, Sha3_224, Sha3_256, Sha3_384,
|
|
Sha3_512,
|
|
};
|
|
|
|
use super::{DigestCreate, DigestPart, DigestVerify};
|
|
|
|
fn create(digest: &mut impl sha3::Digest, input: &[u8]) -> String {
|
|
digest.update(input);
|
|
base64::encode(&digest.finalize_reset())
|
|
}
|
|
|
|
fn verify(digest: &mut impl sha3::Digest, name: &str, parts: &[DigestPart]) -> bool {
|
|
if let Some(part) = parts
|
|
.iter()
|
|
.find(|p| p.algorithm.to_lowercase() == name.to_lowercase())
|
|
{
|
|
debug!("Verifying digest type, {}", name);
|
|
let encoded = base64::encode(&digest.finalize_reset());
|
|
|
|
return part.digest == encoded;
|
|
}
|
|
warn!("No matching digest algorithm found for {}", name);
|
|
warn!(
|
|
"Provided: [{}]",
|
|
parts.iter().fold(String::new(), |mut acc, item| {
|
|
if acc.is_empty() {
|
|
} else {
|
|
acc.push_str(", ");
|
|
}
|
|
acc.push_str(&item.algorithm);
|
|
acc
|
|
})
|
|
);
|
|
|
|
false
|
|
}
|
|
|
|
impl DigestCreate for Sha3_224 {
|
|
const NAME: &'static str = "SHA3-224";
|
|
|
|
fn compute(&mut self, input: &[u8]) -> String {
|
|
create(self, input)
|
|
}
|
|
}
|
|
|
|
impl DigestVerify for Sha3_224 {
|
|
fn update(&mut self, part: &[u8]) {
|
|
sha3::Digest::update(self, part);
|
|
}
|
|
|
|
fn verify(&mut self, parts: &[DigestPart]) -> bool {
|
|
verify(self, <Self as DigestCreate>::NAME, parts)
|
|
}
|
|
}
|
|
|
|
impl DigestCreate for Sha3_256 {
|
|
const NAME: &'static str = "SHA3-256";
|
|
|
|
fn compute(&mut self, input: &[u8]) -> String {
|
|
create(self, input)
|
|
}
|
|
}
|
|
|
|
impl DigestVerify for Sha3_256 {
|
|
fn update(&mut self, part: &[u8]) {
|
|
sha3::Digest::update(self, part);
|
|
}
|
|
|
|
fn verify(&mut self, parts: &[DigestPart]) -> bool {
|
|
verify(self, <Self as DigestCreate>::NAME, parts)
|
|
}
|
|
}
|
|
|
|
impl DigestCreate for Sha3_384 {
|
|
const NAME: &'static str = "SHA3-384";
|
|
|
|
fn compute(&mut self, input: &[u8]) -> String {
|
|
create(self, input)
|
|
}
|
|
}
|
|
|
|
impl DigestVerify for Sha3_384 {
|
|
fn update(&mut self, part: &[u8]) {
|
|
sha3::Digest::update(self, part);
|
|
}
|
|
|
|
fn verify(&mut self, parts: &[DigestPart]) -> bool {
|
|
verify(self, <Self as DigestCreate>::NAME, parts)
|
|
}
|
|
}
|
|
|
|
impl DigestCreate for Sha3_512 {
|
|
const NAME: &'static str = "SHA3-512";
|
|
|
|
fn compute(&mut self, input: &[u8]) -> String {
|
|
create(self, input)
|
|
}
|
|
}
|
|
|
|
impl DigestVerify for Sha3_512 {
|
|
fn update(&mut self, part: &[u8]) {
|
|
sha3::Digest::update(self, part);
|
|
}
|
|
|
|
fn verify(&mut self, parts: &[DigestPart]) -> bool {
|
|
verify(self, <Self as DigestCreate>::NAME, parts)
|
|
}
|
|
}
|
|
|
|
impl DigestCreate for Keccak224 {
|
|
const NAME: &'static str = "keccak-224";
|
|
|
|
fn compute(&mut self, input: &[u8]) -> String {
|
|
create(self, input)
|
|
}
|
|
}
|
|
|
|
impl DigestVerify for Keccak224 {
|
|
fn update(&mut self, part: &[u8]) {
|
|
sha3::Digest::update(self, part);
|
|
}
|
|
|
|
fn verify(&mut self, parts: &[DigestPart]) -> bool {
|
|
verify(self, <Self as DigestCreate>::NAME, parts)
|
|
}
|
|
}
|
|
|
|
impl DigestCreate for Keccak256 {
|
|
const NAME: &'static str = "keccak-256";
|
|
|
|
fn compute(&mut self, input: &[u8]) -> String {
|
|
create(self, input)
|
|
}
|
|
}
|
|
|
|
impl DigestVerify for Keccak256 {
|
|
fn update(&mut self, part: &[u8]) {
|
|
sha3::Digest::update(self, part);
|
|
}
|
|
|
|
fn verify(&mut self, parts: &[DigestPart]) -> bool {
|
|
verify(self, <Self as DigestCreate>::NAME, parts)
|
|
}
|
|
}
|
|
|
|
impl DigestCreate for Keccak256Full {
|
|
const NAME: &'static str = "keccak-256-full";
|
|
|
|
fn compute(&mut self, input: &[u8]) -> String {
|
|
create(self, input)
|
|
}
|
|
}
|
|
|
|
impl DigestVerify for Keccak256Full {
|
|
fn update(&mut self, part: &[u8]) {
|
|
sha3::Digest::update(self, part);
|
|
}
|
|
|
|
fn verify(&mut self, parts: &[DigestPart]) -> bool {
|
|
verify(self, <Self as DigestCreate>::NAME, parts)
|
|
}
|
|
}
|
|
|
|
impl DigestCreate for Keccak384 {
|
|
const NAME: &'static str = "keccak-384";
|
|
|
|
fn compute(&mut self, input: &[u8]) -> String {
|
|
create(self, input)
|
|
}
|
|
}
|
|
|
|
impl DigestVerify for Keccak384 {
|
|
fn update(&mut self, part: &[u8]) {
|
|
sha3::Digest::update(self, part);
|
|
}
|
|
|
|
fn verify(&mut self, parts: &[DigestPart]) -> bool {
|
|
verify(self, <Self as DigestCreate>::NAME, parts)
|
|
}
|
|
}
|
|
|
|
impl DigestCreate for Keccak512 {
|
|
const NAME: &'static str = "keccak-512";
|
|
|
|
fn compute(&mut self, input: &[u8]) -> String {
|
|
create(self, input)
|
|
}
|
|
}
|
|
|
|
impl DigestVerify for Keccak512 {
|
|
fn update(&mut self, part: &[u8]) {
|
|
sha3::Digest::update(self, part);
|
|
}
|
|
|
|
fn verify(&mut self, parts: &[DigestPart]) -> bool {
|
|
verify(self, <Self as DigestCreate>::NAME, parts)
|
|
}
|
|
}
|