--- v10
+++ v15
@@ -1,136 +1,96 @@
-delete_fdt_mem_rsv() retrieves the memory reserve map entry, for
-the given starting address and size, from the device tree blob, and
-removes the entry from the device tree blob. This function is called
-to free the resources reserved for the buffer used for carrying forward
-the IMA measurement logs on kexec. This function does not have
-architecture specific code, but is currently limited to powerpc.
-
-Move delete_fdt_mem_rsv() to "drivers/of/kexec_fdt.c" so that it is
-accessible for other architectures as well.
-
-Co-developed-by: Prakhar Srivastava <prsriva@linux.microsoft.com>
-Signed-off-by: Prakhar Srivastava <prsriva@linux.microsoft.com>
-Signed-off-by: Lakshmi Ramasubramanian <nramas@linux.microsoft.com>
+From: Rob Herring <robh@kernel.org>
+
+Both arm64 and powerpc do essentially the same FDT /chosen setup for
+kexec. The differences are either omissions that arm64 should have
+or additional properties that will be ignored. The setup code can be
+combined and shared by both powerpc and arm64.
+
+The differences relative to the arm64 version:
+ - If /chosen doesn't exist, it will be created (should never happen).
+ - Any old dtb and initrd reserved memory will be released.
+ - The new initrd and elfcorehdr are marked reserved.
+ - "linux,booted-from-kexec" is set.
+
+The differences relative to the powerpc version:
+ - "kaslr-seed" and "rng-seed" may be set.
+ - "linux,elfcorehdr" is set.
+ - Any existing "linux,usable-memory-range" is removed.
+
+Combine the code for setting up the /chosen node in the FDT and updating
+the memory reservation for kexec, for powerpc and arm64, in
+of_kexec_setup_new_fdt() and move it to "drivers/of/kexec.c".
+
+Signed-off-by: Rob Herring <robh@kernel.org>
+Reviewed-by: Thiago Jung Bauermann <bauerman@linux.ibm.com>
+Reviewed-by: Lakshmi Ramasubramanian <nramas@linux.microsoft.com>
---
- arch/powerpc/include/asm/kexec.h | 1 -
- arch/powerpc/kexec/file_load.c | 32 -----------------
- drivers/of/Makefile | 1 +
- drivers/of/kexec.c | 61 ++++++++++++++++++++++++++++++++
- include/linux/kexec.h | 5 +++
- 5 files changed, 67 insertions(+), 33 deletions(-)
+ drivers/of/Makefile | 1 +
+ drivers/of/kexec.c | 236 ++++++++++++++++++++++++++++++++++++++++++++
+ include/linux/of.h | 5 +
+ 3 files changed, 242 insertions(+)
create mode 100644 drivers/of/kexec.c
-diff --git a/arch/powerpc/include/asm/kexec.h b/arch/powerpc/include/asm/kexec.h
-index 55d6ede30c19..7c223031ecdd 100644
---- a/arch/powerpc/include/asm/kexec.h
-+++ b/arch/powerpc/include/asm/kexec.h
-@@ -126,7 +126,6 @@ int setup_purgatory(struct kimage *image, const void *slave_code,
- int setup_new_fdt(const struct kimage *image, void *fdt,
- unsigned long initrd_load_addr, unsigned long initrd_len,
- const char *cmdline);
--int delete_fdt_mem_rsv(void *fdt, unsigned long start, unsigned long size);
-
- #ifdef CONFIG_PPC64
- struct kexec_buf;
-diff --git a/arch/powerpc/kexec/file_load.c b/arch/powerpc/kexec/file_load.c
-index 9a232bc36c8f..9efc98b1e2ae 100644
---- a/arch/powerpc/kexec/file_load.c
-+++ b/arch/powerpc/kexec/file_load.c
-@@ -109,38 +109,6 @@ int setup_purgatory(struct kimage *image, const void *slave_code,
- return 0;
- }
-
--/**
-- * delete_fdt_mem_rsv - delete memory reservation with given address and size
-- *
-- * Return: 0 on success, or negative errno on error.
-- */
--int delete_fdt_mem_rsv(void *fdt, unsigned long start, unsigned long size)
--{
-- int i, ret, num_rsvs = fdt_num_mem_rsv(fdt);
--
-- for (i = 0; i < num_rsvs; i++) {
-- uint64_t rsv_start, rsv_size;
--
-- ret = fdt_get_mem_rsv(fdt, i, &rsv_start, &rsv_size);
-- if (ret) {
-- pr_err("Malformed device tree.\n");
-- return -EINVAL;
-- }
--
-- if (rsv_start == start && rsv_size == size) {
-- ret = fdt_del_mem_rsv(fdt, i);
-- if (ret) {
-- pr_err("Error deleting device tree reservation.\n");
-- return -EINVAL;
-- }
--
-- return 0;
-- }
-- }
--
-- return -ENOENT;
--}
--
- /*
- * setup_new_fdt - modify /chosen and memory reservation for the next kernel
- * @image: kexec image being loaded.
diff --git a/drivers/of/Makefile b/drivers/of/Makefile
-index 6e1e5212f058..77d24712c0c8 100644
+index 6e1e5212f058..8ce11955afde 100644
--- a/drivers/of/Makefile
+++ b/drivers/of/Makefile
@@ -13,5 +13,6 @@ obj-$(CONFIG_OF_RESERVED_MEM) += of_reserved_mem.o
obj-$(CONFIG_OF_RESOLVE) += resolver.o
obj-$(CONFIG_OF_OVERLAY) += overlay.o
obj-$(CONFIG_OF_NUMA) += of_numa.o
-+obj-$(CONFIG_OF_FLATTREE) += kexec.o
++obj-$(CONFIG_KEXEC_FILE) += kexec.o
obj-$(CONFIG_OF_UNITTEST) += unittest-data/
diff --git a/drivers/of/kexec.c b/drivers/of/kexec.c
new file mode 100644
-index 000000000000..b7d59105fcb8
+index 000000000000..4afd3cc1c04a
--- /dev/null
+++ b/drivers/of/kexec.c
-@@ -0,0 +1,61 @@
-+// SPDX-License-Identifier: GPL-2.0+
+@@ -0,0 +1,236 @@
++// SPDX-License-Identifier: GPL-2.0-only
+/*
-+ * Copyright (C) 2020 Microsoft Corporation
-+ *
-+ * Author: Lakshmi Ramasubramanian (nramas@linux.microsoft.com)
-+ *
-+ * File: kexec.c
-+ * Defines kexec related functions.
++ * Copyright (C) 2020 Arm Limited
++ *
++ * Based on arch/arm64/kernel/machine_kexec_file.c:
++ * Copyright (C) 2018 Linaro Limited
++ *
++ * And arch/powerpc/kexec/file_load.c:
++ * Copyright (C) 2016 IBM Corporation
+ */
+
-+#define pr_fmt(fmt) "OF: kexec: " fmt
-+
+#include <linux/kernel.h>
-+#include <linux/slab.h>
-+#include <linux/memblock.h>
+#include <linux/kexec.h>
++#include <linux/libfdt.h>
+#include <linux/of.h>
+#include <linux/of_fdt.h>
-+#include <linux/libfdt.h>
++#include <linux/random.h>
++#include <linux/types.h>
++
++/* relevant device tree properties */
++#define FDT_PROP_KEXEC_ELFHDR "linux,elfcorehdr"
++#define FDT_PROP_MEM_RANGE "linux,usable-memory-range"
++#define FDT_PROP_INITRD_START "linux,initrd-start"
++#define FDT_PROP_INITRD_END "linux,initrd-end"
++#define FDT_PROP_BOOTARGS "bootargs"
++#define FDT_PROP_KASLR_SEED "kaslr-seed"
++#define FDT_PROP_RNG_SEED "rng-seed"
++#define RNG_SEED_SIZE 128
+
+/**
-+ * delete_fdt_mem_rsv - delete memory reservation with given address and size
-+ *
-+ * @fdt: Flattened Device Tree to update
-+ * @start: Starting address of the reservation to delete
-+ * @size: Size of the reservation to delete
++ * fdt_find_and_del_mem_rsv - delete memory reservation with given address and size
++ *
++ * @fdt: Flattened device tree for the current kernel.
++ * @start: Starting address of the reserved memory.
++ * @size: Size of the reserved memory.
+ *
+ * Return: 0 on success, or negative errno on error.
+ */
-+int delete_fdt_mem_rsv(void *fdt, unsigned long start, unsigned long size)
++static int fdt_find_and_del_mem_rsv(void *fdt, unsigned long start, unsigned long size)
+{
-+ int i, ret, num_rsvs;
-+
-+ if (!IS_ENABLED(CONFIG_KEXEC_FILE))
-+ return 0;
-+
-+ num_rsvs = fdt_num_mem_rsv(fdt);
++ int i, ret, num_rsvs = fdt_num_mem_rsv(fdt);
++
+ for (i = 0; i < num_rsvs; i++) {
-+ uint64_t rsv_start, rsv_size;
++ u64 rsv_start, rsv_size;
+
+ ret = fdt_get_mem_rsv(fdt, i, &rsv_start, &rsv_size);
+ if (ret) {
@@ -145,30 +105,200 @@
+ return -EINVAL;
+ }
+
-+ pr_debug("Freed reserved memory at %lu of size %lu\n",
-+ start, size);
+ return 0;
+ }
+ }
+
+ return -ENOENT;
+}
-diff --git a/include/linux/kexec.h b/include/linux/kexec.h
-index 9e93bef52968..d0234c4815da 100644
---- a/include/linux/kexec.h
-+++ b/include/linux/kexec.h
-@@ -407,6 +407,11 @@ static inline int kexec_crash_loaded(void) { return 0; }
- #define kexec_in_progress false
- #endif /* CONFIG_KEXEC_CORE */
++
++/*
++ * of_kexec_setup_new_fdt - modify /chosen and memory reservation for the next kernel
++ *
++ * @image: kexec image being loaded.
++ * @fdt: Flattened device tree for the next kernel.
++ * @initrd_load_addr: Address where the next initrd will be loaded.
++ * @initrd_len: Size of the next initrd, or 0 if there will be none.
++ * @cmdline: Command line for the next kernel, or NULL if there will
++ * be none.
++ *
++ * Return: 0 on success, or negative errno on error.
++ */
++int of_kexec_setup_new_fdt(const struct kimage *image, void *fdt,
++ unsigned long initrd_load_addr, unsigned long initrd_len,
++ const char *cmdline)
++{
++ int ret, chosen_node;
++ const void *prop;
++
++ /* Remove memory reservation for the current device tree. */
++ ret = fdt_find_and_del_mem_rsv(fdt, __pa(initial_boot_params),
++ fdt_totalsize(initial_boot_params));
++ if (ret == -EINVAL)
++ return ret;
++
++ chosen_node = fdt_path_offset(fdt, "/chosen");
++ if (chosen_node == -FDT_ERR_NOTFOUND)
++ chosen_node = fdt_add_subnode(fdt, fdt_path_offset(fdt, "/"),
++ "chosen");
++ if (chosen_node < 0) {
++ ret = chosen_node;
++ goto out;
++ }
++
++ ret = fdt_delprop(fdt, chosen_node, FDT_PROP_KEXEC_ELFHDR);
++ if (ret && ret != -FDT_ERR_NOTFOUND)
++ goto out;
++ ret = fdt_delprop(fdt, chosen_node, FDT_PROP_MEM_RANGE);
++ if (ret && ret != -FDT_ERR_NOTFOUND)
++ goto out;
++
++ /* Did we boot using an initrd? */
++ prop = fdt_getprop(fdt, chosen_node, "linux,initrd-start", NULL);
++ if (prop) {
++ u64 tmp_start, tmp_end, tmp_size;
++
++ tmp_start = fdt64_to_cpu(*((const fdt64_t *) prop));
++
++ prop = fdt_getprop(fdt, chosen_node, "linux,initrd-end", NULL);
++ if (!prop)
++ return -EINVAL;
++
++ tmp_end = fdt64_to_cpu(*((const fdt64_t *) prop));
++
++ /*
++ * kexec reserves exact initrd size, while firmware may
++ * reserve a multiple of PAGE_SIZE, so check for both.
++ */
++ tmp_size = tmp_end - tmp_start;
++ ret = fdt_find_and_del_mem_rsv(fdt, tmp_start, tmp_size);
++ if (ret == -ENOENT)
++ ret = fdt_find_and_del_mem_rsv(fdt, tmp_start,
++ round_up(tmp_size, PAGE_SIZE));
++ if (ret == -EINVAL)
++ return ret;
++ }
++
++ /* add initrd-* */
++ if (initrd_load_addr) {
++ ret = fdt_setprop_u64(fdt, chosen_node, FDT_PROP_INITRD_START,
++ initrd_load_addr);
++ if (ret)
++ goto out;
++
++ ret = fdt_setprop_u64(fdt, chosen_node, FDT_PROP_INITRD_END,
++ initrd_load_addr + initrd_len);
++ if (ret)
++ goto out;
++
++ ret = fdt_add_mem_rsv(fdt, initrd_load_addr, initrd_len);
++ if (ret)
++ goto out;
++
++ } else {
++ ret = fdt_delprop(fdt, chosen_node, FDT_PROP_INITRD_START);
++ if (ret && (ret != -FDT_ERR_NOTFOUND))
++ goto out;
++
++ ret = fdt_delprop(fdt, chosen_node, FDT_PROP_INITRD_END);
++ if (ret && (ret != -FDT_ERR_NOTFOUND))
++ goto out;
++ }
++
++ if (image->type == KEXEC_TYPE_CRASH) {
++ /* add linux,elfcorehdr */
++ ret = fdt_appendprop_addrrange(fdt, 0, chosen_node,
++ FDT_PROP_KEXEC_ELFHDR,
++ image->arch.elf_headers_mem,
++ image->arch.elf_headers_sz);
++ if (ret)
++ goto out;
++
++ /*
++ * Avoid elfcorehdr from being stomped on in kdump kernel by
++ * setting up memory reserve map.
++ */
++ ret = fdt_add_mem_rsv(fdt, image->arch.elf_headers_mem,
++ image->arch.elf_headers_sz);
++ if (ret)
++ goto out;
++
++ /* add linux,usable-memory-range */
++ ret = fdt_appendprop_addrrange(fdt, 0, chosen_node,
++ FDT_PROP_MEM_RANGE,
++ crashk_res.start,
++ crashk_res.end - crashk_res.start + 1);
++ if (ret)
++ goto out;
++ }
++
++ /* add bootargs */
++ if (cmdline) {
++ ret = fdt_setprop_string(fdt, chosen_node, FDT_PROP_BOOTARGS, cmdline);
++ if (ret)
++ goto out;
++ } else {
++ ret = fdt_delprop(fdt, chosen_node, FDT_PROP_BOOTARGS);
++ if (ret && (ret != -FDT_ERR_NOTFOUND))
++ goto out;
++ }
++
++ /* add kaslr-seed */
++ ret = fdt_delprop(fdt, chosen_node, FDT_PROP_KASLR_SEED);
++ if (ret == -FDT_ERR_NOTFOUND)
++ ret = 0;
++ else if (ret)
++ goto out;
++
++ if (rng_is_initialized()) {
++ u64 seed = get_random_u64();
++
++ ret = fdt_setprop_u64(fdt, chosen_node, FDT_PROP_KASLR_SEED, seed);
++ if (ret)
++ goto out;
++ } else {
++ pr_notice("RNG is not initialised: omitting \"%s\" property\n",
++ FDT_PROP_KASLR_SEED);
++ }
++
++ /* add rng-seed */
++ if (rng_is_initialized()) {
++ void *rng_seed;
++
++ ret = fdt_setprop_placeholder(fdt, chosen_node, FDT_PROP_RNG_SEED,
++ RNG_SEED_SIZE, &rng_seed);
++ if (ret)
++ goto out;
++ get_random_bytes(rng_seed, RNG_SEED_SIZE);
++ } else {
++ pr_notice("RNG is not initialised: omitting \"%s\" property\n",
++ FDT_PROP_RNG_SEED);
++ }
++
++ ret = fdt_setprop(fdt, chosen_node, "linux,booted-from-kexec", NULL, 0);
++
++out:
++ if (ret)
++ return (ret == -FDT_ERR_NOSPACE) ? -ENOMEM : -EINVAL;
++
++ return 0;
++}
+diff --git a/include/linux/of.h b/include/linux/of.h
+index 5d51891cbf1a..3375f5295875 100644
+--- a/include/linux/of.h
++++ b/include/linux/of.h
+@@ -558,6 +558,11 @@ int of_map_id(struct device_node *np, u32 id,
+ const char *map_name, const char *map_mask_name,
+ struct device_node **target, u32 *id_out);
-+#if defined(CONFIG_OF_FLATTREE)
-+extern int delete_fdt_mem_rsv(void *fdt, unsigned long start,
-+ unsigned long size);
-+#endif /* CONFIG_OF_FLATTREE */
-+
- #endif /* !defined(__ASSEBMLY__) */
++struct kimage;
++int of_kexec_setup_new_fdt(const struct kimage *image, void *fdt,
++ unsigned long initrd_load_addr, unsigned long initrd_len,
++ const char *cmdline);
++
+ #else /* CONFIG_OF */
- #endif /* LINUX_KEXEC_H */
+ static inline void of_core_init(void)
--
-2.29.2
-
+2.30.0
+