testutils.c 46.3 KB
Newer Older
Niels Möller's avatar
Niels Möller committed
1
/* testutils.c */
Niels Möller's avatar
Niels Möller committed
2
3
4

#include "testutils.h"

5
#include "base16.h"
6
#include "cbc.h"
Dmitry Baryshkov's avatar
Dmitry Baryshkov committed
7
#include "cfb.h"
8
#include "ctr.h"
Niels Möller's avatar
Niels Möller committed
9
#include "knuth-lfib.h"
10
#include "macros.h"
11
#include "nettle-internal.h"
12

Niels Möller's avatar
Niels Möller committed
13
#include <assert.h>
Niels Möller's avatar
Niels Möller committed
14
15
#include <ctype.h>

16
17
18
19
20
21
22
23
24
25
26
void
die(const char *format, ...)
{
  va_list args;
  va_start(args, format);
  vfprintf(stderr, format, args);
  va_end(args);

  abort ();
}

27
28
29
30
void *
xalloc(size_t size)
{
  void *p = malloc(size);
31
  if (size && !p)
32
33
34
35
36
37
38
39
    {
      fprintf(stderr, "Virtual memory exhausted.\n");
      abort();
    }

  return p;
}

40
41
42
static struct tstring *tstring_first = NULL;

struct tstring *
Niels Möller's avatar
Niels Möller committed
43
tstring_alloc (size_t length)
44
{
Niels Möller's avatar
Niels Möller committed
45
  struct tstring *s = xalloc(sizeof(struct tstring) + length);
46
47
  s->length = length;
  s->next = tstring_first;
Niels Möller's avatar
Niels Möller committed
48
49
  /* NUL-terminate, for convenience. */
  s->data[length] = '\0';
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
  tstring_first = s;
  return s;
}

void
tstring_clear(void)
{
  while (tstring_first)
    {
      struct tstring *s = tstring_first;
      tstring_first = s->next;
      free(s);
    }
}

struct tstring *
66
tstring_data(size_t length, const uint8_t *data)
67
68
69
70
71
72
73
74
{
  struct tstring *s = tstring_alloc (length);
  memcpy (s->data, data, length);
  return s;
}

struct tstring *
tstring_hex(const char *hex)
Niels Möller's avatar
Niels Möller committed
75
{
76
  struct base16_decode_ctx ctx;
77
  struct tstring *s;
78
  size_t length = strlen(hex);
Niels Möller's avatar
Niels Möller committed
79

80
81
82
83
84
  s = tstring_alloc(BASE16_DECODE_LENGTH (length));
  base16_decode_init (&ctx);
  ASSERT (base16_decode_update (&ctx, &s->length, s->data,
				length, hex));
  ASSERT (base16_decode_final (&ctx));
Niels Möller's avatar
Niels Möller committed
85

86
87
88
89
90
91
92
  return s;
}

void
tstring_print_hex(const struct tstring *s)
{
  print_hex (s->length, s->data);
Niels Möller's avatar
Niels Möller committed
93
}
94

95
void
Niels Möller's avatar
Niels Möller committed
96
print_hex(size_t length, const uint8_t *data)
97
{
Niels Möller's avatar
Niels Möller committed
98
  size_t i;
99
100
101
  
  for (i = 0; i < length; i++)
    {
Niels Möller's avatar
Niels Möller committed
102
103
104
105
106
107
108
109
110
111
112
      switch (i % 16)
	{
	default:
	  break;
	case 0:
	  printf("\n");
	  break;
	case 8:
	  printf(" ");
	  break;
	}
113
114
      printf("%02x", data[i]);
    }
Niels Möller's avatar
Niels Möller committed
115
  printf("\n");
116
117
}

Niels Möller's avatar
Niels Möller committed
118
119
120
121
122
int verbose = 0;

int
main(int argc, char **argv)
{
123
124
  if (argc > 1)
    {
125
      if (argc == 2 && !strcmp(argv[1], "-v"))
Niels Möller's avatar
Niels Möller committed
126
	verbose = 1;
127
128
129
130
131
132
133
      else
	{
	  fprintf(stderr, "Invalid argument `%s', only accepted option is `-v'.\n",
		  argv[1]);
	  return 1;
	}
    }
Niels Möller's avatar
Niels Möller committed
134

135
136
137
138
  test_main();

  tstring_clear();
  return EXIT_SUCCESS;
Niels Möller's avatar
Niels Möller committed
139
140
}

141
142
void
test_cipher(const struct nettle_cipher *cipher,
143
144
145
	    const struct tstring *key,
	    const struct tstring *cleartext,
	    const struct tstring *ciphertext)
146
{
147
  void *ctx = xalloc(cipher->context_size);
148
  uint8_t *data = xalloc(cleartext->length);
Niels Möller's avatar
Niels Möller committed
149
  size_t length;
150
151
  ASSERT (cleartext->length == ciphertext->length);
  length = cleartext->length;
152

153
154
  ASSERT (key->length == cipher->key_size);
  cipher->set_encrypt_key(ctx, key->data);
155
  cipher->encrypt(ctx, length, data, cleartext->data);
156

157
  if (!MEMEQ(length, data, ciphertext->data))
158
159
    {
      fprintf(stderr, "Encrypt failed:\nInput:");
160
      tstring_print_hex(cleartext);
161
162
163
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
164
      tstring_print_hex(ciphertext);
165
166
167
      fprintf(stderr, "\n");
      FAIL();
    }
168
  cipher->set_decrypt_key(ctx, key->data);
169
170
  cipher->decrypt(ctx, length, data, data);

171
  if (!MEMEQ(length, data, cleartext->data))
172
173
    {
      fprintf(stderr, "Decrypt failed:\nInput:");
174
      tstring_print_hex(ciphertext);
175
176
177
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
178
      tstring_print_hex(cleartext);
179
180
181
      fprintf(stderr, "\n");
      FAIL();
    }
182
183
184

  free(ctx);
  free(data);
185
186
187
188
}

void
test_cipher_cbc(const struct nettle_cipher *cipher,
189
190
191
192
		const struct tstring *key,
		const struct tstring *cleartext,
		const struct tstring *ciphertext,
		const struct tstring *iiv)
193
{
194
  void *ctx = xalloc(cipher->context_size);
195
  uint8_t *data;
196
  uint8_t *iv = xalloc(cipher->block_size);
Niels Möller's avatar
Niels Möller committed
197
  size_t length;
198
199
200
201

  ASSERT (cleartext->length == ciphertext->length);
  length = cleartext->length;

202
  ASSERT (key->length == cipher->key_size);
203
204
205
  ASSERT (iiv->length == cipher->block_size);

  data = xalloc(length);  
206
  cipher->set_encrypt_key(ctx, key->data);
207
  memcpy(iv, iiv->data, cipher->block_size);
208
209
210

  cbc_encrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
211
	      length, data, cleartext->data);
212

213
  if (!MEMEQ(length, data, ciphertext->data))
214
215
    {
      fprintf(stderr, "CBC encrypt failed:\nInput:");
216
      tstring_print_hex(cleartext);
217
218
219
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
220
      tstring_print_hex(ciphertext);
221
222
223
      fprintf(stderr, "\n");
      FAIL();
    }
224
  cipher->set_decrypt_key(ctx, key->data);
225
  memcpy(iv, iiv->data, cipher->block_size);
226
227
228
229
230

  cbc_decrypt(ctx, cipher->decrypt,
	      cipher->block_size, iv,
	      length, data, data);

231
  if (!MEMEQ(length, data, cleartext->data))
232
233
    {
      fprintf(stderr, "CBC decrypt failed:\nInput:");
234
      tstring_print_hex(ciphertext);
235
236
237
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
238
      tstring_print_hex(cleartext);
239
240
241
      fprintf(stderr, "\n");
      FAIL();
    }
242

243
244
245
  free(ctx);
  free(data);
  free(iv);
246
247
}

Dmitry Baryshkov's avatar
Dmitry Baryshkov committed
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
void
test_cipher_cfb(const struct nettle_cipher *cipher,
		const struct tstring *key,
		const struct tstring *cleartext,
		const struct tstring *ciphertext,
		const struct tstring *iiv)
{
  void *ctx = xalloc(cipher->context_size);
  uint8_t *data, *data2;
  uint8_t *iv = xalloc(cipher->block_size);
  size_t length;

  ASSERT (cleartext->length == ciphertext->length);
  length = cleartext->length;

  ASSERT (key->length == cipher->key_size);
  ASSERT (iiv->length == cipher->block_size);

  data = xalloc(length);
  data2 = xalloc(length);

  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);

  cfb_encrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data, cleartext->data);

  if (!MEMEQ(length, data, ciphertext->data))
    {
      fprintf(stderr, "CFB encrypt failed:\nInput:");
      tstring_print_hex(cleartext);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
      tstring_print_hex(ciphertext);
      fprintf(stderr, "\n");
      FAIL();
    }
  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);

  cfb_decrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data2, data);

  if (!MEMEQ(length, data2, cleartext->data))
    {
      fprintf(stderr, "CFB decrypt failed:\nInput:");
      tstring_print_hex(ciphertext);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data2);
      fprintf(stderr, "\nExpected:");
      tstring_print_hex(cleartext);
      fprintf(stderr, "\n");
      FAIL();
    }
  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);
  memcpy(data, cleartext->data, length);

  cfb_encrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data, data);

  if (!MEMEQ(length, data, ciphertext->data))
    {
      fprintf(stderr, "CFB inplace encrypt failed:\nInput:");
      tstring_print_hex(cleartext);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
      tstring_print_hex(ciphertext);
      fprintf(stderr, "\n");
      FAIL();
    }
  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);

  cfb_decrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data, data);

  if (!MEMEQ(length, data, cleartext->data))
    {
      fprintf(stderr, "CFB inplace decrypt failed:\nInput:");
      tstring_print_hex(ciphertext);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
      tstring_print_hex(cleartext);
      fprintf(stderr, "\n");
      FAIL();
    }

  /* Repeat all tests with incomplete last block */
  length -= 1;

  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);

  cfb_encrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data, cleartext->data);

  if (!MEMEQ(length, data, ciphertext->data))
    {
      fprintf(stderr, "CFB encrypt failed:\nInput:");
      print_hex(length, cleartext->data);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
      print_hex(length, ciphertext->data);
      fprintf(stderr, "\n");
      FAIL();
    }
  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);

  cfb_decrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data2, data);

  if (!MEMEQ(length, data2, cleartext->data))
    {
      fprintf(stderr, "CFB decrypt failed:\nInput:");
      print_hex(length, ciphertext->data);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data2);
      fprintf(stderr, "\nExpected:");
      print_hex(length, cleartext->data);
      fprintf(stderr, "\n");
      FAIL();
    }
  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);
  memcpy(data, cleartext->data, length);

  cfb_encrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data, data);

  if (!MEMEQ(length, data, ciphertext->data))
    {
      fprintf(stderr, "CFB inplace encrypt failed:\nInput:");
      print_hex(length, cleartext->data);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
      print_hex(length, ciphertext->data);
      fprintf(stderr, "\n");
      FAIL();
    }
  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);

  cfb_decrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data, data);

  if (!MEMEQ(length, data, cleartext->data))
    {
      fprintf(stderr, "CFB inplace decrypt failed:\nInput:");
      print_hex(length, ciphertext->data);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
      print_hex(length, cleartext->data);
      fprintf(stderr, "\n");
      FAIL();
    }

  free(ctx);
  free(data);
  free(data2);
  free(iv);
}

426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
void
test_cipher_cfb8(const struct nettle_cipher *cipher,
		 const struct tstring *key,
		 const struct tstring *cleartext,
		 const struct tstring *ciphertext,
		 const struct tstring *iiv)
{
  void *ctx = xalloc(cipher->context_size);
  uint8_t *data, *data2;
  uint8_t *iv = xalloc(cipher->block_size);
  size_t length;

  ASSERT (cleartext->length == ciphertext->length);
  length = cleartext->length;

  ASSERT (key->length == cipher->key_size);
  ASSERT (iiv->length == cipher->block_size);

  data = xalloc(length);
  data2 = xalloc(length);

  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);

  cfb8_encrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data, cleartext->data);

  if (!MEMEQ(length, data, ciphertext->data))
    {
      fprintf(stderr, "CFB8 encrypt failed:\nInput:");
      tstring_print_hex(cleartext);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
      tstring_print_hex(ciphertext);
      fprintf(stderr, "\n");
      FAIL();
    }
  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);

  cfb8_decrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data2, data);

  if (!MEMEQ(length, data2, cleartext->data))
    {
      fprintf(stderr, "CFB8 decrypt failed:\nInput:");
      tstring_print_hex(ciphertext);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data2);
      fprintf(stderr, "\nExpected:");
      tstring_print_hex(cleartext);
      fprintf(stderr, "\n");
      FAIL();
    }
  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);
  memcpy(data, cleartext->data, length);

  cfb8_encrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data, data);

  if (!MEMEQ(length, data, ciphertext->data))
    {
      fprintf(stderr, "CFB8 inplace encrypt failed:\nInput:");
      tstring_print_hex(cleartext);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
      tstring_print_hex(ciphertext);
      fprintf(stderr, "\n");
      FAIL();
    }
  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);

  cfb8_decrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data, data);

  if (!MEMEQ(length, data, cleartext->data))
    {
      fprintf(stderr, "CFB8 inplace decrypt failed:\nInput:");
      tstring_print_hex(ciphertext);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
      tstring_print_hex(cleartext);
      fprintf(stderr, "\n");
      FAIL();
    }

  /* Repeat all tests with incomplete last block */
  length -= 1;

  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);

  cfb8_encrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data, cleartext->data);

  if (!MEMEQ(length, data, ciphertext->data))
    {
      fprintf(stderr, "CFB8 encrypt failed:\nInput:");
      print_hex(length, cleartext->data);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
      print_hex(length, ciphertext->data);
      fprintf(stderr, "\n");
      FAIL();
    }
  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);

  cfb8_decrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data2, data);

  if (!MEMEQ(length, data2, cleartext->data))
    {
      fprintf(stderr, "CFB8 decrypt failed:\nInput:");
      print_hex(length, ciphertext->data);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data2);
      fprintf(stderr, "\nExpected:");
      print_hex(length, cleartext->data);
      fprintf(stderr, "\n");
      FAIL();
    }
  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);
  memcpy(data, cleartext->data, length);

  cfb8_encrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data, data);

  if (!MEMEQ(length, data, ciphertext->data))
    {
      fprintf(stderr, "CFB8 inplace encrypt failed:\nInput:");
      print_hex(length, cleartext->data);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
      print_hex(length, ciphertext->data);
      fprintf(stderr, "\n");
      FAIL();
    }
  cipher->set_encrypt_key(ctx, key->data);
  memcpy(iv, iiv->data, cipher->block_size);

  cfb8_decrypt(ctx, cipher->encrypt,
	      cipher->block_size, iv,
	      length, data, data);

  if (!MEMEQ(length, data, cleartext->data))
    {
      fprintf(stderr, "CFB8 inplace decrypt failed:\nInput:");
      print_hex(length, ciphertext->data);
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
      print_hex(length, cleartext->data);
      fprintf(stderr, "\n");
      FAIL();
    }

  free(ctx);
  free(data);
  free(data2);
  free(iv);
}

604
605
void
test_cipher_ctr(const struct nettle_cipher *cipher,
606
607
608
609
		const struct tstring *key,
		const struct tstring *cleartext,
		const struct tstring *ciphertext,
		const struct tstring *ictr)
610
611
{
  void *ctx = xalloc(cipher->context_size);
612
  uint8_t *data;
613
  uint8_t *ctr = xalloc(cipher->block_size);
614
  uint8_t *octr = xalloc(cipher->block_size);
Niels Möller's avatar
Niels Möller committed
615
616
  size_t length, nblocks;
  unsigned low;
Niels Möller's avatar
Niels Möller committed
617
  size_t i;
618
619
620
621

  ASSERT (cleartext->length == ciphertext->length);
  length = cleartext->length;

622
  ASSERT (key->length == cipher->key_size);
623
  ASSERT (ictr->length == cipher->block_size);
624
625
626
627
628
629
630
631
632
633
634
635

  /* Compute expected counter value after the operation. */
  nblocks = (length + cipher->block_size - 1) / cipher->block_size;
  ASSERT (nblocks < 0x100);

  memcpy (octr, ictr->data, cipher->block_size - 1);
  low = ictr->data[cipher->block_size - 1] + nblocks;
  octr[cipher->block_size - 1] = low;

  if (low >= 0x100)
    INCREMENT (cipher->block_size - 1, octr);

636
637
  data = xalloc(length);  

638
  cipher->set_encrypt_key(ctx, key->data);
639

Niels Möller's avatar
Niels Möller committed
640
  for (i = 0; i <= length; i++)
641
    {
Niels Möller's avatar
Niels Möller committed
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
      memcpy(ctr, ictr->data, cipher->block_size);
      memset(data, 17, length);

      ctr_crypt(ctx, cipher->encrypt,
		cipher->block_size, ctr,
		i, data, cleartext->data);

      if (!MEMEQ(i, data, ciphertext->data)
	  || (i < length && data[i] != 17))
	{
	  fprintf(stderr, "CTR encrypt failed (length %d of %d):\nInput:",
		  (int) i, (int) length);
	  tstring_print_hex(cleartext);
	  fprintf(stderr, "\nOutput: ");
	  print_hex(length, data);
	  fprintf(stderr, "\nExpected:");
	  tstring_print_hex(ciphertext);
	  fprintf(stderr, "\n");
	  FAIL();
	}
662
    }
663

664
665
  ASSERT (MEMEQ (cipher->block_size, ctr, octr));

666
  memcpy(ctr, ictr->data, cipher->block_size);
667
668
669
670
671

  ctr_crypt(ctx, cipher->encrypt,
	    cipher->block_size, ctr,
	    length, data, data);

672
  if (!MEMEQ(length, data, cleartext->data))
673
674
    {
      fprintf(stderr, "CTR decrypt failed:\nInput:");
675
      tstring_print_hex(ciphertext);
676
677
678
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
679
      tstring_print_hex(cleartext);
680
681
682
      fprintf(stderr, "\n");
      FAIL();
    }
683

684
685
  ASSERT (MEMEQ (cipher->block_size, ctr, octr));

686
687
  free(ctx);
  free(data);
688
  free(octr);
689
690
691
  free(ctr);
}

692
#if 0
693
694
void
test_cipher_stream(const struct nettle_cipher *cipher,
695
696
697
		   const struct tstring *key,
		   const struct tstring *cleartext,
		   const struct tstring *ciphertext)
698
{
Niels Möller's avatar
Niels Möller committed
699
  size_t block;
700
  
701
  void *ctx = xalloc(cipher->context_size);
702
  uint8_t *data;
Niels Möller's avatar
Niels Möller committed
703
  size_t length;
704
705
706
707
708
709

  ASSERT (cleartext->length == ciphertext->length);
  length = cleartext->length;

  data = xalloc(length + 1);

710
711
  for (block = 1; block <= length; block++)
    {
Niels Möller's avatar
Niels Möller committed
712
      size_t i;
713
714

      memset(data, 0x17, length + 1);
715
      cipher->set_encrypt_key(ctx, key->length, key->data);
716
717
718

      for (i = 0; i + block < length; i += block)
	{
719
720
	  cipher->encrypt(ctx, block, data + i, cleartext->data + i);
	  ASSERT (data[i + block] == 0x17);
721
	}
722

723
724
      cipher->encrypt(ctx, length - i, data + i, cleartext->data + i);
      ASSERT (data[length] == 0x17);
725
      
726
      if (!MEMEQ(length, data, ciphertext->data))
727
	{
Niels Möller's avatar
Niels Möller committed
728
729
	  fprintf(stderr, "Encrypt failed, block size %lu\nInput:",
		  (unsigned long) block);
730
	  tstring_print_hex(cleartext);
731
732
733
	  fprintf(stderr, "\nOutput: ");
	  print_hex(length, data);
	  fprintf(stderr, "\nExpected:");
734
	  tstring_print_hex(ciphertext);
735
736
737
	  fprintf(stderr, "\n");
	  FAIL();	    
	}
738
739
    }
  
740
  cipher->set_decrypt_key(ctx, key->length, key->data);
741
742
  cipher->decrypt(ctx, length, data, data);

743
  ASSERT (data[length] == 0x17);
744

745
  if (!MEMEQ(length, data, cleartext->data))
746
747
    {
      fprintf(stderr, "Decrypt failed\nInput:");
748
      tstring_print_hex(ciphertext);
749
750
751
      fprintf(stderr, "\nOutput: ");
      print_hex(length, data);
      fprintf(stderr, "\nExpected:");
752
      tstring_print_hex(cleartext);
753
754
755
      fprintf(stderr, "\n");
      FAIL();	    
    }
756
757
758

  free(ctx);
  free(data);
759
}
760
#endif
761

762
763
void
test_aead(const struct nettle_aead *aead,
764
	  nettle_hash_update_func *set_nonce,
765
766
767
768
	  const struct tstring *key,
	  const struct tstring *authtext,
	  const struct tstring *cleartext,
	  const struct tstring *ciphertext,
769
	  const struct tstring *nonce,
770
	  const struct tstring *digest)
771
772
{
  void *ctx = xalloc(aead->context_size);
773
  uint8_t *data;
774
  uint8_t *buffer = xalloc(aead->digest_size);
Niels Möller's avatar
Niels Möller committed
775
  size_t length;
776
777
778

  ASSERT (cleartext->length == ciphertext->length);
  length = cleartext->length;
779

780
  ASSERT (key->length == aead->key_size);
781
  ASSERT (digest->length <= aead->digest_size);
782
783
784

  data = xalloc(length);
  
785
  /* encryption */
786
  memset(buffer, 0, aead->digest_size);
787
  aead->set_encrypt_key(ctx, key->data);
788

789
790
791
792
793
794
795
  if (nonce->length != aead->nonce_size)
    {
      ASSERT (set_nonce);
      set_nonce (ctx, nonce->length, nonce->data);
    }
  else
    aead->set_nonce(ctx, nonce->data);
796

797
798
  if (authtext->length)
    aead->update(ctx, authtext->length, authtext->data);
799
800
    
  if (length)
801
    aead->encrypt(ctx, length, data, cleartext->data);
802

803
  aead->digest(ctx, digest->length, buffer);
804

805
  ASSERT(MEMEQ(length, data, ciphertext->data));
806
  ASSERT(MEMEQ(digest->length, buffer, digest->data));
807
808

  /* decryption */
809
  memset(buffer, 0, aead->digest_size);
810
811
812
813
814
815
816
817
818
819

  aead->set_decrypt_key(ctx, key->data);

  if (nonce->length != aead->nonce_size)
    {
      ASSERT (set_nonce);
      set_nonce (ctx, nonce->length, nonce->data);
    }
  else
    aead->set_nonce(ctx, nonce->data);
820

821
822
  if (authtext->length)
    aead->update(ctx, authtext->length, authtext->data);
823
824
825
826
    
  if (length)
    aead->decrypt(ctx, length, data, data);

827
  aead->digest(ctx, digest->length, buffer);
828

829
  ASSERT(MEMEQ(length, data, cleartext->data));
830
  ASSERT(MEMEQ(digest->length, buffer, digest->data));
831
832
833
834
835
836

  free(ctx);
  free(data);
  free(buffer);
}

837
838
void
test_hash(const struct nettle_hash *hash,
839
840
	  const struct tstring *msg,
	  const struct tstring *digest)
841
{
842
843
  void *ctx = xalloc(hash->context_size);
  uint8_t *buffer = xalloc(hash->digest_size);
844
845
  uint8_t *input;
  unsigned offset;
846

847
848
  ASSERT (digest->length == hash->digest_size);

849
  hash->init(ctx);
850
  hash->update(ctx, msg->length, msg->data);
851
852
  hash->digest(ctx, hash->digest_size, buffer);

853
854
855
856
857
858
859
860
  if (MEMEQ(hash->digest_size, digest->data, buffer) == 0)
    {
      fprintf(stdout, "\nGot:\n");
      print_hex(hash->digest_size, buffer);
      fprintf(stdout, "\nExpected:\n");
      print_hex(hash->digest_size, digest->data);
      abort();
    }
861
862
863

  memset(buffer, 0, hash->digest_size);

864
  hash->update(ctx, msg->length, msg->data);
865
866
  hash->digest(ctx, hash->digest_size - 1, buffer);

867
  ASSERT(MEMEQ(hash->digest_size - 1, digest->data, buffer));
868

869
  ASSERT(buffer[hash->digest_size - 1] == 0);
870

871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
  input = xalloc (msg->length + 16);
  for (offset = 0; offset < 16; offset++)
    {
      memset (input, 0, msg->length + 16);
      memcpy (input + offset, msg->data, msg->length);
      hash->update (ctx, msg->length, input + offset);
      hash->digest (ctx, hash->digest_size, buffer);
      if (MEMEQ(hash->digest_size, digest->data, buffer) == 0)
	{
	  fprintf(stdout, "hash input address: %p\nGot:\n", input + offset);
	  print_hex(hash->digest_size, buffer);
	  fprintf(stdout, "\nExpected:\n");
	  print_hex(hash->digest_size, digest->data);
	  abort();
	}      
    }
887
888
  free(ctx);
  free(buffer);
889
  free(input);
890
}
891

892
893
void
test_hash_large(const struct nettle_hash *hash,
Niels Möller's avatar
Niels Möller committed
894
		size_t count, size_t length,
895
		uint8_t c,
896
		const struct tstring *digest)
897
898
899
900
{
  void *ctx = xalloc(hash->context_size);
  uint8_t *buffer = xalloc(hash->digest_size);
  uint8_t *data = xalloc(length);
Niels Möller's avatar
Niels Möller committed
901
  size_t i;
902

903
904
  ASSERT (digest->length == hash->digest_size);

905
906
907
908
  memset(data, c, length);

  hash->init(ctx);
  for (i = 0; i < count; i++)
909
910
911
912
913
914
915
    {
      hash->update(ctx, length, data);
      if (i % (count / 50) == 0)
	fprintf (stderr, ".");
    }
  fprintf (stderr, "\n");
  
916
917
918
919
  hash->digest(ctx, hash->digest_size, buffer);

  print_hex(hash->digest_size, buffer);

920
  ASSERT (MEMEQ(hash->digest_size, digest->data, buffer));
921
922
923
924
925
926

  free(ctx);
  free(buffer);
  free(data);
}

Niels Möller's avatar
Niels Möller committed
927
928
void
test_armor(const struct nettle_armor *armor,
Niels Möller's avatar
Niels Möller committed
929
           size_t data_length,
Niels Möller's avatar
Niels Möller committed
930
           const uint8_t *data,
931
           const char *ascii)
Niels Möller's avatar
Niels Möller committed
932
{
933
  size_t ascii_length = strlen(ascii);
934
  char *buffer = xalloc(1 + ascii_length);
935
936
937
  uint8_t *check = xalloc(1 + armor->decode_length(ascii_length));
  void *encode = xalloc(armor->encode_context_size);
  void *decode = xalloc(armor->decode_context_size);
938
  size_t done;
939
940
941
942

  ASSERT(ascii_length
	 <= (armor->encode_length(data_length) + armor->encode_final_length));
  ASSERT(data_length <= armor->decode_length(ascii_length));
943
944
  
  memset(buffer, 0x33, 1 + ascii_length);
945
  memset(check, 0x55, 1 + data_length);
Niels Möller's avatar
Niels Möller committed
946

947
948
949
950
951
952
  armor->encode_init(encode);
  
  done = armor->encode_update(encode, buffer, data_length, data);
  done += armor->encode_final(encode, buffer + done);
  ASSERT(done == ascii_length);

953
954
  ASSERT (MEMEQ(ascii_length, buffer, ascii));
  ASSERT (0x33 == buffer[strlen(ascii)]);
Niels Möller's avatar
Niels Möller committed
955

956
  armor->decode_init(decode);
957
  done = armor->decode_length(ascii_length);
Niels Möller's avatar
Niels Möller committed
958

959
960
961
  ASSERT(armor->decode_update(decode, &done, check, ascii_length, buffer));
  ASSERT(done == data_length);
  ASSERT(armor->decode_final(decode));
962
  
963
964
  ASSERT (MEMEQ(data_length, check, data));
  ASSERT (0x55 == check[data_length]);
965
966
967
968
969

  free(buffer);
  free(check);
  free(encode);
  free(decode);
Niels Möller's avatar
Niels Möller committed
970
971
}

972
973
#if WITH_HOGWEED

974
975
976
977
978
979
980
981
982
983
984
985
986
#ifndef mpn_zero_p
int
mpn_zero_p (mp_srcptr ap, mp_size_t n)
{
  while (--n >= 0)
    {
      if (ap[n] != 0)
	return 0;
    }
  return 1;
}
#endif

987
988
989
990
991
992
993
void
mpn_out_str (FILE *f, int base, const mp_limb_t *xp, mp_size_t xn)
{
  mpz_t x;
  mpz_out_str (f, base, mpz_roinit_n (x, xp, xn));
}

994
995
996
997
998
999
1000
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
1011
1012
#if NETTLE_USE_MINI_GMP
void
gmp_randinit_default (struct knuth_lfib_ctx *ctx)
{
  knuth_lfib_init (ctx, 17);
}
void
mpz_urandomb (mpz_t r, struct knuth_lfib_ctx *ctx, mp_bitcnt_t bits)
{
  size_t bytes = (bits+7)/8;
  uint8_t *buf = xalloc (bytes);

  knuth_lfib_random (ctx, bytes, buf);
  buf[bytes-1] &= 0xff >> (8*bytes - bits);
  nettle_mpz_set_str_256_u (r, bytes, buf);
  free (buf);
}
#endif /* NETTLE_USE_MINI_GMP */

Niels Möller's avatar
Niels Möller committed
1013
1014
1015
1016
1017
1018
mp_limb_t *
xalloc_limbs (mp_size_t n)
{
  return xalloc (n * sizeof (mp_limb_t));
}

1019
1020
1021
1022
1023
1024
1025
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042
1043
1044
1045
/* Expects local variables pub, key, rstate, digest, signature */
#define SIGN(hash, msg, expected) do { \
  hash##_update(&hash, LDATA(msg));					\
  ASSERT(rsa_##hash##_sign(key, &hash, signature));			\
  if (verbose)								\
    {									\
      fprintf(stderr, "rsa-%s signature: ", #hash);			\
      mpz_out_str(stderr, 16, signature);				\
      fprintf(stderr, "\n");						\
    }									\
  ASSERT(mpz_cmp (signature, expected) == 0);				\
									\
  hash##_update(&hash, LDATA(msg));					\
  ASSERT(rsa_##hash##_sign_tr(pub, key, &rstate,			\
			      (nettle_random_func *) knuth_lfib_random,	\
			      &hash, signature));			\
  ASSERT(mpz_cmp (signature, expected) == 0);				\
									\
  hash##_update(&hash, LDATA(msg));					\
  hash##_digest(&hash, sizeof(digest), digest);				\
  ASSERT(rsa_##hash##_sign_digest(key, digest, signature));		\
  ASSERT(mpz_cmp (signature, expected) == 0);				\
									\
  ASSERT(rsa_##hash##_sign_digest_tr(pub, key, &rstate,			\
				     (nettle_random_func *)knuth_lfib_random, \
				     digest, signature));		\
  ASSERT(mpz_cmp (signature, expected) == 0);				\
1046
1047
1048
1049
1050
1051
} while(0)

#define VERIFY(key, hash, msg, signature) (	\
  hash##_update(&hash, LDATA(msg)),		\
  rsa_##hash##_verify(key, &hash, signature)	\
)
1052

1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
1081
1082
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
void
test_rsa_set_key_1(struct rsa_public_key *pub,
		   struct rsa_private_key *key)
{
  /* Initialize key pair for test programs */
  /* 1000-bit key, generated by
   *
   *   lsh-keygen -a rsa -l 1000 -f advanced-hex
   *
   * (private-key (rsa-pkcs1 
   *        (n #69abd505285af665 36ddc7c8f027e6f0 ed435d6748b16088
   *            4fd60842b3a8d7fb bd8a3c98f0cc50ae 4f6a9f7dd73122cc
   *            ec8afa3f77134406 f53721973115fc2d 8cfbba23b145f28d
   *            84f81d3b6ae8ce1e 2850580c026e809b cfbb52566ea3a3b3
   *            df7edf52971872a7 e35c1451b8636d22 279a8fb299368238
   *            e545fbb4cf#)
   *        (e #0db2ad57#)
   *        (d #3240a56f4cd0dcc2 4a413eb4ea545259 5c83d771a1c2ba7b
   *            ec47c5b43eb4b374 09bd2aa1e236dd86 481eb1768811412f
   *            f8d91be3545912af b55c014cb55ceac6 54216af3b85d5c4f
   *            4a32894e3b5dfcde 5b2875aa4dc8d9a8 6afd0ca92ef50d35
   *            bd09f1c47efb4c8d c631e07698d362aa 4a83fd304e66d6c5
   *            468863c307#)
   *        (p #0a66399919be4b4d e5a78c5ea5c85bf9 aba8c013cb4a8732
   *            14557a12bd67711e bb4073fd39ad9a86 f4e80253ad809e5b
   *            f2fad3bc37f6f013 273c9552c9f489#)
   *        (q #0a294f069f118625 f5eae2538db9338c 776a298eae953329
   *            9fd1eed4eba04e82 b2593bc98ba8db27 de034da7daaea795
   *            2d55b07b5f9a5875 d1ca5f6dcab897#)
   *        (a #011b6c48eb592eee e85d1bb35cfb6e07 344ea0b5e5f03a28
   *            5b405396cbc78c5c 868e961db160ba8d 4b984250930cf79a
   *            1bf8a9f28963de53 128aa7d690eb87#)
   *        (b #0409ecf3d2557c88 214f1af5e1f17853 d8b2d63782fa5628
   *            60cf579b0833b7ff 5c0529f2a97c6452 2fa1a8878a9635ab
   *            ce56debf431bdec2 70b308fa5bf387#)
   *        (c #04e103ee925cb5e6 6653949fa5e1a462 c9e65e1adcd60058
   *            e2df9607cee95fa8 daec7a389a7d9afc 8dd21fef9d83805a
   *            40d46f49676a2f6b 2926f70c572c00#)))
   */
  
  mpz_set_str(pub->n,
	      "69abd505285af665" "36ddc7c8f027e6f0" "ed435d6748b16088"
	      "4fd60842b3a8d7fb" "bd8a3c98f0cc50ae" "4f6a9f7dd73122cc"
	      "ec8afa3f77134406" "f53721973115fc2d" "8cfbba23b145f28d"
	      "84f81d3b6ae8ce1e" "2850580c026e809b" "cfbb52566ea3a3b3"
	      "df7edf52971872a7" "e35c1451b8636d22" "279a8fb299368238"
	      "e545fbb4cf", 16);
  mpz_set_str(pub->e, "0db2ad57", 16);

1102
  ASSERT (rsa_public_key_prepare(pub));
1103
1104
1105
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
  
  /* d is not used */
#if 0  
  mpz_set_str(key->d,
	      "3240a56f4cd0dcc2" "4a413eb4ea545259" "5c83d771a1c2ba7b"
	      "ec47c5b43eb4b374" "09bd2aa1e236dd86" "481eb1768811412f"
	      "f8d91be3545912af" "b55c014cb55ceac6" "54216af3b85d5c4f"
	      "4a32894e3b5dfcde" "5b2875aa4dc8d9a8" "6afd0ca92ef50d35"
	      "bd09f1c47efb4c8d" "c631e07698d362aa" "4a83fd304e66d6c5"
	      "468863c307", 16);
#endif
  
  mpz_set_str(key->p,
	      "0a66399919be4b4d" "e5a78c5ea5c85bf9" "aba8c013cb4a8732"
	      "14557a12bd67711e" "bb4073fd39ad9a86" "f4e80253ad809e5b"
	      "f2fad3bc37f6f013" "273c9552c9f489", 16);

  mpz_set_str(key->q,
	      "0a294f069f118625" "f5eae2538db9338c" "776a298eae953329"
	      "9fd1eed4eba04e82" "b2593bc98ba8db27" "de034da7daaea795"
	      "2d55b07b5f9a5875" "d1ca5f6dcab897", 16);
  
  mpz_set_str(key->a,
	      "011b6c48eb592eee" "e85d1bb35cfb6e07" "344ea0b5e5f03a28"
	      "5b405396cbc78c5c" "868e961db160ba8d" "4b984250930cf79a"
	      "1bf8a9f28963de53" "128aa7d690eb87", 16);
  
  mpz_set_str(key->b,
	      "0409ecf3d2557c88" "214f1af5e1f17853" "d8b2d63782fa5628"
	      "60cf579b0833b7ff" "5c0529f2a97c6452" "2fa1a8878a9635ab"
	      "ce56debf431bdec2" "70b308fa5bf387", 16);
  
  mpz_set_str(key->c,
	      "04e103ee925cb5e6" "6653949fa5e1a462" "c9e65e1adcd60058"
	      "e2df9607cee95fa8" "daec7a389a7d9afc" "8dd21fef9d83805a"
	      "40d46f49676a2f6b" "2926f70c572c00", 16);

1140
1141
  ASSERT (rsa_private_key_prepare(key));
  ASSERT (pub->size == key->size);
1142
1143
}

1144
1145
1146
1147
1148
1149
void
test_rsa_md5(struct rsa_public_key *pub,
	     struct rsa_private_key *key,
	     mpz_t expected)
{
  struct md5_ctx md5;
1150
1151
  struct knuth_lfib_ctx rstate;
  uint8_t digest[MD5_DIGEST_SIZE];
1152
1153
1154
1155
  mpz_t signature;

  md5_init(&md5);
  mpz_init(signature);
1156
  knuth_lfib_init (&rstate, 15);
1157

1158
  SIGN(md5, "The magic words are squeamish ossifrage", expected);
1159
1160

  /* Try bad data */
1161
1162
  ASSERT (!VERIFY(pub, md5,
		  "The magick words are squeamish ossifrage", signature));
1163
1164

  /* Try correct data */
1165
1166
  ASSERT (VERIFY(pub, md5,
		 "The magic words are squeamish ossifrage", signature));
1167
1168

  /* Try bad signature */
1169
  mpz_combit(signature, 17);
1170
1171
  ASSERT (!VERIFY(pub, md5,
		  "The magic words are squeamish ossifrage", signature));
1172
1173
1174
1175
1176
1177

  mpz_clear(signature);
}

void
test_rsa_sha1(struct rsa_public_key *pub,
1178
1179
	      struct rsa_private_key *key,
	      mpz_t expected)
1180
1181
{
  struct sha1_ctx sha1;
1182
1183
  struct knuth_lfib_ctx rstate;
  uint8_t digest[SHA1_DIGEST_SIZE];
1184
1185
1186
1187
  mpz_t signature;

  sha1_init(&sha1);
  mpz_init(signature);
1188
  knuth_lfib_init (&rstate, 16);
1189

1190
  SIGN(sha1, "The magic words are squeamish ossifrage", expected);
1191
1192

  /* Try bad data */
1193
1194
  ASSERT (!VERIFY(pub, sha1,
		  "The magick words are squeamish ossifrage", signature));
1195
1196

  /* Try correct data */
1197
1198
  ASSERT (VERIFY(pub, sha1,
		 "The magic words are squeamish ossifrage", signature));
1199
1200

  /* Try bad signature */
1201
  mpz_combit(signature, 17);
1202
1203
  ASSERT (!VERIFY(pub, sha1,
		  "The magic words are squeamish ossifrage", signature));
1204
1205
1206

  mpz_clear(signature);
}
1207

1208
1209
void
test_rsa_sha256(struct rsa_public_key *pub,
1210
1211
		struct rsa_private_key *key,
		mpz_t expected)
1212
1213
{
  struct sha256_ctx sha256;
1214
1215
  struct knuth_lfib_ctx rstate;
  uint8_t digest[SHA256_DIGEST_SIZE];
1216
1217
1218
1219
  mpz_t signature;

  sha256_init(&sha256);
  mpz_init(signature);
1220
  knuth_lfib_init (&rstate, 17);
1221

1222
  SIGN(sha256, "The magic words are squeamish ossifrage", expected);
1223
1224

  /* Try bad data */
1225
1226
  ASSERT (!VERIFY(pub, sha256,
		  "The magick words are squeamish ossifrage", signature));
1227
1228

  /* Try correct data */
1229
1230
  ASSERT (VERIFY(pub, sha256,
		 "The magic words are squeamish ossifrage", signature));
1231
1232

  /* Try bad signature */
1233
  mpz_combit(signature, 17);
1234
1235
  ASSERT (!VERIFY(pub, sha256,
		  "The magic words are squeamish ossifrage", signature));
1236
1237
1238
1239

  mpz_clear(signature);
}

1240
1241
1242
1243
1244
1245
void
test_rsa_sha512(struct rsa_public_key *pub,
		struct rsa_private_key *key,
		mpz_t expected)
{
  struct sha512_ctx sha512;
1246
1247
  struct knuth_lfib_ctx rstate;
  uint8_t digest[SHA512_DIGEST_SIZE];
1248
1249
1250
1251
  mpz_t signature;

  sha512_init(&sha512);
  mpz_init(signature);
1252
  knuth_lfib_init (&rstate, 18);
1253

1254
  SIGN(sha512, "The magic words are squeamish ossifrage", expected);
1255
1256

  /* Try bad data */
1257
1258
  ASSERT (!VERIFY(pub, sha512,
		  "The magick words are squeamish ossifrage", signature));
1259
1260

  /* Try correct data */
1261
1262
  ASSERT (VERIFY(pub, sha512,
		 "The magic words are squeamish ossifrage", signature));
1263
1264

  /* Try bad signature */
1265
  mpz_combit(signature, 17);
1266
1267
  ASSERT (!VERIFY(pub, sha512,
		  "The magic words are squeamish ossifrage", signature));
1268
1269
1270
1271

  mpz_clear(signature);
}

Niels Möller's avatar
Niels Möller committed
1272
1273
1274
#undef SIGN
#undef VERIFY

1275
1276
1277
1278
1279
1280
1281
1282
1283
1284
1285
1286
1287
1288
1289
1290
1291
1292
1293
1294
1295
1296
1297
1298
1299
1300
1301
1302
1303
1304
1305
1306
1307
1308
void
test_rsa_key(struct rsa_public_key *pub,
	     struct rsa_private_key *key)
{
  mpz_t tmp;
  mpz_t phi;
  
  mpz_init(tmp); mpz_init(phi);
  
  if (verbose)
    {
      /* FIXME: Use gmp_printf */
      fprintf(stderr, "Public key: n=");
      mpz_out_str(stderr, 16, pub->n);
      fprintf(stderr, "\n    e=");
      mpz_out_str(stderr, 16, pub->e);

      fprintf(stderr, "\n\nPrivate key: d=");
      mpz_out_str(stderr, 16, key->d);
      fprintf(stderr, "\n    p=");
      mpz_out_str(stderr, 16, key->p);
      fprintf(stderr, "\n    q=");
      mpz_out_str(stderr, 16, key->q);
      fprintf(stderr, "\n    a=");
      mpz_out_str(stderr, 16, key->a);
      fprintf(stderr, "\n    b=");
      mpz_out_str(stderr, 16, key->b);
      fprintf(stderr, "\n    c=");
      mpz_out_str(stderr, 16, key->c);
      fprintf(stderr, "\n\n");
    }

  /* Check n = p q */
  mpz_mul(tmp, key->p, key->q);
1309
  ASSERT (mpz_cmp(tmp, pub->n)== 0);
1310
1311
1312
1313

  /* Check c q = 1 mod p */
  mpz_mul(tmp, key->c, key->q);
  mpz_fdiv_r(tmp, tmp, key->p);
1314
  ASSERT (mpz_cmp_ui(tmp, 1) == 0);
1315
1316
1317
1318
1319
1320
1321
1322
1323

  /* Check ed = 1 (mod phi) */
  mpz_sub_ui(phi, key->p, 1);
  mpz_sub_ui(tmp, key->q, 1);

  mpz_mul(phi, phi, tmp);

  mpz_mul(tmp, pub->e, key->d);
  mpz_fdiv_r(tmp, tmp, phi);
1324
  ASSERT (mpz_cmp_ui(tmp, 1) == 0);
1325
1326
1327
1328
1329

  /* Check a e = 1 (mod (p-1) ) */
  mpz_sub_ui(phi, key->p, 1);
  mpz_mul(tmp, pub->e, key->a);
  mpz_fdiv_r(tmp, tmp, phi);
1330
  ASSERT (mpz_cmp_ui(tmp, 1) == 0);
1331
1332
1333
1334
1335
  
  /* Check b e = 1 (mod (q-1) ) */
  mpz_sub_ui(phi, key->q, 1);
  mpz_mul(tmp, pub->e, key->b);
  mpz_fdiv_r(tmp, tmp, phi);
1336
  ASSERT (mpz_cmp_ui(tmp, 1) == 0);
1337
1338
1339
  
  mpz_clear(tmp); mpz_clear(phi);
}
Niels Möller's avatar
Niels Möller committed
1340