forked from erickt/rustcrypto
-
Notifications
You must be signed in to change notification settings - Fork 2
/
rand.rs
34 lines (27 loc) · 626 Bytes
/
rand.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
use std::libc::c_int;
use std::vec;
#[link_args = "-lcrypto"]
#[abi = "cdecl"]
extern {
fn RAND_bytes(buf: *mut u8, num: c_int) -> c_int;
}
pub fn rand_bytes(len: uint) -> ~[u8] {
let mut out = vec::with_capacity(len);
do out.as_mut_buf |out_buf, len| {
unsafe {
let r = RAND_bytes(out_buf, len as c_int);
if r != 1 as c_int { fail!() }
}
}
unsafe { vec::raw::set_len(&mut out, len); }
out
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_rand_bytes() {
let bytes = rand_bytes(32u);
println(fmt!("%?", bytes));
}
}