crypto/dh/dh_lib.c - third_party/openssl - Git at Google

 /*
  * Copyright 1995-2020 The OpenSSL Project Authors. All Rights Reserved.
  *
  * Licensed under the Apache License 2.0 (the "License").  You may not use
  * this file except in compliance with the License.  You can obtain a copy
  * in the file LICENSE in the source distribution or at
  * https://www.openssl.org/source/license.html
  */

 /*
  * DH low level APIs are deprecated for public use, but still ok for
  * internal use.
  */
 #include "internal/deprecated.h"

 #include <stdio.h>
 #include <openssl/bn.h>
 #include <openssl/engine.h>
 #include <openssl/obj_mac.h>
 #include <openssl/core_names.h>
 #include "internal/cryptlib.h"
 #include "internal/refcount.h"
 #include "crypto/evp.h"
 #include "crypto/dh.h"
 #include "dh_local.h"

 static DH *dh_new_intern(ENGINE *engine, OPENSSL_CTX *libctx);

 #ifndef FIPS_MODE
 int DH_set_method(DH *dh, const DH_METHOD *meth)
 {
     /*
      * NB: The caller is specifically setting a method, so it's not up to us
      * to deal with which ENGINE it comes from.
      */
     const DH_METHOD *mtmp;
     mtmp = dh->meth;
     if (mtmp->finish)
         mtmp->finish(dh);
 #ifndef OPENSSL_NO_ENGINE
     ENGINE_finish(dh->engine);
     dh->engine = NULL;
 #endif
     dh->meth = meth;
     if (meth->init)
         meth->init(dh);
     return 1;
 }

 const DH_METHOD *dh_get_method(const DH *dh)
 {
     return dh->meth;
 }

 DH *DH_new(void)
 {
     return dh_new_intern(NULL, NULL);
 }

 DH *DH_new_method(ENGINE *engine)
 {
     return dh_new_intern(engine, NULL);
 }
 #endif /* !FIPS_MODE */

 DH *dh_new_with_libctx(OPENSSL_CTX *libctx)
 {
     return dh_new_intern(NULL, libctx);
 }

 static DH *dh_new_intern(ENGINE *engine, OPENSSL_CTX *libctx)
 {
     DH *ret = OPENSSL_zalloc(sizeof(*ret));

     if (ret == NULL) {
         DHerr(0, ERR_R_MALLOC_FAILURE);
         return NULL;
     }

     ret->references = 1;
     ret->lock = CRYPTO_THREAD_lock_new();
     if (ret->lock == NULL) {
         DHerr(0, ERR_R_MALLOC_FAILURE);
         OPENSSL_free(ret);
         return NULL;
     }

     ret->libctx = libctx;
     ret->meth = DH_get_default_method();
 #if !defined(FIPS_MODE) && !defined(OPENSSL_NO_ENGINE)
     ret->flags = ret->meth->flags;  /* early default init */
     if (engine) {
         if (!ENGINE_init(engine)) {
             DHerr(0, ERR_R_ENGINE_LIB);
             goto err;
         }
         ret->engine = engine;
     } else
         ret->engine = ENGINE_get_default_DH();
     if (ret->engine) {
         ret->meth = ENGINE_get_DH(ret->engine);
         if (ret->meth == NULL) {
             DHerr(0, ERR_R_ENGINE_LIB);
             goto err;
         }
     }
 #endif

     ret->flags = ret->meth->flags;

 #ifndef FIPS_MODE
     if (!CRYPTO_new_ex_data(CRYPTO_EX_INDEX_DH, ret, &ret->ex_data))
         goto err;
 #endif /* FIPS_MODE */

     if ((ret->meth->init != NULL) && !ret->meth->init(ret)) {
         DHerr(0, ERR_R_INIT_FAIL);
         goto err;
     }

     return ret;

  err:
     DH_free(ret);
     return NULL;
 }

 void DH_free(DH *r)
 {
     int i;

     if (r == NULL)
         return;

     CRYPTO_DOWN_REF(&r->references, &i, r->lock);
     REF_PRINT_COUNT("DH", r);
     if (i > 0)
         return;
     REF_ASSERT_ISNT(i < 0);

     if (r->meth != NULL && r->meth->finish != NULL)
         r->meth->finish(r);
 #if !defined(FIPS_MODE)
 # if !defined(OPENSSL_NO_ENGINE)
     ENGINE_finish(r->engine);
 # endif
     CRYPTO_free_ex_data(CRYPTO_EX_INDEX_DH, r, &r->ex_data);
 #endif

     CRYPTO_THREAD_lock_free(r->lock);

     ffc_params_cleanup(&r->params);
     BN_clear_free(r->pub_key);
     BN_clear_free(r->priv_key);
     OPENSSL_free(r);
 }

 int DH_up_ref(DH *r)
 {
     int i;

     if (CRYPTO_UP_REF(&r->references, &i, r->lock) <= 0)
         return 0;

     REF_PRINT_COUNT("DH", r);
     REF_ASSERT_ISNT(i < 2);
     return ((i > 1) ? 1 : 0);
 }

 #ifndef FIPS_MODE
 int DH_set_ex_data(DH *d, int idx, void *arg)
 {
     return CRYPTO_set_ex_data(&d->ex_data, idx, arg);
 }

 void *DH_get_ex_data(const DH *d, int idx)
 {
     return CRYPTO_get_ex_data(&d->ex_data, idx);
 }
 #endif

 int DH_bits(const DH *dh)
 {
     return BN_num_bits(dh->params.p);
 }

 int DH_size(const DH *dh)
 {
     return BN_num_bytes(dh->params.p);
 }

 int DH_security_bits(const DH *dh)
 {
     int N;
     if (dh->params.q != NULL)
         N = BN_num_bits(dh->params.q);
     else if (dh->length)
         N = dh->length;
     else
         N = -1;
     return BN_security_bits(BN_num_bits(dh->params.p), N);
 }

 void DH_get0_pqg(const DH *dh,
                  const BIGNUM **p, const BIGNUM **q, const BIGNUM **g)
 {
     ffc_params_get0_pqg(&dh->params, p, q, g);
 }

 int DH_set0_pqg(DH *dh, BIGNUM *p, BIGNUM *q, BIGNUM *g)
 {
     /*
      * If the fields p and g in d are NULL, the corresponding input
      * parameters MUST be non-NULL.  q may remain NULL.
      */
     if ((dh->params.p == NULL && p == NULL)
         || (dh->params.g == NULL && g == NULL))
         return 0;

     ffc_params_set0_pqg(&dh->params, p, q, g);
     dh_cache_named_group(dh);
     if (q != NULL)
         dh->length = BN_num_bits(q);
     /*
      * Check if this is a named group. If it finds a named group then the
      * 'q' and 'length' value are either already set or are set by the
      * call.
      */
     if (DH_get_nid(dh) == NID_undef) {
         /* If its not a named group then set the 'length' if q is not NULL */
         if (q != NULL)
             dh->length = BN_num_bits(q);
     }
     dh->dirty_cnt++;
     return 1;
 }

 long DH_get_length(const DH *dh)
 {
     return dh->length;
 }

 int DH_set_length(DH *dh, long length)
 {
     dh->length = length;
     return 1;
 }

 void DH_get0_key(const DH *dh, const BIGNUM **pub_key, const BIGNUM **priv_key)
 {
     if (pub_key != NULL)
         *pub_key = dh->pub_key;
     if (priv_key != NULL)
         *priv_key = dh->priv_key;
 }

 int DH_set0_key(DH *dh, BIGNUM *pub_key, BIGNUM *priv_key)
 {
     if (pub_key != NULL) {
         BN_clear_free(dh->pub_key);
         dh->pub_key = pub_key;
     }
     if (priv_key != NULL) {
         BN_clear_free(dh->priv_key);
         dh->priv_key = priv_key;
         dh->length = BN_num_bits(priv_key);
     }

     dh->dirty_cnt++;
     return 1;
 }

 const BIGNUM *DH_get0_p(const DH *dh)
 {
     return dh->params.p;
 }

 const BIGNUM *DH_get0_q(const DH *dh)
 {
     return dh->params.q;
 }

 const BIGNUM *DH_get0_g(const DH *dh)
 {
     return dh->params.g;
 }

 const BIGNUM *DH_get0_priv_key(const DH *dh)
 {
     return dh->priv_key;
 }

 const BIGNUM *DH_get0_pub_key(const DH *dh)
 {
     return dh->pub_key;
 }

 void DH_clear_flags(DH *dh, int flags)
 {
     dh->flags &= ~flags;
 }

 int DH_test_flags(const DH *dh, int flags)
 {
     return dh->flags & flags;
 }

 void DH_set_flags(DH *dh, int flags)
 {
     dh->flags |= flags;
 }

 #ifndef FIPS_MODE
 ENGINE *DH_get0_engine(DH *dh)
 {
     return dh->engine;
 }
 #endif /*FIPS_MODE */

 FFC_PARAMS *dh_get0_params(DH *dh)
 {
     return &dh->params;
 }
 int dh_get0_nid(const DH *dh)
 {
     return dh->params.nid;
 }

 int dh_ffc_params_fromdata(DH *dh, const OSSL_PARAM params[])
 {
     int ret;
     FFC_PARAMS *ffc;

     if (dh == NULL)
         return 0;
     ffc = dh_get0_params(dh);
     if (ffc == NULL)
         return 0;

     ret = ffc_params_fromdata(ffc, params);
     if (ret) {
         dh_cache_named_group(dh);
         dh->dirty_cnt++;
     }
     return ret;
 }

 static int dh_paramgen_check(EVP_PKEY_CTX *ctx)
 {
     if (ctx == NULL || !EVP_PKEY_CTX_IS_GEN_OP(ctx)) {
         ERR_raise(ERR_LIB_EVP, EVP_R_COMMAND_NOT_SUPPORTED);
         /* Uses the same return values as EVP_PKEY_CTX_ctrl */
         return -2;
     }
     /* If key type not DH return error */
     if (ctx->pmeth != NULL
         && ctx->pmeth->pkey_id != EVP_PKEY_DH
         && ctx->pmeth->pkey_id != EVP_PKEY_DHX)
         return -1;
     return 1;
 }

 int EVP_PKEY_CTX_set_dh_paramgen_gindex(EVP_PKEY_CTX *ctx, int gindex)
 {
     int ret;
     OSSL_PARAM params[2], *p = params;

     if ((ret = dh_paramgen_check(ctx)) <= 0)
         return ret;

     *p++ = OSSL_PARAM_construct_int(OSSL_PKEY_PARAM_FFC_GINDEX, &gindex);
     *p++ = OSSL_PARAM_construct_end();

     return EVP_PKEY_CTX_set_params(ctx, params);
 }

 int EVP_PKEY_CTX_set_dh_paramgen_seed(EVP_PKEY_CTX *ctx,
                                       const unsigned char *seed,
                                       size_t seedlen)
 {
     int ret;
     OSSL_PARAM params[2], *p = params;

     if ((ret = dh_paramgen_check(ctx)) <= 0)
         return ret;

     *p++ = OSSL_PARAM_construct_octet_string(OSSL_PKEY_PARAM_FFC_SEED,
                                              (void *)seed, seedlen);
     *p++ = OSSL_PARAM_construct_end();

     return EVP_PKEY_CTX_set_params(ctx, params);
 }

 int EVP_PKEY_CTX_set_dh_paramgen_type(EVP_PKEY_CTX *ctx, int typ)
 {
     int ret;
     OSSL_PARAM params[2], *p = params;
     const char *name;

     if ((ret = dh_paramgen_check(ctx)) <= 0)
         return ret;

 #if !defined(FIPS_MODE)
     /* TODO(3.0): Remove this eventually when no more legacy */
     if (ctx->op.keymgmt.genctx == NULL)
         return EVP_PKEY_CTX_ctrl(ctx, EVP_PKEY_DH, EVP_PKEY_OP_PARAMGEN,
                                  EVP_PKEY_CTRL_DH_PARAMGEN_TYPE, typ, NULL);
 #endif

     name = dh_gen_type_id2name(typ);
     if (name == NULL)
         return 0;
     *p++ = OSSL_PARAM_construct_utf8_string(OSSL_PKEY_PARAM_FFC_TYPE,
                                             (char *) name, 0);
     *p++ = OSSL_PARAM_construct_end();

     return EVP_PKEY_CTX_set_params(ctx, params);
 }

 int EVP_PKEY_CTX_set_dh_paramgen_prime_len(EVP_PKEY_CTX *ctx, int pbits)
 {
     int ret;
     OSSL_PARAM params[2], *p = params;
     size_t bits = pbits;

     if ((ret = dh_paramgen_check(ctx)) <= 0)
         return ret;

 #if !defined(FIPS_MODE)
     /* TODO(3.0): Remove this eventually when no more legacy */
     if (ctx->op.keymgmt.genctx == NULL)
         return EVP_PKEY_CTX_ctrl(ctx, EVP_PKEY_DH, EVP_PKEY_OP_PARAMGEN,
                                  EVP_PKEY_CTRL_DH_PARAMGEN_PRIME_LEN, pbits,
                                  NULL);
 #endif
     *p++ = OSSL_PARAM_construct_size_t(OSSL_PKEY_PARAM_FFC_PBITS, &bits);
     *p++ = OSSL_PARAM_construct_end();
     return EVP_PKEY_CTX_set_params(ctx, params);
 }

 int EVP_PKEY_CTX_set_dh_paramgen_subprime_len(EVP_PKEY_CTX *ctx, int qbits)
 {
     int ret;
     OSSL_PARAM params[2], *p = params;
     size_t bits2 = qbits;

     if ((ret = dh_paramgen_check(ctx)) <= 0)
         return ret;

 #if !defined(FIPS_MODE)
     /* TODO(3.0): Remove this eventually when no more legacy */
     if (ctx->op.keymgmt.genctx == NULL)
         return EVP_PKEY_CTX_ctrl(ctx, EVP_PKEY_DH, EVP_PKEY_OP_PARAMGEN,
                                  EVP_PKEY_CTRL_DH_PARAMGEN_SUBPRIME_LEN, qbits,
                                  NULL);
 #endif
     *p++ = OSSL_PARAM_construct_size_t(OSSL_PKEY_PARAM_FFC_QBITS, &bits2);
     *p++ = OSSL_PARAM_construct_end();

     return EVP_PKEY_CTX_set_params(ctx, params);
 }

 int EVP_PKEY_CTX_set_dh_paramgen_generator(EVP_PKEY_CTX *ctx, int gen)
 {
     int ret;
     OSSL_PARAM params[2], *p = params;

     if ((ret = dh_paramgen_check(ctx)) <= 0)
         return ret;

 #if !defined(FIPS_MODE)
     /* TODO(3.0): Remove this eventually when no more legacy */
     if (ctx->op.keymgmt.genctx == NULL)
         return EVP_PKEY_CTX_ctrl(ctx, EVP_PKEY_DH, EVP_PKEY_OP_PARAMGEN,
                                  EVP_PKEY_CTRL_DH_PARAMGEN_GENERATOR, gen, NULL);
 #endif

     *p++ = OSSL_PARAM_construct_int(OSSL_PKEY_PARAM_FFC_GENERATOR, &gen);
     *p++ = OSSL_PARAM_construct_end();

     return EVP_PKEY_CTX_set_params(ctx, params);
 }

 int EVP_PKEY_CTX_set_dh_rfc5114(EVP_PKEY_CTX *ctx, int gen)
 {
     int ret;
     OSSL_PARAM params[2], *p = params;
     const char *name;

     if ((ret = dh_paramgen_check(ctx)) <= 0)
         return ret;

 #if !defined(FIPS_MODE)
     /* TODO(3.0): Remove this eventually when no more legacy */
     if (ctx->op.keymgmt.genctx == NULL)
         return EVP_PKEY_CTX_ctrl(ctx, EVP_PKEY_DHX, EVP_PKEY_OP_PARAMGEN,
                                  EVP_PKEY_CTRL_DH_RFC5114, gen, NULL);
 #endif
     name = ffc_named_group_from_uid(gen);
     if (name == NULL)
         return 0;

     *p++ = OSSL_PARAM_construct_utf8_string(OSSL_PKEY_PARAM_FFC_GROUP,
                                             (void *)name, 0);
     *p++ = OSSL_PARAM_construct_end();
     return EVP_PKEY_CTX_set_params(ctx, params);
 }

 int EVP_PKEY_CTX_set_dhx_rfc5114(EVP_PKEY_CTX *ctx, int gen)
 {
     return EVP_PKEY_CTX_set_dh_rfc5114(ctx, gen);
 }

 int EVP_PKEY_CTX_set_dh_nid(EVP_PKEY_CTX *ctx, int nid)
 {
     int ret;
     OSSL_PARAM params[2], *p = params;
     const char *name;

     if ((ret = dh_paramgen_check(ctx)) <= 0)
         return ret;

 #if !defined(FIPS_MODE)
     /* TODO(3.0): Remove this eventually when no more legacy */
     if (ctx->op.keymgmt.genctx == NULL)
         return EVP_PKEY_CTX_ctrl(ctx, EVP_PKEY_DH,
                                  EVP_PKEY_OP_PARAMGEN | EVP_PKEY_OP_KEYGEN,
                                  EVP_PKEY_CTRL_DH_NID, nid, NULL);
 #endif
     name = ffc_named_group_from_uid(nid);
     if (name == NULL)
         return 0;

     *p++ = OSSL_PARAM_construct_utf8_string(OSSL_PKEY_PARAM_FFC_GROUP,
                                             (void *)name, 0);
     *p++ = OSSL_PARAM_construct_end();
     return EVP_PKEY_CTX_set_params(ctx, params);
 }
	/*
	* Copyright 1995-2020 The OpenSSL Project Authors. All Rights Reserved.
	*
	* Licensed under the Apache License 2.0 (the "License"). You may not use
	* this file except in compliance with the License. You can obtain a copy
	* in the file LICENSE in the source distribution or at
	* https://www.openssl.org/source/license.html
	*/

	/*
	* DH low level APIs are deprecated for public use, but still ok for
	* internal use.
	*/
	#include "internal/deprecated.h"

	#include <stdio.h>
	#include <openssl/bn.h>
	#include <openssl/engine.h>
	#include <openssl/obj_mac.h>
	#include <openssl/core_names.h>
	#include "internal/cryptlib.h"
	#include "internal/refcount.h"
	#include "crypto/evp.h"
	#include "crypto/dh.h"
	#include "dh_local.h"

	static DH dh_new_intern(ENGINE engine, OPENSSL_CTX *libctx);

	#ifndef FIPS_MODE
	int DH_set_method(DH dh, const DH_METHOD meth)
	{
	/*
	* NB: The caller is specifically setting a method, so it's not up to us
	* to deal with which ENGINE it comes from.
	*/
	const DH_METHOD *mtmp;
	mtmp = dh->meth;
	if (mtmp->finish)
	mtmp->finish(dh);
	#ifndef OPENSSL_NO_ENGINE
	ENGINE_finish(dh->engine);
	dh->engine = NULL;
	#endif
	dh->meth = meth;
	if (meth->init)
	meth->init(dh);
	return 1;
	}

	const DH_METHOD dh_get_method(const DH dh)
	{
	return dh->meth;
	}

	DH *DH_new(void)
	{
	return dh_new_intern(NULL, NULL);
	}

	DH DH_new_method(ENGINE engine)
	{
	return dh_new_intern(engine, NULL);
	}
	#endif /* !FIPS_MODE */

	DH dh_new_with_libctx(OPENSSL_CTX libctx)
	{
	return dh_new_intern(NULL, libctx);
	}

	static DH dh_new_intern(ENGINE engine, OPENSSL_CTX *libctx)
	{
	DH ret = OPENSSL_zalloc(sizeof(ret));

	if (ret == NULL) {
	DHerr(0, ERR_R_MALLOC_FAILURE);
	return NULL;
	}

	ret->references = 1;
	ret->lock = CRYPTO_THREAD_lock_new();
	if (ret->lock == NULL) {
	DHerr(0, ERR_R_MALLOC_FAILURE);
	OPENSSL_free(ret);
	return NULL;
	}

	ret->libctx = libctx;
	ret->meth = DH_get_default_method();
	#if !defined(FIPS_MODE) && !defined(OPENSSL_NO_ENGINE)
	ret->flags = ret->meth->flags; /* early default init */
	if (engine) {
	if (!ENGINE_init(engine)) {
	DHerr(0, ERR_R_ENGINE_LIB);
	goto err;
	}
	ret->engine = engine;
	} else
	ret->engine = ENGINE_get_default_DH();
	if (ret->engine) {
	ret->meth = ENGINE_get_DH(ret->engine);
	if (ret->meth == NULL) {
	DHerr(0, ERR_R_ENGINE_LIB);
	goto err;
	}
	}
	#endif

	ret->flags = ret->meth->flags;

	#ifndef FIPS_MODE
	if (!CRYPTO_new_ex_data(CRYPTO_EX_INDEX_DH, ret, &ret->ex_data))
	goto err;
	#endif /* FIPS_MODE */

	if ((ret->meth->init != NULL) && !ret->meth->init(ret)) {
	DHerr(0, ERR_R_INIT_FAIL);
	goto err;
	}

	return ret;

	err:
	DH_free(ret);
	return NULL;
	}

	void DH_free(DH *r)
	{
	int i;

	if (r == NULL)
	return;

	CRYPTO_DOWN_REF(&r->references, &i, r->lock);
	REF_PRINT_COUNT("DH", r);
	if (i > 0)
	return;
	REF_ASSERT_ISNT(i < 0);

	if (r->meth != NULL && r->meth->finish != NULL)
	r->meth->finish(r);
	#if !defined(FIPS_MODE)
	# if !defined(OPENSSL_NO_ENGINE)
	ENGINE_finish(r->engine);
	# endif
	CRYPTO_free_ex_data(CRYPTO_EX_INDEX_DH, r, &r->ex_data);
	#endif

	CRYPTO_THREAD_lock_free(r->lock);

	ffc_params_cleanup(&r->params);
	BN_clear_free(r->pub_key);
	BN_clear_free(r->priv_key);
	OPENSSL_free(r);
	}

	int DH_up_ref(DH *r)
	{
	int i;

	if (CRYPTO_UP_REF(&r->references, &i, r->lock) <= 0)
	return 0;

	REF_PRINT_COUNT("DH", r);
	REF_ASSERT_ISNT(i < 2);
	return ((i > 1) ? 1 : 0);
	}

	#ifndef FIPS_MODE
	int DH_set_ex_data(DH d, int idx, void arg)
	{
	return CRYPTO_set_ex_data(&d->ex_data, idx, arg);
	}

	void DH_get_ex_data(const DH d, int idx)
	{
	return CRYPTO_get_ex_data(&d->ex_data, idx);
	}
	#endif

	int DH_bits(const DH *dh)
	{
	return BN_num_bits(dh->params.p);
	}

	int DH_size(const DH *dh)
	{
	return BN_num_bytes(dh->params.p);
	}

	int DH_security_bits(const DH *dh)
	{
	int N;
	if (dh->params.q != NULL)
	N = BN_num_bits(dh->params.q);
	else if (dh->length)
	N = dh->length;
	else
	N = -1;
	return BN_security_bits(BN_num_bits(dh->params.p), N);
	}

	void DH_get0_pqg(const DH *dh,
	const BIGNUM p, const BIGNUM q, const BIGNUM **g)
	{
	ffc_params_get0_pqg(&dh->params, p, q, g);
	}

	int DH_set0_pqg(DH dh, BIGNUM p, BIGNUM q, BIGNUM g)
	{
	/*
	* If the fields p and g in d are NULL, the corresponding input
	* parameters MUST be non-NULL. q may remain NULL.
	*/
	if ((dh->params.p == NULL && p == NULL)
	\|\| (dh->params.g == NULL && g == NULL))
	return 0;

	ffc_params_set0_pqg(&dh->params, p, q, g);
	dh_cache_named_group(dh);
	if (q != NULL)
	dh->length = BN_num_bits(q);
	/*
	* Check if this is a named group. If it finds a named group then the
	* 'q' and 'length' value are either already set or are set by the
	* call.
	*/
	if (DH_get_nid(dh) == NID_undef) {
	/* If its not a named group then set the 'length' if q is not NULL */
	if (q != NULL)
	dh->length = BN_num_bits(q);
	}
	dh->dirty_cnt++;
	return 1;
	}

	long DH_get_length(const DH *dh)
	{
	return dh->length;
	}

	int DH_set_length(DH *dh, long length)
	{
	dh->length = length;
	return 1;
	}

	void DH_get0_key(const DH dh, const BIGNUM pub_key, const BIGNUM *priv_key)
	{
	if (pub_key != NULL)
	*pub_key = dh->pub_key;
	if (priv_key != NULL)
	*priv_key = dh->priv_key;
	}

	int DH_set0_key(DH dh, BIGNUM pub_key, BIGNUM *priv_key)
	{
	if (pub_key != NULL) {
	BN_clear_free(dh->pub_key);
	dh->pub_key = pub_key;
	}
	if (priv_key != NULL) {
	BN_clear_free(dh->priv_key);
	dh->priv_key = priv_key;
	dh->length = BN_num_bits(priv_key);
	}

	dh->dirty_cnt++;
	return 1;
	}

	const BIGNUM DH_get0_p(const DH dh)
	{
	return dh->params.p;
	}

	const BIGNUM DH_get0_q(const DH dh)
	{
	return dh->params.q;
	}

	const BIGNUM DH_get0_g(const DH dh)
	{
	return dh->params.g;
	}

	const BIGNUM DH_get0_priv_key(const DH dh)
	{
	return dh->priv_key;
	}

	const BIGNUM DH_get0_pub_key(const DH dh)
	{
	return dh->pub_key;
	}

	void DH_clear_flags(DH *dh, int flags)
	{
	dh->flags &= ~flags;
	}

	int DH_test_flags(const DH *dh, int flags)
	{
	return dh->flags & flags;
	}

	void DH_set_flags(DH *dh, int flags)
	{
	dh->flags \|= flags;
	}

	#ifndef FIPS_MODE
	ENGINE DH_get0_engine(DH dh)
	{
	return dh->engine;
	}
	#endif /FIPS_MODE /

	FFC_PARAMS dh_get0_params(DH dh)
	{
	return &dh->params;
	}
	int dh_get0_nid(const DH *dh)
	{
	return dh->params.nid;
	}

	int dh_ffc_params_fromdata(DH *dh, const OSSL_PARAM params[])
	{
	int ret;
	FFC_PARAMS *ffc;

	if (dh == NULL)
	return 0;
	ffc = dh_get0_params(dh);
	if (ffc == NULL)
	return 0;

	ret = ffc_params_fromdata(ffc, params);
	if (ret) {
	dh_cache_named_group(dh);
	dh->dirty_cnt++;
	}
	return ret;
	}

	static int dh_paramgen_check(EVP_PKEY_CTX *ctx)
	{
	if (ctx == NULL \|\| !EVP_PKEY_CTX_IS_GEN_OP(ctx)) {
	ERR_raise(ERR_LIB_EVP, EVP_R_COMMAND_NOT_SUPPORTED);
	/* Uses the same return values as EVP_PKEY_CTX_ctrl */
	return -2;
	}
	/* If key type not DH return error */
	if (ctx->pmeth != NULL
	&& ctx->pmeth->pkey_id != EVP_PKEY_DH
	&& ctx->pmeth->pkey_id != EVP_PKEY_DHX)
	return -1;
	return 1;
	}

	int EVP_PKEY_CTX_set_dh_paramgen_gindex(EVP_PKEY_CTX *ctx, int gindex)
	{
	int ret;
	OSSL_PARAM params[2], *p = params;

	if ((ret = dh_paramgen_check(ctx)) <= 0)
	return ret;

	*p++ = OSSL_PARAM_construct_int(OSSL_PKEY_PARAM_FFC_GINDEX, &gindex);
	*p++ = OSSL_PARAM_construct_end();

	return EVP_PKEY_CTX_set_params(ctx, params);
	}

	int EVP_PKEY_CTX_set_dh_paramgen_seed(EVP_PKEY_CTX *ctx,
	const unsigned char *seed,
	size_t seedlen)
	{
	int ret;
	OSSL_PARAM params[2], *p = params;

	if ((ret = dh_paramgen_check(ctx)) <= 0)
	return ret;

	*p++ = OSSL_PARAM_construct_octet_string(OSSL_PKEY_PARAM_FFC_SEED,
	(void *)seed, seedlen);
	*p++ = OSSL_PARAM_construct_end();

	return EVP_PKEY_CTX_set_params(ctx, params);
	}

	int EVP_PKEY_CTX_set_dh_paramgen_type(EVP_PKEY_CTX *ctx, int typ)
	{
	int ret;
	OSSL_PARAM params[2], *p = params;
	const char *name;

	if ((ret = dh_paramgen_check(ctx)) <= 0)
	return ret;

	#if !defined(FIPS_MODE)
	/* TODO(3.0): Remove this eventually when no more legacy */
	if (ctx->op.keymgmt.genctx == NULL)
	return EVP_PKEY_CTX_ctrl(ctx, EVP_PKEY_DH, EVP_PKEY_OP_PARAMGEN,
	EVP_PKEY_CTRL_DH_PARAMGEN_TYPE, typ, NULL);
	#endif

	name = dh_gen_type_id2name(typ);
	if (name == NULL)
	return 0;
	*p++ = OSSL_PARAM_construct_utf8_string(OSSL_PKEY_PARAM_FFC_TYPE,
	(char *) name, 0);
	*p++ = OSSL_PARAM_construct_end();

	return EVP_PKEY_CTX_set_params(ctx, params);
	}

	int EVP_PKEY_CTX_set_dh_paramgen_prime_len(EVP_PKEY_CTX *ctx, int pbits)
	{
	int ret;
	OSSL_PARAM params[2], *p = params;
	size_t bits = pbits;

	if ((ret = dh_paramgen_check(ctx)) <= 0)
	return ret;

	#if !defined(FIPS_MODE)
	/* TODO(3.0): Remove this eventually when no more legacy */
	if (ctx->op.keymgmt.genctx == NULL)
	return EVP_PKEY_CTX_ctrl(ctx, EVP_PKEY_DH, EVP_PKEY_OP_PARAMGEN,
	EVP_PKEY_CTRL_DH_PARAMGEN_PRIME_LEN, pbits,
	NULL);
	#endif
	*p++ = OSSL_PARAM_construct_size_t(OSSL_PKEY_PARAM_FFC_PBITS, &bits);
	*p++ = OSSL_PARAM_construct_end();
	return EVP_PKEY_CTX_set_params(ctx, params);
	}

	int EVP_PKEY_CTX_set_dh_paramgen_subprime_len(EVP_PKEY_CTX *ctx, int qbits)
	{
	int ret;
	OSSL_PARAM params[2], *p = params;
	size_t bits2 = qbits;

	if ((ret = dh_paramgen_check(ctx)) <= 0)
	return ret;

	#if !defined(FIPS_MODE)
	/* TODO(3.0): Remove this eventually when no more legacy */
	if (ctx->op.keymgmt.genctx == NULL)
	return EVP_PKEY_CTX_ctrl(ctx, EVP_PKEY_DH, EVP_PKEY_OP_PARAMGEN,
	EVP_PKEY_CTRL_DH_PARAMGEN_SUBPRIME_LEN, qbits,
	NULL);
	#endif
	*p++ = OSSL_PARAM_construct_size_t(OSSL_PKEY_PARAM_FFC_QBITS, &bits2);
	*p++ = OSSL_PARAM_construct_end();

	return EVP_PKEY_CTX_set_params(ctx, params);
	}

	int EVP_PKEY_CTX_set_dh_paramgen_generator(EVP_PKEY_CTX *ctx, int gen)
	{
	int ret;
	OSSL_PARAM params[2], *p = params;

	if ((ret = dh_paramgen_check(ctx)) <= 0)
	return ret;

	#if !defined(FIPS_MODE)
	/* TODO(3.0): Remove this eventually when no more legacy */
	if (ctx->op.keymgmt.genctx == NULL)
	return EVP_PKEY_CTX_ctrl(ctx, EVP_PKEY_DH, EVP_PKEY_OP_PARAMGEN,
	EVP_PKEY_CTRL_DH_PARAMGEN_GENERATOR, gen, NULL);
	#endif

	*p++ = OSSL_PARAM_construct_int(OSSL_PKEY_PARAM_FFC_GENERATOR, &gen);
	*p++ = OSSL_PARAM_construct_end();

	return EVP_PKEY_CTX_set_params(ctx, params);
	}

	int EVP_PKEY_CTX_set_dh_rfc5114(EVP_PKEY_CTX *ctx, int gen)
	{
	int ret;
	OSSL_PARAM params[2], *p = params;
	const char *name;

	if ((ret = dh_paramgen_check(ctx)) <= 0)
	return ret;

	#if !defined(FIPS_MODE)
	/* TODO(3.0): Remove this eventually when no more legacy */
	if (ctx->op.keymgmt.genctx == NULL)
	return EVP_PKEY_CTX_ctrl(ctx, EVP_PKEY_DHX, EVP_PKEY_OP_PARAMGEN,
	EVP_PKEY_CTRL_DH_RFC5114, gen, NULL);
	#endif
	name = ffc_named_group_from_uid(gen);
	if (name == NULL)
	return 0;

	*p++ = OSSL_PARAM_construct_utf8_string(OSSL_PKEY_PARAM_FFC_GROUP,
	(void *)name, 0);
	*p++ = OSSL_PARAM_construct_end();
	return EVP_PKEY_CTX_set_params(ctx, params);
	}

	int EVP_PKEY_CTX_set_dhx_rfc5114(EVP_PKEY_CTX *ctx, int gen)
	{
	return EVP_PKEY_CTX_set_dh_rfc5114(ctx, gen);
	}

	int EVP_PKEY_CTX_set_dh_nid(EVP_PKEY_CTX *ctx, int nid)
	{
	int ret;
	OSSL_PARAM params[2], *p = params;
	const char *name;

	if ((ret = dh_paramgen_check(ctx)) <= 0)
	return ret;

	#if !defined(FIPS_MODE)
	/* TODO(3.0): Remove this eventually when no more legacy */
	if (ctx->op.keymgmt.genctx == NULL)
	return EVP_PKEY_CTX_ctrl(ctx, EVP_PKEY_DH,
	EVP_PKEY_OP_PARAMGEN \| EVP_PKEY_OP_KEYGEN,
	EVP_PKEY_CTRL_DH_NID, nid, NULL);
	#endif
	name = ffc_named_group_from_uid(nid);
	if (name == NULL)
	return 0;

	*p++ = OSSL_PARAM_construct_utf8_string(OSSL_PKEY_PARAM_FFC_GROUP,
	(void *)name, 0);
	*p++ = OSSL_PARAM_construct_end();
	return EVP_PKEY_CTX_set_params(ctx, params);
	}