providers/implementations/keymgmt/dh_kmgmt.c

   1 /*
   2  * Copyright 2019-2020 The OpenSSL Project Authors. All Rights Reserved.
   3  *
   4  * Licensed under the Apache License 2.0 (the "License").  You may not use
   5  * this file except in compliance with the License.  You can obtain a copy
   6  * in the file LICENSE in the source distribution or at
   7  * https://www.openssl.org/source/license.html
   8  */
   9
  10 /*
  11  * DH low level APIs are deprecated for public use, but still ok for
  12  * internal use.
  13  */
  14 #include "internal/deprecated.h"
  15
  16 #include <string.h> /* strcmp */
  17 #include <openssl/core_numbers.h>
  18 #include <openssl/core_names.h>
  19 #include <openssl/bn.h>
  20 #include <openssl/err.h>
  21 #include "prov/implementations.h"
  22 #include "prov/providercommon.h"
  23 #include "prov/provider_ctx.h"
  24 #include "crypto/dh.h"
  25 #include "internal/sizes.h"
  26 #include "internal/nelem.h"
  27 #include "internal/param_build_set.h"
  28
  29 static OSSL_OP_keymgmt_new_fn dh_newdata;
  30 static OSSL_OP_keymgmt_free_fn dh_freedata;
  31 static OSSL_OP_keymgmt_gen_init_fn dh_gen_init;
  32 static OSSL_OP_keymgmt_gen_set_template_fn dh_gen_set_template;
  33 static OSSL_OP_keymgmt_gen_set_params_fn dh_gen_set_params;
  34 static OSSL_OP_keymgmt_gen_settable_params_fn dh_gen_settable_params;
  35 static OSSL_OP_keymgmt_gen_fn dh_gen;
  36 static OSSL_OP_keymgmt_gen_cleanup_fn dh_gen_cleanup;
  37 static OSSL_OP_keymgmt_get_params_fn dh_get_params;
  38 static OSSL_OP_keymgmt_gettable_params_fn dh_gettable_params;
  39 static OSSL_OP_keymgmt_has_fn dh_has;
  40 static OSSL_OP_keymgmt_match_fn dh_match;
  41 static OSSL_OP_keymgmt_validate_fn dh_validate;
  42 static OSSL_OP_keymgmt_import_fn dh_import;
  43 static OSSL_OP_keymgmt_import_types_fn dh_import_types;
  44 static OSSL_OP_keymgmt_export_fn dh_export;
  45 static OSSL_OP_keymgmt_export_types_fn dh_export_types;
  46
  47 #define DH_POSSIBLE_SELECTIONS                                                 \
  48     (OSSL_KEYMGMT_SELECT_KEYPAIR | OSSL_KEYMGMT_SELECT_DOMAIN_PARAMETERS)
  49
  50 struct dh_gen_ctx {
  51     OPENSSL_CTX *libctx;
  52
  53     FFC_PARAMS *ffc_params;
  54     int selection;
  55     /* All these parameters are used for parameter generation only */
  56     /* If there is a group name then the remaining parameters are not needed */
  57     int group_nid;
  58     size_t pbits;
  59     size_t qbits;
  60     EVP_MD *md;
  61     unsigned char *seed; /* optional FIPS186-4 param for testing */
  62     size_t seedlen;
  63     int gindex; /* optional  FIPS186-4 generator index (ignored if -1) */
  64     int gen_type; /* see dhtype2id */
  65     int generator; /* Used by DH_PARAMGEN_TYPE_GENERATOR in non fips mode only */
  66     int pcounter;
  67     int hindex;
  68
  69     OSSL_CALLBACK *cb;
  70     void *cbarg;
  71 };
  72
  73 typedef struct dh_name2id_st{
  74     const char *name;
  75     int id;
  76 } DH_GENTYPE_NAME2ID;
  77
  78 static const DH_GENTYPE_NAME2ID dhtype2id[]=
  79 {
  80     { "default", DH_PARAMGEN_TYPE_FIPS_186_4 },
  81     { "fips186_4", DH_PARAMGEN_TYPE_FIPS_186_4 },
  82     { "fips186_2", DH_PARAMGEN_TYPE_FIPS_186_2 },
  83     { "group", DH_PARAMGEN_TYPE_GROUP },
  84     { "generator", DH_PARAMGEN_TYPE_GENERATOR }
  85 };
  86
  87 const char *dh_gen_type_id2name(int id)
  88 {
  89     size_t i;
  90
  91     for (i = 0; i < OSSL_NELEM(dhtype2id); ++i) {
  92         if (dhtype2id[i].id == id)
  93             return dhtype2id[i].name;
  94     }
  95     return NULL;
  96 }
  97
  98 static int dh_gen_type_name2id(const char *name)
  99 {
 100     size_t i;
 101
 102     for (i = 0; i < OSSL_NELEM(dhtype2id); ++i) {
 103         if (strcmp(dhtype2id[i].name, name) == 0)
 104             return dhtype2id[i].id;
 105     }
 106     return -1;
 107 }
 108
 109 static int dh_key_todata(DH *dh, OSSL_PARAM_BLD *bld, OSSL_PARAM params[])
 110 {
 111     const BIGNUM *priv = NULL, *pub = NULL;
 112
 113     if (dh == NULL)
 114         return 0;
 115
 116     DH_get0_key(dh, &pub, &priv);
 117     if (priv != NULL
 118         && !ossl_param_build_set_bn(bld, params, OSSL_PKEY_PARAM_PRIV_KEY, priv))
 119         return 0;
 120     if (pub != NULL
 121         && !ossl_param_build_set_bn(bld, params, OSSL_PKEY_PARAM_PUB_KEY, pub))
 122         return 0;
 123
 124     return 1;
 125 }
 126
 127 static void *dh_newdata(void *provctx)
 128 {
 129     return dh_new_with_libctx(PROV_LIBRARY_CONTEXT_OF(provctx));
 130 }
 131
 132 static void dh_freedata(void *keydata)
 133 {
 134     DH_free(keydata);
 135 }
 136
 137 static int dh_has(void *keydata, int selection)
 138 {
 139     DH *dh = keydata;
 140     int ok = 0;
 141
 142     if (dh != NULL) {
 143         if ((selection & DH_POSSIBLE_SELECTIONS) != 0)
 144             ok = 1;
 145
 146         if ((selection & OSSL_KEYMGMT_SELECT_PUBLIC_KEY) != 0)
 147             ok = ok && (DH_get0_pub_key(dh) != NULL);
 148         if ((selection & OSSL_KEYMGMT_SELECT_PRIVATE_KEY) != 0)
 149             ok = ok && (DH_get0_priv_key(dh) != NULL);
 150         if ((selection & OSSL_KEYMGMT_SELECT_DOMAIN_PARAMETERS) != 0)
 151             ok = ok && (DH_get0_p(dh) != NULL && DH_get0_g(dh) != NULL);
 152     }
 153     return ok;
 154 }
 155
 156 static int dh_match(const void *keydata1, const void *keydata2, int selection)
 157 {
 158     const DH *dh1 = keydata1;
 159     const DH *dh2 = keydata2;
 160     int ok = 1;
 161
 162     if ((selection & OSSL_KEYMGMT_SELECT_PUBLIC_KEY) != 0)
 163         ok = ok && BN_cmp(DH_get0_pub_key(dh1), DH_get0_pub_key(dh2)) == 0;
 164     if ((selection & OSSL_KEYMGMT_SELECT_PRIVATE_KEY) != 0)
 165         ok = ok && BN_cmp(DH_get0_priv_key(dh1), DH_get0_priv_key(dh2)) == 0;
 166     if ((selection & OSSL_KEYMGMT_SELECT_DOMAIN_PARAMETERS) != 0) {
 167         FFC_PARAMS *dhparams1 = dh_get0_params((DH *)dh1);
 168         FFC_PARAMS *dhparams2 = dh_get0_params((DH *)dh2);
 169
 170         ok = ok && ffc_params_cmp(dhparams1, dhparams2, 1);
 171     }
 172     return ok;
 173 }
 174
 175 static int dh_import(void *keydata, int selection, const OSSL_PARAM params[])
 176 {
 177     DH *dh = keydata;
 178     int ok = 1;
 179
 180     if (dh == NULL)
 181         return 0;
 182
 183     if ((selection & DH_POSSIBLE_SELECTIONS) == 0)
 184         return 0;
 185
 186     if ((selection & OSSL_KEYMGMT_SELECT_ALL_PARAMETERS) != 0)
 187         ok = ok && dh_ffc_params_fromdata(dh, params);
 188
 189     if ((selection & OSSL_KEYMGMT_SELECT_KEYPAIR) != 0)
 190         ok = ok && dh_key_fromdata(dh, params);
 191
 192     return ok;
 193 }
 194
 195 static int dh_export(void *keydata, int selection, OSSL_CALLBACK *param_cb,
 196                      void *cbarg)
 197 {
 198     DH *dh = keydata;
 199     OSSL_PARAM_BLD *tmpl = NULL;
 200     OSSL_PARAM *params = NULL;
 201     int ok = 1;
 202
 203     if (dh == NULL)
 204         return 0;
 205
 206     tmpl = OSSL_PARAM_BLD_new();
 207     if (tmpl == NULL)
 208         return 0;
 209
 210     if ((selection & OSSL_KEYMGMT_SELECT_ALL_PARAMETERS) != 0)
 211         ok = ok && ffc_params_todata(dh_get0_params(dh), tmpl, NULL);
 212     if ((selection & OSSL_KEYMGMT_SELECT_KEYPAIR) != 0)
 213         ok = ok && dh_key_todata(dh, tmpl, NULL);
 214
 215     if (!ok
 216         || (params = OSSL_PARAM_BLD_to_param(tmpl)) == NULL) {
 217         ok = 0;
 218         goto err;
 219     }
 220     ok = param_cb(params, cbarg);
 221     OSSL_PARAM_BLD_free_params(params);
 222 err:
 223     OSSL_PARAM_BLD_free(tmpl);
 224     return ok;
 225 }
 226
 227 /* IMEXPORT = IMPORT + EXPORT */
 228
 229 # define DH_IMEXPORTABLE_PARAMETERS                                            \
 230     OSSL_PARAM_BN(OSSL_PKEY_PARAM_FFC_P, NULL, 0),                             \
 231     OSSL_PARAM_BN(OSSL_PKEY_PARAM_FFC_Q, NULL, 0),                             \
 232     OSSL_PARAM_BN(OSSL_PKEY_PARAM_FFC_G, NULL, 0),                             \
 233     OSSL_PARAM_BN(OSSL_PKEY_PARAM_FFC_COFACTOR, NULL, 0),                      \
 234     OSSL_PARAM_int(OSSL_PKEY_PARAM_FFC_GINDEX, NULL),                          \
 235     OSSL_PARAM_int(OSSL_PKEY_PARAM_FFC_PCOUNTER, NULL),                        \
 236     OSSL_PARAM_int(OSSL_PKEY_PARAM_FFC_H, NULL),                               \
 237     OSSL_PARAM_utf8_string(OSSL_PKEY_PARAM_FFC_GROUP, NULL, 0),                \
 238     OSSL_PARAM_octet_string(OSSL_PKEY_PARAM_FFC_SEED, NULL, 0)
 239 # define DH_IMEXPORTABLE_PUBLIC_KEY                                            \
 240     OSSL_PARAM_BN(OSSL_PKEY_PARAM_PUB_KEY, NULL, 0)
 241 # define DH_IMEXPORTABLE_PRIVATE_KEY                                           \
 242     OSSL_PARAM_BN(OSSL_PKEY_PARAM_PRIV_KEY, NULL, 0)
 243 static const OSSL_PARAM dh_all_types[] = {
 244     DH_IMEXPORTABLE_PARAMETERS,
 245     DH_IMEXPORTABLE_PUBLIC_KEY,
 246     DH_IMEXPORTABLE_PRIVATE_KEY,
 247     OSSL_PARAM_END
 248 };
 249 static const OSSL_PARAM dh_parameter_types[] = {
 250     DH_IMEXPORTABLE_PARAMETERS,
 251     OSSL_PARAM_END
 252 };
 253 static const OSSL_PARAM dh_key_types[] = {
 254     DH_IMEXPORTABLE_PUBLIC_KEY,
 255     DH_IMEXPORTABLE_PRIVATE_KEY,
 256     OSSL_PARAM_END
 257 };
 258 static const OSSL_PARAM *dh_types[] = {
 259     NULL,                        /* Index 0 = none of them */
 260     dh_parameter_types,          /* Index 1 = parameter types */
 261     dh_key_types,                /* Index 2 = key types */
 262     dh_all_types                 /* Index 3 = 1 + 2 */
 263 };
 264
 265 static const OSSL_PARAM *dh_imexport_types(int selection)
 266 {
 267     int type_select = 0;
 268
 269     if ((selection & OSSL_KEYMGMT_SELECT_ALL_PARAMETERS) != 0)
 270         type_select += 1;
 271     if ((selection & OSSL_KEYMGMT_SELECT_KEYPAIR) != 0)
 272         type_select += 2;
 273     return dh_types[type_select];
 274 }
 275
 276 static const OSSL_PARAM *dh_import_types(int selection)
 277 {
 278     return dh_imexport_types(selection);
 279 }
 280
 281 static const OSSL_PARAM *dh_export_types(int selection)
 282 {
 283     return dh_imexport_types(selection);
 284 }
 285
 286 static ossl_inline int dh_get_params(void *key, OSSL_PARAM params[])
 287 {
 288     DH *dh = key;
 289     OSSL_PARAM *p;
 290
 291     if ((p = OSSL_PARAM_locate(params, OSSL_PKEY_PARAM_BITS)) != NULL
 292         && !OSSL_PARAM_set_int(p, DH_bits(dh)))
 293         return 0;
 294     if ((p = OSSL_PARAM_locate(params, OSSL_PKEY_PARAM_SECURITY_BITS)) != NULL
 295         && !OSSL_PARAM_set_int(p, DH_security_bits(dh)))
 296         return 0;
 297     if ((p = OSSL_PARAM_locate(params, OSSL_PKEY_PARAM_MAX_SIZE)) != NULL
 298         && !OSSL_PARAM_set_int(p, DH_size(dh)))
 299         return 0;
 300     return ffc_params_todata(dh_get0_params(dh), NULL, params)
 301            && dh_key_todata(dh, NULL, params);
 302 }
 303
 304 static const OSSL_PARAM dh_params[] = {
 305     OSSL_PARAM_int(OSSL_PKEY_PARAM_BITS, NULL),
 306     OSSL_PARAM_int(OSSL_PKEY_PARAM_SECURITY_BITS, NULL),
 307     OSSL_PARAM_int(OSSL_PKEY_PARAM_MAX_SIZE, NULL),
 308     DH_IMEXPORTABLE_PARAMETERS,
 309     DH_IMEXPORTABLE_PUBLIC_KEY,
 310     DH_IMEXPORTABLE_PRIVATE_KEY,
 311     OSSL_PARAM_END
 312 };
 313
 314 static const OSSL_PARAM *dh_gettable_params(void)
 315 {
 316     return dh_params;
 317 }
 318
 319 static int dh_validate_public(DH *dh)
 320 {
 321     const BIGNUM *pub_key = NULL;
 322
 323     DH_get0_key(dh, &pub_key, NULL);
 324     return DH_check_pub_key_ex(dh, pub_key);
 325 }
 326
 327 static int dh_validate_private(DH *dh)
 328 {
 329     int status = 0;
 330     const BIGNUM *priv_key = NULL;
 331
 332     DH_get0_key(dh, NULL, &priv_key);
 333     return dh_check_priv_key(dh, priv_key, &status);;
 334 }
 335
 336 static int dh_validate(void *keydata, int selection)
 337 {
 338     DH *dh = keydata;
 339     int ok = 0;
 340
 341     if ((selection & DH_POSSIBLE_SELECTIONS) != 0)
 342         ok = 1;
 343
 344     if ((selection & OSSL_KEYMGMT_SELECT_DOMAIN_PARAMETERS) != 0)
 345         ok = ok && DH_check_params_ex(dh);
 346
 347     if ((selection & OSSL_KEYMGMT_SELECT_PUBLIC_KEY) != 0)
 348         ok = ok && dh_validate_public(dh);
 349
 350     if ((selection & OSSL_KEYMGMT_SELECT_PRIVATE_KEY) != 0)
 351         ok = ok && dh_validate_private(dh);
 352
 353     if ((selection & OSSL_KEYMGMT_SELECT_KEYPAIR)
 354             == OSSL_KEYMGMT_SELECT_KEYPAIR)
 355         ok = ok && dh_check_pairwise(dh);
 356     return ok;
 357 }
 358
 359 static void *dh_gen_init(void *provctx, int selection)
 360 {
 361     OPENSSL_CTX *libctx = PROV_LIBRARY_CONTEXT_OF(provctx);
 362     struct dh_gen_ctx *gctx = NULL;
 363
 364     if ((selection & (OSSL_KEYMGMT_SELECT_KEYPAIR
 365                       | OSSL_KEYMGMT_SELECT_DOMAIN_PARAMETERS)) == 0)
 366         return NULL;
 367
 368     if ((gctx = OPENSSL_zalloc(sizeof(*gctx))) != NULL) {
 369         gctx->selection = selection;
 370         gctx->libctx = libctx;
 371         gctx->pbits = 2048;
 372         gctx->qbits = 224;
 373         gctx->md = NULL;
 374         gctx->gen_type = DH_PARAMGEN_TYPE_FIPS_186_4;
 375         gctx->gindex = -1;
 376         gctx->hindex = 0;
 377         gctx->pcounter = -1;
 378         gctx->generator = DH_GENERATOR_2;
 379     }
 380     return gctx;
 381 }
 382
 383 static int dh_gen_set_template(void *genctx, void *templ)
 384 {
 385     struct dh_gen_ctx *gctx = genctx;
 386     DH *dh = templ;
 387
 388     if (gctx == NULL || dh == NULL)
 389         return 0;
 390     gctx->ffc_params = dh_get0_params(dh);
 391     return 1;
 392 }
 393
 394 static int dh_set_gen_seed(struct dh_gen_ctx *gctx, unsigned char *seed,
 395                            size_t seedlen)
 396 {
 397     OPENSSL_clear_free(gctx->seed, gctx->seedlen);
 398     gctx->seed = NULL;
 399     gctx->seedlen = 0;
 400     if (seed != NULL && seedlen > 0) {
 401         gctx->seed = OPENSSL_memdup(seed, seedlen);
 402         if (gctx->seed == NULL)
 403             return 0;
 404         gctx->seedlen = seedlen;
 405     }
 406     return 1;
 407 }
 408
 409 static int dh_gen_set_params(void *genctx, const OSSL_PARAM params[])
 410 {
 411     struct dh_gen_ctx *gctx = genctx;
 412     const OSSL_PARAM *p;
 413
 414     if (gctx == NULL)
 415         return 0;
 416
 417     p = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_FFC_TYPE);
 418     if (p != NULL) {
 419         if (p->data_type != OSSL_PARAM_UTF8_STRING
 420             || ((gctx->gen_type = dh_gen_type_name2id(p->data)) == -1)) {
 421             ERR_raise(ERR_LIB_PROV, ERR_R_PASSED_INVALID_ARGUMENT);
 422             return 0;
 423         }
 424     }
 425     p = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_FFC_GROUP);
 426     if (p != NULL) {
 427         if (p->data_type != OSSL_PARAM_UTF8_STRING
 428            || ((gctx->group_nid = ffc_named_group_to_uid(p->data)) == NID_undef)) {
 429             ERR_raise(ERR_LIB_PROV, ERR_R_PASSED_INVALID_ARGUMENT);
 430             return 0;
 431         }
 432         gctx->gen_type = DH_PARAMGEN_TYPE_GROUP;
 433     }
 434     p = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_FFC_GENERATOR);
 435     if (p != NULL
 436         && !OSSL_PARAM_get_int(p, &gctx->generator))
 437         return 0;
 438     p = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_FFC_GINDEX);
 439     if (p != NULL
 440         && !OSSL_PARAM_get_int(p, &gctx->gindex))
 441         return 0;
 442     p = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_FFC_PCOUNTER);
 443     if (p != NULL
 444         && !OSSL_PARAM_get_int(p, &gctx->pcounter))
 445         return 0;
 446     p = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_FFC_H);
 447     if (p != NULL
 448         && !OSSL_PARAM_get_int(p, &gctx->hindex))
 449         return 0;
 450     p = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_FFC_SEED);
 451     if (p != NULL
 452         && (p->data_type != OSSL_PARAM_OCTET_STRING
 453             || !dh_set_gen_seed(gctx, p->data, p->data_size)))
 454             return 0;
 455
 456     if ((p = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_FFC_PBITS)) != NULL
 457         && !OSSL_PARAM_get_size_t(p, &gctx->pbits))
 458         return 0;
 459     if ((p = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_FFC_QBITS)) != NULL
 460         && !OSSL_PARAM_get_size_t(p, &gctx->qbits))
 461         return 0;
 462     p = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_FFC_DIGEST);
 463     if (p != NULL) {
 464         const OSSL_PARAM *p1;
 465         char mdprops[OSSL_MAX_PROPQUERY_SIZE] = { '\0' };
 466         char *str = mdprops;
 467
 468         if (p->data_type != OSSL_PARAM_UTF8_STRING)
 469             return 0;
 470         p1 = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_FFC_DIGEST_PROPS);
 471         if (p1 != NULL
 472             && !OSSL_PARAM_get_utf8_string(p1, &str, sizeof(mdprops)))
 473             return 0;
 474         EVP_MD_free(gctx->md);
 475         gctx->md = EVP_MD_fetch(gctx->libctx, p->data, mdprops);
 476         if (gctx->md == NULL)
 477             return 0;
 478     }
 479     return 1;
 480 }
 481
 482 static const OSSL_PARAM *dh_gen_settable_params(void *provctx)
 483 {
 484     static OSSL_PARAM settable[] = {
 485         OSSL_PARAM_utf8_string(OSSL_PKEY_PARAM_FFC_GROUP, NULL, 0),
 486         OSSL_PARAM_utf8_string(OSSL_PKEY_PARAM_FFC_TYPE, NULL, 0),
 487         OSSL_PARAM_size_t(OSSL_PKEY_PARAM_FFC_PBITS, NULL),
 488         OSSL_PARAM_size_t(OSSL_PKEY_PARAM_FFC_QBITS, NULL),
 489         OSSL_PARAM_utf8_string(OSSL_PKEY_PARAM_FFC_DIGEST, NULL, 0),
 490         OSSL_PARAM_utf8_string(OSSL_PKEY_PARAM_FFC_DIGEST_PROPS, NULL, 0),
 491         OSSL_PARAM_int(OSSL_PKEY_PARAM_FFC_GINDEX, NULL),
 492         OSSL_PARAM_octet_string(OSSL_PKEY_PARAM_FFC_SEED, NULL, 0),
 493         OSSL_PARAM_int(OSSL_PKEY_PARAM_FFC_GENERATOR, NULL),
 494         OSSL_PARAM_int(OSSL_PKEY_PARAM_FFC_PCOUNTER, NULL),
 495         OSSL_PARAM_int(OSSL_PKEY_PARAM_FFC_H, NULL),
 496         OSSL_PARAM_END
 497     };
 498     return settable;
 499 }
 500
 501 static int dh_gencb(int p, int n, BN_GENCB *cb)
 502 {
 503     struct dh_gen_ctx *gctx = BN_GENCB_get_arg(cb);
 504     OSSL_PARAM params[] = { OSSL_PARAM_END, OSSL_PARAM_END, OSSL_PARAM_END };
 505
 506     params[0] = OSSL_PARAM_construct_int(OSSL_GEN_PARAM_POTENTIAL, &p);
 507     params[1] = OSSL_PARAM_construct_int(OSSL_GEN_PARAM_ITERATION, &n);
 508
 509     return gctx->cb(params, gctx->cbarg);
 510 }
 511
 512 static void *dh_gen(void *genctx, OSSL_CALLBACK *osslcb, void *cbarg)
 513 {
 514     int ret = 0;
 515     struct dh_gen_ctx *gctx = genctx;
 516     DH *dh = NULL;
 517     BN_GENCB *gencb = NULL;
 518     FFC_PARAMS *ffc;
 519
 520     if (gctx == NULL)
 521         return NULL;
 522
 523     /* For parameter generation - If there is a group name just create it */
 524     if (gctx->gen_type == DH_PARAMGEN_TYPE_GROUP) {
 525         /* Select a named group if there is not one already */
 526         if (gctx->group_nid == NID_undef)
 527             gctx->group_nid = dh_get_named_group_uid_from_size(gctx->pbits);
 528         if (gctx->group_nid == NID_undef)
 529             return NULL;
 530         dh = dh_new_by_nid_with_libctx(gctx->libctx, gctx->group_nid);
 531         if (dh == NULL)
 532             return NULL;
 533         ffc = dh_get0_params(dh);
 534     } else {
 535         dh = dh_new_with_libctx(gctx->libctx);
 536         if (dh == NULL)
 537             return NULL;
 538         ffc = dh_get0_params(dh);
 539
 540         /* Copy the template value if one was passed */
 541         if (gctx->ffc_params != NULL
 542             && !ffc_params_copy(ffc, gctx->ffc_params))
 543             goto end;
 544
 545         if (!ffc_params_set_seed(ffc, gctx->seed, gctx->seedlen))
 546             goto end;
 547         if (gctx->gindex != -1) {
 548             ffc_params_set_gindex(ffc, gctx->gindex);
 549             if (gctx->pcounter != -1)
 550                 ffc_params_set_pcounter(ffc, gctx->pcounter);
 551         } else if (gctx->hindex != 0) {
 552             ffc_params_set_h(ffc, gctx->hindex);
 553         }
 554         gctx->cb = osslcb;
 555         gctx->cbarg = cbarg;
 556         gencb = BN_GENCB_new();
 557         if (gencb != NULL)
 558             BN_GENCB_set(gencb, dh_gencb, genctx);
 559
 560         if ((gctx->selection & OSSL_KEYMGMT_SELECT_DOMAIN_PARAMETERS) != 0) {
 561             /*
 562              * NOTE: The old safe prime generator code is not used in fips mode,
 563              * (i.e internally it ignores the generator and chooses a named
 564              * group based on pbits.
 565              */
 566             if (gctx->gen_type == DH_PARAMGEN_TYPE_GENERATOR)
 567                 ret = DH_generate_parameters_ex(dh, gctx->pbits,
 568                                                 gctx->generator, gencb);
 569             else
 570                 ret = dh_generate_ffc_parameters(dh, gctx->gen_type, gctx->pbits,
 571                                                  gctx->qbits, gctx->md, gencb);
 572             if (ret <= 0)
 573                 goto end;
 574         }
 575     }
 576
 577     if ((gctx->selection & OSSL_KEYMGMT_SELECT_KEYPAIR) != 0) {
 578         if (ffc->p == NULL || ffc->g == NULL)
 579             goto end;
 580         if (DH_generate_key(dh) <= 0)
 581             goto end;
 582     }
 583     ret = 1;
 584 end:
 585     if (ret <= 0) {
 586         DH_free(dh);
 587         dh = NULL;
 588     }
 589     BN_GENCB_free(gencb);
 590     return dh;
 591 }
 592
 593 static void dh_gen_cleanup(void *genctx)
 594 {
 595     struct dh_gen_ctx *gctx = genctx;
 596
 597     if (gctx == NULL)
 598         return;
 599
 600     OPENSSL_clear_free(gctx->seed, gctx->seedlen);
 601     EVP_MD_free(gctx->md);
 602     OPENSSL_free(gctx);
 603 }
 604
 605 const OSSL_DISPATCH dh_keymgmt_functions[] = {
 606     { OSSL_FUNC_KEYMGMT_NEW, (void (*)(void))dh_newdata },
 607     { OSSL_FUNC_KEYMGMT_GEN_INIT, (void (*)(void))dh_gen_init },
 608     { OSSL_FUNC_KEYMGMT_GEN_SET_TEMPLATE, (void (*)(void))dh_gen_set_template },
 609     { OSSL_FUNC_KEYMGMT_GEN_SET_PARAMS, (void (*)(void))dh_gen_set_params },
 610     { OSSL_FUNC_KEYMGMT_GEN_SETTABLE_PARAMS,
 611       (void (*)(void))dh_gen_settable_params },
 612     { OSSL_FUNC_KEYMGMT_GEN, (void (*)(void))dh_gen },
 613     { OSSL_FUNC_KEYMGMT_GEN_CLEANUP, (void (*)(void))dh_gen_cleanup },
 614     { OSSL_FUNC_KEYMGMT_FREE, (void (*)(void))dh_freedata },
 615     { OSSL_FUNC_KEYMGMT_GET_PARAMS, (void (*) (void))dh_get_params },
 616     { OSSL_FUNC_KEYMGMT_GETTABLE_PARAMS, (void (*) (void))dh_gettable_params },
 617     { OSSL_FUNC_KEYMGMT_HAS, (void (*)(void))dh_has },
 618     { OSSL_FUNC_KEYMGMT_MATCH, (void (*)(void))dh_match },
 619     { OSSL_FUNC_KEYMGMT_VALIDATE, (void (*)(void))dh_validate },
 620     { OSSL_FUNC_KEYMGMT_IMPORT, (void (*)(void))dh_import },
 621     { OSSL_FUNC_KEYMGMT_IMPORT_TYPES, (void (*)(void))dh_import_types },
 622     { OSSL_FUNC_KEYMGMT_EXPORT, (void (*)(void))dh_export },
 623     { OSSL_FUNC_KEYMGMT_EXPORT_TYPES, (void (*)(void))dh_export_types },
 624     { 0, NULL }
 625 };