blob: d8c722bb1b0a21e3bc2ee43ca8da650dc1098b1d [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
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200163 mbedtls_cipher_init( &operation->ctx.cipher );
Ronald Crond6d28882020-12-14 14:56:02 +0100164
Ronald Cron6e412a72021-03-10 09:58:47 +0100165 operation->alg = alg;
Ronald Crond6d28882020-12-14 14:56:02 +0100166 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-arm42cdb2a2021-04-12 15:47:35 +0200172 ret = mbedtls_cipher_setup( &operation->ctx.cipher, cipher_info );
Ronald Crond6d28882020-12-14 14:56:02 +0100173 if( ret != 0 )
174 goto exit;
175
Ronald Cron5d9b00d2021-03-10 14:43:20 +0100176#if defined(BUILTIN_KEY_TYPE_DES)
Ronald Crond6d28882020-12-14 14:56:02 +0100177 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-arm42cdb2a2021-04-12 15:47:35 +0200183 ret = mbedtls_cipher_setkey( &operation->ctx.cipher,
Ronald Crond6d28882020-12-14 14:56:02 +0100184 keys,
185 192, cipher_operation );
186 }
187 else
188#endif
189 {
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200190 ret = mbedtls_cipher_setkey( &operation->ctx.cipher, key_buffer,
Ronald Crond6d28882020-12-14 14:56:02 +0100191 (int) key_bits, cipher_operation );
192 }
193 if( ret != 0 )
194 goto exit;
195
Ronald Cron5d9b00d2021-03-10 14:43:20 +0100196#if defined(BUILTIN_ALG_CBC_NO_PADDING) || \
197 defined(BUILTIN_ALG_CBC_PKCS7)
Ronald Crond6d28882020-12-14 14:56:02 +0100198 switch( alg )
199 {
200 case PSA_ALG_CBC_NO_PADDING:
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200201 ret = mbedtls_cipher_set_padding_mode( &operation->ctx.cipher,
Ronald Crond6d28882020-12-14 14:56:02 +0100202 MBEDTLS_PADDING_NONE );
203 break;
204 case PSA_ALG_CBC_PKCS7:
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200205 ret = mbedtls_cipher_set_padding_mode( &operation->ctx.cipher,
Ronald Crond6d28882020-12-14 14:56:02 +0100206 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 Cron5d9b00d2021-03-10 14:43:20 +0100215#endif /* BUILTIN_ALG_CBC_NO_PADDING || BUILTIN_ALG_CBC_PKCS7 */
Ronald Crond6d28882020-12-14 14:56:02 +0100216
Ronald Cron6ad554c2021-03-26 09:29:09 +0100217 operation->block_length = ( PSA_ALG_IS_STREAM_CIPHER( alg ) ? 1 :
218 PSA_BLOCK_CIPHER_BLOCK_LENGTH( key_type ) );
Ronald Cronc17e8a92021-03-19 14:12:26 +0100219 operation->iv_length = PSA_CIPHER_IV_LENGTH( key_type, alg );
Ronald Crond6d28882020-12-14 14:56:02 +0100220
221exit:
222 return( mbedtls_to_psa_error( ret ) );
223}
224
Ronald Cron8287e6b2021-03-12 10:35:18 +0100225static psa_status_t cipher_encrypt_setup(
Ronald Cron6e412a72021-03-10 09:58:47 +0100226 mbedtls_psa_cipher_operation_t *operation,
Ronald Crond6d28882020-12-14 14:56:02 +0100227 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 Cron8287e6b2021-03-12 10:35:18 +0100236static psa_status_t cipher_decrypt_setup(
Ronald Cron6e412a72021-03-10 09:58:47 +0100237 mbedtls_psa_cipher_operation_t *operation,
Ronald Crond6d28882020-12-14 14:56:02 +0100238 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 Cron6d051732020-10-01 14:10:20 +0200246
Ronald Cron8287e6b2021-03-12 10:35:18 +0100247static psa_status_t cipher_set_iv( mbedtls_psa_cipher_operation_t *operation,
248 const uint8_t *iv, size_t iv_length )
249{
Ronald Cron6ad554c2021-03-26 09:29:09 +0100250 if( iv_length != operation->iv_length )
Ronald Cron8287e6b2021-03-12 10:35:18 +0100251 return( PSA_ERROR_INVALID_ARGUMENT );
252
253 return( mbedtls_to_psa_error(
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200254 mbedtls_cipher_set_iv( &operation->ctx.cipher,
Ronald Cron8287e6b2021-03-12 10:35:18 +0100255 iv, iv_length ) ) );
256}
257
Ronald Cron6d051732020-10-01 14:10:20 +0200258/* 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. */
262static 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
342exit:
343 return( status );
344}
345
Ronald Cron8287e6b2021-03-12 10:35:18 +0100346static 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 Cron6d051732020-10-01 14:10:20 +0200352{
353 psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED;
354 size_t expected_output_size;
355
Ronald Cron6e412a72021-03-10 09:58:47 +0100356 if( ! PSA_ALG_IS_STREAM_CIPHER( operation->alg ) )
Ronald Cron6d051732020-10-01 14:10:20 +0200357 {
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-arm42cdb2a2021-04-12 15:47:35 +0200363 ( operation->ctx.cipher.unprocessed_len + input_length )
Ronald Cron6ad554c2021-03-26 09:29:09 +0100364 / operation->block_length * operation->block_length;
Ronald Cron6d051732020-10-01 14:10:20 +0200365 }
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 Cron6e412a72021-03-10 09:58:47 +0100374 if( operation->alg == PSA_ALG_ECB_NO_PADDING )
Ronald Cron6d051732020-10-01 14:10:20 +0200375 {
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-arm42cdb2a2021-04-12 15:47:35 +0200379 status = psa_cipher_update_ecb( &operation->ctx.cipher,
Ronald Cron6d051732020-10-01 14:10:20 +0200380 input,
381 input_length,
382 output,
383 output_size,
384 output_length );
385 }
386 else
387 {
388 status = mbedtls_to_psa_error(
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200389 mbedtls_cipher_update( &operation->ctx.cipher, input,
Ronald Cron6d051732020-10-01 14:10:20 +0200390 input_length, output, output_length ) );
gabor-mezei-arm58c17272021-06-29 16:41:25 +0200391
392 if( *output_length > output_size )
gabor-mezei-arm00e54f12021-06-29 19:06:30 +0200393 return( PSA_ERROR_CORRUPTION_DETECTED );
Ronald Cron6d051732020-10-01 14:10:20 +0200394 }
395
396 return( status );
397}
398
Ronald Cron8287e6b2021-03-12 10:35:18 +0100399static 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 Cron6d051732020-10-01 14:10:20 +0200403{
404 psa_status_t status = PSA_ERROR_GENERIC_ERROR;
405 uint8_t temp_output_buffer[MBEDTLS_MAX_BLOCK_LENGTH];
406
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200407 if( operation->ctx.cipher.unprocessed_len != 0 )
Ronald Cron6d051732020-10-01 14:10:20 +0200408 {
Ronald Cron6e412a72021-03-10 09:58:47 +0100409 if( operation->alg == PSA_ALG_ECB_NO_PADDING ||
410 operation->alg == PSA_ALG_CBC_NO_PADDING )
Ronald Cron6d051732020-10-01 14:10:20 +0200411 {
412 status = PSA_ERROR_INVALID_ARGUMENT;
413 goto exit;
414 }
415 }
416
417 status = mbedtls_to_psa_error(
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200418 mbedtls_cipher_finish( &operation->ctx.cipher,
Ronald Cron6d051732020-10-01 14:10:20 +0200419 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
431exit:
432 mbedtls_platform_zeroize( temp_output_buffer,
433 sizeof( temp_output_buffer ) );
434
435 return( status );
436}
437
Ronald Cron8287e6b2021-03-12 10:35:18 +0100438static psa_status_t cipher_abort( mbedtls_psa_cipher_operation_t *operation )
Ronald Cron6d051732020-10-01 14:10:20 +0200439{
Ronald Cron937dfee2021-03-10 09:17:32 +0100440 /* Sanity check (shouldn't happen: operation->alg should
441 * always have been initialized to a valid value). */
Ronald Cron6e412a72021-03-10 09:58:47 +0100442 if( ! PSA_ALG_IS_CIPHER( operation->alg ) )
Ronald Cron937dfee2021-03-10 09:17:32 +0100443 return( PSA_ERROR_BAD_STATE );
444
gabor-mezei-arm42cdb2a2021-04-12 15:47:35 +0200445 mbedtls_cipher_free( &operation->ctx.cipher );
Ronald Cron6d051732020-10-01 14:10:20 +0200446
447 return( PSA_SUCCESS );
448}
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100449
450static 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-arme5ff8f42021-06-25 15:23:05 +0200462 size_t olength, accumulated_length;
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100463
gabor-mezei-arm47a8e142021-06-25 15:44:47 +0200464 status = cipher_encrypt_setup( &operation, attributes,
465 key_buffer, key_buffer_size, alg );
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100466 if( status != PSA_SUCCESS )
467 goto exit;
468
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200469 accumulated_length = 0;
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100470 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-arme5ff8f42021-06-25 15:23:05 +0200476 accumulated_length = operation.iv_length;
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100477 }
478
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100479 status = cipher_update( &operation, input, input_length,
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200480 output + operation.iv_length,
481 output_size - operation.iv_length,
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100482 &olength );
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100483 if( status != PSA_SUCCESS )
484 goto exit;
485
gabor-mezei-arm6158e282021-06-29 16:42:13 +0200486 accumulated_length += olength;
487
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200488 status = cipher_finish( &operation, output + accumulated_length,
489 output_size - accumulated_length, &olength );
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100490 if( status != PSA_SUCCESS )
491 goto exit;
492
gabor-mezei-arm00e54f12021-06-29 19:06:30 +0200493 *output_length = accumulated_length + olength;
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200494
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100495exit:
496 if( status == PSA_SUCCESS )
497 status = cipher_abort( &operation );
498 else
499 cipher_abort( &operation );
500 return( status );
501}
502
503static 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-arme5ff8f42021-06-25 15:23:05 +0200515 size_t olength, accumulated_length;
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100516
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-arma9449a02021-03-25 11:17:10 +0100529 status = cipher_update( &operation, input + operation.iv_length,
530 input_length - operation.iv_length,
531 output, output_size, &olength );
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100532 if( status != PSA_SUCCESS )
533 goto exit;
534
gabor-mezei-arm6158e282021-06-29 16:42:13 +0200535 accumulated_length = olength;
gabor-mezei-arm258ae072021-06-25 15:25:38 +0200536
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200537 status = cipher_finish( &operation, output + accumulated_length,
538 output_size - accumulated_length, &olength );
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100539 if( status != PSA_SUCCESS )
540 goto exit;
541
gabor-mezei-arm00e54f12021-06-29 19:06:30 +0200542 *output_length = accumulated_length + olength;
gabor-mezei-arme5ff8f42021-06-25 15:23:05 +0200543
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100544exit:
545 if ( status == PSA_SUCCESS )
546 status = cipher_abort( &operation );
547 else
548 cipher_abort( &operation );
549 return( status );
550}
Ronald Cron5d9b00d2021-03-10 14:43:20 +0100551#endif /* MBEDTLS_PSA_BUILTIN_CIPHER || PSA_CRYPTO_DRIVER_TEST */
Ronald Cron6d051732020-10-01 14:10:20 +0200552
Ronald Cron5d9b00d2021-03-10 14:43:20 +0100553#if defined(MBEDTLS_PSA_BUILTIN_CIPHER)
Ronald Cron8287e6b2021-03-12 10:35:18 +0100554psa_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
564psa_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 Cron8287e6b2021-03-12 10:35:18 +0100574psa_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
581psa_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
592psa_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
600psa_status_t mbedtls_psa_cipher_abort( mbedtls_psa_cipher_operation_t *operation )
601{
602 return( cipher_abort( operation ) );
603}
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100604
605psa_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
620psa_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 Cron5d9b00d2021-03-10 14:43:20 +0100634#endif /* MBEDTLS_PSA_BUILTIN_CIPHER */
Ronald Cron8287e6b2021-03-12 10:35:18 +0100635
Ronald Cron3522e322021-03-12 11:08:49 +0100636/*
637 * BEYOND THIS POINT, TEST DRIVER ENTRY POINTS ONLY.
638 */
639
640#if defined(PSA_CRYPTO_DRIVER_TEST)
641psa_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
651psa_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 Cron3522e322021-03-12 11:08:49 +0100661psa_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
668psa_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
677psa_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
684psa_status_t mbedtls_transparent_test_driver_cipher_abort(
685 mbedtls_psa_cipher_operation_t *operation )
686{
687 return( cipher_abort( operation ) );
688}
gabor-mezei-arma9449a02021-03-25 11:17:10 +0100689
690psa_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
706psa_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 Cron3522e322021-03-12 11:08:49 +0100721#endif /* PSA_CRYPTO_DRIVER_TEST */
722
Ronald Cron0ff57952021-03-08 16:46:35 +0100723#endif /* MBEDTLS_PSA_CRYPTO_C */