Check the return values where memory allocation failures may happen.
[openssl.git] / crypto / asn1 / x_pubkey.c
1 /* crypto/asn1/x_pubkey.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/asn1t.h>
62 #include <openssl/x509.h>
63
64 /* Minor tweak to operation: free up EVP_PKEY */
65 static int pubkey_cb(int operation, ASN1_VALUE **pval, const ASN1_ITEM *it)
66         {
67         if (operation == ASN1_OP_FREE_POST)
68                 {
69                 X509_PUBKEY *pubkey = (X509_PUBKEY *)*pval;
70                 EVP_PKEY_free(pubkey->pkey);
71                 }
72         return 1;
73         }
74
75 ASN1_SEQUENCE_cb(X509_PUBKEY, pubkey_cb) = {
76         ASN1_SIMPLE(X509_PUBKEY, algor, X509_ALGOR),
77         ASN1_SIMPLE(X509_PUBKEY, public_key, ASN1_BIT_STRING)
78 } ASN1_SEQUENCE_END_cb(X509_PUBKEY, X509_PUBKEY)
79
80 IMPLEMENT_ASN1_FUNCTIONS(X509_PUBKEY)
81
82 int X509_PUBKEY_set(X509_PUBKEY **x, EVP_PKEY *pkey)
83         {
84         int ok=0;
85         X509_PUBKEY *pk;
86         X509_ALGOR *a;
87         ASN1_OBJECT *o;
88         unsigned char *s,*p = NULL;
89         int i;
90
91         if (x == NULL) return(0);
92
93         if ((pk=X509_PUBKEY_new()) == NULL) goto err;
94         a=pk->algor;
95
96         /* set the algorithm id */
97         if ((o=OBJ_nid2obj(pkey->type)) == NULL) goto err;
98         ASN1_OBJECT_free(a->algorithm);
99         a->algorithm=o;
100
101         /* Set the parameter list */
102         if (!pkey->save_parameters || (pkey->type == EVP_PKEY_RSA))
103                 {
104                 if ((a->parameter == NULL) ||
105                         (a->parameter->type != V_ASN1_NULL))
106                         {
107                         ASN1_TYPE_free(a->parameter);
108                         a->parameter=ASN1_TYPE_new();
109                         a->parameter->type=V_ASN1_NULL;
110                         }
111                 }
112 #ifndef OPENSSL_NO_DSA
113         else if (pkey->type == EVP_PKEY_DSA)
114                 {
115                 unsigned char *pp;
116                 DSA *dsa;
117                 
118                 dsa=pkey->pkey.dsa;
119                 dsa->write_params=0;
120                 ASN1_TYPE_free(a->parameter);
121                 i=i2d_DSAparams(dsa,NULL);
122                 if ((p=(unsigned char *)OPENSSL_malloc(i)) == NULL) goto err;
123                 pp=p;
124                 i2d_DSAparams(dsa,&pp);
125                 a->parameter=ASN1_TYPE_new();
126                 a->parameter->type=V_ASN1_SEQUENCE;
127                 a->parameter->value.sequence=ASN1_STRING_new();
128                 ASN1_STRING_set(a->parameter->value.sequence,p,i);
129                 OPENSSL_free(p);
130                 }
131 #endif
132 #ifndef OPENSSL_NO_ECDSA
133         else if (pkey->type == EVP_PKEY_ECDSA)
134                 {
135                 int nid=0;
136                 unsigned char *pp;
137                 ECDSA *ecdsa;
138                 
139                 ecdsa = pkey->pkey.ecdsa;
140                 ASN1_TYPE_free(a->parameter);
141
142                 if ((a->parameter = ASN1_TYPE_new()) == NULL)
143                         {
144                         X509err(X509_F_X509_PUBKEY_SET, ERR_R_ASN1_LIB);
145                         goto err;
146                         }
147
148                 if ((ECDSA_get_parameter_flags(ecdsa) & ECDSA_FLAG_NAMED_CURVE) && (nid = EC_GROUP_get_nid(ecdsa->group)))
149                         {
150                         /* just set the OID */
151                         a->parameter->type = V_ASN1_OBJECT;
152                         a->parameter->value.object = OBJ_nid2obj(nid);
153                         }
154                 else /* explicit parameters */
155                         {
156                         if ((i = i2d_ECDSAParameters(ecdsa, NULL)) == 0)
157                                 {
158                                 X509err(X509_F_X509_PUBKEY_SET, ERR_R_ECDSA_LIB);
159                                 goto err;
160                                 }
161                         if ((p = (unsigned char *) OPENSSL_malloc(i)) == NULL)
162                                 {
163                                 X509err(X509_F_X509_PUBKEY_SET, ERR_R_MALLOC_FAILURE);
164                                 goto err;
165                                 }       
166                         pp = p;
167                         if (!i2d_ECDSAParameters(ecdsa, &pp))
168                                 {
169                                 X509err(X509_F_X509_PUBKEY_SET, ERR_R_ECDSA_LIB);
170                                 OPENSSL_free(p);
171                                 goto err;
172                                 }
173                         a->parameter->type = V_ASN1_SEQUENCE;
174                         if ((a->parameter->value.sequence = ASN1_STRING_new()) == NULL)
175                                 {
176                                 X509err(X509_F_X509_PUBKEY_SET, ERR_R_ASN1_LIB);
177                                 OPENSSL_free(p);
178                                 goto err;
179                                 }
180                         ASN1_STRING_set(a->parameter->value.sequence, p, i);
181                         OPENSSL_free(p);
182                         }
183                 }
184 #endif
185         else if (1)
186                 {
187                 X509err(X509_F_X509_PUBKEY_SET,X509_R_UNSUPPORTED_ALGORITHM);
188                 goto err;
189                 }
190
191         if ((i=i2d_PublicKey(pkey,NULL)) <= 0) goto err;
192         if ((s=(unsigned char *)OPENSSL_malloc(i+1)) == NULL)
193                 {
194                 X509err(X509_F_X509_PUBKEY_SET,ERR_R_MALLOC_FAILURE);
195                 goto err;
196                 }
197         p=s;
198         i2d_PublicKey(pkey,&p);
199         if (!M_ASN1_BIT_STRING_set(pk->public_key,s,i)) goto err;
200         /* Set number of unused bits to zero */
201         pk->public_key->flags&= ~(ASN1_STRING_FLAG_BITS_LEFT|0x07);
202         pk->public_key->flags|=ASN1_STRING_FLAG_BITS_LEFT;
203
204         OPENSSL_free(s);
205
206 #if 0
207         CRYPTO_add(&pkey->references,1,CRYPTO_LOCK_EVP_PKEY);
208         pk->pkey=pkey;
209 #endif
210
211         if (*x != NULL)
212                 X509_PUBKEY_free(*x);
213
214         *x=pk;
215         pk=NULL;
216
217         ok=1;
218 err:
219         if (pk != NULL) X509_PUBKEY_free(pk);
220         return(ok);
221         }
222
223 EVP_PKEY *X509_PUBKEY_get(X509_PUBKEY *key)
224         {
225         EVP_PKEY *ret=NULL;
226         long j;
227         int type;
228         unsigned char *p;
229 #if !defined(OPENSSL_NO_DSA) || !defined(OPENSSL_NO_ECDSA)
230         const unsigned char *cp;
231         X509_ALGOR *a;
232 #endif
233
234         if (key == NULL) goto err;
235
236         if (key->pkey != NULL)
237                 {
238                 CRYPTO_add(&key->pkey->references, 1, CRYPTO_LOCK_EVP_PKEY);
239                 return(key->pkey);
240                 }
241
242         if (key->public_key == NULL) goto err;
243
244         type=OBJ_obj2nid(key->algor->algorithm);
245         if ((ret = EVP_PKEY_new()) == NULL)
246                 {
247                 X509err(X509_F_X509_PUBKEY_GET, ERR_R_MALLOC_FAILURE);
248                 goto err;
249                 }
250         ret->type = EVP_PKEY_type(type);
251
252         /* the parameters must be extracted before the public key (ECDSA!) */
253         
254 #if !defined(OPENSSL_NO_DSA) || !defined(OPENSSL_NO_ECDSA)
255         a=key->algor;
256 #endif
257
258         if (0)
259                 ;
260 #ifndef OPENSSL_NO_DSA
261         else if (ret->type == EVP_PKEY_DSA)
262                 {
263                 if (a->parameter && (a->parameter->type == V_ASN1_SEQUENCE))
264                         {
265                         if ((ret->pkey.dsa = DSA_new()) == NULL)
266                                 {
267                                 X509err(X509_F_X509_PUBKEY_GET, ERR_R_MALLOC_FAILURE);
268                                 goto err;
269                                 }
270                         ret->pkey.dsa->write_params=0;
271                         cp=p=a->parameter->value.sequence->data;
272                         j=a->parameter->value.sequence->length;
273                         if (!d2i_DSAparams(&ret->pkey.dsa, &cp, (long)j))
274                                 goto err;
275                         }
276                 ret->save_parameters=1;
277                 }
278 #endif
279 #ifndef OPENSSL_NO_ECDSA
280         else if (ret->type == EVP_PKEY_ECDSA)
281                 {
282                 if (a->parameter && (a->parameter->type == V_ASN1_SEQUENCE))
283                         {
284                         /* type == V_ASN1_SEQUENCE => we have explicit parameters
285                          * (e.g. parameters in the X9_62_EC_PARAMETERS-structure )
286                          */
287                         if ((ret->pkey.ecdsa= ECDSA_new()) == NULL)
288                                 {
289                                 X509err(X509_F_X509_PUBKEY_GET, ERR_R_MALLOC_FAILURE);
290                                 goto err;
291                                 }
292                         cp = p = a->parameter->value.sequence->data;
293                         j = a->parameter->value.sequence->length;
294                         if (!d2i_ECDSAParameters(&ret->pkey.ecdsa, &cp, (long)j))
295                                 {
296                                 X509err(X509_F_X509_PUBKEY_GET, ERR_R_ECDSA_LIB);
297                                 goto err;
298                                 }
299                         }
300                 else if (a->parameter && (a->parameter->type == V_ASN1_OBJECT))
301                         {
302                         /* type == V_ASN1_OBJECT => the parameters are given
303                          * by an asn1 OID
304                          */
305                         if (ret->pkey.ecdsa == NULL)
306                                 ret->pkey.ecdsa = ECDSA_new();
307                         if (ret->pkey.ecdsa->group)
308                                 EC_GROUP_free(ret->pkey.ecdsa->group);
309                         ret->pkey.ecdsa->parameter_flags |= ECDSA_FLAG_NAMED_CURVE;
310                         if ((ret->pkey.ecdsa->group = EC_GROUP_new_by_name(OBJ_obj2nid(a->parameter->value.object))) == NULL)
311                                 goto err;
312                         }
313                         /* the case implicitlyCA is currently not implemented */
314                 ret->save_parameters = 1;
315                 }
316 #endif
317
318         p=key->public_key->data;
319         j=key->public_key->length;
320         if ((ret = d2i_PublicKey(type, &ret, &p, (long)j)) == NULL)
321                 {
322                 X509err(X509_F_X509_PUBKEY_GET, X509_R_ERR_ASN1_LIB);
323                 goto err;
324                 }
325
326         key->pkey = ret;
327         CRYPTO_add(&ret->references, 1, CRYPTO_LOCK_EVP_PKEY);
328         return(ret);
329 err:
330         if (ret != NULL)
331                 EVP_PKEY_free(ret);
332         return(NULL);
333         }
334
335 /* Now two pseudo ASN1 routines that take an EVP_PKEY structure
336  * and encode or decode as X509_PUBKEY
337  */
338
339 EVP_PKEY *d2i_PUBKEY(EVP_PKEY **a, unsigned char **pp,
340              long length)
341         {
342         X509_PUBKEY *xpk;
343         EVP_PKEY *pktmp;
344         xpk = d2i_X509_PUBKEY(NULL, pp, length);
345         if(!xpk) return NULL;
346         pktmp = X509_PUBKEY_get(xpk);
347         X509_PUBKEY_free(xpk);
348         if(!pktmp) return NULL;
349         if(a)
350                 {
351                 EVP_PKEY_free(*a);
352                 *a = pktmp;
353                 }
354         return pktmp;
355         }
356
357 int i2d_PUBKEY(EVP_PKEY *a, unsigned char **pp)
358         {
359         X509_PUBKEY *xpk=NULL;
360         int ret;
361         if(!a) return 0;
362         if(!X509_PUBKEY_set(&xpk, a)) return 0;
363         ret = i2d_X509_PUBKEY(xpk, pp);
364         X509_PUBKEY_free(xpk);
365         return ret;
366         }
367
368 /* The following are equivalents but which return RSA and DSA
369  * keys
370  */
371 #ifndef OPENSSL_NO_RSA
372 RSA *d2i_RSA_PUBKEY(RSA **a, unsigned char **pp,
373              long length)
374         {
375         EVP_PKEY *pkey;
376         RSA *key;
377         unsigned char *q;
378         q = *pp;
379         pkey = d2i_PUBKEY(NULL, &q, length);
380         if (!pkey) return NULL;
381         key = EVP_PKEY_get1_RSA(pkey);
382         EVP_PKEY_free(pkey);
383         if (!key) return NULL;
384         *pp = q;
385         if (a)
386                 {
387                 RSA_free(*a);
388                 *a = key;
389                 }
390         return key;
391         }
392
393 int i2d_RSA_PUBKEY(RSA *a, unsigned char **pp)
394         {
395         EVP_PKEY *pktmp;
396         int ret;
397         if (!a) return 0;
398         pktmp = EVP_PKEY_new();
399         if (!pktmp)
400                 {
401                 ASN1err(ASN1_F_I2D_RSA_PUBKEY, ERR_R_MALLOC_FAILURE);
402                 return 0;
403                 }
404         EVP_PKEY_set1_RSA(pktmp, a);
405         ret = i2d_PUBKEY(pktmp, pp);
406         EVP_PKEY_free(pktmp);
407         return ret;
408         }
409 #endif
410
411 #ifndef OPENSSL_NO_DSA
412 DSA *d2i_DSA_PUBKEY(DSA **a, unsigned char **pp,
413              long length)
414         {
415         EVP_PKEY *pkey;
416         DSA *key;
417         unsigned char *q;
418         q = *pp;
419         pkey = d2i_PUBKEY(NULL, &q, length);
420         if (!pkey) return NULL;
421         key = EVP_PKEY_get1_DSA(pkey);
422         EVP_PKEY_free(pkey);
423         if (!key) return NULL;
424         *pp = q;
425         if (a)
426                 {
427                 DSA_free(*a);
428                 *a = key;
429                 }
430         return key;
431         }
432
433 int i2d_DSA_PUBKEY(DSA *a, unsigned char **pp)
434         {
435         EVP_PKEY *pktmp;
436         int ret;
437         if(!a) return 0;
438         pktmp = EVP_PKEY_new();
439         if(!pktmp)
440                 {
441                 ASN1err(ASN1_F_I2D_DSA_PUBKEY, ERR_R_MALLOC_FAILURE);
442                 return 0;
443                 }
444         EVP_PKEY_set1_DSA(pktmp, a);
445         ret = i2d_PUBKEY(pktmp, pp);
446         EVP_PKEY_free(pktmp);
447         return ret;
448         }
449 #endif
450
451 #ifndef OPENSSL_NO_ECDSA
452 ECDSA *d2i_ECDSA_PUBKEY(ECDSA **a, unsigned char **pp, long length)
453         {
454         EVP_PKEY *pkey;
455         ECDSA *key;
456         unsigned char *q;
457         q = *pp;
458         pkey = d2i_PUBKEY(NULL, &q, length);
459         if (!pkey) return(NULL);
460         key = EVP_PKEY_get1_ECDSA(pkey);
461         EVP_PKEY_free(pkey);
462         if (!key)  return(NULL);
463         *pp = q;
464         if (a)
465                 {
466                 ECDSA_free(*a);
467                 *a = key;
468                 }
469         return(key);
470         }
471
472 int i2d_ECDSA_PUBKEY(ECDSA *a, unsigned char **pp)
473         {
474         EVP_PKEY *pktmp;
475         int ret;
476         if (!a) return(0);
477         if ((pktmp = EVP_PKEY_new()) == NULL)
478                 {
479                 ASN1err(ASN1_F_I2D_ECDSA_PUBKEY, ERR_R_MALLOC_FAILURE);
480                 return(0);
481                 }
482         EVP_PKEY_set1_ECDSA(pktmp, a);
483         ret = i2d_PUBKEY(pktmp, pp);
484         EVP_PKEY_free(pktmp);
485         return(ret);
486         }
487 #endif