Handling probing code for a particular uclass between
dm vs nodm always confusing and requires additional
ifdefs to handle them properly.

But, having separate low-level code bases for dm and
nodm can make it easy for the command level to use same
function name to probe the devices. This would indeed
avoid extra ifdef call in source code.

So, this patch probes the spi flash in common legacy
call spi_flash_probe for both dm and nodm devices and
give a chance to handle on respective code bases based
on the build files.

Cc: Simon Glass <s...@chromium.org>
Cc: Vignesh R <vigne...@ti.com>
Cc: Daniel Schwierzeck <daniel.schwierz...@gmail.com>
Signed-off-by: Jagan Teki <ja...@amarulasolutions.com>
---
 cmd/sf.c                    | 22 ---------------------
 drivers/mtd/spi/sf-uclass.c | 38 +++++++++++++------------------------
 drivers/net/fm/fm.c         | 20 -------------------
 env/sf.c                    | 17 +----------------
 include/spi_flash.h         | 20 +++++--------------
 5 files changed, 19 insertions(+), 98 deletions(-)

diff --git a/cmd/sf.c b/cmd/sf.c
index 302201c2b0..3835dd857d 100644
--- a/cmd/sf.c
+++ b/cmd/sf.c
@@ -87,12 +87,7 @@ static int do_spi_flash_probe(int argc, char * const argv[])
        unsigned int speed = CONFIG_SF_DEFAULT_SPEED;
        unsigned int mode = CONFIG_SF_DEFAULT_MODE;
        char *endp;
-#ifdef CONFIG_DM_SPI_FLASH
-       struct udevice *new, *bus_dev;
-       int ret;
-#else
        struct spi_flash *new;
-#endif
 
        if (argc >= 2) {
                cs = simple_strtoul(argv[1], &endp, 0);
@@ -120,22 +115,6 @@ static int do_spi_flash_probe(int argc, char * const 
argv[])
                        return -1;
        }
 
-#ifdef CONFIG_DM_SPI_FLASH
-       /* Remove the old device, otherwise probe will just be a nop */
-       ret = spi_find_bus_and_cs(bus, cs, &bus_dev, &new);
-       if (!ret) {
-               device_remove(new, DM_REMOVE_NORMAL);
-       }
-       flash = NULL;
-       ret = spi_flash_probe_bus_cs(bus, cs, speed, mode, &new);
-       if (ret) {
-               printf("Failed to initialize SPI flash at %u:%u (error %d)\n",
-                      bus, cs, ret);
-               return 1;
-       }
-
-       flash = dev_get_uclass_priv(new);
-#else
        if (flash)
                spi_flash_free(flash);
 
@@ -145,7 +124,6 @@ static int do_spi_flash_probe(int argc, char * const argv[])
                printf("Failed to initialize SPI flash at %u:%u\n", bus, cs);
                return 1;
        }
-#endif
 
        return 0;
 }
diff --git a/drivers/mtd/spi/sf-uclass.c b/drivers/mtd/spi/sf-uclass.c
index 97a3f5d2c7..4993f7cd02 100644
--- a/drivers/mtd/spi/sf-uclass.c
+++ b/drivers/mtd/spi/sf-uclass.c
@@ -29,50 +29,38 @@ int spi_flash_erase_dm(struct udevice *dev, u32 offset, 
size_t len)
        return log_ret(sf_get_ops(dev)->erase(dev, offset, len));
 }
 
-/*
- * TODO(s...@chromium.org): This is an old-style function. We should remove
- * it when all SPI flash drivers use dm
- */
-struct spi_flash *spi_flash_probe(unsigned int bus, unsigned int cs,
-                                 unsigned int max_hz, unsigned int spi_mode)
-{
-       struct udevice *dev;
-
-       if (spi_flash_probe_bus_cs(bus, cs, max_hz, spi_mode, &dev))
-               return NULL;
-
-       return dev_get_uclass_priv(dev);
-}
-
 void spi_flash_free(struct spi_flash *flash)
 {
        device_remove(flash->spi->dev, DM_REMOVE_NORMAL);
 }
 
-int spi_flash_probe_bus_cs(unsigned int busnum, unsigned int cs,
-                          unsigned int max_hz, unsigned int spi_mode,
-                          struct udevice **devp)
+struct spi_flash *spi_flash_probe(unsigned int bus, unsigned int cs,
+                                 unsigned int max_hz, unsigned int spi_mode)
 {
        struct spi_slave *slave;
-       struct udevice *bus;
+       struct udevice *new, *bus_dev;
        char *str;
        int ret;
 
+       /* Remove the old device, otherwise probe will just be a nop */
+       ret = spi_find_bus_and_cs(bus, cs, &bus_dev, &new);
+       if (!ret)
+               device_remove(new, DM_REMOVE_NORMAL);
+
 #if defined(CONFIG_SPL_BUILD) && CONFIG_IS_ENABLED(USE_TINY_PRINTF)
        str = "spi_flash";
 #else
        char name[30];
 
-       snprintf(name, sizeof(name), "spi_flash@%d:%d", busnum, cs);
+       snprintf(name, sizeof(name), "spi_flash@%d:%d", bus, cs);
        str = strdup(name);
 #endif
-       ret = spi_get_bus_and_cs(busnum, cs, max_hz, spi_mode,
-                                 "spi_flash_std", str, &bus, &slave);
+       ret = spi_get_bus_and_cs(bus, cs, max_hz, spi_mode,
+                                "spi_flash_std", str, &bus_dev, &slave);
        if (ret)
-               return ret;
+               return NULL;
 
-       *devp = slave->dev;
-       return 0;
+       return dev_get_uclass_priv(slave->dev);
 }
 
 static int spi_flash_post_bind(struct udevice *dev)
diff --git a/drivers/net/fm/fm.c b/drivers/net/fm/fm.c
index 8ab1816395..811d1cd4fa 100644
--- a/drivers/net/fm/fm.c
+++ b/drivers/net/fm/fm.c
@@ -383,20 +383,10 @@ int fm_init_common(int index, struct ccsr_fman *reg)
                addr = malloc(CONFIG_SYS_QE_FMAN_FW_LENGTH);
                int ret = 0;
 
-#ifdef CONFIG_DM_SPI_FLASH
-               struct udevice *new;
-
-               /* speed and mode will be read from DT */
-               ret = spi_flash_probe_bus_cs(CONFIG_ENV_SPI_BUS,
-                                            CONFIG_ENV_SPI_CS, 0, 0, &new);
-
-               ucode_flash = dev_get_uclass_priv(new);
-#else
                ucode_flash = spi_flash_probe(CONFIG_ENV_SPI_BUS,
                                              CONFIG_ENV_SPI_CS,
                                              CONFIG_ENV_SPI_MAX_HZ,
                                              CONFIG_ENV_SPI_MODE);
-#endif
                if (!ucode_flash) {
                        printf("SF: probe for ucode failed\n");
                } else {
@@ -470,18 +460,8 @@ int fm_init_common(int index, struct ccsr_fman *reg)
        void *addr = malloc(CONFIG_SYS_QE_FMAN_FW_LENGTH);
        int ret = 0;
 
-#ifdef CONFIG_DM_SPI_FLASH
-       struct udevice *new;
-
-       /* speed and mode will be read from DT */
-       ret = spi_flash_probe_bus_cs(CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS,
-                                    0, 0, &new);
-
-       ucode_flash = dev_get_uclass_priv(new);
-#else
        ucode_flash = spi_flash_probe(CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS,
                        CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE);
-#endif
        if (!ucode_flash)
                printf("SF: probe for ucode failed\n");
        else {
diff --git a/env/sf.c b/env/sf.c
index af59c8375c..253f467832 100644
--- a/env/sf.c
+++ b/env/sf.c
@@ -35,21 +35,6 @@ static struct spi_flash *env_flash;
 
 static int setup_flash_device(void)
 {
-#ifdef CONFIG_DM_SPI_FLASH
-       struct udevice *new;
-       int     ret;
-
-       /* speed and mode will be read from DT */
-       ret = spi_flash_probe_bus_cs(CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS,
-                                    CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE,
-                                    &new);
-       if (ret) {
-               env_set_default("spi_flash_probe_bus_cs() failed", 0);
-               return ret;
-       }
-
-       env_flash = dev_get_uclass_priv(new);
-#else
        struct spi_flash *new;
 
        if (env_flash)
@@ -62,7 +47,7 @@ static int setup_flash_device(void)
                env_set_default("spi_flash_probe() failed", 0);
                return -EIO;
        }
-#endif
+
        return 0;
 }
 
diff --git a/include/spi_flash.h b/include/spi_flash.h
index 3b33b970ef..c270c6eff7 100644
--- a/include/spi_flash.h
+++ b/include/spi_flash.h
@@ -75,17 +75,6 @@ int spi_flash_write_dm(struct udevice *dev, u32 offset, 
size_t len,
  */
 int spi_flash_erase_dm(struct udevice *dev, u32 offset, size_t len);
 
-int spi_flash_probe_bus_cs(unsigned int busnum, unsigned int cs,
-                          unsigned int max_hz, unsigned int spi_mode,
-                          struct udevice **devp);
-
-/* Compatibility function - this is the old U-Boot API */
-struct spi_flash *spi_flash_probe(unsigned int bus, unsigned int cs,
-                                 unsigned int max_hz, unsigned int spi_mode);
-
-/* Compatibility function - this is the old U-Boot API */
-void spi_flash_free(struct spi_flash *flash);
-
 static inline int spi_flash_read(struct spi_flash *flash, u32 offset,
                                 size_t len, void *buf)
 {
@@ -112,10 +101,6 @@ int sandbox_sf_bind_emul(struct sandbox_state *state, int 
busnum, int cs,
 void sandbox_sf_unbind_emul(struct sandbox_state *state, int busnum, int cs);
 
 #else
-struct spi_flash *spi_flash_probe(unsigned int bus, unsigned int cs,
-               unsigned int max_hz, unsigned int spi_mode);
-
-void spi_flash_free(struct spi_flash *flash);
 
 static inline int spi_flash_read(struct spi_flash *flash, u32 offset,
                size_t len, void *buf)
@@ -154,6 +139,11 @@ static inline int spi_flash_erase(struct spi_flash *flash, 
u32 offset,
 }
 #endif
 
+struct spi_flash *spi_flash_probe(unsigned int bus, unsigned int cs,
+                                 unsigned int max_hz, unsigned int spi_mode);
+
+void spi_flash_free(struct spi_flash *flash);
+
 static inline int spi_flash_protect(struct spi_flash *flash, u32 ofs, u32 len,
                                        bool prot)
 {
-- 
2.20.1

Reply via email to