common/cmd_sata.c

   1 /*
   2  * Copyright (C) 2000-2005, DENX Software Engineering
   3  *              Wolfgang Denk <wd@denx.de>
   4  * Copyright (C) Procsys. All rights reserved.
   5  *              Mushtaq Khan <mushtaq_k@procsys.com>
   6  *                      <mushtaqk_921@yahoo.co.in>
   7  * Copyright (C) 2008 Freescale Semiconductor, Inc.
   8  *              Dave Liu <daveliu@freescale.com>
   9  *
  10  * SPDX-License-Identifier:     GPL-2.0+
  11  */
  12
  13 #include <common.h>
  14 #include <command.h>
  15 #include <part.h>
  16 #include <sata.h>
  17
  18 static int sata_curr_device = -1;
  19 block_dev_desc_t sata_dev_desc[CONFIG_SYS_SATA_MAX_DEVICE];
  20
  21 int __sata_initialize(void)
  22 {
  23         int rc;
  24         int i;
  25
  26         for (i = 0; i < CONFIG_SYS_SATA_MAX_DEVICE; i++) {
  27                 memset(&sata_dev_desc[i], 0, sizeof(struct block_dev_desc));
  28                 sata_dev_desc[i].if_type = IF_TYPE_SATA;
  29                 sata_dev_desc[i].dev = i;
  30                 sata_dev_desc[i].part_type = PART_TYPE_UNKNOWN;
  31                 sata_dev_desc[i].type = DEV_TYPE_HARDDISK;
  32                 sata_dev_desc[i].lba = 0;
  33                 sata_dev_desc[i].blksz = 512;
  34                 sata_dev_desc[i].log2blksz = LOG2(sata_dev_desc[i].blksz);
  35                 sata_dev_desc[i].block_read = sata_read;
  36                 sata_dev_desc[i].block_write = sata_write;
  37
  38                 rc = init_sata(i);
  39                 if (!rc) {
  40                         rc = scan_sata(i);
  41                         if (!rc && (sata_dev_desc[i].lba > 0) &&
  42                                 (sata_dev_desc[i].blksz > 0))
  43                                 init_part(&sata_dev_desc[i]);
  44                 }
  45         }
  46         sata_curr_device = 0;
  47         return rc;
  48 }
  49 int sata_initialize(void) __attribute__((weak,alias("__sata_initialize")));
  50
  51 __weak int __sata_stop(void)
  52 {
  53         int i, err = 0;
  54
  55         for (i = 0; i < CONFIG_SYS_SATA_MAX_DEVICE; i++)
  56                 err |= reset_sata(i);
  57
  58         if (err)
  59                 printf("Could not reset some SATA devices\n");
  60
  61         return err;
  62 }
  63 int sata_stop(void) __attribute__((weak, alias("__sata_stop")));
  64
  65 #ifdef CONFIG_PARTITIONS
  66 block_dev_desc_t *sata_get_dev(int dev)
  67 {
  68         return (dev < CONFIG_SYS_SATA_MAX_DEVICE) ? &sata_dev_desc[dev] : NULL;
  69 }
  70 #endif
  71
  72 static int do_sata(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  73 {
  74         int rc = 0;
  75
  76         if (argc == 2 && strcmp(argv[1], "stop") == 0)
  77                 return sata_stop();
  78
  79         if (argc == 2 && strcmp(argv[1], "init") == 0) {
  80                 if (sata_curr_device != -1)
  81                         sata_stop();
  82
  83                 return sata_initialize();
  84         }
  85
  86         /* If the user has not yet run `sata init`, do it now */
  87         if (sata_curr_device == -1)
  88                 if (sata_initialize())
  89                         return 1;
  90
  91         switch (argc) {
  92         case 0:
  93         case 1:
  94                 return CMD_RET_USAGE;
  95         case 2:
  96                 if (strncmp(argv[1],"inf", 3) == 0) {
  97                         int i;
  98                         putc('\n');
  99                         for (i = 0; i < CONFIG_SYS_SATA_MAX_DEVICE; ++i) {
 100                                 if (sata_dev_desc[i].type == DEV_TYPE_UNKNOWN)
 101                                         continue;
 102                                 printf ("SATA device %d: ", i);
 103                                 dev_print(&sata_dev_desc[i]);
 104                         }
 105                         return 0;
 106                 } else if (strncmp(argv[1],"dev", 3) == 0) {
 107                         if ((sata_curr_device < 0) || (sata_curr_device >= CONFIG_SYS_SATA_MAX_DEVICE)) {
 108                                 puts("\nno SATA devices available\n");
 109                                 return 1;
 110                         }
 111                         printf("\nSATA device %d: ", sata_curr_device);
 112                         dev_print(&sata_dev_desc[sata_curr_device]);
 113                         return 0;
 114                 } else if (strncmp(argv[1],"part",4) == 0) {
 115                         int dev, ok;
 116
 117                         for (ok = 0, dev = 0; dev < CONFIG_SYS_SATA_MAX_DEVICE; ++dev) {
 118                                 if (sata_dev_desc[dev].part_type != PART_TYPE_UNKNOWN) {
 119                                         ++ok;
 120                                         if (dev)
 121                                                 putc ('\n');
 122                                         print_part(&sata_dev_desc[dev]);
 123                                 }
 124                         }
 125                         if (!ok) {
 126                                 puts("\nno SATA devices available\n");
 127                                 rc ++;
 128                         }
 129                         return rc;
 130                 }
 131                 return CMD_RET_USAGE;
 132         case 3:
 133                 if (strncmp(argv[1], "dev", 3) == 0) {
 134                         int dev = (int)simple_strtoul(argv[2], NULL, 10);
 135
 136                         printf("\nSATA device %d: ", dev);
 137                         if (dev >= CONFIG_SYS_SATA_MAX_DEVICE) {
 138                                 puts ("unknown device\n");
 139                                 return 1;
 140                         }
 141                         dev_print(&sata_dev_desc[dev]);
 142
 143                         if (sata_dev_desc[dev].type == DEV_TYPE_UNKNOWN)
 144                                 return 1;
 145
 146                         sata_curr_device = dev;
 147
 148                         puts("... is now current device\n");
 149
 150                         return 0;
 151                 } else if (strncmp(argv[1], "part", 4) == 0) {
 152                         int dev = (int)simple_strtoul(argv[2], NULL, 10);
 153
 154                         if (sata_dev_desc[dev].part_type != PART_TYPE_UNKNOWN) {
 155                                 print_part(&sata_dev_desc[dev]);
 156                         } else {
 157                                 printf("\nSATA device %d not available\n", dev);
 158                                 rc = 1;
 159                         }
 160                         return rc;
 161                 }
 162                 return CMD_RET_USAGE;
 163
 164         default: /* at least 4 args */
 165                 if (strcmp(argv[1], "read") == 0) {
 166                         ulong addr = simple_strtoul(argv[2], NULL, 16);
 167                         ulong cnt = simple_strtoul(argv[4], NULL, 16);
 168                         ulong n;
 169                         lbaint_t blk = simple_strtoul(argv[3], NULL, 16);
 170
 171                         printf("\nSATA read: device %d block # %ld, count %ld ... ",
 172                                 sata_curr_device, blk, cnt);
 173
 174                         n = sata_read(sata_curr_device, blk, cnt, (u32 *)addr);
 175
 176                         /* flush cache after read */
 177                         flush_cache(addr, cnt * sata_dev_desc[sata_curr_device].blksz);
 178
 179                         printf("%ld blocks read: %s\n",
 180                                 n, (n==cnt) ? "OK" : "ERROR");
 181                         return (n == cnt) ? 0 : 1;
 182                 } else if (strcmp(argv[1], "write") == 0) {
 183                         ulong addr = simple_strtoul(argv[2], NULL, 16);
 184                         ulong cnt = simple_strtoul(argv[4], NULL, 16);
 185                         ulong n;
 186
 187                         lbaint_t blk = simple_strtoul(argv[3], NULL, 16);
 188
 189                         printf("\nSATA write: device %d block # %ld, count %ld ... ",
 190                                 sata_curr_device, blk, cnt);
 191
 192                         n = sata_write(sata_curr_device, blk, cnt, (u32 *)addr);
 193
 194                         printf("%ld blocks written: %s\n",
 195                                 n, (n == cnt) ? "OK" : "ERROR");
 196                         return (n == cnt) ? 0 : 1;
 197                 } else {
 198                         return CMD_RET_USAGE;
 199                 }
 200
 201                 return rc;
 202         }
 203 }
 204
 205 U_BOOT_CMD(
 206         sata, 5, 1, do_sata,
 207         "SATA sub system",
 208         "init - init SATA sub system\n"
 209         "sata stop - disable SATA sub system\n"
 210         "sata info - show available SATA devices\n"
 211         "sata device [dev] - show or set current device\n"
 212         "sata part [dev] - print partition table\n"
 213         "sata read addr blk# cnt\n"
 214         "sata write addr blk# cnt"
 215 );