blob: 36c7e73c618cf20fb7fde9bc2d08a9e049c90aa2 [file] [log] [blame]
Ronald Cron0ff57952021-03-08 16:46:35 +01001/*
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 Crond6d28882020-12-14 14:56:02 +010026#include "psa_crypto_core.h"
Ronald Cron6d051732020-10-01 14:10:20 +020027#include "psa_crypto_random_impl.h"
28
Ronald Crond6d28882020-12-14 14:56:02 +010029#include "mbedtls/cipher.h"
Ronald Cron6d051732020-10-01 14:10:20 +020030#include "mbedtls/error.h"
Ronald Cron0ff57952021-03-08 16:46:35 +010031
Ronald Crond6d28882020-12-14 14:56:02 +010032#include <string.h>
33
Ronald Cron5d9b00d2021-03-10 14:43:20 +010034#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 Cron75e6ae22021-03-17 14:46:05 +010058const 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 Cron75e6ae22021-03-17 14:46:05 +0100134 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 Cron5d9b00d2021-03-10 14:43:20 +0100147#if defined(MBEDTLS_PSA_BUILTIN_CIPHER) || defined(PSA_CRYPTO_DRIVER_TEST)
148
Ronald Crond6d28882020-12-14 14:56:02 +0100149static psa_status_t cipher_setup(
Ronald Cron6e412a72021-03-10 09:58:47 +0100150 mbedtls_psa_cipher_operation_t *operation,
Ronald Crond6d28882020-12-14 14:56:02 +0100151 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
163 /* Proceed with initializing an mbed TLS cipher context if no driver is
164 * available for the given algorithm & key. */
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200165 mbedtls_cipher_init( &operation->ctx.cipher );
Ronald Crond6d28882020-12-14 14:56:02 +0100166
Ronald Cron6e412a72021-03-10 09:58:47 +0100167 operation->alg = alg;
Ronald Crond6d28882020-12-14 14:56:02 +0100168 key_bits = attributes->core.bits;
169 cipher_info = mbedtls_cipher_info_from_psa( alg, key_type,
170 key_bits, NULL );
171 if( cipher_info == NULL )
172 return( PSA_ERROR_NOT_SUPPORTED );
173
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200174 ret = mbedtls_cipher_setup( &operation->ctx.cipher, cipher_info );
Ronald Crond6d28882020-12-14 14:56:02 +0100175 if( ret != 0 )
176 goto exit;
177
Ronald Cron5d9b00d2021-03-10 14:43:20 +0100178#if defined(BUILTIN_KEY_TYPE_DES)
Ronald Crond6d28882020-12-14 14:56:02 +0100179 if( key_type == PSA_KEY_TYPE_DES && key_bits == 128 )
180 {
181 /* Two-key Triple-DES is 3-key Triple-DES with K1=K3 */
182 uint8_t keys[24];
183 memcpy( keys, key_buffer, 16 );
184 memcpy( keys + 16, key_buffer, 8 );
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200185 ret = mbedtls_cipher_setkey( &operation->ctx.cipher,
Ronald Crond6d28882020-12-14 14:56:02 +0100186 keys,
187 192, cipher_operation );
188 }
189 else
190#endif
191 {
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200192 ret = mbedtls_cipher_setkey( &operation->ctx.cipher, key_buffer,
Ronald Crond6d28882020-12-14 14:56:02 +0100193 (int) key_bits, cipher_operation );
194 }
195 if( ret != 0 )
196 goto exit;
197
Ronald Cron5d9b00d2021-03-10 14:43:20 +0100198#if defined(BUILTIN_ALG_CBC_NO_PADDING) || \
199 defined(BUILTIN_ALG_CBC_PKCS7)
Ronald Crond6d28882020-12-14 14:56:02 +0100200 switch( alg )
201 {
202 case PSA_ALG_CBC_NO_PADDING:
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200203 ret = mbedtls_cipher_set_padding_mode( &operation->ctx.cipher,
Ronald Crond6d28882020-12-14 14:56:02 +0100204 MBEDTLS_PADDING_NONE );
205 break;
206 case PSA_ALG_CBC_PKCS7:
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200207 ret = mbedtls_cipher_set_padding_mode( &operation->ctx.cipher,
Ronald Crond6d28882020-12-14 14:56:02 +0100208 MBEDTLS_PADDING_PKCS7 );
209 break;
210 default:
211 /* The algorithm doesn't involve padding. */
212 ret = 0;
213 break;
214 }
215 if( ret != 0 )
216 goto exit;
Ronald Cron5d9b00d2021-03-10 14:43:20 +0100217#endif /* BUILTIN_ALG_CBC_NO_PADDING || BUILTIN_ALG_CBC_PKCS7 */
Ronald Crond6d28882020-12-14 14:56:02 +0100218
Ronald Cron6ad554c2021-03-26 09:29:09 +0100219 operation->block_length = ( PSA_ALG_IS_STREAM_CIPHER( alg ) ? 1 :
220 PSA_BLOCK_CIPHER_BLOCK_LENGTH( key_type ) );
Ronald Cronc17e8a92021-03-19 14:12:26 +0100221 operation->iv_length = PSA_CIPHER_IV_LENGTH( key_type, alg );
Ronald Crond6d28882020-12-14 14:56:02 +0100222
223exit:
224 return( mbedtls_to_psa_error( ret ) );
225}
226
Ronald Cron8287e6b2021-03-12 10:35:18 +0100227static psa_status_t cipher_encrypt_setup(
Ronald Cron6e412a72021-03-10 09:58:47 +0100228 mbedtls_psa_cipher_operation_t *operation,
Ronald Crond6d28882020-12-14 14:56:02 +0100229 const psa_key_attributes_t *attributes,
230 const uint8_t *key_buffer, size_t key_buffer_size,
231 psa_algorithm_t alg )
232{
233 return( cipher_setup( operation, attributes,
234 key_buffer, key_buffer_size,
235 alg, MBEDTLS_ENCRYPT ) );
236}
237
Ronald Cron8287e6b2021-03-12 10:35:18 +0100238static psa_status_t cipher_decrypt_setup(
Ronald Cron6e412a72021-03-10 09:58:47 +0100239 mbedtls_psa_cipher_operation_t *operation,
Ronald Crond6d28882020-12-14 14:56:02 +0100240 const psa_key_attributes_t *attributes,
241 const uint8_t *key_buffer, size_t key_buffer_size,
242 psa_algorithm_t alg )
243{
244 return( cipher_setup( operation, attributes,
245 key_buffer, key_buffer_size,
246 alg, MBEDTLS_DECRYPT ) );
247}
Ronald Cron6d051732020-10-01 14:10:20 +0200248
Ronald Cron8287e6b2021-03-12 10:35:18 +0100249static psa_status_t cipher_set_iv( mbedtls_psa_cipher_operation_t *operation,
250 const uint8_t *iv, size_t iv_length )
251{
Ronald Cron6ad554c2021-03-26 09:29:09 +0100252 if( iv_length != operation->iv_length )
Ronald Cron8287e6b2021-03-12 10:35:18 +0100253 return( PSA_ERROR_INVALID_ARGUMENT );
254
255 return( mbedtls_to_psa_error(
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200256 mbedtls_cipher_set_iv( &operation->ctx.cipher,
Ronald Cron8287e6b2021-03-12 10:35:18 +0100257 iv, iv_length ) ) );
258}
259
Ronald Cron6d051732020-10-01 14:10:20 +0200260/* Process input for which the algorithm is set to ECB mode. This requires
261 * manual processing, since the PSA API is defined as being able to process
262 * arbitrary-length calls to psa_cipher_update() with ECB mode, but the
263 * underlying mbedtls_cipher_update only takes full blocks. */
264static psa_status_t psa_cipher_update_ecb(
265 mbedtls_cipher_context_t *ctx,
266 const uint8_t *input,
267 size_t input_length,
268 uint8_t *output,
269 size_t output_size,
270 size_t *output_length )
271{
272 psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED;
273 size_t block_size = ctx->cipher_info->block_size;
274 size_t internal_output_length = 0;
275 *output_length = 0;
276
277 if( input_length == 0 )
278 {
279 status = PSA_SUCCESS;
280 goto exit;
281 }
282
283 if( ctx->unprocessed_len > 0 )
284 {
285 /* Fill up to block size, and run the block if there's a full one. */
286 size_t bytes_to_copy = block_size - ctx->unprocessed_len;
287
288 if( input_length < bytes_to_copy )
289 bytes_to_copy = input_length;
290
291 memcpy( &( ctx->unprocessed_data[ctx->unprocessed_len] ),
292 input, bytes_to_copy );
293 input_length -= bytes_to_copy;
294 input += bytes_to_copy;
295 ctx->unprocessed_len += bytes_to_copy;
296
297 if( ctx->unprocessed_len == block_size )
298 {
299 status = mbedtls_to_psa_error(
300 mbedtls_cipher_update( ctx,
301 ctx->unprocessed_data,
302 block_size,
303 output, &internal_output_length ) );
304
305 if( status != PSA_SUCCESS )
306 goto exit;
307
308 output += internal_output_length;
309 output_size -= internal_output_length;
310 *output_length += internal_output_length;
311 ctx->unprocessed_len = 0;
312 }
313 }
314
315 while( input_length >= block_size )
316 {
317 /* Run all full blocks we have, one by one */
318 status = mbedtls_to_psa_error(
319 mbedtls_cipher_update( ctx, input,
320 block_size,
321 output, &internal_output_length ) );
322
323 if( status != PSA_SUCCESS )
324 goto exit;
325
326 input_length -= block_size;
327 input += block_size;
328
329 output += internal_output_length;
330 output_size -= internal_output_length;
331 *output_length += internal_output_length;
332 }
333
334 if( input_length > 0 )
335 {
336 /* Save unprocessed bytes for later processing */
337 memcpy( &( ctx->unprocessed_data[ctx->unprocessed_len] ),
338 input, input_length );
339 ctx->unprocessed_len += input_length;
340 }
341
342 status = PSA_SUCCESS;
343
344exit:
345 return( status );
346}
347
Ronald Cron8287e6b2021-03-12 10:35:18 +0100348static psa_status_t cipher_update( mbedtls_psa_cipher_operation_t *operation,
349 const uint8_t *input,
350 size_t input_length,
351 uint8_t *output,
352 size_t output_size,
353 size_t *output_length )
Ronald Cron6d051732020-10-01 14:10:20 +0200354{
355 psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED;
356 size_t expected_output_size;
357
Ronald Cron6e412a72021-03-10 09:58:47 +0100358 if( ! PSA_ALG_IS_STREAM_CIPHER( operation->alg ) )
Ronald Cron6d051732020-10-01 14:10:20 +0200359 {
360 /* Take the unprocessed partial block left over from previous
361 * update calls, if any, plus the input to this call. Remove
362 * the last partial block, if any. You get the data that will be
363 * output in this call. */
364 expected_output_size =
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200365 ( operation->ctx.cipher.unprocessed_len + input_length )
Ronald Cron6ad554c2021-03-26 09:29:09 +0100366 / operation->block_length * operation->block_length;
Ronald Cron6d051732020-10-01 14:10:20 +0200367 }
368 else
369 {
370 expected_output_size = input_length;
371 }
372
373 if( output_size < expected_output_size )
374 return( PSA_ERROR_BUFFER_TOO_SMALL );
375
Ronald Cron6e412a72021-03-10 09:58:47 +0100376 if( operation->alg == PSA_ALG_ECB_NO_PADDING )
Ronald Cron6d051732020-10-01 14:10:20 +0200377 {
378 /* mbedtls_cipher_update has an API inconsistency: it will only
379 * process a single block at a time in ECB mode. Abstract away that
380 * inconsistency here to match the PSA API behaviour. */
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200381 status = psa_cipher_update_ecb( &operation->ctx.cipher,
Ronald Cron6d051732020-10-01 14:10:20 +0200382 input,
383 input_length,
384 output,
385 output_size,
386 output_length );
387 }
388 else
389 {
390 status = mbedtls_to_psa_error(
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200391 mbedtls_cipher_update( &operation->ctx.cipher, input,
Ronald Cron6d051732020-10-01 14:10:20 +0200392 input_length, output, output_length ) );
gabor-mezei-arm58c17272021-06-29 16:41:25 +0200393
394 if( *output_length > output_size )
gabor-mezei-arm00e54f12021-06-29 19:06:30 +0200395 return( PSA_ERROR_CORRUPTION_DETECTED );
Ronald Cron6d051732020-10-01 14:10:20 +0200396 }
397
398 return( status );
399}
400
Ronald Cron8287e6b2021-03-12 10:35:18 +0100401static psa_status_t cipher_finish( mbedtls_psa_cipher_operation_t *operation,
402 uint8_t *output,
403 size_t output_size,
404 size_t *output_length )
Ronald Cron6d051732020-10-01 14:10:20 +0200405{
406 psa_status_t status = PSA_ERROR_GENERIC_ERROR;
407 uint8_t temp_output_buffer[MBEDTLS_MAX_BLOCK_LENGTH];
408
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200409 if( operation->ctx.cipher.unprocessed_len != 0 )
Ronald Cron6d051732020-10-01 14:10:20 +0200410 {
Ronald Cron6e412a72021-03-10 09:58:47 +0100411 if( operation->alg == PSA_ALG_ECB_NO_PADDING ||
412 operation->alg == PSA_ALG_CBC_NO_PADDING )
Ronald Cron6d051732020-10-01 14:10:20 +0200413 {
414 status = PSA_ERROR_INVALID_ARGUMENT;
415 goto exit;
416 }
417 }
418
419 status = mbedtls_to_psa_error(
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200420 mbedtls_cipher_finish( &operation->ctx.cipher,
Ronald Cron6d051732020-10-01 14:10:20 +0200421 temp_output_buffer,
422 output_length ) );
423 if( status != PSA_SUCCESS )
424 goto exit;
425
426 if( *output_length == 0 )
427 ; /* Nothing to copy. Note that output may be NULL in this case. */
428 else if( output_size >= *output_length )
429 memcpy( output, temp_output_buffer, *output_length );
430 else
431 status = PSA_ERROR_BUFFER_TOO_SMALL;
432
433exit:
434 mbedtls_platform_zeroize( temp_output_buffer,
435 sizeof( temp_output_buffer ) );
436
437 return( status );
438}
439
Ronald Cron8287e6b2021-03-12 10:35:18 +0100440static psa_status_t cipher_abort( mbedtls_psa_cipher_operation_t *operation )
Ronald Cron6d051732020-10-01 14:10:20 +0200441{
Ronald Cron937dfee2021-03-10 09:17:32 +0100442 /* Sanity check (shouldn't happen: operation->alg should
443 * always have been initialized to a valid value). */
Ronald Cron6e412a72021-03-10 09:58:47 +0100444 if( ! PSA_ALG_IS_CIPHER( operation->alg ) )
Ronald Cron937dfee2021-03-10 09:17:32 +0100445 return( PSA_ERROR_BAD_STATE );
446
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200447 mbedtls_cipher_free( &operation->ctx.cipher );
Ronald Cron6d051732020-10-01 14:10:20 +0200448
449 return( PSA_SUCCESS );
450}
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100451
452static psa_status_t cipher_encrypt( const psa_key_attributes_t *attributes,
453 const uint8_t *key_buffer,
454 size_t key_buffer_size,
455 psa_algorithm_t alg,
456 const uint8_t *input,
457 size_t input_length,
458 uint8_t *output,
459 size_t output_size,
460 size_t *output_length )
461{
462 psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED;
463 mbedtls_psa_cipher_operation_t operation = MBEDTLS_PSA_CIPHER_OPERATION_INIT;
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200464 size_t olength, accumulated_length;
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100465
gabor-mezei-arm47a8e142021-06-25 15:44:47 +0200466 status = cipher_encrypt_setup( &operation, attributes,
467 key_buffer, key_buffer_size, alg );
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100468 if( status != PSA_SUCCESS )
469 goto exit;
470
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200471 accumulated_length = 0;
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100472 if( operation.iv_length > 0 )
473 {
474 status = cipher_set_iv( &operation, output, operation.iv_length );
475 if( status != PSA_SUCCESS )
476 goto exit;
477
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200478 accumulated_length = operation.iv_length;
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100479 }
480
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100481 status = cipher_update( &operation, input, input_length,
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200482 output + operation.iv_length,
483 output_size - operation.iv_length,
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100484 &olength );
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100485 if( status != PSA_SUCCESS )
486 goto exit;
487
gabor-mezei-arm6158e282021-06-29 16:42:13 +0200488 accumulated_length += olength;
489
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200490 status = cipher_finish( &operation, output + accumulated_length,
491 output_size - accumulated_length, &olength );
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100492 if( status != PSA_SUCCESS )
493 goto exit;
494
gabor-mezei-arm00e54f12021-06-29 19:06:30 +0200495 *output_length = accumulated_length + olength;
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200496
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100497exit:
498 if( status == PSA_SUCCESS )
499 status = cipher_abort( &operation );
500 else
501 cipher_abort( &operation );
502 return( status );
503}
504
505static psa_status_t cipher_decrypt( const psa_key_attributes_t *attributes,
506 const uint8_t *key_buffer,
507 size_t key_buffer_size,
508 psa_algorithm_t alg,
509 const uint8_t *input,
510 size_t input_length,
511 uint8_t *output,
512 size_t output_size,
513 size_t *output_length )
514{
515 psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED;
516 mbedtls_psa_cipher_operation_t operation = MBEDTLS_PSA_CIPHER_OPERATION_INIT;
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200517 size_t olength, accumulated_length;
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100518
519 status = cipher_decrypt_setup( &operation, attributes,
520 key_buffer, key_buffer_size, alg );
521 if( status != PSA_SUCCESS )
522 goto exit;
523
524 if( operation.iv_length > 0 )
525 {
526 status = cipher_set_iv( &operation, input, operation.iv_length );
527 if( status != PSA_SUCCESS )
528 goto exit;
529 }
530
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100531 status = cipher_update( &operation, input + operation.iv_length,
532 input_length - operation.iv_length,
533 output, output_size, &olength );
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100534 if( status != PSA_SUCCESS )
535 goto exit;
536
gabor-mezei-arm6158e282021-06-29 16:42:13 +0200537 accumulated_length = olength;
gabor-mezei-arm258ae072021-06-25 15:25:38 +0200538
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200539 status = cipher_finish( &operation, output + accumulated_length,
540 output_size - accumulated_length, &olength );
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100541 if( status != PSA_SUCCESS )
542 goto exit;
543
gabor-mezei-arm00e54f12021-06-29 19:06:30 +0200544 *output_length = accumulated_length + olength;
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200545
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100546exit:
547 if ( status == PSA_SUCCESS )
548 status = cipher_abort( &operation );
549 else
550 cipher_abort( &operation );
551 return( status );
552}
Ronald Cron5d9b00d2021-03-10 14:43:20 +0100553#endif /* MBEDTLS_PSA_BUILTIN_CIPHER || PSA_CRYPTO_DRIVER_TEST */
Ronald Cron6d051732020-10-01 14:10:20 +0200554
Ronald Cron5d9b00d2021-03-10 14:43:20 +0100555#if defined(MBEDTLS_PSA_BUILTIN_CIPHER)
Ronald Cron8287e6b2021-03-12 10:35:18 +0100556psa_status_t mbedtls_psa_cipher_encrypt_setup(
557 mbedtls_psa_cipher_operation_t *operation,
558 const psa_key_attributes_t *attributes,
559 const uint8_t *key_buffer, size_t key_buffer_size,
560 psa_algorithm_t alg )
561{
562 return( cipher_encrypt_setup(
563 operation, attributes, key_buffer, key_buffer_size, alg ) );
564}
565
566psa_status_t mbedtls_psa_cipher_decrypt_setup(
567 mbedtls_psa_cipher_operation_t *operation,
568 const psa_key_attributes_t *attributes,
569 const uint8_t *key_buffer, size_t key_buffer_size,
570 psa_algorithm_t alg )
571{
572 return( cipher_decrypt_setup(
573 operation, attributes, key_buffer, key_buffer_size, alg ) );
574}
575
Ronald Cron8287e6b2021-03-12 10:35:18 +0100576psa_status_t mbedtls_psa_cipher_set_iv( mbedtls_psa_cipher_operation_t *operation,
577 const uint8_t *iv,
578 size_t iv_length )
579{
580 return( cipher_set_iv( operation, iv, iv_length ) );
581}
582
583psa_status_t mbedtls_psa_cipher_update( mbedtls_psa_cipher_operation_t *operation,
584 const uint8_t *input,
585 size_t input_length,
586 uint8_t *output,
587 size_t output_size,
588 size_t *output_length )
589{
590 return( cipher_update( operation, input, input_length,
591 output, output_size, output_length ) );
592}
593
594psa_status_t mbedtls_psa_cipher_finish( mbedtls_psa_cipher_operation_t *operation,
595 uint8_t *output,
596 size_t output_size,
597 size_t *output_length )
598{
599 return( cipher_finish( operation, output, output_size, output_length ) );
600}
601
602psa_status_t mbedtls_psa_cipher_abort( mbedtls_psa_cipher_operation_t *operation )
603{
604 return( cipher_abort( operation ) );
605}
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100606
607psa_status_t mbedtls_psa_cipher_encrypt( const psa_key_attributes_t *attributes,
608 const uint8_t *key_buffer,
609 size_t key_buffer_size,
610 psa_algorithm_t alg,
611 const uint8_t *input,
612 size_t input_length,
613 uint8_t *output,
614 size_t output_size,
615 size_t *output_length )
616{
617 return( cipher_encrypt( attributes, key_buffer, key_buffer_size,
618 alg, input, input_length,
619 output, output_size, output_length ) );
620}
621
622psa_status_t mbedtls_psa_cipher_decrypt( const psa_key_attributes_t *attributes,
623 const uint8_t *key_buffer,
624 size_t key_buffer_size,
625 psa_algorithm_t alg,
626 const uint8_t *input,
627 size_t input_length,
628 uint8_t *output,
629 size_t output_size,
630 size_t *output_length )
631{
632 return( cipher_decrypt( attributes, key_buffer, key_buffer_size,
633 alg, input, input_length,
634 output, output_size, output_length ) );
635}
Ronald Cron5d9b00d2021-03-10 14:43:20 +0100636#endif /* MBEDTLS_PSA_BUILTIN_CIPHER */
Ronald Cron8287e6b2021-03-12 10:35:18 +0100637
Ronald Cron3522e322021-03-12 11:08:49 +0100638/*
639 * BEYOND THIS POINT, TEST DRIVER ENTRY POINTS ONLY.
640 */
641
642#if defined(PSA_CRYPTO_DRIVER_TEST)
643psa_status_t mbedtls_transparent_test_driver_cipher_encrypt_setup(
644 mbedtls_psa_cipher_operation_t *operation,
645 const psa_key_attributes_t *attributes,
646 const uint8_t *key_buffer, size_t key_buffer_size,
647 psa_algorithm_t alg )
648{
649 return( cipher_encrypt_setup(
650 operation, attributes, key_buffer, key_buffer_size, alg ) );
651}
652
653psa_status_t mbedtls_transparent_test_driver_cipher_decrypt_setup(
654 mbedtls_psa_cipher_operation_t *operation,
655 const psa_key_attributes_t *attributes,
656 const uint8_t *key_buffer, size_t key_buffer_size,
657 psa_algorithm_t alg )
658{
659 return( cipher_decrypt_setup(
660 operation, attributes, key_buffer, key_buffer_size, alg ) );
661}
662
Ronald Cron3522e322021-03-12 11:08:49 +0100663psa_status_t mbedtls_transparent_test_driver_cipher_set_iv(
664 mbedtls_psa_cipher_operation_t *operation,
665 const uint8_t *iv, size_t iv_length )
666{
667 return( cipher_set_iv( operation, iv, iv_length ) );
668}
669
670psa_status_t mbedtls_transparent_test_driver_cipher_update(
671 mbedtls_psa_cipher_operation_t *operation,
672 const uint8_t *input, size_t input_length,
673 uint8_t *output, size_t output_size, size_t *output_length )
674{
675 return( cipher_update( operation, input, input_length,
676 output, output_size, output_length ) );
677}
678
679psa_status_t mbedtls_transparent_test_driver_cipher_finish(
680 mbedtls_psa_cipher_operation_t *operation,
681 uint8_t *output, size_t output_size, size_t *output_length )
682{
683 return( cipher_finish( operation, output, output_size, output_length ) );
684}
685
686psa_status_t mbedtls_transparent_test_driver_cipher_abort(
687 mbedtls_psa_cipher_operation_t *operation )
688{
689 return( cipher_abort( operation ) );
690}
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100691
692psa_status_t mbedtls_transparent_test_driver_cipher_encrypt(
693 const psa_key_attributes_t *attributes,
694 const uint8_t *key_buffer,
695 size_t key_buffer_size,
696 psa_algorithm_t alg,
697 const uint8_t *input,
698 size_t input_length,
699 uint8_t *output,
700 size_t output_size,
701 size_t *output_length )
702{
703 return( cipher_encrypt( attributes, key_buffer, key_buffer_size,
704 alg, input, input_length,
705 output, output_size, output_length ) );
706}
707
708psa_status_t mbedtls_transparent_test_driver_cipher_decrypt(
709 const psa_key_attributes_t *attributes,
710 const uint8_t *key_buffer,
711 size_t key_buffer_size,
712 psa_algorithm_t alg,
713 const uint8_t *input,
714 size_t input_length,
715 uint8_t *output,
716 size_t output_size,
717 size_t *output_length )
718{
719 return( cipher_decrypt( attributes, key_buffer, key_buffer_size,
720 alg, input, input_length,
721 output, output_size, output_length ) );
722}
Ronald Cron3522e322021-03-12 11:08:49 +0100723#endif /* PSA_CRYPTO_DRIVER_TEST */
724
Ronald Cron0ff57952021-03-08 16:46:35 +0100725#endif /* MBEDTLS_PSA_CRYPTO_C */