Austin Blackstone / Mbed 2 deprecated client-lwip-lightswitch-demo-expanded

mbed client lightswitch demo

Dependencies:   mbed Socket lwip-eth lwip-sys lwip

Fork of mbed-client-classic-example-lwip by Austin Blackstone

mbedtls/source/cipher.c@11:cada08fc8a70, 2016-06-09 (annotated)

Committer:
mbedAustin
Date:
Thu Jun 09 17:08:36 2016 +0000
Revision:
11:cada08fc8a70
Commit for public Consumption

Who changed what in which revision?

UserRevisionLine numberNew contents of line
mbedAustin 11:cada08fc8a70 1 /**
mbedAustin 11:cada08fc8a70 2 * \file cipher.c
mbedAustin 11:cada08fc8a70 3 *
mbedAustin 11:cada08fc8a70 4 * \brief Generic cipher wrapper for mbed TLS
mbedAustin 11:cada08fc8a70 5 *
mbedAustin 11:cada08fc8a70 6 * \author Adriaan de Jong <dejong@fox-it.com>
mbedAustin 11:cada08fc8a70 7 *
mbedAustin 11:cada08fc8a70 8 * Copyright (C) 2006-2015, ARM Limited, All Rights Reserved
mbedAustin 11:cada08fc8a70 9 * SPDX-License-Identifier: Apache-2.0
mbedAustin 11:cada08fc8a70 10 *
mbedAustin 11:cada08fc8a70 11 * Licensed under the Apache License, Version 2.0 (the "License"); you may
mbedAustin 11:cada08fc8a70 12 * not use this file except in compliance with the License.
mbedAustin 11:cada08fc8a70 13 * You may obtain a copy of the License at
mbedAustin 11:cada08fc8a70 14 *
mbedAustin 11:cada08fc8a70 15 * http://www.apache.org/licenses/LICENSE-2.0
mbedAustin 11:cada08fc8a70 16 *
mbedAustin 11:cada08fc8a70 17 * Unless required by applicable law or agreed to in writing, software
mbedAustin 11:cada08fc8a70 18 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
mbedAustin 11:cada08fc8a70 19 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
mbedAustin 11:cada08fc8a70 20 * See the License for the specific language governing permissions and
mbedAustin 11:cada08fc8a70 21 * limitations under the License.
mbedAustin 11:cada08fc8a70 22 *
mbedAustin 11:cada08fc8a70 23 * This file is part of mbed TLS (https://tls.mbed.org)
mbedAustin 11:cada08fc8a70 24 */
mbedAustin 11:cada08fc8a70 25
mbedAustin 11:cada08fc8a70 26 #if !defined(MBEDTLS_CONFIG_FILE)
mbedAustin 11:cada08fc8a70 27 #include "mbedtls/config.h"
mbedAustin 11:cada08fc8a70 28 #else
mbedAustin 11:cada08fc8a70 29 #include MBEDTLS_CONFIG_FILE
mbedAustin 11:cada08fc8a70 30 #endif
mbedAustin 11:cada08fc8a70 31
mbedAustin 11:cada08fc8a70 32 #if defined(MBEDTLS_CIPHER_C)
mbedAustin 11:cada08fc8a70 33
mbedAustin 11:cada08fc8a70 34 #include "mbedtls/cipher.h"
mbedAustin 11:cada08fc8a70 35 #include "mbedtls/cipher_internal.h"
mbedAustin 11:cada08fc8a70 36
mbedAustin 11:cada08fc8a70 37 #include <stdlib.h>
mbedAustin 11:cada08fc8a70 38 #include <string.h>
mbedAustin 11:cada08fc8a70 39
mbedAustin 11:cada08fc8a70 40 #if defined(MBEDTLS_GCM_C)
mbedAustin 11:cada08fc8a70 41 #include "mbedtls/gcm.h"
mbedAustin 11:cada08fc8a70 42 #endif
mbedAustin 11:cada08fc8a70 43
mbedAustin 11:cada08fc8a70 44 #if defined(MBEDTLS_CCM_C)
mbedAustin 11:cada08fc8a70 45 #include "mbedtls/ccm.h"
mbedAustin 11:cada08fc8a70 46 #endif
mbedAustin 11:cada08fc8a70 47
mbedAustin 11:cada08fc8a70 48 #if defined(MBEDTLS_ARC4_C) || defined(MBEDTLS_CIPHER_NULL_CIPHER)
mbedAustin 11:cada08fc8a70 49 #define MBEDTLS_CIPHER_MODE_STREAM
mbedAustin 11:cada08fc8a70 50 #endif
mbedAustin 11:cada08fc8a70 51
mbedAustin 11:cada08fc8a70 52 /* Implementation that should never be optimized out by the compiler */
mbedAustin 11:cada08fc8a70 53 static void mbedtls_zeroize( void *v, size_t n ) {
mbedAustin 11:cada08fc8a70 54 volatile unsigned char *p = v; while( n-- ) *p++ = 0;
mbedAustin 11:cada08fc8a70 55 }
mbedAustin 11:cada08fc8a70 56
mbedAustin 11:cada08fc8a70 57 static int supported_init = 0;
mbedAustin 11:cada08fc8a70 58
mbedAustin 11:cada08fc8a70 59 const int *mbedtls_cipher_list( void )
mbedAustin 11:cada08fc8a70 60 {
mbedAustin 11:cada08fc8a70 61 const mbedtls_cipher_definition_t *def;
mbedAustin 11:cada08fc8a70 62 int *type;
mbedAustin 11:cada08fc8a70 63
mbedAustin 11:cada08fc8a70 64 if( ! supported_init )
mbedAustin 11:cada08fc8a70 65 {
mbedAustin 11:cada08fc8a70 66 def = mbedtls_cipher_definitions;
mbedAustin 11:cada08fc8a70 67 type = mbedtls_cipher_supported;
mbedAustin 11:cada08fc8a70 68
mbedAustin 11:cada08fc8a70 69 while( def->type != 0 )
mbedAustin 11:cada08fc8a70 70 *type++ = (*def++).type;
mbedAustin 11:cada08fc8a70 71
mbedAustin 11:cada08fc8a70 72 *type = 0;
mbedAustin 11:cada08fc8a70 73
mbedAustin 11:cada08fc8a70 74 supported_init = 1;
mbedAustin 11:cada08fc8a70 75 }
mbedAustin 11:cada08fc8a70 76
mbedAustin 11:cada08fc8a70 77 return( mbedtls_cipher_supported );
mbedAustin 11:cada08fc8a70 78 }
mbedAustin 11:cada08fc8a70 79
mbedAustin 11:cada08fc8a70 80 const mbedtls_cipher_info_t *mbedtls_cipher_info_from_type( const mbedtls_cipher_type_t cipher_type )
mbedAustin 11:cada08fc8a70 81 {
mbedAustin 11:cada08fc8a70 82 const mbedtls_cipher_definition_t *def;
mbedAustin 11:cada08fc8a70 83
mbedAustin 11:cada08fc8a70 84 for( def = mbedtls_cipher_definitions; def->info != NULL; def++ )
mbedAustin 11:cada08fc8a70 85 if( def->type == cipher_type )
mbedAustin 11:cada08fc8a70 86 return( def->info );
mbedAustin 11:cada08fc8a70 87
mbedAustin 11:cada08fc8a70 88 return( NULL );
mbedAustin 11:cada08fc8a70 89 }
mbedAustin 11:cada08fc8a70 90
mbedAustin 11:cada08fc8a70 91 const mbedtls_cipher_info_t *mbedtls_cipher_info_from_string( const char *cipher_name )
mbedAustin 11:cada08fc8a70 92 {
mbedAustin 11:cada08fc8a70 93 const mbedtls_cipher_definition_t *def;
mbedAustin 11:cada08fc8a70 94
mbedAustin 11:cada08fc8a70 95 if( NULL == cipher_name )
mbedAustin 11:cada08fc8a70 96 return( NULL );
mbedAustin 11:cada08fc8a70 97
mbedAustin 11:cada08fc8a70 98 for( def = mbedtls_cipher_definitions; def->info != NULL; def++ )
mbedAustin 11:cada08fc8a70 99 if( ! strcmp( def->info->name, cipher_name ) )
mbedAustin 11:cada08fc8a70 100 return( def->info );
mbedAustin 11:cada08fc8a70 101
mbedAustin 11:cada08fc8a70 102 return( NULL );
mbedAustin 11:cada08fc8a70 103 }
mbedAustin 11:cada08fc8a70 104
mbedAustin 11:cada08fc8a70 105 const mbedtls_cipher_info_t *mbedtls_cipher_info_from_values( const mbedtls_cipher_id_t cipher_id,
mbedAustin 11:cada08fc8a70 106 int key_bitlen,
mbedAustin 11:cada08fc8a70 107 const mbedtls_cipher_mode_t mode )
mbedAustin 11:cada08fc8a70 108 {
mbedAustin 11:cada08fc8a70 109 const mbedtls_cipher_definition_t *def;
mbedAustin 11:cada08fc8a70 110
mbedAustin 11:cada08fc8a70 111 for( def = mbedtls_cipher_definitions; def->info != NULL; def++ )
mbedAustin 11:cada08fc8a70 112 if( def->info->base->cipher == cipher_id &&
mbedAustin 11:cada08fc8a70 113 def->info->key_bitlen == (unsigned) key_bitlen &&
mbedAustin 11:cada08fc8a70 114 def->info->mode == mode )
mbedAustin 11:cada08fc8a70 115 return( def->info );
mbedAustin 11:cada08fc8a70 116
mbedAustin 11:cada08fc8a70 117 return( NULL );
mbedAustin 11:cada08fc8a70 118 }
mbedAustin 11:cada08fc8a70 119
mbedAustin 11:cada08fc8a70 120 void mbedtls_cipher_init( mbedtls_cipher_context_t *ctx )
mbedAustin 11:cada08fc8a70 121 {
mbedAustin 11:cada08fc8a70 122 memset( ctx, 0, sizeof( mbedtls_cipher_context_t ) );
mbedAustin 11:cada08fc8a70 123 }
mbedAustin 11:cada08fc8a70 124
mbedAustin 11:cada08fc8a70 125 void mbedtls_cipher_free( mbedtls_cipher_context_t *ctx )
mbedAustin 11:cada08fc8a70 126 {
mbedAustin 11:cada08fc8a70 127 if( ctx == NULL )
mbedAustin 11:cada08fc8a70 128 return;
mbedAustin 11:cada08fc8a70 129
mbedAustin 11:cada08fc8a70 130 if( ctx->cipher_ctx )
mbedAustin 11:cada08fc8a70 131 ctx->cipher_info->base->ctx_free_func( ctx->cipher_ctx );
mbedAustin 11:cada08fc8a70 132
mbedAustin 11:cada08fc8a70 133 mbedtls_zeroize( ctx, sizeof(mbedtls_cipher_context_t) );
mbedAustin 11:cada08fc8a70 134 }
mbedAustin 11:cada08fc8a70 135
mbedAustin 11:cada08fc8a70 136 int mbedtls_cipher_setup( mbedtls_cipher_context_t *ctx, const mbedtls_cipher_info_t *cipher_info )
mbedAustin 11:cada08fc8a70 137 {
mbedAustin 11:cada08fc8a70 138 if( NULL == cipher_info || NULL == ctx )
mbedAustin 11:cada08fc8a70 139 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 140
mbedAustin 11:cada08fc8a70 141 memset( ctx, 0, sizeof( mbedtls_cipher_context_t ) );
mbedAustin 11:cada08fc8a70 142
mbedAustin 11:cada08fc8a70 143 if( NULL == ( ctx->cipher_ctx = cipher_info->base->ctx_alloc_func() ) )
mbedAustin 11:cada08fc8a70 144 return( MBEDTLS_ERR_CIPHER_ALLOC_FAILED );
mbedAustin 11:cada08fc8a70 145
mbedAustin 11:cada08fc8a70 146 ctx->cipher_info = cipher_info;
mbedAustin 11:cada08fc8a70 147
mbedAustin 11:cada08fc8a70 148 #if defined(MBEDTLS_CIPHER_MODE_WITH_PADDING)
mbedAustin 11:cada08fc8a70 149 /*
mbedAustin 11:cada08fc8a70 150 * Ignore possible errors caused by a cipher mode that doesn't use padding
mbedAustin 11:cada08fc8a70 151 */
mbedAustin 11:cada08fc8a70 152 #if defined(MBEDTLS_CIPHER_PADDING_PKCS7)
mbedAustin 11:cada08fc8a70 153 (void) mbedtls_cipher_set_padding_mode( ctx, MBEDTLS_PADDING_PKCS7 );
mbedAustin 11:cada08fc8a70 154 #else
mbedAustin 11:cada08fc8a70 155 (void) mbedtls_cipher_set_padding_mode( ctx, MBEDTLS_PADDING_NONE );
mbedAustin 11:cada08fc8a70 156 #endif
mbedAustin 11:cada08fc8a70 157 #endif /* MBEDTLS_CIPHER_MODE_WITH_PADDING */
mbedAustin 11:cada08fc8a70 158
mbedAustin 11:cada08fc8a70 159 return( 0 );
mbedAustin 11:cada08fc8a70 160 }
mbedAustin 11:cada08fc8a70 161
mbedAustin 11:cada08fc8a70 162 int mbedtls_cipher_setkey( mbedtls_cipher_context_t *ctx, const unsigned char *key,
mbedAustin 11:cada08fc8a70 163 int key_bitlen, const mbedtls_operation_t operation )
mbedAustin 11:cada08fc8a70 164 {
mbedAustin 11:cada08fc8a70 165 if( NULL == ctx || NULL == ctx->cipher_info )
mbedAustin 11:cada08fc8a70 166 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 167
mbedAustin 11:cada08fc8a70 168 if( ( ctx->cipher_info->flags & MBEDTLS_CIPHER_VARIABLE_KEY_LEN ) == 0 &&
mbedAustin 11:cada08fc8a70 169 (int) ctx->cipher_info->key_bitlen != key_bitlen )
mbedAustin 11:cada08fc8a70 170 {
mbedAustin 11:cada08fc8a70 171 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 172 }
mbedAustin 11:cada08fc8a70 173
mbedAustin 11:cada08fc8a70 174 ctx->key_bitlen = key_bitlen;
mbedAustin 11:cada08fc8a70 175 ctx->operation = operation;
mbedAustin 11:cada08fc8a70 176
mbedAustin 11:cada08fc8a70 177 /*
mbedAustin 11:cada08fc8a70 178 * For CFB and CTR mode always use the encryption key schedule
mbedAustin 11:cada08fc8a70 179 */
mbedAustin 11:cada08fc8a70 180 if( MBEDTLS_ENCRYPT == operation ||
mbedAustin 11:cada08fc8a70 181 MBEDTLS_MODE_CFB == ctx->cipher_info->mode ||
mbedAustin 11:cada08fc8a70 182 MBEDTLS_MODE_CTR == ctx->cipher_info->mode )
mbedAustin 11:cada08fc8a70 183 {
mbedAustin 11:cada08fc8a70 184 return ctx->cipher_info->base->setkey_enc_func( ctx->cipher_ctx, key,
mbedAustin 11:cada08fc8a70 185 ctx->key_bitlen );
mbedAustin 11:cada08fc8a70 186 }
mbedAustin 11:cada08fc8a70 187
mbedAustin 11:cada08fc8a70 188 if( MBEDTLS_DECRYPT == operation )
mbedAustin 11:cada08fc8a70 189 return ctx->cipher_info->base->setkey_dec_func( ctx->cipher_ctx, key,
mbedAustin 11:cada08fc8a70 190 ctx->key_bitlen );
mbedAustin 11:cada08fc8a70 191
mbedAustin 11:cada08fc8a70 192 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 193 }
mbedAustin 11:cada08fc8a70 194
mbedAustin 11:cada08fc8a70 195 int mbedtls_cipher_set_iv( mbedtls_cipher_context_t *ctx,
mbedAustin 11:cada08fc8a70 196 const unsigned char *iv, size_t iv_len )
mbedAustin 11:cada08fc8a70 197 {
mbedAustin 11:cada08fc8a70 198 size_t actual_iv_size;
mbedAustin 11:cada08fc8a70 199
mbedAustin 11:cada08fc8a70 200 if( NULL == ctx || NULL == ctx->cipher_info || NULL == iv )
mbedAustin 11:cada08fc8a70 201 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 202
mbedAustin 11:cada08fc8a70 203 /* avoid buffer overflow in ctx->iv */
mbedAustin 11:cada08fc8a70 204 if( iv_len > MBEDTLS_MAX_IV_LENGTH )
mbedAustin 11:cada08fc8a70 205 return( MBEDTLS_ERR_CIPHER_FEATURE_UNAVAILABLE );
mbedAustin 11:cada08fc8a70 206
mbedAustin 11:cada08fc8a70 207 if( ( ctx->cipher_info->flags & MBEDTLS_CIPHER_VARIABLE_IV_LEN ) != 0 )
mbedAustin 11:cada08fc8a70 208 actual_iv_size = iv_len;
mbedAustin 11:cada08fc8a70 209 else
mbedAustin 11:cada08fc8a70 210 {
mbedAustin 11:cada08fc8a70 211 actual_iv_size = ctx->cipher_info->iv_size;
mbedAustin 11:cada08fc8a70 212
mbedAustin 11:cada08fc8a70 213 /* avoid reading past the end of input buffer */
mbedAustin 11:cada08fc8a70 214 if( actual_iv_size > iv_len )
mbedAustin 11:cada08fc8a70 215 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 216 }
mbedAustin 11:cada08fc8a70 217
mbedAustin 11:cada08fc8a70 218 memcpy( ctx->iv, iv, actual_iv_size );
mbedAustin 11:cada08fc8a70 219 ctx->iv_size = actual_iv_size;
mbedAustin 11:cada08fc8a70 220
mbedAustin 11:cada08fc8a70 221 return( 0 );
mbedAustin 11:cada08fc8a70 222 }
mbedAustin 11:cada08fc8a70 223
mbedAustin 11:cada08fc8a70 224 int mbedtls_cipher_reset( mbedtls_cipher_context_t *ctx )
mbedAustin 11:cada08fc8a70 225 {
mbedAustin 11:cada08fc8a70 226 if( NULL == ctx || NULL == ctx->cipher_info )
mbedAustin 11:cada08fc8a70 227 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 228
mbedAustin 11:cada08fc8a70 229 ctx->unprocessed_len = 0;
mbedAustin 11:cada08fc8a70 230
mbedAustin 11:cada08fc8a70 231 return( 0 );
mbedAustin 11:cada08fc8a70 232 }
mbedAustin 11:cada08fc8a70 233
mbedAustin 11:cada08fc8a70 234 #if defined(MBEDTLS_GCM_C)
mbedAustin 11:cada08fc8a70 235 int mbedtls_cipher_update_ad( mbedtls_cipher_context_t *ctx,
mbedAustin 11:cada08fc8a70 236 const unsigned char *ad, size_t ad_len )
mbedAustin 11:cada08fc8a70 237 {
mbedAustin 11:cada08fc8a70 238 if( NULL == ctx || NULL == ctx->cipher_info )
mbedAustin 11:cada08fc8a70 239 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 240
mbedAustin 11:cada08fc8a70 241 if( MBEDTLS_MODE_GCM == ctx->cipher_info->mode )
mbedAustin 11:cada08fc8a70 242 {
mbedAustin 11:cada08fc8a70 243 return mbedtls_gcm_starts( (mbedtls_gcm_context *) ctx->cipher_ctx, ctx->operation,
mbedAustin 11:cada08fc8a70 244 ctx->iv, ctx->iv_size, ad, ad_len );
mbedAustin 11:cada08fc8a70 245 }
mbedAustin 11:cada08fc8a70 246
mbedAustin 11:cada08fc8a70 247 return( 0 );
mbedAustin 11:cada08fc8a70 248 }
mbedAustin 11:cada08fc8a70 249 #endif /* MBEDTLS_GCM_C */
mbedAustin 11:cada08fc8a70 250
mbedAustin 11:cada08fc8a70 251 int mbedtls_cipher_update( mbedtls_cipher_context_t *ctx, const unsigned char *input,
mbedAustin 11:cada08fc8a70 252 size_t ilen, unsigned char *output, size_t *olen )
mbedAustin 11:cada08fc8a70 253 {
mbedAustin 11:cada08fc8a70 254 int ret;
mbedAustin 11:cada08fc8a70 255
mbedAustin 11:cada08fc8a70 256 if( NULL == ctx || NULL == ctx->cipher_info || NULL == olen )
mbedAustin 11:cada08fc8a70 257 {
mbedAustin 11:cada08fc8a70 258 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 259 }
mbedAustin 11:cada08fc8a70 260
mbedAustin 11:cada08fc8a70 261 *olen = 0;
mbedAustin 11:cada08fc8a70 262
mbedAustin 11:cada08fc8a70 263 if( ctx->cipher_info->mode == MBEDTLS_MODE_ECB )
mbedAustin 11:cada08fc8a70 264 {
mbedAustin 11:cada08fc8a70 265 if( ilen != mbedtls_cipher_get_block_size( ctx ) )
mbedAustin 11:cada08fc8a70 266 return( MBEDTLS_ERR_CIPHER_FULL_BLOCK_EXPECTED );
mbedAustin 11:cada08fc8a70 267
mbedAustin 11:cada08fc8a70 268 *olen = ilen;
mbedAustin 11:cada08fc8a70 269
mbedAustin 11:cada08fc8a70 270 if( 0 != ( ret = ctx->cipher_info->base->ecb_func( ctx->cipher_ctx,
mbedAustin 11:cada08fc8a70 271 ctx->operation, input, output ) ) )
mbedAustin 11:cada08fc8a70 272 {
mbedAustin 11:cada08fc8a70 273 return( ret );
mbedAustin 11:cada08fc8a70 274 }
mbedAustin 11:cada08fc8a70 275
mbedAustin 11:cada08fc8a70 276 return( 0 );
mbedAustin 11:cada08fc8a70 277 }
mbedAustin 11:cada08fc8a70 278
mbedAustin 11:cada08fc8a70 279 #if defined(MBEDTLS_GCM_C)
mbedAustin 11:cada08fc8a70 280 if( ctx->cipher_info->mode == MBEDTLS_MODE_GCM )
mbedAustin 11:cada08fc8a70 281 {
mbedAustin 11:cada08fc8a70 282 *olen = ilen;
mbedAustin 11:cada08fc8a70 283 return mbedtls_gcm_update( (mbedtls_gcm_context *) ctx->cipher_ctx, ilen, input,
mbedAustin 11:cada08fc8a70 284 output );
mbedAustin 11:cada08fc8a70 285 }
mbedAustin 11:cada08fc8a70 286 #endif
mbedAustin 11:cada08fc8a70 287
mbedAustin 11:cada08fc8a70 288 if( input == output &&
mbedAustin 11:cada08fc8a70 289 ( ctx->unprocessed_len != 0 || ilen % mbedtls_cipher_get_block_size( ctx ) ) )
mbedAustin 11:cada08fc8a70 290 {
mbedAustin 11:cada08fc8a70 291 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 292 }
mbedAustin 11:cada08fc8a70 293
mbedAustin 11:cada08fc8a70 294 #if defined(MBEDTLS_CIPHER_MODE_CBC)
mbedAustin 11:cada08fc8a70 295 if( ctx->cipher_info->mode == MBEDTLS_MODE_CBC )
mbedAustin 11:cada08fc8a70 296 {
mbedAustin 11:cada08fc8a70 297 size_t copy_len = 0;
mbedAustin 11:cada08fc8a70 298
mbedAustin 11:cada08fc8a70 299 /*
mbedAustin 11:cada08fc8a70 300 * If there is not enough data for a full block, cache it.
mbedAustin 11:cada08fc8a70 301 */
mbedAustin 11:cada08fc8a70 302 if( ( ctx->operation == MBEDTLS_DECRYPT &&
mbedAustin 11:cada08fc8a70 303 ilen + ctx->unprocessed_len <= mbedtls_cipher_get_block_size( ctx ) ) ||
mbedAustin 11:cada08fc8a70 304 ( ctx->operation == MBEDTLS_ENCRYPT &&
mbedAustin 11:cada08fc8a70 305 ilen + ctx->unprocessed_len < mbedtls_cipher_get_block_size( ctx ) ) )
mbedAustin 11:cada08fc8a70 306 {
mbedAustin 11:cada08fc8a70 307 memcpy( &( ctx->unprocessed_data[ctx->unprocessed_len] ), input,
mbedAustin 11:cada08fc8a70 308 ilen );
mbedAustin 11:cada08fc8a70 309
mbedAustin 11:cada08fc8a70 310 ctx->unprocessed_len += ilen;
mbedAustin 11:cada08fc8a70 311 return( 0 );
mbedAustin 11:cada08fc8a70 312 }
mbedAustin 11:cada08fc8a70 313
mbedAustin 11:cada08fc8a70 314 /*
mbedAustin 11:cada08fc8a70 315 * Process cached data first
mbedAustin 11:cada08fc8a70 316 */
mbedAustin 11:cada08fc8a70 317 if( ctx->unprocessed_len != 0 )
mbedAustin 11:cada08fc8a70 318 {
mbedAustin 11:cada08fc8a70 319 copy_len = mbedtls_cipher_get_block_size( ctx ) - ctx->unprocessed_len;
mbedAustin 11:cada08fc8a70 320
mbedAustin 11:cada08fc8a70 321 memcpy( &( ctx->unprocessed_data[ctx->unprocessed_len] ), input,
mbedAustin 11:cada08fc8a70 322 copy_len );
mbedAustin 11:cada08fc8a70 323
mbedAustin 11:cada08fc8a70 324 if( 0 != ( ret = ctx->cipher_info->base->cbc_func( ctx->cipher_ctx,
mbedAustin 11:cada08fc8a70 325 ctx->operation, mbedtls_cipher_get_block_size( ctx ), ctx->iv,
mbedAustin 11:cada08fc8a70 326 ctx->unprocessed_data, output ) ) )
mbedAustin 11:cada08fc8a70 327 {
mbedAustin 11:cada08fc8a70 328 return( ret );
mbedAustin 11:cada08fc8a70 329 }
mbedAustin 11:cada08fc8a70 330
mbedAustin 11:cada08fc8a70 331 *olen += mbedtls_cipher_get_block_size( ctx );
mbedAustin 11:cada08fc8a70 332 output += mbedtls_cipher_get_block_size( ctx );
mbedAustin 11:cada08fc8a70 333 ctx->unprocessed_len = 0;
mbedAustin 11:cada08fc8a70 334
mbedAustin 11:cada08fc8a70 335 input += copy_len;
mbedAustin 11:cada08fc8a70 336 ilen -= copy_len;
mbedAustin 11:cada08fc8a70 337 }
mbedAustin 11:cada08fc8a70 338
mbedAustin 11:cada08fc8a70 339 /*
mbedAustin 11:cada08fc8a70 340 * Cache final, incomplete block
mbedAustin 11:cada08fc8a70 341 */
mbedAustin 11:cada08fc8a70 342 if( 0 != ilen )
mbedAustin 11:cada08fc8a70 343 {
mbedAustin 11:cada08fc8a70 344 copy_len = ilen % mbedtls_cipher_get_block_size( ctx );
mbedAustin 11:cada08fc8a70 345 if( copy_len == 0 && ctx->operation == MBEDTLS_DECRYPT )
mbedAustin 11:cada08fc8a70 346 copy_len = mbedtls_cipher_get_block_size( ctx );
mbedAustin 11:cada08fc8a70 347
mbedAustin 11:cada08fc8a70 348 memcpy( ctx->unprocessed_data, &( input[ilen - copy_len] ),
mbedAustin 11:cada08fc8a70 349 copy_len );
mbedAustin 11:cada08fc8a70 350
mbedAustin 11:cada08fc8a70 351 ctx->unprocessed_len += copy_len;
mbedAustin 11:cada08fc8a70 352 ilen -= copy_len;
mbedAustin 11:cada08fc8a70 353 }
mbedAustin 11:cada08fc8a70 354
mbedAustin 11:cada08fc8a70 355 /*
mbedAustin 11:cada08fc8a70 356 * Process remaining full blocks
mbedAustin 11:cada08fc8a70 357 */
mbedAustin 11:cada08fc8a70 358 if( ilen )
mbedAustin 11:cada08fc8a70 359 {
mbedAustin 11:cada08fc8a70 360 if( 0 != ( ret = ctx->cipher_info->base->cbc_func( ctx->cipher_ctx,
mbedAustin 11:cada08fc8a70 361 ctx->operation, ilen, ctx->iv, input, output ) ) )
mbedAustin 11:cada08fc8a70 362 {
mbedAustin 11:cada08fc8a70 363 return( ret );
mbedAustin 11:cada08fc8a70 364 }
mbedAustin 11:cada08fc8a70 365
mbedAustin 11:cada08fc8a70 366 *olen += ilen;
mbedAustin 11:cada08fc8a70 367 }
mbedAustin 11:cada08fc8a70 368
mbedAustin 11:cada08fc8a70 369 return( 0 );
mbedAustin 11:cada08fc8a70 370 }
mbedAustin 11:cada08fc8a70 371 #endif /* MBEDTLS_CIPHER_MODE_CBC */
mbedAustin 11:cada08fc8a70 372
mbedAustin 11:cada08fc8a70 373 #if defined(MBEDTLS_CIPHER_MODE_CFB)
mbedAustin 11:cada08fc8a70 374 if( ctx->cipher_info->mode == MBEDTLS_MODE_CFB )
mbedAustin 11:cada08fc8a70 375 {
mbedAustin 11:cada08fc8a70 376 if( 0 != ( ret = ctx->cipher_info->base->cfb_func( ctx->cipher_ctx,
mbedAustin 11:cada08fc8a70 377 ctx->operation, ilen, &ctx->unprocessed_len, ctx->iv,
mbedAustin 11:cada08fc8a70 378 input, output ) ) )
mbedAustin 11:cada08fc8a70 379 {
mbedAustin 11:cada08fc8a70 380 return( ret );
mbedAustin 11:cada08fc8a70 381 }
mbedAustin 11:cada08fc8a70 382
mbedAustin 11:cada08fc8a70 383 *olen = ilen;
mbedAustin 11:cada08fc8a70 384
mbedAustin 11:cada08fc8a70 385 return( 0 );
mbedAustin 11:cada08fc8a70 386 }
mbedAustin 11:cada08fc8a70 387 #endif /* MBEDTLS_CIPHER_MODE_CFB */
mbedAustin 11:cada08fc8a70 388
mbedAustin 11:cada08fc8a70 389 #if defined(MBEDTLS_CIPHER_MODE_CTR)
mbedAustin 11:cada08fc8a70 390 if( ctx->cipher_info->mode == MBEDTLS_MODE_CTR )
mbedAustin 11:cada08fc8a70 391 {
mbedAustin 11:cada08fc8a70 392 if( 0 != ( ret = ctx->cipher_info->base->ctr_func( ctx->cipher_ctx,
mbedAustin 11:cada08fc8a70 393 ilen, &ctx->unprocessed_len, ctx->iv,
mbedAustin 11:cada08fc8a70 394 ctx->unprocessed_data, input, output ) ) )
mbedAustin 11:cada08fc8a70 395 {
mbedAustin 11:cada08fc8a70 396 return( ret );
mbedAustin 11:cada08fc8a70 397 }
mbedAustin 11:cada08fc8a70 398
mbedAustin 11:cada08fc8a70 399 *olen = ilen;
mbedAustin 11:cada08fc8a70 400
mbedAustin 11:cada08fc8a70 401 return( 0 );
mbedAustin 11:cada08fc8a70 402 }
mbedAustin 11:cada08fc8a70 403 #endif /* MBEDTLS_CIPHER_MODE_CTR */
mbedAustin 11:cada08fc8a70 404
mbedAustin 11:cada08fc8a70 405 #if defined(MBEDTLS_CIPHER_MODE_STREAM)
mbedAustin 11:cada08fc8a70 406 if( ctx->cipher_info->mode == MBEDTLS_MODE_STREAM )
mbedAustin 11:cada08fc8a70 407 {
mbedAustin 11:cada08fc8a70 408 if( 0 != ( ret = ctx->cipher_info->base->stream_func( ctx->cipher_ctx,
mbedAustin 11:cada08fc8a70 409 ilen, input, output ) ) )
mbedAustin 11:cada08fc8a70 410 {
mbedAustin 11:cada08fc8a70 411 return( ret );
mbedAustin 11:cada08fc8a70 412 }
mbedAustin 11:cada08fc8a70 413
mbedAustin 11:cada08fc8a70 414 *olen = ilen;
mbedAustin 11:cada08fc8a70 415
mbedAustin 11:cada08fc8a70 416 return( 0 );
mbedAustin 11:cada08fc8a70 417 }
mbedAustin 11:cada08fc8a70 418 #endif /* MBEDTLS_CIPHER_MODE_STREAM */
mbedAustin 11:cada08fc8a70 419
mbedAustin 11:cada08fc8a70 420 return( MBEDTLS_ERR_CIPHER_FEATURE_UNAVAILABLE );
mbedAustin 11:cada08fc8a70 421 }
mbedAustin 11:cada08fc8a70 422
mbedAustin 11:cada08fc8a70 423 #if defined(MBEDTLS_CIPHER_MODE_WITH_PADDING)
mbedAustin 11:cada08fc8a70 424 #if defined(MBEDTLS_CIPHER_PADDING_PKCS7)
mbedAustin 11:cada08fc8a70 425 /*
mbedAustin 11:cada08fc8a70 426 * PKCS7 (and PKCS5) padding: fill with ll bytes, with ll = padding_len
mbedAustin 11:cada08fc8a70 427 */
mbedAustin 11:cada08fc8a70 428 static void add_pkcs_padding( unsigned char *output, size_t output_len,
mbedAustin 11:cada08fc8a70 429 size_t data_len )
mbedAustin 11:cada08fc8a70 430 {
mbedAustin 11:cada08fc8a70 431 size_t padding_len = output_len - data_len;
mbedAustin 11:cada08fc8a70 432 unsigned char i;
mbedAustin 11:cada08fc8a70 433
mbedAustin 11:cada08fc8a70 434 for( i = 0; i < padding_len; i++ )
mbedAustin 11:cada08fc8a70 435 output[data_len + i] = (unsigned char) padding_len;
mbedAustin 11:cada08fc8a70 436 }
mbedAustin 11:cada08fc8a70 437
mbedAustin 11:cada08fc8a70 438 static int get_pkcs_padding( unsigned char *input, size_t input_len,
mbedAustin 11:cada08fc8a70 439 size_t *data_len )
mbedAustin 11:cada08fc8a70 440 {
mbedAustin 11:cada08fc8a70 441 size_t i, pad_idx;
mbedAustin 11:cada08fc8a70 442 unsigned char padding_len, bad = 0;
mbedAustin 11:cada08fc8a70 443
mbedAustin 11:cada08fc8a70 444 if( NULL == input || NULL == data_len )
mbedAustin 11:cada08fc8a70 445 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 446
mbedAustin 11:cada08fc8a70 447 padding_len = input[input_len - 1];
mbedAustin 11:cada08fc8a70 448 *data_len = input_len - padding_len;
mbedAustin 11:cada08fc8a70 449
mbedAustin 11:cada08fc8a70 450 /* Avoid logical || since it results in a branch */
mbedAustin 11:cada08fc8a70 451 bad |= padding_len > input_len;
mbedAustin 11:cada08fc8a70 452 bad |= padding_len == 0;
mbedAustin 11:cada08fc8a70 453
mbedAustin 11:cada08fc8a70 454 /* The number of bytes checked must be independent of padding_len,
mbedAustin 11:cada08fc8a70 455 * so pick input_len, which is usually 8 or 16 (one block) */
mbedAustin 11:cada08fc8a70 456 pad_idx = input_len - padding_len;
mbedAustin 11:cada08fc8a70 457 for( i = 0; i < input_len; i++ )
mbedAustin 11:cada08fc8a70 458 bad |= ( input[i] ^ padding_len ) * ( i >= pad_idx );
mbedAustin 11:cada08fc8a70 459
mbedAustin 11:cada08fc8a70 460 return( MBEDTLS_ERR_CIPHER_INVALID_PADDING * ( bad != 0 ) );
mbedAustin 11:cada08fc8a70 461 }
mbedAustin 11:cada08fc8a70 462 #endif /* MBEDTLS_CIPHER_PADDING_PKCS7 */
mbedAustin 11:cada08fc8a70 463
mbedAustin 11:cada08fc8a70 464 #if defined(MBEDTLS_CIPHER_PADDING_ONE_AND_ZEROS)
mbedAustin 11:cada08fc8a70 465 /*
mbedAustin 11:cada08fc8a70 466 * One and zeros padding: fill with 80 00 ... 00
mbedAustin 11:cada08fc8a70 467 */
mbedAustin 11:cada08fc8a70 468 static void add_one_and_zeros_padding( unsigned char *output,
mbedAustin 11:cada08fc8a70 469 size_t output_len, size_t data_len )
mbedAustin 11:cada08fc8a70 470 {
mbedAustin 11:cada08fc8a70 471 size_t padding_len = output_len - data_len;
mbedAustin 11:cada08fc8a70 472 unsigned char i = 0;
mbedAustin 11:cada08fc8a70 473
mbedAustin 11:cada08fc8a70 474 output[data_len] = 0x80;
mbedAustin 11:cada08fc8a70 475 for( i = 1; i < padding_len; i++ )
mbedAustin 11:cada08fc8a70 476 output[data_len + i] = 0x00;
mbedAustin 11:cada08fc8a70 477 }
mbedAustin 11:cada08fc8a70 478
mbedAustin 11:cada08fc8a70 479 static int get_one_and_zeros_padding( unsigned char *input, size_t input_len,
mbedAustin 11:cada08fc8a70 480 size_t *data_len )
mbedAustin 11:cada08fc8a70 481 {
mbedAustin 11:cada08fc8a70 482 size_t i;
mbedAustin 11:cada08fc8a70 483 unsigned char done = 0, prev_done, bad;
mbedAustin 11:cada08fc8a70 484
mbedAustin 11:cada08fc8a70 485 if( NULL == input || NULL == data_len )
mbedAustin 11:cada08fc8a70 486 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 487
mbedAustin 11:cada08fc8a70 488 bad = 0xFF;
mbedAustin 11:cada08fc8a70 489 *data_len = 0;
mbedAustin 11:cada08fc8a70 490 for( i = input_len; i > 0; i-- )
mbedAustin 11:cada08fc8a70 491 {
mbedAustin 11:cada08fc8a70 492 prev_done = done;
mbedAustin 11:cada08fc8a70 493 done |= ( input[i-1] != 0 );
mbedAustin 11:cada08fc8a70 494 *data_len |= ( i - 1 ) * ( done != prev_done );
mbedAustin 11:cada08fc8a70 495 bad &= ( input[i-1] ^ 0x80 ) | ( done == prev_done );
mbedAustin 11:cada08fc8a70 496 }
mbedAustin 11:cada08fc8a70 497
mbedAustin 11:cada08fc8a70 498 return( MBEDTLS_ERR_CIPHER_INVALID_PADDING * ( bad != 0 ) );
mbedAustin 11:cada08fc8a70 499
mbedAustin 11:cada08fc8a70 500 }
mbedAustin 11:cada08fc8a70 501 #endif /* MBEDTLS_CIPHER_PADDING_ONE_AND_ZEROS */
mbedAustin 11:cada08fc8a70 502
mbedAustin 11:cada08fc8a70 503 #if defined(MBEDTLS_CIPHER_PADDING_ZEROS_AND_LEN)
mbedAustin 11:cada08fc8a70 504 /*
mbedAustin 11:cada08fc8a70 505 * Zeros and len padding: fill with 00 ... 00 ll, where ll is padding length
mbedAustin 11:cada08fc8a70 506 */
mbedAustin 11:cada08fc8a70 507 static void add_zeros_and_len_padding( unsigned char *output,
mbedAustin 11:cada08fc8a70 508 size_t output_len, size_t data_len )
mbedAustin 11:cada08fc8a70 509 {
mbedAustin 11:cada08fc8a70 510 size_t padding_len = output_len - data_len;
mbedAustin 11:cada08fc8a70 511 unsigned char i = 0;
mbedAustin 11:cada08fc8a70 512
mbedAustin 11:cada08fc8a70 513 for( i = 1; i < padding_len; i++ )
mbedAustin 11:cada08fc8a70 514 output[data_len + i - 1] = 0x00;
mbedAustin 11:cada08fc8a70 515 output[output_len - 1] = (unsigned char) padding_len;
mbedAustin 11:cada08fc8a70 516 }
mbedAustin 11:cada08fc8a70 517
mbedAustin 11:cada08fc8a70 518 static int get_zeros_and_len_padding( unsigned char *input, size_t input_len,
mbedAustin 11:cada08fc8a70 519 size_t *data_len )
mbedAustin 11:cada08fc8a70 520 {
mbedAustin 11:cada08fc8a70 521 size_t i, pad_idx;
mbedAustin 11:cada08fc8a70 522 unsigned char padding_len, bad = 0;
mbedAustin 11:cada08fc8a70 523
mbedAustin 11:cada08fc8a70 524 if( NULL == input || NULL == data_len )
mbedAustin 11:cada08fc8a70 525 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 526
mbedAustin 11:cada08fc8a70 527 padding_len = input[input_len - 1];
mbedAustin 11:cada08fc8a70 528 *data_len = input_len - padding_len;
mbedAustin 11:cada08fc8a70 529
mbedAustin 11:cada08fc8a70 530 /* Avoid logical || since it results in a branch */
mbedAustin 11:cada08fc8a70 531 bad |= padding_len > input_len;
mbedAustin 11:cada08fc8a70 532 bad |= padding_len == 0;
mbedAustin 11:cada08fc8a70 533
mbedAustin 11:cada08fc8a70 534 /* The number of bytes checked must be independent of padding_len */
mbedAustin 11:cada08fc8a70 535 pad_idx = input_len - padding_len;
mbedAustin 11:cada08fc8a70 536 for( i = 0; i < input_len - 1; i++ )
mbedAustin 11:cada08fc8a70 537 bad |= input[i] * ( i >= pad_idx );
mbedAustin 11:cada08fc8a70 538
mbedAustin 11:cada08fc8a70 539 return( MBEDTLS_ERR_CIPHER_INVALID_PADDING * ( bad != 0 ) );
mbedAustin 11:cada08fc8a70 540 }
mbedAustin 11:cada08fc8a70 541 #endif /* MBEDTLS_CIPHER_PADDING_ZEROS_AND_LEN */
mbedAustin 11:cada08fc8a70 542
mbedAustin 11:cada08fc8a70 543 #if defined(MBEDTLS_CIPHER_PADDING_ZEROS)
mbedAustin 11:cada08fc8a70 544 /*
mbedAustin 11:cada08fc8a70 545 * Zero padding: fill with 00 ... 00
mbedAustin 11:cada08fc8a70 546 */
mbedAustin 11:cada08fc8a70 547 static void add_zeros_padding( unsigned char *output,
mbedAustin 11:cada08fc8a70 548 size_t output_len, size_t data_len )
mbedAustin 11:cada08fc8a70 549 {
mbedAustin 11:cada08fc8a70 550 size_t i;
mbedAustin 11:cada08fc8a70 551
mbedAustin 11:cada08fc8a70 552 for( i = data_len; i < output_len; i++ )
mbedAustin 11:cada08fc8a70 553 output[i] = 0x00;
mbedAustin 11:cada08fc8a70 554 }
mbedAustin 11:cada08fc8a70 555
mbedAustin 11:cada08fc8a70 556 static int get_zeros_padding( unsigned char *input, size_t input_len,
mbedAustin 11:cada08fc8a70 557 size_t *data_len )
mbedAustin 11:cada08fc8a70 558 {
mbedAustin 11:cada08fc8a70 559 size_t i;
mbedAustin 11:cada08fc8a70 560 unsigned char done = 0, prev_done;
mbedAustin 11:cada08fc8a70 561
mbedAustin 11:cada08fc8a70 562 if( NULL == input || NULL == data_len )
mbedAustin 11:cada08fc8a70 563 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 564
mbedAustin 11:cada08fc8a70 565 *data_len = 0;
mbedAustin 11:cada08fc8a70 566 for( i = input_len; i > 0; i-- )
mbedAustin 11:cada08fc8a70 567 {
mbedAustin 11:cada08fc8a70 568 prev_done = done;
mbedAustin 11:cada08fc8a70 569 done |= ( input[i-1] != 0 );
mbedAustin 11:cada08fc8a70 570 *data_len |= i * ( done != prev_done );
mbedAustin 11:cada08fc8a70 571 }
mbedAustin 11:cada08fc8a70 572
mbedAustin 11:cada08fc8a70 573 return( 0 );
mbedAustin 11:cada08fc8a70 574 }
mbedAustin 11:cada08fc8a70 575 #endif /* MBEDTLS_CIPHER_PADDING_ZEROS */
mbedAustin 11:cada08fc8a70 576
mbedAustin 11:cada08fc8a70 577 /*
mbedAustin 11:cada08fc8a70 578 * No padding: don't pad :)
mbedAustin 11:cada08fc8a70 579 *
mbedAustin 11:cada08fc8a70 580 * There is no add_padding function (check for NULL in mbedtls_cipher_finish)
mbedAustin 11:cada08fc8a70 581 * but a trivial get_padding function
mbedAustin 11:cada08fc8a70 582 */
mbedAustin 11:cada08fc8a70 583 static int get_no_padding( unsigned char *input, size_t input_len,
mbedAustin 11:cada08fc8a70 584 size_t *data_len )
mbedAustin 11:cada08fc8a70 585 {
mbedAustin 11:cada08fc8a70 586 if( NULL == input || NULL == data_len )
mbedAustin 11:cada08fc8a70 587 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 588
mbedAustin 11:cada08fc8a70 589 *data_len = input_len;
mbedAustin 11:cada08fc8a70 590
mbedAustin 11:cada08fc8a70 591 return( 0 );
mbedAustin 11:cada08fc8a70 592 }
mbedAustin 11:cada08fc8a70 593 #endif /* MBEDTLS_CIPHER_MODE_WITH_PADDING */
mbedAustin 11:cada08fc8a70 594
mbedAustin 11:cada08fc8a70 595 int mbedtls_cipher_finish( mbedtls_cipher_context_t *ctx,
mbedAustin 11:cada08fc8a70 596 unsigned char *output, size_t *olen )
mbedAustin 11:cada08fc8a70 597 {
mbedAustin 11:cada08fc8a70 598 if( NULL == ctx || NULL == ctx->cipher_info || NULL == olen )
mbedAustin 11:cada08fc8a70 599 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 600
mbedAustin 11:cada08fc8a70 601 *olen = 0;
mbedAustin 11:cada08fc8a70 602
mbedAustin 11:cada08fc8a70 603 if( MBEDTLS_MODE_CFB == ctx->cipher_info->mode ||
mbedAustin 11:cada08fc8a70 604 MBEDTLS_MODE_CTR == ctx->cipher_info->mode ||
mbedAustin 11:cada08fc8a70 605 MBEDTLS_MODE_GCM == ctx->cipher_info->mode ||
mbedAustin 11:cada08fc8a70 606 MBEDTLS_MODE_STREAM == ctx->cipher_info->mode )
mbedAustin 11:cada08fc8a70 607 {
mbedAustin 11:cada08fc8a70 608 return( 0 );
mbedAustin 11:cada08fc8a70 609 }
mbedAustin 11:cada08fc8a70 610
mbedAustin 11:cada08fc8a70 611 if( MBEDTLS_MODE_ECB == ctx->cipher_info->mode )
mbedAustin 11:cada08fc8a70 612 {
mbedAustin 11:cada08fc8a70 613 if( ctx->unprocessed_len != 0 )
mbedAustin 11:cada08fc8a70 614 return( MBEDTLS_ERR_CIPHER_FULL_BLOCK_EXPECTED );
mbedAustin 11:cada08fc8a70 615
mbedAustin 11:cada08fc8a70 616 return( 0 );
mbedAustin 11:cada08fc8a70 617 }
mbedAustin 11:cada08fc8a70 618
mbedAustin 11:cada08fc8a70 619 #if defined(MBEDTLS_CIPHER_MODE_CBC)
mbedAustin 11:cada08fc8a70 620 if( MBEDTLS_MODE_CBC == ctx->cipher_info->mode )
mbedAustin 11:cada08fc8a70 621 {
mbedAustin 11:cada08fc8a70 622 int ret = 0;
mbedAustin 11:cada08fc8a70 623
mbedAustin 11:cada08fc8a70 624 if( MBEDTLS_ENCRYPT == ctx->operation )
mbedAustin 11:cada08fc8a70 625 {
mbedAustin 11:cada08fc8a70 626 /* check for 'no padding' mode */
mbedAustin 11:cada08fc8a70 627 if( NULL == ctx->add_padding )
mbedAustin 11:cada08fc8a70 628 {
mbedAustin 11:cada08fc8a70 629 if( 0 != ctx->unprocessed_len )
mbedAustin 11:cada08fc8a70 630 return( MBEDTLS_ERR_CIPHER_FULL_BLOCK_EXPECTED );
mbedAustin 11:cada08fc8a70 631
mbedAustin 11:cada08fc8a70 632 return( 0 );
mbedAustin 11:cada08fc8a70 633 }
mbedAustin 11:cada08fc8a70 634
mbedAustin 11:cada08fc8a70 635 ctx->add_padding( ctx->unprocessed_data, mbedtls_cipher_get_iv_size( ctx ),
mbedAustin 11:cada08fc8a70 636 ctx->unprocessed_len );
mbedAustin 11:cada08fc8a70 637 }
mbedAustin 11:cada08fc8a70 638 else if( mbedtls_cipher_get_block_size( ctx ) != ctx->unprocessed_len )
mbedAustin 11:cada08fc8a70 639 {
mbedAustin 11:cada08fc8a70 640 /*
mbedAustin 11:cada08fc8a70 641 * For decrypt operations, expect a full block,
mbedAustin 11:cada08fc8a70 642 * or an empty block if no padding
mbedAustin 11:cada08fc8a70 643 */
mbedAustin 11:cada08fc8a70 644 if( NULL == ctx->add_padding && 0 == ctx->unprocessed_len )
mbedAustin 11:cada08fc8a70 645 return( 0 );
mbedAustin 11:cada08fc8a70 646
mbedAustin 11:cada08fc8a70 647 return( MBEDTLS_ERR_CIPHER_FULL_BLOCK_EXPECTED );
mbedAustin 11:cada08fc8a70 648 }
mbedAustin 11:cada08fc8a70 649
mbedAustin 11:cada08fc8a70 650 /* cipher block */
mbedAustin 11:cada08fc8a70 651 if( 0 != ( ret = ctx->cipher_info->base->cbc_func( ctx->cipher_ctx,
mbedAustin 11:cada08fc8a70 652 ctx->operation, mbedtls_cipher_get_block_size( ctx ), ctx->iv,
mbedAustin 11:cada08fc8a70 653 ctx->unprocessed_data, output ) ) )
mbedAustin 11:cada08fc8a70 654 {
mbedAustin 11:cada08fc8a70 655 return( ret );
mbedAustin 11:cada08fc8a70 656 }
mbedAustin 11:cada08fc8a70 657
mbedAustin 11:cada08fc8a70 658 /* Set output size for decryption */
mbedAustin 11:cada08fc8a70 659 if( MBEDTLS_DECRYPT == ctx->operation )
mbedAustin 11:cada08fc8a70 660 return ctx->get_padding( output, mbedtls_cipher_get_block_size( ctx ),
mbedAustin 11:cada08fc8a70 661 olen );
mbedAustin 11:cada08fc8a70 662
mbedAustin 11:cada08fc8a70 663 /* Set output size for encryption */
mbedAustin 11:cada08fc8a70 664 *olen = mbedtls_cipher_get_block_size( ctx );
mbedAustin 11:cada08fc8a70 665 return( 0 );
mbedAustin 11:cada08fc8a70 666 }
mbedAustin 11:cada08fc8a70 667 #else
mbedAustin 11:cada08fc8a70 668 ((void) output);
mbedAustin 11:cada08fc8a70 669 #endif /* MBEDTLS_CIPHER_MODE_CBC */
mbedAustin 11:cada08fc8a70 670
mbedAustin 11:cada08fc8a70 671 return( MBEDTLS_ERR_CIPHER_FEATURE_UNAVAILABLE );
mbedAustin 11:cada08fc8a70 672 }
mbedAustin 11:cada08fc8a70 673
mbedAustin 11:cada08fc8a70 674 #if defined(MBEDTLS_CIPHER_MODE_WITH_PADDING)
mbedAustin 11:cada08fc8a70 675 int mbedtls_cipher_set_padding_mode( mbedtls_cipher_context_t *ctx, mbedtls_cipher_padding_t mode )
mbedAustin 11:cada08fc8a70 676 {
mbedAustin 11:cada08fc8a70 677 if( NULL == ctx ||
mbedAustin 11:cada08fc8a70 678 MBEDTLS_MODE_CBC != ctx->cipher_info->mode )
mbedAustin 11:cada08fc8a70 679 {
mbedAustin 11:cada08fc8a70 680 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 681 }
mbedAustin 11:cada08fc8a70 682
mbedAustin 11:cada08fc8a70 683 switch( mode )
mbedAustin 11:cada08fc8a70 684 {
mbedAustin 11:cada08fc8a70 685 #if defined(MBEDTLS_CIPHER_PADDING_PKCS7)
mbedAustin 11:cada08fc8a70 686 case MBEDTLS_PADDING_PKCS7:
mbedAustin 11:cada08fc8a70 687 ctx->add_padding = add_pkcs_padding;
mbedAustin 11:cada08fc8a70 688 ctx->get_padding = get_pkcs_padding;
mbedAustin 11:cada08fc8a70 689 break;
mbedAustin 11:cada08fc8a70 690 #endif
mbedAustin 11:cada08fc8a70 691 #if defined(MBEDTLS_CIPHER_PADDING_ONE_AND_ZEROS)
mbedAustin 11:cada08fc8a70 692 case MBEDTLS_PADDING_ONE_AND_ZEROS:
mbedAustin 11:cada08fc8a70 693 ctx->add_padding = add_one_and_zeros_padding;
mbedAustin 11:cada08fc8a70 694 ctx->get_padding = get_one_and_zeros_padding;
mbedAustin 11:cada08fc8a70 695 break;
mbedAustin 11:cada08fc8a70 696 #endif
mbedAustin 11:cada08fc8a70 697 #if defined(MBEDTLS_CIPHER_PADDING_ZEROS_AND_LEN)
mbedAustin 11:cada08fc8a70 698 case MBEDTLS_PADDING_ZEROS_AND_LEN:
mbedAustin 11:cada08fc8a70 699 ctx->add_padding = add_zeros_and_len_padding;
mbedAustin 11:cada08fc8a70 700 ctx->get_padding = get_zeros_and_len_padding;
mbedAustin 11:cada08fc8a70 701 break;
mbedAustin 11:cada08fc8a70 702 #endif
mbedAustin 11:cada08fc8a70 703 #if defined(MBEDTLS_CIPHER_PADDING_ZEROS)
mbedAustin 11:cada08fc8a70 704 case MBEDTLS_PADDING_ZEROS:
mbedAustin 11:cada08fc8a70 705 ctx->add_padding = add_zeros_padding;
mbedAustin 11:cada08fc8a70 706 ctx->get_padding = get_zeros_padding;
mbedAustin 11:cada08fc8a70 707 break;
mbedAustin 11:cada08fc8a70 708 #endif
mbedAustin 11:cada08fc8a70 709 case MBEDTLS_PADDING_NONE:
mbedAustin 11:cada08fc8a70 710 ctx->add_padding = NULL;
mbedAustin 11:cada08fc8a70 711 ctx->get_padding = get_no_padding;
mbedAustin 11:cada08fc8a70 712 break;
mbedAustin 11:cada08fc8a70 713
mbedAustin 11:cada08fc8a70 714 default:
mbedAustin 11:cada08fc8a70 715 return( MBEDTLS_ERR_CIPHER_FEATURE_UNAVAILABLE );
mbedAustin 11:cada08fc8a70 716 }
mbedAustin 11:cada08fc8a70 717
mbedAustin 11:cada08fc8a70 718 return( 0 );
mbedAustin 11:cada08fc8a70 719 }
mbedAustin 11:cada08fc8a70 720 #endif /* MBEDTLS_CIPHER_MODE_WITH_PADDING */
mbedAustin 11:cada08fc8a70 721
mbedAustin 11:cada08fc8a70 722 #if defined(MBEDTLS_GCM_C)
mbedAustin 11:cada08fc8a70 723 int mbedtls_cipher_write_tag( mbedtls_cipher_context_t *ctx,
mbedAustin 11:cada08fc8a70 724 unsigned char *tag, size_t tag_len )
mbedAustin 11:cada08fc8a70 725 {
mbedAustin 11:cada08fc8a70 726 if( NULL == ctx || NULL == ctx->cipher_info || NULL == tag )
mbedAustin 11:cada08fc8a70 727 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 728
mbedAustin 11:cada08fc8a70 729 if( MBEDTLS_ENCRYPT != ctx->operation )
mbedAustin 11:cada08fc8a70 730 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 731
mbedAustin 11:cada08fc8a70 732 if( MBEDTLS_MODE_GCM == ctx->cipher_info->mode )
mbedAustin 11:cada08fc8a70 733 return mbedtls_gcm_finish( (mbedtls_gcm_context *) ctx->cipher_ctx, tag, tag_len );
mbedAustin 11:cada08fc8a70 734
mbedAustin 11:cada08fc8a70 735 return( 0 );
mbedAustin 11:cada08fc8a70 736 }
mbedAustin 11:cada08fc8a70 737
mbedAustin 11:cada08fc8a70 738 int mbedtls_cipher_check_tag( mbedtls_cipher_context_t *ctx,
mbedAustin 11:cada08fc8a70 739 const unsigned char *tag, size_t tag_len )
mbedAustin 11:cada08fc8a70 740 {
mbedAustin 11:cada08fc8a70 741 int ret;
mbedAustin 11:cada08fc8a70 742
mbedAustin 11:cada08fc8a70 743 if( NULL == ctx || NULL == ctx->cipher_info ||
mbedAustin 11:cada08fc8a70 744 MBEDTLS_DECRYPT != ctx->operation )
mbedAustin 11:cada08fc8a70 745 {
mbedAustin 11:cada08fc8a70 746 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 747 }
mbedAustin 11:cada08fc8a70 748
mbedAustin 11:cada08fc8a70 749 if( MBEDTLS_MODE_GCM == ctx->cipher_info->mode )
mbedAustin 11:cada08fc8a70 750 {
mbedAustin 11:cada08fc8a70 751 unsigned char check_tag[16];
mbedAustin 11:cada08fc8a70 752 size_t i;
mbedAustin 11:cada08fc8a70 753 int diff;
mbedAustin 11:cada08fc8a70 754
mbedAustin 11:cada08fc8a70 755 if( tag_len > sizeof( check_tag ) )
mbedAustin 11:cada08fc8a70 756 return( MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );
mbedAustin 11:cada08fc8a70 757
mbedAustin 11:cada08fc8a70 758 if( 0 != ( ret = mbedtls_gcm_finish( (mbedtls_gcm_context *) ctx->cipher_ctx,
mbedAustin 11:cada08fc8a70 759 check_tag, tag_len ) ) )
mbedAustin 11:cada08fc8a70 760 {
mbedAustin 11:cada08fc8a70 761 return( ret );
mbedAustin 11:cada08fc8a70 762 }
mbedAustin 11:cada08fc8a70 763
mbedAustin 11:cada08fc8a70 764 /* Check the tag in "constant-time" */
mbedAustin 11:cada08fc8a70 765 for( diff = 0, i = 0; i < tag_len; i++ )
mbedAustin 11:cada08fc8a70 766 diff |= tag[i] ^ check_tag[i];
mbedAustin 11:cada08fc8a70 767
mbedAustin 11:cada08fc8a70 768 if( diff != 0 )
mbedAustin 11:cada08fc8a70 769 return( MBEDTLS_ERR_CIPHER_AUTH_FAILED );
mbedAustin 11:cada08fc8a70 770
mbedAustin 11:cada08fc8a70 771 return( 0 );
mbedAustin 11:cada08fc8a70 772 }
mbedAustin 11:cada08fc8a70 773
mbedAustin 11:cada08fc8a70 774 return( 0 );
mbedAustin 11:cada08fc8a70 775 }
mbedAustin 11:cada08fc8a70 776 #endif /* MBEDTLS_GCM_C */
mbedAustin 11:cada08fc8a70 777
mbedAustin 11:cada08fc8a70 778 /*
mbedAustin 11:cada08fc8a70 779 * Packet-oriented wrapper for non-AEAD modes
mbedAustin 11:cada08fc8a70 780 */
mbedAustin 11:cada08fc8a70 781 int mbedtls_cipher_crypt( mbedtls_cipher_context_t *ctx,
mbedAustin 11:cada08fc8a70 782 const unsigned char *iv, size_t iv_len,
mbedAustin 11:cada08fc8a70 783 const unsigned char *input, size_t ilen,
mbedAustin 11:cada08fc8a70 784 unsigned char *output, size_t *olen )
mbedAustin 11:cada08fc8a70 785 {
mbedAustin 11:cada08fc8a70 786 int ret;
mbedAustin 11:cada08fc8a70 787 size_t finish_olen;
mbedAustin 11:cada08fc8a70 788
mbedAustin 11:cada08fc8a70 789 if( ( ret = mbedtls_cipher_set_iv( ctx, iv, iv_len ) ) != 0 )
mbedAustin 11:cada08fc8a70 790 return( ret );
mbedAustin 11:cada08fc8a70 791
mbedAustin 11:cada08fc8a70 792 if( ( ret = mbedtls_cipher_reset( ctx ) ) != 0 )
mbedAustin 11:cada08fc8a70 793 return( ret );
mbedAustin 11:cada08fc8a70 794
mbedAustin 11:cada08fc8a70 795 if( ( ret = mbedtls_cipher_update( ctx, input, ilen, output, olen ) ) != 0 )
mbedAustin 11:cada08fc8a70 796 return( ret );
mbedAustin 11:cada08fc8a70 797
mbedAustin 11:cada08fc8a70 798 if( ( ret = mbedtls_cipher_finish( ctx, output + *olen, &finish_olen ) ) != 0 )
mbedAustin 11:cada08fc8a70 799 return( ret );
mbedAustin 11:cada08fc8a70 800
mbedAustin 11:cada08fc8a70 801 *olen += finish_olen;
mbedAustin 11:cada08fc8a70 802
mbedAustin 11:cada08fc8a70 803 return( 0 );
mbedAustin 11:cada08fc8a70 804 }
mbedAustin 11:cada08fc8a70 805
mbedAustin 11:cada08fc8a70 806 #if defined(MBEDTLS_CIPHER_MODE_AEAD)
mbedAustin 11:cada08fc8a70 807 /*
mbedAustin 11:cada08fc8a70 808 * Packet-oriented encryption for AEAD modes
mbedAustin 11:cada08fc8a70 809 */
mbedAustin 11:cada08fc8a70 810 int mbedtls_cipher_auth_encrypt( mbedtls_cipher_context_t *ctx,
mbedAustin 11:cada08fc8a70 811 const unsigned char *iv, size_t iv_len,
mbedAustin 11:cada08fc8a70 812 const unsigned char *ad, size_t ad_len,
mbedAustin 11:cada08fc8a70 813 const unsigned char *input, size_t ilen,
mbedAustin 11:cada08fc8a70 814 unsigned char *output, size_t *olen,
mbedAustin 11:cada08fc8a70 815 unsigned char *tag, size_t tag_len )
mbedAustin 11:cada08fc8a70 816 {
mbedAustin 11:cada08fc8a70 817 #if defined(MBEDTLS_GCM_C)
mbedAustin 11:cada08fc8a70 818 if( MBEDTLS_MODE_GCM == ctx->cipher_info->mode )
mbedAustin 11:cada08fc8a70 819 {
mbedAustin 11:cada08fc8a70 820 *olen = ilen;
mbedAustin 11:cada08fc8a70 821 return( mbedtls_gcm_crypt_and_tag( ctx->cipher_ctx, MBEDTLS_GCM_ENCRYPT, ilen,
mbedAustin 11:cada08fc8a70 822 iv, iv_len, ad, ad_len, input, output,
mbedAustin 11:cada08fc8a70 823 tag_len, tag ) );
mbedAustin 11:cada08fc8a70 824 }
mbedAustin 11:cada08fc8a70 825 #endif /* MBEDTLS_GCM_C */
mbedAustin 11:cada08fc8a70 826 #if defined(MBEDTLS_CCM_C)
mbedAustin 11:cada08fc8a70 827 if( MBEDTLS_MODE_CCM == ctx->cipher_info->mode )
mbedAustin 11:cada08fc8a70 828 {
mbedAustin 11:cada08fc8a70 829 *olen = ilen;
mbedAustin 11:cada08fc8a70 830 return( mbedtls_ccm_encrypt_and_tag( ctx->cipher_ctx, ilen,
mbedAustin 11:cada08fc8a70 831 iv, iv_len, ad, ad_len, input, output,
mbedAustin 11:cada08fc8a70 832 tag, tag_len ) );
mbedAustin 11:cada08fc8a70 833 }
mbedAustin 11:cada08fc8a70 834 #endif /* MBEDTLS_CCM_C */
mbedAustin 11:cada08fc8a70 835
mbedAustin 11:cada08fc8a70 836 return( MBEDTLS_ERR_CIPHER_FEATURE_UNAVAILABLE );
mbedAustin 11:cada08fc8a70 837 }
mbedAustin 11:cada08fc8a70 838
mbedAustin 11:cada08fc8a70 839 /*
mbedAustin 11:cada08fc8a70 840 * Packet-oriented decryption for AEAD modes
mbedAustin 11:cada08fc8a70 841 */
mbedAustin 11:cada08fc8a70 842 int mbedtls_cipher_auth_decrypt( mbedtls_cipher_context_t *ctx,
mbedAustin 11:cada08fc8a70 843 const unsigned char *iv, size_t iv_len,
mbedAustin 11:cada08fc8a70 844 const unsigned char *ad, size_t ad_len,
mbedAustin 11:cada08fc8a70 845 const unsigned char *input, size_t ilen,
mbedAustin 11:cada08fc8a70 846 unsigned char *output, size_t *olen,
mbedAustin 11:cada08fc8a70 847 const unsigned char *tag, size_t tag_len )
mbedAustin 11:cada08fc8a70 848 {
mbedAustin 11:cada08fc8a70 849 #if defined(MBEDTLS_GCM_C)
mbedAustin 11:cada08fc8a70 850 if( MBEDTLS_MODE_GCM == ctx->cipher_info->mode )
mbedAustin 11:cada08fc8a70 851 {
mbedAustin 11:cada08fc8a70 852 int ret;
mbedAustin 11:cada08fc8a70 853
mbedAustin 11:cada08fc8a70 854 *olen = ilen;
mbedAustin 11:cada08fc8a70 855 ret = mbedtls_gcm_auth_decrypt( ctx->cipher_ctx, ilen,
mbedAustin 11:cada08fc8a70 856 iv, iv_len, ad, ad_len,
mbedAustin 11:cada08fc8a70 857 tag, tag_len, input, output );
mbedAustin 11:cada08fc8a70 858
mbedAustin 11:cada08fc8a70 859 if( ret == MBEDTLS_ERR_GCM_AUTH_FAILED )
mbedAustin 11:cada08fc8a70 860 ret = MBEDTLS_ERR_CIPHER_AUTH_FAILED;
mbedAustin 11:cada08fc8a70 861
mbedAustin 11:cada08fc8a70 862 return( ret );
mbedAustin 11:cada08fc8a70 863 }
mbedAustin 11:cada08fc8a70 864 #endif /* MBEDTLS_GCM_C */
mbedAustin 11:cada08fc8a70 865 #if defined(MBEDTLS_CCM_C)
mbedAustin 11:cada08fc8a70 866 if( MBEDTLS_MODE_CCM == ctx->cipher_info->mode )
mbedAustin 11:cada08fc8a70 867 {
mbedAustin 11:cada08fc8a70 868 int ret;
mbedAustin 11:cada08fc8a70 869
mbedAustin 11:cada08fc8a70 870 *olen = ilen;
mbedAustin 11:cada08fc8a70 871 ret = mbedtls_ccm_auth_decrypt( ctx->cipher_ctx, ilen,
mbedAustin 11:cada08fc8a70 872 iv, iv_len, ad, ad_len,
mbedAustin 11:cada08fc8a70 873 input, output, tag, tag_len );
mbedAustin 11:cada08fc8a70 874
mbedAustin 11:cada08fc8a70 875 if( ret == MBEDTLS_ERR_CCM_AUTH_FAILED )
mbedAustin 11:cada08fc8a70 876 ret = MBEDTLS_ERR_CIPHER_AUTH_FAILED;
mbedAustin 11:cada08fc8a70 877
mbedAustin 11:cada08fc8a70 878 return( ret );
mbedAustin 11:cada08fc8a70 879 }
mbedAustin 11:cada08fc8a70 880 #endif /* MBEDTLS_CCM_C */
mbedAustin 11:cada08fc8a70 881
mbedAustin 11:cada08fc8a70 882 return( MBEDTLS_ERR_CIPHER_FEATURE_UNAVAILABLE );
mbedAustin 11:cada08fc8a70 883 }
mbedAustin 11:cada08fc8a70 884 #endif /* MBEDTLS_CIPHER_MODE_AEAD */
mbedAustin 11:cada08fc8a70 885
mbedAustin 11:cada08fc8a70 886 #endif /* MBEDTLS_CIPHER_C */