wolfSSL/wolfssl
1
0
Fork 1
mirror of https://github.com/wolfSSL/wolfssl.git synced 2025-08-01 11:44:38 +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

8
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,6 +17789,7 @@ WOLFSSL_TEST_SUBROUTINE int dh_test(void)
done: done:
#ifndef WC_NO_RNG #ifndef WC_NO_RNG
if (rngInit)
wc_FreeRng(&rng); wc_FreeRng(&rng);
#endif #endif
@@ -17796,6 +17800,7 @@ done:
XFREE(key, HEAP_HINT, DYNAMIC_TYPE_TMP_BUFFER); XFREE(key, HEAP_HINT, DYNAMIC_TYPE_TMP_BUFFER);
} }
if (key2) { if (key2) {
if (key2Init)
wc_FreeDhKey(key2); wc_FreeDhKey(key2);
XFREE(key2, HEAP_HINT, DYNAMIC_TYPE_TMP_BUFFER); XFREE(key2, HEAP_HINT, DYNAMIC_TYPE_TMP_BUFFER);
} }
@@ -17816,6 +17821,7 @@ done:
#else #else
if (keyInit) if (keyInit)
wc_FreeDhKey(key); wc_FreeDhKey(key);
if (key2Init)
wc_FreeDhKey(key2); wc_FreeDhKey(key2);
#endif #endif