-
Notifications
You must be signed in to change notification settings - Fork 172
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add test for custom RNG handler (#197)
See #194 This uses the fact that `wasm32-unknown-unknown` is an "unsupported" target. This means we can use the `"custom"` feature to define a custom handler, and then write tests to make sure that function is called.
- Loading branch information
Showing
4 changed files
with
61 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,50 @@ | ||
// Test that a custom handler works on wasm32-unknown-unknown | ||
#![cfg(all( | ||
target_arch = "wasm32", | ||
target_os = "unknown", | ||
feature = "custom", | ||
not(feature = "js") | ||
))] | ||
|
||
use wasm_bindgen_test::wasm_bindgen_test as test; | ||
#[cfg(feature = "test-in-browser")] | ||
wasm_bindgen_test::wasm_bindgen_test_configure!(run_in_browser); | ||
|
||
use core::{ | ||
num::NonZeroU32, | ||
sync::atomic::{AtomicU8, Ordering}, | ||
}; | ||
use getrandom::{getrandom, register_custom_getrandom, Error}; | ||
|
||
fn len7_err() -> Error { | ||
NonZeroU32::new(Error::INTERNAL_START + 7).unwrap().into() | ||
} | ||
|
||
fn super_insecure_rng(buf: &mut [u8]) -> Result<(), Error> { | ||
// Length 7 buffers return a custom error | ||
if buf.len() == 7 { | ||
return Err(len7_err()); | ||
} | ||
// Otherwise, increment an atomic counter | ||
static COUNTER: AtomicU8 = AtomicU8::new(0); | ||
for b in buf { | ||
*b = COUNTER.fetch_add(1, Ordering::Relaxed); | ||
} | ||
Ok(()) | ||
} | ||
|
||
register_custom_getrandom!(super_insecure_rng); | ||
|
||
#[test] | ||
fn custom_rng_output() { | ||
let mut buf = [0u8; 4]; | ||
assert_eq!(getrandom(&mut buf), Ok(())); | ||
assert_eq!(buf, [0, 1, 2, 3]); | ||
assert_eq!(getrandom(&mut buf), Ok(())); | ||
assert_eq!(buf, [4, 5, 6, 7]); | ||
} | ||
|
||
#[test] | ||
fn rng_err_output() { | ||
assert_eq!(getrandom(&mut [0; 7]), Err(len7_err())); | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,3 +1,11 @@ | ||
// Don't test on custom wasm32-unknown-unknown | ||
#![cfg(not(all( | ||
target_arch = "wasm32", | ||
target_os = "unknown", | ||
feature = "custom", | ||
not(feature = "js") | ||
)))] | ||
|
||
// Use the normal getrandom implementation on this architecture. | ||
use getrandom::getrandom as getrandom_impl; | ||
mod common; |