Move src/ into core/src. Top-level crate is now called solana-workspace

core/src/gen_keys.rs

@@ -0,0 +1,68 @@
+//! The `signature` module provides functionality for public, and private keys.
+
+use rand::{Rng, SeedableRng};
+use rand_chacha::ChaChaRng;
+use rayon::prelude::*;
+use solana_sdk::signature::Keypair;
+use untrusted::Input;
+
+pub struct GenKeys {
+    generator: ChaChaRng,
+}
+
+impl GenKeys {
+    pub fn new(seed: [u8; 32]) -> GenKeys {
+        let generator = ChaChaRng::from_seed(seed);
+        GenKeys { generator }
+    }
+
+    fn gen_seed(&mut self) -> [u8; 32] {
+        let mut seed = [0u8; 32];
+        self.generator.fill(&mut seed);
+        seed
+    }
+
+    fn gen_n_seeds(&mut self, n: u64) -> Vec<[u8; 32]> {
+        (0..n).map(|_| self.gen_seed()).collect()
+    }
+
+    pub fn gen_n_keypairs(&mut self, n: u64) -> Vec<Keypair> {
+        self.gen_n_seeds(n)
+            .into_par_iter()
+            .map(|seed| Keypair::from_seed_unchecked(Input::from(&seed)).unwrap())
+            .collect()
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    pub use solana_sdk::pubkey::Pubkey;
+    use solana_sdk::signature::KeypairUtil;
+    use std::collections::HashSet;
+
+    #[test]
+    fn test_new_key_is_deterministic() {
+        let seed = [0u8; 32];
+        let mut gen0 = GenKeys::new(seed);
+        let mut gen1 = GenKeys::new(seed);
+
+        for _ in 0..100 {
+            assert_eq!(gen0.gen_seed().to_vec(), gen1.gen_seed().to_vec());
+        }
+    }
+
+    fn gen_n_pubkeys(seed: [u8; 32], n: u64) -> HashSet<Pubkey> {
+        GenKeys::new(seed)
+            .gen_n_keypairs(n)
+            .into_iter()
+            .map(|x| x.pubkey())
+            .collect()
+    }
+
+    #[test]
+    fn test_gen_n_pubkeys_deterministic() {
+        let seed = [0u8; 32];
+        assert_eq!(gen_n_pubkeys(seed, 50), gen_n_pubkeys(seed, 50));
+    }
+}