memcontrol.h 29.2 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,
56 57
	MEMCG_SWAP_MAX,
	MEMCG_SWAP_FAIL,
58
	MEMCG_NR_MEMORY_EVENTS,
59 60
};

61
struct mem_cgroup_reclaim_cookie {
62
	pg_data_t *pgdat;
63 64 65 66
	int priority;
	unsigned int generation;
};

67 68 69 70 71 72 73 74 75 76
#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 已提交
77 78 79 80 81 82 83 84 85 86 87 88 89 90
/*
 * 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 {
91
	long count[MEMCG_NR_STAT];
92
	unsigned long events[NR_VM_EVENT_ITEMS];
M
Michal Hocko 已提交
93 94 95 96 97 98 99 100 101 102
	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;
};

103 104 105 106
struct lruvec_stat {
	long count[NR_VM_NODE_STAT_ITEMS];
};

M
Michal Hocko 已提交
107 108 109
/*
 * per-zone information in memory controller.
 */
110
struct mem_cgroup_per_node {
M
Michal Hocko 已提交
111
	struct lruvec		lruvec;
112 113 114 115

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

116
	unsigned long		lru_zone_size[MAX_NR_ZONES][NR_LRU_LISTS];
M
Michal Hocko 已提交
117 118 119 120 121 122 123

	struct mem_cgroup_reclaim_iter	iter[DEF_PRIORITY + 1];

	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;
124 125 126
	bool			congested;	/* memcg has many dirty pages */
						/* backed by a congested BDI */

M
Michal Hocko 已提交
127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156
	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;
};

157 158 159 160 161 162
enum memcg_kmem_state {
	KMEM_NONE,
	KMEM_ALLOCATED,
	KMEM_ONLINE,
};

M
Michal Hocko 已提交
163 164 165 166 167 168 169 170 171
/*
 * 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;

172 173 174
	/* Private memcg ID. Used to ID objects that outlive the cgroup */
	struct mem_cgroup_id id;

M
Michal Hocko 已提交
175 176
	/* Accounted resources */
	struct page_counter memory;
177
	struct page_counter swap;
178 179

	/* Legacy consumer-oriented counters */
M
Michal Hocko 已提交
180 181
	struct page_counter memsw;
	struct page_counter kmem;
182
	struct page_counter tcpmem;
M
Michal Hocko 已提交
183 184 185 186 187

	/* Normal memory consumption range */
	unsigned long low;
	unsigned long high;

188 189 190
	/* Range enforcement for interrupt charges */
	struct work_struct high_work;

M
Michal Hocko 已提交
191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208
	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;

209 210
	/* memory.events */
	atomic_long_t memory_events[MEMCG_NR_MEMORY_EVENTS];
211 212
	struct cgroup_file events_file;

213 214 215
	/* handle for "memory.swap.events" */
	struct cgroup_file swap_events_file;

M
Michal Hocko 已提交
216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240
	/* 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;
	/*
	 * set > 0 if pages under this cgroup are moving to other cgroup.
	 */
	atomic_t		moving_account;
	/* taken only while moving_account > 0 */
	spinlock_t		move_lock;
	struct task_struct	*move_lock_task;
	unsigned long		move_lock_flags;
241

242
	/* memory.stat */
243 244
	struct mem_cgroup_stat_cpu __percpu *stat_cpu;
	atomic_long_t		stat[MEMCG_NR_STAT];
245
	atomic_long_t		events[NR_VM_EVENT_ITEMS];
M
Michal Hocko 已提交
246

247 248 249
	unsigned long		socket_pressure;

	/* Legacy tcp memory accounting */
250 251
	bool			tcpmem_active;
	int			tcpmem_pressure;
252

253
#ifndef CONFIG_SLOB
M
Michal Hocko 已提交
254 255
        /* Index in the kmem_cache->memcg_params.memcg_caches array */
	int kmemcg_id;
256
	enum memcg_kmem_state kmem_state;
257
	struct list_head kmem_caches;
M
Michal Hocko 已提交
258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278
#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 */
};
279

280 281 282 283 284 285
/*
 * 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

286
extern struct mem_cgroup *root_mem_cgroup;
T
Tejun Heo 已提交
287

288 289 290 291 292
static inline bool mem_cgroup_disabled(void)
{
	return !cgroup_subsys_enabled(memory_cgrp_subsys);
}

293 294
bool mem_cgroup_low(struct mem_cgroup *root, struct mem_cgroup *memcg);

295
int mem_cgroup_try_charge(struct page *page, struct mm_struct *mm,
296 297
			  gfp_t gfp_mask, struct mem_cgroup **memcgp,
			  bool compound);
298
void mem_cgroup_commit_charge(struct page *page, struct mem_cgroup *memcg,
299 300 301
			      bool lrucare, bool compound);
void mem_cgroup_cancel_charge(struct page *page, struct mem_cgroup *memcg,
		bool compound);
302
void mem_cgroup_uncharge(struct page *page);
303
void mem_cgroup_uncharge_list(struct list_head *page_list);
304

305
void mem_cgroup_migrate(struct page *oldpage, struct page *newpage);
306

307 308
static struct mem_cgroup_per_node *
mem_cgroup_nodeinfo(struct mem_cgroup *memcg, int nid)
309
{
310
	return memcg->nodeinfo[nid];
311 312 313
}

/**
314 315
 * mem_cgroup_lruvec - get the lru list vector for a node or a memcg zone
 * @node: node of the wanted lruvec
316 317
 * @memcg: memcg of the wanted lruvec
 *
318 319
 * 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
320 321
 * is disabled.
 */
322
static inline struct lruvec *mem_cgroup_lruvec(struct pglist_data *pgdat,
323
				struct mem_cgroup *memcg)
324
{
325
	struct mem_cgroup_per_node *mz;
326 327 328
	struct lruvec *lruvec;

	if (mem_cgroup_disabled()) {
329
		lruvec = node_lruvec(pgdat);
330 331 332
		goto out;
	}

333
	mz = mem_cgroup_nodeinfo(memcg, pgdat->node_id);
334 335 336 337
	lruvec = &mz->lruvec;
out:
	/*
	 * Since a node can be onlined after the mem_cgroup was created,
M
Mel Gorman 已提交
338
	 * we have to be prepared to initialize lruvec->pgdat here;
339 340
	 * and if offlined then reonlined, we need to reinitialize it.
	 */
341 342
	if (unlikely(lruvec->pgdat != pgdat))
		lruvec->pgdat = pgdat;
343 344 345
	return lruvec;
}

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

348
bool task_in_mem_cgroup(struct task_struct *task, struct mem_cgroup *memcg);
349
struct mem_cgroup *mem_cgroup_from_task(struct task_struct *p);
350

M
Michal Hocko 已提交
351 352 353 354 355
static inline
struct mem_cgroup *mem_cgroup_from_css(struct cgroup_subsys_state *css){
	return css ? container_of(css, struct mem_cgroup, css) : NULL;
}

356 357 358
#define mem_cgroup_from_counter(counter, member)	\
	container_of(counter, struct mem_cgroup, member)

M
Michal Hocko 已提交
359 360 361 362
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 *);
363 364
int mem_cgroup_scan_tasks(struct mem_cgroup *,
			  int (*)(struct task_struct *, void *), void *);
M
Michal Hocko 已提交
365

366 367 368 369 370
static inline unsigned short mem_cgroup_id(struct mem_cgroup *memcg)
{
	if (mem_cgroup_disabled())
		return 0;

371
	return memcg->id.id;
372
}
373
struct mem_cgroup *mem_cgroup_from_id(unsigned short id);
374

375 376 377 378 379 380 381 382 383 384 385
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;
}

386 387 388 389 390 391 392 393 394 395 396 397 398 399
/**
 * 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 已提交
400 401 402 403 404 405 406 407 408
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 已提交
409

410 411
static inline bool mm_match_cgroup(struct mm_struct *mm,
				   struct mem_cgroup *memcg)
412
{
413
	struct mem_cgroup *task_memcg;
414
	bool match = false;
415

416
	rcu_read_lock();
417
	task_memcg = mem_cgroup_from_task(rcu_dereference(mm->owner));
418
	if (task_memcg)
419
		match = mem_cgroup_is_descendant(task_memcg, memcg);
420
	rcu_read_unlock();
421
	return match;
422
}
423

424
struct cgroup_subsys_state *mem_cgroup_css_from_page(struct page *page);
425
ino_t page_cgroup_ino(struct page *page);
426

427 428 429 430 431 432 433
static inline bool mem_cgroup_online(struct mem_cgroup *memcg)
{
	if (mem_cgroup_disabled())
		return true;
	return !!(memcg->css.flags & CSS_ONLINE);
}

434 435 436
/*
 * For memory reclaim.
 */
437
int mem_cgroup_select_victim_node(struct mem_cgroup *memcg);
M
Michal Hocko 已提交
438 439

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

442 443 444
unsigned long mem_cgroup_node_nr_lru_pages(struct mem_cgroup *memcg,
					   int nid, unsigned int lru_mask);

M
Michal Hocko 已提交
445 446 447
static inline
unsigned long mem_cgroup_get_lru_size(struct lruvec *lruvec, enum lru_list lru)
{
448
	struct mem_cgroup_per_node *mz;
449 450
	unsigned long nr_pages = 0;
	int zid;
M
Michal Hocko 已提交
451

452
	mz = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
453 454 455 456 457 458 459 460 461 462 463 464 465
	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 已提交
466 467
}

468 469
void mem_cgroup_handle_over_high(void);

470 471
unsigned long mem_cgroup_get_limit(struct mem_cgroup *memcg);

472 473
void mem_cgroup_print_oom_info(struct mem_cgroup *memcg,
				struct task_struct *p);
474

475
static inline void mem_cgroup_oom_enable(void)
476
{
T
Tejun Heo 已提交
477 478
	WARN_ON(current->memcg_may_oom);
	current->memcg_may_oom = 1;
479 480
}

481
static inline void mem_cgroup_oom_disable(void)
482
{
T
Tejun Heo 已提交
483 484
	WARN_ON(!current->memcg_may_oom);
	current->memcg_may_oom = 0;
485 486
}

487 488
static inline bool task_in_memcg_oom(struct task_struct *p)
{
T
Tejun Heo 已提交
489
	return p->memcg_in_oom;
490 491
}

492
bool mem_cgroup_oom_synchronize(bool wait);
493

A
Andrew Morton 已提交
494
#ifdef CONFIG_MEMCG_SWAP
495 496
extern int do_swap_account;
#endif
497

498 499
struct mem_cgroup *lock_page_memcg(struct page *page);
void __unlock_page_memcg(struct mem_cgroup *memcg);
J
Johannes Weiner 已提交
500
void unlock_page_memcg(struct page *page);
501

502
/* idx can be of type enum memcg_stat_item or node_stat_item */
503
static inline unsigned long memcg_page_state(struct mem_cgroup *memcg,
504
					     int idx)
505
{
506 507 508 509 510 511
	long x = atomic_long_read(&memcg->stat[idx]);
#ifdef CONFIG_SMP
	if (x < 0)
		x = 0;
#endif
	return x;
512 513
}

514
/* idx can be of type enum memcg_stat_item or node_stat_item */
515
static inline void __mod_memcg_state(struct mem_cgroup *memcg,
516
				     int idx, int val)
517
{
518 519 520 521 522 523 524 525 526 527 528
	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);
529 530
}

531
/* idx can be of type enum memcg_stat_item or node_stat_item */
532
static inline void mod_memcg_state(struct mem_cgroup *memcg,
533
				   int idx, int val)
534
{
535 536 537
	unsigned long flags;

	local_irq_save(flags);
538
	__mod_memcg_state(memcg, idx, val);
539
	local_irq_restore(flags);
540 541
}

M
Michal Hocko 已提交
542
/**
543
 * mod_memcg_page_state - update page state statistics
J
Johannes Weiner 已提交
544
 * @page: the page
M
Michal Hocko 已提交
545 546 547
 * @idx: page state item to account
 * @val: number of pages (positive or negative)
 *
548 549 550
 * 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:
551
 *
552
 *   lock_page(page) or lock_page_memcg(page)
553
 *   if (TestClearPageState(page))
554
 *     mod_memcg_page_state(page, state, -1);
555
 *   unlock_page(page) or unlock_page_memcg(page)
556 557
 *
 * Kernel pages are an exception to this, since they'll never move.
M
Michal Hocko 已提交
558
 */
559
static inline void __mod_memcg_page_state(struct page *page,
560
					  int idx, int val)
561 562 563 564 565
{
	if (page->mem_cgroup)
		__mod_memcg_state(page->mem_cgroup, idx, val);
}

566
static inline void mod_memcg_page_state(struct page *page,
567
					int idx, int val)
M
Michal Hocko 已提交
568
{
J
Johannes Weiner 已提交
569
	if (page->mem_cgroup)
570
		mod_memcg_state(page->mem_cgroup, idx, val);
M
Michal Hocko 已提交
571 572
}

573 574
static inline unsigned long lruvec_page_state(struct lruvec *lruvec,
					      enum node_stat_item idx)
575
{
576
	struct mem_cgroup_per_node *pn;
577
	long x;
578 579 580 581 582

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

	pn = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
583 584 585 586 587 588
	x = atomic_long_read(&pn->lruvec_stat[idx]);
#ifdef CONFIG_SMP
	if (x < 0)
		x = 0;
#endif
	return x;
589 590
}

591 592
static inline void __mod_lruvec_state(struct lruvec *lruvec,
				      enum node_stat_item idx, int val)
593
{
594
	struct mem_cgroup_per_node *pn;
595
	long x;
596

597
	/* Update node */
598
	__mod_node_page_state(lruvec_pgdat(lruvec), idx, val);
599

600 601
	if (mem_cgroup_disabled())
		return;
602

603
	pn = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
604 605

	/* Update memcg */
606
	__mod_memcg_state(pn->memcg, idx, val);
607 608

	/* Update lruvec */
609 610 611 612 613 614
	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);
615 616 617 618 619
}

static inline void mod_lruvec_state(struct lruvec *lruvec,
				    enum node_stat_item idx, int val)
{
620 621 622
	unsigned long flags;

	local_irq_save(flags);
623
	__mod_lruvec_state(lruvec, idx, val);
624
	local_irq_restore(flags);
625 626 627 628 629
}

static inline void __mod_lruvec_page_state(struct page *page,
					   enum node_stat_item idx, int val)
{
630 631
	pg_data_t *pgdat = page_pgdat(page);
	struct lruvec *lruvec;
632

633 634 635
	/* Untracked pages have no memcg, no lruvec. Update only the node */
	if (!page->mem_cgroup) {
		__mod_node_page_state(pgdat, idx, val);
636
		return;
637 638 639 640
	}

	lruvec = mem_cgroup_lruvec(pgdat, page->mem_cgroup);
	__mod_lruvec_state(lruvec, idx, val);
641 642 643 644 645
}

static inline void mod_lruvec_page_state(struct page *page,
					 enum node_stat_item idx, int val)
{
646 647 648
	unsigned long flags;

	local_irq_save(flags);
649
	__mod_lruvec_page_state(page, idx, val);
650
	local_irq_restore(flags);
651 652
}

653
unsigned long mem_cgroup_soft_limit_reclaim(pg_data_t *pgdat, int order,
654 655
						gfp_t gfp_mask,
						unsigned long *total_scanned);
D
David Rientjes 已提交
656

657
static inline void __count_memcg_events(struct mem_cgroup *memcg,
658 659
					enum vm_event_item idx,
					unsigned long count)
660
{
661 662 663 664 665 666 667 668 669 670 671
	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);
672 673
}

674
static inline void count_memcg_events(struct mem_cgroup *memcg,
675 676
				      enum vm_event_item idx,
				      unsigned long count)
677
{
678 679 680
	unsigned long flags;

	local_irq_save(flags);
681
	__count_memcg_events(memcg, idx, count);
682
	local_irq_restore(flags);
683 684 685
}

static inline void count_memcg_page_event(struct page *page,
686
					  enum vm_event_item idx)
687 688 689 690 691 692 693
{
	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)
694
{
M
Michal Hocko 已提交
695 696
	struct mem_cgroup *memcg;

697 698
	if (mem_cgroup_disabled())
		return;
M
Michal Hocko 已提交
699 700 701

	rcu_read_lock();
	memcg = mem_cgroup_from_task(rcu_dereference(mm->owner));
702
	if (likely(memcg)) {
703
		count_memcg_events(memcg, idx, 1);
704 705 706
		if (idx == OOM_KILL)
			cgroup_file_notify(&memcg->events_file);
	}
M
Michal Hocko 已提交
707
	rcu_read_unlock();
708
}
709

710 711
static inline void memcg_memory_event(struct mem_cgroup *memcg,
				      enum memcg_memory_event event)
712
{
713
	atomic_long_inc(&memcg->memory_events[event]);
714 715 716
	cgroup_file_notify(&memcg->events_file);
}

717
#ifdef CONFIG_TRANSPARENT_HUGEPAGE
718
void mem_cgroup_split_huge_fixup(struct page *head);
719 720
#endif

A
Andrew Morton 已提交
721
#else /* CONFIG_MEMCG */
722 723 724 725

#define MEM_CGROUP_ID_SHIFT	0
#define MEM_CGROUP_ID_MAX	0

726 727
struct mem_cgroup;

728 729 730 731 732
static inline bool mem_cgroup_disabled(void)
{
	return true;
}

733 734
static inline void memcg_memory_event(struct mem_cgroup *memcg,
				      enum memcg_memory_event event)
735 736 737 738 739 740 741 742 743
{
}

static inline bool mem_cgroup_low(struct mem_cgroup *root,
				  struct mem_cgroup *memcg)
{
	return false;
}

744 745
static inline int mem_cgroup_try_charge(struct page *page, struct mm_struct *mm,
					gfp_t gfp_mask,
746 747
					struct mem_cgroup **memcgp,
					bool compound)
748
{
749
	*memcgp = NULL;
750 751 752
	return 0;
}

753 754
static inline void mem_cgroup_commit_charge(struct page *page,
					    struct mem_cgroup *memcg,
755
					    bool lrucare, bool compound)
756 757 758
{
}

759
static inline void mem_cgroup_cancel_charge(struct page *page,
760 761
					    struct mem_cgroup *memcg,
					    bool compound)
762 763 764
{
}

765
static inline void mem_cgroup_uncharge(struct page *page)
766 767 768
{
}

769
static inline void mem_cgroup_uncharge_list(struct list_head *page_list)
770 771 772
{
}

773
static inline void mem_cgroup_migrate(struct page *old, struct page *new)
774 775 776
{
}

777
static inline struct lruvec *mem_cgroup_lruvec(struct pglist_data *pgdat,
778
				struct mem_cgroup *memcg)
K
KAMEZAWA Hiroyuki 已提交
779
{
780
	return node_lruvec(pgdat);
K
KAMEZAWA Hiroyuki 已提交
781 782
}

783
static inline struct lruvec *mem_cgroup_page_lruvec(struct page *page,
M
Mel Gorman 已提交
784
						    struct pglist_data *pgdat)
785
{
M
Mel Gorman 已提交
786
	return &pgdat->lruvec;
787 788
}

789
static inline bool mm_match_cgroup(struct mm_struct *mm,
790
		struct mem_cgroup *memcg)
791
{
792
	return true;
793 794
}

795 796
static inline bool task_in_mem_cgroup(struct task_struct *task,
				      const struct mem_cgroup *memcg)
797
{
798
	return true;
799 800
}

801 802 803 804 805 806 807 808 809 810 811 812 813
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)
{
}

814 815 816 817 818 819
static inline int mem_cgroup_scan_tasks(struct mem_cgroup *memcg,
		int (*fn)(struct task_struct *, void *), void *arg)
{
	return 0;
}

820
static inline unsigned short mem_cgroup_id(struct mem_cgroup *memcg)
821
{
822 823 824 825 826 827 828 829
	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;
830
}
831

832 833 834 835 836
static inline struct mem_cgroup *lruvec_memcg(struct lruvec *lruvec)
{
	return NULL;
}

837
static inline bool mem_cgroup_online(struct mem_cgroup *memcg)
838
{
839
	return true;
840 841
}

842
static inline unsigned long
843
mem_cgroup_get_lru_size(struct lruvec *lruvec, enum lru_list lru)
844 845 846
{
	return 0;
}
847 848 849 850 851 852
static inline
unsigned long mem_cgroup_get_zone_lru_size(struct lruvec *lruvec,
		enum lru_list lru, int zone_idx)
{
	return 0;
}
853

854 855 856 857 858 859 860
static inline unsigned long
mem_cgroup_node_nr_lru_pages(struct mem_cgroup *memcg,
			     int nid, unsigned int lru_mask)
{
	return 0;
}

861 862 863 864 865
static inline unsigned long mem_cgroup_get_limit(struct mem_cgroup *memcg)
{
	return 0;
}

866 867 868 869 870
static inline void
mem_cgroup_print_oom_info(struct mem_cgroup *memcg, struct task_struct *p)
{
}

871 872 873 874 875 876
static inline struct mem_cgroup *lock_page_memcg(struct page *page)
{
	return NULL;
}

static inline void __unlock_page_memcg(struct mem_cgroup *memcg)
877 878 879
{
}

J
Johannes Weiner 已提交
880
static inline void unlock_page_memcg(struct page *page)
881 882 883
{
}

884 885 886 887
static inline void mem_cgroup_handle_over_high(void)
{
}

888
static inline void mem_cgroup_oom_enable(void)
889 890 891
{
}

892
static inline void mem_cgroup_oom_disable(void)
893 894 895
{
}

896 897 898 899 900
static inline bool task_in_memcg_oom(struct task_struct *p)
{
	return false;
}

901
static inline bool mem_cgroup_oom_synchronize(bool wait)
902 903 904 905
{
	return false;
}

906
static inline unsigned long memcg_page_state(struct mem_cgroup *memcg,
907
					     int idx)
908 909 910 911
{
	return 0;
}

912
static inline void __mod_memcg_state(struct mem_cgroup *memcg,
913
				     int idx,
914
				     int nr)
915 916 917
{
}

918
static inline void mod_memcg_state(struct mem_cgroup *memcg,
919
				   int idx,
920
				   int nr)
921 922 923
{
}

924
static inline void __mod_memcg_page_state(struct page *page,
925
					  int idx,
926
					  int nr)
927 928 929
{
}

930
static inline void mod_memcg_page_state(struct page *page,
931
					int idx,
932
					int nr)
933 934 935
{
}

936 937
static inline unsigned long lruvec_page_state(struct lruvec *lruvec,
					      enum node_stat_item idx)
938
{
939
	return node_page_state(lruvec_pgdat(lruvec), idx);
940 941
}

942 943
static inline void __mod_lruvec_state(struct lruvec *lruvec,
				      enum node_stat_item idx, int val)
944
{
945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963
	__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);
964 965
}

966
static inline
967
unsigned long mem_cgroup_soft_limit_reclaim(pg_data_t *pgdat, int order,
968 969
					    gfp_t gfp_mask,
					    unsigned long *total_scanned)
970
{
971
	return 0;
972 973
}

974
static inline void mem_cgroup_split_huge_fixup(struct page *head)
975 976 977
{
}

978 979 980 981 982 983 984
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,
985
					  int idx)
986 987 988
{
}

989
static inline
990
void count_memcg_event_mm(struct mm_struct *mm, enum vm_event_item idx)
991 992
{
}
A
Andrew Morton 已提交
993
#endif /* CONFIG_MEMCG */
994

995
/* idx can be of type enum memcg_stat_item or node_stat_item */
996
static inline void __inc_memcg_state(struct mem_cgroup *memcg,
997
				     int idx)
998 999 1000 1001
{
	__mod_memcg_state(memcg, idx, 1);
}

1002
/* idx can be of type enum memcg_stat_item or node_stat_item */
1003
static inline void __dec_memcg_state(struct mem_cgroup *memcg,
1004
				     int idx)
1005 1006 1007 1008
{
	__mod_memcg_state(memcg, idx, -1);
}

1009
/* idx can be of type enum memcg_stat_item or node_stat_item */
1010
static inline void __inc_memcg_page_state(struct page *page,
1011
					  int idx)
1012 1013 1014 1015
{
	__mod_memcg_page_state(page, idx, 1);
}

1016
/* idx can be of type enum memcg_stat_item or node_stat_item */
1017
static inline void __dec_memcg_page_state(struct page *page,
1018
					  int idx)
1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046
{
	__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);
}

1047
/* idx can be of type enum memcg_stat_item or node_stat_item */
1048
static inline void inc_memcg_state(struct mem_cgroup *memcg,
1049
				   int idx)
1050 1051 1052 1053
{
	mod_memcg_state(memcg, idx, 1);
}

1054
/* idx can be of type enum memcg_stat_item or node_stat_item */
1055
static inline void dec_memcg_state(struct mem_cgroup *memcg,
1056
				   int idx)
1057 1058 1059 1060
{
	mod_memcg_state(memcg, idx, -1);
}

1061
/* idx can be of type enum memcg_stat_item or node_stat_item */
1062
static inline void inc_memcg_page_state(struct page *page,
1063
					int idx)
1064 1065 1066 1067
{
	mod_memcg_page_state(page, idx, 1);
}

1068
/* idx can be of type enum memcg_stat_item or node_stat_item */
1069
static inline void dec_memcg_page_state(struct page *page,
1070
					int idx)
1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098
{
	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);
}

1099
#ifdef CONFIG_CGROUP_WRITEBACK
T
Tejun Heo 已提交
1100

1101
struct list_head *mem_cgroup_cgwb_list(struct mem_cgroup *memcg);
T
Tejun Heo 已提交
1102
struct wb_domain *mem_cgroup_wb_domain(struct bdi_writeback *wb);
1103 1104 1105
void mem_cgroup_wb_stats(struct bdi_writeback *wb, unsigned long *pfilepages,
			 unsigned long *pheadroom, unsigned long *pdirty,
			 unsigned long *pwriteback);
T
Tejun Heo 已提交
1106 1107 1108 1109 1110 1111 1112 1113

#else	/* CONFIG_CGROUP_WRITEBACK */

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

1114
static inline void mem_cgroup_wb_stats(struct bdi_writeback *wb,
1115 1116
				       unsigned long *pfilepages,
				       unsigned long *pheadroom,
1117 1118 1119 1120 1121
				       unsigned long *pdirty,
				       unsigned long *pwriteback)
{
}

T
Tejun Heo 已提交
1122
#endif	/* CONFIG_CGROUP_WRITEBACK */
1123

G
Glauber Costa 已提交
1124
struct sock;
1125 1126
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);
1127
#ifdef CONFIG_MEMCG
1128 1129
extern struct static_key_false memcg_sockets_enabled_key;
#define mem_cgroup_sockets_enabled static_branch_unlikely(&memcg_sockets_enabled_key)
1130 1131
void mem_cgroup_sk_alloc(struct sock *sk);
void mem_cgroup_sk_free(struct sock *sk);
1132
static inline bool mem_cgroup_under_socket_pressure(struct mem_cgroup *memcg)
1133
{
1134
	if (!cgroup_subsys_on_dfl(memory_cgrp_subsys) && memcg->tcpmem_pressure)
1135 1136 1137 1138 1139 1140
		return true;
	do {
		if (time_before(jiffies, memcg->socket_pressure))
			return true;
	} while ((memcg = parent_mem_cgroup(memcg)));
	return false;
1141 1142
}
#else
1143
#define mem_cgroup_sockets_enabled 0
1144 1145
static inline void mem_cgroup_sk_alloc(struct sock *sk) { };
static inline void mem_cgroup_sk_free(struct sock *sk) { };
1146
static inline bool mem_cgroup_under_socket_pressure(struct mem_cgroup *memcg)
1147 1148 1149 1150
{
	return false;
}
#endif
1151

1152 1153 1154 1155 1156 1157 1158
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);

1159
#if defined(CONFIG_MEMCG) && !defined(CONFIG_SLOB)
1160
extern struct static_key_false memcg_kmem_enabled_key;
1161
extern struct workqueue_struct *memcg_kmem_cache_wq;
1162

1163
extern int memcg_nr_cache_ids;
1164 1165
void memcg_get_cache_ids(void);
void memcg_put_cache_ids(void);
1166 1167 1168 1169 1170 1171

/*
 * 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
 */
1172
#define for_each_memcg_cache_index(_idx)	\
1173
	for ((_idx) = 0; (_idx) < memcg_nr_cache_ids; (_idx)++)
1174

1175 1176
static inline bool memcg_kmem_enabled(void)
{
1177
	return static_branch_unlikely(&memcg_kmem_enabled_key);
1178 1179
}

M
Michal Hocko 已提交
1180
/*
J
Jesper Dangaard Brouer 已提交
1181
 * helper for accessing a memcg's index. It will be used as an index in the
M
Michal Hocko 已提交
1182 1183 1184 1185 1186 1187 1188
 * 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;
}
1189

1190
#else
1191 1192 1193
#define for_each_memcg_cache_index(_idx)	\
	for (; NULL; )

1194 1195 1196 1197 1198
static inline bool memcg_kmem_enabled(void)
{
	return false;
}

1199 1200 1201 1202 1203
static inline int memcg_cache_id(struct mem_cgroup *memcg)
{
	return -1;
}

1204 1205 1206 1207 1208 1209 1210 1211
static inline void memcg_get_cache_ids(void)
{
}

static inline void memcg_put_cache_ids(void)
{
}

1212 1213
#endif /* CONFIG_MEMCG && !CONFIG_SLOB */

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