kern/efi/mm: Extract function to add memory regions

In preparation of support for runtime-allocating additional memory
region, this patch extracts the function to retrieve the EFI memory
map and add a subset of it to GRUB's own memory regions.

Signed-off-by: Patrick Steinhardt <ps@pks.im>
Signed-off-by: Daniel Axtens <dja@axtens.net>
Reviewed-by: Daniel Kiper <daniel.kiper@oracle.com>
Tested-by: Patrick Steinhardt <ps@pks.im>
This commit is contained in:
Patrick Steinhardt authored and Daniel Kiper committed 2022-07-04 14:36:15 +02:00
1 parent 938c3760b8
commit 96a7ea29e3
1 file changed
+15 -6
+15 -6
View File
@@ -556,8 +556,8 @@ print_memory_map (grub_efi_memory_descriptor_t *memory_map,
} }
#endif #endif
void static grub_err_t
grub_efi_mm_init (void) grub_efi_mm_add_regions (grub_size_t required_bytes)
{ {
grub_efi_memory_descriptor_t *memory_map; grub_efi_memory_descriptor_t *memory_map;
grub_efi_memory_descriptor_t *memory_map_end; grub_efi_memory_descriptor_t *memory_map_end;
@@ -570,7 +570,7 @@ grub_efi_mm_init (void)
/* Prepare a memory region to store two memory maps. */ /* Prepare a memory region to store two memory maps. */
memory_map = grub_efi_allocate_any_pages (2 * BYTES_TO_PAGES (MEMORY_MAP_SIZE)); memory_map = grub_efi_allocate_any_pages (2 * BYTES_TO_PAGES (MEMORY_MAP_SIZE));
if (! memory_map) if (! memory_map)
grub_fatal ("cannot allocate memory"); return grub_error (GRUB_ERR_OUT_OF_MEMORY, "cannot allocate memory for memory map");
/* Obtain descriptors for available memory. */ /* Obtain descriptors for available memory. */
map_size = MEMORY_MAP_SIZE; map_size = MEMORY_MAP_SIZE;
@@ -588,14 +588,14 @@ grub_efi_mm_init (void)
memory_map = grub_efi_allocate_any_pages (2 * BYTES_TO_PAGES (map_size)); memory_map = grub_efi_allocate_any_pages (2 * BYTES_TO_PAGES (map_size));
if (! memory_map) if (! memory_map)
grub_fatal ("cannot allocate memory"); return grub_error (GRUB_ERR_OUT_OF_MEMORY, "cannot allocate memory for new memory map");
mm_status = grub_efi_get_memory_map (&map_size, memory_map, 0, mm_status = grub_efi_get_memory_map (&map_size, memory_map, 0,
&desc_size, 0); &desc_size, 0);
} }
if (mm_status < 0) if (mm_status < 0)
grub_fatal ("cannot get memory map"); return grub_error (GRUB_ERR_OUT_OF_MEMORY, "error fetching memory map from EFI");
memory_map_end = NEXT_MEMORY_DESCRIPTOR (memory_map, map_size); memory_map_end = NEXT_MEMORY_DESCRIPTOR (memory_map, map_size);
@@ -610,7 +610,7 @@ grub_efi_mm_init (void)
/* Allocate memory regions for GRUB's memory management. */ /* Allocate memory regions for GRUB's memory management. */
add_memory_regions (filtered_memory_map, desc_size, add_memory_regions (filtered_memory_map, desc_size,
filtered_memory_map_end, BYTES_TO_PAGES (DEFAULT_HEAP_SIZE)); filtered_memory_map_end, BYTES_TO_PAGES (required_bytes));
#if 0 #if 0
/* For debug. */ /* For debug. */
@@ -628,6 +628,15 @@ grub_efi_mm_init (void)
/* Release the memory maps. */ /* Release the memory maps. */
grub_efi_free_pages ((grub_addr_t) memory_map, grub_efi_free_pages ((grub_addr_t) memory_map,
2 * BYTES_TO_PAGES (MEMORY_MAP_SIZE)); 2 * BYTES_TO_PAGES (MEMORY_MAP_SIZE));
return GRUB_ERR_NONE;
}
void
grub_efi_mm_init (void)
{
if (grub_efi_mm_add_regions (DEFAULT_HEAP_SIZE) != GRUB_ERR_NONE)
grub_fatal ("%s", grub_errmsg);
} }
#if defined (__aarch64__) || defined (__arm__) || defined (__riscv) #if defined (__aarch64__) || defined (__arm__) || defined (__riscv)