blob: dac6edc958e7c73a6c889e3f2cbeb6682080206c [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 */
Azim Khan5fcca462018-06-29 11:05:32 +010011void aes_encrypt_ecb( data_t * key_str, data_t * src_str,
Ronald Cronac6ae352020-06-26 14:33:03 +020012 data_t * dst, int setkey_result )
Paul Bakker367dae42009-06-28 21:50:27 +000013{
Paul Bakker367dae42009-06-28 21:50:27 +000014 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020015 mbedtls_aes_context ctx;
Paul Bakker367dae42009-06-28 21:50:27 +000016
Paul Bakker367dae42009-06-28 21:50:27 +000017 memset(output, 0x00, 100);
18
Andrzej Kurekc470b6b2019-01-31 08:20:20 -050019 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +000020
Azim Khand30ca132017-06-09 04:32:58 +010021 TEST_ASSERT( mbedtls_aes_setkey_enc( &ctx, key_str->x, key_str->len * 8 ) == setkey_result );
Paul Bakker33b43f12013-08-20 11:48:36 +020022 if( setkey_result == 0 )
Paul Bakker2b222c82009-07-27 21:03:45 +000023 {
Azim Khand30ca132017-06-09 04:32:58 +010024 TEST_ASSERT( mbedtls_aes_crypt_ecb( &ctx, MBEDTLS_AES_ENCRYPT, src_str->x, output ) == 0 );
Paul Bakker367dae42009-06-28 21:50:27 +000025
Ronald Cronac6ae352020-06-26 14:33:03 +020026 TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x, 16, dst->len ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +000027 }
Paul Bakker8cfd9d82014-06-18 11:16:11 +020028
Paul Bakkerbd51b262014-07-10 15:26:12 +020029exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020030 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +000031}
Paul Bakker33b43f12013-08-20 11:48:36 +020032/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +000033
Paul Bakker33b43f12013-08-20 11:48:36 +020034/* BEGIN_CASE */
Azim Khan5fcca462018-06-29 11:05:32 +010035void aes_decrypt_ecb( data_t * key_str, data_t * src_str,
Ronald Cronac6ae352020-06-26 14:33:03 +020036 data_t * dst, int setkey_result )
Paul Bakker367dae42009-06-28 21:50:27 +000037{
Paul Bakker367dae42009-06-28 21:50:27 +000038 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020039 mbedtls_aes_context ctx;
Paul Bakker367dae42009-06-28 21:50:27 +000040
Paul Bakker367dae42009-06-28 21:50:27 +000041 memset(output, 0x00, 100);
42
Andrzej Kurekc470b6b2019-01-31 08:20:20 -050043 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +000044
Azim Khand30ca132017-06-09 04:32:58 +010045 TEST_ASSERT( mbedtls_aes_setkey_dec( &ctx, key_str->x, key_str->len * 8 ) == setkey_result );
Paul Bakker33b43f12013-08-20 11:48:36 +020046 if( setkey_result == 0 )
Paul Bakker2b222c82009-07-27 21:03:45 +000047 {
Azim Khand30ca132017-06-09 04:32:58 +010048 TEST_ASSERT( mbedtls_aes_crypt_ecb( &ctx, MBEDTLS_AES_DECRYPT, src_str->x, output ) == 0 );
Paul Bakker367dae42009-06-28 21:50:27 +000049
Ronald Cronac6ae352020-06-26 14:33:03 +020050 TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x, 16, dst->len ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +000051 }
Paul Bakker8cfd9d82014-06-18 11:16:11 +020052
Paul Bakkerbd51b262014-07-10 15:26:12 +020053exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020054 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +000055}
Paul Bakker33b43f12013-08-20 11:48:36 +020056/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +000057
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020058/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CBC */
Azim Khan5fcca462018-06-29 11:05:32 +010059void aes_encrypt_cbc( data_t * key_str, data_t * iv_str,
Ronald Cronac6ae352020-06-26 14:33:03 +020060 data_t * src_str, data_t * dst,
Azim Khand30ca132017-06-09 04:32:58 +010061 int cbc_result )
Paul Bakker367dae42009-06-28 21:50:27 +000062{
Paul Bakker367dae42009-06-28 21:50:27 +000063 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020064 mbedtls_aes_context ctx;
Paul Bakker367dae42009-06-28 21:50:27 +000065
Paul Bakker367dae42009-06-28 21:50:27 +000066 memset(output, 0x00, 100);
67
Andrzej Kurekc470b6b2019-01-31 08:20:20 -050068 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +000069
Azim Khand30ca132017-06-09 04:32:58 +010070 mbedtls_aes_setkey_enc( &ctx, key_str->x, key_str->len * 8 );
71 TEST_ASSERT( mbedtls_aes_crypt_cbc( &ctx, MBEDTLS_AES_ENCRYPT, src_str->len, iv_str->x, src_str->x, output ) == cbc_result );
Paul Bakker33b43f12013-08-20 11:48:36 +020072 if( cbc_result == 0 )
Paul Bakkerf3ccc682010-03-18 21:21:02 +000073 {
Paul Bakker367dae42009-06-28 21:50:27 +000074
Ronald Cronac6ae352020-06-26 14:33:03 +020075 TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x,
76 src_str->len, dst->len ) == 0 );
Paul Bakkerf3ccc682010-03-18 21:21:02 +000077 }
Paul Bakker8cfd9d82014-06-18 11:16:11 +020078
Paul Bakkerbd51b262014-07-10 15:26:12 +020079exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020080 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +000081}
Paul Bakker33b43f12013-08-20 11:48:36 +020082/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +000083
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020084/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CBC */
Azim Khan5fcca462018-06-29 11:05:32 +010085void aes_decrypt_cbc( data_t * key_str, data_t * iv_str,
Ronald Cronac6ae352020-06-26 14:33:03 +020086 data_t * src_str, data_t * dst,
Azim Khand30ca132017-06-09 04:32:58 +010087 int cbc_result )
Paul Bakker367dae42009-06-28 21:50:27 +000088{
Paul Bakker367dae42009-06-28 21:50:27 +000089 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020090 mbedtls_aes_context ctx;
Paul Bakker367dae42009-06-28 21:50:27 +000091
Paul Bakker367dae42009-06-28 21:50:27 +000092 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020093 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +000094
Azim Khand30ca132017-06-09 04:32:58 +010095 mbedtls_aes_setkey_dec( &ctx, key_str->x, key_str->len * 8 );
96 TEST_ASSERT( mbedtls_aes_crypt_cbc( &ctx, MBEDTLS_AES_DECRYPT, src_str->len, iv_str->x, src_str->x, output ) == cbc_result );
Paul Bakker33b43f12013-08-20 11:48:36 +020097 if( cbc_result == 0)
Paul Bakkerf3ccc682010-03-18 21:21:02 +000098 {
Paul Bakker367dae42009-06-28 21:50:27 +000099
Ronald Cronac6ae352020-06-26 14:33:03 +0200100 TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x,
101 src_str->len, dst->len ) == 0 );
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000102 }
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200103
Paul Bakkerbd51b262014-07-10 15:26:12 +0200104exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200105 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000106}
Paul Bakker33b43f12013-08-20 11:48:36 +0200107/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000108
Aorimn5f778012016-06-09 23:22:58 +0200109/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_XTS */
Jaeden Amerocd9fc5e2018-05-30 15:23:24 +0100110void aes_encrypt_xts( char *hex_key_string, char *hex_data_unit_string,
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100111 char *hex_src_string, char *hex_dst_string )
Aorimn5f778012016-06-09 23:22:58 +0200112{
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100113 enum { AES_BLOCK_SIZE = 16 };
114 unsigned char *data_unit = NULL;
115 unsigned char *key = NULL;
116 unsigned char *src = NULL;
117 unsigned char *dst = NULL;
118 unsigned char *output = NULL;
Jaeden Amero9366feb2018-05-29 18:55:17 +0100119 mbedtls_aes_xts_context ctx;
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100120 size_t key_len, src_len, dst_len, data_unit_len;
Aorimn5f778012016-06-09 23:22:58 +0200121
Jaeden Amero9366feb2018-05-29 18:55:17 +0100122 mbedtls_aes_xts_init( &ctx );
Aorimn5f778012016-06-09 23:22:58 +0200123
Ronald Cronf73ab002020-06-10 10:57:28 +0200124 data_unit = mbedtls_test_unhexify_alloc( hex_data_unit_string,
125 &data_unit_len );
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100126 TEST_ASSERT( data_unit_len == AES_BLOCK_SIZE );
Aorimn5f778012016-06-09 23:22:58 +0200127
Ronald Crona256c702020-06-10 10:53:11 +0200128 key = mbedtls_test_unhexify_alloc( hex_key_string, &key_len );
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100129 TEST_ASSERT( key_len % 2 == 0 );
Aorimn5f778012016-06-09 23:22:58 +0200130
Ronald Crona256c702020-06-10 10:53:11 +0200131 src = mbedtls_test_unhexify_alloc( hex_src_string, &src_len );
132 dst = mbedtls_test_unhexify_alloc( hex_dst_string, &dst_len );
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100133 TEST_ASSERT( src_len == dst_len );
Aorimn5f778012016-06-09 23:22:58 +0200134
Ronald Cron690f3eb2020-06-10 10:42:18 +0200135 output = mbedtls_test_zero_alloc( dst_len );
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100136
137 TEST_ASSERT( mbedtls_aes_xts_setkey_enc( &ctx, key, key_len * 8 ) == 0 );
138 TEST_ASSERT( mbedtls_aes_crypt_xts( &ctx, MBEDTLS_AES_ENCRYPT, src_len,
139 data_unit, src, output ) == 0 );
140
141 TEST_ASSERT( memcmp( output, dst, dst_len ) == 0 );
Aorimn5f778012016-06-09 23:22:58 +0200142
143exit:
Jaeden Amero9366feb2018-05-29 18:55:17 +0100144 mbedtls_aes_xts_free( &ctx );
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100145 mbedtls_free( data_unit );
146 mbedtls_free( key );
147 mbedtls_free( src );
148 mbedtls_free( dst );
149 mbedtls_free( output );
Aorimn5f778012016-06-09 23:22:58 +0200150}
151/* END_CASE */
152
153/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_XTS */
Jaeden Amerocd9fc5e2018-05-30 15:23:24 +0100154void aes_decrypt_xts( char *hex_key_string, char *hex_data_unit_string,
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100155 char *hex_dst_string, char *hex_src_string )
Aorimn5f778012016-06-09 23:22:58 +0200156{
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100157 enum { AES_BLOCK_SIZE = 16 };
158 unsigned char *data_unit = NULL;
159 unsigned char *key = NULL;
160 unsigned char *src = NULL;
161 unsigned char *dst = NULL;
162 unsigned char *output = NULL;
Jaeden Amero9366feb2018-05-29 18:55:17 +0100163 mbedtls_aes_xts_context ctx;
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100164 size_t key_len, src_len, dst_len, data_unit_len;
Aorimn5f778012016-06-09 23:22:58 +0200165
Jaeden Amero9366feb2018-05-29 18:55:17 +0100166 mbedtls_aes_xts_init( &ctx );
Aorimn5f778012016-06-09 23:22:58 +0200167
Ronald Cronf73ab002020-06-10 10:57:28 +0200168 data_unit = mbedtls_test_unhexify_alloc( hex_data_unit_string,
169 &data_unit_len );
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100170 TEST_ASSERT( data_unit_len == AES_BLOCK_SIZE );
Aorimn5f778012016-06-09 23:22:58 +0200171
Ronald Crona256c702020-06-10 10:53:11 +0200172 key = mbedtls_test_unhexify_alloc( hex_key_string, &key_len );
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100173 TEST_ASSERT( key_len % 2 == 0 );
Aorimn5f778012016-06-09 23:22:58 +0200174
Ronald Crona256c702020-06-10 10:53:11 +0200175 src = mbedtls_test_unhexify_alloc( hex_src_string, &src_len );
176 dst = mbedtls_test_unhexify_alloc( hex_dst_string, &dst_len );
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100177 TEST_ASSERT( src_len == dst_len );
Aorimn5f778012016-06-09 23:22:58 +0200178
Ronald Cron690f3eb2020-06-10 10:42:18 +0200179 output = mbedtls_test_zero_alloc( dst_len );
Jaeden Ameroe5c4b072018-04-28 17:26:25 +0100180
181 TEST_ASSERT( mbedtls_aes_xts_setkey_dec( &ctx, key, key_len * 8 ) == 0 );
182 TEST_ASSERT( mbedtls_aes_crypt_xts( &ctx, MBEDTLS_AES_DECRYPT, 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
Jaeden Amero425382d2018-04-28 17:26:25 +0100197/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_XTS */
198void aes_crypt_xts_size( int size, int retval )
199{
200 mbedtls_aes_xts_context ctx;
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500201 const unsigned char src[16] = { 0 };
202 unsigned char output[16];
Jaeden Amero425382d2018-04-28 17:26:25 +0100203 unsigned char data_unit[16];
204 size_t length = size;
205
206 mbedtls_aes_xts_init( &ctx );
207 memset( data_unit, 0x00, sizeof( data_unit ) );
208
Jaeden Amero425382d2018-04-28 17:26:25 +0100209 TEST_ASSERT( mbedtls_aes_crypt_xts( &ctx, MBEDTLS_AES_ENCRYPT, length, data_unit, src, output ) == retval );
JoeSubbiani57face22021-06-17 15:00:01 +0100210exit:
211 mbedtls_aes_xts_free(&ctx);
Jaeden Amero425382d2018-04-28 17:26:25 +0100212}
213/* END_CASE */
214
Jaeden Amero142383e2018-05-31 10:40:34 +0100215/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_XTS */
216void aes_crypt_xts_keysize( int size, int retval )
217{
218 mbedtls_aes_xts_context ctx;
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500219 const unsigned char key[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06 };
Jaeden Amero142383e2018-05-31 10:40:34 +0100220 size_t key_len = size;
221
222 mbedtls_aes_xts_init( &ctx );
223
224 TEST_ASSERT( mbedtls_aes_xts_setkey_enc( &ctx, key, key_len * 8 ) == retval );
225 TEST_ASSERT( mbedtls_aes_xts_setkey_dec( &ctx, key, key_len * 8 ) == retval );
226exit:
227 mbedtls_aes_xts_free( &ctx );
228}
229/* END_CASE */
Jaeden Amero425382d2018-04-28 17:26:25 +0100230
231
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200232/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */
Azim Khan5fcca462018-06-29 11:05:32 +0100233void aes_encrypt_cfb128( data_t * key_str, data_t * iv_str,
Ronald Cronac6ae352020-06-26 14:33:03 +0200234 data_t * src_str, data_t * dst )
Paul Bakker367dae42009-06-28 21:50:27 +0000235{
Paul Bakker367dae42009-06-28 21:50:27 +0000236 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200237 mbedtls_aes_context ctx;
Paul Bakkercd43a0b2011-06-09 13:55:44 +0000238 size_t iv_offset = 0;
Paul Bakker367dae42009-06-28 21:50:27 +0000239
Paul Bakker367dae42009-06-28 21:50:27 +0000240 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200241 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000242
Paul Bakker367dae42009-06-28 21:50:27 +0000243
Azim Khand30ca132017-06-09 04:32:58 +0100244 mbedtls_aes_setkey_enc( &ctx, key_str->x, key_str->len * 8 );
245 TEST_ASSERT( mbedtls_aes_crypt_cfb128( &ctx, MBEDTLS_AES_ENCRYPT, 16, &iv_offset, iv_str->x, src_str->x, output ) == 0 );
Paul Bakker367dae42009-06-28 21:50:27 +0000246
Ronald Cronac6ae352020-06-26 14:33:03 +0200247 TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x, 16, dst->len ) == 0 );
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200248
Paul Bakkerbd51b262014-07-10 15:26:12 +0200249exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200250 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000251}
Paul Bakker33b43f12013-08-20 11:48:36 +0200252/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000253
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200254/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */
Azim Khan5fcca462018-06-29 11:05:32 +0100255void aes_decrypt_cfb128( data_t * key_str, data_t * iv_str,
Ronald Cronac6ae352020-06-26 14:33:03 +0200256 data_t * src_str, data_t * dst )
Paul Bakker367dae42009-06-28 21:50:27 +0000257{
Paul Bakker367dae42009-06-28 21:50:27 +0000258 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200259 mbedtls_aes_context ctx;
Paul Bakkercd43a0b2011-06-09 13:55:44 +0000260 size_t iv_offset = 0;
Paul Bakker367dae42009-06-28 21:50:27 +0000261
Paul Bakker367dae42009-06-28 21:50:27 +0000262 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200263 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000264
Paul Bakker367dae42009-06-28 21:50:27 +0000265
Azim Khand30ca132017-06-09 04:32:58 +0100266 mbedtls_aes_setkey_enc( &ctx, key_str->x, key_str->len * 8 );
267 TEST_ASSERT( mbedtls_aes_crypt_cfb128( &ctx, MBEDTLS_AES_DECRYPT, 16, &iv_offset, iv_str->x, src_str->x, output ) == 0 );
Paul Bakker367dae42009-06-28 21:50:27 +0000268
Ronald Cronac6ae352020-06-26 14:33:03 +0200269 TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x, 16, dst->len ) == 0 );
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200270
Paul Bakkerbd51b262014-07-10 15:26:12 +0200271exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200272 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000273}
Paul Bakker33b43f12013-08-20 11:48:36 +0200274/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000275
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200276/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */
Azim Khan5fcca462018-06-29 11:05:32 +0100277void aes_encrypt_cfb8( data_t * key_str, data_t * iv_str,
Ronald Cronac6ae352020-06-26 14:33:03 +0200278 data_t * src_str, data_t * dst )
Paul Bakker556efba2014-01-24 15:38:12 +0100279{
Paul Bakker556efba2014-01-24 15:38:12 +0100280 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200281 mbedtls_aes_context ctx;
Paul Bakker556efba2014-01-24 15:38:12 +0100282
Paul Bakker556efba2014-01-24 15:38:12 +0100283 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200284 mbedtls_aes_init( &ctx );
Paul Bakker556efba2014-01-24 15:38:12 +0100285
Paul Bakker556efba2014-01-24 15:38:12 +0100286
Azim Khand30ca132017-06-09 04:32:58 +0100287 mbedtls_aes_setkey_enc( &ctx, key_str->x, key_str->len * 8 );
288 TEST_ASSERT( mbedtls_aes_crypt_cfb8( &ctx, MBEDTLS_AES_ENCRYPT, src_str->len, iv_str->x, src_str->x, output ) == 0 );
Paul Bakker556efba2014-01-24 15:38:12 +0100289
Ronald Cronac6ae352020-06-26 14:33:03 +0200290 TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x,
291 src_str->len, dst->len ) == 0 );
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200292
Paul Bakkerbd51b262014-07-10 15:26:12 +0200293exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200294 mbedtls_aes_free( &ctx );
Paul Bakker556efba2014-01-24 15:38:12 +0100295}
296/* END_CASE */
297
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200298/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */
Azim Khan5fcca462018-06-29 11:05:32 +0100299void aes_decrypt_cfb8( data_t * key_str, data_t * iv_str,
Ronald Cronac6ae352020-06-26 14:33:03 +0200300 data_t * src_str, data_t * dst )
Paul Bakker556efba2014-01-24 15:38:12 +0100301{
Paul Bakker556efba2014-01-24 15:38:12 +0100302 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200303 mbedtls_aes_context ctx;
Paul Bakker556efba2014-01-24 15:38:12 +0100304
Paul Bakker556efba2014-01-24 15:38:12 +0100305 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200306 mbedtls_aes_init( &ctx );
Paul Bakker556efba2014-01-24 15:38:12 +0100307
Paul Bakker556efba2014-01-24 15:38:12 +0100308
Azim Khand30ca132017-06-09 04:32:58 +0100309 mbedtls_aes_setkey_enc( &ctx, key_str->x, key_str->len * 8 );
310 TEST_ASSERT( mbedtls_aes_crypt_cfb8( &ctx, MBEDTLS_AES_DECRYPT, src_str->len, iv_str->x, src_str->x, output ) == 0 );
Paul Bakker556efba2014-01-24 15:38:12 +0100311
Ronald Cronac6ae352020-06-26 14:33:03 +0200312 TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x,
313 src_str->len, dst->len ) == 0 );
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200314
Paul Bakkerbd51b262014-07-10 15:26:12 +0200315exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200316 mbedtls_aes_free( &ctx );
Paul Bakker556efba2014-01-24 15:38:12 +0100317}
318/* END_CASE */
319
Simon Butcher03018842018-04-22 22:57:58 +0100320/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_OFB */
Ronald Cron9ed40732020-06-25 09:03:34 +0200321void aes_encrypt_ofb( int fragment_size, data_t *key_str,
322 data_t *iv_str, data_t *src_str,
Ronald Cron55d97f22020-06-26 17:00:30 +0200323 data_t *expected_output )
Simon Butcher03018842018-04-22 22:57:58 +0100324{
Simon Butchere416bf92018-06-02 18:28:32 +0100325 unsigned char output[32];
Simon Butcher03018842018-04-22 22:57:58 +0100326 mbedtls_aes_context ctx;
327 size_t iv_offset = 0;
328 int in_buffer_len;
329 unsigned char* src_str_next;
Simon Butcher03018842018-04-22 22:57:58 +0100330
Simon Butcherb7836e12018-06-02 18:36:49 +0100331 memset( output, 0x00, sizeof( output ) );
Simon Butcher03018842018-04-22 22:57:58 +0100332 mbedtls_aes_init( &ctx );
333
Ronald Crondf02eb02020-06-25 13:57:05 +0200334 TEST_ASSERT( (size_t)fragment_size < sizeof( output ) );
Simon Butchere416bf92018-06-02 18:28:32 +0100335
Ronald Cron9ed40732020-06-25 09:03:34 +0200336 TEST_ASSERT( mbedtls_aes_setkey_enc( &ctx, key_str->x,
337 key_str->len * 8 ) == 0 );
338 in_buffer_len = src_str->len;
339 src_str_next = src_str->x;
Simon Butcher03018842018-04-22 22:57:58 +0100340
341 while( in_buffer_len > 0 )
342 {
343 TEST_ASSERT( mbedtls_aes_crypt_ofb( &ctx, fragment_size, &iv_offset,
Ronald Cron9ed40732020-06-25 09:03:34 +0200344 iv_str->x, src_str_next, output ) == 0 );
Simon Butcher03018842018-04-22 22:57:58 +0100345
Ronald Cron55d97f22020-06-26 17:00:30 +0200346 TEST_ASSERT( memcmp( output, expected_output->x, fragment_size ) == 0 );
Simon Butcher03018842018-04-22 22:57:58 +0100347
348 in_buffer_len -= fragment_size;
Ronald Cron55d97f22020-06-26 17:00:30 +0200349 expected_output->x += fragment_size;
Simon Butcher03018842018-04-22 22:57:58 +0100350 src_str_next += fragment_size;
351
352 if( in_buffer_len < fragment_size )
353 fragment_size = in_buffer_len;
354 }
355
356exit:
357 mbedtls_aes_free( &ctx );
358}
359/* END_CASE */
360
TRodziewicz062f3532021-05-25 15:15:57 +0200361/* BEGIN_CASE depends_on:NOT_DEFINED */
Ronald Cron875b5fb2021-05-21 08:50:00 +0200362void aes_invalid_mode( )
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500363{
364 mbedtls_aes_context aes_ctx;
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500365 const unsigned char in[16] = { 0 };
366 unsigned char out[16];
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500367 const int invalid_mode = 42;
368
Ronald Cron875b5fb2021-05-21 08:50:00 +0200369 TEST_EQUAL( MBEDTLS_ERR_AES_BAD_INPUT_DATA,
370 mbedtls_aes_crypt_ecb( &aes_ctx, invalid_mode, in, out ) );
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500371
372#if defined(MBEDTLS_CIPHER_MODE_CBC)
Ronald Cron875b5fb2021-05-21 08:50:00 +0200373 TEST_EQUAL( MBEDTLS_ERR_AES_BAD_INPUT_DATA,
374 mbedtls_aes_crypt_cbc( &aes_ctx, invalid_mode, 16,
375 out, in, out ) );
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500376#endif /* MBEDTLS_CIPHER_MODE_CBC */
377
378#if defined(MBEDTLS_CIPHER_MODE_XTS)
Ronald Cron875b5fb2021-05-21 08:50:00 +0200379 mbedtls_aes_xts_context xts_ctx;
380
381 TEST_EQUAL( MBEDTLS_ERR_AES_BAD_INPUT_DATA,
382 mbedtls_aes_crypt_xts( &xts_ctx, invalid_mode, 16,
383 in, in, out ) );
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500384#endif /* MBEDTLS_CIPHER_MODE_XTS */
385
386#if defined(MBEDTLS_CIPHER_MODE_CFB)
Ronald Cron875b5fb2021-05-21 08:50:00 +0200387 size_t size;
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500388
Ronald Cron875b5fb2021-05-21 08:50:00 +0200389 TEST_EQUAL( MBEDTLS_ERR_AES_BAD_INPUT_DATA,
390 mbedtls_aes_crypt_cfb128( &aes_ctx, invalid_mode, 16,
391 &size, out, in, out ) );
392 TEST_EQUAL( MBEDTLS_ERR_AES_BAD_INPUT_DATA,
393 mbedtls_aes_crypt_cfb8( &aes_ctx, invalid_mode, 16,
394 out, in, out ) );
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500395#endif /* MBEDTLS_CIPHER_MODE_CFB */
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500396}
397/* END_CASE */
398
399/* BEGIN_CASE */
400void aes_misc_params( )
401{
402#if defined(MBEDTLS_CIPHER_MODE_CBC) || \
403 defined(MBEDTLS_CIPHER_MODE_XTS) || \
404 defined(MBEDTLS_CIPHER_MODE_CFB) || \
405 defined(MBEDTLS_CIPHER_MODE_OFB)
406 mbedtls_aes_context aes_ctx;
407 const unsigned char in[16] = { 0 };
408 unsigned char out[16];
409#endif
410#if defined(MBEDTLS_CIPHER_MODE_XTS)
411 mbedtls_aes_xts_context xts_ctx;
412#endif
413#if defined(MBEDTLS_CIPHER_MODE_CFB) || \
414 defined(MBEDTLS_CIPHER_MODE_OFB)
415 size_t size;
416#endif
417
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500418#if defined(MBEDTLS_CIPHER_MODE_CBC)
419 TEST_ASSERT( mbedtls_aes_crypt_cbc( &aes_ctx, MBEDTLS_AES_ENCRYPT,
420 15,
421 out, in, out )
422 == MBEDTLS_ERR_AES_INVALID_INPUT_LENGTH );
423 TEST_ASSERT( mbedtls_aes_crypt_cbc( &aes_ctx, MBEDTLS_AES_ENCRYPT,
424 17,
425 out, in, out )
426 == MBEDTLS_ERR_AES_INVALID_INPUT_LENGTH );
427#endif
428
429#if defined(MBEDTLS_CIPHER_MODE_XTS)
430 TEST_ASSERT( mbedtls_aes_crypt_xts( &xts_ctx, MBEDTLS_AES_ENCRYPT,
431 15,
432 in, in, out )
433 == MBEDTLS_ERR_AES_INVALID_INPUT_LENGTH );
434 TEST_ASSERT( mbedtls_aes_crypt_xts( &xts_ctx, MBEDTLS_AES_ENCRYPT,
435 (1 << 24) + 1,
436 in, in, out )
437 == MBEDTLS_ERR_AES_INVALID_INPUT_LENGTH );
438#endif
439
440#if defined(MBEDTLS_CIPHER_MODE_CFB)
441 size = 16;
442 TEST_ASSERT( mbedtls_aes_crypt_cfb128( &aes_ctx, MBEDTLS_AES_ENCRYPT, 16,
443 &size, out, in, out )
444 == MBEDTLS_ERR_AES_BAD_INPUT_DATA );
445#endif
446
447#if defined(MBEDTLS_CIPHER_MODE_OFB)
448 size = 16;
449 TEST_ASSERT( mbedtls_aes_crypt_ofb( &aes_ctx, 16, &size, out, in, out )
450 == MBEDTLS_ERR_AES_BAD_INPUT_DATA );
451#endif
TRodziewicz358e5882021-05-28 12:58:46 +0200452
453/*
454 * The following line needs to be added to make the code compilable
455 * when all the conditions above will be not define in a specific
456 * choice of features.
457 */
458 TEST_ASSERT( 1 );
459/* TODO: It will be removed when the whole test will be reworked */
Andrzej Kurekc470b6b2019-01-31 08:20:20 -0500460}
461/* END_CASE */
462
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200463/* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */
Azim Khanf1aaec92017-05-30 14:23:15 +0100464void aes_selftest( )
Paul Bakker3d360822009-07-05 11:29:38 +0000465{
Andres AG93012e82016-09-09 09:10:28 +0100466 TEST_ASSERT( mbedtls_aes_self_test( 1 ) == 0 );
Paul Bakker3d360822009-07-05 11:29:38 +0000467}
Paul Bakker33b43f12013-08-20 11:48:36 +0200468/* END_CASE */