mm/hugetlb: add prot_modify_start/commit sequence for hugetlb update

Architectures like ppc64 require to do a conditional tlb flush based on
the old and new value of pte.  Follow the regular pte change protection
sequence for hugetlb too.  This allows the architectures to override the
update sequence.

Link: http://lkml.kernel.org/r/20190116085035.29729-5-aneesh.kumar@linux.ibm.com
Signed-off-by: Aneesh Kumar K.V <aneesh.kumar@linux.ibm.com>
Reviewed-by: Michael Ellerman <mpe@ellerman.id.au>
Cc: Benjamin Herrenschmidt <benh@kernel.crashing.org>
Cc: Heiko Carstens <heiko.carstens@de.ibm.com>
Cc: "H. Peter Anvin" <hpa@zytor.com>
Cc: Ingo Molnar <mingo@elte.hu>
Cc: Martin Schwidefsky <schwidefsky@de.ibm.com>
Cc: Nicholas Piggin <npiggin@gmail.com>
Cc: Paul Mackerras <paulus@samba.org>
Cc: Thomas Gleixner <tglx@linutronix.de>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>

1 files changed, 5 insertions, 3 deletions

diff --git a/mm/hugetlb.c b/mm/hugetlb.c
index 2fb3062a3595..0c7848fccf93 100644
--- a/mm/hugetlb.c
+++ b/mm/hugetlb.c
@@ -4399,10 +4399,12 @@ unsigned long hugetlb_change_protection(struct vm_area_struct *vma,
                         continue;
                 }
                 if (!huge_pte_none(pte)) {
-                        pte = huge_ptep_get_and_clear(mm, address, ptep);
-                        pte = pte_mkhuge(huge_pte_modify(pte, newprot));
+                        pte_t old_pte;
+
+                        old_pte = huge_ptep_modify_prot_start(vma, address, ptep);
+                        pte = pte_mkhuge(huge_pte_modify(old_pte, newprot));
                         pte = arch_make_huge_pte(pte, vma, NULL, 0);
-                        set_huge_pte_at(mm, address, ptep, pte);
+                        huge_ptep_modify_prot_commit(vma, address, ptep, old_pte, pte);
                         pages++;
                 }
                 spin_unlock(ptl);