add EVP_PKEY_RSA_PSS
[openssl.git] / crypto / rsa / rsa_ameth.c
1 /*
2  * Copyright 2006-2016 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 <stdio.h>
11 #include "internal/cryptlib.h"
12 #include <openssl/asn1t.h>
13 #include <openssl/x509.h>
14 #include <openssl/bn.h>
15 #include <openssl/cms.h>
16 #include "internal/asn1_int.h"
17 #include "internal/evp_int.h"
18 #include "rsa_locl.h"
19
20 #ifndef OPENSSL_NO_CMS
21 static int rsa_cms_sign(CMS_SignerInfo *si);
22 static int rsa_cms_verify(CMS_SignerInfo *si);
23 static int rsa_cms_decrypt(CMS_RecipientInfo *ri);
24 static int rsa_cms_encrypt(CMS_RecipientInfo *ri);
25 #endif
26
27 static int rsa_pub_encode(X509_PUBKEY *pk, const EVP_PKEY *pkey)
28 {
29     unsigned char *penc = NULL;
30     int penclen;
31     penclen = i2d_RSAPublicKey(pkey->pkey.rsa, &penc);
32     if (penclen <= 0)
33         return 0;
34     if (X509_PUBKEY_set0_param(pk, OBJ_nid2obj(EVP_PKEY_RSA),
35                                V_ASN1_NULL, NULL, penc, penclen))
36         return 1;
37
38     OPENSSL_free(penc);
39     return 0;
40 }
41
42 static int rsa_pub_decode(EVP_PKEY *pkey, X509_PUBKEY *pubkey)
43 {
44     const unsigned char *p;
45     int pklen;
46     RSA *rsa = NULL;
47
48     if (!X509_PUBKEY_get0_param(NULL, &p, &pklen, NULL, pubkey))
49         return 0;
50     if ((rsa = d2i_RSAPublicKey(NULL, &p, pklen)) == NULL) {
51         RSAerr(RSA_F_RSA_PUB_DECODE, ERR_R_RSA_LIB);
52         return 0;
53     }
54     EVP_PKEY_assign_RSA(pkey, rsa);
55     return 1;
56 }
57
58 static int rsa_pub_cmp(const EVP_PKEY *a, const EVP_PKEY *b)
59 {
60     if (BN_cmp(b->pkey.rsa->n, a->pkey.rsa->n) != 0
61         || BN_cmp(b->pkey.rsa->e, a->pkey.rsa->e) != 0)
62         return 0;
63     return 1;
64 }
65
66 static int old_rsa_priv_decode(EVP_PKEY *pkey,
67                                const unsigned char **pder, int derlen)
68 {
69     RSA *rsa;
70
71     if ((rsa = d2i_RSAPrivateKey(NULL, pder, derlen)) == NULL) {
72         RSAerr(RSA_F_OLD_RSA_PRIV_DECODE, ERR_R_RSA_LIB);
73         return 0;
74     }
75     EVP_PKEY_assign_RSA(pkey, rsa);
76     return 1;
77 }
78
79 static int old_rsa_priv_encode(const EVP_PKEY *pkey, unsigned char **pder)
80 {
81     return i2d_RSAPrivateKey(pkey->pkey.rsa, pder);
82 }
83
84 static int rsa_priv_encode(PKCS8_PRIV_KEY_INFO *p8, const EVP_PKEY *pkey)
85 {
86     unsigned char *rk = NULL;
87     int rklen;
88     rklen = i2d_RSAPrivateKey(pkey->pkey.rsa, &rk);
89
90     if (rklen <= 0) {
91         RSAerr(RSA_F_RSA_PRIV_ENCODE, ERR_R_MALLOC_FAILURE);
92         return 0;
93     }
94
95     if (!PKCS8_pkey_set0(p8, OBJ_nid2obj(NID_rsaEncryption), 0,
96                          V_ASN1_NULL, NULL, rk, rklen)) {
97         RSAerr(RSA_F_RSA_PRIV_ENCODE, ERR_R_MALLOC_FAILURE);
98         return 0;
99     }
100
101     return 1;
102 }
103
104 static int rsa_priv_decode(EVP_PKEY *pkey, const PKCS8_PRIV_KEY_INFO *p8)
105 {
106     const unsigned char *p;
107     int pklen;
108     if (!PKCS8_pkey_get0(NULL, &p, &pklen, NULL, p8))
109         return 0;
110     return old_rsa_priv_decode(pkey, &p, pklen);
111 }
112
113 static int int_rsa_size(const EVP_PKEY *pkey)
114 {
115     return RSA_size(pkey->pkey.rsa);
116 }
117
118 static int rsa_bits(const EVP_PKEY *pkey)
119 {
120     return BN_num_bits(pkey->pkey.rsa->n);
121 }
122
123 static int rsa_security_bits(const EVP_PKEY *pkey)
124 {
125     return RSA_security_bits(pkey->pkey.rsa);
126 }
127
128 static void int_rsa_free(EVP_PKEY *pkey)
129 {
130     RSA_free(pkey->pkey.rsa);
131 }
132
133 static int do_rsa_print(BIO *bp, const RSA *x, int off, int priv)
134 {
135     char *str;
136     const char *s;
137     int ret = 0, mod_len = 0;
138
139     if (x->n != NULL)
140         mod_len = BN_num_bits(x->n);
141
142     if (!BIO_indent(bp, off, 128))
143         goto err;
144
145     if (priv && x->d) {
146         if (BIO_printf(bp, "Private-Key: (%d bit)\n", mod_len) <= 0)
147             goto err;
148         str = "modulus:";
149         s = "publicExponent:";
150     } else {
151         if (BIO_printf(bp, "Public-Key: (%d bit)\n", mod_len) <= 0)
152             goto err;
153         str = "Modulus:";
154         s = "Exponent:";
155     }
156     if (!ASN1_bn_print(bp, str, x->n, NULL, off))
157         goto err;
158     if (!ASN1_bn_print(bp, s, x->e, NULL, off))
159         goto err;
160     if (priv) {
161         if (!ASN1_bn_print(bp, "privateExponent:", x->d, NULL, off))
162             goto err;
163         if (!ASN1_bn_print(bp, "prime1:", x->p, NULL, off))
164             goto err;
165         if (!ASN1_bn_print(bp, "prime2:", x->q, NULL, off))
166             goto err;
167         if (!ASN1_bn_print(bp, "exponent1:", x->dmp1, NULL, off))
168             goto err;
169         if (!ASN1_bn_print(bp, "exponent2:", x->dmq1, NULL, off))
170             goto err;
171         if (!ASN1_bn_print(bp, "coefficient:", x->iqmp, NULL, off))
172             goto err;
173     }
174     ret = 1;
175  err:
176     return (ret);
177 }
178
179 static int rsa_pub_print(BIO *bp, const EVP_PKEY *pkey, int indent,
180                          ASN1_PCTX *ctx)
181 {
182     return do_rsa_print(bp, pkey->pkey.rsa, indent, 0);
183 }
184
185 static int rsa_priv_print(BIO *bp, const EVP_PKEY *pkey, int indent,
186                           ASN1_PCTX *ctx)
187 {
188     return do_rsa_print(bp, pkey->pkey.rsa, indent, 1);
189 }
190
191 static X509_ALGOR *rsa_mgf1_decode(X509_ALGOR *alg)
192 {
193     if (OBJ_obj2nid(alg->algorithm) != NID_mgf1)
194         return NULL;
195     return ASN1_TYPE_unpack_sequence(ASN1_ITEM_rptr(X509_ALGOR),
196                                      alg->parameter);
197 }
198
199 static RSA_PSS_PARAMS *rsa_pss_decode(const X509_ALGOR *alg)
200 {
201     RSA_PSS_PARAMS *pss;
202
203     pss = ASN1_TYPE_unpack_sequence(ASN1_ITEM_rptr(RSA_PSS_PARAMS),
204                                     alg->parameter);
205
206     if (pss == NULL)
207         return NULL;
208
209     if (pss->maskGenAlgorithm != NULL) {
210         pss->maskHash = rsa_mgf1_decode(pss->maskGenAlgorithm);
211         if (pss->maskHash == NULL) {
212             RSA_PSS_PARAMS_free(pss);
213             return NULL;
214         }
215     }
216
217     return pss;
218 }
219
220 static int rsa_pss_param_print(BIO *bp, RSA_PSS_PARAMS *pss, int indent)
221 {
222     int rv = 0;
223     if (!pss) {
224         if (BIO_puts(bp, " (INVALID PSS PARAMETERS)\n") <= 0)
225             return 0;
226         return 1;
227     }
228     if (BIO_puts(bp, "\n") <= 0)
229         goto err;
230     if (!BIO_indent(bp, indent, 128))
231         goto err;
232     if (BIO_puts(bp, "Hash Algorithm: ") <= 0)
233         goto err;
234
235     if (pss->hashAlgorithm) {
236         if (i2a_ASN1_OBJECT(bp, pss->hashAlgorithm->algorithm) <= 0)
237             goto err;
238     } else if (BIO_puts(bp, "sha1 (default)") <= 0)
239         goto err;
240
241     if (BIO_puts(bp, "\n") <= 0)
242         goto err;
243
244     if (!BIO_indent(bp, indent, 128))
245         goto err;
246
247     if (BIO_puts(bp, "Mask Algorithm: ") <= 0)
248         goto err;
249     if (pss->maskGenAlgorithm) {
250         if (i2a_ASN1_OBJECT(bp, pss->maskGenAlgorithm->algorithm) <= 0)
251             goto err;
252         if (BIO_puts(bp, " with ") <= 0)
253             goto err;
254         if (pss->maskHash) {
255             if (i2a_ASN1_OBJECT(bp, pss->maskHash->algorithm) <= 0)
256                 goto err;
257         } else if (BIO_puts(bp, "INVALID") <= 0)
258             goto err;
259     } else if (BIO_puts(bp, "mgf1 with sha1 (default)") <= 0)
260         goto err;
261     BIO_puts(bp, "\n");
262
263     if (!BIO_indent(bp, indent, 128))
264         goto err;
265     if (BIO_puts(bp, "Salt Length: 0x") <= 0)
266         goto err;
267     if (pss->saltLength) {
268         if (i2a_ASN1_INTEGER(bp, pss->saltLength) <= 0)
269             goto err;
270     } else if (BIO_puts(bp, "14 (default)") <= 0)
271         goto err;
272     BIO_puts(bp, "\n");
273
274     if (!BIO_indent(bp, indent, 128))
275         goto err;
276     if (BIO_puts(bp, "Trailer Field: 0x") <= 0)
277         goto err;
278     if (pss->trailerField) {
279         if (i2a_ASN1_INTEGER(bp, pss->trailerField) <= 0)
280             goto err;
281     } else if (BIO_puts(bp, "BC (default)") <= 0)
282         goto err;
283     BIO_puts(bp, "\n");
284
285     rv = 1;
286
287  err:
288     return rv;
289
290 }
291
292 static int rsa_sig_print(BIO *bp, const X509_ALGOR *sigalg,
293                          const ASN1_STRING *sig, int indent, ASN1_PCTX *pctx)
294 {
295     if (OBJ_obj2nid(sigalg->algorithm) == EVP_PKEY_RSA_PSS) {
296         int rv;
297         RSA_PSS_PARAMS *pss;
298         pss = rsa_pss_decode(sigalg);
299         rv = rsa_pss_param_print(bp, pss, indent);
300         RSA_PSS_PARAMS_free(pss);
301         if (!rv)
302             return 0;
303     } else if (!sig && BIO_puts(bp, "\n") <= 0)
304         return 0;
305     if (sig)
306         return X509_signature_dump(bp, sig, indent);
307     return 1;
308 }
309
310 static int rsa_pkey_ctrl(EVP_PKEY *pkey, int op, long arg1, void *arg2)
311 {
312     X509_ALGOR *alg = NULL;
313     switch (op) {
314
315     case ASN1_PKEY_CTRL_PKCS7_SIGN:
316         if (arg1 == 0)
317             PKCS7_SIGNER_INFO_get0_algs(arg2, NULL, NULL, &alg);
318         break;
319
320     case ASN1_PKEY_CTRL_PKCS7_ENCRYPT:
321         if (arg1 == 0)
322             PKCS7_RECIP_INFO_get0_alg(arg2, &alg);
323         break;
324 #ifndef OPENSSL_NO_CMS
325     case ASN1_PKEY_CTRL_CMS_SIGN:
326         if (arg1 == 0)
327             return rsa_cms_sign(arg2);
328         else if (arg1 == 1)
329             return rsa_cms_verify(arg2);
330         break;
331
332     case ASN1_PKEY_CTRL_CMS_ENVELOPE:
333         if (arg1 == 0)
334             return rsa_cms_encrypt(arg2);
335         else if (arg1 == 1)
336             return rsa_cms_decrypt(arg2);
337         break;
338
339     case ASN1_PKEY_CTRL_CMS_RI_TYPE:
340         *(int *)arg2 = CMS_RECIPINFO_TRANS;
341         return 1;
342 #endif
343
344     case ASN1_PKEY_CTRL_DEFAULT_MD_NID:
345         *(int *)arg2 = NID_sha256;
346         return 1;
347
348     default:
349         return -2;
350
351     }
352
353     if (alg)
354         X509_ALGOR_set0(alg, OBJ_nid2obj(NID_rsaEncryption), V_ASN1_NULL, 0);
355
356     return 1;
357
358 }
359
360 /* allocate and set algorithm ID from EVP_MD, default SHA1 */
361 static int rsa_md_to_algor(X509_ALGOR **palg, const EVP_MD *md)
362 {
363     if (EVP_MD_type(md) == NID_sha1)
364         return 1;
365     *palg = X509_ALGOR_new();
366     if (*palg == NULL)
367         return 0;
368     X509_ALGOR_set_md(*palg, md);
369     return 1;
370 }
371
372 /* Allocate and set MGF1 algorithm ID from EVP_MD */
373 static int rsa_md_to_mgf1(X509_ALGOR **palg, const EVP_MD *mgf1md)
374 {
375     X509_ALGOR *algtmp = NULL;
376     ASN1_STRING *stmp = NULL;
377     *palg = NULL;
378     if (EVP_MD_type(mgf1md) == NID_sha1)
379         return 1;
380     /* need to embed algorithm ID inside another */
381     if (!rsa_md_to_algor(&algtmp, mgf1md))
382         goto err;
383     if (!ASN1_item_pack(algtmp, ASN1_ITEM_rptr(X509_ALGOR), &stmp))
384          goto err;
385     *palg = X509_ALGOR_new();
386     if (*palg == NULL)
387         goto err;
388     X509_ALGOR_set0(*palg, OBJ_nid2obj(NID_mgf1), V_ASN1_SEQUENCE, stmp);
389     stmp = NULL;
390  err:
391     ASN1_STRING_free(stmp);
392     X509_ALGOR_free(algtmp);
393     if (*palg)
394         return 1;
395     return 0;
396 }
397
398 /* convert algorithm ID to EVP_MD, default SHA1 */
399 static const EVP_MD *rsa_algor_to_md(X509_ALGOR *alg)
400 {
401     const EVP_MD *md;
402     if (!alg)
403         return EVP_sha1();
404     md = EVP_get_digestbyobj(alg->algorithm);
405     if (md == NULL)
406         RSAerr(RSA_F_RSA_ALGOR_TO_MD, RSA_R_UNKNOWN_DIGEST);
407     return md;
408 }
409
410 /*
411  * Convert EVP_PKEY_CTX is PSS mode into corresponding algorithm parameter,
412  * suitable for setting an AlgorithmIdentifier.
413  */
414
415 static ASN1_STRING *rsa_ctx_to_pss(EVP_PKEY_CTX *pkctx)
416 {
417     const EVP_MD *sigmd, *mgf1md;
418     RSA_PSS_PARAMS *pss = NULL;
419     ASN1_STRING *os = NULL;
420     EVP_PKEY *pk = EVP_PKEY_CTX_get0_pkey(pkctx);
421     int saltlen, rv = 0;
422     if (EVP_PKEY_CTX_get_signature_md(pkctx, &sigmd) <= 0)
423         goto err;
424     if (EVP_PKEY_CTX_get_rsa_mgf1_md(pkctx, &mgf1md) <= 0)
425         goto err;
426     if (!EVP_PKEY_CTX_get_rsa_pss_saltlen(pkctx, &saltlen))
427         goto err;
428     if (saltlen == -1)
429         saltlen = EVP_MD_size(sigmd);
430     else if (saltlen == -2) {
431         saltlen = EVP_PKEY_size(pk) - EVP_MD_size(sigmd) - 2;
432         if (((EVP_PKEY_bits(pk) - 1) & 0x7) == 0)
433             saltlen--;
434     }
435     pss = RSA_PSS_PARAMS_new();
436     if (pss == NULL)
437         goto err;
438     if (saltlen != 20) {
439         pss->saltLength = ASN1_INTEGER_new();
440         if (pss->saltLength == NULL)
441             goto err;
442         if (!ASN1_INTEGER_set(pss->saltLength, saltlen))
443             goto err;
444     }
445     if (!rsa_md_to_algor(&pss->hashAlgorithm, sigmd))
446         goto err;
447     if (!rsa_md_to_mgf1(&pss->maskGenAlgorithm, mgf1md))
448         goto err;
449     /* Finally create string with pss parameter encoding. */
450     if (!ASN1_item_pack(pss, ASN1_ITEM_rptr(RSA_PSS_PARAMS), &os))
451          goto err;
452     rv = 1;
453  err:
454     RSA_PSS_PARAMS_free(pss);
455     if (rv)
456         return os;
457     ASN1_STRING_free(os);
458     return NULL;
459 }
460
461 /*
462  * From PSS AlgorithmIdentifier set public key parameters. If pkey isn't NULL
463  * then the EVP_MD_CTX is setup and initialised. If it is NULL parameters are
464  * passed to pkctx instead.
465  */
466
467 static int rsa_pss_to_ctx(EVP_MD_CTX *ctx, EVP_PKEY_CTX *pkctx,
468                           X509_ALGOR *sigalg, EVP_PKEY *pkey)
469 {
470     int rv = -1;
471     int saltlen;
472     const EVP_MD *mgf1md = NULL, *md = NULL;
473     RSA_PSS_PARAMS *pss;
474     /* Sanity check: make sure it is PSS */
475     if (OBJ_obj2nid(sigalg->algorithm) != EVP_PKEY_RSA_PSS) {
476         RSAerr(RSA_F_RSA_PSS_TO_CTX, RSA_R_UNSUPPORTED_SIGNATURE_TYPE);
477         return -1;
478     }
479     /* Decode PSS parameters */
480     pss = rsa_pss_decode(sigalg);
481
482     if (pss == NULL) {
483         RSAerr(RSA_F_RSA_PSS_TO_CTX, RSA_R_INVALID_PSS_PARAMETERS);
484         goto err;
485     }
486     mgf1md = rsa_algor_to_md(pss->maskHash);
487     if (!mgf1md)
488         goto err;
489     md = rsa_algor_to_md(pss->hashAlgorithm);
490     if (!md)
491         goto err;
492
493     if (pss->saltLength) {
494         saltlen = ASN1_INTEGER_get(pss->saltLength);
495
496         /*
497          * Could perform more salt length sanity checks but the main RSA
498          * routines will trap other invalid values anyway.
499          */
500         if (saltlen < 0) {
501             RSAerr(RSA_F_RSA_PSS_TO_CTX, RSA_R_INVALID_SALT_LENGTH);
502             goto err;
503         }
504     } else
505         saltlen = 20;
506
507     /*
508      * low-level routines support only trailer field 0xbc (value 1) and
509      * PKCS#1 says we should reject any other value anyway.
510      */
511     if (pss->trailerField && ASN1_INTEGER_get(pss->trailerField) != 1) {
512         RSAerr(RSA_F_RSA_PSS_TO_CTX, RSA_R_INVALID_TRAILER);
513         goto err;
514     }
515
516     /* We have all parameters now set up context */
517
518     if (pkey) {
519         if (!EVP_DigestVerifyInit(ctx, &pkctx, md, NULL, pkey))
520             goto err;
521     } else {
522         const EVP_MD *checkmd;
523         if (EVP_PKEY_CTX_get_signature_md(pkctx, &checkmd) <= 0)
524             goto err;
525         if (EVP_MD_type(md) != EVP_MD_type(checkmd)) {
526             RSAerr(RSA_F_RSA_PSS_TO_CTX, RSA_R_DIGEST_DOES_NOT_MATCH);
527             goto err;
528         }
529     }
530
531     if (EVP_PKEY_CTX_set_rsa_padding(pkctx, RSA_PKCS1_PSS_PADDING) <= 0)
532         goto err;
533
534     if (EVP_PKEY_CTX_set_rsa_pss_saltlen(pkctx, saltlen) <= 0)
535         goto err;
536
537     if (EVP_PKEY_CTX_set_rsa_mgf1_md(pkctx, mgf1md) <= 0)
538         goto err;
539     /* Carry on */
540     rv = 1;
541
542  err:
543     RSA_PSS_PARAMS_free(pss);
544     return rv;
545 }
546
547 #ifndef OPENSSL_NO_CMS
548 static int rsa_cms_verify(CMS_SignerInfo *si)
549 {
550     int nid, nid2;
551     X509_ALGOR *alg;
552     EVP_PKEY_CTX *pkctx = CMS_SignerInfo_get0_pkey_ctx(si);
553     CMS_SignerInfo_get0_algs(si, NULL, NULL, NULL, &alg);
554     nid = OBJ_obj2nid(alg->algorithm);
555     if (nid == NID_rsaEncryption)
556         return 1;
557     if (nid == EVP_PKEY_RSA_PSS)
558         return rsa_pss_to_ctx(NULL, pkctx, alg, NULL);
559     /* Workaround for some implementation that use a signature OID */
560     if (OBJ_find_sigid_algs(nid, NULL, &nid2)) {
561         if (nid2 == NID_rsaEncryption)
562             return 1;
563     }
564     return 0;
565 }
566 #endif
567
568 /*
569  * Customised RSA item verification routine. This is called when a signature
570  * is encountered requiring special handling. We currently only handle PSS.
571  */
572
573 static int rsa_item_verify(EVP_MD_CTX *ctx, const ASN1_ITEM *it, void *asn,
574                            X509_ALGOR *sigalg, ASN1_BIT_STRING *sig,
575                            EVP_PKEY *pkey)
576 {
577     /* Sanity check: make sure it is PSS */
578     if (OBJ_obj2nid(sigalg->algorithm) != EVP_PKEY_RSA_PSS) {
579         RSAerr(RSA_F_RSA_ITEM_VERIFY, RSA_R_UNSUPPORTED_SIGNATURE_TYPE);
580         return -1;
581     }
582     if (rsa_pss_to_ctx(ctx, NULL, sigalg, pkey) > 0) {
583         /* Carry on */
584         return 2;
585     }
586     return -1;
587 }
588
589 #ifndef OPENSSL_NO_CMS
590 static int rsa_cms_sign(CMS_SignerInfo *si)
591 {
592     int pad_mode = RSA_PKCS1_PADDING;
593     X509_ALGOR *alg;
594     EVP_PKEY_CTX *pkctx = CMS_SignerInfo_get0_pkey_ctx(si);
595     ASN1_STRING *os = NULL;
596     CMS_SignerInfo_get0_algs(si, NULL, NULL, NULL, &alg);
597     if (pkctx) {
598         if (EVP_PKEY_CTX_get_rsa_padding(pkctx, &pad_mode) <= 0)
599             return 0;
600     }
601     if (pad_mode == RSA_PKCS1_PADDING) {
602         X509_ALGOR_set0(alg, OBJ_nid2obj(NID_rsaEncryption), V_ASN1_NULL, 0);
603         return 1;
604     }
605     /* We don't support it */
606     if (pad_mode != RSA_PKCS1_PSS_PADDING)
607         return 0;
608     os = rsa_ctx_to_pss(pkctx);
609     if (!os)
610         return 0;
611     X509_ALGOR_set0(alg, OBJ_nid2obj(EVP_PKEY_RSA_PSS), V_ASN1_SEQUENCE, os);
612     return 1;
613 }
614 #endif
615
616 static int rsa_item_sign(EVP_MD_CTX *ctx, const ASN1_ITEM *it, void *asn,
617                          X509_ALGOR *alg1, X509_ALGOR *alg2,
618                          ASN1_BIT_STRING *sig)
619 {
620     int pad_mode;
621     EVP_PKEY_CTX *pkctx = EVP_MD_CTX_pkey_ctx(ctx);
622     if (EVP_PKEY_CTX_get_rsa_padding(pkctx, &pad_mode) <= 0)
623         return 0;
624     if (pad_mode == RSA_PKCS1_PADDING)
625         return 2;
626     if (pad_mode == RSA_PKCS1_PSS_PADDING) {
627         ASN1_STRING *os1 = NULL;
628         os1 = rsa_ctx_to_pss(pkctx);
629         if (!os1)
630             return 0;
631         /* Duplicate parameters if we have to */
632         if (alg2) {
633             ASN1_STRING *os2 = ASN1_STRING_dup(os1);
634             if (!os2) {
635                 ASN1_STRING_free(os1);
636                 return 0;
637             }
638             X509_ALGOR_set0(alg2, OBJ_nid2obj(EVP_PKEY_RSA_PSS),
639                             V_ASN1_SEQUENCE, os2);
640         }
641         X509_ALGOR_set0(alg1, OBJ_nid2obj(EVP_PKEY_RSA_PSS),
642                         V_ASN1_SEQUENCE, os1);
643         return 3;
644     }
645     return 2;
646 }
647
648 #ifndef OPENSSL_NO_CMS
649 static RSA_OAEP_PARAMS *rsa_oaep_decode(const X509_ALGOR *alg)
650 {
651     RSA_OAEP_PARAMS *oaep;
652
653     oaep = ASN1_TYPE_unpack_sequence(ASN1_ITEM_rptr(RSA_OAEP_PARAMS),
654                                     alg->parameter);
655
656     if (oaep == NULL)
657         return NULL;
658
659     if (oaep->maskGenFunc != NULL) {
660         oaep->maskHash = rsa_mgf1_decode(oaep->maskGenFunc);
661         if (oaep->maskHash == NULL) {
662             RSA_OAEP_PARAMS_free(oaep);
663             return NULL;
664         }
665     }
666     return oaep;
667 }
668
669 static int rsa_cms_decrypt(CMS_RecipientInfo *ri)
670 {
671     EVP_PKEY_CTX *pkctx;
672     X509_ALGOR *cmsalg;
673     int nid;
674     int rv = -1;
675     unsigned char *label = NULL;
676     int labellen = 0;
677     const EVP_MD *mgf1md = NULL, *md = NULL;
678     RSA_OAEP_PARAMS *oaep;
679     pkctx = CMS_RecipientInfo_get0_pkey_ctx(ri);
680     if (!pkctx)
681         return 0;
682     if (!CMS_RecipientInfo_ktri_get0_algs(ri, NULL, NULL, &cmsalg))
683         return -1;
684     nid = OBJ_obj2nid(cmsalg->algorithm);
685     if (nid == NID_rsaEncryption)
686         return 1;
687     if (nid != NID_rsaesOaep) {
688         RSAerr(RSA_F_RSA_CMS_DECRYPT, RSA_R_UNSUPPORTED_ENCRYPTION_TYPE);
689         return -1;
690     }
691     /* Decode OAEP parameters */
692     oaep = rsa_oaep_decode(cmsalg);
693
694     if (oaep == NULL) {
695         RSAerr(RSA_F_RSA_CMS_DECRYPT, RSA_R_INVALID_OAEP_PARAMETERS);
696         goto err;
697     }
698
699     mgf1md = rsa_algor_to_md(oaep->maskHash);
700     if (!mgf1md)
701         goto err;
702     md = rsa_algor_to_md(oaep->hashFunc);
703     if (!md)
704         goto err;
705
706     if (oaep->pSourceFunc) {
707         X509_ALGOR *plab = oaep->pSourceFunc;
708         if (OBJ_obj2nid(plab->algorithm) != NID_pSpecified) {
709             RSAerr(RSA_F_RSA_CMS_DECRYPT, RSA_R_UNSUPPORTED_LABEL_SOURCE);
710             goto err;
711         }
712         if (plab->parameter->type != V_ASN1_OCTET_STRING) {
713             RSAerr(RSA_F_RSA_CMS_DECRYPT, RSA_R_INVALID_LABEL);
714             goto err;
715         }
716
717         label = plab->parameter->value.octet_string->data;
718         /* Stop label being freed when OAEP parameters are freed */
719         plab->parameter->value.octet_string->data = NULL;
720         labellen = plab->parameter->value.octet_string->length;
721     }
722
723     if (EVP_PKEY_CTX_set_rsa_padding(pkctx, RSA_PKCS1_OAEP_PADDING) <= 0)
724         goto err;
725     if (EVP_PKEY_CTX_set_rsa_oaep_md(pkctx, md) <= 0)
726         goto err;
727     if (EVP_PKEY_CTX_set_rsa_mgf1_md(pkctx, mgf1md) <= 0)
728         goto err;
729     if (EVP_PKEY_CTX_set0_rsa_oaep_label(pkctx, label, labellen) <= 0)
730         goto err;
731     /* Carry on */
732     rv = 1;
733
734  err:
735     RSA_OAEP_PARAMS_free(oaep);
736     return rv;
737 }
738
739 static int rsa_cms_encrypt(CMS_RecipientInfo *ri)
740 {
741     const EVP_MD *md, *mgf1md;
742     RSA_OAEP_PARAMS *oaep = NULL;
743     ASN1_STRING *os = NULL;
744     X509_ALGOR *alg;
745     EVP_PKEY_CTX *pkctx = CMS_RecipientInfo_get0_pkey_ctx(ri);
746     int pad_mode = RSA_PKCS1_PADDING, rv = 0, labellen;
747     unsigned char *label;
748     CMS_RecipientInfo_ktri_get0_algs(ri, NULL, NULL, &alg);
749     if (pkctx) {
750         if (EVP_PKEY_CTX_get_rsa_padding(pkctx, &pad_mode) <= 0)
751             return 0;
752     }
753     if (pad_mode == RSA_PKCS1_PADDING) {
754         X509_ALGOR_set0(alg, OBJ_nid2obj(NID_rsaEncryption), V_ASN1_NULL, 0);
755         return 1;
756     }
757     /* Not supported */
758     if (pad_mode != RSA_PKCS1_OAEP_PADDING)
759         return 0;
760     if (EVP_PKEY_CTX_get_rsa_oaep_md(pkctx, &md) <= 0)
761         goto err;
762     if (EVP_PKEY_CTX_get_rsa_mgf1_md(pkctx, &mgf1md) <= 0)
763         goto err;
764     labellen = EVP_PKEY_CTX_get0_rsa_oaep_label(pkctx, &label);
765     if (labellen < 0)
766         goto err;
767     oaep = RSA_OAEP_PARAMS_new();
768     if (oaep == NULL)
769         goto err;
770     if (!rsa_md_to_algor(&oaep->hashFunc, md))
771         goto err;
772     if (!rsa_md_to_mgf1(&oaep->maskGenFunc, mgf1md))
773         goto err;
774     if (labellen > 0) {
775         ASN1_OCTET_STRING *los;
776         oaep->pSourceFunc = X509_ALGOR_new();
777         if (oaep->pSourceFunc == NULL)
778             goto err;
779         los = ASN1_OCTET_STRING_new();
780         if (los == NULL)
781             goto err;
782         if (!ASN1_OCTET_STRING_set(los, label, labellen)) {
783             ASN1_OCTET_STRING_free(los);
784             goto err;
785         }
786         X509_ALGOR_set0(oaep->pSourceFunc, OBJ_nid2obj(NID_pSpecified),
787                         V_ASN1_OCTET_STRING, los);
788     }
789     /* create string with pss parameter encoding. */
790     if (!ASN1_item_pack(oaep, ASN1_ITEM_rptr(RSA_OAEP_PARAMS), &os))
791          goto err;
792     X509_ALGOR_set0(alg, OBJ_nid2obj(NID_rsaesOaep), V_ASN1_SEQUENCE, os);
793     os = NULL;
794     rv = 1;
795  err:
796     RSA_OAEP_PARAMS_free(oaep);
797     ASN1_STRING_free(os);
798     return rv;
799 }
800 #endif
801
802 const EVP_PKEY_ASN1_METHOD rsa_asn1_meths[2] = {
803     {
804      EVP_PKEY_RSA,
805      EVP_PKEY_RSA,
806      ASN1_PKEY_SIGPARAM_NULL,
807
808      "RSA",
809      "OpenSSL RSA method",
810
811      rsa_pub_decode,
812      rsa_pub_encode,
813      rsa_pub_cmp,
814      rsa_pub_print,
815
816      rsa_priv_decode,
817      rsa_priv_encode,
818      rsa_priv_print,
819
820      int_rsa_size,
821      rsa_bits,
822      rsa_security_bits,
823
824      0, 0, 0, 0, 0, 0,
825
826      rsa_sig_print,
827      int_rsa_free,
828      rsa_pkey_ctrl,
829      old_rsa_priv_decode,
830      old_rsa_priv_encode,
831      rsa_item_verify,
832      rsa_item_sign},
833
834     {
835      EVP_PKEY_RSA2,
836      EVP_PKEY_RSA,
837      ASN1_PKEY_ALIAS}
838 };