4 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License version 2 only,
8 * as published by the Free Software Foundation.
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License version 2 for more details (a copy is included
14 * in the LICENSE file that accompanied this code).
16 * You should have received a copy of the GNU General Public License
17 * version 2 along with this program; If not, see
18 * http://www.gnu.org/licenses/gpl-2.0.html
23 * Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
24 * Use is subject to license terms.
26 * Copyright (c) 2011, 2014, Intel Corporation.
29 * This file is part of Lustre, http://www.lustre.org/
30 * Lustre is a trademark of Sun Microsystems, Inc.
32 * libcfs/include/libcfs/linux/linux-mem.h
34 * Basic library routines.
37 #ifndef __LIBCFS_LINUX_CFS_MEM_H__
38 #define __LIBCFS_LINUX_CFS_MEM_H__
41 #include <linux/vmalloc.h>
42 #include <linux/pagemap.h>
43 #include <linux/slab.h>
45 # include <linux/mm_inline.h>
47 #include <linux/sched.h>
48 #ifdef HAVE_SCHED_HEADERS
49 #include <linux/sched/mm.h>
52 #ifdef HAVE_TOTALRAM_PAGES_AS_FUNC
53 #ifndef cfs_totalram_pages
54 #define cfs_totalram_pages() totalram_pages()
57 #ifndef cfs_totalram_pages
58 #define cfs_totalram_pages() totalram_pages
62 #ifndef HAVE_MEMALLOC_RECLAIM
63 static inline unsigned int memalloc_noreclaim_save(void)
65 unsigned int flags = current->flags & PF_MEMALLOC;
67 current->flags |= PF_MEMALLOC;
71 static inline void memalloc_noreclaim_restore(unsigned int flags)
73 current->flags = (current->flags & ~PF_MEMALLOC) | flags;
75 #endif /* !HAVE_MEMALLOC_RECLAIM */
80 # define SHRINKER_ARGS(sc, nr_to_scan, gfp_mask) \
81 struct shrinker *shrinker, \
82 struct shrink_control *sc
83 # define shrink_param(sc, var) ((sc)->var)
85 #ifdef HAVE_SHRINKER_COUNT
87 unsigned long (*count)(struct shrinker *,
88 struct shrink_control *sc);
89 unsigned long (*scan)(struct shrinker *,
90 struct shrink_control *sc);
92 # define DEF_SHRINKER_VAR(name, shrink, count_obj, scan_obj) \
93 struct shrinker_var name = { .count = count_obj, .scan = scan_obj }
96 int (*shrink)(SHRINKER_ARGS(sc, nr_to_scan, gfp_mask));
98 # define DEF_SHRINKER_VAR(name, shrinker, count, scan) \
99 struct shrinker_var name = { .shrink = shrinker }
100 # define SHRINK_STOP (~0UL)
104 struct shrinker *set_shrinker(int seek, struct shrinker_var *var)
108 s = kzalloc(sizeof(*s), GFP_KERNEL);
112 #ifdef HAVE_SHRINKER_COUNT
113 s->count_objects = var->count;
114 s->scan_objects = var->scan;
116 s->shrink = var->shrink;
120 register_shrinker(s);
126 void remove_shrinker(struct shrinker *shrinker)
128 if (shrinker == NULL)
131 unregister_shrinker(shrinker);
135 #endif /* __LINUX_CFS_MEM_H__ */