[PATCH net-next 14/17] virtio_net: xsk: rx: support recv small mode

[Xuan Zhuo]

receive the xsk buffer for small mode.

Signed-off-by: Xuan Zhuo <xuanz...@linux.alibaba.com>
---
 drivers/net/virtio/xsk.c | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/drivers/net/virtio/xsk.c b/drivers/net/virtio/xsk.c
index 6f4fa32b1184..a73559faadf6 100644
--- a/drivers/net/virtio/xsk.c
+++ b/drivers/net/virtio/xsk.c
@@ -200,6 +200,37 @@ static struct sk_buff *virtnet_receive_xsk_merge(struct 
net_device *dev, struct
        return NULL;
 }
 
+static struct sk_buff *virtnet_receive_xsk_small(struct net_device *dev, 
struct virtnet_info *vi,
+                                                struct virtnet_rq *rq, struct 
xdp_buff *xdp,
+                                                unsigned int *xdp_xmit,
+                                                struct virtnet_rq_stats *stats)
+{
+       struct bpf_prog *prog;
+       u32 ret;
+
+       ret = XDP_PASS;
+       rcu_read_lock();
+       prog = rcu_dereference(rq->xdp_prog);
+       if (prog)
+               ret = virtnet_xdp_handler(prog, xdp, dev, xdp_xmit, stats);
+       rcu_read_unlock();
+
+       switch (ret) {
+       case XDP_PASS:
+               return xdp_construct_skb(rq, xdp);
+
+       case XDP_TX:
+       case XDP_REDIRECT:
+               return NULL;
+
+       default:
+               /* drop packet */
+               xsk_buff_free(xdp);
+               u64_stats_inc(&stats->drops);
+               return NULL;
+       }
+}
+
 struct sk_buff *virtnet_receive_xsk_buf(struct virtnet_info *vi, struct 
virtnet_rq *rq,
                                        void *buf, u32 len,
                                        unsigned int *xdp_xmit,
@@ -226,6 +257,8 @@ struct sk_buff *virtnet_receive_xsk_buf(struct virtnet_info 
*vi, struct virtnet_
 
        if (vi->mergeable_rx_bufs)
                skb = virtnet_receive_xsk_merge(dev, vi, rq, xdp, xdp_xmit, 
stats);
+       else
+               skb = virtnet_receive_xsk_small(dev, vi, rq, xdp, xdp_xmit, 
stats);
 
        return skb;
 }
-- 
2.32.0.3.g01195cf9f