From cedec3ae28d1bb2177604b5668c6709946fa77d5 Mon Sep 17 00:00:00 2001
From: Tesfa Mael <tesfa@wolfssl.com>
Date: Mon, 28 Dec 2020 15:53:56 -0800
Subject: [PATCH] Add parameter check

---
 wolfcrypt/src/random.c | 2 +-
 wolfcrypt/src/sp_int.c | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/wolfcrypt/src/random.c b/wolfcrypt/src/random.c
index e8070a355..05ad76c7b 100644
--- a/wolfcrypt/src/random.c
+++ b/wolfcrypt/src/random.c
@@ -914,7 +914,7 @@ int wc_RNG_GenerateBlock(WC_RNG* rng, byte* output, word32 sz)
 {
     int ret;
 
-    if (rng == NULL || output == NULL)
+    if (rng == NULL || output == NULL || sz == 0 )
         return BAD_FUNC_ARG;
 
 #ifdef WOLF_CRYPTO_CB
diff --git a/wolfcrypt/src/sp_int.c b/wolfcrypt/src/sp_int.c
index 53fe07a11..7e436caef 100644
--- a/wolfcrypt/src/sp_int.c
+++ b/wolfcrypt/src/sp_int.c
@@ -12979,7 +12979,7 @@ int sp_rand_prime(sp_int* r, int len, WC_RNG* rng, void* heap)
 
     (void)heap;
 
-    if ((r == NULL) || (rng == NULL)) {
+    if ((r == NULL) || (rng == NULL) || len <= 0 ) {
         err = MP_VAL;
     }