| /* BEGIN_HEADER */ |
| #include "mbedtls/chacha20.h" |
| /* END_HEADER */ |
| |
| /* BEGIN_DEPENDENCIES |
| * depends_on:MBEDTLS_CHACHA20_C |
| * END_DEPENDENCIES |
| */ |
| |
| /* BEGIN_CASE */ |
| void chacha20_crypt( char *hex_key_string, |
| char *hex_nonce_string, |
| int counter, |
| char *hex_src_string, |
| char *hex_dst_string ) |
| { |
| unsigned char key_str[32]; /* size set by the standard */ |
| unsigned char nonce_str[12]; /* size set by the standard */ |
| unsigned char src_str[375]; /* max size of binary input */ |
| unsigned char dst_str[751]; /* hex expansion of the above */ |
| unsigned char output[751]; |
| size_t key_len; |
| size_t nonce_len; |
| size_t src_len; |
| size_t dst_len; |
| |
| memset( key_str, 0x00, sizeof( key_str ) ); |
| memset( nonce_str, 0x00, sizeof( nonce_str ) ); |
| memset( src_str, 0x00, sizeof( src_str ) ); |
| memset( dst_str, 0x00, sizeof( dst_str ) ); |
| memset( output, 0x00, sizeof( output ) ); |
| |
| key_len = unhexify( key_str, hex_key_string ); |
| nonce_len = unhexify( nonce_str, hex_nonce_string ); |
| src_len = unhexify( src_str, hex_src_string ); |
| dst_len = unhexify( dst_str, hex_dst_string ); |
| |
| TEST_ASSERT( src_len == dst_len ); |
| TEST_ASSERT( key_len == 32U ); |
| TEST_ASSERT( nonce_len == 12U ); |
| |
| TEST_ASSERT( mbedtls_chacha20_crypt( key_str, nonce_str, counter, src_len, src_str, output ) == 0 ); |
| |
| hexify( dst_str, output, src_len ); |
| |
| TEST_ASSERT( strcmp( (char*) dst_str, hex_dst_string ) == 0); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */ |
| void chacha20_self_test() |
| { |
| TEST_ASSERT( mbedtls_chacha20_self_test( 0 ) == 0 ); |
| } |
| /* END_CASE */ |