third_party/mbedtls/repo/tests/suites/test_suite_poly1305.function

   1 /* BEGIN_HEADER */
   2 #include "mbedtls/poly1305.h"
   3 #include <stddef.h>
   4 /* END_HEADER */
   5
   6 /* BEGIN_DEPENDENCIES
   7  * depends_on:MBEDTLS_POLY1305_C
   8  * END_DEPENDENCIES
   9  */
  10
  11 /* BEGIN_CASE */
  12 void mbedtls_poly1305( char *hex_key_string, char *hex_mac_string, char *hex_src_string  )
  13 {
  14     unsigned char src_str[375]; /* max size of binary input */
  15     unsigned char key[32]; /* size set by the standard */
  16     unsigned char mac[16]; /* size set by the standard */
  17     unsigned char mac_str[33]; /* hex expansion of the above */
  18     size_t src_len;
  19     mbedtls_poly1305_context ctx;
  20
  21     memset( src_str, 0x00, sizeof( src_str ) );
  22     memset( mac_str, 0x00, sizeof( mac_str ) );
  23     memset( key,     0x00, sizeof( key ) );
  24     memset( mac,     0x00, sizeof( mac ) );
  25
  26     src_len = unhexify( src_str, hex_src_string );
  27     unhexify( key, hex_key_string );
  28
  29     /*
  30      * Test the integrated API
  31      */
  32     TEST_ASSERT( mbedtls_poly1305_mac( key, src_str, src_len, mac ) == 0 );
  33
  34     hexify( mac_str, mac, 16 );
  35     TEST_ASSERT( strcmp( (char *) mac_str, hex_mac_string ) == 0 );
  36
  37     /*
  38      * Test the streaming API
  39      */
  40     mbedtls_poly1305_init( &ctx );
  41
  42     TEST_ASSERT( mbedtls_poly1305_starts( &ctx, key ) == 0 );
  43
  44     TEST_ASSERT( mbedtls_poly1305_update( &ctx, src_str, src_len ) == 0 );
  45
  46     TEST_ASSERT( mbedtls_poly1305_finish( &ctx, mac ) == 0 );
  47
  48     hexify( mac_str, mac, 16 );
  49     TEST_ASSERT( strcmp( (char *) mac_str, hex_mac_string ) == 0 );
  50
  51     /*
  52      * Test the streaming API again, piecewise
  53      */
  54
  55     /* Don't free/init the context, in order to test that starts() does the
  56      * right thing. */
  57     if( src_len >= 1 )
  58     {
  59         TEST_ASSERT( mbedtls_poly1305_starts( &ctx, key ) == 0 );
  60
  61         TEST_ASSERT( mbedtls_poly1305_update( &ctx, src_str, 1 ) == 0 );
  62         TEST_ASSERT( mbedtls_poly1305_update( &ctx, src_str + 1, src_len - 1 ) == 0 );
  63
  64         TEST_ASSERT( mbedtls_poly1305_finish( &ctx, mac ) == 0 );
  65
  66         hexify( mac_str, mac, 16 );
  67         TEST_ASSERT( strcmp( (char *) mac_str, hex_mac_string ) == 0 );
  68     }
  69
  70     /*
  71      * Again with more pieces
  72      */
  73     if( src_len >= 2 )
  74     {
  75         TEST_ASSERT( mbedtls_poly1305_starts( &ctx, key ) == 0 );
  76
  77         TEST_ASSERT( mbedtls_poly1305_update( &ctx, src_str, 1 ) == 0 );
  78         TEST_ASSERT( mbedtls_poly1305_update( &ctx, src_str + 1, 1 ) == 0 );
  79         TEST_ASSERT( mbedtls_poly1305_update( &ctx, src_str + 2, src_len - 2 ) == 0 );
  80
  81         TEST_ASSERT( mbedtls_poly1305_finish( &ctx, mac ) == 0 );
  82
  83         hexify( mac_str, mac, 16 );
  84         TEST_ASSERT( strcmp( (char *) mac_str, hex_mac_string ) == 0 );
  85     }
  86
  87     mbedtls_poly1305_free( &ctx );
  88 }
  89 /* END_CASE */
  90
  91 /* BEGIN_CASE depends_on:MBEDTLS_CHECK_PARAMS:!MBEDTLS_PARAM_FAILED_ALT */
  92 void poly1305_bad_params()
  93 {
  94     unsigned char src[1];
  95     unsigned char key[32];
  96     unsigned char mac[16];
  97     size_t src_len = sizeof( src );
  98     mbedtls_poly1305_context ctx;
  99
 100     TEST_INVALID_PARAM( mbedtls_poly1305_init( NULL ) );
 101     TEST_VALID_PARAM( mbedtls_poly1305_free( NULL ) );
 102
 103     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
 104                  mbedtls_poly1305_starts( NULL, key ) );
 105     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
 106                  mbedtls_poly1305_starts( &ctx, NULL ) );
 107
 108     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
 109                  mbedtls_poly1305_update( NULL, src, 0 ) );
 110     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
 111                  mbedtls_poly1305_update( &ctx, NULL, src_len ) );
 112
 113     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
 114                  mbedtls_poly1305_finish( NULL, mac ) );
 115     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
 116                  mbedtls_poly1305_finish( &ctx, NULL ) );
 117
 118     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
 119                  mbedtls_poly1305_mac( NULL, src, 0, mac ) );
 120     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
 121                  mbedtls_poly1305_mac( key, NULL, src_len, mac ) );
 122     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
 123                  mbedtls_poly1305_mac( key, src, 0, NULL ) );
 124
 125 exit:
 126     return;
 127 }
 128 /* END_CASE */
 129
 130 /* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */
 131 void poly1305_selftest()
 132 {
 133     TEST_ASSERT( mbedtls_poly1305_self_test( 1 ) == 0 );
 134 }
 135 /* END_CASE */