Lines
92.59 %
Functions
100 %
Branches
use std::collections::HashMap;
use serde::{Deserialize, Serialize};
use zbus::zvariant::Type;
use super::{Error, UnlockedItem};
use crate::{Key, Mac, crypto};
#[derive(Deserialize, Serialize, Type, Debug, Clone)]
pub(crate) struct EncryptedItem {
pub(crate) hashed_attributes: HashMap<String, Mac>,
#[serde(with = "serde_bytes")]
pub(crate) blob: Vec<u8>,
}
impl EncryptedItem {
pub fn has_attribute(&self, key: &str, value_mac: &Mac) -> bool {
self.hashed_attributes.get(key) == Some(value_mac)
fn try_decrypt_inner(&self, key: &Key) -> Result<UnlockedItem, Error> {
let n = self.blob.len();
let n_mac = crypto::mac_len();
let n_iv = crypto::iv_len();
// The encrypted data, the iv, and the mac are concatenated into blob.
let (encrypted_data_with_iv, mac_tag) = &self.blob.split_at(n - n_mac);
// verify item
if !crypto::verify_mac(encrypted_data_with_iv, key, mac_tag)? {
return Err(Error::MacError);
let (encrypted_data, iv) = encrypted_data_with_iv.split_at(n - n_mac - n_iv);
// decrypt item
let decrypted = crypto::decrypt(encrypted_data, key, iv)?;
let item = UnlockedItem::try_from(decrypted.as_slice())?;
Self::validate(&self.hashed_attributes, &item, key)?;
Ok(item)
pub fn is_valid(&self, key: &Key) -> bool {
self.try_decrypt_inner(key).is_ok()
pub fn decrypt(self, key: &Key) -> Result<UnlockedItem, Error> {
self.try_decrypt_inner(key)
fn validate(
hashed_attributes: &HashMap<String, Mac>,
item: &UnlockedItem,
key: &Key,
) -> Result<(), Error> {
for (attribute_key, hashed_attribute) in hashed_attributes.iter() {
if let Some(attribute_plaintext) = item.attributes().get(attribute_key) {
if !crypto::verify_mac(
attribute_plaintext.as_bytes(),
key,
hashed_attribute.as_slice(),
)? {
return Err(Error::HashedAttributeMac(attribute_key.to_owned()));
} else {
Ok(())