On 15.07.2016 09:17, Yuanhan Liu wrote:
> On Thu, Jul 14, 2016 at 11:18:39AM +0300, Ilya Maximets wrote:
>> In current implementation vhost will crash with segmentation fault
>> if malicious or buggy virtio application breaks addresses of descriptors.
>>
>> Before commit 0823c1cb0a73 this crash was reproducible even with
>> normal DPDK application that tries to change number of virtqueues
>> dynamically inside VM.
>>
>> Fix that by checking addresses of descriptors before using.
>>
>> Also fixed return value on error for 'copy_mbuf_to_desc_mergeable()'
>> from '-1' to '0' because it returns unsigned value and it means
>> number of used descriptors.
> 
> Yeah, that's a good fix. Thanks.
> 
> Maybe you'd better make it a standalone patch.

Ok. Maybe I should split this patch in two:
1. Fix return value + using of this value (vq->last_used_idx += nr_used;)
2. Check addresses of descriptors.
What do you think?

>> Signed-off-by: Ilya Maximets <i.maximets at samsung.com>
>> ---
>> Version 2:
>>      * Rebased on top of current master.
>>      * host's address now checked in meargeable case,
>>        because needed refactoring already done.
>>      * Commit-message changed because old issue with
>>        virtio reload accidentially fixed by commit
>>        0823c1cb0a73.
>>
>>  lib/librte_vhost/vhost_rxtx.c | 28 +++++++++++++++++++++-------
>>  1 file changed, 21 insertions(+), 7 deletions(-)
>>
>> diff --git a/lib/librte_vhost/vhost_rxtx.c b/lib/librte_vhost/vhost_rxtx.c
>> index 15ca956..31e8b58 100644
>> --- a/lib/librte_vhost/vhost_rxtx.c
>> +++ b/lib/librte_vhost/vhost_rxtx.c
>> @@ -147,10 +147,10 @@ copy_mbuf_to_desc(struct virtio_net *dev, struct 
>> vhost_virtqueue *vq,
>>      struct virtio_net_hdr_mrg_rxbuf virtio_hdr = {{0, 0, 0, 0, 0, 0}, 0};
>>  
>>      desc = &vq->desc[desc_idx];
>> -    if (unlikely(desc->len < dev->vhost_hlen))
>> +    desc_addr = gpa_to_vva(dev, desc->addr);
>> +    if (unlikely(desc->len < dev->vhost_hlen || !desc_addr))
>>              return -1;
> 
> So, you discards the workaround from Rich?

I can apply it, if you wish. Should I?

>>  
>> -    desc_addr = gpa_to_vva(dev, desc->addr);
>>      rte_prefetch0((void *)(uintptr_t)desc_addr);
>>  
>>      virtio_enqueue_offload(m, &virtio_hdr.hdr);
>> @@ -182,7 +182,10 @@ copy_mbuf_to_desc(struct virtio_net *dev, struct 
>> vhost_virtqueue *vq,
>>                              return -1;
>>  
>>                      desc = &vq->desc[desc->next];
>> -                    desc_addr   = gpa_to_vva(dev, desc->addr);
>> +                    desc_addr = gpa_to_vva(dev, desc->addr);
>> +                    if (unlikely(!desc_addr))
>> +                            return -1;
>> +
>>                      desc_offset = 0;
>>                      desc_avail  = desc->len;
>>              }
>> @@ -387,10 +390,10 @@ copy_mbuf_to_desc_mergeable(struct virtio_net *dev, 
>> struct vhost_virtqueue *vq,
>>      LOG_DEBUG(VHOST_DATA, "(%d) current index %d | end index %d\n",
>>              dev->vid, cur_idx, end_idx);
>>  
>> -    if (buf_vec[vec_idx].buf_len < dev->vhost_hlen)
>> -            return -1;
>> -
>>      desc_addr = gpa_to_vva(dev, buf_vec[vec_idx].buf_addr);
>> +    if (buf_vec[vec_idx].buf_len < dev->vhost_hlen || !desc_addr)
>> +            return 0;
>> +
>>      rte_prefetch0((void *)(uintptr_t)desc_addr);
>>  
>>      virtio_hdr.num_buffers = end_idx - start_idx;
>> @@ -425,6 +428,8 @@ copy_mbuf_to_desc_mergeable(struct virtio_net *dev, 
>> struct vhost_virtqueue *vq,
>>  
>>                      vec_idx++;
>>                      desc_addr = gpa_to_vva(dev, buf_vec[vec_idx].buf_addr);
>> +                    if (unlikely(!desc_addr))
>> +                            return 0;
>>  
>>                      /* Prefetch buffer address. */
>>                      rte_prefetch0((void *)(uintptr_t)desc_addr);
>> @@ -507,7 +512,7 @@ virtio_dev_merge_rx(struct virtio_net *dev, uint16_t 
>> queue_id,
>>              *(volatile uint16_t *)&vq->used->idx += nr_used;
>>              vhost_log_used_vring(dev, vq, offsetof(struct vring_used, idx),
>>                      sizeof(vq->used->idx));
>> -            vq->last_used_idx = end;
>> +            vq->last_used_idx += nr_used;
> 
> Ditto, this may deserve another patch, too.
> 
>       --yliu
> 
> 

Reply via email to