cms encrypt, better OBJ_nid2obj() return check
[openssl.git] / crypto / cms / cms_enc.c
1 /*
2  * Copyright 2008-2021 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 "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/rand.h>
17 #include "crypto/evp.h"
18 #include "cms_local.h"
19
20 /* CMS EncryptedData Utilities */
21
22 /* Return BIO based on EncryptedContentInfo and key */
23
24 BIO *ossl_cms_EncryptedContent_init_bio(CMS_EncryptedContentInfo *ec,
25                                         const CMS_CTX *cms_ctx)
26 {
27     BIO *b;
28     EVP_CIPHER_CTX *ctx;
29     EVP_CIPHER *fetched_ciph = NULL;
30     const EVP_CIPHER *cipher = NULL;
31     X509_ALGOR *calg = ec->contentEncryptionAlgorithm;
32     evp_cipher_aead_asn1_params aparams;
33     unsigned char iv[EVP_MAX_IV_LENGTH], *piv = NULL;
34     unsigned char *tkey = NULL;
35     int len;
36     int ivlen = 0;
37     size_t tkeylen = 0;
38     int ok = 0;
39     int enc, keep_key = 0;
40     OSSL_LIB_CTX *libctx = ossl_cms_ctx_get0_libctx(cms_ctx);
41     const char *propq = ossl_cms_ctx_get0_propq(cms_ctx);
42
43     enc = ec->cipher ? 1 : 0;
44
45     b = BIO_new(BIO_f_cipher());
46     if (b == NULL) {
47         ERR_raise(ERR_LIB_CMS, ERR_R_BIO_LIB);
48         return NULL;
49     }
50
51     BIO_get_cipher_ctx(b, &ctx);
52
53     (void)ERR_set_mark();
54     if (enc) {
55         cipher = ec->cipher;
56         /*
57          * If not keeping key set cipher to NULL so subsequent calls decrypt.
58          */
59         if (ec->key != NULL)
60             ec->cipher = NULL;
61     } else {
62         cipher = EVP_get_cipherbyobj(calg->algorithm);
63     }
64     if (cipher != NULL) {
65         fetched_ciph = EVP_CIPHER_fetch(libctx, EVP_CIPHER_get0_name(cipher),
66                                         propq);
67         if (fetched_ciph != NULL)
68             cipher = fetched_ciph;
69     }
70     if (cipher == NULL) {
71         (void)ERR_clear_last_mark();
72         ERR_raise(ERR_LIB_CMS, CMS_R_UNKNOWN_CIPHER);
73         goto err;
74     }
75     (void)ERR_pop_to_mark();
76
77     if (EVP_CipherInit_ex(ctx, cipher, NULL, NULL, NULL, enc) <= 0) {
78         ERR_raise(ERR_LIB_CMS, CMS_R_CIPHER_INITIALISATION_ERROR);
79         goto err;
80     }
81
82     if (enc) {
83         calg->algorithm = OBJ_nid2obj(EVP_CIPHER_CTX_get_type(ctx));
84         if (calg->algorithm == NULL || calg->algorithm->nid == NID_undef) {
85             ERR_raise(ERR_LIB_CMS, CMS_R_UNSUPPORTED_CONTENT_ENCRYPTION_ALGORITHM);
86             goto err;
87         }
88         /* Generate a random IV if we need one */
89         ivlen = EVP_CIPHER_CTX_get_iv_length(ctx);
90         if (ivlen < 0) {
91             ERR_raise(ERR_LIB_CMS, ERR_R_EVP_LIB);
92             goto err;
93         }
94
95         if (ivlen > 0) {
96             if (RAND_bytes_ex(libctx, iv, ivlen, 0) <= 0)
97                 goto err;
98             piv = iv;
99         }
100     } else {
101         if (evp_cipher_asn1_to_param_ex(ctx, calg->parameter, &aparams) <= 0) {
102             ERR_raise(ERR_LIB_CMS, CMS_R_CIPHER_PARAMETER_INITIALISATION_ERROR);
103             goto err;
104         }
105         if ((EVP_CIPHER_get_flags(cipher) & EVP_CIPH_FLAG_AEAD_CIPHER)) {
106             piv = aparams.iv;
107             if (ec->taglen > 0
108                     && EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_AEAD_SET_TAG,
109                                            ec->taglen, ec->tag) <= 0) {
110                 ERR_raise(ERR_LIB_CMS, CMS_R_CIPHER_AEAD_SET_TAG_ERROR);
111                 goto err;
112             }
113         }
114     }
115     len = EVP_CIPHER_CTX_get_key_length(ctx);
116     if (len <= 0)
117         goto err;
118     tkeylen = (size_t)len;
119
120     /* Generate random session key */
121     if (!enc || !ec->key) {
122         tkey = OPENSSL_malloc(tkeylen);
123         if (tkey == NULL)
124             goto err;
125         if (EVP_CIPHER_CTX_rand_key(ctx, tkey) <= 0)
126             goto err;
127     }
128
129     if (!ec->key) {
130         ec->key = tkey;
131         ec->keylen = tkeylen;
132         tkey = NULL;
133         if (enc)
134             keep_key = 1;
135         else
136             ERR_clear_error();
137
138     }
139
140     if (ec->keylen != tkeylen) {
141         /* If necessary set key length */
142         if (EVP_CIPHER_CTX_set_key_length(ctx, ec->keylen) <= 0) {
143             /*
144              * Only reveal failure if debugging so we don't leak information
145              * which may be useful in MMA.
146              */
147             if (enc || ec->debug) {
148                 ERR_raise(ERR_LIB_CMS, CMS_R_INVALID_KEY_LENGTH);
149                 goto err;
150             } else {
151                 /* Use random key */
152                 OPENSSL_clear_free(ec->key, ec->keylen);
153                 ec->key = tkey;
154                 ec->keylen = tkeylen;
155                 tkey = NULL;
156                 ERR_clear_error();
157             }
158         }
159     }
160
161     if (EVP_CipherInit_ex(ctx, NULL, NULL, ec->key, piv, enc) <= 0) {
162         ERR_raise(ERR_LIB_CMS, CMS_R_CIPHER_INITIALISATION_ERROR);
163         goto err;
164     }
165     if (enc) {
166         calg->parameter = ASN1_TYPE_new();
167         if (calg->parameter == NULL) {
168             ERR_raise(ERR_LIB_CMS, ERR_R_ASN1_LIB);
169             goto err;
170         }
171         if ((EVP_CIPHER_get_flags(cipher) & EVP_CIPH_FLAG_AEAD_CIPHER)) {
172             memcpy(aparams.iv, piv, ivlen);
173             aparams.iv_len = ivlen;
174             aparams.tag_len = EVP_CIPHER_CTX_get_tag_length(ctx);
175             if (aparams.tag_len <= 0)
176                 goto err;
177         }
178
179         if (evp_cipher_param_to_asn1_ex(ctx, calg->parameter, &aparams) <= 0) {
180             ERR_raise(ERR_LIB_CMS, CMS_R_CIPHER_PARAMETER_INITIALISATION_ERROR);
181             goto err;
182         }
183         /* If parameter type not set omit parameter */
184         if (calg->parameter->type == V_ASN1_UNDEF) {
185             ASN1_TYPE_free(calg->parameter);
186             calg->parameter = NULL;
187         }
188     }
189     ok = 1;
190
191  err:
192     EVP_CIPHER_free(fetched_ciph);
193     if (!keep_key || !ok) {
194         OPENSSL_clear_free(ec->key, ec->keylen);
195         ec->key = NULL;
196     }
197     OPENSSL_clear_free(tkey, tkeylen);
198     if (ok)
199         return b;
200     BIO_free(b);
201     return NULL;
202 }
203
204 int ossl_cms_EncryptedContent_init(CMS_EncryptedContentInfo *ec,
205                                    const EVP_CIPHER *cipher,
206                                    const unsigned char *key, size_t keylen,
207                                    const CMS_CTX *cms_ctx)
208 {
209     ec->cipher = cipher;
210     if (key) {
211         if ((ec->key = OPENSSL_malloc(keylen)) == NULL)
212             return 0;
213         memcpy(ec->key, key, keylen);
214     }
215     ec->keylen = keylen;
216     if (cipher != NULL)
217         ec->contentType = OBJ_nid2obj(NID_pkcs7_data);
218     return 1;
219 }
220
221 int CMS_EncryptedData_set1_key(CMS_ContentInfo *cms, const EVP_CIPHER *ciph,
222                                const unsigned char *key, size_t keylen)
223 {
224     CMS_EncryptedContentInfo *ec;
225
226     if (!key || !keylen) {
227         ERR_raise(ERR_LIB_CMS, CMS_R_NO_KEY);
228         return 0;
229     }
230     if (ciph) {
231         cms->d.encryptedData = M_ASN1_new_of(CMS_EncryptedData);
232         if (!cms->d.encryptedData) {
233             ERR_raise(ERR_LIB_CMS, ERR_R_ASN1_LIB);
234             return 0;
235         }
236         cms->contentType = OBJ_nid2obj(NID_pkcs7_encrypted);
237         cms->d.encryptedData->version = 0;
238     } else if (OBJ_obj2nid(cms->contentType) != NID_pkcs7_encrypted) {
239         ERR_raise(ERR_LIB_CMS, CMS_R_NOT_ENCRYPTED_DATA);
240         return 0;
241     }
242     ec = cms->d.encryptedData->encryptedContentInfo;
243     return ossl_cms_EncryptedContent_init(ec, ciph, key, keylen,
244                                           ossl_cms_get0_cmsctx(cms));
245 }
246
247 BIO *ossl_cms_EncryptedData_init_bio(const CMS_ContentInfo *cms)
248 {
249     CMS_EncryptedData *enc = cms->d.encryptedData;
250     if (enc->encryptedContentInfo->cipher && enc->unprotectedAttrs)
251         enc->version = 2;
252     return ossl_cms_EncryptedContent_init_bio(enc->encryptedContentInfo,
253                                               ossl_cms_get0_cmsctx(cms));
254 }