Added support for ECC keypair generation.

Signed-off-by: Pol Henarejos <pol.henarejos@cttc.es>
This commit is contained in:
Pol Henarejos
2022-04-21 23:13:05 +02:00
parent c006246aa0
commit d69342d2d0

View File

@@ -833,23 +833,23 @@ int load_private_key_rsa(mbedtls_rsa_context *ctx, file_t *fkey) {
} }
mbedtls_ecp_group_id get_ec_group_id_from_attr(const uint8_t *algo, size_t algo_len) { mbedtls_ecp_group_id get_ec_group_id_from_attr(const uint8_t *algo, size_t algo_len) {
if (memcmp(algorithm_attr_p256k1, algo, algo_len) == 0) if (memcmp(algorithm_attr_p256k1+2, algo, algo_len) == 0)
return MBEDTLS_ECP_DP_SECP256K1; return MBEDTLS_ECP_DP_SECP256K1;
else if (memcmp(algorithm_attr_p256r1, algo, algo_len) == 0) else if (memcmp(algorithm_attr_p256r1+2, algo, algo_len) == 0)
return MBEDTLS_ECP_DP_SECP256R1; return MBEDTLS_ECP_DP_SECP256R1;
else if (memcmp(algorithm_attr_p384r1, algo, algo_len) == 0) else if (memcmp(algorithm_attr_p384r1+2, algo, algo_len) == 0)
return MBEDTLS_ECP_DP_SECP384R1; return MBEDTLS_ECP_DP_SECP384R1;
else if (memcmp(algorithm_attr_p521r1, algo, algo_len) == 0) else if (memcmp(algorithm_attr_p521r1+2, algo, algo_len) == 0)
return MBEDTLS_ECP_DP_SECP521R1; return MBEDTLS_ECP_DP_SECP521R1;
else if (memcmp(algorithm_attr_bp256r1, algo, algo_len) == 0) else if (memcmp(algorithm_attr_bp256r1+2, algo, algo_len) == 0)
return MBEDTLS_ECP_DP_BP256R1; return MBEDTLS_ECP_DP_BP256R1;
else if (memcmp(algorithm_attr_bp384r1, algo, algo_len) == 0) else if (memcmp(algorithm_attr_bp384r1+2, algo, algo_len) == 0)
return MBEDTLS_ECP_DP_BP384R1; return MBEDTLS_ECP_DP_BP384R1;
else if (memcmp(algorithm_attr_bp512r1, algo, algo_len) == 0) else if (memcmp(algorithm_attr_bp512r1+2, algo, algo_len) == 0)
return MBEDTLS_ECP_DP_BP512R1; return MBEDTLS_ECP_DP_BP512R1;
else if (memcmp(algorithm_attr_cv25519, algo, algo_len) == 0) else if (memcmp(algorithm_attr_cv25519+2, algo, algo_len) == 0)
return MBEDTLS_ECP_DP_CURVE25519; return MBEDTLS_ECP_DP_CURVE25519;
else if (memcmp(algorithm_attr_x448, algo, algo_len) == 0) else if (memcmp(algorithm_attr_x448+2, algo, algo_len) == 0)
return MBEDTLS_ECP_DP_CURVE448; return MBEDTLS_ECP_DP_CURVE448;
return MBEDTLS_ECP_DP_NONE; return MBEDTLS_ECP_DP_NONE;
} }
@@ -867,6 +867,16 @@ void make_rsa_response(mbedtls_rsa_context *rsa) {
put_uint16_t(res_APDU_size-5, res_APDU+3); put_uint16_t(res_APDU_size-5, res_APDU+3);
} }
void make_ecdsa_response(mbedtls_ecdsa_context *ecdsa) {
memcpy(res_APDU, "\x7f\x49\x81\x00", 4);
res_APDU_size = 4;
res_APDU[res_APDU_size++] = 0x86;
size_t olen = 0;
mbedtls_ecp_point_write_binary(&ecdsa->grp, &ecdsa->Q, MBEDTLS_ECP_PF_UNCOMPRESSED, &olen, res_APDU+res_APDU_size, 4096);
res_APDU_size += olen;
res_APDU[3] = res_APDU_size-4;
}
static int cmd_keypair_gen() { static int cmd_keypair_gen() {
if (P2(apdu) != 0x0) if (P2(apdu) != 0x0)
return SW_INCORRECT_P1P2(); return SW_INCORRECT_P1P2();
@@ -890,9 +900,10 @@ static int cmd_keypair_gen() {
const uint8_t *algo = algorithm_attr_rsa2k+1; const uint8_t *algo = algorithm_attr_rsa2k+1;
uint16_t algo_len = algorithm_attr_rsa2k[0]; uint16_t algo_len = algorithm_attr_rsa2k[0];
if (algo_ef && algo_ef->data) { if (algo_ef && algo_ef->data) {
algo_len = file_read_uint16(algo_ef->data);
algo = file_read(algo_ef->data+2); algo = file_read(algo_ef->data+2);
algo_len = file_read_uint16(algo_ef->data);
} }
DEBUG_PAYLOAD(algo,algo_len);
if (P1(apdu) == 0x80) { //generate if (P1(apdu) == 0x80) { //generate
if (algo[0] == ALGO_RSA) { if (algo[0] == ALGO_RSA) {
int exponent = 65537, nlen = (algo[1] << 8) | algo[2]; int exponent = 65537, nlen = (algo[1] << 8) | algo[2];
@@ -917,11 +928,9 @@ static int cmd_keypair_gen() {
} }
else if (algo[0] == ALGO_ECDH || algo[0] == ALGO_ECDSA) { else if (algo[0] == ALGO_ECDH || algo[0] == ALGO_ECDSA) {
printf("KEYPAIR ECDSA\r\n"); printf("KEYPAIR ECDSA\r\n");
mbedtls_ecp_group_id gid = get_ec_group_id_from_attr(algo, algo_len); mbedtls_ecp_group_id gid = get_ec_group_id_from_attr(algo+1, algo_len-1);
if (gid == MBEDTLS_ECP_DP_NONE) if (gid == MBEDTLS_ECP_DP_NONE)
return SW_FUNC_NOT_SUPPORTED(); return SW_FUNC_NOT_SUPPORTED();
if ((gid == MBEDTLS_ECP_DP_CURVE25519 || gid == MBEDTLS_ECP_DP_CURVE448) && (fid == EF_PK_SIG || fid == EF_PK_AUT))
return SW_FUNC_NOT_SUPPORTED();
mbedtls_ecdsa_context ecdsa; mbedtls_ecdsa_context ecdsa;
mbedtls_ecdsa_init(&ecdsa); mbedtls_ecdsa_init(&ecdsa);
uint8_t index = 0; uint8_t index = 0;
@@ -931,6 +940,7 @@ static int cmd_keypair_gen() {
return SW_EXEC_ERROR(); return SW_EXEC_ERROR();
} }
r = store_keys(&ecdsa, algo[0], fid); r = store_keys(&ecdsa, algo[0], fid);
make_ecdsa_response(&ecdsa);
mbedtls_ecdsa_free(&ecdsa); mbedtls_ecdsa_free(&ecdsa);
if (r != CCID_OK) if (r != CCID_OK)
return SW_EXEC_ERROR(); return SW_EXEC_ERROR();
@@ -947,17 +957,11 @@ static int cmd_keypair_gen() {
return SW_OK(); return SW_OK();
} }
else if (P1(apdu) == 0x81) { //read else if (P1(apdu) == 0x81) { //read
if (algo[0] == ALGO_RSA) { file_t *ef = search_by_fid(fid+3, NULL, SPECIFY_EF);
file_t *ef = search_by_fid(fid+3, NULL, SPECIFY_EF); if (!ef || !ef->data)
if (!ef || !ef->data) return SW_FILE_NOT_FOUND();
return SW_FILE_NOT_FOUND(); res_APDU_size = file_read_uint16(ef->data);
res_APDU_size = file_read_uint16(ef->data); memcpy(res_APDU, file_read(ef->data+2), res_APDU_size);
memcpy(res_APDU, file_read(ef->data+2), res_APDU_size);
}
else if (algo[0] == ALGO_ECDH || algo[0] == ALGO_ECDSA) {
}
else
return SW_FUNC_NOT_SUPPORTED();
return SW_OK(); return SW_OK();
} }
return SW_INCORRECT_P1P2(); return SW_INCORRECT_P1P2();
@@ -1013,6 +1017,9 @@ static int cmd_pso_sig() {
} }
res_APDU_size = key_size; res_APDU_size = key_size;
//apdu.expected_res_size = key_size; //apdu.expected_res_size = key_size;
}
else if (algo[0] == ALGO_ECDH || algo[0] == ALGO_ECDSA) {
} }
return SW_OK(); return SW_OK();
} }