takashi kadono / Mbed OS Nucleo_446

Important changes to repositories hosted on mbed.com

Mbed hosted mercurial repositories are deprecated and are due to be permanently deleted in July 2026.

To keep a copy of this software download the repository Zip archive or clone locally using Mercurial.

It is also possible to export all your personal repositories from the account settings page.

Dependencies:   ssd1331

mbed-os/features/mbedtls/src/pk.c@0:8fdf9a60065b, 2018-10-10 (annotated)

Committer:
kadonotakashi
Date:
Wed Oct 10 00:33:53 2018 +0000
Revision:
0:8fdf9a60065b
how to make mbed librry

Who changed what in which revision?

UserRevisionLine numberNew contents of line
kadonotakashi 0:8fdf9a60065b 1 /*
kadonotakashi 0:8fdf9a60065b 2 * Public Key abstraction layer
kadonotakashi 0:8fdf9a60065b 3 *
kadonotakashi 0:8fdf9a60065b 4 * Copyright (C) 2006-2015, ARM Limited, All Rights Reserved
kadonotakashi 0:8fdf9a60065b 5 * SPDX-License-Identifier: Apache-2.0
kadonotakashi 0:8fdf9a60065b 6 *
kadonotakashi 0:8fdf9a60065b 7 * Licensed under the Apache License, Version 2.0 (the "License"); you may
kadonotakashi 0:8fdf9a60065b 8 * not use this file except in compliance with the License.
kadonotakashi 0:8fdf9a60065b 9 * You may obtain a copy of the License at
kadonotakashi 0:8fdf9a60065b 10 *
kadonotakashi 0:8fdf9a60065b 11 * http://www.apache.org/licenses/LICENSE-2.0
kadonotakashi 0:8fdf9a60065b 12 *
kadonotakashi 0:8fdf9a60065b 13 * Unless required by applicable law or agreed to in writing, software
kadonotakashi 0:8fdf9a60065b 14 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
kadonotakashi 0:8fdf9a60065b 15 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
kadonotakashi 0:8fdf9a60065b 16 * See the License for the specific language governing permissions and
kadonotakashi 0:8fdf9a60065b 17 * limitations under the License.
kadonotakashi 0:8fdf9a60065b 18 *
kadonotakashi 0:8fdf9a60065b 19 * This file is part of mbed TLS (https://tls.mbed.org)
kadonotakashi 0:8fdf9a60065b 20 */
kadonotakashi 0:8fdf9a60065b 21
kadonotakashi 0:8fdf9a60065b 22 #if !defined(MBEDTLS_CONFIG_FILE)
kadonotakashi 0:8fdf9a60065b 23 #include "mbedtls/config.h"
kadonotakashi 0:8fdf9a60065b 24 #else
kadonotakashi 0:8fdf9a60065b 25 #include MBEDTLS_CONFIG_FILE
kadonotakashi 0:8fdf9a60065b 26 #endif
kadonotakashi 0:8fdf9a60065b 27
kadonotakashi 0:8fdf9a60065b 28 #if defined(MBEDTLS_PK_C)
kadonotakashi 0:8fdf9a60065b 29 #include "mbedtls/pk.h"
kadonotakashi 0:8fdf9a60065b 30 #include "mbedtls/pk_internal.h"
kadonotakashi 0:8fdf9a60065b 31
kadonotakashi 0:8fdf9a60065b 32 #include "mbedtls/platform_util.h"
kadonotakashi 0:8fdf9a60065b 33
kadonotakashi 0:8fdf9a60065b 34 #if defined(MBEDTLS_RSA_C)
kadonotakashi 0:8fdf9a60065b 35 #include "mbedtls/rsa.h"
kadonotakashi 0:8fdf9a60065b 36 #endif
kadonotakashi 0:8fdf9a60065b 37 #if defined(MBEDTLS_ECP_C)
kadonotakashi 0:8fdf9a60065b 38 #include "mbedtls/ecp.h"
kadonotakashi 0:8fdf9a60065b 39 #endif
kadonotakashi 0:8fdf9a60065b 40 #if defined(MBEDTLS_ECDSA_C)
kadonotakashi 0:8fdf9a60065b 41 #include "mbedtls/ecdsa.h"
kadonotakashi 0:8fdf9a60065b 42 #endif
kadonotakashi 0:8fdf9a60065b 43
kadonotakashi 0:8fdf9a60065b 44 #include <limits.h>
kadonotakashi 0:8fdf9a60065b 45 #include <stdint.h>
kadonotakashi 0:8fdf9a60065b 46
kadonotakashi 0:8fdf9a60065b 47 /*
kadonotakashi 0:8fdf9a60065b 48 * Initialise a mbedtls_pk_context
kadonotakashi 0:8fdf9a60065b 49 */
kadonotakashi 0:8fdf9a60065b 50 void mbedtls_pk_init( mbedtls_pk_context *ctx )
kadonotakashi 0:8fdf9a60065b 51 {
kadonotakashi 0:8fdf9a60065b 52 if( ctx == NULL )
kadonotakashi 0:8fdf9a60065b 53 return;
kadonotakashi 0:8fdf9a60065b 54
kadonotakashi 0:8fdf9a60065b 55 ctx->pk_info = NULL;
kadonotakashi 0:8fdf9a60065b 56 ctx->pk_ctx = NULL;
kadonotakashi 0:8fdf9a60065b 57 }
kadonotakashi 0:8fdf9a60065b 58
kadonotakashi 0:8fdf9a60065b 59 /*
kadonotakashi 0:8fdf9a60065b 60 * Free (the components of) a mbedtls_pk_context
kadonotakashi 0:8fdf9a60065b 61 */
kadonotakashi 0:8fdf9a60065b 62 void mbedtls_pk_free( mbedtls_pk_context *ctx )
kadonotakashi 0:8fdf9a60065b 63 {
kadonotakashi 0:8fdf9a60065b 64 if( ctx == NULL || ctx->pk_info == NULL )
kadonotakashi 0:8fdf9a60065b 65 return;
kadonotakashi 0:8fdf9a60065b 66
kadonotakashi 0:8fdf9a60065b 67 ctx->pk_info->ctx_free_func( ctx->pk_ctx );
kadonotakashi 0:8fdf9a60065b 68
kadonotakashi 0:8fdf9a60065b 69 mbedtls_platform_zeroize( ctx, sizeof( mbedtls_pk_context ) );
kadonotakashi 0:8fdf9a60065b 70 }
kadonotakashi 0:8fdf9a60065b 71
kadonotakashi 0:8fdf9a60065b 72 /*
kadonotakashi 0:8fdf9a60065b 73 * Get pk_info structure from type
kadonotakashi 0:8fdf9a60065b 74 */
kadonotakashi 0:8fdf9a60065b 75 const mbedtls_pk_info_t * mbedtls_pk_info_from_type( mbedtls_pk_type_t pk_type )
kadonotakashi 0:8fdf9a60065b 76 {
kadonotakashi 0:8fdf9a60065b 77 switch( pk_type ) {
kadonotakashi 0:8fdf9a60065b 78 #if defined(MBEDTLS_RSA_C)
kadonotakashi 0:8fdf9a60065b 79 case MBEDTLS_PK_RSA:
kadonotakashi 0:8fdf9a60065b 80 return( &mbedtls_rsa_info );
kadonotakashi 0:8fdf9a60065b 81 #endif
kadonotakashi 0:8fdf9a60065b 82 #if defined(MBEDTLS_ECP_C)
kadonotakashi 0:8fdf9a60065b 83 case MBEDTLS_PK_ECKEY:
kadonotakashi 0:8fdf9a60065b 84 return( &mbedtls_eckey_info );
kadonotakashi 0:8fdf9a60065b 85 case MBEDTLS_PK_ECKEY_DH:
kadonotakashi 0:8fdf9a60065b 86 return( &mbedtls_eckeydh_info );
kadonotakashi 0:8fdf9a60065b 87 #endif
kadonotakashi 0:8fdf9a60065b 88 #if defined(MBEDTLS_ECDSA_C)
kadonotakashi 0:8fdf9a60065b 89 case MBEDTLS_PK_ECDSA:
kadonotakashi 0:8fdf9a60065b 90 return( &mbedtls_ecdsa_info );
kadonotakashi 0:8fdf9a60065b 91 #endif
kadonotakashi 0:8fdf9a60065b 92 /* MBEDTLS_PK_RSA_ALT omitted on purpose */
kadonotakashi 0:8fdf9a60065b 93 default:
kadonotakashi 0:8fdf9a60065b 94 return( NULL );
kadonotakashi 0:8fdf9a60065b 95 }
kadonotakashi 0:8fdf9a60065b 96 }
kadonotakashi 0:8fdf9a60065b 97
kadonotakashi 0:8fdf9a60065b 98 /*
kadonotakashi 0:8fdf9a60065b 99 * Initialise context
kadonotakashi 0:8fdf9a60065b 100 */
kadonotakashi 0:8fdf9a60065b 101 int mbedtls_pk_setup( mbedtls_pk_context *ctx, const mbedtls_pk_info_t *info )
kadonotakashi 0:8fdf9a60065b 102 {
kadonotakashi 0:8fdf9a60065b 103 if( ctx == NULL || info == NULL || ctx->pk_info != NULL )
kadonotakashi 0:8fdf9a60065b 104 return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 105
kadonotakashi 0:8fdf9a60065b 106 if( ( ctx->pk_ctx = info->ctx_alloc_func() ) == NULL )
kadonotakashi 0:8fdf9a60065b 107 return( MBEDTLS_ERR_PK_ALLOC_FAILED );
kadonotakashi 0:8fdf9a60065b 108
kadonotakashi 0:8fdf9a60065b 109 ctx->pk_info = info;
kadonotakashi 0:8fdf9a60065b 110
kadonotakashi 0:8fdf9a60065b 111 return( 0 );
kadonotakashi 0:8fdf9a60065b 112 }
kadonotakashi 0:8fdf9a60065b 113
kadonotakashi 0:8fdf9a60065b 114 #if defined(MBEDTLS_PK_RSA_ALT_SUPPORT)
kadonotakashi 0:8fdf9a60065b 115 /*
kadonotakashi 0:8fdf9a60065b 116 * Initialize an RSA-alt context
kadonotakashi 0:8fdf9a60065b 117 */
kadonotakashi 0:8fdf9a60065b 118 int mbedtls_pk_setup_rsa_alt( mbedtls_pk_context *ctx, void * key,
kadonotakashi 0:8fdf9a60065b 119 mbedtls_pk_rsa_alt_decrypt_func decrypt_func,
kadonotakashi 0:8fdf9a60065b 120 mbedtls_pk_rsa_alt_sign_func sign_func,
kadonotakashi 0:8fdf9a60065b 121 mbedtls_pk_rsa_alt_key_len_func key_len_func )
kadonotakashi 0:8fdf9a60065b 122 {
kadonotakashi 0:8fdf9a60065b 123 mbedtls_rsa_alt_context *rsa_alt;
kadonotakashi 0:8fdf9a60065b 124 const mbedtls_pk_info_t *info = &mbedtls_rsa_alt_info;
kadonotakashi 0:8fdf9a60065b 125
kadonotakashi 0:8fdf9a60065b 126 if( ctx == NULL || ctx->pk_info != NULL )
kadonotakashi 0:8fdf9a60065b 127 return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 128
kadonotakashi 0:8fdf9a60065b 129 if( ( ctx->pk_ctx = info->ctx_alloc_func() ) == NULL )
kadonotakashi 0:8fdf9a60065b 130 return( MBEDTLS_ERR_PK_ALLOC_FAILED );
kadonotakashi 0:8fdf9a60065b 131
kadonotakashi 0:8fdf9a60065b 132 ctx->pk_info = info;
kadonotakashi 0:8fdf9a60065b 133
kadonotakashi 0:8fdf9a60065b 134 rsa_alt = (mbedtls_rsa_alt_context *) ctx->pk_ctx;
kadonotakashi 0:8fdf9a60065b 135
kadonotakashi 0:8fdf9a60065b 136 rsa_alt->key = key;
kadonotakashi 0:8fdf9a60065b 137 rsa_alt->decrypt_func = decrypt_func;
kadonotakashi 0:8fdf9a60065b 138 rsa_alt->sign_func = sign_func;
kadonotakashi 0:8fdf9a60065b 139 rsa_alt->key_len_func = key_len_func;
kadonotakashi 0:8fdf9a60065b 140
kadonotakashi 0:8fdf9a60065b 141 return( 0 );
kadonotakashi 0:8fdf9a60065b 142 }
kadonotakashi 0:8fdf9a60065b 143 #endif /* MBEDTLS_PK_RSA_ALT_SUPPORT */
kadonotakashi 0:8fdf9a60065b 144
kadonotakashi 0:8fdf9a60065b 145 /*
kadonotakashi 0:8fdf9a60065b 146 * Tell if a PK can do the operations of the given type
kadonotakashi 0:8fdf9a60065b 147 */
kadonotakashi 0:8fdf9a60065b 148 int mbedtls_pk_can_do( const mbedtls_pk_context *ctx, mbedtls_pk_type_t type )
kadonotakashi 0:8fdf9a60065b 149 {
kadonotakashi 0:8fdf9a60065b 150 /* null or NONE context can't do anything */
kadonotakashi 0:8fdf9a60065b 151 if( ctx == NULL || ctx->pk_info == NULL )
kadonotakashi 0:8fdf9a60065b 152 return( 0 );
kadonotakashi 0:8fdf9a60065b 153
kadonotakashi 0:8fdf9a60065b 154 return( ctx->pk_info->can_do( type ) );
kadonotakashi 0:8fdf9a60065b 155 }
kadonotakashi 0:8fdf9a60065b 156
kadonotakashi 0:8fdf9a60065b 157 /*
kadonotakashi 0:8fdf9a60065b 158 * Helper for mbedtls_pk_sign and mbedtls_pk_verify
kadonotakashi 0:8fdf9a60065b 159 */
kadonotakashi 0:8fdf9a60065b 160 static inline int pk_hashlen_helper( mbedtls_md_type_t md_alg, size_t *hash_len )
kadonotakashi 0:8fdf9a60065b 161 {
kadonotakashi 0:8fdf9a60065b 162 const mbedtls_md_info_t *md_info;
kadonotakashi 0:8fdf9a60065b 163
kadonotakashi 0:8fdf9a60065b 164 if( *hash_len != 0 )
kadonotakashi 0:8fdf9a60065b 165 return( 0 );
kadonotakashi 0:8fdf9a60065b 166
kadonotakashi 0:8fdf9a60065b 167 if( ( md_info = mbedtls_md_info_from_type( md_alg ) ) == NULL )
kadonotakashi 0:8fdf9a60065b 168 return( -1 );
kadonotakashi 0:8fdf9a60065b 169
kadonotakashi 0:8fdf9a60065b 170 *hash_len = mbedtls_md_get_size( md_info );
kadonotakashi 0:8fdf9a60065b 171 return( 0 );
kadonotakashi 0:8fdf9a60065b 172 }
kadonotakashi 0:8fdf9a60065b 173
kadonotakashi 0:8fdf9a60065b 174 /*
kadonotakashi 0:8fdf9a60065b 175 * Verify a signature
kadonotakashi 0:8fdf9a60065b 176 */
kadonotakashi 0:8fdf9a60065b 177 int mbedtls_pk_verify( mbedtls_pk_context *ctx, mbedtls_md_type_t md_alg,
kadonotakashi 0:8fdf9a60065b 178 const unsigned char *hash, size_t hash_len,
kadonotakashi 0:8fdf9a60065b 179 const unsigned char *sig, size_t sig_len )
kadonotakashi 0:8fdf9a60065b 180 {
kadonotakashi 0:8fdf9a60065b 181 if( ctx == NULL || ctx->pk_info == NULL ||
kadonotakashi 0:8fdf9a60065b 182 pk_hashlen_helper( md_alg, &hash_len ) != 0 )
kadonotakashi 0:8fdf9a60065b 183 return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 184
kadonotakashi 0:8fdf9a60065b 185 if( ctx->pk_info->verify_func == NULL )
kadonotakashi 0:8fdf9a60065b 186 return( MBEDTLS_ERR_PK_TYPE_MISMATCH );
kadonotakashi 0:8fdf9a60065b 187
kadonotakashi 0:8fdf9a60065b 188 return( ctx->pk_info->verify_func( ctx->pk_ctx, md_alg, hash, hash_len,
kadonotakashi 0:8fdf9a60065b 189 sig, sig_len ) );
kadonotakashi 0:8fdf9a60065b 190 }
kadonotakashi 0:8fdf9a60065b 191
kadonotakashi 0:8fdf9a60065b 192 /*
kadonotakashi 0:8fdf9a60065b 193 * Verify a signature with options
kadonotakashi 0:8fdf9a60065b 194 */
kadonotakashi 0:8fdf9a60065b 195 int mbedtls_pk_verify_ext( mbedtls_pk_type_t type, const void *options,
kadonotakashi 0:8fdf9a60065b 196 mbedtls_pk_context *ctx, mbedtls_md_type_t md_alg,
kadonotakashi 0:8fdf9a60065b 197 const unsigned char *hash, size_t hash_len,
kadonotakashi 0:8fdf9a60065b 198 const unsigned char *sig, size_t sig_len )
kadonotakashi 0:8fdf9a60065b 199 {
kadonotakashi 0:8fdf9a60065b 200 if( ctx == NULL || ctx->pk_info == NULL )
kadonotakashi 0:8fdf9a60065b 201 return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 202
kadonotakashi 0:8fdf9a60065b 203 if( ! mbedtls_pk_can_do( ctx, type ) )
kadonotakashi 0:8fdf9a60065b 204 return( MBEDTLS_ERR_PK_TYPE_MISMATCH );
kadonotakashi 0:8fdf9a60065b 205
kadonotakashi 0:8fdf9a60065b 206 if( type == MBEDTLS_PK_RSASSA_PSS )
kadonotakashi 0:8fdf9a60065b 207 {
kadonotakashi 0:8fdf9a60065b 208 #if defined(MBEDTLS_RSA_C) && defined(MBEDTLS_PKCS1_V21)
kadonotakashi 0:8fdf9a60065b 209 int ret;
kadonotakashi 0:8fdf9a60065b 210 const mbedtls_pk_rsassa_pss_options *pss_opts;
kadonotakashi 0:8fdf9a60065b 211
kadonotakashi 0:8fdf9a60065b 212 #if SIZE_MAX > UINT_MAX
kadonotakashi 0:8fdf9a60065b 213 if( md_alg == MBEDTLS_MD_NONE && UINT_MAX < hash_len )
kadonotakashi 0:8fdf9a60065b 214 return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 215 #endif /* SIZE_MAX > UINT_MAX */
kadonotakashi 0:8fdf9a60065b 216
kadonotakashi 0:8fdf9a60065b 217 if( options == NULL )
kadonotakashi 0:8fdf9a60065b 218 return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 219
kadonotakashi 0:8fdf9a60065b 220 pss_opts = (const mbedtls_pk_rsassa_pss_options *) options;
kadonotakashi 0:8fdf9a60065b 221
kadonotakashi 0:8fdf9a60065b 222 if( sig_len < mbedtls_pk_get_len( ctx ) )
kadonotakashi 0:8fdf9a60065b 223 return( MBEDTLS_ERR_RSA_VERIFY_FAILED );
kadonotakashi 0:8fdf9a60065b 224
kadonotakashi 0:8fdf9a60065b 225 ret = mbedtls_rsa_rsassa_pss_verify_ext( mbedtls_pk_rsa( *ctx ),
kadonotakashi 0:8fdf9a60065b 226 NULL, NULL, MBEDTLS_RSA_PUBLIC,
kadonotakashi 0:8fdf9a60065b 227 md_alg, (unsigned int) hash_len, hash,
kadonotakashi 0:8fdf9a60065b 228 pss_opts->mgf1_hash_id,
kadonotakashi 0:8fdf9a60065b 229 pss_opts->expected_salt_len,
kadonotakashi 0:8fdf9a60065b 230 sig );
kadonotakashi 0:8fdf9a60065b 231 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 232 return( ret );
kadonotakashi 0:8fdf9a60065b 233
kadonotakashi 0:8fdf9a60065b 234 if( sig_len > mbedtls_pk_get_len( ctx ) )
kadonotakashi 0:8fdf9a60065b 235 return( MBEDTLS_ERR_PK_SIG_LEN_MISMATCH );
kadonotakashi 0:8fdf9a60065b 236
kadonotakashi 0:8fdf9a60065b 237 return( 0 );
kadonotakashi 0:8fdf9a60065b 238 #else
kadonotakashi 0:8fdf9a60065b 239 return( MBEDTLS_ERR_PK_FEATURE_UNAVAILABLE );
kadonotakashi 0:8fdf9a60065b 240 #endif /* MBEDTLS_RSA_C && MBEDTLS_PKCS1_V21 */
kadonotakashi 0:8fdf9a60065b 241 }
kadonotakashi 0:8fdf9a60065b 242
kadonotakashi 0:8fdf9a60065b 243 /* General case: no options */
kadonotakashi 0:8fdf9a60065b 244 if( options != NULL )
kadonotakashi 0:8fdf9a60065b 245 return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 246
kadonotakashi 0:8fdf9a60065b 247 return( mbedtls_pk_verify( ctx, md_alg, hash, hash_len, sig, sig_len ) );
kadonotakashi 0:8fdf9a60065b 248 }
kadonotakashi 0:8fdf9a60065b 249
kadonotakashi 0:8fdf9a60065b 250 /*
kadonotakashi 0:8fdf9a60065b 251 * Make a signature
kadonotakashi 0:8fdf9a60065b 252 */
kadonotakashi 0:8fdf9a60065b 253 int mbedtls_pk_sign( mbedtls_pk_context *ctx, mbedtls_md_type_t md_alg,
kadonotakashi 0:8fdf9a60065b 254 const unsigned char *hash, size_t hash_len,
kadonotakashi 0:8fdf9a60065b 255 unsigned char *sig, size_t *sig_len,
kadonotakashi 0:8fdf9a60065b 256 int (*f_rng)(void *, unsigned char *, size_t), void *p_rng )
kadonotakashi 0:8fdf9a60065b 257 {
kadonotakashi 0:8fdf9a60065b 258 if( ctx == NULL || ctx->pk_info == NULL ||
kadonotakashi 0:8fdf9a60065b 259 pk_hashlen_helper( md_alg, &hash_len ) != 0 )
kadonotakashi 0:8fdf9a60065b 260 return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 261
kadonotakashi 0:8fdf9a60065b 262 if( ctx->pk_info->sign_func == NULL )
kadonotakashi 0:8fdf9a60065b 263 return( MBEDTLS_ERR_PK_TYPE_MISMATCH );
kadonotakashi 0:8fdf9a60065b 264
kadonotakashi 0:8fdf9a60065b 265 return( ctx->pk_info->sign_func( ctx->pk_ctx, md_alg, hash, hash_len,
kadonotakashi 0:8fdf9a60065b 266 sig, sig_len, f_rng, p_rng ) );
kadonotakashi 0:8fdf9a60065b 267 }
kadonotakashi 0:8fdf9a60065b 268
kadonotakashi 0:8fdf9a60065b 269 /*
kadonotakashi 0:8fdf9a60065b 270 * Decrypt message
kadonotakashi 0:8fdf9a60065b 271 */
kadonotakashi 0:8fdf9a60065b 272 int mbedtls_pk_decrypt( mbedtls_pk_context *ctx,
kadonotakashi 0:8fdf9a60065b 273 const unsigned char *input, size_t ilen,
kadonotakashi 0:8fdf9a60065b 274 unsigned char *output, size_t *olen, size_t osize,
kadonotakashi 0:8fdf9a60065b 275 int (*f_rng)(void *, unsigned char *, size_t), void *p_rng )
kadonotakashi 0:8fdf9a60065b 276 {
kadonotakashi 0:8fdf9a60065b 277 if( ctx == NULL || ctx->pk_info == NULL )
kadonotakashi 0:8fdf9a60065b 278 return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 279
kadonotakashi 0:8fdf9a60065b 280 if( ctx->pk_info->decrypt_func == NULL )
kadonotakashi 0:8fdf9a60065b 281 return( MBEDTLS_ERR_PK_TYPE_MISMATCH );
kadonotakashi 0:8fdf9a60065b 282
kadonotakashi 0:8fdf9a60065b 283 return( ctx->pk_info->decrypt_func( ctx->pk_ctx, input, ilen,
kadonotakashi 0:8fdf9a60065b 284 output, olen, osize, f_rng, p_rng ) );
kadonotakashi 0:8fdf9a60065b 285 }
kadonotakashi 0:8fdf9a60065b 286
kadonotakashi 0:8fdf9a60065b 287 /*
kadonotakashi 0:8fdf9a60065b 288 * Encrypt message
kadonotakashi 0:8fdf9a60065b 289 */
kadonotakashi 0:8fdf9a60065b 290 int mbedtls_pk_encrypt( mbedtls_pk_context *ctx,
kadonotakashi 0:8fdf9a60065b 291 const unsigned char *input, size_t ilen,
kadonotakashi 0:8fdf9a60065b 292 unsigned char *output, size_t *olen, size_t osize,
kadonotakashi 0:8fdf9a60065b 293 int (*f_rng)(void *, unsigned char *, size_t), void *p_rng )
kadonotakashi 0:8fdf9a60065b 294 {
kadonotakashi 0:8fdf9a60065b 295 if( ctx == NULL || ctx->pk_info == NULL )
kadonotakashi 0:8fdf9a60065b 296 return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 297
kadonotakashi 0:8fdf9a60065b 298 if( ctx->pk_info->encrypt_func == NULL )
kadonotakashi 0:8fdf9a60065b 299 return( MBEDTLS_ERR_PK_TYPE_MISMATCH );
kadonotakashi 0:8fdf9a60065b 300
kadonotakashi 0:8fdf9a60065b 301 return( ctx->pk_info->encrypt_func( ctx->pk_ctx, input, ilen,
kadonotakashi 0:8fdf9a60065b 302 output, olen, osize, f_rng, p_rng ) );
kadonotakashi 0:8fdf9a60065b 303 }
kadonotakashi 0:8fdf9a60065b 304
kadonotakashi 0:8fdf9a60065b 305 /*
kadonotakashi 0:8fdf9a60065b 306 * Check public-private key pair
kadonotakashi 0:8fdf9a60065b 307 */
kadonotakashi 0:8fdf9a60065b 308 int mbedtls_pk_check_pair( const mbedtls_pk_context *pub, const mbedtls_pk_context *prv )
kadonotakashi 0:8fdf9a60065b 309 {
kadonotakashi 0:8fdf9a60065b 310 if( pub == NULL || pub->pk_info == NULL ||
kadonotakashi 0:8fdf9a60065b 311 prv == NULL || prv->pk_info == NULL ||
kadonotakashi 0:8fdf9a60065b 312 prv->pk_info->check_pair_func == NULL )
kadonotakashi 0:8fdf9a60065b 313 {
kadonotakashi 0:8fdf9a60065b 314 return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 315 }
kadonotakashi 0:8fdf9a60065b 316
kadonotakashi 0:8fdf9a60065b 317 if( prv->pk_info->type == MBEDTLS_PK_RSA_ALT )
kadonotakashi 0:8fdf9a60065b 318 {
kadonotakashi 0:8fdf9a60065b 319 if( pub->pk_info->type != MBEDTLS_PK_RSA )
kadonotakashi 0:8fdf9a60065b 320 return( MBEDTLS_ERR_PK_TYPE_MISMATCH );
kadonotakashi 0:8fdf9a60065b 321 }
kadonotakashi 0:8fdf9a60065b 322 else
kadonotakashi 0:8fdf9a60065b 323 {
kadonotakashi 0:8fdf9a60065b 324 if( pub->pk_info != prv->pk_info )
kadonotakashi 0:8fdf9a60065b 325 return( MBEDTLS_ERR_PK_TYPE_MISMATCH );
kadonotakashi 0:8fdf9a60065b 326 }
kadonotakashi 0:8fdf9a60065b 327
kadonotakashi 0:8fdf9a60065b 328 return( prv->pk_info->check_pair_func( pub->pk_ctx, prv->pk_ctx ) );
kadonotakashi 0:8fdf9a60065b 329 }
kadonotakashi 0:8fdf9a60065b 330
kadonotakashi 0:8fdf9a60065b 331 /*
kadonotakashi 0:8fdf9a60065b 332 * Get key size in bits
kadonotakashi 0:8fdf9a60065b 333 */
kadonotakashi 0:8fdf9a60065b 334 size_t mbedtls_pk_get_bitlen( const mbedtls_pk_context *ctx )
kadonotakashi 0:8fdf9a60065b 335 {
kadonotakashi 0:8fdf9a60065b 336 if( ctx == NULL || ctx->pk_info == NULL )
kadonotakashi 0:8fdf9a60065b 337 return( 0 );
kadonotakashi 0:8fdf9a60065b 338
kadonotakashi 0:8fdf9a60065b 339 return( ctx->pk_info->get_bitlen( ctx->pk_ctx ) );
kadonotakashi 0:8fdf9a60065b 340 }
kadonotakashi 0:8fdf9a60065b 341
kadonotakashi 0:8fdf9a60065b 342 /*
kadonotakashi 0:8fdf9a60065b 343 * Export debug information
kadonotakashi 0:8fdf9a60065b 344 */
kadonotakashi 0:8fdf9a60065b 345 int mbedtls_pk_debug( const mbedtls_pk_context *ctx, mbedtls_pk_debug_item *items )
kadonotakashi 0:8fdf9a60065b 346 {
kadonotakashi 0:8fdf9a60065b 347 if( ctx == NULL || ctx->pk_info == NULL )
kadonotakashi 0:8fdf9a60065b 348 return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 349
kadonotakashi 0:8fdf9a60065b 350 if( ctx->pk_info->debug_func == NULL )
kadonotakashi 0:8fdf9a60065b 351 return( MBEDTLS_ERR_PK_TYPE_MISMATCH );
kadonotakashi 0:8fdf9a60065b 352
kadonotakashi 0:8fdf9a60065b 353 ctx->pk_info->debug_func( ctx->pk_ctx, items );
kadonotakashi 0:8fdf9a60065b 354 return( 0 );
kadonotakashi 0:8fdf9a60065b 355 }
kadonotakashi 0:8fdf9a60065b 356
kadonotakashi 0:8fdf9a60065b 357 /*
kadonotakashi 0:8fdf9a60065b 358 * Access the PK type name
kadonotakashi 0:8fdf9a60065b 359 */
kadonotakashi 0:8fdf9a60065b 360 const char *mbedtls_pk_get_name( const mbedtls_pk_context *ctx )
kadonotakashi 0:8fdf9a60065b 361 {
kadonotakashi 0:8fdf9a60065b 362 if( ctx == NULL || ctx->pk_info == NULL )
kadonotakashi 0:8fdf9a60065b 363 return( "invalid PK" );
kadonotakashi 0:8fdf9a60065b 364
kadonotakashi 0:8fdf9a60065b 365 return( ctx->pk_info->name );
kadonotakashi 0:8fdf9a60065b 366 }
kadonotakashi 0:8fdf9a60065b 367
kadonotakashi 0:8fdf9a60065b 368 /*
kadonotakashi 0:8fdf9a60065b 369 * Access the PK type
kadonotakashi 0:8fdf9a60065b 370 */
kadonotakashi 0:8fdf9a60065b 371 mbedtls_pk_type_t mbedtls_pk_get_type( const mbedtls_pk_context *ctx )
kadonotakashi 0:8fdf9a60065b 372 {
kadonotakashi 0:8fdf9a60065b 373 if( ctx == NULL || ctx->pk_info == NULL )
kadonotakashi 0:8fdf9a60065b 374 return( MBEDTLS_PK_NONE );
kadonotakashi 0:8fdf9a60065b 375
kadonotakashi 0:8fdf9a60065b 376 return( ctx->pk_info->type );
kadonotakashi 0:8fdf9a60065b 377 }
kadonotakashi 0:8fdf9a60065b 378
kadonotakashi 0:8fdf9a60065b 379 #endif /* MBEDTLS_PK_C */