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