a3c1c95538
The strdup() function returns a pointer to a new string which is a duplicate of the string "ifname". Memory for the new string is obtained with malloc(), and need to be freed with free(). This patch adds this missing "free(saved_hwtstamp_ifname)" in cleanup() to avoid a potential memory leak in xdp_hw_metadata.c. Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> Link: https://lore.kernel.org/r/af9bcccb96655e82de5ce2b4510b88c9c8ed5ed0.1715417367.git.tanggeliang@kylinos.cn Signed-off-by: Alexei Starovoitov <ast@kernel.org>
736 lines
18 KiB
C
736 lines
18 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
|
|
/* Reference program for verifying XDP metadata on real HW. Functional test
|
|
* only, doesn't test the performance.
|
|
*
|
|
* RX:
|
|
* - UDP 9091 packets are diverted into AF_XDP
|
|
* - Metadata verified:
|
|
* - rx_timestamp
|
|
* - rx_hash
|
|
*
|
|
* TX:
|
|
* - UDP 9091 packets trigger TX reply
|
|
* - TX HW timestamp is requested and reported back upon completion
|
|
* - TX checksum is requested
|
|
*/
|
|
|
|
#include <test_progs.h>
|
|
#include <network_helpers.h>
|
|
#include "xdp_hw_metadata.skel.h"
|
|
#include "xsk.h"
|
|
|
|
#include <error.h>
|
|
#include <linux/kernel.h>
|
|
#include <linux/bits.h>
|
|
#include <linux/bitfield.h>
|
|
#include <linux/errqueue.h>
|
|
#include <linux/if_link.h>
|
|
#include <linux/net_tstamp.h>
|
|
#include <linux/udp.h>
|
|
#include <linux/sockios.h>
|
|
#include <linux/if_xdp.h>
|
|
#include <sys/mman.h>
|
|
#include <net/if.h>
|
|
#include <ctype.h>
|
|
#include <poll.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
#include <libgen.h>
|
|
|
|
#include "xdp_metadata.h"
|
|
|
|
#define UMEM_NUM 256
|
|
#define UMEM_FRAME_SIZE XSK_UMEM__DEFAULT_FRAME_SIZE
|
|
#define UMEM_SIZE (UMEM_FRAME_SIZE * UMEM_NUM)
|
|
#define XDP_FLAGS (XDP_FLAGS_DRV_MODE | XDP_FLAGS_REPLACE)
|
|
|
|
struct xsk {
|
|
void *umem_area;
|
|
struct xsk_umem *umem;
|
|
struct xsk_ring_prod fill;
|
|
struct xsk_ring_cons comp;
|
|
struct xsk_ring_prod tx;
|
|
struct xsk_ring_cons rx;
|
|
struct xsk_socket *socket;
|
|
};
|
|
|
|
struct xdp_hw_metadata *bpf_obj;
|
|
__u16 bind_flags = XDP_USE_NEED_WAKEUP | XDP_ZEROCOPY;
|
|
struct xsk *rx_xsk;
|
|
const char *ifname;
|
|
int ifindex;
|
|
int rxq;
|
|
bool skip_tx;
|
|
__u64 last_hw_rx_timestamp;
|
|
__u64 last_xdp_rx_timestamp;
|
|
|
|
void test__fail(void) { /* for network_helpers.c */ }
|
|
|
|
static int open_xsk(int ifindex, struct xsk *xsk, __u32 queue_id)
|
|
{
|
|
int mmap_flags = MAP_PRIVATE | MAP_ANONYMOUS | MAP_NORESERVE;
|
|
const struct xsk_socket_config socket_config = {
|
|
.rx_size = XSK_RING_PROD__DEFAULT_NUM_DESCS,
|
|
.tx_size = XSK_RING_PROD__DEFAULT_NUM_DESCS,
|
|
.bind_flags = bind_flags,
|
|
};
|
|
const struct xsk_umem_config umem_config = {
|
|
.fill_size = XSK_RING_PROD__DEFAULT_NUM_DESCS,
|
|
.comp_size = XSK_RING_CONS__DEFAULT_NUM_DESCS,
|
|
.frame_size = XSK_UMEM__DEFAULT_FRAME_SIZE,
|
|
.flags = XSK_UMEM__DEFAULT_FLAGS,
|
|
.tx_metadata_len = sizeof(struct xsk_tx_metadata),
|
|
};
|
|
__u32 idx = 0;
|
|
u64 addr;
|
|
int ret;
|
|
int i;
|
|
|
|
xsk->umem_area = mmap(NULL, UMEM_SIZE, PROT_READ | PROT_WRITE, mmap_flags, -1, 0);
|
|
if (xsk->umem_area == MAP_FAILED)
|
|
return -ENOMEM;
|
|
|
|
ret = xsk_umem__create(&xsk->umem,
|
|
xsk->umem_area, UMEM_SIZE,
|
|
&xsk->fill,
|
|
&xsk->comp,
|
|
&umem_config);
|
|
if (ret)
|
|
return ret;
|
|
|
|
ret = xsk_socket__create(&xsk->socket, ifindex, queue_id,
|
|
xsk->umem,
|
|
&xsk->rx,
|
|
&xsk->tx,
|
|
&socket_config);
|
|
if (ret)
|
|
return ret;
|
|
|
|
/* First half of umem is for TX. This way address matches 1-to-1
|
|
* to the completion queue index.
|
|
*/
|
|
|
|
for (i = 0; i < UMEM_NUM / 2; i++) {
|
|
addr = i * UMEM_FRAME_SIZE;
|
|
printf("%p: tx_desc[%d] -> %lx\n", xsk, i, addr);
|
|
}
|
|
|
|
/* Second half of umem is for RX. */
|
|
|
|
ret = xsk_ring_prod__reserve(&xsk->fill, UMEM_NUM / 2, &idx);
|
|
for (i = 0; i < UMEM_NUM / 2; i++) {
|
|
addr = (UMEM_NUM / 2 + i) * UMEM_FRAME_SIZE;
|
|
printf("%p: rx_desc[%d] -> %lx\n", xsk, i, addr);
|
|
*xsk_ring_prod__fill_addr(&xsk->fill, idx + i) = addr;
|
|
}
|
|
xsk_ring_prod__submit(&xsk->fill, ret);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void close_xsk(struct xsk *xsk)
|
|
{
|
|
if (xsk->umem)
|
|
xsk_umem__delete(xsk->umem);
|
|
if (xsk->socket)
|
|
xsk_socket__delete(xsk->socket);
|
|
munmap(xsk->umem_area, UMEM_SIZE);
|
|
}
|
|
|
|
static void refill_rx(struct xsk *xsk, __u64 addr)
|
|
{
|
|
__u32 idx;
|
|
|
|
if (xsk_ring_prod__reserve(&xsk->fill, 1, &idx) == 1) {
|
|
printf("%p: complete rx idx=%u addr=%llx\n", xsk, idx, addr);
|
|
*xsk_ring_prod__fill_addr(&xsk->fill, idx) = addr;
|
|
xsk_ring_prod__submit(&xsk->fill, 1);
|
|
}
|
|
}
|
|
|
|
static int kick_tx(struct xsk *xsk)
|
|
{
|
|
return sendto(xsk_socket__fd(xsk->socket), NULL, 0, MSG_DONTWAIT, NULL, 0);
|
|
}
|
|
|
|
static int kick_rx(struct xsk *xsk)
|
|
{
|
|
return recvfrom(xsk_socket__fd(xsk->socket), NULL, 0, MSG_DONTWAIT, NULL, NULL);
|
|
}
|
|
|
|
#define NANOSEC_PER_SEC 1000000000 /* 10^9 */
|
|
static __u64 gettime(clockid_t clock_id)
|
|
{
|
|
struct timespec t;
|
|
int res;
|
|
|
|
/* See man clock_gettime(2) for type of clock_id's */
|
|
res = clock_gettime(clock_id, &t);
|
|
|
|
if (res < 0)
|
|
error(res, errno, "Error with clock_gettime()");
|
|
|
|
return (__u64) t.tv_sec * NANOSEC_PER_SEC + t.tv_nsec;
|
|
}
|
|
|
|
static void print_tstamp_delta(const char *name, const char *refname,
|
|
__u64 tstamp, __u64 reference)
|
|
{
|
|
__s64 delta = (__s64)reference - (__s64)tstamp;
|
|
|
|
printf("%s: %llu (sec:%0.4f) delta to %s sec:%0.4f (%0.3f usec)\n",
|
|
name, tstamp, (double)tstamp / NANOSEC_PER_SEC, refname,
|
|
(double)delta / NANOSEC_PER_SEC,
|
|
(double)delta / 1000);
|
|
}
|
|
|
|
#define VLAN_PRIO_MASK GENMASK(15, 13) /* Priority Code Point */
|
|
#define VLAN_DEI_MASK GENMASK(12, 12) /* Drop Eligible Indicator */
|
|
#define VLAN_VID_MASK GENMASK(11, 0) /* VLAN Identifier */
|
|
static void print_vlan_tci(__u16 tag)
|
|
{
|
|
__u16 vlan_id = FIELD_GET(VLAN_VID_MASK, tag);
|
|
__u8 pcp = FIELD_GET(VLAN_PRIO_MASK, tag);
|
|
bool dei = FIELD_GET(VLAN_DEI_MASK, tag);
|
|
|
|
printf("PCP=%u, DEI=%d, VID=0x%X\n", pcp, dei, vlan_id);
|
|
}
|
|
|
|
static void verify_xdp_metadata(void *data, clockid_t clock_id)
|
|
{
|
|
struct xdp_meta *meta;
|
|
|
|
meta = data - sizeof(*meta);
|
|
|
|
if (meta->hint_valid & XDP_META_FIELD_RSS)
|
|
printf("rx_hash: 0x%X with RSS type:0x%X\n",
|
|
meta->rx_hash, meta->rx_hash_type);
|
|
else
|
|
printf("No rx_hash, err=%d\n", meta->rx_hash_err);
|
|
|
|
if (meta->hint_valid & XDP_META_FIELD_TS) {
|
|
__u64 ref_tstamp = gettime(clock_id);
|
|
|
|
/* store received timestamps to calculate a delta at tx */
|
|
last_hw_rx_timestamp = meta->rx_timestamp;
|
|
last_xdp_rx_timestamp = meta->xdp_timestamp;
|
|
|
|
print_tstamp_delta("HW RX-time", "User RX-time",
|
|
meta->rx_timestamp, ref_tstamp);
|
|
print_tstamp_delta("XDP RX-time", "User RX-time",
|
|
meta->xdp_timestamp, ref_tstamp);
|
|
} else {
|
|
printf("No rx_timestamp, err=%d\n", meta->rx_timestamp_err);
|
|
}
|
|
|
|
if (meta->hint_valid & XDP_META_FIELD_VLAN_TAG) {
|
|
printf("rx_vlan_proto: 0x%X\n", ntohs(meta->rx_vlan_proto));
|
|
printf("rx_vlan_tci: ");
|
|
print_vlan_tci(meta->rx_vlan_tci);
|
|
} else {
|
|
printf("No rx_vlan_tci or rx_vlan_proto, err=%d\n",
|
|
meta->rx_vlan_tag_err);
|
|
}
|
|
}
|
|
|
|
static void verify_skb_metadata(int fd)
|
|
{
|
|
char cmsg_buf[1024];
|
|
char packet_buf[128];
|
|
|
|
struct scm_timestamping *ts;
|
|
struct iovec packet_iov;
|
|
struct cmsghdr *cmsg;
|
|
struct msghdr hdr;
|
|
|
|
memset(&hdr, 0, sizeof(hdr));
|
|
hdr.msg_iov = &packet_iov;
|
|
hdr.msg_iovlen = 1;
|
|
packet_iov.iov_base = packet_buf;
|
|
packet_iov.iov_len = sizeof(packet_buf);
|
|
|
|
hdr.msg_control = cmsg_buf;
|
|
hdr.msg_controllen = sizeof(cmsg_buf);
|
|
|
|
if (recvmsg(fd, &hdr, 0) < 0)
|
|
error(1, errno, "recvmsg");
|
|
|
|
for (cmsg = CMSG_FIRSTHDR(&hdr); cmsg != NULL;
|
|
cmsg = CMSG_NXTHDR(&hdr, cmsg)) {
|
|
|
|
if (cmsg->cmsg_level != SOL_SOCKET)
|
|
continue;
|
|
|
|
switch (cmsg->cmsg_type) {
|
|
case SCM_TIMESTAMPING:
|
|
ts = (struct scm_timestamping *)CMSG_DATA(cmsg);
|
|
if (ts->ts[2].tv_sec || ts->ts[2].tv_nsec) {
|
|
printf("found skb hwtstamp = %lu.%lu\n",
|
|
ts->ts[2].tv_sec, ts->ts[2].tv_nsec);
|
|
return;
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
printf("skb hwtstamp is not found!\n");
|
|
}
|
|
|
|
static bool complete_tx(struct xsk *xsk, clockid_t clock_id)
|
|
{
|
|
struct xsk_tx_metadata *meta;
|
|
__u64 addr;
|
|
void *data;
|
|
__u32 idx;
|
|
|
|
if (!xsk_ring_cons__peek(&xsk->comp, 1, &idx))
|
|
return false;
|
|
|
|
addr = *xsk_ring_cons__comp_addr(&xsk->comp, idx);
|
|
data = xsk_umem__get_data(xsk->umem_area, addr);
|
|
meta = data - sizeof(struct xsk_tx_metadata);
|
|
|
|
printf("%p: complete tx idx=%u addr=%llx\n", xsk, idx, addr);
|
|
|
|
if (meta->completion.tx_timestamp) {
|
|
__u64 ref_tstamp = gettime(clock_id);
|
|
|
|
print_tstamp_delta("HW TX-complete-time", "User TX-complete-time",
|
|
meta->completion.tx_timestamp, ref_tstamp);
|
|
print_tstamp_delta("XDP RX-time", "User TX-complete-time",
|
|
last_xdp_rx_timestamp, ref_tstamp);
|
|
print_tstamp_delta("HW RX-time", "HW TX-complete-time",
|
|
last_hw_rx_timestamp, meta->completion.tx_timestamp);
|
|
} else {
|
|
printf("No tx_timestamp\n");
|
|
}
|
|
|
|
xsk_ring_cons__release(&xsk->comp, 1);
|
|
|
|
return true;
|
|
}
|
|
|
|
#define swap(a, b, len) do { \
|
|
for (int i = 0; i < len; i++) { \
|
|
__u8 tmp = ((__u8 *)a)[i]; \
|
|
((__u8 *)a)[i] = ((__u8 *)b)[i]; \
|
|
((__u8 *)b)[i] = tmp; \
|
|
} \
|
|
} while (0)
|
|
|
|
static void ping_pong(struct xsk *xsk, void *rx_packet, clockid_t clock_id)
|
|
{
|
|
struct xsk_tx_metadata *meta;
|
|
struct ipv6hdr *ip6h = NULL;
|
|
struct iphdr *iph = NULL;
|
|
struct xdp_desc *tx_desc;
|
|
struct udphdr *udph;
|
|
struct ethhdr *eth;
|
|
__sum16 want_csum;
|
|
void *data;
|
|
__u32 idx;
|
|
int ret;
|
|
int len;
|
|
|
|
ret = xsk_ring_prod__reserve(&xsk->tx, 1, &idx);
|
|
if (ret != 1) {
|
|
printf("%p: failed to reserve tx slot\n", xsk);
|
|
return;
|
|
}
|
|
|
|
tx_desc = xsk_ring_prod__tx_desc(&xsk->tx, idx);
|
|
tx_desc->addr = idx % (UMEM_NUM / 2) * UMEM_FRAME_SIZE + sizeof(struct xsk_tx_metadata);
|
|
data = xsk_umem__get_data(xsk->umem_area, tx_desc->addr);
|
|
|
|
meta = data - sizeof(struct xsk_tx_metadata);
|
|
memset(meta, 0, sizeof(*meta));
|
|
meta->flags = XDP_TXMD_FLAGS_TIMESTAMP;
|
|
|
|
eth = rx_packet;
|
|
|
|
if (eth->h_proto == htons(ETH_P_IP)) {
|
|
iph = (void *)(eth + 1);
|
|
udph = (void *)(iph + 1);
|
|
} else if (eth->h_proto == htons(ETH_P_IPV6)) {
|
|
ip6h = (void *)(eth + 1);
|
|
udph = (void *)(ip6h + 1);
|
|
} else {
|
|
printf("%p: failed to detect IP version for ping pong %04x\n", xsk, eth->h_proto);
|
|
xsk_ring_prod__cancel(&xsk->tx, 1);
|
|
return;
|
|
}
|
|
|
|
len = ETH_HLEN;
|
|
if (ip6h)
|
|
len += sizeof(*ip6h) + ntohs(ip6h->payload_len);
|
|
if (iph)
|
|
len += ntohs(iph->tot_len);
|
|
|
|
swap(eth->h_dest, eth->h_source, ETH_ALEN);
|
|
if (iph)
|
|
swap(&iph->saddr, &iph->daddr, 4);
|
|
else
|
|
swap(&ip6h->saddr, &ip6h->daddr, 16);
|
|
swap(&udph->source, &udph->dest, 2);
|
|
|
|
want_csum = udph->check;
|
|
if (ip6h)
|
|
udph->check = ~csum_ipv6_magic(&ip6h->saddr, &ip6h->daddr,
|
|
ntohs(udph->len), IPPROTO_UDP, 0);
|
|
else
|
|
udph->check = ~csum_tcpudp_magic(iph->saddr, iph->daddr,
|
|
ntohs(udph->len), IPPROTO_UDP, 0);
|
|
|
|
meta->flags |= XDP_TXMD_FLAGS_CHECKSUM;
|
|
if (iph)
|
|
meta->request.csum_start = sizeof(*eth) + sizeof(*iph);
|
|
else
|
|
meta->request.csum_start = sizeof(*eth) + sizeof(*ip6h);
|
|
meta->request.csum_offset = offsetof(struct udphdr, check);
|
|
|
|
printf("%p: ping-pong with csum=%04x (want %04x) csum_start=%d csum_offset=%d\n",
|
|
xsk, ntohs(udph->check), ntohs(want_csum),
|
|
meta->request.csum_start, meta->request.csum_offset);
|
|
|
|
memcpy(data, rx_packet, len); /* don't share umem chunk for simplicity */
|
|
tx_desc->options |= XDP_TX_METADATA;
|
|
tx_desc->len = len;
|
|
|
|
xsk_ring_prod__submit(&xsk->tx, 1);
|
|
}
|
|
|
|
static int verify_metadata(struct xsk *rx_xsk, int rxq, int server_fd, clockid_t clock_id)
|
|
{
|
|
const struct xdp_desc *rx_desc;
|
|
struct pollfd fds[rxq + 1];
|
|
__u64 comp_addr;
|
|
__u64 addr;
|
|
__u32 idx = 0;
|
|
int ret;
|
|
int i;
|
|
|
|
for (i = 0; i < rxq; i++) {
|
|
fds[i].fd = xsk_socket__fd(rx_xsk[i].socket);
|
|
fds[i].events = POLLIN;
|
|
fds[i].revents = 0;
|
|
}
|
|
|
|
fds[rxq].fd = server_fd;
|
|
fds[rxq].events = POLLIN;
|
|
fds[rxq].revents = 0;
|
|
|
|
while (true) {
|
|
errno = 0;
|
|
|
|
for (i = 0; i < rxq; i++) {
|
|
ret = kick_rx(&rx_xsk[i]);
|
|
if (ret)
|
|
printf("kick_rx ret=%d\n", ret);
|
|
}
|
|
|
|
ret = poll(fds, rxq + 1, 1000);
|
|
printf("poll: %d (%d) skip=%llu fail=%llu redir=%llu\n",
|
|
ret, errno, bpf_obj->bss->pkts_skip,
|
|
bpf_obj->bss->pkts_fail, bpf_obj->bss->pkts_redir);
|
|
if (ret < 0)
|
|
break;
|
|
if (ret == 0)
|
|
continue;
|
|
|
|
if (fds[rxq].revents)
|
|
verify_skb_metadata(server_fd);
|
|
|
|
for (i = 0; i < rxq; i++) {
|
|
bool first_seg = true;
|
|
bool is_eop = true;
|
|
|
|
if (fds[i].revents == 0)
|
|
continue;
|
|
|
|
struct xsk *xsk = &rx_xsk[i];
|
|
peek:
|
|
ret = xsk_ring_cons__peek(&xsk->rx, 1, &idx);
|
|
printf("xsk_ring_cons__peek: %d\n", ret);
|
|
if (ret != 1)
|
|
continue;
|
|
|
|
rx_desc = xsk_ring_cons__rx_desc(&xsk->rx, idx);
|
|
comp_addr = xsk_umem__extract_addr(rx_desc->addr);
|
|
addr = xsk_umem__add_offset_to_addr(rx_desc->addr);
|
|
is_eop = !(rx_desc->options & XDP_PKT_CONTD);
|
|
printf("%p: rx_desc[%u]->addr=%llx addr=%llx comp_addr=%llx%s\n",
|
|
xsk, idx, rx_desc->addr, addr, comp_addr, is_eop ? " EoP" : "");
|
|
if (first_seg) {
|
|
verify_xdp_metadata(xsk_umem__get_data(xsk->umem_area, addr),
|
|
clock_id);
|
|
first_seg = false;
|
|
|
|
if (!skip_tx) {
|
|
/* mirror first chunk back */
|
|
ping_pong(xsk, xsk_umem__get_data(xsk->umem_area, addr),
|
|
clock_id);
|
|
|
|
ret = kick_tx(xsk);
|
|
if (ret)
|
|
printf("kick_tx ret=%d\n", ret);
|
|
|
|
for (int j = 0; j < 500; j++) {
|
|
if (complete_tx(xsk, clock_id))
|
|
break;
|
|
usleep(10);
|
|
}
|
|
}
|
|
}
|
|
|
|
xsk_ring_cons__release(&xsk->rx, 1);
|
|
refill_rx(xsk, comp_addr);
|
|
if (!is_eop)
|
|
goto peek;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int rxq_num(const char *ifname)
|
|
{
|
|
struct ethtool_channels ch = {
|
|
.cmd = ETHTOOL_GCHANNELS,
|
|
};
|
|
|
|
struct ifreq ifr = {
|
|
.ifr_data = (void *)&ch,
|
|
};
|
|
strncpy(ifr.ifr_name, ifname, IF_NAMESIZE - 1);
|
|
int fd, ret;
|
|
|
|
fd = socket(AF_UNIX, SOCK_DGRAM, 0);
|
|
if (fd < 0)
|
|
error(1, errno, "socket");
|
|
|
|
ret = ioctl(fd, SIOCETHTOOL, &ifr);
|
|
if (ret < 0)
|
|
error(1, errno, "ioctl(SIOCETHTOOL)");
|
|
|
|
close(fd);
|
|
|
|
return ch.rx_count + ch.combined_count;
|
|
}
|
|
|
|
static void hwtstamp_ioctl(int op, const char *ifname, struct hwtstamp_config *cfg)
|
|
{
|
|
struct ifreq ifr = {
|
|
.ifr_data = (void *)cfg,
|
|
};
|
|
strncpy(ifr.ifr_name, ifname, IF_NAMESIZE - 1);
|
|
int fd, ret;
|
|
|
|
fd = socket(AF_UNIX, SOCK_DGRAM, 0);
|
|
if (fd < 0)
|
|
error(1, errno, "socket");
|
|
|
|
ret = ioctl(fd, op, &ifr);
|
|
if (ret < 0)
|
|
error(1, errno, "ioctl(%d)", op);
|
|
|
|
close(fd);
|
|
}
|
|
|
|
static struct hwtstamp_config saved_hwtstamp_cfg;
|
|
static const char *saved_hwtstamp_ifname;
|
|
|
|
static void hwtstamp_restore(void)
|
|
{
|
|
hwtstamp_ioctl(SIOCSHWTSTAMP, saved_hwtstamp_ifname, &saved_hwtstamp_cfg);
|
|
}
|
|
|
|
static void hwtstamp_enable(const char *ifname)
|
|
{
|
|
struct hwtstamp_config cfg = {
|
|
.rx_filter = HWTSTAMP_FILTER_ALL,
|
|
};
|
|
|
|
hwtstamp_ioctl(SIOCGHWTSTAMP, ifname, &saved_hwtstamp_cfg);
|
|
saved_hwtstamp_ifname = strdup(ifname);
|
|
atexit(hwtstamp_restore);
|
|
|
|
hwtstamp_ioctl(SIOCSHWTSTAMP, ifname, &cfg);
|
|
}
|
|
|
|
static void cleanup(void)
|
|
{
|
|
LIBBPF_OPTS(bpf_xdp_attach_opts, opts);
|
|
int ret;
|
|
int i;
|
|
|
|
if (bpf_obj) {
|
|
opts.old_prog_fd = bpf_program__fd(bpf_obj->progs.rx);
|
|
if (opts.old_prog_fd >= 0) {
|
|
printf("detaching bpf program....\n");
|
|
ret = bpf_xdp_detach(ifindex, XDP_FLAGS, &opts);
|
|
if (ret)
|
|
printf("failed to detach XDP program: %d\n", ret);
|
|
}
|
|
}
|
|
|
|
for (i = 0; i < rxq; i++)
|
|
close_xsk(&rx_xsk[i]);
|
|
|
|
if (bpf_obj)
|
|
xdp_hw_metadata__destroy(bpf_obj);
|
|
|
|
free((void *)saved_hwtstamp_ifname);
|
|
}
|
|
|
|
static void handle_signal(int sig)
|
|
{
|
|
/* interrupting poll() is all we need */
|
|
}
|
|
|
|
static void timestamping_enable(int fd, int val)
|
|
{
|
|
int ret;
|
|
|
|
ret = setsockopt(fd, SOL_SOCKET, SO_TIMESTAMPING, &val, sizeof(val));
|
|
if (ret < 0)
|
|
error(1, errno, "setsockopt(SO_TIMESTAMPING)");
|
|
}
|
|
|
|
static void print_usage(void)
|
|
{
|
|
const char *usage =
|
|
"Usage: xdp_hw_metadata [OPTIONS] [IFNAME]\n"
|
|
" -c Run in copy mode (zerocopy is default)\n"
|
|
" -h Display this help and exit\n\n"
|
|
" -m Enable multi-buffer XDP for larger MTU\n"
|
|
" -r Don't generate AF_XDP reply (rx metadata only)\n"
|
|
"Generate test packets on the other machine with:\n"
|
|
" echo -n xdp | nc -u -q1 <dst_ip> 9091\n";
|
|
|
|
printf("%s", usage);
|
|
}
|
|
|
|
static void read_args(int argc, char *argv[])
|
|
{
|
|
int opt;
|
|
|
|
while ((opt = getopt(argc, argv, "chmr")) != -1) {
|
|
switch (opt) {
|
|
case 'c':
|
|
bind_flags &= ~XDP_USE_NEED_WAKEUP;
|
|
bind_flags &= ~XDP_ZEROCOPY;
|
|
bind_flags |= XDP_COPY;
|
|
break;
|
|
case 'h':
|
|
print_usage();
|
|
exit(0);
|
|
case 'm':
|
|
bind_flags |= XDP_USE_SG;
|
|
break;
|
|
case 'r':
|
|
skip_tx = true;
|
|
break;
|
|
case '?':
|
|
if (isprint(optopt))
|
|
fprintf(stderr, "Unknown option: -%c\n", optopt);
|
|
fallthrough;
|
|
default:
|
|
print_usage();
|
|
error(-1, opterr, "Command line options error");
|
|
}
|
|
}
|
|
|
|
if (optind >= argc) {
|
|
fprintf(stderr, "No device name provided\n");
|
|
print_usage();
|
|
exit(-1);
|
|
}
|
|
|
|
ifname = argv[optind];
|
|
ifindex = if_nametoindex(ifname);
|
|
|
|
if (!ifname)
|
|
error(-1, errno, "Invalid interface name");
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
clockid_t clock_id = CLOCK_TAI;
|
|
int server_fd = -1;
|
|
int ret;
|
|
int i;
|
|
|
|
struct bpf_program *prog;
|
|
|
|
read_args(argc, argv);
|
|
|
|
rxq = rxq_num(ifname);
|
|
|
|
printf("rxq: %d\n", rxq);
|
|
|
|
hwtstamp_enable(ifname);
|
|
|
|
rx_xsk = malloc(sizeof(struct xsk) * rxq);
|
|
if (!rx_xsk)
|
|
error(1, ENOMEM, "malloc");
|
|
|
|
for (i = 0; i < rxq; i++) {
|
|
printf("open_xsk(%s, %p, %d)\n", ifname, &rx_xsk[i], i);
|
|
ret = open_xsk(ifindex, &rx_xsk[i], i);
|
|
if (ret)
|
|
error(1, -ret, "open_xsk");
|
|
|
|
printf("xsk_socket__fd() -> %d\n", xsk_socket__fd(rx_xsk[i].socket));
|
|
}
|
|
|
|
printf("open bpf program...\n");
|
|
bpf_obj = xdp_hw_metadata__open();
|
|
if (libbpf_get_error(bpf_obj))
|
|
error(1, libbpf_get_error(bpf_obj), "xdp_hw_metadata__open");
|
|
|
|
prog = bpf_object__find_program_by_name(bpf_obj->obj, "rx");
|
|
bpf_program__set_ifindex(prog, ifindex);
|
|
bpf_program__set_flags(prog, BPF_F_XDP_DEV_BOUND_ONLY);
|
|
|
|
printf("load bpf program...\n");
|
|
ret = xdp_hw_metadata__load(bpf_obj);
|
|
if (ret)
|
|
error(1, -ret, "xdp_hw_metadata__load");
|
|
|
|
printf("prepare skb endpoint...\n");
|
|
server_fd = start_server(AF_INET6, SOCK_DGRAM, NULL, 9092, 1000);
|
|
if (server_fd < 0)
|
|
error(1, errno, "start_server");
|
|
timestamping_enable(server_fd,
|
|
SOF_TIMESTAMPING_SOFTWARE |
|
|
SOF_TIMESTAMPING_RAW_HARDWARE);
|
|
|
|
printf("prepare xsk map...\n");
|
|
for (i = 0; i < rxq; i++) {
|
|
int sock_fd = xsk_socket__fd(rx_xsk[i].socket);
|
|
__u32 queue_id = i;
|
|
|
|
printf("map[%d] = %d\n", queue_id, sock_fd);
|
|
ret = bpf_map_update_elem(bpf_map__fd(bpf_obj->maps.xsk), &queue_id, &sock_fd, 0);
|
|
if (ret)
|
|
error(1, -ret, "bpf_map_update_elem");
|
|
}
|
|
|
|
printf("attach bpf program...\n");
|
|
ret = bpf_xdp_attach(ifindex,
|
|
bpf_program__fd(bpf_obj->progs.rx),
|
|
XDP_FLAGS, NULL);
|
|
if (ret)
|
|
error(1, -ret, "bpf_xdp_attach");
|
|
|
|
signal(SIGINT, handle_signal);
|
|
ret = verify_metadata(rx_xsk, rxq, server_fd, clock_id);
|
|
close(server_fd);
|
|
cleanup();
|
|
if (ret)
|
|
error(1, -ret, "verify_metadata");
|
|
}
|