1 files changed, 4 insertions, 2 deletions

diff --git a/tools/virtio/virtio_test.c b/tools/virtio/virtio_test.c
index da7a19558281..bdb71a26ae35 100644
--- a/tools/virtio/virtio_test.c
+++ b/tools/virtio/virtio_test.c
@@ -41,13 +41,14 @@ struct vdev_info {
         struct vhost_memory *mem;
 };
 
-void vq_notify(struct virtqueue *vq)
+bool vq_notify(struct virtqueue *vq)
 {
         struct vq_info *info = vq->priv;
         unsigned long long v = 1;
         int r;
         r = write(info->kick, &v, sizeof v);
         assert(r == sizeof v);
+        return true;
 }
 
 void vq_callback(struct virtqueue *vq)
@@ -171,7 +172,8 @@ static void run_test(struct vdev_info *dev, struct vq_info *vq,
                                                          GFP_ATOMIC);
                                 if (likely(r == 0)) {
                                         ++started;
-                                        virtqueue_kick(vq->vq);
+                                        if (unlikely(!virtqueue_kick(vq->vq))
+                                                r = -1;
                                 }
                         } else
                                 r = -1;