xref: /freebsd/crypto/openssl/crypto/dh/dh_key.c (revision e0c4386e7e71d93b0edc0c8fa156263fc4a8b0b6)
1 /*
2  * Copyright 1995-2023 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 <stdio.h>
17 #include "internal/cryptlib.h"
18 #include "dh_local.h"
19 #include "crypto/bn.h"
20 #include "crypto/dh.h"
21 #include "crypto/security_bits.h"
22 
23 #ifdef FIPS_MODULE
24 # define MIN_STRENGTH 112
25 #else
26 # define MIN_STRENGTH 80
27 #endif
28 
29 static int generate_key(DH *dh);
30 static int dh_bn_mod_exp(const DH *dh, BIGNUM *r,
31                          const BIGNUM *a, const BIGNUM *p,
32                          const BIGNUM *m, BN_CTX *ctx, BN_MONT_CTX *m_ctx);
33 static int dh_init(DH *dh);
34 static int dh_finish(DH *dh);
35 
36 /*
37  * See SP800-56Ar3 Section 5.7.1.1
38  * Finite Field Cryptography Diffie-Hellman (FFC DH) Primitive
39  */
ossl_dh_compute_key(unsigned char * key,const BIGNUM * pub_key,DH * dh)40 int ossl_dh_compute_key(unsigned char *key, const BIGNUM *pub_key, DH *dh)
41 {
42     BN_CTX *ctx = NULL;
43     BN_MONT_CTX *mont = NULL;
44     BIGNUM *z = NULL, *pminus1;
45     int ret = -1;
46 
47     if (BN_num_bits(dh->params.p) > OPENSSL_DH_MAX_MODULUS_BITS) {
48         ERR_raise(ERR_LIB_DH, DH_R_MODULUS_TOO_LARGE);
49         goto err;
50     }
51 
52     if (dh->params.q != NULL
53         && BN_num_bits(dh->params.q) > OPENSSL_DH_MAX_MODULUS_BITS) {
54         ERR_raise(ERR_LIB_DH, DH_R_Q_TOO_LARGE);
55         goto err;
56     }
57 
58     if (BN_num_bits(dh->params.p) < DH_MIN_MODULUS_BITS) {
59         ERR_raise(ERR_LIB_DH, DH_R_MODULUS_TOO_SMALL);
60         return 0;
61     }
62 
63     ctx = BN_CTX_new_ex(dh->libctx);
64     if (ctx == NULL)
65         goto err;
66     BN_CTX_start(ctx);
67     pminus1 = BN_CTX_get(ctx);
68     z = BN_CTX_get(ctx);
69     if (z == NULL)
70         goto err;
71 
72     if (dh->priv_key == NULL) {
73         ERR_raise(ERR_LIB_DH, DH_R_NO_PRIVATE_VALUE);
74         goto err;
75     }
76 
77     if (dh->flags & DH_FLAG_CACHE_MONT_P) {
78         mont = BN_MONT_CTX_set_locked(&dh->method_mont_p,
79                                       dh->lock, dh->params.p, ctx);
80         BN_set_flags(dh->priv_key, BN_FLG_CONSTTIME);
81         if (!mont)
82             goto err;
83     }
84 
85     /* (Step 1) Z = pub_key^priv_key mod p */
86     if (!dh->meth->bn_mod_exp(dh, z, pub_key, dh->priv_key, dh->params.p, ctx,
87                               mont)) {
88         ERR_raise(ERR_LIB_DH, ERR_R_BN_LIB);
89         goto err;
90     }
91 
92     /* (Step 2) Error if z <= 1 or z = p - 1 */
93     if (BN_copy(pminus1, dh->params.p) == NULL
94         || !BN_sub_word(pminus1, 1)
95         || BN_cmp(z, BN_value_one()) <= 0
96         || BN_cmp(z, pminus1) == 0) {
97         ERR_raise(ERR_LIB_DH, DH_R_INVALID_SECRET);
98         goto err;
99     }
100 
101     /* return the padded key, i.e. same number of bytes as the modulus */
102     ret = BN_bn2binpad(z, key, BN_num_bytes(dh->params.p));
103  err:
104     BN_clear(z); /* (Step 2) destroy intermediate values */
105     BN_CTX_end(ctx);
106     BN_CTX_free(ctx);
107     return ret;
108 }
109 
110 /*-
111  * NB: This function is inherently not constant time due to the
112  * RFC 5246 (8.1.2) padding style that strips leading zero bytes.
113  */
DH_compute_key(unsigned char * key,const BIGNUM * pub_key,DH * dh)114 int DH_compute_key(unsigned char *key, const BIGNUM *pub_key, DH *dh)
115 {
116     int ret = 0, i;
117     volatile size_t npad = 0, mask = 1;
118 
119     /* compute the key; ret is constant unless compute_key is external */
120 #ifdef FIPS_MODULE
121     ret = ossl_dh_compute_key(key, pub_key, dh);
122 #else
123     ret = dh->meth->compute_key(key, pub_key, dh);
124 #endif
125     if (ret <= 0)
126         return ret;
127 
128     /* count leading zero bytes, yet still touch all bytes */
129     for (i = 0; i < ret; i++) {
130         mask &= !key[i];
131         npad += mask;
132     }
133 
134     /* unpad key */
135     ret -= npad;
136     /* key-dependent memory access, potentially leaking npad / ret */
137     memmove(key, key + npad, ret);
138     /* key-dependent memory access, potentially leaking npad / ret */
139     memset(key + ret, 0, npad);
140 
141     return ret;
142 }
143 
DH_compute_key_padded(unsigned char * key,const BIGNUM * pub_key,DH * dh)144 int DH_compute_key_padded(unsigned char *key, const BIGNUM *pub_key, DH *dh)
145 {
146     int rv, pad;
147 
148     /* rv is constant unless compute_key is external */
149 #ifdef FIPS_MODULE
150     rv = ossl_dh_compute_key(key, pub_key, dh);
151 #else
152     rv = dh->meth->compute_key(key, pub_key, dh);
153 #endif
154     if (rv <= 0)
155         return rv;
156     pad = BN_num_bytes(dh->params.p) - rv;
157     /* pad is constant (zero) unless compute_key is external */
158     if (pad > 0) {
159         memmove(key + pad, key, rv);
160         memset(key, 0, pad);
161     }
162     return rv + pad;
163 }
164 
165 static DH_METHOD dh_ossl = {
166     "OpenSSL DH Method",
167     generate_key,
168     ossl_dh_compute_key,
169     dh_bn_mod_exp,
170     dh_init,
171     dh_finish,
172     DH_FLAG_FIPS_METHOD,
173     NULL,
174     NULL
175 };
176 
177 static const DH_METHOD *default_DH_method = &dh_ossl;
178 
DH_OpenSSL(void)179 const DH_METHOD *DH_OpenSSL(void)
180 {
181     return &dh_ossl;
182 }
183 
DH_get_default_method(void)184 const DH_METHOD *DH_get_default_method(void)
185 {
186     return default_DH_method;
187 }
188 
dh_bn_mod_exp(const DH * dh,BIGNUM * r,const BIGNUM * a,const BIGNUM * p,const BIGNUM * m,BN_CTX * ctx,BN_MONT_CTX * m_ctx)189 static int dh_bn_mod_exp(const DH *dh, BIGNUM *r,
190                          const BIGNUM *a, const BIGNUM *p,
191                          const BIGNUM *m, BN_CTX *ctx, BN_MONT_CTX *m_ctx)
192 {
193     return BN_mod_exp_mont(r, a, p, m, ctx, m_ctx);
194 }
195 
dh_init(DH * dh)196 static int dh_init(DH *dh)
197 {
198     dh->flags |= DH_FLAG_CACHE_MONT_P;
199     dh->dirty_cnt++;
200     return 1;
201 }
202 
dh_finish(DH * dh)203 static int dh_finish(DH *dh)
204 {
205     BN_MONT_CTX_free(dh->method_mont_p);
206     return 1;
207 }
208 
209 #ifndef FIPS_MODULE
DH_set_default_method(const DH_METHOD * meth)210 void DH_set_default_method(const DH_METHOD *meth)
211 {
212     default_DH_method = meth;
213 }
214 #endif /* FIPS_MODULE */
215 
DH_generate_key(DH * dh)216 int DH_generate_key(DH *dh)
217 {
218 #ifdef FIPS_MODULE
219     return generate_key(dh);
220 #else
221     return dh->meth->generate_key(dh);
222 #endif
223 }
224 
ossl_dh_generate_public_key(BN_CTX * ctx,const DH * dh,const BIGNUM * priv_key,BIGNUM * pub_key)225 int ossl_dh_generate_public_key(BN_CTX *ctx, const DH *dh,
226                                 const BIGNUM *priv_key, BIGNUM *pub_key)
227 {
228     int ret = 0;
229     BIGNUM *prk = BN_new();
230     BN_MONT_CTX *mont = NULL;
231 
232     if (prk == NULL)
233         return 0;
234 
235     if (dh->flags & DH_FLAG_CACHE_MONT_P) {
236         /*
237          * We take the input DH as const, but we lie, because in some cases we
238          * want to get a hold of its Montgomery context.
239          *
240          * We cast to remove the const qualifier in this case, it should be
241          * fine...
242          */
243         BN_MONT_CTX **pmont = (BN_MONT_CTX **)&dh->method_mont_p;
244 
245         mont = BN_MONT_CTX_set_locked(pmont, dh->lock, dh->params.p, ctx);
246         if (mont == NULL)
247             goto err;
248     }
249     BN_with_flags(prk, priv_key, BN_FLG_CONSTTIME);
250 
251     /* pub_key = g^priv_key mod p */
252     if (!dh->meth->bn_mod_exp(dh, pub_key, dh->params.g, prk, dh->params.p,
253                               ctx, mont))
254         goto err;
255     ret = 1;
256 err:
257     BN_clear_free(prk);
258     return ret;
259 }
260 
generate_key(DH * dh)261 static int generate_key(DH *dh)
262 {
263     int ok = 0;
264     int generate_new_key = 0;
265 #ifndef FIPS_MODULE
266     unsigned l;
267 #endif
268     BN_CTX *ctx = NULL;
269     BIGNUM *pub_key = NULL, *priv_key = NULL;
270 
271     if (BN_num_bits(dh->params.p) > OPENSSL_DH_MAX_MODULUS_BITS) {
272         ERR_raise(ERR_LIB_DH, DH_R_MODULUS_TOO_LARGE);
273         return 0;
274     }
275 
276     if (dh->params.q != NULL
277         && BN_num_bits(dh->params.q) > OPENSSL_DH_MAX_MODULUS_BITS) {
278         ERR_raise(ERR_LIB_DH, DH_R_Q_TOO_LARGE);
279         return 0;
280     }
281 
282     if (BN_num_bits(dh->params.p) < DH_MIN_MODULUS_BITS) {
283         ERR_raise(ERR_LIB_DH, DH_R_MODULUS_TOO_SMALL);
284         return 0;
285     }
286 
287     ctx = BN_CTX_new_ex(dh->libctx);
288     if (ctx == NULL)
289         goto err;
290 
291     if (dh->priv_key == NULL) {
292         priv_key = BN_secure_new();
293         if (priv_key == NULL)
294             goto err;
295         generate_new_key = 1;
296     } else {
297         priv_key = dh->priv_key;
298     }
299 
300     if (dh->pub_key == NULL) {
301         pub_key = BN_new();
302         if (pub_key == NULL)
303             goto err;
304     } else {
305         pub_key = dh->pub_key;
306     }
307     if (generate_new_key) {
308         /* Is it an approved safe prime ?*/
309         if (DH_get_nid(dh) != NID_undef) {
310             int max_strength =
311                     ossl_ifc_ffc_compute_security_bits(BN_num_bits(dh->params.p));
312 
313             if (dh->params.q == NULL
314                 || dh->length > BN_num_bits(dh->params.q))
315                 goto err;
316             /* dh->length = maximum bit length of generated private key */
317             if (!ossl_ffc_generate_private_key(ctx, &dh->params, dh->length,
318                                                max_strength, priv_key))
319                 goto err;
320         } else {
321 #ifdef FIPS_MODULE
322             if (dh->params.q == NULL)
323                 goto err;
324 #else
325             if (dh->params.q == NULL) {
326                 /* secret exponent length, must satisfy 2^(l-1) <= p */
327                 if (dh->length != 0
328                     && dh->length >= BN_num_bits(dh->params.p))
329                     goto err;
330                 l = dh->length ? dh->length : BN_num_bits(dh->params.p) - 1;
331                 if (!BN_priv_rand_ex(priv_key, l, BN_RAND_TOP_ONE,
332                                      BN_RAND_BOTTOM_ANY, 0, ctx))
333                     goto err;
334                 /*
335                  * We handle just one known case where g is a quadratic non-residue:
336                  * for g = 2: p % 8 == 3
337                  */
338                 if (BN_is_word(dh->params.g, DH_GENERATOR_2)
339                     && !BN_is_bit_set(dh->params.p, 2)) {
340                     /* clear bit 0, since it won't be a secret anyway */
341                     if (!BN_clear_bit(priv_key, 0))
342                         goto err;
343                 }
344             } else
345 #endif
346             {
347                 /* Do a partial check for invalid p, q, g */
348                 if (!ossl_ffc_params_simple_validate(dh->libctx, &dh->params,
349                                                      FFC_PARAM_TYPE_DH, NULL))
350                     goto err;
351                 /*
352                  * For FFC FIPS 186-4 keygen
353                  * security strength s = 112,
354                  * Max Private key size N = len(q)
355                  */
356                 if (!ossl_ffc_generate_private_key(ctx, &dh->params,
357                                                    BN_num_bits(dh->params.q),
358                                                    MIN_STRENGTH,
359                                                    priv_key))
360                     goto err;
361             }
362         }
363     }
364 
365     if (!ossl_dh_generate_public_key(ctx, dh, priv_key, pub_key))
366         goto err;
367 
368     dh->pub_key = pub_key;
369     dh->priv_key = priv_key;
370     dh->dirty_cnt++;
371     ok = 1;
372  err:
373     if (ok != 1)
374         ERR_raise(ERR_LIB_DH, ERR_R_BN_LIB);
375 
376     if (pub_key != dh->pub_key)
377         BN_free(pub_key);
378     if (priv_key != dh->priv_key)
379         BN_free(priv_key);
380     BN_CTX_free(ctx);
381     return ok;
382 }
383 
ossl_dh_buf2key(DH * dh,const unsigned char * buf,size_t len)384 int ossl_dh_buf2key(DH *dh, const unsigned char *buf, size_t len)
385 {
386     int err_reason = DH_R_BN_ERROR;
387     BIGNUM *pubkey = NULL;
388     const BIGNUM *p;
389     int ret;
390 
391     if ((pubkey = BN_bin2bn(buf, len, NULL)) == NULL)
392         goto err;
393     DH_get0_pqg(dh, &p, NULL, NULL);
394     if (p == NULL || BN_num_bytes(p) == 0) {
395         err_reason = DH_R_NO_PARAMETERS_SET;
396         goto err;
397     }
398     /* Prevent small subgroup attacks per RFC 8446 Section 4.2.8.1 */
399     if (!ossl_dh_check_pub_key_partial(dh, pubkey, &ret)) {
400         err_reason = DH_R_INVALID_PUBKEY;
401         goto err;
402     }
403     if (DH_set0_key(dh, pubkey, NULL) != 1)
404         goto err;
405     return 1;
406 err:
407     ERR_raise(ERR_LIB_DH, err_reason);
408     BN_free(pubkey);
409     return 0;
410 }
411 
ossl_dh_key2buf(const DH * dh,unsigned char ** pbuf_out,size_t size,int alloc)412 size_t ossl_dh_key2buf(const DH *dh, unsigned char **pbuf_out, size_t size,
413                        int alloc)
414 {
415     const BIGNUM *pubkey;
416     unsigned char *pbuf = NULL;
417     const BIGNUM *p;
418     int p_size;
419 
420     DH_get0_pqg(dh, &p, NULL, NULL);
421     DH_get0_key(dh, &pubkey, NULL);
422     if (p == NULL || pubkey == NULL
423             || (p_size = BN_num_bytes(p)) == 0
424             || BN_num_bytes(pubkey) == 0) {
425         ERR_raise(ERR_LIB_DH, DH_R_INVALID_PUBKEY);
426         return 0;
427     }
428     if (pbuf_out != NULL && (alloc || *pbuf_out != NULL)) {
429         if (!alloc) {
430             if (size >= (size_t)p_size)
431                 pbuf = *pbuf_out;
432         } else {
433             pbuf = OPENSSL_malloc(p_size);
434         }
435 
436         if (pbuf == NULL) {
437             ERR_raise(ERR_LIB_DH, ERR_R_MALLOC_FAILURE);
438             return 0;
439         }
440         /*
441          * As per Section 4.2.8.1 of RFC 8446 left pad public
442          * key with zeros to the size of p
443          */
444         if (BN_bn2binpad(pubkey, pbuf, p_size) < 0) {
445             if (alloc)
446                 OPENSSL_free(pbuf);
447             ERR_raise(ERR_LIB_DH, DH_R_BN_ERROR);
448             return 0;
449         }
450         *pbuf_out = pbuf;
451     }
452     return p_size;
453 }
454