]>
Commit | Line | Data |
---|---|---|
1 | /* | |
2 | * Copyright 1995-2018 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 "dh_locl.h" | |
15 | #include <openssl/engine.h> | |
16 | ||
17 | int DH_set_method(DH *dh, const DH_METHOD *meth) | |
18 | { | |
19 | /* | |
20 | * NB: The caller is specifically setting a method, so it's not up to us | |
21 | * to deal with which ENGINE it comes from. | |
22 | */ | |
23 | const DH_METHOD *mtmp; | |
24 | mtmp = dh->meth; | |
25 | if (mtmp->finish) | |
26 | mtmp->finish(dh); | |
27 | #ifndef OPENSSL_NO_ENGINE | |
28 | ENGINE_finish(dh->engine); | |
29 | dh->engine = NULL; | |
30 | #endif | |
31 | dh->meth = meth; | |
32 | if (meth->init) | |
33 | meth->init(dh); | |
34 | return 1; | |
35 | } | |
36 | ||
37 | DH *DH_new(void) | |
38 | { | |
39 | return DH_new_method(NULL); | |
40 | } | |
41 | ||
42 | DH *DH_new_method(ENGINE *engine) | |
43 | { | |
44 | DH *ret = OPENSSL_zalloc(sizeof(*ret)); | |
45 | ||
46 | if (ret == NULL) { | |
47 | DHerr(DH_F_DH_NEW_METHOD, ERR_R_MALLOC_FAILURE); | |
48 | return NULL; | |
49 | } | |
50 | ||
51 | ret->references = 1; | |
52 | ret->lock = CRYPTO_THREAD_lock_new(); | |
53 | if (ret->lock == NULL) { | |
54 | DHerr(DH_F_DH_NEW_METHOD, ERR_R_MALLOC_FAILURE); | |
55 | OPENSSL_free(ret); | |
56 | return NULL; | |
57 | } | |
58 | ||
59 | ret->meth = DH_get_default_method(); | |
60 | #ifndef OPENSSL_NO_ENGINE | |
61 | ret->flags = ret->meth->flags; /* early default init */ | |
62 | if (engine) { | |
63 | if (!ENGINE_init(engine)) { | |
64 | DHerr(DH_F_DH_NEW_METHOD, ERR_R_ENGINE_LIB); | |
65 | goto err; | |
66 | } | |
67 | ret->engine = engine; | |
68 | } else | |
69 | ret->engine = ENGINE_get_default_DH(); | |
70 | if (ret->engine) { | |
71 | ret->meth = ENGINE_get_DH(ret->engine); | |
72 | if (ret->meth == NULL) { | |
73 | DHerr(DH_F_DH_NEW_METHOD, ERR_R_ENGINE_LIB); | |
74 | goto err; | |
75 | } | |
76 | } | |
77 | #endif | |
78 | ||
79 | ret->flags = ret->meth->flags; | |
80 | ||
81 | if (!CRYPTO_new_ex_data(CRYPTO_EX_INDEX_DH, ret, &ret->ex_data)) | |
82 | goto err; | |
83 | ||
84 | if ((ret->meth->init != NULL) && !ret->meth->init(ret)) { | |
85 | DHerr(DH_F_DH_NEW_METHOD, ERR_R_INIT_FAIL); | |
86 | goto err; | |
87 | } | |
88 | ||
89 | return ret; | |
90 | ||
91 | err: | |
92 | DH_free(ret); | |
93 | return NULL; | |
94 | } | |
95 | ||
96 | void DH_free(DH *r) | |
97 | { | |
98 | int i; | |
99 | ||
100 | if (r == NULL) | |
101 | return; | |
102 | ||
103 | CRYPTO_DOWN_REF(&r->references, &i, r->lock); | |
104 | REF_PRINT_COUNT("DH", r); | |
105 | if (i > 0) | |
106 | return; | |
107 | REF_ASSERT_ISNT(i < 0); | |
108 | ||
109 | if (r->meth != NULL && r->meth->finish != NULL) | |
110 | r->meth->finish(r); | |
111 | #ifndef OPENSSL_NO_ENGINE | |
112 | ENGINE_finish(r->engine); | |
113 | #endif | |
114 | ||
115 | CRYPTO_free_ex_data(CRYPTO_EX_INDEX_DH, r, &r->ex_data); | |
116 | ||
117 | CRYPTO_THREAD_lock_free(r->lock); | |
118 | ||
119 | BN_clear_free(r->p); | |
120 | BN_clear_free(r->g); | |
121 | BN_clear_free(r->q); | |
122 | BN_clear_free(r->j); | |
123 | OPENSSL_free(r->seed); | |
124 | BN_clear_free(r->counter); | |
125 | BN_clear_free(r->pub_key); | |
126 | BN_clear_free(r->priv_key); | |
127 | OPENSSL_free(r); | |
128 | } | |
129 | ||
130 | int DH_up_ref(DH *r) | |
131 | { | |
132 | int i; | |
133 | ||
134 | if (CRYPTO_UP_REF(&r->references, &i, r->lock) <= 0) | |
135 | return 0; | |
136 | ||
137 | REF_PRINT_COUNT("DH", r); | |
138 | REF_ASSERT_ISNT(i < 2); | |
139 | return ((i > 1) ? 1 : 0); | |
140 | } | |
141 | ||
142 | int DH_set_ex_data(DH *d, int idx, void *arg) | |
143 | { | |
144 | return CRYPTO_set_ex_data(&d->ex_data, idx, arg); | |
145 | } | |
146 | ||
147 | void *DH_get_ex_data(DH *d, int idx) | |
148 | { | |
149 | return CRYPTO_get_ex_data(&d->ex_data, idx); | |
150 | } | |
151 | ||
152 | int DH_bits(const DH *dh) | |
153 | { | |
154 | return BN_num_bits(dh->p); | |
155 | } | |
156 | ||
157 | int DH_size(const DH *dh) | |
158 | { | |
159 | return BN_num_bytes(dh->p); | |
160 | } | |
161 | ||
162 | int DH_security_bits(const DH *dh) | |
163 | { | |
164 | int N; | |
165 | if (dh->q) | |
166 | N = BN_num_bits(dh->q); | |
167 | else if (dh->length) | |
168 | N = dh->length; | |
169 | else | |
170 | N = -1; | |
171 | return BN_security_bits(BN_num_bits(dh->p), N); | |
172 | } | |
173 | ||
174 | ||
175 | void DH_get0_pqg(const DH *dh, | |
176 | const BIGNUM **p, const BIGNUM **q, const BIGNUM **g) | |
177 | { | |
178 | if (p != NULL) | |
179 | *p = dh->p; | |
180 | if (q != NULL) | |
181 | *q = dh->q; | |
182 | if (g != NULL) | |
183 | *g = dh->g; | |
184 | } | |
185 | ||
186 | int DH_set0_pqg(DH *dh, BIGNUM *p, BIGNUM *q, BIGNUM *g) | |
187 | { | |
188 | /* If the fields p and g in d are NULL, the corresponding input | |
189 | * parameters MUST be non-NULL. q may remain NULL. | |
190 | */ | |
191 | if ((dh->p == NULL && p == NULL) | |
192 | || (dh->g == NULL && g == NULL)) | |
193 | return 0; | |
194 | ||
195 | if (p != NULL) { | |
196 | BN_free(dh->p); | |
197 | dh->p = p; | |
198 | } | |
199 | if (q != NULL) { | |
200 | BN_free(dh->q); | |
201 | dh->q = q; | |
202 | } | |
203 | if (g != NULL) { | |
204 | BN_free(dh->g); | |
205 | dh->g = g; | |
206 | } | |
207 | ||
208 | if (q != NULL) { | |
209 | dh->length = BN_num_bits(q); | |
210 | } | |
211 | ||
212 | dh->dirty_cnt++; | |
213 | return 1; | |
214 | } | |
215 | ||
216 | long DH_get_length(const DH *dh) | |
217 | { | |
218 | return dh->length; | |
219 | } | |
220 | ||
221 | int DH_set_length(DH *dh, long length) | |
222 | { | |
223 | dh->length = length; | |
224 | return 1; | |
225 | } | |
226 | ||
227 | void DH_get0_key(const DH *dh, const BIGNUM **pub_key, const BIGNUM **priv_key) | |
228 | { | |
229 | if (pub_key != NULL) | |
230 | *pub_key = dh->pub_key; | |
231 | if (priv_key != NULL) | |
232 | *priv_key = dh->priv_key; | |
233 | } | |
234 | ||
235 | int DH_set0_key(DH *dh, BIGNUM *pub_key, BIGNUM *priv_key) | |
236 | { | |
237 | if (pub_key != NULL) { | |
238 | BN_free(dh->pub_key); | |
239 | dh->pub_key = pub_key; | |
240 | } | |
241 | if (priv_key != NULL) { | |
242 | BN_free(dh->priv_key); | |
243 | dh->priv_key = priv_key; | |
244 | } | |
245 | ||
246 | dh->dirty_cnt++; | |
247 | return 1; | |
248 | } | |
249 | ||
250 | const BIGNUM *DH_get0_p(const DH *dh) | |
251 | { | |
252 | return dh->p; | |
253 | } | |
254 | ||
255 | const BIGNUM *DH_get0_q(const DH *dh) | |
256 | { | |
257 | return dh->q; | |
258 | } | |
259 | ||
260 | const BIGNUM *DH_get0_g(const DH *dh) | |
261 | { | |
262 | return dh->g; | |
263 | } | |
264 | ||
265 | const BIGNUM *DH_get0_priv_key(const DH *dh) | |
266 | { | |
267 | return dh->priv_key; | |
268 | } | |
269 | ||
270 | const BIGNUM *DH_get0_pub_key(const DH *dh) | |
271 | { | |
272 | return dh->pub_key; | |
273 | } | |
274 | ||
275 | void DH_clear_flags(DH *dh, int flags) | |
276 | { | |
277 | dh->flags &= ~flags; | |
278 | } | |
279 | ||
280 | int DH_test_flags(const DH *dh, int flags) | |
281 | { | |
282 | return dh->flags & flags; | |
283 | } | |
284 | ||
285 | void DH_set_flags(DH *dh, int flags) | |
286 | { | |
287 | dh->flags |= flags; | |
288 | } | |
289 | ||
290 | ENGINE *DH_get0_engine(DH *dh) | |
291 | { | |
292 | return dh->engine; | |
293 | } |