Marco Zecchini / Mbed OS Example_RTOS

Rtos API example

mbed-os/features/mbedtls/src/pk.c@0:9fca2b23d0ba, 2019-02-23 (annotated)

Committer:
marcozecchini
Date:
Sat Feb 23 12:13:36 2019 +0000
Revision:
0:9fca2b23d0ba
final commit

Who changed what in which revision?

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