From 5e84eedfbb29fd0d49ca119eb3716118cf6bae57 Mon Sep 17 00:00:00 2001
From: Isis Lovecruft <isis@torproject.org>
Date: Wed, 15 Mar 2017 18:44:59 +0000
Subject: [PATCH] Add benchmarks with blake2b.

---
 src/ed25519.rs | 28 ++++++++++++++++++++++++++++
 src/lib.rs     |  4 ++++
 2 files changed, 32 insertions(+)

diff --git a/src/ed25519.rs b/src/ed25519.rs
index 215eab8..74863c6 100644
--- a/src/ed25519.rs
+++ b/src/ed25519.rs
@@ -495,6 +495,8 @@ mod test {
 
 #[cfg(all(test, feature = "bench"))]
 mod bench {
+    use blake2::Blake2b;
+    use digest::Digest;
     use generic_array::GenericArray;
     use generic_array::typenum::{U64, U128};
     use test::Bencher;
@@ -566,4 +568,30 @@ mod bench {
 
         b.iter(| | Keypair::generate::<Sha512>(&mut rng));
     }
+
+    #[bench]
+    fn blake2b_sign(b: &mut Bencher) {
+        let mut cspring: OsRng = OsRng::new().unwrap();
+        let keypair: Keypair = Keypair::generate::<Blake2b>(&mut cspring);
+        let msg: &[u8] = "".as_bytes();
+
+        b.iter(| | keypair.sign::<Blake2b>(msg));
+    }
+
+    #[bench]
+    fn blake2b_verify(b: &mut Bencher) {
+        let mut cspring: OsRng = OsRng::new().unwrap();
+        let keypair: Keypair = Keypair::generate::<Blake2b>(&mut cspring);
+        let msg: &[u8] = "".as_bytes();
+        let sig: Signature = keypair.sign::<Blake2b>(msg);
+
+        b.iter(| | keypair.verify::<Blake2b>(msg, &sig));
+    }
+
+    #[bench]
+    fn blake2b_key_generation(b: &mut Bencher) {
+        let mut rng: ZeroRng = ZeroRng::new();
+
+        b.iter(| | Keypair::generate::<Blake2b>(&mut rng));
+    }
 }
diff --git a/src/lib.rs b/src/lib.rs
index 73daa41..41b11c0 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -131,6 +131,10 @@ extern crate test;
 #[cfg(test)]
 extern crate rustc_serialize;
 
+#[cfg(all(test, feature = "bench"))]
+extern crate blake2;
+
+
 mod ed25519;
 
 // Export everything public in ed25519.