free NULL cleanup
[openssl.git] / crypto / cms / cms_sd.c
1 /* crypto/cms/cms_sd.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 "cryptlib.h"
56 #include <openssl/asn1t.h>
57 #include <openssl/pem.h>
58 #include <openssl/x509.h>
59 #include <openssl/x509v3.h>
60 #include <openssl/err.h>
61 #include <openssl/cms.h>
62 #include "cms_lcl.h"
63 #include "internal/asn1_int.h"
64
65 /* CMS SignedData Utilities */
66
67 DECLARE_ASN1_ITEM(CMS_SignedData)
68
69 static CMS_SignedData *cms_get0_signed(CMS_ContentInfo *cms)
70 {
71     if (OBJ_obj2nid(cms->contentType) != NID_pkcs7_signed) {
72         CMSerr(CMS_F_CMS_GET0_SIGNED, CMS_R_CONTENT_TYPE_NOT_SIGNED_DATA);
73         return NULL;
74     }
75     return cms->d.signedData;
76 }
77
78 static CMS_SignedData *cms_signed_data_init(CMS_ContentInfo *cms)
79 {
80     if (cms->d.other == NULL) {
81         cms->d.signedData = M_ASN1_new_of(CMS_SignedData);
82         if (!cms->d.signedData) {
83             CMSerr(CMS_F_CMS_SIGNED_DATA_INIT, ERR_R_MALLOC_FAILURE);
84             return NULL;
85         }
86         cms->d.signedData->version = 1;
87         cms->d.signedData->encapContentInfo->eContentType =
88             OBJ_nid2obj(NID_pkcs7_data);
89         cms->d.signedData->encapContentInfo->partial = 1;
90         ASN1_OBJECT_free(cms->contentType);
91         cms->contentType = OBJ_nid2obj(NID_pkcs7_signed);
92         return cms->d.signedData;
93     }
94     return cms_get0_signed(cms);
95 }
96
97 /* Just initialize SignedData e.g. for certs only structure */
98
99 int CMS_SignedData_init(CMS_ContentInfo *cms)
100 {
101     if (cms_signed_data_init(cms))
102         return 1;
103     else
104         return 0;
105 }
106
107 /* Check structures and fixup version numbers (if necessary) */
108
109 static void cms_sd_set_version(CMS_SignedData *sd)
110 {
111     int i;
112     CMS_CertificateChoices *cch;
113     CMS_RevocationInfoChoice *rch;
114     CMS_SignerInfo *si;
115
116     for (i = 0; i < sk_CMS_CertificateChoices_num(sd->certificates); i++) {
117         cch = sk_CMS_CertificateChoices_value(sd->certificates, i);
118         if (cch->type == CMS_CERTCHOICE_OTHER) {
119             if (sd->version < 5)
120                 sd->version = 5;
121         } else if (cch->type == CMS_CERTCHOICE_V2ACERT) {
122             if (sd->version < 4)
123                 sd->version = 4;
124         } else if (cch->type == CMS_CERTCHOICE_V1ACERT) {
125             if (sd->version < 3)
126                 sd->version = 3;
127         }
128     }
129
130     for (i = 0; i < sk_CMS_RevocationInfoChoice_num(sd->crls); i++) {
131         rch = sk_CMS_RevocationInfoChoice_value(sd->crls, i);
132         if (rch->type == CMS_REVCHOICE_OTHER) {
133             if (sd->version < 5)
134                 sd->version = 5;
135         }
136     }
137
138     if ((OBJ_obj2nid(sd->encapContentInfo->eContentType) != NID_pkcs7_data)
139         && (sd->version < 3))
140         sd->version = 3;
141
142     for (i = 0; i < sk_CMS_SignerInfo_num(sd->signerInfos); i++) {
143         si = sk_CMS_SignerInfo_value(sd->signerInfos, i);
144         if (si->sid->type == CMS_SIGNERINFO_KEYIDENTIFIER) {
145             if (si->version < 3)
146                 si->version = 3;
147             if (sd->version < 3)
148                 sd->version = 3;
149         } else if (si->version < 1)
150             si->version = 1;
151     }
152
153     if (sd->version < 1)
154         sd->version = 1;
155
156 }
157
158 /* Copy an existing messageDigest value */
159
160 static int cms_copy_messageDigest(CMS_ContentInfo *cms, CMS_SignerInfo *si)
161 {
162     STACK_OF(CMS_SignerInfo) *sinfos;
163     CMS_SignerInfo *sitmp;
164     int i;
165     sinfos = CMS_get0_SignerInfos(cms);
166     for (i = 0; i < sk_CMS_SignerInfo_num(sinfos); i++) {
167         ASN1_OCTET_STRING *messageDigest;
168         sitmp = sk_CMS_SignerInfo_value(sinfos, i);
169         if (sitmp == si)
170             continue;
171         if (CMS_signed_get_attr_count(sitmp) < 0)
172             continue;
173         if (OBJ_cmp(si->digestAlgorithm->algorithm,
174                     sitmp->digestAlgorithm->algorithm))
175             continue;
176         messageDigest = CMS_signed_get0_data_by_OBJ(sitmp,
177                                                     OBJ_nid2obj
178                                                     (NID_pkcs9_messageDigest),
179                                                     -3, V_ASN1_OCTET_STRING);
180         if (!messageDigest) {
181             CMSerr(CMS_F_CMS_COPY_MESSAGEDIGEST,
182                    CMS_R_ERROR_READING_MESSAGEDIGEST_ATTRIBUTE);
183             return 0;
184         }
185
186         if (CMS_signed_add1_attr_by_NID(si, NID_pkcs9_messageDigest,
187                                         V_ASN1_OCTET_STRING,
188                                         messageDigest, -1))
189             return 1;
190         else
191             return 0;
192     }
193     CMSerr(CMS_F_CMS_COPY_MESSAGEDIGEST, CMS_R_NO_MATCHING_DIGEST);
194     return 0;
195 }
196
197 int cms_set1_SignerIdentifier(CMS_SignerIdentifier *sid, X509 *cert, int type)
198 {
199     switch (type) {
200     case CMS_SIGNERINFO_ISSUER_SERIAL:
201         if (!cms_set1_ias(&sid->d.issuerAndSerialNumber, cert))
202             return 0;
203         break;
204
205     case CMS_SIGNERINFO_KEYIDENTIFIER:
206         if (!cms_set1_keyid(&sid->d.subjectKeyIdentifier, cert))
207             return 0;
208         break;
209
210     default:
211         CMSerr(CMS_F_CMS_SET1_SIGNERIDENTIFIER, CMS_R_UNKNOWN_ID);
212         return 0;
213     }
214
215     sid->type = type;
216
217     return 1;
218 }
219
220 int cms_SignerIdentifier_get0_signer_id(CMS_SignerIdentifier *sid,
221                                         ASN1_OCTET_STRING **keyid,
222                                         X509_NAME **issuer,
223                                         ASN1_INTEGER **sno)
224 {
225     if (sid->type == CMS_SIGNERINFO_ISSUER_SERIAL) {
226         if (issuer)
227             *issuer = sid->d.issuerAndSerialNumber->issuer;
228         if (sno)
229             *sno = sid->d.issuerAndSerialNumber->serialNumber;
230     } else if (sid->type == CMS_SIGNERINFO_KEYIDENTIFIER) {
231         if (keyid)
232             *keyid = sid->d.subjectKeyIdentifier;
233     } else
234         return 0;
235     return 1;
236 }
237
238 int cms_SignerIdentifier_cert_cmp(CMS_SignerIdentifier *sid, X509 *cert)
239 {
240     if (sid->type == CMS_SIGNERINFO_ISSUER_SERIAL)
241         return cms_ias_cert_cmp(sid->d.issuerAndSerialNumber, cert);
242     else if (sid->type == CMS_SIGNERINFO_KEYIDENTIFIER)
243         return cms_keyid_cert_cmp(sid->d.subjectKeyIdentifier, cert);
244     else
245         return -1;
246 }
247
248 static int cms_sd_asn1_ctrl(CMS_SignerInfo *si, int cmd)
249 {
250     EVP_PKEY *pkey = si->pkey;
251     int i;
252     if (!pkey->ameth || !pkey->ameth->pkey_ctrl)
253         return 1;
254     i = pkey->ameth->pkey_ctrl(pkey, ASN1_PKEY_CTRL_CMS_SIGN, cmd, si);
255     if (i == -2) {
256         CMSerr(CMS_F_CMS_SD_ASN1_CTRL, CMS_R_NOT_SUPPORTED_FOR_THIS_KEY_TYPE);
257         return 0;
258     }
259     if (i <= 0) {
260         CMSerr(CMS_F_CMS_SD_ASN1_CTRL, CMS_R_CTRL_FAILURE);
261         return 0;
262     }
263     return 1;
264 }
265
266 CMS_SignerInfo *CMS_add1_signer(CMS_ContentInfo *cms,
267                                 X509 *signer, EVP_PKEY *pk, const EVP_MD *md,
268                                 unsigned int flags)
269 {
270     CMS_SignedData *sd;
271     CMS_SignerInfo *si = NULL;
272     X509_ALGOR *alg;
273     int i, type;
274     if (!X509_check_private_key(signer, pk)) {
275         CMSerr(CMS_F_CMS_ADD1_SIGNER,
276                CMS_R_PRIVATE_KEY_DOES_NOT_MATCH_CERTIFICATE);
277         return NULL;
278     }
279     sd = cms_signed_data_init(cms);
280     if (!sd)
281         goto err;
282     si = M_ASN1_new_of(CMS_SignerInfo);
283     if (!si)
284         goto merr;
285     X509_check_purpose(signer, -1, -1);
286
287     CRYPTO_add(&pk->references, 1, CRYPTO_LOCK_EVP_PKEY);
288     CRYPTO_add(&signer->references, 1, CRYPTO_LOCK_X509);
289
290     si->pkey = pk;
291     si->signer = signer;
292     EVP_MD_CTX_init(&si->mctx);
293     si->pctx = NULL;
294
295     if (flags & CMS_USE_KEYID) {
296         si->version = 3;
297         if (sd->version < 3)
298             sd->version = 3;
299         type = CMS_SIGNERINFO_KEYIDENTIFIER;
300     } else {
301         type = CMS_SIGNERINFO_ISSUER_SERIAL;
302         si->version = 1;
303     }
304
305     if (!cms_set1_SignerIdentifier(si->sid, signer, type))
306         goto err;
307
308     if (md == NULL) {
309         int def_nid;
310         if (EVP_PKEY_get_default_digest_nid(pk, &def_nid) <= 0)
311             goto err;
312         md = EVP_get_digestbynid(def_nid);
313         if (md == NULL) {
314             CMSerr(CMS_F_CMS_ADD1_SIGNER, CMS_R_NO_DEFAULT_DIGEST);
315             goto err;
316         }
317     }
318
319     if (!md) {
320         CMSerr(CMS_F_CMS_ADD1_SIGNER, CMS_R_NO_DIGEST_SET);
321         goto err;
322     }
323
324     X509_ALGOR_set_md(si->digestAlgorithm, md);
325
326     /* See if digest is present in digestAlgorithms */
327     for (i = 0; i < sk_X509_ALGOR_num(sd->digestAlgorithms); i++) {
328         ASN1_OBJECT *aoid;
329         alg = sk_X509_ALGOR_value(sd->digestAlgorithms, i);
330         X509_ALGOR_get0(&aoid, NULL, NULL, alg);
331         if (OBJ_obj2nid(aoid) == EVP_MD_type(md))
332             break;
333     }
334
335     if (i == sk_X509_ALGOR_num(sd->digestAlgorithms)) {
336         alg = X509_ALGOR_new();
337         if (!alg)
338             goto merr;
339         X509_ALGOR_set_md(alg, md);
340         if (!sk_X509_ALGOR_push(sd->digestAlgorithms, alg)) {
341             X509_ALGOR_free(alg);
342             goto merr;
343         }
344     }
345
346     if (!(flags & CMS_KEY_PARAM) && !cms_sd_asn1_ctrl(si, 0))
347         goto err;
348     if (!(flags & CMS_NOATTR)) {
349         /*
350          * Initialialize signed attributes strutucture so other attributes
351          * such as signing time etc are added later even if we add none here.
352          */
353         if (!si->signedAttrs) {
354             si->signedAttrs = sk_X509_ATTRIBUTE_new_null();
355             if (!si->signedAttrs)
356                 goto merr;
357         }
358
359         if (!(flags & CMS_NOSMIMECAP)) {
360             STACK_OF(X509_ALGOR) *smcap = NULL;
361             i = CMS_add_standard_smimecap(&smcap);
362             if (i)
363                 i = CMS_add_smimecap(si, smcap);
364             sk_X509_ALGOR_pop_free(smcap, X509_ALGOR_free);
365             if (!i)
366                 goto merr;
367         }
368         if (flags & CMS_REUSE_DIGEST) {
369             if (!cms_copy_messageDigest(cms, si))
370                 goto err;
371             if (!(flags & (CMS_PARTIAL | CMS_KEY_PARAM)) &&
372                 !CMS_SignerInfo_sign(si))
373                 goto err;
374         }
375     }
376
377     if (!(flags & CMS_NOCERTS)) {
378         /* NB ignore -1 return for duplicate cert */
379         if (!CMS_add1_cert(cms, signer))
380             goto merr;
381     }
382
383     if (flags & CMS_KEY_PARAM) {
384         if (flags & CMS_NOATTR) {
385             si->pctx = EVP_PKEY_CTX_new(si->pkey, NULL);
386             if (!si->pctx)
387                 goto err;
388             if (EVP_PKEY_sign_init(si->pctx) <= 0)
389                 goto err;
390             if (EVP_PKEY_CTX_set_signature_md(si->pctx, md) <= 0)
391                 goto err;
392         } else if (EVP_DigestSignInit(&si->mctx, &si->pctx, md, NULL, pk) <=
393                    0)
394             goto err;
395     }
396
397     if (!sd->signerInfos)
398         sd->signerInfos = sk_CMS_SignerInfo_new_null();
399     if (!sd->signerInfos || !sk_CMS_SignerInfo_push(sd->signerInfos, si))
400         goto merr;
401
402     return si;
403
404  merr:
405     CMSerr(CMS_F_CMS_ADD1_SIGNER, ERR_R_MALLOC_FAILURE);
406  err:
407     if (si)
408         M_ASN1_free_of(si, CMS_SignerInfo);
409     return NULL;
410
411 }
412
413 static int cms_add1_signingTime(CMS_SignerInfo *si, ASN1_TIME *t)
414 {
415     ASN1_TIME *tt;
416     int r = 0;
417     if (t)
418         tt = t;
419     else
420         tt = X509_gmtime_adj(NULL, 0);
421
422     if (!tt)
423         goto merr;
424
425     if (CMS_signed_add1_attr_by_NID(si, NID_pkcs9_signingTime,
426                                     tt->type, tt, -1) <= 0)
427         goto merr;
428
429     r = 1;
430
431  merr:
432
433     if (!t)
434         ASN1_TIME_free(tt);
435
436     if (!r)
437         CMSerr(CMS_F_CMS_ADD1_SIGNINGTIME, ERR_R_MALLOC_FAILURE);
438
439     return r;
440
441 }
442
443 EVP_PKEY_CTX *CMS_SignerInfo_get0_pkey_ctx(CMS_SignerInfo *si)
444 {
445     return si->pctx;
446 }
447
448 EVP_MD_CTX *CMS_SignerInfo_get0_md_ctx(CMS_SignerInfo *si)
449 {
450     return &si->mctx;
451 }
452
453 STACK_OF(CMS_SignerInfo) *CMS_get0_SignerInfos(CMS_ContentInfo *cms)
454 {
455     CMS_SignedData *sd;
456     sd = cms_get0_signed(cms);
457     if (!sd)
458         return NULL;
459     return sd->signerInfos;
460 }
461
462 STACK_OF(X509) *CMS_get0_signers(CMS_ContentInfo *cms)
463 {
464     STACK_OF(X509) *signers = NULL;
465     STACK_OF(CMS_SignerInfo) *sinfos;
466     CMS_SignerInfo *si;
467     int i;
468     sinfos = CMS_get0_SignerInfos(cms);
469     for (i = 0; i < sk_CMS_SignerInfo_num(sinfos); i++) {
470         si = sk_CMS_SignerInfo_value(sinfos, i);
471         if (si->signer) {
472             if (!signers) {
473                 signers = sk_X509_new_null();
474                 if (!signers)
475                     return NULL;
476             }
477             if (!sk_X509_push(signers, si->signer)) {
478                 sk_X509_free(signers);
479                 return NULL;
480             }
481         }
482     }
483     return signers;
484 }
485
486 void CMS_SignerInfo_set1_signer_cert(CMS_SignerInfo *si, X509 *signer)
487 {
488     if (signer) {
489         CRYPTO_add(&signer->references, 1, CRYPTO_LOCK_X509);
490         if (si->pkey)
491             EVP_PKEY_free(si->pkey);
492         si->pkey = X509_get_pubkey(signer);
493     }
494     if (si->signer)
495         X509_free(si->signer);
496     si->signer = signer;
497 }
498
499 int CMS_SignerInfo_get0_signer_id(CMS_SignerInfo *si,
500                                   ASN1_OCTET_STRING **keyid,
501                                   X509_NAME **issuer, ASN1_INTEGER **sno)
502 {
503     return cms_SignerIdentifier_get0_signer_id(si->sid, keyid, issuer, sno);
504 }
505
506 int CMS_SignerInfo_cert_cmp(CMS_SignerInfo *si, X509 *cert)
507 {
508     return cms_SignerIdentifier_cert_cmp(si->sid, cert);
509 }
510
511 int CMS_set1_signers_certs(CMS_ContentInfo *cms, STACK_OF(X509) *scerts,
512                            unsigned int flags)
513 {
514     CMS_SignedData *sd;
515     CMS_SignerInfo *si;
516     CMS_CertificateChoices *cch;
517     STACK_OF(CMS_CertificateChoices) *certs;
518     X509 *x;
519     int i, j;
520     int ret = 0;
521     sd = cms_get0_signed(cms);
522     if (!sd)
523         return -1;
524     certs = sd->certificates;
525     for (i = 0; i < sk_CMS_SignerInfo_num(sd->signerInfos); i++) {
526         si = sk_CMS_SignerInfo_value(sd->signerInfos, i);
527         if (si->signer)
528             continue;
529
530         for (j = 0; j < sk_X509_num(scerts); j++) {
531             x = sk_X509_value(scerts, j);
532             if (CMS_SignerInfo_cert_cmp(si, x) == 0) {
533                 CMS_SignerInfo_set1_signer_cert(si, x);
534                 ret++;
535                 break;
536             }
537         }
538
539         if (si->signer || (flags & CMS_NOINTERN))
540             continue;
541
542         for (j = 0; j < sk_CMS_CertificateChoices_num(certs); j++) {
543             cch = sk_CMS_CertificateChoices_value(certs, j);
544             if (cch->type != 0)
545                 continue;
546             x = cch->d.certificate;
547             if (CMS_SignerInfo_cert_cmp(si, x) == 0) {
548                 CMS_SignerInfo_set1_signer_cert(si, x);
549                 ret++;
550                 break;
551             }
552         }
553     }
554     return ret;
555 }
556
557 void CMS_SignerInfo_get0_algs(CMS_SignerInfo *si, EVP_PKEY **pk,
558                               X509 **signer, X509_ALGOR **pdig,
559                               X509_ALGOR **psig)
560 {
561     if (pk)
562         *pk = si->pkey;
563     if (signer)
564         *signer = si->signer;
565     if (pdig)
566         *pdig = si->digestAlgorithm;
567     if (psig)
568         *psig = si->signatureAlgorithm;
569 }
570
571 ASN1_OCTET_STRING *CMS_SignerInfo_get0_signature(CMS_SignerInfo *si)
572 {
573     return si->signature;
574 }
575
576 static int cms_SignerInfo_content_sign(CMS_ContentInfo *cms,
577                                        CMS_SignerInfo *si, BIO *chain)
578 {
579     EVP_MD_CTX mctx;
580     int r = 0;
581     EVP_PKEY_CTX *pctx = NULL;
582     EVP_MD_CTX_init(&mctx);
583
584     if (!si->pkey) {
585         CMSerr(CMS_F_CMS_SIGNERINFO_CONTENT_SIGN, CMS_R_NO_PRIVATE_KEY);
586         return 0;
587     }
588
589     if (!cms_DigestAlgorithm_find_ctx(&mctx, chain, si->digestAlgorithm))
590         goto err;
591     /* Set SignerInfo algortihm details if we used custom parametsr */
592     if (si->pctx && !cms_sd_asn1_ctrl(si, 0))
593         goto err;
594
595     /*
596      * If any signed attributes calculate and add messageDigest attribute
597      */
598
599     if (CMS_signed_get_attr_count(si) >= 0) {
600         ASN1_OBJECT *ctype =
601             cms->d.signedData->encapContentInfo->eContentType;
602         unsigned char md[EVP_MAX_MD_SIZE];
603         unsigned int mdlen;
604         if (!EVP_DigestFinal_ex(&mctx, md, &mdlen))
605             goto err;
606         if (!CMS_signed_add1_attr_by_NID(si, NID_pkcs9_messageDigest,
607                                          V_ASN1_OCTET_STRING, md, mdlen))
608             goto err;
609         /* Copy content type across */
610         if (CMS_signed_add1_attr_by_NID(si, NID_pkcs9_contentType,
611                                         V_ASN1_OBJECT, ctype, -1) <= 0)
612             goto err;
613         if (!CMS_SignerInfo_sign(si))
614             goto err;
615     } else if (si->pctx) {
616         unsigned char *sig;
617         size_t siglen;
618         unsigned char md[EVP_MAX_MD_SIZE];
619         unsigned int mdlen;
620         pctx = si->pctx;
621         if (!EVP_DigestFinal_ex(&mctx, md, &mdlen))
622             goto err;
623         siglen = EVP_PKEY_size(si->pkey);
624         sig = OPENSSL_malloc(siglen);
625         if (!sig) {
626             CMSerr(CMS_F_CMS_SIGNERINFO_CONTENT_SIGN, ERR_R_MALLOC_FAILURE);
627             goto err;
628         }
629         if (EVP_PKEY_sign(pctx, sig, &siglen, md, mdlen) <= 0) {
630             OPENSSL_free(sig);
631             goto err;
632         }
633         ASN1_STRING_set0(si->signature, sig, siglen);
634     } else {
635         unsigned char *sig;
636         unsigned int siglen;
637         sig = OPENSSL_malloc(EVP_PKEY_size(si->pkey));
638         if (!sig) {
639             CMSerr(CMS_F_CMS_SIGNERINFO_CONTENT_SIGN, ERR_R_MALLOC_FAILURE);
640             goto err;
641         }
642         if (!EVP_SignFinal(&mctx, sig, &siglen, si->pkey)) {
643             CMSerr(CMS_F_CMS_SIGNERINFO_CONTENT_SIGN, CMS_R_SIGNFINAL_ERROR);
644             OPENSSL_free(sig);
645             goto err;
646         }
647         ASN1_STRING_set0(si->signature, sig, siglen);
648     }
649
650     r = 1;
651
652  err:
653     EVP_MD_CTX_cleanup(&mctx);
654     if (pctx)
655         EVP_PKEY_CTX_free(pctx);
656     return r;
657
658 }
659
660 int cms_SignedData_final(CMS_ContentInfo *cms, BIO *chain)
661 {
662     STACK_OF(CMS_SignerInfo) *sinfos;
663     CMS_SignerInfo *si;
664     int i;
665     sinfos = CMS_get0_SignerInfos(cms);
666     for (i = 0; i < sk_CMS_SignerInfo_num(sinfos); i++) {
667         si = sk_CMS_SignerInfo_value(sinfos, i);
668         if (!cms_SignerInfo_content_sign(cms, si, chain))
669             return 0;
670     }
671     cms->d.signedData->encapContentInfo->partial = 0;
672     return 1;
673 }
674
675 int CMS_SignerInfo_sign(CMS_SignerInfo *si)
676 {
677     EVP_MD_CTX *mctx = &si->mctx;
678     EVP_PKEY_CTX *pctx;
679     unsigned char *abuf = NULL;
680     int alen;
681     size_t siglen;
682     const EVP_MD *md = NULL;
683
684     md = EVP_get_digestbyobj(si->digestAlgorithm->algorithm);
685     if (md == NULL)
686         return 0;
687
688     if (CMS_signed_get_attr_by_NID(si, NID_pkcs9_signingTime, -1) < 0) {
689         if (!cms_add1_signingTime(si, NULL))
690             goto err;
691     }
692
693     if (si->pctx)
694         pctx = si->pctx;
695     else {
696         EVP_MD_CTX_init(mctx);
697         if (EVP_DigestSignInit(mctx, &pctx, md, NULL, si->pkey) <= 0)
698             goto err;
699     }
700
701     if (EVP_PKEY_CTX_ctrl(pctx, -1, EVP_PKEY_OP_SIGN,
702                           EVP_PKEY_CTRL_CMS_SIGN, 0, si) <= 0) {
703         CMSerr(CMS_F_CMS_SIGNERINFO_SIGN, CMS_R_CTRL_ERROR);
704         goto err;
705     }
706
707     alen = ASN1_item_i2d((ASN1_VALUE *)si->signedAttrs, &abuf,
708                          ASN1_ITEM_rptr(CMS_Attributes_Sign));
709     if (!abuf)
710         goto err;
711     if (EVP_DigestSignUpdate(mctx, abuf, alen) <= 0)
712         goto err;
713     if (EVP_DigestSignFinal(mctx, NULL, &siglen) <= 0)
714         goto err;
715     OPENSSL_free(abuf);
716     abuf = OPENSSL_malloc(siglen);
717     if (!abuf)
718         goto err;
719     if (EVP_DigestSignFinal(mctx, abuf, &siglen) <= 0)
720         goto err;
721
722     if (EVP_PKEY_CTX_ctrl(pctx, -1, EVP_PKEY_OP_SIGN,
723                           EVP_PKEY_CTRL_CMS_SIGN, 1, si) <= 0) {
724         CMSerr(CMS_F_CMS_SIGNERINFO_SIGN, CMS_R_CTRL_ERROR);
725         goto err;
726     }
727
728     EVP_MD_CTX_cleanup(mctx);
729
730     ASN1_STRING_set0(si->signature, abuf, siglen);
731
732     return 1;
733
734  err:
735     if (abuf)
736         OPENSSL_free(abuf);
737     EVP_MD_CTX_cleanup(mctx);
738     return 0;
739
740 }
741
742 int CMS_SignerInfo_verify(CMS_SignerInfo *si)
743 {
744     EVP_MD_CTX *mctx = &si->mctx;
745     unsigned char *abuf = NULL;
746     int alen, r = -1;
747     const EVP_MD *md = NULL;
748
749     if (!si->pkey) {
750         CMSerr(CMS_F_CMS_SIGNERINFO_VERIFY, CMS_R_NO_PUBLIC_KEY);
751         return -1;
752     }
753
754     md = EVP_get_digestbyobj(si->digestAlgorithm->algorithm);
755     if (md == NULL)
756         return -1;
757     EVP_MD_CTX_init(mctx);
758     if (EVP_DigestVerifyInit(mctx, &si->pctx, md, NULL, si->pkey) <= 0)
759         goto err;
760
761     if (!cms_sd_asn1_ctrl(si, 1))
762         goto err;
763
764     alen = ASN1_item_i2d((ASN1_VALUE *)si->signedAttrs, &abuf,
765                          ASN1_ITEM_rptr(CMS_Attributes_Verify));
766     if (!abuf)
767         goto err;
768     r = EVP_DigestVerifyUpdate(mctx, abuf, alen);
769     OPENSSL_free(abuf);
770     if (r <= 0) {
771         r = -1;
772         goto err;
773     }
774     r = EVP_DigestVerifyFinal(mctx,
775                               si->signature->data, si->signature->length);
776     if (r <= 0)
777         CMSerr(CMS_F_CMS_SIGNERINFO_VERIFY, CMS_R_VERIFICATION_FAILURE);
778  err:
779     EVP_MD_CTX_cleanup(mctx);
780     return r;
781 }
782
783 /* Create a chain of digest BIOs from a CMS ContentInfo */
784
785 BIO *cms_SignedData_init_bio(CMS_ContentInfo *cms)
786 {
787     int i;
788     CMS_SignedData *sd;
789     BIO *chain = NULL;
790     sd = cms_get0_signed(cms);
791     if (!sd)
792         return NULL;
793     if (cms->d.signedData->encapContentInfo->partial)
794         cms_sd_set_version(sd);
795     for (i = 0; i < sk_X509_ALGOR_num(sd->digestAlgorithms); i++) {
796         X509_ALGOR *digestAlgorithm;
797         BIO *mdbio;
798         digestAlgorithm = sk_X509_ALGOR_value(sd->digestAlgorithms, i);
799         mdbio = cms_DigestAlgorithm_init_bio(digestAlgorithm);
800         if (!mdbio)
801             goto err;
802         if (chain)
803             BIO_push(chain, mdbio);
804         else
805             chain = mdbio;
806     }
807     return chain;
808  err:
809     BIO_free_all(chain);
810     return NULL;
811 }
812
813 int CMS_SignerInfo_verify_content(CMS_SignerInfo *si, BIO *chain)
814 {
815     ASN1_OCTET_STRING *os = NULL;
816     EVP_MD_CTX mctx;
817     EVP_PKEY_CTX *pkctx = NULL;
818     int r = -1;
819     unsigned char mval[EVP_MAX_MD_SIZE];
820     unsigned int mlen;
821     EVP_MD_CTX_init(&mctx);
822     /* If we have any signed attributes look for messageDigest value */
823     if (CMS_signed_get_attr_count(si) >= 0) {
824         os = CMS_signed_get0_data_by_OBJ(si,
825                                          OBJ_nid2obj(NID_pkcs9_messageDigest),
826                                          -3, V_ASN1_OCTET_STRING);
827         if (!os) {
828             CMSerr(CMS_F_CMS_SIGNERINFO_VERIFY_CONTENT,
829                    CMS_R_ERROR_READING_MESSAGEDIGEST_ATTRIBUTE);
830             goto err;
831         }
832     }
833
834     if (!cms_DigestAlgorithm_find_ctx(&mctx, chain, si->digestAlgorithm))
835         goto err;
836
837     if (EVP_DigestFinal_ex(&mctx, mval, &mlen) <= 0) {
838         CMSerr(CMS_F_CMS_SIGNERINFO_VERIFY_CONTENT,
839                CMS_R_UNABLE_TO_FINALIZE_CONTEXT);
840         goto err;
841     }
842
843     /* If messageDigest found compare it */
844
845     if (os) {
846         if (mlen != (unsigned int)os->length) {
847             CMSerr(CMS_F_CMS_SIGNERINFO_VERIFY_CONTENT,
848                    CMS_R_MESSAGEDIGEST_ATTRIBUTE_WRONG_LENGTH);
849             goto err;
850         }
851
852         if (memcmp(mval, os->data, mlen)) {
853             CMSerr(CMS_F_CMS_SIGNERINFO_VERIFY_CONTENT,
854                    CMS_R_VERIFICATION_FAILURE);
855             r = 0;
856         } else
857             r = 1;
858     } else {
859         const EVP_MD *md = EVP_MD_CTX_md(&mctx);
860         pkctx = EVP_PKEY_CTX_new(si->pkey, NULL);
861         if (EVP_PKEY_verify_init(pkctx) <= 0)
862             goto err;
863         if (EVP_PKEY_CTX_set_signature_md(pkctx, md) <= 0)
864             goto err;
865         si->pctx = pkctx;
866         if (!cms_sd_asn1_ctrl(si, 1))
867             goto err;
868         r = EVP_PKEY_verify(pkctx, si->signature->data,
869                             si->signature->length, mval, mlen);
870         if (r <= 0) {
871             CMSerr(CMS_F_CMS_SIGNERINFO_VERIFY_CONTENT,
872                    CMS_R_VERIFICATION_FAILURE);
873             r = 0;
874         }
875     }
876
877  err:
878     if (pkctx)
879         EVP_PKEY_CTX_free(pkctx);
880     EVP_MD_CTX_cleanup(&mctx);
881     return r;
882
883 }
884
885 int CMS_add_smimecap(CMS_SignerInfo *si, STACK_OF(X509_ALGOR) *algs)
886 {
887     unsigned char *smder = NULL;
888     int smderlen, r;
889     smderlen = i2d_X509_ALGORS(algs, &smder);
890     if (smderlen <= 0)
891         return 0;
892     r = CMS_signed_add1_attr_by_NID(si, NID_SMIMECapabilities,
893                                     V_ASN1_SEQUENCE, smder, smderlen);
894     OPENSSL_free(smder);
895     return r;
896 }
897
898 int CMS_add_simple_smimecap(STACK_OF(X509_ALGOR) **algs,
899                             int algnid, int keysize)
900 {
901     X509_ALGOR *alg;
902     ASN1_INTEGER *key = NULL;
903     if (keysize > 0) {
904         key = ASN1_INTEGER_new();
905         if (!key || !ASN1_INTEGER_set(key, keysize))
906             return 0;
907     }
908     alg = X509_ALGOR_new();
909     if (!alg) {
910         if (key)
911             ASN1_INTEGER_free(key);
912         return 0;
913     }
914
915     X509_ALGOR_set0(alg, OBJ_nid2obj(algnid),
916                     key ? V_ASN1_INTEGER : V_ASN1_UNDEF, key);
917     if (!*algs)
918         *algs = sk_X509_ALGOR_new_null();
919     if (!*algs || !sk_X509_ALGOR_push(*algs, alg)) {
920         X509_ALGOR_free(alg);
921         return 0;
922     }
923     return 1;
924 }
925
926 /* Check to see if a cipher exists and if so add S/MIME capabilities */
927
928 static int cms_add_cipher_smcap(STACK_OF(X509_ALGOR) **sk, int nid, int arg)
929 {
930     if (EVP_get_cipherbynid(nid))
931         return CMS_add_simple_smimecap(sk, nid, arg);
932     return 1;
933 }
934
935 static int cms_add_digest_smcap(STACK_OF(X509_ALGOR) **sk, int nid, int arg)
936 {
937     if (EVP_get_digestbynid(nid))
938         return CMS_add_simple_smimecap(sk, nid, arg);
939     return 1;
940 }
941
942 int CMS_add_standard_smimecap(STACK_OF(X509_ALGOR) **smcap)
943 {
944     if (!cms_add_cipher_smcap(smcap, NID_aes_256_cbc, -1)
945         || !cms_add_digest_smcap(smcap, NID_id_GostR3411_94, -1)
946         || !cms_add_cipher_smcap(smcap, NID_id_Gost28147_89, -1)
947         || !cms_add_cipher_smcap(smcap, NID_aes_192_cbc, -1)
948         || !cms_add_cipher_smcap(smcap, NID_aes_128_cbc, -1)
949         || !cms_add_cipher_smcap(smcap, NID_des_ede3_cbc, -1)
950         || !cms_add_cipher_smcap(smcap, NID_rc2_cbc, 128)
951         || !cms_add_cipher_smcap(smcap, NID_rc2_cbc, 64)
952         || !cms_add_cipher_smcap(smcap, NID_des_cbc, -1)
953         || !cms_add_cipher_smcap(smcap, NID_rc2_cbc, 40))
954         return 0;
955     return 1;
956 }