efi_loader: Extract adding a conventional memory in separate routine

[oweals/u-boot.git] / lib / efi_loader / efi_memory.c

diff --git a/lib/efi_loader/efi_memory.c b/lib/efi_loader/efi_memory.c
index b5775e0399ea517a7653f7f4ffcf088a37aa2487..83cbc9154f3f47fc4f9cd478ad7630f551021a61 100644 (file)
--- a/lib/efi_loader/efi_memory.c
+++ b/lib/efi_loader/efi_memory.c
@@ -655,6 +655,54 @@ efi_status_t efi_get_memory_map(efi_uintn_t *memory_map_size,
        return EFI_SUCCESS;
 }
 
+/**
+ * efi_add_conventional_memory_map() - add a RAM memory area to the map
+ *
+ * @ram_start:         start address of a RAM memory area
+ * @ram_end:           end address of a RAM memory area
+ * @ram_top:           max address to be used as conventional memory
+ * Return:             status code
+ */
+efi_status_t efi_add_conventional_memory_map(u64 ram_start, u64 ram_end,
+                                            u64 ram_top)
+{
+       u64 pages;
+
+       /* Remove partial pages */
+       ram_end &= ~EFI_PAGE_MASK;
+       ram_start = (ram_start + EFI_PAGE_MASK) & ~EFI_PAGE_MASK;
+
+       if (ram_end <= ram_start) {
+               /* Invalid mapping */
+               return EFI_INVALID_PARAMETER;
+       }
+
+       pages = (ram_end - ram_start) >> EFI_PAGE_SHIFT;
+
+       efi_add_memory_map(ram_start, pages,
+                          EFI_CONVENTIONAL_MEMORY, false);
+
+       /*
+        * Boards may indicate to the U-Boot memory core that they
+        * can not support memory above ram_top. Let's honor this
+        * in the efi_loader subsystem too by declaring any memory
+        * above ram_top as "already occupied by firmware".
+        */
+       if (ram_top < ram_start) {
+               /* ram_top is before this region, reserve all */
+               efi_add_memory_map(ram_start, pages,
+                                  EFI_BOOT_SERVICES_DATA, true);
+       } else if ((ram_top >= ram_start) && (ram_top < ram_end)) {
+               /* ram_top is inside this region, reserve parts */
+               pages = (ram_end - ram_top) >> EFI_PAGE_SHIFT;
+
+               efi_add_memory_map(ram_top, pages,
+                                  EFI_BOOT_SERVICES_DATA, true);
+       }
+
+       return EFI_SUCCESS;
+}
+
 __weak void efi_add_known_memory(void)
 {
        u64 ram_top = board_get_usable_ram_top(0) & ~EFI_PAGE_MASK;
@@ -672,42 +720,12 @@ __weak void efi_add_known_memory(void)
 
        /* Add RAM */
        for (i = 0; i < CONFIG_NR_DRAM_BANKS; i++) {
-               u64 ram_end, ram_start, pages;
+               u64 ram_end, ram_start;
 
                ram_start = (uintptr_t)map_sysmem(gd->bd->bi_dram[i].start, 0);
                ram_end = ram_start + gd->bd->bi_dram[i].size;
 
-               /* Remove partial pages */
-               ram_end &= ~EFI_PAGE_MASK;
-               ram_start = (ram_start + EFI_PAGE_MASK) & ~EFI_PAGE_MASK;
-
-               if (ram_end <= ram_start) {
-                       /* Invalid mapping, keep going. */
-                       continue;
-               }
-
-               pages = (ram_end - ram_start) >> EFI_PAGE_SHIFT;
-
-               efi_add_memory_map(ram_start, pages,
-                                  EFI_CONVENTIONAL_MEMORY, false);
-
-               /*
-                * Boards may indicate to the U-Boot memory core that they
-                * can not support memory above ram_top. Let's honor this
-                * in the efi_loader subsystem too by declaring any memory
-                * above ram_top as "already occupied by firmware".
-                */
-               if (ram_top < ram_start) {
-                       /* ram_top is before this region, reserve all */
-                       efi_add_memory_map(ram_start, pages,
-                                          EFI_BOOT_SERVICES_DATA, true);
-               } else if ((ram_top >= ram_start) && (ram_top < ram_end)) {
-                       /* ram_top is inside this region, reserve parts */
-                       pages = (ram_end - ram_top) >> EFI_PAGE_SHIFT;
-
-                       efi_add_memory_map(ram_top, pages,
-                                          EFI_BOOT_SERVICES_DATA, true);
-               }
+               efi_add_conventional_memory_map(ram_start, ram_end, ram_top);
        }
 }