1*1dcdf01fSchristos /*
2*1dcdf01fSchristos * Copyright 2016-2021 The OpenSSL Project Authors. All Rights Reserved.
3*1dcdf01fSchristos *
4*1dcdf01fSchristos * Licensed under the OpenSSL license (the "License"). You may not use
5*1dcdf01fSchristos * this file except in compliance with the License. You can obtain a copy
6*1dcdf01fSchristos * in the file LICENSE in the source distribution or at
7*1dcdf01fSchristos * https://www.openssl.org/source/license.html
8*1dcdf01fSchristos */
9*1dcdf01fSchristos
10*1dcdf01fSchristos #include <stdio.h>
11*1dcdf01fSchristos #include <openssl/opensslconf.h>
12*1dcdf01fSchristos
13*1dcdf01fSchristos #include <string.h>
14*1dcdf01fSchristos #include <openssl/engine.h>
15*1dcdf01fSchristos #include <openssl/evp.h>
16*1dcdf01fSchristos #include <openssl/rand.h>
17*1dcdf01fSchristos #include "testutil.h"
18*1dcdf01fSchristos
19*1dcdf01fSchristos /* Use a buffer size which is not aligned to block size */
20*1dcdf01fSchristos #define BUFFER_SIZE 17
21*1dcdf01fSchristos
22*1dcdf01fSchristos #ifndef OPENSSL_NO_ENGINE
23*1dcdf01fSchristos static ENGINE *e;
24*1dcdf01fSchristos
test_afalg_aes_cbc(int keysize_idx)25*1dcdf01fSchristos static int test_afalg_aes_cbc(int keysize_idx)
26*1dcdf01fSchristos {
27*1dcdf01fSchristos EVP_CIPHER_CTX *ctx;
28*1dcdf01fSchristos const EVP_CIPHER *cipher;
29*1dcdf01fSchristos unsigned char key[] = "\x06\xa9\x21\x40\x36\xb8\xa1\x5b"
30*1dcdf01fSchristos "\x51\x2e\x03\xd5\x34\x12\x00\x06"
31*1dcdf01fSchristos "\x06\xa9\x21\x40\x36\xb8\xa1\x5b"
32*1dcdf01fSchristos "\x51\x2e\x03\xd5\x34\x12\x00\x06";
33*1dcdf01fSchristos unsigned char iv[] = "\x3d\xaf\xba\x42\x9d\x9e\xb4\x30"
34*1dcdf01fSchristos "\xb4\x22\xda\x80\x2c\x9f\xac\x41";
35*1dcdf01fSchristos /* input = "Single block msg\n" 17Bytes*/
36*1dcdf01fSchristos unsigned char in[BUFFER_SIZE] = "\x53\x69\x6e\x67\x6c\x65\x20\x62"
37*1dcdf01fSchristos "\x6c\x6f\x63\x6b\x20\x6d\x73\x67\x0a";
38*1dcdf01fSchristos unsigned char ebuf[BUFFER_SIZE + 32];
39*1dcdf01fSchristos unsigned char dbuf[BUFFER_SIZE + 32];
40*1dcdf01fSchristos unsigned char encresult_128[] = "\xe3\x53\x77\x9c\x10\x79\xae\xb8"
41*1dcdf01fSchristos "\x27\x08\x94\x2d\xbe\x77\x18\x1a\x2d";
42*1dcdf01fSchristos unsigned char encresult_192[] = "\xf7\xe4\x26\xd1\xd5\x4f\x8f\x39"
43*1dcdf01fSchristos "\xb1\x9e\xe0\xdf\x61\xb9\xc2\x55\xeb";
44*1dcdf01fSchristos unsigned char encresult_256[] = "\xa0\x76\x85\xfd\xc1\x65\x71\x9d"
45*1dcdf01fSchristos "\xc7\xe9\x13\x6e\xae\x55\x49\xb4\x13";
46*1dcdf01fSchristos unsigned char *enc_result = NULL;
47*1dcdf01fSchristos
48*1dcdf01fSchristos int encl, encf, decl, decf;
49*1dcdf01fSchristos int ret = 0;
50*1dcdf01fSchristos
51*1dcdf01fSchristos switch (keysize_idx) {
52*1dcdf01fSchristos case 0:
53*1dcdf01fSchristos cipher = EVP_aes_128_cbc();
54*1dcdf01fSchristos enc_result = &encresult_128[0];
55*1dcdf01fSchristos break;
56*1dcdf01fSchristos case 1:
57*1dcdf01fSchristos cipher = EVP_aes_192_cbc();
58*1dcdf01fSchristos enc_result = &encresult_192[0];
59*1dcdf01fSchristos break;
60*1dcdf01fSchristos case 2:
61*1dcdf01fSchristos cipher = EVP_aes_256_cbc();
62*1dcdf01fSchristos enc_result = &encresult_256[0];
63*1dcdf01fSchristos break;
64*1dcdf01fSchristos default:
65*1dcdf01fSchristos cipher = NULL;
66*1dcdf01fSchristos }
67*1dcdf01fSchristos if (!TEST_ptr(ctx = EVP_CIPHER_CTX_new()))
68*1dcdf01fSchristos return 0;
69*1dcdf01fSchristos
70*1dcdf01fSchristos if (!TEST_true(EVP_CipherInit_ex(ctx, cipher, e, key, iv, 1))
71*1dcdf01fSchristos || !TEST_true(EVP_CipherUpdate(ctx, ebuf, &encl, in, BUFFER_SIZE))
72*1dcdf01fSchristos || !TEST_true(EVP_CipherFinal_ex(ctx, ebuf+encl, &encf)))
73*1dcdf01fSchristos goto end;
74*1dcdf01fSchristos encl += encf;
75*1dcdf01fSchristos
76*1dcdf01fSchristos if (!TEST_mem_eq(enc_result, BUFFER_SIZE, ebuf, BUFFER_SIZE))
77*1dcdf01fSchristos goto end;
78*1dcdf01fSchristos
79*1dcdf01fSchristos if (!TEST_true(EVP_CIPHER_CTX_reset(ctx))
80*1dcdf01fSchristos || !TEST_true(EVP_CipherInit_ex(ctx, cipher, e, key, iv, 0))
81*1dcdf01fSchristos || !TEST_true(EVP_CipherUpdate(ctx, dbuf, &decl, ebuf, encl))
82*1dcdf01fSchristos || !TEST_true(EVP_CipherFinal_ex(ctx, dbuf+decl, &decf)))
83*1dcdf01fSchristos goto end;
84*1dcdf01fSchristos decl += decf;
85*1dcdf01fSchristos
86*1dcdf01fSchristos if (!TEST_int_eq(decl, BUFFER_SIZE)
87*1dcdf01fSchristos || !TEST_mem_eq(dbuf, BUFFER_SIZE, in, BUFFER_SIZE))
88*1dcdf01fSchristos goto end;
89*1dcdf01fSchristos
90*1dcdf01fSchristos ret = 1;
91*1dcdf01fSchristos
92*1dcdf01fSchristos end:
93*1dcdf01fSchristos EVP_CIPHER_CTX_free(ctx);
94*1dcdf01fSchristos return ret;
95*1dcdf01fSchristos }
96*1dcdf01fSchristos
test_pr16743(void)97*1dcdf01fSchristos static int test_pr16743(void)
98*1dcdf01fSchristos {
99*1dcdf01fSchristos int ret = 0;
100*1dcdf01fSchristos const EVP_CIPHER * cipher;
101*1dcdf01fSchristos EVP_CIPHER_CTX *ctx;
102*1dcdf01fSchristos
103*1dcdf01fSchristos if (!TEST_true(ENGINE_init(e)))
104*1dcdf01fSchristos return 0;
105*1dcdf01fSchristos cipher = ENGINE_get_cipher(e, NID_aes_128_cbc);
106*1dcdf01fSchristos ctx = EVP_CIPHER_CTX_new();
107*1dcdf01fSchristos if (cipher != NULL && ctx != NULL)
108*1dcdf01fSchristos ret = EVP_EncryptInit_ex(ctx, cipher, e, NULL, NULL);
109*1dcdf01fSchristos TEST_true(ret);
110*1dcdf01fSchristos EVP_CIPHER_CTX_free(ctx);
111*1dcdf01fSchristos ENGINE_finish(e);
112*1dcdf01fSchristos return ret;
113*1dcdf01fSchristos }
114*1dcdf01fSchristos
global_init(void)115*1dcdf01fSchristos int global_init(void)
116*1dcdf01fSchristos {
117*1dcdf01fSchristos ENGINE_load_builtin_engines();
118*1dcdf01fSchristos # ifndef OPENSSL_NO_STATIC_ENGINE
119*1dcdf01fSchristos OPENSSL_init_crypto(OPENSSL_INIT_ENGINE_AFALG, NULL);
120*1dcdf01fSchristos # endif
121*1dcdf01fSchristos return 1;
122*1dcdf01fSchristos }
123*1dcdf01fSchristos #endif
124*1dcdf01fSchristos
setup_tests(void)125*1dcdf01fSchristos int setup_tests(void)
126*1dcdf01fSchristos {
127*1dcdf01fSchristos #ifndef OPENSSL_NO_ENGINE
128*1dcdf01fSchristos if ((e = ENGINE_by_id("afalg")) == NULL) {
129*1dcdf01fSchristos /* Probably a platform env issue, not a test failure. */
130*1dcdf01fSchristos TEST_info("Can't load AFALG engine");
131*1dcdf01fSchristos } else {
132*1dcdf01fSchristos ADD_ALL_TESTS(test_afalg_aes_cbc, 3);
133*1dcdf01fSchristos ADD_TEST(test_pr16743);
134*1dcdf01fSchristos }
135*1dcdf01fSchristos #endif
136*1dcdf01fSchristos
137*1dcdf01fSchristos return 1;
138*1dcdf01fSchristos }
139*1dcdf01fSchristos
140*1dcdf01fSchristos #ifndef OPENSSL_NO_ENGINE
cleanup_tests(void)141*1dcdf01fSchristos void cleanup_tests(void)
142*1dcdf01fSchristos {
143*1dcdf01fSchristos ENGINE_free(e);
144*1dcdf01fSchristos }
145*1dcdf01fSchristos #endif
146