[z180-stamp.git] / avr / cmd_sd.c

#include "common.h"
#include <stdlib.h>

#include "command.h"
#include "diskio.h"
#include "ff.h"
#include "z80-if.h"
#include "print-utils.h"


/*
 * di <pd#> - Initialize disk
 *
 */
static
command_ret_t do_init(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	DSTATUS res;
	BYTE dev;

	(void) cmdtp; (void) flag;

	if (argc < 2)
		return CMD_RET_USAGE;

	dev = (BYTE) strtoul(argv[1], 0, 10);
	res = disk_status(dev);
	printf_P(PSTR("disk_status=%.2x\n"), res);

	if ((res & STA_NODISK) == 0) {
		res = disk_initialize(dev);
	}
	printf_P(PSTR("rc=%.2x\n"), res);
	if (res) {
		return CMD_RET_FAILURE;
	}

	return CMD_RET_SUCCESS;
}

/*
 * ds <pd#> - Show disk status
 *
 */
static
command_ret_t do_status(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	BYTE dev;

	union {
		unsigned char uca[64];
		unsigned long ul;
		unsigned char uc;
	} dat;

	(void) cmdtp; (void) flag;

	if (argc < 2)
		return CMD_RET_USAGE;

	dev = (BYTE) strtoul(argv[1], 0, 10);


	if (disk_ioctl(dev, GET_SECTOR_COUNT, &dat.ul) == RES_OK)
		printf_P(PSTR("Drive size: %lu sectors\n"), dat.ul);
	if (disk_ioctl(dev, GET_BLOCK_SIZE, &dat.ul) == RES_OK)
		printf_P(PSTR("Erase block: %lu sectors\n"), dat.ul);
	if (disk_ioctl(dev, MMC_GET_TYPE, &dat.uc) == RES_OK)
		printf_P(PSTR("Card type: %u\n"), dat.uc);
	if (disk_ioctl(dev, MMC_GET_CSD, dat.uca) == RES_OK)
		dump_ram(dat.uca, 0, 16, "CSD:");
	if (disk_ioctl(dev, MMC_GET_CID, dat.uca) == RES_OK)
		dump_ram(dat.uca, 0, 16, "CID:");
	if (disk_ioctl(dev, MMC_GET_OCR, dat.uca) == RES_OK)
		dump_ram(dat.uca, 0, 4, "OCR:");
	if (disk_ioctl(dev, MMC_GET_SDSTAT, dat.uca) == RES_OK)
		dump_ram(dat.uca, 0, 64, "SD Status:");

	return CMD_RET_SUCCESS;
}


/*
 * dump <pd#> [<sector> [count]] - Dump sector
 *
 */
static
command_ret_t do_dump(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	static BYTE dev_last;
	static 	DWORD sec_last;
	BYTE buffer[_MAX_SS];
	char header[20];

	DRESULT res;
	BYTE dev;
	DWORD sec;
	UINT count;

	(void) cmdtp; (void) flag;

	if (argc < 2)
		return CMD_RET_USAGE;

	dev = (BYTE) strtoul(argv[1], NULL, 10);
	if (dev != dev_last)
		sec_last = 0;
	sec = sec_last;
	count = 1;

	if ((flag & CMD_FLAG_REPEAT) == 0) {
		/* If another parameter, it is the sector to dump. */
		if (argc > 2)
			sec = strtoul(argv[2], 0, 10);
		if (argc > 3)
			count = (UINT) strtoul(argv[3], 0, 10);
	}

	for ( ; count; count--, sec++) {
		res = disk_read(dev, buffer, sec, 1);

		if (res) {
			printf_P(PSTR("rc=%.2x\n"), res);
			return CMD_RET_FAILURE;
		}

		sprintf_P(header, PSTR("Sector: %lu"), sec);
		dump_ram(buffer, 0, _MAX_SS, header);
	}
	dev_last = dev;
	sec_last = sec;

	return CMD_RET_SUCCESS;
}

/*
 * read drive sector count memaddr - Read disk into memory
 *
 */
static
command_ret_t do_read(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	DRESULT res;
	BYTE dev;
	DWORD sec;
	uint32_t addr;
	int count, nr;
	BYTE buffer[_MAX_SS];

	static DWORD sec_last;
	static uint32_t addr_last;

	(void) cmdtp;

	if (argc < 2)
		return CMD_RET_USAGE;

	dev = (BYTE) strtoul(argv[1], NULL, 10);
	sec = sec_last;
	count = 1;
	addr = addr_last;

	if ((flag & CMD_FLAG_REPEAT) == 0) {
		/* If another parameter, it is the sector to dump. */
		if (argc > 2)
			sec = strtoul(argv[2], 0, 10);
		if (argc > 3)
			count = strtoul(argv[3], 0, 10);
		if (argc > 4)
			addr = strtoul(argv[4], 0, 16);
	}

	for (nr = 0; nr < count;) {
		nr++;
		if ((res = disk_read(dev, buffer, sec, 1)) == RES_OK) {
			if (!(z80_bus_cmd(Request) & ZST_ACQUIRED)) {
				my_puts_P(PSTR("Bus timeout\n"));
				return  CMD_RET_FAILURE;
			}
			z80_write_block(buffer, addr /*+ base*/, _MAX_SS);
			z80_bus_cmd(Release);
			sec++; addr += _MAX_SS;
		} else
			break;
	}

	printf_P(PSTR("Read %d sector(s), rc=%.2x.\n"), nr, res);
	if (res)
		printf_P(PSTR("Last sector not written!\n"));

	sec_last = sec;
	addr_last = addr;

	return res ? CMD_RET_FAILURE : CMD_RET_SUCCESS;
}


/*
 * write <pd#> <sector> <memaddr> [<n>] - Write memory to disk
 *
 */
static
command_ret_t do_write(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	DRESULT res;
	BYTE dev;
	DWORD sec;
	uint32_t addr;
	int count, nr;
	BYTE buffer[_MAX_SS];

	static DWORD sec_last;
	static uint32_t addr_last;

	(void) cmdtp;

	if (argc < 2)
		return CMD_RET_USAGE;

	dev = (BYTE) strtoul(argv[1], NULL, 10);
	sec = sec_last;
	addr = addr_last;
	count = 1;

	if ((flag & CMD_FLAG_REPEAT) == 0) {
		/* If another parameter, it is the sector to dump. */
		if (argc > 2)
			sec = strtoul(argv[2], 0, 10);
		if (argc > 3)
			count = strtoul(argv[3], 0, 10);
		if (argc > 4)
			addr = strtoul(argv[4], 0, 16);
	}

	for (nr = 0; nr < count;) {
		nr++;
		if (!(z80_bus_cmd(Request) & ZST_ACQUIRED)) {
			my_puts_P(PSTR("Bus timeout\n"));
			return  CMD_RET_FAILURE;
		}
		z80_read_block(buffer, addr /*+ base*/, _MAX_SS);
		z80_bus_cmd(Release);

		res = disk_write(dev, buffer, sec, 1);
		if (res != RES_OK)
			break;
		sec++; addr += _MAX_SS;
	}

	printf_P(PSTR("%d sector(s) written, rc=%.2x.\n"), nr, res);

	sec_last = sec;
	addr_last = addr;

	return res ? CMD_RET_FAILURE : CMD_RET_SUCCESS;
}


/*
 * Disk ioctl
 * dcs <pd#> - CTRL_SYNC
 *
 */
static
command_ret_t do_ioctl_sync(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	BYTE dev;

	(void) cmdtp; (void) flag;

	if (argc < 2)
		return CMD_RET_USAGE;

	dev = (BYTE) strtoul(argv[1], 0, 10);
	printf_P(PSTR("rc=%.2x\n"), disk_ioctl(dev, CTRL_SYNC, 0));

	return CMD_RET_SUCCESS;
}


static
command_ret_t do_help(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]);

cmd_tbl_t cmd_sd_sub[] = {
CMD_TBL_ITEM(
	init,	2,			1,	do_init,
	"Initialize disk",
	""
),
CMD_TBL_ITEM(
	status,	2,			1,	do_status,
	"Disk status",
	""
),
CMD_TBL_ITEM(
	dump,	CONFIG_SYS_MAXARGS,	1,	do_dump,
	"Dump sector",
	""
),
CMD_TBL_ITEM(
	read,	2,			1,	do_read,
	"read disk sector(s) into meomory",
	"drive [sector [count [memaddr]]]"
),
CMD_TBL_ITEM(
	write,	2,			1,	do_write,
	"write sector(s) from meomory to disk",
	"drive [sector [count [memaddr]]]"
),
CMD_TBL_ITEM(
	sync,	2,			1,	do_ioctl_sync,
	"Device control: SYNC",
	""
),

CMD_TBL_ITEM(
	help,	CONFIG_SYS_MAXARGS,	1,	do_help,
	"print sub command description/usage",
	"\n"
	"       - print brief description of all sub commands\n"
	"sd help command ...\n"
	"       - print detailed usage of sub cmd 'command'"
),

/* This does not use the CMD_TBL_ITEM macro as ? can't be used in symbol names */
	{FSTR("?"),   CONFIG_SYS_MAXARGS, 1, do_help,
	 FSTR("alias for 'help'"),
#ifdef  CONFIG_SYS_LONGHELP
	FSTR(""),
#endif /* CONFIG_SYS_LONGHELP */
#ifdef CONFIG_AUTO_COMPLETE
	0,
#endif
},
};

static
command_ret_t do_help(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	return _do_help(cmd_sd_sub, ARRAY_SIZE(cmd_sd_sub), cmdtp, flag, argc, argv);
}


command_ret_t do_sd(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	cmd_tbl_t *cp;

	if (argc < 2)
		return CMD_RET_USAGE;

	/* drop initial "sd" arg */
	argc--;
	argv++;

	cp = find_cmd_tbl(argv[0], cmd_sd_sub, ARRAY_SIZE(cmd_sd_sub));

	if (cp)
		return cp->cmd(cmdtp, flag, argc, argv);

	return CMD_RET_USAGE;
}
Commit	Line	Data
7f552300 L	1	#include "common.h"
	2	#include <stdlib.h>
	3
a882a089	4	#include "command.h"
7f552300 L	5	#include "diskio.h"
7f552300 L	6	#include "ff.h"
a882a089	7	#include "z80-if.h"
7f552300 L	8	#include "print-utils.h"
	9
	10
7f552300	11	/*
a882a089	12	* di <pd#> - Initialize disk
7f552300 L	13	*
7f552300 L	14	*/
a882a089 L	15	static
a882a089 L	16	command_ret_t do_init(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
7f552300	17	{
a882a089	18	DSTATUS res;
7f552300	19	BYTE dev;
7f552300 L	20
	21	(void) cmdtp; (void) flag;
	22
	23	if (argc < 2)
	24	return CMD_RET_USAGE;
	25
	26	dev = (BYTE) strtoul(argv[1], 0, 10);
a882a089 L	27	res = disk_status(dev);
a882a089 L	28	printf_P(PSTR("disk_status=%.2x\n"), res);
7f552300	29
a882a089 L	30	if ((res & STA_NODISK) == 0) {
	31	res = disk_initialize(dev);
	32	}
	33	printf_P(PSTR("rc=%.2x\n"), res);
7f552300	34	if (res) {
7f552300 L	35	return CMD_RET_FAILURE;
	36	}
	37
7f552300 L	38	return CMD_RET_SUCCESS;
	39	}
	40
	41	/*
a882a089	42	* ds <pd#> - Show disk status
7f552300 L	43	*
7f552300 L	44	*/
a882a089 L	45	static
a882a089 L	46	command_ret_t do_status(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
7f552300	47	{
7f552300 L	48	BYTE dev;
7f552300 L	49
a882a089 L	50	union {
	51	unsigned char uca[64];
	52	unsigned long ul;
	53	unsigned char uc;
	54	} dat;
	55
7f552300 L	56	(void) cmdtp; (void) flag;
	57
	58	if (argc < 2)
	59	return CMD_RET_USAGE;
	60
	61	dev = (BYTE) strtoul(argv[1], 0, 10);
7f552300	62
a882a089 L	63
	64	if (disk_ioctl(dev, GET_SECTOR_COUNT, &dat.ul) == RES_OK)
	65	printf_P(PSTR("Drive size: %lu sectors\n"), dat.ul);
a882a089 L	66	if (disk_ioctl(dev, GET_BLOCK_SIZE, &dat.ul) == RES_OK)
a882a089 L	67	printf_P(PSTR("Erase block: %lu sectors\n"), dat.ul);
a882a089 L	68	if (disk_ioctl(dev, MMC_GET_TYPE, &dat.uc) == RES_OK)
a882a089 L	69	printf_P(PSTR("Card type: %u\n"), dat.uc);
a882a089	70	if (disk_ioctl(dev, MMC_GET_CSD, dat.uca) == RES_OK)
cca42593	71	dump_ram(dat.uca, 0, 16, "CSD:");
a882a089	72	if (disk_ioctl(dev, MMC_GET_CID, dat.uca) == RES_OK)
cca42593	73	dump_ram(dat.uca, 0, 16, "CID:");
a882a089	74	if (disk_ioctl(dev, MMC_GET_OCR, dat.uca) == RES_OK)
cca42593	75	dump_ram(dat.uca, 0, 4, "OCR:");
a882a089	76	if (disk_ioctl(dev, MMC_GET_SDSTAT, dat.uca) == RES_OK)
cca42593	77	dump_ram(dat.uca, 0, 64, "SD Status:");
a882a089	78
7f552300 L	79	return CMD_RET_SUCCESS;
	80	}
	81
a882a089	82
7f552300	83	/*
a882a089	84	* dump <pd#> [<sector> [count]] - Dump sector
7f552300 L	85	*
7f552300 L	86	*/
a882a089 L	87	static
a882a089 L	88	command_ret_t do_dump(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
7f552300	89	{
a882a089 L	90	static BYTE dev_last;
	91	static DWORD sec_last;
	92	BYTE buffer[_MAX_SS];
	93	char header[20];
7f552300	94
a882a089 L	95	DRESULT res;
	96	BYTE dev;
	97	DWORD sec;
	98	UINT count;
7f552300 L	99
	100	(void) cmdtp; (void) flag;
	101
	102	if (argc < 2)
	103	return CMD_RET_USAGE;
	104
a882a089 L	105	dev = (BYTE) strtoul(argv[1], NULL, 10);
	106	if (dev != dev_last)
	107	sec_last = 0;
	108	sec = sec_last;
	109	count = 1;
	110
	111	if ((flag & CMD_FLAG_REPEAT) == 0) {
	112	/* If another parameter, it is the sector to dump. */
	113	if (argc > 2)
	114	sec = strtoul(argv[2], 0, 10);
	115	if (argc > 3)
	116	count = (UINT) strtoul(argv[3], 0, 10);
	117	}
7f552300	118
a882a089 L	119	for ( ; count; count--, sec++) {
a882a089 L	120	res = disk_read(dev, buffer, sec, 1);
7f552300	121
a882a089 L	122	if (res) {
	123	printf_P(PSTR("rc=%.2x\n"), res);
	124	return CMD_RET_FAILURE;
	125	}
7f552300	126
a882a089	127	sprintf_P(header, PSTR("Sector: %lu"), sec);
cca42593	128	dump_ram(buffer, 0, _MAX_SS, header);
a882a089 L	129	}
	130	dev_last = dev;
	131	sec_last = sec;
7f552300	132
a882a089 L	133	return CMD_RET_SUCCESS;
a882a089 L	134	}
7f552300	135
a882a089 L	136	/*
	137	* read drive sector count memaddr - Read disk into memory
	138	*
	139	*/
	140	static
	141	command_ret_t do_read(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
	142	{
	143	DRESULT res;
	144	BYTE dev;
	145	DWORD sec;
	146	uint32_t addr;
	147	int count, nr;
	148	BYTE buffer[_MAX_SS];
7f552300	149
a882a089 L	150	static DWORD sec_last;
a882a089 L	151	static uint32_t addr_last;
7f552300	152
a882a089	153	(void) cmdtp;
7f552300	154
a882a089 L	155	if (argc < 2)
a882a089 L	156	return CMD_RET_USAGE;
7f552300	157
a882a089 L	158	dev = (BYTE) strtoul(argv[1], NULL, 10);
	159	sec = sec_last;
	160	count = 1;
	161	addr = addr_last;
	162
	163	if ((flag & CMD_FLAG_REPEAT) == 0) {
	164	/* If another parameter, it is the sector to dump. */
	165	if (argc > 2)
	166	sec = strtoul(argv[2], 0, 10);
	167	if (argc > 3)
	168	count = strtoul(argv[3], 0, 10);
	169	if (argc > 4)
	170	addr = strtoul(argv[4], 0, 16);
7f552300	171	}
a882a089 L	172
	173	for (nr = 0; nr < count;) {
	174	nr++;
	175	if ((res = disk_read(dev, buffer, sec, 1)) == RES_OK) {
	176	if (!(z80_bus_cmd(Request) & ZST_ACQUIRED)) {
	177	my_puts_P(PSTR("Bus timeout\n"));
	178	return CMD_RET_FAILURE;
	179	}
	180	z80_write_block(buffer, addr /+ base/, _MAX_SS);
	181	z80_bus_cmd(Release);
	182	sec++; addr += _MAX_SS;
	183	} else
	184	break;
7f552300 L	185	}
7f552300 L	186
a882a089 L	187	printf_P(PSTR("Read %d sector(s), rc=%.2x.\n"), nr, res);
	188	if (res)
	189	printf_P(PSTR("Last sector not written!\n"));
	190
	191	sec_last = sec;
	192	addr_last = addr;
7f552300	193
a882a089 L	194	return res ? CMD_RET_FAILURE : CMD_RET_SUCCESS;
a882a089 L	195	}
7f552300 L	196
	197
	198	/*
a882a089	199	* write <pd#> <sector> <memaddr> [<n>] - Write memory to disk
7f552300 L	200	*
7f552300 L	201	*/
a882a089 L	202	static
a882a089 L	203	command_ret_t do_write(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
7f552300	204	{
a882a089	205	DRESULT res;
7f552300	206	BYTE dev;
a882a089 L	207	DWORD sec;
	208	uint32_t addr;
	209	int count, nr;
	210	BYTE buffer[_MAX_SS];
7f552300	211
a882a089 L	212	static DWORD sec_last;
	213	static uint32_t addr_last;
	214
	215	(void) cmdtp;
7f552300 L	216
	217	if (argc < 2)
	218	return CMD_RET_USAGE;
	219
a882a089 L	220	dev = (BYTE) strtoul(argv[1], NULL, 10);
	221	sec = sec_last;
	222	addr = addr_last;
	223	count = 1;
	224
	225	if ((flag & CMD_FLAG_REPEAT) == 0) {
	226	/* If another parameter, it is the sector to dump. */
	227	if (argc > 2)
	228	sec = strtoul(argv[2], 0, 10);
	229	if (argc > 3)
	230	count = strtoul(argv[3], 0, 10);
	231	if (argc > 4)
	232	addr = strtoul(argv[4], 0, 16);
	233	}
7f552300	234
a882a089 L	235	for (nr = 0; nr < count;) {
	236	nr++;
	237	if (!(z80_bus_cmd(Request) & ZST_ACQUIRED)) {
	238	my_puts_P(PSTR("Bus timeout\n"));
	239	return CMD_RET_FAILURE;
	240	}
	241	z80_read_block(buffer, addr /+ base/, _MAX_SS);
	242	z80_bus_cmd(Release);
	243
	244	res = disk_write(dev, buffer, sec, 1);
	245	if (res != RES_OK)
	246	break;
	247	sec++; addr += _MAX_SS;
	248	}
	249
	250	printf_P(PSTR("%d sector(s) written, rc=%.2x.\n"), nr, res);
	251
	252	sec_last = sec;
	253	addr_last = addr;
	254
	255	return res ? CMD_RET_FAILURE : CMD_RET_SUCCESS;
7f552300 L	256	}
	257
	258
	259
a882a089 L	260	/*
	261	* Disk ioctl
	262	* dcs <pd#> - CTRL_SYNC
	263	*
	264	*/
	265	static
	266	command_ret_t do_ioctl_sync(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
7f552300	267	{
a882a089	268	BYTE dev;
7f552300	269
a882a089	270	(void) cmdtp; (void) flag;
7f552300	271
a882a089 L	272	if (argc < 2)
a882a089 L	273	return CMD_RET_USAGE;
7f552300	274
a882a089 L	275	dev = (BYTE) strtoul(argv[1], 0, 10);
a882a089 L	276	printf_P(PSTR("rc=%.2x\n"), disk_ioctl(dev, CTRL_SYNC, 0));
7f552300 L	277
	278	return CMD_RET_SUCCESS;
	279	}
	280
7f552300 L	281
	282	static
	283	command_ret_t do_help(cmd_tbl_t cmdtp, int flag, int argc, char const argv[]);
	284
	285	cmd_tbl_t cmd_sd_sub[] = {
	286	CMD_TBL_ITEM(
a882a089	287	init, 2, 1, do_init,
7f552300 L	288	"Initialize disk",
	289	""
	290	),
	291	CMD_TBL_ITEM(
a882a089	292	status, 2, 1, do_status,
7f552300 L	293	"Disk status",
	294	""
	295	),
	296	CMD_TBL_ITEM(
a882a089 L	297	dump, CONFIG_SYS_MAXARGS, 1, do_dump,
a882a089 L	298	"Dump sector",
7f552300 L	299	""
	300	),
	301	CMD_TBL_ITEM(
a882a089 L	302	read, 2, 1, do_read,
	303	"read disk sector(s) into meomory",
	304	"drive [sector [count [memaddr]]]"
	305	),
	306	CMD_TBL_ITEM(
	307	write, 2, 1, do_write,
	308	"write sector(s) from meomory to disk",
	309	"drive [sector [count [memaddr]]]"
7f552300 L	310	),
7f552300 L	311	CMD_TBL_ITEM(
a882a089 L	312	sync, 2, 1, do_ioctl_sync,
	313	"Device control: SYNC",
	314	""
7f552300	315	),
a882a089	316
7f552300 L	317	CMD_TBL_ITEM(
	318	help, CONFIG_SYS_MAXARGS, 1, do_help,
	319	"print sub command description/usage",
	320	"\n"
	321	" - print brief description of all sub commands\n"
	322	"sd help command ...\n"
	323	" - print detailed usage of sub cmd 'command'"
	324	),
	325
	326	/* This does not use the CMD_TBL_ITEM macro as ? can't be used in symbol names */
	327	{FSTR("?"), CONFIG_SYS_MAXARGS, 1, do_help,
	328	FSTR("alias for 'help'"),
	329	#ifdef CONFIG_SYS_LONGHELP
	330	FSTR(""),
	331	#endif /* CONFIG_SYS_LONGHELP */
	332	#ifdef CONFIG_AUTO_COMPLETE
	333	0,
	334	#endif
	335	},
	336	};
	337
	338	static
	339	command_ret_t do_help(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
	340	{
	341	return _do_help(cmd_sd_sub, ARRAY_SIZE(cmd_sd_sub), cmdtp, flag, argc, argv);
	342	}
	343
	344
	345	command_ret_t do_sd(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
	346	{
	347	cmd_tbl_t *cp;
	348
	349	if (argc < 2)
	350	return CMD_RET_USAGE;
	351
	352	/* drop initial "sd" arg */
	353	argc--;
	354	argv++;
	355
	356	cp = find_cmd_tbl(argv[0], cmd_sd_sub, ARRAY_SIZE(cmd_sd_sub));
	357
	358	if (cp)
	359	return cp->cmd(cmdtp, flag, argc, argv);
	360
	361	return CMD_RET_USAGE;
	362	}