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
use hmac::Hmac;
use pbkdf2::{Params, Pbkdf2, password_hash::{Ident, PasswordHasher, Salt, SaltString}, pbkdf2};
use anyhow::*;
use rand_core::OsRng;
use sha1::Sha1;
use sha2::{Sha256, Sha512};
use crate::{KDF, PBKDF2Errors, hash::Hash};
use wasm_bindgen::prelude::*;
use wasm_bindgen::{JsValue, throw_str};
#[wasm_bindgen]
#[derive(Debug, Clone, Copy)]
pub enum PBKDF2Hashes {
SHA1,
SHA256,
SHA512
}
impl KDF {
pub fn pbkdf2_impl(password: &[u8], salt: &[u8], hash_algo: PBKDF2Hashes, rounds: u32, output_length: usize)-> KDF {
let pbkdf2_fn = match hash_algo {
PBKDF2Hashes::SHA1 => pbkdf2::<Hmac<Sha1>>,
PBKDF2Hashes::SHA256 => pbkdf2::<Hmac<Sha256>>,
PBKDF2Hashes::SHA512 => pbkdf2::<Hmac<Sha512>>,
};
let mut result = vec![0; output_length];
pbkdf2_fn(password, salt, rounds, &mut result);
KDF{ hash: Hash(result), salt: salt.to_vec() }
}
pub fn pbkdf2_random_salt_impl(password: &[u8], hash_algo: PBKDF2Hashes, rounds: u32, output_length: usize)-> KDF {
let salt = SaltString::generate(&mut OsRng);
KDF::pbkdf2_impl(password, salt.as_bytes(), hash_algo, rounds, output_length)
}
}
#[wasm_bindgen]
impl KDF {
pub fn pbkdf2(password: Vec<u8>, salt: Option<Vec<u8>>, hash_algo: PBKDF2Hashes, rounds: u32, output_length: usize) -> KDF {
match salt {
Some(s) => KDF::pbkdf2_impl(&password, &s, hash_algo, rounds, output_length),
None => KDF::pbkdf2_random_salt_impl(&password, hash_algo, rounds, output_length)
}
}
}