1 files changed, 9 insertions, 4 deletions

diff --git a/tools/virtio/virtio_test.c b/tools/virtio/virtio_test.c
index fcc9aa25fd08..da7a19558281 100644
--- a/tools/virtio/virtio_test.c
+++ b/tools/virtio/virtio_test.c
@@ -10,11 +10,15 @@
 #include <sys/stat.h>
 #include <sys/types.h>
 #include <fcntl.h>
+#include <stdbool.h>
 #include <linux/vhost.h>
 #include <linux/virtio.h>
 #include <linux/virtio_ring.h>
 #include "../../drivers/vhost/test.h"
 
+/* Unused */
+void *__kmalloc_fake, *__kfree_ignore_start, *__kfree_ignore_end;
+
 struct vq_info {
         int kick;
         int call;
@@ -92,7 +96,8 @@ static void vq_info_add(struct vdev_info *dev, int num)
         assert(r >= 0);
         memset(info->ring, 0, vring_size(num, 4096));
         vring_init(&info->vring, num, info->ring, 4096);
-        info->vq = vring_new_virtqueue(info->vring.num, 4096, &dev->vdev,
+        info->vq = vring_new_virtqueue(info->idx,
+                                       info->vring.num, 4096, &dev->vdev,
                                        true, info->ring,
                                        vq_notify, vq_callback, "test");
         assert(info->vq);
@@ -161,9 +166,9 @@ static void run_test(struct vdev_info *dev, struct vq_info *vq,
                 do {
                         if (started < bufs) {
                                 sg_init_one(&sl, dev->buf, dev->buf_size);
-                                r = virtqueue_add_buf(vq->vq, &sl, 1, 0,
-                                                      dev->buf + started,
-                                                      GFP_ATOMIC);
+                                r = virtqueue_add_outbuf(vq->vq, &sl, 1,
+                                                         dev->buf + started,
+                                                         GFP_ATOMIC);
                                 if (likely(r == 0)) {
                                         ++started;
                                         virtqueue_kick(vq->vq);