]> git.karo-electronics.de Git - linux-beck.git/blob - fs/proc/task_mmu.c
pagemap: use walk->vma instead of calling find_vma()
[linux-beck.git] / fs / proc / task_mmu.c
1 #include <linux/mm.h>
2 #include <linux/vmacache.h>
3 #include <linux/hugetlb.h>
4 #include <linux/huge_mm.h>
5 #include <linux/mount.h>
6 #include <linux/seq_file.h>
7 #include <linux/highmem.h>
8 #include <linux/ptrace.h>
9 #include <linux/slab.h>
10 #include <linux/pagemap.h>
11 #include <linux/mempolicy.h>
12 #include <linux/rmap.h>
13 #include <linux/swap.h>
14 #include <linux/swapops.h>
15 #include <linux/mmu_notifier.h>
16
17 #include <asm/elf.h>
18 #include <asm/uaccess.h>
19 #include <asm/tlbflush.h>
20 #include "internal.h"
21
22 void task_mem(struct seq_file *m, struct mm_struct *mm)
23 {
24         unsigned long data, text, lib, swap, ptes, pmds;
25         unsigned long hiwater_vm, total_vm, hiwater_rss, total_rss;
26
27         /*
28          * Note: to minimize their overhead, mm maintains hiwater_vm and
29          * hiwater_rss only when about to *lower* total_vm or rss.  Any
30          * collector of these hiwater stats must therefore get total_vm
31          * and rss too, which will usually be the higher.  Barriers? not
32          * worth the effort, such snapshots can always be inconsistent.
33          */
34         hiwater_vm = total_vm = mm->total_vm;
35         if (hiwater_vm < mm->hiwater_vm)
36                 hiwater_vm = mm->hiwater_vm;
37         hiwater_rss = total_rss = get_mm_rss(mm);
38         if (hiwater_rss < mm->hiwater_rss)
39                 hiwater_rss = mm->hiwater_rss;
40
41         data = mm->total_vm - mm->shared_vm - mm->stack_vm;
42         text = (PAGE_ALIGN(mm->end_code) - (mm->start_code & PAGE_MASK)) >> 10;
43         lib = (mm->exec_vm << (PAGE_SHIFT-10)) - text;
44         swap = get_mm_counter(mm, MM_SWAPENTS);
45         ptes = PTRS_PER_PTE * sizeof(pte_t) * atomic_long_read(&mm->nr_ptes);
46         pmds = PTRS_PER_PMD * sizeof(pmd_t) * mm_nr_pmds(mm);
47         seq_printf(m,
48                 "VmPeak:\t%8lu kB\n"
49                 "VmSize:\t%8lu kB\n"
50                 "VmLck:\t%8lu kB\n"
51                 "VmPin:\t%8lu kB\n"
52                 "VmHWM:\t%8lu kB\n"
53                 "VmRSS:\t%8lu kB\n"
54                 "VmData:\t%8lu kB\n"
55                 "VmStk:\t%8lu kB\n"
56                 "VmExe:\t%8lu kB\n"
57                 "VmLib:\t%8lu kB\n"
58                 "VmPTE:\t%8lu kB\n"
59                 "VmPMD:\t%8lu kB\n"
60                 "VmSwap:\t%8lu kB\n",
61                 hiwater_vm << (PAGE_SHIFT-10),
62                 total_vm << (PAGE_SHIFT-10),
63                 mm->locked_vm << (PAGE_SHIFT-10),
64                 mm->pinned_vm << (PAGE_SHIFT-10),
65                 hiwater_rss << (PAGE_SHIFT-10),
66                 total_rss << (PAGE_SHIFT-10),
67                 data << (PAGE_SHIFT-10),
68                 mm->stack_vm << (PAGE_SHIFT-10), text, lib,
69                 ptes >> 10,
70                 pmds >> 10,
71                 swap << (PAGE_SHIFT-10));
72 }
73
74 unsigned long task_vsize(struct mm_struct *mm)
75 {
76         return PAGE_SIZE * mm->total_vm;
77 }
78
79 unsigned long task_statm(struct mm_struct *mm,
80                          unsigned long *shared, unsigned long *text,
81                          unsigned long *data, unsigned long *resident)
82 {
83         *shared = get_mm_counter(mm, MM_FILEPAGES);
84         *text = (PAGE_ALIGN(mm->end_code) - (mm->start_code & PAGE_MASK))
85                                                                 >> PAGE_SHIFT;
86         *data = mm->total_vm - mm->shared_vm;
87         *resident = *shared + get_mm_counter(mm, MM_ANONPAGES);
88         return mm->total_vm;
89 }
90
91 #ifdef CONFIG_NUMA
92 /*
93  * Save get_task_policy() for show_numa_map().
94  */
95 static void hold_task_mempolicy(struct proc_maps_private *priv)
96 {
97         struct task_struct *task = priv->task;
98
99         task_lock(task);
100         priv->task_mempolicy = get_task_policy(task);
101         mpol_get(priv->task_mempolicy);
102         task_unlock(task);
103 }
104 static void release_task_mempolicy(struct proc_maps_private *priv)
105 {
106         mpol_put(priv->task_mempolicy);
107 }
108 #else
109 static void hold_task_mempolicy(struct proc_maps_private *priv)
110 {
111 }
112 static void release_task_mempolicy(struct proc_maps_private *priv)
113 {
114 }
115 #endif
116
117 static void vma_stop(struct proc_maps_private *priv)
118 {
119         struct mm_struct *mm = priv->mm;
120
121         release_task_mempolicy(priv);
122         up_read(&mm->mmap_sem);
123         mmput(mm);
124 }
125
126 static struct vm_area_struct *
127 m_next_vma(struct proc_maps_private *priv, struct vm_area_struct *vma)
128 {
129         if (vma == priv->tail_vma)
130                 return NULL;
131         return vma->vm_next ?: priv->tail_vma;
132 }
133
134 static void m_cache_vma(struct seq_file *m, struct vm_area_struct *vma)
135 {
136         if (m->count < m->size) /* vma is copied successfully */
137                 m->version = m_next_vma(m->private, vma) ? vma->vm_start : -1UL;
138 }
139
140 static void *m_start(struct seq_file *m, loff_t *ppos)
141 {
142         struct proc_maps_private *priv = m->private;
143         unsigned long last_addr = m->version;
144         struct mm_struct *mm;
145         struct vm_area_struct *vma;
146         unsigned int pos = *ppos;
147
148         /* See m_cache_vma(). Zero at the start or after lseek. */
149         if (last_addr == -1UL)
150                 return NULL;
151
152         priv->task = get_proc_task(priv->inode);
153         if (!priv->task)
154                 return ERR_PTR(-ESRCH);
155
156         mm = priv->mm;
157         if (!mm || !atomic_inc_not_zero(&mm->mm_users))
158                 return NULL;
159
160         down_read(&mm->mmap_sem);
161         hold_task_mempolicy(priv);
162         priv->tail_vma = get_gate_vma(mm);
163
164         if (last_addr) {
165                 vma = find_vma(mm, last_addr);
166                 if (vma && (vma = m_next_vma(priv, vma)))
167                         return vma;
168         }
169
170         m->version = 0;
171         if (pos < mm->map_count) {
172                 for (vma = mm->mmap; pos; pos--) {
173                         m->version = vma->vm_start;
174                         vma = vma->vm_next;
175                 }
176                 return vma;
177         }
178
179         /* we do not bother to update m->version in this case */
180         if (pos == mm->map_count && priv->tail_vma)
181                 return priv->tail_vma;
182
183         vma_stop(priv);
184         return NULL;
185 }
186
187 static void *m_next(struct seq_file *m, void *v, loff_t *pos)
188 {
189         struct proc_maps_private *priv = m->private;
190         struct vm_area_struct *next;
191
192         (*pos)++;
193         next = m_next_vma(priv, v);
194         if (!next)
195                 vma_stop(priv);
196         return next;
197 }
198
199 static void m_stop(struct seq_file *m, void *v)
200 {
201         struct proc_maps_private *priv = m->private;
202
203         if (!IS_ERR_OR_NULL(v))
204                 vma_stop(priv);
205         if (priv->task) {
206                 put_task_struct(priv->task);
207                 priv->task = NULL;
208         }
209 }
210
211 static int proc_maps_open(struct inode *inode, struct file *file,
212                         const struct seq_operations *ops, int psize)
213 {
214         struct proc_maps_private *priv = __seq_open_private(file, ops, psize);
215
216         if (!priv)
217                 return -ENOMEM;
218
219         priv->inode = inode;
220         priv->mm = proc_mem_open(inode, PTRACE_MODE_READ);
221         if (IS_ERR(priv->mm)) {
222                 int err = PTR_ERR(priv->mm);
223
224                 seq_release_private(inode, file);
225                 return err;
226         }
227
228         return 0;
229 }
230
231 static int proc_map_release(struct inode *inode, struct file *file)
232 {
233         struct seq_file *seq = file->private_data;
234         struct proc_maps_private *priv = seq->private;
235
236         if (priv->mm)
237                 mmdrop(priv->mm);
238
239         return seq_release_private(inode, file);
240 }
241
242 static int do_maps_open(struct inode *inode, struct file *file,
243                         const struct seq_operations *ops)
244 {
245         return proc_maps_open(inode, file, ops,
246                                 sizeof(struct proc_maps_private));
247 }
248
249 static pid_t pid_of_stack(struct proc_maps_private *priv,
250                                 struct vm_area_struct *vma, bool is_pid)
251 {
252         struct inode *inode = priv->inode;
253         struct task_struct *task;
254         pid_t ret = 0;
255
256         rcu_read_lock();
257         task = pid_task(proc_pid(inode), PIDTYPE_PID);
258         if (task) {
259                 task = task_of_stack(task, vma, is_pid);
260                 if (task)
261                         ret = task_pid_nr_ns(task, inode->i_sb->s_fs_info);
262         }
263         rcu_read_unlock();
264
265         return ret;
266 }
267
268 static void
269 show_map_vma(struct seq_file *m, struct vm_area_struct *vma, int is_pid)
270 {
271         struct mm_struct *mm = vma->vm_mm;
272         struct file *file = vma->vm_file;
273         struct proc_maps_private *priv = m->private;
274         vm_flags_t flags = vma->vm_flags;
275         unsigned long ino = 0;
276         unsigned long long pgoff = 0;
277         unsigned long start, end;
278         dev_t dev = 0;
279         const char *name = NULL;
280
281         if (file) {
282                 struct inode *inode = file_inode(vma->vm_file);
283                 dev = inode->i_sb->s_dev;
284                 ino = inode->i_ino;
285                 pgoff = ((loff_t)vma->vm_pgoff) << PAGE_SHIFT;
286         }
287
288         /* We don't show the stack guard page in /proc/maps */
289         start = vma->vm_start;
290         if (stack_guard_page_start(vma, start))
291                 start += PAGE_SIZE;
292         end = vma->vm_end;
293         if (stack_guard_page_end(vma, end))
294                 end -= PAGE_SIZE;
295
296         seq_setwidth(m, 25 + sizeof(void *) * 6 - 1);
297         seq_printf(m, "%08lx-%08lx %c%c%c%c %08llx %02x:%02x %lu ",
298                         start,
299                         end,
300                         flags & VM_READ ? 'r' : '-',
301                         flags & VM_WRITE ? 'w' : '-',
302                         flags & VM_EXEC ? 'x' : '-',
303                         flags & VM_MAYSHARE ? 's' : 'p',
304                         pgoff,
305                         MAJOR(dev), MINOR(dev), ino);
306
307         /*
308          * Print the dentry name for named mappings, and a
309          * special [heap] marker for the heap:
310          */
311         if (file) {
312                 seq_pad(m, ' ');
313                 seq_path(m, &file->f_path, "\n");
314                 goto done;
315         }
316
317         if (vma->vm_ops && vma->vm_ops->name) {
318                 name = vma->vm_ops->name(vma);
319                 if (name)
320                         goto done;
321         }
322
323         name = arch_vma_name(vma);
324         if (!name) {
325                 pid_t tid;
326
327                 if (!mm) {
328                         name = "[vdso]";
329                         goto done;
330                 }
331
332                 if (vma->vm_start <= mm->brk &&
333                     vma->vm_end >= mm->start_brk) {
334                         name = "[heap]";
335                         goto done;
336                 }
337
338                 tid = pid_of_stack(priv, vma, is_pid);
339                 if (tid != 0) {
340                         /*
341                          * Thread stack in /proc/PID/task/TID/maps or
342                          * the main process stack.
343                          */
344                         if (!is_pid || (vma->vm_start <= mm->start_stack &&
345                             vma->vm_end >= mm->start_stack)) {
346                                 name = "[stack]";
347                         } else {
348                                 /* Thread stack in /proc/PID/maps */
349                                 seq_pad(m, ' ');
350                                 seq_printf(m, "[stack:%d]", tid);
351                         }
352                 }
353         }
354
355 done:
356         if (name) {
357                 seq_pad(m, ' ');
358                 seq_puts(m, name);
359         }
360         seq_putc(m, '\n');
361 }
362
363 static int show_map(struct seq_file *m, void *v, int is_pid)
364 {
365         show_map_vma(m, v, is_pid);
366         m_cache_vma(m, v);
367         return 0;
368 }
369
370 static int show_pid_map(struct seq_file *m, void *v)
371 {
372         return show_map(m, v, 1);
373 }
374
375 static int show_tid_map(struct seq_file *m, void *v)
376 {
377         return show_map(m, v, 0);
378 }
379
380 static const struct seq_operations proc_pid_maps_op = {
381         .start  = m_start,
382         .next   = m_next,
383         .stop   = m_stop,
384         .show   = show_pid_map
385 };
386
387 static const struct seq_operations proc_tid_maps_op = {
388         .start  = m_start,
389         .next   = m_next,
390         .stop   = m_stop,
391         .show   = show_tid_map
392 };
393
394 static int pid_maps_open(struct inode *inode, struct file *file)
395 {
396         return do_maps_open(inode, file, &proc_pid_maps_op);
397 }
398
399 static int tid_maps_open(struct inode *inode, struct file *file)
400 {
401         return do_maps_open(inode, file, &proc_tid_maps_op);
402 }
403
404 const struct file_operations proc_pid_maps_operations = {
405         .open           = pid_maps_open,
406         .read           = seq_read,
407         .llseek         = seq_lseek,
408         .release        = proc_map_release,
409 };
410
411 const struct file_operations proc_tid_maps_operations = {
412         .open           = tid_maps_open,
413         .read           = seq_read,
414         .llseek         = seq_lseek,
415         .release        = proc_map_release,
416 };
417
418 /*
419  * Proportional Set Size(PSS): my share of RSS.
420  *
421  * PSS of a process is the count of pages it has in memory, where each
422  * page is divided by the number of processes sharing it.  So if a
423  * process has 1000 pages all to itself, and 1000 shared with one other
424  * process, its PSS will be 1500.
425  *
426  * To keep (accumulated) division errors low, we adopt a 64bit
427  * fixed-point pss counter to minimize division errors. So (pss >>
428  * PSS_SHIFT) would be the real byte count.
429  *
430  * A shift of 12 before division means (assuming 4K page size):
431  *      - 1M 3-user-pages add up to 8KB errors;
432  *      - supports mapcount up to 2^24, or 16M;
433  *      - supports PSS up to 2^52 bytes, or 4PB.
434  */
435 #define PSS_SHIFT 12
436
437 #ifdef CONFIG_PROC_PAGE_MONITOR
438 struct mem_size_stats {
439         unsigned long resident;
440         unsigned long shared_clean;
441         unsigned long shared_dirty;
442         unsigned long private_clean;
443         unsigned long private_dirty;
444         unsigned long referenced;
445         unsigned long anonymous;
446         unsigned long anonymous_thp;
447         unsigned long swap;
448         u64 pss;
449 };
450
451 static void smaps_account(struct mem_size_stats *mss, struct page *page,
452                 unsigned long size, bool young, bool dirty)
453 {
454         int mapcount;
455
456         if (PageAnon(page))
457                 mss->anonymous += size;
458
459         mss->resident += size;
460         /* Accumulate the size in pages that have been accessed. */
461         if (young || PageReferenced(page))
462                 mss->referenced += size;
463         mapcount = page_mapcount(page);
464         if (mapcount >= 2) {
465                 u64 pss_delta;
466
467                 if (dirty || PageDirty(page))
468                         mss->shared_dirty += size;
469                 else
470                         mss->shared_clean += size;
471                 pss_delta = (u64)size << PSS_SHIFT;
472                 do_div(pss_delta, mapcount);
473                 mss->pss += pss_delta;
474         } else {
475                 if (dirty || PageDirty(page))
476                         mss->private_dirty += size;
477                 else
478                         mss->private_clean += size;
479                 mss->pss += (u64)size << PSS_SHIFT;
480         }
481 }
482
483 static void smaps_pte_entry(pte_t *pte, unsigned long addr,
484                 struct mm_walk *walk)
485 {
486         struct mem_size_stats *mss = walk->private;
487         struct vm_area_struct *vma = walk->vma;
488         struct page *page = NULL;
489
490         if (pte_present(*pte)) {
491                 page = vm_normal_page(vma, addr, *pte);
492         } else if (is_swap_pte(*pte)) {
493                 swp_entry_t swpent = pte_to_swp_entry(*pte);
494
495                 if (!non_swap_entry(swpent))
496                         mss->swap += PAGE_SIZE;
497                 else if (is_migration_entry(swpent))
498                         page = migration_entry_to_page(swpent);
499         }
500
501         if (!page)
502                 return;
503         smaps_account(mss, page, PAGE_SIZE, pte_young(*pte), pte_dirty(*pte));
504 }
505
506 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
507 static void smaps_pmd_entry(pmd_t *pmd, unsigned long addr,
508                 struct mm_walk *walk)
509 {
510         struct mem_size_stats *mss = walk->private;
511         struct vm_area_struct *vma = walk->vma;
512         struct page *page;
513
514         /* FOLL_DUMP will return -EFAULT on huge zero page */
515         page = follow_trans_huge_pmd(vma, addr, pmd, FOLL_DUMP);
516         if (IS_ERR_OR_NULL(page))
517                 return;
518         mss->anonymous_thp += HPAGE_PMD_SIZE;
519         smaps_account(mss, page, HPAGE_PMD_SIZE,
520                         pmd_young(*pmd), pmd_dirty(*pmd));
521 }
522 #else
523 static void smaps_pmd_entry(pmd_t *pmd, unsigned long addr,
524                 struct mm_walk *walk)
525 {
526 }
527 #endif
528
529 static int smaps_pte_range(pmd_t *pmd, unsigned long addr, unsigned long end,
530                            struct mm_walk *walk)
531 {
532         struct vm_area_struct *vma = walk->vma;
533         pte_t *pte;
534         spinlock_t *ptl;
535
536         if (pmd_trans_huge_lock(pmd, vma, &ptl) == 1) {
537                 smaps_pmd_entry(pmd, addr, walk);
538                 spin_unlock(ptl);
539                 return 0;
540         }
541
542         if (pmd_trans_unstable(pmd))
543                 return 0;
544         /*
545          * The mmap_sem held all the way back in m_start() is what
546          * keeps khugepaged out of here and from collapsing things
547          * in here.
548          */
549         pte = pte_offset_map_lock(vma->vm_mm, pmd, addr, &ptl);
550         for (; addr != end; pte++, addr += PAGE_SIZE)
551                 smaps_pte_entry(pte, addr, walk);
552         pte_unmap_unlock(pte - 1, ptl);
553         cond_resched();
554         return 0;
555 }
556
557 static void show_smap_vma_flags(struct seq_file *m, struct vm_area_struct *vma)
558 {
559         /*
560          * Don't forget to update Documentation/ on changes.
561          */
562         static const char mnemonics[BITS_PER_LONG][2] = {
563                 /*
564                  * In case if we meet a flag we don't know about.
565                  */
566                 [0 ... (BITS_PER_LONG-1)] = "??",
567
568                 [ilog2(VM_READ)]        = "rd",
569                 [ilog2(VM_WRITE)]       = "wr",
570                 [ilog2(VM_EXEC)]        = "ex",
571                 [ilog2(VM_SHARED)]      = "sh",
572                 [ilog2(VM_MAYREAD)]     = "mr",
573                 [ilog2(VM_MAYWRITE)]    = "mw",
574                 [ilog2(VM_MAYEXEC)]     = "me",
575                 [ilog2(VM_MAYSHARE)]    = "ms",
576                 [ilog2(VM_GROWSDOWN)]   = "gd",
577                 [ilog2(VM_PFNMAP)]      = "pf",
578                 [ilog2(VM_DENYWRITE)]   = "dw",
579 #ifdef CONFIG_X86_INTEL_MPX
580                 [ilog2(VM_MPX)]         = "mp",
581 #endif
582                 [ilog2(VM_LOCKED)]      = "lo",
583                 [ilog2(VM_IO)]          = "io",
584                 [ilog2(VM_SEQ_READ)]    = "sr",
585                 [ilog2(VM_RAND_READ)]   = "rr",
586                 [ilog2(VM_DONTCOPY)]    = "dc",
587                 [ilog2(VM_DONTEXPAND)]  = "de",
588                 [ilog2(VM_ACCOUNT)]     = "ac",
589                 [ilog2(VM_NORESERVE)]   = "nr",
590                 [ilog2(VM_HUGETLB)]     = "ht",
591                 [ilog2(VM_ARCH_1)]      = "ar",
592                 [ilog2(VM_DONTDUMP)]    = "dd",
593 #ifdef CONFIG_MEM_SOFT_DIRTY
594                 [ilog2(VM_SOFTDIRTY)]   = "sd",
595 #endif
596                 [ilog2(VM_MIXEDMAP)]    = "mm",
597                 [ilog2(VM_HUGEPAGE)]    = "hg",
598                 [ilog2(VM_NOHUGEPAGE)]  = "nh",
599                 [ilog2(VM_MERGEABLE)]   = "mg",
600         };
601         size_t i;
602
603         seq_puts(m, "VmFlags: ");
604         for (i = 0; i < BITS_PER_LONG; i++) {
605                 if (vma->vm_flags & (1UL << i)) {
606                         seq_printf(m, "%c%c ",
607                                    mnemonics[i][0], mnemonics[i][1]);
608                 }
609         }
610         seq_putc(m, '\n');
611 }
612
613 static int show_smap(struct seq_file *m, void *v, int is_pid)
614 {
615         struct vm_area_struct *vma = v;
616         struct mem_size_stats mss;
617         struct mm_walk smaps_walk = {
618                 .pmd_entry = smaps_pte_range,
619                 .mm = vma->vm_mm,
620                 .private = &mss,
621         };
622
623         memset(&mss, 0, sizeof mss);
624         /* mmap_sem is held in m_start */
625         walk_page_vma(vma, &smaps_walk);
626
627         show_map_vma(m, vma, is_pid);
628
629         seq_printf(m,
630                    "Size:           %8lu kB\n"
631                    "Rss:            %8lu kB\n"
632                    "Pss:            %8lu kB\n"
633                    "Shared_Clean:   %8lu kB\n"
634                    "Shared_Dirty:   %8lu kB\n"
635                    "Private_Clean:  %8lu kB\n"
636                    "Private_Dirty:  %8lu kB\n"
637                    "Referenced:     %8lu kB\n"
638                    "Anonymous:      %8lu kB\n"
639                    "AnonHugePages:  %8lu kB\n"
640                    "Swap:           %8lu kB\n"
641                    "KernelPageSize: %8lu kB\n"
642                    "MMUPageSize:    %8lu kB\n"
643                    "Locked:         %8lu kB\n",
644                    (vma->vm_end - vma->vm_start) >> 10,
645                    mss.resident >> 10,
646                    (unsigned long)(mss.pss >> (10 + PSS_SHIFT)),
647                    mss.shared_clean  >> 10,
648                    mss.shared_dirty  >> 10,
649                    mss.private_clean >> 10,
650                    mss.private_dirty >> 10,
651                    mss.referenced >> 10,
652                    mss.anonymous >> 10,
653                    mss.anonymous_thp >> 10,
654                    mss.swap >> 10,
655                    vma_kernel_pagesize(vma) >> 10,
656                    vma_mmu_pagesize(vma) >> 10,
657                    (vma->vm_flags & VM_LOCKED) ?
658                         (unsigned long)(mss.pss >> (10 + PSS_SHIFT)) : 0);
659
660         show_smap_vma_flags(m, vma);
661         m_cache_vma(m, vma);
662         return 0;
663 }
664
665 static int show_pid_smap(struct seq_file *m, void *v)
666 {
667         return show_smap(m, v, 1);
668 }
669
670 static int show_tid_smap(struct seq_file *m, void *v)
671 {
672         return show_smap(m, v, 0);
673 }
674
675 static const struct seq_operations proc_pid_smaps_op = {
676         .start  = m_start,
677         .next   = m_next,
678         .stop   = m_stop,
679         .show   = show_pid_smap
680 };
681
682 static const struct seq_operations proc_tid_smaps_op = {
683         .start  = m_start,
684         .next   = m_next,
685         .stop   = m_stop,
686         .show   = show_tid_smap
687 };
688
689 static int pid_smaps_open(struct inode *inode, struct file *file)
690 {
691         return do_maps_open(inode, file, &proc_pid_smaps_op);
692 }
693
694 static int tid_smaps_open(struct inode *inode, struct file *file)
695 {
696         return do_maps_open(inode, file, &proc_tid_smaps_op);
697 }
698
699 const struct file_operations proc_pid_smaps_operations = {
700         .open           = pid_smaps_open,
701         .read           = seq_read,
702         .llseek         = seq_lseek,
703         .release        = proc_map_release,
704 };
705
706 const struct file_operations proc_tid_smaps_operations = {
707         .open           = tid_smaps_open,
708         .read           = seq_read,
709         .llseek         = seq_lseek,
710         .release        = proc_map_release,
711 };
712
713 /*
714  * We do not want to have constant page-shift bits sitting in
715  * pagemap entries and are about to reuse them some time soon.
716  *
717  * Here's the "migration strategy":
718  * 1. when the system boots these bits remain what they are,
719  *    but a warning about future change is printed in log;
720  * 2. once anyone clears soft-dirty bits via clear_refs file,
721  *    these flag is set to denote, that user is aware of the
722  *    new API and those page-shift bits change their meaning.
723  *    The respective warning is printed in dmesg;
724  * 3. In a couple of releases we will remove all the mentions
725  *    of page-shift in pagemap entries.
726  */
727
728 static bool soft_dirty_cleared __read_mostly;
729
730 enum clear_refs_types {
731         CLEAR_REFS_ALL = 1,
732         CLEAR_REFS_ANON,
733         CLEAR_REFS_MAPPED,
734         CLEAR_REFS_SOFT_DIRTY,
735         CLEAR_REFS_LAST,
736 };
737
738 struct clear_refs_private {
739         enum clear_refs_types type;
740 };
741
742 static inline void clear_soft_dirty(struct vm_area_struct *vma,
743                 unsigned long addr, pte_t *pte)
744 {
745 #ifdef CONFIG_MEM_SOFT_DIRTY
746         /*
747          * The soft-dirty tracker uses #PF-s to catch writes
748          * to pages, so write-protect the pte as well. See the
749          * Documentation/vm/soft-dirty.txt for full description
750          * of how soft-dirty works.
751          */
752         pte_t ptent = *pte;
753
754         if (pte_present(ptent)) {
755                 ptent = pte_wrprotect(ptent);
756                 ptent = pte_clear_flags(ptent, _PAGE_SOFT_DIRTY);
757         } else if (is_swap_pte(ptent)) {
758                 ptent = pte_swp_clear_soft_dirty(ptent);
759         }
760
761         set_pte_at(vma->vm_mm, addr, pte, ptent);
762 #endif
763 }
764
765 static int clear_refs_pte_range(pmd_t *pmd, unsigned long addr,
766                                 unsigned long end, struct mm_walk *walk)
767 {
768         struct clear_refs_private *cp = walk->private;
769         struct vm_area_struct *vma = walk->vma;
770         pte_t *pte, ptent;
771         spinlock_t *ptl;
772         struct page *page;
773
774         split_huge_page_pmd(vma, addr, pmd);
775         if (pmd_trans_unstable(pmd))
776                 return 0;
777
778         pte = pte_offset_map_lock(vma->vm_mm, pmd, addr, &ptl);
779         for (; addr != end; pte++, addr += PAGE_SIZE) {
780                 ptent = *pte;
781
782                 if (cp->type == CLEAR_REFS_SOFT_DIRTY) {
783                         clear_soft_dirty(vma, addr, pte);
784                         continue;
785                 }
786
787                 if (!pte_present(ptent))
788                         continue;
789
790                 page = vm_normal_page(vma, addr, ptent);
791                 if (!page)
792                         continue;
793
794                 /* Clear accessed and referenced bits. */
795                 ptep_test_and_clear_young(vma, addr, pte);
796                 ClearPageReferenced(page);
797         }
798         pte_unmap_unlock(pte - 1, ptl);
799         cond_resched();
800         return 0;
801 }
802
803 static int clear_refs_test_walk(unsigned long start, unsigned long end,
804                                 struct mm_walk *walk)
805 {
806         struct clear_refs_private *cp = walk->private;
807         struct vm_area_struct *vma = walk->vma;
808
809         /*
810          * Writing 1 to /proc/pid/clear_refs affects all pages.
811          * Writing 2 to /proc/pid/clear_refs only affects anonymous pages.
812          * Writing 3 to /proc/pid/clear_refs only affects file mapped pages.
813          * Writing 4 to /proc/pid/clear_refs affects all pages.
814          */
815         if (cp->type == CLEAR_REFS_ANON && vma->vm_file)
816                 return 1;
817         if (cp->type == CLEAR_REFS_MAPPED && !vma->vm_file)
818                 return 1;
819         return 0;
820 }
821
822 static ssize_t clear_refs_write(struct file *file, const char __user *buf,
823                                 size_t count, loff_t *ppos)
824 {
825         struct task_struct *task;
826         char buffer[PROC_NUMBUF];
827         struct mm_struct *mm;
828         struct vm_area_struct *vma;
829         enum clear_refs_types type;
830         int itype;
831         int rv;
832
833         memset(buffer, 0, sizeof(buffer));
834         if (count > sizeof(buffer) - 1)
835                 count = sizeof(buffer) - 1;
836         if (copy_from_user(buffer, buf, count))
837                 return -EFAULT;
838         rv = kstrtoint(strstrip(buffer), 10, &itype);
839         if (rv < 0)
840                 return rv;
841         type = (enum clear_refs_types)itype;
842         if (type < CLEAR_REFS_ALL || type >= CLEAR_REFS_LAST)
843                 return -EINVAL;
844
845         if (type == CLEAR_REFS_SOFT_DIRTY) {
846                 soft_dirty_cleared = true;
847                 pr_warn_once("The pagemap bits 55-60 has changed their meaning!"
848                              " See the linux/Documentation/vm/pagemap.txt for "
849                              "details.\n");
850         }
851
852         task = get_proc_task(file_inode(file));
853         if (!task)
854                 return -ESRCH;
855         mm = get_task_mm(task);
856         if (mm) {
857                 struct clear_refs_private cp = {
858                         .type = type,
859                 };
860                 struct mm_walk clear_refs_walk = {
861                         .pmd_entry = clear_refs_pte_range,
862                         .test_walk = clear_refs_test_walk,
863                         .mm = mm,
864                         .private = &cp,
865                 };
866                 down_read(&mm->mmap_sem);
867                 if (type == CLEAR_REFS_SOFT_DIRTY) {
868                         for (vma = mm->mmap; vma; vma = vma->vm_next) {
869                                 if (!(vma->vm_flags & VM_SOFTDIRTY))
870                                         continue;
871                                 up_read(&mm->mmap_sem);
872                                 down_write(&mm->mmap_sem);
873                                 for (vma = mm->mmap; vma; vma = vma->vm_next) {
874                                         vma->vm_flags &= ~VM_SOFTDIRTY;
875                                         vma_set_page_prot(vma);
876                                 }
877                                 downgrade_write(&mm->mmap_sem);
878                                 break;
879                         }
880                         mmu_notifier_invalidate_range_start(mm, 0, -1);
881                 }
882                 walk_page_range(0, ~0UL, &clear_refs_walk);
883                 if (type == CLEAR_REFS_SOFT_DIRTY)
884                         mmu_notifier_invalidate_range_end(mm, 0, -1);
885                 flush_tlb_mm(mm);
886                 up_read(&mm->mmap_sem);
887                 mmput(mm);
888         }
889         put_task_struct(task);
890
891         return count;
892 }
893
894 const struct file_operations proc_clear_refs_operations = {
895         .write          = clear_refs_write,
896         .llseek         = noop_llseek,
897 };
898
899 typedef struct {
900         u64 pme;
901 } pagemap_entry_t;
902
903 struct pagemapread {
904         int pos, len;           /* units: PM_ENTRY_BYTES, not bytes */
905         pagemap_entry_t *buffer;
906         bool v2;
907 };
908
909 #define PAGEMAP_WALK_SIZE       (PMD_SIZE)
910 #define PAGEMAP_WALK_MASK       (PMD_MASK)
911
912 #define PM_ENTRY_BYTES      sizeof(pagemap_entry_t)
913 #define PM_STATUS_BITS      3
914 #define PM_STATUS_OFFSET    (64 - PM_STATUS_BITS)
915 #define PM_STATUS_MASK      (((1LL << PM_STATUS_BITS) - 1) << PM_STATUS_OFFSET)
916 #define PM_STATUS(nr)       (((nr) << PM_STATUS_OFFSET) & PM_STATUS_MASK)
917 #define PM_PSHIFT_BITS      6
918 #define PM_PSHIFT_OFFSET    (PM_STATUS_OFFSET - PM_PSHIFT_BITS)
919 #define PM_PSHIFT_MASK      (((1LL << PM_PSHIFT_BITS) - 1) << PM_PSHIFT_OFFSET)
920 #define __PM_PSHIFT(x)      (((u64) (x) << PM_PSHIFT_OFFSET) & PM_PSHIFT_MASK)
921 #define PM_PFRAME_MASK      ((1LL << PM_PSHIFT_OFFSET) - 1)
922 #define PM_PFRAME(x)        ((x) & PM_PFRAME_MASK)
923 /* in "new" pagemap pshift bits are occupied with more status bits */
924 #define PM_STATUS2(v2, x)   (__PM_PSHIFT(v2 ? x : PAGE_SHIFT))
925
926 #define __PM_SOFT_DIRTY      (1LL)
927 #define PM_PRESENT          PM_STATUS(4LL)
928 #define PM_SWAP             PM_STATUS(2LL)
929 #define PM_FILE             PM_STATUS(1LL)
930 #define PM_NOT_PRESENT(v2)  PM_STATUS2(v2, 0)
931 #define PM_END_OF_BUFFER    1
932
933 static inline pagemap_entry_t make_pme(u64 val)
934 {
935         return (pagemap_entry_t) { .pme = val };
936 }
937
938 static int add_to_pagemap(unsigned long addr, pagemap_entry_t *pme,
939                           struct pagemapread *pm)
940 {
941         pm->buffer[pm->pos++] = *pme;
942         if (pm->pos >= pm->len)
943                 return PM_END_OF_BUFFER;
944         return 0;
945 }
946
947 static int pagemap_pte_hole(unsigned long start, unsigned long end,
948                                 struct mm_walk *walk)
949 {
950         struct pagemapread *pm = walk->private;
951         unsigned long addr = start;
952         int err = 0;
953
954         while (addr < end) {
955                 struct vm_area_struct *vma = find_vma(walk->mm, addr);
956                 pagemap_entry_t pme = make_pme(PM_NOT_PRESENT(pm->v2));
957                 /* End of address space hole, which we mark as non-present. */
958                 unsigned long hole_end;
959
960                 if (vma)
961                         hole_end = min(end, vma->vm_start);
962                 else
963                         hole_end = end;
964
965                 for (; addr < hole_end; addr += PAGE_SIZE) {
966                         err = add_to_pagemap(addr, &pme, pm);
967                         if (err)
968                                 goto out;
969                 }
970
971                 if (!vma)
972                         break;
973
974                 /* Addresses in the VMA. */
975                 if (vma->vm_flags & VM_SOFTDIRTY)
976                         pme.pme |= PM_STATUS2(pm->v2, __PM_SOFT_DIRTY);
977                 for (; addr < min(end, vma->vm_end); addr += PAGE_SIZE) {
978                         err = add_to_pagemap(addr, &pme, pm);
979                         if (err)
980                                 goto out;
981                 }
982         }
983 out:
984         return err;
985 }
986
987 static void pte_to_pagemap_entry(pagemap_entry_t *pme, struct pagemapread *pm,
988                 struct vm_area_struct *vma, unsigned long addr, pte_t pte)
989 {
990         u64 frame, flags;
991         struct page *page = NULL;
992         int flags2 = 0;
993
994         if (pte_present(pte)) {
995                 frame = pte_pfn(pte);
996                 flags = PM_PRESENT;
997                 page = vm_normal_page(vma, addr, pte);
998                 if (pte_soft_dirty(pte))
999                         flags2 |= __PM_SOFT_DIRTY;
1000         } else if (is_swap_pte(pte)) {
1001                 swp_entry_t entry;
1002                 if (pte_swp_soft_dirty(pte))
1003                         flags2 |= __PM_SOFT_DIRTY;
1004                 entry = pte_to_swp_entry(pte);
1005                 frame = swp_type(entry) |
1006                         (swp_offset(entry) << MAX_SWAPFILES_SHIFT);
1007                 flags = PM_SWAP;
1008                 if (is_migration_entry(entry))
1009                         page = migration_entry_to_page(entry);
1010         } else {
1011                 if (vma->vm_flags & VM_SOFTDIRTY)
1012                         flags2 |= __PM_SOFT_DIRTY;
1013                 *pme = make_pme(PM_NOT_PRESENT(pm->v2) | PM_STATUS2(pm->v2, flags2));
1014                 return;
1015         }
1016
1017         if (page && !PageAnon(page))
1018                 flags |= PM_FILE;
1019         if ((vma->vm_flags & VM_SOFTDIRTY))
1020                 flags2 |= __PM_SOFT_DIRTY;
1021
1022         *pme = make_pme(PM_PFRAME(frame) | PM_STATUS2(pm->v2, flags2) | flags);
1023 }
1024
1025 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
1026 static void thp_pmd_to_pagemap_entry(pagemap_entry_t *pme, struct pagemapread *pm,
1027                 pmd_t pmd, int offset, int pmd_flags2)
1028 {
1029         /*
1030          * Currently pmd for thp is always present because thp can not be
1031          * swapped-out, migrated, or HWPOISONed (split in such cases instead.)
1032          * This if-check is just to prepare for future implementation.
1033          */
1034         if (pmd_present(pmd))
1035                 *pme = make_pme(PM_PFRAME(pmd_pfn(pmd) + offset)
1036                                 | PM_STATUS2(pm->v2, pmd_flags2) | PM_PRESENT);
1037         else
1038                 *pme = make_pme(PM_NOT_PRESENT(pm->v2) | PM_STATUS2(pm->v2, pmd_flags2));
1039 }
1040 #else
1041 static inline void thp_pmd_to_pagemap_entry(pagemap_entry_t *pme, struct pagemapread *pm,
1042                 pmd_t pmd, int offset, int pmd_flags2)
1043 {
1044 }
1045 #endif
1046
1047 static int pagemap_pte_range(pmd_t *pmd, unsigned long addr, unsigned long end,
1048                              struct mm_walk *walk)
1049 {
1050         struct vm_area_struct *vma = walk->vma;
1051         struct pagemapread *pm = walk->private;
1052         spinlock_t *ptl;
1053         pte_t *pte, *orig_pte;
1054         int err = 0;
1055
1056         if (pmd_trans_huge_lock(pmd, vma, &ptl) == 1) {
1057                 int pmd_flags2;
1058
1059                 if ((vma->vm_flags & VM_SOFTDIRTY) || pmd_soft_dirty(*pmd))
1060                         pmd_flags2 = __PM_SOFT_DIRTY;
1061                 else
1062                         pmd_flags2 = 0;
1063
1064                 for (; addr != end; addr += PAGE_SIZE) {
1065                         unsigned long offset;
1066                         pagemap_entry_t pme;
1067
1068                         offset = (addr & ~PAGEMAP_WALK_MASK) >>
1069                                         PAGE_SHIFT;
1070                         thp_pmd_to_pagemap_entry(&pme, pm, *pmd, offset, pmd_flags2);
1071                         err = add_to_pagemap(addr, &pme, pm);
1072                         if (err)
1073                                 break;
1074                 }
1075                 spin_unlock(ptl);
1076                 return err;
1077         }
1078
1079         if (pmd_trans_unstable(pmd))
1080                 return 0;
1081
1082         /*
1083          * We can assume that @vma always points to a valid one and @end never
1084          * goes beyond vma->vm_end.
1085          */
1086         orig_pte = pte = pte_offset_map_lock(walk->mm, pmd, addr, &ptl);
1087         for (; addr < end; pte++, addr += PAGE_SIZE) {
1088                 pagemap_entry_t pme;
1089
1090                 pte_to_pagemap_entry(&pme, pm, vma, addr, *pte);
1091                 err = add_to_pagemap(addr, &pme, pm);
1092                 if (err)
1093                         break;
1094         }
1095         pte_unmap_unlock(orig_pte, ptl);
1096
1097         cond_resched();
1098
1099         return err;
1100 }
1101
1102 #ifdef CONFIG_HUGETLB_PAGE
1103 static void huge_pte_to_pagemap_entry(pagemap_entry_t *pme, struct pagemapread *pm,
1104                                         pte_t pte, int offset, int flags2)
1105 {
1106         if (pte_present(pte))
1107                 *pme = make_pme(PM_PFRAME(pte_pfn(pte) + offset)        |
1108                                 PM_STATUS2(pm->v2, flags2)              |
1109                                 PM_PRESENT);
1110         else
1111                 *pme = make_pme(PM_NOT_PRESENT(pm->v2)                  |
1112                                 PM_STATUS2(pm->v2, flags2));
1113 }
1114
1115 /* This function walks within one hugetlb entry in the single call */
1116 static int pagemap_hugetlb_range(pte_t *pte, unsigned long hmask,
1117                                  unsigned long addr, unsigned long end,
1118                                  struct mm_walk *walk)
1119 {
1120         struct pagemapread *pm = walk->private;
1121         struct vm_area_struct *vma = walk->vma;
1122         int err = 0;
1123         int flags2;
1124         pagemap_entry_t pme;
1125
1126         if (vma->vm_flags & VM_SOFTDIRTY)
1127                 flags2 = __PM_SOFT_DIRTY;
1128         else
1129                 flags2 = 0;
1130
1131         for (; addr != end; addr += PAGE_SIZE) {
1132                 int offset = (addr & ~hmask) >> PAGE_SHIFT;
1133                 huge_pte_to_pagemap_entry(&pme, pm, *pte, offset, flags2);
1134                 err = add_to_pagemap(addr, &pme, pm);
1135                 if (err)
1136                         return err;
1137         }
1138
1139         cond_resched();
1140
1141         return err;
1142 }
1143 #endif /* HUGETLB_PAGE */
1144
1145 /*
1146  * /proc/pid/pagemap - an array mapping virtual pages to pfns
1147  *
1148  * For each page in the address space, this file contains one 64-bit entry
1149  * consisting of the following:
1150  *
1151  * Bits 0-54  page frame number (PFN) if present
1152  * Bits 0-4   swap type if swapped
1153  * Bits 5-54  swap offset if swapped
1154  * Bits 55-60 page shift (page size = 1<<page shift)
1155  * Bit  61    page is file-page or shared-anon
1156  * Bit  62    page swapped
1157  * Bit  63    page present
1158  *
1159  * If the page is not present but in swap, then the PFN contains an
1160  * encoding of the swap file number and the page's offset into the
1161  * swap. Unmapped pages return a null PFN. This allows determining
1162  * precisely which pages are mapped (or in swap) and comparing mapped
1163  * pages between processes.
1164  *
1165  * Efficient users of this interface will use /proc/pid/maps to
1166  * determine which areas of memory are actually mapped and llseek to
1167  * skip over unmapped regions.
1168  */
1169 static ssize_t pagemap_read(struct file *file, char __user *buf,
1170                             size_t count, loff_t *ppos)
1171 {
1172         struct task_struct *task = get_proc_task(file_inode(file));
1173         struct mm_struct *mm;
1174         struct pagemapread pm;
1175         int ret = -ESRCH;
1176         struct mm_walk pagemap_walk = {};
1177         unsigned long src;
1178         unsigned long svpfn;
1179         unsigned long start_vaddr;
1180         unsigned long end_vaddr;
1181         int copied = 0;
1182
1183         if (!task)
1184                 goto out;
1185
1186         ret = -EINVAL;
1187         /* file position must be aligned */
1188         if ((*ppos % PM_ENTRY_BYTES) || (count % PM_ENTRY_BYTES))
1189                 goto out_task;
1190
1191         ret = 0;
1192         if (!count)
1193                 goto out_task;
1194
1195         pm.v2 = soft_dirty_cleared;
1196         pm.len = (PAGEMAP_WALK_SIZE >> PAGE_SHIFT);
1197         pm.buffer = kmalloc(pm.len * PM_ENTRY_BYTES, GFP_TEMPORARY);
1198         ret = -ENOMEM;
1199         if (!pm.buffer)
1200                 goto out_task;
1201
1202         mm = mm_access(task, PTRACE_MODE_READ);
1203         ret = PTR_ERR(mm);
1204         if (!mm || IS_ERR(mm))
1205                 goto out_free;
1206
1207         pagemap_walk.pmd_entry = pagemap_pte_range;
1208         pagemap_walk.pte_hole = pagemap_pte_hole;
1209 #ifdef CONFIG_HUGETLB_PAGE
1210         pagemap_walk.hugetlb_entry = pagemap_hugetlb_range;
1211 #endif
1212         pagemap_walk.mm = mm;
1213         pagemap_walk.private = &pm;
1214
1215         src = *ppos;
1216         svpfn = src / PM_ENTRY_BYTES;
1217         start_vaddr = svpfn << PAGE_SHIFT;
1218         end_vaddr = TASK_SIZE_OF(task);
1219
1220         /* watch out for wraparound */
1221         if (svpfn > TASK_SIZE_OF(task) >> PAGE_SHIFT)
1222                 start_vaddr = end_vaddr;
1223
1224         /*
1225          * The odds are that this will stop walking way
1226          * before end_vaddr, because the length of the
1227          * user buffer is tracked in "pm", and the walk
1228          * will stop when we hit the end of the buffer.
1229          */
1230         ret = 0;
1231         while (count && (start_vaddr < end_vaddr)) {
1232                 int len;
1233                 unsigned long end;
1234
1235                 pm.pos = 0;
1236                 end = (start_vaddr + PAGEMAP_WALK_SIZE) & PAGEMAP_WALK_MASK;
1237                 /* overflow ? */
1238                 if (end < start_vaddr || end > end_vaddr)
1239                         end = end_vaddr;
1240                 down_read(&mm->mmap_sem);
1241                 ret = walk_page_range(start_vaddr, end, &pagemap_walk);
1242                 up_read(&mm->mmap_sem);
1243                 start_vaddr = end;
1244
1245                 len = min(count, PM_ENTRY_BYTES * pm.pos);
1246                 if (copy_to_user(buf, pm.buffer, len)) {
1247                         ret = -EFAULT;
1248                         goto out_mm;
1249                 }
1250                 copied += len;
1251                 buf += len;
1252                 count -= len;
1253         }
1254         *ppos += copied;
1255         if (!ret || ret == PM_END_OF_BUFFER)
1256                 ret = copied;
1257
1258 out_mm:
1259         mmput(mm);
1260 out_free:
1261         kfree(pm.buffer);
1262 out_task:
1263         put_task_struct(task);
1264 out:
1265         return ret;
1266 }
1267
1268 static int pagemap_open(struct inode *inode, struct file *file)
1269 {
1270         pr_warn_once("Bits 55-60 of /proc/PID/pagemap entries are about "
1271                         "to stop being page-shift some time soon. See the "
1272                         "linux/Documentation/vm/pagemap.txt for details.\n");
1273         return 0;
1274 }
1275
1276 const struct file_operations proc_pagemap_operations = {
1277         .llseek         = mem_lseek, /* borrow this */
1278         .read           = pagemap_read,
1279         .open           = pagemap_open,
1280 };
1281 #endif /* CONFIG_PROC_PAGE_MONITOR */
1282
1283 #ifdef CONFIG_NUMA
1284
1285 struct numa_maps {
1286         struct vm_area_struct *vma;
1287         unsigned long pages;
1288         unsigned long anon;
1289         unsigned long active;
1290         unsigned long writeback;
1291         unsigned long mapcount_max;
1292         unsigned long dirty;
1293         unsigned long swapcache;
1294         unsigned long node[MAX_NUMNODES];
1295 };
1296
1297 struct numa_maps_private {
1298         struct proc_maps_private proc_maps;
1299         struct numa_maps md;
1300 };
1301
1302 static void gather_stats(struct page *page, struct numa_maps *md, int pte_dirty,
1303                         unsigned long nr_pages)
1304 {
1305         int count = page_mapcount(page);
1306
1307         md->pages += nr_pages;
1308         if (pte_dirty || PageDirty(page))
1309                 md->dirty += nr_pages;
1310
1311         if (PageSwapCache(page))
1312                 md->swapcache += nr_pages;
1313
1314         if (PageActive(page) || PageUnevictable(page))
1315                 md->active += nr_pages;
1316
1317         if (PageWriteback(page))
1318                 md->writeback += nr_pages;
1319
1320         if (PageAnon(page))
1321                 md->anon += nr_pages;
1322
1323         if (count > md->mapcount_max)
1324                 md->mapcount_max = count;
1325
1326         md->node[page_to_nid(page)] += nr_pages;
1327 }
1328
1329 static struct page *can_gather_numa_stats(pte_t pte, struct vm_area_struct *vma,
1330                 unsigned long addr)
1331 {
1332         struct page *page;
1333         int nid;
1334
1335         if (!pte_present(pte))
1336                 return NULL;
1337
1338         page = vm_normal_page(vma, addr, pte);
1339         if (!page)
1340                 return NULL;
1341
1342         if (PageReserved(page))
1343                 return NULL;
1344
1345         nid = page_to_nid(page);
1346         if (!node_isset(nid, node_states[N_MEMORY]))
1347                 return NULL;
1348
1349         return page;
1350 }
1351
1352 static int gather_pte_stats(pmd_t *pmd, unsigned long addr,
1353                 unsigned long end, struct mm_walk *walk)
1354 {
1355         struct numa_maps *md;
1356         spinlock_t *ptl;
1357         pte_t *orig_pte;
1358         pte_t *pte;
1359
1360         md = walk->private;
1361
1362         if (pmd_trans_huge_lock(pmd, md->vma, &ptl) == 1) {
1363                 pte_t huge_pte = *(pte_t *)pmd;
1364                 struct page *page;
1365
1366                 page = can_gather_numa_stats(huge_pte, md->vma, addr);
1367                 if (page)
1368                         gather_stats(page, md, pte_dirty(huge_pte),
1369                                      HPAGE_PMD_SIZE/PAGE_SIZE);
1370                 spin_unlock(ptl);
1371                 return 0;
1372         }
1373
1374         if (pmd_trans_unstable(pmd))
1375                 return 0;
1376         orig_pte = pte = pte_offset_map_lock(walk->mm, pmd, addr, &ptl);
1377         do {
1378                 struct page *page = can_gather_numa_stats(*pte, md->vma, addr);
1379                 if (!page)
1380                         continue;
1381                 gather_stats(page, md, pte_dirty(*pte), 1);
1382
1383         } while (pte++, addr += PAGE_SIZE, addr != end);
1384         pte_unmap_unlock(orig_pte, ptl);
1385         return 0;
1386 }
1387 #ifdef CONFIG_HUGETLB_PAGE
1388 static int gather_hugetbl_stats(pte_t *pte, unsigned long hmask,
1389                 unsigned long addr, unsigned long end, struct mm_walk *walk)
1390 {
1391         struct numa_maps *md;
1392         struct page *page;
1393
1394         if (!pte_present(*pte))
1395                 return 0;
1396
1397         page = pte_page(*pte);
1398         if (!page)
1399                 return 0;
1400
1401         md = walk->private;
1402         gather_stats(page, md, pte_dirty(*pte), 1);
1403         return 0;
1404 }
1405
1406 #else
1407 static int gather_hugetbl_stats(pte_t *pte, unsigned long hmask,
1408                 unsigned long addr, unsigned long end, struct mm_walk *walk)
1409 {
1410         return 0;
1411 }
1412 #endif
1413
1414 /*
1415  * Display pages allocated per node and memory policy via /proc.
1416  */
1417 static int show_numa_map(struct seq_file *m, void *v, int is_pid)
1418 {
1419         struct numa_maps_private *numa_priv = m->private;
1420         struct proc_maps_private *proc_priv = &numa_priv->proc_maps;
1421         struct vm_area_struct *vma = v;
1422         struct numa_maps *md = &numa_priv->md;
1423         struct file *file = vma->vm_file;
1424         struct mm_struct *mm = vma->vm_mm;
1425         struct mm_walk walk = {};
1426         struct mempolicy *pol;
1427         char buffer[64];
1428         int nid;
1429
1430         if (!mm)
1431                 return 0;
1432
1433         /* Ensure we start with an empty set of numa_maps statistics. */
1434         memset(md, 0, sizeof(*md));
1435
1436         md->vma = vma;
1437
1438         walk.hugetlb_entry = gather_hugetbl_stats;
1439         walk.pmd_entry = gather_pte_stats;
1440         walk.private = md;
1441         walk.mm = mm;
1442
1443         pol = __get_vma_policy(vma, vma->vm_start);
1444         if (pol) {
1445                 mpol_to_str(buffer, sizeof(buffer), pol);
1446                 mpol_cond_put(pol);
1447         } else {
1448                 mpol_to_str(buffer, sizeof(buffer), proc_priv->task_mempolicy);
1449         }
1450
1451         seq_printf(m, "%08lx %s", vma->vm_start, buffer);
1452
1453         if (file) {
1454                 seq_puts(m, " file=");
1455                 seq_path(m, &file->f_path, "\n\t= ");
1456         } else if (vma->vm_start <= mm->brk && vma->vm_end >= mm->start_brk) {
1457                 seq_puts(m, " heap");
1458         } else {
1459                 pid_t tid = pid_of_stack(proc_priv, vma, is_pid);
1460                 if (tid != 0) {
1461                         /*
1462                          * Thread stack in /proc/PID/task/TID/maps or
1463                          * the main process stack.
1464                          */
1465                         if (!is_pid || (vma->vm_start <= mm->start_stack &&
1466                             vma->vm_end >= mm->start_stack))
1467                                 seq_puts(m, " stack");
1468                         else
1469                                 seq_printf(m, " stack:%d", tid);
1470                 }
1471         }
1472
1473         if (is_vm_hugetlb_page(vma))
1474                 seq_puts(m, " huge");
1475
1476         walk_page_range(vma->vm_start, vma->vm_end, &walk);
1477
1478         if (!md->pages)
1479                 goto out;
1480
1481         if (md->anon)
1482                 seq_printf(m, " anon=%lu", md->anon);
1483
1484         if (md->dirty)
1485                 seq_printf(m, " dirty=%lu", md->dirty);
1486
1487         if (md->pages != md->anon && md->pages != md->dirty)
1488                 seq_printf(m, " mapped=%lu", md->pages);
1489
1490         if (md->mapcount_max > 1)
1491                 seq_printf(m, " mapmax=%lu", md->mapcount_max);
1492
1493         if (md->swapcache)
1494                 seq_printf(m, " swapcache=%lu", md->swapcache);
1495
1496         if (md->active < md->pages && !is_vm_hugetlb_page(vma))
1497                 seq_printf(m, " active=%lu", md->active);
1498
1499         if (md->writeback)
1500                 seq_printf(m, " writeback=%lu", md->writeback);
1501
1502         for_each_node_state(nid, N_MEMORY)
1503                 if (md->node[nid])
1504                         seq_printf(m, " N%d=%lu", nid, md->node[nid]);
1505 out:
1506         seq_putc(m, '\n');
1507         m_cache_vma(m, vma);
1508         return 0;
1509 }
1510
1511 static int show_pid_numa_map(struct seq_file *m, void *v)
1512 {
1513         return show_numa_map(m, v, 1);
1514 }
1515
1516 static int show_tid_numa_map(struct seq_file *m, void *v)
1517 {
1518         return show_numa_map(m, v, 0);
1519 }
1520
1521 static const struct seq_operations proc_pid_numa_maps_op = {
1522         .start  = m_start,
1523         .next   = m_next,
1524         .stop   = m_stop,
1525         .show   = show_pid_numa_map,
1526 };
1527
1528 static const struct seq_operations proc_tid_numa_maps_op = {
1529         .start  = m_start,
1530         .next   = m_next,
1531         .stop   = m_stop,
1532         .show   = show_tid_numa_map,
1533 };
1534
1535 static int numa_maps_open(struct inode *inode, struct file *file,
1536                           const struct seq_operations *ops)
1537 {
1538         return proc_maps_open(inode, file, ops,
1539                                 sizeof(struct numa_maps_private));
1540 }
1541
1542 static int pid_numa_maps_open(struct inode *inode, struct file *file)
1543 {
1544         return numa_maps_open(inode, file, &proc_pid_numa_maps_op);
1545 }
1546
1547 static int tid_numa_maps_open(struct inode *inode, struct file *file)
1548 {
1549         return numa_maps_open(inode, file, &proc_tid_numa_maps_op);
1550 }
1551
1552 const struct file_operations proc_pid_numa_maps_operations = {
1553         .open           = pid_numa_maps_open,
1554         .read           = seq_read,
1555         .llseek         = seq_lseek,
1556         .release        = proc_map_release,
1557 };
1558
1559 const struct file_operations proc_tid_numa_maps_operations = {
1560         .open           = tid_numa_maps_open,
1561         .read           = seq_read,
1562         .llseek         = seq_lseek,
1563         .release        = proc_map_release,
1564 };
1565 #endif /* CONFIG_NUMA */