quickjs-tart

quickjs-based runtime for wallet-core logic
Log | Files | Refs | README | LICENSE

test_suite_ecdsa.function (20614B)


      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_zero(int id)
     12 {
     13     mbedtls_ecp_group grp;
     14     mbedtls_ecp_point Q;
     15     mbedtls_mpi d, r, s;
     16     mbedtls_test_rnd_pseudo_info rnd_info;
     17     unsigned char buf[MBEDTLS_MD_MAX_SIZE];
     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(mbedtls_test_rnd_pseudo_info));
     23     memset(buf, 0, sizeof(buf));
     24 
     25     TEST_ASSERT(mbedtls_ecp_group_load(&grp, id) == 0);
     26     TEST_ASSERT(mbedtls_ecp_gen_keypair(&grp, &d, &Q,
     27                                         &mbedtls_test_rnd_pseudo_rand,
     28                                         &rnd_info) == 0);
     29 
     30     TEST_ASSERT(mbedtls_ecdsa_sign(&grp, &r, &s, &d, buf, sizeof(buf),
     31                                    &mbedtls_test_rnd_pseudo_rand,
     32                                    &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_random(int id)
     44 {
     45     mbedtls_ecp_group grp;
     46     mbedtls_ecp_point Q;
     47     mbedtls_mpi d, r, s;
     48     mbedtls_test_rnd_pseudo_info rnd_info;
     49     unsigned char buf[MBEDTLS_MD_MAX_SIZE];
     50 
     51     mbedtls_ecp_group_init(&grp);
     52     mbedtls_ecp_point_init(&Q);
     53     mbedtls_mpi_init(&d); mbedtls_mpi_init(&r); mbedtls_mpi_init(&s);
     54     memset(&rnd_info, 0x00, sizeof(mbedtls_test_rnd_pseudo_info));
     55     memset(buf, 0, sizeof(buf));
     56 
     57     /* prepare material for signature */
     58     TEST_ASSERT(mbedtls_test_rnd_pseudo_rand(&rnd_info,
     59                                              buf, sizeof(buf)) == 0);
     60     TEST_ASSERT(mbedtls_ecp_group_load(&grp, id) == 0);
     61     TEST_ASSERT(mbedtls_ecp_gen_keypair(&grp, &d, &Q,
     62                                         &mbedtls_test_rnd_pseudo_rand,
     63                                         &rnd_info) == 0);
     64 
     65     TEST_ASSERT(mbedtls_ecdsa_sign(&grp, &r, &s, &d, buf, sizeof(buf),
     66                                    &mbedtls_test_rnd_pseudo_rand,
     67                                    &rnd_info) == 0);
     68     TEST_ASSERT(mbedtls_ecdsa_verify(&grp, buf, sizeof(buf), &Q, &r, &s) == 0);
     69 
     70 exit:
     71     mbedtls_ecp_group_free(&grp);
     72     mbedtls_ecp_point_free(&Q);
     73     mbedtls_mpi_free(&d); mbedtls_mpi_free(&r); mbedtls_mpi_free(&s);
     74 }
     75 /* END_CASE */
     76 
     77 /* BEGIN_CASE */
     78 void ecdsa_prim_test_vectors(int id, char *d_str, char *xQ_str,
     79                              char *yQ_str, data_t *rnd_buf,
     80                              data_t *hash, char *r_str, char *s_str,
     81                              int result)
     82 {
     83     mbedtls_ecp_group grp;
     84     mbedtls_ecp_point Q;
     85     mbedtls_mpi d, r, s, r_check, s_check, zero;
     86     mbedtls_test_rnd_buf_info rnd_info;
     87 
     88     mbedtls_ecp_group_init(&grp);
     89     mbedtls_ecp_point_init(&Q);
     90     mbedtls_mpi_init(&d); mbedtls_mpi_init(&r); mbedtls_mpi_init(&s);
     91     mbedtls_mpi_init(&r_check); mbedtls_mpi_init(&s_check);
     92     mbedtls_mpi_init(&zero);
     93 
     94     TEST_ASSERT(mbedtls_ecp_group_load(&grp, id) == 0);
     95     TEST_ASSERT(mbedtls_ecp_point_read_string(&Q, 16, xQ_str, yQ_str) == 0);
     96     TEST_ASSERT(mbedtls_test_read_mpi(&d, d_str) == 0);
     97     TEST_ASSERT(mbedtls_test_read_mpi(&r_check, r_str) == 0);
     98     TEST_ASSERT(mbedtls_test_read_mpi(&s_check, s_str) == 0);
     99     rnd_info.fallback_f_rng = mbedtls_test_rnd_std_rand;
    100     rnd_info.fallback_p_rng = NULL;
    101     rnd_info.buf = rnd_buf->x;
    102     rnd_info.length = rnd_buf->len;
    103 
    104     /* Fix rnd_buf->x by shifting it left if necessary */
    105     if (grp.nbits % 8 != 0) {
    106         unsigned char shift = 8 - (grp.nbits % 8);
    107         size_t i;
    108 
    109         for (i = 0; i < rnd_info.length - 1; i++) {
    110             rnd_buf->x[i] = rnd_buf->x[i] << shift | rnd_buf->x[i+1] >> (8 - shift);
    111         }
    112 
    113         rnd_buf->x[rnd_info.length-1] <<= shift;
    114     }
    115 
    116     TEST_ASSERT(mbedtls_ecdsa_sign(&grp, &r, &s, &d, hash->x, hash->len,
    117                                    mbedtls_test_rnd_buffer_rand, &rnd_info) == result);
    118 
    119     if (result == 0) {
    120         /* Check we generated the expected values */
    121         TEST_EQUAL(mbedtls_mpi_cmp_mpi(&r, &r_check), 0);
    122         TEST_EQUAL(mbedtls_mpi_cmp_mpi(&s, &s_check), 0);
    123 
    124         /* Valid signature */
    125         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len,
    126                                         &Q, &r_check, &s_check), 0);
    127 
    128         /* Invalid signature: wrong public key (G instead of Q) */
    129         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len,
    130                                         &grp.G, &r_check, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    131 
    132         /* Invalid signatures: r or s or both one off */
    133         TEST_EQUAL(mbedtls_mpi_sub_int(&r, &r_check, 1), 0);
    134         TEST_EQUAL(mbedtls_mpi_add_int(&s, &s_check, 1), 0);
    135 
    136         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    137                                         &r, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    138         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    139                                         &r_check, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    140         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    141                                         &r, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    142 
    143         /* Invalid signatures: r, s or both (CVE-2022-21449) are zero */
    144         TEST_EQUAL(mbedtls_mpi_lset(&zero, 0), 0);
    145 
    146         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    147                                         &zero, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    148         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    149                                         &r_check, &zero), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    150         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    151                                         &zero, &zero), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    152 
    153         /* Invalid signatures: r, s or both are == N */
    154         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    155                                         &grp.N, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    156         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    157                                         &r_check, &grp.N), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    158         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    159                                         &grp.N, &grp.N), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    160 
    161         /* Invalid signatures: r, s or both are negative */
    162         TEST_EQUAL(mbedtls_mpi_sub_mpi(&r, &r_check, &grp.N), 0);
    163         TEST_EQUAL(mbedtls_mpi_sub_mpi(&s, &s_check, &grp.N), 0);
    164 
    165         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    166                                         &r, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    167         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    168                                         &r_check, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    169         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    170                                         &r, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    171 
    172         /* Invalid signatures: r or s or both are > N */
    173         TEST_EQUAL(mbedtls_mpi_add_mpi(&r, &r_check, &grp.N), 0);
    174         TEST_EQUAL(mbedtls_mpi_add_mpi(&s, &s_check, &grp.N), 0);
    175 
    176         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    177                                         &r, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    178         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    179                                         &r_check, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    180         TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
    181                                         &r, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
    182     }
    183 
    184 exit:
    185     mbedtls_ecp_group_free(&grp);
    186     mbedtls_ecp_point_free(&Q);
    187     mbedtls_mpi_free(&d); mbedtls_mpi_free(&r); mbedtls_mpi_free(&s);
    188     mbedtls_mpi_free(&r_check); mbedtls_mpi_free(&s_check);
    189     mbedtls_mpi_free(&zero);
    190 }
    191 /* END_CASE */
    192 
    193 /* BEGIN_CASE depends_on:MBEDTLS_ECDSA_DETERMINISTIC */
    194 void ecdsa_det_test_vectors(int id, char *d_str, int md_alg, data_t *hash,
    195                             char *r_str, char *s_str)
    196 {
    197     mbedtls_ecp_group grp;
    198     mbedtls_mpi d, r, s, r_check, s_check;
    199     mbedtls_ecp_group_init(&grp);
    200     mbedtls_mpi_init(&d); mbedtls_mpi_init(&r); mbedtls_mpi_init(&s);
    201     mbedtls_mpi_init(&r_check); mbedtls_mpi_init(&s_check);
    202 
    203     MD_PSA_INIT();
    204 
    205     TEST_ASSERT(mbedtls_ecp_group_load(&grp, id) == 0);
    206     TEST_ASSERT(mbedtls_test_read_mpi(&d, d_str) == 0);
    207     TEST_ASSERT(mbedtls_test_read_mpi(&r_check, r_str) == 0);
    208     TEST_ASSERT(mbedtls_test_read_mpi(&s_check, s_str) == 0);
    209 
    210     TEST_ASSERT(
    211         mbedtls_ecdsa_sign_det_ext(&grp, &r, &s, &d,
    212                                    hash->x, hash->len, md_alg,
    213                                    mbedtls_test_rnd_std_rand,
    214                                    NULL)
    215         == 0);
    216 
    217     TEST_ASSERT(mbedtls_mpi_cmp_mpi(&r, &r_check) == 0);
    218     TEST_ASSERT(mbedtls_mpi_cmp_mpi(&s, &s_check) == 0);
    219 
    220 exit:
    221     mbedtls_ecp_group_free(&grp);
    222     mbedtls_mpi_free(&d); mbedtls_mpi_free(&r); mbedtls_mpi_free(&s);
    223     mbedtls_mpi_free(&r_check); mbedtls_mpi_free(&s_check);
    224     MD_PSA_DONE();
    225 }
    226 /* END_CASE */
    227 
    228 /* BEGIN_CASE depends_on:MBEDTLS_MD_CAN_SHA256 */
    229 void ecdsa_write_read_zero(int id)
    230 {
    231     mbedtls_ecdsa_context ctx;
    232     mbedtls_test_rnd_pseudo_info rnd_info;
    233     unsigned char hash[32];
    234     unsigned char sig[200];
    235     size_t sig_len, i;
    236 
    237     mbedtls_ecdsa_init(&ctx);
    238     memset(&rnd_info, 0x00, sizeof(mbedtls_test_rnd_pseudo_info));
    239     memset(hash, 0, sizeof(hash));
    240     memset(sig, 0x2a, sizeof(sig));
    241 
    242     MD_PSA_INIT();
    243 
    244     /* generate signing key */
    245     TEST_ASSERT(mbedtls_ecdsa_genkey(&ctx, id,
    246                                      &mbedtls_test_rnd_pseudo_rand,
    247                                      &rnd_info) == 0);
    248 
    249     /* generate and write signature, then read and verify it */
    250     TEST_ASSERT(mbedtls_ecdsa_write_signature(&ctx, MBEDTLS_MD_SHA256,
    251                                               hash, sizeof(hash),
    252                                               sig, sizeof(sig), &sig_len,
    253                                               &mbedtls_test_rnd_pseudo_rand,
    254                                               &rnd_info) == 0);
    255     TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
    256                                              sig, sig_len) == 0);
    257 
    258     /* check we didn't write past the announced length */
    259     for (i = sig_len; i < sizeof(sig); i++) {
    260         TEST_ASSERT(sig[i] == 0x2a);
    261     }
    262 
    263     /* try verification with invalid length */
    264     TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
    265                                              sig, sig_len - 1) != 0);
    266     TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
    267                                              sig, sig_len + 1) != 0);
    268 
    269     /* try invalid sequence tag */
    270     sig[0]++;
    271     TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
    272                                              sig, sig_len) != 0);
    273     sig[0]--;
    274 
    275     /* try modifying r */
    276     sig[10]++;
    277     TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
    278                                              sig, sig_len) == MBEDTLS_ERR_ECP_VERIFY_FAILED);
    279     sig[10]--;
    280 
    281     /* try modifying s */
    282     sig[sig_len - 1]++;
    283     TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
    284                                              sig, sig_len) == MBEDTLS_ERR_ECP_VERIFY_FAILED);
    285     sig[sig_len - 1]--;
    286 
    287 exit:
    288     mbedtls_ecdsa_free(&ctx);
    289     MD_PSA_DONE();
    290 }
    291 /* END_CASE */
    292 
    293 /* BEGIN_CASE depends_on:MBEDTLS_MD_CAN_SHA256 */
    294 void ecdsa_write_read_random(int id)
    295 {
    296     mbedtls_ecdsa_context ctx;
    297     mbedtls_test_rnd_pseudo_info rnd_info;
    298     unsigned char hash[32];
    299     unsigned char sig[200];
    300     size_t sig_len, i;
    301 
    302     mbedtls_ecdsa_init(&ctx);
    303     memset(&rnd_info, 0x00, sizeof(mbedtls_test_rnd_pseudo_info));
    304     memset(hash, 0, sizeof(hash));
    305     memset(sig, 0x2a, sizeof(sig));
    306 
    307     MD_PSA_INIT();
    308 
    309     /* prepare material for signature */
    310     TEST_ASSERT(mbedtls_test_rnd_pseudo_rand(&rnd_info,
    311                                              hash, sizeof(hash)) == 0);
    312 
    313     /* generate signing key */
    314     TEST_ASSERT(mbedtls_ecdsa_genkey(&ctx, id,
    315                                      &mbedtls_test_rnd_pseudo_rand,
    316                                      &rnd_info) == 0);
    317 
    318     /* generate and write signature, then read and verify it */
    319     TEST_ASSERT(mbedtls_ecdsa_write_signature(&ctx, MBEDTLS_MD_SHA256,
    320                                               hash, sizeof(hash),
    321                                               sig, sizeof(sig), &sig_len,
    322                                               &mbedtls_test_rnd_pseudo_rand,
    323                                               &rnd_info) == 0);
    324     TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
    325                                              sig, sig_len) == 0);
    326 
    327     /* check we didn't write past the announced length */
    328     for (i = sig_len; i < sizeof(sig); i++) {
    329         TEST_ASSERT(sig[i] == 0x2a);
    330     }
    331 
    332     /* try verification with invalid length */
    333     TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
    334                                              sig, sig_len - 1) != 0);
    335     TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
    336                                              sig, sig_len + 1) != 0);
    337 
    338     /* try invalid sequence tag */
    339     sig[0]++;
    340     TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
    341                                              sig, sig_len) != 0);
    342     sig[0]--;
    343 
    344     /* try modifying r */
    345     sig[10]++;
    346     TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
    347                                              sig, sig_len) == MBEDTLS_ERR_ECP_VERIFY_FAILED);
    348     sig[10]--;
    349 
    350     /* try modifying s */
    351     sig[sig_len - 1]++;
    352     TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
    353                                              sig, sig_len) == MBEDTLS_ERR_ECP_VERIFY_FAILED);
    354     sig[sig_len - 1]--;
    355 
    356 exit:
    357     mbedtls_ecdsa_free(&ctx);
    358     MD_PSA_DONE();
    359 }
    360 /* END_CASE */
    361 
    362 /* BEGIN_CASE depends_on:MBEDTLS_ECP_RESTARTABLE */
    363 void ecdsa_read_restart(int id, data_t *pk, data_t *hash, data_t *sig,
    364                         int max_ops, int min_restart, int max_restart)
    365 {
    366     mbedtls_ecdsa_context ctx;
    367     mbedtls_ecdsa_restart_ctx rs_ctx;
    368     int ret, cnt_restart;
    369 
    370     mbedtls_ecdsa_init(&ctx);
    371     mbedtls_ecdsa_restart_init(&rs_ctx);
    372 
    373     TEST_ASSERT(mbedtls_ecp_group_load(&ctx.grp, id) == 0);
    374     TEST_ASSERT(mbedtls_ecp_point_read_binary(&ctx.grp, &ctx.Q,
    375                                               pk->x, pk->len) == 0);
    376 
    377     mbedtls_ecp_set_max_ops(max_ops);
    378 
    379     cnt_restart = 0;
    380     do {
    381         ret = mbedtls_ecdsa_read_signature_restartable(&ctx,
    382                                                        hash->x, hash->len, sig->x, sig->len,
    383                                                        &rs_ctx);
    384     } while (ret == MBEDTLS_ERR_ECP_IN_PROGRESS && ++cnt_restart);
    385 
    386     TEST_ASSERT(ret == 0);
    387     TEST_ASSERT(cnt_restart >= min_restart);
    388     TEST_ASSERT(cnt_restart <= max_restart);
    389 
    390     /* try modifying r */
    391 
    392     TEST_ASSERT(sig->len > 10);
    393     sig->x[10]++;
    394     do {
    395         ret = mbedtls_ecdsa_read_signature_restartable(&ctx,
    396                                                        hash->x, hash->len, sig->x, sig->len,
    397                                                        &rs_ctx);
    398     } while (ret == MBEDTLS_ERR_ECP_IN_PROGRESS);
    399     TEST_ASSERT(ret == MBEDTLS_ERR_ECP_VERIFY_FAILED);
    400     sig->x[10]--;
    401 
    402     /* try modifying s */
    403     sig->x[sig->len - 1]++;
    404     do {
    405         ret = mbedtls_ecdsa_read_signature_restartable(&ctx,
    406                                                        hash->x, hash->len, sig->x, sig->len,
    407                                                        &rs_ctx);
    408     } while (ret == MBEDTLS_ERR_ECP_IN_PROGRESS);
    409     TEST_ASSERT(ret == MBEDTLS_ERR_ECP_VERIFY_FAILED);
    410     sig->x[sig->len - 1]--;
    411 
    412     /* Do we leak memory when aborting an operation?
    413      * This test only makes sense when we actually restart */
    414     if (min_restart > 0) {
    415         ret = mbedtls_ecdsa_read_signature_restartable(&ctx,
    416                                                        hash->x, hash->len, sig->x, sig->len,
    417                                                        &rs_ctx);
    418         TEST_ASSERT(ret == MBEDTLS_ERR_ECP_IN_PROGRESS);
    419     }
    420 
    421 exit:
    422     mbedtls_ecdsa_free(&ctx);
    423     mbedtls_ecdsa_restart_free(&rs_ctx);
    424 }
    425 /* END_CASE */
    426 
    427 /* BEGIN_CASE depends_on:MBEDTLS_ECP_RESTARTABLE:MBEDTLS_ECDSA_DETERMINISTIC */
    428 void ecdsa_write_restart(int id, char *d_str, int md_alg,
    429                          data_t *hash, data_t *sig_check,
    430                          int max_ops, int min_restart, int max_restart)
    431 {
    432     int ret, cnt_restart;
    433     mbedtls_ecdsa_restart_ctx rs_ctx;
    434     mbedtls_ecdsa_context ctx;
    435     unsigned char sig[MBEDTLS_ECDSA_MAX_LEN];
    436     size_t slen;
    437 
    438     mbedtls_ecdsa_restart_init(&rs_ctx);
    439     mbedtls_ecdsa_init(&ctx);
    440     memset(sig, 0, sizeof(sig));
    441 
    442     MD_PSA_INIT();
    443 
    444     TEST_ASSERT(mbedtls_ecp_group_load(&ctx.grp, id) == 0);
    445     TEST_ASSERT(mbedtls_test_read_mpi(&ctx.d, d_str) == 0);
    446 
    447     mbedtls_ecp_set_max_ops(max_ops);
    448 
    449     slen = sizeof(sig);
    450     cnt_restart = 0;
    451     do {
    452         ret = mbedtls_ecdsa_write_signature_restartable(&ctx,
    453                                                         md_alg,
    454                                                         hash->x,
    455                                                         hash->len,
    456                                                         sig,
    457                                                         sizeof(sig),
    458                                                         &slen,
    459                                                         mbedtls_test_rnd_std_rand,
    460                                                         NULL,
    461                                                         &rs_ctx);
    462     } while (ret == MBEDTLS_ERR_ECP_IN_PROGRESS && ++cnt_restart);
    463 
    464     TEST_ASSERT(ret == 0);
    465     TEST_ASSERT(slen == sig_check->len);
    466     TEST_ASSERT(memcmp(sig, sig_check->x, slen) == 0);
    467 
    468     TEST_ASSERT(cnt_restart >= min_restart);
    469     TEST_ASSERT(cnt_restart <= max_restart);
    470 
    471     /* Do we leak memory when aborting an operation?
    472      * This test only makes sense when we actually restart */
    473     if (min_restart > 0) {
    474         ret = mbedtls_ecdsa_write_signature_restartable(&ctx,
    475                                                         md_alg,
    476                                                         hash->x,
    477                                                         hash->len,
    478                                                         sig,
    479                                                         sizeof(sig),
    480                                                         &slen,
    481                                                         mbedtls_test_rnd_std_rand,
    482                                                         NULL,
    483                                                         &rs_ctx);
    484         TEST_ASSERT(ret == MBEDTLS_ERR_ECP_IN_PROGRESS);
    485     }
    486 
    487 exit:
    488     mbedtls_ecdsa_restart_free(&rs_ctx);
    489     mbedtls_ecdsa_free(&ctx);
    490     MD_PSA_DONE();
    491 }
    492 /* END_CASE */
    493 
    494 /* BEGIN_CASE */
    495 void ecdsa_verify(int grp_id, char *x, char *y, char *r, char *s, data_t *content, int expected)
    496 {
    497     mbedtls_ecdsa_context ctx;
    498     mbedtls_mpi sig_r, sig_s;
    499 
    500     mbedtls_ecdsa_init(&ctx);
    501     mbedtls_mpi_init(&sig_r);
    502     mbedtls_mpi_init(&sig_s);
    503 
    504     /* Prepare ECP group context */
    505     TEST_EQUAL(mbedtls_ecp_group_load(&ctx.grp, grp_id), 0);
    506 
    507     /* Prepare public key */
    508     TEST_EQUAL(mbedtls_test_read_mpi(&ctx.Q.X, x), 0);
    509     TEST_EQUAL(mbedtls_test_read_mpi(&ctx.Q.Y, y), 0);
    510     TEST_EQUAL(mbedtls_mpi_lset(&ctx.Q.Z, 1), 0);
    511 
    512     /* Prepare signature R & S */
    513     TEST_EQUAL(mbedtls_test_read_mpi(&sig_r, r), 0);
    514     TEST_EQUAL(mbedtls_test_read_mpi(&sig_s, s), 0);
    515 
    516     /* Test whether public key has expected validity */
    517     TEST_EQUAL(mbedtls_ecp_check_pubkey(&ctx.grp, &ctx.Q),
    518                expected == MBEDTLS_ERR_ECP_INVALID_KEY ? MBEDTLS_ERR_ECP_INVALID_KEY : 0);
    519 
    520     /* Verification */
    521     int result = mbedtls_ecdsa_verify(&ctx.grp, content->x, content->len, &ctx.Q, &sig_r, &sig_s);
    522 
    523     TEST_EQUAL(result, expected);
    524 exit:
    525     mbedtls_ecdsa_free(&ctx);
    526     mbedtls_mpi_free(&sig_r);
    527     mbedtls_mpi_free(&sig_s);
    528 }
    529 /* END_CASE */