Whamcloud - gitweb
c8861d0808a3f2ab240713b0fe823450e9a8088b
[fs/lustre-release.git] / libcfs / include / libcfs / linux / linux-mem.h
1 /*
2  * GPL HEADER START
3  *
4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
5  *
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.
9  *
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).
15  *
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.sun.com/software/products/lustre/docs/GPLv2.pdf
19  *
20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
21  * CA 95054 USA or visit www.sun.com if you need additional information or
22  * have any questions.
23  *
24  * GPL HEADER END
25  */
26 /*
27  * Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
28  * Use is subject to license terms.
29  *
30  * Copyright (c) 2011, 2012, Intel Corporation.
31  */
32 /*
33  * This file is part of Lustre, http://www.lustre.org/
34  * Lustre is a trademark of Sun Microsystems, Inc.
35  *
36  * libcfs/include/libcfs/linux/linux-mem.h
37  *
38  * Basic library routines.
39  */
40
41 #ifndef __LIBCFS_LINUX_CFS_MEM_H__
42 #define __LIBCFS_LINUX_CFS_MEM_H__
43
44 #ifndef __LIBCFS_LIBCFS_H__
45 #error Do not #include this file directly. #include <libcfs/libcfs.h> instead
46 #endif
47
48 #ifndef __KERNEL__
49 #error This include is only for kernel use.
50 #endif
51
52 #include <linux/mm.h>
53 #include <linux/vmalloc.h>
54 #include <linux/pagemap.h>
55 #include <linux/slab.h>
56 #ifdef HAVE_MM_INLINE
57 # include <linux/mm_inline.h>
58 #endif
59
60 #define CFS_PAGE_MASK                   (~((__u64)PAGE_CACHE_SIZE-1))
61
62 #define page_index(p)       ((p)->index)
63
64
65 #define memory_pressure_get() (current->flags & PF_MEMALLOC)
66 #define memory_pressure_set() do { current->flags |= PF_MEMALLOC; } while (0)
67 #define memory_pressure_clr() do { current->flags &= ~PF_MEMALLOC; } while (0)
68
69 #if BITS_PER_LONG == 32
70 /* limit to lowmem on 32-bit systems */
71 #define NUM_CACHEPAGES \
72         min(totalram_pages, 1UL << (30 - PAGE_CACHE_SHIFT) * 3 / 4)
73 #else
74 #define NUM_CACHEPAGES totalram_pages
75 #endif
76
77 /*
78  * In Linux there is no way to determine whether current execution context is
79  * blockable.
80  */
81 #define ALLOC_ATOMIC_TRY   GFP_ATOMIC
82 /* GFP_IOFS was added in 2.6.33 kernel */
83 #ifndef GFP_IOFS
84 #define GFP_IOFS       (__GFP_IO | __GFP_FS)
85 #endif
86
87 #define DECL_MMSPACE                mm_segment_t __oldfs
88 #define MMSPACE_OPEN \
89         do { __oldfs = get_fs(); set_fs(get_ds());} while(0)
90 #define MMSPACE_CLOSE               set_fs(__oldfs)
91
92
93 extern void *cfs_cpt_malloc(struct cfs_cpt_table *cptab, int cpt,
94                             size_t nr_bytes, unsigned int flags);
95 extern void *cfs_cpt_vmalloc(struct cfs_cpt_table *cptab, int cpt,
96                              size_t nr_bytes);
97 extern struct page *cfs_page_cpt_alloc(struct cfs_cpt_table *cptab,
98                                       int cpt, unsigned int flags);
99 extern void *cfs_mem_cache_cpt_alloc(struct kmem_cache *cachep,
100                                      struct cfs_cpt_table *cptab,
101                                      int cpt, unsigned int flags);
102
103 /*
104  * Shrinker
105  */
106
107 #ifdef HAVE_SHRINK_CONTROL
108 # define SHRINKER_ARGS(sc, nr_to_scan, gfp_mask)  \
109                        struct shrinker *shrinker, \
110                        struct shrink_control *sc
111 # define shrink_param(sc, var) ((sc)->var)
112 #else
113 # ifdef HAVE_SHRINKER_WANT_SHRINK_PTR
114 #  define SHRINKER_ARGS(sc, nr_to_scan, gfp_mask)  \
115                         struct shrinker *shrinker, \
116                         int nr_to_scan, gfp_t gfp_mask
117 # else
118 #  define SHRINKER_ARGS(sc, nr_to_scan, gfp_mask)  \
119                         int nr_to_scan, gfp_t gfp_mask
120 # endif
121 # define shrink_param(sc, var) (var)
122 #endif
123
124 typedef int (*shrinker_t)(SHRINKER_ARGS(sc, nr_to_scan, gfp_mask));
125
126 static inline
127 struct shrinker *set_shrinker(int seek, shrinker_t func)
128 {
129         struct shrinker *s;
130
131         s = kzalloc(sizeof(*s), GFP_KERNEL);
132         if (s == NULL)
133                 return (NULL);
134
135         s->shrink = func;
136         s->seeks = seek;
137
138         register_shrinker(s);
139
140         return s;
141 }
142
143 static inline
144 void remove_shrinker(struct shrinker *shrinker)
145 {
146         if (shrinker == NULL)
147                 return;
148
149         unregister_shrinker(shrinker);
150         kfree(shrinker);
151 }
152
153 #endif /* __LINUX_CFS_MEM_H__ */