From 20200ee319bf2349e0b28d8b282769d8aa5b94ec Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Love=20H=C3=B6rnquist=20=C3=85strand?= <lha@kth.se>
Date: Wed, 18 Jan 2006 13:45:02 +0000
Subject: [PATCH] (DH_check_pubkey): New function.

git-svn-id: svn://svn.h5l.se/heimdal/trunk/heimdal@16584 ec53bebd-3082-4978-b11e-865c3cabbd6b
---
 lib/des/dh.c | 38 ++++++++++++++++++++++++++++++++++++--
 1 file changed, 36 insertions(+), 2 deletions(-)

diff --git a/lib/des/dh.c b/lib/des/dh.c
index 4d63e0689..a598ca72b 100644
--- a/lib/des/dh.c
+++ b/lib/des/dh.c
@@ -150,9 +150,43 @@ DH_generate_parameters_ex(DH *dh, int prime_len, int generator, BN_GENCB *cb)
 }
 
 int
-DH_check(const DH *dh, int *num)
+DH_check_pubkey(const DH *dh, const BIGNUM *pub_key, int *codes)
 {
-    return 1; /* XXX */
+    BIGNUM *bn = NULL, *sum = NULL;
+    int ret = 0;
+
+    *codes = 0;
+
+    bn = BN_new();
+    if (bn == NULL)
+	goto out;
+
+    if (!BN_set_word(bn, 1))
+	goto out;
+
+    if (BN_cmp(bn, pub_key) >= 0)
+	*codes |= DH_CHECK_PUBKEY_TOO_SMALL;
+
+    if (!BN_set_word(bn, 2))
+	goto out;
+
+    sum = BN_new();
+    if (sum == NULL)
+	goto out;
+
+    BN_uadd(sum, pub_key, bn);
+
+    if (BN_cmp(sum, dh->p) >= 0)
+	*codes |= DH_CHECK_PUBKEY_TOO_LARGE;
+
+    ret = 1;
+out:
+    if (bn)
+	BN_free(bn);
+    if (sum)
+	BN_free(sum);
+
+    return ret;
 }
 
 int