wc_MakeRsaKey(): add missing allocation failure checks for WOLFSSL_SMALL_STACK.

This commit is contained in:
Daniel Pouzzner
2020-09-21 18:26:45 -05:00
parent 0f3283ce7b
commit 6a7a87545a

View File

@ -3968,6 +3968,17 @@ int wc_MakeRsaKey(RsaKey* key, int size, long e, WC_RNG* rng)
int err, i, failCount, primeSz, isPrime = 0;
byte* buf = NULL;
#ifdef WOLFSSL_SMALL_STACK
if ((p == NULL) ||
(q == NULL) ||
(tmp1 == NULL) ||
(tmp2 == NULL) ||
(tmp3 == NULL)) {
err = MEMORY_E;
goto out;
}
#endif
if (key == NULL || rng == NULL) {
err = BAD_FUNC_ARG;
goto out;