Fix memory leak
[openssl.git] / crypto / dh / dh_ameth.c
1 /*
2  * Written by Dr Stephen N Henson (steve@openssl.org) for the OpenSSL project
3  * 2006.
4  */
5 /* ====================================================================
6  * Copyright (c) 2006 The OpenSSL Project.  All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  *
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer.
14  *
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in
17  *    the documentation and/or other materials provided with the
18  *    distribution.
19  *
20  * 3. All advertising materials mentioning features or use of this
21  *    software must display the following acknowledgment:
22  *    "This product includes software developed by the OpenSSL Project
23  *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
24  *
25  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
26  *    endorse or promote products derived from this software without
27  *    prior written permission. For written permission, please contact
28  *    licensing@OpenSSL.org.
29  *
30  * 5. Products derived from this software may not be called "OpenSSL"
31  *    nor may "OpenSSL" appear in their names without prior written
32  *    permission of the OpenSSL Project.
33  *
34  * 6. Redistributions of any form whatsoever must retain the following
35  *    acknowledgment:
36  *    "This product includes software developed by the OpenSSL Project
37  *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
38  *
39  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
40  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
41  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
42  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
43  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
44  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
45  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
46  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
47  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
48  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
49  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
50  * OF THE POSSIBILITY OF SUCH DAMAGE.
51  * ====================================================================
52  *
53  * This product includes cryptographic software written by Eric Young
54  * (eay@cryptsoft.com).  This product includes software written by Tim
55  * Hudson (tjh@cryptsoft.com).
56  *
57  */
58
59 #include <stdio.h>
60 #include "cryptlib.h"
61 #include <openssl/x509.h>
62 #include <openssl/asn1.h>
63 #include <openssl/dh.h>
64 #include <openssl/bn.h>
65 #include "asn1_locl.h"
66
67 static void int_dh_free(EVP_PKEY *pkey)
68 {
69     DH_free(pkey->pkey.dh);
70 }
71
72 static int dh_pub_decode(EVP_PKEY *pkey, X509_PUBKEY *pubkey)
73 {
74     const unsigned char *p, *pm;
75     int pklen, pmlen;
76     int ptype;
77     void *pval;
78     ASN1_STRING *pstr;
79     X509_ALGOR *palg;
80     ASN1_INTEGER *public_key = NULL;
81
82     DH *dh = NULL;
83
84     if (!X509_PUBKEY_get0_param(NULL, &p, &pklen, &palg, pubkey))
85         return 0;
86     X509_ALGOR_get0(NULL, &ptype, &pval, palg);
87
88     if (ptype != V_ASN1_SEQUENCE) {
89         DHerr(DH_F_DH_PUB_DECODE, DH_R_PARAMETER_ENCODING_ERROR);
90         goto err;
91     }
92
93     pstr = pval;
94     pm = pstr->data;
95     pmlen = pstr->length;
96
97     if (!(dh = d2i_DHparams(NULL, &pm, pmlen))) {
98         DHerr(DH_F_DH_PUB_DECODE, DH_R_DECODE_ERROR);
99         goto err;
100     }
101
102     if (!(public_key = d2i_ASN1_INTEGER(NULL, &p, pklen))) {
103         DHerr(DH_F_DH_PUB_DECODE, DH_R_DECODE_ERROR);
104         goto err;
105     }
106
107     /* We have parameters now set public key */
108     if (!(dh->pub_key = ASN1_INTEGER_to_BN(public_key, NULL))) {
109         DHerr(DH_F_DH_PUB_DECODE, DH_R_BN_DECODE_ERROR);
110         goto err;
111     }
112
113     ASN1_INTEGER_free(public_key);
114     EVP_PKEY_assign_DH(pkey, dh);
115     return 1;
116
117  err:
118     if (public_key)
119         ASN1_INTEGER_free(public_key);
120     if (dh)
121         DH_free(dh);
122     return 0;
123
124 }
125
126 static int dh_pub_encode(X509_PUBKEY *pk, const EVP_PKEY *pkey)
127 {
128     DH *dh;
129     void *pval = NULL;
130     int ptype;
131     unsigned char *penc = NULL;
132     int penclen;
133     ASN1_STRING *str;
134     ASN1_INTEGER *pub_key = NULL;
135
136     dh = pkey->pkey.dh;
137
138     str = ASN1_STRING_new();
139     str->length = i2d_DHparams(dh, &str->data);
140     if (str->length <= 0) {
141         DHerr(DH_F_DH_PUB_ENCODE, ERR_R_MALLOC_FAILURE);
142         goto err;
143     }
144     pval = str;
145     ptype = V_ASN1_SEQUENCE;
146
147     pub_key = BN_to_ASN1_INTEGER(dh->pub_key, NULL);
148     if (!pub_key)
149         goto err;
150
151     penclen = i2d_ASN1_INTEGER(pub_key, &penc);
152
153     ASN1_INTEGER_free(pub_key);
154
155     if (penclen <= 0) {
156         DHerr(DH_F_DH_PUB_ENCODE, ERR_R_MALLOC_FAILURE);
157         goto err;
158     }
159
160     if (X509_PUBKEY_set0_param(pk, OBJ_nid2obj(EVP_PKEY_DH),
161                                ptype, pval, penc, penclen))
162         return 1;
163
164  err:
165     if (penc)
166         OPENSSL_free(penc);
167     if (pval)
168         ASN1_STRING_free(pval);
169
170     return 0;
171 }
172
173 /*
174  * PKCS#8 DH is defined in PKCS#11 of all places. It is similar to DH in that
175  * the AlgorithmIdentifier contains the paramaters, the private key is
176  * explcitly included and the pubkey must be recalculated.
177  */
178
179 static int dh_priv_decode(EVP_PKEY *pkey, PKCS8_PRIV_KEY_INFO *p8)
180 {
181     const unsigned char *p, *pm;
182     int pklen, pmlen;
183     int ptype;
184     void *pval;
185     ASN1_STRING *pstr;
186     X509_ALGOR *palg;
187     ASN1_INTEGER *privkey = NULL;
188
189     DH *dh = NULL;
190
191     if (!PKCS8_pkey_get0(NULL, &p, &pklen, &palg, p8))
192         return 0;
193
194     X509_ALGOR_get0(NULL, &ptype, &pval, palg);
195
196     if (ptype != V_ASN1_SEQUENCE)
197         goto decerr;
198
199     if (!(privkey = d2i_ASN1_INTEGER(NULL, &p, pklen)))
200         goto decerr;
201
202     pstr = pval;
203     pm = pstr->data;
204     pmlen = pstr->length;
205     if (!(dh = d2i_DHparams(NULL, &pm, pmlen)))
206         goto decerr;
207     /* We have parameters now set private key */
208     if (!(dh->priv_key = ASN1_INTEGER_to_BN(privkey, NULL))) {
209         DHerr(DH_F_DH_PRIV_DECODE, DH_R_BN_ERROR);
210         goto dherr;
211     }
212     /* Calculate public key */
213     if (!DH_generate_key(dh))
214         goto dherr;
215
216     EVP_PKEY_assign_DH(pkey, dh);
217
218     ASN1_INTEGER_free(privkey);
219
220     return 1;
221
222  decerr:
223     DHerr(DH_F_DH_PRIV_DECODE, EVP_R_DECODE_ERROR);
224  dherr:
225     DH_free(dh);
226     ASN1_INTEGER_free(privkey);
227     return 0;
228 }
229
230 static int dh_priv_encode(PKCS8_PRIV_KEY_INFO *p8, const EVP_PKEY *pkey)
231 {
232     ASN1_STRING *params = NULL;
233     ASN1_INTEGER *prkey = NULL;
234     unsigned char *dp = NULL;
235     int dplen;
236
237     params = ASN1_STRING_new();
238
239     if (!params) {
240         DHerr(DH_F_DH_PRIV_ENCODE, ERR_R_MALLOC_FAILURE);
241         goto err;
242     }
243
244     params->length = i2d_DHparams(pkey->pkey.dh, &params->data);
245     if (params->length <= 0) {
246         DHerr(DH_F_DH_PRIV_ENCODE, ERR_R_MALLOC_FAILURE);
247         goto err;
248     }
249     params->type = V_ASN1_SEQUENCE;
250
251     /* Get private key into integer */
252     prkey = BN_to_ASN1_INTEGER(pkey->pkey.dh->priv_key, NULL);
253
254     if (!prkey) {
255         DHerr(DH_F_DH_PRIV_ENCODE, DH_R_BN_ERROR);
256         goto err;
257     }
258
259     dplen = i2d_ASN1_INTEGER(prkey, &dp);
260
261     ASN1_INTEGER_free(prkey);
262     prkey = NULL;
263
264     if (!PKCS8_pkey_set0(p8, OBJ_nid2obj(NID_dhKeyAgreement), 0,
265                          V_ASN1_SEQUENCE, params, dp, dplen))
266         goto err;
267
268     return 1;
269
270  err:
271     if (dp != NULL)
272         OPENSSL_free(dp);
273     if (params != NULL)
274         ASN1_STRING_free(params);
275     if (prkey != NULL)
276         ASN1_INTEGER_free(prkey);
277     return 0;
278 }
279
280 static void update_buflen(const BIGNUM *b, size_t *pbuflen)
281 {
282     size_t i;
283     if (!b)
284         return;
285     if (*pbuflen < (i = (size_t)BN_num_bytes(b)))
286         *pbuflen = i;
287 }
288
289 static int dh_param_decode(EVP_PKEY *pkey,
290                            const unsigned char **pder, int derlen)
291 {
292     DH *dh;
293     if (!(dh = d2i_DHparams(NULL, pder, derlen))) {
294         DHerr(DH_F_DH_PARAM_DECODE, ERR_R_DH_LIB);
295         return 0;
296     }
297     EVP_PKEY_assign_DH(pkey, dh);
298     return 1;
299 }
300
301 static int dh_param_encode(const EVP_PKEY *pkey, unsigned char **pder)
302 {
303     return i2d_DHparams(pkey->pkey.dh, pder);
304 }
305
306 static int do_dh_print(BIO *bp, const DH *x, int indent,
307                        ASN1_PCTX *ctx, int ptype)
308 {
309     unsigned char *m = NULL;
310     int reason = ERR_R_BUF_LIB, ret = 0;
311     size_t buf_len = 0;
312
313     const char *ktype = NULL;
314
315     BIGNUM *priv_key, *pub_key;
316
317     if (ptype == 2)
318         priv_key = x->priv_key;
319     else
320         priv_key = NULL;
321
322     if (ptype > 0)
323         pub_key = x->pub_key;
324     else
325         pub_key = NULL;
326
327     update_buflen(x->p, &buf_len);
328
329     if (buf_len == 0) {
330         reason = ERR_R_PASSED_NULL_PARAMETER;
331         goto err;
332     }
333
334     update_buflen(x->g, &buf_len);
335     update_buflen(pub_key, &buf_len);
336     update_buflen(priv_key, &buf_len);
337
338     if (ptype == 2)
339         ktype = "PKCS#3 DH Private-Key";
340     else if (ptype == 1)
341         ktype = "PKCS#3 DH Public-Key";
342     else
343         ktype = "PKCS#3 DH Parameters";
344
345     m = OPENSSL_malloc(buf_len + 10);
346     if (m == NULL) {
347         reason = ERR_R_MALLOC_FAILURE;
348         goto err;
349     }
350
351     BIO_indent(bp, indent, 128);
352     if (BIO_printf(bp, "%s: (%d bit)\n", ktype, BN_num_bits(x->p)) <= 0)
353         goto err;
354     indent += 4;
355
356     if (!ASN1_bn_print(bp, "private-key:", priv_key, m, indent))
357         goto err;
358     if (!ASN1_bn_print(bp, "public-key:", pub_key, m, indent))
359         goto err;
360
361     if (!ASN1_bn_print(bp, "prime:", x->p, m, indent))
362         goto err;
363     if (!ASN1_bn_print(bp, "generator:", x->g, m, indent))
364         goto err;
365     if (x->length != 0) {
366         BIO_indent(bp, indent, 128);
367         if (BIO_printf(bp, "recommended-private-length: %d bits\n",
368                        (int)x->length) <= 0)
369             goto err;
370     }
371
372     ret = 1;
373     if (0) {
374  err:
375         DHerr(DH_F_DO_DH_PRINT, reason);
376     }
377     if (m != NULL)
378         OPENSSL_free(m);
379     return (ret);
380 }
381
382 static int int_dh_size(const EVP_PKEY *pkey)
383 {
384     return (DH_size(pkey->pkey.dh));
385 }
386
387 static int dh_bits(const EVP_PKEY *pkey)
388 {
389     return BN_num_bits(pkey->pkey.dh->p);
390 }
391
392 static int dh_cmp_parameters(const EVP_PKEY *a, const EVP_PKEY *b)
393 {
394     if (BN_cmp(a->pkey.dh->p, b->pkey.dh->p) ||
395         BN_cmp(a->pkey.dh->g, b->pkey.dh->g))
396         return 0;
397     else
398         return 1;
399 }
400
401 static int dh_copy_parameters(EVP_PKEY *to, const EVP_PKEY *from)
402 {
403     BIGNUM *a;
404
405     if ((a = BN_dup(from->pkey.dh->p)) == NULL)
406         return 0;
407     if (to->pkey.dh->p != NULL)
408         BN_free(to->pkey.dh->p);
409     to->pkey.dh->p = a;
410
411     if ((a = BN_dup(from->pkey.dh->g)) == NULL)
412         return 0;
413     if (to->pkey.dh->g != NULL)
414         BN_free(to->pkey.dh->g);
415     to->pkey.dh->g = a;
416
417     return 1;
418 }
419
420 static int dh_missing_parameters(const EVP_PKEY *a)
421 {
422     if (!a->pkey.dh->p || !a->pkey.dh->g)
423         return 1;
424     return 0;
425 }
426
427 static int dh_pub_cmp(const EVP_PKEY *a, const EVP_PKEY *b)
428 {
429     if (dh_cmp_parameters(a, b) == 0)
430         return 0;
431     if (BN_cmp(b->pkey.dh->pub_key, a->pkey.dh->pub_key) != 0)
432         return 0;
433     else
434         return 1;
435 }
436
437 static int dh_param_print(BIO *bp, const EVP_PKEY *pkey, int indent,
438                           ASN1_PCTX *ctx)
439 {
440     return do_dh_print(bp, pkey->pkey.dh, indent, ctx, 0);
441 }
442
443 static int dh_public_print(BIO *bp, const EVP_PKEY *pkey, int indent,
444                            ASN1_PCTX *ctx)
445 {
446     return do_dh_print(bp, pkey->pkey.dh, indent, ctx, 1);
447 }
448
449 static int dh_private_print(BIO *bp, const EVP_PKEY *pkey, int indent,
450                             ASN1_PCTX *ctx)
451 {
452     return do_dh_print(bp, pkey->pkey.dh, indent, ctx, 2);
453 }
454
455 int DHparams_print(BIO *bp, const DH *x)
456 {
457     return do_dh_print(bp, x, 4, NULL, 0);
458 }
459
460 const EVP_PKEY_ASN1_METHOD dh_asn1_meth = {
461     EVP_PKEY_DH,
462     EVP_PKEY_DH,
463     0,
464
465     "DH",
466     "OpenSSL PKCS#3 DH method",
467
468     dh_pub_decode,
469     dh_pub_encode,
470     dh_pub_cmp,
471     dh_public_print,
472
473     dh_priv_decode,
474     dh_priv_encode,
475     dh_private_print,
476
477     int_dh_size,
478     dh_bits,
479
480     dh_param_decode,
481     dh_param_encode,
482     dh_missing_parameters,
483     dh_copy_parameters,
484     dh_cmp_parameters,
485     dh_param_print,
486     0,
487
488     int_dh_free,
489     0
490 };