1use crate::ffi;
2use libc::{c_int, c_uint};
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#[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(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 as usize];
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
84pub 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 assert!(pass.len() <= c_int::MAX as usize);
94 assert!(salt.len() <= c_int::MAX as usize);
95 assert!(key.len() <= c_int::MAX as usize);
96
97 ffi::init();
98 cvt(ffi::PKCS5_PBKDF2_HMAC(
99 pass.as_ptr() as *const _,
100 pass.len(),
101 salt.as_ptr(),
102 salt.len(),
103 iter as c_uint,
104 hash.as_ptr(),
105 key.len(),
106 key.as_mut_ptr(),
107 ))
108 .map(|_| ())
109 }
110}
111
112pub fn scrypt(
114 pass: &[u8],
115 salt: &[u8],
116 n: u64,
117 r: u64,
118 p: u64,
119 maxmem: usize,
120 key: &mut [u8],
121) -> Result<(), ErrorStack> {
122 unsafe {
123 ffi::init();
124 cvt(ffi::EVP_PBE_scrypt(
125 pass.as_ptr() as *const _,
126 pass.len(),
127 salt.as_ptr() as *const _,
128 salt.len(),
129 n,
130 r,
131 p,
132 maxmem,
133 key.as_mut_ptr() as *mut _,
134 key.len(),
135 ))
136 .map(|_| ())
137 }
138}
139
140#[cfg(test)]
141mod tests {
142 use crate::hash::MessageDigest;
143 use crate::symm::Cipher;
144
145 #[test]
148 fn pbkdf2_hmac_sha256() {
149 let mut buf = [0; 16];
150
151 super::pbkdf2_hmac(b"passwd", b"salt", 1, MessageDigest::sha256(), &mut buf).unwrap();
152 assert_eq!(
153 buf,
154 &[
155 0x55_u8, 0xac_u8, 0x04_u8, 0x6e_u8, 0x56_u8, 0xe3_u8, 0x08_u8, 0x9f_u8, 0xec_u8,
156 0x16_u8, 0x91_u8, 0xc2_u8, 0x25_u8, 0x44_u8, 0xb6_u8, 0x05_u8,
157 ][..]
158 );
159
160 super::pbkdf2_hmac(
161 b"Password",
162 b"NaCl",
163 80000,
164 MessageDigest::sha256(),
165 &mut buf,
166 )
167 .unwrap();
168 assert_eq!(
169 buf,
170 &[
171 0x4d_u8, 0xdc_u8, 0xd8_u8, 0xf6_u8, 0x0b_u8, 0x98_u8, 0xbe_u8, 0x21_u8, 0x83_u8,
172 0x0c_u8, 0xee_u8, 0x5e_u8, 0xf2_u8, 0x27_u8, 0x01_u8, 0xf9_u8,
173 ][..]
174 );
175 }
176
177 #[test]
180 fn pbkdf2_hmac_sha512() {
181 let mut buf = [0; 64];
182
183 super::pbkdf2_hmac(b"password", b"NaCL", 1, MessageDigest::sha512(), &mut buf).unwrap();
184 assert_eq!(
185 &buf[..],
186 &[
187 0x73_u8, 0xde_u8, 0xcf_u8, 0xa5_u8, 0x8a_u8, 0xa2_u8, 0xe8_u8, 0x4f_u8, 0x94_u8,
188 0x77_u8, 0x1a_u8, 0x75_u8, 0x73_u8, 0x6b_u8, 0xb8_u8, 0x8b_u8, 0xd3_u8, 0xc7_u8,
189 0xb3_u8, 0x82_u8, 0x70_u8, 0xcf_u8, 0xb5_u8, 0x0c_u8, 0xb3_u8, 0x90_u8, 0xed_u8,
190 0x78_u8, 0xb3_u8, 0x05_u8, 0x65_u8, 0x6a_u8, 0xf8_u8, 0x14_u8, 0x8e_u8, 0x52_u8,
191 0x45_u8, 0x2b_u8, 0x22_u8, 0x16_u8, 0xb2_u8, 0xb8_u8, 0x09_u8, 0x8b_u8, 0x76_u8,
192 0x1f_u8, 0xc6_u8, 0x33_u8, 0x60_u8, 0x60_u8, 0xa0_u8, 0x9f_u8, 0x76_u8, 0x41_u8,
193 0x5e_u8, 0x9f_u8, 0x71_u8, 0xea_u8, 0x47_u8, 0xf9_u8, 0xe9_u8, 0x06_u8, 0x43_u8,
194 0x06_u8,
195 ][..]
196 );
197
198 super::pbkdf2_hmac(
199 b"pass\0word",
200 b"sa\0lt",
201 1,
202 MessageDigest::sha512(),
203 &mut buf,
204 )
205 .unwrap();
206 assert_eq!(
207 &buf[..],
208 &[
209 0x71_u8, 0xa0_u8, 0xec_u8, 0x84_u8, 0x2a_u8, 0xbd_u8, 0x5c_u8, 0x67_u8, 0x8b_u8,
210 0xcf_u8, 0xd1_u8, 0x45_u8, 0xf0_u8, 0x9d_u8, 0x83_u8, 0x52_u8, 0x2f_u8, 0x93_u8,
211 0x36_u8, 0x15_u8, 0x60_u8, 0x56_u8, 0x3c_u8, 0x4d_u8, 0x0d_u8, 0x63_u8, 0xb8_u8,
212 0x83_u8, 0x29_u8, 0x87_u8, 0x10_u8, 0x90_u8, 0xe7_u8, 0x66_u8, 0x04_u8, 0xa4_u8,
213 0x9a_u8, 0xf0_u8, 0x8f_u8, 0xe7_u8, 0xc9_u8, 0xf5_u8, 0x71_u8, 0x56_u8, 0xc8_u8,
214 0x79_u8, 0x09_u8, 0x96_u8, 0xb2_u8, 0x0f_u8, 0x06_u8, 0xbc_u8, 0x53_u8, 0x5e_u8,
215 0x5a_u8, 0xb5_u8, 0x44_u8, 0x0d_u8, 0xf7_u8, 0xe8_u8, 0x78_u8, 0x29_u8, 0x6f_u8,
216 0xa7_u8,
217 ][..]
218 );
219
220 super::pbkdf2_hmac(
221 b"passwordPASSWORDpassword",
222 b"salt\0\0\0",
223 50,
224 MessageDigest::sha512(),
225 &mut buf,
226 )
227 .unwrap();
228 assert_eq!(
229 &buf[..],
230 &[
231 0x01_u8, 0x68_u8, 0x71_u8, 0xa4_u8, 0xc4_u8, 0xb7_u8, 0x5f_u8, 0x96_u8, 0x85_u8,
232 0x7f_u8, 0xd2_u8, 0xb9_u8, 0xf8_u8, 0xca_u8, 0x28_u8, 0x02_u8, 0x3b_u8, 0x30_u8,
233 0xee_u8, 0x2a_u8, 0x39_u8, 0xf5_u8, 0xad_u8, 0xca_u8, 0xc8_u8, 0xc9_u8, 0x37_u8,
234 0x5f_u8, 0x9b_u8, 0xda_u8, 0x1c_u8, 0xcd_u8, 0x1b_u8, 0x6f_u8, 0x0b_u8, 0x2f_u8,
235 0xc3_u8, 0xad_u8, 0xda_u8, 0x50_u8, 0x54_u8, 0x12_u8, 0xe7_u8, 0x9d_u8, 0x89_u8,
236 0x00_u8, 0x56_u8, 0xc6_u8, 0x2e_u8, 0x52_u8, 0x4c_u8, 0x7d_u8, 0x51_u8, 0x15_u8,
237 0x4b_u8, 0x1a_u8, 0x85_u8, 0x34_u8, 0x57_u8, 0x5b_u8, 0xd0_u8, 0x2d_u8, 0xee_u8,
238 0x39_u8,
239 ][..]
240 );
241 }
242
243 #[test]
244 fn bytes_to_key() {
245 let salt = [16_u8, 34_u8, 19_u8, 23_u8, 141_u8, 4_u8, 207_u8, 221_u8];
246
247 let data = [
248 143_u8, 210_u8, 75_u8, 63_u8, 214_u8, 179_u8, 155_u8, 241_u8, 242_u8, 31_u8, 154_u8,
249 56_u8, 198_u8, 145_u8, 192_u8, 64_u8, 2_u8, 245_u8, 167_u8, 220_u8, 55_u8, 119_u8,
250 233_u8, 136_u8, 139_u8, 27_u8, 71_u8, 242_u8, 119_u8, 175_u8, 65_u8, 207_u8,
251 ];
252
253 let expected_key = vec![
254 249_u8, 115_u8, 114_u8, 97_u8, 32_u8, 213_u8, 165_u8, 146_u8, 58_u8, 87_u8, 234_u8,
255 3_u8, 43_u8, 250_u8, 97_u8, 114_u8, 26_u8, 98_u8, 245_u8, 246_u8, 238_u8, 177_u8,
256 229_u8, 161_u8, 183_u8, 224_u8, 174_u8, 3_u8, 6_u8, 244_u8, 236_u8, 255_u8,
257 ];
258 let expected_iv = vec![
259 4_u8, 223_u8, 153_u8, 219_u8, 28_u8, 142_u8, 234_u8, 68_u8, 227_u8, 69_u8, 98_u8,
260 107_u8, 208_u8, 14_u8, 236_u8, 60_u8,
261 ];
262
263 assert_eq!(
264 super::bytes_to_key(
265 Cipher::aes_256_cbc(),
266 MessageDigest::sha1(),
267 &data,
268 Some(&salt),
269 1,
270 )
271 .unwrap(),
272 super::KeyIvPair {
273 key: expected_key,
274 iv: Some(expected_iv),
275 }
276 );
277 }
278
279 #[test]
280 fn scrypt() {
281 use hex;
282
283 let pass = "pleaseletmein";
284 let salt = "SodiumChloride";
285 let expected =
286 "7023bdcb3afd7348461c06cd81fd38ebfda8fbba904f8e3ea9b543f6545da1f2d5432955613\
287 f0fcf62d49705242a9af9e61e85dc0d651e40dfcf017b45575887";
288
289 let mut actual = [0; 64];
290 super::scrypt(
291 pass.as_bytes(),
292 salt.as_bytes(),
293 16384,
294 8,
295 1,
296 0,
297 &mut actual,
298 )
299 .unwrap();
300 assert_eq!(hex::encode(&actual[..]), expected);
301 }
302}