ddr: altera: Add ECC DRAM scrubbing support for Arria10

[u-boot] / cmd / sata.c

diff --git a/cmd/sata.c b/cmd/sata.c
index d34a56dccce04fb43a961ddd60823d28b66a5e7d..cc12afb07e697728dce15460fd3189d3d44d5ddf 100644 (file)
--- a/cmd/sata.c
+++ b/cmd/sata.c
@@ -1,3 +1,4 @@
+// SPDX-License-Identifier: GPL-2.0+
 /*
  * Copyright (C) 2000-2005, DENX Software Engineering
  *             Wolfgang Denk <wd@denx.de>
@@ -6,38 +7,109 @@
  *                     <mushtaqk_921@yahoo.co.in>
  * Copyright (C) 2008 Freescale Semiconductor, Inc.
  *             Dave Liu <daveliu@freescale.com>
- *
- * SPDX-License-Identifier:    GPL-2.0+
  */
 
 #include <common.h>
+#include <ahci.h>
+#include <dm.h>
 #include <command.h>
 #include <part.h>
 #include <sata.h>
+#include <dm/device-internal.h>
+#include <dm/uclass-internal.h>
 
 static int sata_curr_device = -1;
 
+int sata_remove(int devnum)
+{
+#ifdef CONFIG_AHCI
+       struct udevice *dev;
+       int rc;
+
+       rc = uclass_find_device(UCLASS_AHCI, devnum, &dev);
+       if (!rc && !dev)
+               rc = uclass_find_first_device(UCLASS_AHCI, &dev);
+       if (rc || !dev) {
+               printf("Cannot find SATA device %d (err=%d)\n", devnum, rc);
+               return CMD_RET_FAILURE;
+       }
+
+       rc = device_remove(dev, DM_REMOVE_NORMAL);
+       if (rc) {
+               printf("Cannot remove SATA device '%s' (err=%d)\n", dev->name,
+                      rc);
+               return CMD_RET_FAILURE;
+       }
+
+       return 0;
+#else
+       return sata_stop();
+#endif
+}
+
+int sata_probe(int devnum)
+{
+#ifdef CONFIG_AHCI
+       struct udevice *dev;
+       struct udevice *blk;
+       int rc;
+
+       rc = uclass_get_device(UCLASS_AHCI, devnum, &dev);
+       if (rc)
+               rc = uclass_find_first_device(UCLASS_AHCI, &dev);
+       if (rc) {
+               printf("Cannot probe SATA device %d (err=%d)\n", devnum, rc);
+               return CMD_RET_FAILURE;
+       }
+       rc = sata_scan(dev);
+       if (rc) {
+               printf("Cannot scan SATA device %d (err=%d)\n", devnum, rc);
+               return CMD_RET_FAILURE;
+       }
+
+       rc = blk_get_from_parent(dev, &blk);
+       if (!rc) {
+               struct blk_desc *desc = dev_get_uclass_platdata(blk);
+
+               if (desc->lba > 0 && desc->blksz > 0)
+                       part_init(desc);
+       }
+
+       return 0;
+#else
+       return sata_initialize() < 0 ? CMD_RET_FAILURE : CMD_RET_SUCCESS;
+#endif
+}
+
 static int do_sata(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 {
        int rc = 0;
 
-       if (argc == 2 && strcmp(argv[1], "stop") == 0)
-               return sata_stop();
+       if (argc >= 2) {
+               int devnum = 0;
+
+               if (argc == 3)
+                       devnum = (int)simple_strtoul(argv[2], NULL, 10);
+               if (!strcmp(argv[1], "stop"))
+                       return sata_remove(devnum);
 
-       if (argc == 2 && strcmp(argv[1], "init") == 0) {
-               if (sata_curr_device != -1)
-                       sata_stop();
+               if (!strcmp(argv[1], "init")) {
+                       if (sata_curr_device != -1) {
+                               rc = sata_remove(devnum);
+                               if (rc)
+                                       return rc;
+                       }
 
-               return (sata_initialize() < 0) ?
-                       CMD_RET_FAILURE : CMD_RET_SUCCESS;
+                       return sata_probe(devnum);
+               }
        }
 
        /* If the user has not yet run `sata init`, do it now */
        if (sata_curr_device == -1) {
-               rc = sata_initialize();
-               if (rc == -1)
+               rc = sata_probe(0);
+               if (rc < 0)
                        return CMD_RET_FAILURE;
-               sata_curr_device = rc;
+               sata_curr_device = 0;
        }
 
        return blk_common_cmd(argc, argv, IF_TYPE_SATA, &sata_curr_device);
@@ -47,7 +119,7 @@ U_BOOT_CMD(
        sata, 5, 1, do_sata,
        "SATA sub system",
        "init - init SATA sub system\n"
-       "sata stop - disable SATA sub system\n"
+       "sata stop [dev] - disable SATA sub system or device\n"
        "sata info - show available SATA devices\n"
        "sata device [dev] - show or set current device\n"
        "sata part [dev] - print partition table\n"