1 files changed, 14 insertions, 16 deletions

diff --git a/arch/microblaze/mm/init.c b/arch/microblaze/mm/init.c
index 65eb00419d19..c8437866d3b7 100644
--- a/arch/microblaze/mm/init.c
+++ b/arch/microblaze/mm/init.c
@@ -70,16 +70,16 @@ static void __init paging_init(void)
 
 void __init setup_memory(void)
 {
-        int i;
         unsigned long map_size;
+        struct memblock_region *reg;
+
 #ifndef CONFIG_MMU
         u32 kernel_align_start, kernel_align_size;
 
         /* Find main memory where is the kernel */
-        for (i = 0; i < memblock.memory.cnt; i++) {
-                memory_start = (u32) memblock.memory.region[i].base;
-                memory_end = (u32) memblock.memory.region[i].base
-                                + (u32) memblock.memory.region[i].size;
+        for_each_memblock(memory, reg) {
+                memory_start = (u32)reg->base;
+                memory_end = (u32) reg->base + reg->size;
                 if ((memory_start <= (u32)_text) &&
                                         ((u32)_text <= memory_end)) {
                         memory_size = memory_end - memory_start;
@@ -142,12 +142,10 @@ void __init setup_memory(void)
         free_bootmem(memory_start, memory_size);
 
         /* reserve allocate blocks */
-        for (i = 0; i < memblock.reserved.cnt; i++) {
-                pr_debug("reserved %d - 0x%08x-0x%08x\n", i,
-                        (u32) memblock.reserved.region[i].base,
-                        (u32) memblock_size_bytes(&memblock.reserved, i));
-                reserve_bootmem(memblock.reserved.region[i].base,
-                        memblock_size_bytes(&memblock.reserved, i) - 1, BOOTMEM_DEFAULT);
+        for_each_memblock(reserved, reg) {
+                pr_debug("reserved - 0x%08x-0x%08x\n",
+                         (u32) reg->base, (u32) reg->size);
+                reserve_bootmem(reg->base, reg->size, BOOTMEM_DEFAULT);
         }
 #ifdef CONFIG_MMU
         init_bootmem_done = 1;
@@ -230,7 +228,7 @@ static void mm_cmdline_setup(void)
                 if (maxmem && memory_size > maxmem) {
                         memory_size = maxmem;
                         memory_end = memory_start + memory_size;
-                        memblock.memory.region[0].size = memory_size;
+                        memblock.memory.regions[0].size = memory_size;
                 }
         }
 }
@@ -273,14 +271,14 @@ asmlinkage void __init mmu_init(void)
                 machine_restart(NULL);
         }
 
-        if ((u32) memblock.memory.region[0].size < 0x1000000) {
+        if ((u32) memblock.memory.regions[0].size < 0x1000000) {
                 printk(KERN_EMERG "Memory must be greater than 16MB\n");
                 machine_restart(NULL);
         }
         /* Find main memory where the kernel is */
-        memory_start = (u32) memblock.memory.region[0].base;
-        memory_end = (u32) memblock.memory.region[0].base +
-                                (u32) memblock.memory.region[0].size;
+        memory_start = (u32) memblock.memory.regions[0].base;
+        memory_end = (u32) memblock.memory.regions[0].base +
+                                (u32) memblock.memory.regions[0].size;
         memory_size = memory_end - memory_start;
 
         mm_cmdline_setup(); /* FIXME parse args from command line - not used */