145 lines
4.5 KiB
C
145 lines
4.5 KiB
C
/***********************************************************************
|
|
* Copyright (c) 2014 Pieter Wuille *
|
|
* Distributed under the MIT software license, see the accompanying *
|
|
* file COPYING or https://www.opensource.org/licenses/mit-license.php.*
|
|
***********************************************************************/
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#include "../include/secp256k1.h"
|
|
#include "util.h"
|
|
#include "bench.h"
|
|
|
|
typedef struct {
|
|
secp256k1_context *ctx;
|
|
unsigned char msg[32];
|
|
unsigned char key[32];
|
|
unsigned char sig[72];
|
|
size_t siglen;
|
|
unsigned char pubkey[33];
|
|
size_t pubkeylen;
|
|
} bench_verify_data;
|
|
|
|
static void bench_verify(void* arg, int iters) {
|
|
int i;
|
|
bench_verify_data* data = (bench_verify_data*)arg;
|
|
|
|
for (i = 0; i < iters; i++) {
|
|
secp256k1_pubkey pubkey;
|
|
secp256k1_ecdsa_signature sig;
|
|
data->sig[data->siglen - 1] ^= (i & 0xFF);
|
|
data->sig[data->siglen - 2] ^= ((i >> 8) & 0xFF);
|
|
data->sig[data->siglen - 3] ^= ((i >> 16) & 0xFF);
|
|
CHECK(secp256k1_ec_pubkey_parse(data->ctx, &pubkey, data->pubkey, data->pubkeylen) == 1);
|
|
CHECK(secp256k1_ecdsa_signature_parse_der(data->ctx, &sig, data->sig, data->siglen) == 1);
|
|
CHECK(secp256k1_ecdsa_verify(data->ctx, &sig, data->msg, &pubkey) == (i == 0));
|
|
data->sig[data->siglen - 1] ^= (i & 0xFF);
|
|
data->sig[data->siglen - 2] ^= ((i >> 8) & 0xFF);
|
|
data->sig[data->siglen - 3] ^= ((i >> 16) & 0xFF);
|
|
}
|
|
}
|
|
|
|
typedef struct {
|
|
secp256k1_context* ctx;
|
|
unsigned char msg[32];
|
|
unsigned char key[32];
|
|
} bench_sign_data;
|
|
|
|
static void bench_sign_setup(void* arg) {
|
|
int i;
|
|
bench_sign_data *data = (bench_sign_data*)arg;
|
|
|
|
for (i = 0; i < 32; i++) {
|
|
data->msg[i] = i + 1;
|
|
}
|
|
for (i = 0; i < 32; i++) {
|
|
data->key[i] = i + 65;
|
|
}
|
|
}
|
|
|
|
static void bench_sign_run(void* arg, int iters) {
|
|
int i;
|
|
bench_sign_data *data = (bench_sign_data*)arg;
|
|
|
|
unsigned char sig[74];
|
|
for (i = 0; i < iters; i++) {
|
|
size_t siglen = 74;
|
|
int j;
|
|
secp256k1_ecdsa_signature signature;
|
|
CHECK(secp256k1_ecdsa_sign(data->ctx, &signature, data->msg, data->key, NULL, NULL));
|
|
CHECK(secp256k1_ecdsa_signature_serialize_der(data->ctx, sig, &siglen, &signature));
|
|
for (j = 0; j < 32; j++) {
|
|
data->msg[j] = sig[j];
|
|
data->key[j] = sig[j + 32];
|
|
}
|
|
}
|
|
}
|
|
|
|
#ifdef ENABLE_MODULE_ECDH
|
|
# include "modules/ecdh/bench_impl.h"
|
|
#endif
|
|
|
|
#ifdef ENABLE_MODULE_RECOVERY
|
|
# include "modules/recovery/bench_impl.h"
|
|
#endif
|
|
|
|
#ifdef ENABLE_MODULE_SCHNORRSIG
|
|
# include "modules/schnorrsig/bench_impl.h"
|
|
#endif
|
|
|
|
int main(int argc, char** argv) {
|
|
int i;
|
|
secp256k1_pubkey pubkey;
|
|
secp256k1_ecdsa_signature sig;
|
|
bench_verify_data data;
|
|
|
|
int d = argc == 1;
|
|
int iters = get_iters(20000);
|
|
|
|
/* ECDSA verification benchmark */
|
|
data.ctx = secp256k1_context_create(SECP256K1_CONTEXT_SIGN | SECP256K1_CONTEXT_VERIFY);
|
|
|
|
for (i = 0; i < 32; i++) {
|
|
data.msg[i] = 1 + i;
|
|
}
|
|
for (i = 0; i < 32; i++) {
|
|
data.key[i] = 33 + i;
|
|
}
|
|
data.siglen = 72;
|
|
CHECK(secp256k1_ecdsa_sign(data.ctx, &sig, data.msg, data.key, NULL, NULL));
|
|
CHECK(secp256k1_ecdsa_signature_serialize_der(data.ctx, data.sig, &data.siglen, &sig));
|
|
CHECK(secp256k1_ec_pubkey_create(data.ctx, &pubkey, data.key));
|
|
data.pubkeylen = 33;
|
|
CHECK(secp256k1_ec_pubkey_serialize(data.ctx, data.pubkey, &data.pubkeylen, &pubkey, SECP256K1_EC_COMPRESSED) == 1);
|
|
|
|
print_output_table_header_row();
|
|
if (d || have_flag(argc, argv, "ecdsa") || have_flag(argc, argv, "verify") || have_flag(argc, argv, "ecdsa_verify")) run_benchmark("ecdsa_verify", bench_verify, NULL, NULL, &data, 10, iters);
|
|
|
|
secp256k1_context_destroy(data.ctx);
|
|
|
|
/* ECDSA signing benchmark */
|
|
data.ctx = secp256k1_context_create(SECP256K1_CONTEXT_SIGN);
|
|
|
|
if (d || have_flag(argc, argv, "ecdsa") || have_flag(argc, argv, "sign") || have_flag(argc, argv, "ecdsa_sign")) run_benchmark("ecdsa_sign", bench_sign_run, bench_sign_setup, NULL, &data, 10, iters);
|
|
|
|
secp256k1_context_destroy(data.ctx);
|
|
|
|
#ifdef ENABLE_MODULE_ECDH
|
|
/* ECDH benchmarks */
|
|
run_ecdh_bench(iters, argc, argv);
|
|
#endif
|
|
|
|
#ifdef ENABLE_MODULE_RECOVERY
|
|
/* ECDSA recovery benchmarks */
|
|
run_recovery_bench(iters, argc, argv);
|
|
#endif
|
|
|
|
#ifdef ENABLE_MODULE_SCHNORRSIG
|
|
/* Schnorr signature benchmarks */
|
|
run_schnorrsig_bench(iters, argc, argv);
|
|
#endif
|
|
|
|
return 0;
|
|
}
|