use super::buffer::TailBuffer;
use crate::{Error, Key};
use aes_siv::{aead::KeyInit, siv::Aes256Siv};
const MIN_PAYLOAD_LEN: usize = 17;
#[inline]
pub fn encrypt(plaintext: &[u8], key: &Key) -> Result<Vec<u8>, Error> {
if plaintext.is_empty() {
return Err(Error::EmptyPlaintext);
}
let mut cipher = Aes256Siv::new(key.as_bytes().into());
let headers: &[&[u8]] = &[];
cipher
.encrypt(headers, plaintext)
.map_err(|_| Error::EncryptionFailed)
}
#[inline]
pub fn encrypt_into(plaintext: &[u8], key: &Key, out: &mut Vec<u8>) -> Result<(), Error> {
if plaintext.is_empty() {
return Err(Error::EmptyPlaintext);
}
let start = out.len();
out.extend_from_slice(plaintext);
let mut cipher = Aes256Siv::new(key.as_bytes().into());
let headers: &[&[u8]] = &[];
let mut tail = TailBuffer::new(out, start);
cipher
.encrypt_in_place(headers, &mut tail)
.map_err(|_| Error::EncryptionFailed)
}
#[inline]
pub fn decrypt(ciphertext: &[u8], key: &Key) -> Result<Vec<u8>, Error> {
if ciphertext.len() < MIN_PAYLOAD_LEN {
return Err(Error::PayloadTooShort);
}
let mut cipher = Aes256Siv::new(key.as_bytes().into());
let headers: &[&[u8]] = &[];
cipher
.decrypt(headers, ciphertext)
.map_err(|_| Error::DecryptionFailed)
}
#[inline]
pub fn decrypt_into(ciphertext: &[u8], key: &Key, out: &mut Vec<u8>) -> Result<(), Error> {
if ciphertext.len() < MIN_PAYLOAD_LEN {
return Err(Error::PayloadTooShort);
}
let start = out.len();
out.extend_from_slice(ciphertext);
let mut cipher = Aes256Siv::new(key.as_bytes().into());
let headers: &[&[u8]] = &[];
let mut tail = TailBuffer::new(out, start);
cipher
.decrypt_in_place(headers, &mut tail)
.map_err(|_| Error::DecryptionFailed)
}