Re: [PATCH v19 05/13] of: Add a common kexec FDT setup function

2021-06-16 Thread Rob Herring
On Tue, Jun 15, 2021 at 8:23 PM Michael Ellerman  wrote:
>
> Rob Herring  writes:
> > On Tue, Jun 15, 2021 at 10:13 AM nramas  wrote:
> >>
> >> On Tue, 2021-06-15 at 08:01 -0600, Rob Herring wrote:
> >> > On Tue, Jun 15, 2021 at 6:18 AM Geert Uytterhoeven <
> >> > ge...@linux-m68k.org> wrote:
> >> > >
> >> > > > +void *of_kexec_alloc_and_setup_fdt(const struct kimage *image,
> >> > > > +  unsigned long
> >> > > > initrd_load_addr,
> >> > > > +  unsigned long initrd_len,
> >> > > > +  const char *cmdline, size_t
> >> > > > extra_fdt_size)
> >> > > > +{
> >> > > > +   /* 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) {
> >> > > > +   ret = -EINVAL;
> >> > > > +   goto out;
> >> > > > +   }
> >> > > > +
> >> > > > +   tmp_end = fdt64_to_cpu(*((const fdt64_t *)
> >> > > > prop));
> >> > >
> >> > > Some kernel code assumes "linux,initrd-{start,end}" are 64-bit,
> >> > > other code assumes 32-bit.
> >> >
> >> > It can be either. The above code was a merge of arm64 and powerpc >> > 
> >> > both
> >> > of which use 64-bit and still only runs on those arches. It looks >> > 
> >> > like
> >> > some powerpc platforms may use 32-bit, but this would have been >> > 
> >> > broken
> >> > before.
>
> >> of_kexec_alloc_and_setup_fdt() is called from elf_64.c (in
> >> arch/powerpc/kexec) which is for 64-bit powerpc platform only.
> >
> > 64-bit PPC could be writing 32-bit property values. The architecture
> > size doesn't necessarily matter. And if the values came from the
> > bootloader, who knows what size it used.
> >
> > This code is 32-bit powerpc only?:
> >
> > arch/powerpc/boot/main.c-   /* Tell the kernel initrd address via 
> > device tree */
> > arch/powerpc/boot/main.c:   setprop_val(chosen, "linux,initrd-start", 
> > (u32)(initrd_addr));
> > arch/powerpc/boot/main.c-   setprop_val(chosen, "linux,initrd-end", 
> > (u32)(initrd_addr+initrd_size));
>
> Historically that code was always built 32-bit, even when used with a
> 64-bit kernel.
>
> These days it is also built 64-bit (for ppc64le).

How it is built is immaterial. It's always writing a 32-bit value due
to the u32 cast.

> It looks like the drivers/of/fdt.c code can handle either 64 or 32-bit,
> so I guess that's why it seems to be working.

Yes, that works, but that's not the issue. The question is does the
main.c code run in combination with kexec. The kexec code above
(copied straight from PPC code) would not work if linux,initrd-* are
written by the main.c code.

Rob


Re: [PATCH v19 05/13] of: Add a common kexec FDT setup function

2021-06-15 Thread Michael Ellerman
Rob Herring  writes:
> On Tue, Jun 15, 2021 at 10:13 AM nramas  wrote:
>>
>> On Tue, 2021-06-15 at 08:01 -0600, Rob Herring wrote:
>> > On Tue, Jun 15, 2021 at 6:18 AM Geert Uytterhoeven <
>> > ge...@linux-m68k.org> wrote:
>> > >
>> > > > +void *of_kexec_alloc_and_setup_fdt(const struct kimage *image,
>> > > > +  unsigned long
>> > > > initrd_load_addr,
>> > > > +  unsigned long initrd_len,
>> > > > +  const char *cmdline, size_t
>> > > > extra_fdt_size)
>> > > > +{
>> > > > +   /* 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) {
>> > > > +   ret = -EINVAL;
>> > > > +   goto out;
>> > > > +   }
>> > > > +
>> > > > +   tmp_end = fdt64_to_cpu(*((const fdt64_t *)
>> > > > prop));
>> > >
>> > > Some kernel code assumes "linux,initrd-{start,end}" are 64-bit,
>> > > other code assumes 32-bit.
>> >
>> > It can be either. The above code was a merge of arm64 and powerpc >> > both
>> > of which use 64-bit and still only runs on those arches. It looks >> > like
>> > some powerpc platforms may use 32-bit, but this would have been >> > broken
>> > before.

>> of_kexec_alloc_and_setup_fdt() is called from elf_64.c (in
>> arch/powerpc/kexec) which is for 64-bit powerpc platform only.
>
> 64-bit PPC could be writing 32-bit property values. The architecture
> size doesn't necessarily matter. And if the values came from the
> bootloader, who knows what size it used.
>
> This code is 32-bit powerpc only?:
>
> arch/powerpc/boot/main.c-   /* Tell the kernel initrd address via device 
> tree */
> arch/powerpc/boot/main.c:   setprop_val(chosen, "linux,initrd-start", 
> (u32)(initrd_addr));
> arch/powerpc/boot/main.c-   setprop_val(chosen, "linux,initrd-end", 
> (u32)(initrd_addr+initrd_size));

Historically that code was always built 32-bit, even when used with a
64-bit kernel.

These days it is also built 64-bit (for ppc64le).

It looks like the drivers/of/fdt.c code can handle either 64 or 32-bit,
so I guess that's why it seems to be working.

Although I'm not sure how much testing the 64-bit case gets, because the
distros tend to just use the vmlinux.

cheers


Re: [PATCH v19 05/13] of: Add a common kexec FDT setup function

2021-06-15 Thread Rob Herring
On Tue, Jun 15, 2021 at 10:13 AM nramas  wrote:
>
> On Tue, 2021-06-15 at 08:01 -0600, Rob Herring wrote:
> > On Tue, Jun 15, 2021 at 6:18 AM Geert Uytterhoeven <
> > ge...@linux-m68k.org> wrote:
> > >
> > > > +void *of_kexec_alloc_and_setup_fdt(const struct kimage *image,
> > > > +  unsigned long
> > > > initrd_load_addr,
> > > > +  unsigned long initrd_len,
> > > > +  const char *cmdline, size_t
> > > > extra_fdt_size)
> > > > +{
> > > > +   /* 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) {
> > > > +   ret = -EINVAL;
> > > > +   goto out;
> > > > +   }
> > > > +
> > > > +   tmp_end = fdt64_to_cpu(*((const fdt64_t *)
> > > > prop));
> > >
> > > Some kernel code assumes "linux,initrd-{start,end}" are 64-bit,
> > > other code assumes 32-bit.
> >
> > It can be either. The above code was a merge of arm64 and powerpc
> > both
> > of which use 64-bit and still only runs on those arches. It looks
> > like
> > some powerpc platforms may use 32-bit, but this would have been
> > broken
> > before.
> of_kexec_alloc_and_setup_fdt() is called from elf_64.c (in
> arch/powerpc/kexec) which is for 64-bit powerpc platform only.

64-bit PPC could be writing 32-bit property values. The architecture
size doesn't necessarily matter. And if the values came from the
bootloader, who knows what size it used.

This code is 32-bit powerpc only?:

arch/powerpc/boot/main.c-   /* Tell the kernel initrd address via
device tree */
arch/powerpc/boot/main.c:   setprop_val(chosen,
"linux,initrd-start", (u32)(initrd_addr));
arch/powerpc/boot/main.c-   setprop_val(chosen,
"linux,initrd-end", (u32)(initrd_addr+initrd_size));

Rob


Re: [PATCH v19 05/13] of: Add a common kexec FDT setup function

2021-06-15 Thread nramas
On Tue, 2021-06-15 at 08:01 -0600, Rob Herring wrote:
> On Tue, Jun 15, 2021 at 6:18 AM Geert Uytterhoeven <
> ge...@linux-m68k.org> wrote:
> > 
> > > +void *of_kexec_alloc_and_setup_fdt(const struct kimage *image,
> > > +  unsigned long
> > > initrd_load_addr,
> > > +  unsigned long initrd_len,
> > > +  const char *cmdline, size_t
> > > extra_fdt_size)
> > > +{
> > > +   /* 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) {
> > > +   ret = -EINVAL;
> > > +   goto out;
> > > +   }
> > > +
> > > +   tmp_end = fdt64_to_cpu(*((const fdt64_t *)
> > > prop));
> > 
> > Some kernel code assumes "linux,initrd-{start,end}" are 64-bit,
> > other code assumes 32-bit.
> 
> It can be either. The above code was a merge of arm64 and powerpc
> both
> of which use 64-bit and still only runs on those arches. It looks
> like
> some powerpc platforms may use 32-bit, but this would have been
> broken
> before.
of_kexec_alloc_and_setup_fdt() is called from elf_64.c (in
arch/powerpc/kexec) which is for 64-bit powerpc platform only.

thanks,
 -lakshmi

> 
> The code in drivers/of/fdt.c handles either case. We should probably
> refactor early_init_dt_check_for_initrd() and this function to use a
> common routine.
> 
> > linux/Documentation/arm/uefi.rst says 64-bit,
> > dt-schema/schemas/chosen.yaml says 32-bit.
> 
> We should fix that.
> 
> Rob



Re: [PATCH v19 05/13] of: Add a common kexec FDT setup function

2021-06-15 Thread Rob Herring
On Tue, Jun 15, 2021 at 6:18 AM Geert Uytterhoeven  wrote:
>
> Hi Lakshmi and Rob,
>
> On Sun, Feb 21, 2021 at 6:52 PM Lakshmi Ramasubramanian
>  wrote:
> > From: Rob Herring 
> >
> > 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_alloc_and_setup_fdt() and move it to "drivers/of/kexec.c".
> >
> > Signed-off-by: Rob Herring 
> > Signed-off-by: Lakshmi Ramasubramanian 
>
> > --- /dev/null
> > +++ b/drivers/of/kexec.c
>
> > +/*
> > + * of_kexec_alloc_and_setup_fdt - Alloc and setup a new Flattened Device 
> > Tree
> > + *
> > + * @image: kexec image being loaded.
> > + * @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.
> > + * @extra_fdt_size:Additional size for the new FDT buffer.
> > + *
> > + * Return: fdt on success, or NULL errno on error.
> > + */
> > +void *of_kexec_alloc_and_setup_fdt(const struct kimage *image,
> > +  unsigned long initrd_load_addr,
> > +  unsigned long initrd_len,
> > +  const char *cmdline, size_t 
> > extra_fdt_size)
> > +{
>
> > +   /* 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) {
> > +   ret = -EINVAL;
> > +   goto out;
> > +   }
> > +
> > +   tmp_end = fdt64_to_cpu(*((const fdt64_t *) prop));
>
> Some kernel code assumes "linux,initrd-{start,end}" are 64-bit,
> other code assumes 32-bit.

It can be either. The above code was a merge of arm64 and powerpc both
of which use 64-bit and still only runs on those arches. It looks like
some powerpc platforms may use 32-bit, but this would have been broken
before.

The code in drivers/of/fdt.c handles either case. We should probably
refactor early_init_dt_check_for_initrd() and this function to use a
common routine.

> linux/Documentation/arm/uefi.rst says 64-bit,
> dt-schema/schemas/chosen.yaml says 32-bit.

We should fix that.

Rob


Re: [PATCH v19 05/13] of: Add a common kexec FDT setup function

2021-06-15 Thread Geert Uytterhoeven
Hi Lakshmi and Rob,

On Sun, Feb 21, 2021 at 6:52 PM Lakshmi Ramasubramanian
 wrote:
> From: Rob Herring 
>
> 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_alloc_and_setup_fdt() and move it to "drivers/of/kexec.c".
>
> Signed-off-by: Rob Herring 
> Signed-off-by: Lakshmi Ramasubramanian 

> --- /dev/null
> +++ b/drivers/of/kexec.c

> +/*
> + * of_kexec_alloc_and_setup_fdt - Alloc and setup a new Flattened Device Tree
> + *
> + * @image: kexec image being loaded.
> + * @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.
> + * @extra_fdt_size:Additional size for the new FDT buffer.
> + *
> + * Return: fdt on success, or NULL errno on error.
> + */
> +void *of_kexec_alloc_and_setup_fdt(const struct kimage *image,
> +  unsigned long initrd_load_addr,
> +  unsigned long initrd_len,
> +  const char *cmdline, size_t extra_fdt_size)
> +{

> +   /* 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) {
> +   ret = -EINVAL;
> +   goto out;
> +   }
> +
> +   tmp_end = fdt64_to_cpu(*((const fdt64_t *) prop));

Some kernel code assumes "linux,initrd-{start,end}" are 64-bit,
other code assumes 32-bit.
linux/Documentation/arm/uefi.rst says 64-bit,
dt-schema/schemas/chosen.yaml says 32-bit.

> +
> +   /*
> +* 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)
> +   goto out;
> +   }
> +
> +   /* 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;
> +   }

Gr{oetje,eeting}s,

Geert

-- 
Geert Uytterhoeven -- There's lots of Linux beyond ia32 -- ge...@linux-m68k.org

In personal conversations with technical people, I call myself a hacker. But
when I'm talking to journalists I just say "programmer" or something like that.
-- Linus Torvalds


Re: [PATCH v19 05/13] of: Add a common kexec FDT setup function

2021-02-23 Thread Lakshmi Ramasubramanian

On 2/23/21 5:20 PM, Thiago Jung Bauermann wrote:


Lakshmi Ramasubramanian  writes:


From: Rob Herring 

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_alloc_and_setup_fdt() and move it to "drivers/of/kexec.c".

Signed-off-by: Rob Herring 
Signed-off-by: Lakshmi Ramasubramanian 
Fixes: 33488dc4d61f ("of: Add a common kexec FDT setup function")


A patch cannot fix itself. The world would be a much better place if it
could. :-)


:)




Reported-by: kernel test robot 
---
  drivers/of/Makefile |   6 +
  drivers/of/kexec.c  | 265 
  include/linux/of.h  |   5 +
  3 files changed, 276 insertions(+)
  create mode 100644 drivers/of/kexec.c


With that fixed:

Reviewed-by: Thiago Jung Bauermann 


Thanks for reviewing the patches Thiago.

 -lakshmi




Re: [PATCH v19 05/13] of: Add a common kexec FDT setup function

2021-02-23 Thread Thiago Jung Bauermann


Lakshmi Ramasubramanian  writes:

> From: Rob Herring 
>
> 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_alloc_and_setup_fdt() and move it to "drivers/of/kexec.c".
>
> Signed-off-by: Rob Herring 
> Signed-off-by: Lakshmi Ramasubramanian 
> Fixes: 33488dc4d61f ("of: Add a common kexec FDT setup function")

A patch cannot fix itself. The world would be a much better place if it
could. :-)

> Reported-by: kernel test robot 
> ---
>  drivers/of/Makefile |   6 +
>  drivers/of/kexec.c  | 265 
>  include/linux/of.h  |   5 +
>  3 files changed, 276 insertions(+)
>  create mode 100644 drivers/of/kexec.c

With that fixed:

Reviewed-by: Thiago Jung Bauermann 

-- 
Thiago Jung Bauermann
IBM Linux Technology Center


[PATCH v19 05/13] of: Add a common kexec FDT setup function

2021-02-21 Thread Lakshmi Ramasubramanian
From: Rob Herring 

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_alloc_and_setup_fdt() and move it to "drivers/of/kexec.c".

Signed-off-by: Rob Herring 
Signed-off-by: Lakshmi Ramasubramanian 
Fixes: 33488dc4d61f ("of: Add a common kexec FDT setup function")
Reported-by: kernel test robot 
---
 drivers/of/Makefile |   6 +
 drivers/of/kexec.c  | 265 
 include/linux/of.h  |   5 +
 3 files changed, 276 insertions(+)
 create mode 100644 drivers/of/kexec.c

diff --git a/drivers/of/Makefile b/drivers/of/Makefile
index 6e1e5212f058..c13b982084a3 100644
--- a/drivers/of/Makefile
+++ b/drivers/of/Makefile
@@ -14,4 +14,10 @@ obj-$(CONFIG_OF_RESOLVE)  += resolver.o
 obj-$(CONFIG_OF_OVERLAY) += overlay.o
 obj-$(CONFIG_OF_NUMA) += of_numa.o
 
+ifdef CONFIG_KEXEC_FILE
+ifdef CONFIG_OF_FLATTREE
+obj-y  += kexec.o
+endif
+endif
+
 obj-$(CONFIG_OF_UNITTEST) += unittest-data/
diff --git a/drivers/of/kexec.c b/drivers/of/kexec.c
new file mode 100644
index ..8fa8946cda42
--- /dev/null
+++ b/drivers/of/kexec.c
@@ -0,0 +1,265 @@
+// SPDX-License-Identifier: GPL-2.0-only
+/*
+ * 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
+ */
+
+#include 
+#include 
+#include 
+#include 
+#include 
+#include 
+#include 
+
+/* 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
+
+/*
+ * Additional space needed for the FDT buffer so that we can add initrd,
+ * bootargs, kaslr-seed, rng-seed, useable-memory-range and elfcorehdr.
+ */
+#define FDT_EXTRA_SPACE 0x1000
+
+/**
+ * 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.
+ */
+static int fdt_find_and_del_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++) {
+   u64 rsv_start, rsv_size;
+
+   ret = fdt_get_mem_rsv(fdt, i, _start, _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;
+}
+
+/*
+ * of_kexec_alloc_and_setup_fdt - Alloc and setup a new Flattened Device Tree
+ *
+ * @image: kexec image being loaded.
+ * @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.
+ * @extra_fdt_size:Additional size for the new FDT buffer.
+ *
+ * Return: fdt on success, or NULL errno on error.
+ */
+void *of_kexec_alloc_and_setup_fdt(const struct kimage *image,
+  unsigned long initrd_load_addr,
+  unsigned long initrd_len,
+  const char *cmdline, size_t extra_fdt_size)
+{
+   void *fdt;
+   int ret, chosen_node;
+   const void *prop;
+   size_t fdt_size;
+
+   fdt_size = fdt_totalsize(initial_boot_params) +
+