10a714186SBjörn Töpel // SPDX-License-Identifier: GPL-2.0 20a714186SBjörn Töpel /* Copyright(c) 2018 Intel Corporation. */ 30a714186SBjörn Töpel 40a714186SBjörn Töpel #include <linux/bpf_trace.h> 53106c580SMagnus Karlsson #include <linux/stringify.h> 6a71506a4SMagnus Karlsson #include <net/xdp_sock_drv.h> 70a714186SBjörn Töpel #include <net/xdp.h> 80a714186SBjörn Töpel 90a714186SBjörn Töpel #include "i40e.h" 100a714186SBjörn Töpel #include "i40e_txrx_common.h" 110a714186SBjörn Töpel #include "i40e_xsk.h" 120a714186SBjörn Töpel 13be1222b5SBjörn Töpel int i40e_alloc_rx_bi_zc(struct i40e_ring *rx_ring) 14e1675f97SBjörn Töpel { 15be1222b5SBjörn Töpel unsigned long sz = sizeof(*rx_ring->rx_bi_zc) * rx_ring->count; 16be1222b5SBjörn Töpel 17be1222b5SBjörn Töpel rx_ring->rx_bi_zc = kzalloc(sz, GFP_KERNEL); 18be1222b5SBjörn Töpel return rx_ring->rx_bi_zc ? 0 : -ENOMEM; 19be1222b5SBjörn Töpel } 20be1222b5SBjörn Töpel 21be1222b5SBjörn Töpel void i40e_clear_rx_bi_zc(struct i40e_ring *rx_ring) 22be1222b5SBjörn Töpel { 23be1222b5SBjörn Töpel memset(rx_ring->rx_bi_zc, 0, 24be1222b5SBjörn Töpel sizeof(*rx_ring->rx_bi_zc) * rx_ring->count); 25be1222b5SBjörn Töpel } 26be1222b5SBjörn Töpel 273b4f0b66SBjörn Töpel static struct xdp_buff **i40e_rx_bi(struct i40e_ring *rx_ring, u32 idx) 28be1222b5SBjörn Töpel { 29be1222b5SBjörn Töpel return &rx_ring->rx_bi_zc[idx]; 30e1675f97SBjörn Töpel } 31e1675f97SBjörn Töpel 320a714186SBjörn Töpel /** 331742b3d5SMagnus Karlsson * i40e_xsk_pool_enable - Enable/associate an AF_XDP buffer pool to a 341742b3d5SMagnus Karlsson * certain ring/qid 350a714186SBjörn Töpel * @vsi: Current VSI 361742b3d5SMagnus Karlsson * @pool: buffer pool 371742b3d5SMagnus Karlsson * @qid: Rx ring to associate buffer pool with 380a714186SBjörn Töpel * 390a714186SBjörn Töpel * Returns 0 on success, <0 on failure 400a714186SBjörn Töpel **/ 411742b3d5SMagnus Karlsson static int i40e_xsk_pool_enable(struct i40e_vsi *vsi, 421742b3d5SMagnus Karlsson struct xsk_buff_pool *pool, 430a714186SBjörn Töpel u16 qid) 440a714186SBjörn Töpel { 45f3fef2b6SJan Sokolowski struct net_device *netdev = vsi->netdev; 460a714186SBjörn Töpel bool if_running; 470a714186SBjörn Töpel int err; 480a714186SBjörn Töpel 490a714186SBjörn Töpel if (vsi->type != I40E_VSI_MAIN) 500a714186SBjörn Töpel return -EINVAL; 510a714186SBjörn Töpel 520a714186SBjörn Töpel if (qid >= vsi->num_queue_pairs) 530a714186SBjörn Töpel return -EINVAL; 540a714186SBjörn Töpel 55f3fef2b6SJan Sokolowski if (qid >= netdev->real_num_rx_queues || 56f3fef2b6SJan Sokolowski qid >= netdev->real_num_tx_queues) 570a714186SBjörn Töpel return -EINVAL; 580a714186SBjörn Töpel 59c4655761SMagnus Karlsson err = xsk_pool_dma_map(pool, &vsi->back->pdev->dev, I40E_RX_DMA_ATTR); 600a714186SBjörn Töpel if (err) 610a714186SBjörn Töpel return err; 620a714186SBjörn Töpel 6344ddd4f1SBjörn Töpel set_bit(qid, vsi->af_xdp_zc_qps); 6444ddd4f1SBjörn Töpel 650a714186SBjörn Töpel if_running = netif_running(vsi->netdev) && i40e_enabled_xdp_vsi(vsi); 660a714186SBjörn Töpel 670a714186SBjörn Töpel if (if_running) { 680a714186SBjörn Töpel err = i40e_queue_pair_disable(vsi, qid); 690a714186SBjörn Töpel if (err) 700a714186SBjörn Töpel return err; 710a714186SBjörn Töpel 720a714186SBjörn Töpel err = i40e_queue_pair_enable(vsi, qid); 730a714186SBjörn Töpel if (err) 740a714186SBjörn Töpel return err; 7514ffeb52SMagnus Karlsson 7614ffeb52SMagnus Karlsson /* Kick start the NAPI context so that receiving will start */ 779116e5e2SMagnus Karlsson err = i40e_xsk_wakeup(vsi->netdev, qid, XDP_WAKEUP_RX); 7814ffeb52SMagnus Karlsson if (err) 7914ffeb52SMagnus Karlsson return err; 800a714186SBjörn Töpel } 810a714186SBjörn Töpel 820a714186SBjörn Töpel return 0; 830a714186SBjörn Töpel } 840a714186SBjörn Töpel 850a714186SBjörn Töpel /** 861742b3d5SMagnus Karlsson * i40e_xsk_pool_disable - Disassociate an AF_XDP buffer pool from a 871742b3d5SMagnus Karlsson * certain ring/qid 880a714186SBjörn Töpel * @vsi: Current VSI 891742b3d5SMagnus Karlsson * @qid: Rx ring to associate buffer pool with 900a714186SBjörn Töpel * 910a714186SBjörn Töpel * Returns 0 on success, <0 on failure 920a714186SBjörn Töpel **/ 931742b3d5SMagnus Karlsson static int i40e_xsk_pool_disable(struct i40e_vsi *vsi, u16 qid) 940a714186SBjörn Töpel { 95f3fef2b6SJan Sokolowski struct net_device *netdev = vsi->netdev; 961742b3d5SMagnus Karlsson struct xsk_buff_pool *pool; 970a714186SBjörn Töpel bool if_running; 980a714186SBjörn Töpel int err; 990a714186SBjörn Töpel 100c4655761SMagnus Karlsson pool = xsk_get_pool_from_qid(netdev, qid); 1011742b3d5SMagnus Karlsson if (!pool) 1020a714186SBjörn Töpel return -EINVAL; 1030a714186SBjörn Töpel 1040a714186SBjörn Töpel if_running = netif_running(vsi->netdev) && i40e_enabled_xdp_vsi(vsi); 1050a714186SBjörn Töpel 1060a714186SBjörn Töpel if (if_running) { 1070a714186SBjörn Töpel err = i40e_queue_pair_disable(vsi, qid); 1080a714186SBjörn Töpel if (err) 1090a714186SBjörn Töpel return err; 1100a714186SBjörn Töpel } 1110a714186SBjörn Töpel 11244ddd4f1SBjörn Töpel clear_bit(qid, vsi->af_xdp_zc_qps); 113c4655761SMagnus Karlsson xsk_pool_dma_unmap(pool, I40E_RX_DMA_ATTR); 1140a714186SBjörn Töpel 1150a714186SBjörn Töpel if (if_running) { 1160a714186SBjörn Töpel err = i40e_queue_pair_enable(vsi, qid); 1170a714186SBjörn Töpel if (err) 1180a714186SBjörn Töpel return err; 1190a714186SBjörn Töpel } 1200a714186SBjörn Töpel 1210a714186SBjörn Töpel return 0; 1220a714186SBjörn Töpel } 1230a714186SBjörn Töpel 1240a714186SBjörn Töpel /** 1251742b3d5SMagnus Karlsson * i40e_xsk_pool_setup - Enable/disassociate an AF_XDP buffer pool to/from 1261742b3d5SMagnus Karlsson * a ring/qid 1270a714186SBjörn Töpel * @vsi: Current VSI 1281742b3d5SMagnus Karlsson * @pool: Buffer pool to enable/associate to a ring, or NULL to disable 1291742b3d5SMagnus Karlsson * @qid: Rx ring to (dis)associate buffer pool (from)to 1300a714186SBjörn Töpel * 1311742b3d5SMagnus Karlsson * This function enables or disables a buffer pool to a certain ring. 1320a714186SBjörn Töpel * 1330a714186SBjörn Töpel * Returns 0 on success, <0 on failure 1340a714186SBjörn Töpel **/ 1351742b3d5SMagnus Karlsson int i40e_xsk_pool_setup(struct i40e_vsi *vsi, struct xsk_buff_pool *pool, 1360a714186SBjörn Töpel u16 qid) 1370a714186SBjörn Töpel { 1381742b3d5SMagnus Karlsson return pool ? i40e_xsk_pool_enable(vsi, pool, qid) : 1391742b3d5SMagnus Karlsson i40e_xsk_pool_disable(vsi, qid); 1400a714186SBjörn Töpel } 1410a714186SBjörn Töpel 1420a714186SBjörn Töpel /** 1430a714186SBjörn Töpel * i40e_run_xdp_zc - Executes an XDP program on an xdp_buff 1440a714186SBjörn Töpel * @rx_ring: Rx ring 1450a714186SBjörn Töpel * @xdp: xdp_buff used as input to the XDP program 1460a714186SBjörn Töpel * 1470a714186SBjörn Töpel * Returns any of I40E_XDP_{PASS, CONSUMED, TX, REDIR} 1480a714186SBjörn Töpel **/ 1490a714186SBjörn Töpel static int i40e_run_xdp_zc(struct i40e_ring *rx_ring, struct xdp_buff *xdp) 1500a714186SBjörn Töpel { 1510a714186SBjörn Töpel int err, result = I40E_XDP_PASS; 1520a714186SBjörn Töpel struct i40e_ring *xdp_ring; 1530a714186SBjörn Töpel struct bpf_prog *xdp_prog; 1540a714186SBjörn Töpel u32 act; 1550a714186SBjörn Töpel 1560a714186SBjörn Töpel rcu_read_lock(); 1570a714186SBjörn Töpel /* NB! xdp_prog will always be !NULL, due to the fact that 1580a714186SBjörn Töpel * this path is enabled by setting an XDP program. 1590a714186SBjörn Töpel */ 1600a714186SBjörn Töpel xdp_prog = READ_ONCE(rx_ring->xdp_prog); 1610a714186SBjörn Töpel act = bpf_prog_run_xdp(xdp_prog, xdp); 1622f86c806SKevin Laatz 1630a714186SBjörn Töpel switch (act) { 1640a714186SBjörn Töpel case XDP_PASS: 1650a714186SBjörn Töpel break; 1660a714186SBjörn Töpel case XDP_TX: 1670a714186SBjörn Töpel xdp_ring = rx_ring->vsi->xdp_rings[rx_ring->queue_index]; 1680a714186SBjörn Töpel result = i40e_xmit_xdp_tx_ring(xdp, xdp_ring); 1690a714186SBjörn Töpel break; 1700a714186SBjörn Töpel case XDP_REDIRECT: 1710a714186SBjörn Töpel err = xdp_do_redirect(rx_ring->netdev, xdp, xdp_prog); 1720a714186SBjörn Töpel result = !err ? I40E_XDP_REDIR : I40E_XDP_CONSUMED; 1730a714186SBjörn Töpel break; 1740a714186SBjörn Töpel default: 1750a714186SBjörn Töpel bpf_warn_invalid_xdp_action(act); 1765463fce6SJeff Kirsher fallthrough; 1770a714186SBjörn Töpel case XDP_ABORTED: 1780a714186SBjörn Töpel trace_xdp_exception(rx_ring->netdev, xdp_prog, act); 1795463fce6SJeff Kirsher fallthrough; /* handle aborts by dropping packet */ 1800a714186SBjörn Töpel case XDP_DROP: 1810a714186SBjörn Töpel result = I40E_XDP_CONSUMED; 1820a714186SBjörn Töpel break; 1830a714186SBjörn Töpel } 1840a714186SBjörn Töpel rcu_read_unlock(); 1850a714186SBjörn Töpel return result; 1860a714186SBjörn Töpel } 1870a714186SBjörn Töpel 1883b4f0b66SBjörn Töpel bool i40e_alloc_rx_buffers_zc(struct i40e_ring *rx_ring, u16 count) 1890a714186SBjörn Töpel { 1900a714186SBjörn Töpel u16 ntu = rx_ring->next_to_use; 1910a714186SBjörn Töpel union i40e_rx_desc *rx_desc; 1923b4f0b66SBjörn Töpel struct xdp_buff **bi, *xdp; 1933b4f0b66SBjörn Töpel dma_addr_t dma; 1940a714186SBjörn Töpel bool ok = true; 1950a714186SBjörn Töpel 1960a714186SBjörn Töpel rx_desc = I40E_RX_DESC(rx_ring, ntu); 197e1675f97SBjörn Töpel bi = i40e_rx_bi(rx_ring, ntu); 1980a714186SBjörn Töpel do { 199c4655761SMagnus Karlsson xdp = xsk_buff_alloc(rx_ring->xsk_pool); 2003b4f0b66SBjörn Töpel if (!xdp) { 2010a714186SBjörn Töpel ok = false; 2020a714186SBjörn Töpel goto no_buffers; 2030a714186SBjörn Töpel } 2043b4f0b66SBjörn Töpel *bi = xdp; 2053b4f0b66SBjörn Töpel dma = xsk_buff_xdp_get_dma(xdp); 2063b4f0b66SBjörn Töpel rx_desc->read.pkt_addr = cpu_to_le64(dma); 2073b4f0b66SBjörn Töpel rx_desc->read.hdr_addr = 0; 2080a714186SBjörn Töpel 2090a714186SBjörn Töpel rx_desc++; 2100a714186SBjörn Töpel bi++; 2110a714186SBjörn Töpel ntu++; 2120a714186SBjörn Töpel 2130a714186SBjörn Töpel if (unlikely(ntu == rx_ring->count)) { 2140a714186SBjörn Töpel rx_desc = I40E_RX_DESC(rx_ring, 0); 215e1675f97SBjörn Töpel bi = i40e_rx_bi(rx_ring, 0); 2160a714186SBjörn Töpel ntu = 0; 2170a714186SBjörn Töpel } 2180a714186SBjörn Töpel 2190a714186SBjörn Töpel count--; 2200a714186SBjörn Töpel } while (count); 2210a714186SBjörn Töpel 2220a714186SBjörn Töpel no_buffers: 22364050b5bSBjörn Töpel if (rx_ring->next_to_use != ntu) { 22464050b5bSBjörn Töpel /* clear the status bits for the next_to_use descriptor */ 22564050b5bSBjörn Töpel rx_desc->wb.qword1.status_error_len = 0; 2260a714186SBjörn Töpel i40e_release_rx_desc(rx_ring, ntu); 22764050b5bSBjörn Töpel } 2280a714186SBjörn Töpel 2290a714186SBjörn Töpel return ok; 2300a714186SBjörn Töpel } 2310a714186SBjörn Töpel 2320a714186SBjörn Töpel /** 233e92c0e02SJesper Dangaard Brouer * i40e_construct_skb_zc - Create skbuff from zero-copy Rx buffer 2340a714186SBjörn Töpel * @rx_ring: Rx ring 2350a714186SBjörn Töpel * @xdp: xdp_buff 2360a714186SBjörn Töpel * 2370a714186SBjörn Töpel * This functions allocates a new skb from a zero-copy Rx buffer. 2380a714186SBjörn Töpel * 2390a714186SBjörn Töpel * Returns the skb, or NULL on failure. 2400a714186SBjörn Töpel **/ 2410a714186SBjörn Töpel static struct sk_buff *i40e_construct_skb_zc(struct i40e_ring *rx_ring, 2420a714186SBjörn Töpel struct xdp_buff *xdp) 2430a714186SBjörn Töpel { 2440a714186SBjörn Töpel unsigned int metasize = xdp->data - xdp->data_meta; 2450a714186SBjörn Töpel unsigned int datasize = xdp->data_end - xdp->data; 2460a714186SBjörn Töpel struct sk_buff *skb; 2470a714186SBjörn Töpel 2480a714186SBjörn Töpel /* allocate a skb to store the frags */ 2490a714186SBjörn Töpel skb = __napi_alloc_skb(&rx_ring->q_vector->napi, 2500a714186SBjörn Töpel xdp->data_end - xdp->data_hard_start, 2510a714186SBjörn Töpel GFP_ATOMIC | __GFP_NOWARN); 2520a714186SBjörn Töpel if (unlikely(!skb)) 2530a714186SBjörn Töpel return NULL; 2540a714186SBjörn Töpel 2550a714186SBjörn Töpel skb_reserve(skb, xdp->data - xdp->data_hard_start); 2560a714186SBjörn Töpel memcpy(__skb_put(skb, datasize), xdp->data, datasize); 2570a714186SBjörn Töpel if (metasize) 2580a714186SBjörn Töpel skb_metadata_set(skb, metasize); 2590a714186SBjörn Töpel 2603b4f0b66SBjörn Töpel xsk_buff_free(xdp); 2610a714186SBjörn Töpel return skb; 2620a714186SBjörn Töpel } 2630a714186SBjörn Töpel 2640a714186SBjörn Töpel /** 2650a714186SBjörn Töpel * i40e_clean_rx_irq_zc - Consumes Rx packets from the hardware ring 2660a714186SBjörn Töpel * @rx_ring: Rx ring 2670a714186SBjörn Töpel * @budget: NAPI budget 2680a714186SBjörn Töpel * 2690a714186SBjörn Töpel * Returns amount of work completed 2700a714186SBjörn Töpel **/ 2710a714186SBjörn Töpel int i40e_clean_rx_irq_zc(struct i40e_ring *rx_ring, int budget) 2720a714186SBjörn Töpel { 2730a714186SBjörn Töpel unsigned int total_rx_bytes = 0, total_rx_packets = 0; 2740a714186SBjörn Töpel u16 cleaned_count = I40E_DESC_UNUSED(rx_ring); 275c8a8ca34SCristian Dumitrescu u16 next_to_clean = rx_ring->next_to_clean; 276c8a8ca34SCristian Dumitrescu u16 count_mask = rx_ring->count - 1; 2770a714186SBjörn Töpel unsigned int xdp_res, xdp_xmit = 0; 2781773482fSDan Carpenter bool failure = false; 2790a714186SBjörn Töpel struct sk_buff *skb; 2800a714186SBjörn Töpel 2810a714186SBjörn Töpel while (likely(total_rx_packets < (unsigned int)budget)) { 2820a714186SBjörn Töpel union i40e_rx_desc *rx_desc; 283*d4178c31SCristian Dumitrescu struct xdp_buff *bi; 2840a714186SBjörn Töpel unsigned int size; 2850a714186SBjörn Töpel u64 qword; 2860a714186SBjörn Töpel 287c8a8ca34SCristian Dumitrescu rx_desc = I40E_RX_DESC(rx_ring, next_to_clean); 2880a714186SBjörn Töpel qword = le64_to_cpu(rx_desc->wb.qword1.status_error_len); 2890a714186SBjörn Töpel 2900a714186SBjörn Töpel /* This memory barrier is needed to keep us from reading 2910a714186SBjörn Töpel * any other fields out of the rx_desc until we have 2920a714186SBjörn Töpel * verified the descriptor has been written back. 2930a714186SBjörn Töpel */ 2940a714186SBjörn Töpel dma_rmb(); 2950a714186SBjörn Töpel 296be1222b5SBjörn Töpel if (i40e_rx_is_programming_status(qword)) { 297be1222b5SBjörn Töpel i40e_clean_programming_status(rx_ring, 298be1222b5SBjörn Töpel rx_desc->raw.qword[0], 2990a714186SBjörn Töpel qword); 300*d4178c31SCristian Dumitrescu bi = *i40e_rx_bi(rx_ring, next_to_clean); 301*d4178c31SCristian Dumitrescu xsk_buff_free(bi); 302c8a8ca34SCristian Dumitrescu next_to_clean = (next_to_clean + 1) & count_mask; 3030a714186SBjörn Töpel continue; 3040a714186SBjörn Töpel } 3050a714186SBjörn Töpel 3060a714186SBjörn Töpel size = (qword & I40E_RXD_QW1_LENGTH_PBUF_MASK) >> 3070a714186SBjörn Töpel I40E_RXD_QW1_LENGTH_PBUF_SHIFT; 3080a714186SBjörn Töpel if (!size) 3090a714186SBjörn Töpel break; 3100a714186SBjörn Töpel 311*d4178c31SCristian Dumitrescu bi = *i40e_rx_bi(rx_ring, next_to_clean); 312*d4178c31SCristian Dumitrescu bi->data_end = bi->data + size; 313*d4178c31SCristian Dumitrescu xsk_buff_dma_sync_for_cpu(bi, rx_ring->xsk_pool); 3140a714186SBjörn Töpel 315*d4178c31SCristian Dumitrescu xdp_res = i40e_run_xdp_zc(rx_ring, bi); 3160a714186SBjörn Töpel if (xdp_res) { 3173b4f0b66SBjörn Töpel if (xdp_res & (I40E_XDP_TX | I40E_XDP_REDIR)) 3180a714186SBjörn Töpel xdp_xmit |= xdp_res; 3193b4f0b66SBjörn Töpel else 320*d4178c31SCristian Dumitrescu xsk_buff_free(bi); 3210a714186SBjörn Töpel 3220a714186SBjörn Töpel total_rx_bytes += size; 3230a714186SBjörn Töpel total_rx_packets++; 3240a714186SBjörn Töpel 325c8a8ca34SCristian Dumitrescu next_to_clean = (next_to_clean + 1) & count_mask; 3260a714186SBjörn Töpel continue; 3270a714186SBjörn Töpel } 3280a714186SBjörn Töpel 3290a714186SBjörn Töpel /* XDP_PASS path */ 3300a714186SBjörn Töpel 3310a714186SBjörn Töpel /* NB! We are not checking for errors using 3320a714186SBjörn Töpel * i40e_test_staterr with 3330a714186SBjörn Töpel * BIT(I40E_RXD_QW1_ERROR_SHIFT). This is due to that 3340a714186SBjörn Töpel * SBP is *not* set in PRT_SBPVSI (default not set). 3350a714186SBjörn Töpel */ 336*d4178c31SCristian Dumitrescu skb = i40e_construct_skb_zc(rx_ring, bi); 3370a714186SBjörn Töpel if (!skb) { 3380a714186SBjörn Töpel rx_ring->rx_stats.alloc_buff_failed++; 3390a714186SBjörn Töpel break; 3400a714186SBjörn Töpel } 3410a714186SBjörn Töpel 342c8a8ca34SCristian Dumitrescu next_to_clean = (next_to_clean + 1) & count_mask; 3430a714186SBjörn Töpel 3440a714186SBjörn Töpel if (eth_skb_pad(skb)) 3450a714186SBjörn Töpel continue; 3460a714186SBjörn Töpel 3470a714186SBjörn Töpel total_rx_bytes += skb->len; 3480a714186SBjörn Töpel total_rx_packets++; 3490a714186SBjörn Töpel 350800b8f63SMichał Mirosław i40e_process_skb_fields(rx_ring, rx_desc, skb); 3512a508c64SMichał Mirosław napi_gro_receive(&rx_ring->q_vector->napi, skb); 3520a714186SBjörn Töpel } 3530a714186SBjörn Töpel 354c8a8ca34SCristian Dumitrescu rx_ring->next_to_clean = next_to_clean; 355f12738b6SCristian Dumitrescu cleaned_count = (next_to_clean - rx_ring->next_to_use - 1) & count_mask; 356c8a8ca34SCristian Dumitrescu 3578cbf7414SBjörn Töpel if (cleaned_count >= I40E_RX_BUFFER_WRITE) 3588cbf7414SBjörn Töpel failure = !i40e_alloc_rx_buffers_zc(rx_ring, cleaned_count); 3598cbf7414SBjörn Töpel 3600a714186SBjörn Töpel i40e_finalize_xdp_rx(rx_ring, xdp_xmit); 3610a714186SBjörn Töpel i40e_update_rx_stats(rx_ring, total_rx_bytes, total_rx_packets); 3623d0c5f1cSMagnus Karlsson 363c4655761SMagnus Karlsson if (xsk_uses_need_wakeup(rx_ring->xsk_pool)) { 364c8a8ca34SCristian Dumitrescu if (failure || next_to_clean == rx_ring->next_to_use) 365c4655761SMagnus Karlsson xsk_set_rx_need_wakeup(rx_ring->xsk_pool); 3663d0c5f1cSMagnus Karlsson else 367c4655761SMagnus Karlsson xsk_clear_rx_need_wakeup(rx_ring->xsk_pool); 3683d0c5f1cSMagnus Karlsson 3693d0c5f1cSMagnus Karlsson return (int)total_rx_packets; 3703d0c5f1cSMagnus Karlsson } 3710a714186SBjörn Töpel return failure ? budget : (int)total_rx_packets; 3720a714186SBjörn Töpel } 3730a714186SBjörn Töpel 3743106c580SMagnus Karlsson static void i40e_xmit_pkt(struct i40e_ring *xdp_ring, struct xdp_desc *desc, 3753106c580SMagnus Karlsson unsigned int *total_bytes) 3763106c580SMagnus Karlsson { 3773106c580SMagnus Karlsson struct i40e_tx_desc *tx_desc; 3783106c580SMagnus Karlsson dma_addr_t dma; 3793106c580SMagnus Karlsson 3803106c580SMagnus Karlsson dma = xsk_buff_raw_get_dma(xdp_ring->xsk_pool, desc->addr); 3813106c580SMagnus Karlsson xsk_buff_raw_dma_sync_for_device(xdp_ring->xsk_pool, dma, desc->len); 3823106c580SMagnus Karlsson 3833106c580SMagnus Karlsson tx_desc = I40E_TX_DESC(xdp_ring, xdp_ring->next_to_use++); 3843106c580SMagnus Karlsson tx_desc->buffer_addr = cpu_to_le64(dma); 3853106c580SMagnus Karlsson tx_desc->cmd_type_offset_bsz = build_ctob(I40E_TX_DESC_CMD_ICRC | I40E_TX_DESC_CMD_EOP, 3863106c580SMagnus Karlsson 0, desc->len, 0); 3873106c580SMagnus Karlsson 3883106c580SMagnus Karlsson *total_bytes += desc->len; 3893106c580SMagnus Karlsson } 3903106c580SMagnus Karlsson 3913106c580SMagnus Karlsson static void i40e_xmit_pkt_batch(struct i40e_ring *xdp_ring, struct xdp_desc *desc, 3923106c580SMagnus Karlsson unsigned int *total_bytes) 3933106c580SMagnus Karlsson { 3943106c580SMagnus Karlsson u16 ntu = xdp_ring->next_to_use; 3953106c580SMagnus Karlsson struct i40e_tx_desc *tx_desc; 3963106c580SMagnus Karlsson dma_addr_t dma; 3973106c580SMagnus Karlsson u32 i; 3983106c580SMagnus Karlsson 3993106c580SMagnus Karlsson loop_unrolled_for(i = 0; i < PKTS_PER_BATCH; i++) { 4003106c580SMagnus Karlsson dma = xsk_buff_raw_get_dma(xdp_ring->xsk_pool, desc[i].addr); 4013106c580SMagnus Karlsson xsk_buff_raw_dma_sync_for_device(xdp_ring->xsk_pool, dma, desc[i].len); 4023106c580SMagnus Karlsson 4033106c580SMagnus Karlsson tx_desc = I40E_TX_DESC(xdp_ring, ntu++); 4043106c580SMagnus Karlsson tx_desc->buffer_addr = cpu_to_le64(dma); 4053106c580SMagnus Karlsson tx_desc->cmd_type_offset_bsz = build_ctob(I40E_TX_DESC_CMD_ICRC | 4063106c580SMagnus Karlsson I40E_TX_DESC_CMD_EOP, 4073106c580SMagnus Karlsson 0, desc[i].len, 0); 4083106c580SMagnus Karlsson 4093106c580SMagnus Karlsson *total_bytes += desc[i].len; 4103106c580SMagnus Karlsson } 4113106c580SMagnus Karlsson 4123106c580SMagnus Karlsson xdp_ring->next_to_use = ntu; 4133106c580SMagnus Karlsson } 4143106c580SMagnus Karlsson 4153106c580SMagnus Karlsson static void i40e_fill_tx_hw_ring(struct i40e_ring *xdp_ring, struct xdp_desc *descs, u32 nb_pkts, 4163106c580SMagnus Karlsson unsigned int *total_bytes) 4173106c580SMagnus Karlsson { 4183106c580SMagnus Karlsson u32 batched, leftover, i; 4193106c580SMagnus Karlsson 4203106c580SMagnus Karlsson batched = nb_pkts & ~(PKTS_PER_BATCH - 1); 4213106c580SMagnus Karlsson leftover = nb_pkts & (PKTS_PER_BATCH - 1); 4223106c580SMagnus Karlsson for (i = 0; i < batched; i += PKTS_PER_BATCH) 4233106c580SMagnus Karlsson i40e_xmit_pkt_batch(xdp_ring, &descs[i], total_bytes); 4243106c580SMagnus Karlsson for (i = batched; i < batched + leftover; i++) 4253106c580SMagnus Karlsson i40e_xmit_pkt(xdp_ring, &descs[i], total_bytes); 4263106c580SMagnus Karlsson } 4273106c580SMagnus Karlsson 4283106c580SMagnus Karlsson static void i40e_set_rs_bit(struct i40e_ring *xdp_ring) 4293106c580SMagnus Karlsson { 4303106c580SMagnus Karlsson u16 ntu = xdp_ring->next_to_use ? xdp_ring->next_to_use - 1 : xdp_ring->count - 1; 4313106c580SMagnus Karlsson struct i40e_tx_desc *tx_desc; 4323106c580SMagnus Karlsson 4333106c580SMagnus Karlsson tx_desc = I40E_TX_DESC(xdp_ring, ntu); 4343106c580SMagnus Karlsson tx_desc->cmd_type_offset_bsz |= (I40E_TX_DESC_CMD_RS << I40E_TXD_QW1_CMD_SHIFT); 4353106c580SMagnus Karlsson } 4363106c580SMagnus Karlsson 4371328dcddSMagnus Karlsson /** 4381328dcddSMagnus Karlsson * i40e_xmit_zc - Performs zero-copy Tx AF_XDP 4391328dcddSMagnus Karlsson * @xdp_ring: XDP Tx ring 4401328dcddSMagnus Karlsson * @budget: NAPI budget 4411328dcddSMagnus Karlsson * 4421328dcddSMagnus Karlsson * Returns true if the work is finished. 4431328dcddSMagnus Karlsson **/ 4441328dcddSMagnus Karlsson static bool i40e_xmit_zc(struct i40e_ring *xdp_ring, unsigned int budget) 4451328dcddSMagnus Karlsson { 4463106c580SMagnus Karlsson struct xdp_desc *descs = xdp_ring->xsk_descs; 4473106c580SMagnus Karlsson u32 nb_pkts, nb_processed = 0; 4483106c580SMagnus Karlsson unsigned int total_bytes = 0; 4491328dcddSMagnus Karlsson 4503106c580SMagnus Karlsson nb_pkts = xsk_tx_peek_release_desc_batch(xdp_ring->xsk_pool, descs, budget); 4513106c580SMagnus Karlsson if (!nb_pkts) 4523106c580SMagnus Karlsson return false; 4531328dcddSMagnus Karlsson 4543106c580SMagnus Karlsson if (xdp_ring->next_to_use + nb_pkts >= xdp_ring->count) { 4553106c580SMagnus Karlsson nb_processed = xdp_ring->count - xdp_ring->next_to_use; 4563106c580SMagnus Karlsson i40e_fill_tx_hw_ring(xdp_ring, descs, nb_processed, &total_bytes); 4571328dcddSMagnus Karlsson xdp_ring->next_to_use = 0; 4581328dcddSMagnus Karlsson } 4591328dcddSMagnus Karlsson 4603106c580SMagnus Karlsson i40e_fill_tx_hw_ring(xdp_ring, &descs[nb_processed], nb_pkts - nb_processed, 4613106c580SMagnus Karlsson &total_bytes); 4623106c580SMagnus Karlsson 4631328dcddSMagnus Karlsson /* Request an interrupt for the last frame and bump tail ptr. */ 4643106c580SMagnus Karlsson i40e_set_rs_bit(xdp_ring); 4651328dcddSMagnus Karlsson i40e_xdp_ring_update_tail(xdp_ring); 4661328dcddSMagnus Karlsson 4673106c580SMagnus Karlsson i40e_update_tx_stats(xdp_ring, nb_pkts, total_bytes); 4681328dcddSMagnus Karlsson 4693106c580SMagnus Karlsson return true; 4701328dcddSMagnus Karlsson } 4711328dcddSMagnus Karlsson 4721328dcddSMagnus Karlsson /** 4731328dcddSMagnus Karlsson * i40e_clean_xdp_tx_buffer - Frees and unmaps an XDP Tx entry 4741328dcddSMagnus Karlsson * @tx_ring: XDP Tx ring 4751328dcddSMagnus Karlsson * @tx_bi: Tx buffer info to clean 4761328dcddSMagnus Karlsson **/ 4771328dcddSMagnus Karlsson static void i40e_clean_xdp_tx_buffer(struct i40e_ring *tx_ring, 4781328dcddSMagnus Karlsson struct i40e_tx_buffer *tx_bi) 4791328dcddSMagnus Karlsson { 4801328dcddSMagnus Karlsson xdp_return_frame(tx_bi->xdpf); 4815574ff7bSMagnus Karlsson tx_ring->xdp_tx_active--; 4821328dcddSMagnus Karlsson dma_unmap_single(tx_ring->dev, 4831328dcddSMagnus Karlsson dma_unmap_addr(tx_bi, dma), 4841328dcddSMagnus Karlsson dma_unmap_len(tx_bi, len), DMA_TO_DEVICE); 4851328dcddSMagnus Karlsson dma_unmap_len_set(tx_bi, len, 0); 4861328dcddSMagnus Karlsson } 4871328dcddSMagnus Karlsson 4881328dcddSMagnus Karlsson /** 4891328dcddSMagnus Karlsson * i40e_clean_xdp_tx_irq - Completes AF_XDP entries, and cleans XDP entries 4901fd972ebSMagnus Karlsson * @vsi: Current VSI 4911328dcddSMagnus Karlsson * @tx_ring: XDP Tx ring 4921328dcddSMagnus Karlsson * 4931328dcddSMagnus Karlsson * Returns true if cleanup/tranmission is done. 4941328dcddSMagnus Karlsson **/ 4955574ff7bSMagnus Karlsson bool i40e_clean_xdp_tx_irq(struct i40e_vsi *vsi, struct i40e_ring *tx_ring) 4961328dcddSMagnus Karlsson { 4971742b3d5SMagnus Karlsson struct xsk_buff_pool *bp = tx_ring->xsk_pool; 4985574ff7bSMagnus Karlsson u32 i, completed_frames, xsk_frames = 0; 4991328dcddSMagnus Karlsson u32 head_idx = i40e_get_head(tx_ring); 5001328dcddSMagnus Karlsson struct i40e_tx_buffer *tx_bi; 5011fd972ebSMagnus Karlsson unsigned int ntc; 5021328dcddSMagnus Karlsson 5031328dcddSMagnus Karlsson if (head_idx < tx_ring->next_to_clean) 5041328dcddSMagnus Karlsson head_idx += tx_ring->count; 5055574ff7bSMagnus Karlsson completed_frames = head_idx - tx_ring->next_to_clean; 5061328dcddSMagnus Karlsson 5075574ff7bSMagnus Karlsson if (completed_frames == 0) 5081328dcddSMagnus Karlsson goto out_xmit; 5095574ff7bSMagnus Karlsson 5105574ff7bSMagnus Karlsson if (likely(!tx_ring->xdp_tx_active)) { 5115574ff7bSMagnus Karlsson xsk_frames = completed_frames; 5125574ff7bSMagnus Karlsson goto skip; 5131328dcddSMagnus Karlsson } 5141328dcddSMagnus Karlsson 5151328dcddSMagnus Karlsson ntc = tx_ring->next_to_clean; 5161328dcddSMagnus Karlsson 5171328dcddSMagnus Karlsson for (i = 0; i < completed_frames; i++) { 5181328dcddSMagnus Karlsson tx_bi = &tx_ring->tx_bi[ntc]; 5191328dcddSMagnus Karlsson 5205574ff7bSMagnus Karlsson if (tx_bi->xdpf) { 5211328dcddSMagnus Karlsson i40e_clean_xdp_tx_buffer(tx_ring, tx_bi); 5221328dcddSMagnus Karlsson tx_bi->xdpf = NULL; 5235574ff7bSMagnus Karlsson } else { 5245574ff7bSMagnus Karlsson xsk_frames++; 5255574ff7bSMagnus Karlsson } 5261328dcddSMagnus Karlsson 5271328dcddSMagnus Karlsson if (++ntc >= tx_ring->count) 5281328dcddSMagnus Karlsson ntc = 0; 5291328dcddSMagnus Karlsson } 5301328dcddSMagnus Karlsson 5315574ff7bSMagnus Karlsson skip: 5321328dcddSMagnus Karlsson tx_ring->next_to_clean += completed_frames; 5331328dcddSMagnus Karlsson if (unlikely(tx_ring->next_to_clean >= tx_ring->count)) 5341328dcddSMagnus Karlsson tx_ring->next_to_clean -= tx_ring->count; 5351328dcddSMagnus Karlsson 5361328dcddSMagnus Karlsson if (xsk_frames) 537c4655761SMagnus Karlsson xsk_tx_completed(bp, xsk_frames); 5381328dcddSMagnus Karlsson 5395574ff7bSMagnus Karlsson i40e_arm_wb(tx_ring, vsi, completed_frames); 5401328dcddSMagnus Karlsson 5411328dcddSMagnus Karlsson out_xmit: 542c4655761SMagnus Karlsson if (xsk_uses_need_wakeup(tx_ring->xsk_pool)) 543c4655761SMagnus Karlsson xsk_set_tx_need_wakeup(tx_ring->xsk_pool); 5443d0c5f1cSMagnus Karlsson 5451fd972ebSMagnus Karlsson return i40e_xmit_zc(tx_ring, I40E_DESC_UNUSED(tx_ring)); 5461328dcddSMagnus Karlsson } 5471328dcddSMagnus Karlsson 5481328dcddSMagnus Karlsson /** 5499116e5e2SMagnus Karlsson * i40e_xsk_wakeup - Implements the ndo_xsk_wakeup 5501328dcddSMagnus Karlsson * @dev: the netdevice 5511328dcddSMagnus Karlsson * @queue_id: queue id to wake up 5529116e5e2SMagnus Karlsson * @flags: ignored in our case since we have Rx and Tx in the same NAPI. 5531328dcddSMagnus Karlsson * 5541328dcddSMagnus Karlsson * Returns <0 for errors, 0 otherwise. 5551328dcddSMagnus Karlsson **/ 5569116e5e2SMagnus Karlsson int i40e_xsk_wakeup(struct net_device *dev, u32 queue_id, u32 flags) 5571328dcddSMagnus Karlsson { 5581328dcddSMagnus Karlsson struct i40e_netdev_priv *np = netdev_priv(dev); 5591328dcddSMagnus Karlsson struct i40e_vsi *vsi = np->vsi; 560b3873a5bSMaxim Mikityanskiy struct i40e_pf *pf = vsi->back; 5611328dcddSMagnus Karlsson struct i40e_ring *ring; 5621328dcddSMagnus Karlsson 563b3873a5bSMaxim Mikityanskiy if (test_bit(__I40E_CONFIG_BUSY, pf->state)) 564c77e9f09SMaciej Fijalkowski return -EAGAIN; 565b3873a5bSMaxim Mikityanskiy 5661328dcddSMagnus Karlsson if (test_bit(__I40E_VSI_DOWN, vsi->state)) 5671328dcddSMagnus Karlsson return -ENETDOWN; 5681328dcddSMagnus Karlsson 5691328dcddSMagnus Karlsson if (!i40e_enabled_xdp_vsi(vsi)) 5701328dcddSMagnus Karlsson return -ENXIO; 5711328dcddSMagnus Karlsson 5721328dcddSMagnus Karlsson if (queue_id >= vsi->num_queue_pairs) 5731328dcddSMagnus Karlsson return -ENXIO; 5741328dcddSMagnus Karlsson 5751742b3d5SMagnus Karlsson if (!vsi->xdp_rings[queue_id]->xsk_pool) 5761328dcddSMagnus Karlsson return -ENXIO; 5771328dcddSMagnus Karlsson 5781328dcddSMagnus Karlsson ring = vsi->xdp_rings[queue_id]; 5791328dcddSMagnus Karlsson 5801328dcddSMagnus Karlsson /* The idea here is that if NAPI is running, mark a miss, so 5811328dcddSMagnus Karlsson * it will run again. If not, trigger an interrupt and 5821328dcddSMagnus Karlsson * schedule the NAPI from interrupt context. If NAPI would be 5831328dcddSMagnus Karlsson * scheduled here, the interrupt affinity would not be 5841328dcddSMagnus Karlsson * honored. 5851328dcddSMagnus Karlsson */ 5861328dcddSMagnus Karlsson if (!napi_if_scheduled_mark_missed(&ring->q_vector->napi)) 5871328dcddSMagnus Karlsson i40e_force_wb(vsi, ring->q_vector); 5881328dcddSMagnus Karlsson 5891328dcddSMagnus Karlsson return 0; 5901328dcddSMagnus Karlsson } 5919dbb1370SBjörn Töpel 592411dc16fSBjörn Töpel void i40e_xsk_clean_rx_ring(struct i40e_ring *rx_ring) 593411dc16fSBjörn Töpel { 594*d4178c31SCristian Dumitrescu u16 count_mask = rx_ring->count - 1; 595*d4178c31SCristian Dumitrescu u16 ntc = rx_ring->next_to_clean; 596*d4178c31SCristian Dumitrescu u16 ntu = rx_ring->next_to_use; 597411dc16fSBjörn Töpel 598*d4178c31SCristian Dumitrescu for ( ; ntc != ntu; ntc = (ntc + 1) & count_mask) { 599*d4178c31SCristian Dumitrescu struct xdp_buff *rx_bi = *i40e_rx_bi(rx_ring, ntc); 600411dc16fSBjörn Töpel 6013b4f0b66SBjörn Töpel xsk_buff_free(rx_bi); 602411dc16fSBjörn Töpel } 603411dc16fSBjörn Töpel } 604411dc16fSBjörn Töpel 6059dbb1370SBjörn Töpel /** 6069dbb1370SBjörn Töpel * i40e_xsk_clean_xdp_ring - Clean the XDP Tx ring on shutdown 6071fd972ebSMagnus Karlsson * @tx_ring: XDP Tx ring 6089dbb1370SBjörn Töpel **/ 6099dbb1370SBjörn Töpel void i40e_xsk_clean_tx_ring(struct i40e_ring *tx_ring) 6109dbb1370SBjörn Töpel { 6119dbb1370SBjörn Töpel u16 ntc = tx_ring->next_to_clean, ntu = tx_ring->next_to_use; 6121742b3d5SMagnus Karlsson struct xsk_buff_pool *bp = tx_ring->xsk_pool; 6139dbb1370SBjörn Töpel struct i40e_tx_buffer *tx_bi; 6149dbb1370SBjörn Töpel u32 xsk_frames = 0; 6159dbb1370SBjörn Töpel 6169dbb1370SBjörn Töpel while (ntc != ntu) { 6179dbb1370SBjörn Töpel tx_bi = &tx_ring->tx_bi[ntc]; 6189dbb1370SBjörn Töpel 6199dbb1370SBjörn Töpel if (tx_bi->xdpf) 6209dbb1370SBjörn Töpel i40e_clean_xdp_tx_buffer(tx_ring, tx_bi); 6219dbb1370SBjörn Töpel else 6229dbb1370SBjörn Töpel xsk_frames++; 6239dbb1370SBjörn Töpel 6249dbb1370SBjörn Töpel tx_bi->xdpf = NULL; 6259dbb1370SBjörn Töpel 6269dbb1370SBjörn Töpel ntc++; 6279dbb1370SBjörn Töpel if (ntc >= tx_ring->count) 6289dbb1370SBjörn Töpel ntc = 0; 6299dbb1370SBjörn Töpel } 6309dbb1370SBjörn Töpel 6319dbb1370SBjörn Töpel if (xsk_frames) 632c4655761SMagnus Karlsson xsk_tx_completed(bp, xsk_frames); 6339dbb1370SBjörn Töpel } 6343ab52af5SBjörn Töpel 6353ab52af5SBjörn Töpel /** 6361742b3d5SMagnus Karlsson * i40e_xsk_any_rx_ring_enabled - Checks if Rx rings have an AF_XDP 6371742b3d5SMagnus Karlsson * buffer pool attached 6383ab52af5SBjörn Töpel * @vsi: vsi 6393ab52af5SBjörn Töpel * 6401742b3d5SMagnus Karlsson * Returns true if any of the Rx rings has an AF_XDP buffer pool attached 6413ab52af5SBjörn Töpel **/ 6423ab52af5SBjörn Töpel bool i40e_xsk_any_rx_ring_enabled(struct i40e_vsi *vsi) 6433ab52af5SBjörn Töpel { 644f3fef2b6SJan Sokolowski struct net_device *netdev = vsi->netdev; 6453ab52af5SBjörn Töpel int i; 6463ab52af5SBjörn Töpel 6473ab52af5SBjörn Töpel for (i = 0; i < vsi->num_queue_pairs; i++) { 648c4655761SMagnus Karlsson if (xsk_get_pool_from_qid(netdev, i)) 6493ab52af5SBjörn Töpel return true; 6503ab52af5SBjörn Töpel } 6513ab52af5SBjörn Töpel 6523ab52af5SBjörn Töpel return false; 6533ab52af5SBjörn Töpel } 654