-
Notifications
You must be signed in to change notification settings - Fork 1.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
linux 6.7 compat: rework shrinker setup for heap allocations
6.7 changes the shrinker API such that shrinkers must be allocated dynamically by the kernel. To accomodate this, this commit reworks spl_register_shrinker() to do something similar against earlier kernels. Signed-off-by: Rob Norris <[email protected]> Sponsored-by: https://github.com/sponsors/robn
- Loading branch information
1 parent
18a9185
commit 03b8409
Showing
5 changed files
with
189 additions
and
56 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,115 @@ | ||
| /* | ||
| * Copyright (C) 2007-2010 Lawrence Livermore National Security, LLC. | ||
| * Copyright (C) 2007 The Regents of the University of California. | ||
| * Produced at Lawrence Livermore National Laboratory (cf, DISCLAIMER). | ||
| * Written by Brian Behlendorf <[email protected]>. | ||
| * UCRL-CODE-235197 | ||
| * | ||
| * This file is part of the SPL, Solaris Porting Layer. | ||
| * | ||
| * The SPL is free software; you can redistribute it and/or modify it | ||
| * under the terms of the GNU General Public License as published by the | ||
| * Free Software Foundation; either version 2 of the License, or (at your | ||
| * option) any later version. | ||
| * | ||
| * The SPL is distributed in the hope that it will be useful, but WITHOUT | ||
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | ||
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | ||
| * for more details. | ||
| * | ||
| * You should have received a copy of the GNU General Public License along | ||
| * with the SPL. If not, see <http://www.gnu.org/licenses/>. | ||
| * | ||
| * Solaris Porting Layer (SPL) Shrinker Implementation. | ||
| */ | ||
|
|
||
| #include <sys/kmem.h> | ||
| #include <sys/shrinker.h> | ||
|
|
||
| #ifdef HAVE_SINGLE_SHRINKER_CALLBACK | ||
| /* 3.0-3.11: single shrink() callback, which we wrap to carry both functions */ | ||
| struct spl_shrinker_wrap { | ||
| struct shrinker shrinker; | ||
| spl_shrinker_cb countfunc; | ||
| spl_shrinker_cb scanfunc; | ||
| }; | ||
|
|
||
| static int | ||
| spl_shrinker_single_cb(struct shrinker *shrinker, struct shrink_control *sc) | ||
| { | ||
| struct spl_shrinker_wrap *sw = (struct spl_shrinker_wrap *)shrinker; | ||
|
|
||
| if (sc->nr_to_scan != 0) | ||
| (void) sw->scanfunc(&sw->shrinker, sc); | ||
| return (sw->countfunc(&sw->shrinker, sc)); | ||
| } | ||
| #endif | ||
|
|
||
| struct shrinker * | ||
| spl_register_shrinker(const char *name, spl_shrinker_cb countfunc, | ||
| spl_shrinker_cb scanfunc, int seek_cost) | ||
| { | ||
| struct shrinker *shrinker; | ||
|
|
||
| /* allocate shrinker */ | ||
| #if defined(HAVE_SHRINKER_REGISTER) | ||
| /* 6.7: kernel will allocate the shrinker for us */ | ||
| shrinker = shrinker_alloc(0, name); | ||
| #elif defined(HAVE_SPLIT_SHRINKER_CALLBACK) | ||
| /* 3.12-6.6: we allocate the shrinker */ | ||
| shrinker = kmem_zalloc(sizeof (struct shrinker), KM_SLEEP); | ||
| #elif defined(HAVE_SINGLE_SHRINKER_CALLBACK) | ||
| /* 3.0-3.11: allocate a wrapper */ | ||
| struct spl_shrinker_wrap *sw = | ||
| kmem_zalloc(sizeof (struct spl_shrinker_wrap), KM_SLEEP); | ||
| shrinker = &sw->shrinker; | ||
| #else | ||
| /* 2.x-2.6.22, or a newer shrinker API has been introduced. */ | ||
| #error "Unknown shrinker API" | ||
| #endif | ||
|
|
||
| if (shrinker == NULL) | ||
| return (NULL); | ||
|
|
||
| /* set callbacks */ | ||
| #ifdef HAVE_SINGLE_SHRINKER_CALLBACK | ||
| sw->countfunc = countfunc; | ||
| sw->scanfunc = scanfunc; | ||
| shrinker->shrink = spl_shrinker_single_cb; | ||
| #else | ||
| shrinker->count_objects = countfunc; | ||
| shrinker->scan_objects = scanfunc; | ||
| #endif | ||
|
|
||
| /* set params */ | ||
| shrinker->seeks = seek_cost; | ||
|
|
||
| /* register with kernel */ | ||
| #if defined(HAVE_SHRINKER_REGISTER) | ||
| shrinker_register(shrinker); | ||
| #elif defined(HAVE_REGISTER_SHRINKER_VARARG) | ||
| register_shrinker(shrinker, name); | ||
| #else | ||
| register_shrinker(shrinker); | ||
| #endif | ||
|
|
||
| return (shrinker); | ||
| } | ||
| EXPORT_SYMBOL(spl_register_shrinker); | ||
|
|
||
| void | ||
| spl_unregister_shrinker(struct shrinker *shrinker) | ||
| { | ||
| #if defined(HAVE_SHRINKER_REGISTER) | ||
| shrinker_free(shrinker); | ||
| #elif defined(HAVE_SPLIT_SHRINKER_CALLBACK) | ||
| unregister_shrinker(shrinker); | ||
| kmem_free(shrinker, sizeof (struct shrinker)); | ||
| #elif defined(HAVE_SINGLE_SHRINKER_CALLBACK) | ||
| unregister_shrinker(shrinker); | ||
| kmem_free(shrinker, sizeof (struct spl_shrinker_wrap)); | ||
| #else | ||
| #error "Unknown shrinker API" | ||
| #endif | ||
| } | ||
| EXPORT_SYMBOL(spl_unregister_shrinker); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters