3 files changed, 11 insertions, 4 deletions

diff --git a/arch/x86/kernel/head64.c b/arch/x86/kernel/head64.c
index 816fc85c9bb3..f3b19685918e 100644
--- a/arch/x86/kernel/head64.c
+++ b/arch/x86/kernel/head64.c
@@ -148,9 +148,6 @@ void __init x86_64_start_kernel(char * real_mode_data)
         /* clear bss before set_intr_gate with early_idt_handler */
         clear_bss();
 
-        /* XXX - this is wrong... we need to build page tables from scratch */
-        max_pfn_mapped = KERNEL_IMAGE_SIZE >> PAGE_SHIFT;
-
         for (i = 0; i < NUM_EXCEPTION_VECTORS; i++) {
 #ifdef CONFIG_EARLY_PRINTK
                 set_intr_gate(i, &early_idt_handlers[i]);
diff --git a/arch/x86/kernel/setup.c b/arch/x86/kernel/setup.c
index db9c41dae8d7..d58083a2e158 100644
--- a/arch/x86/kernel/setup.c
+++ b/arch/x86/kernel/setup.c
@@ -996,8 +996,10 @@ void __init setup_arch(char **cmdline_p)
         setup_bios_corruption_check();
 #endif
 
+#ifdef CONFIG_X86_32
         printk(KERN_DEBUG "initial memory mapped: [mem 0x00000000-%#010lx]\n",
                         (max_pfn_mapped<<PAGE_SHIFT) - 1);
+#endif
 
         reserve_real_mode();
 
diff --git a/arch/x86/mm/init_64.c b/arch/x86/mm/init_64.c
index 9fbb85c8d930..dc67337d9bf0 100644
--- a/arch/x86/mm/init_64.c
+++ b/arch/x86/mm/init_64.c
@@ -378,10 +378,18 @@ void __init init_extra_mapping_uc(unsigned long phys, unsigned long size)
 void __init cleanup_highmap(void)
 {
         unsigned long vaddr = __START_KERNEL_map;
-        unsigned long vaddr_end = __START_KERNEL_map + (max_pfn_mapped << PAGE_SHIFT);
+        unsigned long vaddr_end = __START_KERNEL_map + KERNEL_IMAGE_SIZE;
         unsigned long end = roundup((unsigned long)_brk_end, PMD_SIZE) - 1;
         pmd_t *pmd = level2_kernel_pgt;
 
+        /*
+         * Native path, max_pfn_mapped is not set yet.
+         * Xen has valid max_pfn_mapped set in
+         *      arch/x86/xen/mmu.c:xen_setup_kernel_pagetable().
+         */
+        if (max_pfn_mapped)
+                vaddr_end = __START_KERNEL_map + (max_pfn_mapped << PAGE_SHIFT);
+
         for (; vaddr + PMD_SIZE - 1 < vaddr_end; pmd++, vaddr += PMD_SIZE) {
                 if (pmd_none(*pmd))
                         continue;