Commit | Line | Data |
---|---|---|
eb42cebb PB |
1 | // SPDX-License-Identifier: GPL-2.0 |
2 | ||
3 | #include <linux/net.h> | |
4 | #include <linux/uio.h> | |
5 | #include <net/sock.h> | |
6 | #include <linux/nospec.h> | |
7 | ||
6a9ce66f PB |
8 | #include "rsrc.h" |
9 | ||
eb4a299b PB |
10 | #define IO_NOTIF_SPLICE_BATCH 32 |
11 | ||
14b146b6 PB |
12 | struct io_notif_data { |
13 | struct file *file; | |
eb42cebb | 14 | struct ubuf_info uarg; |
6a9ce66f | 15 | unsigned long account_pages; |
eb42cebb PB |
16 | }; |
17 | ||
b48c312b PB |
18 | void io_notif_flush(struct io_kiocb *notif); |
19 | struct io_kiocb *io_alloc_notif(struct io_ring_ctx *ctx); | |
eb42cebb | 20 | |
14b146b6 PB |
21 | static inline struct io_notif_data *io_notif_to_data(struct io_kiocb *notif) |
22 | { | |
f2ccb5ae | 23 | return io_kiocb_to_cmd(notif, struct io_notif_data); |
14b146b6 PB |
24 | } |
25 | ||
14b146b6 | 26 | static inline int io_notif_account_mem(struct io_kiocb *notif, unsigned len) |
6a9ce66f PB |
27 | { |
28 | struct io_ring_ctx *ctx = notif->ctx; | |
14b146b6 | 29 | struct io_notif_data *nd = io_notif_to_data(notif); |
6a9ce66f PB |
30 | unsigned nr_pages = (len >> PAGE_SHIFT) + 2; |
31 | int ret; | |
32 | ||
33 | if (ctx->user) { | |
34 | ret = __io_account_mem(ctx->user, nr_pages); | |
35 | if (ret) | |
36 | return ret; | |
14b146b6 | 37 | nd->account_pages += nr_pages; |
6a9ce66f PB |
38 | } |
39 | return 0; | |
40 | } |