Loading...
1// SPDX-License-Identifier: GPL-2.0
2/*
3 * Cryptographic API.
4 *
5 * s390 implementation of the AES Cipher Algorithm with protected keys.
6 *
7 * s390 Version:
8 * Copyright IBM Corp. 2017, 2023
9 * Author(s): Martin Schwidefsky <schwidefsky@de.ibm.com>
10 * Harald Freudenberger <freude@de.ibm.com>
11 */
12
13#define KMSG_COMPONENT "paes_s390"
14#define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
15
16#include <crypto/aes.h>
17#include <crypto/algapi.h>
18#include <linux/bug.h>
19#include <linux/err.h>
20#include <linux/module.h>
21#include <linux/cpufeature.h>
22#include <linux/init.h>
23#include <linux/mutex.h>
24#include <linux/spinlock.h>
25#include <linux/delay.h>
26#include <crypto/internal/skcipher.h>
27#include <crypto/xts.h>
28#include <asm/cpacf.h>
29#include <asm/pkey.h>
30
31/*
32 * Key blobs smaller/bigger than these defines are rejected
33 * by the common code even before the individual setkey function
34 * is called. As paes can handle different kinds of key blobs
35 * and padding is also possible, the limits need to be generous.
36 */
37#define PAES_MIN_KEYSIZE 16
38#define PAES_MAX_KEYSIZE MAXEP11AESKEYBLOBSIZE
39
40static u8 *ctrblk;
41static DEFINE_MUTEX(ctrblk_lock);
42
43static cpacf_mask_t km_functions, kmc_functions, kmctr_functions;
44
45struct key_blob {
46 /*
47 * Small keys will be stored in the keybuf. Larger keys are
48 * stored in extra allocated memory. In both cases does
49 * key point to the memory where the key is stored.
50 * The code distinguishes by checking keylen against
51 * sizeof(keybuf). See the two following helper functions.
52 */
53 u8 *key;
54 u8 keybuf[128];
55 unsigned int keylen;
56};
57
58static inline int _key_to_kb(struct key_blob *kb,
59 const u8 *key,
60 unsigned int keylen)
61{
62 struct clearkey_header {
63 u8 type;
64 u8 res0[3];
65 u8 version;
66 u8 res1[3];
67 u32 keytype;
68 u32 len;
69 } __packed * h;
70
71 switch (keylen) {
72 case 16:
73 case 24:
74 case 32:
75 /* clear key value, prepare pkey clear key token in keybuf */
76 memset(kb->keybuf, 0, sizeof(kb->keybuf));
77 h = (struct clearkey_header *) kb->keybuf;
78 h->version = 0x02; /* TOKVER_CLEAR_KEY */
79 h->keytype = (keylen - 8) >> 3;
80 h->len = keylen;
81 memcpy(kb->keybuf + sizeof(*h), key, keylen);
82 kb->keylen = sizeof(*h) + keylen;
83 kb->key = kb->keybuf;
84 break;
85 default:
86 /* other key material, let pkey handle this */
87 if (keylen <= sizeof(kb->keybuf))
88 kb->key = kb->keybuf;
89 else {
90 kb->key = kmalloc(keylen, GFP_KERNEL);
91 if (!kb->key)
92 return -ENOMEM;
93 }
94 memcpy(kb->key, key, keylen);
95 kb->keylen = keylen;
96 break;
97 }
98
99 return 0;
100}
101
102static inline void _free_kb_keybuf(struct key_blob *kb)
103{
104 if (kb->key && kb->key != kb->keybuf
105 && kb->keylen > sizeof(kb->keybuf)) {
106 kfree_sensitive(kb->key);
107 kb->key = NULL;
108 }
109}
110
111struct s390_paes_ctx {
112 struct key_blob kb;
113 struct pkey_protkey pk;
114 spinlock_t pk_lock;
115 unsigned long fc;
116};
117
118struct s390_pxts_ctx {
119 struct key_blob kb[2];
120 struct pkey_protkey pk[2];
121 spinlock_t pk_lock;
122 unsigned long fc;
123};
124
125static inline int __paes_keyblob2pkey(struct key_blob *kb,
126 struct pkey_protkey *pk)
127{
128 int i, ret;
129
130 /* try three times in case of failure */
131 for (i = 0; i < 3; i++) {
132 if (i > 0 && ret == -EAGAIN && in_task())
133 if (msleep_interruptible(1000))
134 return -EINTR;
135 ret = pkey_keyblob2pkey(kb->key, kb->keylen,
136 pk->protkey, &pk->len, &pk->type);
137 if (ret == 0)
138 break;
139 }
140
141 return ret;
142}
143
144static inline int __paes_convert_key(struct s390_paes_ctx *ctx)
145{
146 int ret;
147 struct pkey_protkey pkey;
148
149 pkey.len = sizeof(pkey.protkey);
150 ret = __paes_keyblob2pkey(&ctx->kb, &pkey);
151 if (ret)
152 return ret;
153
154 spin_lock_bh(&ctx->pk_lock);
155 memcpy(&ctx->pk, &pkey, sizeof(pkey));
156 spin_unlock_bh(&ctx->pk_lock);
157
158 return 0;
159}
160
161static int ecb_paes_init(struct crypto_skcipher *tfm)
162{
163 struct s390_paes_ctx *ctx = crypto_skcipher_ctx(tfm);
164
165 ctx->kb.key = NULL;
166 spin_lock_init(&ctx->pk_lock);
167
168 return 0;
169}
170
171static void ecb_paes_exit(struct crypto_skcipher *tfm)
172{
173 struct s390_paes_ctx *ctx = crypto_skcipher_ctx(tfm);
174
175 _free_kb_keybuf(&ctx->kb);
176}
177
178static inline int __ecb_paes_set_key(struct s390_paes_ctx *ctx)
179{
180 int rc;
181 unsigned long fc;
182
183 rc = __paes_convert_key(ctx);
184 if (rc)
185 return rc;
186
187 /* Pick the correct function code based on the protected key type */
188 fc = (ctx->pk.type == PKEY_KEYTYPE_AES_128) ? CPACF_KM_PAES_128 :
189 (ctx->pk.type == PKEY_KEYTYPE_AES_192) ? CPACF_KM_PAES_192 :
190 (ctx->pk.type == PKEY_KEYTYPE_AES_256) ? CPACF_KM_PAES_256 : 0;
191
192 /* Check if the function code is available */
193 ctx->fc = (fc && cpacf_test_func(&km_functions, fc)) ? fc : 0;
194
195 return ctx->fc ? 0 : -EINVAL;
196}
197
198static int ecb_paes_set_key(struct crypto_skcipher *tfm, const u8 *in_key,
199 unsigned int key_len)
200{
201 int rc;
202 struct s390_paes_ctx *ctx = crypto_skcipher_ctx(tfm);
203
204 _free_kb_keybuf(&ctx->kb);
205 rc = _key_to_kb(&ctx->kb, in_key, key_len);
206 if (rc)
207 return rc;
208
209 return __ecb_paes_set_key(ctx);
210}
211
212static int ecb_paes_crypt(struct skcipher_request *req, unsigned long modifier)
213{
214 struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
215 struct s390_paes_ctx *ctx = crypto_skcipher_ctx(tfm);
216 struct skcipher_walk walk;
217 unsigned int nbytes, n, k;
218 int ret;
219 struct {
220 u8 key[MAXPROTKEYSIZE];
221 } param;
222
223 ret = skcipher_walk_virt(&walk, req, false);
224 if (ret)
225 return ret;
226
227 spin_lock_bh(&ctx->pk_lock);
228 memcpy(param.key, ctx->pk.protkey, MAXPROTKEYSIZE);
229 spin_unlock_bh(&ctx->pk_lock);
230
231 while ((nbytes = walk.nbytes) != 0) {
232 /* only use complete blocks */
233 n = nbytes & ~(AES_BLOCK_SIZE - 1);
234 k = cpacf_km(ctx->fc | modifier, ¶m,
235 walk.dst.virt.addr, walk.src.virt.addr, n);
236 if (k)
237 ret = skcipher_walk_done(&walk, nbytes - k);
238 if (k < n) {
239 if (__paes_convert_key(ctx))
240 return skcipher_walk_done(&walk, -EIO);
241 spin_lock_bh(&ctx->pk_lock);
242 memcpy(param.key, ctx->pk.protkey, MAXPROTKEYSIZE);
243 spin_unlock_bh(&ctx->pk_lock);
244 }
245 }
246 return ret;
247}
248
249static int ecb_paes_encrypt(struct skcipher_request *req)
250{
251 return ecb_paes_crypt(req, 0);
252}
253
254static int ecb_paes_decrypt(struct skcipher_request *req)
255{
256 return ecb_paes_crypt(req, CPACF_DECRYPT);
257}
258
259static struct skcipher_alg ecb_paes_alg = {
260 .base.cra_name = "ecb(paes)",
261 .base.cra_driver_name = "ecb-paes-s390",
262 .base.cra_priority = 401, /* combo: aes + ecb + 1 */
263 .base.cra_blocksize = AES_BLOCK_SIZE,
264 .base.cra_ctxsize = sizeof(struct s390_paes_ctx),
265 .base.cra_module = THIS_MODULE,
266 .base.cra_list = LIST_HEAD_INIT(ecb_paes_alg.base.cra_list),
267 .init = ecb_paes_init,
268 .exit = ecb_paes_exit,
269 .min_keysize = PAES_MIN_KEYSIZE,
270 .max_keysize = PAES_MAX_KEYSIZE,
271 .setkey = ecb_paes_set_key,
272 .encrypt = ecb_paes_encrypt,
273 .decrypt = ecb_paes_decrypt,
274};
275
276static int cbc_paes_init(struct crypto_skcipher *tfm)
277{
278 struct s390_paes_ctx *ctx = crypto_skcipher_ctx(tfm);
279
280 ctx->kb.key = NULL;
281 spin_lock_init(&ctx->pk_lock);
282
283 return 0;
284}
285
286static void cbc_paes_exit(struct crypto_skcipher *tfm)
287{
288 struct s390_paes_ctx *ctx = crypto_skcipher_ctx(tfm);
289
290 _free_kb_keybuf(&ctx->kb);
291}
292
293static inline int __cbc_paes_set_key(struct s390_paes_ctx *ctx)
294{
295 int rc;
296 unsigned long fc;
297
298 rc = __paes_convert_key(ctx);
299 if (rc)
300 return rc;
301
302 /* Pick the correct function code based on the protected key type */
303 fc = (ctx->pk.type == PKEY_KEYTYPE_AES_128) ? CPACF_KMC_PAES_128 :
304 (ctx->pk.type == PKEY_KEYTYPE_AES_192) ? CPACF_KMC_PAES_192 :
305 (ctx->pk.type == PKEY_KEYTYPE_AES_256) ? CPACF_KMC_PAES_256 : 0;
306
307 /* Check if the function code is available */
308 ctx->fc = (fc && cpacf_test_func(&kmc_functions, fc)) ? fc : 0;
309
310 return ctx->fc ? 0 : -EINVAL;
311}
312
313static int cbc_paes_set_key(struct crypto_skcipher *tfm, const u8 *in_key,
314 unsigned int key_len)
315{
316 int rc;
317 struct s390_paes_ctx *ctx = crypto_skcipher_ctx(tfm);
318
319 _free_kb_keybuf(&ctx->kb);
320 rc = _key_to_kb(&ctx->kb, in_key, key_len);
321 if (rc)
322 return rc;
323
324 return __cbc_paes_set_key(ctx);
325}
326
327static int cbc_paes_crypt(struct skcipher_request *req, unsigned long modifier)
328{
329 struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
330 struct s390_paes_ctx *ctx = crypto_skcipher_ctx(tfm);
331 struct skcipher_walk walk;
332 unsigned int nbytes, n, k;
333 int ret;
334 struct {
335 u8 iv[AES_BLOCK_SIZE];
336 u8 key[MAXPROTKEYSIZE];
337 } param;
338
339 ret = skcipher_walk_virt(&walk, req, false);
340 if (ret)
341 return ret;
342
343 memcpy(param.iv, walk.iv, AES_BLOCK_SIZE);
344 spin_lock_bh(&ctx->pk_lock);
345 memcpy(param.key, ctx->pk.protkey, MAXPROTKEYSIZE);
346 spin_unlock_bh(&ctx->pk_lock);
347
348 while ((nbytes = walk.nbytes) != 0) {
349 /* only use complete blocks */
350 n = nbytes & ~(AES_BLOCK_SIZE - 1);
351 k = cpacf_kmc(ctx->fc | modifier, ¶m,
352 walk.dst.virt.addr, walk.src.virt.addr, n);
353 if (k) {
354 memcpy(walk.iv, param.iv, AES_BLOCK_SIZE);
355 ret = skcipher_walk_done(&walk, nbytes - k);
356 }
357 if (k < n) {
358 if (__paes_convert_key(ctx))
359 return skcipher_walk_done(&walk, -EIO);
360 spin_lock_bh(&ctx->pk_lock);
361 memcpy(param.key, ctx->pk.protkey, MAXPROTKEYSIZE);
362 spin_unlock_bh(&ctx->pk_lock);
363 }
364 }
365 return ret;
366}
367
368static int cbc_paes_encrypt(struct skcipher_request *req)
369{
370 return cbc_paes_crypt(req, 0);
371}
372
373static int cbc_paes_decrypt(struct skcipher_request *req)
374{
375 return cbc_paes_crypt(req, CPACF_DECRYPT);
376}
377
378static struct skcipher_alg cbc_paes_alg = {
379 .base.cra_name = "cbc(paes)",
380 .base.cra_driver_name = "cbc-paes-s390",
381 .base.cra_priority = 402, /* ecb-paes-s390 + 1 */
382 .base.cra_blocksize = AES_BLOCK_SIZE,
383 .base.cra_ctxsize = sizeof(struct s390_paes_ctx),
384 .base.cra_module = THIS_MODULE,
385 .base.cra_list = LIST_HEAD_INIT(cbc_paes_alg.base.cra_list),
386 .init = cbc_paes_init,
387 .exit = cbc_paes_exit,
388 .min_keysize = PAES_MIN_KEYSIZE,
389 .max_keysize = PAES_MAX_KEYSIZE,
390 .ivsize = AES_BLOCK_SIZE,
391 .setkey = cbc_paes_set_key,
392 .encrypt = cbc_paes_encrypt,
393 .decrypt = cbc_paes_decrypt,
394};
395
396static int xts_paes_init(struct crypto_skcipher *tfm)
397{
398 struct s390_pxts_ctx *ctx = crypto_skcipher_ctx(tfm);
399
400 ctx->kb[0].key = NULL;
401 ctx->kb[1].key = NULL;
402 spin_lock_init(&ctx->pk_lock);
403
404 return 0;
405}
406
407static void xts_paes_exit(struct crypto_skcipher *tfm)
408{
409 struct s390_pxts_ctx *ctx = crypto_skcipher_ctx(tfm);
410
411 _free_kb_keybuf(&ctx->kb[0]);
412 _free_kb_keybuf(&ctx->kb[1]);
413}
414
415static inline int __xts_paes_convert_key(struct s390_pxts_ctx *ctx)
416{
417 struct pkey_protkey pkey0, pkey1;
418
419 pkey0.len = sizeof(pkey0.protkey);
420 pkey1.len = sizeof(pkey1.protkey);
421
422 if (__paes_keyblob2pkey(&ctx->kb[0], &pkey0) ||
423 __paes_keyblob2pkey(&ctx->kb[1], &pkey1))
424 return -EINVAL;
425
426 spin_lock_bh(&ctx->pk_lock);
427 memcpy(&ctx->pk[0], &pkey0, sizeof(pkey0));
428 memcpy(&ctx->pk[1], &pkey1, sizeof(pkey1));
429 spin_unlock_bh(&ctx->pk_lock);
430
431 return 0;
432}
433
434static inline int __xts_paes_set_key(struct s390_pxts_ctx *ctx)
435{
436 unsigned long fc;
437
438 if (__xts_paes_convert_key(ctx))
439 return -EINVAL;
440
441 if (ctx->pk[0].type != ctx->pk[1].type)
442 return -EINVAL;
443
444 /* Pick the correct function code based on the protected key type */
445 fc = (ctx->pk[0].type == PKEY_KEYTYPE_AES_128) ? CPACF_KM_PXTS_128 :
446 (ctx->pk[0].type == PKEY_KEYTYPE_AES_256) ?
447 CPACF_KM_PXTS_256 : 0;
448
449 /* Check if the function code is available */
450 ctx->fc = (fc && cpacf_test_func(&km_functions, fc)) ? fc : 0;
451
452 return ctx->fc ? 0 : -EINVAL;
453}
454
455static int xts_paes_set_key(struct crypto_skcipher *tfm, const u8 *in_key,
456 unsigned int xts_key_len)
457{
458 int rc;
459 struct s390_pxts_ctx *ctx = crypto_skcipher_ctx(tfm);
460 u8 ckey[2 * AES_MAX_KEY_SIZE];
461 unsigned int ckey_len, key_len;
462
463 if (xts_key_len % 2)
464 return -EINVAL;
465
466 key_len = xts_key_len / 2;
467
468 _free_kb_keybuf(&ctx->kb[0]);
469 _free_kb_keybuf(&ctx->kb[1]);
470 rc = _key_to_kb(&ctx->kb[0], in_key, key_len);
471 if (rc)
472 return rc;
473 rc = _key_to_kb(&ctx->kb[1], in_key + key_len, key_len);
474 if (rc)
475 return rc;
476
477 rc = __xts_paes_set_key(ctx);
478 if (rc)
479 return rc;
480
481 /*
482 * xts_verify_key verifies the key length is not odd and makes
483 * sure that the two keys are not the same. This can be done
484 * on the two protected keys as well
485 */
486 ckey_len = (ctx->pk[0].type == PKEY_KEYTYPE_AES_128) ?
487 AES_KEYSIZE_128 : AES_KEYSIZE_256;
488 memcpy(ckey, ctx->pk[0].protkey, ckey_len);
489 memcpy(ckey + ckey_len, ctx->pk[1].protkey, ckey_len);
490 return xts_verify_key(tfm, ckey, 2*ckey_len);
491}
492
493static int xts_paes_crypt(struct skcipher_request *req, unsigned long modifier)
494{
495 struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
496 struct s390_pxts_ctx *ctx = crypto_skcipher_ctx(tfm);
497 struct skcipher_walk walk;
498 unsigned int keylen, offset, nbytes, n, k;
499 int ret;
500 struct {
501 u8 key[MAXPROTKEYSIZE]; /* key + verification pattern */
502 u8 tweak[16];
503 u8 block[16];
504 u8 bit[16];
505 u8 xts[16];
506 } pcc_param;
507 struct {
508 u8 key[MAXPROTKEYSIZE]; /* key + verification pattern */
509 u8 init[16];
510 } xts_param;
511
512 ret = skcipher_walk_virt(&walk, req, false);
513 if (ret)
514 return ret;
515
516 keylen = (ctx->pk[0].type == PKEY_KEYTYPE_AES_128) ? 48 : 64;
517 offset = (ctx->pk[0].type == PKEY_KEYTYPE_AES_128) ? 16 : 0;
518
519 memset(&pcc_param, 0, sizeof(pcc_param));
520 memcpy(pcc_param.tweak, walk.iv, sizeof(pcc_param.tweak));
521 spin_lock_bh(&ctx->pk_lock);
522 memcpy(pcc_param.key + offset, ctx->pk[1].protkey, keylen);
523 memcpy(xts_param.key + offset, ctx->pk[0].protkey, keylen);
524 spin_unlock_bh(&ctx->pk_lock);
525 cpacf_pcc(ctx->fc, pcc_param.key + offset);
526 memcpy(xts_param.init, pcc_param.xts, 16);
527
528 while ((nbytes = walk.nbytes) != 0) {
529 /* only use complete blocks */
530 n = nbytes & ~(AES_BLOCK_SIZE - 1);
531 k = cpacf_km(ctx->fc | modifier, xts_param.key + offset,
532 walk.dst.virt.addr, walk.src.virt.addr, n);
533 if (k)
534 ret = skcipher_walk_done(&walk, nbytes - k);
535 if (k < n) {
536 if (__xts_paes_convert_key(ctx))
537 return skcipher_walk_done(&walk, -EIO);
538 spin_lock_bh(&ctx->pk_lock);
539 memcpy(xts_param.key + offset,
540 ctx->pk[0].protkey, keylen);
541 spin_unlock_bh(&ctx->pk_lock);
542 }
543 }
544
545 return ret;
546}
547
548static int xts_paes_encrypt(struct skcipher_request *req)
549{
550 return xts_paes_crypt(req, 0);
551}
552
553static int xts_paes_decrypt(struct skcipher_request *req)
554{
555 return xts_paes_crypt(req, CPACF_DECRYPT);
556}
557
558static struct skcipher_alg xts_paes_alg = {
559 .base.cra_name = "xts(paes)",
560 .base.cra_driver_name = "xts-paes-s390",
561 .base.cra_priority = 402, /* ecb-paes-s390 + 1 */
562 .base.cra_blocksize = AES_BLOCK_SIZE,
563 .base.cra_ctxsize = sizeof(struct s390_pxts_ctx),
564 .base.cra_module = THIS_MODULE,
565 .base.cra_list = LIST_HEAD_INIT(xts_paes_alg.base.cra_list),
566 .init = xts_paes_init,
567 .exit = xts_paes_exit,
568 .min_keysize = 2 * PAES_MIN_KEYSIZE,
569 .max_keysize = 2 * PAES_MAX_KEYSIZE,
570 .ivsize = AES_BLOCK_SIZE,
571 .setkey = xts_paes_set_key,
572 .encrypt = xts_paes_encrypt,
573 .decrypt = xts_paes_decrypt,
574};
575
576static int ctr_paes_init(struct crypto_skcipher *tfm)
577{
578 struct s390_paes_ctx *ctx = crypto_skcipher_ctx(tfm);
579
580 ctx->kb.key = NULL;
581 spin_lock_init(&ctx->pk_lock);
582
583 return 0;
584}
585
586static void ctr_paes_exit(struct crypto_skcipher *tfm)
587{
588 struct s390_paes_ctx *ctx = crypto_skcipher_ctx(tfm);
589
590 _free_kb_keybuf(&ctx->kb);
591}
592
593static inline int __ctr_paes_set_key(struct s390_paes_ctx *ctx)
594{
595 int rc;
596 unsigned long fc;
597
598 rc = __paes_convert_key(ctx);
599 if (rc)
600 return rc;
601
602 /* Pick the correct function code based on the protected key type */
603 fc = (ctx->pk.type == PKEY_KEYTYPE_AES_128) ? CPACF_KMCTR_PAES_128 :
604 (ctx->pk.type == PKEY_KEYTYPE_AES_192) ? CPACF_KMCTR_PAES_192 :
605 (ctx->pk.type == PKEY_KEYTYPE_AES_256) ?
606 CPACF_KMCTR_PAES_256 : 0;
607
608 /* Check if the function code is available */
609 ctx->fc = (fc && cpacf_test_func(&kmctr_functions, fc)) ? fc : 0;
610
611 return ctx->fc ? 0 : -EINVAL;
612}
613
614static int ctr_paes_set_key(struct crypto_skcipher *tfm, const u8 *in_key,
615 unsigned int key_len)
616{
617 int rc;
618 struct s390_paes_ctx *ctx = crypto_skcipher_ctx(tfm);
619
620 _free_kb_keybuf(&ctx->kb);
621 rc = _key_to_kb(&ctx->kb, in_key, key_len);
622 if (rc)
623 return rc;
624
625 return __ctr_paes_set_key(ctx);
626}
627
628static unsigned int __ctrblk_init(u8 *ctrptr, u8 *iv, unsigned int nbytes)
629{
630 unsigned int i, n;
631
632 /* only use complete blocks, max. PAGE_SIZE */
633 memcpy(ctrptr, iv, AES_BLOCK_SIZE);
634 n = (nbytes > PAGE_SIZE) ? PAGE_SIZE : nbytes & ~(AES_BLOCK_SIZE - 1);
635 for (i = (n / AES_BLOCK_SIZE) - 1; i > 0; i--) {
636 memcpy(ctrptr + AES_BLOCK_SIZE, ctrptr, AES_BLOCK_SIZE);
637 crypto_inc(ctrptr + AES_BLOCK_SIZE, AES_BLOCK_SIZE);
638 ctrptr += AES_BLOCK_SIZE;
639 }
640 return n;
641}
642
643static int ctr_paes_crypt(struct skcipher_request *req)
644{
645 struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
646 struct s390_paes_ctx *ctx = crypto_skcipher_ctx(tfm);
647 u8 buf[AES_BLOCK_SIZE], *ctrptr;
648 struct skcipher_walk walk;
649 unsigned int nbytes, n, k;
650 int ret, locked;
651 struct {
652 u8 key[MAXPROTKEYSIZE];
653 } param;
654
655 ret = skcipher_walk_virt(&walk, req, false);
656 if (ret)
657 return ret;
658
659 spin_lock_bh(&ctx->pk_lock);
660 memcpy(param.key, ctx->pk.protkey, MAXPROTKEYSIZE);
661 spin_unlock_bh(&ctx->pk_lock);
662
663 locked = mutex_trylock(&ctrblk_lock);
664
665 while ((nbytes = walk.nbytes) >= AES_BLOCK_SIZE) {
666 n = AES_BLOCK_SIZE;
667 if (nbytes >= 2*AES_BLOCK_SIZE && locked)
668 n = __ctrblk_init(ctrblk, walk.iv, nbytes);
669 ctrptr = (n > AES_BLOCK_SIZE) ? ctrblk : walk.iv;
670 k = cpacf_kmctr(ctx->fc, ¶m, walk.dst.virt.addr,
671 walk.src.virt.addr, n, ctrptr);
672 if (k) {
673 if (ctrptr == ctrblk)
674 memcpy(walk.iv, ctrptr + k - AES_BLOCK_SIZE,
675 AES_BLOCK_SIZE);
676 crypto_inc(walk.iv, AES_BLOCK_SIZE);
677 ret = skcipher_walk_done(&walk, nbytes - k);
678 }
679 if (k < n) {
680 if (__paes_convert_key(ctx)) {
681 if (locked)
682 mutex_unlock(&ctrblk_lock);
683 return skcipher_walk_done(&walk, -EIO);
684 }
685 spin_lock_bh(&ctx->pk_lock);
686 memcpy(param.key, ctx->pk.protkey, MAXPROTKEYSIZE);
687 spin_unlock_bh(&ctx->pk_lock);
688 }
689 }
690 if (locked)
691 mutex_unlock(&ctrblk_lock);
692 /*
693 * final block may be < AES_BLOCK_SIZE, copy only nbytes
694 */
695 if (nbytes) {
696 memset(buf, 0, AES_BLOCK_SIZE);
697 memcpy(buf, walk.src.virt.addr, nbytes);
698 while (1) {
699 if (cpacf_kmctr(ctx->fc, ¶m, buf,
700 buf, AES_BLOCK_SIZE,
701 walk.iv) == AES_BLOCK_SIZE)
702 break;
703 if (__paes_convert_key(ctx))
704 return skcipher_walk_done(&walk, -EIO);
705 spin_lock_bh(&ctx->pk_lock);
706 memcpy(param.key, ctx->pk.protkey, MAXPROTKEYSIZE);
707 spin_unlock_bh(&ctx->pk_lock);
708 }
709 memcpy(walk.dst.virt.addr, buf, nbytes);
710 crypto_inc(walk.iv, AES_BLOCK_SIZE);
711 ret = skcipher_walk_done(&walk, nbytes);
712 }
713
714 return ret;
715}
716
717static struct skcipher_alg ctr_paes_alg = {
718 .base.cra_name = "ctr(paes)",
719 .base.cra_driver_name = "ctr-paes-s390",
720 .base.cra_priority = 402, /* ecb-paes-s390 + 1 */
721 .base.cra_blocksize = 1,
722 .base.cra_ctxsize = sizeof(struct s390_paes_ctx),
723 .base.cra_module = THIS_MODULE,
724 .base.cra_list = LIST_HEAD_INIT(ctr_paes_alg.base.cra_list),
725 .init = ctr_paes_init,
726 .exit = ctr_paes_exit,
727 .min_keysize = PAES_MIN_KEYSIZE,
728 .max_keysize = PAES_MAX_KEYSIZE,
729 .ivsize = AES_BLOCK_SIZE,
730 .setkey = ctr_paes_set_key,
731 .encrypt = ctr_paes_crypt,
732 .decrypt = ctr_paes_crypt,
733 .chunksize = AES_BLOCK_SIZE,
734};
735
736static inline void __crypto_unregister_skcipher(struct skcipher_alg *alg)
737{
738 if (!list_empty(&alg->base.cra_list))
739 crypto_unregister_skcipher(alg);
740}
741
742static void paes_s390_fini(void)
743{
744 __crypto_unregister_skcipher(&ctr_paes_alg);
745 __crypto_unregister_skcipher(&xts_paes_alg);
746 __crypto_unregister_skcipher(&cbc_paes_alg);
747 __crypto_unregister_skcipher(&ecb_paes_alg);
748 if (ctrblk)
749 free_page((unsigned long) ctrblk);
750}
751
752static int __init paes_s390_init(void)
753{
754 int ret;
755
756 /* Query available functions for KM, KMC and KMCTR */
757 cpacf_query(CPACF_KM, &km_functions);
758 cpacf_query(CPACF_KMC, &kmc_functions);
759 cpacf_query(CPACF_KMCTR, &kmctr_functions);
760
761 if (cpacf_test_func(&km_functions, CPACF_KM_PAES_128) ||
762 cpacf_test_func(&km_functions, CPACF_KM_PAES_192) ||
763 cpacf_test_func(&km_functions, CPACF_KM_PAES_256)) {
764 ret = crypto_register_skcipher(&ecb_paes_alg);
765 if (ret)
766 goto out_err;
767 }
768
769 if (cpacf_test_func(&kmc_functions, CPACF_KMC_PAES_128) ||
770 cpacf_test_func(&kmc_functions, CPACF_KMC_PAES_192) ||
771 cpacf_test_func(&kmc_functions, CPACF_KMC_PAES_256)) {
772 ret = crypto_register_skcipher(&cbc_paes_alg);
773 if (ret)
774 goto out_err;
775 }
776
777 if (cpacf_test_func(&km_functions, CPACF_KM_PXTS_128) ||
778 cpacf_test_func(&km_functions, CPACF_KM_PXTS_256)) {
779 ret = crypto_register_skcipher(&xts_paes_alg);
780 if (ret)
781 goto out_err;
782 }
783
784 if (cpacf_test_func(&kmctr_functions, CPACF_KMCTR_PAES_128) ||
785 cpacf_test_func(&kmctr_functions, CPACF_KMCTR_PAES_192) ||
786 cpacf_test_func(&kmctr_functions, CPACF_KMCTR_PAES_256)) {
787 ctrblk = (u8 *) __get_free_page(GFP_KERNEL);
788 if (!ctrblk) {
789 ret = -ENOMEM;
790 goto out_err;
791 }
792 ret = crypto_register_skcipher(&ctr_paes_alg);
793 if (ret)
794 goto out_err;
795 }
796
797 return 0;
798out_err:
799 paes_s390_fini();
800 return ret;
801}
802
803module_init(paes_s390_init);
804module_exit(paes_s390_fini);
805
806MODULE_ALIAS_CRYPTO("paes");
807
808MODULE_DESCRIPTION("Rijndael (AES) Cipher Algorithm with protected keys");
809MODULE_LICENSE("GPL");
1// SPDX-License-Identifier: GPL-2.0
2/*
3 * Cryptographic API.
4 *
5 * s390 implementation of the AES Cipher Algorithm with protected keys.
6 *
7 * s390 Version:
8 * Copyright IBM Corp. 2017
9 * Author(s): Martin Schwidefsky <schwidefsky@de.ibm.com>
10 * Harald Freudenberger <freude@de.ibm.com>
11 */
12
13#define KMSG_COMPONENT "paes_s390"
14#define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
15
16#include <crypto/aes.h>
17#include <crypto/algapi.h>
18#include <linux/bug.h>
19#include <linux/err.h>
20#include <linux/module.h>
21#include <linux/cpufeature.h>
22#include <linux/init.h>
23#include <linux/spinlock.h>
24#include <crypto/xts.h>
25#include <asm/cpacf.h>
26#include <asm/pkey.h>
27
28static u8 *ctrblk;
29static DEFINE_SPINLOCK(ctrblk_lock);
30
31static cpacf_mask_t km_functions, kmc_functions, kmctr_functions;
32
33struct s390_paes_ctx {
34 struct pkey_seckey sk;
35 struct pkey_protkey pk;
36 unsigned long fc;
37};
38
39struct s390_pxts_ctx {
40 struct pkey_seckey sk[2];
41 struct pkey_protkey pk[2];
42 unsigned long fc;
43};
44
45static inline int __paes_convert_key(struct pkey_seckey *sk,
46 struct pkey_protkey *pk)
47{
48 int i, ret;
49
50 /* try three times in case of failure */
51 for (i = 0; i < 3; i++) {
52 ret = pkey_skey2pkey(sk, pk);
53 if (ret == 0)
54 break;
55 }
56
57 return ret;
58}
59
60static int __paes_set_key(struct s390_paes_ctx *ctx)
61{
62 unsigned long fc;
63
64 if (__paes_convert_key(&ctx->sk, &ctx->pk))
65 return -EINVAL;
66
67 /* Pick the correct function code based on the protected key type */
68 fc = (ctx->pk.type == PKEY_KEYTYPE_AES_128) ? CPACF_KM_PAES_128 :
69 (ctx->pk.type == PKEY_KEYTYPE_AES_192) ? CPACF_KM_PAES_192 :
70 (ctx->pk.type == PKEY_KEYTYPE_AES_256) ? CPACF_KM_PAES_256 : 0;
71
72 /* Check if the function code is available */
73 ctx->fc = (fc && cpacf_test_func(&km_functions, fc)) ? fc : 0;
74
75 return ctx->fc ? 0 : -EINVAL;
76}
77
78static int ecb_paes_set_key(struct crypto_tfm *tfm, const u8 *in_key,
79 unsigned int key_len)
80{
81 struct s390_paes_ctx *ctx = crypto_tfm_ctx(tfm);
82
83 if (key_len != SECKEYBLOBSIZE)
84 return -EINVAL;
85
86 memcpy(ctx->sk.seckey, in_key, SECKEYBLOBSIZE);
87 if (__paes_set_key(ctx)) {
88 tfm->crt_flags |= CRYPTO_TFM_RES_BAD_KEY_LEN;
89 return -EINVAL;
90 }
91 return 0;
92}
93
94static int ecb_paes_crypt(struct blkcipher_desc *desc,
95 unsigned long modifier,
96 struct blkcipher_walk *walk)
97{
98 struct s390_paes_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
99 unsigned int nbytes, n, k;
100 int ret;
101
102 ret = blkcipher_walk_virt(desc, walk);
103 while ((nbytes = walk->nbytes) >= AES_BLOCK_SIZE) {
104 /* only use complete blocks */
105 n = nbytes & ~(AES_BLOCK_SIZE - 1);
106 k = cpacf_km(ctx->fc | modifier, ctx->pk.protkey,
107 walk->dst.virt.addr, walk->src.virt.addr, n);
108 if (k)
109 ret = blkcipher_walk_done(desc, walk, nbytes - k);
110 if (k < n) {
111 if (__paes_set_key(ctx) != 0)
112 return blkcipher_walk_done(desc, walk, -EIO);
113 }
114 }
115 return ret;
116}
117
118static int ecb_paes_encrypt(struct blkcipher_desc *desc,
119 struct scatterlist *dst, struct scatterlist *src,
120 unsigned int nbytes)
121{
122 struct blkcipher_walk walk;
123
124 blkcipher_walk_init(&walk, dst, src, nbytes);
125 return ecb_paes_crypt(desc, CPACF_ENCRYPT, &walk);
126}
127
128static int ecb_paes_decrypt(struct blkcipher_desc *desc,
129 struct scatterlist *dst, struct scatterlist *src,
130 unsigned int nbytes)
131{
132 struct blkcipher_walk walk;
133
134 blkcipher_walk_init(&walk, dst, src, nbytes);
135 return ecb_paes_crypt(desc, CPACF_DECRYPT, &walk);
136}
137
138static struct crypto_alg ecb_paes_alg = {
139 .cra_name = "ecb(paes)",
140 .cra_driver_name = "ecb-paes-s390",
141 .cra_priority = 401, /* combo: aes + ecb + 1 */
142 .cra_flags = CRYPTO_ALG_TYPE_BLKCIPHER,
143 .cra_blocksize = AES_BLOCK_SIZE,
144 .cra_ctxsize = sizeof(struct s390_paes_ctx),
145 .cra_type = &crypto_blkcipher_type,
146 .cra_module = THIS_MODULE,
147 .cra_list = LIST_HEAD_INIT(ecb_paes_alg.cra_list),
148 .cra_u = {
149 .blkcipher = {
150 .min_keysize = SECKEYBLOBSIZE,
151 .max_keysize = SECKEYBLOBSIZE,
152 .setkey = ecb_paes_set_key,
153 .encrypt = ecb_paes_encrypt,
154 .decrypt = ecb_paes_decrypt,
155 }
156 }
157};
158
159static int __cbc_paes_set_key(struct s390_paes_ctx *ctx)
160{
161 unsigned long fc;
162
163 if (__paes_convert_key(&ctx->sk, &ctx->pk))
164 return -EINVAL;
165
166 /* Pick the correct function code based on the protected key type */
167 fc = (ctx->pk.type == PKEY_KEYTYPE_AES_128) ? CPACF_KMC_PAES_128 :
168 (ctx->pk.type == PKEY_KEYTYPE_AES_192) ? CPACF_KMC_PAES_192 :
169 (ctx->pk.type == PKEY_KEYTYPE_AES_256) ? CPACF_KMC_PAES_256 : 0;
170
171 /* Check if the function code is available */
172 ctx->fc = (fc && cpacf_test_func(&kmc_functions, fc)) ? fc : 0;
173
174 return ctx->fc ? 0 : -EINVAL;
175}
176
177static int cbc_paes_set_key(struct crypto_tfm *tfm, const u8 *in_key,
178 unsigned int key_len)
179{
180 struct s390_paes_ctx *ctx = crypto_tfm_ctx(tfm);
181
182 memcpy(ctx->sk.seckey, in_key, SECKEYBLOBSIZE);
183 if (__cbc_paes_set_key(ctx)) {
184 tfm->crt_flags |= CRYPTO_TFM_RES_BAD_KEY_LEN;
185 return -EINVAL;
186 }
187 return 0;
188}
189
190static int cbc_paes_crypt(struct blkcipher_desc *desc, unsigned long modifier,
191 struct blkcipher_walk *walk)
192{
193 struct s390_paes_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
194 unsigned int nbytes, n, k;
195 int ret;
196 struct {
197 u8 iv[AES_BLOCK_SIZE];
198 u8 key[MAXPROTKEYSIZE];
199 } param;
200
201 ret = blkcipher_walk_virt(desc, walk);
202 memcpy(param.iv, walk->iv, AES_BLOCK_SIZE);
203 memcpy(param.key, ctx->pk.protkey, MAXPROTKEYSIZE);
204 while ((nbytes = walk->nbytes) >= AES_BLOCK_SIZE) {
205 /* only use complete blocks */
206 n = nbytes & ~(AES_BLOCK_SIZE - 1);
207 k = cpacf_kmc(ctx->fc | modifier, ¶m,
208 walk->dst.virt.addr, walk->src.virt.addr, n);
209 if (k)
210 ret = blkcipher_walk_done(desc, walk, nbytes - k);
211 if (n < k) {
212 if (__cbc_paes_set_key(ctx) != 0)
213 return blkcipher_walk_done(desc, walk, -EIO);
214 memcpy(param.key, ctx->pk.protkey, MAXPROTKEYSIZE);
215 }
216 }
217 memcpy(walk->iv, param.iv, AES_BLOCK_SIZE);
218 return ret;
219}
220
221static int cbc_paes_encrypt(struct blkcipher_desc *desc,
222 struct scatterlist *dst, struct scatterlist *src,
223 unsigned int nbytes)
224{
225 struct blkcipher_walk walk;
226
227 blkcipher_walk_init(&walk, dst, src, nbytes);
228 return cbc_paes_crypt(desc, 0, &walk);
229}
230
231static int cbc_paes_decrypt(struct blkcipher_desc *desc,
232 struct scatterlist *dst, struct scatterlist *src,
233 unsigned int nbytes)
234{
235 struct blkcipher_walk walk;
236
237 blkcipher_walk_init(&walk, dst, src, nbytes);
238 return cbc_paes_crypt(desc, CPACF_DECRYPT, &walk);
239}
240
241static struct crypto_alg cbc_paes_alg = {
242 .cra_name = "cbc(paes)",
243 .cra_driver_name = "cbc-paes-s390",
244 .cra_priority = 402, /* ecb-paes-s390 + 1 */
245 .cra_flags = CRYPTO_ALG_TYPE_BLKCIPHER,
246 .cra_blocksize = AES_BLOCK_SIZE,
247 .cra_ctxsize = sizeof(struct s390_paes_ctx),
248 .cra_type = &crypto_blkcipher_type,
249 .cra_module = THIS_MODULE,
250 .cra_list = LIST_HEAD_INIT(cbc_paes_alg.cra_list),
251 .cra_u = {
252 .blkcipher = {
253 .min_keysize = SECKEYBLOBSIZE,
254 .max_keysize = SECKEYBLOBSIZE,
255 .ivsize = AES_BLOCK_SIZE,
256 .setkey = cbc_paes_set_key,
257 .encrypt = cbc_paes_encrypt,
258 .decrypt = cbc_paes_decrypt,
259 }
260 }
261};
262
263static int __xts_paes_set_key(struct s390_pxts_ctx *ctx)
264{
265 unsigned long fc;
266
267 if (__paes_convert_key(&ctx->sk[0], &ctx->pk[0]) ||
268 __paes_convert_key(&ctx->sk[1], &ctx->pk[1]))
269 return -EINVAL;
270
271 if (ctx->pk[0].type != ctx->pk[1].type)
272 return -EINVAL;
273
274 /* Pick the correct function code based on the protected key type */
275 fc = (ctx->pk[0].type == PKEY_KEYTYPE_AES_128) ? CPACF_KM_PXTS_128 :
276 (ctx->pk[0].type == PKEY_KEYTYPE_AES_256) ?
277 CPACF_KM_PXTS_256 : 0;
278
279 /* Check if the function code is available */
280 ctx->fc = (fc && cpacf_test_func(&km_functions, fc)) ? fc : 0;
281
282 return ctx->fc ? 0 : -EINVAL;
283}
284
285static int xts_paes_set_key(struct crypto_tfm *tfm, const u8 *in_key,
286 unsigned int key_len)
287{
288 struct s390_pxts_ctx *ctx = crypto_tfm_ctx(tfm);
289 u8 ckey[2 * AES_MAX_KEY_SIZE];
290 unsigned int ckey_len;
291
292 memcpy(ctx->sk[0].seckey, in_key, SECKEYBLOBSIZE);
293 memcpy(ctx->sk[1].seckey, in_key + SECKEYBLOBSIZE, SECKEYBLOBSIZE);
294 if (__xts_paes_set_key(ctx)) {
295 tfm->crt_flags |= CRYPTO_TFM_RES_BAD_KEY_LEN;
296 return -EINVAL;
297 }
298
299 /*
300 * xts_check_key verifies the key length is not odd and makes
301 * sure that the two keys are not the same. This can be done
302 * on the two protected keys as well
303 */
304 ckey_len = (ctx->pk[0].type == PKEY_KEYTYPE_AES_128) ?
305 AES_KEYSIZE_128 : AES_KEYSIZE_256;
306 memcpy(ckey, ctx->pk[0].protkey, ckey_len);
307 memcpy(ckey + ckey_len, ctx->pk[1].protkey, ckey_len);
308 return xts_check_key(tfm, ckey, 2*ckey_len);
309}
310
311static int xts_paes_crypt(struct blkcipher_desc *desc, unsigned long modifier,
312 struct blkcipher_walk *walk)
313{
314 struct s390_pxts_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
315 unsigned int keylen, offset, nbytes, n, k;
316 int ret;
317 struct {
318 u8 key[MAXPROTKEYSIZE]; /* key + verification pattern */
319 u8 tweak[16];
320 u8 block[16];
321 u8 bit[16];
322 u8 xts[16];
323 } pcc_param;
324 struct {
325 u8 key[MAXPROTKEYSIZE]; /* key + verification pattern */
326 u8 init[16];
327 } xts_param;
328
329 ret = blkcipher_walk_virt(desc, walk);
330 keylen = (ctx->pk[0].type == PKEY_KEYTYPE_AES_128) ? 48 : 64;
331 offset = (ctx->pk[0].type == PKEY_KEYTYPE_AES_128) ? 16 : 0;
332retry:
333 memset(&pcc_param, 0, sizeof(pcc_param));
334 memcpy(pcc_param.tweak, walk->iv, sizeof(pcc_param.tweak));
335 memcpy(pcc_param.key + offset, ctx->pk[1].protkey, keylen);
336 cpacf_pcc(ctx->fc, pcc_param.key + offset);
337
338 memcpy(xts_param.key + offset, ctx->pk[0].protkey, keylen);
339 memcpy(xts_param.init, pcc_param.xts, 16);
340
341 while ((nbytes = walk->nbytes) >= AES_BLOCK_SIZE) {
342 /* only use complete blocks */
343 n = nbytes & ~(AES_BLOCK_SIZE - 1);
344 k = cpacf_km(ctx->fc | modifier, xts_param.key + offset,
345 walk->dst.virt.addr, walk->src.virt.addr, n);
346 if (k)
347 ret = blkcipher_walk_done(desc, walk, nbytes - k);
348 if (k < n) {
349 if (__xts_paes_set_key(ctx) != 0)
350 return blkcipher_walk_done(desc, walk, -EIO);
351 goto retry;
352 }
353 }
354 return ret;
355}
356
357static int xts_paes_encrypt(struct blkcipher_desc *desc,
358 struct scatterlist *dst, struct scatterlist *src,
359 unsigned int nbytes)
360{
361 struct blkcipher_walk walk;
362
363 blkcipher_walk_init(&walk, dst, src, nbytes);
364 return xts_paes_crypt(desc, 0, &walk);
365}
366
367static int xts_paes_decrypt(struct blkcipher_desc *desc,
368 struct scatterlist *dst, struct scatterlist *src,
369 unsigned int nbytes)
370{
371 struct blkcipher_walk walk;
372
373 blkcipher_walk_init(&walk, dst, src, nbytes);
374 return xts_paes_crypt(desc, CPACF_DECRYPT, &walk);
375}
376
377static struct crypto_alg xts_paes_alg = {
378 .cra_name = "xts(paes)",
379 .cra_driver_name = "xts-paes-s390",
380 .cra_priority = 402, /* ecb-paes-s390 + 1 */
381 .cra_flags = CRYPTO_ALG_TYPE_BLKCIPHER,
382 .cra_blocksize = AES_BLOCK_SIZE,
383 .cra_ctxsize = sizeof(struct s390_pxts_ctx),
384 .cra_type = &crypto_blkcipher_type,
385 .cra_module = THIS_MODULE,
386 .cra_list = LIST_HEAD_INIT(xts_paes_alg.cra_list),
387 .cra_u = {
388 .blkcipher = {
389 .min_keysize = 2 * SECKEYBLOBSIZE,
390 .max_keysize = 2 * SECKEYBLOBSIZE,
391 .ivsize = AES_BLOCK_SIZE,
392 .setkey = xts_paes_set_key,
393 .encrypt = xts_paes_encrypt,
394 .decrypt = xts_paes_decrypt,
395 }
396 }
397};
398
399static int __ctr_paes_set_key(struct s390_paes_ctx *ctx)
400{
401 unsigned long fc;
402
403 if (__paes_convert_key(&ctx->sk, &ctx->pk))
404 return -EINVAL;
405
406 /* Pick the correct function code based on the protected key type */
407 fc = (ctx->pk.type == PKEY_KEYTYPE_AES_128) ? CPACF_KMCTR_PAES_128 :
408 (ctx->pk.type == PKEY_KEYTYPE_AES_192) ? CPACF_KMCTR_PAES_192 :
409 (ctx->pk.type == PKEY_KEYTYPE_AES_256) ?
410 CPACF_KMCTR_PAES_256 : 0;
411
412 /* Check if the function code is available */
413 ctx->fc = (fc && cpacf_test_func(&kmctr_functions, fc)) ? fc : 0;
414
415 return ctx->fc ? 0 : -EINVAL;
416}
417
418static int ctr_paes_set_key(struct crypto_tfm *tfm, const u8 *in_key,
419 unsigned int key_len)
420{
421 struct s390_paes_ctx *ctx = crypto_tfm_ctx(tfm);
422
423 memcpy(ctx->sk.seckey, in_key, key_len);
424 if (__ctr_paes_set_key(ctx)) {
425 tfm->crt_flags |= CRYPTO_TFM_RES_BAD_KEY_LEN;
426 return -EINVAL;
427 }
428 return 0;
429}
430
431static unsigned int __ctrblk_init(u8 *ctrptr, u8 *iv, unsigned int nbytes)
432{
433 unsigned int i, n;
434
435 /* only use complete blocks, max. PAGE_SIZE */
436 memcpy(ctrptr, iv, AES_BLOCK_SIZE);
437 n = (nbytes > PAGE_SIZE) ? PAGE_SIZE : nbytes & ~(AES_BLOCK_SIZE - 1);
438 for (i = (n / AES_BLOCK_SIZE) - 1; i > 0; i--) {
439 memcpy(ctrptr + AES_BLOCK_SIZE, ctrptr, AES_BLOCK_SIZE);
440 crypto_inc(ctrptr + AES_BLOCK_SIZE, AES_BLOCK_SIZE);
441 ctrptr += AES_BLOCK_SIZE;
442 }
443 return n;
444}
445
446static int ctr_paes_crypt(struct blkcipher_desc *desc, unsigned long modifier,
447 struct blkcipher_walk *walk)
448{
449 struct s390_paes_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
450 u8 buf[AES_BLOCK_SIZE], *ctrptr;
451 unsigned int nbytes, n, k;
452 int ret, locked;
453
454 locked = spin_trylock(&ctrblk_lock);
455
456 ret = blkcipher_walk_virt_block(desc, walk, AES_BLOCK_SIZE);
457 while ((nbytes = walk->nbytes) >= AES_BLOCK_SIZE) {
458 n = AES_BLOCK_SIZE;
459 if (nbytes >= 2*AES_BLOCK_SIZE && locked)
460 n = __ctrblk_init(ctrblk, walk->iv, nbytes);
461 ctrptr = (n > AES_BLOCK_SIZE) ? ctrblk : walk->iv;
462 k = cpacf_kmctr(ctx->fc | modifier, ctx->pk.protkey,
463 walk->dst.virt.addr, walk->src.virt.addr,
464 n, ctrptr);
465 if (k) {
466 if (ctrptr == ctrblk)
467 memcpy(walk->iv, ctrptr + k - AES_BLOCK_SIZE,
468 AES_BLOCK_SIZE);
469 crypto_inc(walk->iv, AES_BLOCK_SIZE);
470 ret = blkcipher_walk_done(desc, walk, nbytes - n);
471 }
472 if (k < n) {
473 if (__ctr_paes_set_key(ctx) != 0) {
474 if (locked)
475 spin_unlock(&ctrblk_lock);
476 return blkcipher_walk_done(desc, walk, -EIO);
477 }
478 }
479 }
480 if (locked)
481 spin_unlock(&ctrblk_lock);
482 /*
483 * final block may be < AES_BLOCK_SIZE, copy only nbytes
484 */
485 if (nbytes) {
486 while (1) {
487 if (cpacf_kmctr(ctx->fc | modifier,
488 ctx->pk.protkey, buf,
489 walk->src.virt.addr, AES_BLOCK_SIZE,
490 walk->iv) == AES_BLOCK_SIZE)
491 break;
492 if (__ctr_paes_set_key(ctx) != 0)
493 return blkcipher_walk_done(desc, walk, -EIO);
494 }
495 memcpy(walk->dst.virt.addr, buf, nbytes);
496 crypto_inc(walk->iv, AES_BLOCK_SIZE);
497 ret = blkcipher_walk_done(desc, walk, 0);
498 }
499
500 return ret;
501}
502
503static int ctr_paes_encrypt(struct blkcipher_desc *desc,
504 struct scatterlist *dst, struct scatterlist *src,
505 unsigned int nbytes)
506{
507 struct blkcipher_walk walk;
508
509 blkcipher_walk_init(&walk, dst, src, nbytes);
510 return ctr_paes_crypt(desc, 0, &walk);
511}
512
513static int ctr_paes_decrypt(struct blkcipher_desc *desc,
514 struct scatterlist *dst, struct scatterlist *src,
515 unsigned int nbytes)
516{
517 struct blkcipher_walk walk;
518
519 blkcipher_walk_init(&walk, dst, src, nbytes);
520 return ctr_paes_crypt(desc, CPACF_DECRYPT, &walk);
521}
522
523static struct crypto_alg ctr_paes_alg = {
524 .cra_name = "ctr(paes)",
525 .cra_driver_name = "ctr-paes-s390",
526 .cra_priority = 402, /* ecb-paes-s390 + 1 */
527 .cra_flags = CRYPTO_ALG_TYPE_BLKCIPHER,
528 .cra_blocksize = 1,
529 .cra_ctxsize = sizeof(struct s390_paes_ctx),
530 .cra_type = &crypto_blkcipher_type,
531 .cra_module = THIS_MODULE,
532 .cra_list = LIST_HEAD_INIT(ctr_paes_alg.cra_list),
533 .cra_u = {
534 .blkcipher = {
535 .min_keysize = SECKEYBLOBSIZE,
536 .max_keysize = SECKEYBLOBSIZE,
537 .ivsize = AES_BLOCK_SIZE,
538 .setkey = ctr_paes_set_key,
539 .encrypt = ctr_paes_encrypt,
540 .decrypt = ctr_paes_decrypt,
541 }
542 }
543};
544
545static inline void __crypto_unregister_alg(struct crypto_alg *alg)
546{
547 if (!list_empty(&alg->cra_list))
548 crypto_unregister_alg(alg);
549}
550
551static void paes_s390_fini(void)
552{
553 if (ctrblk)
554 free_page((unsigned long) ctrblk);
555 __crypto_unregister_alg(&ctr_paes_alg);
556 __crypto_unregister_alg(&xts_paes_alg);
557 __crypto_unregister_alg(&cbc_paes_alg);
558 __crypto_unregister_alg(&ecb_paes_alg);
559}
560
561static int __init paes_s390_init(void)
562{
563 int ret;
564
565 /* Query available functions for KM, KMC and KMCTR */
566 cpacf_query(CPACF_KM, &km_functions);
567 cpacf_query(CPACF_KMC, &kmc_functions);
568 cpacf_query(CPACF_KMCTR, &kmctr_functions);
569
570 if (cpacf_test_func(&km_functions, CPACF_KM_PAES_128) ||
571 cpacf_test_func(&km_functions, CPACF_KM_PAES_192) ||
572 cpacf_test_func(&km_functions, CPACF_KM_PAES_256)) {
573 ret = crypto_register_alg(&ecb_paes_alg);
574 if (ret)
575 goto out_err;
576 }
577
578 if (cpacf_test_func(&kmc_functions, CPACF_KMC_PAES_128) ||
579 cpacf_test_func(&kmc_functions, CPACF_KMC_PAES_192) ||
580 cpacf_test_func(&kmc_functions, CPACF_KMC_PAES_256)) {
581 ret = crypto_register_alg(&cbc_paes_alg);
582 if (ret)
583 goto out_err;
584 }
585
586 if (cpacf_test_func(&km_functions, CPACF_KM_PXTS_128) ||
587 cpacf_test_func(&km_functions, CPACF_KM_PXTS_256)) {
588 ret = crypto_register_alg(&xts_paes_alg);
589 if (ret)
590 goto out_err;
591 }
592
593 if (cpacf_test_func(&kmctr_functions, CPACF_KMCTR_PAES_128) ||
594 cpacf_test_func(&kmctr_functions, CPACF_KMCTR_PAES_192) ||
595 cpacf_test_func(&kmctr_functions, CPACF_KMCTR_PAES_256)) {
596 ret = crypto_register_alg(&ctr_paes_alg);
597 if (ret)
598 goto out_err;
599 ctrblk = (u8 *) __get_free_page(GFP_KERNEL);
600 if (!ctrblk) {
601 ret = -ENOMEM;
602 goto out_err;
603 }
604 }
605
606 return 0;
607out_err:
608 paes_s390_fini();
609 return ret;
610}
611
612module_init(paes_s390_init);
613module_exit(paes_s390_fini);
614
615MODULE_ALIAS_CRYPTO("paes");
616
617MODULE_DESCRIPTION("Rijndael (AES) Cipher Algorithm with protected keys");
618MODULE_LICENSE("GPL");