4 files changed, 47 insertions, 14 deletions

diff --git a/arch/mips/cavium-octeon/setup.c b/arch/mips/cavium-octeon/setup.c
index 2d9028f1474c..b394552e5ca0 100644
--- a/arch/mips/cavium-octeon/setup.c
+++ b/arch/mips/cavium-octeon/setup.c
@@ -642,14 +642,6 @@ void __init plat_mem_setup(void)
 
         total = 0;
 
-        /* First add the init memory we will be returning.  */
-        memory = __pa_symbol(&__init_begin) & PAGE_MASK;
-        mem_alloc_size = (__pa_symbol(&__init_end) & PAGE_MASK) - memory;
-        if (mem_alloc_size > 0) {
-                add_memory_region(memory, mem_alloc_size, BOOT_MEM_RAM);
-                total += mem_alloc_size;
-        }
-
         /*
          * The Mips memory init uses the first memory location for
          * some memory vectors. When SPARSEMEM is in use, it doesn't
diff --git a/arch/mips/include/asm/bootinfo.h b/arch/mips/include/asm/bootinfo.h
index 35cd1bab69c3..7a51d879e6ca 100644
--- a/arch/mips/include/asm/bootinfo.h
+++ b/arch/mips/include/asm/bootinfo.h
@@ -86,6 +86,7 @@ extern unsigned long mips_machtype;
 #define BOOT_MEM_RAM            1
 #define BOOT_MEM_ROM_DATA       2
 #define BOOT_MEM_RESERVED       3
+#define BOOT_MEM_INIT_RAM       4
 
 /*
  * A memory map that's built upon what was determined
diff --git a/arch/mips/kernel/setup.c b/arch/mips/kernel/setup.c
index 84af26ab2212..e86c2cf554aa 100644
--- a/arch/mips/kernel/setup.c
+++ b/arch/mips/kernel/setup.c
@@ -121,6 +121,9 @@ static void __init print_memory_map(void)
                 case BOOT_MEM_RAM:
                         printk(KERN_CONT "(usable)\n");
                         break;
+                case BOOT_MEM_INIT_RAM:
+                        printk(KERN_CONT "(usable after init)\n");
+                        break;
                 case BOOT_MEM_ROM_DATA:
                         printk(KERN_CONT "(ROM data)\n");
                         break;
@@ -361,15 +364,24 @@ static void __init bootmem_init(void)
         for (i = 0; i < boot_mem_map.nr_map; i++) {
                 unsigned long start, end, size;
 
+                start = PFN_UP(boot_mem_map.map[i].addr);
+                end   = PFN_DOWN(boot_mem_map.map[i].addr
+                                    + boot_mem_map.map[i].size);
+
                 /*
                  * Reserve usable memory.
                  */
-                if (boot_mem_map.map[i].type != BOOT_MEM_RAM)
+                switch (boot_mem_map.map[i].type) {
+                case BOOT_MEM_RAM:
+                        break;
+                case BOOT_MEM_INIT_RAM:
+                        memory_present(0, start, end);
                         continue;
+                default:
+                        /* Not usable memory */
+                        continue;
+                }
 
-                start = PFN_UP(boot_mem_map.map[i].addr);
-                end   = PFN_DOWN(boot_mem_map.map[i].addr
-                                    + boot_mem_map.map[i].size);
                 /*
                  * We are rounding up the start address of usable memory
                  * and at the end of the usable range downwards.
@@ -455,11 +467,33 @@ early_param("mem", early_parse_mem);
 
 static void __init arch_mem_init(char **cmdline_p)
 {
+        phys_t init_mem, init_end, init_size;
+
         extern void plat_mem_setup(void);
 
         /* call board setup routine */
         plat_mem_setup();
 
+        init_mem = PFN_UP(__pa_symbol(&__init_begin)) << PAGE_SHIFT;
+        init_end = PFN_DOWN(__pa_symbol(&__init_end)) << PAGE_SHIFT;
+        init_size = init_end - init_mem;
+        if (init_size) {
+                /* Make sure it is in the boot_mem_map */
+                int i, found;
+                found = 0;
+                for (i = 0; i < boot_mem_map.nr_map; i++) {
+                        if (init_mem >= boot_mem_map.map[i].addr &&
+                            init_mem < (boot_mem_map.map[i].addr +
+                                        boot_mem_map.map[i].size)) {
+                                found = 1;
+                                break;
+                        }
+                }
+                if (!found)
+                        add_memory_region(init_mem, init_size,
+                                          BOOT_MEM_INIT_RAM);
+        }
+
         pr_info("Determined physical RAM map:\n");
         print_memory_map();
 
@@ -523,6 +557,7 @@ static void __init resource_init(void)
                 res = alloc_bootmem(sizeof(struct resource));
                 switch (boot_mem_map.map[i].type) {
                 case BOOT_MEM_RAM:
+                case BOOT_MEM_INIT_RAM:
                 case BOOT_MEM_ROM_DATA:
                         res->name = "System RAM";
                         break;
diff --git a/arch/mips/mm/init.c b/arch/mips/mm/init.c
index b7ebc4fa89bc..3b3ffd439cd7 100644
--- a/arch/mips/mm/init.c
+++ b/arch/mips/mm/init.c
@@ -304,9 +304,14 @@ int page_is_ram(unsigned long pagenr)
         for (i = 0; i < boot_mem_map.nr_map; i++) {
                 unsigned long addr, end;
 
-                if (boot_mem_map.map[i].type != BOOT_MEM_RAM)
+                switch (boot_mem_map.map[i].type) {
+                case BOOT_MEM_RAM:
+                case BOOT_MEM_INIT_RAM:
+                        break;
+                default:
                         /* not usable memory */
                         continue;
+                }
 
                 addr = PFN_UP(boot_mem_map.map[i].addr);
                 end = PFN_DOWN(boot_mem_map.map[i].addr +
@@ -379,7 +384,7 @@ void __init mem_init(void)
 
         reservedpages = ram = 0;
         for (tmp = 0; tmp < max_low_pfn; tmp++)
-                if (page_is_ram(tmp)) {
+                if (page_is_ram(tmp) && pfn_valid(tmp)) {
                         ram++;
                         if (PageReserved(pfn_to_page(tmp)))
                                 reservedpages++;