use crate::{VariableOutput, VariableOutputReset, dev::TestVector};
pub fn variable_reset_test<D: VariableOutputReset + Clone>(
&TestVector { input, output }: &TestVector,
) -> Result<(), &'static str> {
let mut hasher = D::new(output.len()).unwrap();
let mut buf = [0u8; 128];
let buf = &mut buf[..output.len()];
hasher.update(input);
let mut hasher2 = hasher.clone();
hasher.finalize_variable(buf).unwrap();
if buf != output {
return Err("whole message");
}
buf.iter_mut().for_each(|b| *b = 0);
hasher2.reset();
hasher2.update(input);
hasher2.finalize_variable_reset(buf).unwrap();
if buf != output {
return Err("whole message after reset");
}
buf.iter_mut().for_each(|b| *b = 0);
for n in 1..core::cmp::min(17, input.len()) {
let mut hasher = D::new(output.len()).unwrap();
for chunk in input.chunks(n) {
hasher.update(chunk);
hasher2.update(chunk);
}
hasher.finalize_variable(buf).unwrap();
if buf != output {
return Err("message in chunks");
}
buf.iter_mut().for_each(|b| *b = 0);
hasher2.finalize_variable_reset(buf).unwrap();
if buf != output {
return Err("message in chunks");
}
buf.iter_mut().for_each(|b| *b = 0);
}
Ok(())
}
pub fn variable_test<D: VariableOutput>(
&TestVector { input, output }: &TestVector,
) -> Result<(), &'static str> {
let mut hasher = D::new(output.len()).unwrap();
let mut buf = [0u8; 128];
let buf = &mut buf[..output.len()];
hasher.update(input);
hasher.finalize_variable(buf).unwrap();
if buf != output {
return Err("whole message");
}
buf.iter_mut().for_each(|b| *b = 0);
for n in 1..core::cmp::min(17, input.len()) {
let mut hasher = D::new(output.len()).unwrap();
for chunk in input.chunks(n) {
hasher.update(chunk);
}
hasher.finalize_variable(buf).unwrap();
if buf != output {
return Err("message in chunks");
}
buf.iter_mut().for_each(|b| *b = 0);
}
Ok(())
}