feat: allow generating random SigningKey's

Allow generating random SigningKey structs directly. Signed-off-by: Dmitry Baryshkov <dmitry.baryshkov@linaro.org>
RustCrypto · Nov 7, 2022 · a972877 · a972877
1 parent 5f7ef14
commit a972877
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 1 deletion.
diff --git a/src/pkcs1v15.rs b/src/pkcs1v15.rs
@@ -328,6 +328,15 @@ where
             phantom: Default::default(),
         }
     }
+
+    /// Generate a new signing key.
+    pub fn random<R: CryptoRngCore>(rng: &mut R, bit_size: usize) -> Result<Self> {
+        Ok(Self {
+            inner: RsaPrivateKey::new(rng, bit_size)?,
+            prefix: Vec::new(),
+            phantom: Default::default(),
+        })
+    }
 }
 
 impl<D> From<RsaPrivateKey> for SigningKey<D>
@@ -352,14 +361,23 @@ impl<D> SigningKey<D>
 where
     D: Digest + AssociatedOid,
 {
-    /// Create a new verifying key with a prefix for the digest `D`.
+    /// Create a new signing key with a prefix for the digest `D`.
     pub fn new_with_prefix(key: RsaPrivateKey) -> Self {
         Self {
             inner: key,
             prefix: generate_prefix::<D>(),
             phantom: Default::default(),
         }
     }
+
+    /// Generate a new signing key with a prefix for the digest `D`.
+    pub fn random_with_prefix<R: CryptoRngCore>(rng: &mut R, bit_size: usize) -> Result<Self> {
+        Ok(Self {
+            inner: RsaPrivateKey::new(rng, bit_size)?,
+            prefix: generate_prefix::<D>(),
+            phantom: Default::default(),
+        })
+    }
 }
 
 impl<D> AsRef<RsaPrivateKey> for SigningKey<D>

diff --git a/src/pss.rs b/src/pss.rs
@@ -553,6 +553,28 @@ where
             phantom: Default::default(),
         }
     }
+
+    /// Generate a new random RSASSA-PSS signing key.
+    pub fn random<R: CryptoRngCore>(rng: &mut R, bit_size: usize) -> Result<Self> {
+        Ok(Self {
+            inner: RsaPrivateKey::new(rng, bit_size)?,
+            salt_len: None,
+            phantom: Default::default(),
+        })
+    }
+
+    /// Generate a new random RSASSA-PSS signing key with a salt of the given length.
+    pub fn random_with_salt_len<R: CryptoRngCore>(
+        rng: &mut R,
+        bit_size: usize,
+        salt_len: usize,
+    ) -> Result<Self> {
+        Ok(Self {
+            inner: RsaPrivateKey::new(rng, bit_size)?,
+            salt_len: Some(salt_len),
+            phantom: Default::default(),
+        })
+    }
 }
 
 impl<D> From<RsaPrivateKey> for SigningKey<D>