mirror of
https://mirrors.bfsu.edu.cn/git/linux.git
synced 2024-12-02 08:34:20 +08:00
5a712e1363
Preserve the offset of the address of the received descriptor, and include
it in the address set for the tx descriptor, so the kernel can correctly
locate the start of the packet data.
Fixes: 03895e63ff
("samples/bpf: add buffer recycling for unaligned chunks to xdpsock")
Signed-off-by: Ciara Loftus <ciara.loftus@intel.com>
Signed-off-by: Daniel Borkmann <daniel@iogearbox.net>
787 lines
18 KiB
C
787 lines
18 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/* Copyright(c) 2017 - 2018 Intel Corporation. */
|
|
|
|
#include <asm/barrier.h>
|
|
#include <errno.h>
|
|
#include <getopt.h>
|
|
#include <libgen.h>
|
|
#include <linux/bpf.h>
|
|
#include <linux/compiler.h>
|
|
#include <linux/if_link.h>
|
|
#include <linux/if_xdp.h>
|
|
#include <linux/if_ether.h>
|
|
#include <locale.h>
|
|
#include <net/ethernet.h>
|
|
#include <net/if.h>
|
|
#include <poll.h>
|
|
#include <pthread.h>
|
|
#include <signal.h>
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/mman.h>
|
|
#include <sys/resource.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
|
|
#include "libbpf.h"
|
|
#include "xsk.h"
|
|
#include <bpf/bpf.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
|
|
|
|
#define NUM_FRAMES (4 * 1024)
|
|
#define BATCH_SIZE 64
|
|
|
|
#define DEBUG_HEXDUMP 0
|
|
#define MAX_SOCKS 8
|
|
|
|
typedef __u64 u64;
|
|
typedef __u32 u32;
|
|
|
|
static unsigned long prev_time;
|
|
|
|
enum benchmark_type {
|
|
BENCH_RXDROP = 0,
|
|
BENCH_TXONLY = 1,
|
|
BENCH_L2FWD = 2,
|
|
};
|
|
|
|
static enum benchmark_type opt_bench = BENCH_RXDROP;
|
|
static u32 opt_xdp_flags = XDP_FLAGS_UPDATE_IF_NOEXIST;
|
|
static const char *opt_if = "";
|
|
static int opt_ifindex;
|
|
static int opt_queue;
|
|
static int opt_poll;
|
|
static int opt_interval = 1;
|
|
static u32 opt_xdp_bind_flags = XDP_USE_NEED_WAKEUP;
|
|
static u32 opt_umem_flags;
|
|
static int opt_unaligned_chunks;
|
|
static int opt_mmap_flags;
|
|
static u32 opt_xdp_bind_flags;
|
|
static int opt_xsk_frame_size = XSK_UMEM__DEFAULT_FRAME_SIZE;
|
|
static int opt_timeout = 1000;
|
|
static bool opt_need_wakeup = true;
|
|
static __u32 prog_id;
|
|
|
|
struct xsk_umem_info {
|
|
struct xsk_ring_prod fq;
|
|
struct xsk_ring_cons cq;
|
|
struct xsk_umem *umem;
|
|
void *buffer;
|
|
};
|
|
|
|
struct xsk_socket_info {
|
|
struct xsk_ring_cons rx;
|
|
struct xsk_ring_prod tx;
|
|
struct xsk_umem_info *umem;
|
|
struct xsk_socket *xsk;
|
|
unsigned long rx_npkts;
|
|
unsigned long tx_npkts;
|
|
unsigned long prev_rx_npkts;
|
|
unsigned long prev_tx_npkts;
|
|
u32 outstanding_tx;
|
|
};
|
|
|
|
static int num_socks;
|
|
struct xsk_socket_info *xsks[MAX_SOCKS];
|
|
|
|
static unsigned long get_nsecs(void)
|
|
{
|
|
struct timespec ts;
|
|
|
|
clock_gettime(CLOCK_MONOTONIC, &ts);
|
|
return ts.tv_sec * 1000000000UL + ts.tv_nsec;
|
|
}
|
|
|
|
static void print_benchmark(bool running)
|
|
{
|
|
const char *bench_str = "INVALID";
|
|
|
|
if (opt_bench == BENCH_RXDROP)
|
|
bench_str = "rxdrop";
|
|
else if (opt_bench == BENCH_TXONLY)
|
|
bench_str = "txonly";
|
|
else if (opt_bench == BENCH_L2FWD)
|
|
bench_str = "l2fwd";
|
|
|
|
printf("%s:%d %s ", opt_if, opt_queue, bench_str);
|
|
if (opt_xdp_flags & XDP_FLAGS_SKB_MODE)
|
|
printf("xdp-skb ");
|
|
else if (opt_xdp_flags & XDP_FLAGS_DRV_MODE)
|
|
printf("xdp-drv ");
|
|
else
|
|
printf(" ");
|
|
|
|
if (opt_poll)
|
|
printf("poll() ");
|
|
|
|
if (running) {
|
|
printf("running...");
|
|
fflush(stdout);
|
|
}
|
|
}
|
|
|
|
static void dump_stats(void)
|
|
{
|
|
unsigned long now = get_nsecs();
|
|
long dt = now - prev_time;
|
|
int i;
|
|
|
|
prev_time = now;
|
|
|
|
for (i = 0; i < num_socks && xsks[i]; i++) {
|
|
char *fmt = "%-15s %'-11.0f %'-11lu\n";
|
|
double rx_pps, tx_pps;
|
|
|
|
rx_pps = (xsks[i]->rx_npkts - xsks[i]->prev_rx_npkts) *
|
|
1000000000. / dt;
|
|
tx_pps = (xsks[i]->tx_npkts - xsks[i]->prev_tx_npkts) *
|
|
1000000000. / dt;
|
|
|
|
printf("\n sock%d@", i);
|
|
print_benchmark(false);
|
|
printf("\n");
|
|
|
|
printf("%-15s %-11s %-11s %-11.2f\n", "", "pps", "pkts",
|
|
dt / 1000000000.);
|
|
printf(fmt, "rx", rx_pps, xsks[i]->rx_npkts);
|
|
printf(fmt, "tx", tx_pps, xsks[i]->tx_npkts);
|
|
|
|
xsks[i]->prev_rx_npkts = xsks[i]->rx_npkts;
|
|
xsks[i]->prev_tx_npkts = xsks[i]->tx_npkts;
|
|
}
|
|
}
|
|
|
|
static void *poller(void *arg)
|
|
{
|
|
(void)arg;
|
|
for (;;) {
|
|
sleep(opt_interval);
|
|
dump_stats();
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static void remove_xdp_program(void)
|
|
{
|
|
__u32 curr_prog_id = 0;
|
|
|
|
if (bpf_get_link_xdp_id(opt_ifindex, &curr_prog_id, opt_xdp_flags)) {
|
|
printf("bpf_get_link_xdp_id failed\n");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
if (prog_id == curr_prog_id)
|
|
bpf_set_link_xdp_fd(opt_ifindex, -1, opt_xdp_flags);
|
|
else if (!curr_prog_id)
|
|
printf("couldn't find a prog id on a given interface\n");
|
|
else
|
|
printf("program on interface changed, not removing\n");
|
|
}
|
|
|
|
static void int_exit(int sig)
|
|
{
|
|
struct xsk_umem *umem = xsks[0]->umem->umem;
|
|
|
|
(void)sig;
|
|
|
|
dump_stats();
|
|
xsk_socket__delete(xsks[0]->xsk);
|
|
(void)xsk_umem__delete(umem);
|
|
remove_xdp_program();
|
|
|
|
exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
static void __exit_with_error(int error, const char *file, const char *func,
|
|
int line)
|
|
{
|
|
fprintf(stderr, "%s:%s:%i: errno: %d/\"%s\"\n", file, func,
|
|
line, error, strerror(error));
|
|
dump_stats();
|
|
remove_xdp_program();
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
#define exit_with_error(error) __exit_with_error(error, __FILE__, __func__, \
|
|
__LINE__)
|
|
|
|
static const char pkt_data[] =
|
|
"\x3c\xfd\xfe\x9e\x7f\x71\xec\xb1\xd7\x98\x3a\xc0\x08\x00\x45\x00"
|
|
"\x00\x2e\x00\x00\x00\x00\x40\x11\x88\x97\x05\x08\x07\x08\xc8\x14"
|
|
"\x1e\x04\x10\x92\x10\x92\x00\x1a\x6d\xa3\x34\x33\x1f\x69\x40\x6b"
|
|
"\x54\x59\xb6\x14\x2d\x11\x44\xbf\xaf\xd9\xbe\xaa";
|
|
|
|
static void swap_mac_addresses(void *data)
|
|
{
|
|
struct ether_header *eth = (struct ether_header *)data;
|
|
struct ether_addr *src_addr = (struct ether_addr *)ð->ether_shost;
|
|
struct ether_addr *dst_addr = (struct ether_addr *)ð->ether_dhost;
|
|
struct ether_addr tmp;
|
|
|
|
tmp = *src_addr;
|
|
*src_addr = *dst_addr;
|
|
*dst_addr = tmp;
|
|
}
|
|
|
|
static void hex_dump(void *pkt, size_t length, u64 addr)
|
|
{
|
|
const unsigned char *address = (unsigned char *)pkt;
|
|
const unsigned char *line = address;
|
|
size_t line_size = 32;
|
|
unsigned char c;
|
|
char buf[32];
|
|
int i = 0;
|
|
|
|
if (!DEBUG_HEXDUMP)
|
|
return;
|
|
|
|
sprintf(buf, "addr=%llu", addr);
|
|
printf("length = %zu\n", length);
|
|
printf("%s | ", buf);
|
|
while (length-- > 0) {
|
|
printf("%02X ", *address++);
|
|
if (!(++i % line_size) || (length == 0 && i % line_size)) {
|
|
if (length == 0) {
|
|
while (i++ % line_size)
|
|
printf("__ ");
|
|
}
|
|
printf(" | "); /* right close */
|
|
while (line < address) {
|
|
c = *line++;
|
|
printf("%c", (c < 33 || c == 255) ? 0x2E : c);
|
|
}
|
|
printf("\n");
|
|
if (length > 0)
|
|
printf("%s | ", buf);
|
|
}
|
|
}
|
|
printf("\n");
|
|
}
|
|
|
|
static size_t gen_eth_frame(struct xsk_umem_info *umem, u64 addr)
|
|
{
|
|
memcpy(xsk_umem__get_data(umem->buffer, addr), pkt_data,
|
|
sizeof(pkt_data) - 1);
|
|
return sizeof(pkt_data) - 1;
|
|
}
|
|
|
|
static struct xsk_umem_info *xsk_configure_umem(void *buffer, u64 size)
|
|
{
|
|
struct xsk_umem_info *umem;
|
|
struct xsk_umem_config cfg = {
|
|
.fill_size = XSK_RING_PROD__DEFAULT_NUM_DESCS,
|
|
.comp_size = XSK_RING_CONS__DEFAULT_NUM_DESCS,
|
|
.frame_size = opt_xsk_frame_size,
|
|
.frame_headroom = XSK_UMEM__DEFAULT_FRAME_HEADROOM,
|
|
.flags = opt_umem_flags
|
|
};
|
|
|
|
int ret;
|
|
|
|
umem = calloc(1, sizeof(*umem));
|
|
if (!umem)
|
|
exit_with_error(errno);
|
|
|
|
ret = xsk_umem__create(&umem->umem, buffer, size, &umem->fq, &umem->cq,
|
|
&cfg);
|
|
|
|
if (ret)
|
|
exit_with_error(-ret);
|
|
|
|
umem->buffer = buffer;
|
|
return umem;
|
|
}
|
|
|
|
static struct xsk_socket_info *xsk_configure_socket(struct xsk_umem_info *umem)
|
|
{
|
|
struct xsk_socket_config cfg;
|
|
struct xsk_socket_info *xsk;
|
|
int ret;
|
|
u32 idx;
|
|
int i;
|
|
|
|
xsk = calloc(1, sizeof(*xsk));
|
|
if (!xsk)
|
|
exit_with_error(errno);
|
|
|
|
xsk->umem = umem;
|
|
cfg.rx_size = XSK_RING_CONS__DEFAULT_NUM_DESCS;
|
|
cfg.tx_size = XSK_RING_PROD__DEFAULT_NUM_DESCS;
|
|
cfg.libbpf_flags = 0;
|
|
cfg.xdp_flags = opt_xdp_flags;
|
|
cfg.bind_flags = opt_xdp_bind_flags;
|
|
ret = xsk_socket__create(&xsk->xsk, opt_if, opt_queue, umem->umem,
|
|
&xsk->rx, &xsk->tx, &cfg);
|
|
if (ret)
|
|
exit_with_error(-ret);
|
|
|
|
ret = bpf_get_link_xdp_id(opt_ifindex, &prog_id, opt_xdp_flags);
|
|
if (ret)
|
|
exit_with_error(-ret);
|
|
|
|
ret = xsk_ring_prod__reserve(&xsk->umem->fq,
|
|
XSK_RING_PROD__DEFAULT_NUM_DESCS,
|
|
&idx);
|
|
if (ret != XSK_RING_PROD__DEFAULT_NUM_DESCS)
|
|
exit_with_error(-ret);
|
|
for (i = 0; i < XSK_RING_PROD__DEFAULT_NUM_DESCS; i++)
|
|
*xsk_ring_prod__fill_addr(&xsk->umem->fq, idx++) =
|
|
i * opt_xsk_frame_size;
|
|
xsk_ring_prod__submit(&xsk->umem->fq,
|
|
XSK_RING_PROD__DEFAULT_NUM_DESCS);
|
|
|
|
return xsk;
|
|
}
|
|
|
|
static struct option long_options[] = {
|
|
{"rxdrop", no_argument, 0, 'r'},
|
|
{"txonly", no_argument, 0, 't'},
|
|
{"l2fwd", no_argument, 0, 'l'},
|
|
{"interface", required_argument, 0, 'i'},
|
|
{"queue", required_argument, 0, 'q'},
|
|
{"poll", no_argument, 0, 'p'},
|
|
{"xdp-skb", no_argument, 0, 'S'},
|
|
{"xdp-native", no_argument, 0, 'N'},
|
|
{"interval", required_argument, 0, 'n'},
|
|
{"zero-copy", no_argument, 0, 'z'},
|
|
{"copy", no_argument, 0, 'c'},
|
|
{"frame-size", required_argument, 0, 'f'},
|
|
{"no-need-wakeup", no_argument, 0, 'm'},
|
|
{"unaligned", no_argument, 0, 'u'},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static void usage(const char *prog)
|
|
{
|
|
const char *str =
|
|
" Usage: %s [OPTIONS]\n"
|
|
" Options:\n"
|
|
" -r, --rxdrop Discard all incoming packets (default)\n"
|
|
" -t, --txonly Only send packets\n"
|
|
" -l, --l2fwd MAC swap L2 forwarding\n"
|
|
" -i, --interface=n Run on interface n\n"
|
|
" -q, --queue=n Use queue n (default 0)\n"
|
|
" -p, --poll Use poll syscall\n"
|
|
" -S, --xdp-skb=n Use XDP skb-mod\n"
|
|
" -N, --xdp-native=n Enfore XDP native mode\n"
|
|
" -n, --interval=n Specify statistics update interval (default 1 sec).\n"
|
|
" -z, --zero-copy Force zero-copy mode.\n"
|
|
" -c, --copy Force copy mode.\n"
|
|
" -f, --frame-size=n Set the frame size (must be a power of two, default is %d).\n"
|
|
" -m, --no-need-wakeup Turn off use of driver need wakeup flag.\n"
|
|
" -f, --frame-size=n Set the frame size (must be a power of two in aligned mode, default is %d).\n"
|
|
" -u, --unaligned Enable unaligned chunk placement\n"
|
|
"\n";
|
|
fprintf(stderr, str, prog, XSK_UMEM__DEFAULT_FRAME_SIZE);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
static void parse_command_line(int argc, char **argv)
|
|
{
|
|
int option_index, c;
|
|
|
|
opterr = 0;
|
|
|
|
for (;;) {
|
|
c = getopt_long(argc, argv, "Frtli:q:psSNn:czf:mu",
|
|
long_options, &option_index);
|
|
if (c == -1)
|
|
break;
|
|
|
|
switch (c) {
|
|
case 'r':
|
|
opt_bench = BENCH_RXDROP;
|
|
break;
|
|
case 't':
|
|
opt_bench = BENCH_TXONLY;
|
|
break;
|
|
case 'l':
|
|
opt_bench = BENCH_L2FWD;
|
|
break;
|
|
case 'i':
|
|
opt_if = optarg;
|
|
break;
|
|
case 'q':
|
|
opt_queue = atoi(optarg);
|
|
break;
|
|
case 'p':
|
|
opt_poll = 1;
|
|
break;
|
|
case 'S':
|
|
opt_xdp_flags |= XDP_FLAGS_SKB_MODE;
|
|
opt_xdp_bind_flags |= XDP_COPY;
|
|
break;
|
|
case 'N':
|
|
opt_xdp_flags |= XDP_FLAGS_DRV_MODE;
|
|
break;
|
|
case 'n':
|
|
opt_interval = atoi(optarg);
|
|
break;
|
|
case 'z':
|
|
opt_xdp_bind_flags |= XDP_ZEROCOPY;
|
|
break;
|
|
case 'c':
|
|
opt_xdp_bind_flags |= XDP_COPY;
|
|
break;
|
|
case 'u':
|
|
opt_umem_flags |= XDP_UMEM_UNALIGNED_CHUNK_FLAG;
|
|
opt_unaligned_chunks = 1;
|
|
opt_mmap_flags = MAP_HUGETLB;
|
|
break;
|
|
case 'F':
|
|
opt_xdp_flags &= ~XDP_FLAGS_UPDATE_IF_NOEXIST;
|
|
break;
|
|
case 'f':
|
|
opt_xsk_frame_size = atoi(optarg);
|
|
case 'm':
|
|
opt_need_wakeup = false;
|
|
opt_xdp_bind_flags &= ~XDP_USE_NEED_WAKEUP;
|
|
break;
|
|
|
|
default:
|
|
usage(basename(argv[0]));
|
|
}
|
|
}
|
|
|
|
opt_ifindex = if_nametoindex(opt_if);
|
|
if (!opt_ifindex) {
|
|
fprintf(stderr, "ERROR: interface \"%s\" does not exist\n",
|
|
opt_if);
|
|
usage(basename(argv[0]));
|
|
}
|
|
|
|
if ((opt_xsk_frame_size & (opt_xsk_frame_size - 1)) &&
|
|
!opt_unaligned_chunks) {
|
|
fprintf(stderr, "--frame-size=%d is not a power of two\n",
|
|
opt_xsk_frame_size);
|
|
usage(basename(argv[0]));
|
|
}
|
|
}
|
|
|
|
static void kick_tx(struct xsk_socket_info *xsk)
|
|
{
|
|
int ret;
|
|
|
|
ret = sendto(xsk_socket__fd(xsk->xsk), NULL, 0, MSG_DONTWAIT, NULL, 0);
|
|
if (ret >= 0 || errno == ENOBUFS || errno == EAGAIN || errno == EBUSY)
|
|
return;
|
|
exit_with_error(errno);
|
|
}
|
|
|
|
static inline void complete_tx_l2fwd(struct xsk_socket_info *xsk,
|
|
struct pollfd *fds)
|
|
{
|
|
struct xsk_umem_info *umem = xsk->umem;
|
|
u32 idx_cq = 0, idx_fq = 0;
|
|
unsigned int rcvd;
|
|
size_t ndescs;
|
|
|
|
if (!xsk->outstanding_tx)
|
|
return;
|
|
|
|
if (!opt_need_wakeup || xsk_ring_prod__needs_wakeup(&xsk->tx))
|
|
kick_tx(xsk);
|
|
|
|
ndescs = (xsk->outstanding_tx > BATCH_SIZE) ? BATCH_SIZE :
|
|
xsk->outstanding_tx;
|
|
|
|
/* re-add completed Tx buffers */
|
|
rcvd = xsk_ring_cons__peek(&umem->cq, ndescs, &idx_cq);
|
|
if (rcvd > 0) {
|
|
unsigned int i;
|
|
int ret;
|
|
|
|
ret = xsk_ring_prod__reserve(&umem->fq, rcvd, &idx_fq);
|
|
while (ret != rcvd) {
|
|
if (ret < 0)
|
|
exit_with_error(-ret);
|
|
if (xsk_ring_prod__needs_wakeup(&umem->fq))
|
|
ret = poll(fds, num_socks, opt_timeout);
|
|
ret = xsk_ring_prod__reserve(&umem->fq, rcvd, &idx_fq);
|
|
}
|
|
|
|
for (i = 0; i < rcvd; i++)
|
|
*xsk_ring_prod__fill_addr(&umem->fq, idx_fq++) =
|
|
*xsk_ring_cons__comp_addr(&umem->cq, idx_cq++);
|
|
|
|
xsk_ring_prod__submit(&xsk->umem->fq, rcvd);
|
|
xsk_ring_cons__release(&xsk->umem->cq, rcvd);
|
|
xsk->outstanding_tx -= rcvd;
|
|
xsk->tx_npkts += rcvd;
|
|
}
|
|
}
|
|
|
|
static inline void complete_tx_only(struct xsk_socket_info *xsk)
|
|
{
|
|
unsigned int rcvd;
|
|
u32 idx;
|
|
|
|
if (!xsk->outstanding_tx)
|
|
return;
|
|
|
|
if (!opt_need_wakeup || xsk_ring_prod__needs_wakeup(&xsk->tx))
|
|
kick_tx(xsk);
|
|
|
|
rcvd = xsk_ring_cons__peek(&xsk->umem->cq, BATCH_SIZE, &idx);
|
|
if (rcvd > 0) {
|
|
xsk_ring_cons__release(&xsk->umem->cq, rcvd);
|
|
xsk->outstanding_tx -= rcvd;
|
|
xsk->tx_npkts += rcvd;
|
|
}
|
|
}
|
|
|
|
static void rx_drop(struct xsk_socket_info *xsk, struct pollfd *fds)
|
|
{
|
|
unsigned int rcvd, i;
|
|
u32 idx_rx = 0, idx_fq = 0;
|
|
int ret;
|
|
|
|
rcvd = xsk_ring_cons__peek(&xsk->rx, BATCH_SIZE, &idx_rx);
|
|
if (!rcvd) {
|
|
if (xsk_ring_prod__needs_wakeup(&xsk->umem->fq))
|
|
ret = poll(fds, num_socks, opt_timeout);
|
|
return;
|
|
}
|
|
|
|
ret = xsk_ring_prod__reserve(&xsk->umem->fq, rcvd, &idx_fq);
|
|
while (ret != rcvd) {
|
|
if (ret < 0)
|
|
exit_with_error(-ret);
|
|
if (xsk_ring_prod__needs_wakeup(&xsk->umem->fq))
|
|
ret = poll(fds, num_socks, opt_timeout);
|
|
ret = xsk_ring_prod__reserve(&xsk->umem->fq, rcvd, &idx_fq);
|
|
}
|
|
|
|
for (i = 0; i < rcvd; i++) {
|
|
u64 addr = xsk_ring_cons__rx_desc(&xsk->rx, idx_rx)->addr;
|
|
u32 len = xsk_ring_cons__rx_desc(&xsk->rx, idx_rx++)->len;
|
|
u64 orig = xsk_umem__extract_addr(addr);
|
|
|
|
addr = xsk_umem__add_offset_to_addr(addr);
|
|
char *pkt = xsk_umem__get_data(xsk->umem->buffer, addr);
|
|
|
|
hex_dump(pkt, len, addr);
|
|
*xsk_ring_prod__fill_addr(&xsk->umem->fq, idx_fq++) = orig;
|
|
}
|
|
|
|
xsk_ring_prod__submit(&xsk->umem->fq, rcvd);
|
|
xsk_ring_cons__release(&xsk->rx, rcvd);
|
|
xsk->rx_npkts += rcvd;
|
|
}
|
|
|
|
static void rx_drop_all(void)
|
|
{
|
|
struct pollfd fds[MAX_SOCKS + 1];
|
|
int i, ret;
|
|
|
|
memset(fds, 0, sizeof(fds));
|
|
|
|
for (i = 0; i < num_socks; i++) {
|
|
fds[i].fd = xsk_socket__fd(xsks[i]->xsk);
|
|
fds[i].events = POLLIN;
|
|
}
|
|
|
|
for (;;) {
|
|
if (opt_poll) {
|
|
ret = poll(fds, num_socks, opt_timeout);
|
|
if (ret <= 0)
|
|
continue;
|
|
}
|
|
|
|
for (i = 0; i < num_socks; i++)
|
|
rx_drop(xsks[i], fds);
|
|
}
|
|
}
|
|
|
|
static void tx_only(struct xsk_socket_info *xsk, u32 frame_nb)
|
|
{
|
|
u32 idx;
|
|
|
|
if (xsk_ring_prod__reserve(&xsk->tx, BATCH_SIZE, &idx) == BATCH_SIZE) {
|
|
unsigned int i;
|
|
|
|
for (i = 0; i < BATCH_SIZE; i++) {
|
|
xsk_ring_prod__tx_desc(&xsk->tx, idx + i)->addr =
|
|
(frame_nb + i) << XSK_UMEM__DEFAULT_FRAME_SHIFT;
|
|
xsk_ring_prod__tx_desc(&xsk->tx, idx + i)->len =
|
|
sizeof(pkt_data) - 1;
|
|
}
|
|
|
|
xsk_ring_prod__submit(&xsk->tx, BATCH_SIZE);
|
|
xsk->outstanding_tx += BATCH_SIZE;
|
|
frame_nb += BATCH_SIZE;
|
|
frame_nb %= NUM_FRAMES;
|
|
}
|
|
|
|
complete_tx_only(xsk);
|
|
}
|
|
|
|
static void tx_only_all(void)
|
|
{
|
|
struct pollfd fds[MAX_SOCKS];
|
|
u32 frame_nb[MAX_SOCKS] = {};
|
|
int i, ret;
|
|
|
|
memset(fds, 0, sizeof(fds));
|
|
for (i = 0; i < num_socks; i++) {
|
|
fds[0].fd = xsk_socket__fd(xsks[i]->xsk);
|
|
fds[0].events = POLLOUT;
|
|
}
|
|
|
|
for (;;) {
|
|
if (opt_poll) {
|
|
ret = poll(fds, num_socks, opt_timeout);
|
|
if (ret <= 0)
|
|
continue;
|
|
|
|
if (!(fds[0].revents & POLLOUT))
|
|
continue;
|
|
}
|
|
|
|
for (i = 0; i < num_socks; i++)
|
|
tx_only(xsks[i], frame_nb[i]);
|
|
}
|
|
}
|
|
|
|
static void l2fwd(struct xsk_socket_info *xsk, struct pollfd *fds)
|
|
{
|
|
unsigned int rcvd, i;
|
|
u32 idx_rx = 0, idx_tx = 0;
|
|
int ret;
|
|
|
|
complete_tx_l2fwd(xsk, fds);
|
|
|
|
rcvd = xsk_ring_cons__peek(&xsk->rx, BATCH_SIZE, &idx_rx);
|
|
if (!rcvd) {
|
|
if (xsk_ring_prod__needs_wakeup(&xsk->umem->fq))
|
|
ret = poll(fds, num_socks, opt_timeout);
|
|
return;
|
|
}
|
|
|
|
ret = xsk_ring_prod__reserve(&xsk->tx, rcvd, &idx_tx);
|
|
while (ret != rcvd) {
|
|
if (ret < 0)
|
|
exit_with_error(-ret);
|
|
if (xsk_ring_prod__needs_wakeup(&xsk->tx))
|
|
kick_tx(xsk);
|
|
ret = xsk_ring_prod__reserve(&xsk->tx, rcvd, &idx_tx);
|
|
}
|
|
|
|
for (i = 0; i < rcvd; i++) {
|
|
u64 addr = xsk_ring_cons__rx_desc(&xsk->rx, idx_rx)->addr;
|
|
u32 len = xsk_ring_cons__rx_desc(&xsk->rx, idx_rx++)->len;
|
|
u64 orig = addr;
|
|
|
|
addr = xsk_umem__add_offset_to_addr(addr);
|
|
char *pkt = xsk_umem__get_data(xsk->umem->buffer, addr);
|
|
|
|
swap_mac_addresses(pkt);
|
|
|
|
hex_dump(pkt, len, addr);
|
|
xsk_ring_prod__tx_desc(&xsk->tx, idx_tx)->addr = orig;
|
|
xsk_ring_prod__tx_desc(&xsk->tx, idx_tx++)->len = len;
|
|
}
|
|
|
|
xsk_ring_prod__submit(&xsk->tx, rcvd);
|
|
xsk_ring_cons__release(&xsk->rx, rcvd);
|
|
|
|
xsk->rx_npkts += rcvd;
|
|
xsk->outstanding_tx += rcvd;
|
|
}
|
|
|
|
static void l2fwd_all(void)
|
|
{
|
|
struct pollfd fds[MAX_SOCKS];
|
|
int i, ret;
|
|
|
|
memset(fds, 0, sizeof(fds));
|
|
|
|
for (i = 0; i < num_socks; i++) {
|
|
fds[i].fd = xsk_socket__fd(xsks[i]->xsk);
|
|
fds[i].events = POLLOUT | POLLIN;
|
|
}
|
|
|
|
for (;;) {
|
|
if (opt_poll) {
|
|
ret = poll(fds, num_socks, opt_timeout);
|
|
if (ret <= 0)
|
|
continue;
|
|
}
|
|
|
|
for (i = 0; i < num_socks; i++)
|
|
l2fwd(xsks[i], fds);
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
struct rlimit r = {RLIM_INFINITY, RLIM_INFINITY};
|
|
struct xsk_umem_info *umem;
|
|
pthread_t pt;
|
|
void *bufs;
|
|
int ret;
|
|
|
|
parse_command_line(argc, argv);
|
|
|
|
if (setrlimit(RLIMIT_MEMLOCK, &r)) {
|
|
fprintf(stderr, "ERROR: setrlimit(RLIMIT_MEMLOCK) \"%s\"\n",
|
|
strerror(errno));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
/* Reserve memory for the umem. Use hugepages if unaligned chunk mode */
|
|
bufs = mmap(NULL, NUM_FRAMES * opt_xsk_frame_size,
|
|
PROT_READ | PROT_WRITE,
|
|
MAP_PRIVATE | MAP_ANONYMOUS | opt_mmap_flags, -1, 0);
|
|
if (bufs == MAP_FAILED) {
|
|
printf("ERROR: mmap failed\n");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
/* Create sockets... */
|
|
umem = xsk_configure_umem(bufs, NUM_FRAMES * opt_xsk_frame_size);
|
|
xsks[num_socks++] = xsk_configure_socket(umem);
|
|
|
|
if (opt_bench == BENCH_TXONLY) {
|
|
int i;
|
|
|
|
for (i = 0; i < NUM_FRAMES; i++)
|
|
(void)gen_eth_frame(umem, i * opt_xsk_frame_size);
|
|
}
|
|
|
|
signal(SIGINT, int_exit);
|
|
signal(SIGTERM, int_exit);
|
|
signal(SIGABRT, int_exit);
|
|
|
|
setlocale(LC_ALL, "");
|
|
|
|
ret = pthread_create(&pt, NULL, poller, NULL);
|
|
if (ret)
|
|
exit_with_error(ret);
|
|
|
|
prev_time = get_nsecs();
|
|
|
|
if (opt_bench == BENCH_RXDROP)
|
|
rx_drop_all();
|
|
else if (opt_bench == BENCH_TXONLY)
|
|
tx_only_all();
|
|
else
|
|
l2fwd_all();
|
|
|
|
return 0;
|
|
}
|