[PATCH] riscv: Fix arch_fixup_fdt always failing without /chosen

[Sean Anderson]

If /chosen was missing, chosen_offset would never get updated with the new
/chosen node. This would cause fdt_setprop_u32 to fail. This patch fixes
this by setting chosen_offset. In addition, log any errors from setting
boot-hartid as well.

Fixes: 177c53fe6c6 ("riscv: Move all fdt fixups together")
Signed-off-by: Sean Anderson <sean...@gmail.com>
---
I have not actually tested this (nor observed the original failure). But this
seemed buggy from inspection.

 arch/riscv/lib/fdt_fixup.c | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/arch/riscv/lib/fdt_fixup.c b/arch/riscv/lib/fdt_fixup.c
index 1f3f23621c..f636b28449 100644
--- a/arch/riscv/lib/fdt_fixup.c
+++ b/arch/riscv/lib/fdt_fixup.c
@@ -151,14 +151,17 @@ int arch_fixup_fdt(void *blob)
        }
        chosen_offset = fdt_path_offset(blob, "/chosen");
        if (chosen_offset < 0) {
-               err = fdt_add_subnode(blob, 0, "chosen");
-               if (err < 0) {
+               chosen_offset = fdt_add_subnode(blob, 0, "chosen");
+               if (chosen_offset < 0) {
                        log_err("chosen node cannot be added\n");
-                       return err;
+                       return chosen_offset;
                }
        }
        /* Overwrite the boot-hartid as U-Boot is the last stage BL */
-       fdt_setprop_u32(blob, chosen_offset, "boot-hartid", gd->arch.boot_hart);
+       err = fdt_setprop_u32(blob, chosen_offset, "boot-hartid",
+                             gd->arch.boot_hart);
+       if (err < 0)
+               return log_msg_ret("could not set boot-hartid", err);
 #endif
 
        /* Copy the reserved-memory node to the DT used by OS */
-- 
2.31.0