Merge tag 'perf-tools-for-v6.6-1-2023-09-05' of git://git.kernel.org/pub/scm/linux...
[linux-2.6-microblaze.git] / mm / vmscan.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  *  Copyright (C) 1991, 1992, 1993, 1994  Linus Torvalds
4  *
5  *  Swap reorganised 29.12.95, Stephen Tweedie.
6  *  kswapd added: 7.1.96  sct
7  *  Removed kswapd_ctl limits, and swap out as many pages as needed
8  *  to bring the system back to freepages.high: 2.4.97, Rik van Riel.
9  *  Zone aware kswapd started 02/00, Kanoj Sarcar (kanoj@sgi.com).
10  *  Multiqueue VM started 5.8.00, Rik van Riel.
11  */
12
13 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
14
15 #include <linux/mm.h>
16 #include <linux/sched/mm.h>
17 #include <linux/module.h>
18 #include <linux/gfp.h>
19 #include <linux/kernel_stat.h>
20 #include <linux/swap.h>
21 #include <linux/pagemap.h>
22 #include <linux/init.h>
23 #include <linux/highmem.h>
24 #include <linux/vmpressure.h>
25 #include <linux/vmstat.h>
26 #include <linux/file.h>
27 #include <linux/writeback.h>
28 #include <linux/blkdev.h>
29 #include <linux/buffer_head.h>  /* for buffer_heads_over_limit */
30 #include <linux/mm_inline.h>
31 #include <linux/backing-dev.h>
32 #include <linux/rmap.h>
33 #include <linux/topology.h>
34 #include <linux/cpu.h>
35 #include <linux/cpuset.h>
36 #include <linux/compaction.h>
37 #include <linux/notifier.h>
38 #include <linux/rwsem.h>
39 #include <linux/delay.h>
40 #include <linux/kthread.h>
41 #include <linux/freezer.h>
42 #include <linux/memcontrol.h>
43 #include <linux/migrate.h>
44 #include <linux/delayacct.h>
45 #include <linux/sysctl.h>
46 #include <linux/memory-tiers.h>
47 #include <linux/oom.h>
48 #include <linux/pagevec.h>
49 #include <linux/prefetch.h>
50 #include <linux/printk.h>
51 #include <linux/dax.h>
52 #include <linux/psi.h>
53 #include <linux/pagewalk.h>
54 #include <linux/shmem_fs.h>
55 #include <linux/ctype.h>
56 #include <linux/debugfs.h>
57 #include <linux/khugepaged.h>
58 #include <linux/rculist_nulls.h>
59 #include <linux/random.h>
60
61 #include <asm/tlbflush.h>
62 #include <asm/div64.h>
63
64 #include <linux/swapops.h>
65 #include <linux/balloon_compaction.h>
66 #include <linux/sched/sysctl.h>
67
68 #include "internal.h"
69 #include "swap.h"
70
71 #define CREATE_TRACE_POINTS
72 #include <trace/events/vmscan.h>
73
74 struct scan_control {
75         /* How many pages shrink_list() should reclaim */
76         unsigned long nr_to_reclaim;
77
78         /*
79          * Nodemask of nodes allowed by the caller. If NULL, all nodes
80          * are scanned.
81          */
82         nodemask_t      *nodemask;
83
84         /*
85          * The memory cgroup that hit its limit and as a result is the
86          * primary target of this reclaim invocation.
87          */
88         struct mem_cgroup *target_mem_cgroup;
89
90         /*
91          * Scan pressure balancing between anon and file LRUs
92          */
93         unsigned long   anon_cost;
94         unsigned long   file_cost;
95
96         /* Can active folios be deactivated as part of reclaim? */
97 #define DEACTIVATE_ANON 1
98 #define DEACTIVATE_FILE 2
99         unsigned int may_deactivate:2;
100         unsigned int force_deactivate:1;
101         unsigned int skipped_deactivate:1;
102
103         /* Writepage batching in laptop mode; RECLAIM_WRITE */
104         unsigned int may_writepage:1;
105
106         /* Can mapped folios be reclaimed? */
107         unsigned int may_unmap:1;
108
109         /* Can folios be swapped as part of reclaim? */
110         unsigned int may_swap:1;
111
112         /* Proactive reclaim invoked by userspace through memory.reclaim */
113         unsigned int proactive:1;
114
115         /*
116          * Cgroup memory below memory.low is protected as long as we
117          * don't threaten to OOM. If any cgroup is reclaimed at
118          * reduced force or passed over entirely due to its memory.low
119          * setting (memcg_low_skipped), and nothing is reclaimed as a
120          * result, then go back for one more cycle that reclaims the protected
121          * memory (memcg_low_reclaim) to avert OOM.
122          */
123         unsigned int memcg_low_reclaim:1;
124         unsigned int memcg_low_skipped:1;
125
126         unsigned int hibernation_mode:1;
127
128         /* One of the zones is ready for compaction */
129         unsigned int compaction_ready:1;
130
131         /* There is easily reclaimable cold cache in the current node */
132         unsigned int cache_trim_mode:1;
133
134         /* The file folios on the current node are dangerously low */
135         unsigned int file_is_tiny:1;
136
137         /* Always discard instead of demoting to lower tier memory */
138         unsigned int no_demotion:1;
139
140         /* Allocation order */
141         s8 order;
142
143         /* Scan (total_size >> priority) pages at once */
144         s8 priority;
145
146         /* The highest zone to isolate folios for reclaim from */
147         s8 reclaim_idx;
148
149         /* This context's GFP mask */
150         gfp_t gfp_mask;
151
152         /* Incremented by the number of inactive pages that were scanned */
153         unsigned long nr_scanned;
154
155         /* Number of pages freed so far during a call to shrink_zones() */
156         unsigned long nr_reclaimed;
157
158         struct {
159                 unsigned int dirty;
160                 unsigned int unqueued_dirty;
161                 unsigned int congested;
162                 unsigned int writeback;
163                 unsigned int immediate;
164                 unsigned int file_taken;
165                 unsigned int taken;
166         } nr;
167
168         /* for recording the reclaimed slab by now */
169         struct reclaim_state reclaim_state;
170 };
171
172 #ifdef ARCH_HAS_PREFETCHW
173 #define prefetchw_prev_lru_folio(_folio, _base, _field)                 \
174         do {                                                            \
175                 if ((_folio)->lru.prev != _base) {                      \
176                         struct folio *prev;                             \
177                                                                         \
178                         prev = lru_to_folio(&(_folio->lru));            \
179                         prefetchw(&prev->_field);                       \
180                 }                                                       \
181         } while (0)
182 #else
183 #define prefetchw_prev_lru_folio(_folio, _base, _field) do { } while (0)
184 #endif
185
186 /*
187  * From 0 .. 200.  Higher means more swappy.
188  */
189 int vm_swappiness = 60;
190
191 LIST_HEAD(shrinker_list);
192 DECLARE_RWSEM(shrinker_rwsem);
193
194 #ifdef CONFIG_MEMCG
195 static int shrinker_nr_max;
196
197 /* The shrinker_info is expanded in a batch of BITS_PER_LONG */
198 static inline int shrinker_map_size(int nr_items)
199 {
200         return (DIV_ROUND_UP(nr_items, BITS_PER_LONG) * sizeof(unsigned long));
201 }
202
203 static inline int shrinker_defer_size(int nr_items)
204 {
205         return (round_up(nr_items, BITS_PER_LONG) * sizeof(atomic_long_t));
206 }
207
208 static struct shrinker_info *shrinker_info_protected(struct mem_cgroup *memcg,
209                                                      int nid)
210 {
211         return rcu_dereference_protected(memcg->nodeinfo[nid]->shrinker_info,
212                                          lockdep_is_held(&shrinker_rwsem));
213 }
214
215 static int expand_one_shrinker_info(struct mem_cgroup *memcg,
216                                     int map_size, int defer_size,
217                                     int old_map_size, int old_defer_size,
218                                     int new_nr_max)
219 {
220         struct shrinker_info *new, *old;
221         struct mem_cgroup_per_node *pn;
222         int nid;
223         int size = map_size + defer_size;
224
225         for_each_node(nid) {
226                 pn = memcg->nodeinfo[nid];
227                 old = shrinker_info_protected(memcg, nid);
228                 /* Not yet online memcg */
229                 if (!old)
230                         return 0;
231
232                 /* Already expanded this shrinker_info */
233                 if (new_nr_max <= old->map_nr_max)
234                         continue;
235
236                 new = kvmalloc_node(sizeof(*new) + size, GFP_KERNEL, nid);
237                 if (!new)
238                         return -ENOMEM;
239
240                 new->nr_deferred = (atomic_long_t *)(new + 1);
241                 new->map = (void *)new->nr_deferred + defer_size;
242                 new->map_nr_max = new_nr_max;
243
244                 /* map: set all old bits, clear all new bits */
245                 memset(new->map, (int)0xff, old_map_size);
246                 memset((void *)new->map + old_map_size, 0, map_size - old_map_size);
247                 /* nr_deferred: copy old values, clear all new values */
248                 memcpy(new->nr_deferred, old->nr_deferred, old_defer_size);
249                 memset((void *)new->nr_deferred + old_defer_size, 0,
250                        defer_size - old_defer_size);
251
252                 rcu_assign_pointer(pn->shrinker_info, new);
253                 kvfree_rcu(old, rcu);
254         }
255
256         return 0;
257 }
258
259 void free_shrinker_info(struct mem_cgroup *memcg)
260 {
261         struct mem_cgroup_per_node *pn;
262         struct shrinker_info *info;
263         int nid;
264
265         for_each_node(nid) {
266                 pn = memcg->nodeinfo[nid];
267                 info = rcu_dereference_protected(pn->shrinker_info, true);
268                 kvfree(info);
269                 rcu_assign_pointer(pn->shrinker_info, NULL);
270         }
271 }
272
273 int alloc_shrinker_info(struct mem_cgroup *memcg)
274 {
275         struct shrinker_info *info;
276         int nid, size, ret = 0;
277         int map_size, defer_size = 0;
278
279         down_write(&shrinker_rwsem);
280         map_size = shrinker_map_size(shrinker_nr_max);
281         defer_size = shrinker_defer_size(shrinker_nr_max);
282         size = map_size + defer_size;
283         for_each_node(nid) {
284                 info = kvzalloc_node(sizeof(*info) + size, GFP_KERNEL, nid);
285                 if (!info) {
286                         free_shrinker_info(memcg);
287                         ret = -ENOMEM;
288                         break;
289                 }
290                 info->nr_deferred = (atomic_long_t *)(info + 1);
291                 info->map = (void *)info->nr_deferred + defer_size;
292                 info->map_nr_max = shrinker_nr_max;
293                 rcu_assign_pointer(memcg->nodeinfo[nid]->shrinker_info, info);
294         }
295         up_write(&shrinker_rwsem);
296
297         return ret;
298 }
299
300 static int expand_shrinker_info(int new_id)
301 {
302         int ret = 0;
303         int new_nr_max = round_up(new_id + 1, BITS_PER_LONG);
304         int map_size, defer_size = 0;
305         int old_map_size, old_defer_size = 0;
306         struct mem_cgroup *memcg;
307
308         if (!root_mem_cgroup)
309                 goto out;
310
311         lockdep_assert_held(&shrinker_rwsem);
312
313         map_size = shrinker_map_size(new_nr_max);
314         defer_size = shrinker_defer_size(new_nr_max);
315         old_map_size = shrinker_map_size(shrinker_nr_max);
316         old_defer_size = shrinker_defer_size(shrinker_nr_max);
317
318         memcg = mem_cgroup_iter(NULL, NULL, NULL);
319         do {
320                 ret = expand_one_shrinker_info(memcg, map_size, defer_size,
321                                                old_map_size, old_defer_size,
322                                                new_nr_max);
323                 if (ret) {
324                         mem_cgroup_iter_break(NULL, memcg);
325                         goto out;
326                 }
327         } while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)) != NULL);
328 out:
329         if (!ret)
330                 shrinker_nr_max = new_nr_max;
331
332         return ret;
333 }
334
335 void set_shrinker_bit(struct mem_cgroup *memcg, int nid, int shrinker_id)
336 {
337         if (shrinker_id >= 0 && memcg && !mem_cgroup_is_root(memcg)) {
338                 struct shrinker_info *info;
339
340                 rcu_read_lock();
341                 info = rcu_dereference(memcg->nodeinfo[nid]->shrinker_info);
342                 if (!WARN_ON_ONCE(shrinker_id >= info->map_nr_max)) {
343                         /* Pairs with smp mb in shrink_slab() */
344                         smp_mb__before_atomic();
345                         set_bit(shrinker_id, info->map);
346                 }
347                 rcu_read_unlock();
348         }
349 }
350
351 static DEFINE_IDR(shrinker_idr);
352
353 static int prealloc_memcg_shrinker(struct shrinker *shrinker)
354 {
355         int id, ret = -ENOMEM;
356
357         if (mem_cgroup_disabled())
358                 return -ENOSYS;
359
360         down_write(&shrinker_rwsem);
361         /* This may call shrinker, so it must use down_read_trylock() */
362         id = idr_alloc(&shrinker_idr, shrinker, 0, 0, GFP_KERNEL);
363         if (id < 0)
364                 goto unlock;
365
366         if (id >= shrinker_nr_max) {
367                 if (expand_shrinker_info(id)) {
368                         idr_remove(&shrinker_idr, id);
369                         goto unlock;
370                 }
371         }
372         shrinker->id = id;
373         ret = 0;
374 unlock:
375         up_write(&shrinker_rwsem);
376         return ret;
377 }
378
379 static void unregister_memcg_shrinker(struct shrinker *shrinker)
380 {
381         int id = shrinker->id;
382
383         BUG_ON(id < 0);
384
385         lockdep_assert_held(&shrinker_rwsem);
386
387         idr_remove(&shrinker_idr, id);
388 }
389
390 static long xchg_nr_deferred_memcg(int nid, struct shrinker *shrinker,
391                                    struct mem_cgroup *memcg)
392 {
393         struct shrinker_info *info;
394
395         info = shrinker_info_protected(memcg, nid);
396         return atomic_long_xchg(&info->nr_deferred[shrinker->id], 0);
397 }
398
399 static long add_nr_deferred_memcg(long nr, int nid, struct shrinker *shrinker,
400                                   struct mem_cgroup *memcg)
401 {
402         struct shrinker_info *info;
403
404         info = shrinker_info_protected(memcg, nid);
405         return atomic_long_add_return(nr, &info->nr_deferred[shrinker->id]);
406 }
407
408 void reparent_shrinker_deferred(struct mem_cgroup *memcg)
409 {
410         int i, nid;
411         long nr;
412         struct mem_cgroup *parent;
413         struct shrinker_info *child_info, *parent_info;
414
415         parent = parent_mem_cgroup(memcg);
416         if (!parent)
417                 parent = root_mem_cgroup;
418
419         /* Prevent from concurrent shrinker_info expand */
420         down_read(&shrinker_rwsem);
421         for_each_node(nid) {
422                 child_info = shrinker_info_protected(memcg, nid);
423                 parent_info = shrinker_info_protected(parent, nid);
424                 for (i = 0; i < child_info->map_nr_max; i++) {
425                         nr = atomic_long_read(&child_info->nr_deferred[i]);
426                         atomic_long_add(nr, &parent_info->nr_deferred[i]);
427                 }
428         }
429         up_read(&shrinker_rwsem);
430 }
431
432 /* Returns true for reclaim through cgroup limits or cgroup interfaces. */
433 static bool cgroup_reclaim(struct scan_control *sc)
434 {
435         return sc->target_mem_cgroup;
436 }
437
438 /*
439  * Returns true for reclaim on the root cgroup. This is true for direct
440  * allocator reclaim and reclaim through cgroup interfaces on the root cgroup.
441  */
442 static bool root_reclaim(struct scan_control *sc)
443 {
444         return !sc->target_mem_cgroup || mem_cgroup_is_root(sc->target_mem_cgroup);
445 }
446
447 /**
448  * writeback_throttling_sane - is the usual dirty throttling mechanism available?
449  * @sc: scan_control in question
450  *
451  * The normal page dirty throttling mechanism in balance_dirty_pages() is
452  * completely broken with the legacy memcg and direct stalling in
453  * shrink_folio_list() is used for throttling instead, which lacks all the
454  * niceties such as fairness, adaptive pausing, bandwidth proportional
455  * allocation and configurability.
456  *
457  * This function tests whether the vmscan currently in progress can assume
458  * that the normal dirty throttling mechanism is operational.
459  */
460 static bool writeback_throttling_sane(struct scan_control *sc)
461 {
462         if (!cgroup_reclaim(sc))
463                 return true;
464 #ifdef CONFIG_CGROUP_WRITEBACK
465         if (cgroup_subsys_on_dfl(memory_cgrp_subsys))
466                 return true;
467 #endif
468         return false;
469 }
470 #else
471 static int prealloc_memcg_shrinker(struct shrinker *shrinker)
472 {
473         return -ENOSYS;
474 }
475
476 static void unregister_memcg_shrinker(struct shrinker *shrinker)
477 {
478 }
479
480 static long xchg_nr_deferred_memcg(int nid, struct shrinker *shrinker,
481                                    struct mem_cgroup *memcg)
482 {
483         return 0;
484 }
485
486 static long add_nr_deferred_memcg(long nr, int nid, struct shrinker *shrinker,
487                                   struct mem_cgroup *memcg)
488 {
489         return 0;
490 }
491
492 static bool cgroup_reclaim(struct scan_control *sc)
493 {
494         return false;
495 }
496
497 static bool root_reclaim(struct scan_control *sc)
498 {
499         return true;
500 }
501
502 static bool writeback_throttling_sane(struct scan_control *sc)
503 {
504         return true;
505 }
506 #endif
507
508 static void set_task_reclaim_state(struct task_struct *task,
509                                    struct reclaim_state *rs)
510 {
511         /* Check for an overwrite */
512         WARN_ON_ONCE(rs && task->reclaim_state);
513
514         /* Check for the nulling of an already-nulled member */
515         WARN_ON_ONCE(!rs && !task->reclaim_state);
516
517         task->reclaim_state = rs;
518 }
519
520 /*
521  * flush_reclaim_state(): add pages reclaimed outside of LRU-based reclaim to
522  * scan_control->nr_reclaimed.
523  */
524 static void flush_reclaim_state(struct scan_control *sc)
525 {
526         /*
527          * Currently, reclaim_state->reclaimed includes three types of pages
528          * freed outside of vmscan:
529          * (1) Slab pages.
530          * (2) Clean file pages from pruned inodes (on highmem systems).
531          * (3) XFS freed buffer pages.
532          *
533          * For all of these cases, we cannot universally link the pages to a
534          * single memcg. For example, a memcg-aware shrinker can free one object
535          * charged to the target memcg, causing an entire page to be freed.
536          * If we count the entire page as reclaimed from the memcg, we end up
537          * overestimating the reclaimed amount (potentially under-reclaiming).
538          *
539          * Only count such pages for global reclaim to prevent under-reclaiming
540          * from the target memcg; preventing unnecessary retries during memcg
541          * charging and false positives from proactive reclaim.
542          *
543          * For uncommon cases where the freed pages were actually mostly
544          * charged to the target memcg, we end up underestimating the reclaimed
545          * amount. This should be fine. The freed pages will be uncharged
546          * anyway, even if they are not counted here properly, and we will be
547          * able to make forward progress in charging (which is usually in a
548          * retry loop).
549          *
550          * We can go one step further, and report the uncharged objcg pages in
551          * memcg reclaim, to make reporting more accurate and reduce
552          * underestimation, but it's probably not worth the complexity for now.
553          */
554         if (current->reclaim_state && root_reclaim(sc)) {
555                 sc->nr_reclaimed += current->reclaim_state->reclaimed;
556                 current->reclaim_state->reclaimed = 0;
557         }
558 }
559
560 static long xchg_nr_deferred(struct shrinker *shrinker,
561                              struct shrink_control *sc)
562 {
563         int nid = sc->nid;
564
565         if (!(shrinker->flags & SHRINKER_NUMA_AWARE))
566                 nid = 0;
567
568         if (sc->memcg &&
569             (shrinker->flags & SHRINKER_MEMCG_AWARE))
570                 return xchg_nr_deferred_memcg(nid, shrinker,
571                                               sc->memcg);
572
573         return atomic_long_xchg(&shrinker->nr_deferred[nid], 0);
574 }
575
576
577 static long add_nr_deferred(long nr, struct shrinker *shrinker,
578                             struct shrink_control *sc)
579 {
580         int nid = sc->nid;
581
582         if (!(shrinker->flags & SHRINKER_NUMA_AWARE))
583                 nid = 0;
584
585         if (sc->memcg &&
586             (shrinker->flags & SHRINKER_MEMCG_AWARE))
587                 return add_nr_deferred_memcg(nr, nid, shrinker,
588                                              sc->memcg);
589
590         return atomic_long_add_return(nr, &shrinker->nr_deferred[nid]);
591 }
592
593 static bool can_demote(int nid, struct scan_control *sc)
594 {
595         if (!numa_demotion_enabled)
596                 return false;
597         if (sc && sc->no_demotion)
598                 return false;
599         if (next_demotion_node(nid) == NUMA_NO_NODE)
600                 return false;
601
602         return true;
603 }
604
605 static inline bool can_reclaim_anon_pages(struct mem_cgroup *memcg,
606                                           int nid,
607                                           struct scan_control *sc)
608 {
609         if (memcg == NULL) {
610                 /*
611                  * For non-memcg reclaim, is there
612                  * space in any swap device?
613                  */
614                 if (get_nr_swap_pages() > 0)
615                         return true;
616         } else {
617                 /* Is the memcg below its swap limit? */
618                 if (mem_cgroup_get_nr_swap_pages(memcg) > 0)
619                         return true;
620         }
621
622         /*
623          * The page can not be swapped.
624          *
625          * Can it be reclaimed from this node via demotion?
626          */
627         return can_demote(nid, sc);
628 }
629
630 /*
631  * This misses isolated folios which are not accounted for to save counters.
632  * As the data only determines if reclaim or compaction continues, it is
633  * not expected that isolated folios will be a dominating factor.
634  */
635 unsigned long zone_reclaimable_pages(struct zone *zone)
636 {
637         unsigned long nr;
638
639         nr = zone_page_state_snapshot(zone, NR_ZONE_INACTIVE_FILE) +
640                 zone_page_state_snapshot(zone, NR_ZONE_ACTIVE_FILE);
641         if (can_reclaim_anon_pages(NULL, zone_to_nid(zone), NULL))
642                 nr += zone_page_state_snapshot(zone, NR_ZONE_INACTIVE_ANON) +
643                         zone_page_state_snapshot(zone, NR_ZONE_ACTIVE_ANON);
644
645         return nr;
646 }
647
648 /**
649  * lruvec_lru_size -  Returns the number of pages on the given LRU list.
650  * @lruvec: lru vector
651  * @lru: lru to use
652  * @zone_idx: zones to consider (use MAX_NR_ZONES - 1 for the whole LRU list)
653  */
654 static unsigned long lruvec_lru_size(struct lruvec *lruvec, enum lru_list lru,
655                                      int zone_idx)
656 {
657         unsigned long size = 0;
658         int zid;
659
660         for (zid = 0; zid <= zone_idx; zid++) {
661                 struct zone *zone = &lruvec_pgdat(lruvec)->node_zones[zid];
662
663                 if (!managed_zone(zone))
664                         continue;
665
666                 if (!mem_cgroup_disabled())
667                         size += mem_cgroup_get_zone_lru_size(lruvec, lru, zid);
668                 else
669                         size += zone_page_state(zone, NR_ZONE_LRU_BASE + lru);
670         }
671         return size;
672 }
673
674 /*
675  * Add a shrinker callback to be called from the vm.
676  */
677 static int __prealloc_shrinker(struct shrinker *shrinker)
678 {
679         unsigned int size;
680         int err;
681
682         if (shrinker->flags & SHRINKER_MEMCG_AWARE) {
683                 err = prealloc_memcg_shrinker(shrinker);
684                 if (err != -ENOSYS)
685                         return err;
686
687                 shrinker->flags &= ~SHRINKER_MEMCG_AWARE;
688         }
689
690         size = sizeof(*shrinker->nr_deferred);
691         if (shrinker->flags & SHRINKER_NUMA_AWARE)
692                 size *= nr_node_ids;
693
694         shrinker->nr_deferred = kzalloc(size, GFP_KERNEL);
695         if (!shrinker->nr_deferred)
696                 return -ENOMEM;
697
698         return 0;
699 }
700
701 #ifdef CONFIG_SHRINKER_DEBUG
702 int prealloc_shrinker(struct shrinker *shrinker, const char *fmt, ...)
703 {
704         va_list ap;
705         int err;
706
707         va_start(ap, fmt);
708         shrinker->name = kvasprintf_const(GFP_KERNEL, fmt, ap);
709         va_end(ap);
710         if (!shrinker->name)
711                 return -ENOMEM;
712
713         err = __prealloc_shrinker(shrinker);
714         if (err) {
715                 kfree_const(shrinker->name);
716                 shrinker->name = NULL;
717         }
718
719         return err;
720 }
721 #else
722 int prealloc_shrinker(struct shrinker *shrinker, const char *fmt, ...)
723 {
724         return __prealloc_shrinker(shrinker);
725 }
726 #endif
727
728 void free_prealloced_shrinker(struct shrinker *shrinker)
729 {
730 #ifdef CONFIG_SHRINKER_DEBUG
731         kfree_const(shrinker->name);
732         shrinker->name = NULL;
733 #endif
734         if (shrinker->flags & SHRINKER_MEMCG_AWARE) {
735                 down_write(&shrinker_rwsem);
736                 unregister_memcg_shrinker(shrinker);
737                 up_write(&shrinker_rwsem);
738                 return;
739         }
740
741         kfree(shrinker->nr_deferred);
742         shrinker->nr_deferred = NULL;
743 }
744
745 void register_shrinker_prepared(struct shrinker *shrinker)
746 {
747         down_write(&shrinker_rwsem);
748         list_add_tail(&shrinker->list, &shrinker_list);
749         shrinker->flags |= SHRINKER_REGISTERED;
750         shrinker_debugfs_add(shrinker);
751         up_write(&shrinker_rwsem);
752 }
753
754 static int __register_shrinker(struct shrinker *shrinker)
755 {
756         int err = __prealloc_shrinker(shrinker);
757
758         if (err)
759                 return err;
760         register_shrinker_prepared(shrinker);
761         return 0;
762 }
763
764 #ifdef CONFIG_SHRINKER_DEBUG
765 int register_shrinker(struct shrinker *shrinker, const char *fmt, ...)
766 {
767         va_list ap;
768         int err;
769
770         va_start(ap, fmt);
771         shrinker->name = kvasprintf_const(GFP_KERNEL, fmt, ap);
772         va_end(ap);
773         if (!shrinker->name)
774                 return -ENOMEM;
775
776         err = __register_shrinker(shrinker);
777         if (err) {
778                 kfree_const(shrinker->name);
779                 shrinker->name = NULL;
780         }
781         return err;
782 }
783 #else
784 int register_shrinker(struct shrinker *shrinker, const char *fmt, ...)
785 {
786         return __register_shrinker(shrinker);
787 }
788 #endif
789 EXPORT_SYMBOL(register_shrinker);
790
791 /*
792  * Remove one
793  */
794 void unregister_shrinker(struct shrinker *shrinker)
795 {
796         struct dentry *debugfs_entry;
797         int debugfs_id;
798
799         if (!(shrinker->flags & SHRINKER_REGISTERED))
800                 return;
801
802         down_write(&shrinker_rwsem);
803         list_del(&shrinker->list);
804         shrinker->flags &= ~SHRINKER_REGISTERED;
805         if (shrinker->flags & SHRINKER_MEMCG_AWARE)
806                 unregister_memcg_shrinker(shrinker);
807         debugfs_entry = shrinker_debugfs_detach(shrinker, &debugfs_id);
808         up_write(&shrinker_rwsem);
809
810         shrinker_debugfs_remove(debugfs_entry, debugfs_id);
811
812         kfree(shrinker->nr_deferred);
813         shrinker->nr_deferred = NULL;
814 }
815 EXPORT_SYMBOL(unregister_shrinker);
816
817 /**
818  * synchronize_shrinkers - Wait for all running shrinkers to complete.
819  *
820  * This is equivalent to calling unregister_shrink() and register_shrinker(),
821  * but atomically and with less overhead. This is useful to guarantee that all
822  * shrinker invocations have seen an update, before freeing memory, similar to
823  * rcu.
824  */
825 void synchronize_shrinkers(void)
826 {
827         down_write(&shrinker_rwsem);
828         up_write(&shrinker_rwsem);
829 }
830 EXPORT_SYMBOL(synchronize_shrinkers);
831
832 #define SHRINK_BATCH 128
833
834 static unsigned long do_shrink_slab(struct shrink_control *shrinkctl,
835                                     struct shrinker *shrinker, int priority)
836 {
837         unsigned long freed = 0;
838         unsigned long long delta;
839         long total_scan;
840         long freeable;
841         long nr;
842         long new_nr;
843         long batch_size = shrinker->batch ? shrinker->batch
844                                           : SHRINK_BATCH;
845         long scanned = 0, next_deferred;
846
847         freeable = shrinker->count_objects(shrinker, shrinkctl);
848         if (freeable == 0 || freeable == SHRINK_EMPTY)
849                 return freeable;
850
851         /*
852          * copy the current shrinker scan count into a local variable
853          * and zero it so that other concurrent shrinker invocations
854          * don't also do this scanning work.
855          */
856         nr = xchg_nr_deferred(shrinker, shrinkctl);
857
858         if (shrinker->seeks) {
859                 delta = freeable >> priority;
860                 delta *= 4;
861                 do_div(delta, shrinker->seeks);
862         } else {
863                 /*
864                  * These objects don't require any IO to create. Trim
865                  * them aggressively under memory pressure to keep
866                  * them from causing refetches in the IO caches.
867                  */
868                 delta = freeable / 2;
869         }
870
871         total_scan = nr >> priority;
872         total_scan += delta;
873         total_scan = min(total_scan, (2 * freeable));
874
875         trace_mm_shrink_slab_start(shrinker, shrinkctl, nr,
876                                    freeable, delta, total_scan, priority);
877
878         /*
879          * Normally, we should not scan less than batch_size objects in one
880          * pass to avoid too frequent shrinker calls, but if the slab has less
881          * than batch_size objects in total and we are really tight on memory,
882          * we will try to reclaim all available objects, otherwise we can end
883          * up failing allocations although there are plenty of reclaimable
884          * objects spread over several slabs with usage less than the
885          * batch_size.
886          *
887          * We detect the "tight on memory" situations by looking at the total
888          * number of objects we want to scan (total_scan). If it is greater
889          * than the total number of objects on slab (freeable), we must be
890          * scanning at high prio and therefore should try to reclaim as much as
891          * possible.
892          */
893         while (total_scan >= batch_size ||
894                total_scan >= freeable) {
895                 unsigned long ret;
896                 unsigned long nr_to_scan = min(batch_size, total_scan);
897
898                 shrinkctl->nr_to_scan = nr_to_scan;
899                 shrinkctl->nr_scanned = nr_to_scan;
900                 ret = shrinker->scan_objects(shrinker, shrinkctl);
901                 if (ret == SHRINK_STOP)
902                         break;
903                 freed += ret;
904
905                 count_vm_events(SLABS_SCANNED, shrinkctl->nr_scanned);
906                 total_scan -= shrinkctl->nr_scanned;
907                 scanned += shrinkctl->nr_scanned;
908
909                 cond_resched();
910         }
911
912         /*
913          * The deferred work is increased by any new work (delta) that wasn't
914          * done, decreased by old deferred work that was done now.
915          *
916          * And it is capped to two times of the freeable items.
917          */
918         next_deferred = max_t(long, (nr + delta - scanned), 0);
919         next_deferred = min(next_deferred, (2 * freeable));
920
921         /*
922          * move the unused scan count back into the shrinker in a
923          * manner that handles concurrent updates.
924          */
925         new_nr = add_nr_deferred(next_deferred, shrinker, shrinkctl);
926
927         trace_mm_shrink_slab_end(shrinker, shrinkctl->nid, freed, nr, new_nr, total_scan);
928         return freed;
929 }
930
931 #ifdef CONFIG_MEMCG
932 static unsigned long shrink_slab_memcg(gfp_t gfp_mask, int nid,
933                         struct mem_cgroup *memcg, int priority)
934 {
935         struct shrinker_info *info;
936         unsigned long ret, freed = 0;
937         int i;
938
939         if (!mem_cgroup_online(memcg))
940                 return 0;
941
942         if (!down_read_trylock(&shrinker_rwsem))
943                 return 0;
944
945         info = shrinker_info_protected(memcg, nid);
946         if (unlikely(!info))
947                 goto unlock;
948
949         for_each_set_bit(i, info->map, info->map_nr_max) {
950                 struct shrink_control sc = {
951                         .gfp_mask = gfp_mask,
952                         .nid = nid,
953                         .memcg = memcg,
954                 };
955                 struct shrinker *shrinker;
956
957                 shrinker = idr_find(&shrinker_idr, i);
958                 if (unlikely(!shrinker || !(shrinker->flags & SHRINKER_REGISTERED))) {
959                         if (!shrinker)
960                                 clear_bit(i, info->map);
961                         continue;
962                 }
963
964                 /* Call non-slab shrinkers even though kmem is disabled */
965                 if (!memcg_kmem_online() &&
966                     !(shrinker->flags & SHRINKER_NONSLAB))
967                         continue;
968
969                 ret = do_shrink_slab(&sc, shrinker, priority);
970                 if (ret == SHRINK_EMPTY) {
971                         clear_bit(i, info->map);
972                         /*
973                          * After the shrinker reported that it had no objects to
974                          * free, but before we cleared the corresponding bit in
975                          * the memcg shrinker map, a new object might have been
976                          * added. To make sure, we have the bit set in this
977                          * case, we invoke the shrinker one more time and reset
978                          * the bit if it reports that it is not empty anymore.
979                          * The memory barrier here pairs with the barrier in
980                          * set_shrinker_bit():
981                          *
982                          * list_lru_add()     shrink_slab_memcg()
983                          *   list_add_tail()    clear_bit()
984                          *   <MB>               <MB>
985                          *   set_bit()          do_shrink_slab()
986                          */
987                         smp_mb__after_atomic();
988                         ret = do_shrink_slab(&sc, shrinker, priority);
989                         if (ret == SHRINK_EMPTY)
990                                 ret = 0;
991                         else
992                                 set_shrinker_bit(memcg, nid, i);
993                 }
994                 freed += ret;
995
996                 if (rwsem_is_contended(&shrinker_rwsem)) {
997                         freed = freed ? : 1;
998                         break;
999                 }
1000         }
1001 unlock:
1002         up_read(&shrinker_rwsem);
1003         return freed;
1004 }
1005 #else /* CONFIG_MEMCG */
1006 static unsigned long shrink_slab_memcg(gfp_t gfp_mask, int nid,
1007                         struct mem_cgroup *memcg, int priority)
1008 {
1009         return 0;
1010 }
1011 #endif /* CONFIG_MEMCG */
1012
1013 /**
1014  * shrink_slab - shrink slab caches
1015  * @gfp_mask: allocation context
1016  * @nid: node whose slab caches to target
1017  * @memcg: memory cgroup whose slab caches to target
1018  * @priority: the reclaim priority
1019  *
1020  * Call the shrink functions to age shrinkable caches.
1021  *
1022  * @nid is passed along to shrinkers with SHRINKER_NUMA_AWARE set,
1023  * unaware shrinkers will receive a node id of 0 instead.
1024  *
1025  * @memcg specifies the memory cgroup to target. Unaware shrinkers
1026  * are called only if it is the root cgroup.
1027  *
1028  * @priority is sc->priority, we take the number of objects and >> by priority
1029  * in order to get the scan target.
1030  *
1031  * Returns the number of reclaimed slab objects.
1032  */
1033 static unsigned long shrink_slab(gfp_t gfp_mask, int nid,
1034                                  struct mem_cgroup *memcg,
1035                                  int priority)
1036 {
1037         unsigned long ret, freed = 0;
1038         struct shrinker *shrinker;
1039
1040         /*
1041          * The root memcg might be allocated even though memcg is disabled
1042          * via "cgroup_disable=memory" boot parameter.  This could make
1043          * mem_cgroup_is_root() return false, then just run memcg slab
1044          * shrink, but skip global shrink.  This may result in premature
1045          * oom.
1046          */
1047         if (!mem_cgroup_disabled() && !mem_cgroup_is_root(memcg))
1048                 return shrink_slab_memcg(gfp_mask, nid, memcg, priority);
1049
1050         if (!down_read_trylock(&shrinker_rwsem))
1051                 goto out;
1052
1053         list_for_each_entry(shrinker, &shrinker_list, list) {
1054                 struct shrink_control sc = {
1055                         .gfp_mask = gfp_mask,
1056                         .nid = nid,
1057                         .memcg = memcg,
1058                 };
1059
1060                 ret = do_shrink_slab(&sc, shrinker, priority);
1061                 if (ret == SHRINK_EMPTY)
1062                         ret = 0;
1063                 freed += ret;
1064                 /*
1065                  * Bail out if someone want to register a new shrinker to
1066                  * prevent the registration from being stalled for long periods
1067                  * by parallel ongoing shrinking.
1068                  */
1069                 if (rwsem_is_contended(&shrinker_rwsem)) {
1070                         freed = freed ? : 1;
1071                         break;
1072                 }
1073         }
1074
1075         up_read(&shrinker_rwsem);
1076 out:
1077         cond_resched();
1078         return freed;
1079 }
1080
1081 static unsigned long drop_slab_node(int nid)
1082 {
1083         unsigned long freed = 0;
1084         struct mem_cgroup *memcg = NULL;
1085
1086         memcg = mem_cgroup_iter(NULL, NULL, NULL);
1087         do {
1088                 freed += shrink_slab(GFP_KERNEL, nid, memcg, 0);
1089         } while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)) != NULL);
1090
1091         return freed;
1092 }
1093
1094 void drop_slab(void)
1095 {
1096         int nid;
1097         int shift = 0;
1098         unsigned long freed;
1099
1100         do {
1101                 freed = 0;
1102                 for_each_online_node(nid) {
1103                         if (fatal_signal_pending(current))
1104                                 return;
1105
1106                         freed += drop_slab_node(nid);
1107                 }
1108         } while ((freed >> shift++) > 1);
1109 }
1110
1111 static int reclaimer_offset(void)
1112 {
1113         BUILD_BUG_ON(PGSTEAL_DIRECT - PGSTEAL_KSWAPD !=
1114                         PGDEMOTE_DIRECT - PGDEMOTE_KSWAPD);
1115         BUILD_BUG_ON(PGSTEAL_DIRECT - PGSTEAL_KSWAPD !=
1116                         PGSCAN_DIRECT - PGSCAN_KSWAPD);
1117         BUILD_BUG_ON(PGSTEAL_KHUGEPAGED - PGSTEAL_KSWAPD !=
1118                         PGDEMOTE_KHUGEPAGED - PGDEMOTE_KSWAPD);
1119         BUILD_BUG_ON(PGSTEAL_KHUGEPAGED - PGSTEAL_KSWAPD !=
1120                         PGSCAN_KHUGEPAGED - PGSCAN_KSWAPD);
1121
1122         if (current_is_kswapd())
1123                 return 0;
1124         if (current_is_khugepaged())
1125                 return PGSTEAL_KHUGEPAGED - PGSTEAL_KSWAPD;
1126         return PGSTEAL_DIRECT - PGSTEAL_KSWAPD;
1127 }
1128
1129 static inline int is_page_cache_freeable(struct folio *folio)
1130 {
1131         /*
1132          * A freeable page cache folio is referenced only by the caller
1133          * that isolated the folio, the page cache and optional filesystem
1134          * private data at folio->private.
1135          */
1136         return folio_ref_count(folio) - folio_test_private(folio) ==
1137                 1 + folio_nr_pages(folio);
1138 }
1139
1140 /*
1141  * We detected a synchronous write error writing a folio out.  Probably
1142  * -ENOSPC.  We need to propagate that into the address_space for a subsequent
1143  * fsync(), msync() or close().
1144  *
1145  * The tricky part is that after writepage we cannot touch the mapping: nothing
1146  * prevents it from being freed up.  But we have a ref on the folio and once
1147  * that folio is locked, the mapping is pinned.
1148  *
1149  * We're allowed to run sleeping folio_lock() here because we know the caller has
1150  * __GFP_FS.
1151  */
1152 static void handle_write_error(struct address_space *mapping,
1153                                 struct folio *folio, int error)
1154 {
1155         folio_lock(folio);
1156         if (folio_mapping(folio) == mapping)
1157                 mapping_set_error(mapping, error);
1158         folio_unlock(folio);
1159 }
1160
1161 static bool skip_throttle_noprogress(pg_data_t *pgdat)
1162 {
1163         int reclaimable = 0, write_pending = 0;
1164         int i;
1165
1166         /*
1167          * If kswapd is disabled, reschedule if necessary but do not
1168          * throttle as the system is likely near OOM.
1169          */
1170         if (pgdat->kswapd_failures >= MAX_RECLAIM_RETRIES)
1171                 return true;
1172
1173         /*
1174          * If there are a lot of dirty/writeback folios then do not
1175          * throttle as throttling will occur when the folios cycle
1176          * towards the end of the LRU if still under writeback.
1177          */
1178         for (i = 0; i < MAX_NR_ZONES; i++) {
1179                 struct zone *zone = pgdat->node_zones + i;
1180
1181                 if (!managed_zone(zone))
1182                         continue;
1183
1184                 reclaimable += zone_reclaimable_pages(zone);
1185                 write_pending += zone_page_state_snapshot(zone,
1186                                                   NR_ZONE_WRITE_PENDING);
1187         }
1188         if (2 * write_pending <= reclaimable)
1189                 return true;
1190
1191         return false;
1192 }
1193
1194 void reclaim_throttle(pg_data_t *pgdat, enum vmscan_throttle_state reason)
1195 {
1196         wait_queue_head_t *wqh = &pgdat->reclaim_wait[reason];
1197         long timeout, ret;
1198         DEFINE_WAIT(wait);
1199
1200         /*
1201          * Do not throttle user workers, kthreads other than kswapd or
1202          * workqueues. They may be required for reclaim to make
1203          * forward progress (e.g. journalling workqueues or kthreads).
1204          */
1205         if (!current_is_kswapd() &&
1206             current->flags & (PF_USER_WORKER|PF_KTHREAD)) {
1207                 cond_resched();
1208                 return;
1209         }
1210
1211         /*
1212          * These figures are pulled out of thin air.
1213          * VMSCAN_THROTTLE_ISOLATED is a transient condition based on too many
1214          * parallel reclaimers which is a short-lived event so the timeout is
1215          * short. Failing to make progress or waiting on writeback are
1216          * potentially long-lived events so use a longer timeout. This is shaky
1217          * logic as a failure to make progress could be due to anything from
1218          * writeback to a slow device to excessive referenced folios at the tail
1219          * of the inactive LRU.
1220          */
1221         switch(reason) {
1222         case VMSCAN_THROTTLE_WRITEBACK:
1223                 timeout = HZ/10;
1224
1225                 if (atomic_inc_return(&pgdat->nr_writeback_throttled) == 1) {
1226                         WRITE_ONCE(pgdat->nr_reclaim_start,
1227                                 node_page_state(pgdat, NR_THROTTLED_WRITTEN));
1228                 }
1229
1230                 break;
1231         case VMSCAN_THROTTLE_CONGESTED:
1232                 fallthrough;
1233         case VMSCAN_THROTTLE_NOPROGRESS:
1234                 if (skip_throttle_noprogress(pgdat)) {
1235                         cond_resched();
1236                         return;
1237                 }
1238
1239                 timeout = 1;
1240
1241                 break;
1242         case VMSCAN_THROTTLE_ISOLATED:
1243                 timeout = HZ/50;
1244                 break;
1245         default:
1246                 WARN_ON_ONCE(1);
1247                 timeout = HZ;
1248                 break;
1249         }
1250
1251         prepare_to_wait(wqh, &wait, TASK_UNINTERRUPTIBLE);
1252         ret = schedule_timeout(timeout);
1253         finish_wait(wqh, &wait);
1254
1255         if (reason == VMSCAN_THROTTLE_WRITEBACK)
1256                 atomic_dec(&pgdat->nr_writeback_throttled);
1257
1258         trace_mm_vmscan_throttled(pgdat->node_id, jiffies_to_usecs(timeout),
1259                                 jiffies_to_usecs(timeout - ret),
1260                                 reason);
1261 }
1262
1263 /*
1264  * Account for folios written if tasks are throttled waiting on dirty
1265  * folios to clean. If enough folios have been cleaned since throttling
1266  * started then wakeup the throttled tasks.
1267  */
1268 void __acct_reclaim_writeback(pg_data_t *pgdat, struct folio *folio,
1269                                                         int nr_throttled)
1270 {
1271         unsigned long nr_written;
1272
1273         node_stat_add_folio(folio, NR_THROTTLED_WRITTEN);
1274
1275         /*
1276          * This is an inaccurate read as the per-cpu deltas may not
1277          * be synchronised. However, given that the system is
1278          * writeback throttled, it is not worth taking the penalty
1279          * of getting an accurate count. At worst, the throttle
1280          * timeout guarantees forward progress.
1281          */
1282         nr_written = node_page_state(pgdat, NR_THROTTLED_WRITTEN) -
1283                 READ_ONCE(pgdat->nr_reclaim_start);
1284
1285         if (nr_written > SWAP_CLUSTER_MAX * nr_throttled)
1286                 wake_up(&pgdat->reclaim_wait[VMSCAN_THROTTLE_WRITEBACK]);
1287 }
1288
1289 /* possible outcome of pageout() */
1290 typedef enum {
1291         /* failed to write folio out, folio is locked */
1292         PAGE_KEEP,
1293         /* move folio to the active list, folio is locked */
1294         PAGE_ACTIVATE,
1295         /* folio has been sent to the disk successfully, folio is unlocked */
1296         PAGE_SUCCESS,
1297         /* folio is clean and locked */
1298         PAGE_CLEAN,
1299 } pageout_t;
1300
1301 /*
1302  * pageout is called by shrink_folio_list() for each dirty folio.
1303  * Calls ->writepage().
1304  */
1305 static pageout_t pageout(struct folio *folio, struct address_space *mapping,
1306                          struct swap_iocb **plug)
1307 {
1308         /*
1309          * If the folio is dirty, only perform writeback if that write
1310          * will be non-blocking.  To prevent this allocation from being
1311          * stalled by pagecache activity.  But note that there may be
1312          * stalls if we need to run get_block().  We could test
1313          * PagePrivate for that.
1314          *
1315          * If this process is currently in __generic_file_write_iter() against
1316          * this folio's queue, we can perform writeback even if that
1317          * will block.
1318          *
1319          * If the folio is swapcache, write it back even if that would
1320          * block, for some throttling. This happens by accident, because
1321          * swap_backing_dev_info is bust: it doesn't reflect the
1322          * congestion state of the swapdevs.  Easy to fix, if needed.
1323          */
1324         if (!is_page_cache_freeable(folio))
1325                 return PAGE_KEEP;
1326         if (!mapping) {
1327                 /*
1328                  * Some data journaling orphaned folios can have
1329                  * folio->mapping == NULL while being dirty with clean buffers.
1330                  */
1331                 if (folio_test_private(folio)) {
1332                         if (try_to_free_buffers(folio)) {
1333                                 folio_clear_dirty(folio);
1334                                 pr_info("%s: orphaned folio\n", __func__);
1335                                 return PAGE_CLEAN;
1336                         }
1337                 }
1338                 return PAGE_KEEP;
1339         }
1340         if (mapping->a_ops->writepage == NULL)
1341                 return PAGE_ACTIVATE;
1342
1343         if (folio_clear_dirty_for_io(folio)) {
1344                 int res;
1345                 struct writeback_control wbc = {
1346                         .sync_mode = WB_SYNC_NONE,
1347                         .nr_to_write = SWAP_CLUSTER_MAX,
1348                         .range_start = 0,
1349                         .range_end = LLONG_MAX,
1350                         .for_reclaim = 1,
1351                         .swap_plug = plug,
1352                 };
1353
1354                 folio_set_reclaim(folio);
1355                 res = mapping->a_ops->writepage(&folio->page, &wbc);
1356                 if (res < 0)
1357                         handle_write_error(mapping, folio, res);
1358                 if (res == AOP_WRITEPAGE_ACTIVATE) {
1359                         folio_clear_reclaim(folio);
1360                         return PAGE_ACTIVATE;
1361                 }
1362
1363                 if (!folio_test_writeback(folio)) {
1364                         /* synchronous write or broken a_ops? */
1365                         folio_clear_reclaim(folio);
1366                 }
1367                 trace_mm_vmscan_write_folio(folio);
1368                 node_stat_add_folio(folio, NR_VMSCAN_WRITE);
1369                 return PAGE_SUCCESS;
1370         }
1371
1372         return PAGE_CLEAN;
1373 }
1374
1375 /*
1376  * Same as remove_mapping, but if the folio is removed from the mapping, it
1377  * gets returned with a refcount of 0.
1378  */
1379 static int __remove_mapping(struct address_space *mapping, struct folio *folio,
1380                             bool reclaimed, struct mem_cgroup *target_memcg)
1381 {
1382         int refcount;
1383         void *shadow = NULL;
1384
1385         BUG_ON(!folio_test_locked(folio));
1386         BUG_ON(mapping != folio_mapping(folio));
1387
1388         if (!folio_test_swapcache(folio))
1389                 spin_lock(&mapping->host->i_lock);
1390         xa_lock_irq(&mapping->i_pages);
1391         /*
1392          * The non racy check for a busy folio.
1393          *
1394          * Must be careful with the order of the tests. When someone has
1395          * a ref to the folio, it may be possible that they dirty it then
1396          * drop the reference. So if the dirty flag is tested before the
1397          * refcount here, then the following race may occur:
1398          *
1399          * get_user_pages(&page);
1400          * [user mapping goes away]
1401          * write_to(page);
1402          *                              !folio_test_dirty(folio)    [good]
1403          * folio_set_dirty(folio);
1404          * folio_put(folio);
1405          *                              !refcount(folio)   [good, discard it]
1406          *
1407          * [oops, our write_to data is lost]
1408          *
1409          * Reversing the order of the tests ensures such a situation cannot
1410          * escape unnoticed. The smp_rmb is needed to ensure the folio->flags
1411          * load is not satisfied before that of folio->_refcount.
1412          *
1413          * Note that if the dirty flag is always set via folio_mark_dirty,
1414          * and thus under the i_pages lock, then this ordering is not required.
1415          */
1416         refcount = 1 + folio_nr_pages(folio);
1417         if (!folio_ref_freeze(folio, refcount))
1418                 goto cannot_free;
1419         /* note: atomic_cmpxchg in folio_ref_freeze provides the smp_rmb */
1420         if (unlikely(folio_test_dirty(folio))) {
1421                 folio_ref_unfreeze(folio, refcount);
1422                 goto cannot_free;
1423         }
1424
1425         if (folio_test_swapcache(folio)) {
1426                 swp_entry_t swap = folio->swap;
1427
1428                 if (reclaimed && !mapping_exiting(mapping))
1429                         shadow = workingset_eviction(folio, target_memcg);
1430                 __delete_from_swap_cache(folio, swap, shadow);
1431                 mem_cgroup_swapout(folio, swap);
1432                 xa_unlock_irq(&mapping->i_pages);
1433                 put_swap_folio(folio, swap);
1434         } else {
1435                 void (*free_folio)(struct folio *);
1436
1437                 free_folio = mapping->a_ops->free_folio;
1438                 /*
1439                  * Remember a shadow entry for reclaimed file cache in
1440                  * order to detect refaults, thus thrashing, later on.
1441                  *
1442                  * But don't store shadows in an address space that is
1443                  * already exiting.  This is not just an optimization,
1444                  * inode reclaim needs to empty out the radix tree or
1445                  * the nodes are lost.  Don't plant shadows behind its
1446                  * back.
1447                  *
1448                  * We also don't store shadows for DAX mappings because the
1449                  * only page cache folios found in these are zero pages
1450                  * covering holes, and because we don't want to mix DAX
1451                  * exceptional entries and shadow exceptional entries in the
1452                  * same address_space.
1453                  */
1454                 if (reclaimed && folio_is_file_lru(folio) &&
1455                     !mapping_exiting(mapping) && !dax_mapping(mapping))
1456                         shadow = workingset_eviction(folio, target_memcg);
1457                 __filemap_remove_folio(folio, shadow);
1458                 xa_unlock_irq(&mapping->i_pages);
1459                 if (mapping_shrinkable(mapping))
1460                         inode_add_lru(mapping->host);
1461                 spin_unlock(&mapping->host->i_lock);
1462
1463                 if (free_folio)
1464                         free_folio(folio);
1465         }
1466
1467         return 1;
1468
1469 cannot_free:
1470         xa_unlock_irq(&mapping->i_pages);
1471         if (!folio_test_swapcache(folio))
1472                 spin_unlock(&mapping->host->i_lock);
1473         return 0;
1474 }
1475
1476 /**
1477  * remove_mapping() - Attempt to remove a folio from its mapping.
1478  * @mapping: The address space.
1479  * @folio: The folio to remove.
1480  *
1481  * If the folio is dirty, under writeback or if someone else has a ref
1482  * on it, removal will fail.
1483  * Return: The number of pages removed from the mapping.  0 if the folio
1484  * could not be removed.
1485  * Context: The caller should have a single refcount on the folio and
1486  * hold its lock.
1487  */
1488 long remove_mapping(struct address_space *mapping, struct folio *folio)
1489 {
1490         if (__remove_mapping(mapping, folio, false, NULL)) {
1491                 /*
1492                  * Unfreezing the refcount with 1 effectively
1493                  * drops the pagecache ref for us without requiring another
1494                  * atomic operation.
1495                  */
1496                 folio_ref_unfreeze(folio, 1);
1497                 return folio_nr_pages(folio);
1498         }
1499         return 0;
1500 }
1501
1502 /**
1503  * folio_putback_lru - Put previously isolated folio onto appropriate LRU list.
1504  * @folio: Folio to be returned to an LRU list.
1505  *
1506  * Add previously isolated @folio to appropriate LRU list.
1507  * The folio may still be unevictable for other reasons.
1508  *
1509  * Context: lru_lock must not be held, interrupts must be enabled.
1510  */
1511 void folio_putback_lru(struct folio *folio)
1512 {
1513         folio_add_lru(folio);
1514         folio_put(folio);               /* drop ref from isolate */
1515 }
1516
1517 enum folio_references {
1518         FOLIOREF_RECLAIM,
1519         FOLIOREF_RECLAIM_CLEAN,
1520         FOLIOREF_KEEP,
1521         FOLIOREF_ACTIVATE,
1522 };
1523
1524 static enum folio_references folio_check_references(struct folio *folio,
1525                                                   struct scan_control *sc)
1526 {
1527         int referenced_ptes, referenced_folio;
1528         unsigned long vm_flags;
1529
1530         referenced_ptes = folio_referenced(folio, 1, sc->target_mem_cgroup,
1531                                            &vm_flags);
1532         referenced_folio = folio_test_clear_referenced(folio);
1533
1534         /*
1535          * The supposedly reclaimable folio was found to be in a VM_LOCKED vma.
1536          * Let the folio, now marked Mlocked, be moved to the unevictable list.
1537          */
1538         if (vm_flags & VM_LOCKED)
1539                 return FOLIOREF_ACTIVATE;
1540
1541         /* rmap lock contention: rotate */
1542         if (referenced_ptes == -1)
1543                 return FOLIOREF_KEEP;
1544
1545         if (referenced_ptes) {
1546                 /*
1547                  * All mapped folios start out with page table
1548                  * references from the instantiating fault, so we need
1549                  * to look twice if a mapped file/anon folio is used more
1550                  * than once.
1551                  *
1552                  * Mark it and spare it for another trip around the
1553                  * inactive list.  Another page table reference will
1554                  * lead to its activation.
1555                  *
1556                  * Note: the mark is set for activated folios as well
1557                  * so that recently deactivated but used folios are
1558                  * quickly recovered.
1559                  */
1560                 folio_set_referenced(folio);
1561
1562                 if (referenced_folio || referenced_ptes > 1)
1563                         return FOLIOREF_ACTIVATE;
1564
1565                 /*
1566                  * Activate file-backed executable folios after first usage.
1567                  */
1568                 if ((vm_flags & VM_EXEC) && folio_is_file_lru(folio))
1569                         return FOLIOREF_ACTIVATE;
1570
1571                 return FOLIOREF_KEEP;
1572         }
1573
1574         /* Reclaim if clean, defer dirty folios to writeback */
1575         if (referenced_folio && folio_is_file_lru(folio))
1576                 return FOLIOREF_RECLAIM_CLEAN;
1577
1578         return FOLIOREF_RECLAIM;
1579 }
1580
1581 /* Check if a folio is dirty or under writeback */
1582 static void folio_check_dirty_writeback(struct folio *folio,
1583                                        bool *dirty, bool *writeback)
1584 {
1585         struct address_space *mapping;
1586
1587         /*
1588          * Anonymous folios are not handled by flushers and must be written
1589          * from reclaim context. Do not stall reclaim based on them.
1590          * MADV_FREE anonymous folios are put into inactive file list too.
1591          * They could be mistakenly treated as file lru. So further anon
1592          * test is needed.
1593          */
1594         if (!folio_is_file_lru(folio) ||
1595             (folio_test_anon(folio) && !folio_test_swapbacked(folio))) {
1596                 *dirty = false;
1597                 *writeback = false;
1598                 return;
1599         }
1600
1601         /* By default assume that the folio flags are accurate */
1602         *dirty = folio_test_dirty(folio);
1603         *writeback = folio_test_writeback(folio);
1604
1605         /* Verify dirty/writeback state if the filesystem supports it */
1606         if (!folio_test_private(folio))
1607                 return;
1608
1609         mapping = folio_mapping(folio);
1610         if (mapping && mapping->a_ops->is_dirty_writeback)
1611                 mapping->a_ops->is_dirty_writeback(folio, dirty, writeback);
1612 }
1613
1614 static struct folio *alloc_demote_folio(struct folio *src,
1615                 unsigned long private)
1616 {
1617         struct folio *dst;
1618         nodemask_t *allowed_mask;
1619         struct migration_target_control *mtc;
1620
1621         mtc = (struct migration_target_control *)private;
1622
1623         allowed_mask = mtc->nmask;
1624         /*
1625          * make sure we allocate from the target node first also trying to
1626          * demote or reclaim pages from the target node via kswapd if we are
1627          * low on free memory on target node. If we don't do this and if
1628          * we have free memory on the slower(lower) memtier, we would start
1629          * allocating pages from slower(lower) memory tiers without even forcing
1630          * a demotion of cold pages from the target memtier. This can result
1631          * in the kernel placing hot pages in slower(lower) memory tiers.
1632          */
1633         mtc->nmask = NULL;
1634         mtc->gfp_mask |= __GFP_THISNODE;
1635         dst = alloc_migration_target(src, (unsigned long)mtc);
1636         if (dst)
1637                 return dst;
1638
1639         mtc->gfp_mask &= ~__GFP_THISNODE;
1640         mtc->nmask = allowed_mask;
1641
1642         return alloc_migration_target(src, (unsigned long)mtc);
1643 }
1644
1645 /*
1646  * Take folios on @demote_folios and attempt to demote them to another node.
1647  * Folios which are not demoted are left on @demote_folios.
1648  */
1649 static unsigned int demote_folio_list(struct list_head *demote_folios,
1650                                      struct pglist_data *pgdat)
1651 {
1652         int target_nid = next_demotion_node(pgdat->node_id);
1653         unsigned int nr_succeeded;
1654         nodemask_t allowed_mask;
1655
1656         struct migration_target_control mtc = {
1657                 /*
1658                  * Allocate from 'node', or fail quickly and quietly.
1659                  * When this happens, 'page' will likely just be discarded
1660                  * instead of migrated.
1661                  */
1662                 .gfp_mask = (GFP_HIGHUSER_MOVABLE & ~__GFP_RECLAIM) | __GFP_NOWARN |
1663                         __GFP_NOMEMALLOC | GFP_NOWAIT,
1664                 .nid = target_nid,
1665                 .nmask = &allowed_mask
1666         };
1667
1668         if (list_empty(demote_folios))
1669                 return 0;
1670
1671         if (target_nid == NUMA_NO_NODE)
1672                 return 0;
1673
1674         node_get_allowed_targets(pgdat, &allowed_mask);
1675
1676         /* Demotion ignores all cpuset and mempolicy settings */
1677         migrate_pages(demote_folios, alloc_demote_folio, NULL,
1678                       (unsigned long)&mtc, MIGRATE_ASYNC, MR_DEMOTION,
1679                       &nr_succeeded);
1680
1681         __count_vm_events(PGDEMOTE_KSWAPD + reclaimer_offset(), nr_succeeded);
1682
1683         return nr_succeeded;
1684 }
1685
1686 static bool may_enter_fs(struct folio *folio, gfp_t gfp_mask)
1687 {
1688         if (gfp_mask & __GFP_FS)
1689                 return true;
1690         if (!folio_test_swapcache(folio) || !(gfp_mask & __GFP_IO))
1691                 return false;
1692         /*
1693          * We can "enter_fs" for swap-cache with only __GFP_IO
1694          * providing this isn't SWP_FS_OPS.
1695          * ->flags can be updated non-atomicially (scan_swap_map_slots),
1696          * but that will never affect SWP_FS_OPS, so the data_race
1697          * is safe.
1698          */
1699         return !data_race(folio_swap_flags(folio) & SWP_FS_OPS);
1700 }
1701
1702 /*
1703  * shrink_folio_list() returns the number of reclaimed pages
1704  */
1705 static unsigned int shrink_folio_list(struct list_head *folio_list,
1706                 struct pglist_data *pgdat, struct scan_control *sc,
1707                 struct reclaim_stat *stat, bool ignore_references)
1708 {
1709         LIST_HEAD(ret_folios);
1710         LIST_HEAD(free_folios);
1711         LIST_HEAD(demote_folios);
1712         unsigned int nr_reclaimed = 0;
1713         unsigned int pgactivate = 0;
1714         bool do_demote_pass;
1715         struct swap_iocb *plug = NULL;
1716
1717         memset(stat, 0, sizeof(*stat));
1718         cond_resched();
1719         do_demote_pass = can_demote(pgdat->node_id, sc);
1720
1721 retry:
1722         while (!list_empty(folio_list)) {
1723                 struct address_space *mapping;
1724                 struct folio *folio;
1725                 enum folio_references references = FOLIOREF_RECLAIM;
1726                 bool dirty, writeback;
1727                 unsigned int nr_pages;
1728
1729                 cond_resched();
1730
1731                 folio = lru_to_folio(folio_list);
1732                 list_del(&folio->lru);
1733
1734                 if (!folio_trylock(folio))
1735                         goto keep;
1736
1737                 VM_BUG_ON_FOLIO(folio_test_active(folio), folio);
1738
1739                 nr_pages = folio_nr_pages(folio);
1740
1741                 /* Account the number of base pages */
1742                 sc->nr_scanned += nr_pages;
1743
1744                 if (unlikely(!folio_evictable(folio)))
1745                         goto activate_locked;
1746
1747                 if (!sc->may_unmap && folio_mapped(folio))
1748                         goto keep_locked;
1749
1750                 /* folio_update_gen() tried to promote this page? */
1751                 if (lru_gen_enabled() && !ignore_references &&
1752                     folio_mapped(folio) && folio_test_referenced(folio))
1753                         goto keep_locked;
1754
1755                 /*
1756                  * The number of dirty pages determines if a node is marked
1757                  * reclaim_congested. kswapd will stall and start writing
1758                  * folios if the tail of the LRU is all dirty unqueued folios.
1759                  */
1760                 folio_check_dirty_writeback(folio, &dirty, &writeback);
1761                 if (dirty || writeback)
1762                         stat->nr_dirty += nr_pages;
1763
1764                 if (dirty && !writeback)
1765                         stat->nr_unqueued_dirty += nr_pages;
1766
1767                 /*
1768                  * Treat this folio as congested if folios are cycling
1769                  * through the LRU so quickly that the folios marked
1770                  * for immediate reclaim are making it to the end of
1771                  * the LRU a second time.
1772                  */
1773                 if (writeback && folio_test_reclaim(folio))
1774                         stat->nr_congested += nr_pages;
1775
1776                 /*
1777                  * If a folio at the tail of the LRU is under writeback, there
1778                  * are three cases to consider.
1779                  *
1780                  * 1) If reclaim is encountering an excessive number
1781                  *    of folios under writeback and this folio has both
1782                  *    the writeback and reclaim flags set, then it
1783                  *    indicates that folios are being queued for I/O but
1784                  *    are being recycled through the LRU before the I/O
1785                  *    can complete. Waiting on the folio itself risks an
1786                  *    indefinite stall if it is impossible to writeback
1787                  *    the folio due to I/O error or disconnected storage
1788                  *    so instead note that the LRU is being scanned too
1789                  *    quickly and the caller can stall after the folio
1790                  *    list has been processed.
1791                  *
1792                  * 2) Global or new memcg reclaim encounters a folio that is
1793                  *    not marked for immediate reclaim, or the caller does not
1794                  *    have __GFP_FS (or __GFP_IO if it's simply going to swap,
1795                  *    not to fs). In this case mark the folio for immediate
1796                  *    reclaim and continue scanning.
1797                  *
1798                  *    Require may_enter_fs() because we would wait on fs, which
1799                  *    may not have submitted I/O yet. And the loop driver might
1800                  *    enter reclaim, and deadlock if it waits on a folio for
1801                  *    which it is needed to do the write (loop masks off
1802                  *    __GFP_IO|__GFP_FS for this reason); but more thought
1803                  *    would probably show more reasons.
1804                  *
1805                  * 3) Legacy memcg encounters a folio that already has the
1806                  *    reclaim flag set. memcg does not have any dirty folio
1807                  *    throttling so we could easily OOM just because too many
1808                  *    folios are in writeback and there is nothing else to
1809                  *    reclaim. Wait for the writeback to complete.
1810                  *
1811                  * In cases 1) and 2) we activate the folios to get them out of
1812                  * the way while we continue scanning for clean folios on the
1813                  * inactive list and refilling from the active list. The
1814                  * observation here is that waiting for disk writes is more
1815                  * expensive than potentially causing reloads down the line.
1816                  * Since they're marked for immediate reclaim, they won't put
1817                  * memory pressure on the cache working set any longer than it
1818                  * takes to write them to disk.
1819                  */
1820                 if (folio_test_writeback(folio)) {
1821                         /* Case 1 above */
1822                         if (current_is_kswapd() &&
1823                             folio_test_reclaim(folio) &&
1824                             test_bit(PGDAT_WRITEBACK, &pgdat->flags)) {
1825                                 stat->nr_immediate += nr_pages;
1826                                 goto activate_locked;
1827
1828                         /* Case 2 above */
1829                         } else if (writeback_throttling_sane(sc) ||
1830                             !folio_test_reclaim(folio) ||
1831                             !may_enter_fs(folio, sc->gfp_mask)) {
1832                                 /*
1833                                  * This is slightly racy -
1834                                  * folio_end_writeback() might have
1835                                  * just cleared the reclaim flag, then
1836                                  * setting the reclaim flag here ends up
1837                                  * interpreted as the readahead flag - but
1838                                  * that does not matter enough to care.
1839                                  * What we do want is for this folio to
1840                                  * have the reclaim flag set next time
1841                                  * memcg reclaim reaches the tests above,
1842                                  * so it will then wait for writeback to
1843                                  * avoid OOM; and it's also appropriate
1844                                  * in global reclaim.
1845                                  */
1846                                 folio_set_reclaim(folio);
1847                                 stat->nr_writeback += nr_pages;
1848                                 goto activate_locked;
1849
1850                         /* Case 3 above */
1851                         } else {
1852                                 folio_unlock(folio);
1853                                 folio_wait_writeback(folio);
1854                                 /* then go back and try same folio again */
1855                                 list_add_tail(&folio->lru, folio_list);
1856                                 continue;
1857                         }
1858                 }
1859
1860                 if (!ignore_references)
1861                         references = folio_check_references(folio, sc);
1862
1863                 switch (references) {
1864                 case FOLIOREF_ACTIVATE:
1865                         goto activate_locked;
1866                 case FOLIOREF_KEEP:
1867                         stat->nr_ref_keep += nr_pages;
1868                         goto keep_locked;
1869                 case FOLIOREF_RECLAIM:
1870                 case FOLIOREF_RECLAIM_CLEAN:
1871                         ; /* try to reclaim the folio below */
1872                 }
1873
1874                 /*
1875                  * Before reclaiming the folio, try to relocate
1876                  * its contents to another node.
1877                  */
1878                 if (do_demote_pass &&
1879                     (thp_migration_supported() || !folio_test_large(folio))) {
1880                         list_add(&folio->lru, &demote_folios);
1881                         folio_unlock(folio);
1882                         continue;
1883                 }
1884
1885                 /*
1886                  * Anonymous process memory has backing store?
1887                  * Try to allocate it some swap space here.
1888                  * Lazyfree folio could be freed directly
1889                  */
1890                 if (folio_test_anon(folio) && folio_test_swapbacked(folio)) {
1891                         if (!folio_test_swapcache(folio)) {
1892                                 if (!(sc->gfp_mask & __GFP_IO))
1893                                         goto keep_locked;
1894                                 if (folio_maybe_dma_pinned(folio))
1895                                         goto keep_locked;
1896                                 if (folio_test_large(folio)) {
1897                                         /* cannot split folio, skip it */
1898                                         if (!can_split_folio(folio, NULL))
1899                                                 goto activate_locked;
1900                                         /*
1901                                          * Split folios without a PMD map right
1902                                          * away. Chances are some or all of the
1903                                          * tail pages can be freed without IO.
1904                                          */
1905                                         if (!folio_entire_mapcount(folio) &&
1906                                             split_folio_to_list(folio,
1907                                                                 folio_list))
1908                                                 goto activate_locked;
1909                                 }
1910                                 if (!add_to_swap(folio)) {
1911                                         if (!folio_test_large(folio))
1912                                                 goto activate_locked_split;
1913                                         /* Fallback to swap normal pages */
1914                                         if (split_folio_to_list(folio,
1915                                                                 folio_list))
1916                                                 goto activate_locked;
1917 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
1918                                         count_vm_event(THP_SWPOUT_FALLBACK);
1919 #endif
1920                                         if (!add_to_swap(folio))
1921                                                 goto activate_locked_split;
1922                                 }
1923                         }
1924                 } else if (folio_test_swapbacked(folio) &&
1925                            folio_test_large(folio)) {
1926                         /* Split shmem folio */
1927                         if (split_folio_to_list(folio, folio_list))
1928                                 goto keep_locked;
1929                 }
1930
1931                 /*
1932                  * If the folio was split above, the tail pages will make
1933                  * their own pass through this function and be accounted
1934                  * then.
1935                  */
1936                 if ((nr_pages > 1) && !folio_test_large(folio)) {
1937                         sc->nr_scanned -= (nr_pages - 1);
1938                         nr_pages = 1;
1939                 }
1940
1941                 /*
1942                  * The folio is mapped into the page tables of one or more
1943                  * processes. Try to unmap it here.
1944                  */
1945                 if (folio_mapped(folio)) {
1946                         enum ttu_flags flags = TTU_BATCH_FLUSH;
1947                         bool was_swapbacked = folio_test_swapbacked(folio);
1948
1949                         if (folio_test_pmd_mappable(folio))
1950                                 flags |= TTU_SPLIT_HUGE_PMD;
1951
1952                         try_to_unmap(folio, flags);
1953                         if (folio_mapped(folio)) {
1954                                 stat->nr_unmap_fail += nr_pages;
1955                                 if (!was_swapbacked &&
1956                                     folio_test_swapbacked(folio))
1957                                         stat->nr_lazyfree_fail += nr_pages;
1958                                 goto activate_locked;
1959                         }
1960                 }
1961
1962                 /*
1963                  * Folio is unmapped now so it cannot be newly pinned anymore.
1964                  * No point in trying to reclaim folio if it is pinned.
1965                  * Furthermore we don't want to reclaim underlying fs metadata
1966                  * if the folio is pinned and thus potentially modified by the
1967                  * pinning process as that may upset the filesystem.
1968                  */
1969                 if (folio_maybe_dma_pinned(folio))
1970                         goto activate_locked;
1971
1972                 mapping = folio_mapping(folio);
1973                 if (folio_test_dirty(folio)) {
1974                         /*
1975                          * Only kswapd can writeback filesystem folios
1976                          * to avoid risk of stack overflow. But avoid
1977                          * injecting inefficient single-folio I/O into
1978                          * flusher writeback as much as possible: only
1979                          * write folios when we've encountered many
1980                          * dirty folios, and when we've already scanned
1981                          * the rest of the LRU for clean folios and see
1982                          * the same dirty folios again (with the reclaim
1983                          * flag set).
1984                          */
1985                         if (folio_is_file_lru(folio) &&
1986                             (!current_is_kswapd() ||
1987                              !folio_test_reclaim(folio) ||
1988                              !test_bit(PGDAT_DIRTY, &pgdat->flags))) {
1989                                 /*
1990                                  * Immediately reclaim when written back.
1991                                  * Similar in principle to folio_deactivate()
1992                                  * except we already have the folio isolated
1993                                  * and know it's dirty
1994                                  */
1995                                 node_stat_mod_folio(folio, NR_VMSCAN_IMMEDIATE,
1996                                                 nr_pages);
1997                                 folio_set_reclaim(folio);
1998
1999                                 goto activate_locked;
2000                         }
2001
2002                         if (references == FOLIOREF_RECLAIM_CLEAN)
2003                                 goto keep_locked;
2004                         if (!may_enter_fs(folio, sc->gfp_mask))
2005                                 goto keep_locked;
2006                         if (!sc->may_writepage)
2007                                 goto keep_locked;
2008
2009                         /*
2010                          * Folio is dirty. Flush the TLB if a writable entry
2011                          * potentially exists to avoid CPU writes after I/O
2012                          * starts and then write it out here.
2013                          */
2014                         try_to_unmap_flush_dirty();
2015                         switch (pageout(folio, mapping, &plug)) {
2016                         case PAGE_KEEP:
2017                                 goto keep_locked;
2018                         case PAGE_ACTIVATE:
2019                                 goto activate_locked;
2020                         case PAGE_SUCCESS:
2021                                 stat->nr_pageout += nr_pages;
2022
2023                                 if (folio_test_writeback(folio))
2024                                         goto keep;
2025                                 if (folio_test_dirty(folio))
2026                                         goto keep;
2027
2028                                 /*
2029                                  * A synchronous write - probably a ramdisk.  Go
2030                                  * ahead and try to reclaim the folio.
2031                                  */
2032                                 if (!folio_trylock(folio))
2033                                         goto keep;
2034                                 if (folio_test_dirty(folio) ||
2035                                     folio_test_writeback(folio))
2036                                         goto keep_locked;
2037                                 mapping = folio_mapping(folio);
2038                                 fallthrough;
2039                         case PAGE_CLEAN:
2040                                 ; /* try to free the folio below */
2041                         }
2042                 }
2043
2044                 /*
2045                  * If the folio has buffers, try to free the buffer
2046                  * mappings associated with this folio. If we succeed
2047                  * we try to free the folio as well.
2048                  *
2049                  * We do this even if the folio is dirty.
2050                  * filemap_release_folio() does not perform I/O, but it
2051                  * is possible for a folio to have the dirty flag set,
2052                  * but it is actually clean (all its buffers are clean).
2053                  * This happens if the buffers were written out directly,
2054                  * with submit_bh(). ext3 will do this, as well as
2055                  * the blockdev mapping.  filemap_release_folio() will
2056                  * discover that cleanness and will drop the buffers
2057                  * and mark the folio clean - it can be freed.
2058                  *
2059                  * Rarely, folios can have buffers and no ->mapping.
2060                  * These are the folios which were not successfully
2061                  * invalidated in truncate_cleanup_folio().  We try to
2062                  * drop those buffers here and if that worked, and the
2063                  * folio is no longer mapped into process address space
2064                  * (refcount == 1) it can be freed.  Otherwise, leave
2065                  * the folio on the LRU so it is swappable.
2066                  */
2067                 if (folio_needs_release(folio)) {
2068                         if (!filemap_release_folio(folio, sc->gfp_mask))
2069                                 goto activate_locked;
2070                         if (!mapping && folio_ref_count(folio) == 1) {
2071                                 folio_unlock(folio);
2072                                 if (folio_put_testzero(folio))
2073                                         goto free_it;
2074                                 else {
2075                                         /*
2076                                          * rare race with speculative reference.
2077                                          * the speculative reference will free
2078                                          * this folio shortly, so we may
2079                                          * increment nr_reclaimed here (and
2080                                          * leave it off the LRU).
2081                                          */
2082                                         nr_reclaimed += nr_pages;
2083                                         continue;
2084                                 }
2085                         }
2086                 }
2087
2088                 if (folio_test_anon(folio) && !folio_test_swapbacked(folio)) {
2089                         /* follow __remove_mapping for reference */
2090                         if (!folio_ref_freeze(folio, 1))
2091                                 goto keep_locked;
2092                         /*
2093                          * The folio has only one reference left, which is
2094                          * from the isolation. After the caller puts the
2095                          * folio back on the lru and drops the reference, the
2096                          * folio will be freed anyway. It doesn't matter
2097                          * which lru it goes on. So we don't bother checking
2098                          * the dirty flag here.
2099                          */
2100                         count_vm_events(PGLAZYFREED, nr_pages);
2101                         count_memcg_folio_events(folio, PGLAZYFREED, nr_pages);
2102                 } else if (!mapping || !__remove_mapping(mapping, folio, true,
2103                                                          sc->target_mem_cgroup))
2104                         goto keep_locked;
2105
2106                 folio_unlock(folio);
2107 free_it:
2108                 /*
2109                  * Folio may get swapped out as a whole, need to account
2110                  * all pages in it.
2111                  */
2112                 nr_reclaimed += nr_pages;
2113
2114                 /*
2115                  * Is there need to periodically free_folio_list? It would
2116                  * appear not as the counts should be low
2117                  */
2118                 if (unlikely(folio_test_large(folio)))
2119                         destroy_large_folio(folio);
2120                 else
2121                         list_add(&folio->lru, &free_folios);
2122                 continue;
2123
2124 activate_locked_split:
2125                 /*
2126                  * The tail pages that are failed to add into swap cache
2127                  * reach here.  Fixup nr_scanned and nr_pages.
2128                  */
2129                 if (nr_pages > 1) {
2130                         sc->nr_scanned -= (nr_pages - 1);
2131                         nr_pages = 1;
2132                 }
2133 activate_locked:
2134                 /* Not a candidate for swapping, so reclaim swap space. */
2135                 if (folio_test_swapcache(folio) &&
2136                     (mem_cgroup_swap_full(folio) || folio_test_mlocked(folio)))
2137                         folio_free_swap(folio);
2138                 VM_BUG_ON_FOLIO(folio_test_active(folio), folio);
2139                 if (!folio_test_mlocked(folio)) {
2140                         int type = folio_is_file_lru(folio);
2141                         folio_set_active(folio);
2142                         stat->nr_activate[type] += nr_pages;
2143                         count_memcg_folio_events(folio, PGACTIVATE, nr_pages);
2144                 }
2145 keep_locked:
2146                 folio_unlock(folio);
2147 keep:
2148                 list_add(&folio->lru, &ret_folios);
2149                 VM_BUG_ON_FOLIO(folio_test_lru(folio) ||
2150                                 folio_test_unevictable(folio), folio);
2151         }
2152         /* 'folio_list' is always empty here */
2153
2154         /* Migrate folios selected for demotion */
2155         nr_reclaimed += demote_folio_list(&demote_folios, pgdat);
2156         /* Folios that could not be demoted are still in @demote_folios */
2157         if (!list_empty(&demote_folios)) {
2158                 /* Folios which weren't demoted go back on @folio_list */
2159                 list_splice_init(&demote_folios, folio_list);
2160
2161                 /*
2162                  * goto retry to reclaim the undemoted folios in folio_list if
2163                  * desired.
2164                  *
2165                  * Reclaiming directly from top tier nodes is not often desired
2166                  * due to it breaking the LRU ordering: in general memory
2167                  * should be reclaimed from lower tier nodes and demoted from
2168                  * top tier nodes.
2169                  *
2170                  * However, disabling reclaim from top tier nodes entirely
2171                  * would cause ooms in edge scenarios where lower tier memory
2172                  * is unreclaimable for whatever reason, eg memory being
2173                  * mlocked or too hot to reclaim. We can disable reclaim
2174                  * from top tier nodes in proactive reclaim though as that is
2175                  * not real memory pressure.
2176                  */
2177                 if (!sc->proactive) {
2178                         do_demote_pass = false;
2179                         goto retry;
2180                 }
2181         }
2182
2183         pgactivate = stat->nr_activate[0] + stat->nr_activate[1];
2184
2185         mem_cgroup_uncharge_list(&free_folios);
2186         try_to_unmap_flush();
2187         free_unref_page_list(&free_folios);
2188
2189         list_splice(&ret_folios, folio_list);
2190         count_vm_events(PGACTIVATE, pgactivate);
2191
2192         if (plug)
2193                 swap_write_unplug(plug);
2194         return nr_reclaimed;
2195 }
2196
2197 unsigned int reclaim_clean_pages_from_list(struct zone *zone,
2198                                            struct list_head *folio_list)
2199 {
2200         struct scan_control sc = {
2201                 .gfp_mask = GFP_KERNEL,
2202                 .may_unmap = 1,
2203         };
2204         struct reclaim_stat stat;
2205         unsigned int nr_reclaimed;
2206         struct folio *folio, *next;
2207         LIST_HEAD(clean_folios);
2208         unsigned int noreclaim_flag;
2209
2210         list_for_each_entry_safe(folio, next, folio_list, lru) {
2211                 if (!folio_test_hugetlb(folio) && folio_is_file_lru(folio) &&
2212                     !folio_test_dirty(folio) && !__folio_test_movable(folio) &&
2213                     !folio_test_unevictable(folio)) {
2214                         folio_clear_active(folio);
2215                         list_move(&folio->lru, &clean_folios);
2216                 }
2217         }
2218
2219         /*
2220          * We should be safe here since we are only dealing with file pages and
2221          * we are not kswapd and therefore cannot write dirty file pages. But
2222          * call memalloc_noreclaim_save() anyway, just in case these conditions
2223          * change in the future.
2224          */
2225         noreclaim_flag = memalloc_noreclaim_save();
2226         nr_reclaimed = shrink_folio_list(&clean_folios, zone->zone_pgdat, &sc,
2227                                         &stat, true);
2228         memalloc_noreclaim_restore(noreclaim_flag);
2229
2230         list_splice(&clean_folios, folio_list);
2231         mod_node_page_state(zone->zone_pgdat, NR_ISOLATED_FILE,
2232                             -(long)nr_reclaimed);
2233         /*
2234          * Since lazyfree pages are isolated from file LRU from the beginning,
2235          * they will rotate back to anonymous LRU in the end if it failed to
2236          * discard so isolated count will be mismatched.
2237          * Compensate the isolated count for both LRU lists.
2238          */
2239         mod_node_page_state(zone->zone_pgdat, NR_ISOLATED_ANON,
2240                             stat.nr_lazyfree_fail);
2241         mod_node_page_state(zone->zone_pgdat, NR_ISOLATED_FILE,
2242                             -(long)stat.nr_lazyfree_fail);
2243         return nr_reclaimed;
2244 }
2245
2246 /*
2247  * Update LRU sizes after isolating pages. The LRU size updates must
2248  * be complete before mem_cgroup_update_lru_size due to a sanity check.
2249  */
2250 static __always_inline void update_lru_sizes(struct lruvec *lruvec,
2251                         enum lru_list lru, unsigned long *nr_zone_taken)
2252 {
2253         int zid;
2254
2255         for (zid = 0; zid < MAX_NR_ZONES; zid++) {
2256                 if (!nr_zone_taken[zid])
2257                         continue;
2258
2259                 update_lru_size(lruvec, lru, zid, -nr_zone_taken[zid]);
2260         }
2261
2262 }
2263
2264 #ifdef CONFIG_CMA
2265 /*
2266  * It is waste of effort to scan and reclaim CMA pages if it is not available
2267  * for current allocation context. Kswapd can not be enrolled as it can not
2268  * distinguish this scenario by using sc->gfp_mask = GFP_KERNEL
2269  */
2270 static bool skip_cma(struct folio *folio, struct scan_control *sc)
2271 {
2272         return !current_is_kswapd() &&
2273                         gfp_migratetype(sc->gfp_mask) != MIGRATE_MOVABLE &&
2274                         get_pageblock_migratetype(&folio->page) == MIGRATE_CMA;
2275 }
2276 #else
2277 static bool skip_cma(struct folio *folio, struct scan_control *sc)
2278 {
2279         return false;
2280 }
2281 #endif
2282
2283 /*
2284  * Isolating page from the lruvec to fill in @dst list by nr_to_scan times.
2285  *
2286  * lruvec->lru_lock is heavily contended.  Some of the functions that
2287  * shrink the lists perform better by taking out a batch of pages
2288  * and working on them outside the LRU lock.
2289  *
2290  * For pagecache intensive workloads, this function is the hottest
2291  * spot in the kernel (apart from copy_*_user functions).
2292  *
2293  * Lru_lock must be held before calling this function.
2294  *
2295  * @nr_to_scan: The number of eligible pages to look through on the list.
2296  * @lruvec:     The LRU vector to pull pages from.
2297  * @dst:        The temp list to put pages on to.
2298  * @nr_scanned: The number of pages that were scanned.
2299  * @sc:         The scan_control struct for this reclaim session
2300  * @lru:        LRU list id for isolating
2301  *
2302  * returns how many pages were moved onto *@dst.
2303  */
2304 static unsigned long isolate_lru_folios(unsigned long nr_to_scan,
2305                 struct lruvec *lruvec, struct list_head *dst,
2306                 unsigned long *nr_scanned, struct scan_control *sc,
2307                 enum lru_list lru)
2308 {
2309         struct list_head *src = &lruvec->lists[lru];
2310         unsigned long nr_taken = 0;
2311         unsigned long nr_zone_taken[MAX_NR_ZONES] = { 0 };
2312         unsigned long nr_skipped[MAX_NR_ZONES] = { 0, };
2313         unsigned long skipped = 0;
2314         unsigned long scan, total_scan, nr_pages;
2315         LIST_HEAD(folios_skipped);
2316
2317         total_scan = 0;
2318         scan = 0;
2319         while (scan < nr_to_scan && !list_empty(src)) {
2320                 struct list_head *move_to = src;
2321                 struct folio *folio;
2322
2323                 folio = lru_to_folio(src);
2324                 prefetchw_prev_lru_folio(folio, src, flags);
2325
2326                 nr_pages = folio_nr_pages(folio);
2327                 total_scan += nr_pages;
2328
2329                 if (folio_zonenum(folio) > sc->reclaim_idx ||
2330                                 skip_cma(folio, sc)) {
2331                         nr_skipped[folio_zonenum(folio)] += nr_pages;
2332                         move_to = &folios_skipped;
2333                         goto move;
2334                 }
2335
2336                 /*
2337                  * Do not count skipped folios because that makes the function
2338                  * return with no isolated folios if the LRU mostly contains
2339                  * ineligible folios.  This causes the VM to not reclaim any
2340                  * folios, triggering a premature OOM.
2341                  * Account all pages in a folio.
2342                  */
2343                 scan += nr_pages;
2344
2345                 if (!folio_test_lru(folio))
2346                         goto move;
2347                 if (!sc->may_unmap && folio_mapped(folio))
2348                         goto move;
2349
2350                 /*
2351                  * Be careful not to clear the lru flag until after we're
2352                  * sure the folio is not being freed elsewhere -- the
2353                  * folio release code relies on it.
2354                  */
2355                 if (unlikely(!folio_try_get(folio)))
2356                         goto move;
2357
2358                 if (!folio_test_clear_lru(folio)) {
2359                         /* Another thread is already isolating this folio */
2360                         folio_put(folio);
2361                         goto move;
2362                 }
2363
2364                 nr_taken += nr_pages;
2365                 nr_zone_taken[folio_zonenum(folio)] += nr_pages;
2366                 move_to = dst;
2367 move:
2368                 list_move(&folio->lru, move_to);
2369         }
2370
2371         /*
2372          * Splice any skipped folios to the start of the LRU list. Note that
2373          * this disrupts the LRU order when reclaiming for lower zones but
2374          * we cannot splice to the tail. If we did then the SWAP_CLUSTER_MAX
2375          * scanning would soon rescan the same folios to skip and waste lots
2376          * of cpu cycles.
2377          */
2378         if (!list_empty(&folios_skipped)) {
2379                 int zid;
2380
2381                 list_splice(&folios_skipped, src);
2382                 for (zid = 0; zid < MAX_NR_ZONES; zid++) {
2383                         if (!nr_skipped[zid])
2384                                 continue;
2385
2386                         __count_zid_vm_events(PGSCAN_SKIP, zid, nr_skipped[zid]);
2387                         skipped += nr_skipped[zid];
2388                 }
2389         }
2390         *nr_scanned = total_scan;
2391         trace_mm_vmscan_lru_isolate(sc->reclaim_idx, sc->order, nr_to_scan,
2392                                     total_scan, skipped, nr_taken,
2393                                     sc->may_unmap ? 0 : ISOLATE_UNMAPPED, lru);
2394         update_lru_sizes(lruvec, lru, nr_zone_taken);
2395         return nr_taken;
2396 }
2397
2398 /**
2399  * folio_isolate_lru() - Try to isolate a folio from its LRU list.
2400  * @folio: Folio to isolate from its LRU list.
2401  *
2402  * Isolate a @folio from an LRU list and adjust the vmstat statistic
2403  * corresponding to whatever LRU list the folio was on.
2404  *
2405  * The folio will have its LRU flag cleared.  If it was found on the
2406  * active list, it will have the Active flag set.  If it was found on the
2407  * unevictable list, it will have the Unevictable flag set.  These flags
2408  * may need to be cleared by the caller before letting the page go.
2409  *
2410  * Context:
2411  *
2412  * (1) Must be called with an elevated refcount on the folio. This is a
2413  *     fundamental difference from isolate_lru_folios() (which is called
2414  *     without a stable reference).
2415  * (2) The lru_lock must not be held.
2416  * (3) Interrupts must be enabled.
2417  *
2418  * Return: true if the folio was removed from an LRU list.
2419  * false if the folio was not on an LRU list.
2420  */
2421 bool folio_isolate_lru(struct folio *folio)
2422 {
2423         bool ret = false;
2424
2425         VM_BUG_ON_FOLIO(!folio_ref_count(folio), folio);
2426
2427         if (folio_test_clear_lru(folio)) {
2428                 struct lruvec *lruvec;
2429
2430                 folio_get(folio);
2431                 lruvec = folio_lruvec_lock_irq(folio);
2432                 lruvec_del_folio(lruvec, folio);
2433                 unlock_page_lruvec_irq(lruvec);
2434                 ret = true;
2435         }
2436
2437         return ret;
2438 }
2439
2440 /*
2441  * A direct reclaimer may isolate SWAP_CLUSTER_MAX pages from the LRU list and
2442  * then get rescheduled. When there are massive number of tasks doing page
2443  * allocation, such sleeping direct reclaimers may keep piling up on each CPU,
2444  * the LRU list will go small and be scanned faster than necessary, leading to
2445  * unnecessary swapping, thrashing and OOM.
2446  */
2447 static int too_many_isolated(struct pglist_data *pgdat, int file,
2448                 struct scan_control *sc)
2449 {
2450         unsigned long inactive, isolated;
2451         bool too_many;
2452
2453         if (current_is_kswapd())
2454                 return 0;
2455
2456         if (!writeback_throttling_sane(sc))
2457                 return 0;
2458
2459         if (file) {
2460                 inactive = node_page_state(pgdat, NR_INACTIVE_FILE);
2461                 isolated = node_page_state(pgdat, NR_ISOLATED_FILE);
2462         } else {
2463                 inactive = node_page_state(pgdat, NR_INACTIVE_ANON);
2464                 isolated = node_page_state(pgdat, NR_ISOLATED_ANON);
2465         }
2466
2467         /*
2468          * GFP_NOIO/GFP_NOFS callers are allowed to isolate more pages, so they
2469          * won't get blocked by normal direct-reclaimers, forming a circular
2470          * deadlock.
2471          */
2472         if (gfp_has_io_fs(sc->gfp_mask))
2473                 inactive >>= 3;
2474
2475         too_many = isolated > inactive;
2476
2477         /* Wake up tasks throttled due to too_many_isolated. */
2478         if (!too_many)
2479                 wake_throttle_isolated(pgdat);
2480
2481         return too_many;
2482 }
2483
2484 /*
2485  * move_folios_to_lru() moves folios from private @list to appropriate LRU list.
2486  * On return, @list is reused as a list of folios to be freed by the caller.
2487  *
2488  * Returns the number of pages moved to the given lruvec.
2489  */
2490 static unsigned int move_folios_to_lru(struct lruvec *lruvec,
2491                 struct list_head *list)
2492 {
2493         int nr_pages, nr_moved = 0;
2494         LIST_HEAD(folios_to_free);
2495
2496         while (!list_empty(list)) {
2497                 struct folio *folio = lru_to_folio(list);
2498
2499                 VM_BUG_ON_FOLIO(folio_test_lru(folio), folio);
2500                 list_del(&folio->lru);
2501                 if (unlikely(!folio_evictable(folio))) {
2502                         spin_unlock_irq(&lruvec->lru_lock);
2503                         folio_putback_lru(folio);
2504                         spin_lock_irq(&lruvec->lru_lock);
2505                         continue;
2506                 }
2507
2508                 /*
2509                  * The folio_set_lru needs to be kept here for list integrity.
2510                  * Otherwise:
2511                  *   #0 move_folios_to_lru             #1 release_pages
2512                  *   if (!folio_put_testzero())
2513                  *                                    if (folio_put_testzero())
2514                  *                                      !lru //skip lru_lock
2515                  *     folio_set_lru()
2516                  *     list_add(&folio->lru,)
2517                  *                                        list_add(&folio->lru,)
2518                  */
2519                 folio_set_lru(folio);
2520
2521                 if (unlikely(folio_put_testzero(folio))) {
2522                         __folio_clear_lru_flags(folio);
2523
2524                         if (unlikely(folio_test_large(folio))) {
2525                                 spin_unlock_irq(&lruvec->lru_lock);
2526                                 destroy_large_folio(folio);
2527                                 spin_lock_irq(&lruvec->lru_lock);
2528                         } else
2529                                 list_add(&folio->lru, &folios_to_free);
2530
2531                         continue;
2532                 }
2533
2534                 /*
2535                  * All pages were isolated from the same lruvec (and isolation
2536                  * inhibits memcg migration).
2537                  */
2538                 VM_BUG_ON_FOLIO(!folio_matches_lruvec(folio, lruvec), folio);
2539                 lruvec_add_folio(lruvec, folio);
2540                 nr_pages = folio_nr_pages(folio);
2541                 nr_moved += nr_pages;
2542                 if (folio_test_active(folio))
2543                         workingset_age_nonresident(lruvec, nr_pages);
2544         }
2545
2546         /*
2547          * To save our caller's stack, now use input list for pages to free.
2548          */
2549         list_splice(&folios_to_free, list);
2550
2551         return nr_moved;
2552 }
2553
2554 /*
2555  * If a kernel thread (such as nfsd for loop-back mounts) services a backing
2556  * device by writing to the page cache it sets PF_LOCAL_THROTTLE. In this case
2557  * we should not throttle.  Otherwise it is safe to do so.
2558  */
2559 static int current_may_throttle(void)
2560 {
2561         return !(current->flags & PF_LOCAL_THROTTLE);
2562 }
2563
2564 /*
2565  * shrink_inactive_list() is a helper for shrink_node().  It returns the number
2566  * of reclaimed pages
2567  */
2568 static unsigned long shrink_inactive_list(unsigned long nr_to_scan,
2569                 struct lruvec *lruvec, struct scan_control *sc,
2570                 enum lru_list lru)
2571 {
2572         LIST_HEAD(folio_list);
2573         unsigned long nr_scanned;
2574         unsigned int nr_reclaimed = 0;
2575         unsigned long nr_taken;
2576         struct reclaim_stat stat;
2577         bool file = is_file_lru(lru);
2578         enum vm_event_item item;
2579         struct pglist_data *pgdat = lruvec_pgdat(lruvec);
2580         bool stalled = false;
2581
2582         while (unlikely(too_many_isolated(pgdat, file, sc))) {
2583                 if (stalled)
2584                         return 0;
2585
2586                 /* wait a bit for the reclaimer. */
2587                 stalled = true;
2588                 reclaim_throttle(pgdat, VMSCAN_THROTTLE_ISOLATED);
2589
2590                 /* We are about to die and free our memory. Return now. */
2591                 if (fatal_signal_pending(current))
2592                         return SWAP_CLUSTER_MAX;
2593         }
2594
2595         lru_add_drain();
2596
2597         spin_lock_irq(&lruvec->lru_lock);
2598
2599         nr_taken = isolate_lru_folios(nr_to_scan, lruvec, &folio_list,
2600                                      &nr_scanned, sc, lru);
2601
2602         __mod_node_page_state(pgdat, NR_ISOLATED_ANON + file, nr_taken);
2603         item = PGSCAN_KSWAPD + reclaimer_offset();
2604         if (!cgroup_reclaim(sc))
2605                 __count_vm_events(item, nr_scanned);
2606         __count_memcg_events(lruvec_memcg(lruvec), item, nr_scanned);
2607         __count_vm_events(PGSCAN_ANON + file, nr_scanned);
2608
2609         spin_unlock_irq(&lruvec->lru_lock);
2610
2611         if (nr_taken == 0)
2612                 return 0;
2613
2614         nr_reclaimed = shrink_folio_list(&folio_list, pgdat, sc, &stat, false);
2615
2616         spin_lock_irq(&lruvec->lru_lock);
2617         move_folios_to_lru(lruvec, &folio_list);
2618
2619         __mod_node_page_state(pgdat, NR_ISOLATED_ANON + file, -nr_taken);
2620         item = PGSTEAL_KSWAPD + reclaimer_offset();
2621         if (!cgroup_reclaim(sc))
2622                 __count_vm_events(item, nr_reclaimed);
2623         __count_memcg_events(lruvec_memcg(lruvec), item, nr_reclaimed);
2624         __count_vm_events(PGSTEAL_ANON + file, nr_reclaimed);
2625         spin_unlock_irq(&lruvec->lru_lock);
2626
2627         lru_note_cost(lruvec, file, stat.nr_pageout, nr_scanned - nr_reclaimed);
2628         mem_cgroup_uncharge_list(&folio_list);
2629         free_unref_page_list(&folio_list);
2630
2631         /*
2632          * If dirty folios are scanned that are not queued for IO, it
2633          * implies that flushers are not doing their job. This can
2634          * happen when memory pressure pushes dirty folios to the end of
2635          * the LRU before the dirty limits are breached and the dirty
2636          * data has expired. It can also happen when the proportion of
2637          * dirty folios grows not through writes but through memory
2638          * pressure reclaiming all the clean cache. And in some cases,
2639          * the flushers simply cannot keep up with the allocation
2640          * rate. Nudge the flusher threads in case they are asleep.
2641          */
2642         if (stat.nr_unqueued_dirty == nr_taken) {
2643                 wakeup_flusher_threads(WB_REASON_VMSCAN);
2644                 /*
2645                  * For cgroupv1 dirty throttling is achieved by waking up
2646                  * the kernel flusher here and later waiting on folios
2647                  * which are in writeback to finish (see shrink_folio_list()).
2648                  *
2649                  * Flusher may not be able to issue writeback quickly
2650                  * enough for cgroupv1 writeback throttling to work
2651                  * on a large system.
2652                  */
2653                 if (!writeback_throttling_sane(sc))
2654                         reclaim_throttle(pgdat, VMSCAN_THROTTLE_WRITEBACK);
2655         }
2656
2657         sc->nr.dirty += stat.nr_dirty;
2658         sc->nr.congested += stat.nr_congested;
2659         sc->nr.unqueued_dirty += stat.nr_unqueued_dirty;
2660         sc->nr.writeback += stat.nr_writeback;
2661         sc->nr.immediate += stat.nr_immediate;
2662         sc->nr.taken += nr_taken;
2663         if (file)
2664                 sc->nr.file_taken += nr_taken;
2665
2666         trace_mm_vmscan_lru_shrink_inactive(pgdat->node_id,
2667                         nr_scanned, nr_reclaimed, &stat, sc->priority, file);
2668         return nr_reclaimed;
2669 }
2670
2671 /*
2672  * shrink_active_list() moves folios from the active LRU to the inactive LRU.
2673  *
2674  * We move them the other way if the folio is referenced by one or more
2675  * processes.
2676  *
2677  * If the folios are mostly unmapped, the processing is fast and it is
2678  * appropriate to hold lru_lock across the whole operation.  But if
2679  * the folios are mapped, the processing is slow (folio_referenced()), so
2680  * we should drop lru_lock around each folio.  It's impossible to balance
2681  * this, so instead we remove the folios from the LRU while processing them.
2682  * It is safe to rely on the active flag against the non-LRU folios in here
2683  * because nobody will play with that bit on a non-LRU folio.
2684  *
2685  * The downside is that we have to touch folio->_refcount against each folio.
2686  * But we had to alter folio->flags anyway.
2687  */
2688 static void shrink_active_list(unsigned long nr_to_scan,
2689                                struct lruvec *lruvec,
2690                                struct scan_control *sc,
2691                                enum lru_list lru)
2692 {
2693         unsigned long nr_taken;
2694         unsigned long nr_scanned;
2695         unsigned long vm_flags;
2696         LIST_HEAD(l_hold);      /* The folios which were snipped off */
2697         LIST_HEAD(l_active);
2698         LIST_HEAD(l_inactive);
2699         unsigned nr_deactivate, nr_activate;
2700         unsigned nr_rotated = 0;
2701         int file = is_file_lru(lru);
2702         struct pglist_data *pgdat = lruvec_pgdat(lruvec);
2703
2704         lru_add_drain();
2705
2706         spin_lock_irq(&lruvec->lru_lock);
2707
2708         nr_taken = isolate_lru_folios(nr_to_scan, lruvec, &l_hold,
2709                                      &nr_scanned, sc, lru);
2710
2711         __mod_node_page_state(pgdat, NR_ISOLATED_ANON + file, nr_taken);
2712
2713         if (!cgroup_reclaim(sc))
2714                 __count_vm_events(PGREFILL, nr_scanned);
2715         __count_memcg_events(lruvec_memcg(lruvec), PGREFILL, nr_scanned);
2716
2717         spin_unlock_irq(&lruvec->lru_lock);
2718
2719         while (!list_empty(&l_hold)) {
2720                 struct folio *folio;
2721
2722                 cond_resched();
2723                 folio = lru_to_folio(&l_hold);
2724                 list_del(&folio->lru);
2725
2726                 if (unlikely(!folio_evictable(folio))) {
2727                         folio_putback_lru(folio);
2728                         continue;
2729                 }
2730
2731                 if (unlikely(buffer_heads_over_limit)) {
2732                         if (folio_needs_release(folio) &&
2733                             folio_trylock(folio)) {
2734                                 filemap_release_folio(folio, 0);
2735                                 folio_unlock(folio);
2736                         }
2737                 }
2738
2739                 /* Referenced or rmap lock contention: rotate */
2740                 if (folio_referenced(folio, 0, sc->target_mem_cgroup,
2741                                      &vm_flags) != 0) {
2742                         /*
2743                          * Identify referenced, file-backed active folios and
2744                          * give them one more trip around the active list. So
2745                          * that executable code get better chances to stay in
2746                          * memory under moderate memory pressure.  Anon folios
2747                          * are not likely to be evicted by use-once streaming
2748                          * IO, plus JVM can create lots of anon VM_EXEC folios,
2749                          * so we ignore them here.
2750                          */
2751                         if ((vm_flags & VM_EXEC) && folio_is_file_lru(folio)) {
2752                                 nr_rotated += folio_nr_pages(folio);
2753                                 list_add(&folio->lru, &l_active);
2754                                 continue;
2755                         }
2756                 }
2757
2758                 folio_clear_active(folio);      /* we are de-activating */
2759                 folio_set_workingset(folio);
2760                 list_add(&folio->lru, &l_inactive);
2761         }
2762
2763         /*
2764          * Move folios back to the lru list.
2765          */
2766         spin_lock_irq(&lruvec->lru_lock);
2767
2768         nr_activate = move_folios_to_lru(lruvec, &l_active);
2769         nr_deactivate = move_folios_to_lru(lruvec, &l_inactive);
2770         /* Keep all free folios in l_active list */
2771         list_splice(&l_inactive, &l_active);
2772
2773         __count_vm_events(PGDEACTIVATE, nr_deactivate);
2774         __count_memcg_events(lruvec_memcg(lruvec), PGDEACTIVATE, nr_deactivate);
2775
2776         __mod_node_page_state(pgdat, NR_ISOLATED_ANON + file, -nr_taken);
2777         spin_unlock_irq(&lruvec->lru_lock);
2778
2779         if (nr_rotated)
2780                 lru_note_cost(lruvec, file, 0, nr_rotated);
2781         mem_cgroup_uncharge_list(&l_active);
2782         free_unref_page_list(&l_active);
2783         trace_mm_vmscan_lru_shrink_active(pgdat->node_id, nr_taken, nr_activate,
2784                         nr_deactivate, nr_rotated, sc->priority, file);
2785 }
2786
2787 static unsigned int reclaim_folio_list(struct list_head *folio_list,
2788                                       struct pglist_data *pgdat)
2789 {
2790         struct reclaim_stat dummy_stat;
2791         unsigned int nr_reclaimed;
2792         struct folio *folio;
2793         struct scan_control sc = {
2794                 .gfp_mask = GFP_KERNEL,
2795                 .may_writepage = 1,
2796                 .may_unmap = 1,
2797                 .may_swap = 1,
2798                 .no_demotion = 1,
2799         };
2800
2801         nr_reclaimed = shrink_folio_list(folio_list, pgdat, &sc, &dummy_stat, false);
2802         while (!list_empty(folio_list)) {
2803                 folio = lru_to_folio(folio_list);
2804                 list_del(&folio->lru);
2805                 folio_putback_lru(folio);
2806         }
2807
2808         return nr_reclaimed;
2809 }
2810
2811 unsigned long reclaim_pages(struct list_head *folio_list)
2812 {
2813         int nid;
2814         unsigned int nr_reclaimed = 0;
2815         LIST_HEAD(node_folio_list);
2816         unsigned int noreclaim_flag;
2817
2818         if (list_empty(folio_list))
2819                 return nr_reclaimed;
2820
2821         noreclaim_flag = memalloc_noreclaim_save();
2822
2823         nid = folio_nid(lru_to_folio(folio_list));
2824         do {
2825                 struct folio *folio = lru_to_folio(folio_list);
2826
2827                 if (nid == folio_nid(folio)) {
2828                         folio_clear_active(folio);
2829                         list_move(&folio->lru, &node_folio_list);
2830                         continue;
2831                 }
2832
2833                 nr_reclaimed += reclaim_folio_list(&node_folio_list, NODE_DATA(nid));
2834                 nid = folio_nid(lru_to_folio(folio_list));
2835         } while (!list_empty(folio_list));
2836
2837         nr_reclaimed += reclaim_folio_list(&node_folio_list, NODE_DATA(nid));
2838
2839         memalloc_noreclaim_restore(noreclaim_flag);
2840
2841         return nr_reclaimed;
2842 }
2843
2844 static unsigned long shrink_list(enum lru_list lru, unsigned long nr_to_scan,
2845                                  struct lruvec *lruvec, struct scan_control *sc)
2846 {
2847         if (is_active_lru(lru)) {
2848                 if (sc->may_deactivate & (1 << is_file_lru(lru)))
2849                         shrink_active_list(nr_to_scan, lruvec, sc, lru);
2850                 else
2851                         sc->skipped_deactivate = 1;
2852                 return 0;
2853         }
2854
2855         return shrink_inactive_list(nr_to_scan, lruvec, sc, lru);
2856 }
2857
2858 /*
2859  * The inactive anon list should be small enough that the VM never has
2860  * to do too much work.
2861  *
2862  * The inactive file list should be small enough to leave most memory
2863  * to the established workingset on the scan-resistant active list,
2864  * but large enough to avoid thrashing the aggregate readahead window.
2865  *
2866  * Both inactive lists should also be large enough that each inactive
2867  * folio has a chance to be referenced again before it is reclaimed.
2868  *
2869  * If that fails and refaulting is observed, the inactive list grows.
2870  *
2871  * The inactive_ratio is the target ratio of ACTIVE to INACTIVE folios
2872  * on this LRU, maintained by the pageout code. An inactive_ratio
2873  * of 3 means 3:1 or 25% of the folios are kept on the inactive list.
2874  *
2875  * total     target    max
2876  * memory    ratio     inactive
2877  * -------------------------------------
2878  *   10MB       1         5MB
2879  *  100MB       1        50MB
2880  *    1GB       3       250MB
2881  *   10GB      10       0.9GB
2882  *  100GB      31         3GB
2883  *    1TB     101        10GB
2884  *   10TB     320        32GB
2885  */
2886 static bool inactive_is_low(struct lruvec *lruvec, enum lru_list inactive_lru)
2887 {
2888         enum lru_list active_lru = inactive_lru + LRU_ACTIVE;
2889         unsigned long inactive, active;
2890         unsigned long inactive_ratio;
2891         unsigned long gb;
2892
2893         inactive = lruvec_page_state(lruvec, NR_LRU_BASE + inactive_lru);
2894         active = lruvec_page_state(lruvec, NR_LRU_BASE + active_lru);
2895
2896         gb = (inactive + active) >> (30 - PAGE_SHIFT);
2897         if (gb)
2898                 inactive_ratio = int_sqrt(10 * gb);
2899         else
2900                 inactive_ratio = 1;
2901
2902         return inactive * inactive_ratio < active;
2903 }
2904
2905 enum scan_balance {
2906         SCAN_EQUAL,
2907         SCAN_FRACT,
2908         SCAN_ANON,
2909         SCAN_FILE,
2910 };
2911
2912 static void prepare_scan_count(pg_data_t *pgdat, struct scan_control *sc)
2913 {
2914         unsigned long file;
2915         struct lruvec *target_lruvec;
2916
2917         if (lru_gen_enabled())
2918                 return;
2919
2920         target_lruvec = mem_cgroup_lruvec(sc->target_mem_cgroup, pgdat);
2921
2922         /*
2923          * Flush the memory cgroup stats, so that we read accurate per-memcg
2924          * lruvec stats for heuristics.
2925          */
2926         mem_cgroup_flush_stats();
2927
2928         /*
2929          * Determine the scan balance between anon and file LRUs.
2930          */
2931         spin_lock_irq(&target_lruvec->lru_lock);
2932         sc->anon_cost = target_lruvec->anon_cost;
2933         sc->file_cost = target_lruvec->file_cost;
2934         spin_unlock_irq(&target_lruvec->lru_lock);
2935
2936         /*
2937          * Target desirable inactive:active list ratios for the anon
2938          * and file LRU lists.
2939          */
2940         if (!sc->force_deactivate) {
2941                 unsigned long refaults;
2942
2943                 /*
2944                  * When refaults are being observed, it means a new
2945                  * workingset is being established. Deactivate to get
2946                  * rid of any stale active pages quickly.
2947                  */
2948                 refaults = lruvec_page_state(target_lruvec,
2949                                 WORKINGSET_ACTIVATE_ANON);
2950                 if (refaults != target_lruvec->refaults[WORKINGSET_ANON] ||
2951                         inactive_is_low(target_lruvec, LRU_INACTIVE_ANON))
2952                         sc->may_deactivate |= DEACTIVATE_ANON;
2953                 else
2954                         sc->may_deactivate &= ~DEACTIVATE_ANON;
2955
2956                 refaults = lruvec_page_state(target_lruvec,
2957                                 WORKINGSET_ACTIVATE_FILE);
2958                 if (refaults != target_lruvec->refaults[WORKINGSET_FILE] ||
2959                     inactive_is_low(target_lruvec, LRU_INACTIVE_FILE))
2960                         sc->may_deactivate |= DEACTIVATE_FILE;
2961                 else
2962                         sc->may_deactivate &= ~DEACTIVATE_FILE;
2963         } else
2964                 sc->may_deactivate = DEACTIVATE_ANON | DEACTIVATE_FILE;
2965
2966         /*
2967          * If we have plenty of inactive file pages that aren't
2968          * thrashing, try to reclaim those first before touching
2969          * anonymous pages.
2970          */
2971         file = lruvec_page_state(target_lruvec, NR_INACTIVE_FILE);
2972         if (file >> sc->priority && !(sc->may_deactivate & DEACTIVATE_FILE))
2973                 sc->cache_trim_mode = 1;
2974         else
2975                 sc->cache_trim_mode = 0;
2976
2977         /*
2978          * Prevent the reclaimer from falling into the cache trap: as
2979          * cache pages start out inactive, every cache fault will tip
2980          * the scan balance towards the file LRU.  And as the file LRU
2981          * shrinks, so does the window for rotation from references.
2982          * This means we have a runaway feedback loop where a tiny
2983          * thrashing file LRU becomes infinitely more attractive than
2984          * anon pages.  Try to detect this based on file LRU size.
2985          */
2986         if (!cgroup_reclaim(sc)) {
2987                 unsigned long total_high_wmark = 0;
2988                 unsigned long free, anon;
2989                 int z;
2990
2991                 free = sum_zone_node_page_state(pgdat->node_id, NR_FREE_PAGES);
2992                 file = node_page_state(pgdat, NR_ACTIVE_FILE) +
2993                            node_page_state(pgdat, NR_INACTIVE_FILE);
2994
2995                 for (z = 0; z < MAX_NR_ZONES; z++) {
2996                         struct zone *zone = &pgdat->node_zones[z];
2997
2998                         if (!managed_zone(zone))
2999                                 continue;
3000
3001                         total_high_wmark += high_wmark_pages(zone);
3002                 }
3003
3004                 /*
3005                  * Consider anon: if that's low too, this isn't a
3006                  * runaway file reclaim problem, but rather just
3007                  * extreme pressure. Reclaim as per usual then.
3008                  */
3009                 anon = node_page_state(pgdat, NR_INACTIVE_ANON);
3010
3011                 sc->file_is_tiny =
3012                         file + free <= total_high_wmark &&
3013                         !(sc->may_deactivate & DEACTIVATE_ANON) &&
3014                         anon >> sc->priority;
3015         }
3016 }
3017
3018 /*
3019  * Determine how aggressively the anon and file LRU lists should be
3020  * scanned.
3021  *
3022  * nr[0] = anon inactive folios to scan; nr[1] = anon active folios to scan
3023  * nr[2] = file inactive folios to scan; nr[3] = file active folios to scan
3024  */
3025 static void get_scan_count(struct lruvec *lruvec, struct scan_control *sc,
3026                            unsigned long *nr)
3027 {
3028         struct pglist_data *pgdat = lruvec_pgdat(lruvec);
3029         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
3030         unsigned long anon_cost, file_cost, total_cost;
3031         int swappiness = mem_cgroup_swappiness(memcg);
3032         u64 fraction[ANON_AND_FILE];
3033         u64 denominator = 0;    /* gcc */
3034         enum scan_balance scan_balance;
3035         unsigned long ap, fp;
3036         enum lru_list lru;
3037
3038         /* If we have no swap space, do not bother scanning anon folios. */
3039         if (!sc->may_swap || !can_reclaim_anon_pages(memcg, pgdat->node_id, sc)) {
3040                 scan_balance = SCAN_FILE;
3041                 goto out;
3042         }
3043
3044         /*
3045          * Global reclaim will swap to prevent OOM even with no
3046          * swappiness, but memcg users want to use this knob to
3047          * disable swapping for individual groups completely when
3048          * using the memory controller's swap limit feature would be
3049          * too expensive.
3050          */
3051         if (cgroup_reclaim(sc) && !swappiness) {
3052                 scan_balance = SCAN_FILE;
3053                 goto out;
3054         }
3055
3056         /*
3057          * Do not apply any pressure balancing cleverness when the
3058          * system is close to OOM, scan both anon and file equally
3059          * (unless the swappiness setting disagrees with swapping).
3060          */
3061         if (!sc->priority && swappiness) {
3062                 scan_balance = SCAN_EQUAL;
3063                 goto out;
3064         }
3065
3066         /*
3067          * If the system is almost out of file pages, force-scan anon.
3068          */
3069         if (sc->file_is_tiny) {
3070                 scan_balance = SCAN_ANON;
3071                 goto out;
3072         }
3073
3074         /*
3075          * If there is enough inactive page cache, we do not reclaim
3076          * anything from the anonymous working right now.
3077          */
3078         if (sc->cache_trim_mode) {
3079                 scan_balance = SCAN_FILE;
3080                 goto out;
3081         }
3082
3083         scan_balance = SCAN_FRACT;
3084         /*
3085          * Calculate the pressure balance between anon and file pages.
3086          *
3087          * The amount of pressure we put on each LRU is inversely
3088          * proportional to the cost of reclaiming each list, as
3089          * determined by the share of pages that are refaulting, times
3090          * the relative IO cost of bringing back a swapped out
3091          * anonymous page vs reloading a filesystem page (swappiness).
3092          *
3093          * Although we limit that influence to ensure no list gets
3094          * left behind completely: at least a third of the pressure is
3095          * applied, before swappiness.
3096          *
3097          * With swappiness at 100, anon and file have equal IO cost.
3098          */
3099         total_cost = sc->anon_cost + sc->file_cost;
3100         anon_cost = total_cost + sc->anon_cost;
3101         file_cost = total_cost + sc->file_cost;
3102         total_cost = anon_cost + file_cost;
3103
3104         ap = swappiness * (total_cost + 1);
3105         ap /= anon_cost + 1;
3106
3107         fp = (200 - swappiness) * (total_cost + 1);
3108         fp /= file_cost + 1;
3109
3110         fraction[0] = ap;
3111         fraction[1] = fp;
3112         denominator = ap + fp;
3113 out:
3114         for_each_evictable_lru(lru) {
3115                 int file = is_file_lru(lru);
3116                 unsigned long lruvec_size;
3117                 unsigned long low, min;
3118                 unsigned long scan;
3119
3120                 lruvec_size = lruvec_lru_size(lruvec, lru, sc->reclaim_idx);
3121                 mem_cgroup_protection(sc->target_mem_cgroup, memcg,
3122                                       &min, &low);
3123
3124                 if (min || low) {
3125                         /*
3126                          * Scale a cgroup's reclaim pressure by proportioning
3127                          * its current usage to its memory.low or memory.min
3128                          * setting.
3129                          *
3130                          * This is important, as otherwise scanning aggression
3131                          * becomes extremely binary -- from nothing as we
3132                          * approach the memory protection threshold, to totally
3133                          * nominal as we exceed it.  This results in requiring
3134                          * setting extremely liberal protection thresholds. It
3135                          * also means we simply get no protection at all if we
3136                          * set it too low, which is not ideal.
3137                          *
3138                          * If there is any protection in place, we reduce scan
3139                          * pressure by how much of the total memory used is
3140                          * within protection thresholds.
3141                          *
3142                          * There is one special case: in the first reclaim pass,
3143                          * we skip over all groups that are within their low
3144                          * protection. If that fails to reclaim enough pages to
3145                          * satisfy the reclaim goal, we come back and override
3146                          * the best-effort low protection. However, we still
3147                          * ideally want to honor how well-behaved groups are in
3148                          * that case instead of simply punishing them all
3149                          * equally. As such, we reclaim them based on how much
3150                          * memory they are using, reducing the scan pressure
3151                          * again by how much of the total memory used is under
3152                          * hard protection.
3153                          */
3154                         unsigned long cgroup_size = mem_cgroup_size(memcg);
3155                         unsigned long protection;
3156
3157                         /* memory.low scaling, make sure we retry before OOM */
3158                         if (!sc->memcg_low_reclaim && low > min) {
3159                                 protection = low;
3160                                 sc->memcg_low_skipped = 1;
3161                         } else {
3162                                 protection = min;
3163                         }
3164
3165                         /* Avoid TOCTOU with earlier protection check */
3166                         cgroup_size = max(cgroup_size, protection);
3167
3168                         scan = lruvec_size - lruvec_size * protection /
3169                                 (cgroup_size + 1);
3170
3171                         /*
3172                          * Minimally target SWAP_CLUSTER_MAX pages to keep
3173                          * reclaim moving forwards, avoiding decrementing
3174                          * sc->priority further than desirable.
3175                          */
3176                         scan = max(scan, SWAP_CLUSTER_MAX);
3177                 } else {
3178                         scan = lruvec_size;
3179                 }
3180
3181                 scan >>= sc->priority;
3182
3183                 /*
3184                  * If the cgroup's already been deleted, make sure to
3185                  * scrape out the remaining cache.
3186                  */
3187                 if (!scan && !mem_cgroup_online(memcg))
3188                         scan = min(lruvec_size, SWAP_CLUSTER_MAX);
3189
3190                 switch (scan_balance) {
3191                 case SCAN_EQUAL:
3192                         /* Scan lists relative to size */
3193                         break;
3194                 case SCAN_FRACT:
3195                         /*
3196                          * Scan types proportional to swappiness and
3197                          * their relative recent reclaim efficiency.
3198                          * Make sure we don't miss the last page on
3199                          * the offlined memory cgroups because of a
3200                          * round-off error.
3201                          */
3202                         scan = mem_cgroup_online(memcg) ?
3203                                div64_u64(scan * fraction[file], denominator) :
3204                                DIV64_U64_ROUND_UP(scan * fraction[file],
3205                                                   denominator);
3206                         break;
3207                 case SCAN_FILE:
3208                 case SCAN_ANON:
3209                         /* Scan one type exclusively */
3210                         if ((scan_balance == SCAN_FILE) != file)
3211                                 scan = 0;
3212                         break;
3213                 default:
3214                         /* Look ma, no brain */
3215                         BUG();
3216                 }
3217
3218                 nr[lru] = scan;
3219         }
3220 }
3221
3222 /*
3223  * Anonymous LRU management is a waste if there is
3224  * ultimately no way to reclaim the memory.
3225  */
3226 static bool can_age_anon_pages(struct pglist_data *pgdat,
3227                                struct scan_control *sc)
3228 {
3229         /* Aging the anon LRU is valuable if swap is present: */
3230         if (total_swap_pages > 0)
3231                 return true;
3232
3233         /* Also valuable if anon pages can be demoted: */
3234         return can_demote(pgdat->node_id, sc);
3235 }
3236
3237 #ifdef CONFIG_LRU_GEN
3238
3239 #ifdef CONFIG_LRU_GEN_ENABLED
3240 DEFINE_STATIC_KEY_ARRAY_TRUE(lru_gen_caps, NR_LRU_GEN_CAPS);
3241 #define get_cap(cap)    static_branch_likely(&lru_gen_caps[cap])
3242 #else
3243 DEFINE_STATIC_KEY_ARRAY_FALSE(lru_gen_caps, NR_LRU_GEN_CAPS);
3244 #define get_cap(cap)    static_branch_unlikely(&lru_gen_caps[cap])
3245 #endif
3246
3247 static bool should_walk_mmu(void)
3248 {
3249         return arch_has_hw_pte_young() && get_cap(LRU_GEN_MM_WALK);
3250 }
3251
3252 static bool should_clear_pmd_young(void)
3253 {
3254         return arch_has_hw_nonleaf_pmd_young() && get_cap(LRU_GEN_NONLEAF_YOUNG);
3255 }
3256
3257 /******************************************************************************
3258  *                          shorthand helpers
3259  ******************************************************************************/
3260
3261 #define LRU_REFS_FLAGS  (BIT(PG_referenced) | BIT(PG_workingset))
3262
3263 #define DEFINE_MAX_SEQ(lruvec)                                          \
3264         unsigned long max_seq = READ_ONCE((lruvec)->lrugen.max_seq)
3265
3266 #define DEFINE_MIN_SEQ(lruvec)                                          \
3267         unsigned long min_seq[ANON_AND_FILE] = {                        \
3268                 READ_ONCE((lruvec)->lrugen.min_seq[LRU_GEN_ANON]),      \
3269                 READ_ONCE((lruvec)->lrugen.min_seq[LRU_GEN_FILE]),      \
3270         }
3271
3272 #define for_each_gen_type_zone(gen, type, zone)                         \
3273         for ((gen) = 0; (gen) < MAX_NR_GENS; (gen)++)                   \
3274                 for ((type) = 0; (type) < ANON_AND_FILE; (type)++)      \
3275                         for ((zone) = 0; (zone) < MAX_NR_ZONES; (zone)++)
3276
3277 #define get_memcg_gen(seq)      ((seq) % MEMCG_NR_GENS)
3278 #define get_memcg_bin(bin)      ((bin) % MEMCG_NR_BINS)
3279
3280 static struct lruvec *get_lruvec(struct mem_cgroup *memcg, int nid)
3281 {
3282         struct pglist_data *pgdat = NODE_DATA(nid);
3283
3284 #ifdef CONFIG_MEMCG
3285         if (memcg) {
3286                 struct lruvec *lruvec = &memcg->nodeinfo[nid]->lruvec;
3287
3288                 /* see the comment in mem_cgroup_lruvec() */
3289                 if (!lruvec->pgdat)
3290                         lruvec->pgdat = pgdat;
3291
3292                 return lruvec;
3293         }
3294 #endif
3295         VM_WARN_ON_ONCE(!mem_cgroup_disabled());
3296
3297         return &pgdat->__lruvec;
3298 }
3299
3300 static int get_swappiness(struct lruvec *lruvec, struct scan_control *sc)
3301 {
3302         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
3303         struct pglist_data *pgdat = lruvec_pgdat(lruvec);
3304
3305         if (!sc->may_swap)
3306                 return 0;
3307
3308         if (!can_demote(pgdat->node_id, sc) &&
3309             mem_cgroup_get_nr_swap_pages(memcg) < MIN_LRU_BATCH)
3310                 return 0;
3311
3312         return mem_cgroup_swappiness(memcg);
3313 }
3314
3315 static int get_nr_gens(struct lruvec *lruvec, int type)
3316 {
3317         return lruvec->lrugen.max_seq - lruvec->lrugen.min_seq[type] + 1;
3318 }
3319
3320 static bool __maybe_unused seq_is_valid(struct lruvec *lruvec)
3321 {
3322         /* see the comment on lru_gen_folio */
3323         return get_nr_gens(lruvec, LRU_GEN_FILE) >= MIN_NR_GENS &&
3324                get_nr_gens(lruvec, LRU_GEN_FILE) <= get_nr_gens(lruvec, LRU_GEN_ANON) &&
3325                get_nr_gens(lruvec, LRU_GEN_ANON) <= MAX_NR_GENS;
3326 }
3327
3328 /******************************************************************************
3329  *                          Bloom filters
3330  ******************************************************************************/
3331
3332 /*
3333  * Bloom filters with m=1<<15, k=2 and the false positive rates of ~1/5 when
3334  * n=10,000 and ~1/2 when n=20,000, where, conventionally, m is the number of
3335  * bits in a bitmap, k is the number of hash functions and n is the number of
3336  * inserted items.
3337  *
3338  * Page table walkers use one of the two filters to reduce their search space.
3339  * To get rid of non-leaf entries that no longer have enough leaf entries, the
3340  * aging uses the double-buffering technique to flip to the other filter each
3341  * time it produces a new generation. For non-leaf entries that have enough
3342  * leaf entries, the aging carries them over to the next generation in
3343  * walk_pmd_range(); the eviction also report them when walking the rmap
3344  * in lru_gen_look_around().
3345  *
3346  * For future optimizations:
3347  * 1. It's not necessary to keep both filters all the time. The spare one can be
3348  *    freed after the RCU grace period and reallocated if needed again.
3349  * 2. And when reallocating, it's worth scaling its size according to the number
3350  *    of inserted entries in the other filter, to reduce the memory overhead on
3351  *    small systems and false positives on large systems.
3352  * 3. Jenkins' hash function is an alternative to Knuth's.
3353  */
3354 #define BLOOM_FILTER_SHIFT      15
3355
3356 static inline int filter_gen_from_seq(unsigned long seq)
3357 {
3358         return seq % NR_BLOOM_FILTERS;
3359 }
3360
3361 static void get_item_key(void *item, int *key)
3362 {
3363         u32 hash = hash_ptr(item, BLOOM_FILTER_SHIFT * 2);
3364
3365         BUILD_BUG_ON(BLOOM_FILTER_SHIFT * 2 > BITS_PER_TYPE(u32));
3366
3367         key[0] = hash & (BIT(BLOOM_FILTER_SHIFT) - 1);
3368         key[1] = hash >> BLOOM_FILTER_SHIFT;
3369 }
3370
3371 static bool test_bloom_filter(struct lruvec *lruvec, unsigned long seq, void *item)
3372 {
3373         int key[2];
3374         unsigned long *filter;
3375         int gen = filter_gen_from_seq(seq);
3376
3377         filter = READ_ONCE(lruvec->mm_state.filters[gen]);
3378         if (!filter)
3379                 return true;
3380
3381         get_item_key(item, key);
3382
3383         return test_bit(key[0], filter) && test_bit(key[1], filter);
3384 }
3385
3386 static void update_bloom_filter(struct lruvec *lruvec, unsigned long seq, void *item)
3387 {
3388         int key[2];
3389         unsigned long *filter;
3390         int gen = filter_gen_from_seq(seq);
3391
3392         filter = READ_ONCE(lruvec->mm_state.filters[gen]);
3393         if (!filter)
3394                 return;
3395
3396         get_item_key(item, key);
3397
3398         if (!test_bit(key[0], filter))
3399                 set_bit(key[0], filter);
3400         if (!test_bit(key[1], filter))
3401                 set_bit(key[1], filter);
3402 }
3403
3404 static void reset_bloom_filter(struct lruvec *lruvec, unsigned long seq)
3405 {
3406         unsigned long *filter;
3407         int gen = filter_gen_from_seq(seq);
3408
3409         filter = lruvec->mm_state.filters[gen];
3410         if (filter) {
3411                 bitmap_clear(filter, 0, BIT(BLOOM_FILTER_SHIFT));
3412                 return;
3413         }
3414
3415         filter = bitmap_zalloc(BIT(BLOOM_FILTER_SHIFT),
3416                                __GFP_HIGH | __GFP_NOMEMALLOC | __GFP_NOWARN);
3417         WRITE_ONCE(lruvec->mm_state.filters[gen], filter);
3418 }
3419
3420 /******************************************************************************
3421  *                          mm_struct list
3422  ******************************************************************************/
3423
3424 static struct lru_gen_mm_list *get_mm_list(struct mem_cgroup *memcg)
3425 {
3426         static struct lru_gen_mm_list mm_list = {
3427                 .fifo = LIST_HEAD_INIT(mm_list.fifo),
3428                 .lock = __SPIN_LOCK_UNLOCKED(mm_list.lock),
3429         };
3430
3431 #ifdef CONFIG_MEMCG
3432         if (memcg)
3433                 return &memcg->mm_list;
3434 #endif
3435         VM_WARN_ON_ONCE(!mem_cgroup_disabled());
3436
3437         return &mm_list;
3438 }
3439
3440 void lru_gen_add_mm(struct mm_struct *mm)
3441 {
3442         int nid;
3443         struct mem_cgroup *memcg = get_mem_cgroup_from_mm(mm);
3444         struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
3445
3446         VM_WARN_ON_ONCE(!list_empty(&mm->lru_gen.list));
3447 #ifdef CONFIG_MEMCG
3448         VM_WARN_ON_ONCE(mm->lru_gen.memcg);
3449         mm->lru_gen.memcg = memcg;
3450 #endif
3451         spin_lock(&mm_list->lock);
3452
3453         for_each_node_state(nid, N_MEMORY) {
3454                 struct lruvec *lruvec = get_lruvec(memcg, nid);
3455
3456                 /* the first addition since the last iteration */
3457                 if (lruvec->mm_state.tail == &mm_list->fifo)
3458                         lruvec->mm_state.tail = &mm->lru_gen.list;
3459         }
3460
3461         list_add_tail(&mm->lru_gen.list, &mm_list->fifo);
3462
3463         spin_unlock(&mm_list->lock);
3464 }
3465
3466 void lru_gen_del_mm(struct mm_struct *mm)
3467 {
3468         int nid;
3469         struct lru_gen_mm_list *mm_list;
3470         struct mem_cgroup *memcg = NULL;
3471
3472         if (list_empty(&mm->lru_gen.list))
3473                 return;
3474
3475 #ifdef CONFIG_MEMCG
3476         memcg = mm->lru_gen.memcg;
3477 #endif
3478         mm_list = get_mm_list(memcg);
3479
3480         spin_lock(&mm_list->lock);
3481
3482         for_each_node(nid) {
3483                 struct lruvec *lruvec = get_lruvec(memcg, nid);
3484
3485                 /* where the current iteration continues after */
3486                 if (lruvec->mm_state.head == &mm->lru_gen.list)
3487                         lruvec->mm_state.head = lruvec->mm_state.head->prev;
3488
3489                 /* where the last iteration ended before */
3490                 if (lruvec->mm_state.tail == &mm->lru_gen.list)
3491                         lruvec->mm_state.tail = lruvec->mm_state.tail->next;
3492         }
3493
3494         list_del_init(&mm->lru_gen.list);
3495
3496         spin_unlock(&mm_list->lock);
3497
3498 #ifdef CONFIG_MEMCG
3499         mem_cgroup_put(mm->lru_gen.memcg);
3500         mm->lru_gen.memcg = NULL;
3501 #endif
3502 }
3503
3504 #ifdef CONFIG_MEMCG
3505 void lru_gen_migrate_mm(struct mm_struct *mm)
3506 {
3507         struct mem_cgroup *memcg;
3508         struct task_struct *task = rcu_dereference_protected(mm->owner, true);
3509
3510         VM_WARN_ON_ONCE(task->mm != mm);
3511         lockdep_assert_held(&task->alloc_lock);
3512
3513         /* for mm_update_next_owner() */
3514         if (mem_cgroup_disabled())
3515                 return;
3516
3517         /* migration can happen before addition */
3518         if (!mm->lru_gen.memcg)
3519                 return;
3520
3521         rcu_read_lock();
3522         memcg = mem_cgroup_from_task(task);
3523         rcu_read_unlock();
3524         if (memcg == mm->lru_gen.memcg)
3525                 return;
3526
3527         VM_WARN_ON_ONCE(list_empty(&mm->lru_gen.list));
3528
3529         lru_gen_del_mm(mm);
3530         lru_gen_add_mm(mm);
3531 }
3532 #endif
3533
3534 static void reset_mm_stats(struct lruvec *lruvec, struct lru_gen_mm_walk *walk, bool last)
3535 {
3536         int i;
3537         int hist;
3538
3539         lockdep_assert_held(&get_mm_list(lruvec_memcg(lruvec))->lock);
3540
3541         if (walk) {
3542                 hist = lru_hist_from_seq(walk->max_seq);
3543
3544                 for (i = 0; i < NR_MM_STATS; i++) {
3545                         WRITE_ONCE(lruvec->mm_state.stats[hist][i],
3546                                    lruvec->mm_state.stats[hist][i] + walk->mm_stats[i]);
3547                         walk->mm_stats[i] = 0;
3548                 }
3549         }
3550
3551         if (NR_HIST_GENS > 1 && last) {
3552                 hist = lru_hist_from_seq(lruvec->mm_state.seq + 1);
3553
3554                 for (i = 0; i < NR_MM_STATS; i++)
3555                         WRITE_ONCE(lruvec->mm_state.stats[hist][i], 0);
3556         }
3557 }
3558
3559 static bool should_skip_mm(struct mm_struct *mm, struct lru_gen_mm_walk *walk)
3560 {
3561         int type;
3562         unsigned long size = 0;
3563         struct pglist_data *pgdat = lruvec_pgdat(walk->lruvec);
3564         int key = pgdat->node_id % BITS_PER_TYPE(mm->lru_gen.bitmap);
3565
3566         if (!walk->force_scan && !test_bit(key, &mm->lru_gen.bitmap))
3567                 return true;
3568
3569         clear_bit(key, &mm->lru_gen.bitmap);
3570
3571         for (type = !walk->can_swap; type < ANON_AND_FILE; type++) {
3572                 size += type ? get_mm_counter(mm, MM_FILEPAGES) :
3573                                get_mm_counter(mm, MM_ANONPAGES) +
3574                                get_mm_counter(mm, MM_SHMEMPAGES);
3575         }
3576
3577         if (size < MIN_LRU_BATCH)
3578                 return true;
3579
3580         return !mmget_not_zero(mm);
3581 }
3582
3583 static bool iterate_mm_list(struct lruvec *lruvec, struct lru_gen_mm_walk *walk,
3584                             struct mm_struct **iter)
3585 {
3586         bool first = false;
3587         bool last = false;
3588         struct mm_struct *mm = NULL;
3589         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
3590         struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
3591         struct lru_gen_mm_state *mm_state = &lruvec->mm_state;
3592
3593         /*
3594          * mm_state->seq is incremented after each iteration of mm_list. There
3595          * are three interesting cases for this page table walker:
3596          * 1. It tries to start a new iteration with a stale max_seq: there is
3597          *    nothing left to do.
3598          * 2. It started the next iteration: it needs to reset the Bloom filter
3599          *    so that a fresh set of PTE tables can be recorded.
3600          * 3. It ended the current iteration: it needs to reset the mm stats
3601          *    counters and tell its caller to increment max_seq.
3602          */
3603         spin_lock(&mm_list->lock);
3604
3605         VM_WARN_ON_ONCE(mm_state->seq + 1 < walk->max_seq);
3606
3607         if (walk->max_seq <= mm_state->seq)
3608                 goto done;
3609
3610         if (!mm_state->head)
3611                 mm_state->head = &mm_list->fifo;
3612
3613         if (mm_state->head == &mm_list->fifo)
3614                 first = true;
3615
3616         do {
3617                 mm_state->head = mm_state->head->next;
3618                 if (mm_state->head == &mm_list->fifo) {
3619                         WRITE_ONCE(mm_state->seq, mm_state->seq + 1);
3620                         last = true;
3621                         break;
3622                 }
3623
3624                 /* force scan for those added after the last iteration */
3625                 if (!mm_state->tail || mm_state->tail == mm_state->head) {
3626                         mm_state->tail = mm_state->head->next;
3627                         walk->force_scan = true;
3628                 }
3629
3630                 mm = list_entry(mm_state->head, struct mm_struct, lru_gen.list);
3631                 if (should_skip_mm(mm, walk))
3632                         mm = NULL;
3633         } while (!mm);
3634 done:
3635         if (*iter || last)
3636                 reset_mm_stats(lruvec, walk, last);
3637
3638         spin_unlock(&mm_list->lock);
3639
3640         if (mm && first)
3641                 reset_bloom_filter(lruvec, walk->max_seq + 1);
3642
3643         if (*iter)
3644                 mmput_async(*iter);
3645
3646         *iter = mm;
3647
3648         return last;
3649 }
3650
3651 static bool iterate_mm_list_nowalk(struct lruvec *lruvec, unsigned long max_seq)
3652 {
3653         bool success = false;
3654         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
3655         struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
3656         struct lru_gen_mm_state *mm_state = &lruvec->mm_state;
3657
3658         spin_lock(&mm_list->lock);
3659
3660         VM_WARN_ON_ONCE(mm_state->seq + 1 < max_seq);
3661
3662         if (max_seq > mm_state->seq) {
3663                 mm_state->head = NULL;
3664                 mm_state->tail = NULL;
3665                 WRITE_ONCE(mm_state->seq, mm_state->seq + 1);
3666                 reset_mm_stats(lruvec, NULL, true);
3667                 success = true;
3668         }
3669
3670         spin_unlock(&mm_list->lock);
3671
3672         return success;
3673 }
3674
3675 /******************************************************************************
3676  *                          PID controller
3677  ******************************************************************************/
3678
3679 /*
3680  * A feedback loop based on Proportional-Integral-Derivative (PID) controller.
3681  *
3682  * The P term is refaulted/(evicted+protected) from a tier in the generation
3683  * currently being evicted; the I term is the exponential moving average of the
3684  * P term over the generations previously evicted, using the smoothing factor
3685  * 1/2; the D term isn't supported.
3686  *
3687  * The setpoint (SP) is always the first tier of one type; the process variable
3688  * (PV) is either any tier of the other type or any other tier of the same
3689  * type.
3690  *
3691  * The error is the difference between the SP and the PV; the correction is to
3692  * turn off protection when SP>PV or turn on protection when SP<PV.
3693  *
3694  * For future optimizations:
3695  * 1. The D term may discount the other two terms over time so that long-lived
3696  *    generations can resist stale information.
3697  */
3698 struct ctrl_pos {
3699         unsigned long refaulted;
3700         unsigned long total;
3701         int gain;
3702 };
3703
3704 static void read_ctrl_pos(struct lruvec *lruvec, int type, int tier, int gain,
3705                           struct ctrl_pos *pos)
3706 {
3707         struct lru_gen_folio *lrugen = &lruvec->lrugen;
3708         int hist = lru_hist_from_seq(lrugen->min_seq[type]);
3709
3710         pos->refaulted = lrugen->avg_refaulted[type][tier] +
3711                          atomic_long_read(&lrugen->refaulted[hist][type][tier]);
3712         pos->total = lrugen->avg_total[type][tier] +
3713                      atomic_long_read(&lrugen->evicted[hist][type][tier]);
3714         if (tier)
3715                 pos->total += lrugen->protected[hist][type][tier - 1];
3716         pos->gain = gain;
3717 }
3718
3719 static void reset_ctrl_pos(struct lruvec *lruvec, int type, bool carryover)
3720 {
3721         int hist, tier;
3722         struct lru_gen_folio *lrugen = &lruvec->lrugen;
3723         bool clear = carryover ? NR_HIST_GENS == 1 : NR_HIST_GENS > 1;
3724         unsigned long seq = carryover ? lrugen->min_seq[type] : lrugen->max_seq + 1;
3725
3726         lockdep_assert_held(&lruvec->lru_lock);
3727
3728         if (!carryover && !clear)
3729                 return;
3730
3731         hist = lru_hist_from_seq(seq);
3732
3733         for (tier = 0; tier < MAX_NR_TIERS; tier++) {
3734                 if (carryover) {
3735                         unsigned long sum;
3736
3737                         sum = lrugen->avg_refaulted[type][tier] +
3738                               atomic_long_read(&lrugen->refaulted[hist][type][tier]);
3739                         WRITE_ONCE(lrugen->avg_refaulted[type][tier], sum / 2);
3740
3741                         sum = lrugen->avg_total[type][tier] +
3742                               atomic_long_read(&lrugen->evicted[hist][type][tier]);
3743                         if (tier)
3744                                 sum += lrugen->protected[hist][type][tier - 1];
3745                         WRITE_ONCE(lrugen->avg_total[type][tier], sum / 2);
3746                 }
3747
3748                 if (clear) {
3749                         atomic_long_set(&lrugen->refaulted[hist][type][tier], 0);
3750                         atomic_long_set(&lrugen->evicted[hist][type][tier], 0);
3751                         if (tier)
3752                                 WRITE_ONCE(lrugen->protected[hist][type][tier - 1], 0);
3753                 }
3754         }
3755 }
3756
3757 static bool positive_ctrl_err(struct ctrl_pos *sp, struct ctrl_pos *pv)
3758 {
3759         /*
3760          * Return true if the PV has a limited number of refaults or a lower
3761          * refaulted/total than the SP.
3762          */
3763         return pv->refaulted < MIN_LRU_BATCH ||
3764                pv->refaulted * (sp->total + MIN_LRU_BATCH) * sp->gain <=
3765                (sp->refaulted + 1) * pv->total * pv->gain;
3766 }
3767
3768 /******************************************************************************
3769  *                          the aging
3770  ******************************************************************************/
3771
3772 /* promote pages accessed through page tables */
3773 static int folio_update_gen(struct folio *folio, int gen)
3774 {
3775         unsigned long new_flags, old_flags = READ_ONCE(folio->flags);
3776
3777         VM_WARN_ON_ONCE(gen >= MAX_NR_GENS);
3778         VM_WARN_ON_ONCE(!rcu_read_lock_held());
3779
3780         do {
3781                 /* lru_gen_del_folio() has isolated this page? */
3782                 if (!(old_flags & LRU_GEN_MASK)) {
3783                         /* for shrink_folio_list() */
3784                         new_flags = old_flags | BIT(PG_referenced);
3785                         continue;
3786                 }
3787
3788                 new_flags = old_flags & ~(LRU_GEN_MASK | LRU_REFS_MASK | LRU_REFS_FLAGS);
3789                 new_flags |= (gen + 1UL) << LRU_GEN_PGOFF;
3790         } while (!try_cmpxchg(&folio->flags, &old_flags, new_flags));
3791
3792         return ((old_flags & LRU_GEN_MASK) >> LRU_GEN_PGOFF) - 1;
3793 }
3794
3795 /* protect pages accessed multiple times through file descriptors */
3796 static int folio_inc_gen(struct lruvec *lruvec, struct folio *folio, bool reclaiming)
3797 {
3798         int type = folio_is_file_lru(folio);
3799         struct lru_gen_folio *lrugen = &lruvec->lrugen;
3800         int new_gen, old_gen = lru_gen_from_seq(lrugen->min_seq[type]);
3801         unsigned long new_flags, old_flags = READ_ONCE(folio->flags);
3802
3803         VM_WARN_ON_ONCE_FOLIO(!(old_flags & LRU_GEN_MASK), folio);
3804
3805         do {
3806                 new_gen = ((old_flags & LRU_GEN_MASK) >> LRU_GEN_PGOFF) - 1;
3807                 /* folio_update_gen() has promoted this page? */
3808                 if (new_gen >= 0 && new_gen != old_gen)
3809                         return new_gen;
3810
3811                 new_gen = (old_gen + 1) % MAX_NR_GENS;
3812
3813                 new_flags = old_flags & ~(LRU_GEN_MASK | LRU_REFS_MASK | LRU_REFS_FLAGS);
3814                 new_flags |= (new_gen + 1UL) << LRU_GEN_PGOFF;
3815                 /* for folio_end_writeback() */
3816                 if (reclaiming)
3817                         new_flags |= BIT(PG_reclaim);
3818         } while (!try_cmpxchg(&folio->flags, &old_flags, new_flags));
3819
3820         lru_gen_update_size(lruvec, folio, old_gen, new_gen);
3821
3822         return new_gen;
3823 }
3824
3825 static void update_batch_size(struct lru_gen_mm_walk *walk, struct folio *folio,
3826                               int old_gen, int new_gen)
3827 {
3828         int type = folio_is_file_lru(folio);
3829         int zone = folio_zonenum(folio);
3830         int delta = folio_nr_pages(folio);
3831
3832         VM_WARN_ON_ONCE(old_gen >= MAX_NR_GENS);
3833         VM_WARN_ON_ONCE(new_gen >= MAX_NR_GENS);
3834
3835         walk->batched++;
3836
3837         walk->nr_pages[old_gen][type][zone] -= delta;
3838         walk->nr_pages[new_gen][type][zone] += delta;
3839 }
3840
3841 static void reset_batch_size(struct lruvec *lruvec, struct lru_gen_mm_walk *walk)
3842 {
3843         int gen, type, zone;
3844         struct lru_gen_folio *lrugen = &lruvec->lrugen;
3845
3846         walk->batched = 0;
3847
3848         for_each_gen_type_zone(gen, type, zone) {
3849                 enum lru_list lru = type * LRU_INACTIVE_FILE;
3850                 int delta = walk->nr_pages[gen][type][zone];
3851
3852                 if (!delta)
3853                         continue;
3854
3855                 walk->nr_pages[gen][type][zone] = 0;
3856                 WRITE_ONCE(lrugen->nr_pages[gen][type][zone],
3857                            lrugen->nr_pages[gen][type][zone] + delta);
3858
3859                 if (lru_gen_is_active(lruvec, gen))
3860                         lru += LRU_ACTIVE;
3861                 __update_lru_size(lruvec, lru, zone, delta);
3862         }
3863 }
3864
3865 static int should_skip_vma(unsigned long start, unsigned long end, struct mm_walk *args)
3866 {
3867         struct address_space *mapping;
3868         struct vm_area_struct *vma = args->vma;
3869         struct lru_gen_mm_walk *walk = args->private;
3870
3871         if (!vma_is_accessible(vma))
3872                 return true;
3873
3874         if (is_vm_hugetlb_page(vma))
3875                 return true;
3876
3877         if (!vma_has_recency(vma))
3878                 return true;
3879
3880         if (vma->vm_flags & (VM_LOCKED | VM_SPECIAL))
3881                 return true;
3882
3883         if (vma == get_gate_vma(vma->vm_mm))
3884                 return true;
3885
3886         if (vma_is_anonymous(vma))
3887                 return !walk->can_swap;
3888
3889         if (WARN_ON_ONCE(!vma->vm_file || !vma->vm_file->f_mapping))
3890                 return true;
3891
3892         mapping = vma->vm_file->f_mapping;
3893         if (mapping_unevictable(mapping))
3894                 return true;
3895
3896         if (shmem_mapping(mapping))
3897                 return !walk->can_swap;
3898
3899         /* to exclude special mappings like dax, etc. */
3900         return !mapping->a_ops->read_folio;
3901 }
3902
3903 /*
3904  * Some userspace memory allocators map many single-page VMAs. Instead of
3905  * returning back to the PGD table for each of such VMAs, finish an entire PMD
3906  * table to reduce zigzags and improve cache performance.
3907  */
3908 static bool get_next_vma(unsigned long mask, unsigned long size, struct mm_walk *args,
3909                          unsigned long *vm_start, unsigned long *vm_end)
3910 {
3911         unsigned long start = round_up(*vm_end, size);
3912         unsigned long end = (start | ~mask) + 1;
3913         VMA_ITERATOR(vmi, args->mm, start);
3914
3915         VM_WARN_ON_ONCE(mask & size);
3916         VM_WARN_ON_ONCE((start & mask) != (*vm_start & mask));
3917
3918         for_each_vma(vmi, args->vma) {
3919                 if (end && end <= args->vma->vm_start)
3920                         return false;
3921
3922                 if (should_skip_vma(args->vma->vm_start, args->vma->vm_end, args))
3923                         continue;
3924
3925                 *vm_start = max(start, args->vma->vm_start);
3926                 *vm_end = min(end - 1, args->vma->vm_end - 1) + 1;
3927
3928                 return true;
3929         }
3930
3931         return false;
3932 }
3933
3934 static unsigned long get_pte_pfn(pte_t pte, struct vm_area_struct *vma, unsigned long addr)
3935 {
3936         unsigned long pfn = pte_pfn(pte);
3937
3938         VM_WARN_ON_ONCE(addr < vma->vm_start || addr >= vma->vm_end);
3939
3940         if (!pte_present(pte) || is_zero_pfn(pfn))
3941                 return -1;
3942
3943         if (WARN_ON_ONCE(pte_devmap(pte) || pte_special(pte)))
3944                 return -1;
3945
3946         if (WARN_ON_ONCE(!pfn_valid(pfn)))
3947                 return -1;
3948
3949         return pfn;
3950 }
3951
3952 #if defined(CONFIG_TRANSPARENT_HUGEPAGE) || defined(CONFIG_ARCH_HAS_NONLEAF_PMD_YOUNG)
3953 static unsigned long get_pmd_pfn(pmd_t pmd, struct vm_area_struct *vma, unsigned long addr)
3954 {
3955         unsigned long pfn = pmd_pfn(pmd);
3956
3957         VM_WARN_ON_ONCE(addr < vma->vm_start || addr >= vma->vm_end);
3958
3959         if (!pmd_present(pmd) || is_huge_zero_pmd(pmd))
3960                 return -1;
3961
3962         if (WARN_ON_ONCE(pmd_devmap(pmd)))
3963                 return -1;
3964
3965         if (WARN_ON_ONCE(!pfn_valid(pfn)))
3966                 return -1;
3967
3968         return pfn;
3969 }
3970 #endif
3971
3972 static struct folio *get_pfn_folio(unsigned long pfn, struct mem_cgroup *memcg,
3973                                    struct pglist_data *pgdat, bool can_swap)
3974 {
3975         struct folio *folio;
3976
3977         /* try to avoid unnecessary memory loads */
3978         if (pfn < pgdat->node_start_pfn || pfn >= pgdat_end_pfn(pgdat))
3979                 return NULL;
3980
3981         folio = pfn_folio(pfn);
3982         if (folio_nid(folio) != pgdat->node_id)
3983                 return NULL;
3984
3985         if (folio_memcg_rcu(folio) != memcg)
3986                 return NULL;
3987
3988         /* file VMAs can contain anon pages from COW */
3989         if (!folio_is_file_lru(folio) && !can_swap)
3990                 return NULL;
3991
3992         return folio;
3993 }
3994
3995 static bool suitable_to_scan(int total, int young)
3996 {
3997         int n = clamp_t(int, cache_line_size() / sizeof(pte_t), 2, 8);
3998
3999         /* suitable if the average number of young PTEs per cacheline is >=1 */
4000         return young * n >= total;
4001 }
4002
4003 static bool walk_pte_range(pmd_t *pmd, unsigned long start, unsigned long end,
4004                            struct mm_walk *args)
4005 {
4006         int i;
4007         pte_t *pte;
4008         spinlock_t *ptl;
4009         unsigned long addr;
4010         int total = 0;
4011         int young = 0;
4012         struct lru_gen_mm_walk *walk = args->private;
4013         struct mem_cgroup *memcg = lruvec_memcg(walk->lruvec);
4014         struct pglist_data *pgdat = lruvec_pgdat(walk->lruvec);
4015         int old_gen, new_gen = lru_gen_from_seq(walk->max_seq);
4016
4017         pte = pte_offset_map_nolock(args->mm, pmd, start & PMD_MASK, &ptl);
4018         if (!pte)
4019                 return false;
4020         if (!spin_trylock(ptl)) {
4021                 pte_unmap(pte);
4022                 return false;
4023         }
4024
4025         arch_enter_lazy_mmu_mode();
4026 restart:
4027         for (i = pte_index(start), addr = start; addr != end; i++, addr += PAGE_SIZE) {
4028                 unsigned long pfn;
4029                 struct folio *folio;
4030                 pte_t ptent = ptep_get(pte + i);
4031
4032                 total++;
4033                 walk->mm_stats[MM_LEAF_TOTAL]++;
4034
4035                 pfn = get_pte_pfn(ptent, args->vma, addr);
4036                 if (pfn == -1)
4037                         continue;
4038
4039                 if (!pte_young(ptent)) {
4040                         walk->mm_stats[MM_LEAF_OLD]++;
4041                         continue;
4042                 }
4043
4044                 folio = get_pfn_folio(pfn, memcg, pgdat, walk->can_swap);
4045                 if (!folio)
4046                         continue;
4047
4048                 if (!ptep_test_and_clear_young(args->vma, addr, pte + i))
4049                         VM_WARN_ON_ONCE(true);
4050
4051                 young++;
4052                 walk->mm_stats[MM_LEAF_YOUNG]++;
4053
4054                 if (pte_dirty(ptent) && !folio_test_dirty(folio) &&
4055                     !(folio_test_anon(folio) && folio_test_swapbacked(folio) &&
4056                       !folio_test_swapcache(folio)))
4057                         folio_mark_dirty(folio);
4058
4059                 old_gen = folio_update_gen(folio, new_gen);
4060                 if (old_gen >= 0 && old_gen != new_gen)
4061                         update_batch_size(walk, folio, old_gen, new_gen);
4062         }
4063
4064         if (i < PTRS_PER_PTE && get_next_vma(PMD_MASK, PAGE_SIZE, args, &start, &end))
4065                 goto restart;
4066
4067         arch_leave_lazy_mmu_mode();
4068         pte_unmap_unlock(pte, ptl);
4069
4070         return suitable_to_scan(total, young);
4071 }
4072
4073 #if defined(CONFIG_TRANSPARENT_HUGEPAGE) || defined(CONFIG_ARCH_HAS_NONLEAF_PMD_YOUNG)
4074 static void walk_pmd_range_locked(pud_t *pud, unsigned long addr, struct vm_area_struct *vma,
4075                                   struct mm_walk *args, unsigned long *bitmap, unsigned long *first)
4076 {
4077         int i;
4078         pmd_t *pmd;
4079         spinlock_t *ptl;
4080         struct lru_gen_mm_walk *walk = args->private;
4081         struct mem_cgroup *memcg = lruvec_memcg(walk->lruvec);
4082         struct pglist_data *pgdat = lruvec_pgdat(walk->lruvec);
4083         int old_gen, new_gen = lru_gen_from_seq(walk->max_seq);
4084
4085         VM_WARN_ON_ONCE(pud_leaf(*pud));
4086
4087         /* try to batch at most 1+MIN_LRU_BATCH+1 entries */
4088         if (*first == -1) {
4089                 *first = addr;
4090                 bitmap_zero(bitmap, MIN_LRU_BATCH);
4091                 return;
4092         }
4093
4094         i = addr == -1 ? 0 : pmd_index(addr) - pmd_index(*first);
4095         if (i && i <= MIN_LRU_BATCH) {
4096                 __set_bit(i - 1, bitmap);
4097                 return;
4098         }
4099
4100         pmd = pmd_offset(pud, *first);
4101
4102         ptl = pmd_lockptr(args->mm, pmd);
4103         if (!spin_trylock(ptl))
4104                 goto done;
4105
4106         arch_enter_lazy_mmu_mode();
4107
4108         do {
4109                 unsigned long pfn;
4110                 struct folio *folio;
4111
4112                 /* don't round down the first address */
4113                 addr = i ? (*first & PMD_MASK) + i * PMD_SIZE : *first;
4114
4115                 pfn = get_pmd_pfn(pmd[i], vma, addr);
4116                 if (pfn == -1)
4117                         goto next;
4118
4119                 if (!pmd_trans_huge(pmd[i])) {
4120                         if (should_clear_pmd_young())
4121                                 pmdp_test_and_clear_young(vma, addr, pmd + i);
4122                         goto next;
4123                 }
4124
4125                 folio = get_pfn_folio(pfn, memcg, pgdat, walk->can_swap);
4126                 if (!folio)
4127                         goto next;
4128
4129                 if (!pmdp_test_and_clear_young(vma, addr, pmd + i))
4130                         goto next;
4131
4132                 walk->mm_stats[MM_LEAF_YOUNG]++;
4133
4134                 if (pmd_dirty(pmd[i]) && !folio_test_dirty(folio) &&
4135                     !(folio_test_anon(folio) && folio_test_swapbacked(folio) &&
4136                       !folio_test_swapcache(folio)))
4137                         folio_mark_dirty(folio);
4138
4139                 old_gen = folio_update_gen(folio, new_gen);
4140                 if (old_gen >= 0 && old_gen != new_gen)
4141                         update_batch_size(walk, folio, old_gen, new_gen);
4142 next:
4143                 i = i > MIN_LRU_BATCH ? 0 : find_next_bit(bitmap, MIN_LRU_BATCH, i) + 1;
4144         } while (i <= MIN_LRU_BATCH);
4145
4146         arch_leave_lazy_mmu_mode();
4147         spin_unlock(ptl);
4148 done:
4149         *first = -1;
4150 }
4151 #else
4152 static void walk_pmd_range_locked(pud_t *pud, unsigned long addr, struct vm_area_struct *vma,
4153                                   struct mm_walk *args, unsigned long *bitmap, unsigned long *first)
4154 {
4155 }
4156 #endif
4157
4158 static void walk_pmd_range(pud_t *pud, unsigned long start, unsigned long end,
4159                            struct mm_walk *args)
4160 {
4161         int i;
4162         pmd_t *pmd;
4163         unsigned long next;
4164         unsigned long addr;
4165         struct vm_area_struct *vma;
4166         DECLARE_BITMAP(bitmap, MIN_LRU_BATCH);
4167         unsigned long first = -1;
4168         struct lru_gen_mm_walk *walk = args->private;
4169
4170         VM_WARN_ON_ONCE(pud_leaf(*pud));
4171
4172         /*
4173          * Finish an entire PMD in two passes: the first only reaches to PTE
4174          * tables to avoid taking the PMD lock; the second, if necessary, takes
4175          * the PMD lock to clear the accessed bit in PMD entries.
4176          */
4177         pmd = pmd_offset(pud, start & PUD_MASK);
4178 restart:
4179         /* walk_pte_range() may call get_next_vma() */
4180         vma = args->vma;
4181         for (i = pmd_index(start), addr = start; addr != end; i++, addr = next) {
4182                 pmd_t val = pmdp_get_lockless(pmd + i);
4183
4184                 next = pmd_addr_end(addr, end);
4185
4186                 if (!pmd_present(val) || is_huge_zero_pmd(val)) {
4187                         walk->mm_stats[MM_LEAF_TOTAL]++;
4188                         continue;
4189                 }
4190
4191 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
4192                 if (pmd_trans_huge(val)) {
4193                         unsigned long pfn = pmd_pfn(val);
4194                         struct pglist_data *pgdat = lruvec_pgdat(walk->lruvec);
4195
4196                         walk->mm_stats[MM_LEAF_TOTAL]++;
4197
4198                         if (!pmd_young(val)) {
4199                                 walk->mm_stats[MM_LEAF_OLD]++;
4200                                 continue;
4201                         }
4202
4203                         /* try to avoid unnecessary memory loads */
4204                         if (pfn < pgdat->node_start_pfn || pfn >= pgdat_end_pfn(pgdat))
4205                                 continue;
4206
4207                         walk_pmd_range_locked(pud, addr, vma, args, bitmap, &first);
4208                         continue;
4209                 }
4210 #endif
4211                 walk->mm_stats[MM_NONLEAF_TOTAL]++;
4212
4213                 if (should_clear_pmd_young()) {
4214                         if (!pmd_young(val))
4215                                 continue;
4216
4217                         walk_pmd_range_locked(pud, addr, vma, args, bitmap, &first);
4218                 }
4219
4220                 if (!walk->force_scan && !test_bloom_filter(walk->lruvec, walk->max_seq, pmd + i))
4221                         continue;
4222
4223                 walk->mm_stats[MM_NONLEAF_FOUND]++;
4224
4225                 if (!walk_pte_range(&val, addr, next, args))
4226                         continue;
4227
4228                 walk->mm_stats[MM_NONLEAF_ADDED]++;
4229
4230                 /* carry over to the next generation */
4231                 update_bloom_filter(walk->lruvec, walk->max_seq + 1, pmd + i);
4232         }
4233
4234         walk_pmd_range_locked(pud, -1, vma, args, bitmap, &first);
4235
4236         if (i < PTRS_PER_PMD && get_next_vma(PUD_MASK, PMD_SIZE, args, &start, &end))
4237                 goto restart;
4238 }
4239
4240 static int walk_pud_range(p4d_t *p4d, unsigned long start, unsigned long end,
4241                           struct mm_walk *args)
4242 {
4243         int i;
4244         pud_t *pud;
4245         unsigned long addr;
4246         unsigned long next;
4247         struct lru_gen_mm_walk *walk = args->private;
4248
4249         VM_WARN_ON_ONCE(p4d_leaf(*p4d));
4250
4251         pud = pud_offset(p4d, start & P4D_MASK);
4252 restart:
4253         for (i = pud_index(start), addr = start; addr != end; i++, addr = next) {
4254                 pud_t val = READ_ONCE(pud[i]);
4255
4256                 next = pud_addr_end(addr, end);
4257
4258                 if (!pud_present(val) || WARN_ON_ONCE(pud_leaf(val)))
4259                         continue;
4260
4261                 walk_pmd_range(&val, addr, next, args);
4262
4263                 if (need_resched() || walk->batched >= MAX_LRU_BATCH) {
4264                         end = (addr | ~PUD_MASK) + 1;
4265                         goto done;
4266                 }
4267         }
4268
4269         if (i < PTRS_PER_PUD && get_next_vma(P4D_MASK, PUD_SIZE, args, &start, &end))
4270                 goto restart;
4271
4272         end = round_up(end, P4D_SIZE);
4273 done:
4274         if (!end || !args->vma)
4275                 return 1;
4276
4277         walk->next_addr = max(end, args->vma->vm_start);
4278
4279         return -EAGAIN;
4280 }
4281
4282 static void walk_mm(struct lruvec *lruvec, struct mm_struct *mm, struct lru_gen_mm_walk *walk)
4283 {
4284         static const struct mm_walk_ops mm_walk_ops = {
4285                 .test_walk = should_skip_vma,
4286                 .p4d_entry = walk_pud_range,
4287                 .walk_lock = PGWALK_RDLOCK,
4288         };
4289
4290         int err;
4291         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
4292
4293         walk->next_addr = FIRST_USER_ADDRESS;
4294
4295         do {
4296                 DEFINE_MAX_SEQ(lruvec);
4297
4298                 err = -EBUSY;
4299
4300                 /* another thread might have called inc_max_seq() */
4301                 if (walk->max_seq != max_seq)
4302                         break;
4303
4304                 /* folio_update_gen() requires stable folio_memcg() */
4305                 if (!mem_cgroup_trylock_pages(memcg))
4306                         break;
4307
4308                 /* the caller might be holding the lock for write */
4309                 if (mmap_read_trylock(mm)) {
4310                         err = walk_page_range(mm, walk->next_addr, ULONG_MAX, &mm_walk_ops, walk);
4311
4312                         mmap_read_unlock(mm);
4313                 }
4314
4315                 mem_cgroup_unlock_pages();
4316
4317                 if (walk->batched) {
4318                         spin_lock_irq(&lruvec->lru_lock);
4319                         reset_batch_size(lruvec, walk);
4320                         spin_unlock_irq(&lruvec->lru_lock);
4321                 }
4322
4323                 cond_resched();
4324         } while (err == -EAGAIN);
4325 }
4326
4327 static struct lru_gen_mm_walk *set_mm_walk(struct pglist_data *pgdat, bool force_alloc)
4328 {
4329         struct lru_gen_mm_walk *walk = current->reclaim_state->mm_walk;
4330
4331         if (pgdat && current_is_kswapd()) {
4332                 VM_WARN_ON_ONCE(walk);
4333
4334                 walk = &pgdat->mm_walk;
4335         } else if (!walk && force_alloc) {
4336                 VM_WARN_ON_ONCE(current_is_kswapd());
4337
4338                 walk = kzalloc(sizeof(*walk), __GFP_HIGH | __GFP_NOMEMALLOC | __GFP_NOWARN);
4339         }
4340
4341         current->reclaim_state->mm_walk = walk;
4342
4343         return walk;
4344 }
4345
4346 static void clear_mm_walk(void)
4347 {
4348         struct lru_gen_mm_walk *walk = current->reclaim_state->mm_walk;
4349
4350         VM_WARN_ON_ONCE(walk && memchr_inv(walk->nr_pages, 0, sizeof(walk->nr_pages)));
4351         VM_WARN_ON_ONCE(walk && memchr_inv(walk->mm_stats, 0, sizeof(walk->mm_stats)));
4352
4353         current->reclaim_state->mm_walk = NULL;
4354
4355         if (!current_is_kswapd())
4356                 kfree(walk);
4357 }
4358
4359 static bool inc_min_seq(struct lruvec *lruvec, int type, bool can_swap)
4360 {
4361         int zone;
4362         int remaining = MAX_LRU_BATCH;
4363         struct lru_gen_folio *lrugen = &lruvec->lrugen;
4364         int new_gen, old_gen = lru_gen_from_seq(lrugen->min_seq[type]);
4365
4366         if (type == LRU_GEN_ANON && !can_swap)
4367                 goto done;
4368
4369         /* prevent cold/hot inversion if force_scan is true */
4370         for (zone = 0; zone < MAX_NR_ZONES; zone++) {
4371                 struct list_head *head = &lrugen->folios[old_gen][type][zone];
4372
4373                 while (!list_empty(head)) {
4374                         struct folio *folio = lru_to_folio(head);
4375
4376                         VM_WARN_ON_ONCE_FOLIO(folio_test_unevictable(folio), folio);
4377                         VM_WARN_ON_ONCE_FOLIO(folio_test_active(folio), folio);
4378                         VM_WARN_ON_ONCE_FOLIO(folio_is_file_lru(folio) != type, folio);
4379                         VM_WARN_ON_ONCE_FOLIO(folio_zonenum(folio) != zone, folio);
4380
4381                         new_gen = folio_inc_gen(lruvec, folio, false);
4382                         list_move_tail(&folio->lru, &lrugen->folios[new_gen][type][zone]);
4383
4384                         if (!--remaining)
4385                                 return false;
4386                 }
4387         }
4388 done:
4389         reset_ctrl_pos(lruvec, type, true);
4390         WRITE_ONCE(lrugen->min_seq[type], lrugen->min_seq[type] + 1);
4391
4392         return true;
4393 }
4394
4395 static bool try_to_inc_min_seq(struct lruvec *lruvec, bool can_swap)
4396 {
4397         int gen, type, zone;
4398         bool success = false;
4399         struct lru_gen_folio *lrugen = &lruvec->lrugen;
4400         DEFINE_MIN_SEQ(lruvec);
4401
4402         VM_WARN_ON_ONCE(!seq_is_valid(lruvec));
4403
4404         /* find the oldest populated generation */
4405         for (type = !can_swap; type < ANON_AND_FILE; type++) {
4406                 while (min_seq[type] + MIN_NR_GENS <= lrugen->max_seq) {
4407                         gen = lru_gen_from_seq(min_seq[type]);
4408
4409                         for (zone = 0; zone < MAX_NR_ZONES; zone++) {
4410                                 if (!list_empty(&lrugen->folios[gen][type][zone]))
4411                                         goto next;
4412                         }
4413
4414                         min_seq[type]++;
4415                 }
4416 next:
4417                 ;
4418         }
4419
4420         /* see the comment on lru_gen_folio */
4421         if (can_swap) {
4422                 min_seq[LRU_GEN_ANON] = min(min_seq[LRU_GEN_ANON], min_seq[LRU_GEN_FILE]);
4423                 min_seq[LRU_GEN_FILE] = max(min_seq[LRU_GEN_ANON], lrugen->min_seq[LRU_GEN_FILE]);
4424         }
4425
4426         for (type = !can_swap; type < ANON_AND_FILE; type++) {
4427                 if (min_seq[type] == lrugen->min_seq[type])
4428                         continue;
4429
4430                 reset_ctrl_pos(lruvec, type, true);
4431                 WRITE_ONCE(lrugen->min_seq[type], min_seq[type]);
4432                 success = true;
4433         }
4434
4435         return success;
4436 }
4437
4438 static void inc_max_seq(struct lruvec *lruvec, bool can_swap, bool force_scan)
4439 {
4440         int prev, next;
4441         int type, zone;
4442         struct lru_gen_folio *lrugen = &lruvec->lrugen;
4443 restart:
4444         spin_lock_irq(&lruvec->lru_lock);
4445
4446         VM_WARN_ON_ONCE(!seq_is_valid(lruvec));
4447
4448         for (type = ANON_AND_FILE - 1; type >= 0; type--) {
4449                 if (get_nr_gens(lruvec, type) != MAX_NR_GENS)
4450                         continue;
4451
4452                 VM_WARN_ON_ONCE(!force_scan && (type == LRU_GEN_FILE || can_swap));
4453
4454                 if (inc_min_seq(lruvec, type, can_swap))
4455                         continue;
4456
4457                 spin_unlock_irq(&lruvec->lru_lock);
4458                 cond_resched();
4459                 goto restart;
4460         }
4461
4462         /*
4463          * Update the active/inactive LRU sizes for compatibility. Both sides of
4464          * the current max_seq need to be covered, since max_seq+1 can overlap
4465          * with min_seq[LRU_GEN_ANON] if swapping is constrained. And if they do
4466          * overlap, cold/hot inversion happens.
4467          */
4468         prev = lru_gen_from_seq(lrugen->max_seq - 1);
4469         next = lru_gen_from_seq(lrugen->max_seq + 1);
4470
4471         for (type = 0; type < ANON_AND_FILE; type++) {
4472                 for (zone = 0; zone < MAX_NR_ZONES; zone++) {
4473                         enum lru_list lru = type * LRU_INACTIVE_FILE;
4474                         long delta = lrugen->nr_pages[prev][type][zone] -
4475                                      lrugen->nr_pages[next][type][zone];
4476
4477                         if (!delta)
4478                                 continue;
4479
4480                         __update_lru_size(lruvec, lru, zone, delta);
4481                         __update_lru_size(lruvec, lru + LRU_ACTIVE, zone, -delta);
4482                 }
4483         }
4484
4485         for (type = 0; type < ANON_AND_FILE; type++)
4486                 reset_ctrl_pos(lruvec, type, false);
4487
4488         WRITE_ONCE(lrugen->timestamps[next], jiffies);
4489         /* make sure preceding modifications appear */
4490         smp_store_release(&lrugen->max_seq, lrugen->max_seq + 1);
4491
4492         spin_unlock_irq(&lruvec->lru_lock);
4493 }
4494
4495 static bool try_to_inc_max_seq(struct lruvec *lruvec, unsigned long max_seq,
4496                                struct scan_control *sc, bool can_swap, bool force_scan)
4497 {
4498         bool success;
4499         struct lru_gen_mm_walk *walk;
4500         struct mm_struct *mm = NULL;
4501         struct lru_gen_folio *lrugen = &lruvec->lrugen;
4502
4503         VM_WARN_ON_ONCE(max_seq > READ_ONCE(lrugen->max_seq));
4504
4505         /* see the comment in iterate_mm_list() */
4506         if (max_seq <= READ_ONCE(lruvec->mm_state.seq)) {
4507                 success = false;
4508                 goto done;
4509         }
4510
4511         /*
4512          * If the hardware doesn't automatically set the accessed bit, fallback
4513          * to lru_gen_look_around(), which only clears the accessed bit in a
4514          * handful of PTEs. Spreading the work out over a period of time usually
4515          * is less efficient, but it avoids bursty page faults.
4516          */
4517         if (!should_walk_mmu()) {
4518                 success = iterate_mm_list_nowalk(lruvec, max_seq);
4519                 goto done;
4520         }
4521
4522         walk = set_mm_walk(NULL, true);
4523         if (!walk) {
4524                 success = iterate_mm_list_nowalk(lruvec, max_seq);
4525                 goto done;
4526         }
4527
4528         walk->lruvec = lruvec;
4529         walk->max_seq = max_seq;
4530         walk->can_swap = can_swap;
4531         walk->force_scan = force_scan;
4532
4533         do {
4534                 success = iterate_mm_list(lruvec, walk, &mm);
4535                 if (mm)
4536                         walk_mm(lruvec, mm, walk);
4537         } while (mm);
4538 done:
4539         if (success)
4540                 inc_max_seq(lruvec, can_swap, force_scan);
4541
4542         return success;
4543 }
4544
4545 /******************************************************************************
4546  *                          working set protection
4547  ******************************************************************************/
4548
4549 static bool lruvec_is_sizable(struct lruvec *lruvec, struct scan_control *sc)
4550 {
4551         int gen, type, zone;
4552         unsigned long total = 0;
4553         bool can_swap = get_swappiness(lruvec, sc);
4554         struct lru_gen_folio *lrugen = &lruvec->lrugen;
4555         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
4556         DEFINE_MAX_SEQ(lruvec);
4557         DEFINE_MIN_SEQ(lruvec);
4558
4559         for (type = !can_swap; type < ANON_AND_FILE; type++) {
4560                 unsigned long seq;
4561
4562                 for (seq = min_seq[type]; seq <= max_seq; seq++) {
4563                         gen = lru_gen_from_seq(seq);
4564
4565                         for (zone = 0; zone < MAX_NR_ZONES; zone++)
4566                                 total += max(READ_ONCE(lrugen->nr_pages[gen][type][zone]), 0L);
4567                 }
4568         }
4569
4570         /* whether the size is big enough to be helpful */
4571         return mem_cgroup_online(memcg) ? (total >> sc->priority) : total;
4572 }
4573
4574 static bool lruvec_is_reclaimable(struct lruvec *lruvec, struct scan_control *sc,
4575                                   unsigned long min_ttl)
4576 {
4577         int gen;
4578         unsigned long birth;
4579         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
4580         DEFINE_MIN_SEQ(lruvec);
4581
4582         /* see the comment on lru_gen_folio */
4583         gen = lru_gen_from_seq(min_seq[LRU_GEN_FILE]);
4584         birth = READ_ONCE(lruvec->lrugen.timestamps[gen]);
4585
4586         if (time_is_after_jiffies(birth + min_ttl))
4587                 return false;
4588
4589         if (!lruvec_is_sizable(lruvec, sc))
4590                 return false;
4591
4592         mem_cgroup_calculate_protection(NULL, memcg);
4593
4594         return !mem_cgroup_below_min(NULL, memcg);
4595 }
4596
4597 /* to protect the working set of the last N jiffies */
4598 static unsigned long lru_gen_min_ttl __read_mostly;
4599
4600 static void lru_gen_age_node(struct pglist_data *pgdat, struct scan_control *sc)
4601 {
4602         struct mem_cgroup *memcg;
4603         unsigned long min_ttl = READ_ONCE(lru_gen_min_ttl);
4604
4605         VM_WARN_ON_ONCE(!current_is_kswapd());
4606
4607         /* check the order to exclude compaction-induced reclaim */
4608         if (!min_ttl || sc->order || sc->priority == DEF_PRIORITY)
4609                 return;
4610
4611         memcg = mem_cgroup_iter(NULL, NULL, NULL);
4612         do {
4613                 struct lruvec *lruvec = mem_cgroup_lruvec(memcg, pgdat);
4614
4615                 if (lruvec_is_reclaimable(lruvec, sc, min_ttl)) {
4616                         mem_cgroup_iter_break(NULL, memcg);
4617                         return;
4618                 }
4619
4620                 cond_resched();
4621         } while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)));
4622
4623         /*
4624          * The main goal is to OOM kill if every generation from all memcgs is
4625          * younger than min_ttl. However, another possibility is all memcgs are
4626          * either too small or below min.
4627          */
4628         if (mutex_trylock(&oom_lock)) {
4629                 struct oom_control oc = {
4630                         .gfp_mask = sc->gfp_mask,
4631                 };
4632
4633                 out_of_memory(&oc);
4634
4635                 mutex_unlock(&oom_lock);
4636         }
4637 }
4638
4639 /******************************************************************************
4640  *                          rmap/PT walk feedback
4641  ******************************************************************************/
4642
4643 /*
4644  * This function exploits spatial locality when shrink_folio_list() walks the
4645  * rmap. It scans the adjacent PTEs of a young PTE and promotes hot pages. If
4646  * the scan was done cacheline efficiently, it adds the PMD entry pointing to
4647  * the PTE table to the Bloom filter. This forms a feedback loop between the
4648  * eviction and the aging.
4649  */
4650 void lru_gen_look_around(struct page_vma_mapped_walk *pvmw)
4651 {
4652         int i;
4653         unsigned long start;
4654         unsigned long end;
4655         struct lru_gen_mm_walk *walk;
4656         int young = 0;
4657         pte_t *pte = pvmw->pte;
4658         unsigned long addr = pvmw->address;
4659         struct folio *folio = pfn_folio(pvmw->pfn);
4660         bool can_swap = !folio_is_file_lru(folio);
4661         struct mem_cgroup *memcg = folio_memcg(folio);
4662         struct pglist_data *pgdat = folio_pgdat(folio);
4663         struct lruvec *lruvec = mem_cgroup_lruvec(memcg, pgdat);
4664         DEFINE_MAX_SEQ(lruvec);
4665         int old_gen, new_gen = lru_gen_from_seq(max_seq);
4666
4667         lockdep_assert_held(pvmw->ptl);
4668         VM_WARN_ON_ONCE_FOLIO(folio_test_lru(folio), folio);
4669
4670         if (spin_is_contended(pvmw->ptl))
4671                 return;
4672
4673         /* avoid taking the LRU lock under the PTL when possible */
4674         walk = current->reclaim_state ? current->reclaim_state->mm_walk : NULL;
4675
4676         start = max(addr & PMD_MASK, pvmw->vma->vm_start);
4677         end = min(addr | ~PMD_MASK, pvmw->vma->vm_end - 1) + 1;
4678
4679         if (end - start > MIN_LRU_BATCH * PAGE_SIZE) {
4680                 if (addr - start < MIN_LRU_BATCH * PAGE_SIZE / 2)
4681                         end = start + MIN_LRU_BATCH * PAGE_SIZE;
4682                 else if (end - addr < MIN_LRU_BATCH * PAGE_SIZE / 2)
4683                         start = end - MIN_LRU_BATCH * PAGE_SIZE;
4684                 else {
4685                         start = addr - MIN_LRU_BATCH * PAGE_SIZE / 2;
4686                         end = addr + MIN_LRU_BATCH * PAGE_SIZE / 2;
4687                 }
4688         }
4689
4690         /* folio_update_gen() requires stable folio_memcg() */
4691         if (!mem_cgroup_trylock_pages(memcg))
4692                 return;
4693
4694         arch_enter_lazy_mmu_mode();
4695
4696         pte -= (addr - start) / PAGE_SIZE;
4697
4698         for (i = 0, addr = start; addr != end; i++, addr += PAGE_SIZE) {
4699                 unsigned long pfn;
4700                 pte_t ptent = ptep_get(pte + i);
4701
4702                 pfn = get_pte_pfn(ptent, pvmw->vma, addr);
4703                 if (pfn == -1)
4704                         continue;
4705
4706                 if (!pte_young(ptent))
4707                         continue;
4708
4709                 folio = get_pfn_folio(pfn, memcg, pgdat, can_swap);
4710                 if (!folio)
4711                         continue;
4712
4713                 if (!ptep_test_and_clear_young(pvmw->vma, addr, pte + i))
4714                         VM_WARN_ON_ONCE(true);
4715
4716                 young++;
4717
4718                 if (pte_dirty(ptent) && !folio_test_dirty(folio) &&
4719                     !(folio_test_anon(folio) && folio_test_swapbacked(folio) &&
4720                       !folio_test_swapcache(folio)))
4721                         folio_mark_dirty(folio);
4722
4723                 if (walk) {
4724                         old_gen = folio_update_gen(folio, new_gen);
4725                         if (old_gen >= 0 && old_gen != new_gen)
4726                                 update_batch_size(walk, folio, old_gen, new_gen);
4727
4728                         continue;
4729                 }
4730
4731                 old_gen = folio_lru_gen(folio);
4732                 if (old_gen < 0)
4733                         folio_set_referenced(folio);
4734                 else if (old_gen != new_gen)
4735                         folio_activate(folio);
4736         }
4737
4738         arch_leave_lazy_mmu_mode();
4739         mem_cgroup_unlock_pages();
4740
4741         /* feedback from rmap walkers to page table walkers */
4742         if (suitable_to_scan(i, young))
4743                 update_bloom_filter(lruvec, max_seq, pvmw->pmd);
4744 }
4745
4746 /******************************************************************************
4747  *                          memcg LRU
4748  ******************************************************************************/
4749
4750 /* see the comment on MEMCG_NR_GENS */
4751 enum {
4752         MEMCG_LRU_NOP,
4753         MEMCG_LRU_HEAD,
4754         MEMCG_LRU_TAIL,
4755         MEMCG_LRU_OLD,
4756         MEMCG_LRU_YOUNG,
4757 };
4758
4759 #ifdef CONFIG_MEMCG
4760
4761 static int lru_gen_memcg_seg(struct lruvec *lruvec)
4762 {
4763         return READ_ONCE(lruvec->lrugen.seg);
4764 }
4765
4766 static void lru_gen_rotate_memcg(struct lruvec *lruvec, int op)
4767 {
4768         int seg;
4769         int old, new;
4770         unsigned long flags;
4771         int bin = get_random_u32_below(MEMCG_NR_BINS);
4772         struct pglist_data *pgdat = lruvec_pgdat(lruvec);
4773
4774         spin_lock_irqsave(&pgdat->memcg_lru.lock, flags);
4775
4776         VM_WARN_ON_ONCE(hlist_nulls_unhashed(&lruvec->lrugen.list));
4777
4778         seg = 0;
4779         new = old = lruvec->lrugen.gen;
4780
4781         /* see the comment on MEMCG_NR_GENS */
4782         if (op == MEMCG_LRU_HEAD)
4783                 seg = MEMCG_LRU_HEAD;
4784         else if (op == MEMCG_LRU_TAIL)
4785                 seg = MEMCG_LRU_TAIL;
4786         else if (op == MEMCG_LRU_OLD)
4787                 new = get_memcg_gen(pgdat->memcg_lru.seq);
4788         else if (op == MEMCG_LRU_YOUNG)
4789                 new = get_memcg_gen(pgdat->memcg_lru.seq + 1);
4790         else
4791                 VM_WARN_ON_ONCE(true);
4792
4793         hlist_nulls_del_rcu(&lruvec->lrugen.list);
4794
4795         if (op == MEMCG_LRU_HEAD || op == MEMCG_LRU_OLD)
4796                 hlist_nulls_add_head_rcu(&lruvec->lrugen.list, &pgdat->memcg_lru.fifo[new][bin]);
4797         else
4798                 hlist_nulls_add_tail_rcu(&lruvec->lrugen.list, &pgdat->memcg_lru.fifo[new][bin]);
4799
4800         pgdat->memcg_lru.nr_memcgs[old]--;
4801         pgdat->memcg_lru.nr_memcgs[new]++;
4802
4803         lruvec->lrugen.gen = new;
4804         WRITE_ONCE(lruvec->lrugen.seg, seg);
4805
4806         if (!pgdat->memcg_lru.nr_memcgs[old] && old == get_memcg_gen(pgdat->memcg_lru.seq))
4807                 WRITE_ONCE(pgdat->memcg_lru.seq, pgdat->memcg_lru.seq + 1);
4808
4809         spin_unlock_irqrestore(&pgdat->memcg_lru.lock, flags);
4810 }
4811
4812 void lru_gen_online_memcg(struct mem_cgroup *memcg)
4813 {
4814         int gen;
4815         int nid;
4816         int bin = get_random_u32_below(MEMCG_NR_BINS);
4817
4818         for_each_node(nid) {
4819                 struct pglist_data *pgdat = NODE_DATA(nid);
4820                 struct lruvec *lruvec = get_lruvec(memcg, nid);
4821
4822                 spin_lock_irq(&pgdat->memcg_lru.lock);
4823
4824                 VM_WARN_ON_ONCE(!hlist_nulls_unhashed(&lruvec->lrugen.list));
4825
4826                 gen = get_memcg_gen(pgdat->memcg_lru.seq);
4827
4828                 hlist_nulls_add_tail_rcu(&lruvec->lrugen.list, &pgdat->memcg_lru.fifo[gen][bin]);
4829                 pgdat->memcg_lru.nr_memcgs[gen]++;
4830
4831                 lruvec->lrugen.gen = gen;
4832
4833                 spin_unlock_irq(&pgdat->memcg_lru.lock);
4834         }
4835 }
4836
4837 void lru_gen_offline_memcg(struct mem_cgroup *memcg)
4838 {
4839         int nid;
4840
4841         for_each_node(nid) {
4842                 struct lruvec *lruvec = get_lruvec(memcg, nid);
4843
4844                 lru_gen_rotate_memcg(lruvec, MEMCG_LRU_OLD);
4845         }
4846 }
4847
4848 void lru_gen_release_memcg(struct mem_cgroup *memcg)
4849 {
4850         int gen;
4851         int nid;
4852
4853         for_each_node(nid) {
4854                 struct pglist_data *pgdat = NODE_DATA(nid);
4855                 struct lruvec *lruvec = get_lruvec(memcg, nid);
4856
4857                 spin_lock_irq(&pgdat->memcg_lru.lock);
4858
4859                 if (hlist_nulls_unhashed(&lruvec->lrugen.list))
4860                         goto unlock;
4861
4862                 gen = lruvec->lrugen.gen;
4863
4864                 hlist_nulls_del_init_rcu(&lruvec->lrugen.list);
4865                 pgdat->memcg_lru.nr_memcgs[gen]--;
4866
4867                 if (!pgdat->memcg_lru.nr_memcgs[gen] && gen == get_memcg_gen(pgdat->memcg_lru.seq))
4868                         WRITE_ONCE(pgdat->memcg_lru.seq, pgdat->memcg_lru.seq + 1);
4869 unlock:
4870                 spin_unlock_irq(&pgdat->memcg_lru.lock);
4871         }
4872 }
4873
4874 void lru_gen_soft_reclaim(struct mem_cgroup *memcg, int nid)
4875 {
4876         struct lruvec *lruvec = get_lruvec(memcg, nid);
4877
4878         /* see the comment on MEMCG_NR_GENS */
4879         if (lru_gen_memcg_seg(lruvec) != MEMCG_LRU_HEAD)
4880                 lru_gen_rotate_memcg(lruvec, MEMCG_LRU_HEAD);
4881 }
4882
4883 #else /* !CONFIG_MEMCG */
4884
4885 static int lru_gen_memcg_seg(struct lruvec *lruvec)
4886 {
4887         return 0;
4888 }
4889
4890 #endif
4891
4892 /******************************************************************************
4893  *                          the eviction
4894  ******************************************************************************/
4895
4896 static bool sort_folio(struct lruvec *lruvec, struct folio *folio, struct scan_control *sc,
4897                        int tier_idx)
4898 {
4899         bool success;
4900         int gen = folio_lru_gen(folio);
4901         int type = folio_is_file_lru(folio);
4902         int zone = folio_zonenum(folio);
4903         int delta = folio_nr_pages(folio);
4904         int refs = folio_lru_refs(folio);
4905         int tier = lru_tier_from_refs(refs);
4906         struct lru_gen_folio *lrugen = &lruvec->lrugen;
4907
4908         VM_WARN_ON_ONCE_FOLIO(gen >= MAX_NR_GENS, folio);
4909
4910         /* unevictable */
4911         if (!folio_evictable(folio)) {
4912                 success = lru_gen_del_folio(lruvec, folio, true);
4913                 VM_WARN_ON_ONCE_FOLIO(!success, folio);
4914                 folio_set_unevictable(folio);
4915                 lruvec_add_folio(lruvec, folio);
4916                 __count_vm_events(UNEVICTABLE_PGCULLED, delta);
4917                 return true;
4918         }
4919
4920         /* dirty lazyfree */
4921         if (type == LRU_GEN_FILE && folio_test_anon(folio) && folio_test_dirty(folio)) {
4922                 success = lru_gen_del_folio(lruvec, folio, true);
4923                 VM_WARN_ON_ONCE_FOLIO(!success, folio);
4924                 folio_set_swapbacked(folio);
4925                 lruvec_add_folio_tail(lruvec, folio);
4926                 return true;
4927         }
4928
4929         /* promoted */
4930         if (gen != lru_gen_from_seq(lrugen->min_seq[type])) {
4931                 list_move(&folio->lru, &lrugen->folios[gen][type][zone]);
4932                 return true;
4933         }
4934
4935         /* protected */
4936         if (tier > tier_idx) {
4937                 int hist = lru_hist_from_seq(lrugen->min_seq[type]);
4938
4939                 gen = folio_inc_gen(lruvec, folio, false);
4940                 list_move_tail(&folio->lru, &lrugen->folios[gen][type][zone]);
4941
4942                 WRITE_ONCE(lrugen->protected[hist][type][tier - 1],
4943                            lrugen->protected[hist][type][tier - 1] + delta);
4944                 return true;
4945         }
4946
4947         /* ineligible */
4948         if (zone > sc->reclaim_idx || skip_cma(folio, sc)) {
4949                 gen = folio_inc_gen(lruvec, folio, false);
4950                 list_move_tail(&folio->lru, &lrugen->folios[gen][type][zone]);
4951                 return true;
4952         }
4953
4954         /* waiting for writeback */
4955         if (folio_test_locked(folio) || folio_test_writeback(folio) ||
4956             (type == LRU_GEN_FILE && folio_test_dirty(folio))) {
4957                 gen = folio_inc_gen(lruvec, folio, true);
4958                 list_move(&folio->lru, &lrugen->folios[gen][type][zone]);
4959                 return true;
4960         }
4961
4962         return false;
4963 }
4964
4965 static bool isolate_folio(struct lruvec *lruvec, struct folio *folio, struct scan_control *sc)
4966 {
4967         bool success;
4968
4969         /* swapping inhibited */
4970         if (!(sc->gfp_mask & __GFP_IO) &&
4971             (folio_test_dirty(folio) ||
4972              (folio_test_anon(folio) && !folio_test_swapcache(folio))))
4973                 return false;
4974
4975         /* raced with release_pages() */
4976         if (!folio_try_get(folio))
4977                 return false;
4978
4979         /* raced with another isolation */
4980         if (!folio_test_clear_lru(folio)) {
4981                 folio_put(folio);
4982                 return false;
4983         }
4984
4985         /* see the comment on MAX_NR_TIERS */
4986         if (!folio_test_referenced(folio))
4987                 set_mask_bits(&folio->flags, LRU_REFS_MASK | LRU_REFS_FLAGS, 0);
4988
4989         /* for shrink_folio_list() */
4990         folio_clear_reclaim(folio);
4991         folio_clear_referenced(folio);
4992
4993         success = lru_gen_del_folio(lruvec, folio, true);
4994         VM_WARN_ON_ONCE_FOLIO(!success, folio);
4995
4996         return true;
4997 }
4998
4999 static int scan_folios(struct lruvec *lruvec, struct scan_control *sc,
5000                        int type, int tier, struct list_head *list)
5001 {
5002         int i;
5003         int gen;
5004         enum vm_event_item item;
5005         int sorted = 0;
5006         int scanned = 0;
5007         int isolated = 0;
5008         int remaining = MAX_LRU_BATCH;
5009         struct lru_gen_folio *lrugen = &lruvec->lrugen;
5010         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
5011
5012         VM_WARN_ON_ONCE(!list_empty(list));
5013
5014         if (get_nr_gens(lruvec, type) == MIN_NR_GENS)
5015                 return 0;
5016
5017         gen = lru_gen_from_seq(lrugen->min_seq[type]);
5018
5019         for (i = MAX_NR_ZONES; i > 0; i--) {
5020                 LIST_HEAD(moved);
5021                 int skipped = 0;
5022                 int zone = (sc->reclaim_idx + i) % MAX_NR_ZONES;
5023                 struct list_head *head = &lrugen->folios[gen][type][zone];
5024
5025                 while (!list_empty(head)) {
5026                         struct folio *folio = lru_to_folio(head);
5027                         int delta = folio_nr_pages(folio);
5028
5029                         VM_WARN_ON_ONCE_FOLIO(folio_test_unevictable(folio), folio);
5030                         VM_WARN_ON_ONCE_FOLIO(folio_test_active(folio), folio);
5031                         VM_WARN_ON_ONCE_FOLIO(folio_is_file_lru(folio) != type, folio);
5032                         VM_WARN_ON_ONCE_FOLIO(folio_zonenum(folio) != zone, folio);
5033
5034                         scanned += delta;
5035
5036                         if (sort_folio(lruvec, folio, sc, tier))
5037                                 sorted += delta;
5038                         else if (isolate_folio(lruvec, folio, sc)) {
5039                                 list_add(&folio->lru, list);
5040                                 isolated += delta;
5041                         } else {
5042                                 list_move(&folio->lru, &moved);
5043                                 skipped += delta;
5044                         }
5045
5046                         if (!--remaining || max(isolated, skipped) >= MIN_LRU_BATCH)
5047                                 break;
5048                 }
5049
5050                 if (skipped) {
5051                         list_splice(&moved, head);
5052                         __count_zid_vm_events(PGSCAN_SKIP, zone, skipped);
5053                 }
5054
5055                 if (!remaining || isolated >= MIN_LRU_BATCH)
5056                         break;
5057         }
5058
5059         item = PGSCAN_KSWAPD + reclaimer_offset();
5060         if (!cgroup_reclaim(sc)) {
5061                 __count_vm_events(item, isolated);
5062                 __count_vm_events(PGREFILL, sorted);
5063         }
5064         __count_memcg_events(memcg, item, isolated);
5065         __count_memcg_events(memcg, PGREFILL, sorted);
5066         __count_vm_events(PGSCAN_ANON + type, isolated);
5067
5068         /*
5069          * There might not be eligible folios due to reclaim_idx. Check the
5070          * remaining to prevent livelock if it's not making progress.
5071          */
5072         return isolated || !remaining ? scanned : 0;
5073 }
5074
5075 static int get_tier_idx(struct lruvec *lruvec, int type)
5076 {
5077         int tier;
5078         struct ctrl_pos sp, pv;
5079
5080         /*
5081          * To leave a margin for fluctuations, use a larger gain factor (1:2).
5082          * This value is chosen because any other tier would have at least twice
5083          * as many refaults as the first tier.
5084          */
5085         read_ctrl_pos(lruvec, type, 0, 1, &sp);
5086         for (tier = 1; tier < MAX_NR_TIERS; tier++) {
5087                 read_ctrl_pos(lruvec, type, tier, 2, &pv);
5088                 if (!positive_ctrl_err(&sp, &pv))
5089                         break;
5090         }
5091
5092         return tier - 1;
5093 }
5094
5095 static int get_type_to_scan(struct lruvec *lruvec, int swappiness, int *tier_idx)
5096 {
5097         int type, tier;
5098         struct ctrl_pos sp, pv;
5099         int gain[ANON_AND_FILE] = { swappiness, 200 - swappiness };
5100
5101         /*
5102          * Compare the first tier of anon with that of file to determine which
5103          * type to scan. Also need to compare other tiers of the selected type
5104          * with the first tier of the other type to determine the last tier (of
5105          * the selected type) to evict.
5106          */
5107         read_ctrl_pos(lruvec, LRU_GEN_ANON, 0, gain[LRU_GEN_ANON], &sp);
5108         read_ctrl_pos(lruvec, LRU_GEN_FILE, 0, gain[LRU_GEN_FILE], &pv);
5109         type = positive_ctrl_err(&sp, &pv);
5110
5111         read_ctrl_pos(lruvec, !type, 0, gain[!type], &sp);
5112         for (tier = 1; tier < MAX_NR_TIERS; tier++) {
5113                 read_ctrl_pos(lruvec, type, tier, gain[type], &pv);
5114                 if (!positive_ctrl_err(&sp, &pv))
5115                         break;
5116         }
5117
5118         *tier_idx = tier - 1;
5119
5120         return type;
5121 }
5122
5123 static int isolate_folios(struct lruvec *lruvec, struct scan_control *sc, int swappiness,
5124                           int *type_scanned, struct list_head *list)
5125 {
5126         int i;
5127         int type;
5128         int scanned;
5129         int tier = -1;
5130         DEFINE_MIN_SEQ(lruvec);
5131
5132         /*
5133          * Try to make the obvious choice first. When anon and file are both
5134          * available from the same generation, interpret swappiness 1 as file
5135          * first and 200 as anon first.
5136          */
5137         if (!swappiness)
5138                 type = LRU_GEN_FILE;
5139         else if (min_seq[LRU_GEN_ANON] < min_seq[LRU_GEN_FILE])
5140                 type = LRU_GEN_ANON;
5141         else if (swappiness == 1)
5142                 type = LRU_GEN_FILE;
5143         else if (swappiness == 200)
5144                 type = LRU_GEN_ANON;
5145         else
5146                 type = get_type_to_scan(lruvec, swappiness, &tier);
5147
5148         for (i = !swappiness; i < ANON_AND_FILE; i++) {
5149                 if (tier < 0)
5150                         tier = get_tier_idx(lruvec, type);
5151
5152                 scanned = scan_folios(lruvec, sc, type, tier, list);
5153                 if (scanned)
5154                         break;
5155
5156                 type = !type;
5157                 tier = -1;
5158         }
5159
5160         *type_scanned = type;
5161
5162         return scanned;
5163 }
5164
5165 static int evict_folios(struct lruvec *lruvec, struct scan_control *sc, int swappiness)
5166 {
5167         int type;
5168         int scanned;
5169         int reclaimed;
5170         LIST_HEAD(list);
5171         LIST_HEAD(clean);
5172         struct folio *folio;
5173         struct folio *next;
5174         enum vm_event_item item;
5175         struct reclaim_stat stat;
5176         struct lru_gen_mm_walk *walk;
5177         bool skip_retry = false;
5178         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
5179         struct pglist_data *pgdat = lruvec_pgdat(lruvec);
5180
5181         spin_lock_irq(&lruvec->lru_lock);
5182
5183         scanned = isolate_folios(lruvec, sc, swappiness, &type, &list);
5184
5185         scanned += try_to_inc_min_seq(lruvec, swappiness);
5186
5187         if (get_nr_gens(lruvec, !swappiness) == MIN_NR_GENS)
5188                 scanned = 0;
5189
5190         spin_unlock_irq(&lruvec->lru_lock);
5191
5192         if (list_empty(&list))
5193                 return scanned;
5194 retry:
5195         reclaimed = shrink_folio_list(&list, pgdat, sc, &stat, false);
5196         sc->nr_reclaimed += reclaimed;
5197
5198         list_for_each_entry_safe_reverse(folio, next, &list, lru) {
5199                 if (!folio_evictable(folio)) {
5200                         list_del(&folio->lru);
5201                         folio_putback_lru(folio);
5202                         continue;
5203                 }
5204
5205                 if (folio_test_reclaim(folio) &&
5206                     (folio_test_dirty(folio) || folio_test_writeback(folio))) {
5207                         /* restore LRU_REFS_FLAGS cleared by isolate_folio() */
5208                         if (folio_test_workingset(folio))
5209                                 folio_set_referenced(folio);
5210                         continue;
5211                 }
5212
5213                 if (skip_retry || folio_test_active(folio) || folio_test_referenced(folio) ||
5214                     folio_mapped(folio) || folio_test_locked(folio) ||
5215                     folio_test_dirty(folio) || folio_test_writeback(folio)) {
5216                         /* don't add rejected folios to the oldest generation */
5217                         set_mask_bits(&folio->flags, LRU_REFS_MASK | LRU_REFS_FLAGS,
5218                                       BIT(PG_active));
5219                         continue;
5220                 }
5221
5222                 /* retry folios that may have missed folio_rotate_reclaimable() */
5223                 list_move(&folio->lru, &clean);
5224                 sc->nr_scanned -= folio_nr_pages(folio);
5225         }
5226
5227         spin_lock_irq(&lruvec->lru_lock);
5228
5229         move_folios_to_lru(lruvec, &list);
5230
5231         walk = current->reclaim_state->mm_walk;
5232         if (walk && walk->batched)
5233                 reset_batch_size(lruvec, walk);
5234
5235         item = PGSTEAL_KSWAPD + reclaimer_offset();
5236         if (!cgroup_reclaim(sc))
5237                 __count_vm_events(item, reclaimed);
5238         __count_memcg_events(memcg, item, reclaimed);
5239         __count_vm_events(PGSTEAL_ANON + type, reclaimed);
5240
5241         spin_unlock_irq(&lruvec->lru_lock);
5242
5243         mem_cgroup_uncharge_list(&list);
5244         free_unref_page_list(&list);
5245
5246         INIT_LIST_HEAD(&list);
5247         list_splice_init(&clean, &list);
5248
5249         if (!list_empty(&list)) {
5250                 skip_retry = true;
5251                 goto retry;
5252         }
5253
5254         return scanned;
5255 }
5256
5257 static bool should_run_aging(struct lruvec *lruvec, unsigned long max_seq,
5258                              struct scan_control *sc, bool can_swap, unsigned long *nr_to_scan)
5259 {
5260         int gen, type, zone;
5261         unsigned long old = 0;
5262         unsigned long young = 0;
5263         unsigned long total = 0;
5264         struct lru_gen_folio *lrugen = &lruvec->lrugen;
5265         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
5266         DEFINE_MIN_SEQ(lruvec);
5267
5268         /* whether this lruvec is completely out of cold folios */
5269         if (min_seq[!can_swap] + MIN_NR_GENS > max_seq) {
5270                 *nr_to_scan = 0;
5271                 return true;
5272         }
5273
5274         for (type = !can_swap; type < ANON_AND_FILE; type++) {
5275                 unsigned long seq;
5276
5277                 for (seq = min_seq[type]; seq <= max_seq; seq++) {
5278                         unsigned long size = 0;
5279
5280                         gen = lru_gen_from_seq(seq);
5281
5282                         for (zone = 0; zone < MAX_NR_ZONES; zone++)
5283                                 size += max(READ_ONCE(lrugen->nr_pages[gen][type][zone]), 0L);
5284
5285                         total += size;
5286                         if (seq == max_seq)
5287                                 young += size;
5288                         else if (seq + MIN_NR_GENS == max_seq)
5289                                 old += size;
5290                 }
5291         }
5292
5293         /* try to scrape all its memory if this memcg was deleted */
5294         *nr_to_scan = mem_cgroup_online(memcg) ? (total >> sc->priority) : total;
5295
5296         /*
5297          * The aging tries to be lazy to reduce the overhead, while the eviction
5298          * stalls when the number of generations reaches MIN_NR_GENS. Hence, the
5299          * ideal number of generations is MIN_NR_GENS+1.
5300          */
5301         if (min_seq[!can_swap] + MIN_NR_GENS < max_seq)
5302                 return false;
5303
5304         /*
5305          * It's also ideal to spread pages out evenly, i.e., 1/(MIN_NR_GENS+1)
5306          * of the total number of pages for each generation. A reasonable range
5307          * for this average portion is [1/MIN_NR_GENS, 1/(MIN_NR_GENS+2)]. The
5308          * aging cares about the upper bound of hot pages, while the eviction
5309          * cares about the lower bound of cold pages.
5310          */
5311         if (young * MIN_NR_GENS > total)
5312                 return true;
5313         if (old * (MIN_NR_GENS + 2) < total)
5314                 return true;
5315
5316         return false;
5317 }
5318
5319 /*
5320  * For future optimizations:
5321  * 1. Defer try_to_inc_max_seq() to workqueues to reduce latency for memcg
5322  *    reclaim.
5323  */
5324 static long get_nr_to_scan(struct lruvec *lruvec, struct scan_control *sc, bool can_swap)
5325 {
5326         unsigned long nr_to_scan;
5327         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
5328         DEFINE_MAX_SEQ(lruvec);
5329
5330         if (mem_cgroup_below_min(sc->target_mem_cgroup, memcg))
5331                 return 0;
5332
5333         if (!should_run_aging(lruvec, max_seq, sc, can_swap, &nr_to_scan))
5334                 return nr_to_scan;
5335
5336         /* skip the aging path at the default priority */
5337         if (sc->priority == DEF_PRIORITY)
5338                 return nr_to_scan;
5339
5340         /* skip this lruvec as it's low on cold folios */
5341         return try_to_inc_max_seq(lruvec, max_seq, sc, can_swap, false) ? -1 : 0;
5342 }
5343
5344 static unsigned long get_nr_to_reclaim(struct scan_control *sc)
5345 {
5346         /* don't abort memcg reclaim to ensure fairness */
5347         if (!root_reclaim(sc))
5348                 return -1;
5349
5350         return max(sc->nr_to_reclaim, compact_gap(sc->order));
5351 }
5352
5353 static bool try_to_shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
5354 {
5355         long nr_to_scan;
5356         unsigned long scanned = 0;
5357         unsigned long nr_to_reclaim = get_nr_to_reclaim(sc);
5358         int swappiness = get_swappiness(lruvec, sc);
5359
5360         /* clean file folios are more likely to exist */
5361         if (swappiness && !(sc->gfp_mask & __GFP_IO))
5362                 swappiness = 1;
5363
5364         while (true) {
5365                 int delta;
5366
5367                 nr_to_scan = get_nr_to_scan(lruvec, sc, swappiness);
5368                 if (nr_to_scan <= 0)
5369                         break;
5370
5371                 delta = evict_folios(lruvec, sc, swappiness);
5372                 if (!delta)
5373                         break;
5374
5375                 scanned += delta;
5376                 if (scanned >= nr_to_scan)
5377                         break;
5378
5379                 if (sc->nr_reclaimed >= nr_to_reclaim)
5380                         break;
5381
5382                 cond_resched();
5383         }
5384
5385         /* whether try_to_inc_max_seq() was successful */
5386         return nr_to_scan < 0;
5387 }
5388
5389 static int shrink_one(struct lruvec *lruvec, struct scan_control *sc)
5390 {
5391         bool success;
5392         unsigned long scanned = sc->nr_scanned;
5393         unsigned long reclaimed = sc->nr_reclaimed;
5394         int seg = lru_gen_memcg_seg(lruvec);
5395         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
5396         struct pglist_data *pgdat = lruvec_pgdat(lruvec);
5397
5398         /* see the comment on MEMCG_NR_GENS */
5399         if (!lruvec_is_sizable(lruvec, sc))
5400                 return seg != MEMCG_LRU_TAIL ? MEMCG_LRU_TAIL : MEMCG_LRU_YOUNG;
5401
5402         mem_cgroup_calculate_protection(NULL, memcg);
5403
5404         if (mem_cgroup_below_min(NULL, memcg))
5405                 return MEMCG_LRU_YOUNG;
5406
5407         if (mem_cgroup_below_low(NULL, memcg)) {
5408                 /* see the comment on MEMCG_NR_GENS */
5409                 if (seg != MEMCG_LRU_TAIL)
5410                         return MEMCG_LRU_TAIL;
5411
5412                 memcg_memory_event(memcg, MEMCG_LOW);
5413         }
5414
5415         success = try_to_shrink_lruvec(lruvec, sc);
5416
5417         shrink_slab(sc->gfp_mask, pgdat->node_id, memcg, sc->priority);
5418
5419         if (!sc->proactive)
5420                 vmpressure(sc->gfp_mask, memcg, false, sc->nr_scanned - scanned,
5421                            sc->nr_reclaimed - reclaimed);
5422
5423         flush_reclaim_state(sc);
5424
5425         return success ? MEMCG_LRU_YOUNG : 0;
5426 }
5427
5428 #ifdef CONFIG_MEMCG
5429
5430 static void shrink_many(struct pglist_data *pgdat, struct scan_control *sc)
5431 {
5432         int op;
5433         int gen;
5434         int bin;
5435         int first_bin;
5436         struct lruvec *lruvec;
5437         struct lru_gen_folio *lrugen;
5438         struct mem_cgroup *memcg;
5439         const struct hlist_nulls_node *pos;
5440         unsigned long nr_to_reclaim = get_nr_to_reclaim(sc);
5441
5442         bin = first_bin = get_random_u32_below(MEMCG_NR_BINS);
5443 restart:
5444         op = 0;
5445         memcg = NULL;
5446         gen = get_memcg_gen(READ_ONCE(pgdat->memcg_lru.seq));
5447
5448         rcu_read_lock();
5449
5450         hlist_nulls_for_each_entry_rcu(lrugen, pos, &pgdat->memcg_lru.fifo[gen][bin], list) {
5451                 if (op) {
5452                         lru_gen_rotate_memcg(lruvec, op);
5453                         op = 0;
5454                 }
5455
5456                 mem_cgroup_put(memcg);
5457
5458                 lruvec = container_of(lrugen, struct lruvec, lrugen);
5459                 memcg = lruvec_memcg(lruvec);
5460
5461                 if (!mem_cgroup_tryget(memcg)) {
5462                         lru_gen_release_memcg(memcg);
5463                         memcg = NULL;
5464                         continue;
5465                 }
5466
5467                 rcu_read_unlock();
5468
5469                 op = shrink_one(lruvec, sc);
5470
5471                 rcu_read_lock();
5472
5473                 if (sc->nr_reclaimed >= nr_to_reclaim)
5474                         break;
5475         }
5476
5477         rcu_read_unlock();
5478
5479         if (op)
5480                 lru_gen_rotate_memcg(lruvec, op);
5481
5482         mem_cgroup_put(memcg);
5483
5484         if (sc->nr_reclaimed >= nr_to_reclaim)
5485                 return;
5486
5487         /* restart if raced with lru_gen_rotate_memcg() */
5488         if (gen != get_nulls_value(pos))
5489                 goto restart;
5490
5491         /* try the rest of the bins of the current generation */
5492         bin = get_memcg_bin(bin + 1);
5493         if (bin != first_bin)
5494                 goto restart;
5495 }
5496
5497 static void lru_gen_shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
5498 {
5499         struct blk_plug plug;
5500
5501         VM_WARN_ON_ONCE(root_reclaim(sc));
5502         VM_WARN_ON_ONCE(!sc->may_writepage || !sc->may_unmap);
5503
5504         lru_add_drain();
5505
5506         blk_start_plug(&plug);
5507
5508         set_mm_walk(NULL, sc->proactive);
5509
5510         if (try_to_shrink_lruvec(lruvec, sc))
5511                 lru_gen_rotate_memcg(lruvec, MEMCG_LRU_YOUNG);
5512
5513         clear_mm_walk();
5514
5515         blk_finish_plug(&plug);
5516 }
5517
5518 #else /* !CONFIG_MEMCG */
5519
5520 static void shrink_many(struct pglist_data *pgdat, struct scan_control *sc)
5521 {
5522         BUILD_BUG();
5523 }
5524
5525 static void lru_gen_shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
5526 {
5527         BUILD_BUG();
5528 }
5529
5530 #endif
5531
5532 static void set_initial_priority(struct pglist_data *pgdat, struct scan_control *sc)
5533 {
5534         int priority;
5535         unsigned long reclaimable;
5536         struct lruvec *lruvec = mem_cgroup_lruvec(NULL, pgdat);
5537
5538         if (sc->priority != DEF_PRIORITY || sc->nr_to_reclaim < MIN_LRU_BATCH)
5539                 return;
5540         /*
5541          * Determine the initial priority based on ((total / MEMCG_NR_GENS) >>
5542          * priority) * reclaimed_to_scanned_ratio = nr_to_reclaim, where the
5543          * estimated reclaimed_to_scanned_ratio = inactive / total.
5544          */
5545         reclaimable = node_page_state(pgdat, NR_INACTIVE_FILE);
5546         if (get_swappiness(lruvec, sc))
5547                 reclaimable += node_page_state(pgdat, NR_INACTIVE_ANON);
5548
5549         reclaimable /= MEMCG_NR_GENS;
5550
5551         /* round down reclaimable and round up sc->nr_to_reclaim */
5552         priority = fls_long(reclaimable) - 1 - fls_long(sc->nr_to_reclaim - 1);
5553
5554         sc->priority = clamp(priority, 0, DEF_PRIORITY);
5555 }
5556
5557 static void lru_gen_shrink_node(struct pglist_data *pgdat, struct scan_control *sc)
5558 {
5559         struct blk_plug plug;
5560         unsigned long reclaimed = sc->nr_reclaimed;
5561
5562         VM_WARN_ON_ONCE(!root_reclaim(sc));
5563
5564         /*
5565          * Unmapped clean folios are already prioritized. Scanning for more of
5566          * them is likely futile and can cause high reclaim latency when there
5567          * is a large number of memcgs.
5568          */
5569         if (!sc->may_writepage || !sc->may_unmap)
5570                 goto done;
5571
5572         lru_add_drain();
5573
5574         blk_start_plug(&plug);
5575
5576         set_mm_walk(pgdat, sc->proactive);
5577
5578         set_initial_priority(pgdat, sc);
5579
5580         if (current_is_kswapd())
5581                 sc->nr_reclaimed = 0;
5582
5583         if (mem_cgroup_disabled())
5584                 shrink_one(&pgdat->__lruvec, sc);
5585         else
5586                 shrink_many(pgdat, sc);
5587
5588         if (current_is_kswapd())
5589                 sc->nr_reclaimed += reclaimed;
5590
5591         clear_mm_walk();
5592
5593         blk_finish_plug(&plug);
5594 done:
5595         /* kswapd should never fail */
5596         pgdat->kswapd_failures = 0;
5597 }
5598
5599 /******************************************************************************
5600  *                          state change
5601  ******************************************************************************/
5602
5603 static bool __maybe_unused state_is_valid(struct lruvec *lruvec)
5604 {
5605         struct lru_gen_folio *lrugen = &lruvec->lrugen;
5606
5607         if (lrugen->enabled) {
5608                 enum lru_list lru;
5609
5610                 for_each_evictable_lru(lru) {
5611                         if (!list_empty(&lruvec->lists[lru]))
5612                                 return false;
5613                 }
5614         } else {
5615                 int gen, type, zone;
5616
5617                 for_each_gen_type_zone(gen, type, zone) {
5618                         if (!list_empty(&lrugen->folios[gen][type][zone]))
5619                                 return false;
5620                 }
5621         }
5622
5623         return true;
5624 }
5625
5626 static bool fill_evictable(struct lruvec *lruvec)
5627 {
5628         enum lru_list lru;
5629         int remaining = MAX_LRU_BATCH;
5630
5631         for_each_evictable_lru(lru) {
5632                 int type = is_file_lru(lru);
5633                 bool active = is_active_lru(lru);
5634                 struct list_head *head = &lruvec->lists[lru];
5635
5636                 while (!list_empty(head)) {
5637                         bool success;
5638                         struct folio *folio = lru_to_folio(head);
5639
5640                         VM_WARN_ON_ONCE_FOLIO(folio_test_unevictable(folio), folio);
5641                         VM_WARN_ON_ONCE_FOLIO(folio_test_active(folio) != active, folio);
5642                         VM_WARN_ON_ONCE_FOLIO(folio_is_file_lru(folio) != type, folio);
5643                         VM_WARN_ON_ONCE_FOLIO(folio_lru_gen(folio) != -1, folio);
5644
5645                         lruvec_del_folio(lruvec, folio);
5646                         success = lru_gen_add_folio(lruvec, folio, false);
5647                         VM_WARN_ON_ONCE(!success);
5648
5649                         if (!--remaining)
5650                                 return false;
5651                 }
5652         }
5653
5654         return true;
5655 }
5656
5657 static bool drain_evictable(struct lruvec *lruvec)
5658 {
5659         int gen, type, zone;
5660         int remaining = MAX_LRU_BATCH;
5661
5662         for_each_gen_type_zone(gen, type, zone) {
5663                 struct list_head *head = &lruvec->lrugen.folios[gen][type][zone];
5664
5665                 while (!list_empty(head)) {
5666                         bool success;
5667                         struct folio *folio = lru_to_folio(head);
5668
5669                         VM_WARN_ON_ONCE_FOLIO(folio_test_unevictable(folio), folio);
5670                         VM_WARN_ON_ONCE_FOLIO(folio_test_active(folio), folio);
5671                         VM_WARN_ON_ONCE_FOLIO(folio_is_file_lru(folio) != type, folio);
5672                         VM_WARN_ON_ONCE_FOLIO(folio_zonenum(folio) != zone, folio);
5673
5674                         success = lru_gen_del_folio(lruvec, folio, false);
5675                         VM_WARN_ON_ONCE(!success);
5676                         lruvec_add_folio(lruvec, folio);
5677
5678                         if (!--remaining)
5679                                 return false;
5680                 }
5681         }
5682
5683         return true;
5684 }
5685
5686 static void lru_gen_change_state(bool enabled)
5687 {
5688         static DEFINE_MUTEX(state_mutex);
5689
5690         struct mem_cgroup *memcg;
5691
5692         cgroup_lock();
5693         cpus_read_lock();
5694         get_online_mems();
5695         mutex_lock(&state_mutex);
5696
5697         if (enabled == lru_gen_enabled())
5698                 goto unlock;
5699
5700         if (enabled)
5701                 static_branch_enable_cpuslocked(&lru_gen_caps[LRU_GEN_CORE]);
5702         else
5703                 static_branch_disable_cpuslocked(&lru_gen_caps[LRU_GEN_CORE]);
5704
5705         memcg = mem_cgroup_iter(NULL, NULL, NULL);
5706         do {
5707                 int nid;
5708
5709                 for_each_node(nid) {
5710                         struct lruvec *lruvec = get_lruvec(memcg, nid);
5711
5712                         spin_lock_irq(&lruvec->lru_lock);
5713
5714                         VM_WARN_ON_ONCE(!seq_is_valid(lruvec));
5715                         VM_WARN_ON_ONCE(!state_is_valid(lruvec));
5716
5717                         lruvec->lrugen.enabled = enabled;
5718
5719                         while (!(enabled ? fill_evictable(lruvec) : drain_evictable(lruvec))) {
5720                                 spin_unlock_irq(&lruvec->lru_lock);
5721                                 cond_resched();
5722                                 spin_lock_irq(&lruvec->lru_lock);
5723                         }
5724
5725                         spin_unlock_irq(&lruvec->lru_lock);
5726                 }
5727
5728                 cond_resched();
5729         } while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)));
5730 unlock:
5731         mutex_unlock(&state_mutex);
5732         put_online_mems();
5733         cpus_read_unlock();
5734         cgroup_unlock();
5735 }
5736
5737 /******************************************************************************
5738  *                          sysfs interface
5739  ******************************************************************************/
5740
5741 static ssize_t min_ttl_ms_show(struct kobject *kobj, struct kobj_attribute *attr, char *buf)
5742 {
5743         return sysfs_emit(buf, "%u\n", jiffies_to_msecs(READ_ONCE(lru_gen_min_ttl)));
5744 }
5745
5746 /* see Documentation/admin-guide/mm/multigen_lru.rst for details */
5747 static ssize_t min_ttl_ms_store(struct kobject *kobj, struct kobj_attribute *attr,
5748                                 const char *buf, size_t len)
5749 {
5750         unsigned int msecs;
5751
5752         if (kstrtouint(buf, 0, &msecs))
5753                 return -EINVAL;
5754
5755         WRITE_ONCE(lru_gen_min_ttl, msecs_to_jiffies(msecs));
5756
5757         return len;
5758 }
5759
5760 static struct kobj_attribute lru_gen_min_ttl_attr = __ATTR_RW(min_ttl_ms);
5761
5762 static ssize_t enabled_show(struct kobject *kobj, struct kobj_attribute *attr, char *buf)
5763 {
5764         unsigned int caps = 0;
5765
5766         if (get_cap(LRU_GEN_CORE))
5767                 caps |= BIT(LRU_GEN_CORE);
5768
5769         if (should_walk_mmu())
5770                 caps |= BIT(LRU_GEN_MM_WALK);
5771
5772         if (should_clear_pmd_young())
5773                 caps |= BIT(LRU_GEN_NONLEAF_YOUNG);
5774
5775         return sysfs_emit(buf, "0x%04x\n", caps);
5776 }
5777
5778 /* see Documentation/admin-guide/mm/multigen_lru.rst for details */
5779 static ssize_t enabled_store(struct kobject *kobj, struct kobj_attribute *attr,
5780                              const char *buf, size_t len)
5781 {
5782         int i;
5783         unsigned int caps;
5784
5785         if (tolower(*buf) == 'n')
5786                 caps = 0;
5787         else if (tolower(*buf) == 'y')
5788                 caps = -1;
5789         else if (kstrtouint(buf, 0, &caps))
5790                 return -EINVAL;
5791
5792         for (i = 0; i < NR_LRU_GEN_CAPS; i++) {
5793                 bool enabled = caps & BIT(i);
5794
5795                 if (i == LRU_GEN_CORE)
5796                         lru_gen_change_state(enabled);
5797                 else if (enabled)
5798                         static_branch_enable(&lru_gen_caps[i]);
5799                 else
5800                         static_branch_disable(&lru_gen_caps[i]);
5801         }
5802
5803         return len;
5804 }
5805
5806 static struct kobj_attribute lru_gen_enabled_attr = __ATTR_RW(enabled);
5807
5808 static struct attribute *lru_gen_attrs[] = {
5809         &lru_gen_min_ttl_attr.attr,
5810         &lru_gen_enabled_attr.attr,
5811         NULL
5812 };
5813
5814 static const struct attribute_group lru_gen_attr_group = {
5815         .name = "lru_gen",
5816         .attrs = lru_gen_attrs,
5817 };
5818
5819 /******************************************************************************
5820  *                          debugfs interface
5821  ******************************************************************************/
5822
5823 static void *lru_gen_seq_start(struct seq_file *m, loff_t *pos)
5824 {
5825         struct mem_cgroup *memcg;
5826         loff_t nr_to_skip = *pos;
5827
5828         m->private = kvmalloc(PATH_MAX, GFP_KERNEL);
5829         if (!m->private)
5830                 return ERR_PTR(-ENOMEM);
5831
5832         memcg = mem_cgroup_iter(NULL, NULL, NULL);
5833         do {
5834                 int nid;
5835
5836                 for_each_node_state(nid, N_MEMORY) {
5837                         if (!nr_to_skip--)
5838                                 return get_lruvec(memcg, nid);
5839                 }
5840         } while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)));
5841
5842         return NULL;
5843 }
5844
5845 static void lru_gen_seq_stop(struct seq_file *m, void *v)
5846 {
5847         if (!IS_ERR_OR_NULL(v))
5848                 mem_cgroup_iter_break(NULL, lruvec_memcg(v));
5849
5850         kvfree(m->private);
5851         m->private = NULL;
5852 }
5853
5854 static void *lru_gen_seq_next(struct seq_file *m, void *v, loff_t *pos)
5855 {
5856         int nid = lruvec_pgdat(v)->node_id;
5857         struct mem_cgroup *memcg = lruvec_memcg(v);
5858
5859         ++*pos;
5860
5861         nid = next_memory_node(nid);
5862         if (nid == MAX_NUMNODES) {
5863                 memcg = mem_cgroup_iter(NULL, memcg, NULL);
5864                 if (!memcg)
5865                         return NULL;
5866
5867                 nid = first_memory_node;
5868         }
5869
5870         return get_lruvec(memcg, nid);
5871 }
5872
5873 static void lru_gen_seq_show_full(struct seq_file *m, struct lruvec *lruvec,
5874                                   unsigned long max_seq, unsigned long *min_seq,
5875                                   unsigned long seq)
5876 {
5877         int i;
5878         int type, tier;
5879         int hist = lru_hist_from_seq(seq);
5880         struct lru_gen_folio *lrugen = &lruvec->lrugen;
5881
5882         for (tier = 0; tier < MAX_NR_TIERS; tier++) {
5883                 seq_printf(m, "            %10d", tier);
5884                 for (type = 0; type < ANON_AND_FILE; type++) {
5885                         const char *s = "   ";
5886                         unsigned long n[3] = {};
5887
5888                         if (seq == max_seq) {
5889                                 s = "RT ";
5890                                 n[0] = READ_ONCE(lrugen->avg_refaulted[type][tier]);
5891                                 n[1] = READ_ONCE(lrugen->avg_total[type][tier]);
5892                         } else if (seq == min_seq[type] || NR_HIST_GENS > 1) {
5893                                 s = "rep";
5894                                 n[0] = atomic_long_read(&lrugen->refaulted[hist][type][tier]);
5895                                 n[1] = atomic_long_read(&lrugen->evicted[hist][type][tier]);
5896                                 if (tier)
5897                                         n[2] = READ_ONCE(lrugen->protected[hist][type][tier - 1]);
5898                         }
5899
5900                         for (i = 0; i < 3; i++)
5901                                 seq_printf(m, " %10lu%c", n[i], s[i]);
5902                 }
5903                 seq_putc(m, '\n');
5904         }
5905
5906         seq_puts(m, "                      ");
5907         for (i = 0; i < NR_MM_STATS; i++) {
5908                 const char *s = "      ";
5909                 unsigned long n = 0;
5910
5911                 if (seq == max_seq && NR_HIST_GENS == 1) {
5912                         s = "LOYNFA";
5913                         n = READ_ONCE(lruvec->mm_state.stats[hist][i]);
5914                 } else if (seq != max_seq && NR_HIST_GENS > 1) {
5915                         s = "loynfa";
5916                         n = READ_ONCE(lruvec->mm_state.stats[hist][i]);
5917                 }
5918
5919                 seq_printf(m, " %10lu%c", n, s[i]);
5920         }
5921         seq_putc(m, '\n');
5922 }
5923
5924 /* see Documentation/admin-guide/mm/multigen_lru.rst for details */
5925 static int lru_gen_seq_show(struct seq_file *m, void *v)
5926 {
5927         unsigned long seq;
5928         bool full = !debugfs_real_fops(m->file)->write;
5929         struct lruvec *lruvec = v;
5930         struct lru_gen_folio *lrugen = &lruvec->lrugen;
5931         int nid = lruvec_pgdat(lruvec)->node_id;
5932         struct mem_cgroup *memcg = lruvec_memcg(lruvec);
5933         DEFINE_MAX_SEQ(lruvec);
5934         DEFINE_MIN_SEQ(lruvec);
5935
5936         if (nid == first_memory_node) {
5937                 const char *path = memcg ? m->private : "";
5938
5939 #ifdef CONFIG_MEMCG
5940                 if (memcg)
5941                         cgroup_path(memcg->css.cgroup, m->private, PATH_MAX);
5942 #endif
5943                 seq_printf(m, "memcg %5hu %s\n", mem_cgroup_id(memcg), path);
5944         }
5945
5946         seq_printf(m, " node %5d\n", nid);
5947
5948         if (!full)
5949                 seq = min_seq[LRU_GEN_ANON];
5950         else if (max_seq >= MAX_NR_GENS)
5951                 seq = max_seq - MAX_NR_GENS + 1;
5952         else
5953                 seq = 0;
5954
5955         for (; seq <= max_seq; seq++) {
5956                 int type, zone;
5957                 int gen = lru_gen_from_seq(seq);
5958                 unsigned long birth = READ_ONCE(lruvec->lrugen.timestamps[gen]);
5959
5960                 seq_printf(m, " %10lu %10u", seq, jiffies_to_msecs(jiffies - birth));
5961
5962                 for (type = 0; type < ANON_AND_FILE; type++) {
5963                         unsigned long size = 0;
5964                         char mark = full && seq < min_seq[type] ? 'x' : ' ';
5965
5966                         for (zone = 0; zone < MAX_NR_ZONES; zone++)
5967                                 size += max(READ_ONCE(lrugen->nr_pages[gen][type][zone]), 0L);
5968
5969                         seq_printf(m, " %10lu%c", size, mark);
5970                 }
5971
5972                 seq_putc(m, '\n');
5973
5974                 if (full)
5975                         lru_gen_seq_show_full(m, lruvec, max_seq, min_seq, seq);
5976         }
5977
5978         return 0;
5979 }
5980
5981 static const struct seq_operations lru_gen_seq_ops = {
5982         .start = lru_gen_seq_start,
5983         .stop = lru_gen_seq_stop,
5984         .next = lru_gen_seq_next,
5985         .show = lru_gen_seq_show,
5986 };
5987
5988 static int run_aging(struct lruvec *lruvec, unsigned long seq, struct scan_control *sc,
5989                      bool can_swap, bool force_scan)
5990 {
5991         DEFINE_MAX_SEQ(lruvec);
5992         DEFINE_MIN_SEQ(lruvec);
5993
5994         if (seq < max_seq)
5995                 return 0;
5996
5997         if (seq > max_seq)
5998                 return -EINVAL;
5999
6000         if (!force_scan && min_seq[!can_swap] + MAX_NR_GENS - 1 <= max_seq)
6001                 return -ERANGE;
6002
6003         try_to_inc_max_seq(lruvec, max_seq, sc, can_swap, force_scan);
6004
6005         return 0;
6006 }
6007
6008 static int run_eviction(struct lruvec *lruvec, unsigned long seq, struct scan_control *sc,
6009                         int swappiness, unsigned long nr_to_reclaim)
6010 {
6011         DEFINE_MAX_SEQ(lruvec);
6012
6013         if (seq + MIN_NR_GENS > max_seq)
6014                 return -EINVAL;
6015
6016         sc->nr_reclaimed = 0;
6017
6018         while (!signal_pending(current)) {
6019                 DEFINE_MIN_SEQ(lruvec);
6020
6021                 if (seq < min_seq[!swappiness])
6022                         return 0;
6023
6024                 if (sc->nr_reclaimed >= nr_to_reclaim)
6025                         return 0;
6026
6027                 if (!evict_folios(lruvec, sc, swappiness))
6028                         return 0;
6029
6030                 cond_resched();
6031         }
6032
6033         return -EINTR;
6034 }
6035
6036 static int run_cmd(char cmd, int memcg_id, int nid, unsigned long seq,
6037                    struct scan_control *sc, int swappiness, unsigned long opt)
6038 {
6039         struct lruvec *lruvec;
6040         int err = -EINVAL;
6041         struct mem_cgroup *memcg = NULL;
6042
6043         if (nid < 0 || nid >= MAX_NUMNODES || !node_state(nid, N_MEMORY))
6044                 return -EINVAL;
6045
6046         if (!mem_cgroup_disabled()) {
6047                 rcu_read_lock();
6048
6049                 memcg = mem_cgroup_from_id(memcg_id);
6050                 if (!mem_cgroup_tryget(memcg))
6051                         memcg = NULL;
6052
6053                 rcu_read_unlock();
6054
6055                 if (!memcg)
6056                         return -EINVAL;
6057         }
6058
6059         if (memcg_id != mem_cgroup_id(memcg))
6060                 goto done;
6061
6062         lruvec = get_lruvec(memcg, nid);
6063
6064         if (swappiness < 0)
6065                 swappiness = get_swappiness(lruvec, sc);
6066         else if (swappiness > 200)
6067                 goto done;
6068
6069         switch (cmd) {
6070         case '+':
6071                 err = run_aging(lruvec, seq, sc, swappiness, opt);
6072                 break;
6073         case '-':
6074                 err = run_eviction(lruvec, seq, sc, swappiness, opt);
6075                 break;
6076         }
6077 done:
6078         mem_cgroup_put(memcg);
6079
6080         return err;
6081 }
6082
6083 /* see Documentation/admin-guide/mm/multigen_lru.rst for details */
6084 static ssize_t lru_gen_seq_write(struct file *file, const char __user *src,
6085                                  size_t len, loff_t *pos)
6086 {
6087         void *buf;
6088         char *cur, *next;
6089         unsigned int flags;
6090         struct blk_plug plug;
6091         int err = -EINVAL;
6092         struct scan_control sc = {
6093                 .may_writepage = true,
6094                 .may_unmap = true,
6095                 .may_swap = true,
6096                 .reclaim_idx = MAX_NR_ZONES - 1,
6097                 .gfp_mask = GFP_KERNEL,
6098         };
6099
6100         buf = kvmalloc(len + 1, GFP_KERNEL);
6101         if (!buf)
6102                 return -ENOMEM;
6103
6104         if (copy_from_user(buf, src, len)) {
6105                 kvfree(buf);
6106                 return -EFAULT;
6107         }
6108
6109         set_task_reclaim_state(current, &sc.reclaim_state);
6110         flags = memalloc_noreclaim_save();
6111         blk_start_plug(&plug);
6112         if (!set_mm_walk(NULL, true)) {
6113                 err = -ENOMEM;
6114                 goto done;
6115         }
6116
6117         next = buf;
6118         next[len] = '\0';
6119
6120         while ((cur = strsep(&next, ",;\n"))) {
6121                 int n;
6122                 int end;
6123                 char cmd;
6124                 unsigned int memcg_id;
6125                 unsigned int nid;
6126                 unsigned long seq;
6127                 unsigned int swappiness = -1;
6128                 unsigned long opt = -1;
6129
6130                 cur = skip_spaces(cur);
6131                 if (!*cur)
6132                         continue;
6133
6134                 n = sscanf(cur, "%c %u %u %lu %n %u %n %lu %n", &cmd, &memcg_id, &nid,
6135                            &seq, &end, &swappiness, &end, &opt, &end);
6136                 if (n < 4 || cur[end]) {
6137                         err = -EINVAL;
6138                         break;
6139                 }
6140
6141                 err = run_cmd(cmd, memcg_id, nid, seq, &sc, swappiness, opt);
6142                 if (err)
6143                         break;
6144         }
6145 done:
6146         clear_mm_walk();
6147         blk_finish_plug(&plug);
6148         memalloc_noreclaim_restore(flags);
6149         set_task_reclaim_state(current, NULL);
6150
6151         kvfree(buf);
6152
6153         return err ? : len;
6154 }
6155
6156 static int lru_gen_seq_open(struct inode *inode, struct file *file)
6157 {
6158         return seq_open(file, &lru_gen_seq_ops);
6159 }
6160
6161 static const struct file_operations lru_gen_rw_fops = {
6162         .open = lru_gen_seq_open,
6163         .read = seq_read,
6164         .write = lru_gen_seq_write,
6165         .llseek = seq_lseek,
6166         .release = seq_release,
6167 };
6168
6169 static const struct file_operations lru_gen_ro_fops = {
6170         .open = lru_gen_seq_open,
6171         .read = seq_read,
6172         .llseek = seq_lseek,
6173         .release = seq_release,
6174 };
6175
6176 /******************************************************************************
6177  *                          initialization
6178  ******************************************************************************/
6179
6180 void lru_gen_init_lruvec(struct lruvec *lruvec)
6181 {
6182         int i;
6183         int gen, type, zone;
6184         struct lru_gen_folio *lrugen = &lruvec->lrugen;
6185
6186         lrugen->max_seq = MIN_NR_GENS + 1;
6187         lrugen->enabled = lru_gen_enabled();
6188
6189         for (i = 0; i <= MIN_NR_GENS + 1; i++)
6190                 lrugen->timestamps[i] = jiffies;
6191
6192         for_each_gen_type_zone(gen, type, zone)
6193                 INIT_LIST_HEAD(&lrugen->folios[gen][type][zone]);
6194
6195         lruvec->mm_state.seq = MIN_NR_GENS;
6196 }
6197
6198 #ifdef CONFIG_MEMCG
6199
6200 void lru_gen_init_pgdat(struct pglist_data *pgdat)
6201 {
6202         int i, j;
6203
6204         spin_lock_init(&pgdat->memcg_lru.lock);
6205
6206         for (i = 0; i < MEMCG_NR_GENS; i++) {
6207                 for (j = 0; j < MEMCG_NR_BINS; j++)
6208                         INIT_HLIST_NULLS_HEAD(&pgdat->memcg_lru.fifo[i][j], i);
6209         }
6210 }
6211
6212 void lru_gen_init_memcg(struct mem_cgroup *memcg)
6213 {
6214         INIT_LIST_HEAD(&memcg->mm_list.fifo);
6215         spin_lock_init(&memcg->mm_list.lock);
6216 }
6217
6218 void lru_gen_exit_memcg(struct mem_cgroup *memcg)
6219 {
6220         int i;
6221         int nid;
6222
6223         VM_WARN_ON_ONCE(!list_empty(&memcg->mm_list.fifo));
6224
6225         for_each_node(nid) {
6226                 struct lruvec *lruvec = get_lruvec(memcg, nid);
6227
6228                 VM_WARN_ON_ONCE(memchr_inv(lruvec->lrugen.nr_pages, 0,
6229                                            sizeof(lruvec->lrugen.nr_pages)));
6230
6231                 lruvec->lrugen.list.next = LIST_POISON1;
6232
6233                 for (i = 0; i < NR_BLOOM_FILTERS; i++) {
6234                         bitmap_free(lruvec->mm_state.filters[i]);
6235                         lruvec->mm_state.filters[i] = NULL;
6236                 }
6237         }
6238 }
6239
6240 #endif /* CONFIG_MEMCG */
6241
6242 static int __init init_lru_gen(void)
6243 {
6244         BUILD_BUG_ON(MIN_NR_GENS + 1 >= MAX_NR_GENS);
6245         BUILD_BUG_ON(BIT(LRU_GEN_WIDTH) <= MAX_NR_GENS);
6246
6247         if (sysfs_create_group(mm_kobj, &lru_gen_attr_group))
6248                 pr_err("lru_gen: failed to create sysfs group\n");
6249
6250         debugfs_create_file("lru_gen", 0644, NULL, NULL, &lru_gen_rw_fops);
6251         debugfs_create_file("lru_gen_full", 0444, NULL, NULL, &lru_gen_ro_fops);
6252
6253         return 0;
6254 };
6255 late_initcall(init_lru_gen);
6256
6257 #else /* !CONFIG_LRU_GEN */
6258
6259 static void lru_gen_age_node(struct pglist_data *pgdat, struct scan_control *sc)
6260 {
6261 }
6262
6263 static void lru_gen_shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
6264 {
6265 }
6266
6267 static void lru_gen_shrink_node(struct pglist_data *pgdat, struct scan_control *sc)
6268 {
6269 }
6270
6271 #endif /* CONFIG_LRU_GEN */
6272
6273 static void shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
6274 {
6275         unsigned long nr[NR_LRU_LISTS];
6276         unsigned long targets[NR_LRU_LISTS];
6277         unsigned long nr_to_scan;
6278         enum lru_list lru;
6279         unsigned long nr_reclaimed = 0;
6280         unsigned long nr_to_reclaim = sc->nr_to_reclaim;
6281         bool proportional_reclaim;
6282         struct blk_plug plug;
6283
6284         if (lru_gen_enabled() && !root_reclaim(sc)) {
6285                 lru_gen_shrink_lruvec(lruvec, sc);
6286                 return;
6287         }
6288
6289         get_scan_count(lruvec, sc, nr);
6290
6291         /* Record the original scan target for proportional adjustments later */
6292         memcpy(targets, nr, sizeof(nr));
6293
6294         /*
6295          * Global reclaiming within direct reclaim at DEF_PRIORITY is a normal
6296          * event that can occur when there is little memory pressure e.g.
6297          * multiple streaming readers/writers. Hence, we do not abort scanning
6298          * when the requested number of pages are reclaimed when scanning at
6299          * DEF_PRIORITY on the assumption that the fact we are direct
6300          * reclaiming implies that kswapd is not keeping up and it is best to
6301          * do a batch of work at once. For memcg reclaim one check is made to
6302          * abort proportional reclaim if either the file or anon lru has already
6303          * dropped to zero at the first pass.
6304          */
6305         proportional_reclaim = (!cgroup_reclaim(sc) && !current_is_kswapd() &&
6306                                 sc->priority == DEF_PRIORITY);
6307
6308         blk_start_plug(&plug);
6309         while (nr[LRU_INACTIVE_ANON] || nr[LRU_ACTIVE_FILE] ||
6310                                         nr[LRU_INACTIVE_FILE]) {
6311                 unsigned long nr_anon, nr_file, percentage;
6312                 unsigned long nr_scanned;
6313
6314                 for_each_evictable_lru(lru) {
6315                         if (nr[lru]) {
6316                                 nr_to_scan = min(nr[lru], SWAP_CLUSTER_MAX);
6317                                 nr[lru] -= nr_to_scan;
6318
6319                                 nr_reclaimed += shrink_list(lru, nr_to_scan,
6320                                                             lruvec, sc);
6321                         }
6322                 }
6323
6324                 cond_resched();
6325
6326                 if (nr_reclaimed < nr_to_reclaim || proportional_reclaim)
6327                         continue;
6328
6329                 /*
6330                  * For kswapd and memcg, reclaim at least the number of pages
6331                  * requested. Ensure that the anon and file LRUs are scanned
6332                  * proportionally what was requested by get_scan_count(). We
6333                  * stop reclaiming one LRU and reduce the amount scanning
6334                  * proportional to the original scan target.
6335                  */
6336                 nr_file = nr[LRU_INACTIVE_FILE] + nr[LRU_ACTIVE_FILE];
6337                 nr_anon = nr[LRU_INACTIVE_ANON] + nr[LRU_ACTIVE_ANON];
6338
6339                 /*
6340                  * It's just vindictive to attack the larger once the smaller
6341                  * has gone to zero.  And given the way we stop scanning the
6342                  * smaller below, this makes sure that we only make one nudge
6343                  * towards proportionality once we've got nr_to_reclaim.
6344                  */
6345                 if (!nr_file || !nr_anon)
6346                         break;
6347
6348                 if (nr_file > nr_anon) {
6349                         unsigned long scan_target = targets[LRU_INACTIVE_ANON] +
6350                                                 targets[LRU_ACTIVE_ANON] + 1;
6351                         lru = LRU_BASE;
6352                         percentage = nr_anon * 100 / scan_target;
6353                 } else {
6354                         unsigned long scan_target = targets[LRU_INACTIVE_FILE] +
6355                                                 targets[LRU_ACTIVE_FILE] + 1;
6356                         lru = LRU_FILE;
6357                         percentage = nr_file * 100 / scan_target;
6358                 }
6359
6360                 /* Stop scanning the smaller of the LRU */
6361                 nr[lru] = 0;
6362                 nr[lru + LRU_ACTIVE] = 0;
6363
6364                 /*
6365                  * Recalculate the other LRU scan count based on its original
6366                  * scan target and the percentage scanning already complete
6367                  */
6368                 lru = (lru == LRU_FILE) ? LRU_BASE : LRU_FILE;
6369                 nr_scanned = targets[lru] - nr[lru];
6370                 nr[lru] = targets[lru] * (100 - percentage) / 100;
6371                 nr[lru] -= min(nr[lru], nr_scanned);
6372
6373                 lru += LRU_ACTIVE;
6374                 nr_scanned = targets[lru] - nr[lru];
6375                 nr[lru] = targets[lru] * (100 - percentage) / 100;
6376                 nr[lru] -= min(nr[lru], nr_scanned);
6377         }
6378         blk_finish_plug(&plug);
6379         sc->nr_reclaimed += nr_reclaimed;
6380
6381         /*
6382          * Even if we did not try to evict anon pages at all, we want to
6383          * rebalance the anon lru active/inactive ratio.
6384          */
6385         if (can_age_anon_pages(lruvec_pgdat(lruvec), sc) &&
6386             inactive_is_low(lruvec, LRU_INACTIVE_ANON))
6387                 shrink_active_list(SWAP_CLUSTER_MAX, lruvec,
6388                                    sc, LRU_ACTIVE_ANON);
6389 }
6390
6391 /* Use reclaim/compaction for costly allocs or under memory pressure */
6392 static bool in_reclaim_compaction(struct scan_control *sc)
6393 {
6394         if (IS_ENABLED(CONFIG_COMPACTION) && sc->order &&
6395                         (sc->order > PAGE_ALLOC_COSTLY_ORDER ||
6396                          sc->priority < DEF_PRIORITY - 2))
6397                 return true;
6398
6399         return false;
6400 }
6401
6402 /*
6403  * Reclaim/compaction is used for high-order allocation requests. It reclaims
6404  * order-0 pages before compacting the zone. should_continue_reclaim() returns
6405  * true if more pages should be reclaimed such that when the page allocator
6406  * calls try_to_compact_pages() that it will have enough free pages to succeed.
6407  * It will give up earlier than that if there is difficulty reclaiming pages.
6408  */
6409 static inline bool should_continue_reclaim(struct pglist_data *pgdat,
6410                                         unsigned long nr_reclaimed,
6411                                         struct scan_control *sc)
6412 {
6413         unsigned long pages_for_compaction;
6414         unsigned long inactive_lru_pages;
6415         int z;
6416
6417         /* If not in reclaim/compaction mode, stop */
6418         if (!in_reclaim_compaction(sc))
6419                 return false;
6420
6421         /*
6422          * Stop if we failed to reclaim any pages from the last SWAP_CLUSTER_MAX
6423          * number of pages that were scanned. This will return to the caller
6424          * with the risk reclaim/compaction and the resulting allocation attempt
6425          * fails. In the past we have tried harder for __GFP_RETRY_MAYFAIL
6426          * allocations through requiring that the full LRU list has been scanned
6427          * first, by assuming that zero delta of sc->nr_scanned means full LRU
6428          * scan, but that approximation was wrong, and there were corner cases
6429          * where always a non-zero amount of pages were scanned.
6430          */
6431         if (!nr_reclaimed)
6432                 return false;
6433
6434         /* If compaction would go ahead or the allocation would succeed, stop */
6435         for (z = 0; z <= sc->reclaim_idx; z++) {
6436                 struct zone *zone = &pgdat->node_zones[z];
6437                 if (!managed_zone(zone))
6438                         continue;
6439
6440                 /* Allocation can already succeed, nothing to do */
6441                 if (zone_watermark_ok(zone, sc->order, min_wmark_pages(zone),
6442                                       sc->reclaim_idx, 0))
6443                         return false;
6444
6445                 if (compaction_suitable(zone, sc->order, sc->reclaim_idx))
6446                         return false;
6447         }
6448
6449         /*
6450          * If we have not reclaimed enough pages for compaction and the
6451          * inactive lists are large enough, continue reclaiming
6452          */
6453         pages_for_compaction = compact_gap(sc->order);
6454         inactive_lru_pages = node_page_state(pgdat, NR_INACTIVE_FILE);
6455         if (can_reclaim_anon_pages(NULL, pgdat->node_id, sc))
6456                 inactive_lru_pages += node_page_state(pgdat, NR_INACTIVE_ANON);
6457
6458         return inactive_lru_pages > pages_for_compaction;
6459 }
6460
6461 static void shrink_node_memcgs(pg_data_t *pgdat, struct scan_control *sc)
6462 {
6463         struct mem_cgroup *target_memcg = sc->target_mem_cgroup;
6464         struct mem_cgroup *memcg;
6465
6466         memcg = mem_cgroup_iter(target_memcg, NULL, NULL);
6467         do {
6468                 struct lruvec *lruvec = mem_cgroup_lruvec(memcg, pgdat);
6469                 unsigned long reclaimed;
6470                 unsigned long scanned;
6471
6472                 /*
6473                  * This loop can become CPU-bound when target memcgs
6474                  * aren't eligible for reclaim - either because they
6475                  * don't have any reclaimable pages, or because their
6476                  * memory is explicitly protected. Avoid soft lockups.
6477                  */
6478                 cond_resched();
6479
6480                 mem_cgroup_calculate_protection(target_memcg, memcg);
6481
6482                 if (mem_cgroup_below_min(target_memcg, memcg)) {
6483                         /*
6484                          * Hard protection.
6485                          * If there is no reclaimable memory, OOM.
6486                          */
6487                         continue;
6488                 } else if (mem_cgroup_below_low(target_memcg, memcg)) {
6489                         /*
6490                          * Soft protection.
6491                          * Respect the protection only as long as
6492                          * there is an unprotected supply
6493                          * of reclaimable memory from other cgroups.
6494                          */
6495                         if (!sc->memcg_low_reclaim) {
6496                                 sc->memcg_low_skipped = 1;
6497                                 continue;
6498                         }
6499                         memcg_memory_event(memcg, MEMCG_LOW);
6500                 }
6501
6502                 reclaimed = sc->nr_reclaimed;
6503                 scanned = sc->nr_scanned;
6504
6505                 shrink_lruvec(lruvec, sc);
6506
6507                 shrink_slab(sc->gfp_mask, pgdat->node_id, memcg,
6508                             sc->priority);
6509
6510                 /* Record the group's reclaim efficiency */
6511                 if (!sc->proactive)
6512                         vmpressure(sc->gfp_mask, memcg, false,
6513                                    sc->nr_scanned - scanned,
6514                                    sc->nr_reclaimed - reclaimed);
6515
6516         } while ((memcg = mem_cgroup_iter(target_memcg, memcg, NULL)));
6517 }
6518
6519 static void shrink_node(pg_data_t *pgdat, struct scan_control *sc)
6520 {
6521         unsigned long nr_reclaimed, nr_scanned, nr_node_reclaimed;
6522         struct lruvec *target_lruvec;
6523         bool reclaimable = false;
6524
6525         if (lru_gen_enabled() && root_reclaim(sc)) {
6526                 lru_gen_shrink_node(pgdat, sc);
6527                 return;
6528         }
6529
6530         target_lruvec = mem_cgroup_lruvec(sc->target_mem_cgroup, pgdat);
6531
6532 again:
6533         memset(&sc->nr, 0, sizeof(sc->nr));
6534
6535         nr_reclaimed = sc->nr_reclaimed;
6536         nr_scanned = sc->nr_scanned;
6537
6538         prepare_scan_count(pgdat, sc);
6539
6540         shrink_node_memcgs(pgdat, sc);
6541
6542         flush_reclaim_state(sc);
6543
6544         nr_node_reclaimed = sc->nr_reclaimed - nr_reclaimed;
6545
6546         /* Record the subtree's reclaim efficiency */
6547         if (!sc->proactive)
6548                 vmpressure(sc->gfp_mask, sc->target_mem_cgroup, true,
6549                            sc->nr_scanned - nr_scanned, nr_node_reclaimed);
6550
6551         if (nr_node_reclaimed)
6552                 reclaimable = true;
6553
6554         if (current_is_kswapd()) {
6555                 /*
6556                  * If reclaim is isolating dirty pages under writeback,
6557                  * it implies that the long-lived page allocation rate
6558                  * is exceeding the page laundering rate. Either the
6559                  * global limits are not being effective at throttling
6560                  * processes due to the page distribution throughout
6561                  * zones or there is heavy usage of a slow backing
6562                  * device. The only option is to throttle from reclaim
6563                  * context which is not ideal as there is no guarantee
6564                  * the dirtying process is throttled in the same way
6565                  * balance_dirty_pages() manages.
6566                  *
6567                  * Once a node is flagged PGDAT_WRITEBACK, kswapd will
6568                  * count the number of pages under pages flagged for
6569                  * immediate reclaim and stall if any are encountered
6570                  * in the nr_immediate check below.
6571                  */
6572                 if (sc->nr.writeback && sc->nr.writeback == sc->nr.taken)
6573                         set_bit(PGDAT_WRITEBACK, &pgdat->flags);
6574
6575                 /* Allow kswapd to start writing pages during reclaim.*/
6576                 if (sc->nr.unqueued_dirty == sc->nr.file_taken)
6577                         set_bit(PGDAT_DIRTY, &pgdat->flags);
6578
6579                 /*
6580                  * If kswapd scans pages marked for immediate
6581                  * reclaim and under writeback (nr_immediate), it
6582                  * implies that pages are cycling through the LRU
6583                  * faster than they are written so forcibly stall
6584                  * until some pages complete writeback.
6585                  */
6586                 if (sc->nr.immediate)
6587                         reclaim_throttle(pgdat, VMSCAN_THROTTLE_WRITEBACK);
6588         }
6589
6590         /*
6591          * Tag a node/memcg as congested if all the dirty pages were marked
6592          * for writeback and immediate reclaim (counted in nr.congested).
6593          *
6594          * Legacy memcg will stall in page writeback so avoid forcibly
6595          * stalling in reclaim_throttle().
6596          */
6597         if (sc->nr.dirty && sc->nr.dirty == sc->nr.congested) {
6598                 if (cgroup_reclaim(sc) && writeback_throttling_sane(sc))
6599                         set_bit(LRUVEC_CGROUP_CONGESTED, &target_lruvec->flags);
6600
6601                 if (current_is_kswapd())
6602                         set_bit(LRUVEC_NODE_CONGESTED, &target_lruvec->flags);
6603         }
6604
6605         /*
6606          * Stall direct reclaim for IO completions if the lruvec is
6607          * node is congested. Allow kswapd to continue until it
6608          * starts encountering unqueued dirty pages or cycling through
6609          * the LRU too quickly.
6610          */
6611         if (!current_is_kswapd() && current_may_throttle() &&
6612             !sc->hibernation_mode &&
6613             (test_bit(LRUVEC_CGROUP_CONGESTED, &target_lruvec->flags) ||
6614              test_bit(LRUVEC_NODE_CONGESTED, &target_lruvec->flags)))
6615                 reclaim_throttle(pgdat, VMSCAN_THROTTLE_CONGESTED);
6616
6617         if (should_continue_reclaim(pgdat, nr_node_reclaimed, sc))
6618                 goto again;
6619
6620         /*
6621          * Kswapd gives up on balancing particular nodes after too
6622          * many failures to reclaim anything from them and goes to
6623          * sleep. On reclaim progress, reset the failure counter. A
6624          * successful direct reclaim run will revive a dormant kswapd.
6625          */
6626         if (reclaimable)
6627                 pgdat->kswapd_failures = 0;
6628 }
6629
6630 /*
6631  * Returns true if compaction should go ahead for a costly-order request, or
6632  * the allocation would already succeed without compaction. Return false if we
6633  * should reclaim first.
6634  */
6635 static inline bool compaction_ready(struct zone *zone, struct scan_control *sc)
6636 {
6637         unsigned long watermark;
6638
6639         /* Allocation can already succeed, nothing to do */
6640         if (zone_watermark_ok(zone, sc->order, min_wmark_pages(zone),
6641                               sc->reclaim_idx, 0))
6642                 return true;
6643
6644         /* Compaction cannot yet proceed. Do reclaim. */
6645         if (!compaction_suitable(zone, sc->order, sc->reclaim_idx))
6646                 return false;
6647
6648         /*
6649          * Compaction is already possible, but it takes time to run and there
6650          * are potentially other callers using the pages just freed. So proceed
6651          * with reclaim to make a buffer of free pages available to give
6652          * compaction a reasonable chance of completing and allocating the page.
6653          * Note that we won't actually reclaim the whole buffer in one attempt
6654          * as the target watermark in should_continue_reclaim() is lower. But if
6655          * we are already above the high+gap watermark, don't reclaim at all.
6656          */
6657         watermark = high_wmark_pages(zone) + compact_gap(sc->order);
6658
6659         return zone_watermark_ok_safe(zone, 0, watermark, sc->reclaim_idx);
6660 }
6661
6662 static void consider_reclaim_throttle(pg_data_t *pgdat, struct scan_control *sc)
6663 {
6664         /*
6665          * If reclaim is making progress greater than 12% efficiency then
6666          * wake all the NOPROGRESS throttled tasks.
6667          */
6668         if (sc->nr_reclaimed > (sc->nr_scanned >> 3)) {
6669                 wait_queue_head_t *wqh;
6670
6671                 wqh = &pgdat->reclaim_wait[VMSCAN_THROTTLE_NOPROGRESS];
6672                 if (waitqueue_active(wqh))
6673                         wake_up(wqh);
6674
6675                 return;
6676         }
6677
6678         /*
6679          * Do not throttle kswapd or cgroup reclaim on NOPROGRESS as it will
6680          * throttle on VMSCAN_THROTTLE_WRITEBACK if there are too many pages
6681          * under writeback and marked for immediate reclaim at the tail of the
6682          * LRU.
6683          */
6684         if (current_is_kswapd() || cgroup_reclaim(sc))
6685                 return;
6686
6687         /* Throttle if making no progress at high prioities. */
6688         if (sc->priority == 1 && !sc->nr_reclaimed)
6689                 reclaim_throttle(pgdat, VMSCAN_THROTTLE_NOPROGRESS);
6690 }
6691
6692 /*
6693  * This is the direct reclaim path, for page-allocating processes.  We only
6694  * try to reclaim pages from zones which will satisfy the caller's allocation
6695  * request.
6696  *
6697  * If a zone is deemed to be full of pinned pages then just give it a light
6698  * scan then give up on it.
6699  */
6700 static void shrink_zones(struct zonelist *zonelist, struct scan_control *sc)
6701 {
6702         struct zoneref *z;
6703         struct zone *zone;
6704         unsigned long nr_soft_reclaimed;
6705         unsigned long nr_soft_scanned;
6706         gfp_t orig_mask;
6707         pg_data_t *last_pgdat = NULL;
6708         pg_data_t *first_pgdat = NULL;
6709
6710         /*
6711          * If the number of buffer_heads in the machine exceeds the maximum
6712          * allowed level, force direct reclaim to scan the highmem zone as
6713          * highmem pages could be pinning lowmem pages storing buffer_heads
6714          */
6715         orig_mask = sc->gfp_mask;
6716         if (buffer_heads_over_limit) {
6717                 sc->gfp_mask |= __GFP_HIGHMEM;
6718                 sc->reclaim_idx = gfp_zone(sc->gfp_mask);
6719         }
6720
6721         for_each_zone_zonelist_nodemask(zone, z, zonelist,
6722                                         sc->reclaim_idx, sc->nodemask) {
6723                 /*
6724                  * Take care memory controller reclaiming has small influence
6725                  * to global LRU.
6726                  */
6727                 if (!cgroup_reclaim(sc)) {
6728                         if (!cpuset_zone_allowed(zone,
6729                                                  GFP_KERNEL | __GFP_HARDWALL))
6730                                 continue;
6731
6732                         /*
6733                          * If we already have plenty of memory free for
6734                          * compaction in this zone, don't free any more.
6735                          * Even though compaction is invoked for any
6736                          * non-zero order, only frequent costly order
6737                          * reclamation is disruptive enough to become a
6738                          * noticeable problem, like transparent huge
6739                          * page allocations.
6740                          */
6741                         if (IS_ENABLED(CONFIG_COMPACTION) &&
6742                             sc->order > PAGE_ALLOC_COSTLY_ORDER &&
6743                             compaction_ready(zone, sc)) {
6744                                 sc->compaction_ready = true;
6745                                 continue;
6746                         }
6747
6748                         /*
6749                          * Shrink each node in the zonelist once. If the
6750                          * zonelist is ordered by zone (not the default) then a
6751                          * node may be shrunk multiple times but in that case
6752                          * the user prefers lower zones being preserved.
6753                          */
6754                         if (zone->zone_pgdat == last_pgdat)
6755                                 continue;
6756
6757                         /*
6758                          * This steals pages from memory cgroups over softlimit
6759                          * and returns the number of reclaimed pages and
6760                          * scanned pages. This works for global memory pressure
6761                          * and balancing, not for a memcg's limit.
6762                          */
6763                         nr_soft_scanned = 0;
6764                         nr_soft_reclaimed = mem_cgroup_soft_limit_reclaim(zone->zone_pgdat,
6765                                                 sc->order, sc->gfp_mask,
6766                                                 &nr_soft_scanned);
6767                         sc->nr_reclaimed += nr_soft_reclaimed;
6768                         sc->nr_scanned += nr_soft_scanned;
6769                         /* need some check for avoid more shrink_zone() */
6770                 }
6771
6772                 if (!first_pgdat)
6773                         first_pgdat = zone->zone_pgdat;
6774
6775                 /* See comment about same check for global reclaim above */
6776                 if (zone->zone_pgdat == last_pgdat)
6777                         continue;
6778                 last_pgdat = zone->zone_pgdat;
6779                 shrink_node(zone->zone_pgdat, sc);
6780         }
6781
6782         if (first_pgdat)
6783                 consider_reclaim_throttle(first_pgdat, sc);
6784
6785         /*
6786          * Restore to original mask to avoid the impact on the caller if we
6787          * promoted it to __GFP_HIGHMEM.
6788          */
6789         sc->gfp_mask = orig_mask;
6790 }
6791
6792 static void snapshot_refaults(struct mem_cgroup *target_memcg, pg_data_t *pgdat)
6793 {
6794         struct lruvec *target_lruvec;
6795         unsigned long refaults;
6796
6797         if (lru_gen_enabled())
6798                 return;
6799
6800         target_lruvec = mem_cgroup_lruvec(target_memcg, pgdat);
6801         refaults = lruvec_page_state(target_lruvec, WORKINGSET_ACTIVATE_ANON);
6802         target_lruvec->refaults[WORKINGSET_ANON] = refaults;
6803         refaults = lruvec_page_state(target_lruvec, WORKINGSET_ACTIVATE_FILE);
6804         target_lruvec->refaults[WORKINGSET_FILE] = refaults;
6805 }
6806
6807 /*
6808  * This is the main entry point to direct page reclaim.
6809  *
6810  * If a full scan of the inactive list fails to free enough memory then we
6811  * are "out of memory" and something needs to be killed.
6812  *
6813  * If the caller is !__GFP_FS then the probability of a failure is reasonably
6814  * high - the zone may be full of dirty or under-writeback pages, which this
6815  * caller can't do much about.  We kick the writeback threads and take explicit
6816  * naps in the hope that some of these pages can be written.  But if the
6817  * allocating task holds filesystem locks which prevent writeout this might not
6818  * work, and the allocation attempt will fail.
6819  *
6820  * returns:     0, if no pages reclaimed
6821  *              else, the number of pages reclaimed
6822  */
6823 static unsigned long do_try_to_free_pages(struct zonelist *zonelist,
6824                                           struct scan_control *sc)
6825 {
6826         int initial_priority = sc->priority;
6827         pg_data_t *last_pgdat;
6828         struct zoneref *z;
6829         struct zone *zone;
6830 retry:
6831         delayacct_freepages_start();
6832
6833         if (!cgroup_reclaim(sc))
6834                 __count_zid_vm_events(ALLOCSTALL, sc->reclaim_idx, 1);
6835
6836         do {
6837                 if (!sc->proactive)
6838                         vmpressure_prio(sc->gfp_mask, sc->target_mem_cgroup,
6839                                         sc->priority);
6840                 sc->nr_scanned = 0;
6841                 shrink_zones(zonelist, sc);
6842
6843                 if (sc->nr_reclaimed >= sc->nr_to_reclaim)
6844                         break;
6845
6846                 if (sc->compaction_ready)
6847                         break;
6848
6849                 /*
6850                  * If we're getting trouble reclaiming, start doing
6851                  * writepage even in laptop mode.
6852                  */
6853                 if (sc->priority < DEF_PRIORITY - 2)
6854                         sc->may_writepage = 1;
6855         } while (--sc->priority >= 0);
6856
6857         last_pgdat = NULL;
6858         for_each_zone_zonelist_nodemask(zone, z, zonelist, sc->reclaim_idx,
6859                                         sc->nodemask) {
6860                 if (zone->zone_pgdat == last_pgdat)
6861                         continue;
6862                 last_pgdat = zone->zone_pgdat;
6863
6864                 snapshot_refaults(sc->target_mem_cgroup, zone->zone_pgdat);
6865
6866                 if (cgroup_reclaim(sc)) {
6867                         struct lruvec *lruvec;
6868
6869                         lruvec = mem_cgroup_lruvec(sc->target_mem_cgroup,
6870                                                    zone->zone_pgdat);
6871                         clear_bit(LRUVEC_CGROUP_CONGESTED, &lruvec->flags);
6872                 }
6873         }
6874
6875         delayacct_freepages_end();
6876
6877         if (sc->nr_reclaimed)
6878                 return sc->nr_reclaimed;
6879
6880         /* Aborted reclaim to try compaction? don't OOM, then */
6881         if (sc->compaction_ready)
6882                 return 1;
6883
6884         /*
6885          * We make inactive:active ratio decisions based on the node's
6886          * composition of memory, but a restrictive reclaim_idx or a
6887          * memory.low cgroup setting can exempt large amounts of
6888          * memory from reclaim. Neither of which are very common, so
6889          * instead of doing costly eligibility calculations of the
6890          * entire cgroup subtree up front, we assume the estimates are
6891          * good, and retry with forcible deactivation if that fails.
6892          */
6893         if (sc->skipped_deactivate) {
6894                 sc->priority = initial_priority;
6895                 sc->force_deactivate = 1;
6896                 sc->skipped_deactivate = 0;
6897                 goto retry;
6898         }
6899
6900         /* Untapped cgroup reserves?  Don't OOM, retry. */
6901         if (sc->memcg_low_skipped) {
6902                 sc->priority = initial_priority;
6903                 sc->force_deactivate = 0;
6904                 sc->memcg_low_reclaim = 1;
6905                 sc->memcg_low_skipped = 0;
6906                 goto retry;
6907         }
6908
6909         return 0;
6910 }
6911
6912 static bool allow_direct_reclaim(pg_data_t *pgdat)
6913 {
6914         struct zone *zone;
6915         unsigned long pfmemalloc_reserve = 0;
6916         unsigned long free_pages = 0;
6917         int i;
6918         bool wmark_ok;
6919
6920         if (pgdat->kswapd_failures >= MAX_RECLAIM_RETRIES)
6921                 return true;
6922
6923         for (i = 0; i <= ZONE_NORMAL; i++) {
6924                 zone = &pgdat->node_zones[i];
6925                 if (!managed_zone(zone))
6926                         continue;
6927
6928                 if (!zone_reclaimable_pages(zone))
6929                         continue;
6930
6931                 pfmemalloc_reserve += min_wmark_pages(zone);
6932                 free_pages += zone_page_state_snapshot(zone, NR_FREE_PAGES);
6933         }
6934
6935         /* If there are no reserves (unexpected config) then do not throttle */
6936         if (!pfmemalloc_reserve)
6937                 return true;
6938
6939         wmark_ok = free_pages > pfmemalloc_reserve / 2;
6940
6941         /* kswapd must be awake if processes are being throttled */
6942         if (!wmark_ok && waitqueue_active(&pgdat->kswapd_wait)) {
6943                 if (READ_ONCE(pgdat->kswapd_highest_zoneidx) > ZONE_NORMAL)
6944                         WRITE_ONCE(pgdat->kswapd_highest_zoneidx, ZONE_NORMAL);
6945
6946                 wake_up_interruptible(&pgdat->kswapd_wait);
6947         }
6948
6949         return wmark_ok;
6950 }
6951
6952 /*
6953  * Throttle direct reclaimers if backing storage is backed by the network
6954  * and the PFMEMALLOC reserve for the preferred node is getting dangerously
6955  * depleted. kswapd will continue to make progress and wake the processes
6956  * when the low watermark is reached.
6957  *
6958  * Returns true if a fatal signal was delivered during throttling. If this
6959  * happens, the page allocator should not consider triggering the OOM killer.
6960  */
6961 static bool throttle_direct_reclaim(gfp_t gfp_mask, struct zonelist *zonelist,
6962                                         nodemask_t *nodemask)
6963 {
6964         struct zoneref *z;
6965         struct zone *zone;
6966         pg_data_t *pgdat = NULL;
6967
6968         /*
6969          * Kernel threads should not be throttled as they may be indirectly
6970          * responsible for cleaning pages necessary for reclaim to make forward
6971          * progress. kjournald for example may enter direct reclaim while
6972          * committing a transaction where throttling it could forcing other
6973          * processes to block on log_wait_commit().
6974          */
6975         if (current->flags & PF_KTHREAD)
6976                 goto out;
6977
6978         /*
6979          * If a fatal signal is pending, this process should not throttle.
6980          * It should return quickly so it can exit and free its memory
6981          */
6982         if (fatal_signal_pending(current))
6983                 goto out;
6984
6985         /*
6986          * Check if the pfmemalloc reserves are ok by finding the first node
6987          * with a usable ZONE_NORMAL or lower zone. The expectation is that
6988          * GFP_KERNEL will be required for allocating network buffers when
6989          * swapping over the network so ZONE_HIGHMEM is unusable.
6990          *
6991          * Throttling is based on the first usable node and throttled processes
6992          * wait on a queue until kswapd makes progress and wakes them. There
6993          * is an affinity then between processes waking up and where reclaim
6994          * progress has been made assuming the process wakes on the same node.
6995          * More importantly, processes running on remote nodes will not compete
6996          * for remote pfmemalloc reserves and processes on different nodes
6997          * should make reasonable progress.
6998          */
6999         for_each_zone_zonelist_nodemask(zone, z, zonelist,
7000                                         gfp_zone(gfp_mask), nodemask) {
7001                 if (zone_idx(zone) > ZONE_NORMAL)
7002                         continue;
7003
7004                 /* Throttle based on the first usable node */
7005                 pgdat = zone->zone_pgdat;
7006                 if (allow_direct_reclaim(pgdat))
7007                         goto out;
7008                 break;
7009         }
7010
7011         /* If no zone was usable by the allocation flags then do not throttle */
7012         if (!pgdat)
7013                 goto out;
7014
7015         /* Account for the throttling */
7016         count_vm_event(PGSCAN_DIRECT_THROTTLE);
7017
7018         /*
7019          * If the caller cannot enter the filesystem, it's possible that it
7020          * is due to the caller holding an FS lock or performing a journal
7021          * transaction in the case of a filesystem like ext[3|4]. In this case,
7022          * it is not safe to block on pfmemalloc_wait as kswapd could be
7023          * blocked waiting on the same lock. Instead, throttle for up to a
7024          * second before continuing.
7025          */
7026         if (!(gfp_mask & __GFP_FS))
7027                 wait_event_interruptible_timeout(pgdat->pfmemalloc_wait,
7028                         allow_direct_reclaim(pgdat), HZ);
7029         else
7030                 /* Throttle until kswapd wakes the process */
7031                 wait_event_killable(zone->zone_pgdat->pfmemalloc_wait,
7032                         allow_direct_reclaim(pgdat));
7033
7034         if (fatal_signal_pending(current))
7035                 return true;
7036
7037 out:
7038         return false;
7039 }
7040
7041 unsigned long try_to_free_pages(struct zonelist *zonelist, int order,
7042                                 gfp_t gfp_mask, nodemask_t *nodemask)
7043 {
7044         unsigned long nr_reclaimed;
7045         struct scan_control sc = {
7046                 .nr_to_reclaim = SWAP_CLUSTER_MAX,
7047                 .gfp_mask = current_gfp_context(gfp_mask),
7048                 .reclaim_idx = gfp_zone(gfp_mask),
7049                 .order = order,
7050                 .nodemask = nodemask,
7051                 .priority = DEF_PRIORITY,
7052                 .may_writepage = !laptop_mode,
7053                 .may_unmap = 1,
7054                 .may_swap = 1,
7055         };
7056
7057         /*
7058          * scan_control uses s8 fields for order, priority, and reclaim_idx.
7059          * Confirm they are large enough for max values.
7060          */
7061         BUILD_BUG_ON(MAX_ORDER >= S8_MAX);
7062         BUILD_BUG_ON(DEF_PRIORITY > S8_MAX);
7063         BUILD_BUG_ON(MAX_NR_ZONES > S8_MAX);
7064
7065         /*
7066          * Do not enter reclaim if fatal signal was delivered while throttled.
7067          * 1 is returned so that the page allocator does not OOM kill at this
7068          * point.
7069          */
7070         if (throttle_direct_reclaim(sc.gfp_mask, zonelist, nodemask))
7071                 return 1;
7072
7073         set_task_reclaim_state(current, &sc.reclaim_state);
7074         trace_mm_vmscan_direct_reclaim_begin(order, sc.gfp_mask);
7075
7076         nr_reclaimed = do_try_to_free_pages(zonelist, &sc);
7077
7078         trace_mm_vmscan_direct_reclaim_end(nr_reclaimed);
7079         set_task_reclaim_state(current, NULL);
7080
7081         return nr_reclaimed;
7082 }
7083
7084 #ifdef CONFIG_MEMCG
7085
7086 /* Only used by soft limit reclaim. Do not reuse for anything else. */
7087 unsigned long mem_cgroup_shrink_node(struct mem_cgroup *memcg,
7088                                                 gfp_t gfp_mask, bool noswap,
7089                                                 pg_data_t *pgdat,
7090                                                 unsigned long *nr_scanned)
7091 {
7092         struct lruvec *lruvec = mem_cgroup_lruvec(memcg, pgdat);
7093         struct scan_control sc = {
7094                 .nr_to_reclaim = SWAP_CLUSTER_MAX,
7095                 .target_mem_cgroup = memcg,
7096                 .may_writepage = !laptop_mode,
7097                 .may_unmap = 1,
7098                 .reclaim_idx = MAX_NR_ZONES - 1,
7099                 .may_swap = !noswap,
7100         };
7101
7102         WARN_ON_ONCE(!current->reclaim_state);
7103
7104         sc.gfp_mask = (gfp_mask & GFP_RECLAIM_MASK) |
7105                         (GFP_HIGHUSER_MOVABLE & ~GFP_RECLAIM_MASK);
7106
7107         trace_mm_vmscan_memcg_softlimit_reclaim_begin(sc.order,
7108                                                       sc.gfp_mask);
7109
7110         /*
7111          * NOTE: Although we can get the priority field, using it
7112          * here is not a good idea, since it limits the pages we can scan.
7113          * if we don't reclaim here, the shrink_node from balance_pgdat
7114          * will pick up pages from other mem cgroup's as well. We hack
7115          * the priority and make it zero.
7116          */
7117         shrink_lruvec(lruvec, &sc);
7118
7119         trace_mm_vmscan_memcg_softlimit_reclaim_end(sc.nr_reclaimed);
7120
7121         *nr_scanned = sc.nr_scanned;
7122
7123         return sc.nr_reclaimed;
7124 }
7125
7126 unsigned long try_to_free_mem_cgroup_pages(struct mem_cgroup *memcg,
7127                                            unsigned long nr_pages,
7128                                            gfp_t gfp_mask,
7129                                            unsigned int reclaim_options)
7130 {
7131         unsigned long nr_reclaimed;
7132         unsigned int noreclaim_flag;
7133         struct scan_control sc = {
7134                 .nr_to_reclaim = max(nr_pages, SWAP_CLUSTER_MAX),
7135                 .gfp_mask = (current_gfp_context(gfp_mask) & GFP_RECLAIM_MASK) |
7136                                 (GFP_HIGHUSER_MOVABLE & ~GFP_RECLAIM_MASK),
7137                 .reclaim_idx = MAX_NR_ZONES - 1,
7138                 .target_mem_cgroup = memcg,
7139                 .priority = DEF_PRIORITY,
7140                 .may_writepage = !laptop_mode,
7141                 .may_unmap = 1,
7142                 .may_swap = !!(reclaim_options & MEMCG_RECLAIM_MAY_SWAP),
7143                 .proactive = !!(reclaim_options & MEMCG_RECLAIM_PROACTIVE),
7144         };
7145         /*
7146          * Traverse the ZONELIST_FALLBACK zonelist of the current node to put
7147          * equal pressure on all the nodes. This is based on the assumption that
7148          * the reclaim does not bail out early.
7149          */
7150         struct zonelist *zonelist = node_zonelist(numa_node_id(), sc.gfp_mask);
7151
7152         set_task_reclaim_state(current, &sc.reclaim_state);
7153         trace_mm_vmscan_memcg_reclaim_begin(0, sc.gfp_mask);
7154         noreclaim_flag = memalloc_noreclaim_save();
7155
7156         nr_reclaimed = do_try_to_free_pages(zonelist, &sc);
7157
7158         memalloc_noreclaim_restore(noreclaim_flag);
7159         trace_mm_vmscan_memcg_reclaim_end(nr_reclaimed);
7160         set_task_reclaim_state(current, NULL);
7161
7162         return nr_reclaimed;
7163 }
7164 #endif
7165
7166 static void kswapd_age_node(struct pglist_data *pgdat, struct scan_control *sc)
7167 {
7168         struct mem_cgroup *memcg;
7169         struct lruvec *lruvec;
7170
7171         if (lru_gen_enabled()) {
7172                 lru_gen_age_node(pgdat, sc);
7173                 return;
7174         }
7175
7176         if (!can_age_anon_pages(pgdat, sc))
7177                 return;
7178
7179         lruvec = mem_cgroup_lruvec(NULL, pgdat);
7180         if (!inactive_is_low(lruvec, LRU_INACTIVE_ANON))
7181                 return;
7182
7183         memcg = mem_cgroup_iter(NULL, NULL, NULL);
7184         do {
7185                 lruvec = mem_cgroup_lruvec(memcg, pgdat);
7186                 shrink_active_list(SWAP_CLUSTER_MAX, lruvec,
7187                                    sc, LRU_ACTIVE_ANON);
7188                 memcg = mem_cgroup_iter(NULL, memcg, NULL);
7189         } while (memcg);
7190 }
7191
7192 static bool pgdat_watermark_boosted(pg_data_t *pgdat, int highest_zoneidx)
7193 {
7194         int i;
7195         struct zone *zone;
7196
7197         /*
7198          * Check for watermark boosts top-down as the higher zones
7199          * are more likely to be boosted. Both watermarks and boosts
7200          * should not be checked at the same time as reclaim would
7201          * start prematurely when there is no boosting and a lower
7202          * zone is balanced.
7203          */
7204         for (i = highest_zoneidx; i >= 0; i--) {
7205                 zone = pgdat->node_zones + i;
7206                 if (!managed_zone(zone))
7207                         continue;
7208
7209                 if (zone->watermark_boost)
7210                         return true;
7211         }
7212
7213         return false;
7214 }
7215
7216 /*
7217  * Returns true if there is an eligible zone balanced for the request order
7218  * and highest_zoneidx
7219  */
7220 static bool pgdat_balanced(pg_data_t *pgdat, int order, int highest_zoneidx)
7221 {
7222         int i;
7223         unsigned long mark = -1;
7224         struct zone *zone;
7225
7226         /*
7227          * Check watermarks bottom-up as lower zones are more likely to
7228          * meet watermarks.
7229          */
7230         for (i = 0; i <= highest_zoneidx; i++) {
7231                 zone = pgdat->node_zones + i;
7232
7233                 if (!managed_zone(zone))
7234                         continue;
7235
7236                 if (sysctl_numa_balancing_mode & NUMA_BALANCING_MEMORY_TIERING)
7237                         mark = wmark_pages(zone, WMARK_PROMO);
7238                 else
7239                         mark = high_wmark_pages(zone);
7240                 if (zone_watermark_ok_safe(zone, order, mark, highest_zoneidx))
7241                         return true;
7242         }
7243
7244         /*
7245          * If a node has no managed zone within highest_zoneidx, it does not
7246          * need balancing by definition. This can happen if a zone-restricted
7247          * allocation tries to wake a remote kswapd.
7248          */
7249         if (mark == -1)
7250                 return true;
7251
7252         return false;
7253 }
7254
7255 /* Clear pgdat state for congested, dirty or under writeback. */
7256 static void clear_pgdat_congested(pg_data_t *pgdat)
7257 {
7258         struct lruvec *lruvec = mem_cgroup_lruvec(NULL, pgdat);
7259
7260         clear_bit(LRUVEC_NODE_CONGESTED, &lruvec->flags);
7261         clear_bit(LRUVEC_CGROUP_CONGESTED, &lruvec->flags);
7262         clear_bit(PGDAT_DIRTY, &pgdat->flags);
7263         clear_bit(PGDAT_WRITEBACK, &pgdat->flags);
7264 }
7265
7266 /*
7267  * Prepare kswapd for sleeping. This verifies that there are no processes
7268  * waiting in throttle_direct_reclaim() and that watermarks have been met.
7269  *
7270  * Returns true if kswapd is ready to sleep
7271  */
7272 static bool prepare_kswapd_sleep(pg_data_t *pgdat, int order,
7273                                 int highest_zoneidx)
7274 {
7275         /*
7276          * The throttled processes are normally woken up in balance_pgdat() as
7277          * soon as allow_direct_reclaim() is true. But there is a potential
7278          * race between when kswapd checks the watermarks and a process gets
7279          * throttled. There is also a potential race if processes get
7280          * throttled, kswapd wakes, a large process exits thereby balancing the
7281          * zones, which causes kswapd to exit balance_pgdat() before reaching
7282          * the wake up checks. If kswapd is going to sleep, no process should
7283          * be sleeping on pfmemalloc_wait, so wake them now if necessary. If
7284          * the wake up is premature, processes will wake kswapd and get
7285          * throttled again. The difference from wake ups in balance_pgdat() is
7286          * that here we are under prepare_to_wait().
7287          */
7288         if (waitqueue_active(&pgdat->pfmemalloc_wait))
7289                 wake_up_all(&pgdat->pfmemalloc_wait);
7290
7291         /* Hopeless node, leave it to direct reclaim */
7292         if (pgdat->kswapd_failures >= MAX_RECLAIM_RETRIES)
7293                 return true;
7294
7295         if (pgdat_balanced(pgdat, order, highest_zoneidx)) {
7296                 clear_pgdat_congested(pgdat);
7297                 return true;
7298         }
7299
7300         return false;
7301 }
7302
7303 /*
7304  * kswapd shrinks a node of pages that are at or below the highest usable
7305  * zone that is currently unbalanced.
7306  *
7307  * Returns true if kswapd scanned at least the requested number of pages to
7308  * reclaim or if the lack of progress was due to pages under writeback.
7309  * This is used to determine if the scanning priority needs to be raised.
7310  */
7311 static bool kswapd_shrink_node(pg_data_t *pgdat,
7312                                struct scan_control *sc)
7313 {
7314         struct zone *zone;
7315         int z;
7316
7317         /* Reclaim a number of pages proportional to the number of zones */
7318         sc->nr_to_reclaim = 0;
7319         for (z = 0; z <= sc->reclaim_idx; z++) {
7320                 zone = pgdat->node_zones + z;
7321                 if (!managed_zone(zone))
7322                         continue;
7323
7324                 sc->nr_to_reclaim += max(high_wmark_pages(zone), SWAP_CLUSTER_MAX);
7325         }
7326
7327         /*
7328          * Historically care was taken to put equal pressure on all zones but
7329          * now pressure is applied based on node LRU order.
7330          */
7331         shrink_node(pgdat, sc);
7332
7333         /*
7334          * Fragmentation may mean that the system cannot be rebalanced for
7335          * high-order allocations. If twice the allocation size has been
7336          * reclaimed then recheck watermarks only at order-0 to prevent
7337          * excessive reclaim. Assume that a process requested a high-order
7338          * can direct reclaim/compact.
7339          */
7340         if (sc->order && sc->nr_reclaimed >= compact_gap(sc->order))
7341                 sc->order = 0;
7342
7343         return sc->nr_scanned >= sc->nr_to_reclaim;
7344 }
7345
7346 /* Page allocator PCP high watermark is lowered if reclaim is active. */
7347 static inline void
7348 update_reclaim_active(pg_data_t *pgdat, int highest_zoneidx, bool active)
7349 {
7350         int i;
7351         struct zone *zone;
7352
7353         for (i = 0; i <= highest_zoneidx; i++) {
7354                 zone = pgdat->node_zones + i;
7355
7356                 if (!managed_zone(zone))
7357                         continue;
7358
7359                 if (active)
7360                         set_bit(ZONE_RECLAIM_ACTIVE, &zone->flags);
7361                 else
7362                         clear_bit(ZONE_RECLAIM_ACTIVE, &zone->flags);
7363         }
7364 }
7365
7366 static inline void
7367 set_reclaim_active(pg_data_t *pgdat, int highest_zoneidx)
7368 {
7369         update_reclaim_active(pgdat, highest_zoneidx, true);
7370 }
7371
7372 static inline void
7373 clear_reclaim_active(pg_data_t *pgdat, int highest_zoneidx)
7374 {
7375         update_reclaim_active(pgdat, highest_zoneidx, false);
7376 }
7377
7378 /*
7379  * For kswapd, balance_pgdat() will reclaim pages across a node from zones
7380  * that are eligible for use by the caller until at least one zone is
7381  * balanced.
7382  *
7383  * Returns the order kswapd finished reclaiming at.
7384  *
7385  * kswapd scans the zones in the highmem->normal->dma direction.  It skips
7386  * zones which have free_pages > high_wmark_pages(zone), but once a zone is
7387  * found to have free_pages <= high_wmark_pages(zone), any page in that zone
7388  * or lower is eligible for reclaim until at least one usable zone is
7389  * balanced.
7390  */
7391 static int balance_pgdat(pg_data_t *pgdat, int order, int highest_zoneidx)
7392 {
7393         int i;
7394         unsigned long nr_soft_reclaimed;
7395         unsigned long nr_soft_scanned;
7396         unsigned long pflags;
7397         unsigned long nr_boost_reclaim;
7398         unsigned long zone_boosts[MAX_NR_ZONES] = { 0, };
7399         bool boosted;
7400         struct zone *zone;
7401         struct scan_control sc = {
7402                 .gfp_mask = GFP_KERNEL,
7403                 .order = order,
7404                 .may_unmap = 1,
7405         };
7406
7407         set_task_reclaim_state(current, &sc.reclaim_state);
7408         psi_memstall_enter(&pflags);
7409         __fs_reclaim_acquire(_THIS_IP_);
7410
7411         count_vm_event(PAGEOUTRUN);
7412
7413         /*
7414          * Account for the reclaim boost. Note that the zone boost is left in
7415          * place so that parallel allocations that are near the watermark will
7416          * stall or direct reclaim until kswapd is finished.
7417          */
7418         nr_boost_reclaim = 0;
7419         for (i = 0; i <= highest_zoneidx; i++) {
7420                 zone = pgdat->node_zones + i;
7421                 if (!managed_zone(zone))
7422                         continue;
7423
7424                 nr_boost_reclaim += zone->watermark_boost;
7425                 zone_boosts[i] = zone->watermark_boost;
7426         }
7427         boosted = nr_boost_reclaim;
7428
7429 restart:
7430         set_reclaim_active(pgdat, highest_zoneidx);
7431         sc.priority = DEF_PRIORITY;
7432         do {
7433                 unsigned long nr_reclaimed = sc.nr_reclaimed;
7434                 bool raise_priority = true;
7435                 bool balanced;
7436                 bool ret;
7437
7438                 sc.reclaim_idx = highest_zoneidx;
7439
7440                 /*
7441                  * If the number of buffer_heads exceeds the maximum allowed
7442                  * then consider reclaiming from all zones. This has a dual
7443                  * purpose -- on 64-bit systems it is expected that
7444                  * buffer_heads are stripped during active rotation. On 32-bit
7445                  * systems, highmem pages can pin lowmem memory and shrinking
7446                  * buffers can relieve lowmem pressure. Reclaim may still not
7447                  * go ahead if all eligible zones for the original allocation
7448                  * request are balanced to avoid excessive reclaim from kswapd.
7449                  */
7450                 if (buffer_heads_over_limit) {
7451                         for (i = MAX_NR_ZONES - 1; i >= 0; i--) {
7452                                 zone = pgdat->node_zones + i;
7453                                 if (!managed_zone(zone))
7454                                         continue;
7455
7456                                 sc.reclaim_idx = i;
7457                                 break;
7458                         }
7459                 }
7460
7461                 /*
7462                  * If the pgdat is imbalanced then ignore boosting and preserve
7463                  * the watermarks for a later time and restart. Note that the
7464                  * zone watermarks will be still reset at the end of balancing
7465                  * on the grounds that the normal reclaim should be enough to
7466                  * re-evaluate if boosting is required when kswapd next wakes.
7467                  */
7468                 balanced = pgdat_balanced(pgdat, sc.order, highest_zoneidx);
7469                 if (!balanced && nr_boost_reclaim) {
7470                         nr_boost_reclaim = 0;
7471                         goto restart;
7472                 }
7473
7474                 /*
7475                  * If boosting is not active then only reclaim if there are no
7476                  * eligible zones. Note that sc.reclaim_idx is not used as
7477                  * buffer_heads_over_limit may have adjusted it.
7478                  */
7479                 if (!nr_boost_reclaim && balanced)
7480                         goto out;
7481
7482                 /* Limit the priority of boosting to avoid reclaim writeback */
7483                 if (nr_boost_reclaim && sc.priority == DEF_PRIORITY - 2)
7484                         raise_priority = false;
7485
7486                 /*
7487                  * Do not writeback or swap pages for boosted reclaim. The
7488                  * intent is to relieve pressure not issue sub-optimal IO
7489                  * from reclaim context. If no pages are reclaimed, the
7490                  * reclaim will be aborted.
7491                  */
7492                 sc.may_writepage = !laptop_mode && !nr_boost_reclaim;
7493                 sc.may_swap = !nr_boost_reclaim;
7494
7495                 /*
7496                  * Do some background aging, to give pages a chance to be
7497                  * referenced before reclaiming. All pages are rotated
7498                  * regardless of classzone as this is about consistent aging.
7499                  */
7500                 kswapd_age_node(pgdat, &sc);
7501
7502                 /*
7503                  * If we're getting trouble reclaiming, start doing writepage
7504                  * even in laptop mode.
7505                  */
7506                 if (sc.priority < DEF_PRIORITY - 2)
7507                         sc.may_writepage = 1;
7508
7509                 /* Call soft limit reclaim before calling shrink_node. */
7510                 sc.nr_scanned = 0;
7511                 nr_soft_scanned = 0;
7512                 nr_soft_reclaimed = mem_cgroup_soft_limit_reclaim(pgdat, sc.order,
7513                                                 sc.gfp_mask, &nr_soft_scanned);
7514                 sc.nr_reclaimed += nr_soft_reclaimed;
7515
7516                 /*
7517                  * There should be no need to raise the scanning priority if
7518                  * enough pages are already being scanned that that high
7519                  * watermark would be met at 100% efficiency.
7520                  */
7521                 if (kswapd_shrink_node(pgdat, &sc))
7522                         raise_priority = false;
7523
7524                 /*
7525                  * If the low watermark is met there is no need for processes
7526                  * to be throttled on pfmemalloc_wait as they should not be
7527                  * able to safely make forward progress. Wake them
7528                  */
7529                 if (waitqueue_active(&pgdat->pfmemalloc_wait) &&
7530                                 allow_direct_reclaim(pgdat))
7531                         wake_up_all(&pgdat->pfmemalloc_wait);
7532
7533                 /* Check if kswapd should be suspending */
7534                 __fs_reclaim_release(_THIS_IP_);
7535                 ret = try_to_freeze();
7536                 __fs_reclaim_acquire(_THIS_IP_);
7537                 if (ret || kthread_should_stop())
7538                         break;
7539
7540                 /*
7541                  * Raise priority if scanning rate is too low or there was no
7542                  * progress in reclaiming pages
7543                  */
7544                 nr_reclaimed = sc.nr_reclaimed - nr_reclaimed;
7545                 nr_boost_reclaim -= min(nr_boost_reclaim, nr_reclaimed);
7546
7547                 /*
7548                  * If reclaim made no progress for a boost, stop reclaim as
7549                  * IO cannot be queued and it could be an infinite loop in
7550                  * extreme circumstances.
7551                  */
7552                 if (nr_boost_reclaim && !nr_reclaimed)
7553                         break;
7554
7555                 if (raise_priority || !nr_reclaimed)
7556                         sc.priority--;
7557         } while (sc.priority >= 1);
7558
7559         if (!sc.nr_reclaimed)
7560                 pgdat->kswapd_failures++;
7561
7562 out:
7563         clear_reclaim_active(pgdat, highest_zoneidx);
7564
7565         /* If reclaim was boosted, account for the reclaim done in this pass */
7566         if (boosted) {
7567                 unsigned long flags;
7568
7569                 for (i = 0; i <= highest_zoneidx; i++) {
7570                         if (!zone_boosts[i])
7571                                 continue;
7572
7573                         /* Increments are under the zone lock */
7574                         zone = pgdat->node_zones + i;
7575                         spin_lock_irqsave(&zone->lock, flags);
7576                         zone->watermark_boost -= min(zone->watermark_boost, zone_boosts[i]);
7577                         spin_unlock_irqrestore(&zone->lock, flags);
7578                 }
7579
7580                 /*
7581                  * As there is now likely space, wakeup kcompact to defragment
7582                  * pageblocks.
7583                  */
7584                 wakeup_kcompactd(pgdat, pageblock_order, highest_zoneidx);
7585         }
7586
7587         snapshot_refaults(NULL, pgdat);
7588         __fs_reclaim_release(_THIS_IP_);
7589         psi_memstall_leave(&pflags);
7590         set_task_reclaim_state(current, NULL);
7591
7592         /*
7593          * Return the order kswapd stopped reclaiming at as
7594          * prepare_kswapd_sleep() takes it into account. If another caller
7595          * entered the allocator slow path while kswapd was awake, order will
7596          * remain at the higher level.
7597          */
7598         return sc.order;
7599 }
7600
7601 /*
7602  * The pgdat->kswapd_highest_zoneidx is used to pass the highest zone index to
7603  * be reclaimed by kswapd from the waker. If the value is MAX_NR_ZONES which is
7604  * not a valid index then either kswapd runs for first time or kswapd couldn't
7605  * sleep after previous reclaim attempt (node is still unbalanced). In that
7606  * case return the zone index of the previous kswapd reclaim cycle.
7607  */
7608 static enum zone_type kswapd_highest_zoneidx(pg_data_t *pgdat,
7609                                            enum zone_type prev_highest_zoneidx)
7610 {
7611         enum zone_type curr_idx = READ_ONCE(pgdat->kswapd_highest_zoneidx);
7612
7613         return curr_idx == MAX_NR_ZONES ? prev_highest_zoneidx : curr_idx;
7614 }
7615
7616 static void kswapd_try_to_sleep(pg_data_t *pgdat, int alloc_order, int reclaim_order,
7617                                 unsigned int highest_zoneidx)
7618 {
7619         long remaining = 0;
7620         DEFINE_WAIT(wait);
7621
7622         if (freezing(current) || kthread_should_stop())
7623                 return;
7624
7625         prepare_to_wait(&pgdat->kswapd_wait, &wait, TASK_INTERRUPTIBLE);
7626
7627         /*
7628          * Try to sleep for a short interval. Note that kcompactd will only be
7629          * woken if it is possible to sleep for a short interval. This is
7630          * deliberate on the assumption that if reclaim cannot keep an
7631          * eligible zone balanced that it's also unlikely that compaction will
7632          * succeed.
7633          */
7634         if (prepare_kswapd_sleep(pgdat, reclaim_order, highest_zoneidx)) {
7635                 /*
7636                  * Compaction records what page blocks it recently failed to
7637                  * isolate pages from and skips them in the future scanning.
7638                  * When kswapd is going to sleep, it is reasonable to assume
7639                  * that pages and compaction may succeed so reset the cache.
7640                  */
7641                 reset_isolation_suitable(pgdat);
7642
7643                 /*
7644                  * We have freed the memory, now we should compact it to make
7645                  * allocation of the requested order possible.
7646                  */
7647                 wakeup_kcompactd(pgdat, alloc_order, highest_zoneidx);
7648
7649                 remaining = schedule_timeout(HZ/10);
7650
7651                 /*
7652                  * If woken prematurely then reset kswapd_highest_zoneidx and
7653                  * order. The values will either be from a wakeup request or
7654                  * the previous request that slept prematurely.
7655                  */
7656                 if (remaining) {
7657                         WRITE_ONCE(pgdat->kswapd_highest_zoneidx,
7658                                         kswapd_highest_zoneidx(pgdat,
7659                                                         highest_zoneidx));
7660
7661                         if (READ_ONCE(pgdat->kswapd_order) < reclaim_order)
7662                                 WRITE_ONCE(pgdat->kswapd_order, reclaim_order);
7663                 }
7664
7665                 finish_wait(&pgdat->kswapd_wait, &wait);
7666                 prepare_to_wait(&pgdat->kswapd_wait, &wait, TASK_INTERRUPTIBLE);
7667         }
7668
7669         /*
7670          * After a short sleep, check if it was a premature sleep. If not, then
7671          * go fully to sleep until explicitly woken up.
7672          */
7673         if (!remaining &&
7674             prepare_kswapd_sleep(pgdat, reclaim_order, highest_zoneidx)) {
7675                 trace_mm_vmscan_kswapd_sleep(pgdat->node_id);
7676
7677                 /*
7678                  * vmstat counters are not perfectly accurate and the estimated
7679                  * value for counters such as NR_FREE_PAGES can deviate from the
7680                  * true value by nr_online_cpus * threshold. To avoid the zone
7681                  * watermarks being breached while under pressure, we reduce the
7682                  * per-cpu vmstat threshold while kswapd is awake and restore
7683                  * them before going back to sleep.
7684                  */
7685                 set_pgdat_percpu_threshold(pgdat, calculate_normal_threshold);
7686
7687                 if (!kthread_should_stop())
7688                         schedule();
7689
7690                 set_pgdat_percpu_threshold(pgdat, calculate_pressure_threshold);
7691         } else {
7692                 if (remaining)
7693                         count_vm_event(KSWAPD_LOW_WMARK_HIT_QUICKLY);
7694                 else
7695                         count_vm_event(KSWAPD_HIGH_WMARK_HIT_QUICKLY);
7696         }
7697         finish_wait(&pgdat->kswapd_wait, &wait);
7698 }
7699
7700 /*
7701  * The background pageout daemon, started as a kernel thread
7702  * from the init process.
7703  *
7704  * This basically trickles out pages so that we have _some_
7705  * free memory available even if there is no other activity
7706  * that frees anything up. This is needed for things like routing
7707  * etc, where we otherwise might have all activity going on in
7708  * asynchronous contexts that cannot page things out.
7709  *
7710  * If there are applications that are active memory-allocators
7711  * (most normal use), this basically shouldn't matter.
7712  */
7713 static int kswapd(void *p)
7714 {
7715         unsigned int alloc_order, reclaim_order;
7716         unsigned int highest_zoneidx = MAX_NR_ZONES - 1;
7717         pg_data_t *pgdat = (pg_data_t *)p;
7718         struct task_struct *tsk = current;
7719         const struct cpumask *cpumask = cpumask_of_node(pgdat->node_id);
7720
7721         if (!cpumask_empty(cpumask))
7722                 set_cpus_allowed_ptr(tsk, cpumask);
7723
7724         /*
7725          * Tell the memory management that we're a "memory allocator",
7726          * and that if we need more memory we should get access to it
7727          * regardless (see "__alloc_pages()"). "kswapd" should
7728          * never get caught in the normal page freeing logic.
7729          *
7730          * (Kswapd normally doesn't need memory anyway, but sometimes
7731          * you need a small amount of memory in order to be able to
7732          * page out something else, and this flag essentially protects
7733          * us from recursively trying to free more memory as we're
7734          * trying to free the first piece of memory in the first place).
7735          */
7736         tsk->flags |= PF_MEMALLOC | PF_KSWAPD;
7737         set_freezable();
7738
7739         WRITE_ONCE(pgdat->kswapd_order, 0);
7740         WRITE_ONCE(pgdat->kswapd_highest_zoneidx, MAX_NR_ZONES);
7741         atomic_set(&pgdat->nr_writeback_throttled, 0);
7742         for ( ; ; ) {
7743                 bool ret;
7744
7745                 alloc_order = reclaim_order = READ_ONCE(pgdat->kswapd_order);
7746                 highest_zoneidx = kswapd_highest_zoneidx(pgdat,
7747                                                         highest_zoneidx);
7748
7749 kswapd_try_sleep:
7750                 kswapd_try_to_sleep(pgdat, alloc_order, reclaim_order,
7751                                         highest_zoneidx);
7752
7753                 /* Read the new order and highest_zoneidx */
7754                 alloc_order = READ_ONCE(pgdat->kswapd_order);
7755                 highest_zoneidx = kswapd_highest_zoneidx(pgdat,
7756                                                         highest_zoneidx);
7757                 WRITE_ONCE(pgdat->kswapd_order, 0);
7758                 WRITE_ONCE(pgdat->kswapd_highest_zoneidx, MAX_NR_ZONES);
7759
7760                 ret = try_to_freeze();
7761                 if (kthread_should_stop())
7762                         break;
7763
7764                 /*
7765                  * We can speed up thawing tasks if we don't call balance_pgdat
7766                  * after returning from the refrigerator
7767                  */
7768                 if (ret)
7769                         continue;
7770
7771                 /*
7772                  * Reclaim begins at the requested order but if a high-order
7773                  * reclaim fails then kswapd falls back to reclaiming for
7774                  * order-0. If that happens, kswapd will consider sleeping
7775                  * for the order it finished reclaiming at (reclaim_order)
7776                  * but kcompactd is woken to compact for the original
7777                  * request (alloc_order).
7778                  */
7779                 trace_mm_vmscan_kswapd_wake(pgdat->node_id, highest_zoneidx,
7780                                                 alloc_order);
7781                 reclaim_order = balance_pgdat(pgdat, alloc_order,
7782                                                 highest_zoneidx);
7783                 if (reclaim_order < alloc_order)
7784                         goto kswapd_try_sleep;
7785         }
7786
7787         tsk->flags &= ~(PF_MEMALLOC | PF_KSWAPD);
7788
7789         return 0;
7790 }
7791
7792 /*
7793  * A zone is low on free memory or too fragmented for high-order memory.  If
7794  * kswapd should reclaim (direct reclaim is deferred), wake it up for the zone's
7795  * pgdat.  It will wake up kcompactd after reclaiming memory.  If kswapd reclaim
7796  * has failed or is not needed, still wake up kcompactd if only compaction is
7797  * needed.
7798  */
7799 void wakeup_kswapd(struct zone *zone, gfp_t gfp_flags, int order,
7800                    enum zone_type highest_zoneidx)
7801 {
7802         pg_data_t *pgdat;
7803         enum zone_type curr_idx;
7804
7805         if (!managed_zone(zone))
7806                 return;
7807
7808         if (!cpuset_zone_allowed(zone, gfp_flags))
7809                 return;
7810
7811         pgdat = zone->zone_pgdat;
7812         curr_idx = READ_ONCE(pgdat->kswapd_highest_zoneidx);
7813
7814         if (curr_idx == MAX_NR_ZONES || curr_idx < highest_zoneidx)
7815                 WRITE_ONCE(pgdat->kswapd_highest_zoneidx, highest_zoneidx);
7816
7817         if (READ_ONCE(pgdat->kswapd_order) < order)
7818                 WRITE_ONCE(pgdat->kswapd_order, order);
7819
7820         if (!waitqueue_active(&pgdat->kswapd_wait))
7821                 return;
7822
7823         /* Hopeless node, leave it to direct reclaim if possible */
7824         if (pgdat->kswapd_failures >= MAX_RECLAIM_RETRIES ||
7825             (pgdat_balanced(pgdat, order, highest_zoneidx) &&
7826              !pgdat_watermark_boosted(pgdat, highest_zoneidx))) {
7827                 /*
7828                  * There may be plenty of free memory available, but it's too
7829                  * fragmented for high-order allocations.  Wake up kcompactd
7830                  * and rely on compaction_suitable() to determine if it's
7831                  * needed.  If it fails, it will defer subsequent attempts to
7832                  * ratelimit its work.
7833                  */
7834                 if (!(gfp_flags & __GFP_DIRECT_RECLAIM))
7835                         wakeup_kcompactd(pgdat, order, highest_zoneidx);
7836                 return;
7837         }
7838
7839         trace_mm_vmscan_wakeup_kswapd(pgdat->node_id, highest_zoneidx, order,
7840                                       gfp_flags);
7841         wake_up_interruptible(&pgdat->kswapd_wait);
7842 }
7843
7844 #ifdef CONFIG_HIBERNATION
7845 /*
7846  * Try to free `nr_to_reclaim' of memory, system-wide, and return the number of
7847  * freed pages.
7848  *
7849  * Rather than trying to age LRUs the aim is to preserve the overall
7850  * LRU order by reclaiming preferentially
7851  * inactive > active > active referenced > active mapped
7852  */
7853 unsigned long shrink_all_memory(unsigned long nr_to_reclaim)
7854 {
7855         struct scan_control sc = {
7856                 .nr_to_reclaim = nr_to_reclaim,
7857                 .gfp_mask = GFP_HIGHUSER_MOVABLE,
7858                 .reclaim_idx = MAX_NR_ZONES - 1,
7859                 .priority = DEF_PRIORITY,
7860                 .may_writepage = 1,
7861                 .may_unmap = 1,
7862                 .may_swap = 1,
7863                 .hibernation_mode = 1,
7864         };
7865         struct zonelist *zonelist = node_zonelist(numa_node_id(), sc.gfp_mask);
7866         unsigned long nr_reclaimed;
7867         unsigned int noreclaim_flag;
7868
7869         fs_reclaim_acquire(sc.gfp_mask);
7870         noreclaim_flag = memalloc_noreclaim_save();
7871         set_task_reclaim_state(current, &sc.reclaim_state);
7872
7873         nr_reclaimed = do_try_to_free_pages(zonelist, &sc);
7874
7875         set_task_reclaim_state(current, NULL);
7876         memalloc_noreclaim_restore(noreclaim_flag);
7877         fs_reclaim_release(sc.gfp_mask);
7878
7879         return nr_reclaimed;
7880 }
7881 #endif /* CONFIG_HIBERNATION */
7882
7883 /*
7884  * This kswapd start function will be called by init and node-hot-add.
7885  */
7886 void __meminit kswapd_run(int nid)
7887 {
7888         pg_data_t *pgdat = NODE_DATA(nid);
7889
7890         pgdat_kswapd_lock(pgdat);
7891         if (!pgdat->kswapd) {
7892                 pgdat->kswapd = kthread_run(kswapd, pgdat, "kswapd%d", nid);
7893                 if (IS_ERR(pgdat->kswapd)) {
7894                         /* failure at boot is fatal */
7895                         BUG_ON(system_state < SYSTEM_RUNNING);
7896                         pr_err("Failed to start kswapd on node %d\n", nid);
7897                         pgdat->kswapd = NULL;
7898                 }
7899         }
7900         pgdat_kswapd_unlock(pgdat);
7901 }
7902
7903 /*
7904  * Called by memory hotplug when all memory in a node is offlined.  Caller must
7905  * be holding mem_hotplug_begin/done().
7906  */
7907 void __meminit kswapd_stop(int nid)
7908 {
7909         pg_data_t *pgdat = NODE_DATA(nid);
7910         struct task_struct *kswapd;
7911
7912         pgdat_kswapd_lock(pgdat);
7913         kswapd = pgdat->kswapd;
7914         if (kswapd) {
7915                 kthread_stop(kswapd);
7916                 pgdat->kswapd = NULL;
7917         }
7918         pgdat_kswapd_unlock(pgdat);
7919 }
7920
7921 static int __init kswapd_init(void)
7922 {
7923         int nid;
7924
7925         swap_setup();
7926         for_each_node_state(nid, N_MEMORY)
7927                 kswapd_run(nid);
7928         return 0;
7929 }
7930
7931 module_init(kswapd_init)
7932
7933 #ifdef CONFIG_NUMA
7934 /*
7935  * Node reclaim mode
7936  *
7937  * If non-zero call node_reclaim when the number of free pages falls below
7938  * the watermarks.
7939  */
7940 int node_reclaim_mode __read_mostly;
7941
7942 /*
7943  * Priority for NODE_RECLAIM. This determines the fraction of pages
7944  * of a node considered for each zone_reclaim. 4 scans 1/16th of
7945  * a zone.
7946  */
7947 #define NODE_RECLAIM_PRIORITY 4
7948
7949 /*
7950  * Percentage of pages in a zone that must be unmapped for node_reclaim to
7951  * occur.
7952  */
7953 int sysctl_min_unmapped_ratio = 1;
7954
7955 /*
7956  * If the number of slab pages in a zone grows beyond this percentage then
7957  * slab reclaim needs to occur.
7958  */
7959 int sysctl_min_slab_ratio = 5;
7960
7961 static inline unsigned long node_unmapped_file_pages(struct pglist_data *pgdat)
7962 {
7963         unsigned long file_mapped = node_page_state(pgdat, NR_FILE_MAPPED);
7964         unsigned long file_lru = node_page_state(pgdat, NR_INACTIVE_FILE) +
7965                 node_page_state(pgdat, NR_ACTIVE_FILE);
7966
7967         /*
7968          * It's possible for there to be more file mapped pages than
7969          * accounted for by the pages on the file LRU lists because
7970          * tmpfs pages accounted for as ANON can also be FILE_MAPPED
7971          */
7972         return (file_lru > file_mapped) ? (file_lru - file_mapped) : 0;
7973 }
7974
7975 /* Work out how many page cache pages we can reclaim in this reclaim_mode */
7976 static unsigned long node_pagecache_reclaimable(struct pglist_data *pgdat)
7977 {
7978         unsigned long nr_pagecache_reclaimable;
7979         unsigned long delta = 0;
7980
7981         /*
7982          * If RECLAIM_UNMAP is set, then all file pages are considered
7983          * potentially reclaimable. Otherwise, we have to worry about
7984          * pages like swapcache and node_unmapped_file_pages() provides
7985          * a better estimate
7986          */
7987         if (node_reclaim_mode & RECLAIM_UNMAP)
7988                 nr_pagecache_reclaimable = node_page_state(pgdat, NR_FILE_PAGES);
7989         else
7990                 nr_pagecache_reclaimable = node_unmapped_file_pages(pgdat);
7991
7992         /* If we can't clean pages, remove dirty pages from consideration */
7993         if (!(node_reclaim_mode & RECLAIM_WRITE))
7994                 delta += node_page_state(pgdat, NR_FILE_DIRTY);
7995
7996         /* Watch for any possible underflows due to delta */
7997         if (unlikely(delta > nr_pagecache_reclaimable))
7998                 delta = nr_pagecache_reclaimable;
7999
8000         return nr_pagecache_reclaimable - delta;
8001 }
8002
8003 /*
8004  * Try to free up some pages from this node through reclaim.
8005  */
8006 static int __node_reclaim(struct pglist_data *pgdat, gfp_t gfp_mask, unsigned int order)
8007 {
8008         /* Minimum pages needed in order to stay on node */
8009         const unsigned long nr_pages = 1 << order;
8010         struct task_struct *p = current;
8011         unsigned int noreclaim_flag;
8012         struct scan_control sc = {
8013                 .nr_to_reclaim = max(nr_pages, SWAP_CLUSTER_MAX),
8014                 .gfp_mask = current_gfp_context(gfp_mask),
8015                 .order = order,
8016                 .priority = NODE_RECLAIM_PRIORITY,
8017                 .may_writepage = !!(node_reclaim_mode & RECLAIM_WRITE),
8018                 .may_unmap = !!(node_reclaim_mode & RECLAIM_UNMAP),
8019                 .may_swap = 1,
8020                 .reclaim_idx = gfp_zone(gfp_mask),
8021         };
8022         unsigned long pflags;
8023
8024         trace_mm_vmscan_node_reclaim_begin(pgdat->node_id, order,
8025                                            sc.gfp_mask);
8026
8027         cond_resched();
8028         psi_memstall_enter(&pflags);
8029         fs_reclaim_acquire(sc.gfp_mask);
8030         /*
8031          * We need to be able to allocate from the reserves for RECLAIM_UNMAP
8032          */
8033         noreclaim_flag = memalloc_noreclaim_save();
8034         set_task_reclaim_state(p, &sc.reclaim_state);
8035
8036         if (node_pagecache_reclaimable(pgdat) > pgdat->min_unmapped_pages ||
8037             node_page_state_pages(pgdat, NR_SLAB_RECLAIMABLE_B) > pgdat->min_slab_pages) {
8038                 /*
8039                  * Free memory by calling shrink node with increasing
8040                  * priorities until we have enough memory freed.
8041                  */
8042                 do {
8043                         shrink_node(pgdat, &sc);
8044                 } while (sc.nr_reclaimed < nr_pages && --sc.priority >= 0);
8045         }
8046
8047         set_task_reclaim_state(p, NULL);
8048         memalloc_noreclaim_restore(noreclaim_flag);
8049         fs_reclaim_release(sc.gfp_mask);
8050         psi_memstall_leave(&pflags);
8051
8052         trace_mm_vmscan_node_reclaim_end(sc.nr_reclaimed);
8053
8054         return sc.nr_reclaimed >= nr_pages;
8055 }
8056
8057 int node_reclaim(struct pglist_data *pgdat, gfp_t gfp_mask, unsigned int order)
8058 {
8059         int ret;
8060
8061         /*
8062          * Node reclaim reclaims unmapped file backed pages and
8063          * slab pages if we are over the defined limits.
8064          *
8065          * A small portion of unmapped file backed pages is needed for
8066          * file I/O otherwise pages read by file I/O will be immediately
8067          * thrown out if the node is overallocated. So we do not reclaim
8068          * if less than a specified percentage of the node is used by
8069          * unmapped file backed pages.
8070          */
8071         if (node_pagecache_reclaimable(pgdat) <= pgdat->min_unmapped_pages &&
8072             node_page_state_pages(pgdat, NR_SLAB_RECLAIMABLE_B) <=
8073             pgdat->min_slab_pages)
8074                 return NODE_RECLAIM_FULL;
8075
8076         /*
8077          * Do not scan if the allocation should not be delayed.
8078          */
8079         if (!gfpflags_allow_blocking(gfp_mask) || (current->flags & PF_MEMALLOC))
8080                 return NODE_RECLAIM_NOSCAN;
8081
8082         /*
8083          * Only run node reclaim on the local node or on nodes that do not
8084          * have associated processors. This will favor the local processor
8085          * over remote processors and spread off node memory allocations
8086          * as wide as possible.
8087          */
8088         if (node_state(pgdat->node_id, N_CPU) && pgdat->node_id != numa_node_id())
8089                 return NODE_RECLAIM_NOSCAN;
8090
8091         if (test_and_set_bit(PGDAT_RECLAIM_LOCKED, &pgdat->flags))
8092                 return NODE_RECLAIM_NOSCAN;
8093
8094         ret = __node_reclaim(pgdat, gfp_mask, order);
8095         clear_bit(PGDAT_RECLAIM_LOCKED, &pgdat->flags);
8096
8097         if (!ret)
8098                 count_vm_event(PGSCAN_ZONE_RECLAIM_FAILED);
8099
8100         return ret;
8101 }
8102 #endif
8103
8104 /**
8105  * check_move_unevictable_folios - Move evictable folios to appropriate zone
8106  * lru list
8107  * @fbatch: Batch of lru folios to check.
8108  *
8109  * Checks folios for evictability, if an evictable folio is in the unevictable
8110  * lru list, moves it to the appropriate evictable lru list. This function
8111  * should be only used for lru folios.
8112  */
8113 void check_move_unevictable_folios(struct folio_batch *fbatch)
8114 {
8115         struct lruvec *lruvec = NULL;
8116         int pgscanned = 0;
8117         int pgrescued = 0;
8118         int i;
8119
8120         for (i = 0; i < fbatch->nr; i++) {
8121                 struct folio *folio = fbatch->folios[i];
8122                 int nr_pages = folio_nr_pages(folio);
8123
8124                 pgscanned += nr_pages;
8125
8126                 /* block memcg migration while the folio moves between lrus */
8127                 if (!folio_test_clear_lru(folio))
8128                         continue;
8129
8130                 lruvec = folio_lruvec_relock_irq(folio, lruvec);
8131                 if (folio_evictable(folio) && folio_test_unevictable(folio)) {
8132                         lruvec_del_folio(lruvec, folio);
8133                         folio_clear_unevictable(folio);
8134                         lruvec_add_folio(lruvec, folio);
8135                         pgrescued += nr_pages;
8136                 }
8137                 folio_set_lru(folio);
8138         }
8139
8140         if (lruvec) {
8141                 __count_vm_events(UNEVICTABLE_PGRESCUED, pgrescued);
8142                 __count_vm_events(UNEVICTABLE_PGSCANNED, pgscanned);
8143                 unlock_page_lruvec_irq(lruvec);
8144         } else if (pgscanned) {
8145                 count_vm_events(UNEVICTABLE_PGSCANNED, pgscanned);
8146         }
8147 }
8148 EXPORT_SYMBOL_GPL(check_move_unevictable_folios);