]> git.ipfire.org Git - thirdparty/openssl.git/blame_incremental - test/evp_test.c
projects / thirdparty / openssl.git / blame_incremental
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (non-incremental) | history | HEAD
Add a test where we reuse the EVP_PKEY_CTX for two HKDF test runs
[thirdparty/openssl.git] / test / evp_test.c
... / ...
CommitLineData
1/*
2 * Copyright 2015-2018 The OpenSSL Project Authors. All Rights Reserved.
3 *
4 * Licensed under the OpenSSL license (the "License"). You may not use
5 * this file except in compliance with the License. You can obtain a copy
6 * in the file LICENSE in the source distribution or at
7 * https://www.openssl.org/source/license.html
8 */
9
10#include <stdio.h>
11#include <string.h>
12#include <stdlib.h>
13#include <ctype.h>
14#include <openssl/evp.h>
15#include <openssl/pem.h>
16#include <openssl/err.h>
17#include <openssl/x509v3.h>
18#include <openssl/pkcs12.h>
19#include <openssl/kdf.h>
20#include "internal/numbers.h"
21#include "testutil.h"
22#include "evp_test.h"
23
24
25typedef struct evp_test_method_st EVP_TEST_METHOD;
26
27/*
28 * Structure holding test information
29 */
30typedef struct evp_test_st {
31 STANZA s; /* Common test stanza */
32 char *name;
33 int skip; /* Current test should be skipped */
34 const EVP_TEST_METHOD *meth; /* method for this test */
35 const char *err, *aux_err; /* Error string for test */
36 char *expected_err; /* Expected error value of test */
37 char *func; /* Expected error function string */
38 char *reason; /* Expected error reason string */
39 void *data; /* test specific data */
40} EVP_TEST;
41
42/*
43 * Test method structure
44 */
45struct evp_test_method_st {
46 /* Name of test as it appears in file */
47 const char *name;
48 /* Initialise test for "alg" */
49 int (*init) (EVP_TEST * t, const char *alg);
50 /* Clean up method */
51 void (*cleanup) (EVP_TEST * t);
52 /* Test specific name value pair processing */
53 int (*parse) (EVP_TEST * t, const char *name, const char *value);
54 /* Run the test itself */
55 int (*run_test) (EVP_TEST * t);
56};
57
58
59/*
60 * Linked list of named keys.
61 */
62typedef struct key_list_st {
63 char *name;
64 EVP_PKEY *key;
65 struct key_list_st *next;
66} KEY_LIST;
67
68/*
69 * List of public and private keys
70 */
71static KEY_LIST *private_keys;
72static KEY_LIST *public_keys;
73static int find_key(EVP_PKEY **ppk, const char *name, KEY_LIST *lst);
74
75static int parse_bin(const char *value, unsigned char **buf, size_t *buflen);
76static int pkey_test_ctrl(EVP_TEST *t, EVP_PKEY_CTX *pctx,
77 const char *value);
78
79/*
80 * Compare two memory regions for equality, returning zero if they differ.
81 * However, if there is expected to be an error and the actual error
82 * matches then the memory is expected to be different so handle this
83 * case without producing unnecessary test framework output.
84 */
85static int memory_err_compare(EVP_TEST *t, const char *err,
86 const void *expected, size_t expected_len,
87 const void *got, size_t got_len)
88{
89 int r;
90
91 if (t->expected_err != NULL && strcmp(t->expected_err, err) == 0)
92 r = !TEST_mem_ne(expected, expected_len, got, got_len);
93 else
94 r = TEST_mem_eq(expected, expected_len, got, got_len);
95 if (!r)
96 t->err = err;
97 return r;
98}
99
100/*
101 * Structure used to hold a list of blocks of memory to test
102 * calls to "update" like functions.
103 */
104struct evp_test_buffer_st {
105 unsigned char *buf;
106 size_t buflen;
107 size_t count;
108 int count_set;
109};
110
111static void evp_test_buffer_free(EVP_TEST_BUFFER *db)
112{
113 if (db != NULL) {
114 OPENSSL_free(db->buf);
115 OPENSSL_free(db);
116 }
117}
118
119/*
120 * append buffer to a list
121 */
122static int evp_test_buffer_append(const char *value,
123 STACK_OF(EVP_TEST_BUFFER) **sk)
124{
125 EVP_TEST_BUFFER *db = NULL;
126
127 if (!TEST_ptr(db = OPENSSL_malloc(sizeof(*db))))
128 goto err;
129
130 if (!parse_bin(value, &db->buf, &db->buflen))
131 goto err;
132 db->count = 1;
133 db->count_set = 0;
134
135 if (*sk == NULL && !TEST_ptr(*sk = sk_EVP_TEST_BUFFER_new_null()))
136 goto err;
137 if (!sk_EVP_TEST_BUFFER_push(*sk, db))
138 goto err;
139
140 return 1;
141
142err:
143 evp_test_buffer_free(db);
144 return 0;
145}
146
147/*
148 * replace last buffer in list with copies of itself
149 */
150static int evp_test_buffer_ncopy(const char *value,
151 STACK_OF(EVP_TEST_BUFFER) *sk)
152{
153 EVP_TEST_BUFFER *db;
154 unsigned char *tbuf, *p;
155 size_t tbuflen;
156 int ncopy = atoi(value);
157 int i;
158
159 if (ncopy <= 0)
160 return 0;
161 if (sk == NULL || sk_EVP_TEST_BUFFER_num(sk) == 0)
162 return 0;
163 db = sk_EVP_TEST_BUFFER_value(sk, sk_EVP_TEST_BUFFER_num(sk) - 1);
164
165 tbuflen = db->buflen * ncopy;
166 if (!TEST_ptr(tbuf = OPENSSL_malloc(tbuflen)))
167 return 0;
168 for (i = 0, p = tbuf; i < ncopy; i++, p += db->buflen)
169 memcpy(p, db->buf, db->buflen);
170
171 OPENSSL_free(db->buf);
172 db->buf = tbuf;
173 db->buflen = tbuflen;
174 return 1;
175}
176
177/*
178 * set repeat count for last buffer in list
179 */
180static int evp_test_buffer_set_count(const char *value,
181 STACK_OF(EVP_TEST_BUFFER) *sk)
182{
183 EVP_TEST_BUFFER *db;
184 int count = atoi(value);
185
186 if (count <= 0)
187 return 0;
188
189 if (sk == NULL || sk_EVP_TEST_BUFFER_num(sk) == 0)
190 return 0;
191
192 db = sk_EVP_TEST_BUFFER_value(sk, sk_EVP_TEST_BUFFER_num(sk) - 1);
193 if (db->count_set != 0)
194 return 0;
195
196 db->count = (size_t)count;
197 db->count_set = 1;
198 return 1;
199}
200
201/*
202 * call "fn" with each element of the list in turn
203 */
204static int evp_test_buffer_do(STACK_OF(EVP_TEST_BUFFER) *sk,
205 int (*fn)(void *ctx,
206 const unsigned char *buf,
207 size_t buflen),
208 void *ctx)
209{
210 int i;
211
212 for (i = 0; i < sk_EVP_TEST_BUFFER_num(sk); i++) {
213 EVP_TEST_BUFFER *tb = sk_EVP_TEST_BUFFER_value(sk, i);
214 size_t j;
215
216 for (j = 0; j < tb->count; j++) {
217 if (fn(ctx, tb->buf, tb->buflen) <= 0)
218 return 0;
219 }
220 }
221 return 1;
222}
223
224/*
225 * Unescape some sequences in string literals (only \n for now).
226 * Return an allocated buffer, set |out_len|. If |input_len|
227 * is zero, get an empty buffer but set length to zero.
228 */
229static unsigned char* unescape(const char *input, size_t input_len,
230 size_t *out_len)
231{
232 unsigned char *ret, *p;
233 size_t i;
234
235 if (input_len == 0) {
236 *out_len = 0;
237 return OPENSSL_zalloc(1);
238 }
239
240 /* Escaping is non-expanding; over-allocate original size for simplicity. */
241 if (!TEST_ptr(ret = p = OPENSSL_malloc(input_len)))
242 return NULL;
243
244 for (i = 0; i < input_len; i++) {
245 if (*input == '\\') {
246 if (i == input_len - 1 || *++input != 'n') {
247 TEST_error("Bad escape sequence in file");
248 goto err;
249 }
250 *p++ = '\n';
251 i++;
252 input++;
253 } else {
254 *p++ = *input++;
255 }
256 }
257
258 *out_len = p - ret;
259 return ret;
260
261 err:
262 OPENSSL_free(ret);
263 return NULL;
264}
265
266/*
267 * For a hex string "value" convert to a binary allocated buffer.
268 * Return 1 on success or 0 on failure.
269 */
270static int parse_bin(const char *value, unsigned char **buf, size_t *buflen)
271{
272 long len;
273
274 /* Check for NULL literal */
275 if (strcmp(value, "NULL") == 0) {
276 *buf = NULL;
277 *buflen = 0;
278 return 1;
279 }
280
281 /* Check for empty value */
282 if (*value == '\0') {
283 /*
284 * Don't return NULL for zero length buffer. This is needed for
285 * some tests with empty keys: HMAC_Init_ex() expects a non-NULL key
286 * buffer even if the key length is 0, in order to detect key reset.
287 */
288 *buf = OPENSSL_malloc(1);
289 if (*buf == NULL)
290 return 0;
291 **buf = 0;
292 *buflen = 0;
293 return 1;
294 }
295
296 /* Check for string literal */
297 if (value[0] == '"') {
298 size_t vlen = strlen(++value);
299
300 if (vlen == 0 || value[vlen - 1] != '"')
301 return 0;
302 vlen--;
303 *buf = unescape(value, vlen, buflen);
304 return *buf == NULL ? 0 : 1;
305 }
306
307 /* Otherwise assume as hex literal and convert it to binary buffer */
308 if (!TEST_ptr(*buf = OPENSSL_hexstr2buf(value, &len))) {
309 TEST_info("Can't convert %s", value);
310 TEST_openssl_errors();
311 return -1;
312 }
313 /* Size of input buffer means we'll never overflow */
314 *buflen = len;
315 return 1;
316}
317
318
319/**
320*** MESSAGE DIGEST TESTS
321**/
322
323typedef struct digest_data_st {
324 /* Digest this test is for */
325 const EVP_MD *digest;
326 /* Input to digest */
327 STACK_OF(EVP_TEST_BUFFER) *input;
328 /* Expected output */
329 unsigned char *output;
330 size_t output_len;
331} DIGEST_DATA;
332
333static int digest_test_init(EVP_TEST *t, const char *alg)
334{
335 DIGEST_DATA *mdat;
336 const EVP_MD *digest;
337
338 if ((digest = EVP_get_digestbyname(alg)) == NULL) {
339 /* If alg has an OID assume disabled algorithm */
340 if (OBJ_sn2nid(alg) != NID_undef || OBJ_ln2nid(alg) != NID_undef) {
341 t->skip = 1;
342 return 1;
343 }
344 return 0;
345 }
346 if (!TEST_ptr(mdat = OPENSSL_zalloc(sizeof(*mdat))))
347 return 0;
348 t->data = mdat;
349 mdat->digest = digest;
350 return 1;
351}
352
353static void digest_test_cleanup(EVP_TEST *t)
354{
355 DIGEST_DATA *mdat = t->data;
356
357 sk_EVP_TEST_BUFFER_pop_free(mdat->input, evp_test_buffer_free);
358 OPENSSL_free(mdat->output);
359}
360
361static int digest_test_parse(EVP_TEST *t,
362 const char *keyword, const char *value)
363{
364 DIGEST_DATA *mdata = t->data;
365
366 if (strcmp(keyword, "Input") == 0)
367 return evp_test_buffer_append(value, &mdata->input);
368 if (strcmp(keyword, "Output") == 0)
369 return parse_bin(value, &mdata->output, &mdata->output_len);
370 if (strcmp(keyword, "Count") == 0)
371 return evp_test_buffer_set_count(value, mdata->input);
372 if (strcmp(keyword, "Ncopy") == 0)
373 return evp_test_buffer_ncopy(value, mdata->input);
374 return 0;
375}
376
377static int digest_update_fn(void *ctx, const unsigned char *buf, size_t buflen)
378{
379 return EVP_DigestUpdate(ctx, buf, buflen);
380}
381
382static int digest_test_run(EVP_TEST *t)
383{
384 DIGEST_DATA *expected = t->data;
385 EVP_MD_CTX *mctx;
386 unsigned char *got = NULL;
387 unsigned int got_len;
388
389 t->err = "TEST_FAILURE";
390 if (!TEST_ptr(mctx = EVP_MD_CTX_new()))
391 goto err;
392
393 got = OPENSSL_malloc(expected->output_len > EVP_MAX_MD_SIZE ?
394 expected->output_len : EVP_MAX_MD_SIZE);
395 if (!TEST_ptr(got))
396 goto err;
397
398 if (!EVP_DigestInit_ex(mctx, expected->digest, NULL)) {
399 t->err = "DIGESTINIT_ERROR";
400 goto err;
401 }
402 if (!evp_test_buffer_do(expected->input, digest_update_fn, mctx)) {
403 t->err = "DIGESTUPDATE_ERROR";
404 goto err;
405 }
406
407 if (EVP_MD_flags(expected->digest) & EVP_MD_FLAG_XOF) {
408 got_len = expected->output_len;
409 if (!EVP_DigestFinalXOF(mctx, got, got_len)) {
410 t->err = "DIGESTFINALXOF_ERROR";
411 goto err;
412 }
413 } else {
414 if (!EVP_DigestFinal(mctx, got, &got_len)) {
415 t->err = "DIGESTFINAL_ERROR";
416 goto err;
417 }
418 }
419 if (!TEST_int_eq(expected->output_len, got_len)) {
420 t->err = "DIGEST_LENGTH_MISMATCH";
421 goto err;
422 }
423 if (!memory_err_compare(t, "DIGEST_MISMATCH",
424 expected->output, expected->output_len,
425 got, got_len))
426 goto err;
427
428 t->err = NULL;
429
430 err:
431 OPENSSL_free(got);
432 EVP_MD_CTX_free(mctx);
433 return 1;
434}
435
436static const EVP_TEST_METHOD digest_test_method = {
437 "Digest",
438 digest_test_init,
439 digest_test_cleanup,
440 digest_test_parse,
441 digest_test_run
442};
443
444
445/**
446*** CIPHER TESTS
447**/
448
449typedef struct cipher_data_st {
450 const EVP_CIPHER *cipher;
451 int enc;
452 /* EVP_CIPH_GCM_MODE, EVP_CIPH_CCM_MODE or EVP_CIPH_OCB_MODE if AEAD */
453 int aead;
454 unsigned char *key;
455 size_t key_len;
456 unsigned char *iv;
457 size_t iv_len;
458 unsigned char *plaintext;
459 size_t plaintext_len;
460 unsigned char *ciphertext;
461 size_t ciphertext_len;
462 /* GCM, CCM and OCB only */
463 unsigned char *aad;
464 size_t aad_len;
465 unsigned char *tag;
466 size_t tag_len;
467} CIPHER_DATA;
468
469static int cipher_test_init(EVP_TEST *t, const char *alg)
470{
471 const EVP_CIPHER *cipher;
472 CIPHER_DATA *cdat;
473 int m;
474
475 if ((cipher = EVP_get_cipherbyname(alg)) == NULL) {
476 /* If alg has an OID assume disabled algorithm */
477 if (OBJ_sn2nid(alg) != NID_undef || OBJ_ln2nid(alg) != NID_undef) {
478 t->skip = 1;
479 return 1;
480 }
481 return 0;
482 }
483 cdat = OPENSSL_zalloc(sizeof(*cdat));
484 cdat->cipher = cipher;
485 cdat->enc = -1;
486 m = EVP_CIPHER_mode(cipher);
487 if (m == EVP_CIPH_GCM_MODE
488 || m == EVP_CIPH_OCB_MODE
489 || m == EVP_CIPH_CCM_MODE)
490 cdat->aead = m;
491 else if (EVP_CIPHER_flags(cipher) & EVP_CIPH_FLAG_AEAD_CIPHER)
492 cdat->aead = -1;
493 else
494 cdat->aead = 0;
495
496 t->data = cdat;
497 return 1;
498}
499
500static void cipher_test_cleanup(EVP_TEST *t)
501{
502 CIPHER_DATA *cdat = t->data;
503
504 OPENSSL_free(cdat->key);
505 OPENSSL_free(cdat->iv);
506 OPENSSL_free(cdat->ciphertext);
507 OPENSSL_free(cdat->plaintext);
508 OPENSSL_free(cdat->aad);
509 OPENSSL_free(cdat->tag);
510}
511
512static int cipher_test_parse(EVP_TEST *t, const char *keyword,
513 const char *value)
514{
515 CIPHER_DATA *cdat = t->data;
516
517 if (strcmp(keyword, "Key") == 0)
518 return parse_bin(value, &cdat->key, &cdat->key_len);
519 if (strcmp(keyword, "IV") == 0)
520 return parse_bin(value, &cdat->iv, &cdat->iv_len);
521 if (strcmp(keyword, "Plaintext") == 0)
522 return parse_bin(value, &cdat->plaintext, &cdat->plaintext_len);
523 if (strcmp(keyword, "Ciphertext") == 0)
524 return parse_bin(value, &cdat->ciphertext, &cdat->ciphertext_len);
525 if (cdat->aead) {
526 if (strcmp(keyword, "AAD") == 0)
527 return parse_bin(value, &cdat->aad, &cdat->aad_len);
528 if (strcmp(keyword, "Tag") == 0)
529 return parse_bin(value, &cdat->tag, &cdat->tag_len);
530 }
531
532 if (strcmp(keyword, "Operation") == 0) {
533 if (strcmp(value, "ENCRYPT") == 0)
534 cdat->enc = 1;
535 else if (strcmp(value, "DECRYPT") == 0)
536 cdat->enc = 0;
537 else
538 return 0;
539 return 1;
540 }
541 return 0;
542}
543
544static int cipher_test_enc(EVP_TEST *t, int enc,
545 size_t out_misalign, size_t inp_misalign, int frag)
546{
547 CIPHER_DATA *expected = t->data;
548 unsigned char *in, *expected_out, *tmp = NULL;
549 size_t in_len, out_len, donelen = 0;
550 int ok = 0, tmplen, chunklen, tmpflen;
551 EVP_CIPHER_CTX *ctx = NULL;
552
553 t->err = "TEST_FAILURE";
554 if (!TEST_ptr(ctx = EVP_CIPHER_CTX_new()))
555 goto err;
556 EVP_CIPHER_CTX_set_flags(ctx, EVP_CIPHER_CTX_FLAG_WRAP_ALLOW);
557 if (enc) {
558 in = expected->plaintext;
559 in_len = expected->plaintext_len;
560 expected_out = expected->ciphertext;
561 out_len = expected->ciphertext_len;
562 } else {
563 in = expected->ciphertext;
564 in_len = expected->ciphertext_len;
565 expected_out = expected->plaintext;
566 out_len = expected->plaintext_len;
567 }
568 if (inp_misalign == (size_t)-1) {
569 /*
570 * Exercise in-place encryption
571 */
572 tmp = OPENSSL_malloc(out_misalign + in_len + 2 * EVP_MAX_BLOCK_LENGTH);
573 if (!tmp)
574 goto err;
575 in = memcpy(tmp + out_misalign, in, in_len);
576 } else {
577 inp_misalign += 16 - ((out_misalign + in_len) & 15);
578 /*
579 * 'tmp' will store both output and copy of input. We make the copy
580 * of input to specifically aligned part of 'tmp'. So we just
581 * figured out how much padding would ensure the required alignment,
582 * now we allocate extended buffer and finally copy the input just
583 * past inp_misalign in expression below. Output will be written
584 * past out_misalign...
585 */
586 tmp = OPENSSL_malloc(out_misalign + in_len + 2 * EVP_MAX_BLOCK_LENGTH +
587 inp_misalign + in_len);
588 if (!tmp)
589 goto err;
590 in = memcpy(tmp + out_misalign + in_len + 2 * EVP_MAX_BLOCK_LENGTH +
591 inp_misalign, in, in_len);
592 }
593 if (!EVP_CipherInit_ex(ctx, expected->cipher, NULL, NULL, NULL, enc)) {
594 t->err = "CIPHERINIT_ERROR";
595 goto err;
596 }
597 if (expected->iv) {
598 if (expected->aead) {
599 if (!EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_AEAD_SET_IVLEN,
600 expected->iv_len, 0)) {
601 t->err = "INVALID_IV_LENGTH";
602 goto err;
603 }
604 } else if (expected->iv_len != (size_t)EVP_CIPHER_CTX_iv_length(ctx)) {
605 t->err = "INVALID_IV_LENGTH";
606 goto err;
607 }
608 }
609 if (expected->aead) {
610 unsigned char *tag;
611 /*
612 * If encrypting or OCB just set tag length initially, otherwise
613 * set tag length and value.
614 */
615 if (enc || expected->aead == EVP_CIPH_OCB_MODE) {
616 t->err = "TAG_LENGTH_SET_ERROR";
617 tag = NULL;
618 } else {
619 t->err = "TAG_SET_ERROR";
620 tag = expected->tag;
621 }
622 if (tag || expected->aead != EVP_CIPH_GCM_MODE) {
623 if (!EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_AEAD_SET_TAG,
624 expected->tag_len, tag))
625 goto err;
626 }
627 }
628
629 if (!EVP_CIPHER_CTX_set_key_length(ctx, expected->key_len)) {
630 t->err = "INVALID_KEY_LENGTH";
631 goto err;
632 }
633 if (!EVP_CipherInit_ex(ctx, NULL, NULL, expected->key, expected->iv, -1)) {
634 t->err = "KEY_SET_ERROR";
635 goto err;
636 }
637
638 if (!enc && expected->aead == EVP_CIPH_OCB_MODE) {
639 if (!EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_AEAD_SET_TAG,
640 expected->tag_len, expected->tag)) {
641 t->err = "TAG_SET_ERROR";
642 goto err;
643 }
644 }
645
646 if (expected->aead == EVP_CIPH_CCM_MODE) {
647 if (!EVP_CipherUpdate(ctx, NULL, &tmplen, NULL, out_len)) {
648 t->err = "CCM_PLAINTEXT_LENGTH_SET_ERROR";
649 goto err;
650 }
651 }
652 if (expected->aad) {
653 t->err = "AAD_SET_ERROR";
654 if (!frag) {
655 if (!EVP_CipherUpdate(ctx, NULL, &chunklen, expected->aad,
656 expected->aad_len))
657 goto err;
658 } else {
659 /*
660 * Supply the AAD in chunks less than the block size where possible
661 */
662 if (expected->aad_len > 0) {
663 if (!EVP_CipherUpdate(ctx, NULL, &chunklen, expected->aad, 1))
664 goto err;
665 donelen++;
666 }
667 if (expected->aad_len > 2) {
668 if (!EVP_CipherUpdate(ctx, NULL, &chunklen,
669 expected->aad + donelen,
670 expected->aad_len - 2))
671 goto err;
672 donelen += expected->aad_len - 2;
673 }
674 if (expected->aad_len > 1
675 && !EVP_CipherUpdate(ctx, NULL, &chunklen,
676 expected->aad + donelen, 1))
677 goto err;
678 }
679 }
680 EVP_CIPHER_CTX_set_padding(ctx, 0);
681 t->err = "CIPHERUPDATE_ERROR";
682 tmplen = 0;
683 if (!frag) {
684 /* We supply the data all in one go */
685 if (!EVP_CipherUpdate(ctx, tmp + out_misalign, &tmplen, in, in_len))
686 goto err;
687 } else {
688 /* Supply the data in chunks less than the block size where possible */
689 if (in_len > 0) {
690 if (!EVP_CipherUpdate(ctx, tmp + out_misalign, &chunklen, in, 1))
691 goto err;
692 tmplen += chunklen;
693 in++;
694 in_len--;
695 }
696 if (in_len > 1) {
697 if (!EVP_CipherUpdate(ctx, tmp + out_misalign + tmplen, &chunklen,
698 in, in_len - 1))
699 goto err;
700 tmplen += chunklen;
701 in += in_len - 1;
702 in_len = 1;
703 }
704 if (in_len > 0 ) {
705 if (!EVP_CipherUpdate(ctx, tmp + out_misalign + tmplen, &chunklen,
706 in, 1))
707 goto err;
708 tmplen += chunklen;
709 }
710 }
711 if (!EVP_CipherFinal_ex(ctx, tmp + out_misalign + tmplen, &tmpflen)) {
712 t->err = "CIPHERFINAL_ERROR";
713 goto err;
714 }
715 if (!memory_err_compare(t, "VALUE_MISMATCH", expected_out, out_len,
716 tmp + out_misalign, tmplen + tmpflen))
717 goto err;
718 if (enc && expected->aead) {
719 unsigned char rtag[16];
720
721 if (!TEST_size_t_le(expected->tag_len, sizeof(rtag))) {
722 t->err = "TAG_LENGTH_INTERNAL_ERROR";
723 goto err;
724 }
725 if (!EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_AEAD_GET_TAG,
726 expected->tag_len, rtag)) {
727 t->err = "TAG_RETRIEVE_ERROR";
728 goto err;
729 }
730 if (!memory_err_compare(t, "TAG_VALUE_MISMATCH",
731 expected->tag, expected->tag_len,
732 rtag, expected->tag_len))
733 goto err;
734 }
735 t->err = NULL;
736 ok = 1;
737 err:
738 OPENSSL_free(tmp);
739 EVP_CIPHER_CTX_free(ctx);
740 return ok;
741}
742
743static int cipher_test_run(EVP_TEST *t)
744{
745 CIPHER_DATA *cdat = t->data;
746 int rv, frag = 0;
747 size_t out_misalign, inp_misalign;
748
749 if (!cdat->key) {
750 t->err = "NO_KEY";
751 return 0;
752 }
753 if (!cdat->iv && EVP_CIPHER_iv_length(cdat->cipher)) {
754 /* IV is optional and usually omitted in wrap mode */
755 if (EVP_CIPHER_mode(cdat->cipher) != EVP_CIPH_WRAP_MODE) {
756 t->err = "NO_IV";
757 return 0;
758 }
759 }
760 if (cdat->aead && !cdat->tag) {
761 t->err = "NO_TAG";
762 return 0;
763 }
764 for (out_misalign = 0; out_misalign <= 1;) {
765 static char aux_err[64];
766 t->aux_err = aux_err;
767 for (inp_misalign = (size_t)-1; inp_misalign != 2; inp_misalign++) {
768 if (inp_misalign == (size_t)-1) {
769 /* kludge: inp_misalign == -1 means "exercise in-place" */
770 BIO_snprintf(aux_err, sizeof(aux_err),
771 "%s in-place, %sfragmented",
772 out_misalign ? "misaligned" : "aligned",
773 frag ? "" : "not ");
774 } else {
775 BIO_snprintf(aux_err, sizeof(aux_err),
776 "%s output and %s input, %sfragmented",
777 out_misalign ? "misaligned" : "aligned",
778 inp_misalign ? "misaligned" : "aligned",
779 frag ? "" : "not ");
780 }
781 if (cdat->enc) {
782 rv = cipher_test_enc(t, 1, out_misalign, inp_misalign, frag);
783 /* Not fatal errors: return */
784 if (rv != 1) {
785 if (rv < 0)
786 return 0;
787 return 1;
788 }
789 }
790 if (cdat->enc != 1) {
791 rv = cipher_test_enc(t, 0, out_misalign, inp_misalign, frag);
792 /* Not fatal errors: return */
793 if (rv != 1) {
794 if (rv < 0)
795 return 0;
796 return 1;
797 }
798 }
799 }
800
801 if (out_misalign == 1 && frag == 0) {
802 /*
803 * XTS, CCM and Wrap modes have special requirements about input
804 * lengths so we don't fragment for those
805 */
806 if (cdat->aead == EVP_CIPH_CCM_MODE
807 || EVP_CIPHER_mode(cdat->cipher) == EVP_CIPH_XTS_MODE
808 || EVP_CIPHER_mode(cdat->cipher) == EVP_CIPH_WRAP_MODE)
809 break;
810 out_misalign = 0;
811 frag++;
812 } else {
813 out_misalign++;
814 }
815 }
816 t->aux_err = NULL;
817
818 return 1;
819}
820
821static const EVP_TEST_METHOD cipher_test_method = {
822 "Cipher",
823 cipher_test_init,
824 cipher_test_cleanup,
825 cipher_test_parse,
826 cipher_test_run
827};
828
829
830/**
831*** MAC TESTS
832**/
833
834typedef struct mac_data_st {
835 /* MAC type in one form or another */
836 const EVP_MAC *mac; /* for mac_test_run_mac */
837 int type; /* for mac_test_run_pkey */
838 /* Algorithm string for this MAC */
839 char *alg;
840 /* MAC key */
841 unsigned char *key;
842 size_t key_len;
843 /* Input to MAC */
844 unsigned char *input;
845 size_t input_len;
846 /* Expected output */
847 unsigned char *output;
848 size_t output_len;
849 /* Collection of controls */
850 STACK_OF(OPENSSL_STRING) *controls;
851} MAC_DATA;
852
853static int mac_test_init(EVP_TEST *t, const char *alg)
854{
855 const EVP_MAC *mac = NULL;
856 int type = NID_undef;
857 MAC_DATA *mdat;
858
859 if ((mac = EVP_get_macbyname(alg)) == NULL) {
860 /*
861 * Since we didn't find an EVP_MAC, we check for known EVP_PKEY methods
862 * For debugging purposes, we allow 'NNNN by EVP_PKEY' to force running
863 * the EVP_PKEY method.
864 */
865 size_t sz = strlen(alg);
866 static const char epilogue[] = " by EVP_PKEY";
867
868 if (strcmp(alg + sz - (sizeof(epilogue) - 1), epilogue) == 0)
869 sz -= sizeof(epilogue) - 1;
870
871 if (strncmp(alg, "HMAC", sz) == 0) {
872 type = EVP_PKEY_HMAC;
873 } else if (strncmp(alg, "CMAC", sz) == 0) {
874#ifndef OPENSSL_NO_CMAC
875 type = EVP_PKEY_CMAC;
876#else
877 t->skip = 1;
878 return 1;
879#endif
880 } else if (strncmp(alg, "Poly1305", sz) == 0) {
881#ifndef OPENSSL_NO_POLY1305
882 type = EVP_PKEY_POLY1305;
883#else
884 t->skip = 1;
885 return 1;
886#endif
887 } else if (strncmp(alg, "SipHash", sz) == 0) {
888#ifndef OPENSSL_NO_SIPHASH
889 type = EVP_PKEY_SIPHASH;
890#else
891 t->skip = 1;
892 return 1;
893#endif
894 } else {
895 /*
896 * Not a known EVP_PKEY method either. If it's a known OID, then
897 * assume it's been disabled.
898 */
899 if (OBJ_sn2nid(alg) != NID_undef || OBJ_ln2nid(alg) != NID_undef) {
900 t->skip = 1;
901 return 1;
902 }
903
904 return 0;
905 }
906 }
907
908 mdat = OPENSSL_zalloc(sizeof(*mdat));
909 mdat->type = type;
910 mdat->mac = mac;
911 mdat->controls = sk_OPENSSL_STRING_new_null();
912 t->data = mdat;
913 return 1;
914}
915
916/* Because OPENSSL_free is a macro, it can't be passed as a function pointer */
917static void openssl_free(char *m)
918{
919 OPENSSL_free(m);
920}
921
922static void mac_test_cleanup(EVP_TEST *t)
923{
924 MAC_DATA *mdat = t->data;
925
926 sk_OPENSSL_STRING_pop_free(mdat->controls, openssl_free);
927 OPENSSL_free(mdat->alg);
928 OPENSSL_free(mdat->key);
929 OPENSSL_free(mdat->input);
930 OPENSSL_free(mdat->output);
931}
932
933static int mac_test_parse(EVP_TEST *t,
934 const char *keyword, const char *value)
935{
936 MAC_DATA *mdata = t->data;
937
938 if (strcmp(keyword, "Key") == 0)
939 return parse_bin(value, &mdata->key, &mdata->key_len);
940 if (strcmp(keyword, "Algorithm") == 0) {
941 mdata->alg = OPENSSL_strdup(value);
942 if (!mdata->alg)
943 return 0;
944 return 1;
945 }
946 if (strcmp(keyword, "Input") == 0)
947 return parse_bin(value, &mdata->input, &mdata->input_len);
948 if (strcmp(keyword, "Output") == 0)
949 return parse_bin(value, &mdata->output, &mdata->output_len);
950 if (strcmp(keyword, "Ctrl") == 0)
951 return sk_OPENSSL_STRING_push(mdata->controls,
952 OPENSSL_strdup(value)) != 0;
953 return 0;
954}
955
956static int mac_test_run_pkey(EVP_TEST *t)
957{
958 MAC_DATA *expected = t->data;
959 EVP_MD_CTX *mctx = NULL;
960 EVP_PKEY_CTX *pctx = NULL, *genctx = NULL;
961 EVP_PKEY *key = NULL;
962 const EVP_MD *md = NULL;
963 unsigned char *got = NULL;
964 size_t got_len;
965 int i;
966
967 if (expected->alg == NULL)
968 TEST_info("Trying the EVP_PKEY %s test", OBJ_nid2sn(expected->type));
969 else
970 TEST_info("Trying the EVP_PKEY %s test with %s",
971 OBJ_nid2sn(expected->type), expected->alg);
972
973#ifdef OPENSSL_NO_DES
974 if (expected->alg != NULL && strstr(expected->alg, "DES") != NULL) {
975 /* Skip DES */
976 t->err = NULL;
977 goto err;
978 }
979#endif
980
981 if (expected->type == EVP_PKEY_CMAC)
982 key = EVP_PKEY_new_CMAC_key(NULL, expected->key, expected->key_len,
983 EVP_get_cipherbyname(expected->alg));
984 else
985 key = EVP_PKEY_new_raw_private_key(expected->type, NULL, expected->key,
986 expected->key_len);
987 if (key == NULL) {
988 t->err = "MAC_KEY_CREATE_ERROR";
989 goto err;
990 }
991
992 if (expected->type == EVP_PKEY_HMAC) {
993 if (!TEST_ptr(md = EVP_get_digestbyname(expected->alg))) {
994 t->err = "MAC_ALGORITHM_SET_ERROR";
995 goto err;
996 }
997 }
998 if (!TEST_ptr(mctx = EVP_MD_CTX_new())) {
999 t->err = "INTERNAL_ERROR";
1000 goto err;
1001 }
1002 if (!EVP_DigestSignInit(mctx, &pctx, md, NULL, key)) {
1003 t->err = "DIGESTSIGNINIT_ERROR";
1004 goto err;
1005 }
1006 for (i = 0; i < sk_OPENSSL_STRING_num(expected->controls); i++)
1007 if (!pkey_test_ctrl(t, pctx,
1008 sk_OPENSSL_STRING_value(expected->controls, i))) {
1009 t->err = "EVPPKEYCTXCTRL_ERROR";
1010 goto err;
1011 }
1012 if (!EVP_DigestSignUpdate(mctx, expected->input, expected->input_len)) {
1013 t->err = "DIGESTSIGNUPDATE_ERROR";
1014 goto err;
1015 }
1016 if (!EVP_DigestSignFinal(mctx, NULL, &got_len)) {
1017 t->err = "DIGESTSIGNFINAL_LENGTH_ERROR";
1018 goto err;
1019 }
1020 if (!TEST_ptr(got = OPENSSL_malloc(got_len))) {
1021 t->err = "TEST_FAILURE";
1022 goto err;
1023 }
1024 if (!EVP_DigestSignFinal(mctx, got, &got_len)
1025 || !memory_err_compare(t, "TEST_MAC_ERR",
1026 expected->output, expected->output_len,
1027 got, got_len)) {
1028 t->err = "TEST_MAC_ERR";
1029 goto err;
1030 }
1031 t->err = NULL;
1032 err:
1033 EVP_MD_CTX_free(mctx);
1034 OPENSSL_free(got);
1035 EVP_PKEY_CTX_free(genctx);
1036 EVP_PKEY_free(key);
1037 return 1;
1038}
1039
1040static int mac_test_run_mac(EVP_TEST *t)
1041{
1042 MAC_DATA *expected = t->data;
1043 EVP_MAC_CTX *ctx = NULL;
1044 const void *algo = NULL;
1045 int algo_ctrl = 0;
1046 unsigned char *got = NULL;
1047 size_t got_len;
1048 int rv, i;
1049
1050 if (expected->alg == NULL)
1051 TEST_info("Trying the EVP_MAC %s test", EVP_MAC_name(expected->mac));
1052 else
1053 TEST_info("Trying the EVP_MAC %s test with %s",
1054 EVP_MAC_name(expected->mac), expected->alg);
1055
1056#ifdef OPENSSL_NO_DES
1057 if (expected->alg != NULL && strstr(expected->alg, "DES") != NULL) {
1058 /* Skip DES */
1059 t->err = NULL;
1060 goto err;
1061 }
1062#endif
1063
1064 if ((ctx = EVP_MAC_CTX_new(expected->mac)) == NULL) {
1065 t->err = "MAC_CREATE_ERROR";
1066 goto err;
1067 }
1068
1069 if (expected->alg != NULL
1070 && ((algo_ctrl = EVP_MAC_CTRL_SET_CIPHER,
1071 algo = EVP_get_cipherbyname(expected->alg)) == NULL
1072 && (algo_ctrl = EVP_MAC_CTRL_SET_MD,
1073 algo = EVP_get_digestbyname(expected->alg)) == NULL)) {
1074 t->err = "MAC_BAD_ALGORITHM";
1075 goto err;
1076 }
1077
1078
1079 if (algo_ctrl != 0) {
1080 rv = EVP_MAC_ctrl(ctx, algo_ctrl, algo);
1081 if (rv == -2) {
1082 t->err = "MAC_CTRL_INVALID";
1083 goto err;
1084 } else if (rv <= 0) {
1085 t->err = "MAC_CTRL_ERROR";
1086 goto err;
1087 }
1088 }
1089
1090 rv = EVP_MAC_ctrl(ctx, EVP_MAC_CTRL_SET_KEY,
1091 expected->key, expected->key_len);
1092 if (rv == -2) {
1093 t->err = "MAC_CTRL_INVALID";
1094 goto err;
1095 } else if (rv <= 0) {
1096 t->err = "MAC_CTRL_ERROR";
1097 goto err;
1098 }
1099
1100 if (!EVP_MAC_init(ctx)) {
1101 t->err = "MAC_INIT_ERROR";
1102 goto err;
1103 }
1104 for (i = 0; i < sk_OPENSSL_STRING_num(expected->controls); i++) {
1105 char *p, *tmpval;
1106 char *value = sk_OPENSSL_STRING_value(expected->controls, i);
1107
1108 if (!TEST_ptr(tmpval = OPENSSL_strdup(value))) {
1109 t->err = "MAC_CTRL_ERROR";
1110 goto err;
1111 }
1112 p = strchr(tmpval, ':');
1113 if (p != NULL)
1114 *p++ = '\0';
1115 rv = EVP_MAC_ctrl_str(ctx, tmpval, p);
1116 OPENSSL_free(tmpval);
1117 if (rv == -2) {
1118 t->err = "MAC_CTRL_INVALID";
1119 goto err;
1120 } else if (rv <= 0) {
1121 t->err = "MAC_CTRL_ERROR";
1122 goto err;
1123 }
1124 }
1125 if (!EVP_MAC_update(ctx, expected->input, expected->input_len)) {
1126 t->err = "MAC_UPDATE_ERROR";
1127 goto err;
1128 }
1129 if (!EVP_MAC_final(ctx, NULL, &got_len)) {
1130 t->err = "MAC_FINAL_LENGTH_ERROR";
1131 goto err;
1132 }
1133 if (!TEST_ptr(got = OPENSSL_malloc(got_len))) {
1134 t->err = "TEST_FAILURE";
1135 goto err;
1136 }
1137 if (!EVP_MAC_final(ctx, got, &got_len)
1138 || !memory_err_compare(t, "TEST_MAC_ERR",
1139 expected->output, expected->output_len,
1140 got, got_len)) {
1141 t->err = "TEST_MAC_ERR";
1142 goto err;
1143 }
1144 t->err = NULL;
1145 err:
1146 EVP_MAC_CTX_free(ctx);
1147 OPENSSL_free(got);
1148 return 1;
1149}
1150
1151static int mac_test_run(EVP_TEST *t)
1152{
1153 MAC_DATA *expected = t->data;
1154
1155 if (expected->mac != NULL)
1156 return mac_test_run_mac(t);
1157 return mac_test_run_pkey(t);
1158}
1159
1160static const EVP_TEST_METHOD mac_test_method = {
1161 "MAC",
1162 mac_test_init,
1163 mac_test_cleanup,
1164 mac_test_parse,
1165 mac_test_run
1166};
1167
1168
1169/**
1170*** PUBLIC KEY TESTS
1171*** These are all very similar and share much common code.
1172**/
1173
1174typedef struct pkey_data_st {
1175 /* Context for this operation */
1176 EVP_PKEY_CTX *ctx;
1177 /* Key operation to perform */
1178 int (*keyop) (EVP_PKEY_CTX *ctx,
1179 unsigned char *sig, size_t *siglen,
1180 const unsigned char *tbs, size_t tbslen);
1181 /* Input to MAC */
1182 unsigned char *input;
1183 size_t input_len;
1184 /* Expected output */
1185 unsigned char *output;
1186 size_t output_len;
1187} PKEY_DATA;
1188
1189/*
1190 * Perform public key operation setup: lookup key, allocated ctx and call
1191 * the appropriate initialisation function
1192 */
1193static int pkey_test_init(EVP_TEST *t, const char *name,
1194 int use_public,
1195 int (*keyopinit) (EVP_PKEY_CTX *ctx),
1196 int (*keyop)(EVP_PKEY_CTX *ctx,
1197 unsigned char *sig, size_t *siglen,
1198 const unsigned char *tbs,
1199 size_t tbslen))
1200{
1201 PKEY_DATA *kdata;
1202 EVP_PKEY *pkey = NULL;
1203 int rv = 0;
1204
1205 if (use_public)
1206 rv = find_key(&pkey, name, public_keys);
1207 if (rv == 0)
1208 rv = find_key(&pkey, name, private_keys);
1209 if (rv == 0 || pkey == NULL) {
1210 t->skip = 1;
1211 return 1;
1212 }
1213
1214 if (!TEST_ptr(kdata = OPENSSL_zalloc(sizeof(*kdata)))) {
1215 EVP_PKEY_free(pkey);
1216 return 0;
1217 }
1218 kdata->keyop = keyop;
1219 if (!TEST_ptr(kdata->ctx = EVP_PKEY_CTX_new(pkey, NULL))) {
1220 EVP_PKEY_free(pkey);
1221 OPENSSL_free(kdata);
1222 return 0;
1223 }
1224 if (keyopinit(kdata->ctx) <= 0)
1225 t->err = "KEYOP_INIT_ERROR";
1226 t->data = kdata;
1227 return 1;
1228}
1229
1230static void pkey_test_cleanup(EVP_TEST *t)
1231{
1232 PKEY_DATA *kdata = t->data;
1233
1234 OPENSSL_free(kdata->input);
1235 OPENSSL_free(kdata->output);
1236 EVP_PKEY_CTX_free(kdata->ctx);
1237}
1238
1239static int pkey_test_ctrl(EVP_TEST *t, EVP_PKEY_CTX *pctx,
1240 const char *value)
1241{
1242 int rv;
1243 char *p, *tmpval;
1244
1245 if (!TEST_ptr(tmpval = OPENSSL_strdup(value)))
1246 return 0;
1247 p = strchr(tmpval, ':');
1248 if (p != NULL)
1249 *p++ = '\0';
1250 rv = EVP_PKEY_CTX_ctrl_str(pctx, tmpval, p);
1251 if (rv == -2) {
1252 t->err = "PKEY_CTRL_INVALID";
1253 rv = 1;
1254 } else if (p != NULL && rv <= 0) {
1255 /* If p has an OID and lookup fails assume disabled algorithm */
1256 int nid = OBJ_sn2nid(p);
1257
1258 if (nid == NID_undef)
1259 nid = OBJ_ln2nid(p);
1260 if (nid != NID_undef
1261 && EVP_get_digestbynid(nid) == NULL
1262 && EVP_get_cipherbynid(nid) == NULL) {
1263 t->skip = 1;
1264 rv = 1;
1265 } else {
1266 t->err = "PKEY_CTRL_ERROR";
1267 rv = 1;
1268 }
1269 }
1270 OPENSSL_free(tmpval);
1271 return rv > 0;
1272}
1273
1274static int pkey_test_parse(EVP_TEST *t,
1275 const char *keyword, const char *value)
1276{
1277 PKEY_DATA *kdata = t->data;
1278 if (strcmp(keyword, "Input") == 0)
1279 return parse_bin(value, &kdata->input, &kdata->input_len);
1280 if (strcmp(keyword, "Output") == 0)
1281 return parse_bin(value, &kdata->output, &kdata->output_len);
1282 if (strcmp(keyword, "Ctrl") == 0)
1283 return pkey_test_ctrl(t, kdata->ctx, value);
1284 return 0;
1285}
1286
1287static int pkey_test_run(EVP_TEST *t)
1288{
1289 PKEY_DATA *expected = t->data;
1290 unsigned char *got = NULL;
1291 size_t got_len;
1292
1293 if (expected->keyop(expected->ctx, NULL, &got_len,
1294 expected->input, expected->input_len) <= 0
1295 || !TEST_ptr(got = OPENSSL_malloc(got_len))) {
1296 t->err = "KEYOP_LENGTH_ERROR";
1297 goto err;
1298 }
1299 if (expected->keyop(expected->ctx, got, &got_len,
1300 expected->input, expected->input_len) <= 0) {
1301 t->err = "KEYOP_ERROR";
1302 goto err;
1303 }
1304 if (!memory_err_compare(t, "KEYOP_MISMATCH",
1305 expected->output, expected->output_len,
1306 got, got_len))
1307 goto err;
1308
1309 t->err = NULL;
1310 err:
1311 OPENSSL_free(got);
1312 return 1;
1313}
1314
1315static int sign_test_init(EVP_TEST *t, const char *name)
1316{
1317 return pkey_test_init(t, name, 0, EVP_PKEY_sign_init, EVP_PKEY_sign);
1318}
1319
1320static const EVP_TEST_METHOD psign_test_method = {
1321 "Sign",
1322 sign_test_init,
1323 pkey_test_cleanup,
1324 pkey_test_parse,
1325 pkey_test_run
1326};
1327
1328static int verify_recover_test_init(EVP_TEST *t, const char *name)
1329{
1330 return pkey_test_init(t, name, 1, EVP_PKEY_verify_recover_init,
1331 EVP_PKEY_verify_recover);
1332}
1333
1334static const EVP_TEST_METHOD pverify_recover_test_method = {
1335 "VerifyRecover",
1336 verify_recover_test_init,
1337 pkey_test_cleanup,
1338 pkey_test_parse,
1339 pkey_test_run
1340};
1341
1342static int decrypt_test_init(EVP_TEST *t, const char *name)
1343{
1344 return pkey_test_init(t, name, 0, EVP_PKEY_decrypt_init,
1345 EVP_PKEY_decrypt);
1346}
1347
1348static const EVP_TEST_METHOD pdecrypt_test_method = {
1349 "Decrypt",
1350 decrypt_test_init,
1351 pkey_test_cleanup,
1352 pkey_test_parse,
1353 pkey_test_run
1354};
1355
1356static int verify_test_init(EVP_TEST *t, const char *name)
1357{
1358 return pkey_test_init(t, name, 1, EVP_PKEY_verify_init, 0);
1359}
1360
1361static int verify_test_run(EVP_TEST *t)
1362{
1363 PKEY_DATA *kdata = t->data;
1364
1365 if (EVP_PKEY_verify(kdata->ctx, kdata->output, kdata->output_len,
1366 kdata->input, kdata->input_len) <= 0)
1367 t->err = "VERIFY_ERROR";
1368 return 1;
1369}
1370
1371static const EVP_TEST_METHOD pverify_test_method = {
1372 "Verify",
1373 verify_test_init,
1374 pkey_test_cleanup,
1375 pkey_test_parse,
1376 verify_test_run
1377};
1378
1379
1380static int pderive_test_init(EVP_TEST *t, const char *name)
1381{
1382 return pkey_test_init(t, name, 0, EVP_PKEY_derive_init, 0);
1383}
1384
1385static int pderive_test_parse(EVP_TEST *t,
1386 const char *keyword, const char *value)
1387{
1388 PKEY_DATA *kdata = t->data;
1389
1390 if (strcmp(keyword, "PeerKey") == 0) {
1391 EVP_PKEY *peer;
1392 if (find_key(&peer, value, public_keys) == 0)
1393 return 0;
1394 if (EVP_PKEY_derive_set_peer(kdata->ctx, peer) <= 0)
1395 return 0;
1396 return 1;
1397 }
1398 if (strcmp(keyword, "SharedSecret") == 0)
1399 return parse_bin(value, &kdata->output, &kdata->output_len);
1400 if (strcmp(keyword, "Ctrl") == 0)
1401 return pkey_test_ctrl(t, kdata->ctx, value);
1402 return 0;
1403}
1404
1405static int pderive_test_run(EVP_TEST *t)
1406{
1407 PKEY_DATA *expected = t->data;
1408 unsigned char *got = NULL;
1409 size_t got_len;
1410
1411 if (EVP_PKEY_derive(expected->ctx, NULL, &got_len) <= 0) {
1412 t->err = "DERIVE_ERROR";
1413 goto err;
1414 }
1415 if (!TEST_ptr(got = OPENSSL_malloc(got_len))) {
1416 t->err = "DERIVE_ERROR";
1417 goto err;
1418 }
1419 if (EVP_PKEY_derive(expected->ctx, got, &got_len) <= 0) {
1420 t->err = "DERIVE_ERROR";
1421 goto err;
1422 }
1423 if (!memory_err_compare(t, "SHARED_SECRET_MISMATCH",
1424 expected->output, expected->output_len,
1425 got, got_len))
1426 goto err;
1427
1428 t->err = NULL;
1429 err:
1430 OPENSSL_free(got);
1431 return 1;
1432}
1433
1434static const EVP_TEST_METHOD pderive_test_method = {
1435 "Derive",
1436 pderive_test_init,
1437 pkey_test_cleanup,
1438 pderive_test_parse,
1439 pderive_test_run
1440};
1441
1442
1443/**
1444*** PBE TESTS
1445**/
1446
1447typedef enum pbe_type_enum {
1448 PBE_TYPE_INVALID = 0,
1449 PBE_TYPE_SCRYPT, PBE_TYPE_PBKDF2, PBE_TYPE_PKCS12
1450} PBE_TYPE;
1451
1452typedef struct pbe_data_st {
1453 PBE_TYPE pbe_type;
1454 /* scrypt parameters */
1455 uint64_t N, r, p, maxmem;
1456 /* PKCS#12 parameters */
1457 int id, iter;
1458 const EVP_MD *md;
1459 /* password */
1460 unsigned char *pass;
1461 size_t pass_len;
1462 /* salt */
1463 unsigned char *salt;
1464 size_t salt_len;
1465 /* Expected output */
1466 unsigned char *key;
1467 size_t key_len;
1468} PBE_DATA;
1469
1470#ifndef OPENSSL_NO_SCRYPT
1471/*
1472 * Parse unsigned decimal 64 bit integer value
1473 */
1474static int parse_uint64(const char *value, uint64_t *pr)
1475{
1476 const char *p = value;
1477
1478 if (!TEST_true(*p)) {
1479 TEST_info("Invalid empty integer value");
1480 return -1;
1481 }
1482 for (*pr = 0; *p; ) {
1483 if (*pr > UINT64_MAX / 10) {
1484 TEST_error("Integer overflow in string %s", value);
1485 return -1;
1486 }
1487 *pr *= 10;
1488 if (!TEST_true(isdigit((unsigned char)*p))) {
1489 TEST_error("Invalid character in string %s", value);
1490 return -1;
1491 }
1492 *pr += *p - '0';
1493 p++;
1494 }
1495 return 1;
1496}
1497
1498static int scrypt_test_parse(EVP_TEST *t,
1499 const char *keyword, const char *value)
1500{
1501 PBE_DATA *pdata = t->data;
1502
1503 if (strcmp(keyword, "N") == 0)
1504 return parse_uint64(value, &pdata->N);
1505 if (strcmp(keyword, "p") == 0)
1506 return parse_uint64(value, &pdata->p);
1507 if (strcmp(keyword, "r") == 0)
1508 return parse_uint64(value, &pdata->r);
1509 if (strcmp(keyword, "maxmem") == 0)
1510 return parse_uint64(value, &pdata->maxmem);
1511 return 0;
1512}
1513#endif
1514
1515static int pbkdf2_test_parse(EVP_TEST *t,
1516 const char *keyword, const char *value)
1517{
1518 PBE_DATA *pdata = t->data;
1519
1520 if (strcmp(keyword, "iter") == 0) {
1521 pdata->iter = atoi(value);
1522 if (pdata->iter <= 0)
1523 return -1;
1524 return 1;
1525 }
1526 if (strcmp(keyword, "MD") == 0) {
1527 pdata->md = EVP_get_digestbyname(value);
1528 if (pdata->md == NULL)
1529 return -1;
1530 return 1;
1531 }
1532 return 0;
1533}
1534
1535static int pkcs12_test_parse(EVP_TEST *t,
1536 const char *keyword, const char *value)
1537{
1538 PBE_DATA *pdata = t->data;
1539
1540 if (strcmp(keyword, "id") == 0) {
1541 pdata->id = atoi(value);
1542 if (pdata->id <= 0)
1543 return -1;
1544 return 1;
1545 }
1546 return pbkdf2_test_parse(t, keyword, value);
1547}
1548
1549static int pbe_test_init(EVP_TEST *t, const char *alg)
1550{
1551 PBE_DATA *pdat;
1552 PBE_TYPE pbe_type = PBE_TYPE_INVALID;
1553
1554 if (strcmp(alg, "scrypt") == 0) {
1555#ifndef OPENSSL_NO_SCRYPT
1556 pbe_type = PBE_TYPE_SCRYPT;
1557#else
1558 t->skip = 1;
1559 return 1;
1560#endif
1561 } else if (strcmp(alg, "pbkdf2") == 0) {
1562 pbe_type = PBE_TYPE_PBKDF2;
1563 } else if (strcmp(alg, "pkcs12") == 0) {
1564 pbe_type = PBE_TYPE_PKCS12;
1565 } else {
1566 TEST_error("Unknown pbe algorithm %s", alg);
1567 }
1568 pdat = OPENSSL_zalloc(sizeof(*pdat));
1569 pdat->pbe_type = pbe_type;
1570 t->data = pdat;
1571 return 1;
1572}
1573
1574static void pbe_test_cleanup(EVP_TEST *t)
1575{
1576 PBE_DATA *pdat = t->data;
1577
1578 OPENSSL_free(pdat->pass);
1579 OPENSSL_free(pdat->salt);
1580 OPENSSL_free(pdat->key);
1581}
1582
1583static int pbe_test_parse(EVP_TEST *t,
1584 const char *keyword, const char *value)
1585{
1586 PBE_DATA *pdata = t->data;
1587
1588 if (strcmp(keyword, "Password") == 0)
1589 return parse_bin(value, &pdata->pass, &pdata->pass_len);
1590 if (strcmp(keyword, "Salt") == 0)
1591 return parse_bin(value, &pdata->salt, &pdata->salt_len);
1592 if (strcmp(keyword, "Key") == 0)
1593 return parse_bin(value, &pdata->key, &pdata->key_len);
1594 if (pdata->pbe_type == PBE_TYPE_PBKDF2)
1595 return pbkdf2_test_parse(t, keyword, value);
1596 else if (pdata->pbe_type == PBE_TYPE_PKCS12)
1597 return pkcs12_test_parse(t, keyword, value);
1598#ifndef OPENSSL_NO_SCRYPT
1599 else if (pdata->pbe_type == PBE_TYPE_SCRYPT)
1600 return scrypt_test_parse(t, keyword, value);
1601#endif
1602 return 0;
1603}
1604
1605static int pbe_test_run(EVP_TEST *t)
1606{
1607 PBE_DATA *expected = t->data;
1608 unsigned char *key;
1609
1610 if (!TEST_ptr(key = OPENSSL_malloc(expected->key_len))) {
1611 t->err = "INTERNAL_ERROR";
1612 goto err;
1613 }
1614 if (expected->pbe_type == PBE_TYPE_PBKDF2) {
1615 if (PKCS5_PBKDF2_HMAC((char *)expected->pass, expected->pass_len,
1616 expected->salt, expected->salt_len,
1617 expected->iter, expected->md,
1618 expected->key_len, key) == 0) {
1619 t->err = "PBKDF2_ERROR";
1620 goto err;
1621 }
1622#ifndef OPENSSL_NO_SCRYPT
1623 } else if (expected->pbe_type == PBE_TYPE_SCRYPT) {
1624 if (EVP_PBE_scrypt((const char *)expected->pass, expected->pass_len,
1625 expected->salt, expected->salt_len, expected->N,
1626 expected->r, expected->p, expected->maxmem,
1627 key, expected->key_len) == 0) {
1628 t->err = "SCRYPT_ERROR";
1629 goto err;
1630 }
1631#endif
1632 } else if (expected->pbe_type == PBE_TYPE_PKCS12) {
1633 if (PKCS12_key_gen_uni(expected->pass, expected->pass_len,
1634 expected->salt, expected->salt_len,
1635 expected->id, expected->iter, expected->key_len,
1636 key, expected->md) == 0) {
1637 t->err = "PKCS12_ERROR";
1638 goto err;
1639 }
1640 }
1641 if (!memory_err_compare(t, "KEY_MISMATCH", expected->key, expected->key_len,
1642 key, expected->key_len))
1643 goto err;
1644
1645 t->err = NULL;
1646err:
1647 OPENSSL_free(key);
1648 return 1;
1649}
1650
1651static const EVP_TEST_METHOD pbe_test_method = {
1652 "PBE",
1653 pbe_test_init,
1654 pbe_test_cleanup,
1655 pbe_test_parse,
1656 pbe_test_run
1657};
1658
1659
1660/**
1661*** BASE64 TESTS
1662**/
1663
1664typedef enum {
1665 BASE64_CANONICAL_ENCODING = 0,
1666 BASE64_VALID_ENCODING = 1,
1667 BASE64_INVALID_ENCODING = 2
1668} base64_encoding_type;
1669
1670typedef struct encode_data_st {
1671 /* Input to encoding */
1672 unsigned char *input;
1673 size_t input_len;
1674 /* Expected output */
1675 unsigned char *output;
1676 size_t output_len;
1677 base64_encoding_type encoding;
1678} ENCODE_DATA;
1679
1680static int encode_test_init(EVP_TEST *t, const char *encoding)
1681{
1682 ENCODE_DATA *edata;
1683
1684 if (!TEST_ptr(edata = OPENSSL_zalloc(sizeof(*edata))))
1685 return 0;
1686 if (strcmp(encoding, "canonical") == 0) {
1687 edata->encoding = BASE64_CANONICAL_ENCODING;
1688 } else if (strcmp(encoding, "valid") == 0) {
1689 edata->encoding = BASE64_VALID_ENCODING;
1690 } else if (strcmp(encoding, "invalid") == 0) {
1691 edata->encoding = BASE64_INVALID_ENCODING;
1692 if (!TEST_ptr(t->expected_err = OPENSSL_strdup("DECODE_ERROR")))
1693 return 0;
1694 } else {
1695 TEST_error("Bad encoding: %s."
1696 " Should be one of {canonical, valid, invalid}",
1697 encoding);
1698 return 0;
1699 }
1700 t->data = edata;
1701 return 1;
1702}
1703
1704static void encode_test_cleanup(EVP_TEST *t)
1705{
1706 ENCODE_DATA *edata = t->data;
1707
1708 OPENSSL_free(edata->input);
1709 OPENSSL_free(edata->output);
1710 memset(edata, 0, sizeof(*edata));
1711}
1712
1713static int encode_test_parse(EVP_TEST *t,
1714 const char *keyword, const char *value)
1715{
1716 ENCODE_DATA *edata = t->data;
1717
1718 if (strcmp(keyword, "Input") == 0)
1719 return parse_bin(value, &edata->input, &edata->input_len);
1720 if (strcmp(keyword, "Output") == 0)
1721 return parse_bin(value, &edata->output, &edata->output_len);
1722 return 0;
1723}
1724
1725static int encode_test_run(EVP_TEST *t)
1726{
1727 ENCODE_DATA *expected = t->data;
1728 unsigned char *encode_out = NULL, *decode_out = NULL;
1729 int output_len, chunk_len;
1730 EVP_ENCODE_CTX *decode_ctx;
1731
1732 if (!TEST_ptr(decode_ctx = EVP_ENCODE_CTX_new())) {
1733 t->err = "INTERNAL_ERROR";
1734 goto err;
1735 }
1736
1737 if (expected->encoding == BASE64_CANONICAL_ENCODING) {
1738 EVP_ENCODE_CTX *encode_ctx;
1739
1740 if (!TEST_ptr(encode_ctx = EVP_ENCODE_CTX_new())
1741 || !TEST_ptr(encode_out =
1742 OPENSSL_malloc(EVP_ENCODE_LENGTH(expected->input_len))))
1743 goto err;
1744
1745 EVP_EncodeInit(encode_ctx);
1746 EVP_EncodeUpdate(encode_ctx, encode_out, &chunk_len,
1747 expected->input, expected->input_len);
1748 output_len = chunk_len;
1749
1750 EVP_EncodeFinal(encode_ctx, encode_out + chunk_len, &chunk_len);
1751 output_len += chunk_len;
1752
1753 EVP_ENCODE_CTX_free(encode_ctx);
1754
1755 if (!memory_err_compare(t, "BAD_ENCODING",
1756 expected->output, expected->output_len,
1757 encode_out, output_len))
1758 goto err;
1759 }
1760
1761 if (!TEST_ptr(decode_out =
1762 OPENSSL_malloc(EVP_DECODE_LENGTH(expected->output_len))))
1763 goto err;
1764
1765 EVP_DecodeInit(decode_ctx);
1766 if (EVP_DecodeUpdate(decode_ctx, decode_out, &chunk_len, expected->output,
1767 expected->output_len) < 0) {
1768 t->err = "DECODE_ERROR";
1769 goto err;
1770 }
1771 output_len = chunk_len;
1772
1773 if (EVP_DecodeFinal(decode_ctx, decode_out + chunk_len, &chunk_len) != 1) {
1774 t->err = "DECODE_ERROR";
1775 goto err;
1776 }
1777 output_len += chunk_len;
1778
1779 if (expected->encoding != BASE64_INVALID_ENCODING
1780 && !memory_err_compare(t, "BAD_DECODING",
1781 expected->input, expected->input_len,
1782 decode_out, output_len)) {
1783 t->err = "BAD_DECODING";
1784 goto err;
1785 }
1786
1787 t->err = NULL;
1788 err:
1789 OPENSSL_free(encode_out);
1790 OPENSSL_free(decode_out);
1791 EVP_ENCODE_CTX_free(decode_ctx);
1792 return 1;
1793}
1794
1795static const EVP_TEST_METHOD encode_test_method = {
1796 "Encoding",
1797 encode_test_init,
1798 encode_test_cleanup,
1799 encode_test_parse,
1800 encode_test_run,
1801};
1802
1803/**
1804*** KDF TESTS
1805**/
1806
1807typedef struct kdf_data_st {
1808 /* Context for this operation */
1809 EVP_PKEY_CTX *ctx;
1810 /* Expected output */
1811 unsigned char *output;
1812 size_t output_len;
1813} KDF_DATA;
1814
1815/*
1816 * Perform public key operation setup: lookup key, allocated ctx and call
1817 * the appropriate initialisation function
1818 */
1819static int kdf_test_init(EVP_TEST *t, const char *name)
1820{
1821 KDF_DATA *kdata;
1822 int kdf_nid = OBJ_sn2nid(name);
1823
1824#ifdef OPENSSL_NO_SCRYPT
1825 if (strcmp(name, "scrypt") == 0) {
1826 t->skip = 1;
1827 return 1;
1828 }
1829#endif
1830
1831 if (kdf_nid == NID_undef)
1832 kdf_nid = OBJ_ln2nid(name);
1833
1834 if (!TEST_ptr(kdata = OPENSSL_zalloc(sizeof(*kdata))))
1835 return 0;
1836 kdata->ctx = EVP_PKEY_CTX_new_id(kdf_nid, NULL);
1837 if (kdata->ctx == NULL) {
1838 OPENSSL_free(kdata);
1839 return 0;
1840 }
1841 if (EVP_PKEY_derive_init(kdata->ctx) <= 0) {
1842 EVP_PKEY_CTX_free(kdata->ctx);
1843 OPENSSL_free(kdata);
1844 return 0;
1845 }
1846 t->data = kdata;
1847 return 1;
1848}
1849
1850static void kdf_test_cleanup(EVP_TEST *t)
1851{
1852 KDF_DATA *kdata = t->data;
1853 OPENSSL_free(kdata->output);
1854 EVP_PKEY_CTX_free(kdata->ctx);
1855}
1856
1857static int kdf_test_parse(EVP_TEST *t,
1858 const char *keyword, const char *value)
1859{
1860 KDF_DATA *kdata = t->data;
1861
1862 if (strcmp(keyword, "Output") == 0)
1863 return parse_bin(value, &kdata->output, &kdata->output_len);
1864 if (strncmp(keyword, "Ctrl", 4) == 0)
1865 return pkey_test_ctrl(t, kdata->ctx, value);
1866 return 0;
1867}
1868
1869static int kdf_test_run(EVP_TEST *t)
1870{
1871 KDF_DATA *expected = t->data;
1872 unsigned char *got = NULL;
1873 size_t got_len = expected->output_len;
1874
1875 if (!TEST_ptr(got = OPENSSL_malloc(got_len))) {
1876 t->err = "INTERNAL_ERROR";
1877 goto err;
1878 }
1879 if (EVP_PKEY_derive(expected->ctx, got, &got_len) <= 0) {
1880 t->err = "KDF_DERIVE_ERROR";
1881 goto err;
1882 }
1883 if (!memory_err_compare(t, "KDF_MISMATCH",
1884 expected->output, expected->output_len,
1885 got, got_len))
1886 goto err;
1887
1888 t->err = NULL;
1889
1890 err:
1891 OPENSSL_free(got);
1892 return 1;
1893}
1894
1895static const EVP_TEST_METHOD kdf_test_method = {
1896 "KDF",
1897 kdf_test_init,
1898 kdf_test_cleanup,
1899 kdf_test_parse,
1900 kdf_test_run
1901};
1902
1903
1904/**
1905*** KEYPAIR TESTS
1906**/
1907
1908typedef struct keypair_test_data_st {
1909 EVP_PKEY *privk;
1910 EVP_PKEY *pubk;
1911} KEYPAIR_TEST_DATA;
1912
1913static int keypair_test_init(EVP_TEST *t, const char *pair)
1914{
1915 KEYPAIR_TEST_DATA *data;
1916 int rv = 0;
1917 EVP_PKEY *pk = NULL, *pubk = NULL;
1918 char *pub, *priv = NULL;
1919
1920 /* Split private and public names. */
1921 if (!TEST_ptr(priv = OPENSSL_strdup(pair))
1922 || !TEST_ptr(pub = strchr(priv, ':'))) {
1923 t->err = "PARSING_ERROR";
1924 goto end;
1925 }
1926 *pub++ = '\0';
1927
1928 if (!TEST_true(find_key(&pk, priv, private_keys))) {
1929 TEST_info("Can't find private key: %s", priv);
1930 t->err = "MISSING_PRIVATE_KEY";
1931 goto end;
1932 }
1933 if (!TEST_true(find_key(&pubk, pub, public_keys))) {
1934 TEST_info("Can't find public key: %s", pub);
1935 t->err = "MISSING_PUBLIC_KEY";
1936 goto end;
1937 }
1938
1939 if (pk == NULL && pubk == NULL) {
1940 /* Both keys are listed but unsupported: skip this test */
1941 t->skip = 1;
1942 rv = 1;
1943 goto end;
1944 }
1945
1946 if (!TEST_ptr(data = OPENSSL_malloc(sizeof(*data))))
1947 goto end;
1948 data->privk = pk;
1949 data->pubk = pubk;
1950 t->data = data;
1951 rv = 1;
1952 t->err = NULL;
1953
1954end:
1955 OPENSSL_free(priv);
1956 return rv;
1957}
1958
1959static void keypair_test_cleanup(EVP_TEST *t)
1960{
1961 OPENSSL_free(t->data);
1962 t->data = NULL;
1963}
1964
1965/*
1966 * For tests that do not accept any custom keywords.
1967 */
1968static int void_test_parse(EVP_TEST *t, const char *keyword, const char *value)
1969{
1970 return 0;
1971}
1972
1973static int keypair_test_run(EVP_TEST *t)
1974{
1975 int rv = 0;
1976 const KEYPAIR_TEST_DATA *pair = t->data;
1977
1978 if (pair->privk == NULL || pair->pubk == NULL) {
1979 /*
1980 * this can only happen if only one of the keys is not set
1981 * which means that one of them was unsupported while the
1982 * other isn't: hence a key type mismatch.
1983 */
1984 t->err = "KEYPAIR_TYPE_MISMATCH";
1985 rv = 1;
1986 goto end;
1987 }
1988
1989 if ((rv = EVP_PKEY_cmp(pair->privk, pair->pubk)) != 1 ) {
1990 if ( 0 == rv ) {
1991 t->err = "KEYPAIR_MISMATCH";
1992 } else if ( -1 == rv ) {
1993 t->err = "KEYPAIR_TYPE_MISMATCH";
1994 } else if ( -2 == rv ) {
1995 t->err = "UNSUPPORTED_KEY_COMPARISON";
1996 } else {
1997 TEST_error("Unexpected error in key comparison");
1998 rv = 0;
1999 goto end;
2000 }
2001 rv = 1;
2002 goto end;
2003 }
2004
2005 rv = 1;
2006 t->err = NULL;
2007
2008end:
2009 return rv;
2010}
2011
2012static const EVP_TEST_METHOD keypair_test_method = {
2013 "PrivPubKeyPair",
2014 keypair_test_init,
2015 keypair_test_cleanup,
2016 void_test_parse,
2017 keypair_test_run
2018};
2019
2020/**
2021*** KEYGEN TEST
2022**/
2023
2024typedef struct keygen_test_data_st {
2025 EVP_PKEY_CTX *genctx; /* Keygen context to use */
2026 char *keyname; /* Key name to store key or NULL */
2027} KEYGEN_TEST_DATA;
2028
2029static int keygen_test_init(EVP_TEST *t, const char *alg)
2030{
2031 KEYGEN_TEST_DATA *data;
2032 EVP_PKEY_CTX *genctx;
2033 int nid = OBJ_sn2nid(alg);
2034
2035 if (nid == NID_undef) {
2036 nid = OBJ_ln2nid(alg);
2037 if (nid == NID_undef)
2038 return 0;
2039 }
2040
2041 if (!TEST_ptr(genctx = EVP_PKEY_CTX_new_id(nid, NULL))) {
2042 /* assume algorithm disabled */
2043 t->skip = 1;
2044 return 1;
2045 }
2046
2047 if (EVP_PKEY_keygen_init(genctx) <= 0) {
2048 t->err = "KEYGEN_INIT_ERROR";
2049 goto err;
2050 }
2051
2052 if (!TEST_ptr(data = OPENSSL_malloc(sizeof(*data))))
2053 goto err;
2054 data->genctx = genctx;
2055 data->keyname = NULL;
2056 t->data = data;
2057 t->err = NULL;
2058 return 1;
2059
2060err:
2061 EVP_PKEY_CTX_free(genctx);
2062 return 0;
2063}
2064
2065static void keygen_test_cleanup(EVP_TEST *t)
2066{
2067 KEYGEN_TEST_DATA *keygen = t->data;
2068
2069 EVP_PKEY_CTX_free(keygen->genctx);
2070 OPENSSL_free(keygen->keyname);
2071 OPENSSL_free(t->data);
2072 t->data = NULL;
2073}
2074
2075static int keygen_test_parse(EVP_TEST *t,
2076 const char *keyword, const char *value)
2077{
2078 KEYGEN_TEST_DATA *keygen = t->data;
2079
2080 if (strcmp(keyword, "KeyName") == 0)
2081 return TEST_ptr(keygen->keyname = OPENSSL_strdup(value));
2082 if (strcmp(keyword, "Ctrl") == 0)
2083 return pkey_test_ctrl(t, keygen->genctx, value);
2084 return 0;
2085}
2086
2087static int keygen_test_run(EVP_TEST *t)
2088{
2089 KEYGEN_TEST_DATA *keygen = t->data;
2090 EVP_PKEY *pkey = NULL;
2091
2092 t->err = NULL;
2093 if (EVP_PKEY_keygen(keygen->genctx, &pkey) <= 0) {
2094 t->err = "KEYGEN_GENERATE_ERROR";
2095 goto err;
2096 }
2097
2098 if (keygen->keyname != NULL) {
2099 KEY_LIST *key;
2100
2101 if (find_key(NULL, keygen->keyname, private_keys)) {
2102 TEST_info("Duplicate key %s", keygen->keyname);
2103 goto err;
2104 }
2105
2106 if (!TEST_ptr(key = OPENSSL_malloc(sizeof(*key))))
2107 goto err;
2108 key->name = keygen->keyname;
2109 keygen->keyname = NULL;
2110 key->key = pkey;
2111 key->next = private_keys;
2112 private_keys = key;
2113 } else {
2114 EVP_PKEY_free(pkey);
2115 }
2116
2117 return 1;
2118
2119err:
2120 EVP_PKEY_free(pkey);
2121 return 0;
2122}
2123
2124static const EVP_TEST_METHOD keygen_test_method = {
2125 "KeyGen",
2126 keygen_test_init,
2127 keygen_test_cleanup,
2128 keygen_test_parse,
2129 keygen_test_run,
2130};
2131
2132/**
2133*** DIGEST SIGN+VERIFY TESTS
2134**/
2135
2136typedef struct {
2137 int is_verify; /* Set to 1 if verifying */
2138 int is_oneshot; /* Set to 1 for one shot operation */
2139 const EVP_MD *md; /* Digest to use */
2140 EVP_MD_CTX *ctx; /* Digest context */
2141 EVP_PKEY_CTX *pctx;
2142 STACK_OF(EVP_TEST_BUFFER) *input; /* Input data: streaming */
2143 unsigned char *osin; /* Input data if one shot */
2144 size_t osin_len; /* Input length data if one shot */
2145 unsigned char *output; /* Expected output */
2146 size_t output_len; /* Expected output length */
2147} DIGESTSIGN_DATA;
2148
2149static int digestsigver_test_init(EVP_TEST *t, const char *alg, int is_verify,
2150 int is_oneshot)
2151{
2152 const EVP_MD *md = NULL;
2153 DIGESTSIGN_DATA *mdat;
2154
2155 if (strcmp(alg, "NULL") != 0) {
2156 if ((md = EVP_get_digestbyname(alg)) == NULL) {
2157 /* If alg has an OID assume disabled algorithm */
2158 if (OBJ_sn2nid(alg) != NID_undef || OBJ_ln2nid(alg) != NID_undef) {
2159 t->skip = 1;
2160 return 1;
2161 }
2162 return 0;
2163 }
2164 }
2165 if (!TEST_ptr(mdat = OPENSSL_zalloc(sizeof(*mdat))))
2166 return 0;
2167 mdat->md = md;
2168 if (!TEST_ptr(mdat->ctx = EVP_MD_CTX_new())) {
2169 OPENSSL_free(mdat);
2170 return 0;
2171 }
2172 mdat->is_verify = is_verify;
2173 mdat->is_oneshot = is_oneshot;
2174 t->data = mdat;
2175 return 1;
2176}
2177
2178static int digestsign_test_init(EVP_TEST *t, const char *alg)
2179{
2180 return digestsigver_test_init(t, alg, 0, 0);
2181}
2182
2183static void digestsigver_test_cleanup(EVP_TEST *t)
2184{
2185 DIGESTSIGN_DATA *mdata = t->data;
2186
2187 EVP_MD_CTX_free(mdata->ctx);
2188 sk_EVP_TEST_BUFFER_pop_free(mdata->input, evp_test_buffer_free);
2189 OPENSSL_free(mdata->osin);
2190 OPENSSL_free(mdata->output);
2191 OPENSSL_free(mdata);
2192 t->data = NULL;
2193}
2194
2195static int digestsigver_test_parse(EVP_TEST *t,
2196 const char *keyword, const char *value)
2197{
2198 DIGESTSIGN_DATA *mdata = t->data;
2199
2200 if (strcmp(keyword, "Key") == 0) {
2201 EVP_PKEY *pkey = NULL;
2202 int rv = 0;
2203
2204 if (mdata->is_verify)
2205 rv = find_key(&pkey, value, public_keys);
2206 if (rv == 0)
2207 rv = find_key(&pkey, value, private_keys);
2208 if (rv == 0 || pkey == NULL) {
2209 t->skip = 1;
2210 return 1;
2211 }
2212 if (mdata->is_verify) {
2213 if (!EVP_DigestVerifyInit(mdata->ctx, &mdata->pctx, mdata->md,
2214 NULL, pkey))
2215 t->err = "DIGESTVERIFYINIT_ERROR";
2216 return 1;
2217 }
2218 if (!EVP_DigestSignInit(mdata->ctx, &mdata->pctx, mdata->md, NULL,
2219 pkey))
2220 t->err = "DIGESTSIGNINIT_ERROR";
2221 return 1;
2222 }
2223
2224 if (strcmp(keyword, "Input") == 0) {
2225 if (mdata->is_oneshot)
2226 return parse_bin(value, &mdata->osin, &mdata->osin_len);
2227 return evp_test_buffer_append(value, &mdata->input);
2228 }
2229 if (strcmp(keyword, "Output") == 0)
2230 return parse_bin(value, &mdata->output, &mdata->output_len);
2231
2232 if (!mdata->is_oneshot) {
2233 if (strcmp(keyword, "Count") == 0)
2234 return evp_test_buffer_set_count(value, mdata->input);
2235 if (strcmp(keyword, "Ncopy") == 0)
2236 return evp_test_buffer_ncopy(value, mdata->input);
2237 }
2238 if (strcmp(keyword, "Ctrl") == 0) {
2239 if (mdata->pctx == NULL)
2240 return 0;
2241 return pkey_test_ctrl(t, mdata->pctx, value);
2242 }
2243 return 0;
2244}
2245
2246static int digestsign_update_fn(void *ctx, const unsigned char *buf,
2247 size_t buflen)
2248{
2249 return EVP_DigestSignUpdate(ctx, buf, buflen);
2250}
2251
2252static int digestsign_test_run(EVP_TEST *t)
2253{
2254 DIGESTSIGN_DATA *expected = t->data;
2255 unsigned char *got = NULL;
2256 size_t got_len;
2257
2258 if (!evp_test_buffer_do(expected->input, digestsign_update_fn,
2259 expected->ctx)) {
2260 t->err = "DIGESTUPDATE_ERROR";
2261 goto err;
2262 }
2263
2264 if (!EVP_DigestSignFinal(expected->ctx, NULL, &got_len)) {
2265 t->err = "DIGESTSIGNFINAL_LENGTH_ERROR";
2266 goto err;
2267 }
2268 if (!TEST_ptr(got = OPENSSL_malloc(got_len))) {
2269 t->err = "MALLOC_FAILURE";
2270 goto err;
2271 }
2272 if (!EVP_DigestSignFinal(expected->ctx, got, &got_len)) {
2273 t->err = "DIGESTSIGNFINAL_ERROR";
2274 goto err;
2275 }
2276 if (!memory_err_compare(t, "SIGNATURE_MISMATCH",
2277 expected->output, expected->output_len,
2278 got, got_len))
2279 goto err;
2280
2281 t->err = NULL;
2282 err:
2283 OPENSSL_free(got);
2284 return 1;
2285}
2286
2287static const EVP_TEST_METHOD digestsign_test_method = {
2288 "DigestSign",
2289 digestsign_test_init,
2290 digestsigver_test_cleanup,
2291 digestsigver_test_parse,
2292 digestsign_test_run
2293};
2294
2295static int digestverify_test_init(EVP_TEST *t, const char *alg)
2296{
2297 return digestsigver_test_init(t, alg, 1, 0);
2298}
2299
2300static int digestverify_update_fn(void *ctx, const unsigned char *buf,
2301 size_t buflen)
2302{
2303 return EVP_DigestVerifyUpdate(ctx, buf, buflen);
2304}
2305
2306static int digestverify_test_run(EVP_TEST *t)
2307{
2308 DIGESTSIGN_DATA *mdata = t->data;
2309
2310 if (!evp_test_buffer_do(mdata->input, digestverify_update_fn, mdata->ctx)) {
2311 t->err = "DIGESTUPDATE_ERROR";
2312 return 1;
2313 }
2314
2315 if (EVP_DigestVerifyFinal(mdata->ctx, mdata->output,
2316 mdata->output_len) <= 0)
2317 t->err = "VERIFY_ERROR";
2318 return 1;
2319}
2320
2321static const EVP_TEST_METHOD digestverify_test_method = {
2322 "DigestVerify",
2323 digestverify_test_init,
2324 digestsigver_test_cleanup,
2325 digestsigver_test_parse,
2326 digestverify_test_run
2327};
2328
2329static int oneshot_digestsign_test_init(EVP_TEST *t, const char *alg)
2330{
2331 return digestsigver_test_init(t, alg, 0, 1);
2332}
2333
2334static int oneshot_digestsign_test_run(EVP_TEST *t)
2335{
2336 DIGESTSIGN_DATA *expected = t->data;
2337 unsigned char *got = NULL;
2338 size_t got_len;
2339
2340 if (!EVP_DigestSign(expected->ctx, NULL, &got_len,
2341 expected->osin, expected->osin_len)) {
2342 t->err = "DIGESTSIGN_LENGTH_ERROR";
2343 goto err;
2344 }
2345 if (!TEST_ptr(got = OPENSSL_malloc(got_len))) {
2346 t->err = "MALLOC_FAILURE";
2347 goto err;
2348 }
2349 if (!EVP_DigestSign(expected->ctx, got, &got_len,
2350 expected->osin, expected->osin_len)) {
2351 t->err = "DIGESTSIGN_ERROR";
2352 goto err;
2353 }
2354 if (!memory_err_compare(t, "SIGNATURE_MISMATCH",
2355 expected->output, expected->output_len,
2356 got, got_len))
2357 goto err;
2358
2359 t->err = NULL;
2360 err:
2361 OPENSSL_free(got);
2362 return 1;
2363}
2364
2365static const EVP_TEST_METHOD oneshot_digestsign_test_method = {
2366 "OneShotDigestSign",
2367 oneshot_digestsign_test_init,
2368 digestsigver_test_cleanup,
2369 digestsigver_test_parse,
2370 oneshot_digestsign_test_run
2371};
2372
2373static int oneshot_digestverify_test_init(EVP_TEST *t, const char *alg)
2374{
2375 return digestsigver_test_init(t, alg, 1, 1);
2376}
2377
2378static int oneshot_digestverify_test_run(EVP_TEST *t)
2379{
2380 DIGESTSIGN_DATA *mdata = t->data;
2381
2382 if (EVP_DigestVerify(mdata->ctx, mdata->output, mdata->output_len,
2383 mdata->osin, mdata->osin_len) <= 0)
2384 t->err = "VERIFY_ERROR";
2385 return 1;
2386}
2387
2388static const EVP_TEST_METHOD oneshot_digestverify_test_method = {
2389 "OneShotDigestVerify",
2390 oneshot_digestverify_test_init,
2391 digestsigver_test_cleanup,
2392 digestsigver_test_parse,
2393 oneshot_digestverify_test_run
2394};
2395
2396
2397/**
2398*** PARSING AND DISPATCH
2399**/
2400
2401static const EVP_TEST_METHOD *evp_test_list[] = {
2402 &cipher_test_method,
2403 &digest_test_method,
2404 &digestsign_test_method,
2405 &digestverify_test_method,
2406 &encode_test_method,
2407 &kdf_test_method,
2408 &keypair_test_method,
2409 &keygen_test_method,
2410 &mac_test_method,
2411 &oneshot_digestsign_test_method,
2412 &oneshot_digestverify_test_method,
2413 &pbe_test_method,
2414 &pdecrypt_test_method,
2415 &pderive_test_method,
2416 &psign_test_method,
2417 &pverify_recover_test_method,
2418 &pverify_test_method,
2419 NULL
2420};
2421
2422static const EVP_TEST_METHOD *find_test(const char *name)
2423{
2424 const EVP_TEST_METHOD **tt;
2425
2426 for (tt = evp_test_list; *tt; tt++) {
2427 if (strcmp(name, (*tt)->name) == 0)
2428 return *tt;
2429 }
2430 return NULL;
2431}
2432
2433static void clear_test(EVP_TEST *t)
2434{
2435 test_clearstanza(&t->s);
2436 ERR_clear_error();
2437 if (t->data != NULL) {
2438 if (t->meth != NULL)
2439 t->meth->cleanup(t);
2440 OPENSSL_free(t->data);
2441 t->data = NULL;
2442 }
2443 OPENSSL_free(t->expected_err);
2444 t->expected_err = NULL;
2445 OPENSSL_free(t->func);
2446 t->func = NULL;
2447 OPENSSL_free(t->reason);
2448 t->reason = NULL;
2449
2450 /* Text literal. */
2451 t->err = NULL;
2452 t->skip = 0;
2453 t->meth = NULL;
2454}
2455
2456/*
2457 * Check for errors in the test structure; return 1 if okay, else 0.
2458 */
2459static int check_test_error(EVP_TEST *t)
2460{
2461 unsigned long err;
2462 const char *func;
2463 const char *reason;
2464
2465 if (t->err == NULL && t->expected_err == NULL)
2466 return 1;
2467 if (t->err != NULL && t->expected_err == NULL) {
2468 if (t->aux_err != NULL) {
2469 TEST_info("%s:%d: Source of above error (%s); unexpected error %s",
2470 t->s.test_file, t->s.start, t->aux_err, t->err);
2471 } else {
2472 TEST_info("%s:%d: Source of above error; unexpected error %s",
2473 t->s.test_file, t->s.start, t->err);
2474 }
2475 return 0;
2476 }
2477 if (t->err == NULL && t->expected_err != NULL) {
2478 TEST_info("%s:%d: Succeeded but was expecting %s",
2479 t->s.test_file, t->s.start, t->expected_err);
2480 return 0;
2481 }
2482
2483 if (strcmp(t->err, t->expected_err) != 0) {
2484 TEST_info("%s:%d: Expected %s got %s",
2485 t->s.test_file, t->s.start, t->expected_err, t->err);
2486 return 0;
2487 }
2488
2489 if (t->func == NULL && t->reason == NULL)
2490 return 1;
2491
2492 if (t->func == NULL || t->reason == NULL) {
2493 TEST_info("%s:%d: Test is missing function or reason code",
2494 t->s.test_file, t->s.start);
2495 return 0;
2496 }
2497
2498 err = ERR_peek_error();
2499 if (err == 0) {
2500 TEST_info("%s:%d: Expected error \"%s:%s\" not set",
2501 t->s.test_file, t->s.start, t->func, t->reason);
2502 return 0;
2503 }
2504
2505 func = ERR_func_error_string(err);
2506 reason = ERR_reason_error_string(err);
2507 if (func == NULL && reason == NULL) {
2508 TEST_info("%s:%d: Expected error \"%s:%s\", no strings available."
2509 " Assuming ok.",
2510 t->s.test_file, t->s.start, t->func, t->reason);
2511 return 1;
2512 }
2513
2514 if (strcmp(func, t->func) == 0 && strcmp(reason, t->reason) == 0)
2515 return 1;
2516
2517 TEST_info("%s:%d: Expected error \"%s:%s\", got \"%s:%s\"",
2518 t->s.test_file, t->s.start, t->func, t->reason, func, reason);
2519
2520 return 0;
2521}
2522
2523/*
2524 * Run a parsed test. Log a message and return 0 on error.
2525 */
2526static int run_test(EVP_TEST *t)
2527{
2528 if (t->meth == NULL)
2529 return 1;
2530 t->s.numtests++;
2531 if (t->skip) {
2532 t->s.numskip++;
2533 } else {
2534 /* run the test */
2535 if (t->err == NULL && t->meth->run_test(t) != 1) {
2536 TEST_info("%s:%d %s error",
2537 t->s.test_file, t->s.start, t->meth->name);
2538 return 0;
2539 }
2540 if (!check_test_error(t)) {
2541 TEST_openssl_errors();
2542 t->s.errors++;
2543 }
2544 }
2545
2546 /* clean it up */
2547 return 1;
2548}
2549
2550static int find_key(EVP_PKEY **ppk, const char *name, KEY_LIST *lst)
2551{
2552 for (; lst != NULL; lst = lst->next) {
2553 if (strcmp(lst->name, name) == 0) {
2554 if (ppk != NULL)
2555 *ppk = lst->key;
2556 return 1;
2557 }
2558 }
2559 return 0;
2560}
2561
2562static void free_key_list(KEY_LIST *lst)
2563{
2564 while (lst != NULL) {
2565 KEY_LIST *next = lst->next;
2566
2567 EVP_PKEY_free(lst->key);
2568 OPENSSL_free(lst->name);
2569 OPENSSL_free(lst);
2570 lst = next;
2571 }
2572}
2573
2574/*
2575 * Is the key type an unsupported algorithm?
2576 */
2577static int key_unsupported(void)
2578{
2579 long err = ERR_peek_error();
2580
2581 if (ERR_GET_LIB(err) == ERR_LIB_EVP
2582 && ERR_GET_REASON(err) == EVP_R_UNSUPPORTED_ALGORITHM) {
2583 ERR_clear_error();
2584 return 1;
2585 }
2586#ifndef OPENSSL_NO_EC
2587 /*
2588 * If EC support is enabled we should catch also EC_R_UNKNOWN_GROUP as an
2589 * hint to an unsupported algorithm/curve (e.g. if binary EC support is
2590 * disabled).
2591 */
2592 if (ERR_GET_LIB(err) == ERR_LIB_EC
2593 && ERR_GET_REASON(err) == EC_R_UNKNOWN_GROUP) {
2594 ERR_clear_error();
2595 return 1;
2596 }
2597#endif /* OPENSSL_NO_EC */
2598 return 0;
2599}
2600
2601/*
2602 * NULL out the value from |pp| but return it. This "steals" a pointer.
2603 */
2604static char *take_value(PAIR *pp)
2605{
2606 char *p = pp->value;
2607
2608 pp->value = NULL;
2609 return p;
2610}
2611
2612/*
2613 * Read and parse one test. Return 0 if failure, 1 if okay.
2614 */
2615static int parse(EVP_TEST *t)
2616{
2617 KEY_LIST *key, **klist;
2618 EVP_PKEY *pkey;
2619 PAIR *pp;
2620 int i;
2621
2622top:
2623 do {
2624 if (BIO_eof(t->s.fp))
2625 return EOF;
2626 clear_test(t);
2627 if (!test_readstanza(&t->s))
2628 return 0;
2629 } while (t->s.numpairs == 0);
2630 pp = &t->s.pairs[0];
2631
2632 /* Are we adding a key? */
2633 klist = NULL;
2634 pkey = NULL;
2635 if (strcmp(pp->key, "PrivateKey") == 0) {
2636 pkey = PEM_read_bio_PrivateKey(t->s.key, NULL, 0, NULL);
2637 if (pkey == NULL && !key_unsupported()) {
2638 EVP_PKEY_free(pkey);
2639 TEST_info("Can't read private key %s", pp->value);
2640 TEST_openssl_errors();
2641 return 0;
2642 }
2643 klist = &private_keys;
2644 } else if (strcmp(pp->key, "PublicKey") == 0) {
2645 pkey = PEM_read_bio_PUBKEY(t->s.key, NULL, 0, NULL);
2646 if (pkey == NULL && !key_unsupported()) {
2647 EVP_PKEY_free(pkey);
2648 TEST_info("Can't read public key %s", pp->value);
2649 TEST_openssl_errors();
2650 return 0;
2651 }
2652 klist = &public_keys;
2653 } else if (strcmp(pp->key, "PrivateKeyRaw") == 0
2654 || strcmp(pp->key, "PublicKeyRaw") == 0 ) {
2655 char *strnid = NULL, *keydata = NULL;
2656 unsigned char *keybin;
2657 size_t keylen;
2658 int nid;
2659
2660 if (strcmp(pp->key, "PrivateKeyRaw") == 0)
2661 klist = &private_keys;
2662 else
2663 klist = &public_keys;
2664
2665 strnid = strchr(pp->value, ':');
2666 if (strnid != NULL) {
2667 *strnid++ = '\0';
2668 keydata = strchr(strnid, ':');
2669 if (keydata != NULL)
2670 *keydata++ = '\0';
2671 }
2672 if (keydata == NULL) {
2673 TEST_info("Failed to parse %s value", pp->key);
2674 return 0;
2675 }
2676
2677 nid = OBJ_txt2nid(strnid);
2678 if (nid == NID_undef) {
2679 TEST_info("Uncrecognised algorithm NID");
2680 return 0;
2681 }
2682 if (!parse_bin(keydata, &keybin, &keylen)) {
2683 TEST_info("Failed to create binary key");
2684 return 0;
2685 }
2686 if (klist == &private_keys)
2687 pkey = EVP_PKEY_new_raw_private_key(nid, NULL, keybin, keylen);
2688 else
2689 pkey = EVP_PKEY_new_raw_public_key(nid, NULL, keybin, keylen);
2690 if (pkey == NULL && !key_unsupported()) {
2691 TEST_info("Can't read %s data", pp->key);
2692 OPENSSL_free(keybin);
2693 TEST_openssl_errors();
2694 return 0;
2695 }
2696 OPENSSL_free(keybin);
2697 }
2698
2699 /* If we have a key add to list */
2700 if (klist != NULL) {
2701 if (find_key(NULL, pp->value, *klist)) {
2702 TEST_info("Duplicate key %s", pp->value);
2703 return 0;
2704 }
2705 if (!TEST_ptr(key = OPENSSL_malloc(sizeof(*key))))
2706 return 0;
2707 key->name = take_value(pp);
2708
2709 /* Hack to detect SM2 keys */
2710 if(pkey != NULL && strstr(key->name, "SM2") != NULL) {
2711#ifdef OPENSSL_NO_SM2
2712 EVP_PKEY_free(pkey);
2713 pkey = NULL;
2714#else
2715 EVP_PKEY_set_alias_type(pkey, EVP_PKEY_SM2);
2716#endif
2717 }
2718
2719 key->key = pkey;
2720 key->next = *klist;
2721 *klist = key;
2722
2723 /* Go back and start a new stanza. */
2724 if (t->s.numpairs != 1)
2725 TEST_info("Line %d: missing blank line\n", t->s.curr);
2726 goto top;
2727 }
2728
2729 /* Find the test, based on first keyword. */
2730 if (!TEST_ptr(t->meth = find_test(pp->key)))
2731 return 0;
2732 if (!t->meth->init(t, pp->value)) {
2733 TEST_error("unknown %s: %s\n", pp->key, pp->value);
2734 return 0;
2735 }
2736 if (t->skip == 1) {
2737 /* TEST_info("skipping %s %s", pp->key, pp->value); */
2738 return 0;
2739 }
2740
2741 for (pp++, i = 1; i < t->s.numpairs; pp++, i++) {
2742 if (strcmp(pp->key, "Result") == 0) {
2743 if (t->expected_err != NULL) {
2744 TEST_info("Line %d: multiple result lines", t->s.curr);
2745 return 0;
2746 }
2747 t->expected_err = take_value(pp);
2748 } else if (strcmp(pp->key, "Function") == 0) {
2749 if (t->func != NULL) {
2750 TEST_info("Line %d: multiple function lines\n", t->s.curr);
2751 return 0;
2752 }
2753 t->func = take_value(pp);
2754 } else if (strcmp(pp->key, "Reason") == 0) {
2755 if (t->reason != NULL) {
2756 TEST_info("Line %d: multiple reason lines", t->s.curr);
2757 return 0;
2758 }
2759 t->reason = take_value(pp);
2760 } else {
2761 /* Must be test specific line: try to parse it */
2762 int rv = t->meth->parse(t, pp->key, pp->value);
2763
2764 if (rv == 0) {
2765 TEST_info("Line %d: unknown keyword %s", t->s.curr, pp->key);
2766 return 0;
2767 }
2768 if (rv < 0) {
2769 TEST_info("Line %d: error processing keyword %s\n",
2770 t->s.curr, pp->key);
2771 return 0;
2772 }
2773 }
2774 }
2775
2776 return 1;
2777}
2778
2779static int run_file_tests(int i)
2780{
2781 EVP_TEST *t;
2782 const char *testfile = test_get_argument(i);
2783 int c;
2784
2785 if (!TEST_ptr(t = OPENSSL_zalloc(sizeof(*t))))
2786 return 0;
2787 if (!test_start_file(&t->s, testfile)) {
2788 OPENSSL_free(t);
2789 return 0;
2790 }
2791
2792 while (!BIO_eof(t->s.fp)) {
2793 c = parse(t);
2794 if (t->skip)
2795 continue;
2796 if (c == 0 || !run_test(t)) {
2797 t->s.errors++;
2798 break;
2799 }
2800 }
2801 test_end_file(&t->s);
2802 clear_test(t);
2803
2804 free_key_list(public_keys);
2805 free_key_list(private_keys);
2806 BIO_free(t->s.key);
2807 c = t->s.errors;
2808 OPENSSL_free(t);
2809 return c == 0;
2810}
2811
2812int setup_tests(void)
2813{
2814 size_t n = test_get_argument_count();
2815
2816 if (n == 0) {
2817 TEST_error("Usage: %s file...", test_get_program_name());
2818 return 0;
2819 }
2820
2821 ADD_ALL_TESTS(run_file_tests, n);
2822 return 1;
2823}
A mirror of the official openssl repository