ARM: 8167/1: extend the reserved memory for initrd to be page aligned

This patch extends the start and end address of initrd to be page aligned,
so that we can free all memory including the un-page aligned head or tail
page of initrd, if the start or end address of initrd are not page
aligned, the page can't be freed by free_initrd_mem() function.

Signed-off-by: Yalin Wang <yalin.wang@sonymobile.com>
Acked-by: Catalin Marinas <catalin.marinas@arm.com>
Signed-off-by: Russell King <rmk+kernel@arm.linux.org.uk>

2 files changed, 12 insertions, 1 deletions

diff --git a/arch/arm/mm/init.c b/arch/arm/mm/init.c
index 659c75d808dc..9221645dd192 100644
--- a/arch/arm/mm/init.c
+++ b/arch/arm/mm/init.c
@@ -636,6 +636,11 @@ static int keep_initrd;
 void free_initrd_mem(unsigned long start, unsigned long end)
 {
         if (!keep_initrd) {
+                if (start == initrd_start)
+                        start = round_down(start, PAGE_SIZE);
+                if (end == initrd_end)
+                        end = round_up(end, PAGE_SIZE);
+
                 poison_init_mem((void *)start, PAGE_ALIGN(end) - start);
                 free_reserved_area((void *)start, (void *)end, -1, "initrd");
         }
diff --git a/arch/arm64/mm/init.c b/arch/arm64/mm/init.c
index 5472c2401876..c5512f694814 100644
--- a/arch/arm64/mm/init.c
+++ b/arch/arm64/mm/init.c
@@ -334,8 +334,14 @@ static int keep_initrd;
 
 void free_initrd_mem(unsigned long start, unsigned long end)
 {
-        if (!keep_initrd)
+        if (!keep_initrd) {
+                if (start == initrd_start)
+                        start = round_down(start, PAGE_SIZE);
+                if (end == initrd_end)
+                        end = round_up(end, PAGE_SIZE);
+
                 free_reserved_area((void *)start, (void *)end, 0, "initrd");
+        }
 }
 
 static int __init keepinitrd_setup(char *__unused)