mirror of
https://github.com/cuberite/polarssl.git
synced 2025-11-04 04:32:24 -05:00
532 lines
20 KiB
C
532 lines
20 KiB
C
/* BEGIN_HEADER */
|
|
#include "mbedtls/ecdsa.h"
|
|
#include "hash_info.h"
|
|
/* END_HEADER */
|
|
|
|
/* BEGIN_DEPENDENCIES
|
|
* depends_on:MBEDTLS_ECDSA_C
|
|
* END_DEPENDENCIES
|
|
*/
|
|
|
|
/* BEGIN_CASE */
|
|
void ecdsa_prim_zero(int id)
|
|
{
|
|
mbedtls_ecp_group grp;
|
|
mbedtls_ecp_point Q;
|
|
mbedtls_mpi d, r, s;
|
|
mbedtls_test_rnd_pseudo_info rnd_info;
|
|
unsigned char buf[MBEDTLS_HASH_MAX_SIZE];
|
|
|
|
mbedtls_ecp_group_init(&grp);
|
|
mbedtls_ecp_point_init(&Q);
|
|
mbedtls_mpi_init(&d); mbedtls_mpi_init(&r); mbedtls_mpi_init(&s);
|
|
memset(&rnd_info, 0x00, sizeof(mbedtls_test_rnd_pseudo_info));
|
|
memset(buf, 0, sizeof(buf));
|
|
|
|
TEST_ASSERT(mbedtls_ecp_group_load(&grp, id) == 0);
|
|
TEST_ASSERT(mbedtls_ecp_gen_keypair(&grp, &d, &Q,
|
|
&mbedtls_test_rnd_pseudo_rand,
|
|
&rnd_info) == 0);
|
|
|
|
TEST_ASSERT(mbedtls_ecdsa_sign(&grp, &r, &s, &d, buf, sizeof(buf),
|
|
&mbedtls_test_rnd_pseudo_rand,
|
|
&rnd_info) == 0);
|
|
TEST_ASSERT(mbedtls_ecdsa_verify(&grp, buf, sizeof(buf), &Q, &r, &s) == 0);
|
|
|
|
exit:
|
|
mbedtls_ecp_group_free(&grp);
|
|
mbedtls_ecp_point_free(&Q);
|
|
mbedtls_mpi_free(&d); mbedtls_mpi_free(&r); mbedtls_mpi_free(&s);
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecdsa_prim_random(int id)
|
|
{
|
|
mbedtls_ecp_group grp;
|
|
mbedtls_ecp_point Q;
|
|
mbedtls_mpi d, r, s;
|
|
mbedtls_test_rnd_pseudo_info rnd_info;
|
|
unsigned char buf[MBEDTLS_HASH_MAX_SIZE];
|
|
|
|
mbedtls_ecp_group_init(&grp);
|
|
mbedtls_ecp_point_init(&Q);
|
|
mbedtls_mpi_init(&d); mbedtls_mpi_init(&r); mbedtls_mpi_init(&s);
|
|
memset(&rnd_info, 0x00, sizeof(mbedtls_test_rnd_pseudo_info));
|
|
memset(buf, 0, sizeof(buf));
|
|
|
|
/* prepare material for signature */
|
|
TEST_ASSERT(mbedtls_test_rnd_pseudo_rand(&rnd_info,
|
|
buf, sizeof(buf)) == 0);
|
|
TEST_ASSERT(mbedtls_ecp_group_load(&grp, id) == 0);
|
|
TEST_ASSERT(mbedtls_ecp_gen_keypair(&grp, &d, &Q,
|
|
&mbedtls_test_rnd_pseudo_rand,
|
|
&rnd_info) == 0);
|
|
|
|
TEST_ASSERT(mbedtls_ecdsa_sign(&grp, &r, &s, &d, buf, sizeof(buf),
|
|
&mbedtls_test_rnd_pseudo_rand,
|
|
&rnd_info) == 0);
|
|
TEST_ASSERT(mbedtls_ecdsa_verify(&grp, buf, sizeof(buf), &Q, &r, &s) == 0);
|
|
|
|
exit:
|
|
mbedtls_ecp_group_free(&grp);
|
|
mbedtls_ecp_point_free(&Q);
|
|
mbedtls_mpi_free(&d); mbedtls_mpi_free(&r); mbedtls_mpi_free(&s);
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecdsa_prim_test_vectors(int id, char *d_str, char *xQ_str,
|
|
char *yQ_str, data_t *rnd_buf,
|
|
data_t *hash, char *r_str, char *s_str,
|
|
int result)
|
|
{
|
|
mbedtls_ecp_group grp;
|
|
mbedtls_ecp_point Q;
|
|
mbedtls_mpi d, r, s, r_check, s_check, zero;
|
|
mbedtls_test_rnd_buf_info rnd_info;
|
|
|
|
mbedtls_ecp_group_init(&grp);
|
|
mbedtls_ecp_point_init(&Q);
|
|
mbedtls_mpi_init(&d); mbedtls_mpi_init(&r); mbedtls_mpi_init(&s);
|
|
mbedtls_mpi_init(&r_check); mbedtls_mpi_init(&s_check);
|
|
mbedtls_mpi_init(&zero);
|
|
|
|
TEST_ASSERT(mbedtls_ecp_group_load(&grp, id) == 0);
|
|
TEST_ASSERT(mbedtls_ecp_point_read_string(&Q, 16, xQ_str, yQ_str) == 0);
|
|
TEST_ASSERT(mbedtls_test_read_mpi(&d, d_str) == 0);
|
|
TEST_ASSERT(mbedtls_test_read_mpi(&r_check, r_str) == 0);
|
|
TEST_ASSERT(mbedtls_test_read_mpi(&s_check, s_str) == 0);
|
|
rnd_info.fallback_f_rng = mbedtls_test_rnd_std_rand;
|
|
rnd_info.fallback_p_rng = NULL;
|
|
rnd_info.buf = rnd_buf->x;
|
|
rnd_info.length = rnd_buf->len;
|
|
|
|
/* Fix rnd_buf->x by shifting it left if necessary */
|
|
if (grp.nbits % 8 != 0) {
|
|
unsigned char shift = 8 - (grp.nbits % 8);
|
|
size_t i;
|
|
|
|
for (i = 0; i < rnd_info.length - 1; i++) {
|
|
rnd_buf->x[i] = rnd_buf->x[i] << shift | rnd_buf->x[i+1] >> (8 - shift);
|
|
}
|
|
|
|
rnd_buf->x[rnd_info.length-1] <<= shift;
|
|
}
|
|
|
|
TEST_ASSERT(mbedtls_ecdsa_sign(&grp, &r, &s, &d, hash->x, hash->len,
|
|
mbedtls_test_rnd_buffer_rand, &rnd_info) == result);
|
|
|
|
if (result == 0) {
|
|
/* Check we generated the expected values */
|
|
TEST_EQUAL(mbedtls_mpi_cmp_mpi(&r, &r_check), 0);
|
|
TEST_EQUAL(mbedtls_mpi_cmp_mpi(&s, &s_check), 0);
|
|
|
|
/* Valid signature */
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len,
|
|
&Q, &r_check, &s_check), 0);
|
|
|
|
/* Invalid signature: wrong public key (G instead of Q) */
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len,
|
|
&grp.G, &r_check, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
|
|
/* Invalid signatures: r or s or both one off */
|
|
TEST_EQUAL(mbedtls_mpi_sub_int(&r, &r_check, 1), 0);
|
|
TEST_EQUAL(mbedtls_mpi_add_int(&s, &s_check, 1), 0);
|
|
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&r, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&r_check, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&r, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
|
|
/* Invalid signatures: r, s or both (CVE-2022-21449) are zero */
|
|
TEST_EQUAL(mbedtls_mpi_lset(&zero, 0), 0);
|
|
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&zero, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&r_check, &zero), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&zero, &zero), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
|
|
/* Invalid signatures: r, s or both are == N */
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&grp.N, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&r_check, &grp.N), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&grp.N, &grp.N), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
|
|
/* Invalid signatures: r, s or both are negative */
|
|
TEST_EQUAL(mbedtls_mpi_sub_mpi(&r, &r_check, &grp.N), 0);
|
|
TEST_EQUAL(mbedtls_mpi_sub_mpi(&s, &s_check, &grp.N), 0);
|
|
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&r, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&r_check, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&r, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
|
|
/* Invalid signatures: r or s or both are > N */
|
|
TEST_EQUAL(mbedtls_mpi_add_mpi(&r, &r_check, &grp.N), 0);
|
|
TEST_EQUAL(mbedtls_mpi_add_mpi(&s, &s_check, &grp.N), 0);
|
|
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&r, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&r_check, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
|
|
&r, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
}
|
|
|
|
exit:
|
|
mbedtls_ecp_group_free(&grp);
|
|
mbedtls_ecp_point_free(&Q);
|
|
mbedtls_mpi_free(&d); mbedtls_mpi_free(&r); mbedtls_mpi_free(&s);
|
|
mbedtls_mpi_free(&r_check); mbedtls_mpi_free(&s_check);
|
|
mbedtls_mpi_free(&zero);
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE depends_on:MBEDTLS_ECDSA_DETERMINISTIC */
|
|
void ecdsa_det_test_vectors(int id, char *d_str, int md_alg, data_t *hash,
|
|
char *r_str, char *s_str)
|
|
{
|
|
mbedtls_ecp_group grp;
|
|
mbedtls_mpi d, r, s, r_check, s_check;
|
|
|
|
MD_PSA_INIT();
|
|
|
|
mbedtls_ecp_group_init(&grp);
|
|
mbedtls_mpi_init(&d); mbedtls_mpi_init(&r); mbedtls_mpi_init(&s);
|
|
mbedtls_mpi_init(&r_check); mbedtls_mpi_init(&s_check);
|
|
|
|
TEST_ASSERT(mbedtls_ecp_group_load(&grp, id) == 0);
|
|
TEST_ASSERT(mbedtls_test_read_mpi(&d, d_str) == 0);
|
|
TEST_ASSERT(mbedtls_test_read_mpi(&r_check, r_str) == 0);
|
|
TEST_ASSERT(mbedtls_test_read_mpi(&s_check, s_str) == 0);
|
|
|
|
TEST_ASSERT(
|
|
mbedtls_ecdsa_sign_det_ext(&grp, &r, &s, &d,
|
|
hash->x, hash->len, md_alg,
|
|
mbedtls_test_rnd_std_rand,
|
|
NULL)
|
|
== 0);
|
|
|
|
TEST_ASSERT(mbedtls_mpi_cmp_mpi(&r, &r_check) == 0);
|
|
TEST_ASSERT(mbedtls_mpi_cmp_mpi(&s, &s_check) == 0);
|
|
|
|
exit:
|
|
mbedtls_ecp_group_free(&grp);
|
|
mbedtls_mpi_free(&d); mbedtls_mpi_free(&r); mbedtls_mpi_free(&s);
|
|
mbedtls_mpi_free(&r_check); mbedtls_mpi_free(&s_check);
|
|
MD_PSA_DONE();
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE depends_on:MBEDTLS_MD_CAN_SHA256 */
|
|
void ecdsa_write_read_zero(int id)
|
|
{
|
|
mbedtls_ecdsa_context ctx;
|
|
mbedtls_test_rnd_pseudo_info rnd_info;
|
|
unsigned char hash[32];
|
|
unsigned char sig[200];
|
|
size_t sig_len, i;
|
|
|
|
MD_PSA_INIT();
|
|
|
|
mbedtls_ecdsa_init(&ctx);
|
|
memset(&rnd_info, 0x00, sizeof(mbedtls_test_rnd_pseudo_info));
|
|
memset(hash, 0, sizeof(hash));
|
|
memset(sig, 0x2a, sizeof(sig));
|
|
|
|
/* generate signing key */
|
|
TEST_ASSERT(mbedtls_ecdsa_genkey(&ctx, id,
|
|
&mbedtls_test_rnd_pseudo_rand,
|
|
&rnd_info) == 0);
|
|
|
|
/* generate and write signature, then read and verify it */
|
|
TEST_ASSERT(mbedtls_ecdsa_write_signature(&ctx, MBEDTLS_MD_SHA256,
|
|
hash, sizeof(hash),
|
|
sig, sizeof(sig), &sig_len,
|
|
&mbedtls_test_rnd_pseudo_rand,
|
|
&rnd_info) == 0);
|
|
TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
|
|
sig, sig_len) == 0);
|
|
|
|
/* check we didn't write past the announced length */
|
|
for (i = sig_len; i < sizeof(sig); i++) {
|
|
TEST_ASSERT(sig[i] == 0x2a);
|
|
}
|
|
|
|
/* try verification with invalid length */
|
|
TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
|
|
sig, sig_len - 1) != 0);
|
|
TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
|
|
sig, sig_len + 1) != 0);
|
|
|
|
/* try invalid sequence tag */
|
|
sig[0]++;
|
|
TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
|
|
sig, sig_len) != 0);
|
|
sig[0]--;
|
|
|
|
/* try modifying r */
|
|
sig[10]++;
|
|
TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
|
|
sig, sig_len) == MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
sig[10]--;
|
|
|
|
/* try modifying s */
|
|
sig[sig_len - 1]++;
|
|
TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
|
|
sig, sig_len) == MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
sig[sig_len - 1]--;
|
|
|
|
exit:
|
|
mbedtls_ecdsa_free(&ctx);
|
|
MD_PSA_DONE();
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE depends_on:MBEDTLS_MD_CAN_SHA256 */
|
|
void ecdsa_write_read_random(int id)
|
|
{
|
|
mbedtls_ecdsa_context ctx;
|
|
mbedtls_test_rnd_pseudo_info rnd_info;
|
|
unsigned char hash[32];
|
|
unsigned char sig[200];
|
|
size_t sig_len, i;
|
|
|
|
MD_PSA_INIT();
|
|
|
|
mbedtls_ecdsa_init(&ctx);
|
|
memset(&rnd_info, 0x00, sizeof(mbedtls_test_rnd_pseudo_info));
|
|
memset(hash, 0, sizeof(hash));
|
|
memset(sig, 0x2a, sizeof(sig));
|
|
|
|
/* prepare material for signature */
|
|
TEST_ASSERT(mbedtls_test_rnd_pseudo_rand(&rnd_info,
|
|
hash, sizeof(hash)) == 0);
|
|
|
|
/* generate signing key */
|
|
TEST_ASSERT(mbedtls_ecdsa_genkey(&ctx, id,
|
|
&mbedtls_test_rnd_pseudo_rand,
|
|
&rnd_info) == 0);
|
|
|
|
/* generate and write signature, then read and verify it */
|
|
TEST_ASSERT(mbedtls_ecdsa_write_signature(&ctx, MBEDTLS_MD_SHA256,
|
|
hash, sizeof(hash),
|
|
sig, sizeof(sig), &sig_len,
|
|
&mbedtls_test_rnd_pseudo_rand,
|
|
&rnd_info) == 0);
|
|
TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
|
|
sig, sig_len) == 0);
|
|
|
|
/* check we didn't write past the announced length */
|
|
for (i = sig_len; i < sizeof(sig); i++) {
|
|
TEST_ASSERT(sig[i] == 0x2a);
|
|
}
|
|
|
|
/* try verification with invalid length */
|
|
TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
|
|
sig, sig_len - 1) != 0);
|
|
TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
|
|
sig, sig_len + 1) != 0);
|
|
|
|
/* try invalid sequence tag */
|
|
sig[0]++;
|
|
TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
|
|
sig, sig_len) != 0);
|
|
sig[0]--;
|
|
|
|
/* try modifying r */
|
|
sig[10]++;
|
|
TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
|
|
sig, sig_len) == MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
sig[10]--;
|
|
|
|
/* try modifying s */
|
|
sig[sig_len - 1]++;
|
|
TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
|
|
sig, sig_len) == MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
sig[sig_len - 1]--;
|
|
|
|
exit:
|
|
mbedtls_ecdsa_free(&ctx);
|
|
MD_PSA_DONE();
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE depends_on:MBEDTLS_ECP_RESTARTABLE */
|
|
void ecdsa_read_restart(int id, data_t *pk, data_t *hash, data_t *sig,
|
|
int max_ops, int min_restart, int max_restart)
|
|
{
|
|
mbedtls_ecdsa_context ctx;
|
|
mbedtls_ecdsa_restart_ctx rs_ctx;
|
|
int ret, cnt_restart;
|
|
|
|
mbedtls_ecdsa_init(&ctx);
|
|
mbedtls_ecdsa_restart_init(&rs_ctx);
|
|
|
|
TEST_ASSERT(mbedtls_ecp_group_load(&ctx.grp, id) == 0);
|
|
TEST_ASSERT(mbedtls_ecp_point_read_binary(&ctx.grp, &ctx.Q,
|
|
pk->x, pk->len) == 0);
|
|
|
|
mbedtls_ecp_set_max_ops(max_ops);
|
|
|
|
cnt_restart = 0;
|
|
do {
|
|
ret = mbedtls_ecdsa_read_signature_restartable(&ctx,
|
|
hash->x, hash->len, sig->x, sig->len,
|
|
&rs_ctx);
|
|
} while (ret == MBEDTLS_ERR_ECP_IN_PROGRESS && ++cnt_restart);
|
|
|
|
TEST_ASSERT(ret == 0);
|
|
TEST_ASSERT(cnt_restart >= min_restart);
|
|
TEST_ASSERT(cnt_restart <= max_restart);
|
|
|
|
/* try modifying r */
|
|
|
|
TEST_ASSERT(sig->len > 10);
|
|
sig->x[10]++;
|
|
do {
|
|
ret = mbedtls_ecdsa_read_signature_restartable(&ctx,
|
|
hash->x, hash->len, sig->x, sig->len,
|
|
&rs_ctx);
|
|
} while (ret == MBEDTLS_ERR_ECP_IN_PROGRESS);
|
|
TEST_ASSERT(ret == MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
sig->x[10]--;
|
|
|
|
/* try modifying s */
|
|
sig->x[sig->len - 1]++;
|
|
do {
|
|
ret = mbedtls_ecdsa_read_signature_restartable(&ctx,
|
|
hash->x, hash->len, sig->x, sig->len,
|
|
&rs_ctx);
|
|
} while (ret == MBEDTLS_ERR_ECP_IN_PROGRESS);
|
|
TEST_ASSERT(ret == MBEDTLS_ERR_ECP_VERIFY_FAILED);
|
|
sig->x[sig->len - 1]--;
|
|
|
|
/* Do we leak memory when aborting an operation?
|
|
* This test only makes sense when we actually restart */
|
|
if (min_restart > 0) {
|
|
ret = mbedtls_ecdsa_read_signature_restartable(&ctx,
|
|
hash->x, hash->len, sig->x, sig->len,
|
|
&rs_ctx);
|
|
TEST_ASSERT(ret == MBEDTLS_ERR_ECP_IN_PROGRESS);
|
|
}
|
|
|
|
exit:
|
|
mbedtls_ecdsa_free(&ctx);
|
|
mbedtls_ecdsa_restart_free(&rs_ctx);
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE depends_on:MBEDTLS_ECP_RESTARTABLE:MBEDTLS_ECDSA_DETERMINISTIC */
|
|
void ecdsa_write_restart(int id, char *d_str, int md_alg,
|
|
data_t *hash, data_t *sig_check,
|
|
int max_ops, int min_restart, int max_restart)
|
|
{
|
|
int ret, cnt_restart;
|
|
mbedtls_ecdsa_restart_ctx rs_ctx;
|
|
mbedtls_ecdsa_context ctx;
|
|
unsigned char sig[MBEDTLS_ECDSA_MAX_LEN];
|
|
size_t slen;
|
|
|
|
MD_PSA_INIT();
|
|
|
|
mbedtls_ecdsa_restart_init(&rs_ctx);
|
|
mbedtls_ecdsa_init(&ctx);
|
|
memset(sig, 0, sizeof(sig));
|
|
|
|
TEST_ASSERT(mbedtls_ecp_group_load(&ctx.grp, id) == 0);
|
|
TEST_ASSERT(mbedtls_test_read_mpi(&ctx.d, d_str) == 0);
|
|
|
|
mbedtls_ecp_set_max_ops(max_ops);
|
|
|
|
slen = sizeof(sig);
|
|
cnt_restart = 0;
|
|
do {
|
|
ret = mbedtls_ecdsa_write_signature_restartable(&ctx,
|
|
md_alg,
|
|
hash->x,
|
|
hash->len,
|
|
sig,
|
|
sizeof(sig),
|
|
&slen,
|
|
mbedtls_test_rnd_std_rand,
|
|
NULL,
|
|
&rs_ctx);
|
|
} while (ret == MBEDTLS_ERR_ECP_IN_PROGRESS && ++cnt_restart);
|
|
|
|
TEST_ASSERT(ret == 0);
|
|
TEST_ASSERT(slen == sig_check->len);
|
|
TEST_ASSERT(memcmp(sig, sig_check->x, slen) == 0);
|
|
|
|
TEST_ASSERT(cnt_restart >= min_restart);
|
|
TEST_ASSERT(cnt_restart <= max_restart);
|
|
|
|
/* Do we leak memory when aborting an operation?
|
|
* This test only makes sense when we actually restart */
|
|
if (min_restart > 0) {
|
|
ret = mbedtls_ecdsa_write_signature_restartable(&ctx,
|
|
md_alg,
|
|
hash->x,
|
|
hash->len,
|
|
sig,
|
|
sizeof(sig),
|
|
&slen,
|
|
mbedtls_test_rnd_std_rand,
|
|
NULL,
|
|
&rs_ctx);
|
|
TEST_ASSERT(ret == MBEDTLS_ERR_ECP_IN_PROGRESS);
|
|
}
|
|
|
|
exit:
|
|
mbedtls_ecdsa_restart_free(&rs_ctx);
|
|
mbedtls_ecdsa_free(&ctx);
|
|
MD_PSA_DONE();
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecdsa_verify(int grp_id, char *x, char *y, char *r, char *s, data_t *content, int expected)
|
|
{
|
|
mbedtls_ecdsa_context ctx;
|
|
mbedtls_mpi sig_r, sig_s;
|
|
|
|
mbedtls_ecdsa_init(&ctx);
|
|
mbedtls_mpi_init(&sig_r);
|
|
mbedtls_mpi_init(&sig_s);
|
|
|
|
/* Prepare ECP group context */
|
|
TEST_EQUAL(mbedtls_ecp_group_load(&ctx.grp, grp_id), 0);
|
|
|
|
/* Prepare public key */
|
|
TEST_EQUAL(mbedtls_test_read_mpi(&ctx.Q.X, x), 0);
|
|
TEST_EQUAL(mbedtls_test_read_mpi(&ctx.Q.Y, y), 0);
|
|
TEST_EQUAL(mbedtls_mpi_lset(&ctx.Q.Z, 1), 0);
|
|
|
|
/* Prepare signature R & S */
|
|
TEST_EQUAL(mbedtls_test_read_mpi(&sig_r, r), 0);
|
|
TEST_EQUAL(mbedtls_test_read_mpi(&sig_s, s), 0);
|
|
|
|
/* Test whether public key has expected validity */
|
|
TEST_EQUAL(mbedtls_ecp_check_pubkey(&ctx.grp, &ctx.Q),
|
|
expected == MBEDTLS_ERR_ECP_INVALID_KEY ? MBEDTLS_ERR_ECP_INVALID_KEY : 0);
|
|
|
|
/* Verification */
|
|
int result = mbedtls_ecdsa_verify(&ctx.grp, content->x, content->len, &ctx.Q, &sig_r, &sig_s);
|
|
|
|
TEST_EQUAL(result, expected);
|
|
exit:
|
|
mbedtls_ecdsa_free(&ctx);
|
|
mbedtls_mpi_free(&sig_r);
|
|
mbedtls_mpi_free(&sig_s);
|
|
}
|
|
/* END_CASE */
|