cmd_sata.c 5.48 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
/*
 * Copyright (C) 2000-2005, DENX Software Engineering
 *		Wolfgang Denk <wd@denx.de>
 * Copyright (C) Procsys. All rights reserved.
 *		Mushtaq Khan <mushtaq_k@procsys.com>
 *			<mushtaqk_921@yahoo.co.in>
 * Copyright (C) 2008 Freescale Semiconductor, Inc.
 *		Dave Liu <daveliu@freescale.com>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of
 * the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
 * MA 02111-1307 USA
 */

#include <common.h>
#include <command.h>
#include <part.h>
#include <sata.h>

31
int sata_curr_device = -1;
32
block_dev_desc_t sata_dev_desc[CONFIG_SYS_SATA_MAX_DEVICE];
33

34
int __sata_initialize(void)
35 36 37 38
{
	int rc;
	int i;

39
	for (i = 0; i < CONFIG_SYS_SATA_MAX_DEVICE; i++) {
40 41 42 43 44 45 46 47 48 49 50
		memset(&sata_dev_desc[i], 0, sizeof(struct block_dev_desc));
		sata_dev_desc[i].if_type = IF_TYPE_SATA;
		sata_dev_desc[i].dev = i;
		sata_dev_desc[i].part_type = PART_TYPE_UNKNOWN;
		sata_dev_desc[i].type = DEV_TYPE_HARDDISK;
		sata_dev_desc[i].lba = 0;
		sata_dev_desc[i].blksz = 512;
		sata_dev_desc[i].block_read = sata_read;
		sata_dev_desc[i].block_write = sata_write;

		rc = init_sata(i);
51 52 53 54 55 56
		if (!rc) {
			rc = scan_sata(i);
			if (!rc && (sata_dev_desc[i].lba > 0) &&
				(sata_dev_desc[i].blksz > 0))
				init_part(&sata_dev_desc[i]);
		}
57
	}
58
	sata_curr_device = 0;
59 60
	return rc;
}
61
int sata_initialize(void) __attribute__((weak,alias("__sata_initialize")));
62

63
#ifdef CONFIG_PARTITIONS
64 65
block_dev_desc_t *sata_get_dev(int dev)
{
66
	return (dev < CONFIG_SYS_SATA_MAX_DEVICE) ? &sata_dev_desc[dev] : NULL;
67
}
68
#endif
69

70
int do_sata(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
71 72 73
{
	int rc = 0;

74 75 76 77
	if (argc == 2 && strcmp(argv[1], "init") == 0)
		return sata_initialize();

	/* If the user has not yet run `sata init`, do it now */
78
	if (sata_curr_device == -1)
79 80 81
		if (sata_initialize())
			return 1;

82 83 84
	switch (argc) {
	case 0:
	case 1:
85
		return CMD_RET_USAGE;
86 87 88 89
	case 2:
		if (strncmp(argv[1],"inf", 3) == 0) {
			int i;
			putc('\n');
90
			for (i = 0; i < CONFIG_SYS_SATA_MAX_DEVICE; ++i) {
91 92 93 94 95 96 97
				if (sata_dev_desc[i].type == DEV_TYPE_UNKNOWN)
					continue;
				printf ("SATA device %d: ", i);
				dev_print(&sata_dev_desc[i]);
			}
			return 0;
		} else if (strncmp(argv[1],"dev", 3) == 0) {
98
			if ((sata_curr_device < 0) || (sata_curr_device >= CONFIG_SYS_SATA_MAX_DEVICE)) {
99 100 101
				puts("\nno SATA devices available\n");
				return 1;
			}
102 103
			printf("\nSATA device %d: ", sata_curr_device);
			dev_print(&sata_dev_desc[sata_curr_device]);
104 105 106 107
			return 0;
		} else if (strncmp(argv[1],"part",4) == 0) {
			int dev, ok;

108
			for (ok = 0, dev = 0; dev < CONFIG_SYS_SATA_MAX_DEVICE; ++dev) {
109 110 111 112 113 114 115 116 117 118 119 120 121
				if (sata_dev_desc[dev].part_type != PART_TYPE_UNKNOWN) {
					++ok;
					if (dev)
						putc ('\n');
					print_part(&sata_dev_desc[dev]);
				}
			}
			if (!ok) {
				puts("\nno SATA devices available\n");
				rc ++;
			}
			return rc;
		}
122
		return CMD_RET_USAGE;
123 124 125 126 127
	case 3:
		if (strncmp(argv[1], "dev", 3) == 0) {
			int dev = (int)simple_strtoul(argv[2], NULL, 10);

			printf("\nSATA device %d: ", dev);
128
			if (dev >= CONFIG_SYS_SATA_MAX_DEVICE) {
129 130 131 132 133 134 135 136
				puts ("unknown device\n");
				return 1;
			}
			dev_print(&sata_dev_desc[dev]);

			if (sata_dev_desc[dev].type == DEV_TYPE_UNKNOWN)
				return 1;

137
			sata_curr_device = dev;
138 139 140 141 142 143 144 145 146 147 148 149 150 151 152

			puts("... is now current device\n");

			return 0;
		} else if (strncmp(argv[1], "part", 4) == 0) {
			int dev = (int)simple_strtoul(argv[2], NULL, 10);

			if (sata_dev_desc[dev].part_type != PART_TYPE_UNKNOWN) {
				print_part(&sata_dev_desc[dev]);
			} else {
				printf("\nSATA device %d not available\n", dev);
				rc = 1;
			}
			return rc;
		}
153
		return CMD_RET_USAGE;
154 155 156 157 158 159 160 161 162

	default: /* at least 4 args */
		if (strcmp(argv[1], "read") == 0) {
			ulong addr = simple_strtoul(argv[2], NULL, 16);
			ulong cnt = simple_strtoul(argv[4], NULL, 16);
			ulong n;
			lbaint_t blk = simple_strtoul(argv[3], NULL, 16);

			printf("\nSATA read: device %d block # %ld, count %ld ... ",
163
				sata_curr_device, blk, cnt);
164

165
			n = sata_read(sata_curr_device, blk, cnt, (u32 *)addr);
166 167

			/* flush cache after read */
168
			flush_cache(addr, cnt * sata_dev_desc[sata_curr_device].blksz);
169 170 171 172 173 174 175 176 177 178 179 180

			printf("%ld blocks read: %s\n",
				n, (n==cnt) ? "OK" : "ERROR");
			return (n == cnt) ? 0 : 1;
		} else if (strcmp(argv[1], "write") == 0) {
			ulong addr = simple_strtoul(argv[2], NULL, 16);
			ulong cnt = simple_strtoul(argv[4], NULL, 16);
			ulong n;

			lbaint_t blk = simple_strtoul(argv[3], NULL, 16);

			printf("\nSATA write: device %d block # %ld, count %ld ... ",
181
				sata_curr_device, blk, cnt);
182

183
			n = sata_write(sata_curr_device, blk, cnt, (u32 *)addr);
184 185 186 187 188

			printf("%ld blocks written: %s\n",
				n, (n == cnt) ? "OK" : "ERROR");
			return (n == cnt) ? 0 : 1;
		} else {
189
			return CMD_RET_USAGE;
190 191 192 193 194 195 196 197
		}

		return rc;
	}
}

U_BOOT_CMD(
	sata, 5, 1, do_sata,
Peter Tyser's avatar
Peter Tyser committed
198 199
	"SATA sub system",
	"sata init - init SATA sub system\n"
200 201 202 203
	"sata info - show available SATA devices\n"
	"sata device [dev] - show or set current device\n"
	"sata part [dev] - print partition table\n"
	"sata read addr blk# cnt\n"
204 205
	"sata write addr blk# cnt"
);