blob: 9c0b98522bc54aea266f4d451cfafebb0266ae77 [file] [log] [blame]
Daniel King34b822c2016-05-15 17:28:08 -03001/* BEGIN_HEADER */
2#include "mbedtls/chacha20.h"
3/* END_HEADER */
4
5/* BEGIN_DEPENDENCIES
6 * depends_on:MBEDTLS_CHACHA20_C
7 * END_DEPENDENCIES
8 */
Daniel King6155cc82016-05-18 11:51:22 -03009
10/* BEGIN_CASE */
11void chacha20_crypt( char *hex_key_string,
12 char *hex_nonce_string,
13 int counter,
14 char *hex_src_string,
15 char *hex_dst_string )
16{
Manuel Pégourié-Gonnard528524b2018-05-09 11:21:21 +020017 unsigned char key_str[32]; /* size set by the standard */
18 unsigned char nonce_str[12]; /* size set by the standard */
19 unsigned char src_str[375]; /* max size of binary input */
20 unsigned char dst_str[751]; /* hex expansion of the above */
21 unsigned char output[751];
Daniel King6155cc82016-05-18 11:51:22 -030022 size_t key_len;
23 size_t nonce_len;
24 size_t src_len;
25 size_t dst_len;
26
Manuel Pégourié-Gonnard528524b2018-05-09 11:21:21 +020027 memset( key_str, 0x00, sizeof( key_str ) );
28 memset( nonce_str, 0x00, sizeof( nonce_str ) );
29 memset( src_str, 0x00, sizeof( src_str ) );
30 memset( dst_str, 0x00, sizeof( dst_str ) );
31 memset( output, 0x00, sizeof( output ) );
Daniel King6155cc82016-05-18 11:51:22 -030032
33 key_len = unhexify( key_str, hex_key_string );
34 nonce_len = unhexify( nonce_str, hex_nonce_string );
35 src_len = unhexify( src_str, hex_src_string );
36 dst_len = unhexify( dst_str, hex_dst_string );
37
38 TEST_ASSERT( src_len == dst_len );
39 TEST_ASSERT( key_len == 32U );
40 TEST_ASSERT( nonce_len == 12U );
41
42 TEST_ASSERT( mbedtls_chacha20_crypt( key_str, nonce_str, counter, src_len, src_str, output ) == 0 );
43
44 hexify( dst_str, output, src_len );
45
46 TEST_ASSERT( strcmp( (char*) dst_str, hex_dst_string ) == 0);
47}
48/* END_CASE */
49
Daniel King34b822c2016-05-15 17:28:08 -030050/* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */
51void chacha20_self_test()
52{
53 TEST_ASSERT( mbedtls_chacha20_self_test( 0 ) == 0 );
54}
Manuel Pégourié-Gonnard528524b2018-05-09 11:21:21 +020055/* END_CASE */