better rng function definition

fortanix · Apr 15, 2024 · a2da09c · a2da09c
1 parent 64cfa72
commit a2da09c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/rustls-mbedcrypto-provider/src/lib.rs b/rustls-mbedcrypto-provider/src/lib.rs
@@ -118,9 +118,6 @@ pub mod rng {
     #[cfg(not(any(target_env = "sgx", feature = "rdrand")))]
     use mbedtls::rng::{CtrDrbg, OsEntropy};
 
-    #[cfg(any(target_env = "sgx", feature = "rdrand"))]
-    use mbedtls::rng::Rdrand;
-
     /// Type alias for the RNG supported by *mbedtls*
     #[cfg(not(any(target_env = "sgx", feature = "rdrand")))]
     pub type MbedRng = CtrDrbg;
@@ -132,13 +129,16 @@ pub mod rng {
         CtrDrbg::new(entropy, None).ok()
     }
 
+    #[cfg(any(target_env = "sgx", feature = "rdrand"))]
+    use mbedtls::rng::Rdrand;
+
     /// Type alias for the RNG supported by *mbedtls*
     #[cfg(any(target_env = "sgx", feature = "rdrand"))]
     pub type MbedRng = Rdrand;
 
     /// Get a RNG supported by *mbedtls*
     #[cfg(any(target_env = "sgx", feature = "rdrand"))]
-    pub fn rng_new() -> Option<Rdrand> {
+    pub fn rng_new() -> Option<MbedRng> {
         Some(Rdrand)
     }
 }