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

/*
 * (C) Copyright 2016 Leo C. <erbl259-lmu@yahoo.de>
 *
 * SPDX-License-Identifier:	GPL-2.0
 */

/*
 * attach channels to devices
 */

#include "cmd_attach.h"

#include "command.h"
#include "z180-serv.h"
#include "getopt-min.h"


static const FLASH char * const FLASH rc_messages[] = {
			FSTR("OK"),
			FSTR("Unknown error"),
			FSTR("Disk number out of range 0..7"),
			FSTR("Disk allready attached"),
			FSTR("Disk not attached"),
			FSTR("File not found"),
			FSTR("Not enough memory"),
			FSTR("Error opening file"),
			FSTR("File allready attached to other drive"),
		};

static
void printerror(int rc, uint8_t unit, char *fn)
{
	if (rc < 0 || (unsigned) rc >= ARRAY_SIZE(rc_messages))
		rc = 1;

#if GCC_BUG_61443
		printf_P(PSTR("Attachment of '%s' to dsk%d failed: "), fn, unit);
		my_puts_P(rc_messages[rc]);
		my_puts_P(PSTR("!\n"));
#else
		printf_P(PSTR("Attachment of '%s' to dsk%d failed: %S!\n"),
				fn, unit, rc_messages[rc]);
#endif
}

/*
 * attach [[options] [unit [diskfile]]]
 *
 * detach unit
 * attach -d unit
 *
 * attach -o reattach unit
 * attach -o reattach unit diskfile
 *
 * attach unit diskfile
 *
 */

command_ret_t do_attach(cmd_tbl_t *cmdtp, uint_fast8_t flag, int argc, char * const argv[])
{
	uint8_t unit;
	char *filename = NULL;
	bool detach = false;
	bool detach_all = false;
	drv_opt_t options = 0;
	int res;

	(void) cmdtp; (void) flag;


	if (argv[0][0] == 'd') {
		/* we are called as 'detach' */
		detach = true;
	} else if (argc == 1) {
		/* no arguments */
		drv_list();
		return CMD_RET_SUCCESS;
	}

	/* reset getopt() */
	optind = 0;

	int opt;
	while ((opt = getopt(argc, argv, PSTR("darwo:"))) != -1) {
		switch (opt) {
		case 'd':
			detach = true;
			break;
		case 'a':
			detach_all = true;
			break;
		case 'r':
			options |= DRV_OPT_RO;
			break;
		case 'w':
			options &= ~DRV_OPT_RO;
			break;
		case 'o':
			{
				static const FLASH char delim[] = {", "};
				char *p = strtok_P(optarg, delim);
				while (p != NULL) {
					if (!strcmp_P(p, PSTR("ro")))
						options |= DRV_OPT_RO;
					else if (!strcmp_P(p, PSTR("rw")))
						options &= ~DRV_OPT_RO;
					else if (!strcmp_P(p, PSTR("debug")))
						options |= DRV_OPT_DEBUG;
					else if (!strcmp_P(p, PSTR("nodebug")))
						options &= ~DRV_OPT_DEBUG;
					else if (!strcmp_P(p, PSTR("reattach")))
						options |= DRV_OPT_REATTATCH;
					else
						return CMD_RET_USAGE;

					p = strtok_P(NULL, delim);
				}
			}
			break;
		default: /* '?' */
			return CMD_RET_USAGE;
		}
	}

	/* remaining arguments */
	argc -= optind;
	if ( !( (argc == 0 && detach && detach_all) ||
			(argc == 1 && detach) ||
			(argc == 1 && (options & DRV_OPT_REATTATCH)) ||
			 argc == 2) )
		return CMD_RET_USAGE;

	if (argc > 0 && ((strlen(argv[optind]) != 4) ||
			strncmp_P(argv[optind], PSTR("dsk"), 3) ||
			(unit = argv[optind][3] - '0') >= CONFIG_CPM_MAX_DRIVE)) {

		printf_P(PSTR("Unknown device: '%s'\n"), argv[optind]);
		return CMD_RET_FAILURE;
	}

	if (detach) {
		if (detach_all)
			for (uint8_t i = 0; i < CONFIG_CPM_MAX_DRIVE; i++)
				drv_detach(i);
		else
			drv_detach(unit);
		return CMD_RET_SUCCESS;
	}

	if (argc == 2)
		filename = argv[++optind];

	res = drv_attach(unit, filename, options);
	if (res)
		printerror(res, unit, filename);

	return res ? CMD_RET_FAILURE : CMD_RET_SUCCESS;
}
Commit	Line	Data
cb4fb1ed L	1	/*
	2	* (C) Copyright 2016 Leo C. <erbl259-lmu@yahoo.de>
	3	*
	4	* SPDX-License-Identifier: GPL-2.0
	5	*/
	6
	7	/*
4122fe90	8	* attach channels to devices
cb4fb1ed L	9	*/
cb4fb1ed L	10
228b1c5f	11	#include "cmd_attach.h"
cb4fb1ed L	12
	13	#include "command.h"
	14	#include "z180-serv.h"
	15	#include "getopt-min.h"
cb4fb1ed L	16
	17
	18	static const FLASH char * const FLASH rc_messages[] = {
	19	FSTR("OK"),
	20	FSTR("Unknown error"),
	21	FSTR("Disk number out of range 0..7"),
	22	FSTR("Disk allready attached"),
	23	FSTR("Disk not attached"),
	24	FSTR("File not found"),
	25	FSTR("Not enough memory"),
	26	FSTR("Error opening file"),
	27	FSTR("File allready attached to other drive"),
	28	};
	29
	30	static
	31	void printerror(int rc, uint8_t unit, char *fn)
	32	{
	33	if (rc < 0 \|\| (unsigned) rc >= ARRAY_SIZE(rc_messages))
	34	rc = 1;
	35
3f88ef88 L	36	#if GCC_BUG_61443
	37	printf_P(PSTR("Attachment of '%s' to dsk%d failed: "), fn, unit);
	38	my_puts_P(rc_messages[rc]);
	39	my_puts_P(PSTR("!\n"));
cb4fb1ed L	40	#else
	41	printf_P(PSTR("Attachment of '%s' to dsk%d failed: %S!\n"),
	42	fn, unit, rc_messages[rc]);
	43	#endif
	44	}
	45
	46	/*
	47	* attach [[options] [unit [diskfile]]]
	48	*
	49	* detach unit
	50	* attach -d unit
	51	*
	52	* attach -o reattach unit
	53	* attach -o reattach unit diskfile
	54	*
	55	* attach unit diskfile
	56	*
	57	*/
	58
93ea25f2	59	command_ret_t do_attach(cmd_tbl_t cmdtp, uint_fast8_t flag, int argc, char const argv[])
cb4fb1ed L	60	{
	61	uint8_t unit;
	62	char *filename = NULL;
	63	bool detach = false;
4122fe90	64	bool detach_all = false;
cb4fb1ed L	65	drv_opt_t options = 0;
	66	int res;
	67
	68	(void) cmdtp; (void) flag;
	69
	70
4122fe90	71	if (argv[0][0] == 'd') {
cb4fb1ed L	72	/* we are called as 'detach' */
cb4fb1ed L	73	detach = true;
4122fe90	74	} else if (argc == 1) {
cb4fb1ed L	75	/* no arguments */
	76	drv_list();
	77	return CMD_RET_SUCCESS;
	78	}
	79
	80	/* reset getopt() */
beafa6d6	81	optind = 0;
cb4fb1ed L	82
cb4fb1ed L	83	int opt;
4122fe90	84	while ((opt = getopt(argc, argv, PSTR("darwo:"))) != -1) {
cb4fb1ed L	85	switch (opt) {
	86	case 'd':
	87	detach = true;
	88	break;
4122fe90 L	89	case 'a':
	90	detach_all = true;
	91	break;
cb4fb1ed L	92	case 'r':
	93	options \|= DRV_OPT_RO;
	94	break;
	95	case 'w':
	96	options &= ~DRV_OPT_RO;
	97	break;
	98	case 'o':
	99	{
	100	static const FLASH char delim[] = {", "};
	101	char *p = strtok_P(optarg, delim);
	102	while (p != NULL) {
	103	if (!strcmp_P(p, PSTR("ro")))
	104	options \|= DRV_OPT_RO;
	105	else if (!strcmp_P(p, PSTR("rw")))
	106	options &= ~DRV_OPT_RO;
	107	else if (!strcmp_P(p, PSTR("debug")))
	108	options \|= DRV_OPT_DEBUG;
	109	else if (!strcmp_P(p, PSTR("nodebug")))
	110	options &= ~DRV_OPT_DEBUG;
	111	else if (!strcmp_P(p, PSTR("reattach")))
	112	options \|= DRV_OPT_REATTATCH;
	113	else
	114	return CMD_RET_USAGE;
	115
	116	p = strtok_P(NULL, delim);
	117	}
	118	}
	119	break;
	120	default: /* '?' */
	121	return CMD_RET_USAGE;
	122	}
	123	}
	124
	125	/* remaining arguments */
	126	argc -= optind;
4122fe90 L	127	if ( !( (argc == 0 && detach && detach_all) \|\|
	128	(argc == 1 && detach) \|\|
	129	(argc == 1 && (options & DRV_OPT_REATTATCH)) \|\|
	130	argc == 2) )
cb4fb1ed L	131	return CMD_RET_USAGE;
cb4fb1ed L	132
4122fe90	133	if (argc > 0 && ((strlen(argv[optind]) != 4) \|\|
cb4fb1ed	134	strncmp_P(argv[optind], PSTR("dsk"), 3) \|\|
4122fe90	135	(unit = argv[optind][3] - '0') >= CONFIG_CPM_MAX_DRIVE)) {
cb4fb1ed L	136
	137	printf_P(PSTR("Unknown device: '%s'\n"), argv[optind]);
	138	return CMD_RET_FAILURE;
	139	}
	140
	141	if (detach) {
4122fe90 L	142	if (detach_all)
	143	for (uint8_t i = 0; i < CONFIG_CPM_MAX_DRIVE; i++)
	144	drv_detach(i);
	145	else
	146	drv_detach(unit);
cb4fb1ed L	147	return CMD_RET_SUCCESS;
	148	}
	149
	150	if (argc == 2)
	151	filename = argv[++optind];
	152
	153	res = drv_attach(unit, filename, options);
	154	if (res)
	155	printerror(res, unit, filename);
	156
	157	return res ? CMD_RET_FAILURE : CMD_RET_SUCCESS;
	158	}