Received-SPF: pass (google.com: domain of linux-kernel-owner@vger.kernel.org designates 2620:137:e000::1:20 as permitted sender) client-ip=2620:137:e000::1:20;
Content-Type: text/plain; charset=utf-8
Content-Transfer-Encoding: quoted-printable
From:   Joel Fernandes <joel@joelfernandes.org>
Mime-Version: 1.0 (1.0)
Subject: Re: [PATCH v2 1/2] mm/vmalloc: Add a safer version of find_vm_area() for debug
Date:   Fri, 1 Sep 2023 12:41:24 -0400
Message-Id: <AF34FDF1-CAD9-43FD-B97D-F6A7439F310A@joelfernandes.org>
References: <ZPHdjvFwvtzXO/6z@pc636>
Cc:     linux-kernel@vger.kernel.org,
        Andrew Morton <akpm@linux-foundation.org>,
        Christoph Hellwig <hch@infradead.org>,
        Lorenzo Stoakes <lstoakes@gmail.com>,
        Zhen Lei <thunder.leizhen@huaweicloud.com>,
        "Paul E . McKenney" <paulmck@kernel.org>, rcu@vger.kernel.org,
        Zqiang <qiang.zhang1211@gmail.com>,
        Matthew Wilcox <willy@infradead.org>, linux-mm@kvack.org
In-Reply-To: <ZPHdjvFwvtzXO/6z@pc636>
To:     Uladzislau Rezki <urezki@gmail.com>
Precedence: bulk


> On Sep 1, 2023, at 8:48 AM, Uladzislau Rezki <urezki@gmail.com> wrote:
>=20
> =EF=BB=BFOn Fri, Sep 01, 2023 at 12:33:21AM +0000, Joel Fernandes wrote:
>>> On Fri, Sep 01, 2023 at 12:19:17AM +0000, Joel Fernandes wrote:
>>> On Thu, Aug 31, 2023 at 09:47:52PM +0200, Uladzislau Rezki wrote:
>>>> On Thu, Aug 31, 2023 at 05:18:25PM +0000, Joel Fernandes (Google) wrote=
:
>>>>> It is unsafe to dump vmalloc area information when trying to do so fro=
m
>>>>> some contexts. Add a safer trylock version of the same function to do a=

>>>>> best-effort VMA finding and use it from vmalloc_dump_obj().
>>>>>=20
>>>>> [apply test robot feedback on unused function fix.]
>>>>>=20
>>>>> Reported-by: Zhen Lei <thunder.leizhen@huaweicloud.com>
>>>>> Cc: Paul E. McKenney <paulmck@kernel.org>
>>>>> Cc: rcu@vger.kernel.org
>>>>> Cc: Zqiang <qiang.zhang1211@gmail.com>
>>>>> Reviewed-by: Matthew Wilcox (Oracle) <willy@infradead.org>
>>>>> Signed-off-by: Joel Fernandes (Google) <joel@joelfernandes.org>
>>>>> ---
>>>>> v1->v2: Apply review tags and test robot feedback.
>>>>>=20
>>>>> mm/vmalloc.c | 39 ++++++++++++++++++++++++++++++++++++++-
>>>>> 1 file changed, 38 insertions(+), 1 deletion(-)
>>>>>=20
>>>>> diff --git a/mm/vmalloc.c b/mm/vmalloc.c
>>>>> index 93cf99aba335..f09e882ae3b8 100644
>>>>> --- a/mm/vmalloc.c
>>>>> +++ b/mm/vmalloc.c
>>>>> @@ -1865,6 +1865,20 @@ struct vmap_area *find_vmap_area(unsigned long a=
ddr)
>>>>>    return va;
>>>>> }
>>>>>=20
>>>>> +#ifdef CONFIG_PRINTK
>>>>> +static struct vmap_area *find_vmap_area_trylock(unsigned long addr)
>>>>> +{
>>>>> +    struct vmap_area *va;
>>>>> +
>>>>> +    if (!spin_trylock(&vmap_area_lock))
>>>>> +        return NULL;
>>>>> +    va =3D __find_vmap_area(addr, &vmap_area_root);
>>>>> +    spin_unlock(&vmap_area_lock);
>>>>> +
>>>>> +    return va;
>>>>> +}
>>>>> +#endif
>>>>> +
>>>>> static struct vmap_area *find_unlink_vmap_area(unsigned long addr)
>>>>> {
>>>>>    struct vmap_area *va;
>>>>> @@ -2671,6 +2685,29 @@ struct vm_struct *find_vm_area(const void *addr=
)
>>>>>    return va->vm;
>>>>> }
>>>>>=20
>>>>> +/**
>>>>> + * try_to_find_vm_area - find a continuous kernel virtual area
>>>>> + * @addr:      base address
>>>>> + *
>>>>> + * This function is the same as find_vm_area() except that it is
>>>>> + * safe to call if vmap_area_lock is already held and returns NULL
>>>>> + * if it is. See comments in find_vmap_area() for other details.
>>>>> + *
>>>>> + * Return: the area descriptor on success or %NULL on failure.
>>>>> + */
>>>>> +#ifdef CONFIG_PRINTK
>>>>> +static struct vm_struct *try_to_find_vm_area(const void *addr)
>>>>> +{
>>>>> +    struct vmap_area *va;
>>>>> +
>>>>> +    va =3D find_vmap_area_trylock((unsigned long)addr);
>>>>> +    if (!va)
>>>>> +        return NULL;
>>>>> +
>>>>> +    return va->vm;
>>>>> +}
>>>>> +#endif
>>>>> +
>>>>> /**
>>>>>  * remove_vm_area - find and remove a continuous kernel virtual area
>>>>>  * @addr:        base address
>>>>> @@ -4277,7 +4314,7 @@ bool vmalloc_dump_obj(void *object)
>>>>>    struct vm_struct *vm;
>>>>>    void *objp =3D (void *)PAGE_ALIGN((unsigned long)object);
>>>>>=20
>>>>> -    vm =3D find_vm_area(objp);
>>>>> +    vm =3D try_to_find_vm_area(objp);
>>>>>    if (!vm)
>>>>>        return false;
>>>>>    pr_cont(" %u-page vmalloc region starting at %#lx allocated at %pS\=
n",
>>>=20
>>> Hi Vlad,
>>> Thanks for taking a look.
>>>=20
>>>> I am not sure if this patch makes a lot of sense. I agree, this is a
>>>> problem and it mitigates it. But it is broken in terms of once you drop=

>>>> the lock, the VA should not be accessed.
>>>=20
>>> Just to note the lockless-access issue you are referring to is not intro=
duced
>>> by this patch but is rather in the existing code. Also just to note this=
 is
>>> debug code.
>>>=20
>>>> Is that a real issue or it gets triggered due to some syntetic test cas=
e?
>>>=20
>>> It is a real issue. See 2/2.
>>>=20
>>>> If i were you, i would go with open-coded version of trylock. Because
>>>> there is only one user so far.
>>>=20
>>> Taking your open coding and locking suggestions, I came up with the belo=
w
>>> which actually results in a smaller patch. Does it look good to you?
>>>=20
>>> diff --git a/mm/vmalloc.c b/mm/vmalloc.c
>>> index 93cf99aba335..aaf6bad997a7 100644
>>=20
>> And with some trivial compiler errors fixed (sorry should have build test=
ed
>> but wanted to just share the idea earlier):
>>=20
>> diff --git a/mm/vmalloc.c b/mm/vmalloc.c
>> index 93cf99aba335..2c6a0e2ff404 100644
>> --- a/mm/vmalloc.c
>> +++ b/mm/vmalloc.c
>> @@ -4274,14 +4274,32 @@ void pcpu_free_vm_areas(struct vm_struct **vms, i=
nt nr_vms)
>> #ifdef CONFIG_PRINTK
>> bool vmalloc_dump_obj(void *object)
>> {
>> -    struct vm_struct *vm;
>>    void *objp =3D (void *)PAGE_ALIGN((unsigned long)object);
>> +    const void *caller;
>> +    struct vm_struct *vm;
>> +    struct vmap_area *va;
>> +    unsigned long addr;
>> +    unsigned int nr_pages;
>>=20
>> -    vm =3D find_vm_area(objp);
>> -    if (!vm)
>> +    if (!spin_trylock(&vmap_area_lock))
>> +        return false;
>> +    va =3D __find_vmap_area((unsigned long)objp, &vmap_area_root);
>> +    if (!va) {
>> +        spin_unlock(&vmap_area_lock);
>>        return false;
>> +    }
>> +
>> +    vm =3D va->vm;
>> +    if (!vm) {
>> +        spin_unlock(&vmap_area_lock);
>> +        return false;
>> +    }
>> +    addr =3D (unsigned long)vm->addr;
>> +    caller =3D vm->caller;
>> +    nr_pages =3D vm->nr_pages;
>> +    spin_unlock(&vmap_area_lock);
>>    pr_cont(" %u-page vmalloc region starting at %#lx allocated at %pS\n",=

>> -        vm->nr_pages, (unsigned long)vm->addr, vm->caller);
>> +        nr_pages, addr, caller);
>>    return true;
>> }
>> #endif
>>=20
> Looks good to me and thank you for fixing a locking issue :)
> I think you will re-spin and resend it one more time?

Yes. May I add your Reviewed-by tag to both patches after re-spinning as men=
tioned above?

thanks!

 - Joel

>=20
> --
> Uladzislau Rezki