Linux Audio

Check our new training course

Loading...
v6.8
   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/delay.h>
  39#include <linux/kthread.h>
  40#include <linux/freezer.h>
  41#include <linux/memcontrol.h>
  42#include <linux/migrate.h>
  43#include <linux/delayacct.h>
  44#include <linux/sysctl.h>
  45#include <linux/memory-tiers.h>
  46#include <linux/oom.h>
  47#include <linux/pagevec.h>
  48#include <linux/prefetch.h>
  49#include <linux/printk.h>
  50#include <linux/dax.h>
  51#include <linux/psi.h>
  52#include <linux/pagewalk.h>
  53#include <linux/shmem_fs.h>
  54#include <linux/ctype.h>
  55#include <linux/debugfs.h>
  56#include <linux/khugepaged.h>
  57#include <linux/rculist_nulls.h>
  58#include <linux/random.h>
  59
  60#include <asm/tlbflush.h>
  61#include <asm/div64.h>
  62
  63#include <linux/swapops.h>
  64#include <linux/balloon_compaction.h>
  65#include <linux/sched/sysctl.h>
  66
  67#include "internal.h"
  68#include "swap.h"
  69
  70#define CREATE_TRACE_POINTS
  71#include <trace/events/vmscan.h>
  72
  73struct scan_control {
  74	/* How many pages shrink_list() should reclaim */
  75	unsigned long nr_to_reclaim;
  76
  77	/*
  78	 * Nodemask of nodes allowed by the caller. If NULL, all nodes
  79	 * are scanned.
  80	 */
  81	nodemask_t	*nodemask;
  82
  83	/*
  84	 * The memory cgroup that hit its limit and as a result is the
  85	 * primary target of this reclaim invocation.
  86	 */
  87	struct mem_cgroup *target_mem_cgroup;
  88
  89	/*
  90	 * Scan pressure balancing between anon and file LRUs
  91	 */
  92	unsigned long	anon_cost;
  93	unsigned long	file_cost;
  94
  95	/* Can active folios be deactivated as part of reclaim? */
  96#define DEACTIVATE_ANON 1
  97#define DEACTIVATE_FILE 2
  98	unsigned int may_deactivate:2;
  99	unsigned int force_deactivate:1;
 100	unsigned int skipped_deactivate:1;
 101
 102	/* Writepage batching in laptop mode; RECLAIM_WRITE */
 103	unsigned int may_writepage:1;
 104
 105	/* Can mapped folios be reclaimed? */
 106	unsigned int may_unmap:1;
 107
 108	/* Can folios be swapped as part of reclaim? */
 109	unsigned int may_swap:1;
 110
 
 
 
 
 
 
 111	/* Proactive reclaim invoked by userspace through memory.reclaim */
 112	unsigned int proactive:1;
 113
 114	/*
 115	 * Cgroup memory below memory.low is protected as long as we
 116	 * don't threaten to OOM. If any cgroup is reclaimed at
 117	 * reduced force or passed over entirely due to its memory.low
 118	 * setting (memcg_low_skipped), and nothing is reclaimed as a
 119	 * result, then go back for one more cycle that reclaims the protected
 120	 * memory (memcg_low_reclaim) to avert OOM.
 121	 */
 122	unsigned int memcg_low_reclaim:1;
 123	unsigned int memcg_low_skipped:1;
 124
 125	unsigned int hibernation_mode:1;
 126
 127	/* One of the zones is ready for compaction */
 128	unsigned int compaction_ready:1;
 129
 130	/* There is easily reclaimable cold cache in the current node */
 131	unsigned int cache_trim_mode:1;
 132
 133	/* The file folios on the current node are dangerously low */
 134	unsigned int file_is_tiny:1;
 135
 136	/* Always discard instead of demoting to lower tier memory */
 137	unsigned int no_demotion:1;
 138
 139	/* Allocation order */
 140	s8 order;
 141
 142	/* Scan (total_size >> priority) pages at once */
 143	s8 priority;
 144
 145	/* The highest zone to isolate folios for reclaim from */
 146	s8 reclaim_idx;
 147
 148	/* This context's GFP mask */
 149	gfp_t gfp_mask;
 150
 151	/* Incremented by the number of inactive pages that were scanned */
 152	unsigned long nr_scanned;
 153
 154	/* Number of pages freed so far during a call to shrink_zones() */
 155	unsigned long nr_reclaimed;
 156
 157	struct {
 158		unsigned int dirty;
 159		unsigned int unqueued_dirty;
 160		unsigned int congested;
 161		unsigned int writeback;
 162		unsigned int immediate;
 163		unsigned int file_taken;
 164		unsigned int taken;
 165	} nr;
 166
 167	/* for recording the reclaimed slab by now */
 168	struct reclaim_state reclaim_state;
 169};
 170
 171#ifdef ARCH_HAS_PREFETCHW
 172#define prefetchw_prev_lru_folio(_folio, _base, _field)			\
 173	do {								\
 174		if ((_folio)->lru.prev != _base) {			\
 175			struct folio *prev;				\
 176									\
 177			prev = lru_to_folio(&(_folio->lru));		\
 178			prefetchw(&prev->_field);			\
 179		}							\
 180	} while (0)
 181#else
 182#define prefetchw_prev_lru_folio(_folio, _base, _field) do { } while (0)
 183#endif
 184
 185/*
 186 * From 0 .. 200.  Higher means more swappy.
 187 */
 188int vm_swappiness = 60;
 189
 190#ifdef CONFIG_MEMCG
 191
 192/* Returns true for reclaim through cgroup limits or cgroup interfaces. */
 193static bool cgroup_reclaim(struct scan_control *sc)
 194{
 195	return sc->target_mem_cgroup;
 196}
 197
 198/*
 199 * Returns true for reclaim on the root cgroup. This is true for direct
 200 * allocator reclaim and reclaim through cgroup interfaces on the root cgroup.
 201 */
 202static bool root_reclaim(struct scan_control *sc)
 203{
 204	return !sc->target_mem_cgroup || mem_cgroup_is_root(sc->target_mem_cgroup);
 205}
 206
 207/**
 208 * writeback_throttling_sane - is the usual dirty throttling mechanism available?
 209 * @sc: scan_control in question
 210 *
 211 * The normal page dirty throttling mechanism in balance_dirty_pages() is
 212 * completely broken with the legacy memcg and direct stalling in
 213 * shrink_folio_list() is used for throttling instead, which lacks all the
 214 * niceties such as fairness, adaptive pausing, bandwidth proportional
 215 * allocation and configurability.
 216 *
 217 * This function tests whether the vmscan currently in progress can assume
 218 * that the normal dirty throttling mechanism is operational.
 219 */
 220static bool writeback_throttling_sane(struct scan_control *sc)
 221{
 222	if (!cgroup_reclaim(sc))
 223		return true;
 224#ifdef CONFIG_CGROUP_WRITEBACK
 225	if (cgroup_subsys_on_dfl(memory_cgrp_subsys))
 226		return true;
 227#endif
 228	return false;
 229}
 230#else
 231static bool cgroup_reclaim(struct scan_control *sc)
 232{
 233	return false;
 234}
 235
 236static bool root_reclaim(struct scan_control *sc)
 237{
 238	return true;
 239}
 240
 241static bool writeback_throttling_sane(struct scan_control *sc)
 242{
 243	return true;
 244}
 245#endif
 246
 247static void set_task_reclaim_state(struct task_struct *task,
 248				   struct reclaim_state *rs)
 249{
 250	/* Check for an overwrite */
 251	WARN_ON_ONCE(rs && task->reclaim_state);
 252
 253	/* Check for the nulling of an already-nulled member */
 254	WARN_ON_ONCE(!rs && !task->reclaim_state);
 255
 256	task->reclaim_state = rs;
 257}
 258
 259/*
 260 * flush_reclaim_state(): add pages reclaimed outside of LRU-based reclaim to
 261 * scan_control->nr_reclaimed.
 262 */
 263static void flush_reclaim_state(struct scan_control *sc)
 264{
 265	/*
 266	 * Currently, reclaim_state->reclaimed includes three types of pages
 267	 * freed outside of vmscan:
 268	 * (1) Slab pages.
 269	 * (2) Clean file pages from pruned inodes (on highmem systems).
 270	 * (3) XFS freed buffer pages.
 271	 *
 272	 * For all of these cases, we cannot universally link the pages to a
 273	 * single memcg. For example, a memcg-aware shrinker can free one object
 274	 * charged to the target memcg, causing an entire page to be freed.
 275	 * If we count the entire page as reclaimed from the memcg, we end up
 276	 * overestimating the reclaimed amount (potentially under-reclaiming).
 277	 *
 278	 * Only count such pages for global reclaim to prevent under-reclaiming
 279	 * from the target memcg; preventing unnecessary retries during memcg
 280	 * charging and false positives from proactive reclaim.
 281	 *
 282	 * For uncommon cases where the freed pages were actually mostly
 283	 * charged to the target memcg, we end up underestimating the reclaimed
 284	 * amount. This should be fine. The freed pages will be uncharged
 285	 * anyway, even if they are not counted here properly, and we will be
 286	 * able to make forward progress in charging (which is usually in a
 287	 * retry loop).
 288	 *
 289	 * We can go one step further, and report the uncharged objcg pages in
 290	 * memcg reclaim, to make reporting more accurate and reduce
 291	 * underestimation, but it's probably not worth the complexity for now.
 292	 */
 293	if (current->reclaim_state && root_reclaim(sc)) {
 294		sc->nr_reclaimed += current->reclaim_state->reclaimed;
 295		current->reclaim_state->reclaimed = 0;
 296	}
 297}
 298
 299static bool can_demote(int nid, struct scan_control *sc)
 300{
 301	if (!numa_demotion_enabled)
 302		return false;
 303	if (sc && sc->no_demotion)
 304		return false;
 305	if (next_demotion_node(nid) == NUMA_NO_NODE)
 306		return false;
 307
 308	return true;
 309}
 310
 311static inline bool can_reclaim_anon_pages(struct mem_cgroup *memcg,
 312					  int nid,
 313					  struct scan_control *sc)
 314{
 315	if (memcg == NULL) {
 316		/*
 317		 * For non-memcg reclaim, is there
 318		 * space in any swap device?
 319		 */
 320		if (get_nr_swap_pages() > 0)
 321			return true;
 322	} else {
 323		/* Is the memcg below its swap limit? */
 324		if (mem_cgroup_get_nr_swap_pages(memcg) > 0)
 325			return true;
 326	}
 327
 328	/*
 329	 * The page can not be swapped.
 330	 *
 331	 * Can it be reclaimed from this node via demotion?
 332	 */
 333	return can_demote(nid, sc);
 334}
 335
 336/*
 337 * This misses isolated folios which are not accounted for to save counters.
 338 * As the data only determines if reclaim or compaction continues, it is
 339 * not expected that isolated folios will be a dominating factor.
 340 */
 341unsigned long zone_reclaimable_pages(struct zone *zone)
 342{
 343	unsigned long nr;
 344
 345	nr = zone_page_state_snapshot(zone, NR_ZONE_INACTIVE_FILE) +
 346		zone_page_state_snapshot(zone, NR_ZONE_ACTIVE_FILE);
 347	if (can_reclaim_anon_pages(NULL, zone_to_nid(zone), NULL))
 348		nr += zone_page_state_snapshot(zone, NR_ZONE_INACTIVE_ANON) +
 349			zone_page_state_snapshot(zone, NR_ZONE_ACTIVE_ANON);
 350
 351	return nr;
 352}
 353
 354/**
 355 * lruvec_lru_size -  Returns the number of pages on the given LRU list.
 356 * @lruvec: lru vector
 357 * @lru: lru to use
 358 * @zone_idx: zones to consider (use MAX_NR_ZONES - 1 for the whole LRU list)
 359 */
 360static unsigned long lruvec_lru_size(struct lruvec *lruvec, enum lru_list lru,
 361				     int zone_idx)
 362{
 363	unsigned long size = 0;
 364	int zid;
 365
 366	for (zid = 0; zid <= zone_idx; zid++) {
 367		struct zone *zone = &lruvec_pgdat(lruvec)->node_zones[zid];
 368
 369		if (!managed_zone(zone))
 370			continue;
 371
 372		if (!mem_cgroup_disabled())
 373			size += mem_cgroup_get_zone_lru_size(lruvec, lru, zid);
 374		else
 375			size += zone_page_state(zone, NR_ZONE_LRU_BASE + lru);
 376	}
 377	return size;
 378}
 379
 380static unsigned long drop_slab_node(int nid)
 381{
 382	unsigned long freed = 0;
 383	struct mem_cgroup *memcg = NULL;
 384
 385	memcg = mem_cgroup_iter(NULL, NULL, NULL);
 386	do {
 387		freed += shrink_slab(GFP_KERNEL, nid, memcg, 0);
 388	} while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)) != NULL);
 389
 390	return freed;
 391}
 392
 393void drop_slab(void)
 394{
 395	int nid;
 396	int shift = 0;
 397	unsigned long freed;
 398
 399	do {
 400		freed = 0;
 401		for_each_online_node(nid) {
 402			if (fatal_signal_pending(current))
 403				return;
 404
 405			freed += drop_slab_node(nid);
 406		}
 407	} while ((freed >> shift++) > 1);
 408}
 409
 410static int reclaimer_offset(void)
 411{
 412	BUILD_BUG_ON(PGSTEAL_DIRECT - PGSTEAL_KSWAPD !=
 413			PGDEMOTE_DIRECT - PGDEMOTE_KSWAPD);
 414	BUILD_BUG_ON(PGSTEAL_KHUGEPAGED - PGSTEAL_KSWAPD !=
 415			PGDEMOTE_KHUGEPAGED - PGDEMOTE_KSWAPD);
 416	BUILD_BUG_ON(PGSTEAL_DIRECT - PGSTEAL_KSWAPD !=
 417			PGSCAN_DIRECT - PGSCAN_KSWAPD);
 418	BUILD_BUG_ON(PGSTEAL_KHUGEPAGED - PGSTEAL_KSWAPD !=
 419			PGSCAN_KHUGEPAGED - PGSCAN_KSWAPD);
 420
 421	if (current_is_kswapd())
 422		return 0;
 423	if (current_is_khugepaged())
 424		return PGSTEAL_KHUGEPAGED - PGSTEAL_KSWAPD;
 425	return PGSTEAL_DIRECT - PGSTEAL_KSWAPD;
 426}
 427
 428static inline int is_page_cache_freeable(struct folio *folio)
 429{
 430	/*
 431	 * A freeable page cache folio is referenced only by the caller
 432	 * that isolated the folio, the page cache and optional filesystem
 433	 * private data at folio->private.
 434	 */
 435	return folio_ref_count(folio) - folio_test_private(folio) ==
 436		1 + folio_nr_pages(folio);
 437}
 438
 439/*
 440 * We detected a synchronous write error writing a folio out.  Probably
 441 * -ENOSPC.  We need to propagate that into the address_space for a subsequent
 442 * fsync(), msync() or close().
 443 *
 444 * The tricky part is that after writepage we cannot touch the mapping: nothing
 445 * prevents it from being freed up.  But we have a ref on the folio and once
 446 * that folio is locked, the mapping is pinned.
 447 *
 448 * We're allowed to run sleeping folio_lock() here because we know the caller has
 449 * __GFP_FS.
 450 */
 451static void handle_write_error(struct address_space *mapping,
 452				struct folio *folio, int error)
 453{
 454	folio_lock(folio);
 455	if (folio_mapping(folio) == mapping)
 456		mapping_set_error(mapping, error);
 457	folio_unlock(folio);
 458}
 459
 460static bool skip_throttle_noprogress(pg_data_t *pgdat)
 461{
 462	int reclaimable = 0, write_pending = 0;
 463	int i;
 464
 465	/*
 466	 * If kswapd is disabled, reschedule if necessary but do not
 467	 * throttle as the system is likely near OOM.
 468	 */
 469	if (pgdat->kswapd_failures >= MAX_RECLAIM_RETRIES)
 470		return true;
 471
 472	/*
 473	 * If there are a lot of dirty/writeback folios then do not
 474	 * throttle as throttling will occur when the folios cycle
 475	 * towards the end of the LRU if still under writeback.
 476	 */
 477	for (i = 0; i < MAX_NR_ZONES; i++) {
 478		struct zone *zone = pgdat->node_zones + i;
 479
 480		if (!managed_zone(zone))
 481			continue;
 482
 483		reclaimable += zone_reclaimable_pages(zone);
 484		write_pending += zone_page_state_snapshot(zone,
 485						  NR_ZONE_WRITE_PENDING);
 486	}
 487	if (2 * write_pending <= reclaimable)
 488		return true;
 489
 490	return false;
 491}
 492
 493void reclaim_throttle(pg_data_t *pgdat, enum vmscan_throttle_state reason)
 494{
 495	wait_queue_head_t *wqh = &pgdat->reclaim_wait[reason];
 496	long timeout, ret;
 497	DEFINE_WAIT(wait);
 498
 499	/*
 500	 * Do not throttle user workers, kthreads other than kswapd or
 501	 * workqueues. They may be required for reclaim to make
 502	 * forward progress (e.g. journalling workqueues or kthreads).
 503	 */
 504	if (!current_is_kswapd() &&
 505	    current->flags & (PF_USER_WORKER|PF_KTHREAD)) {
 506		cond_resched();
 507		return;
 508	}
 509
 510	/*
 511	 * These figures are pulled out of thin air.
 512	 * VMSCAN_THROTTLE_ISOLATED is a transient condition based on too many
 513	 * parallel reclaimers which is a short-lived event so the timeout is
 514	 * short. Failing to make progress or waiting on writeback are
 515	 * potentially long-lived events so use a longer timeout. This is shaky
 516	 * logic as a failure to make progress could be due to anything from
 517	 * writeback to a slow device to excessive referenced folios at the tail
 518	 * of the inactive LRU.
 519	 */
 520	switch(reason) {
 521	case VMSCAN_THROTTLE_WRITEBACK:
 522		timeout = HZ/10;
 523
 524		if (atomic_inc_return(&pgdat->nr_writeback_throttled) == 1) {
 525			WRITE_ONCE(pgdat->nr_reclaim_start,
 526				node_page_state(pgdat, NR_THROTTLED_WRITTEN));
 527		}
 528
 529		break;
 530	case VMSCAN_THROTTLE_CONGESTED:
 531		fallthrough;
 532	case VMSCAN_THROTTLE_NOPROGRESS:
 533		if (skip_throttle_noprogress(pgdat)) {
 534			cond_resched();
 535			return;
 536		}
 537
 538		timeout = 1;
 539
 540		break;
 541	case VMSCAN_THROTTLE_ISOLATED:
 542		timeout = HZ/50;
 543		break;
 544	default:
 545		WARN_ON_ONCE(1);
 546		timeout = HZ;
 547		break;
 548	}
 549
 550	prepare_to_wait(wqh, &wait, TASK_UNINTERRUPTIBLE);
 551	ret = schedule_timeout(timeout);
 552	finish_wait(wqh, &wait);
 553
 554	if (reason == VMSCAN_THROTTLE_WRITEBACK)
 555		atomic_dec(&pgdat->nr_writeback_throttled);
 556
 557	trace_mm_vmscan_throttled(pgdat->node_id, jiffies_to_usecs(timeout),
 558				jiffies_to_usecs(timeout - ret),
 559				reason);
 560}
 561
 562/*
 563 * Account for folios written if tasks are throttled waiting on dirty
 564 * folios to clean. If enough folios have been cleaned since throttling
 565 * started then wakeup the throttled tasks.
 566 */
 567void __acct_reclaim_writeback(pg_data_t *pgdat, struct folio *folio,
 568							int nr_throttled)
 569{
 570	unsigned long nr_written;
 571
 572	node_stat_add_folio(folio, NR_THROTTLED_WRITTEN);
 573
 574	/*
 575	 * This is an inaccurate read as the per-cpu deltas may not
 576	 * be synchronised. However, given that the system is
 577	 * writeback throttled, it is not worth taking the penalty
 578	 * of getting an accurate count. At worst, the throttle
 579	 * timeout guarantees forward progress.
 580	 */
 581	nr_written = node_page_state(pgdat, NR_THROTTLED_WRITTEN) -
 582		READ_ONCE(pgdat->nr_reclaim_start);
 583
 584	if (nr_written > SWAP_CLUSTER_MAX * nr_throttled)
 585		wake_up(&pgdat->reclaim_wait[VMSCAN_THROTTLE_WRITEBACK]);
 586}
 587
 588/* possible outcome of pageout() */
 589typedef enum {
 590	/* failed to write folio out, folio is locked */
 591	PAGE_KEEP,
 592	/* move folio to the active list, folio is locked */
 593	PAGE_ACTIVATE,
 594	/* folio has been sent to the disk successfully, folio is unlocked */
 595	PAGE_SUCCESS,
 596	/* folio is clean and locked */
 597	PAGE_CLEAN,
 598} pageout_t;
 599
 600/*
 601 * pageout is called by shrink_folio_list() for each dirty folio.
 602 * Calls ->writepage().
 603 */
 604static pageout_t pageout(struct folio *folio, struct address_space *mapping,
 605			 struct swap_iocb **plug)
 606{
 607	/*
 608	 * If the folio is dirty, only perform writeback if that write
 609	 * will be non-blocking.  To prevent this allocation from being
 610	 * stalled by pagecache activity.  But note that there may be
 611	 * stalls if we need to run get_block().  We could test
 612	 * PagePrivate for that.
 613	 *
 614	 * If this process is currently in __generic_file_write_iter() against
 615	 * this folio's queue, we can perform writeback even if that
 616	 * will block.
 617	 *
 618	 * If the folio is swapcache, write it back even if that would
 619	 * block, for some throttling. This happens by accident, because
 620	 * swap_backing_dev_info is bust: it doesn't reflect the
 621	 * congestion state of the swapdevs.  Easy to fix, if needed.
 622	 */
 623	if (!is_page_cache_freeable(folio))
 624		return PAGE_KEEP;
 625	if (!mapping) {
 626		/*
 627		 * Some data journaling orphaned folios can have
 628		 * folio->mapping == NULL while being dirty with clean buffers.
 629		 */
 630		if (folio_test_private(folio)) {
 631			if (try_to_free_buffers(folio)) {
 632				folio_clear_dirty(folio);
 633				pr_info("%s: orphaned folio\n", __func__);
 634				return PAGE_CLEAN;
 635			}
 636		}
 637		return PAGE_KEEP;
 638	}
 639	if (mapping->a_ops->writepage == NULL)
 640		return PAGE_ACTIVATE;
 641
 642	if (folio_clear_dirty_for_io(folio)) {
 643		int res;
 644		struct writeback_control wbc = {
 645			.sync_mode = WB_SYNC_NONE,
 646			.nr_to_write = SWAP_CLUSTER_MAX,
 647			.range_start = 0,
 648			.range_end = LLONG_MAX,
 649			.for_reclaim = 1,
 650			.swap_plug = plug,
 651		};
 652
 653		folio_set_reclaim(folio);
 654		res = mapping->a_ops->writepage(&folio->page, &wbc);
 655		if (res < 0)
 656			handle_write_error(mapping, folio, res);
 657		if (res == AOP_WRITEPAGE_ACTIVATE) {
 658			folio_clear_reclaim(folio);
 659			return PAGE_ACTIVATE;
 660		}
 661
 662		if (!folio_test_writeback(folio)) {
 663			/* synchronous write or broken a_ops? */
 664			folio_clear_reclaim(folio);
 665		}
 666		trace_mm_vmscan_write_folio(folio);
 667		node_stat_add_folio(folio, NR_VMSCAN_WRITE);
 668		return PAGE_SUCCESS;
 669	}
 670
 671	return PAGE_CLEAN;
 672}
 673
 674/*
 675 * Same as remove_mapping, but if the folio is removed from the mapping, it
 676 * gets returned with a refcount of 0.
 677 */
 678static int __remove_mapping(struct address_space *mapping, struct folio *folio,
 679			    bool reclaimed, struct mem_cgroup *target_memcg)
 680{
 681	int refcount;
 682	void *shadow = NULL;
 683
 684	BUG_ON(!folio_test_locked(folio));
 685	BUG_ON(mapping != folio_mapping(folio));
 686
 687	if (!folio_test_swapcache(folio))
 688		spin_lock(&mapping->host->i_lock);
 689	xa_lock_irq(&mapping->i_pages);
 690	/*
 691	 * The non racy check for a busy folio.
 692	 *
 693	 * Must be careful with the order of the tests. When someone has
 694	 * a ref to the folio, it may be possible that they dirty it then
 695	 * drop the reference. So if the dirty flag is tested before the
 696	 * refcount here, then the following race may occur:
 697	 *
 698	 * get_user_pages(&page);
 699	 * [user mapping goes away]
 700	 * write_to(page);
 701	 *				!folio_test_dirty(folio)    [good]
 702	 * folio_set_dirty(folio);
 703	 * folio_put(folio);
 704	 *				!refcount(folio)   [good, discard it]
 705	 *
 706	 * [oops, our write_to data is lost]
 707	 *
 708	 * Reversing the order of the tests ensures such a situation cannot
 709	 * escape unnoticed. The smp_rmb is needed to ensure the folio->flags
 710	 * load is not satisfied before that of folio->_refcount.
 711	 *
 712	 * Note that if the dirty flag is always set via folio_mark_dirty,
 713	 * and thus under the i_pages lock, then this ordering is not required.
 714	 */
 715	refcount = 1 + folio_nr_pages(folio);
 716	if (!folio_ref_freeze(folio, refcount))
 717		goto cannot_free;
 718	/* note: atomic_cmpxchg in folio_ref_freeze provides the smp_rmb */
 719	if (unlikely(folio_test_dirty(folio))) {
 720		folio_ref_unfreeze(folio, refcount);
 721		goto cannot_free;
 722	}
 723
 724	if (folio_test_swapcache(folio)) {
 725		swp_entry_t swap = folio->swap;
 726
 727		if (reclaimed && !mapping_exiting(mapping))
 728			shadow = workingset_eviction(folio, target_memcg);
 729		__delete_from_swap_cache(folio, swap, shadow);
 730		mem_cgroup_swapout(folio, swap);
 731		xa_unlock_irq(&mapping->i_pages);
 732		put_swap_folio(folio, swap);
 733	} else {
 734		void (*free_folio)(struct folio *);
 735
 736		free_folio = mapping->a_ops->free_folio;
 737		/*
 738		 * Remember a shadow entry for reclaimed file cache in
 739		 * order to detect refaults, thus thrashing, later on.
 740		 *
 741		 * But don't store shadows in an address space that is
 742		 * already exiting.  This is not just an optimization,
 743		 * inode reclaim needs to empty out the radix tree or
 744		 * the nodes are lost.  Don't plant shadows behind its
 745		 * back.
 746		 *
 747		 * We also don't store shadows for DAX mappings because the
 748		 * only page cache folios found in these are zero pages
 749		 * covering holes, and because we don't want to mix DAX
 750		 * exceptional entries and shadow exceptional entries in the
 751		 * same address_space.
 752		 */
 753		if (reclaimed && folio_is_file_lru(folio) &&
 754		    !mapping_exiting(mapping) && !dax_mapping(mapping))
 755			shadow = workingset_eviction(folio, target_memcg);
 756		__filemap_remove_folio(folio, shadow);
 757		xa_unlock_irq(&mapping->i_pages);
 758		if (mapping_shrinkable(mapping))
 759			inode_add_lru(mapping->host);
 760		spin_unlock(&mapping->host->i_lock);
 761
 762		if (free_folio)
 763			free_folio(folio);
 764	}
 765
 766	return 1;
 767
 768cannot_free:
 769	xa_unlock_irq(&mapping->i_pages);
 770	if (!folio_test_swapcache(folio))
 771		spin_unlock(&mapping->host->i_lock);
 772	return 0;
 773}
 774
 775/**
 776 * remove_mapping() - Attempt to remove a folio from its mapping.
 777 * @mapping: The address space.
 778 * @folio: The folio to remove.
 779 *
 780 * If the folio is dirty, under writeback or if someone else has a ref
 781 * on it, removal will fail.
 782 * Return: The number of pages removed from the mapping.  0 if the folio
 783 * could not be removed.
 784 * Context: The caller should have a single refcount on the folio and
 785 * hold its lock.
 786 */
 787long remove_mapping(struct address_space *mapping, struct folio *folio)
 788{
 789	if (__remove_mapping(mapping, folio, false, NULL)) {
 790		/*
 791		 * Unfreezing the refcount with 1 effectively
 792		 * drops the pagecache ref for us without requiring another
 793		 * atomic operation.
 794		 */
 795		folio_ref_unfreeze(folio, 1);
 796		return folio_nr_pages(folio);
 797	}
 798	return 0;
 799}
 800
 801/**
 802 * folio_putback_lru - Put previously isolated folio onto appropriate LRU list.
 803 * @folio: Folio to be returned to an LRU list.
 804 *
 805 * Add previously isolated @folio to appropriate LRU list.
 806 * The folio may still be unevictable for other reasons.
 807 *
 808 * Context: lru_lock must not be held, interrupts must be enabled.
 809 */
 810void folio_putback_lru(struct folio *folio)
 811{
 812	folio_add_lru(folio);
 813	folio_put(folio);		/* drop ref from isolate */
 814}
 815
 816enum folio_references {
 817	FOLIOREF_RECLAIM,
 818	FOLIOREF_RECLAIM_CLEAN,
 819	FOLIOREF_KEEP,
 820	FOLIOREF_ACTIVATE,
 821};
 822
 823static enum folio_references folio_check_references(struct folio *folio,
 824						  struct scan_control *sc)
 825{
 826	int referenced_ptes, referenced_folio;
 827	unsigned long vm_flags;
 828
 829	referenced_ptes = folio_referenced(folio, 1, sc->target_mem_cgroup,
 830					   &vm_flags);
 831	referenced_folio = folio_test_clear_referenced(folio);
 832
 833	/*
 834	 * The supposedly reclaimable folio was found to be in a VM_LOCKED vma.
 835	 * Let the folio, now marked Mlocked, be moved to the unevictable list.
 836	 */
 837	if (vm_flags & VM_LOCKED)
 838		return FOLIOREF_ACTIVATE;
 839
 840	/* rmap lock contention: rotate */
 841	if (referenced_ptes == -1)
 842		return FOLIOREF_KEEP;
 843
 844	if (referenced_ptes) {
 845		/*
 846		 * All mapped folios start out with page table
 847		 * references from the instantiating fault, so we need
 848		 * to look twice if a mapped file/anon folio is used more
 849		 * than once.
 850		 *
 851		 * Mark it and spare it for another trip around the
 852		 * inactive list.  Another page table reference will
 853		 * lead to its activation.
 854		 *
 855		 * Note: the mark is set for activated folios as well
 856		 * so that recently deactivated but used folios are
 857		 * quickly recovered.
 858		 */
 859		folio_set_referenced(folio);
 860
 861		if (referenced_folio || referenced_ptes > 1)
 862			return FOLIOREF_ACTIVATE;
 863
 864		/*
 865		 * Activate file-backed executable folios after first usage.
 866		 */
 867		if ((vm_flags & VM_EXEC) && folio_is_file_lru(folio))
 868			return FOLIOREF_ACTIVATE;
 869
 870		return FOLIOREF_KEEP;
 871	}
 872
 873	/* Reclaim if clean, defer dirty folios to writeback */
 874	if (referenced_folio && folio_is_file_lru(folio))
 875		return FOLIOREF_RECLAIM_CLEAN;
 876
 877	return FOLIOREF_RECLAIM;
 878}
 879
 880/* Check if a folio is dirty or under writeback */
 881static void folio_check_dirty_writeback(struct folio *folio,
 882				       bool *dirty, bool *writeback)
 883{
 884	struct address_space *mapping;
 885
 886	/*
 887	 * Anonymous folios are not handled by flushers and must be written
 888	 * from reclaim context. Do not stall reclaim based on them.
 889	 * MADV_FREE anonymous folios are put into inactive file list too.
 890	 * They could be mistakenly treated as file lru. So further anon
 891	 * test is needed.
 892	 */
 893	if (!folio_is_file_lru(folio) ||
 894	    (folio_test_anon(folio) && !folio_test_swapbacked(folio))) {
 895		*dirty = false;
 896		*writeback = false;
 897		return;
 898	}
 899
 900	/* By default assume that the folio flags are accurate */
 901	*dirty = folio_test_dirty(folio);
 902	*writeback = folio_test_writeback(folio);
 903
 904	/* Verify dirty/writeback state if the filesystem supports it */
 905	if (!folio_test_private(folio))
 906		return;
 907
 908	mapping = folio_mapping(folio);
 909	if (mapping && mapping->a_ops->is_dirty_writeback)
 910		mapping->a_ops->is_dirty_writeback(folio, dirty, writeback);
 911}
 912
 913static struct folio *alloc_demote_folio(struct folio *src,
 914		unsigned long private)
 915{
 916	struct folio *dst;
 917	nodemask_t *allowed_mask;
 918	struct migration_target_control *mtc;
 919
 920	mtc = (struct migration_target_control *)private;
 921
 922	allowed_mask = mtc->nmask;
 923	/*
 924	 * make sure we allocate from the target node first also trying to
 925	 * demote or reclaim pages from the target node via kswapd if we are
 926	 * low on free memory on target node. If we don't do this and if
 927	 * we have free memory on the slower(lower) memtier, we would start
 928	 * allocating pages from slower(lower) memory tiers without even forcing
 929	 * a demotion of cold pages from the target memtier. This can result
 930	 * in the kernel placing hot pages in slower(lower) memory tiers.
 931	 */
 932	mtc->nmask = NULL;
 933	mtc->gfp_mask |= __GFP_THISNODE;
 934	dst = alloc_migration_target(src, (unsigned long)mtc);
 935	if (dst)
 936		return dst;
 937
 938	mtc->gfp_mask &= ~__GFP_THISNODE;
 939	mtc->nmask = allowed_mask;
 940
 941	return alloc_migration_target(src, (unsigned long)mtc);
 942}
 943
 944/*
 945 * Take folios on @demote_folios and attempt to demote them to another node.
 946 * Folios which are not demoted are left on @demote_folios.
 947 */
 948static unsigned int demote_folio_list(struct list_head *demote_folios,
 949				     struct pglist_data *pgdat)
 950{
 951	int target_nid = next_demotion_node(pgdat->node_id);
 952	unsigned int nr_succeeded;
 953	nodemask_t allowed_mask;
 954
 955	struct migration_target_control mtc = {
 956		/*
 957		 * Allocate from 'node', or fail quickly and quietly.
 958		 * When this happens, 'page' will likely just be discarded
 959		 * instead of migrated.
 960		 */
 961		.gfp_mask = (GFP_HIGHUSER_MOVABLE & ~__GFP_RECLAIM) | __GFP_NOWARN |
 962			__GFP_NOMEMALLOC | GFP_NOWAIT,
 963		.nid = target_nid,
 964		.nmask = &allowed_mask
 965	};
 966
 967	if (list_empty(demote_folios))
 968		return 0;
 969
 970	if (target_nid == NUMA_NO_NODE)
 971		return 0;
 972
 973	node_get_allowed_targets(pgdat, &allowed_mask);
 974
 975	/* Demotion ignores all cpuset and mempolicy settings */
 976	migrate_pages(demote_folios, alloc_demote_folio, NULL,
 977		      (unsigned long)&mtc, MIGRATE_ASYNC, MR_DEMOTION,
 978		      &nr_succeeded);
 979
 980	mod_node_page_state(pgdat, PGDEMOTE_KSWAPD + reclaimer_offset(),
 981			    nr_succeeded);
 982
 983	return nr_succeeded;
 984}
 985
 986static bool may_enter_fs(struct folio *folio, gfp_t gfp_mask)
 987{
 988	if (gfp_mask & __GFP_FS)
 989		return true;
 990	if (!folio_test_swapcache(folio) || !(gfp_mask & __GFP_IO))
 991		return false;
 992	/*
 993	 * We can "enter_fs" for swap-cache with only __GFP_IO
 994	 * providing this isn't SWP_FS_OPS.
 995	 * ->flags can be updated non-atomicially (scan_swap_map_slots),
 996	 * but that will never affect SWP_FS_OPS, so the data_race
 997	 * is safe.
 998	 */
 999	return !data_race(folio_swap_flags(folio) & SWP_FS_OPS);
1000}
1001
1002/*
1003 * shrink_folio_list() returns the number of reclaimed pages
1004 */
1005static unsigned int shrink_folio_list(struct list_head *folio_list,
1006		struct pglist_data *pgdat, struct scan_control *sc,
1007		struct reclaim_stat *stat, bool ignore_references)
1008{
 
1009	LIST_HEAD(ret_folios);
1010	LIST_HEAD(free_folios);
1011	LIST_HEAD(demote_folios);
1012	unsigned int nr_reclaimed = 0;
1013	unsigned int pgactivate = 0;
1014	bool do_demote_pass;
1015	struct swap_iocb *plug = NULL;
1016
 
1017	memset(stat, 0, sizeof(*stat));
1018	cond_resched();
1019	do_demote_pass = can_demote(pgdat->node_id, sc);
1020
1021retry:
1022	while (!list_empty(folio_list)) {
1023		struct address_space *mapping;
1024		struct folio *folio;
1025		enum folio_references references = FOLIOREF_RECLAIM;
1026		bool dirty, writeback;
1027		unsigned int nr_pages;
1028
1029		cond_resched();
1030
1031		folio = lru_to_folio(folio_list);
1032		list_del(&folio->lru);
1033
1034		if (!folio_trylock(folio))
1035			goto keep;
1036
1037		VM_BUG_ON_FOLIO(folio_test_active(folio), folio);
1038
1039		nr_pages = folio_nr_pages(folio);
1040
1041		/* Account the number of base pages */
1042		sc->nr_scanned += nr_pages;
1043
1044		if (unlikely(!folio_evictable(folio)))
1045			goto activate_locked;
1046
1047		if (!sc->may_unmap && folio_mapped(folio))
1048			goto keep_locked;
1049
1050		/* folio_update_gen() tried to promote this page? */
1051		if (lru_gen_enabled() && !ignore_references &&
1052		    folio_mapped(folio) && folio_test_referenced(folio))
1053			goto keep_locked;
1054
1055		/*
1056		 * The number of dirty pages determines if a node is marked
1057		 * reclaim_congested. kswapd will stall and start writing
1058		 * folios if the tail of the LRU is all dirty unqueued folios.
1059		 */
1060		folio_check_dirty_writeback(folio, &dirty, &writeback);
1061		if (dirty || writeback)
1062			stat->nr_dirty += nr_pages;
1063
1064		if (dirty && !writeback)
1065			stat->nr_unqueued_dirty += nr_pages;
1066
1067		/*
1068		 * Treat this folio as congested if folios are cycling
1069		 * through the LRU so quickly that the folios marked
1070		 * for immediate reclaim are making it to the end of
1071		 * the LRU a second time.
1072		 */
1073		if (writeback && folio_test_reclaim(folio))
1074			stat->nr_congested += nr_pages;
1075
1076		/*
1077		 * If a folio at the tail of the LRU is under writeback, there
1078		 * are three cases to consider.
1079		 *
1080		 * 1) If reclaim is encountering an excessive number
1081		 *    of folios under writeback and this folio has both
1082		 *    the writeback and reclaim flags set, then it
1083		 *    indicates that folios are being queued for I/O but
1084		 *    are being recycled through the LRU before the I/O
1085		 *    can complete. Waiting on the folio itself risks an
1086		 *    indefinite stall if it is impossible to writeback
1087		 *    the folio due to I/O error or disconnected storage
1088		 *    so instead note that the LRU is being scanned too
1089		 *    quickly and the caller can stall after the folio
1090		 *    list has been processed.
1091		 *
1092		 * 2) Global or new memcg reclaim encounters a folio that is
1093		 *    not marked for immediate reclaim, or the caller does not
1094		 *    have __GFP_FS (or __GFP_IO if it's simply going to swap,
1095		 *    not to fs). In this case mark the folio for immediate
1096		 *    reclaim and continue scanning.
1097		 *
1098		 *    Require may_enter_fs() because we would wait on fs, which
1099		 *    may not have submitted I/O yet. And the loop driver might
1100		 *    enter reclaim, and deadlock if it waits on a folio for
1101		 *    which it is needed to do the write (loop masks off
1102		 *    __GFP_IO|__GFP_FS for this reason); but more thought
1103		 *    would probably show more reasons.
1104		 *
1105		 * 3) Legacy memcg encounters a folio that already has the
1106		 *    reclaim flag set. memcg does not have any dirty folio
1107		 *    throttling so we could easily OOM just because too many
1108		 *    folios are in writeback and there is nothing else to
1109		 *    reclaim. Wait for the writeback to complete.
1110		 *
1111		 * In cases 1) and 2) we activate the folios to get them out of
1112		 * the way while we continue scanning for clean folios on the
1113		 * inactive list and refilling from the active list. The
1114		 * observation here is that waiting for disk writes is more
1115		 * expensive than potentially causing reloads down the line.
1116		 * Since they're marked for immediate reclaim, they won't put
1117		 * memory pressure on the cache working set any longer than it
1118		 * takes to write them to disk.
1119		 */
1120		if (folio_test_writeback(folio)) {
1121			/* Case 1 above */
1122			if (current_is_kswapd() &&
1123			    folio_test_reclaim(folio) &&
1124			    test_bit(PGDAT_WRITEBACK, &pgdat->flags)) {
1125				stat->nr_immediate += nr_pages;
1126				goto activate_locked;
1127
1128			/* Case 2 above */
1129			} else if (writeback_throttling_sane(sc) ||
1130			    !folio_test_reclaim(folio) ||
1131			    !may_enter_fs(folio, sc->gfp_mask)) {
1132				/*
1133				 * This is slightly racy -
1134				 * folio_end_writeback() might have
1135				 * just cleared the reclaim flag, then
1136				 * setting the reclaim flag here ends up
1137				 * interpreted as the readahead flag - but
1138				 * that does not matter enough to care.
1139				 * What we do want is for this folio to
1140				 * have the reclaim flag set next time
1141				 * memcg reclaim reaches the tests above,
1142				 * so it will then wait for writeback to
1143				 * avoid OOM; and it's also appropriate
1144				 * in global reclaim.
1145				 */
1146				folio_set_reclaim(folio);
1147				stat->nr_writeback += nr_pages;
1148				goto activate_locked;
1149
1150			/* Case 3 above */
1151			} else {
1152				folio_unlock(folio);
1153				folio_wait_writeback(folio);
1154				/* then go back and try same folio again */
1155				list_add_tail(&folio->lru, folio_list);
1156				continue;
1157			}
1158		}
1159
1160		if (!ignore_references)
1161			references = folio_check_references(folio, sc);
1162
1163		switch (references) {
1164		case FOLIOREF_ACTIVATE:
1165			goto activate_locked;
1166		case FOLIOREF_KEEP:
1167			stat->nr_ref_keep += nr_pages;
1168			goto keep_locked;
1169		case FOLIOREF_RECLAIM:
1170		case FOLIOREF_RECLAIM_CLEAN:
1171			; /* try to reclaim the folio below */
1172		}
1173
1174		/*
1175		 * Before reclaiming the folio, try to relocate
1176		 * its contents to another node.
1177		 */
1178		if (do_demote_pass &&
1179		    (thp_migration_supported() || !folio_test_large(folio))) {
1180			list_add(&folio->lru, &demote_folios);
1181			folio_unlock(folio);
1182			continue;
1183		}
1184
1185		/*
1186		 * Anonymous process memory has backing store?
1187		 * Try to allocate it some swap space here.
1188		 * Lazyfree folio could be freed directly
1189		 */
1190		if (folio_test_anon(folio) && folio_test_swapbacked(folio)) {
1191			if (!folio_test_swapcache(folio)) {
1192				if (!(sc->gfp_mask & __GFP_IO))
1193					goto keep_locked;
1194				if (folio_maybe_dma_pinned(folio))
1195					goto keep_locked;
1196				if (folio_test_large(folio)) {
1197					/* cannot split folio, skip it */
1198					if (!can_split_folio(folio, NULL))
1199						goto activate_locked;
1200					/*
1201					 * Split folios without a PMD map right
1202					 * away. Chances are some or all of the
1203					 * tail pages can be freed without IO.
1204					 */
1205					if (!folio_entire_mapcount(folio) &&
1206					    split_folio_to_list(folio,
1207								folio_list))
1208						goto activate_locked;
1209				}
1210				if (!add_to_swap(folio)) {
1211					if (!folio_test_large(folio))
1212						goto activate_locked_split;
1213					/* Fallback to swap normal pages */
1214					if (split_folio_to_list(folio,
1215								folio_list))
1216						goto activate_locked;
1217#ifdef CONFIG_TRANSPARENT_HUGEPAGE
1218					count_memcg_folio_events(folio, THP_SWPOUT_FALLBACK, 1);
1219					count_vm_event(THP_SWPOUT_FALLBACK);
1220#endif
1221					if (!add_to_swap(folio))
1222						goto activate_locked_split;
1223				}
1224			}
1225		} else if (folio_test_swapbacked(folio) &&
1226			   folio_test_large(folio)) {
1227			/* Split shmem folio */
1228			if (split_folio_to_list(folio, folio_list))
1229				goto keep_locked;
1230		}
1231
1232		/*
1233		 * If the folio was split above, the tail pages will make
1234		 * their own pass through this function and be accounted
1235		 * then.
1236		 */
1237		if ((nr_pages > 1) && !folio_test_large(folio)) {
1238			sc->nr_scanned -= (nr_pages - 1);
1239			nr_pages = 1;
1240		}
1241
1242		/*
1243		 * The folio is mapped into the page tables of one or more
1244		 * processes. Try to unmap it here.
1245		 */
1246		if (folio_mapped(folio)) {
1247			enum ttu_flags flags = TTU_BATCH_FLUSH;
1248			bool was_swapbacked = folio_test_swapbacked(folio);
1249
1250			if (folio_test_pmd_mappable(folio))
1251				flags |= TTU_SPLIT_HUGE_PMD;
1252
1253			try_to_unmap(folio, flags);
1254			if (folio_mapped(folio)) {
1255				stat->nr_unmap_fail += nr_pages;
1256				if (!was_swapbacked &&
1257				    folio_test_swapbacked(folio))
1258					stat->nr_lazyfree_fail += nr_pages;
1259				goto activate_locked;
1260			}
1261		}
1262
1263		/*
1264		 * Folio is unmapped now so it cannot be newly pinned anymore.
1265		 * No point in trying to reclaim folio if it is pinned.
1266		 * Furthermore we don't want to reclaim underlying fs metadata
1267		 * if the folio is pinned and thus potentially modified by the
1268		 * pinning process as that may upset the filesystem.
1269		 */
1270		if (folio_maybe_dma_pinned(folio))
1271			goto activate_locked;
1272
1273		mapping = folio_mapping(folio);
1274		if (folio_test_dirty(folio)) {
1275			/*
1276			 * Only kswapd can writeback filesystem folios
1277			 * to avoid risk of stack overflow. But avoid
1278			 * injecting inefficient single-folio I/O into
1279			 * flusher writeback as much as possible: only
1280			 * write folios when we've encountered many
1281			 * dirty folios, and when we've already scanned
1282			 * the rest of the LRU for clean folios and see
1283			 * the same dirty folios again (with the reclaim
1284			 * flag set).
1285			 */
1286			if (folio_is_file_lru(folio) &&
1287			    (!current_is_kswapd() ||
1288			     !folio_test_reclaim(folio) ||
1289			     !test_bit(PGDAT_DIRTY, &pgdat->flags))) {
1290				/*
1291				 * Immediately reclaim when written back.
1292				 * Similar in principle to folio_deactivate()
1293				 * except we already have the folio isolated
1294				 * and know it's dirty
1295				 */
1296				node_stat_mod_folio(folio, NR_VMSCAN_IMMEDIATE,
1297						nr_pages);
1298				folio_set_reclaim(folio);
1299
1300				goto activate_locked;
1301			}
1302
1303			if (references == FOLIOREF_RECLAIM_CLEAN)
1304				goto keep_locked;
1305			if (!may_enter_fs(folio, sc->gfp_mask))
1306				goto keep_locked;
1307			if (!sc->may_writepage)
1308				goto keep_locked;
1309
1310			/*
1311			 * Folio is dirty. Flush the TLB if a writable entry
1312			 * potentially exists to avoid CPU writes after I/O
1313			 * starts and then write it out here.
1314			 */
1315			try_to_unmap_flush_dirty();
1316			switch (pageout(folio, mapping, &plug)) {
1317			case PAGE_KEEP:
1318				goto keep_locked;
1319			case PAGE_ACTIVATE:
1320				goto activate_locked;
1321			case PAGE_SUCCESS:
1322				stat->nr_pageout += nr_pages;
1323
1324				if (folio_test_writeback(folio))
1325					goto keep;
1326				if (folio_test_dirty(folio))
1327					goto keep;
1328
1329				/*
1330				 * A synchronous write - probably a ramdisk.  Go
1331				 * ahead and try to reclaim the folio.
1332				 */
1333				if (!folio_trylock(folio))
1334					goto keep;
1335				if (folio_test_dirty(folio) ||
1336				    folio_test_writeback(folio))
1337					goto keep_locked;
1338				mapping = folio_mapping(folio);
1339				fallthrough;
1340			case PAGE_CLEAN:
1341				; /* try to free the folio below */
1342			}
1343		}
1344
1345		/*
1346		 * If the folio has buffers, try to free the buffer
1347		 * mappings associated with this folio. If we succeed
1348		 * we try to free the folio as well.
1349		 *
1350		 * We do this even if the folio is dirty.
1351		 * filemap_release_folio() does not perform I/O, but it
1352		 * is possible for a folio to have the dirty flag set,
1353		 * but it is actually clean (all its buffers are clean).
1354		 * This happens if the buffers were written out directly,
1355		 * with submit_bh(). ext3 will do this, as well as
1356		 * the blockdev mapping.  filemap_release_folio() will
1357		 * discover that cleanness and will drop the buffers
1358		 * and mark the folio clean - it can be freed.
1359		 *
1360		 * Rarely, folios can have buffers and no ->mapping.
1361		 * These are the folios which were not successfully
1362		 * invalidated in truncate_cleanup_folio().  We try to
1363		 * drop those buffers here and if that worked, and the
1364		 * folio is no longer mapped into process address space
1365		 * (refcount == 1) it can be freed.  Otherwise, leave
1366		 * the folio on the LRU so it is swappable.
1367		 */
1368		if (folio_needs_release(folio)) {
1369			if (!filemap_release_folio(folio, sc->gfp_mask))
1370				goto activate_locked;
1371			if (!mapping && folio_ref_count(folio) == 1) {
1372				folio_unlock(folio);
1373				if (folio_put_testzero(folio))
1374					goto free_it;
1375				else {
1376					/*
1377					 * rare race with speculative reference.
1378					 * the speculative reference will free
1379					 * this folio shortly, so we may
1380					 * increment nr_reclaimed here (and
1381					 * leave it off the LRU).
1382					 */
1383					nr_reclaimed += nr_pages;
1384					continue;
1385				}
1386			}
1387		}
1388
1389		if (folio_test_anon(folio) && !folio_test_swapbacked(folio)) {
1390			/* follow __remove_mapping for reference */
1391			if (!folio_ref_freeze(folio, 1))
1392				goto keep_locked;
1393			/*
1394			 * The folio has only one reference left, which is
1395			 * from the isolation. After the caller puts the
1396			 * folio back on the lru and drops the reference, the
1397			 * folio will be freed anyway. It doesn't matter
1398			 * which lru it goes on. So we don't bother checking
1399			 * the dirty flag here.
1400			 */
1401			count_vm_events(PGLAZYFREED, nr_pages);
1402			count_memcg_folio_events(folio, PGLAZYFREED, nr_pages);
1403		} else if (!mapping || !__remove_mapping(mapping, folio, true,
1404							 sc->target_mem_cgroup))
1405			goto keep_locked;
1406
1407		folio_unlock(folio);
1408free_it:
1409		/*
1410		 * Folio may get swapped out as a whole, need to account
1411		 * all pages in it.
1412		 */
1413		nr_reclaimed += nr_pages;
1414
1415		/*
1416		 * Is there need to periodically free_folio_list? It would
1417		 * appear not as the counts should be low
1418		 */
1419		if (unlikely(folio_test_large(folio)))
1420			destroy_large_folio(folio);
1421		else
1422			list_add(&folio->lru, &free_folios);
1423		continue;
1424
1425activate_locked_split:
1426		/*
1427		 * The tail pages that are failed to add into swap cache
1428		 * reach here.  Fixup nr_scanned and nr_pages.
1429		 */
1430		if (nr_pages > 1) {
1431			sc->nr_scanned -= (nr_pages - 1);
1432			nr_pages = 1;
1433		}
1434activate_locked:
1435		/* Not a candidate for swapping, so reclaim swap space. */
1436		if (folio_test_swapcache(folio) &&
1437		    (mem_cgroup_swap_full(folio) || folio_test_mlocked(folio)))
1438			folio_free_swap(folio);
1439		VM_BUG_ON_FOLIO(folio_test_active(folio), folio);
1440		if (!folio_test_mlocked(folio)) {
1441			int type = folio_is_file_lru(folio);
1442			folio_set_active(folio);
1443			stat->nr_activate[type] += nr_pages;
1444			count_memcg_folio_events(folio, PGACTIVATE, nr_pages);
1445		}
1446keep_locked:
1447		folio_unlock(folio);
1448keep:
1449		list_add(&folio->lru, &ret_folios);
1450		VM_BUG_ON_FOLIO(folio_test_lru(folio) ||
1451				folio_test_unevictable(folio), folio);
1452	}
1453	/* 'folio_list' is always empty here */
1454
1455	/* Migrate folios selected for demotion */
1456	nr_reclaimed += demote_folio_list(&demote_folios, pgdat);
1457	/* Folios that could not be demoted are still in @demote_folios */
1458	if (!list_empty(&demote_folios)) {
1459		/* Folios which weren't demoted go back on @folio_list */
1460		list_splice_init(&demote_folios, folio_list);
1461
1462		/*
1463		 * goto retry to reclaim the undemoted folios in folio_list if
1464		 * desired.
1465		 *
1466		 * Reclaiming directly from top tier nodes is not often desired
1467		 * due to it breaking the LRU ordering: in general memory
1468		 * should be reclaimed from lower tier nodes and demoted from
1469		 * top tier nodes.
1470		 *
1471		 * However, disabling reclaim from top tier nodes entirely
1472		 * would cause ooms in edge scenarios where lower tier memory
1473		 * is unreclaimable for whatever reason, eg memory being
1474		 * mlocked or too hot to reclaim. We can disable reclaim
1475		 * from top tier nodes in proactive reclaim though as that is
1476		 * not real memory pressure.
1477		 */
1478		if (!sc->proactive) {
1479			do_demote_pass = false;
1480			goto retry;
1481		}
1482	}
1483
1484	pgactivate = stat->nr_activate[0] + stat->nr_activate[1];
1485
1486	mem_cgroup_uncharge_list(&free_folios);
1487	try_to_unmap_flush();
1488	free_unref_page_list(&free_folios);
1489
1490	list_splice(&ret_folios, folio_list);
1491	count_vm_events(PGACTIVATE, pgactivate);
1492
1493	if (plug)
1494		swap_write_unplug(plug);
1495	return nr_reclaimed;
1496}
1497
1498unsigned int reclaim_clean_pages_from_list(struct zone *zone,
1499					   struct list_head *folio_list)
1500{
1501	struct scan_control sc = {
1502		.gfp_mask = GFP_KERNEL,
1503		.may_unmap = 1,
1504	};
1505	struct reclaim_stat stat;
1506	unsigned int nr_reclaimed;
1507	struct folio *folio, *next;
1508	LIST_HEAD(clean_folios);
1509	unsigned int noreclaim_flag;
1510
1511	list_for_each_entry_safe(folio, next, folio_list, lru) {
1512		if (!folio_test_hugetlb(folio) && folio_is_file_lru(folio) &&
1513		    !folio_test_dirty(folio) && !__folio_test_movable(folio) &&
1514		    !folio_test_unevictable(folio)) {
1515			folio_clear_active(folio);
1516			list_move(&folio->lru, &clean_folios);
1517		}
1518	}
1519
1520	/*
1521	 * We should be safe here since we are only dealing with file pages and
1522	 * we are not kswapd and therefore cannot write dirty file pages. But
1523	 * call memalloc_noreclaim_save() anyway, just in case these conditions
1524	 * change in the future.
1525	 */
1526	noreclaim_flag = memalloc_noreclaim_save();
1527	nr_reclaimed = shrink_folio_list(&clean_folios, zone->zone_pgdat, &sc,
1528					&stat, true);
1529	memalloc_noreclaim_restore(noreclaim_flag);
1530
1531	list_splice(&clean_folios, folio_list);
1532	mod_node_page_state(zone->zone_pgdat, NR_ISOLATED_FILE,
1533			    -(long)nr_reclaimed);
1534	/*
1535	 * Since lazyfree pages are isolated from file LRU from the beginning,
1536	 * they will rotate back to anonymous LRU in the end if it failed to
1537	 * discard so isolated count will be mismatched.
1538	 * Compensate the isolated count for both LRU lists.
1539	 */
1540	mod_node_page_state(zone->zone_pgdat, NR_ISOLATED_ANON,
1541			    stat.nr_lazyfree_fail);
1542	mod_node_page_state(zone->zone_pgdat, NR_ISOLATED_FILE,
1543			    -(long)stat.nr_lazyfree_fail);
1544	return nr_reclaimed;
1545}
1546
1547/*
1548 * Update LRU sizes after isolating pages. The LRU size updates must
1549 * be complete before mem_cgroup_update_lru_size due to a sanity check.
1550 */
1551static __always_inline void update_lru_sizes(struct lruvec *lruvec,
1552			enum lru_list lru, unsigned long *nr_zone_taken)
1553{
1554	int zid;
1555
1556	for (zid = 0; zid < MAX_NR_ZONES; zid++) {
1557		if (!nr_zone_taken[zid])
1558			continue;
1559
1560		update_lru_size(lruvec, lru, zid, -nr_zone_taken[zid]);
1561	}
1562
1563}
1564
1565#ifdef CONFIG_CMA
1566/*
1567 * It is waste of effort to scan and reclaim CMA pages if it is not available
1568 * for current allocation context. Kswapd can not be enrolled as it can not
1569 * distinguish this scenario by using sc->gfp_mask = GFP_KERNEL
1570 */
1571static bool skip_cma(struct folio *folio, struct scan_control *sc)
1572{
1573	return !current_is_kswapd() &&
1574			gfp_migratetype(sc->gfp_mask) != MIGRATE_MOVABLE &&
1575			folio_migratetype(folio) == MIGRATE_CMA;
1576}
1577#else
1578static bool skip_cma(struct folio *folio, struct scan_control *sc)
1579{
1580	return false;
1581}
1582#endif
1583
1584/*
1585 * Isolating page from the lruvec to fill in @dst list by nr_to_scan times.
1586 *
1587 * lruvec->lru_lock is heavily contended.  Some of the functions that
1588 * shrink the lists perform better by taking out a batch of pages
1589 * and working on them outside the LRU lock.
1590 *
1591 * For pagecache intensive workloads, this function is the hottest
1592 * spot in the kernel (apart from copy_*_user functions).
1593 *
1594 * Lru_lock must be held before calling this function.
1595 *
1596 * @nr_to_scan:	The number of eligible pages to look through on the list.
1597 * @lruvec:	The LRU vector to pull pages from.
1598 * @dst:	The temp list to put pages on to.
1599 * @nr_scanned:	The number of pages that were scanned.
1600 * @sc:		The scan_control struct for this reclaim session
1601 * @lru:	LRU list id for isolating
1602 *
1603 * returns how many pages were moved onto *@dst.
1604 */
1605static unsigned long isolate_lru_folios(unsigned long nr_to_scan,
1606		struct lruvec *lruvec, struct list_head *dst,
1607		unsigned long *nr_scanned, struct scan_control *sc,
1608		enum lru_list lru)
1609{
1610	struct list_head *src = &lruvec->lists[lru];
1611	unsigned long nr_taken = 0;
1612	unsigned long nr_zone_taken[MAX_NR_ZONES] = { 0 };
1613	unsigned long nr_skipped[MAX_NR_ZONES] = { 0, };
1614	unsigned long skipped = 0;
1615	unsigned long scan, total_scan, nr_pages;
1616	LIST_HEAD(folios_skipped);
1617
1618	total_scan = 0;
1619	scan = 0;
1620	while (scan < nr_to_scan && !list_empty(src)) {
1621		struct list_head *move_to = src;
1622		struct folio *folio;
1623
1624		folio = lru_to_folio(src);
1625		prefetchw_prev_lru_folio(folio, src, flags);
1626
1627		nr_pages = folio_nr_pages(folio);
1628		total_scan += nr_pages;
1629
1630		if (folio_zonenum(folio) > sc->reclaim_idx ||
1631				skip_cma(folio, sc)) {
1632			nr_skipped[folio_zonenum(folio)] += nr_pages;
1633			move_to = &folios_skipped;
1634			goto move;
1635		}
1636
1637		/*
1638		 * Do not count skipped folios because that makes the function
1639		 * return with no isolated folios if the LRU mostly contains
1640		 * ineligible folios.  This causes the VM to not reclaim any
1641		 * folios, triggering a premature OOM.
1642		 * Account all pages in a folio.
1643		 */
1644		scan += nr_pages;
1645
1646		if (!folio_test_lru(folio))
1647			goto move;
1648		if (!sc->may_unmap && folio_mapped(folio))
1649			goto move;
1650
1651		/*
1652		 * Be careful not to clear the lru flag until after we're
1653		 * sure the folio is not being freed elsewhere -- the
1654		 * folio release code relies on it.
1655		 */
1656		if (unlikely(!folio_try_get(folio)))
1657			goto move;
1658
1659		if (!folio_test_clear_lru(folio)) {
1660			/* Another thread is already isolating this folio */
1661			folio_put(folio);
1662			goto move;
1663		}
1664
1665		nr_taken += nr_pages;
1666		nr_zone_taken[folio_zonenum(folio)] += nr_pages;
1667		move_to = dst;
1668move:
1669		list_move(&folio->lru, move_to);
1670	}
1671
1672	/*
1673	 * Splice any skipped folios to the start of the LRU list. Note that
1674	 * this disrupts the LRU order when reclaiming for lower zones but
1675	 * we cannot splice to the tail. If we did then the SWAP_CLUSTER_MAX
1676	 * scanning would soon rescan the same folios to skip and waste lots
1677	 * of cpu cycles.
1678	 */
1679	if (!list_empty(&folios_skipped)) {
1680		int zid;
1681
1682		list_splice(&folios_skipped, src);
1683		for (zid = 0; zid < MAX_NR_ZONES; zid++) {
1684			if (!nr_skipped[zid])
1685				continue;
1686
1687			__count_zid_vm_events(PGSCAN_SKIP, zid, nr_skipped[zid]);
1688			skipped += nr_skipped[zid];
1689		}
1690	}
1691	*nr_scanned = total_scan;
1692	trace_mm_vmscan_lru_isolate(sc->reclaim_idx, sc->order, nr_to_scan,
1693				    total_scan, skipped, nr_taken, lru);
1694	update_lru_sizes(lruvec, lru, nr_zone_taken);
1695	return nr_taken;
1696}
1697
1698/**
1699 * folio_isolate_lru() - Try to isolate a folio from its LRU list.
1700 * @folio: Folio to isolate from its LRU list.
1701 *
1702 * Isolate a @folio from an LRU list and adjust the vmstat statistic
1703 * corresponding to whatever LRU list the folio was on.
1704 *
1705 * The folio will have its LRU flag cleared.  If it was found on the
1706 * active list, it will have the Active flag set.  If it was found on the
1707 * unevictable list, it will have the Unevictable flag set.  These flags
1708 * may need to be cleared by the caller before letting the page go.
1709 *
1710 * Context:
1711 *
1712 * (1) Must be called with an elevated refcount on the folio. This is a
1713 *     fundamental difference from isolate_lru_folios() (which is called
1714 *     without a stable reference).
1715 * (2) The lru_lock must not be held.
1716 * (3) Interrupts must be enabled.
1717 *
1718 * Return: true if the folio was removed from an LRU list.
1719 * false if the folio was not on an LRU list.
1720 */
1721bool folio_isolate_lru(struct folio *folio)
1722{
1723	bool ret = false;
1724
1725	VM_BUG_ON_FOLIO(!folio_ref_count(folio), folio);
1726
1727	if (folio_test_clear_lru(folio)) {
1728		struct lruvec *lruvec;
1729
1730		folio_get(folio);
1731		lruvec = folio_lruvec_lock_irq(folio);
1732		lruvec_del_folio(lruvec, folio);
1733		unlock_page_lruvec_irq(lruvec);
1734		ret = true;
1735	}
1736
1737	return ret;
1738}
1739
1740/*
1741 * A direct reclaimer may isolate SWAP_CLUSTER_MAX pages from the LRU list and
1742 * then get rescheduled. When there are massive number of tasks doing page
1743 * allocation, such sleeping direct reclaimers may keep piling up on each CPU,
1744 * the LRU list will go small and be scanned faster than necessary, leading to
1745 * unnecessary swapping, thrashing and OOM.
1746 */
1747static int too_many_isolated(struct pglist_data *pgdat, int file,
1748		struct scan_control *sc)
1749{
1750	unsigned long inactive, isolated;
1751	bool too_many;
1752
1753	if (current_is_kswapd())
1754		return 0;
1755
1756	if (!writeback_throttling_sane(sc))
1757		return 0;
1758
1759	if (file) {
1760		inactive = node_page_state(pgdat, NR_INACTIVE_FILE);
1761		isolated = node_page_state(pgdat, NR_ISOLATED_FILE);
1762	} else {
1763		inactive = node_page_state(pgdat, NR_INACTIVE_ANON);
1764		isolated = node_page_state(pgdat, NR_ISOLATED_ANON);
1765	}
1766
1767	/*
1768	 * GFP_NOIO/GFP_NOFS callers are allowed to isolate more pages, so they
1769	 * won't get blocked by normal direct-reclaimers, forming a circular
1770	 * deadlock.
1771	 */
1772	if (gfp_has_io_fs(sc->gfp_mask))
1773		inactive >>= 3;
1774
1775	too_many = isolated > inactive;
1776
1777	/* Wake up tasks throttled due to too_many_isolated. */
1778	if (!too_many)
1779		wake_throttle_isolated(pgdat);
1780
1781	return too_many;
1782}
1783
1784/*
1785 * move_folios_to_lru() moves folios from private @list to appropriate LRU list.
1786 * On return, @list is reused as a list of folios to be freed by the caller.
1787 *
1788 * Returns the number of pages moved to the given lruvec.
1789 */
1790static unsigned int move_folios_to_lru(struct lruvec *lruvec,
1791		struct list_head *list)
1792{
1793	int nr_pages, nr_moved = 0;
1794	LIST_HEAD(folios_to_free);
1795
 
1796	while (!list_empty(list)) {
1797		struct folio *folio = lru_to_folio(list);
1798
1799		VM_BUG_ON_FOLIO(folio_test_lru(folio), folio);
1800		list_del(&folio->lru);
1801		if (unlikely(!folio_evictable(folio))) {
1802			spin_unlock_irq(&lruvec->lru_lock);
1803			folio_putback_lru(folio);
1804			spin_lock_irq(&lruvec->lru_lock);
1805			continue;
1806		}
1807
1808		/*
1809		 * The folio_set_lru needs to be kept here for list integrity.
1810		 * Otherwise:
1811		 *   #0 move_folios_to_lru             #1 release_pages
1812		 *   if (!folio_put_testzero())
1813		 *				      if (folio_put_testzero())
1814		 *				        !lru //skip lru_lock
1815		 *     folio_set_lru()
1816		 *     list_add(&folio->lru,)
1817		 *                                        list_add(&folio->lru,)
1818		 */
1819		folio_set_lru(folio);
1820
1821		if (unlikely(folio_put_testzero(folio))) {
1822			__folio_clear_lru_flags(folio);
1823
1824			if (unlikely(folio_test_large(folio))) {
 
 
 
1825				spin_unlock_irq(&lruvec->lru_lock);
1826				destroy_large_folio(folio);
 
1827				spin_lock_irq(&lruvec->lru_lock);
1828			} else
1829				list_add(&folio->lru, &folios_to_free);
1830
1831			continue;
1832		}
1833
1834		/*
1835		 * All pages were isolated from the same lruvec (and isolation
1836		 * inhibits memcg migration).
1837		 */
1838		VM_BUG_ON_FOLIO(!folio_matches_lruvec(folio, lruvec), folio);
1839		lruvec_add_folio(lruvec, folio);
1840		nr_pages = folio_nr_pages(folio);
1841		nr_moved += nr_pages;
1842		if (folio_test_active(folio))
1843			workingset_age_nonresident(lruvec, nr_pages);
1844	}
1845
1846	/*
1847	 * To save our caller's stack, now use input list for pages to free.
1848	 */
1849	list_splice(&folios_to_free, list);
 
 
1850
1851	return nr_moved;
1852}
1853
1854/*
1855 * If a kernel thread (such as nfsd for loop-back mounts) services a backing
1856 * device by writing to the page cache it sets PF_LOCAL_THROTTLE. In this case
1857 * we should not throttle.  Otherwise it is safe to do so.
1858 */
1859static int current_may_throttle(void)
1860{
1861	return !(current->flags & PF_LOCAL_THROTTLE);
1862}
1863
1864/*
1865 * shrink_inactive_list() is a helper for shrink_node().  It returns the number
1866 * of reclaimed pages
1867 */
1868static unsigned long shrink_inactive_list(unsigned long nr_to_scan,
1869		struct lruvec *lruvec, struct scan_control *sc,
1870		enum lru_list lru)
1871{
1872	LIST_HEAD(folio_list);
1873	unsigned long nr_scanned;
1874	unsigned int nr_reclaimed = 0;
1875	unsigned long nr_taken;
1876	struct reclaim_stat stat;
1877	bool file = is_file_lru(lru);
1878	enum vm_event_item item;
1879	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
1880	bool stalled = false;
1881
1882	while (unlikely(too_many_isolated(pgdat, file, sc))) {
1883		if (stalled)
1884			return 0;
1885
1886		/* wait a bit for the reclaimer. */
1887		stalled = true;
1888		reclaim_throttle(pgdat, VMSCAN_THROTTLE_ISOLATED);
1889
1890		/* We are about to die and free our memory. Return now. */
1891		if (fatal_signal_pending(current))
1892			return SWAP_CLUSTER_MAX;
1893	}
1894
1895	lru_add_drain();
1896
1897	spin_lock_irq(&lruvec->lru_lock);
1898
1899	nr_taken = isolate_lru_folios(nr_to_scan, lruvec, &folio_list,
1900				     &nr_scanned, sc, lru);
1901
1902	__mod_node_page_state(pgdat, NR_ISOLATED_ANON + file, nr_taken);
1903	item = PGSCAN_KSWAPD + reclaimer_offset();
1904	if (!cgroup_reclaim(sc))
1905		__count_vm_events(item, nr_scanned);
1906	__count_memcg_events(lruvec_memcg(lruvec), item, nr_scanned);
1907	__count_vm_events(PGSCAN_ANON + file, nr_scanned);
1908
1909	spin_unlock_irq(&lruvec->lru_lock);
1910
1911	if (nr_taken == 0)
1912		return 0;
1913
1914	nr_reclaimed = shrink_folio_list(&folio_list, pgdat, sc, &stat, false);
1915
1916	spin_lock_irq(&lruvec->lru_lock);
1917	move_folios_to_lru(lruvec, &folio_list);
1918
1919	__mod_node_page_state(pgdat, NR_ISOLATED_ANON + file, -nr_taken);
1920	item = PGSTEAL_KSWAPD + reclaimer_offset();
1921	if (!cgroup_reclaim(sc))
1922		__count_vm_events(item, nr_reclaimed);
1923	__count_memcg_events(lruvec_memcg(lruvec), item, nr_reclaimed);
1924	__count_vm_events(PGSTEAL_ANON + file, nr_reclaimed);
1925	spin_unlock_irq(&lruvec->lru_lock);
1926
1927	lru_note_cost(lruvec, file, stat.nr_pageout, nr_scanned - nr_reclaimed);
1928	mem_cgroup_uncharge_list(&folio_list);
1929	free_unref_page_list(&folio_list);
1930
1931	/*
1932	 * If dirty folios are scanned that are not queued for IO, it
1933	 * implies that flushers are not doing their job. This can
1934	 * happen when memory pressure pushes dirty folios to the end of
1935	 * the LRU before the dirty limits are breached and the dirty
1936	 * data has expired. It can also happen when the proportion of
1937	 * dirty folios grows not through writes but through memory
1938	 * pressure reclaiming all the clean cache. And in some cases,
1939	 * the flushers simply cannot keep up with the allocation
1940	 * rate. Nudge the flusher threads in case they are asleep.
1941	 */
1942	if (stat.nr_unqueued_dirty == nr_taken) {
1943		wakeup_flusher_threads(WB_REASON_VMSCAN);
1944		/*
1945		 * For cgroupv1 dirty throttling is achieved by waking up
1946		 * the kernel flusher here and later waiting on folios
1947		 * which are in writeback to finish (see shrink_folio_list()).
1948		 *
1949		 * Flusher may not be able to issue writeback quickly
1950		 * enough for cgroupv1 writeback throttling to work
1951		 * on a large system.
1952		 */
1953		if (!writeback_throttling_sane(sc))
1954			reclaim_throttle(pgdat, VMSCAN_THROTTLE_WRITEBACK);
1955	}
1956
1957	sc->nr.dirty += stat.nr_dirty;
1958	sc->nr.congested += stat.nr_congested;
1959	sc->nr.unqueued_dirty += stat.nr_unqueued_dirty;
1960	sc->nr.writeback += stat.nr_writeback;
1961	sc->nr.immediate += stat.nr_immediate;
1962	sc->nr.taken += nr_taken;
1963	if (file)
1964		sc->nr.file_taken += nr_taken;
1965
1966	trace_mm_vmscan_lru_shrink_inactive(pgdat->node_id,
1967			nr_scanned, nr_reclaimed, &stat, sc->priority, file);
1968	return nr_reclaimed;
1969}
1970
1971/*
1972 * shrink_active_list() moves folios from the active LRU to the inactive LRU.
1973 *
1974 * We move them the other way if the folio is referenced by one or more
1975 * processes.
1976 *
1977 * If the folios are mostly unmapped, the processing is fast and it is
1978 * appropriate to hold lru_lock across the whole operation.  But if
1979 * the folios are mapped, the processing is slow (folio_referenced()), so
1980 * we should drop lru_lock around each folio.  It's impossible to balance
1981 * this, so instead we remove the folios from the LRU while processing them.
1982 * It is safe to rely on the active flag against the non-LRU folios in here
1983 * because nobody will play with that bit on a non-LRU folio.
1984 *
1985 * The downside is that we have to touch folio->_refcount against each folio.
1986 * But we had to alter folio->flags anyway.
1987 */
1988static void shrink_active_list(unsigned long nr_to_scan,
1989			       struct lruvec *lruvec,
1990			       struct scan_control *sc,
1991			       enum lru_list lru)
1992{
1993	unsigned long nr_taken;
1994	unsigned long nr_scanned;
1995	unsigned long vm_flags;
1996	LIST_HEAD(l_hold);	/* The folios which were snipped off */
1997	LIST_HEAD(l_active);
1998	LIST_HEAD(l_inactive);
1999	unsigned nr_deactivate, nr_activate;
2000	unsigned nr_rotated = 0;
2001	int file = is_file_lru(lru);
2002	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
2003
2004	lru_add_drain();
2005
2006	spin_lock_irq(&lruvec->lru_lock);
2007
2008	nr_taken = isolate_lru_folios(nr_to_scan, lruvec, &l_hold,
2009				     &nr_scanned, sc, lru);
2010
2011	__mod_node_page_state(pgdat, NR_ISOLATED_ANON + file, nr_taken);
2012
2013	if (!cgroup_reclaim(sc))
2014		__count_vm_events(PGREFILL, nr_scanned);
2015	__count_memcg_events(lruvec_memcg(lruvec), PGREFILL, nr_scanned);
2016
2017	spin_unlock_irq(&lruvec->lru_lock);
2018
2019	while (!list_empty(&l_hold)) {
2020		struct folio *folio;
2021
2022		cond_resched();
2023		folio = lru_to_folio(&l_hold);
2024		list_del(&folio->lru);
2025
2026		if (unlikely(!folio_evictable(folio))) {
2027			folio_putback_lru(folio);
2028			continue;
2029		}
2030
2031		if (unlikely(buffer_heads_over_limit)) {
2032			if (folio_needs_release(folio) &&
2033			    folio_trylock(folio)) {
2034				filemap_release_folio(folio, 0);
2035				folio_unlock(folio);
2036			}
2037		}
2038
2039		/* Referenced or rmap lock contention: rotate */
2040		if (folio_referenced(folio, 0, sc->target_mem_cgroup,
2041				     &vm_flags) != 0) {
2042			/*
2043			 * Identify referenced, file-backed active folios and
2044			 * give them one more trip around the active list. So
2045			 * that executable code get better chances to stay in
2046			 * memory under moderate memory pressure.  Anon folios
2047			 * are not likely to be evicted by use-once streaming
2048			 * IO, plus JVM can create lots of anon VM_EXEC folios,
2049			 * so we ignore them here.
2050			 */
2051			if ((vm_flags & VM_EXEC) && folio_is_file_lru(folio)) {
2052				nr_rotated += folio_nr_pages(folio);
2053				list_add(&folio->lru, &l_active);
2054				continue;
2055			}
2056		}
2057
2058		folio_clear_active(folio);	/* we are de-activating */
2059		folio_set_workingset(folio);
2060		list_add(&folio->lru, &l_inactive);
2061	}
2062
2063	/*
2064	 * Move folios back to the lru list.
2065	 */
2066	spin_lock_irq(&lruvec->lru_lock);
2067
2068	nr_activate = move_folios_to_lru(lruvec, &l_active);
2069	nr_deactivate = move_folios_to_lru(lruvec, &l_inactive);
2070	/* Keep all free folios in l_active list */
2071	list_splice(&l_inactive, &l_active);
2072
2073	__count_vm_events(PGDEACTIVATE, nr_deactivate);
2074	__count_memcg_events(lruvec_memcg(lruvec), PGDEACTIVATE, nr_deactivate);
2075
2076	__mod_node_page_state(pgdat, NR_ISOLATED_ANON + file, -nr_taken);
2077	spin_unlock_irq(&lruvec->lru_lock);
2078
2079	if (nr_rotated)
2080		lru_note_cost(lruvec, file, 0, nr_rotated);
2081	mem_cgroup_uncharge_list(&l_active);
2082	free_unref_page_list(&l_active);
2083	trace_mm_vmscan_lru_shrink_active(pgdat->node_id, nr_taken, nr_activate,
2084			nr_deactivate, nr_rotated, sc->priority, file);
2085}
2086
2087static unsigned int reclaim_folio_list(struct list_head *folio_list,
2088				      struct pglist_data *pgdat)
 
2089{
2090	struct reclaim_stat dummy_stat;
2091	unsigned int nr_reclaimed;
2092	struct folio *folio;
2093	struct scan_control sc = {
2094		.gfp_mask = GFP_KERNEL,
2095		.may_writepage = 1,
2096		.may_unmap = 1,
2097		.may_swap = 1,
2098		.no_demotion = 1,
2099	};
2100
2101	nr_reclaimed = shrink_folio_list(folio_list, pgdat, &sc, &dummy_stat, false);
2102	while (!list_empty(folio_list)) {
2103		folio = lru_to_folio(folio_list);
2104		list_del(&folio->lru);
2105		folio_putback_lru(folio);
2106	}
2107
2108	return nr_reclaimed;
2109}
2110
2111unsigned long reclaim_pages(struct list_head *folio_list)
2112{
2113	int nid;
2114	unsigned int nr_reclaimed = 0;
2115	LIST_HEAD(node_folio_list);
2116	unsigned int noreclaim_flag;
2117
2118	if (list_empty(folio_list))
2119		return nr_reclaimed;
2120
2121	noreclaim_flag = memalloc_noreclaim_save();
2122
2123	nid = folio_nid(lru_to_folio(folio_list));
2124	do {
2125		struct folio *folio = lru_to_folio(folio_list);
2126
2127		if (nid == folio_nid(folio)) {
2128			folio_clear_active(folio);
2129			list_move(&folio->lru, &node_folio_list);
2130			continue;
2131		}
2132
2133		nr_reclaimed += reclaim_folio_list(&node_folio_list, NODE_DATA(nid));
 
2134		nid = folio_nid(lru_to_folio(folio_list));
2135	} while (!list_empty(folio_list));
2136
2137	nr_reclaimed += reclaim_folio_list(&node_folio_list, NODE_DATA(nid));
2138
2139	memalloc_noreclaim_restore(noreclaim_flag);
2140
2141	return nr_reclaimed;
2142}
2143
2144static unsigned long shrink_list(enum lru_list lru, unsigned long nr_to_scan,
2145				 struct lruvec *lruvec, struct scan_control *sc)
2146{
2147	if (is_active_lru(lru)) {
2148		if (sc->may_deactivate & (1 << is_file_lru(lru)))
2149			shrink_active_list(nr_to_scan, lruvec, sc, lru);
2150		else
2151			sc->skipped_deactivate = 1;
2152		return 0;
2153	}
2154
2155	return shrink_inactive_list(nr_to_scan, lruvec, sc, lru);
2156}
2157
2158/*
2159 * The inactive anon list should be small enough that the VM never has
2160 * to do too much work.
2161 *
2162 * The inactive file list should be small enough to leave most memory
2163 * to the established workingset on the scan-resistant active list,
2164 * but large enough to avoid thrashing the aggregate readahead window.
2165 *
2166 * Both inactive lists should also be large enough that each inactive
2167 * folio has a chance to be referenced again before it is reclaimed.
2168 *
2169 * If that fails and refaulting is observed, the inactive list grows.
2170 *
2171 * The inactive_ratio is the target ratio of ACTIVE to INACTIVE folios
2172 * on this LRU, maintained by the pageout code. An inactive_ratio
2173 * of 3 means 3:1 or 25% of the folios are kept on the inactive list.
2174 *
2175 * total     target    max
2176 * memory    ratio     inactive
2177 * -------------------------------------
2178 *   10MB       1         5MB
2179 *  100MB       1        50MB
2180 *    1GB       3       250MB
2181 *   10GB      10       0.9GB
2182 *  100GB      31         3GB
2183 *    1TB     101        10GB
2184 *   10TB     320        32GB
2185 */
2186static bool inactive_is_low(struct lruvec *lruvec, enum lru_list inactive_lru)
2187{
2188	enum lru_list active_lru = inactive_lru + LRU_ACTIVE;
2189	unsigned long inactive, active;
2190	unsigned long inactive_ratio;
2191	unsigned long gb;
2192
2193	inactive = lruvec_page_state(lruvec, NR_LRU_BASE + inactive_lru);
2194	active = lruvec_page_state(lruvec, NR_LRU_BASE + active_lru);
2195
2196	gb = (inactive + active) >> (30 - PAGE_SHIFT);
2197	if (gb)
2198		inactive_ratio = int_sqrt(10 * gb);
2199	else
2200		inactive_ratio = 1;
2201
2202	return inactive * inactive_ratio < active;
2203}
2204
2205enum scan_balance {
2206	SCAN_EQUAL,
2207	SCAN_FRACT,
2208	SCAN_ANON,
2209	SCAN_FILE,
2210};
2211
2212static void prepare_scan_control(pg_data_t *pgdat, struct scan_control *sc)
2213{
2214	unsigned long file;
2215	struct lruvec *target_lruvec;
2216
2217	if (lru_gen_enabled())
2218		return;
2219
2220	target_lruvec = mem_cgroup_lruvec(sc->target_mem_cgroup, pgdat);
2221
2222	/*
2223	 * Flush the memory cgroup stats, so that we read accurate per-memcg
2224	 * lruvec stats for heuristics.
2225	 */
2226	mem_cgroup_flush_stats(sc->target_mem_cgroup);
2227
2228	/*
2229	 * Determine the scan balance between anon and file LRUs.
2230	 */
2231	spin_lock_irq(&target_lruvec->lru_lock);
2232	sc->anon_cost = target_lruvec->anon_cost;
2233	sc->file_cost = target_lruvec->file_cost;
2234	spin_unlock_irq(&target_lruvec->lru_lock);
2235
2236	/*
2237	 * Target desirable inactive:active list ratios for the anon
2238	 * and file LRU lists.
2239	 */
2240	if (!sc->force_deactivate) {
2241		unsigned long refaults;
2242
2243		/*
2244		 * When refaults are being observed, it means a new
2245		 * workingset is being established. Deactivate to get
2246		 * rid of any stale active pages quickly.
2247		 */
2248		refaults = lruvec_page_state(target_lruvec,
2249				WORKINGSET_ACTIVATE_ANON);
2250		if (refaults != target_lruvec->refaults[WORKINGSET_ANON] ||
2251			inactive_is_low(target_lruvec, LRU_INACTIVE_ANON))
2252			sc->may_deactivate |= DEACTIVATE_ANON;
2253		else
2254			sc->may_deactivate &= ~DEACTIVATE_ANON;
2255
2256		refaults = lruvec_page_state(target_lruvec,
2257				WORKINGSET_ACTIVATE_FILE);
2258		if (refaults != target_lruvec->refaults[WORKINGSET_FILE] ||
2259		    inactive_is_low(target_lruvec, LRU_INACTIVE_FILE))
2260			sc->may_deactivate |= DEACTIVATE_FILE;
2261		else
2262			sc->may_deactivate &= ~DEACTIVATE_FILE;
2263	} else
2264		sc->may_deactivate = DEACTIVATE_ANON | DEACTIVATE_FILE;
2265
2266	/*
2267	 * If we have plenty of inactive file pages that aren't
2268	 * thrashing, try to reclaim those first before touching
2269	 * anonymous pages.
2270	 */
2271	file = lruvec_page_state(target_lruvec, NR_INACTIVE_FILE);
2272	if (file >> sc->priority && !(sc->may_deactivate & DEACTIVATE_FILE))
 
2273		sc->cache_trim_mode = 1;
2274	else
2275		sc->cache_trim_mode = 0;
2276
2277	/*
2278	 * Prevent the reclaimer from falling into the cache trap: as
2279	 * cache pages start out inactive, every cache fault will tip
2280	 * the scan balance towards the file LRU.  And as the file LRU
2281	 * shrinks, so does the window for rotation from references.
2282	 * This means we have a runaway feedback loop where a tiny
2283	 * thrashing file LRU becomes infinitely more attractive than
2284	 * anon pages.  Try to detect this based on file LRU size.
2285	 */
2286	if (!cgroup_reclaim(sc)) {
2287		unsigned long total_high_wmark = 0;
2288		unsigned long free, anon;
2289		int z;
2290
2291		free = sum_zone_node_page_state(pgdat->node_id, NR_FREE_PAGES);
2292		file = node_page_state(pgdat, NR_ACTIVE_FILE) +
2293			   node_page_state(pgdat, NR_INACTIVE_FILE);
2294
2295		for (z = 0; z < MAX_NR_ZONES; z++) {
2296			struct zone *zone = &pgdat->node_zones[z];
2297
2298			if (!managed_zone(zone))
2299				continue;
2300
2301			total_high_wmark += high_wmark_pages(zone);
2302		}
2303
2304		/*
2305		 * Consider anon: if that's low too, this isn't a
2306		 * runaway file reclaim problem, but rather just
2307		 * extreme pressure. Reclaim as per usual then.
2308		 */
2309		anon = node_page_state(pgdat, NR_INACTIVE_ANON);
2310
2311		sc->file_is_tiny =
2312			file + free <= total_high_wmark &&
2313			!(sc->may_deactivate & DEACTIVATE_ANON) &&
2314			anon >> sc->priority;
2315	}
2316}
2317
2318/*
2319 * Determine how aggressively the anon and file LRU lists should be
2320 * scanned.
2321 *
2322 * nr[0] = anon inactive folios to scan; nr[1] = anon active folios to scan
2323 * nr[2] = file inactive folios to scan; nr[3] = file active folios to scan
2324 */
2325static void get_scan_count(struct lruvec *lruvec, struct scan_control *sc,
2326			   unsigned long *nr)
2327{
2328	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
2329	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
2330	unsigned long anon_cost, file_cost, total_cost;
2331	int swappiness = mem_cgroup_swappiness(memcg);
2332	u64 fraction[ANON_AND_FILE];
2333	u64 denominator = 0;	/* gcc */
2334	enum scan_balance scan_balance;
2335	unsigned long ap, fp;
2336	enum lru_list lru;
2337
2338	/* If we have no swap space, do not bother scanning anon folios. */
2339	if (!sc->may_swap || !can_reclaim_anon_pages(memcg, pgdat->node_id, sc)) {
2340		scan_balance = SCAN_FILE;
2341		goto out;
2342	}
2343
2344	/*
2345	 * Global reclaim will swap to prevent OOM even with no
2346	 * swappiness, but memcg users want to use this knob to
2347	 * disable swapping for individual groups completely when
2348	 * using the memory controller's swap limit feature would be
2349	 * too expensive.
2350	 */
2351	if (cgroup_reclaim(sc) && !swappiness) {
2352		scan_balance = SCAN_FILE;
2353		goto out;
2354	}
2355
2356	/*
2357	 * Do not apply any pressure balancing cleverness when the
2358	 * system is close to OOM, scan both anon and file equally
2359	 * (unless the swappiness setting disagrees with swapping).
2360	 */
2361	if (!sc->priority && swappiness) {
2362		scan_balance = SCAN_EQUAL;
2363		goto out;
2364	}
2365
2366	/*
2367	 * If the system is almost out of file pages, force-scan anon.
2368	 */
2369	if (sc->file_is_tiny) {
2370		scan_balance = SCAN_ANON;
2371		goto out;
2372	}
2373
2374	/*
2375	 * If there is enough inactive page cache, we do not reclaim
2376	 * anything from the anonymous working right now.
2377	 */
2378	if (sc->cache_trim_mode) {
2379		scan_balance = SCAN_FILE;
2380		goto out;
2381	}
2382
2383	scan_balance = SCAN_FRACT;
2384	/*
2385	 * Calculate the pressure balance between anon and file pages.
2386	 *
2387	 * The amount of pressure we put on each LRU is inversely
2388	 * proportional to the cost of reclaiming each list, as
2389	 * determined by the share of pages that are refaulting, times
2390	 * the relative IO cost of bringing back a swapped out
2391	 * anonymous page vs reloading a filesystem page (swappiness).
2392	 *
2393	 * Although we limit that influence to ensure no list gets
2394	 * left behind completely: at least a third of the pressure is
2395	 * applied, before swappiness.
2396	 *
2397	 * With swappiness at 100, anon and file have equal IO cost.
2398	 */
2399	total_cost = sc->anon_cost + sc->file_cost;
2400	anon_cost = total_cost + sc->anon_cost;
2401	file_cost = total_cost + sc->file_cost;
2402	total_cost = anon_cost + file_cost;
2403
2404	ap = swappiness * (total_cost + 1);
2405	ap /= anon_cost + 1;
2406
2407	fp = (200 - swappiness) * (total_cost + 1);
2408	fp /= file_cost + 1;
2409
2410	fraction[0] = ap;
2411	fraction[1] = fp;
2412	denominator = ap + fp;
2413out:
2414	for_each_evictable_lru(lru) {
2415		int file = is_file_lru(lru);
2416		unsigned long lruvec_size;
2417		unsigned long low, min;
2418		unsigned long scan;
2419
2420		lruvec_size = lruvec_lru_size(lruvec, lru, sc->reclaim_idx);
2421		mem_cgroup_protection(sc->target_mem_cgroup, memcg,
2422				      &min, &low);
2423
2424		if (min || low) {
2425			/*
2426			 * Scale a cgroup's reclaim pressure by proportioning
2427			 * its current usage to its memory.low or memory.min
2428			 * setting.
2429			 *
2430			 * This is important, as otherwise scanning aggression
2431			 * becomes extremely binary -- from nothing as we
2432			 * approach the memory protection threshold, to totally
2433			 * nominal as we exceed it.  This results in requiring
2434			 * setting extremely liberal protection thresholds. It
2435			 * also means we simply get no protection at all if we
2436			 * set it too low, which is not ideal.
2437			 *
2438			 * If there is any protection in place, we reduce scan
2439			 * pressure by how much of the total memory used is
2440			 * within protection thresholds.
2441			 *
2442			 * There is one special case: in the first reclaim pass,
2443			 * we skip over all groups that are within their low
2444			 * protection. If that fails to reclaim enough pages to
2445			 * satisfy the reclaim goal, we come back and override
2446			 * the best-effort low protection. However, we still
2447			 * ideally want to honor how well-behaved groups are in
2448			 * that case instead of simply punishing them all
2449			 * equally. As such, we reclaim them based on how much
2450			 * memory they are using, reducing the scan pressure
2451			 * again by how much of the total memory used is under
2452			 * hard protection.
2453			 */
2454			unsigned long cgroup_size = mem_cgroup_size(memcg);
2455			unsigned long protection;
2456
2457			/* memory.low scaling, make sure we retry before OOM */
2458			if (!sc->memcg_low_reclaim && low > min) {
2459				protection = low;
2460				sc->memcg_low_skipped = 1;
2461			} else {
2462				protection = min;
2463			}
2464
2465			/* Avoid TOCTOU with earlier protection check */
2466			cgroup_size = max(cgroup_size, protection);
2467
2468			scan = lruvec_size - lruvec_size * protection /
2469				(cgroup_size + 1);
2470
2471			/*
2472			 * Minimally target SWAP_CLUSTER_MAX pages to keep
2473			 * reclaim moving forwards, avoiding decrementing
2474			 * sc->priority further than desirable.
2475			 */
2476			scan = max(scan, SWAP_CLUSTER_MAX);
2477		} else {
2478			scan = lruvec_size;
2479		}
2480
2481		scan >>= sc->priority;
2482
2483		/*
2484		 * If the cgroup's already been deleted, make sure to
2485		 * scrape out the remaining cache.
2486		 */
2487		if (!scan && !mem_cgroup_online(memcg))
2488			scan = min(lruvec_size, SWAP_CLUSTER_MAX);
2489
2490		switch (scan_balance) {
2491		case SCAN_EQUAL:
2492			/* Scan lists relative to size */
2493			break;
2494		case SCAN_FRACT:
2495			/*
2496			 * Scan types proportional to swappiness and
2497			 * their relative recent reclaim efficiency.
2498			 * Make sure we don't miss the last page on
2499			 * the offlined memory cgroups because of a
2500			 * round-off error.
2501			 */
2502			scan = mem_cgroup_online(memcg) ?
2503			       div64_u64(scan * fraction[file], denominator) :
2504			       DIV64_U64_ROUND_UP(scan * fraction[file],
2505						  denominator);
2506			break;
2507		case SCAN_FILE:
2508		case SCAN_ANON:
2509			/* Scan one type exclusively */
2510			if ((scan_balance == SCAN_FILE) != file)
2511				scan = 0;
2512			break;
2513		default:
2514			/* Look ma, no brain */
2515			BUG();
2516		}
2517
2518		nr[lru] = scan;
2519	}
2520}
2521
2522/*
2523 * Anonymous LRU management is a waste if there is
2524 * ultimately no way to reclaim the memory.
2525 */
2526static bool can_age_anon_pages(struct pglist_data *pgdat,
2527			       struct scan_control *sc)
2528{
2529	/* Aging the anon LRU is valuable if swap is present: */
2530	if (total_swap_pages > 0)
2531		return true;
2532
2533	/* Also valuable if anon pages can be demoted: */
2534	return can_demote(pgdat->node_id, sc);
2535}
2536
2537#ifdef CONFIG_LRU_GEN
2538
2539#ifdef CONFIG_LRU_GEN_ENABLED
2540DEFINE_STATIC_KEY_ARRAY_TRUE(lru_gen_caps, NR_LRU_GEN_CAPS);
2541#define get_cap(cap)	static_branch_likely(&lru_gen_caps[cap])
2542#else
2543DEFINE_STATIC_KEY_ARRAY_FALSE(lru_gen_caps, NR_LRU_GEN_CAPS);
2544#define get_cap(cap)	static_branch_unlikely(&lru_gen_caps[cap])
2545#endif
2546
2547static bool should_walk_mmu(void)
2548{
2549	return arch_has_hw_pte_young() && get_cap(LRU_GEN_MM_WALK);
2550}
2551
2552static bool should_clear_pmd_young(void)
2553{
2554	return arch_has_hw_nonleaf_pmd_young() && get_cap(LRU_GEN_NONLEAF_YOUNG);
2555}
2556
2557/******************************************************************************
2558 *                          shorthand helpers
2559 ******************************************************************************/
2560
2561#define LRU_REFS_FLAGS	(BIT(PG_referenced) | BIT(PG_workingset))
2562
2563#define DEFINE_MAX_SEQ(lruvec)						\
2564	unsigned long max_seq = READ_ONCE((lruvec)->lrugen.max_seq)
2565
2566#define DEFINE_MIN_SEQ(lruvec)						\
2567	unsigned long min_seq[ANON_AND_FILE] = {			\
2568		READ_ONCE((lruvec)->lrugen.min_seq[LRU_GEN_ANON]),	\
2569		READ_ONCE((lruvec)->lrugen.min_seq[LRU_GEN_FILE]),	\
2570	}
2571
2572#define for_each_gen_type_zone(gen, type, zone)				\
2573	for ((gen) = 0; (gen) < MAX_NR_GENS; (gen)++)			\
2574		for ((type) = 0; (type) < ANON_AND_FILE; (type)++)	\
2575			for ((zone) = 0; (zone) < MAX_NR_ZONES; (zone)++)
2576
2577#define get_memcg_gen(seq)	((seq) % MEMCG_NR_GENS)
2578#define get_memcg_bin(bin)	((bin) % MEMCG_NR_BINS)
2579
2580static struct lruvec *get_lruvec(struct mem_cgroup *memcg, int nid)
2581{
2582	struct pglist_data *pgdat = NODE_DATA(nid);
2583
2584#ifdef CONFIG_MEMCG
2585	if (memcg) {
2586		struct lruvec *lruvec = &memcg->nodeinfo[nid]->lruvec;
2587
2588		/* see the comment in mem_cgroup_lruvec() */
2589		if (!lruvec->pgdat)
2590			lruvec->pgdat = pgdat;
2591
2592		return lruvec;
2593	}
2594#endif
2595	VM_WARN_ON_ONCE(!mem_cgroup_disabled());
2596
2597	return &pgdat->__lruvec;
2598}
2599
2600static int get_swappiness(struct lruvec *lruvec, struct scan_control *sc)
2601{
2602	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
2603	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
2604
2605	if (!sc->may_swap)
2606		return 0;
2607
2608	if (!can_demote(pgdat->node_id, sc) &&
2609	    mem_cgroup_get_nr_swap_pages(memcg) < MIN_LRU_BATCH)
2610		return 0;
2611
2612	return mem_cgroup_swappiness(memcg);
2613}
2614
2615static int get_nr_gens(struct lruvec *lruvec, int type)
2616{
2617	return lruvec->lrugen.max_seq - lruvec->lrugen.min_seq[type] + 1;
2618}
2619
2620static bool __maybe_unused seq_is_valid(struct lruvec *lruvec)
2621{
2622	/* see the comment on lru_gen_folio */
2623	return get_nr_gens(lruvec, LRU_GEN_FILE) >= MIN_NR_GENS &&
2624	       get_nr_gens(lruvec, LRU_GEN_FILE) <= get_nr_gens(lruvec, LRU_GEN_ANON) &&
2625	       get_nr_gens(lruvec, LRU_GEN_ANON) <= MAX_NR_GENS;
2626}
2627
2628/******************************************************************************
2629 *                          Bloom filters
2630 ******************************************************************************/
2631
2632/*
2633 * Bloom filters with m=1<<15, k=2 and the false positive rates of ~1/5 when
2634 * n=10,000 and ~1/2 when n=20,000, where, conventionally, m is the number of
2635 * bits in a bitmap, k is the number of hash functions and n is the number of
2636 * inserted items.
2637 *
2638 * Page table walkers use one of the two filters to reduce their search space.
2639 * To get rid of non-leaf entries that no longer have enough leaf entries, the
2640 * aging uses the double-buffering technique to flip to the other filter each
2641 * time it produces a new generation. For non-leaf entries that have enough
2642 * leaf entries, the aging carries them over to the next generation in
2643 * walk_pmd_range(); the eviction also report them when walking the rmap
2644 * in lru_gen_look_around().
2645 *
2646 * For future optimizations:
2647 * 1. It's not necessary to keep both filters all the time. The spare one can be
2648 *    freed after the RCU grace period and reallocated if needed again.
2649 * 2. And when reallocating, it's worth scaling its size according to the number
2650 *    of inserted entries in the other filter, to reduce the memory overhead on
2651 *    small systems and false positives on large systems.
2652 * 3. Jenkins' hash function is an alternative to Knuth's.
2653 */
2654#define BLOOM_FILTER_SHIFT	15
2655
2656static inline int filter_gen_from_seq(unsigned long seq)
2657{
2658	return seq % NR_BLOOM_FILTERS;
2659}
2660
2661static void get_item_key(void *item, int *key)
2662{
2663	u32 hash = hash_ptr(item, BLOOM_FILTER_SHIFT * 2);
2664
2665	BUILD_BUG_ON(BLOOM_FILTER_SHIFT * 2 > BITS_PER_TYPE(u32));
2666
2667	key[0] = hash & (BIT(BLOOM_FILTER_SHIFT) - 1);
2668	key[1] = hash >> BLOOM_FILTER_SHIFT;
2669}
2670
2671static bool test_bloom_filter(struct lru_gen_mm_state *mm_state, unsigned long seq,
2672			      void *item)
2673{
2674	int key[2];
2675	unsigned long *filter;
2676	int gen = filter_gen_from_seq(seq);
2677
2678	filter = READ_ONCE(mm_state->filters[gen]);
2679	if (!filter)
2680		return true;
2681
2682	get_item_key(item, key);
2683
2684	return test_bit(key[0], filter) && test_bit(key[1], filter);
2685}
2686
2687static void update_bloom_filter(struct lru_gen_mm_state *mm_state, unsigned long seq,
2688				void *item)
2689{
2690	int key[2];
2691	unsigned long *filter;
2692	int gen = filter_gen_from_seq(seq);
2693
2694	filter = READ_ONCE(mm_state->filters[gen]);
2695	if (!filter)
2696		return;
2697
2698	get_item_key(item, key);
2699
2700	if (!test_bit(key[0], filter))
2701		set_bit(key[0], filter);
2702	if (!test_bit(key[1], filter))
2703		set_bit(key[1], filter);
2704}
2705
2706static void reset_bloom_filter(struct lru_gen_mm_state *mm_state, unsigned long seq)
2707{
2708	unsigned long *filter;
2709	int gen = filter_gen_from_seq(seq);
2710
2711	filter = mm_state->filters[gen];
2712	if (filter) {
2713		bitmap_clear(filter, 0, BIT(BLOOM_FILTER_SHIFT));
2714		return;
2715	}
2716
2717	filter = bitmap_zalloc(BIT(BLOOM_FILTER_SHIFT),
2718			       __GFP_HIGH | __GFP_NOMEMALLOC | __GFP_NOWARN);
2719	WRITE_ONCE(mm_state->filters[gen], filter);
2720}
2721
2722/******************************************************************************
2723 *                          mm_struct list
2724 ******************************************************************************/
2725
2726#ifdef CONFIG_LRU_GEN_WALKS_MMU
2727
2728static struct lru_gen_mm_list *get_mm_list(struct mem_cgroup *memcg)
2729{
2730	static struct lru_gen_mm_list mm_list = {
2731		.fifo = LIST_HEAD_INIT(mm_list.fifo),
2732		.lock = __SPIN_LOCK_UNLOCKED(mm_list.lock),
2733	};
2734
2735#ifdef CONFIG_MEMCG
2736	if (memcg)
2737		return &memcg->mm_list;
2738#endif
2739	VM_WARN_ON_ONCE(!mem_cgroup_disabled());
2740
2741	return &mm_list;
2742}
2743
2744static struct lru_gen_mm_state *get_mm_state(struct lruvec *lruvec)
2745{
2746	return &lruvec->mm_state;
2747}
2748
2749static struct mm_struct *get_next_mm(struct lru_gen_mm_walk *walk)
2750{
2751	int key;
2752	struct mm_struct *mm;
2753	struct pglist_data *pgdat = lruvec_pgdat(walk->lruvec);
2754	struct lru_gen_mm_state *mm_state = get_mm_state(walk->lruvec);
2755
2756	mm = list_entry(mm_state->head, struct mm_struct, lru_gen.list);
2757	key = pgdat->node_id % BITS_PER_TYPE(mm->lru_gen.bitmap);
2758
2759	if (!walk->force_scan && !test_bit(key, &mm->lru_gen.bitmap))
2760		return NULL;
2761
2762	clear_bit(key, &mm->lru_gen.bitmap);
2763
2764	return mmget_not_zero(mm) ? mm : NULL;
2765}
2766
2767void lru_gen_add_mm(struct mm_struct *mm)
2768{
2769	int nid;
2770	struct mem_cgroup *memcg = get_mem_cgroup_from_mm(mm);
2771	struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
2772
2773	VM_WARN_ON_ONCE(!list_empty(&mm->lru_gen.list));
2774#ifdef CONFIG_MEMCG
2775	VM_WARN_ON_ONCE(mm->lru_gen.memcg);
2776	mm->lru_gen.memcg = memcg;
2777#endif
2778	spin_lock(&mm_list->lock);
2779
2780	for_each_node_state(nid, N_MEMORY) {
2781		struct lruvec *lruvec = get_lruvec(memcg, nid);
2782		struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
2783
2784		/* the first addition since the last iteration */
2785		if (mm_state->tail == &mm_list->fifo)
2786			mm_state->tail = &mm->lru_gen.list;
2787	}
2788
2789	list_add_tail(&mm->lru_gen.list, &mm_list->fifo);
2790
2791	spin_unlock(&mm_list->lock);
2792}
2793
2794void lru_gen_del_mm(struct mm_struct *mm)
2795{
2796	int nid;
2797	struct lru_gen_mm_list *mm_list;
2798	struct mem_cgroup *memcg = NULL;
2799
2800	if (list_empty(&mm->lru_gen.list))
2801		return;
2802
2803#ifdef CONFIG_MEMCG
2804	memcg = mm->lru_gen.memcg;
2805#endif
2806	mm_list = get_mm_list(memcg);
2807
2808	spin_lock(&mm_list->lock);
2809
2810	for_each_node(nid) {
2811		struct lruvec *lruvec = get_lruvec(memcg, nid);
2812		struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
2813
2814		/* where the current iteration continues after */
2815		if (mm_state->head == &mm->lru_gen.list)
2816			mm_state->head = mm_state->head->prev;
2817
2818		/* where the last iteration ended before */
2819		if (mm_state->tail == &mm->lru_gen.list)
2820			mm_state->tail = mm_state->tail->next;
2821	}
2822
2823	list_del_init(&mm->lru_gen.list);
2824
2825	spin_unlock(&mm_list->lock);
2826
2827#ifdef CONFIG_MEMCG
2828	mem_cgroup_put(mm->lru_gen.memcg);
2829	mm->lru_gen.memcg = NULL;
2830#endif
2831}
2832
2833#ifdef CONFIG_MEMCG
2834void lru_gen_migrate_mm(struct mm_struct *mm)
2835{
2836	struct mem_cgroup *memcg;
2837	struct task_struct *task = rcu_dereference_protected(mm->owner, true);
2838
2839	VM_WARN_ON_ONCE(task->mm != mm);
2840	lockdep_assert_held(&task->alloc_lock);
2841
2842	/* for mm_update_next_owner() */
2843	if (mem_cgroup_disabled())
2844		return;
2845
2846	/* migration can happen before addition */
2847	if (!mm->lru_gen.memcg)
2848		return;
2849
2850	rcu_read_lock();
2851	memcg = mem_cgroup_from_task(task);
2852	rcu_read_unlock();
2853	if (memcg == mm->lru_gen.memcg)
2854		return;
2855
2856	VM_WARN_ON_ONCE(list_empty(&mm->lru_gen.list));
2857
2858	lru_gen_del_mm(mm);
2859	lru_gen_add_mm(mm);
2860}
2861#endif
2862
2863#else /* !CONFIG_LRU_GEN_WALKS_MMU */
2864
2865static struct lru_gen_mm_list *get_mm_list(struct mem_cgroup *memcg)
2866{
2867	return NULL;
2868}
2869
2870static struct lru_gen_mm_state *get_mm_state(struct lruvec *lruvec)
2871{
2872	return NULL;
2873}
2874
2875static struct mm_struct *get_next_mm(struct lru_gen_mm_walk *walk)
2876{
2877	return NULL;
2878}
2879
2880#endif
2881
2882static void reset_mm_stats(struct lruvec *lruvec, struct lru_gen_mm_walk *walk, bool last)
2883{
2884	int i;
2885	int hist;
 
2886	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
2887
2888	lockdep_assert_held(&get_mm_list(lruvec_memcg(lruvec))->lock);
2889
2890	if (walk) {
2891		hist = lru_hist_from_seq(walk->max_seq);
2892
2893		for (i = 0; i < NR_MM_STATS; i++) {
2894			WRITE_ONCE(mm_state->stats[hist][i],
2895				   mm_state->stats[hist][i] + walk->mm_stats[i]);
2896			walk->mm_stats[i] = 0;
2897		}
2898	}
2899
2900	if (NR_HIST_GENS > 1 && last) {
2901		hist = lru_hist_from_seq(mm_state->seq + 1);
2902
2903		for (i = 0; i < NR_MM_STATS; i++)
2904			WRITE_ONCE(mm_state->stats[hist][i], 0);
2905	}
2906}
2907
2908static bool iterate_mm_list(struct lruvec *lruvec, struct lru_gen_mm_walk *walk,
2909			    struct mm_struct **iter)
2910{
2911	bool first = false;
2912	bool last = false;
2913	struct mm_struct *mm = NULL;
 
2914	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
2915	struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
2916	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
2917
2918	/*
2919	 * mm_state->seq is incremented after each iteration of mm_list. There
2920	 * are three interesting cases for this page table walker:
2921	 * 1. It tries to start a new iteration with a stale max_seq: there is
2922	 *    nothing left to do.
2923	 * 2. It started the next iteration: it needs to reset the Bloom filter
2924	 *    so that a fresh set of PTE tables can be recorded.
2925	 * 3. It ended the current iteration: it needs to reset the mm stats
2926	 *    counters and tell its caller to increment max_seq.
2927	 */
2928	spin_lock(&mm_list->lock);
2929
2930	VM_WARN_ON_ONCE(mm_state->seq + 1 < walk->max_seq);
2931
2932	if (walk->max_seq <= mm_state->seq)
2933		goto done;
2934
2935	if (!mm_state->head)
2936		mm_state->head = &mm_list->fifo;
2937
2938	if (mm_state->head == &mm_list->fifo)
2939		first = true;
2940
2941	do {
2942		mm_state->head = mm_state->head->next;
2943		if (mm_state->head == &mm_list->fifo) {
2944			WRITE_ONCE(mm_state->seq, mm_state->seq + 1);
2945			last = true;
2946			break;
2947		}
2948
2949		/* force scan for those added after the last iteration */
2950		if (!mm_state->tail || mm_state->tail == mm_state->head) {
2951			mm_state->tail = mm_state->head->next;
2952			walk->force_scan = true;
2953		}
2954	} while (!(mm = get_next_mm(walk)));
2955done:
2956	if (*iter || last)
2957		reset_mm_stats(lruvec, walk, last);
2958
2959	spin_unlock(&mm_list->lock);
2960
2961	if (mm && first)
2962		reset_bloom_filter(mm_state, walk->max_seq + 1);
2963
2964	if (*iter)
2965		mmput_async(*iter);
2966
2967	*iter = mm;
2968
2969	return last;
2970}
2971
2972static bool iterate_mm_list_nowalk(struct lruvec *lruvec, unsigned long max_seq)
2973{
2974	bool success = false;
2975	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
2976	struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
2977	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
2978
2979	spin_lock(&mm_list->lock);
2980
2981	VM_WARN_ON_ONCE(mm_state->seq + 1 < max_seq);
2982
2983	if (max_seq > mm_state->seq) {
2984		mm_state->head = NULL;
2985		mm_state->tail = NULL;
2986		WRITE_ONCE(mm_state->seq, mm_state->seq + 1);
2987		reset_mm_stats(lruvec, NULL, true);
2988		success = true;
2989	}
2990
2991	spin_unlock(&mm_list->lock);
2992
2993	return success;
2994}
2995
2996/******************************************************************************
2997 *                          PID controller
2998 ******************************************************************************/
2999
3000/*
3001 * A feedback loop based on Proportional-Integral-Derivative (PID) controller.
3002 *
3003 * The P term is refaulted/(evicted+protected) from a tier in the generation
3004 * currently being evicted; the I term is the exponential moving average of the
3005 * P term over the generations previously evicted, using the smoothing factor
3006 * 1/2; the D term isn't supported.
3007 *
3008 * The setpoint (SP) is always the first tier of one type; the process variable
3009 * (PV) is either any tier of the other type or any other tier of the same
3010 * type.
3011 *
3012 * The error is the difference between the SP and the PV; the correction is to
3013 * turn off protection when SP>PV or turn on protection when SP<PV.
3014 *
3015 * For future optimizations:
3016 * 1. The D term may discount the other two terms over time so that long-lived
3017 *    generations can resist stale information.
3018 */
3019struct ctrl_pos {
3020	unsigned long refaulted;
3021	unsigned long total;
3022	int gain;
3023};
3024
3025static void read_ctrl_pos(struct lruvec *lruvec, int type, int tier, int gain,
3026			  struct ctrl_pos *pos)
3027{
3028	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3029	int hist = lru_hist_from_seq(lrugen->min_seq[type]);
3030
3031	pos->refaulted = lrugen->avg_refaulted[type][tier] +
3032			 atomic_long_read(&lrugen->refaulted[hist][type][tier]);
3033	pos->total = lrugen->avg_total[type][tier] +
3034		     atomic_long_read(&lrugen->evicted[hist][type][tier]);
3035	if (tier)
3036		pos->total += lrugen->protected[hist][type][tier - 1];
3037	pos->gain = gain;
3038}
3039
3040static void reset_ctrl_pos(struct lruvec *lruvec, int type, bool carryover)
3041{
3042	int hist, tier;
3043	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3044	bool clear = carryover ? NR_HIST_GENS == 1 : NR_HIST_GENS > 1;
3045	unsigned long seq = carryover ? lrugen->min_seq[type] : lrugen->max_seq + 1;
3046
3047	lockdep_assert_held(&lruvec->lru_lock);
3048
3049	if (!carryover && !clear)
3050		return;
3051
3052	hist = lru_hist_from_seq(seq);
3053
3054	for (tier = 0; tier < MAX_NR_TIERS; tier++) {
3055		if (carryover) {
3056			unsigned long sum;
3057
3058			sum = lrugen->avg_refaulted[type][tier] +
3059			      atomic_long_read(&lrugen->refaulted[hist][type][tier]);
3060			WRITE_ONCE(lrugen->avg_refaulted[type][tier], sum / 2);
3061
3062			sum = lrugen->avg_total[type][tier] +
3063			      atomic_long_read(&lrugen->evicted[hist][type][tier]);
3064			if (tier)
3065				sum += lrugen->protected[hist][type][tier - 1];
3066			WRITE_ONCE(lrugen->avg_total[type][tier], sum / 2);
3067		}
3068
3069		if (clear) {
3070			atomic_long_set(&lrugen->refaulted[hist][type][tier], 0);
3071			atomic_long_set(&lrugen->evicted[hist][type][tier], 0);
3072			if (tier)
3073				WRITE_ONCE(lrugen->protected[hist][type][tier - 1], 0);
3074		}
3075	}
3076}
3077
3078static bool positive_ctrl_err(struct ctrl_pos *sp, struct ctrl_pos *pv)
3079{
3080	/*
3081	 * Return true if the PV has a limited number of refaults or a lower
3082	 * refaulted/total than the SP.
3083	 */
3084	return pv->refaulted < MIN_LRU_BATCH ||
3085	       pv->refaulted * (sp->total + MIN_LRU_BATCH) * sp->gain <=
3086	       (sp->refaulted + 1) * pv->total * pv->gain;
3087}
3088
3089/******************************************************************************
3090 *                          the aging
3091 ******************************************************************************/
3092
3093/* promote pages accessed through page tables */
3094static int folio_update_gen(struct folio *folio, int gen)
3095{
3096	unsigned long new_flags, old_flags = READ_ONCE(folio->flags);
3097
3098	VM_WARN_ON_ONCE(gen >= MAX_NR_GENS);
3099	VM_WARN_ON_ONCE(!rcu_read_lock_held());
3100
3101	do {
3102		/* lru_gen_del_folio() has isolated this page? */
3103		if (!(old_flags & LRU_GEN_MASK)) {
3104			/* for shrink_folio_list() */
3105			new_flags = old_flags | BIT(PG_referenced);
3106			continue;
3107		}
3108
3109		new_flags = old_flags & ~(LRU_GEN_MASK | LRU_REFS_MASK | LRU_REFS_FLAGS);
3110		new_flags |= (gen + 1UL) << LRU_GEN_PGOFF;
3111	} while (!try_cmpxchg(&folio->flags, &old_flags, new_flags));
3112
3113	return ((old_flags & LRU_GEN_MASK) >> LRU_GEN_PGOFF) - 1;
3114}
3115
3116/* protect pages accessed multiple times through file descriptors */
3117static int folio_inc_gen(struct lruvec *lruvec, struct folio *folio, bool reclaiming)
3118{
3119	int type = folio_is_file_lru(folio);
3120	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3121	int new_gen, old_gen = lru_gen_from_seq(lrugen->min_seq[type]);
3122	unsigned long new_flags, old_flags = READ_ONCE(folio->flags);
3123
3124	VM_WARN_ON_ONCE_FOLIO(!(old_flags & LRU_GEN_MASK), folio);
3125
3126	do {
3127		new_gen = ((old_flags & LRU_GEN_MASK) >> LRU_GEN_PGOFF) - 1;
3128		/* folio_update_gen() has promoted this page? */
3129		if (new_gen >= 0 && new_gen != old_gen)
3130			return new_gen;
3131
3132		new_gen = (old_gen + 1) % MAX_NR_GENS;
3133
3134		new_flags = old_flags & ~(LRU_GEN_MASK | LRU_REFS_MASK | LRU_REFS_FLAGS);
3135		new_flags |= (new_gen + 1UL) << LRU_GEN_PGOFF;
3136		/* for folio_end_writeback() */
3137		if (reclaiming)
3138			new_flags |= BIT(PG_reclaim);
3139	} while (!try_cmpxchg(&folio->flags, &old_flags, new_flags));
3140
3141	lru_gen_update_size(lruvec, folio, old_gen, new_gen);
3142
3143	return new_gen;
3144}
3145
3146static void update_batch_size(struct lru_gen_mm_walk *walk, struct folio *folio,
3147			      int old_gen, int new_gen)
3148{
3149	int type = folio_is_file_lru(folio);
3150	int zone = folio_zonenum(folio);
3151	int delta = folio_nr_pages(folio);
3152
3153	VM_WARN_ON_ONCE(old_gen >= MAX_NR_GENS);
3154	VM_WARN_ON_ONCE(new_gen >= MAX_NR_GENS);
3155
3156	walk->batched++;
3157
3158	walk->nr_pages[old_gen][type][zone] -= delta;
3159	walk->nr_pages[new_gen][type][zone] += delta;
3160}
3161
3162static void reset_batch_size(struct lruvec *lruvec, struct lru_gen_mm_walk *walk)
3163{
3164	int gen, type, zone;
 
3165	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3166
3167	walk->batched = 0;
3168
3169	for_each_gen_type_zone(gen, type, zone) {
3170		enum lru_list lru = type * LRU_INACTIVE_FILE;
3171		int delta = walk->nr_pages[gen][type][zone];
3172
3173		if (!delta)
3174			continue;
3175
3176		walk->nr_pages[gen][type][zone] = 0;
3177		WRITE_ONCE(lrugen->nr_pages[gen][type][zone],
3178			   lrugen->nr_pages[gen][type][zone] + delta);
3179
3180		if (lru_gen_is_active(lruvec, gen))
3181			lru += LRU_ACTIVE;
3182		__update_lru_size(lruvec, lru, zone, delta);
3183	}
3184}
3185
3186static int should_skip_vma(unsigned long start, unsigned long end, struct mm_walk *args)
3187{
3188	struct address_space *mapping;
3189	struct vm_area_struct *vma = args->vma;
3190	struct lru_gen_mm_walk *walk = args->private;
3191
3192	if (!vma_is_accessible(vma))
3193		return true;
3194
3195	if (is_vm_hugetlb_page(vma))
3196		return true;
3197
3198	if (!vma_has_recency(vma))
3199		return true;
3200
3201	if (vma->vm_flags & (VM_LOCKED | VM_SPECIAL))
3202		return true;
3203
3204	if (vma == get_gate_vma(vma->vm_mm))
3205		return true;
3206
3207	if (vma_is_anonymous(vma))
3208		return !walk->can_swap;
3209
3210	if (WARN_ON_ONCE(!vma->vm_file || !vma->vm_file->f_mapping))
3211		return true;
3212
3213	mapping = vma->vm_file->f_mapping;
3214	if (mapping_unevictable(mapping))
3215		return true;
3216
3217	if (shmem_mapping(mapping))
3218		return !walk->can_swap;
3219
3220	/* to exclude special mappings like dax, etc. */
3221	return !mapping->a_ops->read_folio;
3222}
3223
3224/*
3225 * Some userspace memory allocators map many single-page VMAs. Instead of
3226 * returning back to the PGD table for each of such VMAs, finish an entire PMD
3227 * table to reduce zigzags and improve cache performance.
3228 */
3229static bool get_next_vma(unsigned long mask, unsigned long size, struct mm_walk *args,
3230			 unsigned long *vm_start, unsigned long *vm_end)
3231{
3232	unsigned long start = round_up(*vm_end, size);
3233	unsigned long end = (start | ~mask) + 1;
3234	VMA_ITERATOR(vmi, args->mm, start);
3235
3236	VM_WARN_ON_ONCE(mask & size);
3237	VM_WARN_ON_ONCE((start & mask) != (*vm_start & mask));
3238
3239	for_each_vma(vmi, args->vma) {
3240		if (end && end <= args->vma->vm_start)
3241			return false;
3242
3243		if (should_skip_vma(args->vma->vm_start, args->vma->vm_end, args))
3244			continue;
3245
3246		*vm_start = max(start, args->vma->vm_start);
3247		*vm_end = min(end - 1, args->vma->vm_end - 1) + 1;
3248
3249		return true;
3250	}
3251
3252	return false;
3253}
3254
3255static unsigned long get_pte_pfn(pte_t pte, struct vm_area_struct *vma, unsigned long addr)
3256{
3257	unsigned long pfn = pte_pfn(pte);
3258
3259	VM_WARN_ON_ONCE(addr < vma->vm_start || addr >= vma->vm_end);
3260
3261	if (!pte_present(pte) || is_zero_pfn(pfn))
3262		return -1;
3263
3264	if (WARN_ON_ONCE(pte_devmap(pte) || pte_special(pte)))
3265		return -1;
3266
3267	if (WARN_ON_ONCE(!pfn_valid(pfn)))
3268		return -1;
3269
3270	return pfn;
3271}
3272
3273static unsigned long get_pmd_pfn(pmd_t pmd, struct vm_area_struct *vma, unsigned long addr)
3274{
3275	unsigned long pfn = pmd_pfn(pmd);
3276
3277	VM_WARN_ON_ONCE(addr < vma->vm_start || addr >= vma->vm_end);
3278
3279	if (!pmd_present(pmd) || is_huge_zero_pmd(pmd))
3280		return -1;
3281
3282	if (WARN_ON_ONCE(pmd_devmap(pmd)))
3283		return -1;
3284
3285	if (WARN_ON_ONCE(!pfn_valid(pfn)))
3286		return -1;
3287
3288	return pfn;
3289}
3290
3291static struct folio *get_pfn_folio(unsigned long pfn, struct mem_cgroup *memcg,
3292				   struct pglist_data *pgdat, bool can_swap)
3293{
3294	struct folio *folio;
3295
3296	/* try to avoid unnecessary memory loads */
3297	if (pfn < pgdat->node_start_pfn || pfn >= pgdat_end_pfn(pgdat))
3298		return NULL;
3299
3300	folio = pfn_folio(pfn);
3301	if (folio_nid(folio) != pgdat->node_id)
3302		return NULL;
3303
3304	if (folio_memcg_rcu(folio) != memcg)
3305		return NULL;
3306
3307	/* file VMAs can contain anon pages from COW */
3308	if (!folio_is_file_lru(folio) && !can_swap)
3309		return NULL;
3310
3311	return folio;
3312}
3313
3314static bool suitable_to_scan(int total, int young)
3315{
3316	int n = clamp_t(int, cache_line_size() / sizeof(pte_t), 2, 8);
3317
3318	/* suitable if the average number of young PTEs per cacheline is >=1 */
3319	return young * n >= total;
3320}
3321
3322static bool walk_pte_range(pmd_t *pmd, unsigned long start, unsigned long end,
3323			   struct mm_walk *args)
3324{
3325	int i;
3326	pte_t *pte;
3327	spinlock_t *ptl;
3328	unsigned long addr;
3329	int total = 0;
3330	int young = 0;
3331	struct lru_gen_mm_walk *walk = args->private;
3332	struct mem_cgroup *memcg = lruvec_memcg(walk->lruvec);
3333	struct pglist_data *pgdat = lruvec_pgdat(walk->lruvec);
3334	int old_gen, new_gen = lru_gen_from_seq(walk->max_seq);
 
3335
3336	pte = pte_offset_map_nolock(args->mm, pmd, start & PMD_MASK, &ptl);
3337	if (!pte)
3338		return false;
3339	if (!spin_trylock(ptl)) {
3340		pte_unmap(pte);
3341		return false;
3342	}
3343
3344	arch_enter_lazy_mmu_mode();
3345restart:
3346	for (i = pte_index(start), addr = start; addr != end; i++, addr += PAGE_SIZE) {
3347		unsigned long pfn;
3348		struct folio *folio;
3349		pte_t ptent = ptep_get(pte + i);
3350
3351		total++;
3352		walk->mm_stats[MM_LEAF_TOTAL]++;
3353
3354		pfn = get_pte_pfn(ptent, args->vma, addr);
3355		if (pfn == -1)
3356			continue;
3357
3358		if (!pte_young(ptent)) {
3359			walk->mm_stats[MM_LEAF_OLD]++;
3360			continue;
3361		}
3362
3363		folio = get_pfn_folio(pfn, memcg, pgdat, walk->can_swap);
3364		if (!folio)
3365			continue;
3366
3367		if (!ptep_test_and_clear_young(args->vma, addr, pte + i))
3368			VM_WARN_ON_ONCE(true);
3369
3370		young++;
3371		walk->mm_stats[MM_LEAF_YOUNG]++;
3372
3373		if (pte_dirty(ptent) && !folio_test_dirty(folio) &&
3374		    !(folio_test_anon(folio) && folio_test_swapbacked(folio) &&
3375		      !folio_test_swapcache(folio)))
3376			folio_mark_dirty(folio);
3377
3378		old_gen = folio_update_gen(folio, new_gen);
3379		if (old_gen >= 0 && old_gen != new_gen)
3380			update_batch_size(walk, folio, old_gen, new_gen);
3381	}
3382
3383	if (i < PTRS_PER_PTE && get_next_vma(PMD_MASK, PAGE_SIZE, args, &start, &end))
3384		goto restart;
3385
3386	arch_leave_lazy_mmu_mode();
3387	pte_unmap_unlock(pte, ptl);
3388
3389	return suitable_to_scan(total, young);
3390}
3391
3392static void walk_pmd_range_locked(pud_t *pud, unsigned long addr, struct vm_area_struct *vma,
3393				  struct mm_walk *args, unsigned long *bitmap, unsigned long *first)
3394{
3395	int i;
3396	pmd_t *pmd;
3397	spinlock_t *ptl;
3398	struct lru_gen_mm_walk *walk = args->private;
3399	struct mem_cgroup *memcg = lruvec_memcg(walk->lruvec);
3400	struct pglist_data *pgdat = lruvec_pgdat(walk->lruvec);
3401	int old_gen, new_gen = lru_gen_from_seq(walk->max_seq);
 
3402
3403	VM_WARN_ON_ONCE(pud_leaf(*pud));
3404
3405	/* try to batch at most 1+MIN_LRU_BATCH+1 entries */
3406	if (*first == -1) {
3407		*first = addr;
3408		bitmap_zero(bitmap, MIN_LRU_BATCH);
3409		return;
3410	}
3411
3412	i = addr == -1 ? 0 : pmd_index(addr) - pmd_index(*first);
3413	if (i && i <= MIN_LRU_BATCH) {
3414		__set_bit(i - 1, bitmap);
3415		return;
3416	}
3417
3418	pmd = pmd_offset(pud, *first);
3419
3420	ptl = pmd_lockptr(args->mm, pmd);
3421	if (!spin_trylock(ptl))
3422		goto done;
3423
3424	arch_enter_lazy_mmu_mode();
3425
3426	do {
3427		unsigned long pfn;
3428		struct folio *folio;
3429
3430		/* don't round down the first address */
3431		addr = i ? (*first & PMD_MASK) + i * PMD_SIZE : *first;
3432
3433		pfn = get_pmd_pfn(pmd[i], vma, addr);
3434		if (pfn == -1)
3435			goto next;
3436
3437		if (!pmd_trans_huge(pmd[i])) {
3438			if (should_clear_pmd_young())
3439				pmdp_test_and_clear_young(vma, addr, pmd + i);
3440			goto next;
3441		}
3442
3443		folio = get_pfn_folio(pfn, memcg, pgdat, walk->can_swap);
3444		if (!folio)
3445			goto next;
3446
3447		if (!pmdp_test_and_clear_young(vma, addr, pmd + i))
3448			goto next;
3449
3450		walk->mm_stats[MM_LEAF_YOUNG]++;
3451
3452		if (pmd_dirty(pmd[i]) && !folio_test_dirty(folio) &&
3453		    !(folio_test_anon(folio) && folio_test_swapbacked(folio) &&
3454		      !folio_test_swapcache(folio)))
3455			folio_mark_dirty(folio);
3456
3457		old_gen = folio_update_gen(folio, new_gen);
3458		if (old_gen >= 0 && old_gen != new_gen)
3459			update_batch_size(walk, folio, old_gen, new_gen);
3460next:
3461		i = i > MIN_LRU_BATCH ? 0 : find_next_bit(bitmap, MIN_LRU_BATCH, i) + 1;
3462	} while (i <= MIN_LRU_BATCH);
3463
3464	arch_leave_lazy_mmu_mode();
3465	spin_unlock(ptl);
3466done:
3467	*first = -1;
3468}
3469
3470static void walk_pmd_range(pud_t *pud, unsigned long start, unsigned long end,
3471			   struct mm_walk *args)
3472{
3473	int i;
3474	pmd_t *pmd;
3475	unsigned long next;
3476	unsigned long addr;
3477	struct vm_area_struct *vma;
3478	DECLARE_BITMAP(bitmap, MIN_LRU_BATCH);
3479	unsigned long first = -1;
3480	struct lru_gen_mm_walk *walk = args->private;
3481	struct lru_gen_mm_state *mm_state = get_mm_state(walk->lruvec);
3482
3483	VM_WARN_ON_ONCE(pud_leaf(*pud));
3484
3485	/*
3486	 * Finish an entire PMD in two passes: the first only reaches to PTE
3487	 * tables to avoid taking the PMD lock; the second, if necessary, takes
3488	 * the PMD lock to clear the accessed bit in PMD entries.
3489	 */
3490	pmd = pmd_offset(pud, start & PUD_MASK);
3491restart:
3492	/* walk_pte_range() may call get_next_vma() */
3493	vma = args->vma;
3494	for (i = pmd_index(start), addr = start; addr != end; i++, addr = next) {
3495		pmd_t val = pmdp_get_lockless(pmd + i);
3496
3497		next = pmd_addr_end(addr, end);
3498
3499		if (!pmd_present(val) || is_huge_zero_pmd(val)) {
3500			walk->mm_stats[MM_LEAF_TOTAL]++;
3501			continue;
3502		}
3503
3504		if (pmd_trans_huge(val)) {
3505			unsigned long pfn = pmd_pfn(val);
3506			struct pglist_data *pgdat = lruvec_pgdat(walk->lruvec);
3507
3508			walk->mm_stats[MM_LEAF_TOTAL]++;
3509
3510			if (!pmd_young(val)) {
3511				walk->mm_stats[MM_LEAF_OLD]++;
3512				continue;
3513			}
3514
3515			/* try to avoid unnecessary memory loads */
3516			if (pfn < pgdat->node_start_pfn || pfn >= pgdat_end_pfn(pgdat))
3517				continue;
3518
3519			walk_pmd_range_locked(pud, addr, vma, args, bitmap, &first);
3520			continue;
3521		}
3522
3523		walk->mm_stats[MM_NONLEAF_TOTAL]++;
3524
3525		if (should_clear_pmd_young()) {
3526			if (!pmd_young(val))
3527				continue;
3528
3529			walk_pmd_range_locked(pud, addr, vma, args, bitmap, &first);
3530		}
3531
3532		if (!walk->force_scan && !test_bloom_filter(mm_state, walk->max_seq, pmd + i))
3533			continue;
3534
3535		walk->mm_stats[MM_NONLEAF_FOUND]++;
3536
3537		if (!walk_pte_range(&val, addr, next, args))
3538			continue;
3539
3540		walk->mm_stats[MM_NONLEAF_ADDED]++;
3541
3542		/* carry over to the next generation */
3543		update_bloom_filter(mm_state, walk->max_seq + 1, pmd + i);
3544	}
3545
3546	walk_pmd_range_locked(pud, -1, vma, args, bitmap, &first);
3547
3548	if (i < PTRS_PER_PMD && get_next_vma(PUD_MASK, PMD_SIZE, args, &start, &end))
3549		goto restart;
3550}
3551
3552static int walk_pud_range(p4d_t *p4d, unsigned long start, unsigned long end,
3553			  struct mm_walk *args)
3554{
3555	int i;
3556	pud_t *pud;
3557	unsigned long addr;
3558	unsigned long next;
3559	struct lru_gen_mm_walk *walk = args->private;
3560
3561	VM_WARN_ON_ONCE(p4d_leaf(*p4d));
3562
3563	pud = pud_offset(p4d, start & P4D_MASK);
3564restart:
3565	for (i = pud_index(start), addr = start; addr != end; i++, addr = next) {
3566		pud_t val = READ_ONCE(pud[i]);
3567
3568		next = pud_addr_end(addr, end);
3569
3570		if (!pud_present(val) || WARN_ON_ONCE(pud_leaf(val)))
3571			continue;
3572
3573		walk_pmd_range(&val, addr, next, args);
3574
3575		if (need_resched() || walk->batched >= MAX_LRU_BATCH) {
3576			end = (addr | ~PUD_MASK) + 1;
3577			goto done;
3578		}
3579	}
3580
3581	if (i < PTRS_PER_PUD && get_next_vma(P4D_MASK, PUD_SIZE, args, &start, &end))
3582		goto restart;
3583
3584	end = round_up(end, P4D_SIZE);
3585done:
3586	if (!end || !args->vma)
3587		return 1;
3588
3589	walk->next_addr = max(end, args->vma->vm_start);
3590
3591	return -EAGAIN;
3592}
3593
3594static void walk_mm(struct lruvec *lruvec, struct mm_struct *mm, struct lru_gen_mm_walk *walk)
3595{
3596	static const struct mm_walk_ops mm_walk_ops = {
3597		.test_walk = should_skip_vma,
3598		.p4d_entry = walk_pud_range,
3599		.walk_lock = PGWALK_RDLOCK,
3600	};
3601
3602	int err;
 
3603	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
3604
3605	walk->next_addr = FIRST_USER_ADDRESS;
3606
3607	do {
3608		DEFINE_MAX_SEQ(lruvec);
3609
3610		err = -EBUSY;
3611
3612		/* another thread might have called inc_max_seq() */
3613		if (walk->max_seq != max_seq)
3614			break;
3615
3616		/* folio_update_gen() requires stable folio_memcg() */
3617		if (!mem_cgroup_trylock_pages(memcg))
3618			break;
3619
3620		/* the caller might be holding the lock for write */
3621		if (mmap_read_trylock(mm)) {
3622			err = walk_page_range(mm, walk->next_addr, ULONG_MAX, &mm_walk_ops, walk);
3623
3624			mmap_read_unlock(mm);
3625		}
3626
3627		mem_cgroup_unlock_pages();
3628
3629		if (walk->batched) {
3630			spin_lock_irq(&lruvec->lru_lock);
3631			reset_batch_size(lruvec, walk);
3632			spin_unlock_irq(&lruvec->lru_lock);
3633		}
3634
3635		cond_resched();
3636	} while (err == -EAGAIN);
3637}
3638
3639static struct lru_gen_mm_walk *set_mm_walk(struct pglist_data *pgdat, bool force_alloc)
3640{
3641	struct lru_gen_mm_walk *walk = current->reclaim_state->mm_walk;
3642
3643	if (pgdat && current_is_kswapd()) {
3644		VM_WARN_ON_ONCE(walk);
3645
3646		walk = &pgdat->mm_walk;
3647	} else if (!walk && force_alloc) {
3648		VM_WARN_ON_ONCE(current_is_kswapd());
3649
3650		walk = kzalloc(sizeof(*walk), __GFP_HIGH | __GFP_NOMEMALLOC | __GFP_NOWARN);
3651	}
3652
3653	current->reclaim_state->mm_walk = walk;
3654
3655	return walk;
3656}
3657
3658static void clear_mm_walk(void)
3659{
3660	struct lru_gen_mm_walk *walk = current->reclaim_state->mm_walk;
3661
3662	VM_WARN_ON_ONCE(walk && memchr_inv(walk->nr_pages, 0, sizeof(walk->nr_pages)));
3663	VM_WARN_ON_ONCE(walk && memchr_inv(walk->mm_stats, 0, sizeof(walk->mm_stats)));
3664
3665	current->reclaim_state->mm_walk = NULL;
3666
3667	if (!current_is_kswapd())
3668		kfree(walk);
3669}
3670
3671static bool inc_min_seq(struct lruvec *lruvec, int type, bool can_swap)
3672{
3673	int zone;
3674	int remaining = MAX_LRU_BATCH;
3675	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3676	int new_gen, old_gen = lru_gen_from_seq(lrugen->min_seq[type]);
3677
3678	if (type == LRU_GEN_ANON && !can_swap)
3679		goto done;
3680
3681	/* prevent cold/hot inversion if force_scan is true */
3682	for (zone = 0; zone < MAX_NR_ZONES; zone++) {
3683		struct list_head *head = &lrugen->folios[old_gen][type][zone];
3684
3685		while (!list_empty(head)) {
3686			struct folio *folio = lru_to_folio(head);
3687
3688			VM_WARN_ON_ONCE_FOLIO(folio_test_unevictable(folio), folio);
3689			VM_WARN_ON_ONCE_FOLIO(folio_test_active(folio), folio);
3690			VM_WARN_ON_ONCE_FOLIO(folio_is_file_lru(folio) != type, folio);
3691			VM_WARN_ON_ONCE_FOLIO(folio_zonenum(folio) != zone, folio);
3692
3693			new_gen = folio_inc_gen(lruvec, folio, false);
3694			list_move_tail(&folio->lru, &lrugen->folios[new_gen][type][zone]);
3695
3696			if (!--remaining)
3697				return false;
3698		}
3699	}
3700done:
3701	reset_ctrl_pos(lruvec, type, true);
3702	WRITE_ONCE(lrugen->min_seq[type], lrugen->min_seq[type] + 1);
3703
3704	return true;
3705}
3706
3707static bool try_to_inc_min_seq(struct lruvec *lruvec, bool can_swap)
3708{
3709	int gen, type, zone;
3710	bool success = false;
3711	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3712	DEFINE_MIN_SEQ(lruvec);
3713
3714	VM_WARN_ON_ONCE(!seq_is_valid(lruvec));
3715
3716	/* find the oldest populated generation */
3717	for (type = !can_swap; type < ANON_AND_FILE; type++) {
3718		while (min_seq[type] + MIN_NR_GENS <= lrugen->max_seq) {
3719			gen = lru_gen_from_seq(min_seq[type]);
3720
3721			for (zone = 0; zone < MAX_NR_ZONES; zone++) {
3722				if (!list_empty(&lrugen->folios[gen][type][zone]))
3723					goto next;
3724			}
3725
3726			min_seq[type]++;
3727		}
3728next:
3729		;
3730	}
3731
3732	/* see the comment on lru_gen_folio */
3733	if (can_swap) {
3734		min_seq[LRU_GEN_ANON] = min(min_seq[LRU_GEN_ANON], min_seq[LRU_GEN_FILE]);
3735		min_seq[LRU_GEN_FILE] = max(min_seq[LRU_GEN_ANON], lrugen->min_seq[LRU_GEN_FILE]);
3736	}
3737
3738	for (type = !can_swap; type < ANON_AND_FILE; type++) {
3739		if (min_seq[type] == lrugen->min_seq[type])
3740			continue;
3741
3742		reset_ctrl_pos(lruvec, type, true);
3743		WRITE_ONCE(lrugen->min_seq[type], min_seq[type]);
3744		success = true;
3745	}
3746
3747	return success;
3748}
3749
3750static bool inc_max_seq(struct lruvec *lruvec, unsigned long max_seq,
3751			bool can_swap, bool force_scan)
3752{
3753	bool success;
3754	int prev, next;
3755	int type, zone;
3756	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3757restart:
3758	if (max_seq < READ_ONCE(lrugen->max_seq))
3759		return false;
3760
3761	spin_lock_irq(&lruvec->lru_lock);
3762
3763	VM_WARN_ON_ONCE(!seq_is_valid(lruvec));
3764
3765	success = max_seq == lrugen->max_seq;
3766	if (!success)
3767		goto unlock;
3768
3769	for (type = ANON_AND_FILE - 1; type >= 0; type--) {
3770		if (get_nr_gens(lruvec, type) != MAX_NR_GENS)
3771			continue;
3772
3773		VM_WARN_ON_ONCE(!force_scan && (type == LRU_GEN_FILE || can_swap));
3774
3775		if (inc_min_seq(lruvec, type, can_swap))
3776			continue;
3777
3778		spin_unlock_irq(&lruvec->lru_lock);
3779		cond_resched();
3780		goto restart;
3781	}
3782
3783	/*
3784	 * Update the active/inactive LRU sizes for compatibility. Both sides of
3785	 * the current max_seq need to be covered, since max_seq+1 can overlap
3786	 * with min_seq[LRU_GEN_ANON] if swapping is constrained. And if they do
3787	 * overlap, cold/hot inversion happens.
3788	 */
3789	prev = lru_gen_from_seq(lrugen->max_seq - 1);
3790	next = lru_gen_from_seq(lrugen->max_seq + 1);
3791
3792	for (type = 0; type < ANON_AND_FILE; type++) {
3793		for (zone = 0; zone < MAX_NR_ZONES; zone++) {
3794			enum lru_list lru = type * LRU_INACTIVE_FILE;
3795			long delta = lrugen->nr_pages[prev][type][zone] -
3796				     lrugen->nr_pages[next][type][zone];
3797
3798			if (!delta)
3799				continue;
3800
3801			__update_lru_size(lruvec, lru, zone, delta);
3802			__update_lru_size(lruvec, lru + LRU_ACTIVE, zone, -delta);
3803		}
3804	}
3805
3806	for (type = 0; type < ANON_AND_FILE; type++)
3807		reset_ctrl_pos(lruvec, type, false);
3808
3809	WRITE_ONCE(lrugen->timestamps[next], jiffies);
3810	/* make sure preceding modifications appear */
3811	smp_store_release(&lrugen->max_seq, lrugen->max_seq + 1);
3812unlock:
3813	spin_unlock_irq(&lruvec->lru_lock);
3814
3815	return success;
3816}
3817
3818static bool try_to_inc_max_seq(struct lruvec *lruvec, unsigned long max_seq,
3819			       struct scan_control *sc, bool can_swap, bool force_scan)
3820{
3821	bool success;
3822	struct lru_gen_mm_walk *walk;
3823	struct mm_struct *mm = NULL;
3824	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3825	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
3826
3827	VM_WARN_ON_ONCE(max_seq > READ_ONCE(lrugen->max_seq));
3828
3829	if (!mm_state)
3830		return inc_max_seq(lruvec, max_seq, can_swap, force_scan);
3831
3832	/* see the comment in iterate_mm_list() */
3833	if (max_seq <= READ_ONCE(mm_state->seq))
3834		return false;
3835
3836	/*
3837	 * If the hardware doesn't automatically set the accessed bit, fallback
3838	 * to lru_gen_look_around(), which only clears the accessed bit in a
3839	 * handful of PTEs. Spreading the work out over a period of time usually
3840	 * is less efficient, but it avoids bursty page faults.
3841	 */
3842	if (!should_walk_mmu()) {
3843		success = iterate_mm_list_nowalk(lruvec, max_seq);
3844		goto done;
3845	}
3846
3847	walk = set_mm_walk(NULL, true);
3848	if (!walk) {
3849		success = iterate_mm_list_nowalk(lruvec, max_seq);
3850		goto done;
3851	}
3852
3853	walk->lruvec = lruvec;
3854	walk->max_seq = max_seq;
3855	walk->can_swap = can_swap;
3856	walk->force_scan = force_scan;
3857
3858	do {
3859		success = iterate_mm_list(lruvec, walk, &mm);
3860		if (mm)
3861			walk_mm(lruvec, mm, walk);
3862	} while (mm);
3863done:
3864	if (success) {
3865		success = inc_max_seq(lruvec, max_seq, can_swap, force_scan);
3866		WARN_ON_ONCE(!success);
3867	}
3868
3869	return success;
3870}
3871
3872/******************************************************************************
3873 *                          working set protection
3874 ******************************************************************************/
3875
3876static bool lruvec_is_sizable(struct lruvec *lruvec, struct scan_control *sc)
3877{
3878	int gen, type, zone;
3879	unsigned long total = 0;
3880	bool can_swap = get_swappiness(lruvec, sc);
3881	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3882	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
3883	DEFINE_MAX_SEQ(lruvec);
3884	DEFINE_MIN_SEQ(lruvec);
3885
3886	for (type = !can_swap; type < ANON_AND_FILE; type++) {
3887		unsigned long seq;
3888
3889		for (seq = min_seq[type]; seq <= max_seq; seq++) {
3890			gen = lru_gen_from_seq(seq);
3891
3892			for (zone = 0; zone < MAX_NR_ZONES; zone++)
3893				total += max(READ_ONCE(lrugen->nr_pages[gen][type][zone]), 0L);
3894		}
3895	}
3896
3897	/* whether the size is big enough to be helpful */
3898	return mem_cgroup_online(memcg) ? (total >> sc->priority) : total;
3899}
3900
3901static bool lruvec_is_reclaimable(struct lruvec *lruvec, struct scan_control *sc,
3902				  unsigned long min_ttl)
3903{
3904	int gen;
3905	unsigned long birth;
3906	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
3907	DEFINE_MIN_SEQ(lruvec);
3908
3909	/* see the comment on lru_gen_folio */
3910	gen = lru_gen_from_seq(min_seq[LRU_GEN_FILE]);
3911	birth = READ_ONCE(lruvec->lrugen.timestamps[gen]);
3912
3913	if (time_is_after_jiffies(birth + min_ttl))
3914		return false;
3915
3916	if (!lruvec_is_sizable(lruvec, sc))
3917		return false;
3918
3919	mem_cgroup_calculate_protection(NULL, memcg);
3920
3921	return !mem_cgroup_below_min(NULL, memcg);
3922}
3923
3924/* to protect the working set of the last N jiffies */
3925static unsigned long lru_gen_min_ttl __read_mostly;
3926
3927static void lru_gen_age_node(struct pglist_data *pgdat, struct scan_control *sc)
3928{
3929	struct mem_cgroup *memcg;
3930	unsigned long min_ttl = READ_ONCE(lru_gen_min_ttl);
3931
3932	VM_WARN_ON_ONCE(!current_is_kswapd());
3933
3934	/* check the order to exclude compaction-induced reclaim */
3935	if (!min_ttl || sc->order || sc->priority == DEF_PRIORITY)
3936		return;
3937
3938	memcg = mem_cgroup_iter(NULL, NULL, NULL);
3939	do {
3940		struct lruvec *lruvec = mem_cgroup_lruvec(memcg, pgdat);
3941
3942		if (lruvec_is_reclaimable(lruvec, sc, min_ttl)) {
3943			mem_cgroup_iter_break(NULL, memcg);
3944			return;
3945		}
3946
3947		cond_resched();
3948	} while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)));
3949
3950	/*
3951	 * The main goal is to OOM kill if every generation from all memcgs is
3952	 * younger than min_ttl. However, another possibility is all memcgs are
3953	 * either too small or below min.
3954	 */
3955	if (mutex_trylock(&oom_lock)) {
3956		struct oom_control oc = {
3957			.gfp_mask = sc->gfp_mask,
3958		};
3959
3960		out_of_memory(&oc);
3961
3962		mutex_unlock(&oom_lock);
3963	}
3964}
3965
3966/******************************************************************************
3967 *                          rmap/PT walk feedback
3968 ******************************************************************************/
3969
3970/*
3971 * This function exploits spatial locality when shrink_folio_list() walks the
3972 * rmap. It scans the adjacent PTEs of a young PTE and promotes hot pages. If
3973 * the scan was done cacheline efficiently, it adds the PMD entry pointing to
3974 * the PTE table to the Bloom filter. This forms a feedback loop between the
3975 * eviction and the aging.
3976 */
3977void lru_gen_look_around(struct page_vma_mapped_walk *pvmw)
3978{
3979	int i;
3980	unsigned long start;
3981	unsigned long end;
3982	struct lru_gen_mm_walk *walk;
3983	int young = 0;
3984	pte_t *pte = pvmw->pte;
3985	unsigned long addr = pvmw->address;
3986	struct vm_area_struct *vma = pvmw->vma;
3987	struct folio *folio = pfn_folio(pvmw->pfn);
3988	bool can_swap = !folio_is_file_lru(folio);
3989	struct mem_cgroup *memcg = folio_memcg(folio);
3990	struct pglist_data *pgdat = folio_pgdat(folio);
3991	struct lruvec *lruvec = mem_cgroup_lruvec(memcg, pgdat);
3992	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
3993	DEFINE_MAX_SEQ(lruvec);
3994	int old_gen, new_gen = lru_gen_from_seq(max_seq);
3995
3996	lockdep_assert_held(pvmw->ptl);
3997	VM_WARN_ON_ONCE_FOLIO(folio_test_lru(folio), folio);
3998
3999	if (spin_is_contended(pvmw->ptl))
4000		return;
4001
4002	/* exclude special VMAs containing anon pages from COW */
4003	if (vma->vm_flags & VM_SPECIAL)
4004		return;
4005
4006	/* avoid taking the LRU lock under the PTL when possible */
4007	walk = current->reclaim_state ? current->reclaim_state->mm_walk : NULL;
4008
4009	start = max(addr & PMD_MASK, vma->vm_start);
4010	end = min(addr | ~PMD_MASK, vma->vm_end - 1) + 1;
4011
4012	if (end - start > MIN_LRU_BATCH * PAGE_SIZE) {
4013		if (addr - start < MIN_LRU_BATCH * PAGE_SIZE / 2)
4014			end = start + MIN_LRU_BATCH * PAGE_SIZE;
4015		else if (end - addr < MIN_LRU_BATCH * PAGE_SIZE / 2)
4016			start = end - MIN_LRU_BATCH * PAGE_SIZE;
4017		else {
4018			start = addr - MIN_LRU_BATCH * PAGE_SIZE / 2;
4019			end = addr + MIN_LRU_BATCH * PAGE_SIZE / 2;
4020		}
4021	}
4022
4023	/* folio_update_gen() requires stable folio_memcg() */
4024	if (!mem_cgroup_trylock_pages(memcg))
4025		return;
4026
4027	arch_enter_lazy_mmu_mode();
4028
4029	pte -= (addr - start) / PAGE_SIZE;
4030
4031	for (i = 0, addr = start; addr != end; i++, addr += PAGE_SIZE) {
4032		unsigned long pfn;
4033		pte_t ptent = ptep_get(pte + i);
4034
4035		pfn = get_pte_pfn(ptent, vma, addr);
4036		if (pfn == -1)
4037			continue;
4038
4039		if (!pte_young(ptent))
4040			continue;
4041
4042		folio = get_pfn_folio(pfn, memcg, pgdat, can_swap);
4043		if (!folio)
4044			continue;
4045
4046		if (!ptep_test_and_clear_young(vma, addr, pte + i))
4047			VM_WARN_ON_ONCE(true);
4048
4049		young++;
4050
4051		if (pte_dirty(ptent) && !folio_test_dirty(folio) &&
4052		    !(folio_test_anon(folio) && folio_test_swapbacked(folio) &&
4053		      !folio_test_swapcache(folio)))
4054			folio_mark_dirty(folio);
4055
4056		if (walk) {
4057			old_gen = folio_update_gen(folio, new_gen);
4058			if (old_gen >= 0 && old_gen != new_gen)
4059				update_batch_size(walk, folio, old_gen, new_gen);
4060
4061			continue;
4062		}
4063
4064		old_gen = folio_lru_gen(folio);
4065		if (old_gen < 0)
4066			folio_set_referenced(folio);
4067		else if (old_gen != new_gen)
4068			folio_activate(folio);
4069	}
4070
4071	arch_leave_lazy_mmu_mode();
4072	mem_cgroup_unlock_pages();
4073
4074	/* feedback from rmap walkers to page table walkers */
4075	if (mm_state && suitable_to_scan(i, young))
4076		update_bloom_filter(mm_state, max_seq, pvmw->pmd);
4077}
4078
4079/******************************************************************************
4080 *                          memcg LRU
4081 ******************************************************************************/
4082
4083/* see the comment on MEMCG_NR_GENS */
4084enum {
4085	MEMCG_LRU_NOP,
4086	MEMCG_LRU_HEAD,
4087	MEMCG_LRU_TAIL,
4088	MEMCG_LRU_OLD,
4089	MEMCG_LRU_YOUNG,
4090};
4091
4092static void lru_gen_rotate_memcg(struct lruvec *lruvec, int op)
4093{
4094	int seg;
4095	int old, new;
4096	unsigned long flags;
4097	int bin = get_random_u32_below(MEMCG_NR_BINS);
4098	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
4099
4100	spin_lock_irqsave(&pgdat->memcg_lru.lock, flags);
4101
4102	VM_WARN_ON_ONCE(hlist_nulls_unhashed(&lruvec->lrugen.list));
4103
4104	seg = 0;
4105	new = old = lruvec->lrugen.gen;
4106
4107	/* see the comment on MEMCG_NR_GENS */
4108	if (op == MEMCG_LRU_HEAD)
4109		seg = MEMCG_LRU_HEAD;
4110	else if (op == MEMCG_LRU_TAIL)
4111		seg = MEMCG_LRU_TAIL;
4112	else if (op == MEMCG_LRU_OLD)
4113		new = get_memcg_gen(pgdat->memcg_lru.seq);
4114	else if (op == MEMCG_LRU_YOUNG)
4115		new = get_memcg_gen(pgdat->memcg_lru.seq + 1);
4116	else
4117		VM_WARN_ON_ONCE(true);
4118
4119	WRITE_ONCE(lruvec->lrugen.seg, seg);
4120	WRITE_ONCE(lruvec->lrugen.gen, new);
4121
4122	hlist_nulls_del_rcu(&lruvec->lrugen.list);
4123
4124	if (op == MEMCG_LRU_HEAD || op == MEMCG_LRU_OLD)
4125		hlist_nulls_add_head_rcu(&lruvec->lrugen.list, &pgdat->memcg_lru.fifo[new][bin]);
4126	else
4127		hlist_nulls_add_tail_rcu(&lruvec->lrugen.list, &pgdat->memcg_lru.fifo[new][bin]);
4128
4129	pgdat->memcg_lru.nr_memcgs[old]--;
4130	pgdat->memcg_lru.nr_memcgs[new]++;
4131
4132	if (!pgdat->memcg_lru.nr_memcgs[old] && old == get_memcg_gen(pgdat->memcg_lru.seq))
4133		WRITE_ONCE(pgdat->memcg_lru.seq, pgdat->memcg_lru.seq + 1);
4134
4135	spin_unlock_irqrestore(&pgdat->memcg_lru.lock, flags);
4136}
4137
4138#ifdef CONFIG_MEMCG
4139
4140void lru_gen_online_memcg(struct mem_cgroup *memcg)
4141{
4142	int gen;
4143	int nid;
4144	int bin = get_random_u32_below(MEMCG_NR_BINS);
4145
4146	for_each_node(nid) {
4147		struct pglist_data *pgdat = NODE_DATA(nid);
4148		struct lruvec *lruvec = get_lruvec(memcg, nid);
4149
4150		spin_lock_irq(&pgdat->memcg_lru.lock);
4151
4152		VM_WARN_ON_ONCE(!hlist_nulls_unhashed(&lruvec->lrugen.list));
4153
4154		gen = get_memcg_gen(pgdat->memcg_lru.seq);
4155
4156		lruvec->lrugen.gen = gen;
4157
4158		hlist_nulls_add_tail_rcu(&lruvec->lrugen.list, &pgdat->memcg_lru.fifo[gen][bin]);
4159		pgdat->memcg_lru.nr_memcgs[gen]++;
4160
4161		spin_unlock_irq(&pgdat->memcg_lru.lock);
4162	}
4163}
4164
4165void lru_gen_offline_memcg(struct mem_cgroup *memcg)
4166{
4167	int nid;
4168
4169	for_each_node(nid) {
4170		struct lruvec *lruvec = get_lruvec(memcg, nid);
4171
4172		lru_gen_rotate_memcg(lruvec, MEMCG_LRU_OLD);
4173	}
4174}
4175
4176void lru_gen_release_memcg(struct mem_cgroup *memcg)
4177{
4178	int gen;
4179	int nid;
4180
4181	for_each_node(nid) {
4182		struct pglist_data *pgdat = NODE_DATA(nid);
4183		struct lruvec *lruvec = get_lruvec(memcg, nid);
4184
4185		spin_lock_irq(&pgdat->memcg_lru.lock);
4186
4187		if (hlist_nulls_unhashed(&lruvec->lrugen.list))
4188			goto unlock;
4189
4190		gen = lruvec->lrugen.gen;
4191
4192		hlist_nulls_del_init_rcu(&lruvec->lrugen.list);
4193		pgdat->memcg_lru.nr_memcgs[gen]--;
4194
4195		if (!pgdat->memcg_lru.nr_memcgs[gen] && gen == get_memcg_gen(pgdat->memcg_lru.seq))
4196			WRITE_ONCE(pgdat->memcg_lru.seq, pgdat->memcg_lru.seq + 1);
4197unlock:
4198		spin_unlock_irq(&pgdat->memcg_lru.lock);
4199	}
4200}
4201
4202void lru_gen_soft_reclaim(struct mem_cgroup *memcg, int nid)
4203{
4204	struct lruvec *lruvec = get_lruvec(memcg, nid);
4205
4206	/* see the comment on MEMCG_NR_GENS */
4207	if (READ_ONCE(lruvec->lrugen.seg) != MEMCG_LRU_HEAD)
4208		lru_gen_rotate_memcg(lruvec, MEMCG_LRU_HEAD);
4209}
4210
4211#endif /* CONFIG_MEMCG */
4212
4213/******************************************************************************
4214 *                          the eviction
4215 ******************************************************************************/
4216
4217static bool sort_folio(struct lruvec *lruvec, struct folio *folio, struct scan_control *sc,
4218		       int tier_idx)
4219{
4220	bool success;
4221	int gen = folio_lru_gen(folio);
4222	int type = folio_is_file_lru(folio);
4223	int zone = folio_zonenum(folio);
4224	int delta = folio_nr_pages(folio);
4225	int refs = folio_lru_refs(folio);
4226	int tier = lru_tier_from_refs(refs);
4227	struct lru_gen_folio *lrugen = &lruvec->lrugen;
4228
4229	VM_WARN_ON_ONCE_FOLIO(gen >= MAX_NR_GENS, folio);
4230
4231	/* unevictable */
4232	if (!folio_evictable(folio)) {
4233		success = lru_gen_del_folio(lruvec, folio, true);
4234		VM_WARN_ON_ONCE_FOLIO(!success, folio);
4235		folio_set_unevictable(folio);
4236		lruvec_add_folio(lruvec, folio);
4237		__count_vm_events(UNEVICTABLE_PGCULLED, delta);
4238		return true;
4239	}
4240
4241	/* dirty lazyfree */
4242	if (type == LRU_GEN_FILE && folio_test_anon(folio) && folio_test_dirty(folio)) {
4243		success = lru_gen_del_folio(lruvec, folio, true);
4244		VM_WARN_ON_ONCE_FOLIO(!success, folio);
4245		folio_set_swapbacked(folio);
4246		lruvec_add_folio_tail(lruvec, folio);
4247		return true;
4248	}
4249
4250	/* promoted */
4251	if (gen != lru_gen_from_seq(lrugen->min_seq[type])) {
4252		list_move(&folio->lru, &lrugen->folios[gen][type][zone]);
4253		return true;
4254	}
4255
4256	/* protected */
4257	if (tier > tier_idx || refs == BIT(LRU_REFS_WIDTH)) {
4258		int hist = lru_hist_from_seq(lrugen->min_seq[type]);
4259
4260		gen = folio_inc_gen(lruvec, folio, false);
4261		list_move_tail(&folio->lru, &lrugen->folios[gen][type][zone]);
4262
4263		WRITE_ONCE(lrugen->protected[hist][type][tier - 1],
4264			   lrugen->protected[hist][type][tier - 1] + delta);
4265		return true;
4266	}
4267
4268	/* ineligible */
4269	if (zone > sc->reclaim_idx || skip_cma(folio, sc)) {
4270		gen = folio_inc_gen(lruvec, folio, false);
4271		list_move_tail(&folio->lru, &lrugen->folios[gen][type][zone]);
4272		return true;
4273	}
4274
4275	/* waiting for writeback */
4276	if (folio_test_locked(folio) || folio_test_writeback(folio) ||
4277	    (type == LRU_GEN_FILE && folio_test_dirty(folio))) {
4278		gen = folio_inc_gen(lruvec, folio, true);
4279		list_move(&folio->lru, &lrugen->folios[gen][type][zone]);
4280		return true;
4281	}
4282
4283	return false;
4284}
4285
4286static bool isolate_folio(struct lruvec *lruvec, struct folio *folio, struct scan_control *sc)
4287{
4288	bool success;
4289
4290	/* swapping inhibited */
4291	if (!(sc->gfp_mask & __GFP_IO) &&
4292	    (folio_test_dirty(folio) ||
4293	     (folio_test_anon(folio) && !folio_test_swapcache(folio))))
4294		return false;
4295
4296	/* raced with release_pages() */
4297	if (!folio_try_get(folio))
4298		return false;
4299
4300	/* raced with another isolation */
4301	if (!folio_test_clear_lru(folio)) {
4302		folio_put(folio);
4303		return false;
4304	}
4305
4306	/* see the comment on MAX_NR_TIERS */
4307	if (!folio_test_referenced(folio))
4308		set_mask_bits(&folio->flags, LRU_REFS_MASK | LRU_REFS_FLAGS, 0);
4309
4310	/* for shrink_folio_list() */
4311	folio_clear_reclaim(folio);
4312	folio_clear_referenced(folio);
4313
4314	success = lru_gen_del_folio(lruvec, folio, true);
4315	VM_WARN_ON_ONCE_FOLIO(!success, folio);
4316
4317	return true;
4318}
4319
4320static int scan_folios(struct lruvec *lruvec, struct scan_control *sc,
4321		       int type, int tier, struct list_head *list)
4322{
4323	int i;
4324	int gen;
4325	enum vm_event_item item;
4326	int sorted = 0;
4327	int scanned = 0;
4328	int isolated = 0;
4329	int skipped = 0;
4330	int remaining = MAX_LRU_BATCH;
4331	struct lru_gen_folio *lrugen = &lruvec->lrugen;
4332	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
4333
4334	VM_WARN_ON_ONCE(!list_empty(list));
4335
4336	if (get_nr_gens(lruvec, type) == MIN_NR_GENS)
4337		return 0;
4338
4339	gen = lru_gen_from_seq(lrugen->min_seq[type]);
4340
4341	for (i = MAX_NR_ZONES; i > 0; i--) {
4342		LIST_HEAD(moved);
4343		int skipped_zone = 0;
4344		int zone = (sc->reclaim_idx + i) % MAX_NR_ZONES;
4345		struct list_head *head = &lrugen->folios[gen][type][zone];
4346
4347		while (!list_empty(head)) {
4348			struct folio *folio = lru_to_folio(head);
4349			int delta = folio_nr_pages(folio);
4350
4351			VM_WARN_ON_ONCE_FOLIO(folio_test_unevictable(folio), folio);
4352			VM_WARN_ON_ONCE_FOLIO(folio_test_active(folio), folio);
4353			VM_WARN_ON_ONCE_FOLIO(folio_is_file_lru(folio) != type, folio);
4354			VM_WARN_ON_ONCE_FOLIO(folio_zonenum(folio) != zone, folio);
4355
4356			scanned += delta;
4357
4358			if (sort_folio(lruvec, folio, sc, tier))
4359				sorted += delta;
4360			else if (isolate_folio(lruvec, folio, sc)) {
4361				list_add(&folio->lru, list);
4362				isolated += delta;
4363			} else {
4364				list_move(&folio->lru, &moved);
4365				skipped_zone += delta;
4366			}
4367
4368			if (!--remaining || max(isolated, skipped_zone) >= MIN_LRU_BATCH)
4369				break;
4370		}
4371
4372		if (skipped_zone) {
4373			list_splice(&moved, head);
4374			__count_zid_vm_events(PGSCAN_SKIP, zone, skipped_zone);
4375			skipped += skipped_zone;
4376		}
4377
4378		if (!remaining || isolated >= MIN_LRU_BATCH)
4379			break;
4380	}
4381
4382	item = PGSCAN_KSWAPD + reclaimer_offset();
4383	if (!cgroup_reclaim(sc)) {
4384		__count_vm_events(item, isolated);
4385		__count_vm_events(PGREFILL, sorted);
4386	}
4387	__count_memcg_events(memcg, item, isolated);
4388	__count_memcg_events(memcg, PGREFILL, sorted);
4389	__count_vm_events(PGSCAN_ANON + type, isolated);
4390	trace_mm_vmscan_lru_isolate(sc->reclaim_idx, sc->order, MAX_LRU_BATCH,
4391				scanned, skipped, isolated,
4392				type ? LRU_INACTIVE_FILE : LRU_INACTIVE_ANON);
4393
4394	/*
4395	 * There might not be eligible folios due to reclaim_idx. Check the
4396	 * remaining to prevent livelock if it's not making progress.
4397	 */
4398	return isolated || !remaining ? scanned : 0;
4399}
4400
4401static int get_tier_idx(struct lruvec *lruvec, int type)
4402{
4403	int tier;
4404	struct ctrl_pos sp, pv;
4405
4406	/*
4407	 * To leave a margin for fluctuations, use a larger gain factor (1:2).
4408	 * This value is chosen because any other tier would have at least twice
4409	 * as many refaults as the first tier.
4410	 */
4411	read_ctrl_pos(lruvec, type, 0, 1, &sp);
4412	for (tier = 1; tier < MAX_NR_TIERS; tier++) {
4413		read_ctrl_pos(lruvec, type, tier, 2, &pv);
4414		if (!positive_ctrl_err(&sp, &pv))
4415			break;
4416	}
4417
4418	return tier - 1;
4419}
4420
4421static int get_type_to_scan(struct lruvec *lruvec, int swappiness, int *tier_idx)
4422{
4423	int type, tier;
4424	struct ctrl_pos sp, pv;
4425	int gain[ANON_AND_FILE] = { swappiness, 200 - swappiness };
4426
4427	/*
4428	 * Compare the first tier of anon with that of file to determine which
4429	 * type to scan. Also need to compare other tiers of the selected type
4430	 * with the first tier of the other type to determine the last tier (of
4431	 * the selected type) to evict.
4432	 */
4433	read_ctrl_pos(lruvec, LRU_GEN_ANON, 0, gain[LRU_GEN_ANON], &sp);
4434	read_ctrl_pos(lruvec, LRU_GEN_FILE, 0, gain[LRU_GEN_FILE], &pv);
4435	type = positive_ctrl_err(&sp, &pv);
4436
4437	read_ctrl_pos(lruvec, !type, 0, gain[!type], &sp);
4438	for (tier = 1; tier < MAX_NR_TIERS; tier++) {
4439		read_ctrl_pos(lruvec, type, tier, gain[type], &pv);
4440		if (!positive_ctrl_err(&sp, &pv))
4441			break;
4442	}
4443
4444	*tier_idx = tier - 1;
4445
4446	return type;
4447}
4448
4449static int isolate_folios(struct lruvec *lruvec, struct scan_control *sc, int swappiness,
4450			  int *type_scanned, struct list_head *list)
4451{
4452	int i;
4453	int type;
4454	int scanned;
4455	int tier = -1;
4456	DEFINE_MIN_SEQ(lruvec);
4457
4458	/*
4459	 * Try to make the obvious choice first. When anon and file are both
4460	 * available from the same generation, interpret swappiness 1 as file
4461	 * first and 200 as anon first.
 
 
 
4462	 */
4463	if (!swappiness)
4464		type = LRU_GEN_FILE;
4465	else if (min_seq[LRU_GEN_ANON] < min_seq[LRU_GEN_FILE])
4466		type = LRU_GEN_ANON;
4467	else if (swappiness == 1)
4468		type = LRU_GEN_FILE;
4469	else if (swappiness == 200)
4470		type = LRU_GEN_ANON;
 
 
4471	else
4472		type = get_type_to_scan(lruvec, swappiness, &tier);
4473
4474	for (i = !swappiness; i < ANON_AND_FILE; i++) {
4475		if (tier < 0)
4476			tier = get_tier_idx(lruvec, type);
4477
4478		scanned = scan_folios(lruvec, sc, type, tier, list);
4479		if (scanned)
4480			break;
4481
4482		type = !type;
4483		tier = -1;
4484	}
4485
4486	*type_scanned = type;
4487
4488	return scanned;
4489}
4490
4491static int evict_folios(struct lruvec *lruvec, struct scan_control *sc, int swappiness)
4492{
4493	int type;
4494	int scanned;
4495	int reclaimed;
4496	LIST_HEAD(list);
4497	LIST_HEAD(clean);
4498	struct folio *folio;
4499	struct folio *next;
4500	enum vm_event_item item;
4501	struct reclaim_stat stat;
4502	struct lru_gen_mm_walk *walk;
4503	bool skip_retry = false;
4504	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
4505	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
4506
4507	spin_lock_irq(&lruvec->lru_lock);
4508
4509	scanned = isolate_folios(lruvec, sc, swappiness, &type, &list);
4510
4511	scanned += try_to_inc_min_seq(lruvec, swappiness);
4512
4513	if (get_nr_gens(lruvec, !swappiness) == MIN_NR_GENS)
4514		scanned = 0;
4515
4516	spin_unlock_irq(&lruvec->lru_lock);
4517
4518	if (list_empty(&list))
4519		return scanned;
4520retry:
4521	reclaimed = shrink_folio_list(&list, pgdat, sc, &stat, false);
4522	sc->nr_reclaimed += reclaimed;
4523	trace_mm_vmscan_lru_shrink_inactive(pgdat->node_id,
4524			scanned, reclaimed, &stat, sc->priority,
4525			type ? LRU_INACTIVE_FILE : LRU_INACTIVE_ANON);
4526
4527	list_for_each_entry_safe_reverse(folio, next, &list, lru) {
4528		if (!folio_evictable(folio)) {
4529			list_del(&folio->lru);
4530			folio_putback_lru(folio);
4531			continue;
4532		}
4533
4534		if (folio_test_reclaim(folio) &&
4535		    (folio_test_dirty(folio) || folio_test_writeback(folio))) {
4536			/* restore LRU_REFS_FLAGS cleared by isolate_folio() */
4537			if (folio_test_workingset(folio))
4538				folio_set_referenced(folio);
4539			continue;
4540		}
4541
4542		if (skip_retry || folio_test_active(folio) || folio_test_referenced(folio) ||
4543		    folio_mapped(folio) || folio_test_locked(folio) ||
4544		    folio_test_dirty(folio) || folio_test_writeback(folio)) {
4545			/* don't add rejected folios to the oldest generation */
4546			set_mask_bits(&folio->flags, LRU_REFS_MASK | LRU_REFS_FLAGS,
4547				      BIT(PG_active));
4548			continue;
4549		}
4550
4551		/* retry folios that may have missed folio_rotate_reclaimable() */
4552		list_move(&folio->lru, &clean);
4553		sc->nr_scanned -= folio_nr_pages(folio);
4554	}
4555
4556	spin_lock_irq(&lruvec->lru_lock);
4557
4558	move_folios_to_lru(lruvec, &list);
4559
4560	walk = current->reclaim_state->mm_walk;
4561	if (walk && walk->batched)
4562		reset_batch_size(lruvec, walk);
 
 
4563
4564	item = PGSTEAL_KSWAPD + reclaimer_offset();
4565	if (!cgroup_reclaim(sc))
4566		__count_vm_events(item, reclaimed);
4567	__count_memcg_events(memcg, item, reclaimed);
4568	__count_vm_events(PGSTEAL_ANON + type, reclaimed);
4569
4570	spin_unlock_irq(&lruvec->lru_lock);
4571
4572	mem_cgroup_uncharge_list(&list);
4573	free_unref_page_list(&list);
4574
4575	INIT_LIST_HEAD(&list);
4576	list_splice_init(&clean, &list);
4577
4578	if (!list_empty(&list)) {
4579		skip_retry = true;
4580		goto retry;
4581	}
4582
4583	return scanned;
4584}
4585
4586static bool should_run_aging(struct lruvec *lruvec, unsigned long max_seq,
4587			     struct scan_control *sc, bool can_swap, unsigned long *nr_to_scan)
4588{
4589	int gen, type, zone;
4590	unsigned long old = 0;
4591	unsigned long young = 0;
4592	unsigned long total = 0;
4593	struct lru_gen_folio *lrugen = &lruvec->lrugen;
4594	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
4595	DEFINE_MIN_SEQ(lruvec);
4596
4597	/* whether this lruvec is completely out of cold folios */
4598	if (min_seq[!can_swap] + MIN_NR_GENS > max_seq) {
4599		*nr_to_scan = 0;
4600		return true;
4601	}
4602
4603	for (type = !can_swap; type < ANON_AND_FILE; type++) {
4604		unsigned long seq;
4605
4606		for (seq = min_seq[type]; seq <= max_seq; seq++) {
4607			unsigned long size = 0;
4608
4609			gen = lru_gen_from_seq(seq);
4610
4611			for (zone = 0; zone < MAX_NR_ZONES; zone++)
4612				size += max(READ_ONCE(lrugen->nr_pages[gen][type][zone]), 0L);
4613
4614			total += size;
4615			if (seq == max_seq)
4616				young += size;
4617			else if (seq + MIN_NR_GENS == max_seq)
4618				old += size;
4619		}
4620	}
4621
4622	/* try to scrape all its memory if this memcg was deleted */
4623	if (!mem_cgroup_online(memcg)) {
4624		*nr_to_scan = total;
4625		return false;
4626	}
4627
4628	*nr_to_scan = total >> sc->priority;
4629
4630	/*
4631	 * The aging tries to be lazy to reduce the overhead, while the eviction
4632	 * stalls when the number of generations reaches MIN_NR_GENS. Hence, the
4633	 * ideal number of generations is MIN_NR_GENS+1.
4634	 */
4635	if (min_seq[!can_swap] + MIN_NR_GENS < max_seq)
4636		return false;
4637
4638	/*
4639	 * It's also ideal to spread pages out evenly, i.e., 1/(MIN_NR_GENS+1)
4640	 * of the total number of pages for each generation. A reasonable range
4641	 * for this average portion is [1/MIN_NR_GENS, 1/(MIN_NR_GENS+2)]. The
4642	 * aging cares about the upper bound of hot pages, while the eviction
4643	 * cares about the lower bound of cold pages.
4644	 */
4645	if (young * MIN_NR_GENS > total)
4646		return true;
4647	if (old * (MIN_NR_GENS + 2) < total)
4648		return true;
4649
4650	return false;
4651}
4652
4653/*
4654 * For future optimizations:
4655 * 1. Defer try_to_inc_max_seq() to workqueues to reduce latency for memcg
4656 *    reclaim.
4657 */
4658static long get_nr_to_scan(struct lruvec *lruvec, struct scan_control *sc, bool can_swap)
4659{
 
4660	unsigned long nr_to_scan;
4661	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
4662	DEFINE_MAX_SEQ(lruvec);
4663
4664	if (mem_cgroup_below_min(sc->target_mem_cgroup, memcg))
4665		return -1;
4666
4667	if (!should_run_aging(lruvec, max_seq, sc, can_swap, &nr_to_scan))
4668		return nr_to_scan;
4669
4670	/* skip the aging path at the default priority */
4671	if (sc->priority == DEF_PRIORITY)
4672		return nr_to_scan;
4673
4674	/* skip this lruvec as it's low on cold folios */
4675	return try_to_inc_max_seq(lruvec, max_seq, sc, can_swap, false) ? -1 : 0;
 
 
 
 
4676}
4677
4678static bool should_abort_scan(struct lruvec *lruvec, struct scan_control *sc)
4679{
4680	int i;
4681	enum zone_watermarks mark;
4682
4683	/* don't abort memcg reclaim to ensure fairness */
4684	if (!root_reclaim(sc))
4685		return false;
4686
4687	if (sc->nr_reclaimed >= max(sc->nr_to_reclaim, compact_gap(sc->order)))
4688		return true;
4689
4690	/* check the order to exclude compaction-induced reclaim */
4691	if (!current_is_kswapd() || sc->order)
4692		return false;
4693
4694	mark = sysctl_numa_balancing_mode & NUMA_BALANCING_MEMORY_TIERING ?
4695	       WMARK_PROMO : WMARK_HIGH;
4696
4697	for (i = 0; i <= sc->reclaim_idx; i++) {
4698		struct zone *zone = lruvec_pgdat(lruvec)->node_zones + i;
4699		unsigned long size = wmark_pages(zone, mark) + MIN_LRU_BATCH;
4700
4701		if (managed_zone(zone) && !zone_watermark_ok(zone, 0, size, sc->reclaim_idx, 0))
4702			return false;
4703	}
4704
4705	/* kswapd should abort if all eligible zones are safe */
4706	return true;
4707}
4708
4709static bool try_to_shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
4710{
4711	long nr_to_scan;
4712	unsigned long scanned = 0;
4713	int swappiness = get_swappiness(lruvec, sc);
4714
4715	/* clean file folios are more likely to exist */
4716	if (swappiness && !(sc->gfp_mask & __GFP_IO))
4717		swappiness = 1;
4718
4719	while (true) {
4720		int delta;
4721
4722		nr_to_scan = get_nr_to_scan(lruvec, sc, swappiness);
4723		if (nr_to_scan <= 0)
4724			break;
4725
4726		delta = evict_folios(lruvec, sc, swappiness);
4727		if (!delta)
4728			break;
4729
4730		scanned += delta;
4731		if (scanned >= nr_to_scan)
4732			break;
4733
4734		if (should_abort_scan(lruvec, sc))
4735			break;
4736
4737		cond_resched();
4738	}
4739
4740	/* whether this lruvec should be rotated */
4741	return nr_to_scan < 0;
4742}
4743
4744static int shrink_one(struct lruvec *lruvec, struct scan_control *sc)
4745{
4746	bool success;
4747	unsigned long scanned = sc->nr_scanned;
4748	unsigned long reclaimed = sc->nr_reclaimed;
4749	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
4750	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
4751
4752	mem_cgroup_calculate_protection(NULL, memcg);
4753
4754	if (mem_cgroup_below_min(NULL, memcg))
4755		return MEMCG_LRU_YOUNG;
4756
4757	if (mem_cgroup_below_low(NULL, memcg)) {
4758		/* see the comment on MEMCG_NR_GENS */
4759		if (READ_ONCE(lruvec->lrugen.seg) != MEMCG_LRU_TAIL)
4760			return MEMCG_LRU_TAIL;
4761
4762		memcg_memory_event(memcg, MEMCG_LOW);
4763	}
4764
4765	success = try_to_shrink_lruvec(lruvec, sc);
4766
4767	shrink_slab(sc->gfp_mask, pgdat->node_id, memcg, sc->priority);
4768
4769	if (!sc->proactive)
4770		vmpressure(sc->gfp_mask, memcg, false, sc->nr_scanned - scanned,
4771			   sc->nr_reclaimed - reclaimed);
4772
4773	flush_reclaim_state(sc);
4774
4775	if (success && mem_cgroup_online(memcg))
4776		return MEMCG_LRU_YOUNG;
4777
4778	if (!success && lruvec_is_sizable(lruvec, sc))
4779		return 0;
4780
4781	/* one retry if offlined or too small */
4782	return READ_ONCE(lruvec->lrugen.seg) != MEMCG_LRU_TAIL ?
4783	       MEMCG_LRU_TAIL : MEMCG_LRU_YOUNG;
4784}
4785
4786static void shrink_many(struct pglist_data *pgdat, struct scan_control *sc)
4787{
4788	int op;
4789	int gen;
4790	int bin;
4791	int first_bin;
4792	struct lruvec *lruvec;
4793	struct lru_gen_folio *lrugen;
4794	struct mem_cgroup *memcg;
4795	struct hlist_nulls_node *pos;
4796
4797	gen = get_memcg_gen(READ_ONCE(pgdat->memcg_lru.seq));
4798	bin = first_bin = get_random_u32_below(MEMCG_NR_BINS);
4799restart:
4800	op = 0;
4801	memcg = NULL;
4802
4803	rcu_read_lock();
4804
4805	hlist_nulls_for_each_entry_rcu(lrugen, pos, &pgdat->memcg_lru.fifo[gen][bin], list) {
4806		if (op) {
4807			lru_gen_rotate_memcg(lruvec, op);
4808			op = 0;
4809		}
4810
4811		mem_cgroup_put(memcg);
4812		memcg = NULL;
4813
4814		if (gen != READ_ONCE(lrugen->gen))
4815			continue;
4816
4817		lruvec = container_of(lrugen, struct lruvec, lrugen);
4818		memcg = lruvec_memcg(lruvec);
4819
4820		if (!mem_cgroup_tryget(memcg)) {
4821			lru_gen_release_memcg(memcg);
4822			memcg = NULL;
4823			continue;
4824		}
4825
4826		rcu_read_unlock();
4827
4828		op = shrink_one(lruvec, sc);
4829
4830		rcu_read_lock();
4831
4832		if (should_abort_scan(lruvec, sc))
4833			break;
4834	}
4835
4836	rcu_read_unlock();
4837
4838	if (op)
4839		lru_gen_rotate_memcg(lruvec, op);
4840
4841	mem_cgroup_put(memcg);
4842
4843	if (!is_a_nulls(pos))
4844		return;
4845
4846	/* restart if raced with lru_gen_rotate_memcg() */
4847	if (gen != get_nulls_value(pos))
4848		goto restart;
4849
4850	/* try the rest of the bins of the current generation */
4851	bin = get_memcg_bin(bin + 1);
4852	if (bin != first_bin)
4853		goto restart;
4854}
4855
4856static void lru_gen_shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
4857{
4858	struct blk_plug plug;
4859
4860	VM_WARN_ON_ONCE(root_reclaim(sc));
4861	VM_WARN_ON_ONCE(!sc->may_writepage || !sc->may_unmap);
4862
4863	lru_add_drain();
4864
4865	blk_start_plug(&plug);
4866
4867	set_mm_walk(NULL, sc->proactive);
4868
4869	if (try_to_shrink_lruvec(lruvec, sc))
4870		lru_gen_rotate_memcg(lruvec, MEMCG_LRU_YOUNG);
4871
4872	clear_mm_walk();
4873
4874	blk_finish_plug(&plug);
4875}
4876
4877static void set_initial_priority(struct pglist_data *pgdat, struct scan_control *sc)
4878{
4879	int priority;
4880	unsigned long reclaimable;
4881	struct lruvec *lruvec = mem_cgroup_lruvec(NULL, pgdat);
4882
4883	if (sc->priority != DEF_PRIORITY || sc->nr_to_reclaim < MIN_LRU_BATCH)
4884		return;
4885	/*
4886	 * Determine the initial priority based on
4887	 * (total >> priority) * reclaimed_to_scanned_ratio = nr_to_reclaim,
4888	 * where reclaimed_to_scanned_ratio = inactive / total.
4889	 */
4890	reclaimable = node_page_state(pgdat, NR_INACTIVE_FILE);
4891	if (get_swappiness(lruvec, sc))
4892		reclaimable += node_page_state(pgdat, NR_INACTIVE_ANON);
4893
4894	/* round down reclaimable and round up sc->nr_to_reclaim */
4895	priority = fls_long(reclaimable) - 1 - fls_long(sc->nr_to_reclaim - 1);
4896
4897	sc->priority = clamp(priority, 0, DEF_PRIORITY);
4898}
4899
4900static void lru_gen_shrink_node(struct pglist_data *pgdat, struct scan_control *sc)
4901{
4902	struct blk_plug plug;
4903	unsigned long reclaimed = sc->nr_reclaimed;
4904
4905	VM_WARN_ON_ONCE(!root_reclaim(sc));
4906
4907	/*
4908	 * Unmapped clean folios are already prioritized. Scanning for more of
4909	 * them is likely futile and can cause high reclaim latency when there
4910	 * is a large number of memcgs.
4911	 */
4912	if (!sc->may_writepage || !sc->may_unmap)
4913		goto done;
4914
4915	lru_add_drain();
4916
4917	blk_start_plug(&plug);
4918
4919	set_mm_walk(pgdat, sc->proactive);
4920
4921	set_initial_priority(pgdat, sc);
4922
4923	if (current_is_kswapd())
4924		sc->nr_reclaimed = 0;
4925
4926	if (mem_cgroup_disabled())
4927		shrink_one(&pgdat->__lruvec, sc);
4928	else
4929		shrink_many(pgdat, sc);
4930
4931	if (current_is_kswapd())
4932		sc->nr_reclaimed += reclaimed;
4933
4934	clear_mm_walk();
4935
4936	blk_finish_plug(&plug);
4937done:
4938	/* kswapd should never fail */
4939	pgdat->kswapd_failures = 0;
4940}
4941
4942/******************************************************************************
4943 *                          state change
4944 ******************************************************************************/
4945
4946static bool __maybe_unused state_is_valid(struct lruvec *lruvec)
4947{
4948	struct lru_gen_folio *lrugen = &lruvec->lrugen;
4949
4950	if (lrugen->enabled) {
4951		enum lru_list lru;
4952
4953		for_each_evictable_lru(lru) {
4954			if (!list_empty(&lruvec->lists[lru]))
4955				return false;
4956		}
4957	} else {
4958		int gen, type, zone;
4959
4960		for_each_gen_type_zone(gen, type, zone) {
4961			if (!list_empty(&lrugen->folios[gen][type][zone]))
4962				return false;
4963		}
4964	}
4965
4966	return true;
4967}
4968
4969static bool fill_evictable(struct lruvec *lruvec)
4970{
4971	enum lru_list lru;
4972	int remaining = MAX_LRU_BATCH;
4973
4974	for_each_evictable_lru(lru) {
4975		int type = is_file_lru(lru);
4976		bool active = is_active_lru(lru);
4977		struct list_head *head = &lruvec->lists[lru];
4978
4979		while (!list_empty(head)) {
4980			bool success;
4981			struct folio *folio = lru_to_folio(head);
4982
4983			VM_WARN_ON_ONCE_FOLIO(folio_test_unevictable(folio), folio);
4984			VM_WARN_ON_ONCE_FOLIO(folio_test_active(folio) != active, folio);
4985			VM_WARN_ON_ONCE_FOLIO(folio_is_file_lru(folio) != type, folio);
4986			VM_WARN_ON_ONCE_FOLIO(folio_lru_gen(folio) != -1, folio);
4987
4988			lruvec_del_folio(lruvec, folio);
4989			success = lru_gen_add_folio(lruvec, folio, false);
4990			VM_WARN_ON_ONCE(!success);
4991
4992			if (!--remaining)
4993				return false;
4994		}
4995	}
4996
4997	return true;
4998}
4999
5000static bool drain_evictable(struct lruvec *lruvec)
5001{
5002	int gen, type, zone;
5003	int remaining = MAX_LRU_BATCH;
5004
5005	for_each_gen_type_zone(gen, type, zone) {
5006		struct list_head *head = &lruvec->lrugen.folios[gen][type][zone];
5007
5008		while (!list_empty(head)) {
5009			bool success;
5010			struct folio *folio = lru_to_folio(head);
5011
5012			VM_WARN_ON_ONCE_FOLIO(folio_test_unevictable(folio), folio);
5013			VM_WARN_ON_ONCE_FOLIO(folio_test_active(folio), folio);
5014			VM_WARN_ON_ONCE_FOLIO(folio_is_file_lru(folio) != type, folio);
5015			VM_WARN_ON_ONCE_FOLIO(folio_zonenum(folio) != zone, folio);
5016
5017			success = lru_gen_del_folio(lruvec, folio, false);
5018			VM_WARN_ON_ONCE(!success);
5019			lruvec_add_folio(lruvec, folio);
5020
5021			if (!--remaining)
5022				return false;
5023		}
5024	}
5025
5026	return true;
5027}
5028
5029static void lru_gen_change_state(bool enabled)
5030{
5031	static DEFINE_MUTEX(state_mutex);
5032
5033	struct mem_cgroup *memcg;
5034
5035	cgroup_lock();
5036	cpus_read_lock();
5037	get_online_mems();
5038	mutex_lock(&state_mutex);
5039
5040	if (enabled == lru_gen_enabled())
5041		goto unlock;
5042
5043	if (enabled)
5044		static_branch_enable_cpuslocked(&lru_gen_caps[LRU_GEN_CORE]);
5045	else
5046		static_branch_disable_cpuslocked(&lru_gen_caps[LRU_GEN_CORE]);
5047
5048	memcg = mem_cgroup_iter(NULL, NULL, NULL);
5049	do {
5050		int nid;
5051
5052		for_each_node(nid) {
5053			struct lruvec *lruvec = get_lruvec(memcg, nid);
5054
5055			spin_lock_irq(&lruvec->lru_lock);
5056
5057			VM_WARN_ON_ONCE(!seq_is_valid(lruvec));
5058			VM_WARN_ON_ONCE(!state_is_valid(lruvec));
5059
5060			lruvec->lrugen.enabled = enabled;
5061
5062			while (!(enabled ? fill_evictable(lruvec) : drain_evictable(lruvec))) {
5063				spin_unlock_irq(&lruvec->lru_lock);
5064				cond_resched();
5065				spin_lock_irq(&lruvec->lru_lock);
5066			}
5067
5068			spin_unlock_irq(&lruvec->lru_lock);
5069		}
5070
5071		cond_resched();
5072	} while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)));
5073unlock:
5074	mutex_unlock(&state_mutex);
5075	put_online_mems();
5076	cpus_read_unlock();
5077	cgroup_unlock();
5078}
5079
5080/******************************************************************************
5081 *                          sysfs interface
5082 ******************************************************************************/
5083
5084static ssize_t min_ttl_ms_show(struct kobject *kobj, struct kobj_attribute *attr, char *buf)
5085{
5086	return sysfs_emit(buf, "%u\n", jiffies_to_msecs(READ_ONCE(lru_gen_min_ttl)));
5087}
5088
5089/* see Documentation/admin-guide/mm/multigen_lru.rst for details */
5090static ssize_t min_ttl_ms_store(struct kobject *kobj, struct kobj_attribute *attr,
5091				const char *buf, size_t len)
5092{
5093	unsigned int msecs;
5094
5095	if (kstrtouint(buf, 0, &msecs))
5096		return -EINVAL;
5097
5098	WRITE_ONCE(lru_gen_min_ttl, msecs_to_jiffies(msecs));
5099
5100	return len;
5101}
5102
5103static struct kobj_attribute lru_gen_min_ttl_attr = __ATTR_RW(min_ttl_ms);
5104
5105static ssize_t enabled_show(struct kobject *kobj, struct kobj_attribute *attr, char *buf)
5106{
5107	unsigned int caps = 0;
5108
5109	if (get_cap(LRU_GEN_CORE))
5110		caps |= BIT(LRU_GEN_CORE);
5111
5112	if (should_walk_mmu())
5113		caps |= BIT(LRU_GEN_MM_WALK);
5114
5115	if (should_clear_pmd_young())
5116		caps |= BIT(LRU_GEN_NONLEAF_YOUNG);
5117
5118	return sysfs_emit(buf, "0x%04x\n", caps);
5119}
5120
5121/* see Documentation/admin-guide/mm/multigen_lru.rst for details */
5122static ssize_t enabled_store(struct kobject *kobj, struct kobj_attribute *attr,
5123			     const char *buf, size_t len)
5124{
5125	int i;
5126	unsigned int caps;
5127
5128	if (tolower(*buf) == 'n')
5129		caps = 0;
5130	else if (tolower(*buf) == 'y')
5131		caps = -1;
5132	else if (kstrtouint(buf, 0, &caps))
5133		return -EINVAL;
5134
5135	for (i = 0; i < NR_LRU_GEN_CAPS; i++) {
5136		bool enabled = caps & BIT(i);
5137
5138		if (i == LRU_GEN_CORE)
5139			lru_gen_change_state(enabled);
5140		else if (enabled)
5141			static_branch_enable(&lru_gen_caps[i]);
5142		else
5143			static_branch_disable(&lru_gen_caps[i]);
5144	}
5145
5146	return len;
5147}
5148
5149static struct kobj_attribute lru_gen_enabled_attr = __ATTR_RW(enabled);
5150
5151static struct attribute *lru_gen_attrs[] = {
5152	&lru_gen_min_ttl_attr.attr,
5153	&lru_gen_enabled_attr.attr,
5154	NULL
5155};
5156
5157static const struct attribute_group lru_gen_attr_group = {
5158	.name = "lru_gen",
5159	.attrs = lru_gen_attrs,
5160};
5161
5162/******************************************************************************
5163 *                          debugfs interface
5164 ******************************************************************************/
5165
5166static void *lru_gen_seq_start(struct seq_file *m, loff_t *pos)
5167{
5168	struct mem_cgroup *memcg;
5169	loff_t nr_to_skip = *pos;
5170
5171	m->private = kvmalloc(PATH_MAX, GFP_KERNEL);
5172	if (!m->private)
5173		return ERR_PTR(-ENOMEM);
5174
5175	memcg = mem_cgroup_iter(NULL, NULL, NULL);
5176	do {
5177		int nid;
5178
5179		for_each_node_state(nid, N_MEMORY) {
5180			if (!nr_to_skip--)
5181				return get_lruvec(memcg, nid);
5182		}
5183	} while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)));
5184
5185	return NULL;
5186}
5187
5188static void lru_gen_seq_stop(struct seq_file *m, void *v)
5189{
5190	if (!IS_ERR_OR_NULL(v))
5191		mem_cgroup_iter_break(NULL, lruvec_memcg(v));
5192
5193	kvfree(m->private);
5194	m->private = NULL;
5195}
5196
5197static void *lru_gen_seq_next(struct seq_file *m, void *v, loff_t *pos)
5198{
5199	int nid = lruvec_pgdat(v)->node_id;
5200	struct mem_cgroup *memcg = lruvec_memcg(v);
5201
5202	++*pos;
5203
5204	nid = next_memory_node(nid);
5205	if (nid == MAX_NUMNODES) {
5206		memcg = mem_cgroup_iter(NULL, memcg, NULL);
5207		if (!memcg)
5208			return NULL;
5209
5210		nid = first_memory_node;
5211	}
5212
5213	return get_lruvec(memcg, nid);
5214}
5215
5216static void lru_gen_seq_show_full(struct seq_file *m, struct lruvec *lruvec,
5217				  unsigned long max_seq, unsigned long *min_seq,
5218				  unsigned long seq)
5219{
5220	int i;
5221	int type, tier;
5222	int hist = lru_hist_from_seq(seq);
5223	struct lru_gen_folio *lrugen = &lruvec->lrugen;
5224	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
5225
5226	for (tier = 0; tier < MAX_NR_TIERS; tier++) {
5227		seq_printf(m, "            %10d", tier);
5228		for (type = 0; type < ANON_AND_FILE; type++) {
5229			const char *s = "   ";
5230			unsigned long n[3] = {};
5231
5232			if (seq == max_seq) {
5233				s = "RT ";
5234				n[0] = READ_ONCE(lrugen->avg_refaulted[type][tier]);
5235				n[1] = READ_ONCE(lrugen->avg_total[type][tier]);
5236			} else if (seq == min_seq[type] || NR_HIST_GENS > 1) {
5237				s = "rep";
5238				n[0] = atomic_long_read(&lrugen->refaulted[hist][type][tier]);
5239				n[1] = atomic_long_read(&lrugen->evicted[hist][type][tier]);
5240				if (tier)
5241					n[2] = READ_ONCE(lrugen->protected[hist][type][tier - 1]);
5242			}
5243
5244			for (i = 0; i < 3; i++)
5245				seq_printf(m, " %10lu%c", n[i], s[i]);
5246		}
5247		seq_putc(m, '\n');
5248	}
5249
5250	if (!mm_state)
5251		return;
5252
5253	seq_puts(m, "                      ");
5254	for (i = 0; i < NR_MM_STATS; i++) {
5255		const char *s = "      ";
5256		unsigned long n = 0;
5257
5258		if (seq == max_seq && NR_HIST_GENS == 1) {
5259			s = "LOYNFA";
5260			n = READ_ONCE(mm_state->stats[hist][i]);
5261		} else if (seq != max_seq && NR_HIST_GENS > 1) {
5262			s = "loynfa";
5263			n = READ_ONCE(mm_state->stats[hist][i]);
5264		}
5265
5266		seq_printf(m, " %10lu%c", n, s[i]);
5267	}
5268	seq_putc(m, '\n');
5269}
5270
5271/* see Documentation/admin-guide/mm/multigen_lru.rst for details */
5272static int lru_gen_seq_show(struct seq_file *m, void *v)
5273{
5274	unsigned long seq;
5275	bool full = !debugfs_real_fops(m->file)->write;
5276	struct lruvec *lruvec = v;
5277	struct lru_gen_folio *lrugen = &lruvec->lrugen;
5278	int nid = lruvec_pgdat(lruvec)->node_id;
5279	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
5280	DEFINE_MAX_SEQ(lruvec);
5281	DEFINE_MIN_SEQ(lruvec);
5282
5283	if (nid == first_memory_node) {
5284		const char *path = memcg ? m->private : "";
5285
5286#ifdef CONFIG_MEMCG
5287		if (memcg)
5288			cgroup_path(memcg->css.cgroup, m->private, PATH_MAX);
5289#endif
5290		seq_printf(m, "memcg %5hu %s\n", mem_cgroup_id(memcg), path);
5291	}
5292
5293	seq_printf(m, " node %5d\n", nid);
5294
5295	if (!full)
5296		seq = min_seq[LRU_GEN_ANON];
5297	else if (max_seq >= MAX_NR_GENS)
5298		seq = max_seq - MAX_NR_GENS + 1;
5299	else
5300		seq = 0;
5301
5302	for (; seq <= max_seq; seq++) {
5303		int type, zone;
5304		int gen = lru_gen_from_seq(seq);
5305		unsigned long birth = READ_ONCE(lruvec->lrugen.timestamps[gen]);
5306
5307		seq_printf(m, " %10lu %10u", seq, jiffies_to_msecs(jiffies - birth));
5308
5309		for (type = 0; type < ANON_AND_FILE; type++) {
5310			unsigned long size = 0;
5311			char mark = full && seq < min_seq[type] ? 'x' : ' ';
5312
5313			for (zone = 0; zone < MAX_NR_ZONES; zone++)
5314				size += max(READ_ONCE(lrugen->nr_pages[gen][type][zone]), 0L);
5315
5316			seq_printf(m, " %10lu%c", size, mark);
5317		}
5318
5319		seq_putc(m, '\n');
5320
5321		if (full)
5322			lru_gen_seq_show_full(m, lruvec, max_seq, min_seq, seq);
5323	}
5324
5325	return 0;
5326}
5327
5328static const struct seq_operations lru_gen_seq_ops = {
5329	.start = lru_gen_seq_start,
5330	.stop = lru_gen_seq_stop,
5331	.next = lru_gen_seq_next,
5332	.show = lru_gen_seq_show,
5333};
5334
5335static int run_aging(struct lruvec *lruvec, unsigned long seq, struct scan_control *sc,
5336		     bool can_swap, bool force_scan)
5337{
5338	DEFINE_MAX_SEQ(lruvec);
5339	DEFINE_MIN_SEQ(lruvec);
5340
5341	if (seq < max_seq)
5342		return 0;
5343
5344	if (seq > max_seq)
5345		return -EINVAL;
5346
5347	if (!force_scan && min_seq[!can_swap] + MAX_NR_GENS - 1 <= max_seq)
5348		return -ERANGE;
5349
5350	try_to_inc_max_seq(lruvec, max_seq, sc, can_swap, force_scan);
5351
5352	return 0;
5353}
5354
5355static int run_eviction(struct lruvec *lruvec, unsigned long seq, struct scan_control *sc,
5356			int swappiness, unsigned long nr_to_reclaim)
5357{
5358	DEFINE_MAX_SEQ(lruvec);
5359
5360	if (seq + MIN_NR_GENS > max_seq)
5361		return -EINVAL;
5362
5363	sc->nr_reclaimed = 0;
5364
5365	while (!signal_pending(current)) {
5366		DEFINE_MIN_SEQ(lruvec);
5367
5368		if (seq < min_seq[!swappiness])
5369			return 0;
5370
5371		if (sc->nr_reclaimed >= nr_to_reclaim)
5372			return 0;
5373
5374		if (!evict_folios(lruvec, sc, swappiness))
5375			return 0;
5376
5377		cond_resched();
5378	}
5379
5380	return -EINTR;
5381}
5382
5383static int run_cmd(char cmd, int memcg_id, int nid, unsigned long seq,
5384		   struct scan_control *sc, int swappiness, unsigned long opt)
5385{
5386	struct lruvec *lruvec;
5387	int err = -EINVAL;
5388	struct mem_cgroup *memcg = NULL;
5389
5390	if (nid < 0 || nid >= MAX_NUMNODES || !node_state(nid, N_MEMORY))
5391		return -EINVAL;
5392
5393	if (!mem_cgroup_disabled()) {
5394		rcu_read_lock();
5395
5396		memcg = mem_cgroup_from_id(memcg_id);
5397		if (!mem_cgroup_tryget(memcg))
5398			memcg = NULL;
5399
5400		rcu_read_unlock();
5401
5402		if (!memcg)
5403			return -EINVAL;
5404	}
5405
5406	if (memcg_id != mem_cgroup_id(memcg))
5407		goto done;
5408
5409	lruvec = get_lruvec(memcg, nid);
5410
5411	if (swappiness < 0)
5412		swappiness = get_swappiness(lruvec, sc);
5413	else if (swappiness > 200)
5414		goto done;
5415
5416	switch (cmd) {
5417	case '+':
5418		err = run_aging(lruvec, seq, sc, swappiness, opt);
5419		break;
5420	case '-':
5421		err = run_eviction(lruvec, seq, sc, swappiness, opt);
5422		break;
5423	}
5424done:
5425	mem_cgroup_put(memcg);
5426
5427	return err;
5428}
5429
5430/* see Documentation/admin-guide/mm/multigen_lru.rst for details */
5431static ssize_t lru_gen_seq_write(struct file *file, const char __user *src,
5432				 size_t len, loff_t *pos)
5433{
5434	void *buf;
5435	char *cur, *next;
5436	unsigned int flags;
5437	struct blk_plug plug;
5438	int err = -EINVAL;
5439	struct scan_control sc = {
5440		.may_writepage = true,
5441		.may_unmap = true,
5442		.may_swap = true,
5443		.reclaim_idx = MAX_NR_ZONES - 1,
5444		.gfp_mask = GFP_KERNEL,
5445	};
5446
5447	buf = kvmalloc(len + 1, GFP_KERNEL);
5448	if (!buf)
5449		return -ENOMEM;
5450
5451	if (copy_from_user(buf, src, len)) {
5452		kvfree(buf);
5453		return -EFAULT;
5454	}
5455
5456	set_task_reclaim_state(current, &sc.reclaim_state);
5457	flags = memalloc_noreclaim_save();
5458	blk_start_plug(&plug);
5459	if (!set_mm_walk(NULL, true)) {
5460		err = -ENOMEM;
5461		goto done;
5462	}
5463
5464	next = buf;
5465	next[len] = '\0';
5466
5467	while ((cur = strsep(&next, ",;\n"))) {
5468		int n;
5469		int end;
5470		char cmd;
5471		unsigned int memcg_id;
5472		unsigned int nid;
5473		unsigned long seq;
5474		unsigned int swappiness = -1;
5475		unsigned long opt = -1;
5476
5477		cur = skip_spaces(cur);
5478		if (!*cur)
5479			continue;
5480
5481		n = sscanf(cur, "%c %u %u %lu %n %u %n %lu %n", &cmd, &memcg_id, &nid,
5482			   &seq, &end, &swappiness, &end, &opt, &end);
5483		if (n < 4 || cur[end]) {
5484			err = -EINVAL;
5485			break;
5486		}
5487
5488		err = run_cmd(cmd, memcg_id, nid, seq, &sc, swappiness, opt);
5489		if (err)
5490			break;
5491	}
5492done:
5493	clear_mm_walk();
5494	blk_finish_plug(&plug);
5495	memalloc_noreclaim_restore(flags);
5496	set_task_reclaim_state(current, NULL);
5497
5498	kvfree(buf);
5499
5500	return err ? : len;
5501}
5502
5503static int lru_gen_seq_open(struct inode *inode, struct file *file)
5504{
5505	return seq_open(file, &lru_gen_seq_ops);
5506}
5507
5508static const struct file_operations lru_gen_rw_fops = {
5509	.open = lru_gen_seq_open,
5510	.read = seq_read,
5511	.write = lru_gen_seq_write,
5512	.llseek = seq_lseek,
5513	.release = seq_release,
5514};
5515
5516static const struct file_operations lru_gen_ro_fops = {
5517	.open = lru_gen_seq_open,
5518	.read = seq_read,
5519	.llseek = seq_lseek,
5520	.release = seq_release,
5521};
5522
5523/******************************************************************************
5524 *                          initialization
5525 ******************************************************************************/
5526
5527void lru_gen_init_pgdat(struct pglist_data *pgdat)
5528{
5529	int i, j;
5530
5531	spin_lock_init(&pgdat->memcg_lru.lock);
5532
5533	for (i = 0; i < MEMCG_NR_GENS; i++) {
5534		for (j = 0; j < MEMCG_NR_BINS; j++)
5535			INIT_HLIST_NULLS_HEAD(&pgdat->memcg_lru.fifo[i][j], i);
5536	}
5537}
5538
5539void lru_gen_init_lruvec(struct lruvec *lruvec)
5540{
5541	int i;
5542	int gen, type, zone;
5543	struct lru_gen_folio *lrugen = &lruvec->lrugen;
5544	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
5545
5546	lrugen->max_seq = MIN_NR_GENS + 1;
5547	lrugen->enabled = lru_gen_enabled();
5548
5549	for (i = 0; i <= MIN_NR_GENS + 1; i++)
5550		lrugen->timestamps[i] = jiffies;
5551
5552	for_each_gen_type_zone(gen, type, zone)
5553		INIT_LIST_HEAD(&lrugen->folios[gen][type][zone]);
5554
5555	if (mm_state)
5556		mm_state->seq = MIN_NR_GENS;
5557}
5558
5559#ifdef CONFIG_MEMCG
5560
5561void lru_gen_init_memcg(struct mem_cgroup *memcg)
5562{
5563	struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
5564
5565	if (!mm_list)
5566		return;
5567
5568	INIT_LIST_HEAD(&mm_list->fifo);
5569	spin_lock_init(&mm_list->lock);
5570}
5571
5572void lru_gen_exit_memcg(struct mem_cgroup *memcg)
5573{
5574	int i;
5575	int nid;
5576	struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
5577
5578	VM_WARN_ON_ONCE(mm_list && !list_empty(&mm_list->fifo));
5579
5580	for_each_node(nid) {
5581		struct lruvec *lruvec = get_lruvec(memcg, nid);
5582		struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
5583
5584		VM_WARN_ON_ONCE(memchr_inv(lruvec->lrugen.nr_pages, 0,
5585					   sizeof(lruvec->lrugen.nr_pages)));
5586
5587		lruvec->lrugen.list.next = LIST_POISON1;
5588
5589		if (!mm_state)
5590			continue;
5591
5592		for (i = 0; i < NR_BLOOM_FILTERS; i++) {
5593			bitmap_free(mm_state->filters[i]);
5594			mm_state->filters[i] = NULL;
5595		}
5596	}
5597}
5598
5599#endif /* CONFIG_MEMCG */
5600
5601static int __init init_lru_gen(void)
5602{
5603	BUILD_BUG_ON(MIN_NR_GENS + 1 >= MAX_NR_GENS);
5604	BUILD_BUG_ON(BIT(LRU_GEN_WIDTH) <= MAX_NR_GENS);
5605
5606	if (sysfs_create_group(mm_kobj, &lru_gen_attr_group))
5607		pr_err("lru_gen: failed to create sysfs group\n");
5608
5609	debugfs_create_file("lru_gen", 0644, NULL, NULL, &lru_gen_rw_fops);
5610	debugfs_create_file("lru_gen_full", 0444, NULL, NULL, &lru_gen_ro_fops);
5611
5612	return 0;
5613};
5614late_initcall(init_lru_gen);
5615
5616#else /* !CONFIG_LRU_GEN */
5617
5618static void lru_gen_age_node(struct pglist_data *pgdat, struct scan_control *sc)
5619{
5620	BUILD_BUG();
5621}
5622
5623static void lru_gen_shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
5624{
5625	BUILD_BUG();
5626}
5627
5628static void lru_gen_shrink_node(struct pglist_data *pgdat, struct scan_control *sc)
5629{
5630	BUILD_BUG();
5631}
5632
5633#endif /* CONFIG_LRU_GEN */
5634
5635static void shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
5636{
5637	unsigned long nr[NR_LRU_LISTS];
5638	unsigned long targets[NR_LRU_LISTS];
5639	unsigned long nr_to_scan;
5640	enum lru_list lru;
5641	unsigned long nr_reclaimed = 0;
5642	unsigned long nr_to_reclaim = sc->nr_to_reclaim;
5643	bool proportional_reclaim;
5644	struct blk_plug plug;
5645
5646	if (lru_gen_enabled() && !root_reclaim(sc)) {
5647		lru_gen_shrink_lruvec(lruvec, sc);
5648		return;
5649	}
5650
5651	get_scan_count(lruvec, sc, nr);
5652
5653	/* Record the original scan target for proportional adjustments later */
5654	memcpy(targets, nr, sizeof(nr));
5655
5656	/*
5657	 * Global reclaiming within direct reclaim at DEF_PRIORITY is a normal
5658	 * event that can occur when there is little memory pressure e.g.
5659	 * multiple streaming readers/writers. Hence, we do not abort scanning
5660	 * when the requested number of pages are reclaimed when scanning at
5661	 * DEF_PRIORITY on the assumption that the fact we are direct
5662	 * reclaiming implies that kswapd is not keeping up and it is best to
5663	 * do a batch of work at once. For memcg reclaim one check is made to
5664	 * abort proportional reclaim if either the file or anon lru has already
5665	 * dropped to zero at the first pass.
5666	 */
5667	proportional_reclaim = (!cgroup_reclaim(sc) && !current_is_kswapd() &&
5668				sc->priority == DEF_PRIORITY);
5669
5670	blk_start_plug(&plug);
5671	while (nr[LRU_INACTIVE_ANON] || nr[LRU_ACTIVE_FILE] ||
5672					nr[LRU_INACTIVE_FILE]) {
5673		unsigned long nr_anon, nr_file, percentage;
5674		unsigned long nr_scanned;
5675
5676		for_each_evictable_lru(lru) {
5677			if (nr[lru]) {
5678				nr_to_scan = min(nr[lru], SWAP_CLUSTER_MAX);
5679				nr[lru] -= nr_to_scan;
5680
5681				nr_reclaimed += shrink_list(lru, nr_to_scan,
5682							    lruvec, sc);
5683			}
5684		}
5685
5686		cond_resched();
5687
5688		if (nr_reclaimed < nr_to_reclaim || proportional_reclaim)
5689			continue;
5690
5691		/*
5692		 * For kswapd and memcg, reclaim at least the number of pages
5693		 * requested. Ensure that the anon and file LRUs are scanned
5694		 * proportionally what was requested by get_scan_count(). We
5695		 * stop reclaiming one LRU and reduce the amount scanning
5696		 * proportional to the original scan target.
5697		 */
5698		nr_file = nr[LRU_INACTIVE_FILE] + nr[LRU_ACTIVE_FILE];
5699		nr_anon = nr[LRU_INACTIVE_ANON] + nr[LRU_ACTIVE_ANON];
5700
5701		/*
5702		 * It's just vindictive to attack the larger once the smaller
5703		 * has gone to zero.  And given the way we stop scanning the
5704		 * smaller below, this makes sure that we only make one nudge
5705		 * towards proportionality once we've got nr_to_reclaim.
5706		 */
5707		if (!nr_file || !nr_anon)
5708			break;
5709
5710		if (nr_file > nr_anon) {
5711			unsigned long scan_target = targets[LRU_INACTIVE_ANON] +
5712						targets[LRU_ACTIVE_ANON] + 1;
5713			lru = LRU_BASE;
5714			percentage = nr_anon * 100 / scan_target;
5715		} else {
5716			unsigned long scan_target = targets[LRU_INACTIVE_FILE] +
5717						targets[LRU_ACTIVE_FILE] + 1;
5718			lru = LRU_FILE;
5719			percentage = nr_file * 100 / scan_target;
5720		}
5721
5722		/* Stop scanning the smaller of the LRU */
5723		nr[lru] = 0;
5724		nr[lru + LRU_ACTIVE] = 0;
5725
5726		/*
5727		 * Recalculate the other LRU scan count based on its original
5728		 * scan target and the percentage scanning already complete
5729		 */
5730		lru = (lru == LRU_FILE) ? LRU_BASE : LRU_FILE;
5731		nr_scanned = targets[lru] - nr[lru];
5732		nr[lru] = targets[lru] * (100 - percentage) / 100;
5733		nr[lru] -= min(nr[lru], nr_scanned);
5734
5735		lru += LRU_ACTIVE;
5736		nr_scanned = targets[lru] - nr[lru];
5737		nr[lru] = targets[lru] * (100 - percentage) / 100;
5738		nr[lru] -= min(nr[lru], nr_scanned);
5739	}
5740	blk_finish_plug(&plug);
5741	sc->nr_reclaimed += nr_reclaimed;
5742
5743	/*
5744	 * Even if we did not try to evict anon pages at all, we want to
5745	 * rebalance the anon lru active/inactive ratio.
5746	 */
5747	if (can_age_anon_pages(lruvec_pgdat(lruvec), sc) &&
5748	    inactive_is_low(lruvec, LRU_INACTIVE_ANON))
5749		shrink_active_list(SWAP_CLUSTER_MAX, lruvec,
5750				   sc, LRU_ACTIVE_ANON);
5751}
5752
5753/* Use reclaim/compaction for costly allocs or under memory pressure */
5754static bool in_reclaim_compaction(struct scan_control *sc)
5755{
5756	if (gfp_compaction_allowed(sc->gfp_mask) && sc->order &&
5757			(sc->order > PAGE_ALLOC_COSTLY_ORDER ||
5758			 sc->priority < DEF_PRIORITY - 2))
5759		return true;
5760
5761	return false;
5762}
5763
5764/*
5765 * Reclaim/compaction is used for high-order allocation requests. It reclaims
5766 * order-0 pages before compacting the zone. should_continue_reclaim() returns
5767 * true if more pages should be reclaimed such that when the page allocator
5768 * calls try_to_compact_pages() that it will have enough free pages to succeed.
5769 * It will give up earlier than that if there is difficulty reclaiming pages.
5770 */
5771static inline bool should_continue_reclaim(struct pglist_data *pgdat,
5772					unsigned long nr_reclaimed,
5773					struct scan_control *sc)
5774{
5775	unsigned long pages_for_compaction;
5776	unsigned long inactive_lru_pages;
5777	int z;
5778
5779	/* If not in reclaim/compaction mode, stop */
5780	if (!in_reclaim_compaction(sc))
5781		return false;
5782
5783	/*
5784	 * Stop if we failed to reclaim any pages from the last SWAP_CLUSTER_MAX
5785	 * number of pages that were scanned. This will return to the caller
5786	 * with the risk reclaim/compaction and the resulting allocation attempt
5787	 * fails. In the past we have tried harder for __GFP_RETRY_MAYFAIL
5788	 * allocations through requiring that the full LRU list has been scanned
5789	 * first, by assuming that zero delta of sc->nr_scanned means full LRU
5790	 * scan, but that approximation was wrong, and there were corner cases
5791	 * where always a non-zero amount of pages were scanned.
5792	 */
5793	if (!nr_reclaimed)
5794		return false;
5795
5796	/* If compaction would go ahead or the allocation would succeed, stop */
5797	for (z = 0; z <= sc->reclaim_idx; z++) {
5798		struct zone *zone = &pgdat->node_zones[z];
5799		if (!managed_zone(zone))
5800			continue;
5801
5802		/* Allocation can already succeed, nothing to do */
5803		if (zone_watermark_ok(zone, sc->order, min_wmark_pages(zone),
5804				      sc->reclaim_idx, 0))
5805			return false;
5806
5807		if (compaction_suitable(zone, sc->order, sc->reclaim_idx))
5808			return false;
5809	}
5810
5811	/*
5812	 * If we have not reclaimed enough pages for compaction and the
5813	 * inactive lists are large enough, continue reclaiming
5814	 */
5815	pages_for_compaction = compact_gap(sc->order);
5816	inactive_lru_pages = node_page_state(pgdat, NR_INACTIVE_FILE);
5817	if (can_reclaim_anon_pages(NULL, pgdat->node_id, sc))
5818		inactive_lru_pages += node_page_state(pgdat, NR_INACTIVE_ANON);
5819
5820	return inactive_lru_pages > pages_for_compaction;
5821}
5822
5823static void shrink_node_memcgs(pg_data_t *pgdat, struct scan_control *sc)
5824{
5825	struct mem_cgroup *target_memcg = sc->target_mem_cgroup;
5826	struct mem_cgroup *memcg;
5827
5828	memcg = mem_cgroup_iter(target_memcg, NULL, NULL);
5829	do {
5830		struct lruvec *lruvec = mem_cgroup_lruvec(memcg, pgdat);
5831		unsigned long reclaimed;
5832		unsigned long scanned;
5833
5834		/*
5835		 * This loop can become CPU-bound when target memcgs
5836		 * aren't eligible for reclaim - either because they
5837		 * don't have any reclaimable pages, or because their
5838		 * memory is explicitly protected. Avoid soft lockups.
5839		 */
5840		cond_resched();
5841
5842		mem_cgroup_calculate_protection(target_memcg, memcg);
5843
5844		if (mem_cgroup_below_min(target_memcg, memcg)) {
5845			/*
5846			 * Hard protection.
5847			 * If there is no reclaimable memory, OOM.
5848			 */
5849			continue;
5850		} else if (mem_cgroup_below_low(target_memcg, memcg)) {
5851			/*
5852			 * Soft protection.
5853			 * Respect the protection only as long as
5854			 * there is an unprotected supply
5855			 * of reclaimable memory from other cgroups.
5856			 */
5857			if (!sc->memcg_low_reclaim) {
5858				sc->memcg_low_skipped = 1;
5859				continue;
5860			}
5861			memcg_memory_event(memcg, MEMCG_LOW);
5862		}
5863
5864		reclaimed = sc->nr_reclaimed;
5865		scanned = sc->nr_scanned;
5866
5867		shrink_lruvec(lruvec, sc);
5868
5869		shrink_slab(sc->gfp_mask, pgdat->node_id, memcg,
5870			    sc->priority);
5871
5872		/* Record the group's reclaim efficiency */
5873		if (!sc->proactive)
5874			vmpressure(sc->gfp_mask, memcg, false,
5875				   sc->nr_scanned - scanned,
5876				   sc->nr_reclaimed - reclaimed);
5877
5878	} while ((memcg = mem_cgroup_iter(target_memcg, memcg, NULL)));
5879}
5880
5881static void shrink_node(pg_data_t *pgdat, struct scan_control *sc)
5882{
5883	unsigned long nr_reclaimed, nr_scanned, nr_node_reclaimed;
5884	struct lruvec *target_lruvec;
5885	bool reclaimable = false;
5886
5887	if (lru_gen_enabled() && root_reclaim(sc)) {
5888		lru_gen_shrink_node(pgdat, sc);
5889		return;
5890	}
5891
5892	target_lruvec = mem_cgroup_lruvec(sc->target_mem_cgroup, pgdat);
5893
5894again:
5895	memset(&sc->nr, 0, sizeof(sc->nr));
5896
5897	nr_reclaimed = sc->nr_reclaimed;
5898	nr_scanned = sc->nr_scanned;
5899
5900	prepare_scan_control(pgdat, sc);
5901
5902	shrink_node_memcgs(pgdat, sc);
5903
5904	flush_reclaim_state(sc);
5905
5906	nr_node_reclaimed = sc->nr_reclaimed - nr_reclaimed;
5907
5908	/* Record the subtree's reclaim efficiency */
5909	if (!sc->proactive)
5910		vmpressure(sc->gfp_mask, sc->target_mem_cgroup, true,
5911			   sc->nr_scanned - nr_scanned, nr_node_reclaimed);
5912
5913	if (nr_node_reclaimed)
5914		reclaimable = true;
5915
5916	if (current_is_kswapd()) {
5917		/*
5918		 * If reclaim is isolating dirty pages under writeback,
5919		 * it implies that the long-lived page allocation rate
5920		 * is exceeding the page laundering rate. Either the
5921		 * global limits are not being effective at throttling
5922		 * processes due to the page distribution throughout
5923		 * zones or there is heavy usage of a slow backing
5924		 * device. The only option is to throttle from reclaim
5925		 * context which is not ideal as there is no guarantee
5926		 * the dirtying process is throttled in the same way
5927		 * balance_dirty_pages() manages.
5928		 *
5929		 * Once a node is flagged PGDAT_WRITEBACK, kswapd will
5930		 * count the number of pages under pages flagged for
5931		 * immediate reclaim and stall if any are encountered
5932		 * in the nr_immediate check below.
5933		 */
5934		if (sc->nr.writeback && sc->nr.writeback == sc->nr.taken)
5935			set_bit(PGDAT_WRITEBACK, &pgdat->flags);
5936
5937		/* Allow kswapd to start writing pages during reclaim.*/
5938		if (sc->nr.unqueued_dirty == sc->nr.file_taken)
5939			set_bit(PGDAT_DIRTY, &pgdat->flags);
5940
5941		/*
5942		 * If kswapd scans pages marked for immediate
5943		 * reclaim and under writeback (nr_immediate), it
5944		 * implies that pages are cycling through the LRU
5945		 * faster than they are written so forcibly stall
5946		 * until some pages complete writeback.
5947		 */
5948		if (sc->nr.immediate)
5949			reclaim_throttle(pgdat, VMSCAN_THROTTLE_WRITEBACK);
5950	}
5951
5952	/*
5953	 * Tag a node/memcg as congested if all the dirty pages were marked
5954	 * for writeback and immediate reclaim (counted in nr.congested).
5955	 *
5956	 * Legacy memcg will stall in page writeback so avoid forcibly
5957	 * stalling in reclaim_throttle().
5958	 */
5959	if (sc->nr.dirty && sc->nr.dirty == sc->nr.congested) {
5960		if (cgroup_reclaim(sc) && writeback_throttling_sane(sc))
5961			set_bit(LRUVEC_CGROUP_CONGESTED, &target_lruvec->flags);
5962
5963		if (current_is_kswapd())
5964			set_bit(LRUVEC_NODE_CONGESTED, &target_lruvec->flags);
5965	}
5966
5967	/*
5968	 * Stall direct reclaim for IO completions if the lruvec is
5969	 * node is congested. Allow kswapd to continue until it
5970	 * starts encountering unqueued dirty pages or cycling through
5971	 * the LRU too quickly.
5972	 */
5973	if (!current_is_kswapd() && current_may_throttle() &&
5974	    !sc->hibernation_mode &&
5975	    (test_bit(LRUVEC_CGROUP_CONGESTED, &target_lruvec->flags) ||
5976	     test_bit(LRUVEC_NODE_CONGESTED, &target_lruvec->flags)))
5977		reclaim_throttle(pgdat, VMSCAN_THROTTLE_CONGESTED);
5978
5979	if (should_continue_reclaim(pgdat, nr_node_reclaimed, sc))
5980		goto again;
5981
5982	/*
5983	 * Kswapd gives up on balancing particular nodes after too
5984	 * many failures to reclaim anything from them and goes to
5985	 * sleep. On reclaim progress, reset the failure counter. A
5986	 * successful direct reclaim run will revive a dormant kswapd.
5987	 */
5988	if (reclaimable)
5989		pgdat->kswapd_failures = 0;
 
 
5990}
5991
5992/*
5993 * Returns true if compaction should go ahead for a costly-order request, or
5994 * the allocation would already succeed without compaction. Return false if we
5995 * should reclaim first.
5996 */
5997static inline bool compaction_ready(struct zone *zone, struct scan_control *sc)
5998{
5999	unsigned long watermark;
6000
6001	if (!gfp_compaction_allowed(sc->gfp_mask))
6002		return false;
6003
6004	/* Allocation can already succeed, nothing to do */
6005	if (zone_watermark_ok(zone, sc->order, min_wmark_pages(zone),
6006			      sc->reclaim_idx, 0))
6007		return true;
6008
6009	/* Compaction cannot yet proceed. Do reclaim. */
6010	if (!compaction_suitable(zone, sc->order, sc->reclaim_idx))
6011		return false;
6012
6013	/*
6014	 * Compaction is already possible, but it takes time to run and there
6015	 * are potentially other callers using the pages just freed. So proceed
6016	 * with reclaim to make a buffer of free pages available to give
6017	 * compaction a reasonable chance of completing and allocating the page.
6018	 * Note that we won't actually reclaim the whole buffer in one attempt
6019	 * as the target watermark in should_continue_reclaim() is lower. But if
6020	 * we are already above the high+gap watermark, don't reclaim at all.
6021	 */
6022	watermark = high_wmark_pages(zone) + compact_gap(sc->order);
6023
6024	return zone_watermark_ok_safe(zone, 0, watermark, sc->reclaim_idx);
6025}
6026
6027static void consider_reclaim_throttle(pg_data_t *pgdat, struct scan_control *sc)
6028{
6029	/*
6030	 * If reclaim is making progress greater than 12% efficiency then
6031	 * wake all the NOPROGRESS throttled tasks.
6032	 */
6033	if (sc->nr_reclaimed > (sc->nr_scanned >> 3)) {
6034		wait_queue_head_t *wqh;
6035
6036		wqh = &pgdat->reclaim_wait[VMSCAN_THROTTLE_NOPROGRESS];
6037		if (waitqueue_active(wqh))
6038			wake_up(wqh);
6039
6040		return;
6041	}
6042
6043	/*
6044	 * Do not throttle kswapd or cgroup reclaim on NOPROGRESS as it will
6045	 * throttle on VMSCAN_THROTTLE_WRITEBACK if there are too many pages
6046	 * under writeback and marked for immediate reclaim at the tail of the
6047	 * LRU.
6048	 */
6049	if (current_is_kswapd() || cgroup_reclaim(sc))
6050		return;
6051
6052	/* Throttle if making no progress at high prioities. */
6053	if (sc->priority == 1 && !sc->nr_reclaimed)
6054		reclaim_throttle(pgdat, VMSCAN_THROTTLE_NOPROGRESS);
6055}
6056
6057/*
6058 * This is the direct reclaim path, for page-allocating processes.  We only
6059 * try to reclaim pages from zones which will satisfy the caller's allocation
6060 * request.
6061 *
6062 * If a zone is deemed to be full of pinned pages then just give it a light
6063 * scan then give up on it.
6064 */
6065static void shrink_zones(struct zonelist *zonelist, struct scan_control *sc)
6066{
6067	struct zoneref *z;
6068	struct zone *zone;
6069	unsigned long nr_soft_reclaimed;
6070	unsigned long nr_soft_scanned;
6071	gfp_t orig_mask;
6072	pg_data_t *last_pgdat = NULL;
6073	pg_data_t *first_pgdat = NULL;
6074
6075	/*
6076	 * If the number of buffer_heads in the machine exceeds the maximum
6077	 * allowed level, force direct reclaim to scan the highmem zone as
6078	 * highmem pages could be pinning lowmem pages storing buffer_heads
6079	 */
6080	orig_mask = sc->gfp_mask;
6081	if (buffer_heads_over_limit) {
6082		sc->gfp_mask |= __GFP_HIGHMEM;
6083		sc->reclaim_idx = gfp_zone(sc->gfp_mask);
6084	}
6085
6086	for_each_zone_zonelist_nodemask(zone, z, zonelist,
6087					sc->reclaim_idx, sc->nodemask) {
6088		/*
6089		 * Take care memory controller reclaiming has small influence
6090		 * to global LRU.
6091		 */
6092		if (!cgroup_reclaim(sc)) {
6093			if (!cpuset_zone_allowed(zone,
6094						 GFP_KERNEL | __GFP_HARDWALL))
6095				continue;
6096
6097			/*
6098			 * If we already have plenty of memory free for
6099			 * compaction in this zone, don't free any more.
6100			 * Even though compaction is invoked for any
6101			 * non-zero order, only frequent costly order
6102			 * reclamation is disruptive enough to become a
6103			 * noticeable problem, like transparent huge
6104			 * page allocations.
6105			 */
6106			if (IS_ENABLED(CONFIG_COMPACTION) &&
6107			    sc->order > PAGE_ALLOC_COSTLY_ORDER &&
6108			    compaction_ready(zone, sc)) {
6109				sc->compaction_ready = true;
6110				continue;
6111			}
6112
6113			/*
6114			 * Shrink each node in the zonelist once. If the
6115			 * zonelist is ordered by zone (not the default) then a
6116			 * node may be shrunk multiple times but in that case
6117			 * the user prefers lower zones being preserved.
6118			 */
6119			if (zone->zone_pgdat == last_pgdat)
6120				continue;
6121
6122			/*
6123			 * This steals pages from memory cgroups over softlimit
6124			 * and returns the number of reclaimed pages and
6125			 * scanned pages. This works for global memory pressure
6126			 * and balancing, not for a memcg's limit.
6127			 */
6128			nr_soft_scanned = 0;
6129			nr_soft_reclaimed = mem_cgroup_soft_limit_reclaim(zone->zone_pgdat,
6130						sc->order, sc->gfp_mask,
6131						&nr_soft_scanned);
6132			sc->nr_reclaimed += nr_soft_reclaimed;
6133			sc->nr_scanned += nr_soft_scanned;
6134			/* need some check for avoid more shrink_zone() */
6135		}
6136
6137		if (!first_pgdat)
6138			first_pgdat = zone->zone_pgdat;
6139
6140		/* See comment about same check for global reclaim above */
6141		if (zone->zone_pgdat == last_pgdat)
6142			continue;
6143		last_pgdat = zone->zone_pgdat;
6144		shrink_node(zone->zone_pgdat, sc);
6145	}
6146
6147	if (first_pgdat)
6148		consider_reclaim_throttle(first_pgdat, sc);
6149
6150	/*
6151	 * Restore to original mask to avoid the impact on the caller if we
6152	 * promoted it to __GFP_HIGHMEM.
6153	 */
6154	sc->gfp_mask = orig_mask;
6155}
6156
6157static void snapshot_refaults(struct mem_cgroup *target_memcg, pg_data_t *pgdat)
6158{
6159	struct lruvec *target_lruvec;
6160	unsigned long refaults;
6161
6162	if (lru_gen_enabled())
6163		return;
6164
6165	target_lruvec = mem_cgroup_lruvec(target_memcg, pgdat);
6166	refaults = lruvec_page_state(target_lruvec, WORKINGSET_ACTIVATE_ANON);
6167	target_lruvec->refaults[WORKINGSET_ANON] = refaults;
6168	refaults = lruvec_page_state(target_lruvec, WORKINGSET_ACTIVATE_FILE);
6169	target_lruvec->refaults[WORKINGSET_FILE] = refaults;
6170}
6171
6172/*
6173 * This is the main entry point to direct page reclaim.
6174 *
6175 * If a full scan of the inactive list fails to free enough memory then we
6176 * are "out of memory" and something needs to be killed.
6177 *
6178 * If the caller is !__GFP_FS then the probability of a failure is reasonably
6179 * high - the zone may be full of dirty or under-writeback pages, which this
6180 * caller can't do much about.  We kick the writeback threads and take explicit
6181 * naps in the hope that some of these pages can be written.  But if the
6182 * allocating task holds filesystem locks which prevent writeout this might not
6183 * work, and the allocation attempt will fail.
6184 *
6185 * returns:	0, if no pages reclaimed
6186 * 		else, the number of pages reclaimed
6187 */
6188static unsigned long do_try_to_free_pages(struct zonelist *zonelist,
6189					  struct scan_control *sc)
6190{
6191	int initial_priority = sc->priority;
6192	pg_data_t *last_pgdat;
6193	struct zoneref *z;
6194	struct zone *zone;
6195retry:
6196	delayacct_freepages_start();
6197
6198	if (!cgroup_reclaim(sc))
6199		__count_zid_vm_events(ALLOCSTALL, sc->reclaim_idx, 1);
6200
6201	do {
6202		if (!sc->proactive)
6203			vmpressure_prio(sc->gfp_mask, sc->target_mem_cgroup,
6204					sc->priority);
6205		sc->nr_scanned = 0;
6206		shrink_zones(zonelist, sc);
6207
6208		if (sc->nr_reclaimed >= sc->nr_to_reclaim)
6209			break;
6210
6211		if (sc->compaction_ready)
6212			break;
6213
6214		/*
6215		 * If we're getting trouble reclaiming, start doing
6216		 * writepage even in laptop mode.
6217		 */
6218		if (sc->priority < DEF_PRIORITY - 2)
6219			sc->may_writepage = 1;
6220	} while (--sc->priority >= 0);
6221
6222	last_pgdat = NULL;
6223	for_each_zone_zonelist_nodemask(zone, z, zonelist, sc->reclaim_idx,
6224					sc->nodemask) {
6225		if (zone->zone_pgdat == last_pgdat)
6226			continue;
6227		last_pgdat = zone->zone_pgdat;
6228
6229		snapshot_refaults(sc->target_mem_cgroup, zone->zone_pgdat);
6230
6231		if (cgroup_reclaim(sc)) {
6232			struct lruvec *lruvec;
6233
6234			lruvec = mem_cgroup_lruvec(sc->target_mem_cgroup,
6235						   zone->zone_pgdat);
6236			clear_bit(LRUVEC_CGROUP_CONGESTED, &lruvec->flags);
6237		}
6238	}
6239
6240	delayacct_freepages_end();
6241
6242	if (sc->nr_reclaimed)
6243		return sc->nr_reclaimed;
6244
6245	/* Aborted reclaim to try compaction? don't OOM, then */
6246	if (sc->compaction_ready)
6247		return 1;
6248
6249	/*
6250	 * We make inactive:active ratio decisions based on the node's
6251	 * composition of memory, but a restrictive reclaim_idx or a
6252	 * memory.low cgroup setting can exempt large amounts of
6253	 * memory from reclaim. Neither of which are very common, so
6254	 * instead of doing costly eligibility calculations of the
6255	 * entire cgroup subtree up front, we assume the estimates are
6256	 * good, and retry with forcible deactivation if that fails.
6257	 */
6258	if (sc->skipped_deactivate) {
6259		sc->priority = initial_priority;
6260		sc->force_deactivate = 1;
6261		sc->skipped_deactivate = 0;
6262		goto retry;
6263	}
6264
6265	/* Untapped cgroup reserves?  Don't OOM, retry. */
6266	if (sc->memcg_low_skipped) {
6267		sc->priority = initial_priority;
6268		sc->force_deactivate = 0;
6269		sc->memcg_low_reclaim = 1;
6270		sc->memcg_low_skipped = 0;
6271		goto retry;
6272	}
6273
6274	return 0;
6275}
6276
6277static bool allow_direct_reclaim(pg_data_t *pgdat)
6278{
6279	struct zone *zone;
6280	unsigned long pfmemalloc_reserve = 0;
6281	unsigned long free_pages = 0;
6282	int i;
6283	bool wmark_ok;
6284
6285	if (pgdat->kswapd_failures >= MAX_RECLAIM_RETRIES)
6286		return true;
6287
6288	for (i = 0; i <= ZONE_NORMAL; i++) {
6289		zone = &pgdat->node_zones[i];
6290		if (!managed_zone(zone))
6291			continue;
6292
6293		if (!zone_reclaimable_pages(zone))
6294			continue;
6295
6296		pfmemalloc_reserve += min_wmark_pages(zone);
6297		free_pages += zone_page_state_snapshot(zone, NR_FREE_PAGES);
6298	}
6299
6300	/* If there are no reserves (unexpected config) then do not throttle */
6301	if (!pfmemalloc_reserve)
6302		return true;
6303
6304	wmark_ok = free_pages > pfmemalloc_reserve / 2;
6305
6306	/* kswapd must be awake if processes are being throttled */
6307	if (!wmark_ok && waitqueue_active(&pgdat->kswapd_wait)) {
6308		if (READ_ONCE(pgdat->kswapd_highest_zoneidx) > ZONE_NORMAL)
6309			WRITE_ONCE(pgdat->kswapd_highest_zoneidx, ZONE_NORMAL);
6310
6311		wake_up_interruptible(&pgdat->kswapd_wait);
6312	}
6313
6314	return wmark_ok;
6315}
6316
6317/*
6318 * Throttle direct reclaimers if backing storage is backed by the network
6319 * and the PFMEMALLOC reserve for the preferred node is getting dangerously
6320 * depleted. kswapd will continue to make progress and wake the processes
6321 * when the low watermark is reached.
6322 *
6323 * Returns true if a fatal signal was delivered during throttling. If this
6324 * happens, the page allocator should not consider triggering the OOM killer.
6325 */
6326static bool throttle_direct_reclaim(gfp_t gfp_mask, struct zonelist *zonelist,
6327					nodemask_t *nodemask)
6328{
6329	struct zoneref *z;
6330	struct zone *zone;
6331	pg_data_t *pgdat = NULL;
6332
6333	/*
6334	 * Kernel threads should not be throttled as they may be indirectly
6335	 * responsible for cleaning pages necessary for reclaim to make forward
6336	 * progress. kjournald for example may enter direct reclaim while
6337	 * committing a transaction where throttling it could forcing other
6338	 * processes to block on log_wait_commit().
6339	 */
6340	if (current->flags & PF_KTHREAD)
6341		goto out;
6342
6343	/*
6344	 * If a fatal signal is pending, this process should not throttle.
6345	 * It should return quickly so it can exit and free its memory
6346	 */
6347	if (fatal_signal_pending(current))
6348		goto out;
6349
6350	/*
6351	 * Check if the pfmemalloc reserves are ok by finding the first node
6352	 * with a usable ZONE_NORMAL or lower zone. The expectation is that
6353	 * GFP_KERNEL will be required for allocating network buffers when
6354	 * swapping over the network so ZONE_HIGHMEM is unusable.
6355	 *
6356	 * Throttling is based on the first usable node and throttled processes
6357	 * wait on a queue until kswapd makes progress and wakes them. There
6358	 * is an affinity then between processes waking up and where reclaim
6359	 * progress has been made assuming the process wakes on the same node.
6360	 * More importantly, processes running on remote nodes will not compete
6361	 * for remote pfmemalloc reserves and processes on different nodes
6362	 * should make reasonable progress.
6363	 */
6364	for_each_zone_zonelist_nodemask(zone, z, zonelist,
6365					gfp_zone(gfp_mask), nodemask) {
6366		if (zone_idx(zone) > ZONE_NORMAL)
6367			continue;
6368
6369		/* Throttle based on the first usable node */
6370		pgdat = zone->zone_pgdat;
6371		if (allow_direct_reclaim(pgdat))
6372			goto out;
6373		break;
6374	}
6375
6376	/* If no zone was usable by the allocation flags then do not throttle */
6377	if (!pgdat)
6378		goto out;
6379
6380	/* Account for the throttling */
6381	count_vm_event(PGSCAN_DIRECT_THROTTLE);
6382
6383	/*
6384	 * If the caller cannot enter the filesystem, it's possible that it
6385	 * is due to the caller holding an FS lock or performing a journal
6386	 * transaction in the case of a filesystem like ext[3|4]. In this case,
6387	 * it is not safe to block on pfmemalloc_wait as kswapd could be
6388	 * blocked waiting on the same lock. Instead, throttle for up to a
6389	 * second before continuing.
6390	 */
6391	if (!(gfp_mask & __GFP_FS))
6392		wait_event_interruptible_timeout(pgdat->pfmemalloc_wait,
6393			allow_direct_reclaim(pgdat), HZ);
6394	else
6395		/* Throttle until kswapd wakes the process */
6396		wait_event_killable(zone->zone_pgdat->pfmemalloc_wait,
6397			allow_direct_reclaim(pgdat));
6398
6399	if (fatal_signal_pending(current))
6400		return true;
6401
6402out:
6403	return false;
6404}
6405
6406unsigned long try_to_free_pages(struct zonelist *zonelist, int order,
6407				gfp_t gfp_mask, nodemask_t *nodemask)
6408{
6409	unsigned long nr_reclaimed;
6410	struct scan_control sc = {
6411		.nr_to_reclaim = SWAP_CLUSTER_MAX,
6412		.gfp_mask = current_gfp_context(gfp_mask),
6413		.reclaim_idx = gfp_zone(gfp_mask),
6414		.order = order,
6415		.nodemask = nodemask,
6416		.priority = DEF_PRIORITY,
6417		.may_writepage = !laptop_mode,
6418		.may_unmap = 1,
6419		.may_swap = 1,
6420	};
6421
6422	/*
6423	 * scan_control uses s8 fields for order, priority, and reclaim_idx.
6424	 * Confirm they are large enough for max values.
6425	 */
6426	BUILD_BUG_ON(MAX_PAGE_ORDER >= S8_MAX);
6427	BUILD_BUG_ON(DEF_PRIORITY > S8_MAX);
6428	BUILD_BUG_ON(MAX_NR_ZONES > S8_MAX);
6429
6430	/*
6431	 * Do not enter reclaim if fatal signal was delivered while throttled.
6432	 * 1 is returned so that the page allocator does not OOM kill at this
6433	 * point.
6434	 */
6435	if (throttle_direct_reclaim(sc.gfp_mask, zonelist, nodemask))
6436		return 1;
6437
6438	set_task_reclaim_state(current, &sc.reclaim_state);
6439	trace_mm_vmscan_direct_reclaim_begin(order, sc.gfp_mask);
6440
6441	nr_reclaimed = do_try_to_free_pages(zonelist, &sc);
6442
6443	trace_mm_vmscan_direct_reclaim_end(nr_reclaimed);
6444	set_task_reclaim_state(current, NULL);
6445
6446	return nr_reclaimed;
6447}
6448
6449#ifdef CONFIG_MEMCG
6450
6451/* Only used by soft limit reclaim. Do not reuse for anything else. */
6452unsigned long mem_cgroup_shrink_node(struct mem_cgroup *memcg,
6453						gfp_t gfp_mask, bool noswap,
6454						pg_data_t *pgdat,
6455						unsigned long *nr_scanned)
6456{
6457	struct lruvec *lruvec = mem_cgroup_lruvec(memcg, pgdat);
6458	struct scan_control sc = {
6459		.nr_to_reclaim = SWAP_CLUSTER_MAX,
6460		.target_mem_cgroup = memcg,
6461		.may_writepage = !laptop_mode,
6462		.may_unmap = 1,
6463		.reclaim_idx = MAX_NR_ZONES - 1,
6464		.may_swap = !noswap,
6465	};
6466
6467	WARN_ON_ONCE(!current->reclaim_state);
6468
6469	sc.gfp_mask = (gfp_mask & GFP_RECLAIM_MASK) |
6470			(GFP_HIGHUSER_MOVABLE & ~GFP_RECLAIM_MASK);
6471
6472	trace_mm_vmscan_memcg_softlimit_reclaim_begin(sc.order,
6473						      sc.gfp_mask);
6474
6475	/*
6476	 * NOTE: Although we can get the priority field, using it
6477	 * here is not a good idea, since it limits the pages we can scan.
6478	 * if we don't reclaim here, the shrink_node from balance_pgdat
6479	 * will pick up pages from other mem cgroup's as well. We hack
6480	 * the priority and make it zero.
6481	 */
6482	shrink_lruvec(lruvec, &sc);
6483
6484	trace_mm_vmscan_memcg_softlimit_reclaim_end(sc.nr_reclaimed);
6485
6486	*nr_scanned = sc.nr_scanned;
6487
6488	return sc.nr_reclaimed;
6489}
6490
6491unsigned long try_to_free_mem_cgroup_pages(struct mem_cgroup *memcg,
6492					   unsigned long nr_pages,
6493					   gfp_t gfp_mask,
6494					   unsigned int reclaim_options)
6495{
6496	unsigned long nr_reclaimed;
6497	unsigned int noreclaim_flag;
6498	struct scan_control sc = {
6499		.nr_to_reclaim = max(nr_pages, SWAP_CLUSTER_MAX),
6500		.gfp_mask = (current_gfp_context(gfp_mask) & GFP_RECLAIM_MASK) |
6501				(GFP_HIGHUSER_MOVABLE & ~GFP_RECLAIM_MASK),
6502		.reclaim_idx = MAX_NR_ZONES - 1,
6503		.target_mem_cgroup = memcg,
6504		.priority = DEF_PRIORITY,
6505		.may_writepage = !laptop_mode,
6506		.may_unmap = 1,
6507		.may_swap = !!(reclaim_options & MEMCG_RECLAIM_MAY_SWAP),
6508		.proactive = !!(reclaim_options & MEMCG_RECLAIM_PROACTIVE),
6509	};
6510	/*
6511	 * Traverse the ZONELIST_FALLBACK zonelist of the current node to put
6512	 * equal pressure on all the nodes. This is based on the assumption that
6513	 * the reclaim does not bail out early.
6514	 */
6515	struct zonelist *zonelist = node_zonelist(numa_node_id(), sc.gfp_mask);
6516
6517	set_task_reclaim_state(current, &sc.reclaim_state);
6518	trace_mm_vmscan_memcg_reclaim_begin(0, sc.gfp_mask);
6519	noreclaim_flag = memalloc_noreclaim_save();
6520
6521	nr_reclaimed = do_try_to_free_pages(zonelist, &sc);
6522
6523	memalloc_noreclaim_restore(noreclaim_flag);
6524	trace_mm_vmscan_memcg_reclaim_end(nr_reclaimed);
6525	set_task_reclaim_state(current, NULL);
6526
6527	return nr_reclaimed;
6528}
6529#endif
6530
6531static void kswapd_age_node(struct pglist_data *pgdat, struct scan_control *sc)
6532{
6533	struct mem_cgroup *memcg;
6534	struct lruvec *lruvec;
6535
6536	if (lru_gen_enabled()) {
6537		lru_gen_age_node(pgdat, sc);
6538		return;
6539	}
6540
6541	if (!can_age_anon_pages(pgdat, sc))
6542		return;
6543
6544	lruvec = mem_cgroup_lruvec(NULL, pgdat);
6545	if (!inactive_is_low(lruvec, LRU_INACTIVE_ANON))
6546		return;
6547
6548	memcg = mem_cgroup_iter(NULL, NULL, NULL);
6549	do {
6550		lruvec = mem_cgroup_lruvec(memcg, pgdat);
6551		shrink_active_list(SWAP_CLUSTER_MAX, lruvec,
6552				   sc, LRU_ACTIVE_ANON);
6553		memcg = mem_cgroup_iter(NULL, memcg, NULL);
6554	} while (memcg);
6555}
6556
6557static bool pgdat_watermark_boosted(pg_data_t *pgdat, int highest_zoneidx)
6558{
6559	int i;
6560	struct zone *zone;
6561
6562	/*
6563	 * Check for watermark boosts top-down as the higher zones
6564	 * are more likely to be boosted. Both watermarks and boosts
6565	 * should not be checked at the same time as reclaim would
6566	 * start prematurely when there is no boosting and a lower
6567	 * zone is balanced.
6568	 */
6569	for (i = highest_zoneidx; i >= 0; i--) {
6570		zone = pgdat->node_zones + i;
6571		if (!managed_zone(zone))
6572			continue;
6573
6574		if (zone->watermark_boost)
6575			return true;
6576	}
6577
6578	return false;
6579}
6580
6581/*
6582 * Returns true if there is an eligible zone balanced for the request order
6583 * and highest_zoneidx
6584 */
6585static bool pgdat_balanced(pg_data_t *pgdat, int order, int highest_zoneidx)
6586{
6587	int i;
6588	unsigned long mark = -1;
6589	struct zone *zone;
6590
6591	/*
6592	 * Check watermarks bottom-up as lower zones are more likely to
6593	 * meet watermarks.
6594	 */
6595	for (i = 0; i <= highest_zoneidx; i++) {
6596		zone = pgdat->node_zones + i;
6597
6598		if (!managed_zone(zone))
6599			continue;
6600
6601		if (sysctl_numa_balancing_mode & NUMA_BALANCING_MEMORY_TIERING)
6602			mark = wmark_pages(zone, WMARK_PROMO);
6603		else
6604			mark = high_wmark_pages(zone);
6605		if (zone_watermark_ok_safe(zone, order, mark, highest_zoneidx))
6606			return true;
6607	}
6608
6609	/*
6610	 * If a node has no managed zone within highest_zoneidx, it does not
6611	 * need balancing by definition. This can happen if a zone-restricted
6612	 * allocation tries to wake a remote kswapd.
6613	 */
6614	if (mark == -1)
6615		return true;
6616
6617	return false;
6618}
6619
6620/* Clear pgdat state for congested, dirty or under writeback. */
6621static void clear_pgdat_congested(pg_data_t *pgdat)
6622{
6623	struct lruvec *lruvec = mem_cgroup_lruvec(NULL, pgdat);
6624
6625	clear_bit(LRUVEC_NODE_CONGESTED, &lruvec->flags);
6626	clear_bit(LRUVEC_CGROUP_CONGESTED, &lruvec->flags);
6627	clear_bit(PGDAT_DIRTY, &pgdat->flags);
6628	clear_bit(PGDAT_WRITEBACK, &pgdat->flags);
6629}
6630
6631/*
6632 * Prepare kswapd for sleeping. This verifies that there are no processes
6633 * waiting in throttle_direct_reclaim() and that watermarks have been met.
6634 *
6635 * Returns true if kswapd is ready to sleep
6636 */
6637static bool prepare_kswapd_sleep(pg_data_t *pgdat, int order,
6638				int highest_zoneidx)
6639{
6640	/*
6641	 * The throttled processes are normally woken up in balance_pgdat() as
6642	 * soon as allow_direct_reclaim() is true. But there is a potential
6643	 * race between when kswapd checks the watermarks and a process gets
6644	 * throttled. There is also a potential race if processes get
6645	 * throttled, kswapd wakes, a large process exits thereby balancing the
6646	 * zones, which causes kswapd to exit balance_pgdat() before reaching
6647	 * the wake up checks. If kswapd is going to sleep, no process should
6648	 * be sleeping on pfmemalloc_wait, so wake them now if necessary. If
6649	 * the wake up is premature, processes will wake kswapd and get
6650	 * throttled again. The difference from wake ups in balance_pgdat() is
6651	 * that here we are under prepare_to_wait().
6652	 */
6653	if (waitqueue_active(&pgdat->pfmemalloc_wait))
6654		wake_up_all(&pgdat->pfmemalloc_wait);
6655
6656	/* Hopeless node, leave it to direct reclaim */
6657	if (pgdat->kswapd_failures >= MAX_RECLAIM_RETRIES)
6658		return true;
6659
6660	if (pgdat_balanced(pgdat, order, highest_zoneidx)) {
6661		clear_pgdat_congested(pgdat);
6662		return true;
6663	}
6664
6665	return false;
6666}
6667
6668/*
6669 * kswapd shrinks a node of pages that are at or below the highest usable
6670 * zone that is currently unbalanced.
6671 *
6672 * Returns true if kswapd scanned at least the requested number of pages to
6673 * reclaim or if the lack of progress was due to pages under writeback.
6674 * This is used to determine if the scanning priority needs to be raised.
6675 */
6676static bool kswapd_shrink_node(pg_data_t *pgdat,
6677			       struct scan_control *sc)
6678{
6679	struct zone *zone;
6680	int z;
6681
6682	/* Reclaim a number of pages proportional to the number of zones */
6683	sc->nr_to_reclaim = 0;
6684	for (z = 0; z <= sc->reclaim_idx; z++) {
6685		zone = pgdat->node_zones + z;
6686		if (!managed_zone(zone))
6687			continue;
6688
6689		sc->nr_to_reclaim += max(high_wmark_pages(zone), SWAP_CLUSTER_MAX);
6690	}
6691
6692	/*
6693	 * Historically care was taken to put equal pressure on all zones but
6694	 * now pressure is applied based on node LRU order.
6695	 */
6696	shrink_node(pgdat, sc);
6697
6698	/*
6699	 * Fragmentation may mean that the system cannot be rebalanced for
6700	 * high-order allocations. If twice the allocation size has been
6701	 * reclaimed then recheck watermarks only at order-0 to prevent
6702	 * excessive reclaim. Assume that a process requested a high-order
6703	 * can direct reclaim/compact.
6704	 */
6705	if (sc->order && sc->nr_reclaimed >= compact_gap(sc->order))
6706		sc->order = 0;
6707
6708	return sc->nr_scanned >= sc->nr_to_reclaim;
6709}
6710
6711/* Page allocator PCP high watermark is lowered if reclaim is active. */
6712static inline void
6713update_reclaim_active(pg_data_t *pgdat, int highest_zoneidx, bool active)
6714{
6715	int i;
6716	struct zone *zone;
6717
6718	for (i = 0; i <= highest_zoneidx; i++) {
6719		zone = pgdat->node_zones + i;
6720
6721		if (!managed_zone(zone))
6722			continue;
6723
6724		if (active)
6725			set_bit(ZONE_RECLAIM_ACTIVE, &zone->flags);
6726		else
6727			clear_bit(ZONE_RECLAIM_ACTIVE, &zone->flags);
6728	}
6729}
6730
6731static inline void
6732set_reclaim_active(pg_data_t *pgdat, int highest_zoneidx)
6733{
6734	update_reclaim_active(pgdat, highest_zoneidx, true);
6735}
6736
6737static inline void
6738clear_reclaim_active(pg_data_t *pgdat, int highest_zoneidx)
6739{
6740	update_reclaim_active(pgdat, highest_zoneidx, false);
6741}
6742
6743/*
6744 * For kswapd, balance_pgdat() will reclaim pages across a node from zones
6745 * that are eligible for use by the caller until at least one zone is
6746 * balanced.
6747 *
6748 * Returns the order kswapd finished reclaiming at.
6749 *
6750 * kswapd scans the zones in the highmem->normal->dma direction.  It skips
6751 * zones which have free_pages > high_wmark_pages(zone), but once a zone is
6752 * found to have free_pages <= high_wmark_pages(zone), any page in that zone
6753 * or lower is eligible for reclaim until at least one usable zone is
6754 * balanced.
6755 */
6756static int balance_pgdat(pg_data_t *pgdat, int order, int highest_zoneidx)
6757{
6758	int i;
6759	unsigned long nr_soft_reclaimed;
6760	unsigned long nr_soft_scanned;
6761	unsigned long pflags;
6762	unsigned long nr_boost_reclaim;
6763	unsigned long zone_boosts[MAX_NR_ZONES] = { 0, };
6764	bool boosted;
6765	struct zone *zone;
6766	struct scan_control sc = {
6767		.gfp_mask = GFP_KERNEL,
6768		.order = order,
6769		.may_unmap = 1,
6770	};
6771
6772	set_task_reclaim_state(current, &sc.reclaim_state);
6773	psi_memstall_enter(&pflags);
6774	__fs_reclaim_acquire(_THIS_IP_);
6775
6776	count_vm_event(PAGEOUTRUN);
6777
6778	/*
6779	 * Account for the reclaim boost. Note that the zone boost is left in
6780	 * place so that parallel allocations that are near the watermark will
6781	 * stall or direct reclaim until kswapd is finished.
6782	 */
6783	nr_boost_reclaim = 0;
6784	for (i = 0; i <= highest_zoneidx; i++) {
6785		zone = pgdat->node_zones + i;
6786		if (!managed_zone(zone))
6787			continue;
6788
6789		nr_boost_reclaim += zone->watermark_boost;
6790		zone_boosts[i] = zone->watermark_boost;
6791	}
6792	boosted = nr_boost_reclaim;
6793
6794restart:
6795	set_reclaim_active(pgdat, highest_zoneidx);
6796	sc.priority = DEF_PRIORITY;
6797	do {
6798		unsigned long nr_reclaimed = sc.nr_reclaimed;
6799		bool raise_priority = true;
6800		bool balanced;
6801		bool ret;
 
6802
6803		sc.reclaim_idx = highest_zoneidx;
6804
6805		/*
6806		 * If the number of buffer_heads exceeds the maximum allowed
6807		 * then consider reclaiming from all zones. This has a dual
6808		 * purpose -- on 64-bit systems it is expected that
6809		 * buffer_heads are stripped during active rotation. On 32-bit
6810		 * systems, highmem pages can pin lowmem memory and shrinking
6811		 * buffers can relieve lowmem pressure. Reclaim may still not
6812		 * go ahead if all eligible zones for the original allocation
6813		 * request are balanced to avoid excessive reclaim from kswapd.
6814		 */
6815		if (buffer_heads_over_limit) {
6816			for (i = MAX_NR_ZONES - 1; i >= 0; i--) {
6817				zone = pgdat->node_zones + i;
6818				if (!managed_zone(zone))
6819					continue;
6820
6821				sc.reclaim_idx = i;
6822				break;
6823			}
6824		}
6825
6826		/*
6827		 * If the pgdat is imbalanced then ignore boosting and preserve
6828		 * the watermarks for a later time and restart. Note that the
6829		 * zone watermarks will be still reset at the end of balancing
6830		 * on the grounds that the normal reclaim should be enough to
6831		 * re-evaluate if boosting is required when kswapd next wakes.
6832		 */
6833		balanced = pgdat_balanced(pgdat, sc.order, highest_zoneidx);
6834		if (!balanced && nr_boost_reclaim) {
6835			nr_boost_reclaim = 0;
6836			goto restart;
6837		}
6838
6839		/*
6840		 * If boosting is not active then only reclaim if there are no
6841		 * eligible zones. Note that sc.reclaim_idx is not used as
6842		 * buffer_heads_over_limit may have adjusted it.
6843		 */
6844		if (!nr_boost_reclaim && balanced)
6845			goto out;
6846
6847		/* Limit the priority of boosting to avoid reclaim writeback */
6848		if (nr_boost_reclaim && sc.priority == DEF_PRIORITY - 2)
6849			raise_priority = false;
6850
6851		/*
6852		 * Do not writeback or swap pages for boosted reclaim. The
6853		 * intent is to relieve pressure not issue sub-optimal IO
6854		 * from reclaim context. If no pages are reclaimed, the
6855		 * reclaim will be aborted.
6856		 */
6857		sc.may_writepage = !laptop_mode && !nr_boost_reclaim;
6858		sc.may_swap = !nr_boost_reclaim;
6859
6860		/*
6861		 * Do some background aging, to give pages a chance to be
6862		 * referenced before reclaiming. All pages are rotated
6863		 * regardless of classzone as this is about consistent aging.
6864		 */
6865		kswapd_age_node(pgdat, &sc);
6866
6867		/*
6868		 * If we're getting trouble reclaiming, start doing writepage
6869		 * even in laptop mode.
6870		 */
6871		if (sc.priority < DEF_PRIORITY - 2)
6872			sc.may_writepage = 1;
6873
6874		/* Call soft limit reclaim before calling shrink_node. */
6875		sc.nr_scanned = 0;
6876		nr_soft_scanned = 0;
6877		nr_soft_reclaimed = mem_cgroup_soft_limit_reclaim(pgdat, sc.order,
6878						sc.gfp_mask, &nr_soft_scanned);
6879		sc.nr_reclaimed += nr_soft_reclaimed;
6880
6881		/*
6882		 * There should be no need to raise the scanning priority if
6883		 * enough pages are already being scanned that that high
6884		 * watermark would be met at 100% efficiency.
6885		 */
6886		if (kswapd_shrink_node(pgdat, &sc))
6887			raise_priority = false;
6888
6889		/*
6890		 * If the low watermark is met there is no need for processes
6891		 * to be throttled on pfmemalloc_wait as they should not be
6892		 * able to safely make forward progress. Wake them
6893		 */
6894		if (waitqueue_active(&pgdat->pfmemalloc_wait) &&
6895				allow_direct_reclaim(pgdat))
6896			wake_up_all(&pgdat->pfmemalloc_wait);
6897
6898		/* Check if kswapd should be suspending */
6899		__fs_reclaim_release(_THIS_IP_);
6900		ret = try_to_freeze();
6901		__fs_reclaim_acquire(_THIS_IP_);
6902		if (ret || kthread_should_stop())
6903			break;
6904
6905		/*
6906		 * Raise priority if scanning rate is too low or there was no
6907		 * progress in reclaiming pages
6908		 */
6909		nr_reclaimed = sc.nr_reclaimed - nr_reclaimed;
6910		nr_boost_reclaim -= min(nr_boost_reclaim, nr_reclaimed);
6911
6912		/*
6913		 * If reclaim made no progress for a boost, stop reclaim as
6914		 * IO cannot be queued and it could be an infinite loop in
6915		 * extreme circumstances.
6916		 */
6917		if (nr_boost_reclaim && !nr_reclaimed)
6918			break;
6919
6920		if (raise_priority || !nr_reclaimed)
6921			sc.priority--;
6922	} while (sc.priority >= 1);
6923
 
 
 
 
 
 
 
 
 
 
6924	if (!sc.nr_reclaimed)
6925		pgdat->kswapd_failures++;
6926
6927out:
6928	clear_reclaim_active(pgdat, highest_zoneidx);
6929
6930	/* If reclaim was boosted, account for the reclaim done in this pass */
6931	if (boosted) {
6932		unsigned long flags;
6933
6934		for (i = 0; i <= highest_zoneidx; i++) {
6935			if (!zone_boosts[i])
6936				continue;
6937
6938			/* Increments are under the zone lock */
6939			zone = pgdat->node_zones + i;
6940			spin_lock_irqsave(&zone->lock, flags);
6941			zone->watermark_boost -= min(zone->watermark_boost, zone_boosts[i]);
6942			spin_unlock_irqrestore(&zone->lock, flags);
6943		}
6944
6945		/*
6946		 * As there is now likely space, wakeup kcompact to defragment
6947		 * pageblocks.
6948		 */
6949		wakeup_kcompactd(pgdat, pageblock_order, highest_zoneidx);
6950	}
6951
6952	snapshot_refaults(NULL, pgdat);
6953	__fs_reclaim_release(_THIS_IP_);
6954	psi_memstall_leave(&pflags);
6955	set_task_reclaim_state(current, NULL);
6956
6957	/*
6958	 * Return the order kswapd stopped reclaiming at as
6959	 * prepare_kswapd_sleep() takes it into account. If another caller
6960	 * entered the allocator slow path while kswapd was awake, order will
6961	 * remain at the higher level.
6962	 */
6963	return sc.order;
6964}
6965
6966/*
6967 * The pgdat->kswapd_highest_zoneidx is used to pass the highest zone index to
6968 * be reclaimed by kswapd from the waker. If the value is MAX_NR_ZONES which is
6969 * not a valid index then either kswapd runs for first time or kswapd couldn't
6970 * sleep after previous reclaim attempt (node is still unbalanced). In that
6971 * case return the zone index of the previous kswapd reclaim cycle.
6972 */
6973static enum zone_type kswapd_highest_zoneidx(pg_data_t *pgdat,
6974					   enum zone_type prev_highest_zoneidx)
6975{
6976	enum zone_type curr_idx = READ_ONCE(pgdat->kswapd_highest_zoneidx);
6977
6978	return curr_idx == MAX_NR_ZONES ? prev_highest_zoneidx : curr_idx;
6979}
6980
6981static void kswapd_try_to_sleep(pg_data_t *pgdat, int alloc_order, int reclaim_order,
6982				unsigned int highest_zoneidx)
6983{
6984	long remaining = 0;
6985	DEFINE_WAIT(wait);
6986
6987	if (freezing(current) || kthread_should_stop())
6988		return;
6989
6990	prepare_to_wait(&pgdat->kswapd_wait, &wait, TASK_INTERRUPTIBLE);
6991
6992	/*
6993	 * Try to sleep for a short interval. Note that kcompactd will only be
6994	 * woken if it is possible to sleep for a short interval. This is
6995	 * deliberate on the assumption that if reclaim cannot keep an
6996	 * eligible zone balanced that it's also unlikely that compaction will
6997	 * succeed.
6998	 */
6999	if (prepare_kswapd_sleep(pgdat, reclaim_order, highest_zoneidx)) {
7000		/*
7001		 * Compaction records what page blocks it recently failed to
7002		 * isolate pages from and skips them in the future scanning.
7003		 * When kswapd is going to sleep, it is reasonable to assume
7004		 * that pages and compaction may succeed so reset the cache.
7005		 */
7006		reset_isolation_suitable(pgdat);
7007
7008		/*
7009		 * We have freed the memory, now we should compact it to make
7010		 * allocation of the requested order possible.
7011		 */
7012		wakeup_kcompactd(pgdat, alloc_order, highest_zoneidx);
7013
7014		remaining = schedule_timeout(HZ/10);
7015
7016		/*
7017		 * If woken prematurely then reset kswapd_highest_zoneidx and
7018		 * order. The values will either be from a wakeup request or
7019		 * the previous request that slept prematurely.
7020		 */
7021		if (remaining) {
7022			WRITE_ONCE(pgdat->kswapd_highest_zoneidx,
7023					kswapd_highest_zoneidx(pgdat,
7024							highest_zoneidx));
7025
7026			if (READ_ONCE(pgdat->kswapd_order) < reclaim_order)
7027				WRITE_ONCE(pgdat->kswapd_order, reclaim_order);
7028		}
7029
7030		finish_wait(&pgdat->kswapd_wait, &wait);
7031		prepare_to_wait(&pgdat->kswapd_wait, &wait, TASK_INTERRUPTIBLE);
7032	}
7033
7034	/*
7035	 * After a short sleep, check if it was a premature sleep. If not, then
7036	 * go fully to sleep until explicitly woken up.
7037	 */
7038	if (!remaining &&
7039	    prepare_kswapd_sleep(pgdat, reclaim_order, highest_zoneidx)) {
7040		trace_mm_vmscan_kswapd_sleep(pgdat->node_id);
7041
7042		/*
7043		 * vmstat counters are not perfectly accurate and the estimated
7044		 * value for counters such as NR_FREE_PAGES can deviate from the
7045		 * true value by nr_online_cpus * threshold. To avoid the zone
7046		 * watermarks being breached while under pressure, we reduce the
7047		 * per-cpu vmstat threshold while kswapd is awake and restore
7048		 * them before going back to sleep.
7049		 */
7050		set_pgdat_percpu_threshold(pgdat, calculate_normal_threshold);
7051
7052		if (!kthread_should_stop())
7053			schedule();
7054
7055		set_pgdat_percpu_threshold(pgdat, calculate_pressure_threshold);
7056	} else {
7057		if (remaining)
7058			count_vm_event(KSWAPD_LOW_WMARK_HIT_QUICKLY);
7059		else
7060			count_vm_event(KSWAPD_HIGH_WMARK_HIT_QUICKLY);
7061	}
7062	finish_wait(&pgdat->kswapd_wait, &wait);
7063}
7064
7065/*
7066 * The background pageout daemon, started as a kernel thread
7067 * from the init process.
7068 *
7069 * This basically trickles out pages so that we have _some_
7070 * free memory available even if there is no other activity
7071 * that frees anything up. This is needed for things like routing
7072 * etc, where we otherwise might have all activity going on in
7073 * asynchronous contexts that cannot page things out.
7074 *
7075 * If there are applications that are active memory-allocators
7076 * (most normal use), this basically shouldn't matter.
7077 */
7078static int kswapd(void *p)
7079{
7080	unsigned int alloc_order, reclaim_order;
7081	unsigned int highest_zoneidx = MAX_NR_ZONES - 1;
7082	pg_data_t *pgdat = (pg_data_t *)p;
7083	struct task_struct *tsk = current;
7084	const struct cpumask *cpumask = cpumask_of_node(pgdat->node_id);
7085
7086	if (!cpumask_empty(cpumask))
7087		set_cpus_allowed_ptr(tsk, cpumask);
7088
7089	/*
7090	 * Tell the memory management that we're a "memory allocator",
7091	 * and that if we need more memory we should get access to it
7092	 * regardless (see "__alloc_pages()"). "kswapd" should
7093	 * never get caught in the normal page freeing logic.
7094	 *
7095	 * (Kswapd normally doesn't need memory anyway, but sometimes
7096	 * you need a small amount of memory in order to be able to
7097	 * page out something else, and this flag essentially protects
7098	 * us from recursively trying to free more memory as we're
7099	 * trying to free the first piece of memory in the first place).
7100	 */
7101	tsk->flags |= PF_MEMALLOC | PF_KSWAPD;
7102	set_freezable();
7103
7104	WRITE_ONCE(pgdat->kswapd_order, 0);
7105	WRITE_ONCE(pgdat->kswapd_highest_zoneidx, MAX_NR_ZONES);
7106	atomic_set(&pgdat->nr_writeback_throttled, 0);
7107	for ( ; ; ) {
7108		bool ret;
7109
7110		alloc_order = reclaim_order = READ_ONCE(pgdat->kswapd_order);
7111		highest_zoneidx = kswapd_highest_zoneidx(pgdat,
7112							highest_zoneidx);
7113
7114kswapd_try_sleep:
7115		kswapd_try_to_sleep(pgdat, alloc_order, reclaim_order,
7116					highest_zoneidx);
7117
7118		/* Read the new order and highest_zoneidx */
7119		alloc_order = READ_ONCE(pgdat->kswapd_order);
7120		highest_zoneidx = kswapd_highest_zoneidx(pgdat,
7121							highest_zoneidx);
7122		WRITE_ONCE(pgdat->kswapd_order, 0);
7123		WRITE_ONCE(pgdat->kswapd_highest_zoneidx, MAX_NR_ZONES);
7124
7125		ret = try_to_freeze();
7126		if (kthread_should_stop())
7127			break;
7128
7129		/*
7130		 * We can speed up thawing tasks if we don't call balance_pgdat
7131		 * after returning from the refrigerator
7132		 */
7133		if (ret)
7134			continue;
7135
7136		/*
7137		 * Reclaim begins at the requested order but if a high-order
7138		 * reclaim fails then kswapd falls back to reclaiming for
7139		 * order-0. If that happens, kswapd will consider sleeping
7140		 * for the order it finished reclaiming at (reclaim_order)
7141		 * but kcompactd is woken to compact for the original
7142		 * request (alloc_order).
7143		 */
7144		trace_mm_vmscan_kswapd_wake(pgdat->node_id, highest_zoneidx,
7145						alloc_order);
7146		reclaim_order = balance_pgdat(pgdat, alloc_order,
7147						highest_zoneidx);
7148		if (reclaim_order < alloc_order)
7149			goto kswapd_try_sleep;
7150	}
7151
7152	tsk->flags &= ~(PF_MEMALLOC | PF_KSWAPD);
7153
7154	return 0;
7155}
7156
7157/*
7158 * A zone is low on free memory or too fragmented for high-order memory.  If
7159 * kswapd should reclaim (direct reclaim is deferred), wake it up for the zone's
7160 * pgdat.  It will wake up kcompactd after reclaiming memory.  If kswapd reclaim
7161 * has failed or is not needed, still wake up kcompactd if only compaction is
7162 * needed.
7163 */
7164void wakeup_kswapd(struct zone *zone, gfp_t gfp_flags, int order,
7165		   enum zone_type highest_zoneidx)
7166{
7167	pg_data_t *pgdat;
7168	enum zone_type curr_idx;
7169
7170	if (!managed_zone(zone))
7171		return;
7172
7173	if (!cpuset_zone_allowed(zone, gfp_flags))
7174		return;
7175
7176	pgdat = zone->zone_pgdat;
7177	curr_idx = READ_ONCE(pgdat->kswapd_highest_zoneidx);
7178
7179	if (curr_idx == MAX_NR_ZONES || curr_idx < highest_zoneidx)
7180		WRITE_ONCE(pgdat->kswapd_highest_zoneidx, highest_zoneidx);
7181
7182	if (READ_ONCE(pgdat->kswapd_order) < order)
7183		WRITE_ONCE(pgdat->kswapd_order, order);
7184
7185	if (!waitqueue_active(&pgdat->kswapd_wait))
7186		return;
7187
7188	/* Hopeless node, leave it to direct reclaim if possible */
7189	if (pgdat->kswapd_failures >= MAX_RECLAIM_RETRIES ||
7190	    (pgdat_balanced(pgdat, order, highest_zoneidx) &&
7191	     !pgdat_watermark_boosted(pgdat, highest_zoneidx))) {
7192		/*
7193		 * There may be plenty of free memory available, but it's too
7194		 * fragmented for high-order allocations.  Wake up kcompactd
7195		 * and rely on compaction_suitable() to determine if it's
7196		 * needed.  If it fails, it will defer subsequent attempts to
7197		 * ratelimit its work.
7198		 */
7199		if (!(gfp_flags & __GFP_DIRECT_RECLAIM))
7200			wakeup_kcompactd(pgdat, order, highest_zoneidx);
7201		return;
7202	}
7203
7204	trace_mm_vmscan_wakeup_kswapd(pgdat->node_id, highest_zoneidx, order,
7205				      gfp_flags);
7206	wake_up_interruptible(&pgdat->kswapd_wait);
7207}
7208
7209#ifdef CONFIG_HIBERNATION
7210/*
7211 * Try to free `nr_to_reclaim' of memory, system-wide, and return the number of
7212 * freed pages.
7213 *
7214 * Rather than trying to age LRUs the aim is to preserve the overall
7215 * LRU order by reclaiming preferentially
7216 * inactive > active > active referenced > active mapped
7217 */
7218unsigned long shrink_all_memory(unsigned long nr_to_reclaim)
7219{
7220	struct scan_control sc = {
7221		.nr_to_reclaim = nr_to_reclaim,
7222		.gfp_mask = GFP_HIGHUSER_MOVABLE,
7223		.reclaim_idx = MAX_NR_ZONES - 1,
7224		.priority = DEF_PRIORITY,
7225		.may_writepage = 1,
7226		.may_unmap = 1,
7227		.may_swap = 1,
7228		.hibernation_mode = 1,
7229	};
7230	struct zonelist *zonelist = node_zonelist(numa_node_id(), sc.gfp_mask);
7231	unsigned long nr_reclaimed;
7232	unsigned int noreclaim_flag;
7233
7234	fs_reclaim_acquire(sc.gfp_mask);
7235	noreclaim_flag = memalloc_noreclaim_save();
7236	set_task_reclaim_state(current, &sc.reclaim_state);
7237
7238	nr_reclaimed = do_try_to_free_pages(zonelist, &sc);
7239
7240	set_task_reclaim_state(current, NULL);
7241	memalloc_noreclaim_restore(noreclaim_flag);
7242	fs_reclaim_release(sc.gfp_mask);
7243
7244	return nr_reclaimed;
7245}
7246#endif /* CONFIG_HIBERNATION */
7247
7248/*
7249 * This kswapd start function will be called by init and node-hot-add.
7250 */
7251void __meminit kswapd_run(int nid)
7252{
7253	pg_data_t *pgdat = NODE_DATA(nid);
7254
7255	pgdat_kswapd_lock(pgdat);
7256	if (!pgdat->kswapd) {
7257		pgdat->kswapd = kthread_run(kswapd, pgdat, "kswapd%d", nid);
7258		if (IS_ERR(pgdat->kswapd)) {
7259			/* failure at boot is fatal */
7260			pr_err("Failed to start kswapd on node %d,ret=%ld\n",
7261				   nid, PTR_ERR(pgdat->kswapd));
7262			BUG_ON(system_state < SYSTEM_RUNNING);
7263			pgdat->kswapd = NULL;
7264		}
7265	}
7266	pgdat_kswapd_unlock(pgdat);
7267}
7268
7269/*
7270 * Called by memory hotplug when all memory in a node is offlined.  Caller must
7271 * be holding mem_hotplug_begin/done().
7272 */
7273void __meminit kswapd_stop(int nid)
7274{
7275	pg_data_t *pgdat = NODE_DATA(nid);
7276	struct task_struct *kswapd;
7277
7278	pgdat_kswapd_lock(pgdat);
7279	kswapd = pgdat->kswapd;
7280	if (kswapd) {
7281		kthread_stop(kswapd);
7282		pgdat->kswapd = NULL;
7283	}
7284	pgdat_kswapd_unlock(pgdat);
7285}
7286
7287static int __init kswapd_init(void)
7288{
7289	int nid;
7290
7291	swap_setup();
7292	for_each_node_state(nid, N_MEMORY)
7293 		kswapd_run(nid);
7294	return 0;
7295}
7296
7297module_init(kswapd_init)
7298
7299#ifdef CONFIG_NUMA
7300/*
7301 * Node reclaim mode
7302 *
7303 * If non-zero call node_reclaim when the number of free pages falls below
7304 * the watermarks.
7305 */
7306int node_reclaim_mode __read_mostly;
7307
7308/*
7309 * Priority for NODE_RECLAIM. This determines the fraction of pages
7310 * of a node considered for each zone_reclaim. 4 scans 1/16th of
7311 * a zone.
7312 */
7313#define NODE_RECLAIM_PRIORITY 4
7314
7315/*
7316 * Percentage of pages in a zone that must be unmapped for node_reclaim to
7317 * occur.
7318 */
7319int sysctl_min_unmapped_ratio = 1;
7320
7321/*
7322 * If the number of slab pages in a zone grows beyond this percentage then
7323 * slab reclaim needs to occur.
7324 */
7325int sysctl_min_slab_ratio = 5;
7326
7327static inline unsigned long node_unmapped_file_pages(struct pglist_data *pgdat)
7328{
7329	unsigned long file_mapped = node_page_state(pgdat, NR_FILE_MAPPED);
7330	unsigned long file_lru = node_page_state(pgdat, NR_INACTIVE_FILE) +
7331		node_page_state(pgdat, NR_ACTIVE_FILE);
7332
7333	/*
7334	 * It's possible for there to be more file mapped pages than
7335	 * accounted for by the pages on the file LRU lists because
7336	 * tmpfs pages accounted for as ANON can also be FILE_MAPPED
7337	 */
7338	return (file_lru > file_mapped) ? (file_lru - file_mapped) : 0;
7339}
7340
7341/* Work out how many page cache pages we can reclaim in this reclaim_mode */
7342static unsigned long node_pagecache_reclaimable(struct pglist_data *pgdat)
7343{
7344	unsigned long nr_pagecache_reclaimable;
7345	unsigned long delta = 0;
7346
7347	/*
7348	 * If RECLAIM_UNMAP is set, then all file pages are considered
7349	 * potentially reclaimable. Otherwise, we have to worry about
7350	 * pages like swapcache and node_unmapped_file_pages() provides
7351	 * a better estimate
7352	 */
7353	if (node_reclaim_mode & RECLAIM_UNMAP)
7354		nr_pagecache_reclaimable = node_page_state(pgdat, NR_FILE_PAGES);
7355	else
7356		nr_pagecache_reclaimable = node_unmapped_file_pages(pgdat);
7357
7358	/* If we can't clean pages, remove dirty pages from consideration */
7359	if (!(node_reclaim_mode & RECLAIM_WRITE))
7360		delta += node_page_state(pgdat, NR_FILE_DIRTY);
7361
7362	/* Watch for any possible underflows due to delta */
7363	if (unlikely(delta > nr_pagecache_reclaimable))
7364		delta = nr_pagecache_reclaimable;
7365
7366	return nr_pagecache_reclaimable - delta;
7367}
7368
7369/*
7370 * Try to free up some pages from this node through reclaim.
7371 */
7372static int __node_reclaim(struct pglist_data *pgdat, gfp_t gfp_mask, unsigned int order)
7373{
7374	/* Minimum pages needed in order to stay on node */
7375	const unsigned long nr_pages = 1 << order;
7376	struct task_struct *p = current;
7377	unsigned int noreclaim_flag;
7378	struct scan_control sc = {
7379		.nr_to_reclaim = max(nr_pages, SWAP_CLUSTER_MAX),
7380		.gfp_mask = current_gfp_context(gfp_mask),
7381		.order = order,
7382		.priority = NODE_RECLAIM_PRIORITY,
7383		.may_writepage = !!(node_reclaim_mode & RECLAIM_WRITE),
7384		.may_unmap = !!(node_reclaim_mode & RECLAIM_UNMAP),
7385		.may_swap = 1,
7386		.reclaim_idx = gfp_zone(gfp_mask),
7387	};
7388	unsigned long pflags;
7389
7390	trace_mm_vmscan_node_reclaim_begin(pgdat->node_id, order,
7391					   sc.gfp_mask);
7392
7393	cond_resched();
7394	psi_memstall_enter(&pflags);
7395	delayacct_freepages_start();
7396	fs_reclaim_acquire(sc.gfp_mask);
7397	/*
7398	 * We need to be able to allocate from the reserves for RECLAIM_UNMAP
7399	 */
7400	noreclaim_flag = memalloc_noreclaim_save();
7401	set_task_reclaim_state(p, &sc.reclaim_state);
7402
7403	if (node_pagecache_reclaimable(pgdat) > pgdat->min_unmapped_pages ||
7404	    node_page_state_pages(pgdat, NR_SLAB_RECLAIMABLE_B) > pgdat->min_slab_pages) {
7405		/*
7406		 * Free memory by calling shrink node with increasing
7407		 * priorities until we have enough memory freed.
7408		 */
7409		do {
7410			shrink_node(pgdat, &sc);
7411		} while (sc.nr_reclaimed < nr_pages && --sc.priority >= 0);
7412	}
7413
7414	set_task_reclaim_state(p, NULL);
7415	memalloc_noreclaim_restore(noreclaim_flag);
7416	fs_reclaim_release(sc.gfp_mask);
7417	psi_memstall_leave(&pflags);
7418	delayacct_freepages_end();
7419
7420	trace_mm_vmscan_node_reclaim_end(sc.nr_reclaimed);
7421
7422	return sc.nr_reclaimed >= nr_pages;
7423}
7424
7425int node_reclaim(struct pglist_data *pgdat, gfp_t gfp_mask, unsigned int order)
7426{
7427	int ret;
7428
7429	/*
7430	 * Node reclaim reclaims unmapped file backed pages and
7431	 * slab pages if we are over the defined limits.
7432	 *
7433	 * A small portion of unmapped file backed pages is needed for
7434	 * file I/O otherwise pages read by file I/O will be immediately
7435	 * thrown out if the node is overallocated. So we do not reclaim
7436	 * if less than a specified percentage of the node is used by
7437	 * unmapped file backed pages.
7438	 */
7439	if (node_pagecache_reclaimable(pgdat) <= pgdat->min_unmapped_pages &&
7440	    node_page_state_pages(pgdat, NR_SLAB_RECLAIMABLE_B) <=
7441	    pgdat->min_slab_pages)
7442		return NODE_RECLAIM_FULL;
7443
7444	/*
7445	 * Do not scan if the allocation should not be delayed.
7446	 */
7447	if (!gfpflags_allow_blocking(gfp_mask) || (current->flags & PF_MEMALLOC))
7448		return NODE_RECLAIM_NOSCAN;
7449
7450	/*
7451	 * Only run node reclaim on the local node or on nodes that do not
7452	 * have associated processors. This will favor the local processor
7453	 * over remote processors and spread off node memory allocations
7454	 * as wide as possible.
7455	 */
7456	if (node_state(pgdat->node_id, N_CPU) && pgdat->node_id != numa_node_id())
7457		return NODE_RECLAIM_NOSCAN;
7458
7459	if (test_and_set_bit(PGDAT_RECLAIM_LOCKED, &pgdat->flags))
7460		return NODE_RECLAIM_NOSCAN;
7461
7462	ret = __node_reclaim(pgdat, gfp_mask, order);
7463	clear_bit(PGDAT_RECLAIM_LOCKED, &pgdat->flags);
7464
7465	if (!ret)
7466		count_vm_event(PGSCAN_ZONE_RECLAIM_FAILED);
7467
7468	return ret;
7469}
7470#endif
7471
7472/**
7473 * check_move_unevictable_folios - Move evictable folios to appropriate zone
7474 * lru list
7475 * @fbatch: Batch of lru folios to check.
7476 *
7477 * Checks folios for evictability, if an evictable folio is in the unevictable
7478 * lru list, moves it to the appropriate evictable lru list. This function
7479 * should be only used for lru folios.
7480 */
7481void check_move_unevictable_folios(struct folio_batch *fbatch)
7482{
7483	struct lruvec *lruvec = NULL;
7484	int pgscanned = 0;
7485	int pgrescued = 0;
7486	int i;
7487
7488	for (i = 0; i < fbatch->nr; i++) {
7489		struct folio *folio = fbatch->folios[i];
7490		int nr_pages = folio_nr_pages(folio);
7491
7492		pgscanned += nr_pages;
7493
7494		/* block memcg migration while the folio moves between lrus */
7495		if (!folio_test_clear_lru(folio))
7496			continue;
7497
7498		lruvec = folio_lruvec_relock_irq(folio, lruvec);
7499		if (folio_evictable(folio) && folio_test_unevictable(folio)) {
7500			lruvec_del_folio(lruvec, folio);
7501			folio_clear_unevictable(folio);
7502			lruvec_add_folio(lruvec, folio);
7503			pgrescued += nr_pages;
7504		}
7505		folio_set_lru(folio);
7506	}
7507
7508	if (lruvec) {
7509		__count_vm_events(UNEVICTABLE_PGRESCUED, pgrescued);
7510		__count_vm_events(UNEVICTABLE_PGSCANNED, pgscanned);
7511		unlock_page_lruvec_irq(lruvec);
7512	} else if (pgscanned) {
7513		count_vm_events(UNEVICTABLE_PGSCANNED, pgscanned);
7514	}
7515}
7516EXPORT_SYMBOL_GPL(check_move_unevictable_folios);
v6.9.4
   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/delay.h>
  39#include <linux/kthread.h>
  40#include <linux/freezer.h>
  41#include <linux/memcontrol.h>
  42#include <linux/migrate.h>
  43#include <linux/delayacct.h>
  44#include <linux/sysctl.h>
  45#include <linux/memory-tiers.h>
  46#include <linux/oom.h>
  47#include <linux/pagevec.h>
  48#include <linux/prefetch.h>
  49#include <linux/printk.h>
  50#include <linux/dax.h>
  51#include <linux/psi.h>
  52#include <linux/pagewalk.h>
  53#include <linux/shmem_fs.h>
  54#include <linux/ctype.h>
  55#include <linux/debugfs.h>
  56#include <linux/khugepaged.h>
  57#include <linux/rculist_nulls.h>
  58#include <linux/random.h>
  59
  60#include <asm/tlbflush.h>
  61#include <asm/div64.h>
  62
  63#include <linux/swapops.h>
  64#include <linux/balloon_compaction.h>
  65#include <linux/sched/sysctl.h>
  66
  67#include "internal.h"
  68#include "swap.h"
  69
  70#define CREATE_TRACE_POINTS
  71#include <trace/events/vmscan.h>
  72
  73struct scan_control {
  74	/* How many pages shrink_list() should reclaim */
  75	unsigned long nr_to_reclaim;
  76
  77	/*
  78	 * Nodemask of nodes allowed by the caller. If NULL, all nodes
  79	 * are scanned.
  80	 */
  81	nodemask_t	*nodemask;
  82
  83	/*
  84	 * The memory cgroup that hit its limit and as a result is the
  85	 * primary target of this reclaim invocation.
  86	 */
  87	struct mem_cgroup *target_mem_cgroup;
  88
  89	/*
  90	 * Scan pressure balancing between anon and file LRUs
  91	 */
  92	unsigned long	anon_cost;
  93	unsigned long	file_cost;
  94
  95	/* Can active folios be deactivated as part of reclaim? */
  96#define DEACTIVATE_ANON 1
  97#define DEACTIVATE_FILE 2
  98	unsigned int may_deactivate:2;
  99	unsigned int force_deactivate:1;
 100	unsigned int skipped_deactivate:1;
 101
 102	/* Writepage batching in laptop mode; RECLAIM_WRITE */
 103	unsigned int may_writepage:1;
 104
 105	/* Can mapped folios be reclaimed? */
 106	unsigned int may_unmap:1;
 107
 108	/* Can folios be swapped as part of reclaim? */
 109	unsigned int may_swap:1;
 110
 111	/* Not allow cache_trim_mode to be turned on as part of reclaim? */
 112	unsigned int no_cache_trim_mode:1;
 113
 114	/* Has cache_trim_mode failed at least once? */
 115	unsigned int cache_trim_mode_failed:1;
 116
 117	/* Proactive reclaim invoked by userspace through memory.reclaim */
 118	unsigned int proactive:1;
 119
 120	/*
 121	 * Cgroup memory below memory.low is protected as long as we
 122	 * don't threaten to OOM. If any cgroup is reclaimed at
 123	 * reduced force or passed over entirely due to its memory.low
 124	 * setting (memcg_low_skipped), and nothing is reclaimed as a
 125	 * result, then go back for one more cycle that reclaims the protected
 126	 * memory (memcg_low_reclaim) to avert OOM.
 127	 */
 128	unsigned int memcg_low_reclaim:1;
 129	unsigned int memcg_low_skipped:1;
 130
 131	unsigned int hibernation_mode:1;
 132
 133	/* One of the zones is ready for compaction */
 134	unsigned int compaction_ready:1;
 135
 136	/* There is easily reclaimable cold cache in the current node */
 137	unsigned int cache_trim_mode:1;
 138
 139	/* The file folios on the current node are dangerously low */
 140	unsigned int file_is_tiny:1;
 141
 142	/* Always discard instead of demoting to lower tier memory */
 143	unsigned int no_demotion:1;
 144
 145	/* Allocation order */
 146	s8 order;
 147
 148	/* Scan (total_size >> priority) pages at once */
 149	s8 priority;
 150
 151	/* The highest zone to isolate folios for reclaim from */
 152	s8 reclaim_idx;
 153
 154	/* This context's GFP mask */
 155	gfp_t gfp_mask;
 156
 157	/* Incremented by the number of inactive pages that were scanned */
 158	unsigned long nr_scanned;
 159
 160	/* Number of pages freed so far during a call to shrink_zones() */
 161	unsigned long nr_reclaimed;
 162
 163	struct {
 164		unsigned int dirty;
 165		unsigned int unqueued_dirty;
 166		unsigned int congested;
 167		unsigned int writeback;
 168		unsigned int immediate;
 169		unsigned int file_taken;
 170		unsigned int taken;
 171	} nr;
 172
 173	/* for recording the reclaimed slab by now */
 174	struct reclaim_state reclaim_state;
 175};
 176
 177#ifdef ARCH_HAS_PREFETCHW
 178#define prefetchw_prev_lru_folio(_folio, _base, _field)			\
 179	do {								\
 180		if ((_folio)->lru.prev != _base) {			\
 181			struct folio *prev;				\
 182									\
 183			prev = lru_to_folio(&(_folio->lru));		\
 184			prefetchw(&prev->_field);			\
 185		}							\
 186	} while (0)
 187#else
 188#define prefetchw_prev_lru_folio(_folio, _base, _field) do { } while (0)
 189#endif
 190
 191/*
 192 * From 0 .. 200.  Higher means more swappy.
 193 */
 194int vm_swappiness = 60;
 195
 196#ifdef CONFIG_MEMCG
 197
 198/* Returns true for reclaim through cgroup limits or cgroup interfaces. */
 199static bool cgroup_reclaim(struct scan_control *sc)
 200{
 201	return sc->target_mem_cgroup;
 202}
 203
 204/*
 205 * Returns true for reclaim on the root cgroup. This is true for direct
 206 * allocator reclaim and reclaim through cgroup interfaces on the root cgroup.
 207 */
 208static bool root_reclaim(struct scan_control *sc)
 209{
 210	return !sc->target_mem_cgroup || mem_cgroup_is_root(sc->target_mem_cgroup);
 211}
 212
 213/**
 214 * writeback_throttling_sane - is the usual dirty throttling mechanism available?
 215 * @sc: scan_control in question
 216 *
 217 * The normal page dirty throttling mechanism in balance_dirty_pages() is
 218 * completely broken with the legacy memcg and direct stalling in
 219 * shrink_folio_list() is used for throttling instead, which lacks all the
 220 * niceties such as fairness, adaptive pausing, bandwidth proportional
 221 * allocation and configurability.
 222 *
 223 * This function tests whether the vmscan currently in progress can assume
 224 * that the normal dirty throttling mechanism is operational.
 225 */
 226static bool writeback_throttling_sane(struct scan_control *sc)
 227{
 228	if (!cgroup_reclaim(sc))
 229		return true;
 230#ifdef CONFIG_CGROUP_WRITEBACK
 231	if (cgroup_subsys_on_dfl(memory_cgrp_subsys))
 232		return true;
 233#endif
 234	return false;
 235}
 236#else
 237static bool cgroup_reclaim(struct scan_control *sc)
 238{
 239	return false;
 240}
 241
 242static bool root_reclaim(struct scan_control *sc)
 243{
 244	return true;
 245}
 246
 247static bool writeback_throttling_sane(struct scan_control *sc)
 248{
 249	return true;
 250}
 251#endif
 252
 253static void set_task_reclaim_state(struct task_struct *task,
 254				   struct reclaim_state *rs)
 255{
 256	/* Check for an overwrite */
 257	WARN_ON_ONCE(rs && task->reclaim_state);
 258
 259	/* Check for the nulling of an already-nulled member */
 260	WARN_ON_ONCE(!rs && !task->reclaim_state);
 261
 262	task->reclaim_state = rs;
 263}
 264
 265/*
 266 * flush_reclaim_state(): add pages reclaimed outside of LRU-based reclaim to
 267 * scan_control->nr_reclaimed.
 268 */
 269static void flush_reclaim_state(struct scan_control *sc)
 270{
 271	/*
 272	 * Currently, reclaim_state->reclaimed includes three types of pages
 273	 * freed outside of vmscan:
 274	 * (1) Slab pages.
 275	 * (2) Clean file pages from pruned inodes (on highmem systems).
 276	 * (3) XFS freed buffer pages.
 277	 *
 278	 * For all of these cases, we cannot universally link the pages to a
 279	 * single memcg. For example, a memcg-aware shrinker can free one object
 280	 * charged to the target memcg, causing an entire page to be freed.
 281	 * If we count the entire page as reclaimed from the memcg, we end up
 282	 * overestimating the reclaimed amount (potentially under-reclaiming).
 283	 *
 284	 * Only count such pages for global reclaim to prevent under-reclaiming
 285	 * from the target memcg; preventing unnecessary retries during memcg
 286	 * charging and false positives from proactive reclaim.
 287	 *
 288	 * For uncommon cases where the freed pages were actually mostly
 289	 * charged to the target memcg, we end up underestimating the reclaimed
 290	 * amount. This should be fine. The freed pages will be uncharged
 291	 * anyway, even if they are not counted here properly, and we will be
 292	 * able to make forward progress in charging (which is usually in a
 293	 * retry loop).
 294	 *
 295	 * We can go one step further, and report the uncharged objcg pages in
 296	 * memcg reclaim, to make reporting more accurate and reduce
 297	 * underestimation, but it's probably not worth the complexity for now.
 298	 */
 299	if (current->reclaim_state && root_reclaim(sc)) {
 300		sc->nr_reclaimed += current->reclaim_state->reclaimed;
 301		current->reclaim_state->reclaimed = 0;
 302	}
 303}
 304
 305static bool can_demote(int nid, struct scan_control *sc)
 306{
 307	if (!numa_demotion_enabled)
 308		return false;
 309	if (sc && sc->no_demotion)
 310		return false;
 311	if (next_demotion_node(nid) == NUMA_NO_NODE)
 312		return false;
 313
 314	return true;
 315}
 316
 317static inline bool can_reclaim_anon_pages(struct mem_cgroup *memcg,
 318					  int nid,
 319					  struct scan_control *sc)
 320{
 321	if (memcg == NULL) {
 322		/*
 323		 * For non-memcg reclaim, is there
 324		 * space in any swap device?
 325		 */
 326		if (get_nr_swap_pages() > 0)
 327			return true;
 328	} else {
 329		/* Is the memcg below its swap limit? */
 330		if (mem_cgroup_get_nr_swap_pages(memcg) > 0)
 331			return true;
 332	}
 333
 334	/*
 335	 * The page can not be swapped.
 336	 *
 337	 * Can it be reclaimed from this node via demotion?
 338	 */
 339	return can_demote(nid, sc);
 340}
 341
 342/*
 343 * This misses isolated folios which are not accounted for to save counters.
 344 * As the data only determines if reclaim or compaction continues, it is
 345 * not expected that isolated folios will be a dominating factor.
 346 */
 347unsigned long zone_reclaimable_pages(struct zone *zone)
 348{
 349	unsigned long nr;
 350
 351	nr = zone_page_state_snapshot(zone, NR_ZONE_INACTIVE_FILE) +
 352		zone_page_state_snapshot(zone, NR_ZONE_ACTIVE_FILE);
 353	if (can_reclaim_anon_pages(NULL, zone_to_nid(zone), NULL))
 354		nr += zone_page_state_snapshot(zone, NR_ZONE_INACTIVE_ANON) +
 355			zone_page_state_snapshot(zone, NR_ZONE_ACTIVE_ANON);
 356
 357	return nr;
 358}
 359
 360/**
 361 * lruvec_lru_size -  Returns the number of pages on the given LRU list.
 362 * @lruvec: lru vector
 363 * @lru: lru to use
 364 * @zone_idx: zones to consider (use MAX_NR_ZONES - 1 for the whole LRU list)
 365 */
 366static unsigned long lruvec_lru_size(struct lruvec *lruvec, enum lru_list lru,
 367				     int zone_idx)
 368{
 369	unsigned long size = 0;
 370	int zid;
 371
 372	for (zid = 0; zid <= zone_idx; zid++) {
 373		struct zone *zone = &lruvec_pgdat(lruvec)->node_zones[zid];
 374
 375		if (!managed_zone(zone))
 376			continue;
 377
 378		if (!mem_cgroup_disabled())
 379			size += mem_cgroup_get_zone_lru_size(lruvec, lru, zid);
 380		else
 381			size += zone_page_state(zone, NR_ZONE_LRU_BASE + lru);
 382	}
 383	return size;
 384}
 385
 386static unsigned long drop_slab_node(int nid)
 387{
 388	unsigned long freed = 0;
 389	struct mem_cgroup *memcg = NULL;
 390
 391	memcg = mem_cgroup_iter(NULL, NULL, NULL);
 392	do {
 393		freed += shrink_slab(GFP_KERNEL, nid, memcg, 0);
 394	} while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)) != NULL);
 395
 396	return freed;
 397}
 398
 399void drop_slab(void)
 400{
 401	int nid;
 402	int shift = 0;
 403	unsigned long freed;
 404
 405	do {
 406		freed = 0;
 407		for_each_online_node(nid) {
 408			if (fatal_signal_pending(current))
 409				return;
 410
 411			freed += drop_slab_node(nid);
 412		}
 413	} while ((freed >> shift++) > 1);
 414}
 415
 416static int reclaimer_offset(void)
 417{
 418	BUILD_BUG_ON(PGSTEAL_DIRECT - PGSTEAL_KSWAPD !=
 419			PGDEMOTE_DIRECT - PGDEMOTE_KSWAPD);
 420	BUILD_BUG_ON(PGSTEAL_KHUGEPAGED - PGSTEAL_KSWAPD !=
 421			PGDEMOTE_KHUGEPAGED - PGDEMOTE_KSWAPD);
 422	BUILD_BUG_ON(PGSTEAL_DIRECT - PGSTEAL_KSWAPD !=
 423			PGSCAN_DIRECT - PGSCAN_KSWAPD);
 424	BUILD_BUG_ON(PGSTEAL_KHUGEPAGED - PGSTEAL_KSWAPD !=
 425			PGSCAN_KHUGEPAGED - PGSCAN_KSWAPD);
 426
 427	if (current_is_kswapd())
 428		return 0;
 429	if (current_is_khugepaged())
 430		return PGSTEAL_KHUGEPAGED - PGSTEAL_KSWAPD;
 431	return PGSTEAL_DIRECT - PGSTEAL_KSWAPD;
 432}
 433
 434static inline int is_page_cache_freeable(struct folio *folio)
 435{
 436	/*
 437	 * A freeable page cache folio is referenced only by the caller
 438	 * that isolated the folio, the page cache and optional filesystem
 439	 * private data at folio->private.
 440	 */
 441	return folio_ref_count(folio) - folio_test_private(folio) ==
 442		1 + folio_nr_pages(folio);
 443}
 444
 445/*
 446 * We detected a synchronous write error writing a folio out.  Probably
 447 * -ENOSPC.  We need to propagate that into the address_space for a subsequent
 448 * fsync(), msync() or close().
 449 *
 450 * The tricky part is that after writepage we cannot touch the mapping: nothing
 451 * prevents it from being freed up.  But we have a ref on the folio and once
 452 * that folio is locked, the mapping is pinned.
 453 *
 454 * We're allowed to run sleeping folio_lock() here because we know the caller has
 455 * __GFP_FS.
 456 */
 457static void handle_write_error(struct address_space *mapping,
 458				struct folio *folio, int error)
 459{
 460	folio_lock(folio);
 461	if (folio_mapping(folio) == mapping)
 462		mapping_set_error(mapping, error);
 463	folio_unlock(folio);
 464}
 465
 466static bool skip_throttle_noprogress(pg_data_t *pgdat)
 467{
 468	int reclaimable = 0, write_pending = 0;
 469	int i;
 470
 471	/*
 472	 * If kswapd is disabled, reschedule if necessary but do not
 473	 * throttle as the system is likely near OOM.
 474	 */
 475	if (pgdat->kswapd_failures >= MAX_RECLAIM_RETRIES)
 476		return true;
 477
 478	/*
 479	 * If there are a lot of dirty/writeback folios then do not
 480	 * throttle as throttling will occur when the folios cycle
 481	 * towards the end of the LRU if still under writeback.
 482	 */
 483	for (i = 0; i < MAX_NR_ZONES; i++) {
 484		struct zone *zone = pgdat->node_zones + i;
 485
 486		if (!managed_zone(zone))
 487			continue;
 488
 489		reclaimable += zone_reclaimable_pages(zone);
 490		write_pending += zone_page_state_snapshot(zone,
 491						  NR_ZONE_WRITE_PENDING);
 492	}
 493	if (2 * write_pending <= reclaimable)
 494		return true;
 495
 496	return false;
 497}
 498
 499void reclaim_throttle(pg_data_t *pgdat, enum vmscan_throttle_state reason)
 500{
 501	wait_queue_head_t *wqh = &pgdat->reclaim_wait[reason];
 502	long timeout, ret;
 503	DEFINE_WAIT(wait);
 504
 505	/*
 506	 * Do not throttle user workers, kthreads other than kswapd or
 507	 * workqueues. They may be required for reclaim to make
 508	 * forward progress (e.g. journalling workqueues or kthreads).
 509	 */
 510	if (!current_is_kswapd() &&
 511	    current->flags & (PF_USER_WORKER|PF_KTHREAD)) {
 512		cond_resched();
 513		return;
 514	}
 515
 516	/*
 517	 * These figures are pulled out of thin air.
 518	 * VMSCAN_THROTTLE_ISOLATED is a transient condition based on too many
 519	 * parallel reclaimers which is a short-lived event so the timeout is
 520	 * short. Failing to make progress or waiting on writeback are
 521	 * potentially long-lived events so use a longer timeout. This is shaky
 522	 * logic as a failure to make progress could be due to anything from
 523	 * writeback to a slow device to excessive referenced folios at the tail
 524	 * of the inactive LRU.
 525	 */
 526	switch(reason) {
 527	case VMSCAN_THROTTLE_WRITEBACK:
 528		timeout = HZ/10;
 529
 530		if (atomic_inc_return(&pgdat->nr_writeback_throttled) == 1) {
 531			WRITE_ONCE(pgdat->nr_reclaim_start,
 532				node_page_state(pgdat, NR_THROTTLED_WRITTEN));
 533		}
 534
 535		break;
 536	case VMSCAN_THROTTLE_CONGESTED:
 537		fallthrough;
 538	case VMSCAN_THROTTLE_NOPROGRESS:
 539		if (skip_throttle_noprogress(pgdat)) {
 540			cond_resched();
 541			return;
 542		}
 543
 544		timeout = 1;
 545
 546		break;
 547	case VMSCAN_THROTTLE_ISOLATED:
 548		timeout = HZ/50;
 549		break;
 550	default:
 551		WARN_ON_ONCE(1);
 552		timeout = HZ;
 553		break;
 554	}
 555
 556	prepare_to_wait(wqh, &wait, TASK_UNINTERRUPTIBLE);
 557	ret = schedule_timeout(timeout);
 558	finish_wait(wqh, &wait);
 559
 560	if (reason == VMSCAN_THROTTLE_WRITEBACK)
 561		atomic_dec(&pgdat->nr_writeback_throttled);
 562
 563	trace_mm_vmscan_throttled(pgdat->node_id, jiffies_to_usecs(timeout),
 564				jiffies_to_usecs(timeout - ret),
 565				reason);
 566}
 567
 568/*
 569 * Account for folios written if tasks are throttled waiting on dirty
 570 * folios to clean. If enough folios have been cleaned since throttling
 571 * started then wakeup the throttled tasks.
 572 */
 573void __acct_reclaim_writeback(pg_data_t *pgdat, struct folio *folio,
 574							int nr_throttled)
 575{
 576	unsigned long nr_written;
 577
 578	node_stat_add_folio(folio, NR_THROTTLED_WRITTEN);
 579
 580	/*
 581	 * This is an inaccurate read as the per-cpu deltas may not
 582	 * be synchronised. However, given that the system is
 583	 * writeback throttled, it is not worth taking the penalty
 584	 * of getting an accurate count. At worst, the throttle
 585	 * timeout guarantees forward progress.
 586	 */
 587	nr_written = node_page_state(pgdat, NR_THROTTLED_WRITTEN) -
 588		READ_ONCE(pgdat->nr_reclaim_start);
 589
 590	if (nr_written > SWAP_CLUSTER_MAX * nr_throttled)
 591		wake_up(&pgdat->reclaim_wait[VMSCAN_THROTTLE_WRITEBACK]);
 592}
 593
 594/* possible outcome of pageout() */
 595typedef enum {
 596	/* failed to write folio out, folio is locked */
 597	PAGE_KEEP,
 598	/* move folio to the active list, folio is locked */
 599	PAGE_ACTIVATE,
 600	/* folio has been sent to the disk successfully, folio is unlocked */
 601	PAGE_SUCCESS,
 602	/* folio is clean and locked */
 603	PAGE_CLEAN,
 604} pageout_t;
 605
 606/*
 607 * pageout is called by shrink_folio_list() for each dirty folio.
 608 * Calls ->writepage().
 609 */
 610static pageout_t pageout(struct folio *folio, struct address_space *mapping,
 611			 struct swap_iocb **plug)
 612{
 613	/*
 614	 * If the folio is dirty, only perform writeback if that write
 615	 * will be non-blocking.  To prevent this allocation from being
 616	 * stalled by pagecache activity.  But note that there may be
 617	 * stalls if we need to run get_block().  We could test
 618	 * PagePrivate for that.
 619	 *
 620	 * If this process is currently in __generic_file_write_iter() against
 621	 * this folio's queue, we can perform writeback even if that
 622	 * will block.
 623	 *
 624	 * If the folio is swapcache, write it back even if that would
 625	 * block, for some throttling. This happens by accident, because
 626	 * swap_backing_dev_info is bust: it doesn't reflect the
 627	 * congestion state of the swapdevs.  Easy to fix, if needed.
 628	 */
 629	if (!is_page_cache_freeable(folio))
 630		return PAGE_KEEP;
 631	if (!mapping) {
 632		/*
 633		 * Some data journaling orphaned folios can have
 634		 * folio->mapping == NULL while being dirty with clean buffers.
 635		 */
 636		if (folio_test_private(folio)) {
 637			if (try_to_free_buffers(folio)) {
 638				folio_clear_dirty(folio);
 639				pr_info("%s: orphaned folio\n", __func__);
 640				return PAGE_CLEAN;
 641			}
 642		}
 643		return PAGE_KEEP;
 644	}
 645	if (mapping->a_ops->writepage == NULL)
 646		return PAGE_ACTIVATE;
 647
 648	if (folio_clear_dirty_for_io(folio)) {
 649		int res;
 650		struct writeback_control wbc = {
 651			.sync_mode = WB_SYNC_NONE,
 652			.nr_to_write = SWAP_CLUSTER_MAX,
 653			.range_start = 0,
 654			.range_end = LLONG_MAX,
 655			.for_reclaim = 1,
 656			.swap_plug = plug,
 657		};
 658
 659		folio_set_reclaim(folio);
 660		res = mapping->a_ops->writepage(&folio->page, &wbc);
 661		if (res < 0)
 662			handle_write_error(mapping, folio, res);
 663		if (res == AOP_WRITEPAGE_ACTIVATE) {
 664			folio_clear_reclaim(folio);
 665			return PAGE_ACTIVATE;
 666		}
 667
 668		if (!folio_test_writeback(folio)) {
 669			/* synchronous write or broken a_ops? */
 670			folio_clear_reclaim(folio);
 671		}
 672		trace_mm_vmscan_write_folio(folio);
 673		node_stat_add_folio(folio, NR_VMSCAN_WRITE);
 674		return PAGE_SUCCESS;
 675	}
 676
 677	return PAGE_CLEAN;
 678}
 679
 680/*
 681 * Same as remove_mapping, but if the folio is removed from the mapping, it
 682 * gets returned with a refcount of 0.
 683 */
 684static int __remove_mapping(struct address_space *mapping, struct folio *folio,
 685			    bool reclaimed, struct mem_cgroup *target_memcg)
 686{
 687	int refcount;
 688	void *shadow = NULL;
 689
 690	BUG_ON(!folio_test_locked(folio));
 691	BUG_ON(mapping != folio_mapping(folio));
 692
 693	if (!folio_test_swapcache(folio))
 694		spin_lock(&mapping->host->i_lock);
 695	xa_lock_irq(&mapping->i_pages);
 696	/*
 697	 * The non racy check for a busy folio.
 698	 *
 699	 * Must be careful with the order of the tests. When someone has
 700	 * a ref to the folio, it may be possible that they dirty it then
 701	 * drop the reference. So if the dirty flag is tested before the
 702	 * refcount here, then the following race may occur:
 703	 *
 704	 * get_user_pages(&page);
 705	 * [user mapping goes away]
 706	 * write_to(page);
 707	 *				!folio_test_dirty(folio)    [good]
 708	 * folio_set_dirty(folio);
 709	 * folio_put(folio);
 710	 *				!refcount(folio)   [good, discard it]
 711	 *
 712	 * [oops, our write_to data is lost]
 713	 *
 714	 * Reversing the order of the tests ensures such a situation cannot
 715	 * escape unnoticed. The smp_rmb is needed to ensure the folio->flags
 716	 * load is not satisfied before that of folio->_refcount.
 717	 *
 718	 * Note that if the dirty flag is always set via folio_mark_dirty,
 719	 * and thus under the i_pages lock, then this ordering is not required.
 720	 */
 721	refcount = 1 + folio_nr_pages(folio);
 722	if (!folio_ref_freeze(folio, refcount))
 723		goto cannot_free;
 724	/* note: atomic_cmpxchg in folio_ref_freeze provides the smp_rmb */
 725	if (unlikely(folio_test_dirty(folio))) {
 726		folio_ref_unfreeze(folio, refcount);
 727		goto cannot_free;
 728	}
 729
 730	if (folio_test_swapcache(folio)) {
 731		swp_entry_t swap = folio->swap;
 732
 733		if (reclaimed && !mapping_exiting(mapping))
 734			shadow = workingset_eviction(folio, target_memcg);
 735		__delete_from_swap_cache(folio, swap, shadow);
 736		mem_cgroup_swapout(folio, swap);
 737		xa_unlock_irq(&mapping->i_pages);
 738		put_swap_folio(folio, swap);
 739	} else {
 740		void (*free_folio)(struct folio *);
 741
 742		free_folio = mapping->a_ops->free_folio;
 743		/*
 744		 * Remember a shadow entry for reclaimed file cache in
 745		 * order to detect refaults, thus thrashing, later on.
 746		 *
 747		 * But don't store shadows in an address space that is
 748		 * already exiting.  This is not just an optimization,
 749		 * inode reclaim needs to empty out the radix tree or
 750		 * the nodes are lost.  Don't plant shadows behind its
 751		 * back.
 752		 *
 753		 * We also don't store shadows for DAX mappings because the
 754		 * only page cache folios found in these are zero pages
 755		 * covering holes, and because we don't want to mix DAX
 756		 * exceptional entries and shadow exceptional entries in the
 757		 * same address_space.
 758		 */
 759		if (reclaimed && folio_is_file_lru(folio) &&
 760		    !mapping_exiting(mapping) && !dax_mapping(mapping))
 761			shadow = workingset_eviction(folio, target_memcg);
 762		__filemap_remove_folio(folio, shadow);
 763		xa_unlock_irq(&mapping->i_pages);
 764		if (mapping_shrinkable(mapping))
 765			inode_add_lru(mapping->host);
 766		spin_unlock(&mapping->host->i_lock);
 767
 768		if (free_folio)
 769			free_folio(folio);
 770	}
 771
 772	return 1;
 773
 774cannot_free:
 775	xa_unlock_irq(&mapping->i_pages);
 776	if (!folio_test_swapcache(folio))
 777		spin_unlock(&mapping->host->i_lock);
 778	return 0;
 779}
 780
 781/**
 782 * remove_mapping() - Attempt to remove a folio from its mapping.
 783 * @mapping: The address space.
 784 * @folio: The folio to remove.
 785 *
 786 * If the folio is dirty, under writeback or if someone else has a ref
 787 * on it, removal will fail.
 788 * Return: The number of pages removed from the mapping.  0 if the folio
 789 * could not be removed.
 790 * Context: The caller should have a single refcount on the folio and
 791 * hold its lock.
 792 */
 793long remove_mapping(struct address_space *mapping, struct folio *folio)
 794{
 795	if (__remove_mapping(mapping, folio, false, NULL)) {
 796		/*
 797		 * Unfreezing the refcount with 1 effectively
 798		 * drops the pagecache ref for us without requiring another
 799		 * atomic operation.
 800		 */
 801		folio_ref_unfreeze(folio, 1);
 802		return folio_nr_pages(folio);
 803	}
 804	return 0;
 805}
 806
 807/**
 808 * folio_putback_lru - Put previously isolated folio onto appropriate LRU list.
 809 * @folio: Folio to be returned to an LRU list.
 810 *
 811 * Add previously isolated @folio to appropriate LRU list.
 812 * The folio may still be unevictable for other reasons.
 813 *
 814 * Context: lru_lock must not be held, interrupts must be enabled.
 815 */
 816void folio_putback_lru(struct folio *folio)
 817{
 818	folio_add_lru(folio);
 819	folio_put(folio);		/* drop ref from isolate */
 820}
 821
 822enum folio_references {
 823	FOLIOREF_RECLAIM,
 824	FOLIOREF_RECLAIM_CLEAN,
 825	FOLIOREF_KEEP,
 826	FOLIOREF_ACTIVATE,
 827};
 828
 829static enum folio_references folio_check_references(struct folio *folio,
 830						  struct scan_control *sc)
 831{
 832	int referenced_ptes, referenced_folio;
 833	unsigned long vm_flags;
 834
 835	referenced_ptes = folio_referenced(folio, 1, sc->target_mem_cgroup,
 836					   &vm_flags);
 837	referenced_folio = folio_test_clear_referenced(folio);
 838
 839	/*
 840	 * The supposedly reclaimable folio was found to be in a VM_LOCKED vma.
 841	 * Let the folio, now marked Mlocked, be moved to the unevictable list.
 842	 */
 843	if (vm_flags & VM_LOCKED)
 844		return FOLIOREF_ACTIVATE;
 845
 846	/* rmap lock contention: rotate */
 847	if (referenced_ptes == -1)
 848		return FOLIOREF_KEEP;
 849
 850	if (referenced_ptes) {
 851		/*
 852		 * All mapped folios start out with page table
 853		 * references from the instantiating fault, so we need
 854		 * to look twice if a mapped file/anon folio is used more
 855		 * than once.
 856		 *
 857		 * Mark it and spare it for another trip around the
 858		 * inactive list.  Another page table reference will
 859		 * lead to its activation.
 860		 *
 861		 * Note: the mark is set for activated folios as well
 862		 * so that recently deactivated but used folios are
 863		 * quickly recovered.
 864		 */
 865		folio_set_referenced(folio);
 866
 867		if (referenced_folio || referenced_ptes > 1)
 868			return FOLIOREF_ACTIVATE;
 869
 870		/*
 871		 * Activate file-backed executable folios after first usage.
 872		 */
 873		if ((vm_flags & VM_EXEC) && folio_is_file_lru(folio))
 874			return FOLIOREF_ACTIVATE;
 875
 876		return FOLIOREF_KEEP;
 877	}
 878
 879	/* Reclaim if clean, defer dirty folios to writeback */
 880	if (referenced_folio && folio_is_file_lru(folio))
 881		return FOLIOREF_RECLAIM_CLEAN;
 882
 883	return FOLIOREF_RECLAIM;
 884}
 885
 886/* Check if a folio is dirty or under writeback */
 887static void folio_check_dirty_writeback(struct folio *folio,
 888				       bool *dirty, bool *writeback)
 889{
 890	struct address_space *mapping;
 891
 892	/*
 893	 * Anonymous folios are not handled by flushers and must be written
 894	 * from reclaim context. Do not stall reclaim based on them.
 895	 * MADV_FREE anonymous folios are put into inactive file list too.
 896	 * They could be mistakenly treated as file lru. So further anon
 897	 * test is needed.
 898	 */
 899	if (!folio_is_file_lru(folio) ||
 900	    (folio_test_anon(folio) && !folio_test_swapbacked(folio))) {
 901		*dirty = false;
 902		*writeback = false;
 903		return;
 904	}
 905
 906	/* By default assume that the folio flags are accurate */
 907	*dirty = folio_test_dirty(folio);
 908	*writeback = folio_test_writeback(folio);
 909
 910	/* Verify dirty/writeback state if the filesystem supports it */
 911	if (!folio_test_private(folio))
 912		return;
 913
 914	mapping = folio_mapping(folio);
 915	if (mapping && mapping->a_ops->is_dirty_writeback)
 916		mapping->a_ops->is_dirty_writeback(folio, dirty, writeback);
 917}
 918
 919static struct folio *alloc_demote_folio(struct folio *src,
 920		unsigned long private)
 921{
 922	struct folio *dst;
 923	nodemask_t *allowed_mask;
 924	struct migration_target_control *mtc;
 925
 926	mtc = (struct migration_target_control *)private;
 927
 928	allowed_mask = mtc->nmask;
 929	/*
 930	 * make sure we allocate from the target node first also trying to
 931	 * demote or reclaim pages from the target node via kswapd if we are
 932	 * low on free memory on target node. If we don't do this and if
 933	 * we have free memory on the slower(lower) memtier, we would start
 934	 * allocating pages from slower(lower) memory tiers without even forcing
 935	 * a demotion of cold pages from the target memtier. This can result
 936	 * in the kernel placing hot pages in slower(lower) memory tiers.
 937	 */
 938	mtc->nmask = NULL;
 939	mtc->gfp_mask |= __GFP_THISNODE;
 940	dst = alloc_migration_target(src, (unsigned long)mtc);
 941	if (dst)
 942		return dst;
 943
 944	mtc->gfp_mask &= ~__GFP_THISNODE;
 945	mtc->nmask = allowed_mask;
 946
 947	return alloc_migration_target(src, (unsigned long)mtc);
 948}
 949
 950/*
 951 * Take folios on @demote_folios and attempt to demote them to another node.
 952 * Folios which are not demoted are left on @demote_folios.
 953 */
 954static unsigned int demote_folio_list(struct list_head *demote_folios,
 955				     struct pglist_data *pgdat)
 956{
 957	int target_nid = next_demotion_node(pgdat->node_id);
 958	unsigned int nr_succeeded;
 959	nodemask_t allowed_mask;
 960
 961	struct migration_target_control mtc = {
 962		/*
 963		 * Allocate from 'node', or fail quickly and quietly.
 964		 * When this happens, 'page' will likely just be discarded
 965		 * instead of migrated.
 966		 */
 967		.gfp_mask = (GFP_HIGHUSER_MOVABLE & ~__GFP_RECLAIM) | __GFP_NOWARN |
 968			__GFP_NOMEMALLOC | GFP_NOWAIT,
 969		.nid = target_nid,
 970		.nmask = &allowed_mask
 971	};
 972
 973	if (list_empty(demote_folios))
 974		return 0;
 975
 976	if (target_nid == NUMA_NO_NODE)
 977		return 0;
 978
 979	node_get_allowed_targets(pgdat, &allowed_mask);
 980
 981	/* Demotion ignores all cpuset and mempolicy settings */
 982	migrate_pages(demote_folios, alloc_demote_folio, NULL,
 983		      (unsigned long)&mtc, MIGRATE_ASYNC, MR_DEMOTION,
 984		      &nr_succeeded);
 985
 986	mod_node_page_state(pgdat, PGDEMOTE_KSWAPD + reclaimer_offset(),
 987			    nr_succeeded);
 988
 989	return nr_succeeded;
 990}
 991
 992static bool may_enter_fs(struct folio *folio, gfp_t gfp_mask)
 993{
 994	if (gfp_mask & __GFP_FS)
 995		return true;
 996	if (!folio_test_swapcache(folio) || !(gfp_mask & __GFP_IO))
 997		return false;
 998	/*
 999	 * We can "enter_fs" for swap-cache with only __GFP_IO
1000	 * providing this isn't SWP_FS_OPS.
1001	 * ->flags can be updated non-atomicially (scan_swap_map_slots),
1002	 * but that will never affect SWP_FS_OPS, so the data_race
1003	 * is safe.
1004	 */
1005	return !data_race(folio_swap_flags(folio) & SWP_FS_OPS);
1006}
1007
1008/*
1009 * shrink_folio_list() returns the number of reclaimed pages
1010 */
1011static unsigned int shrink_folio_list(struct list_head *folio_list,
1012		struct pglist_data *pgdat, struct scan_control *sc,
1013		struct reclaim_stat *stat, bool ignore_references)
1014{
1015	struct folio_batch free_folios;
1016	LIST_HEAD(ret_folios);
 
1017	LIST_HEAD(demote_folios);
1018	unsigned int nr_reclaimed = 0;
1019	unsigned int pgactivate = 0;
1020	bool do_demote_pass;
1021	struct swap_iocb *plug = NULL;
1022
1023	folio_batch_init(&free_folios);
1024	memset(stat, 0, sizeof(*stat));
1025	cond_resched();
1026	do_demote_pass = can_demote(pgdat->node_id, sc);
1027
1028retry:
1029	while (!list_empty(folio_list)) {
1030		struct address_space *mapping;
1031		struct folio *folio;
1032		enum folio_references references = FOLIOREF_RECLAIM;
1033		bool dirty, writeback;
1034		unsigned int nr_pages;
1035
1036		cond_resched();
1037
1038		folio = lru_to_folio(folio_list);
1039		list_del(&folio->lru);
1040
1041		if (!folio_trylock(folio))
1042			goto keep;
1043
1044		VM_BUG_ON_FOLIO(folio_test_active(folio), folio);
1045
1046		nr_pages = folio_nr_pages(folio);
1047
1048		/* Account the number of base pages */
1049		sc->nr_scanned += nr_pages;
1050
1051		if (unlikely(!folio_evictable(folio)))
1052			goto activate_locked;
1053
1054		if (!sc->may_unmap && folio_mapped(folio))
1055			goto keep_locked;
1056
1057		/* folio_update_gen() tried to promote this page? */
1058		if (lru_gen_enabled() && !ignore_references &&
1059		    folio_mapped(folio) && folio_test_referenced(folio))
1060			goto keep_locked;
1061
1062		/*
1063		 * The number of dirty pages determines if a node is marked
1064		 * reclaim_congested. kswapd will stall and start writing
1065		 * folios if the tail of the LRU is all dirty unqueued folios.
1066		 */
1067		folio_check_dirty_writeback(folio, &dirty, &writeback);
1068		if (dirty || writeback)
1069			stat->nr_dirty += nr_pages;
1070
1071		if (dirty && !writeback)
1072			stat->nr_unqueued_dirty += nr_pages;
1073
1074		/*
1075		 * Treat this folio as congested if folios are cycling
1076		 * through the LRU so quickly that the folios marked
1077		 * for immediate reclaim are making it to the end of
1078		 * the LRU a second time.
1079		 */
1080		if (writeback && folio_test_reclaim(folio))
1081			stat->nr_congested += nr_pages;
1082
1083		/*
1084		 * If a folio at the tail of the LRU is under writeback, there
1085		 * are three cases to consider.
1086		 *
1087		 * 1) If reclaim is encountering an excessive number
1088		 *    of folios under writeback and this folio has both
1089		 *    the writeback and reclaim flags set, then it
1090		 *    indicates that folios are being queued for I/O but
1091		 *    are being recycled through the LRU before the I/O
1092		 *    can complete. Waiting on the folio itself risks an
1093		 *    indefinite stall if it is impossible to writeback
1094		 *    the folio due to I/O error or disconnected storage
1095		 *    so instead note that the LRU is being scanned too
1096		 *    quickly and the caller can stall after the folio
1097		 *    list has been processed.
1098		 *
1099		 * 2) Global or new memcg reclaim encounters a folio that is
1100		 *    not marked for immediate reclaim, or the caller does not
1101		 *    have __GFP_FS (or __GFP_IO if it's simply going to swap,
1102		 *    not to fs). In this case mark the folio for immediate
1103		 *    reclaim and continue scanning.
1104		 *
1105		 *    Require may_enter_fs() because we would wait on fs, which
1106		 *    may not have submitted I/O yet. And the loop driver might
1107		 *    enter reclaim, and deadlock if it waits on a folio for
1108		 *    which it is needed to do the write (loop masks off
1109		 *    __GFP_IO|__GFP_FS for this reason); but more thought
1110		 *    would probably show more reasons.
1111		 *
1112		 * 3) Legacy memcg encounters a folio that already has the
1113		 *    reclaim flag set. memcg does not have any dirty folio
1114		 *    throttling so we could easily OOM just because too many
1115		 *    folios are in writeback and there is nothing else to
1116		 *    reclaim. Wait for the writeback to complete.
1117		 *
1118		 * In cases 1) and 2) we activate the folios to get them out of
1119		 * the way while we continue scanning for clean folios on the
1120		 * inactive list and refilling from the active list. The
1121		 * observation here is that waiting for disk writes is more
1122		 * expensive than potentially causing reloads down the line.
1123		 * Since they're marked for immediate reclaim, they won't put
1124		 * memory pressure on the cache working set any longer than it
1125		 * takes to write them to disk.
1126		 */
1127		if (folio_test_writeback(folio)) {
1128			/* Case 1 above */
1129			if (current_is_kswapd() &&
1130			    folio_test_reclaim(folio) &&
1131			    test_bit(PGDAT_WRITEBACK, &pgdat->flags)) {
1132				stat->nr_immediate += nr_pages;
1133				goto activate_locked;
1134
1135			/* Case 2 above */
1136			} else if (writeback_throttling_sane(sc) ||
1137			    !folio_test_reclaim(folio) ||
1138			    !may_enter_fs(folio, sc->gfp_mask)) {
1139				/*
1140				 * This is slightly racy -
1141				 * folio_end_writeback() might have
1142				 * just cleared the reclaim flag, then
1143				 * setting the reclaim flag here ends up
1144				 * interpreted as the readahead flag - but
1145				 * that does not matter enough to care.
1146				 * What we do want is for this folio to
1147				 * have the reclaim flag set next time
1148				 * memcg reclaim reaches the tests above,
1149				 * so it will then wait for writeback to
1150				 * avoid OOM; and it's also appropriate
1151				 * in global reclaim.
1152				 */
1153				folio_set_reclaim(folio);
1154				stat->nr_writeback += nr_pages;
1155				goto activate_locked;
1156
1157			/* Case 3 above */
1158			} else {
1159				folio_unlock(folio);
1160				folio_wait_writeback(folio);
1161				/* then go back and try same folio again */
1162				list_add_tail(&folio->lru, folio_list);
1163				continue;
1164			}
1165		}
1166
1167		if (!ignore_references)
1168			references = folio_check_references(folio, sc);
1169
1170		switch (references) {
1171		case FOLIOREF_ACTIVATE:
1172			goto activate_locked;
1173		case FOLIOREF_KEEP:
1174			stat->nr_ref_keep += nr_pages;
1175			goto keep_locked;
1176		case FOLIOREF_RECLAIM:
1177		case FOLIOREF_RECLAIM_CLEAN:
1178			; /* try to reclaim the folio below */
1179		}
1180
1181		/*
1182		 * Before reclaiming the folio, try to relocate
1183		 * its contents to another node.
1184		 */
1185		if (do_demote_pass &&
1186		    (thp_migration_supported() || !folio_test_large(folio))) {
1187			list_add(&folio->lru, &demote_folios);
1188			folio_unlock(folio);
1189			continue;
1190		}
1191
1192		/*
1193		 * Anonymous process memory has backing store?
1194		 * Try to allocate it some swap space here.
1195		 * Lazyfree folio could be freed directly
1196		 */
1197		if (folio_test_anon(folio) && folio_test_swapbacked(folio)) {
1198			if (!folio_test_swapcache(folio)) {
1199				if (!(sc->gfp_mask & __GFP_IO))
1200					goto keep_locked;
1201				if (folio_maybe_dma_pinned(folio))
1202					goto keep_locked;
1203				if (folio_test_large(folio)) {
1204					/* cannot split folio, skip it */
1205					if (!can_split_folio(folio, NULL))
1206						goto activate_locked;
1207					/*
1208					 * Split folios without a PMD map right
1209					 * away. Chances are some or all of the
1210					 * tail pages can be freed without IO.
1211					 */
1212					if (!folio_entire_mapcount(folio) &&
1213					    split_folio_to_list(folio,
1214								folio_list))
1215						goto activate_locked;
1216				}
1217				if (!add_to_swap(folio)) {
1218					if (!folio_test_large(folio))
1219						goto activate_locked_split;
1220					/* Fallback to swap normal pages */
1221					if (split_folio_to_list(folio,
1222								folio_list))
1223						goto activate_locked;
1224#ifdef CONFIG_TRANSPARENT_HUGEPAGE
1225					count_memcg_folio_events(folio, THP_SWPOUT_FALLBACK, 1);
1226					count_vm_event(THP_SWPOUT_FALLBACK);
1227#endif
1228					if (!add_to_swap(folio))
1229						goto activate_locked_split;
1230				}
1231			}
1232		} else if (folio_test_swapbacked(folio) &&
1233			   folio_test_large(folio)) {
1234			/* Split shmem folio */
1235			if (split_folio_to_list(folio, folio_list))
1236				goto keep_locked;
1237		}
1238
1239		/*
1240		 * If the folio was split above, the tail pages will make
1241		 * their own pass through this function and be accounted
1242		 * then.
1243		 */
1244		if ((nr_pages > 1) && !folio_test_large(folio)) {
1245			sc->nr_scanned -= (nr_pages - 1);
1246			nr_pages = 1;
1247		}
1248
1249		/*
1250		 * The folio is mapped into the page tables of one or more
1251		 * processes. Try to unmap it here.
1252		 */
1253		if (folio_mapped(folio)) {
1254			enum ttu_flags flags = TTU_BATCH_FLUSH;
1255			bool was_swapbacked = folio_test_swapbacked(folio);
1256
1257			if (folio_test_pmd_mappable(folio))
1258				flags |= TTU_SPLIT_HUGE_PMD;
1259
1260			try_to_unmap(folio, flags);
1261			if (folio_mapped(folio)) {
1262				stat->nr_unmap_fail += nr_pages;
1263				if (!was_swapbacked &&
1264				    folio_test_swapbacked(folio))
1265					stat->nr_lazyfree_fail += nr_pages;
1266				goto activate_locked;
1267			}
1268		}
1269
1270		/*
1271		 * Folio is unmapped now so it cannot be newly pinned anymore.
1272		 * No point in trying to reclaim folio if it is pinned.
1273		 * Furthermore we don't want to reclaim underlying fs metadata
1274		 * if the folio is pinned and thus potentially modified by the
1275		 * pinning process as that may upset the filesystem.
1276		 */
1277		if (folio_maybe_dma_pinned(folio))
1278			goto activate_locked;
1279
1280		mapping = folio_mapping(folio);
1281		if (folio_test_dirty(folio)) {
1282			/*
1283			 * Only kswapd can writeback filesystem folios
1284			 * to avoid risk of stack overflow. But avoid
1285			 * injecting inefficient single-folio I/O into
1286			 * flusher writeback as much as possible: only
1287			 * write folios when we've encountered many
1288			 * dirty folios, and when we've already scanned
1289			 * the rest of the LRU for clean folios and see
1290			 * the same dirty folios again (with the reclaim
1291			 * flag set).
1292			 */
1293			if (folio_is_file_lru(folio) &&
1294			    (!current_is_kswapd() ||
1295			     !folio_test_reclaim(folio) ||
1296			     !test_bit(PGDAT_DIRTY, &pgdat->flags))) {
1297				/*
1298				 * Immediately reclaim when written back.
1299				 * Similar in principle to folio_deactivate()
1300				 * except we already have the folio isolated
1301				 * and know it's dirty
1302				 */
1303				node_stat_mod_folio(folio, NR_VMSCAN_IMMEDIATE,
1304						nr_pages);
1305				folio_set_reclaim(folio);
1306
1307				goto activate_locked;
1308			}
1309
1310			if (references == FOLIOREF_RECLAIM_CLEAN)
1311				goto keep_locked;
1312			if (!may_enter_fs(folio, sc->gfp_mask))
1313				goto keep_locked;
1314			if (!sc->may_writepage)
1315				goto keep_locked;
1316
1317			/*
1318			 * Folio is dirty. Flush the TLB if a writable entry
1319			 * potentially exists to avoid CPU writes after I/O
1320			 * starts and then write it out here.
1321			 */
1322			try_to_unmap_flush_dirty();
1323			switch (pageout(folio, mapping, &plug)) {
1324			case PAGE_KEEP:
1325				goto keep_locked;
1326			case PAGE_ACTIVATE:
1327				goto activate_locked;
1328			case PAGE_SUCCESS:
1329				stat->nr_pageout += nr_pages;
1330
1331				if (folio_test_writeback(folio))
1332					goto keep;
1333				if (folio_test_dirty(folio))
1334					goto keep;
1335
1336				/*
1337				 * A synchronous write - probably a ramdisk.  Go
1338				 * ahead and try to reclaim the folio.
1339				 */
1340				if (!folio_trylock(folio))
1341					goto keep;
1342				if (folio_test_dirty(folio) ||
1343				    folio_test_writeback(folio))
1344					goto keep_locked;
1345				mapping = folio_mapping(folio);
1346				fallthrough;
1347			case PAGE_CLEAN:
1348				; /* try to free the folio below */
1349			}
1350		}
1351
1352		/*
1353		 * If the folio has buffers, try to free the buffer
1354		 * mappings associated with this folio. If we succeed
1355		 * we try to free the folio as well.
1356		 *
1357		 * We do this even if the folio is dirty.
1358		 * filemap_release_folio() does not perform I/O, but it
1359		 * is possible for a folio to have the dirty flag set,
1360		 * but it is actually clean (all its buffers are clean).
1361		 * This happens if the buffers were written out directly,
1362		 * with submit_bh(). ext3 will do this, as well as
1363		 * the blockdev mapping.  filemap_release_folio() will
1364		 * discover that cleanness and will drop the buffers
1365		 * and mark the folio clean - it can be freed.
1366		 *
1367		 * Rarely, folios can have buffers and no ->mapping.
1368		 * These are the folios which were not successfully
1369		 * invalidated in truncate_cleanup_folio().  We try to
1370		 * drop those buffers here and if that worked, and the
1371		 * folio is no longer mapped into process address space
1372		 * (refcount == 1) it can be freed.  Otherwise, leave
1373		 * the folio on the LRU so it is swappable.
1374		 */
1375		if (folio_needs_release(folio)) {
1376			if (!filemap_release_folio(folio, sc->gfp_mask))
1377				goto activate_locked;
1378			if (!mapping && folio_ref_count(folio) == 1) {
1379				folio_unlock(folio);
1380				if (folio_put_testzero(folio))
1381					goto free_it;
1382				else {
1383					/*
1384					 * rare race with speculative reference.
1385					 * the speculative reference will free
1386					 * this folio shortly, so we may
1387					 * increment nr_reclaimed here (and
1388					 * leave it off the LRU).
1389					 */
1390					nr_reclaimed += nr_pages;
1391					continue;
1392				}
1393			}
1394		}
1395
1396		if (folio_test_anon(folio) && !folio_test_swapbacked(folio)) {
1397			/* follow __remove_mapping for reference */
1398			if (!folio_ref_freeze(folio, 1))
1399				goto keep_locked;
1400			/*
1401			 * The folio has only one reference left, which is
1402			 * from the isolation. After the caller puts the
1403			 * folio back on the lru and drops the reference, the
1404			 * folio will be freed anyway. It doesn't matter
1405			 * which lru it goes on. So we don't bother checking
1406			 * the dirty flag here.
1407			 */
1408			count_vm_events(PGLAZYFREED, nr_pages);
1409			count_memcg_folio_events(folio, PGLAZYFREED, nr_pages);
1410		} else if (!mapping || !__remove_mapping(mapping, folio, true,
1411							 sc->target_mem_cgroup))
1412			goto keep_locked;
1413
1414		folio_unlock(folio);
1415free_it:
1416		/*
1417		 * Folio may get swapped out as a whole, need to account
1418		 * all pages in it.
1419		 */
1420		nr_reclaimed += nr_pages;
1421
1422		if (folio_test_large(folio) &&
1423		    folio_test_large_rmappable(folio))
1424			folio_undo_large_rmappable(folio);
1425		if (folio_batch_add(&free_folios, folio) == 0) {
1426			mem_cgroup_uncharge_folios(&free_folios);
1427			try_to_unmap_flush();
1428			free_unref_folios(&free_folios);
1429		}
1430		continue;
1431
1432activate_locked_split:
1433		/*
1434		 * The tail pages that are failed to add into swap cache
1435		 * reach here.  Fixup nr_scanned and nr_pages.
1436		 */
1437		if (nr_pages > 1) {
1438			sc->nr_scanned -= (nr_pages - 1);
1439			nr_pages = 1;
1440		}
1441activate_locked:
1442		/* Not a candidate for swapping, so reclaim swap space. */
1443		if (folio_test_swapcache(folio) &&
1444		    (mem_cgroup_swap_full(folio) || folio_test_mlocked(folio)))
1445			folio_free_swap(folio);
1446		VM_BUG_ON_FOLIO(folio_test_active(folio), folio);
1447		if (!folio_test_mlocked(folio)) {
1448			int type = folio_is_file_lru(folio);
1449			folio_set_active(folio);
1450			stat->nr_activate[type] += nr_pages;
1451			count_memcg_folio_events(folio, PGACTIVATE, nr_pages);
1452		}
1453keep_locked:
1454		folio_unlock(folio);
1455keep:
1456		list_add(&folio->lru, &ret_folios);
1457		VM_BUG_ON_FOLIO(folio_test_lru(folio) ||
1458				folio_test_unevictable(folio), folio);
1459	}
1460	/* 'folio_list' is always empty here */
1461
1462	/* Migrate folios selected for demotion */
1463	nr_reclaimed += demote_folio_list(&demote_folios, pgdat);
1464	/* Folios that could not be demoted are still in @demote_folios */
1465	if (!list_empty(&demote_folios)) {
1466		/* Folios which weren't demoted go back on @folio_list */
1467		list_splice_init(&demote_folios, folio_list);
1468
1469		/*
1470		 * goto retry to reclaim the undemoted folios in folio_list if
1471		 * desired.
1472		 *
1473		 * Reclaiming directly from top tier nodes is not often desired
1474		 * due to it breaking the LRU ordering: in general memory
1475		 * should be reclaimed from lower tier nodes and demoted from
1476		 * top tier nodes.
1477		 *
1478		 * However, disabling reclaim from top tier nodes entirely
1479		 * would cause ooms in edge scenarios where lower tier memory
1480		 * is unreclaimable for whatever reason, eg memory being
1481		 * mlocked or too hot to reclaim. We can disable reclaim
1482		 * from top tier nodes in proactive reclaim though as that is
1483		 * not real memory pressure.
1484		 */
1485		if (!sc->proactive) {
1486			do_demote_pass = false;
1487			goto retry;
1488		}
1489	}
1490
1491	pgactivate = stat->nr_activate[0] + stat->nr_activate[1];
1492
1493	mem_cgroup_uncharge_folios(&free_folios);
1494	try_to_unmap_flush();
1495	free_unref_folios(&free_folios);
1496
1497	list_splice(&ret_folios, folio_list);
1498	count_vm_events(PGACTIVATE, pgactivate);
1499
1500	if (plug)
1501		swap_write_unplug(plug);
1502	return nr_reclaimed;
1503}
1504
1505unsigned int reclaim_clean_pages_from_list(struct zone *zone,
1506					   struct list_head *folio_list)
1507{
1508	struct scan_control sc = {
1509		.gfp_mask = GFP_KERNEL,
1510		.may_unmap = 1,
1511	};
1512	struct reclaim_stat stat;
1513	unsigned int nr_reclaimed;
1514	struct folio *folio, *next;
1515	LIST_HEAD(clean_folios);
1516	unsigned int noreclaim_flag;
1517
1518	list_for_each_entry_safe(folio, next, folio_list, lru) {
1519		if (!folio_test_hugetlb(folio) && folio_is_file_lru(folio) &&
1520		    !folio_test_dirty(folio) && !__folio_test_movable(folio) &&
1521		    !folio_test_unevictable(folio)) {
1522			folio_clear_active(folio);
1523			list_move(&folio->lru, &clean_folios);
1524		}
1525	}
1526
1527	/*
1528	 * We should be safe here since we are only dealing with file pages and
1529	 * we are not kswapd and therefore cannot write dirty file pages. But
1530	 * call memalloc_noreclaim_save() anyway, just in case these conditions
1531	 * change in the future.
1532	 */
1533	noreclaim_flag = memalloc_noreclaim_save();
1534	nr_reclaimed = shrink_folio_list(&clean_folios, zone->zone_pgdat, &sc,
1535					&stat, true);
1536	memalloc_noreclaim_restore(noreclaim_flag);
1537
1538	list_splice(&clean_folios, folio_list);
1539	mod_node_page_state(zone->zone_pgdat, NR_ISOLATED_FILE,
1540			    -(long)nr_reclaimed);
1541	/*
1542	 * Since lazyfree pages are isolated from file LRU from the beginning,
1543	 * they will rotate back to anonymous LRU in the end if it failed to
1544	 * discard so isolated count will be mismatched.
1545	 * Compensate the isolated count for both LRU lists.
1546	 */
1547	mod_node_page_state(zone->zone_pgdat, NR_ISOLATED_ANON,
1548			    stat.nr_lazyfree_fail);
1549	mod_node_page_state(zone->zone_pgdat, NR_ISOLATED_FILE,
1550			    -(long)stat.nr_lazyfree_fail);
1551	return nr_reclaimed;
1552}
1553
1554/*
1555 * Update LRU sizes after isolating pages. The LRU size updates must
1556 * be complete before mem_cgroup_update_lru_size due to a sanity check.
1557 */
1558static __always_inline void update_lru_sizes(struct lruvec *lruvec,
1559			enum lru_list lru, unsigned long *nr_zone_taken)
1560{
1561	int zid;
1562
1563	for (zid = 0; zid < MAX_NR_ZONES; zid++) {
1564		if (!nr_zone_taken[zid])
1565			continue;
1566
1567		update_lru_size(lruvec, lru, zid, -nr_zone_taken[zid]);
1568	}
1569
1570}
1571
1572#ifdef CONFIG_CMA
1573/*
1574 * It is waste of effort to scan and reclaim CMA pages if it is not available
1575 * for current allocation context. Kswapd can not be enrolled as it can not
1576 * distinguish this scenario by using sc->gfp_mask = GFP_KERNEL
1577 */
1578static bool skip_cma(struct folio *folio, struct scan_control *sc)
1579{
1580	return !current_is_kswapd() &&
1581			gfp_migratetype(sc->gfp_mask) != MIGRATE_MOVABLE &&
1582			folio_migratetype(folio) == MIGRATE_CMA;
1583}
1584#else
1585static bool skip_cma(struct folio *folio, struct scan_control *sc)
1586{
1587	return false;
1588}
1589#endif
1590
1591/*
1592 * Isolating page from the lruvec to fill in @dst list by nr_to_scan times.
1593 *
1594 * lruvec->lru_lock is heavily contended.  Some of the functions that
1595 * shrink the lists perform better by taking out a batch of pages
1596 * and working on them outside the LRU lock.
1597 *
1598 * For pagecache intensive workloads, this function is the hottest
1599 * spot in the kernel (apart from copy_*_user functions).
1600 *
1601 * Lru_lock must be held before calling this function.
1602 *
1603 * @nr_to_scan:	The number of eligible pages to look through on the list.
1604 * @lruvec:	The LRU vector to pull pages from.
1605 * @dst:	The temp list to put pages on to.
1606 * @nr_scanned:	The number of pages that were scanned.
1607 * @sc:		The scan_control struct for this reclaim session
1608 * @lru:	LRU list id for isolating
1609 *
1610 * returns how many pages were moved onto *@dst.
1611 */
1612static unsigned long isolate_lru_folios(unsigned long nr_to_scan,
1613		struct lruvec *lruvec, struct list_head *dst,
1614		unsigned long *nr_scanned, struct scan_control *sc,
1615		enum lru_list lru)
1616{
1617	struct list_head *src = &lruvec->lists[lru];
1618	unsigned long nr_taken = 0;
1619	unsigned long nr_zone_taken[MAX_NR_ZONES] = { 0 };
1620	unsigned long nr_skipped[MAX_NR_ZONES] = { 0, };
1621	unsigned long skipped = 0;
1622	unsigned long scan, total_scan, nr_pages;
1623	LIST_HEAD(folios_skipped);
1624
1625	total_scan = 0;
1626	scan = 0;
1627	while (scan < nr_to_scan && !list_empty(src)) {
1628		struct list_head *move_to = src;
1629		struct folio *folio;
1630
1631		folio = lru_to_folio(src);
1632		prefetchw_prev_lru_folio(folio, src, flags);
1633
1634		nr_pages = folio_nr_pages(folio);
1635		total_scan += nr_pages;
1636
1637		if (folio_zonenum(folio) > sc->reclaim_idx ||
1638				skip_cma(folio, sc)) {
1639			nr_skipped[folio_zonenum(folio)] += nr_pages;
1640			move_to = &folios_skipped;
1641			goto move;
1642		}
1643
1644		/*
1645		 * Do not count skipped folios because that makes the function
1646		 * return with no isolated folios if the LRU mostly contains
1647		 * ineligible folios.  This causes the VM to not reclaim any
1648		 * folios, triggering a premature OOM.
1649		 * Account all pages in a folio.
1650		 */
1651		scan += nr_pages;
1652
1653		if (!folio_test_lru(folio))
1654			goto move;
1655		if (!sc->may_unmap && folio_mapped(folio))
1656			goto move;
1657
1658		/*
1659		 * Be careful not to clear the lru flag until after we're
1660		 * sure the folio is not being freed elsewhere -- the
1661		 * folio release code relies on it.
1662		 */
1663		if (unlikely(!folio_try_get(folio)))
1664			goto move;
1665
1666		if (!folio_test_clear_lru(folio)) {
1667			/* Another thread is already isolating this folio */
1668			folio_put(folio);
1669			goto move;
1670		}
1671
1672		nr_taken += nr_pages;
1673		nr_zone_taken[folio_zonenum(folio)] += nr_pages;
1674		move_to = dst;
1675move:
1676		list_move(&folio->lru, move_to);
1677	}
1678
1679	/*
1680	 * Splice any skipped folios to the start of the LRU list. Note that
1681	 * this disrupts the LRU order when reclaiming for lower zones but
1682	 * we cannot splice to the tail. If we did then the SWAP_CLUSTER_MAX
1683	 * scanning would soon rescan the same folios to skip and waste lots
1684	 * of cpu cycles.
1685	 */
1686	if (!list_empty(&folios_skipped)) {
1687		int zid;
1688
1689		list_splice(&folios_skipped, src);
1690		for (zid = 0; zid < MAX_NR_ZONES; zid++) {
1691			if (!nr_skipped[zid])
1692				continue;
1693
1694			__count_zid_vm_events(PGSCAN_SKIP, zid, nr_skipped[zid]);
1695			skipped += nr_skipped[zid];
1696		}
1697	}
1698	*nr_scanned = total_scan;
1699	trace_mm_vmscan_lru_isolate(sc->reclaim_idx, sc->order, nr_to_scan,
1700				    total_scan, skipped, nr_taken, lru);
1701	update_lru_sizes(lruvec, lru, nr_zone_taken);
1702	return nr_taken;
1703}
1704
1705/**
1706 * folio_isolate_lru() - Try to isolate a folio from its LRU list.
1707 * @folio: Folio to isolate from its LRU list.
1708 *
1709 * Isolate a @folio from an LRU list and adjust the vmstat statistic
1710 * corresponding to whatever LRU list the folio was on.
1711 *
1712 * The folio will have its LRU flag cleared.  If it was found on the
1713 * active list, it will have the Active flag set.  If it was found on the
1714 * unevictable list, it will have the Unevictable flag set.  These flags
1715 * may need to be cleared by the caller before letting the page go.
1716 *
1717 * Context:
1718 *
1719 * (1) Must be called with an elevated refcount on the folio. This is a
1720 *     fundamental difference from isolate_lru_folios() (which is called
1721 *     without a stable reference).
1722 * (2) The lru_lock must not be held.
1723 * (3) Interrupts must be enabled.
1724 *
1725 * Return: true if the folio was removed from an LRU list.
1726 * false if the folio was not on an LRU list.
1727 */
1728bool folio_isolate_lru(struct folio *folio)
1729{
1730	bool ret = false;
1731
1732	VM_BUG_ON_FOLIO(!folio_ref_count(folio), folio);
1733
1734	if (folio_test_clear_lru(folio)) {
1735		struct lruvec *lruvec;
1736
1737		folio_get(folio);
1738		lruvec = folio_lruvec_lock_irq(folio);
1739		lruvec_del_folio(lruvec, folio);
1740		unlock_page_lruvec_irq(lruvec);
1741		ret = true;
1742	}
1743
1744	return ret;
1745}
1746
1747/*
1748 * A direct reclaimer may isolate SWAP_CLUSTER_MAX pages from the LRU list and
1749 * then get rescheduled. When there are massive number of tasks doing page
1750 * allocation, such sleeping direct reclaimers may keep piling up on each CPU,
1751 * the LRU list will go small and be scanned faster than necessary, leading to
1752 * unnecessary swapping, thrashing and OOM.
1753 */
1754static bool too_many_isolated(struct pglist_data *pgdat, int file,
1755		struct scan_control *sc)
1756{
1757	unsigned long inactive, isolated;
1758	bool too_many;
1759
1760	if (current_is_kswapd())
1761		return false;
1762
1763	if (!writeback_throttling_sane(sc))
1764		return false;
1765
1766	if (file) {
1767		inactive = node_page_state(pgdat, NR_INACTIVE_FILE);
1768		isolated = node_page_state(pgdat, NR_ISOLATED_FILE);
1769	} else {
1770		inactive = node_page_state(pgdat, NR_INACTIVE_ANON);
1771		isolated = node_page_state(pgdat, NR_ISOLATED_ANON);
1772	}
1773
1774	/*
1775	 * GFP_NOIO/GFP_NOFS callers are allowed to isolate more pages, so they
1776	 * won't get blocked by normal direct-reclaimers, forming a circular
1777	 * deadlock.
1778	 */
1779	if (gfp_has_io_fs(sc->gfp_mask))
1780		inactive >>= 3;
1781
1782	too_many = isolated > inactive;
1783
1784	/* Wake up tasks throttled due to too_many_isolated. */
1785	if (!too_many)
1786		wake_throttle_isolated(pgdat);
1787
1788	return too_many;
1789}
1790
1791/*
1792 * move_folios_to_lru() moves folios from private @list to appropriate LRU list.
 
1793 *
1794 * Returns the number of pages moved to the given lruvec.
1795 */
1796static unsigned int move_folios_to_lru(struct lruvec *lruvec,
1797		struct list_head *list)
1798{
1799	int nr_pages, nr_moved = 0;
1800	struct folio_batch free_folios;
1801
1802	folio_batch_init(&free_folios);
1803	while (!list_empty(list)) {
1804		struct folio *folio = lru_to_folio(list);
1805
1806		VM_BUG_ON_FOLIO(folio_test_lru(folio), folio);
1807		list_del(&folio->lru);
1808		if (unlikely(!folio_evictable(folio))) {
1809			spin_unlock_irq(&lruvec->lru_lock);
1810			folio_putback_lru(folio);
1811			spin_lock_irq(&lruvec->lru_lock);
1812			continue;
1813		}
1814
1815		/*
1816		 * The folio_set_lru needs to be kept here for list integrity.
1817		 * Otherwise:
1818		 *   #0 move_folios_to_lru             #1 release_pages
1819		 *   if (!folio_put_testzero())
1820		 *				      if (folio_put_testzero())
1821		 *				        !lru //skip lru_lock
1822		 *     folio_set_lru()
1823		 *     list_add(&folio->lru,)
1824		 *                                        list_add(&folio->lru,)
1825		 */
1826		folio_set_lru(folio);
1827
1828		if (unlikely(folio_put_testzero(folio))) {
1829			__folio_clear_lru_flags(folio);
1830
1831			if (folio_test_large(folio) &&
1832			    folio_test_large_rmappable(folio))
1833				folio_undo_large_rmappable(folio);
1834			if (folio_batch_add(&free_folios, folio) == 0) {
1835				spin_unlock_irq(&lruvec->lru_lock);
1836				mem_cgroup_uncharge_folios(&free_folios);
1837				free_unref_folios(&free_folios);
1838				spin_lock_irq(&lruvec->lru_lock);
1839			}
 
1840
1841			continue;
1842		}
1843
1844		/*
1845		 * All pages were isolated from the same lruvec (and isolation
1846		 * inhibits memcg migration).
1847		 */
1848		VM_BUG_ON_FOLIO(!folio_matches_lruvec(folio, lruvec), folio);
1849		lruvec_add_folio(lruvec, folio);
1850		nr_pages = folio_nr_pages(folio);
1851		nr_moved += nr_pages;
1852		if (folio_test_active(folio))
1853			workingset_age_nonresident(lruvec, nr_pages);
1854	}
1855
1856	if (free_folios.nr) {
1857		spin_unlock_irq(&lruvec->lru_lock);
1858		mem_cgroup_uncharge_folios(&free_folios);
1859		free_unref_folios(&free_folios);
1860		spin_lock_irq(&lruvec->lru_lock);
1861	}
1862
1863	return nr_moved;
1864}
1865
1866/*
1867 * If a kernel thread (such as nfsd for loop-back mounts) services a backing
1868 * device by writing to the page cache it sets PF_LOCAL_THROTTLE. In this case
1869 * we should not throttle.  Otherwise it is safe to do so.
1870 */
1871static int current_may_throttle(void)
1872{
1873	return !(current->flags & PF_LOCAL_THROTTLE);
1874}
1875
1876/*
1877 * shrink_inactive_list() is a helper for shrink_node().  It returns the number
1878 * of reclaimed pages
1879 */
1880static unsigned long shrink_inactive_list(unsigned long nr_to_scan,
1881		struct lruvec *lruvec, struct scan_control *sc,
1882		enum lru_list lru)
1883{
1884	LIST_HEAD(folio_list);
1885	unsigned long nr_scanned;
1886	unsigned int nr_reclaimed = 0;
1887	unsigned long nr_taken;
1888	struct reclaim_stat stat;
1889	bool file = is_file_lru(lru);
1890	enum vm_event_item item;
1891	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
1892	bool stalled = false;
1893
1894	while (unlikely(too_many_isolated(pgdat, file, sc))) {
1895		if (stalled)
1896			return 0;
1897
1898		/* wait a bit for the reclaimer. */
1899		stalled = true;
1900		reclaim_throttle(pgdat, VMSCAN_THROTTLE_ISOLATED);
1901
1902		/* We are about to die and free our memory. Return now. */
1903		if (fatal_signal_pending(current))
1904			return SWAP_CLUSTER_MAX;
1905	}
1906
1907	lru_add_drain();
1908
1909	spin_lock_irq(&lruvec->lru_lock);
1910
1911	nr_taken = isolate_lru_folios(nr_to_scan, lruvec, &folio_list,
1912				     &nr_scanned, sc, lru);
1913
1914	__mod_node_page_state(pgdat, NR_ISOLATED_ANON + file, nr_taken);
1915	item = PGSCAN_KSWAPD + reclaimer_offset();
1916	if (!cgroup_reclaim(sc))
1917		__count_vm_events(item, nr_scanned);
1918	__count_memcg_events(lruvec_memcg(lruvec), item, nr_scanned);
1919	__count_vm_events(PGSCAN_ANON + file, nr_scanned);
1920
1921	spin_unlock_irq(&lruvec->lru_lock);
1922
1923	if (nr_taken == 0)
1924		return 0;
1925
1926	nr_reclaimed = shrink_folio_list(&folio_list, pgdat, sc, &stat, false);
1927
1928	spin_lock_irq(&lruvec->lru_lock);
1929	move_folios_to_lru(lruvec, &folio_list);
1930
1931	__mod_node_page_state(pgdat, NR_ISOLATED_ANON + file, -nr_taken);
1932	item = PGSTEAL_KSWAPD + reclaimer_offset();
1933	if (!cgroup_reclaim(sc))
1934		__count_vm_events(item, nr_reclaimed);
1935	__count_memcg_events(lruvec_memcg(lruvec), item, nr_reclaimed);
1936	__count_vm_events(PGSTEAL_ANON + file, nr_reclaimed);
1937	spin_unlock_irq(&lruvec->lru_lock);
1938
1939	lru_note_cost(lruvec, file, stat.nr_pageout, nr_scanned - nr_reclaimed);
 
 
1940
1941	/*
1942	 * If dirty folios are scanned that are not queued for IO, it
1943	 * implies that flushers are not doing their job. This can
1944	 * happen when memory pressure pushes dirty folios to the end of
1945	 * the LRU before the dirty limits are breached and the dirty
1946	 * data has expired. It can also happen when the proportion of
1947	 * dirty folios grows not through writes but through memory
1948	 * pressure reclaiming all the clean cache. And in some cases,
1949	 * the flushers simply cannot keep up with the allocation
1950	 * rate. Nudge the flusher threads in case they are asleep.
1951	 */
1952	if (stat.nr_unqueued_dirty == nr_taken) {
1953		wakeup_flusher_threads(WB_REASON_VMSCAN);
1954		/*
1955		 * For cgroupv1 dirty throttling is achieved by waking up
1956		 * the kernel flusher here and later waiting on folios
1957		 * which are in writeback to finish (see shrink_folio_list()).
1958		 *
1959		 * Flusher may not be able to issue writeback quickly
1960		 * enough for cgroupv1 writeback throttling to work
1961		 * on a large system.
1962		 */
1963		if (!writeback_throttling_sane(sc))
1964			reclaim_throttle(pgdat, VMSCAN_THROTTLE_WRITEBACK);
1965	}
1966
1967	sc->nr.dirty += stat.nr_dirty;
1968	sc->nr.congested += stat.nr_congested;
1969	sc->nr.unqueued_dirty += stat.nr_unqueued_dirty;
1970	sc->nr.writeback += stat.nr_writeback;
1971	sc->nr.immediate += stat.nr_immediate;
1972	sc->nr.taken += nr_taken;
1973	if (file)
1974		sc->nr.file_taken += nr_taken;
1975
1976	trace_mm_vmscan_lru_shrink_inactive(pgdat->node_id,
1977			nr_scanned, nr_reclaimed, &stat, sc->priority, file);
1978	return nr_reclaimed;
1979}
1980
1981/*
1982 * shrink_active_list() moves folios from the active LRU to the inactive LRU.
1983 *
1984 * We move them the other way if the folio is referenced by one or more
1985 * processes.
1986 *
1987 * If the folios are mostly unmapped, the processing is fast and it is
1988 * appropriate to hold lru_lock across the whole operation.  But if
1989 * the folios are mapped, the processing is slow (folio_referenced()), so
1990 * we should drop lru_lock around each folio.  It's impossible to balance
1991 * this, so instead we remove the folios from the LRU while processing them.
1992 * It is safe to rely on the active flag against the non-LRU folios in here
1993 * because nobody will play with that bit on a non-LRU folio.
1994 *
1995 * The downside is that we have to touch folio->_refcount against each folio.
1996 * But we had to alter folio->flags anyway.
1997 */
1998static void shrink_active_list(unsigned long nr_to_scan,
1999			       struct lruvec *lruvec,
2000			       struct scan_control *sc,
2001			       enum lru_list lru)
2002{
2003	unsigned long nr_taken;
2004	unsigned long nr_scanned;
2005	unsigned long vm_flags;
2006	LIST_HEAD(l_hold);	/* The folios which were snipped off */
2007	LIST_HEAD(l_active);
2008	LIST_HEAD(l_inactive);
2009	unsigned nr_deactivate, nr_activate;
2010	unsigned nr_rotated = 0;
2011	bool file = is_file_lru(lru);
2012	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
2013
2014	lru_add_drain();
2015
2016	spin_lock_irq(&lruvec->lru_lock);
2017
2018	nr_taken = isolate_lru_folios(nr_to_scan, lruvec, &l_hold,
2019				     &nr_scanned, sc, lru);
2020
2021	__mod_node_page_state(pgdat, NR_ISOLATED_ANON + file, nr_taken);
2022
2023	if (!cgroup_reclaim(sc))
2024		__count_vm_events(PGREFILL, nr_scanned);
2025	__count_memcg_events(lruvec_memcg(lruvec), PGREFILL, nr_scanned);
2026
2027	spin_unlock_irq(&lruvec->lru_lock);
2028
2029	while (!list_empty(&l_hold)) {
2030		struct folio *folio;
2031
2032		cond_resched();
2033		folio = lru_to_folio(&l_hold);
2034		list_del(&folio->lru);
2035
2036		if (unlikely(!folio_evictable(folio))) {
2037			folio_putback_lru(folio);
2038			continue;
2039		}
2040
2041		if (unlikely(buffer_heads_over_limit)) {
2042			if (folio_needs_release(folio) &&
2043			    folio_trylock(folio)) {
2044				filemap_release_folio(folio, 0);
2045				folio_unlock(folio);
2046			}
2047		}
2048
2049		/* Referenced or rmap lock contention: rotate */
2050		if (folio_referenced(folio, 0, sc->target_mem_cgroup,
2051				     &vm_flags) != 0) {
2052			/*
2053			 * Identify referenced, file-backed active folios and
2054			 * give them one more trip around the active list. So
2055			 * that executable code get better chances to stay in
2056			 * memory under moderate memory pressure.  Anon folios
2057			 * are not likely to be evicted by use-once streaming
2058			 * IO, plus JVM can create lots of anon VM_EXEC folios,
2059			 * so we ignore them here.
2060			 */
2061			if ((vm_flags & VM_EXEC) && folio_is_file_lru(folio)) {
2062				nr_rotated += folio_nr_pages(folio);
2063				list_add(&folio->lru, &l_active);
2064				continue;
2065			}
2066		}
2067
2068		folio_clear_active(folio);	/* we are de-activating */
2069		folio_set_workingset(folio);
2070		list_add(&folio->lru, &l_inactive);
2071	}
2072
2073	/*
2074	 * Move folios back to the lru list.
2075	 */
2076	spin_lock_irq(&lruvec->lru_lock);
2077
2078	nr_activate = move_folios_to_lru(lruvec, &l_active);
2079	nr_deactivate = move_folios_to_lru(lruvec, &l_inactive);
 
 
2080
2081	__count_vm_events(PGDEACTIVATE, nr_deactivate);
2082	__count_memcg_events(lruvec_memcg(lruvec), PGDEACTIVATE, nr_deactivate);
2083
2084	__mod_node_page_state(pgdat, NR_ISOLATED_ANON + file, -nr_taken);
2085	spin_unlock_irq(&lruvec->lru_lock);
2086
2087	if (nr_rotated)
2088		lru_note_cost(lruvec, file, 0, nr_rotated);
 
 
2089	trace_mm_vmscan_lru_shrink_active(pgdat->node_id, nr_taken, nr_activate,
2090			nr_deactivate, nr_rotated, sc->priority, file);
2091}
2092
2093static unsigned int reclaim_folio_list(struct list_head *folio_list,
2094				      struct pglist_data *pgdat,
2095				      bool ignore_references)
2096{
2097	struct reclaim_stat dummy_stat;
2098	unsigned int nr_reclaimed;
2099	struct folio *folio;
2100	struct scan_control sc = {
2101		.gfp_mask = GFP_KERNEL,
2102		.may_writepage = 1,
2103		.may_unmap = 1,
2104		.may_swap = 1,
2105		.no_demotion = 1,
2106	};
2107
2108	nr_reclaimed = shrink_folio_list(folio_list, pgdat, &sc, &dummy_stat, ignore_references);
2109	while (!list_empty(folio_list)) {
2110		folio = lru_to_folio(folio_list);
2111		list_del(&folio->lru);
2112		folio_putback_lru(folio);
2113	}
2114
2115	return nr_reclaimed;
2116}
2117
2118unsigned long reclaim_pages(struct list_head *folio_list, bool ignore_references)
2119{
2120	int nid;
2121	unsigned int nr_reclaimed = 0;
2122	LIST_HEAD(node_folio_list);
2123	unsigned int noreclaim_flag;
2124
2125	if (list_empty(folio_list))
2126		return nr_reclaimed;
2127
2128	noreclaim_flag = memalloc_noreclaim_save();
2129
2130	nid = folio_nid(lru_to_folio(folio_list));
2131	do {
2132		struct folio *folio = lru_to_folio(folio_list);
2133
2134		if (nid == folio_nid(folio)) {
2135			folio_clear_active(folio);
2136			list_move(&folio->lru, &node_folio_list);
2137			continue;
2138		}
2139
2140		nr_reclaimed += reclaim_folio_list(&node_folio_list, NODE_DATA(nid),
2141						   ignore_references);
2142		nid = folio_nid(lru_to_folio(folio_list));
2143	} while (!list_empty(folio_list));
2144
2145	nr_reclaimed += reclaim_folio_list(&node_folio_list, NODE_DATA(nid), ignore_references);
2146
2147	memalloc_noreclaim_restore(noreclaim_flag);
2148
2149	return nr_reclaimed;
2150}
2151
2152static unsigned long shrink_list(enum lru_list lru, unsigned long nr_to_scan,
2153				 struct lruvec *lruvec, struct scan_control *sc)
2154{
2155	if (is_active_lru(lru)) {
2156		if (sc->may_deactivate & (1 << is_file_lru(lru)))
2157			shrink_active_list(nr_to_scan, lruvec, sc, lru);
2158		else
2159			sc->skipped_deactivate = 1;
2160		return 0;
2161	}
2162
2163	return shrink_inactive_list(nr_to_scan, lruvec, sc, lru);
2164}
2165
2166/*
2167 * The inactive anon list should be small enough that the VM never has
2168 * to do too much work.
2169 *
2170 * The inactive file list should be small enough to leave most memory
2171 * to the established workingset on the scan-resistant active list,
2172 * but large enough to avoid thrashing the aggregate readahead window.
2173 *
2174 * Both inactive lists should also be large enough that each inactive
2175 * folio has a chance to be referenced again before it is reclaimed.
2176 *
2177 * If that fails and refaulting is observed, the inactive list grows.
2178 *
2179 * The inactive_ratio is the target ratio of ACTIVE to INACTIVE folios
2180 * on this LRU, maintained by the pageout code. An inactive_ratio
2181 * of 3 means 3:1 or 25% of the folios are kept on the inactive list.
2182 *
2183 * total     target    max
2184 * memory    ratio     inactive
2185 * -------------------------------------
2186 *   10MB       1         5MB
2187 *  100MB       1        50MB
2188 *    1GB       3       250MB
2189 *   10GB      10       0.9GB
2190 *  100GB      31         3GB
2191 *    1TB     101        10GB
2192 *   10TB     320        32GB
2193 */
2194static bool inactive_is_low(struct lruvec *lruvec, enum lru_list inactive_lru)
2195{
2196	enum lru_list active_lru = inactive_lru + LRU_ACTIVE;
2197	unsigned long inactive, active;
2198	unsigned long inactive_ratio;
2199	unsigned long gb;
2200
2201	inactive = lruvec_page_state(lruvec, NR_LRU_BASE + inactive_lru);
2202	active = lruvec_page_state(lruvec, NR_LRU_BASE + active_lru);
2203
2204	gb = (inactive + active) >> (30 - PAGE_SHIFT);
2205	if (gb)
2206		inactive_ratio = int_sqrt(10 * gb);
2207	else
2208		inactive_ratio = 1;
2209
2210	return inactive * inactive_ratio < active;
2211}
2212
2213enum scan_balance {
2214	SCAN_EQUAL,
2215	SCAN_FRACT,
2216	SCAN_ANON,
2217	SCAN_FILE,
2218};
2219
2220static void prepare_scan_control(pg_data_t *pgdat, struct scan_control *sc)
2221{
2222	unsigned long file;
2223	struct lruvec *target_lruvec;
2224
2225	if (lru_gen_enabled())
2226		return;
2227
2228	target_lruvec = mem_cgroup_lruvec(sc->target_mem_cgroup, pgdat);
2229
2230	/*
2231	 * Flush the memory cgroup stats, so that we read accurate per-memcg
2232	 * lruvec stats for heuristics.
2233	 */
2234	mem_cgroup_flush_stats(sc->target_mem_cgroup);
2235
2236	/*
2237	 * Determine the scan balance between anon and file LRUs.
2238	 */
2239	spin_lock_irq(&target_lruvec->lru_lock);
2240	sc->anon_cost = target_lruvec->anon_cost;
2241	sc->file_cost = target_lruvec->file_cost;
2242	spin_unlock_irq(&target_lruvec->lru_lock);
2243
2244	/*
2245	 * Target desirable inactive:active list ratios for the anon
2246	 * and file LRU lists.
2247	 */
2248	if (!sc->force_deactivate) {
2249		unsigned long refaults;
2250
2251		/*
2252		 * When refaults are being observed, it means a new
2253		 * workingset is being established. Deactivate to get
2254		 * rid of any stale active pages quickly.
2255		 */
2256		refaults = lruvec_page_state(target_lruvec,
2257				WORKINGSET_ACTIVATE_ANON);
2258		if (refaults != target_lruvec->refaults[WORKINGSET_ANON] ||
2259			inactive_is_low(target_lruvec, LRU_INACTIVE_ANON))
2260			sc->may_deactivate |= DEACTIVATE_ANON;
2261		else
2262			sc->may_deactivate &= ~DEACTIVATE_ANON;
2263
2264		refaults = lruvec_page_state(target_lruvec,
2265				WORKINGSET_ACTIVATE_FILE);
2266		if (refaults != target_lruvec->refaults[WORKINGSET_FILE] ||
2267		    inactive_is_low(target_lruvec, LRU_INACTIVE_FILE))
2268			sc->may_deactivate |= DEACTIVATE_FILE;
2269		else
2270			sc->may_deactivate &= ~DEACTIVATE_FILE;
2271	} else
2272		sc->may_deactivate = DEACTIVATE_ANON | DEACTIVATE_FILE;
2273
2274	/*
2275	 * If we have plenty of inactive file pages that aren't
2276	 * thrashing, try to reclaim those first before touching
2277	 * anonymous pages.
2278	 */
2279	file = lruvec_page_state(target_lruvec, NR_INACTIVE_FILE);
2280	if (file >> sc->priority && !(sc->may_deactivate & DEACTIVATE_FILE) &&
2281	    !sc->no_cache_trim_mode)
2282		sc->cache_trim_mode = 1;
2283	else
2284		sc->cache_trim_mode = 0;
2285
2286	/*
2287	 * Prevent the reclaimer from falling into the cache trap: as
2288	 * cache pages start out inactive, every cache fault will tip
2289	 * the scan balance towards the file LRU.  And as the file LRU
2290	 * shrinks, so does the window for rotation from references.
2291	 * This means we have a runaway feedback loop where a tiny
2292	 * thrashing file LRU becomes infinitely more attractive than
2293	 * anon pages.  Try to detect this based on file LRU size.
2294	 */
2295	if (!cgroup_reclaim(sc)) {
2296		unsigned long total_high_wmark = 0;
2297		unsigned long free, anon;
2298		int z;
2299
2300		free = sum_zone_node_page_state(pgdat->node_id, NR_FREE_PAGES);
2301		file = node_page_state(pgdat, NR_ACTIVE_FILE) +
2302			   node_page_state(pgdat, NR_INACTIVE_FILE);
2303
2304		for (z = 0; z < MAX_NR_ZONES; z++) {
2305			struct zone *zone = &pgdat->node_zones[z];
2306
2307			if (!managed_zone(zone))
2308				continue;
2309
2310			total_high_wmark += high_wmark_pages(zone);
2311		}
2312
2313		/*
2314		 * Consider anon: if that's low too, this isn't a
2315		 * runaway file reclaim problem, but rather just
2316		 * extreme pressure. Reclaim as per usual then.
2317		 */
2318		anon = node_page_state(pgdat, NR_INACTIVE_ANON);
2319
2320		sc->file_is_tiny =
2321			file + free <= total_high_wmark &&
2322			!(sc->may_deactivate & DEACTIVATE_ANON) &&
2323			anon >> sc->priority;
2324	}
2325}
2326
2327/*
2328 * Determine how aggressively the anon and file LRU lists should be
2329 * scanned.
2330 *
2331 * nr[0] = anon inactive folios to scan; nr[1] = anon active folios to scan
2332 * nr[2] = file inactive folios to scan; nr[3] = file active folios to scan
2333 */
2334static void get_scan_count(struct lruvec *lruvec, struct scan_control *sc,
2335			   unsigned long *nr)
2336{
2337	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
2338	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
2339	unsigned long anon_cost, file_cost, total_cost;
2340	int swappiness = mem_cgroup_swappiness(memcg);
2341	u64 fraction[ANON_AND_FILE];
2342	u64 denominator = 0;	/* gcc */
2343	enum scan_balance scan_balance;
2344	unsigned long ap, fp;
2345	enum lru_list lru;
2346
2347	/* If we have no swap space, do not bother scanning anon folios. */
2348	if (!sc->may_swap || !can_reclaim_anon_pages(memcg, pgdat->node_id, sc)) {
2349		scan_balance = SCAN_FILE;
2350		goto out;
2351	}
2352
2353	/*
2354	 * Global reclaim will swap to prevent OOM even with no
2355	 * swappiness, but memcg users want to use this knob to
2356	 * disable swapping for individual groups completely when
2357	 * using the memory controller's swap limit feature would be
2358	 * too expensive.
2359	 */
2360	if (cgroup_reclaim(sc) && !swappiness) {
2361		scan_balance = SCAN_FILE;
2362		goto out;
2363	}
2364
2365	/*
2366	 * Do not apply any pressure balancing cleverness when the
2367	 * system is close to OOM, scan both anon and file equally
2368	 * (unless the swappiness setting disagrees with swapping).
2369	 */
2370	if (!sc->priority && swappiness) {
2371		scan_balance = SCAN_EQUAL;
2372		goto out;
2373	}
2374
2375	/*
2376	 * If the system is almost out of file pages, force-scan anon.
2377	 */
2378	if (sc->file_is_tiny) {
2379		scan_balance = SCAN_ANON;
2380		goto out;
2381	}
2382
2383	/*
2384	 * If there is enough inactive page cache, we do not reclaim
2385	 * anything from the anonymous working right now.
2386	 */
2387	if (sc->cache_trim_mode) {
2388		scan_balance = SCAN_FILE;
2389		goto out;
2390	}
2391
2392	scan_balance = SCAN_FRACT;
2393	/*
2394	 * Calculate the pressure balance between anon and file pages.
2395	 *
2396	 * The amount of pressure we put on each LRU is inversely
2397	 * proportional to the cost of reclaiming each list, as
2398	 * determined by the share of pages that are refaulting, times
2399	 * the relative IO cost of bringing back a swapped out
2400	 * anonymous page vs reloading a filesystem page (swappiness).
2401	 *
2402	 * Although we limit that influence to ensure no list gets
2403	 * left behind completely: at least a third of the pressure is
2404	 * applied, before swappiness.
2405	 *
2406	 * With swappiness at 100, anon and file have equal IO cost.
2407	 */
2408	total_cost = sc->anon_cost + sc->file_cost;
2409	anon_cost = total_cost + sc->anon_cost;
2410	file_cost = total_cost + sc->file_cost;
2411	total_cost = anon_cost + file_cost;
2412
2413	ap = swappiness * (total_cost + 1);
2414	ap /= anon_cost + 1;
2415
2416	fp = (200 - swappiness) * (total_cost + 1);
2417	fp /= file_cost + 1;
2418
2419	fraction[0] = ap;
2420	fraction[1] = fp;
2421	denominator = ap + fp;
2422out:
2423	for_each_evictable_lru(lru) {
2424		bool file = is_file_lru(lru);
2425		unsigned long lruvec_size;
2426		unsigned long low, min;
2427		unsigned long scan;
2428
2429		lruvec_size = lruvec_lru_size(lruvec, lru, sc->reclaim_idx);
2430		mem_cgroup_protection(sc->target_mem_cgroup, memcg,
2431				      &min, &low);
2432
2433		if (min || low) {
2434			/*
2435			 * Scale a cgroup's reclaim pressure by proportioning
2436			 * its current usage to its memory.low or memory.min
2437			 * setting.
2438			 *
2439			 * This is important, as otherwise scanning aggression
2440			 * becomes extremely binary -- from nothing as we
2441			 * approach the memory protection threshold, to totally
2442			 * nominal as we exceed it.  This results in requiring
2443			 * setting extremely liberal protection thresholds. It
2444			 * also means we simply get no protection at all if we
2445			 * set it too low, which is not ideal.
2446			 *
2447			 * If there is any protection in place, we reduce scan
2448			 * pressure by how much of the total memory used is
2449			 * within protection thresholds.
2450			 *
2451			 * There is one special case: in the first reclaim pass,
2452			 * we skip over all groups that are within their low
2453			 * protection. If that fails to reclaim enough pages to
2454			 * satisfy the reclaim goal, we come back and override
2455			 * the best-effort low protection. However, we still
2456			 * ideally want to honor how well-behaved groups are in
2457			 * that case instead of simply punishing them all
2458			 * equally. As such, we reclaim them based on how much
2459			 * memory they are using, reducing the scan pressure
2460			 * again by how much of the total memory used is under
2461			 * hard protection.
2462			 */
2463			unsigned long cgroup_size = mem_cgroup_size(memcg);
2464			unsigned long protection;
2465
2466			/* memory.low scaling, make sure we retry before OOM */
2467			if (!sc->memcg_low_reclaim && low > min) {
2468				protection = low;
2469				sc->memcg_low_skipped = 1;
2470			} else {
2471				protection = min;
2472			}
2473
2474			/* Avoid TOCTOU with earlier protection check */
2475			cgroup_size = max(cgroup_size, protection);
2476
2477			scan = lruvec_size - lruvec_size * protection /
2478				(cgroup_size + 1);
2479
2480			/*
2481			 * Minimally target SWAP_CLUSTER_MAX pages to keep
2482			 * reclaim moving forwards, avoiding decrementing
2483			 * sc->priority further than desirable.
2484			 */
2485			scan = max(scan, SWAP_CLUSTER_MAX);
2486		} else {
2487			scan = lruvec_size;
2488		}
2489
2490		scan >>= sc->priority;
2491
2492		/*
2493		 * If the cgroup's already been deleted, make sure to
2494		 * scrape out the remaining cache.
2495		 */
2496		if (!scan && !mem_cgroup_online(memcg))
2497			scan = min(lruvec_size, SWAP_CLUSTER_MAX);
2498
2499		switch (scan_balance) {
2500		case SCAN_EQUAL:
2501			/* Scan lists relative to size */
2502			break;
2503		case SCAN_FRACT:
2504			/*
2505			 * Scan types proportional to swappiness and
2506			 * their relative recent reclaim efficiency.
2507			 * Make sure we don't miss the last page on
2508			 * the offlined memory cgroups because of a
2509			 * round-off error.
2510			 */
2511			scan = mem_cgroup_online(memcg) ?
2512			       div64_u64(scan * fraction[file], denominator) :
2513			       DIV64_U64_ROUND_UP(scan * fraction[file],
2514						  denominator);
2515			break;
2516		case SCAN_FILE:
2517		case SCAN_ANON:
2518			/* Scan one type exclusively */
2519			if ((scan_balance == SCAN_FILE) != file)
2520				scan = 0;
2521			break;
2522		default:
2523			/* Look ma, no brain */
2524			BUG();
2525		}
2526
2527		nr[lru] = scan;
2528	}
2529}
2530
2531/*
2532 * Anonymous LRU management is a waste if there is
2533 * ultimately no way to reclaim the memory.
2534 */
2535static bool can_age_anon_pages(struct pglist_data *pgdat,
2536			       struct scan_control *sc)
2537{
2538	/* Aging the anon LRU is valuable if swap is present: */
2539	if (total_swap_pages > 0)
2540		return true;
2541
2542	/* Also valuable if anon pages can be demoted: */
2543	return can_demote(pgdat->node_id, sc);
2544}
2545
2546#ifdef CONFIG_LRU_GEN
2547
2548#ifdef CONFIG_LRU_GEN_ENABLED
2549DEFINE_STATIC_KEY_ARRAY_TRUE(lru_gen_caps, NR_LRU_GEN_CAPS);
2550#define get_cap(cap)	static_branch_likely(&lru_gen_caps[cap])
2551#else
2552DEFINE_STATIC_KEY_ARRAY_FALSE(lru_gen_caps, NR_LRU_GEN_CAPS);
2553#define get_cap(cap)	static_branch_unlikely(&lru_gen_caps[cap])
2554#endif
2555
2556static bool should_walk_mmu(void)
2557{
2558	return arch_has_hw_pte_young() && get_cap(LRU_GEN_MM_WALK);
2559}
2560
2561static bool should_clear_pmd_young(void)
2562{
2563	return arch_has_hw_nonleaf_pmd_young() && get_cap(LRU_GEN_NONLEAF_YOUNG);
2564}
2565
2566/******************************************************************************
2567 *                          shorthand helpers
2568 ******************************************************************************/
2569
2570#define LRU_REFS_FLAGS	(BIT(PG_referenced) | BIT(PG_workingset))
2571
2572#define DEFINE_MAX_SEQ(lruvec)						\
2573	unsigned long max_seq = READ_ONCE((lruvec)->lrugen.max_seq)
2574
2575#define DEFINE_MIN_SEQ(lruvec)						\
2576	unsigned long min_seq[ANON_AND_FILE] = {			\
2577		READ_ONCE((lruvec)->lrugen.min_seq[LRU_GEN_ANON]),	\
2578		READ_ONCE((lruvec)->lrugen.min_seq[LRU_GEN_FILE]),	\
2579	}
2580
2581#define for_each_gen_type_zone(gen, type, zone)				\
2582	for ((gen) = 0; (gen) < MAX_NR_GENS; (gen)++)			\
2583		for ((type) = 0; (type) < ANON_AND_FILE; (type)++)	\
2584			for ((zone) = 0; (zone) < MAX_NR_ZONES; (zone)++)
2585
2586#define get_memcg_gen(seq)	((seq) % MEMCG_NR_GENS)
2587#define get_memcg_bin(bin)	((bin) % MEMCG_NR_BINS)
2588
2589static struct lruvec *get_lruvec(struct mem_cgroup *memcg, int nid)
2590{
2591	struct pglist_data *pgdat = NODE_DATA(nid);
2592
2593#ifdef CONFIG_MEMCG
2594	if (memcg) {
2595		struct lruvec *lruvec = &memcg->nodeinfo[nid]->lruvec;
2596
2597		/* see the comment in mem_cgroup_lruvec() */
2598		if (!lruvec->pgdat)
2599			lruvec->pgdat = pgdat;
2600
2601		return lruvec;
2602	}
2603#endif
2604	VM_WARN_ON_ONCE(!mem_cgroup_disabled());
2605
2606	return &pgdat->__lruvec;
2607}
2608
2609static int get_swappiness(struct lruvec *lruvec, struct scan_control *sc)
2610{
2611	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
2612	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
2613
2614	if (!sc->may_swap)
2615		return 0;
2616
2617	if (!can_demote(pgdat->node_id, sc) &&
2618	    mem_cgroup_get_nr_swap_pages(memcg) < MIN_LRU_BATCH)
2619		return 0;
2620
2621	return mem_cgroup_swappiness(memcg);
2622}
2623
2624static int get_nr_gens(struct lruvec *lruvec, int type)
2625{
2626	return lruvec->lrugen.max_seq - lruvec->lrugen.min_seq[type] + 1;
2627}
2628
2629static bool __maybe_unused seq_is_valid(struct lruvec *lruvec)
2630{
2631	/* see the comment on lru_gen_folio */
2632	return get_nr_gens(lruvec, LRU_GEN_FILE) >= MIN_NR_GENS &&
2633	       get_nr_gens(lruvec, LRU_GEN_FILE) <= get_nr_gens(lruvec, LRU_GEN_ANON) &&
2634	       get_nr_gens(lruvec, LRU_GEN_ANON) <= MAX_NR_GENS;
2635}
2636
2637/******************************************************************************
2638 *                          Bloom filters
2639 ******************************************************************************/
2640
2641/*
2642 * Bloom filters with m=1<<15, k=2 and the false positive rates of ~1/5 when
2643 * n=10,000 and ~1/2 when n=20,000, where, conventionally, m is the number of
2644 * bits in a bitmap, k is the number of hash functions and n is the number of
2645 * inserted items.
2646 *
2647 * Page table walkers use one of the two filters to reduce their search space.
2648 * To get rid of non-leaf entries that no longer have enough leaf entries, the
2649 * aging uses the double-buffering technique to flip to the other filter each
2650 * time it produces a new generation. For non-leaf entries that have enough
2651 * leaf entries, the aging carries them over to the next generation in
2652 * walk_pmd_range(); the eviction also report them when walking the rmap
2653 * in lru_gen_look_around().
2654 *
2655 * For future optimizations:
2656 * 1. It's not necessary to keep both filters all the time. The spare one can be
2657 *    freed after the RCU grace period and reallocated if needed again.
2658 * 2. And when reallocating, it's worth scaling its size according to the number
2659 *    of inserted entries in the other filter, to reduce the memory overhead on
2660 *    small systems and false positives on large systems.
2661 * 3. Jenkins' hash function is an alternative to Knuth's.
2662 */
2663#define BLOOM_FILTER_SHIFT	15
2664
2665static inline int filter_gen_from_seq(unsigned long seq)
2666{
2667	return seq % NR_BLOOM_FILTERS;
2668}
2669
2670static void get_item_key(void *item, int *key)
2671{
2672	u32 hash = hash_ptr(item, BLOOM_FILTER_SHIFT * 2);
2673
2674	BUILD_BUG_ON(BLOOM_FILTER_SHIFT * 2 > BITS_PER_TYPE(u32));
2675
2676	key[0] = hash & (BIT(BLOOM_FILTER_SHIFT) - 1);
2677	key[1] = hash >> BLOOM_FILTER_SHIFT;
2678}
2679
2680static bool test_bloom_filter(struct lru_gen_mm_state *mm_state, unsigned long seq,
2681			      void *item)
2682{
2683	int key[2];
2684	unsigned long *filter;
2685	int gen = filter_gen_from_seq(seq);
2686
2687	filter = READ_ONCE(mm_state->filters[gen]);
2688	if (!filter)
2689		return true;
2690
2691	get_item_key(item, key);
2692
2693	return test_bit(key[0], filter) && test_bit(key[1], filter);
2694}
2695
2696static void update_bloom_filter(struct lru_gen_mm_state *mm_state, unsigned long seq,
2697				void *item)
2698{
2699	int key[2];
2700	unsigned long *filter;
2701	int gen = filter_gen_from_seq(seq);
2702
2703	filter = READ_ONCE(mm_state->filters[gen]);
2704	if (!filter)
2705		return;
2706
2707	get_item_key(item, key);
2708
2709	if (!test_bit(key[0], filter))
2710		set_bit(key[0], filter);
2711	if (!test_bit(key[1], filter))
2712		set_bit(key[1], filter);
2713}
2714
2715static void reset_bloom_filter(struct lru_gen_mm_state *mm_state, unsigned long seq)
2716{
2717	unsigned long *filter;
2718	int gen = filter_gen_from_seq(seq);
2719
2720	filter = mm_state->filters[gen];
2721	if (filter) {
2722		bitmap_clear(filter, 0, BIT(BLOOM_FILTER_SHIFT));
2723		return;
2724	}
2725
2726	filter = bitmap_zalloc(BIT(BLOOM_FILTER_SHIFT),
2727			       __GFP_HIGH | __GFP_NOMEMALLOC | __GFP_NOWARN);
2728	WRITE_ONCE(mm_state->filters[gen], filter);
2729}
2730
2731/******************************************************************************
2732 *                          mm_struct list
2733 ******************************************************************************/
2734
2735#ifdef CONFIG_LRU_GEN_WALKS_MMU
2736
2737static struct lru_gen_mm_list *get_mm_list(struct mem_cgroup *memcg)
2738{
2739	static struct lru_gen_mm_list mm_list = {
2740		.fifo = LIST_HEAD_INIT(mm_list.fifo),
2741		.lock = __SPIN_LOCK_UNLOCKED(mm_list.lock),
2742	};
2743
2744#ifdef CONFIG_MEMCG
2745	if (memcg)
2746		return &memcg->mm_list;
2747#endif
2748	VM_WARN_ON_ONCE(!mem_cgroup_disabled());
2749
2750	return &mm_list;
2751}
2752
2753static struct lru_gen_mm_state *get_mm_state(struct lruvec *lruvec)
2754{
2755	return &lruvec->mm_state;
2756}
2757
2758static struct mm_struct *get_next_mm(struct lru_gen_mm_walk *walk)
2759{
2760	int key;
2761	struct mm_struct *mm;
2762	struct pglist_data *pgdat = lruvec_pgdat(walk->lruvec);
2763	struct lru_gen_mm_state *mm_state = get_mm_state(walk->lruvec);
2764
2765	mm = list_entry(mm_state->head, struct mm_struct, lru_gen.list);
2766	key = pgdat->node_id % BITS_PER_TYPE(mm->lru_gen.bitmap);
2767
2768	if (!walk->force_scan && !test_bit(key, &mm->lru_gen.bitmap))
2769		return NULL;
2770
2771	clear_bit(key, &mm->lru_gen.bitmap);
2772
2773	return mmget_not_zero(mm) ? mm : NULL;
2774}
2775
2776void lru_gen_add_mm(struct mm_struct *mm)
2777{
2778	int nid;
2779	struct mem_cgroup *memcg = get_mem_cgroup_from_mm(mm);
2780	struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
2781
2782	VM_WARN_ON_ONCE(!list_empty(&mm->lru_gen.list));
2783#ifdef CONFIG_MEMCG
2784	VM_WARN_ON_ONCE(mm->lru_gen.memcg);
2785	mm->lru_gen.memcg = memcg;
2786#endif
2787	spin_lock(&mm_list->lock);
2788
2789	for_each_node_state(nid, N_MEMORY) {
2790		struct lruvec *lruvec = get_lruvec(memcg, nid);
2791		struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
2792
2793		/* the first addition since the last iteration */
2794		if (mm_state->tail == &mm_list->fifo)
2795			mm_state->tail = &mm->lru_gen.list;
2796	}
2797
2798	list_add_tail(&mm->lru_gen.list, &mm_list->fifo);
2799
2800	spin_unlock(&mm_list->lock);
2801}
2802
2803void lru_gen_del_mm(struct mm_struct *mm)
2804{
2805	int nid;
2806	struct lru_gen_mm_list *mm_list;
2807	struct mem_cgroup *memcg = NULL;
2808
2809	if (list_empty(&mm->lru_gen.list))
2810		return;
2811
2812#ifdef CONFIG_MEMCG
2813	memcg = mm->lru_gen.memcg;
2814#endif
2815	mm_list = get_mm_list(memcg);
2816
2817	spin_lock(&mm_list->lock);
2818
2819	for_each_node(nid) {
2820		struct lruvec *lruvec = get_lruvec(memcg, nid);
2821		struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
2822
2823		/* where the current iteration continues after */
2824		if (mm_state->head == &mm->lru_gen.list)
2825			mm_state->head = mm_state->head->prev;
2826
2827		/* where the last iteration ended before */
2828		if (mm_state->tail == &mm->lru_gen.list)
2829			mm_state->tail = mm_state->tail->next;
2830	}
2831
2832	list_del_init(&mm->lru_gen.list);
2833
2834	spin_unlock(&mm_list->lock);
2835
2836#ifdef CONFIG_MEMCG
2837	mem_cgroup_put(mm->lru_gen.memcg);
2838	mm->lru_gen.memcg = NULL;
2839#endif
2840}
2841
2842#ifdef CONFIG_MEMCG
2843void lru_gen_migrate_mm(struct mm_struct *mm)
2844{
2845	struct mem_cgroup *memcg;
2846	struct task_struct *task = rcu_dereference_protected(mm->owner, true);
2847
2848	VM_WARN_ON_ONCE(task->mm != mm);
2849	lockdep_assert_held(&task->alloc_lock);
2850
2851	/* for mm_update_next_owner() */
2852	if (mem_cgroup_disabled())
2853		return;
2854
2855	/* migration can happen before addition */
2856	if (!mm->lru_gen.memcg)
2857		return;
2858
2859	rcu_read_lock();
2860	memcg = mem_cgroup_from_task(task);
2861	rcu_read_unlock();
2862	if (memcg == mm->lru_gen.memcg)
2863		return;
2864
2865	VM_WARN_ON_ONCE(list_empty(&mm->lru_gen.list));
2866
2867	lru_gen_del_mm(mm);
2868	lru_gen_add_mm(mm);
2869}
2870#endif
2871
2872#else /* !CONFIG_LRU_GEN_WALKS_MMU */
2873
2874static struct lru_gen_mm_list *get_mm_list(struct mem_cgroup *memcg)
2875{
2876	return NULL;
2877}
2878
2879static struct lru_gen_mm_state *get_mm_state(struct lruvec *lruvec)
2880{
2881	return NULL;
2882}
2883
2884static struct mm_struct *get_next_mm(struct lru_gen_mm_walk *walk)
2885{
2886	return NULL;
2887}
2888
2889#endif
2890
2891static void reset_mm_stats(struct lru_gen_mm_walk *walk, bool last)
2892{
2893	int i;
2894	int hist;
2895	struct lruvec *lruvec = walk->lruvec;
2896	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
2897
2898	lockdep_assert_held(&get_mm_list(lruvec_memcg(lruvec))->lock);
2899
2900	hist = lru_hist_from_seq(walk->seq);
 
2901
2902	for (i = 0; i < NR_MM_STATS; i++) {
2903		WRITE_ONCE(mm_state->stats[hist][i],
2904			   mm_state->stats[hist][i] + walk->mm_stats[i]);
2905		walk->mm_stats[i] = 0;
 
2906	}
2907
2908	if (NR_HIST_GENS > 1 && last) {
2909		hist = lru_hist_from_seq(walk->seq + 1);
2910
2911		for (i = 0; i < NR_MM_STATS; i++)
2912			WRITE_ONCE(mm_state->stats[hist][i], 0);
2913	}
2914}
2915
2916static bool iterate_mm_list(struct lru_gen_mm_walk *walk, struct mm_struct **iter)
 
2917{
2918	bool first = false;
2919	bool last = false;
2920	struct mm_struct *mm = NULL;
2921	struct lruvec *lruvec = walk->lruvec;
2922	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
2923	struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
2924	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
2925
2926	/*
2927	 * mm_state->seq is incremented after each iteration of mm_list. There
2928	 * are three interesting cases for this page table walker:
2929	 * 1. It tries to start a new iteration with a stale max_seq: there is
2930	 *    nothing left to do.
2931	 * 2. It started the next iteration: it needs to reset the Bloom filter
2932	 *    so that a fresh set of PTE tables can be recorded.
2933	 * 3. It ended the current iteration: it needs to reset the mm stats
2934	 *    counters and tell its caller to increment max_seq.
2935	 */
2936	spin_lock(&mm_list->lock);
2937
2938	VM_WARN_ON_ONCE(mm_state->seq + 1 < walk->seq);
2939
2940	if (walk->seq <= mm_state->seq)
2941		goto done;
2942
2943	if (!mm_state->head)
2944		mm_state->head = &mm_list->fifo;
2945
2946	if (mm_state->head == &mm_list->fifo)
2947		first = true;
2948
2949	do {
2950		mm_state->head = mm_state->head->next;
2951		if (mm_state->head == &mm_list->fifo) {
2952			WRITE_ONCE(mm_state->seq, mm_state->seq + 1);
2953			last = true;
2954			break;
2955		}
2956
2957		/* force scan for those added after the last iteration */
2958		if (!mm_state->tail || mm_state->tail == mm_state->head) {
2959			mm_state->tail = mm_state->head->next;
2960			walk->force_scan = true;
2961		}
2962	} while (!(mm = get_next_mm(walk)));
2963done:
2964	if (*iter || last)
2965		reset_mm_stats(walk, last);
2966
2967	spin_unlock(&mm_list->lock);
2968
2969	if (mm && first)
2970		reset_bloom_filter(mm_state, walk->seq + 1);
2971
2972	if (*iter)
2973		mmput_async(*iter);
2974
2975	*iter = mm;
2976
2977	return last;
2978}
2979
2980static bool iterate_mm_list_nowalk(struct lruvec *lruvec, unsigned long seq)
2981{
2982	bool success = false;
2983	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
2984	struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
2985	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
2986
2987	spin_lock(&mm_list->lock);
2988
2989	VM_WARN_ON_ONCE(mm_state->seq + 1 < seq);
2990
2991	if (seq > mm_state->seq) {
2992		mm_state->head = NULL;
2993		mm_state->tail = NULL;
2994		WRITE_ONCE(mm_state->seq, mm_state->seq + 1);
 
2995		success = true;
2996	}
2997
2998	spin_unlock(&mm_list->lock);
2999
3000	return success;
3001}
3002
3003/******************************************************************************
3004 *                          PID controller
3005 ******************************************************************************/
3006
3007/*
3008 * A feedback loop based on Proportional-Integral-Derivative (PID) controller.
3009 *
3010 * The P term is refaulted/(evicted+protected) from a tier in the generation
3011 * currently being evicted; the I term is the exponential moving average of the
3012 * P term over the generations previously evicted, using the smoothing factor
3013 * 1/2; the D term isn't supported.
3014 *
3015 * The setpoint (SP) is always the first tier of one type; the process variable
3016 * (PV) is either any tier of the other type or any other tier of the same
3017 * type.
3018 *
3019 * The error is the difference between the SP and the PV; the correction is to
3020 * turn off protection when SP>PV or turn on protection when SP<PV.
3021 *
3022 * For future optimizations:
3023 * 1. The D term may discount the other two terms over time so that long-lived
3024 *    generations can resist stale information.
3025 */
3026struct ctrl_pos {
3027	unsigned long refaulted;
3028	unsigned long total;
3029	int gain;
3030};
3031
3032static void read_ctrl_pos(struct lruvec *lruvec, int type, int tier, int gain,
3033			  struct ctrl_pos *pos)
3034{
3035	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3036	int hist = lru_hist_from_seq(lrugen->min_seq[type]);
3037
3038	pos->refaulted = lrugen->avg_refaulted[type][tier] +
3039			 atomic_long_read(&lrugen->refaulted[hist][type][tier]);
3040	pos->total = lrugen->avg_total[type][tier] +
3041		     atomic_long_read(&lrugen->evicted[hist][type][tier]);
3042	if (tier)
3043		pos->total += lrugen->protected[hist][type][tier - 1];
3044	pos->gain = gain;
3045}
3046
3047static void reset_ctrl_pos(struct lruvec *lruvec, int type, bool carryover)
3048{
3049	int hist, tier;
3050	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3051	bool clear = carryover ? NR_HIST_GENS == 1 : NR_HIST_GENS > 1;
3052	unsigned long seq = carryover ? lrugen->min_seq[type] : lrugen->max_seq + 1;
3053
3054	lockdep_assert_held(&lruvec->lru_lock);
3055
3056	if (!carryover && !clear)
3057		return;
3058
3059	hist = lru_hist_from_seq(seq);
3060
3061	for (tier = 0; tier < MAX_NR_TIERS; tier++) {
3062		if (carryover) {
3063			unsigned long sum;
3064
3065			sum = lrugen->avg_refaulted[type][tier] +
3066			      atomic_long_read(&lrugen->refaulted[hist][type][tier]);
3067			WRITE_ONCE(lrugen->avg_refaulted[type][tier], sum / 2);
3068
3069			sum = lrugen->avg_total[type][tier] +
3070			      atomic_long_read(&lrugen->evicted[hist][type][tier]);
3071			if (tier)
3072				sum += lrugen->protected[hist][type][tier - 1];
3073			WRITE_ONCE(lrugen->avg_total[type][tier], sum / 2);
3074		}
3075
3076		if (clear) {
3077			atomic_long_set(&lrugen->refaulted[hist][type][tier], 0);
3078			atomic_long_set(&lrugen->evicted[hist][type][tier], 0);
3079			if (tier)
3080				WRITE_ONCE(lrugen->protected[hist][type][tier - 1], 0);
3081		}
3082	}
3083}
3084
3085static bool positive_ctrl_err(struct ctrl_pos *sp, struct ctrl_pos *pv)
3086{
3087	/*
3088	 * Return true if the PV has a limited number of refaults or a lower
3089	 * refaulted/total than the SP.
3090	 */
3091	return pv->refaulted < MIN_LRU_BATCH ||
3092	       pv->refaulted * (sp->total + MIN_LRU_BATCH) * sp->gain <=
3093	       (sp->refaulted + 1) * pv->total * pv->gain;
3094}
3095
3096/******************************************************************************
3097 *                          the aging
3098 ******************************************************************************/
3099
3100/* promote pages accessed through page tables */
3101static int folio_update_gen(struct folio *folio, int gen)
3102{
3103	unsigned long new_flags, old_flags = READ_ONCE(folio->flags);
3104
3105	VM_WARN_ON_ONCE(gen >= MAX_NR_GENS);
3106	VM_WARN_ON_ONCE(!rcu_read_lock_held());
3107
3108	do {
3109		/* lru_gen_del_folio() has isolated this page? */
3110		if (!(old_flags & LRU_GEN_MASK)) {
3111			/* for shrink_folio_list() */
3112			new_flags = old_flags | BIT(PG_referenced);
3113			continue;
3114		}
3115
3116		new_flags = old_flags & ~(LRU_GEN_MASK | LRU_REFS_MASK | LRU_REFS_FLAGS);
3117		new_flags |= (gen + 1UL) << LRU_GEN_PGOFF;
3118	} while (!try_cmpxchg(&folio->flags, &old_flags, new_flags));
3119
3120	return ((old_flags & LRU_GEN_MASK) >> LRU_GEN_PGOFF) - 1;
3121}
3122
3123/* protect pages accessed multiple times through file descriptors */
3124static int folio_inc_gen(struct lruvec *lruvec, struct folio *folio, bool reclaiming)
3125{
3126	int type = folio_is_file_lru(folio);
3127	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3128	int new_gen, old_gen = lru_gen_from_seq(lrugen->min_seq[type]);
3129	unsigned long new_flags, old_flags = READ_ONCE(folio->flags);
3130
3131	VM_WARN_ON_ONCE_FOLIO(!(old_flags & LRU_GEN_MASK), folio);
3132
3133	do {
3134		new_gen = ((old_flags & LRU_GEN_MASK) >> LRU_GEN_PGOFF) - 1;
3135		/* folio_update_gen() has promoted this page? */
3136		if (new_gen >= 0 && new_gen != old_gen)
3137			return new_gen;
3138
3139		new_gen = (old_gen + 1) % MAX_NR_GENS;
3140
3141		new_flags = old_flags & ~(LRU_GEN_MASK | LRU_REFS_MASK | LRU_REFS_FLAGS);
3142		new_flags |= (new_gen + 1UL) << LRU_GEN_PGOFF;
3143		/* for folio_end_writeback() */
3144		if (reclaiming)
3145			new_flags |= BIT(PG_reclaim);
3146	} while (!try_cmpxchg(&folio->flags, &old_flags, new_flags));
3147
3148	lru_gen_update_size(lruvec, folio, old_gen, new_gen);
3149
3150	return new_gen;
3151}
3152
3153static void update_batch_size(struct lru_gen_mm_walk *walk, struct folio *folio,
3154			      int old_gen, int new_gen)
3155{
3156	int type = folio_is_file_lru(folio);
3157	int zone = folio_zonenum(folio);
3158	int delta = folio_nr_pages(folio);
3159
3160	VM_WARN_ON_ONCE(old_gen >= MAX_NR_GENS);
3161	VM_WARN_ON_ONCE(new_gen >= MAX_NR_GENS);
3162
3163	walk->batched++;
3164
3165	walk->nr_pages[old_gen][type][zone] -= delta;
3166	walk->nr_pages[new_gen][type][zone] += delta;
3167}
3168
3169static void reset_batch_size(struct lru_gen_mm_walk *walk)
3170{
3171	int gen, type, zone;
3172	struct lruvec *lruvec = walk->lruvec;
3173	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3174
3175	walk->batched = 0;
3176
3177	for_each_gen_type_zone(gen, type, zone) {
3178		enum lru_list lru = type * LRU_INACTIVE_FILE;
3179		int delta = walk->nr_pages[gen][type][zone];
3180
3181		if (!delta)
3182			continue;
3183
3184		walk->nr_pages[gen][type][zone] = 0;
3185		WRITE_ONCE(lrugen->nr_pages[gen][type][zone],
3186			   lrugen->nr_pages[gen][type][zone] + delta);
3187
3188		if (lru_gen_is_active(lruvec, gen))
3189			lru += LRU_ACTIVE;
3190		__update_lru_size(lruvec, lru, zone, delta);
3191	}
3192}
3193
3194static int should_skip_vma(unsigned long start, unsigned long end, struct mm_walk *args)
3195{
3196	struct address_space *mapping;
3197	struct vm_area_struct *vma = args->vma;
3198	struct lru_gen_mm_walk *walk = args->private;
3199
3200	if (!vma_is_accessible(vma))
3201		return true;
3202
3203	if (is_vm_hugetlb_page(vma))
3204		return true;
3205
3206	if (!vma_has_recency(vma))
3207		return true;
3208
3209	if (vma->vm_flags & (VM_LOCKED | VM_SPECIAL))
3210		return true;
3211
3212	if (vma == get_gate_vma(vma->vm_mm))
3213		return true;
3214
3215	if (vma_is_anonymous(vma))
3216		return !walk->can_swap;
3217
3218	if (WARN_ON_ONCE(!vma->vm_file || !vma->vm_file->f_mapping))
3219		return true;
3220
3221	mapping = vma->vm_file->f_mapping;
3222	if (mapping_unevictable(mapping))
3223		return true;
3224
3225	if (shmem_mapping(mapping))
3226		return !walk->can_swap;
3227
3228	/* to exclude special mappings like dax, etc. */
3229	return !mapping->a_ops->read_folio;
3230}
3231
3232/*
3233 * Some userspace memory allocators map many single-page VMAs. Instead of
3234 * returning back to the PGD table for each of such VMAs, finish an entire PMD
3235 * table to reduce zigzags and improve cache performance.
3236 */
3237static bool get_next_vma(unsigned long mask, unsigned long size, struct mm_walk *args,
3238			 unsigned long *vm_start, unsigned long *vm_end)
3239{
3240	unsigned long start = round_up(*vm_end, size);
3241	unsigned long end = (start | ~mask) + 1;
3242	VMA_ITERATOR(vmi, args->mm, start);
3243
3244	VM_WARN_ON_ONCE(mask & size);
3245	VM_WARN_ON_ONCE((start & mask) != (*vm_start & mask));
3246
3247	for_each_vma(vmi, args->vma) {
3248		if (end && end <= args->vma->vm_start)
3249			return false;
3250
3251		if (should_skip_vma(args->vma->vm_start, args->vma->vm_end, args))
3252			continue;
3253
3254		*vm_start = max(start, args->vma->vm_start);
3255		*vm_end = min(end - 1, args->vma->vm_end - 1) + 1;
3256
3257		return true;
3258	}
3259
3260	return false;
3261}
3262
3263static unsigned long get_pte_pfn(pte_t pte, struct vm_area_struct *vma, unsigned long addr)
3264{
3265	unsigned long pfn = pte_pfn(pte);
3266
3267	VM_WARN_ON_ONCE(addr < vma->vm_start || addr >= vma->vm_end);
3268
3269	if (!pte_present(pte) || is_zero_pfn(pfn))
3270		return -1;
3271
3272	if (WARN_ON_ONCE(pte_devmap(pte) || pte_special(pte)))
3273		return -1;
3274
3275	if (WARN_ON_ONCE(!pfn_valid(pfn)))
3276		return -1;
3277
3278	return pfn;
3279}
3280
3281static unsigned long get_pmd_pfn(pmd_t pmd, struct vm_area_struct *vma, unsigned long addr)
3282{
3283	unsigned long pfn = pmd_pfn(pmd);
3284
3285	VM_WARN_ON_ONCE(addr < vma->vm_start || addr >= vma->vm_end);
3286
3287	if (!pmd_present(pmd) || is_huge_zero_pmd(pmd))
3288		return -1;
3289
3290	if (WARN_ON_ONCE(pmd_devmap(pmd)))
3291		return -1;
3292
3293	if (WARN_ON_ONCE(!pfn_valid(pfn)))
3294		return -1;
3295
3296	return pfn;
3297}
3298
3299static struct folio *get_pfn_folio(unsigned long pfn, struct mem_cgroup *memcg,
3300				   struct pglist_data *pgdat, bool can_swap)
3301{
3302	struct folio *folio;
3303
3304	/* try to avoid unnecessary memory loads */
3305	if (pfn < pgdat->node_start_pfn || pfn >= pgdat_end_pfn(pgdat))
3306		return NULL;
3307
3308	folio = pfn_folio(pfn);
3309	if (folio_nid(folio) != pgdat->node_id)
3310		return NULL;
3311
3312	if (folio_memcg_rcu(folio) != memcg)
3313		return NULL;
3314
3315	/* file VMAs can contain anon pages from COW */
3316	if (!folio_is_file_lru(folio) && !can_swap)
3317		return NULL;
3318
3319	return folio;
3320}
3321
3322static bool suitable_to_scan(int total, int young)
3323{
3324	int n = clamp_t(int, cache_line_size() / sizeof(pte_t), 2, 8);
3325
3326	/* suitable if the average number of young PTEs per cacheline is >=1 */
3327	return young * n >= total;
3328}
3329
3330static bool walk_pte_range(pmd_t *pmd, unsigned long start, unsigned long end,
3331			   struct mm_walk *args)
3332{
3333	int i;
3334	pte_t *pte;
3335	spinlock_t *ptl;
3336	unsigned long addr;
3337	int total = 0;
3338	int young = 0;
3339	struct lru_gen_mm_walk *walk = args->private;
3340	struct mem_cgroup *memcg = lruvec_memcg(walk->lruvec);
3341	struct pglist_data *pgdat = lruvec_pgdat(walk->lruvec);
3342	DEFINE_MAX_SEQ(walk->lruvec);
3343	int old_gen, new_gen = lru_gen_from_seq(max_seq);
3344
3345	pte = pte_offset_map_nolock(args->mm, pmd, start & PMD_MASK, &ptl);
3346	if (!pte)
3347		return false;
3348	if (!spin_trylock(ptl)) {
3349		pte_unmap(pte);
3350		return false;
3351	}
3352
3353	arch_enter_lazy_mmu_mode();
3354restart:
3355	for (i = pte_index(start), addr = start; addr != end; i++, addr += PAGE_SIZE) {
3356		unsigned long pfn;
3357		struct folio *folio;
3358		pte_t ptent = ptep_get(pte + i);
3359
3360		total++;
3361		walk->mm_stats[MM_LEAF_TOTAL]++;
3362
3363		pfn = get_pte_pfn(ptent, args->vma, addr);
3364		if (pfn == -1)
3365			continue;
3366
3367		if (!pte_young(ptent)) {
3368			walk->mm_stats[MM_LEAF_OLD]++;
3369			continue;
3370		}
3371
3372		folio = get_pfn_folio(pfn, memcg, pgdat, walk->can_swap);
3373		if (!folio)
3374			continue;
3375
3376		if (!ptep_test_and_clear_young(args->vma, addr, pte + i))
3377			VM_WARN_ON_ONCE(true);
3378
3379		young++;
3380		walk->mm_stats[MM_LEAF_YOUNG]++;
3381
3382		if (pte_dirty(ptent) && !folio_test_dirty(folio) &&
3383		    !(folio_test_anon(folio) && folio_test_swapbacked(folio) &&
3384		      !folio_test_swapcache(folio)))
3385			folio_mark_dirty(folio);
3386
3387		old_gen = folio_update_gen(folio, new_gen);
3388		if (old_gen >= 0 && old_gen != new_gen)
3389			update_batch_size(walk, folio, old_gen, new_gen);
3390	}
3391
3392	if (i < PTRS_PER_PTE && get_next_vma(PMD_MASK, PAGE_SIZE, args, &start, &end))
3393		goto restart;
3394
3395	arch_leave_lazy_mmu_mode();
3396	pte_unmap_unlock(pte, ptl);
3397
3398	return suitable_to_scan(total, young);
3399}
3400
3401static void walk_pmd_range_locked(pud_t *pud, unsigned long addr, struct vm_area_struct *vma,
3402				  struct mm_walk *args, unsigned long *bitmap, unsigned long *first)
3403{
3404	int i;
3405	pmd_t *pmd;
3406	spinlock_t *ptl;
3407	struct lru_gen_mm_walk *walk = args->private;
3408	struct mem_cgroup *memcg = lruvec_memcg(walk->lruvec);
3409	struct pglist_data *pgdat = lruvec_pgdat(walk->lruvec);
3410	DEFINE_MAX_SEQ(walk->lruvec);
3411	int old_gen, new_gen = lru_gen_from_seq(max_seq);
3412
3413	VM_WARN_ON_ONCE(pud_leaf(*pud));
3414
3415	/* try to batch at most 1+MIN_LRU_BATCH+1 entries */
3416	if (*first == -1) {
3417		*first = addr;
3418		bitmap_zero(bitmap, MIN_LRU_BATCH);
3419		return;
3420	}
3421
3422	i = addr == -1 ? 0 : pmd_index(addr) - pmd_index(*first);
3423	if (i && i <= MIN_LRU_BATCH) {
3424		__set_bit(i - 1, bitmap);
3425		return;
3426	}
3427
3428	pmd = pmd_offset(pud, *first);
3429
3430	ptl = pmd_lockptr(args->mm, pmd);
3431	if (!spin_trylock(ptl))
3432		goto done;
3433
3434	arch_enter_lazy_mmu_mode();
3435
3436	do {
3437		unsigned long pfn;
3438		struct folio *folio;
3439
3440		/* don't round down the first address */
3441		addr = i ? (*first & PMD_MASK) + i * PMD_SIZE : *first;
3442
3443		pfn = get_pmd_pfn(pmd[i], vma, addr);
3444		if (pfn == -1)
3445			goto next;
3446
3447		if (!pmd_trans_huge(pmd[i])) {
3448			if (should_clear_pmd_young())
3449				pmdp_test_and_clear_young(vma, addr, pmd + i);
3450			goto next;
3451		}
3452
3453		folio = get_pfn_folio(pfn, memcg, pgdat, walk->can_swap);
3454		if (!folio)
3455			goto next;
3456
3457		if (!pmdp_test_and_clear_young(vma, addr, pmd + i))
3458			goto next;
3459
3460		walk->mm_stats[MM_LEAF_YOUNG]++;
3461
3462		if (pmd_dirty(pmd[i]) && !folio_test_dirty(folio) &&
3463		    !(folio_test_anon(folio) && folio_test_swapbacked(folio) &&
3464		      !folio_test_swapcache(folio)))
3465			folio_mark_dirty(folio);
3466
3467		old_gen = folio_update_gen(folio, new_gen);
3468		if (old_gen >= 0 && old_gen != new_gen)
3469			update_batch_size(walk, folio, old_gen, new_gen);
3470next:
3471		i = i > MIN_LRU_BATCH ? 0 : find_next_bit(bitmap, MIN_LRU_BATCH, i) + 1;
3472	} while (i <= MIN_LRU_BATCH);
3473
3474	arch_leave_lazy_mmu_mode();
3475	spin_unlock(ptl);
3476done:
3477	*first = -1;
3478}
3479
3480static void walk_pmd_range(pud_t *pud, unsigned long start, unsigned long end,
3481			   struct mm_walk *args)
3482{
3483	int i;
3484	pmd_t *pmd;
3485	unsigned long next;
3486	unsigned long addr;
3487	struct vm_area_struct *vma;
3488	DECLARE_BITMAP(bitmap, MIN_LRU_BATCH);
3489	unsigned long first = -1;
3490	struct lru_gen_mm_walk *walk = args->private;
3491	struct lru_gen_mm_state *mm_state = get_mm_state(walk->lruvec);
3492
3493	VM_WARN_ON_ONCE(pud_leaf(*pud));
3494
3495	/*
3496	 * Finish an entire PMD in two passes: the first only reaches to PTE
3497	 * tables to avoid taking the PMD lock; the second, if necessary, takes
3498	 * the PMD lock to clear the accessed bit in PMD entries.
3499	 */
3500	pmd = pmd_offset(pud, start & PUD_MASK);
3501restart:
3502	/* walk_pte_range() may call get_next_vma() */
3503	vma = args->vma;
3504	for (i = pmd_index(start), addr = start; addr != end; i++, addr = next) {
3505		pmd_t val = pmdp_get_lockless(pmd + i);
3506
3507		next = pmd_addr_end(addr, end);
3508
3509		if (!pmd_present(val) || is_huge_zero_pmd(val)) {
3510			walk->mm_stats[MM_LEAF_TOTAL]++;
3511			continue;
3512		}
3513
3514		if (pmd_trans_huge(val)) {
3515			unsigned long pfn = pmd_pfn(val);
3516			struct pglist_data *pgdat = lruvec_pgdat(walk->lruvec);
3517
3518			walk->mm_stats[MM_LEAF_TOTAL]++;
3519
3520			if (!pmd_young(val)) {
3521				walk->mm_stats[MM_LEAF_OLD]++;
3522				continue;
3523			}
3524
3525			/* try to avoid unnecessary memory loads */
3526			if (pfn < pgdat->node_start_pfn || pfn >= pgdat_end_pfn(pgdat))
3527				continue;
3528
3529			walk_pmd_range_locked(pud, addr, vma, args, bitmap, &first);
3530			continue;
3531		}
3532
3533		walk->mm_stats[MM_NONLEAF_TOTAL]++;
3534
3535		if (should_clear_pmd_young()) {
3536			if (!pmd_young(val))
3537				continue;
3538
3539			walk_pmd_range_locked(pud, addr, vma, args, bitmap, &first);
3540		}
3541
3542		if (!walk->force_scan && !test_bloom_filter(mm_state, walk->seq, pmd + i))
3543			continue;
3544
3545		walk->mm_stats[MM_NONLEAF_FOUND]++;
3546
3547		if (!walk_pte_range(&val, addr, next, args))
3548			continue;
3549
3550		walk->mm_stats[MM_NONLEAF_ADDED]++;
3551
3552		/* carry over to the next generation */
3553		update_bloom_filter(mm_state, walk->seq + 1, pmd + i);
3554	}
3555
3556	walk_pmd_range_locked(pud, -1, vma, args, bitmap, &first);
3557
3558	if (i < PTRS_PER_PMD && get_next_vma(PUD_MASK, PMD_SIZE, args, &start, &end))
3559		goto restart;
3560}
3561
3562static int walk_pud_range(p4d_t *p4d, unsigned long start, unsigned long end,
3563			  struct mm_walk *args)
3564{
3565	int i;
3566	pud_t *pud;
3567	unsigned long addr;
3568	unsigned long next;
3569	struct lru_gen_mm_walk *walk = args->private;
3570
3571	VM_WARN_ON_ONCE(p4d_leaf(*p4d));
3572
3573	pud = pud_offset(p4d, start & P4D_MASK);
3574restart:
3575	for (i = pud_index(start), addr = start; addr != end; i++, addr = next) {
3576		pud_t val = READ_ONCE(pud[i]);
3577
3578		next = pud_addr_end(addr, end);
3579
3580		if (!pud_present(val) || WARN_ON_ONCE(pud_leaf(val)))
3581			continue;
3582
3583		walk_pmd_range(&val, addr, next, args);
3584
3585		if (need_resched() || walk->batched >= MAX_LRU_BATCH) {
3586			end = (addr | ~PUD_MASK) + 1;
3587			goto done;
3588		}
3589	}
3590
3591	if (i < PTRS_PER_PUD && get_next_vma(P4D_MASK, PUD_SIZE, args, &start, &end))
3592		goto restart;
3593
3594	end = round_up(end, P4D_SIZE);
3595done:
3596	if (!end || !args->vma)
3597		return 1;
3598
3599	walk->next_addr = max(end, args->vma->vm_start);
3600
3601	return -EAGAIN;
3602}
3603
3604static void walk_mm(struct mm_struct *mm, struct lru_gen_mm_walk *walk)
3605{
3606	static const struct mm_walk_ops mm_walk_ops = {
3607		.test_walk = should_skip_vma,
3608		.p4d_entry = walk_pud_range,
3609		.walk_lock = PGWALK_RDLOCK,
3610	};
3611
3612	int err;
3613	struct lruvec *lruvec = walk->lruvec;
3614	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
3615
3616	walk->next_addr = FIRST_USER_ADDRESS;
3617
3618	do {
3619		DEFINE_MAX_SEQ(lruvec);
3620
3621		err = -EBUSY;
3622
3623		/* another thread might have called inc_max_seq() */
3624		if (walk->seq != max_seq)
3625			break;
3626
3627		/* folio_update_gen() requires stable folio_memcg() */
3628		if (!mem_cgroup_trylock_pages(memcg))
3629			break;
3630
3631		/* the caller might be holding the lock for write */
3632		if (mmap_read_trylock(mm)) {
3633			err = walk_page_range(mm, walk->next_addr, ULONG_MAX, &mm_walk_ops, walk);
3634
3635			mmap_read_unlock(mm);
3636		}
3637
3638		mem_cgroup_unlock_pages();
3639
3640		if (walk->batched) {
3641			spin_lock_irq(&lruvec->lru_lock);
3642			reset_batch_size(walk);
3643			spin_unlock_irq(&lruvec->lru_lock);
3644		}
3645
3646		cond_resched();
3647	} while (err == -EAGAIN);
3648}
3649
3650static struct lru_gen_mm_walk *set_mm_walk(struct pglist_data *pgdat, bool force_alloc)
3651{
3652	struct lru_gen_mm_walk *walk = current->reclaim_state->mm_walk;
3653
3654	if (pgdat && current_is_kswapd()) {
3655		VM_WARN_ON_ONCE(walk);
3656
3657		walk = &pgdat->mm_walk;
3658	} else if (!walk && force_alloc) {
3659		VM_WARN_ON_ONCE(current_is_kswapd());
3660
3661		walk = kzalloc(sizeof(*walk), __GFP_HIGH | __GFP_NOMEMALLOC | __GFP_NOWARN);
3662	}
3663
3664	current->reclaim_state->mm_walk = walk;
3665
3666	return walk;
3667}
3668
3669static void clear_mm_walk(void)
3670{
3671	struct lru_gen_mm_walk *walk = current->reclaim_state->mm_walk;
3672
3673	VM_WARN_ON_ONCE(walk && memchr_inv(walk->nr_pages, 0, sizeof(walk->nr_pages)));
3674	VM_WARN_ON_ONCE(walk && memchr_inv(walk->mm_stats, 0, sizeof(walk->mm_stats)));
3675
3676	current->reclaim_state->mm_walk = NULL;
3677
3678	if (!current_is_kswapd())
3679		kfree(walk);
3680}
3681
3682static bool inc_min_seq(struct lruvec *lruvec, int type, bool can_swap)
3683{
3684	int zone;
3685	int remaining = MAX_LRU_BATCH;
3686	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3687	int new_gen, old_gen = lru_gen_from_seq(lrugen->min_seq[type]);
3688
3689	if (type == LRU_GEN_ANON && !can_swap)
3690		goto done;
3691
3692	/* prevent cold/hot inversion if force_scan is true */
3693	for (zone = 0; zone < MAX_NR_ZONES; zone++) {
3694		struct list_head *head = &lrugen->folios[old_gen][type][zone];
3695
3696		while (!list_empty(head)) {
3697			struct folio *folio = lru_to_folio(head);
3698
3699			VM_WARN_ON_ONCE_FOLIO(folio_test_unevictable(folio), folio);
3700			VM_WARN_ON_ONCE_FOLIO(folio_test_active(folio), folio);
3701			VM_WARN_ON_ONCE_FOLIO(folio_is_file_lru(folio) != type, folio);
3702			VM_WARN_ON_ONCE_FOLIO(folio_zonenum(folio) != zone, folio);
3703
3704			new_gen = folio_inc_gen(lruvec, folio, false);
3705			list_move_tail(&folio->lru, &lrugen->folios[new_gen][type][zone]);
3706
3707			if (!--remaining)
3708				return false;
3709		}
3710	}
3711done:
3712	reset_ctrl_pos(lruvec, type, true);
3713	WRITE_ONCE(lrugen->min_seq[type], lrugen->min_seq[type] + 1);
3714
3715	return true;
3716}
3717
3718static bool try_to_inc_min_seq(struct lruvec *lruvec, bool can_swap)
3719{
3720	int gen, type, zone;
3721	bool success = false;
3722	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3723	DEFINE_MIN_SEQ(lruvec);
3724
3725	VM_WARN_ON_ONCE(!seq_is_valid(lruvec));
3726
3727	/* find the oldest populated generation */
3728	for (type = !can_swap; type < ANON_AND_FILE; type++) {
3729		while (min_seq[type] + MIN_NR_GENS <= lrugen->max_seq) {
3730			gen = lru_gen_from_seq(min_seq[type]);
3731
3732			for (zone = 0; zone < MAX_NR_ZONES; zone++) {
3733				if (!list_empty(&lrugen->folios[gen][type][zone]))
3734					goto next;
3735			}
3736
3737			min_seq[type]++;
3738		}
3739next:
3740		;
3741	}
3742
3743	/* see the comment on lru_gen_folio */
3744	if (can_swap) {
3745		min_seq[LRU_GEN_ANON] = min(min_seq[LRU_GEN_ANON], min_seq[LRU_GEN_FILE]);
3746		min_seq[LRU_GEN_FILE] = max(min_seq[LRU_GEN_ANON], lrugen->min_seq[LRU_GEN_FILE]);
3747	}
3748
3749	for (type = !can_swap; type < ANON_AND_FILE; type++) {
3750		if (min_seq[type] == lrugen->min_seq[type])
3751			continue;
3752
3753		reset_ctrl_pos(lruvec, type, true);
3754		WRITE_ONCE(lrugen->min_seq[type], min_seq[type]);
3755		success = true;
3756	}
3757
3758	return success;
3759}
3760
3761static bool inc_max_seq(struct lruvec *lruvec, unsigned long seq,
3762			bool can_swap, bool force_scan)
3763{
3764	bool success;
3765	int prev, next;
3766	int type, zone;
3767	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3768restart:
3769	if (seq < READ_ONCE(lrugen->max_seq))
3770		return false;
3771
3772	spin_lock_irq(&lruvec->lru_lock);
3773
3774	VM_WARN_ON_ONCE(!seq_is_valid(lruvec));
3775
3776	success = seq == lrugen->max_seq;
3777	if (!success)
3778		goto unlock;
3779
3780	for (type = ANON_AND_FILE - 1; type >= 0; type--) {
3781		if (get_nr_gens(lruvec, type) != MAX_NR_GENS)
3782			continue;
3783
3784		VM_WARN_ON_ONCE(!force_scan && (type == LRU_GEN_FILE || can_swap));
3785
3786		if (inc_min_seq(lruvec, type, can_swap))
3787			continue;
3788
3789		spin_unlock_irq(&lruvec->lru_lock);
3790		cond_resched();
3791		goto restart;
3792	}
3793
3794	/*
3795	 * Update the active/inactive LRU sizes for compatibility. Both sides of
3796	 * the current max_seq need to be covered, since max_seq+1 can overlap
3797	 * with min_seq[LRU_GEN_ANON] if swapping is constrained. And if they do
3798	 * overlap, cold/hot inversion happens.
3799	 */
3800	prev = lru_gen_from_seq(lrugen->max_seq - 1);
3801	next = lru_gen_from_seq(lrugen->max_seq + 1);
3802
3803	for (type = 0; type < ANON_AND_FILE; type++) {
3804		for (zone = 0; zone < MAX_NR_ZONES; zone++) {
3805			enum lru_list lru = type * LRU_INACTIVE_FILE;
3806			long delta = lrugen->nr_pages[prev][type][zone] -
3807				     lrugen->nr_pages[next][type][zone];
3808
3809			if (!delta)
3810				continue;
3811
3812			__update_lru_size(lruvec, lru, zone, delta);
3813			__update_lru_size(lruvec, lru + LRU_ACTIVE, zone, -delta);
3814		}
3815	}
3816
3817	for (type = 0; type < ANON_AND_FILE; type++)
3818		reset_ctrl_pos(lruvec, type, false);
3819
3820	WRITE_ONCE(lrugen->timestamps[next], jiffies);
3821	/* make sure preceding modifications appear */
3822	smp_store_release(&lrugen->max_seq, lrugen->max_seq + 1);
3823unlock:
3824	spin_unlock_irq(&lruvec->lru_lock);
3825
3826	return success;
3827}
3828
3829static bool try_to_inc_max_seq(struct lruvec *lruvec, unsigned long seq,
3830			       bool can_swap, bool force_scan)
3831{
3832	bool success;
3833	struct lru_gen_mm_walk *walk;
3834	struct mm_struct *mm = NULL;
3835	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3836	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
3837
3838	VM_WARN_ON_ONCE(seq > READ_ONCE(lrugen->max_seq));
3839
3840	if (!mm_state)
3841		return inc_max_seq(lruvec, seq, can_swap, force_scan);
3842
3843	/* see the comment in iterate_mm_list() */
3844	if (seq <= READ_ONCE(mm_state->seq))
3845		return false;
3846
3847	/*
3848	 * If the hardware doesn't automatically set the accessed bit, fallback
3849	 * to lru_gen_look_around(), which only clears the accessed bit in a
3850	 * handful of PTEs. Spreading the work out over a period of time usually
3851	 * is less efficient, but it avoids bursty page faults.
3852	 */
3853	if (!should_walk_mmu()) {
3854		success = iterate_mm_list_nowalk(lruvec, seq);
3855		goto done;
3856	}
3857
3858	walk = set_mm_walk(NULL, true);
3859	if (!walk) {
3860		success = iterate_mm_list_nowalk(lruvec, seq);
3861		goto done;
3862	}
3863
3864	walk->lruvec = lruvec;
3865	walk->seq = seq;
3866	walk->can_swap = can_swap;
3867	walk->force_scan = force_scan;
3868
3869	do {
3870		success = iterate_mm_list(walk, &mm);
3871		if (mm)
3872			walk_mm(mm, walk);
3873	} while (mm);
3874done:
3875	if (success) {
3876		success = inc_max_seq(lruvec, seq, can_swap, force_scan);
3877		WARN_ON_ONCE(!success);
3878	}
3879
3880	return success;
3881}
3882
3883/******************************************************************************
3884 *                          working set protection
3885 ******************************************************************************/
3886
3887static bool lruvec_is_sizable(struct lruvec *lruvec, struct scan_control *sc)
3888{
3889	int gen, type, zone;
3890	unsigned long total = 0;
3891	bool can_swap = get_swappiness(lruvec, sc);
3892	struct lru_gen_folio *lrugen = &lruvec->lrugen;
3893	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
3894	DEFINE_MAX_SEQ(lruvec);
3895	DEFINE_MIN_SEQ(lruvec);
3896
3897	for (type = !can_swap; type < ANON_AND_FILE; type++) {
3898		unsigned long seq;
3899
3900		for (seq = min_seq[type]; seq <= max_seq; seq++) {
3901			gen = lru_gen_from_seq(seq);
3902
3903			for (zone = 0; zone < MAX_NR_ZONES; zone++)
3904				total += max(READ_ONCE(lrugen->nr_pages[gen][type][zone]), 0L);
3905		}
3906	}
3907
3908	/* whether the size is big enough to be helpful */
3909	return mem_cgroup_online(memcg) ? (total >> sc->priority) : total;
3910}
3911
3912static bool lruvec_is_reclaimable(struct lruvec *lruvec, struct scan_control *sc,
3913				  unsigned long min_ttl)
3914{
3915	int gen;
3916	unsigned long birth;
3917	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
3918	DEFINE_MIN_SEQ(lruvec);
3919
3920	/* see the comment on lru_gen_folio */
3921	gen = lru_gen_from_seq(min_seq[LRU_GEN_FILE]);
3922	birth = READ_ONCE(lruvec->lrugen.timestamps[gen]);
3923
3924	if (time_is_after_jiffies(birth + min_ttl))
3925		return false;
3926
3927	if (!lruvec_is_sizable(lruvec, sc))
3928		return false;
3929
3930	mem_cgroup_calculate_protection(NULL, memcg);
3931
3932	return !mem_cgroup_below_min(NULL, memcg);
3933}
3934
3935/* to protect the working set of the last N jiffies */
3936static unsigned long lru_gen_min_ttl __read_mostly;
3937
3938static void lru_gen_age_node(struct pglist_data *pgdat, struct scan_control *sc)
3939{
3940	struct mem_cgroup *memcg;
3941	unsigned long min_ttl = READ_ONCE(lru_gen_min_ttl);
3942
3943	VM_WARN_ON_ONCE(!current_is_kswapd());
3944
3945	/* check the order to exclude compaction-induced reclaim */
3946	if (!min_ttl || sc->order || sc->priority == DEF_PRIORITY)
3947		return;
3948
3949	memcg = mem_cgroup_iter(NULL, NULL, NULL);
3950	do {
3951		struct lruvec *lruvec = mem_cgroup_lruvec(memcg, pgdat);
3952
3953		if (lruvec_is_reclaimable(lruvec, sc, min_ttl)) {
3954			mem_cgroup_iter_break(NULL, memcg);
3955			return;
3956		}
3957
3958		cond_resched();
3959	} while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)));
3960
3961	/*
3962	 * The main goal is to OOM kill if every generation from all memcgs is
3963	 * younger than min_ttl. However, another possibility is all memcgs are
3964	 * either too small or below min.
3965	 */
3966	if (mutex_trylock(&oom_lock)) {
3967		struct oom_control oc = {
3968			.gfp_mask = sc->gfp_mask,
3969		};
3970
3971		out_of_memory(&oc);
3972
3973		mutex_unlock(&oom_lock);
3974	}
3975}
3976
3977/******************************************************************************
3978 *                          rmap/PT walk feedback
3979 ******************************************************************************/
3980
3981/*
3982 * This function exploits spatial locality when shrink_folio_list() walks the
3983 * rmap. It scans the adjacent PTEs of a young PTE and promotes hot pages. If
3984 * the scan was done cacheline efficiently, it adds the PMD entry pointing to
3985 * the PTE table to the Bloom filter. This forms a feedback loop between the
3986 * eviction and the aging.
3987 */
3988void lru_gen_look_around(struct page_vma_mapped_walk *pvmw)
3989{
3990	int i;
3991	unsigned long start;
3992	unsigned long end;
3993	struct lru_gen_mm_walk *walk;
3994	int young = 0;
3995	pte_t *pte = pvmw->pte;
3996	unsigned long addr = pvmw->address;
3997	struct vm_area_struct *vma = pvmw->vma;
3998	struct folio *folio = pfn_folio(pvmw->pfn);
3999	bool can_swap = !folio_is_file_lru(folio);
4000	struct mem_cgroup *memcg = folio_memcg(folio);
4001	struct pglist_data *pgdat = folio_pgdat(folio);
4002	struct lruvec *lruvec = mem_cgroup_lruvec(memcg, pgdat);
4003	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
4004	DEFINE_MAX_SEQ(lruvec);
4005	int old_gen, new_gen = lru_gen_from_seq(max_seq);
4006
4007	lockdep_assert_held(pvmw->ptl);
4008	VM_WARN_ON_ONCE_FOLIO(folio_test_lru(folio), folio);
4009
4010	if (spin_is_contended(pvmw->ptl))
4011		return;
4012
4013	/* exclude special VMAs containing anon pages from COW */
4014	if (vma->vm_flags & VM_SPECIAL)
4015		return;
4016
4017	/* avoid taking the LRU lock under the PTL when possible */
4018	walk = current->reclaim_state ? current->reclaim_state->mm_walk : NULL;
4019
4020	start = max(addr & PMD_MASK, vma->vm_start);
4021	end = min(addr | ~PMD_MASK, vma->vm_end - 1) + 1;
4022
4023	if (end - start > MIN_LRU_BATCH * PAGE_SIZE) {
4024		if (addr - start < MIN_LRU_BATCH * PAGE_SIZE / 2)
4025			end = start + MIN_LRU_BATCH * PAGE_SIZE;
4026		else if (end - addr < MIN_LRU_BATCH * PAGE_SIZE / 2)
4027			start = end - MIN_LRU_BATCH * PAGE_SIZE;
4028		else {
4029			start = addr - MIN_LRU_BATCH * PAGE_SIZE / 2;
4030			end = addr + MIN_LRU_BATCH * PAGE_SIZE / 2;
4031		}
4032	}
4033
4034	/* folio_update_gen() requires stable folio_memcg() */
4035	if (!mem_cgroup_trylock_pages(memcg))
4036		return;
4037
4038	arch_enter_lazy_mmu_mode();
4039
4040	pte -= (addr - start) / PAGE_SIZE;
4041
4042	for (i = 0, addr = start; addr != end; i++, addr += PAGE_SIZE) {
4043		unsigned long pfn;
4044		pte_t ptent = ptep_get(pte + i);
4045
4046		pfn = get_pte_pfn(ptent, vma, addr);
4047		if (pfn == -1)
4048			continue;
4049
4050		if (!pte_young(ptent))
4051			continue;
4052
4053		folio = get_pfn_folio(pfn, memcg, pgdat, can_swap);
4054		if (!folio)
4055			continue;
4056
4057		if (!ptep_test_and_clear_young(vma, addr, pte + i))
4058			VM_WARN_ON_ONCE(true);
4059
4060		young++;
4061
4062		if (pte_dirty(ptent) && !folio_test_dirty(folio) &&
4063		    !(folio_test_anon(folio) && folio_test_swapbacked(folio) &&
4064		      !folio_test_swapcache(folio)))
4065			folio_mark_dirty(folio);
4066
4067		if (walk) {
4068			old_gen = folio_update_gen(folio, new_gen);
4069			if (old_gen >= 0 && old_gen != new_gen)
4070				update_batch_size(walk, folio, old_gen, new_gen);
4071
4072			continue;
4073		}
4074
4075		old_gen = folio_lru_gen(folio);
4076		if (old_gen < 0)
4077			folio_set_referenced(folio);
4078		else if (old_gen != new_gen)
4079			folio_activate(folio);
4080	}
4081
4082	arch_leave_lazy_mmu_mode();
4083	mem_cgroup_unlock_pages();
4084
4085	/* feedback from rmap walkers to page table walkers */
4086	if (mm_state && suitable_to_scan(i, young))
4087		update_bloom_filter(mm_state, max_seq, pvmw->pmd);
4088}
4089
4090/******************************************************************************
4091 *                          memcg LRU
4092 ******************************************************************************/
4093
4094/* see the comment on MEMCG_NR_GENS */
4095enum {
4096	MEMCG_LRU_NOP,
4097	MEMCG_LRU_HEAD,
4098	MEMCG_LRU_TAIL,
4099	MEMCG_LRU_OLD,
4100	MEMCG_LRU_YOUNG,
4101};
4102
4103static void lru_gen_rotate_memcg(struct lruvec *lruvec, int op)
4104{
4105	int seg;
4106	int old, new;
4107	unsigned long flags;
4108	int bin = get_random_u32_below(MEMCG_NR_BINS);
4109	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
4110
4111	spin_lock_irqsave(&pgdat->memcg_lru.lock, flags);
4112
4113	VM_WARN_ON_ONCE(hlist_nulls_unhashed(&lruvec->lrugen.list));
4114
4115	seg = 0;
4116	new = old = lruvec->lrugen.gen;
4117
4118	/* see the comment on MEMCG_NR_GENS */
4119	if (op == MEMCG_LRU_HEAD)
4120		seg = MEMCG_LRU_HEAD;
4121	else if (op == MEMCG_LRU_TAIL)
4122		seg = MEMCG_LRU_TAIL;
4123	else if (op == MEMCG_LRU_OLD)
4124		new = get_memcg_gen(pgdat->memcg_lru.seq);
4125	else if (op == MEMCG_LRU_YOUNG)
4126		new = get_memcg_gen(pgdat->memcg_lru.seq + 1);
4127	else
4128		VM_WARN_ON_ONCE(true);
4129
4130	WRITE_ONCE(lruvec->lrugen.seg, seg);
4131	WRITE_ONCE(lruvec->lrugen.gen, new);
4132
4133	hlist_nulls_del_rcu(&lruvec->lrugen.list);
4134
4135	if (op == MEMCG_LRU_HEAD || op == MEMCG_LRU_OLD)
4136		hlist_nulls_add_head_rcu(&lruvec->lrugen.list, &pgdat->memcg_lru.fifo[new][bin]);
4137	else
4138		hlist_nulls_add_tail_rcu(&lruvec->lrugen.list, &pgdat->memcg_lru.fifo[new][bin]);
4139
4140	pgdat->memcg_lru.nr_memcgs[old]--;
4141	pgdat->memcg_lru.nr_memcgs[new]++;
4142
4143	if (!pgdat->memcg_lru.nr_memcgs[old] && old == get_memcg_gen(pgdat->memcg_lru.seq))
4144		WRITE_ONCE(pgdat->memcg_lru.seq, pgdat->memcg_lru.seq + 1);
4145
4146	spin_unlock_irqrestore(&pgdat->memcg_lru.lock, flags);
4147}
4148
4149#ifdef CONFIG_MEMCG
4150
4151void lru_gen_online_memcg(struct mem_cgroup *memcg)
4152{
4153	int gen;
4154	int nid;
4155	int bin = get_random_u32_below(MEMCG_NR_BINS);
4156
4157	for_each_node(nid) {
4158		struct pglist_data *pgdat = NODE_DATA(nid);
4159		struct lruvec *lruvec = get_lruvec(memcg, nid);
4160
4161		spin_lock_irq(&pgdat->memcg_lru.lock);
4162
4163		VM_WARN_ON_ONCE(!hlist_nulls_unhashed(&lruvec->lrugen.list));
4164
4165		gen = get_memcg_gen(pgdat->memcg_lru.seq);
4166
4167		lruvec->lrugen.gen = gen;
4168
4169		hlist_nulls_add_tail_rcu(&lruvec->lrugen.list, &pgdat->memcg_lru.fifo[gen][bin]);
4170		pgdat->memcg_lru.nr_memcgs[gen]++;
4171
4172		spin_unlock_irq(&pgdat->memcg_lru.lock);
4173	}
4174}
4175
4176void lru_gen_offline_memcg(struct mem_cgroup *memcg)
4177{
4178	int nid;
4179
4180	for_each_node(nid) {
4181		struct lruvec *lruvec = get_lruvec(memcg, nid);
4182
4183		lru_gen_rotate_memcg(lruvec, MEMCG_LRU_OLD);
4184	}
4185}
4186
4187void lru_gen_release_memcg(struct mem_cgroup *memcg)
4188{
4189	int gen;
4190	int nid;
4191
4192	for_each_node(nid) {
4193		struct pglist_data *pgdat = NODE_DATA(nid);
4194		struct lruvec *lruvec = get_lruvec(memcg, nid);
4195
4196		spin_lock_irq(&pgdat->memcg_lru.lock);
4197
4198		if (hlist_nulls_unhashed(&lruvec->lrugen.list))
4199			goto unlock;
4200
4201		gen = lruvec->lrugen.gen;
4202
4203		hlist_nulls_del_init_rcu(&lruvec->lrugen.list);
4204		pgdat->memcg_lru.nr_memcgs[gen]--;
4205
4206		if (!pgdat->memcg_lru.nr_memcgs[gen] && gen == get_memcg_gen(pgdat->memcg_lru.seq))
4207			WRITE_ONCE(pgdat->memcg_lru.seq, pgdat->memcg_lru.seq + 1);
4208unlock:
4209		spin_unlock_irq(&pgdat->memcg_lru.lock);
4210	}
4211}
4212
4213void lru_gen_soft_reclaim(struct mem_cgroup *memcg, int nid)
4214{
4215	struct lruvec *lruvec = get_lruvec(memcg, nid);
4216
4217	/* see the comment on MEMCG_NR_GENS */
4218	if (READ_ONCE(lruvec->lrugen.seg) != MEMCG_LRU_HEAD)
4219		lru_gen_rotate_memcg(lruvec, MEMCG_LRU_HEAD);
4220}
4221
4222#endif /* CONFIG_MEMCG */
4223
4224/******************************************************************************
4225 *                          the eviction
4226 ******************************************************************************/
4227
4228static bool sort_folio(struct lruvec *lruvec, struct folio *folio, struct scan_control *sc,
4229		       int tier_idx)
4230{
4231	bool success;
4232	int gen = folio_lru_gen(folio);
4233	int type = folio_is_file_lru(folio);
4234	int zone = folio_zonenum(folio);
4235	int delta = folio_nr_pages(folio);
4236	int refs = folio_lru_refs(folio);
4237	int tier = lru_tier_from_refs(refs);
4238	struct lru_gen_folio *lrugen = &lruvec->lrugen;
4239
4240	VM_WARN_ON_ONCE_FOLIO(gen >= MAX_NR_GENS, folio);
4241
4242	/* unevictable */
4243	if (!folio_evictable(folio)) {
4244		success = lru_gen_del_folio(lruvec, folio, true);
4245		VM_WARN_ON_ONCE_FOLIO(!success, folio);
4246		folio_set_unevictable(folio);
4247		lruvec_add_folio(lruvec, folio);
4248		__count_vm_events(UNEVICTABLE_PGCULLED, delta);
4249		return true;
4250	}
4251
4252	/* dirty lazyfree */
4253	if (type == LRU_GEN_FILE && folio_test_anon(folio) && folio_test_dirty(folio)) {
4254		success = lru_gen_del_folio(lruvec, folio, true);
4255		VM_WARN_ON_ONCE_FOLIO(!success, folio);
4256		folio_set_swapbacked(folio);
4257		lruvec_add_folio_tail(lruvec, folio);
4258		return true;
4259	}
4260
4261	/* promoted */
4262	if (gen != lru_gen_from_seq(lrugen->min_seq[type])) {
4263		list_move(&folio->lru, &lrugen->folios[gen][type][zone]);
4264		return true;
4265	}
4266
4267	/* protected */
4268	if (tier > tier_idx || refs == BIT(LRU_REFS_WIDTH)) {
4269		int hist = lru_hist_from_seq(lrugen->min_seq[type]);
4270
4271		gen = folio_inc_gen(lruvec, folio, false);
4272		list_move_tail(&folio->lru, &lrugen->folios[gen][type][zone]);
4273
4274		WRITE_ONCE(lrugen->protected[hist][type][tier - 1],
4275			   lrugen->protected[hist][type][tier - 1] + delta);
4276		return true;
4277	}
4278
4279	/* ineligible */
4280	if (zone > sc->reclaim_idx || skip_cma(folio, sc)) {
4281		gen = folio_inc_gen(lruvec, folio, false);
4282		list_move_tail(&folio->lru, &lrugen->folios[gen][type][zone]);
4283		return true;
4284	}
4285
4286	/* waiting for writeback */
4287	if (folio_test_locked(folio) || folio_test_writeback(folio) ||
4288	    (type == LRU_GEN_FILE && folio_test_dirty(folio))) {
4289		gen = folio_inc_gen(lruvec, folio, true);
4290		list_move(&folio->lru, &lrugen->folios[gen][type][zone]);
4291		return true;
4292	}
4293
4294	return false;
4295}
4296
4297static bool isolate_folio(struct lruvec *lruvec, struct folio *folio, struct scan_control *sc)
4298{
4299	bool success;
4300
4301	/* swap constrained */
4302	if (!(sc->gfp_mask & __GFP_IO) &&
4303	    (folio_test_dirty(folio) ||
4304	     (folio_test_anon(folio) && !folio_test_swapcache(folio))))
4305		return false;
4306
4307	/* raced with release_pages() */
4308	if (!folio_try_get(folio))
4309		return false;
4310
4311	/* raced with another isolation */
4312	if (!folio_test_clear_lru(folio)) {
4313		folio_put(folio);
4314		return false;
4315	}
4316
4317	/* see the comment on MAX_NR_TIERS */
4318	if (!folio_test_referenced(folio))
4319		set_mask_bits(&folio->flags, LRU_REFS_MASK | LRU_REFS_FLAGS, 0);
4320
4321	/* for shrink_folio_list() */
4322	folio_clear_reclaim(folio);
4323	folio_clear_referenced(folio);
4324
4325	success = lru_gen_del_folio(lruvec, folio, true);
4326	VM_WARN_ON_ONCE_FOLIO(!success, folio);
4327
4328	return true;
4329}
4330
4331static int scan_folios(struct lruvec *lruvec, struct scan_control *sc,
4332		       int type, int tier, struct list_head *list)
4333{
4334	int i;
4335	int gen;
4336	enum vm_event_item item;
4337	int sorted = 0;
4338	int scanned = 0;
4339	int isolated = 0;
4340	int skipped = 0;
4341	int remaining = MAX_LRU_BATCH;
4342	struct lru_gen_folio *lrugen = &lruvec->lrugen;
4343	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
4344
4345	VM_WARN_ON_ONCE(!list_empty(list));
4346
4347	if (get_nr_gens(lruvec, type) == MIN_NR_GENS)
4348		return 0;
4349
4350	gen = lru_gen_from_seq(lrugen->min_seq[type]);
4351
4352	for (i = MAX_NR_ZONES; i > 0; i--) {
4353		LIST_HEAD(moved);
4354		int skipped_zone = 0;
4355		int zone = (sc->reclaim_idx + i) % MAX_NR_ZONES;
4356		struct list_head *head = &lrugen->folios[gen][type][zone];
4357
4358		while (!list_empty(head)) {
4359			struct folio *folio = lru_to_folio(head);
4360			int delta = folio_nr_pages(folio);
4361
4362			VM_WARN_ON_ONCE_FOLIO(folio_test_unevictable(folio), folio);
4363			VM_WARN_ON_ONCE_FOLIO(folio_test_active(folio), folio);
4364			VM_WARN_ON_ONCE_FOLIO(folio_is_file_lru(folio) != type, folio);
4365			VM_WARN_ON_ONCE_FOLIO(folio_zonenum(folio) != zone, folio);
4366
4367			scanned += delta;
4368
4369			if (sort_folio(lruvec, folio, sc, tier))
4370				sorted += delta;
4371			else if (isolate_folio(lruvec, folio, sc)) {
4372				list_add(&folio->lru, list);
4373				isolated += delta;
4374			} else {
4375				list_move(&folio->lru, &moved);
4376				skipped_zone += delta;
4377			}
4378
4379			if (!--remaining || max(isolated, skipped_zone) >= MIN_LRU_BATCH)
4380				break;
4381		}
4382
4383		if (skipped_zone) {
4384			list_splice(&moved, head);
4385			__count_zid_vm_events(PGSCAN_SKIP, zone, skipped_zone);
4386			skipped += skipped_zone;
4387		}
4388
4389		if (!remaining || isolated >= MIN_LRU_BATCH)
4390			break;
4391	}
4392
4393	item = PGSCAN_KSWAPD + reclaimer_offset();
4394	if (!cgroup_reclaim(sc)) {
4395		__count_vm_events(item, isolated);
4396		__count_vm_events(PGREFILL, sorted);
4397	}
4398	__count_memcg_events(memcg, item, isolated);
4399	__count_memcg_events(memcg, PGREFILL, sorted);
4400	__count_vm_events(PGSCAN_ANON + type, isolated);
4401	trace_mm_vmscan_lru_isolate(sc->reclaim_idx, sc->order, MAX_LRU_BATCH,
4402				scanned, skipped, isolated,
4403				type ? LRU_INACTIVE_FILE : LRU_INACTIVE_ANON);
4404
4405	/*
4406	 * There might not be eligible folios due to reclaim_idx. Check the
4407	 * remaining to prevent livelock if it's not making progress.
4408	 */
4409	return isolated || !remaining ? scanned : 0;
4410}
4411
4412static int get_tier_idx(struct lruvec *lruvec, int type)
4413{
4414	int tier;
4415	struct ctrl_pos sp, pv;
4416
4417	/*
4418	 * To leave a margin for fluctuations, use a larger gain factor (1:2).
4419	 * This value is chosen because any other tier would have at least twice
4420	 * as many refaults as the first tier.
4421	 */
4422	read_ctrl_pos(lruvec, type, 0, 1, &sp);
4423	for (tier = 1; tier < MAX_NR_TIERS; tier++) {
4424		read_ctrl_pos(lruvec, type, tier, 2, &pv);
4425		if (!positive_ctrl_err(&sp, &pv))
4426			break;
4427	}
4428
4429	return tier - 1;
4430}
4431
4432static int get_type_to_scan(struct lruvec *lruvec, int swappiness, int *tier_idx)
4433{
4434	int type, tier;
4435	struct ctrl_pos sp, pv;
4436	int gain[ANON_AND_FILE] = { swappiness, 200 - swappiness };
4437
4438	/*
4439	 * Compare the first tier of anon with that of file to determine which
4440	 * type to scan. Also need to compare other tiers of the selected type
4441	 * with the first tier of the other type to determine the last tier (of
4442	 * the selected type) to evict.
4443	 */
4444	read_ctrl_pos(lruvec, LRU_GEN_ANON, 0, gain[LRU_GEN_ANON], &sp);
4445	read_ctrl_pos(lruvec, LRU_GEN_FILE, 0, gain[LRU_GEN_FILE], &pv);
4446	type = positive_ctrl_err(&sp, &pv);
4447
4448	read_ctrl_pos(lruvec, !type, 0, gain[!type], &sp);
4449	for (tier = 1; tier < MAX_NR_TIERS; tier++) {
4450		read_ctrl_pos(lruvec, type, tier, gain[type], &pv);
4451		if (!positive_ctrl_err(&sp, &pv))
4452			break;
4453	}
4454
4455	*tier_idx = tier - 1;
4456
4457	return type;
4458}
4459
4460static int isolate_folios(struct lruvec *lruvec, struct scan_control *sc, int swappiness,
4461			  int *type_scanned, struct list_head *list)
4462{
4463	int i;
4464	int type;
4465	int scanned;
4466	int tier = -1;
4467	DEFINE_MIN_SEQ(lruvec);
4468
4469	/*
4470	 * Try to make the obvious choice first, and if anon and file are both
4471	 * available from the same generation,
4472	 * 1. Interpret swappiness 1 as file first and MAX_SWAPPINESS as anon
4473	 *    first.
4474	 * 2. If !__GFP_IO, file first since clean pagecache is more likely to
4475	 *    exist than clean swapcache.
4476	 */
4477	if (!swappiness)
4478		type = LRU_GEN_FILE;
4479	else if (min_seq[LRU_GEN_ANON] < min_seq[LRU_GEN_FILE])
4480		type = LRU_GEN_ANON;
4481	else if (swappiness == 1)
4482		type = LRU_GEN_FILE;
4483	else if (swappiness == 200)
4484		type = LRU_GEN_ANON;
4485	else if (!(sc->gfp_mask & __GFP_IO))
4486		type = LRU_GEN_FILE;
4487	else
4488		type = get_type_to_scan(lruvec, swappiness, &tier);
4489
4490	for (i = !swappiness; i < ANON_AND_FILE; i++) {
4491		if (tier < 0)
4492			tier = get_tier_idx(lruvec, type);
4493
4494		scanned = scan_folios(lruvec, sc, type, tier, list);
4495		if (scanned)
4496			break;
4497
4498		type = !type;
4499		tier = -1;
4500	}
4501
4502	*type_scanned = type;
4503
4504	return scanned;
4505}
4506
4507static int evict_folios(struct lruvec *lruvec, struct scan_control *sc, int swappiness)
4508{
4509	int type;
4510	int scanned;
4511	int reclaimed;
4512	LIST_HEAD(list);
4513	LIST_HEAD(clean);
4514	struct folio *folio;
4515	struct folio *next;
4516	enum vm_event_item item;
4517	struct reclaim_stat stat;
4518	struct lru_gen_mm_walk *walk;
4519	bool skip_retry = false;
4520	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
4521	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
4522
4523	spin_lock_irq(&lruvec->lru_lock);
4524
4525	scanned = isolate_folios(lruvec, sc, swappiness, &type, &list);
4526
4527	scanned += try_to_inc_min_seq(lruvec, swappiness);
4528
4529	if (get_nr_gens(lruvec, !swappiness) == MIN_NR_GENS)
4530		scanned = 0;
4531
4532	spin_unlock_irq(&lruvec->lru_lock);
4533
4534	if (list_empty(&list))
4535		return scanned;
4536retry:
4537	reclaimed = shrink_folio_list(&list, pgdat, sc, &stat, false);
4538	sc->nr_reclaimed += reclaimed;
4539	trace_mm_vmscan_lru_shrink_inactive(pgdat->node_id,
4540			scanned, reclaimed, &stat, sc->priority,
4541			type ? LRU_INACTIVE_FILE : LRU_INACTIVE_ANON);
4542
4543	list_for_each_entry_safe_reverse(folio, next, &list, lru) {
4544		if (!folio_evictable(folio)) {
4545			list_del(&folio->lru);
4546			folio_putback_lru(folio);
4547			continue;
4548		}
4549
4550		if (folio_test_reclaim(folio) &&
4551		    (folio_test_dirty(folio) || folio_test_writeback(folio))) {
4552			/* restore LRU_REFS_FLAGS cleared by isolate_folio() */
4553			if (folio_test_workingset(folio))
4554				folio_set_referenced(folio);
4555			continue;
4556		}
4557
4558		if (skip_retry || folio_test_active(folio) || folio_test_referenced(folio) ||
4559		    folio_mapped(folio) || folio_test_locked(folio) ||
4560		    folio_test_dirty(folio) || folio_test_writeback(folio)) {
4561			/* don't add rejected folios to the oldest generation */
4562			set_mask_bits(&folio->flags, LRU_REFS_MASK | LRU_REFS_FLAGS,
4563				      BIT(PG_active));
4564			continue;
4565		}
4566
4567		/* retry folios that may have missed folio_rotate_reclaimable() */
4568		list_move(&folio->lru, &clean);
4569		sc->nr_scanned -= folio_nr_pages(folio);
4570	}
4571
4572	spin_lock_irq(&lruvec->lru_lock);
4573
4574	move_folios_to_lru(lruvec, &list);
4575
4576	walk = current->reclaim_state->mm_walk;
4577	if (walk && walk->batched) {
4578		walk->lruvec = lruvec;
4579		reset_batch_size(walk);
4580	}
4581
4582	item = PGSTEAL_KSWAPD + reclaimer_offset();
4583	if (!cgroup_reclaim(sc))
4584		__count_vm_events(item, reclaimed);
4585	__count_memcg_events(memcg, item, reclaimed);
4586	__count_vm_events(PGSTEAL_ANON + type, reclaimed);
4587
4588	spin_unlock_irq(&lruvec->lru_lock);
4589
 
 
 
 
4590	list_splice_init(&clean, &list);
4591
4592	if (!list_empty(&list)) {
4593		skip_retry = true;
4594		goto retry;
4595	}
4596
4597	return scanned;
4598}
4599
4600static bool should_run_aging(struct lruvec *lruvec, unsigned long max_seq,
4601			     bool can_swap, unsigned long *nr_to_scan)
4602{
4603	int gen, type, zone;
4604	unsigned long old = 0;
4605	unsigned long young = 0;
4606	unsigned long total = 0;
4607	struct lru_gen_folio *lrugen = &lruvec->lrugen;
 
4608	DEFINE_MIN_SEQ(lruvec);
4609
4610	/* whether this lruvec is completely out of cold folios */
4611	if (min_seq[!can_swap] + MIN_NR_GENS > max_seq) {
4612		*nr_to_scan = 0;
4613		return true;
4614	}
4615
4616	for (type = !can_swap; type < ANON_AND_FILE; type++) {
4617		unsigned long seq;
4618
4619		for (seq = min_seq[type]; seq <= max_seq; seq++) {
4620			unsigned long size = 0;
4621
4622			gen = lru_gen_from_seq(seq);
4623
4624			for (zone = 0; zone < MAX_NR_ZONES; zone++)
4625				size += max(READ_ONCE(lrugen->nr_pages[gen][type][zone]), 0L);
4626
4627			total += size;
4628			if (seq == max_seq)
4629				young += size;
4630			else if (seq + MIN_NR_GENS == max_seq)
4631				old += size;
4632		}
4633	}
4634
4635	*nr_to_scan = total;
 
 
 
 
 
 
4636
4637	/*
4638	 * The aging tries to be lazy to reduce the overhead, while the eviction
4639	 * stalls when the number of generations reaches MIN_NR_GENS. Hence, the
4640	 * ideal number of generations is MIN_NR_GENS+1.
4641	 */
4642	if (min_seq[!can_swap] + MIN_NR_GENS < max_seq)
4643		return false;
4644
4645	/*
4646	 * It's also ideal to spread pages out evenly, i.e., 1/(MIN_NR_GENS+1)
4647	 * of the total number of pages for each generation. A reasonable range
4648	 * for this average portion is [1/MIN_NR_GENS, 1/(MIN_NR_GENS+2)]. The
4649	 * aging cares about the upper bound of hot pages, while the eviction
4650	 * cares about the lower bound of cold pages.
4651	 */
4652	if (young * MIN_NR_GENS > total)
4653		return true;
4654	if (old * (MIN_NR_GENS + 2) < total)
4655		return true;
4656
4657	return false;
4658}
4659
4660/*
4661 * For future optimizations:
4662 * 1. Defer try_to_inc_max_seq() to workqueues to reduce latency for memcg
4663 *    reclaim.
4664 */
4665static long get_nr_to_scan(struct lruvec *lruvec, struct scan_control *sc, bool can_swap)
4666{
4667	bool success;
4668	unsigned long nr_to_scan;
4669	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
4670	DEFINE_MAX_SEQ(lruvec);
4671
4672	if (mem_cgroup_below_min(sc->target_mem_cgroup, memcg))
4673		return -1;
4674
4675	success = should_run_aging(lruvec, max_seq, can_swap, &nr_to_scan);
 
4676
4677	/* try to scrape all its memory if this memcg was deleted */
4678	if (nr_to_scan && !mem_cgroup_online(memcg))
4679		return nr_to_scan;
4680
4681	/* try to get away with not aging at the default priority */
4682	if (!success || sc->priority == DEF_PRIORITY)
4683		return nr_to_scan >> sc->priority;
4684
4685	/* stop scanning this lruvec as it's low on cold folios */
4686	return try_to_inc_max_seq(lruvec, max_seq, can_swap, false) ? -1 : 0;
4687}
4688
4689static bool should_abort_scan(struct lruvec *lruvec, struct scan_control *sc)
4690{
4691	int i;
4692	enum zone_watermarks mark;
4693
4694	/* don't abort memcg reclaim to ensure fairness */
4695	if (!root_reclaim(sc))
4696		return false;
4697
4698	if (sc->nr_reclaimed >= max(sc->nr_to_reclaim, compact_gap(sc->order)))
4699		return true;
4700
4701	/* check the order to exclude compaction-induced reclaim */
4702	if (!current_is_kswapd() || sc->order)
4703		return false;
4704
4705	mark = sysctl_numa_balancing_mode & NUMA_BALANCING_MEMORY_TIERING ?
4706	       WMARK_PROMO : WMARK_HIGH;
4707
4708	for (i = 0; i <= sc->reclaim_idx; i++) {
4709		struct zone *zone = lruvec_pgdat(lruvec)->node_zones + i;
4710		unsigned long size = wmark_pages(zone, mark) + MIN_LRU_BATCH;
4711
4712		if (managed_zone(zone) && !zone_watermark_ok(zone, 0, size, sc->reclaim_idx, 0))
4713			return false;
4714	}
4715
4716	/* kswapd should abort if all eligible zones are safe */
4717	return true;
4718}
4719
4720static bool try_to_shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
4721{
4722	long nr_to_scan;
4723	unsigned long scanned = 0;
4724	int swappiness = get_swappiness(lruvec, sc);
4725
 
 
 
 
4726	while (true) {
4727		int delta;
4728
4729		nr_to_scan = get_nr_to_scan(lruvec, sc, swappiness);
4730		if (nr_to_scan <= 0)
4731			break;
4732
4733		delta = evict_folios(lruvec, sc, swappiness);
4734		if (!delta)
4735			break;
4736
4737		scanned += delta;
4738		if (scanned >= nr_to_scan)
4739			break;
4740
4741		if (should_abort_scan(lruvec, sc))
4742			break;
4743
4744		cond_resched();
4745	}
4746
4747	/* whether this lruvec should be rotated */
4748	return nr_to_scan < 0;
4749}
4750
4751static int shrink_one(struct lruvec *lruvec, struct scan_control *sc)
4752{
4753	bool success;
4754	unsigned long scanned = sc->nr_scanned;
4755	unsigned long reclaimed = sc->nr_reclaimed;
4756	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
4757	struct pglist_data *pgdat = lruvec_pgdat(lruvec);
4758
4759	mem_cgroup_calculate_protection(NULL, memcg);
4760
4761	if (mem_cgroup_below_min(NULL, memcg))
4762		return MEMCG_LRU_YOUNG;
4763
4764	if (mem_cgroup_below_low(NULL, memcg)) {
4765		/* see the comment on MEMCG_NR_GENS */
4766		if (READ_ONCE(lruvec->lrugen.seg) != MEMCG_LRU_TAIL)
4767			return MEMCG_LRU_TAIL;
4768
4769		memcg_memory_event(memcg, MEMCG_LOW);
4770	}
4771
4772	success = try_to_shrink_lruvec(lruvec, sc);
4773
4774	shrink_slab(sc->gfp_mask, pgdat->node_id, memcg, sc->priority);
4775
4776	if (!sc->proactive)
4777		vmpressure(sc->gfp_mask, memcg, false, sc->nr_scanned - scanned,
4778			   sc->nr_reclaimed - reclaimed);
4779
4780	flush_reclaim_state(sc);
4781
4782	if (success && mem_cgroup_online(memcg))
4783		return MEMCG_LRU_YOUNG;
4784
4785	if (!success && lruvec_is_sizable(lruvec, sc))
4786		return 0;
4787
4788	/* one retry if offlined or too small */
4789	return READ_ONCE(lruvec->lrugen.seg) != MEMCG_LRU_TAIL ?
4790	       MEMCG_LRU_TAIL : MEMCG_LRU_YOUNG;
4791}
4792
4793static void shrink_many(struct pglist_data *pgdat, struct scan_control *sc)
4794{
4795	int op;
4796	int gen;
4797	int bin;
4798	int first_bin;
4799	struct lruvec *lruvec;
4800	struct lru_gen_folio *lrugen;
4801	struct mem_cgroup *memcg;
4802	struct hlist_nulls_node *pos;
4803
4804	gen = get_memcg_gen(READ_ONCE(pgdat->memcg_lru.seq));
4805	bin = first_bin = get_random_u32_below(MEMCG_NR_BINS);
4806restart:
4807	op = 0;
4808	memcg = NULL;
4809
4810	rcu_read_lock();
4811
4812	hlist_nulls_for_each_entry_rcu(lrugen, pos, &pgdat->memcg_lru.fifo[gen][bin], list) {
4813		if (op) {
4814			lru_gen_rotate_memcg(lruvec, op);
4815			op = 0;
4816		}
4817
4818		mem_cgroup_put(memcg);
4819		memcg = NULL;
4820
4821		if (gen != READ_ONCE(lrugen->gen))
4822			continue;
4823
4824		lruvec = container_of(lrugen, struct lruvec, lrugen);
4825		memcg = lruvec_memcg(lruvec);
4826
4827		if (!mem_cgroup_tryget(memcg)) {
4828			lru_gen_release_memcg(memcg);
4829			memcg = NULL;
4830			continue;
4831		}
4832
4833		rcu_read_unlock();
4834
4835		op = shrink_one(lruvec, sc);
4836
4837		rcu_read_lock();
4838
4839		if (should_abort_scan(lruvec, sc))
4840			break;
4841	}
4842
4843	rcu_read_unlock();
4844
4845	if (op)
4846		lru_gen_rotate_memcg(lruvec, op);
4847
4848	mem_cgroup_put(memcg);
4849
4850	if (!is_a_nulls(pos))
4851		return;
4852
4853	/* restart if raced with lru_gen_rotate_memcg() */
4854	if (gen != get_nulls_value(pos))
4855		goto restart;
4856
4857	/* try the rest of the bins of the current generation */
4858	bin = get_memcg_bin(bin + 1);
4859	if (bin != first_bin)
4860		goto restart;
4861}
4862
4863static void lru_gen_shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
4864{
4865	struct blk_plug plug;
4866
4867	VM_WARN_ON_ONCE(root_reclaim(sc));
4868	VM_WARN_ON_ONCE(!sc->may_writepage || !sc->may_unmap);
4869
4870	lru_add_drain();
4871
4872	blk_start_plug(&plug);
4873
4874	set_mm_walk(NULL, sc->proactive);
4875
4876	if (try_to_shrink_lruvec(lruvec, sc))
4877		lru_gen_rotate_memcg(lruvec, MEMCG_LRU_YOUNG);
4878
4879	clear_mm_walk();
4880
4881	blk_finish_plug(&plug);
4882}
4883
4884static void set_initial_priority(struct pglist_data *pgdat, struct scan_control *sc)
4885{
4886	int priority;
4887	unsigned long reclaimable;
 
4888
4889	if (sc->priority != DEF_PRIORITY || sc->nr_to_reclaim < MIN_LRU_BATCH)
4890		return;
4891	/*
4892	 * Determine the initial priority based on
4893	 * (total >> priority) * reclaimed_to_scanned_ratio = nr_to_reclaim,
4894	 * where reclaimed_to_scanned_ratio = inactive / total.
4895	 */
4896	reclaimable = node_page_state(pgdat, NR_INACTIVE_FILE);
4897	if (can_reclaim_anon_pages(NULL, pgdat->node_id, sc))
4898		reclaimable += node_page_state(pgdat, NR_INACTIVE_ANON);
4899
4900	/* round down reclaimable and round up sc->nr_to_reclaim */
4901	priority = fls_long(reclaimable) - 1 - fls_long(sc->nr_to_reclaim - 1);
4902
4903	sc->priority = clamp(priority, 0, DEF_PRIORITY);
4904}
4905
4906static void lru_gen_shrink_node(struct pglist_data *pgdat, struct scan_control *sc)
4907{
4908	struct blk_plug plug;
4909	unsigned long reclaimed = sc->nr_reclaimed;
4910
4911	VM_WARN_ON_ONCE(!root_reclaim(sc));
4912
4913	/*
4914	 * Unmapped clean folios are already prioritized. Scanning for more of
4915	 * them is likely futile and can cause high reclaim latency when there
4916	 * is a large number of memcgs.
4917	 */
4918	if (!sc->may_writepage || !sc->may_unmap)
4919		goto done;
4920
4921	lru_add_drain();
4922
4923	blk_start_plug(&plug);
4924
4925	set_mm_walk(pgdat, sc->proactive);
4926
4927	set_initial_priority(pgdat, sc);
4928
4929	if (current_is_kswapd())
4930		sc->nr_reclaimed = 0;
4931
4932	if (mem_cgroup_disabled())
4933		shrink_one(&pgdat->__lruvec, sc);
4934	else
4935		shrink_many(pgdat, sc);
4936
4937	if (current_is_kswapd())
4938		sc->nr_reclaimed += reclaimed;
4939
4940	clear_mm_walk();
4941
4942	blk_finish_plug(&plug);
4943done:
4944	/* kswapd should never fail */
4945	pgdat->kswapd_failures = 0;
4946}
4947
4948/******************************************************************************
4949 *                          state change
4950 ******************************************************************************/
4951
4952static bool __maybe_unused state_is_valid(struct lruvec *lruvec)
4953{
4954	struct lru_gen_folio *lrugen = &lruvec->lrugen;
4955
4956	if (lrugen->enabled) {
4957		enum lru_list lru;
4958
4959		for_each_evictable_lru(lru) {
4960			if (!list_empty(&lruvec->lists[lru]))
4961				return false;
4962		}
4963	} else {
4964		int gen, type, zone;
4965
4966		for_each_gen_type_zone(gen, type, zone) {
4967			if (!list_empty(&lrugen->folios[gen][type][zone]))
4968				return false;
4969		}
4970	}
4971
4972	return true;
4973}
4974
4975static bool fill_evictable(struct lruvec *lruvec)
4976{
4977	enum lru_list lru;
4978	int remaining = MAX_LRU_BATCH;
4979
4980	for_each_evictable_lru(lru) {
4981		int type = is_file_lru(lru);
4982		bool active = is_active_lru(lru);
4983		struct list_head *head = &lruvec->lists[lru];
4984
4985		while (!list_empty(head)) {
4986			bool success;
4987			struct folio *folio = lru_to_folio(head);
4988
4989			VM_WARN_ON_ONCE_FOLIO(folio_test_unevictable(folio), folio);
4990			VM_WARN_ON_ONCE_FOLIO(folio_test_active(folio) != active, folio);
4991			VM_WARN_ON_ONCE_FOLIO(folio_is_file_lru(folio) != type, folio);
4992			VM_WARN_ON_ONCE_FOLIO(folio_lru_gen(folio) != -1, folio);
4993
4994			lruvec_del_folio(lruvec, folio);
4995			success = lru_gen_add_folio(lruvec, folio, false);
4996			VM_WARN_ON_ONCE(!success);
4997
4998			if (!--remaining)
4999				return false;
5000		}
5001	}
5002
5003	return true;
5004}
5005
5006static bool drain_evictable(struct lruvec *lruvec)
5007{
5008	int gen, type, zone;
5009	int remaining = MAX_LRU_BATCH;
5010
5011	for_each_gen_type_zone(gen, type, zone) {
5012		struct list_head *head = &lruvec->lrugen.folios[gen][type][zone];
5013
5014		while (!list_empty(head)) {
5015			bool success;
5016			struct folio *folio = lru_to_folio(head);
5017
5018			VM_WARN_ON_ONCE_FOLIO(folio_test_unevictable(folio), folio);
5019			VM_WARN_ON_ONCE_FOLIO(folio_test_active(folio), folio);
5020			VM_WARN_ON_ONCE_FOLIO(folio_is_file_lru(folio) != type, folio);
5021			VM_WARN_ON_ONCE_FOLIO(folio_zonenum(folio) != zone, folio);
5022
5023			success = lru_gen_del_folio(lruvec, folio, false);
5024			VM_WARN_ON_ONCE(!success);
5025			lruvec_add_folio(lruvec, folio);
5026
5027			if (!--remaining)
5028				return false;
5029		}
5030	}
5031
5032	return true;
5033}
5034
5035static void lru_gen_change_state(bool enabled)
5036{
5037	static DEFINE_MUTEX(state_mutex);
5038
5039	struct mem_cgroup *memcg;
5040
5041	cgroup_lock();
5042	cpus_read_lock();
5043	get_online_mems();
5044	mutex_lock(&state_mutex);
5045
5046	if (enabled == lru_gen_enabled())
5047		goto unlock;
5048
5049	if (enabled)
5050		static_branch_enable_cpuslocked(&lru_gen_caps[LRU_GEN_CORE]);
5051	else
5052		static_branch_disable_cpuslocked(&lru_gen_caps[LRU_GEN_CORE]);
5053
5054	memcg = mem_cgroup_iter(NULL, NULL, NULL);
5055	do {
5056		int nid;
5057
5058		for_each_node(nid) {
5059			struct lruvec *lruvec = get_lruvec(memcg, nid);
5060
5061			spin_lock_irq(&lruvec->lru_lock);
5062
5063			VM_WARN_ON_ONCE(!seq_is_valid(lruvec));
5064			VM_WARN_ON_ONCE(!state_is_valid(lruvec));
5065
5066			lruvec->lrugen.enabled = enabled;
5067
5068			while (!(enabled ? fill_evictable(lruvec) : drain_evictable(lruvec))) {
5069				spin_unlock_irq(&lruvec->lru_lock);
5070				cond_resched();
5071				spin_lock_irq(&lruvec->lru_lock);
5072			}
5073
5074			spin_unlock_irq(&lruvec->lru_lock);
5075		}
5076
5077		cond_resched();
5078	} while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)));
5079unlock:
5080	mutex_unlock(&state_mutex);
5081	put_online_mems();
5082	cpus_read_unlock();
5083	cgroup_unlock();
5084}
5085
5086/******************************************************************************
5087 *                          sysfs interface
5088 ******************************************************************************/
5089
5090static ssize_t min_ttl_ms_show(struct kobject *kobj, struct kobj_attribute *attr, char *buf)
5091{
5092	return sysfs_emit(buf, "%u\n", jiffies_to_msecs(READ_ONCE(lru_gen_min_ttl)));
5093}
5094
5095/* see Documentation/admin-guide/mm/multigen_lru.rst for details */
5096static ssize_t min_ttl_ms_store(struct kobject *kobj, struct kobj_attribute *attr,
5097				const char *buf, size_t len)
5098{
5099	unsigned int msecs;
5100
5101	if (kstrtouint(buf, 0, &msecs))
5102		return -EINVAL;
5103
5104	WRITE_ONCE(lru_gen_min_ttl, msecs_to_jiffies(msecs));
5105
5106	return len;
5107}
5108
5109static struct kobj_attribute lru_gen_min_ttl_attr = __ATTR_RW(min_ttl_ms);
5110
5111static ssize_t enabled_show(struct kobject *kobj, struct kobj_attribute *attr, char *buf)
5112{
5113	unsigned int caps = 0;
5114
5115	if (get_cap(LRU_GEN_CORE))
5116		caps |= BIT(LRU_GEN_CORE);
5117
5118	if (should_walk_mmu())
5119		caps |= BIT(LRU_GEN_MM_WALK);
5120
5121	if (should_clear_pmd_young())
5122		caps |= BIT(LRU_GEN_NONLEAF_YOUNG);
5123
5124	return sysfs_emit(buf, "0x%04x\n", caps);
5125}
5126
5127/* see Documentation/admin-guide/mm/multigen_lru.rst for details */
5128static ssize_t enabled_store(struct kobject *kobj, struct kobj_attribute *attr,
5129			     const char *buf, size_t len)
5130{
5131	int i;
5132	unsigned int caps;
5133
5134	if (tolower(*buf) == 'n')
5135		caps = 0;
5136	else if (tolower(*buf) == 'y')
5137		caps = -1;
5138	else if (kstrtouint(buf, 0, &caps))
5139		return -EINVAL;
5140
5141	for (i = 0; i < NR_LRU_GEN_CAPS; i++) {
5142		bool enabled = caps & BIT(i);
5143
5144		if (i == LRU_GEN_CORE)
5145			lru_gen_change_state(enabled);
5146		else if (enabled)
5147			static_branch_enable(&lru_gen_caps[i]);
5148		else
5149			static_branch_disable(&lru_gen_caps[i]);
5150	}
5151
5152	return len;
5153}
5154
5155static struct kobj_attribute lru_gen_enabled_attr = __ATTR_RW(enabled);
5156
5157static struct attribute *lru_gen_attrs[] = {
5158	&lru_gen_min_ttl_attr.attr,
5159	&lru_gen_enabled_attr.attr,
5160	NULL
5161};
5162
5163static const struct attribute_group lru_gen_attr_group = {
5164	.name = "lru_gen",
5165	.attrs = lru_gen_attrs,
5166};
5167
5168/******************************************************************************
5169 *                          debugfs interface
5170 ******************************************************************************/
5171
5172static void *lru_gen_seq_start(struct seq_file *m, loff_t *pos)
5173{
5174	struct mem_cgroup *memcg;
5175	loff_t nr_to_skip = *pos;
5176
5177	m->private = kvmalloc(PATH_MAX, GFP_KERNEL);
5178	if (!m->private)
5179		return ERR_PTR(-ENOMEM);
5180
5181	memcg = mem_cgroup_iter(NULL, NULL, NULL);
5182	do {
5183		int nid;
5184
5185		for_each_node_state(nid, N_MEMORY) {
5186			if (!nr_to_skip--)
5187				return get_lruvec(memcg, nid);
5188		}
5189	} while ((memcg = mem_cgroup_iter(NULL, memcg, NULL)));
5190
5191	return NULL;
5192}
5193
5194static void lru_gen_seq_stop(struct seq_file *m, void *v)
5195{
5196	if (!IS_ERR_OR_NULL(v))
5197		mem_cgroup_iter_break(NULL, lruvec_memcg(v));
5198
5199	kvfree(m->private);
5200	m->private = NULL;
5201}
5202
5203static void *lru_gen_seq_next(struct seq_file *m, void *v, loff_t *pos)
5204{
5205	int nid = lruvec_pgdat(v)->node_id;
5206	struct mem_cgroup *memcg = lruvec_memcg(v);
5207
5208	++*pos;
5209
5210	nid = next_memory_node(nid);
5211	if (nid == MAX_NUMNODES) {
5212		memcg = mem_cgroup_iter(NULL, memcg, NULL);
5213		if (!memcg)
5214			return NULL;
5215
5216		nid = first_memory_node;
5217	}
5218
5219	return get_lruvec(memcg, nid);
5220}
5221
5222static void lru_gen_seq_show_full(struct seq_file *m, struct lruvec *lruvec,
5223				  unsigned long max_seq, unsigned long *min_seq,
5224				  unsigned long seq)
5225{
5226	int i;
5227	int type, tier;
5228	int hist = lru_hist_from_seq(seq);
5229	struct lru_gen_folio *lrugen = &lruvec->lrugen;
5230	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
5231
5232	for (tier = 0; tier < MAX_NR_TIERS; tier++) {
5233		seq_printf(m, "            %10d", tier);
5234		for (type = 0; type < ANON_AND_FILE; type++) {
5235			const char *s = "   ";
5236			unsigned long n[3] = {};
5237
5238			if (seq == max_seq) {
5239				s = "RT ";
5240				n[0] = READ_ONCE(lrugen->avg_refaulted[type][tier]);
5241				n[1] = READ_ONCE(lrugen->avg_total[type][tier]);
5242			} else if (seq == min_seq[type] || NR_HIST_GENS > 1) {
5243				s = "rep";
5244				n[0] = atomic_long_read(&lrugen->refaulted[hist][type][tier]);
5245				n[1] = atomic_long_read(&lrugen->evicted[hist][type][tier]);
5246				if (tier)
5247					n[2] = READ_ONCE(lrugen->protected[hist][type][tier - 1]);
5248			}
5249
5250			for (i = 0; i < 3; i++)
5251				seq_printf(m, " %10lu%c", n[i], s[i]);
5252		}
5253		seq_putc(m, '\n');
5254	}
5255
5256	if (!mm_state)
5257		return;
5258
5259	seq_puts(m, "                      ");
5260	for (i = 0; i < NR_MM_STATS; i++) {
5261		const char *s = "      ";
5262		unsigned long n = 0;
5263
5264		if (seq == max_seq && NR_HIST_GENS == 1) {
5265			s = "LOYNFA";
5266			n = READ_ONCE(mm_state->stats[hist][i]);
5267		} else if (seq != max_seq && NR_HIST_GENS > 1) {
5268			s = "loynfa";
5269			n = READ_ONCE(mm_state->stats[hist][i]);
5270		}
5271
5272		seq_printf(m, " %10lu%c", n, s[i]);
5273	}
5274	seq_putc(m, '\n');
5275}
5276
5277/* see Documentation/admin-guide/mm/multigen_lru.rst for details */
5278static int lru_gen_seq_show(struct seq_file *m, void *v)
5279{
5280	unsigned long seq;
5281	bool full = !debugfs_real_fops(m->file)->write;
5282	struct lruvec *lruvec = v;
5283	struct lru_gen_folio *lrugen = &lruvec->lrugen;
5284	int nid = lruvec_pgdat(lruvec)->node_id;
5285	struct mem_cgroup *memcg = lruvec_memcg(lruvec);
5286	DEFINE_MAX_SEQ(lruvec);
5287	DEFINE_MIN_SEQ(lruvec);
5288
5289	if (nid == first_memory_node) {
5290		const char *path = memcg ? m->private : "";
5291
5292#ifdef CONFIG_MEMCG
5293		if (memcg)
5294			cgroup_path(memcg->css.cgroup, m->private, PATH_MAX);
5295#endif
5296		seq_printf(m, "memcg %5hu %s\n", mem_cgroup_id(memcg), path);
5297	}
5298
5299	seq_printf(m, " node %5d\n", nid);
5300
5301	if (!full)
5302		seq = min_seq[LRU_GEN_ANON];
5303	else if (max_seq >= MAX_NR_GENS)
5304		seq = max_seq - MAX_NR_GENS + 1;
5305	else
5306		seq = 0;
5307
5308	for (; seq <= max_seq; seq++) {
5309		int type, zone;
5310		int gen = lru_gen_from_seq(seq);
5311		unsigned long birth = READ_ONCE(lruvec->lrugen.timestamps[gen]);
5312
5313		seq_printf(m, " %10lu %10u", seq, jiffies_to_msecs(jiffies - birth));
5314
5315		for (type = 0; type < ANON_AND_FILE; type++) {
5316			unsigned long size = 0;
5317			char mark = full && seq < min_seq[type] ? 'x' : ' ';
5318
5319			for (zone = 0; zone < MAX_NR_ZONES; zone++)
5320				size += max(READ_ONCE(lrugen->nr_pages[gen][type][zone]), 0L);
5321
5322			seq_printf(m, " %10lu%c", size, mark);
5323		}
5324
5325		seq_putc(m, '\n');
5326
5327		if (full)
5328			lru_gen_seq_show_full(m, lruvec, max_seq, min_seq, seq);
5329	}
5330
5331	return 0;
5332}
5333
5334static const struct seq_operations lru_gen_seq_ops = {
5335	.start = lru_gen_seq_start,
5336	.stop = lru_gen_seq_stop,
5337	.next = lru_gen_seq_next,
5338	.show = lru_gen_seq_show,
5339};
5340
5341static int run_aging(struct lruvec *lruvec, unsigned long seq,
5342		     bool can_swap, bool force_scan)
5343{
5344	DEFINE_MAX_SEQ(lruvec);
5345	DEFINE_MIN_SEQ(lruvec);
5346
5347	if (seq < max_seq)
5348		return 0;
5349
5350	if (seq > max_seq)
5351		return -EINVAL;
5352
5353	if (!force_scan && min_seq[!can_swap] + MAX_NR_GENS - 1 <= max_seq)
5354		return -ERANGE;
5355
5356	try_to_inc_max_seq(lruvec, max_seq, can_swap, force_scan);
5357
5358	return 0;
5359}
5360
5361static int run_eviction(struct lruvec *lruvec, unsigned long seq, struct scan_control *sc,
5362			int swappiness, unsigned long nr_to_reclaim)
5363{
5364	DEFINE_MAX_SEQ(lruvec);
5365
5366	if (seq + MIN_NR_GENS > max_seq)
5367		return -EINVAL;
5368
5369	sc->nr_reclaimed = 0;
5370
5371	while (!signal_pending(current)) {
5372		DEFINE_MIN_SEQ(lruvec);
5373
5374		if (seq < min_seq[!swappiness])
5375			return 0;
5376
5377		if (sc->nr_reclaimed >= nr_to_reclaim)
5378			return 0;
5379
5380		if (!evict_folios(lruvec, sc, swappiness))
5381			return 0;
5382
5383		cond_resched();
5384	}
5385
5386	return -EINTR;
5387}
5388
5389static int run_cmd(char cmd, int memcg_id, int nid, unsigned long seq,
5390		   struct scan_control *sc, int swappiness, unsigned long opt)
5391{
5392	struct lruvec *lruvec;
5393	int err = -EINVAL;
5394	struct mem_cgroup *memcg = NULL;
5395
5396	if (nid < 0 || nid >= MAX_NUMNODES || !node_state(nid, N_MEMORY))
5397		return -EINVAL;
5398
5399	if (!mem_cgroup_disabled()) {
5400		rcu_read_lock();
5401
5402		memcg = mem_cgroup_from_id(memcg_id);
5403		if (!mem_cgroup_tryget(memcg))
5404			memcg = NULL;
5405
5406		rcu_read_unlock();
5407
5408		if (!memcg)
5409			return -EINVAL;
5410	}
5411
5412	if (memcg_id != mem_cgroup_id(memcg))
5413		goto done;
5414
5415	lruvec = get_lruvec(memcg, nid);
5416
5417	if (swappiness < 0)
5418		swappiness = get_swappiness(lruvec, sc);
5419	else if (swappiness > 200)
5420		goto done;
5421
5422	switch (cmd) {
5423	case '+':
5424		err = run_aging(lruvec, seq, swappiness, opt);
5425		break;
5426	case '-':
5427		err = run_eviction(lruvec, seq, sc, swappiness, opt);
5428		break;
5429	}
5430done:
5431	mem_cgroup_put(memcg);
5432
5433	return err;
5434}
5435
5436/* see Documentation/admin-guide/mm/multigen_lru.rst for details */
5437static ssize_t lru_gen_seq_write(struct file *file, const char __user *src,
5438				 size_t len, loff_t *pos)
5439{
5440	void *buf;
5441	char *cur, *next;
5442	unsigned int flags;
5443	struct blk_plug plug;
5444	int err = -EINVAL;
5445	struct scan_control sc = {
5446		.may_writepage = true,
5447		.may_unmap = true,
5448		.may_swap = true,
5449		.reclaim_idx = MAX_NR_ZONES - 1,
5450		.gfp_mask = GFP_KERNEL,
5451	};
5452
5453	buf = kvmalloc(len + 1, GFP_KERNEL);
5454	if (!buf)
5455		return -ENOMEM;
5456
5457	if (copy_from_user(buf, src, len)) {
5458		kvfree(buf);
5459		return -EFAULT;
5460	}
5461
5462	set_task_reclaim_state(current, &sc.reclaim_state);
5463	flags = memalloc_noreclaim_save();
5464	blk_start_plug(&plug);
5465	if (!set_mm_walk(NULL, true)) {
5466		err = -ENOMEM;
5467		goto done;
5468	}
5469
5470	next = buf;
5471	next[len] = '\0';
5472
5473	while ((cur = strsep(&next, ",;\n"))) {
5474		int n;
5475		int end;
5476		char cmd;
5477		unsigned int memcg_id;
5478		unsigned int nid;
5479		unsigned long seq;
5480		unsigned int swappiness = -1;
5481		unsigned long opt = -1;
5482
5483		cur = skip_spaces(cur);
5484		if (!*cur)
5485			continue;
5486
5487		n = sscanf(cur, "%c %u %u %lu %n %u %n %lu %n", &cmd, &memcg_id, &nid,
5488			   &seq, &end, &swappiness, &end, &opt, &end);
5489		if (n < 4 || cur[end]) {
5490			err = -EINVAL;
5491			break;
5492		}
5493
5494		err = run_cmd(cmd, memcg_id, nid, seq, &sc, swappiness, opt);
5495		if (err)
5496			break;
5497	}
5498done:
5499	clear_mm_walk();
5500	blk_finish_plug(&plug);
5501	memalloc_noreclaim_restore(flags);
5502	set_task_reclaim_state(current, NULL);
5503
5504	kvfree(buf);
5505
5506	return err ? : len;
5507}
5508
5509static int lru_gen_seq_open(struct inode *inode, struct file *file)
5510{
5511	return seq_open(file, &lru_gen_seq_ops);
5512}
5513
5514static const struct file_operations lru_gen_rw_fops = {
5515	.open = lru_gen_seq_open,
5516	.read = seq_read,
5517	.write = lru_gen_seq_write,
5518	.llseek = seq_lseek,
5519	.release = seq_release,
5520};
5521
5522static const struct file_operations lru_gen_ro_fops = {
5523	.open = lru_gen_seq_open,
5524	.read = seq_read,
5525	.llseek = seq_lseek,
5526	.release = seq_release,
5527};
5528
5529/******************************************************************************
5530 *                          initialization
5531 ******************************************************************************/
5532
5533void lru_gen_init_pgdat(struct pglist_data *pgdat)
5534{
5535	int i, j;
5536
5537	spin_lock_init(&pgdat->memcg_lru.lock);
5538
5539	for (i = 0; i < MEMCG_NR_GENS; i++) {
5540		for (j = 0; j < MEMCG_NR_BINS; j++)
5541			INIT_HLIST_NULLS_HEAD(&pgdat->memcg_lru.fifo[i][j], i);
5542	}
5543}
5544
5545void lru_gen_init_lruvec(struct lruvec *lruvec)
5546{
5547	int i;
5548	int gen, type, zone;
5549	struct lru_gen_folio *lrugen = &lruvec->lrugen;
5550	struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
5551
5552	lrugen->max_seq = MIN_NR_GENS + 1;
5553	lrugen->enabled = lru_gen_enabled();
5554
5555	for (i = 0; i <= MIN_NR_GENS + 1; i++)
5556		lrugen->timestamps[i] = jiffies;
5557
5558	for_each_gen_type_zone(gen, type, zone)
5559		INIT_LIST_HEAD(&lrugen->folios[gen][type][zone]);
5560
5561	if (mm_state)
5562		mm_state->seq = MIN_NR_GENS;
5563}
5564
5565#ifdef CONFIG_MEMCG
5566
5567void lru_gen_init_memcg(struct mem_cgroup *memcg)
5568{
5569	struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
5570
5571	if (!mm_list)
5572		return;
5573
5574	INIT_LIST_HEAD(&mm_list->fifo);
5575	spin_lock_init(&mm_list->lock);
5576}
5577
5578void lru_gen_exit_memcg(struct mem_cgroup *memcg)
5579{
5580	int i;
5581	int nid;
5582	struct lru_gen_mm_list *mm_list = get_mm_list(memcg);
5583
5584	VM_WARN_ON_ONCE(mm_list && !list_empty(&mm_list->fifo));
5585
5586	for_each_node(nid) {
5587		struct lruvec *lruvec = get_lruvec(memcg, nid);
5588		struct lru_gen_mm_state *mm_state = get_mm_state(lruvec);
5589
5590		VM_WARN_ON_ONCE(memchr_inv(lruvec->lrugen.nr_pages, 0,
5591					   sizeof(lruvec->lrugen.nr_pages)));
5592
5593		lruvec->lrugen.list.next = LIST_POISON1;
5594
5595		if (!mm_state)
5596			continue;
5597
5598		for (i = 0; i < NR_BLOOM_FILTERS; i++) {
5599			bitmap_free(mm_state->filters[i]);
5600			mm_state->filters[i] = NULL;
5601		}
5602	}
5603}
5604
5605#endif /* CONFIG_MEMCG */
5606
5607static int __init init_lru_gen(void)
5608{
5609	BUILD_BUG_ON(MIN_NR_GENS + 1 >= MAX_NR_GENS);
5610	BUILD_BUG_ON(BIT(LRU_GEN_WIDTH) <= MAX_NR_GENS);
5611
5612	if (sysfs_create_group(mm_kobj, &lru_gen_attr_group))
5613		pr_err("lru_gen: failed to create sysfs group\n");
5614
5615	debugfs_create_file("lru_gen", 0644, NULL, NULL, &lru_gen_rw_fops);
5616	debugfs_create_file("lru_gen_full", 0444, NULL, NULL, &lru_gen_ro_fops);
5617
5618	return 0;
5619};
5620late_initcall(init_lru_gen);
5621
5622#else /* !CONFIG_LRU_GEN */
5623
5624static void lru_gen_age_node(struct pglist_data *pgdat, struct scan_control *sc)
5625{
5626	BUILD_BUG();
5627}
5628
5629static void lru_gen_shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
5630{
5631	BUILD_BUG();
5632}
5633
5634static void lru_gen_shrink_node(struct pglist_data *pgdat, struct scan_control *sc)
5635{
5636	BUILD_BUG();
5637}
5638
5639#endif /* CONFIG_LRU_GEN */
5640
5641static void shrink_lruvec(struct lruvec *lruvec, struct scan_control *sc)
5642{
5643	unsigned long nr[NR_LRU_LISTS];
5644	unsigned long targets[NR_LRU_LISTS];
5645	unsigned long nr_to_scan;
5646	enum lru_list lru;
5647	unsigned long nr_reclaimed = 0;
5648	unsigned long nr_to_reclaim = sc->nr_to_reclaim;
5649	bool proportional_reclaim;
5650	struct blk_plug plug;
5651
5652	if (lru_gen_enabled() && !root_reclaim(sc)) {
5653		lru_gen_shrink_lruvec(lruvec, sc);
5654		return;
5655	}
5656
5657	get_scan_count(lruvec, sc, nr);
5658
5659	/* Record the original scan target for proportional adjustments later */
5660	memcpy(targets, nr, sizeof(nr));
5661
5662	/*
5663	 * Global reclaiming within direct reclaim at DEF_PRIORITY is a normal
5664	 * event that can occur when there is little memory pressure e.g.
5665	 * multiple streaming readers/writers. Hence, we do not abort scanning
5666	 * when the requested number of pages are reclaimed when scanning at
5667	 * DEF_PRIORITY on the assumption that the fact we are direct
5668	 * reclaiming implies that kswapd is not keeping up and it is best to
5669	 * do a batch of work at once. For memcg reclaim one check is made to
5670	 * abort proportional reclaim if either the file or anon lru has already
5671	 * dropped to zero at the first pass.
5672	 */
5673	proportional_reclaim = (!cgroup_reclaim(sc) && !current_is_kswapd() &&
5674				sc->priority == DEF_PRIORITY);
5675
5676	blk_start_plug(&plug);
5677	while (nr[LRU_INACTIVE_ANON] || nr[LRU_ACTIVE_FILE] ||
5678					nr[LRU_INACTIVE_FILE]) {
5679		unsigned long nr_anon, nr_file, percentage;
5680		unsigned long nr_scanned;
5681
5682		for_each_evictable_lru(lru) {
5683			if (nr[lru]) {
5684				nr_to_scan = min(nr[lru], SWAP_CLUSTER_MAX);
5685				nr[lru] -= nr_to_scan;
5686
5687				nr_reclaimed += shrink_list(lru, nr_to_scan,
5688							    lruvec, sc);
5689			}
5690		}
5691
5692		cond_resched();
5693
5694		if (nr_reclaimed < nr_to_reclaim || proportional_reclaim)
5695			continue;
5696
5697		/*
5698		 * For kswapd and memcg, reclaim at least the number of pages
5699		 * requested. Ensure that the anon and file LRUs are scanned
5700		 * proportionally what was requested by get_scan_count(). We
5701		 * stop reclaiming one LRU and reduce the amount scanning
5702		 * proportional to the original scan target.
5703		 */
5704		nr_file = nr[LRU_INACTIVE_FILE] + nr[LRU_ACTIVE_FILE];
5705		nr_anon = nr[LRU_INACTIVE_ANON] + nr[LRU_ACTIVE_ANON];
5706
5707		/*
5708		 * It's just vindictive to attack the larger once the smaller
5709		 * has gone to zero.  And given the way we stop scanning the
5710		 * smaller below, this makes sure that we only make one nudge
5711		 * towards proportionality once we've got nr_to_reclaim.
5712		 */
5713		if (!nr_file || !nr_anon)
5714			break;
5715
5716		if (nr_file > nr_anon) {
5717			unsigned long scan_target = targets[LRU_INACTIVE_ANON] +
5718						targets[LRU_ACTIVE_ANON] + 1;
5719			lru = LRU_BASE;
5720			percentage = nr_anon * 100 / scan_target;
5721		} else {
5722			unsigned long scan_target = targets[LRU_INACTIVE_FILE] +
5723						targets[LRU_ACTIVE_FILE] + 1;
5724			lru = LRU_FILE;
5725			percentage = nr_file * 100 / scan_target;
5726		}
5727
5728		/* Stop scanning the smaller of the LRU */
5729		nr[lru] = 0;
5730		nr[lru + LRU_ACTIVE] = 0;
5731
5732		/*
5733		 * Recalculate the other LRU scan count based on its original
5734		 * scan target and the percentage scanning already complete
5735		 */
5736		lru = (lru == LRU_FILE) ? LRU_BASE : LRU_FILE;
5737		nr_scanned = targets[lru] - nr[lru];
5738		nr[lru] = targets[lru] * (100 - percentage) / 100;
5739		nr[lru] -= min(nr[lru], nr_scanned);
5740
5741		lru += LRU_ACTIVE;
5742		nr_scanned = targets[lru] - nr[lru];
5743		nr[lru] = targets[lru] * (100 - percentage) / 100;
5744		nr[lru] -= min(nr[lru], nr_scanned);
5745	}
5746	blk_finish_plug(&plug);
5747	sc->nr_reclaimed += nr_reclaimed;
5748
5749	/*
5750	 * Even if we did not try to evict anon pages at all, we want to
5751	 * rebalance the anon lru active/inactive ratio.
5752	 */
5753	if (can_age_anon_pages(lruvec_pgdat(lruvec), sc) &&
5754	    inactive_is_low(lruvec, LRU_INACTIVE_ANON))
5755		shrink_active_list(SWAP_CLUSTER_MAX, lruvec,
5756				   sc, LRU_ACTIVE_ANON);
5757}
5758
5759/* Use reclaim/compaction for costly allocs or under memory pressure */
5760static bool in_reclaim_compaction(struct scan_control *sc)
5761{
5762	if (gfp_compaction_allowed(sc->gfp_mask) && sc->order &&
5763			(sc->order > PAGE_ALLOC_COSTLY_ORDER ||
5764			 sc->priority < DEF_PRIORITY - 2))
5765		return true;
5766
5767	return false;
5768}
5769
5770/*
5771 * Reclaim/compaction is used for high-order allocation requests. It reclaims
5772 * order-0 pages before compacting the zone. should_continue_reclaim() returns
5773 * true if more pages should be reclaimed such that when the page allocator
5774 * calls try_to_compact_pages() that it will have enough free pages to succeed.
5775 * It will give up earlier than that if there is difficulty reclaiming pages.
5776 */
5777static inline bool should_continue_reclaim(struct pglist_data *pgdat,
5778					unsigned long nr_reclaimed,
5779					struct scan_control *sc)
5780{
5781	unsigned long pages_for_compaction;
5782	unsigned long inactive_lru_pages;
5783	int z;
5784
5785	/* If not in reclaim/compaction mode, stop */
5786	if (!in_reclaim_compaction(sc))
5787		return false;
5788
5789	/*
5790	 * Stop if we failed to reclaim any pages from the last SWAP_CLUSTER_MAX
5791	 * number of pages that were scanned. This will return to the caller
5792	 * with the risk reclaim/compaction and the resulting allocation attempt
5793	 * fails. In the past we have tried harder for __GFP_RETRY_MAYFAIL
5794	 * allocations through requiring that the full LRU list has been scanned
5795	 * first, by assuming that zero delta of sc->nr_scanned means full LRU
5796	 * scan, but that approximation was wrong, and there were corner cases
5797	 * where always a non-zero amount of pages were scanned.
5798	 */
5799	if (!nr_reclaimed)
5800		return false;
5801
5802	/* If compaction would go ahead or the allocation would succeed, stop */
5803	for (z = 0; z <= sc->reclaim_idx; z++) {
5804		struct zone *zone = &pgdat->node_zones[z];
5805		if (!managed_zone(zone))
5806			continue;
5807
5808		/* Allocation can already succeed, nothing to do */
5809		if (zone_watermark_ok(zone, sc->order, min_wmark_pages(zone),
5810				      sc->reclaim_idx, 0))
5811			return false;
5812
5813		if (compaction_suitable(zone, sc->order, sc->reclaim_idx))
5814			return false;
5815	}
5816
5817	/*
5818	 * If we have not reclaimed enough pages for compaction and the
5819	 * inactive lists are large enough, continue reclaiming
5820	 */
5821	pages_for_compaction = compact_gap(sc->order);
5822	inactive_lru_pages = node_page_state(pgdat, NR_INACTIVE_FILE);
5823	if (can_reclaim_anon_pages(NULL, pgdat->node_id, sc))
5824		inactive_lru_pages += node_page_state(pgdat, NR_INACTIVE_ANON);
5825
5826	return inactive_lru_pages > pages_for_compaction;
5827}
5828
5829static void shrink_node_memcgs(pg_data_t *pgdat, struct scan_control *sc)
5830{
5831	struct mem_cgroup *target_memcg = sc->target_mem_cgroup;
5832	struct mem_cgroup *memcg;
5833
5834	memcg = mem_cgroup_iter(target_memcg, NULL, NULL);
5835	do {
5836		struct lruvec *lruvec = mem_cgroup_lruvec(memcg, pgdat);
5837		unsigned long reclaimed;
5838		unsigned long scanned;
5839
5840		/*
5841		 * This loop can become CPU-bound when target memcgs
5842		 * aren't eligible for reclaim - either because they
5843		 * don't have any reclaimable pages, or because their
5844		 * memory is explicitly protected. Avoid soft lockups.
5845		 */
5846		cond_resched();
5847
5848		mem_cgroup_calculate_protection(target_memcg, memcg);
5849
5850		if (mem_cgroup_below_min(target_memcg, memcg)) {
5851			/*
5852			 * Hard protection.
5853			 * If there is no reclaimable memory, OOM.
5854			 */
5855			continue;
5856		} else if (mem_cgroup_below_low(target_memcg, memcg)) {
5857			/*
5858			 * Soft protection.
5859			 * Respect the protection only as long as
5860			 * there is an unprotected supply
5861			 * of reclaimable memory from other cgroups.
5862			 */
5863			if (!sc->memcg_low_reclaim) {
5864				sc->memcg_low_skipped = 1;
5865				continue;
5866			}
5867			memcg_memory_event(memcg, MEMCG_LOW);
5868		}
5869
5870		reclaimed = sc->nr_reclaimed;
5871		scanned = sc->nr_scanned;
5872
5873		shrink_lruvec(lruvec, sc);
5874
5875		shrink_slab(sc->gfp_mask, pgdat->node_id, memcg,
5876			    sc->priority);
5877
5878		/* Record the group's reclaim efficiency */
5879		if (!sc->proactive)
5880			vmpressure(sc->gfp_mask, memcg, false,
5881				   sc->nr_scanned - scanned,
5882				   sc->nr_reclaimed - reclaimed);
5883
5884	} while ((memcg = mem_cgroup_iter(target_memcg, memcg, NULL)));
5885}
5886
5887static void shrink_node(pg_data_t *pgdat, struct scan_control *sc)
5888{
5889	unsigned long nr_reclaimed, nr_scanned, nr_node_reclaimed;
5890	struct lruvec *target_lruvec;
5891	bool reclaimable = false;
5892
5893	if (lru_gen_enabled() && root_reclaim(sc)) {
5894		lru_gen_shrink_node(pgdat, sc);
5895		return;
5896	}
5897
5898	target_lruvec = mem_cgroup_lruvec(sc->target_mem_cgroup, pgdat);
5899
5900again:
5901	memset(&sc->nr, 0, sizeof(sc->nr));
5902
5903	nr_reclaimed = sc->nr_reclaimed;
5904	nr_scanned = sc->nr_scanned;
5905
5906	prepare_scan_control(pgdat, sc);
5907
5908	shrink_node_memcgs(pgdat, sc);
5909
5910	flush_reclaim_state(sc);
5911
5912	nr_node_reclaimed = sc->nr_reclaimed - nr_reclaimed;
5913
5914	/* Record the subtree's reclaim efficiency */
5915	if (!sc->proactive)
5916		vmpressure(sc->gfp_mask, sc->target_mem_cgroup, true,
5917			   sc->nr_scanned - nr_scanned, nr_node_reclaimed);
5918
5919	if (nr_node_reclaimed)
5920		reclaimable = true;
5921
5922	if (current_is_kswapd()) {
5923		/*
5924		 * If reclaim is isolating dirty pages under writeback,
5925		 * it implies that the long-lived page allocation rate
5926		 * is exceeding the page laundering rate. Either the
5927		 * global limits are not being effective at throttling
5928		 * processes due to the page distribution throughout
5929		 * zones or there is heavy usage of a slow backing
5930		 * device. The only option is to throttle from reclaim
5931		 * context which is not ideal as there is no guarantee
5932		 * the dirtying process is throttled in the same way
5933		 * balance_dirty_pages() manages.
5934		 *
5935		 * Once a node is flagged PGDAT_WRITEBACK, kswapd will
5936		 * count the number of pages under pages flagged for
5937		 * immediate reclaim and stall if any are encountered
5938		 * in the nr_immediate check below.
5939		 */
5940		if (sc->nr.writeback && sc->nr.writeback == sc->nr.taken)
5941			set_bit(PGDAT_WRITEBACK, &pgdat->flags);
5942
5943		/* Allow kswapd to start writing pages during reclaim.*/
5944		if (sc->nr.unqueued_dirty == sc->nr.file_taken)
5945			set_bit(PGDAT_DIRTY, &pgdat->flags);
5946
5947		/*
5948		 * If kswapd scans pages marked for immediate
5949		 * reclaim and under writeback (nr_immediate), it
5950		 * implies that pages are cycling through the LRU
5951		 * faster than they are written so forcibly stall
5952		 * until some pages complete writeback.
5953		 */
5954		if (sc->nr.immediate)
5955			reclaim_throttle(pgdat, VMSCAN_THROTTLE_WRITEBACK);
5956	}
5957
5958	/*
5959	 * Tag a node/memcg as congested if all the dirty pages were marked
5960	 * for writeback and immediate reclaim (counted in nr.congested).
5961	 *
5962	 * Legacy memcg will stall in page writeback so avoid forcibly
5963	 * stalling in reclaim_throttle().
5964	 */
5965	if (sc->nr.dirty && sc->nr.dirty == sc->nr.congested) {
5966		if (cgroup_reclaim(sc) && writeback_throttling_sane(sc))
5967			set_bit(LRUVEC_CGROUP_CONGESTED, &target_lruvec->flags);
5968
5969		if (current_is_kswapd())
5970			set_bit(LRUVEC_NODE_CONGESTED, &target_lruvec->flags);
5971	}
5972
5973	/*
5974	 * Stall direct reclaim for IO completions if the lruvec is
5975	 * node is congested. Allow kswapd to continue until it
5976	 * starts encountering unqueued dirty pages or cycling through
5977	 * the LRU too quickly.
5978	 */
5979	if (!current_is_kswapd() && current_may_throttle() &&
5980	    !sc->hibernation_mode &&
5981	    (test_bit(LRUVEC_CGROUP_CONGESTED, &target_lruvec->flags) ||
5982	     test_bit(LRUVEC_NODE_CONGESTED, &target_lruvec->flags)))
5983		reclaim_throttle(pgdat, VMSCAN_THROTTLE_CONGESTED);
5984
5985	if (should_continue_reclaim(pgdat, nr_node_reclaimed, sc))
5986		goto again;
5987
5988	/*
5989	 * Kswapd gives up on balancing particular nodes after too
5990	 * many failures to reclaim anything from them and goes to
5991	 * sleep. On reclaim progress, reset the failure counter. A
5992	 * successful direct reclaim run will revive a dormant kswapd.
5993	 */
5994	if (reclaimable)
5995		pgdat->kswapd_failures = 0;
5996	else if (sc->cache_trim_mode)
5997		sc->cache_trim_mode_failed = 1;
5998}
5999
6000/*
6001 * Returns true if compaction should go ahead for a costly-order request, or
6002 * the allocation would already succeed without compaction. Return false if we
6003 * should reclaim first.
6004 */
6005static inline bool compaction_ready(struct zone *zone, struct scan_control *sc)
6006{
6007	unsigned long watermark;
6008
6009	if (!gfp_compaction_allowed(sc->gfp_mask))
6010		return false;
6011
6012	/* Allocation can already succeed, nothing to do */
6013	if (zone_watermark_ok(zone, sc->order, min_wmark_pages(zone),
6014			      sc->reclaim_idx, 0))
6015		return true;
6016
6017	/* Compaction cannot yet proceed. Do reclaim. */
6018	if (!compaction_suitable(zone, sc->order, sc->reclaim_idx))
6019		return false;
6020
6021	/*
6022	 * Compaction is already possible, but it takes time to run and there
6023	 * are potentially other callers using the pages just freed. So proceed
6024	 * with reclaim to make a buffer of free pages available to give
6025	 * compaction a reasonable chance of completing and allocating the page.
6026	 * Note that we won't actually reclaim the whole buffer in one attempt
6027	 * as the target watermark in should_continue_reclaim() is lower. But if
6028	 * we are already above the high+gap watermark, don't reclaim at all.
6029	 */
6030	watermark = high_wmark_pages(zone) + compact_gap(sc->order);
6031
6032	return zone_watermark_ok_safe(zone, 0, watermark, sc->reclaim_idx);
6033}
6034
6035static void consider_reclaim_throttle(pg_data_t *pgdat, struct scan_control *sc)
6036{
6037	/*
6038	 * If reclaim is making progress greater than 12% efficiency then
6039	 * wake all the NOPROGRESS throttled tasks.
6040	 */
6041	if (sc->nr_reclaimed > (sc->nr_scanned >> 3)) {
6042		wait_queue_head_t *wqh;
6043
6044		wqh = &pgdat->reclaim_wait[VMSCAN_THROTTLE_NOPROGRESS];
6045		if (waitqueue_active(wqh))
6046			wake_up(wqh);
6047
6048		return;
6049	}
6050
6051	/*
6052	 * Do not throttle kswapd or cgroup reclaim on NOPROGRESS as it will
6053	 * throttle on VMSCAN_THROTTLE_WRITEBACK if there are too many pages
6054	 * under writeback and marked for immediate reclaim at the tail of the
6055	 * LRU.
6056	 */
6057	if (current_is_kswapd() || cgroup_reclaim(sc))
6058		return;
6059
6060	/* Throttle if making no progress at high prioities. */
6061	if (sc->priority == 1 && !sc->nr_reclaimed)
6062		reclaim_throttle(pgdat, VMSCAN_THROTTLE_NOPROGRESS);
6063}
6064
6065/*
6066 * This is the direct reclaim path, for page-allocating processes.  We only
6067 * try to reclaim pages from zones which will satisfy the caller's allocation
6068 * request.
6069 *
6070 * If a zone is deemed to be full of pinned pages then just give it a light
6071 * scan then give up on it.
6072 */
6073static void shrink_zones(struct zonelist *zonelist, struct scan_control *sc)
6074{
6075	struct zoneref *z;
6076	struct zone *zone;
6077	unsigned long nr_soft_reclaimed;
6078	unsigned long nr_soft_scanned;
6079	gfp_t orig_mask;
6080	pg_data_t *last_pgdat = NULL;
6081	pg_data_t *first_pgdat = NULL;
6082
6083	/*
6084	 * If the number of buffer_heads in the machine exceeds the maximum
6085	 * allowed level, force direct reclaim to scan the highmem zone as
6086	 * highmem pages could be pinning lowmem pages storing buffer_heads
6087	 */
6088	orig_mask = sc->gfp_mask;
6089	if (buffer_heads_over_limit) {
6090		sc->gfp_mask |= __GFP_HIGHMEM;
6091		sc->reclaim_idx = gfp_zone(sc->gfp_mask);
6092	}
6093
6094	for_each_zone_zonelist_nodemask(zone, z, zonelist,
6095					sc->reclaim_idx, sc->nodemask) {
6096		/*
6097		 * Take care memory controller reclaiming has small influence
6098		 * to global LRU.
6099		 */
6100		if (!cgroup_reclaim(sc)) {
6101			if (!cpuset_zone_allowed(zone,
6102						 GFP_KERNEL | __GFP_HARDWALL))
6103				continue;
6104
6105			/*
6106			 * If we already have plenty of memory free for
6107			 * compaction in this zone, don't free any more.
6108			 * Even though compaction is invoked for any
6109			 * non-zero order, only frequent costly order
6110			 * reclamation is disruptive enough to become a
6111			 * noticeable problem, like transparent huge
6112			 * page allocations.
6113			 */
6114			if (IS_ENABLED(CONFIG_COMPACTION) &&
6115			    sc->order > PAGE_ALLOC_COSTLY_ORDER &&
6116			    compaction_ready(zone, sc)) {
6117				sc->compaction_ready = true;
6118				continue;
6119			}
6120
6121			/*
6122			 * Shrink each node in the zonelist once. If the
6123			 * zonelist is ordered by zone (not the default) then a
6124			 * node may be shrunk multiple times but in that case
6125			 * the user prefers lower zones being preserved.
6126			 */
6127			if (zone->zone_pgdat == last_pgdat)
6128				continue;
6129
6130			/*
6131			 * This steals pages from memory cgroups over softlimit
6132			 * and returns the number of reclaimed pages and
6133			 * scanned pages. This works for global memory pressure
6134			 * and balancing, not for a memcg's limit.
6135			 */
6136			nr_soft_scanned = 0;
6137			nr_soft_reclaimed = mem_cgroup_soft_limit_reclaim(zone->zone_pgdat,
6138						sc->order, sc->gfp_mask,
6139						&nr_soft_scanned);
6140			sc->nr_reclaimed += nr_soft_reclaimed;
6141			sc->nr_scanned += nr_soft_scanned;
6142			/* need some check for avoid more shrink_zone() */
6143		}
6144
6145		if (!first_pgdat)
6146			first_pgdat = zone->zone_pgdat;
6147
6148		/* See comment about same check for global reclaim above */
6149		if (zone->zone_pgdat == last_pgdat)
6150			continue;
6151		last_pgdat = zone->zone_pgdat;
6152		shrink_node(zone->zone_pgdat, sc);
6153	}
6154
6155	if (first_pgdat)
6156		consider_reclaim_throttle(first_pgdat, sc);
6157
6158	/*
6159	 * Restore to original mask to avoid the impact on the caller if we
6160	 * promoted it to __GFP_HIGHMEM.
6161	 */
6162	sc->gfp_mask = orig_mask;
6163}
6164
6165static void snapshot_refaults(struct mem_cgroup *target_memcg, pg_data_t *pgdat)
6166{
6167	struct lruvec *target_lruvec;
6168	unsigned long refaults;
6169
6170	if (lru_gen_enabled())
6171		return;
6172
6173	target_lruvec = mem_cgroup_lruvec(target_memcg, pgdat);
6174	refaults = lruvec_page_state(target_lruvec, WORKINGSET_ACTIVATE_ANON);
6175	target_lruvec->refaults[WORKINGSET_ANON] = refaults;
6176	refaults = lruvec_page_state(target_lruvec, WORKINGSET_ACTIVATE_FILE);
6177	target_lruvec->refaults[WORKINGSET_FILE] = refaults;
6178}
6179
6180/*
6181 * This is the main entry point to direct page reclaim.
6182 *
6183 * If a full scan of the inactive list fails to free enough memory then we
6184 * are "out of memory" and something needs to be killed.
6185 *
6186 * If the caller is !__GFP_FS then the probability of a failure is reasonably
6187 * high - the zone may be full of dirty or under-writeback pages, which this
6188 * caller can't do much about.  We kick the writeback threads and take explicit
6189 * naps in the hope that some of these pages can be written.  But if the
6190 * allocating task holds filesystem locks which prevent writeout this might not
6191 * work, and the allocation attempt will fail.
6192 *
6193 * returns:	0, if no pages reclaimed
6194 * 		else, the number of pages reclaimed
6195 */
6196static unsigned long do_try_to_free_pages(struct zonelist *zonelist,
6197					  struct scan_control *sc)
6198{
6199	int initial_priority = sc->priority;
6200	pg_data_t *last_pgdat;
6201	struct zoneref *z;
6202	struct zone *zone;
6203retry:
6204	delayacct_freepages_start();
6205
6206	if (!cgroup_reclaim(sc))
6207		__count_zid_vm_events(ALLOCSTALL, sc->reclaim_idx, 1);
6208
6209	do {
6210		if (!sc->proactive)
6211			vmpressure_prio(sc->gfp_mask, sc->target_mem_cgroup,
6212					sc->priority);
6213		sc->nr_scanned = 0;
6214		shrink_zones(zonelist, sc);
6215
6216		if (sc->nr_reclaimed >= sc->nr_to_reclaim)
6217			break;
6218
6219		if (sc->compaction_ready)
6220			break;
6221
6222		/*
6223		 * If we're getting trouble reclaiming, start doing
6224		 * writepage even in laptop mode.
6225		 */
6226		if (sc->priority < DEF_PRIORITY - 2)
6227			sc->may_writepage = 1;
6228	} while (--sc->priority >= 0);
6229
6230	last_pgdat = NULL;
6231	for_each_zone_zonelist_nodemask(zone, z, zonelist, sc->reclaim_idx,
6232					sc->nodemask) {
6233		if (zone->zone_pgdat == last_pgdat)
6234			continue;
6235		last_pgdat = zone->zone_pgdat;
6236
6237		snapshot_refaults(sc->target_mem_cgroup, zone->zone_pgdat);
6238
6239		if (cgroup_reclaim(sc)) {
6240			struct lruvec *lruvec;
6241
6242			lruvec = mem_cgroup_lruvec(sc->target_mem_cgroup,
6243						   zone->zone_pgdat);
6244			clear_bit(LRUVEC_CGROUP_CONGESTED, &lruvec->flags);
6245		}
6246	}
6247
6248	delayacct_freepages_end();
6249
6250	if (sc->nr_reclaimed)
6251		return sc->nr_reclaimed;
6252
6253	/* Aborted reclaim to try compaction? don't OOM, then */
6254	if (sc->compaction_ready)
6255		return 1;
6256
6257	/*
6258	 * We make inactive:active ratio decisions based on the node's
6259	 * composition of memory, but a restrictive reclaim_idx or a
6260	 * memory.low cgroup setting can exempt large amounts of
6261	 * memory from reclaim. Neither of which are very common, so
6262	 * instead of doing costly eligibility calculations of the
6263	 * entire cgroup subtree up front, we assume the estimates are
6264	 * good, and retry with forcible deactivation if that fails.
6265	 */
6266	if (sc->skipped_deactivate) {
6267		sc->priority = initial_priority;
6268		sc->force_deactivate = 1;
6269		sc->skipped_deactivate = 0;
6270		goto retry;
6271	}
6272
6273	/* Untapped cgroup reserves?  Don't OOM, retry. */
6274	if (sc->memcg_low_skipped) {
6275		sc->priority = initial_priority;
6276		sc->force_deactivate = 0;
6277		sc->memcg_low_reclaim = 1;
6278		sc->memcg_low_skipped = 0;
6279		goto retry;
6280	}
6281
6282	return 0;
6283}
6284
6285static bool allow_direct_reclaim(pg_data_t *pgdat)
6286{
6287	struct zone *zone;
6288	unsigned long pfmemalloc_reserve = 0;
6289	unsigned long free_pages = 0;
6290	int i;
6291	bool wmark_ok;
6292
6293	if (pgdat->kswapd_failures >= MAX_RECLAIM_RETRIES)
6294		return true;
6295
6296	for (i = 0; i <= ZONE_NORMAL; i++) {
6297		zone = &pgdat->node_zones[i];
6298		if (!managed_zone(zone))
6299			continue;
6300
6301		if (!zone_reclaimable_pages(zone))
6302			continue;
6303
6304		pfmemalloc_reserve += min_wmark_pages(zone);
6305		free_pages += zone_page_state_snapshot(zone, NR_FREE_PAGES);
6306	}
6307
6308	/* If there are no reserves (unexpected config) then do not throttle */
6309	if (!pfmemalloc_reserve)
6310		return true;
6311
6312	wmark_ok = free_pages > pfmemalloc_reserve / 2;
6313
6314	/* kswapd must be awake if processes are being throttled */
6315	if (!wmark_ok && waitqueue_active(&pgdat->kswapd_wait)) {
6316		if (READ_ONCE(pgdat->kswapd_highest_zoneidx) > ZONE_NORMAL)
6317			WRITE_ONCE(pgdat->kswapd_highest_zoneidx, ZONE_NORMAL);
6318
6319		wake_up_interruptible(&pgdat->kswapd_wait);
6320	}
6321
6322	return wmark_ok;
6323}
6324
6325/*
6326 * Throttle direct reclaimers if backing storage is backed by the network
6327 * and the PFMEMALLOC reserve for the preferred node is getting dangerously
6328 * depleted. kswapd will continue to make progress and wake the processes
6329 * when the low watermark is reached.
6330 *
6331 * Returns true if a fatal signal was delivered during throttling. If this
6332 * happens, the page allocator should not consider triggering the OOM killer.
6333 */
6334static bool throttle_direct_reclaim(gfp_t gfp_mask, struct zonelist *zonelist,
6335					nodemask_t *nodemask)
6336{
6337	struct zoneref *z;
6338	struct zone *zone;
6339	pg_data_t *pgdat = NULL;
6340
6341	/*
6342	 * Kernel threads should not be throttled as they may be indirectly
6343	 * responsible for cleaning pages necessary for reclaim to make forward
6344	 * progress. kjournald for example may enter direct reclaim while
6345	 * committing a transaction where throttling it could forcing other
6346	 * processes to block on log_wait_commit().
6347	 */
6348	if (current->flags & PF_KTHREAD)
6349		goto out;
6350
6351	/*
6352	 * If a fatal signal is pending, this process should not throttle.
6353	 * It should return quickly so it can exit and free its memory
6354	 */
6355	if (fatal_signal_pending(current))
6356		goto out;
6357
6358	/*
6359	 * Check if the pfmemalloc reserves are ok by finding the first node
6360	 * with a usable ZONE_NORMAL or lower zone. The expectation is that
6361	 * GFP_KERNEL will be required for allocating network buffers when
6362	 * swapping over the network so ZONE_HIGHMEM is unusable.
6363	 *
6364	 * Throttling is based on the first usable node and throttled processes
6365	 * wait on a queue until kswapd makes progress and wakes them. There
6366	 * is an affinity then between processes waking up and where reclaim
6367	 * progress has been made assuming the process wakes on the same node.
6368	 * More importantly, processes running on remote nodes will not compete
6369	 * for remote pfmemalloc reserves and processes on different nodes
6370	 * should make reasonable progress.
6371	 */
6372	for_each_zone_zonelist_nodemask(zone, z, zonelist,
6373					gfp_zone(gfp_mask), nodemask) {
6374		if (zone_idx(zone) > ZONE_NORMAL)
6375			continue;
6376
6377		/* Throttle based on the first usable node */
6378		pgdat = zone->zone_pgdat;
6379		if (allow_direct_reclaim(pgdat))
6380			goto out;
6381		break;
6382	}
6383
6384	/* If no zone was usable by the allocation flags then do not throttle */
6385	if (!pgdat)
6386		goto out;
6387
6388	/* Account for the throttling */
6389	count_vm_event(PGSCAN_DIRECT_THROTTLE);
6390
6391	/*
6392	 * If the caller cannot enter the filesystem, it's possible that it
6393	 * is due to the caller holding an FS lock or performing a journal
6394	 * transaction in the case of a filesystem like ext[3|4]. In this case,
6395	 * it is not safe to block on pfmemalloc_wait as kswapd could be
6396	 * blocked waiting on the same lock. Instead, throttle for up to a
6397	 * second before continuing.
6398	 */
6399	if (!(gfp_mask & __GFP_FS))
6400		wait_event_interruptible_timeout(pgdat->pfmemalloc_wait,
6401			allow_direct_reclaim(pgdat), HZ);
6402	else
6403		/* Throttle until kswapd wakes the process */
6404		wait_event_killable(zone->zone_pgdat->pfmemalloc_wait,
6405			allow_direct_reclaim(pgdat));
6406
6407	if (fatal_signal_pending(current))
6408		return true;
6409
6410out:
6411	return false;
6412}
6413
6414unsigned long try_to_free_pages(struct zonelist *zonelist, int order,
6415				gfp_t gfp_mask, nodemask_t *nodemask)
6416{
6417	unsigned long nr_reclaimed;
6418	struct scan_control sc = {
6419		.nr_to_reclaim = SWAP_CLUSTER_MAX,
6420		.gfp_mask = current_gfp_context(gfp_mask),
6421		.reclaim_idx = gfp_zone(gfp_mask),
6422		.order = order,
6423		.nodemask = nodemask,
6424		.priority = DEF_PRIORITY,
6425		.may_writepage = !laptop_mode,
6426		.may_unmap = 1,
6427		.may_swap = 1,
6428	};
6429
6430	/*
6431	 * scan_control uses s8 fields for order, priority, and reclaim_idx.
6432	 * Confirm they are large enough for max values.
6433	 */
6434	BUILD_BUG_ON(MAX_PAGE_ORDER >= S8_MAX);
6435	BUILD_BUG_ON(DEF_PRIORITY > S8_MAX);
6436	BUILD_BUG_ON(MAX_NR_ZONES > S8_MAX);
6437
6438	/*
6439	 * Do not enter reclaim if fatal signal was delivered while throttled.
6440	 * 1 is returned so that the page allocator does not OOM kill at this
6441	 * point.
6442	 */
6443	if (throttle_direct_reclaim(sc.gfp_mask, zonelist, nodemask))
6444		return 1;
6445
6446	set_task_reclaim_state(current, &sc.reclaim_state);
6447	trace_mm_vmscan_direct_reclaim_begin(order, sc.gfp_mask);
6448
6449	nr_reclaimed = do_try_to_free_pages(zonelist, &sc);
6450
6451	trace_mm_vmscan_direct_reclaim_end(nr_reclaimed);
6452	set_task_reclaim_state(current, NULL);
6453
6454	return nr_reclaimed;
6455}
6456
6457#ifdef CONFIG_MEMCG
6458
6459/* Only used by soft limit reclaim. Do not reuse for anything else. */
6460unsigned long mem_cgroup_shrink_node(struct mem_cgroup *memcg,
6461						gfp_t gfp_mask, bool noswap,
6462						pg_data_t *pgdat,
6463						unsigned long *nr_scanned)
6464{
6465	struct lruvec *lruvec = mem_cgroup_lruvec(memcg, pgdat);
6466	struct scan_control sc = {
6467		.nr_to_reclaim = SWAP_CLUSTER_MAX,
6468		.target_mem_cgroup = memcg,
6469		.may_writepage = !laptop_mode,
6470		.may_unmap = 1,
6471		.reclaim_idx = MAX_NR_ZONES - 1,
6472		.may_swap = !noswap,
6473	};
6474
6475	WARN_ON_ONCE(!current->reclaim_state);
6476
6477	sc.gfp_mask = (gfp_mask & GFP_RECLAIM_MASK) |
6478			(GFP_HIGHUSER_MOVABLE & ~GFP_RECLAIM_MASK);
6479
6480	trace_mm_vmscan_memcg_softlimit_reclaim_begin(sc.order,
6481						      sc.gfp_mask);
6482
6483	/*
6484	 * NOTE: Although we can get the priority field, using it
6485	 * here is not a good idea, since it limits the pages we can scan.
6486	 * if we don't reclaim here, the shrink_node from balance_pgdat
6487	 * will pick up pages from other mem cgroup's as well. We hack
6488	 * the priority and make it zero.
6489	 */
6490	shrink_lruvec(lruvec, &sc);
6491
6492	trace_mm_vmscan_memcg_softlimit_reclaim_end(sc.nr_reclaimed);
6493
6494	*nr_scanned = sc.nr_scanned;
6495
6496	return sc.nr_reclaimed;
6497}
6498
6499unsigned long try_to_free_mem_cgroup_pages(struct mem_cgroup *memcg,
6500					   unsigned long nr_pages,
6501					   gfp_t gfp_mask,
6502					   unsigned int reclaim_options)
6503{
6504	unsigned long nr_reclaimed;
6505	unsigned int noreclaim_flag;
6506	struct scan_control sc = {
6507		.nr_to_reclaim = max(nr_pages, SWAP_CLUSTER_MAX),
6508		.gfp_mask = (current_gfp_context(gfp_mask) & GFP_RECLAIM_MASK) |
6509				(GFP_HIGHUSER_MOVABLE & ~GFP_RECLAIM_MASK),
6510		.reclaim_idx = MAX_NR_ZONES - 1,
6511		.target_mem_cgroup = memcg,
6512		.priority = DEF_PRIORITY,
6513		.may_writepage = !laptop_mode,
6514		.may_unmap = 1,
6515		.may_swap = !!(reclaim_options & MEMCG_RECLAIM_MAY_SWAP),
6516		.proactive = !!(reclaim_options & MEMCG_RECLAIM_PROACTIVE),
6517	};
6518	/*
6519	 * Traverse the ZONELIST_FALLBACK zonelist of the current node to put
6520	 * equal pressure on all the nodes. This is based on the assumption that
6521	 * the reclaim does not bail out early.
6522	 */
6523	struct zonelist *zonelist = node_zonelist(numa_node_id(), sc.gfp_mask);
6524
6525	set_task_reclaim_state(current, &sc.reclaim_state);
6526	trace_mm_vmscan_memcg_reclaim_begin(0, sc.gfp_mask);
6527	noreclaim_flag = memalloc_noreclaim_save();
6528
6529	nr_reclaimed = do_try_to_free_pages(zonelist, &sc);
6530
6531	memalloc_noreclaim_restore(noreclaim_flag);
6532	trace_mm_vmscan_memcg_reclaim_end(nr_reclaimed);
6533	set_task_reclaim_state(current, NULL);
6534
6535	return nr_reclaimed;
6536}
6537#endif
6538
6539static void kswapd_age_node(struct pglist_data *pgdat, struct scan_control *sc)
6540{
6541	struct mem_cgroup *memcg;
6542	struct lruvec *lruvec;
6543
6544	if (lru_gen_enabled()) {
6545		lru_gen_age_node(pgdat, sc);
6546		return;
6547	}
6548
6549	if (!can_age_anon_pages(pgdat, sc))
6550		return;
6551
6552	lruvec = mem_cgroup_lruvec(NULL, pgdat);
6553	if (!inactive_is_low(lruvec, LRU_INACTIVE_ANON))
6554		return;
6555
6556	memcg = mem_cgroup_iter(NULL, NULL, NULL);
6557	do {
6558		lruvec = mem_cgroup_lruvec(memcg, pgdat);
6559		shrink_active_list(SWAP_CLUSTER_MAX, lruvec,
6560				   sc, LRU_ACTIVE_ANON);
6561		memcg = mem_cgroup_iter(NULL, memcg, NULL);
6562	} while (memcg);
6563}
6564
6565static bool pgdat_watermark_boosted(pg_data_t *pgdat, int highest_zoneidx)
6566{
6567	int i;
6568	struct zone *zone;
6569
6570	/*
6571	 * Check for watermark boosts top-down as the higher zones
6572	 * are more likely to be boosted. Both watermarks and boosts
6573	 * should not be checked at the same time as reclaim would
6574	 * start prematurely when there is no boosting and a lower
6575	 * zone is balanced.
6576	 */
6577	for (i = highest_zoneidx; i >= 0; i--) {
6578		zone = pgdat->node_zones + i;
6579		if (!managed_zone(zone))
6580			continue;
6581
6582		if (zone->watermark_boost)
6583			return true;
6584	}
6585
6586	return false;
6587}
6588
6589/*
6590 * Returns true if there is an eligible zone balanced for the request order
6591 * and highest_zoneidx
6592 */
6593static bool pgdat_balanced(pg_data_t *pgdat, int order, int highest_zoneidx)
6594{
6595	int i;
6596	unsigned long mark = -1;
6597	struct zone *zone;
6598
6599	/*
6600	 * Check watermarks bottom-up as lower zones are more likely to
6601	 * meet watermarks.
6602	 */
6603	for (i = 0; i <= highest_zoneidx; i++) {
6604		zone = pgdat->node_zones + i;
6605
6606		if (!managed_zone(zone))
6607			continue;
6608
6609		if (sysctl_numa_balancing_mode & NUMA_BALANCING_MEMORY_TIERING)
6610			mark = wmark_pages(zone, WMARK_PROMO);
6611		else
6612			mark = high_wmark_pages(zone);
6613		if (zone_watermark_ok_safe(zone, order, mark, highest_zoneidx))
6614			return true;
6615	}
6616
6617	/*
6618	 * If a node has no managed zone within highest_zoneidx, it does not
6619	 * need balancing by definition. This can happen if a zone-restricted
6620	 * allocation tries to wake a remote kswapd.
6621	 */
6622	if (mark == -1)
6623		return true;
6624
6625	return false;
6626}
6627
6628/* Clear pgdat state for congested, dirty or under writeback. */
6629static void clear_pgdat_congested(pg_data_t *pgdat)
6630{
6631	struct lruvec *lruvec = mem_cgroup_lruvec(NULL, pgdat);
6632
6633	clear_bit(LRUVEC_NODE_CONGESTED, &lruvec->flags);
6634	clear_bit(LRUVEC_CGROUP_CONGESTED, &lruvec->flags);
6635	clear_bit(PGDAT_DIRTY, &pgdat->flags);
6636	clear_bit(PGDAT_WRITEBACK, &pgdat->flags);
6637}
6638
6639/*
6640 * Prepare kswapd for sleeping. This verifies that there are no processes
6641 * waiting in throttle_direct_reclaim() and that watermarks have been met.
6642 *
6643 * Returns true if kswapd is ready to sleep
6644 */
6645static bool prepare_kswapd_sleep(pg_data_t *pgdat, int order,
6646				int highest_zoneidx)
6647{
6648	/*
6649	 * The throttled processes are normally woken up in balance_pgdat() as
6650	 * soon as allow_direct_reclaim() is true. But there is a potential
6651	 * race between when kswapd checks the watermarks and a process gets
6652	 * throttled. There is also a potential race if processes get
6653	 * throttled, kswapd wakes, a large process exits thereby balancing the
6654	 * zones, which causes kswapd to exit balance_pgdat() before reaching
6655	 * the wake up checks. If kswapd is going to sleep, no process should
6656	 * be sleeping on pfmemalloc_wait, so wake them now if necessary. If
6657	 * the wake up is premature, processes will wake kswapd and get
6658	 * throttled again. The difference from wake ups in balance_pgdat() is
6659	 * that here we are under prepare_to_wait().
6660	 */
6661	if (waitqueue_active(&pgdat->pfmemalloc_wait))
6662		wake_up_all(&pgdat->pfmemalloc_wait);
6663
6664	/* Hopeless node, leave it to direct reclaim */
6665	if (pgdat->kswapd_failures >= MAX_RECLAIM_RETRIES)
6666		return true;
6667
6668	if (pgdat_balanced(pgdat, order, highest_zoneidx)) {
6669		clear_pgdat_congested(pgdat);
6670		return true;
6671	}
6672
6673	return false;
6674}
6675
6676/*
6677 * kswapd shrinks a node of pages that are at or below the highest usable
6678 * zone that is currently unbalanced.
6679 *
6680 * Returns true if kswapd scanned at least the requested number of pages to
6681 * reclaim or if the lack of progress was due to pages under writeback.
6682 * This is used to determine if the scanning priority needs to be raised.
6683 */
6684static bool kswapd_shrink_node(pg_data_t *pgdat,
6685			       struct scan_control *sc)
6686{
6687	struct zone *zone;
6688	int z;
6689
6690	/* Reclaim a number of pages proportional to the number of zones */
6691	sc->nr_to_reclaim = 0;
6692	for (z = 0; z <= sc->reclaim_idx; z++) {
6693		zone = pgdat->node_zones + z;
6694		if (!managed_zone(zone))
6695			continue;
6696
6697		sc->nr_to_reclaim += max(high_wmark_pages(zone), SWAP_CLUSTER_MAX);
6698	}
6699
6700	/*
6701	 * Historically care was taken to put equal pressure on all zones but
6702	 * now pressure is applied based on node LRU order.
6703	 */
6704	shrink_node(pgdat, sc);
6705
6706	/*
6707	 * Fragmentation may mean that the system cannot be rebalanced for
6708	 * high-order allocations. If twice the allocation size has been
6709	 * reclaimed then recheck watermarks only at order-0 to prevent
6710	 * excessive reclaim. Assume that a process requested a high-order
6711	 * can direct reclaim/compact.
6712	 */
6713	if (sc->order && sc->nr_reclaimed >= compact_gap(sc->order))
6714		sc->order = 0;
6715
6716	return sc->nr_scanned >= sc->nr_to_reclaim;
6717}
6718
6719/* Page allocator PCP high watermark is lowered if reclaim is active. */
6720static inline void
6721update_reclaim_active(pg_data_t *pgdat, int highest_zoneidx, bool active)
6722{
6723	int i;
6724	struct zone *zone;
6725
6726	for (i = 0; i <= highest_zoneidx; i++) {
6727		zone = pgdat->node_zones + i;
6728
6729		if (!managed_zone(zone))
6730			continue;
6731
6732		if (active)
6733			set_bit(ZONE_RECLAIM_ACTIVE, &zone->flags);
6734		else
6735			clear_bit(ZONE_RECLAIM_ACTIVE, &zone->flags);
6736	}
6737}
6738
6739static inline void
6740set_reclaim_active(pg_data_t *pgdat, int highest_zoneidx)
6741{
6742	update_reclaim_active(pgdat, highest_zoneidx, true);
6743}
6744
6745static inline void
6746clear_reclaim_active(pg_data_t *pgdat, int highest_zoneidx)
6747{
6748	update_reclaim_active(pgdat, highest_zoneidx, false);
6749}
6750
6751/*
6752 * For kswapd, balance_pgdat() will reclaim pages across a node from zones
6753 * that are eligible for use by the caller until at least one zone is
6754 * balanced.
6755 *
6756 * Returns the order kswapd finished reclaiming at.
6757 *
6758 * kswapd scans the zones in the highmem->normal->dma direction.  It skips
6759 * zones which have free_pages > high_wmark_pages(zone), but once a zone is
6760 * found to have free_pages <= high_wmark_pages(zone), any page in that zone
6761 * or lower is eligible for reclaim until at least one usable zone is
6762 * balanced.
6763 */
6764static int balance_pgdat(pg_data_t *pgdat, int order, int highest_zoneidx)
6765{
6766	int i;
6767	unsigned long nr_soft_reclaimed;
6768	unsigned long nr_soft_scanned;
6769	unsigned long pflags;
6770	unsigned long nr_boost_reclaim;
6771	unsigned long zone_boosts[MAX_NR_ZONES] = { 0, };
6772	bool boosted;
6773	struct zone *zone;
6774	struct scan_control sc = {
6775		.gfp_mask = GFP_KERNEL,
6776		.order = order,
6777		.may_unmap = 1,
6778	};
6779
6780	set_task_reclaim_state(current, &sc.reclaim_state);
6781	psi_memstall_enter(&pflags);
6782	__fs_reclaim_acquire(_THIS_IP_);
6783
6784	count_vm_event(PAGEOUTRUN);
6785
6786	/*
6787	 * Account for the reclaim boost. Note that the zone boost is left in
6788	 * place so that parallel allocations that are near the watermark will
6789	 * stall or direct reclaim until kswapd is finished.
6790	 */
6791	nr_boost_reclaim = 0;
6792	for (i = 0; i <= highest_zoneidx; i++) {
6793		zone = pgdat->node_zones + i;
6794		if (!managed_zone(zone))
6795			continue;
6796
6797		nr_boost_reclaim += zone->watermark_boost;
6798		zone_boosts[i] = zone->watermark_boost;
6799	}
6800	boosted = nr_boost_reclaim;
6801
6802restart:
6803	set_reclaim_active(pgdat, highest_zoneidx);
6804	sc.priority = DEF_PRIORITY;
6805	do {
6806		unsigned long nr_reclaimed = sc.nr_reclaimed;
6807		bool raise_priority = true;
6808		bool balanced;
6809		bool ret;
6810		bool was_frozen;
6811
6812		sc.reclaim_idx = highest_zoneidx;
6813
6814		/*
6815		 * If the number of buffer_heads exceeds the maximum allowed
6816		 * then consider reclaiming from all zones. This has a dual
6817		 * purpose -- on 64-bit systems it is expected that
6818		 * buffer_heads are stripped during active rotation. On 32-bit
6819		 * systems, highmem pages can pin lowmem memory and shrinking
6820		 * buffers can relieve lowmem pressure. Reclaim may still not
6821		 * go ahead if all eligible zones for the original allocation
6822		 * request are balanced to avoid excessive reclaim from kswapd.
6823		 */
6824		if (buffer_heads_over_limit) {
6825			for (i = MAX_NR_ZONES - 1; i >= 0; i--) {
6826				zone = pgdat->node_zones + i;
6827				if (!managed_zone(zone))
6828					continue;
6829
6830				sc.reclaim_idx = i;
6831				break;
6832			}
6833		}
6834
6835		/*
6836		 * If the pgdat is imbalanced then ignore boosting and preserve
6837		 * the watermarks for a later time and restart. Note that the
6838		 * zone watermarks will be still reset at the end of balancing
6839		 * on the grounds that the normal reclaim should be enough to
6840		 * re-evaluate if boosting is required when kswapd next wakes.
6841		 */
6842		balanced = pgdat_balanced(pgdat, sc.order, highest_zoneidx);
6843		if (!balanced && nr_boost_reclaim) {
6844			nr_boost_reclaim = 0;
6845			goto restart;
6846		}
6847
6848		/*
6849		 * If boosting is not active then only reclaim if there are no
6850		 * eligible zones. Note that sc.reclaim_idx is not used as
6851		 * buffer_heads_over_limit may have adjusted it.
6852		 */
6853		if (!nr_boost_reclaim && balanced)
6854			goto out;
6855
6856		/* Limit the priority of boosting to avoid reclaim writeback */
6857		if (nr_boost_reclaim && sc.priority == DEF_PRIORITY - 2)
6858			raise_priority = false;
6859
6860		/*
6861		 * Do not writeback or swap pages for boosted reclaim. The
6862		 * intent is to relieve pressure not issue sub-optimal IO
6863		 * from reclaim context. If no pages are reclaimed, the
6864		 * reclaim will be aborted.
6865		 */
6866		sc.may_writepage = !laptop_mode && !nr_boost_reclaim;
6867		sc.may_swap = !nr_boost_reclaim;
6868
6869		/*
6870		 * Do some background aging, to give pages a chance to be
6871		 * referenced before reclaiming. All pages are rotated
6872		 * regardless of classzone as this is about consistent aging.
6873		 */
6874		kswapd_age_node(pgdat, &sc);
6875
6876		/*
6877		 * If we're getting trouble reclaiming, start doing writepage
6878		 * even in laptop mode.
6879		 */
6880		if (sc.priority < DEF_PRIORITY - 2)
6881			sc.may_writepage = 1;
6882
6883		/* Call soft limit reclaim before calling shrink_node. */
6884		sc.nr_scanned = 0;
6885		nr_soft_scanned = 0;
6886		nr_soft_reclaimed = mem_cgroup_soft_limit_reclaim(pgdat, sc.order,
6887						sc.gfp_mask, &nr_soft_scanned);
6888		sc.nr_reclaimed += nr_soft_reclaimed;
6889
6890		/*
6891		 * There should be no need to raise the scanning priority if
6892		 * enough pages are already being scanned that that high
6893		 * watermark would be met at 100% efficiency.
6894		 */
6895		if (kswapd_shrink_node(pgdat, &sc))
6896			raise_priority = false;
6897
6898		/*
6899		 * If the low watermark is met there is no need for processes
6900		 * to be throttled on pfmemalloc_wait as they should not be
6901		 * able to safely make forward progress. Wake them
6902		 */
6903		if (waitqueue_active(&pgdat->pfmemalloc_wait) &&
6904				allow_direct_reclaim(pgdat))
6905			wake_up_all(&pgdat->pfmemalloc_wait);
6906
6907		/* Check if kswapd should be suspending */
6908		__fs_reclaim_release(_THIS_IP_);
6909		ret = kthread_freezable_should_stop(&was_frozen);
6910		__fs_reclaim_acquire(_THIS_IP_);
6911		if (was_frozen || ret)
6912			break;
6913
6914		/*
6915		 * Raise priority if scanning rate is too low or there was no
6916		 * progress in reclaiming pages
6917		 */
6918		nr_reclaimed = sc.nr_reclaimed - nr_reclaimed;
6919		nr_boost_reclaim -= min(nr_boost_reclaim, nr_reclaimed);
6920
6921		/*
6922		 * If reclaim made no progress for a boost, stop reclaim as
6923		 * IO cannot be queued and it could be an infinite loop in
6924		 * extreme circumstances.
6925		 */
6926		if (nr_boost_reclaim && !nr_reclaimed)
6927			break;
6928
6929		if (raise_priority || !nr_reclaimed)
6930			sc.priority--;
6931	} while (sc.priority >= 1);
6932
6933	/*
6934	 * Restart only if it went through the priority loop all the way,
6935	 * but cache_trim_mode didn't work.
6936	 */
6937	if (!sc.nr_reclaimed && sc.priority < 1 &&
6938	    !sc.no_cache_trim_mode && sc.cache_trim_mode_failed) {
6939		sc.no_cache_trim_mode = 1;
6940		goto restart;
6941	}
6942
6943	if (!sc.nr_reclaimed)
6944		pgdat->kswapd_failures++;
6945
6946out:
6947	clear_reclaim_active(pgdat, highest_zoneidx);
6948
6949	/* If reclaim was boosted, account for the reclaim done in this pass */
6950	if (boosted) {
6951		unsigned long flags;
6952
6953		for (i = 0; i <= highest_zoneidx; i++) {
6954			if (!zone_boosts[i])
6955				continue;
6956
6957			/* Increments are under the zone lock */
6958			zone = pgdat->node_zones + i;
6959			spin_lock_irqsave(&zone->lock, flags);
6960			zone->watermark_boost -= min(zone->watermark_boost, zone_boosts[i]);
6961			spin_unlock_irqrestore(&zone->lock, flags);
6962		}
6963
6964		/*
6965		 * As there is now likely space, wakeup kcompact to defragment
6966		 * pageblocks.
6967		 */
6968		wakeup_kcompactd(pgdat, pageblock_order, highest_zoneidx);
6969	}
6970
6971	snapshot_refaults(NULL, pgdat);
6972	__fs_reclaim_release(_THIS_IP_);
6973	psi_memstall_leave(&pflags);
6974	set_task_reclaim_state(current, NULL);
6975
6976	/*
6977	 * Return the order kswapd stopped reclaiming at as
6978	 * prepare_kswapd_sleep() takes it into account. If another caller
6979	 * entered the allocator slow path while kswapd was awake, order will
6980	 * remain at the higher level.
6981	 */
6982	return sc.order;
6983}
6984
6985/*
6986 * The pgdat->kswapd_highest_zoneidx is used to pass the highest zone index to
6987 * be reclaimed by kswapd from the waker. If the value is MAX_NR_ZONES which is
6988 * not a valid index then either kswapd runs for first time or kswapd couldn't
6989 * sleep after previous reclaim attempt (node is still unbalanced). In that
6990 * case return the zone index of the previous kswapd reclaim cycle.
6991 */
6992static enum zone_type kswapd_highest_zoneidx(pg_data_t *pgdat,
6993					   enum zone_type prev_highest_zoneidx)
6994{
6995	enum zone_type curr_idx = READ_ONCE(pgdat->kswapd_highest_zoneidx);
6996
6997	return curr_idx == MAX_NR_ZONES ? prev_highest_zoneidx : curr_idx;
6998}
6999
7000static void kswapd_try_to_sleep(pg_data_t *pgdat, int alloc_order, int reclaim_order,
7001				unsigned int highest_zoneidx)
7002{
7003	long remaining = 0;
7004	DEFINE_WAIT(wait);
7005
7006	if (freezing(current) || kthread_should_stop())
7007		return;
7008
7009	prepare_to_wait(&pgdat->kswapd_wait, &wait, TASK_INTERRUPTIBLE);
7010
7011	/*
7012	 * Try to sleep for a short interval. Note that kcompactd will only be
7013	 * woken if it is possible to sleep for a short interval. This is
7014	 * deliberate on the assumption that if reclaim cannot keep an
7015	 * eligible zone balanced that it's also unlikely that compaction will
7016	 * succeed.
7017	 */
7018	if (prepare_kswapd_sleep(pgdat, reclaim_order, highest_zoneidx)) {
7019		/*
7020		 * Compaction records what page blocks it recently failed to
7021		 * isolate pages from and skips them in the future scanning.
7022		 * When kswapd is going to sleep, it is reasonable to assume
7023		 * that pages and compaction may succeed so reset the cache.
7024		 */
7025		reset_isolation_suitable(pgdat);
7026
7027		/*
7028		 * We have freed the memory, now we should compact it to make
7029		 * allocation of the requested order possible.
7030		 */
7031		wakeup_kcompactd(pgdat, alloc_order, highest_zoneidx);
7032
7033		remaining = schedule_timeout(HZ/10);
7034
7035		/*
7036		 * If woken prematurely then reset kswapd_highest_zoneidx and
7037		 * order. The values will either be from a wakeup request or
7038		 * the previous request that slept prematurely.
7039		 */
7040		if (remaining) {
7041			WRITE_ONCE(pgdat->kswapd_highest_zoneidx,
7042					kswapd_highest_zoneidx(pgdat,
7043							highest_zoneidx));
7044
7045			if (READ_ONCE(pgdat->kswapd_order) < reclaim_order)
7046				WRITE_ONCE(pgdat->kswapd_order, reclaim_order);
7047		}
7048
7049		finish_wait(&pgdat->kswapd_wait, &wait);
7050		prepare_to_wait(&pgdat->kswapd_wait, &wait, TASK_INTERRUPTIBLE);
7051	}
7052
7053	/*
7054	 * After a short sleep, check if it was a premature sleep. If not, then
7055	 * go fully to sleep until explicitly woken up.
7056	 */
7057	if (!remaining &&
7058	    prepare_kswapd_sleep(pgdat, reclaim_order, highest_zoneidx)) {
7059		trace_mm_vmscan_kswapd_sleep(pgdat->node_id);
7060
7061		/*
7062		 * vmstat counters are not perfectly accurate and the estimated
7063		 * value for counters such as NR_FREE_PAGES can deviate from the
7064		 * true value by nr_online_cpus * threshold. To avoid the zone
7065		 * watermarks being breached while under pressure, we reduce the
7066		 * per-cpu vmstat threshold while kswapd is awake and restore
7067		 * them before going back to sleep.
7068		 */
7069		set_pgdat_percpu_threshold(pgdat, calculate_normal_threshold);
7070
7071		if (!kthread_should_stop())
7072			schedule();
7073
7074		set_pgdat_percpu_threshold(pgdat, calculate_pressure_threshold);
7075	} else {
7076		if (remaining)
7077			count_vm_event(KSWAPD_LOW_WMARK_HIT_QUICKLY);
7078		else
7079			count_vm_event(KSWAPD_HIGH_WMARK_HIT_QUICKLY);
7080	}
7081	finish_wait(&pgdat->kswapd_wait, &wait);
7082}
7083
7084/*
7085 * The background pageout daemon, started as a kernel thread
7086 * from the init process.
7087 *
7088 * This basically trickles out pages so that we have _some_
7089 * free memory available even if there is no other activity
7090 * that frees anything up. This is needed for things like routing
7091 * etc, where we otherwise might have all activity going on in
7092 * asynchronous contexts that cannot page things out.
7093 *
7094 * If there are applications that are active memory-allocators
7095 * (most normal use), this basically shouldn't matter.
7096 */
7097static int kswapd(void *p)
7098{
7099	unsigned int alloc_order, reclaim_order;
7100	unsigned int highest_zoneidx = MAX_NR_ZONES - 1;
7101	pg_data_t *pgdat = (pg_data_t *)p;
7102	struct task_struct *tsk = current;
7103	const struct cpumask *cpumask = cpumask_of_node(pgdat->node_id);
7104
7105	if (!cpumask_empty(cpumask))
7106		set_cpus_allowed_ptr(tsk, cpumask);
7107
7108	/*
7109	 * Tell the memory management that we're a "memory allocator",
7110	 * and that if we need more memory we should get access to it
7111	 * regardless (see "__alloc_pages()"). "kswapd" should
7112	 * never get caught in the normal page freeing logic.
7113	 *
7114	 * (Kswapd normally doesn't need memory anyway, but sometimes
7115	 * you need a small amount of memory in order to be able to
7116	 * page out something else, and this flag essentially protects
7117	 * us from recursively trying to free more memory as we're
7118	 * trying to free the first piece of memory in the first place).
7119	 */
7120	tsk->flags |= PF_MEMALLOC | PF_KSWAPD;
7121	set_freezable();
7122
7123	WRITE_ONCE(pgdat->kswapd_order, 0);
7124	WRITE_ONCE(pgdat->kswapd_highest_zoneidx, MAX_NR_ZONES);
7125	atomic_set(&pgdat->nr_writeback_throttled, 0);
7126	for ( ; ; ) {
7127		bool was_frozen;
7128
7129		alloc_order = reclaim_order = READ_ONCE(pgdat->kswapd_order);
7130		highest_zoneidx = kswapd_highest_zoneidx(pgdat,
7131							highest_zoneidx);
7132
7133kswapd_try_sleep:
7134		kswapd_try_to_sleep(pgdat, alloc_order, reclaim_order,
7135					highest_zoneidx);
7136
7137		/* Read the new order and highest_zoneidx */
7138		alloc_order = READ_ONCE(pgdat->kswapd_order);
7139		highest_zoneidx = kswapd_highest_zoneidx(pgdat,
7140							highest_zoneidx);
7141		WRITE_ONCE(pgdat->kswapd_order, 0);
7142		WRITE_ONCE(pgdat->kswapd_highest_zoneidx, MAX_NR_ZONES);
7143
7144		if (kthread_freezable_should_stop(&was_frozen))
 
7145			break;
7146
7147		/*
7148		 * We can speed up thawing tasks if we don't call balance_pgdat
7149		 * after returning from the refrigerator
7150		 */
7151		if (was_frozen)
7152			continue;
7153
7154		/*
7155		 * Reclaim begins at the requested order but if a high-order
7156		 * reclaim fails then kswapd falls back to reclaiming for
7157		 * order-0. If that happens, kswapd will consider sleeping
7158		 * for the order it finished reclaiming at (reclaim_order)
7159		 * but kcompactd is woken to compact for the original
7160		 * request (alloc_order).
7161		 */
7162		trace_mm_vmscan_kswapd_wake(pgdat->node_id, highest_zoneidx,
7163						alloc_order);
7164		reclaim_order = balance_pgdat(pgdat, alloc_order,
7165						highest_zoneidx);
7166		if (reclaim_order < alloc_order)
7167			goto kswapd_try_sleep;
7168	}
7169
7170	tsk->flags &= ~(PF_MEMALLOC | PF_KSWAPD);
7171
7172	return 0;
7173}
7174
7175/*
7176 * A zone is low on free memory or too fragmented for high-order memory.  If
7177 * kswapd should reclaim (direct reclaim is deferred), wake it up for the zone's
7178 * pgdat.  It will wake up kcompactd after reclaiming memory.  If kswapd reclaim
7179 * has failed or is not needed, still wake up kcompactd if only compaction is
7180 * needed.
7181 */
7182void wakeup_kswapd(struct zone *zone, gfp_t gfp_flags, int order,
7183		   enum zone_type highest_zoneidx)
7184{
7185	pg_data_t *pgdat;
7186	enum zone_type curr_idx;
7187
7188	if (!managed_zone(zone))
7189		return;
7190
7191	if (!cpuset_zone_allowed(zone, gfp_flags))
7192		return;
7193
7194	pgdat = zone->zone_pgdat;
7195	curr_idx = READ_ONCE(pgdat->kswapd_highest_zoneidx);
7196
7197	if (curr_idx == MAX_NR_ZONES || curr_idx < highest_zoneidx)
7198		WRITE_ONCE(pgdat->kswapd_highest_zoneidx, highest_zoneidx);
7199
7200	if (READ_ONCE(pgdat->kswapd_order) < order)
7201		WRITE_ONCE(pgdat->kswapd_order, order);
7202
7203	if (!waitqueue_active(&pgdat->kswapd_wait))
7204		return;
7205
7206	/* Hopeless node, leave it to direct reclaim if possible */
7207	if (pgdat->kswapd_failures >= MAX_RECLAIM_RETRIES ||
7208	    (pgdat_balanced(pgdat, order, highest_zoneidx) &&
7209	     !pgdat_watermark_boosted(pgdat, highest_zoneidx))) {
7210		/*
7211		 * There may be plenty of free memory available, but it's too
7212		 * fragmented for high-order allocations.  Wake up kcompactd
7213		 * and rely on compaction_suitable() to determine if it's
7214		 * needed.  If it fails, it will defer subsequent attempts to
7215		 * ratelimit its work.
7216		 */
7217		if (!(gfp_flags & __GFP_DIRECT_RECLAIM))
7218			wakeup_kcompactd(pgdat, order, highest_zoneidx);
7219		return;
7220	}
7221
7222	trace_mm_vmscan_wakeup_kswapd(pgdat->node_id, highest_zoneidx, order,
7223				      gfp_flags);
7224	wake_up_interruptible(&pgdat->kswapd_wait);
7225}
7226
7227#ifdef CONFIG_HIBERNATION
7228/*
7229 * Try to free `nr_to_reclaim' of memory, system-wide, and return the number of
7230 * freed pages.
7231 *
7232 * Rather than trying to age LRUs the aim is to preserve the overall
7233 * LRU order by reclaiming preferentially
7234 * inactive > active > active referenced > active mapped
7235 */
7236unsigned long shrink_all_memory(unsigned long nr_to_reclaim)
7237{
7238	struct scan_control sc = {
7239		.nr_to_reclaim = nr_to_reclaim,
7240		.gfp_mask = GFP_HIGHUSER_MOVABLE,
7241		.reclaim_idx = MAX_NR_ZONES - 1,
7242		.priority = DEF_PRIORITY,
7243		.may_writepage = 1,
7244		.may_unmap = 1,
7245		.may_swap = 1,
7246		.hibernation_mode = 1,
7247	};
7248	struct zonelist *zonelist = node_zonelist(numa_node_id(), sc.gfp_mask);
7249	unsigned long nr_reclaimed;
7250	unsigned int noreclaim_flag;
7251
7252	fs_reclaim_acquire(sc.gfp_mask);
7253	noreclaim_flag = memalloc_noreclaim_save();
7254	set_task_reclaim_state(current, &sc.reclaim_state);
7255
7256	nr_reclaimed = do_try_to_free_pages(zonelist, &sc);
7257
7258	set_task_reclaim_state(current, NULL);
7259	memalloc_noreclaim_restore(noreclaim_flag);
7260	fs_reclaim_release(sc.gfp_mask);
7261
7262	return nr_reclaimed;
7263}
7264#endif /* CONFIG_HIBERNATION */
7265
7266/*
7267 * This kswapd start function will be called by init and node-hot-add.
7268 */
7269void __meminit kswapd_run(int nid)
7270{
7271	pg_data_t *pgdat = NODE_DATA(nid);
7272
7273	pgdat_kswapd_lock(pgdat);
7274	if (!pgdat->kswapd) {
7275		pgdat->kswapd = kthread_run(kswapd, pgdat, "kswapd%d", nid);
7276		if (IS_ERR(pgdat->kswapd)) {
7277			/* failure at boot is fatal */
7278			pr_err("Failed to start kswapd on node %d,ret=%ld\n",
7279				   nid, PTR_ERR(pgdat->kswapd));
7280			BUG_ON(system_state < SYSTEM_RUNNING);
7281			pgdat->kswapd = NULL;
7282		}
7283	}
7284	pgdat_kswapd_unlock(pgdat);
7285}
7286
7287/*
7288 * Called by memory hotplug when all memory in a node is offlined.  Caller must
7289 * be holding mem_hotplug_begin/done().
7290 */
7291void __meminit kswapd_stop(int nid)
7292{
7293	pg_data_t *pgdat = NODE_DATA(nid);
7294	struct task_struct *kswapd;
7295
7296	pgdat_kswapd_lock(pgdat);
7297	kswapd = pgdat->kswapd;
7298	if (kswapd) {
7299		kthread_stop(kswapd);
7300		pgdat->kswapd = NULL;
7301	}
7302	pgdat_kswapd_unlock(pgdat);
7303}
7304
7305static int __init kswapd_init(void)
7306{
7307	int nid;
7308
7309	swap_setup();
7310	for_each_node_state(nid, N_MEMORY)
7311 		kswapd_run(nid);
7312	return 0;
7313}
7314
7315module_init(kswapd_init)
7316
7317#ifdef CONFIG_NUMA
7318/*
7319 * Node reclaim mode
7320 *
7321 * If non-zero call node_reclaim when the number of free pages falls below
7322 * the watermarks.
7323 */
7324int node_reclaim_mode __read_mostly;
7325
7326/*
7327 * Priority for NODE_RECLAIM. This determines the fraction of pages
7328 * of a node considered for each zone_reclaim. 4 scans 1/16th of
7329 * a zone.
7330 */
7331#define NODE_RECLAIM_PRIORITY 4
7332
7333/*
7334 * Percentage of pages in a zone that must be unmapped for node_reclaim to
7335 * occur.
7336 */
7337int sysctl_min_unmapped_ratio = 1;
7338
7339/*
7340 * If the number of slab pages in a zone grows beyond this percentage then
7341 * slab reclaim needs to occur.
7342 */
7343int sysctl_min_slab_ratio = 5;
7344
7345static inline unsigned long node_unmapped_file_pages(struct pglist_data *pgdat)
7346{
7347	unsigned long file_mapped = node_page_state(pgdat, NR_FILE_MAPPED);
7348	unsigned long file_lru = node_page_state(pgdat, NR_INACTIVE_FILE) +
7349		node_page_state(pgdat, NR_ACTIVE_FILE);
7350
7351	/*
7352	 * It's possible for there to be more file mapped pages than
7353	 * accounted for by the pages on the file LRU lists because
7354	 * tmpfs pages accounted for as ANON can also be FILE_MAPPED
7355	 */
7356	return (file_lru > file_mapped) ? (file_lru - file_mapped) : 0;
7357}
7358
7359/* Work out how many page cache pages we can reclaim in this reclaim_mode */
7360static unsigned long node_pagecache_reclaimable(struct pglist_data *pgdat)
7361{
7362	unsigned long nr_pagecache_reclaimable;
7363	unsigned long delta = 0;
7364
7365	/*
7366	 * If RECLAIM_UNMAP is set, then all file pages are considered
7367	 * potentially reclaimable. Otherwise, we have to worry about
7368	 * pages like swapcache and node_unmapped_file_pages() provides
7369	 * a better estimate
7370	 */
7371	if (node_reclaim_mode & RECLAIM_UNMAP)
7372		nr_pagecache_reclaimable = node_page_state(pgdat, NR_FILE_PAGES);
7373	else
7374		nr_pagecache_reclaimable = node_unmapped_file_pages(pgdat);
7375
7376	/* If we can't clean pages, remove dirty pages from consideration */
7377	if (!(node_reclaim_mode & RECLAIM_WRITE))
7378		delta += node_page_state(pgdat, NR_FILE_DIRTY);
7379
7380	/* Watch for any possible underflows due to delta */
7381	if (unlikely(delta > nr_pagecache_reclaimable))
7382		delta = nr_pagecache_reclaimable;
7383
7384	return nr_pagecache_reclaimable - delta;
7385}
7386
7387/*
7388 * Try to free up some pages from this node through reclaim.
7389 */
7390static int __node_reclaim(struct pglist_data *pgdat, gfp_t gfp_mask, unsigned int order)
7391{
7392	/* Minimum pages needed in order to stay on node */
7393	const unsigned long nr_pages = 1 << order;
7394	struct task_struct *p = current;
7395	unsigned int noreclaim_flag;
7396	struct scan_control sc = {
7397		.nr_to_reclaim = max(nr_pages, SWAP_CLUSTER_MAX),
7398		.gfp_mask = current_gfp_context(gfp_mask),
7399		.order = order,
7400		.priority = NODE_RECLAIM_PRIORITY,
7401		.may_writepage = !!(node_reclaim_mode & RECLAIM_WRITE),
7402		.may_unmap = !!(node_reclaim_mode & RECLAIM_UNMAP),
7403		.may_swap = 1,
7404		.reclaim_idx = gfp_zone(gfp_mask),
7405	};
7406	unsigned long pflags;
7407
7408	trace_mm_vmscan_node_reclaim_begin(pgdat->node_id, order,
7409					   sc.gfp_mask);
7410
7411	cond_resched();
7412	psi_memstall_enter(&pflags);
7413	delayacct_freepages_start();
7414	fs_reclaim_acquire(sc.gfp_mask);
7415	/*
7416	 * We need to be able to allocate from the reserves for RECLAIM_UNMAP
7417	 */
7418	noreclaim_flag = memalloc_noreclaim_save();
7419	set_task_reclaim_state(p, &sc.reclaim_state);
7420
7421	if (node_pagecache_reclaimable(pgdat) > pgdat->min_unmapped_pages ||
7422	    node_page_state_pages(pgdat, NR_SLAB_RECLAIMABLE_B) > pgdat->min_slab_pages) {
7423		/*
7424		 * Free memory by calling shrink node with increasing
7425		 * priorities until we have enough memory freed.
7426		 */
7427		do {
7428			shrink_node(pgdat, &sc);
7429		} while (sc.nr_reclaimed < nr_pages && --sc.priority >= 0);
7430	}
7431
7432	set_task_reclaim_state(p, NULL);
7433	memalloc_noreclaim_restore(noreclaim_flag);
7434	fs_reclaim_release(sc.gfp_mask);
7435	psi_memstall_leave(&pflags);
7436	delayacct_freepages_end();
7437
7438	trace_mm_vmscan_node_reclaim_end(sc.nr_reclaimed);
7439
7440	return sc.nr_reclaimed >= nr_pages;
7441}
7442
7443int node_reclaim(struct pglist_data *pgdat, gfp_t gfp_mask, unsigned int order)
7444{
7445	int ret;
7446
7447	/*
7448	 * Node reclaim reclaims unmapped file backed pages and
7449	 * slab pages if we are over the defined limits.
7450	 *
7451	 * A small portion of unmapped file backed pages is needed for
7452	 * file I/O otherwise pages read by file I/O will be immediately
7453	 * thrown out if the node is overallocated. So we do not reclaim
7454	 * if less than a specified percentage of the node is used by
7455	 * unmapped file backed pages.
7456	 */
7457	if (node_pagecache_reclaimable(pgdat) <= pgdat->min_unmapped_pages &&
7458	    node_page_state_pages(pgdat, NR_SLAB_RECLAIMABLE_B) <=
7459	    pgdat->min_slab_pages)
7460		return NODE_RECLAIM_FULL;
7461
7462	/*
7463	 * Do not scan if the allocation should not be delayed.
7464	 */
7465	if (!gfpflags_allow_blocking(gfp_mask) || (current->flags & PF_MEMALLOC))
7466		return NODE_RECLAIM_NOSCAN;
7467
7468	/*
7469	 * Only run node reclaim on the local node or on nodes that do not
7470	 * have associated processors. This will favor the local processor
7471	 * over remote processors and spread off node memory allocations
7472	 * as wide as possible.
7473	 */
7474	if (node_state(pgdat->node_id, N_CPU) && pgdat->node_id != numa_node_id())
7475		return NODE_RECLAIM_NOSCAN;
7476
7477	if (test_and_set_bit(PGDAT_RECLAIM_LOCKED, &pgdat->flags))
7478		return NODE_RECLAIM_NOSCAN;
7479
7480	ret = __node_reclaim(pgdat, gfp_mask, order);
7481	clear_bit(PGDAT_RECLAIM_LOCKED, &pgdat->flags);
7482
7483	if (!ret)
7484		count_vm_event(PGSCAN_ZONE_RECLAIM_FAILED);
7485
7486	return ret;
7487}
7488#endif
7489
7490/**
7491 * check_move_unevictable_folios - Move evictable folios to appropriate zone
7492 * lru list
7493 * @fbatch: Batch of lru folios to check.
7494 *
7495 * Checks folios for evictability, if an evictable folio is in the unevictable
7496 * lru list, moves it to the appropriate evictable lru list. This function
7497 * should be only used for lru folios.
7498 */
7499void check_move_unevictable_folios(struct folio_batch *fbatch)
7500{
7501	struct lruvec *lruvec = NULL;
7502	int pgscanned = 0;
7503	int pgrescued = 0;
7504	int i;
7505
7506	for (i = 0; i < fbatch->nr; i++) {
7507		struct folio *folio = fbatch->folios[i];
7508		int nr_pages = folio_nr_pages(folio);
7509
7510		pgscanned += nr_pages;
7511
7512		/* block memcg migration while the folio moves between lrus */
7513		if (!folio_test_clear_lru(folio))
7514			continue;
7515
7516		lruvec = folio_lruvec_relock_irq(folio, lruvec);
7517		if (folio_evictable(folio) && folio_test_unevictable(folio)) {
7518			lruvec_del_folio(lruvec, folio);
7519			folio_clear_unevictable(folio);
7520			lruvec_add_folio(lruvec, folio);
7521			pgrescued += nr_pages;
7522		}
7523		folio_set_lru(folio);
7524	}
7525
7526	if (lruvec) {
7527		__count_vm_events(UNEVICTABLE_PGRESCUED, pgrescued);
7528		__count_vm_events(UNEVICTABLE_PGSCANNED, pgscanned);
7529		unlock_page_lruvec_irq(lruvec);
7530	} else if (pgscanned) {
7531		count_vm_events(UNEVICTABLE_PGSCANNED, pgscanned);
7532	}
7533}
7534EXPORT_SYMBOL_GPL(check_move_unevictable_folios);