stm32mp: stm32prog: adapt the MTD partitions

[oweals/u-boot.git] / arch / arm / mach-stm32mp / cmd_stm32prog / stm32prog.c

diff --git a/arch/arm/mach-stm32mp/cmd_stm32prog/stm32prog.c b/arch/arm/mach-stm32mp/cmd_stm32prog/stm32prog.c
index 93ee6a55a1b1e63e080a720b48792e9c593ac97b..0140fd479dd3a64f6ddc79c6a8662e0fb2866abe 100644 (file)
--- a/arch/arm/mach-stm32mp/cmd_stm32prog/stm32prog.c
+++ b/arch/arm/mach-stm32mp/cmd_stm32prog/stm32prog.c
@@ -762,6 +762,8 @@ static int treat_partition_list(struct stm32prog_data *data)
                INIT_LIST_HEAD(&data->dev[j].part_list);
        }
 
+       data->tee_detected = false;
+       data->fsbl_nor_detected = false;
        for (i = 0; i < data->part_nb; i++) {
                part = &data->part_array[i];
                part->alt_id = -1;
@@ -806,6 +808,21 @@ static int treat_partition_list(struct stm32prog_data *data)
                        stm32prog_err("Layout: too many device");
                        return -EINVAL;
                }
+               switch (part->target)  {
+               case STM32PROG_NOR:
+                       if (!data->fsbl_nor_detected &&
+                           !strncmp(part->name, "fsbl", 4))
+                               data->fsbl_nor_detected = true;
+                       /* fallthrough */
+               case STM32PROG_NAND:
+               case STM32PROG_SPI_NAND:
+                       if (!data->tee_detected &&
+                           !strncmp(part->name, "tee", 3))
+                               data->tee_detected = true;
+                       break;
+               default:
+                       break;
+               }
                part->dev = &data->dev[j];
                if (!IS_SELECT(part))
                        part->dev->full_update = false;