162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-or-later
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * RSA key extract helper
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (c) 2015, Intel Corporation
662306a36Sopenharmony_ci * Authors: Tadeusz Struk <tadeusz.struk@intel.com>
762306a36Sopenharmony_ci */
862306a36Sopenharmony_ci#include <linux/kernel.h>
962306a36Sopenharmony_ci#include <linux/export.h>
1062306a36Sopenharmony_ci#include <linux/err.h>
1162306a36Sopenharmony_ci#include <linux/fips.h>
1262306a36Sopenharmony_ci#include <crypto/internal/rsa.h>
1362306a36Sopenharmony_ci#include "rsapubkey.asn1.h"
1462306a36Sopenharmony_ci#include "rsaprivkey.asn1.h"
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_ciint rsa_get_n(void *context, size_t hdrlen, unsigned char tag,
1762306a36Sopenharmony_ci	      const void *value, size_t vlen)
1862306a36Sopenharmony_ci{
1962306a36Sopenharmony_ci	struct rsa_key *key = context;
2062306a36Sopenharmony_ci	const u8 *ptr = value;
2162306a36Sopenharmony_ci	size_t n_sz = vlen;
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_ci	/* invalid key provided */
2462306a36Sopenharmony_ci	if (!value || !vlen)
2562306a36Sopenharmony_ci		return -EINVAL;
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_ci	if (fips_enabled) {
2862306a36Sopenharmony_ci		while (n_sz && !*ptr) {
2962306a36Sopenharmony_ci			ptr++;
3062306a36Sopenharmony_ci			n_sz--;
3162306a36Sopenharmony_ci		}
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_ci		/* In FIPS mode only allow key size 2K and higher */
3462306a36Sopenharmony_ci		if (n_sz < 256) {
3562306a36Sopenharmony_ci			pr_err("RSA: key size not allowed in FIPS mode\n");
3662306a36Sopenharmony_ci			return -EINVAL;
3762306a36Sopenharmony_ci		}
3862306a36Sopenharmony_ci	}
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_ci	key->n = value;
4162306a36Sopenharmony_ci	key->n_sz = vlen;
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_ci	return 0;
4462306a36Sopenharmony_ci}
4562306a36Sopenharmony_ci
4662306a36Sopenharmony_ciint rsa_get_e(void *context, size_t hdrlen, unsigned char tag,
4762306a36Sopenharmony_ci	      const void *value, size_t vlen)
4862306a36Sopenharmony_ci{
4962306a36Sopenharmony_ci	struct rsa_key *key = context;
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_ci	/* invalid key provided */
5262306a36Sopenharmony_ci	if (!value || !key->n_sz || !vlen || vlen > key->n_sz)
5362306a36Sopenharmony_ci		return -EINVAL;
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci	key->e = value;
5662306a36Sopenharmony_ci	key->e_sz = vlen;
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci	return 0;
5962306a36Sopenharmony_ci}
6062306a36Sopenharmony_ci
6162306a36Sopenharmony_ciint rsa_get_d(void *context, size_t hdrlen, unsigned char tag,
6262306a36Sopenharmony_ci	      const void *value, size_t vlen)
6362306a36Sopenharmony_ci{
6462306a36Sopenharmony_ci	struct rsa_key *key = context;
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_ci	/* invalid key provided */
6762306a36Sopenharmony_ci	if (!value || !key->n_sz || !vlen || vlen > key->n_sz)
6862306a36Sopenharmony_ci		return -EINVAL;
6962306a36Sopenharmony_ci
7062306a36Sopenharmony_ci	key->d = value;
7162306a36Sopenharmony_ci	key->d_sz = vlen;
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_ci	return 0;
7462306a36Sopenharmony_ci}
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ciint rsa_get_p(void *context, size_t hdrlen, unsigned char tag,
7762306a36Sopenharmony_ci	      const void *value, size_t vlen)
7862306a36Sopenharmony_ci{
7962306a36Sopenharmony_ci	struct rsa_key *key = context;
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ci	/* invalid key provided */
8262306a36Sopenharmony_ci	if (!value || !vlen || vlen > key->n_sz)
8362306a36Sopenharmony_ci		return -EINVAL;
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci	key->p = value;
8662306a36Sopenharmony_ci	key->p_sz = vlen;
8762306a36Sopenharmony_ci
8862306a36Sopenharmony_ci	return 0;
8962306a36Sopenharmony_ci}
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_ciint rsa_get_q(void *context, size_t hdrlen, unsigned char tag,
9262306a36Sopenharmony_ci	      const void *value, size_t vlen)
9362306a36Sopenharmony_ci{
9462306a36Sopenharmony_ci	struct rsa_key *key = context;
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_ci	/* invalid key provided */
9762306a36Sopenharmony_ci	if (!value || !vlen || vlen > key->n_sz)
9862306a36Sopenharmony_ci		return -EINVAL;
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_ci	key->q = value;
10162306a36Sopenharmony_ci	key->q_sz = vlen;
10262306a36Sopenharmony_ci
10362306a36Sopenharmony_ci	return 0;
10462306a36Sopenharmony_ci}
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_ciint rsa_get_dp(void *context, size_t hdrlen, unsigned char tag,
10762306a36Sopenharmony_ci	       const void *value, size_t vlen)
10862306a36Sopenharmony_ci{
10962306a36Sopenharmony_ci	struct rsa_key *key = context;
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_ci	/* invalid key provided */
11262306a36Sopenharmony_ci	if (!value || !vlen || vlen > key->n_sz)
11362306a36Sopenharmony_ci		return -EINVAL;
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_ci	key->dp = value;
11662306a36Sopenharmony_ci	key->dp_sz = vlen;
11762306a36Sopenharmony_ci
11862306a36Sopenharmony_ci	return 0;
11962306a36Sopenharmony_ci}
12062306a36Sopenharmony_ci
12162306a36Sopenharmony_ciint rsa_get_dq(void *context, size_t hdrlen, unsigned char tag,
12262306a36Sopenharmony_ci	       const void *value, size_t vlen)
12362306a36Sopenharmony_ci{
12462306a36Sopenharmony_ci	struct rsa_key *key = context;
12562306a36Sopenharmony_ci
12662306a36Sopenharmony_ci	/* invalid key provided */
12762306a36Sopenharmony_ci	if (!value || !vlen || vlen > key->n_sz)
12862306a36Sopenharmony_ci		return -EINVAL;
12962306a36Sopenharmony_ci
13062306a36Sopenharmony_ci	key->dq = value;
13162306a36Sopenharmony_ci	key->dq_sz = vlen;
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_ci	return 0;
13462306a36Sopenharmony_ci}
13562306a36Sopenharmony_ci
13662306a36Sopenharmony_ciint rsa_get_qinv(void *context, size_t hdrlen, unsigned char tag,
13762306a36Sopenharmony_ci		 const void *value, size_t vlen)
13862306a36Sopenharmony_ci{
13962306a36Sopenharmony_ci	struct rsa_key *key = context;
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_ci	/* invalid key provided */
14262306a36Sopenharmony_ci	if (!value || !vlen || vlen > key->n_sz)
14362306a36Sopenharmony_ci		return -EINVAL;
14462306a36Sopenharmony_ci
14562306a36Sopenharmony_ci	key->qinv = value;
14662306a36Sopenharmony_ci	key->qinv_sz = vlen;
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci	return 0;
14962306a36Sopenharmony_ci}
15062306a36Sopenharmony_ci
15162306a36Sopenharmony_ci/**
15262306a36Sopenharmony_ci * rsa_parse_pub_key() - decodes the BER encoded buffer and stores in the
15362306a36Sopenharmony_ci *                       provided struct rsa_key, pointers to the raw key as is,
15462306a36Sopenharmony_ci *                       so that the caller can copy it or MPI parse it, etc.
15562306a36Sopenharmony_ci *
15662306a36Sopenharmony_ci * @rsa_key:	struct rsa_key key representation
15762306a36Sopenharmony_ci * @key:	key in BER format
15862306a36Sopenharmony_ci * @key_len:	length of key
15962306a36Sopenharmony_ci *
16062306a36Sopenharmony_ci * Return:	0 on success or error code in case of error
16162306a36Sopenharmony_ci */
16262306a36Sopenharmony_ciint rsa_parse_pub_key(struct rsa_key *rsa_key, const void *key,
16362306a36Sopenharmony_ci		      unsigned int key_len)
16462306a36Sopenharmony_ci{
16562306a36Sopenharmony_ci	return asn1_ber_decoder(&rsapubkey_decoder, rsa_key, key, key_len);
16662306a36Sopenharmony_ci}
16762306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(rsa_parse_pub_key);
16862306a36Sopenharmony_ci
16962306a36Sopenharmony_ci/**
17062306a36Sopenharmony_ci * rsa_parse_priv_key() - decodes the BER encoded buffer and stores in the
17162306a36Sopenharmony_ci *                        provided struct rsa_key, pointers to the raw key
17262306a36Sopenharmony_ci *                        as is, so that the caller can copy it or MPI parse it,
17362306a36Sopenharmony_ci *                        etc.
17462306a36Sopenharmony_ci *
17562306a36Sopenharmony_ci * @rsa_key:	struct rsa_key key representation
17662306a36Sopenharmony_ci * @key:	key in BER format
17762306a36Sopenharmony_ci * @key_len:	length of key
17862306a36Sopenharmony_ci *
17962306a36Sopenharmony_ci * Return:	0 on success or error code in case of error
18062306a36Sopenharmony_ci */
18162306a36Sopenharmony_ciint rsa_parse_priv_key(struct rsa_key *rsa_key, const void *key,
18262306a36Sopenharmony_ci		       unsigned int key_len)
18362306a36Sopenharmony_ci{
18462306a36Sopenharmony_ci	return asn1_ber_decoder(&rsaprivkey_decoder, rsa_key, key, key_len);
18562306a36Sopenharmony_ci}
18662306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(rsa_parse_priv_key);
187