Add benchmarks

Author: sanket1729

src/bench_bulletproofs_pp.c

@@ -0,0 +1,87 @@
+/**********************************************************************
+ * Copyright (c) 2020 Andrew Poelstra                                 *
+ * Distributed under the MIT software license, see the accompanying   *
+ * file COPYING or http://www.opensource.org/licenses/mit-license.php.*
+ **********************************************************************/
+
+#include <stdint.h>
+
+#include "include/secp256k1.h"
+#include "include/secp256k1_bulletproofs.h"
+#include "util.h"
+#include "bench.h"
+
+typedef struct {
+    secp256k1_context* ctx;
+    secp256k1_bulletproofs_generators* gens;
+    secp256k1_scratch_space *scratch;
+    secp256k1_pedersen_commitment commit;
+    unsigned char proof[1000];
+    unsigned char blind[32];
+    unsigned char nonce[32];
+    size_t proof_len;
+    size_t n_bits;
+    size_t base;
+    uint64_t min_value;
+    uint64_t value;
+} bench_bulletproofs_data;
+
+static void bench_bulletproofs_setup(void* arg) {
+    bench_bulletproofs_data *data = (bench_bulletproofs_data*)arg;
+
+    data->min_value = 0;
+    data->value = 100;
+    data->proof_len = sizeof(data->proof);
+    memset(data->blind, 0x77, 32);
+    memset(data->nonce, 0x0, 32);
+    CHECK(secp256k1_pedersen_commit(data->ctx, &data->commit, data->blind, data->value, secp256k1_generator_h));
+
+    CHECK(secp256k1_bulletproofs_pp_rangeproof_prove(data->ctx, data->scratch, data->gens, secp256k1_generator_h, data->proof, &data->proof_len, data->n_bits, data->base, data->value, 0, &data->commit, data->blind, data->nonce, NULL, 0));
+    CHECK(secp256k1_bulletproofs_pp_rangeproof_verify(data->ctx, data->scratch, data->gens, secp256k1_generator_h, data->proof, data->proof_len, data->n_bits, data->base, data->min_value, &data->commit, NULL, 0));
+}
+
+static void bench_bulletproofs_prove(void* arg, int iters) {
+    bench_bulletproofs_data *data = (bench_bulletproofs_data*)arg;
+    int i;
+
+    for (i = 0; i < iters; i++) {
+        data->nonce[1] = i;
+        data->nonce[2] = i >> 8;
+        data->nonce[3] = i >> 16;
+        data->proof_len = sizeof(data->proof);
+        CHECK(secp256k1_bulletproofs_pp_rangeproof_prove(data->ctx, data->scratch, data->gens, secp256k1_generator_h, data->proof, &data->proof_len, data->n_bits, data->base, data->value, 0, &data->commit, data->blind, data->nonce, NULL, 0));
+    }
+}
+
+static void bench_bulletproofs_verify(void* arg, int iters) {
+    bench_bulletproofs_data *data = (bench_bulletproofs_data*)arg;
+    int i;
+
+    for (i = 0; i < iters; i++) {
+        CHECK(secp256k1_bulletproofs_pp_rangeproof_verify(data->ctx, data->scratch, data->gens, secp256k1_generator_h, data->proof, data->proof_len, data->n_bits, data->base, data->min_value, &data->commit, NULL, 0));
+    }
+}
+
+int main(void) {
+    bench_bulletproofs_data data;
+    int iters = get_iters(32);
+    char test_name[64];
+
+    data.ctx = secp256k1_context_create(SECP256K1_CONTEXT_SIGN | SECP256K1_CONTEXT_VERIFY);
+    data.gens = secp256k1_bulletproofs_generators_create(data.ctx, 24);
+    data.scratch = secp256k1_scratch_space_create(data.ctx, 8000 * 1024);
+
+    data.n_bits = 1ul << 6;
+    data.base = 16;
+    sprintf(test_name, "bulletproofs_uncompressed_prove_64bits_16base");
+    run_benchmark(test_name, bench_bulletproofs_prove, bench_bulletproofs_setup, NULL, &data, 20, iters);
+
+    sprintf(test_name, "bulletproofs_uncompressed_verify_64bits_16base");
+    run_benchmark(test_name, bench_bulletproofs_verify, bench_bulletproofs_setup, NULL, &data, 20, iters);
+
+    secp256k1_scratch_space_destroy(data.ctx, data.scratch);
+    secp256k1_bulletproofs_generators_destroy(data.ctx, data.gens);
+    secp256k1_context_destroy(data.ctx);
+
+    return 0;
+}

src/modules/bulletproofs/Makefile.am.include

@@ -8,7 +8,7 @@ noinst_HEADERS += src/modules/bulletproofs/tests_impl.h
 
 if USE_BENCHMARK
 noinst_PROGRAMS += bench_bulletproofs
-bench_bulletproofs_SOURCES = src/bench_bulletproofs.c
+bench_bulletproofs_SOURCES = src/bench_bulletproofs_pp.c
 bench_bulletproofs_LDADD = libsecp256k1.la $(SECP_LIBS)
 bench_bulletproofs_LDFLAGS = -static
 endif