include/linux/memcontrol.h

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