Mbed-TLS/mbedtls
1
0
Fork 0
mirror of https://github.com/Mbed-TLS/mbedtls.git synced 2026-10-29 12:03:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Check the return status of the functions first

Browse Source 
Signed-off-by: gabor-mezei-arm <[email protected]>
This commit is contained in:
gabor-mezei-arm
2021-06-29 16:49:51 +02:00
parent 42373bdde7
commit 809634d5d9
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
library/psa_crypto_cipher.c
+5 -4
View File
@@ -485,16 +485,17 @@ static psa_status_t cipher_encrypt( const psa_key_attributes_t *attributes,
output + operation.iv_length,
output_size - operation.iv_length,
&olength );
accumulated_length += olength;
if( status != PSA_SUCCESS )
goto exit;
accumulated_length += olength;
status = cipher_finish( &operation, output + accumulated_length,
output_size - accumulated_length, &olength );
accumulated_length += olength;
if( status != PSA_SUCCESS )
goto exit;
accumulated_length += olength;
*output_length = accumulated_length;
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,
input_length - operation.iv_length,
output, output_size, &olength );
accumulated_length = olength;
if( status != PSA_SUCCESS )
goto exit;
accumulated_length = olength;
status = cipher_finish( &operation, output + accumulated_length,
output_size - accumulated_length, &olength );
accumulated_length += olength;
if( status != PSA_SUCCESS )
goto exit;
accumulated_length += olength;
*output_length = accumulated_length;
exit: