nettle-benchmark.c 6.58 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
/* nettle-benchmark.c
 *
 * Tries the performance of the various algorithms.
 *
 */
 
/* nettle, low-level cryptographics library
 *
 * Copyright (C) 2001 Niels Mller
 *  
 * The nettle library is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or (at your
 * option) any later version.
 * 
 * The nettle library is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
 * License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with the nettle library; see the file COPYING.LIB.  If not, write to
 * the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
 * MA 02111-1307, USA.
 */

27
28
#if HAVE_CONFIG_H
# include "config.h"
29
30
31
32
33
34
35
36
37
#endif

#include <assert.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <time.h>
38

39
#include "aes.h"
Niels Möller's avatar
Niels Möller committed
40
41
42
43
44
45
46
#include "arcfour.h"
#include "blowfish.h"
#include "cast128.h"
#include "des.h"
#include "serpent.h"
#include "twofish.h"

47
48
49
#include "nettle-meta.h"
#include "nettle-internal.h"

50
51
#include "cbc.h"

Niels Möller's avatar
Niels Möller committed
52

53
54
55
/* Process BENCH_BLOCK bytes at a time, for BENCH_INTERVAL clocks. */
#define BENCH_BLOCK 10240
#define BENCH_INTERVAL (CLOCKS_PER_SEC / 4)
56

Niels Möller's avatar
Niels Möller committed
57
58
59
/* Total MB:s, for MB/s figures. */
#define BENCH_TOTAL 10.0

60
/* Returns second per function call */
61
62
63
64
65
static double
time_function(void (*f)(void *arg), void *arg)
{
  clock_t before;
  clock_t after;
66
67
68
  clock_t done;
  unsigned ncalls;
  
69
  before = clock();
70
71
  done = before + BENCH_INTERVAL;
  ncalls = 0;
72
  
73
74
75
76
77
78
79
  do 
    {
      f(arg);
      after = clock();
      ncalls++;
    }
  while (after < done);
80
  
81
  return ((double)(after - before)) / CLOCKS_PER_SEC / ncalls;
82
83
}

84
85
86
87
88
89
90
91
92
93
94
95
96
97
struct bench_hash_info
{
  void *ctx;
  nettle_hash_update_func update;
  const uint8_t *data;
};

static void
bench_hash(void *arg)
{
  struct bench_hash_info *info = arg;
  info->update(info->ctx, BENCH_BLOCK, info->data);
}

98
99
100
struct bench_cipher_info
{
  void *ctx;
101
  nettle_crypt_func crypt;
102
103
104
105
106
107
108
  uint8_t *data;
};

static void
bench_cipher(void *arg)
{
  struct bench_cipher_info *info = arg;
109
  info->crypt(info->ctx, BENCH_BLOCK, info->data, info->data);
110
111
112
113
114
}

struct bench_cbc_info
{
  void *ctx;
115
116
  nettle_crypt_func crypt;
 
117
  uint8_t *data;
118
  
119
120
121
122
123
124
125
126
  unsigned block_size;
  uint8_t *iv;
};

static void
bench_cbc_encrypt(void *arg)
{
  struct bench_cbc_info *info = arg;
127
128
129
  cbc_encrypt(info->ctx, info->crypt,
	      info->block_size, info->iv,
	      BENCH_BLOCK, info->data, info->data);
130
131
132
133
134
135
}

static void
bench_cbc_decrypt(void *arg)
{
  struct bench_cbc_info *info = arg;
136
137
138
  cbc_decrypt(info->ctx, info->crypt,
	      info->block_size, info->iv,
	      BENCH_BLOCK, info->data, info->data);
139
140
141
142
143
144
145
}

/* Set data[i] = floor(sqrt(i)) */
static void
init_data(uint8_t *data)
{
  unsigned i,j;
Niels Möller's avatar
Niels Möller committed
146
  for (i = j = 0; i<BENCH_BLOCK;  i++)
147
148
149
150
151
152
153
154
    {
      if (j*j < i)
	j++;
      data[i] = j;
    }
}

static void
Niels Möller's avatar
Niels Möller committed
155
156
init_key(unsigned length,
         uint8_t *key)
157
{
Niels Möller's avatar
Niels Möller committed
158
159
160
  unsigned i;
  for (i = 0; i<length; i++)
    key[i] = i;
161
162
}

Niels Möller's avatar
Niels Möller committed
163
164
static void
display(const char *name, const char *mode,
165
	double speed)
Niels Möller's avatar
Niels Möller committed
166
{
167
  printf("%13s (%s): %.3fMB/s\n",
Niels Möller's avatar
Niels Möller committed
168
	 name, mode,
169
	 1 / (speed * 1048576.0 / BENCH_BLOCK));
Niels Möller's avatar
Niels Möller committed
170
171
}

172
173
174
175
176
177
178
179
180
181
182
183
184
static void *
xalloc(size_t size)
{
  void *p = malloc(size);
  if (!p)
    {
      fprintf(stderr, "Virtual memory exhausted.\n");
      abort();
    }

  return p;
}

185
186
187
188
189
static void
time_hash(const struct nettle_hash *hash)
{
  static uint8_t data[BENCH_BLOCK];
  struct bench_hash_info info;
190
  info.ctx = xalloc(hash->context_size); 
191
192
193
194
195
196
197
198
  info.update = hash->update;
  info.data = data;

  init_data(data);
  hash->init(info.ctx);

  display(hash->name, "Update",
	  time_function(bench_hash, &info));
199
200

  free(info.ctx);
201
202
}

Niels Möller's avatar
Niels Möller committed
203
static void
204
time_cipher(const struct nettle_cipher *cipher)
Niels Möller's avatar
Niels Möller committed
205
{
206
207
  void *ctx = xalloc(cipher->context_size);
  uint8_t *key = xalloc(cipher->key_size);
Niels Möller's avatar
Niels Möller committed
208

209
  static uint8_t data[BENCH_BLOCK];
Niels Möller's avatar
Niels Möller committed
210
211
212
213

  printf("\n");
  
  init_data(data);
214
215

  {
Niels Möller's avatar
Niels Möller committed
216
217
218
219
220
    /* Decent initializers are a GNU extension, so don't use it here. */
    struct bench_cipher_info info;
    info.ctx = ctx;
    info.crypt = cipher->encrypt;
    info.data = data;
221
    
Niels Möller's avatar
Niels Möller committed
222
    init_key(cipher->key_size, key);
223
    cipher->set_encrypt_key(ctx, cipher->key_size, key);
Niels Möller's avatar
Niels Möller committed
224
225
226

    display(cipher->name, "ECB encrypt",
	    time_function(bench_cipher, &info));
227
  }
Niels Möller's avatar
Niels Möller committed
228
  
229
  {
Niels Möller's avatar
Niels Möller committed
230
231
232
233
    struct bench_cipher_info info;
    info.ctx = ctx;
    info.crypt = cipher->decrypt;
    info.data = data;
234
    
Niels Möller's avatar
Niels Möller committed
235
    init_key(cipher->key_size, key);
236
    cipher->set_decrypt_key(ctx, cipher->key_size, key);
Niels Möller's avatar
Niels Möller committed
237
238
239

    display(cipher->name, "ECB decrypt",
	    time_function(bench_cipher, &info));
240
241
  }

Niels Möller's avatar
Niels Möller committed
242
243
  if (cipher->block_size)
    {
244
      uint8_t *iv = xalloc(cipher->block_size);
Niels Möller's avatar
Niels Möller committed
245
246
247
      
      /* Do CBC mode */
      {
Niels Möller's avatar
Niels Möller committed
248
249
250
251
252
253
        struct bench_cbc_info info;
	info.ctx = ctx;
	info.crypt = cipher->encrypt;
	info.data = data;
	info.block_size = cipher->block_size;
	info.iv = iv;
254
    
Niels Möller's avatar
Niels Möller committed
255
        memset(iv, 0, sizeof(iv));
256
    
257
        cipher->set_encrypt_key(ctx, cipher->key_size, key);
258

Niels Möller's avatar
Niels Möller committed
259
260
	display(cipher->name, "CBC encrypt",
		time_function(bench_cbc_encrypt, &info));
Niels Möller's avatar
Niels Möller committed
261
      }
262

Niels Möller's avatar
Niels Möller committed
263
      {
Niels Möller's avatar
Niels Möller committed
264
265
266
267
268
269
        struct bench_cbc_info info;
	info.ctx = ctx;
	info.crypt = cipher->decrypt;
	info.data = data;
	info.block_size = cipher->block_size;
	info.iv = iv;
270
    
Niels Möller's avatar
Niels Möller committed
271
        memset(iv, 0, sizeof(iv));
272

273
        cipher->set_decrypt_key(ctx, cipher->key_size, key);
274

Niels Möller's avatar
Niels Möller committed
275
276
	display(cipher->name, "CBC decrypt",
		time_function(bench_cbc_decrypt, &info));
Niels Möller's avatar
Niels Möller committed
277
      }
278
      free(iv);
Niels Möller's avatar
Niels Möller committed
279
    }
280
281
  free(ctx);
  free(key);
Niels Möller's avatar
Niels Möller committed
282
283
}

Niels Möller's avatar
Niels Möller committed
284
#if WITH_OPENSSL
285
286
287
288
# define OPENSSL(x) x,
#else
# define OPENSSL(x)
#endif
Niels Möller's avatar
Niels Möller committed
289
290

int
Niels Möller's avatar
Niels Möller committed
291
main(int argc UNUSED, char **argv UNUSED)
Niels Möller's avatar
Niels Möller committed
292
293
{
  unsigned i;
294
295
296
297
298
299
300
301

  const struct nettle_hash *hashes[] =
    {
      &nettle_md2, &nettle_md4, &nettle_md5,
      &nettle_sha1, &nettle_sha256,
      NULL
    };

302
  const struct nettle_cipher *ciphers[] =
Niels Möller's avatar
Niels Möller committed
303
    {
304
      &nettle_aes128, &nettle_aes192, &nettle_aes256,
Niels Möller's avatar
Niels Möller committed
305
306
307
308
309
      OPENSSL(&nettle_openssl_aes128)
      OPENSSL(&nettle_openssl_aes192)
      OPENSSL(&nettle_openssl_aes256)
      &nettle_arcfour128, OPENSSL(&nettle_openssl_arcfour128)
      &nettle_blowfish128, OPENSSL(&nettle_openssl_blowfish128)
310
311
312
      &nettle_cast128, OPENSSL(&nettle_openssl_cast128)
      &nettle_des, OPENSSL(&nettle_openssl_des)
      &nettle_des3,
313
314
      &nettle_serpent256,
      &nettle_twofish128, &nettle_twofish192, &nettle_twofish256,
315
      NULL
Niels Möller's avatar
Niels Möller committed
316
    };
317

318
319
320
  for (i = 0; hashes[i]; i++)
    time_hash(hashes[i]);
  
321
  for (i = 0; ciphers[i]; i++)
322
    time_cipher(ciphers[i]);
Niels Möller's avatar
Niels Möller committed
323
  
324
325
  return 0;
}