blob: a7da0ac7c00c06494ef21f688d8f2d3924733eac [file] [log] [blame]
Paul Bakker33b43f12013-08-20 11:48:36 +02001/* BEGIN_HEADER */
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +00002#include "mbedtls/aes.h"
Paul Bakker33b43f12013-08-20 11:48:36 +02003/* END_HEADER */
Paul Bakker367dae42009-06-28 21:50:27 +00004
Paul Bakker33b43f12013-08-20 11:48:36 +02005/* BEGIN_DEPENDENCIES
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02006 * depends_on:MBEDTLS_AES_C
Paul Bakker33b43f12013-08-20 11:48:36 +02007 * END_DEPENDENCIES
8 */
Paul Bakker5690efc2011-05-26 13:16:06 +00009
Paul Bakker33b43f12013-08-20 11:48:36 +020010/* BEGIN_CASE */
11void aes_encrypt_ecb( char *hex_key_string, char *hex_src_string,
12 char *hex_dst_string, int setkey_result )
Paul Bakker367dae42009-06-28 21:50:27 +000013{
14 unsigned char key_str[100];
15 unsigned char src_str[100];
16 unsigned char dst_str[100];
17 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020018 mbedtls_aes_context ctx;
Paul Bakker69998dd2009-07-11 19:15:20 +000019 int key_len;
Paul Bakker367dae42009-06-28 21:50:27 +000020
21 memset(key_str, 0x00, 100);
22 memset(src_str, 0x00, 100);
23 memset(dst_str, 0x00, 100);
24 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020025 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +000026
Paul Bakker33b43f12013-08-20 11:48:36 +020027 key_len = unhexify( key_str, hex_key_string );
28 unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +000029
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020030 TEST_ASSERT( mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 ) == setkey_result );
Paul Bakker33b43f12013-08-20 11:48:36 +020031 if( setkey_result == 0 )
Paul Bakker2b222c82009-07-27 21:03:45 +000032 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020033 TEST_ASSERT( mbedtls_aes_crypt_ecb( &ctx, MBEDTLS_AES_ENCRYPT, src_str, output ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +000034 hexify( dst_str, output, 16 );
Paul Bakker367dae42009-06-28 21:50:27 +000035
Paul Bakker33b43f12013-08-20 11:48:36 +020036 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +000037 }
Paul Bakker8cfd9d82014-06-18 11:16:11 +020038
Paul Bakkerbd51b262014-07-10 15:26:12 +020039exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020040 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +000041}
Paul Bakker33b43f12013-08-20 11:48:36 +020042/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +000043
Paul Bakker33b43f12013-08-20 11:48:36 +020044/* BEGIN_CASE */
45void aes_decrypt_ecb( char *hex_key_string, char *hex_src_string,
46 char *hex_dst_string, int setkey_result )
Paul Bakker367dae42009-06-28 21:50:27 +000047{
48 unsigned char key_str[100];
49 unsigned char src_str[100];
50 unsigned char dst_str[100];
51 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020052 mbedtls_aes_context ctx;
Paul Bakker69998dd2009-07-11 19:15:20 +000053 int key_len;
Paul Bakker367dae42009-06-28 21:50:27 +000054
55 memset(key_str, 0x00, 100);
56 memset(src_str, 0x00, 100);
57 memset(dst_str, 0x00, 100);
58 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020059 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +000060
Paul Bakker33b43f12013-08-20 11:48:36 +020061 key_len = unhexify( key_str, hex_key_string );
62 unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +000063
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020064 TEST_ASSERT( mbedtls_aes_setkey_dec( &ctx, key_str, key_len * 8 ) == setkey_result );
Paul Bakker33b43f12013-08-20 11:48:36 +020065 if( setkey_result == 0 )
Paul Bakker2b222c82009-07-27 21:03:45 +000066 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020067 TEST_ASSERT( mbedtls_aes_crypt_ecb( &ctx, MBEDTLS_AES_DECRYPT, src_str, output ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +000068 hexify( dst_str, output, 16 );
Paul Bakker367dae42009-06-28 21:50:27 +000069
Paul Bakker33b43f12013-08-20 11:48:36 +020070 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +000071 }
Paul Bakker8cfd9d82014-06-18 11:16:11 +020072
Paul Bakkerbd51b262014-07-10 15:26:12 +020073exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020074 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +000075}
Paul Bakker33b43f12013-08-20 11:48:36 +020076/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +000077
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020078/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CBC */
Paul Bakker33b43f12013-08-20 11:48:36 +020079void aes_encrypt_cbc( char *hex_key_string, char *hex_iv_string,
80 char *hex_src_string, char *hex_dst_string,
81 int cbc_result )
Paul Bakker367dae42009-06-28 21:50:27 +000082{
83 unsigned char key_str[100];
84 unsigned char iv_str[100];
85 unsigned char src_str[100];
86 unsigned char dst_str[100];
87 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020088 mbedtls_aes_context ctx;
Paul Bakkerf3ccc682010-03-18 21:21:02 +000089 int key_len, data_len;
Paul Bakker367dae42009-06-28 21:50:27 +000090
91 memset(key_str, 0x00, 100);
92 memset(iv_str, 0x00, 100);
93 memset(src_str, 0x00, 100);
94 memset(dst_str, 0x00, 100);
95 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020096 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +000097
Paul Bakker33b43f12013-08-20 11:48:36 +020098 key_len = unhexify( key_str, hex_key_string );
99 unhexify( iv_str, hex_iv_string );
100 data_len = unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000101
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200102 mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 );
103 TEST_ASSERT( mbedtls_aes_crypt_cbc( &ctx, MBEDTLS_AES_ENCRYPT, data_len, iv_str, src_str, output ) == cbc_result );
Paul Bakker33b43f12013-08-20 11:48:36 +0200104 if( cbc_result == 0 )
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000105 {
106 hexify( dst_str, output, data_len );
Paul Bakker367dae42009-06-28 21:50:27 +0000107
Paul Bakker33b43f12013-08-20 11:48:36 +0200108 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000109 }
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200110
Paul Bakkerbd51b262014-07-10 15:26:12 +0200111exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200112 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000113}
Paul Bakker33b43f12013-08-20 11:48:36 +0200114/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000115
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200116/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CBC */
Paul Bakker33b43f12013-08-20 11:48:36 +0200117void aes_decrypt_cbc( char *hex_key_string, char *hex_iv_string,
118 char *hex_src_string, char *hex_dst_string,
119 int cbc_result )
Paul Bakker367dae42009-06-28 21:50:27 +0000120{
121 unsigned char key_str[100];
122 unsigned char iv_str[100];
123 unsigned char src_str[100];
124 unsigned char dst_str[100];
125 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200126 mbedtls_aes_context ctx;
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000127 int key_len, data_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000128
129 memset(key_str, 0x00, 100);
130 memset(iv_str, 0x00, 100);
131 memset(src_str, 0x00, 100);
132 memset(dst_str, 0x00, 100);
133 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200134 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000135
Paul Bakker33b43f12013-08-20 11:48:36 +0200136 key_len = unhexify( key_str, hex_key_string );
137 unhexify( iv_str, hex_iv_string );
138 data_len = unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000139
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200140 mbedtls_aes_setkey_dec( &ctx, key_str, key_len * 8 );
141 TEST_ASSERT( mbedtls_aes_crypt_cbc( &ctx, MBEDTLS_AES_DECRYPT, data_len, iv_str, src_str, output ) == cbc_result );
Paul Bakker33b43f12013-08-20 11:48:36 +0200142 if( cbc_result == 0)
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000143 {
144 hexify( dst_str, output, data_len );
Paul Bakker367dae42009-06-28 21:50:27 +0000145
Paul Bakker33b43f12013-08-20 11:48:36 +0200146 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000147 }
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200148
Paul Bakkerbd51b262014-07-10 15:26:12 +0200149exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200150 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000151}
Paul Bakker33b43f12013-08-20 11:48:36 +0200152/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000153
Aorimn5f778012016-06-09 23:22:58 +0200154/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_XTS */
Jaeden Amerocd9fc5e2018-05-30 15:23:24 +0100155void aes_encrypt_xts( char *hex_key_string, char *hex_data_unit_string,
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100156 char *hex_src_string, char *hex_dst_string )
Aorimn5f778012016-06-09 23:22:58 +0200157{
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100158 enum { AES_BLOCK_SIZE = 16 };
159 unsigned char *data_unit = NULL;
160 unsigned char *key = NULL;
161 unsigned char *src = NULL;
162 unsigned char *dst = NULL;
163 unsigned char *output = NULL;
Jaeden Amero9366feb2018-05-29 18:55:17 +0100164 mbedtls_aes_xts_context ctx;
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100165 size_t key_len, src_len, dst_len, data_unit_len;
Aorimn5f778012016-06-09 23:22:58 +0200166
Jaeden Amero9366feb2018-05-29 18:55:17 +0100167 mbedtls_aes_xts_init( &ctx );
Aorimn5f778012016-06-09 23:22:58 +0200168
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100169 data_unit = unhexify_alloc( hex_data_unit_string, &data_unit_len );
170 TEST_ASSERT( data_unit_len == AES_BLOCK_SIZE );
Aorimn5f778012016-06-09 23:22:58 +0200171
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100172 key = unhexify_alloc( hex_key_string, &key_len );
173 TEST_ASSERT( key_len % 2 == 0 );
Aorimn5f778012016-06-09 23:22:58 +0200174
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100175 src = unhexify_alloc( hex_src_string, &src_len );
176 dst = unhexify_alloc( hex_dst_string, &dst_len );
177 TEST_ASSERT( src_len == dst_len );
Aorimn5f778012016-06-09 23:22:58 +0200178
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100179 output = zero_alloc( dst_len );
180
181 TEST_ASSERT( mbedtls_aes_xts_setkey_enc( &ctx, key, key_len * 8 ) == 0 );
182 TEST_ASSERT( mbedtls_aes_crypt_xts( &ctx, MBEDTLS_AES_ENCRYPT, src_len,
183 data_unit, src, output ) == 0 );
184
185 TEST_ASSERT( memcmp( output, dst, dst_len ) == 0 );
Aorimn5f778012016-06-09 23:22:58 +0200186
187exit:
Jaeden Amero9366feb2018-05-29 18:55:17 +0100188 mbedtls_aes_xts_free( &ctx );
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100189 mbedtls_free( data_unit );
190 mbedtls_free( key );
191 mbedtls_free( src );
192 mbedtls_free( dst );
193 mbedtls_free( output );
Aorimn5f778012016-06-09 23:22:58 +0200194}
195/* END_CASE */
196
197/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_XTS */
Jaeden Amerocd9fc5e2018-05-30 15:23:24 +0100198void aes_decrypt_xts( char *hex_key_string, char *hex_data_unit_string,
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100199 char *hex_dst_string, char *hex_src_string )
Aorimn5f778012016-06-09 23:22:58 +0200200{
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100201 enum { AES_BLOCK_SIZE = 16 };
202 unsigned char *data_unit = NULL;
203 unsigned char *key = NULL;
204 unsigned char *src = NULL;
205 unsigned char *dst = NULL;
206 unsigned char *output = NULL;
Jaeden Amero9366feb2018-05-29 18:55:17 +0100207 mbedtls_aes_xts_context ctx;
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100208 size_t key_len, src_len, dst_len, data_unit_len;
Aorimn5f778012016-06-09 23:22:58 +0200209
Jaeden Amero9366feb2018-05-29 18:55:17 +0100210 mbedtls_aes_xts_init( &ctx );
Aorimn5f778012016-06-09 23:22:58 +0200211
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100212 data_unit = unhexify_alloc( hex_data_unit_string, &data_unit_len );
213 TEST_ASSERT( data_unit_len == AES_BLOCK_SIZE );
Aorimn5f778012016-06-09 23:22:58 +0200214
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100215 key = unhexify_alloc( hex_key_string, &key_len );
216 TEST_ASSERT( key_len % 2 == 0 );
Aorimn5f778012016-06-09 23:22:58 +0200217
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100218 src = unhexify_alloc( hex_src_string, &src_len );
219 dst = unhexify_alloc( hex_dst_string, &dst_len );
220 TEST_ASSERT( src_len == dst_len );
Aorimn5f778012016-06-09 23:22:58 +0200221
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100222 output = zero_alloc( dst_len );
223
224 TEST_ASSERT( mbedtls_aes_xts_setkey_dec( &ctx, key, key_len * 8 ) == 0 );
225 TEST_ASSERT( mbedtls_aes_crypt_xts( &ctx, MBEDTLS_AES_DECRYPT, src_len,
226 data_unit, src, output ) == 0 );
227
228 TEST_ASSERT( memcmp( output, dst, dst_len ) == 0 );
Aorimn5f778012016-06-09 23:22:58 +0200229
230exit:
Jaeden Amero9366feb2018-05-29 18:55:17 +0100231 mbedtls_aes_xts_free( &ctx );
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100232 mbedtls_free( data_unit );
233 mbedtls_free( key );
234 mbedtls_free( src );
235 mbedtls_free( dst );
236 mbedtls_free( output );
Aorimn5f778012016-06-09 23:22:58 +0200237}
238/* END_CASE */
239
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200240/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */
Paul Bakker33b43f12013-08-20 11:48:36 +0200241void aes_encrypt_cfb128( char *hex_key_string, char *hex_iv_string,
242 char *hex_src_string, char *hex_dst_string )
Paul Bakker367dae42009-06-28 21:50:27 +0000243{
244 unsigned char key_str[100];
245 unsigned char iv_str[100];
246 unsigned char src_str[100];
247 unsigned char dst_str[100];
248 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200249 mbedtls_aes_context ctx;
Paul Bakkercd43a0b2011-06-09 13:55:44 +0000250 size_t iv_offset = 0;
Paul Bakker69998dd2009-07-11 19:15:20 +0000251 int key_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000252
253 memset(key_str, 0x00, 100);
254 memset(iv_str, 0x00, 100);
255 memset(src_str, 0x00, 100);
256 memset(dst_str, 0x00, 100);
257 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200258 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000259
Paul Bakker33b43f12013-08-20 11:48:36 +0200260 key_len = unhexify( key_str, hex_key_string );
261 unhexify( iv_str, hex_iv_string );
262 unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000263
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200264 mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 );
265 TEST_ASSERT( mbedtls_aes_crypt_cfb128( &ctx, MBEDTLS_AES_ENCRYPT, 16, &iv_offset, iv_str, src_str, output ) == 0 );
Paul Bakker367dae42009-06-28 21:50:27 +0000266 hexify( dst_str, output, 16 );
267
Paul Bakker33b43f12013-08-20 11:48:36 +0200268 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200269
Paul Bakkerbd51b262014-07-10 15:26:12 +0200270exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200271 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000272}
Paul Bakker33b43f12013-08-20 11:48:36 +0200273/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000274
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200275/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */
Paul Bakker33b43f12013-08-20 11:48:36 +0200276void aes_decrypt_cfb128( char *hex_key_string, char *hex_iv_string,
277 char *hex_src_string, char *hex_dst_string )
Paul Bakker367dae42009-06-28 21:50:27 +0000278{
279 unsigned char key_str[100];
280 unsigned char iv_str[100];
281 unsigned char src_str[100];
282 unsigned char dst_str[100];
283 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200284 mbedtls_aes_context ctx;
Paul Bakkercd43a0b2011-06-09 13:55:44 +0000285 size_t iv_offset = 0;
Paul Bakker69998dd2009-07-11 19:15:20 +0000286 int key_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000287
288 memset(key_str, 0x00, 100);
289 memset(iv_str, 0x00, 100);
290 memset(src_str, 0x00, 100);
291 memset(dst_str, 0x00, 100);
292 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200293 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000294
Paul Bakker33b43f12013-08-20 11:48:36 +0200295 key_len = unhexify( key_str, hex_key_string );
296 unhexify( iv_str, hex_iv_string );
297 unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000298
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200299 mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 );
300 TEST_ASSERT( mbedtls_aes_crypt_cfb128( &ctx, MBEDTLS_AES_DECRYPT, 16, &iv_offset, iv_str, src_str, output ) == 0 );
Paul Bakker367dae42009-06-28 21:50:27 +0000301 hexify( dst_str, output, 16 );
302
Paul Bakker33b43f12013-08-20 11:48:36 +0200303 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200304
Paul Bakkerbd51b262014-07-10 15:26:12 +0200305exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200306 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000307}
Paul Bakker33b43f12013-08-20 11:48:36 +0200308/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000309
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200310/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */
Paul Bakker556efba2014-01-24 15:38:12 +0100311void aes_encrypt_cfb8( char *hex_key_string, char *hex_iv_string,
312 char *hex_src_string, char *hex_dst_string )
313{
314 unsigned char key_str[100];
315 unsigned char iv_str[100];
316 unsigned char src_str[100];
317 unsigned char dst_str[100];
318 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200319 mbedtls_aes_context ctx;
Paul Bakker556efba2014-01-24 15:38:12 +0100320 int key_len, src_len;
321
322 memset(key_str, 0x00, 100);
323 memset(iv_str, 0x00, 100);
324 memset(src_str, 0x00, 100);
325 memset(dst_str, 0x00, 100);
326 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200327 mbedtls_aes_init( &ctx );
Paul Bakker556efba2014-01-24 15:38:12 +0100328
329 key_len = unhexify( key_str, hex_key_string );
330 unhexify( iv_str, hex_iv_string );
331 src_len = unhexify( src_str, hex_src_string );
332
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200333 mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 );
334 TEST_ASSERT( mbedtls_aes_crypt_cfb8( &ctx, MBEDTLS_AES_ENCRYPT, src_len, iv_str, src_str, output ) == 0 );
Paul Bakker556efba2014-01-24 15:38:12 +0100335 hexify( dst_str, output, src_len );
336
337 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200338
Paul Bakkerbd51b262014-07-10 15:26:12 +0200339exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200340 mbedtls_aes_free( &ctx );
Paul Bakker556efba2014-01-24 15:38:12 +0100341}
342/* END_CASE */
343
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200344/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */
Paul Bakker556efba2014-01-24 15:38:12 +0100345void aes_decrypt_cfb8( char *hex_key_string, char *hex_iv_string,
346 char *hex_src_string, char *hex_dst_string )
347{
348 unsigned char key_str[100];
349 unsigned char iv_str[100];
350 unsigned char src_str[100];
351 unsigned char dst_str[100];
352 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200353 mbedtls_aes_context ctx;
Paul Bakker556efba2014-01-24 15:38:12 +0100354 int key_len, src_len;
355
356 memset(key_str, 0x00, 100);
357 memset(iv_str, 0x00, 100);
358 memset(src_str, 0x00, 100);
359 memset(dst_str, 0x00, 100);
360 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200361 mbedtls_aes_init( &ctx );
Paul Bakker556efba2014-01-24 15:38:12 +0100362
363 key_len = unhexify( key_str, hex_key_string );
364 unhexify( iv_str, hex_iv_string );
365 src_len = unhexify( src_str, hex_src_string );
366
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200367 mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 );
368 TEST_ASSERT( mbedtls_aes_crypt_cfb8( &ctx, MBEDTLS_AES_DECRYPT, src_len, iv_str, src_str, output ) == 0 );
Paul Bakker556efba2014-01-24 15:38:12 +0100369 hexify( dst_str, output, src_len );
370
371 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200372
Paul Bakkerbd51b262014-07-10 15:26:12 +0200373exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200374 mbedtls_aes_free( &ctx );
Paul Bakker556efba2014-01-24 15:38:12 +0100375}
376/* END_CASE */
377
Simon Butcher03018842018-04-22 22:57:58 +0100378/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_OFB */
379void aes_encrypt_ofb( int fragment_size, char *hex_key_string,
Simon Butcher00131442018-05-22 22:40:36 +0100380 char *hex_iv_string, char *hex_src_string,
381 char *hex_dst_string )
Simon Butcher03018842018-04-22 22:57:58 +0100382{
Simon Butchere416bf92018-06-02 18:28:32 +0100383 unsigned char key_str[32];
384 unsigned char iv_str[16];
385 unsigned char src_str[64];
386 unsigned char dst_str[64];
387 unsigned char output[32];
Simon Butcher03018842018-04-22 22:57:58 +0100388 mbedtls_aes_context ctx;
389 size_t iv_offset = 0;
390 int in_buffer_len;
391 unsigned char* src_str_next;
Simon Butcherdbe7fbf2018-04-29 14:51:35 +0100392 int key_len;
Simon Butcher03018842018-04-22 22:57:58 +0100393
Simon Butcherb7836e12018-06-02 18:36:49 +0100394 memset( key_str, 0x00, sizeof( key_str ) );
395 memset( iv_str, 0x00, sizeof( iv_str ) );
396 memset( src_str, 0x00, sizeof( src_str ) );
397 memset( dst_str, 0x00, sizeof( dst_str ) );
398 memset( output, 0x00, sizeof( output ) );
Simon Butcher03018842018-04-22 22:57:58 +0100399 mbedtls_aes_init( &ctx );
400
Simon Butchere416bf92018-06-02 18:28:32 +0100401 TEST_ASSERT( strlen( hex_key_string ) <= ( 32 * 2 ) );
402 TEST_ASSERT( strlen( hex_iv_string ) <= ( 16 * 2 ) );
403 TEST_ASSERT( strlen( hex_src_string ) <= ( 64 * 2 ) );
404 TEST_ASSERT( strlen( hex_dst_string ) <= ( 64 * 2 ) );
405
Simon Butcher03018842018-04-22 22:57:58 +0100406 key_len = unhexify( key_str, hex_key_string );
Simon Butcherdbe7fbf2018-04-29 14:51:35 +0100407 unhexify( iv_str, hex_iv_string );
Simon Butcher03018842018-04-22 22:57:58 +0100408 in_buffer_len = unhexify( src_str, hex_src_string );
409
Simon Butcherad4e4932018-04-29 00:43:47 +0100410 TEST_ASSERT( mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 ) == 0 );
Simon Butcher03018842018-04-22 22:57:58 +0100411 src_str_next = src_str;
412
413 while( in_buffer_len > 0 )
414 {
415 TEST_ASSERT( mbedtls_aes_crypt_ofb( &ctx, fragment_size, &iv_offset,
416 iv_str, src_str_next, output ) == 0 );
417
418 hexify( dst_str, output, fragment_size );
419 TEST_ASSERT( strncmp( (char *) dst_str, hex_dst_string,
Simon Butcher00131442018-05-22 22:40:36 +0100420 ( 2 * fragment_size ) ) == 0 );
Simon Butcher03018842018-04-22 22:57:58 +0100421
422 in_buffer_len -= fragment_size;
423 hex_dst_string += ( fragment_size * 2 );
424 src_str_next += fragment_size;
425
426 if( in_buffer_len < fragment_size )
427 fragment_size = in_buffer_len;
428 }
429
430exit:
431 mbedtls_aes_free( &ctx );
432}
433/* END_CASE */
434
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200435/* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */
Paul Bakker33b43f12013-08-20 11:48:36 +0200436void aes_selftest()
Paul Bakker3d360822009-07-05 11:29:38 +0000437{
Andres AG93012e82016-09-09 09:10:28 +0100438 TEST_ASSERT( mbedtls_aes_self_test( 1 ) == 0 );
Paul Bakker3d360822009-07-05 11:29:38 +0000439}
Paul Bakker33b43f12013-08-20 11:48:36 +0200440/* END_CASE */