feedc0de/wolfssl
1
0
Fork 0
forked from wolfSSL/wolfssl
Code Pull Requests Activity

Review comments

Browse Source
This commit is contained in:
Tesfa Mael
2022-05-17 09:21:20 -07:00
parent 74b3df4f6c
commit d6935ed3be
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
wolfcrypt/src/rsa.c
View File

@@ -886,14 +886,15 @@ static int RsaMGF1(enum wc_HashType hType, byte* seed, word32 seedSz,
/* find largest amount of memory needed which will be the max of /* find largest amount of memory needed which will be the max of
* hLen and (seedSz + 4) since tmp is used to store the hash digest */ * hLen and (seedSz + 4) since tmp is used to store the hash digest */
tmpSz = ((seedSz + 4) > (word32)hLen)? seedSz + 4: (word32)hLen; tmpSz = ((seedSz + 4) > (word32)hLen)? seedSz + 4: (word32)hLen;
if (tmpSz > RSA_MAX_SIZE/8)
return BAD_FUNC_ARG;
#if defined(WOLFSSL_SMALL_STACK) && !defined(WOLFSSL_NO_MALLOC) #if defined(WOLFSSL_SMALL_STACK) && !defined(WOLFSSL_NO_MALLOC)
tmp = (byte*)XMALLOC(tmpSz, heap, DYNAMIC_TYPE_RSA_BUFFER); tmp = (byte*)XMALLOC(tmpSz, heap, DYNAMIC_TYPE_RSA_BUFFER);
if (tmp == NULL) { if (tmp == NULL) {
return MEMORY_E; return MEMORY_E;
} }
tmpF = 1; /* make sure to free memory when done */ tmpF = 1; /* make sure to free memory when done */
#else
if (tmpSz > RSA_MAX_SIZE/8)
return BAD_FUNC_ARG;
#endif #endif
} }
else { else {