liz/veilid
1
0
Fork 0
Code Pull Requests Packages Releases Activity

Use getrandom for RNG implementation in WASM

Browse Source
This commit is contained in:
Victor Graf
2023-09-03 00:30:34 +00:00
committed by Christien Rioux
parent 20f55745b0
commit f454f56816
2 changed files with 14 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
veilid-tools/src/random.rs
View File

@@ -25,56 +25,17 @@ impl RngCore for VeilidRng {
}
}
cfg_if! {
if #[cfg(target_arch = "wasm32")] {
use js_sys::Math;
pub fn random_bytes(dest: &mut [u8]) {
let len = dest.len();
let u32len = len / 4;
let remlen = len % 4;
for n in 0..u32len {
let r = (Math::random() * (u32::max_value() as f64)) as u32;
dest[n * 4 + 0] = (r & 0xFF) as u8;
dest[n * 4 + 1] = ((r >> 8) & 0xFF) as u8;
dest[n * 4 + 2] = ((r >> 16) & 0xFF) as u8;
dest[n * 4 + 3] = ((r >> 24) & 0xFF) as u8;
}
if remlen > 0 {
let r = (Math::random() * (u32::max_value() as f64)) as u32;
for n in 0..remlen {
dest[u32len * 4 + n] = ((r >> (n * 8)) & 0xFF) as u8;
}
}
}
pub fn get_random_u32() -> u32 {
(Math::random() * (u32::max_value() as f64)) as u32
}
pub fn get_random_u64() -> u64 {
let v1: u32 = get_random_u32();
let v2: u32 = get_random_u32();
((v1 as u64) << 32) | ((v2 as u32) as u64)
}
} else {
pub fn random_bytes(dest: &mut [u8]) {
let mut rng = rand::thread_rng();
rng.fill_bytes(dest);
}
pub fn get_random_u32() -> u32 {
let mut rng = rand::thread_rng();
rng.next_u32()
}
pub fn get_random_u64() -> u64 {
let mut rng = rand::thread_rng();
rng.next_u64()
}
}
pub fn random_bytes(dest: &mut [u8]) {
let mut rng = rand::thread_rng();
rng.fill_bytes(dest);
}
pub fn get_random_u32() -> u32 {
let mut rng = rand::thread_rng();
rng.next_u32()
}
pub fn get_random_u64() -> u64 {
let mut rng = rand::thread_rng();
rng.next_u64()
}