From 1c1ddaa6c22e2b121ab9b9022d968bd244c8122c Mon Sep 17 00:00:00 2001
From: Ethan Looney <ethan@wolfssl.com>
Date: Thu, 25 Jun 2020 16:40:38 -0700
Subject: [PATCH] Added checks to initial returns and free rng

---
 tests/api.c | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/tests/api.c b/tests/api.c
index 4350a88cd..84da8368b 100644
--- a/tests/api.c
+++ b/tests/api.c
@@ -16122,7 +16122,9 @@ static int test_wc_curve448_make_key (void)
     printf(testingFmt, "wc_curve448_make_key()");
     
     ret = wc_curve448_init(&key);
-    ret = wc_InitRng(&rng);   
+    if (ret == 0) {
+        ret = wc_InitRng(&rng); 
+    }      
     if (ret == 0) {
     
         ret = wc_curve448_make_key(&rng, CURVE448_KEY_SIZE, &key);
@@ -16161,6 +16163,10 @@ static int test_wc_curve448_make_key (void)
             ret = 0;
         }
     }
+    
+    if (wc_FreeRng(&rng) != 0 && ret == 0) {
+        ret = WOLFSSL_FATAL_ERROR;
+    }
 
     printf(resultFmt, ret == 0 ? passed : failed);
     wc_curve448_free(&key);
@@ -16185,8 +16191,9 @@ static int test_wc_curve448_import_private_raw_ex(void)
     printf(testingFmt, "wc_curve448_import_private_raw_ex()");
        
     ret = wc_curve448_init(&key);
-    
-    ret = wc_InitRng(&rng);   
+    if (ret == 0) {
+        ret = wc_InitRng(&rng); 
+    } 
     if (ret == 0) {
     
         ret = wc_curve448_make_key(&rng, CURVE448_KEY_SIZE, &key);
@@ -16244,6 +16251,10 @@ static int test_wc_curve448_import_private_raw_ex(void)
         }
     }
     
+    if (wc_FreeRng(&rng) != 0 && ret == 0) {
+        ret = WOLFSSL_FATAL_ERROR;
+    }
+    
     printf(resultFmt, ret == 0 ? passed : failed);
     wc_curve448_free(&key);