blob: 28556fce42671e2f182d5239d3dc6468e5b1d970 [file] [log] [blame]
Herbert Xu8ff59092010-10-19 21:31:55 +08001/*
2 * algif_skcipher: User-space interface for skcipher algorithms
3 *
4 * This file provides the user-space API for symmetric key ciphers.
5 *
6 * Copyright (c) 2010 Herbert Xu <herbert@gondor.apana.org.au>
7 *
8 * This program is free software; you can redistribute it and/or modify it
9 * under the terms of the GNU General Public License as published by the Free
10 * Software Foundation; either version 2 of the License, or (at your option)
11 * any later version.
12 *
13 */
14
15#include <crypto/scatterwalk.h>
16#include <crypto/skcipher.h>
17#include <crypto/if_alg.h>
18#include <linux/init.h>
19#include <linux/list.h>
20#include <linux/kernel.h>
21#include <linux/mm.h>
22#include <linux/module.h>
23#include <linux/net.h>
24#include <net/sock.h>
25
26struct skcipher_sg_list {
27 struct list_head list;
28
29 int cur;
30
31 struct scatterlist sg[0];
32};
33
Herbert Xudd504582015-12-25 15:40:05 +080034struct skcipher_tfm {
35 struct crypto_skcipher *skcipher;
36 bool has_key;
37};
38
Herbert Xu8ff59092010-10-19 21:31:55 +080039struct skcipher_ctx {
40 struct list_head tsgl;
41 struct af_alg_sgl rsgl;
42
43 void *iv;
44
45 struct af_alg_completion completion;
46
Tadeusz Struka5969992015-03-19 12:31:40 -070047 atomic_t inflight;
LABBE Corentin652d5b82015-10-23 14:10:36 +020048 size_t used;
Herbert Xu8ff59092010-10-19 21:31:55 +080049
50 unsigned int len;
51 bool more;
52 bool merge;
53 bool enc;
54
Herbert Xu0d96e4b2015-12-18 19:16:57 +080055 struct skcipher_request req;
Herbert Xu8ff59092010-10-19 21:31:55 +080056};
57
Tadeusz Struka5969992015-03-19 12:31:40 -070058struct skcipher_async_rsgl {
59 struct af_alg_sgl sgl;
60 struct list_head list;
61};
62
63struct skcipher_async_req {
64 struct kiocb *iocb;
65 struct skcipher_async_rsgl first_sgl;
66 struct list_head list;
67 struct scatterlist *tsg;
Herbert Xuec69bbf2016-02-03 21:39:24 +080068 atomic_t *inflight;
69 struct skcipher_request req;
Tadeusz Struka5969992015-03-19 12:31:40 -070070};
71
Ondrej Kozinae2cffb52014-08-25 11:49:54 +020072#define MAX_SGL_ENTS ((4096 - sizeof(struct skcipher_sg_list)) / \
Herbert Xu8ff59092010-10-19 21:31:55 +080073 sizeof(struct scatterlist) - 1)
74
Tadeusz Struka5969992015-03-19 12:31:40 -070075static void skcipher_free_async_sgls(struct skcipher_async_req *sreq)
76{
77 struct skcipher_async_rsgl *rsgl, *tmp;
78 struct scatterlist *sgl;
79 struct scatterlist *sg;
80 int i, n;
81
82 list_for_each_entry_safe(rsgl, tmp, &sreq->list, list) {
83 af_alg_free_sg(&rsgl->sgl);
84 if (rsgl != &sreq->first_sgl)
85 kfree(rsgl);
86 }
87 sgl = sreq->tsg;
88 n = sg_nents(sgl);
89 for_each_sg(sgl, sg, n, i)
90 put_page(sg_page(sg));
91
92 kfree(sreq->tsg);
93}
94
95static void skcipher_async_cb(struct crypto_async_request *req, int err)
96{
Herbert Xuec69bbf2016-02-03 21:39:24 +080097 struct skcipher_async_req *sreq = req->data;
Tadeusz Struka5969992015-03-19 12:31:40 -070098 struct kiocb *iocb = sreq->iocb;
99
Herbert Xuec69bbf2016-02-03 21:39:24 +0800100 atomic_dec(sreq->inflight);
Tadeusz Struka5969992015-03-19 12:31:40 -0700101 skcipher_free_async_sgls(sreq);
Herbert Xuec69bbf2016-02-03 21:39:24 +0800102 kzfree(sreq);
Al Viro237dae82015-04-09 00:00:30 -0400103 iocb->ki_complete(iocb, err, err);
Tadeusz Struka5969992015-03-19 12:31:40 -0700104}
105
Herbert Xu0f6bb832010-11-30 16:49:02 +0800106static inline int skcipher_sndbuf(struct sock *sk)
Herbert Xu8ff59092010-10-19 21:31:55 +0800107{
108 struct alg_sock *ask = alg_sk(sk);
109 struct skcipher_ctx *ctx = ask->private;
110
Herbert Xu0f6bb832010-11-30 16:49:02 +0800111 return max_t(int, max_t(int, sk->sk_sndbuf & PAGE_MASK, PAGE_SIZE) -
112 ctx->used, 0);
113}
114
115static inline bool skcipher_writable(struct sock *sk)
116{
117 return PAGE_SIZE <= skcipher_sndbuf(sk);
Herbert Xu8ff59092010-10-19 21:31:55 +0800118}
119
120static int skcipher_alloc_sgl(struct sock *sk)
121{
122 struct alg_sock *ask = alg_sk(sk);
123 struct skcipher_ctx *ctx = ask->private;
124 struct skcipher_sg_list *sgl;
125 struct scatterlist *sg = NULL;
126
127 sgl = list_entry(ctx->tsgl.prev, struct skcipher_sg_list, list);
128 if (!list_empty(&ctx->tsgl))
129 sg = sgl->sg;
130
131 if (!sg || sgl->cur >= MAX_SGL_ENTS) {
132 sgl = sock_kmalloc(sk, sizeof(*sgl) +
133 sizeof(sgl->sg[0]) * (MAX_SGL_ENTS + 1),
134 GFP_KERNEL);
135 if (!sgl)
136 return -ENOMEM;
137
138 sg_init_table(sgl->sg, MAX_SGL_ENTS + 1);
139 sgl->cur = 0;
140
141 if (sg)
Dan Williamsc56f6d12015-08-07 18:15:13 +0200142 sg_chain(sg, MAX_SGL_ENTS + 1, sgl->sg);
Herbert Xu8ff59092010-10-19 21:31:55 +0800143
144 list_add_tail(&sgl->list, &ctx->tsgl);
145 }
146
147 return 0;
148}
149
LABBE Corentin652d5b82015-10-23 14:10:36 +0200150static void skcipher_pull_sgl(struct sock *sk, size_t used, int put)
Herbert Xu8ff59092010-10-19 21:31:55 +0800151{
152 struct alg_sock *ask = alg_sk(sk);
153 struct skcipher_ctx *ctx = ask->private;
154 struct skcipher_sg_list *sgl;
155 struct scatterlist *sg;
156 int i;
157
158 while (!list_empty(&ctx->tsgl)) {
159 sgl = list_first_entry(&ctx->tsgl, struct skcipher_sg_list,
160 list);
161 sg = sgl->sg;
162
163 for (i = 0; i < sgl->cur; i++) {
LABBE Corentin652d5b82015-10-23 14:10:36 +0200164 size_t plen = min_t(size_t, used, sg[i].length);
Herbert Xu8ff59092010-10-19 21:31:55 +0800165
166 if (!sg_page(sg + i))
167 continue;
168
169 sg[i].length -= plen;
170 sg[i].offset += plen;
171
172 used -= plen;
173 ctx->used -= plen;
174
175 if (sg[i].length)
176 return;
Tadeusz Struka5969992015-03-19 12:31:40 -0700177 if (put)
178 put_page(sg_page(sg + i));
Herbert Xu8ff59092010-10-19 21:31:55 +0800179 sg_assign_page(sg + i, NULL);
180 }
181
182 list_del(&sgl->list);
183 sock_kfree_s(sk, sgl,
184 sizeof(*sgl) + sizeof(sgl->sg[0]) *
185 (MAX_SGL_ENTS + 1));
186 }
187
188 if (!ctx->used)
189 ctx->merge = 0;
190}
191
192static void skcipher_free_sgl(struct sock *sk)
193{
194 struct alg_sock *ask = alg_sk(sk);
195 struct skcipher_ctx *ctx = ask->private;
196
Tadeusz Struka5969992015-03-19 12:31:40 -0700197 skcipher_pull_sgl(sk, ctx->used, 1);
Herbert Xu8ff59092010-10-19 21:31:55 +0800198}
199
200static int skcipher_wait_for_wmem(struct sock *sk, unsigned flags)
201{
202 long timeout;
203 DEFINE_WAIT(wait);
204 int err = -ERESTARTSYS;
205
206 if (flags & MSG_DONTWAIT)
207 return -EAGAIN;
208
Eric Dumazet9cd3e072015-11-29 20:03:10 -0800209 sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
Herbert Xu8ff59092010-10-19 21:31:55 +0800210
211 for (;;) {
212 if (signal_pending(current))
213 break;
214 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
215 timeout = MAX_SCHEDULE_TIMEOUT;
216 if (sk_wait_event(sk, &timeout, skcipher_writable(sk))) {
217 err = 0;
218 break;
219 }
220 }
221 finish_wait(sk_sleep(sk), &wait);
222
223 return err;
224}
225
226static void skcipher_wmem_wakeup(struct sock *sk)
227{
228 struct socket_wq *wq;
229
230 if (!skcipher_writable(sk))
231 return;
232
233 rcu_read_lock();
234 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +0800235 if (skwq_has_sleeper(wq))
Herbert Xu8ff59092010-10-19 21:31:55 +0800236 wake_up_interruptible_sync_poll(&wq->wait, POLLIN |
237 POLLRDNORM |
238 POLLRDBAND);
239 sk_wake_async(sk, SOCK_WAKE_WAITD, POLL_IN);
240 rcu_read_unlock();
241}
242
243static int skcipher_wait_for_data(struct sock *sk, unsigned flags)
244{
245 struct alg_sock *ask = alg_sk(sk);
246 struct skcipher_ctx *ctx = ask->private;
247 long timeout;
248 DEFINE_WAIT(wait);
249 int err = -ERESTARTSYS;
250
251 if (flags & MSG_DONTWAIT) {
252 return -EAGAIN;
253 }
254
Eric Dumazet9cd3e072015-11-29 20:03:10 -0800255 sk_set_bit(SOCKWQ_ASYNC_WAITDATA, sk);
Herbert Xu8ff59092010-10-19 21:31:55 +0800256
257 for (;;) {
258 if (signal_pending(current))
259 break;
260 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
261 timeout = MAX_SCHEDULE_TIMEOUT;
262 if (sk_wait_event(sk, &timeout, ctx->used)) {
263 err = 0;
264 break;
265 }
266 }
267 finish_wait(sk_sleep(sk), &wait);
268
Eric Dumazet9cd3e072015-11-29 20:03:10 -0800269 sk_clear_bit(SOCKWQ_ASYNC_WAITDATA, sk);
Herbert Xu8ff59092010-10-19 21:31:55 +0800270
271 return err;
272}
273
274static void skcipher_data_wakeup(struct sock *sk)
275{
276 struct alg_sock *ask = alg_sk(sk);
277 struct skcipher_ctx *ctx = ask->private;
278 struct socket_wq *wq;
279
280 if (!ctx->used)
281 return;
282
283 rcu_read_lock();
284 wq = rcu_dereference(sk->sk_wq);
Herbert Xu1ce0bf52015-11-26 13:55:39 +0800285 if (skwq_has_sleeper(wq))
Herbert Xu8ff59092010-10-19 21:31:55 +0800286 wake_up_interruptible_sync_poll(&wq->wait, POLLOUT |
287 POLLRDNORM |
288 POLLRDBAND);
289 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
290 rcu_read_unlock();
291}
292
Ying Xue1b784142015-03-02 15:37:48 +0800293static int skcipher_sendmsg(struct socket *sock, struct msghdr *msg,
294 size_t size)
Herbert Xu8ff59092010-10-19 21:31:55 +0800295{
296 struct sock *sk = sock->sk;
297 struct alg_sock *ask = alg_sk(sk);
Herbert Xu6454c2b2016-02-03 21:39:26 +0800298 struct sock *psk = ask->parent;
299 struct alg_sock *pask = alg_sk(psk);
Herbert Xu8ff59092010-10-19 21:31:55 +0800300 struct skcipher_ctx *ctx = ask->private;
Herbert Xu6454c2b2016-02-03 21:39:26 +0800301 struct skcipher_tfm *skc = pask->private;
302 struct crypto_skcipher *tfm = skc->skcipher;
Herbert Xu0d96e4b2015-12-18 19:16:57 +0800303 unsigned ivsize = crypto_skcipher_ivsize(tfm);
Herbert Xu8ff59092010-10-19 21:31:55 +0800304 struct skcipher_sg_list *sgl;
305 struct af_alg_control con = {};
306 long copied = 0;
307 bool enc = 0;
Stephan Muellerf26b7b82014-11-30 10:55:26 +0100308 bool init = 0;
Herbert Xu8ff59092010-10-19 21:31:55 +0800309 int err;
310 int i;
311
312 if (msg->msg_controllen) {
313 err = af_alg_cmsg_send(msg, &con);
314 if (err)
315 return err;
316
Stephan Muellerf26b7b82014-11-30 10:55:26 +0100317 init = 1;
Herbert Xu8ff59092010-10-19 21:31:55 +0800318 switch (con.op) {
319 case ALG_OP_ENCRYPT:
320 enc = 1;
321 break;
322 case ALG_OP_DECRYPT:
323 enc = 0;
324 break;
325 default:
326 return -EINVAL;
327 }
328
329 if (con.iv && con.iv->ivlen != ivsize)
330 return -EINVAL;
331 }
332
333 err = -EINVAL;
334
335 lock_sock(sk);
336 if (!ctx->more && ctx->used)
337 goto unlock;
338
Stephan Muellerf26b7b82014-11-30 10:55:26 +0100339 if (init) {
Herbert Xu8ff59092010-10-19 21:31:55 +0800340 ctx->enc = enc;
341 if (con.iv)
342 memcpy(ctx->iv, con.iv->iv, ivsize);
343 }
344
Herbert Xu8ff59092010-10-19 21:31:55 +0800345 while (size) {
346 struct scatterlist *sg;
347 unsigned long len = size;
LABBE Corentin652d5b82015-10-23 14:10:36 +0200348 size_t plen;
Herbert Xu8ff59092010-10-19 21:31:55 +0800349
350 if (ctx->merge) {
351 sgl = list_entry(ctx->tsgl.prev,
352 struct skcipher_sg_list, list);
353 sg = sgl->sg + sgl->cur - 1;
354 len = min_t(unsigned long, len,
355 PAGE_SIZE - sg->offset - sg->length);
356
Al Viro6ce8e9c2014-04-06 21:25:44 -0400357 err = memcpy_from_msg(page_address(sg_page(sg)) +
358 sg->offset + sg->length,
359 msg, len);
Herbert Xu8ff59092010-10-19 21:31:55 +0800360 if (err)
361 goto unlock;
362
363 sg->length += len;
364 ctx->merge = (sg->offset + sg->length) &
365 (PAGE_SIZE - 1);
366
367 ctx->used += len;
368 copied += len;
369 size -= len;
Herbert Xu8ff59092010-10-19 21:31:55 +0800370 continue;
371 }
372
Herbert Xu0f6bb832010-11-30 16:49:02 +0800373 if (!skcipher_writable(sk)) {
Herbert Xu8ff59092010-10-19 21:31:55 +0800374 err = skcipher_wait_for_wmem(sk, msg->msg_flags);
375 if (err)
376 goto unlock;
Herbert Xu8ff59092010-10-19 21:31:55 +0800377 }
378
Herbert Xu0f6bb832010-11-30 16:49:02 +0800379 len = min_t(unsigned long, len, skcipher_sndbuf(sk));
Herbert Xu8ff59092010-10-19 21:31:55 +0800380
381 err = skcipher_alloc_sgl(sk);
382 if (err)
383 goto unlock;
384
385 sgl = list_entry(ctx->tsgl.prev, struct skcipher_sg_list, list);
386 sg = sgl->sg;
Herbert Xu202736d2016-01-19 21:23:57 +0800387 if (sgl->cur)
388 sg_unmark_end(sg + sgl->cur - 1);
Herbert Xu8ff59092010-10-19 21:31:55 +0800389 do {
390 i = sgl->cur;
LABBE Corentin652d5b82015-10-23 14:10:36 +0200391 plen = min_t(size_t, len, PAGE_SIZE);
Herbert Xu8ff59092010-10-19 21:31:55 +0800392
393 sg_assign_page(sg + i, alloc_page(GFP_KERNEL));
394 err = -ENOMEM;
395 if (!sg_page(sg + i))
396 goto unlock;
397
Al Viro6ce8e9c2014-04-06 21:25:44 -0400398 err = memcpy_from_msg(page_address(sg_page(sg + i)),
399 msg, plen);
Herbert Xu8ff59092010-10-19 21:31:55 +0800400 if (err) {
401 __free_page(sg_page(sg + i));
402 sg_assign_page(sg + i, NULL);
403 goto unlock;
404 }
405
406 sg[i].length = plen;
407 len -= plen;
408 ctx->used += plen;
409 copied += plen;
410 size -= plen;
Herbert Xu8ff59092010-10-19 21:31:55 +0800411 sgl->cur++;
412 } while (len && sgl->cur < MAX_SGL_ENTS);
413
Tadeusz Struk0f477b62014-12-08 12:03:42 -0800414 if (!size)
415 sg_mark_end(sg + sgl->cur - 1);
416
Herbert Xu8ff59092010-10-19 21:31:55 +0800417 ctx->merge = plen & (PAGE_SIZE - 1);
418 }
419
420 err = 0;
421
422 ctx->more = msg->msg_flags & MSG_MORE;
Herbert Xu8ff59092010-10-19 21:31:55 +0800423
424unlock:
425 skcipher_data_wakeup(sk);
426 release_sock(sk);
427
428 return copied ?: err;
429}
430
431static ssize_t skcipher_sendpage(struct socket *sock, struct page *page,
432 int offset, size_t size, int flags)
433{
434 struct sock *sk = sock->sk;
435 struct alg_sock *ask = alg_sk(sk);
436 struct skcipher_ctx *ctx = ask->private;
437 struct skcipher_sg_list *sgl;
438 int err = -EINVAL;
Herbert Xu8ff59092010-10-19 21:31:55 +0800439
Shawn Landdend3f7d562013-11-24 22:36:28 -0800440 if (flags & MSG_SENDPAGE_NOTLAST)
441 flags |= MSG_MORE;
442
Herbert Xu8ff59092010-10-19 21:31:55 +0800443 lock_sock(sk);
444 if (!ctx->more && ctx->used)
445 goto unlock;
446
447 if (!size)
448 goto done;
449
Herbert Xu0f6bb832010-11-30 16:49:02 +0800450 if (!skcipher_writable(sk)) {
Herbert Xu8ff59092010-10-19 21:31:55 +0800451 err = skcipher_wait_for_wmem(sk, flags);
452 if (err)
453 goto unlock;
Herbert Xu8ff59092010-10-19 21:31:55 +0800454 }
455
456 err = skcipher_alloc_sgl(sk);
457 if (err)
458 goto unlock;
459
460 ctx->merge = 0;
461 sgl = list_entry(ctx->tsgl.prev, struct skcipher_sg_list, list);
462
Tadeusz Struk0f477b62014-12-08 12:03:42 -0800463 if (sgl->cur)
464 sg_unmark_end(sgl->sg + sgl->cur - 1);
465
466 sg_mark_end(sgl->sg + sgl->cur);
Herbert Xu8ff59092010-10-19 21:31:55 +0800467 get_page(page);
468 sg_set_page(sgl->sg + sgl->cur, page, size, offset);
469 sgl->cur++;
470 ctx->used += size;
471
472done:
473 ctx->more = flags & MSG_MORE;
Herbert Xu8ff59092010-10-19 21:31:55 +0800474
475unlock:
476 skcipher_data_wakeup(sk);
477 release_sock(sk);
478
479 return err ?: size;
480}
481
Tadeusz Struka5969992015-03-19 12:31:40 -0700482static int skcipher_all_sg_nents(struct skcipher_ctx *ctx)
483{
484 struct skcipher_sg_list *sgl;
485 struct scatterlist *sg;
486 int nents = 0;
487
488 list_for_each_entry(sgl, &ctx->tsgl, list) {
489 sg = sgl->sg;
490
491 while (!sg->length)
492 sg++;
493
494 nents += sg_nents(sg);
495 }
496 return nents;
497}
498
499static int skcipher_recvmsg_async(struct socket *sock, struct msghdr *msg,
500 int flags)
501{
502 struct sock *sk = sock->sk;
503 struct alg_sock *ask = alg_sk(sk);
Herbert Xuec69bbf2016-02-03 21:39:24 +0800504 struct sock *psk = ask->parent;
505 struct alg_sock *pask = alg_sk(psk);
Tadeusz Struka5969992015-03-19 12:31:40 -0700506 struct skcipher_ctx *ctx = ask->private;
Herbert Xuec69bbf2016-02-03 21:39:24 +0800507 struct skcipher_tfm *skc = pask->private;
508 struct crypto_skcipher *tfm = skc->skcipher;
Tadeusz Struka5969992015-03-19 12:31:40 -0700509 struct skcipher_sg_list *sgl;
510 struct scatterlist *sg;
511 struct skcipher_async_req *sreq;
Herbert Xu0d96e4b2015-12-18 19:16:57 +0800512 struct skcipher_request *req;
Tadeusz Struka5969992015-03-19 12:31:40 -0700513 struct skcipher_async_rsgl *last_rsgl = NULL;
Herbert Xu6454c2b2016-02-03 21:39:26 +0800514 unsigned int txbufs = 0, len = 0, tx_nents;
Herbert Xuec69bbf2016-02-03 21:39:24 +0800515 unsigned int reqsize = crypto_skcipher_reqsize(tfm);
516 unsigned int ivsize = crypto_skcipher_ivsize(tfm);
Tadeusz Struka5969992015-03-19 12:31:40 -0700517 int err = -ENOMEM;
tadeusz.struk@intel.com033f46b2015-04-01 13:53:06 -0700518 bool mark = false;
Herbert Xuec69bbf2016-02-03 21:39:24 +0800519 char *iv;
520
521 sreq = kzalloc(sizeof(*sreq) + reqsize + ivsize, GFP_KERNEL);
522 if (unlikely(!sreq))
523 goto out;
524
525 req = &sreq->req;
526 iv = (char *)(req + 1) + reqsize;
527 sreq->iocb = msg->msg_iocb;
528 INIT_LIST_HEAD(&sreq->list);
529 sreq->inflight = &ctx->inflight;
Tadeusz Struka5969992015-03-19 12:31:40 -0700530
531 lock_sock(sk);
Herbert Xu6454c2b2016-02-03 21:39:26 +0800532 tx_nents = skcipher_all_sg_nents(ctx);
Tadeusz Struka5969992015-03-19 12:31:40 -0700533 sreq->tsg = kcalloc(tx_nents, sizeof(*sg), GFP_KERNEL);
Herbert Xuec69bbf2016-02-03 21:39:24 +0800534 if (unlikely(!sreq->tsg))
Tadeusz Struka5969992015-03-19 12:31:40 -0700535 goto unlock;
Tadeusz Struka5969992015-03-19 12:31:40 -0700536 sg_init_table(sreq->tsg, tx_nents);
Herbert Xuec69bbf2016-02-03 21:39:24 +0800537 memcpy(iv, ctx->iv, ivsize);
538 skcipher_request_set_tfm(req, tfm);
Herbert Xudad41992016-02-03 21:39:27 +0800539 skcipher_request_set_callback(req, CRYPTO_TFM_REQ_MAY_SLEEP,
Herbert Xuec69bbf2016-02-03 21:39:24 +0800540 skcipher_async_cb, sreq);
Tadeusz Struka5969992015-03-19 12:31:40 -0700541
542 while (iov_iter_count(&msg->msg_iter)) {
543 struct skcipher_async_rsgl *rsgl;
tadeusz.struk@intel.comac110f42015-03-25 07:29:19 -0700544 int used;
Tadeusz Struka5969992015-03-19 12:31:40 -0700545
546 if (!ctx->used) {
547 err = skcipher_wait_for_data(sk, flags);
548 if (err)
549 goto free;
550 }
551 sgl = list_first_entry(&ctx->tsgl,
552 struct skcipher_sg_list, list);
553 sg = sgl->sg;
554
555 while (!sg->length)
556 sg++;
557
558 used = min_t(unsigned long, ctx->used,
559 iov_iter_count(&msg->msg_iter));
560 used = min_t(unsigned long, used, sg->length);
561
tadeusz.struk@intel.com033f46b2015-04-01 13:53:06 -0700562 if (txbufs == tx_nents) {
Tadeusz Struka5969992015-03-19 12:31:40 -0700563 struct scatterlist *tmp;
564 int x;
565 /* Ran out of tx slots in async request
566 * need to expand */
567 tmp = kcalloc(tx_nents * 2, sizeof(*tmp),
568 GFP_KERNEL);
569 if (!tmp)
570 goto free;
571
572 sg_init_table(tmp, tx_nents * 2);
573 for (x = 0; x < tx_nents; x++)
574 sg_set_page(&tmp[x], sg_page(&sreq->tsg[x]),
575 sreq->tsg[x].length,
576 sreq->tsg[x].offset);
577 kfree(sreq->tsg);
578 sreq->tsg = tmp;
579 tx_nents *= 2;
tadeusz.struk@intel.com033f46b2015-04-01 13:53:06 -0700580 mark = true;
Tadeusz Struka5969992015-03-19 12:31:40 -0700581 }
582 /* Need to take over the tx sgl from ctx
583 * to the asynch req - these sgls will be freed later */
tadeusz.struk@intel.com033f46b2015-04-01 13:53:06 -0700584 sg_set_page(sreq->tsg + txbufs++, sg_page(sg), sg->length,
Tadeusz Struka5969992015-03-19 12:31:40 -0700585 sg->offset);
586
587 if (list_empty(&sreq->list)) {
588 rsgl = &sreq->first_sgl;
589 list_add_tail(&rsgl->list, &sreq->list);
590 } else {
Tadeusz Struk82d92922015-03-30 14:25:44 -0700591 rsgl = kmalloc(sizeof(*rsgl), GFP_KERNEL);
Tadeusz Struka5969992015-03-19 12:31:40 -0700592 if (!rsgl) {
593 err = -ENOMEM;
594 goto free;
595 }
596 list_add_tail(&rsgl->list, &sreq->list);
597 }
598
599 used = af_alg_make_sg(&rsgl->sgl, &msg->msg_iter, used);
600 err = used;
601 if (used < 0)
602 goto free;
603 if (last_rsgl)
604 af_alg_link_sg(&last_rsgl->sgl, &rsgl->sgl);
605
606 last_rsgl = rsgl;
607 len += used;
608 skcipher_pull_sgl(sk, used, 0);
609 iov_iter_advance(&msg->msg_iter, used);
610 }
611
tadeusz.struk@intel.com033f46b2015-04-01 13:53:06 -0700612 if (mark)
613 sg_mark_end(sreq->tsg + txbufs - 1);
614
Herbert Xu0d96e4b2015-12-18 19:16:57 +0800615 skcipher_request_set_crypt(req, sreq->tsg, sreq->first_sgl.sgl.sg,
Herbert Xuec69bbf2016-02-03 21:39:24 +0800616 len, iv);
Herbert Xu0d96e4b2015-12-18 19:16:57 +0800617 err = ctx->enc ? crypto_skcipher_encrypt(req) :
618 crypto_skcipher_decrypt(req);
Tadeusz Struka5969992015-03-19 12:31:40 -0700619 if (err == -EINPROGRESS) {
620 atomic_inc(&ctx->inflight);
621 err = -EIOCBQUEUED;
Herbert Xuec69bbf2016-02-03 21:39:24 +0800622 sreq = NULL;
Tadeusz Struka5969992015-03-19 12:31:40 -0700623 goto unlock;
624 }
625free:
626 skcipher_free_async_sgls(sreq);
Tadeusz Struka5969992015-03-19 12:31:40 -0700627unlock:
628 skcipher_wmem_wakeup(sk);
629 release_sock(sk);
Herbert Xuec69bbf2016-02-03 21:39:24 +0800630 kzfree(sreq);
631out:
Tadeusz Struka5969992015-03-19 12:31:40 -0700632 return err;
633}
634
635static int skcipher_recvmsg_sync(struct socket *sock, struct msghdr *msg,
636 int flags)
Herbert Xu8ff59092010-10-19 21:31:55 +0800637{
638 struct sock *sk = sock->sk;
639 struct alg_sock *ask = alg_sk(sk);
Herbert Xu6454c2b2016-02-03 21:39:26 +0800640 struct sock *psk = ask->parent;
641 struct alg_sock *pask = alg_sk(psk);
Herbert Xu8ff59092010-10-19 21:31:55 +0800642 struct skcipher_ctx *ctx = ask->private;
Herbert Xu6454c2b2016-02-03 21:39:26 +0800643 struct skcipher_tfm *skc = pask->private;
644 struct crypto_skcipher *tfm = skc->skcipher;
645 unsigned bs = crypto_skcipher_blocksize(tfm);
Herbert Xu8ff59092010-10-19 21:31:55 +0800646 struct skcipher_sg_list *sgl;
647 struct scatterlist *sg;
Herbert Xu8ff59092010-10-19 21:31:55 +0800648 int err = -EAGAIN;
649 int used;
650 long copied = 0;
651
652 lock_sock(sk);
Al Viro01e97e62014-12-15 21:39:31 -0500653 while (msg_data_left(msg)) {
Linus Torvalds9399f0c2015-02-10 19:55:45 -0800654 if (!ctx->used) {
Al Viro1d10eb22014-11-28 16:39:25 -0500655 err = skcipher_wait_for_data(sk, flags);
656 if (err)
Herbert Xubc97e572010-11-30 17:04:31 +0800657 goto unlock;
Al Viro1d10eb22014-11-28 16:39:25 -0500658 }
Herbert Xubc97e572010-11-30 17:04:31 +0800659
Al Viro01e97e62014-12-15 21:39:31 -0500660 used = min_t(unsigned long, ctx->used, msg_data_left(msg));
Herbert Xu8ff59092010-10-19 21:31:55 +0800661
Al Viro1d10eb22014-11-28 16:39:25 -0500662 used = af_alg_make_sg(&ctx->rsgl, &msg->msg_iter, used);
663 err = used;
664 if (err < 0)
665 goto unlock;
Herbert Xu8ff59092010-10-19 21:31:55 +0800666
Al Viro1d10eb22014-11-28 16:39:25 -0500667 if (ctx->more || used < ctx->used)
668 used -= used % bs;
Herbert Xu8ff59092010-10-19 21:31:55 +0800669
Al Viro1d10eb22014-11-28 16:39:25 -0500670 err = -EINVAL;
671 if (!used)
672 goto free;
673
Herbert Xu4f0414e2016-01-18 18:46:10 +0800674 sgl = list_first_entry(&ctx->tsgl,
675 struct skcipher_sg_list, list);
676 sg = sgl->sg;
677
678 while (!sg->length)
679 sg++;
680
Herbert Xu0d96e4b2015-12-18 19:16:57 +0800681 skcipher_request_set_crypt(&ctx->req, sg, ctx->rsgl.sg, used,
682 ctx->iv);
Al Viro1d10eb22014-11-28 16:39:25 -0500683
684 err = af_alg_wait_for_completion(
Herbert Xu8ff59092010-10-19 21:31:55 +0800685 ctx->enc ?
Herbert Xu0d96e4b2015-12-18 19:16:57 +0800686 crypto_skcipher_encrypt(&ctx->req) :
687 crypto_skcipher_decrypt(&ctx->req),
Herbert Xu8ff59092010-10-19 21:31:55 +0800688 &ctx->completion);
689
Herbert Xubc97e572010-11-30 17:04:31 +0800690free:
Al Viro1d10eb22014-11-28 16:39:25 -0500691 af_alg_free_sg(&ctx->rsgl);
Herbert Xu8ff59092010-10-19 21:31:55 +0800692
Al Viro1d10eb22014-11-28 16:39:25 -0500693 if (err)
694 goto unlock;
Herbert Xu8ff59092010-10-19 21:31:55 +0800695
Al Viro1d10eb22014-11-28 16:39:25 -0500696 copied += used;
Tadeusz Struka5969992015-03-19 12:31:40 -0700697 skcipher_pull_sgl(sk, used, 1);
Al Viro1d10eb22014-11-28 16:39:25 -0500698 iov_iter_advance(&msg->msg_iter, used);
Herbert Xu8ff59092010-10-19 21:31:55 +0800699 }
700
701 err = 0;
702
703unlock:
704 skcipher_wmem_wakeup(sk);
705 release_sock(sk);
706
707 return copied ?: err;
708}
709
Tadeusz Struka5969992015-03-19 12:31:40 -0700710static int skcipher_recvmsg(struct socket *sock, struct msghdr *msg,
711 size_t ignored, int flags)
712{
713 return (msg->msg_iocb && !is_sync_kiocb(msg->msg_iocb)) ?
714 skcipher_recvmsg_async(sock, msg, flags) :
715 skcipher_recvmsg_sync(sock, msg, flags);
716}
Herbert Xu8ff59092010-10-19 21:31:55 +0800717
718static unsigned int skcipher_poll(struct file *file, struct socket *sock,
719 poll_table *wait)
720{
721 struct sock *sk = sock->sk;
722 struct alg_sock *ask = alg_sk(sk);
723 struct skcipher_ctx *ctx = ask->private;
724 unsigned int mask;
725
726 sock_poll_wait(file, sk_sleep(sk), wait);
727 mask = 0;
728
729 if (ctx->used)
730 mask |= POLLIN | POLLRDNORM;
731
732 if (skcipher_writable(sk))
733 mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
734
735 return mask;
736}
737
738static struct proto_ops algif_skcipher_ops = {
739 .family = PF_ALG,
740
741 .connect = sock_no_connect,
742 .socketpair = sock_no_socketpair,
743 .getname = sock_no_getname,
744 .ioctl = sock_no_ioctl,
745 .listen = sock_no_listen,
746 .shutdown = sock_no_shutdown,
747 .getsockopt = sock_no_getsockopt,
748 .mmap = sock_no_mmap,
749 .bind = sock_no_bind,
750 .accept = sock_no_accept,
751 .setsockopt = sock_no_setsockopt,
752
753 .release = af_alg_release,
754 .sendmsg = skcipher_sendmsg,
755 .sendpage = skcipher_sendpage,
756 .recvmsg = skcipher_recvmsg,
757 .poll = skcipher_poll,
758};
759
Herbert Xua0fa2d02016-01-04 13:36:12 +0900760static int skcipher_check_key(struct socket *sock)
761{
Herbert Xu18227932016-01-15 22:02:20 +0800762 int err = 0;
Herbert Xua0fa2d02016-01-04 13:36:12 +0900763 struct sock *psk;
764 struct alg_sock *pask;
765 struct skcipher_tfm *tfm;
766 struct sock *sk = sock->sk;
767 struct alg_sock *ask = alg_sk(sk);
768
Herbert Xu18227932016-01-15 22:02:20 +0800769 lock_sock(sk);
Herbert Xua0fa2d02016-01-04 13:36:12 +0900770 if (ask->refcnt)
Herbert Xu18227932016-01-15 22:02:20 +0800771 goto unlock_child;
Herbert Xua0fa2d02016-01-04 13:36:12 +0900772
773 psk = ask->parent;
774 pask = alg_sk(ask->parent);
775 tfm = pask->private;
776
777 err = -ENOKEY;
Herbert Xu18227932016-01-15 22:02:20 +0800778 lock_sock_nested(psk, SINGLE_DEPTH_NESTING);
Herbert Xua0fa2d02016-01-04 13:36:12 +0900779 if (!tfm->has_key)
780 goto unlock;
781
782 if (!pask->refcnt++)
783 sock_hold(psk);
784
785 ask->refcnt = 1;
786 sock_put(psk);
787
788 err = 0;
789
790unlock:
791 release_sock(psk);
Herbert Xu18227932016-01-15 22:02:20 +0800792unlock_child:
793 release_sock(sk);
Herbert Xua0fa2d02016-01-04 13:36:12 +0900794
795 return err;
796}
797
798static int skcipher_sendmsg_nokey(struct socket *sock, struct msghdr *msg,
799 size_t size)
800{
801 int err;
802
803 err = skcipher_check_key(sock);
804 if (err)
805 return err;
806
807 return skcipher_sendmsg(sock, msg, size);
808}
809
810static ssize_t skcipher_sendpage_nokey(struct socket *sock, struct page *page,
811 int offset, size_t size, int flags)
812{
813 int err;
814
815 err = skcipher_check_key(sock);
816 if (err)
817 return err;
818
819 return skcipher_sendpage(sock, page, offset, size, flags);
820}
821
822static int skcipher_recvmsg_nokey(struct socket *sock, struct msghdr *msg,
823 size_t ignored, int flags)
824{
825 int err;
826
827 err = skcipher_check_key(sock);
828 if (err)
829 return err;
830
831 return skcipher_recvmsg(sock, msg, ignored, flags);
832}
833
834static struct proto_ops algif_skcipher_ops_nokey = {
835 .family = PF_ALG,
836
837 .connect = sock_no_connect,
838 .socketpair = sock_no_socketpair,
839 .getname = sock_no_getname,
840 .ioctl = sock_no_ioctl,
841 .listen = sock_no_listen,
842 .shutdown = sock_no_shutdown,
843 .getsockopt = sock_no_getsockopt,
844 .mmap = sock_no_mmap,
845 .bind = sock_no_bind,
846 .accept = sock_no_accept,
847 .setsockopt = sock_no_setsockopt,
848
849 .release = af_alg_release,
850 .sendmsg = skcipher_sendmsg_nokey,
851 .sendpage = skcipher_sendpage_nokey,
852 .recvmsg = skcipher_recvmsg_nokey,
853 .poll = skcipher_poll,
854};
855
Herbert Xu8ff59092010-10-19 21:31:55 +0800856static void *skcipher_bind(const char *name, u32 type, u32 mask)
857{
Herbert Xudd504582015-12-25 15:40:05 +0800858 struct skcipher_tfm *tfm;
859 struct crypto_skcipher *skcipher;
860
861 tfm = kzalloc(sizeof(*tfm), GFP_KERNEL);
862 if (!tfm)
863 return ERR_PTR(-ENOMEM);
864
865 skcipher = crypto_alloc_skcipher(name, type, mask);
866 if (IS_ERR(skcipher)) {
867 kfree(tfm);
868 return ERR_CAST(skcipher);
869 }
870
871 tfm->skcipher = skcipher;
872
873 return tfm;
Herbert Xu8ff59092010-10-19 21:31:55 +0800874}
875
876static void skcipher_release(void *private)
877{
Herbert Xudd504582015-12-25 15:40:05 +0800878 struct skcipher_tfm *tfm = private;
879
880 crypto_free_skcipher(tfm->skcipher);
881 kfree(tfm);
Herbert Xu8ff59092010-10-19 21:31:55 +0800882}
883
884static int skcipher_setkey(void *private, const u8 *key, unsigned int keylen)
885{
Herbert Xudd504582015-12-25 15:40:05 +0800886 struct skcipher_tfm *tfm = private;
887 int err;
888
889 err = crypto_skcipher_setkey(tfm->skcipher, key, keylen);
890 tfm->has_key = !err;
891
892 return err;
Herbert Xu8ff59092010-10-19 21:31:55 +0800893}
894
Tadeusz Struka5969992015-03-19 12:31:40 -0700895static void skcipher_wait(struct sock *sk)
896{
897 struct alg_sock *ask = alg_sk(sk);
898 struct skcipher_ctx *ctx = ask->private;
899 int ctr = 0;
900
901 while (atomic_read(&ctx->inflight) && ctr++ < 100)
902 msleep(100);
903}
904
Herbert Xu8ff59092010-10-19 21:31:55 +0800905static void skcipher_sock_destruct(struct sock *sk)
906{
907 struct alg_sock *ask = alg_sk(sk);
908 struct skcipher_ctx *ctx = ask->private;
Herbert Xu0d96e4b2015-12-18 19:16:57 +0800909 struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(&ctx->req);
Herbert Xu8ff59092010-10-19 21:31:55 +0800910
Tadeusz Struka5969992015-03-19 12:31:40 -0700911 if (atomic_read(&ctx->inflight))
912 skcipher_wait(sk);
913
Herbert Xu8ff59092010-10-19 21:31:55 +0800914 skcipher_free_sgl(sk);
Herbert Xu0d96e4b2015-12-18 19:16:57 +0800915 sock_kzfree_s(sk, ctx->iv, crypto_skcipher_ivsize(tfm));
Herbert Xu8ff59092010-10-19 21:31:55 +0800916 sock_kfree_s(sk, ctx, ctx->len);
917 af_alg_release_parent(sk);
918}
919
Herbert Xud7b65ae2016-01-13 15:01:06 +0800920static int skcipher_accept_parent_nokey(void *private, struct sock *sk)
Herbert Xu8ff59092010-10-19 21:31:55 +0800921{
922 struct skcipher_ctx *ctx;
923 struct alg_sock *ask = alg_sk(sk);
Herbert Xudd504582015-12-25 15:40:05 +0800924 struct skcipher_tfm *tfm = private;
925 struct crypto_skcipher *skcipher = tfm->skcipher;
926 unsigned int len = sizeof(*ctx) + crypto_skcipher_reqsize(skcipher);
Herbert Xu8ff59092010-10-19 21:31:55 +0800927
928 ctx = sock_kmalloc(sk, len, GFP_KERNEL);
929 if (!ctx)
930 return -ENOMEM;
931
Herbert Xudd504582015-12-25 15:40:05 +0800932 ctx->iv = sock_kmalloc(sk, crypto_skcipher_ivsize(skcipher),
Herbert Xu8ff59092010-10-19 21:31:55 +0800933 GFP_KERNEL);
934 if (!ctx->iv) {
935 sock_kfree_s(sk, ctx, len);
936 return -ENOMEM;
937 }
938
Herbert Xudd504582015-12-25 15:40:05 +0800939 memset(ctx->iv, 0, crypto_skcipher_ivsize(skcipher));
Herbert Xu8ff59092010-10-19 21:31:55 +0800940
941 INIT_LIST_HEAD(&ctx->tsgl);
942 ctx->len = len;
943 ctx->used = 0;
944 ctx->more = 0;
945 ctx->merge = 0;
946 ctx->enc = 0;
Tadeusz Struka5969992015-03-19 12:31:40 -0700947 atomic_set(&ctx->inflight, 0);
Herbert Xu8ff59092010-10-19 21:31:55 +0800948 af_alg_init_completion(&ctx->completion);
949
950 ask->private = ctx;
951
Herbert Xudd504582015-12-25 15:40:05 +0800952 skcipher_request_set_tfm(&ctx->req, skcipher);
Herbert Xudad41992016-02-03 21:39:27 +0800953 skcipher_request_set_callback(&ctx->req, CRYPTO_TFM_REQ_MAY_SLEEP |
954 CRYPTO_TFM_REQ_MAY_BACKLOG,
Herbert Xu0d96e4b2015-12-18 19:16:57 +0800955 af_alg_complete, &ctx->completion);
Herbert Xu8ff59092010-10-19 21:31:55 +0800956
957 sk->sk_destruct = skcipher_sock_destruct;
958
959 return 0;
960}
961
Herbert Xua0fa2d02016-01-04 13:36:12 +0900962static int skcipher_accept_parent(void *private, struct sock *sk)
963{
964 struct skcipher_tfm *tfm = private;
965
Herbert Xu6e8d8ec2016-01-11 21:29:41 +0800966 if (!tfm->has_key && crypto_skcipher_has_setkey(tfm->skcipher))
Herbert Xua0fa2d02016-01-04 13:36:12 +0900967 return -ENOKEY;
968
Herbert Xud7b65ae2016-01-13 15:01:06 +0800969 return skcipher_accept_parent_nokey(private, sk);
Herbert Xua0fa2d02016-01-04 13:36:12 +0900970}
971
Herbert Xu8ff59092010-10-19 21:31:55 +0800972static const struct af_alg_type algif_type_skcipher = {
973 .bind = skcipher_bind,
974 .release = skcipher_release,
975 .setkey = skcipher_setkey,
976 .accept = skcipher_accept_parent,
Herbert Xua0fa2d02016-01-04 13:36:12 +0900977 .accept_nokey = skcipher_accept_parent_nokey,
Herbert Xu8ff59092010-10-19 21:31:55 +0800978 .ops = &algif_skcipher_ops,
Herbert Xua0fa2d02016-01-04 13:36:12 +0900979 .ops_nokey = &algif_skcipher_ops_nokey,
Herbert Xu8ff59092010-10-19 21:31:55 +0800980 .name = "skcipher",
981 .owner = THIS_MODULE
982};
983
984static int __init algif_skcipher_init(void)
985{
986 return af_alg_register_type(&algif_type_skcipher);
987}
988
989static void __exit algif_skcipher_exit(void)
990{
991 int err = af_alg_unregister_type(&algif_type_skcipher);
992 BUG_ON(err);
993}
994
995module_init(algif_skcipher_init);
996module_exit(algif_skcipher_exit);
997MODULE_LICENSE("GPL");