Add RDRAND feature

Author: Jethro Beekman

Cargo.toml

@@ -20,6 +20,7 @@ core_io = { version = "0.0", optional = true }
 [features]
 default = ["libc"]
 no_std = ["core_io"]
+rdrand = []
 
 [dev-dependencies]
 log = "0.3.0"

src/lib.rs

@@ -243,6 +243,7 @@
 
 #![cfg_attr(feature="no_std",no_std)]
 #![cfg_attr(feature="no_std",feature(alloc,collections))]
+#![cfg_attr(feature="rdrand",feature(asm))]
 
 #[cfg(test)] #[macro_use] extern crate log;
 
@@ -251,16 +252,16 @@
 #[cfg(feature="no_std")] extern crate alloc;
 #[cfg(feature="no_std")] extern crate collections;
 
-#[cfg(not(feature="no_std"))] use core::cell::RefCell;
+#[cfg(all(not(feature="no_std"),not(feature="rdrand")))] use core::cell::RefCell;
 use core::marker;
 use core::mem;
 #[cfg(not(feature="no_std"))] use std::io;
-#[cfg(not(feature="no_std"))] use std::rc::Rc;
+#[cfg(all(not(feature="no_std"),not(feature="rdrand")))] use std::rc::Rc;
 use core::num::Wrapping as w;
 #[cfg(feature="no_std")] use alloc::boxed::Box;
 #[cfg(feature="no_std")] use collections::vec::Vec;
 
-#[cfg(not(feature="no_std"))] pub use os::OsRng;
+#[cfg(any(not(feature="no_std"),feature="rdrand"))] pub use os::OsRng;
 
 pub use isaac::{IsaacRng, Isaac64Rng};
 pub use chacha::ChaChaRng;
@@ -278,7 +279,7 @@ pub mod isaac;
 pub mod chacha;
 pub mod reseeding;
 mod rand_impls;
-#[cfg(not(feature="no_std"))] pub mod os;
+#[cfg(any(not(feature="no_std"),feature="rdrand"))] pub mod os;
 pub mod read;
 
 #[allow(bad_style)]
@@ -814,7 +815,7 @@ impl StdRng {
     ///
     /// Reading the randomness from the OS may fail, and any error is
     /// propagated via the `io::Result` return value.
-    #[cfg(not(feature="no_std"))]
+    #[cfg(any(not(feature="no_std"),feature="rdrand"))]
     pub fn new() -> io::Result<StdRng> {
         OsRng::new().map(|mut r| StdRng { rng: r.gen() })
     }
@@ -853,7 +854,7 @@ impl<'a> SeedableRng<&'a [usize]> for StdRng {
 ///
 /// This will read randomness from the operating system to seed the
 /// generator.
-#[cfg(not(feature="no_std"))]
+#[cfg(any(not(feature="no_std"),feature="rdrand"))]
 pub fn weak_rng() -> XorShiftRng {
     match OsRng::new() {
         Ok(mut r) => r.gen(),
@@ -862,10 +863,10 @@ pub fn weak_rng() -> XorShiftRng {
 }
 
 /// Controls how the thread-local RNG is reseeded.
-#[cfg(not(feature="no_std"))]
+#[cfg(all(not(feature="no_std"),not(feature="rdrand")))]
 struct ThreadRngReseeder;
 
-#[cfg(not(feature="no_std"))]
+#[cfg(all(not(feature="no_std"),not(feature="rdrand")))]
 impl reseeding::Reseeder<StdRng> for ThreadRngReseeder {
     fn reseed(&mut self, rng: &mut StdRng) {
         *rng = match StdRng::new() {
@@ -874,14 +875,14 @@ impl reseeding::Reseeder<StdRng> for ThreadRngReseeder {
         }
     }
 }
-#[cfg(not(feature="no_std"))]
+#[cfg(all(not(feature="no_std"),not(feature="rdrand")))]
 const THREAD_RNG_RESEED_THRESHOLD: u64 = 32_768;
-#[cfg(not(feature="no_std"))]
+#[cfg(all(not(feature="no_std"),not(feature="rdrand")))]
 type ThreadRngInner = reseeding::ReseedingRng<StdRng, ThreadRngReseeder>;
 
 /// The thread-local RNG.
+#[cfg(all(not(feature="no_std"),not(feature="rdrand")))]
 #[derive(Clone)]
-#[cfg(not(feature="no_std"))]
 pub struct ThreadRng {
     rng: Rc<RefCell<ThreadRngInner>>,
 }
@@ -897,7 +898,7 @@ pub struct ThreadRng {
 /// if the operating system random number generator is rigged to give
 /// the same sequence always. If absolute consistency is required,
 /// explicitly select an RNG, e.g. `IsaacRng` or `Isaac64Rng`.
-#[cfg(not(feature="no_std"))]
+#[cfg(all(not(feature="no_std"),not(feature="rdrand")))]
 pub fn thread_rng() -> ThreadRng {
     // used to make space in TLS for a random number generator
     thread_local!(static THREAD_RNG_KEY: Rc<RefCell<ThreadRngInner>> = {
@@ -914,7 +915,7 @@ pub fn thread_rng() -> ThreadRng {
     ThreadRng { rng: THREAD_RNG_KEY.with(|t| t.clone()) }
 }
 
-#[cfg(not(feature="no_std"))]
+#[cfg(all(not(feature="no_std"),not(feature="rdrand")))]
 impl Rng for ThreadRng {
     fn next_u32(&mut self) -> u32 {
         self.rng.borrow_mut().next_u32()
@@ -930,6 +931,14 @@ impl Rng for ThreadRng {
     }
 }
 
+#[cfg(feature="rdrand")]
+pub use os::OsRng as ThreadRng;
+
+#[cfg(feature="rdrand")]
+pub fn thread_rng() -> ThreadRng {
+    OsRng::new().unwrap()
+}
+
 /// Generates a random value using the thread-local random number generator.
 ///
 /// `random()` can generate various types of random things, and so may require
@@ -972,7 +981,7 @@ impl Rng for ThreadRng {
 ///     *x = rng.gen();
 /// }
 /// ```
-#[cfg(not(feature="no_std"))]
+#[cfg(any(not(feature="no_std"),feature="rdrand"))]
 #[inline]
 pub fn random<T: Rand>() -> T {
     thread_rng().gen()

src/os.rs

@@ -11,7 +11,8 @@
 //! Interfaces to the operating system provided random number
 //! generators.
 
-use std::io;
+#[cfg(not(feature="no_std"))] use std::io;
+#[cfg(feature="no_std")] use io;
 use Rng;
 
 /// A random number generator that retrieves randomness straight from
@@ -41,7 +42,7 @@ impl Rng for OsRng {
 }
 
 #[cfg(all(unix, not(target_os = "ios"),
-          not(target_os = "nacl")))]
+          not(target_os = "nacl"), not(feature = "rdrand")))]
 mod imp {
     extern crate libc;
 
@@ -198,7 +199,7 @@ mod imp {
     }
 }
 
-#[cfg(target_os = "ios")]
+#[cfg(all(target_os = "ios", not(feature = "rdrand")))]
 mod imp {
     extern crate libc;
 
@@ -248,7 +249,7 @@ mod imp {
     }
 }
 
-#[cfg(windows)]
+#[cfg(all(windows, not(feature = "rdrand")))]
 mod imp {
     use std::io;
     use std::mem;
@@ -339,7 +340,7 @@ mod imp {
     }
 }
 
-#[cfg(target_os = "nacl")]
+#[cfg(all(target_os = "nacl", not(feature = "rdrand")))]
 mod imp {
     extern crate libc;
 
@@ -417,6 +418,36 @@ mod imp {
 }
 
 
+#[cfg(feature = "rdrand")]
+mod imp {
+    #[cfg(not(feature="no_std"))] use std::io;
+    #[cfg(feature="no_std")] use io;
+
+    use Rng;
+
+    pub struct OsRng;
+
+    impl OsRng {
+        pub fn new() -> io::Result<OsRng> {
+            Ok(OsRng)
+        }
+    }
+
+    impl Rng for OsRng {
+        fn next_u32(&mut self) -> u32 {
+            let ret;
+            unsafe{asm!("rdrand $0":"=r"(ret):::"volatile")};
+            ret
+        }
+        #[cfg(target_arch="x86_64")]
+        fn next_u64(&mut self) -> u64 {
+            let ret;
+            unsafe{asm!("rdrand $0":"=r"(ret):::"volatile")};
+            ret
+        }
+    }
+}
+
 #[cfg(test)]
 mod test {
     use std::sync::mpsc::channel;