receive_small() use virtnet_xdp_handler().

Signed-off-by: Xuan Zhuo <[email protected]>
---
 drivers/net/virtio/main.c | 40 +++++++--------------------------------
 1 file changed, 7 insertions(+), 33 deletions(-)

diff --git a/drivers/net/virtio/main.c b/drivers/net/virtio/main.c
index 9d4b84b23ef7..d7a856bd8862 100644
--- a/drivers/net/virtio/main.c
+++ b/drivers/net/virtio/main.c
@@ -618,7 +618,6 @@ static struct sk_buff *receive_small(struct net_device *dev,
        struct page *page = virt_to_head_page(buf);
        unsigned int delta = 0;
        struct page *xdp_page;
-       int err;
        unsigned int metasize = 0;
 
        len -= vi->hdr_len;
@@ -640,7 +639,6 @@ static struct sk_buff *receive_small(struct net_device *dev,
        xdp_prog = rcu_dereference(rq->xdp_prog);
        if (xdp_prog) {
                struct virtio_net_hdr_mrg_rxbuf *hdr = buf + header_offset;
-               struct xdp_frame *xdpf;
                struct xdp_buff xdp;
                void *orig_data;
                u32 act;
@@ -673,46 +671,22 @@ static struct sk_buff *receive_small(struct net_device 
*dev,
                xdp_prepare_buff(&xdp, buf + VIRTNET_RX_PAD + vi->hdr_len,
                                 xdp_headroom, len, true);
                orig_data = xdp.data;
-               act = bpf_prog_run_xdp(xdp_prog, &xdp);
-               stats->xdp_packets++;
+
+               act = virtnet_xdp_handler(xdp_prog, &xdp, dev, xdp_xmit, stats);
 
                switch (act) {
-               case XDP_PASS:
+               case VIRTNET_XDP_RES_PASS:
                        /* Recalculate length in case bpf program changed it */
                        delta = orig_data - xdp.data;
                        len = xdp.data_end - xdp.data;
                        metasize = xdp.data - xdp.data_meta;
                        break;
-               case XDP_TX:
-                       stats->xdp_tx++;
-                       xdpf = xdp_convert_buff_to_frame(&xdp);
-                       if (unlikely(!xdpf))
-                               goto err_xdp;
-                       err = virtnet_xdp_xmit(dev, 1, &xdpf, 0);
-                       if (unlikely(!err)) {
-                               xdp_return_frame_rx_napi(xdpf);
-                       } else if (unlikely(err < 0)) {
-                               trace_xdp_exception(vi->dev, xdp_prog, act);
-                               goto err_xdp;
-                       }
-                       *xdp_xmit |= VIRTIO_XDP_TX;
-                       rcu_read_unlock();
-                       goto xdp_xmit;
-               case XDP_REDIRECT:
-                       stats->xdp_redirects++;
-                       err = xdp_do_redirect(dev, &xdp, xdp_prog);
-                       if (err)
-                               goto err_xdp;
-                       *xdp_xmit |= VIRTIO_XDP_REDIR;
+
+               case VIRTNET_XDP_RES_CONSUMED:
                        rcu_read_unlock();
                        goto xdp_xmit;
-               default:
-                       bpf_warn_invalid_xdp_action(vi->dev, xdp_prog, act);
-                       fallthrough;
-               case XDP_ABORTED:
-                       trace_xdp_exception(vi->dev, xdp_prog, act);
-                       goto err_xdp;
-               case XDP_DROP:
+
+               case VIRTNET_XDP_RES_DROP:
                        goto err_xdp;
                }
        }
-- 
2.32.0.3.g01195cf9f

_______________________________________________
Virtualization mailing list
[email protected]
https://lists.linuxfoundation.org/mailman/listinfo/virtualization

Reply via email to