1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

//! Development-related functionality
pub use blobby;

/// Define AEAD test
#[macro_export]
#[cfg_attr(docsrs, doc(cfg(feature = "dev")))]
macro_rules! new_test {
    ($name:ident, $test_name:expr, $cipher:ty $(,)?) => {
        #[test]
        fn $name() {
            use aead::dev::blobby::Blob6Iterator;
            use aead::generic_array::typenum::Unsigned;
            use aead::{generic_array::GenericArray, Aead, NewAead, Payload};
            use core::convert::TryInto;

            fn run_test(
                key: &[u8],
                nonce: &[u8],
                aad: &[u8],
                pt: &[u8],
                ct: &[u8],
                pass: bool,
            ) -> Result<(), &'static str> {
                let key = key.try_into().map_err(|_| "wrong key size")?;
                let cipher = <$cipher>::new(key);
                let nonce = nonce.try_into().map_err(|_| "wrong nonce size")?;

                if !pass {
                    let res = cipher.decrypt(nonce, Payload { aad: aad, msg: ct });
                    if res.is_ok() {
                        return Err("decryption must return error");
                    }
                    return Ok(());
                }

                let res = cipher
                    .encrypt(nonce, Payload { aad: aad, msg: pt })
                    .map_err(|_| "encryption failure")?;
                if res != ct {
                    return Err("encrypted data is different from target ciphertext");
                }
                let res = cipher
                    .decrypt(nonce, Payload { aad: aad, msg: ct })
                    .map_err(|_| "decryption failure")?;
                if res != pt {
                    return Err("decrypted data is different from target plaintext");
                }
                Ok(())
            }

            let data = include_bytes!(concat!("data/", $test_name, ".blb"));
            for (i, row) in Blob6Iterator::new(data).unwrap().enumerate() {
                let [key, nonce, aad, pt, ct, status] = row.unwrap();
                let pass = match status[0] {
                    0 => false,
                    1 => true,
                    _ => panic!("invalid value for pass flag"),
                };
                if let Err(reason) = run_test(key, nonce, aad, pt, ct, pass) {
                    panic!(
                        "\n\
                            Failed test №{}\n\
                            reason: \t{:?}\n\
                            key:\t{:?}\n\
                            nonce:\t{:?}\n\
                            aad:\t{:?}\n\
                            plaintext:\t{:?}\n\
                            ciphertext:\t{:?}\n\
                            pass:\t{}\n\
                        ",
                        i, reason, key, nonce, aad, pt, ct, pass,
                    );
                }
            }
        }
    };
}