[thirdparty/openssl.git] / crypto / cms / cms_pwri.c

/* crypto/cms/cms_pwri.c */
/*
 * Written by Dr Stephen N Henson (steve@openssl.org) for the OpenSSL
 * project.
 */
/* ====================================================================
 * Copyright (c) 2009 The OpenSSL Project.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *
 * 3. All advertising materials mentioning features or use of this
 *    software must display the following acknowledgment:
 *    "This product includes software developed by the OpenSSL Project
 *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
 *
 * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
 *    endorse or promote products derived from this software without
 *    prior written permission. For written permission, please contact
 *    licensing@OpenSSL.org.
 *
 * 5. Products derived from this software may not be called "OpenSSL"
 *    nor may "OpenSSL" appear in their names without prior written
 *    permission of the OpenSSL Project.
 *
 * 6. Redistributions of any form whatsoever must retain the following
 *    acknowledgment:
 *    "This product includes software developed by the OpenSSL Project
 *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
 *
 * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
 * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
 * OF THE POSSIBILITY OF SUCH DAMAGE.
 * ====================================================================
 */

#include "cryptlib.h"
#include <openssl/asn1t.h>
#include <openssl/pem.h>
#include <openssl/x509v3.h>
#include <openssl/err.h>
#include <openssl/cms.h>
#include <openssl/rand.h>
#include <openssl/aes.h>
#include "cms_lcl.h"
#include "internal/asn1_int.h"

int CMS_RecipientInfo_set0_password(CMS_RecipientInfo *ri,
                                    unsigned char *pass, ossl_ssize_t passlen)
{
    CMS_PasswordRecipientInfo *pwri;
    if (ri->type != CMS_RECIPINFO_PASS) {
        CMSerr(CMS_F_CMS_RECIPIENTINFO_SET0_PASSWORD, CMS_R_NOT_PWRI);
        return 0;
    }

    pwri = ri->d.pwri;
    pwri->pass = pass;
    if (pass && passlen < 0)
        passlen = strlen((char *)pass);
    pwri->passlen = passlen;
    return 1;
}

CMS_RecipientInfo *CMS_add0_recipient_password(CMS_ContentInfo *cms,
                                               int iter, int wrap_nid,
                                               int pbe_nid,
                                               unsigned char *pass,
                                               ossl_ssize_t passlen,
                                               const EVP_CIPHER *kekciph)
{
    CMS_RecipientInfo *ri = NULL;
    CMS_EnvelopedData *env;
    CMS_PasswordRecipientInfo *pwri;
    EVP_CIPHER_CTX ctx;
    X509_ALGOR *encalg = NULL;
    unsigned char iv[EVP_MAX_IV_LENGTH];
    int ivlen;

    env = cms_get0_enveloped(cms);
    if (!env)
        return NULL;

    if (wrap_nid <= 0)
        wrap_nid = NID_id_alg_PWRI_KEK;

    if (pbe_nid <= 0)
        pbe_nid = NID_id_pbkdf2;

    /* Get from enveloped data */
    if (kekciph == NULL)
        kekciph = env->encryptedContentInfo->cipher;

    if (kekciph == NULL) {
        CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD, CMS_R_NO_CIPHER);
        return NULL;
    }
    if (wrap_nid != NID_id_alg_PWRI_KEK) {
        CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD,
               CMS_R_UNSUPPORTED_KEY_ENCRYPTION_ALGORITHM);
        return NULL;
    }

    /* Setup algorithm identifier for cipher */
    encalg = X509_ALGOR_new();
    EVP_CIPHER_CTX_init(&ctx);

    if (EVP_EncryptInit_ex(&ctx, kekciph, NULL, NULL, NULL) <= 0) {
        CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD, ERR_R_EVP_LIB);
        goto err;
    }

    ivlen = EVP_CIPHER_CTX_iv_length(&ctx);

    if (ivlen > 0) {
        if (RAND_bytes(iv, ivlen) <= 0)
            goto err;
        if (EVP_EncryptInit_ex(&ctx, NULL, NULL, NULL, iv) <= 0) {
            CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD, ERR_R_EVP_LIB);
            goto err;
        }
        encalg->parameter = ASN1_TYPE_new();
        if (!encalg->parameter) {
            CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD, ERR_R_MALLOC_FAILURE);
            goto err;
        }
        if (EVP_CIPHER_param_to_asn1(&ctx, encalg->parameter) <= 0) {
            CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD,
                   CMS_R_CIPHER_PARAMETER_INITIALISATION_ERROR);
            goto err;
        }
    }

    encalg->algorithm = OBJ_nid2obj(EVP_CIPHER_CTX_type(&ctx));

    EVP_CIPHER_CTX_cleanup(&ctx);

    /* Initialize recipient info */
    ri = M_ASN1_new_of(CMS_RecipientInfo);
    if (!ri)
        goto merr;

    ri->d.pwri = M_ASN1_new_of(CMS_PasswordRecipientInfo);
    if (!ri->d.pwri)
        goto merr;
    ri->type = CMS_RECIPINFO_PASS;

    pwri = ri->d.pwri;
    /* Since this is overwritten, free up empty structure already there */
    X509_ALGOR_free(pwri->keyEncryptionAlgorithm);
    pwri->keyEncryptionAlgorithm = X509_ALGOR_new();
    if (!pwri->keyEncryptionAlgorithm)
        goto merr;
    pwri->keyEncryptionAlgorithm->algorithm = OBJ_nid2obj(wrap_nid);
    pwri->keyEncryptionAlgorithm->parameter = ASN1_TYPE_new();
    if (!pwri->keyEncryptionAlgorithm->parameter)
        goto merr;

    if (!ASN1_item_pack(encalg, ASN1_ITEM_rptr(X509_ALGOR),
                        &pwri->keyEncryptionAlgorithm->parameter->
                        value.sequence))
         goto merr;
    pwri->keyEncryptionAlgorithm->parameter->type = V_ASN1_SEQUENCE;

    X509_ALGOR_free(encalg);
    encalg = NULL;

    /* Setup PBE algorithm */

    pwri->keyDerivationAlgorithm = PKCS5_pbkdf2_set(iter, NULL, 0, -1, -1);

    if (!pwri->keyDerivationAlgorithm)
        goto err;

    CMS_RecipientInfo_set0_password(ri, pass, passlen);
    pwri->version = 0;

    if (!sk_CMS_RecipientInfo_push(env->recipientInfos, ri))
        goto merr;

    return ri;

 merr:
    CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD, ERR_R_MALLOC_FAILURE);
 err:
    EVP_CIPHER_CTX_cleanup(&ctx);
    if (ri)
        M_ASN1_free_of(ri, CMS_RecipientInfo);
    if (encalg)
        X509_ALGOR_free(encalg);
    return NULL;

}

/*
 * This is an implementation of the key wrapping mechanism in RFC3211, at
 * some point this should go into EVP.
 */

static int kek_unwrap_key(unsigned char *out, size_t *outlen,
                          const unsigned char *in, size_t inlen,
                          EVP_CIPHER_CTX *ctx)
{
    size_t blocklen = EVP_CIPHER_CTX_block_size(ctx);
    unsigned char *tmp;
    int outl, rv = 0;
    if (inlen < 2 * blocklen) {
        /* too small */
        return 0;
    }
    if (inlen % blocklen) {
        /* Invalid size */
        return 0;
    }
    tmp = OPENSSL_malloc(inlen);
    if(!tmp)
        return 0;
    /* setup IV by decrypting last two blocks */
    if (!EVP_DecryptUpdate(ctx, tmp + inlen - 2 * blocklen, &outl,
                           in + inlen - 2 * blocklen, blocklen * 2)
        /*
         * Do a decrypt of last decrypted block to set IV to correct value
         * output it to start of buffer so we don't corrupt decrypted block
         * this works because buffer is at least two block lengths long.
         */
        || !EVP_DecryptUpdate(ctx, tmp, &outl,
                              tmp + inlen - blocklen, blocklen)
        /* Can now decrypt first n - 1 blocks */
        || !EVP_DecryptUpdate(ctx, tmp, &outl, in, inlen - blocklen)

        /* Reset IV to original value */
        || !EVP_DecryptInit_ex(ctx, NULL, NULL, NULL, NULL)
        /* Decrypt again */
        || !EVP_DecryptUpdate(ctx, tmp, &outl, tmp, inlen))
        goto err;
    /* Check check bytes */
    if (((tmp[1] ^ tmp[4]) & (tmp[2] ^ tmp[5]) & (tmp[3] ^ tmp[6])) != 0xff) {
        /* Check byte failure */
        goto err;
    }
    if (inlen < (size_t)(tmp[0] - 4)) {
        /* Invalid length value */
        goto err;
    }
    *outlen = (size_t)tmp[0];
    memcpy(out, tmp + 4, *outlen);
    rv = 1;
 err:
    OPENSSL_cleanse(tmp, inlen);
    OPENSSL_free(tmp);
    return rv;

}

static int kek_wrap_key(unsigned char *out, size_t *outlen,
                        const unsigned char *in, size_t inlen,
                        EVP_CIPHER_CTX *ctx)
{
    size_t blocklen = EVP_CIPHER_CTX_block_size(ctx);
    size_t olen;
    int dummy;
    /*
     * First decide length of output buffer: need header and round up to
     * multiple of block length.
     */
    olen = (inlen + 4 + blocklen - 1) / blocklen;
    olen *= blocklen;
    if (olen < 2 * blocklen) {
        /* Key too small */
        return 0;
    }
    if (inlen > 0xFF) {
        /* Key too large */
        return 0;
    }
    if (out) {
        /* Set header */
        out[0] = (unsigned char)inlen;
        out[1] = in[0] ^ 0xFF;
        out[2] = in[1] ^ 0xFF;
        out[3] = in[2] ^ 0xFF;
        memcpy(out + 4, in, inlen);
        /* Add random padding to end */
        if (olen > inlen + 4
            && RAND_bytes(out + 4 + inlen, olen - 4 - inlen) <= 0)
            return 0;
        /* Encrypt twice */
        if (!EVP_EncryptUpdate(ctx, out, &dummy, out, olen)
            || !EVP_EncryptUpdate(ctx, out, &dummy, out, olen))
            return 0;
    }

    *outlen = olen;

    return 1;
}

/* Encrypt/Decrypt content key in PWRI recipient info */

int cms_RecipientInfo_pwri_crypt(CMS_ContentInfo *cms, CMS_RecipientInfo *ri,
                                 int en_de)
{
    CMS_EncryptedContentInfo *ec;
    CMS_PasswordRecipientInfo *pwri;
    const unsigned char *p = NULL;
    int plen;
    int r = 0;
    X509_ALGOR *algtmp, *kekalg = NULL;
    EVP_CIPHER_CTX kekctx;
    const EVP_CIPHER *kekcipher;
    unsigned char *key = NULL;
    size_t keylen;

    ec = cms->d.envelopedData->encryptedContentInfo;

    pwri = ri->d.pwri;
    EVP_CIPHER_CTX_init(&kekctx);

    if (!pwri->pass) {
        CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT, CMS_R_NO_PASSWORD);
        return 0;
    }
    algtmp = pwri->keyEncryptionAlgorithm;

    if (!algtmp || OBJ_obj2nid(algtmp->algorithm) != NID_id_alg_PWRI_KEK) {
        CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT,
               CMS_R_UNSUPPORTED_KEY_ENCRYPTION_ALGORITHM);
        return 0;
    }

    if (algtmp->parameter->type == V_ASN1_SEQUENCE) {
        p = algtmp->parameter->value.sequence->data;
        plen = algtmp->parameter->value.sequence->length;
        kekalg = d2i_X509_ALGOR(NULL, &p, plen);
    }
    if (kekalg == NULL) {
        CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT,
               CMS_R_INVALID_KEY_ENCRYPTION_PARAMETER);
        return 0;
    }

    kekcipher = EVP_get_cipherbyobj(kekalg->algorithm);

    if (!kekcipher) {
        CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT, CMS_R_UNKNOWN_CIPHER);
        goto err;
    }

    /* Fixup cipher based on AlgorithmIdentifier to set IV etc */
    if (!EVP_CipherInit_ex(&kekctx, kekcipher, NULL, NULL, NULL, en_de))
        goto err;
    EVP_CIPHER_CTX_set_padding(&kekctx, 0);
    if (EVP_CIPHER_asn1_to_param(&kekctx, kekalg->parameter) < 0) {
        CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT,
               CMS_R_CIPHER_PARAMETER_INITIALISATION_ERROR);
        goto err;
    }

    algtmp = pwri->keyDerivationAlgorithm;

    /* Finish password based key derivation to setup key in "ctx" */

    if (EVP_PBE_CipherInit(algtmp->algorithm,
                           (char *)pwri->pass, pwri->passlen,
                           algtmp->parameter, &kekctx, en_de) < 0) {
        CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT, ERR_R_EVP_LIB);
        goto err;
    }

    /* Finally wrap/unwrap the key */

    if (en_de) {

        if (!kek_wrap_key(NULL, &keylen, ec->key, ec->keylen, &kekctx))
            goto err;

        key = OPENSSL_malloc(keylen);

        if (!key)
            goto err;

        if (!kek_wrap_key(key, &keylen, ec->key, ec->keylen, &kekctx))
            goto err;
        pwri->encryptedKey->data = key;
        pwri->encryptedKey->length = keylen;
    } else {
        key = OPENSSL_malloc(pwri->encryptedKey->length);

        if (!key) {
            CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT, ERR_R_MALLOC_FAILURE);
            goto err;
        }
        if (!kek_unwrap_key(key, &keylen,
                            pwri->encryptedKey->data,
                            pwri->encryptedKey->length, &kekctx)) {
            CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT, CMS_R_UNWRAP_FAILURE);
            goto err;
        }

        ec->key = key;
        ec->keylen = keylen;

    }

    r = 1;

 err:

    EVP_CIPHER_CTX_cleanup(&kekctx);

    if (!r && key)
        OPENSSL_free(key);
    X509_ALGOR_free(kekalg);

    return r;

}
Commit	Line	Data
d2a53c22	1	/* crypto/cms/cms_pwri.c */
0f113f3e MC	2	/*
0f113f3e MC	3	* Written by Dr Stephen N Henson (steve@openssl.org) for the OpenSSL
d2a53c22 DSH	4	* project.
	5	*/
	6	/* ====================================================================
	7	* Copyright (c) 2009 The OpenSSL Project. All rights reserved.
	8	*
	9	* Redistribution and use in source and binary forms, with or without
	10	* modification, are permitted provided that the following conditions
	11	* are met:
	12	*
	13	* 1. Redistributions of source code must retain the above copyright
0f113f3e	14	* notice, this list of conditions and the following disclaimer.
d2a53c22 DSH	15	*
	16	* 2. Redistributions in binary form must reproduce the above copyright
	17	* notice, this list of conditions and the following disclaimer in
	18	* the documentation and/or other materials provided with the
	19	* distribution.
	20	*
	21	* 3. All advertising materials mentioning features or use of this
	22	* software must display the following acknowledgment:
	23	* "This product includes software developed by the OpenSSL Project
	24	* for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
	25	*
	26	* 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
	27	* endorse or promote products derived from this software without
	28	* prior written permission. For written permission, please contact
	29	* licensing@OpenSSL.org.
	30	*
	31	* 5. Products derived from this software may not be called "OpenSSL"
	32	* nor may "OpenSSL" appear in their names without prior written
	33	* permission of the OpenSSL Project.
	34	*
	35	* 6. Redistributions of any form whatsoever must retain the following
	36	* acknowledgment:
	37	* "This product includes software developed by the OpenSSL Project
	38	* for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
	39	*
	40	* THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
	41	* EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
	42	* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
	43	* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE OpenSSL PROJECT OR
	44	* ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
	45	* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
	46	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
	47	* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
	48	* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
	49	* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
	50	* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
	51	* OF THE POSSIBILITY OF SUCH DAMAGE.
	52	* ====================================================================
	53	*/
	54
	55	#include "cryptlib.h"
	56	#include <openssl/asn1t.h>
	57	#include <openssl/pem.h>
	58	#include <openssl/x509v3.h>
	59	#include <openssl/err.h>
	60	#include <openssl/cms.h>
	61	#include <openssl/rand.h>
	62	#include <openssl/aes.h>
	63	#include "cms_lcl.h"
5fe736e5	64	#include "internal/asn1_int.h"
d2a53c22	65
0f113f3e MC	66	int CMS_RecipientInfo_set0_password(CMS_RecipientInfo *ri,
	67	unsigned char *pass, ossl_ssize_t passlen)
	68	{
	69	CMS_PasswordRecipientInfo *pwri;
	70	if (ri->type != CMS_RECIPINFO_PASS) {
	71	CMSerr(CMS_F_CMS_RECIPIENTINFO_SET0_PASSWORD, CMS_R_NOT_PWRI);
	72	return 0;
	73	}
	74
	75	pwri = ri->d.pwri;
	76	pwri->pass = pass;
	77	if (pass && passlen < 0)
	78	passlen = strlen((char *)pass);
	79	pwri->passlen = passlen;
	80	return 1;
	81	}
d2a53c22 DSH	82
d2a53c22 DSH	83	CMS_RecipientInfo CMS_add0_recipient_password(CMS_ContentInfo cms,
0f113f3e MC	84	int iter, int wrap_nid,
	85	int pbe_nid,
	86	unsigned char *pass,
	87	ossl_ssize_t passlen,
	88	const EVP_CIPHER *kekciph)
	89	{
	90	CMS_RecipientInfo *ri = NULL;
	91	CMS_EnvelopedData *env;
	92	CMS_PasswordRecipientInfo *pwri;
	93	EVP_CIPHER_CTX ctx;
	94	X509_ALGOR *encalg = NULL;
	95	unsigned char iv[EVP_MAX_IV_LENGTH];
	96	int ivlen;
	97
	98	env = cms_get0_enveloped(cms);
	99	if (!env)
	100	return NULL;
	101
	102	if (wrap_nid <= 0)
	103	wrap_nid = NID_id_alg_PWRI_KEK;
	104
	105	if (pbe_nid <= 0)
	106	pbe_nid = NID_id_pbkdf2;
	107
	108	/* Get from enveloped data */
	109	if (kekciph == NULL)
	110	kekciph = env->encryptedContentInfo->cipher;
	111
	112	if (kekciph == NULL) {
	113	CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD, CMS_R_NO_CIPHER);
	114	return NULL;
	115	}
	116	if (wrap_nid != NID_id_alg_PWRI_KEK) {
	117	CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD,
	118	CMS_R_UNSUPPORTED_KEY_ENCRYPTION_ALGORITHM);
	119	return NULL;
	120	}
	121
	122	/* Setup algorithm identifier for cipher */
	123	encalg = X509_ALGOR_new();
	124	EVP_CIPHER_CTX_init(&ctx);
	125
	126	if (EVP_EncryptInit_ex(&ctx, kekciph, NULL, NULL, NULL) <= 0) {
	127	CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD, ERR_R_EVP_LIB);
	128	goto err;
	129	}
	130
	131	ivlen = EVP_CIPHER_CTX_iv_length(&ctx);
	132
	133	if (ivlen > 0) {
266483d2	134	if (RAND_bytes(iv, ivlen) <= 0)
0f113f3e MC	135	goto err;
	136	if (EVP_EncryptInit_ex(&ctx, NULL, NULL, NULL, iv) <= 0) {
	137	CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD, ERR_R_EVP_LIB);
	138	goto err;
	139	}
	140	encalg->parameter = ASN1_TYPE_new();
	141	if (!encalg->parameter) {
	142	CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD, ERR_R_MALLOC_FAILURE);
	143	goto err;
	144	}
	145	if (EVP_CIPHER_param_to_asn1(&ctx, encalg->parameter) <= 0) {
	146	CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD,
	147	CMS_R_CIPHER_PARAMETER_INITIALISATION_ERROR);
	148	goto err;
	149	}
	150	}
	151
	152	encalg->algorithm = OBJ_nid2obj(EVP_CIPHER_CTX_type(&ctx));
	153
	154	EVP_CIPHER_CTX_cleanup(&ctx);
	155
	156	/* Initialize recipient info */
	157	ri = M_ASN1_new_of(CMS_RecipientInfo);
	158	if (!ri)
	159	goto merr;
	160
	161	ri->d.pwri = M_ASN1_new_of(CMS_PasswordRecipientInfo);
	162	if (!ri->d.pwri)
	163	goto merr;
	164	ri->type = CMS_RECIPINFO_PASS;
	165
	166	pwri = ri->d.pwri;
	167	/* Since this is overwritten, free up empty structure already there */
	168	X509_ALGOR_free(pwri->keyEncryptionAlgorithm);
	169	pwri->keyEncryptionAlgorithm = X509_ALGOR_new();
	170	if (!pwri->keyEncryptionAlgorithm)
	171	goto merr;
	172	pwri->keyEncryptionAlgorithm->algorithm = OBJ_nid2obj(wrap_nid);
	173	pwri->keyEncryptionAlgorithm->parameter = ASN1_TYPE_new();
	174	if (!pwri->keyEncryptionAlgorithm->parameter)
	175	goto merr;
	176
	177	if (!ASN1_item_pack(encalg, ASN1_ITEM_rptr(X509_ALGOR),
	178	&pwri->keyEncryptionAlgorithm->parameter->
	179	value.sequence))
	180	goto merr;
	181	pwri->keyEncryptionAlgorithm->parameter->type = V_ASN1_SEQUENCE;
	182
	183	X509_ALGOR_free(encalg);
	184	encalg = NULL;
	185
	186	/* Setup PBE algorithm */
	187
	188	pwri->keyDerivationAlgorithm = PKCS5_pbkdf2_set(iter, NULL, 0, -1, -1);
	189
	190	if (!pwri->keyDerivationAlgorithm)
	191	goto err;
	192
	193	CMS_RecipientInfo_set0_password(ri, pass, passlen);
	194	pwri->version = 0;
	195
	196	if (!sk_CMS_RecipientInfo_push(env->recipientInfos, ri))
	197	goto merr;
	198
199	return ri;
200
201	merr:
202	CMSerr(CMS_F_CMS_ADD0_RECIPIENT_PASSWORD, ERR_R_MALLOC_FAILURE);
203	err:
204	EVP_CIPHER_CTX_cleanup(&ctx);
205	if (ri)
206	M_ASN1_free_of(ri, CMS_RecipientInfo);
207	if (encalg)
208	X509_ALGOR_free(encalg);
209	return NULL;
210
211	}
212
213	/*
214	* This is an implementation of the key wrapping mechanism in RFC3211, at
215	* some point this should go into EVP.
d2a53c22 DSH	216	*/
	217
	218	static int kek_unwrap_key(unsigned char out, size_t outlen,
0f113f3e MC	219	const unsigned char *in, size_t inlen,
	220	EVP_CIPHER_CTX *ctx)
	221	{
	222	size_t blocklen = EVP_CIPHER_CTX_block_size(ctx);
	223	unsigned char *tmp;
	224	int outl, rv = 0;
	225	if (inlen < 2 * blocklen) {
	226	/* too small */
	227	return 0;
	228	}
	229	if (inlen % blocklen) {
	230	/* Invalid size */
	231	return 0;
	232	}
	233	tmp = OPENSSL_malloc(inlen);
918bb865 MC	234	if(!tmp)
918bb865 MC	235	return 0;
0f113f3e MC	236	/* setup IV by decrypting last two blocks */
	237	if (!EVP_DecryptUpdate(ctx, tmp + inlen - 2 * blocklen, &outl,
	238	in + inlen - 2 * blocklen, blocklen * 2)
	239	/*
	240	* Do a decrypt of last decrypted block to set IV to correct value
	241	* output it to start of buffer so we don't corrupt decrypted block
	242	* this works because buffer is at least two block lengths long.
	243	*/
	244	\|\| !EVP_DecryptUpdate(ctx, tmp, &outl,
	245	tmp + inlen - blocklen, blocklen)
	246	/* Can now decrypt first n - 1 blocks */
	247	\|\| !EVP_DecryptUpdate(ctx, tmp, &outl, in, inlen - blocklen)
	248
	249	/* Reset IV to original value */
	250	\|\| !EVP_DecryptInit_ex(ctx, NULL, NULL, NULL, NULL)
	251	/* Decrypt again */
	252	\|\| !EVP_DecryptUpdate(ctx, tmp, &outl, tmp, inlen))
	253	goto err;
	254	/* Check check bytes */
	255	if (((tmp[1] ^ tmp[4]) & (tmp[2] ^ tmp[5]) & (tmp[3] ^ tmp[6])) != 0xff) {
	256	/* Check byte failure */
	257	goto err;
	258	}
	259	if (inlen < (size_t)(tmp[0] - 4)) {
	260	/* Invalid length value */
	261	goto err;
	262	}
	263	*outlen = (size_t)tmp[0];
	264	memcpy(out, tmp + 4, *outlen);
	265	rv = 1;
	266	err:
	267	OPENSSL_cleanse(tmp, inlen);
	268	OPENSSL_free(tmp);
	269	return rv;
	270
	271	}
d2a53c22 DSH	272
d2a53c22 DSH	273	static int kek_wrap_key(unsigned char out, size_t outlen,
0f113f3e MC	274	const unsigned char *in, size_t inlen,
	275	EVP_CIPHER_CTX *ctx)
	276	{
	277	size_t blocklen = EVP_CIPHER_CTX_block_size(ctx);
	278	size_t olen;
	279	int dummy;
	280	/*
	281	* First decide length of output buffer: need header and round up to
	282	* multiple of block length.
	283	*/
	284	olen = (inlen + 4 + blocklen - 1) / blocklen;
	285	olen *= blocklen;
	286	if (olen < 2 * blocklen) {
	287	/* Key too small */
	288	return 0;
	289	}
	290	if (inlen > 0xFF) {
	291	/* Key too large */
	292	return 0;
	293	}
	294	if (out) {
	295	/* Set header */
	296	out[0] = (unsigned char)inlen;
	297	out[1] = in[0] ^ 0xFF;
	298	out[2] = in[1] ^ 0xFF;
	299	out[3] = in[2] ^ 0xFF;
	300	memcpy(out + 4, in, inlen);
	301	/* Add random padding to end */
266483d2 MC	302	if (olen > inlen + 4
	303	&& RAND_bytes(out + 4 + inlen, olen - 4 - inlen) <= 0)
	304	return 0;
0f113f3e MC	305	/* Encrypt twice */
	306	if (!EVP_EncryptUpdate(ctx, out, &dummy, out, olen)
	307	\|\| !EVP_EncryptUpdate(ctx, out, &dummy, out, olen))
	308	return 0;
	309	}
	310
	311	*outlen = olen;
	312
	313	return 1;
	314	}
d2a53c22 DSH	315
	316	/* Encrypt/Decrypt content key in PWRI recipient info */
	317
	318	int cms_RecipientInfo_pwri_crypt(CMS_ContentInfo cms, CMS_RecipientInfo ri,
0f113f3e MC	319	int en_de)
	320	{
	321	CMS_EncryptedContentInfo *ec;
	322	CMS_PasswordRecipientInfo *pwri;
	323	const unsigned char *p = NULL;
	324	int plen;
	325	int r = 0;
	326	X509_ALGOR algtmp, kekalg = NULL;
	327	EVP_CIPHER_CTX kekctx;
	328	const EVP_CIPHER *kekcipher;
	329	unsigned char *key = NULL;
	330	size_t keylen;
	331
	332	ec = cms->d.envelopedData->encryptedContentInfo;
	333
	334	pwri = ri->d.pwri;
	335	EVP_CIPHER_CTX_init(&kekctx);
	336
	337	if (!pwri->pass) {
	338	CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT, CMS_R_NO_PASSWORD);
	339	return 0;
	340	}
	341	algtmp = pwri->keyEncryptionAlgorithm;
	342
	343	if (!algtmp \|\| OBJ_obj2nid(algtmp->algorithm) != NID_id_alg_PWRI_KEK) {
	344	CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT,
	345	CMS_R_UNSUPPORTED_KEY_ENCRYPTION_ALGORITHM);
	346	return 0;
	347	}
	348
	349	if (algtmp->parameter->type == V_ASN1_SEQUENCE) {
	350	p = algtmp->parameter->value.sequence->data;
	351	plen = algtmp->parameter->value.sequence->length;
	352	kekalg = d2i_X509_ALGOR(NULL, &p, plen);
	353	}
	354	if (kekalg == NULL) {
	355	CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT,
	356	CMS_R_INVALID_KEY_ENCRYPTION_PARAMETER);
	357	return 0;
	358	}
	359
	360	kekcipher = EVP_get_cipherbyobj(kekalg->algorithm);
	361
	362	if (!kekcipher) {
	363	CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT, CMS_R_UNKNOWN_CIPHER);
	364	goto err;
	365	}
	366
	367	/* Fixup cipher based on AlgorithmIdentifier to set IV etc */
	368	if (!EVP_CipherInit_ex(&kekctx, kekcipher, NULL, NULL, NULL, en_de))
	369	goto err;
	370	EVP_CIPHER_CTX_set_padding(&kekctx, 0);
	371	if (EVP_CIPHER_asn1_to_param(&kekctx, kekalg->parameter) < 0) {
	372	CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT,
	373	CMS_R_CIPHER_PARAMETER_INITIALISATION_ERROR);
	374	goto err;
	375	}
	376
	377	algtmp = pwri->keyDerivationAlgorithm;
	378
	379	/* Finish password based key derivation to setup key in "ctx" */
	380
	381	if (EVP_PBE_CipherInit(algtmp->algorithm,
	382	(char *)pwri->pass, pwri->passlen,
383	algtmp->parameter, &kekctx, en_de) < 0) {
384	CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT, ERR_R_EVP_LIB);
385	goto err;
386	}
387
388	/* Finally wrap/unwrap the key */
389
390	if (en_de) {
391
392	if (!kek_wrap_key(NULL, &keylen, ec->key, ec->keylen, &kekctx))
393	goto err;
394
395	key = OPENSSL_malloc(keylen);
396
397	if (!key)
398	goto err;
399
400	if (!kek_wrap_key(key, &keylen, ec->key, ec->keylen, &kekctx))
401	goto err;
402	pwri->encryptedKey->data = key;
403	pwri->encryptedKey->length = keylen;
404	} else {
405	key = OPENSSL_malloc(pwri->encryptedKey->length);
406
407	if (!key) {
408	CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT, ERR_R_MALLOC_FAILURE);
409	goto err;
410	}
411	if (!kek_unwrap_key(key, &keylen,
412	pwri->encryptedKey->data,
413	pwri->encryptedKey->length, &kekctx)) {
414	CMSerr(CMS_F_CMS_RECIPIENTINFO_PWRI_CRYPT, CMS_R_UNWRAP_FAILURE);
415	goto err;
416	}
417
418	ec->key = key;
419	ec->keylen = keylen;
420
421	}
422
423	r = 1;
424
425	err:
426
427	EVP_CIPHER_CTX_cleanup(&kekctx);
428
429	if (!r && key)
430	OPENSSL_free(key);
431	X509_ALGOR_free(kekalg);
432
433	return r;
434
435	}