98 lines
3.0 KiB
Rust
98 lines
3.0 KiB
Rust
// Copyright 2015 Brian Smith.
|
|
//
|
|
// Permission to use, copy, modify, and/or distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice appear in all copies.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHORS DISCLAIM ALL WARRANTIES
|
|
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY
|
|
// SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
|
|
// OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
|
|
// CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
#![forbid(
|
|
anonymous_parameters,
|
|
box_pointers,
|
|
legacy_directory_ownership,
|
|
missing_copy_implementations,
|
|
missing_debug_implementations,
|
|
missing_docs,
|
|
trivial_casts,
|
|
trivial_numeric_casts,
|
|
unsafe_code,
|
|
unstable_features,
|
|
unused_extern_crates,
|
|
unused_import_braces,
|
|
unused_qualifications,
|
|
unused_results,
|
|
variant_size_differences,
|
|
warnings
|
|
)]
|
|
|
|
use ring::{digest, error, hkdf, test, test_file};
|
|
|
|
#[cfg(target_arch = "wasm32")]
|
|
use wasm_bindgen_test::wasm_bindgen_test;
|
|
|
|
#[cfg(target_arch = "wasm32")]
|
|
use wasm_bindgen_test::wasm_bindgen_test_configure;
|
|
|
|
#[cfg(target_arch = "wasm32")]
|
|
wasm_bindgen_test_configure!(run_in_browser);
|
|
|
|
#[cfg_attr(not(target_arch = "wasm32"), test)]
|
|
#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
|
|
fn hkdf_tests() {
|
|
test::run(test_file!("hkdf_tests.txt"), |section, test_case| {
|
|
assert_eq!(section, "");
|
|
let alg = {
|
|
let digest_alg = test_case
|
|
.consume_digest_alg("Hash")
|
|
.ok_or(error::Unspecified)?;
|
|
if digest_alg == &digest::SHA256 {
|
|
hkdf::HKDF_SHA256
|
|
} else {
|
|
// TODO: add test vectors for other algorithms
|
|
panic!("unsupported algorithm: {:?}", digest_alg);
|
|
}
|
|
};
|
|
let secret = test_case.consume_bytes("IKM");
|
|
let salt = test_case.consume_bytes("salt");
|
|
let info = test_case.consume_bytes("info");
|
|
let _ = test_case.consume_bytes("PRK");
|
|
let expected_out = test_case.consume_bytes("OKM");
|
|
|
|
let salt = hkdf::Salt::new(alg, &salt);
|
|
|
|
// TODO: test multi-part info, especially with empty parts.
|
|
let My(out) = salt
|
|
.extract(&secret)
|
|
.expand(&[&info], My(expected_out.len()))
|
|
.unwrap()
|
|
.into();
|
|
assert_eq!(out, expected_out);
|
|
|
|
Ok(())
|
|
});
|
|
}
|
|
|
|
/// Generic newtype wrapper that lets us implement traits for externally-defined
|
|
/// types.
|
|
struct My<T>(T);
|
|
|
|
impl hkdf::KeyType for My<usize> {
|
|
fn len(&self) -> usize {
|
|
self.0
|
|
}
|
|
}
|
|
|
|
impl From<hkdf::Okm<'_, My<usize>>> for My<Vec<u8>> {
|
|
fn from(okm: hkdf::Okm<My<usize>>) -> Self {
|
|
let mut r = vec![0u8; okm.len().0];
|
|
okm.fill(&mut r).unwrap();
|
|
My(r)
|
|
}
|
|
}
|