projects / linux-block.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Merge tag 'for_linus' of git://git.kernel.org/pub/scm/linux/kernel/git/mst/vhost
[linux-block.git] / fs / proc / task_mmu.c
CommitLineData
b2441318 1// SPDX-License-Identifier: GPL-2.0
a520110e 2#include <linux/pagewalk.h>
17fca131 3#include <linux/mm_inline.h>
1da177e4 4#include <linux/hugetlb.h>
22e057c5 5#include <linux/huge_mm.h>
1da177e4 6#include <linux/mount.h>
8b479335 7#include <linux/ksm.h>
1da177e4 8#include <linux/seq_file.h>
e070ad49 9#include <linux/highmem.h>
5096add8 10#include <linux/ptrace.h>
5a0e3ad6 11#include <linux/slab.h>
6e21c8f1
CL		 12#include <linux/pagemap.h>
13#include <linux/mempolicy.h>
22e057c5 14#include <linux/rmap.h>
85863e47 15#include <linux/swap.h>
6e84f315 16#include <linux/sched/mm.h>
85863e47 17#include <linux/swapops.h>
0f8975ec 18#include <linux/mmu_notifier.h>
33c3fc71 19#include <linux/page_idle.h>
6a15a370 20#include <linux/shmem_fs.h>
b3a81d08 21#include <linux/uaccess.h>
27cca866 22#include <linux/pkeys.h>
52526ca7
MUA		 23#include <linux/minmax.h>
24#include <linux/overflow.h>
e070ad49 25
1da177e4 26#include <asm/elf.h>
b3a81d08 27#include <asm/tlb.h>
e070ad49 28#include <asm/tlbflush.h>
1da177e4
LT		 29#include "internal.h"
30
d1be35cb
AV		 31#define SEQ_PUT_DEC(str, val) \
32 seq_put_decimal_ull_width(m, str, (val) << (PAGE_SHIFT-10), 8)
df5f8314 33void task_mem(struct seq_file *m, struct mm_struct *mm)
1da177e4 34{
af5b0f6a 35 unsigned long text, lib, swap, anon, file, shmem;
365e9c87
HD		 36 unsigned long hiwater_vm, total_vm, hiwater_rss, total_rss;
37
8cee852e
JM		 38 anon = get_mm_counter(mm, MM_ANONPAGES);
39 file = get_mm_counter(mm, MM_FILEPAGES);
40 shmem = get_mm_counter(mm, MM_SHMEMPAGES);
41
365e9c87
HD		 42 /*
43 * Note: to minimize their overhead, mm maintains hiwater_vm and
44 * hiwater_rss only when about to *lower* total_vm or rss. Any
45 * collector of these hiwater stats must therefore get total_vm
46 * and rss too, which will usually be the higher. Barriers? not
47 * worth the effort, such snapshots can always be inconsistent.
48 */
49 hiwater_vm = total_vm = mm->total_vm;
50 if (hiwater_vm < mm->hiwater_vm)
51 hiwater_vm = mm->hiwater_vm;
8cee852e 52 hiwater_rss = total_rss = anon + file + shmem;
365e9c87
HD		 53 if (hiwater_rss < mm->hiwater_rss)
54 hiwater_rss = mm->hiwater_rss;
1da177e4 55
8526d84f
KK		 56 /* split executable areas between text and lib */
57 text = PAGE_ALIGN(mm->end_code) - (mm->start_code & PAGE_MASK);
58 text = min(text, mm->exec_vm << PAGE_SHIFT);
59 lib = (mm->exec_vm << PAGE_SHIFT) - text;
60
b084d435 61 swap = get_mm_counter(mm, MM_SWAPENTS);
d1be35cb
AV		 62 SEQ_PUT_DEC("VmPeak:\t", hiwater_vm);
63 SEQ_PUT_DEC(" kB\nVmSize:\t", total_vm);
64 SEQ_PUT_DEC(" kB\nVmLck:\t", mm->locked_vm);
70f8a3ca 65 SEQ_PUT_DEC(" kB\nVmPin:\t", atomic64_read(&mm->pinned_vm));
d1be35cb
AV		 66 SEQ_PUT_DEC(" kB\nVmHWM:\t", hiwater_rss);
67 SEQ_PUT_DEC(" kB\nVmRSS:\t", total_rss);
68 SEQ_PUT_DEC(" kB\nRssAnon:\t", anon);
69 SEQ_PUT_DEC(" kB\nRssFile:\t", file);
70 SEQ_PUT_DEC(" kB\nRssShmem:\t", shmem);
71 SEQ_PUT_DEC(" kB\nVmData:\t", mm->data_vm);
72 SEQ_PUT_DEC(" kB\nVmStk:\t", mm->stack_vm);
73 seq_put_decimal_ull_width(m,
74 " kB\nVmExe:\t", text >> 10, 8);
75 seq_put_decimal_ull_width(m,
76 " kB\nVmLib:\t", lib >> 10, 8);
77 seq_put_decimal_ull_width(m,
78 " kB\nVmPTE:\t", mm_pgtables_bytes(mm) >> 10, 8);
79 SEQ_PUT_DEC(" kB\nVmSwap:\t", swap);
80 seq_puts(m, " kB\n");
5d317b2b 81 hugetlb_report_usage(m, mm);
1da177e4 82}
d1be35cb 83#undef SEQ_PUT_DEC
1da177e4
LT		 84
85unsigned long task_vsize(struct mm_struct *mm)
86{
87 return PAGE_SIZE * mm->total_vm;
88}
89
a2ade7b6
AD		 90unsigned long task_statm(struct mm_struct *mm,
91 unsigned long *shared, unsigned long *text,
92 unsigned long *data, unsigned long *resident)
1da177e4 93{
eca56ff9
JM		 94 *shared = get_mm_counter(mm, MM_FILEPAGES) +
95 get_mm_counter(mm, MM_SHMEMPAGES);
1da177e4
LT		 96 *text = (PAGE_ALIGN(mm->end_code) - (mm->start_code & PAGE_MASK))
97 >> PAGE_SHIFT;
84638335 98 *data = mm->data_vm + mm->stack_vm;
d559db08 99 *resident = *shared + get_mm_counter(mm, MM_ANONPAGES);
1da177e4
LT		 100 return mm->total_vm;
101}
102
9e781440
KH		 103#ifdef CONFIG_NUMA
104/*
498f2371 105 * Save get_task_policy() for show_numa_map().
9e781440
KH		 106 */
107static void hold_task_mempolicy(struct proc_maps_private *priv)
108{
109 struct task_struct *task = priv->task;
110
111 task_lock(task);
498f2371 112 priv->task_mempolicy = get_task_policy(task);
9e781440
KH		 113 mpol_get(priv->task_mempolicy);
114 task_unlock(task);
115}
116static void release_task_mempolicy(struct proc_maps_private *priv)
117{
118 mpol_put(priv->task_mempolicy);
119}
120#else
121static void hold_task_mempolicy(struct proc_maps_private *priv)
122{
123}
124static void release_task_mempolicy(struct proc_maps_private *priv)
125{
126}
127#endif
128
c4c84f06
MWO		 129static struct vm_area_struct *proc_get_vma(struct proc_maps_private *priv,
130 loff_t *ppos)
131{
132 struct vm_area_struct *vma = vma_next(&priv->iter);
133
134 if (vma) {
135 *ppos = vma->vm_start;
136 } else {
137 *ppos = -2UL;
138 vma = get_gate_vma(priv->mm);
139 }
140
141 return vma;
142}
143
0c255321 144static void *m_start(struct seq_file *m, loff_t *ppos)
e070ad49 145{
a6198797 146 struct proc_maps_private *priv = m->private;
4781f2c3 147 unsigned long last_addr = *ppos;
a6198797 148 struct mm_struct *mm;
a6198797 149
c2e88d22 150 /* See m_next(). Zero at the start or after lseek. */
b8c20a9b
ON		 151 if (last_addr == -1UL)
152 return NULL;
153
2c03376d 154 priv->task = get_proc_task(priv->inode);
a6198797 155 if (!priv->task)
ec6fd8a4 156 return ERR_PTR(-ESRCH);
a6198797 157
29a40ace 158 mm = priv->mm;
d07ded61
MWO		 159 if (!mm || !mmget_not_zero(mm)) {
160 put_task_struct(priv->task);
161 priv->task = NULL;
29a40ace 162 return NULL;
d07ded61 163 }
a6198797 164
d8ed45c5 165 if (mmap_read_lock_killable(mm)) {
8a713e7d 166 mmput(mm);
d07ded61
MWO		 167 put_task_struct(priv->task);
168 priv->task = NULL;
8a713e7d
KK		 169 return ERR_PTR(-EINTR);
170 }
171
c4c84f06 172 vma_iter_init(&priv->iter, mm, last_addr);
9e781440 173 hold_task_mempolicy(priv);
c4c84f06
MWO		 174 if (last_addr == -2UL)
175 return get_gate_vma(mm);
59b4bf12 176
c4c84f06 177 return proc_get_vma(priv, ppos);
a6198797
MM		 178}
179
4781f2c3 180static void *m_next(struct seq_file *m, void *v, loff_t *ppos)
a6198797 181{
c4c84f06
MWO		 182 if (*ppos == -2UL) {
183 *ppos = -1UL;
184 return NULL;
185 }
186 return proc_get_vma(m->private, ppos);
a6198797
MM		 187}
188
189static void m_stop(struct seq_file *m, void *v)
190{
191 struct proc_maps_private *priv = m->private;
d07ded61 192 struct mm_struct *mm = priv->mm;
a6198797 193
d07ded61
MWO		 194 if (!priv->task)
195 return;
196
197 release_task_mempolicy(priv);
d8ed45c5 198 mmap_read_unlock(mm);
d07ded61
MWO		 199 mmput(mm);
200 put_task_struct(priv->task);
201 priv->task = NULL;
a6198797
MM		 202}
203
4db7d0ee
ON		 204static int proc_maps_open(struct inode *inode, struct file *file,
205 const struct seq_operations *ops, int psize)
206{
207 struct proc_maps_private *priv = __seq_open_private(file, ops, psize);
208
209 if (!priv)
210 return -ENOMEM;
211
2c03376d 212 priv->inode = inode;
29a40ace
ON		 213 priv->mm = proc_mem_open(inode, PTRACE_MODE_READ);
214 if (IS_ERR(priv->mm)) {
215 int err = PTR_ERR(priv->mm);
216
217 seq_release_private(inode, file);
218 return err;
219 }
220
4db7d0ee
ON		 221 return 0;
222}
223
29a40ace
ON		 224static int proc_map_release(struct inode *inode, struct file *file)
225{
226 struct seq_file *seq = file->private_data;
227 struct proc_maps_private *priv = seq->private;
228
229 if (priv->mm)
230 mmdrop(priv->mm);
231
232 return seq_release_private(inode, file);
233}
234
a6198797 235static int do_maps_open(struct inode *inode, struct file *file,
03a44825 236 const struct seq_operations *ops)
a6198797 237{
4db7d0ee
ON		 238 return proc_maps_open(inode, file, ops,
239 sizeof(struct proc_maps_private));
a6198797 240}
e070ad49 241
493b0e9d
DC		 242static void show_vma_header_prefix(struct seq_file *m,
243 unsigned long start, unsigned long end,
244 vm_flags_t flags, unsigned long long pgoff,
245 dev_t dev, unsigned long ino)
246{
247 seq_setwidth(m, 25 + sizeof(void *) * 6 - 1);
0e3dc019
AV		 248 seq_put_hex_ll(m, NULL, start, 8);
249 seq_put_hex_ll(m, "-", end, 8);
250 seq_putc(m, ' ');
251 seq_putc(m, flags & VM_READ ? 'r' : '-');
252 seq_putc(m, flags & VM_WRITE ? 'w' : '-');
253 seq_putc(m, flags & VM_EXEC ? 'x' : '-');
254 seq_putc(m, flags & VM_MAYSHARE ? 's' : 'p');
255 seq_put_hex_ll(m, " ", pgoff, 8);
256 seq_put_hex_ll(m, " ", MAJOR(dev), 2);
257 seq_put_hex_ll(m, ":", MINOR(dev), 2);
258 seq_put_decimal_ull(m, " ", ino);
259 seq_putc(m, ' ');
493b0e9d
DC		 260}
261
b7643757 262static void
871305bb 263show_map_vma(struct seq_file *m, struct vm_area_struct *vma)
1da177e4 264{
d09e8ca6 265 struct anon_vma_name *anon_name = NULL;
e070ad49
ML		 266 struct mm_struct *mm = vma->vm_mm;
267 struct file *file = vma->vm_file;
ca16d140 268 vm_flags_t flags = vma->vm_flags;
1da177e4 269 unsigned long ino = 0;
6260a4b0 270 unsigned long long pgoff = 0;
a09a79f6 271 unsigned long start, end;
1da177e4 272 dev_t dev = 0;
b7643757 273 const char *name = NULL;
1da177e4
LT		 274
275 if (file) {
3efdc78f
AV		 276 const struct inode *inode = file_user_inode(vma->vm_file);
277
1da177e4
LT		 278 dev = inode->i_sb->s_dev;
279 ino = inode->i_ino;
6260a4b0 280 pgoff = ((loff_t)vma->vm_pgoff) << PAGE_SHIFT;
1da177e4
LT		 281 }
282
d7824370 283 start = vma->vm_start;
a09a79f6 284 end = vma->vm_end;
493b0e9d 285 show_vma_header_prefix(m, start, end, flags, pgoff, dev, ino);
d09e8ca6
PT		 286 if (mm)
287 anon_name = anon_vma_name(vma);
1da177e4
LT		 288
289 /*
290 * Print the dentry name for named mappings, and a
291 * special [heap] marker for the heap:
292 */
e070ad49 293 if (file) {
652586df 294 seq_pad(m, ' ');
d09e8ca6
PT		 295 /*
296 * If user named this anon shared memory via
297 * prctl(PR_SET_VMA ..., use the provided name.
298 */
299 if (anon_name)
300 seq_printf(m, "[anon_shmem:%s]", anon_name->name);
301 else
08582d67 302 seq_path(m, file_user_path(file), "\n");
b7643757
SP		 303 goto done;
304 }
305
78d683e8
AL		 306 if (vma->vm_ops && vma->vm_ops->name) {
307 name = vma->vm_ops->name(vma);
308 if (name)
309 goto done;
310 }
311
b7643757
SP		 312 name = arch_vma_name(vma);
313 if (!name) {
b7643757
SP		 314 if (!mm) {
315 name = "[vdso]";
316 goto done;
317 }
318
11250fd1 319 if (vma_is_initial_heap(vma)) {
b7643757
SP		 320 name = "[heap]";
321 goto done;
322 }
323
11250fd1 324 if (vma_is_initial_stack(vma)) {
65376df5 325 name = "[stack]";
9a10064f
CC		 326 goto done;
327 }
328
9a10064f
CC		 329 if (anon_name) {
330 seq_pad(m, ' ');
5c26f6ac 331 seq_printf(m, "[anon:%s]", anon_name->name);
9a10064f 332 }
b7643757
SP		 333 }
334
335done:
336 if (name) {
652586df 337 seq_pad(m, ' ');
b7643757 338 seq_puts(m, name);
1da177e4
LT		 339 }
340 seq_putc(m, '\n');
7c88db0c
JK		 341}
342
871305bb 343static int show_map(struct seq_file *m, void *v)
7c88db0c 344{
871305bb 345 show_map_vma(m, v);
1da177e4
LT		 346 return 0;
347}
348
03a44825 349static const struct seq_operations proc_pid_maps_op = {
a6198797
MM		 350 .start = m_start,
351 .next = m_next,
352 .stop = m_stop,
871305bb 353 .show = show_map
a6198797
MM		 354};
355
b7643757 356static int pid_maps_open(struct inode *inode, struct file *file)
a6198797
MM		 357{
358 return do_maps_open(inode, file, &proc_pid_maps_op);
359}
360
b7643757
SP		 361const struct file_operations proc_pid_maps_operations = {
362 .open = pid_maps_open,
363 .read = seq_read,
364 .llseek = seq_lseek,
29a40ace 365 .release = proc_map_release,
b7643757
SP		 366};
367
a6198797
MM		 368/*
369 * Proportional Set Size(PSS): my share of RSS.
370 *
371 * PSS of a process is the count of pages it has in memory, where each
372 * page is divided by the number of processes sharing it. So if a
373 * process has 1000 pages all to itself, and 1000 shared with one other
374 * process, its PSS will be 1500.
375 *
376 * To keep (accumulated) division errors low, we adopt a 64bit
377 * fixed-point pss counter to minimize division errors. So (pss >>
378 * PSS_SHIFT) would be the real byte count.
379 *
380 * A shift of 12 before division means (assuming 4K page size):
381 * - 1M 3-user-pages add up to 8KB errors;
382 * - supports mapcount up to 2^24, or 16M;
383 * - supports PSS up to 2^52 bytes, or 4PB.
384 */
385#define PSS_SHIFT 12
386
1e883281 387#ifdef CONFIG_PROC_PAGE_MONITOR
214e471f 388struct mem_size_stats {
a6198797
MM		 389 unsigned long resident;
390 unsigned long shared_clean;
391 unsigned long shared_dirty;
392 unsigned long private_clean;
393 unsigned long private_dirty;
394 unsigned long referenced;
b40d4f84 395 unsigned long anonymous;
cf8496ea 396 unsigned long lazyfree;
4031a219 397 unsigned long anonymous_thp;
65c45377 398 unsigned long shmem_thp;
60fbf0ab 399 unsigned long file_thp;
214e471f 400 unsigned long swap;
25ee01a2
NH		 401 unsigned long shared_hugetlb;
402 unsigned long private_hugetlb;
8b479335 403 unsigned long ksm;
a6198797 404 u64 pss;
ee2ad71b
LS		 405 u64 pss_anon;
406 u64 pss_file;
407 u64 pss_shmem;
30934843 408 u64 pss_dirty;
493b0e9d 409 u64 pss_locked;
8334b962 410 u64 swap_pss;
a6198797
MM		 411};
412
ee2ad71b
LS		 413static void smaps_page_accumulate(struct mem_size_stats *mss,
414 struct page *page, unsigned long size, unsigned long pss,
415 bool dirty, bool locked, bool private)
416{
417 mss->pss += pss;
418
419 if (PageAnon(page))
420 mss->pss_anon += pss;
421 else if (PageSwapBacked(page))
422 mss->pss_shmem += pss;
423 else
424 mss->pss_file += pss;
425
426 if (locked)
427 mss->pss_locked += pss;
428
429 if (dirty || PageDirty(page)) {
30934843 430 mss->pss_dirty += pss;
ee2ad71b
LS		 431 if (private)
432 mss->private_dirty += size;
433 else
434 mss->shared_dirty += size;
435 } else {
436 if (private)
437 mss->private_clean += size;
438 else
439 mss->shared_clean += size;
440 }
441}
442
c164e038 443static void smaps_account(struct mem_size_stats *mss, struct page *page,
24d7275c
YS		 444 bool compound, bool young, bool dirty, bool locked,
445 bool migration)
c164e038 446{
d8c6546b 447 int i, nr = compound ? compound_nr(page) : 1;
afd9883f 448 unsigned long size = nr * PAGE_SIZE;
c164e038 449
ee2ad71b
LS		 450 /*
451 * First accumulate quantities that depend only on |size| and the type
452 * of the compound page.
453 */
cf8496ea 454 if (PageAnon(page)) {
c164e038 455 mss->anonymous += size;
cf8496ea
SL		 456 if (!PageSwapBacked(page) && !dirty && !PageDirty(page))
457 mss->lazyfree += size;
458 }
c164e038 459
8b479335
SR		 460 if (PageKsm(page))
461 mss->ksm += size;
462
c164e038
KS		 463 mss->resident += size;
464 /* Accumulate the size in pages that have been accessed. */
33c3fc71 465 if (young || page_is_young(page) || PageReferenced(page))
c164e038 466 mss->referenced += size;
c164e038 467
afd9883f 468 /*
ee2ad71b
LS		 469 * Then accumulate quantities that may depend on sharing, or that may
470 * differ page-by-page.
471 *
afd9883f
KS		 472 * page_count(page) == 1 guarantees the page is mapped exactly once.
473 * If any subpage of the compound page mapped with PTE it would elevate
474 * page_count().
24d7275c
YS		 475 *
476 * The page_mapcount() is called to get a snapshot of the mapcount.
477 * Without holding the page lock this snapshot can be slightly wrong as
478 * we cannot always read the mapcount atomically. It is not safe to
479 * call page_mapcount() even with PTL held if the page is not mapped,
480 * especially for migration entries. Treat regular migration entries
481 * as mapcount == 1.
afd9883f 482 */
24d7275c 483 if ((page_count(page) == 1) || migration) {
ee2ad71b
LS		 484 smaps_page_accumulate(mss, page, size, size << PSS_SHIFT, dirty,
485 locked, true);
afd9883f
KS		 486 return;
487 }
afd9883f
KS		 488 for (i = 0; i < nr; i++, page++) {
489 int mapcount = page_mapcount(page);
ee2ad71b
LS		 490 unsigned long pss = PAGE_SIZE << PSS_SHIFT;
491 if (mapcount >= 2)
492 pss /= mapcount;
493 smaps_page_accumulate(mss, page, PAGE_SIZE, pss, dirty, locked,
494 mapcount < 2);
c164e038
KS		 495 }
496}
ae11c4d9 497
c261e7d9 498#ifdef CONFIG_SHMEM
c261e7d9 499static int smaps_pte_hole(unsigned long addr, unsigned long end,
b7a16c7a 500 __always_unused int depth, struct mm_walk *walk)
c261e7d9
VB		 501{
502 struct mem_size_stats *mss = walk->private;
10c848c8 503 struct vm_area_struct *vma = walk->vma;
c261e7d9 504
10c848c8
PX		 505 mss->swap += shmem_partial_swap_usage(walk->vma->vm_file->f_mapping,
506 linear_page_index(vma, addr),
507 linear_page_index(vma, end));
c261e7d9
VB		 508
509 return 0;
510}
7b86ac33
CH		 511#else
512#define smaps_pte_hole NULL
513#endif /* CONFIG_SHMEM */
c261e7d9 514
23010032
PX		 515static void smaps_pte_hole_lookup(unsigned long addr, struct mm_walk *walk)
516{
517#ifdef CONFIG_SHMEM
518 if (walk->ops->pte_hole) {
519 /* depth is not used */
520 smaps_pte_hole(addr, addr + PAGE_SIZE, 0, walk);
521 }
522#endif
523}
524
c164e038
KS		 525static void smaps_pte_entry(pte_t *pte, unsigned long addr,
526 struct mm_walk *walk)
ae11c4d9
DH		 527{
528 struct mem_size_stats *mss = walk->private;
14eb6fdd 529 struct vm_area_struct *vma = walk->vma;
27dd768e 530 bool locked = !!(vma->vm_flags & VM_LOCKED);
b1d4d9e0 531 struct page *page = NULL;
efd41493 532 bool migration = false, young = false, dirty = false;
c33c7948 533 pte_t ptent = ptep_get(pte);
ae11c4d9 534
c33c7948
RR		 535 if (pte_present(ptent)) {
536 page = vm_normal_page(vma, addr, ptent);
537 young = pte_young(ptent);
538 dirty = pte_dirty(ptent);
539 } else if (is_swap_pte(ptent)) {
540 swp_entry_t swpent = pte_to_swp_entry(ptent);
ae11c4d9 541
8334b962
MK		 542 if (!non_swap_entry(swpent)) {
543 int mapcount;
544
c164e038 545 mss->swap += PAGE_SIZE;
8334b962
MK		 546 mapcount = swp_swapcount(swpent);
547 if (mapcount >= 2) {
548 u64 pss_delta = (u64)PAGE_SIZE << PSS_SHIFT;
549
550 do_div(pss_delta, mapcount);
551 mss->swap_pss += pss_delta;
552 } else {
553 mss->swap_pss += (u64)PAGE_SIZE << PSS_SHIFT;
554 }
24d7275c
YS		 555 } else if (is_pfn_swap_entry(swpent)) {
556 if (is_migration_entry(swpent))
557 migration = true;
af5cdaf8 558 page = pfn_swap_entry_to_page(swpent);
24d7275c 559 }
23010032
PX		 560 } else {
561 smaps_pte_hole_lookup(addr, walk);
48131e03 562 return;
b1d4d9e0 563 }
ae11c4d9 564
ae11c4d9
DH		 565 if (!page)
566 return;
afd9883f 567
efd41493 568 smaps_account(mss, page, false, young, dirty, locked, migration);
ae11c4d9
DH		 569}
570
c164e038
KS		 571#ifdef CONFIG_TRANSPARENT_HUGEPAGE
572static void smaps_pmd_entry(pmd_t *pmd, unsigned long addr,
573 struct mm_walk *walk)
574{
575 struct mem_size_stats *mss = walk->private;
14eb6fdd 576 struct vm_area_struct *vma = walk->vma;
27dd768e 577 bool locked = !!(vma->vm_flags & VM_LOCKED);
c94b6923 578 struct page *page = NULL;
24d7275c 579 bool migration = false;
c94b6923
HY		 580
581 if (pmd_present(*pmd)) {
8b9c1cc0 582 page = vm_normal_page_pmd(vma, addr, *pmd);
c94b6923
HY		 583 } else if (unlikely(thp_migration_supported() && is_swap_pmd(*pmd))) {
584 swp_entry_t entry = pmd_to_swp_entry(*pmd);
c164e038 585
24d7275c
YS		 586 if (is_migration_entry(entry)) {
587 migration = true;
af5cdaf8 588 page = pfn_swap_entry_to_page(entry);
24d7275c 589 }
c94b6923 590 }
c164e038
KS		 591 if (IS_ERR_OR_NULL(page))
592 return;
65c45377
KS		 593 if (PageAnon(page))
594 mss->anonymous_thp += HPAGE_PMD_SIZE;
595 else if (PageSwapBacked(page))
596 mss->shmem_thp += HPAGE_PMD_SIZE;
ca120cf6
DW		 597 else if (is_zone_device_page(page))
598 /* pass */;
65c45377 599 else
60fbf0ab 600 mss->file_thp += HPAGE_PMD_SIZE;
24d7275c
YS		 601
602 smaps_account(mss, page, true, pmd_young(*pmd), pmd_dirty(*pmd),
603 locked, migration);
c164e038
KS		 604}
605#else
606static void smaps_pmd_entry(pmd_t *pmd, unsigned long addr,
607 struct mm_walk *walk)
608{
609}
610#endif
611
b3ae5acb 612static int smaps_pte_range(pmd_t *pmd, unsigned long addr, unsigned long end,
2165009b 613 struct mm_walk *walk)
e070ad49 614{
14eb6fdd 615 struct vm_area_struct *vma = walk->vma;
ae11c4d9 616 pte_t *pte;
705e87c0 617 spinlock_t *ptl;
e070ad49 618
b6ec57f4
KS		 619 ptl = pmd_trans_huge_lock(pmd, vma);
620 if (ptl) {
c94b6923 621 smaps_pmd_entry(pmd, addr, walk);
bf929152 622 spin_unlock(ptl);
14038302 623 goto out;
22e057c5 624 }
1a5a9906 625
705e87c0 626 pte = pte_offset_map_lock(vma->vm_mm, pmd, addr, &ptl);
7780d040
HD		 627 if (!pte) {
628 walk->action = ACTION_AGAIN;
629 return 0;
630 }
ae11c4d9 631 for (; addr != end; pte++, addr += PAGE_SIZE)
c164e038 632 smaps_pte_entry(pte, addr, walk);
705e87c0 633 pte_unmap_unlock(pte - 1, ptl);
14038302 634out:
705e87c0 635 cond_resched();
b3ae5acb 636 return 0;
e070ad49
ML		 637}
638
834f82e2
CG		 639static void show_smap_vma_flags(struct seq_file *m, struct vm_area_struct *vma)
640{
641 /*
642 * Don't forget to update Documentation/ on changes.
643 */
644 static const char mnemonics[BITS_PER_LONG][2] = {
645 /*
646 * In case if we meet a flag we don't know about.
647 */
648 [0 ... (BITS_PER_LONG-1)] = "??",
649
650 [ilog2(VM_READ)] = "rd",
651 [ilog2(VM_WRITE)] = "wr",
652 [ilog2(VM_EXEC)] = "ex",
653 [ilog2(VM_SHARED)] = "sh",
654 [ilog2(VM_MAYREAD)] = "mr",
655 [ilog2(VM_MAYWRITE)] = "mw",
656 [ilog2(VM_MAYEXEC)] = "me",
657 [ilog2(VM_MAYSHARE)] = "ms",
658 [ilog2(VM_GROWSDOWN)] = "gd",
659 [ilog2(VM_PFNMAP)] = "pf",
834f82e2
CG		 660 [ilog2(VM_LOCKED)] = "lo",
661 [ilog2(VM_IO)] = "io",
662 [ilog2(VM_SEQ_READ)] = "sr",
663 [ilog2(VM_RAND_READ)] = "rr",
664 [ilog2(VM_DONTCOPY)] = "dc",
665 [ilog2(VM_DONTEXPAND)] = "de",
8614d6c5 666 [ilog2(VM_LOCKONFAULT)] = "lf",
834f82e2
CG		 667 [ilog2(VM_ACCOUNT)] = "ac",
668 [ilog2(VM_NORESERVE)] = "nr",
669 [ilog2(VM_HUGETLB)] = "ht",
b6fb293f 670 [ilog2(VM_SYNC)] = "sf",
834f82e2 671 [ilog2(VM_ARCH_1)] = "ar",
d2cd9ede 672 [ilog2(VM_WIPEONFORK)] = "wf",
834f82e2 673 [ilog2(VM_DONTDUMP)] = "dd",
424037b7
DK		 674#ifdef CONFIG_ARM64_BTI
675 [ilog2(VM_ARM64_BTI)] = "bt",
676#endif
ec8e41ae
NH		 677#ifdef CONFIG_MEM_SOFT_DIRTY
678 [ilog2(VM_SOFTDIRTY)] = "sd",
679#endif
834f82e2
CG		 680 [ilog2(VM_MIXEDMAP)] = "mm",
681 [ilog2(VM_HUGEPAGE)] = "hg",
682 [ilog2(VM_NOHUGEPAGE)] = "nh",
683 [ilog2(VM_MERGEABLE)] = "mg",
16ba6f81
AA		 684 [ilog2(VM_UFFD_MISSING)]= "um",
685 [ilog2(VM_UFFD_WP)] = "uw",
9f341931
CM		 686#ifdef CONFIG_ARM64_MTE
687 [ilog2(VM_MTE)] = "mt",
688 [ilog2(VM_MTE_ALLOWED)] = "",
689#endif
5212213a 690#ifdef CONFIG_ARCH_HAS_PKEYS
c1192f84
DH		 691 /* These come out via ProtectionKey: */
692 [ilog2(VM_PKEY_BIT0)] = "",
693 [ilog2(VM_PKEY_BIT1)] = "",
694 [ilog2(VM_PKEY_BIT2)] = "",
695 [ilog2(VM_PKEY_BIT3)] = "",
2c9e0a6f
RP		 696#if VM_PKEY_BIT4
697 [ilog2(VM_PKEY_BIT4)] = "",
c1192f84 698#endif
5212213a 699#endif /* CONFIG_ARCH_HAS_PKEYS */
7677f7fd
AR		 700#ifdef CONFIG_HAVE_ARCH_USERFAULTFD_MINOR
701 [ilog2(VM_UFFD_MINOR)] = "ui",
702#endif /* CONFIG_HAVE_ARCH_USERFAULTFD_MINOR */
54007f81
YY		 703#ifdef CONFIG_X86_USER_SHADOW_STACK
704 [ilog2(VM_SHADOW_STACK)] = "ss",
705#endif
834f82e2
CG		 706 };
707 size_t i;
708
709 seq_puts(m, "VmFlags: ");
710 for (i = 0; i < BITS_PER_LONG; i++) {
c1192f84
DH		 711 if (!mnemonics[i][0])
712 continue;
834f82e2 713 if (vma->vm_flags & (1UL << i)) {
f6640663
AV		 714 seq_putc(m, mnemonics[i][0]);
715 seq_putc(m, mnemonics[i][1]);
716 seq_putc(m, ' ');
834f82e2
CG		 717 }
718 }
719 seq_putc(m, '\n');
720}
721
25ee01a2
NH		 722#ifdef CONFIG_HUGETLB_PAGE
723static int smaps_hugetlb_range(pte_t *pte, unsigned long hmask,
724 unsigned long addr, unsigned long end,
725 struct mm_walk *walk)
726{
727 struct mem_size_stats *mss = walk->private;
728 struct vm_area_struct *vma = walk->vma;
729 struct page *page = NULL;
c33c7948 730 pte_t ptent = ptep_get(pte);
25ee01a2 731
c33c7948
RR		 732 if (pte_present(ptent)) {
733 page = vm_normal_page(vma, addr, ptent);
734 } else if (is_swap_pte(ptent)) {
735 swp_entry_t swpent = pte_to_swp_entry(ptent);
25ee01a2 736
af5cdaf8
AP		 737 if (is_pfn_swap_entry(swpent))
738 page = pfn_swap_entry_to_page(swpent);
25ee01a2
NH		 739 }
740 if (page) {
3489dbb6 741 if (page_mapcount(page) >= 2 || hugetlb_pmd_shared(pte))
25ee01a2
NH		 742 mss->shared_hugetlb += huge_page_size(hstate_vma(vma));
743 else
744 mss->private_hugetlb += huge_page_size(hstate_vma(vma));
745 }
746 return 0;
747}
7b86ac33
CH		 748#else
749#define smaps_hugetlb_range NULL
25ee01a2
NH		 750#endif /* HUGETLB_PAGE */
751
7b86ac33
CH		 752static const struct mm_walk_ops smaps_walk_ops = {
753 .pmd_entry = smaps_pte_range,
754 .hugetlb_entry = smaps_hugetlb_range,
49b06385 755 .walk_lock = PGWALK_RDLOCK,
7b86ac33
CH		 756};
757
758static const struct mm_walk_ops smaps_shmem_walk_ops = {
759 .pmd_entry = smaps_pte_range,
760 .hugetlb_entry = smaps_hugetlb_range,
761 .pte_hole = smaps_pte_hole,
49b06385 762 .walk_lock = PGWALK_RDLOCK,
7b86ac33
CH		 763};
764
03b4b114
CC		 765/*
766 * Gather mem stats from @vma with the indicated beginning
767 * address @start, and keep them in @mss.
768 *
769 * Use vm_start of @vma as the beginning address if @start is 0.
770 */
8e68d689 771static void smap_gather_stats(struct vm_area_struct *vma,
03b4b114 772 struct mem_size_stats *mss, unsigned long start)
e070ad49 773{
03b4b114
CC		 774 const struct mm_walk_ops *ops = &smaps_walk_ops;
775
776 /* Invalid start */
777 if (start >= vma->vm_end)
778 return;
779
c261e7d9 780 if (vma->vm_file && shmem_mapping(vma->vm_file->f_mapping)) {
6a15a370
VB		 781 /*
782 * For shared or readonly shmem mappings we know that all
783 * swapped out pages belong to the shmem object, and we can
784 * obtain the swap value much more efficiently. For private
785 * writable mappings, we might have COW pages that are
786 * not affected by the parent swapped out pages of the shmem
787 * object, so we have to distinguish them during the page walk.
788 * Unless we know that the shmem object (or the part mapped by
789 * our VMA) has no swapped out pages at all.
790 */
791 unsigned long shmem_swapped = shmem_swap_usage(vma);
792
03b4b114
CC		 793 if (!start && (!shmem_swapped || (vma->vm_flags & VM_SHARED) ||
794 !(vma->vm_flags & VM_WRITE))) {
fa76da46 795 mss->swap += shmem_swapped;
6a15a370 796 } else {
03b4b114 797 ops = &smaps_shmem_walk_ops;
6a15a370 798 }
c261e7d9 799 }
b4aca547 800
c1e8d7c6 801 /* mmap_lock is held in m_start */
03b4b114
CC		 802 if (!start)
803 walk_page_vma(vma, ops, mss);
804 else
805 walk_page_range(vma->vm_mm, start, vma->vm_end, ops, mss);
8e68d689
VB		 806}
807
808#define SEQ_PUT_DEC(str, val) \
809 seq_put_decimal_ull_width(m, str, (val) >> 10, 8)
f1547959
VB		 810
811/* Show the contents common for smaps and smaps_rollup */
ee2ad71b
LS		 812static void __show_smap(struct seq_file *m, const struct mem_size_stats *mss,
813 bool rollup_mode)
f1547959
VB		 814{
815 SEQ_PUT_DEC("Rss: ", mss->resident);
816 SEQ_PUT_DEC(" kB\nPss: ", mss->pss >> PSS_SHIFT);
30934843 817 SEQ_PUT_DEC(" kB\nPss_Dirty: ", mss->pss_dirty >> PSS_SHIFT);
ee2ad71b
LS		 818 if (rollup_mode) {
819 /*
820 * These are meaningful only for smaps_rollup, otherwise two of
821 * them are zero, and the other one is the same as Pss.
822 */
823 SEQ_PUT_DEC(" kB\nPss_Anon: ",
824 mss->pss_anon >> PSS_SHIFT);
825 SEQ_PUT_DEC(" kB\nPss_File: ",
826 mss->pss_file >> PSS_SHIFT);
827 SEQ_PUT_DEC(" kB\nPss_Shmem: ",
828 mss->pss_shmem >> PSS_SHIFT);
829 }
f1547959
VB		 830 SEQ_PUT_DEC(" kB\nShared_Clean: ", mss->shared_clean);
831 SEQ_PUT_DEC(" kB\nShared_Dirty: ", mss->shared_dirty);
832 SEQ_PUT_DEC(" kB\nPrivate_Clean: ", mss->private_clean);
833 SEQ_PUT_DEC(" kB\nPrivate_Dirty: ", mss->private_dirty);
834 SEQ_PUT_DEC(" kB\nReferenced: ", mss->referenced);
835 SEQ_PUT_DEC(" kB\nAnonymous: ", mss->anonymous);
8b479335 836 SEQ_PUT_DEC(" kB\nKSM: ", mss->ksm);
f1547959
VB		 837 SEQ_PUT_DEC(" kB\nLazyFree: ", mss->lazyfree);
838 SEQ_PUT_DEC(" kB\nAnonHugePages: ", mss->anonymous_thp);
839 SEQ_PUT_DEC(" kB\nShmemPmdMapped: ", mss->shmem_thp);
471e78cc 840 SEQ_PUT_DEC(" kB\nFilePmdMapped: ", mss->file_thp);
f1547959
VB		 841 SEQ_PUT_DEC(" kB\nShared_Hugetlb: ", mss->shared_hugetlb);
842 seq_put_decimal_ull_width(m, " kB\nPrivate_Hugetlb: ",
843 mss->private_hugetlb >> 10, 7);
844 SEQ_PUT_DEC(" kB\nSwap: ", mss->swap);
845 SEQ_PUT_DEC(" kB\nSwapPss: ",
846 mss->swap_pss >> PSS_SHIFT);
847 SEQ_PUT_DEC(" kB\nLocked: ",
848 mss->pss_locked >> PSS_SHIFT);
849 seq_puts(m, " kB\n");
850}
851
8e68d689
VB		 852static int show_smap(struct seq_file *m, void *v)
853{
8e68d689 854 struct vm_area_struct *vma = v;
860a2e7f 855 struct mem_size_stats mss = {};
258f669e 856
03b4b114 857 smap_gather_stats(vma, &mss, 0);
258f669e
VB		 858
859 show_map_vma(m, vma);
860
861 SEQ_PUT_DEC("Size: ", vma->vm_end - vma->vm_start);
862 SEQ_PUT_DEC(" kB\nKernelPageSize: ", vma_kernel_pagesize(vma));
863 SEQ_PUT_DEC(" kB\nMMUPageSize: ", vma_mmu_pagesize(vma));
864 seq_puts(m, " kB\n");
865
ee2ad71b 866 __show_smap(m, &mss, false);
258f669e 867
daa60ae6 868 seq_printf(m, "THPeligible: %8u\n",
3485b883
RR		 869 !!thp_vma_allowable_orders(vma, vma->vm_flags, true, false,
870 true, THP_ORDERS_ALL));
7635d9cb 871
258f669e
VB		 872 if (arch_pkeys_enabled())
873 seq_printf(m, "ProtectionKey: %8u\n", vma_pkey(vma));
874 show_smap_vma_flags(m, vma);
875
258f669e
VB		 876 return 0;
877}
878
879static int show_smaps_rollup(struct seq_file *m, void *v)
880{
881 struct proc_maps_private *priv = m->private;
860a2e7f 882 struct mem_size_stats mss = {};
c4c84f06 883 struct mm_struct *mm = priv->mm;
258f669e 884 struct vm_area_struct *vma;
c4c84f06 885 unsigned long vma_start = 0, last_vma_end = 0;
8e68d689 886 int ret = 0;
250cb40f 887 VMA_ITERATOR(vmi, mm, 0);
8e68d689 888
258f669e
VB		 889 priv->task = get_proc_task(priv->inode);
890 if (!priv->task)
891 return -ESRCH;
493b0e9d 892
258f669e
VB		 893 if (!mm || !mmget_not_zero(mm)) {
894 ret = -ESRCH;
895 goto out_put_task;
493b0e9d 896 }
4752c369 897
d8ed45c5 898 ret = mmap_read_lock_killable(mm);
a26a9781
KK		 899 if (ret)
900 goto out_put_mm;
901
258f669e 902 hold_task_mempolicy(priv);
250cb40f 903 vma = vma_next(&vmi);
f1547959 904
c4c84f06
MWO		 905 if (unlikely(!vma))
906 goto empty_set;
907
908 vma_start = vma->vm_start;
909 do {
03b4b114 910 smap_gather_stats(vma, &mss, 0);
258f669e 911 last_vma_end = vma->vm_end;
ff9f47f6
CC		 912
913 /*
914 * Release mmap_lock temporarily if someone wants to
915 * access it for write request.
916 */
917 if (mmap_lock_is_contended(mm)) {
250cb40f 918 vma_iter_invalidate(&vmi);
ff9f47f6
CC		 919 mmap_read_unlock(mm);
920 ret = mmap_read_lock_killable(mm);
921 if (ret) {
922 release_task_mempolicy(priv);
923 goto out_put_mm;
924 }
925
926 /*
927 * After dropping the lock, there are four cases to
928 * consider. See the following example for explanation.
929 *
930 * +------+------+-----------+
931 * | VMA1 | VMA2 | VMA3 |
932 * +------+------+-----------+
933 * | | | |
934 * 4k 8k 16k 400k
935 *
936 * Suppose we drop the lock after reading VMA2 due to
937 * contention, then we get:
938 *
939 * last_vma_end = 16k
940 *
941 * 1) VMA2 is freed, but VMA3 exists:
942 *
250cb40f 943 * vma_next(vmi) will return VMA3.
ff9f47f6
CC		 944 * In this case, just continue from VMA3.
945 *
946 * 2) VMA2 still exists:
947 *
250cb40f
LH		 948 * vma_next(vmi) will return VMA3.
949 * In this case, just continue from VMA3.
ff9f47f6
CC		 950 *
951 * 3) No more VMAs can be found:
952 *
250cb40f 953 * vma_next(vmi) will return NULL.
ff9f47f6
CC		 954 * No more things to do, just break.
955 *
956 * 4) (last_vma_end - 1) is the middle of a vma (VMA'):
957 *
250cb40f 958 * vma_next(vmi) will return VMA' whose range
ff9f47f6
CC		 959 * contains last_vma_end.
960 * Iterate VMA' from last_vma_end.
961 */
250cb40f 962 vma = vma_next(&vmi);
ff9f47f6
CC		 963 /* Case 3 above */
964 if (!vma)
965 break;
966
250cb40f 967 /* Case 1 and 2 above */
ff9f47f6
CC		 968 if (vma->vm_start >= last_vma_end)
969 continue;
970
971 /* Case 4 above */
972 if (vma->vm_end > last_vma_end)
973 smap_gather_stats(vma, &mss, last_vma_end);
974 }
250cb40f 975 } for_each_vma(vmi, vma);
258f669e 976
c4c84f06
MWO		 977empty_set:
978 show_vma_header_prefix(m, vma_start, last_vma_end, 0, 0, 0, 0);
258f669e
VB		 979 seq_pad(m, ' ');
980 seq_puts(m, "[rollup]\n");
981
ee2ad71b 982 __show_smap(m, &mss, true);
258f669e
VB		 983
984 release_task_mempolicy(priv);
d8ed45c5 985 mmap_read_unlock(mm);
258f669e 986
a26a9781
KK		 987out_put_mm:
988 mmput(mm);
258f669e
VB		 989out_put_task:
990 put_task_struct(priv->task);
991 priv->task = NULL;
992
493b0e9d 993 return ret;
e070ad49 994}
d1be35cb 995#undef SEQ_PUT_DEC
e070ad49 996
03a44825 997static const struct seq_operations proc_pid_smaps_op = {
a6198797
MM		 998 .start = m_start,
999 .next = m_next,
1000 .stop = m_stop,
871305bb 1001 .show = show_smap
a6198797
MM		 1002};
1003
b7643757 1004static int pid_smaps_open(struct inode *inode, struct file *file)
a6198797
MM		 1005{
1006 return do_maps_open(inode, file, &proc_pid_smaps_op);
1007}
1008
258f669e 1009static int smaps_rollup_open(struct inode *inode, struct file *file)
493b0e9d 1010{
258f669e 1011 int ret;
493b0e9d 1012 struct proc_maps_private *priv;
258f669e
VB		 1013
1014 priv = kzalloc(sizeof(*priv), GFP_KERNEL_ACCOUNT);
1015 if (!priv)
493b0e9d 1016 return -ENOMEM;
258f669e
VB		 1017
1018 ret = single_open(file, show_smaps_rollup, priv);
1019 if (ret)
1020 goto out_free;
1021
1022 priv->inode = inode;
1023 priv->mm = proc_mem_open(inode, PTRACE_MODE_READ);
1024 if (IS_ERR(priv->mm)) {
1025 ret = PTR_ERR(priv->mm);
1026
1027 single_release(inode, file);
1028 goto out_free;
493b0e9d 1029 }
258f669e 1030
493b0e9d 1031 return 0;
258f669e
VB		 1032
1033out_free:
1034 kfree(priv);
1035 return ret;
1036}
1037
1038static int smaps_rollup_release(struct inode *inode, struct file *file)
1039{
1040 struct seq_file *seq = file->private_data;
1041 struct proc_maps_private *priv = seq->private;
1042
1043 if (priv->mm)
1044 mmdrop(priv->mm);
1045
1046 kfree(priv);
1047 return single_release(inode, file);
493b0e9d
DC		 1048}
1049
b7643757
SP		 1050const struct file_operations proc_pid_smaps_operations = {
1051 .open = pid_smaps_open,
1052 .read = seq_read,
1053 .llseek = seq_lseek,
29a40ace 1054 .release = proc_map_release,
b7643757
SP		 1055};
1056
493b0e9d 1057const struct file_operations proc_pid_smaps_rollup_operations = {
258f669e 1058 .open = smaps_rollup_open,
493b0e9d
DC		 1059 .read = seq_read,
1060 .llseek = seq_lseek,
258f669e 1061 .release = smaps_rollup_release,
493b0e9d
DC		 1062};
1063
040fa020
PE		 1064enum clear_refs_types {
1065 CLEAR_REFS_ALL = 1,
1066 CLEAR_REFS_ANON,
1067 CLEAR_REFS_MAPPED,
0f8975ec 1068 CLEAR_REFS_SOFT_DIRTY,
695f0559 1069 CLEAR_REFS_MM_HIWATER_RSS,
040fa020
PE		 1070 CLEAR_REFS_LAST,
1071};
1072
af9de7eb 1073struct clear_refs_private {
0f8975ec 1074 enum clear_refs_types type;
af9de7eb
PE		 1075};
1076
7d5b3bfa 1077#ifdef CONFIG_MEM_SOFT_DIRTY
9348b73c 1078
9348b73c
LT		 1079static inline bool pte_is_pinned(struct vm_area_struct *vma, unsigned long addr, pte_t pte)
1080{
1081 struct page *page;
1082
1083 if (!pte_write(pte))
1084 return false;
1085 if (!is_cow_mapping(vma->vm_flags))
1086 return false;
a458b76a 1087 if (likely(!test_bit(MMF_HAS_PINNED, &vma->vm_mm->flags)))
9348b73c
LT		 1088 return false;
1089 page = vm_normal_page(vma, addr, pte);
1090 if (!page)
1091 return false;
1092 return page_maybe_dma_pinned(page);
1093}
1094
0f8975ec
PE		 1095static inline void clear_soft_dirty(struct vm_area_struct *vma,
1096 unsigned long addr, pte_t *pte)
1097{
0f8975ec
PE		 1098 /*
1099 * The soft-dirty tracker uses #PF-s to catch writes
1100 * to pages, so write-protect the pte as well. See the
1ad1335d 1101 * Documentation/admin-guide/mm/soft-dirty.rst for full description
0f8975ec
PE		 1102 * of how soft-dirty works.
1103 */
c33c7948 1104 pte_t ptent = ptep_get(pte);
179ef71c
CG		 1105
1106 if (pte_present(ptent)) {
04a86453
AK		 1107 pte_t old_pte;
1108
9348b73c
LT		 1109 if (pte_is_pinned(vma, addr, ptent))
1110 return;
04a86453
AK		 1111 old_pte = ptep_modify_prot_start(vma, addr, pte);
1112 ptent = pte_wrprotect(old_pte);
a7b76174 1113 ptent = pte_clear_soft_dirty(ptent);
04a86453 1114 ptep_modify_prot_commit(vma, addr, pte, old_pte, ptent);
179ef71c
CG		 1115 } else if (is_swap_pte(ptent)) {
1116 ptent = pte_swp_clear_soft_dirty(ptent);
326c2597 1117 set_pte_at(vma->vm_mm, addr, pte, ptent);
179ef71c 1118 }
0f8975ec 1119}
5d3875a0
LD		 1120#else
1121static inline void clear_soft_dirty(struct vm_area_struct *vma,
1122 unsigned long addr, pte_t *pte)
1123{
1124}
1125#endif
0f8975ec 1126
5d3875a0 1127#if defined(CONFIG_MEM_SOFT_DIRTY) && defined(CONFIG_TRANSPARENT_HUGEPAGE)
7d5b3bfa
KS		 1128static inline void clear_soft_dirty_pmd(struct vm_area_struct *vma,
1129 unsigned long addr, pmd_t *pmdp)
1130{
a3cf988f 1131 pmd_t old, pmd = *pmdp;
5b7abeae 1132
ab6e3d09
NH		 1133 if (pmd_present(pmd)) {
1134 /* See comment in change_huge_pmd() */
a3cf988f
KS		 1135 old = pmdp_invalidate(vma, addr, pmdp);
1136 if (pmd_dirty(old))
ab6e3d09 1137 pmd = pmd_mkdirty(pmd);
a3cf988f 1138 if (pmd_young(old))
ab6e3d09
NH		 1139 pmd = pmd_mkyoung(pmd);
1140
1141 pmd = pmd_wrprotect(pmd);
1142 pmd = pmd_clear_soft_dirty(pmd);
1143
1144 set_pmd_at(vma->vm_mm, addr, pmdp, pmd);
1145 } else if (is_migration_entry(pmd_to_swp_entry(pmd))) {
1146 pmd = pmd_swp_clear_soft_dirty(pmd);
1147 set_pmd_at(vma->vm_mm, addr, pmdp, pmd);
1148 }
7d5b3bfa 1149}
7d5b3bfa 1150#else
7d5b3bfa
KS		 1151static inline void clear_soft_dirty_pmd(struct vm_area_struct *vma,
1152 unsigned long addr, pmd_t *pmdp)
1153{
1154}
1155#endif
1156
a6198797 1157static int clear_refs_pte_range(pmd_t *pmd, unsigned long addr,
2165009b 1158 unsigned long end, struct mm_walk *walk)
a6198797 1159{
af9de7eb 1160 struct clear_refs_private *cp = walk->private;
5c64f52a 1161 struct vm_area_struct *vma = walk->vma;
a6198797
MM		 1162 pte_t *pte, ptent;
1163 spinlock_t *ptl;
1164 struct page *page;
1165
b6ec57f4
KS		 1166 ptl = pmd_trans_huge_lock(pmd, vma);
1167 if (ptl) {
7d5b3bfa
KS		 1168 if (cp->type == CLEAR_REFS_SOFT_DIRTY) {
1169 clear_soft_dirty_pmd(vma, addr, pmd);
1170 goto out;
1171 }
1172
84c3fc4e
ZY		 1173 if (!pmd_present(*pmd))
1174 goto out;
1175
7d5b3bfa
KS		 1176 page = pmd_page(*pmd);
1177
1178 /* Clear accessed and referenced bits. */
1179 pmdp_test_and_clear_young(vma, addr, pmd);
33c3fc71 1180 test_and_clear_page_young(page);
7d5b3bfa
KS		 1181 ClearPageReferenced(page);
1182out:
1183 spin_unlock(ptl);
1184 return 0;
1185 }
1186
a6198797 1187 pte = pte_offset_map_lock(vma->vm_mm, pmd, addr, &ptl);
7780d040
HD		 1188 if (!pte) {
1189 walk->action = ACTION_AGAIN;
1190 return 0;
1191 }
a6198797 1192 for (; addr != end; pte++, addr += PAGE_SIZE) {
c33c7948 1193 ptent = ptep_get(pte);
a6198797 1194
0f8975ec
PE		 1195 if (cp->type == CLEAR_REFS_SOFT_DIRTY) {
1196 clear_soft_dirty(vma, addr, pte);
1197 continue;
1198 }
1199
179ef71c
CG		 1200 if (!pte_present(ptent))
1201 continue;
1202
a6198797
MM		 1203 page = vm_normal_page(vma, addr, ptent);
1204 if (!page)
1205 continue;
1206
1207 /* Clear accessed and referenced bits. */
1208 ptep_test_and_clear_young(vma, addr, pte);
33c3fc71 1209 test_and_clear_page_young(page);
a6198797
MM		 1210 ClearPageReferenced(page);
1211 }
1212 pte_unmap_unlock(pte - 1, ptl);
1213 cond_resched();
1214 return 0;
1215}
1216
5c64f52a
NH		 1217static int clear_refs_test_walk(unsigned long start, unsigned long end,
1218 struct mm_walk *walk)
1219{
1220 struct clear_refs_private *cp = walk->private;
1221 struct vm_area_struct *vma = walk->vma;
1222
48684a65
NH		 1223 if (vma->vm_flags & VM_PFNMAP)
1224 return 1;
1225
5c64f52a
NH		 1226 /*
1227 * Writing 1 to /proc/pid/clear_refs affects all pages.
1228 * Writing 2 to /proc/pid/clear_refs only affects anonymous pages.
1229 * Writing 3 to /proc/pid/clear_refs only affects file mapped pages.
1230 * Writing 4 to /proc/pid/clear_refs affects all pages.
1231 */
1232 if (cp->type == CLEAR_REFS_ANON && vma->vm_file)
1233 return 1;
1234 if (cp->type == CLEAR_REFS_MAPPED && !vma->vm_file)
1235 return 1;
1236 return 0;
1237}
1238
7b86ac33
CH		 1239static const struct mm_walk_ops clear_refs_walk_ops = {
1240 .pmd_entry = clear_refs_pte_range,
1241 .test_walk = clear_refs_test_walk,
49b06385 1242 .walk_lock = PGWALK_WRLOCK,
7b86ac33
CH		 1243};
1244
f248dcb3
MM		 1245static ssize_t clear_refs_write(struct file *file, const char __user *buf,
1246 size_t count, loff_t *ppos)
b813e931 1247{
f248dcb3 1248 struct task_struct *task;
860a2e7f 1249 char buffer[PROC_NUMBUF] = {};
f248dcb3 1250 struct mm_struct *mm;
b813e931 1251 struct vm_area_struct *vma;
040fa020
PE		 1252 enum clear_refs_types type;
1253 int itype;
0a8cb8e3 1254 int rv;
b813e931 1255
f248dcb3
MM		 1256 if (count > sizeof(buffer) - 1)
1257 count = sizeof(buffer) - 1;
1258 if (copy_from_user(buffer, buf, count))
1259 return -EFAULT;
040fa020 1260 rv = kstrtoint(strstrip(buffer), 10, &itype);
0a8cb8e3
AD		 1261 if (rv < 0)
1262 return rv;
040fa020
PE		 1263 type = (enum clear_refs_types)itype;
1264 if (type < CLEAR_REFS_ALL || type >= CLEAR_REFS_LAST)
f248dcb3 1265 return -EINVAL;
541c237c 1266
496ad9aa 1267 task = get_proc_task(file_inode(file));
f248dcb3
MM		 1268 if (!task)
1269 return -ESRCH;
1270 mm = get_task_mm(task);
1271 if (mm) {
250cb40f 1272 VMA_ITERATOR(vmi, mm, 0);
ac46d4f3 1273 struct mmu_notifier_range range;
af9de7eb 1274 struct clear_refs_private cp = {
0f8975ec 1275 .type = type,
af9de7eb 1276 };
695f0559 1277
29a951df
LT		 1278 if (mmap_write_lock_killable(mm)) {
1279 count = -EINTR;
1280 goto out_mm;
1281 }
695f0559
PC		 1282 if (type == CLEAR_REFS_MM_HIWATER_RSS) {
1283 /*
1284 * Writing 5 to /proc/pid/clear_refs resets the peak
1285 * resident set size to this mm's current rss value.
1286 */
695f0559 1287 reset_mm_hiwater_rss(mm);
29a951df 1288 goto out_unlock;
695f0559
PC		 1289 }
1290
64e45507 1291 if (type == CLEAR_REFS_SOFT_DIRTY) {
250cb40f 1292 for_each_vma(vmi, vma) {
64e45507
PF		 1293 if (!(vma->vm_flags & VM_SOFTDIRTY))
1294 continue;
1c71222e 1295 vm_flags_clear(vma, VM_SOFTDIRTY);
29a951df 1296 vma_set_page_prot(vma);
64e45507 1297 }
ac46d4f3 1298
912efa17 1299 inc_tlb_flush_pending(mm);
7269f999 1300 mmu_notifier_range_init(&range, MMU_NOTIFY_SOFT_DIRTY,
7d4a8be0 1301 0, mm, 0, -1UL);
ac46d4f3 1302 mmu_notifier_invalidate_range_start(&range);
64e45507 1303 }
c4c84f06 1304 walk_page_range(mm, 0, -1, &clear_refs_walk_ops, &cp);
912efa17 1305 if (type == CLEAR_REFS_SOFT_DIRTY) {
ac46d4f3 1306 mmu_notifier_invalidate_range_end(&range);
912efa17
WD		 1307 flush_tlb_mm(mm);
1308 dec_tlb_flush_pending(mm);
1309 }
29a951df
LT		 1310out_unlock:
1311 mmap_write_unlock(mm);
695f0559 1312out_mm:
f248dcb3
MM		 1313 mmput(mm);
1314 }
1315 put_task_struct(task);
fb92a4b0
VL		 1316
1317 return count;
b813e931
DR		 1318}
1319
f248dcb3
MM		 1320const struct file_operations proc_clear_refs_operations = {
1321 .write = clear_refs_write,
6038f373 1322 .llseek = noop_llseek,
f248dcb3
MM		 1323};
1324
092b50ba
NH		 1325typedef struct {
1326 u64 pme;
1327} pagemap_entry_t;
1328
85863e47 1329struct pagemapread {
8c829622 1330 int pos, len; /* units: PM_ENTRY_BYTES, not bytes */
092b50ba 1331 pagemap_entry_t *buffer;
1c90308e 1332 bool show_pfn;
85863e47
MM		 1333};
1334
5aaabe83
NH		 1335#define PAGEMAP_WALK_SIZE (PMD_SIZE)
1336#define PAGEMAP_WALK_MASK (PMD_MASK)
1337
deb94544
KK		 1338#define PM_ENTRY_BYTES sizeof(pagemap_entry_t)
1339#define PM_PFRAME_BITS 55
1340#define PM_PFRAME_MASK GENMASK_ULL(PM_PFRAME_BITS - 1, 0)
1341#define PM_SOFT_DIRTY BIT_ULL(55)
77bb499b 1342#define PM_MMAP_EXCLUSIVE BIT_ULL(56)
fb8e37f3 1343#define PM_UFFD_WP BIT_ULL(57)
deb94544
KK		 1344#define PM_FILE BIT_ULL(61)
1345#define PM_SWAP BIT_ULL(62)
1346#define PM_PRESENT BIT_ULL(63)
1347
85863e47
MM		 1348#define PM_END_OF_BUFFER 1
1349
deb94544 1350static inline pagemap_entry_t make_pme(u64 frame, u64 flags)
092b50ba 1351{
deb94544 1352 return (pagemap_entry_t) { .pme = (frame & PM_PFRAME_MASK) | flags };
092b50ba
NH		 1353}
1354
1355static int add_to_pagemap(unsigned long addr, pagemap_entry_t *pme,
85863e47
MM		 1356 struct pagemapread *pm)
1357{
092b50ba 1358 pm->buffer[pm->pos++] = *pme;
d82ef020 1359 if (pm->pos >= pm->len)
aae8679b 1360 return PM_END_OF_BUFFER;
85863e47
MM		 1361 return 0;
1362}
1363
1364static int pagemap_pte_hole(unsigned long start, unsigned long end,
b7a16c7a 1365 __always_unused int depth, struct mm_walk *walk)
85863e47 1366{
2165009b 1367 struct pagemapread *pm = walk->private;
68b5a652 1368 unsigned long addr = start;
85863e47 1369 int err = 0;
092b50ba 1370
68b5a652
PF		 1371 while (addr < end) {
1372 struct vm_area_struct *vma = find_vma(walk->mm, addr);
deb94544 1373 pagemap_entry_t pme = make_pme(0, 0);
87e6d49a
PF		 1374 /* End of address space hole, which we mark as non-present. */
1375 unsigned long hole_end;
68b5a652 1376
87e6d49a
PF		 1377 if (vma)
1378 hole_end = min(end, vma->vm_start);
1379 else
1380 hole_end = end;
1381
1382 for (; addr < hole_end; addr += PAGE_SIZE) {
1383 err = add_to_pagemap(addr, &pme, pm);
1384 if (err)
1385 goto out;
68b5a652
PF		 1386 }
1387
87e6d49a
PF		 1388 if (!vma)
1389 break;
1390
1391 /* Addresses in the VMA. */
1392 if (vma->vm_flags & VM_SOFTDIRTY)
deb94544 1393 pme = make_pme(0, PM_SOFT_DIRTY);
87e6d49a 1394 for (; addr < min(end, vma->vm_end); addr += PAGE_SIZE) {
68b5a652
PF		 1395 err = add_to_pagemap(addr, &pme, pm);
1396 if (err)
1397 goto out;
1398 }
85863e47 1399 }
68b5a652 1400out:
85863e47
MM		 1401 return err;
1402}
1403
deb94544 1404static pagemap_entry_t pte_to_pagemap_entry(struct pagemapread *pm,
052fb0d6 1405 struct vm_area_struct *vma, unsigned long addr, pte_t pte)
85863e47 1406{
deb94544 1407 u64 frame = 0, flags = 0;
052fb0d6 1408 struct page *page = NULL;
24d7275c 1409 bool migration = false;
85863e47 1410
052fb0d6 1411 if (pte_present(pte)) {
1c90308e
KK		 1412 if (pm->show_pfn)
1413 frame = pte_pfn(pte);
deb94544 1414 flags |= PM_PRESENT;
25b2995a 1415 page = vm_normal_page(vma, addr, pte);
e9cdd6e7 1416 if (pte_soft_dirty(pte))
deb94544 1417 flags |= PM_SOFT_DIRTY;
fb8e37f3
PX		 1418 if (pte_uffd_wp(pte))
1419 flags |= PM_UFFD_WP;
052fb0d6 1420 } else if (is_swap_pte(pte)) {
179ef71c
CG		 1421 swp_entry_t entry;
1422 if (pte_swp_soft_dirty(pte))
deb94544 1423 flags |= PM_SOFT_DIRTY;
fb8e37f3
PX		 1424 if (pte_swp_uffd_wp(pte))
1425 flags |= PM_UFFD_WP;
179ef71c 1426 entry = pte_to_swp_entry(pte);
0d206b5d
PX		 1427 if (pm->show_pfn) {
1428 pgoff_t offset;
1429 /*
1430 * For PFN swap offsets, keeping the offset field
1431 * to be PFN only to be compatible with old smaps.
1432 */
1433 if (is_pfn_swap_entry(entry))
1434 offset = swp_offset_pfn(entry);
1435 else
1436 offset = swp_offset(entry);
ab6ecf24 1437 frame = swp_type(entry) |
0d206b5d
PX		 1438 (offset << MAX_SWAPFILES_SHIFT);
1439 }
deb94544 1440 flags |= PM_SWAP;
24d7275c 1441 migration = is_migration_entry(entry);
af5cdaf8
AP		 1442 if (is_pfn_swap_entry(entry))
1443 page = pfn_swap_entry_to_page(entry);
8e165e73
PX		 1444 if (pte_marker_entry_uffd_wp(entry))
1445 flags |= PM_UFFD_WP;
052fb0d6
KK		 1446 }
1447
1448 if (page && !PageAnon(page))
1449 flags |= PM_FILE;
24d7275c 1450 if (page && !migration && page_mapcount(page) == 1)
77bb499b 1451 flags |= PM_MMAP_EXCLUSIVE;
deb94544
KK		 1452 if (vma->vm_flags & VM_SOFTDIRTY)
1453 flags |= PM_SOFT_DIRTY;
052fb0d6 1454
deb94544 1455 return make_pme(frame, flags);
bcf8039e
DH		 1456}
1457
356515e7 1458static int pagemap_pmd_range(pmd_t *pmdp, unsigned long addr, unsigned long end,
2165009b 1459 struct mm_walk *walk)
85863e47 1460{
f995ece2 1461 struct vm_area_struct *vma = walk->vma;
2165009b 1462 struct pagemapread *pm = walk->private;
bf929152 1463 spinlock_t *ptl;
05fbf357 1464 pte_t *pte, *orig_pte;
85863e47 1465 int err = 0;
356515e7 1466#ifdef CONFIG_TRANSPARENT_HUGEPAGE
24d7275c
YS		 1467 bool migration = false;
1468
b6ec57f4
KS		 1469 ptl = pmd_trans_huge_lock(pmdp, vma);
1470 if (ptl) {
356515e7
KK		 1471 u64 flags = 0, frame = 0;
1472 pmd_t pmd = *pmdp;
84c3fc4e 1473 struct page *page = NULL;
0f8975ec 1474
b83d7e43 1475 if (vma->vm_flags & VM_SOFTDIRTY)
deb94544 1476 flags |= PM_SOFT_DIRTY;
d9104d1c 1477
356515e7 1478 if (pmd_present(pmd)) {
84c3fc4e 1479 page = pmd_page(pmd);
77bb499b 1480
356515e7 1481 flags |= PM_PRESENT;
b83d7e43
HY		 1482 if (pmd_soft_dirty(pmd))
1483 flags |= PM_SOFT_DIRTY;
fb8e37f3
PX		 1484 if (pmd_uffd_wp(pmd))
1485 flags |= PM_UFFD_WP;
1c90308e
KK		 1486 if (pm->show_pfn)
1487 frame = pmd_pfn(pmd) +
1488 ((addr & ~PMD_MASK) >> PAGE_SHIFT);
356515e7 1489 }
84c3fc4e
ZY		 1490#ifdef CONFIG_ARCH_ENABLE_THP_MIGRATION
1491 else if (is_swap_pmd(pmd)) {
1492 swp_entry_t entry = pmd_to_swp_entry(pmd);
ab6ecf24 1493 unsigned long offset;
84c3fc4e 1494
ab6ecf24 1495 if (pm->show_pfn) {
0d206b5d
PX		 1496 if (is_pfn_swap_entry(entry))
1497 offset = swp_offset_pfn(entry);
1498 else
1499 offset = swp_offset(entry);
1500 offset = offset +
ab6ecf24
HY		 1501 ((addr & ~PMD_MASK) >> PAGE_SHIFT);
1502 frame = swp_type(entry) |
1503 (offset << MAX_SWAPFILES_SHIFT);
1504 }
84c3fc4e 1505 flags |= PM_SWAP;
b83d7e43
HY		 1506 if (pmd_swp_soft_dirty(pmd))
1507 flags |= PM_SOFT_DIRTY;
fb8e37f3
PX		 1508 if (pmd_swp_uffd_wp(pmd))
1509 flags |= PM_UFFD_WP;
84c3fc4e 1510 VM_BUG_ON(!is_pmd_migration_entry(pmd));
24d7275c 1511 migration = is_migration_entry(entry);
af5cdaf8 1512 page = pfn_swap_entry_to_page(entry);
84c3fc4e
ZY		 1513 }
1514#endif
1515
24d7275c 1516 if (page && !migration && page_mapcount(page) == 1)
84c3fc4e 1517 flags |= PM_MMAP_EXCLUSIVE;
356515e7 1518
025c5b24 1519 for (; addr != end; addr += PAGE_SIZE) {
356515e7 1520 pagemap_entry_t pme = make_pme(frame, flags);
025c5b24 1521
092b50ba 1522 err = add_to_pagemap(addr, &pme, pm);
025c5b24
NH		 1523 if (err)
1524 break;
ab6ecf24
HY		 1525 if (pm->show_pfn) {
1526 if (flags & PM_PRESENT)
1527 frame++;
1528 else if (flags & PM_SWAP)
1529 frame += (1 << MAX_SWAPFILES_SHIFT);
1530 }
5aaabe83 1531 }
bf929152 1532 spin_unlock(ptl);
025c5b24 1533 return err;
5aaabe83 1534 }
356515e7 1535#endif /* CONFIG_TRANSPARENT_HUGEPAGE */
81d0fa62 1536
f995ece2
NH		 1537 /*
1538 * We can assume that @vma always points to a valid one and @end never
1539 * goes beyond vma->vm_end.
1540 */
356515e7 1541 orig_pte = pte = pte_offset_map_lock(walk->mm, pmdp, addr, &ptl);
7780d040
HD		 1542 if (!pte) {
1543 walk->action = ACTION_AGAIN;
1544 return err;
1545 }
f995ece2
NH		 1546 for (; addr < end; pte++, addr += PAGE_SIZE) {
1547 pagemap_entry_t pme;
05fbf357 1548
c33c7948 1549 pme = pte_to_pagemap_entry(pm, vma, addr, ptep_get(pte));
f995ece2 1550 err = add_to_pagemap(addr, &pme, pm);
05fbf357 1551 if (err)
81d0fa62 1552 break;
85863e47 1553 }
f995ece2 1554 pte_unmap_unlock(orig_pte, ptl);
85863e47
MM		 1555
1556 cond_resched();
1557
1558 return err;
1559}
1560
1a5cb814 1561#ifdef CONFIG_HUGETLB_PAGE
116354d1 1562/* This function walks within one hugetlb entry in the single call */
356515e7 1563static int pagemap_hugetlb_range(pte_t *ptep, unsigned long hmask,
116354d1
NH		 1564 unsigned long addr, unsigned long end,
1565 struct mm_walk *walk)
5dc37642 1566{
5dc37642 1567 struct pagemapread *pm = walk->private;
f995ece2 1568 struct vm_area_struct *vma = walk->vma;
356515e7 1569 u64 flags = 0, frame = 0;
5dc37642 1570 int err = 0;
356515e7 1571 pte_t pte;
5dc37642 1572
f995ece2 1573 if (vma->vm_flags & VM_SOFTDIRTY)
deb94544 1574 flags |= PM_SOFT_DIRTY;
d9104d1c 1575
356515e7
KK		 1576 pte = huge_ptep_get(ptep);
1577 if (pte_present(pte)) {
1578 struct page *page = pte_page(pte);
1579
1580 if (!PageAnon(page))
1581 flags |= PM_FILE;
1582
77bb499b
KK		 1583 if (page_mapcount(page) == 1)
1584 flags |= PM_MMAP_EXCLUSIVE;
1585
8e165e73
PX		 1586 if (huge_pte_uffd_wp(pte))
1587 flags |= PM_UFFD_WP;
1588
356515e7 1589 flags |= PM_PRESENT;
1c90308e
KK		 1590 if (pm->show_pfn)
1591 frame = pte_pfn(pte) +
1592 ((addr & ~hmask) >> PAGE_SHIFT);
8e165e73
PX		 1593 } else if (pte_swp_uffd_wp_any(pte)) {
1594 flags |= PM_UFFD_WP;
356515e7
KK		 1595 }
1596
5dc37642 1597 for (; addr != end; addr += PAGE_SIZE) {
356515e7
KK		 1598 pagemap_entry_t pme = make_pme(frame, flags);
1599
092b50ba 1600 err = add_to_pagemap(addr, &pme, pm);
5dc37642
NH		 1601 if (err)
1602 return err;
1c90308e 1603 if (pm->show_pfn && (flags & PM_PRESENT))
356515e7 1604 frame++;
5dc37642
NH		 1605 }
1606
1607 cond_resched();
1608
1609 return err;
1610}
7b86ac33
CH		 1611#else
1612#define pagemap_hugetlb_range NULL
1a5cb814 1613#endif /* HUGETLB_PAGE */
5dc37642 1614
7b86ac33
CH		 1615static const struct mm_walk_ops pagemap_ops = {
1616 .pmd_entry = pagemap_pmd_range,
1617 .pte_hole = pagemap_pte_hole,
1618 .hugetlb_entry = pagemap_hugetlb_range,
49b06385 1619 .walk_lock = PGWALK_RDLOCK,
7b86ac33
CH		 1620};
1621
85863e47
MM		 1622/*
1623 * /proc/pid/pagemap - an array mapping virtual pages to pfns
1624 *
f16278c6
HR		 1625 * For each page in the address space, this file contains one 64-bit entry
1626 * consisting of the following:
1627 *
052fb0d6 1628 * Bits 0-54 page frame number (PFN) if present
f16278c6 1629 * Bits 0-4 swap type if swapped
052fb0d6 1630 * Bits 5-54 swap offset if swapped
1ad1335d 1631 * Bit 55 pte is soft-dirty (see Documentation/admin-guide/mm/soft-dirty.rst)
77bb499b 1632 * Bit 56 page exclusively mapped
dd21bfa4
YZ		 1633 * Bit 57 pte is uffd-wp write-protected
1634 * Bits 58-60 zero
052fb0d6 1635 * Bit 61 page is file-page or shared-anon
f16278c6
HR		 1636 * Bit 62 page swapped
1637 * Bit 63 page present
1638 *
1639 * If the page is not present but in swap, then the PFN contains an
1640 * encoding of the swap file number and the page's offset into the
1641 * swap. Unmapped pages return a null PFN. This allows determining
85863e47
MM		 1642 * precisely which pages are mapped (or in swap) and comparing mapped
1643 * pages between processes.
1644 *
1645 * Efficient users of this interface will use /proc/pid/maps to
1646 * determine which areas of memory are actually mapped and llseek to
1647 * skip over unmapped regions.
1648 */
1649static ssize_t pagemap_read(struct file *file, char __user *buf,
1650 size_t count, loff_t *ppos)
1651{
a06db751 1652 struct mm_struct *mm = file->private_data;
85863e47 1653 struct pagemapread pm;
5d7e0d2b
AM		 1654 unsigned long src;
1655 unsigned long svpfn;
1656 unsigned long start_vaddr;
1657 unsigned long end_vaddr;
a06db751 1658 int ret = 0, copied = 0;
85863e47 1659
388f7934 1660 if (!mm || !mmget_not_zero(mm))
85863e47
MM		 1661 goto out;
1662
85863e47
MM		 1663 ret = -EINVAL;
1664 /* file position must be aligned */
aae8679b 1665 if ((*ppos % PM_ENTRY_BYTES) || (count % PM_ENTRY_BYTES))
a06db751 1666 goto out_mm;
85863e47
MM		 1667
1668 ret = 0;
08161786 1669 if (!count)
a06db751 1670 goto out_mm;
08161786 1671
1c90308e
KK		 1672 /* do not disclose physical addresses: attack vector */
1673 pm.show_pfn = file_ns_capable(file, &init_user_ns, CAP_SYS_ADMIN);
1674
8c829622 1675 pm.len = (PAGEMAP_WALK_SIZE >> PAGE_SHIFT);
6da2ec56 1676 pm.buffer = kmalloc_array(pm.len, PM_ENTRY_BYTES, GFP_KERNEL);
5d7e0d2b 1677 ret = -ENOMEM;
d82ef020 1678 if (!pm.buffer)
a06db751 1679 goto out_mm;
85863e47 1680
5d7e0d2b
AM		 1681 src = *ppos;
1682 svpfn = src / PM_ENTRY_BYTES;
a06db751 1683 end_vaddr = mm->task_size;
5d7e0d2b
AM		 1684
1685 /* watch out for wraparound */
40d6366e 1686 start_vaddr = end_vaddr;
428e106a 1687 if (svpfn <= (ULONG_MAX >> PAGE_SHIFT)) {
7bab8dfb
YX		 1688 unsigned long end;
1689
428e106a
KS		 1690 ret = mmap_read_lock_killable(mm);
1691 if (ret)
1692 goto out_free;
1693 start_vaddr = untagged_addr_remote(mm, svpfn << PAGE_SHIFT);
1694 mmap_read_unlock(mm);
7bab8dfb
YX		 1695
1696 end = start_vaddr + ((count / PM_ENTRY_BYTES) << PAGE_SHIFT);
1697 if (end >= start_vaddr && end < mm->task_size)
1698 end_vaddr = end;
428e106a 1699 }
40d6366e
MC		 1700
1701 /* Ensure the address is inside the task */
1702 if (start_vaddr > mm->task_size)
5d7e0d2b
AM		 1703 start_vaddr = end_vaddr;
1704
d82ef020
KH		 1705 ret = 0;
1706 while (count && (start_vaddr < end_vaddr)) {
1707 int len;
1708 unsigned long end;
1709
1710 pm.pos = 0;
ea251c1d 1711 end = (start_vaddr + PAGEMAP_WALK_SIZE) & PAGEMAP_WALK_MASK;
d82ef020
KH		 1712 /* overflow ? */
1713 if (end < start_vaddr || end > end_vaddr)
1714 end = end_vaddr;
d8ed45c5 1715 ret = mmap_read_lock_killable(mm);
ad80b932
KK		 1716 if (ret)
1717 goto out_free;
7b86ac33 1718 ret = walk_page_range(mm, start_vaddr, end, &pagemap_ops, &pm);
d8ed45c5 1719 mmap_read_unlock(mm);
d82ef020
KH		 1720 start_vaddr = end;
1721
1722 len = min(count, PM_ENTRY_BYTES * pm.pos);
309361e0 1723 if (copy_to_user(buf, pm.buffer, len)) {
d82ef020 1724 ret = -EFAULT;
a06db751 1725 goto out_free;
d82ef020
KH		 1726 }
1727 copied += len;
1728 buf += len;
1729 count -= len;
85863e47 1730 }
d82ef020
KH		 1731 *ppos += copied;
1732 if (!ret || ret == PM_END_OF_BUFFER)
1733 ret = copied;
1734
98bc93e5
KM		 1735out_free:
1736 kfree(pm.buffer);
a06db751
KK		 1737out_mm:
1738 mmput(mm);
85863e47
MM		 1739out:
1740 return ret;
1741}
1742
541c237c
PE		 1743static int pagemap_open(struct inode *inode, struct file *file)
1744{
a06db751
KK		 1745 struct mm_struct *mm;
1746
a06db751
KK		 1747 mm = proc_mem_open(inode, PTRACE_MODE_READ);
1748 if (IS_ERR(mm))
1749 return PTR_ERR(mm);
1750 file->private_data = mm;
1751 return 0;
1752}
1753
1754static int pagemap_release(struct inode *inode, struct file *file)
1755{
1756 struct mm_struct *mm = file->private_data;
1757
1758 if (mm)
1759 mmdrop(mm);
541c237c
PE		 1760 return 0;
1761}
1762
52526ca7
MUA		 1763#define PM_SCAN_CATEGORIES (PAGE_IS_WPALLOWED | PAGE_IS_WRITTEN | \
1764 PAGE_IS_FILE | PAGE_IS_PRESENT | \
1765 PAGE_IS_SWAPPED | PAGE_IS_PFNZERO | \
e6a9a2cb 1766 PAGE_IS_HUGE | PAGE_IS_SOFT_DIRTY)
52526ca7
MUA		 1767#define PM_SCAN_FLAGS (PM_SCAN_WP_MATCHING | PM_SCAN_CHECK_WPASYNC)
1768
1769struct pagemap_scan_private {
1770 struct pm_scan_arg arg;
1771 unsigned long masks_of_interest, cur_vma_category;
1772 struct page_region *vec_buf;
1773 unsigned long vec_buf_len, vec_buf_index, found_pages;
1774 struct page_region __user *vec_out;
1775};
1776
1777static unsigned long pagemap_page_category(struct pagemap_scan_private *p,
1778 struct vm_area_struct *vma,
1779 unsigned long addr, pte_t pte)
1780{
1781 unsigned long categories = 0;
1782
1783 if (pte_present(pte)) {
1784 struct page *page;
1785
1786 categories |= PAGE_IS_PRESENT;
1787 if (!pte_uffd_wp(pte))
1788 categories |= PAGE_IS_WRITTEN;
1789
1790 if (p->masks_of_interest & PAGE_IS_FILE) {
1791 page = vm_normal_page(vma, addr, pte);
1792 if (page && !PageAnon(page))
1793 categories |= PAGE_IS_FILE;
1794 }
1795
1796 if (is_zero_pfn(pte_pfn(pte)))
1797 categories |= PAGE_IS_PFNZERO;
e6a9a2cb
AV		 1798 if (pte_soft_dirty(pte))
1799 categories |= PAGE_IS_SOFT_DIRTY;
52526ca7
MUA		 1800 } else if (is_swap_pte(pte)) {
1801 swp_entry_t swp;
1802
1803 categories |= PAGE_IS_SWAPPED;
1804 if (!pte_swp_uffd_wp_any(pte))
1805 categories |= PAGE_IS_WRITTEN;
1806
1807 if (p->masks_of_interest & PAGE_IS_FILE) {
1808 swp = pte_to_swp_entry(pte);
1809 if (is_pfn_swap_entry(swp) &&
1810 !PageAnon(pfn_swap_entry_to_page(swp)))
1811 categories |= PAGE_IS_FILE;
1812 }
e6a9a2cb
AV		 1813 if (pte_swp_soft_dirty(pte))
1814 categories |= PAGE_IS_SOFT_DIRTY;
52526ca7
MUA		 1815 }
1816
1817 return categories;
1818}
1819
1820static void make_uffd_wp_pte(struct vm_area_struct *vma,
1821 unsigned long addr, pte_t *pte)
1822{
1823 pte_t ptent = ptep_get(pte);
1824
1825 if (pte_present(ptent)) {
1826 pte_t old_pte;
1827
1828 old_pte = ptep_modify_prot_start(vma, addr, pte);
1829 ptent = pte_mkuffd_wp(ptent);
1830 ptep_modify_prot_commit(vma, addr, pte, old_pte, ptent);
1831 } else if (is_swap_pte(ptent)) {
1832 ptent = pte_swp_mkuffd_wp(ptent);
1833 set_pte_at(vma->vm_mm, addr, pte, ptent);
1834 } else {
1835 set_pte_at(vma->vm_mm, addr, pte,
1836 make_pte_marker(PTE_MARKER_UFFD_WP));
1837 }
1838}
1839
1840#ifdef CONFIG_TRANSPARENT_HUGEPAGE
1841static unsigned long pagemap_thp_category(struct pagemap_scan_private *p,
1842 struct vm_area_struct *vma,
1843 unsigned long addr, pmd_t pmd)
1844{
1845 unsigned long categories = PAGE_IS_HUGE;
1846
1847 if (pmd_present(pmd)) {
1848 struct page *page;
1849
1850 categories |= PAGE_IS_PRESENT;
1851 if (!pmd_uffd_wp(pmd))
1852 categories |= PAGE_IS_WRITTEN;
1853
1854 if (p->masks_of_interest & PAGE_IS_FILE) {
1855 page = vm_normal_page_pmd(vma, addr, pmd);
1856 if (page && !PageAnon(page))
1857 categories |= PAGE_IS_FILE;
1858 }
1859
1860 if (is_zero_pfn(pmd_pfn(pmd)))
1861 categories |= PAGE_IS_PFNZERO;
e6a9a2cb
AV		 1862 if (pmd_soft_dirty(pmd))
1863 categories |= PAGE_IS_SOFT_DIRTY;
52526ca7
MUA		 1864 } else if (is_swap_pmd(pmd)) {
1865 swp_entry_t swp;
1866
1867 categories |= PAGE_IS_SWAPPED;
1868 if (!pmd_swp_uffd_wp(pmd))
1869 categories |= PAGE_IS_WRITTEN;
e6a9a2cb
AV		 1870 if (pmd_swp_soft_dirty(pmd))
1871 categories |= PAGE_IS_SOFT_DIRTY;
52526ca7
MUA		 1872
1873 if (p->masks_of_interest & PAGE_IS_FILE) {
1874 swp = pmd_to_swp_entry(pmd);
1875 if (is_pfn_swap_entry(swp) &&
1876 !PageAnon(pfn_swap_entry_to_page(swp)))
1877 categories |= PAGE_IS_FILE;
1878 }
1879 }
1880
1881 return categories;
1882}
1883
1884static void make_uffd_wp_pmd(struct vm_area_struct *vma,
1885 unsigned long addr, pmd_t *pmdp)
1886{
1887 pmd_t old, pmd = *pmdp;
1888
1889 if (pmd_present(pmd)) {
1890 old = pmdp_invalidate_ad(vma, addr, pmdp);
1891 pmd = pmd_mkuffd_wp(old);
1892 set_pmd_at(vma->vm_mm, addr, pmdp, pmd);
1893 } else if (is_migration_entry(pmd_to_swp_entry(pmd))) {
1894 pmd = pmd_swp_mkuffd_wp(pmd);
1895 set_pmd_at(vma->vm_mm, addr, pmdp, pmd);
1896 }
1897}
1898#endif /* CONFIG_TRANSPARENT_HUGEPAGE */
1899
1900#ifdef CONFIG_HUGETLB_PAGE
1901static unsigned long pagemap_hugetlb_category(pte_t pte)
1902{
1903 unsigned long categories = PAGE_IS_HUGE;
1904
1905 /*
1906 * According to pagemap_hugetlb_range(), file-backed HugeTLB
1907 * page cannot be swapped. So PAGE_IS_FILE is not checked for
1908 * swapped pages.
1909 */
1910 if (pte_present(pte)) {
1911 categories |= PAGE_IS_PRESENT;
1912 if (!huge_pte_uffd_wp(pte))
1913 categories |= PAGE_IS_WRITTEN;
1914 if (!PageAnon(pte_page(pte)))
1915 categories |= PAGE_IS_FILE;
1916 if (is_zero_pfn(pte_pfn(pte)))
1917 categories |= PAGE_IS_PFNZERO;
e6a9a2cb
AV		 1918 if (pte_soft_dirty(pte))
1919 categories |= PAGE_IS_SOFT_DIRTY;
52526ca7
MUA		 1920 } else if (is_swap_pte(pte)) {
1921 categories |= PAGE_IS_SWAPPED;
1922 if (!pte_swp_uffd_wp_any(pte))
1923 categories |= PAGE_IS_WRITTEN;
e6a9a2cb
AV		 1924 if (pte_swp_soft_dirty(pte))
1925 categories |= PAGE_IS_SOFT_DIRTY;
52526ca7
MUA		 1926 }
1927
1928 return categories;
1929}
1930
1931static void make_uffd_wp_huge_pte(struct vm_area_struct *vma,
1932 unsigned long addr, pte_t *ptep,
1933 pte_t ptent)
1934{
1935 unsigned long psize;
1936
1937 if (is_hugetlb_entry_hwpoisoned(ptent) || is_pte_marker(ptent))
1938 return;
1939
1940 psize = huge_page_size(hstate_vma(vma));
1941
1942 if (is_hugetlb_entry_migration(ptent))
1943 set_huge_pte_at(vma->vm_mm, addr, ptep,
1944 pte_swp_mkuffd_wp(ptent), psize);
1945 else if (!huge_pte_none(ptent))
1946 huge_ptep_modify_prot_commit(vma, addr, ptep, ptent,
1947 huge_pte_mkuffd_wp(ptent));
1948 else
1949 set_huge_pte_at(vma->vm_mm, addr, ptep,
1950 make_pte_marker(PTE_MARKER_UFFD_WP), psize);
1951}
1952#endif /* CONFIG_HUGETLB_PAGE */
1953
1954#if defined(CONFIG_TRANSPARENT_HUGEPAGE) || defined(CONFIG_HUGETLB_PAGE)
1955static void pagemap_scan_backout_range(struct pagemap_scan_private *p,
1956 unsigned long addr, unsigned long end)
1957{
1958 struct page_region *cur_buf = &p->vec_buf[p->vec_buf_index];
1959
1960 if (cur_buf->start != addr)
1961 cur_buf->end = addr;
1962 else
1963 cur_buf->start = cur_buf->end = 0;
1964
1965 p->found_pages -= (end - addr) / PAGE_SIZE;
1966}
1967#endif
1968
1969static bool pagemap_scan_is_interesting_page(unsigned long categories,
1970 const struct pagemap_scan_private *p)
1971{
1972 categories ^= p->arg.category_inverted;
1973 if ((categories & p->arg.category_mask) != p->arg.category_mask)
1974 return false;
1975 if (p->arg.category_anyof_mask && !(categories & p->arg.category_anyof_mask))
1976 return false;
1977
1978 return true;
1979}
1980
1981static bool pagemap_scan_is_interesting_vma(unsigned long categories,
1982 const struct pagemap_scan_private *p)
1983{
1984 unsigned long required = p->arg.category_mask & PAGE_IS_WPALLOWED;
1985
1986 categories ^= p->arg.category_inverted;
1987 if ((categories & required) != required)
1988 return false;
1989
1990 return true;
1991}
1992
1993static int pagemap_scan_test_walk(unsigned long start, unsigned long end,
1994 struct mm_walk *walk)
1995{
1996 struct pagemap_scan_private *p = walk->private;
1997 struct vm_area_struct *vma = walk->vma;
1998 unsigned long vma_category = 0;
0dff1b40
PX		 1999 bool wp_allowed = userfaultfd_wp_async(vma) &&
2000 userfaultfd_wp_use_markers(vma);
52526ca7 2001
0dff1b40
PX		 2002 if (!wp_allowed) {
2003 /* User requested explicit failure over wp-async capability */
2004 if (p->arg.flags & PM_SCAN_CHECK_WPASYNC)
2005 return -EPERM;
2006 /*
2007 * User requires wr-protect, and allows silently skipping
2008 * unsupported vmas.
2009 */
2010 if (p->arg.flags & PM_SCAN_WP_MATCHING)
2011 return 1;
2012 /*
2013 * Then the request doesn't involve wr-protects at all,
2014 * fall through to the rest checks, and allow vma walk.
2015 */
2016 }
52526ca7
MUA		 2017
2018 if (vma->vm_flags & VM_PFNMAP)
2019 return 1;
2020
0dff1b40
PX		 2021 if (wp_allowed)
2022 vma_category |= PAGE_IS_WPALLOWED;
2023
e6a9a2cb
AV		 2024 if (vma->vm_flags & VM_SOFTDIRTY)
2025 vma_category |= PAGE_IS_SOFT_DIRTY;
2026
52526ca7
MUA		 2027 if (!pagemap_scan_is_interesting_vma(vma_category, p))
2028 return 1;
2029
2030 p->cur_vma_category = vma_category;
2031
2032 return 0;
2033}
2034
2035static bool pagemap_scan_push_range(unsigned long categories,
2036 struct pagemap_scan_private *p,
2037 unsigned long addr, unsigned long end)
2038{
2039 struct page_region *cur_buf = &p->vec_buf[p->vec_buf_index];
2040
2041 /*
2042 * When there is no output buffer provided at all, the sentinel values
2043 * won't match here. There is no other way for `cur_buf->end` to be
2044 * non-zero other than it being non-empty.
2045 */
2046 if (addr == cur_buf->end && categories == cur_buf->categories) {
2047 cur_buf->end = end;
2048 return true;
2049 }
2050
2051 if (cur_buf->end) {
2052 if (p->vec_buf_index >= p->vec_buf_len - 1)
2053 return false;
2054
2055 cur_buf = &p->vec_buf[++p->vec_buf_index];
2056 }
2057
2058 cur_buf->start = addr;
2059 cur_buf->end = end;
2060 cur_buf->categories = categories;
2061
2062 return true;
2063}
2064
2065static int pagemap_scan_output(unsigned long categories,
2066 struct pagemap_scan_private *p,
2067 unsigned long addr, unsigned long *end)
2068{
2069 unsigned long n_pages, total_pages;
2070 int ret = 0;
2071
2072 if (!p->vec_buf)
2073 return 0;
2074
2075 categories &= p->arg.return_mask;
2076
2077 n_pages = (*end - addr) / PAGE_SIZE;
2078 if (check_add_overflow(p->found_pages, n_pages, &total_pages) ||
2079 total_pages > p->arg.max_pages) {
2080 size_t n_too_much = total_pages - p->arg.max_pages;
2081 *end -= n_too_much * PAGE_SIZE;
2082 n_pages -= n_too_much;
2083 ret = -ENOSPC;
2084 }
2085
2086 if (!pagemap_scan_push_range(categories, p, addr, *end)) {
2087 *end = addr;
2088 n_pages = 0;
2089 ret = -ENOSPC;
2090 }
2091
2092 p->found_pages += n_pages;
2093 if (ret)
2094 p->arg.walk_end = *end;
2095
2096 return ret;
2097}
2098
2099static int pagemap_scan_thp_entry(pmd_t *pmd, unsigned long start,
2100 unsigned long end, struct mm_walk *walk)
2101{
2102#ifdef CONFIG_TRANSPARENT_HUGEPAGE
2103 struct pagemap_scan_private *p = walk->private;
2104 struct vm_area_struct *vma = walk->vma;
2105 unsigned long categories;
2106 spinlock_t *ptl;
2107 int ret = 0;
2108
2109 ptl = pmd_trans_huge_lock(pmd, vma);
2110 if (!ptl)
2111 return -ENOENT;
2112
2113 categories = p->cur_vma_category |
2114 pagemap_thp_category(p, vma, start, *pmd);
2115
2116 if (!pagemap_scan_is_interesting_page(categories, p))
2117 goto out_unlock;
2118
2119 ret = pagemap_scan_output(categories, p, start, &end);
2120 if (start == end)
2121 goto out_unlock;
2122
2123 if (~p->arg.flags & PM_SCAN_WP_MATCHING)
2124 goto out_unlock;
2125 if (~categories & PAGE_IS_WRITTEN)
2126 goto out_unlock;
2127
2128 /*
2129 * Break huge page into small pages if the WP operation
2130 * needs to be performed on a portion of the huge page.
2131 */
2132 if (end != start + HPAGE_SIZE) {
2133 spin_unlock(ptl);
2134 split_huge_pmd(vma, pmd, start);
2135 pagemap_scan_backout_range(p, start, end);
2136 /* Report as if there was no THP */
2137 return -ENOENT;
2138 }
2139
2140 make_uffd_wp_pmd(vma, start, pmd);
2141 flush_tlb_range(vma, start, end);
2142out_unlock:
2143 spin_unlock(ptl);
2144 return ret;
2145#else /* !CONFIG_TRANSPARENT_HUGEPAGE */
2146 return -ENOENT;
2147#endif
2148}
2149
2150static int pagemap_scan_pmd_entry(pmd_t *pmd, unsigned long start,
2151 unsigned long end, struct mm_walk *walk)
2152{
2153 struct pagemap_scan_private *p = walk->private;
2154 struct vm_area_struct *vma = walk->vma;
2155 unsigned long addr, flush_end = 0;
2156 pte_t *pte, *start_pte;
2157 spinlock_t *ptl;
2158 int ret;
2159
2160 arch_enter_lazy_mmu_mode();
2161
2162 ret = pagemap_scan_thp_entry(pmd, start, end, walk);
2163 if (ret != -ENOENT) {
2164 arch_leave_lazy_mmu_mode();
2165 return ret;
2166 }
2167
2168 ret = 0;
2169 start_pte = pte = pte_offset_map_lock(vma->vm_mm, pmd, start, &ptl);
2170 if (!pte) {
2171 arch_leave_lazy_mmu_mode();
2172 walk->action = ACTION_AGAIN;
2173 return 0;
2174 }
2175
4980e837 2176 if ((p->arg.flags & PM_SCAN_WP_MATCHING) && !p->vec_out) {
12f6b01a
MUA		 2177 /* Fast path for performing exclusive WP */
2178 for (addr = start; addr != end; pte++, addr += PAGE_SIZE) {
2179 if (pte_uffd_wp(ptep_get(pte)))
2180 continue;
2181 make_uffd_wp_pte(vma, addr, pte);
2182 if (!flush_end)
2183 start = addr;
2184 flush_end = addr + PAGE_SIZE;
2185 }
2186 goto flush_and_return;
2187 }
2188
2189 if (!p->arg.category_anyof_mask && !p->arg.category_inverted &&
2190 p->arg.category_mask == PAGE_IS_WRITTEN &&
2191 p->arg.return_mask == PAGE_IS_WRITTEN) {
2192 for (addr = start; addr < end; pte++, addr += PAGE_SIZE) {
2193 unsigned long next = addr + PAGE_SIZE;
2194
2195 if (pte_uffd_wp(ptep_get(pte)))
2196 continue;
2197 ret = pagemap_scan_output(p->cur_vma_category | PAGE_IS_WRITTEN,
2198 p, addr, &next);
2199 if (next == addr)
2200 break;
2201 if (~p->arg.flags & PM_SCAN_WP_MATCHING)
2202 continue;
2203 make_uffd_wp_pte(vma, addr, pte);
2204 if (!flush_end)
2205 start = addr;
2206 flush_end = next;
2207 }
2208 goto flush_and_return;
2209 }
2210
52526ca7
MUA		 2211 for (addr = start; addr != end; pte++, addr += PAGE_SIZE) {
2212 unsigned long categories = p->cur_vma_category |
2213 pagemap_page_category(p, vma, addr, ptep_get(pte));
2214 unsigned long next = addr + PAGE_SIZE;
2215
2216 if (!pagemap_scan_is_interesting_page(categories, p))
2217 continue;
2218
2219 ret = pagemap_scan_output(categories, p, addr, &next);
2220 if (next == addr)
2221 break;
2222
2223 if (~p->arg.flags & PM_SCAN_WP_MATCHING)
2224 continue;
2225 if (~categories & PAGE_IS_WRITTEN)
2226 continue;
2227
2228 make_uffd_wp_pte(vma, addr, pte);
2229 if (!flush_end)
2230 start = addr;
2231 flush_end = next;
2232 }
2233
12f6b01a 2234flush_and_return:
52526ca7
MUA		 2235 if (flush_end)
2236 flush_tlb_range(vma, start, addr);
2237
2238 pte_unmap_unlock(start_pte, ptl);
2239 arch_leave_lazy_mmu_mode();
2240
2241 cond_resched();
2242 return ret;
2243}
2244
2245#ifdef CONFIG_HUGETLB_PAGE
2246static int pagemap_scan_hugetlb_entry(pte_t *ptep, unsigned long hmask,
2247 unsigned long start, unsigned long end,
2248 struct mm_walk *walk)
2249{
2250 struct pagemap_scan_private *p = walk->private;
2251 struct vm_area_struct *vma = walk->vma;
2252 unsigned long categories;
2253 spinlock_t *ptl;
2254 int ret = 0;
2255 pte_t pte;
2256
2257 if (~p->arg.flags & PM_SCAN_WP_MATCHING) {
2258 /* Go the short route when not write-protecting pages. */
2259
2260 pte = huge_ptep_get(ptep);
2261 categories = p->cur_vma_category | pagemap_hugetlb_category(pte);
2262
2263 if (!pagemap_scan_is_interesting_page(categories, p))
2264 return 0;
2265
2266 return pagemap_scan_output(categories, p, start, &end);
2267 }
2268
2269 i_mmap_lock_write(vma->vm_file->f_mapping);
2270 ptl = huge_pte_lock(hstate_vma(vma), vma->vm_mm, ptep);
2271
2272 pte = huge_ptep_get(ptep);
2273 categories = p->cur_vma_category | pagemap_hugetlb_category(pte);
2274
2275 if (!pagemap_scan_is_interesting_page(categories, p))
2276 goto out_unlock;
2277
2278 ret = pagemap_scan_output(categories, p, start, &end);
2279 if (start == end)
2280 goto out_unlock;
2281
2282 if (~categories & PAGE_IS_WRITTEN)
2283 goto out_unlock;
2284
2285 if (end != start + HPAGE_SIZE) {
2286 /* Partial HugeTLB page WP isn't possible. */
2287 pagemap_scan_backout_range(p, start, end);
2288 p->arg.walk_end = start;
2289 ret = 0;
2290 goto out_unlock;
2291 }
2292
2293 make_uffd_wp_huge_pte(vma, start, ptep, pte);
2294 flush_hugetlb_tlb_range(vma, start, end);
2295
2296out_unlock:
2297 spin_unlock(ptl);
2298 i_mmap_unlock_write(vma->vm_file->f_mapping);
2299
2300 return ret;
2301}
2302#else
2303#define pagemap_scan_hugetlb_entry NULL
2304#endif
2305
2306static int pagemap_scan_pte_hole(unsigned long addr, unsigned long end,
2307 int depth, struct mm_walk *walk)
2308{
2309 struct pagemap_scan_private *p = walk->private;
2310 struct vm_area_struct *vma = walk->vma;
2311 int ret, err;
2312
2313 if (!vma || !pagemap_scan_is_interesting_page(p->cur_vma_category, p))
2314 return 0;
2315
2316 ret = pagemap_scan_output(p->cur_vma_category, p, addr, &end);
2317 if (addr == end)
2318 return ret;
2319
2320 if (~p->arg.flags & PM_SCAN_WP_MATCHING)
2321 return ret;
2322
2323 err = uffd_wp_range(vma, addr, end - addr, true);
2324 if (err < 0)
2325 ret = err;
2326
2327 return ret;
2328}
2329
2330static const struct mm_walk_ops pagemap_scan_ops = {
2331 .test_walk = pagemap_scan_test_walk,
2332 .pmd_entry = pagemap_scan_pmd_entry,
2333 .pte_hole = pagemap_scan_pte_hole,
2334 .hugetlb_entry = pagemap_scan_hugetlb_entry,
2335};
2336
2337static int pagemap_scan_get_args(struct pm_scan_arg *arg,
2338 unsigned long uarg)
2339{
2340 if (copy_from_user(arg, (void __user *)uarg, sizeof(*arg)))
2341 return -EFAULT;
2342
2343 if (arg->size != sizeof(struct pm_scan_arg))
2344 return -EINVAL;
2345
2346 /* Validate requested features */
2347 if (arg->flags & ~PM_SCAN_FLAGS)
2348 return -EINVAL;
2349 if ((arg->category_inverted | arg->category_mask |
2350 arg->category_anyof_mask | arg->return_mask) & ~PM_SCAN_CATEGORIES)
2351 return -EINVAL;
2352
2353 arg->start = untagged_addr((unsigned long)arg->start);
2354 arg->end = untagged_addr((unsigned long)arg->end);
2355 arg->vec = untagged_addr((unsigned long)arg->vec);
2356
2357 /* Validate memory pointers */
2358 if (!IS_ALIGNED(arg->start, PAGE_SIZE))
2359 return -EINVAL;
2360 if (!access_ok((void __user *)(long)arg->start, arg->end - arg->start))
2361 return -EFAULT;
2362 if (!arg->vec && arg->vec_len)
2363 return -EINVAL;
2364 if (arg->vec && !access_ok((void __user *)(long)arg->vec,
2365 arg->vec_len * sizeof(struct page_region)))
2366 return -EFAULT;
2367
2368 /* Fixup default values */
2369 arg->end = ALIGN(arg->end, PAGE_SIZE);
2370 arg->walk_end = 0;
2371 if (!arg->max_pages)
2372 arg->max_pages = ULONG_MAX;
2373
2374 return 0;
2375}
2376
2377static int pagemap_scan_writeback_args(struct pm_scan_arg *arg,
2378 unsigned long uargl)
2379{
2380 struct pm_scan_arg __user *uarg = (void __user *)uargl;
2381
2382 if (copy_to_user(&uarg->walk_end, &arg->walk_end, sizeof(arg->walk_end)))
2383 return -EFAULT;
2384
2385 return 0;
2386}
2387
2388static int pagemap_scan_init_bounce_buffer(struct pagemap_scan_private *p)
2389{
2390 if (!p->arg.vec_len)
2391 return 0;
2392
2393 p->vec_buf_len = min_t(size_t, PAGEMAP_WALK_SIZE >> PAGE_SHIFT,
2394 p->arg.vec_len);
2395 p->vec_buf = kmalloc_array(p->vec_buf_len, sizeof(*p->vec_buf),
2396 GFP_KERNEL);
2397 if (!p->vec_buf)
2398 return -ENOMEM;
2399
2400 p->vec_buf->start = p->vec_buf->end = 0;
2401 p->vec_out = (struct page_region __user *)(long)p->arg.vec;
2402
2403 return 0;
2404}
2405
2406static long pagemap_scan_flush_buffer(struct pagemap_scan_private *p)
2407{
2408 const struct page_region *buf = p->vec_buf;
2409 long n = p->vec_buf_index;
2410
2411 if (!p->vec_buf)
2412 return 0;
2413
2414 if (buf[n].end != buf[n].start)
2415 n++;
2416
2417 if (!n)
2418 return 0;
2419
2420 if (copy_to_user(p->vec_out, buf, n * sizeof(*buf)))
2421 return -EFAULT;
2422
2423 p->arg.vec_len -= n;
2424 p->vec_out += n;
2425
2426 p->vec_buf_index = 0;
2427 p->vec_buf_len = min_t(size_t, p->vec_buf_len, p->arg.vec_len);
2428 p->vec_buf->start = p->vec_buf->end = 0;
2429
2430 return n;
2431}
2432
2433static long do_pagemap_scan(struct mm_struct *mm, unsigned long uarg)
2434{
52526ca7
MUA		 2435 struct pagemap_scan_private p = {0};
2436 unsigned long walk_start;
2437 size_t n_ranges_out = 0;
2438 int ret;
2439
2440 ret = pagemap_scan_get_args(&p.arg, uarg);
2441 if (ret)
2442 return ret;
2443
2444 p.masks_of_interest = p.arg.category_mask | p.arg.category_anyof_mask |
2445 p.arg.return_mask;
2446 ret = pagemap_scan_init_bounce_buffer(&p);
2447 if (ret)
2448 return ret;
2449
52526ca7
MUA		 2450 for (walk_start = p.arg.start; walk_start < p.arg.end;
2451 walk_start = p.arg.walk_end) {
4cccb622 2452 struct mmu_notifier_range range;
52526ca7
MUA		 2453 long n_out;
2454
2455 if (fatal_signal_pending(current)) {
2456 ret = -EINTR;
2457 break;
2458 }
2459
2460 ret = mmap_read_lock_killable(mm);
2461 if (ret)
2462 break;
4cccb622
MUA		 2463
2464 /* Protection change for the range is going to happen. */
2465 if (p.arg.flags & PM_SCAN_WP_MATCHING) {
2466 mmu_notifier_range_init(&range, MMU_NOTIFY_PROTECTION_VMA, 0,
2467 mm, walk_start, p.arg.end);
2468 mmu_notifier_invalidate_range_start(&range);
2469 }
2470
52526ca7
MUA		 2471 ret = walk_page_range(mm, walk_start, p.arg.end,
2472 &pagemap_scan_ops, &p);
4cccb622
MUA		 2473
2474 if (p.arg.flags & PM_SCAN_WP_MATCHING)
2475 mmu_notifier_invalidate_range_end(&range);
2476
52526ca7
MUA		 2477 mmap_read_unlock(mm);
2478
2479 n_out = pagemap_scan_flush_buffer(&p);
2480 if (n_out < 0)
2481 ret = n_out;
2482 else
2483 n_ranges_out += n_out;
2484
2485 if (ret != -ENOSPC)
2486 break;
2487
2488 if (p.arg.vec_len == 0 || p.found_pages == p.arg.max_pages)
2489 break;
2490 }
2491
2492 /* ENOSPC signifies early stop (buffer full) from the walk. */
2493 if (!ret || ret == -ENOSPC)
2494 ret = n_ranges_out;
2495
2496 /* The walk_end isn't set when ret is zero */
2497 if (!p.arg.walk_end)
2498 p.arg.walk_end = p.arg.end;
2499 if (pagemap_scan_writeback_args(&p.arg, uarg))
2500 ret = -EFAULT;
2501
52526ca7
MUA		 2502 kfree(p.vec_buf);
2503 return ret;
2504}
2505
2506static long do_pagemap_cmd(struct file *file, unsigned int cmd,
2507 unsigned long arg)
2508{
2509 struct mm_struct *mm = file->private_data;
2510
2511 switch (cmd) {
2512 case PAGEMAP_SCAN:
2513 return do_pagemap_scan(mm, arg);
2514
2515 default:
2516 return -EINVAL;
2517 }
2518}
2519
85863e47
MM		 2520const struct file_operations proc_pagemap_operations = {
2521 .llseek = mem_lseek, /* borrow this */
2522 .read = pagemap_read,
541c237c 2523 .open = pagemap_open,
a06db751 2524 .release = pagemap_release,
52526ca7
MUA		 2525 .unlocked_ioctl = do_pagemap_cmd,
2526 .compat_ioctl = do_pagemap_cmd,
85863e47 2527};
1e883281 2528#endif /* CONFIG_PROC_PAGE_MONITOR */
85863e47 2529
6e21c8f1 2530#ifdef CONFIG_NUMA
6e21c8f1 2531
f69ff943 2532struct numa_maps {
f69ff943
SW		 2533 unsigned long pages;
2534 unsigned long anon;
2535 unsigned long active;
2536 unsigned long writeback;
2537 unsigned long mapcount_max;
2538 unsigned long dirty;
2539 unsigned long swapcache;
2540 unsigned long node[MAX_NUMNODES];
2541};
2542
5b52fc89
SW		 2543struct numa_maps_private {
2544 struct proc_maps_private proc_maps;
2545 struct numa_maps md;
2546};
2547
eb4866d0
DH		 2548static void gather_stats(struct page *page, struct numa_maps *md, int pte_dirty,
2549 unsigned long nr_pages)
f69ff943
SW		 2550{
2551 int count = page_mapcount(page);
2552
eb4866d0 2553 md->pages += nr_pages;
f69ff943 2554 if (pte_dirty || PageDirty(page))
eb4866d0 2555 md->dirty += nr_pages;
f69ff943
SW		 2556
2557 if (PageSwapCache(page))
eb4866d0 2558 md->swapcache += nr_pages;
f69ff943
SW		 2559
2560 if (PageActive(page) || PageUnevictable(page))
eb4866d0 2561 md->active += nr_pages;
f69ff943
SW		 2562
2563 if (PageWriteback(page))
eb4866d0 2564 md->writeback += nr_pages;
f69ff943
SW		 2565
2566 if (PageAnon(page))
eb4866d0 2567 md->anon += nr_pages;
f69ff943
SW		 2568
2569 if (count > md->mapcount_max)
2570 md->mapcount_max = count;
2571
eb4866d0 2572 md->node[page_to_nid(page)] += nr_pages;
f69ff943
SW		 2573}
2574
3200a8aa
DH		 2575static struct page *can_gather_numa_stats(pte_t pte, struct vm_area_struct *vma,
2576 unsigned long addr)
2577{
2578 struct page *page;
2579 int nid;
2580
2581 if (!pte_present(pte))
2582 return NULL;
2583
2584 page = vm_normal_page(vma, addr, pte);
3218f871 2585 if (!page || is_zone_device_page(page))
3200a8aa
DH		 2586 return NULL;
2587
2588 if (PageReserved(page))
2589 return NULL;
2590
2591 nid = page_to_nid(page);
4ff1b2c2 2592 if (!node_isset(nid, node_states[N_MEMORY]))
3200a8aa
DH		 2593 return NULL;
2594
2595 return page;
2596}
2597
28093f9f
GS		 2598#ifdef CONFIG_TRANSPARENT_HUGEPAGE
2599static struct page *can_gather_numa_stats_pmd(pmd_t pmd,
2600 struct vm_area_struct *vma,
2601 unsigned long addr)
2602{
2603 struct page *page;
2604 int nid;
2605
2606 if (!pmd_present(pmd))
2607 return NULL;
2608
2609 page = vm_normal_page_pmd(vma, addr, pmd);
2610 if (!page)
2611 return NULL;
2612
2613 if (PageReserved(page))
2614 return NULL;
2615
2616 nid = page_to_nid(page);
2617 if (!node_isset(nid, node_states[N_MEMORY]))
2618 return NULL;
2619
2620 return page;
2621}
2622#endif
2623
f69ff943
SW		 2624static int gather_pte_stats(pmd_t *pmd, unsigned long addr,
2625 unsigned long end, struct mm_walk *walk)
2626{
d85f4d6d
NH		 2627 struct numa_maps *md = walk->private;
2628 struct vm_area_struct *vma = walk->vma;
f69ff943
SW		 2629 spinlock_t *ptl;
2630 pte_t *orig_pte;
2631 pte_t *pte;
2632
28093f9f 2633#ifdef CONFIG_TRANSPARENT_HUGEPAGE
b6ec57f4
KS		 2634 ptl = pmd_trans_huge_lock(pmd, vma);
2635 if (ptl) {
025c5b24
NH		 2636 struct page *page;
2637
28093f9f 2638 page = can_gather_numa_stats_pmd(*pmd, vma, addr);
025c5b24 2639 if (page)
28093f9f 2640 gather_stats(page, md, pmd_dirty(*pmd),
025c5b24 2641 HPAGE_PMD_SIZE/PAGE_SIZE);
bf929152 2642 spin_unlock(ptl);
025c5b24 2643 return 0;
32ef4384 2644 }
28093f9f 2645#endif
f69ff943 2646 orig_pte = pte = pte_offset_map_lock(walk->mm, pmd, addr, &ptl);
7780d040
HD		 2647 if (!pte) {
2648 walk->action = ACTION_AGAIN;
2649 return 0;
2650 }
f69ff943 2651 do {
c33c7948
RR		 2652 pte_t ptent = ptep_get(pte);
2653 struct page *page = can_gather_numa_stats(ptent, vma, addr);
f69ff943
SW		 2654 if (!page)
2655 continue;
c33c7948 2656 gather_stats(page, md, pte_dirty(ptent), 1);
f69ff943
SW		 2657
2658 } while (pte++, addr += PAGE_SIZE, addr != end);
2659 pte_unmap_unlock(orig_pte, ptl);
a66c0410 2660 cond_resched();
f69ff943
SW		 2661 return 0;
2662}
2663#ifdef CONFIG_HUGETLB_PAGE
632fd60f 2664static int gather_hugetlb_stats(pte_t *pte, unsigned long hmask,
f69ff943
SW		 2665 unsigned long addr, unsigned long end, struct mm_walk *walk)
2666{
5c2ff95e 2667 pte_t huge_pte = huge_ptep_get(pte);
f69ff943
SW		 2668 struct numa_maps *md;
2669 struct page *page;
2670
5c2ff95e 2671 if (!pte_present(huge_pte))
f69ff943
SW		 2672 return 0;
2673
5c2ff95e 2674 page = pte_page(huge_pte);
f69ff943
SW		 2675
2676 md = walk->private;
5c2ff95e 2677 gather_stats(page, md, pte_dirty(huge_pte), 1);
f69ff943
SW		 2678 return 0;
2679}
2680
2681#else
632fd60f 2682static int gather_hugetlb_stats(pte_t *pte, unsigned long hmask,
f69ff943
SW		 2683 unsigned long addr, unsigned long end, struct mm_walk *walk)
2684{
2685 return 0;
2686}
2687#endif
2688
7b86ac33
CH		 2689static const struct mm_walk_ops show_numa_ops = {
2690 .hugetlb_entry = gather_hugetlb_stats,
2691 .pmd_entry = gather_pte_stats,
49b06385 2692 .walk_lock = PGWALK_RDLOCK,
7b86ac33
CH		 2693};
2694
f69ff943
SW		 2695/*
2696 * Display pages allocated per node and memory policy via /proc.
2697 */
871305bb 2698static int show_numa_map(struct seq_file *m, void *v)
f69ff943 2699{
5b52fc89
SW		 2700 struct numa_maps_private *numa_priv = m->private;
2701 struct proc_maps_private *proc_priv = &numa_priv->proc_maps;
f69ff943 2702 struct vm_area_struct *vma = v;
5b52fc89 2703 struct numa_maps *md = &numa_priv->md;
f69ff943
SW		 2704 struct file *file = vma->vm_file;
2705 struct mm_struct *mm = vma->vm_mm;
948927ee 2706 char buffer[64];
ddc1a5cb
HD		 2707 struct mempolicy *pol;
2708 pgoff_t ilx;
948927ee 2709 int nid;
f69ff943
SW		 2710
2711 if (!mm)
2712 return 0;
2713
5b52fc89
SW		 2714 /* Ensure we start with an empty set of numa_maps statistics. */
2715 memset(md, 0, sizeof(*md));
f69ff943 2716
ddc1a5cb 2717 pol = __get_vma_policy(vma, vma->vm_start, &ilx);
498f2371
ON		 2718 if (pol) {
2719 mpol_to_str(buffer, sizeof(buffer), pol);
2720 mpol_cond_put(pol);
2721 } else {
2722 mpol_to_str(buffer, sizeof(buffer), proc_priv->task_mempolicy);
2723 }
f69ff943
SW		 2724
2725 seq_printf(m, "%08lx %s", vma->vm_start, buffer);
2726
2727 if (file) {
17c2b4ee 2728 seq_puts(m, " file=");
08582d67 2729 seq_path(m, file_user_path(file), "\n\t= ");
11250fd1 2730 } else if (vma_is_initial_heap(vma)) {
17c2b4ee 2731 seq_puts(m, " heap");
11250fd1 2732 } else if (vma_is_initial_stack(vma)) {
65376df5 2733 seq_puts(m, " stack");
f69ff943
SW		 2734 }
2735
fc360bd9 2736 if (is_vm_hugetlb_page(vma))
17c2b4ee 2737 seq_puts(m, " huge");
fc360bd9 2738
c1e8d7c6 2739 /* mmap_lock is held by m_start */
7b86ac33 2740 walk_page_vma(vma, &show_numa_ops, md);
f69ff943
SW		 2741
2742 if (!md->pages)
2743 goto out;
2744
2745 if (md->anon)
2746 seq_printf(m, " anon=%lu", md->anon);
2747
2748 if (md->dirty)
2749 seq_printf(m, " dirty=%lu", md->dirty);
2750
2751 if (md->pages != md->anon && md->pages != md->dirty)
2752 seq_printf(m, " mapped=%lu", md->pages);
2753
2754 if (md->mapcount_max > 1)
2755 seq_printf(m, " mapmax=%lu", md->mapcount_max);
2756
2757 if (md->swapcache)
2758 seq_printf(m, " swapcache=%lu", md->swapcache);
2759
2760 if (md->active < md->pages && !is_vm_hugetlb_page(vma))
2761 seq_printf(m, " active=%lu", md->active);
2762
2763 if (md->writeback)
2764 seq_printf(m, " writeback=%lu", md->writeback);
2765
948927ee
DR		 2766 for_each_node_state(nid, N_MEMORY)
2767 if (md->node[nid])
2768 seq_printf(m, " N%d=%lu", nid, md->node[nid]);
198d1597
RA		 2769
2770 seq_printf(m, " kernelpagesize_kB=%lu", vma_kernel_pagesize(vma) >> 10);
f69ff943
SW		 2771out:
2772 seq_putc(m, '\n');
f69ff943
SW		 2773 return 0;
2774}
5b52fc89 2775
03a44825 2776static const struct seq_operations proc_pid_numa_maps_op = {
b7643757
SP		 2777 .start = m_start,
2778 .next = m_next,
2779 .stop = m_stop,
871305bb 2780 .show = show_numa_map,
6e21c8f1 2781};
662795de 2782
b7643757
SP		 2783static int pid_numa_maps_open(struct inode *inode, struct file *file)
2784{
871305bb
VB		 2785 return proc_maps_open(inode, file, &proc_pid_numa_maps_op,
2786 sizeof(struct numa_maps_private));
b7643757
SP		 2787}
2788
2789const struct file_operations proc_pid_numa_maps_operations = {
2790 .open = pid_numa_maps_open,
2791 .read = seq_read,
2792 .llseek = seq_lseek,
29a40ace 2793 .release = proc_map_release,
b7643757
SP		 2794};
2795
f69ff943 2796#endif /* CONFIG_NUMA */
Linux 6.x block layer and io_uring tree(s)