Subversion Repositories Kolibri OS

Rev

Blame | Last modification | View Log | Download | RSS feed

  1. /**
  2.  * \file chacha20.c
  3.  *
  4.  * \brief ChaCha20 cipher.
  5.  *
  6.  * \author Daniel King <damaki.gh@gmail.com>
  7.  *
  8.  *  Copyright (C) 2006-2016, ARM Limited, All Rights Reserved
  9.  *  SPDX-License-Identifier: GPL-2.0
  10.  *
  11.  *  This program is free software; you can redistribute it and/or modify
  12.  *  it under the terms of the GNU General Public License as published by
  13.  *  the Free Software Foundation; either version 2 of the License, or
  14.  *  (at your option) any later version.
  15.  *
  16.  *  This program is distributed in the hope that it will be useful,
  17.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  18.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19.  *  GNU General Public License for more details.
  20.  *
  21.  *  You should have received a copy of the GNU General Public License along
  22.  *  with this program; if not, write to the Free Software Foundation, Inc.,
  23.  *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  24.  *
  25.  *  This file is part of mbed TLS (https://tls.mbed.org)
  26.  */
  27.  
  28. #if !defined(MBEDTLS_CONFIG_FILE)
  29. #include "mbedtls/config.h"
  30. #else
  31. #include MBEDTLS_CONFIG_FILE
  32. #endif
  33.  
  34. #if defined(MBEDTLS_CHACHA20_C)
  35.  
  36. #include "mbedtls/chacha20.h"
  37. #include "mbedtls/platform_util.h"
  38.  
  39. #include <stddef.h>
  40. #include <string.h>
  41.  
  42. #if defined(MBEDTLS_SELF_TEST)
  43. #if defined(MBEDTLS_PLATFORM_C)
  44. #include "mbedtls/platform.h"
  45. #else
  46. #include <stdio.h>
  47. #define mbedtls_printf printf
  48. #endif /* MBEDTLS_PLATFORM_C */
  49. #endif /* MBEDTLS_SELF_TEST */
  50.  
  51. #if !defined(MBEDTLS_CHACHA20_ALT)
  52.  
  53. #if ( defined(__ARMCC_VERSION) || defined(_MSC_VER) ) && \
  54.     !defined(inline) && !defined(__cplusplus)
  55. #define inline __inline
  56. #endif
  57.  
  58. /* Parameter validation macros */
  59. #define CHACHA20_VALIDATE_RET( cond )                                       \
  60.     MBEDTLS_INTERNAL_VALIDATE_RET( cond, MBEDTLS_ERR_CHACHA20_BAD_INPUT_DATA )
  61. #define CHACHA20_VALIDATE( cond )                                           \
  62.     MBEDTLS_INTERNAL_VALIDATE( cond )
  63.  
  64. #define BYTES_TO_U32_LE( data, offset )                           \
  65.     ( (uint32_t) (data)[offset]                                   \
  66.       | (uint32_t) ( (uint32_t) (data)[( offset ) + 1] << 8 )     \
  67.       | (uint32_t) ( (uint32_t) (data)[( offset ) + 2] << 16 )    \
  68.       | (uint32_t) ( (uint32_t) (data)[( offset ) + 3] << 24 )    \
  69.     )
  70.  
  71. #define ROTL32( value, amount ) \
  72.     ( (uint32_t) ( (value) << (amount) ) | ( (value) >> ( 32 - (amount) ) ) )
  73.  
  74. #define CHACHA20_CTR_INDEX ( 12U )
  75.  
  76. #define CHACHA20_BLOCK_SIZE_BYTES ( 4U * 16U )
  77.  
  78. /**
  79.  * \brief           ChaCha20 quarter round operation.
  80.  *
  81.  *                  The quarter round is defined as follows (from RFC 7539):
  82.  *                      1.  a += b; d ^= a; d <<<= 16;
  83.  *                      2.  c += d; b ^= c; b <<<= 12;
  84.  *                      3.  a += b; d ^= a; d <<<= 8;
  85.  *                      4.  c += d; b ^= c; b <<<= 7;
  86.  *
  87.  * \param state     ChaCha20 state to modify.
  88.  * \param a         The index of 'a' in the state.
  89.  * \param b         The index of 'b' in the state.
  90.  * \param c         The index of 'c' in the state.
  91.  * \param d         The index of 'd' in the state.
  92.  */
  93. static inline void chacha20_quarter_round( uint32_t state[16],
  94.                                            size_t a,
  95.                                            size_t b,
  96.                                            size_t c,
  97.                                            size_t d )
  98. {
  99.     /* a += b; d ^= a; d <<<= 16; */
  100.     state[a] += state[b];
  101.     state[d] ^= state[a];
  102.     state[d] = ROTL32( state[d], 16 );
  103.  
  104.     /* c += d; b ^= c; b <<<= 12 */
  105.     state[c] += state[d];
  106.     state[b] ^= state[c];
  107.     state[b] = ROTL32( state[b], 12 );
  108.  
  109.     /* a += b; d ^= a; d <<<= 8; */
  110.     state[a] += state[b];
  111.     state[d] ^= state[a];
  112.     state[d] = ROTL32( state[d], 8 );
  113.  
  114.     /* c += d; b ^= c; b <<<= 7; */
  115.     state[c] += state[d];
  116.     state[b] ^= state[c];
  117.     state[b] = ROTL32( state[b], 7 );
  118. }
  119.  
  120. /**
  121.  * \brief           Perform the ChaCha20 inner block operation.
  122.  *
  123.  *                  This function performs two rounds: the column round and the
  124.  *                  diagonal round.
  125.  *
  126.  * \param state     The ChaCha20 state to update.
  127.  */
  128. static void chacha20_inner_block( uint32_t state[16] )
  129. {
  130.     chacha20_quarter_round( state, 0, 4, 8,  12 );
  131.     chacha20_quarter_round( state, 1, 5, 9,  13 );
  132.     chacha20_quarter_round( state, 2, 6, 10, 14 );
  133.     chacha20_quarter_round( state, 3, 7, 11, 15 );
  134.  
  135.     chacha20_quarter_round( state, 0, 5, 10, 15 );
  136.     chacha20_quarter_round( state, 1, 6, 11, 12 );
  137.     chacha20_quarter_round( state, 2, 7, 8,  13 );
  138.     chacha20_quarter_round( state, 3, 4, 9,  14 );
  139. }
  140.  
  141. /**
  142.  * \brief               Generates a keystream block.
  143.  *
  144.  * \param initial_state The initial ChaCha20 state (key, nonce, counter).
  145.  * \param keystream     Generated keystream bytes are written to this buffer.
  146.  */
  147. static void chacha20_block( const uint32_t initial_state[16],
  148.                             unsigned char keystream[64] )
  149. {
  150.     uint32_t working_state[16];
  151.     size_t i;
  152.  
  153.     memcpy( working_state,
  154.             initial_state,
  155.             CHACHA20_BLOCK_SIZE_BYTES );
  156.  
  157.     for( i = 0U; i < 10U; i++ )
  158.         chacha20_inner_block( working_state );
  159.  
  160.     working_state[ 0] += initial_state[ 0];
  161.     working_state[ 1] += initial_state[ 1];
  162.     working_state[ 2] += initial_state[ 2];
  163.     working_state[ 3] += initial_state[ 3];
  164.     working_state[ 4] += initial_state[ 4];
  165.     working_state[ 5] += initial_state[ 5];
  166.     working_state[ 6] += initial_state[ 6];
  167.     working_state[ 7] += initial_state[ 7];
  168.     working_state[ 8] += initial_state[ 8];
  169.     working_state[ 9] += initial_state[ 9];
  170.     working_state[10] += initial_state[10];
  171.     working_state[11] += initial_state[11];
  172.     working_state[12] += initial_state[12];
  173.     working_state[13] += initial_state[13];
  174.     working_state[14] += initial_state[14];
  175.     working_state[15] += initial_state[15];
  176.  
  177.     for( i = 0U; i < 16; i++ )
  178.     {
  179.         size_t offset = i * 4U;
  180.  
  181.         keystream[offset     ] = (unsigned char)( working_state[i]       );
  182.         keystream[offset + 1U] = (unsigned char)( working_state[i] >>  8 );
  183.         keystream[offset + 2U] = (unsigned char)( working_state[i] >> 16 );
  184.         keystream[offset + 3U] = (unsigned char)( working_state[i] >> 24 );
  185.     }
  186.  
  187.     mbedtls_platform_zeroize( working_state, sizeof( working_state ) );
  188. }
  189.  
  190. void mbedtls_chacha20_init( mbedtls_chacha20_context *ctx )
  191. {
  192.     CHACHA20_VALIDATE( ctx != NULL );
  193.  
  194.     mbedtls_platform_zeroize( ctx->state, sizeof( ctx->state ) );
  195.     mbedtls_platform_zeroize( ctx->keystream8, sizeof( ctx->keystream8 ) );
  196.  
  197.     /* Initially, there's no keystream bytes available */
  198.     ctx->keystream_bytes_used = CHACHA20_BLOCK_SIZE_BYTES;
  199. }
  200.  
  201. void mbedtls_chacha20_free( mbedtls_chacha20_context *ctx )
  202. {
  203.     if( ctx != NULL )
  204.     {
  205.         mbedtls_platform_zeroize( ctx, sizeof( mbedtls_chacha20_context ) );
  206.     }
  207. }
  208.  
  209. int mbedtls_chacha20_setkey( mbedtls_chacha20_context *ctx,
  210.                             const unsigned char key[32] )
  211. {
  212.     CHACHA20_VALIDATE_RET( ctx != NULL );
  213.     CHACHA20_VALIDATE_RET( key != NULL );
  214.  
  215.     /* ChaCha20 constants - the string "expand 32-byte k" */
  216.     ctx->state[0] = 0x61707865;
  217.     ctx->state[1] = 0x3320646e;
  218.     ctx->state[2] = 0x79622d32;
  219.     ctx->state[3] = 0x6b206574;
  220.  
  221.     /* Set key */
  222.     ctx->state[4]  = BYTES_TO_U32_LE( key, 0 );
  223.     ctx->state[5]  = BYTES_TO_U32_LE( key, 4 );
  224.     ctx->state[6]  = BYTES_TO_U32_LE( key, 8 );
  225.     ctx->state[7]  = BYTES_TO_U32_LE( key, 12 );
  226.     ctx->state[8]  = BYTES_TO_U32_LE( key, 16 );
  227.     ctx->state[9]  = BYTES_TO_U32_LE( key, 20 );
  228.     ctx->state[10] = BYTES_TO_U32_LE( key, 24 );
  229.     ctx->state[11] = BYTES_TO_U32_LE( key, 28 );
  230.  
  231.     return( 0 );
  232. }
  233.  
  234. int mbedtls_chacha20_starts( mbedtls_chacha20_context* ctx,
  235.                              const unsigned char nonce[12],
  236.                              uint32_t counter )
  237. {
  238.     CHACHA20_VALIDATE_RET( ctx != NULL );
  239.     CHACHA20_VALIDATE_RET( nonce != NULL );
  240.  
  241.     /* Counter */
  242.     ctx->state[12] = counter;
  243.  
  244.     /* Nonce */
  245.     ctx->state[13] = BYTES_TO_U32_LE( nonce, 0 );
  246.     ctx->state[14] = BYTES_TO_U32_LE( nonce, 4 );
  247.     ctx->state[15] = BYTES_TO_U32_LE( nonce, 8 );
  248.  
  249.     mbedtls_platform_zeroize( ctx->keystream8, sizeof( ctx->keystream8 ) );
  250.  
  251.     /* Initially, there's no keystream bytes available */
  252.     ctx->keystream_bytes_used = CHACHA20_BLOCK_SIZE_BYTES;
  253.  
  254.     return( 0 );
  255. }
  256.  
  257. int mbedtls_chacha20_update( mbedtls_chacha20_context *ctx,
  258.                               size_t size,
  259.                               const unsigned char *input,
  260.                               unsigned char *output )
  261. {
  262.     size_t offset = 0U;
  263.     size_t i;
  264.  
  265.     CHACHA20_VALIDATE_RET( ctx != NULL );
  266.     CHACHA20_VALIDATE_RET( size == 0 || input  != NULL );
  267.     CHACHA20_VALIDATE_RET( size == 0 || output != NULL );
  268.  
  269.     /* Use leftover keystream bytes, if available */
  270.     while( size > 0U && ctx->keystream_bytes_used < CHACHA20_BLOCK_SIZE_BYTES )
  271.     {
  272.         output[offset] = input[offset]
  273.                        ^ ctx->keystream8[ctx->keystream_bytes_used];
  274.  
  275.         ctx->keystream_bytes_used++;
  276.         offset++;
  277.         size--;
  278.     }
  279.  
  280.     /* Process full blocks */
  281.     while( size >= CHACHA20_BLOCK_SIZE_BYTES )
  282.     {
  283.         /* Generate new keystream block and increment counter */
  284.         chacha20_block( ctx->state, ctx->keystream8 );
  285.         ctx->state[CHACHA20_CTR_INDEX]++;
  286.  
  287.         for( i = 0U; i < 64U; i += 8U )
  288.         {
  289.             output[offset + i  ] = input[offset + i  ] ^ ctx->keystream8[i  ];
  290.             output[offset + i+1] = input[offset + i+1] ^ ctx->keystream8[i+1];
  291.             output[offset + i+2] = input[offset + i+2] ^ ctx->keystream8[i+2];
  292.             output[offset + i+3] = input[offset + i+3] ^ ctx->keystream8[i+3];
  293.             output[offset + i+4] = input[offset + i+4] ^ ctx->keystream8[i+4];
  294.             output[offset + i+5] = input[offset + i+5] ^ ctx->keystream8[i+5];
  295.             output[offset + i+6] = input[offset + i+6] ^ ctx->keystream8[i+6];
  296.             output[offset + i+7] = input[offset + i+7] ^ ctx->keystream8[i+7];
  297.         }
  298.  
  299.         offset += CHACHA20_BLOCK_SIZE_BYTES;
  300.         size   -= CHACHA20_BLOCK_SIZE_BYTES;
  301.     }
  302.  
  303.     /* Last (partial) block */
  304.     if( size > 0U )
  305.     {
  306.         /* Generate new keystream block and increment counter */
  307.         chacha20_block( ctx->state, ctx->keystream8 );
  308.         ctx->state[CHACHA20_CTR_INDEX]++;
  309.  
  310.         for( i = 0U; i < size; i++)
  311.         {
  312.             output[offset + i] = input[offset + i] ^ ctx->keystream8[i];
  313.         }
  314.  
  315.         ctx->keystream_bytes_used = size;
  316.  
  317.     }
  318.  
  319.     return( 0 );
  320. }
  321.  
  322. int mbedtls_chacha20_crypt( const unsigned char key[32],
  323.                             const unsigned char nonce[12],
  324.                             uint32_t counter,
  325.                             size_t data_len,
  326.                             const unsigned char* input,
  327.                             unsigned char* output )
  328. {
  329.     mbedtls_chacha20_context ctx;
  330.     int ret;
  331.  
  332.     CHACHA20_VALIDATE_RET( key != NULL );
  333.     CHACHA20_VALIDATE_RET( nonce != NULL );
  334.     CHACHA20_VALIDATE_RET( data_len == 0 || input  != NULL );
  335.     CHACHA20_VALIDATE_RET( data_len == 0 || output != NULL );
  336.  
  337.     mbedtls_chacha20_init( &ctx );
  338.  
  339.     ret = mbedtls_chacha20_setkey( &ctx, key );
  340.     if( ret != 0 )
  341.         goto cleanup;
  342.  
  343.     ret = mbedtls_chacha20_starts( &ctx, nonce, counter );
  344.     if( ret != 0 )
  345.         goto cleanup;
  346.  
  347.     ret = mbedtls_chacha20_update( &ctx, data_len, input, output );
  348.  
  349. cleanup:
  350.     mbedtls_chacha20_free( &ctx );
  351.     return( ret );
  352. }
  353.  
  354. #endif /* !MBEDTLS_CHACHA20_ALT */
  355.  
  356. #if defined(MBEDTLS_SELF_TEST)
  357.  
  358. static const unsigned char test_keys[2][32] =
  359. {
  360.     {
  361.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  362.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  363.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  364.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
  365.     },
  366.     {
  367.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  368.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  369.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  370.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01
  371.     }
  372. };
  373.  
  374. static const unsigned char test_nonces[2][12] =
  375. {
  376.     {
  377.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  378.         0x00, 0x00, 0x00, 0x00
  379.     },
  380.     {
  381.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  382.         0x00, 0x00, 0x00, 0x02
  383.     }
  384. };
  385.  
  386. static const uint32_t test_counters[2] =
  387. {
  388.     0U,
  389.     1U
  390. };
  391.  
  392. static const unsigned char test_input[2][375] =
  393. {
  394.     {
  395.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  396.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  397.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  398.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  399.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  400.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  401.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  402.         0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
  403.     },
  404.     {
  405.         0x41, 0x6e, 0x79, 0x20, 0x73, 0x75, 0x62, 0x6d,
  406.         0x69, 0x73, 0x73, 0x69, 0x6f, 0x6e, 0x20, 0x74,
  407.         0x6f, 0x20, 0x74, 0x68, 0x65, 0x20, 0x49, 0x45,
  408.         0x54, 0x46, 0x20, 0x69, 0x6e, 0x74, 0x65, 0x6e,
  409.         0x64, 0x65, 0x64, 0x20, 0x62, 0x79, 0x20, 0x74,
  410.         0x68, 0x65, 0x20, 0x43, 0x6f, 0x6e, 0x74, 0x72,
  411.         0x69, 0x62, 0x75, 0x74, 0x6f, 0x72, 0x20, 0x66,
  412.         0x6f, 0x72, 0x20, 0x70, 0x75, 0x62, 0x6c, 0x69,
  413.         0x63, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x20, 0x61,
  414.         0x73, 0x20, 0x61, 0x6c, 0x6c, 0x20, 0x6f, 0x72,
  415.         0x20, 0x70, 0x61, 0x72, 0x74, 0x20, 0x6f, 0x66,
  416.         0x20, 0x61, 0x6e, 0x20, 0x49, 0x45, 0x54, 0x46,
  417.         0x20, 0x49, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x65,
  418.         0x74, 0x2d, 0x44, 0x72, 0x61, 0x66, 0x74, 0x20,
  419.         0x6f, 0x72, 0x20, 0x52, 0x46, 0x43, 0x20, 0x61,
  420.         0x6e, 0x64, 0x20, 0x61, 0x6e, 0x79, 0x20, 0x73,
  421.         0x74, 0x61, 0x74, 0x65, 0x6d, 0x65, 0x6e, 0x74,
  422.         0x20, 0x6d, 0x61, 0x64, 0x65, 0x20, 0x77, 0x69,
  423.         0x74, 0x68, 0x69, 0x6e, 0x20, 0x74, 0x68, 0x65,
  424.         0x20, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x78, 0x74,
  425.         0x20, 0x6f, 0x66, 0x20, 0x61, 0x6e, 0x20, 0x49,
  426.         0x45, 0x54, 0x46, 0x20, 0x61, 0x63, 0x74, 0x69,
  427.         0x76, 0x69, 0x74, 0x79, 0x20, 0x69, 0x73, 0x20,
  428.         0x63, 0x6f, 0x6e, 0x73, 0x69, 0x64, 0x65, 0x72,
  429.         0x65, 0x64, 0x20, 0x61, 0x6e, 0x20, 0x22, 0x49,
  430.         0x45, 0x54, 0x46, 0x20, 0x43, 0x6f, 0x6e, 0x74,
  431.         0x72, 0x69, 0x62, 0x75, 0x74, 0x69, 0x6f, 0x6e,
  432.         0x22, 0x2e, 0x20, 0x53, 0x75, 0x63, 0x68, 0x20,
  433.         0x73, 0x74, 0x61, 0x74, 0x65, 0x6d, 0x65, 0x6e,
  434.         0x74, 0x73, 0x20, 0x69, 0x6e, 0x63, 0x6c, 0x75,
  435.         0x64, 0x65, 0x20, 0x6f, 0x72, 0x61, 0x6c, 0x20,
  436.         0x73, 0x74, 0x61, 0x74, 0x65, 0x6d, 0x65, 0x6e,
  437.         0x74, 0x73, 0x20, 0x69, 0x6e, 0x20, 0x49, 0x45,
  438.         0x54, 0x46, 0x20, 0x73, 0x65, 0x73, 0x73, 0x69,
  439.         0x6f, 0x6e, 0x73, 0x2c, 0x20, 0x61, 0x73, 0x20,
  440.         0x77, 0x65, 0x6c, 0x6c, 0x20, 0x61, 0x73, 0x20,
  441.         0x77, 0x72, 0x69, 0x74, 0x74, 0x65, 0x6e, 0x20,
  442.         0x61, 0x6e, 0x64, 0x20, 0x65, 0x6c, 0x65, 0x63,
  443.         0x74, 0x72, 0x6f, 0x6e, 0x69, 0x63, 0x20, 0x63,
  444.         0x6f, 0x6d, 0x6d, 0x75, 0x6e, 0x69, 0x63, 0x61,
  445.         0x74, 0x69, 0x6f, 0x6e, 0x73, 0x20, 0x6d, 0x61,
  446.         0x64, 0x65, 0x20, 0x61, 0x74, 0x20, 0x61, 0x6e,
  447.         0x79, 0x20, 0x74, 0x69, 0x6d, 0x65, 0x20, 0x6f,
  448.         0x72, 0x20, 0x70, 0x6c, 0x61, 0x63, 0x65, 0x2c,
  449.         0x20, 0x77, 0x68, 0x69, 0x63, 0x68, 0x20, 0x61,
  450.         0x72, 0x65, 0x20, 0x61, 0x64, 0x64, 0x72, 0x65,
  451.         0x73, 0x73, 0x65, 0x64, 0x20, 0x74, 0x6f
  452.     }
  453. };
  454.  
  455. static const unsigned char test_output[2][375] =
  456. {
  457.     {
  458.         0x76, 0xb8, 0xe0, 0xad, 0xa0, 0xf1, 0x3d, 0x90,
  459.         0x40, 0x5d, 0x6a, 0xe5, 0x53, 0x86, 0xbd, 0x28,
  460.         0xbd, 0xd2, 0x19, 0xb8, 0xa0, 0x8d, 0xed, 0x1a,
  461.         0xa8, 0x36, 0xef, 0xcc, 0x8b, 0x77, 0x0d, 0xc7,
  462.         0xda, 0x41, 0x59, 0x7c, 0x51, 0x57, 0x48, 0x8d,
  463.         0x77, 0x24, 0xe0, 0x3f, 0xb8, 0xd8, 0x4a, 0x37,
  464.         0x6a, 0x43, 0xb8, 0xf4, 0x15, 0x18, 0xa1, 0x1c,
  465.         0xc3, 0x87, 0xb6, 0x69, 0xb2, 0xee, 0x65, 0x86
  466.     },
  467.     {
  468.         0xa3, 0xfb, 0xf0, 0x7d, 0xf3, 0xfa, 0x2f, 0xde,
  469.         0x4f, 0x37, 0x6c, 0xa2, 0x3e, 0x82, 0x73, 0x70,
  470.         0x41, 0x60, 0x5d, 0x9f, 0x4f, 0x4f, 0x57, 0xbd,
  471.         0x8c, 0xff, 0x2c, 0x1d, 0x4b, 0x79, 0x55, 0xec,
  472.         0x2a, 0x97, 0x94, 0x8b, 0xd3, 0x72, 0x29, 0x15,
  473.         0xc8, 0xf3, 0xd3, 0x37, 0xf7, 0xd3, 0x70, 0x05,
  474.         0x0e, 0x9e, 0x96, 0xd6, 0x47, 0xb7, 0xc3, 0x9f,
  475.         0x56, 0xe0, 0x31, 0xca, 0x5e, 0xb6, 0x25, 0x0d,
  476.         0x40, 0x42, 0xe0, 0x27, 0x85, 0xec, 0xec, 0xfa,
  477.         0x4b, 0x4b, 0xb5, 0xe8, 0xea, 0xd0, 0x44, 0x0e,
  478.         0x20, 0xb6, 0xe8, 0xdb, 0x09, 0xd8, 0x81, 0xa7,
  479.         0xc6, 0x13, 0x2f, 0x42, 0x0e, 0x52, 0x79, 0x50,
  480.         0x42, 0xbd, 0xfa, 0x77, 0x73, 0xd8, 0xa9, 0x05,
  481.         0x14, 0x47, 0xb3, 0x29, 0x1c, 0xe1, 0x41, 0x1c,
  482.         0x68, 0x04, 0x65, 0x55, 0x2a, 0xa6, 0xc4, 0x05,
  483.         0xb7, 0x76, 0x4d, 0x5e, 0x87, 0xbe, 0xa8, 0x5a,
  484.         0xd0, 0x0f, 0x84, 0x49, 0xed, 0x8f, 0x72, 0xd0,
  485.         0xd6, 0x62, 0xab, 0x05, 0x26, 0x91, 0xca, 0x66,
  486.         0x42, 0x4b, 0xc8, 0x6d, 0x2d, 0xf8, 0x0e, 0xa4,
  487.         0x1f, 0x43, 0xab, 0xf9, 0x37, 0xd3, 0x25, 0x9d,
  488.         0xc4, 0xb2, 0xd0, 0xdf, 0xb4, 0x8a, 0x6c, 0x91,
  489.         0x39, 0xdd, 0xd7, 0xf7, 0x69, 0x66, 0xe9, 0x28,
  490.         0xe6, 0x35, 0x55, 0x3b, 0xa7, 0x6c, 0x5c, 0x87,
  491.         0x9d, 0x7b, 0x35, 0xd4, 0x9e, 0xb2, 0xe6, 0x2b,
  492.         0x08, 0x71, 0xcd, 0xac, 0x63, 0x89, 0x39, 0xe2,
  493.         0x5e, 0x8a, 0x1e, 0x0e, 0xf9, 0xd5, 0x28, 0x0f,
  494.         0xa8, 0xca, 0x32, 0x8b, 0x35, 0x1c, 0x3c, 0x76,
  495.         0x59, 0x89, 0xcb, 0xcf, 0x3d, 0xaa, 0x8b, 0x6c,
  496.         0xcc, 0x3a, 0xaf, 0x9f, 0x39, 0x79, 0xc9, 0x2b,
  497.         0x37, 0x20, 0xfc, 0x88, 0xdc, 0x95, 0xed, 0x84,
  498.         0xa1, 0xbe, 0x05, 0x9c, 0x64, 0x99, 0xb9, 0xfd,
  499.         0xa2, 0x36, 0xe7, 0xe8, 0x18, 0xb0, 0x4b, 0x0b,
  500.         0xc3, 0x9c, 0x1e, 0x87, 0x6b, 0x19, 0x3b, 0xfe,
  501.         0x55, 0x69, 0x75, 0x3f, 0x88, 0x12, 0x8c, 0xc0,
  502.         0x8a, 0xaa, 0x9b, 0x63, 0xd1, 0xa1, 0x6f, 0x80,
  503.         0xef, 0x25, 0x54, 0xd7, 0x18, 0x9c, 0x41, 0x1f,
  504.         0x58, 0x69, 0xca, 0x52, 0xc5, 0xb8, 0x3f, 0xa3,
  505.         0x6f, 0xf2, 0x16, 0xb9, 0xc1, 0xd3, 0x00, 0x62,
  506.         0xbe, 0xbc, 0xfd, 0x2d, 0xc5, 0xbc, 0xe0, 0x91,
  507.         0x19, 0x34, 0xfd, 0xa7, 0x9a, 0x86, 0xf6, 0xe6,
  508.         0x98, 0xce, 0xd7, 0x59, 0xc3, 0xff, 0x9b, 0x64,
  509.         0x77, 0x33, 0x8f, 0x3d, 0xa4, 0xf9, 0xcd, 0x85,
  510.         0x14, 0xea, 0x99, 0x82, 0xcc, 0xaf, 0xb3, 0x41,
  511.         0xb2, 0x38, 0x4d, 0xd9, 0x02, 0xf3, 0xd1, 0xab,
  512.         0x7a, 0xc6, 0x1d, 0xd2, 0x9c, 0x6f, 0x21, 0xba,
  513.         0x5b, 0x86, 0x2f, 0x37, 0x30, 0xe3, 0x7c, 0xfd,
  514.         0xc4, 0xfd, 0x80, 0x6c, 0x22, 0xf2, 0x21
  515.     }
  516. };
  517.  
  518. static const size_t test_lengths[2] =
  519. {
  520.     64U,
  521.     375U
  522. };
  523.  
  524. #define ASSERT( cond, args )            \
  525.     do                                  \
  526.     {                                   \
  527.         if( ! ( cond ) )                \
  528.         {                               \
  529.             if( verbose != 0 )          \
  530.                 mbedtls_printf args;    \
  531.                                         \
  532.             return( -1 );               \
  533.         }                               \
  534.     }                                   \
  535.     while( 0 )
  536.  
  537. int mbedtls_chacha20_self_test( int verbose )
  538. {
  539.     unsigned char output[381];
  540.     unsigned i;
  541.     int ret;
  542.  
  543.     for( i = 0U; i < 2U; i++ )
  544.     {
  545.         if( verbose != 0 )
  546.             mbedtls_printf( "  ChaCha20 test %u ", i );
  547.  
  548.         ret = mbedtls_chacha20_crypt( test_keys[i],
  549.                                       test_nonces[i],
  550.                                       test_counters[i],
  551.                                       test_lengths[i],
  552.                                       test_input[i],
  553.                                       output );
  554.  
  555.         ASSERT( 0 == ret, ( "error code: %i\n", ret ) );
  556.  
  557.         ASSERT( 0 == memcmp( output, test_output[i], test_lengths[i] ),
  558.                 ( "failed (output)\n" ) );
  559.  
  560.         if( verbose != 0 )
  561.             mbedtls_printf( "passed\n" );
  562.     }
  563.  
  564.     if( verbose != 0 )
  565.         mbedtls_printf( "\n" );
  566.  
  567.     return( 0 );
  568. }
  569.  
  570. #endif /* MBEDTLS_SELF_TEST */
  571.  
  572. #endif /* !MBEDTLS_CHACHA20_C */
  573.