From 937b90febfc3d54373fe2aa2e0265c135cb37ca7 Mon Sep 17 00:00:00 2001
From: Przemek Stekiel <przemyslaw.stekiel@mobica.com>
Date: Wed, 20 Apr 2022 08:33:13 +0200
Subject: [PATCH] Add null check for pms allocation

Signed-off-by: Przemek Stekiel <przemyslaw.stekiel@mobica.com>
---
 library/psa_crypto.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 7f5a7f894..1cb00c801 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -5284,6 +5284,8 @@ static psa_status_t psa_tls12_prf_psk_to_ms_set_key(
         return( PSA_ERROR_INVALID_ARGUMENT );
 
     uint8_t *pms = mbedtls_calloc( 1, pms_len );
+    if( pms == NULL )
+        return( PSA_ERROR_INSUFFICIENT_MEMORY );
     uint8_t *cur = pms;
 
     /* pure-PSK: