memcontrol.h 28 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142
  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/mm.h>
  29. #include <linux/vmstat.h>
  30. #include <linux/writeback.h>
  31. #include <linux/page-flags.h>
  32. struct mem_cgroup;
  33. struct page;
  34. struct mm_struct;
  35. struct kmem_cache;
  36. /* Cgroup-specific page state, on top of universal node page state */
  37. enum memcg_stat_item {
  38. MEMCG_CACHE = NR_VM_NODE_STAT_ITEMS,
  39. MEMCG_RSS,
  40. MEMCG_RSS_HUGE,
  41. MEMCG_SWAP,
  42. MEMCG_SOCK,
  43. /* XXX: why are these zone and not node counters? */
  44. MEMCG_KERNEL_STACK_KB,
  45. MEMCG_NR_STAT,
  46. };
  47. /* Cgroup-specific events, on top of universal VM events */
  48. enum memcg_event_item {
  49. MEMCG_LOW = NR_VM_EVENT_ITEMS,
  50. MEMCG_HIGH,
  51. MEMCG_MAX,
  52. MEMCG_OOM,
  53. MEMCG_NR_EVENTS,
  54. };
  55. struct mem_cgroup_reclaim_cookie {
  56. pg_data_t *pgdat;
  57. int priority;
  58. unsigned int generation;
  59. };
  60. #ifdef CONFIG_MEMCG
  61. #define MEM_CGROUP_ID_SHIFT 16
  62. #define MEM_CGROUP_ID_MAX USHRT_MAX
  63. struct mem_cgroup_id {
  64. int id;
  65. atomic_t ref;
  66. };
  67. /*
  68. * Per memcg event counter is incremented at every pagein/pageout. With THP,
  69. * it will be incremated by the number of pages. This counter is used for
  70. * for trigger some periodic events. This is straightforward and better
  71. * than using jiffies etc. to handle periodic memcg event.
  72. */
  73. enum mem_cgroup_events_target {
  74. MEM_CGROUP_TARGET_THRESH,
  75. MEM_CGROUP_TARGET_SOFTLIMIT,
  76. MEM_CGROUP_TARGET_NUMAINFO,
  77. MEM_CGROUP_NTARGETS,
  78. };
  79. struct mem_cgroup_stat_cpu {
  80. long count[MEMCG_NR_STAT];
  81. unsigned long events[MEMCG_NR_EVENTS];
  82. unsigned long nr_page_events;
  83. unsigned long targets[MEM_CGROUP_NTARGETS];
  84. };
  85. struct mem_cgroup_reclaim_iter {
  86. struct mem_cgroup *position;
  87. /* scan generation, increased every round-trip */
  88. unsigned int generation;
  89. };
  90. struct lruvec_stat {
  91. long count[NR_VM_NODE_STAT_ITEMS];
  92. };
  93. /*
  94. * per-zone information in memory controller.
  95. */
  96. struct mem_cgroup_per_node {
  97. struct lruvec lruvec;
  98. struct lruvec_stat __percpu *lruvec_stat;
  99. unsigned long lru_zone_size[MAX_NR_ZONES][NR_LRU_LISTS];
  100. struct mem_cgroup_reclaim_iter iter[DEF_PRIORITY + 1];
  101. struct rb_node tree_node; /* RB tree node */
  102. unsigned long usage_in_excess;/* Set to the value by which */
  103. /* the soft limit is exceeded*/
  104. bool on_tree;
  105. struct mem_cgroup *memcg; /* Back pointer, we cannot */
  106. /* use container_of */
  107. };
  108. struct mem_cgroup_threshold {
  109. struct eventfd_ctx *eventfd;
  110. unsigned long threshold;
  111. };
  112. /* For threshold */
  113. struct mem_cgroup_threshold_ary {
  114. /* An array index points to threshold just below or equal to usage. */
  115. int current_threshold;
  116. /* Size of entries[] */
  117. unsigned int size;
  118. /* Array of thresholds */
  119. struct mem_cgroup_threshold entries[0];
  120. };
  121. struct mem_cgroup_thresholds {
  122. /* Primary thresholds array */
  123. struct mem_cgroup_threshold_ary *primary;
  124. /*
  125. * Spare threshold array.
  126. * This is needed to make mem_cgroup_unregister_event() "never fail".
  127. * It must be able to store at least primary->size - 1 entries.
  128. */
  129. struct mem_cgroup_threshold_ary *spare;
  130. };
  131. enum memcg_kmem_state {
  132. KMEM_NONE,
  133. KMEM_ALLOCATED,
  134. KMEM_ONLINE,
  135. };
  136. /*
  137. * The memory controller data structure. The memory controller controls both
  138. * page cache and RSS per cgroup. We would eventually like to provide
  139. * statistics based on the statistics developed by Rik Van Riel for clock-pro,
  140. * to help the administrator determine what knobs to tune.
  141. */
  142. struct mem_cgroup {
  143. struct cgroup_subsys_state css;
  144. /* Private memcg ID. Used to ID objects that outlive the cgroup */
  145. struct mem_cgroup_id id;
  146. /* Accounted resources */
  147. struct page_counter memory;
  148. struct page_counter swap;
  149. /* Legacy consumer-oriented counters */
  150. struct page_counter memsw;
  151. struct page_counter kmem;
  152. struct page_counter tcpmem;
  153. /* Normal memory consumption range */
  154. unsigned long low;
  155. unsigned long high;
  156. /* Range enforcement for interrupt charges */
  157. struct work_struct high_work;
  158. unsigned long soft_limit;
  159. /* vmpressure notifications */
  160. struct vmpressure vmpressure;
  161. /*
  162. * Should the accounting and control be hierarchical, per subtree?
  163. */
  164. bool use_hierarchy;
  165. /* protected by memcg_oom_lock */
  166. bool oom_lock;
  167. int under_oom;
  168. int swappiness;
  169. /* OOM-Killer disable */
  170. int oom_kill_disable;
  171. /* handle for "memory.events" */
  172. struct cgroup_file events_file;
  173. /* protect arrays of thresholds */
  174. struct mutex thresholds_lock;
  175. /* thresholds for memory usage. RCU-protected */
  176. struct mem_cgroup_thresholds thresholds;
  177. /* thresholds for mem+swap usage. RCU-protected */
  178. struct mem_cgroup_thresholds memsw_thresholds;
  179. /* For oom notifier event fd */
  180. struct list_head oom_notify;
  181. /*
  182. * Should we move charges of a task when a task is moved into this
  183. * mem_cgroup ? And what type of charges should we move ?
  184. */
  185. unsigned long move_charge_at_immigrate;
  186. /*
  187. * set > 0 if pages under this cgroup are moving to other cgroup.
  188. */
  189. atomic_t moving_account;
  190. /* taken only while moving_account > 0 */
  191. spinlock_t move_lock;
  192. struct task_struct *move_lock_task;
  193. unsigned long move_lock_flags;
  194. /*
  195. * percpu counter.
  196. */
  197. struct mem_cgroup_stat_cpu __percpu *stat;
  198. unsigned long socket_pressure;
  199. /* Legacy tcp memory accounting */
  200. bool tcpmem_active;
  201. int tcpmem_pressure;
  202. #ifndef CONFIG_SLOB
  203. /* Index in the kmem_cache->memcg_params.memcg_caches array */
  204. int kmemcg_id;
  205. enum memcg_kmem_state kmem_state;
  206. struct list_head kmem_caches;
  207. #endif
  208. int last_scanned_node;
  209. #if MAX_NUMNODES > 1
  210. nodemask_t scan_nodes;
  211. atomic_t numainfo_events;
  212. atomic_t numainfo_updating;
  213. #endif
  214. #ifdef CONFIG_CGROUP_WRITEBACK
  215. struct list_head cgwb_list;
  216. struct wb_domain cgwb_domain;
  217. #endif
  218. /* List of events which userspace want to receive */
  219. struct list_head event_list;
  220. spinlock_t event_list_lock;
  221. struct mem_cgroup_per_node *nodeinfo[0];
  222. /* WARNING: nodeinfo must be the last member here */
  223. };
  224. extern struct mem_cgroup *root_mem_cgroup;
  225. static inline bool mem_cgroup_disabled(void)
  226. {
  227. return !cgroup_subsys_enabled(memory_cgrp_subsys);
  228. }
  229. static inline void mem_cgroup_event(struct mem_cgroup *memcg,
  230. enum memcg_event_item event)
  231. {
  232. this_cpu_inc(memcg->stat->events[event]);
  233. cgroup_file_notify(&memcg->events_file);
  234. }
  235. bool mem_cgroup_low(struct mem_cgroup *root, struct mem_cgroup *memcg);
  236. int mem_cgroup_try_charge(struct page *page, struct mm_struct *mm,
  237. gfp_t gfp_mask, struct mem_cgroup **memcgp,
  238. bool compound);
  239. void mem_cgroup_commit_charge(struct page *page, struct mem_cgroup *memcg,
  240. bool lrucare, bool compound);
  241. void mem_cgroup_cancel_charge(struct page *page, struct mem_cgroup *memcg,
  242. bool compound);
  243. void mem_cgroup_uncharge(struct page *page);
  244. void mem_cgroup_uncharge_list(struct list_head *page_list);
  245. void mem_cgroup_migrate(struct page *oldpage, struct page *newpage);
  246. static struct mem_cgroup_per_node *
  247. mem_cgroup_nodeinfo(struct mem_cgroup *memcg, int nid)
  248. {
  249. return memcg->nodeinfo[nid];
  250. }
  251. /**
  252. * mem_cgroup_lruvec - get the lru list vector for a node or a memcg zone
  253. * @node: node of the wanted lruvec
  254. * @memcg: memcg of the wanted lruvec
  255. *
  256. * Returns the lru list vector holding pages for a given @node or a given
  257. * @memcg and @zone. This can be the node lruvec, if the memory controller
  258. * is disabled.
  259. */
  260. static inline struct lruvec *mem_cgroup_lruvec(struct pglist_data *pgdat,
  261. struct mem_cgroup *memcg)
  262. {
  263. struct mem_cgroup_per_node *mz;
  264. struct lruvec *lruvec;
  265. if (mem_cgroup_disabled()) {
  266. lruvec = node_lruvec(pgdat);
  267. goto out;
  268. }
  269. mz = mem_cgroup_nodeinfo(memcg, pgdat->node_id);
  270. lruvec = &mz->lruvec;
  271. out:
  272. /*
  273. * Since a node can be onlined after the mem_cgroup was created,
  274. * we have to be prepared to initialize lruvec->pgdat here;
  275. * and if offlined then reonlined, we need to reinitialize it.
  276. */
  277. if (unlikely(lruvec->pgdat != pgdat))
  278. lruvec->pgdat = pgdat;
  279. return lruvec;
  280. }
  281. struct lruvec *mem_cgroup_page_lruvec(struct page *, struct pglist_data *);
  282. bool task_in_mem_cgroup(struct task_struct *task, struct mem_cgroup *memcg);
  283. struct mem_cgroup *mem_cgroup_from_task(struct task_struct *p);
  284. static inline
  285. struct mem_cgroup *mem_cgroup_from_css(struct cgroup_subsys_state *css){
  286. return css ? container_of(css, struct mem_cgroup, css) : NULL;
  287. }
  288. #define mem_cgroup_from_counter(counter, member) \
  289. container_of(counter, struct mem_cgroup, member)
  290. struct mem_cgroup *mem_cgroup_iter(struct mem_cgroup *,
  291. struct mem_cgroup *,
  292. struct mem_cgroup_reclaim_cookie *);
  293. void mem_cgroup_iter_break(struct mem_cgroup *, struct mem_cgroup *);
  294. int mem_cgroup_scan_tasks(struct mem_cgroup *,
  295. int (*)(struct task_struct *, void *), void *);
  296. static inline unsigned short mem_cgroup_id(struct mem_cgroup *memcg)
  297. {
  298. if (mem_cgroup_disabled())
  299. return 0;
  300. return memcg->id.id;
  301. }
  302. struct mem_cgroup *mem_cgroup_from_id(unsigned short id);
  303. static inline struct mem_cgroup *lruvec_memcg(struct lruvec *lruvec)
  304. {
  305. struct mem_cgroup_per_node *mz;
  306. if (mem_cgroup_disabled())
  307. return NULL;
  308. mz = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
  309. return mz->memcg;
  310. }
  311. /**
  312. * parent_mem_cgroup - find the accounting parent of a memcg
  313. * @memcg: memcg whose parent to find
  314. *
  315. * Returns the parent memcg, or NULL if this is the root or the memory
  316. * controller is in legacy no-hierarchy mode.
  317. */
  318. static inline struct mem_cgroup *parent_mem_cgroup(struct mem_cgroup *memcg)
  319. {
  320. if (!memcg->memory.parent)
  321. return NULL;
  322. return mem_cgroup_from_counter(memcg->memory.parent, memory);
  323. }
  324. static inline bool mem_cgroup_is_descendant(struct mem_cgroup *memcg,
  325. struct mem_cgroup *root)
  326. {
  327. if (root == memcg)
  328. return true;
  329. if (!root->use_hierarchy)
  330. return false;
  331. return cgroup_is_descendant(memcg->css.cgroup, root->css.cgroup);
  332. }
  333. static inline bool mm_match_cgroup(struct mm_struct *mm,
  334. struct mem_cgroup *memcg)
  335. {
  336. struct mem_cgroup *task_memcg;
  337. bool match = false;
  338. rcu_read_lock();
  339. task_memcg = mem_cgroup_from_task(rcu_dereference(mm->owner));
  340. if (task_memcg)
  341. match = mem_cgroup_is_descendant(task_memcg, memcg);
  342. rcu_read_unlock();
  343. return match;
  344. }
  345. struct cgroup_subsys_state *mem_cgroup_css_from_page(struct page *page);
  346. ino_t page_cgroup_ino(struct page *page);
  347. static inline bool mem_cgroup_online(struct mem_cgroup *memcg)
  348. {
  349. if (mem_cgroup_disabled())
  350. return true;
  351. return !!(memcg->css.flags & CSS_ONLINE);
  352. }
  353. /*
  354. * For memory reclaim.
  355. */
  356. int mem_cgroup_select_victim_node(struct mem_cgroup *memcg);
  357. void mem_cgroup_update_lru_size(struct lruvec *lruvec, enum lru_list lru,
  358. int zid, int nr_pages);
  359. unsigned long mem_cgroup_node_nr_lru_pages(struct mem_cgroup *memcg,
  360. int nid, unsigned int lru_mask);
  361. static inline
  362. unsigned long mem_cgroup_get_lru_size(struct lruvec *lruvec, enum lru_list lru)
  363. {
  364. struct mem_cgroup_per_node *mz;
  365. unsigned long nr_pages = 0;
  366. int zid;
  367. mz = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
  368. for (zid = 0; zid < MAX_NR_ZONES; zid++)
  369. nr_pages += mz->lru_zone_size[zid][lru];
  370. return nr_pages;
  371. }
  372. static inline
  373. unsigned long mem_cgroup_get_zone_lru_size(struct lruvec *lruvec,
  374. enum lru_list lru, int zone_idx)
  375. {
  376. struct mem_cgroup_per_node *mz;
  377. mz = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
  378. return mz->lru_zone_size[zone_idx][lru];
  379. }
  380. void mem_cgroup_handle_over_high(void);
  381. unsigned long mem_cgroup_get_limit(struct mem_cgroup *memcg);
  382. void mem_cgroup_print_oom_info(struct mem_cgroup *memcg,
  383. struct task_struct *p);
  384. static inline void mem_cgroup_oom_enable(void)
  385. {
  386. WARN_ON(current->memcg_may_oom);
  387. current->memcg_may_oom = 1;
  388. }
  389. static inline void mem_cgroup_oom_disable(void)
  390. {
  391. WARN_ON(!current->memcg_may_oom);
  392. current->memcg_may_oom = 0;
  393. }
  394. static inline bool task_in_memcg_oom(struct task_struct *p)
  395. {
  396. return p->memcg_in_oom;
  397. }
  398. bool mem_cgroup_oom_synchronize(bool wait);
  399. #ifdef CONFIG_MEMCG_SWAP
  400. extern int do_swap_account;
  401. #endif
  402. void lock_page_memcg(struct page *page);
  403. void unlock_page_memcg(struct page *page);
  404. static inline unsigned long memcg_page_state(struct mem_cgroup *memcg,
  405. enum memcg_stat_item idx)
  406. {
  407. long val = 0;
  408. int cpu;
  409. for_each_possible_cpu(cpu)
  410. val += per_cpu(memcg->stat->count[idx], cpu);
  411. if (val < 0)
  412. val = 0;
  413. return val;
  414. }
  415. static inline void __mod_memcg_state(struct mem_cgroup *memcg,
  416. enum memcg_stat_item idx, int val)
  417. {
  418. if (!mem_cgroup_disabled())
  419. __this_cpu_add(memcg->stat->count[idx], val);
  420. }
  421. static inline void mod_memcg_state(struct mem_cgroup *memcg,
  422. enum memcg_stat_item idx, int val)
  423. {
  424. if (!mem_cgroup_disabled())
  425. this_cpu_add(memcg->stat->count[idx], val);
  426. }
  427. /**
  428. * mod_memcg_page_state - update page state statistics
  429. * @page: the page
  430. * @idx: page state item to account
  431. * @val: number of pages (positive or negative)
  432. *
  433. * The @page must be locked or the caller must use lock_page_memcg()
  434. * to prevent double accounting when the page is concurrently being
  435. * moved to another memcg:
  436. *
  437. * lock_page(page) or lock_page_memcg(page)
  438. * if (TestClearPageState(page))
  439. * mod_memcg_page_state(page, state, -1);
  440. * unlock_page(page) or unlock_page_memcg(page)
  441. *
  442. * Kernel pages are an exception to this, since they'll never move.
  443. */
  444. static inline void __mod_memcg_page_state(struct page *page,
  445. enum memcg_stat_item idx, int val)
  446. {
  447. if (page->mem_cgroup)
  448. __mod_memcg_state(page->mem_cgroup, idx, val);
  449. }
  450. static inline void mod_memcg_page_state(struct page *page,
  451. enum memcg_stat_item idx, int val)
  452. {
  453. if (page->mem_cgroup)
  454. mod_memcg_state(page->mem_cgroup, idx, val);
  455. }
  456. static inline unsigned long lruvec_page_state(struct lruvec *lruvec,
  457. enum node_stat_item idx)
  458. {
  459. struct mem_cgroup_per_node *pn;
  460. long val = 0;
  461. int cpu;
  462. if (mem_cgroup_disabled())
  463. return node_page_state(lruvec_pgdat(lruvec), idx);
  464. pn = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
  465. for_each_possible_cpu(cpu)
  466. val += per_cpu(pn->lruvec_stat->count[idx], cpu);
  467. if (val < 0)
  468. val = 0;
  469. return val;
  470. }
  471. static inline void __mod_lruvec_state(struct lruvec *lruvec,
  472. enum node_stat_item idx, int val)
  473. {
  474. struct mem_cgroup_per_node *pn;
  475. __mod_node_page_state(lruvec_pgdat(lruvec), idx, val);
  476. if (mem_cgroup_disabled())
  477. return;
  478. pn = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
  479. __mod_memcg_state(pn->memcg, idx, val);
  480. __this_cpu_add(pn->lruvec_stat->count[idx], val);
  481. }
  482. static inline void mod_lruvec_state(struct lruvec *lruvec,
  483. enum node_stat_item idx, int val)
  484. {
  485. struct mem_cgroup_per_node *pn;
  486. mod_node_page_state(lruvec_pgdat(lruvec), idx, val);
  487. if (mem_cgroup_disabled())
  488. return;
  489. pn = container_of(lruvec, struct mem_cgroup_per_node, lruvec);
  490. mod_memcg_state(pn->memcg, idx, val);
  491. this_cpu_add(pn->lruvec_stat->count[idx], val);
  492. }
  493. static inline void __mod_lruvec_page_state(struct page *page,
  494. enum node_stat_item idx, int val)
  495. {
  496. struct mem_cgroup_per_node *pn;
  497. __mod_node_page_state(page_pgdat(page), idx, val);
  498. if (mem_cgroup_disabled() || !page->mem_cgroup)
  499. return;
  500. __mod_memcg_state(page->mem_cgroup, idx, val);
  501. pn = page->mem_cgroup->nodeinfo[page_to_nid(page)];
  502. __this_cpu_add(pn->lruvec_stat->count[idx], val);
  503. }
  504. static inline void mod_lruvec_page_state(struct page *page,
  505. enum node_stat_item idx, int val)
  506. {
  507. struct mem_cgroup_per_node *pn;
  508. mod_node_page_state(page_pgdat(page), idx, val);
  509. if (mem_cgroup_disabled() || !page->mem_cgroup)
  510. return;
  511. mod_memcg_state(page->mem_cgroup, idx, val);
  512. pn = page->mem_cgroup->nodeinfo[page_to_nid(page)];
  513. this_cpu_add(pn->lruvec_stat->count[idx], val);
  514. }
  515. unsigned long mem_cgroup_soft_limit_reclaim(pg_data_t *pgdat, int order,
  516. gfp_t gfp_mask,
  517. unsigned long *total_scanned);
  518. static inline void count_memcg_events(struct mem_cgroup *memcg,
  519. enum vm_event_item idx,
  520. unsigned long count)
  521. {
  522. if (!mem_cgroup_disabled())
  523. this_cpu_add(memcg->stat->events[idx], count);
  524. }
  525. static inline void count_memcg_page_event(struct page *page,
  526. enum memcg_stat_item idx)
  527. {
  528. if (page->mem_cgroup)
  529. count_memcg_events(page->mem_cgroup, idx, 1);
  530. }
  531. static inline void count_memcg_event_mm(struct mm_struct *mm,
  532. enum vm_event_item idx)
  533. {
  534. struct mem_cgroup *memcg;
  535. if (mem_cgroup_disabled())
  536. return;
  537. rcu_read_lock();
  538. memcg = mem_cgroup_from_task(rcu_dereference(mm->owner));
  539. if (likely(memcg)) {
  540. this_cpu_inc(memcg->stat->events[idx]);
  541. if (idx == OOM_KILL)
  542. cgroup_file_notify(&memcg->events_file);
  543. }
  544. rcu_read_unlock();
  545. }
  546. #ifdef CONFIG_TRANSPARENT_HUGEPAGE
  547. void mem_cgroup_split_huge_fixup(struct page *head);
  548. #endif
  549. #else /* CONFIG_MEMCG */
  550. #define MEM_CGROUP_ID_SHIFT 0
  551. #define MEM_CGROUP_ID_MAX 0
  552. struct mem_cgroup;
  553. static inline bool mem_cgroup_disabled(void)
  554. {
  555. return true;
  556. }
  557. static inline void mem_cgroup_event(struct mem_cgroup *memcg,
  558. enum memcg_event_item event)
  559. {
  560. }
  561. static inline bool mem_cgroup_low(struct mem_cgroup *root,
  562. struct mem_cgroup *memcg)
  563. {
  564. return false;
  565. }
  566. static inline int mem_cgroup_try_charge(struct page *page, struct mm_struct *mm,
  567. gfp_t gfp_mask,
  568. struct mem_cgroup **memcgp,
  569. bool compound)
  570. {
  571. *memcgp = NULL;
  572. return 0;
  573. }
  574. static inline void mem_cgroup_commit_charge(struct page *page,
  575. struct mem_cgroup *memcg,
  576. bool lrucare, bool compound)
  577. {
  578. }
  579. static inline void mem_cgroup_cancel_charge(struct page *page,
  580. struct mem_cgroup *memcg,
  581. bool compound)
  582. {
  583. }
  584. static inline void mem_cgroup_uncharge(struct page *page)
  585. {
  586. }
  587. static inline void mem_cgroup_uncharge_list(struct list_head *page_list)
  588. {
  589. }
  590. static inline void mem_cgroup_migrate(struct page *old, struct page *new)
  591. {
  592. }
  593. static inline struct lruvec *mem_cgroup_lruvec(struct pglist_data *pgdat,
  594. struct mem_cgroup *memcg)
  595. {
  596. return node_lruvec(pgdat);
  597. }
  598. static inline struct lruvec *mem_cgroup_page_lruvec(struct page *page,
  599. struct pglist_data *pgdat)
  600. {
  601. return &pgdat->lruvec;
  602. }
  603. static inline bool mm_match_cgroup(struct mm_struct *mm,
  604. struct mem_cgroup *memcg)
  605. {
  606. return true;
  607. }
  608. static inline bool task_in_mem_cgroup(struct task_struct *task,
  609. const struct mem_cgroup *memcg)
  610. {
  611. return true;
  612. }
  613. static inline struct mem_cgroup *
  614. mem_cgroup_iter(struct mem_cgroup *root,
  615. struct mem_cgroup *prev,
  616. struct mem_cgroup_reclaim_cookie *reclaim)
  617. {
  618. return NULL;
  619. }
  620. static inline void mem_cgroup_iter_break(struct mem_cgroup *root,
  621. struct mem_cgroup *prev)
  622. {
  623. }
  624. static inline int mem_cgroup_scan_tasks(struct mem_cgroup *memcg,
  625. int (*fn)(struct task_struct *, void *), void *arg)
  626. {
  627. return 0;
  628. }
  629. static inline unsigned short mem_cgroup_id(struct mem_cgroup *memcg)
  630. {
  631. return 0;
  632. }
  633. static inline struct mem_cgroup *mem_cgroup_from_id(unsigned short id)
  634. {
  635. WARN_ON_ONCE(id);
  636. /* XXX: This should always return root_mem_cgroup */
  637. return NULL;
  638. }
  639. static inline struct mem_cgroup *lruvec_memcg(struct lruvec *lruvec)
  640. {
  641. return NULL;
  642. }
  643. static inline bool mem_cgroup_online(struct mem_cgroup *memcg)
  644. {
  645. return true;
  646. }
  647. static inline unsigned long
  648. mem_cgroup_get_lru_size(struct lruvec *lruvec, enum lru_list lru)
  649. {
  650. return 0;
  651. }
  652. static inline
  653. unsigned long mem_cgroup_get_zone_lru_size(struct lruvec *lruvec,
  654. enum lru_list lru, int zone_idx)
  655. {
  656. return 0;
  657. }
  658. static inline unsigned long
  659. mem_cgroup_node_nr_lru_pages(struct mem_cgroup *memcg,
  660. int nid, unsigned int lru_mask)
  661. {
  662. return 0;
  663. }
  664. static inline unsigned long mem_cgroup_get_limit(struct mem_cgroup *memcg)
  665. {
  666. return 0;
  667. }
  668. static inline void
  669. mem_cgroup_print_oom_info(struct mem_cgroup *memcg, struct task_struct *p)
  670. {
  671. }
  672. static inline void lock_page_memcg(struct page *page)
  673. {
  674. }
  675. static inline void unlock_page_memcg(struct page *page)
  676. {
  677. }
  678. static inline void mem_cgroup_handle_over_high(void)
  679. {
  680. }
  681. static inline void mem_cgroup_oom_enable(void)
  682. {
  683. }
  684. static inline void mem_cgroup_oom_disable(void)
  685. {
  686. }
  687. static inline bool task_in_memcg_oom(struct task_struct *p)
  688. {
  689. return false;
  690. }
  691. static inline bool mem_cgroup_oom_synchronize(bool wait)
  692. {
  693. return false;
  694. }
  695. static inline unsigned long memcg_page_state(struct mem_cgroup *memcg,
  696. enum memcg_stat_item idx)
  697. {
  698. return 0;
  699. }
  700. static inline void __mod_memcg_state(struct mem_cgroup *memcg,
  701. enum memcg_stat_item idx,
  702. int nr)
  703. {
  704. }
  705. static inline void mod_memcg_state(struct mem_cgroup *memcg,
  706. enum memcg_stat_item idx,
  707. int nr)
  708. {
  709. }
  710. static inline void __mod_memcg_page_state(struct page *page,
  711. enum memcg_stat_item idx,
  712. int nr)
  713. {
  714. }
  715. static inline void mod_memcg_page_state(struct page *page,
  716. enum memcg_stat_item idx,
  717. int nr)
  718. {
  719. }
  720. static inline unsigned long lruvec_page_state(struct lruvec *lruvec,
  721. enum node_stat_item idx)
  722. {
  723. return node_page_state(lruvec_pgdat(lruvec), idx);
  724. }
  725. static inline void __mod_lruvec_state(struct lruvec *lruvec,
  726. enum node_stat_item idx, int val)
  727. {
  728. __mod_node_page_state(lruvec_pgdat(lruvec), idx, val);
  729. }
  730. static inline void mod_lruvec_state(struct lruvec *lruvec,
  731. enum node_stat_item idx, int val)
  732. {
  733. mod_node_page_state(lruvec_pgdat(lruvec), idx, val);
  734. }
  735. static inline void __mod_lruvec_page_state(struct page *page,
  736. enum node_stat_item idx, int val)
  737. {
  738. __mod_node_page_state(page_pgdat(page), idx, val);
  739. }
  740. static inline void mod_lruvec_page_state(struct page *page,
  741. enum node_stat_item idx, int val)
  742. {
  743. mod_node_page_state(page_pgdat(page), idx, val);
  744. }
  745. static inline
  746. unsigned long mem_cgroup_soft_limit_reclaim(pg_data_t *pgdat, int order,
  747. gfp_t gfp_mask,
  748. unsigned long *total_scanned)
  749. {
  750. return 0;
  751. }
  752. static inline void mem_cgroup_split_huge_fixup(struct page *head)
  753. {
  754. }
  755. static inline void count_memcg_events(struct mem_cgroup *memcg,
  756. enum vm_event_item idx,
  757. unsigned long count)
  758. {
  759. }
  760. static inline void count_memcg_page_event(struct page *page,
  761. enum memcg_stat_item idx)
  762. {
  763. }
  764. static inline
  765. void count_memcg_event_mm(struct mm_struct *mm, enum vm_event_item idx)
  766. {
  767. }
  768. #endif /* CONFIG_MEMCG */
  769. static inline void __inc_memcg_state(struct mem_cgroup *memcg,
  770. enum memcg_stat_item idx)
  771. {
  772. __mod_memcg_state(memcg, idx, 1);
  773. }
  774. static inline void __dec_memcg_state(struct mem_cgroup *memcg,
  775. enum memcg_stat_item idx)
  776. {
  777. __mod_memcg_state(memcg, idx, -1);
  778. }
  779. static inline void __inc_memcg_page_state(struct page *page,
  780. enum memcg_stat_item idx)
  781. {
  782. __mod_memcg_page_state(page, idx, 1);
  783. }
  784. static inline void __dec_memcg_page_state(struct page *page,
  785. enum memcg_stat_item idx)
  786. {
  787. __mod_memcg_page_state(page, idx, -1);
  788. }
  789. static inline void __inc_lruvec_state(struct lruvec *lruvec,
  790. enum node_stat_item idx)
  791. {
  792. __mod_lruvec_state(lruvec, idx, 1);
  793. }
  794. static inline void __dec_lruvec_state(struct lruvec *lruvec,
  795. enum node_stat_item idx)
  796. {
  797. __mod_lruvec_state(lruvec, idx, -1);
  798. }
  799. static inline void __inc_lruvec_page_state(struct page *page,
  800. enum node_stat_item idx)
  801. {
  802. __mod_lruvec_page_state(page, idx, 1);
  803. }
  804. static inline void __dec_lruvec_page_state(struct page *page,
  805. enum node_stat_item idx)
  806. {
  807. __mod_lruvec_page_state(page, idx, -1);
  808. }
  809. static inline void inc_memcg_state(struct mem_cgroup *memcg,
  810. enum memcg_stat_item idx)
  811. {
  812. mod_memcg_state(memcg, idx, 1);
  813. }
  814. static inline void dec_memcg_state(struct mem_cgroup *memcg,
  815. enum memcg_stat_item idx)
  816. {
  817. mod_memcg_state(memcg, idx, -1);
  818. }
  819. static inline void inc_memcg_page_state(struct page *page,
  820. enum memcg_stat_item idx)
  821. {
  822. mod_memcg_page_state(page, idx, 1);
  823. }
  824. static inline void dec_memcg_page_state(struct page *page,
  825. enum memcg_stat_item idx)
  826. {
  827. mod_memcg_page_state(page, idx, -1);
  828. }
  829. static inline void inc_lruvec_state(struct lruvec *lruvec,
  830. enum node_stat_item idx)
  831. {
  832. mod_lruvec_state(lruvec, idx, 1);
  833. }
  834. static inline void dec_lruvec_state(struct lruvec *lruvec,
  835. enum node_stat_item idx)
  836. {
  837. mod_lruvec_state(lruvec, idx, -1);
  838. }
  839. static inline void inc_lruvec_page_state(struct page *page,
  840. enum node_stat_item idx)
  841. {
  842. mod_lruvec_page_state(page, idx, 1);
  843. }
  844. static inline void dec_lruvec_page_state(struct page *page,
  845. enum node_stat_item idx)
  846. {
  847. mod_lruvec_page_state(page, idx, -1);
  848. }
  849. #ifdef CONFIG_CGROUP_WRITEBACK
  850. struct list_head *mem_cgroup_cgwb_list(struct mem_cgroup *memcg);
  851. struct wb_domain *mem_cgroup_wb_domain(struct bdi_writeback *wb);
  852. void mem_cgroup_wb_stats(struct bdi_writeback *wb, unsigned long *pfilepages,
  853. unsigned long *pheadroom, unsigned long *pdirty,
  854. unsigned long *pwriteback);
  855. #else /* CONFIG_CGROUP_WRITEBACK */
  856. static inline struct wb_domain *mem_cgroup_wb_domain(struct bdi_writeback *wb)
  857. {
  858. return NULL;
  859. }
  860. static inline void mem_cgroup_wb_stats(struct bdi_writeback *wb,
  861. unsigned long *pfilepages,
  862. unsigned long *pheadroom,
  863. unsigned long *pdirty,
  864. unsigned long *pwriteback)
  865. {
  866. }
  867. #endif /* CONFIG_CGROUP_WRITEBACK */
  868. struct sock;
  869. bool mem_cgroup_charge_skmem(struct mem_cgroup *memcg, unsigned int nr_pages);
  870. void mem_cgroup_uncharge_skmem(struct mem_cgroup *memcg, unsigned int nr_pages);
  871. #ifdef CONFIG_MEMCG
  872. extern struct static_key_false memcg_sockets_enabled_key;
  873. #define mem_cgroup_sockets_enabled static_branch_unlikely(&memcg_sockets_enabled_key)
  874. void mem_cgroup_sk_alloc(struct sock *sk);
  875. void mem_cgroup_sk_free(struct sock *sk);
  876. static inline bool mem_cgroup_under_socket_pressure(struct mem_cgroup *memcg)
  877. {
  878. if (!cgroup_subsys_on_dfl(memory_cgrp_subsys) && memcg->tcpmem_pressure)
  879. return true;
  880. do {
  881. if (time_before(jiffies, memcg->socket_pressure))
  882. return true;
  883. } while ((memcg = parent_mem_cgroup(memcg)));
  884. return false;
  885. }
  886. #else
  887. #define mem_cgroup_sockets_enabled 0
  888. static inline void mem_cgroup_sk_alloc(struct sock *sk) { };
  889. static inline void mem_cgroup_sk_free(struct sock *sk) { };
  890. static inline bool mem_cgroup_under_socket_pressure(struct mem_cgroup *memcg)
  891. {
  892. return false;
  893. }
  894. #endif
  895. struct kmem_cache *memcg_kmem_get_cache(struct kmem_cache *cachep);
  896. void memcg_kmem_put_cache(struct kmem_cache *cachep);
  897. int memcg_kmem_charge_memcg(struct page *page, gfp_t gfp, int order,
  898. struct mem_cgroup *memcg);
  899. int memcg_kmem_charge(struct page *page, gfp_t gfp, int order);
  900. void memcg_kmem_uncharge(struct page *page, int order);
  901. #if defined(CONFIG_MEMCG) && !defined(CONFIG_SLOB)
  902. extern struct static_key_false memcg_kmem_enabled_key;
  903. extern struct workqueue_struct *memcg_kmem_cache_wq;
  904. extern int memcg_nr_cache_ids;
  905. void memcg_get_cache_ids(void);
  906. void memcg_put_cache_ids(void);
  907. /*
  908. * Helper macro to loop through all memcg-specific caches. Callers must still
  909. * check if the cache is valid (it is either valid or NULL).
  910. * the slab_mutex must be held when looping through those caches
  911. */
  912. #define for_each_memcg_cache_index(_idx) \
  913. for ((_idx) = 0; (_idx) < memcg_nr_cache_ids; (_idx)++)
  914. static inline bool memcg_kmem_enabled(void)
  915. {
  916. return static_branch_unlikely(&memcg_kmem_enabled_key);
  917. }
  918. /*
  919. * helper for accessing a memcg's index. It will be used as an index in the
  920. * child cache array in kmem_cache, and also to derive its name. This function
  921. * will return -1 when this is not a kmem-limited memcg.
  922. */
  923. static inline int memcg_cache_id(struct mem_cgroup *memcg)
  924. {
  925. return memcg ? memcg->kmemcg_id : -1;
  926. }
  927. #else
  928. #define for_each_memcg_cache_index(_idx) \
  929. for (; NULL; )
  930. static inline bool memcg_kmem_enabled(void)
  931. {
  932. return false;
  933. }
  934. static inline int memcg_cache_id(struct mem_cgroup *memcg)
  935. {
  936. return -1;
  937. }
  938. static inline void memcg_get_cache_ids(void)
  939. {
  940. }
  941. static inline void memcg_put_cache_ids(void)
  942. {
  943. }
  944. #endif /* CONFIG_MEMCG && !CONFIG_SLOB */
  945. #endif /* _LINUX_MEMCONTROL_H */