2 files changed, 5 insertions, 4 deletions

diff --git a/arch/powerpc/mm/hash_utils_64.c b/arch/powerpc/mm/hash_utils_64.c
index 1f8b2a05e3d0..1628201c8cea 100644
--- a/arch/powerpc/mm/hash_utils_64.c
+++ b/arch/powerpc/mm/hash_utils_64.c
@@ -531,11 +531,11 @@ static unsigned long __init htab_get_table_size(void)
 }
 
 #ifdef CONFIG_MEMORY_HOTPLUG
-void create_section_mapping(unsigned long start, unsigned long end)
+int create_section_mapping(unsigned long start, unsigned long end)
 {
-        BUG_ON(htab_bolt_mapping(start, end, __pa(start),
+        return htab_bolt_mapping(start, end, __pa(start),
                                  pgprot_val(PAGE_KERNEL), mmu_linear_psize,
-                                 mmu_kernel_ssize));
+                                 mmu_kernel_ssize);
 }
 
 int remove_section_mapping(unsigned long start, unsigned long end)
diff --git a/arch/powerpc/mm/mem.c b/arch/powerpc/mm/mem.c
index ad9cf49dfb89..5db316cad47b 100644
--- a/arch/powerpc/mm/mem.c
+++ b/arch/powerpc/mm/mem.c
@@ -123,7 +123,8 @@ int arch_add_memory(int nid, u64 start, u64 size)
         pgdata = NODE_DATA(nid);
 
         start = (unsigned long)__va(start);
-        create_section_mapping(start, start + size);
+        if (create_section_mapping(start, start + size))
+                return -EINVAL;
 
         /* this should work for most non-highmem platforms */
         zone = pgdata->node_zones;