Make DH_free() free up any ex_data and also call the finish method.
[openssl.git] / crypto / dh / dh_lib.c
1 /* crypto/dh/dh_lib.c */
2 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
3  * All rights reserved.
4  *
5  * This package is an SSL implementation written
6  * by Eric Young (eay@cryptsoft.com).
7  * The implementation was written so as to conform with Netscapes SSL.
8  * 
9  * This library is free for commercial and non-commercial use as long as
10  * the following conditions are aheared to.  The following conditions
11  * apply to all code found in this distribution, be it the RC4, RSA,
12  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
13  * included with this distribution is covered by the same copyright terms
14  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
15  * 
16  * Copyright remains Eric Young's, and as such any Copyright notices in
17  * the code are not to be removed.
18  * If this package is used in a product, Eric Young should be given attribution
19  * as the author of the parts of the library used.
20  * This can be in the form of a textual message at program startup or
21  * in documentation (online or textual) provided with the package.
22  * 
23  * Redistribution and use in source and binary forms, with or without
24  * modification, are permitted provided that the following conditions
25  * are met:
26  * 1. Redistributions of source code must retain the copyright
27  *    notice, this list of conditions and the following disclaimer.
28  * 2. Redistributions in binary form must reproduce the above copyright
29  *    notice, this list of conditions and the following disclaimer in the
30  *    documentation and/or other materials provided with the distribution.
31  * 3. All advertising materials mentioning features or use of this software
32  *    must display the following acknowledgement:
33  *    "This product includes cryptographic software written by
34  *     Eric Young (eay@cryptsoft.com)"
35  *    The word 'cryptographic' can be left out if the rouines from the library
36  *    being used are not cryptographic related :-).
37  * 4. If you include any Windows specific code (or a derivative thereof) from 
38  *    the apps directory (application code) you must include an acknowledgement:
39  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
40  * 
41  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
42  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
43  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
44  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
45  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
46  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
47  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
49  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
50  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
51  * SUCH DAMAGE.
52  * 
53  * The licence and distribution terms for any publically available version or
54  * derivative of this code cannot be changed.  i.e. this code cannot simply be
55  * copied and put under another distribution licence
56  * [including the GNU Public Licence.]
57  */
58
59 #include <stdio.h>
60 #include "cryptlib.h"
61 #include <openssl/bn.h>
62 #include <openssl/dh.h>
63
64 const char *DH_version="Diffie-Hellman" OPENSSL_VERSION_PTEXT;
65
66 static DH_METHOD *default_DH_method;
67 static int dh_meth_num = 0;
68 static STACK *dh_meth = NULL;
69
70 void DH_set_default_method(DH_METHOD *meth)
71 {
72         default_DH_method = meth;
73 }
74
75 DH_METHOD *DH_get_default_method(void)
76 {
77         if(!default_DH_method) default_DH_method = DH_OpenSSL();
78         return default_DH_method;
79 }
80
81 DH_METHOD *DH_set_method(DH *dh, DH_METHOD *meth)
82 {
83         DH_METHOD *mtmp;
84         mtmp = dh->meth;
85         if (mtmp->finish) mtmp->finish(dh);
86         dh->meth = meth;
87         if (meth->init) meth->init(dh);
88         return mtmp;
89 }
90
91 DH *DH_new(void)
92 {
93         return DH_new_method(NULL);
94 }
95
96 DH *DH_new_method(DH_METHOD *meth)
97         {
98         DH *ret;
99         ret=(DH *)Malloc(sizeof(DH));
100
101         if (ret == NULL)
102                 {
103                 DHerr(DH_F_DH_NEW,ERR_R_MALLOC_FAILURE);
104                 return(NULL);
105                 }
106         if(!default_DH_method) default_DH_method = DH_OpenSSL();
107         if(meth) ret->meth = meth;
108         else ret->meth = default_DH_method;
109         ret->pad=0;
110         ret->version=0;
111         ret->p=NULL;
112         ret->g=NULL;
113         ret->length=0;
114         ret->pub_key=NULL;
115         ret->priv_key=NULL;
116         ret->method_mont_p=NULL;
117         ret->references = 1;
118         ret->flags=ret->meth->flags;
119         if ((ret->meth->init != NULL) && !ret->meth->init(ret))
120                 {
121                 Free(ret);
122                 ret=NULL;
123                 }
124         else
125                 CRYPTO_new_ex_data(dh_meth,(char *)ret,&ret->ex_data);
126         return(ret);
127         }
128
129 void DH_free(DH *r)
130         {
131         int i;
132         if(r == NULL) return;
133         i = CRYPTO_add(&r->references, -1, CRYPTO_LOCK_DH);
134 #ifdef REF_PRINT
135         REF_PRINT("DH",r);
136 #endif
137         if (i > 0) return;
138 #ifdef REF_CHECK
139         if (i < 0)
140                 {
141                 fprintf(stderr,"DH_free, bad reference count\n");
142                 abort();
143         }
144 #endif
145
146         CRYPTO_free_ex_data(dh_meth, (char *)r, &r->ex_data);
147
148         if(r->meth->finish) r->meth->finish(r);
149
150         if (r->p != NULL) BN_clear_free(r->p);
151         if (r->g != NULL) BN_clear_free(r->g);
152         if (r->pub_key != NULL) BN_clear_free(r->pub_key);
153         if (r->priv_key != NULL) BN_clear_free(r->priv_key);
154         Free(r);
155         }
156
157 int DH_get_ex_new_index(long argl, char *argp, int (*new_func)(),
158              int (*dup_func)(), void (*free_func)())
159         {
160         dh_meth_num++;
161         return(CRYPTO_get_ex_new_index(dh_meth_num-1,
162                 &dh_meth,argl,argp,new_func,dup_func,free_func));
163         }
164
165 int DH_set_ex_data(DH *d, int idx, char *arg)
166         {
167         return(CRYPTO_set_ex_data(&d->ex_data,idx,arg));
168         }
169
170 char *DH_get_ex_data(DH *d, int idx)
171         {
172         return(CRYPTO_get_ex_data(&d->ex_data,idx));
173         }
174
175 int DH_size(DH *dh)
176         {
177         return(BN_num_bytes(dh->p));
178         }