use crate::{PrivateKeyInfo, Result, SubjectPublicKeyInfo};
use core::convert::TryFrom;
#[cfg(feature = "alloc")]
use crate::{PrivateKeyDocument, PublicKeyDocument};
#[cfg(feature = "encryption")]
use {
crate::{EncryptedPrivateKeyDocument, EncryptedPrivateKeyInfo},
rand_core::{CryptoRng, RngCore},
};
#[cfg(feature = "pem")]
use {crate::LineEnding, alloc::string::String};
#[cfg(feature = "pkcs1")]
use crate::{Error, ObjectIdentifier};
#[cfg(feature = "std")]
use std::path::Path;
#[cfg(any(feature = "pem", feature = "std"))]
use zeroize::Zeroizing;
#[cfg(all(feature = "alloc", feature = "pkcs1"))]
use crate::AlgorithmIdentifier;
#[cfg(feature = "pkcs1")]
const PKCS1_OID: ObjectIdentifier = ObjectIdentifier::new("1.2.840.113549.1.1.1");
pub trait FromPrivateKey: Sized {
fn from_pkcs8_private_key_info(private_key_info: PrivateKeyInfo<'_>) -> Result<Self>;
fn from_pkcs8_der(bytes: &[u8]) -> Result<Self> {
Self::from_pkcs8_private_key_info(PrivateKeyInfo::try_from(bytes)?)
}
#[cfg(feature = "encryption")]
#[cfg_attr(docsrs, doc(cfg(feature = "encryption")))]
fn from_pkcs8_encrypted_der(bytes: &[u8], password: impl AsRef<[u8]>) -> Result<Self> {
EncryptedPrivateKeyInfo::try_from(bytes)?
.decrypt(password)
.and_then(|doc| Self::from_pkcs8_doc(&doc))
}
#[cfg(feature = "alloc")]
#[cfg_attr(docsrs, doc(cfg(feature = "alloc")))]
fn from_pkcs8_doc(doc: &PrivateKeyDocument) -> Result<Self> {
Self::from_pkcs8_private_key_info(doc.private_key_info())
}
#[cfg(feature = "pem")]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
fn from_pkcs8_pem(s: &str) -> Result<Self> {
PrivateKeyDocument::from_pem(s).and_then(|doc| Self::from_pkcs8_doc(&doc))
}
#[cfg(all(feature = "encryption", feature = "pem"))]
#[cfg_attr(docsrs, doc(cfg(feature = "encryption")))]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
fn from_pkcs8_encrypted_pem(s: &str, password: impl AsRef<[u8]>) -> Result<Self> {
EncryptedPrivateKeyDocument::from_pem(s)?
.decrypt(password)
.and_then(|doc| Self::from_pkcs8_doc(&doc))
}
#[cfg(feature = "std")]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
fn read_pkcs8_der_file(path: impl AsRef<Path>) -> Result<Self> {
PrivateKeyDocument::read_der_file(path).and_then(|doc| Self::from_pkcs8_doc(&doc))
}
#[cfg(all(feature = "pem", feature = "std"))]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
fn read_pkcs8_pem_file(path: impl AsRef<Path>) -> Result<Self> {
PrivateKeyDocument::read_pem_file(path).and_then(|doc| Self::from_pkcs8_doc(&doc))
}
}
pub trait FromPublicKey: Sized {
fn from_spki(spki: SubjectPublicKeyInfo<'_>) -> Result<Self>;
fn from_public_key_der(bytes: &[u8]) -> Result<Self> {
Self::from_spki(SubjectPublicKeyInfo::try_from(bytes)?)
}
#[cfg(feature = "alloc")]
#[cfg_attr(docsrs, doc(cfg(feature = "alloc")))]
fn from_public_key_doc(doc: &PublicKeyDocument) -> Result<Self> {
Self::from_spki(doc.spki())
}
#[cfg(feature = "pem")]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
fn from_public_key_pem(s: &str) -> Result<Self> {
PublicKeyDocument::from_pem(s).and_then(|doc| Self::from_public_key_doc(&doc))
}
#[cfg(feature = "std")]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
fn read_public_key_der_file(path: impl AsRef<Path>) -> Result<Self> {
PublicKeyDocument::read_der_file(path).and_then(|doc| Self::from_public_key_doc(&doc))
}
#[cfg(all(feature = "pem", feature = "std"))]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
fn read_public_key_pem_file(path: impl AsRef<Path>) -> Result<Self> {
PublicKeyDocument::read_pem_file(path).and_then(|doc| Self::from_public_key_doc(&doc))
}
}
#[cfg(feature = "alloc")]
#[cfg_attr(docsrs, doc(cfg(feature = "alloc")))]
pub trait ToPrivateKey {
fn to_pkcs8_der(&self) -> Result<PrivateKeyDocument>;
#[cfg(feature = "encryption")]
#[cfg_attr(docsrs, doc(cfg(feature = "encryption")))]
fn to_pkcs8_encrypted_der(
&self,
rng: impl CryptoRng + RngCore,
password: impl AsRef<[u8]>,
) -> Result<EncryptedPrivateKeyDocument> {
self.to_pkcs8_der()?.encrypt(rng, password)
}
#[cfg(feature = "pem")]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
fn to_pkcs8_pem(&self) -> Result<Zeroizing<String>> {
self.to_pkcs8_pem_with_le(LineEnding::default())
}
#[cfg(feature = "pem")]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
fn to_pkcs8_pem_with_le(&self, line_ending: LineEnding) -> Result<Zeroizing<String>> {
Ok(self.to_pkcs8_der()?.to_pem_with_le(line_ending))
}
#[cfg(all(feature = "encryption", feature = "pem"))]
#[cfg_attr(docsrs, doc(cfg(feature = "encryption")))]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
fn to_pkcs8_encrypted_pem(
&self,
rng: impl CryptoRng + RngCore,
password: impl AsRef<[u8]>,
) -> Result<Zeroizing<String>> {
self.to_pkcs8_encrypted_der(rng, password)
.map(|key| key.to_pem())
}
#[cfg(feature = "std")]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
fn write_pkcs8_der_file(&self, path: impl AsRef<Path>) -> Result<()> {
self.to_pkcs8_der()?.write_der_file(path)
}
#[cfg(all(feature = "pem", feature = "std"))]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
fn write_pkcs8_pem_file(&self, path: impl AsRef<Path>) -> Result<()> {
self.to_pkcs8_der()?.write_pem_file(path)
}
}
#[cfg(feature = "alloc")]
#[cfg_attr(docsrs, doc(cfg(feature = "alloc")))]
pub trait ToPublicKey {
fn to_public_key_der(&self) -> Result<PublicKeyDocument>;
#[cfg(feature = "pem")]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
fn to_public_key_pem(&self) -> Result<String> {
self.to_public_key_pem_with_le(LineEnding::default())
}
#[cfg(feature = "pem")]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
fn to_public_key_pem_with_le(&self, line_ending: LineEnding) -> Result<String> {
Ok(self.to_public_key_der()?.to_pem_with_le(line_ending))
}
#[cfg(feature = "std")]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
fn write_public_key_der_file(&self, path: impl AsRef<Path>) -> Result<()> {
self.to_public_key_der()?.write_der_file(path)
}
#[cfg(all(feature = "pem", feature = "std"))]
#[cfg_attr(docsrs, doc(cfg(feature = "pem")))]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
fn write_public_key_pem_file(&self, path: impl AsRef<Path>) -> Result<()> {
self.to_public_key_der()?.write_pem_file(path)
}
}
#[cfg(feature = "pkcs1")]
#[cfg_attr(docsrs, doc(cfg(feature = "pkcs1")))]
impl<K: pkcs1::FromRsaPrivateKey> FromPrivateKey for K {
fn from_pkcs8_private_key_info(pkcs8_key: PrivateKeyInfo<'_>) -> Result<Self> {
pkcs8_key.algorithm.assert_algorithm_oid(PKCS1_OID)?;
if pkcs8_key.algorithm.parameters != Some(der::asn1::Null.into()) {
return Err(Error::ParametersMalformed);
}
let pkcs1_key = pkcs1::RsaPrivateKey::try_from(pkcs8_key.private_key)?;
Ok(K::from_pkcs1_private_key(pkcs1_key)?)
}
}
#[cfg(feature = "pkcs1")]
#[cfg_attr(docsrs, doc(cfg(feature = "pkcs1")))]
impl<K: pkcs1::FromRsaPublicKey> FromPublicKey for K {
fn from_spki(pkcs8_key: SubjectPublicKeyInfo<'_>) -> Result<Self> {
pkcs8_key.algorithm.assert_algorithm_oid(PKCS1_OID)?;
if pkcs8_key.algorithm.parameters != Some(der::asn1::Null.into()) {
return Err(Error::ParametersMalformed);
}
let pkcs1_key = pkcs1::RsaPublicKey::try_from(pkcs8_key.subject_public_key)?;
Ok(K::from_pkcs1_public_key(pkcs1_key)?)
}
}
#[cfg(all(feature = "alloc", feature = "pkcs1"))]
#[cfg_attr(docsrs, doc(cfg(feature = "alloc")))]
#[cfg_attr(docsrs, doc(cfg(feature = "pkcs1")))]
impl<K: pkcs1::ToRsaPrivateKey> ToPrivateKey for K {
fn to_pkcs8_der(&self) -> Result<PrivateKeyDocument> {
let pkcs1_der = self.to_pkcs1_der()?;
let algorithm = AlgorithmIdentifier {
oid: PKCS1_OID,
parameters: Some(der::asn1::Null.into()),
};
Ok(PrivateKeyInfo::new(algorithm, pkcs1_der.as_ref()).to_der())
}
}
#[cfg(all(feature = "alloc", feature = "pkcs1"))]
#[cfg_attr(docsrs, doc(cfg(feature = "alloc")))]
#[cfg_attr(docsrs, doc(cfg(feature = "pkcs1")))]
impl<K: pkcs1::ToRsaPublicKey> ToPublicKey for K {
fn to_public_key_der(&self) -> Result<PublicKeyDocument> {
let pkcs1_der = self.to_pkcs1_der()?;
let algorithm = AlgorithmIdentifier {
oid: PKCS1_OID,
parameters: Some(der::asn1::Null.into()),
};
Ok(SubjectPublicKeyInfo {
algorithm,
subject_public_key: pkcs1_der.as_ref(),
}
.into())
}
}