1 files changed, 12 insertions, 11 deletions

diff --git a/arch/powerpc/mm/pgtable_32.c b/arch/powerpc/mm/pgtable_32.c
index 573b3bd1c45b..a87ead0138b4 100644
--- a/arch/powerpc/mm/pgtable_32.c
+++ b/arch/powerpc/mm/pgtable_32.c
@@ -26,7 +26,8 @@
 #include <linux/vmalloc.h>
 #include <linux/init.h>
 #include <linux/highmem.h>
-#include <linux/lmb.h>
+#include <linux/memblock.h>
+#include <linux/slab.h>
 
 #include <asm/pgtable.h>
 #include <asm/pgalloc.h>
@@ -114,11 +115,7 @@ pgtable_t pte_alloc_one(struct mm_struct *mm, unsigned long address)
 {
         struct page *ptepage;
 
-#ifdef CONFIG_HIGHPTE
-        gfp_t flags = GFP_KERNEL | __GFP_HIGHMEM | __GFP_REPEAT | __GFP_ZERO;
-#else
         gfp_t flags = GFP_KERNEL | __GFP_REPEAT | __GFP_ZERO;
-#endif
 
         ptepage = alloc_pages(flags, 0);
         if (!ptepage)
@@ -145,6 +142,14 @@ ioremap_flags(phys_addr_t addr, unsigned long size, unsigned long flags)
         /* we don't want to let _PAGE_USER and _PAGE_EXEC leak out */
         flags &= ~(_PAGE_USER | _PAGE_EXEC);
 
+#ifdef _PAGE_BAP_SR
+        /* _PAGE_USER contains _PAGE_BAP_SR on BookE using the new PTE format
+         * which means that we just cleared supervisor access... oops ;-) This
+         * restores it
+         */
+        flags |= _PAGE_BAP_SR;
+#endif
+
         return __ioremap_caller(addr, size, flags, __builtin_return_address(0));
 }
 EXPORT_SYMBOL(ioremap_flags);
@@ -193,7 +198,7 @@ __ioremap_caller(phys_addr_t addr, unsigned long size, unsigned long flags,
          * mem_init() sets high_memory so only do the check after that.
          */
         if (mem_init_done && (p < virt_to_phys(high_memory)) &&
-            !(__allow_ioremap_reserved && lmb_is_region_reserved(p, size))) {
+            !(__allow_ioremap_reserved && memblock_is_region_reserved(p, size))) {
                 printk("__ioremap(): phys addr 0x%llx is RAM lr %p\n",
                        (unsigned long long)p, __builtin_return_address(0));
                 return NULL;
@@ -326,7 +331,7 @@ void __init mapin_ram(void)
                 s = mmu_mapin_ram(top);
                 __mapin_ram_chunk(s, top);
 
-                top = lmb_end_of_DRAM();
+                top = memblock_end_of_DRAM();
                 s = wii_mmu_mapin_mem2(top);
                 __mapin_ram_chunk(s, top);
         }
@@ -384,11 +389,7 @@ static int __change_page_attr(struct page *page, pgprot_t prot)
                 return -EINVAL;
         __set_pte_at(&init_mm, address, kpte, mk_pte(page, prot), 0);
         wmb();
-#ifdef CONFIG_PPC_STD_MMU
-        flush_hash_pages(0, address, pmd_val(*kpmd), 1);
-#else
         flush_tlb_page(NULL, address);
-#endif
         pte_unmap(kpte);
 
         return 0;