commit 3f6cdd7aab90fa6b4830349916d994a05d6f8f79
author Raef Coles <raef.coles@arm.com> Fri Oct 07 14:07:59 2022 +0100
committer Raef Coles <raef.coles@arm.com> Thu Oct 13 14:29:24 2022 +0100
tree 4f9845b096872ccb3e0bd72f35a95f5ed7cc3bd3
parent 02cf8234b4d005ecc858bb5f913272b6f70b1bb8

Fix LMS not checking RNG function return value

Signed-off-by: Raef Coles <raef.coles@arm.com>

library/lms.c

diff --git a/library/lms.c b/library/lms.c
index 1e0f319..a4235ad 100644
--- a/library/lms.c
+++ b/library/lms.c
@@ -545,9 +545,13 @@
     ctx->params.type = type;
     ctx->params.otstype = otstype;
 
-    f_rng( p_rng,
-           ctx->params.I_key_identifier,
-           MBEDTLS_LMOTS_I_KEY_ID_LEN );
+    ret = f_rng( p_rng,
+                 ctx->params.I_key_identifier,
+                 MBEDTLS_LMOTS_I_KEY_ID_LEN );
+    if( ret != 0 )
+    {
+        goto exit;
+    }
 
     ctx->ots_private_keys = mbedtls_calloc( ( size_t )MERKLE_TREE_LEAF_NODE_AM(ctx->params.type),
                                             sizeof( *ctx->ots_private_keys ) );