From 80fd096b0749412229a863da82e91054e6ab7363 Mon Sep 17 00:00:00 2001
From: Alexander Block <ablock84@gmail.com>
Date: Fri, 14 Sep 2018 18:04:45 +0200
Subject: [PATCH] Add ECDSA benchmarks

---
 src/Makefile.bench.include |  1 +
 src/bench/bench_dash.cpp   |  2 +
 src/bench/ecdsa.cpp        | 77 ++++++++++++++++++++++++++++++++++++++
 3 files changed, 80 insertions(+)
 create mode 100644 src/bench/ecdsa.cpp

diff --git a/src/Makefile.bench.include b/src/Makefile.bench.include
index 07f4089a5c..96170ff4e9 100644
--- a/src/Makefile.bench.include
+++ b/src/Makefile.bench.include
@@ -19,6 +19,7 @@ bench_bench_dash_SOURCES = \
   bench/bls_dkg.cpp \
   bench/checkblock.cpp \
   bench/checkqueue.cpp \
+  bench/ecdsa.cpp \
   bench/Examples.cpp \
   bench/rollingbloom.cpp \
   bench/crypto_hash.cpp \
diff --git a/src/bench/bench_dash.cpp b/src/bench/bench_dash.cpp
index 3f654e1fdc..533ae7194c 100644
--- a/src/bench/bench_dash.cpp
+++ b/src/bench/bench_dash.cpp
@@ -17,6 +17,8 @@ int
 main(int argc, char** argv)
 {
     ECC_Start();
+    ECCVerifyHandle verifyHandle;
+
     BLSInit();
     SetupEnvironment();
     fPrintToDebugLog = false; // don't want to write to debug.log file
diff --git a/src/bench/ecdsa.cpp b/src/bench/ecdsa.cpp
new file mode 100644
index 0000000000..706608eac2
--- /dev/null
+++ b/src/bench/ecdsa.cpp
@@ -0,0 +1,77 @@
+// Copyright (c) 2018 The Dash Core developers
+// Distributed under the MIT software license, see the accompanying
+// file COPYING or http://www.opensource.org/licenses/mit-license.php.
+
+#include "bench.h"
+
+#include "key.h"
+
+static void ECDSASign(benchmark::State& state)
+{
+    std::vector<CKey> keys;
+    std::vector<uint256> hashes;
+    for (size_t i = 0; i < 100; i++) {
+        CKey k;
+        k.MakeNewKey(false);
+        keys.emplace_back(k);
+        hashes.emplace_back(::SerializeHash((int)i));
+    }
+
+    // Benchmark.
+    size_t i = 0;
+    while (state.KeepRunning()) {
+        std::vector<unsigned char> sig;
+        keys[i].Sign(hashes[i], sig);
+        i = (i + 1) % keys.size();
+    }
+}
+
+static void ECDSAVerify(benchmark::State& state)
+{
+    std::vector<CPubKey> keys;
+    std::vector<uint256> hashes;
+    std::vector<std::vector<unsigned char>> sigs;
+    for (size_t i = 0; i < 100; i++) {
+        CKey k;
+        k.MakeNewKey(false);
+        keys.emplace_back(k.GetPubKey());
+        hashes.emplace_back(::SerializeHash((int)i));
+        std::vector<unsigned char> sig;
+        k.Sign(hashes[i], sig);
+        sigs.emplace_back(sig);
+    }
+
+    // Benchmark.
+    size_t i = 0;
+    while (state.KeepRunning()) {
+        keys[i].Verify(hashes[i], sigs[i]);
+        i = (i + 1) % keys.size();
+    }
+}
+
+static void ECDSAVerify_LargeBlock(benchmark::State& state)
+{
+    std::vector<CPubKey> keys;
+    std::vector<uint256> hashes;
+    std::vector<std::vector<unsigned char>> sigs;
+    for (size_t i = 0; i < 1000; i++) {
+        CKey k;
+        k.MakeNewKey(false);
+        keys.emplace_back(k.GetPubKey());
+        hashes.emplace_back(::SerializeHash((int)i));
+        std::vector<unsigned char> sig;
+        k.Sign(hashes[i], sig);
+        sigs.emplace_back(sig);
+    }
+
+    // Benchmark.
+    while (state.KeepRunning()) {
+        for (size_t i = 0; i < keys.size(); i++) {
+            keys[i].Verify(hashes[i], sigs[i]);
+        }
+    }
+}
+
+BENCHMARK(ECDSASign)
+BENCHMARK(ECDSAVerify)
+BENCHMARK(ECDSAVerify_LargeBlock)