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 8 #include "rsrc.h" 9 10 #define IO_NOTIF_SPLICE_BATCH 32 11 12 struct io_notif_data { 13 struct file *file; 14 struct ubuf_info uarg; 15 unsigned long account_pages; 16 }; 17 18 void io_notif_flush(struct io_kiocb *notif); 19 struct io_kiocb *io_alloc_notif(struct io_ring_ctx *ctx); 20 21 static inline struct io_notif_data *io_notif_to_data(struct io_kiocb *notif) 22 { 23 return io_kiocb_to_cmd(notif, struct io_notif_data); 24 } 25 26 static inline int io_notif_account_mem(struct io_kiocb *notif, unsigned len) 27 { 28 struct io_ring_ctx *ctx = notif->ctx; 29 struct io_notif_data *nd = io_notif_to_data(notif); 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; 37 nd->account_pages += nr_pages; 38 } 39 return 0; 40 } 41