From 0d5378135b4336cdc86d58b1c4336f1becc44642 Mon Sep 17 00:00:00 2001
From: Colton Willey <colton@wolfssl.com>
Date: Wed, 14 May 2025 14:32:23 -0700
Subject: [PATCH] Add get private key handling for DH

---
 src/wp_dh_kmgmt.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/wp_dh_kmgmt.c b/src/wp_dh_kmgmt.c
index cfa9d4f2..47bc78d8 100644
--- a/src/wp_dh_kmgmt.c
+++ b/src/wp_dh_kmgmt.c
@@ -749,6 +749,10 @@ static int wp_dh_get_params(wp_Dh* dh, OSSL_PARAM params[])
             dh->pub, dh->pubSz))) {
         ok = 0;
     }
+    if (ok && (!wp_params_set_octet_string_be(params, OSSL_PKEY_PARAM_PRIV_KEY,
+            dh->priv, dh->privSz))) {
+        ok = 0;
+    }
     if (ok && (!wp_dh_get_params_encoded_public_key(dh, params))) {
         ok = 0;
     }