wolfSSL/wolfssl
1
0
Fork 1
mirror of https://github.com/wolfSSL/wolfssl.git synced 2025-07-30 18:57:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5719 from ejohnstown/dh-test-key2

Browse Source
This commit is contained in:
Hayden Roche
2022-10-20 10:29:26 -07:00
committed by GitHub
parent 86605c92ce 31222618b9
commit 3591ce773c
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
wolfcrypt/test/test.c
View File

@ -17412,8 +17412,9 @@ WOLFSSL_TEST_SUBROUTINE int dh_test(void)
word32 idx = 0, privSz, pubSz, privSz2, pubSz2; word32 idx = 0, privSz, pubSz, privSz2, pubSz2;
#ifndef WC_NO_RNG #ifndef WC_NO_RNG
WC_RNG rng; WC_RNG rng;
int rngInit = 0;
#endif #endif
int keyInit = 0; int keyInit = 0, key2Init = 0;
#define DH_TEST_TMP_SIZE 1024 #define DH_TEST_TMP_SIZE 1024
#if !defined(USE_CERT_BUFFERS_3072) && !defined(USE_CERT_BUFFERS_4096) #if !defined(USE_CERT_BUFFERS_3072) && !defined(USE_CERT_BUFFERS_4096)
@ -17521,6 +17522,7 @@ WOLFSSL_TEST_SUBROUTINE int dh_test(void)
if (ret != 0) { if (ret != 0) {
ERROR_OUT(-8105, done); ERROR_OUT(-8105, done);
} }
key2Init = 1;
#ifdef NO_ASN #ifdef NO_ASN
#ifndef WOLFSSL_SP_MATH #ifndef WOLFSSL_SP_MATH
@ -17568,6 +17570,7 @@ WOLFSSL_TEST_SUBROUTINE int dh_test(void)
if (ret != 0) { if (ret != 0) {
ERROR_OUT(-8110, done); ERROR_OUT(-8110, done);
} }
rngInit = 1;
ret = wc_DhGenerateKeyPair(key, &rng, priv, &privSz, pub, &pubSz); ret = wc_DhGenerateKeyPair(key, &rng, priv, &privSz, pub, &pubSz);
#if defined(WOLFSSL_ASYNC_CRYPT) #if defined(WOLFSSL_ASYNC_CRYPT)
@ -17786,7 +17789,8 @@ WOLFSSL_TEST_SUBROUTINE int dh_test(void)
done: done:
#ifndef WC_NO_RNG #ifndef WC_NO_RNG
wc_FreeRng(&rng); if (rngInit)
wc_FreeRng(&rng);
#endif #endif
#if defined(WOLFSSL_SMALL_STACK) && !defined(WOLFSSL_NO_MALLOC) #if defined(WOLFSSL_SMALL_STACK) && !defined(WOLFSSL_NO_MALLOC)
@ -17796,7 +17800,8 @@ done:
XFREE(key, HEAP_HINT, DYNAMIC_TYPE_TMP_BUFFER); XFREE(key, HEAP_HINT, DYNAMIC_TYPE_TMP_BUFFER);
} }
if (key2) { if (key2) {
wc_FreeDhKey(key2); if (key2Init)
wc_FreeDhKey(key2);
XFREE(key2, HEAP_HINT, DYNAMIC_TYPE_TMP_BUFFER); XFREE(key2, HEAP_HINT, DYNAMIC_TYPE_TMP_BUFFER);
} }
if (tmp) if (tmp)
@ -17816,7 +17821,8 @@ done:
#else #else
if (keyInit) if (keyInit)
wc_FreeDhKey(key); wc_FreeDhKey(key);
wc_FreeDhKey(key2); if (key2Init)
wc_FreeDhKey(key2);
#endif #endif
(void)privSz; (void)privSz;