bench_impl.h (2432B)
1 /*********************************************************************** 2 * Copyright (c) 2014-2015 Pieter Wuille * 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 #ifndef SECP256K1_MODULE_RECOVERY_BENCH_H 8 #define SECP256K1_MODULE_RECOVERY_BENCH_H 9 10 #include "../../../include/secp256k1_recovery.h" 11 12 typedef struct { 13 haskellsecp256k1_v0_1_0_context *ctx; 14 unsigned char msg[32]; 15 unsigned char sig[64]; 16 } bench_recover_data; 17 18 static void bench_recover(void* arg, int iters) { 19 int i; 20 bench_recover_data *data = (bench_recover_data*)arg; 21 haskellsecp256k1_v0_1_0_pubkey pubkey; 22 unsigned char pubkeyc[33]; 23 24 for (i = 0; i < iters; i++) { 25 int j; 26 size_t pubkeylen = 33; 27 haskellsecp256k1_v0_1_0_ecdsa_recoverable_signature sig; 28 CHECK(haskellsecp256k1_v0_1_0_ecdsa_recoverable_signature_parse_compact(data->ctx, &sig, data->sig, i % 2)); 29 CHECK(haskellsecp256k1_v0_1_0_ecdsa_recover(data->ctx, &pubkey, &sig, data->msg)); 30 CHECK(haskellsecp256k1_v0_1_0_ec_pubkey_serialize(data->ctx, pubkeyc, &pubkeylen, &pubkey, SECP256K1_EC_COMPRESSED)); 31 for (j = 0; j < 32; j++) { 32 data->sig[j + 32] = data->msg[j]; /* Move former message to S. */ 33 data->msg[j] = data->sig[j]; /* Move former R to message. */ 34 data->sig[j] = pubkeyc[j + 1]; /* Move recovered pubkey X coordinate to R (which must be a valid X coordinate). */ 35 } 36 } 37 } 38 39 static void bench_recover_setup(void* arg) { 40 int i; 41 bench_recover_data *data = (bench_recover_data*)arg; 42 43 for (i = 0; i < 32; i++) { 44 data->msg[i] = 1 + i; 45 } 46 for (i = 0; i < 64; i++) { 47 data->sig[i] = 65 + i; 48 } 49 } 50 51 static void run_recovery_bench(int iters, int argc, char** argv) { 52 bench_recover_data data; 53 int d = argc == 1; 54 55 data.ctx = haskellsecp256k1_v0_1_0_context_create(SECP256K1_CONTEXT_NONE); 56 57 if (d || have_flag(argc, argv, "ecdsa") || have_flag(argc, argv, "recover") || have_flag(argc, argv, "ecdsa_recover")) run_benchmark("ecdsa_recover", bench_recover, bench_recover_setup, NULL, &data, 10, iters); 58 59 haskellsecp256k1_v0_1_0_context_destroy(data.ctx); 60 } 61 62 #endif /* SECP256K1_MODULE_RECOVERY_BENCH_H */