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

   1 /* BEGIN_HEADER */
   2 #include "mbedtls/rsa.h"
   3 #include "mbedtls/md.h"
   4 /* END_HEADER */
   5
   6 /* BEGIN_DEPENDENCIES
   7  * depends_on:MBEDTLS_PKCS1_V21:MBEDTLS_RSA_C:MBEDTLS_SHA1_C
   8  * END_DEPENDENCIES
   9  */
  10
  11 /* BEGIN_CASE */
  12 void pkcs1_rsaes_oaep_encrypt( int mod, int radix_N, char * input_N,
  13                                int radix_E, char * input_E, int hash,
  14                                data_t * message_str, data_t * rnd_buf,
  15                                data_t * result_hex_str, int result )
  16 {
  17     unsigned char output[1000];
  18     mbedtls_rsa_context ctx;
  19     rnd_buf_info info;
  20     mbedtls_mpi N, E;
  21
  22     info.buf = rnd_buf->x;
  23     info.length = rnd_buf->len;
  24
  25     mbedtls_mpi_init( &N ); mbedtls_mpi_init( &E );
  26     mbedtls_rsa_init( &ctx, MBEDTLS_RSA_PKCS_V21, hash );
  27     memset( output, 0x00, 1000 );
  28
  29     TEST_ASSERT( mbedtls_mpi_read_string( &N, radix_N, input_N ) == 0 );
  30     TEST_ASSERT( mbedtls_mpi_read_string( &E, radix_E, input_E ) == 0 );
  31     TEST_ASSERT( mbedtls_rsa_import( &ctx, &N, NULL, NULL, NULL, &E ) == 0 );
  32     TEST_ASSERT( mbedtls_rsa_get_len( &ctx ) == (size_t) ( ( mod + 7 ) / 8 ) );
  33     TEST_ASSERT( mbedtls_rsa_check_pubkey( &ctx ) == 0 );
  34
  35     if( message_str->len == 0 )
  36         message_str->x = NULL;
  37     TEST_ASSERT( mbedtls_rsa_pkcs1_encrypt( &ctx, &rnd_buffer_rand, &info, MBEDTLS_RSA_PUBLIC, message_str->len, message_str->x, output ) == result );
  38     if( result == 0 )
  39     {
  40         TEST_ASSERT( hexcmp( output, result_hex_str->x, ctx.len, result_hex_str->len ) == 0 );
  41     }
  42
  43 exit:
  44     mbedtls_mpi_free( &N ); mbedtls_mpi_free( &E );
  45     mbedtls_rsa_free( &ctx );
  46 }
  47 /* END_CASE */
  48
  49 /* BEGIN_CASE */
  50 void pkcs1_rsaes_oaep_decrypt( int mod, int radix_P, char * input_P,
  51                                int radix_Q, char * input_Q, int radix_N,
  52                                char * input_N, int radix_E, char * input_E,
  53                                int hash, data_t * result_hex_str,
  54                                char * seed, data_t * message_str,
  55                                int result )
  56 {
  57     unsigned char output[1000];
  58     mbedtls_rsa_context ctx;
  59     size_t output_len;
  60     rnd_pseudo_info rnd_info;
  61     mbedtls_mpi N, P, Q, E;
  62     ((void) seed);
  63
  64     mbedtls_mpi_init( &N ); mbedtls_mpi_init( &P );
  65     mbedtls_mpi_init( &Q ); mbedtls_mpi_init( &E );
  66
  67     mbedtls_rsa_init( &ctx, MBEDTLS_RSA_PKCS_V21, hash );
  68
  69     memset( output, 0x00, 1000 );
  70     memset( &rnd_info, 0, sizeof( rnd_pseudo_info ) );
  71
  72     TEST_ASSERT( mbedtls_mpi_read_string( &P, radix_P, input_P ) == 0 );
  73     TEST_ASSERT( mbedtls_mpi_read_string( &Q, radix_Q, input_Q ) == 0 );
  74     TEST_ASSERT( mbedtls_mpi_read_string( &N, radix_N, input_N ) == 0 );
  75     TEST_ASSERT( mbedtls_mpi_read_string( &E, radix_E, input_E ) == 0 );
  76
  77     TEST_ASSERT( mbedtls_rsa_import( &ctx, &N, &P, &Q, NULL, &E ) == 0 );
  78     TEST_ASSERT( mbedtls_rsa_get_len( &ctx ) == (size_t) ( ( mod + 7 ) / 8 ) );
  79     TEST_ASSERT( mbedtls_rsa_complete( &ctx ) == 0 );
  80     TEST_ASSERT( mbedtls_rsa_check_privkey( &ctx ) == 0 );
  81
  82     if( result_hex_str->len == 0 )
  83     {
  84         TEST_ASSERT( mbedtls_rsa_pkcs1_decrypt( &ctx, &rnd_pseudo_rand, &rnd_info, MBEDTLS_RSA_PRIVATE, &output_len, message_str->x, NULL, 0 ) == result );
  85     }
  86     else
  87     {
  88         TEST_ASSERT( mbedtls_rsa_pkcs1_decrypt( &ctx, &rnd_pseudo_rand, &rnd_info, MBEDTLS_RSA_PRIVATE, &output_len, message_str->x, output, 1000 ) == result );
  89         if( result == 0 )
  90         {
  91             TEST_ASSERT( hexcmp( output, result_hex_str->x, output_len, result_hex_str->len ) == 0 );
  92         }
  93     }
  94
  95 exit:
  96     mbedtls_mpi_free( &N ); mbedtls_mpi_free( &P );
  97     mbedtls_mpi_free( &Q ); mbedtls_mpi_free( &E );
  98     mbedtls_rsa_free( &ctx );
  99 }
 100 /* END_CASE */
 101
 102 /* BEGIN_CASE */
 103 void pkcs1_rsassa_pss_sign( int mod, int radix_P, char * input_P, int radix_Q,
 104                             char * input_Q, int radix_N, char * input_N,
 105                             int radix_E, char * input_E, int digest, int hash,
 106                             data_t * message_str, data_t * rnd_buf,
 107                             data_t * result_hex_str, int result )
 108 {
 109     unsigned char hash_result[1000];
 110     unsigned char output[1000];
 111     mbedtls_rsa_context ctx;
 112     rnd_buf_info info;
 113     mbedtls_mpi N, P, Q, E;
 114
 115     info.buf = rnd_buf->x;
 116     info.length = rnd_buf->len;
 117
 118     mbedtls_mpi_init( &N ); mbedtls_mpi_init( &P );
 119     mbedtls_mpi_init( &Q ); mbedtls_mpi_init( &E );
 120     mbedtls_rsa_init( &ctx, MBEDTLS_RSA_PKCS_V21, hash );
 121
 122     memset( hash_result, 0x00, 1000 );
 123     memset( output, 0x00, 1000 );
 124
 125     TEST_ASSERT( mbedtls_mpi_read_string( &P, radix_P, input_P ) == 0 );
 126     TEST_ASSERT( mbedtls_mpi_read_string( &Q, radix_Q, input_Q ) == 0 );
 127     TEST_ASSERT( mbedtls_mpi_read_string( &N, radix_N, input_N ) == 0 );
 128     TEST_ASSERT( mbedtls_mpi_read_string( &E, radix_E, input_E ) == 0 );
 129
 130     TEST_ASSERT( mbedtls_rsa_import( &ctx, &N, &P, &Q, NULL, &E ) == 0 );
 131     TEST_ASSERT( mbedtls_rsa_get_len( &ctx ) == (size_t) ( ( mod + 7 ) / 8 ) );
 132     TEST_ASSERT( mbedtls_rsa_complete( &ctx ) == 0 );
 133     TEST_ASSERT( mbedtls_rsa_check_privkey( &ctx ) == 0 );
 134
 135
 136     if( mbedtls_md_info_from_type( digest ) != NULL )
 137         TEST_ASSERT( mbedtls_md( mbedtls_md_info_from_type( digest ), message_str->x, message_str->len, hash_result ) == 0 );
 138
 139     TEST_ASSERT( mbedtls_rsa_pkcs1_sign( &ctx, &rnd_buffer_rand, &info, MBEDTLS_RSA_PRIVATE,
 140                                          digest, 0, hash_result, output ) == result );
 141     if( result == 0 )
 142     {
 143
 144         TEST_ASSERT( hexcmp( output, result_hex_str->x, ctx.len, result_hex_str->len ) == 0 );
 145     }
 146
 147 exit:
 148     mbedtls_mpi_free( &N ); mbedtls_mpi_free( &P );
 149     mbedtls_mpi_free( &Q ); mbedtls_mpi_free( &E );
 150     mbedtls_rsa_free( &ctx );
 151 }
 152 /* END_CASE */
 153
 154 /* BEGIN_CASE */
 155 void pkcs1_rsassa_pss_verify( int mod, int radix_N, char * input_N,
 156                               int radix_E, char * input_E, int digest,
 157                               int hash, data_t * message_str, char * salt,
 158                               data_t * result_str, int result )
 159 {
 160     unsigned char hash_result[1000];
 161     mbedtls_rsa_context ctx;
 162     mbedtls_mpi N, E;
 163     ((void) salt);
 164
 165     mbedtls_mpi_init( &N ); mbedtls_mpi_init( &E );
 166     mbedtls_rsa_init( &ctx, MBEDTLS_RSA_PKCS_V21, hash );
 167     memset( hash_result, 0x00, 1000 );
 168
 169     TEST_ASSERT( mbedtls_mpi_read_string( &N, radix_N, input_N ) == 0 );
 170     TEST_ASSERT( mbedtls_mpi_read_string( &E, radix_E, input_E ) == 0 );
 171
 172     TEST_ASSERT( mbedtls_rsa_import( &ctx, &N, NULL, NULL, NULL, &E ) == 0 );
 173     TEST_ASSERT( mbedtls_rsa_get_len( &ctx ) == (size_t) ( ( mod + 7 ) / 8 ) );
 174     TEST_ASSERT( mbedtls_rsa_check_pubkey( &ctx ) == 0 );
 175
 176
 177     if( mbedtls_md_info_from_type( digest ) != NULL )
 178         TEST_ASSERT( mbedtls_md( mbedtls_md_info_from_type( digest ), message_str->x, message_str->len, hash_result ) == 0 );
 179
 180     TEST_ASSERT( mbedtls_rsa_pkcs1_verify( &ctx, NULL, NULL, MBEDTLS_RSA_PUBLIC, digest, 0, hash_result, result_str->x ) == result );
 181
 182 exit:
 183     mbedtls_mpi_free( &N ); mbedtls_mpi_free( &E );
 184     mbedtls_rsa_free( &ctx );
 185 }
 186 /* END_CASE */
 187
 188 /* BEGIN_CASE */
 189 void pkcs1_rsassa_pss_verify_ext( int mod, int radix_N, char * input_N,
 190                                   int radix_E, char * input_E,
 191                                   int msg_digest_id, int ctx_hash,
 192                                   int mgf_hash, int salt_len,
 193                                   data_t * message_str,
 194                                   data_t * result_str, int result_simple,
 195                                   int result_full )
 196 {
 197     unsigned char hash_result[1000];
 198     mbedtls_rsa_context ctx;
 199     size_t hash_len;
 200     mbedtls_mpi N, E;
 201
 202     mbedtls_mpi_init( &N ); mbedtls_mpi_init( &E );
 203     mbedtls_rsa_init( &ctx, MBEDTLS_RSA_PKCS_V21, ctx_hash );
 204     memset( hash_result, 0x00, 1000 );
 205
 206     TEST_ASSERT( mbedtls_mpi_read_string( &N, radix_N, input_N ) == 0 );
 207     TEST_ASSERT( mbedtls_mpi_read_string( &E, radix_E, input_E ) == 0 );
 208
 209     TEST_ASSERT( mbedtls_rsa_import( &ctx, &N, NULL, NULL, NULL, &E ) == 0 );
 210     TEST_ASSERT( mbedtls_rsa_get_len( &ctx ) == (size_t) ( ( mod + 7 ) / 8 ) );
 211     TEST_ASSERT( mbedtls_rsa_check_pubkey( &ctx ) == 0 );
 212
 213
 214     if( msg_digest_id != MBEDTLS_MD_NONE )
 215     {
 216         TEST_ASSERT( mbedtls_md( mbedtls_md_info_from_type( msg_digest_id ),
 217                      message_str->x, message_str->len, hash_result ) == 0 );
 218         hash_len = 0;
 219     }
 220     else
 221     {
 222         memcpy( hash_result, message_str->x, message_str->len );
 223         hash_len = message_str->len;
 224     }
 225
 226     TEST_ASSERT( mbedtls_rsa_pkcs1_verify( &ctx, NULL, NULL, MBEDTLS_RSA_PUBLIC,
 227                                    msg_digest_id, hash_len, hash_result,
 228                                    result_str->x ) == result_simple );
 229
 230     TEST_ASSERT( mbedtls_rsa_rsassa_pss_verify_ext( &ctx, NULL, NULL, MBEDTLS_RSA_PUBLIC,
 231                                         msg_digest_id, hash_len, hash_result,
 232                                         mgf_hash, salt_len,
 233                                         result_str->x ) == result_full );
 234
 235 exit:
 236     mbedtls_mpi_free( &N ); mbedtls_mpi_free( &E );
 237     mbedtls_rsa_free( &ctx );
 238 }
 239 /* END_CASE */