22
33use beam_lib:: { AppId , AppOrProxyId , ProxyId , FailureStrategy , WorkStatus } ;
44use chacha20poly1305:: {
5- aead:: { Aead , AeadCore , KeyInit , OsRng } ,
5+ aead:: { Aead , AeadCore , KeyInit } ,
66 XChaCha20Poly1305 , XNonce ,
77} ;
88use crypto_jwt:: extract_jwt;
@@ -21,7 +21,7 @@ use std::{
2121 time:: { Duration , Instant , SystemTime } , net:: SocketAddr , error:: Error ,
2222} ;
2323
24- use rand:: Rng ;
24+ use rand:: { rng , rngs :: OsRng , Rng } ;
2525use serde:: {
2626 de:: { DeserializeOwned , Visitor } ,
2727 Deserialize , Serialize ,
@@ -302,9 +302,9 @@ pub trait EncryptableMsg: Msg + Serialize + Sized {
302302 receivers_public_keys : & Vec < RsaPublicKey > ,
303303 ) -> Result < Self :: Output , SamplyBeamError > {
304304 // Generate Symmetric Key and Nonce
305- let mut rng = rand :: thread_rng ( ) ;
306- let symmetric_key = XChaCha20Poly1305 :: generate_key ( & mut rng) ;
307- let nonce = XChaCha20Poly1305 :: generate_nonce ( & mut rng) ;
305+ let mut rng = rng ( ) ;
306+ let symmetric_key = XChaCha20Poly1305 :: generate_key_with_rng ( & mut rng) ;
307+ let nonce = XChaCha20Poly1305 :: generate_nonce_with_rng ( & mut rng) ;
308308
309309 // Encrypt symmetric key with receivers' public keys
310310 let Ok ( encrypted_keys) = receivers_public_keys
@@ -753,7 +753,7 @@ mod tests {
753753 } ;
754754
755755 //Setup Keypairs
756- let mut rng = rand:: thread_rng ( ) ;
756+ let mut rng = rand:: rng ( ) ;
757757 let rsa_length: usize = 2048 ;
758758 let p1_private = RsaPrivateKey :: new ( & mut rng, rsa_length)
759759 . expect ( "Failed to generate private key for proxy 1" ) ;
@@ -799,7 +799,7 @@ mod tests {
799799 } ;
800800
801801 //Setup Keypairs
802- let mut rng = rand:: thread_rng ( ) ;
802+ let mut rng = rand:: rng ( ) ;
803803 let rsa_length: usize = 2048 ;
804804 let p1_private = RsaPrivateKey :: new ( & mut rng, rsa_length)
805805 . expect ( "Failed to generate private key for proxy 1" ) ;
0 commit comments