Check the return status of the functions first

Signed-off-by: gabor-mezei-arm <gabor.mezei@arm.com>
This commit is contained in:
gabor-mezei-arm 2021-06-29 16:42:13 +02:00
parent 42373bdde7
commit 809634d5d9
No known key found for this signature in database
GPG Key ID: 106F5A41ECC305BD

View File

@ -485,16 +485,17 @@ static psa_status_t cipher_encrypt( const psa_key_attributes_t *attributes,
output + operation.iv_length, output + operation.iv_length,
output_size - operation.iv_length, output_size - operation.iv_length,
&olength ); &olength );
accumulated_length += olength;
if( status != PSA_SUCCESS ) if( status != PSA_SUCCESS )
goto exit; goto exit;
accumulated_length += olength;
status = cipher_finish( &operation, output + accumulated_length, status = cipher_finish( &operation, output + accumulated_length,
output_size - accumulated_length, &olength ); output_size - accumulated_length, &olength );
accumulated_length += olength;
if( status != PSA_SUCCESS ) if( status != PSA_SUCCESS )
goto exit; goto exit;
accumulated_length += olength;
*output_length = accumulated_length; *output_length = accumulated_length;
exit: exit:
@ -534,17 +535,17 @@ static psa_status_t cipher_decrypt( const psa_key_attributes_t *attributes,
status = cipher_update( &operation, input + operation.iv_length, status = cipher_update( &operation, input + operation.iv_length,
input_length - operation.iv_length, input_length - operation.iv_length,
output, output_size, &olength ); output, output_size, &olength );
accumulated_length = olength;
if( status != PSA_SUCCESS ) if( status != PSA_SUCCESS )
goto exit; goto exit;
accumulated_length = olength;
status = cipher_finish( &operation, output + accumulated_length, status = cipher_finish( &operation, output + accumulated_length,
output_size - accumulated_length, &olength ); output_size - accumulated_length, &olength );
accumulated_length += olength;
if( status != PSA_SUCCESS ) if( status != PSA_SUCCESS )
goto exit; goto exit;
accumulated_length += olength;
*output_length = accumulated_length; *output_length = accumulated_length;
exit: exit: