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