Refactor traits

src/algorithms/rsa.rs

@@ -8,7 +8,7 @@ use rand_core::CryptoRngCore;
 use zeroize::Zeroize;
 
 use crate::errors::{Error, Result};
-use crate::keytraits::{PrivateKeyParts, PublicKeyParts};
+use crate::traits::{PrivateKeyParts, PublicKeyParts};
 
 /// Raw RSA encryption of m with the public key. No padding is performed.
 #[inline]

src/encoding.rs

@@ -4,7 +4,7 @@
 //! `pkcs1` crate's traits for types which impl the `pkcs8` crate's traits.
 
 use crate::{
-    keytraits::{PrivateKeyParts, PublicKeyParts},
+    traits::{PrivateKeyParts, PublicKeyParts},
     BigUint, RsaPrivateKey, RsaPublicKey,
 };
 use core::convert::{TryFrom, TryInto};

src/key.rs

@@ -16,9 +16,8 @@ use zeroize::{Zeroize, ZeroizeOnDrop};
 use crate::algorithms::generate::generate_multi_prime_key_with_exp;
 use crate::dummy_rng::DummyRng;
 use crate::errors::{Error, Result};
-use crate::keytraits::{CrtValue, PrivateKeyParts, PublicKeyParts};
-
-use crate::padding::{PaddingScheme, SignatureScheme};
+use crate::traits::{PaddingScheme, PrivateKeyParts, PublicKeyParts, SignatureScheme};
+use crate::CrtValue;
 
 /// Represents the public part of an RSA key.
 #[derive(Debug, Clone, Hash, PartialEq, Eq)]

src/lib.rs

@@ -224,7 +224,6 @@ pub use signature;
 
 mod algorithms;
 pub mod errors;
-mod keytraits;
 pub mod oaep;
 pub mod pkcs1v15;
 pub mod pss;
@@ -233,7 +232,6 @@ pub mod traits;
 mod dummy_rng;
 mod encoding;
 mod key;
-mod padding;
 
 pub use pkcs1;
 pub use pkcs8;
@@ -242,9 +240,8 @@ pub use sha2;
 
 pub use crate::{
     key::{RsaPrivateKey, RsaPublicKey},
-    keytraits::{CrtValue, PrivateKeyParts, PublicKeyParts},
     oaep::Oaep,
-    padding::{PaddingScheme, SignatureScheme},
     pkcs1v15::{Pkcs1v15Encrypt, Pkcs1v15Sign},
     pss::Pss,
+    traits::keys::CrtValue,
 };

src/oaep.rs

@@ -21,9 +21,9 @@ use crate::algorithms::rsa::{rsa_decrypt_and_check, rsa_encrypt};
 use crate::dummy_rng::DummyRng;
 use crate::errors::{Error, Result};
 use crate::key::{self, RsaPrivateKey, RsaPublicKey};
-use crate::keytraits::PublicKeyParts;
-use crate::padding::PaddingScheme;
+use crate::traits::PaddingScheme;
 use crate::traits::{Decryptor, RandomizedDecryptor, RandomizedEncryptor};
+use crate::traits::PublicKeyParts;
 
 /// Encryption and Decryption using [OAEP padding](https://datatracker.ietf.org/doc/html/rfc8017#section-7.1).
 ///
@@ -422,9 +422,9 @@ where
 #[cfg(test)]
 mod tests {
     use crate::key::{RsaPrivateKey, RsaPublicKey};
-    use crate::keytraits::PublicKeyParts;
     use crate::oaep::{DecryptingKey, EncryptingKey, Oaep};
     use crate::traits::{Decryptor, RandomizedDecryptor, RandomizedEncryptor};
+    use crate::traits::PublicKeyParts;
 
     use alloc::string::String;
     use digest::{Digest, DynDigest, FixedOutputReset};

src/pkcs1v15.rs

@@ -32,8 +32,8 @@ use crate::algorithms::rsa::{rsa_decrypt_and_check, rsa_encrypt};
 use crate::dummy_rng::DummyRng;
 use crate::errors::{Error, Result};
 use crate::key;
-use crate::keytraits::PublicKeyParts;
-use crate::padding::{PaddingScheme, SignatureScheme};
+use crate::traits::{PaddingScheme, SignatureScheme};
+use crate::traits::PublicKeyParts;
 use crate::traits::{Decryptor, EncryptingKeypair, RandomizedDecryptor, RandomizedEncryptor};
 use crate::{RsaPrivateKey, RsaPublicKey};
 
@@ -792,7 +792,7 @@ mod tests {
     use sha3::Sha3_256;
     use signature::{RandomizedSigner, Signer, Verifier};
 
-    use crate::{PublicKeyParts, RsaPrivateKey, RsaPublicKey};
+    use crate::{traits::PublicKeyParts, RsaPrivateKey, RsaPublicKey};
 
     fn get_private_key() -> RsaPrivateKey {
         // In order to generate new test vectors you'll need the PEM form of this key:

src/pss.rs

@@ -36,8 +36,8 @@ use crate::algorithms::pad::{uint_to_be_pad, uint_to_zeroizing_be_pad};
 use crate::algorithms::pss::*;
 use crate::algorithms::rsa::{rsa_decrypt_and_check, rsa_encrypt};
 use crate::errors::{Error, Result};
-use crate::keytraits::PublicKeyParts;
-use crate::padding::SignatureScheme;
+use crate::traits::SignatureScheme;
+use crate::traits::PublicKeyParts;
 use crate::{RsaPrivateKey, RsaPublicKey};
 
 #[cfg(feature = "getrandom")]

src/traits.rs

@@ -1,42 +1,9 @@
-//! Generic traits for message encryption and decryption
+//! RSA-related trait definitions.
 
-use alloc::vec::Vec;
-use rand_core::CryptoRngCore;
+mod encryption;
+pub(crate) mod keys;
+mod padding;
 
-use crate::errors::Result;
-
-/// Encrypt the message using provided random source
-pub trait RandomizedEncryptor {
-    /// Encrypt the given message.
-    fn encrypt_with_rng<R: CryptoRngCore + ?Sized>(
-        &self,
-        rng: &mut R,
-        msg: &[u8],
-    ) -> Result<Vec<u8>>;
-}
-
-/// Decrypt the given message
-pub trait Decryptor {
-    /// Decrypt the given message.
-    fn decrypt(&self, ciphertext: &[u8]) -> Result<Vec<u8>>;
-}
-
-/// Decrypt the given message using provided random source
-pub trait RandomizedDecryptor {
-    /// Decrypt the given message.
-    fn decrypt_with_rng<R: CryptoRngCore + ?Sized>(
-        &self,
-        rng: &mut R,
-        ciphertext: &[u8],
-    ) -> Result<Vec<u8>>;
-}
-
-/// Encryption keypair with an associated encryption key.
-pub trait EncryptingKeypair {
-    /// Encrypting key type for this keypair.
-    type EncryptingKey: Clone;
-
-    /// Get the encrypting key which can encrypt messages to be decrypted by
-    /// the decryption key portion of this keypair.
-    fn encrypting_key(&self) -> Self::EncryptingKey;
-}
+pub use encryption::{Decryptor, EncryptingKeypair, RandomizedDecryptor, RandomizedEncryptor};
+pub use keys::{PrivateKeyParts, PublicKeyParts};
+pub use padding::{PaddingScheme, SignatureScheme};

src/traits/encryption.rs

@@ -0,0 +1,42 @@
+//! Encryption-related traits.
+
+use alloc::vec::Vec;
+use rand_core::CryptoRngCore;
+
+use crate::errors::Result;
+
+/// Encrypt the message using provided random source
+pub trait RandomizedEncryptor {
+    /// Encrypt the given message.
+    fn encrypt_with_rng<R: CryptoRngCore + ?Sized>(
+        &self,
+        rng: &mut R,
+        msg: &[u8],
+    ) -> Result<Vec<u8>>;
+}
+
+/// Decrypt the given message
+pub trait Decryptor {
+    /// Decrypt the given message.
+    fn decrypt(&self, ciphertext: &[u8]) -> Result<Vec<u8>>;
+}
+
+/// Decrypt the given message using provided random source
+pub trait RandomizedDecryptor {
+    /// Decrypt the given message.
+    fn decrypt_with_rng<R: CryptoRngCore + ?Sized>(
+        &self,
+        rng: &mut R,
+        ciphertext: &[u8],
+    ) -> Result<Vec<u8>>;
+}
+
+/// Encryption keypair with an associated encryption key.
+pub trait EncryptingKeypair {
+    /// Encrypting key type for this keypair.
+    type EncryptingKey: Clone;
+
+    /// Get the encrypting key which can encrypt messages to be decrypted by
+    /// the decryption key portion of this keypair.
+    fn encrypting_key(&self) -> Self::EncryptingKey;
+}

tests/pkcs1.rs

@@ -3,7 +3,8 @@
 use hex_literal::hex;
 use rsa::{
     pkcs1::{DecodeRsaPrivateKey, DecodeRsaPublicKey, EncodeRsaPrivateKey, EncodeRsaPublicKey},
-    PrivateKeyParts, PublicKeyParts, RsaPrivateKey, RsaPublicKey,
+    traits::{PrivateKeyParts, PublicKeyParts},
+    RsaPrivateKey, RsaPublicKey,
 };
 
 #[cfg(feature = "pem")]

tests/pkcs8.rs

@@ -17,7 +17,8 @@ const RSA_2048_PUB_PEM: &str = include_str!("examples/pkcs8/rsa2048-pub.pem");
 use hex_literal::hex;
 use rsa::{
     pkcs8::{DecodePrivateKey, DecodePublicKey, EncodePrivateKey, EncodePublicKey},
-    PrivateKeyParts, PublicKeyParts, RsaPrivateKey, RsaPublicKey,
+    traits::{PrivateKeyParts, PublicKeyParts},
+    RsaPrivateKey, RsaPublicKey,
 };
 
 #[cfg(feature = "pem")]