xref: /freebsd/crypto/openssl/crypto/cms/cms_env.c (revision 681ce946)
1 /*
2  * Copyright 2008-2021 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the OpenSSL license (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 "internal/cryptlib.h"
11 #include <openssl/asn1t.h>
12 #include <openssl/pem.h>
13 #include <openssl/x509v3.h>
14 #include <openssl/err.h>
15 #include <openssl/cms.h>
16 #include <openssl/aes.h>
17 #include "cms_local.h"
18 #include "crypto/asn1.h"
19 #include "crypto/evp.h"
20 
21 /* CMS EnvelopedData Utilities */
22 
23 CMS_EnvelopedData *cms_get0_enveloped(CMS_ContentInfo *cms)
24 {
25     if (OBJ_obj2nid(cms->contentType) != NID_pkcs7_enveloped) {
26         CMSerr(CMS_F_CMS_GET0_ENVELOPED,
27                CMS_R_CONTENT_TYPE_NOT_ENVELOPED_DATA);
28         return NULL;
29     }
30     return cms->d.envelopedData;
31 }
32 
33 static CMS_EnvelopedData *cms_enveloped_data_init(CMS_ContentInfo *cms)
34 {
35     if (cms->d.other == NULL) {
36         cms->d.envelopedData = M_ASN1_new_of(CMS_EnvelopedData);
37         if (!cms->d.envelopedData) {
38             CMSerr(CMS_F_CMS_ENVELOPED_DATA_INIT, ERR_R_MALLOC_FAILURE);
39             return NULL;
40         }
41         cms->d.envelopedData->version = 0;
42         cms->d.envelopedData->encryptedContentInfo->contentType =
43             OBJ_nid2obj(NID_pkcs7_data);
44         ASN1_OBJECT_free(cms->contentType);
45         cms->contentType = OBJ_nid2obj(NID_pkcs7_enveloped);
46         return cms->d.envelopedData;
47     }
48     return cms_get0_enveloped(cms);
49 }
50 
51 int cms_env_asn1_ctrl(CMS_RecipientInfo *ri, int cmd)
52 {
53     EVP_PKEY *pkey;
54     int i;
55     if (ri->type == CMS_RECIPINFO_TRANS)
56         pkey = ri->d.ktri->pkey;
57     else if (ri->type == CMS_RECIPINFO_AGREE) {
58         EVP_PKEY_CTX *pctx = ri->d.kari->pctx;
59         if (!pctx)
60             return 0;
61         pkey = EVP_PKEY_CTX_get0_pkey(pctx);
62         if (!pkey)
63             return 0;
64     } else
65         return 0;
66     if (!pkey->ameth || !pkey->ameth->pkey_ctrl)
67         return 1;
68     i = pkey->ameth->pkey_ctrl(pkey, ASN1_PKEY_CTRL_CMS_ENVELOPE, cmd, ri);
69     if (i == -2) {
70         CMSerr(CMS_F_CMS_ENV_ASN1_CTRL,
71                CMS_R_NOT_SUPPORTED_FOR_THIS_KEY_TYPE);
72         return 0;
73     }
74     if (i <= 0) {
75         CMSerr(CMS_F_CMS_ENV_ASN1_CTRL, CMS_R_CTRL_FAILURE);
76         return 0;
77     }
78     return 1;
79 }
80 
81 STACK_OF(CMS_RecipientInfo) *CMS_get0_RecipientInfos(CMS_ContentInfo *cms)
82 {
83     CMS_EnvelopedData *env;
84     env = cms_get0_enveloped(cms);
85     if (!env)
86         return NULL;
87     return env->recipientInfos;
88 }
89 
90 int CMS_RecipientInfo_type(CMS_RecipientInfo *ri)
91 {
92     return ri->type;
93 }
94 
95 EVP_PKEY_CTX *CMS_RecipientInfo_get0_pkey_ctx(CMS_RecipientInfo *ri)
96 {
97     if (ri->type == CMS_RECIPINFO_TRANS)
98         return ri->d.ktri->pctx;
99     else if (ri->type == CMS_RECIPINFO_AGREE)
100         return ri->d.kari->pctx;
101     return NULL;
102 }
103 
104 CMS_ContentInfo *CMS_EnvelopedData_create(const EVP_CIPHER *cipher)
105 {
106     CMS_ContentInfo *cms;
107     CMS_EnvelopedData *env;
108     cms = CMS_ContentInfo_new();
109     if (cms == NULL)
110         goto merr;
111     env = cms_enveloped_data_init(cms);
112     if (env == NULL)
113         goto merr;
114     if (!cms_EncryptedContent_init(env->encryptedContentInfo,
115                                    cipher, NULL, 0))
116         goto merr;
117     return cms;
118  merr:
119     CMS_ContentInfo_free(cms);
120     CMSerr(CMS_F_CMS_ENVELOPEDDATA_CREATE, ERR_R_MALLOC_FAILURE);
121     return NULL;
122 }
123 
124 /* Key Transport Recipient Info (KTRI) routines */
125 
126 /* Initialise a ktri based on passed certificate and key */
127 
128 static int cms_RecipientInfo_ktri_init(CMS_RecipientInfo *ri, X509 *recip,
129                                        EVP_PKEY *pk, unsigned int flags)
130 {
131     CMS_KeyTransRecipientInfo *ktri;
132     int idtype;
133 
134     ri->d.ktri = M_ASN1_new_of(CMS_KeyTransRecipientInfo);
135     if (!ri->d.ktri)
136         return 0;
137     ri->type = CMS_RECIPINFO_TRANS;
138 
139     ktri = ri->d.ktri;
140 
141     if (flags & CMS_USE_KEYID) {
142         ktri->version = 2;
143         idtype = CMS_RECIPINFO_KEYIDENTIFIER;
144     } else {
145         ktri->version = 0;
146         idtype = CMS_RECIPINFO_ISSUER_SERIAL;
147     }
148 
149     /*
150      * Not a typo: RecipientIdentifier and SignerIdentifier are the same
151      * structure.
152      */
153 
154     if (!cms_set1_SignerIdentifier(ktri->rid, recip, idtype))
155         return 0;
156 
157     X509_up_ref(recip);
158     EVP_PKEY_up_ref(pk);
159 
160     ktri->pkey = pk;
161     ktri->recip = recip;
162 
163     if (flags & CMS_KEY_PARAM) {
164         ktri->pctx = EVP_PKEY_CTX_new(ktri->pkey, NULL);
165         if (ktri->pctx == NULL)
166             return 0;
167         if (EVP_PKEY_encrypt_init(ktri->pctx) <= 0)
168             return 0;
169     } else if (!cms_env_asn1_ctrl(ri, 0))
170         return 0;
171     return 1;
172 }
173 
174 /*
175  * Add a recipient certificate using appropriate type of RecipientInfo
176  */
177 
178 CMS_RecipientInfo *CMS_add1_recipient_cert(CMS_ContentInfo *cms,
179                                            X509 *recip, unsigned int flags)
180 {
181     CMS_RecipientInfo *ri = NULL;
182     CMS_EnvelopedData *env;
183     EVP_PKEY *pk = NULL;
184     env = cms_get0_enveloped(cms);
185     if (!env)
186         goto err;
187 
188     /* Initialize recipient info */
189     ri = M_ASN1_new_of(CMS_RecipientInfo);
190     if (!ri)
191         goto merr;
192 
193     pk = X509_get0_pubkey(recip);
194     if (!pk) {
195         CMSerr(CMS_F_CMS_ADD1_RECIPIENT_CERT, CMS_R_ERROR_GETTING_PUBLIC_KEY);
196         goto err;
197     }
198 
199     switch (cms_pkey_get_ri_type(pk)) {
200 
201     case CMS_RECIPINFO_TRANS:
202         if (!cms_RecipientInfo_ktri_init(ri, recip, pk, flags))
203             goto err;
204         break;
205 
206     case CMS_RECIPINFO_AGREE:
207         if (!cms_RecipientInfo_kari_init(ri, recip, pk, flags))
208             goto err;
209         break;
210 
211     default:
212         CMSerr(CMS_F_CMS_ADD1_RECIPIENT_CERT,
213                CMS_R_NOT_SUPPORTED_FOR_THIS_KEY_TYPE);
214         goto err;
215 
216     }
217 
218     if (!sk_CMS_RecipientInfo_push(env->recipientInfos, ri))
219         goto merr;
220 
221     return ri;
222 
223  merr:
224     CMSerr(CMS_F_CMS_ADD1_RECIPIENT_CERT, ERR_R_MALLOC_FAILURE);
225  err:
226     M_ASN1_free_of(ri, CMS_RecipientInfo);
227     return NULL;
228 
229 }
230 
231 int CMS_RecipientInfo_ktri_get0_algs(CMS_RecipientInfo *ri,
232                                      EVP_PKEY **pk, X509 **recip,
233                                      X509_ALGOR **palg)
234 {
235     CMS_KeyTransRecipientInfo *ktri;
236     if (ri->type != CMS_RECIPINFO_TRANS) {
237         CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_GET0_ALGS,
238                CMS_R_NOT_KEY_TRANSPORT);
239         return 0;
240     }
241 
242     ktri = ri->d.ktri;
243 
244     if (pk)
245         *pk = ktri->pkey;
246     if (recip)
247         *recip = ktri->recip;
248     if (palg)
249         *palg = ktri->keyEncryptionAlgorithm;
250     return 1;
251 }
252 
253 int CMS_RecipientInfo_ktri_get0_signer_id(CMS_RecipientInfo *ri,
254                                           ASN1_OCTET_STRING **keyid,
255                                           X509_NAME **issuer,
256                                           ASN1_INTEGER **sno)
257 {
258     CMS_KeyTransRecipientInfo *ktri;
259     if (ri->type != CMS_RECIPINFO_TRANS) {
260         CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_GET0_SIGNER_ID,
261                CMS_R_NOT_KEY_TRANSPORT);
262         return 0;
263     }
264     ktri = ri->d.ktri;
265 
266     return cms_SignerIdentifier_get0_signer_id(ktri->rid, keyid, issuer, sno);
267 }
268 
269 int CMS_RecipientInfo_ktri_cert_cmp(CMS_RecipientInfo *ri, X509 *cert)
270 {
271     if (ri->type != CMS_RECIPINFO_TRANS) {
272         CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_CERT_CMP,
273                CMS_R_NOT_KEY_TRANSPORT);
274         return -2;
275     }
276     return cms_SignerIdentifier_cert_cmp(ri->d.ktri->rid, cert);
277 }
278 
279 int CMS_RecipientInfo_set0_pkey(CMS_RecipientInfo *ri, EVP_PKEY *pkey)
280 {
281     if (ri->type != CMS_RECIPINFO_TRANS) {
282         CMSerr(CMS_F_CMS_RECIPIENTINFO_SET0_PKEY, CMS_R_NOT_KEY_TRANSPORT);
283         return 0;
284     }
285     EVP_PKEY_free(ri->d.ktri->pkey);
286     ri->d.ktri->pkey = pkey;
287     return 1;
288 }
289 
290 /* Encrypt content key in key transport recipient info */
291 
292 static int cms_RecipientInfo_ktri_encrypt(CMS_ContentInfo *cms,
293                                           CMS_RecipientInfo *ri)
294 {
295     CMS_KeyTransRecipientInfo *ktri;
296     CMS_EncryptedContentInfo *ec;
297     EVP_PKEY_CTX *pctx;
298     unsigned char *ek = NULL;
299     size_t eklen;
300 
301     int ret = 0;
302 
303     if (ri->type != CMS_RECIPINFO_TRANS) {
304         CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_ENCRYPT, CMS_R_NOT_KEY_TRANSPORT);
305         return 0;
306     }
307     ktri = ri->d.ktri;
308     ec = cms->d.envelopedData->encryptedContentInfo;
309 
310     pctx = ktri->pctx;
311 
312     if (pctx) {
313         if (!cms_env_asn1_ctrl(ri, 0))
314             goto err;
315     } else {
316         pctx = EVP_PKEY_CTX_new(ktri->pkey, NULL);
317         if (pctx == NULL)
318             return 0;
319 
320         if (EVP_PKEY_encrypt_init(pctx) <= 0)
321             goto err;
322     }
323 
324     if (EVP_PKEY_CTX_ctrl(pctx, -1, EVP_PKEY_OP_ENCRYPT,
325                           EVP_PKEY_CTRL_CMS_ENCRYPT, 0, ri) <= 0) {
326         CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_ENCRYPT, CMS_R_CTRL_ERROR);
327         goto err;
328     }
329 
330     if (EVP_PKEY_encrypt(pctx, NULL, &eklen, ec->key, ec->keylen) <= 0)
331         goto err;
332 
333     ek = OPENSSL_malloc(eklen);
334 
335     if (ek == NULL) {
336         CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_ENCRYPT, ERR_R_MALLOC_FAILURE);
337         goto err;
338     }
339 
340     if (EVP_PKEY_encrypt(pctx, ek, &eklen, ec->key, ec->keylen) <= 0)
341         goto err;
342 
343     ASN1_STRING_set0(ktri->encryptedKey, ek, eklen);
344     ek = NULL;
345 
346     ret = 1;
347 
348  err:
349     EVP_PKEY_CTX_free(pctx);
350     ktri->pctx = NULL;
351     OPENSSL_free(ek);
352     return ret;
353 
354 }
355 
356 /* Decrypt content key from KTRI */
357 
358 static int cms_RecipientInfo_ktri_decrypt(CMS_ContentInfo *cms,
359                                           CMS_RecipientInfo *ri)
360 {
361     CMS_KeyTransRecipientInfo *ktri = ri->d.ktri;
362     EVP_PKEY *pkey = ktri->pkey;
363     unsigned char *ek = NULL;
364     size_t eklen;
365     int ret = 0;
366     size_t fixlen = 0;
367     CMS_EncryptedContentInfo *ec;
368     ec = cms->d.envelopedData->encryptedContentInfo;
369 
370     if (ktri->pkey == NULL) {
371         CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_DECRYPT, CMS_R_NO_PRIVATE_KEY);
372         return 0;
373     }
374 
375     if (cms->d.envelopedData->encryptedContentInfo->havenocert
376             && !cms->d.envelopedData->encryptedContentInfo->debug) {
377         X509_ALGOR *calg = ec->contentEncryptionAlgorithm;
378         const EVP_CIPHER *ciph = EVP_get_cipherbyobj(calg->algorithm);
379 
380         if (ciph == NULL) {
381             CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_DECRYPT, CMS_R_UNKNOWN_CIPHER);
382             return 0;
383         }
384 
385         fixlen = EVP_CIPHER_key_length(ciph);
386     }
387 
388     ktri->pctx = EVP_PKEY_CTX_new(pkey, NULL);
389     if (ktri->pctx == NULL)
390         return 0;
391 
392     if (EVP_PKEY_decrypt_init(ktri->pctx) <= 0)
393         goto err;
394 
395     if (!cms_env_asn1_ctrl(ri, 1))
396         goto err;
397 
398     if (EVP_PKEY_CTX_ctrl(ktri->pctx, -1, EVP_PKEY_OP_DECRYPT,
399                           EVP_PKEY_CTRL_CMS_DECRYPT, 0, ri) <= 0) {
400         CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_DECRYPT, CMS_R_CTRL_ERROR);
401         goto err;
402     }
403 
404     if (EVP_PKEY_decrypt(ktri->pctx, NULL, &eklen,
405                          ktri->encryptedKey->data,
406                          ktri->encryptedKey->length) <= 0)
407         goto err;
408 
409     ek = OPENSSL_malloc(eklen);
410 
411     if (ek == NULL) {
412         CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_DECRYPT, ERR_R_MALLOC_FAILURE);
413         goto err;
414     }
415 
416     if (EVP_PKEY_decrypt(ktri->pctx, ek, &eklen,
417                          ktri->encryptedKey->data,
418                          ktri->encryptedKey->length) <= 0
419             || eklen == 0
420             || (fixlen != 0 && eklen != fixlen)) {
421         CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_DECRYPT, CMS_R_CMS_LIB);
422         goto err;
423     }
424 
425     ret = 1;
426 
427     OPENSSL_clear_free(ec->key, ec->keylen);
428     ec->key = ek;
429     ec->keylen = eklen;
430 
431  err:
432     EVP_PKEY_CTX_free(ktri->pctx);
433     ktri->pctx = NULL;
434     if (!ret)
435         OPENSSL_free(ek);
436 
437     return ret;
438 }
439 
440 /* Key Encrypted Key (KEK) RecipientInfo routines */
441 
442 int CMS_RecipientInfo_kekri_id_cmp(CMS_RecipientInfo *ri,
443                                    const unsigned char *id, size_t idlen)
444 {
445     ASN1_OCTET_STRING tmp_os;
446     CMS_KEKRecipientInfo *kekri;
447     if (ri->type != CMS_RECIPINFO_KEK) {
448         CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_ID_CMP, CMS_R_NOT_KEK);
449         return -2;
450     }
451     kekri = ri->d.kekri;
452     tmp_os.type = V_ASN1_OCTET_STRING;
453     tmp_os.flags = 0;
454     tmp_os.data = (unsigned char *)id;
455     tmp_os.length = (int)idlen;
456     return ASN1_OCTET_STRING_cmp(&tmp_os, kekri->kekid->keyIdentifier);
457 }
458 
459 /* For now hard code AES key wrap info */
460 
461 static size_t aes_wrap_keylen(int nid)
462 {
463     switch (nid) {
464     case NID_id_aes128_wrap:
465         return 16;
466 
467     case NID_id_aes192_wrap:
468         return 24;
469 
470     case NID_id_aes256_wrap:
471         return 32;
472 
473     default:
474         return 0;
475     }
476 }
477 
478 CMS_RecipientInfo *CMS_add0_recipient_key(CMS_ContentInfo *cms, int nid,
479                                           unsigned char *key, size_t keylen,
480                                           unsigned char *id, size_t idlen,
481                                           ASN1_GENERALIZEDTIME *date,
482                                           ASN1_OBJECT *otherTypeId,
483                                           ASN1_TYPE *otherType)
484 {
485     CMS_RecipientInfo *ri = NULL;
486     CMS_EnvelopedData *env;
487     CMS_KEKRecipientInfo *kekri;
488     env = cms_get0_enveloped(cms);
489     if (!env)
490         goto err;
491 
492     if (nid == NID_undef) {
493         switch (keylen) {
494         case 16:
495             nid = NID_id_aes128_wrap;
496             break;
497 
498         case 24:
499             nid = NID_id_aes192_wrap;
500             break;
501 
502         case 32:
503             nid = NID_id_aes256_wrap;
504             break;
505 
506         default:
507             CMSerr(CMS_F_CMS_ADD0_RECIPIENT_KEY, CMS_R_INVALID_KEY_LENGTH);
508             goto err;
509         }
510 
511     } else {
512 
513         size_t exp_keylen = aes_wrap_keylen(nid);
514 
515         if (!exp_keylen) {
516             CMSerr(CMS_F_CMS_ADD0_RECIPIENT_KEY,
517                    CMS_R_UNSUPPORTED_KEK_ALGORITHM);
518             goto err;
519         }
520 
521         if (keylen != exp_keylen) {
522             CMSerr(CMS_F_CMS_ADD0_RECIPIENT_KEY, CMS_R_INVALID_KEY_LENGTH);
523             goto err;
524         }
525 
526     }
527 
528     /* Initialize recipient info */
529     ri = M_ASN1_new_of(CMS_RecipientInfo);
530     if (!ri)
531         goto merr;
532 
533     ri->d.kekri = M_ASN1_new_of(CMS_KEKRecipientInfo);
534     if (!ri->d.kekri)
535         goto merr;
536     ri->type = CMS_RECIPINFO_KEK;
537 
538     kekri = ri->d.kekri;
539 
540     if (otherTypeId) {
541         kekri->kekid->other = M_ASN1_new_of(CMS_OtherKeyAttribute);
542         if (kekri->kekid->other == NULL)
543             goto merr;
544     }
545 
546     if (!sk_CMS_RecipientInfo_push(env->recipientInfos, ri))
547         goto merr;
548 
549     /* After this point no calls can fail */
550 
551     kekri->version = 4;
552 
553     kekri->key = key;
554     kekri->keylen = keylen;
555 
556     ASN1_STRING_set0(kekri->kekid->keyIdentifier, id, idlen);
557 
558     kekri->kekid->date = date;
559 
560     if (kekri->kekid->other) {
561         kekri->kekid->other->keyAttrId = otherTypeId;
562         kekri->kekid->other->keyAttr = otherType;
563     }
564 
565     X509_ALGOR_set0(kekri->keyEncryptionAlgorithm,
566                     OBJ_nid2obj(nid), V_ASN1_UNDEF, NULL);
567 
568     return ri;
569 
570  merr:
571     CMSerr(CMS_F_CMS_ADD0_RECIPIENT_KEY, ERR_R_MALLOC_FAILURE);
572  err:
573     M_ASN1_free_of(ri, CMS_RecipientInfo);
574     return NULL;
575 
576 }
577 
578 int CMS_RecipientInfo_kekri_get0_id(CMS_RecipientInfo *ri,
579                                     X509_ALGOR **palg,
580                                     ASN1_OCTET_STRING **pid,
581                                     ASN1_GENERALIZEDTIME **pdate,
582                                     ASN1_OBJECT **potherid,
583                                     ASN1_TYPE **pothertype)
584 {
585     CMS_KEKIdentifier *rkid;
586     if (ri->type != CMS_RECIPINFO_KEK) {
587         CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_GET0_ID, CMS_R_NOT_KEK);
588         return 0;
589     }
590     rkid = ri->d.kekri->kekid;
591     if (palg)
592         *palg = ri->d.kekri->keyEncryptionAlgorithm;
593     if (pid)
594         *pid = rkid->keyIdentifier;
595     if (pdate)
596         *pdate = rkid->date;
597     if (potherid) {
598         if (rkid->other)
599             *potherid = rkid->other->keyAttrId;
600         else
601             *potherid = NULL;
602     }
603     if (pothertype) {
604         if (rkid->other)
605             *pothertype = rkid->other->keyAttr;
606         else
607             *pothertype = NULL;
608     }
609     return 1;
610 }
611 
612 int CMS_RecipientInfo_set0_key(CMS_RecipientInfo *ri,
613                                unsigned char *key, size_t keylen)
614 {
615     CMS_KEKRecipientInfo *kekri;
616     if (ri->type != CMS_RECIPINFO_KEK) {
617         CMSerr(CMS_F_CMS_RECIPIENTINFO_SET0_KEY, CMS_R_NOT_KEK);
618         return 0;
619     }
620 
621     kekri = ri->d.kekri;
622     kekri->key = key;
623     kekri->keylen = keylen;
624     return 1;
625 }
626 
627 /* Encrypt content key in KEK recipient info */
628 
629 static int cms_RecipientInfo_kekri_encrypt(CMS_ContentInfo *cms,
630                                            CMS_RecipientInfo *ri)
631 {
632     CMS_EncryptedContentInfo *ec;
633     CMS_KEKRecipientInfo *kekri;
634     AES_KEY actx;
635     unsigned char *wkey = NULL;
636     int wkeylen;
637     int r = 0;
638 
639     ec = cms->d.envelopedData->encryptedContentInfo;
640 
641     kekri = ri->d.kekri;
642 
643     if (!kekri->key) {
644         CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_ENCRYPT, CMS_R_NO_KEY);
645         return 0;
646     }
647 
648     if (AES_set_encrypt_key(kekri->key, kekri->keylen << 3, &actx)) {
649         CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_ENCRYPT,
650                CMS_R_ERROR_SETTING_KEY);
651         goto err;
652     }
653 
654     wkey = OPENSSL_malloc(ec->keylen + 8);
655 
656     if (wkey == NULL) {
657         CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_ENCRYPT, ERR_R_MALLOC_FAILURE);
658         goto err;
659     }
660 
661     wkeylen = AES_wrap_key(&actx, NULL, wkey, ec->key, ec->keylen);
662 
663     if (wkeylen <= 0) {
664         CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_ENCRYPT, CMS_R_WRAP_ERROR);
665         goto err;
666     }
667 
668     ASN1_STRING_set0(kekri->encryptedKey, wkey, wkeylen);
669 
670     r = 1;
671 
672  err:
673 
674     if (!r)
675         OPENSSL_free(wkey);
676     OPENSSL_cleanse(&actx, sizeof(actx));
677 
678     return r;
679 
680 }
681 
682 /* Decrypt content key in KEK recipient info */
683 
684 static int cms_RecipientInfo_kekri_decrypt(CMS_ContentInfo *cms,
685                                            CMS_RecipientInfo *ri)
686 {
687     CMS_EncryptedContentInfo *ec;
688     CMS_KEKRecipientInfo *kekri;
689     AES_KEY actx;
690     unsigned char *ukey = NULL;
691     int ukeylen;
692     int r = 0, wrap_nid;
693 
694     ec = cms->d.envelopedData->encryptedContentInfo;
695 
696     kekri = ri->d.kekri;
697 
698     if (!kekri->key) {
699         CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_DECRYPT, CMS_R_NO_KEY);
700         return 0;
701     }
702 
703     wrap_nid = OBJ_obj2nid(kekri->keyEncryptionAlgorithm->algorithm);
704     if (aes_wrap_keylen(wrap_nid) != kekri->keylen) {
705         CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_DECRYPT,
706                CMS_R_INVALID_KEY_LENGTH);
707         return 0;
708     }
709 
710     /* If encrypted key length is invalid don't bother */
711 
712     if (kekri->encryptedKey->length < 16) {
713         CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_DECRYPT,
714                CMS_R_INVALID_ENCRYPTED_KEY_LENGTH);
715         goto err;
716     }
717 
718     if (AES_set_decrypt_key(kekri->key, kekri->keylen << 3, &actx)) {
719         CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_DECRYPT,
720                CMS_R_ERROR_SETTING_KEY);
721         goto err;
722     }
723 
724     ukey = OPENSSL_malloc(kekri->encryptedKey->length - 8);
725 
726     if (ukey == NULL) {
727         CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_DECRYPT, ERR_R_MALLOC_FAILURE);
728         goto err;
729     }
730 
731     ukeylen = AES_unwrap_key(&actx, NULL, ukey,
732                              kekri->encryptedKey->data,
733                              kekri->encryptedKey->length);
734 
735     if (ukeylen <= 0) {
736         CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_DECRYPT, CMS_R_UNWRAP_ERROR);
737         goto err;
738     }
739 
740     OPENSSL_clear_free(ec->key, ec->keylen);
741     ec->key = ukey;
742     ec->keylen = ukeylen;
743 
744     r = 1;
745 
746  err:
747 
748     if (!r)
749         OPENSSL_free(ukey);
750     OPENSSL_cleanse(&actx, sizeof(actx));
751 
752     return r;
753 
754 }
755 
756 int CMS_RecipientInfo_decrypt(CMS_ContentInfo *cms, CMS_RecipientInfo *ri)
757 {
758     switch (ri->type) {
759     case CMS_RECIPINFO_TRANS:
760         return cms_RecipientInfo_ktri_decrypt(cms, ri);
761 
762     case CMS_RECIPINFO_KEK:
763         return cms_RecipientInfo_kekri_decrypt(cms, ri);
764 
765     case CMS_RECIPINFO_PASS:
766         return cms_RecipientInfo_pwri_crypt(cms, ri, 0);
767 
768     default:
769         CMSerr(CMS_F_CMS_RECIPIENTINFO_DECRYPT,
770                CMS_R_UNSUPPORTED_RECIPIENTINFO_TYPE);
771         return 0;
772     }
773 }
774 
775 int CMS_RecipientInfo_encrypt(CMS_ContentInfo *cms, CMS_RecipientInfo *ri)
776 {
777     switch (ri->type) {
778     case CMS_RECIPINFO_TRANS:
779         return cms_RecipientInfo_ktri_encrypt(cms, ri);
780 
781     case CMS_RECIPINFO_AGREE:
782         return cms_RecipientInfo_kari_encrypt(cms, ri);
783 
784     case CMS_RECIPINFO_KEK:
785         return cms_RecipientInfo_kekri_encrypt(cms, ri);
786 
787     case CMS_RECIPINFO_PASS:
788         return cms_RecipientInfo_pwri_crypt(cms, ri, 1);
789 
790     default:
791         CMSerr(CMS_F_CMS_RECIPIENTINFO_ENCRYPT,
792                CMS_R_UNSUPPORTED_RECIPIENT_TYPE);
793         return 0;
794     }
795 }
796 
797 /* Check structures and fixup version numbers (if necessary) */
798 
799 static void cms_env_set_originfo_version(CMS_EnvelopedData *env)
800 {
801     CMS_OriginatorInfo *org = env->originatorInfo;
802     int i;
803     if (org == NULL)
804         return;
805     for (i = 0; i < sk_CMS_CertificateChoices_num(org->certificates); i++) {
806         CMS_CertificateChoices *cch;
807         cch = sk_CMS_CertificateChoices_value(org->certificates, i);
808         if (cch->type == CMS_CERTCHOICE_OTHER) {
809             env->version = 4;
810             return;
811         } else if (cch->type == CMS_CERTCHOICE_V2ACERT) {
812             if (env->version < 3)
813                 env->version = 3;
814         }
815     }
816 
817     for (i = 0; i < sk_CMS_RevocationInfoChoice_num(org->crls); i++) {
818         CMS_RevocationInfoChoice *rch;
819         rch = sk_CMS_RevocationInfoChoice_value(org->crls, i);
820         if (rch->type == CMS_REVCHOICE_OTHER) {
821             env->version = 4;
822             return;
823         }
824     }
825 }
826 
827 static void cms_env_set_version(CMS_EnvelopedData *env)
828 {
829     int i;
830     CMS_RecipientInfo *ri;
831 
832     /*
833      * Can't set version higher than 4 so if 4 or more already nothing to do.
834      */
835     if (env->version >= 4)
836         return;
837 
838     cms_env_set_originfo_version(env);
839 
840     if (env->version >= 3)
841         return;
842 
843     for (i = 0; i < sk_CMS_RecipientInfo_num(env->recipientInfos); i++) {
844         ri = sk_CMS_RecipientInfo_value(env->recipientInfos, i);
845         if (ri->type == CMS_RECIPINFO_PASS || ri->type == CMS_RECIPINFO_OTHER) {
846             env->version = 3;
847             return;
848         } else if (ri->type != CMS_RECIPINFO_TRANS
849                    || ri->d.ktri->version != 0) {
850             env->version = 2;
851         }
852     }
853     if (env->originatorInfo || env->unprotectedAttrs)
854         env->version = 2;
855     if (env->version == 2)
856         return;
857     env->version = 0;
858 }
859 
860 BIO *cms_EnvelopedData_init_bio(CMS_ContentInfo *cms)
861 {
862     CMS_EncryptedContentInfo *ec;
863     STACK_OF(CMS_RecipientInfo) *rinfos;
864     CMS_RecipientInfo *ri;
865     int i, ok = 0;
866     BIO *ret;
867 
868     /* Get BIO first to set up key */
869 
870     ec = cms->d.envelopedData->encryptedContentInfo;
871     ret = cms_EncryptedContent_init_bio(ec);
872 
873     /* If error or no cipher end of processing */
874 
875     if (!ret || !ec->cipher)
876         return ret;
877 
878     /* Now encrypt content key according to each RecipientInfo type */
879 
880     rinfos = cms->d.envelopedData->recipientInfos;
881 
882     for (i = 0; i < sk_CMS_RecipientInfo_num(rinfos); i++) {
883         ri = sk_CMS_RecipientInfo_value(rinfos, i);
884         if (CMS_RecipientInfo_encrypt(cms, ri) <= 0) {
885             CMSerr(CMS_F_CMS_ENVELOPEDDATA_INIT_BIO,
886                    CMS_R_ERROR_SETTING_RECIPIENTINFO);
887             goto err;
888         }
889     }
890     cms_env_set_version(cms->d.envelopedData);
891 
892     ok = 1;
893 
894  err:
895     ec->cipher = NULL;
896     OPENSSL_clear_free(ec->key, ec->keylen);
897     ec->key = NULL;
898     ec->keylen = 0;
899     if (ok)
900         return ret;
901     BIO_free(ret);
902     return NULL;
903 
904 }
905 
906 /*
907  * Get RecipientInfo type (if any) supported by a key (public or private). To
908  * retain compatibility with previous behaviour if the ctrl value isn't
909  * supported we assume key transport.
910  */
911 int cms_pkey_get_ri_type(EVP_PKEY *pk)
912 {
913     if (pk->ameth && pk->ameth->pkey_ctrl) {
914         int i, r;
915         i = pk->ameth->pkey_ctrl(pk, ASN1_PKEY_CTRL_CMS_RI_TYPE, 0, &r);
916         if (i > 0)
917             return r;
918     }
919     return CMS_RECIPINFO_TRANS;
920 }
921