Provide {DH,DSA}_set0_key(). Requested by sthen.

ok jsing

1 files changed, 19 insertions, 1 deletions

diff --git a/src/lib/libcrypto/dsa/dsa_lib.c b/src/lib/libcrypto/dsa/dsa_lib.c
index 2dec8567f5..772c939d31 100644
--- a/src/lib/libcrypto/dsa/dsa_lib.c
+++ b/src/lib/libcrypto/dsa/dsa_lib.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: dsa_lib.c,v 1.25 2018/02/18 12:50:58 tb Exp $ */
+/* $OpenBSD: dsa_lib.c,v 1.26 2018/02/18 14:58:12 tb Exp $ */
 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
  * All rights reserved.
  *
@@ -346,3 +346,21 @@ DSA_get0_key(const DSA *d, const BIGNUM **pub_key, const BIGNUM **priv_key)
         if (priv_key != NULL)
                 *priv_key = d->priv_key;
 }
+
+int
+DSA_set0_key(DSA *d, BIGNUM *pub_key, BIGNUM *priv_key)
+{
+        if (d->pub_key == NULL && pub_key == NULL)
+                return 0;
+
+        if (pub_key != NULL) {
+                BN_free(d->pub_key);
+                d->pub_key = pub_key;
+        }
+        if (priv_key != NULL) {
+                BN_free(d->priv_key);
+                d->priv_key = priv_key;
+        }
+
+        return 1;
+}