Prepare v2023.10
[platform/kernel/u-boot.git] / drivers / mtd / mtd_uboot.c
index 9360d4e..14ce726 100644 (file)
@@ -9,9 +9,11 @@
 #include <malloc.h>
 #include <dm/device.h>
 #include <dm/uclass-internal.h>
+#include <dm/uclass.h>
 #include <linux/err.h>
 #include <linux/mtd/mtd.h>
 #include <linux/mtd/partitions.h>
+#include <asm/global_data.h>
 #include <mtd.h>
 
 #define MTD_NAME_MAX_LEN 20
@@ -28,8 +30,6 @@ static const char *get_mtdids(void)
 
 #if defined(CONFIG_SYS_MTDPARTS_RUNTIME)
        board_mtdparts_default(&mtdids, &mtdparts);
-#elif defined(MTDIDS_DEFAULT)
-       mtdids = MTDIDS_DEFAULT;
 #elif defined(CONFIG_MTDIDS_DEFAULT)
        mtdids = CONFIG_MTDIDS_DEFAULT;
 #endif
@@ -52,7 +52,7 @@ static const char *get_mtdids(void)
  * @altname: Alternate name to return
  * @max_len: Length of the alternate name buffer
  *
- * @return 0 on success, an error otherwise.
+ * Return: 0 on success, an error otherwise.
  */
 int mtd_search_alternate_name(const char *mtdname, char *altname,
                              unsigned int max_len)
@@ -105,18 +105,26 @@ int mtd_search_alternate_name(const char *mtdname, char *altname,
 static void mtd_probe_uclass_mtd_devs(void)
 {
        struct udevice *dev;
-       int idx = 0;
 
-       /* Probe devices with DM compliant drivers */
-       while (!uclass_find_device(UCLASS_MTD, idx, &dev) && dev) {
-               mtd_probe(dev);
-               idx++;
-       }
+       uclass_foreach_dev_probe(UCLASS_MTD, dev)
+               ;
 }
 #else
 static void mtd_probe_uclass_mtd_devs(void) { }
 #endif
 
+#if IS_ENABLED(CONFIG_DM_SPI_FLASH) && IS_ENABLED(CONFIG_SPI_FLASH_MTD)
+static void mtd_probe_uclass_spi_nor_devs(void)
+{
+       struct udevice *dev;
+
+       uclass_foreach_dev_probe(UCLASS_SPI_FLASH, dev)
+               ;
+}
+#else
+static void mtd_probe_uclass_spi_nor_devs(void) { }
+#endif
+
 #if defined(CONFIG_MTD_PARTITIONS)
 
 #define MTDPARTS_MAXLEN         512
@@ -137,8 +145,6 @@ static const char *get_mtdparts(void)
 
 #if defined(CONFIG_SYS_MTDPARTS_RUNTIME)
        board_mtdparts_default(&mtdids, &mtdparts);
-#elif defined(MTDPARTS_DEFAULT)
-       mtdparts = MTDPARTS_DEFAULT;
 #elif defined(CONFIG_MTDPARTS_DEFAULT)
        mtdparts = CONFIG_MTDPARTS_DEFAULT;
 #endif
@@ -197,53 +203,11 @@ static void mtd_del_all_parts(void)
        } while (ret > 0);
 }
 
-int mtd_probe_devices(void)
+static int parse_mtdparts(const char *mtdparts, const char *mtdids)
 {
-       static char *old_mtdparts;
-       static char *old_mtdids;
-       const char *mtdparts = get_mtdparts();
-       const char *mtdids = get_mtdids();
-       const char *mtdparts_next = mtdparts;
+       const char *mtdparts_next;
        struct mtd_info *mtd;
 
-       mtd_probe_uclass_mtd_devs();
-
-       /*
-        * Check if mtdparts/mtdids changed, if the MTD dev list was updated
-        * or if our previous attempt to delete existing partititions failed.
-        * In any of these cases we want to update the partitions, otherwise,
-        * everything is up-to-date and we can return 0 directly.
-        */
-       if ((!mtdparts && !old_mtdparts && !mtdids && !old_mtdids) ||
-           (mtdparts && old_mtdparts && mtdids && old_mtdids &&
-            !mtd_dev_list_updated() && !mtd_del_all_parts_failed &&
-            !strcmp(mtdparts, old_mtdparts) &&
-            !strcmp(mtdids, old_mtdids)))
-               return 0;
-
-       /* Update the local copy of mtdparts */
-       free(old_mtdparts);
-       free(old_mtdids);
-       old_mtdparts = strdup(mtdparts);
-       old_mtdids = strdup(mtdids);
-
-       /*
-        * Remove all old parts. Note that partition removal can fail in case
-        * one of the partition is still being used by an MTD user, so this
-        * does not guarantee that all old partitions are gone.
-        */
-       mtd_del_all_parts();
-
-       /*
-        * Call mtd_dev_list_updated() to clear updates generated by our own
-        * parts removal loop.
-        */
-       mtd_dev_list_updated();
-
-       /* If either mtdparts or mtdids is empty, then exit */
-       if (!mtdparts || !mtdids)
-               return 0;
-
        /* Start the parsing by ignoring the extra 'mtdparts=' prefix, if any */
        if (!strncmp(mtdparts, "mtdparts=", sizeof("mtdparts=") - 1))
                mtdparts += 9;
@@ -342,6 +306,67 @@ int mtd_probe_devices(void)
                put_mtd_device(mtd);
        }
 
+       return 0;
+}
+
+int mtd_probe_devices(void)
+{
+       static char *old_mtdparts;
+       static char *old_mtdids;
+       const char *mtdparts = get_mtdparts();
+       const char *mtdids = get_mtdids();
+       struct mtd_info *mtd;
+
+       mtd_probe_uclass_mtd_devs();
+       mtd_probe_uclass_spi_nor_devs();
+
+       /*
+        * Check if mtdparts/mtdids changed, if the MTD dev list was updated
+        * or if our previous attempt to delete existing partititions failed.
+        * In any of these cases we want to update the partitions, otherwise,
+        * everything is up-to-date and we can return 0 directly.
+        */
+       if ((!mtdparts && !old_mtdparts && !mtdids && !old_mtdids) ||
+           (mtdparts && old_mtdparts && mtdids && old_mtdids &&
+            !mtd_dev_list_updated() && !mtd_del_all_parts_failed &&
+            !strcmp(mtdparts, old_mtdparts) &&
+            !strcmp(mtdids, old_mtdids)))
+               return 0;
+
+       /* Update the local copy of mtdparts */
+       free(old_mtdparts);
+       free(old_mtdids);
+       old_mtdparts = strdup(mtdparts);
+       old_mtdids = strdup(mtdids);
+
+       /*
+        * Remove all old parts. Note that partition removal can fail in case
+        * one of the partition is still being used by an MTD user, so this
+        * does not guarantee that all old partitions are gone.
+        */
+       mtd_del_all_parts();
+
+       /*
+        * Call mtd_dev_list_updated() to clear updates generated by our own
+        * parts removal loop.
+        */
+       mtd_dev_list_updated();
+
+       /* If both mtdparts and mtdids are non-empty, parse */
+       if (mtdparts && mtdids) {
+               if (parse_mtdparts(mtdparts, mtdids) < 0)
+                       printf("Failed parsing MTD partitions from mtdparts!\n");
+       }
+
+       /* Fallback to OF partitions */
+       mtd_for_each_device(mtd) {
+               if (list_empty(&mtd->partitions)) {
+                       if (add_mtd_partitions_of(mtd) < 0)
+                               printf("Failed parsing MTD %s OF partitions!\n",
+                                       mtd->name);
+               }
+       }
+
        /*
         * Call mtd_dev_list_updated() to clear updates generated by our own
         * parts registration loop.
@@ -354,6 +379,7 @@ int mtd_probe_devices(void)
 int mtd_probe_devices(void)
 {
        mtd_probe_uclass_mtd_devs();
+       mtd_probe_uclass_spi_nor_devs();
 
        return 0;
 }