1 // SPDX-License-Identifier: GPL-2.0 2 #ifndef IOU_ZC_RX_H 3 #define IOU_ZC_RX_H 4 5 #include <linux/io_uring_types.h> 6 #include <linux/socket.h> 7 #include <net/page_pool/types.h> 8 #include <net/net_trackers.h> 9 10 struct io_zcrx_area { 11 struct net_iov_area nia; 12 struct io_zcrx_ifq *ifq; 13 atomic_t *user_refs; 14 15 bool is_mapped; 16 u16 area_id; 17 struct page **pages; 18 19 /* freelist */ 20 spinlock_t freelist_lock ____cacheline_aligned_in_smp; 21 u32 free_count; 22 u32 *freelist; 23 }; 24 25 struct io_zcrx_ifq { 26 struct io_ring_ctx *ctx; 27 struct io_zcrx_area *area; 28 29 struct io_uring *rq_ring; 30 struct io_uring_zcrx_rqe *rqes; 31 u32 rq_entries; 32 u32 cached_rq_head; 33 spinlock_t rq_lock; 34 35 u32 if_rxq; 36 struct device *dev; 37 struct net_device *netdev; 38 netdevice_tracker netdev_tracker; 39 spinlock_t lock; 40 }; 41 42 #if defined(CONFIG_IO_URING_ZCRX) 43 int io_register_zcrx_ifq(struct io_ring_ctx *ctx, 44 struct io_uring_zcrx_ifq_reg __user *arg); 45 void io_unregister_zcrx_ifqs(struct io_ring_ctx *ctx); 46 void io_shutdown_zcrx_ifqs(struct io_ring_ctx *ctx); 47 int io_zcrx_recv(struct io_kiocb *req, struct io_zcrx_ifq *ifq, 48 struct socket *sock, unsigned int flags, 49 unsigned issue_flags, unsigned int *len); 50 #else 51 static inline int io_register_zcrx_ifq(struct io_ring_ctx *ctx, 52 struct io_uring_zcrx_ifq_reg __user *arg) 53 { 54 return -EOPNOTSUPP; 55 } 56 static inline void io_unregister_zcrx_ifqs(struct io_ring_ctx *ctx) 57 { 58 } 59 static inline void io_shutdown_zcrx_ifqs(struct io_ring_ctx *ctx) 60 { 61 } 62 static inline int io_zcrx_recv(struct io_kiocb *req, struct io_zcrx_ifq *ifq, 63 struct socket *sock, unsigned int flags, 64 unsigned issue_flags, unsigned int *len) 65 { 66 return -EOPNOTSUPP; 67 } 68 #endif 69 70 int io_recvzc(struct io_kiocb *req, unsigned int issue_flags); 71 int io_recvzc_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe); 72 73 #endif 74