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