dcrypt-api 4.0.1

// File: crates/api/src/types.rs

//! Core types with explicit ownership and memory-hygiene behavior
//!
//! Secret wrappers explicitly clear owned initialized bytes on drop. This is a
//! best-effort safe-Rust behavior, not a physical-erasure guarantee for caller,
//! compiler/register, freed-storage, swap, or crash-dump copies.

use crate::{
    error::Error,
    traits::serialize::{Serialize, SerializeSecret},
    Result,
};
use core::fmt;
use core::ops::{Deref, DerefMut};
use dcrypt_internal::constant_time::ct_eq;
pub use dcrypt_internal::zeroing::ZeroizingBytes;
use dcrypt_internal::{
    random::try_fill_bytes_zeroing_on_error,
    zeroing::{
        boxed_bytes_from_slice, boxed_bytes_zeroed, zeroizing_bytes_from_slice, Zeroize,
        ZeroizeOnDrop, Zeroizing,
    },
};

#[cfg(not(feature = "std"))]
use alloc::{boxed::Box, vec::Vec};
#[cfg(feature = "std")]
use std::{boxed::Box, vec::Vec};

/// A fixed-size byte array that explicitly clears its owned storage on drop.
#[derive(Clone)]
pub struct SecretBytes<const N: usize> {
    data: [u8; N],
}

impl<const N: usize> Zeroize for SecretBytes<N> {
    fn zeroize(&mut self) {
        self.data.zeroize();
    }
}

impl<const N: usize> ZeroizeOnDrop for SecretBytes<N> {}

impl<const N: usize> Drop for SecretBytes<N> {
    fn drop(&mut self) {
        self.zeroize();
    }
}

impl<const N: usize> SecretBytes<N> {
    pub fn new(data: [u8; N]) -> Self {
        Self { data }
    }
    pub fn from_slice(slice: &[u8]) -> Result<Self> {
        if slice.len() != N {
            return Err(Error::InvalidLength {
                context: "SecretBytes::from_slice",
                expected: N,
                actual: slice.len(),
            });
        }
        let mut data = [0u8; N];
        data.copy_from_slice(slice);
        Ok(Self { data })
    }
    pub fn zeroed() -> Self {
        Self { data: [0u8; N] }
    }
    pub fn random<R: dcrypt_internal::random::CryptoRng + ?Sized>(rng: &mut R) -> Result<Self> {
        let mut data = [0u8; N];
        try_fill_bytes_zeroing_on_error(rng, &mut data).map_err(|_| {
            Error::RandomGenerationError {
                context: "SecretBytes::random",
                #[cfg(feature = "std")]
                message: "caller-provided randomness source failed".into(),
            }
        })?;
        Ok(Self { data })
    }
    pub fn len(&self) -> usize {
        N
    }
    pub fn is_empty(&self) -> bool {
        N == 0
    }

    /// Serialize into exact-size storage that clears itself on drop.
    pub fn to_bytes_zeroizing_boxed(&self) -> ZeroizingBytes {
        zeroizing_bytes_from_slice(&self.data)
    }
}

impl<const N: usize> AsRef<[u8]> for SecretBytes<N> {
    fn as_ref(&self) -> &[u8] {
        &self.data
    }
}

impl<const N: usize> AsMut<[u8]> for SecretBytes<N> {
    fn as_mut(&mut self) -> &mut [u8] {
        &mut self.data
    }
}

impl<const N: usize> Deref for SecretBytes<N> {
    type Target = [u8; N];
    fn deref(&self) -> &Self::Target {
        &self.data
    }
}

impl<const N: usize> DerefMut for SecretBytes<N> {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.data
    }
}

impl<const N: usize> PartialEq for SecretBytes<N> {
    fn eq(&self, other: &Self) -> bool {
        ct_eq(self.data.as_slice(), other.data.as_slice())
    }
}

impl<const N: usize> Eq for SecretBytes<N> {}

impl<const N: usize> fmt::Debug for SecretBytes<N> {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "SecretBytes<{}>[REDACTED]", N)
    }
}

impl<const N: usize> SerializeSecret for SecretBytes<N> {
    fn from_bytes(bytes: &[u8]) -> Result<Self> {
        Self::from_slice(bytes)
    }
    fn to_bytes_zeroizing(&self) -> ZeroizingBytes {
        self.to_bytes_zeroizing_boxed()
    }
}

/// Exact-size owned bytes whose initialized storage is explicitly cleared on drop.
///
/// This is best-effort software memory hygiene, not a physical-erasure guarantee
/// for caller, compiler/register, allocator, swap, or crash-dump copies.
pub struct SecretVec {
    data: Box<[u8]>,
}

impl Zeroize for SecretVec {
    fn zeroize(&mut self) {
        self.data.zeroize();
    }
}

impl ZeroizeOnDrop for SecretVec {}

impl Drop for SecretVec {
    fn drop(&mut self) {
        self.zeroize();
    }
}

impl SecretVec {
    /// Take ownership of an exact-size boxed secret allocation.
    ///
    /// This constructor intentionally does not accept `Vec<u8>`: safe Rust
    /// cannot inspect or erase a vector's spare capacity. Use [`from_slice`](Self::from_slice)
    /// when copying from caller-owned storage.
    pub fn new(data: Box<[u8]>) -> Self {
        Self { data }
    }

    /// Copy a slice directly into exact-size owned storage.
    pub fn from_slice(slice: &[u8]) -> Self {
        Self::new(boxed_bytes_from_slice(slice))
    }

    /// Create an empty secret value.
    pub fn empty() -> Self {
        Self::new(boxed_bytes_zeroed(0))
    }

    /// Create an exact-size zero-filled secret value.
    pub fn zeroed(len: usize) -> Self {
        Self::new(boxed_bytes_zeroed(len))
    }

    /// Create an exact-size secret value from caller-provided randomness.
    /// Any partial output is erased if the RNG reports failure.
    pub fn random<R: dcrypt_internal::random::CryptoRng + ?Sized>(
        rng: &mut R,
        len: usize,
    ) -> Result<Self> {
        let mut data = boxed_bytes_zeroed(len);
        try_fill_bytes_zeroing_on_error(rng, &mut data).map_err(|_| {
            Error::RandomGenerationError {
                context: "SecretVec::random",
                #[cfg(feature = "std")]
                message: "caller-provided randomness source failed".into(),
            }
        })?;
        Ok(Self::new(data))
    }
    pub fn len(&self) -> usize {
        self.data.len()
    }
    pub fn is_empty(&self) -> bool {
        self.data.is_empty()
    }

    /// Return the secret bytes as a slice.
    pub fn as_slice(&self) -> &[u8] {
        &self.data
    }

    /// Return the secret bytes as a mutable slice.
    ///
    /// Slice access permits in-place byte changes but cannot resize or
    /// reallocate the backing storage.
    pub fn as_mut_slice(&mut self) -> &mut [u8] {
        &mut self.data
    }

    /// Copy into exact-size storage that clears itself on drop.
    pub fn to_bytes_zeroizing_boxed(&self) -> ZeroizingBytes {
        zeroizing_bytes_from_slice(&self.data)
    }

    /// Consume this value without copying, transferring its exact-size
    /// allocation into a zeroizing wrapper.
    pub fn into_bytes_zeroizing_boxed(mut self) -> ZeroizingBytes {
        let data = core::mem::replace(&mut self.data, boxed_bytes_zeroed(0));
        Zeroizing::new(data)
    }

    /// Return the allocation capacity.
    pub fn capacity(&self) -> usize {
        self.data.len()
    }

    /// Extend the value, wiping the old allocation before freeing it if growth
    /// requires a larger allocation.
    pub fn extend_from_slice(&mut self, slice: &[u8]) {
        let new_len = self
            .data
            .len()
            .checked_add(slice.len())
            .expect("SecretVec length overflow");
        let mut replacement = boxed_bytes_zeroed(new_len);
        replacement[..self.data.len()].copy_from_slice(&self.data);
        replacement[self.data.len()..].copy_from_slice(slice);
        self.replace_and_zeroize(replacement);
    }

    /// Resize the value. Bytes removed by shrinking are wiped first.
    pub fn resize(&mut self, new_len: usize, value: u8) {
        if new_len <= self.data.len() {
            self.truncate(new_len);
            return;
        }

        let mut replacement = boxed_bytes_zeroed(new_len);
        replacement[..self.data.len()].copy_from_slice(&self.data);
        replacement[self.data.len()..].fill(value);
        self.replace_and_zeroize(replacement);
    }

    /// Shorten the value, wiping every removed byte before changing its length.
    pub fn truncate(&mut self, len: usize) {
        if len >= self.data.len() {
            return;
        }

        let replacement = boxed_bytes_from_slice(&self.data[..len]);
        self.replace_and_zeroize(replacement);
    }

    /// Remove all bytes while retaining a fully zeroed allocation.
    pub fn clear(&mut self) {
        self.replace_and_zeroize(boxed_bytes_zeroed(0));
    }

    /// Append one byte, securely replacing the allocation when it is full.
    pub fn push(&mut self, value: u8) {
        self.extend_from_slice(&[value]);
    }

    /// Remove and return the last byte, wiping its allocation slot first.
    pub fn pop(&mut self) -> Option<u8> {
        let value = self.data.last().copied()?;
        self.truncate(self.data.len() - 1);
        Some(value)
    }

    fn replace_and_zeroize(&mut self, replacement: Box<[u8]>) {
        self.data.zeroize();
        self.data = replacement;
    }
}

impl Clone for SecretVec {
    fn clone(&self) -> Self {
        Self::from_slice(&self.data)
    }
}

impl From<Box<[u8]>> for SecretVec {
    fn from(data: Box<[u8]>) -> Self {
        Self::new(data)
    }
}

impl AsRef<[u8]> for SecretVec {
    fn as_ref(&self) -> &[u8] {
        self.data.as_ref()
    }
}

impl AsMut<[u8]> for SecretVec {
    fn as_mut(&mut self) -> &mut [u8] {
        self.data.as_mut()
    }
}

impl Deref for SecretVec {
    type Target = [u8];
    fn deref(&self) -> &Self::Target {
        self.data.as_ref()
    }
}

impl DerefMut for SecretVec {
    fn deref_mut(&mut self) -> &mut Self::Target {
        self.data.as_mut()
    }
}

impl PartialEq for SecretVec {
    fn eq(&self, other: &Self) -> bool {
        ct_eq(self.data.as_ref(), other.data.as_ref())
    }
}

impl Eq for SecretVec {}

impl fmt::Debug for SecretVec {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "SecretVec({})[REDACTED]", self.data.len())
    }
}

impl SerializeSecret for SecretVec {
    fn from_bytes(bytes: &[u8]) -> Result<Self> {
        Ok(Self::from_slice(bytes))
    }
    fn to_bytes_zeroizing(&self) -> ZeroizingBytes {
        self.to_bytes_zeroizing_boxed()
    }
}

#[cfg(test)]
mod secret_vec_tests {
    use super::{Key, SecretBytes, SecretVec};
    use crate::traits::SerializeSecret;
    use dcrypt_internal::random::{CryptoRng, Error as RandomError, RngCore};

    #[test]
    fn constructor_uses_exact_size_storage() {
        let secret = SecretVec::new([0xA5; 4].into());

        assert_eq!(secret.as_slice(), &[0xA5; 4]);
        assert_eq!(secret.capacity(), secret.len());
    }

    #[test]
    fn shrinking_operations_wipe_removed_slots() {
        let mut secret = SecretVec::from_slice(&[1, 2, 3, 4, 5, 6]);

        secret.truncate(4);
        assert_eq!(secret.as_slice(), &[1, 2, 3, 4]);
        assert_eq!(secret.capacity(), secret.len());

        secret.resize(2, 0xFF);
        assert_eq!(secret.as_slice(), &[1, 2]);
        assert_eq!(secret.capacity(), secret.len());

        assert_eq!(secret.pop(), Some(2));
        assert_eq!(secret.as_slice(), &[1]);
        assert_eq!(secret.capacity(), secret.len());

        secret.clear();
        assert!(secret.is_empty());
        assert_eq!(secret.capacity(), secret.len());
    }

    #[test]
    fn growth_and_shrink_replace_allocations_securely() {
        let mut secret = SecretVec::from_slice(&[0x11; 4]);
        secret.extend_from_slice(&[0x22, 0x33]);
        assert_eq!(secret.capacity(), 6);
        assert_eq!(secret.as_slice(), &[0x11, 0x11, 0x11, 0x11, 0x22, 0x33]);
    }

    #[test]
    fn exact_size_serialization_supports_copy_and_ownership_transfer() {
        let fixed = SecretBytes::<4>::new([1, 2, 3, 4]);
        let fixed_bytes = fixed.to_bytes_zeroizing_boxed();
        assert_eq!(&**fixed_bytes, &[1, 2, 3, 4]);
        let trait_fixed_bytes = fixed.to_bytes_zeroizing();
        assert_eq!(&**trait_fixed_bytes, &[1, 2, 3, 4]);

        let secret = SecretVec::from_slice(&[5, 6, 7]);
        let copied = secret.to_bytes_zeroizing_boxed();
        assert_eq!(&**copied, &[5, 6, 7]);
        let trait_copied = secret.to_bytes_zeroizing();
        assert_eq!(&**trait_copied, &[5, 6, 7]);

        let transferred = secret.into_bytes_zeroizing_boxed();
        assert_eq!(&**transferred, &[5, 6, 7]);

        let key = Key::from_slice(&[8, 9]);
        let key_bytes = key.to_bytes_zeroizing_boxed();
        assert_eq!(&**key_bytes, &[8, 9]);
        let trait_key_bytes = key.to_bytes_zeroizing();
        assert_eq!(&**trait_key_bytes, &[8, 9]);
    }

    struct PartiallyFailingRng;

    impl RngCore for PartiallyFailingRng {
        fn try_fill_bytes(&mut self, destination: &mut [u8]) -> Result<(), RandomError> {
            let written = core::cmp::min(destination.len(), 5);
            destination[..written].fill(0xA5);
            Err(RandomError)
        }
    }

    impl CryptoRng for PartiallyFailingRng {}

    #[test]
    fn random_secret_constructors_propagate_caller_rng_failure() {
        let mut rng = PartiallyFailingRng;
        assert!(SecretBytes::<32>::random(&mut rng).is_err());
        assert!(SecretVec::random(&mut rng, 32).is_err());
    }
}

/// Base key type backed by exact-size secret storage.
#[derive(Clone)]
pub struct Key {
    data: Box<[u8]>,
}

impl Zeroize for Key {
    fn zeroize(&mut self) {
        self.data.zeroize();
    }
}

impl ZeroizeOnDrop for Key {}

impl Drop for Key {
    fn drop(&mut self) {
        self.zeroize();
    }
}

impl Key {
    /// Copy key bytes directly into exact-size owned storage.
    ///
    /// The caller retains responsibility for clearing the source buffer. This
    /// constructor intentionally accepts a slice rather than taking a `Vec`:
    /// safe Rust cannot inspect or erase a vector's spare capacity.
    pub fn new(data: &[u8]) -> Self {
        Self::from_boxed_slice(boxed_bytes_from_slice(data))
    }

    /// Take ownership of an exact-size boxed key allocation.
    pub fn from_boxed_slice(data: Box<[u8]>) -> Self {
        Self { data }
    }

    /// Copy key bytes directly into exact-size owned storage.
    pub fn from_slice(data: &[u8]) -> Self {
        Self::new(data)
    }

    pub fn new_zeros(len: usize) -> Self {
        Self::from_boxed_slice(boxed_bytes_zeroed(len))
    }
    pub fn len(&self) -> usize {
        self.data.len()
    }
    pub fn is_empty(&self) -> bool {
        self.data.is_empty()
    }

    /// Serialize into exact-size storage that clears itself on drop.
    pub fn to_bytes_zeroizing_boxed(&self) -> ZeroizingBytes {
        zeroizing_bytes_from_slice(&self.data)
    }
}

impl From<Box<[u8]>> for Key {
    fn from(data: Box<[u8]>) -> Self {
        Self::from_boxed_slice(data)
    }
}

impl AsRef<[u8]> for Key {
    fn as_ref(&self) -> &[u8] {
        self.data.as_ref()
    }
}

impl AsMut<[u8]> for Key {
    fn as_mut(&mut self) -> &mut [u8] {
        self.data.as_mut()
    }
}

impl SerializeSecret for Key {
    fn from_bytes(bytes: &[u8]) -> Result<Self> {
        Ok(Self::from_slice(bytes))
    }
    fn to_bytes_zeroizing(&self) -> ZeroizingBytes {
        self.to_bytes_zeroizing_boxed()
    }
}

/// Wrapper for public key data
#[derive(Clone)]
pub struct PublicKey {
    data: Vec<u8>,
}

impl PublicKey {
    /// Create a new PublicKey.
    ///
    /// Accepts `Vec<u8>` (move) or `&[u8]` (copy).
    /// Moving a `Vec<u8>` avoids unnecessary allocation.
    pub fn new<T: Into<Vec<u8>>>(data: T) -> Self {
        Self { data: data.into() }
    }
    pub fn len(&self) -> usize {
        self.data.len()
    }
    pub fn is_empty(&self) -> bool {
        self.data.is_empty()
    }
}

impl From<Vec<u8>> for PublicKey {
    fn from(data: Vec<u8>) -> Self {
        Self::new(data)
    }
}

impl AsRef<[u8]> for PublicKey {
    fn as_ref(&self) -> &[u8] {
        &self.data
    }
}

impl AsMut<[u8]> for PublicKey {
    fn as_mut(&mut self) -> &mut [u8] {
        &mut self.data
    }
}

impl Serialize for PublicKey {
    fn to_bytes(&self) -> Vec<u8> {
        self.data.clone()
    }
    fn from_bytes(bytes: &[u8]) -> Result<Self> {
        Ok(Self::new(bytes))
    }
}

/// Wrapper for ciphertext data
#[derive(Clone)]
pub struct Ciphertext {
    data: Vec<u8>,
}

impl Ciphertext {
    /// Create a new Ciphertext.
    ///
    /// Accepts `Vec<u8>` (move) or `&[u8]` (copy).
    /// Moving a `Vec<u8>` avoids unnecessary allocation, which is critical
    /// for large ciphertexts.
    pub fn new<T: Into<Vec<u8>>>(data: T) -> Self {
        Self { data: data.into() }
    }
    pub fn len(&self) -> usize {
        self.data.len()
    }
    pub fn is_empty(&self) -> bool {
        self.data.is_empty()
    }
}

impl From<Vec<u8>> for Ciphertext {
    fn from(data: Vec<u8>) -> Self {
        Self::new(data)
    }
}

impl AsRef<[u8]> for Ciphertext {
    fn as_ref(&self) -> &[u8] {
        &self.data
    }
}

impl AsMut<[u8]> for Ciphertext {
    fn as_mut(&mut self) -> &mut [u8] {
        &mut self.data
    }
}

impl Serialize for Ciphertext {
    fn to_bytes(&self) -> Vec<u8> {
        self.data.clone()
    }
    fn from_bytes(bytes: &[u8]) -> Result<Self> {
        Ok(Self::new(bytes))
    }
}