[PATCH] imx: imx8: ahab: refactor do_ahab_close command

[Oleksandr Suvorov]

From: Igor Opaniuk <igor.opan...@foundries.io>

Move an OEM closing logic to ahab_close() function to be able to use
it directly without calling a u-boot command.

Signed-off-by: Igor Opaniuk <igor.opan...@foundries.io>
Co-developed-by: Oleksandr Suvorov <oleksandr.suvo...@foundries.io>
Signed-off-by: Oleksandr Suvorov <oleksandr.suvo...@foundries.io>
---

 arch/arm/include/asm/arch-imx8/sys_proto.h |  1 +
 arch/arm/mach-imx/imx8/ahab.c              | 45 ++++++++++++++--------
 2 files changed, 30 insertions(+), 16 deletions(-)

diff --git a/arch/arm/include/asm/arch-imx8/sys_proto.h 
b/arch/arm/include/asm/arch-imx8/sys_proto.h
index e7625c42985..405e9bd3d81 100644
--- a/arch/arm/include/asm/arch-imx8/sys_proto.h
+++ b/arch/arm/include/asm/arch-imx8/sys_proto.h
@@ -23,6 +23,7 @@ struct pass_over_info_t {
 
 extern unsigned long boot_pointer[];
 void build_info(void);
+int ahab_close(void);
 int print_bootinfo(void);
 int sc_pm_setup_uart(sc_rsrc_t uart_rsrc, sc_pm_clock_rate_t clk_rate);
 int imx8_power_domain_lookup_name(const char *name,
diff --git a/arch/arm/mach-imx/imx8/ahab.c b/arch/arm/mach-imx/imx8/ahab.c
index b58b14ca9b4..a7891bf8d98 100644
--- a/arch/arm/mach-imx/imx8/ahab.c
+++ b/arch/arm/mach-imx/imx8/ahab.c
@@ -340,6 +340,32 @@ static int do_ahab_status(struct cmd_tbl *cmdtp, int flag, 
int argc,
        return 0;
 }
 
+int ahab_close(void)
+{
+       int err;
+       u16 lc;
+
+       err = sc_seco_chip_info(-1, &lc, NULL, NULL, NULL);
+       if (err != SC_ERR_NONE) {
+               printf("Error in get lifecycle\n");
+               return -EIO;
+       }
+
+       if (lc != 0x20) {
+               puts("Current lifecycle is NOT NXP closed, can't move to OEM 
closed\n");
+               display_life_cycle(lc);
+               return -EPERM;
+       }
+
+       err = sc_seco_forward_lifecycle(-1, 16);
+       if (err != SC_ERR_NONE) {
+               printf("Error in forward lifecycle to OEM closed\n");
+               return -EIO;
+       }
+
+       return 0;
+}
+
 static int confirm_close(void)
 {
        puts("Warning: Please ensure your sample is in NXP closed state, "
@@ -361,27 +387,14 @@ static int do_ahab_close(struct cmd_tbl *cmdtp, int flag, 
int argc,
 {
        int confirmed = argc >= 2 && !strcmp(argv[1], "-y");
        int err;
-       u16 lc;
 
        if (!confirmed && !confirm_close())
                return -EACCES;
 
-       err = sc_seco_chip_info(-1, &lc, NULL, NULL, NULL);
+       err = ahab_close();
        if (err) {
-               printf("Error in get lifecycle\n");
-               return -EIO;
-       }
-
-       if (lc != 0x20) {
-               puts("Current lifecycle is NOT NXP closed, can't move to OEM 
closed\n");
-               display_life_cycle(lc);
-               return -EPERM;
-       }
-
-       err = sc_seco_forward_lifecycle(-1, 16);
-       if (err) {
-               printf("Error in forward lifecycle to OEM closed\n");
-               return -EIO;
+               printf("Change to OEM closed failed\n");
+               return err;
        }
 
        printf("Change to OEM closed successfully\n");
-- 
2.41.0