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 */
77 #ifndef HAVE_BITMAP_ALLOC
78 static inline unsigned long *bitmap_alloc(unsigned int nbits, gfp_t flags)
80 return kmalloc_array(BITS_TO_LONGS(nbits), sizeof(unsigned long),
84 static inline unsigned long *bitmap_zalloc(unsigned int nbits, gfp_t flags)
86 return bitmap_alloc(nbits, flags | __GFP_ZERO);
89 static inline void bitmap_free(const unsigned long *bitmap)
93 #endif /* !HAVE_BITMAP_ALLOC */
98 # define SHRINKER_ARGS(sc, nr_to_scan, gfp_mask) \
99 struct shrinker *shrinker, \
100 struct shrink_control *sc
101 # define shrink_param(sc, var) ((sc)->var)
103 #ifdef HAVE_SHRINKER_COUNT
104 struct shrinker_var {
105 unsigned long (*count)(struct shrinker *,
106 struct shrink_control *sc);
107 unsigned long (*scan)(struct shrinker *,
108 struct shrink_control *sc);
110 # define DEF_SHRINKER_VAR(name, shrink, count_obj, scan_obj) \
111 struct shrinker_var name = { .count = count_obj, .scan = scan_obj }
113 struct shrinker_var {
114 int (*shrink)(SHRINKER_ARGS(sc, nr_to_scan, gfp_mask));
116 # define DEF_SHRINKER_VAR(name, shrinker, count, scan) \
117 struct shrinker_var name = { .shrink = shrinker }
118 # define SHRINK_STOP (~0UL)
122 struct shrinker *set_shrinker(int seek, struct shrinker_var *var)
126 s = kzalloc(sizeof(*s), GFP_KERNEL);
130 #ifdef HAVE_SHRINKER_COUNT
131 s->count_objects = var->count;
132 s->scan_objects = var->scan;
134 s->shrink = var->shrink;
138 register_shrinker(s);
144 void remove_shrinker(struct shrinker *shrinker)
146 if (shrinker == NULL)
149 unregister_shrinker(shrinker);
153 #endif /* __LINUX_CFS_MEM_H__ */