On Tue, Aug 02, 2016 at 11:35:27AM +0100, Marc Zyngier wrote:
> On 01/08/16 19:29, Christoffer Dall wrote:
> > During low memory conditions, we could be dereferencing a NULL pointer
> > when vgic_add_lpi fails to allocate memory.
> > 
> > Consider for example this call sequence:
> > 
> >   vgic_its_cmd_handle_mapi
> >       itte->irq = vgic_add_lpi(kvm, lpi_nr);
> >           update_lpi_config(kvm, itte->irq, NULL);
> >               ret = kvm_read_guest(kvm, propbase + irq->intid
> >                                                  ^^^^
> >                                                  kaboom?
> > 
> > Instead, return an error pointer from vgic_add_lpi and check the return
> > value from its single caller.
> > 
> > Signed-off-by: Christoffer Dall <christoffer.d...@linaro.org>
> > ---
> >  virt/kvm/arm/vgic/vgic-its.c | 42 
> > +++++++++++++++++++++++++++++-------------
> >  1 file changed, 29 insertions(+), 13 deletions(-)
> > 
> > diff --git a/virt/kvm/arm/vgic/vgic-its.c b/virt/kvm/arm/vgic/vgic-its.c
> > index 07411cf..3515bdb 100644
> > --- a/virt/kvm/arm/vgic/vgic-its.c
> > +++ b/virt/kvm/arm/vgic/vgic-its.c
> > @@ -51,7 +51,7 @@ static struct vgic_irq *vgic_add_lpi(struct kvm *kvm, u32 
> > intid)
> >  
> >     irq = kzalloc(sizeof(struct vgic_irq), GFP_KERNEL);
> >     if (!irq)
> > -           return NULL;
> > +           return ERR_PTR(-ENOMEM);
> >  
> >     INIT_LIST_HEAD(&irq->lpi_list);
> >     INIT_LIST_HEAD(&irq->ap_list);
> > @@ -697,24 +697,33 @@ static int vgic_its_cmd_handle_mapi(struct kvm *kvm, 
> > struct vgic_its *its,
> >     struct its_device *device;
> >     struct its_collection *collection, *new_coll = NULL;
> >     int lpi_nr;
> > -
> > -   device = find_its_device(its, device_id);
> > -   if (!device)
> > -           return E_ITS_MAPTI_UNMAPPED_DEVICE;
> > +   struct vgic_irq *irq = NULL;
> > +   int err = 0;
> >  
> >     if (its_cmd_get_command(its_cmd) == GITS_CMD_MAPTI)
> >             lpi_nr = its_cmd_get_physical_id(its_cmd);
> >     else
> >             lpi_nr = event_id;
> > +
> >     if (lpi_nr < GIC_LPI_OFFSET ||
> >         lpi_nr >= max_lpis_propbaser(kvm->arch.vgic.propbaser))
> >             return E_ITS_MAPTI_PHYSICALID_OOR;
> >  
> > +   irq = vgic_add_lpi(kvm, lpi_nr);
> > +   if (IS_ERR(irq))
> > +           return PTR_ERR(irq);
> > +
> > +   device = find_its_device(its, device_id);
> > +   if (!device) {
> > +           err = E_ITS_MAPTI_UNMAPPED_DEVICE;
> > +           goto out;
> > +   }
> > +
> >     collection = find_collection(its, coll_id);
> >     if (!collection) {
> > -           int ret = vgic_its_alloc_collection(its, &collection, coll_id);
> > -           if (ret)
> > -                   return ret;
> > +           err = vgic_its_alloc_collection(its, &collection, coll_id);
> > +           if (err)
> > +                   goto out;
> >             new_coll = collection;
> >     }
> >  
> > @@ -722,9 +731,8 @@ static int vgic_its_cmd_handle_mapi(struct kvm *kvm, 
> > struct vgic_its *its,
> >     if (!itte) {
> >             itte = kzalloc(sizeof(struct its_itte), GFP_KERNEL);
> >             if (!itte) {
> > -                   if (new_coll)
> > -                           vgic_its_free_collection(its, coll_id);
> > -                   return -ENOMEM;
> > +                   err = -ENOMEM;
> > +                   goto out;
> >             }
> >  
> >             itte->event_id  = event_id;
> > @@ -733,7 +741,8 @@ static int vgic_its_cmd_handle_mapi(struct kvm *kvm, 
> > struct vgic_its *its,
> >  
> >     itte->collection = collection;
> >     itte->lpi = lpi_nr;
> > -   itte->irq = vgic_add_lpi(kvm, lpi_nr);
> > +   vgic_get_irq_kref(irq);
> > +   itte->irq = irq;
> >     update_affinity_itte(kvm, itte);
> >  
> >     /*
> > @@ -742,8 +751,15 @@ static int vgic_its_cmd_handle_mapi(struct kvm *kvm, 
> > struct vgic_its *its,
> >      * the respective config data from memory here upon mapping the LPI.
> >      */
> >     update_lpi_config(kvm, itte->irq, NULL);
> > +   new_coll = NULL;
> > +   irq = NULL;
> >  
> > -   return 0;
> > +out:
> > +   if (new_coll)
> > +           vgic_its_free_collection(its, coll_id);
> > +   if (irq)
> > +           vgic_put_irq(kvm, irq);
> 
> Ah, it took me a moment to understand why you had the
> vgic_irq_get_kref() above. Maybe a small comment?
> 

actually, now I'm confused.

vgic_add_lpi returns a struct vgic_irq with a reference count for the
reference in the itte struct, right?

So we never get a reference for the irq->lpi_list/dist->lpi_list_head ?

In which case my code above is wrong and I should not be getting the
extra reference.  Right?



Now, this made me think of another issue.  vgic_add_lpi has a blurb in
there about racing with another vgic_add_lpi and then it returns an irq
with an additional reference.  But I don't understand how this can
happen, given that the function only has a single caller which has to
run with a mutex held???

Can two different itte's point to the same struct vgic_irq ?


Thanks,
-Christoffer
_______________________________________________
kvmarm mailing list
kvmarm@lists.cs.columbia.edu
https://lists.cs.columbia.edu/mailman/listinfo/kvmarm

Reply via email to