memcontrol.h 31.0 KB
Newer Older
B
Balbir Singh 已提交
1 2 3 4 5
/* memcontrol.h - Memory Controller
 *
 * Copyright IBM Corporation, 2007
 * Author Balbir Singh <balbir@linux.vnet.ibm.com>
 *
6 7 8
 * Copyright 2007 OpenVZ SWsoft Inc
 * Author: Pavel Emelianov <xemul@openvz.org>
 *
B
Balbir Singh 已提交
9 10 11 12 13 14 15 16 17 18 19 20 21
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 */

#ifndef _LINUX_MEMCONTROL_H
#define _LINUX_MEMCONTROL_H
22
#include <linux/cgroup.h>
23
#include <linux/vm_event_item.h>
24
#include <linux/hardirq.h>
25
#include <linux/jump_label.h>
M
Michal Hocko 已提交
26 27 28
#include <linux/page_counter.h>
#include <linux/vmpressure.h>
#include <linux/eventfd.h>
29 30
#include <linux/mm.h>
#include <linux/vmstat.h>
M
Michal Hocko 已提交
31
#include <linux/writeback.h>
32
#include <linux/page-flags.h>
33

34
struct mem_cgroup;
35 36
struct page;
struct mm_struct;
37
struct kmem_cache;
38

39 40 41 42 43 44 45 46 47
/* Cgroup-specific page state, on top of universal node page state */
enum memcg_stat_item {
	MEMCG_CACHE = NR_VM_NODE_STAT_ITEMS,
	MEMCG_RSS,
	MEMCG_RSS_HUGE,
	MEMCG_SWAP,
	MEMCG_SOCK,
	/* XXX: why are these zone and not node counters? */
	MEMCG_KERNEL_STACK_KB,
48
	MEMCG_NR_STAT,
49 50
};

51 52
enum memcg_memory_event {
	MEMCG_LOW,
53 54 55
	MEMCG_HIGH,
	MEMCG_MAX,
	MEMCG_OOM,
R
Roman Gushchin 已提交
56
	MEMCG_OOM_KILL,
57 58
	MEMCG_SWAP_MAX,
	MEMCG_SWAP_FAIL,
59
	MEMCG_NR_MEMORY_EVENTS,
60 61
};

R
Roman Gushchin 已提交
62 63 64 65 66 67
enum mem_cgroup_protection {
	MEMCG_PROT_NONE,
	MEMCG_PROT_LOW,
	MEMCG_PROT_MIN,
};

68
struct mem_cgroup_reclaim_cookie {
69
	pg_data_t *pgdat;
70 71 72 73
	int priority;
	unsigned int generation;
};

74 75 76 77 78 79 80 81 82 83
#ifdef CONFIG_MEMCG

#define MEM_CGROUP_ID_SHIFT	16
#define MEM_CGROUP_ID_MAX	USHRT_MAX

struct mem_cgroup_id {
	int id;
	atomic_t ref;
};

M
Michal Hocko 已提交
84 85 86 87 88 89 90 91 92 93 94 95 96 97
/*
 * Per memcg event counter is incremented at every pagein/pageout. With THP,
 * it will be incremated by the number of pages. This counter is used for
 * for trigger some periodic events. This is straightforward and better
 * than using jiffies etc. to handle periodic memcg event.
 */
enum mem_cgroup_events_target {
	MEM_CGROUP_TARGET_THRESH,
	MEM_CGROUP_TARGET_SOFTLIMIT,
	MEM_CGROUP_TARGET_NUMAINFO,
	MEM_CGROUP_NTARGETS,
};

struct mem_cgroup_stat_cpu {
98
	long count[MEMCG_NR_STAT];
99
	unsigned long events[NR_VM_EVENT_ITEMS];
M
Michal Hocko 已提交
100 101 102 103 104 105 106 107 108 109
	unsigned long nr_page_events;
	unsigned long targets[MEM_CGROUP_NTARGETS];
};

struct mem_cgroup_reclaim_iter {
	struct mem_cgroup *position;
	/* scan generation, increased every round-trip */
	unsigned int generation;
};

110 111 112 113
struct lruvec_stat {
	long count[NR_VM_NODE_STAT_ITEMS];
};

114 115 116 117 118 119 120 121 122
/*
 * Bitmap of shrinker::id corresponding to memcg-aware shrinkers,
 * which have elements charged to this memcg.
 */
struct memcg_shrinker_map {
	struct rcu_head rcu;
	unsigned long map[0];
};

M
Michal Hocko 已提交
123 124 125
/*
 * per-zone information in memory controller.
 */
126
struct mem_cgroup_per_node {
M
Michal Hocko 已提交
127
	struct lruvec		lruvec;
128 129 130 131

	struct lruvec_stat __percpu *lruvec_stat_cpu;
	atomic_long_t		lruvec_stat[NR_VM_NODE_STAT_ITEMS];

132
	unsigned long		lru_zone_size[MAX_NR_ZONES][NR_LRU_LISTS];
M
Michal Hocko 已提交
133 134 135

	struct mem_cgroup_reclaim_iter	iter[DEF_PRIORITY + 1];

136 137 138
#ifdef CONFIG_MEMCG_KMEM
	struct memcg_shrinker_map __rcu	*shrinker_map;
#endif
M
Michal Hocko 已提交
139 140 141 142
	struct rb_node		tree_node;	/* RB tree node */
	unsigned long		usage_in_excess;/* Set to the value by which */
						/* the soft limit is exceeded*/
	bool			on_tree;
143 144 145
	bool			congested;	/* memcg has many dirty pages */
						/* backed by a congested BDI */

M
Michal Hocko 已提交
146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175
	struct mem_cgroup	*memcg;		/* Back pointer, we cannot */
						/* use container_of	   */
};

struct mem_cgroup_threshold {
	struct eventfd_ctx *eventfd;
	unsigned long threshold;
};

/* For threshold */
struct mem_cgroup_threshold_ary {
	/* An array index points to threshold just below or equal to usage. */
	int current_threshold;
	/* Size of entries[] */
	unsigned int size;
	/* Array of thresholds */
	struct mem_cgroup_threshold entries[0];
};

struct mem_cgroup_thresholds {
	/* Primary thresholds array */
	struct mem_cgroup_threshold_ary *primary;
	/*
	 * Spare threshold array.
	 * This is needed to make mem_cgroup_unregister_event() "never fail".
	 * It must be able to store at least primary->size - 1 entries.
	 */
	struct mem_cgroup_threshold_ary *spare;
};

176 177 178 179 180 181
enum memcg_kmem_state {
	KMEM_NONE,
	KMEM_ALLOCATED,
	KMEM_ONLINE,
};

182 183 184 185 186 187 188 189 190
#if defined(CONFIG_SMP)
struct memcg_padding {
	char x[0];
} ____cacheline_internodealigned_in_smp;
#define MEMCG_PADDING(name)      struct memcg_padding name;
#else
#define MEMCG_PADDING(name)
#endif

M
Michal Hocko 已提交
191 192 193 194 195 196 197 198 199
/*
 * The memory controller data structure. The memory controller controls both
 * page cache and RSS per cgroup. We would eventually like to provide
 * statistics based on the statistics developed by Rik Van Riel for clock-pro,
 * to help the administrator determine what knobs to tune.
 */
struct mem_cgroup {
	struct cgroup_subsys_state css;

200 201 202
	/* Private memcg ID. Used to ID objects that outlive the cgroup */
	struct mem_cgroup_id id;

M
Michal Hocko 已提交
203 204
	/* Accounted resources */
	struct page_counter memory;
205
	struct page_counter swap;
206 207

	/* Legacy consumer-oriented counters */
M
Michal Hocko 已提交
208 209
	struct page_counter memsw;
	struct page_counter kmem;
210
	struct page_counter tcpmem;
M
Michal Hocko 已提交
211

212
	/* Upper bound of normal memory consumption range */
M
Michal Hocko 已提交
213 214
	unsigned long high;

215 216 217
	/* Range enforcement for interrupt charges */
	struct work_struct high_work;

M
Michal Hocko 已提交
218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235
	unsigned long soft_limit;

	/* vmpressure notifications */
	struct vmpressure vmpressure;

	/*
	 * Should the accounting and control be hierarchical, per subtree?
	 */
	bool use_hierarchy;

	/* protected by memcg_oom_lock */
	bool		oom_lock;
	int		under_oom;

	int	swappiness;
	/* OOM-Killer disable */
	int		oom_kill_disable;

236
	/* memory.events */
237 238
	struct cgroup_file events_file;

239 240 241
	/* handle for "memory.swap.events" */
	struct cgroup_file swap_events_file;

M
Michal Hocko 已提交
242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258
	/* protect arrays of thresholds */
	struct mutex thresholds_lock;

	/* thresholds for memory usage. RCU-protected */
	struct mem_cgroup_thresholds thresholds;

	/* thresholds for mem+swap usage. RCU-protected */
	struct mem_cgroup_thresholds memsw_thresholds;

	/* For oom notifier event fd */
	struct list_head oom_notify;

	/*
	 * Should we move charges of a task when a task is moved into this
	 * mem_cgroup ? And what type of charges should we move ?
	 */
	unsigned long move_charge_at_immigrate;
259 260 261 262 263 264
	/* taken only while moving_account > 0 */
	spinlock_t		move_lock;
	unsigned long		move_lock_flags;

	MEMCG_PADDING(_pad1_);

M
Michal Hocko 已提交
265 266 267 268 269
	/*
	 * set > 0 if pages under this cgroup are moving to other cgroup.
	 */
	atomic_t		moving_account;
	struct task_struct	*move_lock_task;
270

271
	/* memory.stat */
272
	struct mem_cgroup_stat_cpu __percpu *stat_cpu;
273 274 275

	MEMCG_PADDING(_pad2_);

276
	atomic_long_t		stat[MEMCG_NR_STAT];
277
	atomic_long_t		events[NR_VM_EVENT_ITEMS];
278
	atomic_long_t memory_events[MEMCG_NR_MEMORY_EVENTS];
M
Michal Hocko 已提交
279

280 281 282
	unsigned long		socket_pressure;

	/* Legacy tcp memory accounting */
283 284
	bool			tcpmem_active;
	int			tcpmem_pressure;
285

286
#ifdef CONFIG_MEMCG_KMEM
M
Michal Hocko 已提交
287 288
        /* Index in the kmem_cache->memcg_params.memcg_caches array */
	int kmemcg_id;
289
	enum memcg_kmem_state kmem_state;
290
	struct list_head kmem_caches;
M
Michal Hocko 已提交
291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311
#endif

	int last_scanned_node;
#if MAX_NUMNODES > 1
	nodemask_t	scan_nodes;
	atomic_t	numainfo_events;
	atomic_t	numainfo_updating;
#endif

#ifdef CONFIG_CGROUP_WRITEBACK
	struct list_head cgwb_list;
	struct wb_domain cgwb_domain;
#endif

	/* List of events which userspace want to receive */
	struct list_head event_list;
	spinlock_t event_list_lock;

	struct mem_cgroup_per_node *nodeinfo[0];
	/* WARNING: nodeinfo must be the last member here */
};
312

313 314 315 316 317 318
/*
 * size of first charge trial. "32" comes from vmscan.c's magic value.
 * TODO: maybe necessary to use big numbers in big irons.
 */
#define MEMCG_CHARGE_BATCH 32U

319
extern struct mem_cgroup *root_mem_cgroup;
T
Tejun Heo 已提交
320

321 322 323 324 325
static inline bool mem_cgroup_disabled(void)
{
	return !cgroup_subsys_enabled(memory_cgrp_subsys);
}

R
Roman Gushchin 已提交
326 327
enum mem_cgroup_protection mem_cgroup_protected(struct mem_cgroup *root,
						struct mem_cgroup *memcg);
328

329
int mem_cgroup_try_charge(struct page *page, struct mm_struct *mm,
330 331
			  gfp_t gfp_mask, struct mem_cgroup **memcgp,
			  bool compound);
332 333 334
int mem_cgroup_try_charge_delay(struct page *page, struct mm_struct *mm,
			  gfp_t gfp_mask, struct mem_cgroup **memcgp,
			  bool compound);
335
void mem_cgroup_commit_charge(struct page *page, struct mem_cgroup *memcg,
336 337 338
			      bool lrucare, bool compound);
void mem_cgroup_cancel_charge(struct page *page, struct mem_cgroup *memcg,
		bool compound);
339
void mem_cgroup_uncharge(struct page *page);
340
void mem_cgroup_uncharge_list(struct list_head *page_list);
341

342
void mem_cgroup_migrate(struct page *oldpage, struct page *newpage);
343

344 345
static struct mem_cgroup_per_node *
mem_cgroup_nodeinfo(struct mem_cgroup *memcg, int nid)
346
{
347
	return memcg->nodeinfo[nid];
348 349 350
}

/**
351 352
 * mem_cgroup_lruvec - get the lru list vector for a node or a memcg zone
 * @node: node of the wanted lruvec
353 354
 * @memcg: memcg of the wanted lruvec
 *
355 356
 * Returns the lru list vector holding pages for a given @node or a given
 * @memcg and @zone. This can be the node lruvec, if the memory controller
357 358
 * is disabled.
 */
359
static inline struct lruvec *mem_cgroup_lruvec(struct pglist_data *pgdat,
360
				struct mem_cgroup *memcg)
361
{
362
	struct mem_cgroup_per_node *mz;
363 364 365
	struct lruvec *lruvec;

	if (mem_cgroup_disabled()) {
366
		lruvec = node_lruvec(pgdat);
367 368 369
		goto out;
	}

370
	mz = mem_cgroup_nodeinfo(memcg, pgdat->node_id);
371 372 373 374
	lruvec = &mz->lruvec;
out:
	/*
	 * Since a node can be onlined after the mem_cgroup was created,
M
Mel Gorman 已提交
375
	 * we have to be prepared to initialize lruvec->pgdat here;
376 377
	 * and if offlined then reonlined, we need to reinitialize it.
	 */
378 379
	if (unlikely(lruvec->pgdat != pgdat))
		lruvec->pgdat = pgdat;
380 381 382
	return lruvec;
}

M
Mel Gorman 已提交
383
struct lruvec *mem_cgroup_page_lruvec(struct page *, struct pglist_data *);
384

385
bool task_in_mem_cgroup(struct task_struct *task, struct mem_cgroup *memcg);
386
struct mem_cgroup *mem_cgroup_from_task(struct task_struct *p);
387

388 389
struct mem_cgroup *get_mem_cgroup_from_mm(struct mm_struct *mm);

390 391
struct mem_cgroup *get_mem_cgroup_from_page(struct page *page);

M
Michal Hocko 已提交
392 393 394 395 396
static inline
struct mem_cgroup *mem_cgroup_from_css(struct cgroup_subsys_state *css){
	return css ? container_of(css, struct mem_cgroup, css) : NULL;
}

397 398
static inline void mem_cgroup_put(struct mem_cgroup *memcg)
{
399 400
	if (memcg)
		css_put(&memcg->css);
401 402
}

403 404 405
#define mem_cgroup_from_counter(counter, member)	\
	container_of(counter, struct mem_cgroup, member)

M
Michal Hocko 已提交
406 407 408 409
struct mem_cgroup *mem_cgroup_iter(struct mem_cgroup *,
				   struct mem_cgroup *,
				   struct mem_cgroup_reclaim_cookie *);
void mem_cgroup_iter_break(struct mem_cgroup *, struct mem_cgroup *);
410 411
int mem_cgroup_scan_tasks(struct mem_cgroup *,
			  int (*)(struct task_struct *, void *), void *);
M
Michal Hocko 已提交
412

413 414 415 416 417
static inline unsigned short mem_cgroup_id(struct mem_cgroup *memcg)
{
	if (mem_cgroup_disabled())
		return 0;

418
	return memcg->id.id;
419
}
420
struct mem_cgroup *mem_cgroup_from_id(unsigned short id);
421

422 423 424 425 426 427 428 429 430 431 432
static inline struct mem_cgroup *lruvec_memcg(struct lruvec *lruvec)
{
	struct mem_cgroup_per_node *mz;

	if (mem_cgroup_disabled())
		return NULL;

	mz = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
	return mz->memcg;
}

433 434 435 436 437 438 439 440 441 442 443 444 445 446
/**
 * parent_mem_cgroup - find the accounting parent of a memcg
 * @memcg: memcg whose parent to find
 *
 * Returns the parent memcg, or NULL if this is the root or the memory
 * controller is in legacy no-hierarchy mode.
 */
static inline struct mem_cgroup *parent_mem_cgroup(struct mem_cgroup *memcg)
{
	if (!memcg->memory.parent)
		return NULL;
	return mem_cgroup_from_counter(memcg->memory.parent, memory);
}

M
Michal Hocko 已提交
447 448 449 450 451 452 453 454 455
static inline bool mem_cgroup_is_descendant(struct mem_cgroup *memcg,
			      struct mem_cgroup *root)
{
	if (root == memcg)
		return true;
	if (!root->use_hierarchy)
		return false;
	return cgroup_is_descendant(memcg->css.cgroup, root->css.cgroup);
}
G
Glauber Costa 已提交
456

457 458
static inline bool mm_match_cgroup(struct mm_struct *mm,
				   struct mem_cgroup *memcg)
459
{
460
	struct mem_cgroup *task_memcg;
461
	bool match = false;
462

463
	rcu_read_lock();
464
	task_memcg = mem_cgroup_from_task(rcu_dereference(mm->owner));
465
	if (task_memcg)
466
		match = mem_cgroup_is_descendant(task_memcg, memcg);
467
	rcu_read_unlock();
468
	return match;
469
}
470

471
struct cgroup_subsys_state *mem_cgroup_css_from_page(struct page *page);
472
ino_t page_cgroup_ino(struct page *page);
473

474 475 476 477 478 479 480
static inline bool mem_cgroup_online(struct mem_cgroup *memcg)
{
	if (mem_cgroup_disabled())
		return true;
	return !!(memcg->css.flags & CSS_ONLINE);
}

481 482 483
/*
 * For memory reclaim.
 */
484
int mem_cgroup_select_victim_node(struct mem_cgroup *memcg);
M
Michal Hocko 已提交
485 486

void mem_cgroup_update_lru_size(struct lruvec *lruvec, enum lru_list lru,
487
		int zid, int nr_pages);
M
Michal Hocko 已提交
488

489 490 491
unsigned long mem_cgroup_node_nr_lru_pages(struct mem_cgroup *memcg,
					   int nid, unsigned int lru_mask);

M
Michal Hocko 已提交
492 493 494
static inline
unsigned long mem_cgroup_get_lru_size(struct lruvec *lruvec, enum lru_list lru)
{
495
	struct mem_cgroup_per_node *mz;
496 497
	unsigned long nr_pages = 0;
	int zid;
M
Michal Hocko 已提交
498

499
	mz = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
500 501 502 503 504 505 506 507 508 509 510 511 512
	for (zid = 0; zid < MAX_NR_ZONES; zid++)
		nr_pages += mz->lru_zone_size[zid][lru];
	return nr_pages;
}

static inline
unsigned long mem_cgroup_get_zone_lru_size(struct lruvec *lruvec,
		enum lru_list lru, int zone_idx)
{
	struct mem_cgroup_per_node *mz;

	mz = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
	return mz->lru_zone_size[zone_idx][lru];
M
Michal Hocko 已提交
513 514
}

515 516
void mem_cgroup_handle_over_high(void);

517
unsigned long mem_cgroup_get_max(struct mem_cgroup *memcg);
518

519 520
void mem_cgroup_print_oom_info(struct mem_cgroup *memcg,
				struct task_struct *p);
521

522
static inline void mem_cgroup_enter_user_fault(void)
523
{
524 525
	WARN_ON(current->in_user_fault);
	current->in_user_fault = 1;
526 527
}

528
static inline void mem_cgroup_exit_user_fault(void)
529
{
530 531
	WARN_ON(!current->in_user_fault);
	current->in_user_fault = 0;
532 533
}

534 535
static inline bool task_in_memcg_oom(struct task_struct *p)
{
T
Tejun Heo 已提交
536
	return p->memcg_in_oom;
537 538
}

539
bool mem_cgroup_oom_synchronize(bool wait);
540

A
Andrew Morton 已提交
541
#ifdef CONFIG_MEMCG_SWAP
542 543
extern int do_swap_account;
#endif
544

545 546
struct mem_cgroup *lock_page_memcg(struct page *page);
void __unlock_page_memcg(struct mem_cgroup *memcg);
J
Johannes Weiner 已提交
547
void unlock_page_memcg(struct page *page);
548

549
/* idx can be of type enum memcg_stat_item or node_stat_item */
550
static inline unsigned long memcg_page_state(struct mem_cgroup *memcg,
551
					     int idx)
552
{
553 554 555 556 557 558
	long x = atomic_long_read(&memcg->stat[idx]);
#ifdef CONFIG_SMP
	if (x < 0)
		x = 0;
#endif
	return x;
559 560
}

561
/* idx can be of type enum memcg_stat_item or node_stat_item */
562
static inline void __mod_memcg_state(struct mem_cgroup *memcg,
563
				     int idx, int val)
564
{
565 566 567 568 569 570 571 572 573 574 575
	long x;

	if (mem_cgroup_disabled())
		return;

	x = val + __this_cpu_read(memcg->stat_cpu->count[idx]);
	if (unlikely(abs(x) > MEMCG_CHARGE_BATCH)) {
		atomic_long_add(x, &memcg->stat[idx]);
		x = 0;
	}
	__this_cpu_write(memcg->stat_cpu->count[idx], x);
576 577
}

578
/* idx can be of type enum memcg_stat_item or node_stat_item */
579
static inline void mod_memcg_state(struct mem_cgroup *memcg,
580
				   int idx, int val)
581
{
582 583 584
	unsigned long flags;

	local_irq_save(flags);
585
	__mod_memcg_state(memcg, idx, val);
586
	local_irq_restore(flags);
587 588
}

M
Michal Hocko 已提交
589
/**
590
 * mod_memcg_page_state - update page state statistics
J
Johannes Weiner 已提交
591
 * @page: the page
M
Michal Hocko 已提交
592 593 594
 * @idx: page state item to account
 * @val: number of pages (positive or negative)
 *
595 596 597
 * The @page must be locked or the caller must use lock_page_memcg()
 * to prevent double accounting when the page is concurrently being
 * moved to another memcg:
598
 *
599
 *   lock_page(page) or lock_page_memcg(page)
600
 *   if (TestClearPageState(page))
601
 *     mod_memcg_page_state(page, state, -1);
602
 *   unlock_page(page) or unlock_page_memcg(page)
603 604
 *
 * Kernel pages are an exception to this, since they'll never move.
M
Michal Hocko 已提交
605
 */
606
static inline void __mod_memcg_page_state(struct page *page,
607
					  int idx, int val)
608 609 610 611 612
{
	if (page->mem_cgroup)
		__mod_memcg_state(page->mem_cgroup, idx, val);
}

613
static inline void mod_memcg_page_state(struct page *page,
614
					int idx, int val)
M
Michal Hocko 已提交
615
{
J
Johannes Weiner 已提交
616
	if (page->mem_cgroup)
617
		mod_memcg_state(page->mem_cgroup, idx, val);
M
Michal Hocko 已提交
618 619
}

620 621
static inline unsigned long lruvec_page_state(struct lruvec *lruvec,
					      enum node_stat_item idx)
622
{
623
	struct mem_cgroup_per_node *pn;
624
	long x;
625 626 627 628 629

	if (mem_cgroup_disabled())
		return node_page_state(lruvec_pgdat(lruvec), idx);

	pn = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
630 631 632 633 634 635
	x = atomic_long_read(&pn->lruvec_stat[idx]);
#ifdef CONFIG_SMP
	if (x < 0)
		x = 0;
#endif
	return x;
636 637
}

638 639
static inline void __mod_lruvec_state(struct lruvec *lruvec,
				      enum node_stat_item idx, int val)
640
{
641
	struct mem_cgroup_per_node *pn;
642
	long x;
643

644
	/* Update node */
645
	__mod_node_page_state(lruvec_pgdat(lruvec), idx, val);
646

647 648
	if (mem_cgroup_disabled())
		return;
649

650
	pn = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
651 652

	/* Update memcg */
653
	__mod_memcg_state(pn->memcg, idx, val);
654 655

	/* Update lruvec */
656 657 658 659 660 661
	x = val + __this_cpu_read(pn->lruvec_stat_cpu->count[idx]);
	if (unlikely(abs(x) > MEMCG_CHARGE_BATCH)) {
		atomic_long_add(x, &pn->lruvec_stat[idx]);
		x = 0;
	}
	__this_cpu_write(pn->lruvec_stat_cpu->count[idx], x);
662 663 664 665 666
}

static inline void mod_lruvec_state(struct lruvec *lruvec,
				    enum node_stat_item idx, int val)
{
667 668 669
	unsigned long flags;

	local_irq_save(flags);
670
	__mod_lruvec_state(lruvec, idx, val);
671
	local_irq_restore(flags);
672 673 674 675 676
}

static inline void __mod_lruvec_page_state(struct page *page,
					   enum node_stat_item idx, int val)
{
677 678
	pg_data_t *pgdat = page_pgdat(page);
	struct lruvec *lruvec;
679

680 681 682
	/* Untracked pages have no memcg, no lruvec. Update only the node */
	if (!page->mem_cgroup) {
		__mod_node_page_state(pgdat, idx, val);
683
		return;
684 685 686 687
	}

	lruvec = mem_cgroup_lruvec(pgdat, page->mem_cgroup);
	__mod_lruvec_state(lruvec, idx, val);
688 689 690 691 692
}

static inline void mod_lruvec_page_state(struct page *page,
					 enum node_stat_item idx, int val)
{
693 694 695
	unsigned long flags;

	local_irq_save(flags);
696
	__mod_lruvec_page_state(page, idx, val);
697
	local_irq_restore(flags);
698 699
}

700
unsigned long mem_cgroup_soft_limit_reclaim(pg_data_t *pgdat, int order,
701 702
						gfp_t gfp_mask,
						unsigned long *total_scanned);
D
David Rientjes 已提交
703

704
static inline void __count_memcg_events(struct mem_cgroup *memcg,
705 706
					enum vm_event_item idx,
					unsigned long count)
707
{
708 709 710 711 712 713 714 715 716 717 718
	unsigned long x;

	if (mem_cgroup_disabled())
		return;

	x = count + __this_cpu_read(memcg->stat_cpu->events[idx]);
	if (unlikely(x > MEMCG_CHARGE_BATCH)) {
		atomic_long_add(x, &memcg->events[idx]);
		x = 0;
	}
	__this_cpu_write(memcg->stat_cpu->events[idx], x);
719 720
}

721
static inline void count_memcg_events(struct mem_cgroup *memcg,
722 723
				      enum vm_event_item idx,
				      unsigned long count)
724
{
725 726 727
	unsigned long flags;

	local_irq_save(flags);
728
	__count_memcg_events(memcg, idx, count);
729
	local_irq_restore(flags);
730 731 732
}

static inline void count_memcg_page_event(struct page *page,
733
					  enum vm_event_item idx)
734 735 736 737 738 739 740
{
	if (page->mem_cgroup)
		count_memcg_events(page->mem_cgroup, idx, 1);
}

static inline void count_memcg_event_mm(struct mm_struct *mm,
					enum vm_event_item idx)
741
{
M
Michal Hocko 已提交
742 743
	struct mem_cgroup *memcg;

744 745
	if (mem_cgroup_disabled())
		return;
M
Michal Hocko 已提交
746 747 748

	rcu_read_lock();
	memcg = mem_cgroup_from_task(rcu_dereference(mm->owner));
R
Roman Gushchin 已提交
749
	if (likely(memcg))
750
		count_memcg_events(memcg, idx, 1);
M
Michal Hocko 已提交
751
	rcu_read_unlock();
752
}
753

754 755
static inline void memcg_memory_event(struct mem_cgroup *memcg,
				      enum memcg_memory_event event)
756
{
757
	atomic_long_inc(&memcg->memory_events[event]);
758 759 760
	cgroup_file_notify(&memcg->events_file);
}

R
Roman Gushchin 已提交
761 762 763 764 765 766 767 768 769 770 771 772 773 774 775
static inline void memcg_memory_event_mm(struct mm_struct *mm,
					 enum memcg_memory_event event)
{
	struct mem_cgroup *memcg;

	if (mem_cgroup_disabled())
		return;

	rcu_read_lock();
	memcg = mem_cgroup_from_task(rcu_dereference(mm->owner));
	if (likely(memcg))
		memcg_memory_event(memcg, event);
	rcu_read_unlock();
}

776
#ifdef CONFIG_TRANSPARENT_HUGEPAGE
777
void mem_cgroup_split_huge_fixup(struct page *head);
778 779
#endif

A
Andrew Morton 已提交
780
#else /* CONFIG_MEMCG */
781 782 783 784

#define MEM_CGROUP_ID_SHIFT	0
#define MEM_CGROUP_ID_MAX	0

785 786
struct mem_cgroup;

787 788 789 790 791
static inline bool mem_cgroup_disabled(void)
{
	return true;
}

792 793
static inline void memcg_memory_event(struct mem_cgroup *memcg,
				      enum memcg_memory_event event)
794 795 796
{
}

R
Roman Gushchin 已提交
797 798 799 800 801
static inline void memcg_memory_event_mm(struct mm_struct *mm,
					 enum memcg_memory_event event)
{
}

R
Roman Gushchin 已提交
802 803
static inline enum mem_cgroup_protection mem_cgroup_protected(
	struct mem_cgroup *root, struct mem_cgroup *memcg)
804
{
R
Roman Gushchin 已提交
805
	return MEMCG_PROT_NONE;
806 807
}

808 809
static inline int mem_cgroup_try_charge(struct page *page, struct mm_struct *mm,
					gfp_t gfp_mask,
810 811
					struct mem_cgroup **memcgp,
					bool compound)
812
{
813
	*memcgp = NULL;
814 815 816
	return 0;
}

817 818 819 820 821 822 823 824 825 826
static inline int mem_cgroup_try_charge_delay(struct page *page,
					      struct mm_struct *mm,
					      gfp_t gfp_mask,
					      struct mem_cgroup **memcgp,
					      bool compound)
{
	*memcgp = NULL;
	return 0;
}

827 828
static inline void mem_cgroup_commit_charge(struct page *page,
					    struct mem_cgroup *memcg,
829
					    bool lrucare, bool compound)
830 831 832
{
}

833
static inline void mem_cgroup_cancel_charge(struct page *page,
834 835
					    struct mem_cgroup *memcg,
					    bool compound)
836 837 838
{
}

839
static inline void mem_cgroup_uncharge(struct page *page)
840 841 842
{
}

843
static inline void mem_cgroup_uncharge_list(struct list_head *page_list)
844 845 846
{
}

847
static inline void mem_cgroup_migrate(struct page *old, struct page *new)
848 849 850
{
}

851
static inline struct lruvec *mem_cgroup_lruvec(struct pglist_data *pgdat,
852
				struct mem_cgroup *memcg)
K
KAMEZAWA Hiroyuki 已提交
853
{
854
	return node_lruvec(pgdat);
K
KAMEZAWA Hiroyuki 已提交
855 856
}

857
static inline struct lruvec *mem_cgroup_page_lruvec(struct page *page,
M
Mel Gorman 已提交
858
						    struct pglist_data *pgdat)
859
{
M
Mel Gorman 已提交
860
	return &pgdat->lruvec;
861 862
}

863
static inline bool mm_match_cgroup(struct mm_struct *mm,
864
		struct mem_cgroup *memcg)
865
{
866
	return true;
867 868
}

869 870
static inline bool task_in_mem_cgroup(struct task_struct *task,
				      const struct mem_cgroup *memcg)
871
{
872
	return true;
873 874
}

875 876 877 878 879
static inline struct mem_cgroup *get_mem_cgroup_from_mm(struct mm_struct *mm)
{
	return NULL;
}

880 881 882 883 884
static inline struct mem_cgroup *get_mem_cgroup_from_page(struct page *page)
{
	return NULL;
}

885 886 887 888
static inline void mem_cgroup_put(struct mem_cgroup *memcg)
{
}

889 890 891 892 893 894 895 896 897 898 899 900 901
static inline struct mem_cgroup *
mem_cgroup_iter(struct mem_cgroup *root,
		struct mem_cgroup *prev,
		struct mem_cgroup_reclaim_cookie *reclaim)
{
	return NULL;
}

static inline void mem_cgroup_iter_break(struct mem_cgroup *root,
					 struct mem_cgroup *prev)
{
}

902 903 904 905 906 907
static inline int mem_cgroup_scan_tasks(struct mem_cgroup *memcg,
		int (*fn)(struct task_struct *, void *), void *arg)
{
	return 0;
}

908
static inline unsigned short mem_cgroup_id(struct mem_cgroup *memcg)
909
{
910 911 912 913 914 915 916 917
	return 0;
}

static inline struct mem_cgroup *mem_cgroup_from_id(unsigned short id)
{
	WARN_ON_ONCE(id);
	/* XXX: This should always return root_mem_cgroup */
	return NULL;
918
}
919

920 921 922 923 924
static inline struct mem_cgroup *lruvec_memcg(struct lruvec *lruvec)
{
	return NULL;
}

925
static inline bool mem_cgroup_online(struct mem_cgroup *memcg)
926
{
927
	return true;
928 929
}

930
static inline unsigned long
931
mem_cgroup_get_lru_size(struct lruvec *lruvec, enum lru_list lru)
932 933 934
{
	return 0;
}
935 936 937 938 939 940
static inline
unsigned long mem_cgroup_get_zone_lru_size(struct lruvec *lruvec,
		enum lru_list lru, int zone_idx)
{
	return 0;
}
941

942 943 944 945 946 947 948
static inline unsigned long
mem_cgroup_node_nr_lru_pages(struct mem_cgroup *memcg,
			     int nid, unsigned int lru_mask)
{
	return 0;
}

949
static inline unsigned long mem_cgroup_get_max(struct mem_cgroup *memcg)
950 951 952 953
{
	return 0;
}

954 955 956 957 958
static inline void
mem_cgroup_print_oom_info(struct mem_cgroup *memcg, struct task_struct *p)
{
}

959 960 961 962 963 964
static inline struct mem_cgroup *lock_page_memcg(struct page *page)
{
	return NULL;
}

static inline void __unlock_page_memcg(struct mem_cgroup *memcg)
965 966 967
{
}

J
Johannes Weiner 已提交
968
static inline void unlock_page_memcg(struct page *page)
969 970 971
{
}

972 973 974 975
static inline void mem_cgroup_handle_over_high(void)
{
}

976
static inline void mem_cgroup_enter_user_fault(void)
977 978 979
{
}

980
static inline void mem_cgroup_exit_user_fault(void)
981 982 983
{
}

984 985 986 987 988
static inline bool task_in_memcg_oom(struct task_struct *p)
{
	return false;
}

989
static inline bool mem_cgroup_oom_synchronize(bool wait)
990 991 992 993
{
	return false;
}

994
static inline unsigned long memcg_page_state(struct mem_cgroup *memcg,
995
					     int idx)
996 997 998 999
{
	return 0;
}

1000
static inline void __mod_memcg_state(struct mem_cgroup *memcg,
1001
				     int idx,
1002
				     int nr)
1003 1004 1005
{
}

1006
static inline void mod_memcg_state(struct mem_cgroup *memcg,
1007
				   int idx,
1008
				   int nr)
1009 1010 1011
{
}

1012
static inline void __mod_memcg_page_state(struct page *page,
1013
					  int idx,
1014
					  int nr)
1015 1016 1017
{
}

1018
static inline void mod_memcg_page_state(struct page *page,
1019
					int idx,
1020
					int nr)
1021 1022 1023
{
}

1024 1025
static inline unsigned long lruvec_page_state(struct lruvec *lruvec,
					      enum node_stat_item idx)
1026
{
1027
	return node_page_state(lruvec_pgdat(lruvec), idx);
1028 1029
}

1030 1031
static inline void __mod_lruvec_state(struct lruvec *lruvec,
				      enum node_stat_item idx, int val)
1032
{
1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051
	__mod_node_page_state(lruvec_pgdat(lruvec), idx, val);
}

static inline void mod_lruvec_state(struct lruvec *lruvec,
				    enum node_stat_item idx, int val)
{
	mod_node_page_state(lruvec_pgdat(lruvec), idx, val);
}

static inline void __mod_lruvec_page_state(struct page *page,
					   enum node_stat_item idx, int val)
{
	__mod_node_page_state(page_pgdat(page), idx, val);
}

static inline void mod_lruvec_page_state(struct page *page,
					 enum node_stat_item idx, int val)
{
	mod_node_page_state(page_pgdat(page), idx, val);
1052 1053
}

1054
static inline
1055
unsigned long mem_cgroup_soft_limit_reclaim(pg_data_t *pgdat, int order,
1056 1057
					    gfp_t gfp_mask,
					    unsigned long *total_scanned)
1058
{
1059
	return 0;
1060 1061
}

1062
static inline void mem_cgroup_split_huge_fixup(struct page *head)
1063 1064 1065
{
}

1066 1067 1068 1069 1070 1071 1072
static inline void count_memcg_events(struct mem_cgroup *memcg,
				      enum vm_event_item idx,
				      unsigned long count)
{
}

static inline void count_memcg_page_event(struct page *page,
1073
					  int idx)
1074 1075 1076
{
}

1077
static inline
1078
void count_memcg_event_mm(struct mm_struct *mm, enum vm_event_item idx)
1079 1080
{
}
A
Andrew Morton 已提交
1081
#endif /* CONFIG_MEMCG */
1082

1083
/* idx can be of type enum memcg_stat_item or node_stat_item */
1084
static inline void __inc_memcg_state(struct mem_cgroup *memcg,
1085
				     int idx)
1086 1087 1088 1089
{
	__mod_memcg_state(memcg, idx, 1);
}

1090
/* idx can be of type enum memcg_stat_item or node_stat_item */
1091
static inline void __dec_memcg_state(struct mem_cgroup *memcg,
1092
				     int idx)
1093 1094 1095 1096
{
	__mod_memcg_state(memcg, idx, -1);
}

1097
/* idx can be of type enum memcg_stat_item or node_stat_item */
1098
static inline void __inc_memcg_page_state(struct page *page,
1099
					  int idx)
1100 1101 1102 1103
{
	__mod_memcg_page_state(page, idx, 1);
}

1104
/* idx can be of type enum memcg_stat_item or node_stat_item */
1105
static inline void __dec_memcg_page_state(struct page *page,
1106
					  int idx)
1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134
{
	__mod_memcg_page_state(page, idx, -1);
}

static inline void __inc_lruvec_state(struct lruvec *lruvec,
				      enum node_stat_item idx)
{
	__mod_lruvec_state(lruvec, idx, 1);
}

static inline void __dec_lruvec_state(struct lruvec *lruvec,
				      enum node_stat_item idx)
{
	__mod_lruvec_state(lruvec, idx, -1);
}

static inline void __inc_lruvec_page_state(struct page *page,
					   enum node_stat_item idx)
{
	__mod_lruvec_page_state(page, idx, 1);
}

static inline void __dec_lruvec_page_state(struct page *page,
					   enum node_stat_item idx)
{
	__mod_lruvec_page_state(page, idx, -1);
}

1135
/* idx can be of type enum memcg_stat_item or node_stat_item */
1136
static inline void inc_memcg_state(struct mem_cgroup *memcg,
1137
				   int idx)
1138 1139 1140 1141
{
	mod_memcg_state(memcg, idx, 1);
}

1142
/* idx can be of type enum memcg_stat_item or node_stat_item */
1143
static inline void dec_memcg_state(struct mem_cgroup *memcg,
1144
				   int idx)
1145 1146 1147 1148
{
	mod_memcg_state(memcg, idx, -1);
}

1149
/* idx can be of type enum memcg_stat_item or node_stat_item */
1150
static inline void inc_memcg_page_state(struct page *page,
1151
					int idx)
1152 1153 1154 1155
{
	mod_memcg_page_state(page, idx, 1);
}

1156
/* idx can be of type enum memcg_stat_item or node_stat_item */
1157
static inline void dec_memcg_page_state(struct page *page,
1158
					int idx)
1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186
{
	mod_memcg_page_state(page, idx, -1);
}

static inline void inc_lruvec_state(struct lruvec *lruvec,
				    enum node_stat_item idx)
{
	mod_lruvec_state(lruvec, idx, 1);
}

static inline void dec_lruvec_state(struct lruvec *lruvec,
				    enum node_stat_item idx)
{
	mod_lruvec_state(lruvec, idx, -1);
}

static inline void inc_lruvec_page_state(struct page *page,
					 enum node_stat_item idx)
{
	mod_lruvec_page_state(page, idx, 1);
}

static inline void dec_lruvec_page_state(struct page *page,
					 enum node_stat_item idx)
{
	mod_lruvec_page_state(page, idx, -1);
}

1187
#ifdef CONFIG_CGROUP_WRITEBACK
T
Tejun Heo 已提交
1188 1189

struct wb_domain *mem_cgroup_wb_domain(struct bdi_writeback *wb);
1190 1191 1192
void mem_cgroup_wb_stats(struct bdi_writeback *wb, unsigned long *pfilepages,
			 unsigned long *pheadroom, unsigned long *pdirty,
			 unsigned long *pwriteback);
T
Tejun Heo 已提交
1193 1194 1195 1196 1197 1198 1199 1200

#else	/* CONFIG_CGROUP_WRITEBACK */

static inline struct wb_domain *mem_cgroup_wb_domain(struct bdi_writeback *wb)
{
	return NULL;
}

1201
static inline void mem_cgroup_wb_stats(struct bdi_writeback *wb,
1202 1203
				       unsigned long *pfilepages,
				       unsigned long *pheadroom,
1204 1205 1206 1207 1208
				       unsigned long *pdirty,
				       unsigned long *pwriteback)
{
}

T
Tejun Heo 已提交
1209
#endif	/* CONFIG_CGROUP_WRITEBACK */
1210

G
Glauber Costa 已提交
1211
struct sock;
1212 1213
bool mem_cgroup_charge_skmem(struct mem_cgroup *memcg, unsigned int nr_pages);
void mem_cgroup_uncharge_skmem(struct mem_cgroup *memcg, unsigned int nr_pages);
1214
#ifdef CONFIG_MEMCG
1215 1216
extern struct static_key_false memcg_sockets_enabled_key;
#define mem_cgroup_sockets_enabled static_branch_unlikely(&memcg_sockets_enabled_key)
1217 1218
void mem_cgroup_sk_alloc(struct sock *sk);
void mem_cgroup_sk_free(struct sock *sk);
1219
static inline bool mem_cgroup_under_socket_pressure(struct mem_cgroup *memcg)
1220
{
1221
	if (!cgroup_subsys_on_dfl(memory_cgrp_subsys) && memcg->tcpmem_pressure)
1222 1223 1224 1225 1226 1227
		return true;
	do {
		if (time_before(jiffies, memcg->socket_pressure))
			return true;
	} while ((memcg = parent_mem_cgroup(memcg)));
	return false;
1228 1229
}
#else
1230
#define mem_cgroup_sockets_enabled 0
1231 1232
static inline void mem_cgroup_sk_alloc(struct sock *sk) { };
static inline void mem_cgroup_sk_free(struct sock *sk) { };
1233
static inline bool mem_cgroup_under_socket_pressure(struct mem_cgroup *memcg)
1234 1235 1236 1237
{
	return false;
}
#endif
1238

1239 1240 1241 1242 1243 1244 1245
struct kmem_cache *memcg_kmem_get_cache(struct kmem_cache *cachep);
void memcg_kmem_put_cache(struct kmem_cache *cachep);
int memcg_kmem_charge_memcg(struct page *page, gfp_t gfp, int order,
			    struct mem_cgroup *memcg);
int memcg_kmem_charge(struct page *page, gfp_t gfp, int order);
void memcg_kmem_uncharge(struct page *page, int order);

1246
#ifdef CONFIG_MEMCG_KMEM
1247
extern struct static_key_false memcg_kmem_enabled_key;
1248
extern struct workqueue_struct *memcg_kmem_cache_wq;
1249

1250
extern int memcg_nr_cache_ids;
1251 1252
void memcg_get_cache_ids(void);
void memcg_put_cache_ids(void);
1253 1254 1255 1256 1257 1258

/*
 * Helper macro to loop through all memcg-specific caches. Callers must still
 * check if the cache is valid (it is either valid or NULL).
 * the slab_mutex must be held when looping through those caches
 */
1259
#define for_each_memcg_cache_index(_idx)	\
1260
	for ((_idx) = 0; (_idx) < memcg_nr_cache_ids; (_idx)++)
1261

1262 1263
static inline bool memcg_kmem_enabled(void)
{
1264
	return static_branch_unlikely(&memcg_kmem_enabled_key);
1265 1266
}

M
Michal Hocko 已提交
1267
/*
J
Jesper Dangaard Brouer 已提交
1268
 * helper for accessing a memcg's index. It will be used as an index in the
M
Michal Hocko 已提交
1269 1270 1271 1272 1273 1274 1275
 * child cache array in kmem_cache, and also to derive its name. This function
 * will return -1 when this is not a kmem-limited memcg.
 */
static inline int memcg_cache_id(struct mem_cgroup *memcg)
{
	return memcg ? memcg->kmemcg_id : -1;
}
1276

1277 1278
extern int memcg_expand_shrinker_maps(int new_id);

1279
#else
1280 1281 1282
#define for_each_memcg_cache_index(_idx)	\
	for (; NULL; )

1283 1284 1285 1286 1287
static inline bool memcg_kmem_enabled(void)
{
	return false;
}

1288 1289 1290 1291 1292
static inline int memcg_cache_id(struct mem_cgroup *memcg)
{
	return -1;
}

1293 1294 1295 1296 1297 1298 1299 1300
static inline void memcg_get_cache_ids(void)
{
}

static inline void memcg_put_cache_ids(void)
{
}

1301
#endif /* CONFIG_MEMCG_KMEM */
1302

B
Balbir Singh 已提交
1303
#endif /* _LINUX_MEMCONTROL_H */