github.com/ethereum/go-ethereum@v1.16.1/crypto/secp256k1/libsecp256k1/src/modules/recovery/bench_impl.h (about)

     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      secp256k1_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      secp256k1_pubkey pubkey;
    22      unsigned char pubkeyc[33];
    23  
    24      for (i = 0; i < iters; i++) {
    25          int j;
    26          size_t pubkeylen = 33;
    27          secp256k1_ecdsa_recoverable_signature sig;
    28          CHECK(secp256k1_ecdsa_recoverable_signature_parse_compact(data->ctx, &sig, data->sig, i % 2));
    29          CHECK(secp256k1_ecdsa_recover(data->ctx, &pubkey, &sig, data->msg));
    30          CHECK(secp256k1_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 = secp256k1_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      secp256k1_context_destroy(data.ctx);
    60  }
    61  
    62  #endif /* SECP256K1_MODULE_RECOVERY_BENCH_H */