469746e65dc2314b5b1cbeeb29b4b638230696be
[openssl.git] / crypto / dsa / dsa_lib.c
1 /*
2  * Copyright 1995-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 #include <stdio.h>
11 #include "internal/cryptlib.h"
12 #include "internal/refcount.h"
13 #include <openssl/bn.h>
14 #include <openssl/asn1.h>
15 #include <openssl/engine.h>
16 #include "dsa_local.h"
17 #include "crypto/dsa.h"
18 #include "crypto/dh.h" /* required by DSA_dup_DH() */
19
20 #ifndef FIPS_MODE
21
22 int DSA_set_ex_data(DSA *d, int idx, void *arg)
23 {
24     return CRYPTO_set_ex_data(&d->ex_data, idx, arg);
25 }
26
27 void *DSA_get_ex_data(DSA *d, int idx)
28 {
29     return CRYPTO_get_ex_data(&d->ex_data, idx);
30 }
31
32 # ifndef OPENSSL_NO_DH
33 DH *DSA_dup_DH(const DSA *r)
34 {
35     /*
36      * DSA has p, q, g, optional pub_key, optional priv_key.
37      * DH has p, optional length, g, optional pub_key,
38      * optional priv_key, optional q.
39      */
40     DH *ret = NULL;
41     BIGNUM *pub_key = NULL, *priv_key = NULL;
42
43     if (r == NULL)
44         goto err;
45     ret = DH_new();
46     if (ret == NULL)
47         goto err;
48
49     if (!ffc_params_copy(dh_get0_params(ret), &r->params))
50         goto err;
51
52     if (r->pub_key != NULL) {
53         pub_key = BN_dup(r->pub_key);
54         if (pub_key == NULL)
55             goto err;
56         if (r->priv_key != NULL) {
57             priv_key = BN_dup(r->priv_key);
58             if (priv_key == NULL)
59                 goto err;
60         }
61         if (!DH_set0_key(ret, pub_key, priv_key))
62             goto err;
63     } else if (r->priv_key != NULL) {
64         /* Shouldn't happen */
65         goto err;
66     }
67
68     return ret;
69
70  err:
71     BN_free(pub_key);
72     BN_free(priv_key);
73     DH_free(ret);
74     return NULL;
75 }
76 # endif /*  OPENSSL_NO_DH */
77
78 const BIGNUM *DSA_get0_p(const DSA *d)
79 {
80     return d->params.p;
81 }
82
83 const BIGNUM *DSA_get0_q(const DSA *d)
84 {
85     return d->params.q;
86 }
87
88 const BIGNUM *DSA_get0_g(const DSA *d)
89 {
90     return d->params.g;
91 }
92
93 const BIGNUM *DSA_get0_pub_key(const DSA *d)
94 {
95     return d->pub_key;
96 }
97
98 const BIGNUM *DSA_get0_priv_key(const DSA *d)
99 {
100     return d->priv_key;
101 }
102
103 void DSA_clear_flags(DSA *d, int flags)
104 {
105     d->flags &= ~flags;
106 }
107
108 int DSA_test_flags(const DSA *d, int flags)
109 {
110     return d->flags & flags;
111 }
112
113 void DSA_set_flags(DSA *d, int flags)
114 {
115     d->flags |= flags;
116 }
117
118 ENGINE *DSA_get0_engine(DSA *d)
119 {
120     return d->engine;
121 }
122
123 int DSA_set_method(DSA *dsa, const DSA_METHOD *meth)
124 {
125     /*
126      * NB: The caller is specifically setting a method, so it's not up to us
127      * to deal with which ENGINE it comes from.
128      */
129     const DSA_METHOD *mtmp;
130     mtmp = dsa->meth;
131     if (mtmp->finish)
132         mtmp->finish(dsa);
133 #ifndef OPENSSL_NO_ENGINE
134     ENGINE_finish(dsa->engine);
135     dsa->engine = NULL;
136 #endif
137     dsa->meth = meth;
138     if (meth->init)
139         meth->init(dsa);
140     return 1;
141 }
142 #endif /* FIPS_MODE */
143
144
145 const DSA_METHOD *DSA_get_method(DSA *d)
146 {
147     return d->meth;
148 }
149
150 static DSA *dsa_new_method(OPENSSL_CTX *libctx, ENGINE *engine)
151 {
152     DSA *ret = OPENSSL_zalloc(sizeof(*ret));
153
154     if (ret == NULL) {
155         DSAerr(DSA_F_DSA_NEW_METHOD, ERR_R_MALLOC_FAILURE);
156         return NULL;
157     }
158
159     ret->references = 1;
160     ret->lock = CRYPTO_THREAD_lock_new();
161     if (ret->lock == NULL) {
162         DSAerr(DSA_F_DSA_NEW_METHOD, ERR_R_MALLOC_FAILURE);
163         OPENSSL_free(ret);
164         return NULL;
165     }
166
167     ret->meth = DSA_get_default_method();
168 #if !defined(FIPS_MODE) && !defined(OPENSSL_NO_ENGINE)
169     ret->flags = ret->meth->flags & ~DSA_FLAG_NON_FIPS_ALLOW; /* early default init */
170     if (engine) {
171         if (!ENGINE_init(engine)) {
172             DSAerr(DSA_F_DSA_NEW_METHOD, ERR_R_ENGINE_LIB);
173             goto err;
174         }
175         ret->engine = engine;
176     } else
177         ret->engine = ENGINE_get_default_DSA();
178     if (ret->engine) {
179         ret->meth = ENGINE_get_DSA(ret->engine);
180         if (ret->meth == NULL) {
181             DSAerr(DSA_F_DSA_NEW_METHOD, ERR_R_ENGINE_LIB);
182             goto err;
183         }
184     }
185 #endif
186
187     ret->flags = ret->meth->flags & ~DSA_FLAG_NON_FIPS_ALLOW;
188
189 #ifndef FIPS_MODE
190     if (!crypto_new_ex_data_ex(libctx, CRYPTO_EX_INDEX_DSA, ret, &ret->ex_data))
191         goto err;
192 #endif
193
194     if ((ret->meth->init != NULL) && !ret->meth->init(ret)) {
195         DSAerr(DSA_F_DSA_NEW_METHOD, ERR_R_INIT_FAIL);
196         goto err;
197     }
198
199     return ret;
200
201  err:
202     DSA_free(ret);
203     return NULL;
204 }
205
206 DSA *DSA_new_method(ENGINE *engine)
207 {
208     return dsa_new_method(NULL, engine);
209 }
210
211 DSA *DSA_new(void)
212 {
213     return DSA_new_method(NULL);
214 }
215
216 void DSA_free(DSA *r)
217 {
218     int i;
219
220     if (r == NULL)
221         return;
222
223     CRYPTO_DOWN_REF(&r->references, &i, r->lock);
224     REF_PRINT_COUNT("DSA", r);
225     if (i > 0)
226         return;
227     REF_ASSERT_ISNT(i < 0);
228
229     if (r->meth != NULL && r->meth->finish != NULL)
230         r->meth->finish(r);
231 #if !defined(FIPS_MODE) && !defined(OPENSSL_NO_ENGINE)
232     ENGINE_finish(r->engine);
233 #endif
234
235 #ifndef FIPS_MODE
236     CRYPTO_free_ex_data(CRYPTO_EX_INDEX_DSA, r, &r->ex_data);
237 #endif
238
239     CRYPTO_THREAD_lock_free(r->lock);
240
241     ffc_params_cleanup(&r->params);
242     BN_clear_free(r->pub_key);
243     BN_clear_free(r->priv_key);
244     OPENSSL_free(r);
245 }
246
247 int DSA_up_ref(DSA *r)
248 {
249     int i;
250
251     if (CRYPTO_UP_REF(&r->references, &i, r->lock) <= 0)
252         return 0;
253
254     REF_PRINT_COUNT("DSA", r);
255     REF_ASSERT_ISNT(i < 2);
256     return ((i > 1) ? 1 : 0);
257 }
258
259 void DSA_get0_pqg(const DSA *d,
260                   const BIGNUM **p, const BIGNUM **q, const BIGNUM **g)
261 {
262     ffc_params_get0_pqg(&d->params, p, q, g);
263 }
264
265 int DSA_set0_pqg(DSA *d, BIGNUM *p, BIGNUM *q, BIGNUM *g)
266 {
267     /* If the fields p, q and g in d are NULL, the corresponding input
268      * parameters MUST be non-NULL.
269      */
270     if ((d->params.p == NULL && p == NULL)
271         || (d->params.q == NULL && q == NULL)
272         || (d->params.g == NULL && g == NULL))
273         return 0;
274
275     ffc_params_set0_pqg(&d->params, p, q, g);
276     d->dirty_cnt++;
277
278     return 1;
279 }
280
281 void DSA_get0_key(const DSA *d,
282                   const BIGNUM **pub_key, const BIGNUM **priv_key)
283 {
284     if (pub_key != NULL)
285         *pub_key = d->pub_key;
286     if (priv_key != NULL)
287         *priv_key = d->priv_key;
288 }
289
290 int DSA_set0_key(DSA *d, BIGNUM *pub_key, BIGNUM *priv_key)
291 {
292     /* If the field pub_key in d is NULL, the corresponding input
293      * parameters MUST be non-NULL.  The priv_key field may
294      * be left NULL.
295      */
296     if (d->pub_key == NULL && pub_key == NULL)
297         return 0;
298
299     if (pub_key != NULL) {
300         BN_free(d->pub_key);
301         d->pub_key = pub_key;
302     }
303     if (priv_key != NULL) {
304         BN_free(d->priv_key);
305         d->priv_key = priv_key;
306     }
307     d->dirty_cnt++;
308
309     return 1;
310 }
311
312 int DSA_security_bits(const DSA *d)
313 {
314     if (d->params.p != NULL && d->params.q != NULL)
315         return BN_security_bits(BN_num_bits(d->params.p),
316                                 BN_num_bits(d->params.q));
317     return -1;
318 }
319
320 int DSA_bits(const DSA *dsa)
321 {
322     return BN_num_bits(dsa->params.p);
323 }