src/bench_verify.c

   1 /**********************************************************************
   2  * Copyright (c) 2014 Pieter Wuille                                   *
   3  * Distributed under the MIT software license, see the accompanying   *
   4  * file COPYING or http://www.opensource.org/licenses/mit-license.php.*
   5  **********************************************************************/
   6
   7 #include <stdio.h>
   8 #include <string.h>
   9
  10 #include "include/secp256k1.h"
  11 #include "util.h"
  12 #include "bench.h"
  13
  14 #ifdef ENABLE_OPENSSL_TESTS
  15 #include <openssl/bn.h>
  16 #include <openssl/ecdsa.h>
  17 #include <openssl/obj_mac.h>
  18 #endif
  19
  20
  21 typedef struct {
  22     secp256k1_context *ctx;
  23     unsigned char msg[32];
  24     unsigned char key[32];
  25     unsigned char sig[72];
  26     size_t siglen;
  27     unsigned char pubkey[33];
  28     size_t pubkeylen;
  29 #ifdef ENABLE_OPENSSL_TESTS
  30     EC_GROUP* ec_group;
  31 #endif
  32 } benchmark_verify_t;
  33
  34 static void benchmark_verify(void* arg, int iters) {
  35     int i;
  36     benchmark_verify_t* data = (benchmark_verify_t*)arg;
  37
  38     for (i = 0; i < iters; i++) {
  39         secp256k1_pubkey pubkey;
  40         secp256k1_ecdsa_signature sig;
  41         data->sig[data->siglen - 1] ^= (i & 0xFF);
  42         data->sig[data->siglen - 2] ^= ((i >> 8) & 0xFF);
  43         data->sig[data->siglen - 3] ^= ((i >> 16) & 0xFF);
  44         CHECK(secp256k1_ec_pubkey_parse(data->ctx, &pubkey, data->pubkey, data->pubkeylen) == 1);
  45         CHECK(secp256k1_ecdsa_signature_parse_der(data->ctx, &sig, data->sig, data->siglen) == 1);
  46         CHECK(secp256k1_ecdsa_verify(data->ctx, &sig, data->msg, &pubkey) == (i == 0));
  47         data->sig[data->siglen - 1] ^= (i & 0xFF);
  48         data->sig[data->siglen - 2] ^= ((i >> 8) & 0xFF);
  49         data->sig[data->siglen - 3] ^= ((i >> 16) & 0xFF);
  50     }
  51 }
  52
  53 #ifdef ENABLE_OPENSSL_TESTS
  54 static void benchmark_verify_openssl(void* arg, int iters) {
  55     int i;
  56     benchmark_verify_t* data = (benchmark_verify_t*)arg;
  57
  58     for (i = 0; i < iters; i++) {
  59         data->sig[data->siglen - 1] ^= (i & 0xFF);
  60         data->sig[data->siglen - 2] ^= ((i >> 8) & 0xFF);
  61         data->sig[data->siglen - 3] ^= ((i >> 16) & 0xFF);
  62         {
  63             EC_KEY *pkey = EC_KEY_new();
  64             const unsigned char *pubkey = &data->pubkey[0];
  65             int result;
  66
  67             CHECK(pkey != NULL);
  68             result = EC_KEY_set_group(pkey, data->ec_group);
  69             CHECK(result);
  70             result = (o2i_ECPublicKey(&pkey, &pubkey, data->pubkeylen)) != NULL;
  71             CHECK(result);
  72             result = ECDSA_verify(0, &data->msg[0], sizeof(data->msg), &data->sig[0], data->siglen, pkey) == (i == 0);
  73             CHECK(result);
  74             EC_KEY_free(pkey);
  75         }
  76         data->sig[data->siglen - 1] ^= (i & 0xFF);
  77         data->sig[data->siglen - 2] ^= ((i >> 8) & 0xFF);
  78         data->sig[data->siglen - 3] ^= ((i >> 16) & 0xFF);
  79     }
  80 }
  81 #endif
  82
  83 int main(void) {
  84     int i;
  85     secp256k1_pubkey pubkey;
  86     secp256k1_ecdsa_signature sig;
  87     benchmark_verify_t data;
  88
  89     int iters = get_iters(20000);
  90
  91     data.ctx = secp256k1_context_create(SECP256K1_CONTEXT_SIGN | SECP256K1_CONTEXT_VERIFY);
  92
  93     for (i = 0; i < 32; i++) {
  94         data.msg[i] = 1 + i;
  95     }
  96     for (i = 0; i < 32; i++) {
  97         data.key[i] = 33 + i;
  98     }
  99     data.siglen = 72;
 100     CHECK(secp256k1_ecdsa_sign(data.ctx, &sig, data.msg, data.key, NULL, NULL));
 101     CHECK(secp256k1_ecdsa_signature_serialize_der(data.ctx, data.sig, &data.siglen, &sig));
 102     CHECK(secp256k1_ec_pubkey_create(data.ctx, &pubkey, data.key));
 103     data.pubkeylen = 33;
 104     CHECK(secp256k1_ec_pubkey_serialize(data.ctx, data.pubkey, &data.pubkeylen, &pubkey, SECP256K1_EC_COMPRESSED) == 1);
 105
 106     run_benchmark("ecdsa_verify", benchmark_verify, NULL, NULL, &data, 10, iters);
 107 #ifdef ENABLE_OPENSSL_TESTS
 108     data.ec_group = EC_GROUP_new_by_curve_name(NID_secp256k1);
 109     run_benchmark("ecdsa_verify_openssl", benchmark_verify_openssl, NULL, NULL, &data, 10, iters);
 110     EC_GROUP_free(data.ec_group);
 111 #endif
 112
 113     secp256k1_context_destroy(data.ctx);
 114     return 0;
 115 }