[PATCH v6 3/5] cmd: kaslrseed: Use common API to fixup FDT

[Sean Edmond]

Use the newly introduced common API fdt_fixup_kaslr_seed() in the
kaslrseed command.

Signed-off-by: Sean Edmond <seanedm...@microsoft.com>
Reviewed-by: Simon Glass <s...@chromium.org>

---

Changes in v6:
- root_ofnode_from_fdt()->ofnode_root_from_fdt()
- "goto CMD_RET_FAILURE" -> "return CMD_RET_FAILURE"

Changes in v5:
- cast buffer to (u8 *) in kaslrseed.c

 cmd/kaslrseed.c | 20 +++++++-------------
 1 file changed, 7 insertions(+), 13 deletions(-)

diff --git a/cmd/kaslrseed.c b/cmd/kaslrseed.c
index 9acb8e16386..9339d82649b 100644
--- a/cmd/kaslrseed.c
+++ b/cmd/kaslrseed.c
@@ -19,7 +19,7 @@ static int do_kaslr_seed(struct cmd_tbl *cmdtp, int flag, int 
argc, char *const
        size_t n = 0x8;
        struct udevice *dev;
        u64 *buf;
-       int nodeoffset;
+       ofnode root;
        int ret = CMD_RET_SUCCESS;
 
        if (uclass_get_device(UCLASS_RNG, 0, &dev) || !dev) {
@@ -45,21 +45,15 @@ static int do_kaslr_seed(struct cmd_tbl *cmdtp, int flag, 
int argc, char *const
                return CMD_RET_FAILURE;
        }
 
-       ret = fdt_check_header(working_fdt);
-       if (ret < 0) {
-               printf("fdt_chosen: %s\n", fdt_strerror(ret));
+       ret = ofnode_root_from_fdt(working_fdt, &root);
+       if (ret) {
+               printf("ERROR: Unable to get root ofnode\n");
                return CMD_RET_FAILURE;
        }
 
-       nodeoffset = fdt_find_or_add_subnode(working_fdt, 0, "chosen");
-       if (nodeoffset < 0) {
-               printf("Reading chosen node failed\n");
-               return CMD_RET_FAILURE;
-       }
-
-       ret = fdt_setprop(working_fdt, nodeoffset, "kaslr-seed", buf, 
sizeof(buf));
-       if (ret < 0) {
-               printf("Unable to set kaslr-seed on chosen node: %s\n", 
fdt_strerror(ret));
+       ret = fdt_fixup_kaslr_seed(root, (u8 *)buf, sizeof(buf));
+       if (ret) {
+               printf("ERROR: failed to add kaslr-seed to fdt\n");
                return CMD_RET_FAILURE;
        }
 
-- 
2.42.0