use crate::{Error, Result};
use core::{convert::TryFrom, fmt};
use der::{
asn1::{Any, OctetString},
Decodable, Encodable, Message,
};
use pkcs5::EncryptionScheme;
#[cfg(feature = "alloc")]
use crate::{EncryptedPrivateKeyDocument, PrivateKeyDocument};
#[cfg(feature = "encryption")]
use core::convert::TryInto;
#[cfg(feature = "pem")]
use {
crate::{error, pem, LineEnding},
zeroize::Zeroizing,
};
#[cfg(feature = "pem")]
pub(crate) const PEM_TYPE_LABEL: &str = "ENCRYPTED PRIVATE KEY";
#[cfg_attr(docsrs, doc(cfg(feature = "pkcs5")))]
#[derive(Clone, Eq, PartialEq)]
pub struct EncryptedPrivateKeyInfo<'a> {
pub encryption_algorithm: EncryptionScheme<'a>,
pub encrypted_data: &'a [u8],
}
impl<'a> EncryptedPrivateKeyInfo<'a> {
#[cfg(feature = "encryption")]
#[cfg_attr(docsrs, doc(cfg(feature = "encryption")))]
pub fn decrypt(&self, password: impl AsRef<[u8]>) -> Result<PrivateKeyDocument> {
self.encryption_algorithm
.decrypt(password, &self.encrypted_data)
.map_err(|_| Error::Crypto)
.and_then(TryInto::try_into)
}
#[cfg(feature = "alloc")]
#[cfg_attr(docsrs, doc(cfg(feature = "alloc")))]
pub fn to_der(&self) -> EncryptedPrivateKeyDocument {
self.into()
}
#[cfg(feature = "pem")]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
pub fn to_pem(&self) -> Zeroizing<alloc::string::String> {
self.to_pem_with_le(LineEnding::default())
}
#[cfg(feature = "pem")]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
pub fn to_pem_with_le(&self, line_ending: LineEnding) -> Zeroizing<alloc::string::String> {
Zeroizing::new(
pem::encode_string(PEM_TYPE_LABEL, line_ending, self.to_der().as_ref())
.expect(error::PEM_ENCODING_MSG),
)
}
}
impl<'a> TryFrom<&'a [u8]> for EncryptedPrivateKeyInfo<'a> {
type Error = Error;
fn try_from(bytes: &'a [u8]) -> Result<Self> {
Ok(Self::from_der(bytes)?)
}
}
impl<'a> TryFrom<Any<'a>> for EncryptedPrivateKeyInfo<'a> {
type Error = der::Error;
fn try_from(any: Any<'a>) -> der::Result<EncryptedPrivateKeyInfo<'a>> {
any.sequence(|decoder| {
Ok(Self {
encryption_algorithm: decoder.decode()?,
encrypted_data: decoder.octet_string()?.as_bytes(),
})
})
}
}
impl<'a> Message<'a> for EncryptedPrivateKeyInfo<'a> {
fn fields<F, T>(&self, f: F) -> der::Result<T>
where
F: FnOnce(&[&dyn Encodable]) -> der::Result<T>,
{
f(&[
&self.encryption_algorithm,
&OctetString::new(self.encrypted_data)?,
])
}
}
impl<'a> fmt::Debug for EncryptedPrivateKeyInfo<'a> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("EncryptedPrivateKeyInfo")
.field("encryption_algorithm", &self.encryption_algorithm)
.finish() }
}