| // SPDX-License-Identifier: GPL-2.0 |
| #include <linux/nospec.h> |
| #define IO_NOTIF_SPLICE_BATCH 32 |
| unsigned long account_pages; |
| void io_notif_flush(struct io_kiocb *notif); |
| struct io_kiocb *io_alloc_notif(struct io_ring_ctx *ctx); |
| static inline struct io_notif_data *io_notif_to_data(struct io_kiocb *notif) |
| return io_kiocb_to_cmd(notif, struct io_notif_data); |
| static inline int io_notif_account_mem(struct io_kiocb *notif, unsigned len) |
| struct io_ring_ctx *ctx = notif->ctx; |
| struct io_notif_data *nd = io_notif_to_data(notif); |
| unsigned nr_pages = (len >> PAGE_SHIFT) + 2; |
| ret = __io_account_mem(ctx->user, nr_pages); |
| nd->account_pages += nr_pages; |