1 /* SPDX-License-Identifier: GPL-2.0
2  * Copyright(c) 2020 Intel Corporation.
3  */
4 
5 #ifndef XSKXCEIVER_H_
6 #define XSKXCEIVER_H_
7 
8 #ifndef SOL_XDP
9 #define SOL_XDP 283
10 #endif
11 
12 #ifndef AF_XDP
13 #define AF_XDP 44
14 #endif
15 
16 #ifndef PF_XDP
17 #define PF_XDP AF_XDP
18 #endif
19 
20 #ifndef SO_BUSY_POLL_BUDGET
21 #define SO_BUSY_POLL_BUDGET 70
22 #endif
23 
24 #ifndef SO_PREFER_BUSY_POLL
25 #define SO_PREFER_BUSY_POLL 69
26 #endif
27 
28 #define TEST_PASS 0
29 #define TEST_FAILURE -1
30 #define TEST_CONTINUE 1
31 #define MAX_INTERFACES 2
32 #define MAX_INTERFACE_NAME_CHARS 7
33 #define MAX_INTERFACES_NAMESPACE_CHARS 10
34 #define MAX_SOCKETS 2
35 #define MAX_TEST_NAME_SIZE 32
36 #define MAX_TEARDOWN_ITER 10
37 #define PKT_HDR_SIZE (sizeof(struct ethhdr) + sizeof(struct iphdr) + \
38 			sizeof(struct udphdr))
39 #define MIN_ETH_PKT_SIZE 64
40 #define ETH_FCS_SIZE 4
41 #define MIN_PKT_SIZE (MIN_ETH_PKT_SIZE - ETH_FCS_SIZE)
42 #define PKT_SIZE (MIN_PKT_SIZE)
43 #define IP_PKT_SIZE (PKT_SIZE - sizeof(struct ethhdr))
44 #define IP_PKT_VER 0x4
45 #define IP_PKT_TOS 0x9
46 #define UDP_PKT_SIZE (IP_PKT_SIZE - sizeof(struct iphdr))
47 #define UDP_PKT_DATA_SIZE (UDP_PKT_SIZE - sizeof(struct udphdr))
48 #define USLEEP_MAX 10000
49 #define SOCK_RECONF_CTR 10
50 #define BATCH_SIZE 64
51 #define POLL_TMOUT 1000
52 #define THREAD_TMOUT 3
53 #define DEFAULT_PKT_CNT (4 * 1024)
54 #define DEFAULT_UMEM_BUFFERS (DEFAULT_PKT_CNT / 4)
55 #define UMEM_SIZE (DEFAULT_UMEM_BUFFERS * XSK_UMEM__DEFAULT_FRAME_SIZE)
56 #define RX_FULL_RXQSIZE 32
57 #define UMEM_HEADROOM_TEST_SIZE 128
58 #define XSK_UMEM__INVALID_FRAME_SIZE (XSK_UMEM__DEFAULT_FRAME_SIZE + 1)
59 
60 #define print_verbose(x...) do { if (opt_verbose) ksft_print_msg(x); } while (0)
61 
62 enum test_mode {
63 	TEST_MODE_SKB,
64 	TEST_MODE_DRV,
65 	TEST_MODE_MAX
66 };
67 
68 enum test_type {
69 	TEST_TYPE_RUN_TO_COMPLETION,
70 	TEST_TYPE_RUN_TO_COMPLETION_2K_FRAME,
71 	TEST_TYPE_RUN_TO_COMPLETION_SINGLE_PKT,
72 	TEST_TYPE_RX_POLL,
73 	TEST_TYPE_TX_POLL,
74 	TEST_TYPE_POLL_RXQ_TMOUT,
75 	TEST_TYPE_POLL_TXQ_TMOUT,
76 	TEST_TYPE_UNALIGNED,
77 	TEST_TYPE_ALIGNED_INV_DESC,
78 	TEST_TYPE_ALIGNED_INV_DESC_2K_FRAME,
79 	TEST_TYPE_UNALIGNED_INV_DESC,
80 	TEST_TYPE_HEADROOM,
81 	TEST_TYPE_TEARDOWN,
82 	TEST_TYPE_BIDI,
83 	TEST_TYPE_STATS_RX_DROPPED,
84 	TEST_TYPE_STATS_TX_INVALID_DESCS,
85 	TEST_TYPE_STATS_RX_FULL,
86 	TEST_TYPE_STATS_FILL_EMPTY,
87 	TEST_TYPE_BPF_RES,
88 	TEST_TYPE_MAX
89 };
90 
91 static bool opt_pkt_dump;
92 static bool opt_verbose;
93 
94 struct xsk_umem_info {
95 	struct xsk_ring_prod fq;
96 	struct xsk_ring_cons cq;
97 	struct xsk_umem *umem;
98 	u32 num_frames;
99 	u32 frame_headroom;
100 	void *buffer;
101 	u32 frame_size;
102 	bool unaligned_mode;
103 };
104 
105 struct xsk_socket_info {
106 	struct xsk_ring_cons rx;
107 	struct xsk_ring_prod tx;
108 	struct xsk_umem_info *umem;
109 	struct xsk_socket *xsk;
110 	u32 outstanding_tx;
111 	u32 rxqsize;
112 };
113 
114 struct pkt {
115 	u64 addr;
116 	u32 len;
117 	u32 payload;
118 	bool valid;
119 };
120 
121 struct pkt_stream {
122 	u32 nb_pkts;
123 	u32 rx_pkt_nb;
124 	struct pkt *pkts;
125 	bool use_addr_for_fill;
126 };
127 
128 struct ifobject;
129 typedef int (*validation_func_t)(struct ifobject *ifobj);
130 typedef void *(*thread_func_t)(void *arg);
131 
132 struct ifobject {
133 	char ifname[MAX_INTERFACE_NAME_CHARS];
134 	char nsname[MAX_INTERFACES_NAMESPACE_CHARS];
135 	struct xsk_socket_info *xsk;
136 	struct xsk_socket_info *xsk_arr;
137 	struct xsk_umem_info *umem;
138 	thread_func_t func_ptr;
139 	validation_func_t validation_func;
140 	struct pkt_stream *pkt_stream;
141 	int ns_fd;
142 	int xsk_map_fd;
143 	u32 dst_ip;
144 	u32 src_ip;
145 	u32 xdp_flags;
146 	u32 bind_flags;
147 	u16 src_port;
148 	u16 dst_port;
149 	bool tx_on;
150 	bool rx_on;
151 	bool use_poll;
152 	bool busy_poll;
153 	bool use_fill_ring;
154 	bool release_rx;
155 	u8 dst_mac[ETH_ALEN];
156 	u8 src_mac[ETH_ALEN];
157 };
158 
159 struct test_spec {
160 	struct ifobject *ifobj_tx;
161 	struct ifobject *ifobj_rx;
162 	struct pkt_stream *pkt_stream_default;
163 	u16 total_steps;
164 	u16 current_step;
165 	u16 nb_sockets;
166 	bool fail;
167 	char name[MAX_TEST_NAME_SIZE];
168 };
169 
170 pthread_barrier_t barr;
171 pthread_mutex_t pacing_mutex = PTHREAD_MUTEX_INITIALIZER;
172 pthread_cond_t pacing_cond = PTHREAD_COND_INITIALIZER;
173 
174 int pkts_in_flight;
175 
176 #endif				/* XSKXCEIVER_H_ */
177