wolfSSL/wolfssl
1
0
Fork 1
mirror of https://github.com/wolfSSL/wolfssl.git synced 2026-07-06 08:50:49 +02:00
Code Issues Packages Projects Releases Wiki Activity
Files
a5ee9604c71294dea6d0d31da75dd24f2f2908a1
wolfssl/wrapper/rust/wolfssl-wolfcrypt/tests/test_aes.rs
T
Josh Holtrop 23cb7ae30c Rust wrapper: add cipher trait support
2026-03-25 10:51:06 -04:00

1426 lines
58 KiB
Rust
Raw Blame History

#![cfg(aes)]
use wolfssl_wolfcrypt::aes::*;
const BIG_MSG: [u8; 384] = [
0x41,0x6c,0x6c,0x20,0x77,0x6f,0x72,0x6b,
0x20,0x61,0x6e,0x64,0x20,0x6e,0x6f,0x20,
0x70,0x6c,0x61,0x79,0x20,0x6d,0x61,0x6b,
0x65,0x73,0x20,0x4a,0x61,0x63,0x6b,0x20,
0x61,0x20,0x64,0x75,0x6c,0x6c,0x20,0x62,
0x6f,0x79,0x2e,0x20,0x41,0x6c,0x6c,0x20,
0x77,0x6f,0x72,0x6b,0x20,0x61,0x6e,0x64,
0x20,0x6e,0x6f,0x20,0x70,0x6c,0x61,0x79,
0x20,0x6d,0x61,0x6b,0x65,0x73,0x20,0x4a,
0x61,0x63,0x6b,0x20,0x61,0x20,0x64,0x75,
0x6c,0x6c,0x20,0x62,0x6f,0x79,0x2e,0x20,
0x41,0x6c,0x6c,0x20,0x77,0x6f,0x72,0x6b,
0x20,0x61,0x6e,0x64,0x20,0x6e,0x6f,0x20,
0x70,0x6c,0x61,0x79,0x20,0x6d,0x61,0x6b,
0x65,0x73,0x20,0x4a,0x61,0x63,0x6b,0x20,
0x61,0x20,0x64,0x75,0x6c,0x6c,0x20,0x62,
0x6f,0x79,0x2e,0x20,0x41,0x6c,0x6c,0x20,
0x77,0x6f,0x72,0x6b,0x20,0x61,0x6e,0x64,
0x20,0x6e,0x6f,0x20,0x70,0x6c,0x61,0x79,
0x20,0x6d,0x61,0x6b,0x65,0x73,0x20,0x4a,
0x61,0x63,0x6b,0x20,0x61,0x20,0x64,0x75,
0x6c,0x6c,0x20,0x62,0x6f,0x79,0x2e,0x20,
0x41,0x6c,0x6c,0x20,0x77,0x6f,0x72,0x6b,
0x20,0x61,0x6e,0x64,0x20,0x6e,0x6f,0x20,
0x70,0x6c,0x61,0x79,0x20,0x6d,0x61,0x6b,
0x65,0x73,0x20,0x4a,0x61,0x63,0x6b,0x20,
0x61,0x20,0x64,0x75,0x6c,0x6c,0x20,0x62,
0x6f,0x79,0x2e,0x20,0x41,0x6c,0x6c,0x20,
0x77,0x6f,0x72,0x6b,0x20,0x61,0x6e,0x64,
0x20,0x6e,0x6f,0x20,0x70,0x6c,0x61,0x79,
0x20,0x6d,0x61,0x6b,0x65,0x73,0x20,0x4a,
0x61,0x63,0x6b,0x20,0x61,0x20,0x64,0x75,
0x6c,0x6c,0x20,0x62,0x6f,0x79,0x2e,0x20,
0x41,0x6c,0x6c,0x20,0x77,0x6f,0x72,0x6b,
0x20,0x61,0x6e,0x64,0x20,0x6e,0x6f,0x20,
0x70,0x6c,0x61,0x79,0x20,0x6d,0x61,0x6b,
0x65,0x73,0x20,0x4a,0x61,0x63,0x6b,0x20,
0x61,0x20,0x64,0x75,0x6c,0x6c,0x20,0x62,
0x6f,0x79,0x2e,0x20,0x41,0x6c,0x6c,0x20,
0x77,0x6f,0x72,0x6b,0x20,0x61,0x6e,0x64,
0x20,0x6e,0x6f,0x20,0x70,0x6c,0x61,0x79,
0x20,0x6d,0x61,0x6b,0x65,0x73,0x20,0x4a,
0x61,0x63,0x6b,0x20,0x61,0x20,0x64,0x75,
0x6c,0x6c,0x20,0x62,0x6f,0x79,0x2e,0x20,
0x41,0x6c,0x6c,0x20,0x77,0x6f,0x72,0x6b,
0x20,0x61,0x6e,0x64,0x20,0x6e,0x6f,0x20,
0x70,0x6c,0x61,0x79,0x20,0x6d,0x61,0x6b,
0x65,0x73,0x20,0x4a,0x61,0x63,0x6b,0x20
];
#[test]
#[cfg(aes_cbc)]
fn test_cbc_encrypt_decrypt() {
let mut cbc = CBC::new().expect("Failed to create CBC");
let key: &[u8; 16] = b"0123456789abcdef";
let iv: &[u8; 16] = b"1234567890abcdef";
let msg: [u8; 16] = [
0x6e, 0x6f, 0x77, 0x20, 0x69, 0x73, 0x20, 0x74,
0x68, 0x65, 0x20, 0x74, 0x69, 0x6d, 0x65, 0x20,
];
let expected_cipher: [u8; 16] = [
0x95, 0x94, 0x92, 0x57, 0x5f, 0x42, 0x81, 0x53,
0x2c, 0xcc, 0x9d, 0x46, 0x77, 0xa2, 0x33, 0xcb
];
cbc.init_encrypt(key, iv).expect("Error with init_encrypt()");
let mut cipher: [u8; 16] = [0; 16];
cbc.encrypt(&msg, &mut cipher).expect("Error with encrypt()");
assert_eq!(&cipher, &expected_cipher);
let mut plain_out = [0; 16];
cbc.init_decrypt(key, iv).expect("Error with init_decrypt()");
cbc.decrypt(&cipher, &mut plain_out).expect("Error with decrypt()");
assert_eq!(&plain_out, &msg);
}
#[test]
#[cfg(aes_cbc)]
fn test_cbc_big_msg() {
let mut cbc = CBC::new().expect("Failed to create CBC");
let big_key = b"0123456789abcdeffedcba9876543210";
let iv: &[u8; 16] = b"1234567890abcdef";
let mut big_cipher: [u8; 384] = [0; 384];
let mut big_plain: [u8; 384] = [0; 384];
cbc.init_encrypt(big_key, iv).expect("Error with init_encrypt()");
for i in (0..384).step_by(16) {
cbc.encrypt(&BIG_MSG[i..(i + 16)], &mut big_cipher[i..(i + 16)]).expect("Error with encrypt()");
}
cbc.init_decrypt(big_key, iv).expect("Error with init_decrypt()");
for i in (0..384).step_by(16) {
cbc.decrypt(&big_cipher[i..(i + 16)], &mut big_plain[i..(i + 16)]).expect("Error with decrypt()");
}
assert_eq!(big_plain, BIG_MSG);
}
#[test]
#[cfg(aes_ccm)]
fn test_ccm_encrypt_decrypt() {
let key: [u8; 16] = [
0xc0, 0xc1, 0xc2, 0xc3, 0xc4, 0xc5, 0xc6, 0xc7,
0xc8, 0xc9, 0xca, 0xcb, 0xcc, 0xcd, 0xce, 0xcf
];
let nonce: [u8; 13] = [
0x00, 0x00, 0x00, 0x03, 0x02, 0x01, 0x00, 0xa0,
0xa1, 0xa2, 0xa3, 0xa4, 0xa5
];
let plaintext: [u8; 23] = [
0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,
0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e
];
let plaintext_long: [u8; 73] = [
0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,
0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f,
0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27,
0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f,
0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37,
0x38, 0x39, 0x3a, 0x3b, 0x3c, 0x3d, 0x3e, 0x3f,
0x40, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47,
0x48, 0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f,
0x50
];
let auth_data: [u8; 8] = [
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07
];
let expected_ciphertext: [u8; 23] = [
0x58, 0x8c, 0x97, 0x9a, 0x61, 0xc6, 0x63, 0xd2,
0xf0, 0x66, 0xd0, 0xc2, 0xc0, 0xf9, 0x89, 0x80,
0x6d, 0x5f, 0x6b, 0x61, 0xda, 0xc3, 0x84
];
let expected_auth_tag: [u8; 8] = [
0x17, 0xe8, 0xd1, 0x2c, 0xfd, 0xf9, 0x26, 0xe0
];
let expected_ciphertext_long: [u8; 73] = [
0x58, 0x8c, 0x97, 0x9a, 0x61, 0xc6, 0x63, 0xd2,
0xf0, 0x66, 0xd0, 0xc2, 0xc0, 0xf9, 0x89, 0x80,
0x6d, 0x5f, 0x6b, 0x61, 0xda, 0xc3, 0x84, 0xe0,
0x44, 0x2d, 0xbe, 0x25, 0xfa, 0x48, 0x2b, 0xa8,
0x36, 0x0b, 0xbf, 0x01, 0xc0, 0x12, 0x45, 0xa4,
0x82, 0x9f, 0x20, 0x6c, 0xc3, 0xd6, 0xae, 0x5b,
0x54, 0x8d, 0xd0, 0xb1, 0x69, 0x2c, 0xec, 0x5e,
0x95, 0xa5, 0x6b, 0x48, 0xc3, 0xc6, 0xc8, 0x9e,
0xc7, 0x92, 0x98, 0x9d, 0x26, 0x7d, 0x2a, 0x10,
0x0b
];
let expected_auth_tag_long: [u8; 8] = [
0x89, 0xd8, 0xd2, 0x02, 0xc5, 0xcf, 0xae, 0xf4
];
let mut ccm = CCM::new().expect("Failed to create CCM");
ccm.init(&key).expect("Error with init()");
let mut auth_tag_out: [u8; 8] = [0; 8];
let mut cipher_out: [u8; 23] = [0; 23];
ccm.encrypt(&plaintext, &mut cipher_out,
&nonce, &auth_data, &mut auth_tag_out).expect("Error with encrypt()");
assert_eq!(cipher_out, expected_ciphertext);
assert_eq!(auth_tag_out, expected_auth_tag);
ccm.init(&key).expect("Error with init()");
let mut plain_out: [u8; 23] = [0; 23];
ccm.decrypt(&cipher_out, &mut plain_out,
&nonce, &auth_data, &auth_tag_out).expect("Error with decrypt()");
assert_eq!(plain_out, plaintext);
ccm.init(&key).expect("Error with init()");
let mut auth_tag_long_out: [u8; 8] = [0; 8];
let mut cipher_long_out: [u8; 73] = [0; 73];
ccm.encrypt(&plaintext_long, &mut cipher_long_out,
&nonce, &auth_data, &mut auth_tag_long_out).expect("Error with encrypt()");
assert_eq!(cipher_long_out, expected_ciphertext_long);
assert_eq!(auth_tag_long_out, expected_auth_tag_long);
ccm.init(&key).expect("Error with init()");
let mut plain_long_out: [u8; 73] = [0; 73];
ccm.decrypt(&cipher_long_out, &mut plain_long_out,
&nonce, &auth_data, &auth_tag_long_out).expect("Error with decrypt()");
assert_eq!(plain_long_out, plaintext_long);
}
#[test]
#[cfg(aes_ccm)]
fn test_ccm_big_msg() {
let mut ccm = CCM::new().expect("Failed to create CCM");
let big_key = b"0123456789abcdeffedcba9876543210";
let nonce: [u8; 7] = [0x00, 0x00, 0x00, 0x03, 0x02, 0x01, 0x00];
let auth_data: [u8; 8] = [0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07];
let mut big_cipher: [u8; 384] = [0; 384];
let mut big_plain: [u8; 384] = [0; 384];
let mut auth_tag: [u8; 8] = [0; 8];
ccm.init(big_key).expect("Error with init()");
ccm.encrypt(&BIG_MSG, &mut big_cipher,
&nonce, &auth_data, &mut auth_tag).expect("Error with encrypt()");
ccm.init(big_key).expect("Error with init()");
ccm.decrypt(&big_cipher, &mut big_plain,
&nonce, &auth_data, &auth_tag).expect("Error with decrypt()");
assert_eq!(big_plain, BIG_MSG);
}
#[test]
#[cfg(aes_cfb)]
fn test_cfb_encrypt_decrypt() {
let mut cfb = CFB::new().expect("Failed to create CFB");
let key: [u8; 16] = [
0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,
0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c
];
let iv: [u8; 16] = [
0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,
0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f
];
let msg: [u8; 48] = [
0x6b,0xc1,0xbe,0xe2,0x2e,0x40,0x9f,0x96,
0xe9,0x3d,0x7e,0x11,0x73,0x93,0x17,0x2a,
0xae,0x2d,0x8a,0x57,0x1e,0x03,0xac,0x9c,
0x9e,0xb7,0x6f,0xac,0x45,0xaf,0x8e,0x51,
0x30,0xc8,0x1c,0x46,0xa3,0x5c,0xe4,0x11,
0xe5,0xfb,0xc1,0x19,0x1a,0x0a,0x52,0xef
];
let cipher: [u8; 48] = [
0x3b,0x3f,0xd9,0x2e,0xb7,0x2d,0xad,0x20,
0x33,0x34,0x49,0xf8,0xe8,0x3c,0xfb,0x4a,
0xc8,0xa6,0x45,0x37,0xa0,0xb3,0xa9,0x3f,
0xcd,0xe3,0xcd,0xad,0x9f,0x1c,0xe5,0x8b,
0x26,0x75,0x1f,0x67,0xa3,0xcb,0xb1,0x40,
0xb1,0x80,0x8c,0xf1,0x87,0xa4,0xf4,0xdf
];
cfb.init(&key, &iv).expect("Error with init()");
let mut outbuf: [u8; 48] = [0; 48];
cfb.encrypt(&msg[0..32], &mut outbuf[0..32]).expect("Error with encrypt()");
/* Break up encrypt over two operations to test continuation. */
cfb.encrypt(&msg[32..48], &mut outbuf[32..48]).expect("Error with encrypt()");
assert_eq!(outbuf, cipher);
cfb.init(&key, &iv).expect("Error with init()");
let mut plain: [u8; 48] = [0; 48];
cfb.decrypt(&outbuf, &mut plain).expect("Error with decrypt()");
assert_eq!(plain, msg);
}
#[test]
#[cfg(aes_cfb)]
fn test_cfb_big_msg() {
let mut cfb = CFB::new().expect("Failed to create CFB");
let big_key = b"0123456789abcdeffedcba9876543210";
let iv: [u8; 16] = [
0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,
0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f
];
let mut big_cipher: [u8; 384] = [0; 384];
let mut big_plain: [u8; 384] = [0; 384];
cfb.init(big_key, &iv).expect("Error with init()");
cfb.encrypt(&BIG_MSG, &mut big_cipher).expect("Error with encrypt()");
cfb.init(big_key, &iv).expect("Error with init()");
cfb.decrypt(&big_cipher, &mut big_plain).expect("Error with decrypt()");
assert_eq!(big_plain, BIG_MSG);
}
#[test]
#[cfg(aes_ctr)]
fn test_ctr_encrypt_decrypt() {
let iv: [u8; 16] = [
0xf0,0xf1,0xf2,0xf3,0xf4,0xf5,0xf6,0xf7,
0xf8,0xf9,0xfa,0xfb,0xfc,0xfd,0xfe,0xff
];
let msg: [u8; 64] = [
0x6b,0xc1,0xbe,0xe2,0x2e,0x40,0x9f,0x96,
0xe9,0x3d,0x7e,0x11,0x73,0x93,0x17,0x2a,
0xae,0x2d,0x8a,0x57,0x1e,0x03,0xac,0x9c,
0x9e,0xb7,0x6f,0xac,0x45,0xaf,0x8e,0x51,
0x30,0xc8,0x1c,0x46,0xa3,0x5c,0xe4,0x11,
0xe5,0xfb,0xc1,0x19,0x1a,0x0a,0x52,0xef,
0xf6,0x9f,0x24,0x45,0xdf,0x4f,0x9b,0x17,
0xad,0x2b,0x41,0x7b,0xe6,0x6c,0x37,0x10
];
let key: [u8; 16] = [
0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,
0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c
];
let cipher: [u8; 64] = [
0x87,0x4d,0x61,0x91,0xb6,0x20,0xe3,0x26,
0x1b,0xef,0x68,0x64,0x99,0x0d,0xb6,0xce,
0x98,0x06,0xf6,0x6b,0x79,0x70,0xfd,0xff,
0x86,0x17,0x18,0x7b,0xb9,0xff,0xfd,0xff,
0x5a,0xe4,0xdf,0x3e,0xdb,0xd5,0xd3,0x5e,
0x5b,0x4f,0x09,0x02,0x0d,0xb0,0x3e,0xab,
0x1e,0x03,0x1d,0xda,0x2f,0xbe,0x03,0xd1,
0x79,0x21,0x70,0xa0,0xf3,0x00,0x9c,0xee
];
let mut ctr = CTR::new().expect("Failed to create CTR");
ctr.init(&key, &iv).expect("Error with init()");
let mut outbuf: [u8; 64] = [0; 64];
ctr.encrypt(&msg, &mut outbuf).expect("Error with encrypt()");
assert_eq!(outbuf, cipher);
ctr.init(&key, &iv).expect("Error with init()");
let mut plain: [u8; 64] = [0; 64];
ctr.decrypt(&outbuf, &mut plain).expect("Error with decrypt()");
assert_eq!(plain, msg);
}
#[test]
#[cfg(aes_ctr)]
fn test_ctr_big_msg() {
let mut ctr = CTR::new().expect("Failed to create CTR");
let big_key = b"0123456789abcdeffedcba9876543210";
let iv: [u8; 16] = [
0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,
0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f
];
let mut big_cipher: [u8; 384] = [0; 384];
let mut big_plain: [u8; 384] = [0; 384];
ctr.init(big_key, &iv).expect("Error with init()");
ctr.encrypt(&BIG_MSG, &mut big_cipher).expect("Error with encrypt()");
ctr.init(big_key, &iv).expect("Error with init()");
ctr.decrypt(&big_cipher, &mut big_plain).expect("Error with decrypt()");
assert_eq!(big_plain, BIG_MSG);
}
#[test]
#[cfg(aes_eax)]
fn test_eax_one_shot_encrypt_decrypt() {
let key: [u8; 16] = [
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f
];
let nonce: [u8; 16] = [
0x3c, 0x8c, 0xc2, 0x97, 0x0a, 0x00, 0x8f, 0x75,
0xcc, 0x5b, 0xea, 0xe2, 0x84, 0x72, 0x58, 0xc2
];
let auth: &[u8] = &[];
let msg: [u8; 32] = [
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11,
0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11
];
let expected_cipher: [u8; 32] = [
0x3c, 0x44, 0x1f, 0x32, 0xce, 0x07, 0x82, 0x23,
0x64, 0xd7, 0xa2, 0x99, 0x0e, 0x50, 0xbb, 0x13,
0xd7, 0xb0, 0x2a, 0x26, 0x96, 0x9e, 0x4a, 0x93,
0x7e, 0x5e, 0x90, 0x73, 0xb0, 0xd9, 0xc9, 0x68
];
let expected_auth_tag: [u8; 16] = [
0xdb, 0x90, 0xbd, 0xb3, 0xda, 0x3d, 0x00, 0xaf,
0xd0, 0xfc, 0x6a, 0x83, 0x55, 0x1d, 0xa9, 0x5e
];
let mut cipher: [u8; 32] = [0; 32];
let mut auth_tag: [u8; 16] = [0; 16];
EAX::encrypt(&msg, &mut cipher, &key, &nonce, auth, &mut auth_tag).expect("Error with encrypt()");
assert_eq!(cipher, expected_cipher);
assert_eq!(auth_tag, expected_auth_tag);
let mut plain: [u8; 32] = [0; 32];
EAX::decrypt(&cipher, &mut plain, &key, &nonce, auth, &auth_tag).expect("Error with decrypt()");
assert_eq!(plain, msg);
}
#[test]
#[cfg(aes_ecb)]
fn test_ecb_encrypt_decrypt() {
let mut ecb = ECB::new().expect("Failed to create ECB");
let key_128: &[u8; 16] = b"0123456789abcdef";
let msg: [u8; 16] = [
0x6e, 0x6f, 0x77, 0x20, 0x69, 0x73, 0x20, 0x74,
0x68, 0x65, 0x20, 0x74, 0x69, 0x6d, 0x65, 0x20
];
let verify_ecb_128: [u8; 16] = [
0xd0, 0xc9, 0xd9, 0xc9, 0x40, 0xe8, 0x97, 0xb6,
0xc8, 0x8c, 0x33, 0x3b, 0xb5, 0x8f, 0x85, 0xd1
];
ecb.init_encrypt(key_128).expect("Error with init_encrypt()");
let mut outbuf: [u8; 16] = [0; 16];
ecb.encrypt(&msg, &mut outbuf).expect("Error with encrypt()");
assert_eq!(&outbuf, &verify_ecb_128);
outbuf = [0; 16];
ecb.init_decrypt(key_128).expect("Error with init_decrypt()");
ecb.decrypt(&verify_ecb_128, &mut outbuf).expect("Error with decrypt()");
assert_eq!(&outbuf, &msg);
}
#[test]
#[cfg(aes_gcm)]
fn test_gcm_encrypt_decrypt() {
let key: [u8; 16] = [
0x29, 0x8e, 0xfa, 0x1c, 0xcf, 0x29, 0xcf, 0x62,
0xae, 0x68, 0x24, 0xbf, 0xc1, 0x95, 0x57, 0xfc
];
let iv: [u8; 12] = [
0x6f, 0x58, 0xa9, 0x3f, 0xe1, 0xd2, 0x07, 0xfa,
0xe4, 0xed, 0x2f, 0x6d
];
let plain: [u8; 32] = [
0xcc, 0x38, 0xbc, 0xcd, 0x6b, 0xc5, 0x36, 0xad,
0x91, 0x9b, 0x13, 0x95, 0xf5, 0xd6, 0x38, 0x01,
0xf9, 0x9f, 0x80, 0x68, 0xd6, 0x5c, 0xa5, 0xac,
0x63, 0x87, 0x2d, 0xaf, 0x16, 0xb9, 0x39, 0x01
];
let auth: [u8; 16] = [
0x02, 0x1f, 0xaf, 0xd2, 0x38, 0x46, 0x39, 0x73,
0xff, 0xe8, 0x02, 0x56, 0xe5, 0xb1, 0xc6, 0xb1
];
let expected_cipher: [u8; 32] = [
0xdf, 0xce, 0x4e, 0x9c, 0xd2, 0x91, 0x10, 0x3d,
0x7f, 0xe4, 0xe6, 0x33, 0x51, 0xd9, 0xe7, 0x9d,
0x3d, 0xfd, 0x39, 0x1e, 0x32, 0x67, 0x10, 0x46,
0x58, 0x21, 0x2d, 0xa9, 0x65, 0x21, 0xb7, 0xdb
];
let expected_auth_tag: [u8; 16] = [
0x54, 0x24, 0x65, 0xef, 0x59, 0x93, 0x16, 0xf7,
0x3a, 0x7a, 0x56, 0x05, 0x09, 0xa2, 0xd9, 0xf2
];
let mut gcm = GCM::new().expect("Failed to create GCM");
gcm.init(&key).expect("Error with init()");
let mut cipher: [u8; 32] = [0; 32];
let mut auth_tag: [u8; 16] = [0; 16];
gcm.encrypt(&plain, &mut cipher, &iv, &auth, &mut auth_tag).expect("Error with encrypt()");
assert_eq!(cipher, expected_cipher);
assert_eq!(auth_tag, expected_auth_tag);
let mut plain_out: [u8; 32] = [0; 32];
gcm.decrypt(&cipher, &mut plain_out, &iv, &auth, &auth_tag).expect("Error with decrypt()");
assert_eq!(plain_out, plain);
}
#[test]
#[cfg(aes_gcm_stream)]
fn test_gcmstream_encrypt_decrypt() {
let plain: [u8; 60] = [
0xd9, 0x31, 0x32, 0x25, 0xf8, 0x84, 0x06, 0xe5,
0xa5, 0x59, 0x09, 0xc5, 0xaf, 0xf5, 0x26, 0x9a,
0x86, 0xa7, 0xa9, 0x53, 0x15, 0x34, 0xf7, 0xda,
0x2e, 0x4c, 0x30, 0x3d, 0x8a, 0x31, 0x8a, 0x72,
0x1c, 0x3c, 0x0c, 0x95, 0x95, 0x68, 0x09, 0x53,
0x2f, 0xcf, 0x0e, 0x24, 0x49, 0xa6, 0xb5, 0x25,
0xb1, 0x6a, 0xed, 0xf5, 0xaa, 0x0d, 0xe6, 0x57,
0xba, 0x63, 0x7b, 0x39
];
let auth: [u8; 20] = [
0xfe, 0xed, 0xfa, 0xce, 0xde, 0xad, 0xbe, 0xef,
0xfe, 0xed, 0xfa, 0xce, 0xde, 0xad, 0xbe, 0xef,
0xab, 0xad, 0xda, 0xd2
];
let key: [u8; 32] = [
0xfe, 0xff, 0xe9, 0x92, 0x86, 0x65, 0x73, 0x1c,
0x6d, 0x6a, 0x8f, 0x94, 0x67, 0x30, 0x83, 0x08,
0xfe, 0xff, 0xe9, 0x92, 0x86, 0x65, 0x73, 0x1c,
0x6d, 0x6a, 0x8f, 0x94, 0x67, 0x30, 0x83, 0x08
];
let iv: [u8; 12] = [
0xca, 0xfe, 0xba, 0xbe, 0xfa, 0xce, 0xdb, 0xad,
0xde, 0xca, 0xf8, 0x88
];
let expected_cipher: [u8; 60] = [
0x52, 0x2d, 0xc1, 0xf0, 0x99, 0x56, 0x7d, 0x07,
0xf4, 0x7f, 0x37, 0xa3, 0x2a, 0x84, 0x42, 0x7d,
0x64, 0x3a, 0x8c, 0xdc, 0xbf, 0xe5, 0xc0, 0xc9,
0x75, 0x98, 0xa2, 0xbd, 0x25, 0x55, 0xd1, 0xaa,
0x8c, 0xb0, 0x8e, 0x48, 0x59, 0x0d, 0xbb, 0x3d,
0xa7, 0xb0, 0x8b, 0x10, 0x56, 0x82, 0x88, 0x38,
0xc5, 0xf6, 0x1e, 0x63, 0x93, 0xba, 0x7a, 0x0a,
0xbc, 0xc9, 0xf6, 0x62
];
let expected_auth_tag: [u8; 16] = [
0x76, 0xfc, 0x6e, 0xce, 0x0f, 0x4e, 0x17, 0x68,
0xcd, 0xdf, 0x88, 0x53, 0xbb, 0x2d, 0x55, 0x1b
];
let mut gcmstream = GCMStream::new().expect("Failed to create GCMStream");
for chunk_size in 1..=auth.len() {
gcmstream.init(&key, &iv).expect("Error with init()");
let mut cipher: [u8; 60] = [0; 60];
let mut i = 0;
while i < auth.len() {
let mut end = i + chunk_size;
if end > auth.len() {
end = auth.len()
}
gcmstream.encrypt_update(&plain[0..0], &mut cipher[0..0], &auth[i..end]).expect("Error with encrypt_update()");
i += chunk_size;
}
i = 0;
while i < plain.len() {
let mut end = i + chunk_size;
if end > plain.len() {
end = plain.len()
}
gcmstream.encrypt_update(&plain[i..end], &mut cipher[i..end], &auth[0..0]).expect("Error with encrypt_update()");
i += chunk_size;
}
let mut auth_tag: [u8; 16] = [0; 16];
gcmstream.encrypt_final(&mut auth_tag).expect("Error with encrypt_final()");
assert_eq!(cipher, expected_cipher);
assert_eq!(auth_tag, expected_auth_tag);
}
}
#[test]
#[cfg(aes_ofb)]
fn test_ofb_encrypt_decrypt() {
let key: [u8; 32] = [
0xc4,0xc7,0xfa,0xd6,0x53,0x5c,0xb8,0x71,
0x4a,0x5c,0x40,0x77,0x9a,0x8b,0xa1,0xd2,
0x53,0x3e,0x23,0xb4,0xb2,0x58,0x73,0x2a,
0x5b,0x78,0x01,0xf4,0xe3,0x71,0xa7,0x94
];
let iv: [u8; 16] = [
0x5e,0xb9,0x33,0x13,0xb8,0x71,0xff,0x16,
0xb9,0x8a,0x9b,0xcb,0x43,0x33,0x0d,0x6f
];
let plain: [u8; 48] = [
0x6d,0x0b,0xb0,0x79,0x63,0x84,0x71,0xe9,
0x39,0xd4,0x53,0x14,0x86,0xc1,0x4c,0x25,
0x9a,0xee,0xc6,0xf3,0xc0,0x0d,0xfd,0xd6,
0xc0,0x50,0xa8,0xba,0xa8,0x20,0xdb,0x71,
0xcc,0x12,0x2c,0x4e,0x0c,0x17,0x15,0xef,
0x55,0xf3,0x99,0x5a,0x6b,0xf0,0x2a,0x4c
];
let expected_cipher: [u8; 48] = [
0x0f,0x54,0x61,0x71,0x59,0xd0,0x3f,0xfc,
0x1b,0xfa,0xfb,0x60,0x29,0x30,0xd7,0x00,
0xf4,0xa4,0xa8,0xe6,0xdd,0x93,0x94,0x46,
0x64,0xd2,0x19,0xc4,0xc5,0x4d,0xde,0x1b,
0x04,0x53,0xe1,0x73,0xf5,0x18,0x74,0xae,
0xfd,0x64,0xa2,0xe1,0xe2,0x76,0x13,0xb0
];
let mut ofb = OFB::new().expect("Failed to create OFB");
ofb.init(&key, &iv).expect("Error with init()");
let mut cipher: [u8; 48] = [0; 48];
ofb.encrypt(&plain, &mut cipher).expect("Error with encrypt()");
assert_eq!(cipher, expected_cipher);
ofb.init(&key, &iv).expect("Error with init()");
let mut plain_out: [u8; 48] = [0; 48];
ofb.decrypt(&cipher, &mut plain_out).expect("Error with decrypt()");
assert_eq!(plain_out, plain);
}
#[test]
#[cfg(aes_xts)]
fn test_xts_one_shot() {
let key: [u8; 32] = [
0xa1, 0xb9, 0x0c, 0xba, 0x3f, 0x06, 0xac, 0x35,
0x3b, 0x2c, 0x34, 0x38, 0x76, 0x08, 0x17, 0x62,
0x09, 0x09, 0x23, 0x02, 0x6e, 0x91, 0x77, 0x18,
0x15, 0xf2, 0x9d, 0xab, 0x01, 0x93, 0x2f, 0x2f
];
let tweak: [u8; 16] = [
0x4f, 0xae, 0xf7, 0x11, 0x7c, 0xda, 0x59, 0xc6,
0x6e, 0x4b, 0x92, 0x01, 0x3e, 0x76, 0x8a, 0xd5
];
let plain: [u8; 16] = [
0xeb, 0xab, 0xce, 0x95, 0xb1, 0x4d, 0x3c, 0x8d,
0x6f, 0xb3, 0x50, 0x39, 0x07, 0x90, 0x31, 0x1c
];
let expected_cipher: [u8; 16] = [
0x77, 0x8a, 0xe8, 0xb4, 0x3c, 0xb9, 0x8d, 0x5a,
0x82, 0x50, 0x81, 0xd5, 0xbe, 0x47, 0x1c, 0x63
];
let partial: [u8; 24] = [
0xeb, 0xab, 0xce, 0x95, 0xb1, 0x4d, 0x3c, 0x8d,
0x6f, 0xb3, 0x50, 0x39, 0x07, 0x90, 0x31, 0x1c,
0x6e, 0x4b, 0x92, 0x01, 0x3e, 0x76, 0x8a, 0xd5
];
let expected_partial_cipher: [u8; 24] = [
0x2b, 0xf7, 0x2c, 0xf3, 0xeb, 0x85, 0xef, 0x7b,
0x0b, 0x76, 0xa0, 0xaa, 0xf3, 0x3f, 0x25, 0x8b,
0x77, 0x8a, 0xe8, 0xb4, 0x3c, 0xb9, 0x8d, 0x5a
];
let mut xts = XTS::new().expect("Failed to create XTS");
xts.init_encrypt(&key).expect("Error with init_encrypt()");
let mut cipher: [u8; 16] = [0; 16];
xts.encrypt(&plain, &mut cipher, &tweak).expect("Error with encrypt()");
assert_eq!(cipher, expected_cipher);
xts.init_decrypt(&key).expect("Error with init_decrypt()");
let mut plain_out: [u8; 16] = [0; 16];
xts.decrypt(&cipher, &mut plain_out, &tweak).expect("Error with decrypt()");
assert_eq!(plain_out, plain);
xts.init_encrypt(&key).expect("Error with init_encrypt()");
let mut partial_cipher: [u8; 24] = [0; 24];
xts.encrypt(&partial, &mut partial_cipher, &tweak).expect("Error with encrypt()");
assert_eq!(partial_cipher, expected_partial_cipher);
xts.init_decrypt(&key).expect("Error with init_decrypt()");
let mut partial_out: [u8; 24] = [0; 24];
xts.decrypt(&partial_cipher, &mut partial_out, &tweak).expect("Error with decrypt()");
assert_eq!(partial_out, partial);
}
#[test]
#[cfg(aes_xts)]
fn test_xts_sector_128() {
let keys: [u8; 32] = [
0xa3, 0xe4, 0x0d, 0x5b, 0xd4, 0xb6, 0xbb, 0xed,
0xb2, 0xd1, 0x8c, 0x70, 0x0a, 0xd2, 0xdb, 0x22,
0x10, 0xc8, 0x11, 0x90, 0x64, 0x6d, 0x67, 0x3c,
0xbc, 0xa5, 0x3f, 0x13, 0x3e, 0xab, 0x37, 0x3c
];
let plain: [u8; 16] =[
0x20, 0xe0, 0x71, 0x94, 0x05, 0x99, 0x3f, 0x09,
0xa6, 0x6a, 0xe5, 0xbb, 0x50, 0x0e, 0x56, 0x2c
];
let expected_cipher: [u8; 16] = [
0x74, 0x62, 0x35, 0x51, 0x21, 0x02, 0x16, 0xac,
0x92, 0x6b, 0x96, 0x50, 0xb6, 0xd3, 0xfa, 0x52
];
let sector = 141;
let mut xts = XTS::new().expect("Failed to create XTS");
xts.init_encrypt(&keys).expect("Error with init_encrypt()");
let mut cipher: [u8; 16] = [0; 16];
xts.encrypt_sector(&plain, &mut cipher, sector).expect("Error with encrypt_sector()");
assert_eq!(cipher, expected_cipher);
xts.init_decrypt(&keys).expect("Error with init_decrypt()");
let mut plain_out: [u8; 16] = [0; 16];
xts.decrypt_sector(&cipher, &mut plain_out, sector).expect("Error with decrypt_sector()");
assert_eq!(plain_out, plain);
}
#[test]
#[cfg(aes_xts)]
fn test_xts_sector_256() {
let keys: [u8; 64] = [
0xef, 0x01, 0x0c, 0xa1, 0xa3, 0x66, 0x3e, 0x32,
0x53, 0x43, 0x49, 0xbc, 0x0b, 0xae, 0x62, 0x23,
0x2a, 0x15, 0x73, 0x34, 0x85, 0x68, 0xfb, 0x9e,
0xf4, 0x17, 0x68, 0xa7, 0x67, 0x4f, 0x50, 0x7a,
0x72, 0x7f, 0x98, 0x75, 0x53, 0x97, 0xd0, 0xe0,
0xaa, 0x32, 0xf8, 0x30, 0x33, 0x8c, 0xc7, 0xa9,
0x26, 0xc7, 0x73, 0xf0, 0x9e, 0x57, 0xb3, 0x57,
0xcd, 0x15, 0x6a, 0xfb, 0xca, 0x46, 0xe1, 0xa0
];
let plain: [u8; 32] =[
0xed, 0x98, 0xe0, 0x17, 0x70, 0xa8, 0x53, 0xb4,
0x9d, 0xb9, 0xe6, 0xaa, 0xf8, 0x8f, 0x0a, 0x41,
0xb9, 0xb5, 0x6e, 0x91, 0xa5, 0xa2, 0xb1, 0x1d,
0x40, 0x52, 0x92, 0x54, 0xf5, 0x52, 0x3e, 0x75
];
let expected_cipher: [u8; 32] = [
0xca, 0x20, 0xc5, 0x5e, 0x8d, 0xc1, 0x49, 0x68,
0x7d, 0x25, 0x41, 0xde, 0x39, 0xc3, 0xdf, 0x63,
0x00, 0xbb, 0x5a, 0x16, 0x3c, 0x10, 0xce, 0xd3,
0x66, 0x6b, 0x13, 0x57, 0xdb, 0x8b, 0xd3, 0x9d
];
let sector = 187;
let mut xts = XTS::new().expect("Failed to create XTS");
xts.init_encrypt(&keys).expect("Error with init_encrypt()");
let mut cipher: [u8; 32] = [0; 32];
xts.encrypt_sector(&plain, &mut cipher, sector).expect("Error with encrypt_sector()");
assert_eq!(cipher, expected_cipher);
xts.init_decrypt(&keys).expect("Error with init_decrypt()");
let mut plain_out: [u8; 32] = [0; 32];
xts.decrypt_sector(&cipher, &mut plain_out, sector).expect("Error with decrypt_sector()");
assert_eq!(plain_out, plain);
}
#[test]
#[cfg(aes_xts)]
fn test_xts_consecutive_sectors() {
let keys: [u8; 32] = [
0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11,
0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11,
0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22,
0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22
];
let plain: [u8; 544] =[
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b,
0x0c, 0x0d, 0x0e, 0x0f, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f, 0x20, 0x21, 0x22, 0x23,
0x24, 0x25, 0x26, 0x27, 0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f,
0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3a, 0x3b,
0x3c, 0x3d, 0x3e, 0x3f, 0x40, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47,
0x48, 0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f, 0x50, 0x51, 0x52, 0x53,
0x54, 0x55, 0x56, 0x57, 0x58, 0x59, 0x5a, 0x5b, 0x5c, 0x5d, 0x5e, 0x5f,
0x60, 0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68, 0x69, 0x6a, 0x6b,
0x6c, 0x6d, 0x6e, 0x6f, 0x70, 0x71, 0x72, 0x73, 0x74, 0x75, 0x76, 0x77,
0x78, 0x79, 0x7a, 0x7b, 0x7c, 0x7d, 0x7e, 0x7f, 0x80, 0x81, 0x82, 0x83,
0x84, 0x85, 0x86, 0x87, 0x88, 0x89, 0x8a, 0x8b, 0x8c, 0x8d, 0x8e, 0x8f,
0x90, 0x91, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, 0x99, 0x9a, 0x9b,
0x9c, 0x9d, 0x9e, 0x9f, 0xa0, 0xa1, 0xa2, 0xa3, 0xa4, 0xa5, 0xa6, 0xa7,
0xa8, 0xa9, 0xaa, 0xab, 0xac, 0xad, 0xae, 0xaf, 0xb0, 0xb1, 0xb2, 0xb3,
0xb4, 0xb5, 0xb6, 0xb7, 0xb8, 0xb9, 0xba, 0xbb, 0xbc, 0xbd, 0xbe, 0xbf,
0xc0, 0xc1, 0xc2, 0xc3, 0xc4, 0xc5, 0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xcb,
0xcc, 0xcd, 0xce, 0xcf, 0xd0, 0xd1, 0xd2, 0xd3, 0xd4, 0xd5, 0xd6, 0xd7,
0xd8, 0xd9, 0xda, 0xdb, 0xdc, 0xdd, 0xde, 0xdf, 0xe0, 0xe1, 0xe2, 0xe3,
0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea, 0xeb, 0xec, 0xed, 0xee, 0xef,
0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8, 0xf9, 0xfa, 0xfb,
0xfc, 0xfd, 0xfe, 0xff,
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b,
0x0c, 0x0d, 0x0e, 0x0f, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f, 0x20, 0x21, 0x22, 0x23,
0x24, 0x25, 0x26, 0x27, 0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f,
0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3a, 0x3b,
0x3c, 0x3d, 0x3e, 0x3f, 0x40, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47,
0x48, 0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f, 0x50, 0x51, 0x52, 0x53,
0x54, 0x55, 0x56, 0x57, 0x58, 0x59, 0x5a, 0x5b, 0x5c, 0x5d, 0x5e, 0x5f,
0x60, 0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68, 0x69, 0x6a, 0x6b,
0x6c, 0x6d, 0x6e, 0x6f, 0x70, 0x71, 0x72, 0x73, 0x74, 0x75, 0x76, 0x77,
0x78, 0x79, 0x7a, 0x7b, 0x7c, 0x7d, 0x7e, 0x7f, 0x80, 0x81, 0x82, 0x83,
0x84, 0x85, 0x86, 0x87, 0x88, 0x89, 0x8a, 0x8b, 0x8c, 0x8d, 0x8e, 0x8f,
0x90, 0x91, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, 0x99, 0x9a, 0x9b,
0x9c, 0x9d, 0x9e, 0x9f, 0xa0, 0xa1, 0xa2, 0xa3, 0xa4, 0xa5, 0xa6, 0xa7,
0xa8, 0xa9, 0xaa, 0xab, 0xac, 0xad, 0xae, 0xaf, 0xb0, 0xb1, 0xb2, 0xb3,
0xb4, 0xb5, 0xb6, 0xb7, 0xb8, 0xb9, 0xba, 0xbb, 0xbc, 0xbd, 0xbe, 0xbf,
0xc0, 0xc1, 0xc2, 0xc3, 0xc4, 0xc5, 0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xcb,
0xcc, 0xcd, 0xce, 0xcf, 0xd0, 0xd1, 0xd2, 0xd3, 0xd4, 0xd5, 0xd6, 0xd7,
0xd8, 0xd9, 0xda, 0xdb, 0xdc, 0xdd, 0xde, 0xdf, 0xe0, 0xe1, 0xe2, 0xe3,
0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea, 0xeb, 0xec, 0xed, 0xee, 0xef,
0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8, 0xf9, 0xfa, 0xfb,
0xfc, 0xfd, 0xfe, 0xff,
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b,
0x0c, 0x0d, 0x0e, 0x0f, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f
];
let expected_cipher: [u8; 544] = [
0xb9, 0x6b, 0x2b, 0xfd, 0x61, 0x87, 0x84, 0xd5, 0x26, 0xd2, 0x8c, 0x62,
0x63, 0x01, 0xca, 0x46, 0xb1, 0x82, 0xfa, 0xdc, 0xbc, 0x32, 0x18, 0xe9,
0xda, 0xe6, 0xda, 0xd1, 0x1a, 0x52, 0x77, 0xca, 0xdb, 0x0e, 0xbe, 0x37,
0x88, 0x36, 0x1c, 0x87, 0x16, 0x60, 0xfe, 0xa8, 0x9e, 0xf6, 0x48, 0x64,
0x94, 0x34, 0x64, 0xed, 0xf6, 0x9a, 0xc5, 0x28, 0xc9, 0xed, 0x64, 0x80,
0x85, 0xd8, 0x93, 0xa7, 0x50, 0xb1, 0x9d, 0x2f, 0x1e, 0x34, 0xcc, 0xb4,
0x03, 0xfb, 0x6b, 0x43, 0x21, 0xa8, 0x5b, 0xc6, 0x59, 0x13, 0xd2, 0xb5,
0xf5, 0x7b, 0xf6, 0xb2, 0xa4, 0x7a, 0xd2, 0x50, 0x26, 0xcb, 0xa4, 0x83,
0xc3, 0x56, 0xb0, 0xb1, 0x14, 0x34, 0x12, 0x1b, 0xea, 0x26, 0x97, 0x24,
0x54, 0xcc, 0x32, 0x4c, 0xa4, 0xc2, 0xa3, 0x07, 0xfa, 0x30, 0xa9, 0xf0,
0x91, 0x17, 0x60, 0x68, 0x88, 0x7f, 0x34, 0x7e, 0xbd, 0x20, 0x33, 0x95,
0x6e, 0xc0, 0xb6, 0x2b, 0xff, 0x7e, 0x61, 0x35, 0x9a, 0x88, 0xff, 0xd9,
0x69, 0x21, 0xe7, 0x8f, 0x45, 0x02, 0xf9, 0xd7, 0xeb, 0xa6, 0x53, 0xf1,
0x73, 0x04, 0xf1, 0x0b, 0x85, 0xc6, 0x1f, 0x4a, 0x51, 0x2f, 0x95, 0x87,
0x5a, 0x67, 0x37, 0xb2, 0x87, 0xf7, 0xbe, 0x2a, 0x17, 0x57, 0xca, 0xfc,
0xdd, 0x5f, 0x37, 0x48, 0x78, 0xbd, 0xfa, 0x75, 0xc9, 0xfa, 0x86, 0x7e,
0xc4, 0x0f, 0x60, 0x85, 0xce, 0x12, 0x44, 0x7c, 0xd9, 0xb2, 0x50, 0xd9,
0x57, 0x85, 0xa5, 0xd7, 0x68, 0x59, 0x03, 0x09, 0x97, 0x2e, 0x8e, 0xa5,
0xe3, 0x98, 0xac, 0x16, 0xfb, 0x6d, 0x54, 0xc5, 0x5d, 0x7a, 0x33, 0x44,
0x0a, 0x39, 0x91, 0xcc, 0x9f, 0x67, 0xf9, 0x89, 0xbb, 0x62, 0x02, 0xc4,
0x22, 0xec, 0xcf, 0x97, 0x69, 0x81, 0x3d, 0x00, 0xfd, 0xeb, 0x55, 0x08,
0xa2, 0xff, 0x97, 0xaa, 0x79, 0xde, 0x3c, 0x8a, 0x78, 0x71, 0x73, 0xa2,
0x98, 0x2f, 0xd8, 0x5c, 0x62, 0x1c, 0x5c, 0x23, 0x0a, 0xd1, 0xf1, 0x81,
0x8a, 0x12, 0xe7, 0x4d, 0xdd, 0x4f, 0xd4, 0xf1, 0xe8, 0x0f, 0x25, 0x79,
0x45, 0x4a, 0x49, 0x49, 0x7e, 0x56, 0x91, 0x4e, 0xaa, 0xba, 0x18, 0xe1,
0xe4, 0xbe, 0x21, 0xdc, 0x58, 0x60, 0x6f, 0x6a, 0x7f, 0xdc, 0x5e, 0x74,
0x47, 0xbf, 0xeb, 0x84, 0xc4, 0x1e, 0x5a, 0x61, 0x64, 0xc8, 0x63, 0x68,
0xfa, 0x17, 0x9c, 0xac, 0x60, 0x1c, 0xa5, 0x6e, 0x00, 0x21, 0x93, 0x3c,
0xd7, 0xbb, 0x73, 0x45, 0xf7, 0x34, 0x81, 0x6c, 0xfa, 0xf2, 0x33, 0xfd,
0xb1, 0x40, 0x30, 0x6b, 0x30, 0xd1, 0x83, 0x5e, 0x2e, 0x7a, 0xce, 0xa6,
0x12, 0x2a, 0x15, 0x03, 0x78, 0x29, 0xb9, 0x07, 0xae, 0xe7, 0xc2, 0x78,
0x74, 0x72, 0xa5, 0x0e, 0x6b, 0x1f, 0x78, 0xf2, 0x5a, 0x69, 0xb6, 0x2b,
0x99, 0x94, 0x1f, 0x89, 0xd1, 0x21, 0x14, 0x4a, 0x54, 0xab, 0x5a, 0x9f,
0xaa, 0xa7, 0x96, 0x0a, 0x21, 0xce, 0x30, 0xb6, 0x70, 0x81, 0xe9, 0xd3,
0x71, 0xc0, 0xf1, 0x15, 0xe2, 0xf6, 0xd3, 0xcc, 0x41, 0x15, 0x9d, 0xd5,
0xa3, 0xa4, 0xe0, 0xf8, 0x62, 0xc4, 0x76, 0x65, 0x63, 0x89, 0xa7, 0xe2,
0xfb, 0xf5, 0xc9, 0x80, 0x15, 0x5b, 0xc1, 0x59, 0xb2, 0xd0, 0x01, 0x3a,
0xf9, 0xab, 0x5b, 0x79, 0x54, 0xed, 0x6b, 0xf9, 0x1d, 0x9d, 0x87, 0x63,
0x80, 0x4f, 0xec, 0x9c, 0x4f, 0xad, 0x97, 0x04, 0xff, 0x62, 0x4a, 0x17,
0xc0, 0x09, 0x2a, 0x2c, 0x23, 0x4b, 0xc3, 0xb6, 0x6d, 0xed, 0xdb, 0x1a,
0x6f, 0x56, 0x2b, 0x78, 0x92, 0x3a, 0x5c, 0x7f, 0xb2, 0x63, 0xd3, 0xd5,
0x1a, 0xbe, 0xc2, 0x34, 0xc8, 0xad, 0x36, 0xb7, 0x12, 0xb8, 0xe1, 0xb7,
0x52, 0x7f, 0x16, 0x84, 0x2c, 0x47, 0x7e, 0xf2, 0xa5, 0x36, 0x2e, 0xad,
0xe7, 0xbb, 0xc0, 0x6f, 0x27, 0x8e, 0x41, 0x08, 0x75, 0xe5, 0xff, 0xde,
0x08, 0x9f, 0x8c, 0x91, 0xba, 0xc9, 0x9d, 0x9f, 0x27, 0x90, 0x50, 0x44,
0x24, 0xe7, 0x3d, 0x6f
];
let sector = 0x000000ffffffffff;
let sector_size = 512;
let mut xts = XTS::new().expect("Failed to create XTS");
xts.init_encrypt(&keys).expect("Error with init_encrypt()");
let mut cipher: [u8; 544] = [0; 544];
xts.encrypt_consecutive_sectors(&plain, &mut cipher, sector, sector_size).expect("Error with encrypt_consecutive_sectors()");
assert_eq!(cipher, expected_cipher);
xts.init_decrypt(&keys).expect("Error with init_decrypt()");
let mut plain_out: [u8; 544] = [0; 544];
xts.decrypt_consecutive_sectors(&cipher, &mut plain_out, sector, sector_size).expect("Error with decrypt_consecutive_sectors()");
assert_eq!(plain_out, plain);
}
#[test]
#[cfg(aes_xts_stream)]
fn test_xtsstream() {
let keys: [u8; 32] = [
0x39, 0x25, 0x79, 0x05, 0xdf, 0xcc, 0x77, 0x76,
0x6c, 0x87, 0x0a, 0x80, 0x6a, 0x60, 0xe3, 0xc0,
0x93, 0xd1, 0x2a, 0xcf, 0xcb, 0x51, 0x42, 0xfa,
0x09, 0x69, 0x89, 0x62, 0x5b, 0x60, 0xdb, 0x16
];
let tweak: [u8; 16] = [
0x5c, 0xf7, 0x9d, 0xb6, 0xc5, 0xcd, 0x99, 0x1a,
0x1c, 0x78, 0x81, 0x42, 0x24, 0x95, 0x1e, 0x84
];
let plain: [u8; 32] = [
0xbd, 0xc5, 0x46, 0x8f, 0xbc, 0x8d, 0x50, 0xa1,
0x0d, 0x1c, 0x85, 0x7f, 0x79, 0x1c, 0x5c, 0xba,
0xb3, 0x81, 0x0d, 0x0d, 0x73, 0xcf, 0x8f, 0x20,
0x46, 0xb1, 0xd1, 0x9e, 0x7d, 0x5d, 0x8a, 0x56
];
let expected_cipher: [u8; 32] = [
0xd6, 0xbe, 0x04, 0x6d, 0x41, 0xf2, 0x3b, 0x5e,
0xd7, 0x0b, 0x6b, 0x3d, 0x5c, 0x8e, 0x66, 0x23,
0x2b, 0xe6, 0xb8, 0x07, 0xd4, 0xdc, 0xc6, 0x0e,
0xff, 0x8d, 0xbc, 0x1d, 0x9f, 0x7f, 0xc8, 0x22
];
let mut xtsstream = XTSStream::new().expect("Failed to create XTSStream");
xtsstream.init_encrypt(&keys, &tweak).expect("Error with init_encrypt()");
let mut cipher: [u8; 32] = [0; 32];
xtsstream.encrypt_update(&plain[0..16], &mut cipher[0..16]).expect("Error with encrypt_update()");
xtsstream.encrypt_final(&plain[16..32], &mut cipher[16..32]).expect("Error with encrypt_final()");
assert_eq!(cipher, expected_cipher);
xtsstream.init_decrypt(&keys, &tweak).expect("Error with init_decrypt()");
let mut plain_out: [u8; 32] = [0; 32];
xtsstream.decrypt_update(&cipher[0..16], &mut plain_out[0..16]).expect("Error with decrypt_update()");
xtsstream.decrypt_final(&cipher[16..32], &mut plain_out[16..32]).expect("Error with decrypt_final()");
assert_eq!(plain_out, plain);
}
#[test]
#[cfg(aes_xts_stream)]
fn test_xtsstream_big_msg() {
let key: [u8; 32] = [
0xa1, 0xb9, 0x0c, 0xba, 0x3f, 0x06, 0xac, 0x35,
0x3b, 0x2c, 0x34, 0x38, 0x76, 0x08, 0x17, 0x62,
0x09, 0x09, 0x23, 0x02, 0x6e, 0x91, 0x77, 0x18,
0x15, 0xf2, 0x9d, 0xab, 0x01, 0x93, 0x2f, 0x2f
];
let tweak: [u8; 16] = [
0x4f, 0xae, 0xf7, 0x11, 0x7c, 0xda, 0x59, 0xc6,
0x6e, 0x4b, 0x92, 0x01, 0x3e, 0x76, 0x8a, 0xd5
];
let mut xtsstream = XTSStream::new().expect("Failed to create XTSStream");
xtsstream.init_encrypt(&key, &tweak).expect("Error with init_encrypt()");
let mut cipher: [u8; 384] = [0; 384];
let mut i = 0;
while i < BIG_MSG.len() {
let remaining = BIG_MSG.len() - i;
if remaining < 32 {
xtsstream.encrypt_final(&BIG_MSG[i..BIG_MSG.len()], &mut cipher[i..BIG_MSG.len()]).expect("Error with encrypt_final()");
i += remaining;
} else {
let end = i + 16;
xtsstream.encrypt_update(&BIG_MSG[i..end], &mut cipher[i..end]).expect("Error with encrypt_update()");
i += 16;
}
}
xtsstream.init_decrypt(&key, &tweak).expect("Error with init_decrypt()");
let mut plain_out: [u8; 384] = [0; 384];
let mut i = 0;
while i < BIG_MSG.len() {
let remaining = BIG_MSG.len() - i;
if remaining < 32 {
xtsstream.decrypt_final(&cipher[i..BIG_MSG.len()], &mut plain_out[i..BIG_MSG.len()]).expect("Error with decrypt_final()");
i += remaining;
} else {
let end = i + 16;
xtsstream.decrypt_update(&cipher[i..end], &mut plain_out[i..end]).expect("Error with decrypt_update()");
i += 16;
}
}
assert_eq!(plain_out, BIG_MSG);
}
// ---------------------------------------------------------------------------
// AES aead trait implementations
// ---------------------------------------------------------------------------
#[cfg(feature = "aead")]
use aead::{Aead, AeadInPlace, KeyInit, Payload};
/// NIST SP 800-38D, Test Case 2:
/// Key = 00000000000000000000000000000000
/// IV = 000000000000000000000000
/// PT = 00000000000000000000000000000000
/// AAD = (empty)
/// CT = 0388dace60b6a392f328c2b971b2fe78
/// Tag = ab6e47d42cec13bdf53a67b21257bddf
#[test]
#[cfg(all(feature = "aead", aes_gcm))]
fn test_aes128gcm_nist_tc2_encrypt() {
let key = [0u8; 16];
let nonce = [0u8; 12];
let expected_ciphertext = [
0x03u8, 0x88, 0xda, 0xce, 0x60, 0xb6, 0xa3, 0x92,
0xf3, 0x28, 0xc2, 0xb9, 0x71, 0xb2, 0xfe, 0x78,
];
let expected_tag = [
0xabu8, 0x6e, 0x47, 0xd4, 0x2c, 0xec, 0x13, 0xbd,
0xf5, 0x3a, 0x67, 0xb2, 0x12, 0x57, 0xbd, 0xdf,
];
let cipher = Aes128Gcm::new_from_slice(&key).unwrap();
let nonce_arr: aead::Nonce<Aes128Gcm> = nonce.into();
let mut buffer = [0u8; 16];
let tag = cipher
.encrypt_in_place_detached(&nonce_arr, &[], &mut buffer)
.expect("AES-128-GCM encrypt failed");
assert_eq!(buffer, expected_ciphertext);
assert_eq!(&tag[..], &expected_tag);
}
#[test]
#[cfg(all(feature = "aead", aes_gcm))]
fn test_aes128gcm_nist_tc2_decrypt() {
let key = [0u8; 16];
let nonce = [0u8; 12];
let mut ciphertext = [
0x03u8, 0x88, 0xda, 0xce, 0x60, 0xb6, 0xa3, 0x92,
0xf3, 0x28, 0xc2, 0xb9, 0x71, 0xb2, 0xfe, 0x78,
];
let tag_bytes = [
0xabu8, 0x6e, 0x47, 0xd4, 0x2c, 0xec, 0x13, 0xbd,
0xf5, 0x3a, 0x67, 0xb2, 0x12, 0x57, 0xbd, 0xdf,
];
let cipher = Aes128Gcm::new_from_slice(&key).unwrap();
let nonce_arr: aead::Nonce<Aes128Gcm> = nonce.into();
let tag: aead::Tag<Aes128Gcm> = tag_bytes.into();
cipher
.decrypt_in_place_detached(&nonce_arr, &[], &mut ciphertext, &tag)
.expect("AES-128-GCM decrypt failed");
assert_eq!(ciphertext, [0u8; 16]);
}
/// Test AES-128-GCM roundtrip using the `aead::Aead` blanket impl.
#[test]
#[cfg(all(feature = "aead", aes_gcm))]
fn test_aes128gcm_aead_roundtrip() {
let key = [0x42u8; 16];
let nonce_bytes = [0x11u8; 12];
let aad = b"associated data";
let plaintext = b"Hello, AEAD world!";
let cipher = Aes128Gcm::new_from_slice(&key).unwrap();
let nonce: aead::Nonce<Aes128Gcm> = nonce_bytes.into();
let ciphertext = cipher
.encrypt(&nonce, Payload { msg: plaintext, aad })
.expect("AES-128-GCM Aead::encrypt failed");
let recovered = cipher
.decrypt(&nonce, Payload { msg: &ciphertext, aad })
.expect("AES-128-GCM Aead::decrypt failed");
assert_eq!(recovered, plaintext);
}
/// Verify that decryption rejects a tampered tag.
#[test]
#[cfg(all(feature = "aead", aes_gcm))]
fn test_aes128gcm_reject_bad_tag() {
let key = [0u8; 16];
let nonce_bytes = [0u8; 12];
let plaintext = b"some plaintext!";
let cipher = Aes128Gcm::new_from_slice(&key).unwrap();
let nonce: aead::Nonce<Aes128Gcm> = nonce_bytes.into();
let mut ct = cipher.encrypt(&nonce, plaintext.as_ref()).expect("encrypt failed");
let last = ct.len() - 1;
ct[last] ^= 0xff;
assert!(cipher.decrypt(&nonce, ct.as_slice()).is_err());
}
/// NIST SP 800-38D, Test Case 14 (256-bit key):
/// Key = feffe9928665731c6d6a8f9467308308feffe9928665731c6d6a8f9467308308
/// IV = cafebabefacedbaddecaf888
/// PT = d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a7
/// 21c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b39 (60 B)
/// AAD = feedfacedeadbeeffeedfacedeadbeefabaddad2
/// CT = 522dc1f099567d07f47f37a32a84427d643a8cdcbfe5c0c97598a2bd2555d1a
/// a8cb08e48590dbb3da7b08b1056828838c5f61e6393ba7a0a (60 B)
/// Tag = 76fc6ece0f4e1768cddf8853bb2d551b
#[test]
#[cfg(all(feature = "aead", aes_gcm))]
fn test_aes256gcm_nist_tc14_encrypt() {
let key = [
0xfeu8, 0xff, 0xe9, 0x92, 0x86, 0x65, 0x73, 0x1c,
0x6d, 0x6a, 0x8f, 0x94, 0x67, 0x30, 0x83, 0x08,
0xfe, 0xff, 0xe9, 0x92, 0x86, 0x65, 0x73, 0x1c,
0x6d, 0x6a, 0x8f, 0x94, 0x67, 0x30, 0x83, 0x08,
];
let nonce = [
0xcau8, 0xfe, 0xba, 0xbe, 0xfa, 0xce, 0xdb, 0xad,
0xde, 0xca, 0xf8, 0x88,
];
let aad = [
0xfeu8, 0xed, 0xfa, 0xce, 0xde, 0xad, 0xbe, 0xef,
0xfe, 0xed, 0xfa, 0xce, 0xde, 0xad, 0xbe, 0xef,
0xab, 0xad, 0xda, 0xd2,
];
let plaintext = [
0xd9u8, 0x31, 0x32, 0x25, 0xf8, 0x84, 0x06, 0xe5,
0xa5, 0x59, 0x09, 0xc5, 0xaf, 0xf5, 0x26, 0x9a,
0x86, 0xa7, 0xa9, 0x53, 0x15, 0x34, 0xf7, 0xda,
0x2e, 0x4c, 0x30, 0x3d, 0x8a, 0x31, 0x8a, 0x72,
0x1c, 0x3c, 0x0c, 0x95, 0x95, 0x68, 0x09, 0x53,
0x2f, 0xcf, 0x0e, 0x24, 0x49, 0xa6, 0xb5, 0x25,
0xb1, 0x6a, 0xed, 0xf5, 0xaa, 0x0d, 0xe6, 0x57,
0xba, 0x63, 0x7b, 0x39,
];
let expected_ciphertext = [
0x52u8, 0x2d, 0xc1, 0xf0, 0x99, 0x56, 0x7d, 0x07,
0xf4, 0x7f, 0x37, 0xa3, 0x2a, 0x84, 0x42, 0x7d,
0x64, 0x3a, 0x8c, 0xdc, 0xbf, 0xe5, 0xc0, 0xc9,
0x75, 0x98, 0xa2, 0xbd, 0x25, 0x55, 0xd1, 0xaa,
0x8c, 0xb0, 0x8e, 0x48, 0x59, 0x0d, 0xbb, 0x3d,
0xa7, 0xb0, 0x8b, 0x10, 0x56, 0x82, 0x88, 0x38,
0xc5, 0xf6, 0x1e, 0x63, 0x93, 0xba, 0x7a, 0x0a,
0xbc, 0xc9, 0xf6, 0x62,
];
let expected_tag = [
0x76u8, 0xfc, 0x6e, 0xce, 0x0f, 0x4e, 0x17, 0x68,
0xcd, 0xdf, 0x88, 0x53, 0xbb, 0x2d, 0x55, 0x1b,
];
let cipher = Aes256Gcm::new_from_slice(&key).unwrap();
let nonce_arr: aead::Nonce<Aes256Gcm> = nonce.into();
let mut buffer = plaintext;
let tag = cipher
.encrypt_in_place_detached(&nonce_arr, &aad, &mut buffer)
.expect("AES-256-GCM encrypt failed");
assert_eq!(buffer, expected_ciphertext);
assert_eq!(&tag[..], &expected_tag);
}
/// Roundtrip test for AES-256-GCM using `aead::Aead`.
#[test]
#[cfg(all(feature = "aead", aes_gcm))]
fn test_aes256gcm_aead_roundtrip() {
let key = [0xabu8; 32];
let nonce_bytes = [0xbcu8; 12];
let aad = b"test aad";
let plaintext = b"AES-256-GCM roundtrip test";
let cipher = Aes256Gcm::new_from_slice(&key).unwrap();
let nonce: aead::Nonce<Aes256Gcm> = nonce_bytes.into();
let ciphertext = cipher
.encrypt(&nonce, Payload { msg: plaintext, aad })
.expect("encrypt failed");
let recovered = cipher
.decrypt(&nonce, Payload { msg: &ciphertext, aad })
.expect("decrypt failed");
assert_eq!(recovered, plaintext);
}
/// Roundtrip test for AES-128-CCM using `aead::Aead`.
#[test]
#[cfg(all(feature = "aead", aes_ccm))]
fn test_aes128ccm_aead_roundtrip() {
let key = [0x01u8; 16];
let nonce_bytes = [0x02u8; 12];
let aad = b"ccm aad";
let plaintext = b"AES-128-CCM plaintext!";
let cipher = Aes128Ccm::new_from_slice(&key).unwrap();
let nonce: aead::Nonce<Aes128Ccm> = nonce_bytes.into();
let ciphertext = cipher
.encrypt(&nonce, Payload { msg: plaintext, aad })
.expect("AES-128-CCM encrypt failed");
let recovered = cipher
.decrypt(&nonce, Payload { msg: &ciphertext, aad })
.expect("AES-128-CCM decrypt failed");
assert_eq!(recovered, plaintext);
}
/// Verify that AES-128-CCM decryption rejects a tampered ciphertext.
#[test]
#[cfg(all(feature = "aead", aes_ccm))]
fn test_aes128ccm_reject_tampered() {
let key = [0x01u8; 16];
let nonce_bytes = [0x02u8; 12];
let plaintext = b"AES-128-CCM tamper test!";
let cipher = Aes128Ccm::new_from_slice(&key).unwrap();
let nonce: aead::Nonce<Aes128Ccm> = nonce_bytes.into();
let mut ct = cipher.encrypt(&nonce, plaintext.as_ref()).expect("encrypt failed");
ct[0] ^= 0x01;
assert!(cipher.decrypt(&nonce, ct.as_slice()).is_err());
}
/// Roundtrip test for AES-256-CCM using `aead::Aead`.
#[test]
#[cfg(all(feature = "aead", aes_ccm))]
fn test_aes256ccm_aead_roundtrip() {
let key = [0xddu8; 32];
let nonce_bytes = [0xeeu8; 12];
let aad = b"aes-256-ccm test";
let plaintext = b"AES-256-CCM plaintext data";
let cipher = Aes256Ccm::new_from_slice(&key).unwrap();
let nonce: aead::Nonce<Aes256Ccm> = nonce_bytes.into();
let ciphertext = cipher
.encrypt(&nonce, Payload { msg: plaintext, aad })
.expect("AES-256-CCM encrypt failed");
let recovered = cipher
.decrypt(&nonce, Payload { msg: &ciphertext, aad })
.expect("AES-256-CCM decrypt failed");
assert_eq!(recovered, plaintext);
}
// ---------------------------------------------------------------------------
// AES cipher crate trait tests
// ---------------------------------------------------------------------------
/// Test AES-128-ECB encryption against the known test vector used in the
/// existing `test_ecb_encrypt_decrypt` test.
#[test]
#[cfg(all(feature = "cipher", aes_ecb))]
fn test_aes128_ecb_enc_block_encrypt() {
use cipher::{BlockModeEncrypt, KeyInit};
use wolfssl_wolfcrypt::aes::Aes128EcbEnc;
let key: [u8; 16] = *b"0123456789abcdef";
let plaintext: [u8; 16] = [
0x6e, 0x6f, 0x77, 0x20, 0x69, 0x73, 0x20, 0x74,
0x68, 0x65, 0x20, 0x74, 0x69, 0x6d, 0x65, 0x20,
];
let expected: [u8; 16] = [
0xd0, 0xc9, 0xd9, 0xc9, 0x40, 0xe8, 0x97, 0xb6,
0xc8, 0x8c, 0x33, 0x3b, 0xb5, 0x8f, 0x85, 0xd1,
];
let mut enc = Aes128EcbEnc::new_from_slice(&key).expect("key init failed");
let mut block = cipher::Block::<Aes128EcbEnc>::try_from(&plaintext[..]).unwrap();
enc.encrypt_block(&mut block);
assert_eq!(block.as_slice(), &expected);
}
/// Test AES-128-ECB decryption matches the plaintext after encryption.
#[test]
#[cfg(all(feature = "cipher", aes_ecb))]
fn test_aes128_ecb_dec_block_decrypt() {
use cipher::{BlockModeDecrypt, BlockModeEncrypt, KeyInit};
use wolfssl_wolfcrypt::aes::{Aes128EcbDec, Aes128EcbEnc};
let key: [u8; 16] = *b"0123456789abcdef";
let plaintext: [u8; 16] = [
0x6e, 0x6f, 0x77, 0x20, 0x69, 0x73, 0x20, 0x74,
0x68, 0x65, 0x20, 0x74, 0x69, 0x6d, 0x65, 0x20,
];
let mut enc = Aes128EcbEnc::new_from_slice(&key).expect("enc init failed");
let mut dec = Aes128EcbDec::new_from_slice(&key).expect("dec init failed");
let mut block = cipher::Block::<Aes128EcbEnc>::try_from(&plaintext[..]).unwrap();
enc.encrypt_block(&mut block);
let mut block2 = cipher::Block::<Aes128EcbDec>::try_from(block.as_slice()).unwrap();
dec.decrypt_block(&mut block2);
assert_eq!(block2.as_slice(), &plaintext);
}
/// Test AES-256-ECB encryption and decryption roundtrip.
#[test]
#[cfg(all(feature = "cipher", aes_ecb))]
fn test_aes256_ecb_roundtrip() {
use cipher::{BlockModeDecrypt, BlockModeEncrypt, KeyInit};
use wolfssl_wolfcrypt::aes::{Aes256EcbDec, Aes256EcbEnc};
let key = [0xabu8; 32];
let plaintext = [0x5cu8; 16];
let mut enc = Aes256EcbEnc::new_from_slice(&key).expect("enc init failed");
let mut dec = Aes256EcbDec::new_from_slice(&key).expect("dec init failed");
let mut block = cipher::Block::<Aes256EcbEnc>::try_from(&plaintext[..]).unwrap();
enc.encrypt_block(&mut block);
assert_ne!(block.as_slice(), &plaintext, "encrypted block should differ from plaintext");
let mut block2 = cipher::Block::<Aes256EcbDec>::try_from(block.as_slice()).unwrap();
dec.decrypt_block(&mut block2);
assert_eq!(block2.as_slice(), &plaintext);
}
/// Test AES-128-CTR `apply_keystream` against the NIST CTR test vector.
#[test]
#[cfg(all(feature = "cipher", aes_ctr))]
fn test_aes128_ctr_apply_keystream() {
use cipher::{KeyIvInit, StreamCipher};
use wolfssl_wolfcrypt::aes::Aes128Ctr;
let key: [u8; 16] = [
0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6,
0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f, 0x3c,
];
let iv: [u8; 16] = [
0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7,
0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff,
];
let plaintext: [u8; 64] = [
0x6b, 0xc1, 0xbe, 0xe2, 0x2e, 0x40, 0x9f, 0x96,
0xe9, 0x3d, 0x7e, 0x11, 0x73, 0x93, 0x17, 0x2a,
0xae, 0x2d, 0x8a, 0x57, 0x1e, 0x03, 0xac, 0x9c,
0x9e, 0xb7, 0x6f, 0xac, 0x45, 0xaf, 0x8e, 0x51,
0x30, 0xc8, 0x1c, 0x46, 0xa3, 0x5c, 0xe4, 0x11,
0xe5, 0xfb, 0xc1, 0x19, 0x1a, 0x0a, 0x52, 0xef,
0xf6, 0x9f, 0x24, 0x45, 0xdf, 0x4f, 0x9b, 0x17,
0xad, 0x2b, 0x41, 0x7b, 0xe6, 0x6c, 0x37, 0x10,
];
let expected_ciphertext: [u8; 64] = [
0x87, 0x4d, 0x61, 0x91, 0xb6, 0x20, 0xe3, 0x26,
0x1b, 0xef, 0x68, 0x64, 0x99, 0x0d, 0xb6, 0xce,
0x98, 0x06, 0xf6, 0x6b, 0x79, 0x70, 0xfd, 0xff,
0x86, 0x17, 0x18, 0x7b, 0xb9, 0xff, 0xfd, 0xff,
0x5a, 0xe4, 0xdf, 0x3e, 0xdb, 0xd5, 0xd3, 0x5e,
0x5b, 0x4f, 0x09, 0x02, 0x0d, 0xb0, 0x3e, 0xab,
0x1e, 0x03, 0x1d, 0xda, 0x2f, 0xbe, 0x03, 0xd1,
0x79, 0x21, 0x70, 0xa0, 0xf3, 0x00, 0x9c, 0xee,
];
let key_arr = cipher::Key::<Aes128Ctr>::try_from(&key[..]).unwrap();
let iv_arr = cipher::Iv::<Aes128Ctr>::try_from(&iv[..]).unwrap();
let mut enc = Aes128Ctr::new(&key_arr, &iv_arr);
let mut data = plaintext;
enc.apply_keystream(&mut data);
assert_eq!(data, expected_ciphertext);
// apply_keystream is self-inverse: applying again must recover plaintext.
let mut dec = Aes128Ctr::new(&key_arr, &iv_arr);
dec.apply_keystream(&mut data);
assert_eq!(data, plaintext);
}
/// Test AES-256-CTR roundtrip via `apply_keystream`.
#[test]
#[cfg(all(feature = "cipher", aes_ctr))]
fn test_aes256_ctr_roundtrip() {
use cipher::{KeyIvInit, StreamCipher};
use wolfssl_wolfcrypt::aes::Aes256Ctr;
let key = [0x01u8; 32];
let iv = [0x02u8; 16];
let plaintext = [0x55u8; 48];
let key_arr = cipher::Key::<Aes256Ctr>::try_from(&key[..]).unwrap();
let iv_arr = cipher::Iv::<Aes256Ctr>::try_from(&iv[..]).unwrap();
let mut enc = Aes256Ctr::new(&key_arr, &iv_arr);
let mut data = plaintext;
enc.apply_keystream(&mut data);
assert_ne!(data, plaintext);
let mut dec = Aes256Ctr::new(&key_arr, &iv_arr);
dec.apply_keystream(&mut data);
assert_eq!(data, plaintext);
}
/// Test AES-256-OFB `apply_keystream` against the known OFB test vector.
#[test]
#[cfg(all(feature = "cipher", aes_ofb))]
fn test_aes256_ofb_apply_keystream() {
use cipher::{KeyIvInit, StreamCipher};
use wolfssl_wolfcrypt::aes::Aes256Ofb;
let key: [u8; 32] = [
0xc4, 0xc7, 0xfa, 0xd6, 0x53, 0x5c, 0xb8, 0x71,
0x4a, 0x5c, 0x40, 0x77, 0x9a, 0x8b, 0xa1, 0xd2,
0x53, 0x3e, 0x23, 0xb4, 0xb2, 0x58, 0x73, 0x2a,
0x5b, 0x78, 0x01, 0xf4, 0xe3, 0x71, 0xa7, 0x94,
];
let iv: [u8; 16] = [
0x5e, 0xb9, 0x33, 0x13, 0xb8, 0x71, 0xff, 0x16,
0xb9, 0x8a, 0x9b, 0xcb, 0x43, 0x33, 0x0d, 0x6f,
];
let plaintext: [u8; 48] = [
0x6d, 0x0b, 0xb0, 0x79, 0x63, 0x84, 0x71, 0xe9,
0x39, 0xd4, 0x53, 0x14, 0x86, 0xc1, 0x4c, 0x25,
0x9a, 0xee, 0xc6, 0xf3, 0xc0, 0x0d, 0xfd, 0xd6,
0xc0, 0x50, 0xa8, 0xba, 0xa8, 0x20, 0xdb, 0x71,
0xcc, 0x12, 0x2c, 0x4e, 0x0c, 0x17, 0x15, 0xef,
0x55, 0xf3, 0x99, 0x5a, 0x6b, 0xf0, 0x2a, 0x4c,
];
let expected_ciphertext: [u8; 48] = [
0x0f, 0x54, 0x61, 0x71, 0x59, 0xd0, 0x3f, 0xfc,
0x1b, 0xfa, 0xfb, 0x60, 0x29, 0x30, 0xd7, 0x00,
0xf4, 0xa4, 0xa8, 0xe6, 0xdd, 0x93, 0x94, 0x46,
0x64, 0xd2, 0x19, 0xc4, 0xc5, 0x4d, 0xde, 0x1b,
0x04, 0x53, 0xe1, 0x73, 0xf5, 0x18, 0x74, 0xae,
0xfd, 0x64, 0xa2, 0xe1, 0xe2, 0x76, 0x13, 0xb0,
];
let key_arr = cipher::Key::<Aes256Ofb>::try_from(&key[..]).unwrap();
let iv_arr = cipher::Iv::<Aes256Ofb>::try_from(&iv[..]).unwrap();
let mut enc = Aes256Ofb::new(&key_arr, &iv_arr);
let mut data = plaintext;
enc.apply_keystream(&mut data);
assert_eq!(data, expected_ciphertext);
// apply_keystream is self-inverse for OFB (same keystream for enc/dec).
let mut dec = Aes256Ofb::new(&key_arr, &iv_arr);
dec.apply_keystream(&mut data);
assert_eq!(data, plaintext);
}
/// Test AES-128-OFB roundtrip via `apply_keystream`.
#[test]
#[cfg(all(feature = "cipher", aes_ofb))]
fn test_aes128_ofb_roundtrip() {
use cipher::{KeyIvInit, StreamCipher};
use wolfssl_wolfcrypt::aes::Aes128Ofb;
let key = [0xddu8; 16];
let iv = [0xeeu8; 16];
let plaintext = [0x42u8; 32];
let key_arr = cipher::Key::<Aes128Ofb>::try_from(&key[..]).unwrap();
let iv_arr = cipher::Iv::<Aes128Ofb>::try_from(&iv[..]).unwrap();
let mut enc = Aes128Ofb::new(&key_arr, &iv_arr);
let mut data = plaintext;
enc.apply_keystream(&mut data);
assert_ne!(data, plaintext);
let mut dec = Aes128Ofb::new(&key_arr, &iv_arr);
dec.apply_keystream(&mut data);
assert_eq!(data, plaintext);
}
/// Test AES-128-CBC encryption against a known vector (same as test_cbc_encrypt_decrypt).
#[test]
#[cfg(all(feature = "cipher", aes_cbc))]
fn test_aes128_cbc_enc_block_mode() {
use cipher::{BlockModeEncrypt, KeyIvInit};
use wolfssl_wolfcrypt::aes::Aes128CbcEnc;
let key: [u8; 16] = *b"0123456789abcdef";
let iv: [u8; 16] = *b"1234567890abcdef";
let plaintext: [u8; 16] = [
0x6e, 0x6f, 0x77, 0x20, 0x69, 0x73, 0x20, 0x74,
0x68, 0x65, 0x20, 0x74, 0x69, 0x6d, 0x65, 0x20,
];
let expected: [u8; 16] = [
0x95, 0x94, 0x92, 0x57, 0x5f, 0x42, 0x81, 0x53,
0x2c, 0xcc, 0x9d, 0x46, 0x77, 0xa2, 0x33, 0xcb,
];
let key_arr = cipher::Key::<Aes128CbcEnc>::try_from(&key[..]).unwrap();
let iv_arr = cipher::Iv::<Aes128CbcEnc>::try_from(&iv[..]).unwrap();
let mut enc = Aes128CbcEnc::new(&key_arr, &iv_arr);
let mut block = cipher::Block::<Aes128CbcEnc>::try_from(&plaintext[..]).unwrap();
enc.encrypt_block(&mut block);
assert_eq!(block.as_slice(), &expected);
}
/// Test AES-128-CBC decryption roundtrip.
#[test]
#[cfg(all(feature = "cipher", aes_cbc))]
fn test_aes128_cbc_dec_block_mode() {
use cipher::{BlockModeDecrypt, BlockModeEncrypt, KeyIvInit};
use wolfssl_wolfcrypt::aes::{Aes128CbcDec, Aes128CbcEnc};
let key: [u8; 16] = *b"0123456789abcdef";
let iv: [u8; 16] = *b"1234567890abcdef";
let plaintext: [u8; 16] = [
0x6e, 0x6f, 0x77, 0x20, 0x69, 0x73, 0x20, 0x74,
0x68, 0x65, 0x20, 0x74, 0x69, 0x6d, 0x65, 0x20,
];
let key_arr = cipher::Key::<Aes128CbcEnc>::try_from(&key[..]).unwrap();
let iv_arr = cipher::Iv::<Aes128CbcEnc>::try_from(&iv[..]).unwrap();
let mut enc = Aes128CbcEnc::new(&key_arr, &iv_arr);
let mut block = cipher::Block::<Aes128CbcEnc>::try_from(&plaintext[..]).unwrap();
enc.encrypt_block(&mut block);
let key_arr = cipher::Key::<Aes128CbcDec>::try_from(&key[..]).unwrap();
let iv_arr = cipher::Iv::<Aes128CbcDec>::try_from(&iv[..]).unwrap();
let mut dec = Aes128CbcDec::new(&key_arr, &iv_arr);
dec.decrypt_block(&mut block);
assert_eq!(block.as_slice(), &plaintext);
}
/// Test AES-256-CBC encryption/decryption roundtrip across multiple blocks.
#[test]
#[cfg(all(feature = "cipher", aes_cbc))]
fn test_aes256_cbc_roundtrip() {
use cipher::{BlockModeDecrypt, BlockModeEncrypt, KeyIvInit};
use wolfssl_wolfcrypt::aes::{Aes256CbcDec, Aes256CbcEnc};
let key = [0xabu8; 32];
let iv = [0xcdu8; 16];
let plaintext = [[0x5cu8; 16], [0x3au8; 16], [0x1eu8; 16]];
let key_arr = cipher::Key::<Aes256CbcEnc>::try_from(&key[..]).unwrap();
let iv_arr = cipher::Iv::<Aes256CbcEnc>::try_from(&iv[..]).unwrap();
let mut enc = Aes256CbcEnc::new(&key_arr, &iv_arr);
let mut blocks: [cipher::Block<Aes256CbcEnc>; 3] = plaintext
.iter()
.map(|b| cipher::Block::<Aes256CbcEnc>::try_from(b.as_ref()).unwrap())
.collect::<Vec<_>>()
.try_into()
.unwrap();
for block in blocks.iter_mut() {
enc.encrypt_block(block);
}
// Ciphertext must differ from plaintext due to key and IV mixing.
assert!(blocks.iter().zip(plaintext.iter()).any(|(c, p)| c.as_slice() != p));
let key_arr = cipher::Key::<Aes256CbcDec>::try_from(&key[..]).unwrap();
let iv_arr = cipher::Iv::<Aes256CbcDec>::try_from(&iv[..]).unwrap();
let mut dec = Aes256CbcDec::new(&key_arr, &iv_arr);
for block in blocks.iter_mut() {
dec.decrypt_block(block);
}
for (block, expected) in blocks.iter().zip(plaintext.iter()) {
assert_eq!(block.as_slice(), expected);
}
}
Reference in New Issue View Git Blame Copy Permalink