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
__sata_initialize(void)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
__sata_stop(void)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
sata_get_dev(int dev)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
do_sata(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])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 );
216