1 files changed, 13 insertions, 10 deletions

diff --git a/arch/x86/mm/init_64.c b/arch/x86/mm/init_64.c
index fc18be0f6f29..2b6b4a3c8beb 100644
--- a/arch/x86/mm/init_64.c
+++ b/arch/x86/mm/init_64.c
@@ -407,12 +407,12 @@ static unsigned long __meminit
 phys_pmd_init(pmd_t *pmd_page, unsigned long address, unsigned long end,
               unsigned long page_size_mask, pgprot_t prot)
 {
-        unsigned long pages = 0;
+        unsigned long pages = 0, next;
         unsigned long last_map_addr = end;
 
         int i = pmd_index(address);
 
-        for (; i < PTRS_PER_PMD; i++, address += PMD_SIZE) {
+        for (; i < PTRS_PER_PMD; i++, address = next) {
                 unsigned long pte_phys;
                 pmd_t *pmd = pmd_page + pmd_index(address);
                 pte_t *pte;
@@ -426,6 +426,8 @@ phys_pmd_init(pmd_t *pmd_page, unsigned long address, unsigned long end,
                         break;
                 }
 
+                next = (address & PMD_MASK) + PMD_SIZE;
+
                 if (pmd_val(*pmd)) {
                         if (!pmd_large(*pmd)) {
                                 spin_lock(&init_mm.page_table_lock);
@@ -449,7 +451,7 @@ phys_pmd_init(pmd_t *pmd_page, unsigned long address, unsigned long end,
                          * attributes.
                          */
                         if (page_size_mask & (1 << PG_LEVEL_2M)) {
-                                pages++;
+                                last_map_addr = next;
                                 continue;
                         }
                         new_prot = pte_pgprot(pte_clrhuge(*(pte_t *)pmd));
@@ -462,7 +464,7 @@ phys_pmd_init(pmd_t *pmd_page, unsigned long address, unsigned long end,
                                 pfn_pte(address >> PAGE_SHIFT,
                                         __pgprot(pgprot_val(prot) | _PAGE_PSE)));
                         spin_unlock(&init_mm.page_table_lock);
-                        last_map_addr = (address & PMD_MASK) + PMD_SIZE;
+                        last_map_addr = next;
                         continue;
                 }
 
@@ -482,11 +484,11 @@ static unsigned long __meminit
 phys_pud_init(pud_t *pud_page, unsigned long addr, unsigned long end,
                          unsigned long page_size_mask)
 {
-        unsigned long pages = 0;
+        unsigned long pages = 0, next;
         unsigned long last_map_addr = end;
         int i = pud_index(addr);
 
-        for (; i < PTRS_PER_PUD; i++, addr = (addr & PUD_MASK) + PUD_SIZE) {
+        for (; i < PTRS_PER_PUD; i++, addr = next) {
                 unsigned long pmd_phys;
                 pud_t *pud = pud_page + pud_index(addr);
                 pmd_t *pmd;
@@ -495,8 +497,9 @@ phys_pud_init(pud_t *pud_page, unsigned long addr, unsigned long end,
                 if (addr >= end)
                         break;
 
-                if (!after_bootmem &&
-                                !e820_any_mapped(addr, addr+PUD_SIZE, 0)) {
+                next = (addr & PUD_MASK) + PUD_SIZE;
+
+                if (!after_bootmem && !e820_any_mapped(addr, next, 0)) {
                         set_pud(pud, __pud(0));
                         continue;
                 }
@@ -523,7 +526,7 @@ phys_pud_init(pud_t *pud_page, unsigned long addr, unsigned long end,
                          * attributes.
                          */
                         if (page_size_mask & (1 << PG_LEVEL_1G)) {
-                                pages++;
+                                last_map_addr = next;
                                 continue;
                         }
                         prot = pte_pgprot(pte_clrhuge(*(pte_t *)pud));
@@ -535,7 +538,7 @@ phys_pud_init(pud_t *pud_page, unsigned long addr, unsigned long end,
                         set_pte((pte_t *)pud,
                                 pfn_pte(addr >> PAGE_SHIFT, PAGE_KERNEL_LARGE));
                         spin_unlock(&init_mm.page_table_lock);
-                        last_map_addr = (addr & PUD_MASK) + PUD_SIZE;
+                        last_map_addr = next;
                         continue;
                 }