1 /***********************************************************************
2 * Copyright (c) 2018-2020 Andrew Poelstra, Jonas Nick *
3 * Distributed under the MIT software license, see the accompanying *
4 * file COPYING or https://www.opensource.org/licenses/mit-license.php.*
5 ***********************************************************************/
6
7 #include <string.h>
8 #include <stdlib.h>
9
10
11 #include "include/secp256k1.h"
12 #include "include/secp256k1_schnorrsig.h"
13 #include "util.h"
14 #include "bench.h"
15
16 typedef struct {
17 secp256k1_context *ctx;
18 int n;
19
20 const secp256k1_keypair **keypairs;
21 const unsigned char **pk;
22 const unsigned char **sigs;
23 const unsigned char **msgs;
24 } bench_schnorrsig_data;
25
bench_schnorrsig_sign(void * arg,int iters)26 void bench_schnorrsig_sign(void* arg, int iters) {
27 bench_schnorrsig_data *data = (bench_schnorrsig_data *)arg;
28 int i;
29 unsigned char msg[32] = "benchmarkexamplemessagetemplate";
30 unsigned char sig[64];
31
32 for (i = 0; i < iters; i++) {
33 msg[0] = i;
34 msg[1] = i >> 8;
35 CHECK(secp256k1_schnorrsig_sign(data->ctx, sig, msg, data->keypairs[i], NULL, NULL));
36 }
37 }
38
bench_schnorrsig_verify(void * arg,int iters)39 void bench_schnorrsig_verify(void* arg, int iters) {
40 bench_schnorrsig_data *data = (bench_schnorrsig_data *)arg;
41 int i;
42
43 for (i = 0; i < iters; i++) {
44 secp256k1_xonly_pubkey pk;
45 CHECK(secp256k1_xonly_pubkey_parse(data->ctx, &pk, data->pk[i]) == 1);
46 CHECK(secp256k1_schnorrsig_verify(data->ctx, data->sigs[i], data->msgs[i], &pk));
47 }
48 }
49
main(void)50 int main(void) {
51 int i;
52 bench_schnorrsig_data data;
53 int iters = get_iters(10000);
54
55 data.ctx = secp256k1_context_create(SECP256K1_CONTEXT_VERIFY | SECP256K1_CONTEXT_SIGN);
56 data.keypairs = (const secp256k1_keypair **)malloc(iters * sizeof(secp256k1_keypair *));
57 data.pk = (const unsigned char **)malloc(iters * sizeof(unsigned char *));
58 data.msgs = (const unsigned char **)malloc(iters * sizeof(unsigned char *));
59 data.sigs = (const unsigned char **)malloc(iters * sizeof(unsigned char *));
60
61 for (i = 0; i < iters; i++) {
62 unsigned char sk[32];
63 unsigned char *msg = (unsigned char *)malloc(32);
64 unsigned char *sig = (unsigned char *)malloc(64);
65 secp256k1_keypair *keypair = (secp256k1_keypair *)malloc(sizeof(*keypair));
66 unsigned char *pk_char = (unsigned char *)malloc(32);
67 secp256k1_xonly_pubkey pk;
68 msg[0] = sk[0] = i;
69 msg[1] = sk[1] = i >> 8;
70 msg[2] = sk[2] = i >> 16;
71 msg[3] = sk[3] = i >> 24;
72 memset(&msg[4], 'm', 28);
73 memset(&sk[4], 's', 28);
74
75 data.keypairs[i] = keypair;
76 data.pk[i] = pk_char;
77 data.msgs[i] = msg;
78 data.sigs[i] = sig;
79
80 CHECK(secp256k1_keypair_create(data.ctx, keypair, sk));
81 CHECK(secp256k1_schnorrsig_sign(data.ctx, sig, msg, keypair, NULL, NULL));
82 CHECK(secp256k1_keypair_xonly_pub(data.ctx, &pk, NULL, keypair));
83 CHECK(secp256k1_xonly_pubkey_serialize(data.ctx, pk_char, &pk) == 1);
84 }
85
86 run_benchmark("schnorrsig_sign", bench_schnorrsig_sign, NULL, NULL, (void *) &data, 10, iters);
87 run_benchmark("schnorrsig_verify", bench_schnorrsig_verify, NULL, NULL, (void *) &data, 10, iters);
88
89 for (i = 0; i < iters; i++) {
90 free((void *)data.keypairs[i]);
91 free((void *)data.pk[i]);
92 free((void *)data.msgs[i]);
93 free((void *)data.sigs[i]);
94 }
95 free(data.keypairs);
96 free(data.pk);
97 free(data.msgs);
98 free(data.sigs);
99
100 secp256k1_context_destroy(data.ctx);
101 return 0;
102 }
103