[mirror_ubuntu-hirsute-kernel.git] / include / net / xdp_sock.h

/* SPDX-License-Identifier: GPL-2.0 */
/* AF_XDP internal functions
 * Copyright(c) 2018 Intel Corporation.
 */

#ifndef _LINUX_XDP_SOCK_H
#define _LINUX_XDP_SOCK_H

#include <linux/workqueue.h>
#include <linux/if_xdp.h>
#include <linux/mutex.h>
#include <linux/spinlock.h>
#include <linux/mm.h>
#include <net/sock.h>

struct net_device;
struct xsk_queue;
struct xdp_buff;

struct xdp_umem {
	void *addrs;
	u64 size;
	u32 headroom;
	u32 chunk_size;
	u32 chunks;
	u32 npgs;
	struct user_struct *user;
	refcount_t users;
	u8 flags;
	bool zc;
	struct page **pgs;
	int id;
	struct list_head xsk_dma_list;
	struct work_struct work;
};

struct xsk_map {
	struct bpf_map map;
	spinlock_t lock; /* Synchronize map updates */
	struct xdp_sock *xsk_map[];
};

struct xdp_sock {
	/* struct sock must be the first member of struct xdp_sock */
	struct sock sk;
	struct xsk_queue *rx ____cacheline_aligned_in_smp;
	struct net_device *dev;
	struct xdp_umem *umem;
	struct list_head flush_node;
	struct xsk_buff_pool *pool;
	u16 queue_id;
	bool zc;
	enum {
		XSK_READY = 0,
		XSK_BOUND,
		XSK_UNBOUND,
	} state;

	struct xsk_queue *tx ____cacheline_aligned_in_smp;
	struct list_head tx_list;
	/* Protects generic receive. */
	spinlock_t rx_lock;

	/* Statistics */
	u64 rx_dropped;
	u64 rx_queue_full;

	struct list_head map_list;
	/* Protects map_list */
	spinlock_t map_list_lock;
	/* Protects multiple processes in the control path */
	struct mutex mutex;
	struct xsk_queue *fq_tmp; /* Only as tmp storage before bind */
	struct xsk_queue *cq_tmp; /* Only as tmp storage before bind */
};

#ifdef CONFIG_XDP_SOCKETS

int xsk_generic_rcv(struct xdp_sock *xs, struct xdp_buff *xdp);
int __xsk_map_redirect(struct xdp_sock *xs, struct xdp_buff *xdp);
void __xsk_map_flush(void);

static inline struct xdp_sock *__xsk_map_lookup_elem(struct bpf_map *map,
						     u32 key)
{
	struct xsk_map *m = container_of(map, struct xsk_map, map);
	struct xdp_sock *xs;

	if (key >= map->max_entries)
		return NULL;

	xs = READ_ONCE(m->xsk_map[key]);
	return xs;
}

#else

static inline int xsk_generic_rcv(struct xdp_sock *xs, struct xdp_buff *xdp)
{
	return -ENOTSUPP;
}

static inline int __xsk_map_redirect(struct xdp_sock *xs, struct xdp_buff *xdp)
{
	return -EOPNOTSUPP;
}

static inline void __xsk_map_flush(void)
{
}

static inline struct xdp_sock *__xsk_map_lookup_elem(struct bpf_map *map,
						     u32 key)
{
	return NULL;
}

#endif /* CONFIG_XDP_SOCKETS */

#endif /* _LINUX_XDP_SOCK_H */
Commit	Line	Data
dac09149 BT	1	/* SPDX-License-Identifier: GPL-2.0 */
dac09149 BT	2	/* AF_XDP internal functions
c0c77d8f	3	* Copyright(c) 2018 Intel Corporation.
c0c77d8f BT	4	*/
	5
	6	#ifndef _LINUX_XDP_SOCK_H
	7	#define _LINUX_XDP_SOCK_H
	8
e61e62b9 BT	9	#include <linux/workqueue.h>
e61e62b9 BT	10	#include <linux/if_xdp.h>
c0c77d8f	11	#include <linux/mutex.h>
ac98d8aa	12	#include <linux/spinlock.h>
e61e62b9	13	#include <linux/mm.h>
c0c77d8f BT	14	#include <net/sock.h>
c0c77d8f BT	15
b9b6b68e BT	16	struct net_device;
b9b6b68e BT	17	struct xsk_queue;
a71506a4	18	struct xdp_buff;
e61e62b9	19
e61e62b9	20	struct xdp_umem {
7f7ffa4e	21	void *addrs;
93ee30f3	22	u64 size;
e61e62b9	23	u32 headroom;
2b43470a	24	u32 chunk_size;
1c1efc2a	25	u32 chunks;
8ef4e27e	26	u32 npgs;
e61e62b9	27	struct user_struct *user;
e61e62b9	28	refcount_t users;
77cd0d7b	29	u8 flags;
173d3adb	30	bool zc;
8ef4e27e MK	31	struct page **pgs;
8ef4e27e MK	32	int id;
921b6869	33	struct list_head xsk_dma_list;
537cf4e3	34	struct work_struct work;
e61e62b9	35	};
c0c77d8f	36
d817991c BT	37	struct xsk_map {
d817991c BT	38	struct bpf_map map;
d817991c BT	39	spinlock_t lock; /* Synchronize map updates */
	40	struct xdp_sock *xsk_map[];
	41	};
	42
c0c77d8f BT	43	struct xdp_sock {
	44	/* struct sock must be the first member of struct xdp_sock */
	45	struct sock sk;
8ef4e27e	46	struct xsk_queue *rx ____cacheline_aligned_in_smp;
b9b6b68e	47	struct net_device *dev;
c0c77d8f	48	struct xdp_umem *umem;
fbfc504a	49	struct list_head flush_node;
c4655761	50	struct xsk_buff_pool *pool;
965a9909	51	u16 queue_id;
ac98d8aa	52	bool zc;
455302d1 IM	53	enum {
	54	XSK_READY = 0,
	55	XSK_BOUND,
	56	XSK_UNBOUND,
	57	} state;
8ef4e27e	58
fada7fdc	59	struct xsk_queue *tx ____cacheline_aligned_in_smp;
a5aa8e52	60	struct list_head tx_list;
bf0bdd13 IM	61	/* Protects generic receive. */
bf0bdd13 IM	62	spinlock_t rx_lock;
8aa5a335 CL	63
8aa5a335 CL	64	/* Statistics */
c497176c	65	u64 rx_dropped;
8aa5a335 CL	66	u64 rx_queue_full;
8aa5a335 CL	67
0402acd6 BT	68	struct list_head map_list;
	69	/* Protects map_list */
	70	spinlock_t map_list_lock;
8ef4e27e MK	71	/* Protects multiple processes in the control path */
8ef4e27e MK	72	struct mutex mutex;
7361f9c3 MK	73	struct xsk_queue fq_tmp; / Only as tmp storage before bind */
7361f9c3 MK	74	struct xsk_queue cq_tmp; / Only as tmp storage before bind */
c0c77d8f BT	75	};
c0c77d8f BT	76
c497176c	77	#ifdef CONFIG_XDP_SOCKETS
90254034	78
a71506a4	79	int xsk_generic_rcv(struct xdp_sock xs, struct xdp_buff xdp);
e312b9e7 BT	80	int __xsk_map_redirect(struct xdp_sock xs, struct xdp_buff xdp);
e312b9e7 BT	81	void __xsk_map_flush(void);
d817991c BT	82
	83	static inline struct xdp_sock __xsk_map_lookup_elem(struct bpf_map map,
	84	u32 key)
	85	{
	86	struct xsk_map *m = container_of(map, struct xsk_map, map);
	87	struct xdp_sock *xs;
	88
	89	if (key >= map->max_entries)
	90	return NULL;
	91
	92	xs = READ_ONCE(m->xsk_map[key]);
	93	return xs;
	94	}
0402acd6	95
c497176c	96	#else
a71506a4	97
c497176c BT	98	static inline int xsk_generic_rcv(struct xdp_sock xs, struct xdp_buff xdp)
	99	{
	100	return -ENOTSUPP;
	101	}
	102
a71506a4	103	static inline int __xsk_map_redirect(struct xdp_sock xs, struct xdp_buff xdp)
f5bd9138	104	{
a71506a4	105	return -EOPNOTSUPP;
f5bd9138 JK	106	}
f5bd9138 JK	107
a71506a4	108	static inline void __xsk_map_flush(void)
f5bd9138 JK	109	{
	110	}
	111
a71506a4 MK	112	static inline struct xdp_sock __xsk_map_lookup_elem(struct bpf_map map,
a71506a4 MK	113	u32 key)
1661d346 JK	114	{
	115	return NULL;
	116	}
	117
c497176c BT	118	#endif /* CONFIG_XDP_SOCKETS */
c497176c BT	119
c0c77d8f	120	#endif /* _LINUX_XDP_SOCK_H */