2 files changed, 14 insertions, 4 deletions

diff --git a/tools/virtio/linux/virtio.h b/tools/virtio/linux/virtio.h
index 5fa612ad932c..6df181a6bcc6 100644
--- a/tools/virtio/linux/virtio.h
+++ b/tools/virtio/linux/virtio.h
@@ -63,6 +63,16 @@ int virtqueue_add_sgs(struct virtqueue *vq,
                       void *data,
                       gfp_t gfp);
 
+int virtqueue_add_outbuf(struct virtqueue *vq,
+                         struct scatterlist sg[], unsigned int num,
+                         void *data,
+                         gfp_t gfp);
+
+int virtqueue_add_inbuf(struct virtqueue *vq,
+                        struct scatterlist sg[], unsigned int num,
+                        void *data,
+                        gfp_t gfp);
+
 void virtqueue_kick(struct virtqueue *vq);
 
 void *virtqueue_get_buf(struct virtqueue *vq, unsigned int *len);
diff --git a/tools/virtio/vringh_test.c b/tools/virtio/vringh_test.c
index 6a48ca5c101f..bb0bd9403e9e 100644
--- a/tools/virtio/vringh_test.c
+++ b/tools/virtio/vringh_test.c
@@ -369,11 +369,11 @@ static int parallel_test(unsigned long features,
                          * user addr */
                         __kmalloc_fake = indirects + (xfers % RINGSIZE) * 4;
                         if (output)
-                                err = virtqueue_add_buf(vq, sg, num_sg, 0, dbuf,
-                                                        GFP_KERNEL);
+                                err = virtqueue_add_outbuf(vq, sg, num_sg, dbuf,
+                                                           GFP_KERNEL);
                         else
-                                err = virtqueue_add_buf(vq, sg, 0, num_sg, dbuf,
-                                                        GFP_KERNEL);
+                                err = virtqueue_add_inbuf(vq, sg, num_sg,
+                                                          dbuf, GFP_KERNEL);
 
                         if (err == -ENOSPC) {
                                 if (!virtqueue_enable_cb_delayed(vq))