Merge tag 'for-linus-5.18-rc1' of git://git.kernel.org/pub/scm/linux/kernel/git/rw/uml
[linux-block.git] / include / linux / gfp.h
1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef __LINUX_GFP_H
3 #define __LINUX_GFP_H
4
5 #include <linux/mmdebug.h>
6 #include <linux/mmzone.h>
7 #include <linux/stddef.h>
8 #include <linux/linkage.h>
9 #include <linux/topology.h>
10
11 /* The typedef is in types.h but we want the documentation here */
12 #if 0
13 /**
14  * typedef gfp_t - Memory allocation flags.
15  *
16  * GFP flags are commonly used throughout Linux to indicate how memory
17  * should be allocated.  The GFP acronym stands for get_free_pages(),
18  * the underlying memory allocation function.  Not every GFP flag is
19  * supported by every function which may allocate memory.  Most users
20  * will want to use a plain ``GFP_KERNEL``.
21  */
22 typedef unsigned int __bitwise gfp_t;
23 #endif
24
25 struct vm_area_struct;
26
27 /*
28  * In case of changes, please don't forget to update
29  * include/trace/events/mmflags.h and tools/perf/builtin-kmem.c
30  */
31
32 /* Plain integer GFP bitmasks. Do not use this directly. */
33 #define ___GFP_DMA              0x01u
34 #define ___GFP_HIGHMEM          0x02u
35 #define ___GFP_DMA32            0x04u
36 #define ___GFP_MOVABLE          0x08u
37 #define ___GFP_RECLAIMABLE      0x10u
38 #define ___GFP_HIGH             0x20u
39 #define ___GFP_IO               0x40u
40 #define ___GFP_FS               0x80u
41 #define ___GFP_ZERO             0x100u
42 #define ___GFP_ATOMIC           0x200u
43 #define ___GFP_DIRECT_RECLAIM   0x400u
44 #define ___GFP_KSWAPD_RECLAIM   0x800u
45 #define ___GFP_WRITE            0x1000u
46 #define ___GFP_NOWARN           0x2000u
47 #define ___GFP_RETRY_MAYFAIL    0x4000u
48 #define ___GFP_NOFAIL           0x8000u
49 #define ___GFP_NORETRY          0x10000u
50 #define ___GFP_MEMALLOC         0x20000u
51 #define ___GFP_COMP             0x40000u
52 #define ___GFP_NOMEMALLOC       0x80000u
53 #define ___GFP_HARDWALL         0x100000u
54 #define ___GFP_THISNODE         0x200000u
55 #define ___GFP_ACCOUNT          0x400000u
56 #define ___GFP_ZEROTAGS         0x800000u
57 #ifdef CONFIG_KASAN_HW_TAGS
58 #define ___GFP_SKIP_ZERO                0x1000000u
59 #define ___GFP_SKIP_KASAN_UNPOISON      0x2000000u
60 #define ___GFP_SKIP_KASAN_POISON        0x4000000u
61 #else
62 #define ___GFP_SKIP_ZERO                0
63 #define ___GFP_SKIP_KASAN_UNPOISON      0
64 #define ___GFP_SKIP_KASAN_POISON        0
65 #endif
66 #ifdef CONFIG_LOCKDEP
67 #define ___GFP_NOLOCKDEP        0x8000000u
68 #else
69 #define ___GFP_NOLOCKDEP        0
70 #endif
71 /* If the above are modified, __GFP_BITS_SHIFT may need updating */
72
73 /*
74  * Physical address zone modifiers (see linux/mmzone.h - low four bits)
75  *
76  * Do not put any conditional on these. If necessary modify the definitions
77  * without the underscores and use them consistently. The definitions here may
78  * be used in bit comparisons.
79  */
80 #define __GFP_DMA       ((__force gfp_t)___GFP_DMA)
81 #define __GFP_HIGHMEM   ((__force gfp_t)___GFP_HIGHMEM)
82 #define __GFP_DMA32     ((__force gfp_t)___GFP_DMA32)
83 #define __GFP_MOVABLE   ((__force gfp_t)___GFP_MOVABLE)  /* ZONE_MOVABLE allowed */
84 #define GFP_ZONEMASK    (__GFP_DMA|__GFP_HIGHMEM|__GFP_DMA32|__GFP_MOVABLE)
85
86 /**
87  * DOC: Page mobility and placement hints
88  *
89  * Page mobility and placement hints
90  * ---------------------------------
91  *
92  * These flags provide hints about how mobile the page is. Pages with similar
93  * mobility are placed within the same pageblocks to minimise problems due
94  * to external fragmentation.
95  *
96  * %__GFP_MOVABLE (also a zone modifier) indicates that the page can be
97  * moved by page migration during memory compaction or can be reclaimed.
98  *
99  * %__GFP_RECLAIMABLE is used for slab allocations that specify
100  * SLAB_RECLAIM_ACCOUNT and whose pages can be freed via shrinkers.
101  *
102  * %__GFP_WRITE indicates the caller intends to dirty the page. Where possible,
103  * these pages will be spread between local zones to avoid all the dirty
104  * pages being in one zone (fair zone allocation policy).
105  *
106  * %__GFP_HARDWALL enforces the cpuset memory allocation policy.
107  *
108  * %__GFP_THISNODE forces the allocation to be satisfied from the requested
109  * node with no fallbacks or placement policy enforcements.
110  *
111  * %__GFP_ACCOUNT causes the allocation to be accounted to kmemcg.
112  */
113 #define __GFP_RECLAIMABLE ((__force gfp_t)___GFP_RECLAIMABLE)
114 #define __GFP_WRITE     ((__force gfp_t)___GFP_WRITE)
115 #define __GFP_HARDWALL   ((__force gfp_t)___GFP_HARDWALL)
116 #define __GFP_THISNODE  ((__force gfp_t)___GFP_THISNODE)
117 #define __GFP_ACCOUNT   ((__force gfp_t)___GFP_ACCOUNT)
118
119 /**
120  * DOC: Watermark modifiers
121  *
122  * Watermark modifiers -- controls access to emergency reserves
123  * ------------------------------------------------------------
124  *
125  * %__GFP_HIGH indicates that the caller is high-priority and that granting
126  * the request is necessary before the system can make forward progress.
127  * For example, creating an IO context to clean pages.
128  *
129  * %__GFP_ATOMIC indicates that the caller cannot reclaim or sleep and is
130  * high priority. Users are typically interrupt handlers. This may be
131  * used in conjunction with %__GFP_HIGH
132  *
133  * %__GFP_MEMALLOC allows access to all memory. This should only be used when
134  * the caller guarantees the allocation will allow more memory to be freed
135  * very shortly e.g. process exiting or swapping. Users either should
136  * be the MM or co-ordinating closely with the VM (e.g. swap over NFS).
137  * Users of this flag have to be extremely careful to not deplete the reserve
138  * completely and implement a throttling mechanism which controls the
139  * consumption of the reserve based on the amount of freed memory.
140  * Usage of a pre-allocated pool (e.g. mempool) should be always considered
141  * before using this flag.
142  *
143  * %__GFP_NOMEMALLOC is used to explicitly forbid access to emergency reserves.
144  * This takes precedence over the %__GFP_MEMALLOC flag if both are set.
145  */
146 #define __GFP_ATOMIC    ((__force gfp_t)___GFP_ATOMIC)
147 #define __GFP_HIGH      ((__force gfp_t)___GFP_HIGH)
148 #define __GFP_MEMALLOC  ((__force gfp_t)___GFP_MEMALLOC)
149 #define __GFP_NOMEMALLOC ((__force gfp_t)___GFP_NOMEMALLOC)
150
151 /**
152  * DOC: Reclaim modifiers
153  *
154  * Reclaim modifiers
155  * -----------------
156  * Please note that all the following flags are only applicable to sleepable
157  * allocations (e.g. %GFP_NOWAIT and %GFP_ATOMIC will ignore them).
158  *
159  * %__GFP_IO can start physical IO.
160  *
161  * %__GFP_FS can call down to the low-level FS. Clearing the flag avoids the
162  * allocator recursing into the filesystem which might already be holding
163  * locks.
164  *
165  * %__GFP_DIRECT_RECLAIM indicates that the caller may enter direct reclaim.
166  * This flag can be cleared to avoid unnecessary delays when a fallback
167  * option is available.
168  *
169  * %__GFP_KSWAPD_RECLAIM indicates that the caller wants to wake kswapd when
170  * the low watermark is reached and have it reclaim pages until the high
171  * watermark is reached. A caller may wish to clear this flag when fallback
172  * options are available and the reclaim is likely to disrupt the system. The
173  * canonical example is THP allocation where a fallback is cheap but
174  * reclaim/compaction may cause indirect stalls.
175  *
176  * %__GFP_RECLAIM is shorthand to allow/forbid both direct and kswapd reclaim.
177  *
178  * The default allocator behavior depends on the request size. We have a concept
179  * of so called costly allocations (with order > %PAGE_ALLOC_COSTLY_ORDER).
180  * !costly allocations are too essential to fail so they are implicitly
181  * non-failing by default (with some exceptions like OOM victims might fail so
182  * the caller still has to check for failures) while costly requests try to be
183  * not disruptive and back off even without invoking the OOM killer.
184  * The following three modifiers might be used to override some of these
185  * implicit rules
186  *
187  * %__GFP_NORETRY: The VM implementation will try only very lightweight
188  * memory direct reclaim to get some memory under memory pressure (thus
189  * it can sleep). It will avoid disruptive actions like OOM killer. The
190  * caller must handle the failure which is quite likely to happen under
191  * heavy memory pressure. The flag is suitable when failure can easily be
192  * handled at small cost, such as reduced throughput
193  *
194  * %__GFP_RETRY_MAYFAIL: The VM implementation will retry memory reclaim
195  * procedures that have previously failed if there is some indication
196  * that progress has been made else where.  It can wait for other
197  * tasks to attempt high level approaches to freeing memory such as
198  * compaction (which removes fragmentation) and page-out.
199  * There is still a definite limit to the number of retries, but it is
200  * a larger limit than with %__GFP_NORETRY.
201  * Allocations with this flag may fail, but only when there is
202  * genuinely little unused memory. While these allocations do not
203  * directly trigger the OOM killer, their failure indicates that
204  * the system is likely to need to use the OOM killer soon.  The
205  * caller must handle failure, but can reasonably do so by failing
206  * a higher-level request, or completing it only in a much less
207  * efficient manner.
208  * If the allocation does fail, and the caller is in a position to
209  * free some non-essential memory, doing so could benefit the system
210  * as a whole.
211  *
212  * %__GFP_NOFAIL: The VM implementation _must_ retry infinitely: the caller
213  * cannot handle allocation failures. The allocation could block
214  * indefinitely but will never return with failure. Testing for
215  * failure is pointless.
216  * New users should be evaluated carefully (and the flag should be
217  * used only when there is no reasonable failure policy) but it is
218  * definitely preferable to use the flag rather than opencode endless
219  * loop around allocator.
220  * Using this flag for costly allocations is _highly_ discouraged.
221  */
222 #define __GFP_IO        ((__force gfp_t)___GFP_IO)
223 #define __GFP_FS        ((__force gfp_t)___GFP_FS)
224 #define __GFP_DIRECT_RECLAIM    ((__force gfp_t)___GFP_DIRECT_RECLAIM) /* Caller can reclaim */
225 #define __GFP_KSWAPD_RECLAIM    ((__force gfp_t)___GFP_KSWAPD_RECLAIM) /* kswapd can wake */
226 #define __GFP_RECLAIM ((__force gfp_t)(___GFP_DIRECT_RECLAIM|___GFP_KSWAPD_RECLAIM))
227 #define __GFP_RETRY_MAYFAIL     ((__force gfp_t)___GFP_RETRY_MAYFAIL)
228 #define __GFP_NOFAIL    ((__force gfp_t)___GFP_NOFAIL)
229 #define __GFP_NORETRY   ((__force gfp_t)___GFP_NORETRY)
230
231 /**
232  * DOC: Action modifiers
233  *
234  * Action modifiers
235  * ----------------
236  *
237  * %__GFP_NOWARN suppresses allocation failure reports.
238  *
239  * %__GFP_COMP address compound page metadata.
240  *
241  * %__GFP_ZERO returns a zeroed page on success.
242  *
243  * %__GFP_ZEROTAGS zeroes memory tags at allocation time if the memory itself
244  * is being zeroed (either via __GFP_ZERO or via init_on_alloc, provided that
245  * __GFP_SKIP_ZERO is not set). This flag is intended for optimization: setting
246  * memory tags at the same time as zeroing memory has minimal additional
247  * performace impact.
248  *
249  * %__GFP_SKIP_KASAN_UNPOISON makes KASAN skip unpoisoning on page allocation.
250  * Only effective in HW_TAGS mode.
251  *
252  * %__GFP_SKIP_KASAN_POISON makes KASAN skip poisoning on page deallocation.
253  * Typically, used for userspace pages. Only effective in HW_TAGS mode.
254  */
255 #define __GFP_NOWARN    ((__force gfp_t)___GFP_NOWARN)
256 #define __GFP_COMP      ((__force gfp_t)___GFP_COMP)
257 #define __GFP_ZERO      ((__force gfp_t)___GFP_ZERO)
258 #define __GFP_ZEROTAGS  ((__force gfp_t)___GFP_ZEROTAGS)
259 #define __GFP_SKIP_ZERO ((__force gfp_t)___GFP_SKIP_ZERO)
260 #define __GFP_SKIP_KASAN_UNPOISON ((__force gfp_t)___GFP_SKIP_KASAN_UNPOISON)
261 #define __GFP_SKIP_KASAN_POISON   ((__force gfp_t)___GFP_SKIP_KASAN_POISON)
262
263 /* Disable lockdep for GFP context tracking */
264 #define __GFP_NOLOCKDEP ((__force gfp_t)___GFP_NOLOCKDEP)
265
266 /* Room for N __GFP_FOO bits */
267 #define __GFP_BITS_SHIFT (24 +                                          \
268                           3 * IS_ENABLED(CONFIG_KASAN_HW_TAGS) +        \
269                           IS_ENABLED(CONFIG_LOCKDEP))
270 #define __GFP_BITS_MASK ((__force gfp_t)((1 << __GFP_BITS_SHIFT) - 1))
271
272 /**
273  * DOC: Useful GFP flag combinations
274  *
275  * Useful GFP flag combinations
276  * ----------------------------
277  *
278  * Useful GFP flag combinations that are commonly used. It is recommended
279  * that subsystems start with one of these combinations and then set/clear
280  * %__GFP_FOO flags as necessary.
281  *
282  * %GFP_ATOMIC users can not sleep and need the allocation to succeed. A lower
283  * watermark is applied to allow access to "atomic reserves".
284  * The current implementation doesn't support NMI and few other strict
285  * non-preemptive contexts (e.g. raw_spin_lock). The same applies to %GFP_NOWAIT.
286  *
287  * %GFP_KERNEL is typical for kernel-internal allocations. The caller requires
288  * %ZONE_NORMAL or a lower zone for direct access but can direct reclaim.
289  *
290  * %GFP_KERNEL_ACCOUNT is the same as GFP_KERNEL, except the allocation is
291  * accounted to kmemcg.
292  *
293  * %GFP_NOWAIT is for kernel allocations that should not stall for direct
294  * reclaim, start physical IO or use any filesystem callback.
295  *
296  * %GFP_NOIO will use direct reclaim to discard clean pages or slab pages
297  * that do not require the starting of any physical IO.
298  * Please try to avoid using this flag directly and instead use
299  * memalloc_noio_{save,restore} to mark the whole scope which cannot
300  * perform any IO with a short explanation why. All allocation requests
301  * will inherit GFP_NOIO implicitly.
302  *
303  * %GFP_NOFS will use direct reclaim but will not use any filesystem interfaces.
304  * Please try to avoid using this flag directly and instead use
305  * memalloc_nofs_{save,restore} to mark the whole scope which cannot/shouldn't
306  * recurse into the FS layer with a short explanation why. All allocation
307  * requests will inherit GFP_NOFS implicitly.
308  *
309  * %GFP_USER is for userspace allocations that also need to be directly
310  * accessibly by the kernel or hardware. It is typically used by hardware
311  * for buffers that are mapped to userspace (e.g. graphics) that hardware
312  * still must DMA to. cpuset limits are enforced for these allocations.
313  *
314  * %GFP_DMA exists for historical reasons and should be avoided where possible.
315  * The flags indicates that the caller requires that the lowest zone be
316  * used (%ZONE_DMA or 16M on x86-64). Ideally, this would be removed but
317  * it would require careful auditing as some users really require it and
318  * others use the flag to avoid lowmem reserves in %ZONE_DMA and treat the
319  * lowest zone as a type of emergency reserve.
320  *
321  * %GFP_DMA32 is similar to %GFP_DMA except that the caller requires a 32-bit
322  * address. Note that kmalloc(..., GFP_DMA32) does not return DMA32 memory
323  * because the DMA32 kmalloc cache array is not implemented.
324  * (Reason: there is no such user in kernel).
325  *
326  * %GFP_HIGHUSER is for userspace allocations that may be mapped to userspace,
327  * do not need to be directly accessible by the kernel but that cannot
328  * move once in use. An example may be a hardware allocation that maps
329  * data directly into userspace but has no addressing limitations.
330  *
331  * %GFP_HIGHUSER_MOVABLE is for userspace allocations that the kernel does not
332  * need direct access to but can use kmap() when access is required. They
333  * are expected to be movable via page reclaim or page migration. Typically,
334  * pages on the LRU would also be allocated with %GFP_HIGHUSER_MOVABLE.
335  *
336  * %GFP_TRANSHUGE and %GFP_TRANSHUGE_LIGHT are used for THP allocations. They
337  * are compound allocations that will generally fail quickly if memory is not
338  * available and will not wake kswapd/kcompactd on failure. The _LIGHT
339  * version does not attempt reclaim/compaction at all and is by default used
340  * in page fault path, while the non-light is used by khugepaged.
341  */
342 #define GFP_ATOMIC      (__GFP_HIGH|__GFP_ATOMIC|__GFP_KSWAPD_RECLAIM)
343 #define GFP_KERNEL      (__GFP_RECLAIM | __GFP_IO | __GFP_FS)
344 #define GFP_KERNEL_ACCOUNT (GFP_KERNEL | __GFP_ACCOUNT)
345 #define GFP_NOWAIT      (__GFP_KSWAPD_RECLAIM)
346 #define GFP_NOIO        (__GFP_RECLAIM)
347 #define GFP_NOFS        (__GFP_RECLAIM | __GFP_IO)
348 #define GFP_USER        (__GFP_RECLAIM | __GFP_IO | __GFP_FS | __GFP_HARDWALL)
349 #define GFP_DMA         __GFP_DMA
350 #define GFP_DMA32       __GFP_DMA32
351 #define GFP_HIGHUSER    (GFP_USER | __GFP_HIGHMEM)
352 #define GFP_HIGHUSER_MOVABLE    (GFP_HIGHUSER | __GFP_MOVABLE | \
353                          __GFP_SKIP_KASAN_POISON)
354 #define GFP_TRANSHUGE_LIGHT     ((GFP_HIGHUSER_MOVABLE | __GFP_COMP | \
355                          __GFP_NOMEMALLOC | __GFP_NOWARN) & ~__GFP_RECLAIM)
356 #define GFP_TRANSHUGE   (GFP_TRANSHUGE_LIGHT | __GFP_DIRECT_RECLAIM)
357
358 /* Convert GFP flags to their corresponding migrate type */
359 #define GFP_MOVABLE_MASK (__GFP_RECLAIMABLE|__GFP_MOVABLE)
360 #define GFP_MOVABLE_SHIFT 3
361
362 static inline int gfp_migratetype(const gfp_t gfp_flags)
363 {
364         VM_WARN_ON((gfp_flags & GFP_MOVABLE_MASK) == GFP_MOVABLE_MASK);
365         BUILD_BUG_ON((1UL << GFP_MOVABLE_SHIFT) != ___GFP_MOVABLE);
366         BUILD_BUG_ON((___GFP_MOVABLE >> GFP_MOVABLE_SHIFT) != MIGRATE_MOVABLE);
367
368         if (unlikely(page_group_by_mobility_disabled))
369                 return MIGRATE_UNMOVABLE;
370
371         /* Group based on mobility */
372         return (gfp_flags & GFP_MOVABLE_MASK) >> GFP_MOVABLE_SHIFT;
373 }
374 #undef GFP_MOVABLE_MASK
375 #undef GFP_MOVABLE_SHIFT
376
377 static inline bool gfpflags_allow_blocking(const gfp_t gfp_flags)
378 {
379         return !!(gfp_flags & __GFP_DIRECT_RECLAIM);
380 }
381
382 /**
383  * gfpflags_normal_context - is gfp_flags a normal sleepable context?
384  * @gfp_flags: gfp_flags to test
385  *
386  * Test whether @gfp_flags indicates that the allocation is from the
387  * %current context and allowed to sleep.
388  *
389  * An allocation being allowed to block doesn't mean it owns the %current
390  * context.  When direct reclaim path tries to allocate memory, the
391  * allocation context is nested inside whatever %current was doing at the
392  * time of the original allocation.  The nested allocation may be allowed
393  * to block but modifying anything %current owns can corrupt the outer
394  * context's expectations.
395  *
396  * %true result from this function indicates that the allocation context
397  * can sleep and use anything that's associated with %current.
398  */
399 static inline bool gfpflags_normal_context(const gfp_t gfp_flags)
400 {
401         return (gfp_flags & (__GFP_DIRECT_RECLAIM | __GFP_MEMALLOC)) ==
402                 __GFP_DIRECT_RECLAIM;
403 }
404
405 #ifdef CONFIG_HIGHMEM
406 #define OPT_ZONE_HIGHMEM ZONE_HIGHMEM
407 #else
408 #define OPT_ZONE_HIGHMEM ZONE_NORMAL
409 #endif
410
411 #ifdef CONFIG_ZONE_DMA
412 #define OPT_ZONE_DMA ZONE_DMA
413 #else
414 #define OPT_ZONE_DMA ZONE_NORMAL
415 #endif
416
417 #ifdef CONFIG_ZONE_DMA32
418 #define OPT_ZONE_DMA32 ZONE_DMA32
419 #else
420 #define OPT_ZONE_DMA32 ZONE_NORMAL
421 #endif
422
423 /*
424  * GFP_ZONE_TABLE is a word size bitstring that is used for looking up the
425  * zone to use given the lowest 4 bits of gfp_t. Entries are GFP_ZONES_SHIFT
426  * bits long and there are 16 of them to cover all possible combinations of
427  * __GFP_DMA, __GFP_DMA32, __GFP_MOVABLE and __GFP_HIGHMEM.
428  *
429  * The zone fallback order is MOVABLE=>HIGHMEM=>NORMAL=>DMA32=>DMA.
430  * But GFP_MOVABLE is not only a zone specifier but also an allocation
431  * policy. Therefore __GFP_MOVABLE plus another zone selector is valid.
432  * Only 1 bit of the lowest 3 bits (DMA,DMA32,HIGHMEM) can be set to "1".
433  *
434  *       bit       result
435  *       =================
436  *       0x0    => NORMAL
437  *       0x1    => DMA or NORMAL
438  *       0x2    => HIGHMEM or NORMAL
439  *       0x3    => BAD (DMA+HIGHMEM)
440  *       0x4    => DMA32 or NORMAL
441  *       0x5    => BAD (DMA+DMA32)
442  *       0x6    => BAD (HIGHMEM+DMA32)
443  *       0x7    => BAD (HIGHMEM+DMA32+DMA)
444  *       0x8    => NORMAL (MOVABLE+0)
445  *       0x9    => DMA or NORMAL (MOVABLE+DMA)
446  *       0xa    => MOVABLE (Movable is valid only if HIGHMEM is set too)
447  *       0xb    => BAD (MOVABLE+HIGHMEM+DMA)
448  *       0xc    => DMA32 or NORMAL (MOVABLE+DMA32)
449  *       0xd    => BAD (MOVABLE+DMA32+DMA)
450  *       0xe    => BAD (MOVABLE+DMA32+HIGHMEM)
451  *       0xf    => BAD (MOVABLE+DMA32+HIGHMEM+DMA)
452  *
453  * GFP_ZONES_SHIFT must be <= 2 on 32 bit platforms.
454  */
455
456 #if defined(CONFIG_ZONE_DEVICE) && (MAX_NR_ZONES-1) <= 4
457 /* ZONE_DEVICE is not a valid GFP zone specifier */
458 #define GFP_ZONES_SHIFT 2
459 #else
460 #define GFP_ZONES_SHIFT ZONES_SHIFT
461 #endif
462
463 #if 16 * GFP_ZONES_SHIFT > BITS_PER_LONG
464 #error GFP_ZONES_SHIFT too large to create GFP_ZONE_TABLE integer
465 #endif
466
467 #define GFP_ZONE_TABLE ( \
468         (ZONE_NORMAL << 0 * GFP_ZONES_SHIFT)                                   \
469         | (OPT_ZONE_DMA << ___GFP_DMA * GFP_ZONES_SHIFT)                       \
470         | (OPT_ZONE_HIGHMEM << ___GFP_HIGHMEM * GFP_ZONES_SHIFT)               \
471         | (OPT_ZONE_DMA32 << ___GFP_DMA32 * GFP_ZONES_SHIFT)                   \
472         | (ZONE_NORMAL << ___GFP_MOVABLE * GFP_ZONES_SHIFT)                    \
473         | (OPT_ZONE_DMA << (___GFP_MOVABLE | ___GFP_DMA) * GFP_ZONES_SHIFT)    \
474         | (ZONE_MOVABLE << (___GFP_MOVABLE | ___GFP_HIGHMEM) * GFP_ZONES_SHIFT)\
475         | (OPT_ZONE_DMA32 << (___GFP_MOVABLE | ___GFP_DMA32) * GFP_ZONES_SHIFT)\
476 )
477
478 /*
479  * GFP_ZONE_BAD is a bitmap for all combinations of __GFP_DMA, __GFP_DMA32
480  * __GFP_HIGHMEM and __GFP_MOVABLE that are not permitted. One flag per
481  * entry starting with bit 0. Bit is set if the combination is not
482  * allowed.
483  */
484 #define GFP_ZONE_BAD ( \
485         1 << (___GFP_DMA | ___GFP_HIGHMEM)                                    \
486         | 1 << (___GFP_DMA | ___GFP_DMA32)                                    \
487         | 1 << (___GFP_DMA32 | ___GFP_HIGHMEM)                                \
488         | 1 << (___GFP_DMA | ___GFP_DMA32 | ___GFP_HIGHMEM)                   \
489         | 1 << (___GFP_MOVABLE | ___GFP_HIGHMEM | ___GFP_DMA)                 \
490         | 1 << (___GFP_MOVABLE | ___GFP_DMA32 | ___GFP_DMA)                   \
491         | 1 << (___GFP_MOVABLE | ___GFP_DMA32 | ___GFP_HIGHMEM)               \
492         | 1 << (___GFP_MOVABLE | ___GFP_DMA32 | ___GFP_DMA | ___GFP_HIGHMEM)  \
493 )
494
495 static inline enum zone_type gfp_zone(gfp_t flags)
496 {
497         enum zone_type z;
498         int bit = (__force int) (flags & GFP_ZONEMASK);
499
500         z = (GFP_ZONE_TABLE >> (bit * GFP_ZONES_SHIFT)) &
501                                          ((1 << GFP_ZONES_SHIFT) - 1);
502         VM_BUG_ON((GFP_ZONE_BAD >> bit) & 1);
503         return z;
504 }
505
506 /*
507  * There is only one page-allocator function, and two main namespaces to
508  * it. The alloc_page*() variants return 'struct page *' and as such
509  * can allocate highmem pages, the *get*page*() variants return
510  * virtual kernel addresses to the allocated page(s).
511  */
512
513 static inline int gfp_zonelist(gfp_t flags)
514 {
515 #ifdef CONFIG_NUMA
516         if (unlikely(flags & __GFP_THISNODE))
517                 return ZONELIST_NOFALLBACK;
518 #endif
519         return ZONELIST_FALLBACK;
520 }
521
522 /*
523  * We get the zone list from the current node and the gfp_mask.
524  * This zone list contains a maximum of MAX_NUMNODES*MAX_NR_ZONES zones.
525  * There are two zonelists per node, one for all zones with memory and
526  * one containing just zones from the node the zonelist belongs to.
527  *
528  * For the case of non-NUMA systems the NODE_DATA() gets optimized to
529  * &contig_page_data at compile-time.
530  */
531 static inline struct zonelist *node_zonelist(int nid, gfp_t flags)
532 {
533         return NODE_DATA(nid)->node_zonelists + gfp_zonelist(flags);
534 }
535
536 #ifndef HAVE_ARCH_FREE_PAGE
537 static inline void arch_free_page(struct page *page, int order) { }
538 #endif
539 #ifndef HAVE_ARCH_ALLOC_PAGE
540 static inline void arch_alloc_page(struct page *page, int order) { }
541 #endif
542
543 struct page *__alloc_pages(gfp_t gfp, unsigned int order, int preferred_nid,
544                 nodemask_t *nodemask);
545 struct folio *__folio_alloc(gfp_t gfp, unsigned int order, int preferred_nid,
546                 nodemask_t *nodemask);
547
548 unsigned long __alloc_pages_bulk(gfp_t gfp, int preferred_nid,
549                                 nodemask_t *nodemask, int nr_pages,
550                                 struct list_head *page_list,
551                                 struct page **page_array);
552
553 unsigned long alloc_pages_bulk_array_mempolicy(gfp_t gfp,
554                                 unsigned long nr_pages,
555                                 struct page **page_array);
556
557 /* Bulk allocate order-0 pages */
558 static inline unsigned long
559 alloc_pages_bulk_list(gfp_t gfp, unsigned long nr_pages, struct list_head *list)
560 {
561         return __alloc_pages_bulk(gfp, numa_mem_id(), NULL, nr_pages, list, NULL);
562 }
563
564 static inline unsigned long
565 alloc_pages_bulk_array(gfp_t gfp, unsigned long nr_pages, struct page **page_array)
566 {
567         return __alloc_pages_bulk(gfp, numa_mem_id(), NULL, nr_pages, NULL, page_array);
568 }
569
570 static inline unsigned long
571 alloc_pages_bulk_array_node(gfp_t gfp, int nid, unsigned long nr_pages, struct page **page_array)
572 {
573         if (nid == NUMA_NO_NODE)
574                 nid = numa_mem_id();
575
576         return __alloc_pages_bulk(gfp, nid, NULL, nr_pages, NULL, page_array);
577 }
578
579 /*
580  * Allocate pages, preferring the node given as nid. The node must be valid and
581  * online. For more general interface, see alloc_pages_node().
582  */
583 static inline struct page *
584 __alloc_pages_node(int nid, gfp_t gfp_mask, unsigned int order)
585 {
586         VM_BUG_ON(nid < 0 || nid >= MAX_NUMNODES);
587         VM_WARN_ON((gfp_mask & __GFP_THISNODE) && !node_online(nid));
588
589         return __alloc_pages(gfp_mask, order, nid, NULL);
590 }
591
592 static inline
593 struct folio *__folio_alloc_node(gfp_t gfp, unsigned int order, int nid)
594 {
595         VM_BUG_ON(nid < 0 || nid >= MAX_NUMNODES);
596         VM_WARN_ON((gfp & __GFP_THISNODE) && !node_online(nid));
597
598         return __folio_alloc(gfp, order, nid, NULL);
599 }
600
601 /*
602  * Allocate pages, preferring the node given as nid. When nid == NUMA_NO_NODE,
603  * prefer the current CPU's closest node. Otherwise node must be valid and
604  * online.
605  */
606 static inline struct page *alloc_pages_node(int nid, gfp_t gfp_mask,
607                                                 unsigned int order)
608 {
609         if (nid == NUMA_NO_NODE)
610                 nid = numa_mem_id();
611
612         return __alloc_pages_node(nid, gfp_mask, order);
613 }
614
615 #ifdef CONFIG_NUMA
616 struct page *alloc_pages(gfp_t gfp, unsigned int order);
617 struct folio *folio_alloc(gfp_t gfp, unsigned order);
618 extern struct page *alloc_pages_vma(gfp_t gfp_mask, int order,
619                         struct vm_area_struct *vma, unsigned long addr,
620                         bool hugepage);
621 #define alloc_hugepage_vma(gfp_mask, vma, addr, order) \
622         alloc_pages_vma(gfp_mask, order, vma, addr, true)
623 #else
624 static inline struct page *alloc_pages(gfp_t gfp_mask, unsigned int order)
625 {
626         return alloc_pages_node(numa_node_id(), gfp_mask, order);
627 }
628 static inline struct folio *folio_alloc(gfp_t gfp, unsigned int order)
629 {
630         return __folio_alloc_node(gfp, order, numa_node_id());
631 }
632 #define alloc_pages_vma(gfp_mask, order, vma, addr, false)\
633         alloc_pages(gfp_mask, order)
634 #define alloc_hugepage_vma(gfp_mask, vma, addr, order) \
635         alloc_pages(gfp_mask, order)
636 #endif
637 #define alloc_page(gfp_mask) alloc_pages(gfp_mask, 0)
638 #define alloc_page_vma(gfp_mask, vma, addr)                     \
639         alloc_pages_vma(gfp_mask, 0, vma, addr, false)
640
641 extern unsigned long __get_free_pages(gfp_t gfp_mask, unsigned int order);
642 extern unsigned long get_zeroed_page(gfp_t gfp_mask);
643
644 void *alloc_pages_exact(size_t size, gfp_t gfp_mask) __alloc_size(1);
645 void free_pages_exact(void *virt, size_t size);
646 __meminit void *alloc_pages_exact_nid(int nid, size_t size, gfp_t gfp_mask) __alloc_size(2);
647
648 #define __get_free_page(gfp_mask) \
649                 __get_free_pages((gfp_mask), 0)
650
651 #define __get_dma_pages(gfp_mask, order) \
652                 __get_free_pages((gfp_mask) | GFP_DMA, (order))
653
654 extern void __free_pages(struct page *page, unsigned int order);
655 extern void free_pages(unsigned long addr, unsigned int order);
656
657 struct page_frag_cache;
658 extern void __page_frag_cache_drain(struct page *page, unsigned int count);
659 extern void *page_frag_alloc_align(struct page_frag_cache *nc,
660                                    unsigned int fragsz, gfp_t gfp_mask,
661                                    unsigned int align_mask);
662
663 static inline void *page_frag_alloc(struct page_frag_cache *nc,
664                              unsigned int fragsz, gfp_t gfp_mask)
665 {
666         return page_frag_alloc_align(nc, fragsz, gfp_mask, ~0u);
667 }
668
669 extern void page_frag_free(void *addr);
670
671 #define __free_page(page) __free_pages((page), 0)
672 #define free_page(addr) free_pages((addr), 0)
673
674 void page_alloc_init(void);
675 void drain_zone_pages(struct zone *zone, struct per_cpu_pages *pcp);
676 void drain_all_pages(struct zone *zone);
677 void drain_local_pages(struct zone *zone);
678
679 void page_alloc_init_late(void);
680
681 /*
682  * gfp_allowed_mask is set to GFP_BOOT_MASK during early boot to restrict what
683  * GFP flags are used before interrupts are enabled. Once interrupts are
684  * enabled, it is set to __GFP_BITS_MASK while the system is running. During
685  * hibernation, it is used by PM to avoid I/O during memory allocation while
686  * devices are suspended.
687  */
688 extern gfp_t gfp_allowed_mask;
689
690 /* Returns true if the gfp_mask allows use of ALLOC_NO_WATERMARK */
691 bool gfp_pfmemalloc_allowed(gfp_t gfp_mask);
692
693 extern void pm_restrict_gfp_mask(void);
694 extern void pm_restore_gfp_mask(void);
695
696 extern gfp_t vma_thp_gfp_mask(struct vm_area_struct *vma);
697
698 #ifdef CONFIG_PM_SLEEP
699 extern bool pm_suspended_storage(void);
700 #else
701 static inline bool pm_suspended_storage(void)
702 {
703         return false;
704 }
705 #endif /* CONFIG_PM_SLEEP */
706
707 #ifdef CONFIG_CONTIG_ALLOC
708 /* The below functions must be run on a range from a single zone. */
709 extern int alloc_contig_range(unsigned long start, unsigned long end,
710                               unsigned migratetype, gfp_t gfp_mask);
711 extern struct page *alloc_contig_pages(unsigned long nr_pages, gfp_t gfp_mask,
712                                        int nid, nodemask_t *nodemask);
713 #endif
714 void free_contig_range(unsigned long pfn, unsigned long nr_pages);
715
716 #ifdef CONFIG_CMA
717 /* CMA stuff */
718 extern void init_cma_reserved_pageblock(struct page *page);
719 #endif
720
721 #endif /* __LINUX_GFP_H */