blob: cfd9d8ca5a933b9795a01a49b7a5d701129390f9 [file] [log] [blame]
Philippe Reynescb69a812020-01-06 15:22:36 +01001// SPDX-License-Identifier: GPL-2.0+
2/*
3 * Copyright (c) 2019 Philippe Reynes <philippe.reynes@softathome.com>
4 *
5 * Unit tests for aes functions
6 */
7
Philippe Reynescb69a812020-01-06 15:22:36 +01008#include <command.h>
9#include <hexdump.h>
Simon Glass274e0b02020-05-10 11:39:56 -060010#include <rand.h>
Philippe Reynescb69a812020-01-06 15:22:36 +010011#include <uboot_aes.h>
12#include <test/lib.h>
13#include <test/test.h>
14#include <test/ut.h>
15
16#define TEST_AES_ONE_BLOCK 0
17#define TEST_AES_CBC_CHAIN 1
18
19struct test_aes_s {
20 int key_len;
21 int key_exp_len;
22 int type;
23 int num_block;
24};
25
26static struct test_aes_s test_aes[] = {
27 { AES128_KEY_LENGTH, AES128_EXPAND_KEY_LENGTH, TEST_AES_ONE_BLOCK, 1 },
28 { AES128_KEY_LENGTH, AES128_EXPAND_KEY_LENGTH, TEST_AES_CBC_CHAIN, 16 },
Philippe Reynesa6abf662020-01-06 15:22:37 +010029 { AES192_KEY_LENGTH, AES192_EXPAND_KEY_LENGTH, TEST_AES_ONE_BLOCK, 1 },
30 { AES192_KEY_LENGTH, AES192_EXPAND_KEY_LENGTH, TEST_AES_CBC_CHAIN, 16 },
31 { AES256_KEY_LENGTH, AES256_EXPAND_KEY_LENGTH, TEST_AES_ONE_BLOCK, 1 },
32 { AES256_KEY_LENGTH, AES256_EXPAND_KEY_LENGTH, TEST_AES_CBC_CHAIN, 16 },
Philippe Reynescb69a812020-01-06 15:22:36 +010033};
34
35static void rand_buf(u8 *buf, int size)
36{
37 int i;
38
39 for (i = 0; i < size; i++)
40 buf[i] = rand() & 0xff;
41}
42
43static int lib_test_aes_one_block(struct unit_test_state *uts, int key_len,
44 u8 *key_exp, u8 *iv, int num_block,
45 u8 *nocipher, u8 *ciphered, u8 *uncipher)
46{
47 aes_encrypt(key_len, nocipher, key_exp, ciphered);
48 aes_decrypt(key_len, ciphered, key_exp, uncipher);
49
50 ut_asserteq_mem(nocipher, uncipher, AES_BLOCK_LENGTH);
51
52 /* corrupt the expanded key */
53 key_exp[0]++;
54 aes_decrypt(key_len, ciphered, key_exp, uncipher);
55 ut_assertf(memcmp(nocipher, uncipher, AES_BLOCK_LENGTH),
56 "nocipher and uncipher should be different\n");
57
58 return 0;
59}
60
61static int lib_test_aes_cbc_chain(struct unit_test_state *uts, int key_len,
62 u8 *key_exp, u8 *iv, int num_block,
63 u8 *nocipher, u8 *ciphered, u8 *uncipher)
64{
65 aes_cbc_encrypt_blocks(key_len, key_exp, iv,
66 nocipher, ciphered, num_block);
67 aes_cbc_decrypt_blocks(key_len, key_exp, iv,
68 ciphered, uncipher, num_block);
69
70 ut_asserteq_mem(nocipher, uncipher, num_block * AES_BLOCK_LENGTH);
71
72 /* corrupt the expanded key */
73 key_exp[0]++;
74 aes_cbc_decrypt_blocks(key_len, key_exp, iv,
75 ciphered, uncipher, num_block);
76 ut_assertf(memcmp(nocipher, uncipher, num_block * AES_BLOCK_LENGTH),
77 "nocipher and uncipher should be different\n");
78
79 return 0;
80}
81
82static int _lib_test_aes_run(struct unit_test_state *uts, int key_len,
83 int key_exp_len, int type, int num_block)
84{
85 u8 *key, *key_exp, *iv;
86 u8 *nocipher, *ciphered, *uncipher;
87 int ret;
88
89 /* Allocate all the buffer */
90 key = malloc(key_len);
Philippe Reynescb69a812020-01-06 15:22:36 +010091 key_exp = malloc(key_exp_len);
Philippe Reynescb69a812020-01-06 15:22:36 +010092 iv = malloc(AES_BLOCK_LENGTH);
Philippe Reynescb69a812020-01-06 15:22:36 +010093 nocipher = malloc(num_block * AES_BLOCK_LENGTH);
Philippe Reynescb69a812020-01-06 15:22:36 +010094 ciphered = malloc((num_block + 1) * AES_BLOCK_LENGTH);
Philippe Reynescb69a812020-01-06 15:22:36 +010095 uncipher = malloc((num_block + 1) * AES_BLOCK_LENGTH);
Philippe Reynes08bb9b72020-02-06 17:12:59 +010096
97 if (!key || !key_exp || !iv || !nocipher || !ciphered || !uncipher) {
98 printf("%s: can't allocate memory\n", __func__);
99 ret = -1;
100 goto out;
101 }
Philippe Reynescb69a812020-01-06 15:22:36 +0100102
103 /* Initialize all buffer */
104 rand_buf(key, key_len);
105 rand_buf(iv, AES_BLOCK_LENGTH);
106 rand_buf(nocipher, num_block * AES_BLOCK_LENGTH);
107 memset(ciphered, 0, (num_block + 1) * AES_BLOCK_LENGTH);
108 memset(uncipher, 0, (num_block + 1) * AES_BLOCK_LENGTH);
109
110 /* Expand the key */
111 aes_expand_key(key, key_len, key_exp);
112
113 /* Encrypt and decrypt */
114 switch (type) {
115 case TEST_AES_ONE_BLOCK:
116 ret = lib_test_aes_one_block(uts, key_len, key_exp, iv,
117 num_block, nocipher,
118 ciphered, uncipher);
119 break;
120 case TEST_AES_CBC_CHAIN:
121 ret = lib_test_aes_cbc_chain(uts, key_len, key_exp, iv,
122 num_block, nocipher,
123 ciphered, uncipher);
124 break;
125 default:
126 printf("%s: unknown type (type=%d)\n", __func__, type);
127 ret = -1;
128 };
129
Philippe Reynes08bb9b72020-02-06 17:12:59 +0100130 out:
Philippe Reynescb69a812020-01-06 15:22:36 +0100131 /* Free all the data */
132 free(key);
133 free(key_exp);
134 free(iv);
135 free(nocipher);
136 free(ciphered);
137 free(uncipher);
138
139 return ret;
140}
141
142static int lib_test_aes_run(struct unit_test_state *uts,
143 struct test_aes_s *test)
144{
145 int key_len = test->key_len;
146 int key_exp_len = test->key_exp_len;
147 int type = test->type;
148 int num_block = test->num_block;
149
150 return _lib_test_aes_run(uts, key_len, key_exp_len,
151 type, num_block);
152}
153
154static int lib_test_aes(struct unit_test_state *uts)
155{
156 int i, ret = 0;
157
158 for (i = 0; i < ARRAY_SIZE(test_aes); i++) {
159 ret = lib_test_aes_run(uts, &test_aes[i]);
160 if (ret)
161 break;
162 }
163
164 return ret;
165}
166
167LIB_TEST(lib_test_aes, 0);