#[cfg(feature = "alloc")]
use alloc::vec::Vec;
use core::cmp::Ordering;
use core::fmt;
use core::hash::{Hash, Hasher};
#[cfg(feature = "alloc")]
use crypto_bigint::Resize as _;
#[cfg(feature = "alloc")]
use crypto_bigint::{
modular::{BoxedMontyForm, BoxedMontyParams},
BoxedUint, ConcatenatingMul, Integer,
};
#[cfg(feature = "alloc")]
use crypto_bigint::{NonZero as CryptoNonZero, Odd as CryptoOdd};
use rand_core::CryptoRng;
use zeroize::{Zeroize, ZeroizeOnDrop};
#[cfg(feature = "serde")]
use {
pkcs8::{DecodePrivateKey, EncodePrivateKey},
serdect::serde::{de, ser, Deserialize, Serialize},
spki::{DecodePublicKey, EncodePublicKey},
};
#[cfg(feature = "private-key")]
use crate::algorithms::generate::generate_multi_prime_key_with_exp;
#[cfg(feature = "private-key")]
use crate::algorithms::rsa::{
compute_modulus, compute_private_exponent_carmicheal, compute_private_exponent_euler_totient,
recover_primes,
};
#[cfg(feature = "private-key")]
use crate::dummy_rng::DummyRng;
use crate::errors::{Error, Result};
use crate::traits::keys::PublicKeyParts;
#[cfg(feature = "private-key")]
use crate::traits::keys::{CrtValue, PrivateKeyParts};
use crate::traits::{
modular::ModulusParams, NonZero, PaddingScheme, SignatureScheme, UnsignedModularInt,
};
#[derive(Debug, Clone)]
pub struct GenericRsaPublicKey<T, M>
where
T: UnsignedModularInt,
M: ModulusParams<Modulus = T>,
{
n: NonZero<T>,
e: T,
n_params: M,
}
#[cfg(feature = "alloc")]
pub type RsaPublicKey = GenericRsaPublicKey<BoxedUint, BoxedMontyParams>;
impl<T, M> Eq for GenericRsaPublicKey<T, M>
where
T: UnsignedModularInt + Eq,
M: ModulusParams<Modulus = T>,
{
}
impl<T, M> PartialEq for GenericRsaPublicKey<T, M>
where
T: UnsignedModularInt + PartialEq,
M: ModulusParams<Modulus = T>,
{
#[inline]
fn eq(&self, other: &GenericRsaPublicKey<T, M>) -> bool {
self.n == other.n && self.e == other.e
}
}
impl<T, M> Hash for GenericRsaPublicKey<T, M>
where
T: UnsignedModularInt,
M: ModulusParams<Modulus = T>,
{
fn hash<H: Hasher>(&self, state: &mut H) {
state.write(b"RsaPublicKey");
state.write(self.n.as_ref().to_be_bytes().as_ref());
state.write(self.e.to_be_bytes().as_ref());
}
}
#[cfg(feature = "private-key")]
#[derive(Clone)]
pub struct RsaPrivateKey {
pubkey_components: RsaPublicKey,
pub(crate) d: BoxedUint,
pub(crate) primes: Vec<BoxedUint>,
pub(crate) precomputed: Option<PrecomputedValues>,
}
#[cfg(feature = "private-key")]
impl fmt::Debug for RsaPrivateKey {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let precomputed = if self.precomputed.is_some() {
"Some(...)"
} else {
"None"
};
f.debug_struct("RsaPrivateKey")
.field("pubkey_components", &self.pubkey_components)
.field("d", &"...")
.field("primes", &"&[...]")
.field("precomputed", &precomputed)
.finish()
}
}
#[cfg(feature = "private-key")]
impl Eq for RsaPrivateKey {}
#[cfg(feature = "private-key")]
impl PartialEq for RsaPrivateKey {
#[inline]
fn eq(&self, other: &RsaPrivateKey) -> bool {
self.pubkey_components == other.pubkey_components
&& self.d == other.d
&& self.primes == other.primes
}
}
#[cfg(feature = "private-key")]
impl AsRef<RsaPublicKey> for RsaPrivateKey {
fn as_ref(&self) -> &RsaPublicKey {
&self.pubkey_components
}
}
#[cfg(feature = "private-key")]
impl Hash for RsaPrivateKey {
fn hash<H: Hasher>(&self, state: &mut H) {
state.write(b"RsaPrivateKey");
Hash::hash(&self.pubkey_components, state);
}
}
#[cfg(feature = "private-key")]
impl Drop for RsaPrivateKey {
fn drop(&mut self) {
self.d.zeroize();
self.primes.zeroize();
self.precomputed.zeroize();
}
}
#[cfg(feature = "private-key")]
impl ZeroizeOnDrop for RsaPrivateKey {}
#[cfg(feature = "private-key")]
#[derive(Clone)]
pub(crate) struct PrecomputedValues {
pub(crate) dp: BoxedUint,
pub(crate) dq: BoxedUint,
pub(crate) qinv: BoxedMontyForm,
pub(crate) p_params: BoxedMontyParams,
pub(crate) q_params: BoxedMontyParams,
}
#[cfg(feature = "private-key")]
impl ZeroizeOnDrop for PrecomputedValues {}
#[cfg(feature = "private-key")]
impl Zeroize for PrecomputedValues {
fn zeroize(&mut self) {
self.dp.zeroize();
self.dq.zeroize();
}
}
#[cfg(feature = "private-key")]
impl Drop for PrecomputedValues {
fn drop(&mut self) {
self.zeroize();
}
}
#[cfg(feature = "private-key")]
impl From<RsaPrivateKey> for GenericRsaPublicKey<BoxedUint, BoxedMontyParams> {
fn from(private_key: RsaPrivateKey) -> Self {
(&private_key).into()
}
}
#[cfg(feature = "private-key")]
impl From<&RsaPrivateKey> for GenericRsaPublicKey<BoxedUint, BoxedMontyParams> {
fn from(private_key: &RsaPrivateKey) -> Self {
let public_key: &dyn PublicKeyParts<BoxedUint, MontyParams = BoxedMontyParams> =
private_key;
GenericRsaPublicKey {
n: public_key.n().clone(),
e: public_key.e().clone(),
n_params: public_key.n_params().clone(),
}
}
}
impl<T, M> PublicKeyParts<T> for GenericRsaPublicKey<T, M>
where
T: UnsignedModularInt,
M: ModulusParams<Modulus = T>,
{
type MontyParams = M;
fn n(&self) -> &NonZero<T> {
&self.n
}
fn e(&self) -> &T {
&self.e
}
fn n_params(&self) -> &M {
&self.n_params
}
}
impl<T, M> GenericRsaPublicKey<T, M>
where
T: UnsignedModularInt,
M: ModulusParams<Modulus = T>,
{
pub fn from_components(n: T, e: T, n_params: M) -> Result<Self> {
let n = NonZero::new(n).ok_or(Error::InvalidModulus)?;
Ok(Self { n, e, n_params })
}
}
impl<T, M> GenericRsaPublicKey<T, M>
where
T: UnsignedModularInt,
M: ModulusParams<Modulus = T>,
{
#[cfg(feature = "alloc")]
pub fn encrypt<R: CryptoRng + ?Sized, P: PaddingScheme>(
&self,
rng: &mut R,
padding: P,
msg: &[u8],
) -> Result<Vec<u8>> {
padding.encrypt(rng, self, msg)
}
pub fn verify<S: SignatureScheme>(&self, scheme: S, hashed: &[u8], sig: &[u8]) -> Result<()> {
scheme.verify(self, hashed, sig)
}
}
#[cfg(feature = "alloc")]
impl GenericRsaPublicKey<BoxedUint, BoxedMontyParams> {
pub const MIN_PUB_EXPONENT: u64 = 2;
pub const MAX_PUB_EXPONENT: u64 = (1 << 33) - 1;
pub const MAX_SIZE: usize = 8192;
pub fn new(n: BoxedUint, e: BoxedUint) -> Result<Self> {
Self::new_with_max_size(n, e, Self::MAX_SIZE)
}
pub fn new_with_max_size(n: BoxedUint, e: BoxedUint, max_size: usize) -> Result<Self> {
check_public_with_max_size(&n, &e, Some(max_size))?;
let n_odd = CryptoOdd::new(n.clone())
.into_option()
.ok_or(Error::InvalidModulus)?;
let n_params = BoxedMontyParams::new(n_odd);
let n = NonZero::new(n).expect("checked above");
Ok(Self { n, e, n_params })
}
pub fn new_unchecked(n: BoxedUint, e: BoxedUint) -> Self {
let n_odd = CryptoOdd::new(n.clone()).expect("n must be odd");
let n_params = BoxedMontyParams::new(n_odd);
let n = NonZero::new(n).expect("odd numbers are non zero");
Self { n, e, n_params }
}
}
#[cfg(feature = "private-key")]
impl PublicKeyParts<BoxedUint> for RsaPrivateKey {
type MontyParams = BoxedMontyParams;
fn n(&self) -> &NonZero<BoxedUint> {
&self.pubkey_components.n
}
fn e(&self) -> &BoxedUint {
&self.pubkey_components.e
}
fn n_params(&self) -> &BoxedMontyParams {
&self.pubkey_components.n_params
}
}
#[cfg(feature = "private-key")]
impl RsaPrivateKey {
const EXP: u64 = 65537;
const MIN_SIZE: u32 = 1024;
pub fn new<R: CryptoRng + ?Sized>(rng: &mut R, bit_size: usize) -> Result<Self> {
Self::new_with_exp(rng, bit_size, Self::EXP.into())
}
#[cfg(feature = "hazmat")]
pub fn new_unchecked<R: CryptoRng + ?Sized>(rng: &mut R, bit_size: usize) -> Result<Self> {
Self::new_with_exp_unchecked(rng, bit_size, Self::EXP.into())
}
pub fn new_with_exp<R: CryptoRng + ?Sized>(
rng: &mut R,
bit_size: usize,
exp: BoxedUint,
) -> Result<RsaPrivateKey> {
if bit_size < Self::MIN_SIZE as usize {
return Err(Error::ModulusTooSmall);
}
let components = generate_multi_prime_key_with_exp(rng, 2, bit_size, exp)?;
RsaPrivateKey::from_components(
components.n.get(),
components.e,
components.d,
components.primes,
)
}
#[cfg(feature = "hazmat")]
pub fn new_with_exp_unchecked<R: CryptoRng + ?Sized>(
rng: &mut R,
bit_size: usize,
exp: BoxedUint,
) -> Result<RsaPrivateKey> {
let components = generate_multi_prime_key_with_exp(rng, 2, bit_size, exp)?;
RsaPrivateKey::from_components(
components.n.get(),
components.e,
components.d,
components.primes,
)
}
fn from_components_inner(
n: BoxedUint,
e: BoxedUint,
d: BoxedUint,
mut primes: Vec<BoxedUint>,
) -> Result<RsaPrivateKey> {
let n = CryptoOdd::new(n)
.into_option()
.ok_or(Error::InvalidModulus)?;
let n_bits = n.bits_vartime();
let n = n.resize_unchecked(n_bits);
let n_params = BoxedMontyParams::new(n.clone());
let n_c = NonZero::new(n.get()).ok_or(Error::InvalidModulus)?;
match primes.len() {
0 => {
let n_for_recovery =
CryptoNonZero::new(n_c.as_ref().clone()).expect("modulus is non-zero");
let (p, q) = recover_primes(&n_for_recovery, &e, &d)?;
primes.push(p);
primes.push(q);
}
1 => return Err(Error::NprimesTooSmall),
_ => {
if primes
.iter()
.fold(BoxedUint::one(), |acc, p| acc.concatenating_mul(&p))
!= n_c.as_ref()
{
return Err(Error::InvalidModulus);
}
}
}
let primes = primes
.into_iter()
.map(|p| {
let p_bits = p.bits();
p.resize_unchecked(p_bits)
})
.collect();
let k = RsaPrivateKey {
pubkey_components: RsaPublicKey {
n: n_c,
e,
n_params,
},
d,
primes,
precomputed: None,
};
Ok(k)
}
#[cfg(all(feature = "hazmat", feature = "private-key"))]
pub fn from_components_with_large_exponent(
n: BoxedUint,
e: BoxedUint,
d: BoxedUint,
primes: Vec<BoxedUint>,
) -> Result<RsaPrivateKey> {
let mut k = Self::from_components_inner(n, e, d, primes)?;
validate_skip_exponent_size(&k)?;
k.precompute().ok();
Ok(k)
}
pub fn from_components(
n: BoxedUint,
e: BoxedUint,
d: BoxedUint,
primes: Vec<BoxedUint>,
) -> Result<RsaPrivateKey> {
let mut k = Self::from_components_inner(n, e, d, primes)?;
k.validate()?;
k.precompute().ok();
Ok(k)
}
pub fn from_p_q(
p: BoxedUint,
q: BoxedUint,
public_exponent: BoxedUint,
) -> Result<RsaPrivateKey> {
if p == q {
return Err(Error::InvalidPrime);
}
let d = compute_private_exponent_carmicheal(&p, &q, &public_exponent)?;
let primes = vec![p, q];
let n = compute_modulus(&primes);
Self::from_components(n.get(), public_exponent, d, primes)
}
pub fn from_primes(
primes: Vec<BoxedUint>,
public_exponent: BoxedUint,
) -> Result<RsaPrivateKey> {
if primes.len() < 2 {
return Err(Error::NprimesTooSmall);
}
for (i, prime1) in primes.iter().enumerate() {
for prime2 in primes.iter().take(i) {
if prime1 == prime2 {
return Err(Error::InvalidPrime);
}
}
}
let n = compute_modulus(&primes);
let d = compute_private_exponent_euler_totient(&primes, &public_exponent)?;
Self::from_components(n.get(), public_exponent, d, primes)
}
pub fn as_public_key(&self) -> &RsaPublicKey {
&self.pubkey_components
}
pub fn to_public_key(&self) -> RsaPublicKey {
self.pubkey_components.clone()
}
pub fn precompute(&mut self) -> Result<()> {
if self.precomputed.is_some() {
return Ok(());
}
let d = &self.d;
let p = self.primes[0].clone();
let q = self.primes[1].clone();
let p_odd = CryptoOdd::new(p.clone())
.into_option()
.ok_or(Error::InvalidPrime)?;
let p_params = BoxedMontyParams::new(p_odd);
let q_odd = CryptoOdd::new(q.clone())
.into_option()
.ok_or(Error::InvalidPrime)?;
let q_params = BoxedMontyParams::new(q_odd);
let x = CryptoNonZero::new(p.wrapping_sub(BoxedUint::one()))
.into_option()
.ok_or(Error::InvalidPrime)?;
let dp = d.rem_vartime(&x);
let x = CryptoNonZero::new(q.wrapping_sub(BoxedUint::one()))
.into_option()
.ok_or(Error::InvalidPrime)?;
let dq = d.rem_vartime(&x);
let q_mod_p = match p.bits_precision().cmp(&q.bits_precision()) {
Ordering::Less => {
let p_wide = CryptoNonZero::new(p.clone())
.expect("`p` is non-zero")
.resize_unchecked(q.bits_precision());
(&q % p_wide).resize_unchecked(p.bits_precision())
}
Ordering::Greater => {
(&q).resize_unchecked(p.bits_precision())
% &CryptoNonZero::new(p.clone()).expect("`p` is non-zero")
}
Ordering::Equal => &q % CryptoNonZero::new(p.clone()).expect("`p` is non-zero"),
};
let q_mod_p = BoxedMontyForm::new(q_mod_p, &p_params);
let qinv = q_mod_p.invert().into_option().ok_or(Error::InvalidPrime)?;
debug_assert_eq!(dp.bits_precision(), p.bits_precision());
debug_assert_eq!(dq.bits_precision(), q.bits_precision());
debug_assert_eq!(qinv.bits_precision(), p.bits_precision());
debug_assert_eq!(p_params.bits_precision(), p.bits_precision());
debug_assert_eq!(q_params.bits_precision(), q.bits_precision());
self.precomputed = Some(PrecomputedValues {
dp,
dq,
qinv,
p_params,
q_params,
});
Ok(())
}
pub fn clear_precomputed(&mut self) {
self.precomputed = None;
}
pub fn crt_coefficient(&self) -> Option<BoxedUint> {
let p = &self.primes[0];
let q = &self.primes[1];
Option::from(q.invert_mod(&CryptoNonZero::new(p.clone()).expect("prime")))
}
pub fn validate(&self) -> Result<()> {
check_public(self)?;
validate_private_key_parts(self)?;
Ok(())
}
pub fn decrypt<P: PaddingScheme>(&self, padding: P, ciphertext: &[u8]) -> Result<Vec<u8>> {
padding.decrypt(Option::<&mut DummyRng>::None, self, ciphertext)
}
pub fn decrypt_blinded<R: CryptoRng + ?Sized, P: PaddingScheme>(
&self,
rng: &mut R,
padding: P,
ciphertext: &[u8],
) -> Result<Vec<u8>> {
padding.decrypt(Some(rng), self, ciphertext)
}
pub fn sign<S: SignatureScheme>(&self, padding: S, digest_in: &[u8]) -> Result<Vec<u8>> {
padding.sign(Option::<&mut DummyRng>::None, self, digest_in)
}
pub fn sign_with_rng<R: CryptoRng + ?Sized, S: SignatureScheme>(
&self,
rng: &mut R,
padding: S,
digest_in: &[u8],
) -> Result<Vec<u8>> {
padding.sign(Some(rng), self, digest_in)
}
}
#[cfg(feature = "private-key")]
impl PrivateKeyParts for RsaPrivateKey {
fn d(&self) -> &BoxedUint {
&self.d
}
fn primes(&self) -> &[BoxedUint] {
&self.primes
}
fn dp(&self) -> Option<&BoxedUint> {
self.precomputed.as_ref().map(|p| &p.dp)
}
fn dq(&self) -> Option<&BoxedUint> {
self.precomputed.as_ref().map(|p| &p.dq)
}
fn qinv(&self) -> Option<&BoxedMontyForm> {
self.precomputed.as_ref().map(|p| &p.qinv)
}
fn crt_values(&self) -> Option<&[CrtValue]> {
None
}
fn p_params(&self) -> Option<&BoxedMontyParams> {
self.precomputed.as_ref().map(|p| &p.p_params)
}
fn q_params(&self) -> Option<&BoxedMontyParams> {
self.precomputed.as_ref().map(|p| &p.q_params)
}
}
#[inline]
#[cfg(feature = "alloc")]
pub fn check_public(public_key: &impl PublicKeyParts<BoxedUint>) -> Result<()> {
check_public_with_max_size(public_key.n().as_ref(), public_key.e(), None)
}
#[inline]
#[cfg(feature = "alloc")]
fn check_public_with_max_size(n: &BoxedUint, e: &BoxedUint, max_size: Option<usize>) -> Result<()> {
if let Some(max_size) = max_size {
if n.bits_vartime() as usize > max_size {
return Err(Error::ModulusTooLarge);
}
}
check_public_skip_exponent_size(n, e)?;
if e < &BoxedUint::from(RsaPublicKey::MIN_PUB_EXPONENT) {
return Err(Error::PublicExponentTooSmall);
}
if e > &BoxedUint::from(RsaPublicKey::MAX_PUB_EXPONENT) {
return Err(Error::PublicExponentTooLarge);
}
Ok(())
}
#[inline]
#[cfg(feature = "alloc")]
fn check_public_skip_exponent_size(n: &BoxedUint, e: &BoxedUint) -> Result<()> {
if e >= n || n.is_even().into() || n.is_zero().into() {
return Err(Error::InvalidModulus);
}
if e.is_even().into() {
return Err(Error::InvalidExponent);
}
Ok(())
}
#[cfg(feature = "private-key")]
fn validate_private_key_parts(key: &RsaPrivateKey) -> Result<()> {
let mut m = BoxedUint::one_with_precision(key.pubkey_components.n.bits_precision());
let one = BoxedUint::one();
for prime in &key.primes {
if prime <= &one {
return Err(Error::InvalidPrime);
}
m = m.wrapping_mul(prime);
}
if m != *key.pubkey_components.n.as_ref() {
return Err(Error::InvalidModulus);
}
let de = key.d.concatenating_mul(&key.pubkey_components.e);
for prime in &key.primes {
let x = CryptoNonZero::new(prime.wrapping_sub(BoxedUint::one())).unwrap();
let congruence = de.rem_vartime(&x);
if !bool::from(congruence.is_one()) {
return Err(Error::InvalidExponent);
}
}
Ok(())
}
#[cfg(all(feature = "hazmat", feature = "private-key"))]
fn validate_skip_exponent_size(key: &RsaPrivateKey) -> Result<()> {
check_public_skip_exponent_size(key.pubkey_components.n.as_ref(), &key.pubkey_components.e)?;
validate_private_key_parts(key)?;
Ok(())
}
#[cfg(feature = "serde")]
impl Serialize for RsaPublicKey {
fn serialize<S>(&self, serializer: S) -> core::prelude::v1::Result<S::Ok, S::Error>
where
S: serdect::serde::Serializer,
{
let der = self.to_public_key_der().map_err(ser::Error::custom)?;
serdect::slice::serialize_hex_lower_or_bin(&der, serializer)
}
}
#[cfg(feature = "serde")]
impl<'de> Deserialize<'de> for RsaPublicKey {
fn deserialize<D>(deserializer: D) -> core::prelude::v1::Result<Self, D::Error>
where
D: serdect::serde::Deserializer<'de>,
{
let der_bytes = serdect::slice::deserialize_hex_or_bin_vec(deserializer)?;
Self::from_public_key_der(&der_bytes).map_err(de::Error::custom)
}
}
#[cfg(feature = "serde")]
impl Serialize for RsaPrivateKey {
fn serialize<S>(&self, serializer: S) -> core::prelude::v1::Result<S::Ok, S::Error>
where
S: ser::Serializer,
{
let der = self.to_pkcs8_der().map_err(ser::Error::custom)?;
serdect::slice::serialize_hex_lower_or_bin(&der.as_bytes(), serializer)
}
}
#[cfg(feature = "serde")]
impl<'de> Deserialize<'de> for RsaPrivateKey {
fn deserialize<D>(deserializer: D) -> core::prelude::v1::Result<Self, D::Error>
where
D: de::Deserializer<'de>,
{
let der_bytes = serdect::slice::deserialize_hex_or_bin_vec(deserializer)?;
Self::from_pkcs8_der(&der_bytes).map_err(de::Error::custom)
}
}
#[cfg(test)]
#[cfg(all(feature = "alloc", feature = "private-key"))]
mod tests {
use super::*;
use crate::algorithms::rsa::{rsa_decrypt_and_check, rsa_encrypt};
use crate::traits::{PrivateKeyParts, PublicKeyParts};
use hex_literal::hex;
use rand::rngs::ChaCha8Rng;
use rand_core::SeedableRng;
#[cfg(feature = "encoding")]
use pkcs8::DecodePrivateKey;
#[test]
fn test_from_into() {
let raw_n = BoxedUint::from(101u64);
let n_odd = CryptoOdd::new(raw_n.clone()).unwrap();
let private_key = RsaPrivateKey {
pubkey_components: RsaPublicKey {
n: NonZero::new(raw_n.clone()).unwrap(),
e: BoxedUint::from(200u64),
n_params: BoxedMontyParams::new(n_odd),
},
d: BoxedUint::from(123u64),
primes: vec![],
precomputed: None,
};
let public_key: RsaPublicKey = private_key.into();
let n_limbs: &[u64] = PublicKeyParts::n(&public_key).as_ref().as_ref();
assert_eq!(n_limbs, &[101u64]);
assert_eq!(PublicKeyParts::e(&public_key), &BoxedUint::from(200u64));
assert_eq!(PublicKeyParts::e_bytes(&public_key), [200].into());
assert_eq!(PublicKeyParts::n_bytes(&public_key), [101].into());
}
fn test_key_basics(private_key: &RsaPrivateKey) {
private_key.validate().expect("invalid private key");
assert!(
PrivateKeyParts::d(private_key) < PublicKeyParts::n(private_key).as_ref(),
"private exponent too large"
);
let pub_key: RsaPublicKey = private_key.clone().into();
let m = BoxedUint::from(42u64);
let c = rsa_encrypt(&pub_key, &m).expect("encryption successful");
let m2 = rsa_decrypt_and_check::<ChaCha8Rng>(private_key, None, &c)
.expect("unable to decrypt without blinding");
assert_eq!(m, m2);
let mut rng = ChaCha8Rng::from_seed([42; 32]);
let m3 = rsa_decrypt_and_check(private_key, Some(&mut rng), &c)
.expect("unable to decrypt with blinding");
assert_eq!(m, m3);
}
macro_rules! key_generation {
($name:ident, $multi:expr, $size:expr) => {
#[cfg(feature = "private-key")]
#[test]
fn $name() {
let mut rng = ChaCha8Rng::from_seed([42; 32]);
let exp = BoxedUint::from(RsaPrivateKey::EXP);
for _ in 0..10 {
let components =
generate_multi_prime_key_with_exp(&mut rng, $multi, $size, exp.clone())
.unwrap();
let private_key = RsaPrivateKey::from_components(
components.n.get(),
components.e,
components.d,
components.primes,
)
.unwrap();
assert_eq!(PublicKeyParts::n(&private_key).bits(), $size);
test_key_basics(&private_key);
}
}
#[cfg(not(feature = "private-key"))]
#[test]
fn $name() {
todo!("generate_multi_prime_key_with_exp is not implemented yet");
}
};
}
key_generation!(key_generation_128, 2, 128);
key_generation!(key_generation_1024, 2, 1024);
key_generation!(key_generation_multi_3_256, 3, 256);
key_generation!(key_generation_multi_4_64, 4, 64);
key_generation!(key_generation_multi_5_64, 5, 64);
key_generation!(key_generation_multi_8_576, 8, 576);
key_generation!(key_generation_multi_16_1024, 16, 1024);
#[test]
fn test_negative_decryption_value() {
let bits = 128;
let private_key = RsaPrivateKey::from_components(
BoxedUint::from_le_slice(
&[
99, 192, 208, 179, 0, 220, 7, 29, 49, 151, 75, 107, 75, 73, 200, 180,
],
bits,
)
.unwrap(),
BoxedUint::from_le_slice(&[1, 0, 1, 0, 0, 0, 0, 0], 64).unwrap(),
BoxedUint::from_le_slice(
&[
81, 163, 254, 144, 171, 159, 144, 42, 244, 133, 51, 249, 28, 12, 63, 65,
],
bits,
)
.unwrap(),
vec![
BoxedUint::from_le_slice(&[105, 101, 60, 173, 19, 153, 3, 192], bits / 2).unwrap(),
BoxedUint::from_le_slice(&[235, 65, 160, 134, 32, 136, 6, 241], bits / 2).unwrap(),
],
)
.unwrap();
for _ in 0..1000 {
test_key_basics(&private_key);
}
}
#[test]
#[cfg(all(feature = "hazmat", feature = "serde"))]
fn test_serde() {
use rand::rngs::ChaCha8Rng;
use rand_core::SeedableRng;
use serde_test::{assert_tokens, Configure, Token};
let mut rng = ChaCha8Rng::from_seed([42; 32]);
let priv_key = RsaPrivateKey::new_unchecked(&mut rng, 64).expect("failed to generate key");
let priv_tokens = [Token::Str(concat!(
"3056020100300d06092a864886f70d010101050004423040020100020900a",
"b240c3361d02e370203010001020811e54a15259d22f9020500ceff5cf302",
"0500d3a7aaad020500ccaddf17020500cb529d3d020500bb526d6f"
))];
assert_tokens(&priv_key.clone().readable(), &priv_tokens);
let priv_tokens = [Token::Str(
"3024300d06092a864886f70d01010105000313003010020900ab240c3361d02e370203010001",
)];
assert_tokens(
&RsaPublicKey::from(priv_key.clone()).readable(),
&priv_tokens,
);
}
#[test]
fn invalid_coeff_private_key_regression() {
use base64ct::{Base64, Encoding};
let n = Base64::decode_vec(
"wC8GyQvTCZOK+iiBR5fGQCmzRCTWX9TQ3aRG5gGFk0wB6EFoLMAyEEqeG3gS8xhA\
m2rSWYx9kKufvNat3iWlbSRVqkcbpVAYlj2vTrpqDpJl+6u+zxFYoUEBevlJJkAh\
l8EuCccOA30fVpcfRvXPTtvRd3yFT9E9EwZljtgSI02w7gZwg7VIxaGeajh5Euz6\
ZVQZ+qNRKgXrRC7gPRqVyI6Dt0Jc+Su5KBGNn0QcPDzOahWha1ieaeMkFisZ9mdp\
sJoZ4tw5eicLaUomKzALHXQVt+/rcZSrCd6/7uUo11B/CYBM4UfSpwXaL88J9AE6\
A5++no9hmJzaF2LLp+Qwx4yY3j9TDutxSAjsraxxJOGZ3XyA9nG++Ybt3cxZ5fP7\
ROjxCfROBmVv5dYn0O9OBIqYeCH6QraNpZMadlLNIhyMv8Y+P3r5l/PaK4VJaEi5\
pPosnEPawp0W0yZDzmjk2z1LthaRx0aZVrAjlH0Rb/6goLUQ9qu1xsDtQVVpN4A8\
9ZUmtTWORnnJr0+595eHHxssd2gpzqf4bPjNITdAEuOCCtpvyi4ls23zwuzryUYj\
cUOEnsXNQ+DrZpLKxdtsD/qNV/j1hfeyBoPllC3cV+6bcGOFcVGbjYqb+Kw1b0+j\
L69RSKQqgmS+qYqr8c48nDRxyq3QXhR8qtzUwBFSLVk=",
)
.unwrap();
let e = Base64::decode_vec("AQAB").unwrap();
let d = Base64::decode_vec(
"qQazSQ+FRN7nVK1bRsROMRB8AmsDwLVEHivlz1V3Td2Dr+oW3YUMgxedhztML1Id\
QJPq/ad6qErJ6yRFNySVIjDaxzBTOEoB1eHa1btOnBJWb8rVvvjaorixvJ6Tn3i4\
EuhsvVy9DoR1k4rGj3qSIiFjUVvLRDAbLyhpGgEfsr0Z577yJmTC5E8JLRMOKX8T\
mxsk3jPVpsgd65Hu1s8S/ZmabwuHCf9SkdMeY/1bd/9i7BqqJeeDLE4B5x1xcC3z\
3scqDUTzqGO+vZPhjgprPDRlBamVwgenhr7KwCn8iaLamFinRVwOAag8BeBqOJj7\
lURiOsKQa9FIX1kdFUS1QMQxgtPycLjkbvCJjriqT7zWKsmJ7l8YLs6Wmm9/+QJR\
wNCEVdMTXKfCP1cJjudaiskEQThfUldtgu8gUDNYbQ/Filb2eKfiX4h1TiMxZqUZ\
HVZyb9nShbQoXJ3vj/MGVF0QM8TxhXM8r2Lv9gDYU5t9nQlUMLhs0jVjai48jHAB\
bFNyH3sEcOmJOIwJrCXw1dzG7AotwyaEVUHOmL04TffmwCFfnyrLjbFgnyOeoyII\
BYjcY7QFRm/9nupXMTH5hZ2qrHfCJIp0KK4tNBdQqmnHapFl5l6Le1s4qBS5bEIz\
jitobLvAFm9abPlDGfxmY6mlrMK4+nytwF9Ct7wc1AE=",
)
.unwrap();
let primes = [
Base64::decode_vec(
"9kQWEAzsbzOcdPa+s5wFfw4XDd7bB1q9foZ31b1+TNjGNxbSBCFlDF1q98vwpV6n\
M8bWDh/wtbNoETSQDgpEnYOQ26LWEw6YY1+q1Q2GGEFceYUf+Myk8/vTc8TN6Zw0\
bKZBWy10Qo8h7xk4JpzuI7NcxvjJYTkS9aErFxi3vVH0aiZC0tmfaCqr8a2rJxyV\
wqreRpOjwAWrotMsf2wGsF4ofx5ScoFy5GB5fJkkdOrW1LyTvZAUCX3cstPr19+T\
NC5zZOk7WzZatnCkN5H5WzalWtZuu0oVL205KPOa3R8V2yv5e6fm0v5fTmqSuvjm\
aMJLXCN4QJkmIzojO99ckQ==",
)
.unwrap(),
Base64::decode_vec(
"x8exdMjVA2CiI+Thx7loHtVcevoeE2sZ7btRVAvmBqo+lkHwxb7FHRnWvuj6eJSl\
D2f0T50EewIhhiW3R9BmktCk7hXjbSCnC1u9Oxc1IAUm/7azRqyfCMx43XhLxpD+\
xkBCpWkKDLxGczsRwTuaP3lKS3bSdBrNlGmdblubvVBIq4YZ2vXVlnYtza0cS+dg\
CK7BGTqUsrCUd/ZbIvwcwZkZtpkhj1KQfto9X/0OMurBzAqbkeq1cyRHXHkOfN/q\
bUIIRqr9Ii7Eswf9Vk8xp2O1Nt8nzcYS9PFD12M5eyaeFEkEYfpNMNGuTzp/31oq\
VjbpoCxS6vuWAZyADxhISQ==",
)
.unwrap(),
Base64::decode_vec(
"is7d0LY4HoXszlC2NO7gejkq7XqL4p1W6hZJPYTNx+r37t1CC2n3Vvzg6kNdpRix\
DhIpXVTLjN9O7UO/XuqSumYKJIKoP52eb4Tg+a3hw5Iz2Zsb5lUTNSLgkQSBPAf7\
1LHxbL82JL4g1nBUog8ae60BwnVArThKY4EwlJguGNw09BAU4lwf6csDl/nX2vfV\
wiAloYpeZkHL+L8m+bueGZM5KE2jEz+7ztZCI+T+E5i69rZEYDjx0lfLKlEhQlCW\
3HbCPELqXgNJJkRfi6MP9kXa9lSfnZmoT081RMvqonB/FUa4HOcKyCrw9XZEtnbN\
CIdbitfDVEX+pSSD7596wQ==",
)
.unwrap(),
Base64::decode_vec(
"GPs0injugfycacaeIP5jMa/WX55VEnKLDHom4k6WlfDF4L4gIGoJdekcPEUfxOI5\
faKvHyFwRP1wObkPoRBDM0qZxRfBl4zEtpvjHrd5MibSyJkM8+J0BIKk/nSjbRIG\
eb3hV5O56PvGB3S0dKhCUnuVObiC+ne7izplsD4OTG70l1Yud33UFntyoMxrxGYL\
USqhBMmZfHquJg4NOWOzKNY/K+EcHDLj1Kjvkcgv9Vf7ocsVxvpFdD9uGPceQ6kw\
RDdEl6mb+6FDgWuXVyqR9+904oanEIkbJ7vfkthagLbEf57dyG6nJlqh5FBZWxGI\
R72YGypPuAh7qnnqXXjY2Q==",
)
.unwrap(),
Base64::decode_vec(
"CUWC+hRWOT421kwRllgVjy6FYv6jQUcgDNHeAiYZnf5HjS9iK2ki7v8G5dL/0f+Y\
f+NhE/4q8w4m8go51hACrVpP1p8GJDjiT09+RsOzITsHwl+ceEKoe56ZW6iDHBLl\
rNw5/MtcYhKpjNU9KJ2udm5J/c9iislcjgckrZG2IB8ADgXHMEByZ5DgaMl4AKZ1\
Gx8/q6KftTvmOT5rNTMLi76VN5KWQcDWK/DqXiOiZHM7Nr4dX4me3XeRgABJyNR8\
Fqxj3N1+HrYLe/zs7LOaK0++F9Ul3tLelhrhsvLxei3oCZkF9A/foD3on3luYA+1\
cRcxWpSY3h2J4/22+yo4+Q==",
)
.unwrap(),
];
let e = BoxedUint::from_be_slice(&e, 64).unwrap();
let bits = 4096;
let n = BoxedUint::from_be_slice(&n, bits).unwrap();
let d = BoxedUint::from_be_slice(&d, bits).unwrap();
let primes = primes
.iter()
.map(|p| BoxedUint::from_be_slice(p, bits / 2).unwrap())
.collect();
let res = RsaPrivateKey::from_components(n, e, d, primes);
assert_eq!(res, Err(Error::InvalidModulus));
}
#[test]
fn reject_oversized_private_key() {
let n = BoxedUint::from_be_slice(
&hex!(
"a909fa3bba5df62a102443b04b6b219c3d9b337f8f68ba2fae8f8e28e13db7b8
f1ae9f9bf57abcd68793eb3379bb9c8f7b06be574f19b0025011b17d0f900c51
990765cfec5bf5c022d0bc0a88d0847a0241f8ce5d9a4e3661ac4319d224d830
2d5e530d52940296d020fcc89bdc83c0ebe5d725c8ca969120d1f3b61cf7eb2b
0799a1e5355659438d011d590e89477ec2ced6f54f6f1cdd1071a4870bc94750
4a9bfc8c9bd054a666cfaa6e6e0d5f2688f74a25a0015fb4e9edf1515b1df4ff
4d74b45703327ac73116cbaf72e4eb5f1f0461cf53b894e15b6db004014e2ba4
a97ed82d67a8b8589aa4d05f09ad3c19dd3ca1ccab0f4707bc29e366b66a98a1
a1a23e3f6550e73ddbc4333b24c44d577ca55ed4a6768e7bd4385872f44269d7
9fad690aee2aa682ee2bc8c1f883bbc627d16478f73cc7ec8c9bab678305e7b2
e0a52ac8eb47a460d875a3a5625ce0c742ecc6c734da1779ffec62130c14170d
14e51ae407efb87293116a3dc951188d30c4408f29ab31a738e189a4ee6bd7a2
a4a79d5c1fa8e57172a2c9c3b7f5f8620ab18ffab6ff5f0b3ee42014ba96473b
077784dd550ca2abdc429b32281948c03f3d50f1c8c85d1a549e9dab6508c448
22de4e37b0ac635b8d2303521745e4cd8e0519a7ed0f171c7094c53f5f63f059
7b275e139e0abc47449289b191b5e884728499b59e11d869efc874e3fc56dfae
29d4dbbd638f70ee0b4efbd9e6c0a729349ab7d8ce906f4776750df0274a0823
d9a5fa7fc34e76be208c306c8f14385108e0a650cd96aa98dbf74f31a6a67a1c
1caede33212d014790654206acf9ab1501262999fdae482bff61f0f13db4abb7
b4944eb922cebdbd33f6c179b5ec1e7328562c91ae358c9a19ea32296cc3adb9
53d95774a8198337f59e00c7725a1991342ff750ea524985d154002148417c0d
5aacddeecae7bdbca6e212eaa0fb8c863914b5372fe98383124ecb07614b0dcc
a6987ff308bdf0e154a2e3e1f2659d7780cec2cfc5a6917e07fa378bd3e25a43
21b50b3a95fc703bf2f52926ffdb6ff275f97d62ff51e5b4b7938ed798e2b061
553d5cc41a1504641a0b08a7740b3e163729fbc7002beff7a4c42086792b187c
ebc7d6086283865ece2f4083069e061a4daeda5c069fb787d68e0e047928ddad
9c2aaab3cdb94704ba77853fd44f7df8fc7f7be486a0858285417d533ba27b6e
d9dda746efe18409204f5dd473af0974cda2981bc7028e830d4b7a9ac58200c3
8011762e3bb680fa7dde7adb6706ab440a8400a0225d08709e6e32299683015c
c2d7cac786b4b2bd6ab238f210ef7314a57fbff5964041e8b80d426409efddb9
4c38a989840867addfe1de59d890a95b4bd4930ceb1f888f6f6872f9305048ec
3592f8cbc86fe51d68f30d06bad31c555bdbb9f43c50b9e1ec34430750e443d6
8157bc52defc0965"
),
8256,
)
.unwrap();
let e = BoxedUint::from(65_537u64);
assert_eq!(
RsaPublicKey::new(n, e).err().unwrap(),
Error::ModulusTooLarge
);
}
#[test]
#[cfg(feature = "encoding")]
fn build_key_from_primes() {
const RSA_2048_PRIV_DER: &[u8] = include_bytes!("../tests/examples/pkcs8/rsa2048-priv.der");
let ref_key = RsaPrivateKey::from_pkcs8_der(RSA_2048_PRIV_DER).unwrap();
assert_eq!(ref_key.validate(), Ok(()));
let primes = PrivateKeyParts::primes(&ref_key).to_vec();
let exp = PublicKeyParts::e(&ref_key);
let key = RsaPrivateKey::from_primes(primes, exp.clone())
.expect("failed to import key from primes");
assert_eq!(key.validate(), Ok(()));
assert_eq!(PublicKeyParts::n(&key), PublicKeyParts::n(&ref_key));
assert_eq!(PrivateKeyParts::dp(&key), PrivateKeyParts::dp(&ref_key));
assert_eq!(PrivateKeyParts::dq(&key), PrivateKeyParts::dq(&ref_key));
assert_eq!(PrivateKeyParts::d(&key), PrivateKeyParts::d(&ref_key));
}
#[test]
#[cfg(feature = "encoding")]
fn build_key_from_p_q() {
const RSA_2048_SP800_PRIV_DER: &[u8] =
include_bytes!("../tests/examples/pkcs8/rsa2048-sp800-56b-priv.der");
let ref_key = RsaPrivateKey::from_pkcs8_der(RSA_2048_SP800_PRIV_DER).unwrap();
assert_eq!(ref_key.validate(), Ok(()));
let primes = PrivateKeyParts::primes(&ref_key).to_vec();
let exp = PublicKeyParts::e(&ref_key);
let key = RsaPrivateKey::from_p_q(primes[0].clone(), primes[1].clone(), exp.clone())
.expect("failed to import key from primes");
assert_eq!(key.validate(), Ok(()));
assert_eq!(PublicKeyParts::n(&key), PublicKeyParts::n(&ref_key));
assert_eq!(PrivateKeyParts::dp(&key), PrivateKeyParts::dp(&ref_key));
assert_eq!(PrivateKeyParts::dq(&key), PrivateKeyParts::dq(&ref_key));
assert_eq!(PrivateKeyParts::d(&key), PrivateKeyParts::d(&ref_key));
}
#[test]
#[cfg(feature = "hazmat")]
fn test_from_components_with_large_exponent() {
use rand::rngs::ChaCha8Rng;
use rand_core::SeedableRng;
let mut rng = ChaCha8Rng::from_seed([42; 32]);
let large_e = BoxedUint::from((1u64 << 34) + 1);
let components =
generate_multi_prime_key_with_exp(&mut rng, 2, 1024, large_e.clone()).unwrap();
let n = components.n.get().clone();
let d = components.d;
let primes = components.primes;
let result =
RsaPrivateKey::from_components(n.clone(), large_e.clone(), d.clone(), primes.clone());
assert!(result.is_err());
assert_eq!(result.unwrap_err(), Error::PublicExponentTooLarge);
let key_with_large_exp = RsaPrivateKey::from_components_with_large_exponent(
n.clone(),
large_e.clone(),
d.clone(),
primes.clone(),
);
assert!(key_with_large_exp.is_ok());
let key_with_large_exp = key_with_large_exp.unwrap();
assert_eq!(PublicKeyParts::e(&key_with_large_exp), &large_e);
assert_eq!(PublicKeyParts::n(&key_with_large_exp).as_ref(), &n);
assert_eq!(PrivateKeyParts::d(&key_with_large_exp), &d);
assert!(validate_skip_exponent_size(&key_with_large_exp).is_ok());
}
#[test]
#[cfg(feature = "hazmat")]
fn test_from_components_with_small_exponent() {
use rand::rngs::ChaCha8Rng;
use rand_core::SeedableRng;
let mut rng = ChaCha8Rng::from_seed([43; 32]);
let small_e = BoxedUint::from(1u64);
let components =
generate_multi_prime_key_with_exp(&mut rng, 2, 1024, small_e.clone()).unwrap();
let n = components.n.get().clone();
let d = components.d;
let primes = components.primes;
let result =
RsaPrivateKey::from_components(n.clone(), small_e.clone(), d.clone(), primes.clone());
assert!(result.is_err());
let key_with_small_exp = RsaPrivateKey::from_components_with_large_exponent(
n.clone(),
small_e.clone(),
d.clone(),
primes,
);
assert!(key_with_small_exp.is_ok());
let key_with_small_exp = key_with_small_exp.unwrap();
assert_eq!(PublicKeyParts::e(&key_with_small_exp), &small_e);
assert!(validate_skip_exponent_size(&key_with_small_exp).is_ok());
}
#[test]
#[cfg(feature = "hazmat")]
fn test_key_invalid_primes() {
let e = RsaPrivateKey::from_components_with_large_exponent(
BoxedUint::from(239u64),
BoxedUint::from(185u64),
BoxedUint::from(0u64),
vec![BoxedUint::from(1u64), BoxedUint::from(239u64)],
)
.unwrap_err();
assert_eq!(e, Error::InvalidPrime);
}
}