In progress

CodeChain-io · Feb 18, 2020 · 61ad23b · 61ad23b
1 parent 370c3d4
commit 61ad23b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 22 deletions.
diff --git a/key/benches/schnorr.rs → key/benches/bls.rs b/key/benches/schnorr.rs → key/benches/bls.rs
@@ -19,34 +19,25 @@
 extern crate codechain_key as ckey;
 extern crate test;
 
-use ckey::{recover_schnorr, sign_schnorr, verify_schnorr, Generator, Message, Random};
+use ckey::{sign_bls, verify_bls, Generator, Message, Random, BLSKeyPair};
+use primitives::H256;
 use test::Bencher;
 
 #[bench]
-fn schnorr_sign(b: &mut Bencher) {
+fn bls_sign(b: &mut Bencher) {
     b.iter(|| {
-        let key_pair = Random.generate().unwrap();
+        let key_pair = BLSKeyPair::from_secret(H256::random());
         let message = Message::random();
-        let _signature = sign_schnorr(key_pair.private(), &message).unwrap();
+        let _signature = sign_bls(key_pair.private(), &message);
     });
 }
 
 #[bench]
-fn schnorr_sign_and_verify(b: &mut Bencher) {
+fn bls_sign_and_verify(b: &mut Bencher) {
     b.iter(|| {
-        let key_pair = Random.generate().unwrap();
+        let key_pair = BLSKeyPair::from_secret(H256::random());
         let message = Message::random();
-        let signature = sign_schnorr(key_pair.private(), &message).unwrap();
-        assert_eq!(Ok(true), verify_schnorr(key_pair.public(), &signature, &message));
-    });
-}
-
-#[bench]
-fn schnorr_sign_and_recover(b: &mut Bencher) {
-    b.iter(|| {
-        let key_pair = Random.generate().unwrap();
-        let message = Message::random();
-        let signature = sign_schnorr(key_pair.private(), &message).unwrap();
-        assert_eq!(Ok(*key_pair.public()), recover_schnorr(&signature, &message));
+        let signature = sign_bls(key_pair.private(), &message);
+        assert_eq!(Ok(true), verify_bls(key_pair.public(), &signature, &message));
     });
 }
diff --git a/key/benches/tendermint.rs b/key/benches/tendermint.rs
@@ -21,7 +21,7 @@ extern crate codechain_key as ckey;
 extern crate primitives;
 extern crate test;
 
-use ckey::{sign_schnorr, verify_schnorr, Generator, Message, Random};
+use ckey::{sign_bls, verify_bls, Generator, Message, Random};
 use test::Bencher;
 
 #[bench]
@@ -39,7 +39,7 @@ fn tendermint_max_step_time(b: &mut Bencher) {
     while i < num_validators - 1 {
         let key_pair = Random.generate().unwrap();
         let message = Message::random();
-        let signature = sign_schnorr(key_pair.private(), &message).unwrap();
+        let signature = sign_bls(key_pair.private(), &message).unwrap();
 
         key_pairs.push(key_pair);
         messages.push(message);
@@ -48,11 +48,11 @@ fn tendermint_max_step_time(b: &mut Bencher) {
         i += 1;
     }
     b.iter(|| {
-        sign_schnorr(key_pair_self.private(), &message_self).unwrap();
+        sign_bls(key_pair_self.private(), &message_self).unwrap();
 
         let mut i = 0;
         while i < num_validators - 1 {
-            assert_eq!(Ok(true), verify_schnorr(key_pairs[i].public(), &signatures[i], &messages[i]));
+            assert_eq!(Ok(true), verify_bls(key_pairs[i].public(), &signatures[i], &messages[i]));
             i += 1;
         }
     });