x86: add PG_LEVEL enum

this way PG_LEVEL_1GB will be an easy change.

Signed-off-by: Ingo Molnar <mingo@elte.hu>
Signed-off-by: Thomas Gleixner <tglx@linutronix.de>

2 files changed, 7 insertions, 6 deletions

diff --git a/arch/x86/mm/pageattr-test.c b/arch/x86/mm/pageattr-test.c
index 6cc106b388a5..d7a93008cc12 100644
--- a/arch/x86/mm/pageattr-test.c
+++ b/arch/x86/mm/pageattr-test.c
@@ -16,14 +16,12 @@
 
 enum {
         NTEST                   = 400,
+        LOWEST_LEVEL            = PG_LEVEL_4K,
 #ifdef CONFIG_X86_64
-        LOWEST_LEVEL            = 4,
         LPS                     = (1 << PMD_SHIFT),
 #elif defined(CONFIG_X86_PAE)
-        LOWEST_LEVEL            = 4,
         LPS                     = (1 << PMD_SHIFT),
 #else
-        LOWEST_LEVEL            = 4, /* lookup_address lies here */
         LPS                     = (1 << 22),
 #endif
         GPS                     = (1<<30)
diff --git a/arch/x86/mm/pageattr.c b/arch/x86/mm/pageattr.c
index a270f9ccebfb..4589a1382fa1 100644
--- a/arch/x86/mm/pageattr.c
+++ b/arch/x86/mm/pageattr.c
@@ -28,6 +28,8 @@ pte_t *lookup_address(unsigned long address, int *level)
         pud_t *pud;
         pmd_t *pmd;
 
+        *level = PG_LEVEL_NONE;
+
         if (pgd_none(*pgd))
                 return NULL;
         pud = pud_offset(pgd, address);
@@ -36,11 +38,12 @@ pte_t *lookup_address(unsigned long address, int *level)
         pmd = pmd_offset(pud, address);
         if (pmd_none(*pmd))
                 return NULL;
-        *level = 3;
+
+        *level = PG_LEVEL_2M;
         if (pmd_large(*pmd))
                 return (pte_t *)pmd;
-        *level = 4;
 
+        *level = PG_LEVEL_4K;
         return pte_offset_kernel(pmd, address);
 }
 
@@ -145,7 +148,7 @@ repeat:
                 address < (unsigned long)&_etext &&
                (pgprot_val(prot) & _PAGE_NX));
 
-        if (level == 4) {
+        if (level == PG_LEVEL_4K) {
                 set_pte_atomic(kpte, mk_pte(page, canon_pgprot(prot)));
         } else {
                 err = split_large_page(kpte, address);