@@ -217,10 +217,14 @@ static int virtballoon_probe(struct virtio_device *vdev)
vb->vdev = vdev;
/* We expect two virtqueues. */
+ err = virtio_request_vqs(vdev, 2);
+ if (err)
+ goto out_free_vb;
+
vb->inflate_vq = vdev->config->find_vq(vdev, 0, balloon_ack);
if (IS_ERR(vb->inflate_vq)) {
err = PTR_ERR(vb->inflate_vq);
- goto out_free_vb;
+ goto out_free_vqs;
}
vb->deflate_vq = vdev->config->find_vq(vdev, 1, balloon_ack);
@@ -244,6 +248,8 @@ out_del_deflate_vq:
vdev->config->del_vq(vb->deflate_vq);
out_del_inflate_vq:
vdev->config->del_vq(vb->inflate_vq);
+out_free_vqs:
+ virtio_free_vqs(vdev);
out_free_vb:
kfree(vb);
out:
@@ -265,6 +271,7 @@ static void virtballoon_remove(struct virtio_device *vdev)
vdev->config->del_vq(vb->deflate_vq);
vdev->config->del_vq(vb->inflate_vq);
+ virtio_free_vqs(vdev);
kfree(vb);
}
Add request_vqs/free_vqs calls to virtio_balloon. These will be required for MSI support. Signed-off-by: Michael S. Tsirkin <mst@redhat.com> --- drivers/virtio/virtio_balloon.c | 9 ++++++++- 1 files changed, 8 insertions(+), 1 deletions(-)