TrustedFirmware Git Browser
Code Review Sign In
review.trustedfirmware.org / mirror / mbed-tls / e416bf93d261cddbb246b80ff4b5357bab34c026 / . / tests / suites / test_suite_aes.function
blob: 24e8f7ab61623bb24d3c7af0130f721ec2a732e8 [file] [log] [blame]
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]1/* BEGIN_HEADER */
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +0000[diff] [blame]2#include "mbedtls/aes.h"
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]3/* END_HEADER */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]4
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]5/* BEGIN_DEPENDENCIES
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]6 * depends_on:MBEDTLS_AES_C
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]7 * END_DEPENDENCIES
8 */
Paul Bakker5690efc2011-05-26 13:16:06 +0000[diff] [blame]9
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]10/* 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 +0000[diff] [blame]13{
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 +0200[diff] [blame]18 mbedtls_aes_context ctx;
Paul Bakker69998dd2009-07-11 19:15:20 +0000[diff] [blame]19 int key_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]20
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 +0200[diff] [blame]25 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]26
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]27 key_len = unhexify( key_str, hex_key_string );
28 unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]29
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]30 TEST_ASSERT( mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 ) == setkey_result );
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]31 if( setkey_result == 0 )
Paul Bakker2b222c82009-07-27 21:03:45 +0000[diff] [blame]32 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]33 TEST_ASSERT( mbedtls_aes_crypt_ecb( &ctx, MBEDTLS_AES_ENCRYPT, src_str, output ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +0000[diff] [blame]34 hexify( dst_str, output, 16 );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]35
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]36 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +0000[diff] [blame]37 }
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200[diff] [blame]38
Paul Bakkerbd51b262014-07-10 15:26:12 +0200[diff] [blame]39exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]40 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]41}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]42/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]43
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]44/* 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 +0000[diff] [blame]47{
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 +0200[diff] [blame]52 mbedtls_aes_context ctx;
Paul Bakker69998dd2009-07-11 19:15:20 +0000[diff] [blame]53 int key_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]54
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 +0200[diff] [blame]59 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]60
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]61 key_len = unhexify( key_str, hex_key_string );
62 unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]63
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]64 TEST_ASSERT( mbedtls_aes_setkey_dec( &ctx, key_str, key_len * 8 ) == setkey_result );
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]65 if( setkey_result == 0 )
Paul Bakker2b222c82009-07-27 21:03:45 +0000[diff] [blame]66 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]67 TEST_ASSERT( mbedtls_aes_crypt_ecb( &ctx, MBEDTLS_AES_DECRYPT, src_str, output ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +0000[diff] [blame]68 hexify( dst_str, output, 16 );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]69
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]70 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +0000[diff] [blame]71 }
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200[diff] [blame]72
Paul Bakkerbd51b262014-07-10 15:26:12 +0200[diff] [blame]73exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]74 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]75}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]76/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]77
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]78/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CBC */
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]79void 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 +0000[diff] [blame]82{
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 +0200[diff] [blame]88 mbedtls_aes_context ctx;
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]89 int key_len, data_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]90
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 +0200[diff] [blame]96 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]97
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]98 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 +0000[diff] [blame]101
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]102 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 +0200[diff] [blame]104 if( cbc_result == 0 )
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]105 {
106 hexify( dst_str, output, data_len );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]107
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]108 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]109 }
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200[diff] [blame]110
Paul Bakkerbd51b262014-07-10 15:26:12 +0200[diff] [blame]111exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]112 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]113}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]114/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]115
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]116/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CBC */
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]117void 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 +0000[diff] [blame]120{
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 +0200[diff] [blame]126 mbedtls_aes_context ctx;
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]127 int key_len, data_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]128
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 +0200[diff] [blame]134 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]135
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]136 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 +0000[diff] [blame]139
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]140 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 +0200[diff] [blame]142 if( cbc_result == 0)
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]143 {
144 hexify( dst_str, output, data_len );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]145
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]146 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]147 }
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200[diff] [blame]148
Paul Bakkerbd51b262014-07-10 15:26:12 +0200[diff] [blame]149exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]150 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]151}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]152/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]153
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]154/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]155void aes_encrypt_cfb128( char *hex_key_string, char *hex_iv_string,
156 char *hex_src_string, char *hex_dst_string )
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]157{
158 unsigned char key_str[100];
159 unsigned char iv_str[100];
160 unsigned char src_str[100];
161 unsigned char dst_str[100];
162 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]163 mbedtls_aes_context ctx;
Paul Bakkercd43a0b2011-06-09 13:55:44 +0000[diff] [blame]164 size_t iv_offset = 0;
Paul Bakker69998dd2009-07-11 19:15:20 +0000[diff] [blame]165 int key_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]166
167 memset(key_str, 0x00, 100);
168 memset(iv_str, 0x00, 100);
169 memset(src_str, 0x00, 100);
170 memset(dst_str, 0x00, 100);
171 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]172 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]173
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]174 key_len = unhexify( key_str, hex_key_string );
175 unhexify( iv_str, hex_iv_string );
176 unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]177
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]178 mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 );
179 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 +0000[diff] [blame]180 hexify( dst_str, output, 16 );
181
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]182 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200[diff] [blame]183
Paul Bakkerbd51b262014-07-10 15:26:12 +0200[diff] [blame]184exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]185 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]186}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]187/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]188
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]189/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]190void aes_decrypt_cfb128( char *hex_key_string, char *hex_iv_string,
191 char *hex_src_string, char *hex_dst_string )
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]192{
193 unsigned char key_str[100];
194 unsigned char iv_str[100];
195 unsigned char src_str[100];
196 unsigned char dst_str[100];
197 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]198 mbedtls_aes_context ctx;
Paul Bakkercd43a0b2011-06-09 13:55:44 +0000[diff] [blame]199 size_t iv_offset = 0;
Paul Bakker69998dd2009-07-11 19:15:20 +0000[diff] [blame]200 int key_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]201
202 memset(key_str, 0x00, 100);
203 memset(iv_str, 0x00, 100);
204 memset(src_str, 0x00, 100);
205 memset(dst_str, 0x00, 100);
206 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]207 mbedtls_aes_init( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]208
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]209 key_len = unhexify( key_str, hex_key_string );
210 unhexify( iv_str, hex_iv_string );
211 unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]212
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]213 mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 );
214 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 +0000[diff] [blame]215 hexify( dst_str, output, 16 );
216
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]217 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200[diff] [blame]218
Paul Bakkerbd51b262014-07-10 15:26:12 +0200[diff] [blame]219exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]220 mbedtls_aes_free( &ctx );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]221}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]222/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]223
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]224/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */
Paul Bakker556efba2014-01-24 15:38:12 +0100[diff] [blame]225void aes_encrypt_cfb8( char *hex_key_string, char *hex_iv_string,
226 char *hex_src_string, char *hex_dst_string )
227{
228 unsigned char key_str[100];
229 unsigned char iv_str[100];
230 unsigned char src_str[100];
231 unsigned char dst_str[100];
232 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]233 mbedtls_aes_context ctx;
Paul Bakker556efba2014-01-24 15:38:12 +0100[diff] [blame]234 int key_len, src_len;
235
236 memset(key_str, 0x00, 100);
237 memset(iv_str, 0x00, 100);
238 memset(src_str, 0x00, 100);
239 memset(dst_str, 0x00, 100);
240 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]241 mbedtls_aes_init( &ctx );
Paul Bakker556efba2014-01-24 15:38:12 +0100[diff] [blame]242
243 key_len = unhexify( key_str, hex_key_string );
244 unhexify( iv_str, hex_iv_string );
245 src_len = unhexify( src_str, hex_src_string );
246
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]247 mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 );
248 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 +0100[diff] [blame]249 hexify( dst_str, output, src_len );
250
251 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200[diff] [blame]252
Paul Bakkerbd51b262014-07-10 15:26:12 +0200[diff] [blame]253exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]254 mbedtls_aes_free( &ctx );
Paul Bakker556efba2014-01-24 15:38:12 +0100[diff] [blame]255}
256/* END_CASE */
257
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]258/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */
Paul Bakker556efba2014-01-24 15:38:12 +0100[diff] [blame]259void aes_decrypt_cfb8( char *hex_key_string, char *hex_iv_string,
260 char *hex_src_string, char *hex_dst_string )
261{
262 unsigned char key_str[100];
263 unsigned char iv_str[100];
264 unsigned char src_str[100];
265 unsigned char dst_str[100];
266 unsigned char output[100];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]267 mbedtls_aes_context ctx;
Paul Bakker556efba2014-01-24 15:38:12 +0100[diff] [blame]268 int key_len, src_len;
269
270 memset(key_str, 0x00, 100);
271 memset(iv_str, 0x00, 100);
272 memset(src_str, 0x00, 100);
273 memset(dst_str, 0x00, 100);
274 memset(output, 0x00, 100);
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]275 mbedtls_aes_init( &ctx );
Paul Bakker556efba2014-01-24 15:38:12 +0100[diff] [blame]276
277 key_len = unhexify( key_str, hex_key_string );
278 unhexify( iv_str, hex_iv_string );
279 src_len = unhexify( src_str, hex_src_string );
280
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]281 mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 );
282 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 +0100[diff] [blame]283 hexify( dst_str, output, src_len );
284
285 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker8cfd9d82014-06-18 11:16:11 +0200[diff] [blame]286
Paul Bakkerbd51b262014-07-10 15:26:12 +0200[diff] [blame]287exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]288 mbedtls_aes_free( &ctx );
Paul Bakker556efba2014-01-24 15:38:12 +0100[diff] [blame]289}
290/* END_CASE */
291
Simon Butcher03018842018-04-22 22:57:58 +0100[diff] [blame]292/* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_OFB */
293void aes_encrypt_ofb( int fragment_size, char *hex_key_string,
Simon Butcher00131442018-05-22 22:40:36 +0100[diff] [blame]294 char *hex_iv_string, char *hex_src_string,
295 char *hex_dst_string )
Simon Butcher03018842018-04-22 22:57:58 +0100[diff] [blame]296{
Simon Butchere416bf92018-06-02 18:28:32 +0100[diff] [blame^]297 unsigned char key_str[32];
298 unsigned char iv_str[16];
299 unsigned char src_str[64];
300 unsigned char dst_str[64];
301 unsigned char output[32];
Simon Butcher03018842018-04-22 22:57:58 +0100[diff] [blame]302 mbedtls_aes_context ctx;
303 size_t iv_offset = 0;
304 int in_buffer_len;
305 unsigned char* src_str_next;
Simon Butcherdbe7fbf2018-04-29 14:51:35 +0100[diff] [blame]306 int key_len;
Simon Butcher03018842018-04-22 22:57:58 +0100[diff] [blame]307
Simon Butchere416bf92018-06-02 18:28:32 +0100[diff] [blame^]308 memset(key_str, 0x00, 32);
309 memset(iv_str, 0x00, 16);
310 memset(src_str, 0x00, 64);
311 memset(dst_str, 0x00, 64);
312 memset(output, 0x00, 32);
Simon Butcher03018842018-04-22 22:57:58 +0100[diff] [blame]313 mbedtls_aes_init( &ctx );
314
Simon Butchere416bf92018-06-02 18:28:32 +0100[diff] [blame^]315 TEST_ASSERT( strlen( hex_key_string ) <= ( 32 * 2 ) );
316 TEST_ASSERT( strlen( hex_iv_string ) <= ( 16 * 2 ) );
317 TEST_ASSERT( strlen( hex_src_string ) <= ( 64 * 2 ) );
318 TEST_ASSERT( strlen( hex_dst_string ) <= ( 64 * 2 ) );
319
Simon Butcher03018842018-04-22 22:57:58 +0100[diff] [blame]320 key_len = unhexify( key_str, hex_key_string );
Simon Butcherdbe7fbf2018-04-29 14:51:35 +0100[diff] [blame]321 unhexify( iv_str, hex_iv_string );
Simon Butcher03018842018-04-22 22:57:58 +0100[diff] [blame]322 in_buffer_len = unhexify( src_str, hex_src_string );
323
Simon Butcherad4e4932018-04-29 00:43:47 +0100[diff] [blame]324 TEST_ASSERT( mbedtls_aes_setkey_enc( &ctx, key_str, key_len * 8 ) == 0 );
Simon Butcher03018842018-04-22 22:57:58 +0100[diff] [blame]325 src_str_next = src_str;
326
327 while( in_buffer_len > 0 )
328 {
329 TEST_ASSERT( mbedtls_aes_crypt_ofb( &ctx, fragment_size, &iv_offset,
330 iv_str, src_str_next, output ) == 0 );
331
332 hexify( dst_str, output, fragment_size );
333 TEST_ASSERT( strncmp( (char *) dst_str, hex_dst_string,
Simon Butcher00131442018-05-22 22:40:36 +0100[diff] [blame]334 ( 2 * fragment_size ) ) == 0 );
Simon Butcher03018842018-04-22 22:57:58 +0100[diff] [blame]335
336 in_buffer_len -= fragment_size;
337 hex_dst_string += ( fragment_size * 2 );
338 src_str_next += fragment_size;
339
340 if( in_buffer_len < fragment_size )
341 fragment_size = in_buffer_len;
342 }
343
344exit:
345 mbedtls_aes_free( &ctx );
346}
347/* END_CASE */
348
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200[diff] [blame]349/* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]350void aes_selftest()
Paul Bakker3d360822009-07-05 11:29:38 +0000[diff] [blame]351{
Andres AG93012e82016-09-09 09:10:28 +0100[diff] [blame]352 TEST_ASSERT( mbedtls_aes_self_test( 1 ) == 0 );
Paul Bakker3d360822009-07-05 11:29:38 +0000[diff] [blame]353}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]354/* END_CASE */