2 files changed, 60 insertions, 0 deletions

diff --git a/arch/ia64/kernel/efi.c b/arch/ia64/kernel/efi.c
index 32ce330cbc64..4061593e5b17 100644
--- a/arch/ia64/kernel/efi.c
+++ b/arch/ia64/kernel/efi.c
@@ -1183,3 +1183,33 @@ kdump_find_rsvd_region (unsigned long size,
   return ~0UL;
 }
 #endif
+
+#ifdef CONFIG_PROC_VMCORE
+/* locate the size find a the descriptor at a certain address */
+unsigned long
+vmcore_find_descriptor_size (unsigned long address)
+{
+        void *efi_map_start, *efi_map_end, *p;
+        efi_memory_desc_t *md;
+        u64 efi_desc_size;
+        unsigned long ret = 0;
+
+        efi_map_start = __va(ia64_boot_param->efi_memmap);
+        efi_map_end   = efi_map_start + ia64_boot_param->efi_memmap_size;
+        efi_desc_size = ia64_boot_param->efi_memdesc_size;
+
+        for (p = efi_map_start; p < efi_map_end; p += efi_desc_size) {
+                md = p;
+                if (efi_wb(md) && md->type == EFI_LOADER_DATA
+                    && md->phys_addr == address) {
+                        ret = efi_md_size(md);
+                        break;
+                }
+        }
+
+        if (ret == 0)
+                printk(KERN_WARNING "Cannot locate EFI vmcore descriptor\n");
+
+        return ret;
+}
+#endif
diff --git a/arch/ia64/kernel/setup.c b/arch/ia64/kernel/setup.c
index 5fa09d141ab7..7d6fe65c93f4 100644
--- a/arch/ia64/kernel/setup.c
+++ b/arch/ia64/kernel/setup.c
@@ -251,6 +251,12 @@ reserve_memory (void)
         }
 #endif
 
+#ifdef CONFIG_PROC_VMCORE
+        if (reserve_elfcorehdr(&rsvd_region[n].start,
+                               &rsvd_region[n].end) == 0)
+                n++;
+#endif
+
         efi_memmap_init(&rsvd_region[n].start, &rsvd_region[n].end);
         n++;
 
@@ -453,6 +459,30 @@ static int __init parse_elfcorehdr(char *arg)
         return 0;
 }
 early_param("elfcorehdr", parse_elfcorehdr);
+
+int __init reserve_elfcorehdr(unsigned long *start, unsigned long *end)
+{
+        unsigned long length;
+
+        /* We get the address using the kernel command line,
+         * but the size is extracted from the EFI tables.
+         * Both address and size are required for reservation
+         * to work properly.
+         */
+
+        if (elfcorehdr_addr >= ELFCORE_ADDR_MAX)
+                return -EINVAL;
+
+        if ((length = vmcore_find_descriptor_size(elfcorehdr_addr)) == 0) {
+                elfcorehdr_addr = ELFCORE_ADDR_MAX;
+                return -EINVAL;
+        }
+
+        *start = (unsigned long)__va(elfcorehdr_addr);
+        *end = *start + length;
+        return 0;
+}
+
 #endif /* CONFIG_PROC_VMCORE */
 
 void __init