Merge branch 'work.uaccess-unaligned' of git://git.kernel.org/pub/scm/linux/kernel...
[platform/kernel/linux-exynos.git] / crypto / rsa.c
1 /* RSA asymmetric public-key algorithm [RFC3447]
2  *
3  * Copyright (c) 2015, Intel Corporation
4  * Authors: Tadeusz Struk <tadeusz.struk@intel.com>
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public Licence
8  * as published by the Free Software Foundation; either version
9  * 2 of the Licence, or (at your option) any later version.
10  */
11
12 #include <linux/module.h>
13 #include <linux/mpi.h>
14 #include <crypto/internal/rsa.h>
15 #include <crypto/internal/akcipher.h>
16 #include <crypto/akcipher.h>
17 #include <crypto/algapi.h>
18
19 struct rsa_mpi_key {
20         MPI n;
21         MPI e;
22         MPI d;
23 };
24
25 /*
26  * RSAEP function [RFC3447 sec 5.1.1]
27  * c = m^e mod n;
28  */
29 static int _rsa_enc(const struct rsa_mpi_key *key, MPI c, MPI m)
30 {
31         /* (1) Validate 0 <= m < n */
32         if (mpi_cmp_ui(m, 0) < 0 || mpi_cmp(m, key->n) >= 0)
33                 return -EINVAL;
34
35         /* (2) c = m^e mod n */
36         return mpi_powm(c, m, key->e, key->n);
37 }
38
39 /*
40  * RSADP function [RFC3447 sec 5.1.2]
41  * m = c^d mod n;
42  */
43 static int _rsa_dec(const struct rsa_mpi_key *key, MPI m, MPI c)
44 {
45         /* (1) Validate 0 <= c < n */
46         if (mpi_cmp_ui(c, 0) < 0 || mpi_cmp(c, key->n) >= 0)
47                 return -EINVAL;
48
49         /* (2) m = c^d mod n */
50         return mpi_powm(m, c, key->d, key->n);
51 }
52
53 /*
54  * RSASP1 function [RFC3447 sec 5.2.1]
55  * s = m^d mod n
56  */
57 static int _rsa_sign(const struct rsa_mpi_key *key, MPI s, MPI m)
58 {
59         /* (1) Validate 0 <= m < n */
60         if (mpi_cmp_ui(m, 0) < 0 || mpi_cmp(m, key->n) >= 0)
61                 return -EINVAL;
62
63         /* (2) s = m^d mod n */
64         return mpi_powm(s, m, key->d, key->n);
65 }
66
67 /*
68  * RSAVP1 function [RFC3447 sec 5.2.2]
69  * m = s^e mod n;
70  */
71 static int _rsa_verify(const struct rsa_mpi_key *key, MPI m, MPI s)
72 {
73         /* (1) Validate 0 <= s < n */
74         if (mpi_cmp_ui(s, 0) < 0 || mpi_cmp(s, key->n) >= 0)
75                 return -EINVAL;
76
77         /* (2) m = s^e mod n */
78         return mpi_powm(m, s, key->e, key->n);
79 }
80
81 static inline struct rsa_mpi_key *rsa_get_key(struct crypto_akcipher *tfm)
82 {
83         return akcipher_tfm_ctx(tfm);
84 }
85
86 static int rsa_enc(struct akcipher_request *req)
87 {
88         struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
89         const struct rsa_mpi_key *pkey = rsa_get_key(tfm);
90         MPI m, c = mpi_alloc(0);
91         int ret = 0;
92         int sign;
93
94         if (!c)
95                 return -ENOMEM;
96
97         if (unlikely(!pkey->n || !pkey->e)) {
98                 ret = -EINVAL;
99                 goto err_free_c;
100         }
101
102         ret = -ENOMEM;
103         m = mpi_read_raw_from_sgl(req->src, req->src_len);
104         if (!m)
105                 goto err_free_c;
106
107         ret = _rsa_enc(pkey, c, m);
108         if (ret)
109                 goto err_free_m;
110
111         ret = mpi_write_to_sgl(c, req->dst, req->dst_len, &sign);
112         if (ret)
113                 goto err_free_m;
114
115         if (sign < 0)
116                 ret = -EBADMSG;
117
118 err_free_m:
119         mpi_free(m);
120 err_free_c:
121         mpi_free(c);
122         return ret;
123 }
124
125 static int rsa_dec(struct akcipher_request *req)
126 {
127         struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
128         const struct rsa_mpi_key *pkey = rsa_get_key(tfm);
129         MPI c, m = mpi_alloc(0);
130         int ret = 0;
131         int sign;
132
133         if (!m)
134                 return -ENOMEM;
135
136         if (unlikely(!pkey->n || !pkey->d)) {
137                 ret = -EINVAL;
138                 goto err_free_m;
139         }
140
141         ret = -ENOMEM;
142         c = mpi_read_raw_from_sgl(req->src, req->src_len);
143         if (!c)
144                 goto err_free_m;
145
146         ret = _rsa_dec(pkey, m, c);
147         if (ret)
148                 goto err_free_c;
149
150         ret = mpi_write_to_sgl(m, req->dst, req->dst_len, &sign);
151         if (ret)
152                 goto err_free_c;
153
154         if (sign < 0)
155                 ret = -EBADMSG;
156 err_free_c:
157         mpi_free(c);
158 err_free_m:
159         mpi_free(m);
160         return ret;
161 }
162
163 static int rsa_sign(struct akcipher_request *req)
164 {
165         struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
166         const struct rsa_mpi_key *pkey = rsa_get_key(tfm);
167         MPI m, s = mpi_alloc(0);
168         int ret = 0;
169         int sign;
170
171         if (!s)
172                 return -ENOMEM;
173
174         if (unlikely(!pkey->n || !pkey->d)) {
175                 ret = -EINVAL;
176                 goto err_free_s;
177         }
178
179         ret = -ENOMEM;
180         m = mpi_read_raw_from_sgl(req->src, req->src_len);
181         if (!m)
182                 goto err_free_s;
183
184         ret = _rsa_sign(pkey, s, m);
185         if (ret)
186                 goto err_free_m;
187
188         ret = mpi_write_to_sgl(s, req->dst, req->dst_len, &sign);
189         if (ret)
190                 goto err_free_m;
191
192         if (sign < 0)
193                 ret = -EBADMSG;
194
195 err_free_m:
196         mpi_free(m);
197 err_free_s:
198         mpi_free(s);
199         return ret;
200 }
201
202 static int rsa_verify(struct akcipher_request *req)
203 {
204         struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
205         const struct rsa_mpi_key *pkey = rsa_get_key(tfm);
206         MPI s, m = mpi_alloc(0);
207         int ret = 0;
208         int sign;
209
210         if (!m)
211                 return -ENOMEM;
212
213         if (unlikely(!pkey->n || !pkey->e)) {
214                 ret = -EINVAL;
215                 goto err_free_m;
216         }
217
218         ret = -ENOMEM;
219         s = mpi_read_raw_from_sgl(req->src, req->src_len);
220         if (!s) {
221                 ret = -ENOMEM;
222                 goto err_free_m;
223         }
224
225         ret = _rsa_verify(pkey, m, s);
226         if (ret)
227                 goto err_free_s;
228
229         ret = mpi_write_to_sgl(m, req->dst, req->dst_len, &sign);
230         if (ret)
231                 goto err_free_s;
232
233         if (sign < 0)
234                 ret = -EBADMSG;
235
236 err_free_s:
237         mpi_free(s);
238 err_free_m:
239         mpi_free(m);
240         return ret;
241 }
242
243 static void rsa_free_mpi_key(struct rsa_mpi_key *key)
244 {
245         mpi_free(key->d);
246         mpi_free(key->e);
247         mpi_free(key->n);
248         key->d = NULL;
249         key->e = NULL;
250         key->n = NULL;
251 }
252
253 static int rsa_check_key_length(unsigned int len)
254 {
255         switch (len) {
256         case 512:
257         case 1024:
258         case 1536:
259         case 2048:
260         case 3072:
261         case 4096:
262                 return 0;
263         }
264
265         return -EINVAL;
266 }
267
268 static int rsa_set_pub_key(struct crypto_akcipher *tfm, const void *key,
269                            unsigned int keylen)
270 {
271         struct rsa_mpi_key *mpi_key = akcipher_tfm_ctx(tfm);
272         struct rsa_key raw_key = {0};
273         int ret;
274
275         /* Free the old MPI key if any */
276         rsa_free_mpi_key(mpi_key);
277
278         ret = rsa_parse_pub_key(&raw_key, key, keylen);
279         if (ret)
280                 return ret;
281
282         mpi_key->e = mpi_read_raw_data(raw_key.e, raw_key.e_sz);
283         if (!mpi_key->e)
284                 goto err;
285
286         mpi_key->n = mpi_read_raw_data(raw_key.n, raw_key.n_sz);
287         if (!mpi_key->n)
288                 goto err;
289
290         if (rsa_check_key_length(mpi_get_size(mpi_key->n) << 3)) {
291                 rsa_free_mpi_key(mpi_key);
292                 return -EINVAL;
293         }
294
295         return 0;
296
297 err:
298         rsa_free_mpi_key(mpi_key);
299         return -ENOMEM;
300 }
301
302 static int rsa_set_priv_key(struct crypto_akcipher *tfm, const void *key,
303                             unsigned int keylen)
304 {
305         struct rsa_mpi_key *mpi_key = akcipher_tfm_ctx(tfm);
306         struct rsa_key raw_key = {0};
307         int ret;
308
309         /* Free the old MPI key if any */
310         rsa_free_mpi_key(mpi_key);
311
312         ret = rsa_parse_priv_key(&raw_key, key, keylen);
313         if (ret)
314                 return ret;
315
316         mpi_key->d = mpi_read_raw_data(raw_key.d, raw_key.d_sz);
317         if (!mpi_key->d)
318                 goto err;
319
320         mpi_key->e = mpi_read_raw_data(raw_key.e, raw_key.e_sz);
321         if (!mpi_key->e)
322                 goto err;
323
324         mpi_key->n = mpi_read_raw_data(raw_key.n, raw_key.n_sz);
325         if (!mpi_key->n)
326                 goto err;
327
328         if (rsa_check_key_length(mpi_get_size(mpi_key->n) << 3)) {
329                 rsa_free_mpi_key(mpi_key);
330                 return -EINVAL;
331         }
332
333         return 0;
334
335 err:
336         rsa_free_mpi_key(mpi_key);
337         return -ENOMEM;
338 }
339
340 static unsigned int rsa_max_size(struct crypto_akcipher *tfm)
341 {
342         struct rsa_mpi_key *pkey = akcipher_tfm_ctx(tfm);
343
344         return mpi_get_size(pkey->n);
345 }
346
347 static void rsa_exit_tfm(struct crypto_akcipher *tfm)
348 {
349         struct rsa_mpi_key *pkey = akcipher_tfm_ctx(tfm);
350
351         rsa_free_mpi_key(pkey);
352 }
353
354 static struct akcipher_alg rsa = {
355         .encrypt = rsa_enc,
356         .decrypt = rsa_dec,
357         .sign = rsa_sign,
358         .verify = rsa_verify,
359         .set_priv_key = rsa_set_priv_key,
360         .set_pub_key = rsa_set_pub_key,
361         .max_size = rsa_max_size,
362         .exit = rsa_exit_tfm,
363         .base = {
364                 .cra_name = "rsa",
365                 .cra_driver_name = "rsa-generic",
366                 .cra_priority = 100,
367                 .cra_module = THIS_MODULE,
368                 .cra_ctxsize = sizeof(struct rsa_mpi_key),
369         },
370 };
371
372 static int rsa_init(void)
373 {
374         int err;
375
376         err = crypto_register_akcipher(&rsa);
377         if (err)
378                 return err;
379
380         err = crypto_register_template(&rsa_pkcs1pad_tmpl);
381         if (err) {
382                 crypto_unregister_akcipher(&rsa);
383                 return err;
384         }
385
386         return 0;
387 }
388
389 static void rsa_exit(void)
390 {
391         crypto_unregister_template(&rsa_pkcs1pad_tmpl);
392         crypto_unregister_akcipher(&rsa);
393 }
394
395 module_init(rsa_init);
396 module_exit(rsa_exit);
397 MODULE_ALIAS_CRYPTO("rsa");
398 MODULE_LICENSE("GPL");
399 MODULE_DESCRIPTION("RSA generic algorithm");