From 40e3cac69534ff938ded90399b2db9ceb5dc56de Mon Sep 17 00:00:00 2001
From: John Safranek <john@wolfssl.com>
Date: Mon, 18 Oct 2021 09:42:57 -0700
Subject: [PATCH] Use correct value for pSz when setting the dhKeySize in the
 session.

---
 src/tls.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/tls.c b/src/tls.c
index 2e754aa97..0dd426858 100644
--- a/src/tls.c
+++ b/src/tls.c
@@ -6975,8 +6975,8 @@ static int TLSX_KeyShare_ProcessDh(WOLFSSL* ssl, KeyShareEntry* keyShareEntry)
     }
     pSz = params->p_len;
 #else
-    pSz = wc_DhGetNamedKeyMinSize(keyShareEntry->group);
-    if (pSz == 0) {
+    ret = wc_DhGetNamedKeyParamSize(keyShareEntry->group, &pSz, NULL, NULL);
+    if (ret != 0 || pSz == 0) {
         return PEER_KEY_ERROR;
     }
 #endif