[people/ms/u-boot.git] / common / cmd_sata.c

/*
 * 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>
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

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

static int sata_curr_device = -1;
block_dev_desc_t sata_dev_desc[CONFIG_SYS_SATA_MAX_DEVICE];

int __sata_initialize(void)
{
	int rc;
	int i;

	for (i = 0; i < CONFIG_SYS_SATA_MAX_DEVICE; i++) {
		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].log2blksz = LOG2(sata_dev_desc[i].blksz);
		sata_dev_desc[i].block_read = sata_read;
		sata_dev_desc[i].block_write = sata_write;

		rc = init_sata(i);
		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]);
		}
	}
	sata_curr_device = 0;
	return rc;
}
int sata_initialize(void) __attribute__((weak,alias("__sata_initialize")));

__weak int __sata_stop(void)
{
	int i, err = 0;

	for (i = 0; i < CONFIG_SYS_SATA_MAX_DEVICE; i++)
		err |= reset_sata(i);

	if (err)
		printf("Could not reset some SATA devices\n");

	return err;
}
int sata_stop(void) __attribute__((weak, alias("__sata_stop")));

#ifdef CONFIG_PARTITIONS
block_dev_desc_t *sata_get_dev(int dev)
{
	return (dev < CONFIG_SYS_SATA_MAX_DEVICE) ? &sata_dev_desc[dev] : NULL;
}
#endif

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

	if (argc == 2 && strcmp(argv[1], "stop") == 0)
		return sata_stop();

	if (argc == 2 && strcmp(argv[1], "init") == 0) {
		if (sata_curr_device != -1)
			sata_stop();

		return sata_initialize();
	}

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

	switch (argc) {
	case 0:
	case 1:
		return CMD_RET_USAGE;
	case 2:
		if (strncmp(argv[1],"inf", 3) == 0) {
			int i;
			putc('\n');
			for (i = 0; i < CONFIG_SYS_SATA_MAX_DEVICE; ++i) {
				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) {
			if ((sata_curr_device < 0) || (sata_curr_device >= CONFIG_SYS_SATA_MAX_DEVICE)) {
				puts("\nno SATA devices available\n");
				return 1;
			}
			printf("\nSATA device %d: ", sata_curr_device);
			dev_print(&sata_dev_desc[sata_curr_device]);
			return 0;
		} else if (strncmp(argv[1],"part",4) == 0) {
			int dev, ok;

			for (ok = 0, dev = 0; dev < CONFIG_SYS_SATA_MAX_DEVICE; ++dev) {
				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;
		}
		return CMD_RET_USAGE;
	case 3:
		if (strncmp(argv[1], "dev", 3) == 0) {
			int dev = (int)simple_strtoul(argv[2], NULL, 10);

			printf("\nSATA device %d: ", dev);
			if (dev >= CONFIG_SYS_SATA_MAX_DEVICE) {
				puts ("unknown device\n");
				return 1;
			}
			dev_print(&sata_dev_desc[dev]);

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

			sata_curr_device = dev;

			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;
		}
		return CMD_RET_USAGE;

	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 ... ",
				sata_curr_device, blk, cnt);

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

			/* flush cache after read */
			flush_cache(addr, cnt * sata_dev_desc[sata_curr_device].blksz);

			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 ... ",
				sata_curr_device, blk, cnt);

			n = sata_write(sata_curr_device, blk, cnt, (u32 *)addr);

			printf("%ld blocks written: %s\n",
				n, (n == cnt) ? "OK" : "ERROR");
			return (n == cnt) ? 0 : 1;
		} else {
			return CMD_RET_USAGE;
		}

		return rc;
	}
}

U_BOOT_CMD(
	sata, 5, 1, do_sata,
	"SATA sub system",
	"init - init SATA sub system\n"
	"sata stop - disable SATA sub system\n"
	"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"
	"sata write addr blk# cnt"
);
Commit	Line	Data
c7057b52 DL	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	*
1a459660	10	* SPDX-License-Identifier: GPL-2.0+
c7057b52 DL	11	*/
	12
	13	#include <common.h>
	14	#include <command.h>
	15	#include <part.h>
	16	#include <sata.h>
	17
088f1b19	18	static int sata_curr_device = -1;
6d0f6bcf	19	block_dev_desc_t sata_dev_desc[CONFIG_SYS_SATA_MAX_DEVICE];
c7057b52	20
cf7e399f	21	int __sata_initialize(void)
c7057b52 DL	22	{
	23	int rc;
	24	int i;
	25
6d0f6bcf	26	for (i = 0; i < CONFIG_SYS_SATA_MAX_DEVICE; i++) {
c7057b52 DL	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;
0472fbfd	34	sata_dev_desc[i].log2blksz = LOG2(sata_dev_desc[i].blksz);
c7057b52 DL	35	sata_dev_desc[i].block_read = sata_read;
	36	sata_dev_desc[i].block_write = sata_write;
	37
	38	rc = init_sata(i);
71cadda3 SB	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	}
c7057b52	45	}
569460eb	46	sata_curr_device = 0;
c7057b52 DL	47	return rc;
c7057b52 DL	48	}
cf7e399f	49	int sata_initialize(void) __attribute__((weak,alias("__sata_initialize")));
c7057b52	50
d957c28a NK	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
df3fc526	65	#ifdef CONFIG_PARTITIONS
c7057b52 DL	66	block_dev_desc_t *sata_get_dev(int dev)
c7057b52 DL	67	{
6d0f6bcf	68	return (dev < CONFIG_SYS_SATA_MAX_DEVICE) ? &sata_dev_desc[dev] : NULL;
c7057b52	69	}
df3fc526	70	#endif
c7057b52	71
088f1b19	72	static int do_sata(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
c7057b52 DL	73	{
	74	int rc = 0;
	75
d957c28a NK	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
cf7e399f	83	return sata_initialize();
d957c28a	84	}
cf7e399f MF	85
cf7e399f MF	86	/* If the user has not yet run `sata init`, do it now */
569460eb	87	if (sata_curr_device == -1)
cf7e399f MF	88	if (sata_initialize())
	89	return 1;
	90
c7057b52 DL	91	switch (argc) {
	92	case 0:
	93	case 1:
4c12eeb8	94	return CMD_RET_USAGE;
c7057b52 DL	95	case 2:
	96	if (strncmp(argv[1],"inf", 3) == 0) {
	97	int i;
	98	putc('\n');
6d0f6bcf	99	for (i = 0; i < CONFIG_SYS_SATA_MAX_DEVICE; ++i) {
c7057b52 DL	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) {
569460eb	107	if ((sata_curr_device < 0) \|\| (sata_curr_device >= CONFIG_SYS_SATA_MAX_DEVICE)) {
c7057b52 DL	108	puts("\nno SATA devices available\n");
	109	return 1;
	110	}
569460eb MF	111	printf("\nSATA device %d: ", sata_curr_device);
569460eb MF	112	dev_print(&sata_dev_desc[sata_curr_device]);
c7057b52 DL	113	return 0;
	114	} else if (strncmp(argv[1],"part",4) == 0) {
	115	int dev, ok;
	116
6d0f6bcf	117	for (ok = 0, dev = 0; dev < CONFIG_SYS_SATA_MAX_DEVICE; ++dev) {
c7057b52 DL	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	}
4c12eeb8	131	return CMD_RET_USAGE;
c7057b52 DL	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);
6d0f6bcf	137	if (dev >= CONFIG_SYS_SATA_MAX_DEVICE) {
c7057b52 DL	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
569460eb	146	sata_curr_device = dev;
c7057b52 DL	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	}
4c12eeb8	162	return CMD_RET_USAGE;
c7057b52 DL	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 ... ",
569460eb	172	sata_curr_device, blk, cnt);
c7057b52	173
569460eb	174	n = sata_read(sata_curr_device, blk, cnt, (u32 *)addr);
c7057b52 DL	175
c7057b52 DL	176	/* flush cache after read */
569460eb	177	flush_cache(addr, cnt * sata_dev_desc[sata_curr_device].blksz);
c7057b52 DL	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 ... ",
569460eb	190	sata_curr_device, blk, cnt);
c7057b52	191
569460eb	192	n = sata_write(sata_curr_device, blk, cnt, (u32 *)addr);
c7057b52 DL	193
	194	printf("%ld blocks written: %s\n",
	195	n, (n == cnt) ? "OK" : "ERROR");
	196	return (n == cnt) ? 0 : 1;
	197	} else {
4c12eeb8	198	return CMD_RET_USAGE;
c7057b52 DL	199	}
	200
	201	return rc;
	202	}
	203	}
	204
	205	U_BOOT_CMD(
	206	sata, 5, 1, do_sata,
2fb2604d	207	"SATA sub system",
85dafbb8	208	"init - init SATA sub system\n"
d957c28a	209	"sata stop - disable SATA sub system\n"
c7057b52 DL	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"
a89c33db WD	214	"sata write addr blk# cnt"
a89c33db WD	215	);