Simplify hash_g2.

Cargo.toml

@@ -9,6 +9,7 @@ byteorder = "1.2.3"
 derive_deref = "1.0.1"
 env_logger = "0.5.10"
 error-chain = "0.11.0"
+init_with = "1.1.0"
 itertools = "0.7"
 log = "0.4.1"
 merkle = { git = "https://github.com/afck/merkle.rs", branch = "public-proof" }

src/crypto/mod.rs

@@ -1,6 +1,7 @@
 mod error;
 
 use byteorder::{BigEndian, ByteOrder};
+use init_with::InitWith;
 
 use pairing::{CurveAffine, CurveProjective, Engine, Field, PrimeField};
 use rand::{ChaChaRng, Rand, Rng, SeedableRng};
@@ -8,6 +9,9 @@ use ring::digest;
 
 use self::error::{ErrorKind, Result};
 
+/// The number of words (`u32`) in a ChaCha RNG seed.
+const CHACHA_RNG_SEED_SIZE: usize = 8;
+
 /// Returns a hash of the given message in `G2`.
 pub fn hash_g2<E, M>(msg: M) -> E::G2
 where
@@ -16,14 +20,10 @@ where
     M: AsRef<[u8]>,
 {
     let digest = digest::digest(&digest::SHA256, msg.as_ref());
-    // The `pairing` crate's `G2` implements `Rand`. We initialize a seedable RNG with the SHA256
-    // digest, and use it to generate the element.
-    let mut msg_u32: Vec<u32> = Vec::with_capacity(256 / 32);
-    for chunk in digest.as_ref().chunks(4) {
-        let word = BigEndian::read_u32(chunk);
-        msg_u32.push(word);
-    }
-    let mut rng = ChaChaRng::from_seed(&msg_u32);
+    let seed = <[u32; CHACHA_RNG_SEED_SIZE]>::init_with_indices(|i| {
+        BigEndian::read_u32(&digest.as_ref()[(4 * i)..(4 * i + 4)])
+    });
+    let mut rng = ChaChaRng::from_seed(&seed);
     rng.gen()
 }
 

src/lib.rs

@@ -11,6 +11,7 @@ extern crate byteorder;
 extern crate derive_deref;
 #[macro_use]
 extern crate error_chain;
+extern crate init_with;
 #[macro_use]
 extern crate log;
 extern crate itertools;