use crate::crypto::digest::*;
use crate::crypto::key::*;
use crate::crypto::secret::Secret;
use crate::crypto::util::randombytes_into;
use rmp_serde;
fn clone_key(key: &Key) -> Key {
Key::deserialize(key.serialize().unwrap()).unwrap()
}
fn new_password(password: &str) -> Secret {
let bytes = password.as_bytes();
let mut s = Secret::with_len(bytes.len()).unwrap();
unsafe { s.as_mut_slice() }.copy_from_slice(bytes);
s
}
fn random_secret(len: usize) -> Secret {
let mut s = Secret::with_len(len).unwrap();
randombytes_into(unsafe { s.as_mut_slice() });
s
}
#[test]
fn test_nonce_increment() {
crate::init().unwrap();
let nonce = Nonce::new();
let next = Nonce::new();
assert_eq!(nonce, next);
assert_ne!(nonce, next.increment());
}
#[test]
fn test_digest_serde_round_trip() {
crate::init().unwrap();
let key = Key::new_random().unwrap();
let digest = key.get_digest();
let serialized = rmp_serde::to_vec(&digest).unwrap();
let deserialized: Digest = rmp_serde::from_slice(serialized.as_slice()).unwrap();
assert_eq!(digest, deserialized);
}
#[test]
fn test_password_key_derivation() {
crate::init().unwrap();
let salt = Salt::default();
let _key = Key::new_password(
&new_password("foobar"),
&salt,
OPS_LIMIT_INTERACTIVE,
MEM_LIMIT_INTERACTIVE,
)
.unwrap();
}
#[test]
fn test_basic_key_digest_comparison() {
crate::init().unwrap();
let a = Key::new_random().unwrap();
let b = Key::new_random().unwrap();
let c = clone_key(&a);
assert_eq!(a.get_digest(), c.get_digest());
assert_ne!(a.get_digest(), b.get_digest());
}
#[test]
fn test_encryption_roundtrip() {
crate::init().unwrap();
let key = Key::new_random().unwrap();
let plaintext = random_secret(1024);
let (nonce, ciphertext) = key.encrypt(&plaintext, None).unwrap();
assert_ne!(unsafe { plaintext.as_slice() }, ciphertext.as_slice());
let decrypted = key.decrypt(nonce.as_ref(), ciphertext.as_slice()).unwrap();
assert_eq!(unsafe { plaintext.as_slice() }, unsafe {
decrypted.as_slice()
});
}
#[test]
fn test_decrypting_with_wrong_key_fails() {
crate::init().unwrap();
let key = Key::new_random().unwrap();
let plaintext = random_secret(1024);
let (nonce, ciphertext) = key.encrypt(&plaintext, None).unwrap();
assert_ne!(unsafe { plaintext.as_slice() }, ciphertext.as_slice());
let wrong_key = Key::new_random().unwrap();
let decrypted_result = wrong_key.decrypt(nonce.as_ref(), ciphertext.as_slice());
assert!(decrypted_result.is_err());
}