powerpc/hugetlb: Don't do runtime allocation of 16G pages in LPAR configuration
[linux-2.6-block.git] / include / linux / memcontrol.h
CommitLineData
8cdea7c0
BS
1/* memcontrol.h - Memory Controller
2 *
3 * Copyright IBM Corporation, 2007
4 * Author Balbir Singh <balbir@linux.vnet.ibm.com>
5 *
78fb7466
PE
6 * Copyright 2007 OpenVZ SWsoft Inc
7 * Author: Pavel Emelianov <xemul@openvz.org>
8 *
8cdea7c0
BS
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
13 *
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
18 */
19
20#ifndef _LINUX_MEMCONTROL_H
21#define _LINUX_MEMCONTROL_H
f8d66542 22#include <linux/cgroup.h>
456f998e 23#include <linux/vm_event_item.h>
7ae1e1d0 24#include <linux/hardirq.h>
a8964b9b 25#include <linux/jump_label.h>
33398cf2
MH
26#include <linux/page_counter.h>
27#include <linux/vmpressure.h>
28#include <linux/eventfd.h>
00f3ca2c
JW
29#include <linux/mm.h>
30#include <linux/vmstat.h>
33398cf2 31#include <linux/writeback.h>
fdf1cdb9 32#include <linux/page-flags.h>
456f998e 33
78fb7466 34struct mem_cgroup;
8697d331
BS
35struct page;
36struct mm_struct;
2633d7a0 37struct kmem_cache;
78fb7466 38
71cd3113
JW
39/* Cgroup-specific page state, on top of universal node page state */
40enum memcg_stat_item {
41 MEMCG_CACHE = NR_VM_NODE_STAT_ITEMS,
42 MEMCG_RSS,
43 MEMCG_RSS_HUGE,
44 MEMCG_SWAP,
45 MEMCG_SOCK,
46 /* XXX: why are these zone and not node counters? */
47 MEMCG_KERNEL_STACK_KB,
b2807f07 48 MEMCG_NR_STAT,
2a7106f2
GT
49};
50
e27be240
JW
51enum memcg_memory_event {
52 MEMCG_LOW,
71cd3113
JW
53 MEMCG_HIGH,
54 MEMCG_MAX,
55 MEMCG_OOM,
fe6bdfc8 56 MEMCG_OOM_KILL,
f3a53a3a
TH
57 MEMCG_SWAP_MAX,
58 MEMCG_SWAP_FAIL,
e27be240 59 MEMCG_NR_MEMORY_EVENTS,
71cd3113
JW
60};
61
bf8d5d52
RG
62enum mem_cgroup_protection {
63 MEMCG_PROT_NONE,
64 MEMCG_PROT_LOW,
65 MEMCG_PROT_MIN,
66};
67
5660048c 68struct mem_cgroup_reclaim_cookie {
ef8f2327 69 pg_data_t *pgdat;
5660048c
JW
70 int priority;
71 unsigned int generation;
72};
73
71cd3113
JW
74#ifdef CONFIG_MEMCG
75
76#define MEM_CGROUP_ID_SHIFT 16
77#define MEM_CGROUP_ID_MAX USHRT_MAX
78
79struct mem_cgroup_id {
80 int id;
1c2d479a 81 refcount_t ref;
71cd3113
JW
82};
83
33398cf2
MH
84/*
85 * Per memcg event counter is incremented at every pagein/pageout. With THP,
86 * it will be incremated by the number of pages. This counter is used for
87 * for trigger some periodic events. This is straightforward and better
88 * than using jiffies etc. to handle periodic memcg event.
89 */
90enum mem_cgroup_events_target {
91 MEM_CGROUP_TARGET_THRESH,
92 MEM_CGROUP_TARGET_SOFTLIMIT,
93 MEM_CGROUP_TARGET_NUMAINFO,
94 MEM_CGROUP_NTARGETS,
95};
96
33398cf2 97struct mem_cgroup_stat_cpu {
b2807f07 98 long count[MEMCG_NR_STAT];
e27be240 99 unsigned long events[NR_VM_EVENT_ITEMS];
33398cf2
MH
100 unsigned long nr_page_events;
101 unsigned long targets[MEM_CGROUP_NTARGETS];
102};
103
104struct mem_cgroup_reclaim_iter {
105 struct mem_cgroup *position;
106 /* scan generation, increased every round-trip */
107 unsigned int generation;
108};
109
00f3ca2c
JW
110struct lruvec_stat {
111 long count[NR_VM_NODE_STAT_ITEMS];
112};
113
0a4465d3
KT
114/*
115 * Bitmap of shrinker::id corresponding to memcg-aware shrinkers,
116 * which have elements charged to this memcg.
117 */
118struct memcg_shrinker_map {
119 struct rcu_head rcu;
120 unsigned long map[0];
121};
122
33398cf2
MH
123/*
124 * per-zone information in memory controller.
125 */
ef8f2327 126struct mem_cgroup_per_node {
33398cf2 127 struct lruvec lruvec;
a983b5eb
JW
128
129 struct lruvec_stat __percpu *lruvec_stat_cpu;
130 atomic_long_t lruvec_stat[NR_VM_NODE_STAT_ITEMS];
131
b4536f0c 132 unsigned long lru_zone_size[MAX_NR_ZONES][NR_LRU_LISTS];
33398cf2
MH
133
134 struct mem_cgroup_reclaim_iter iter[DEF_PRIORITY + 1];
135
0a4465d3
KT
136#ifdef CONFIG_MEMCG_KMEM
137 struct memcg_shrinker_map __rcu *shrinker_map;
138#endif
33398cf2
MH
139 struct rb_node tree_node; /* RB tree node */
140 unsigned long usage_in_excess;/* Set to the value by which */
141 /* the soft limit is exceeded*/
142 bool on_tree;
e3c1ac58
AR
143 bool congested; /* memcg has many dirty pages */
144 /* backed by a congested BDI */
145
33398cf2
MH
146 struct mem_cgroup *memcg; /* Back pointer, we cannot */
147 /* use container_of */
148};
149
33398cf2
MH
150struct mem_cgroup_threshold {
151 struct eventfd_ctx *eventfd;
152 unsigned long threshold;
153};
154
155/* For threshold */
156struct mem_cgroup_threshold_ary {
157 /* An array index points to threshold just below or equal to usage. */
158 int current_threshold;
159 /* Size of entries[] */
160 unsigned int size;
161 /* Array of thresholds */
162 struct mem_cgroup_threshold entries[0];
163};
164
165struct mem_cgroup_thresholds {
166 /* Primary thresholds array */
167 struct mem_cgroup_threshold_ary *primary;
168 /*
169 * Spare threshold array.
170 * This is needed to make mem_cgroup_unregister_event() "never fail".
171 * It must be able to store at least primary->size - 1 entries.
172 */
173 struct mem_cgroup_threshold_ary *spare;
174};
175
567e9ab2
JW
176enum memcg_kmem_state {
177 KMEM_NONE,
178 KMEM_ALLOCATED,
179 KMEM_ONLINE,
180};
181
e81bf979
AL
182#if defined(CONFIG_SMP)
183struct memcg_padding {
184 char x[0];
185} ____cacheline_internodealigned_in_smp;
186#define MEMCG_PADDING(name) struct memcg_padding name;
187#else
188#define MEMCG_PADDING(name)
189#endif
190
33398cf2
MH
191/*
192 * The memory controller data structure. The memory controller controls both
193 * page cache and RSS per cgroup. We would eventually like to provide
194 * statistics based on the statistics developed by Rik Van Riel for clock-pro,
195 * to help the administrator determine what knobs to tune.
196 */
197struct mem_cgroup {
198 struct cgroup_subsys_state css;
199
73f576c0
JW
200 /* Private memcg ID. Used to ID objects that outlive the cgroup */
201 struct mem_cgroup_id id;
202
33398cf2
MH
203 /* Accounted resources */
204 struct page_counter memory;
37e84351 205 struct page_counter swap;
0db15298
JW
206
207 /* Legacy consumer-oriented counters */
33398cf2
MH
208 struct page_counter memsw;
209 struct page_counter kmem;
0db15298 210 struct page_counter tcpmem;
33398cf2 211
23067153 212 /* Upper bound of normal memory consumption range */
33398cf2
MH
213 unsigned long high;
214
f7e1cb6e
JW
215 /* Range enforcement for interrupt charges */
216 struct work_struct high_work;
217
33398cf2
MH
218 unsigned long soft_limit;
219
220 /* vmpressure notifications */
221 struct vmpressure vmpressure;
222
33398cf2
MH
223 /*
224 * Should the accounting and control be hierarchical, per subtree?
225 */
226 bool use_hierarchy;
227
3d8b38eb
RG
228 /*
229 * Should the OOM killer kill all belonging tasks, had it kill one?
230 */
231 bool oom_group;
232
33398cf2
MH
233 /* protected by memcg_oom_lock */
234 bool oom_lock;
235 int under_oom;
236
237 int swappiness;
238 /* OOM-Killer disable */
239 int oom_kill_disable;
240
e27be240 241 /* memory.events */
472912a2
TH
242 struct cgroup_file events_file;
243
f3a53a3a
TH
244 /* handle for "memory.swap.events" */
245 struct cgroup_file swap_events_file;
246
33398cf2
MH
247 /* protect arrays of thresholds */
248 struct mutex thresholds_lock;
249
250 /* thresholds for memory usage. RCU-protected */
251 struct mem_cgroup_thresholds thresholds;
252
253 /* thresholds for mem+swap usage. RCU-protected */
254 struct mem_cgroup_thresholds memsw_thresholds;
255
256 /* For oom notifier event fd */
257 struct list_head oom_notify;
258
259 /*
260 * Should we move charges of a task when a task is moved into this
261 * mem_cgroup ? And what type of charges should we move ?
262 */
263 unsigned long move_charge_at_immigrate;
e81bf979
AL
264 /* taken only while moving_account > 0 */
265 spinlock_t move_lock;
266 unsigned long move_lock_flags;
267
268 MEMCG_PADDING(_pad1_);
269
33398cf2
MH
270 /*
271 * set > 0 if pages under this cgroup are moving to other cgroup.
272 */
273 atomic_t moving_account;
33398cf2 274 struct task_struct *move_lock_task;
a983b5eb 275
e27be240 276 /* memory.stat */
a983b5eb 277 struct mem_cgroup_stat_cpu __percpu *stat_cpu;
e81bf979
AL
278
279 MEMCG_PADDING(_pad2_);
280
a983b5eb 281 atomic_long_t stat[MEMCG_NR_STAT];
e27be240 282 atomic_long_t events[NR_VM_EVENT_ITEMS];
e81bf979 283 atomic_long_t memory_events[MEMCG_NR_MEMORY_EVENTS];
33398cf2 284
d886f4e4
JW
285 unsigned long socket_pressure;
286
287 /* Legacy tcp memory accounting */
0db15298
JW
288 bool tcpmem_active;
289 int tcpmem_pressure;
d886f4e4 290
84c07d11 291#ifdef CONFIG_MEMCG_KMEM
33398cf2
MH
292 /* Index in the kmem_cache->memcg_params.memcg_caches array */
293 int kmemcg_id;
567e9ab2 294 enum memcg_kmem_state kmem_state;
bc2791f8 295 struct list_head kmem_caches;
33398cf2
MH
296#endif
297
298 int last_scanned_node;
299#if MAX_NUMNODES > 1
300 nodemask_t scan_nodes;
301 atomic_t numainfo_events;
302 atomic_t numainfo_updating;
303#endif
304
305#ifdef CONFIG_CGROUP_WRITEBACK
306 struct list_head cgwb_list;
307 struct wb_domain cgwb_domain;
308#endif
309
310 /* List of events which userspace want to receive */
311 struct list_head event_list;
312 spinlock_t event_list_lock;
313
314 struct mem_cgroup_per_node *nodeinfo[0];
315 /* WARNING: nodeinfo must be the last member here */
316};
7d828602 317
a983b5eb
JW
318/*
319 * size of first charge trial. "32" comes from vmscan.c's magic value.
320 * TODO: maybe necessary to use big numbers in big irons.
321 */
322#define MEMCG_CHARGE_BATCH 32U
323
7d828602 324extern struct mem_cgroup *root_mem_cgroup;
56161634 325
dfd2f10c
KT
326static inline bool mem_cgroup_is_root(struct mem_cgroup *memcg)
327{
328 return (memcg == root_mem_cgroup);
329}
330
23047a96
JW
331static inline bool mem_cgroup_disabled(void)
332{
333 return !cgroup_subsys_enabled(memory_cgrp_subsys);
334}
335
bf8d5d52
RG
336enum mem_cgroup_protection mem_cgroup_protected(struct mem_cgroup *root,
337 struct mem_cgroup *memcg);
241994ed 338
00501b53 339int mem_cgroup_try_charge(struct page *page, struct mm_struct *mm,
f627c2f5
KS
340 gfp_t gfp_mask, struct mem_cgroup **memcgp,
341 bool compound);
2cf85583
TH
342int mem_cgroup_try_charge_delay(struct page *page, struct mm_struct *mm,
343 gfp_t gfp_mask, struct mem_cgroup **memcgp,
344 bool compound);
00501b53 345void mem_cgroup_commit_charge(struct page *page, struct mem_cgroup *memcg,
f627c2f5
KS
346 bool lrucare, bool compound);
347void mem_cgroup_cancel_charge(struct page *page, struct mem_cgroup *memcg,
348 bool compound);
0a31bc97 349void mem_cgroup_uncharge(struct page *page);
747db954 350void mem_cgroup_uncharge_list(struct list_head *page_list);
569b846d 351
6a93ca8f 352void mem_cgroup_migrate(struct page *oldpage, struct page *newpage);
569b846d 353
ef8f2327
MG
354static struct mem_cgroup_per_node *
355mem_cgroup_nodeinfo(struct mem_cgroup *memcg, int nid)
55779ec7 356{
ef8f2327 357 return memcg->nodeinfo[nid];
55779ec7
JW
358}
359
360/**
a9dd0a83
MG
361 * mem_cgroup_lruvec - get the lru list vector for a node or a memcg zone
362 * @node: node of the wanted lruvec
55779ec7
JW
363 * @memcg: memcg of the wanted lruvec
364 *
a9dd0a83
MG
365 * Returns the lru list vector holding pages for a given @node or a given
366 * @memcg and @zone. This can be the node lruvec, if the memory controller
55779ec7
JW
367 * is disabled.
368 */
a9dd0a83 369static inline struct lruvec *mem_cgroup_lruvec(struct pglist_data *pgdat,
ef8f2327 370 struct mem_cgroup *memcg)
55779ec7 371{
ef8f2327 372 struct mem_cgroup_per_node *mz;
55779ec7
JW
373 struct lruvec *lruvec;
374
375 if (mem_cgroup_disabled()) {
a9dd0a83 376 lruvec = node_lruvec(pgdat);
55779ec7
JW
377 goto out;
378 }
379
ef8f2327 380 mz = mem_cgroup_nodeinfo(memcg, pgdat->node_id);
55779ec7
JW
381 lruvec = &mz->lruvec;
382out:
383 /*
384 * Since a node can be onlined after the mem_cgroup was created,
599d0c95 385 * we have to be prepared to initialize lruvec->pgdat here;
55779ec7
JW
386 * and if offlined then reonlined, we need to reinitialize it.
387 */
ef8f2327
MG
388 if (unlikely(lruvec->pgdat != pgdat))
389 lruvec->pgdat = pgdat;
55779ec7
JW
390 return lruvec;
391}
392
599d0c95 393struct lruvec *mem_cgroup_page_lruvec(struct page *, struct pglist_data *);
c9b0ed51 394
2314b42d 395bool task_in_mem_cgroup(struct task_struct *task, struct mem_cgroup *memcg);
64219994 396struct mem_cgroup *mem_cgroup_from_task(struct task_struct *p);
e993d905 397
d46eb14b
SB
398struct mem_cgroup *get_mem_cgroup_from_mm(struct mm_struct *mm);
399
f745c6f5
SB
400struct mem_cgroup *get_mem_cgroup_from_page(struct page *page);
401
33398cf2
MH
402static inline
403struct mem_cgroup *mem_cgroup_from_css(struct cgroup_subsys_state *css){
404 return css ? container_of(css, struct mem_cgroup, css) : NULL;
405}
406
dc0b5864
RG
407static inline void mem_cgroup_put(struct mem_cgroup *memcg)
408{
d46eb14b
SB
409 if (memcg)
410 css_put(&memcg->css);
dc0b5864
RG
411}
412
8e8ae645
JW
413#define mem_cgroup_from_counter(counter, member) \
414 container_of(counter, struct mem_cgroup, member)
415
33398cf2
MH
416struct mem_cgroup *mem_cgroup_iter(struct mem_cgroup *,
417 struct mem_cgroup *,
418 struct mem_cgroup_reclaim_cookie *);
419void mem_cgroup_iter_break(struct mem_cgroup *, struct mem_cgroup *);
7c5f64f8
VD
420int mem_cgroup_scan_tasks(struct mem_cgroup *,
421 int (*)(struct task_struct *, void *), void *);
33398cf2 422
23047a96
JW
423static inline unsigned short mem_cgroup_id(struct mem_cgroup *memcg)
424{
425 if (mem_cgroup_disabled())
426 return 0;
427
73f576c0 428 return memcg->id.id;
23047a96 429}
73f576c0 430struct mem_cgroup *mem_cgroup_from_id(unsigned short id);
23047a96 431
2262185c
RG
432static inline struct mem_cgroup *lruvec_memcg(struct lruvec *lruvec)
433{
434 struct mem_cgroup_per_node *mz;
435
436 if (mem_cgroup_disabled())
437 return NULL;
438
439 mz = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
440 return mz->memcg;
441}
442
8e8ae645
JW
443/**
444 * parent_mem_cgroup - find the accounting parent of a memcg
445 * @memcg: memcg whose parent to find
446 *
447 * Returns the parent memcg, or NULL if this is the root or the memory
448 * controller is in legacy no-hierarchy mode.
449 */
450static inline struct mem_cgroup *parent_mem_cgroup(struct mem_cgroup *memcg)
451{
452 if (!memcg->memory.parent)
453 return NULL;
454 return mem_cgroup_from_counter(memcg->memory.parent, memory);
455}
456
33398cf2
MH
457static inline bool mem_cgroup_is_descendant(struct mem_cgroup *memcg,
458 struct mem_cgroup *root)
459{
460 if (root == memcg)
461 return true;
462 if (!root->use_hierarchy)
463 return false;
464 return cgroup_is_descendant(memcg->css.cgroup, root->css.cgroup);
465}
e1aab161 466
2314b42d
JW
467static inline bool mm_match_cgroup(struct mm_struct *mm,
468 struct mem_cgroup *memcg)
2e4d4091 469{
587af308 470 struct mem_cgroup *task_memcg;
413918bb 471 bool match = false;
c3ac9a8a 472
2e4d4091 473 rcu_read_lock();
587af308 474 task_memcg = mem_cgroup_from_task(rcu_dereference(mm->owner));
413918bb 475 if (task_memcg)
2314b42d 476 match = mem_cgroup_is_descendant(task_memcg, memcg);
2e4d4091 477 rcu_read_unlock();
c3ac9a8a 478 return match;
2e4d4091 479}
8a9f3ccd 480
64219994 481struct cgroup_subsys_state *mem_cgroup_css_from_page(struct page *page);
2fc04524 482ino_t page_cgroup_ino(struct page *page);
d324236b 483
eb01aaab
VD
484static inline bool mem_cgroup_online(struct mem_cgroup *memcg)
485{
486 if (mem_cgroup_disabled())
487 return true;
488 return !!(memcg->css.flags & CSS_ONLINE);
489}
490
58ae83db
KH
491/*
492 * For memory reclaim.
493 */
889976db 494int mem_cgroup_select_victim_node(struct mem_cgroup *memcg);
33398cf2
MH
495
496void mem_cgroup_update_lru_size(struct lruvec *lruvec, enum lru_list lru,
b4536f0c 497 int zid, int nr_pages);
33398cf2 498
0a6b76dd
VD
499unsigned long mem_cgroup_node_nr_lru_pages(struct mem_cgroup *memcg,
500 int nid, unsigned int lru_mask);
501
33398cf2
MH
502static inline
503unsigned long mem_cgroup_get_lru_size(struct lruvec *lruvec, enum lru_list lru)
504{
ef8f2327 505 struct mem_cgroup_per_node *mz;
b4536f0c
MH
506 unsigned long nr_pages = 0;
507 int zid;
33398cf2 508
ef8f2327 509 mz = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
b4536f0c
MH
510 for (zid = 0; zid < MAX_NR_ZONES; zid++)
511 nr_pages += mz->lru_zone_size[zid][lru];
512 return nr_pages;
513}
514
515static inline
516unsigned long mem_cgroup_get_zone_lru_size(struct lruvec *lruvec,
517 enum lru_list lru, int zone_idx)
518{
519 struct mem_cgroup_per_node *mz;
520
521 mz = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
522 return mz->lru_zone_size[zone_idx][lru];
33398cf2
MH
523}
524
b23afb93
TH
525void mem_cgroup_handle_over_high(void);
526
bbec2e15 527unsigned long mem_cgroup_get_max(struct mem_cgroup *memcg);
7c5f64f8 528
f0c867d9 529void mem_cgroup_print_oom_context(struct mem_cgroup *memcg,
64219994 530 struct task_struct *p);
58ae83db 531
f0c867d9 532void mem_cgroup_print_oom_meminfo(struct mem_cgroup *memcg);
533
29ef680a 534static inline void mem_cgroup_enter_user_fault(void)
519e5247 535{
29ef680a
MH
536 WARN_ON(current->in_user_fault);
537 current->in_user_fault = 1;
519e5247
JW
538}
539
29ef680a 540static inline void mem_cgroup_exit_user_fault(void)
519e5247 541{
29ef680a
MH
542 WARN_ON(!current->in_user_fault);
543 current->in_user_fault = 0;
519e5247
JW
544}
545
3812c8c8
JW
546static inline bool task_in_memcg_oom(struct task_struct *p)
547{
626ebc41 548 return p->memcg_in_oom;
3812c8c8
JW
549}
550
49426420 551bool mem_cgroup_oom_synchronize(bool wait);
3d8b38eb
RG
552struct mem_cgroup *mem_cgroup_get_oom_group(struct task_struct *victim,
553 struct mem_cgroup *oom_domain);
554void mem_cgroup_print_oom_group(struct mem_cgroup *memcg);
3812c8c8 555
c255a458 556#ifdef CONFIG_MEMCG_SWAP
c077719b
KH
557extern int do_swap_account;
558#endif
f8d66542 559
739f79fc
JW
560struct mem_cgroup *lock_page_memcg(struct page *page);
561void __unlock_page_memcg(struct mem_cgroup *memcg);
62cccb8c 562void unlock_page_memcg(struct page *page);
d7365e78 563
04fecbf5 564/* idx can be of type enum memcg_stat_item or node_stat_item */
ccda7f43 565static inline unsigned long memcg_page_state(struct mem_cgroup *memcg,
04fecbf5 566 int idx)
2a2e4885 567{
a983b5eb
JW
568 long x = atomic_long_read(&memcg->stat[idx]);
569#ifdef CONFIG_SMP
570 if (x < 0)
571 x = 0;
572#endif
573 return x;
2a2e4885
JW
574}
575
04fecbf5 576/* idx can be of type enum memcg_stat_item or node_stat_item */
00f3ca2c 577static inline void __mod_memcg_state(struct mem_cgroup *memcg,
04fecbf5 578 int idx, int val)
2a2e4885 579{
a983b5eb
JW
580 long x;
581
582 if (mem_cgroup_disabled())
583 return;
584
585 x = val + __this_cpu_read(memcg->stat_cpu->count[idx]);
586 if (unlikely(abs(x) > MEMCG_CHARGE_BATCH)) {
587 atomic_long_add(x, &memcg->stat[idx]);
588 x = 0;
589 }
590 __this_cpu_write(memcg->stat_cpu->count[idx], x);
2a2e4885
JW
591}
592
04fecbf5 593/* idx can be of type enum memcg_stat_item or node_stat_item */
00f3ca2c 594static inline void mod_memcg_state(struct mem_cgroup *memcg,
04fecbf5 595 int idx, int val)
2a2e4885 596{
c3cc3911
JW
597 unsigned long flags;
598
599 local_irq_save(flags);
a983b5eb 600 __mod_memcg_state(memcg, idx, val);
c3cc3911 601 local_irq_restore(flags);
2a2e4885
JW
602}
603
33398cf2 604/**
ccda7f43 605 * mod_memcg_page_state - update page state statistics
62cccb8c 606 * @page: the page
33398cf2
MH
607 * @idx: page state item to account
608 * @val: number of pages (positive or negative)
609 *
fdf1cdb9
JW
610 * The @page must be locked or the caller must use lock_page_memcg()
611 * to prevent double accounting when the page is concurrently being
612 * moved to another memcg:
81f8c3a4 613 *
fdf1cdb9 614 * lock_page(page) or lock_page_memcg(page)
81f8c3a4 615 * if (TestClearPageState(page))
ccda7f43 616 * mod_memcg_page_state(page, state, -1);
fdf1cdb9 617 * unlock_page(page) or unlock_page_memcg(page)
2a2e4885
JW
618 *
619 * Kernel pages are an exception to this, since they'll never move.
33398cf2 620 */
00f3ca2c 621static inline void __mod_memcg_page_state(struct page *page,
04fecbf5 622 int idx, int val)
00f3ca2c
JW
623{
624 if (page->mem_cgroup)
625 __mod_memcg_state(page->mem_cgroup, idx, val);
626}
627
ccda7f43 628static inline void mod_memcg_page_state(struct page *page,
04fecbf5 629 int idx, int val)
33398cf2 630{
62cccb8c 631 if (page->mem_cgroup)
ccda7f43 632 mod_memcg_state(page->mem_cgroup, idx, val);
33398cf2
MH
633}
634
00f3ca2c
JW
635static inline unsigned long lruvec_page_state(struct lruvec *lruvec,
636 enum node_stat_item idx)
2a7106f2 637{
00f3ca2c 638 struct mem_cgroup_per_node *pn;
a983b5eb 639 long x;
00f3ca2c
JW
640
641 if (mem_cgroup_disabled())
642 return node_page_state(lruvec_pgdat(lruvec), idx);
643
644 pn = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
a983b5eb
JW
645 x = atomic_long_read(&pn->lruvec_stat[idx]);
646#ifdef CONFIG_SMP
647 if (x < 0)
648 x = 0;
649#endif
650 return x;
2a7106f2
GT
651}
652
00f3ca2c
JW
653static inline void __mod_lruvec_state(struct lruvec *lruvec,
654 enum node_stat_item idx, int val)
2a7106f2 655{
00f3ca2c 656 struct mem_cgroup_per_node *pn;
a983b5eb 657 long x;
00f3ca2c 658
28454265 659 /* Update node */
00f3ca2c 660 __mod_node_page_state(lruvec_pgdat(lruvec), idx, val);
28454265 661
00f3ca2c
JW
662 if (mem_cgroup_disabled())
663 return;
28454265 664
00f3ca2c 665 pn = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
28454265
JW
666
667 /* Update memcg */
00f3ca2c 668 __mod_memcg_state(pn->memcg, idx, val);
28454265
JW
669
670 /* Update lruvec */
a983b5eb
JW
671 x = val + __this_cpu_read(pn->lruvec_stat_cpu->count[idx]);
672 if (unlikely(abs(x) > MEMCG_CHARGE_BATCH)) {
673 atomic_long_add(x, &pn->lruvec_stat[idx]);
674 x = 0;
675 }
676 __this_cpu_write(pn->lruvec_stat_cpu->count[idx], x);
00f3ca2c
JW
677}
678
679static inline void mod_lruvec_state(struct lruvec *lruvec,
680 enum node_stat_item idx, int val)
681{
c3cc3911
JW
682 unsigned long flags;
683
684 local_irq_save(flags);
28454265 685 __mod_lruvec_state(lruvec, idx, val);
c3cc3911 686 local_irq_restore(flags);
00f3ca2c
JW
687}
688
689static inline void __mod_lruvec_page_state(struct page *page,
690 enum node_stat_item idx, int val)
691{
28454265
JW
692 pg_data_t *pgdat = page_pgdat(page);
693 struct lruvec *lruvec;
00f3ca2c 694
28454265
JW
695 /* Untracked pages have no memcg, no lruvec. Update only the node */
696 if (!page->mem_cgroup) {
697 __mod_node_page_state(pgdat, idx, val);
00f3ca2c 698 return;
28454265
JW
699 }
700
701 lruvec = mem_cgroup_lruvec(pgdat, page->mem_cgroup);
702 __mod_lruvec_state(lruvec, idx, val);
00f3ca2c
JW
703}
704
705static inline void mod_lruvec_page_state(struct page *page,
706 enum node_stat_item idx, int val)
707{
c3cc3911
JW
708 unsigned long flags;
709
710 local_irq_save(flags);
28454265 711 __mod_lruvec_page_state(page, idx, val);
c3cc3911 712 local_irq_restore(flags);
2a7106f2
GT
713}
714
ef8f2327 715unsigned long mem_cgroup_soft_limit_reclaim(pg_data_t *pgdat, int order,
0608f43d
AM
716 gfp_t gfp_mask,
717 unsigned long *total_scanned);
a63d83f4 718
c9019e9b 719static inline void __count_memcg_events(struct mem_cgroup *memcg,
e27be240
JW
720 enum vm_event_item idx,
721 unsigned long count)
c9019e9b 722{
a983b5eb
JW
723 unsigned long x;
724
725 if (mem_cgroup_disabled())
726 return;
727
728 x = count + __this_cpu_read(memcg->stat_cpu->events[idx]);
729 if (unlikely(x > MEMCG_CHARGE_BATCH)) {
730 atomic_long_add(x, &memcg->events[idx]);
731 x = 0;
732 }
733 __this_cpu_write(memcg->stat_cpu->events[idx], x);
c9019e9b
JW
734}
735
2262185c 736static inline void count_memcg_events(struct mem_cgroup *memcg,
e27be240
JW
737 enum vm_event_item idx,
738 unsigned long count)
2262185c 739{
c3cc3911
JW
740 unsigned long flags;
741
742 local_irq_save(flags);
a983b5eb 743 __count_memcg_events(memcg, idx, count);
c3cc3911 744 local_irq_restore(flags);
2262185c
RG
745}
746
747static inline void count_memcg_page_event(struct page *page,
e27be240 748 enum vm_event_item idx)
2262185c
RG
749{
750 if (page->mem_cgroup)
751 count_memcg_events(page->mem_cgroup, idx, 1);
752}
753
754static inline void count_memcg_event_mm(struct mm_struct *mm,
755 enum vm_event_item idx)
68ae564b 756{
33398cf2
MH
757 struct mem_cgroup *memcg;
758
68ae564b
DR
759 if (mem_cgroup_disabled())
760 return;
33398cf2
MH
761
762 rcu_read_lock();
763 memcg = mem_cgroup_from_task(rcu_dereference(mm->owner));
fe6bdfc8 764 if (likely(memcg))
c9019e9b 765 count_memcg_events(memcg, idx, 1);
33398cf2 766 rcu_read_unlock();
68ae564b 767}
c9019e9b 768
e27be240
JW
769static inline void memcg_memory_event(struct mem_cgroup *memcg,
770 enum memcg_memory_event event)
c9019e9b 771{
e27be240 772 atomic_long_inc(&memcg->memory_events[event]);
c9019e9b
JW
773 cgroup_file_notify(&memcg->events_file);
774}
775
fe6bdfc8
RG
776static inline void memcg_memory_event_mm(struct mm_struct *mm,
777 enum memcg_memory_event event)
778{
779 struct mem_cgroup *memcg;
780
781 if (mem_cgroup_disabled())
782 return;
783
784 rcu_read_lock();
785 memcg = mem_cgroup_from_task(rcu_dereference(mm->owner));
786 if (likely(memcg))
787 memcg_memory_event(memcg, event);
788 rcu_read_unlock();
789}
790
ca3e0214 791#ifdef CONFIG_TRANSPARENT_HUGEPAGE
e94c8a9c 792void mem_cgroup_split_huge_fixup(struct page *head);
ca3e0214
KH
793#endif
794
c255a458 795#else /* CONFIG_MEMCG */
23047a96
JW
796
797#define MEM_CGROUP_ID_SHIFT 0
798#define MEM_CGROUP_ID_MAX 0
799
7a81b88c
KH
800struct mem_cgroup;
801
dfd2f10c
KT
802static inline bool mem_cgroup_is_root(struct mem_cgroup *memcg)
803{
804 return true;
805}
806
23047a96
JW
807static inline bool mem_cgroup_disabled(void)
808{
809 return true;
810}
811
e27be240
JW
812static inline void memcg_memory_event(struct mem_cgroup *memcg,
813 enum memcg_memory_event event)
241994ed
JW
814{
815}
816
fe6bdfc8
RG
817static inline void memcg_memory_event_mm(struct mm_struct *mm,
818 enum memcg_memory_event event)
819{
820}
821
bf8d5d52
RG
822static inline enum mem_cgroup_protection mem_cgroup_protected(
823 struct mem_cgroup *root, struct mem_cgroup *memcg)
241994ed 824{
bf8d5d52 825 return MEMCG_PROT_NONE;
241994ed
JW
826}
827
00501b53
JW
828static inline int mem_cgroup_try_charge(struct page *page, struct mm_struct *mm,
829 gfp_t gfp_mask,
f627c2f5
KS
830 struct mem_cgroup **memcgp,
831 bool compound)
7a81b88c 832{
00501b53 833 *memcgp = NULL;
7a81b88c
KH
834 return 0;
835}
836
2cf85583
TH
837static inline int mem_cgroup_try_charge_delay(struct page *page,
838 struct mm_struct *mm,
839 gfp_t gfp_mask,
840 struct mem_cgroup **memcgp,
841 bool compound)
842{
843 *memcgp = NULL;
844 return 0;
845}
846
00501b53
JW
847static inline void mem_cgroup_commit_charge(struct page *page,
848 struct mem_cgroup *memcg,
f627c2f5 849 bool lrucare, bool compound)
7a81b88c
KH
850{
851}
852
00501b53 853static inline void mem_cgroup_cancel_charge(struct page *page,
f627c2f5
KS
854 struct mem_cgroup *memcg,
855 bool compound)
7a81b88c
KH
856{
857}
858
0a31bc97 859static inline void mem_cgroup_uncharge(struct page *page)
569b846d
KH
860{
861}
862
747db954 863static inline void mem_cgroup_uncharge_list(struct list_head *page_list)
8a9f3ccd
BS
864{
865}
866
6a93ca8f 867static inline void mem_cgroup_migrate(struct page *old, struct page *new)
69029cd5
KH
868{
869}
870
a9dd0a83 871static inline struct lruvec *mem_cgroup_lruvec(struct pglist_data *pgdat,
ef8f2327 872 struct mem_cgroup *memcg)
08e552c6 873{
a9dd0a83 874 return node_lruvec(pgdat);
08e552c6
KH
875}
876
fa9add64 877static inline struct lruvec *mem_cgroup_page_lruvec(struct page *page,
599d0c95 878 struct pglist_data *pgdat)
66e1707b 879{
599d0c95 880 return &pgdat->lruvec;
66e1707b
BS
881}
882
587af308 883static inline bool mm_match_cgroup(struct mm_struct *mm,
c0ff4b85 884 struct mem_cgroup *memcg)
bed7161a 885{
587af308 886 return true;
bed7161a
BS
887}
888
ffbdccf5
DR
889static inline bool task_in_mem_cgroup(struct task_struct *task,
890 const struct mem_cgroup *memcg)
4c4a2214 891{
ffbdccf5 892 return true;
4c4a2214
DR
893}
894
d46eb14b
SB
895static inline struct mem_cgroup *get_mem_cgroup_from_mm(struct mm_struct *mm)
896{
897 return NULL;
898}
899
f745c6f5
SB
900static inline struct mem_cgroup *get_mem_cgroup_from_page(struct page *page)
901{
902 return NULL;
903}
904
dc0b5864
RG
905static inline void mem_cgroup_put(struct mem_cgroup *memcg)
906{
907}
908
5660048c
JW
909static inline struct mem_cgroup *
910mem_cgroup_iter(struct mem_cgroup *root,
911 struct mem_cgroup *prev,
912 struct mem_cgroup_reclaim_cookie *reclaim)
913{
914 return NULL;
915}
916
917static inline void mem_cgroup_iter_break(struct mem_cgroup *root,
918 struct mem_cgroup *prev)
919{
920}
921
7c5f64f8
VD
922static inline int mem_cgroup_scan_tasks(struct mem_cgroup *memcg,
923 int (*fn)(struct task_struct *, void *), void *arg)
924{
925 return 0;
926}
927
23047a96 928static inline unsigned short mem_cgroup_id(struct mem_cgroup *memcg)
f8d66542 929{
23047a96
JW
930 return 0;
931}
932
933static inline struct mem_cgroup *mem_cgroup_from_id(unsigned short id)
934{
935 WARN_ON_ONCE(id);
936 /* XXX: This should always return root_mem_cgroup */
937 return NULL;
f8d66542 938}
a636b327 939
2262185c
RG
940static inline struct mem_cgroup *lruvec_memcg(struct lruvec *lruvec)
941{
942 return NULL;
943}
944
eb01aaab 945static inline bool mem_cgroup_online(struct mem_cgroup *memcg)
14797e23 946{
13308ca9 947 return true;
14797e23
KM
948}
949
a3d8e054 950static inline unsigned long
4d7dcca2 951mem_cgroup_get_lru_size(struct lruvec *lruvec, enum lru_list lru)
a3d8e054
KM
952{
953 return 0;
954}
b4536f0c
MH
955static inline
956unsigned long mem_cgroup_get_zone_lru_size(struct lruvec *lruvec,
957 enum lru_list lru, int zone_idx)
958{
959 return 0;
960}
a3d8e054 961
0a6b76dd
VD
962static inline unsigned long
963mem_cgroup_node_nr_lru_pages(struct mem_cgroup *memcg,
964 int nid, unsigned int lru_mask)
965{
966 return 0;
967}
968
bbec2e15 969static inline unsigned long mem_cgroup_get_max(struct mem_cgroup *memcg)
7c5f64f8
VD
970{
971 return 0;
972}
973
e222432b 974static inline void
f0c867d9 975mem_cgroup_print_oom_context(struct mem_cgroup *memcg, struct task_struct *p)
976{
977}
978
979static inline void
980mem_cgroup_print_oom_meminfo(struct mem_cgroup *memcg)
e222432b
BS
981{
982}
983
739f79fc
JW
984static inline struct mem_cgroup *lock_page_memcg(struct page *page)
985{
986 return NULL;
987}
988
989static inline void __unlock_page_memcg(struct mem_cgroup *memcg)
89c06bd5
KH
990{
991}
992
62cccb8c 993static inline void unlock_page_memcg(struct page *page)
89c06bd5
KH
994{
995}
996
b23afb93
TH
997static inline void mem_cgroup_handle_over_high(void)
998{
999}
1000
29ef680a 1001static inline void mem_cgroup_enter_user_fault(void)
519e5247
JW
1002{
1003}
1004
29ef680a 1005static inline void mem_cgroup_exit_user_fault(void)
519e5247
JW
1006{
1007}
1008
3812c8c8
JW
1009static inline bool task_in_memcg_oom(struct task_struct *p)
1010{
1011 return false;
1012}
1013
49426420 1014static inline bool mem_cgroup_oom_synchronize(bool wait)
3812c8c8
JW
1015{
1016 return false;
1017}
1018
3d8b38eb
RG
1019static inline struct mem_cgroup *mem_cgroup_get_oom_group(
1020 struct task_struct *victim, struct mem_cgroup *oom_domain)
1021{
1022 return NULL;
1023}
1024
1025static inline void mem_cgroup_print_oom_group(struct mem_cgroup *memcg)
1026{
1027}
1028
ccda7f43 1029static inline unsigned long memcg_page_state(struct mem_cgroup *memcg,
04fecbf5 1030 int idx)
2a2e4885
JW
1031{
1032 return 0;
1033}
1034
00f3ca2c 1035static inline void __mod_memcg_state(struct mem_cgroup *memcg,
04fecbf5 1036 int idx,
00f3ca2c 1037 int nr)
2a2e4885
JW
1038{
1039}
1040
00f3ca2c 1041static inline void mod_memcg_state(struct mem_cgroup *memcg,
04fecbf5 1042 int idx,
00f3ca2c 1043 int nr)
2a2e4885
JW
1044{
1045}
1046
00f3ca2c 1047static inline void __mod_memcg_page_state(struct page *page,
04fecbf5 1048 int idx,
00f3ca2c 1049 int nr)
2a2e4885
JW
1050{
1051}
1052
ccda7f43 1053static inline void mod_memcg_page_state(struct page *page,
04fecbf5 1054 int idx,
ccda7f43 1055 int nr)
553af430
JW
1056{
1057}
1058
00f3ca2c
JW
1059static inline unsigned long lruvec_page_state(struct lruvec *lruvec,
1060 enum node_stat_item idx)
2a7106f2 1061{
00f3ca2c 1062 return node_page_state(lruvec_pgdat(lruvec), idx);
2a7106f2
GT
1063}
1064
00f3ca2c
JW
1065static inline void __mod_lruvec_state(struct lruvec *lruvec,
1066 enum node_stat_item idx, int val)
d69b042f 1067{
00f3ca2c
JW
1068 __mod_node_page_state(lruvec_pgdat(lruvec), idx, val);
1069}
1070
1071static inline void mod_lruvec_state(struct lruvec *lruvec,
1072 enum node_stat_item idx, int val)
1073{
1074 mod_node_page_state(lruvec_pgdat(lruvec), idx, val);
1075}
1076
1077static inline void __mod_lruvec_page_state(struct page *page,
1078 enum node_stat_item idx, int val)
1079{
1080 __mod_node_page_state(page_pgdat(page), idx, val);
1081}
1082
1083static inline void mod_lruvec_page_state(struct page *page,
1084 enum node_stat_item idx, int val)
1085{
1086 mod_node_page_state(page_pgdat(page), idx, val);
d69b042f
BS
1087}
1088
4e416953 1089static inline
ef8f2327 1090unsigned long mem_cgroup_soft_limit_reclaim(pg_data_t *pgdat, int order,
0608f43d
AM
1091 gfp_t gfp_mask,
1092 unsigned long *total_scanned)
4e416953 1093{
0608f43d 1094 return 0;
4e416953
BS
1095}
1096
e94c8a9c 1097static inline void mem_cgroup_split_huge_fixup(struct page *head)
ca3e0214
KH
1098{
1099}
1100
2262185c
RG
1101static inline void count_memcg_events(struct mem_cgroup *memcg,
1102 enum vm_event_item idx,
1103 unsigned long count)
1104{
1105}
1106
1107static inline void count_memcg_page_event(struct page *page,
04fecbf5 1108 int idx)
2262185c
RG
1109{
1110}
1111
456f998e 1112static inline
2262185c 1113void count_memcg_event_mm(struct mm_struct *mm, enum vm_event_item idx)
456f998e
YH
1114{
1115}
c255a458 1116#endif /* CONFIG_MEMCG */
78fb7466 1117
04fecbf5 1118/* idx can be of type enum memcg_stat_item or node_stat_item */
00f3ca2c 1119static inline void __inc_memcg_state(struct mem_cgroup *memcg,
04fecbf5 1120 int idx)
00f3ca2c
JW
1121{
1122 __mod_memcg_state(memcg, idx, 1);
1123}
1124
04fecbf5 1125/* idx can be of type enum memcg_stat_item or node_stat_item */
00f3ca2c 1126static inline void __dec_memcg_state(struct mem_cgroup *memcg,
04fecbf5 1127 int idx)
00f3ca2c
JW
1128{
1129 __mod_memcg_state(memcg, idx, -1);
1130}
1131
04fecbf5 1132/* idx can be of type enum memcg_stat_item or node_stat_item */
00f3ca2c 1133static inline void __inc_memcg_page_state(struct page *page,
04fecbf5 1134 int idx)
00f3ca2c
JW
1135{
1136 __mod_memcg_page_state(page, idx, 1);
1137}
1138
04fecbf5 1139/* idx can be of type enum memcg_stat_item or node_stat_item */
00f3ca2c 1140static inline void __dec_memcg_page_state(struct page *page,
04fecbf5 1141 int idx)
00f3ca2c
JW
1142{
1143 __mod_memcg_page_state(page, idx, -1);
1144}
1145
1146static inline void __inc_lruvec_state(struct lruvec *lruvec,
1147 enum node_stat_item idx)
1148{
1149 __mod_lruvec_state(lruvec, idx, 1);
1150}
1151
1152static inline void __dec_lruvec_state(struct lruvec *lruvec,
1153 enum node_stat_item idx)
1154{
1155 __mod_lruvec_state(lruvec, idx, -1);
1156}
1157
1158static inline void __inc_lruvec_page_state(struct page *page,
1159 enum node_stat_item idx)
1160{
1161 __mod_lruvec_page_state(page, idx, 1);
1162}
1163
1164static inline void __dec_lruvec_page_state(struct page *page,
1165 enum node_stat_item idx)
1166{
1167 __mod_lruvec_page_state(page, idx, -1);
1168}
1169
04fecbf5 1170/* idx can be of type enum memcg_stat_item or node_stat_item */
00f3ca2c 1171static inline void inc_memcg_state(struct mem_cgroup *memcg,
04fecbf5 1172 int idx)
00f3ca2c
JW
1173{
1174 mod_memcg_state(memcg, idx, 1);
1175}
1176
04fecbf5 1177/* idx can be of type enum memcg_stat_item or node_stat_item */
00f3ca2c 1178static inline void dec_memcg_state(struct mem_cgroup *memcg,
04fecbf5 1179 int idx)
00f3ca2c
JW
1180{
1181 mod_memcg_state(memcg, idx, -1);
1182}
1183
04fecbf5 1184/* idx can be of type enum memcg_stat_item or node_stat_item */
00f3ca2c 1185static inline void inc_memcg_page_state(struct page *page,
04fecbf5 1186 int idx)
00f3ca2c
JW
1187{
1188 mod_memcg_page_state(page, idx, 1);
1189}
1190
04fecbf5 1191/* idx can be of type enum memcg_stat_item or node_stat_item */
00f3ca2c 1192static inline void dec_memcg_page_state(struct page *page,
04fecbf5 1193 int idx)
00f3ca2c
JW
1194{
1195 mod_memcg_page_state(page, idx, -1);
1196}
1197
1198static inline void inc_lruvec_state(struct lruvec *lruvec,
1199 enum node_stat_item idx)
1200{
1201 mod_lruvec_state(lruvec, idx, 1);
1202}
1203
1204static inline void dec_lruvec_state(struct lruvec *lruvec,
1205 enum node_stat_item idx)
1206{
1207 mod_lruvec_state(lruvec, idx, -1);
1208}
1209
1210static inline void inc_lruvec_page_state(struct page *page,
1211 enum node_stat_item idx)
1212{
1213 mod_lruvec_page_state(page, idx, 1);
1214}
1215
1216static inline void dec_lruvec_page_state(struct page *page,
1217 enum node_stat_item idx)
1218{
1219 mod_lruvec_page_state(page, idx, -1);
1220}
1221
52ebea74 1222#ifdef CONFIG_CGROUP_WRITEBACK
841710aa 1223
841710aa 1224struct wb_domain *mem_cgroup_wb_domain(struct bdi_writeback *wb);
c5edf9cd
TH
1225void mem_cgroup_wb_stats(struct bdi_writeback *wb, unsigned long *pfilepages,
1226 unsigned long *pheadroom, unsigned long *pdirty,
1227 unsigned long *pwriteback);
841710aa
TH
1228
1229#else /* CONFIG_CGROUP_WRITEBACK */
1230
1231static inline struct wb_domain *mem_cgroup_wb_domain(struct bdi_writeback *wb)
1232{
1233 return NULL;
1234}
1235
c2aa723a 1236static inline void mem_cgroup_wb_stats(struct bdi_writeback *wb,
c5edf9cd
TH
1237 unsigned long *pfilepages,
1238 unsigned long *pheadroom,
c2aa723a
TH
1239 unsigned long *pdirty,
1240 unsigned long *pwriteback)
1241{
1242}
1243
841710aa 1244#endif /* CONFIG_CGROUP_WRITEBACK */
52ebea74 1245
e1aab161 1246struct sock;
baac50bb
JW
1247bool mem_cgroup_charge_skmem(struct mem_cgroup *memcg, unsigned int nr_pages);
1248void mem_cgroup_uncharge_skmem(struct mem_cgroup *memcg, unsigned int nr_pages);
d886f4e4 1249#ifdef CONFIG_MEMCG
ef12947c
JW
1250extern struct static_key_false memcg_sockets_enabled_key;
1251#define mem_cgroup_sockets_enabled static_branch_unlikely(&memcg_sockets_enabled_key)
2d758073
JW
1252void mem_cgroup_sk_alloc(struct sock *sk);
1253void mem_cgroup_sk_free(struct sock *sk);
baac50bb 1254static inline bool mem_cgroup_under_socket_pressure(struct mem_cgroup *memcg)
e805605c 1255{
0db15298 1256 if (!cgroup_subsys_on_dfl(memory_cgrp_subsys) && memcg->tcpmem_pressure)
8e8ae645 1257 return true;
8e8ae645
JW
1258 do {
1259 if (time_before(jiffies, memcg->socket_pressure))
1260 return true;
1261 } while ((memcg = parent_mem_cgroup(memcg)));
1262 return false;
e805605c
JW
1263}
1264#else
80e95fe0 1265#define mem_cgroup_sockets_enabled 0
2d758073
JW
1266static inline void mem_cgroup_sk_alloc(struct sock *sk) { };
1267static inline void mem_cgroup_sk_free(struct sock *sk) { };
baac50bb 1268static inline bool mem_cgroup_under_socket_pressure(struct mem_cgroup *memcg)
e805605c
JW
1269{
1270 return false;
1271}
1272#endif
7ae1e1d0 1273
45264778
VD
1274struct kmem_cache *memcg_kmem_get_cache(struct kmem_cache *cachep);
1275void memcg_kmem_put_cache(struct kmem_cache *cachep);
1276int memcg_kmem_charge_memcg(struct page *page, gfp_t gfp, int order,
1277 struct mem_cgroup *memcg);
9b6f7e16
RG
1278
1279#ifdef CONFIG_MEMCG_KMEM
45264778
VD
1280int memcg_kmem_charge(struct page *page, gfp_t gfp, int order);
1281void memcg_kmem_uncharge(struct page *page, int order);
1282
ef12947c 1283extern struct static_key_false memcg_kmem_enabled_key;
17cc4dfe 1284extern struct workqueue_struct *memcg_kmem_cache_wq;
749c5415 1285
dbcf73e2 1286extern int memcg_nr_cache_ids;
64219994
MH
1287void memcg_get_cache_ids(void);
1288void memcg_put_cache_ids(void);
ebe945c2
GC
1289
1290/*
1291 * Helper macro to loop through all memcg-specific caches. Callers must still
1292 * check if the cache is valid (it is either valid or NULL).
1293 * the slab_mutex must be held when looping through those caches
1294 */
749c5415 1295#define for_each_memcg_cache_index(_idx) \
dbcf73e2 1296 for ((_idx) = 0; (_idx) < memcg_nr_cache_ids; (_idx)++)
749c5415 1297
7ae1e1d0
GC
1298static inline bool memcg_kmem_enabled(void)
1299{
ef12947c 1300 return static_branch_unlikely(&memcg_kmem_enabled_key);
7ae1e1d0
GC
1301}
1302
33398cf2 1303/*
9f706d68 1304 * helper for accessing a memcg's index. It will be used as an index in the
33398cf2
MH
1305 * child cache array in kmem_cache, and also to derive its name. This function
1306 * will return -1 when this is not a kmem-limited memcg.
1307 */
1308static inline int memcg_cache_id(struct mem_cgroup *memcg)
1309{
1310 return memcg ? memcg->kmemcg_id : -1;
1311}
5722d094 1312
0a4465d3
KT
1313extern int memcg_expand_shrinker_maps(int new_id);
1314
fae91d6d
KT
1315extern void memcg_set_shrinker_bit(struct mem_cgroup *memcg,
1316 int nid, int shrinker_id);
7ae1e1d0 1317#else
9b6f7e16
RG
1318
1319static inline int memcg_kmem_charge(struct page *page, gfp_t gfp, int order)
1320{
1321 return 0;
1322}
1323
1324static inline void memcg_kmem_uncharge(struct page *page, int order)
1325{
1326}
1327
749c5415
GC
1328#define for_each_memcg_cache_index(_idx) \
1329 for (; NULL; )
1330
b9ce5ef4
GC
1331static inline bool memcg_kmem_enabled(void)
1332{
1333 return false;
1334}
1335
2633d7a0
GC
1336static inline int memcg_cache_id(struct mem_cgroup *memcg)
1337{
1338 return -1;
1339}
1340
05257a1a
VD
1341static inline void memcg_get_cache_ids(void)
1342{
1343}
1344
1345static inline void memcg_put_cache_ids(void)
1346{
1347}
1348
fae91d6d
KT
1349static inline void memcg_set_shrinker_bit(struct mem_cgroup *memcg,
1350 int nid, int shrinker_id) { }
84c07d11 1351#endif /* CONFIG_MEMCG_KMEM */
127424c8 1352
8cdea7c0 1353#endif /* _LINUX_MEMCONTROL_H */