1 /* SPDX-License-Identifier: GPL-2.0 */
2 /* AF_XDP internal functions
3 * Copyright(c) 2018 Intel Corporation.
4 */
5
6 #ifndef _LINUX_XDP_SOCK_H
7 #define _LINUX_XDP_SOCK_H
8
9 #include <linux/workqueue.h>
10 #include <linux/if_xdp.h>
11 #include <linux/mutex.h>
12 #include <linux/spinlock.h>
13 #include <linux/mm.h>
14 #include <net/sock.h>
15
16 struct net_device;
17 struct xsk_queue;
18
19 struct xdp_umem_props {
20 u64 chunk_mask;
21 u64 size;
22 };
23
24 struct xdp_umem_page {
25 void *addr;
26 dma_addr_t dma;
27 };
28
29 struct xdp_umem {
30 struct xsk_queue *fq;
31 struct xsk_queue *cq;
32 struct xdp_umem_page *pages;
33 struct xdp_umem_props props;
34 u32 headroom;
35 u32 chunk_size_nohr;
36 struct user_struct *user;
37 unsigned long address;
38 refcount_t users;
39 struct work_struct work;
40 struct page **pgs;
41 u32 npgs;
42 struct net_device *dev;
43 u16 queue_id;
44 bool zc;
45 spinlock_t xsk_list_lock;
46 struct list_head xsk_list;
47 };
48
49 struct xdp_sock {
50 /* struct sock must be the first member of struct xdp_sock */
51 struct sock sk;
52 struct xsk_queue *rx;
53 struct net_device *dev;
54 struct xdp_umem *umem;
55 struct list_head flush_node;
56 u16 queue_id;
57 struct xsk_queue *tx ____cacheline_aligned_in_smp;
58 struct list_head list;
59 bool zc;
60 /* Protects multiple processes in the control path */
61 struct mutex mutex;
62 /* Mutual exclusion of NAPI TX thread and sendmsg error paths
63 * in the SKB destructor callback.
64 */
65 spinlock_t tx_completion_lock;
66 u64 rx_dropped;
67 };
68
69 struct xdp_buff;
70 #ifdef CONFIG_XDP_SOCKETS
71 int xsk_generic_rcv(struct xdp_sock *xs, struct xdp_buff *xdp);
72 int xsk_rcv(struct xdp_sock *xs, struct xdp_buff *xdp);
73 void xsk_flush(struct xdp_sock *xs);
74 bool xsk_is_setup_for_bpf_map(struct xdp_sock *xs);
75 /* Used from netdev driver */
76 u64 *xsk_umem_peek_addr(struct xdp_umem *umem, u64 *addr);
77 void xsk_umem_discard_addr(struct xdp_umem *umem);
78 void xsk_umem_complete_tx(struct xdp_umem *umem, u32 nb_entries);
79 bool xsk_umem_consume_tx(struct xdp_umem *umem, dma_addr_t *dma, u32 *len);
80 void xsk_umem_consume_tx_done(struct xdp_umem *umem);
81 #else
xsk_generic_rcv(struct xdp_sock * xs,struct xdp_buff * xdp)82 static inline int xsk_generic_rcv(struct xdp_sock *xs, struct xdp_buff *xdp)
83 {
84 return -ENOTSUPP;
85 }
86
xsk_rcv(struct xdp_sock * xs,struct xdp_buff * xdp)87 static inline int xsk_rcv(struct xdp_sock *xs, struct xdp_buff *xdp)
88 {
89 return -ENOTSUPP;
90 }
91
xsk_flush(struct xdp_sock * xs)92 static inline void xsk_flush(struct xdp_sock *xs)
93 {
94 }
95
xsk_is_setup_for_bpf_map(struct xdp_sock * xs)96 static inline bool xsk_is_setup_for_bpf_map(struct xdp_sock *xs)
97 {
98 return false;
99 }
100 #endif /* CONFIG_XDP_SOCKETS */
101
102 #endif /* _LINUX_XDP_SOCK_H */
103