Err isn't always malloc failure.
[openssl.git] / crypto / cms / cms_smime.c
1 /* crypto/cms/cms_smime.c */
2 /*
3  * Written by Dr Stephen N Henson (steve@openssl.org) for the OpenSSL
4  * project.
5  */
6 /* ====================================================================
7  * Copyright (c) 2008 The OpenSSL Project.  All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  *
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  *
16  * 2. Redistributions in binary form must reproduce the above copyright
17  *    notice, this list of conditions and the following disclaimer in
18  *    the documentation and/or other materials provided with the
19  *    distribution.
20  *
21  * 3. All advertising materials mentioning features or use of this
22  *    software must display the following acknowledgment:
23  *    "This product includes software developed by the OpenSSL Project
24  *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
25  *
26  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
27  *    endorse or promote products derived from this software without
28  *    prior written permission. For written permission, please contact
29  *    licensing@OpenSSL.org.
30  *
31  * 5. Products derived from this software may not be called "OpenSSL"
32  *    nor may "OpenSSL" appear in their names without prior written
33  *    permission of the OpenSSL Project.
34  *
35  * 6. Redistributions of any form whatsoever must retain the following
36  *    acknowledgment:
37  *    "This product includes software developed by the OpenSSL Project
38  *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
39  *
40  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
41  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
42  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
43  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
44  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
45  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
46  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
47  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
49  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
50  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
51  * OF THE POSSIBILITY OF SUCH DAMAGE.
52  * ====================================================================
53  */
54
55 #include "internal/cryptlib.h"
56 #include <openssl/asn1t.h>
57 #include <openssl/x509.h>
58 #include <openssl/x509v3.h>
59 #include <openssl/err.h>
60 #include <openssl/cms.h>
61 #include "cms_lcl.h"
62 #include "internal/asn1_int.h"
63
64 static BIO *cms_get_text_bio(BIO *out, unsigned int flags)
65 {
66     BIO *rbio;
67     if (out == NULL)
68         rbio = BIO_new(BIO_s_null());
69     else if (flags & CMS_TEXT) {
70         rbio = BIO_new(BIO_s_mem());
71         BIO_set_mem_eof_return(rbio, 0);
72     } else
73         rbio = out;
74     return rbio;
75 }
76
77 static int cms_copy_content(BIO *out, BIO *in, unsigned int flags)
78 {
79     unsigned char buf[4096];
80     int r = 0, i;
81     BIO *tmpout;
82
83     tmpout = cms_get_text_bio(out, flags);
84
85     if (!tmpout) {
86         CMSerr(CMS_F_CMS_COPY_CONTENT, ERR_R_MALLOC_FAILURE);
87         goto err;
88     }
89
90     /* Read all content through chain to process digest, decrypt etc */
91     for (;;) {
92         i = BIO_read(in, buf, sizeof(buf));
93         if (i <= 0) {
94             if (BIO_method_type(in) == BIO_TYPE_CIPHER) {
95                 if (!BIO_get_cipher_status(in))
96                     goto err;
97             }
98             if (i < 0)
99                 goto err;
100             break;
101         }
102
103         if (tmpout && (BIO_write(tmpout, buf, i) != i))
104             goto err;
105     }
106
107     if (flags & CMS_TEXT) {
108         if (!SMIME_text(tmpout, out)) {
109             CMSerr(CMS_F_CMS_COPY_CONTENT, CMS_R_SMIME_TEXT_ERROR);
110             goto err;
111         }
112     }
113
114     r = 1;
115
116  err:
117     if (tmpout != out)
118         BIO_free(tmpout);
119     return r;
120
121 }
122
123 static int check_content(CMS_ContentInfo *cms)
124 {
125     ASN1_OCTET_STRING **pos = CMS_get0_content(cms);
126     if (!pos || !*pos) {
127         CMSerr(CMS_F_CHECK_CONTENT, CMS_R_NO_CONTENT);
128         return 0;
129     }
130     return 1;
131 }
132
133 static void do_free_upto(BIO *f, BIO *upto)
134 {
135     if (upto) {
136         BIO *tbio;
137         do {
138             tbio = BIO_pop(f);
139             BIO_free(f);
140             f = tbio;
141         }
142         while (f && f != upto);
143     } else
144         BIO_free_all(f);
145 }
146
147 int CMS_data(CMS_ContentInfo *cms, BIO *out, unsigned int flags)
148 {
149     BIO *cont;
150     int r;
151     if (OBJ_obj2nid(CMS_get0_type(cms)) != NID_pkcs7_data) {
152         CMSerr(CMS_F_CMS_DATA, CMS_R_TYPE_NOT_DATA);
153         return 0;
154     }
155     cont = CMS_dataInit(cms, NULL);
156     if (!cont)
157         return 0;
158     r = cms_copy_content(out, cont, flags);
159     BIO_free_all(cont);
160     return r;
161 }
162
163 CMS_ContentInfo *CMS_data_create(BIO *in, unsigned int flags)
164 {
165     CMS_ContentInfo *cms;
166     cms = cms_Data_create();
167     if (!cms)
168         return NULL;
169
170     if ((flags & CMS_STREAM) || CMS_final(cms, in, NULL, flags))
171         return cms;
172
173     CMS_ContentInfo_free(cms);
174
175     return NULL;
176 }
177
178 int CMS_digest_verify(CMS_ContentInfo *cms, BIO *dcont, BIO *out,
179                       unsigned int flags)
180 {
181     BIO *cont;
182     int r;
183     if (OBJ_obj2nid(CMS_get0_type(cms)) != NID_pkcs7_digest) {
184         CMSerr(CMS_F_CMS_DIGEST_VERIFY, CMS_R_TYPE_NOT_DIGESTED_DATA);
185         return 0;
186     }
187
188     if (!dcont && !check_content(cms))
189         return 0;
190
191     cont = CMS_dataInit(cms, dcont);
192     if (!cont)
193         return 0;
194     r = cms_copy_content(out, cont, flags);
195     if (r)
196         r = cms_DigestedData_do_final(cms, cont, 1);
197     do_free_upto(cont, dcont);
198     return r;
199 }
200
201 CMS_ContentInfo *CMS_digest_create(BIO *in, const EVP_MD *md,
202                                    unsigned int flags)
203 {
204     CMS_ContentInfo *cms;
205     if (!md)
206         md = EVP_sha1();
207     cms = cms_DigestedData_create(md);
208     if (!cms)
209         return NULL;
210
211     if (!(flags & CMS_DETACHED))
212         CMS_set_detached(cms, 0);
213
214     if ((flags & CMS_STREAM) || CMS_final(cms, in, NULL, flags))
215         return cms;
216
217     CMS_ContentInfo_free(cms);
218     return NULL;
219 }
220
221 int CMS_EncryptedData_decrypt(CMS_ContentInfo *cms,
222                               const unsigned char *key, size_t keylen,
223                               BIO *dcont, BIO *out, unsigned int flags)
224 {
225     BIO *cont;
226     int r;
227     if (OBJ_obj2nid(CMS_get0_type(cms)) != NID_pkcs7_encrypted) {
228         CMSerr(CMS_F_CMS_ENCRYPTEDDATA_DECRYPT,
229                CMS_R_TYPE_NOT_ENCRYPTED_DATA);
230         return 0;
231     }
232
233     if (!dcont && !check_content(cms))
234         return 0;
235
236     if (CMS_EncryptedData_set1_key(cms, NULL, key, keylen) <= 0)
237         return 0;
238     cont = CMS_dataInit(cms, dcont);
239     if (!cont)
240         return 0;
241     r = cms_copy_content(out, cont, flags);
242     do_free_upto(cont, dcont);
243     return r;
244 }
245
246 CMS_ContentInfo *CMS_EncryptedData_encrypt(BIO *in, const EVP_CIPHER *cipher,
247                                            const unsigned char *key,
248                                            size_t keylen, unsigned int flags)
249 {
250     CMS_ContentInfo *cms;
251     if (!cipher) {
252         CMSerr(CMS_F_CMS_ENCRYPTEDDATA_ENCRYPT, CMS_R_NO_CIPHER);
253         return NULL;
254     }
255     cms = CMS_ContentInfo_new();
256     if (!cms)
257         return NULL;
258     if (!CMS_EncryptedData_set1_key(cms, cipher, key, keylen))
259         return NULL;
260
261     if (!(flags & CMS_DETACHED))
262         CMS_set_detached(cms, 0);
263
264     if ((flags & (CMS_STREAM | CMS_PARTIAL))
265         || CMS_final(cms, in, NULL, flags))
266         return cms;
267
268     CMS_ContentInfo_free(cms);
269     return NULL;
270 }
271
272 static int cms_signerinfo_verify_cert(CMS_SignerInfo *si,
273                                       X509_STORE *store,
274                                       STACK_OF(X509) *certs,
275                                       STACK_OF(X509_CRL) *crls,
276                                       unsigned int flags)
277 {
278     X509_STORE_CTX ctx;
279     X509 *signer;
280     int i, j, r = 0;
281     CMS_SignerInfo_get0_algs(si, NULL, &signer, NULL, NULL);
282     if (!X509_STORE_CTX_init(&ctx, store, signer, certs)) {
283         CMSerr(CMS_F_CMS_SIGNERINFO_VERIFY_CERT, CMS_R_STORE_INIT_ERROR);
284         goto err;
285     }
286     X509_STORE_CTX_set_default(&ctx, "smime_sign");
287     if (crls)
288         X509_STORE_CTX_set0_crls(&ctx, crls);
289
290     i = X509_verify_cert(&ctx);
291     if (i <= 0) {
292         j = X509_STORE_CTX_get_error(&ctx);
293         CMSerr(CMS_F_CMS_SIGNERINFO_VERIFY_CERT,
294                CMS_R_CERTIFICATE_VERIFY_ERROR);
295         ERR_add_error_data(2, "Verify error:",
296                            X509_verify_cert_error_string(j));
297         goto err;
298     }
299     r = 1;
300  err:
301     X509_STORE_CTX_cleanup(&ctx);
302     return r;
303
304 }
305
306 int CMS_verify(CMS_ContentInfo *cms, STACK_OF(X509) *certs,
307                X509_STORE *store, BIO *dcont, BIO *out, unsigned int flags)
308 {
309     CMS_SignerInfo *si;
310     STACK_OF(CMS_SignerInfo) *sinfos;
311     STACK_OF(X509) *cms_certs = NULL;
312     STACK_OF(X509_CRL) *crls = NULL;
313     X509 *signer;
314     int i, scount = 0, ret = 0;
315     BIO *cmsbio = NULL, *tmpin = NULL, *tmpout = NULL;
316
317     if (!dcont && !check_content(cms))
318         return 0;
319     if (dcont && !(flags & CMS_BINARY)) {
320         const ASN1_OBJECT *coid = CMS_get0_eContentType(cms);
321         if (OBJ_obj2nid(coid) == NID_id_ct_asciiTextWithCRLF)
322             flags |= CMS_ASCIICRLF;
323     }
324
325     /* Attempt to find all signer certificates */
326
327     sinfos = CMS_get0_SignerInfos(cms);
328
329     if (sk_CMS_SignerInfo_num(sinfos) <= 0) {
330         CMSerr(CMS_F_CMS_VERIFY, CMS_R_NO_SIGNERS);
331         goto err;
332     }
333
334     for (i = 0; i < sk_CMS_SignerInfo_num(sinfos); i++) {
335         si = sk_CMS_SignerInfo_value(sinfos, i);
336         CMS_SignerInfo_get0_algs(si, NULL, &signer, NULL, NULL);
337         if (signer)
338             scount++;
339     }
340
341     if (scount != sk_CMS_SignerInfo_num(sinfos))
342         scount += CMS_set1_signers_certs(cms, certs, flags);
343
344     if (scount != sk_CMS_SignerInfo_num(sinfos)) {
345         CMSerr(CMS_F_CMS_VERIFY, CMS_R_SIGNER_CERTIFICATE_NOT_FOUND);
346         goto err;
347     }
348
349     /* Attempt to verify all signers certs */
350
351     if (!(flags & CMS_NO_SIGNER_CERT_VERIFY)) {
352         cms_certs = CMS_get1_certs(cms);
353         if (!(flags & CMS_NOCRL))
354             crls = CMS_get1_crls(cms);
355         for (i = 0; i < sk_CMS_SignerInfo_num(sinfos); i++) {
356             si = sk_CMS_SignerInfo_value(sinfos, i);
357             if (!cms_signerinfo_verify_cert(si, store,
358                                             cms_certs, crls, flags))
359                 goto err;
360         }
361     }
362
363     /* Attempt to verify all SignerInfo signed attribute signatures */
364
365     if (!(flags & CMS_NO_ATTR_VERIFY)) {
366         for (i = 0; i < sk_CMS_SignerInfo_num(sinfos); i++) {
367             si = sk_CMS_SignerInfo_value(sinfos, i);
368             if (CMS_signed_get_attr_count(si) < 0)
369                 continue;
370             if (CMS_SignerInfo_verify(si) <= 0)
371                 goto err;
372         }
373     }
374
375     /*
376      * Performance optimization: if the content is a memory BIO then store
377      * its contents in a temporary read only memory BIO. This avoids
378      * potentially large numbers of slow copies of data which will occur when
379      * reading from a read write memory BIO when signatures are calculated.
380      */
381
382     if (dcont && (BIO_method_type(dcont) == BIO_TYPE_MEM)) {
383         char *ptr;
384         long len;
385         len = BIO_get_mem_data(dcont, &ptr);
386         tmpin = BIO_new_mem_buf(ptr, len);
387         if (tmpin == NULL) {
388             CMSerr(CMS_F_CMS_VERIFY, ERR_R_MALLOC_FAILURE);
389             goto err2;
390         }
391     } else
392         tmpin = dcont;
393     /*
394      * If not binary mode and detached generate digests by *writing* through
395      * the BIO. That makes it possible to canonicalise the input.
396      */
397     if (!(flags & SMIME_BINARY) && dcont) {
398         /*
399          * Create output BIO so we can either handle text or to ensure
400          * included content doesn't override detached content.
401          */
402         tmpout = cms_get_text_bio(out, flags);
403         if (!tmpout) {
404             CMSerr(CMS_F_CMS_VERIFY, ERR_R_MALLOC_FAILURE);
405             goto err;
406         }
407         cmsbio = CMS_dataInit(cms, tmpout);
408         if (!cmsbio)
409             goto err;
410         /*
411          * Don't use SMIME_TEXT for verify: it adds headers and we want to
412          * remove them.
413          */
414         SMIME_crlf_copy(dcont, cmsbio, flags & ~SMIME_TEXT);
415
416         if (flags & CMS_TEXT) {
417             if (!SMIME_text(tmpout, out)) {
418                 CMSerr(CMS_F_CMS_VERIFY, CMS_R_SMIME_TEXT_ERROR);
419                 goto err;
420             }
421         }
422     } else {
423         cmsbio = CMS_dataInit(cms, tmpin);
424         if (!cmsbio)
425             goto err;
426
427         if (!cms_copy_content(out, cmsbio, flags))
428             goto err;
429
430     }
431     if (!(flags & CMS_NO_CONTENT_VERIFY)) {
432         for (i = 0; i < sk_CMS_SignerInfo_num(sinfos); i++) {
433             si = sk_CMS_SignerInfo_value(sinfos, i);
434             if (CMS_SignerInfo_verify_content(si, cmsbio) <= 0) {
435                 CMSerr(CMS_F_CMS_VERIFY, CMS_R_CONTENT_VERIFY_ERROR);
436                 goto err;
437             }
438         }
439     }
440
441     ret = 1;
442
443  err:
444     if (!(flags & SMIME_BINARY) && dcont) {
445         do_free_upto(cmsbio, tmpout);
446         if (tmpin != dcont)
447             BIO_free(tmpin);
448     } else {
449         if (dcont && (tmpin == dcont))
450             do_free_upto(cmsbio, dcont);
451         else
452             BIO_free_all(cmsbio);
453     }
454
455     if (out != tmpout)
456         BIO_free_all(tmpout);
457
458  err2:
459     sk_X509_pop_free(cms_certs, X509_free);
460     sk_X509_CRL_pop_free(crls, X509_CRL_free);
461
462     return ret;
463 }
464
465 int CMS_verify_receipt(CMS_ContentInfo *rcms, CMS_ContentInfo *ocms,
466                        STACK_OF(X509) *certs,
467                        X509_STORE *store, unsigned int flags)
468 {
469     int r;
470     flags &= ~(CMS_DETACHED | CMS_TEXT);
471     r = CMS_verify(rcms, certs, store, NULL, NULL, flags);
472     if (r <= 0)
473         return r;
474     return cms_Receipt_verify(rcms, ocms);
475 }
476
477 CMS_ContentInfo *CMS_sign(X509 *signcert, EVP_PKEY *pkey,
478                           STACK_OF(X509) *certs, BIO *data,
479                           unsigned int flags)
480 {
481     CMS_ContentInfo *cms;
482     int i;
483
484     cms = CMS_ContentInfo_new();
485     if (!cms || !CMS_SignedData_init(cms))
486         goto merr;
487     if (flags & CMS_ASCIICRLF
488         && !CMS_set1_eContentType(cms,
489                                   OBJ_nid2obj(NID_id_ct_asciiTextWithCRLF)))
490         goto err;
491
492     if (pkey && !CMS_add1_signer(cms, signcert, pkey, NULL, flags)) {
493         CMSerr(CMS_F_CMS_SIGN, CMS_R_ADD_SIGNER_ERROR);
494         goto err;
495     }
496
497     for (i = 0; i < sk_X509_num(certs); i++) {
498         X509 *x = sk_X509_value(certs, i);
499         if (!CMS_add1_cert(cms, x))
500             goto merr;
501     }
502
503     if (!(flags & CMS_DETACHED))
504         CMS_set_detached(cms, 0);
505
506     if ((flags & (CMS_STREAM | CMS_PARTIAL))
507         || CMS_final(cms, data, NULL, flags))
508         return cms;
509     else
510         goto err;
511
512  merr:
513     CMSerr(CMS_F_CMS_SIGN, ERR_R_MALLOC_FAILURE);
514
515  err:
516     CMS_ContentInfo_free(cms);
517     return NULL;
518 }
519
520 CMS_ContentInfo *CMS_sign_receipt(CMS_SignerInfo *si,
521                                   X509 *signcert, EVP_PKEY *pkey,
522                                   STACK_OF(X509) *certs, unsigned int flags)
523 {
524     CMS_SignerInfo *rct_si;
525     CMS_ContentInfo *cms = NULL;
526     ASN1_OCTET_STRING **pos, *os;
527     BIO *rct_cont = NULL;
528     int r = 0;
529
530     flags &= ~(CMS_STREAM | CMS_TEXT);
531     /* Not really detached but avoids content being allocated */
532     flags |= CMS_PARTIAL | CMS_BINARY | CMS_DETACHED;
533     if (!pkey || !signcert) {
534         CMSerr(CMS_F_CMS_SIGN_RECEIPT, CMS_R_NO_KEY_OR_CERT);
535         return NULL;
536     }
537
538     /* Initialize signed data */
539
540     cms = CMS_sign(NULL, NULL, certs, NULL, flags);
541     if (!cms)
542         goto err;
543
544     /* Set inner content type to signed receipt */
545     if (!CMS_set1_eContentType(cms, OBJ_nid2obj(NID_id_smime_ct_receipt)))
546         goto err;
547
548     rct_si = CMS_add1_signer(cms, signcert, pkey, NULL, flags);
549     if (!rct_si) {
550         CMSerr(CMS_F_CMS_SIGN_RECEIPT, CMS_R_ADD_SIGNER_ERROR);
551         goto err;
552     }
553
554     os = cms_encode_Receipt(si);
555
556     if (!os)
557         goto err;
558
559     /* Set content to digest */
560     rct_cont = BIO_new_mem_buf(os->data, os->length);
561     if (!rct_cont)
562         goto err;
563
564     /* Add msgSigDigest attribute */
565
566     if (!cms_msgSigDigest_add1(rct_si, si))
567         goto err;
568
569     /* Finalize structure */
570     if (!CMS_final(cms, rct_cont, NULL, flags))
571         goto err;
572
573     /* Set embedded content */
574     pos = CMS_get0_content(cms);
575     *pos = os;
576
577     r = 1;
578
579  err:
580     BIO_free(rct_cont);
581     if (r)
582         return cms;
583     CMS_ContentInfo_free(cms);
584     return NULL;
585
586 }
587
588 CMS_ContentInfo *CMS_encrypt(STACK_OF(X509) *certs, BIO *data,
589                              const EVP_CIPHER *cipher, unsigned int flags)
590 {
591     CMS_ContentInfo *cms;
592     int i;
593     X509 *recip;
594     cms = CMS_EnvelopedData_create(cipher);
595     if (!cms)
596         goto merr;
597     for (i = 0; i < sk_X509_num(certs); i++) {
598         recip = sk_X509_value(certs, i);
599         if (!CMS_add1_recipient_cert(cms, recip, flags)) {
600             CMSerr(CMS_F_CMS_ENCRYPT, CMS_R_RECIPIENT_ERROR);
601             goto err;
602         }
603     }
604
605     if (!(flags & CMS_DETACHED))
606         CMS_set_detached(cms, 0);
607
608     if ((flags & (CMS_STREAM | CMS_PARTIAL))
609         || CMS_final(cms, data, NULL, flags))
610         return cms;
611     else
612         goto err;
613
614  merr:
615     CMSerr(CMS_F_CMS_ENCRYPT, ERR_R_MALLOC_FAILURE);
616  err:
617     CMS_ContentInfo_free(cms);
618     return NULL;
619 }
620
621 static int cms_kari_set1_pkey(CMS_ContentInfo *cms, CMS_RecipientInfo *ri,
622                               EVP_PKEY *pk, X509 *cert)
623 {
624     int i;
625     STACK_OF(CMS_RecipientEncryptedKey) *reks;
626     CMS_RecipientEncryptedKey *rek;
627     reks = CMS_RecipientInfo_kari_get0_reks(ri);
628     if (!cert)
629         return 0;
630     for (i = 0; i < sk_CMS_RecipientEncryptedKey_num(reks); i++) {
631         int rv;
632         rek = sk_CMS_RecipientEncryptedKey_value(reks, i);
633         if (CMS_RecipientEncryptedKey_cert_cmp(rek, cert))
634             continue;
635         CMS_RecipientInfo_kari_set0_pkey(ri, pk);
636         rv = CMS_RecipientInfo_kari_decrypt(cms, ri, rek);
637         CMS_RecipientInfo_kari_set0_pkey(ri, NULL);
638         if (rv > 0)
639             return 1;
640         return -1;
641     }
642     return 0;
643 }
644
645 int CMS_decrypt_set1_pkey(CMS_ContentInfo *cms, EVP_PKEY *pk, X509 *cert)
646 {
647     STACK_OF(CMS_RecipientInfo) *ris;
648     CMS_RecipientInfo *ri;
649     int i, r, ri_type;
650     int debug = 0, match_ri = 0;
651     ris = CMS_get0_RecipientInfos(cms);
652     if (ris)
653         debug = cms->d.envelopedData->encryptedContentInfo->debug;
654     ri_type = cms_pkey_get_ri_type(pk);
655     if (ri_type == CMS_RECIPINFO_NONE) {
656         CMSerr(CMS_F_CMS_DECRYPT_SET1_PKEY,
657                CMS_R_NOT_SUPPORTED_FOR_THIS_KEY_TYPE);
658         return 0;
659     }
660
661     for (i = 0; i < sk_CMS_RecipientInfo_num(ris); i++) {
662         ri = sk_CMS_RecipientInfo_value(ris, i);
663         if (CMS_RecipientInfo_type(ri) != ri_type)
664             continue;
665         match_ri = 1;
666         if (ri_type == CMS_RECIPINFO_AGREE) {
667             r = cms_kari_set1_pkey(cms, ri, pk, cert);
668             if (r > 0)
669                 return 1;
670             if (r < 0)
671                 return 0;
672         }
673         /*
674          * If we have a cert try matching RecipientInfo otherwise try them
675          * all.
676          */
677         else if (!cert || !CMS_RecipientInfo_ktri_cert_cmp(ri, cert)) {
678             CMS_RecipientInfo_set0_pkey(ri, pk);
679             r = CMS_RecipientInfo_decrypt(cms, ri);
680             CMS_RecipientInfo_set0_pkey(ri, NULL);
681             if (cert) {
682                 /*
683                  * If not debugging clear any error and return success to
684                  * avoid leaking of information useful to MMA
685                  */
686                 if (!debug) {
687                     ERR_clear_error();
688                     return 1;
689                 }
690                 if (r > 0)
691                     return 1;
692                 CMSerr(CMS_F_CMS_DECRYPT_SET1_PKEY, CMS_R_DECRYPT_ERROR);
693                 return 0;
694             }
695             /*
696              * If no cert and not debugging don't leave loop after first
697              * successful decrypt. Always attempt to decrypt all recipients
698              * to avoid leaking timing of a successful decrypt.
699              */
700             else if (r > 0 && debug)
701                 return 1;
702         }
703     }
704     /* If no cert and not debugging always return success */
705     if (match_ri && !cert && !debug) {
706         ERR_clear_error();
707         return 1;
708     }
709
710     CMSerr(CMS_F_CMS_DECRYPT_SET1_PKEY, CMS_R_NO_MATCHING_RECIPIENT);
711     return 0;
712
713 }
714
715 int CMS_decrypt_set1_key(CMS_ContentInfo *cms,
716                          unsigned char *key, size_t keylen,
717                          unsigned char *id, size_t idlen)
718 {
719     STACK_OF(CMS_RecipientInfo) *ris;
720     CMS_RecipientInfo *ri;
721     int i, r;
722     ris = CMS_get0_RecipientInfos(cms);
723     for (i = 0; i < sk_CMS_RecipientInfo_num(ris); i++) {
724         ri = sk_CMS_RecipientInfo_value(ris, i);
725         if (CMS_RecipientInfo_type(ri) != CMS_RECIPINFO_KEK)
726             continue;
727
728         /*
729          * If we have an id try matching RecipientInfo otherwise try them
730          * all.
731          */
732         if (!id || (CMS_RecipientInfo_kekri_id_cmp(ri, id, idlen) == 0)) {
733             CMS_RecipientInfo_set0_key(ri, key, keylen);
734             r = CMS_RecipientInfo_decrypt(cms, ri);
735             CMS_RecipientInfo_set0_key(ri, NULL, 0);
736             if (r > 0)
737                 return 1;
738             if (id) {
739                 CMSerr(CMS_F_CMS_DECRYPT_SET1_KEY, CMS_R_DECRYPT_ERROR);
740                 return 0;
741             }
742             ERR_clear_error();
743         }
744     }
745
746     CMSerr(CMS_F_CMS_DECRYPT_SET1_KEY, CMS_R_NO_MATCHING_RECIPIENT);
747     return 0;
748
749 }
750
751 int CMS_decrypt_set1_password(CMS_ContentInfo *cms,
752                               unsigned char *pass, ossl_ssize_t passlen)
753 {
754     STACK_OF(CMS_RecipientInfo) *ris;
755     CMS_RecipientInfo *ri;
756     int i, r;
757     ris = CMS_get0_RecipientInfos(cms);
758     for (i = 0; i < sk_CMS_RecipientInfo_num(ris); i++) {
759         ri = sk_CMS_RecipientInfo_value(ris, i);
760         if (CMS_RecipientInfo_type(ri) != CMS_RECIPINFO_PASS)
761             continue;
762         CMS_RecipientInfo_set0_password(ri, pass, passlen);
763         r = CMS_RecipientInfo_decrypt(cms, ri);
764         CMS_RecipientInfo_set0_password(ri, NULL, 0);
765         if (r > 0)
766             return 1;
767     }
768
769     CMSerr(CMS_F_CMS_DECRYPT_SET1_PASSWORD, CMS_R_NO_MATCHING_RECIPIENT);
770     return 0;
771
772 }
773
774 int CMS_decrypt(CMS_ContentInfo *cms, EVP_PKEY *pk, X509 *cert,
775                 BIO *dcont, BIO *out, unsigned int flags)
776 {
777     int r;
778     BIO *cont;
779     if (OBJ_obj2nid(CMS_get0_type(cms)) != NID_pkcs7_enveloped) {
780         CMSerr(CMS_F_CMS_DECRYPT, CMS_R_TYPE_NOT_ENVELOPED_DATA);
781         return 0;
782     }
783     if (!dcont && !check_content(cms))
784         return 0;
785     if (flags & CMS_DEBUG_DECRYPT)
786         cms->d.envelopedData->encryptedContentInfo->debug = 1;
787     else
788         cms->d.envelopedData->encryptedContentInfo->debug = 0;
789     if (!pk && !cert && !dcont && !out)
790         return 1;
791     if (pk && !CMS_decrypt_set1_pkey(cms, pk, cert))
792         return 0;
793     cont = CMS_dataInit(cms, dcont);
794     if (!cont)
795         return 0;
796     r = cms_copy_content(out, cont, flags);
797     do_free_upto(cont, dcont);
798     return r;
799 }
800
801 int CMS_final(CMS_ContentInfo *cms, BIO *data, BIO *dcont, unsigned int flags)
802 {
803     BIO *cmsbio;
804     int ret = 0;
805
806     if ((cmsbio = CMS_dataInit(cms, dcont)) == NULL) {
807         CMSerr(CMS_F_CMS_FINAL, CMS_R_CMS_LIB);
808         return 0;
809     }
810
811     SMIME_crlf_copy(data, cmsbio, flags);
812
813     (void)BIO_flush(cmsbio);
814
815     if (!CMS_dataFinal(cms, cmsbio)) {
816         CMSerr(CMS_F_CMS_FINAL, CMS_R_CMS_DATAFINAL_ERROR);
817         goto err;
818     }
819
820     ret = 1;
821
822  err:
823     do_free_upto(cmsbio, dcont);
824
825     return ret;
826
827 }
828
829 #ifdef ZLIB
830
831 int CMS_uncompress(CMS_ContentInfo *cms, BIO *dcont, BIO *out,
832                    unsigned int flags)
833 {
834     BIO *cont;
835     int r;
836     if (OBJ_obj2nid(CMS_get0_type(cms)) != NID_id_smime_ct_compressedData) {
837         CMSerr(CMS_F_CMS_UNCOMPRESS, CMS_R_TYPE_NOT_COMPRESSED_DATA);
838         return 0;
839     }
840
841     if (!dcont && !check_content(cms))
842         return 0;
843
844     cont = CMS_dataInit(cms, dcont);
845     if (!cont)
846         return 0;
847     r = cms_copy_content(out, cont, flags);
848     do_free_upto(cont, dcont);
849     return r;
850 }
851
852 CMS_ContentInfo *CMS_compress(BIO *in, int comp_nid, unsigned int flags)
853 {
854     CMS_ContentInfo *cms;
855     if (comp_nid <= 0)
856         comp_nid = NID_zlib_compression;
857     cms = cms_CompressedData_create(comp_nid);
858     if (!cms)
859         return NULL;
860
861     if (!(flags & CMS_DETACHED))
862         CMS_set_detached(cms, 0);
863
864     if ((flags & CMS_STREAM) || CMS_final(cms, in, NULL, flags))
865         return cms;
866
867     CMS_ContentInfo_free(cms);
868     return NULL;
869 }
870
871 #else
872
873 int CMS_uncompress(CMS_ContentInfo *cms, BIO *dcont, BIO *out,
874                    unsigned int flags)
875 {
876     CMSerr(CMS_F_CMS_UNCOMPRESS, CMS_R_UNSUPPORTED_COMPRESSION_ALGORITHM);
877     return 0;
878 }
879
880 CMS_ContentInfo *CMS_compress(BIO *in, int comp_nid, unsigned int flags)
881 {
882     CMSerr(CMS_F_CMS_COMPRESS, CMS_R_UNSUPPORTED_COMPRESSION_ALGORITHM);
883     return NULL;
884 }
885
886 #endif