Received: by 2002:a05:6358:701b:b0:131:369:b2a3 with SMTP id 27csp4210122rwo; Tue, 25 Jul 2023 02:22:10 -0700 (PDT) X-Google-Smtp-Source: APBJJlHBmMx2rbhYVIvG+3H0+nn4PsH+/VZnQvj7uBZN+Ta08Ps/LIjO+PHDtmn5CpLSJyfMqXET X-Received: by 2002:a17:906:2da:b0:994:54af:e282 with SMTP id 26-20020a17090602da00b0099454afe282mr11689946ejk.10.1690276930492; Tue, 25 Jul 2023 02:22:10 -0700 (PDT) ARC-Seal: i=1; a=rsa-sha256; t=1690276930; cv=none; d=google.com; s=arc-20160816; b=UzpQPt8BzJsDNfZMIE+sBJY7wDhd7CUEKTAGEO0W4BFNXqEq/2rSqqoqvXe8tm7E6O 6D8d1JWlqDrBaSuPFAj34Fl8z24vY0JMeni3v1ejcuRCbT2gFEJUseGiBEMKZJkIX4V/ orqVDTbJppEq4Ds2rWLqXm28uiTCitMieLeyUQdDS4RKBAJX8nesfzJNfTRKzD/le+Hc UVmOaUh5uOwxwDTvSdPWzTKusqsh4YO+Zwh6u9MzybJK78b2pNjcQLtH9gnL/xPXzyU3 Ny7rQQQFmzviC8Xp7P4YITBb0LFp+nOcO6CpL1l9PU/NUbLqdDyftCVW4/P8hDTsD1fE S0Zg== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=list-id:precedence:content-transfer-encoding:in-reply-to:from :references:cc:to:subject:mime-version:date:dkim-signature :message-id; bh=jkjBh/Kp2KZsbd5A9bxpjAJZxN02Eft+lFTspqPnOCk=; fh=2aXCQdnPCRGWbb3+3TWBb4ivLG2Pbs4VxrYTn6yMgTk=; b=Gy5nTdhyGPZ3X2i7WklvE6ZoFHdEWgAkXGUmggqbcYMiS8umEe6tWLyvliM7kd9wgh 3W5E0yib/mk8Us8uqqM2A/FqN6EztQy+QbIpBv4mbpGxRvJ+yJE7+TYopH4N21HWKZNh AuUMYv8IdNeyWbhS41w6H13GFSYKbLl1D2z1T3lfFdNWOymOopIYfc4b9uIqdnpjcNOU /ITO3/rWhAVj0LGllPOCb+Zoj8bCOvmBuxlUIHnJ0pGWXjOeT+9Vt0LlYAcPr9lF24u1 /eJRcUOk5Da80QNiv9wpZw6j95Ovny7ZOirjNwb31DTafD4j1UtmyxY/8EnKvZKTRj9f D1qQ== ARC-Authentication-Results: i=1; mx.google.com; dkim=pass header.i=@linux.dev header.s=key1 header.b=C86+NnAg; spf=pass (google.com: domain of linux-ext4-owner@vger.kernel.org designates 2620:137:e000::1:20 as permitted sender) smtp.mailfrom=linux-ext4-owner@vger.kernel.org; dmarc=pass (p=NONE sp=NONE dis=NONE) header.from=linux.dev Return-Path: Received: from out1.vger.email (out1.vger.email. [2620:137:e000::1:20]) by mx.google.com with ESMTP id h11-20020a17090634cb00b00988e62b24ecsi7031186ejb.760.2023.07.25.02.21.43; Tue, 25 Jul 2023 02:22:10 -0700 (PDT) Received-SPF: pass (google.com: domain of linux-ext4-owner@vger.kernel.org designates 2620:137:e000::1:20 as permitted sender) client-ip=2620:137:e000::1:20; Authentication-Results: mx.google.com; dkim=pass header.i=@linux.dev header.s=key1 header.b=C86+NnAg; spf=pass (google.com: domain of linux-ext4-owner@vger.kernel.org designates 2620:137:e000::1:20 as permitted sender) smtp.mailfrom=linux-ext4-owner@vger.kernel.org; dmarc=pass (p=NONE sp=NONE dis=NONE) header.from=linux.dev Received: (majordomo@vger.kernel.org) by vger.kernel.org via listexpand id S232930AbjGYJDh (ORCPT + 99 others); Tue, 25 Jul 2023 05:03:37 -0400 Received: from lindbergh.monkeyblade.net ([23.128.96.19]:46192 "EHLO lindbergh.monkeyblade.net" rhost-flags-OK-OK-OK-OK) by vger.kernel.org with ESMTP id S233220AbjGYJDY (ORCPT ); Tue, 25 Jul 2023 05:03:24 -0400 Received: from out-23.mta1.migadu.com (out-23.mta1.migadu.com [IPv6:2001:41d0:203:375::17]) by lindbergh.monkeyblade.net (Postfix) with ESMTPS id 9DE582109 for ; Tue, 25 Jul 2023 02:03:08 -0700 (PDT) Message-ID: <3648ca69-d65e-8431-135a-a5738586bc25@linux.dev> DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=linux.dev; s=key1; t=1690275782; h=from:from:reply-to:subject:subject:date:date:message-id:message-id: to:to:cc:cc:mime-version:mime-version:content-type:content-type: content-transfer-encoding:content-transfer-encoding: in-reply-to:in-reply-to:references:references; bh=jkjBh/Kp2KZsbd5A9bxpjAJZxN02Eft+lFTspqPnOCk=; b=C86+NnAgVOWWduY0bypLj9pEd8+f7K2hQWQRdMiOl+4lNoex/sNqQF/OzRY+XF6R/pQj3x RJfrwmBw/U62Xz7gGIQfDrsff33dmTRFy6JUY2ZoASoV8vvmjWXkOHfSOC62P8ONIDS3dq 14Z/qjc5TC07VTaeF50qDKBsNsdzhz4= Date: Tue, 25 Jul 2023 17:02:47 +0800 MIME-Version: 1.0 Subject: Re: [PATCH v2 03/47] mm: shrinker: add infrastructure for dynamically allocating shrinker To: Qi Zheng Cc: linux-kernel@vger.kernel.org, linux-mm@kvack.org, x86@kernel.org, kvm@vger.kernel.org, xen-devel@lists.xenproject.org, linux-erofs@lists.ozlabs.org, linux-f2fs-devel@lists.sourceforge.net, cluster-devel@redhat.com, linux-nfs@vger.kernel.org, linux-mtd@lists.infradead.org, rcu@vger.kernel.org, netdev@vger.kernel.org, dri-devel@lists.freedesktop.org, linux-arm-msm@vger.kernel.org, dm-devel@redhat.com, linux-raid@vger.kernel.org, linux-bcache@vger.kernel.org, virtualization@lists.linux-foundation.org, linux-fsdevel@vger.kernel.org, linux-ext4@vger.kernel.org, linux-xfs@vger.kernel.org, linux-btrfs@vger.kernel.org, akpm@linux-foundation.org, david@fromorbit.com, tkhai@ya.ru, vbabka@suse.cz, roman.gushchin@linux.dev, djwong@kernel.org, brauner@kernel.org, paulmck@kernel.org, tytso@mit.edu, steven.price@arm.com, cel@kernel.org, senozhatsky@chromium.org, yujie.liu@intel.com, gregkh@linuxfoundation.org References: <20230724094354.90817-1-zhengqi.arch@bytedance.com> <20230724094354.90817-4-zhengqi.arch@bytedance.com> X-Report-Abuse: Please report any abuse attempt to abuse@migadu.com and include these headers. From: Muchun Song In-Reply-To: <20230724094354.90817-4-zhengqi.arch@bytedance.com> Content-Type: text/plain; charset=UTF-8; format=flowed Content-Transfer-Encoding: 8bit X-Migadu-Flow: FLOW_OUT X-Spam-Status: No, score=-2.1 required=5.0 tests=BAYES_00,DKIM_SIGNED, DKIM_VALID,DKIM_VALID_AU,DKIM_VALID_EF,RCVD_IN_DNSWL_BLOCKED, SPF_HELO_NONE,SPF_PASS,T_SCC_BODY_TEXT_LINE,URIBL_BLOCKED autolearn=unavailable autolearn_force=no version=3.4.6 X-Spam-Checker-Version: SpamAssassin 3.4.6 (2021-04-09) on lindbergh.monkeyblade.net Precedence: bulk List-ID: X-Mailing-List: linux-ext4@vger.kernel.org On 2023/7/24 17:43, Qi Zheng wrote: > Currently, the shrinker instances can be divided into the following three > types: > > a) global shrinker instance statically defined in the kernel, such as > workingset_shadow_shrinker. > > b) global shrinker instance statically defined in the kernel modules, such > as mmu_shrinker in x86. > > c) shrinker instance embedded in other structures. > > For case a, the memory of shrinker instance is never freed. For case b, > the memory of shrinker instance will be freed after synchronize_rcu() when > the module is unloaded. For case c, the memory of shrinker instance will > be freed along with the structure it is embedded in. > > In preparation for implementing lockless slab shrink, we need to > dynamically allocate those shrinker instances in case c, then the memory > can be dynamically freed alone by calling kfree_rcu(). > > So this commit adds the following new APIs for dynamically allocating > shrinker, and add a private_data field to struct shrinker to record and > get the original embedded structure. > > 1. shrinker_alloc() > > Used to allocate shrinker instance itself and related memory, it will > return a pointer to the shrinker instance on success and NULL on failure. > > 2. shrinker_free_non_registered() > > Used to destroy the non-registered shrinker instance. At least I don't like this name. I know you want to tell others this function only should be called when shrinker has not been registed but allocated. Maybe shrinker_free() is more simple. And and a comment to tell the users when to use it. > > 3. shrinker_register() > > Used to register the shrinker instance, which is same as the current > register_shrinker_prepared(). > > 4. shrinker_unregister() > > Used to unregister and free the shrinker instance. > > In order to simplify shrinker-related APIs and make shrinker more > independent of other kernel mechanisms, subsequent submissions will use > the above API to convert all shrinkers (including case a and b) to > dynamically allocated, and then remove all existing APIs. > > This will also have another advantage mentioned by Dave Chinner: > > ``` > The other advantage of this is that it will break all the existing > out of tree code and third party modules using the old API and will > no longer work with a kernel using lockless slab shrinkers. They > need to break (both at the source and binary levels) to stop bad > things from happening due to using uncoverted shrinkers in the new > setup. > ``` > > Signed-off-by: Qi Zheng > --- > include/linux/shrinker.h | 6 +++ > mm/shrinker.c | 113 +++++++++++++++++++++++++++++++++++++++ > 2 files changed, 119 insertions(+) > > diff --git a/include/linux/shrinker.h b/include/linux/shrinker.h > index 961cb84e51f5..296f5e163861 100644 > --- a/include/linux/shrinker.h > +++ b/include/linux/shrinker.h > @@ -70,6 +70,8 @@ struct shrinker { > int seeks; /* seeks to recreate an obj */ > unsigned flags; > > + void *private_data; > + > /* These are for internal use */ > struct list_head list; > #ifdef CONFIG_MEMCG > @@ -98,6 +100,10 @@ struct shrinker { > > unsigned long shrink_slab(gfp_t gfp_mask, int nid, struct mem_cgroup *memcg, > int priority); > +struct shrinker *shrinker_alloc(unsigned int flags, const char *fmt, ...); > +void shrinker_free_non_registered(struct shrinker *shrinker); > +void shrinker_register(struct shrinker *shrinker); > +void shrinker_unregister(struct shrinker *shrinker); > > extern int __printf(2, 3) prealloc_shrinker(struct shrinker *shrinker, > const char *fmt, ...); > diff --git a/mm/shrinker.c b/mm/shrinker.c > index 0a32ef42f2a7..d820e4cc5806 100644 > --- a/mm/shrinker.c > +++ b/mm/shrinker.c > @@ -548,6 +548,119 @@ unsigned long shrink_slab(gfp_t gfp_mask, int nid, struct mem_cgroup *memcg, > return freed; > } > > +struct shrinker *shrinker_alloc(unsigned int flags, const char *fmt, ...) > +{ > + struct shrinker *shrinker; > + unsigned int size; > + va_list __maybe_unused ap; > + int err; > + > + shrinker = kzalloc(sizeof(struct shrinker), GFP_KERNEL); > + if (!shrinker) > + return NULL; > + > +#ifdef CONFIG_SHRINKER_DEBUG > + va_start(ap, fmt); > + shrinker->name = kvasprintf_const(GFP_KERNEL, fmt, ap); > + va_end(ap); > + if (!shrinker->name) > + goto err_name; > +#endif So why not introduce another helper to handle this and declare it as a void function when !CONFIG_SHRINKER_DEBUG? Something like the following: #ifdef CONFIG_SHRINKER_DEBUG static int shrinker_debugfs_name_alloc(struct shrinker *shrinker, const char *fmt,                                        va_list vargs) {     shrinker->name = kvasprintf_const(GFP_KERNEL, fmt, vargs);     return shrinker->name ? 0 : -ENOMEM; } #else static int shrinker_debugfs_name_alloc(struct shrinker *shrinker, const char *fmt,                                        va_list vargs) {     return 0; } #endif > + shrinker->flags = flags; > + > + if (flags & SHRINKER_MEMCG_AWARE) { > + err = prealloc_memcg_shrinker(shrinker); > + if (err == -ENOSYS) > + shrinker->flags &= ~SHRINKER_MEMCG_AWARE; > + else if (err == 0) > + goto done; > + else > + goto err_flags; > + } > + > + /* > + * The nr_deferred is available on per memcg level for memcg aware > + * shrinkers, so only allocate nr_deferred in the following cases: > + * - non memcg aware shrinkers > + * - !CONFIG_MEMCG > + * - memcg is disabled by kernel command line > + */ > + size = sizeof(*shrinker->nr_deferred); > + if (flags & SHRINKER_NUMA_AWARE) > + size *= nr_node_ids; > + > + shrinker->nr_deferred = kzalloc(size, GFP_KERNEL); > + if (!shrinker->nr_deferred) > + goto err_flags; > + > +done: > + return shrinker; > + > +err_flags: > +#ifdef CONFIG_SHRINKER_DEBUG > + kfree_const(shrinker->name); > + shrinker->name = NULL; This could be shrinker_debugfs_name_free() > +err_name: > +#endif > + kfree(shrinker); > + return NULL; > +} > +EXPORT_SYMBOL(shrinker_alloc); > + > +void shrinker_free_non_registered(struct shrinker *shrinker) > +{ > +#ifdef CONFIG_SHRINKER_DEBUG > + kfree_const(shrinker->name); > + shrinker->name = NULL; This could be shrinker_debugfs_name_free() > +#endif > + if (shrinker->flags & SHRINKER_MEMCG_AWARE) { > + down_write(&shrinker_rwsem); > + unregister_memcg_shrinker(shrinker); > + up_write(&shrinker_rwsem); > + } > + > + kfree(shrinker->nr_deferred); > + shrinker->nr_deferred = NULL; > + > + kfree(shrinker); > +} > +EXPORT_SYMBOL(shrinker_free_non_registered); > + > +void shrinker_register(struct shrinker *shrinker) > +{ > + down_write(&shrinker_rwsem); > + list_add_tail(&shrinker->list, &shrinker_list); > + shrinker->flags |= SHRINKER_REGISTERED; > + shrinker_debugfs_add(shrinker); > + up_write(&shrinker_rwsem); > +} > +EXPORT_SYMBOL(shrinker_register); > + > +void shrinker_unregister(struct shrinker *shrinker) You have made all shrinkers to be dynamically allocated, so we should prevent users from allocating shrinkers statically and use this function to unregister it. It is better to add a flag like SHRINKER_ALLOCATED which is set in shrinker_alloc(), and check whether it is set in shrinker_unregister(), if not maybe a warning should be added to tell the users what happened. > +{ > + struct dentry *debugfs_entry; > + int debugfs_id; > + > + if (!shrinker || !(shrinker->flags & SHRINKER_REGISTERED)) > + return; > + > + down_write(&shrinker_rwsem); > + list_del(&shrinker->list); > + shrinker->flags &= ~SHRINKER_REGISTERED; > + if (shrinker->flags & SHRINKER_MEMCG_AWARE) > + unregister_memcg_shrinker(shrinker); > + debugfs_entry = shrinker_debugfs_detach(shrinker, &debugfs_id); In the internal of this function, you also could use shrinker_debugfs_name_free(). Thanks. > + up_write(&shrinker_rwsem); > + > + shrinker_debugfs_remove(debugfs_entry, debugfs_id); > + > + kfree(shrinker->nr_deferred); > + shrinker->nr_deferred = NULL; > + > + kfree(shrinker); > +} > +EXPORT_SYMBOL(shrinker_unregister); > + > /* > * Add a shrinker callback to be called from the vm. > */