40b356
From f7d6a76475d29e0edb5456e62492117b87f4bc41 Mon Sep 17 00:00:00 2001
40b356
From: Ladi Prosek <lprosek@redhat.com>
40b356
Date: Thu, 10 Nov 2016 23:00:50 +0100
40b356
Subject: [PATCH 7/8] virtio: add virtqueue_rewind()
40b356
40b356
RH-Author: Ladi Prosek <lprosek@redhat.com>
40b356
Message-id: <1478797251-10302-1-git-send-email-lprosek@redhat.com>
40b356
Patchwork-id: 72818
40b356
O-Subject: [PATCH v2 7/6] virtio: add virtqueue_rewind()
40b356
Bugzilla: 1393484
40b356
RH-Acked-by: Paolo Bonzini <pbonzini@redhat.com>
40b356
RH-Acked-by: Michael S. Tsirkin <mst@redhat.com>
40b356
RH-Acked-by: Miroslav Rezanina <mrezanin@redhat.com>
40b356
40b356
From: Stefan Hajnoczi <stefanha@redhat.com>
40b356
40b356
virtqueue_discard() requires a VirtQueueElement but virtio-balloon does
40b356
not migrate its in-use element.  Introduce a new function that is
40b356
similar to virtqueue_discard() but doesn't require a VirtQueueElement.
40b356
40b356
This will allow virtio-balloon to access element again after migration
40b356
with the usual proviso that the guest may have modified the vring since
40b356
last time.
40b356
40b356
Cc: Michael S. Tsirkin <mst@redhat.com>
40b356
Cc: Roman Kagan <rkagan@virtuozzo.com>
40b356
Cc: Stefan Hajnoczi <stefanha@redhat.com>
40b356
Signed-off-by: Ladi Prosek <lprosek@redhat.com>
40b356
Reviewed-by: Michael S. Tsirkin <mst@redhat.com>
40b356
Signed-off-by: Michael S. Tsirkin <mst@redhat.com>
40b356
(cherry picked from commit 297a75e6c55d91db2704a3d6e4029d99c7df51fd)
40b356
Signed-off-by: Ladi Prosek <lprosek@redhat.com>
40b356
Signed-off-by: Miroslav Rezanina <mrezanin@redhat.com>
40b356
---
40b356
 hw/virtio/virtio.c         | 22 ++++++++++++++++++++++
40b356
 include/hw/virtio/virtio.h |  1 +
40b356
 2 files changed, 23 insertions(+)
40b356
40b356
diff --git a/hw/virtio/virtio.c b/hw/virtio/virtio.c
40b356
index cdb21b1..fe6b032 100644
40b356
--- a/hw/virtio/virtio.c
40b356
+++ b/hw/virtio/virtio.c
40b356
@@ -259,6 +259,28 @@ void virtqueue_discard(VirtQueue *vq, const VirtQueueElement *elem,
40b356
     virtqueue_unmap_sg(vq, elem, len);
40b356
 }
40b356
 
40b356
+/* virtqueue_rewind:
40b356
+ * @vq: The #VirtQueue
40b356
+ * @num: Number of elements to push back
40b356
+ *
40b356
+ * Pretend that elements weren't popped from the virtqueue.  The next
40b356
+ * virtqueue_pop() will refetch the oldest element.
40b356
+ *
40b356
+ * Use virtqueue_discard() instead if you have a VirtQueueElement.
40b356
+ *
40b356
+ * Returns: true on success, false if @num is greater than the number of in use
40b356
+ * elements.
40b356
+ */
40b356
+bool virtqueue_rewind(VirtQueue *vq, unsigned int num)
40b356
+{
40b356
+    if (num > vq->inuse) {
40b356
+        return false;
40b356
+    }
40b356
+    vq->last_avail_idx -= num;
40b356
+    vq->inuse -= num;
40b356
+    return true;
40b356
+}
40b356
+
40b356
 void virtqueue_fill(VirtQueue *vq, const VirtQueueElement *elem,
40b356
                     unsigned int len, unsigned int idx)
40b356
 {
40b356
diff --git a/include/hw/virtio/virtio.h b/include/hw/virtio/virtio.h
40b356
index de32425..d9bfe4c 100644
40b356
--- a/include/hw/virtio/virtio.h
40b356
+++ b/include/hw/virtio/virtio.h
40b356
@@ -167,6 +167,7 @@ void virtqueue_push(VirtQueue *vq, const VirtQueueElement *elem,
40b356
 void virtqueue_flush(VirtQueue *vq, unsigned int count);
40b356
 void virtqueue_discard(VirtQueue *vq, const VirtQueueElement *elem,
40b356
                        unsigned int len);
40b356
+bool virtqueue_rewind(VirtQueue *vq, unsigned int num);
40b356
 void virtqueue_fill(VirtQueue *vq, const VirtQueueElement *elem,
40b356
                     unsigned int len, unsigned int idx);
40b356
 
40b356
-- 
40b356
1.8.3.1
40b356