proxmark3/common/mbedtls/pk.c

360 lines
10 KiB
C
Raw Normal View History

2018-11-08 00:05:02 +08:00
/*
* Public Key abstraction layer
*
* Copyright (C) 2006-2015, ARM Limited, All Rights Reserved
* SPDX-License-Identifier: GPL-2.0
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*
* This file is part of mbed TLS (https://tls.mbed.org)
*/
#if !defined(MBEDTLS_CONFIG_FILE)
#include "mbedtls/config.h"
#else
#include MBEDTLS_CONFIG_FILE
#endif
#if defined(MBEDTLS_PK_C)
#include "mbedtls/pk.h"
#include "mbedtls/pk_internal.h"
#include "mbedtls/platform_util.h"
#if defined(MBEDTLS_RSA_C)
#include "mbedtls/rsa.h"
#endif
#if defined(MBEDTLS_ECP_C)
#include "mbedtls/ecp.h"
#endif
#if defined(MBEDTLS_ECDSA_C)
#include "mbedtls/ecdsa.h"
#endif
#include <limits.h>
#include <stdint.h>
/*
* Initialise a mbedtls_pk_context
*/
void mbedtls_pk_init(mbedtls_pk_context *ctx) {
2019-03-10 07:00:59 +08:00
if (ctx == NULL)
2018-11-08 00:05:02 +08:00
return;
ctx->pk_info = NULL;
ctx->pk_ctx = NULL;
}
/*
* Free (the components of) a mbedtls_pk_context
*/
void mbedtls_pk_free(mbedtls_pk_context *ctx) {
2019-03-10 07:00:59 +08:00
if (ctx == NULL || ctx->pk_info == NULL)
2018-11-08 00:05:02 +08:00
return;
2019-03-10 07:00:59 +08:00
ctx->pk_info->ctx_free_func(ctx->pk_ctx);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
mbedtls_platform_zeroize(ctx, sizeof(mbedtls_pk_context));
2018-11-08 00:05:02 +08:00
}
/*
* Get pk_info structure from type
*/
const mbedtls_pk_info_t *mbedtls_pk_info_from_type(mbedtls_pk_type_t pk_type) {
2019-03-10 07:00:59 +08:00
switch (pk_type) {
2018-11-08 00:05:02 +08:00
#if defined(MBEDTLS_RSA_C)
case MBEDTLS_PK_RSA:
2019-03-10 07:00:59 +08:00
return (&mbedtls_rsa_info);
2018-11-08 00:05:02 +08:00
#endif
#if defined(MBEDTLS_ECP_C)
case MBEDTLS_PK_ECKEY:
2019-03-10 07:00:59 +08:00
return (&mbedtls_eckey_info);
2018-11-08 00:05:02 +08:00
case MBEDTLS_PK_ECKEY_DH:
2019-03-10 07:00:59 +08:00
return (&mbedtls_eckeydh_info);
2018-11-08 00:05:02 +08:00
#endif
#if defined(MBEDTLS_ECDSA_C)
case MBEDTLS_PK_ECDSA:
2019-03-10 07:00:59 +08:00
return (&mbedtls_ecdsa_info);
2018-11-08 00:05:02 +08:00
#endif
/* MBEDTLS_PK_RSA_ALT omitted on purpose */
default:
2019-03-10 07:00:59 +08:00
return (NULL);
2018-11-08 00:05:02 +08:00
}
}
/*
* Initialise context
*/
int mbedtls_pk_setup(mbedtls_pk_context *ctx, const mbedtls_pk_info_t *info) {
2019-03-10 07:00:59 +08:00
if (ctx == NULL || info == NULL || ctx->pk_info != NULL)
return (MBEDTLS_ERR_PK_BAD_INPUT_DATA);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
if ((ctx->pk_ctx = info->ctx_alloc_func()) == NULL)
return (MBEDTLS_ERR_PK_ALLOC_FAILED);
2018-11-08 00:05:02 +08:00
ctx->pk_info = info;
2019-03-10 07:00:59 +08:00
return (0);
2018-11-08 00:05:02 +08:00
}
#if defined(MBEDTLS_PK_RSA_ALT_SUPPORT)
/*
* Initialize an RSA-alt context
*/
2019-03-10 07:00:59 +08:00
int mbedtls_pk_setup_rsa_alt(mbedtls_pk_context *ctx, void *key,
mbedtls_pk_rsa_alt_decrypt_func decrypt_func,
mbedtls_pk_rsa_alt_sign_func sign_func,
mbedtls_pk_rsa_alt_key_len_func key_len_func) {
2018-11-08 00:05:02 +08:00
mbedtls_rsa_alt_context *rsa_alt;
const mbedtls_pk_info_t *info = &mbedtls_rsa_alt_info;
2019-03-10 07:00:59 +08:00
if (ctx == NULL || ctx->pk_info != NULL)
return (MBEDTLS_ERR_PK_BAD_INPUT_DATA);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
if ((ctx->pk_ctx = info->ctx_alloc_func()) == NULL)
return (MBEDTLS_ERR_PK_ALLOC_FAILED);
2018-11-08 00:05:02 +08:00
ctx->pk_info = info;
rsa_alt = (mbedtls_rsa_alt_context *) ctx->pk_ctx;
rsa_alt->key = key;
rsa_alt->decrypt_func = decrypt_func;
rsa_alt->sign_func = sign_func;
rsa_alt->key_len_func = key_len_func;
2019-03-10 07:00:59 +08:00
return (0);
2018-11-08 00:05:02 +08:00
}
#endif /* MBEDTLS_PK_RSA_ALT_SUPPORT */
/*
* Tell if a PK can do the operations of the given type
*/
int mbedtls_pk_can_do(const mbedtls_pk_context *ctx, mbedtls_pk_type_t type) {
2018-11-08 00:05:02 +08:00
/* null or NONE context can't do anything */
2019-03-10 07:00:59 +08:00
if (ctx == NULL || ctx->pk_info == NULL)
return (0);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
return (ctx->pk_info->can_do(type));
2018-11-08 00:05:02 +08:00
}
/*
* Helper for mbedtls_pk_sign and mbedtls_pk_verify
*/
static inline int pk_hashlen_helper(mbedtls_md_type_t md_alg, size_t *hash_len) {
2018-11-08 00:05:02 +08:00
const mbedtls_md_info_t *md_info;
2019-03-10 07:00:59 +08:00
if (*hash_len != 0)
return (0);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
if ((md_info = mbedtls_md_info_from_type(md_alg)) == NULL)
return (-1);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
*hash_len = mbedtls_md_get_size(md_info);
return (0);
2018-11-08 00:05:02 +08:00
}
/*
* Verify a signature
*/
2019-03-10 07:00:59 +08:00
int mbedtls_pk_verify(mbedtls_pk_context *ctx, mbedtls_md_type_t md_alg,
const unsigned char *hash, size_t hash_len,
const unsigned char *sig, size_t sig_len) {
2019-03-10 07:00:59 +08:00
if (ctx == NULL || ctx->pk_info == NULL ||
pk_hashlen_helper(md_alg, &hash_len) != 0)
2019-03-10 07:00:59 +08:00
return (MBEDTLS_ERR_PK_BAD_INPUT_DATA);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
if (ctx->pk_info->verify_func == NULL)
return (MBEDTLS_ERR_PK_TYPE_MISMATCH);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
return (ctx->pk_info->verify_func(ctx->pk_ctx, md_alg, hash, hash_len,
sig, sig_len));
2018-11-08 00:05:02 +08:00
}
/*
* Verify a signature with options
*/
2019-03-10 07:00:59 +08:00
int mbedtls_pk_verify_ext(mbedtls_pk_type_t type, const void *options,
mbedtls_pk_context *ctx, mbedtls_md_type_t md_alg,
const unsigned char *hash, size_t hash_len,
const unsigned char *sig, size_t sig_len) {
2019-03-10 07:00:59 +08:00
if (ctx == NULL || ctx->pk_info == NULL)
return (MBEDTLS_ERR_PK_BAD_INPUT_DATA);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
if (! mbedtls_pk_can_do(ctx, type))
return (MBEDTLS_ERR_PK_TYPE_MISMATCH);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
if (type == MBEDTLS_PK_RSASSA_PSS) {
2018-11-08 00:05:02 +08:00
#if defined(MBEDTLS_RSA_C) && defined(MBEDTLS_PKCS1_V21)
int ret;
const mbedtls_pk_rsassa_pss_options *pss_opts;
#if SIZE_MAX > UINT_MAX
2019-03-10 07:00:59 +08:00
if (md_alg == MBEDTLS_MD_NONE && UINT_MAX < hash_len)
return (MBEDTLS_ERR_PK_BAD_INPUT_DATA);
2018-11-08 00:05:02 +08:00
#endif /* SIZE_MAX > UINT_MAX */
2019-03-10 07:00:59 +08:00
if (options == NULL)
return (MBEDTLS_ERR_PK_BAD_INPUT_DATA);
2018-11-08 00:05:02 +08:00
pss_opts = (const mbedtls_pk_rsassa_pss_options *) options;
2019-03-10 07:00:59 +08:00
if (sig_len < mbedtls_pk_get_len(ctx))
return (MBEDTLS_ERR_RSA_VERIFY_FAILED);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
ret = mbedtls_rsa_rsassa_pss_verify_ext(mbedtls_pk_rsa(*ctx),
NULL, NULL, MBEDTLS_RSA_PUBLIC,
md_alg, (unsigned int) hash_len, hash,
pss_opts->mgf1_hash_id,
pss_opts->expected_salt_len,
sig);
if (ret != 0)
return (ret);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
if (sig_len > mbedtls_pk_get_len(ctx))
return (MBEDTLS_ERR_PK_SIG_LEN_MISMATCH);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
return (0);
2018-11-08 00:05:02 +08:00
#else
2019-03-10 07:00:59 +08:00
return (MBEDTLS_ERR_PK_FEATURE_UNAVAILABLE);
2018-11-08 00:05:02 +08:00
#endif /* MBEDTLS_RSA_C && MBEDTLS_PKCS1_V21 */
}
/* General case: no options */
2019-03-10 07:00:59 +08:00
if (options != NULL)
return (MBEDTLS_ERR_PK_BAD_INPUT_DATA);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
return (mbedtls_pk_verify(ctx, md_alg, hash, hash_len, sig, sig_len));
2018-11-08 00:05:02 +08:00
}
/*
* Make a signature
*/
2019-03-10 07:00:59 +08:00
int mbedtls_pk_sign(mbedtls_pk_context *ctx, mbedtls_md_type_t md_alg,
const unsigned char *hash, size_t hash_len,
unsigned char *sig, size_t *sig_len,
int (*f_rng)(void *, unsigned char *, size_t), void *p_rng) {
2019-03-10 07:00:59 +08:00
if (ctx == NULL || ctx->pk_info == NULL ||
pk_hashlen_helper(md_alg, &hash_len) != 0)
2019-03-10 07:00:59 +08:00
return (MBEDTLS_ERR_PK_BAD_INPUT_DATA);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
if (ctx->pk_info->sign_func == NULL)
return (MBEDTLS_ERR_PK_TYPE_MISMATCH);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
return (ctx->pk_info->sign_func(ctx->pk_ctx, md_alg, hash, hash_len,
sig, sig_len, f_rng, p_rng));
2018-11-08 00:05:02 +08:00
}
/*
* Decrypt message
*/
2019-03-10 07:00:59 +08:00
int mbedtls_pk_decrypt(mbedtls_pk_context *ctx,
const unsigned char *input, size_t ilen,
unsigned char *output, size_t *olen, size_t osize,
int (*f_rng)(void *, unsigned char *, size_t), void *p_rng) {
2019-03-10 07:00:59 +08:00
if (ctx == NULL || ctx->pk_info == NULL)
return (MBEDTLS_ERR_PK_BAD_INPUT_DATA);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
if (ctx->pk_info->decrypt_func == NULL)
return (MBEDTLS_ERR_PK_TYPE_MISMATCH);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
return (ctx->pk_info->decrypt_func(ctx->pk_ctx, input, ilen,
output, olen, osize, f_rng, p_rng));
2018-11-08 00:05:02 +08:00
}
/*
* Encrypt message
*/
2019-03-10 07:00:59 +08:00
int mbedtls_pk_encrypt(mbedtls_pk_context *ctx,
const unsigned char *input, size_t ilen,
unsigned char *output, size_t *olen, size_t osize,
int (*f_rng)(void *, unsigned char *, size_t), void *p_rng) {
2019-03-10 07:00:59 +08:00
if (ctx == NULL || ctx->pk_info == NULL)
return (MBEDTLS_ERR_PK_BAD_INPUT_DATA);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
if (ctx->pk_info->encrypt_func == NULL)
return (MBEDTLS_ERR_PK_TYPE_MISMATCH);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
return (ctx->pk_info->encrypt_func(ctx->pk_ctx, input, ilen,
output, olen, osize, f_rng, p_rng));
2018-11-08 00:05:02 +08:00
}
/*
* Check public-private key pair
*/
int mbedtls_pk_check_pair(const mbedtls_pk_context *pub, const mbedtls_pk_context *prv) {
2019-03-10 07:00:59 +08:00
if (pub == NULL || pub->pk_info == NULL ||
prv == NULL || prv->pk_info == NULL ||
prv->pk_info->check_pair_func == NULL) {
2019-03-10 07:00:59 +08:00
return (MBEDTLS_ERR_PK_BAD_INPUT_DATA);
2018-11-08 00:05:02 +08:00
}
2019-03-10 07:00:59 +08:00
if (prv->pk_info->type == MBEDTLS_PK_RSA_ALT) {
if (pub->pk_info->type != MBEDTLS_PK_RSA)
return (MBEDTLS_ERR_PK_TYPE_MISMATCH);
} else {
if (pub->pk_info != prv->pk_info)
return (MBEDTLS_ERR_PK_TYPE_MISMATCH);
2018-11-08 00:05:02 +08:00
}
2019-03-10 07:00:59 +08:00
return (prv->pk_info->check_pair_func(pub->pk_ctx, prv->pk_ctx));
2018-11-08 00:05:02 +08:00
}
/*
* Get key size in bits
*/
size_t mbedtls_pk_get_bitlen(const mbedtls_pk_context *ctx) {
2019-03-10 07:00:59 +08:00
if (ctx == NULL || ctx->pk_info == NULL)
return (0);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
return (ctx->pk_info->get_bitlen(ctx->pk_ctx));
2018-11-08 00:05:02 +08:00
}
/*
* Export debug information
*/
int mbedtls_pk_debug(const mbedtls_pk_context *ctx, mbedtls_pk_debug_item *items) {
2019-03-10 07:00:59 +08:00
if (ctx == NULL || ctx->pk_info == NULL)
return (MBEDTLS_ERR_PK_BAD_INPUT_DATA);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
if (ctx->pk_info->debug_func == NULL)
return (MBEDTLS_ERR_PK_TYPE_MISMATCH);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
ctx->pk_info->debug_func(ctx->pk_ctx, items);
return (0);
2018-11-08 00:05:02 +08:00
}
/*
* Access the PK type name
*/
const char *mbedtls_pk_get_name(const mbedtls_pk_context *ctx) {
2019-03-10 07:00:59 +08:00
if (ctx == NULL || ctx->pk_info == NULL)
return ("invalid PK");
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
return (ctx->pk_info->name);
2018-11-08 00:05:02 +08:00
}
/*
* Access the PK type
*/
mbedtls_pk_type_t mbedtls_pk_get_type(const mbedtls_pk_context *ctx) {
2019-03-10 07:00:59 +08:00
if (ctx == NULL || ctx->pk_info == NULL)
return (MBEDTLS_PK_NONE);
2018-11-08 00:05:02 +08:00
2019-03-10 07:00:59 +08:00
return (ctx->pk_info->type);
2018-11-08 00:05:02 +08:00
}
#endif /* MBEDTLS_PK_C */