This patch will check the weight and exit the loop if we exceeds the
weight. This is useful for preventing scsi kthread from hogging cpu
which is guest triggerable.

This addresses CVE-2019-3900.

Cc: Paolo Bonzini <pbonz...@redhat.com>
Cc: Stefan Hajnoczi <stefa...@redhat.com>
Fixes: 057cbf49a1f0 ("tcm_vhost: Initial merge for vhost level target fabric 
driver")
Signed-off-by: Jason Wang <jasow...@redhat.com>
---
 drivers/vhost/scsi.c | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/drivers/vhost/scsi.c b/drivers/vhost/scsi.c
index d830579..3a59f47 100644
--- a/drivers/vhost/scsi.c
+++ b/drivers/vhost/scsi.c
@@ -918,7 +918,7 @@ static void vhost_scsi_submission_work(struct work_struct 
*work)
        struct iov_iter in_iter, prot_iter, data_iter;
        u64 tag;
        u32 exp_data_len, data_direction;
-       int ret, prot_bytes;
+       int ret, prot_bytes, c = 0;
        u16 lun;
        u8 task_attr;
        bool t10_pi = vhost_has_feature(vq, VIRTIO_SCSI_F_T10_PI);
@@ -938,7 +938,7 @@ static void vhost_scsi_submission_work(struct work_struct 
*work)
 
        vhost_disable_notify(&vs->dev, vq);
 
-       for (;;) {
+       do {
                ret = vhost_scsi_get_desc(vs, vq, &vc);
                if (ret)
                        goto err;
@@ -1118,7 +1118,7 @@ static void vhost_scsi_submission_work(struct work_struct 
*work)
                        break;
                else if (ret == -EIO)
                        vhost_scsi_send_bad_target(vs, vq, vc.head, vc.out);
-       }
+       } while (likely(!vhost_exceeds_weight(vq, ++c, 0)));
 out:
        mutex_unlock(&vq->mutex);
 }
@@ -1177,7 +1177,7 @@ static void vhost_scsi_submission_work(struct work_struct 
*work)
        } v_req;
        struct vhost_scsi_ctx vc;
        size_t typ_size;
-       int ret;
+       int ret, c = 0;
 
        mutex_lock(&vq->mutex);
        /*
@@ -1191,7 +1191,7 @@ static void vhost_scsi_submission_work(struct work_struct 
*work)
 
        vhost_disable_notify(&vs->dev, vq);
 
-       for (;;) {
+       do {
                ret = vhost_scsi_get_desc(vs, vq, &vc);
                if (ret)
                        goto err;
@@ -1270,7 +1270,7 @@ static void vhost_scsi_submission_work(struct work_struct 
*work)
                        break;
                else if (ret == -EIO)
                        vhost_scsi_send_bad_target(vs, vq, vc.head, vc.out);
-       }
+       } while (likely(!vhost_exceeds_weight(vq, ++c, 0)));
 out:
        mutex_unlock(&vq->mutex);
 }
-- 
1.8.3.1

_______________________________________________
Virtualization mailing list
Virtualization@lists.linux-foundation.org
https://lists.linuxfoundation.org/mailman/listinfo/virtualization

Reply via email to