rama_boring/
pkcs5.rs

1use crate::libc_types::c_int;
2use crate::{cvt_nz, ffi, try_int};
3use std::ptr;
4
5use crate::cvt;
6use crate::error::ErrorStack;
7use crate::hash::MessageDigest;
8use crate::symm::Cipher;
9
10#[derive(Clone, Eq, PartialEq, Hash, Debug)]
11pub struct KeyIvPair {
12    pub key: Vec<u8>,
13    pub iv: Option<Vec<u8>>,
14}
15
16/// Derives a key and an IV from various parameters.
17///
18/// If specified, `salt` must be 8 bytes in length.
19///
20/// If the total key and IV length is less than 16 bytes and MD5 is used then
21/// the algorithm is compatible with the key derivation algorithm from PKCS#5
22/// v1.5 or PBKDF1 from PKCS#5 v2.0.
23///
24/// New applications should not use this and instead use
25/// `pbkdf2_hmac` or another more modern key derivation algorithm.
26#[allow(clippy::useless_conversion)]
27pub fn bytes_to_key(
28    cipher: Cipher,
29    digest: MessageDigest,
30    data: &[u8],
31    salt: Option<&[u8]>,
32    count: u32,
33) -> Result<KeyIvPair, ErrorStack> {
34    unsafe {
35        assert!(data.len() <= c_int::MAX as usize);
36        let salt_ptr = match salt {
37            Some(salt) => {
38                pub const PKCS5_SALT_LEN: c_int = 8;
39                assert_eq!(salt.len(), PKCS5_SALT_LEN as usize);
40                salt.as_ptr()
41            }
42            None => ptr::null(),
43        };
44
45        ffi::init();
46
47        let mut iv = cipher.iv_len().map(|l| vec![0; l]);
48
49        let cipher = cipher.as_ptr();
50        let digest = digest.as_ptr();
51
52        let len = cvt_nz(ffi::EVP_BytesToKey(
53            cipher,
54            digest,
55            salt_ptr,
56            ptr::null(),
57            data.len(),
58            count,
59            ptr::null_mut(),
60            ptr::null_mut(),
61        ))?;
62
63        let mut key = vec![0; len.get()];
64        let iv_ptr = iv
65            .as_mut()
66            .map(|v| v.as_mut_ptr())
67            .unwrap_or(ptr::null_mut());
68
69        cvt(ffi::EVP_BytesToKey(
70            cipher,
71            digest,
72            salt_ptr,
73            data.as_ptr(),
74            data.len(),
75            count,
76            key.as_mut_ptr(),
77            iv_ptr,
78        ))?;
79
80        Ok(KeyIvPair { key, iv })
81    }
82}
83
84/// Derives a key from a password and salt using the PBKDF2-HMAC algorithm with a digest function.
85pub fn pbkdf2_hmac(
86    pass: &[u8],
87    salt: &[u8],
88    iter: usize,
89    hash: MessageDigest,
90    key: &mut [u8],
91) -> Result<(), ErrorStack> {
92    unsafe {
93        ffi::init();
94        cvt(ffi::PKCS5_PBKDF2_HMAC(
95            pass.as_ptr().cast(),
96            pass.len(),
97            salt.as_ptr(),
98            salt.len(),
99            try_int(iter)?,
100            hash.as_ptr(),
101            key.len(),
102            key.as_mut_ptr(),
103        ))
104    }
105}
106
107/// Derives a key from a password and salt using the scrypt algorithm.
108pub fn scrypt(
109    pass: &[u8],
110    salt: &[u8],
111    n: u64,
112    r: u64,
113    p: u64,
114    maxmem: usize,
115    key: &mut [u8],
116) -> Result<(), ErrorStack> {
117    unsafe {
118        ffi::init();
119        cvt(ffi::EVP_PBE_scrypt(
120            pass.as_ptr().cast(),
121            pass.len(),
122            salt.as_ptr().cast(),
123            salt.len(),
124            n,
125            r,
126            p,
127            maxmem,
128            key.as_mut_ptr(),
129            key.len(),
130        ))
131    }
132}
133
134#[cfg(test)]
135mod tests {
136    use crate::hash::MessageDigest;
137    use crate::symm::Cipher;
138
139    // Test vectors from
140    // https://git.lysator.liu.se/nettle/nettle/blob/nettle_3.1.1_release_20150424/testsuite/pbkdf2-test.c
141    #[test]
142    fn pbkdf2_hmac_sha256() {
143        let mut buf = [0; 16];
144
145        super::pbkdf2_hmac(b"passwd", b"salt", 1, MessageDigest::sha256(), &mut buf).unwrap();
146        assert_eq!(
147            buf,
148            &[
149                0x55_u8, 0xac_u8, 0x04_u8, 0x6e_u8, 0x56_u8, 0xe3_u8, 0x08_u8, 0x9f_u8, 0xec_u8,
150                0x16_u8, 0x91_u8, 0xc2_u8, 0x25_u8, 0x44_u8, 0xb6_u8, 0x05_u8,
151            ][..]
152        );
153
154        super::pbkdf2_hmac(
155            b"Password",
156            b"NaCl",
157            80000,
158            MessageDigest::sha256(),
159            &mut buf,
160        )
161        .unwrap();
162        assert_eq!(
163            buf,
164            &[
165                0x4d_u8, 0xdc_u8, 0xd8_u8, 0xf6_u8, 0x0b_u8, 0x98_u8, 0xbe_u8, 0x21_u8, 0x83_u8,
166                0x0c_u8, 0xee_u8, 0x5e_u8, 0xf2_u8, 0x27_u8, 0x01_u8, 0xf9_u8,
167            ][..]
168        );
169    }
170
171    // Test vectors from
172    // https://git.lysator.liu.se/nettle/nettle/blob/nettle_3.1.1_release_20150424/testsuite/pbkdf2-test.c
173    #[test]
174    fn pbkdf2_hmac_sha512() {
175        let mut buf = [0; 64];
176
177        super::pbkdf2_hmac(b"password", b"NaCL", 1, MessageDigest::sha512(), &mut buf).unwrap();
178        assert_eq!(
179            &buf[..],
180            &[
181                0x73_u8, 0xde_u8, 0xcf_u8, 0xa5_u8, 0x8a_u8, 0xa2_u8, 0xe8_u8, 0x4f_u8, 0x94_u8,
182                0x77_u8, 0x1a_u8, 0x75_u8, 0x73_u8, 0x6b_u8, 0xb8_u8, 0x8b_u8, 0xd3_u8, 0xc7_u8,
183                0xb3_u8, 0x82_u8, 0x70_u8, 0xcf_u8, 0xb5_u8, 0x0c_u8, 0xb3_u8, 0x90_u8, 0xed_u8,
184                0x78_u8, 0xb3_u8, 0x05_u8, 0x65_u8, 0x6a_u8, 0xf8_u8, 0x14_u8, 0x8e_u8, 0x52_u8,
185                0x45_u8, 0x2b_u8, 0x22_u8, 0x16_u8, 0xb2_u8, 0xb8_u8, 0x09_u8, 0x8b_u8, 0x76_u8,
186                0x1f_u8, 0xc6_u8, 0x33_u8, 0x60_u8, 0x60_u8, 0xa0_u8, 0x9f_u8, 0x76_u8, 0x41_u8,
187                0x5e_u8, 0x9f_u8, 0x71_u8, 0xea_u8, 0x47_u8, 0xf9_u8, 0xe9_u8, 0x06_u8, 0x43_u8,
188                0x06_u8,
189            ][..]
190        );
191
192        super::pbkdf2_hmac(
193            b"pass\0word",
194            b"sa\0lt",
195            1,
196            MessageDigest::sha512(),
197            &mut buf,
198        )
199        .unwrap();
200        assert_eq!(
201            &buf[..],
202            &[
203                0x71_u8, 0xa0_u8, 0xec_u8, 0x84_u8, 0x2a_u8, 0xbd_u8, 0x5c_u8, 0x67_u8, 0x8b_u8,
204                0xcf_u8, 0xd1_u8, 0x45_u8, 0xf0_u8, 0x9d_u8, 0x83_u8, 0x52_u8, 0x2f_u8, 0x93_u8,
205                0x36_u8, 0x15_u8, 0x60_u8, 0x56_u8, 0x3c_u8, 0x4d_u8, 0x0d_u8, 0x63_u8, 0xb8_u8,
206                0x83_u8, 0x29_u8, 0x87_u8, 0x10_u8, 0x90_u8, 0xe7_u8, 0x66_u8, 0x04_u8, 0xa4_u8,
207                0x9a_u8, 0xf0_u8, 0x8f_u8, 0xe7_u8, 0xc9_u8, 0xf5_u8, 0x71_u8, 0x56_u8, 0xc8_u8,
208                0x79_u8, 0x09_u8, 0x96_u8, 0xb2_u8, 0x0f_u8, 0x06_u8, 0xbc_u8, 0x53_u8, 0x5e_u8,
209                0x5a_u8, 0xb5_u8, 0x44_u8, 0x0d_u8, 0xf7_u8, 0xe8_u8, 0x78_u8, 0x29_u8, 0x6f_u8,
210                0xa7_u8,
211            ][..]
212        );
213
214        super::pbkdf2_hmac(
215            b"passwordPASSWORDpassword",
216            b"salt\0\0\0",
217            50,
218            MessageDigest::sha512(),
219            &mut buf,
220        )
221        .unwrap();
222        assert_eq!(
223            &buf[..],
224            &[
225                0x01_u8, 0x68_u8, 0x71_u8, 0xa4_u8, 0xc4_u8, 0xb7_u8, 0x5f_u8, 0x96_u8, 0x85_u8,
226                0x7f_u8, 0xd2_u8, 0xb9_u8, 0xf8_u8, 0xca_u8, 0x28_u8, 0x02_u8, 0x3b_u8, 0x30_u8,
227                0xee_u8, 0x2a_u8, 0x39_u8, 0xf5_u8, 0xad_u8, 0xca_u8, 0xc8_u8, 0xc9_u8, 0x37_u8,
228                0x5f_u8, 0x9b_u8, 0xda_u8, 0x1c_u8, 0xcd_u8, 0x1b_u8, 0x6f_u8, 0x0b_u8, 0x2f_u8,
229                0xc3_u8, 0xad_u8, 0xda_u8, 0x50_u8, 0x54_u8, 0x12_u8, 0xe7_u8, 0x9d_u8, 0x89_u8,
230                0x00_u8, 0x56_u8, 0xc6_u8, 0x2e_u8, 0x52_u8, 0x4c_u8, 0x7d_u8, 0x51_u8, 0x15_u8,
231                0x4b_u8, 0x1a_u8, 0x85_u8, 0x34_u8, 0x57_u8, 0x5b_u8, 0xd0_u8, 0x2d_u8, 0xee_u8,
232                0x39_u8,
233            ][..]
234        );
235    }
236
237    #[test]
238    fn bytes_to_key() {
239        let salt = [16_u8, 34_u8, 19_u8, 23_u8, 141_u8, 4_u8, 207_u8, 221_u8];
240
241        let data = [
242            143_u8, 210_u8, 75_u8, 63_u8, 214_u8, 179_u8, 155_u8, 241_u8, 242_u8, 31_u8, 154_u8,
243            56_u8, 198_u8, 145_u8, 192_u8, 64_u8, 2_u8, 245_u8, 167_u8, 220_u8, 55_u8, 119_u8,
244            233_u8, 136_u8, 139_u8, 27_u8, 71_u8, 242_u8, 119_u8, 175_u8, 65_u8, 207_u8,
245        ];
246
247        let expected_key = vec![
248            249_u8, 115_u8, 114_u8, 97_u8, 32_u8, 213_u8, 165_u8, 146_u8, 58_u8, 87_u8, 234_u8,
249            3_u8, 43_u8, 250_u8, 97_u8, 114_u8, 26_u8, 98_u8, 245_u8, 246_u8, 238_u8, 177_u8,
250            229_u8, 161_u8, 183_u8, 224_u8, 174_u8, 3_u8, 6_u8, 244_u8, 236_u8, 255_u8,
251        ];
252        let expected_iv = vec![
253            4_u8, 223_u8, 153_u8, 219_u8, 28_u8, 142_u8, 234_u8, 68_u8, 227_u8, 69_u8, 98_u8,
254            107_u8, 208_u8, 14_u8, 236_u8, 60_u8,
255        ];
256
257        assert_eq!(
258            super::bytes_to_key(
259                Cipher::aes_256_cbc(),
260                MessageDigest::sha1(),
261                &data,
262                Some(&salt),
263                1,
264            )
265            .unwrap(),
266            super::KeyIvPair {
267                key: expected_key,
268                iv: Some(expected_iv),
269            }
270        );
271    }
272
273    #[test]
274    fn scrypt() {
275        use hex;
276
277        let pass = "pleaseletmein";
278        let salt = "SodiumChloride";
279        let expected =
280            "7023bdcb3afd7348461c06cd81fd38ebfda8fbba904f8e3ea9b543f6545da1f2d5432955613\
281             f0fcf62d49705242a9af9e61e85dc0d651e40dfcf017b45575887";
282
283        let mut actual = [0; 64];
284        super::scrypt(
285            pass.as_bytes(),
286            salt.as_bytes(),
287            16384,
288            8,
289            1,
290            0,
291            &mut actual,
292        )
293        .unwrap();
294        assert_eq!(hex::encode(&actual[..]), expected);
295    }
296}
rama_boring/pkcs5.rs

rama_boring/
pkcs5.rs
