From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org X-Spam-Level: X-Spam-Status: No, score=-16.6 required=3.0 tests=BAYES_00,DKIM_INVALID, DKIM_SIGNED,HEADER_FROM_DIFFERENT_DOMAINS,INCLUDES_CR_TRAILER,INCLUDES_PATCH, MAILING_LIST_MULTI,SPF_HELO_NONE,SPF_PASS,URIBL_BLOCKED,USER_AGENT_GIT autolearn=ham autolearn_force=no version=3.4.0 Received: from mail.kernel.org (mail.kernel.org [198.145.29.99]) by smtp.lore.kernel.org (Postfix) with ESMTP id 8C314C433E3 for ; Wed, 24 Mar 2021 15:07:52 +0000 (UTC) Received: from kanga.kvack.org (kanga.kvack.org [205.233.56.17]) by mail.kernel.org (Postfix) with ESMTP id 14CFA619A4 for ; Wed, 24 Mar 2021 15:07:52 +0000 (UTC) DMARC-Filter: OpenDMARC Filter v1.3.2 mail.kernel.org 14CFA619A4 Authentication-Results: mail.kernel.org; dmarc=none (p=none dis=none) header.from=infradead.org Authentication-Results: mail.kernel.org; spf=pass smtp.mailfrom=owner-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix) id 8AD8E6B02BC; Wed, 24 Mar 2021 11:07:51 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id 8873D6B02C4; Wed, 24 Mar 2021 11:07:51 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 74E646B02C7; Wed, 24 Mar 2021 11:07:51 -0400 (EDT) X-Delivered-To: linux-mm@kvack.org Received: from forelay.hostedemail.com (smtprelay0031.hostedemail.com [216.40.44.31]) by kanga.kvack.org (Postfix) with ESMTP id 5AAA96B02BC for ; Wed, 24 Mar 2021 11:07:51 -0400 (EDT) Received: from smtpin14.hostedemail.com (10.5.19.251.rfc1918.com [10.5.19.251]) by forelay03.hostedemail.com (Postfix) with ESMTP id A25E9824999B for ; Wed, 24 Mar 2021 15:07:50 +0000 (UTC) X-FDA: 77955097500.14.482A1C8 Received: from casper.infradead.org (casper.infradead.org [90.155.50.34]) by imf22.hostedemail.com (Postfix) with ESMTP id 073AFC001C6F for ; Wed, 24 Mar 2021 15:07:39 +0000 (UTC) DKIM-Signature: v=1; a=rsa-sha256; q=dns/txt; c=relaxed/relaxed; d=infradead.org; s=casper.20170209; h=Content-Transfer-Encoding:MIME-Version: References:In-Reply-To:Message-Id:Date:Subject:Cc:To:From:Sender:Reply-To: Content-Type:Content-ID:Content-Description; bh=1JCSW6fhPhnUwYncwpS20rorlG/js5sifoyi387v58U=; b=a12sGQgsKwfYuwbQ1gBupszG1C yAeb6nnGZRI3R+69ggOJafJWtp13VdbaAliVZsTER/1hsNxFFU1t2KDbEx5paEL3HHhRlj5KFLf/M eC7LAdaYRv8+w94Ny4nBOE2uvC4n+90QiwFjLDlm7E6a2dl68Qxdb7p0OiwJIvYFEQyx1oMeIrEa3 9wSsZtkcI5F1rPfd+RktLQwnt02h7SKU40Sxu7kaM7EK0o1X8R9r2+Gyeb22Yrx+fAJN/wHb3jmIc X+hlVmuCSKoX+WR/UqcSD+LBC0JiSuYIacUF4+U1TJny4vwITd02UstB1FQGY0UEwowuDPnbyR17G ucX0BvkA==; Received: from willy by casper.infradead.org with local (Exim 4.94 #2 (Red Hat Linux)) id 1lP55t-00BU5L-Jx; Wed, 24 Mar 2021 15:06:56 +0000 From: "Matthew Wilcox (Oracle)" To: linux-mm@kvack.org Cc: "Matthew Wilcox (Oracle)" , linux-kernel@vger.kernel.org, Andrew Morton , Christoph Lameter , Pekka Enberg , David Rientjes , Joonsoo Kim , Vlastimil Babka , Uladzislau Rezki , Nicholas Piggin Subject: [PATCH v2 2/4] mm/util: Add kvmalloc_node_caller Date: Wed, 24 Mar 2021 15:05:16 +0000 Message-Id: <20210324150518.2734402-3-willy@infradead.org> X-Mailer: git-send-email 2.29.2 In-Reply-To: <20210324150518.2734402-1-willy@infradead.org> References: <20210324150518.2734402-1-willy@infradead.org> MIME-Version: 1.0 X-Stat-Signature: zsctmb8mm3bzpzautwnnmsjh67yz8ype X-Rspamd-Server: rspam05 X-Rspamd-Queue-Id: 073AFC001C6F Received-SPF: none (infradead.org>: No applicable sender policy available) receiver=imf22; identity=mailfrom; envelope-from=""; helo=casper.infradead.org; client-ip=90.155.50.34 X-HE-DKIM-Result: pass/pass X-HE-Tag: 1616598459-437283 Content-Transfer-Encoding: quoted-printable X-Bogosity: Ham, tests=bogofilter, spamicity=0.000000, version=1.2.4 Sender: owner-linux-mm@kvack.org Precedence: bulk X-Loop: owner-majordomo@kvack.org List-ID: Allow the caller of kvmalloc to specify who counts as the allocator of the memory instead of assuming it's the immediate caller. Signed-off-by: Matthew Wilcox (Oracle) --- include/linux/mm.h | 4 +++- include/linux/slab.h | 2 ++ mm/util.c | 51 +++++++++++++++++++++++++------------------- 3 files changed, 34 insertions(+), 23 deletions(-) diff --git a/include/linux/mm.h b/include/linux/mm.h index cb1e191da319..b65a7105d9a7 100644 --- a/include/linux/mm.h +++ b/include/linux/mm.h @@ -786,7 +786,9 @@ static inline int is_vmalloc_or_module_addr(const voi= d *x) } #endif =20 -extern void *kvmalloc_node(size_t size, gfp_t flags, int node); +void *kvmalloc_node_caller(size_t size, gfp_t flags, int node, + unsigned long caller); +void *kvmalloc_node(size_t size, gfp_t flags, int node); static inline void *kvmalloc(size_t size, gfp_t flags) { return kvmalloc_node(size, flags, NUMA_NO_NODE); diff --git a/include/linux/slab.h b/include/linux/slab.h index 0c97d788762c..6611b8ee55ee 100644 --- a/include/linux/slab.h +++ b/include/linux/slab.h @@ -663,6 +663,8 @@ extern void *__kmalloc_node_track_caller(size_t, gfp_= t, int, unsigned long); =20 #else /* CONFIG_NUMA */ =20 +#define __kmalloc_node_track_caller(size, flags, node, caller) \ + __kmalloc_track_caller(size, flags, caller) #define kmalloc_node_track_caller(size, flags, node) \ kmalloc_track_caller(size, flags) =20 diff --git a/mm/util.c b/mm/util.c index 5c2304bb02ae..96dd498206ad 100644 --- a/mm/util.c +++ b/mm/util.c @@ -539,26 +539,8 @@ unsigned long vm_mmap(struct file *file, unsigned lo= ng addr, } EXPORT_SYMBOL(vm_mmap); =20 -/** - * kvmalloc_node - attempt to allocate physically contiguous memory, but= upon - * failure, fall back to non-contiguous (vmalloc) allocation. - * @size: size of the request. - * @flags: gfp mask for the allocation - must be compatible (superset) w= ith GFP_KERNEL. - * @node: numa node to allocate from - * - * Uses kmalloc to get the memory but if the allocation fails then falls= back - * to the vmalloc allocator. Use kvfree for freeing the memory. - * - * Reclaim modifiers - __GFP_NORETRY and __GFP_NOFAIL are not supported. - * __GFP_RETRY_MAYFAIL is supported, and it should be used only if kmall= oc is - * preferable to the vmalloc fallback, due to visible performance drawba= cks. - * - * Please note that any use of gfp flags outside of GFP_KERNEL is carefu= l to not - * fall back to vmalloc. - * - * Return: pointer to the allocated memory of %NULL in case of failure - */ -void *kvmalloc_node(size_t size, gfp_t flags, int node) +void *kvmalloc_node_caller(size_t size, gfp_t flags, int node, + unsigned long caller) { gfp_t kmalloc_flags =3D flags; void *ret; @@ -584,7 +566,7 @@ void *kvmalloc_node(size_t size, gfp_t flags, int nod= e) kmalloc_flags |=3D __GFP_NORETRY; } =20 - ret =3D kmalloc_node(size, kmalloc_flags, node); + ret =3D __kmalloc_node_track_caller(size, kmalloc_flags, node, caller); =20 /* * It doesn't really make sense to fallback to vmalloc for sub page @@ -593,7 +575,32 @@ void *kvmalloc_node(size_t size, gfp_t flags, int no= de) if (ret || size <=3D PAGE_SIZE) return ret; =20 - return __vmalloc_node(size, 1, flags, node, _RET_IP_); + return __vmalloc_node(size, 1, flags, node, caller); +} + +/** + * kvmalloc_node - attempt to allocate physically contiguous memory, but= upon + * failure, fall back to non-contiguous (vmalloc) allocation. + * @size: size of the request. + * @flags: GFP flags for the allocation - must be compatible (superset) = with GFP_KERNEL. + * @node: numa node to allocate from + * + * Uses kmalloc to get the memory but if the allocation fails then falls= back + * to the vmalloc allocator. Use kvfree for freeing the memory. + * + * Reclaim modifiers - __GFP_NORETRY and __GFP_NOFAIL are not supported. + * __GFP_RETRY_MAYFAIL is supported, and it should be used only if kmall= oc is + * preferable to the vmalloc fallback, due to visible performance drawba= cks. + * + * Please note that any use of gfp flags outside of GFP_KERNEL is carefu= l to not + * fall back to vmalloc. + * + * Return: pointer to the allocated memory or %NULL in case of failure. + * %ZERO_SIZE_PTR if @size is zero. + */ +void *kvmalloc_node(size_t size, gfp_t flags, int node) +{ + return kvmalloc_node_caller(size, flags, node, _RET_IP_); } EXPORT_SYMBOL(kvmalloc_node); =20 --=20 2.30.2