5c83f50
From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
5c83f50
From: Zhang Boyang <zhangboyang.id@gmail.com>
5c83f50
Date: Sun, 29 Jan 2023 19:49:32 +0800
5c83f50
Subject: [PATCH] mm: Preallocate some space when adding new regions
5c83f50
5c83f50
When grub_memalign() encounters out-of-memory, it will try
5c83f50
grub_mm_add_region_fn() to request more memory from system firmware.
5c83f50
However, it doesn't preallocate memory space for future allocation
5c83f50
requests. In extreme cases, it requires one call to
5c83f50
grub_mm_add_region_fn() for each memory allocation request. This can
5c83f50
be very slow.
5c83f50
5c83f50
This patch introduces GRUB_MM_HEAP_GROW_EXTRA, the minimal heap growth
5c83f50
granularity. The new region size is now set to the bigger one of its
5c83f50
original value and GRUB_MM_HEAP_GROW_EXTRA. Thus, it will result in some
5c83f50
memory space preallocated if current allocations request is small.
5c83f50
5c83f50
The value of GRUB_MM_HEAP_GROW_EXTRA is set to 1MB. If this value is
5c83f50
smaller, the cost of small memory allocations will be higher. If this
5c83f50
value is larger, more memory will be wasted and it might cause
5c83f50
out-of-memory on machines with small amount of RAM.
5c83f50
5c83f50
Signed-off-by: Zhang Boyang <zhangboyang.id@gmail.com>
5c83f50
Reviewed-by: Daniel Kiper <daniel.kiper@oracle.com>
5c83f50
(cherry picked from commit 21869baec15239b6d99122b32b14a778af4c754f)
5c83f50
---
5c83f50
 grub-core/kern/mm.c | 6 ++++++
5c83f50
 1 file changed, 6 insertions(+)
5c83f50
5c83f50
diff --git a/grub-core/kern/mm.c b/grub-core/kern/mm.c
5c83f50
index f29a3e5cbd..cc8a4703bc 100644
5c83f50
--- a/grub-core/kern/mm.c
5c83f50
+++ b/grub-core/kern/mm.c
5c83f50
@@ -123,6 +123,9 @@
5c83f50
 /* The size passed to grub_mm_add_region_fn() is aligned up by this value. */
5c83f50
 #define GRUB_MM_HEAP_GROW_ALIGN	4096
5c83f50
 
5c83f50
+/* Minimal heap growth granularity when existing heap space is exhausted. */
5c83f50
+#define GRUB_MM_HEAP_GROW_EXTRA	0x100000
5c83f50
+
5c83f50
 grub_mm_region_t grub_mm_base;
5c83f50
 grub_mm_add_region_func_t grub_mm_add_region_fn;
5c83f50
 
5c83f50
@@ -471,6 +474,9 @@ grub_memalign (grub_size_t align, grub_size_t size)
5c83f50
   if (grub_add (size + align, GRUB_MM_MGMT_OVERHEAD, &grow))
5c83f50
     goto fail;
5c83f50
 
5c83f50
+  /* Preallocate some extra space if heap growth is small. */
5c83f50
+  grow = grub_max (grow, GRUB_MM_HEAP_GROW_EXTRA);
5c83f50
+
5c83f50
   /* Align up heap growth to make it friendly to CPU/MMU. */
5c83f50
   if (grow > ~(grub_size_t) (GRUB_MM_HEAP_GROW_ALIGN - 1))
5c83f50
     goto fail;