mm: account pmd page tables to the process
[cascardo/linux.git] / fs / proc / task_mmu.c
index f6734c6..e6e0abe 100644 (file)
@@ -21,7 +21,7 @@
 
 void task_mem(struct seq_file *m, struct mm_struct *mm)
 {
-       unsigned long data, text, lib, swap;
+       unsigned long data, text, lib, swap, ptes, pmds;
        unsigned long hiwater_vm, total_vm, hiwater_rss, total_rss;
 
        /*
@@ -42,6 +42,8 @@ void task_mem(struct seq_file *m, struct mm_struct *mm)
        text = (PAGE_ALIGN(mm->end_code) - (mm->start_code & PAGE_MASK)) >> 10;
        lib = (mm->exec_vm << (PAGE_SHIFT-10)) - text;
        swap = get_mm_counter(mm, MM_SWAPENTS);
+       ptes = PTRS_PER_PTE * sizeof(pte_t) * atomic_long_read(&mm->nr_ptes);
+       pmds = PTRS_PER_PMD * sizeof(pmd_t) * mm_nr_pmds(mm);
        seq_printf(m,
                "VmPeak:\t%8lu kB\n"
                "VmSize:\t%8lu kB\n"
@@ -54,6 +56,7 @@ void task_mem(struct seq_file *m, struct mm_struct *mm)
                "VmExe:\t%8lu kB\n"
                "VmLib:\t%8lu kB\n"
                "VmPTE:\t%8lu kB\n"
+               "VmPMD:\t%8lu kB\n"
                "VmSwap:\t%8lu kB\n",
                hiwater_vm << (PAGE_SHIFT-10),
                total_vm << (PAGE_SHIFT-10),
@@ -63,8 +66,8 @@ void task_mem(struct seq_file *m, struct mm_struct *mm)
                total_rss << (PAGE_SHIFT-10),
                data << (PAGE_SHIFT-10),
                mm->stack_vm << (PAGE_SHIFT-10), text, lib,
-               (PTRS_PER_PTE * sizeof(pte_t) *
-                atomic_long_read(&mm->nr_ptes)) >> 10,
+               ptes >> 10,
+               pmds >> 10,
                swap << (PAGE_SHIFT-10));
 }
 
@@ -443,62 +446,86 @@ struct mem_size_stats {
        unsigned long anonymous;
        unsigned long anonymous_thp;
        unsigned long swap;
-       unsigned long nonlinear;
        u64 pss;
 };
 
+static void smaps_account(struct mem_size_stats *mss, struct page *page,
+               unsigned long size, bool young, bool dirty)
+{
+       int mapcount;
+
+       if (PageAnon(page))
+               mss->anonymous += size;
+
+       mss->resident += size;
+       /* Accumulate the size in pages that have been accessed. */
+       if (young || PageReferenced(page))
+               mss->referenced += size;
+       mapcount = page_mapcount(page);
+       if (mapcount >= 2) {
+               u64 pss_delta;
+
+               if (dirty || PageDirty(page))
+                       mss->shared_dirty += size;
+               else
+                       mss->shared_clean += size;
+               pss_delta = (u64)size << PSS_SHIFT;
+               do_div(pss_delta, mapcount);
+               mss->pss += pss_delta;
+       } else {
+               if (dirty || PageDirty(page))
+                       mss->private_dirty += size;
+               else
+                       mss->private_clean += size;
+               mss->pss += (u64)size << PSS_SHIFT;
+       }
+}
 
-static void smaps_pte_entry(pte_t ptent, unsigned long addr,
-               unsigned long ptent_size, struct mm_walk *walk)
+static void smaps_pte_entry(pte_t *pte, unsigned long addr,
+               struct mm_walk *walk)
 {
        struct mem_size_stats *mss = walk->private;
        struct vm_area_struct *vma = mss->vma;
-       pgoff_t pgoff = linear_page_index(vma, addr);
        struct page *page = NULL;
-       int mapcount;
 
-       if (pte_present(ptent)) {
-               page = vm_normal_page(vma, addr, ptent);
-       } else if (is_swap_pte(ptent)) {
-               swp_entry_t swpent = pte_to_swp_entry(ptent);
+       if (pte_present(*pte)) {
+               page = vm_normal_page(vma, addr, *pte);
+       } else if (is_swap_pte(*pte)) {
+               swp_entry_t swpent = pte_to_swp_entry(*pte);
 
                if (!non_swap_entry(swpent))
-                       mss->swap += ptent_size;
+                       mss->swap += PAGE_SIZE;
                else if (is_migration_entry(swpent))
                        page = migration_entry_to_page(swpent);
-       } else if (pte_file(ptent)) {
-               if (pte_to_pgoff(ptent) != pgoff)
-                       mss->nonlinear += ptent_size;
        }
 
        if (!page)
                return;
+       smaps_account(mss, page, PAGE_SIZE, pte_young(*pte), pte_dirty(*pte));
+}
 
-       if (PageAnon(page))
-               mss->anonymous += ptent_size;
-
-       if (page->index != pgoff)
-               mss->nonlinear += ptent_size;
+#ifdef CONFIG_TRANSPARENT_HUGEPAGE
+static void smaps_pmd_entry(pmd_t *pmd, unsigned long addr,
+               struct mm_walk *walk)
+{
+       struct mem_size_stats *mss = walk->private;
+       struct vm_area_struct *vma = mss->vma;
+       struct page *page;
 
-       mss->resident += ptent_size;
-       /* Accumulate the size in pages that have been accessed. */
-       if (pte_young(ptent) || PageReferenced(page))
-               mss->referenced += ptent_size;
-       mapcount = page_mapcount(page);
-       if (mapcount >= 2) {
-               if (pte_dirty(ptent) || PageDirty(page))
-                       mss->shared_dirty += ptent_size;
-               else
-                       mss->shared_clean += ptent_size;
-               mss->pss += (ptent_size << PSS_SHIFT) / mapcount;
-       } else {
-               if (pte_dirty(ptent) || PageDirty(page))
-                       mss->private_dirty += ptent_size;
-               else
-                       mss->private_clean += ptent_size;
-               mss->pss += (ptent_size << PSS_SHIFT);
-       }
+       /* FOLL_DUMP will return -EFAULT on huge zero page */
+       page = follow_trans_huge_pmd(vma, addr, pmd, FOLL_DUMP);
+       if (IS_ERR_OR_NULL(page))
+               return;
+       mss->anonymous_thp += HPAGE_PMD_SIZE;
+       smaps_account(mss, page, HPAGE_PMD_SIZE,
+                       pmd_young(*pmd), pmd_dirty(*pmd));
 }
+#else
+static void smaps_pmd_entry(pmd_t *pmd, unsigned long addr,
+               struct mm_walk *walk)
+{
+}
+#endif
 
 static int smaps_pte_range(pmd_t *pmd, unsigned long addr, unsigned long end,
                           struct mm_walk *walk)
@@ -509,9 +536,8 @@ static int smaps_pte_range(pmd_t *pmd, unsigned long addr, unsigned long end,
        spinlock_t *ptl;
 
        if (pmd_trans_huge_lock(pmd, vma, &ptl) == 1) {
-               smaps_pte_entry(*(pte_t *)pmd, addr, HPAGE_PMD_SIZE, walk);
+               smaps_pmd_entry(pmd, addr, walk);
                spin_unlock(ptl);
-               mss->anonymous_thp += HPAGE_PMD_SIZE;
                return 0;
        }
 
@@ -524,7 +550,7 @@ static int smaps_pte_range(pmd_t *pmd, unsigned long addr, unsigned long end,
         */
        pte = pte_offset_map_lock(vma->vm_mm, pmd, addr, &ptl);
        for (; addr != end; pte++, addr += PAGE_SIZE)
-               smaps_pte_entry(*pte, addr, PAGE_SIZE, walk);
+               smaps_pte_entry(pte, addr, walk);
        pte_unmap_unlock(pte - 1, ptl);
        cond_resched();
        return 0;
@@ -564,7 +590,6 @@ static void show_smap_vma_flags(struct seq_file *m, struct vm_area_struct *vma)
                [ilog2(VM_ACCOUNT)]     = "ac",
                [ilog2(VM_NORESERVE)]   = "nr",
                [ilog2(VM_HUGETLB)]     = "ht",
-               [ilog2(VM_NONLINEAR)]   = "nl",
                [ilog2(VM_ARCH_1)]      = "ar",
                [ilog2(VM_DONTDUMP)]    = "dd",
 #ifdef CONFIG_MEM_SOFT_DIRTY
@@ -636,10 +661,6 @@ static int show_smap(struct seq_file *m, void *v, int is_pid)
                   (vma->vm_flags & VM_LOCKED) ?
                        (unsigned long)(mss.pss >> (10 + PSS_SHIFT)) : 0);
 
-       if (vma->vm_flags & VM_NONLINEAR)
-               seq_printf(m, "Nonlinear:      %8lu kB\n",
-                               mss.nonlinear >> 10);
-
        show_smap_vma_flags(m, vma);
        m_cache_vma(m, vma);
        return 0;
@@ -740,8 +761,6 @@ static inline void clear_soft_dirty(struct vm_area_struct *vma,
                ptent = pte_clear_flags(ptent, _PAGE_SOFT_DIRTY);
        } else if (is_swap_pte(ptent)) {
                ptent = pte_swp_clear_soft_dirty(ptent);
-       } else if (pte_file(ptent)) {
-               ptent = pte_file_clear_soft_dirty(ptent);
        }
 
        set_pte_at(vma->vm_mm, addr, pte, ptent);