1/*
2 * Copyright 2015-2022 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/*
11 * This is the OSSLTEST engine. It provides deliberately crippled digest
12 * implementations for test purposes. It is highly insecure and must NOT be
13 * used for any purpose except testing
14 */
15
16/* We need to use some engine deprecated APIs */
17#define OPENSSL_SUPPRESS_DEPRECATED
18
19/*
20 * SHA low level APIs are deprecated for public use, but still ok for
21 * internal use.  Note, that due to symbols not being exported, only the
22 * #defines and type definitions can be accessed, function calls are not
23 * available.  The digest lengths, block sizes and sizeof(CTX) are used herein
24 * for several different digests.
25 */
26#include "internal/deprecated.h"
27
28#include <stdio.h>
29#include <string.h>
30
31#include <openssl/engine.h>
32#include <openssl/sha.h>
33#include <openssl/md5.h>
34#include <openssl/rsa.h>
35#include <openssl/evp.h>
36#include <openssl/modes.h>
37#include <openssl/aes.h>
38#include <openssl/rand.h>
39#include <openssl/crypto.h>
40#include <openssl/pem.h>
41#include <crypto/evp.h>
42
43#include "e_ossltest_err.c"
44
45/* Engine Id and Name */
46static const char *engine_ossltest_id = "ossltest";
47static const char *engine_ossltest_name = "OpenSSL Test engine support";
48
49
50/* Engine Lifetime functions */
51static int ossltest_destroy(ENGINE *e);
52static int ossltest_init(ENGINE *e);
53static int ossltest_finish(ENGINE *e);
54void ENGINE_load_ossltest(void);
55
56
57/* Set up digests */
58static int ossltest_digests(ENGINE *e, const EVP_MD **digest,
59                          const int **nids, int nid);
60static const RAND_METHOD *ossltest_rand_method(void);
61
62/* MD5 */
63static int digest_md5_init(EVP_MD_CTX *ctx);
64static int digest_md5_update(EVP_MD_CTX *ctx, const void *data,
65                             size_t count);
66static int digest_md5_final(EVP_MD_CTX *ctx, unsigned char *md);
67
68static EVP_MD *_hidden_md5_md = NULL;
69static const EVP_MD *digest_md5(void)
70{
71    if (_hidden_md5_md == NULL) {
72        EVP_MD *md;
73
74        if ((md = EVP_MD_meth_new(NID_md5, NID_md5WithRSAEncryption)) == NULL
75            || !EVP_MD_meth_set_result_size(md, MD5_DIGEST_LENGTH)
76            || !EVP_MD_meth_set_input_blocksize(md, MD5_CBLOCK)
77            || !EVP_MD_meth_set_app_datasize(md,
78                                             sizeof(EVP_MD *) + sizeof(MD5_CTX))
79            || !EVP_MD_meth_set_flags(md, 0)
80            || !EVP_MD_meth_set_init(md, digest_md5_init)
81            || !EVP_MD_meth_set_update(md, digest_md5_update)
82            || !EVP_MD_meth_set_final(md, digest_md5_final)) {
83            EVP_MD_meth_free(md);
84            md = NULL;
85        }
86        _hidden_md5_md = md;
87    }
88    return _hidden_md5_md;
89}
90
91/* SHA1 */
92static int digest_sha1_init(EVP_MD_CTX *ctx);
93static int digest_sha1_update(EVP_MD_CTX *ctx, const void *data,
94                              size_t count);
95static int digest_sha1_final(EVP_MD_CTX *ctx, unsigned char *md);
96
97static EVP_MD *_hidden_sha1_md = NULL;
98static const EVP_MD *digest_sha1(void)
99{
100    if (_hidden_sha1_md == NULL) {
101        EVP_MD *md;
102
103        if ((md = EVP_MD_meth_new(NID_sha1, NID_sha1WithRSAEncryption)) == NULL
104            || !EVP_MD_meth_set_result_size(md, SHA_DIGEST_LENGTH)
105            || !EVP_MD_meth_set_input_blocksize(md, SHA_CBLOCK)
106            || !EVP_MD_meth_set_app_datasize(md,
107                                             sizeof(EVP_MD *) + sizeof(SHA_CTX))
108            || !EVP_MD_meth_set_flags(md, EVP_MD_FLAG_DIGALGID_ABSENT)
109            || !EVP_MD_meth_set_init(md, digest_sha1_init)
110            || !EVP_MD_meth_set_update(md, digest_sha1_update)
111            || !EVP_MD_meth_set_final(md, digest_sha1_final)) {
112            EVP_MD_meth_free(md);
113            md = NULL;
114        }
115        _hidden_sha1_md = md;
116    }
117    return _hidden_sha1_md;
118}
119
120/* SHA256 */
121static int digest_sha256_init(EVP_MD_CTX *ctx);
122static int digest_sha256_update(EVP_MD_CTX *ctx, const void *data,
123                                size_t count);
124static int digest_sha256_final(EVP_MD_CTX *ctx, unsigned char *md);
125
126static EVP_MD *_hidden_sha256_md = NULL;
127static const EVP_MD *digest_sha256(void)
128{
129    if (_hidden_sha256_md == NULL) {
130        EVP_MD *md;
131
132        if ((md = EVP_MD_meth_new(NID_sha256, NID_sha256WithRSAEncryption)) == NULL
133            || !EVP_MD_meth_set_result_size(md, SHA256_DIGEST_LENGTH)
134            || !EVP_MD_meth_set_input_blocksize(md, SHA256_CBLOCK)
135            || !EVP_MD_meth_set_app_datasize(md,
136                                             sizeof(EVP_MD *) + sizeof(SHA256_CTX))
137            || !EVP_MD_meth_set_flags(md, EVP_MD_FLAG_DIGALGID_ABSENT)
138            || !EVP_MD_meth_set_init(md, digest_sha256_init)
139            || !EVP_MD_meth_set_update(md, digest_sha256_update)
140            || !EVP_MD_meth_set_final(md, digest_sha256_final)) {
141            EVP_MD_meth_free(md);
142            md = NULL;
143        }
144        _hidden_sha256_md = md;
145    }
146    return _hidden_sha256_md;
147}
148
149/* SHA384/SHA512 */
150static int digest_sha384_init(EVP_MD_CTX *ctx);
151static int digest_sha384_update(EVP_MD_CTX *ctx, const void *data,
152                                size_t count);
153static int digest_sha384_final(EVP_MD_CTX *ctx, unsigned char *md);
154
155static int digest_sha512_init(EVP_MD_CTX *ctx);
156static int digest_sha512_update(EVP_MD_CTX *ctx, const void *data,
157                                size_t count);
158static int digest_sha512_final(EVP_MD_CTX *ctx, unsigned char *md);
159
160static EVP_MD *_hidden_sha384_md = NULL;
161static const EVP_MD *digest_sha384(void)
162{
163    if (_hidden_sha384_md == NULL) {
164        EVP_MD *md;
165
166        if ((md = EVP_MD_meth_new(NID_sha384, NID_sha384WithRSAEncryption)) == NULL
167            || !EVP_MD_meth_set_result_size(md, SHA384_DIGEST_LENGTH)
168            || !EVP_MD_meth_set_input_blocksize(md, SHA512_CBLOCK)
169            || !EVP_MD_meth_set_app_datasize(md,
170                                             sizeof(EVP_MD *) + sizeof(SHA512_CTX))
171            || !EVP_MD_meth_set_flags(md, EVP_MD_FLAG_DIGALGID_ABSENT)
172            || !EVP_MD_meth_set_init(md, digest_sha384_init)
173            || !EVP_MD_meth_set_update(md, digest_sha384_update)
174            || !EVP_MD_meth_set_final(md, digest_sha384_final)) {
175            EVP_MD_meth_free(md);
176            md = NULL;
177        }
178        _hidden_sha384_md = md;
179    }
180    return _hidden_sha384_md;
181}
182static EVP_MD *_hidden_sha512_md = NULL;
183static const EVP_MD *digest_sha512(void)
184{
185    if (_hidden_sha512_md == NULL) {
186        EVP_MD *md;
187
188        if ((md = EVP_MD_meth_new(NID_sha512, NID_sha512WithRSAEncryption)) == NULL
189            || !EVP_MD_meth_set_result_size(md, SHA512_DIGEST_LENGTH)
190            || !EVP_MD_meth_set_input_blocksize(md, SHA512_CBLOCK)
191            || !EVP_MD_meth_set_app_datasize(md,
192                                             sizeof(EVP_MD *) + sizeof(SHA512_CTX))
193            || !EVP_MD_meth_set_flags(md, EVP_MD_FLAG_DIGALGID_ABSENT)
194            || !EVP_MD_meth_set_init(md, digest_sha512_init)
195            || !EVP_MD_meth_set_update(md, digest_sha512_update)
196            || !EVP_MD_meth_set_final(md, digest_sha512_final)) {
197            EVP_MD_meth_free(md);
198            md = NULL;
199        }
200        _hidden_sha512_md = md;
201    }
202    return _hidden_sha512_md;
203}
204static void destroy_digests(void)
205{
206    EVP_MD_meth_free(_hidden_md5_md);
207    _hidden_md5_md = NULL;
208    EVP_MD_meth_free(_hidden_sha1_md);
209    _hidden_sha1_md = NULL;
210    EVP_MD_meth_free(_hidden_sha256_md);
211    _hidden_sha256_md = NULL;
212    EVP_MD_meth_free(_hidden_sha384_md);
213    _hidden_sha384_md = NULL;
214    EVP_MD_meth_free(_hidden_sha512_md);
215    _hidden_sha512_md = NULL;
216}
217static int ossltest_digest_nids(const int **nids)
218{
219    static int digest_nids[6] = { 0, 0, 0, 0, 0, 0 };
220    static int pos = 0;
221    static int init = 0;
222
223    if (!init) {
224        const EVP_MD *md;
225        if ((md = digest_md5()) != NULL)
226            digest_nids[pos++] = EVP_MD_get_type(md);
227        if ((md = digest_sha1()) != NULL)
228            digest_nids[pos++] = EVP_MD_get_type(md);
229        if ((md = digest_sha256()) != NULL)
230            digest_nids[pos++] = EVP_MD_get_type(md);
231        if ((md = digest_sha384()) != NULL)
232            digest_nids[pos++] = EVP_MD_get_type(md);
233        if ((md = digest_sha512()) != NULL)
234            digest_nids[pos++] = EVP_MD_get_type(md);
235        digest_nids[pos] = 0;
236        init = 1;
237    }
238    *nids = digest_nids;
239    return pos;
240}
241
242/* Setup ciphers */
243static int ossltest_ciphers(ENGINE *, const EVP_CIPHER **,
244                            const int **, int);
245
246static int ossltest_cipher_nids[] = {
247    NID_aes_128_cbc, NID_aes_128_gcm,
248    NID_aes_128_cbc_hmac_sha1, 0
249};
250
251/* AES128 */
252
253static int ossltest_aes128_init_key(EVP_CIPHER_CTX *ctx,
254                                    const unsigned char *key,
255                                    const unsigned char *iv, int enc);
256static int ossltest_aes128_cbc_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
257                                      const unsigned char *in, size_t inl);
258static int ossltest_aes128_gcm_init_key(EVP_CIPHER_CTX *ctx,
259                                        const unsigned char *key,
260                                        const unsigned char *iv, int enc);
261static int ossltest_aes128_gcm_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
262                                      const unsigned char *in, size_t inl);
263static int ossltest_aes128_gcm_ctrl(EVP_CIPHER_CTX *ctx, int type, int arg,
264                                    void *ptr);
265static int ossltest_aes128_cbc_hmac_sha1_init_key(EVP_CIPHER_CTX *ctx,
266                                                  const unsigned char *key,
267                                                  const unsigned char *iv,
268                                                  int enc);
269static int ossltest_aes128_cbc_hmac_sha1_cipher(EVP_CIPHER_CTX *ctx,
270                                                unsigned char *out,
271                                                const unsigned char *in,
272                                                size_t inl);
273static int ossltest_aes128_cbc_hmac_sha1_ctrl(EVP_CIPHER_CTX *ctx, int type,
274                                              int arg, void *ptr);
275
276typedef struct {
277    size_t payload_length;      /* AAD length in decrypt case */
278    unsigned int tls_ver;
279} EVP_AES_HMAC_SHA1;
280
281static EVP_CIPHER *_hidden_aes_128_cbc = NULL;
282static const EVP_CIPHER *ossltest_aes_128_cbc(void)
283{
284    if (_hidden_aes_128_cbc == NULL
285        && ((_hidden_aes_128_cbc = EVP_CIPHER_meth_new(NID_aes_128_cbc,
286                                                       16 /* block size */,
287                                                       16 /* key len */)) == NULL
288            || !EVP_CIPHER_meth_set_iv_length(_hidden_aes_128_cbc,16)
289            || !EVP_CIPHER_meth_set_flags(_hidden_aes_128_cbc,
290                                          EVP_CIPH_FLAG_DEFAULT_ASN1
291                                          | EVP_CIPH_CBC_MODE)
292            || !EVP_CIPHER_meth_set_init(_hidden_aes_128_cbc,
293                                         ossltest_aes128_init_key)
294            || !EVP_CIPHER_meth_set_do_cipher(_hidden_aes_128_cbc,
295                                              ossltest_aes128_cbc_cipher)
296            || !EVP_CIPHER_meth_set_impl_ctx_size(_hidden_aes_128_cbc,
297                    EVP_CIPHER_impl_ctx_size(EVP_aes_128_cbc())))) {
298        EVP_CIPHER_meth_free(_hidden_aes_128_cbc);
299        _hidden_aes_128_cbc = NULL;
300    }
301    return _hidden_aes_128_cbc;
302}
303
304static EVP_CIPHER *_hidden_aes_128_gcm = NULL;
305
306#define AES_GCM_FLAGS   (EVP_CIPH_FLAG_DEFAULT_ASN1 \
307                | EVP_CIPH_CUSTOM_IV | EVP_CIPH_FLAG_CUSTOM_CIPHER \
308                | EVP_CIPH_ALWAYS_CALL_INIT | EVP_CIPH_CTRL_INIT \
309                | EVP_CIPH_CUSTOM_COPY |EVP_CIPH_FLAG_AEAD_CIPHER \
310                | EVP_CIPH_GCM_MODE)
311
312static const EVP_CIPHER *ossltest_aes_128_gcm(void)
313{
314    if (_hidden_aes_128_gcm == NULL
315        && ((_hidden_aes_128_gcm = EVP_CIPHER_meth_new(NID_aes_128_gcm,
316                                                       1 /* block size */,
317                                                       16 /* key len */)) == NULL
318            || !EVP_CIPHER_meth_set_iv_length(_hidden_aes_128_gcm,12)
319            || !EVP_CIPHER_meth_set_flags(_hidden_aes_128_gcm, AES_GCM_FLAGS)
320            || !EVP_CIPHER_meth_set_init(_hidden_aes_128_gcm,
321                                         ossltest_aes128_gcm_init_key)
322            || !EVP_CIPHER_meth_set_do_cipher(_hidden_aes_128_gcm,
323                                              ossltest_aes128_gcm_cipher)
324            || !EVP_CIPHER_meth_set_ctrl(_hidden_aes_128_gcm,
325                                              ossltest_aes128_gcm_ctrl)
326            || !EVP_CIPHER_meth_set_impl_ctx_size(_hidden_aes_128_gcm,
327                    EVP_CIPHER_impl_ctx_size(EVP_aes_128_gcm())))) {
328        EVP_CIPHER_meth_free(_hidden_aes_128_gcm);
329        _hidden_aes_128_gcm = NULL;
330    }
331    return _hidden_aes_128_gcm;
332}
333
334static EVP_CIPHER *_hidden_aes_128_cbc_hmac_sha1 = NULL;
335
336static const EVP_CIPHER *ossltest_aes_128_cbc_hmac_sha1(void)
337{
338    if (_hidden_aes_128_cbc_hmac_sha1 == NULL
339        && ((_hidden_aes_128_cbc_hmac_sha1
340             = EVP_CIPHER_meth_new(NID_aes_128_cbc_hmac_sha1,
341                                   16 /* block size */,
342                                   16 /* key len */)) == NULL
343            || !EVP_CIPHER_meth_set_iv_length(_hidden_aes_128_cbc_hmac_sha1,16)
344            || !EVP_CIPHER_meth_set_flags(_hidden_aes_128_cbc_hmac_sha1,
345                   EVP_CIPH_CBC_MODE | EVP_CIPH_FLAG_DEFAULT_ASN1 |
346                   EVP_CIPH_FLAG_AEAD_CIPHER)
347            || !EVP_CIPHER_meth_set_init(_hidden_aes_128_cbc_hmac_sha1,
348                   ossltest_aes128_cbc_hmac_sha1_init_key)
349            || !EVP_CIPHER_meth_set_do_cipher(_hidden_aes_128_cbc_hmac_sha1,
350                   ossltest_aes128_cbc_hmac_sha1_cipher)
351            || !EVP_CIPHER_meth_set_ctrl(_hidden_aes_128_cbc_hmac_sha1,
352                                         ossltest_aes128_cbc_hmac_sha1_ctrl)
353            || !EVP_CIPHER_meth_set_set_asn1_params(_hidden_aes_128_cbc_hmac_sha1,
354                   EVP_CIPH_FLAG_DEFAULT_ASN1 ? NULL : EVP_CIPHER_set_asn1_iv)
355            || !EVP_CIPHER_meth_set_get_asn1_params(_hidden_aes_128_cbc_hmac_sha1,
356                   EVP_CIPH_FLAG_DEFAULT_ASN1 ? NULL : EVP_CIPHER_get_asn1_iv)
357            || !EVP_CIPHER_meth_set_impl_ctx_size(_hidden_aes_128_cbc_hmac_sha1,
358                   sizeof(EVP_AES_HMAC_SHA1)))) {
359        EVP_CIPHER_meth_free(_hidden_aes_128_cbc_hmac_sha1);
360        _hidden_aes_128_cbc_hmac_sha1 = NULL;
361    }
362    return _hidden_aes_128_cbc_hmac_sha1;
363}
364
365static void destroy_ciphers(void)
366{
367    EVP_CIPHER_meth_free(_hidden_aes_128_cbc);
368    EVP_CIPHER_meth_free(_hidden_aes_128_gcm);
369    EVP_CIPHER_meth_free(_hidden_aes_128_cbc_hmac_sha1);
370    _hidden_aes_128_cbc = NULL;
371    _hidden_aes_128_gcm = NULL;
372    _hidden_aes_128_cbc_hmac_sha1 = NULL;
373}
374
375/* Key loading */
376static EVP_PKEY *load_key(ENGINE *eng, const char *key_id, int pub,
377                          UI_METHOD *ui_method, void *ui_data)
378{
379    BIO *in;
380    EVP_PKEY *key;
381
382    if (OPENSSL_strncasecmp(key_id, "ot:", 3) != 0)
383        return NULL;
384    key_id += 3;
385
386    fprintf(stderr, "[ossltest]Loading %s key %s\n",
387            pub ? "Public" : "Private", key_id);
388    in = BIO_new_file(key_id, "r");
389    if (!in)
390        return NULL;
391    if (pub)
392        key = PEM_read_bio_PUBKEY(in, NULL, 0, NULL);
393    else
394        key = PEM_read_bio_PrivateKey(in, NULL, 0, NULL);
395    BIO_free(in);
396    return key;
397}
398
399static EVP_PKEY *ossltest_load_privkey(ENGINE *eng, const char *key_id,
400                                       UI_METHOD *ui_method, void *ui_data)
401{
402    return load_key(eng, key_id, 0, ui_method, ui_data);
403}
404
405static EVP_PKEY *ossltest_load_pubkey(ENGINE *eng, const char *key_id,
406                                      UI_METHOD *ui_method, void *ui_data)
407{
408    return load_key(eng, key_id, 1, ui_method, ui_data);
409}
410
411
412static int bind_ossltest(ENGINE *e)
413{
414    /* Ensure the ossltest error handling is set up */
415    ERR_load_OSSLTEST_strings();
416
417    if (!ENGINE_set_id(e, engine_ossltest_id)
418        || !ENGINE_set_name(e, engine_ossltest_name)
419        || !ENGINE_set_digests(e, ossltest_digests)
420        || !ENGINE_set_ciphers(e, ossltest_ciphers)
421        || !ENGINE_set_RAND(e, ossltest_rand_method())
422        || !ENGINE_set_destroy_function(e, ossltest_destroy)
423        || !ENGINE_set_load_privkey_function(e, ossltest_load_privkey)
424        || !ENGINE_set_load_pubkey_function(e, ossltest_load_pubkey)
425        || !ENGINE_set_init_function(e, ossltest_init)
426        || !ENGINE_set_finish_function(e, ossltest_finish)) {
427        OSSLTESTerr(OSSLTEST_F_BIND_OSSLTEST, OSSLTEST_R_INIT_FAILED);
428        return 0;
429    }
430
431    return 1;
432}
433
434#ifndef OPENSSL_NO_DYNAMIC_ENGINE
435static int bind_helper(ENGINE *e, const char *id)
436{
437    if (id && (strcmp(id, engine_ossltest_id) != 0))
438        return 0;
439    if (!bind_ossltest(e))
440        return 0;
441    return 1;
442}
443
444IMPLEMENT_DYNAMIC_CHECK_FN()
445    IMPLEMENT_DYNAMIC_BIND_FN(bind_helper)
446#endif
447
448static ENGINE *engine_ossltest(void)
449{
450    ENGINE *ret = ENGINE_new();
451    if (ret == NULL)
452        return NULL;
453    if (!bind_ossltest(ret)) {
454        ENGINE_free(ret);
455        return NULL;
456    }
457    return ret;
458}
459
460void ENGINE_load_ossltest(void)
461{
462    /* Copied from eng_[openssl|dyn].c */
463    ENGINE *toadd = engine_ossltest();
464    if (!toadd)
465        return;
466    ENGINE_add(toadd);
467    ENGINE_free(toadd);
468    ERR_clear_error();
469}
470
471
472static int ossltest_init(ENGINE *e)
473{
474    return 1;
475}
476
477
478static int ossltest_finish(ENGINE *e)
479{
480    return 1;
481}
482
483
484static int ossltest_destroy(ENGINE *e)
485{
486    destroy_digests();
487    destroy_ciphers();
488    ERR_unload_OSSLTEST_strings();
489    return 1;
490}
491
492static int ossltest_digests(ENGINE *e, const EVP_MD **digest,
493                          const int **nids, int nid)
494{
495    int ok = 1;
496    if (!digest) {
497        /* We are returning a list of supported nids */
498        return ossltest_digest_nids(nids);
499    }
500    /* We are being asked for a specific digest */
501    switch (nid) {
502    case NID_md5:
503        *digest = digest_md5();
504        break;
505    case NID_sha1:
506        *digest = digest_sha1();
507        break;
508    case NID_sha256:
509        *digest = digest_sha256();
510        break;
511    case NID_sha384:
512        *digest = digest_sha384();
513        break;
514    case NID_sha512:
515        *digest = digest_sha512();
516        break;
517    default:
518        ok = 0;
519        *digest = NULL;
520        break;
521    }
522    return ok;
523}
524
525static int ossltest_ciphers(ENGINE *e, const EVP_CIPHER **cipher,
526                          const int **nids, int nid)
527{
528    int ok = 1;
529    if (!cipher) {
530        /* We are returning a list of supported nids */
531        *nids = ossltest_cipher_nids;
532        return (sizeof(ossltest_cipher_nids) - 1)
533               / sizeof(ossltest_cipher_nids[0]);
534    }
535    /* We are being asked for a specific cipher */
536    switch (nid) {
537    case NID_aes_128_cbc:
538        *cipher = ossltest_aes_128_cbc();
539        break;
540    case NID_aes_128_gcm:
541        *cipher = ossltest_aes_128_gcm();
542        break;
543    case NID_aes_128_cbc_hmac_sha1:
544        *cipher = ossltest_aes_128_cbc_hmac_sha1();
545        break;
546    default:
547        ok = 0;
548        *cipher = NULL;
549        break;
550    }
551    return ok;
552}
553
554static void fill_known_data(unsigned char *md, unsigned int len)
555{
556    unsigned int i;
557
558    for (i=0; i<len; i++) {
559        md[i] = (unsigned char)(i & 0xff);
560    }
561}
562
563/*
564 * MD5 implementation. We go through the motions of doing MD5 by deferring to
565 * the standard implementation. Then we overwrite the result with a will defined
566 * value, so that all "MD5" digests using the test engine always end up with
567 * the same value.
568 */
569static int digest_md5_init(EVP_MD_CTX *ctx)
570{
571   return EVP_MD_meth_get_init(EVP_md5())(ctx);
572}
573
574static int digest_md5_update(EVP_MD_CTX *ctx, const void *data,
575                             size_t count)
576{
577    return EVP_MD_meth_get_update(EVP_md5())(ctx, data, count);
578}
579
580static int digest_md5_final(EVP_MD_CTX *ctx, unsigned char *md)
581{
582    int ret = EVP_MD_meth_get_final(EVP_md5())(ctx, md);
583
584    if (ret > 0) {
585        fill_known_data(md, MD5_DIGEST_LENGTH);
586    }
587    return ret;
588}
589
590/*
591 * SHA1 implementation.
592 */
593static int digest_sha1_init(EVP_MD_CTX *ctx)
594{
595    return EVP_MD_meth_get_init(EVP_sha1())(ctx);
596}
597
598static int digest_sha1_update(EVP_MD_CTX *ctx, const void *data,
599                              size_t count)
600{
601    return EVP_MD_meth_get_update(EVP_sha1())(ctx, data, count);
602}
603
604static int digest_sha1_final(EVP_MD_CTX *ctx, unsigned char *md)
605{
606    int ret = EVP_MD_meth_get_final(EVP_sha1())(ctx, md);
607
608    if (ret > 0) {
609        fill_known_data(md, SHA_DIGEST_LENGTH);
610    }
611    return ret;
612}
613
614/*
615 * SHA256 implementation.
616 */
617static int digest_sha256_init(EVP_MD_CTX *ctx)
618{
619    return EVP_MD_meth_get_init(EVP_sha256())(ctx);
620}
621
622static int digest_sha256_update(EVP_MD_CTX *ctx, const void *data,
623                                size_t count)
624{
625    return EVP_MD_meth_get_update(EVP_sha256())(ctx, data, count);
626}
627
628static int digest_sha256_final(EVP_MD_CTX *ctx, unsigned char *md)
629{
630    int ret = EVP_MD_meth_get_final(EVP_sha256())(ctx, md);
631
632    if (ret > 0) {
633        fill_known_data(md, SHA256_DIGEST_LENGTH);
634    }
635    return ret;
636}
637
638/*
639 * SHA384 implementation.
640 */
641static int digest_sha384_init(EVP_MD_CTX *ctx)
642{
643    return EVP_MD_meth_get_init(EVP_sha384())(ctx);
644}
645
646static int digest_sha384_update(EVP_MD_CTX *ctx, const void *data,
647                                size_t count)
648{
649    return EVP_MD_meth_get_update(EVP_sha384())(ctx, data, count);
650}
651
652static int digest_sha384_final(EVP_MD_CTX *ctx, unsigned char *md)
653{
654    int ret = EVP_MD_meth_get_final(EVP_sha384())(ctx, md);
655
656    if (ret > 0) {
657        fill_known_data(md, SHA384_DIGEST_LENGTH);
658    }
659    return ret;
660}
661
662/*
663 * SHA512 implementation.
664 */
665static int digest_sha512_init(EVP_MD_CTX *ctx)
666{
667    return EVP_MD_meth_get_init(EVP_sha512())(ctx);
668}
669
670static int digest_sha512_update(EVP_MD_CTX *ctx, const void *data,
671                                size_t count)
672{
673    return EVP_MD_meth_get_update(EVP_sha512())(ctx, data, count);
674}
675
676static int digest_sha512_final(EVP_MD_CTX *ctx, unsigned char *md)
677{
678    int ret = EVP_MD_meth_get_final(EVP_sha512())(ctx, md);
679
680    if (ret > 0) {
681        fill_known_data(md, SHA512_DIGEST_LENGTH);
682    }
683    return ret;
684}
685
686/*
687 * AES128 Implementation
688 */
689
690static int ossltest_aes128_init_key(EVP_CIPHER_CTX *ctx,
691                                    const unsigned char *key,
692                                    const unsigned char *iv, int enc)
693{
694    return EVP_CIPHER_meth_get_init(EVP_aes_128_cbc()) (ctx, key, iv, enc);
695}
696
697static int ossltest_aes128_cbc_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
698                                      const unsigned char *in, size_t inl)
699{
700    unsigned char *tmpbuf;
701    int ret;
702
703    tmpbuf = OPENSSL_malloc(inl);
704
705    /* OPENSSL_malloc will return NULL if inl == 0 */
706    if (tmpbuf == NULL && inl > 0)
707        return -1;
708
709    /* Remember what we were asked to encrypt */
710    if (tmpbuf != NULL)
711        memcpy(tmpbuf, in, inl);
712
713    /* Go through the motions of encrypting it */
714    ret = EVP_CIPHER_meth_get_do_cipher(EVP_aes_128_cbc())(ctx, out, in, inl);
715
716    /* Throw it all away and just use the plaintext as the output */
717    if (tmpbuf != NULL)
718        memcpy(out, tmpbuf, inl);
719    OPENSSL_free(tmpbuf);
720
721    return ret;
722}
723
724static int ossltest_aes128_gcm_init_key(EVP_CIPHER_CTX *ctx,
725                                        const unsigned char *key,
726                                        const unsigned char *iv, int enc)
727{
728    return EVP_CIPHER_meth_get_init(EVP_aes_128_gcm()) (ctx, key, iv, enc);
729}
730
731static int ossltest_aes128_gcm_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
732                                      const unsigned char *in, size_t inl)
733{
734    unsigned char *tmpbuf = OPENSSL_malloc(inl);
735
736    /* OPENSSL_malloc will return NULL if inl == 0 */
737    if (tmpbuf == NULL && inl > 0)
738        return -1;
739
740    /* Remember what we were asked to encrypt */
741    if (tmpbuf != NULL)
742        memcpy(tmpbuf, in, inl);
743
744    /* Go through the motions of encrypting it */
745    EVP_CIPHER_meth_get_do_cipher(EVP_aes_128_gcm())(ctx, out, in, inl);
746
747    /* Throw it all away and just use the plaintext as the output */
748    if (tmpbuf != NULL && out != NULL)
749        memcpy(out, tmpbuf, inl);
750    OPENSSL_free(tmpbuf);
751
752    return inl;
753}
754
755static int ossltest_aes128_gcm_ctrl(EVP_CIPHER_CTX *ctx, int type, int arg,
756                                    void *ptr)
757{
758    /* Pass the ctrl down */
759    int ret = EVP_CIPHER_meth_get_ctrl(EVP_aes_128_gcm())(ctx, type, arg, ptr);
760
761    if (ret <= 0)
762        return ret;
763
764    switch(type) {
765    case EVP_CTRL_AEAD_GET_TAG:
766        /* Always give the same tag */
767        memset(ptr, 0, EVP_GCM_TLS_TAG_LEN);
768        break;
769
770    default:
771        break;
772    }
773
774    return 1;
775}
776
777#define NO_PAYLOAD_LENGTH       ((size_t)-1)
778# define data(ctx) ((EVP_AES_HMAC_SHA1 *)EVP_CIPHER_CTX_get_cipher_data(ctx))
779
780static int ossltest_aes128_cbc_hmac_sha1_init_key(EVP_CIPHER_CTX *ctx,
781                                                  const unsigned char *inkey,
782                                                  const unsigned char *iv,
783                                                  int enc)
784{
785    EVP_AES_HMAC_SHA1 *key = data(ctx);
786    key->payload_length = NO_PAYLOAD_LENGTH;
787    return 1;
788}
789
790static int ossltest_aes128_cbc_hmac_sha1_cipher(EVP_CIPHER_CTX *ctx,
791                                                unsigned char *out,
792                                                const unsigned char *in,
793                                                size_t len)
794{
795    EVP_AES_HMAC_SHA1 *key = data(ctx);
796    unsigned int l;
797    size_t plen = key->payload_length;
798
799    key->payload_length = NO_PAYLOAD_LENGTH;
800
801    if (len % AES_BLOCK_SIZE)
802        return 0;
803
804    if (EVP_CIPHER_CTX_is_encrypting(ctx)) {
805        if (plen == NO_PAYLOAD_LENGTH)
806            plen = len;
807        else if (len !=
808                 ((plen + SHA_DIGEST_LENGTH +
809                   AES_BLOCK_SIZE) & -AES_BLOCK_SIZE))
810            return 0;
811
812        memmove(out, in, plen);
813
814        if (plen != len) {      /* "TLS" mode of operation */
815            /* calculate HMAC and append it to payload */
816            fill_known_data(out + plen, SHA_DIGEST_LENGTH);
817
818            /* pad the payload|hmac */
819            plen += SHA_DIGEST_LENGTH;
820            for (l = len - plen - 1; plen < len; plen++)
821                out[plen] = l;
822        }
823    } else {
824        /* decrypt HMAC|padding at once */
825        memmove(out, in, len);
826
827        if (plen != NO_PAYLOAD_LENGTH) { /* "TLS" mode of operation */
828            unsigned int maxpad, pad;
829
830            if (key->tls_ver >= TLS1_1_VERSION) {
831                if (len < (AES_BLOCK_SIZE + SHA_DIGEST_LENGTH + 1))
832                    return 0;
833
834                /* omit explicit iv */
835                in += AES_BLOCK_SIZE;
836                out += AES_BLOCK_SIZE;
837                len -= AES_BLOCK_SIZE;
838            } else if (len < (SHA_DIGEST_LENGTH + 1))
839                return 0;
840
841            /* figure out payload length */
842            pad = out[len - 1];
843            maxpad = len - (SHA_DIGEST_LENGTH + 1);
844            if (pad > maxpad)
845                return 0;
846            for (plen = len - pad - 1; plen < len; plen++)
847                if (out[plen] != pad)
848                    return 0;
849        }
850    }
851
852    return 1;
853}
854
855static int ossltest_aes128_cbc_hmac_sha1_ctrl(EVP_CIPHER_CTX *ctx, int type,
856                                              int arg, void *ptr)
857{
858    EVP_AES_HMAC_SHA1 *key = data(ctx);
859
860    switch (type) {
861    case EVP_CTRL_AEAD_SET_MAC_KEY:
862        return 1;
863
864    case EVP_CTRL_AEAD_TLS1_AAD:
865        {
866            unsigned char *p = ptr;
867            unsigned int len;
868
869            if (arg != EVP_AEAD_TLS1_AAD_LEN)
870                return -1;
871
872            len = p[arg - 2] << 8 | p[arg - 1];
873            key->tls_ver = p[arg - 4] << 8 | p[arg - 3];
874
875            if (EVP_CIPHER_CTX_is_encrypting(ctx)) {
876                key->payload_length = len;
877                if (key->tls_ver >= TLS1_1_VERSION) {
878                    if (len < AES_BLOCK_SIZE)
879                        return 0;
880                    len -= AES_BLOCK_SIZE;
881                    p[arg - 2] = len >> 8;
882                    p[arg - 1] = len;
883                }
884
885                return (int)(((len + SHA_DIGEST_LENGTH +
886                               AES_BLOCK_SIZE) & -AES_BLOCK_SIZE)
887                             - len);
888            } else {
889                key->payload_length = arg;
890
891                return SHA_DIGEST_LENGTH;
892            }
893        }
894    default:
895        return -1;
896    }
897}
898
899static int ossltest_rand_bytes(unsigned char *buf, int num)
900{
901    unsigned char val = 1;
902
903    while (--num >= 0)
904        *buf++ = val++;
905    return 1;
906}
907
908static int ossltest_rand_status(void)
909{
910    return 1;
911}
912
913static const RAND_METHOD *ossltest_rand_method(void)
914{
915
916    static RAND_METHOD osslt_rand_meth = {
917        NULL,
918        ossltest_rand_bytes,
919        NULL,
920        NULL,
921        ossltest_rand_bytes,
922        ossltest_rand_status
923    };
924
925    return &osslt_rand_meth;
926}
927