Revise how output allocation is checked

Signed-off-by: Thomas Daubney <thomas.daubney@arm.com>
This commit is contained in:
Thomas Daubney 2024-02-21 15:15:00 +00:00
parent 89d8c2a1b4
commit 0576a6a174

View File

@ -7429,10 +7429,7 @@ psa_status_t psa_raw_key_agreement(psa_algorithm_t alg,
output_length);
exit:
/* Check for successful allocation of output. */
if (output != NULL && status != PSA_ERROR_INSUFFICIENT_MEMORY) {
/* output allocated. */
if (status != PSA_SUCCESS) {
if (output != NULL && status != PSA_SUCCESS) {
/* If an error happens and is not handled properly, the output
* may be used as a key to protect sensitive data. Arrange for such
* a key to be random, which is likely to result in decryption or
@ -7442,7 +7439,6 @@ exit:
*/
psa_generate_random(output, output_size);
*output_length = output_size;
}
} else {
/* output allocation failed. */
*output_length = 0;