memcontrol.h 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925
  1. /* memcontrol.h - Memory Controller
  2. *
  3. * Copyright IBM Corporation, 2007
  4. * Author Balbir Singh <balbir@linux.vnet.ibm.com>
  5. *
  6. * Copyright 2007 OpenVZ SWsoft Inc
  7. * Author: Pavel Emelianov <xemul@openvz.org>
  8. *
  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. #ifndef _LINUX_MEMCONTROL_H
  20. #define _LINUX_MEMCONTROL_H
  21. #include <linux/cgroup.h>
  22. #include <linux/vm_event_item.h>
  23. #include <linux/hardirq.h>
  24. #include <linux/jump_label.h>
  25. #include <linux/page_counter.h>
  26. #include <linux/vmpressure.h>
  27. #include <linux/eventfd.h>
  28. #include <linux/mmzone.h>
  29. #include <linux/writeback.h>
  30. struct mem_cgroup;
  31. struct page;
  32. struct mm_struct;
  33. struct kmem_cache;
  34. /*
  35. * The corresponding mem_cgroup_stat_names is defined in mm/memcontrol.c,
  36. * These two lists should keep in accord with each other.
  37. */
  38. enum mem_cgroup_stat_index {
  39. /*
  40. * For MEM_CONTAINER_TYPE_ALL, usage = pagecache + rss.
  41. */
  42. MEM_CGROUP_STAT_CACHE, /* # of pages charged as cache */
  43. MEM_CGROUP_STAT_RSS, /* # of pages charged as anon rss */
  44. MEM_CGROUP_STAT_RSS_HUGE, /* # of pages charged as anon huge */
  45. MEM_CGROUP_STAT_FILE_MAPPED, /* # of pages charged as file rss */
  46. MEM_CGROUP_STAT_DIRTY, /* # of dirty pages in page cache */
  47. MEM_CGROUP_STAT_WRITEBACK, /* # of pages under writeback */
  48. MEM_CGROUP_STAT_SWAP, /* # of pages, swapped out */
  49. MEM_CGROUP_STAT_NSTATS,
  50. };
  51. struct mem_cgroup_reclaim_cookie {
  52. struct zone *zone;
  53. int priority;
  54. unsigned int generation;
  55. };
  56. enum mem_cgroup_events_index {
  57. MEM_CGROUP_EVENTS_PGPGIN, /* # of pages paged in */
  58. MEM_CGROUP_EVENTS_PGPGOUT, /* # of pages paged out */
  59. MEM_CGROUP_EVENTS_PGFAULT, /* # of page-faults */
  60. MEM_CGROUP_EVENTS_PGMAJFAULT, /* # of major page-faults */
  61. MEM_CGROUP_EVENTS_NSTATS,
  62. /* default hierarchy events */
  63. MEMCG_LOW = MEM_CGROUP_EVENTS_NSTATS,
  64. MEMCG_HIGH,
  65. MEMCG_MAX,
  66. MEMCG_OOM,
  67. MEMCG_NR_EVENTS,
  68. };
  69. /*
  70. * Per memcg event counter is incremented at every pagein/pageout. With THP,
  71. * it will be incremated by the number of pages. This counter is used for
  72. * for trigger some periodic events. This is straightforward and better
  73. * than using jiffies etc. to handle periodic memcg event.
  74. */
  75. enum mem_cgroup_events_target {
  76. MEM_CGROUP_TARGET_THRESH,
  77. MEM_CGROUP_TARGET_SOFTLIMIT,
  78. MEM_CGROUP_TARGET_NUMAINFO,
  79. MEM_CGROUP_NTARGETS,
  80. };
  81. /*
  82. * Bits in struct cg_proto.flags
  83. */
  84. enum cg_proto_flags {
  85. /* Currently active and new sockets should be assigned to cgroups */
  86. MEMCG_SOCK_ACTIVE,
  87. /* It was ever activated; we must disarm static keys on destruction */
  88. MEMCG_SOCK_ACTIVATED,
  89. };
  90. struct cg_proto {
  91. struct page_counter memory_allocated; /* Current allocated memory. */
  92. struct percpu_counter sockets_allocated; /* Current number of sockets. */
  93. int memory_pressure;
  94. long sysctl_mem[3];
  95. unsigned long flags;
  96. /*
  97. * memcg field is used to find which memcg we belong directly
  98. * Each memcg struct can hold more than one cg_proto, so container_of
  99. * won't really cut.
  100. *
  101. * The elegant solution would be having an inverse function to
  102. * proto_cgroup in struct proto, but that means polluting the structure
  103. * for everybody, instead of just for memcg users.
  104. */
  105. struct mem_cgroup *memcg;
  106. };
  107. #ifdef CONFIG_MEMCG
  108. struct mem_cgroup_stat_cpu {
  109. long count[MEM_CGROUP_STAT_NSTATS];
  110. unsigned long events[MEMCG_NR_EVENTS];
  111. unsigned long nr_page_events;
  112. unsigned long targets[MEM_CGROUP_NTARGETS];
  113. };
  114. struct mem_cgroup_reclaim_iter {
  115. struct mem_cgroup *position;
  116. /* scan generation, increased every round-trip */
  117. unsigned int generation;
  118. };
  119. /*
  120. * per-zone information in memory controller.
  121. */
  122. struct mem_cgroup_per_zone {
  123. struct lruvec lruvec;
  124. unsigned long lru_size[NR_LRU_LISTS];
  125. struct mem_cgroup_reclaim_iter iter[DEF_PRIORITY + 1];
  126. struct rb_node tree_node; /* RB tree node */
  127. unsigned long usage_in_excess;/* Set to the value by which */
  128. /* the soft limit is exceeded*/
  129. bool on_tree;
  130. struct mem_cgroup *memcg; /* Back pointer, we cannot */
  131. /* use container_of */
  132. };
  133. struct mem_cgroup_per_node {
  134. struct mem_cgroup_per_zone zoneinfo[MAX_NR_ZONES];
  135. };
  136. struct mem_cgroup_threshold {
  137. struct eventfd_ctx *eventfd;
  138. unsigned long threshold;
  139. };
  140. /* For threshold */
  141. struct mem_cgroup_threshold_ary {
  142. /* An array index points to threshold just below or equal to usage. */
  143. int current_threshold;
  144. /* Size of entries[] */
  145. unsigned int size;
  146. /* Array of thresholds */
  147. struct mem_cgroup_threshold entries[0];
  148. };
  149. struct mem_cgroup_thresholds {
  150. /* Primary thresholds array */
  151. struct mem_cgroup_threshold_ary *primary;
  152. /*
  153. * Spare threshold array.
  154. * This is needed to make mem_cgroup_unregister_event() "never fail".
  155. * It must be able to store at least primary->size - 1 entries.
  156. */
  157. struct mem_cgroup_threshold_ary *spare;
  158. };
  159. /*
  160. * The memory controller data structure. The memory controller controls both
  161. * page cache and RSS per cgroup. We would eventually like to provide
  162. * statistics based on the statistics developed by Rik Van Riel for clock-pro,
  163. * to help the administrator determine what knobs to tune.
  164. */
  165. struct mem_cgroup {
  166. struct cgroup_subsys_state css;
  167. /* Accounted resources */
  168. struct page_counter memory;
  169. struct page_counter memsw;
  170. struct page_counter kmem;
  171. /* Normal memory consumption range */
  172. unsigned long low;
  173. unsigned long high;
  174. unsigned long soft_limit;
  175. /* vmpressure notifications */
  176. struct vmpressure vmpressure;
  177. /* css_online() has been completed */
  178. int initialized;
  179. /*
  180. * Should the accounting and control be hierarchical, per subtree?
  181. */
  182. bool use_hierarchy;
  183. /* protected by memcg_oom_lock */
  184. bool oom_lock;
  185. int under_oom;
  186. int swappiness;
  187. /* OOM-Killer disable */
  188. int oom_kill_disable;
  189. /* protect arrays of thresholds */
  190. struct mutex thresholds_lock;
  191. /* thresholds for memory usage. RCU-protected */
  192. struct mem_cgroup_thresholds thresholds;
  193. /* thresholds for mem+swap usage. RCU-protected */
  194. struct mem_cgroup_thresholds memsw_thresholds;
  195. /* For oom notifier event fd */
  196. struct list_head oom_notify;
  197. /*
  198. * Should we move charges of a task when a task is moved into this
  199. * mem_cgroup ? And what type of charges should we move ?
  200. */
  201. unsigned long move_charge_at_immigrate;
  202. /*
  203. * set > 0 if pages under this cgroup are moving to other cgroup.
  204. */
  205. atomic_t moving_account;
  206. /* taken only while moving_account > 0 */
  207. spinlock_t move_lock;
  208. struct task_struct *move_lock_task;
  209. unsigned long move_lock_flags;
  210. /*
  211. * percpu counter.
  212. */
  213. struct mem_cgroup_stat_cpu __percpu *stat;
  214. #if defined(CONFIG_MEMCG_KMEM) && defined(CONFIG_INET)
  215. struct cg_proto tcp_mem;
  216. #endif
  217. #if defined(CONFIG_MEMCG_KMEM)
  218. /* Index in the kmem_cache->memcg_params.memcg_caches array */
  219. int kmemcg_id;
  220. bool kmem_acct_activated;
  221. bool kmem_acct_active;
  222. #endif
  223. int last_scanned_node;
  224. #if MAX_NUMNODES > 1
  225. nodemask_t scan_nodes;
  226. atomic_t numainfo_events;
  227. atomic_t numainfo_updating;
  228. #endif
  229. #ifdef CONFIG_CGROUP_WRITEBACK
  230. struct list_head cgwb_list;
  231. struct wb_domain cgwb_domain;
  232. #endif
  233. /* List of events which userspace want to receive */
  234. struct list_head event_list;
  235. spinlock_t event_list_lock;
  236. struct mem_cgroup_per_node *nodeinfo[0];
  237. /* WARNING: nodeinfo must be the last member here */
  238. };
  239. extern struct cgroup_subsys_state *mem_cgroup_root_css;
  240. /**
  241. * mem_cgroup_events - count memory events against a cgroup
  242. * @memcg: the memory cgroup
  243. * @idx: the event index
  244. * @nr: the number of events to account for
  245. */
  246. static inline void mem_cgroup_events(struct mem_cgroup *memcg,
  247. enum mem_cgroup_events_index idx,
  248. unsigned int nr)
  249. {
  250. this_cpu_add(memcg->stat->events[idx], nr);
  251. }
  252. bool mem_cgroup_low(struct mem_cgroup *root, struct mem_cgroup *memcg);
  253. int mem_cgroup_try_charge(struct page *page, struct mm_struct *mm,
  254. gfp_t gfp_mask, struct mem_cgroup **memcgp);
  255. void mem_cgroup_commit_charge(struct page *page, struct mem_cgroup *memcg,
  256. bool lrucare);
  257. void mem_cgroup_cancel_charge(struct page *page, struct mem_cgroup *memcg);
  258. void mem_cgroup_uncharge(struct page *page);
  259. void mem_cgroup_uncharge_list(struct list_head *page_list);
  260. void mem_cgroup_migrate(struct page *oldpage, struct page *newpage,
  261. bool lrucare);
  262. struct lruvec *mem_cgroup_zone_lruvec(struct zone *, struct mem_cgroup *);
  263. struct lruvec *mem_cgroup_page_lruvec(struct page *, struct zone *);
  264. bool task_in_mem_cgroup(struct task_struct *task, struct mem_cgroup *memcg);
  265. struct mem_cgroup *mem_cgroup_from_task(struct task_struct *p);
  266. struct mem_cgroup *parent_mem_cgroup(struct mem_cgroup *memcg);
  267. static inline
  268. struct mem_cgroup *mem_cgroup_from_css(struct cgroup_subsys_state *css){
  269. return css ? container_of(css, struct mem_cgroup, css) : NULL;
  270. }
  271. struct mem_cgroup *mem_cgroup_iter(struct mem_cgroup *,
  272. struct mem_cgroup *,
  273. struct mem_cgroup_reclaim_cookie *);
  274. void mem_cgroup_iter_break(struct mem_cgroup *, struct mem_cgroup *);
  275. static inline bool mem_cgroup_is_descendant(struct mem_cgroup *memcg,
  276. struct mem_cgroup *root)
  277. {
  278. if (root == memcg)
  279. return true;
  280. if (!root->use_hierarchy)
  281. return false;
  282. return cgroup_is_descendant(memcg->css.cgroup, root->css.cgroup);
  283. }
  284. static inline bool mm_match_cgroup(struct mm_struct *mm,
  285. struct mem_cgroup *memcg)
  286. {
  287. struct mem_cgroup *task_memcg;
  288. bool match = false;
  289. rcu_read_lock();
  290. task_memcg = mem_cgroup_from_task(rcu_dereference(mm->owner));
  291. if (task_memcg)
  292. match = mem_cgroup_is_descendant(task_memcg, memcg);
  293. rcu_read_unlock();
  294. return match;
  295. }
  296. struct cgroup_subsys_state *mem_cgroup_css_from_page(struct page *page);
  297. ino_t page_cgroup_ino(struct page *page);
  298. static inline bool mem_cgroup_disabled(void)
  299. {
  300. if (memory_cgrp_subsys.disabled)
  301. return true;
  302. return false;
  303. }
  304. /*
  305. * For memory reclaim.
  306. */
  307. int mem_cgroup_select_victim_node(struct mem_cgroup *memcg);
  308. void mem_cgroup_update_lru_size(struct lruvec *lruvec, enum lru_list lru,
  309. int nr_pages);
  310. static inline bool mem_cgroup_lruvec_online(struct lruvec *lruvec)
  311. {
  312. struct mem_cgroup_per_zone *mz;
  313. struct mem_cgroup *memcg;
  314. if (mem_cgroup_disabled())
  315. return true;
  316. mz = container_of(lruvec, struct mem_cgroup_per_zone, lruvec);
  317. memcg = mz->memcg;
  318. return !!(memcg->css.flags & CSS_ONLINE);
  319. }
  320. static inline
  321. unsigned long mem_cgroup_get_lru_size(struct lruvec *lruvec, enum lru_list lru)
  322. {
  323. struct mem_cgroup_per_zone *mz;
  324. mz = container_of(lruvec, struct mem_cgroup_per_zone, lruvec);
  325. return mz->lru_size[lru];
  326. }
  327. static inline bool mem_cgroup_inactive_anon_is_low(struct lruvec *lruvec)
  328. {
  329. unsigned long inactive_ratio;
  330. unsigned long inactive;
  331. unsigned long active;
  332. unsigned long gb;
  333. inactive = mem_cgroup_get_lru_size(lruvec, LRU_INACTIVE_ANON);
  334. active = mem_cgroup_get_lru_size(lruvec, LRU_ACTIVE_ANON);
  335. gb = (inactive + active) >> (30 - PAGE_SHIFT);
  336. if (gb)
  337. inactive_ratio = int_sqrt(10 * gb);
  338. else
  339. inactive_ratio = 1;
  340. return inactive * inactive_ratio < active;
  341. }
  342. void mem_cgroup_handle_over_high(void);
  343. void mem_cgroup_print_oom_info(struct mem_cgroup *memcg,
  344. struct task_struct *p);
  345. static inline void mem_cgroup_oom_enable(void)
  346. {
  347. WARN_ON(current->memcg_may_oom);
  348. current->memcg_may_oom = 1;
  349. }
  350. static inline void mem_cgroup_oom_disable(void)
  351. {
  352. WARN_ON(!current->memcg_may_oom);
  353. current->memcg_may_oom = 0;
  354. }
  355. static inline bool task_in_memcg_oom(struct task_struct *p)
  356. {
  357. return p->memcg_in_oom;
  358. }
  359. bool mem_cgroup_oom_synchronize(bool wait);
  360. #ifdef CONFIG_MEMCG_SWAP
  361. extern int do_swap_account;
  362. #endif
  363. struct mem_cgroup *mem_cgroup_begin_page_stat(struct page *page);
  364. void mem_cgroup_end_page_stat(struct mem_cgroup *memcg);
  365. /**
  366. * mem_cgroup_update_page_stat - update page state statistics
  367. * @memcg: memcg to account against
  368. * @idx: page state item to account
  369. * @val: number of pages (positive or negative)
  370. *
  371. * See mem_cgroup_begin_page_stat() for locking requirements.
  372. */
  373. static inline void mem_cgroup_update_page_stat(struct mem_cgroup *memcg,
  374. enum mem_cgroup_stat_index idx, int val)
  375. {
  376. VM_BUG_ON(!rcu_read_lock_held());
  377. if (memcg)
  378. this_cpu_add(memcg->stat->count[idx], val);
  379. }
  380. static inline void mem_cgroup_inc_page_stat(struct mem_cgroup *memcg,
  381. enum mem_cgroup_stat_index idx)
  382. {
  383. mem_cgroup_update_page_stat(memcg, idx, 1);
  384. }
  385. static inline void mem_cgroup_dec_page_stat(struct mem_cgroup *memcg,
  386. enum mem_cgroup_stat_index idx)
  387. {
  388. mem_cgroup_update_page_stat(memcg, idx, -1);
  389. }
  390. unsigned long mem_cgroup_soft_limit_reclaim(struct zone *zone, int order,
  391. gfp_t gfp_mask,
  392. unsigned long *total_scanned);
  393. static inline void mem_cgroup_count_vm_event(struct mm_struct *mm,
  394. enum vm_event_item idx)
  395. {
  396. struct mem_cgroup *memcg;
  397. if (mem_cgroup_disabled())
  398. return;
  399. rcu_read_lock();
  400. memcg = mem_cgroup_from_task(rcu_dereference(mm->owner));
  401. if (unlikely(!memcg))
  402. goto out;
  403. switch (idx) {
  404. case PGFAULT:
  405. this_cpu_inc(memcg->stat->events[MEM_CGROUP_EVENTS_PGFAULT]);
  406. break;
  407. case PGMAJFAULT:
  408. this_cpu_inc(memcg->stat->events[MEM_CGROUP_EVENTS_PGMAJFAULT]);
  409. break;
  410. default:
  411. BUG();
  412. }
  413. out:
  414. rcu_read_unlock();
  415. }
  416. #ifdef CONFIG_TRANSPARENT_HUGEPAGE
  417. void mem_cgroup_split_huge_fixup(struct page *head);
  418. #endif
  419. #else /* CONFIG_MEMCG */
  420. struct mem_cgroup;
  421. static inline void mem_cgroup_events(struct mem_cgroup *memcg,
  422. enum mem_cgroup_events_index idx,
  423. unsigned int nr)
  424. {
  425. }
  426. static inline bool mem_cgroup_low(struct mem_cgroup *root,
  427. struct mem_cgroup *memcg)
  428. {
  429. return false;
  430. }
  431. static inline int mem_cgroup_try_charge(struct page *page, struct mm_struct *mm,
  432. gfp_t gfp_mask,
  433. struct mem_cgroup **memcgp)
  434. {
  435. *memcgp = NULL;
  436. return 0;
  437. }
  438. static inline void mem_cgroup_commit_charge(struct page *page,
  439. struct mem_cgroup *memcg,
  440. bool lrucare)
  441. {
  442. }
  443. static inline void mem_cgroup_cancel_charge(struct page *page,
  444. struct mem_cgroup *memcg)
  445. {
  446. }
  447. static inline void mem_cgroup_uncharge(struct page *page)
  448. {
  449. }
  450. static inline void mem_cgroup_uncharge_list(struct list_head *page_list)
  451. {
  452. }
  453. static inline void mem_cgroup_migrate(struct page *oldpage,
  454. struct page *newpage,
  455. bool lrucare)
  456. {
  457. }
  458. static inline struct lruvec *mem_cgroup_zone_lruvec(struct zone *zone,
  459. struct mem_cgroup *memcg)
  460. {
  461. return &zone->lruvec;
  462. }
  463. static inline struct lruvec *mem_cgroup_page_lruvec(struct page *page,
  464. struct zone *zone)
  465. {
  466. return &zone->lruvec;
  467. }
  468. static inline bool mm_match_cgroup(struct mm_struct *mm,
  469. struct mem_cgroup *memcg)
  470. {
  471. return true;
  472. }
  473. static inline bool task_in_mem_cgroup(struct task_struct *task,
  474. const struct mem_cgroup *memcg)
  475. {
  476. return true;
  477. }
  478. static inline struct mem_cgroup *
  479. mem_cgroup_iter(struct mem_cgroup *root,
  480. struct mem_cgroup *prev,
  481. struct mem_cgroup_reclaim_cookie *reclaim)
  482. {
  483. return NULL;
  484. }
  485. static inline void mem_cgroup_iter_break(struct mem_cgroup *root,
  486. struct mem_cgroup *prev)
  487. {
  488. }
  489. static inline bool mem_cgroup_disabled(void)
  490. {
  491. return true;
  492. }
  493. static inline bool
  494. mem_cgroup_inactive_anon_is_low(struct lruvec *lruvec)
  495. {
  496. return true;
  497. }
  498. static inline bool mem_cgroup_lruvec_online(struct lruvec *lruvec)
  499. {
  500. return true;
  501. }
  502. static inline unsigned long
  503. mem_cgroup_get_lru_size(struct lruvec *lruvec, enum lru_list lru)
  504. {
  505. return 0;
  506. }
  507. static inline void
  508. mem_cgroup_update_lru_size(struct lruvec *lruvec, enum lru_list lru,
  509. int increment)
  510. {
  511. }
  512. static inline void
  513. mem_cgroup_print_oom_info(struct mem_cgroup *memcg, struct task_struct *p)
  514. {
  515. }
  516. static inline struct mem_cgroup *mem_cgroup_begin_page_stat(struct page *page)
  517. {
  518. return NULL;
  519. }
  520. static inline void mem_cgroup_end_page_stat(struct mem_cgroup *memcg)
  521. {
  522. }
  523. static inline void mem_cgroup_handle_over_high(void)
  524. {
  525. }
  526. static inline void mem_cgroup_oom_enable(void)
  527. {
  528. }
  529. static inline void mem_cgroup_oom_disable(void)
  530. {
  531. }
  532. static inline bool task_in_memcg_oom(struct task_struct *p)
  533. {
  534. return false;
  535. }
  536. static inline bool mem_cgroup_oom_synchronize(bool wait)
  537. {
  538. return false;
  539. }
  540. static inline void mem_cgroup_inc_page_stat(struct mem_cgroup *memcg,
  541. enum mem_cgroup_stat_index idx)
  542. {
  543. }
  544. static inline void mem_cgroup_dec_page_stat(struct mem_cgroup *memcg,
  545. enum mem_cgroup_stat_index idx)
  546. {
  547. }
  548. static inline
  549. unsigned long mem_cgroup_soft_limit_reclaim(struct zone *zone, int order,
  550. gfp_t gfp_mask,
  551. unsigned long *total_scanned)
  552. {
  553. return 0;
  554. }
  555. static inline void mem_cgroup_split_huge_fixup(struct page *head)
  556. {
  557. }
  558. static inline
  559. void mem_cgroup_count_vm_event(struct mm_struct *mm, enum vm_event_item idx)
  560. {
  561. }
  562. #endif /* CONFIG_MEMCG */
  563. enum {
  564. UNDER_LIMIT,
  565. SOFT_LIMIT,
  566. OVER_LIMIT,
  567. };
  568. #ifdef CONFIG_CGROUP_WRITEBACK
  569. struct list_head *mem_cgroup_cgwb_list(struct mem_cgroup *memcg);
  570. struct wb_domain *mem_cgroup_wb_domain(struct bdi_writeback *wb);
  571. void mem_cgroup_wb_stats(struct bdi_writeback *wb, unsigned long *pfilepages,
  572. unsigned long *pheadroom, unsigned long *pdirty,
  573. unsigned long *pwriteback);
  574. #else /* CONFIG_CGROUP_WRITEBACK */
  575. static inline struct wb_domain *mem_cgroup_wb_domain(struct bdi_writeback *wb)
  576. {
  577. return NULL;
  578. }
  579. static inline void mem_cgroup_wb_stats(struct bdi_writeback *wb,
  580. unsigned long *pfilepages,
  581. unsigned long *pheadroom,
  582. unsigned long *pdirty,
  583. unsigned long *pwriteback)
  584. {
  585. }
  586. #endif /* CONFIG_CGROUP_WRITEBACK */
  587. struct sock;
  588. #if defined(CONFIG_INET) && defined(CONFIG_MEMCG_KMEM)
  589. void sock_update_memcg(struct sock *sk);
  590. void sock_release_memcg(struct sock *sk);
  591. #else
  592. static inline void sock_update_memcg(struct sock *sk)
  593. {
  594. }
  595. static inline void sock_release_memcg(struct sock *sk)
  596. {
  597. }
  598. #endif /* CONFIG_INET && CONFIG_MEMCG_KMEM */
  599. #ifdef CONFIG_MEMCG_KMEM
  600. extern struct static_key memcg_kmem_enabled_key;
  601. extern int memcg_nr_cache_ids;
  602. void memcg_get_cache_ids(void);
  603. void memcg_put_cache_ids(void);
  604. /*
  605. * Helper macro to loop through all memcg-specific caches. Callers must still
  606. * check if the cache is valid (it is either valid or NULL).
  607. * the slab_mutex must be held when looping through those caches
  608. */
  609. #define for_each_memcg_cache_index(_idx) \
  610. for ((_idx) = 0; (_idx) < memcg_nr_cache_ids; (_idx)++)
  611. static inline bool memcg_kmem_enabled(void)
  612. {
  613. return static_key_false(&memcg_kmem_enabled_key);
  614. }
  615. static inline bool memcg_kmem_is_active(struct mem_cgroup *memcg)
  616. {
  617. return memcg->kmem_acct_active;
  618. }
  619. /*
  620. * In general, we'll do everything in our power to not incur in any overhead
  621. * for non-memcg users for the kmem functions. Not even a function call, if we
  622. * can avoid it.
  623. *
  624. * Therefore, we'll inline all those functions so that in the best case, we'll
  625. * see that kmemcg is off for everybody and proceed quickly. If it is on,
  626. * we'll still do most of the flag checking inline. We check a lot of
  627. * conditions, but because they are pretty simple, they are expected to be
  628. * fast.
  629. */
  630. bool __memcg_kmem_newpage_charge(gfp_t gfp, struct mem_cgroup **memcg,
  631. int order);
  632. void __memcg_kmem_commit_charge(struct page *page,
  633. struct mem_cgroup *memcg, int order);
  634. void __memcg_kmem_uncharge_pages(struct page *page, int order);
  635. /*
  636. * helper for acessing a memcg's index. It will be used as an index in the
  637. * child cache array in kmem_cache, and also to derive its name. This function
  638. * will return -1 when this is not a kmem-limited memcg.
  639. */
  640. static inline int memcg_cache_id(struct mem_cgroup *memcg)
  641. {
  642. return memcg ? memcg->kmemcg_id : -1;
  643. }
  644. struct kmem_cache *__memcg_kmem_get_cache(struct kmem_cache *cachep);
  645. void __memcg_kmem_put_cache(struct kmem_cache *cachep);
  646. struct mem_cgroup *__mem_cgroup_from_kmem(void *ptr);
  647. int memcg_charge_kmem(struct mem_cgroup *memcg, gfp_t gfp,
  648. unsigned long nr_pages);
  649. void memcg_uncharge_kmem(struct mem_cgroup *memcg, unsigned long nr_pages);
  650. static inline bool __memcg_kmem_bypass(gfp_t gfp)
  651. {
  652. if (!memcg_kmem_enabled())
  653. return true;
  654. if (gfp & __GFP_NOACCOUNT)
  655. return true;
  656. if (in_interrupt() || (!current->mm) || (current->flags & PF_KTHREAD))
  657. return true;
  658. return false;
  659. }
  660. /**
  661. * memcg_kmem_newpage_charge: verify if a new kmem allocation is allowed.
  662. * @gfp: the gfp allocation flags.
  663. * @memcg: a pointer to the memcg this was charged against.
  664. * @order: allocation order.
  665. *
  666. * returns true if the memcg where the current task belongs can hold this
  667. * allocation.
  668. *
  669. * We return true automatically if this allocation is not to be accounted to
  670. * any memcg.
  671. */
  672. static inline bool
  673. memcg_kmem_newpage_charge(gfp_t gfp, struct mem_cgroup **memcg, int order)
  674. {
  675. if (__memcg_kmem_bypass(gfp))
  676. return true;
  677. return __memcg_kmem_newpage_charge(gfp, memcg, order);
  678. }
  679. /**
  680. * memcg_kmem_uncharge_pages: uncharge pages from memcg
  681. * @page: pointer to struct page being freed
  682. * @order: allocation order.
  683. */
  684. static inline void
  685. memcg_kmem_uncharge_pages(struct page *page, int order)
  686. {
  687. if (memcg_kmem_enabled())
  688. __memcg_kmem_uncharge_pages(page, order);
  689. }
  690. /**
  691. * memcg_kmem_commit_charge: embeds correct memcg in a page
  692. * @page: pointer to struct page recently allocated
  693. * @memcg: the memcg structure we charged against
  694. * @order: allocation order.
  695. *
  696. * Needs to be called after memcg_kmem_newpage_charge, regardless of success or
  697. * failure of the allocation. if @page is NULL, this function will revert the
  698. * charges. Otherwise, it will commit @page to @memcg.
  699. */
  700. static inline void
  701. memcg_kmem_commit_charge(struct page *page, struct mem_cgroup *memcg, int order)
  702. {
  703. if (memcg_kmem_enabled() && memcg)
  704. __memcg_kmem_commit_charge(page, memcg, order);
  705. }
  706. /**
  707. * memcg_kmem_get_cache: selects the correct per-memcg cache for allocation
  708. * @cachep: the original global kmem cache
  709. * @gfp: allocation flags.
  710. *
  711. * All memory allocated from a per-memcg cache is charged to the owner memcg.
  712. */
  713. static __always_inline struct kmem_cache *
  714. memcg_kmem_get_cache(struct kmem_cache *cachep, gfp_t gfp)
  715. {
  716. if (__memcg_kmem_bypass(gfp))
  717. return cachep;
  718. return __memcg_kmem_get_cache(cachep);
  719. }
  720. static __always_inline void memcg_kmem_put_cache(struct kmem_cache *cachep)
  721. {
  722. if (memcg_kmem_enabled())
  723. __memcg_kmem_put_cache(cachep);
  724. }
  725. static __always_inline struct mem_cgroup *mem_cgroup_from_kmem(void *ptr)
  726. {
  727. if (!memcg_kmem_enabled())
  728. return NULL;
  729. return __mem_cgroup_from_kmem(ptr);
  730. }
  731. #else
  732. #define for_each_memcg_cache_index(_idx) \
  733. for (; NULL; )
  734. static inline bool memcg_kmem_enabled(void)
  735. {
  736. return false;
  737. }
  738. static inline bool memcg_kmem_is_active(struct mem_cgroup *memcg)
  739. {
  740. return false;
  741. }
  742. static inline bool
  743. memcg_kmem_newpage_charge(gfp_t gfp, struct mem_cgroup **memcg, int order)
  744. {
  745. return true;
  746. }
  747. static inline void memcg_kmem_uncharge_pages(struct page *page, int order)
  748. {
  749. }
  750. static inline void
  751. memcg_kmem_commit_charge(struct page *page, struct mem_cgroup *memcg, int order)
  752. {
  753. }
  754. static inline int memcg_cache_id(struct mem_cgroup *memcg)
  755. {
  756. return -1;
  757. }
  758. static inline void memcg_get_cache_ids(void)
  759. {
  760. }
  761. static inline void memcg_put_cache_ids(void)
  762. {
  763. }
  764. static inline struct kmem_cache *
  765. memcg_kmem_get_cache(struct kmem_cache *cachep, gfp_t gfp)
  766. {
  767. return cachep;
  768. }
  769. static inline void memcg_kmem_put_cache(struct kmem_cache *cachep)
  770. {
  771. }
  772. static inline struct mem_cgroup *mem_cgroup_from_kmem(void *ptr)
  773. {
  774. return NULL;
  775. }
  776. #endif /* CONFIG_MEMCG_KMEM */
  777. #endif /* _LINUX_MEMCONTROL_H */