literate-crypto 0.0.3

Literate Cryptography by 12hbender
Documentation 
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

//! Tests for ciphers. The tests ensure that
//! ```
//! decrypt(encrypt(plaintext, key)) == plaintext
//! ```
//! for a random plaintext and key.

use {
    crate::{util::CollectVec, Aes128, Aes192, Aes256, Cbc, Cipher, Ctr, Ecb, Pkcs7},
    rand::Rng,
    std::fmt,
};

#[test]
fn aes_128_ecb_pkcs7() {
    test(Ecb::new(Aes128::default(), Pkcs7::default()), 10);
    test(Ecb::new(Aes128::default(), Pkcs7::default()), 20);
    test(Ecb::new(Aes128::default(), Pkcs7::default()), 30);
    test(Ecb::new(Aes128::default(), Pkcs7::default()), 16);
}

#[test]
fn aes_192_ecb_pkcs7() {
    test(Ecb::new(Aes192::default(), Pkcs7::default()), 10);
    test(Ecb::new(Aes192::default(), Pkcs7::default()), 20);
    test(Ecb::new(Aes192::default(), Pkcs7::default()), 30);
    test(Ecb::new(Aes192::default(), Pkcs7::default()), 16);
}

#[test]
fn aes_256_ecb_pkcs7() {
    test(Ecb::new(Aes256::default(), Pkcs7::default()), 10);
    test(Ecb::new(Aes256::default(), Pkcs7::default()), 20);
    test(Ecb::new(Aes256::default(), Pkcs7::default()), 30);
    test(Ecb::new(Aes256::default(), Pkcs7::default()), 16);
}

#[test]
fn aes_128_cbc_pkcs7() {
    let iv = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16];
    test(Cbc::new(Aes128::default(), Pkcs7::default(), iv), 10);
    test(Cbc::new(Aes128::default(), Pkcs7::default(), iv), 20);
    test(Cbc::new(Aes128::default(), Pkcs7::default(), iv), 30);
    test(Cbc::new(Aes128::default(), Pkcs7::default(), iv), 16);
}

#[test]
fn aes_192_cbc_pkcs7() {
    let iv = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16];
    test(Cbc::new(Aes192::default(), Pkcs7::default(), iv), 10);
    test(Cbc::new(Aes192::default(), Pkcs7::default(), iv), 20);
    test(Cbc::new(Aes192::default(), Pkcs7::default(), iv), 30);
    test(Cbc::new(Aes192::default(), Pkcs7::default(), iv), 16);
}

#[test]
fn aes_256_cbc_pkcs7() {
    let iv = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16];
    test(Cbc::new(Aes256::default(), Pkcs7::default(), iv), 10);
    test(Cbc::new(Aes256::default(), Pkcs7::default(), iv), 20);
    test(Cbc::new(Aes256::default(), Pkcs7::default(), iv), 30);
    test(Cbc::new(Aes256::default(), Pkcs7::default(), iv), 16);
}

#[test]
fn aes_256_ctr() {
    test(
        Ctr::new(Aes256::default(), rand::thread_rng().gen()).unwrap(),
        10,
    );
    test(
        Ctr::new(Aes256::default(), rand::thread_rng().gen()).unwrap(),
        20,
    );
    test(
        Ctr::new(Aes256::default(), rand::thread_rng().gen()).unwrap(),
        30,
    );
    test(
        Ctr::new(Aes256::default(), rand::thread_rng().gen()).unwrap(),
        16,
    );
}

/// Test that a cipher is valid by making sure that
/// ```
/// decrypt(encrypt(plaintext, key)) == plaintext
/// ```
fn test<Cip: Cipher>(cip: Cip, data_size: usize)
where
    Cip::Key: for<'a> TryFrom<&'a [u8]> + fmt::Debug + Clone,
    Cip::EncryptionErr: fmt::Debug,
    Cip::DecryptionErr: fmt::Debug,
{
    let data = (0..data_size)
        .map(|_| rand::thread_rng().gen())
        .collect_vec();
    let key_size = std::mem::size_of::<Cip::Key>();
    let key = (0..key_size)
        .map(|_| rand::thread_rng().gen())
        .collect_vec();
    let key = Cip::Key::try_from(key.as_slice()).unwrap_or_else(|_| unreachable!());

    let ciphertext = cip.encrypt(data.clone(), key.clone()).unwrap();
    let plaintext = cip.decrypt(ciphertext.clone(), key.clone()).unwrap();

    assert_eq!(
        data, plaintext,
        "decrypted plaintext did not match for cipher\ndata: {data:?}\nkey: {key:?}\nciphertext: \
         {ciphertext:?}\nplaintext: {plaintext:?}"
    );
}