cmd/sata.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Copyright (C) 2000-2005, DENX Software Engineering
   4  *              Wolfgang Denk <wd@denx.de>
   5  * Copyright (C) Procsys. All rights reserved.
   6  *              Mushtaq Khan <mushtaq_k@procsys.com>
   7  *                      <mushtaqk_921@yahoo.co.in>
   8  * Copyright (C) 2008 Freescale Semiconductor, Inc.
   9  *              Dave Liu <daveliu@freescale.com>
  10  */
  11
  12 #include <common.h>
  13 #include <ahci.h>
  14 #include <blk.h>
  15 #include <dm.h>
  16 #include <command.h>
  17 #include <part.h>
  18 #include <sata.h>
  19 #include <dm/device-internal.h>
  20 #include <dm/uclass-internal.h>
  21
  22 static int sata_curr_device = -1;
  23
  24 int sata_remove(int devnum)
  25 {
  26 #ifdef CONFIG_AHCI
  27         struct udevice *dev;
  28         int rc;
  29
  30         blk_unbind_all(IF_TYPE_SATA);
  31
  32         rc = uclass_find_device(UCLASS_AHCI, devnum, &dev);
  33         if (!rc && !dev)
  34                 rc = uclass_find_first_device(UCLASS_AHCI, &dev);
  35         if (rc || !dev) {
  36                 printf("Cannot find SATA device %d (err=%d)\n", devnum, rc);
  37                 return CMD_RET_FAILURE;
  38         }
  39
  40         rc = device_remove(dev, DM_REMOVE_NORMAL);
  41         if (rc) {
  42                 printf("Cannot remove SATA device '%s' (err=%d)\n", dev->name,
  43                        rc);
  44                 return CMD_RET_FAILURE;
  45         }
  46
  47         return 0;
  48 #else
  49         return sata_stop();
  50 #endif
  51 }
  52
  53 int sata_probe(int devnum)
  54 {
  55 #ifdef CONFIG_AHCI
  56         struct udevice *dev;
  57         int rc;
  58
  59         rc = uclass_get_device(UCLASS_AHCI, devnum, &dev);
  60         if (rc)
  61                 rc = uclass_find_first_device(UCLASS_AHCI, &dev);
  62         if (rc) {
  63                 printf("Cannot probe SATA device %d (err=%d)\n", devnum, rc);
  64                 return CMD_RET_FAILURE;
  65         }
  66         if (!dev) {
  67                 printf("No SATA device found!\n");
  68                 return CMD_RET_FAILURE;
  69         }
  70         rc = sata_scan(dev);
  71         if (rc) {
  72                 printf("Cannot scan SATA device %d (err=%d)\n", devnum, rc);
  73                 return CMD_RET_FAILURE;
  74         }
  75
  76         return 0;
  77 #else
  78         return sata_initialize() < 0 ? CMD_RET_FAILURE : CMD_RET_SUCCESS;
  79 #endif
  80 }
  81
  82 static int do_sata(struct cmd_tbl *cmdtp, int flag, int argc,
  83                    char *const argv[])
  84 {
  85         int rc = 0;
  86
  87         if (argc >= 2) {
  88                 int devnum = 0;
  89
  90                 if (argc == 3)
  91                         devnum = (int)simple_strtoul(argv[2], NULL, 10);
  92                 if (!strcmp(argv[1], "stop"))
  93                         return sata_remove(devnum);
  94
  95                 if (!strcmp(argv[1], "init")) {
  96                         if (sata_curr_device != -1) {
  97                                 rc = sata_remove(devnum);
  98                                 if (rc)
  99                                         return rc;
 100                         }
 101
 102                         return sata_probe(devnum);
 103                 }
 104         }
 105
 106         /* If the user has not yet run `sata init`, do it now */
 107         if (sata_curr_device == -1) {
 108                 rc = sata_probe(0);
 109                 if (rc)
 110                         return rc;
 111                 sata_curr_device = 0;
 112         }
 113
 114         return blk_common_cmd(argc, argv, IF_TYPE_SATA, &sata_curr_device);
 115 }
 116
 117 U_BOOT_CMD(
 118         sata, 5, 1, do_sata,
 119         "SATA sub system",
 120         "init - init SATA sub system\n"
 121         "sata stop [dev] - disable SATA sub system or device\n"
 122         "sata info - show available SATA devices\n"
 123         "sata device [dev] - show or set current device\n"
 124         "sata part [dev] - print partition table\n"
 125         "sata read addr blk# cnt\n"
 126         "sata write addr blk# cnt"
 127 );