#![allow(missing_docs)]
use ring::aead::quic;
#[allow(deprecated)]
use ring::{test, test_file};
#[test]
fn quic_aes_128() {
test_quic(&quic::AES_128, test_file!("quic_aes_128_tests.txt"));
}
#[test]
fn quic_aes_256() {
test_quic(&quic::AES_256, test_file!("quic_aes_256_tests.txt"));
}
#[test]
fn quic_chacha20() {
test_quic(&quic::CHACHA20, test_file!("quic_chacha20_tests.txt"));
}
fn test_quic(alg: &'static quic::Algorithm, test_file: test::File) {
test_key_len(alg);
test_sample_len(alg);
test::run(test_file, |section, test_case| {
assert_eq!(section, "");
let key_bytes = test_case.consume_bytes("KEY");
let sample = test_case.consume_bytes("SAMPLE");
let mask = test_case.consume_bytes("MASK");
let key = quic::HeaderProtectionKey::new(alg, &key_bytes)?;
assert_eq!(mask.as_ref(), key.new_mask(&sample)?);
Ok(())
});
}
#[allow(clippy::range_plus_one)]
fn test_key_len(alg: &'static quic::Algorithm) {
let key_len = alg.key_len();
let key_data = vec![0u8; key_len + 1];
assert!(quic::HeaderProtectionKey::new(alg, &[]).is_err());
assert!(quic::HeaderProtectionKey::new(alg, &key_data[..key_len]).is_ok());
assert!(quic::HeaderProtectionKey::new(alg, &key_data[..(key_len + 1)]).is_err());
assert!(quic::HeaderProtectionKey::new(alg, &key_data[..(key_len - 1)]).is_err());
}
#[allow(clippy::range_plus_one)]
fn test_sample_len(alg: &'static quic::Algorithm) {
let key_len = alg.key_len();
let key_data = vec![0u8; key_len];
let key = quic::HeaderProtectionKey::new(alg, &key_data).unwrap();
let sample_len = alg.sample_len();
assert_eq!(sample_len, 16); let sample_data = vec![0u8; sample_len + 2];
assert!(key.new_mask(&sample_data[..sample_len]).is_ok());
assert!(key.new_mask(&sample_data[..(sample_len - 1)]).is_err());
assert!(key.new_mask(&sample_data[..(sample_len + 1)]).is_err());
assert!(key.new_mask(&[]).is_err());
}