stm32mp: stm32prog: correctly handle DM_PMIC

Correctly handle number of alternate when DM_PMIC is not activated.
This patch remove the last UNKNOWN partition in this case.

Signed-off-by: Patrick Delaunay <patrick.delaunay@foss.st.com>
Reviewed-by: Patrice Chotard <patrice.chotard@foss.st.com>
Acked-by: Jaehoon Chung <jh80.chung@samsung.com>

diff --git a/arch/arm/mach-stm32mp/cmd_stm32prog/stm32prog.c b/arch/arm/mach-stm32mp/cmd_stm32prog/stm32prog.c
index ea69d5d..ab687c2 100644 (file)
--- a/arch/arm/mach-stm32mp/cmd_stm32prog/stm32prog.c
+++ b/arch/arm/mach-stm32mp/cmd_stm32prog/stm32prog.c
@@ -1150,7 +1150,10 @@ static int dfu_init_entities(struct stm32prog_data *data)
        struct dfu_entity *dfu;
        int alt_nb;
 
-       alt_nb = 3; /* number of virtual = CMD, OTP, PMIC*/
+       alt_nb = 2; /* number of virtual = CMD, OTP*/
+       if (CONFIG_IS_ENABLED(DM_PMIC))
+               alt_nb++; /* PMIC NVMEM*/
+
        if (data->part_nb == 0)
                alt_nb++;  /* +1 for FlashLayout */
        else