You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.8 KiB
Rust
58 lines
1.8 KiB
Rust
use crate::lib::hash::create_hmac;
|
|
use cfb_mode::stream_cipher::{NewStreamCipher, StreamCipher};
|
|
use cfb_mode::Cfb;
|
|
use des::Des;
|
|
use rand::Rng;
|
|
use rayon::prelude::*;
|
|
use std::sync::Mutex;
|
|
|
|
type DesCfb = Cfb<Des>;
|
|
|
|
/// Encrypts data with des
|
|
pub fn encrypt_data(data: &[u8], key: &[u8]) -> Vec<u8> {
|
|
let iv = rand::thread_rng().gen::<[u8; 8]>();
|
|
let mut buffer = data.to_vec();
|
|
let mut cipher = DesCfb::new_var(key, &iv).unwrap();
|
|
cipher.encrypt(&mut buffer);
|
|
let mut cipher_text = iv.to_vec();
|
|
cipher_text.append(&mut buffer.to_vec());
|
|
|
|
cipher_text
|
|
}
|
|
|
|
/// Decrypts data with des
|
|
pub fn decrypt_data(data: &[u8], key: &[u8]) -> Vec<u8> {
|
|
let iv = &data[..8];
|
|
let mut buffer = data[8..].to_vec();
|
|
let mut cipher = DesCfb::new_var(&key, &iv).unwrap();
|
|
cipher.decrypt(&mut buffer);
|
|
|
|
buffer
|
|
}
|
|
|
|
/// Decrypts data using a dictionary
|
|
pub fn decrypt_with_dictionary(data: &[u8], dict: Vec<(&String, &Vec<u8>)>) -> Option<Vec<u8>> {
|
|
let decrypted = Mutex::<Option<Vec<u8>>>::new(None);
|
|
let hmac = &data[data.len() - 32..];
|
|
let encrypted_data = &data[..data.len() - 32];
|
|
let pass = dict.par_iter().find_first(|(_pw, key)| {
|
|
let decrypted_data = decrypt_data(encrypted_data, &key[0..8]);
|
|
let decr_hmac = create_hmac(&key, &decrypted_data).expect("failed to create hmac");
|
|
return if decr_hmac == hmac {
|
|
let mut decry = decrypted.lock().unwrap();
|
|
*decry = Some(decrypted_data);
|
|
true
|
|
} else {
|
|
false
|
|
};
|
|
});
|
|
if let Some((pw, _key)) = pass {
|
|
println!("\nPassword found: {}", pw);
|
|
let decry = decrypted.lock().unwrap();
|
|
if let Some(decrypted_data) = (*decry).clone() {
|
|
return Some(decrypted_data);
|
|
}
|
|
}
|
|
None
|
|
}
|