TrustedFirmware Git Browser
Code Review Sign In
review.trustedfirmware.org / mirror / mbed-tls / 0ace4b3154e88b1ddf5347363e07ed6f1f6d2407 / . / tests / suites / test_suite_aes.function
blob: 051264e115e9acf65718d042bc5ab1c6fec68fb9 [file] [log] [blame]
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]1/* BEGIN_HEADER */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]2#include <polarssl/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
6 * depends_on:POLARSSL_AES_C
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];
18 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);
25
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]26 key_len = unhexify( key_str, hex_key_string );
27 unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]28
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]29 TEST_ASSERT( aes_setkey_enc( &ctx, key_str, key_len * 8 ) == setkey_result );
30 if( setkey_result == 0 )
Paul Bakker2b222c82009-07-27 21:03:45 +0000[diff] [blame]31 {
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]32 TEST_ASSERT( aes_crypt_ecb( &ctx, AES_ENCRYPT, src_str, output ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +0000[diff] [blame]33 hexify( dst_str, output, 16 );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]34
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]35 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +0000[diff] [blame]36 }
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]37}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]38/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]39
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]40/* BEGIN_CASE */
41void aes_decrypt_ecb( char *hex_key_string, char *hex_src_string,
42 char *hex_dst_string, int setkey_result )
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]43{
44 unsigned char key_str[100];
45 unsigned char src_str[100];
46 unsigned char dst_str[100];
47 unsigned char output[100];
48 aes_context ctx;
Paul Bakker69998dd2009-07-11 19:15:20 +0000[diff] [blame]49 int key_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]50
51 memset(key_str, 0x00, 100);
52 memset(src_str, 0x00, 100);
53 memset(dst_str, 0x00, 100);
54 memset(output, 0x00, 100);
55
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]56 key_len = unhexify( key_str, hex_key_string );
57 unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]58
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]59 TEST_ASSERT( aes_setkey_dec( &ctx, key_str, key_len * 8 ) == setkey_result );
60 if( setkey_result == 0 )
Paul Bakker2b222c82009-07-27 21:03:45 +0000[diff] [blame]61 {
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]62 TEST_ASSERT( aes_crypt_ecb( &ctx, AES_DECRYPT, src_str, output ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +0000[diff] [blame]63 hexify( dst_str, output, 16 );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]64
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]65 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker2b222c82009-07-27 21:03:45 +0000[diff] [blame]66 }
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]67}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]68/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]69
Manuel Pégourié-Gonnard92cb1d32013-09-13 16:24:20 +0200[diff] [blame]70/* BEGIN_CASE depends_on:POLARSSL_CIPHER_MODE_CBC */
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]71void aes_encrypt_cbc( char *hex_key_string, char *hex_iv_string,
72 char *hex_src_string, char *hex_dst_string,
73 int cbc_result )
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]74{
75 unsigned char key_str[100];
76 unsigned char iv_str[100];
77 unsigned char src_str[100];
78 unsigned char dst_str[100];
79 unsigned char output[100];
80 aes_context ctx;
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]81 int key_len, data_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]82
83 memset(key_str, 0x00, 100);
84 memset(iv_str, 0x00, 100);
85 memset(src_str, 0x00, 100);
86 memset(dst_str, 0x00, 100);
87 memset(output, 0x00, 100);
88
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]89 key_len = unhexify( key_str, hex_key_string );
90 unhexify( iv_str, hex_iv_string );
91 data_len = unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]92
93 aes_setkey_enc( &ctx, key_str, key_len * 8 );
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]94 TEST_ASSERT( aes_crypt_cbc( &ctx, AES_ENCRYPT, data_len, iv_str, src_str, output ) == cbc_result );
95 if( cbc_result == 0 )
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]96 {
97 hexify( dst_str, output, data_len );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]98
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]99 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]100 }
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]101}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]102/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]103
Manuel Pégourié-Gonnard92cb1d32013-09-13 16:24:20 +0200[diff] [blame]104/* BEGIN_CASE depends_on:POLARSSL_CIPHER_MODE_CBC */
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]105void aes_decrypt_cbc( char *hex_key_string, char *hex_iv_string,
106 char *hex_src_string, char *hex_dst_string,
107 int cbc_result )
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]108{
109 unsigned char key_str[100];
110 unsigned char iv_str[100];
111 unsigned char src_str[100];
112 unsigned char dst_str[100];
113 unsigned char output[100];
114 aes_context ctx;
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]115 int key_len, data_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]116
117 memset(key_str, 0x00, 100);
118 memset(iv_str, 0x00, 100);
119 memset(src_str, 0x00, 100);
120 memset(dst_str, 0x00, 100);
121 memset(output, 0x00, 100);
122
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]123 key_len = unhexify( key_str, hex_key_string );
124 unhexify( iv_str, hex_iv_string );
125 data_len = unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]126
127 aes_setkey_dec( &ctx, key_str, key_len * 8 );
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]128 TEST_ASSERT( aes_crypt_cbc( &ctx, AES_DECRYPT, data_len, iv_str, src_str, output ) == cbc_result );
129 if( cbc_result == 0)
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]130 {
131 hexify( dst_str, output, data_len );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]132
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]133 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]134 }
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]135}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]136/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]137
Manuel Pégourié-Gonnard387a2112013-09-18 18:54:01 +0200[diff] [blame]138/* BEGIN_CASE depends_on:POLARSSL_CIPHER_MODE_CFB */
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]139void aes_encrypt_cfb128( char *hex_key_string, char *hex_iv_string,
140 char *hex_src_string, char *hex_dst_string )
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]141{
142 unsigned char key_str[100];
143 unsigned char iv_str[100];
144 unsigned char src_str[100];
145 unsigned char dst_str[100];
146 unsigned char output[100];
147 aes_context ctx;
Paul Bakkercd43a0b2011-06-09 13:55:44 +0000[diff] [blame]148 size_t iv_offset = 0;
Paul Bakker69998dd2009-07-11 19:15:20 +0000[diff] [blame]149 int key_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]150
151 memset(key_str, 0x00, 100);
152 memset(iv_str, 0x00, 100);
153 memset(src_str, 0x00, 100);
154 memset(dst_str, 0x00, 100);
155 memset(output, 0x00, 100);
156
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]157 key_len = unhexify( key_str, hex_key_string );
158 unhexify( iv_str, hex_iv_string );
159 unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]160
161 aes_setkey_enc( &ctx, key_str, key_len * 8 );
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]162 TEST_ASSERT( aes_crypt_cfb128( &ctx, AES_ENCRYPT, 16, &iv_offset, iv_str, src_str, output ) == 0 );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]163 hexify( dst_str, output, 16 );
164
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]165 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]166}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]167/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]168
Manuel Pégourié-Gonnard387a2112013-09-18 18:54:01 +0200[diff] [blame]169/* BEGIN_CASE depends_on:POLARSSL_CIPHER_MODE_CFB */
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]170void aes_decrypt_cfb128( char *hex_key_string, char *hex_iv_string,
171 char *hex_src_string, char *hex_dst_string )
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]172{
173 unsigned char key_str[100];
174 unsigned char iv_str[100];
175 unsigned char src_str[100];
176 unsigned char dst_str[100];
177 unsigned char output[100];
178 aes_context ctx;
Paul Bakkercd43a0b2011-06-09 13:55:44 +0000[diff] [blame]179 size_t iv_offset = 0;
Paul Bakker69998dd2009-07-11 19:15:20 +0000[diff] [blame]180 int key_len;
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]181
182 memset(key_str, 0x00, 100);
183 memset(iv_str, 0x00, 100);
184 memset(src_str, 0x00, 100);
185 memset(dst_str, 0x00, 100);
186 memset(output, 0x00, 100);
187
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]188 key_len = unhexify( key_str, hex_key_string );
189 unhexify( iv_str, hex_iv_string );
190 unhexify( src_str, hex_src_string );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]191
192 aes_setkey_enc( &ctx, key_str, key_len * 8 );
Paul Bakkerf3ccc682010-03-18 21:21:02 +0000[diff] [blame]193 TEST_ASSERT( aes_crypt_cfb128( &ctx, AES_DECRYPT, 16, &iv_offset, iv_str, src_str, output ) == 0 );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]194 hexify( dst_str, output, 16 );
195
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]196 TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 );
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]197}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]198/* END_CASE */
Paul Bakker367dae42009-06-28 21:50:27 +0000[diff] [blame]199
Manuel Pégourié-Gonnard387a2112013-09-18 18:54:01 +0200[diff] [blame]200/* BEGIN_CASE depends_on:POLARSSL_SELFTEST_C */
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]201void aes_selftest()
Paul Bakker3d360822009-07-05 11:29:38 +0000[diff] [blame]202{
203 TEST_ASSERT( aes_self_test( 0 ) == 0 );
204}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]205/* END_CASE */