Revision control
Copy as Markdown
Other Tools
/// Define MAC test
#[macro_export]
#[cfg(feature = "mac")]
#[cfg_attr(docsrs, doc(cfg(all(feature = "dev", feature = "mac"))))]
macro_rules! new_mac_test {
($name:ident, $test_name:expr, $mac:ty $(,)?) => {
digest::new_mac_test!($name, $test_name, $mac, "");
};
($name:ident, $test_name:expr, $mac:ty, trunc_left $(,)?) => {
digest::new_mac_test!($name, $test_name, $mac, "left");
};
($name:ident, $test_name:expr, $mac:ty, trunc_right $(,)?) => {
digest::new_mac_test!($name, $test_name, $mac, "right");
};
($name:ident, $test_name:expr, $mac:ty, $trunc:expr $(,)?) => {
#[test]
fn $name() {
use core::cmp::min;
use digest::dev::blobby::Blob3Iterator;
use digest::Mac;
fn run_test(key: &[u8], input: &[u8], tag: &[u8]) -> Option<&'static str> {
let mac0 = <$mac as Mac>::new_from_slice(key).unwrap();
let mut mac = mac0.clone();
mac.update(input);
let result = mac.finalize().into_bytes();
let n = tag.len();
let result_bytes = match $trunc {
"left" => &result[..n],
"right" => &result[result.len() - n..],
_ => &result[..],
};
if result_bytes != tag {
return Some("whole message");
}
// test reading different chunk sizes
for chunk_size in 1..min(64, input.len()) {
let mut mac = mac0.clone();
for chunk in input.chunks(chunk_size) {
mac.update(chunk);
}
let res = match $trunc {
"left" => mac.verify_truncated_left(tag),
"right" => mac.verify_truncated_right(tag),
_ => mac.verify_slice(tag),
};
if res.is_err() {
return Some("chunked message");
}
}
None
}
let data = include_bytes!(concat!("data/", $test_name, ".blb"));
for (i, row) in Blob3Iterator::new(data).unwrap().enumerate() {
let [key, input, tag] = row.unwrap();
if let Some(desc) = run_test(key, input, tag) {
panic!(
"\n\
Failed test №{}: {}\n\
key:\t{:?}\n\
input:\t{:?}\n\
tag:\t{:?}\n",
i, desc, key, input, tag,
);
}
}
}
};
}
/// Define resettable MAC test
#[macro_export]
#[cfg(feature = "mac")]
#[cfg_attr(docsrs, doc(cfg(all(feature = "dev", feature = "mac"))))]
macro_rules! new_resettable_mac_test {
($name:ident, $test_name:expr, $mac:ty $(,)?) => {
digest::new_resettable_mac_test!($name, $test_name, $mac, "");
};
($name:ident, $test_name:expr, $mac:ty, trunc_left $(,)?) => {
digest::new_resettable_mac_test!($name, $test_name, $mac, "left");
};
($name:ident, $test_name:expr, $mac:ty, trunc_right $(,)?) => {
digest::new_resettable_mac_test!($name, $test_name, $mac, "right");
};
($name:ident, $test_name:expr, $mac:ty, $trunc:expr $(,)?) => {
#[test]
fn $name() {
use core::cmp::min;
use digest::dev::blobby::Blob3Iterator;
use digest::Mac;
fn run_test(key: &[u8], input: &[u8], tag: &[u8]) -> Option<&'static str> {
let mac0 = <$mac as Mac>::new_from_slice(key).unwrap();
let mut mac = mac0.clone();
mac.update(input);
let result = mac.finalize_reset().into_bytes();
let n = tag.len();
let result_bytes = match $trunc {
"left" => &result[..n],
"right" => &result[result.len() - n..],
_ => &result[..],
};
if result_bytes != tag {
return Some("whole message");
}
// test if reset worked correctly
mac.update(input);
let res = match $trunc {
"left" => mac.verify_truncated_left(tag),
"right" => mac.verify_truncated_right(tag),
_ => mac.verify_slice(tag),
};
if res.is_err() {
return Some("after reset");
}
// test reading different chunk sizes
for chunk_size in 1..min(64, input.len()) {
let mut mac = mac0.clone();
for chunk in input.chunks(chunk_size) {
mac.update(chunk);
}
let res = match $trunc {
"left" => mac.verify_truncated_left(tag),
"right" => mac.verify_truncated_right(tag),
_ => mac.verify_slice(tag),
};
if res.is_err() {
return Some("chunked message");
}
}
None
}
let data = include_bytes!(concat!("data/", $test_name, ".blb"));
for (i, row) in Blob3Iterator::new(data).unwrap().enumerate() {
let [key, input, tag] = row.unwrap();
if let Some(desc) = run_test(key, input, tag) {
panic!(
"\n\
Failed test №{}: {}\n\
key:\t{:?}\n\
input:\t{:?}\n\
tag:\t{:?}\n",
i, desc, key, input, tag,
);
}
}
}
};
}