| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | /* BEGIN_HEADER */ | 
					
						
							|  |  |  | #include "mbedtls/aes.h" | 
					
						
							|  |  |  | /* END_HEADER */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_DEPENDENCIES | 
					
						
							|  |  |  |  * depends_on:MBEDTLS_AES_C | 
					
						
							|  |  |  |  * END_DEPENDENCIES | 
					
						
							|  |  |  |  */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE */ | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  | void aes_encrypt_ecb( data_t * key_str, data_t * src_str, | 
					
						
							|  |  |  |                       data_t * dst, int setkey_result ) | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | { | 
					
						
							|  |  |  |     unsigned char output[100]; | 
					
						
							|  |  |  |     mbedtls_aes_context ctx; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     memset(output, 0x00, 100); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     mbedtls_aes_init( &ctx ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     TEST_ASSERT( mbedtls_aes_setkey_enc( &ctx, key_str->x, key_str->len * 8 ) == setkey_result ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |     if( setkey_result == 0 ) | 
					
						
							|  |  |  |     { | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |         TEST_ASSERT( mbedtls_aes_crypt_ecb( &ctx, MBEDTLS_AES_ENCRYPT, src_str->x, output ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |         TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x, 16, dst->len ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | exit: | 
					
						
							|  |  |  |     mbedtls_aes_free( &ctx ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE */ | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  | void aes_decrypt_ecb( data_t * key_str, data_t * src_str, | 
					
						
							|  |  |  |                       data_t * dst, int setkey_result ) | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | { | 
					
						
							|  |  |  |     unsigned char output[100]; | 
					
						
							|  |  |  |     mbedtls_aes_context ctx; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     memset(output, 0x00, 100); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     mbedtls_aes_init( &ctx ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     TEST_ASSERT( mbedtls_aes_setkey_dec( &ctx, key_str->x, key_str->len * 8 ) == setkey_result ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |     if( setkey_result == 0 ) | 
					
						
							|  |  |  |     { | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |         TEST_ASSERT( mbedtls_aes_crypt_ecb( &ctx, MBEDTLS_AES_DECRYPT, src_str->x, output ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |         TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x, 16, dst->len ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | exit: | 
					
						
							|  |  |  |     mbedtls_aes_free( &ctx ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CBC */ | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  | void aes_encrypt_cbc( data_t * key_str, data_t * iv_str, | 
					
						
							|  |  |  |                       data_t * src_str, data_t * dst, | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |                       int cbc_result ) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     unsigned char output[100]; | 
					
						
							|  |  |  |     mbedtls_aes_context ctx; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     memset(output, 0x00, 100); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     mbedtls_aes_init( &ctx ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     mbedtls_aes_setkey_enc( &ctx, key_str->x, key_str->len * 8 ); | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_cbc( &ctx, MBEDTLS_AES_ENCRYPT, src_str->len, iv_str->x, src_str->x, output ) == cbc_result ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |     if( cbc_result == 0 ) | 
					
						
							|  |  |  |     { | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |         TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x, | 
					
						
							|  |  |  |                                           src_str->len, dst->len ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | exit: | 
					
						
							|  |  |  |     mbedtls_aes_free( &ctx ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CBC */ | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  | void aes_decrypt_cbc( data_t * key_str, data_t * iv_str, | 
					
						
							|  |  |  |                       data_t * src_str, data_t * dst, | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |                       int cbc_result ) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     unsigned char output[100]; | 
					
						
							|  |  |  |     mbedtls_aes_context ctx; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     memset(output, 0x00, 100); | 
					
						
							|  |  |  |     mbedtls_aes_init( &ctx ); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     mbedtls_aes_setkey_dec( &ctx, key_str->x, key_str->len * 8 ); | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_cbc( &ctx, MBEDTLS_AES_DECRYPT, src_str->len, iv_str->x, src_str->x, output ) == cbc_result ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |     if( cbc_result == 0) | 
					
						
							|  |  |  |     { | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |         TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x, | 
					
						
							|  |  |  |                                           src_str->len, dst->len ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | exit: | 
					
						
							|  |  |  |     mbedtls_aes_free( &ctx ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_XTS */ | 
					
						
							|  |  |  | void aes_encrypt_xts( char *hex_key_string, char *hex_data_unit_string, | 
					
						
							|  |  |  |                       char *hex_src_string, char *hex_dst_string ) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     enum { AES_BLOCK_SIZE = 16 }; | 
					
						
							|  |  |  |     unsigned char *data_unit = NULL; | 
					
						
							|  |  |  |     unsigned char *key = NULL; | 
					
						
							|  |  |  |     unsigned char *src = NULL; | 
					
						
							|  |  |  |     unsigned char *dst = NULL; | 
					
						
							|  |  |  |     unsigned char *output = NULL; | 
					
						
							|  |  |  |     mbedtls_aes_xts_context ctx; | 
					
						
							|  |  |  |     size_t key_len, src_len, dst_len, data_unit_len; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     mbedtls_aes_xts_init( &ctx ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     data_unit = unhexify_alloc( hex_data_unit_string, &data_unit_len ); | 
					
						
							|  |  |  |     TEST_ASSERT( data_unit_len == AES_BLOCK_SIZE ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     key = unhexify_alloc( hex_key_string, &key_len ); | 
					
						
							|  |  |  |     TEST_ASSERT( key_len % 2 == 0 ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     src = unhexify_alloc( hex_src_string, &src_len ); | 
					
						
							|  |  |  |     dst = unhexify_alloc( hex_dst_string, &dst_len ); | 
					
						
							|  |  |  |     TEST_ASSERT( src_len == dst_len ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     output = zero_alloc( dst_len ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_xts_setkey_enc( &ctx, key, key_len * 8 ) == 0 ); | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_xts( &ctx, MBEDTLS_AES_ENCRYPT, src_len, | 
					
						
							|  |  |  |                                         data_unit, src, output ) == 0 ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     TEST_ASSERT( memcmp( output, dst, dst_len ) == 0 ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | exit: | 
					
						
							|  |  |  |     mbedtls_aes_xts_free( &ctx ); | 
					
						
							|  |  |  |     mbedtls_free( data_unit ); | 
					
						
							|  |  |  |     mbedtls_free( key ); | 
					
						
							|  |  |  |     mbedtls_free( src ); | 
					
						
							|  |  |  |     mbedtls_free( dst ); | 
					
						
							|  |  |  |     mbedtls_free( output ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_XTS */ | 
					
						
							|  |  |  | void aes_decrypt_xts( char *hex_key_string, char *hex_data_unit_string, | 
					
						
							|  |  |  |                       char *hex_dst_string, char *hex_src_string ) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     enum { AES_BLOCK_SIZE = 16 }; | 
					
						
							|  |  |  |     unsigned char *data_unit = NULL; | 
					
						
							|  |  |  |     unsigned char *key = NULL; | 
					
						
							|  |  |  |     unsigned char *src = NULL; | 
					
						
							|  |  |  |     unsigned char *dst = NULL; | 
					
						
							|  |  |  |     unsigned char *output = NULL; | 
					
						
							|  |  |  |     mbedtls_aes_xts_context ctx; | 
					
						
							|  |  |  |     size_t key_len, src_len, dst_len, data_unit_len; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     mbedtls_aes_xts_init( &ctx ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     data_unit = unhexify_alloc( hex_data_unit_string, &data_unit_len ); | 
					
						
							|  |  |  |     TEST_ASSERT( data_unit_len == AES_BLOCK_SIZE ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     key = unhexify_alloc( hex_key_string, &key_len ); | 
					
						
							|  |  |  |     TEST_ASSERT( key_len % 2 == 0 ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     src = unhexify_alloc( hex_src_string, &src_len ); | 
					
						
							|  |  |  |     dst = unhexify_alloc( hex_dst_string, &dst_len ); | 
					
						
							|  |  |  |     TEST_ASSERT( src_len == dst_len ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     output = zero_alloc( dst_len ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_xts_setkey_dec( &ctx, key, key_len * 8 ) == 0 ); | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_xts( &ctx, MBEDTLS_AES_DECRYPT, src_len, | 
					
						
							|  |  |  |                                         data_unit, src, output ) == 0 ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     TEST_ASSERT( memcmp( output, dst, dst_len ) == 0 ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | exit: | 
					
						
							|  |  |  |     mbedtls_aes_xts_free( &ctx ); | 
					
						
							|  |  |  |     mbedtls_free( data_unit ); | 
					
						
							|  |  |  |     mbedtls_free( key ); | 
					
						
							|  |  |  |     mbedtls_free( src ); | 
					
						
							|  |  |  |     mbedtls_free( dst ); | 
					
						
							|  |  |  |     mbedtls_free( output ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_XTS */ | 
					
						
							|  |  |  | void aes_crypt_xts_size( int size, int retval ) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     mbedtls_aes_xts_context ctx; | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     const unsigned char src[16] = { 0 }; | 
					
						
							|  |  |  |     unsigned char output[16]; | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |     unsigned char data_unit[16]; | 
					
						
							|  |  |  |     size_t length = size; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     mbedtls_aes_xts_init( &ctx ); | 
					
						
							|  |  |  |     memset( data_unit, 0x00, sizeof( data_unit ) ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     /* Valid pointers are passed for builds with MBEDTLS_CHECK_PARAMS, as | 
					
						
							|  |  |  |      * otherwise we wouldn't get to the size check we're interested in. */ | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |     TEST_ASSERT( mbedtls_aes_crypt_xts( &ctx, MBEDTLS_AES_ENCRYPT, length, data_unit, src, output ) == retval ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_XTS */ | 
					
						
							|  |  |  | void aes_crypt_xts_keysize( int size, int retval ) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     mbedtls_aes_xts_context ctx; | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     const unsigned char key[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06 }; | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |     size_t key_len = size; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     mbedtls_aes_xts_init( &ctx ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_xts_setkey_enc( &ctx, key, key_len * 8 ) == retval ); | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_xts_setkey_dec( &ctx, key, key_len * 8 ) == retval ); | 
					
						
							|  |  |  | exit: | 
					
						
							|  |  |  |     mbedtls_aes_xts_free( &ctx ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */ | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  | void aes_encrypt_cfb128( data_t * key_str, data_t * iv_str, | 
					
						
							|  |  |  |                          data_t * src_str, data_t * dst ) | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | { | 
					
						
							|  |  |  |     unsigned char output[100]; | 
					
						
							|  |  |  |     mbedtls_aes_context ctx; | 
					
						
							|  |  |  |     size_t iv_offset = 0; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     memset(output, 0x00, 100); | 
					
						
							|  |  |  |     mbedtls_aes_init( &ctx ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     mbedtls_aes_setkey_enc( &ctx, key_str->x, key_str->len * 8 ); | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_cfb128( &ctx, MBEDTLS_AES_ENCRYPT, 16, &iv_offset, iv_str->x, src_str->x, output ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x, 16, dst->len ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | exit: | 
					
						
							|  |  |  |     mbedtls_aes_free( &ctx ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */ | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  | void aes_decrypt_cfb128( data_t * key_str, data_t * iv_str, | 
					
						
							|  |  |  |                          data_t * src_str, data_t * dst ) | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | { | 
					
						
							|  |  |  |     unsigned char output[100]; | 
					
						
							|  |  |  |     mbedtls_aes_context ctx; | 
					
						
							|  |  |  |     size_t iv_offset = 0; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     memset(output, 0x00, 100); | 
					
						
							|  |  |  |     mbedtls_aes_init( &ctx ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     mbedtls_aes_setkey_enc( &ctx, key_str->x, key_str->len * 8 ); | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_cfb128( &ctx, MBEDTLS_AES_DECRYPT, 16, &iv_offset, iv_str->x, src_str->x, output ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x, 16, dst->len ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | exit: | 
					
						
							|  |  |  |     mbedtls_aes_free( &ctx ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */ | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  | void aes_encrypt_cfb8( data_t * key_str, data_t * iv_str, | 
					
						
							|  |  |  |                        data_t * src_str, data_t * dst ) | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | { | 
					
						
							|  |  |  |     unsigned char output[100]; | 
					
						
							|  |  |  |     mbedtls_aes_context ctx; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     memset(output, 0x00, 100); | 
					
						
							|  |  |  |     mbedtls_aes_init( &ctx ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     mbedtls_aes_setkey_enc( &ctx, key_str->x, key_str->len * 8 ); | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_cfb8( &ctx, MBEDTLS_AES_ENCRYPT, src_str->len, iv_str->x, src_str->x, output ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x, | 
					
						
							|  |  |  |                                       src_str->len, dst->len ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | exit: | 
					
						
							|  |  |  |     mbedtls_aes_free( &ctx ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */ | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  | void aes_decrypt_cfb8( data_t * key_str, data_t * iv_str, | 
					
						
							|  |  |  |                        data_t * src_str, data_t * dst ) | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | { | 
					
						
							|  |  |  |     unsigned char output[100]; | 
					
						
							|  |  |  |     mbedtls_aes_context ctx; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     memset(output, 0x00, 100); | 
					
						
							|  |  |  |     mbedtls_aes_init( &ctx ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     mbedtls_aes_setkey_enc( &ctx, key_str->x, key_str->len * 8 ); | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_cfb8( &ctx, MBEDTLS_AES_DECRYPT, src_str->len, iv_str->x, src_str->x, output ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     TEST_ASSERT( mbedtls_test_hexcmp( output, dst->x, | 
					
						
							|  |  |  |                                       src_str->len, dst->len ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | exit: | 
					
						
							|  |  |  |     mbedtls_aes_free( &ctx ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_OFB */ | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  | void aes_encrypt_ofb( int fragment_size, data_t *key_str, | 
					
						
							|  |  |  |                       data_t *iv_str, data_t *src_str, | 
					
						
							|  |  |  |                       data_t *expected_output ) | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | { | 
					
						
							|  |  |  |     unsigned char output[32]; | 
					
						
							|  |  |  |     mbedtls_aes_context ctx; | 
					
						
							|  |  |  |     size_t iv_offset = 0; | 
					
						
							|  |  |  |     int in_buffer_len; | 
					
						
							|  |  |  |     unsigned char* src_str_next; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     memset( output, 0x00, sizeof( output ) ); | 
					
						
							|  |  |  |     mbedtls_aes_init( &ctx ); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     TEST_ASSERT( (size_t)fragment_size < sizeof( output ) ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |     TEST_ASSERT( mbedtls_aes_setkey_enc( &ctx, key_str->x, | 
					
						
							|  |  |  |                                          key_str->len * 8 ) == 0 ); | 
					
						
							|  |  |  |     in_buffer_len = src_str->len; | 
					
						
							|  |  |  |     src_str_next = src_str->x; | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |     while( in_buffer_len > 0 ) | 
					
						
							|  |  |  |     { | 
					
						
							|  |  |  |         TEST_ASSERT( mbedtls_aes_crypt_ofb( &ctx, fragment_size, &iv_offset, | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |                                             iv_str->x, src_str_next, output ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |         TEST_ASSERT( memcmp( output, expected_output->x, fragment_size ) == 0 ); | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |         in_buffer_len -= fragment_size; | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  |         expected_output->x += fragment_size; | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  |         src_str_next += fragment_size; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if( in_buffer_len < fragment_size ) | 
					
						
							|  |  |  |             fragment_size = in_buffer_len; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | exit: | 
					
						
							|  |  |  |     mbedtls_aes_free( &ctx ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  | /* BEGIN_CASE depends_on:MBEDTLS_CHECK_PARAMS:!MBEDTLS_PARAM_FAILED_ALT */ | 
					
						
							|  |  |  | void aes_check_params( ) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     mbedtls_aes_context aes_ctx; | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_XTS) | 
					
						
							|  |  |  |     mbedtls_aes_xts_context xts_ctx; | 
					
						
							|  |  |  | #endif | 
					
						
							|  |  |  |     const unsigned char key[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06 }; | 
					
						
							|  |  |  |     const unsigned char in[16] = { 0 }; | 
					
						
							|  |  |  |     unsigned char out[16]; | 
					
						
							|  |  |  |     size_t size; | 
					
						
							|  |  |  |     const int valid_mode = MBEDTLS_AES_ENCRYPT; | 
					
						
							|  |  |  |     const int invalid_mode = 42; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     TEST_INVALID_PARAM( mbedtls_aes_init( NULL ) ); | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_XTS) | 
					
						
							|  |  |  |     TEST_INVALID_PARAM( mbedtls_aes_xts_init( NULL ) ); | 
					
						
							|  |  |  | #endif | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_setkey_enc( NULL, key, 128 ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_setkey_enc( &aes_ctx, NULL, 128 ) ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_setkey_dec( NULL, key, 128 ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_setkey_dec( &aes_ctx, NULL, 128 ) ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_XTS) | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_xts_setkey_enc( NULL, key, 128 ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_xts_setkey_enc( &xts_ctx, NULL, 128 ) ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_xts_setkey_dec( NULL, key, 128 ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_xts_setkey_dec( &xts_ctx, NULL, 128 ) ); | 
					
						
							|  |  |  | #endif | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ecb( NULL, | 
					
						
							|  |  |  |                                                    valid_mode, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ecb( &aes_ctx, | 
					
						
							|  |  |  |                                                    invalid_mode, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ecb( &aes_ctx, | 
					
						
							|  |  |  |                                                    valid_mode, NULL, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ecb( &aes_ctx, | 
					
						
							|  |  |  |                                                    valid_mode, in, NULL ) ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_CBC) | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cbc( NULL, | 
					
						
							|  |  |  |                                                    valid_mode, 16, | 
					
						
							|  |  |  |                                                    out, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cbc( &aes_ctx, | 
					
						
							|  |  |  |                                                    invalid_mode, 16, | 
					
						
							|  |  |  |                                                    out, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cbc( &aes_ctx, | 
					
						
							|  |  |  |                                                    valid_mode, 16, | 
					
						
							|  |  |  |                                                    NULL, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cbc( &aes_ctx, | 
					
						
							|  |  |  |                                                    valid_mode, 16, | 
					
						
							|  |  |  |                                                    out, NULL, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cbc( &aes_ctx, | 
					
						
							|  |  |  |                                                    valid_mode, 16, | 
					
						
							|  |  |  |                                                    out, in, NULL ) ); | 
					
						
							|  |  |  | #endif /* MBEDTLS_CIPHER_MODE_CBC */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_XTS) | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_xts( NULL, | 
					
						
							|  |  |  |                                                    valid_mode, 16, | 
					
						
							|  |  |  |                                                    in, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_xts( &xts_ctx, | 
					
						
							|  |  |  |                                                    invalid_mode, 16, | 
					
						
							|  |  |  |                                                    in, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_xts( &xts_ctx, | 
					
						
							|  |  |  |                                                    valid_mode, 16, | 
					
						
							|  |  |  |                                                    NULL, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_xts( &xts_ctx, | 
					
						
							|  |  |  |                                                    valid_mode, 16, | 
					
						
							|  |  |  |                                                    in, NULL, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_xts( &xts_ctx, | 
					
						
							|  |  |  |                                                    valid_mode, 16, | 
					
						
							|  |  |  |                                                    in, in, NULL ) ); | 
					
						
							|  |  |  | #endif /* MBEDTLS_CIPHER_MODE_XTS */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_CFB) | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cfb128( NULL, | 
					
						
							|  |  |  |                                                       valid_mode, 16, | 
					
						
							|  |  |  |                                                       &size, out, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cfb128( &aes_ctx, | 
					
						
							|  |  |  |                                                       invalid_mode, 16, | 
					
						
							|  |  |  |                                                       &size, out, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cfb128( &aes_ctx, | 
					
						
							|  |  |  |                                                       valid_mode, 16, | 
					
						
							|  |  |  |                                                       NULL, out, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cfb128( &aes_ctx, | 
					
						
							|  |  |  |                                                       valid_mode, 16, | 
					
						
							|  |  |  |                                                       &size, NULL, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cfb128( &aes_ctx, | 
					
						
							|  |  |  |                                                       valid_mode, 16, | 
					
						
							|  |  |  |                                                       &size, out, NULL, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cfb128( &aes_ctx, | 
					
						
							|  |  |  |                                                       valid_mode, 16, | 
					
						
							|  |  |  |                                                       &size, out, in, NULL ) ); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cfb8( NULL, | 
					
						
							|  |  |  |                                                     valid_mode, 16, | 
					
						
							|  |  |  |                                                     out, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cfb8( &aes_ctx, | 
					
						
							|  |  |  |                                                     invalid_mode, 16, | 
					
						
							|  |  |  |                                                     out, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cfb8( &aes_ctx, | 
					
						
							|  |  |  |                                                     valid_mode, 16, | 
					
						
							|  |  |  |                                                     NULL, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cfb8( &aes_ctx, | 
					
						
							|  |  |  |                                                     valid_mode, 16, | 
					
						
							|  |  |  |                                                     out, NULL, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_cfb8( &aes_ctx, | 
					
						
							|  |  |  |                                                     valid_mode, 16, | 
					
						
							|  |  |  |                                                     out, in, NULL ) ); | 
					
						
							|  |  |  | #endif /* MBEDTLS_CIPHER_MODE_CFB */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_OFB) | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ofb( NULL, 16, | 
					
						
							|  |  |  |                                                    &size, out, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ofb( &aes_ctx, 16, | 
					
						
							|  |  |  |                                                    NULL, out, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ofb( &aes_ctx, 16, | 
					
						
							|  |  |  |                                                    &size, NULL, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ofb( &aes_ctx, 16, | 
					
						
							|  |  |  |                                                    &size, out, NULL, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ofb( &aes_ctx, 16, | 
					
						
							|  |  |  |                                                    &size, out, in, NULL ) ); | 
					
						
							|  |  |  | #endif /* MBEDTLS_CIPHER_MODE_OFB */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_CTR) | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ctr( NULL, 16, &size, out, | 
					
						
							|  |  |  |                                                    out, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ctr( &aes_ctx, 16, NULL, out, | 
					
						
							|  |  |  |                                                    out, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ctr( &aes_ctx, 16, &size, NULL, | 
					
						
							|  |  |  |                                                    out, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ctr( &aes_ctx, 16, &size, out, | 
					
						
							|  |  |  |                                                    NULL, in, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ctr( &aes_ctx, 16, &size, out, | 
					
						
							|  |  |  |                                                    out, NULL, out ) ); | 
					
						
							|  |  |  |     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA, | 
					
						
							|  |  |  |                             mbedtls_aes_crypt_ctr( &aes_ctx, 16, &size, out, | 
					
						
							|  |  |  |                                                    out, in, NULL ) ); | 
					
						
							|  |  |  | #endif /* MBEDTLS_CIPHER_MODE_CTR */ | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* BEGIN_CASE */ | 
					
						
							|  |  |  | void aes_misc_params( ) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_CBC) || \ | 
					
						
							|  |  |  |     defined(MBEDTLS_CIPHER_MODE_XTS) || \ | 
					
						
							|  |  |  |     defined(MBEDTLS_CIPHER_MODE_CFB) || \ | 
					
						
							|  |  |  |     defined(MBEDTLS_CIPHER_MODE_OFB) | 
					
						
							|  |  |  |     mbedtls_aes_context aes_ctx; | 
					
						
							|  |  |  |     const unsigned char in[16] = { 0 }; | 
					
						
							|  |  |  |     unsigned char out[16]; | 
					
						
							|  |  |  | #endif | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_XTS) | 
					
						
							|  |  |  |     mbedtls_aes_xts_context xts_ctx; | 
					
						
							|  |  |  | #endif | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_CFB) || \ | 
					
						
							|  |  |  |     defined(MBEDTLS_CIPHER_MODE_OFB) | 
					
						
							|  |  |  |     size_t size; | 
					
						
							|  |  |  | #endif | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* These calls accept NULL */ | 
					
						
							|  |  |  |     TEST_VALID_PARAM( mbedtls_aes_free( NULL ) ); | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_XTS) | 
					
						
							|  |  |  |     TEST_VALID_PARAM( mbedtls_aes_xts_free( NULL ) ); | 
					
						
							|  |  |  | #endif | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_CBC) | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_cbc( &aes_ctx, MBEDTLS_AES_ENCRYPT, | 
					
						
							|  |  |  |                                         15, | 
					
						
							|  |  |  |                                         out, in, out ) | 
					
						
							|  |  |  |                  == MBEDTLS_ERR_AES_INVALID_INPUT_LENGTH ); | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_cbc( &aes_ctx, MBEDTLS_AES_ENCRYPT, | 
					
						
							|  |  |  |                                         17, | 
					
						
							|  |  |  |                                         out, in, out ) | 
					
						
							|  |  |  |                  == MBEDTLS_ERR_AES_INVALID_INPUT_LENGTH ); | 
					
						
							|  |  |  | #endif | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_XTS) | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_xts( &xts_ctx, MBEDTLS_AES_ENCRYPT, | 
					
						
							|  |  |  |                                         15, | 
					
						
							|  |  |  |                                         in, in, out ) | 
					
						
							|  |  |  |                  == MBEDTLS_ERR_AES_INVALID_INPUT_LENGTH ); | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_xts( &xts_ctx, MBEDTLS_AES_ENCRYPT, | 
					
						
							|  |  |  |                                         (1 << 24) + 1, | 
					
						
							|  |  |  |                                         in, in, out ) | 
					
						
							|  |  |  |                  == MBEDTLS_ERR_AES_INVALID_INPUT_LENGTH ); | 
					
						
							|  |  |  | #endif | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_CFB) | 
					
						
							|  |  |  |     size = 16; | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_cfb128( &aes_ctx, MBEDTLS_AES_ENCRYPT, 16, | 
					
						
							|  |  |  |                                            &size, out, in, out ) | 
					
						
							|  |  |  |                  == MBEDTLS_ERR_AES_BAD_INPUT_DATA ); | 
					
						
							|  |  |  | #endif | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #if defined(MBEDTLS_CIPHER_MODE_OFB) | 
					
						
							|  |  |  |     size = 16; | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_crypt_ofb( &aes_ctx, 16, &size, out, in, out ) | 
					
						
							|  |  |  |                  == MBEDTLS_ERR_AES_BAD_INPUT_DATA ); | 
					
						
							|  |  |  | #endif | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | /* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */ | 
					
						
							| 
									
										
										
										
											2021-03-06 01:41:47 +01:00
										 |  |  | void aes_selftest(  ) | 
					
						
							| 
									
										
										
										
											2020-12-28 15:15:37 +00:00
										 |  |  | { | 
					
						
							|  |  |  |     TEST_ASSERT( mbedtls_aes_self_test( 1 ) == 0 ); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* END_CASE */ |