> -----Original Message-----
> From: Yunsheng Lin <linyunsh...@huawei.com>
> Sent: Wednesday, April 16, 2025 3:38 AM
> To: Haiyang Zhang <haiya...@microsoft.com>; linux-hyp...@vger.kernel.org;
> a...@linux-foundation.org; cor...@lwn.net; linux...@kvack.org; linux-
> d...@vger.kernel.org
> Cc: Dexuan Cui <de...@microsoft.com>; KY Srinivasan <k...@microsoft.com>;
> Paul Rosswurm <paul...@microsoft.com>; o...@aepfle.de;
> vkuzn...@redhat.com; da...@davemloft.net; wei....@kernel.org; Long Li
> <lon...@microsoft.com>; linux-ker...@vger.kernel.org
> Subject: [EXTERNAL] Re: [PATCH 1/2] mm: page_frag: Check fragsz at the
> beginning of __page_frag_alloc_align()
> 
> On 2025/4/4 5:21, Haiyang Zhang wrote:
> > Frag allocator is not designed for fragsz > PAGE_SIZE. So, check and
> return
> > the error at the beginning of __page_frag_alloc_align(), instead of
> > succeed for a few times, then fail due to not refilling the cache.
> >
> > Signed-off-by: Haiyang Zhang <haiya...@microsoft.com>
> > ---
> >  mm/page_frag_cache.c | 22 +++++++++-------------
> >  1 file changed, 9 insertions(+), 13 deletions(-)
> >
> > diff --git a/mm/page_frag_cache.c b/mm/page_frag_cache.c
> > index d2423f30577e..d6bf022087e7 100644
> > --- a/mm/page_frag_cache.c
> > +++ b/mm/page_frag_cache.c
> > @@ -98,6 +98,15 @@ void *__page_frag_alloc_align(struct page_frag_cache
> *nc,
> >     unsigned int size, offset;
> >     struct page *page;
> >
> > +   if (unlikely(fragsz > PAGE_SIZE)) {
> > +           /*
> > +            * The caller is trying to allocate a fragment
> > +            * with fragsz > PAGE_SIZE which is not supported
> > +            * by design. So we simply return NULL here.
> > +            */
> > +           return NULL;
> > +   }
> 
> The checking is done at below to avoid doing the checking for the
> likely case of cache being enough as the frag API is mostly used
> to allocate small memory.
> 
> And it seems my recent refactoring to frag API have made two
> frag API misuse more obvious if I recalled it correctly. If more
> explicit about that for all the codepath is really helpful, perhaps
> VM_BUG_ON() is an option to make it more explicit while avoiding
> the checking as much as possible.

Thanks, I will use VM_BUG_ON() instead of a checking here. Also, will
keep the checking below unchanged.

> > +
> >     if (unlikely(!encoded_page)) {
> >  refill:
> >             page = __page_frag_cache_refill(nc, gfp_mask);
> > @@ -119,19 +128,6 @@ void *__page_frag_alloc_align(struct
> page_frag_cache *nc,
> >     size = PAGE_SIZE << encoded_page_decode_order(encoded_page);
> >     offset = __ALIGN_KERNEL_MASK(nc->offset, ~align_mask);
> >     if (unlikely(offset + fragsz > size)) {
> > -           if (unlikely(fragsz > PAGE_SIZE)) {
> > -                   /*
> > -                    * The caller is trying to allocate a fragment
> > -                    * with fragsz > PAGE_SIZE but the cache isn't big
> > -                    * enough to satisfy the request, this may
> > -                    * happen in low memory conditions.
> > -                    * We don't release the cache page because
> > -                    * it could make memory pressure worse
> > -                    * so we simply return NULL here.
> > -                    */
> > -                   return NULL;
> > -           }
> > -
> >             page = encoded_page_decode_page(encoded_page);
> >
> >             if (!page_ref_sub_and_test(page, nc->pagecnt_bias))

Reply via email to