| Ronald Cron | 0ff5795 | 2021-03-08 16:46:35 +0100 | [diff] [blame] | 1 | /* | 
|  | 2 | *  PSA cipher driver entry points | 
|  | 3 | */ | 
|  | 4 | /* | 
|  | 5 | *  Copyright The Mbed TLS Contributors | 
|  | 6 | *  SPDX-License-Identifier: Apache-2.0 | 
|  | 7 | * | 
|  | 8 | *  Licensed under the Apache License, Version 2.0 (the "License"); you may | 
|  | 9 | *  not use this file except in compliance with the License. | 
|  | 10 | *  You may obtain a copy of the License at | 
|  | 11 | * | 
|  | 12 | *  http://www.apache.org/licenses/LICENSE-2.0 | 
|  | 13 | * | 
|  | 14 | *  Unless required by applicable law or agreed to in writing, software | 
|  | 15 | *  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | 
|  | 16 | *  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
|  | 17 | *  See the License for the specific language governing permissions and | 
|  | 18 | *  limitations under the License. | 
|  | 19 | */ | 
|  | 20 |  | 
|  | 21 | #include "common.h" | 
|  | 22 |  | 
|  | 23 | #if defined(MBEDTLS_PSA_CRYPTO_C) | 
|  | 24 |  | 
|  | 25 | #include <psa_crypto_cipher.h> | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 26 | #include "psa_crypto_core.h" | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 27 | #include "psa_crypto_random_impl.h" | 
|  | 28 |  | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 29 | #include "mbedtls/cipher.h" | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 30 | #include "mbedtls/error.h" | 
| Ronald Cron | 0ff5795 | 2021-03-08 16:46:35 +0100 | [diff] [blame] | 31 |  | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 32 | #include <string.h> | 
|  | 33 |  | 
| Ronald Cron | 5d9b00d | 2021-03-10 14:43:20 +0100 | [diff] [blame] | 34 | #if ( defined(MBEDTLS_PSA_BUILTIN_KEY_TYPE_DES) || \ | 
|  | 35 | ( defined(PSA_CRYPTO_DRIVER_TEST) && \ | 
|  | 36 | defined(MBEDTLS_PSA_ACCEL_KEY_TYPE_DES) ) ) | 
|  | 37 | #define BUILTIN_KEY_TYPE_DES  1 | 
|  | 38 | #endif | 
|  | 39 |  | 
|  | 40 | #if ( defined(MBEDTLS_PSA_BUILTIN_ALG_CBC_NO_PADDING) || \ | 
|  | 41 | ( defined(PSA_CRYPTO_DRIVER_TEST) && \ | 
|  | 42 | defined(MBEDTLS_PSA_ACCEL_ALG_CBC_NO_PADDING) ) ) | 
|  | 43 | #define BUILTIN_ALG_CBC_NO_PADDING  1 | 
|  | 44 | #endif | 
|  | 45 |  | 
|  | 46 | #if ( defined(MBEDTLS_PSA_BUILTIN_ALG_CBC_PKCS7) || \ | 
|  | 47 | ( defined(PSA_CRYPTO_DRIVER_TEST) && \ | 
|  | 48 | defined(MBEDTLS_PSA_ACCEL_ALG_CBC_PKCS7) ) ) | 
|  | 49 | #define BUILTIN_ALG_CBC_PKCS7  1 | 
|  | 50 | #endif | 
|  | 51 |  | 
|  | 52 | #if ( defined(MBEDTLS_PSA_BUILTIN_KEY_TYPE_CHACHA20) || \ | 
|  | 53 | ( defined(PSA_CRYPTO_DRIVER_TEST) && \ | 
|  | 54 | defined(MBEDTLS_PSA_ACCEL_KEY_TYPE_CHACHA20) ) ) | 
|  | 55 | #define BUILTIN_KEY_TYPE_CHACHA20  1 | 
|  | 56 | #endif | 
|  | 57 |  | 
| Ronald Cron | 75e6ae2 | 2021-03-17 14:46:05 +0100 | [diff] [blame] | 58 | const mbedtls_cipher_info_t *mbedtls_cipher_info_from_psa( | 
|  | 59 | psa_algorithm_t alg, | 
|  | 60 | psa_key_type_t key_type, | 
|  | 61 | size_t key_bits, | 
|  | 62 | mbedtls_cipher_id_t* cipher_id ) | 
|  | 63 | { | 
|  | 64 | mbedtls_cipher_mode_t mode; | 
|  | 65 | mbedtls_cipher_id_t cipher_id_tmp; | 
|  | 66 |  | 
|  | 67 | if( PSA_ALG_IS_AEAD( alg ) ) | 
|  | 68 | alg = PSA_ALG_AEAD_WITH_SHORTENED_TAG( alg, 0 ); | 
|  | 69 |  | 
|  | 70 | if( PSA_ALG_IS_CIPHER( alg ) || PSA_ALG_IS_AEAD( alg ) ) | 
|  | 71 | { | 
|  | 72 | switch( alg ) | 
|  | 73 | { | 
|  | 74 | case PSA_ALG_STREAM_CIPHER: | 
|  | 75 | mode = MBEDTLS_MODE_STREAM; | 
|  | 76 | break; | 
|  | 77 | case PSA_ALG_CTR: | 
|  | 78 | mode = MBEDTLS_MODE_CTR; | 
|  | 79 | break; | 
|  | 80 | case PSA_ALG_CFB: | 
|  | 81 | mode = MBEDTLS_MODE_CFB; | 
|  | 82 | break; | 
|  | 83 | case PSA_ALG_OFB: | 
|  | 84 | mode = MBEDTLS_MODE_OFB; | 
|  | 85 | break; | 
|  | 86 | case PSA_ALG_ECB_NO_PADDING: | 
|  | 87 | mode = MBEDTLS_MODE_ECB; | 
|  | 88 | break; | 
|  | 89 | case PSA_ALG_CBC_NO_PADDING: | 
|  | 90 | mode = MBEDTLS_MODE_CBC; | 
|  | 91 | break; | 
|  | 92 | case PSA_ALG_CBC_PKCS7: | 
|  | 93 | mode = MBEDTLS_MODE_CBC; | 
|  | 94 | break; | 
|  | 95 | case PSA_ALG_AEAD_WITH_SHORTENED_TAG( PSA_ALG_CCM, 0 ): | 
|  | 96 | mode = MBEDTLS_MODE_CCM; | 
|  | 97 | break; | 
|  | 98 | case PSA_ALG_AEAD_WITH_SHORTENED_TAG( PSA_ALG_GCM, 0 ): | 
|  | 99 | mode = MBEDTLS_MODE_GCM; | 
|  | 100 | break; | 
|  | 101 | case PSA_ALG_AEAD_WITH_SHORTENED_TAG( PSA_ALG_CHACHA20_POLY1305, 0 ): | 
|  | 102 | mode = MBEDTLS_MODE_CHACHAPOLY; | 
|  | 103 | break; | 
|  | 104 | default: | 
|  | 105 | return( NULL ); | 
|  | 106 | } | 
|  | 107 | } | 
|  | 108 | else if( alg == PSA_ALG_CMAC ) | 
|  | 109 | mode = MBEDTLS_MODE_ECB; | 
|  | 110 | else | 
|  | 111 | return( NULL ); | 
|  | 112 |  | 
|  | 113 | switch( key_type ) | 
|  | 114 | { | 
|  | 115 | case PSA_KEY_TYPE_AES: | 
|  | 116 | cipher_id_tmp = MBEDTLS_CIPHER_ID_AES; | 
|  | 117 | break; | 
|  | 118 | case PSA_KEY_TYPE_DES: | 
|  | 119 | /* key_bits is 64 for Single-DES, 128 for two-key Triple-DES, | 
|  | 120 | * and 192 for three-key Triple-DES. */ | 
|  | 121 | if( key_bits == 64 ) | 
|  | 122 | cipher_id_tmp = MBEDTLS_CIPHER_ID_DES; | 
|  | 123 | else | 
|  | 124 | cipher_id_tmp = MBEDTLS_CIPHER_ID_3DES; | 
|  | 125 | /* mbedtls doesn't recognize two-key Triple-DES as an algorithm, | 
|  | 126 | * but two-key Triple-DES is functionally three-key Triple-DES | 
|  | 127 | * with K1=K3, so that's how we present it to mbedtls. */ | 
|  | 128 | if( key_bits == 128 ) | 
|  | 129 | key_bits = 192; | 
|  | 130 | break; | 
|  | 131 | case PSA_KEY_TYPE_CAMELLIA: | 
|  | 132 | cipher_id_tmp = MBEDTLS_CIPHER_ID_CAMELLIA; | 
|  | 133 | break; | 
| Ronald Cron | 75e6ae2 | 2021-03-17 14:46:05 +0100 | [diff] [blame] | 134 | case PSA_KEY_TYPE_CHACHA20: | 
|  | 135 | cipher_id_tmp = MBEDTLS_CIPHER_ID_CHACHA20; | 
|  | 136 | break; | 
|  | 137 | default: | 
|  | 138 | return( NULL ); | 
|  | 139 | } | 
|  | 140 | if( cipher_id != NULL ) | 
|  | 141 | *cipher_id = cipher_id_tmp; | 
|  | 142 |  | 
|  | 143 | return( mbedtls_cipher_info_from_values( cipher_id_tmp, | 
|  | 144 | (int) key_bits, mode ) ); | 
|  | 145 | } | 
|  | 146 |  | 
| Ronald Cron | 5d9b00d | 2021-03-10 14:43:20 +0100 | [diff] [blame] | 147 | #if defined(MBEDTLS_PSA_BUILTIN_CIPHER) || defined(PSA_CRYPTO_DRIVER_TEST) | 
|  | 148 |  | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 149 | static psa_status_t cipher_setup( | 
| Ronald Cron | 6e412a7 | 2021-03-10 09:58:47 +0100 | [diff] [blame] | 150 | mbedtls_psa_cipher_operation_t *operation, | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 151 | const psa_key_attributes_t *attributes, | 
|  | 152 | const uint8_t *key_buffer, size_t key_buffer_size, | 
|  | 153 | psa_algorithm_t alg, | 
|  | 154 | mbedtls_operation_t cipher_operation ) | 
|  | 155 | { | 
|  | 156 | int ret = 0; | 
|  | 157 | size_t key_bits; | 
|  | 158 | const mbedtls_cipher_info_t *cipher_info = NULL; | 
|  | 159 | psa_key_type_t key_type = attributes->core.type; | 
|  | 160 |  | 
|  | 161 | (void)key_buffer_size; | 
|  | 162 |  | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 163 | mbedtls_cipher_init( &operation->ctx.cipher ); | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 164 |  | 
| Ronald Cron | 6e412a7 | 2021-03-10 09:58:47 +0100 | [diff] [blame] | 165 | operation->alg = alg; | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 166 | key_bits = attributes->core.bits; | 
|  | 167 | cipher_info = mbedtls_cipher_info_from_psa( alg, key_type, | 
|  | 168 | key_bits, NULL ); | 
|  | 169 | if( cipher_info == NULL ) | 
|  | 170 | return( PSA_ERROR_NOT_SUPPORTED ); | 
|  | 171 |  | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 172 | ret = mbedtls_cipher_setup( &operation->ctx.cipher, cipher_info ); | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 173 | if( ret != 0 ) | 
|  | 174 | goto exit; | 
|  | 175 |  | 
| Ronald Cron | 5d9b00d | 2021-03-10 14:43:20 +0100 | [diff] [blame] | 176 | #if defined(BUILTIN_KEY_TYPE_DES) | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 177 | if( key_type == PSA_KEY_TYPE_DES && key_bits == 128 ) | 
|  | 178 | { | 
|  | 179 | /* Two-key Triple-DES is 3-key Triple-DES with K1=K3 */ | 
|  | 180 | uint8_t keys[24]; | 
|  | 181 | memcpy( keys, key_buffer, 16 ); | 
|  | 182 | memcpy( keys + 16, key_buffer, 8 ); | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 183 | ret = mbedtls_cipher_setkey( &operation->ctx.cipher, | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 184 | keys, | 
|  | 185 | 192, cipher_operation ); | 
|  | 186 | } | 
|  | 187 | else | 
|  | 188 | #endif | 
|  | 189 | { | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 190 | ret = mbedtls_cipher_setkey( &operation->ctx.cipher, key_buffer, | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 191 | (int) key_bits, cipher_operation ); | 
|  | 192 | } | 
|  | 193 | if( ret != 0 ) | 
|  | 194 | goto exit; | 
|  | 195 |  | 
| Ronald Cron | 5d9b00d | 2021-03-10 14:43:20 +0100 | [diff] [blame] | 196 | #if defined(BUILTIN_ALG_CBC_NO_PADDING) || \ | 
|  | 197 | defined(BUILTIN_ALG_CBC_PKCS7) | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 198 | switch( alg ) | 
|  | 199 | { | 
|  | 200 | case PSA_ALG_CBC_NO_PADDING: | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 201 | ret = mbedtls_cipher_set_padding_mode( &operation->ctx.cipher, | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 202 | MBEDTLS_PADDING_NONE ); | 
|  | 203 | break; | 
|  | 204 | case PSA_ALG_CBC_PKCS7: | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 205 | ret = mbedtls_cipher_set_padding_mode( &operation->ctx.cipher, | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 206 | MBEDTLS_PADDING_PKCS7 ); | 
|  | 207 | break; | 
|  | 208 | default: | 
|  | 209 | /* The algorithm doesn't involve padding. */ | 
|  | 210 | ret = 0; | 
|  | 211 | break; | 
|  | 212 | } | 
|  | 213 | if( ret != 0 ) | 
|  | 214 | goto exit; | 
| Ronald Cron | 5d9b00d | 2021-03-10 14:43:20 +0100 | [diff] [blame] | 215 | #endif /* BUILTIN_ALG_CBC_NO_PADDING || BUILTIN_ALG_CBC_PKCS7 */ | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 216 |  | 
| Ronald Cron | 6ad554c | 2021-03-26 09:29:09 +0100 | [diff] [blame] | 217 | operation->block_length = ( PSA_ALG_IS_STREAM_CIPHER( alg ) ? 1 : | 
|  | 218 | PSA_BLOCK_CIPHER_BLOCK_LENGTH( key_type ) ); | 
| Ronald Cron | c17e8a9 | 2021-03-19 14:12:26 +0100 | [diff] [blame] | 219 | operation->iv_length = PSA_CIPHER_IV_LENGTH( key_type, alg ); | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 220 |  | 
|  | 221 | exit: | 
|  | 222 | return( mbedtls_to_psa_error( ret ) ); | 
|  | 223 | } | 
|  | 224 |  | 
| Ronald Cron | 8287e6b | 2021-03-12 10:35:18 +0100 | [diff] [blame] | 225 | static psa_status_t cipher_encrypt_setup( | 
| Ronald Cron | 6e412a7 | 2021-03-10 09:58:47 +0100 | [diff] [blame] | 226 | mbedtls_psa_cipher_operation_t *operation, | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 227 | const psa_key_attributes_t *attributes, | 
|  | 228 | const uint8_t *key_buffer, size_t key_buffer_size, | 
|  | 229 | psa_algorithm_t alg ) | 
|  | 230 | { | 
|  | 231 | return( cipher_setup( operation, attributes, | 
|  | 232 | key_buffer, key_buffer_size, | 
|  | 233 | alg, MBEDTLS_ENCRYPT ) ); | 
|  | 234 | } | 
|  | 235 |  | 
| Ronald Cron | 8287e6b | 2021-03-12 10:35:18 +0100 | [diff] [blame] | 236 | static psa_status_t cipher_decrypt_setup( | 
| Ronald Cron | 6e412a7 | 2021-03-10 09:58:47 +0100 | [diff] [blame] | 237 | mbedtls_psa_cipher_operation_t *operation, | 
| Ronald Cron | d6d2888 | 2020-12-14 14:56:02 +0100 | [diff] [blame] | 238 | const psa_key_attributes_t *attributes, | 
|  | 239 | const uint8_t *key_buffer, size_t key_buffer_size, | 
|  | 240 | psa_algorithm_t alg ) | 
|  | 241 | { | 
|  | 242 | return( cipher_setup( operation, attributes, | 
|  | 243 | key_buffer, key_buffer_size, | 
|  | 244 | alg, MBEDTLS_DECRYPT ) ); | 
|  | 245 | } | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 246 |  | 
| Ronald Cron | 8287e6b | 2021-03-12 10:35:18 +0100 | [diff] [blame] | 247 | static psa_status_t cipher_set_iv( mbedtls_psa_cipher_operation_t *operation, | 
|  | 248 | const uint8_t *iv, size_t iv_length ) | 
|  | 249 | { | 
| Ronald Cron | 6ad554c | 2021-03-26 09:29:09 +0100 | [diff] [blame] | 250 | if( iv_length != operation->iv_length ) | 
| Ronald Cron | 8287e6b | 2021-03-12 10:35:18 +0100 | [diff] [blame] | 251 | return( PSA_ERROR_INVALID_ARGUMENT ); | 
|  | 252 |  | 
|  | 253 | return( mbedtls_to_psa_error( | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 254 | mbedtls_cipher_set_iv( &operation->ctx.cipher, | 
| Ronald Cron | 8287e6b | 2021-03-12 10:35:18 +0100 | [diff] [blame] | 255 | iv, iv_length ) ) ); | 
|  | 256 | } | 
|  | 257 |  | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 258 | /* Process input for which the algorithm is set to ECB mode. This requires | 
|  | 259 | * manual processing, since the PSA API is defined as being able to process | 
|  | 260 | * arbitrary-length calls to psa_cipher_update() with ECB mode, but the | 
|  | 261 | * underlying mbedtls_cipher_update only takes full blocks. */ | 
|  | 262 | static psa_status_t psa_cipher_update_ecb( | 
|  | 263 | mbedtls_cipher_context_t *ctx, | 
|  | 264 | const uint8_t *input, | 
|  | 265 | size_t input_length, | 
|  | 266 | uint8_t *output, | 
|  | 267 | size_t output_size, | 
|  | 268 | size_t *output_length ) | 
|  | 269 | { | 
|  | 270 | psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED; | 
|  | 271 | size_t block_size = ctx->cipher_info->block_size; | 
|  | 272 | size_t internal_output_length = 0; | 
|  | 273 | *output_length = 0; | 
|  | 274 |  | 
|  | 275 | if( input_length == 0 ) | 
|  | 276 | { | 
|  | 277 | status = PSA_SUCCESS; | 
|  | 278 | goto exit; | 
|  | 279 | } | 
|  | 280 |  | 
|  | 281 | if( ctx->unprocessed_len > 0 ) | 
|  | 282 | { | 
|  | 283 | /* Fill up to block size, and run the block if there's a full one. */ | 
|  | 284 | size_t bytes_to_copy = block_size - ctx->unprocessed_len; | 
|  | 285 |  | 
|  | 286 | if( input_length < bytes_to_copy ) | 
|  | 287 | bytes_to_copy = input_length; | 
|  | 288 |  | 
|  | 289 | memcpy( &( ctx->unprocessed_data[ctx->unprocessed_len] ), | 
|  | 290 | input, bytes_to_copy ); | 
|  | 291 | input_length -= bytes_to_copy; | 
|  | 292 | input += bytes_to_copy; | 
|  | 293 | ctx->unprocessed_len += bytes_to_copy; | 
|  | 294 |  | 
|  | 295 | if( ctx->unprocessed_len == block_size ) | 
|  | 296 | { | 
|  | 297 | status = mbedtls_to_psa_error( | 
|  | 298 | mbedtls_cipher_update( ctx, | 
|  | 299 | ctx->unprocessed_data, | 
|  | 300 | block_size, | 
|  | 301 | output, &internal_output_length ) ); | 
|  | 302 |  | 
|  | 303 | if( status != PSA_SUCCESS ) | 
|  | 304 | goto exit; | 
|  | 305 |  | 
|  | 306 | output += internal_output_length; | 
|  | 307 | output_size -= internal_output_length; | 
|  | 308 | *output_length += internal_output_length; | 
|  | 309 | ctx->unprocessed_len = 0; | 
|  | 310 | } | 
|  | 311 | } | 
|  | 312 |  | 
|  | 313 | while( input_length >= block_size ) | 
|  | 314 | { | 
|  | 315 | /* Run all full blocks we have, one by one */ | 
|  | 316 | status = mbedtls_to_psa_error( | 
|  | 317 | mbedtls_cipher_update( ctx, input, | 
|  | 318 | block_size, | 
|  | 319 | output, &internal_output_length ) ); | 
|  | 320 |  | 
|  | 321 | if( status != PSA_SUCCESS ) | 
|  | 322 | goto exit; | 
|  | 323 |  | 
|  | 324 | input_length -= block_size; | 
|  | 325 | input += block_size; | 
|  | 326 |  | 
|  | 327 | output += internal_output_length; | 
|  | 328 | output_size -= internal_output_length; | 
|  | 329 | *output_length += internal_output_length; | 
|  | 330 | } | 
|  | 331 |  | 
|  | 332 | if( input_length > 0 ) | 
|  | 333 | { | 
|  | 334 | /* Save unprocessed bytes for later processing */ | 
|  | 335 | memcpy( &( ctx->unprocessed_data[ctx->unprocessed_len] ), | 
|  | 336 | input, input_length ); | 
|  | 337 | ctx->unprocessed_len += input_length; | 
|  | 338 | } | 
|  | 339 |  | 
|  | 340 | status = PSA_SUCCESS; | 
|  | 341 |  | 
|  | 342 | exit: | 
|  | 343 | return( status ); | 
|  | 344 | } | 
|  | 345 |  | 
| Ronald Cron | 8287e6b | 2021-03-12 10:35:18 +0100 | [diff] [blame] | 346 | static psa_status_t cipher_update( mbedtls_psa_cipher_operation_t *operation, | 
|  | 347 | const uint8_t *input, | 
|  | 348 | size_t input_length, | 
|  | 349 | uint8_t *output, | 
|  | 350 | size_t output_size, | 
|  | 351 | size_t *output_length ) | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 352 | { | 
|  | 353 | psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED; | 
|  | 354 | size_t expected_output_size; | 
|  | 355 |  | 
| Ronald Cron | 6e412a7 | 2021-03-10 09:58:47 +0100 | [diff] [blame] | 356 | if( ! PSA_ALG_IS_STREAM_CIPHER( operation->alg ) ) | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 357 | { | 
|  | 358 | /* Take the unprocessed partial block left over from previous | 
|  | 359 | * update calls, if any, plus the input to this call. Remove | 
|  | 360 | * the last partial block, if any. You get the data that will be | 
|  | 361 | * output in this call. */ | 
|  | 362 | expected_output_size = | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 363 | ( operation->ctx.cipher.unprocessed_len + input_length ) | 
| Ronald Cron | 6ad554c | 2021-03-26 09:29:09 +0100 | [diff] [blame] | 364 | / operation->block_length * operation->block_length; | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 365 | } | 
|  | 366 | else | 
|  | 367 | { | 
|  | 368 | expected_output_size = input_length; | 
|  | 369 | } | 
|  | 370 |  | 
|  | 371 | if( output_size < expected_output_size ) | 
|  | 372 | return( PSA_ERROR_BUFFER_TOO_SMALL ); | 
|  | 373 |  | 
| Ronald Cron | 6e412a7 | 2021-03-10 09:58:47 +0100 | [diff] [blame] | 374 | if( operation->alg == PSA_ALG_ECB_NO_PADDING ) | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 375 | { | 
|  | 376 | /* mbedtls_cipher_update has an API inconsistency: it will only | 
|  | 377 | * process a single block at a time in ECB mode. Abstract away that | 
|  | 378 | * inconsistency here to match the PSA API behaviour. */ | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 379 | status = psa_cipher_update_ecb( &operation->ctx.cipher, | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 380 | input, | 
|  | 381 | input_length, | 
|  | 382 | output, | 
|  | 383 | output_size, | 
|  | 384 | output_length ); | 
|  | 385 | } | 
|  | 386 | else | 
|  | 387 | { | 
|  | 388 | status = mbedtls_to_psa_error( | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 389 | mbedtls_cipher_update( &operation->ctx.cipher, input, | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 390 | input_length, output, output_length ) ); | 
| gabor-mezei-arm | 58c1727 | 2021-06-29 16:41:25 +0200 | [diff] [blame] | 391 |  | 
|  | 392 | if( *output_length > output_size ) | 
| gabor-mezei-arm | 00e54f1 | 2021-06-29 19:06:30 +0200 | [diff] [blame] | 393 | return( PSA_ERROR_CORRUPTION_DETECTED ); | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 394 | } | 
|  | 395 |  | 
|  | 396 | return( status ); | 
|  | 397 | } | 
|  | 398 |  | 
| Ronald Cron | 8287e6b | 2021-03-12 10:35:18 +0100 | [diff] [blame] | 399 | static psa_status_t cipher_finish( mbedtls_psa_cipher_operation_t *operation, | 
|  | 400 | uint8_t *output, | 
|  | 401 | size_t output_size, | 
|  | 402 | size_t *output_length ) | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 403 | { | 
|  | 404 | psa_status_t status = PSA_ERROR_GENERIC_ERROR; | 
|  | 405 | uint8_t temp_output_buffer[MBEDTLS_MAX_BLOCK_LENGTH]; | 
|  | 406 |  | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 407 | if( operation->ctx.cipher.unprocessed_len != 0 ) | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 408 | { | 
| Ronald Cron | 6e412a7 | 2021-03-10 09:58:47 +0100 | [diff] [blame] | 409 | if( operation->alg == PSA_ALG_ECB_NO_PADDING || | 
|  | 410 | operation->alg == PSA_ALG_CBC_NO_PADDING ) | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 411 | { | 
|  | 412 | status = PSA_ERROR_INVALID_ARGUMENT; | 
|  | 413 | goto exit; | 
|  | 414 | } | 
|  | 415 | } | 
|  | 416 |  | 
|  | 417 | status = mbedtls_to_psa_error( | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 418 | mbedtls_cipher_finish( &operation->ctx.cipher, | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 419 | temp_output_buffer, | 
|  | 420 | output_length ) ); | 
|  | 421 | if( status != PSA_SUCCESS ) | 
|  | 422 | goto exit; | 
|  | 423 |  | 
|  | 424 | if( *output_length == 0 ) | 
|  | 425 | ; /* Nothing to copy. Note that output may be NULL in this case. */ | 
|  | 426 | else if( output_size >= *output_length ) | 
|  | 427 | memcpy( output, temp_output_buffer, *output_length ); | 
|  | 428 | else | 
|  | 429 | status = PSA_ERROR_BUFFER_TOO_SMALL; | 
|  | 430 |  | 
|  | 431 | exit: | 
|  | 432 | mbedtls_platform_zeroize( temp_output_buffer, | 
|  | 433 | sizeof( temp_output_buffer ) ); | 
|  | 434 |  | 
|  | 435 | return( status ); | 
|  | 436 | } | 
|  | 437 |  | 
| Ronald Cron | 8287e6b | 2021-03-12 10:35:18 +0100 | [diff] [blame] | 438 | static psa_status_t cipher_abort( mbedtls_psa_cipher_operation_t *operation ) | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 439 | { | 
| Ronald Cron | 937dfee | 2021-03-10 09:17:32 +0100 | [diff] [blame] | 440 | /* Sanity check (shouldn't happen: operation->alg should | 
|  | 441 | * always have been initialized to a valid value). */ | 
| Ronald Cron | 6e412a7 | 2021-03-10 09:58:47 +0100 | [diff] [blame] | 442 | if( ! PSA_ALG_IS_CIPHER( operation->alg ) ) | 
| Ronald Cron | 937dfee | 2021-03-10 09:17:32 +0100 | [diff] [blame] | 443 | return( PSA_ERROR_BAD_STATE ); | 
|  | 444 |  | 
| gabor-mezei-arm | 42cdb2a | 2021-04-12 15:47:35 +0200 | [diff] [blame] | 445 | mbedtls_cipher_free( &operation->ctx.cipher ); | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 446 |  | 
|  | 447 | return( PSA_SUCCESS ); | 
|  | 448 | } | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 449 |  | 
|  | 450 | static psa_status_t cipher_encrypt( const psa_key_attributes_t *attributes, | 
|  | 451 | const uint8_t *key_buffer, | 
|  | 452 | size_t key_buffer_size, | 
|  | 453 | psa_algorithm_t alg, | 
|  | 454 | const uint8_t *input, | 
|  | 455 | size_t input_length, | 
|  | 456 | uint8_t *output, | 
|  | 457 | size_t output_size, | 
|  | 458 | size_t *output_length ) | 
|  | 459 | { | 
|  | 460 | psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED; | 
|  | 461 | mbedtls_psa_cipher_operation_t operation = MBEDTLS_PSA_CIPHER_OPERATION_INIT; | 
| gabor-mezei-arm | e5ff8f4 | 2021-06-25 15:23:05 +0200 | [diff] [blame] | 462 | size_t olength, accumulated_length; | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 463 |  | 
| gabor-mezei-arm | 47a8e14 | 2021-06-25 15:44:47 +0200 | [diff] [blame] | 464 | status = cipher_encrypt_setup( &operation, attributes, | 
|  | 465 | key_buffer, key_buffer_size, alg ); | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 466 | if( status != PSA_SUCCESS ) | 
|  | 467 | goto exit; | 
|  | 468 |  | 
| gabor-mezei-arm | e5ff8f4 | 2021-06-25 15:23:05 +0200 | [diff] [blame] | 469 | accumulated_length = 0; | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 470 | if( operation.iv_length > 0 ) | 
|  | 471 | { | 
|  | 472 | status = cipher_set_iv( &operation, output, operation.iv_length ); | 
|  | 473 | if( status != PSA_SUCCESS ) | 
|  | 474 | goto exit; | 
|  | 475 |  | 
| gabor-mezei-arm | e5ff8f4 | 2021-06-25 15:23:05 +0200 | [diff] [blame] | 476 | accumulated_length = operation.iv_length; | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 477 | } | 
|  | 478 |  | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 479 | status = cipher_update( &operation, input, input_length, | 
| gabor-mezei-arm | e5ff8f4 | 2021-06-25 15:23:05 +0200 | [diff] [blame] | 480 | output + operation.iv_length, | 
|  | 481 | output_size - operation.iv_length, | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 482 | &olength ); | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 483 | if( status != PSA_SUCCESS ) | 
|  | 484 | goto exit; | 
|  | 485 |  | 
| gabor-mezei-arm | 6158e28 | 2021-06-29 16:42:13 +0200 | [diff] [blame] | 486 | accumulated_length += olength; | 
|  | 487 |  | 
| gabor-mezei-arm | e5ff8f4 | 2021-06-25 15:23:05 +0200 | [diff] [blame] | 488 | status = cipher_finish( &operation, output + accumulated_length, | 
|  | 489 | output_size - accumulated_length, &olength ); | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 490 | if( status != PSA_SUCCESS ) | 
|  | 491 | goto exit; | 
|  | 492 |  | 
| gabor-mezei-arm | 00e54f1 | 2021-06-29 19:06:30 +0200 | [diff] [blame] | 493 | *output_length = accumulated_length + olength; | 
| gabor-mezei-arm | e5ff8f4 | 2021-06-25 15:23:05 +0200 | [diff] [blame] | 494 |  | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 495 | exit: | 
|  | 496 | if( status == PSA_SUCCESS ) | 
|  | 497 | status = cipher_abort( &operation ); | 
|  | 498 | else | 
|  | 499 | cipher_abort( &operation ); | 
|  | 500 | return( status ); | 
|  | 501 | } | 
|  | 502 |  | 
|  | 503 | static psa_status_t cipher_decrypt( const psa_key_attributes_t *attributes, | 
|  | 504 | const uint8_t *key_buffer, | 
|  | 505 | size_t key_buffer_size, | 
|  | 506 | psa_algorithm_t alg, | 
|  | 507 | const uint8_t *input, | 
|  | 508 | size_t input_length, | 
|  | 509 | uint8_t *output, | 
|  | 510 | size_t output_size, | 
|  | 511 | size_t *output_length ) | 
|  | 512 | { | 
|  | 513 | psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED; | 
|  | 514 | mbedtls_psa_cipher_operation_t operation = MBEDTLS_PSA_CIPHER_OPERATION_INIT; | 
| gabor-mezei-arm | e5ff8f4 | 2021-06-25 15:23:05 +0200 | [diff] [blame] | 515 | size_t olength, accumulated_length; | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 516 |  | 
|  | 517 | status = cipher_decrypt_setup( &operation, attributes, | 
|  | 518 | key_buffer, key_buffer_size, alg ); | 
|  | 519 | if( status != PSA_SUCCESS ) | 
|  | 520 | goto exit; | 
|  | 521 |  | 
|  | 522 | if( operation.iv_length > 0 ) | 
|  | 523 | { | 
|  | 524 | status = cipher_set_iv( &operation, input, operation.iv_length ); | 
|  | 525 | if( status != PSA_SUCCESS ) | 
|  | 526 | goto exit; | 
|  | 527 | } | 
|  | 528 |  | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 529 | status = cipher_update( &operation, input + operation.iv_length, | 
|  | 530 | input_length - operation.iv_length, | 
|  | 531 | output, output_size, &olength ); | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 532 | if( status != PSA_SUCCESS ) | 
|  | 533 | goto exit; | 
|  | 534 |  | 
| gabor-mezei-arm | 6158e28 | 2021-06-29 16:42:13 +0200 | [diff] [blame] | 535 | accumulated_length = olength; | 
| gabor-mezei-arm | 258ae07 | 2021-06-25 15:25:38 +0200 | [diff] [blame] | 536 |  | 
| gabor-mezei-arm | e5ff8f4 | 2021-06-25 15:23:05 +0200 | [diff] [blame] | 537 | status = cipher_finish( &operation, output + accumulated_length, | 
|  | 538 | output_size - accumulated_length, &olength ); | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 539 | if( status != PSA_SUCCESS ) | 
|  | 540 | goto exit; | 
|  | 541 |  | 
| gabor-mezei-arm | 00e54f1 | 2021-06-29 19:06:30 +0200 | [diff] [blame] | 542 | *output_length = accumulated_length + olength; | 
| gabor-mezei-arm | e5ff8f4 | 2021-06-25 15:23:05 +0200 | [diff] [blame] | 543 |  | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 544 | exit: | 
|  | 545 | if ( status == PSA_SUCCESS ) | 
|  | 546 | status = cipher_abort( &operation ); | 
|  | 547 | else | 
|  | 548 | cipher_abort( &operation ); | 
|  | 549 | return( status ); | 
|  | 550 | } | 
| Ronald Cron | 5d9b00d | 2021-03-10 14:43:20 +0100 | [diff] [blame] | 551 | #endif /* MBEDTLS_PSA_BUILTIN_CIPHER || PSA_CRYPTO_DRIVER_TEST */ | 
| Ronald Cron | 6d05173 | 2020-10-01 14:10:20 +0200 | [diff] [blame] | 552 |  | 
| Ronald Cron | 5d9b00d | 2021-03-10 14:43:20 +0100 | [diff] [blame] | 553 | #if defined(MBEDTLS_PSA_BUILTIN_CIPHER) | 
| Ronald Cron | 8287e6b | 2021-03-12 10:35:18 +0100 | [diff] [blame] | 554 | psa_status_t mbedtls_psa_cipher_encrypt_setup( | 
|  | 555 | mbedtls_psa_cipher_operation_t *operation, | 
|  | 556 | const psa_key_attributes_t *attributes, | 
|  | 557 | const uint8_t *key_buffer, size_t key_buffer_size, | 
|  | 558 | psa_algorithm_t alg ) | 
|  | 559 | { | 
|  | 560 | return( cipher_encrypt_setup( | 
|  | 561 | operation, attributes, key_buffer, key_buffer_size, alg ) ); | 
|  | 562 | } | 
|  | 563 |  | 
|  | 564 | psa_status_t mbedtls_psa_cipher_decrypt_setup( | 
|  | 565 | mbedtls_psa_cipher_operation_t *operation, | 
|  | 566 | const psa_key_attributes_t *attributes, | 
|  | 567 | const uint8_t *key_buffer, size_t key_buffer_size, | 
|  | 568 | psa_algorithm_t alg ) | 
|  | 569 | { | 
|  | 570 | return( cipher_decrypt_setup( | 
|  | 571 | operation, attributes, key_buffer, key_buffer_size, alg ) ); | 
|  | 572 | } | 
|  | 573 |  | 
| Ronald Cron | 8287e6b | 2021-03-12 10:35:18 +0100 | [diff] [blame] | 574 | psa_status_t mbedtls_psa_cipher_set_iv( mbedtls_psa_cipher_operation_t *operation, | 
|  | 575 | const uint8_t *iv, | 
|  | 576 | size_t iv_length ) | 
|  | 577 | { | 
|  | 578 | return( cipher_set_iv( operation, iv, iv_length ) ); | 
|  | 579 | } | 
|  | 580 |  | 
|  | 581 | psa_status_t mbedtls_psa_cipher_update( mbedtls_psa_cipher_operation_t *operation, | 
|  | 582 | const uint8_t *input, | 
|  | 583 | size_t input_length, | 
|  | 584 | uint8_t *output, | 
|  | 585 | size_t output_size, | 
|  | 586 | size_t *output_length ) | 
|  | 587 | { | 
|  | 588 | return( cipher_update( operation, input, input_length, | 
|  | 589 | output, output_size, output_length ) ); | 
|  | 590 | } | 
|  | 591 |  | 
|  | 592 | psa_status_t mbedtls_psa_cipher_finish( mbedtls_psa_cipher_operation_t *operation, | 
|  | 593 | uint8_t *output, | 
|  | 594 | size_t output_size, | 
|  | 595 | size_t *output_length ) | 
|  | 596 | { | 
|  | 597 | return( cipher_finish( operation, output, output_size, output_length ) ); | 
|  | 598 | } | 
|  | 599 |  | 
|  | 600 | psa_status_t mbedtls_psa_cipher_abort( mbedtls_psa_cipher_operation_t *operation ) | 
|  | 601 | { | 
|  | 602 | return( cipher_abort( operation ) ); | 
|  | 603 | } | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 604 |  | 
|  | 605 | psa_status_t mbedtls_psa_cipher_encrypt( const psa_key_attributes_t *attributes, | 
|  | 606 | const uint8_t *key_buffer, | 
|  | 607 | size_t key_buffer_size, | 
|  | 608 | psa_algorithm_t alg, | 
|  | 609 | const uint8_t *input, | 
|  | 610 | size_t input_length, | 
|  | 611 | uint8_t *output, | 
|  | 612 | size_t output_size, | 
|  | 613 | size_t *output_length ) | 
|  | 614 | { | 
|  | 615 | return( cipher_encrypt( attributes, key_buffer, key_buffer_size, | 
|  | 616 | alg, input, input_length, | 
|  | 617 | output, output_size, output_length ) ); | 
|  | 618 | } | 
|  | 619 |  | 
|  | 620 | psa_status_t mbedtls_psa_cipher_decrypt( const psa_key_attributes_t *attributes, | 
|  | 621 | const uint8_t *key_buffer, | 
|  | 622 | size_t key_buffer_size, | 
|  | 623 | psa_algorithm_t alg, | 
|  | 624 | const uint8_t *input, | 
|  | 625 | size_t input_length, | 
|  | 626 | uint8_t *output, | 
|  | 627 | size_t output_size, | 
|  | 628 | size_t *output_length ) | 
|  | 629 | { | 
|  | 630 | return( cipher_decrypt( attributes, key_buffer, key_buffer_size, | 
|  | 631 | alg, input, input_length, | 
|  | 632 | output, output_size, output_length ) ); | 
|  | 633 | } | 
| Ronald Cron | 5d9b00d | 2021-03-10 14:43:20 +0100 | [diff] [blame] | 634 | #endif /* MBEDTLS_PSA_BUILTIN_CIPHER */ | 
| Ronald Cron | 8287e6b | 2021-03-12 10:35:18 +0100 | [diff] [blame] | 635 |  | 
| Ronald Cron | 3522e32 | 2021-03-12 11:08:49 +0100 | [diff] [blame] | 636 | /* | 
|  | 637 | * BEYOND THIS POINT, TEST DRIVER ENTRY POINTS ONLY. | 
|  | 638 | */ | 
|  | 639 |  | 
|  | 640 | #if defined(PSA_CRYPTO_DRIVER_TEST) | 
|  | 641 | psa_status_t mbedtls_transparent_test_driver_cipher_encrypt_setup( | 
|  | 642 | mbedtls_psa_cipher_operation_t *operation, | 
|  | 643 | const psa_key_attributes_t *attributes, | 
|  | 644 | const uint8_t *key_buffer, size_t key_buffer_size, | 
|  | 645 | psa_algorithm_t alg ) | 
|  | 646 | { | 
|  | 647 | return( cipher_encrypt_setup( | 
|  | 648 | operation, attributes, key_buffer, key_buffer_size, alg ) ); | 
|  | 649 | } | 
|  | 650 |  | 
|  | 651 | psa_status_t mbedtls_transparent_test_driver_cipher_decrypt_setup( | 
|  | 652 | mbedtls_psa_cipher_operation_t *operation, | 
|  | 653 | const psa_key_attributes_t *attributes, | 
|  | 654 | const uint8_t *key_buffer, size_t key_buffer_size, | 
|  | 655 | psa_algorithm_t alg ) | 
|  | 656 | { | 
|  | 657 | return( cipher_decrypt_setup( | 
|  | 658 | operation, attributes, key_buffer, key_buffer_size, alg ) ); | 
|  | 659 | } | 
|  | 660 |  | 
| Ronald Cron | 3522e32 | 2021-03-12 11:08:49 +0100 | [diff] [blame] | 661 | psa_status_t mbedtls_transparent_test_driver_cipher_set_iv( | 
|  | 662 | mbedtls_psa_cipher_operation_t *operation, | 
|  | 663 | const uint8_t *iv, size_t iv_length ) | 
|  | 664 | { | 
|  | 665 | return( cipher_set_iv( operation, iv, iv_length ) ); | 
|  | 666 | } | 
|  | 667 |  | 
|  | 668 | psa_status_t mbedtls_transparent_test_driver_cipher_update( | 
|  | 669 | mbedtls_psa_cipher_operation_t *operation, | 
|  | 670 | const uint8_t *input, size_t input_length, | 
|  | 671 | uint8_t *output, size_t output_size, size_t *output_length ) | 
|  | 672 | { | 
|  | 673 | return( cipher_update( operation, input, input_length, | 
|  | 674 | output, output_size, output_length ) ); | 
|  | 675 | } | 
|  | 676 |  | 
|  | 677 | psa_status_t mbedtls_transparent_test_driver_cipher_finish( | 
|  | 678 | mbedtls_psa_cipher_operation_t *operation, | 
|  | 679 | uint8_t *output, size_t output_size, size_t *output_length ) | 
|  | 680 | { | 
|  | 681 | return( cipher_finish( operation, output, output_size, output_length ) ); | 
|  | 682 | } | 
|  | 683 |  | 
|  | 684 | psa_status_t mbedtls_transparent_test_driver_cipher_abort( | 
|  | 685 | mbedtls_psa_cipher_operation_t *operation ) | 
|  | 686 | { | 
|  | 687 | return( cipher_abort( operation ) ); | 
|  | 688 | } | 
| gabor-mezei-arm | a9449a0 | 2021-03-25 11:17:10 +0100 | [diff] [blame] | 689 |  | 
|  | 690 | psa_status_t mbedtls_transparent_test_driver_cipher_encrypt( | 
|  | 691 | const psa_key_attributes_t *attributes, | 
|  | 692 | const uint8_t *key_buffer, | 
|  | 693 | size_t key_buffer_size, | 
|  | 694 | psa_algorithm_t alg, | 
|  | 695 | const uint8_t *input, | 
|  | 696 | size_t input_length, | 
|  | 697 | uint8_t *output, | 
|  | 698 | size_t output_size, | 
|  | 699 | size_t *output_length ) | 
|  | 700 | { | 
|  | 701 | return( cipher_encrypt( attributes, key_buffer, key_buffer_size, | 
|  | 702 | alg, input, input_length, | 
|  | 703 | output, output_size, output_length ) ); | 
|  | 704 | } | 
|  | 705 |  | 
|  | 706 | psa_status_t mbedtls_transparent_test_driver_cipher_decrypt( | 
|  | 707 | const psa_key_attributes_t *attributes, | 
|  | 708 | const uint8_t *key_buffer, | 
|  | 709 | size_t key_buffer_size, | 
|  | 710 | psa_algorithm_t alg, | 
|  | 711 | const uint8_t *input, | 
|  | 712 | size_t input_length, | 
|  | 713 | uint8_t *output, | 
|  | 714 | size_t output_size, | 
|  | 715 | size_t *output_length ) | 
|  | 716 | { | 
|  | 717 | return( cipher_decrypt( attributes, key_buffer, key_buffer_size, | 
|  | 718 | alg, input, input_length, | 
|  | 719 | output, output_size, output_length ) ); | 
|  | 720 | } | 
| Ronald Cron | 3522e32 | 2021-03-12 11:08:49 +0100 | [diff] [blame] | 721 | #endif /* PSA_CRYPTO_DRIVER_TEST */ | 
|  | 722 |  | 
| Ronald Cron | 0ff5795 | 2021-03-08 16:46:35 +0100 | [diff] [blame] | 723 | #endif /* MBEDTLS_PSA_CRYPTO_C */ |