Run the withlibctx.pl script
[openssl.git] / crypto / pkcs7 / pk7_doit.c
1 /*
2  * Copyright 1995-2020 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the Apache License 2.0 (the "License").  You may not use
5  * this file except in compliance with the License.  You can obtain a copy
6  * in the file LICENSE in the source distribution or at
7  * https://www.openssl.org/source/license.html
8  */
9
10 #include <stdio.h>
11 #include "internal/cryptlib.h"
12 #include <openssl/rand.h>
13 #include <openssl/objects.h>
14 #include <openssl/x509.h>
15 #include <openssl/x509v3.h>
16 #include <openssl/err.h>
17 #include "pk7_local.h"
18
19 static int add_attribute(STACK_OF(X509_ATTRIBUTE) **sk, int nid, int atrtype,
20                          void *value);
21 static ASN1_TYPE *get_attribute(STACK_OF(X509_ATTRIBUTE) *sk, int nid);
22
23 static int PKCS7_type_is_other(PKCS7 *p7)
24 {
25     int isOther = 1;
26
27     int nid = OBJ_obj2nid(p7->type);
28
29     switch (nid) {
30     case NID_pkcs7_data:
31     case NID_pkcs7_signed:
32     case NID_pkcs7_enveloped:
33     case NID_pkcs7_signedAndEnveloped:
34     case NID_pkcs7_digest:
35     case NID_pkcs7_encrypted:
36         isOther = 0;
37         break;
38     default:
39         isOther = 1;
40     }
41
42     return isOther;
43
44 }
45
46 static ASN1_OCTET_STRING *PKCS7_get_octet_string(PKCS7 *p7)
47 {
48     if (PKCS7_type_is_data(p7))
49         return p7->d.data;
50     if (PKCS7_type_is_other(p7) && p7->d.other
51         && (p7->d.other->type == V_ASN1_OCTET_STRING))
52         return p7->d.other->value.octet_string;
53     return NULL;
54 }
55
56 static int pkcs7_bio_add_digest(BIO **pbio, X509_ALGOR *alg,
57                                 const PKCS7_CTX *ctx)
58 {
59     BIO *btmp;
60     const char *name;
61     EVP_MD *fetched = NULL;
62     const EVP_MD *md;
63
64     if ((btmp = BIO_new(BIO_f_md())) == NULL) {
65         PKCS7err(PKCS7_F_PKCS7_BIO_ADD_DIGEST, ERR_R_BIO_LIB);
66         goto err;
67     }
68
69     name = OBJ_nid2sn(OBJ_obj2nid(alg->algorithm));
70
71     (void)ERR_set_mark();
72     fetched = EVP_MD_fetch(ctx->libctx, name, ctx->propq);
73     if (fetched != NULL)
74         md = fetched;
75     else
76         md = EVP_get_digestbyname(name);
77
78     if (md == NULL) {
79         (void)ERR_clear_last_mark();
80         PKCS7err(PKCS7_F_PKCS7_BIO_ADD_DIGEST, PKCS7_R_UNKNOWN_DIGEST_TYPE);
81         goto err;
82     }
83     (void)ERR_pop_to_mark();
84
85     BIO_set_md(btmp, md);
86     EVP_MD_free(fetched);
87     if (*pbio == NULL)
88         *pbio = btmp;
89     else if (!BIO_push(*pbio, btmp)) {
90         PKCS7err(PKCS7_F_PKCS7_BIO_ADD_DIGEST, ERR_R_BIO_LIB);
91         goto err;
92     }
93     btmp = NULL;
94
95     return 1;
96
97  err:
98     BIO_free(btmp);
99     return 0;
100 }
101
102 static int pkcs7_encode_rinfo(PKCS7_RECIP_INFO *ri,
103                               unsigned char *key, int keylen)
104 {
105     EVP_PKEY_CTX *pctx = NULL;
106     EVP_PKEY *pkey = NULL;
107     unsigned char *ek = NULL;
108     int ret = 0;
109     size_t eklen;
110     const PKCS7_CTX *ctx = ri->ctx;
111
112     pkey = X509_get0_pubkey(ri->cert);
113     if (pkey == NULL)
114         return 0;
115
116     pctx = EVP_PKEY_CTX_new_from_pkey(ctx->libctx, pkey, ctx->propq);
117     if (pctx == NULL)
118         return 0;
119
120     if (EVP_PKEY_encrypt_init(pctx) <= 0)
121         goto err;
122
123     if (EVP_PKEY_CTX_ctrl(pctx, -1, EVP_PKEY_OP_ENCRYPT,
124                           EVP_PKEY_CTRL_PKCS7_ENCRYPT, 0, ri) <= 0) {
125         PKCS7err(PKCS7_F_PKCS7_ENCODE_RINFO, PKCS7_R_CTRL_ERROR);
126         goto err;
127     }
128
129     if (EVP_PKEY_encrypt(pctx, NULL, &eklen, key, keylen) <= 0)
130         goto err;
131
132     ek = OPENSSL_malloc(eklen);
133
134     if (ek == NULL) {
135         PKCS7err(PKCS7_F_PKCS7_ENCODE_RINFO, ERR_R_MALLOC_FAILURE);
136         goto err;
137     }
138
139     if (EVP_PKEY_encrypt(pctx, ek, &eklen, key, keylen) <= 0)
140         goto err;
141
142     ASN1_STRING_set0(ri->enc_key, ek, eklen);
143     ek = NULL;
144
145     ret = 1;
146
147  err:
148     EVP_PKEY_CTX_free(pctx);
149     OPENSSL_free(ek);
150     return ret;
151
152 }
153
154 static int pkcs7_decrypt_rinfo(unsigned char **pek, int *peklen,
155                                PKCS7_RECIP_INFO *ri, EVP_PKEY *pkey,
156                                size_t fixlen)
157 {
158     EVP_PKEY_CTX *pctx = NULL;
159     unsigned char *ek = NULL;
160     size_t eklen;
161     int ret = -1;
162     const PKCS7_CTX *ctx = ri->ctx;
163
164     pctx = EVP_PKEY_CTX_new_from_pkey(ctx->libctx, pkey, ctx->propq);
165     if (pctx == NULL)
166         return -1;
167
168     if (EVP_PKEY_decrypt_init(pctx) <= 0)
169         goto err;
170
171     if (EVP_PKEY_CTX_ctrl(pctx, -1, EVP_PKEY_OP_DECRYPT,
172                           EVP_PKEY_CTRL_PKCS7_DECRYPT, 0, ri) <= 0) {
173         PKCS7err(PKCS7_F_PKCS7_DECRYPT_RINFO, PKCS7_R_CTRL_ERROR);
174         goto err;
175     }
176
177     if (EVP_PKEY_decrypt(pctx, NULL, &eklen,
178                          ri->enc_key->data, ri->enc_key->length) <= 0)
179         goto err;
180
181     ek = OPENSSL_malloc(eklen);
182
183     if (ek == NULL) {
184         PKCS7err(PKCS7_F_PKCS7_DECRYPT_RINFO, ERR_R_MALLOC_FAILURE);
185         goto err;
186     }
187
188     if (EVP_PKEY_decrypt(pctx, ek, &eklen,
189                          ri->enc_key->data, ri->enc_key->length) <= 0
190             || eklen == 0
191             || (fixlen != 0 && eklen != fixlen)) {
192         ret = 0;
193         PKCS7err(PKCS7_F_PKCS7_DECRYPT_RINFO, ERR_R_EVP_LIB);
194         goto err;
195     }
196
197     ret = 1;
198
199     OPENSSL_clear_free(*pek, *peklen);
200     *pek = ek;
201     *peklen = eklen;
202
203  err:
204     EVP_PKEY_CTX_free(pctx);
205     if (!ret)
206         OPENSSL_free(ek);
207
208     return ret;
209 }
210
211 BIO *PKCS7_dataInit(PKCS7 *p7, BIO *bio)
212 {
213     int i;
214     BIO *out = NULL, *btmp = NULL;
215     X509_ALGOR *xa = NULL;
216     EVP_CIPHER *fetched_cipher = NULL;
217     const EVP_CIPHER *cipher;
218     const EVP_CIPHER *evp_cipher = NULL;
219     STACK_OF(X509_ALGOR) *md_sk = NULL;
220     STACK_OF(PKCS7_RECIP_INFO) *rsk = NULL;
221     X509_ALGOR *xalg = NULL;
222     PKCS7_RECIP_INFO *ri = NULL;
223     ASN1_OCTET_STRING *os = NULL;
224     const PKCS7_CTX *p7_ctx;
225
226     if (p7 == NULL) {
227         PKCS7err(PKCS7_F_PKCS7_DATAINIT, PKCS7_R_INVALID_NULL_POINTER);
228         return NULL;
229     }
230     p7_ctx = pkcs7_get0_ctx(p7);
231
232     /*
233      * The content field in the PKCS7 ContentInfo is optional, but that really
234      * only applies to inner content (precisely, detached signatures).
235      *
236      * When reading content, missing outer content is therefore treated as an
237      * error.
238      *
239      * When creating content, PKCS7_content_new() must be called before
240      * calling this method, so a NULL p7->d is always an error.
241      */
242     if (p7->d.ptr == NULL) {
243         PKCS7err(PKCS7_F_PKCS7_DATAINIT, PKCS7_R_NO_CONTENT);
244         return NULL;
245     }
246
247     i = OBJ_obj2nid(p7->type);
248     p7->state = PKCS7_S_HEADER;
249
250     switch (i) {
251     case NID_pkcs7_signed:
252         md_sk = p7->d.sign->md_algs;
253         os = PKCS7_get_octet_string(p7->d.sign->contents);
254         break;
255     case NID_pkcs7_signedAndEnveloped:
256         rsk = p7->d.signed_and_enveloped->recipientinfo;
257         md_sk = p7->d.signed_and_enveloped->md_algs;
258         xalg = p7->d.signed_and_enveloped->enc_data->algorithm;
259         evp_cipher = p7->d.signed_and_enveloped->enc_data->cipher;
260         if (evp_cipher == NULL) {
261             PKCS7err(PKCS7_F_PKCS7_DATAINIT, PKCS7_R_CIPHER_NOT_INITIALIZED);
262             goto err;
263         }
264         break;
265     case NID_pkcs7_enveloped:
266         rsk = p7->d.enveloped->recipientinfo;
267         xalg = p7->d.enveloped->enc_data->algorithm;
268         evp_cipher = p7->d.enveloped->enc_data->cipher;
269         if (evp_cipher == NULL) {
270             PKCS7err(PKCS7_F_PKCS7_DATAINIT, PKCS7_R_CIPHER_NOT_INITIALIZED);
271             goto err;
272         }
273         break;
274     case NID_pkcs7_digest:
275         xa = p7->d.digest->md;
276         os = PKCS7_get_octet_string(p7->d.digest->contents);
277         break;
278     case NID_pkcs7_data:
279         break;
280     default:
281         PKCS7err(PKCS7_F_PKCS7_DATAINIT, PKCS7_R_UNSUPPORTED_CONTENT_TYPE);
282         goto err;
283     }
284
285     for (i = 0; i < sk_X509_ALGOR_num(md_sk); i++)
286         if (!pkcs7_bio_add_digest(&out, sk_X509_ALGOR_value(md_sk, i), p7_ctx))
287             goto err;
288
289     if (xa && !pkcs7_bio_add_digest(&out, xa, p7_ctx))
290         goto err;
291
292     if (evp_cipher != NULL) {
293         unsigned char key[EVP_MAX_KEY_LENGTH];
294         unsigned char iv[EVP_MAX_IV_LENGTH];
295         int keylen, ivlen;
296         EVP_CIPHER_CTX *ctx;
297
298         if ((btmp = BIO_new(BIO_f_cipher())) == NULL) {
299             PKCS7err(PKCS7_F_PKCS7_DATAINIT, ERR_R_BIO_LIB);
300             goto err;
301         }
302         BIO_get_cipher_ctx(btmp, &ctx);
303         keylen = EVP_CIPHER_key_length(evp_cipher);
304         ivlen = EVP_CIPHER_iv_length(evp_cipher);
305         xalg->algorithm = OBJ_nid2obj(EVP_CIPHER_type(evp_cipher));
306         if (ivlen > 0)
307             if (RAND_bytes_ex(p7_ctx->libctx, iv, ivlen) <= 0)
308                 goto err;
309
310         (void)ERR_set_mark();
311         fetched_cipher = EVP_CIPHER_fetch(p7_ctx->libctx,
312                                           EVP_CIPHER_name(evp_cipher),
313                                           p7_ctx->propq);
314         (void)ERR_pop_to_mark();
315         if (fetched_cipher != NULL)
316             cipher = fetched_cipher;
317         else
318             cipher = evp_cipher;
319
320         if (EVP_CipherInit_ex(ctx, cipher, NULL, NULL, NULL, 1) <= 0)
321             goto err;
322
323         EVP_CIPHER_free(fetched_cipher);
324         fetched_cipher = NULL;
325
326         if (EVP_CIPHER_CTX_rand_key(ctx, key) <= 0)
327             goto err;
328         if (EVP_CipherInit_ex(ctx, NULL, NULL, key, iv, 1) <= 0)
329             goto err;
330
331         if (ivlen > 0) {
332             if (xalg->parameter == NULL) {
333                 xalg->parameter = ASN1_TYPE_new();
334                 if (xalg->parameter == NULL)
335                     goto err;
336             }
337             if (EVP_CIPHER_param_to_asn1(ctx, xalg->parameter) < 0)
338                 goto err;
339         }
340
341         /* Lets do the pub key stuff :-) */
342         for (i = 0; i < sk_PKCS7_RECIP_INFO_num(rsk); i++) {
343             ri = sk_PKCS7_RECIP_INFO_value(rsk, i);
344             if (pkcs7_encode_rinfo(ri, key, keylen) <= 0)
345                 goto err;
346         }
347         OPENSSL_cleanse(key, keylen);
348
349         if (out == NULL)
350             out = btmp;
351         else
352             BIO_push(out, btmp);
353         btmp = NULL;
354     }
355
356     if (bio == NULL) {
357         if (PKCS7_is_detached(p7)) {
358             bio = BIO_new(BIO_s_null());
359         } else if (os && os->length > 0) {
360             bio = BIO_new_mem_buf(os->data, os->length);
361         } else {
362             bio = BIO_new(BIO_s_mem());
363             if (bio == NULL)
364                 goto err;
365             BIO_set_mem_eof_return(bio, 0);
366         }
367         if (bio == NULL)
368             goto err;
369     }
370     if (out)
371         BIO_push(out, bio);
372     else
373         out = bio;
374     return out;
375
376  err:
377     EVP_CIPHER_free(fetched_cipher);
378     BIO_free_all(out);
379     BIO_free_all(btmp);
380     return NULL;
381 }
382
383 static int pkcs7_cmp_ri(PKCS7_RECIP_INFO *ri, X509 *pcert)
384 {
385     int ret;
386     ret = X509_NAME_cmp(ri->issuer_and_serial->issuer,
387                         X509_get_issuer_name(pcert));
388     if (ret)
389         return ret;
390     return ASN1_INTEGER_cmp(X509_get0_serialNumber(pcert),
391                             ri->issuer_and_serial->serial);
392 }
393
394 /* int */
395 BIO *PKCS7_dataDecode(PKCS7 *p7, EVP_PKEY *pkey, BIO *in_bio, X509 *pcert)
396 {
397     int i, len;
398     BIO *out = NULL, *btmp = NULL, *etmp = NULL, *bio = NULL;
399     X509_ALGOR *xa;
400     ASN1_OCTET_STRING *data_body = NULL;
401     EVP_MD *evp_md = NULL;
402     const EVP_MD *md;
403     EVP_CIPHER *evp_cipher = NULL;
404     const EVP_CIPHER *cipher = NULL;
405     EVP_CIPHER_CTX *evp_ctx = NULL;
406     X509_ALGOR *enc_alg = NULL;
407     STACK_OF(X509_ALGOR) *md_sk = NULL;
408     STACK_OF(PKCS7_RECIP_INFO) *rsk = NULL;
409     PKCS7_RECIP_INFO *ri = NULL;
410     unsigned char *ek = NULL, *tkey = NULL;
411     int eklen = 0, tkeylen = 0;
412     const char *name;
413     const PKCS7_CTX *p7_ctx;
414
415     if (p7 == NULL) {
416         PKCS7err(PKCS7_F_PKCS7_DATADECODE, PKCS7_R_INVALID_NULL_POINTER);
417         return NULL;
418     }
419
420     p7_ctx = pkcs7_get0_ctx(p7);
421
422     if (p7->d.ptr == NULL) {
423         PKCS7err(PKCS7_F_PKCS7_DATADECODE, PKCS7_R_NO_CONTENT);
424         return NULL;
425     }
426
427     i = OBJ_obj2nid(p7->type);
428     p7->state = PKCS7_S_HEADER;
429
430     switch (i) {
431     case NID_pkcs7_signed:
432         /*
433          * p7->d.sign->contents is a PKCS7 structure consisting of a contentType
434          * field and optional content.
435          * data_body is NULL if that structure has no (=detached) content
436          * or if the contentType is wrong (i.e., not "data").
437          */
438         data_body = PKCS7_get_octet_string(p7->d.sign->contents);
439         if (!PKCS7_is_detached(p7) && data_body == NULL) {
440             PKCS7err(PKCS7_F_PKCS7_DATADECODE,
441                      PKCS7_R_INVALID_SIGNED_DATA_TYPE);
442             goto err;
443         }
444         md_sk = p7->d.sign->md_algs;
445         break;
446     case NID_pkcs7_signedAndEnveloped:
447         rsk = p7->d.signed_and_enveloped->recipientinfo;
448         md_sk = p7->d.signed_and_enveloped->md_algs;
449         /* data_body is NULL if the optional EncryptedContent is missing. */
450         data_body = p7->d.signed_and_enveloped->enc_data->enc_data;
451         enc_alg = p7->d.signed_and_enveloped->enc_data->algorithm;
452
453         name = OBJ_nid2sn(OBJ_obj2nid(enc_alg->algorithm));
454
455         (void)ERR_set_mark();
456         evp_cipher = EVP_CIPHER_fetch(p7_ctx->libctx, name, p7_ctx->propq);
457         if (evp_cipher != NULL)
458             cipher = evp_cipher;
459         else
460             cipher = EVP_get_cipherbyname(name);
461
462         if (cipher == NULL) {
463             (void)ERR_clear_last_mark();
464             PKCS7err(PKCS7_F_PKCS7_DATADECODE,
465                      PKCS7_R_UNSUPPORTED_CIPHER_TYPE);
466             goto err;
467         }
468         (void)ERR_pop_to_mark();
469         break;
470     case NID_pkcs7_enveloped:
471         rsk = p7->d.enveloped->recipientinfo;
472         enc_alg = p7->d.enveloped->enc_data->algorithm;
473         /* data_body is NULL if the optional EncryptedContent is missing. */
474         data_body = p7->d.enveloped->enc_data->enc_data;
475         name = OBJ_nid2sn(OBJ_obj2nid(enc_alg->algorithm));
476
477         (void)ERR_set_mark();
478         evp_cipher = EVP_CIPHER_fetch(p7_ctx->libctx, name, p7_ctx->propq);
479         if (evp_cipher != NULL)
480             cipher = evp_cipher;
481         else
482             cipher = EVP_get_cipherbyname(name);
483
484         if (cipher == NULL) {
485             (void)ERR_clear_last_mark();
486             PKCS7err(PKCS7_F_PKCS7_DATADECODE,
487                      PKCS7_R_UNSUPPORTED_CIPHER_TYPE);
488             goto err;
489         }
490         (void)ERR_pop_to_mark();
491         break;
492     default:
493         PKCS7err(PKCS7_F_PKCS7_DATADECODE, PKCS7_R_UNSUPPORTED_CONTENT_TYPE);
494         goto err;
495     }
496
497     /* Detached content must be supplied via in_bio instead. */
498     if (data_body == NULL && in_bio == NULL) {
499         PKCS7err(PKCS7_F_PKCS7_DATADECODE, PKCS7_R_NO_CONTENT);
500         goto err;
501     }
502
503     /* We will be checking the signature */
504     if (md_sk != NULL) {
505         for (i = 0; i < sk_X509_ALGOR_num(md_sk); i++) {
506             xa = sk_X509_ALGOR_value(md_sk, i);
507             if ((btmp = BIO_new(BIO_f_md())) == NULL) {
508                 PKCS7err(PKCS7_F_PKCS7_DATADECODE, ERR_R_BIO_LIB);
509                 goto err;
510             }
511
512             name = OBJ_nid2sn(OBJ_obj2nid(xa->algorithm));
513
514             (void)ERR_set_mark();
515             evp_md = EVP_MD_fetch(p7_ctx->libctx, name, p7_ctx->propq);
516             if (evp_md != NULL)
517                 md = evp_md;
518             else
519                 md = EVP_get_digestbyname(name);
520
521             if (md == NULL) {
522                 (void)ERR_clear_last_mark();
523                 PKCS7err(PKCS7_F_PKCS7_DATADECODE,
524                          PKCS7_R_UNKNOWN_DIGEST_TYPE);
525                 goto err;
526             }
527             (void)ERR_pop_to_mark();
528
529             BIO_set_md(btmp, md);
530             EVP_MD_free(evp_md);
531             if (out == NULL)
532                 out = btmp;
533             else
534                 BIO_push(out, btmp);
535             btmp = NULL;
536         }
537     }
538
539     if (cipher != NULL) {
540         if ((etmp = BIO_new(BIO_f_cipher())) == NULL) {
541             PKCS7err(PKCS7_F_PKCS7_DATADECODE, ERR_R_BIO_LIB);
542             goto err;
543         }
544
545         /*
546          * It was encrypted, we need to decrypt the secret key with the
547          * private key
548          */
549
550         /*
551          * Find the recipientInfo which matches the passed certificate (if
552          * any)
553          */
554
555         if (pcert) {
556             for (i = 0; i < sk_PKCS7_RECIP_INFO_num(rsk); i++) {
557                 ri = sk_PKCS7_RECIP_INFO_value(rsk, i);
558                 if (!pkcs7_cmp_ri(ri, pcert))
559                     break;
560                 ri = NULL;
561             }
562             if (ri == NULL) {
563                 PKCS7err(PKCS7_F_PKCS7_DATADECODE,
564                          PKCS7_R_NO_RECIPIENT_MATCHES_CERTIFICATE);
565                 goto err;
566             }
567         }
568
569         /* If we haven't got a certificate try each ri in turn */
570         if (pcert == NULL) {
571             /*
572              * Always attempt to decrypt all rinfo even after success as a
573              * defence against MMA timing attacks.
574              */
575             for (i = 0; i < sk_PKCS7_RECIP_INFO_num(rsk); i++) {
576                 ri = sk_PKCS7_RECIP_INFO_value(rsk, i);
577                 ri->ctx = p7_ctx;
578                 if (pkcs7_decrypt_rinfo(&ek, &eklen, ri, pkey,
579                         EVP_CIPHER_key_length(cipher)) < 0)
580                     goto err;
581                 ERR_clear_error();
582             }
583         } else {
584             ri->ctx = p7_ctx;
585             /* Only exit on fatal errors, not decrypt failure */
586             if (pkcs7_decrypt_rinfo(&ek, &eklen, ri, pkey, 0) < 0)
587                 goto err;
588             ERR_clear_error();
589         }
590
591         evp_ctx = NULL;
592         BIO_get_cipher_ctx(etmp, &evp_ctx);
593         if (EVP_CipherInit_ex(evp_ctx, cipher, NULL, NULL, NULL, 0) <= 0)
594             goto err;
595         if (EVP_CIPHER_asn1_to_param(evp_ctx, enc_alg->parameter) < 0)
596             goto err;
597         /* Generate random key as MMA defence */
598         len = EVP_CIPHER_CTX_key_length(evp_ctx);
599         if (len <= 0)
600             goto err;
601         tkeylen = (size_t)len;
602         tkey = OPENSSL_malloc(tkeylen);
603         if (tkey == NULL)
604             goto err;
605         if (EVP_CIPHER_CTX_rand_key(evp_ctx, tkey) <= 0)
606             goto err;
607         if (ek == NULL) {
608             ek = tkey;
609             eklen = tkeylen;
610             tkey = NULL;
611         }
612
613         if (eklen != EVP_CIPHER_CTX_key_length(evp_ctx)) {
614             /*
615              * Some S/MIME clients don't use the same key and effective key
616              * length. The key length is determined by the size of the
617              * decrypted RSA key.
618              */
619             if (!EVP_CIPHER_CTX_set_key_length(evp_ctx, eklen)) {
620                 /* Use random key as MMA defence */
621                 OPENSSL_clear_free(ek, eklen);
622                 ek = tkey;
623                 eklen = tkeylen;
624                 tkey = NULL;
625             }
626         }
627         /* Clear errors so we don't leak information useful in MMA */
628         ERR_clear_error();
629         if (EVP_CipherInit_ex(evp_ctx, NULL, NULL, ek, NULL, 0) <= 0)
630             goto err;
631
632         OPENSSL_clear_free(ek, eklen);
633         ek = NULL;
634         OPENSSL_clear_free(tkey, tkeylen);
635         tkey = NULL;
636
637         if (out == NULL)
638             out = etmp;
639         else
640             BIO_push(out, etmp);
641         etmp = NULL;
642     }
643     if (in_bio != NULL) {
644         bio = in_bio;
645     } else {
646         if (data_body->length > 0)
647             bio = BIO_new_mem_buf(data_body->data, data_body->length);
648         else {
649             bio = BIO_new(BIO_s_mem());
650             if (bio == NULL)
651                 goto err;
652             BIO_set_mem_eof_return(bio, 0);
653         }
654         if (bio == NULL)
655             goto err;
656     }
657     BIO_push(out, bio);
658     bio = NULL;
659     EVP_CIPHER_free(evp_cipher);
660     return out;
661
662  err:
663     EVP_CIPHER_free(evp_cipher);
664     OPENSSL_clear_free(ek, eklen);
665     OPENSSL_clear_free(tkey, tkeylen);
666     BIO_free_all(out);
667     BIO_free_all(btmp);
668     BIO_free_all(etmp);
669     BIO_free_all(bio);
670     return NULL;
671 }
672
673 static BIO *PKCS7_find_digest(EVP_MD_CTX **pmd, BIO *bio, int nid)
674 {
675     for (;;) {
676         bio = BIO_find_type(bio, BIO_TYPE_MD);
677         if (bio == NULL) {
678             PKCS7err(PKCS7_F_PKCS7_FIND_DIGEST,
679                      PKCS7_R_UNABLE_TO_FIND_MESSAGE_DIGEST);
680             return NULL;
681         }
682         BIO_get_md_ctx(bio, pmd);
683         if (*pmd == NULL) {
684             PKCS7err(PKCS7_F_PKCS7_FIND_DIGEST, ERR_R_INTERNAL_ERROR);
685             return NULL;
686         }
687         if (EVP_MD_CTX_type(*pmd) == nid)
688             return bio;
689         bio = BIO_next(bio);
690     }
691     return NULL;
692 }
693
694 static int do_pkcs7_signed_attrib(PKCS7_SIGNER_INFO *si, EVP_MD_CTX *mctx)
695 {
696     unsigned char md_data[EVP_MAX_MD_SIZE];
697     unsigned int md_len;
698
699     /* Add signing time if not already present */
700     if (!PKCS7_get_signed_attribute(si, NID_pkcs9_signingTime)) {
701         if (!PKCS7_add0_attrib_signing_time(si, NULL)) {
702             PKCS7err(PKCS7_F_DO_PKCS7_SIGNED_ATTRIB, ERR_R_MALLOC_FAILURE);
703             return 0;
704         }
705     }
706
707     /* Add digest */
708     if (!EVP_DigestFinal_ex(mctx, md_data, &md_len)) {
709         PKCS7err(PKCS7_F_DO_PKCS7_SIGNED_ATTRIB, ERR_R_EVP_LIB);
710         return 0;
711     }
712     if (!PKCS7_add1_attrib_digest(si, md_data, md_len)) {
713         PKCS7err(PKCS7_F_DO_PKCS7_SIGNED_ATTRIB, ERR_R_MALLOC_FAILURE);
714         return 0;
715     }
716
717     /* Now sign the attributes */
718     if (!PKCS7_SIGNER_INFO_sign(si))
719         return 0;
720
721     return 1;
722 }
723
724 int PKCS7_dataFinal(PKCS7 *p7, BIO *bio)
725 {
726     int ret = 0;
727     int i, j;
728     BIO *btmp;
729     PKCS7_SIGNER_INFO *si;
730     EVP_MD_CTX *mdc, *ctx_tmp;
731     STACK_OF(X509_ATTRIBUTE) *sk;
732     STACK_OF(PKCS7_SIGNER_INFO) *si_sk = NULL;
733     ASN1_OCTET_STRING *os = NULL;
734     const PKCS7_CTX *p7_ctx;
735
736     if (p7 == NULL) {
737         PKCS7err(PKCS7_F_PKCS7_DATAFINAL, PKCS7_R_INVALID_NULL_POINTER);
738         return 0;
739     }
740
741     p7_ctx = pkcs7_get0_ctx(p7);
742
743     if (p7->d.ptr == NULL) {
744         PKCS7err(PKCS7_F_PKCS7_DATAFINAL, PKCS7_R_NO_CONTENT);
745         return 0;
746     }
747
748     ctx_tmp = EVP_MD_CTX_new();
749     if (ctx_tmp == NULL) {
750         PKCS7err(PKCS7_F_PKCS7_DATAFINAL, ERR_R_MALLOC_FAILURE);
751         return 0;
752     }
753
754     i = OBJ_obj2nid(p7->type);
755     p7->state = PKCS7_S_HEADER;
756
757     switch (i) {
758     case NID_pkcs7_data:
759         os = p7->d.data;
760         break;
761     case NID_pkcs7_signedAndEnveloped:
762         /* XXXXXXXXXXXXXXXX */
763         si_sk = p7->d.signed_and_enveloped->signer_info;
764         os = p7->d.signed_and_enveloped->enc_data->enc_data;
765         if (os == NULL) {
766             os = ASN1_OCTET_STRING_new();
767             if (os == NULL) {
768                 PKCS7err(PKCS7_F_PKCS7_DATAFINAL, ERR_R_MALLOC_FAILURE);
769                 goto err;
770             }
771             p7->d.signed_and_enveloped->enc_data->enc_data = os;
772         }
773         break;
774     case NID_pkcs7_enveloped:
775         /* XXXXXXXXXXXXXXXX */
776         os = p7->d.enveloped->enc_data->enc_data;
777         if (os == NULL) {
778             os = ASN1_OCTET_STRING_new();
779             if (os == NULL) {
780                 PKCS7err(PKCS7_F_PKCS7_DATAFINAL, ERR_R_MALLOC_FAILURE);
781                 goto err;
782             }
783             p7->d.enveloped->enc_data->enc_data = os;
784         }
785         break;
786     case NID_pkcs7_signed:
787         si_sk = p7->d.sign->signer_info;
788         os = PKCS7_get_octet_string(p7->d.sign->contents);
789         /* If detached data then the content is excluded */
790         if (PKCS7_type_is_data(p7->d.sign->contents) && p7->detached) {
791             ASN1_OCTET_STRING_free(os);
792             os = NULL;
793             p7->d.sign->contents->d.data = NULL;
794         }
795         break;
796
797     case NID_pkcs7_digest:
798         os = PKCS7_get_octet_string(p7->d.digest->contents);
799         /* If detached data then the content is excluded */
800         if (PKCS7_type_is_data(p7->d.digest->contents) && p7->detached) {
801             ASN1_OCTET_STRING_free(os);
802             os = NULL;
803             p7->d.digest->contents->d.data = NULL;
804         }
805         break;
806
807     default:
808         PKCS7err(PKCS7_F_PKCS7_DATAFINAL, PKCS7_R_UNSUPPORTED_CONTENT_TYPE);
809         goto err;
810     }
811
812     if (si_sk != NULL) {
813         for (i = 0; i < sk_PKCS7_SIGNER_INFO_num(si_sk); i++) {
814             si = sk_PKCS7_SIGNER_INFO_value(si_sk, i);
815             if (si->pkey == NULL)
816                 continue;
817
818             j = OBJ_obj2nid(si->digest_alg->algorithm);
819
820             btmp = bio;
821
822             btmp = PKCS7_find_digest(&mdc, btmp, j);
823
824             if (btmp == NULL)
825                 goto err;
826
827             /*
828              * We now have the EVP_MD_CTX, lets do the signing.
829              */
830             if (!EVP_MD_CTX_copy_ex(ctx_tmp, mdc))
831                 goto err;
832
833             sk = si->auth_attr;
834
835             /*
836              * If there are attributes, we add the digest attribute and only
837              * sign the attributes
838              */
839             if (sk_X509_ATTRIBUTE_num(sk) > 0) {
840                 if (!do_pkcs7_signed_attrib(si, ctx_tmp))
841                     goto err;
842             } else {
843                 unsigned char *abuf = NULL;
844                 unsigned int abuflen;
845                 abuflen = EVP_PKEY_size(si->pkey);
846                 abuf = OPENSSL_malloc(abuflen);
847                 if (abuf == NULL)
848                     goto err;
849
850                 if (!EVP_SignFinal_ex(ctx_tmp, abuf, &abuflen, si->pkey,
851                                       p7_ctx->libctx, p7_ctx->propq)) {
852                     OPENSSL_free(abuf);
853                     PKCS7err(PKCS7_F_PKCS7_DATAFINAL, ERR_R_EVP_LIB);
854                     goto err;
855                 }
856                 ASN1_STRING_set0(si->enc_digest, abuf, abuflen);
857             }
858         }
859     } else if (i == NID_pkcs7_digest) {
860         unsigned char md_data[EVP_MAX_MD_SIZE];
861         unsigned int md_len;
862         if (!PKCS7_find_digest(&mdc, bio,
863                                OBJ_obj2nid(p7->d.digest->md->algorithm)))
864             goto err;
865         if (!EVP_DigestFinal_ex(mdc, md_data, &md_len))
866             goto err;
867         if (!ASN1_OCTET_STRING_set(p7->d.digest->digest, md_data, md_len))
868             goto err;
869     }
870
871     if (!PKCS7_is_detached(p7)) {
872         /*
873          * NOTE(emilia): I think we only reach os == NULL here because detached
874          * digested data support is broken.
875          */
876         if (os == NULL)
877             goto err;
878         if (!(os->flags & ASN1_STRING_FLAG_NDEF)) {
879             char *cont;
880             long contlen;
881             btmp = BIO_find_type(bio, BIO_TYPE_MEM);
882             if (btmp == NULL) {
883                 PKCS7err(PKCS7_F_PKCS7_DATAFINAL, PKCS7_R_UNABLE_TO_FIND_MEM_BIO);
884                 goto err;
885             }
886             contlen = BIO_get_mem_data(btmp, &cont);
887             /*
888              * Mark the BIO read only then we can use its copy of the data
889              * instead of making an extra copy.
890              */
891             BIO_set_flags(btmp, BIO_FLAGS_MEM_RDONLY);
892             BIO_set_mem_eof_return(btmp, 0);
893             ASN1_STRING_set0(os, (unsigned char *)cont, contlen);
894         }
895     }
896     ret = 1;
897  err:
898     EVP_MD_CTX_free(ctx_tmp);
899     return ret;
900 }
901
902 int PKCS7_SIGNER_INFO_sign(PKCS7_SIGNER_INFO *si)
903 {
904     EVP_MD_CTX *mctx;
905     EVP_PKEY_CTX *pctx = NULL;
906     unsigned char *abuf = NULL;
907     int alen;
908     size_t siglen;
909     const EVP_MD *md = NULL;
910     const PKCS7_CTX *ctx = si->ctx;
911
912     md = EVP_get_digestbyobj(si->digest_alg->algorithm);
913     if (md == NULL)
914         return 0;
915
916     mctx = EVP_MD_CTX_new();
917     if (mctx == NULL) {
918         PKCS7err(0, ERR_R_MALLOC_FAILURE);
919         goto err;
920     }
921
922     if (EVP_DigestSignInit_ex(mctx, &pctx, EVP_MD_name(md), ctx->libctx,
923                               ctx->propq, si->pkey) <= 0)
924         goto err;
925
926     /*
927      * TODO(3.0): This causes problems when providers are in use, so disabled
928      * for now. Can we get rid of this completely? AFAICT this ctrl has never
929      * been used since it was first put in. All internal implementations just
930      * return 1 and ignore this ctrl and have always done so by the looks of
931      * things. To fix this we could convert this ctrl into a param, which would
932      * require us to send all the signer info data as a set of params...but that
933      * is non-trivial and since this isn't used by anything it may be better
934      * just to remove it. The original commit that added it had this
935      * justification in CHANGES:
936      *
937      * "During PKCS7 signing pass the PKCS7 SignerInfo structure to the
938      *  EVP_PKEY_METHOD before and after signing via the
939      *  EVP_PKEY_CTRL_PKCS7_SIGN ctrl. It can then customise the structure
940      *  before and/or after signing if necessary."
941      */
942 #if 0
943     if (EVP_PKEY_CTX_ctrl(pctx, -1, EVP_PKEY_OP_SIGN,
944                           EVP_PKEY_CTRL_PKCS7_SIGN, 0, si) <= 0) {
945         PKCS7err(0, PKCS7_R_CTRL_ERROR);
946         goto err;
947     }
948 #endif
949
950     alen = ASN1_item_i2d((ASN1_VALUE *)si->auth_attr, &abuf,
951                          ASN1_ITEM_rptr(PKCS7_ATTR_SIGN));
952     if (!abuf)
953         goto err;
954     if (EVP_DigestSignUpdate(mctx, abuf, alen) <= 0)
955         goto err;
956     OPENSSL_free(abuf);
957     abuf = NULL;
958     if (EVP_DigestSignFinal(mctx, NULL, &siglen) <= 0)
959         goto err;
960     abuf = OPENSSL_malloc(siglen);
961     if (abuf == NULL)
962         goto err;
963     if (EVP_DigestSignFinal(mctx, abuf, &siglen) <= 0)
964         goto err;
965
966     /*
967      * TODO(3.0): This causes problems when providers are in use, so disabled
968      * for now. Can we get rid of this completely? AFAICT this ctrl has never
969      * been used since it was first put in. All internal implementations just
970      * return 1 and ignore this ctrl and have always done so by the looks of
971      * things. To fix this we could convert this ctrl into a param, which would
972      * require us to send all the signer info data as a set of params...but that
973      * is non-trivial and since this isn't used by anything it may be better
974      * just to remove it. The original commit that added it had this
975      * justification in CHANGES:
976      *
977      * "During PKCS7 signing pass the PKCS7 SignerInfo structure to the
978      *  EVP_PKEY_METHOD before and after signing via the
979      *  EVP_PKEY_CTRL_PKCS7_SIGN ctrl. It can then customise the structure
980      *  before and/or after signing if necessary."
981      */
982 #if 0
983     if (EVP_PKEY_CTX_ctrl(pctx, -1, EVP_PKEY_OP_SIGN,
984                           EVP_PKEY_CTRL_PKCS7_SIGN, 1, si) <= 0) {
985         PKCS7err(0, PKCS7_R_CTRL_ERROR);
986         goto err;
987     }
988 #endif
989
990     EVP_MD_CTX_free(mctx);
991
992     ASN1_STRING_set0(si->enc_digest, abuf, siglen);
993
994     return 1;
995
996  err:
997     OPENSSL_free(abuf);
998     EVP_MD_CTX_free(mctx);
999     return 0;
1000 }
1001
1002 int PKCS7_dataVerify(X509_STORE *cert_store, X509_STORE_CTX *ctx, BIO *bio,
1003                      PKCS7 *p7, PKCS7_SIGNER_INFO *si)
1004 {
1005     PKCS7_ISSUER_AND_SERIAL *ias;
1006     int ret = 0, i;
1007     STACK_OF(X509) *cert;
1008     X509 *x509;
1009
1010     if (p7 == NULL) {
1011         PKCS7err(PKCS7_F_PKCS7_DATAVERIFY, PKCS7_R_INVALID_NULL_POINTER);
1012         return 0;
1013     }
1014
1015     if (p7->d.ptr == NULL) {
1016         PKCS7err(PKCS7_F_PKCS7_DATAVERIFY, PKCS7_R_NO_CONTENT);
1017         return 0;
1018     }
1019
1020     if (PKCS7_type_is_signed(p7)) {
1021         cert = p7->d.sign->cert;
1022     } else if (PKCS7_type_is_signedAndEnveloped(p7)) {
1023         cert = p7->d.signed_and_enveloped->cert;
1024     } else {
1025         PKCS7err(PKCS7_F_PKCS7_DATAVERIFY, PKCS7_R_WRONG_PKCS7_TYPE);
1026         goto err;
1027     }
1028     /* XXXXXXXXXXXXXXXXXXXXXXX */
1029     ias = si->issuer_and_serial;
1030
1031     x509 = X509_find_by_issuer_and_serial(cert, ias->issuer, ias->serial);
1032
1033     /* were we able to find the cert in passed to us */
1034     if (x509 == NULL) {
1035         PKCS7err(PKCS7_F_PKCS7_DATAVERIFY,
1036                  PKCS7_R_UNABLE_TO_FIND_CERTIFICATE);
1037         goto err;
1038     }
1039
1040     /* Lets verify */
1041     if (!X509_STORE_CTX_init(ctx, cert_store, x509, cert)) {
1042         PKCS7err(PKCS7_F_PKCS7_DATAVERIFY, ERR_R_X509_LIB);
1043         goto err;
1044     }
1045     X509_STORE_CTX_set_purpose(ctx, X509_PURPOSE_SMIME_SIGN);
1046     i = X509_verify_cert(ctx);
1047     if (i <= 0) {
1048         PKCS7err(PKCS7_F_PKCS7_DATAVERIFY, ERR_R_X509_LIB);
1049         X509_STORE_CTX_cleanup(ctx);
1050         goto err;
1051     }
1052     X509_STORE_CTX_cleanup(ctx);
1053
1054     return PKCS7_signatureVerify(bio, p7, si, x509);
1055  err:
1056     return ret;
1057 }
1058
1059 int PKCS7_signatureVerify(BIO *bio, PKCS7 *p7, PKCS7_SIGNER_INFO *si,
1060                           X509 *x509)
1061 {
1062     ASN1_OCTET_STRING *os;
1063     EVP_MD_CTX *mdc_tmp, *mdc;
1064     const EVP_MD *md;
1065     EVP_MD *fetched_md = NULL;
1066     int ret = 0, i;
1067     int md_type;
1068     STACK_OF(X509_ATTRIBUTE) *sk;
1069     BIO *btmp;
1070     EVP_PKEY *pkey;
1071     const PKCS7_CTX *ctx = pkcs7_get0_ctx(p7);
1072
1073     mdc_tmp = EVP_MD_CTX_new();
1074     if (mdc_tmp == NULL) {
1075         PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY, ERR_R_MALLOC_FAILURE);
1076         goto err;
1077     }
1078
1079     if (!PKCS7_type_is_signed(p7) && !PKCS7_type_is_signedAndEnveloped(p7)) {
1080         PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY, PKCS7_R_WRONG_PKCS7_TYPE);
1081         goto err;
1082     }
1083
1084     md_type = OBJ_obj2nid(si->digest_alg->algorithm);
1085
1086     btmp = bio;
1087     for (;;) {
1088         if ((btmp == NULL) ||
1089             ((btmp = BIO_find_type(btmp, BIO_TYPE_MD)) == NULL)) {
1090             PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY,
1091                      PKCS7_R_UNABLE_TO_FIND_MESSAGE_DIGEST);
1092             goto err;
1093         }
1094         BIO_get_md_ctx(btmp, &mdc);
1095         if (mdc == NULL) {
1096             PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY, ERR_R_INTERNAL_ERROR);
1097             goto err;
1098         }
1099         if (EVP_MD_CTX_type(mdc) == md_type)
1100             break;
1101         /*
1102          * Workaround for some broken clients that put the signature OID
1103          * instead of the digest OID in digest_alg->algorithm
1104          */
1105         if (EVP_MD_pkey_type(EVP_MD_CTX_md(mdc)) == md_type)
1106             break;
1107         btmp = BIO_next(btmp);
1108     }
1109
1110     /*
1111      * mdc is the digest ctx that we want, unless there are attributes, in
1112      * which case the digest is the signed attributes
1113      */
1114     if (!EVP_MD_CTX_copy_ex(mdc_tmp, mdc))
1115         goto err;
1116
1117     sk = si->auth_attr;
1118     if ((sk != NULL) && (sk_X509_ATTRIBUTE_num(sk) != 0)) {
1119         unsigned char md_dat[EVP_MAX_MD_SIZE], *abuf = NULL;
1120         unsigned int md_len;
1121         int alen;
1122         ASN1_OCTET_STRING *message_digest;
1123
1124         if (!EVP_DigestFinal_ex(mdc_tmp, md_dat, &md_len))
1125             goto err;
1126         message_digest = PKCS7_digest_from_attributes(sk);
1127         if (!message_digest) {
1128             PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY,
1129                      PKCS7_R_UNABLE_TO_FIND_MESSAGE_DIGEST);
1130             goto err;
1131         }
1132         if ((message_digest->length != (int)md_len) ||
1133             (memcmp(message_digest->data, md_dat, md_len))) {
1134             PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY, PKCS7_R_DIGEST_FAILURE);
1135             ret = -1;
1136             goto err;
1137         }
1138
1139         (void)ERR_set_mark();
1140         fetched_md = EVP_MD_fetch(ctx->libctx, OBJ_nid2sn(md_type), ctx->propq);
1141
1142         if (fetched_md != NULL)
1143             md = fetched_md;
1144         else
1145             md = EVP_get_digestbynid(md_type);
1146
1147         if (md == NULL || !EVP_VerifyInit_ex(mdc_tmp, md, NULL)) {
1148             (void)ERR_clear_last_mark();
1149             goto err;
1150         }
1151         (void)ERR_pop_to_mark();
1152
1153         alen = ASN1_item_i2d((ASN1_VALUE *)sk, &abuf,
1154                              ASN1_ITEM_rptr(PKCS7_ATTR_VERIFY));
1155         if (alen <= 0) {
1156             PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY, ERR_R_ASN1_LIB);
1157             ret = -1;
1158             goto err;
1159         }
1160         if (!EVP_VerifyUpdate(mdc_tmp, abuf, alen))
1161             goto err;
1162
1163         OPENSSL_free(abuf);
1164     }
1165
1166     os = si->enc_digest;
1167     pkey = X509_get0_pubkey(x509);
1168     if (pkey == NULL) {
1169         ret = -1;
1170         goto err;
1171     }
1172
1173     i = EVP_VerifyFinal_ex(mdc_tmp, os->data, os->length, pkey, ctx->libctx,
1174                            ctx->propq);
1175     if (i <= 0) {
1176         PKCS7err(PKCS7_F_PKCS7_SIGNATUREVERIFY, PKCS7_R_SIGNATURE_FAILURE);
1177         ret = -1;
1178         goto err;
1179     }
1180     ret = 1;
1181  err:
1182     EVP_MD_CTX_free(mdc_tmp);
1183     EVP_MD_free(fetched_md);
1184     return ret;
1185 }
1186
1187 PKCS7_ISSUER_AND_SERIAL *PKCS7_get_issuer_and_serial(PKCS7 *p7, int idx)
1188 {
1189     STACK_OF(PKCS7_RECIP_INFO) *rsk;
1190     PKCS7_RECIP_INFO *ri;
1191     int i;
1192
1193     i = OBJ_obj2nid(p7->type);
1194     if (i != NID_pkcs7_signedAndEnveloped)
1195         return NULL;
1196     if (p7->d.signed_and_enveloped == NULL)
1197         return NULL;
1198     rsk = p7->d.signed_and_enveloped->recipientinfo;
1199     if (rsk == NULL)
1200         return NULL;
1201     if (sk_PKCS7_RECIP_INFO_num(rsk) <= idx)
1202         return NULL;
1203     ri = sk_PKCS7_RECIP_INFO_value(rsk, idx);
1204     return ri->issuer_and_serial;
1205 }
1206
1207 ASN1_TYPE *PKCS7_get_signed_attribute(PKCS7_SIGNER_INFO *si, int nid)
1208 {
1209     return get_attribute(si->auth_attr, nid);
1210 }
1211
1212 ASN1_TYPE *PKCS7_get_attribute(PKCS7_SIGNER_INFO *si, int nid)
1213 {
1214     return get_attribute(si->unauth_attr, nid);
1215 }
1216
1217 static ASN1_TYPE *get_attribute(STACK_OF(X509_ATTRIBUTE) *sk, int nid)
1218 {
1219     int idx;
1220     X509_ATTRIBUTE *xa;
1221     idx = X509at_get_attr_by_NID(sk, nid, -1);
1222     xa = X509at_get_attr(sk, idx);
1223     return X509_ATTRIBUTE_get0_type(xa, 0);
1224 }
1225
1226 ASN1_OCTET_STRING *PKCS7_digest_from_attributes(STACK_OF(X509_ATTRIBUTE) *sk)
1227 {
1228     ASN1_TYPE *astype;
1229     if ((astype = get_attribute(sk, NID_pkcs9_messageDigest)) == NULL)
1230         return NULL;
1231     return astype->value.octet_string;
1232 }
1233
1234 int PKCS7_set_signed_attributes(PKCS7_SIGNER_INFO *p7si,
1235                                 STACK_OF(X509_ATTRIBUTE) *sk)
1236 {
1237     int i;
1238
1239     sk_X509_ATTRIBUTE_pop_free(p7si->auth_attr, X509_ATTRIBUTE_free);
1240     p7si->auth_attr = sk_X509_ATTRIBUTE_dup(sk);
1241     if (p7si->auth_attr == NULL)
1242         return 0;
1243     for (i = 0; i < sk_X509_ATTRIBUTE_num(sk); i++) {
1244         if ((sk_X509_ATTRIBUTE_set(p7si->auth_attr, i,
1245                                    X509_ATTRIBUTE_dup(sk_X509_ATTRIBUTE_value
1246                                                       (sk, i))))
1247             == NULL)
1248             return 0;
1249     }
1250     return 1;
1251 }
1252
1253 int PKCS7_set_attributes(PKCS7_SIGNER_INFO *p7si,
1254                          STACK_OF(X509_ATTRIBUTE) *sk)
1255 {
1256     int i;
1257
1258     sk_X509_ATTRIBUTE_pop_free(p7si->unauth_attr, X509_ATTRIBUTE_free);
1259     p7si->unauth_attr = sk_X509_ATTRIBUTE_dup(sk);
1260     if (p7si->unauth_attr == NULL)
1261         return 0;
1262     for (i = 0; i < sk_X509_ATTRIBUTE_num(sk); i++) {
1263         if ((sk_X509_ATTRIBUTE_set(p7si->unauth_attr, i,
1264                                    X509_ATTRIBUTE_dup(sk_X509_ATTRIBUTE_value
1265                                                       (sk, i))))
1266             == NULL)
1267             return 0;
1268     }
1269     return 1;
1270 }
1271
1272 int PKCS7_add_signed_attribute(PKCS7_SIGNER_INFO *p7si, int nid, int atrtype,
1273                                void *value)
1274 {
1275     return add_attribute(&(p7si->auth_attr), nid, atrtype, value);
1276 }
1277
1278 int PKCS7_add_attribute(PKCS7_SIGNER_INFO *p7si, int nid, int atrtype,
1279                         void *value)
1280 {
1281     return add_attribute(&(p7si->unauth_attr), nid, atrtype, value);
1282 }
1283
1284 static int add_attribute(STACK_OF(X509_ATTRIBUTE) **sk, int nid, int atrtype,
1285                          void *value)
1286 {
1287     X509_ATTRIBUTE *attr = NULL;
1288
1289     if (*sk == NULL) {
1290         if ((*sk = sk_X509_ATTRIBUTE_new_null()) == NULL)
1291             return 0;
1292  new_attrib:
1293         if ((attr = X509_ATTRIBUTE_create(nid, atrtype, value)) == NULL)
1294             return 0;
1295         if (!sk_X509_ATTRIBUTE_push(*sk, attr)) {
1296             X509_ATTRIBUTE_free(attr);
1297             return 0;
1298         }
1299     } else {
1300         int i;
1301
1302         for (i = 0; i < sk_X509_ATTRIBUTE_num(*sk); i++) {
1303             attr = sk_X509_ATTRIBUTE_value(*sk, i);
1304             if (OBJ_obj2nid(X509_ATTRIBUTE_get0_object(attr)) == nid) {
1305                 X509_ATTRIBUTE_free(attr);
1306                 attr = X509_ATTRIBUTE_create(nid, atrtype, value);
1307                 if (attr == NULL)
1308                     return 0;
1309                 if (!sk_X509_ATTRIBUTE_set(*sk, i, attr)) {
1310                     X509_ATTRIBUTE_free(attr);
1311                     return 0;
1312                 }
1313                 goto end;
1314             }
1315         }
1316         goto new_attrib;
1317     }
1318  end:
1319     return 1;
1320 }