improve random_u64

This commit is contained in:
Simon Gardling 2022-05-24 23:08:08 -04:00
parent 873e5b1ebf
commit 08f3040381
2 changed files with 8 additions and 6 deletions

View File

@ -226,8 +226,10 @@ impl FunctionManager {
/// Create and push new empty function entry /// Create and push new empty function entry
pub fn push_empty(&mut self) { pub fn push_empty(&mut self) {
self.functions self.functions.push((
.push((Id::new_from_u64(random_u64()), FunctionEntry::EMPTY)); Id::new_from_u64(random_u64().expect("unable to generate random id")),
FunctionEntry::EMPTY,
));
} }
/// Detect if any functions are using integrals /// Detect if any functions are using integrals

View File

@ -180,8 +180,8 @@ pub const fn hashed_storage_read(data: &str) -> Option<(HashBytes, &[u8])> {
)) ))
} }
pub fn random_u64() -> u64 { pub fn random_u64() -> Result<u64, getrandom::Error> {
let mut data: [u8; 8] = [0, 0, 0, 0, 0, 0, 0, 0]; let mut buf = [0u8; 8];
getrandom(&mut data).expect("unable to generate random number"); getrandom(&mut buf)?;
u64::from_be_bytes(data) Ok(u64::from_be_bytes(buf))
} }