188 lines
4.0 KiB
C
188 lines
4.0 KiB
C
/* SPDX-License-Identifier: GPL-2.0
|
|
* Copyright(c) 2020 Intel Corporation.
|
|
*/
|
|
|
|
#ifndef XSKXCEIVER_H_
|
|
#define XSKXCEIVER_H_
|
|
|
|
#include "xsk_xdp_progs.skel.h"
|
|
|
|
#ifndef SOL_XDP
|
|
#define SOL_XDP 283
|
|
#endif
|
|
|
|
#ifndef AF_XDP
|
|
#define AF_XDP 44
|
|
#endif
|
|
|
|
#ifndef PF_XDP
|
|
#define PF_XDP AF_XDP
|
|
#endif
|
|
|
|
#ifndef SO_BUSY_POLL_BUDGET
|
|
#define SO_BUSY_POLL_BUDGET 70
|
|
#endif
|
|
|
|
#ifndef SO_PREFER_BUSY_POLL
|
|
#define SO_PREFER_BUSY_POLL 69
|
|
#endif
|
|
|
|
#define TEST_PASS 0
|
|
#define TEST_FAILURE -1
|
|
#define TEST_CONTINUE 1
|
|
#define MAX_INTERFACES 2
|
|
#define MAX_INTERFACE_NAME_CHARS 16
|
|
#define MAX_SOCKETS 2
|
|
#define MAX_TEST_NAME_SIZE 32
|
|
#define MAX_TEARDOWN_ITER 10
|
|
#define PKT_HDR_SIZE (sizeof(struct ethhdr) + sizeof(struct iphdr) + \
|
|
sizeof(struct udphdr))
|
|
#define MIN_ETH_PKT_SIZE 64
|
|
#define ETH_FCS_SIZE 4
|
|
#define MIN_PKT_SIZE (MIN_ETH_PKT_SIZE - ETH_FCS_SIZE)
|
|
#define PKT_SIZE (MIN_PKT_SIZE)
|
|
#define IP_PKT_SIZE (PKT_SIZE - sizeof(struct ethhdr))
|
|
#define IP_PKT_VER 0x4
|
|
#define IP_PKT_TOS 0x9
|
|
#define UDP_PKT_SIZE (IP_PKT_SIZE - sizeof(struct iphdr))
|
|
#define UDP_PKT_DATA_SIZE (UDP_PKT_SIZE - sizeof(struct udphdr))
|
|
#define USLEEP_MAX 10000
|
|
#define SOCK_RECONF_CTR 10
|
|
#define BATCH_SIZE 64
|
|
#define POLL_TMOUT 1000
|
|
#define THREAD_TMOUT 3
|
|
#define DEFAULT_PKT_CNT (4 * 1024)
|
|
#define DEFAULT_UMEM_BUFFERS (DEFAULT_PKT_CNT / 4)
|
|
#define RX_FULL_RXQSIZE 32
|
|
#define UMEM_HEADROOM_TEST_SIZE 128
|
|
#define XSK_UMEM__INVALID_FRAME_SIZE (XSK_UMEM__DEFAULT_FRAME_SIZE + 1)
|
|
|
|
#define print_verbose(x...) do { if (opt_verbose) ksft_print_msg(x); } while (0)
|
|
|
|
enum test_mode {
|
|
TEST_MODE_SKB,
|
|
TEST_MODE_DRV,
|
|
TEST_MODE_ZC,
|
|
TEST_MODE_MAX
|
|
};
|
|
|
|
enum test_type {
|
|
TEST_TYPE_RUN_TO_COMPLETION,
|
|
TEST_TYPE_RUN_TO_COMPLETION_2K_FRAME,
|
|
TEST_TYPE_RUN_TO_COMPLETION_SINGLE_PKT,
|
|
TEST_TYPE_RX_POLL,
|
|
TEST_TYPE_TX_POLL,
|
|
TEST_TYPE_POLL_RXQ_TMOUT,
|
|
TEST_TYPE_POLL_TXQ_TMOUT,
|
|
TEST_TYPE_UNALIGNED,
|
|
TEST_TYPE_ALIGNED_INV_DESC,
|
|
TEST_TYPE_ALIGNED_INV_DESC_2K_FRAME,
|
|
TEST_TYPE_UNALIGNED_INV_DESC,
|
|
TEST_TYPE_HEADROOM,
|
|
TEST_TYPE_TEARDOWN,
|
|
TEST_TYPE_BIDI,
|
|
TEST_TYPE_STATS_RX_DROPPED,
|
|
TEST_TYPE_STATS_TX_INVALID_DESCS,
|
|
TEST_TYPE_STATS_RX_FULL,
|
|
TEST_TYPE_STATS_FILL_EMPTY,
|
|
TEST_TYPE_BPF_RES,
|
|
TEST_TYPE_XDP_DROP_HALF,
|
|
TEST_TYPE_MAX
|
|
};
|
|
|
|
static bool opt_pkt_dump;
|
|
static bool opt_verbose;
|
|
|
|
struct xsk_umem_info {
|
|
struct xsk_ring_prod fq;
|
|
struct xsk_ring_cons cq;
|
|
struct xsk_umem *umem;
|
|
u32 num_frames;
|
|
u32 frame_headroom;
|
|
void *buffer;
|
|
u32 frame_size;
|
|
u32 base_addr;
|
|
bool unaligned_mode;
|
|
};
|
|
|
|
struct xsk_socket_info {
|
|
struct xsk_ring_cons rx;
|
|
struct xsk_ring_prod tx;
|
|
struct xsk_umem_info *umem;
|
|
struct xsk_socket *xsk;
|
|
u32 outstanding_tx;
|
|
u32 rxqsize;
|
|
};
|
|
|
|
struct pkt {
|
|
u64 addr;
|
|
u32 len;
|
|
u32 payload;
|
|
bool valid;
|
|
};
|
|
|
|
struct pkt_stream {
|
|
u32 nb_pkts;
|
|
u32 rx_pkt_nb;
|
|
struct pkt *pkts;
|
|
bool use_addr_for_fill;
|
|
};
|
|
|
|
struct ifobject;
|
|
typedef int (*validation_func_t)(struct ifobject *ifobj);
|
|
typedef void *(*thread_func_t)(void *arg);
|
|
|
|
struct ifobject {
|
|
char ifname[MAX_INTERFACE_NAME_CHARS];
|
|
struct xsk_socket_info *xsk;
|
|
struct xsk_socket_info *xsk_arr;
|
|
struct xsk_umem_info *umem;
|
|
thread_func_t func_ptr;
|
|
validation_func_t validation_func;
|
|
struct pkt_stream *pkt_stream;
|
|
struct xsk_xdp_progs *xdp_progs;
|
|
struct bpf_map *xskmap;
|
|
struct bpf_program *xdp_prog;
|
|
enum test_mode mode;
|
|
int ifindex;
|
|
u32 dst_ip;
|
|
u32 src_ip;
|
|
u32 bind_flags;
|
|
u16 src_port;
|
|
u16 dst_port;
|
|
bool tx_on;
|
|
bool rx_on;
|
|
bool use_poll;
|
|
bool busy_poll;
|
|
bool use_fill_ring;
|
|
bool release_rx;
|
|
bool shared_umem;
|
|
u8 dst_mac[ETH_ALEN];
|
|
u8 src_mac[ETH_ALEN];
|
|
};
|
|
|
|
struct test_spec {
|
|
struct ifobject *ifobj_tx;
|
|
struct ifobject *ifobj_rx;
|
|
struct pkt_stream *tx_pkt_stream_default;
|
|
struct pkt_stream *rx_pkt_stream_default;
|
|
struct bpf_program *xdp_prog_rx;
|
|
struct bpf_program *xdp_prog_tx;
|
|
struct bpf_map *xskmap_rx;
|
|
struct bpf_map *xskmap_tx;
|
|
u16 total_steps;
|
|
u16 current_step;
|
|
u16 nb_sockets;
|
|
bool fail;
|
|
enum test_mode mode;
|
|
char name[MAX_TEST_NAME_SIZE];
|
|
};
|
|
|
|
pthread_barrier_t barr;
|
|
pthread_mutex_t pacing_mutex = PTHREAD_MUTEX_INITIALIZER;
|
|
pthread_cond_t pacing_cond = PTHREAD_COND_INITIALIZER;
|
|
|
|
int pkts_in_flight;
|
|
|
|
#endif /* XSKXCEIVER_H_ */
|