crypto/cms/cms_env.c

   1 /* crypto/cms/cms_env.c */
   2 /* Written by Dr Stephen N Henson (steve@openssl.org) for the OpenSSL
   3  * project.
   4  */
   5 /* ====================================================================
   6  * Copyright (c) 2008 The OpenSSL Project.  All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  *
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  *
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in
  17  *    the documentation and/or other materials provided with the
  18  *    distribution.
  19  *
  20  * 3. All advertising materials mentioning features or use of this
  21  *    software must display the following acknowledgment:
  22  *    "This product includes software developed by the OpenSSL Project
  23  *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
  24  *
  25  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
  26  *    endorse or promote products derived from this software without
  27  *    prior written permission. For written permission, please contact
  28  *    licensing@OpenSSL.org.
  29  *
  30  * 5. Products derived from this software may not be called "OpenSSL"
  31  *    nor may "OpenSSL" appear in their names without prior written
  32  *    permission of the OpenSSL Project.
  33  *
  34  * 6. Redistributions of any form whatsoever must retain the following
  35  *    acknowledgment:
  36  *    "This product includes software developed by the OpenSSL Project
  37  *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
  38  *
  39  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
  40  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  41  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  42  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
  43  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  44  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  45  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  46  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  47  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  48  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  49  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  50  * OF THE POSSIBILITY OF SUCH DAMAGE.
  51  * ====================================================================
  52  */
  53
  54 #include "cryptlib.h"
  55 #include <openssl/asn1t.h>
  56 #include <openssl/pem.h>
  57 #include <openssl/x509v3.h>
  58 #include <openssl/err.h>
  59 #include <openssl/cms.h>
  60 #include <openssl/rand.h>
  61 #include <openssl/aes.h>
  62 #include "cms_lcl.h"
  63 #include "asn1_locl.h"
  64
  65 /* CMS EnvelopedData Utilities */
  66
  67 DECLARE_ASN1_ITEM(CMS_EnvelopedData)
  68 DECLARE_ASN1_ITEM(CMS_RecipientInfo)
  69 DECLARE_ASN1_ITEM(CMS_KeyTransRecipientInfo)
  70 DECLARE_ASN1_ITEM(CMS_KEKRecipientInfo)
  71 DECLARE_ASN1_ITEM(CMS_OtherKeyAttribute)
  72
  73 DECLARE_STACK_OF(CMS_RecipientInfo)
  74
  75 static CMS_EnvelopedData *cms_get0_enveloped(CMS_ContentInfo *cms)
  76         {
  77         if (OBJ_obj2nid(cms->contentType) != NID_pkcs7_enveloped)
  78                 {
  79                 CMSerr(CMS_F_CMS_GET0_ENVELOPED,
  80                                 CMS_R_CONTENT_TYPE_NOT_ENVELOPED_DATA);
  81                 return NULL;
  82                 }
  83         return cms->d.envelopedData;
  84         }
  85
  86 static CMS_EnvelopedData *cms_enveloped_data_init(CMS_ContentInfo *cms)
  87         {
  88         if (cms->d.other == NULL)
  89                 {
  90                 cms->d.envelopedData = M_ASN1_new_of(CMS_EnvelopedData);
  91                 if (!cms->d.envelopedData)
  92                         {
  93                         CMSerr(CMS_F_CMS_ENVELOPED_DATA_INIT,
  94                                                         ERR_R_MALLOC_FAILURE);
  95                         return NULL;
  96                         }
  97                 cms->d.envelopedData->version = 0;
  98                 cms->d.envelopedData->encryptedContentInfo->contentType =
  99                                                 OBJ_nid2obj(NID_pkcs7_data);
 100                 ASN1_OBJECT_free(cms->contentType);
 101                 cms->contentType = OBJ_nid2obj(NID_pkcs7_enveloped);
 102                 return cms->d.envelopedData;
 103                 }
 104         return cms_get0_enveloped(cms);
 105         }
 106
 107 STACK_OF(CMS_RecipientInfo) *CMS_get0_RecipientInfos(CMS_ContentInfo *cms)
 108         {
 109         CMS_EnvelopedData *env;
 110         env = cms_get0_enveloped(cms);
 111         if (!env)
 112                 return NULL;
 113         return env->recipientInfos;
 114         }
 115
 116 int CMS_RecipientInfo_type(CMS_RecipientInfo *ri)
 117         {
 118         return ri->type;
 119         }
 120
 121 CMS_ContentInfo *CMS_EnvelopedData_create(const EVP_CIPHER *cipher)
 122         {
 123         CMS_ContentInfo *cms;
 124         CMS_EnvelopedData *env;
 125         cms = CMS_ContentInfo_new();
 126         if (!cms)
 127                 goto merr;
 128         env = cms_enveloped_data_init(cms);
 129         if (!env)
 130                 goto merr;
 131         if (!cms_EncryptedContent_init(env->encryptedContentInfo,
 132                                         cipher, NULL, 0))
 133                 goto merr;
 134         return cms;
 135         merr:
 136         if (cms)
 137                 CMS_ContentInfo_free(cms);
 138         CMSerr(CMS_F_CMS_ENVELOPEDDATA_CREATE, ERR_R_MALLOC_FAILURE);
 139         return NULL;
 140         }
 141
 142 /* Key Transport Recipient Info (KTRI) routines */
 143
 144 /* Add a recipient certificate. For now only handle key transport.
 145  * If we ever handle key agreement will need updating.
 146  */
 147
 148 CMS_RecipientInfo *CMS_add1_recipient_cert(CMS_ContentInfo *cms,
 149                                         X509 *recip, unsigned int flags)
 150         {
 151         CMS_RecipientInfo *ri = NULL;
 152         CMS_KeyTransRecipientInfo *ktri;
 153         CMS_EnvelopedData *env;
 154         EVP_PKEY *pk = NULL;
 155         int i, type;
 156         env = cms_get0_enveloped(cms);
 157         if (!env)
 158                 goto err;
 159
 160         /* Initialize recipient info */
 161         ri = M_ASN1_new_of(CMS_RecipientInfo);
 162         if (!ri)
 163                 goto merr;
 164
 165         /* Initialize and add key transport recipient info */
 166
 167         ri->d.ktri = M_ASN1_new_of(CMS_KeyTransRecipientInfo);
 168         if (!ri->d.ktri)
 169                 goto merr;
 170         ri->type = CMS_RECIPINFO_TRANS;
 171
 172         ktri = ri->d.ktri;
 173
 174         X509_check_purpose(recip, -1, -1);
 175         pk = X509_get_pubkey(recip);
 176         if (!pk)
 177                 {
 178                 CMSerr(CMS_F_CMS_ADD1_RECIPIENT_CERT,
 179                                 CMS_R_ERROR_GETTING_PUBLIC_KEY);
 180                 goto err;
 181                 }
 182         CRYPTO_add(&recip->references, 1, CRYPTO_LOCK_X509);
 183         ktri->pkey = pk;
 184         ktri->recip = recip;
 185
 186         if (flags & CMS_USE_KEYID)
 187                 {
 188                 ktri->version = 2;
 189                 type = CMS_RECIPINFO_KEYIDENTIFIER;
 190                 }
 191         else
 192                 {
 193                 ktri->version = 0;
 194                 type = CMS_RECIPINFO_ISSUER_SERIAL;
 195                 }
 196
 197         /* Not a typo: RecipientIdentifier and SignerIdentifier are the
 198          * same structure.
 199          */
 200
 201         if (!cms_set1_SignerIdentifier(ktri->rid, recip, type))
 202                 goto err;
 203
 204         if (pk->ameth && pk->ameth->pkey_ctrl)
 205                 {
 206                 i = pk->ameth->pkey_ctrl(pk, ASN1_PKEY_CTRL_CMS_ENVELOPE,
 207                                                 0, ri);
 208                 if (i == -2)
 209                         {
 210                         CMSerr(CMS_F_CMS_ADD1_RECIPIENT_CERT,
 211                                 CMS_R_NOT_SUPPORTED_FOR_THIS_KEY_TYPE);
 212                         goto err;
 213                         }
 214                 if (i <= 0)
 215                         {
 216                         CMSerr(CMS_F_CMS_ADD1_RECIPIENT_CERT,
 217                                 CMS_R_CTRL_FAILURE);
 218                         goto err;
 219                         }
 220                 }
 221
 222         if (!sk_CMS_RecipientInfo_push(env->recipientInfos, ri))
 223                 goto merr;
 224
 225         return ri;
 226
 227         merr:
 228         CMSerr(CMS_F_CMS_ADD1_RECIPIENT_CERT, ERR_R_MALLOC_FAILURE);
 229         err:
 230         if (ri)
 231                 M_ASN1_free_of(ri, CMS_RecipientInfo);
 232         return NULL;
 233
 234         }
 235
 236 int CMS_RecipientInfo_ktri_get0_algs(CMS_RecipientInfo *ri,
 237                                         EVP_PKEY **pk, X509 **recip,
 238                                         X509_ALGOR **palg)
 239         {
 240         CMS_KeyTransRecipientInfo *ktri;
 241         if (ri->type != CMS_RECIPINFO_TRANS)
 242                 {
 243                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_GET0_ALGS,
 244                         CMS_R_NOT_KEY_TRANSPORT);
 245                 return 0;
 246                 }
 247
 248         ktri = ri->d.ktri;
 249
 250         if (pk)
 251                 *pk = ktri->pkey;
 252         if (recip)
 253                 *recip = ktri->recip;
 254         if (palg)
 255                 *palg = ktri->keyEncryptionAlgorithm;
 256         return 1;
 257         }
 258
 259 int CMS_RecipientInfo_ktri_get0_signer_id(CMS_RecipientInfo *ri,
 260                                         ASN1_OCTET_STRING **keyid,
 261                                         X509_NAME **issuer, ASN1_INTEGER **sno)
 262         {
 263         CMS_KeyTransRecipientInfo *ktri;
 264         if (ri->type != CMS_RECIPINFO_TRANS)
 265                 {
 266                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_GET0_SIGNER_ID,
 267                         CMS_R_NOT_KEY_TRANSPORT);
 268                 return 0;
 269                 }
 270         ktri = ri->d.ktri;
 271
 272         return cms_SignerIdentifier_get0_signer_id(ktri->rid,
 273                                                         keyid, issuer, sno);
 274         }
 275
 276 int CMS_RecipientInfo_ktri_cert_cmp(CMS_RecipientInfo *ri, X509 *cert)
 277         {
 278         if (ri->type != CMS_RECIPINFO_TRANS)
 279                 {
 280                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_CERT_CMP,
 281                         CMS_R_NOT_KEY_TRANSPORT);
 282                 return -2;
 283                 }
 284         return cms_SignerIdentifier_cert_cmp(ri->d.ktri->rid, cert);
 285         }
 286
 287 int CMS_RecipientInfo_set0_pkey(CMS_RecipientInfo *ri, EVP_PKEY *pkey)
 288         {
 289         if (ri->type != CMS_RECIPINFO_TRANS)
 290                 {
 291                 CMSerr(CMS_F_CMS_RECIPIENTINFO_SET0_PKEY,
 292                         CMS_R_NOT_KEY_TRANSPORT);
 293                 return 0;
 294                 }
 295         ri->d.ktri->pkey = pkey;
 296         return 1;
 297         }
 298
 299 /* Encrypt content key in key transport recipient info */
 300
 301 static int cms_RecipientInfo_ktri_encrypt(CMS_ContentInfo *cms,
 302                                         CMS_RecipientInfo *ri)
 303         {
 304         CMS_KeyTransRecipientInfo *ktri;
 305         CMS_EncryptedContentInfo *ec;
 306         EVP_PKEY_CTX *pctx = NULL;
 307         unsigned char *ek = NULL;
 308         size_t eklen;
 309
 310         int ret = 0;
 311
 312         if (ri->type != CMS_RECIPINFO_TRANS)
 313                 {
 314                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_ENCRYPT,
 315                         CMS_R_NOT_KEY_TRANSPORT);
 316                 return 0;
 317                 }
 318         ktri = ri->d.ktri;
 319         ec = cms->d.envelopedData->encryptedContentInfo;
 320
 321         pctx = EVP_PKEY_CTX_new(ktri->pkey, NULL);
 322         if (!pctx)
 323                 return 0;
 324
 325         if (EVP_PKEY_encrypt_init(pctx) <= 0)
 326                 goto err;
 327
 328         if (EVP_PKEY_CTX_ctrl(pctx, -1, EVP_PKEY_OP_ENCRYPT,
 329                                 EVP_PKEY_CTRL_CMS_ENCRYPT, 0, ri) <= 0)
 330                 {
 331                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_ENCRYPT, CMS_R_CTRL_ERROR);
 332                 goto err;
 333                 }
 334
 335         if (EVP_PKEY_encrypt(pctx, NULL, &eklen, ec->key, ec->keylen) <= 0)
 336                 goto err;
 337
 338         ek = OPENSSL_malloc(eklen);
 339
 340         if (ek == NULL)
 341                 {
 342                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_ENCRYPT,
 343                                                         ERR_R_MALLOC_FAILURE);
 344                 goto err;
 345                 }
 346
 347         if (EVP_PKEY_encrypt(pctx, ek, &eklen, ec->key, ec->keylen) <= 0)
 348                 goto err;
 349
 350         ASN1_STRING_set0(ktri->encryptedKey, ek, eklen);
 351         ek = NULL;
 352
 353         ret = 1;
 354
 355         err:
 356         if (pctx)
 357                 EVP_PKEY_CTX_free(pctx);
 358         if (ek)
 359                 OPENSSL_free(ek);
 360         return ret;
 361
 362         }
 363
 364 /* Decrypt content key from KTRI */
 365
 366 static int cms_RecipientInfo_ktri_decrypt(CMS_ContentInfo *cms,
 367                                                         CMS_RecipientInfo *ri)
 368         {
 369         CMS_KeyTransRecipientInfo *ktri = ri->d.ktri;
 370         EVP_PKEY_CTX *pctx = NULL;
 371         unsigned char *ek = NULL;
 372         size_t eklen;
 373         int ret = 0;
 374
 375         if (ktri->pkey == NULL)
 376                 {
 377                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_DECRYPT,
 378                         CMS_R_NO_PRIVATE_KEY);
 379                 return 0;
 380                 }
 381
 382         pctx = EVP_PKEY_CTX_new(ktri->pkey, NULL);
 383         if (!pctx)
 384                 return 0;
 385
 386         if (EVP_PKEY_decrypt_init(pctx) <= 0)
 387                 goto err;
 388
 389         if (EVP_PKEY_CTX_ctrl(pctx, -1, EVP_PKEY_OP_DECRYPT,
 390                                 EVP_PKEY_CTRL_CMS_DECRYPT, 0, ri) <= 0)
 391                 {
 392                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_DECRYPT, CMS_R_CTRL_ERROR);
 393                 goto err;
 394                 }
 395
 396         if (EVP_PKEY_decrypt(pctx, NULL, &eklen,
 397                                 ktri->encryptedKey->data,
 398                                 ktri->encryptedKey->length) <= 0)
 399                 goto err;
 400
 401         ek = OPENSSL_malloc(eklen);
 402
 403         if (ek == NULL)
 404                 {
 405                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_DECRYPT,
 406                                                         ERR_R_MALLOC_FAILURE);
 407                 goto err;
 408                 }
 409
 410         if (EVP_PKEY_decrypt(pctx, ek, &eklen,
 411                                 ktri->encryptedKey->data,
 412                                 ktri->encryptedKey->length) <= 0)
 413                 {
 414                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KTRI_DECRYPT, CMS_R_CMS_LIB);
 415                 goto err;
 416                 }
 417
 418         ret = 1;
 419
 420         cms->d.envelopedData->encryptedContentInfo->key = ek;
 421         cms->d.envelopedData->encryptedContentInfo->keylen = eklen;
 422
 423         err:
 424         if (pctx)
 425                 EVP_PKEY_CTX_free(pctx);
 426         if (!ret && ek)
 427                 OPENSSL_free(ek);
 428
 429         return ret;
 430         }
 431
 432 /* Key Encrypted Key (KEK) RecipientInfo routines */
 433
 434 int CMS_RecipientInfo_kekri_id_cmp(CMS_RecipientInfo *ri,
 435                                         const unsigned char *id, size_t idlen)
 436         {
 437         ASN1_OCTET_STRING tmp_os;
 438         CMS_KEKRecipientInfo *kekri;
 439         if (ri->type != CMS_RECIPINFO_KEK)
 440                 {
 441                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_ID_CMP, CMS_R_NOT_KEK);
 442                 return -2;
 443                 }
 444         kekri = ri->d.kekri;
 445         tmp_os.type = V_ASN1_OCTET_STRING;
 446         tmp_os.flags = 0;
 447         tmp_os.data = (unsigned char *)id;
 448         tmp_os.length = (int)idlen;
 449         return ASN1_OCTET_STRING_cmp(&tmp_os, kekri->kekid->keyIdentifier);
 450         }
 451
 452 /* For now hard code AES key wrap info */
 453
 454 static size_t aes_wrap_keylen(int nid)
 455         {
 456         switch (nid)
 457                 {
 458                 case NID_id_aes128_wrap:
 459                 return 16;
 460
 461                 case NID_id_aes192_wrap:
 462                 return  24;
 463
 464                 case NID_id_aes256_wrap:
 465                 return  32;
 466
 467                 default:
 468                 return 0;
 469                 }
 470         }
 471
 472 CMS_RecipientInfo *CMS_add0_recipient_key(CMS_ContentInfo *cms, int nid,
 473                                         unsigned char *key, size_t keylen,
 474                                         unsigned char *id, size_t idlen,
 475                                         ASN1_GENERALIZEDTIME *date,
 476                                         ASN1_OBJECT *otherTypeId,
 477                                         ASN1_TYPE *otherType)
 478         {
 479         CMS_RecipientInfo *ri = NULL;
 480         CMS_EnvelopedData *env;
 481         CMS_KEKRecipientInfo *kekri;
 482         env = cms_get0_enveloped(cms);
 483         if (!env)
 484                 goto err;
 485
 486         if (nid == NID_undef)
 487                 {
 488                 switch (keylen)
 489                         {
 490                         case 16:
 491                         nid = NID_id_aes128_wrap;
 492                         break;
 493
 494                         case  24:
 495                         nid = NID_id_aes192_wrap;
 496                         break;
 497
 498                         case  32:
 499                         nid = NID_id_aes256_wrap;
 500                         break;
 501
 502                         default:
 503                         CMSerr(CMS_F_CMS_ADD0_RECIPIENT_KEY,
 504                                                 CMS_R_INVALID_KEY_LENGTH);
 505                         goto err;
 506                         }
 507
 508                 }
 509         else
 510                 {
 511
 512                 size_t exp_keylen = aes_wrap_keylen(nid);
 513
 514                 if (!exp_keylen)
 515                         {
 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                         {
 523                         CMSerr(CMS_F_CMS_ADD0_RECIPIENT_KEY,
 524                                         CMS_R_INVALID_KEY_LENGTH);
 525                         goto err;
 526                         }
 527
 528                 }
 529
 530         /* Initialize recipient info */
 531         ri = M_ASN1_new_of(CMS_RecipientInfo);
 532         if (!ri)
 533                 goto merr;
 534
 535         ri->d.kekri = M_ASN1_new_of(CMS_KEKRecipientInfo);
 536         if (!ri->d.kekri)
 537                 goto merr;
 538         ri->type = CMS_RECIPINFO_KEK;
 539
 540         kekri = ri->d.kekri;
 541
 542         if (otherTypeId)
 543                 {
 544                 kekri->kekid->other = M_ASN1_new_of(CMS_OtherKeyAttribute);
 545                 if (kekri->kekid->other == NULL)
 546                         goto merr;
 547                 }
 548
 549         if (!sk_CMS_RecipientInfo_push(env->recipientInfos, ri))
 550                 goto merr;
 551
 552
 553         /* After this point no calls can fail */
 554
 555         kekri->version = 4;
 556
 557         kekri->key = key;
 558         kekri->keylen = keylen;
 559
 560         ASN1_STRING_set0(kekri->kekid->keyIdentifier, id, idlen);
 561
 562         kekri->kekid->date = date;
 563
 564         if (kekri->kekid->other)
 565                 {
 566                 kekri->kekid->other->keyAttrId = otherTypeId;
 567                 kekri->kekid->other->keyAttr = otherType;
 568                 }
 569
 570         X509_ALGOR_set0(kekri->keyEncryptionAlgorithm,
 571                                 OBJ_nid2obj(nid), V_ASN1_UNDEF, NULL);
 572
 573         return ri;
 574
 575         merr:
 576         CMSerr(CMS_F_CMS_ADD0_RECIPIENT_KEY, ERR_R_MALLOC_FAILURE);
 577         err:
 578         if (ri)
 579                 M_ASN1_free_of(ri, CMS_RecipientInfo);
 580         return NULL;
 581
 582         }
 583
 584 #if 0
 585 int CMS_RecipientInfo_kekri_get0_id(CMS_RecipientInfo *ri,
 586                                         X509_ALGOR **palg,
 587                                         ASN1_OCTET_STRING **pid,
 588                                         ASN1_GENERALIZEDTIME **pdate,
 589                                         ASN1_OBJECT **potherid,
 590                                         ASN1_TYPE **pothertype)
 591         {
 592         CMS_KEKIdentifier *rkid;
 593         if (ri->type != CMS_RECIPINFO_KEK)
 594                 {
 595                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_GET0_ID, CMS_R_NOT_KEK);
 596                 return 0;
 597                 }
 598         rkid =  ri->d.kekri->kekid;
 599         if (palg)
 600                 *palg = ri->d.kekri->keyEncryptionAlgorithm;
 601         if (pid)
 602                 *pid = rkid->keyIdentifier;
 603         if (pdate)
 604                 *pdate = rkid->date;
 605         if (potherid)
 606                 {
 607                 if (rkid->other)
 608                         *potherid = rkid->other->keyAttrId;
 609                 else
 610                         *potherid = NULL;
 611                 }
 612         if (pothertype)
 613                 {
 614                 if (rkid->other)
 615                         *pothertype = rkid->other->keyAttr;
 616                 else
 617                         *pothertype = NULL;
 618                 }
 619         return 1;
 620         }
 621 #endif
 622
 623
 624 int CMS_RecipientInfo_set0_key(CMS_RecipientInfo *ri,
 625                                 unsigned char *key, size_t keylen)
 626         {
 627         CMS_KEKRecipientInfo *kekri;
 628         if (ri->type != CMS_RECIPINFO_KEK)
 629                 {
 630                 CMSerr(CMS_F_CMS_RECIPIENTINFO_SET0_KEY, CMS_R_NOT_KEK);
 631                 return 0;
 632                 }
 633
 634         kekri = ri->d.kekri;
 635         kekri->key = key;
 636         kekri->keylen = keylen;
 637         return 1;
 638         }
 639
 640
 641 /* Encrypt content key in KEK recipient info */
 642
 643 static int cms_RecipientInfo_kekri_encrypt(CMS_ContentInfo *cms,
 644                                         CMS_RecipientInfo *ri)
 645         {
 646         CMS_EncryptedContentInfo *ec;
 647         CMS_KEKRecipientInfo *kekri;
 648         AES_KEY actx;
 649         unsigned char *wkey = NULL;
 650         int wkeylen;
 651         int r = 0;
 652
 653         ec = cms->d.envelopedData->encryptedContentInfo;
 654
 655         kekri = ri->d.kekri;
 656
 657         if (!kekri->key)
 658                 {
 659                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_ENCRYPT, CMS_R_NO_KEY);
 660                 return 0;
 661                 }
 662
 663         if (AES_set_encrypt_key(kekri->key, kekri->keylen << 3, &actx))
 664                 {
 665                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_ENCRYPT,
 666                                                 CMS_R_ERROR_SETTING_KEY);
 667                 goto err;
 668                 }
 669
 670         wkey = OPENSSL_malloc(ec->keylen + 8);
 671
 672         if (!wkey)
 673                 {
 674                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_ENCRYPT,
 675                                                 ERR_R_MALLOC_FAILURE);
 676                 goto err;
 677                 }
 678
 679         wkeylen = AES_wrap_key(&actx, NULL, wkey, ec->key, ec->keylen);
 680
 681         if (wkeylen <= 0)
 682                 {
 683                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_ENCRYPT, CMS_R_WRAP_ERROR);
 684                 goto err;
 685                 }
 686
 687         ASN1_STRING_set0(kekri->encryptedKey, wkey, wkeylen);
 688
 689         r = 1;
 690
 691         err:
 692
 693         if (!r && wkey)
 694                 OPENSSL_free(wkey);
 695         OPENSSL_cleanse(&actx, sizeof(actx));
 696
 697         return r;
 698
 699         }
 700
 701 /* Decrypt content key in KEK recipient info */
 702
 703 static int cms_RecipientInfo_kekri_decrypt(CMS_ContentInfo *cms,
 704                                         CMS_RecipientInfo *ri)
 705         {
 706         CMS_EncryptedContentInfo *ec;
 707         CMS_KEKRecipientInfo *kekri;
 708         AES_KEY actx;
 709         unsigned char *ukey = NULL;
 710         int ukeylen;
 711         int r = 0, wrap_nid;
 712
 713         ec = cms->d.envelopedData->encryptedContentInfo;
 714
 715         kekri = ri->d.kekri;
 716
 717         if (!kekri->key)
 718                 {
 719                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_DECRYPT, CMS_R_NO_KEY);
 720                 return 0;
 721                 }
 722
 723         wrap_nid = OBJ_obj2nid(kekri->keyEncryptionAlgorithm->algorithm);
 724         if (aes_wrap_keylen(wrap_nid) != kekri->keylen)
 725                 {
 726                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_DECRYPT,
 727                         CMS_R_INVALID_KEY_LENGTH);
 728                 return 0;
 729                 }
 730
 731         /* If encrypted key length is invalid don't bother */
 732
 733         if (kekri->encryptedKey->length < 16)
 734                 {
 735                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_DECRYPT,
 736                                         CMS_R_INVALID_ENCRYPTED_KEY_LENGTH);
 737                 goto err;
 738                 }
 739
 740         if (AES_set_decrypt_key(kekri->key, kekri->keylen << 3, &actx))
 741                 {
 742                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_DECRYPT,
 743                                                 CMS_R_ERROR_SETTING_KEY);
 744                 goto err;
 745                 }
 746
 747         ukey = OPENSSL_malloc(kekri->encryptedKey->length - 8);
 748
 749         if (!ukey)
 750                 {
 751                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_DECRYPT,
 752                                                 ERR_R_MALLOC_FAILURE);
 753                 goto err;
 754                 }
 755
 756         ukeylen = AES_unwrap_key(&actx, NULL, ukey,
 757                                         kekri->encryptedKey->data,
 758                                         kekri->encryptedKey->length);
 759
 760         if (ukeylen <= 0)
 761                 {
 762                 CMSerr(CMS_F_CMS_RECIPIENTINFO_KEKRI_DECRYPT,
 763                                                         CMS_R_UNWRAP_ERROR);
 764                 goto err;
 765                 }
 766
 767         ec->key = ukey;
 768         ec->keylen = ukeylen;
 769
 770         r = 1;
 771
 772         err:
 773
 774         if (!r && ukey)
 775                 OPENSSL_free(ukey);
 776         OPENSSL_cleanse(&actx, sizeof(actx));
 777
 778         return r;
 779
 780         }
 781
 782 int CMS_RecipientInfo_decrypt(CMS_ContentInfo *cms, CMS_RecipientInfo *ri)
 783         {
 784         switch(ri->type)
 785                 {
 786                 case CMS_RECIPINFO_TRANS:
 787                 return cms_RecipientInfo_ktri_decrypt(cms, ri);
 788
 789                 case CMS_RECIPINFO_KEK:
 790                 return cms_RecipientInfo_kekri_decrypt(cms, ri);
 791
 792                 default:
 793                 CMSerr(CMS_F_CMS_RECIPIENTINFO_DECRYPT,
 794                         CMS_R_UNSUPPORTED_RECPIENTINFO_TYPE);
 795                 return 0;
 796                 }
 797         }
 798
 799 BIO *cms_EnvelopedData_init_bio(CMS_ContentInfo *cms)
 800         {
 801         CMS_EncryptedContentInfo *ec;
 802         STACK_OF(CMS_RecipientInfo) *rinfos;
 803         CMS_RecipientInfo *ri;
 804         int i, r, ok = 0;
 805         BIO *ret;
 806
 807         /* Get BIO first to set up key */
 808
 809         ec = cms->d.envelopedData->encryptedContentInfo;
 810         ret = cms_EncryptedContent_init_bio(ec);
 811
 812         /* If error or no cipher end of processing */
 813
 814         if (!ret || !ec->cipher)
 815                 return ret;
 816
 817         /* Now encrypt content key according to each RecipientInfo type */
 818
 819         rinfos = cms->d.envelopedData->recipientInfos;
 820
 821         for (i = 0; i < sk_CMS_RecipientInfo_num(rinfos); i++)
 822                 {
 823                 ri = sk_CMS_RecipientInfo_value(rinfos, i);
 824
 825                 switch (ri->type)
 826                         {
 827                         case CMS_RECIPINFO_TRANS:
 828                         r = cms_RecipientInfo_ktri_encrypt(cms, ri);
 829                         break;
 830
 831                         case CMS_RECIPINFO_KEK:
 832                         r = cms_RecipientInfo_kekri_encrypt(cms, ri);
 833                         break;
 834
 835                         default:
 836                         CMSerr(CMS_F_CMS_ENVELOPEDDATA_INIT_BIO,
 837                                 CMS_R_UNSUPPORTED_RECIPIENT_TYPE);
 838                         goto err;
 839                         }
 840
 841                 if (r <= 0)
 842                         {
 843                         CMSerr(CMS_F_CMS_ENVELOPEDDATA_INIT_BIO,
 844                                 CMS_R_ERROR_SETTING_RECIPIENTINFO);
 845                         goto err;
 846                         }
 847                 }
 848
 849         ok = 1;
 850
 851         err:
 852         ec->cipher = NULL;
 853         if (ec->key)
 854                 {
 855                 OPENSSL_cleanse(ec->key, ec->keylen);
 856                 OPENSSL_free(ec->key);
 857                 ec->key = NULL;
 858                 ec->keylen = 0;
 859                 }
 860         if (ok)
 861                 return ret;
 862         BIO_free(ret);
 863         return NULL;
 864
 865         }