use sha1::Sha1;
use sha2::{Digest, Sha256};
fn xor<T, U>(mut left: T, right: U) -> T
where
T: AsMut<[u8]>,
U: AsRef<[u8]>,
{
left.as_mut()
.iter_mut()
.zip(right.as_ref().iter())
.map(|(l, r)| *l ^= r)
.last();
left
}
fn to_u8_32(bytes: impl AsRef<[u8]>) -> [u8; 32] {
let mut out = [0; 32];
(&mut out[..]).copy_from_slice(bytes.as_ref());
out
}
pub fn scramble_native(nonce: &[u8], password: &[u8]) -> Option<[u8; 20]> {
fn sha1_1(bytes: impl AsRef<[u8]>) -> [u8; 20] {
let mut hasher = Sha1::default();
hasher.update(bytes.as_ref());
hasher.digest().bytes()
}
fn sha1_2(bytes1: impl AsRef<[u8]>, bytes2: impl AsRef<[u8]>) -> [u8; 20] {
let mut hasher = Sha1::default();
hasher.update(bytes1.as_ref());
hasher.update(bytes2.as_ref());
hasher.digest().bytes()
}
if password.len() == 0 {
return None;
}
Some(xor(
sha1_1(password),
sha1_2(nonce, sha1_1(sha1_1(password))),
))
}
pub fn scramble_sha256(nonce: &[u8], password: &[u8]) -> Option<[u8; 32]> {
fn sha256_1(bytes: impl AsRef<[u8]>) -> [u8; 32] {
let mut hasher = Sha256::default();
hasher.input(bytes.as_ref());
to_u8_32(hasher.result())
}
fn sha256_2(bytes1: impl AsRef<[u8]>, bytes2: impl AsRef<[u8]>) -> [u8; 32] {
let mut hasher = Sha256::default();
hasher.input(bytes1.as_ref());
hasher.input(bytes2.as_ref());
to_u8_32(hasher.result())
}
if password.len() == 0 {
return None;
}
Some(xor(
sha256_1(password),
sha256_2(sha256_1(sha256_1(password)), nonce),
))
}
#[cfg(test)]
mod test {
use super::*;
#[test]
fn should_compute_scrambled_password() {
let scr = [
0x4e, 0x52, 0x33, 0x48, 0x50, 0x3a, 0x71, 0x49, 0x59, 0x61, 0x5f, 0x39, 0x3d, 0x64,
0x62, 0x3f, 0x53, 0x64, 0x7b, 0x60,
];
let password = [0x47, 0x21, 0x69, 0x64, 0x65, 0x72, 0x32, 0x37];
let output1 = scramble_native(&scr, &password);
let output2 = scramble_sha256(&scr, &password);
assert!(output1.is_some());
assert!(output2.is_some());
assert_eq!(
output1.unwrap(),
[
0x09, 0xcf, 0xf8, 0x85, 0x5e, 0x9e, 0x70, 0x53, 0x40, 0xff, 0x22, 0x70, 0xd8, 0xfb,
0x9f, 0xad, 0xba, 0x90, 0x6b, 0x70,
]
);
assert_eq!(
output2.unwrap(),
[
0x4f, 0x97, 0xbb, 0xfd, 0x20, 0x24, 0x01, 0xc4, 0x2a, 0x69, 0xde, 0xaa, 0xe5, 0x3b,
0xda, 0x07, 0x7e, 0xd7, 0x57, 0x85, 0x63, 0xc1, 0xa8, 0x0e, 0xb8, 0x16, 0xc8, 0x21,
0x19, 0xb6, 0x8d, 0x2e,
]
);
}
}