extlibs/mbedtls/mbedtls/tests/suites/test_suite_ecdsa.function

   1 /* BEGIN_HEADER */
   2 #include "mbedtls/ecdsa.h"
   3 /* END_HEADER */
   4
   5 /* BEGIN_DEPENDENCIES
   6  * depends_on:MBEDTLS_ECDSA_C
   7  * END_DEPENDENCIES
   8  */
   9
  10 /* BEGIN_CASE */
  11 void ecdsa_prim_random( int id )
  12 {
  13     mbedtls_ecp_group grp;
  14     mbedtls_ecp_point Q;
  15     mbedtls_mpi d, r, s;
  16     rnd_pseudo_info rnd_info;
  17     unsigned char buf[66];
  18
  19     mbedtls_ecp_group_init( &grp );
  20     mbedtls_ecp_point_init( &Q );
  21     mbedtls_mpi_init( &d ); mbedtls_mpi_init( &r ); mbedtls_mpi_init( &s );
  22     memset( &rnd_info, 0x00, sizeof( rnd_pseudo_info ) );
  23     memset( buf, 0, sizeof( buf ) );
  24
  25     /* prepare material for signature */
  26     TEST_ASSERT( rnd_pseudo_rand( &rnd_info, buf, sizeof( buf ) ) == 0 );
  27     TEST_ASSERT( mbedtls_ecp_group_load( &grp, id ) == 0 );
  28     TEST_ASSERT( mbedtls_ecp_gen_keypair( &grp, &d, &Q, &rnd_pseudo_rand, &rnd_info )
  29                  == 0 );
  30
  31     TEST_ASSERT( mbedtls_ecdsa_sign( &grp, &r, &s, &d, buf, sizeof( buf ),
  32                              &rnd_pseudo_rand, &rnd_info ) == 0 );
  33     TEST_ASSERT( mbedtls_ecdsa_verify( &grp, buf, sizeof( buf ), &Q, &r, &s ) == 0 );
  34
  35 exit:
  36     mbedtls_ecp_group_free( &grp );
  37     mbedtls_ecp_point_free( &Q );
  38     mbedtls_mpi_free( &d ); mbedtls_mpi_free( &r ); mbedtls_mpi_free( &s );
  39 }
  40 /* END_CASE */
  41
  42 /* BEGIN_CASE */
  43 void ecdsa_prim_test_vectors( int id, char *d_str, char *xQ_str, char *yQ_str,
  44                               char *k_str, char *hash_str, char *r_str,
  45                               char *s_str )
  46 {
  47     mbedtls_ecp_group grp;
  48     mbedtls_ecp_point Q;
  49     mbedtls_mpi d, r, s, r_check, s_check;
  50     unsigned char hash[66], rnd_buf[66];
  51     size_t hlen;
  52     rnd_buf_info rnd_info;
  53
  54     mbedtls_ecp_group_init( &grp );
  55     mbedtls_ecp_point_init( &Q );
  56     mbedtls_mpi_init( &d ); mbedtls_mpi_init( &r ); mbedtls_mpi_init( &s );
  57     mbedtls_mpi_init( &r_check ); mbedtls_mpi_init( &s_check );
  58     memset( hash, 0, sizeof( hash ) );
  59     memset( rnd_buf, 0, sizeof( rnd_buf ) );
  60
  61     TEST_ASSERT( mbedtls_ecp_group_load( &grp, id ) == 0 );
  62     TEST_ASSERT( mbedtls_ecp_point_read_string( &Q, 16, xQ_str, yQ_str ) == 0 );
  63     TEST_ASSERT( mbedtls_mpi_read_string( &d, 16, d_str ) == 0 );
  64     TEST_ASSERT( mbedtls_mpi_read_string( &r_check, 16, r_str ) == 0 );
  65     TEST_ASSERT( mbedtls_mpi_read_string( &s_check, 16, s_str ) == 0 );
  66     hlen = unhexify(hash, hash_str);
  67     rnd_info.buf = rnd_buf;
  68     rnd_info.length = unhexify( rnd_buf, k_str );
  69
  70     /* Fix rnd_buf by shifting it left if necessary */
  71     if( grp.nbits % 8 != 0 )
  72     {
  73         unsigned char shift = 8 - ( grp.nbits % 8 );
  74         size_t i;
  75
  76         for( i = 0; i < rnd_info.length - 1; i++ )
  77             rnd_buf[i] = rnd_buf[i] << shift | rnd_buf[i+1] >> ( 8 - shift );
  78
  79         rnd_buf[rnd_info.length-1] <<= shift;
  80     }
  81
  82     TEST_ASSERT( mbedtls_ecdsa_sign( &grp, &r, &s, &d, hash, hlen,
  83                  rnd_buffer_rand, &rnd_info ) == 0 );
  84
  85     TEST_ASSERT( mbedtls_mpi_cmp_mpi( &r, &r_check ) == 0 );
  86     TEST_ASSERT( mbedtls_mpi_cmp_mpi( &s, &s_check ) == 0 );
  87
  88     TEST_ASSERT( mbedtls_ecdsa_verify( &grp, hash, hlen, &Q, &r_check, &s_check ) == 0 );
  89
  90 exit:
  91     mbedtls_ecp_group_free( &grp );
  92     mbedtls_ecp_point_free( &Q );
  93     mbedtls_mpi_free( &d ); mbedtls_mpi_free( &r ); mbedtls_mpi_free( &s );
  94     mbedtls_mpi_free( &r_check ); mbedtls_mpi_free( &s_check );
  95 }
  96 /* END_CASE */
  97
  98 /* BEGIN_CASE depends_on:MBEDTLS_ECDSA_DETERMINISTIC */
  99 void ecdsa_det_test_vectors( int id, char *d_str, int md_alg,
 100                              char *msg, char *r_str, char *s_str )
 101 {
 102     mbedtls_ecp_group grp;
 103     mbedtls_mpi d, r, s, r_check, s_check;
 104     unsigned char hash[MBEDTLS_MD_MAX_SIZE];
 105     size_t hlen;
 106     const mbedtls_md_info_t *md_info;
 107
 108     mbedtls_ecp_group_init( &grp );
 109     mbedtls_mpi_init( &d ); mbedtls_mpi_init( &r ); mbedtls_mpi_init( &s );
 110     mbedtls_mpi_init( &r_check ); mbedtls_mpi_init( &s_check );
 111     memset( hash, 0, sizeof( hash ) );
 112
 113     TEST_ASSERT( mbedtls_ecp_group_load( &grp, id ) == 0 );
 114     TEST_ASSERT( mbedtls_mpi_read_string( &d, 16, d_str ) == 0 );
 115     TEST_ASSERT( mbedtls_mpi_read_string( &r_check, 16, r_str ) == 0 );
 116     TEST_ASSERT( mbedtls_mpi_read_string( &s_check, 16, s_str ) == 0 );
 117
 118     md_info = mbedtls_md_info_from_type( md_alg );
 119     TEST_ASSERT( md_info != NULL );
 120     hlen = mbedtls_md_get_size( md_info );
 121     mbedtls_md( md_info, (const unsigned char *) msg, strlen( msg ), hash );
 122
 123     TEST_ASSERT( mbedtls_ecdsa_sign_det( &grp, &r, &s, &d, hash, hlen, md_alg ) == 0 );
 124
 125     TEST_ASSERT( mbedtls_mpi_cmp_mpi( &r, &r_check ) == 0 );
 126     TEST_ASSERT( mbedtls_mpi_cmp_mpi( &s, &s_check ) == 0 );
 127
 128 exit:
 129     mbedtls_ecp_group_free( &grp );
 130     mbedtls_mpi_free( &d ); mbedtls_mpi_free( &r ); mbedtls_mpi_free( &s );
 131     mbedtls_mpi_free( &r_check ); mbedtls_mpi_free( &s_check );
 132 }
 133 /* END_CASE */
 134
 135 /* BEGIN_CASE depends_on:MBEDTLS_SHA256_C */
 136 void ecdsa_write_read_random( int id )
 137 {
 138     mbedtls_ecdsa_context ctx;
 139     rnd_pseudo_info rnd_info;
 140     unsigned char hash[32];
 141     unsigned char sig[200];
 142     size_t sig_len, i;
 143
 144     mbedtls_ecdsa_init( &ctx );
 145     memset( &rnd_info, 0x00, sizeof( rnd_pseudo_info ) );
 146     memset( hash, 0, sizeof( hash ) );
 147     memset( sig, 0x2a, sizeof( sig ) );
 148
 149     /* prepare material for signature */
 150     TEST_ASSERT( rnd_pseudo_rand( &rnd_info, hash, sizeof( hash ) ) == 0 );
 151
 152     /* generate signing key */
 153     TEST_ASSERT( mbedtls_ecdsa_genkey( &ctx, id, &rnd_pseudo_rand, &rnd_info ) == 0 );
 154
 155     /* generate and write signature, then read and verify it */
 156     TEST_ASSERT( mbedtls_ecdsa_write_signature( &ctx, MBEDTLS_MD_SHA256,
 157                  hash, sizeof( hash ),
 158                  sig, &sig_len, &rnd_pseudo_rand, &rnd_info ) == 0 );
 159     TEST_ASSERT( mbedtls_ecdsa_read_signature( &ctx, hash, sizeof( hash ),
 160                  sig, sig_len ) == 0 );
 161
 162     /* check we didn't write past the announced length */
 163     for( i = sig_len; i < sizeof( sig ); i++ )
 164         TEST_ASSERT( sig[i] == 0x2a );
 165
 166     /* try verification with invalid length */
 167     TEST_ASSERT( mbedtls_ecdsa_read_signature( &ctx, hash, sizeof( hash ),
 168                  sig, sig_len - 1 ) != 0 );
 169     TEST_ASSERT( mbedtls_ecdsa_read_signature( &ctx, hash, sizeof( hash ),
 170                  sig, sig_len + 1 ) != 0 );
 171
 172     /* try invalid sequence tag */
 173     sig[0]++;
 174     TEST_ASSERT( mbedtls_ecdsa_read_signature( &ctx, hash, sizeof( hash ),
 175                  sig, sig_len ) != 0 );
 176     sig[0]--;
 177
 178     /* try modifying r */
 179     sig[10]++;
 180     TEST_ASSERT( mbedtls_ecdsa_read_signature( &ctx, hash, sizeof( hash ),
 181                  sig, sig_len ) != 0 );
 182     sig[10]--;
 183
 184     /* try modifying s */
 185     sig[sig_len - 1]++;
 186     TEST_ASSERT( mbedtls_ecdsa_read_signature( &ctx, hash, sizeof( hash ),
 187                  sig, sig_len ) != 0 );
 188     sig[sig_len - 1]--;
 189
 190 exit:
 191     mbedtls_ecdsa_free( &ctx );
 192 }
 193 /* END_CASE */
 194