openethereum/util/src/sha3.rs

60 lines
1.5 KiB
Rust
Raw Normal View History

2015-12-04 13:12:11 +01:00
//! Wrapper around tiny-keccak crate.
2015-11-27 17:54:33 +01:00
use std::mem::uninitialized;
2016-01-16 15:29:36 +01:00
use bytes::{BytesConvertable, Populatable};
2016-01-09 10:22:03 +01:00
use hash::{H256, FixedHash};
2015-11-27 17:54:33 +01:00
2016-01-14 21:24:03 +01:00
pub const SHA3_EMPTY: H256 = H256( [0xc5, 0xd2, 0x46, 0x01, 0x86, 0xf7, 0x23, 0x3c, 0x92, 0x7e, 0x7d, 0xb2, 0xdc, 0xc7, 0x03, 0xc0, 0xe5, 0x00, 0xb6, 0x53, 0xca, 0x82, 0x27, 0x3b, 0x7b, 0xfa, 0xd8, 0x04, 0x5d, 0x85, 0xa4, 0x70] );
2016-01-16 14:29:36 +01:00
extern {
fn sha3_256(out: *mut u8, outlen: usize, input: *const u8, inputlen: usize) -> i32;
}
2015-12-04 13:12:11 +01:00
/// Types implementing this trait are sha3able.
2015-12-26 15:48:41 +01:00
///
2015-12-04 13:12:11 +01:00
/// ```
/// extern crate ethcore_util as util;
/// use std::str::FromStr;
/// use util::sha3::*;
/// use util::hash::*;
2015-12-26 15:48:41 +01:00
///
2015-12-04 13:12:11 +01:00
/// fn main() {
/// assert_eq!([0u8; 0].sha3(), H256::from_str("c5d2460186f7233c927e7db2dcc703c0e500b653ca82273b7bfad8045d85a470").unwrap());
/// }
/// ```
2015-11-28 00:14:40 +01:00
pub trait Hashable {
2016-01-16 15:29:36 +01:00
/// Calculate SHA3 of this object.
2015-11-27 17:54:33 +01:00
fn sha3(&self) -> H256;
2016-01-16 15:29:36 +01:00
/// Calculate SHA3 of this object and place result into dest.
2015-12-26 15:48:41 +01:00
fn sha3_into(&self, dest: &mut [u8]) {
self.sha3().copy_to(dest);
}
2015-11-27 17:54:33 +01:00
}
impl<T> Hashable for T where T: BytesConvertable {
fn sha3(&self) -> H256 {
unsafe {
let mut ret: H256 = uninitialized();
2016-01-16 15:29:36 +01:00
self.sha3_into(ret.as_slice_mut());
2015-11-27 17:54:33 +01:00
ret
}
}
fn sha3_into(&self, dest: &mut [u8]) {
2016-01-16 14:29:36 +01:00
unsafe {
let input: &[u8] = self.bytes();
sha3_256(dest.as_mut_ptr(), dest.len(), input.as_ptr(), input.len());
}
}
2015-11-27 17:54:33 +01:00
}
#[test]
fn sha3_empty() {
2016-01-14 21:24:03 +01:00
assert_eq!([0u8; 0].sha3(), SHA3_EMPTY);
2015-11-27 17:54:33 +01:00
}
#[test]
fn sha3_as() {
2016-01-14 21:24:03 +01:00
assert_eq!([0x41u8; 32].sha3(), From::from("59cad5948673622c1d64e2322488bf01619f7ff45789741b15a9f782ce9290a8"));
2015-11-27 17:54:33 +01:00
}