mm, uprobes: convert __replace_page() to use page_vma_mapped_walk()

For consistency, it worth converting all page_check_address() to
page_vma_mapped_walk(), so we could drop the former.

Link: http://lkml.kernel.org/r/20170129173858.45174-10-kirill.shutemov@linux.intel.com
Signed-off-by: Kirill A. Shutemov <kirill.shutemov@linux.intel.com>
Reviewed-by: Srikar Dronamraju <srikar@linux.vnet.ibm.com>
Cc: Andrea Arcangeli <aarcange@redhat.com>
Cc: Hillf Danton <hillf.zj@alibaba-inc.com>
Cc: Hugh Dickins <hughd@google.com>
Cc: Johannes Weiner <hannes@cmpxchg.org>
Cc: Oleg Nesterov <oleg@redhat.com>
Cc: Peter Zijlstra <peterz@infradead.org>
Cc: Rik van Riel <riel@redhat.com>
Cc: Vladimir Davydov <vdavydov.dev@gmail.com>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>
This commit is contained in:
Kirill A. Shutemov 2017-02-24 14:58:07 -08:00 committed by Linus Torvalds
parent 36eaff3364
commit 14fa2daa15

View File

@ -153,14 +153,19 @@ static int __replace_page(struct vm_area_struct *vma, unsigned long addr,
struct page *old_page, struct page *new_page) struct page *old_page, struct page *new_page)
{ {
struct mm_struct *mm = vma->vm_mm; struct mm_struct *mm = vma->vm_mm;
spinlock_t *ptl; struct page_vma_mapped_walk pvmw = {
pte_t *ptep; .page = old_page,
.vma = vma,
.address = addr,
};
int err; int err;
/* For mmu_notifiers */ /* For mmu_notifiers */
const unsigned long mmun_start = addr; const unsigned long mmun_start = addr;
const unsigned long mmun_end = addr + PAGE_SIZE; const unsigned long mmun_end = addr + PAGE_SIZE;
struct mem_cgroup *memcg; struct mem_cgroup *memcg;
VM_BUG_ON_PAGE(PageTransHuge(old_page), old_page);
err = mem_cgroup_try_charge(new_page, vma->vm_mm, GFP_KERNEL, &memcg, err = mem_cgroup_try_charge(new_page, vma->vm_mm, GFP_KERNEL, &memcg,
false); false);
if (err) if (err)
@ -171,11 +176,11 @@ static int __replace_page(struct vm_area_struct *vma, unsigned long addr,
mmu_notifier_invalidate_range_start(mm, mmun_start, mmun_end); mmu_notifier_invalidate_range_start(mm, mmun_start, mmun_end);
err = -EAGAIN; err = -EAGAIN;
ptep = page_check_address(old_page, mm, addr, &ptl, 0); if (!page_vma_mapped_walk(&pvmw)) {
if (!ptep) {
mem_cgroup_cancel_charge(new_page, memcg, false); mem_cgroup_cancel_charge(new_page, memcg, false);
goto unlock; goto unlock;
} }
VM_BUG_ON_PAGE(addr != pvmw.address, old_page);
get_page(new_page); get_page(new_page);
page_add_new_anon_rmap(new_page, vma, addr, false); page_add_new_anon_rmap(new_page, vma, addr, false);
@ -187,14 +192,15 @@ static int __replace_page(struct vm_area_struct *vma, unsigned long addr,
inc_mm_counter(mm, MM_ANONPAGES); inc_mm_counter(mm, MM_ANONPAGES);
} }
flush_cache_page(vma, addr, pte_pfn(*ptep)); flush_cache_page(vma, addr, pte_pfn(*pvmw.pte));
ptep_clear_flush_notify(vma, addr, ptep); ptep_clear_flush_notify(vma, addr, pvmw.pte);
set_pte_at_notify(mm, addr, ptep, mk_pte(new_page, vma->vm_page_prot)); set_pte_at_notify(mm, addr, pvmw.pte,
mk_pte(new_page, vma->vm_page_prot));
page_remove_rmap(old_page, false); page_remove_rmap(old_page, false);
if (!page_mapped(old_page)) if (!page_mapped(old_page))
try_to_free_swap(old_page); try_to_free_swap(old_page);
pte_unmap_unlock(ptep, ptl); page_vma_mapped_walk_done(&pvmw);
if (vma->vm_flags & VM_LOCKED) if (vma->vm_flags & VM_LOCKED)
munlock_vma_page(old_page); munlock_vma_page(old_page);