Add missing #include.
[oweals/openssl.git] / crypto / pkcs7 / pk7_doit.c
1 /* crypto/pkcs7/pk7_doit.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/rand.h>
62 #include <openssl/objects.h>
63 #include <openssl/x509.h>
64 #include <openssl/x509v3.h>
65
66 static int add_attribute(STACK_OF(X509_ATTRIBUTE) **sk, int nid, int atrtype,
67                          void *value);
68 static ASN1_TYPE *get_attribute(STACK_OF(X509_ATTRIBUTE) *sk, int nid);
69
70 BIO *PKCS7_dataInit(PKCS7 *p7, BIO *bio)
71         {
72         int i,j;
73         BIO *out=NULL,*btmp=NULL;
74         X509_ALGOR *xa;
75         const EVP_MD *evp_md;
76         const EVP_CIPHER *evp_cipher=NULL;
77         STACK_OF(X509_ALGOR) *md_sk=NULL;
78         STACK_OF(PKCS7_RECIP_INFO) *rsk=NULL;
79         X509_ALGOR *xalg=NULL;
80         PKCS7_RECIP_INFO *ri=NULL;
81         EVP_PKEY *pkey;
82
83         i=OBJ_obj2nid(p7->type);
84         p7->state=PKCS7_S_HEADER;
85
86         switch (i)
87                 {
88         case NID_pkcs7_signed:
89                 md_sk=p7->d.sign->md_algs;
90                 break;
91         case NID_pkcs7_signedAndEnveloped:
92                 rsk=p7->d.signed_and_enveloped->recipientinfo;
93                 md_sk=p7->d.signed_and_enveloped->md_algs;
94                 xalg=p7->d.signed_and_enveloped->enc_data->algorithm;
95                 evp_cipher=p7->d.signed_and_enveloped->enc_data->cipher;
96                 if (evp_cipher == NULL)
97                         {
98                         PKCS7err(PKCS7_F_PKCS7_DATAINIT,
99                                                 PKCS7_R_CIPHER_NOT_INITIALIZED);
100                         goto err;
101                         }
102                 break;
103         case NID_pkcs7_enveloped:
104                 rsk=p7->d.enveloped->recipientinfo;
105                 xalg=p7->d.enveloped->enc_data->algorithm;
106                 evp_cipher=p7->d.enveloped->enc_data->cipher;
107                 if (evp_cipher == NULL)
108                         {
109                         PKCS7err(PKCS7_F_PKCS7_DATAINIT,
110                                                 PKCS7_R_CIPHER_NOT_INITIALIZED);
111                         goto err;
112                         }
113                 break;
114         default:
115                 PKCS7err(PKCS7_F_PKCS7_DATAINIT,PKCS7_R_UNSUPPORTED_CONTENT_TYPE);
116                 goto err;
117                 }
118
119         if (md_sk != NULL)
120                 {
121                 for (i=0; i<sk_X509_ALGOR_num(md_sk); i++)
122                         {
123                         xa=sk_X509_ALGOR_value(md_sk,i);
124                         if ((btmp=BIO_new(BIO_f_md())) == NULL)
125                                 {
126                                 PKCS7err(PKCS7_F_PKCS7_DATAINIT,ERR_R_BIO_LIB);
127                                 goto err;
128                                 }
129
130                         j=OBJ_obj2nid(xa->algorithm);
131                         evp_md=EVP_get_digestbyname(OBJ_nid2sn(j));
132                         if (evp_md == NULL)
133                                 {
134                                 PKCS7err(PKCS7_F_PKCS7_DATAINIT,PKCS7_R_UNKNOWN_DIGEST_TYPE);
135                                 goto err;
136                                 }
137
138                         BIO_set_md(btmp,evp_md);
139                         if (out == NULL)
140                                 out=btmp;
141                         else
142                                 BIO_push(out,btmp);
143                         btmp=NULL;
144                         }
145                 }
146
147         if (evp_cipher != NULL)
148                 {
149                 unsigned char key[EVP_MAX_KEY_LENGTH];
150                 unsigned char iv[EVP_MAX_IV_LENGTH];
151                 int keylen,ivlen;
152                 int jj,max;
153                 unsigned char *tmp;
154                 EVP_CIPHER_CTX *ctx;
155
156                 if ((btmp=BIO_new(BIO_f_cipher())) == NULL)
157                         {
158                         PKCS7err(PKCS7_F_PKCS7_DATAINIT,ERR_R_BIO_LIB);
159                         goto err;
160                         }
161                 BIO_get_cipher_ctx(btmp, &ctx);
162                 keylen=EVP_CIPHER_key_length(evp_cipher);
163                 ivlen=EVP_CIPHER_iv_length(evp_cipher);
164                 if (RAND_bytes(key,keylen) <= 0)
165                         goto err;
166                 xalg->algorithm = OBJ_nid2obj(EVP_CIPHER_type(evp_cipher));
167                 if (ivlen > 0) RAND_pseudo_bytes(iv,ivlen);
168                 EVP_CipherInit(ctx, evp_cipher, key, iv, 1);
169
170                 if (ivlen > 0) {
171                         if (xalg->parameter == NULL) 
172                                                 xalg->parameter=ASN1_TYPE_new();
173                         if(EVP_CIPHER_param_to_asn1(ctx, xalg->parameter) < 0)
174                                                                        goto err;
175                 }
176
177                 /* Lets do the pub key stuff :-) */
178                 max=0;
179                 for (i=0; i<sk_PKCS7_RECIP_INFO_num(rsk); i++)
180                         {
181                         ri=sk_PKCS7_RECIP_INFO_value(rsk,i);
182                         if (ri->cert == NULL)
183                                 {
184                                 PKCS7err(PKCS7_F_PKCS7_DATAINIT,PKCS7_R_MISSING_CERIPEND_INFO);
185                                 goto err;
186                                 }
187                         pkey=X509_get_pubkey(ri->cert);
188                         jj=EVP_PKEY_size(pkey);
189                         EVP_PKEY_free(pkey);
190                         if (max < jj) max=jj;
191                         }
192                 if ((tmp=(unsigned char *)Malloc(max)) == NULL)
193                         {
194                         PKCS7err(PKCS7_F_PKCS7_DATAINIT,ERR_R_MALLOC_FAILURE);
195                         goto err;
196                         }
197                 for (i=0; i<sk_PKCS7_RECIP_INFO_num(rsk); i++)
198                         {
199                         ri=sk_PKCS7_RECIP_INFO_value(rsk,i);
200                         pkey=X509_get_pubkey(ri->cert);
201                         jj=EVP_PKEY_encrypt(tmp,key,keylen,pkey);
202                         EVP_PKEY_free(pkey);
203                         if (jj <= 0)
204                                 {
205                                 PKCS7err(PKCS7_F_PKCS7_DATAINIT,ERR_R_EVP_LIB);
206                                 Free(tmp);
207                                 goto err;
208                                 }
209                         M_ASN1_OCTET_STRING_set(ri->enc_key,tmp,jj);
210                         }
211                 Free(tmp);
212                 memset(key, 0, keylen);
213
214                 if (out == NULL)
215                         out=btmp;
216                 else
217                         BIO_push(out,btmp);
218                 btmp=NULL;
219                 }
220
221         if (bio == NULL) {
222                 if (p7->detached)
223                         bio=BIO_new(BIO_s_null());
224                 else {
225                         if (PKCS7_type_is_signed(p7) &&
226                                 PKCS7_type_is_data(p7->d.sign->contents)) {
227                                 ASN1_OCTET_STRING *os;
228                                 os=p7->d.sign->contents->d.data;
229                                 if (os->length > 0) bio = 
230                                         BIO_new_mem_buf(os->data, os->length);
231                         } 
232                         if(bio == NULL) {
233                                 bio=BIO_new(BIO_s_mem());
234                                 BIO_set_mem_eof_return(bio,0);
235                         }
236                 }
237         }
238         BIO_push(out,bio);
239         bio=NULL;
240         if (0)
241                 {
242 err:
243                 if (out != NULL)
244                         BIO_free_all(out);
245                 if (btmp != NULL)
246                         BIO_free_all(btmp);
247                 out=NULL;
248                 }
249         return(out);
250         }
251
252 /* int */
253 BIO *PKCS7_dataDecode(PKCS7 *p7, EVP_PKEY *pkey, BIO *in_bio, X509 *pcert)
254         {
255         int i,j;
256         BIO *out=NULL,*btmp=NULL,*etmp=NULL,*bio=NULL;
257         unsigned char *tmp=NULL;
258         X509_ALGOR *xa;
259         ASN1_OCTET_STRING *data_body=NULL;
260         const EVP_MD *evp_md;
261         const EVP_CIPHER *evp_cipher=NULL;
262         EVP_CIPHER_CTX *evp_ctx=NULL;
263         X509_ALGOR *enc_alg=NULL;
264         STACK_OF(X509_ALGOR) *md_sk=NULL;
265         STACK_OF(PKCS7_RECIP_INFO) *rsk=NULL;
266         X509_ALGOR *xalg=NULL;
267         PKCS7_RECIP_INFO *ri=NULL;
268 #ifndef NO_RC2
269         char is_rc2 = 0;
270 #endif
271 /*      EVP_PKEY *pkey; */
272 #if 0
273         X509_STORE_CTX s_ctx;
274 #endif
275
276         i=OBJ_obj2nid(p7->type);
277         p7->state=PKCS7_S_HEADER;
278
279         switch (i)
280                 {
281         case NID_pkcs7_signed:
282                 data_body=p7->d.sign->contents->d.data;
283                 md_sk=p7->d.sign->md_algs;
284                 break;
285         case NID_pkcs7_signedAndEnveloped:
286                 rsk=p7->d.signed_and_enveloped->recipientinfo;
287                 md_sk=p7->d.signed_and_enveloped->md_algs;
288                 data_body=p7->d.signed_and_enveloped->enc_data->enc_data;
289                 enc_alg=p7->d.signed_and_enveloped->enc_data->algorithm;
290                 evp_cipher=EVP_get_cipherbyname(OBJ_nid2sn(OBJ_obj2nid(enc_alg->algorithm)));
291                 if (evp_cipher == NULL)
292                         {
293                         PKCS7err(PKCS7_F_PKCS7_DATADECODE,PKCS7_R_UNSUPPORTED_CIPHER_TYPE);
294                         goto err;
295                         }
296                 xalg=p7->d.signed_and_enveloped->enc_data->algorithm;
297                 break;
298         case NID_pkcs7_enveloped:
299                 rsk=p7->d.enveloped->recipientinfo;
300                 enc_alg=p7->d.enveloped->enc_data->algorithm;
301                 data_body=p7->d.enveloped->enc_data->enc_data;
302                 evp_cipher=EVP_get_cipherbyname(OBJ_nid2sn(OBJ_obj2nid(enc_alg->algorithm)));
303                 if (evp_cipher == NULL)
304                         {
305                         PKCS7err(PKCS7_F_PKCS7_DATADECODE,PKCS7_R_UNSUPPORTED_CIPHER_TYPE);
306                         goto err;
307                         }
308                 xalg=p7->d.enveloped->enc_data->algorithm;
309                 break;
310         default:
311                 PKCS7err(PKCS7_F_PKCS7_DATADECODE,PKCS7_R_UNSUPPORTED_CONTENT_TYPE);
312                 goto err;
313                 }
314
315         if(EVP_CIPHER_nid(evp_cipher) == NID_rc2_cbc)
316                 {
317 #ifndef NO_RC2          
318                 is_rc2 = 1; 
319 #else
320                 PKCS7err(PKCS7_F_PKCS7_DATADECODE,PKCS7_R_UNSUPPORTED_CIPHER_TYPE);
321                 goto err;
322 #endif
323                 }
324
325         /* We will be checking the signature */
326         if (md_sk != NULL)
327                 {
328                 for (i=0; i<sk_X509_ALGOR_num(md_sk); i++)
329                         {
330                         xa=sk_X509_ALGOR_value(md_sk,i);
331                         if ((btmp=BIO_new(BIO_f_md())) == NULL)
332                                 {
333                                 PKCS7err(PKCS7_F_PKCS7_DATADECODE,ERR_R_BIO_LIB);
334                                 goto err;
335                                 }
336
337                         j=OBJ_obj2nid(xa->algorithm);
338                         evp_md=EVP_get_digestbyname(OBJ_nid2sn(j));
339                         if (evp_md == NULL)
340                                 {
341                                 PKCS7err(PKCS7_F_PKCS7_DATADECODE,PKCS7_R_UNKNOWN_DIGEST_TYPE);
342                                 goto err;
343                                 }
344
345                         BIO_set_md(btmp,evp_md);
346                         if (out == NULL)
347                                 out=btmp;
348                         else
349                                 BIO_push(out,btmp);
350                         btmp=NULL;
351                         }
352                 }
353
354         if (evp_cipher != NULL)
355                 {
356 #if 0
357                 unsigned char key[EVP_MAX_KEY_LENGTH];
358                 unsigned char iv[EVP_MAX_IV_LENGTH];
359                 unsigned char *p;
360                 int keylen,ivlen;
361                 int max;
362                 X509_OBJECT ret;
363 #endif
364                 int jj;
365
366                 if ((etmp=BIO_new(BIO_f_cipher())) == NULL)
367                         {
368                         PKCS7err(PKCS7_F_PKCS7_DATADECODE,ERR_R_BIO_LIB);
369                         goto err;
370                         }
371
372                 /* It was encrypted, we need to decrypt the secret key
373                  * with the private key */
374
375                 /* Find the recipientInfo which matches the passed certificate
376                  * (if any)
377                  */
378
379                 for (i=0; i<sk_PKCS7_RECIP_INFO_num(rsk); i++) {
380                         ri=sk_PKCS7_RECIP_INFO_value(rsk,i);
381                         if(!X509_NAME_cmp(ri->issuer_and_serial->issuer,
382                                         pcert->cert_info->issuer) &&
383                              !M_ASN1_INTEGER_cmp(pcert->cert_info->serialNumber,
384                                         ri->issuer_and_serial->serial)) break;
385                         ri=NULL;
386                 }
387                 if (ri == NULL) {
388                         PKCS7err(PKCS7_F_PKCS7_DATADECODE,
389                                  PKCS7_R_NO_RECIPIENT_MATCHES_CERTIFICATE);
390                         return(NULL);
391                 }
392
393                 jj=EVP_PKEY_size(pkey);
394                 tmp=(unsigned char *)Malloc(jj+10);
395                 if (tmp == NULL)
396                         {
397                         PKCS7err(PKCS7_F_PKCS7_DATADECODE,ERR_R_MALLOC_FAILURE);
398                         goto err;
399                         }
400
401                 jj=EVP_PKEY_decrypt(tmp, M_ASN1_STRING_data(ri->enc_key),
402                         M_ASN1_STRING_length(ri->enc_key), pkey);
403                 if (jj <= 0)
404                         {
405                         PKCS7err(PKCS7_F_PKCS7_DATADECODE,ERR_R_EVP_LIB);
406                         goto err;
407                         }
408
409                 evp_ctx=NULL;
410                 BIO_get_cipher_ctx(etmp,&evp_ctx);
411                 EVP_CipherInit(evp_ctx,evp_cipher,NULL,NULL,0);
412                 if (EVP_CIPHER_asn1_to_param(evp_ctx,enc_alg->parameter) < 0)
413                         return(NULL);
414
415                 if (jj != EVP_CIPHER_CTX_key_length(evp_ctx)) {
416                         /* HACK: some S/MIME clients don't use the same key
417                          * and effective key length. The key length is
418                          * determined by the size of the decrypted RSA key.
419                          * So we hack things to manually set the RC2 key
420                          * because we currently can't do this with the EVP
421                          * interface.
422                          */
423 #ifndef NO_RC2          
424                         if(is_rc2) RC2_set_key(&(evp_ctx->c.rc2_ks),jj, tmp,
425                                         EVP_CIPHER_CTX_key_length(evp_ctx)*8);
426                         else
427 #endif
428                                 {
429                                 PKCS7err(PKCS7_F_PKCS7_DATADECODE,
430                                         PKCS7_R_DECRYPTED_KEY_IS_WRONG_LENGTH);
431                                 goto err;
432                                 }
433                 } else EVP_CipherInit(evp_ctx,NULL,tmp,NULL,0);
434
435                 memset(tmp,0,jj);
436
437                 if (out == NULL)
438                         out=etmp;
439                 else
440                         BIO_push(out,etmp);
441                 etmp=NULL;
442                 }
443
444 #if 1
445         if (p7->detached || (in_bio != NULL))
446                 {
447                 bio=in_bio;
448                 }
449         else 
450                 {
451 #if 0
452                 bio=BIO_new(BIO_s_mem());
453                 /* We need to set this so that when we have read all
454                  * the data, the encrypt BIO, if present, will read
455                  * EOF and encode the last few bytes */
456                 BIO_set_mem_eof_return(bio,0);
457
458                 if (data_body->length > 0)
459                         BIO_write(bio,(char *)data_body->data,data_body->length);
460 #else
461                 if (data_body->length > 0)
462                       bio = BIO_new_mem_buf(data_body->data,data_body->length);
463                 else {
464                         bio=BIO_new(BIO_s_mem());
465                         BIO_set_mem_eof_return(bio,0);
466                 }
467 #endif
468                 }
469         BIO_push(out,bio);
470         bio=NULL;
471 #endif
472         if (0)
473                 {
474 err:
475                 if (out != NULL) BIO_free_all(out);
476                 if (btmp != NULL) BIO_free_all(btmp);
477                 if (etmp != NULL) BIO_free_all(etmp);
478                 if (bio != NULL) BIO_free_all(bio);
479                 out=NULL;
480                 }
481         if (tmp != NULL)
482                 Free(tmp);
483         return(out);
484         }
485
486 int PKCS7_dataFinal(PKCS7 *p7, BIO *bio)
487         {
488         int ret=0;
489         int i,j;
490         BIO *btmp;
491         BUF_MEM *buf_mem=NULL;
492         BUF_MEM *buf=NULL;
493         PKCS7_SIGNER_INFO *si;
494         EVP_MD_CTX *mdc,ctx_tmp;
495         STACK_OF(X509_ATTRIBUTE) *sk;
496         STACK_OF(PKCS7_SIGNER_INFO) *si_sk=NULL;
497         unsigned char *p,*pp=NULL;
498         int x;
499         ASN1_OCTET_STRING *os=NULL;
500
501         i=OBJ_obj2nid(p7->type);
502         p7->state=PKCS7_S_HEADER;
503
504         switch (i)
505                 {
506         case NID_pkcs7_signedAndEnveloped:
507                 /* XXXXXXXXXXXXXXXX */
508                 si_sk=p7->d.signed_and_enveloped->signer_info;
509                 os=M_ASN1_OCTET_STRING_new();
510                 p7->d.signed_and_enveloped->enc_data->enc_data=os;
511                 break;
512         case NID_pkcs7_enveloped:
513                 /* XXXXXXXXXXXXXXXX */
514                 os=M_ASN1_OCTET_STRING_new();
515                 p7->d.enveloped->enc_data->enc_data=os;
516                 break;
517         case NID_pkcs7_signed:
518                 si_sk=p7->d.sign->signer_info;
519                 os=p7->d.sign->contents->d.data;
520                 /* If detached data then the content is excluded */
521                 if(p7->detached) {
522                         M_ASN1_OCTET_STRING_free(os);
523                         p7->d.sign->contents->d.data = NULL;
524                 }
525                 break;
526                 }
527
528         if (si_sk != NULL)
529                 {
530                 if ((buf=BUF_MEM_new()) == NULL)
531                         {
532                         PKCS7err(PKCS7_F_PKCS7_DATASIGN,ERR_R_BIO_LIB);
533                         goto err;
534                         }
535                 for (i=0; i<sk_PKCS7_SIGNER_INFO_num(si_sk); i++)
536                         {
537                         si=sk_PKCS7_SIGNER_INFO_value(si_sk,i);
538                         if (si->pkey == NULL) continue;
539
540                         j=OBJ_obj2nid(si->digest_alg->algorithm);
541
542                         btmp=bio;
543                         for (;;)
544                                 {
545                                 if ((btmp=BIO_find_type(btmp,BIO_TYPE_MD)) 
546                                         == NULL)
547                                         {
548                                         PKCS7err(PKCS7_F_PKCS7_DATASIGN,PKCS7_R_UNABLE_TO_FIND_MESSAGE_DIGEST);
549                                         goto err;
550                                         }
551                                 BIO_get_md_ctx(btmp,&mdc);
552                                 if (mdc == NULL)
553                                         {
554                                         PKCS7err(PKCS7_F_PKCS7_DATASIGN,PKCS7_R_INTERNAL_ERROR);
555                                         goto err;
556                                         }
557                                 if (EVP_MD_CTX_type(mdc) == j)
558                                         break;
559                                 else
560                                         btmp=btmp->next_bio;
561                                 }
562                         
563                         /* We now have the EVP_MD_CTX, lets do the
564                          * signing. */
565                         memcpy(&ctx_tmp,mdc,sizeof(ctx_tmp));
566                         if (!BUF_MEM_grow(buf,EVP_PKEY_size(si->pkey)))
567                                 {
568                                 PKCS7err(PKCS7_F_PKCS7_DATASIGN,ERR_R_BIO_LIB);
569                                 goto err;
570                                 }
571
572                         sk=si->auth_attr;
573
574                         /* If there are attributes, we add the digest
575                          * attribute and only sign the attributes */
576                         if ((sk != NULL) && (sk_X509_ATTRIBUTE_num(sk) != 0))
577                                 {
578                                 unsigned char md_data[EVP_MAX_MD_SIZE];
579                                 unsigned int md_len;
580                                 ASN1_OCTET_STRING *digest;
581                                 ASN1_UTCTIME *sign_time;
582                                 const EVP_MD *md_tmp;
583
584                                 /* Add signing time */
585                                 sign_time=X509_gmtime_adj(NULL,0);
586                                 PKCS7_add_signed_attribute(si,
587                                         NID_pkcs9_signingTime,
588                                         V_ASN1_UTCTIME,sign_time);
589
590                                 /* Add digest */
591                                 md_tmp=EVP_MD_CTX_md(&ctx_tmp);
592                                 EVP_DigestFinal(&ctx_tmp,md_data,&md_len);
593                                 digest=M_ASN1_OCTET_STRING_new();
594                                 M_ASN1_OCTET_STRING_set(digest,md_data,md_len);
595                                 PKCS7_add_signed_attribute(si,
596                                         NID_pkcs9_messageDigest,
597                                         V_ASN1_OCTET_STRING,digest);
598
599                                 /* Now sign the mess */
600                                 EVP_SignInit(&ctx_tmp,md_tmp);
601                                 x=i2d_ASN1_SET_OF_X509_ATTRIBUTE(sk,NULL,
602                                            i2d_X509_ATTRIBUTE,
603                                            V_ASN1_SET,V_ASN1_UNIVERSAL,IS_SET);
604                                 pp=(unsigned char *)Malloc(x);
605                                 p=pp;
606                                 i2d_ASN1_SET_OF_X509_ATTRIBUTE(sk,&p,
607                                            i2d_X509_ATTRIBUTE,
608                                            V_ASN1_SET,V_ASN1_UNIVERSAL,IS_SET);
609                                 EVP_SignUpdate(&ctx_tmp,pp,x);
610                                 Free(pp);
611                                 pp=NULL;
612                                 }
613
614                         if (si->pkey->type == EVP_PKEY_DSA)
615                                 ctx_tmp.digest=EVP_dss1();
616
617                         if (!EVP_SignFinal(&ctx_tmp,(unsigned char *)buf->data,
618                                 (unsigned int *)&buf->length,si->pkey))
619                                 {
620                                 PKCS7err(PKCS7_F_PKCS7_DATASIGN,ERR_R_EVP_LIB);
621                                 goto err;
622                                 }
623                         if (!ASN1_STRING_set(si->enc_digest,
624                                 (unsigned char *)buf->data,buf->length))
625                                 {
626                                 PKCS7err(PKCS7_F_PKCS7_DATASIGN,ERR_R_ASN1_LIB);
627                                 goto err;
628                                 }
629                         }
630                 }
631
632         if (!p7->detached)
633                 {
634                 btmp=BIO_find_type(bio,BIO_TYPE_MEM);
635                 if (btmp == NULL)
636                         {
637                         PKCS7err(PKCS7_F_PKCS7_DATASIGN,PKCS7_R_UNABLE_TO_FIND_MEM_BIO);
638                         goto err;
639                         }
640                 BIO_get_mem_ptr(btmp,&buf_mem);
641                 /* Mark the BIO read only then we can use its copy of the data
642                  * instead of making an extra copy.
643                  */
644                 BIO_set_flags(btmp, BIO_FLAGS_MEM_RDONLY);
645                 BIO_set_mem_eof_return(btmp, 0);
646                 os->data = (unsigned char *)buf_mem->data;
647                 os->length = buf_mem->length;
648 #if 0
649                 M_ASN1_OCTET_STRING_set(os,
650                         (unsigned char *)buf_mem->data,buf_mem->length);
651 #endif
652                 }
653         if (pp != NULL) Free(pp);
654         pp=NULL;
655
656         ret=1;
657 err:
658         if (buf != NULL) BUF_MEM_free(buf);
659         return(ret);
660         }
661
662 int PKCS7_dataVerify(X509_STORE *cert_store, X509_STORE_CTX *ctx, BIO *bio,
663              PKCS7 *p7, PKCS7_SIGNER_INFO *si)
664         {
665         PKCS7_ISSUER_AND_SERIAL *ias;
666         int ret=0,i;
667         STACK_OF(X509) *cert;
668         X509 *x509;
669
670         if (PKCS7_type_is_signed(p7))
671                 {
672                 cert=p7->d.sign->cert;
673                 }
674         else if (PKCS7_type_is_signedAndEnveloped(p7))
675                 {
676                 cert=p7->d.signed_and_enveloped->cert;
677                 }
678         else
679                 {
680                 PKCS7err(PKCS7_F_PKCS7_DATAVERIFY,PKCS7_R_WRONG_PKCS7_TYPE);
681                 goto err;
682                 }
683         /* XXXXXXXXXXXXXXXXXXXXXXX */
684         ias=si->issuer_and_serial;
685
686         x509=X509_find_by_issuer_and_serial(cert,ias->issuer,ias->serial);
687
688         /* were we able to find the cert in passed to us */
689         if (x509 == NULL)
690                 {
691                 PKCS7err(PKCS7_F_PKCS7_DATAVERIFY,PKCS7_R_UNABLE_TO_FIND_CERTIFICATE);
692                 goto err;
693                 }
694
695         /* Lets verify */
696         X509_STORE_CTX_init(ctx,cert_store,x509,cert);
697         X509_STORE_CTX_set_purpose(ctx, X509_PURPOSE_SMIME_SIGN);
698         i=X509_verify_cert(ctx);
699         if (i <= 0) 
700                 {
701                 PKCS7err(PKCS7_F_PKCS7_DATAVERIFY,ERR_R_X509_LIB);
702                 X509_STORE_CTX_cleanup(ctx);
703                 goto err;
704                 }
705         X509_STORE_CTX_cleanup(ctx);
706
707         return PKCS7_signatureVerify(bio, p7, si, x509);
708         err:
709         return ret;
710         }
711
712 int PKCS7_signatureVerify(BIO *bio, PKCS7 *p7, PKCS7_SIGNER_INFO *si,
713                                                                 X509 *x509)
714         {
715         ASN1_OCTET_STRING *os;
716         EVP_MD_CTX mdc_tmp,*mdc;
717         unsigned char *pp,*p;
718         int ret=0,i;
719         int md_type;
720         STACK_OF(X509_ATTRIBUTE) *sk;
721         BIO *btmp;
722         EVP_PKEY *pkey;
723
724         if (!PKCS7_type_is_signed(p7) && 
725                                 !PKCS7_type_is_signedAndEnveloped(p7)) {
726                 PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY,
727                                                 PKCS7_R_WRONG_PKCS7_TYPE);
728                 goto err;
729         }
730
731         md_type=OBJ_obj2nid(si->digest_alg->algorithm);
732
733         btmp=bio;
734         for (;;)
735                 {
736                 if ((btmp == NULL) ||
737                         ((btmp=BIO_find_type(btmp,BIO_TYPE_MD)) == NULL))
738                         {
739                         PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY,
740                                         PKCS7_R_UNABLE_TO_FIND_MESSAGE_DIGEST);
741                         goto err;
742                         }
743                 BIO_get_md_ctx(btmp,&mdc);
744                 if (mdc == NULL)
745                         {
746                         PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY,
747                                                         PKCS7_R_INTERNAL_ERROR);
748                         goto err;
749                         }
750                 if (EVP_MD_CTX_type(mdc) == md_type)
751                         break;
752                 btmp=btmp->next_bio;    
753                 }
754
755         /* mdc is the digest ctx that we want, unless there are attributes,
756          * in which case the digest is the signed attributes */
757         memcpy(&mdc_tmp,mdc,sizeof(mdc_tmp));
758
759         sk=si->auth_attr;
760         if ((sk != NULL) && (sk_X509_ATTRIBUTE_num(sk) != 0))
761                 {
762                 unsigned char md_dat[EVP_MAX_MD_SIZE];
763                 unsigned int md_len;
764                 ASN1_OCTET_STRING *message_digest;
765
766                 EVP_DigestFinal(&mdc_tmp,md_dat,&md_len);
767                 message_digest=PKCS7_digest_from_attributes(sk);
768                 if (!message_digest)
769                         {
770                         PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY,
771                                         PKCS7_R_UNABLE_TO_FIND_MESSAGE_DIGEST);
772                         goto err;
773                         }
774                 if ((message_digest->length != (int)md_len) ||
775                         (memcmp(message_digest->data,md_dat,md_len)))
776                         {
777 #if 0
778 {
779 int ii;
780 for (ii=0; ii<message_digest->length; ii++)
781         printf("%02X",message_digest->data[ii]); printf(" sent\n");
782 for (ii=0; ii<md_len; ii++) printf("%02X",md_dat[ii]); printf(" calc\n");
783 }
784 #endif
785                         PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY,
786                                                         PKCS7_R_DIGEST_FAILURE);
787                         ret= -1;
788                         goto err;
789                         }
790
791                 EVP_VerifyInit(&mdc_tmp,EVP_get_digestbynid(md_type));
792                 /* Note: when forming the encoding of the attributes we
793                  * shouldn't reorder them or this will break the signature.
794                  * This is done by using the IS_SEQUENCE flag.
795                  */
796                 i=i2d_ASN1_SET_OF_X509_ATTRIBUTE(sk,NULL,i2d_X509_ATTRIBUTE,
797                         V_ASN1_SET,V_ASN1_UNIVERSAL, IS_SEQUENCE);
798                 pp=Malloc(i);
799                 p=pp;
800                 i2d_ASN1_SET_OF_X509_ATTRIBUTE(sk,&p,i2d_X509_ATTRIBUTE,
801                         V_ASN1_SET,V_ASN1_UNIVERSAL, IS_SEQUENCE);
802                 EVP_VerifyUpdate(&mdc_tmp,pp,i);
803
804                 Free(pp);
805                 }
806
807         os=si->enc_digest;
808         pkey = X509_get_pubkey(x509);
809         if(pkey->type == EVP_PKEY_DSA) mdc_tmp.digest=EVP_dss1();
810
811         i=EVP_VerifyFinal(&mdc_tmp,os->data,os->length, pkey);
812         EVP_PKEY_free(pkey);
813         if (i <= 0)
814                 {
815                 PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY,
816                                                 PKCS7_R_SIGNATURE_FAILURE);
817                 ret= -1;
818                 goto err;
819                 }
820         else
821                 ret=1;
822 err:
823         return(ret);
824         }
825
826 PKCS7_ISSUER_AND_SERIAL *PKCS7_get_issuer_and_serial(PKCS7 *p7, int idx)
827         {
828         STACK_OF(PKCS7_RECIP_INFO) *rsk;
829         PKCS7_RECIP_INFO *ri;
830         int i;
831
832         i=OBJ_obj2nid(p7->type);
833         if (i != NID_pkcs7_signedAndEnveloped) return(NULL);
834         rsk=p7->d.signed_and_enveloped->recipientinfo;
835         ri=sk_PKCS7_RECIP_INFO_value(rsk,0);
836         if (sk_PKCS7_RECIP_INFO_num(rsk) <= idx) return(NULL);
837         ri=sk_PKCS7_RECIP_INFO_value(rsk,idx);
838         return(ri->issuer_and_serial);
839         }
840
841 ASN1_TYPE *PKCS7_get_signed_attribute(PKCS7_SIGNER_INFO *si, int nid)
842         {
843         return(get_attribute(si->auth_attr,nid));
844         }
845
846 ASN1_TYPE *PKCS7_get_attribute(PKCS7_SIGNER_INFO *si, int nid)
847         {
848         return(get_attribute(si->unauth_attr,nid));
849         }
850
851 static ASN1_TYPE *get_attribute(STACK_OF(X509_ATTRIBUTE) *sk, int nid)
852         {
853         int i;
854         X509_ATTRIBUTE *xa;
855         ASN1_OBJECT *o;
856
857         o=OBJ_nid2obj(nid);
858         if (!o || !sk) return(NULL);
859         for (i=0; i<sk_X509_ATTRIBUTE_num(sk); i++)
860                 {
861                 xa=sk_X509_ATTRIBUTE_value(sk,i);
862                 if (OBJ_cmp(xa->object,o) == 0)
863                         {
864                         if (xa->set && sk_ASN1_TYPE_num(xa->value.set))
865                                 return(sk_ASN1_TYPE_value(xa->value.set,0));
866                         else
867                                 return(NULL);
868                         }
869                 }
870         return(NULL);
871         }
872
873 ASN1_OCTET_STRING *PKCS7_digest_from_attributes(STACK_OF(X509_ATTRIBUTE) *sk)
874 {
875         ASN1_TYPE *astype;
876         if(!(astype = get_attribute(sk, NID_pkcs9_messageDigest))) return NULL;
877         return astype->value.octet_string;
878 }
879
880 int PKCS7_set_signed_attributes(PKCS7_SIGNER_INFO *p7si,
881                                 STACK_OF(X509_ATTRIBUTE) *sk)
882         {
883         int i;
884
885         if (p7si->auth_attr != NULL)
886                 sk_X509_ATTRIBUTE_pop_free(p7si->auth_attr,X509_ATTRIBUTE_free);
887         p7si->auth_attr=sk_X509_ATTRIBUTE_dup(sk);
888         for (i=0; i<sk_X509_ATTRIBUTE_num(sk); i++)
889                 {
890                 if ((sk_X509_ATTRIBUTE_set(p7si->auth_attr,i,
891                         X509_ATTRIBUTE_dup(sk_X509_ATTRIBUTE_value(sk,i))))
892                     == NULL)
893                         return(0);
894                 }
895         return(1);
896         }
897
898 int PKCS7_set_attributes(PKCS7_SIGNER_INFO *p7si, STACK_OF(X509_ATTRIBUTE) *sk)
899         {
900         int i;
901
902         if (p7si->unauth_attr != NULL)
903                 sk_X509_ATTRIBUTE_pop_free(p7si->unauth_attr,
904                                            X509_ATTRIBUTE_free);
905         p7si->unauth_attr=sk_X509_ATTRIBUTE_dup(sk);
906         for (i=0; i<sk_X509_ATTRIBUTE_num(sk); i++)
907                 {
908                 if ((sk_X509_ATTRIBUTE_set(p7si->unauth_attr,i,
909                         X509_ATTRIBUTE_dup(sk_X509_ATTRIBUTE_value(sk,i))))
910                     == NULL)
911                         return(0);
912                 }
913         return(1);
914         }
915
916 int PKCS7_add_signed_attribute(PKCS7_SIGNER_INFO *p7si, int nid, int atrtype,
917              void *value)
918         {
919         return(add_attribute(&(p7si->auth_attr),nid,atrtype,value));
920         }
921
922 int PKCS7_add_attribute(PKCS7_SIGNER_INFO *p7si, int nid, int atrtype,
923              void *value)
924         {
925         return(add_attribute(&(p7si->unauth_attr),nid,atrtype,value));
926         }
927
928 static int add_attribute(STACK_OF(X509_ATTRIBUTE) **sk, int nid, int atrtype,
929                          void *value)
930         {
931         X509_ATTRIBUTE *attr=NULL;
932
933         if (*sk == NULL)
934                 {
935                 *sk = sk_X509_ATTRIBUTE_new(NULL);
936 new_attrib:
937                 attr=X509_ATTRIBUTE_create(nid,atrtype,value);
938                 sk_X509_ATTRIBUTE_push(*sk,attr);
939                 }
940         else
941                 {
942                 int i;
943
944                 for (i=0; i<sk_X509_ATTRIBUTE_num(*sk); i++)
945                         {
946                         attr=sk_X509_ATTRIBUTE_value(*sk,i);
947                         if (OBJ_obj2nid(attr->object) == nid)
948                                 {
949                                 X509_ATTRIBUTE_free(attr);
950                                 attr=X509_ATTRIBUTE_create(nid,atrtype,value);
951                                 sk_X509_ATTRIBUTE_set(*sk,i,attr);
952                                 goto end;
953                                 }
954                         }
955                 goto new_attrib;
956                 }
957 end:
958         return(1);
959         }
960