rsa.c 7.49 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12
/* RSA asymmetric public-key algorithm [RFC3447]
 *
 * Copyright (c) 2015, Intel Corporation
 * Authors: Tadeusz Struk <tadeusz.struk@intel.com>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public Licence
 * as published by the Free Software Foundation; either version
 * 2 of the Licence, or (at your option) any later version.
 */

#include <linux/module.h>
13
#include <linux/mpi.h>
14 15 16
#include <crypto/internal/rsa.h>
#include <crypto/internal/akcipher.h>
#include <crypto/akcipher.h>
17
#include <crypto/algapi.h>
18

19 20 21 22 23 24
struct rsa_mpi_key {
	MPI n;
	MPI e;
	MPI d;
};

25 26 27 28
/*
 * RSAEP function [RFC3447 sec 5.1.1]
 * c = m^e mod n;
 */
29
static int _rsa_enc(const struct rsa_mpi_key *key, MPI c, MPI m)
30 31 32 33 34 35 36 37 38 39 40 41 42
{
	/* (1) Validate 0 <= m < n */
	if (mpi_cmp_ui(m, 0) < 0 || mpi_cmp(m, key->n) >= 0)
		return -EINVAL;

	/* (2) c = m^e mod n */
	return mpi_powm(c, m, key->e, key->n);
}

/*
 * RSADP function [RFC3447 sec 5.1.2]
 * m = c^d mod n;
 */
43
static int _rsa_dec(const struct rsa_mpi_key *key, MPI m, MPI c)
44 45 46 47 48 49 50 51 52 53 54 55 56
{
	/* (1) Validate 0 <= c < n */
	if (mpi_cmp_ui(c, 0) < 0 || mpi_cmp(c, key->n) >= 0)
		return -EINVAL;

	/* (2) m = c^d mod n */
	return mpi_powm(m, c, key->d, key->n);
}

/*
 * RSASP1 function [RFC3447 sec 5.2.1]
 * s = m^d mod n
 */
57
static int _rsa_sign(const struct rsa_mpi_key *key, MPI s, MPI m)
58 59 60 61 62 63 64 65 66 67 68 69 70
{
	/* (1) Validate 0 <= m < n */
	if (mpi_cmp_ui(m, 0) < 0 || mpi_cmp(m, key->n) >= 0)
		return -EINVAL;

	/* (2) s = m^d mod n */
	return mpi_powm(s, m, key->d, key->n);
}

/*
 * RSAVP1 function [RFC3447 sec 5.2.2]
 * m = s^e mod n;
 */
71
static int _rsa_verify(const struct rsa_mpi_key *key, MPI m, MPI s)
72 73 74 75 76 77 78 79 80
{
	/* (1) Validate 0 <= s < n */
	if (mpi_cmp_ui(s, 0) < 0 || mpi_cmp(s, key->n) >= 0)
		return -EINVAL;

	/* (2) m = s^e mod n */
	return mpi_powm(m, s, key->e, key->n);
}

81
static inline struct rsa_mpi_key *rsa_get_key(struct crypto_akcipher *tfm)
82 83 84 85 86 87 88
{
	return akcipher_tfm_ctx(tfm);
}

static int rsa_enc(struct akcipher_request *req)
{
	struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
89
	const struct rsa_mpi_key *pkey = rsa_get_key(tfm);
90 91 92 93 94 95 96 97 98 99 100 101
	MPI m, c = mpi_alloc(0);
	int ret = 0;
	int sign;

	if (!c)
		return -ENOMEM;

	if (unlikely(!pkey->n || !pkey->e)) {
		ret = -EINVAL;
		goto err_free_c;
	}

102 103 104
	ret = -ENOMEM;
	m = mpi_read_raw_from_sgl(req->src, req->src_len);
	if (!m)
105 106 107 108 109 110
		goto err_free_c;

	ret = _rsa_enc(pkey, c, m);
	if (ret)
		goto err_free_m;

111
	ret = mpi_write_to_sgl(c, req->dst, req->dst_len, &sign);
112 113 114
	if (ret)
		goto err_free_m;

115
	if (sign < 0)
116 117 118 119 120 121 122 123 124 125 126 127
		ret = -EBADMSG;

err_free_m:
	mpi_free(m);
err_free_c:
	mpi_free(c);
	return ret;
}

static int rsa_dec(struct akcipher_request *req)
{
	struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
128
	const struct rsa_mpi_key *pkey = rsa_get_key(tfm);
129 130 131 132 133 134 135 136 137 138 139 140
	MPI c, m = mpi_alloc(0);
	int ret = 0;
	int sign;

	if (!m)
		return -ENOMEM;

	if (unlikely(!pkey->n || !pkey->d)) {
		ret = -EINVAL;
		goto err_free_m;
	}

141 142 143
	ret = -ENOMEM;
	c = mpi_read_raw_from_sgl(req->src, req->src_len);
	if (!c)
144 145 146 147 148 149
		goto err_free_m;

	ret = _rsa_dec(pkey, m, c);
	if (ret)
		goto err_free_c;

150
	ret = mpi_write_to_sgl(m, req->dst, req->dst_len, &sign);
151 152 153
	if (ret)
		goto err_free_c;

154
	if (sign < 0)
155 156 157 158 159 160 161 162 163 164 165
		ret = -EBADMSG;
err_free_c:
	mpi_free(c);
err_free_m:
	mpi_free(m);
	return ret;
}

static int rsa_sign(struct akcipher_request *req)
{
	struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
166
	const struct rsa_mpi_key *pkey = rsa_get_key(tfm);
167 168 169 170 171 172 173 174 175 176 177 178
	MPI m, s = mpi_alloc(0);
	int ret = 0;
	int sign;

	if (!s)
		return -ENOMEM;

	if (unlikely(!pkey->n || !pkey->d)) {
		ret = -EINVAL;
		goto err_free_s;
	}

179 180 181
	ret = -ENOMEM;
	m = mpi_read_raw_from_sgl(req->src, req->src_len);
	if (!m)
182 183 184 185 186 187
		goto err_free_s;

	ret = _rsa_sign(pkey, s, m);
	if (ret)
		goto err_free_m;

188
	ret = mpi_write_to_sgl(s, req->dst, req->dst_len, &sign);
189 190 191
	if (ret)
		goto err_free_m;

192
	if (sign < 0)
193 194 195 196 197 198 199 200 201 202 203 204
		ret = -EBADMSG;

err_free_m:
	mpi_free(m);
err_free_s:
	mpi_free(s);
	return ret;
}

static int rsa_verify(struct akcipher_request *req)
{
	struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
205
	const struct rsa_mpi_key *pkey = rsa_get_key(tfm);
206 207 208 209 210 211 212 213 214 215 216 217
	MPI s, m = mpi_alloc(0);
	int ret = 0;
	int sign;

	if (!m)
		return -ENOMEM;

	if (unlikely(!pkey->n || !pkey->e)) {
		ret = -EINVAL;
		goto err_free_m;
	}

218 219
	ret = -ENOMEM;
	s = mpi_read_raw_from_sgl(req->src, req->src_len);
220 221 222 223 224 225 226 227 228
	if (!s) {
		ret = -ENOMEM;
		goto err_free_m;
	}

	ret = _rsa_verify(pkey, m, s);
	if (ret)
		goto err_free_s;

229
	ret = mpi_write_to_sgl(m, req->dst, req->dst_len, &sign);
230 231 232
	if (ret)
		goto err_free_s;

233
	if (sign < 0)
234 235 236 237 238 239 240 241 242
		ret = -EBADMSG;

err_free_s:
	mpi_free(s);
err_free_m:
	mpi_free(m);
	return ret;
}

243 244 245 246 247 248 249 250 251 252
static void rsa_free_mpi_key(struct rsa_mpi_key *key)
{
	mpi_free(key->d);
	mpi_free(key->e);
	mpi_free(key->n);
	key->d = NULL;
	key->e = NULL;
	key->n = NULL;
}

253 254 255 256 257 258 259 260 261 262 263 264 265 266 267
static int rsa_check_key_length(unsigned int len)
{
	switch (len) {
	case 512:
	case 1024:
	case 1536:
	case 2048:
	case 3072:
	case 4096:
		return 0;
	}

	return -EINVAL;
}

268 269
static int rsa_set_pub_key(struct crypto_akcipher *tfm, const void *key,
			   unsigned int keylen)
270
{
271 272
	struct rsa_mpi_key *mpi_key = akcipher_tfm_ctx(tfm);
	struct rsa_key raw_key = {0};
273
	int ret;
274

275 276 277 278
	/* Free the old MPI key if any */
	rsa_free_mpi_key(mpi_key);

	ret = rsa_parse_pub_key(&raw_key, key, keylen);
279 280 281
	if (ret)
		return ret;

282 283 284 285 286 287 288 289 290 291 292
	mpi_key->e = mpi_read_raw_data(raw_key.e, raw_key.e_sz);
	if (!mpi_key->e)
		goto err;

	mpi_key->n = mpi_read_raw_data(raw_key.n, raw_key.n_sz);
	if (!mpi_key->n)
		goto err;

	if (rsa_check_key_length(mpi_get_size(mpi_key->n) << 3)) {
		rsa_free_mpi_key(mpi_key);
		return -EINVAL;
293
	}
294 295 296 297 298 299

	return 0;

err:
	rsa_free_mpi_key(mpi_key);
	return -ENOMEM;
300 301
}

302 303 304
static int rsa_set_priv_key(struct crypto_akcipher *tfm, const void *key,
			    unsigned int keylen)
{
305 306
	struct rsa_mpi_key *mpi_key = akcipher_tfm_ctx(tfm);
	struct rsa_key raw_key = {0};
307 308
	int ret;

309 310 311 312
	/* Free the old MPI key if any */
	rsa_free_mpi_key(mpi_key);

	ret = rsa_parse_priv_key(&raw_key, key, keylen);
313 314 315
	if (ret)
		return ret;

316 317 318 319 320 321 322 323 324 325 326 327 328 329 330
	mpi_key->d = mpi_read_raw_data(raw_key.d, raw_key.d_sz);
	if (!mpi_key->d)
		goto err;

	mpi_key->e = mpi_read_raw_data(raw_key.e, raw_key.e_sz);
	if (!mpi_key->e)
		goto err;

	mpi_key->n = mpi_read_raw_data(raw_key.n, raw_key.n_sz);
	if (!mpi_key->n)
		goto err;

	if (rsa_check_key_length(mpi_get_size(mpi_key->n) << 3)) {
		rsa_free_mpi_key(mpi_key);
		return -EINVAL;
331
	}
332 333 334 335 336 337

	return 0;

err:
	rsa_free_mpi_key(mpi_key);
	return -ENOMEM;
338 339 340 341
}

static int rsa_max_size(struct crypto_akcipher *tfm)
{
342
	struct rsa_mpi_key *pkey = akcipher_tfm_ctx(tfm);
343 344 345 346

	return pkey->n ? mpi_get_size(pkey->n) : -EINVAL;
}

347 348
static void rsa_exit_tfm(struct crypto_akcipher *tfm)
{
349
	struct rsa_mpi_key *pkey = akcipher_tfm_ctx(tfm);
350

351
	rsa_free_mpi_key(pkey);
352 353 354 355 356 357 358
}

static struct akcipher_alg rsa = {
	.encrypt = rsa_enc,
	.decrypt = rsa_dec,
	.sign = rsa_sign,
	.verify = rsa_verify,
359 360 361
	.set_priv_key = rsa_set_priv_key,
	.set_pub_key = rsa_set_pub_key,
	.max_size = rsa_max_size,
362 363 364 365 366 367
	.exit = rsa_exit_tfm,
	.base = {
		.cra_name = "rsa",
		.cra_driver_name = "rsa-generic",
		.cra_priority = 100,
		.cra_module = THIS_MODULE,
368
		.cra_ctxsize = sizeof(struct rsa_mpi_key),
369 370 371 372 373
	},
};

static int rsa_init(void)
{
374 375 376 377 378 379 380 381 382 383 384 385 386
	int err;

	err = crypto_register_akcipher(&rsa);
	if (err)
		return err;

	err = crypto_register_template(&rsa_pkcs1pad_tmpl);
	if (err) {
		crypto_unregister_akcipher(&rsa);
		return err;
	}

	return 0;
387 388 389 390
}

static void rsa_exit(void)
{
391
	crypto_unregister_template(&rsa_pkcs1pad_tmpl);
392 393 394 395 396 397 398 399
	crypto_unregister_akcipher(&rsa);
}

module_init(rsa_init);
module_exit(rsa_exit);
MODULE_ALIAS_CRYPTO("rsa");
MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("RSA generic algorithm");