1 files changed, 20 insertions, 5 deletions

diff --git a/kernel/kexec_file.c b/kernel/kexec_file.c
index efd2c094af7e..0c2df7f73792 100644
--- a/kernel/kexec_file.c
+++ b/kernel/kexec_file.c
@@ -450,6 +450,23 @@ int __weak arch_kexec_walk_mem(struct kexec_buf *kbuf,
 }
 
 /**
+ * kexec_locate_mem_hole - find free memory for the purgatory or the next kernel
+ * @kbuf:       Parameters for the memory search.
+ *
+ * On success, kbuf->mem will have the start address of the memory region found.
+ *
+ * Return: 0 on success, negative errno on error.
+ */
+int kexec_locate_mem_hole(struct kexec_buf *kbuf)
+{
+        int ret;
+
+        ret = arch_kexec_walk_mem(kbuf, locate_mem_hole_callback);
+
+        return ret == 1 ? 0 : -EADDRNOTAVAIL;
+}
+
+/**
  * kexec_add_buffer - place a buffer in a kexec segment
  * @kbuf:       Buffer contents and memory parameters.
  *
@@ -489,11 +506,9 @@ int kexec_add_buffer(struct kexec_buf *kbuf)
         kbuf->buf_align = max(kbuf->buf_align, PAGE_SIZE);
 
         /* Walk the RAM ranges and allocate a suitable range for the buffer */
-        ret = arch_kexec_walk_mem(kbuf, locate_mem_hole_callback);
-        if (ret != 1) {
-                /* A suitable memory range could not be found for buffer */
-                return -EADDRNOTAVAIL;
-        }
+        ret = kexec_locate_mem_hole(kbuf);
+        if (ret)
+                return ret;
 
         /* Found a suitable memory range */
         ksegment = &kbuf->image->segment[kbuf->image->nr_segments];