powerpc/mm: Don't open code pgtable_t size

The slot information of base page size hash pte is stored in the
pgtable_t w.r.t transparent hugepage. We need to make sure we don't
index beyond pgtable_t size.

Signed-off-by: Aneesh Kumar K.V <[email protected]>
Signed-off-by: Michael Ellerman <[email protected]>

diff --git a/arch/powerpc/mm/hugepage-hash64.c b/arch/powerpc/mm/hugepage-hash64.c
index 1f666de0110acf80da3e3e3be067f1dfc58be44c..baf1301ded0ccf4ae5c3280923aa65985f42d8fe 100644 (file)
--- a/arch/powerpc/mm/hugepage-hash64.c
+++ b/arch/powerpc/mm/hugepage-hash64.c
@@ -71,7 +71,7 @@ int __hash_page_thp(unsigned long ea, unsigned long access, unsigned long vsid,
         */
        shift = mmu_psize_defs[psize].shift;
        index = (ea & ~HPAGE_PMD_MASK) >> shift;
-       BUG_ON(index >= 4096);
+       BUG_ON(index >= PTE_FRAG_SIZE);
 
        vpn = hpt_vpn(ea, vsid, ssize);
        hpte_slot_array = get_hpte_slot_array(pmdp);