cmd: sf: Fix sf probe crash

Handle the return value of spi_flash_probe_bus_cs() to avoid sf probe
crashes.

Link: https://lore.kernel.org/all/[email protected]
Signed-off-by: Weizhao Ouyang <[email protected]>
Reviewed-by: Michael Trimarchi <[email protected]>
Acked-by: Michal Simek <[email protected]>
Reviewed-by: Jonas Karlman <[email protected]>
Signed-off-by: Dario Binacchi <[email protected]>

1 files changed, 3 insertions, 2 deletions

diff --git a/cmd/sf.c b/cmd/sf.c
index 730996c02b7..e3866899f6c 100644
--- a/cmd/sf.c
+++ b/cmd/sf.c
@@ -135,8 +135,9 @@ static int do_spi_flash_probe(int argc, char *const argv[])
 	}
 	flash = NULL;
 	if (use_dt) {
-		spi_flash_probe_bus_cs(bus, cs, &new);
-		flash = dev_get_uclass_priv(new);
+		ret = spi_flash_probe_bus_cs(bus, cs, &new);
+		if (!ret)
+			flash = dev_get_uclass_priv(new);
 	} else {
 		flash = spi_flash_probe(bus, cs, speed, mode);
 	}