mirror of
https://github.com/yuzu-emu/mbedtls.git
synced 2024-11-26 14:35:37 +01:00
Fix cleanup in psa_cipher_setup
In some error cases, psa_cipher_setup was leaving a partly-initialized operation context.
This commit is contained in:
parent
f426e0f303
commit
9ab61b603d
@ -2902,8 +2902,8 @@ static psa_status_t psa_cipher_setup( psa_cipher_operation_t *operation,
|
|||||||
psa_algorithm_t alg,
|
psa_algorithm_t alg,
|
||||||
mbedtls_operation_t cipher_operation )
|
mbedtls_operation_t cipher_operation )
|
||||||
{
|
{
|
||||||
int ret = MBEDTLS_ERR_CIPHER_FEATURE_UNAVAILABLE;
|
int ret = 0;
|
||||||
psa_status_t status;
|
psa_status_t status = PSA_ERROR_GENERIC_ERROR;
|
||||||
psa_key_slot_t *slot;
|
psa_key_slot_t *slot;
|
||||||
size_t key_bits;
|
size_t key_bits;
|
||||||
const mbedtls_cipher_info_t *cipher_info = NULL;
|
const mbedtls_cipher_info_t *cipher_info = NULL;
|
||||||
@ -2923,19 +2923,19 @@ static psa_status_t psa_cipher_setup( psa_cipher_operation_t *operation,
|
|||||||
|
|
||||||
status = psa_get_key_from_slot( handle, &slot, usage, alg);
|
status = psa_get_key_from_slot( handle, &slot, usage, alg);
|
||||||
if( status != PSA_SUCCESS )
|
if( status != PSA_SUCCESS )
|
||||||
return( status );
|
goto exit;
|
||||||
key_bits = psa_get_key_bits( slot );
|
key_bits = psa_get_key_bits( slot );
|
||||||
|
|
||||||
cipher_info = mbedtls_cipher_info_from_psa( alg, slot->type, key_bits, NULL );
|
cipher_info = mbedtls_cipher_info_from_psa( alg, slot->type, key_bits, NULL );
|
||||||
if( cipher_info == NULL )
|
if( cipher_info == NULL )
|
||||||
return( PSA_ERROR_NOT_SUPPORTED );
|
{
|
||||||
|
status = PSA_ERROR_NOT_SUPPORTED;
|
||||||
|
goto exit;
|
||||||
|
}
|
||||||
|
|
||||||
ret = mbedtls_cipher_setup( &operation->ctx.cipher, cipher_info );
|
ret = mbedtls_cipher_setup( &operation->ctx.cipher, cipher_info );
|
||||||
if( ret != 0 )
|
if( ret != 0 )
|
||||||
{
|
goto exit;
|
||||||
psa_cipher_abort( operation );
|
|
||||||
return( mbedtls_to_psa_error( ret ) );
|
|
||||||
}
|
|
||||||
|
|
||||||
#if defined(MBEDTLS_DES_C)
|
#if defined(MBEDTLS_DES_C)
|
||||||
if( slot->type == PSA_KEY_TYPE_DES && key_bits == 128 )
|
if( slot->type == PSA_KEY_TYPE_DES && key_bits == 128 )
|
||||||
@ -2956,10 +2956,7 @@ static psa_status_t psa_cipher_setup( psa_cipher_operation_t *operation,
|
|||||||
(int) key_bits, cipher_operation );
|
(int) key_bits, cipher_operation );
|
||||||
}
|
}
|
||||||
if( ret != 0 )
|
if( ret != 0 )
|
||||||
{
|
goto exit;
|
||||||
psa_cipher_abort( operation );
|
|
||||||
return( mbedtls_to_psa_error( ret ) );
|
|
||||||
}
|
|
||||||
|
|
||||||
#if defined(MBEDTLS_CIPHER_MODE_WITH_PADDING)
|
#if defined(MBEDTLS_CIPHER_MODE_WITH_PADDING)
|
||||||
switch( alg )
|
switch( alg )
|
||||||
@ -2978,10 +2975,7 @@ static psa_status_t psa_cipher_setup( psa_cipher_operation_t *operation,
|
|||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
if( ret != 0 )
|
if( ret != 0 )
|
||||||
{
|
goto exit;
|
||||||
psa_cipher_abort( operation );
|
|
||||||
return( mbedtls_to_psa_error( ret ) );
|
|
||||||
}
|
|
||||||
#endif //MBEDTLS_CIPHER_MODE_WITH_PADDING
|
#endif //MBEDTLS_CIPHER_MODE_WITH_PADDING
|
||||||
|
|
||||||
operation->key_set = 1;
|
operation->key_set = 1;
|
||||||
@ -2992,7 +2986,12 @@ static psa_status_t psa_cipher_setup( psa_cipher_operation_t *operation,
|
|||||||
operation->iv_size = PSA_BLOCK_CIPHER_BLOCK_SIZE( slot->type );
|
operation->iv_size = PSA_BLOCK_CIPHER_BLOCK_SIZE( slot->type );
|
||||||
}
|
}
|
||||||
|
|
||||||
return( PSA_SUCCESS );
|
exit:
|
||||||
|
if( status == 0 )
|
||||||
|
status = mbedtls_to_psa_error( ret );
|
||||||
|
if( status != 0 )
|
||||||
|
psa_cipher_abort( operation );
|
||||||
|
return( status );
|
||||||
}
|
}
|
||||||
|
|
||||||
psa_status_t psa_cipher_encrypt_setup( psa_cipher_operation_t *operation,
|
psa_status_t psa_cipher_encrypt_setup( psa_cipher_operation_t *operation,
|
||||||
|
Loading…
Reference in New Issue
Block a user