net/af_xdp: use recvfrom only if busy polling enabled

[dpdk.git] / drivers / net / af_xdp / compat.h

diff --git a/drivers/net/af_xdp/compat.h b/drivers/net/af_xdp/compat.h
index 7aa40d5..52fd447 100644 (file)
--- a/drivers/net/af_xdp/compat.h
+++ b/drivers/net/af_xdp/compat.h
@@ -4,6 +4,7 @@
 
 #include <bpf/xsk.h>
 #include <linux/version.h>
+#include <poll.h>
 
 #if KERNEL_VERSION(5, 10, 0) <= LINUX_VERSION_CODE && \
        defined(RTE_LIBRTE_AF_XDP_PMD_SHARED_UMEM)
@@ -39,3 +40,42 @@ create_shared_socket(struct xsk_socket **xsk_ptr __rte_unused,
        return -1;
 }
 #endif
+
+#ifdef XDP_USE_NEED_WAKEUP
+static void
+rx_syscall_handler(struct xsk_ring_prod *q, uint32_t busy_budget,
+                  struct pollfd *fds, struct xsk_socket *xsk)
+{
+       /* we can assume a kernel >= 5.11 is in use if busy polling is enabled
+        * and thus we can safely use the recvfrom() syscall which is only
+        * supported for AF_XDP sockets in kernels >= 5.11.
+        */
+       if (busy_budget) {
+               (void)recvfrom(xsk_socket__fd(xsk), NULL, 0,
+                       MSG_DONTWAIT, NULL, NULL);
+               return;
+       }
+
+       if (xsk_ring_prod__needs_wakeup(q))
+               (void)poll(fds, 1, 1000);
+}
+static int
+tx_syscall_needed(struct xsk_ring_prod *q)
+{
+       return xsk_ring_prod__needs_wakeup(q);
+}
+#else
+static void
+rx_syscall_handler(struct xsk_ring_prod *q __rte_unused, uint32_t busy_budget,
+                  struct pollfd *fds __rte_unused, struct xsk_socket *xsk)
+{
+       if (busy_budget)
+               (void)recvfrom(xsk_socket__fd(xsk), NULL, 0,
+                       MSG_DONTWAIT, NULL, NULL);
+}
+static int
+tx_syscall_needed(struct xsk_ring_prod *q __rte_unused)
+{
+       return 1;
+}
+#endif