[PATCH net-next 15/17] virtio_net: xsk: rx: free the unused xsk buffer

[Xuan Zhuo]

Since this will be called in other circumstances(freeze), we must check
whether it is xsk's buffer in this function. It cannot be judged outside
this function.

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

diff --git a/drivers/net/virtio/main.c b/drivers/net/virtio/main.c
index 35f0daaea7e5..81d390781a4f 100644
--- a/drivers/net/virtio/main.c
+++ b/drivers/net/virtio/main.c
@@ -4146,6 +4146,14 @@ void virtnet_rq_free_unused_bufs(struct virtqueue *vq)
        rq = &vi->rq[i];
 
        while ((buf = virtqueue_detach_unused_buf(vq)) != NULL) {
+               if (rq->xsk.pool) {
+                       struct xdp_buff *xdp;
+
+                       xdp = (struct xdp_buff *)buf;
+                       xsk_buff_free(xdp);
+                       continue;
+               }
+
                if (virtqueue_get_dma_premapped(rq->vq))
                        virtnet_rq_unmap(rq, buf, 0);
 
-- 
2.32.0.3.g01195cf9f