| extern crate sgx_crypto_helper; |
| extern crate sgx_ucrypto as crypto; |
| |
| use crypto::*; |
| |
| static HASH_TEST_VEC: &'static [&'static str] = &[ |
| &"abc", |
| &"abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq", |
| &"abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmnoijklmnopjklmnopqklmnopqrlmnopqrsmnopqrstnopqrstu", |
| ]; |
| |
| static HASH_SHA256_TRUTH: &'static [&'static str] = &[ |
| &"ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad", |
| &"248d6a61d20638b8e5c026930c3e6039a33ce45964ff2167f6ecedd419db06c1", |
| &"cf5b16a778af8380036ce59e7b0492370b249b11e8f07a51afac45037afee9d1", |
| ]; |
| |
| fn test_rsgx_sha256_slice() { |
| let test_size = HASH_TEST_VEC.len(); |
| for i in 0..test_size { |
| let input_str = String::from(HASH_TEST_VEC[i]); |
| let hash = rsgx_sha256_slice(input_str.as_bytes()).unwrap(); |
| assert_eq!(hex_to_bytes(HASH_SHA256_TRUTH[i]), hash); |
| } |
| } |
| |
| fn test_rsgx_sha256_handle() { |
| let test_size = HASH_TEST_VEC.len(); |
| for i in 0..test_size { |
| let input_str = String::from(HASH_TEST_VEC[i]); |
| let shah = SgxShaHandle::new(); |
| shah.init().unwrap(); |
| shah.update_slice(input_str.as_bytes()).unwrap(); |
| let hash = shah.get_hash().unwrap(); |
| shah.close().unwrap(); |
| assert_eq!(hex_to_bytes(HASH_SHA256_TRUTH[i]), hash); |
| } |
| } |
| |
| fn main() { |
| test_rsgx_sha256_slice(); |
| test_rsgx_sha256_handle(); |
| println!("Complete!"); |
| } |