use bitcoin_io::impl_write;
use crate::{hmac, ripemd160, sha1, sha256, sha256t, sha512, siphash24, HashEngine};
impl_write!(
sha1::Engine,
|us: &mut sha1::Engine, buf| {
us.input(buf);
Ok(buf.len())
},
|_us| { Ok(()) }
);
impl_write!(
sha256::Engine,
|us: &mut sha256::Engine, buf| {
us.input(buf);
Ok(buf.len())
},
|_us| { Ok(()) }
);
impl_write!(
sha512::Engine,
|us: &mut sha512::Engine, buf| {
us.input(buf);
Ok(buf.len())
},
|_us| { Ok(()) }
);
impl_write!(
ripemd160::Engine,
|us: &mut ripemd160::Engine, buf| {
us.input(buf);
Ok(buf.len())
},
|_us| { Ok(()) }
);
impl_write!(
siphash24::Engine,
|us: &mut siphash24::Engine, buf| {
us.input(buf);
Ok(buf.len())
},
|_us| { Ok(()) }
);
impl<E: HashEngine> bitcoin_io::Write for hmac::Engine<E> {
#[inline]
fn write(&mut self, buf: &[u8]) -> Result<usize, bitcoin_io::Error> {
self.input(buf);
Ok(buf.len())
}
#[inline]
fn flush(&mut self) -> Result<(), bitcoin_io::Error> { Ok(()) }
}
#[cfg(feature = "std")]
impl<E: HashEngine> std::io::Write for hmac::Engine<E> {
#[inline]
fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> {
self.input(buf);
Ok(buf.len())
}
#[inline]
fn flush(&mut self) -> std::io::Result<()> { Ok(()) }
}
impl_write!(
sha256t::Engine<T>,
|us: &mut sha256t::Engine<T>, buf| {
us.input(buf);
Ok(buf.len())
},
|_us| { Ok(()) },
T: crate::sha256t::Tag
);
#[cfg(test)]
mod tests {
use bitcoin_io::Write;
use crate::{hmac, ripemd160, sha1, sha256, sha512, siphash24, HashEngine as _};
macro_rules! write_test {
($mod:ident, $exp_empty:expr, $exp_256:expr, $exp_64k:expr,) => {
#[test]
fn $mod() {
let mut engine = $mod::Engine::new();
engine.write_all(&[]).unwrap();
let hash = $mod::Hash::from_engine(engine);
assert_eq!(format!("{}", hash), $exp_empty);
let mut engine = $mod::Engine::new();
engine.write_all(&[1; 256]).unwrap();
let hash = $mod::Hash::from_engine(engine);
assert_eq!(format!("{}", hash), $exp_256);
let mut engine = $mod::Engine::new();
engine.write_all(&[99; 64000]).unwrap();
let hash = $mod::Hash::from_engine(engine);
assert_eq!(format!("{}", hash), $exp_64k);
}
};
}
write_test!(
sha1,
"da39a3ee5e6b4b0d3255bfef95601890afd80709",
"ac458b067c6b021c7e9358229b636e9d1e4cb154",
"e4b66838f9f7b6f91e5be32a02ae78094df402e7",
);
write_test!(
sha256,
"e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
"2661920f2409dd6c8adeb0c44972959f232b6429afa913845d0fd95e7e768234",
"5c5e904f5d4fd587c7a906bf846e08a927286f388c54c39213a4884695271bbc",
);
write_test!(
sha512,
"cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce\
47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e",
"57ecf739d3a7ca647639adae80a05f4f361304bfcbfa1ceba93296b096e74287\
45fc10c142cecdd3bb587a3dba598c072f6f78b31cc0a06a3da0105ee51f75d6",
"dd28f78c53f3bc9bd0c2dca9642a1ad402a70412f985c1f6e54fadb98ce9c458\
4761df8d04ed04bb734ba48dd2106bb9ea54524f1394cdd18e6da3166e71c3ee",
);
write_test!(
ripemd160,
"9c1185a5c5e9fc54612808977ee8f548b2258d31",
"e571a1ca5b780aa52bafdb9ec852544ffca418ba",
"ddd2ecce739e823629c7d46ab18918e9c4a51c75",
);
write_test!(siphash24, "d70077739d4b921e", "3a3ccefde9b5b1e3", "ce456e4e4ecbc5bf",);
#[test]
fn hmac() {
let mut engine = hmac::Engine::<sha256::Engine>::new(&[0xde, 0xad, 0xbe, 0xef]);
engine.write_all(&[]).unwrap();
assert_eq!(
format!("{}", hmac::Hash::from_engine(engine)),
"bf5515149cf797955c4d3194cca42472883281951697c8375d9d9b107f384225"
);
let mut engine = hmac::Engine::<sha256::Engine>::new(&[0xde, 0xad, 0xbe, 0xef]);
engine.write_all(&[1; 256]).unwrap();
assert_eq!(
format!("{}", hmac::Hash::from_engine(engine)),
"59c9aca10c81c73cb4c196d94db741b6bf2050e0153d5a45f2526bff34675ac5"
);
let mut engine = hmac::Engine::<sha256::Engine>::new(&[0xde, 0xad, 0xbe, 0xef]);
engine.write_all(&[99; 64000]).unwrap();
assert_eq!(
format!("{}", hmac::Hash::from_engine(engine)),
"30df499717415a395379a1eaabe50038036e4abb5afc94aa55c952f4aa57be08"
);
}
}