[unix-history] / usr / src / bin / dd / args.c

/*-
 * Copyright (c) 1991 The Regents of the University of California.
 * All rights reserved.
 *
 * This code is derived from software contributed to Berkeley by
 * Keith Muller of the University of California, San Diego and Lance
 * Visser of Convex Computer Corporation.
 *
 * %sccs.include.redist.c%
 */

#ifndef lint
static char sccsid[] = "@(#)args.c	5.5 (Berkeley) %G%";
#endif /* not lint */

#include <sys/types.h>
#include <limits.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "dd.h"
#include "extern.h"

static u_long get_bsz __P((char *));

static void f_bs __P((char *));
static void f_cbs __P((char *));
static void f_conv __P((char *));
static void f_count __P((char *));
static void f_files __P((char *));
static void f_ibs __P((char *));
static void f_if __P((char *));
static void f_obs __P((char *));
static void f_of __P((char *));
static void f_seek __P((char *));
static void f_skip __P((char *));

static struct arg {
	char *name;
	void (*f) __P((char *));
	u_int set, noset;
} args[] = {
	"bs",		f_bs,		C_BS,		C_BS|C_IBS|C_OBS,
	"cbs",		f_cbs,		C_CBS,		C_CBS,
	"conv",		f_conv,		0,		0,
	"count",	f_count,	C_COUNT,	C_COUNT,
	"files",	f_files,	C_FILES,	C_FILES,
	"ibs",		f_ibs,		C_IBS,		C_BS|C_IBS,
	"if",		f_if,		C_IF,		C_IF,
	"obs",		f_obs,		C_OBS,		C_BS|C_OBS,
	"of",		f_of,		C_OF,		C_OF,
	"seek",		f_seek,		C_SEEK,		C_SEEK,
	"skip",		f_skip,		C_SKIP,		C_SKIP,
};

static char *oper;

/*
 * args -- parse JCL syntax of dd.
 */
void
jcl(argv)
	register char **argv;
{
	register struct arg *ap;
	struct arg tmp;
	char *arg;
	static int c_arg __P((const void *, const void *));

	in.dbsz = out.dbsz = 512;

	while (oper = *++argv) {
		if ((arg = index(oper, '=')) == NULL)
			err("unknown operand %s", oper);
		*arg++ = '\0';
		if (!*arg)
			err("no value specified for %s", oper);
		tmp.name = oper;
		if (!(ap = (struct arg *)bsearch(&tmp, args,
		    sizeof(args)/sizeof(struct arg), sizeof(struct arg),
		    c_arg)))
			err("unknown operand %s", tmp.name);
		if (ddflags & ap->noset)
			err("%s: illegal argument combination or already set",
			    tmp.name);
		ddflags |= ap->set;
		ap->f(arg);
	}

	/* Final sanity checks. */

	if (ddflags & C_BS) {
		/*
		 * Bs is turned off by any conversion -- we assume the user
		 * just wanted to set both the input and output block sizes
		 * and didn't want the bs semantics, so we don't warn.
		 */
		if (ddflags & (C_BLOCK|C_LCASE|C_SWAB|C_UCASE|C_UNBLOCK))
			ddflags &= ~C_BS;

		/* Bs supersedes ibs and obs. */
		if (ddflags & C_BS && ddflags & (C_IBS|C_OBS))
			warn("bs supersedes ibs and obs");
	}

	/*
	 * Ascii/ebcdic and cbs implies block/unblock.
	 * Block/unblock requires cbs and vice-versa.
	 */
	if (ddflags & (C_BLOCK|C_UNBLOCK)) {
		if (!(ddflags & C_CBS))
			err("record operations require cbs");
		if (cbsz == 0)
			err("cbs cannot be zero");
		cfunc = ddflags & C_BLOCK ? block : unblock;
	} else if (ddflags & C_CBS) {
		if (ddflags & (C_ASCII|C_EBCDIC)) {
			if (ddflags & C_ASCII) {
				ddflags |= C_UNBLOCK;
				cfunc = unblock;
			} else {
				ddflags |= C_BLOCK;
				cfunc = block;
			}
		} else
			err("cbs meaningless if not doing record operations");
		if (cbsz == 0)
			err("cbs cannot be zero");
	} else
		cfunc = def;

	if (in.dbsz == 0 || out.dbsz == 0)
		err("buffer sizes cannot be zero");

	/*
	 * Read, write and seek calls take ints as arguments.  Seek sizes
	 * could be larger if we wanted to do it in stages or check only
	 * regular files, but it's probably not worth it.
	 */
	if (in.dbsz > INT_MAX || out.dbsz > INT_MAX)
		err("buffer sizes cannot be greater than %d", INT_MAX);
	if (in.offset > INT_MAX / in.dbsz || out.offset > INT_MAX / out.dbsz)
		err("seek offsets cannot be larger than %d", INT_MAX);
}

static int
c_arg(a, b)
	const void *a, *b;
{
	return (strcmp(((struct arg *)a)->name, ((struct arg *)b)->name));
}

static void
f_bs(arg)
	char *arg;
{
	in.dbsz = out.dbsz = (int)get_bsz(arg);
}

static void
f_cbs(arg)
	char *arg;
{
	cbsz = (int)get_bsz(arg);
}

static void
f_count(arg)
	char *arg;
{
	cpy_cnt = (u_int)get_bsz(arg);
	if (!cpy_cnt)
		terminate(0);
}

static void
f_files(arg)
	char *arg;
{
	files_cnt = (int)get_bsz(arg);
}

static void
f_ibs(arg)
	char *arg;
{
	if (!(ddflags & C_BS))
		in.dbsz = (int)get_bsz(arg);
}

static void
f_if(arg)
	char *arg;
{
	in.name = arg;
}

static void
f_obs(arg)
	char *arg;
{
	if (!(ddflags & C_BS))
		out.dbsz = (int)get_bsz(arg);
}

static void
f_of(arg)
	char *arg;
{
	out.name = arg;
}

static void
f_seek(arg)
	char *arg;
{
	out.offset = (u_int)get_bsz(arg);
}

static void
f_skip(arg)
	char *arg;
{
	in.offset = (u_int)get_bsz(arg);
}

static struct conv {
	char *name;
	u_int set, noset;
	u_char *ctab;
} clist[] = {
	"ascii",	C_ASCII,	C_EBCDIC,	e2a_POSIX,
	"block",	C_BLOCK,	C_UNBLOCK,	NULL,
	"ebcdic",	C_EBCDIC,	C_ASCII,	a2e_POSIX,
	"ibm",		C_EBCDIC,	C_ASCII,	a2ibm_POSIX,
	"lcase",	C_LCASE,	C_UCASE,	NULL,
	"noerror",	C_NOERROR,	0,		NULL,
	"notrunc",	C_NOTRUNC,	0,		NULL,
	"oldascii",	C_ASCII,	C_EBCDIC,	e2a_32V,
	"oldebcdic",	C_EBCDIC,	C_ASCII,	a2e_32V,
	"oldibm",	C_EBCDIC,	C_ASCII,	a2ibm_32V,
	"swab",		C_SWAB,		0,		NULL,
	"sync",		C_SYNC,		0,		NULL,
	"ucase",	C_UCASE,	C_LCASE,	NULL,
	"unblock",	C_UNBLOCK,	C_BLOCK,	NULL,
};

static void
f_conv(arg)
	char *arg;
{
	register struct conv *cp;
	struct conv tmp;
	static int c_conv __P((const void *, const void *));

	while (arg != NULL) {
		tmp.name = strsep(&arg, ",");
		if (!(cp = (struct conv *)bsearch(&tmp, clist,
		    sizeof(clist)/sizeof(struct conv), sizeof(struct conv),
		    c_conv)))
			err("unknown conversion %s", tmp.name);
		if (ddflags & cp->noset)
			err("%s: illegal conversion combination", tmp.name);
		ddflags |= cp->set;
		if (cp->ctab)
			ctab = cp->ctab;
	}
}

static int
c_conv(a, b)
	const void *a, *b;
{
	return (strcmp(((struct conv *)a)->name, ((struct conv *)b)->name));
}

/*
 * Convert an expression of the following forms to an unsigned long.
 * 	1) A positive decimal number.
 *	2) A positive decimal number followed by a b (mult by 512).
 *	3) A positive decimal number followed by a k (mult by 1024).
 *	4) A positive decimal number followed by a m (mult by 512).
 *	5) A positive decimal number followed by a w (mult by sizeof int)
 *	6) Two or more positive decimal numbers (with/without k,b or w).
 *	   seperated by x (also * for backwards compatibility), specifying
 *	   the product of the indicated values.
 */
static u_long
get_bsz(val)
	char *val;
{
	char *expr;
	u_long num, t;

	num = strtoul(val, &expr, 0);
	if (num == ULONG_MAX)			/* Overflow. */
		err("%s: %s", oper, strerror(errno));
	if (expr == val)			/* No digits. */
		err("%s: illegal numeric value", oper);

	switch(*expr) {
	case 'b':
		t = num;
		num *= 512;
		if (t > num)
			goto erange;
		++expr;
		break;
	case 'k':
		t = num;
		num *= 1024;
		if (t > num)
			goto erange;
		++expr;
		break;
	case 'm':
		t = num;
		num *= 1048576;
		if (t > num)
			goto erange;
		++expr;
		break;
	case 'w':
		t = num;
		num *= sizeof(int);
		if (t > num)
			goto erange;
		++expr;
		break;
	}

	switch(*expr) {
		case '\0':
			break;
		case '*':			/* Backward compatible. */
		case 'x':
			t = num;
			num *= get_bsz(expr + 1);
			if (t > num)
erange:				err("%s: %s", oper, strerror(ERANGE));
			break;
		default:
			err("%s: illegal numeric value", oper);
	}
	return(num);
}
Commit	Line	Data
ac894d30 KB	1	/*-
	2	* Copyright (c) 1991 The Regents of the University of California.
	3	* All rights reserved.
	4	*
	5	* This code is derived from software contributed to Berkeley by
a08a4cd0 KB	6	* Keith Muller of the University of California, San Diego and Lance
a08a4cd0 KB	7	* Visser of Convex Computer Corporation.
ac894d30 KB	8	*
	9	* %sccs.include.redist.c%
	10	*/
	11
	12	#ifndef lint
acb87732	13	static char sccsid[] = "@(#)args.c 5.5 (Berkeley) %G%";
ac894d30 KB	14	#endif /* not lint */
	15
	16	#include <sys/types.h>
	17	#include <limits.h>
	18	#include <errno.h>
	19	#include <stdio.h>
	20	#include <stdlib.h>
	21	#include <string.h>
	22	#include "dd.h"
	23	#include "extern.h"
	24
	25	static u_long get_bsz __P((char *));
	26
	27	static void f_bs __P((char *));
	28	static void f_cbs __P((char *));
	29	static void f_conv __P((char *));
	30	static void f_count __P((char *));
	31	static void f_files __P((char *));
	32	static void f_ibs __P((char *));
	33	static void f_if __P((char *));
	34	static void f_obs __P((char *));
	35	static void f_of __P((char *));
	36	static void f_seek __P((char *));
	37	static void f_skip __P((char *));
	38
	39	static struct arg {
	40	char *name;
	41	void (f) __P((char ));
	42	u_int set, noset;
	43	} args[] = {
	44	"bs", f_bs, C_BS, C_BS\|C_IBS\|C_OBS,
998e1ac9	45	"cbs", f_cbs, C_CBS, C_CBS,
ac894d30 KB	46	"conv", f_conv, 0, 0,
	47	"count", f_count, C_COUNT, C_COUNT,
	48	"files", f_files, C_FILES, C_FILES,
	49	"ibs", f_ibs, C_IBS, C_BS\|C_IBS,
	50	"if", f_if, C_IF, C_IF,
	51	"obs", f_obs, C_OBS, C_BS\|C_OBS,
	52	"of", f_of, C_OF, C_OF,
	53	"seek", f_seek, C_SEEK, C_SEEK,
	54	"skip", f_skip, C_SKIP, C_SKIP,
	55	};
	56
	57	static char *oper;
	58
	59	/*
998e1ac9	60	* args -- parse JCL syntax of dd.
ac894d30 KB	61	*/
	62	void
	63	jcl(argv)
	64	register char **argv;
	65	{
	66	register struct arg *ap;
	67	struct arg tmp;
	68	char *arg;
	69	static int c_arg __P((const void , const void ));
	70
	71	in.dbsz = out.dbsz = 512;
ac894d30 KB	72
	73	while (oper = *++argv) {
	74	if ((arg = index(oper, '=')) == NULL)
	75	err("unknown operand %s", oper);
	76	*arg++ = '\0';
	77	if (!*arg)
	78	err("no value specified for %s", oper);
	79	tmp.name = oper;
	80	if (!(ap = (struct arg *)bsearch(&tmp, args,
	81	sizeof(args)/sizeof(struct arg), sizeof(struct arg),
	82	c_arg)))
	83	err("unknown operand %s", tmp.name);
	84	if (ddflags & ap->noset)
	85	err("%s: illegal argument combination or already set",
	86	tmp.name);
	87	ddflags \|= ap->set;
	88	ap->f(arg);
	89	}
	90
	91	/* Final sanity checks. */
	92
	93	if (ddflags & C_BS) {
	94	/*
	95	* Bs is turned off by any conversion -- we assume the user
	96	* just wanted to set both the input and output block sizes
	97	* and didn't want the bs semantics, so we don't warn.
	98	*/
acb87732	99	if (ddflags & (C_BLOCK\|C_LCASE\|C_SWAB\|C_UCASE\|C_UNBLOCK))
ac894d30 KB	100	ddflags &= ~C_BS;
	101
	102	/* Bs supersedes ibs and obs. */
	103	if (ddflags & C_BS && ddflags & (C_IBS\|C_OBS))
	104	warn("bs supersedes ibs and obs");
	105	}
	106
998e1ac9 KB	107	/*
	108	* Ascii/ebcdic and cbs implies block/unblock.
	109	* Block/unblock requires cbs and vice-versa.
	110	*/
ac894d30	111	if (ddflags & (C_BLOCK\|C_UNBLOCK)) {
998e1ac9 KB	112	if (!(ddflags & C_CBS))
998e1ac9 KB	113	err("record operations require cbs");
ac894d30	114	if (cbsz == 0)
998e1ac9 KB	115	err("cbs cannot be zero");
	116	cfunc = ddflags & C_BLOCK ? block : unblock;
	117	} else if (ddflags & C_CBS) {
	118	if (ddflags & (C_ASCII\|C_EBCDIC)) {
	119	if (ddflags & C_ASCII) {
	120	ddflags \|= C_UNBLOCK;
	121	cfunc = unblock;
	122	} else {
	123	ddflags \|= C_BLOCK;
	124	cfunc = block;
	125	}
	126	} else
	127	err("cbs meaningless if not doing record operations");
	128	if (cbsz == 0)
	129	err("cbs cannot be zero");
	130	} else
	131	cfunc = def;
ac894d30 KB	132
	133	if (in.dbsz == 0 \|\| out.dbsz == 0)
	134	err("buffer sizes cannot be zero");
	135
998e1ac9 KB	136	/*
	137	* Read, write and seek calls take ints as arguments. Seek sizes
	138	* could be larger if we wanted to do it in stages or check only
	139	* regular files, but it's probably not worth it.
	140	*/
	141	if (in.dbsz > INT_MAX \|\| out.dbsz > INT_MAX)
	142	err("buffer sizes cannot be greater than %d", INT_MAX);
	143	if (in.offset > INT_MAX / in.dbsz \|\| out.offset > INT_MAX / out.dbsz)
	144	err("seek offsets cannot be larger than %d", INT_MAX);
ac894d30 KB	145	}
	146
	147	static int
	148	c_arg(a, b)
	149	const void a, b;
	150	{
	151	return (strcmp(((struct arg )a)->name, ((struct arg )b)->name));
	152	}
	153
	154	static void
	155	f_bs(arg)
	156	char *arg;
	157	{
	158	in.dbsz = out.dbsz = (int)get_bsz(arg);
	159	}
	160
	161	static void
	162	f_cbs(arg)
	163	char *arg;
	164	{
	165	cbsz = (int)get_bsz(arg);
	166	}
	167
	168	static void
	169	f_count(arg)
	170	char *arg;
	171	{
	172	cpy_cnt = (u_int)get_bsz(arg);
998e1ac9	173	if (!cpy_cnt)
ac894d30 KB	174	terminate(0);
	175	}
	176
	177	static void
998e1ac9	178	f_files(arg)
ac894d30 KB	179	char *arg;
	180	{
	181	files_cnt = (int)get_bsz(arg);
	182	}
	183
	184	static void
	185	f_ibs(arg)
	186	char *arg;
	187	{
	188	if (!(ddflags & C_BS))
	189	in.dbsz = (int)get_bsz(arg);
	190	}
	191
	192	static void
	193	f_if(arg)
	194	char *arg;
	195	{
	196	in.name = arg;
	197	}
	198
	199	static void
	200	f_obs(arg)
	201	char *arg;
	202	{
	203	if (!(ddflags & C_BS))
	204	out.dbsz = (int)get_bsz(arg);
	205	}
	206
	207	static void
	208	f_of(arg)
	209	char *arg;
	210	{
	211	out.name = arg;
	212	}
	213
	214	static void
	215	f_seek(arg)
	216	char *arg;
	217	{
	218	out.offset = (u_int)get_bsz(arg);
	219	}
	220
	221	static void
	222	f_skip(arg)
	223	char *arg;
	224	{
	225	in.offset = (u_int)get_bsz(arg);
	226	}
	227
ac894d30 KB	228	static struct conv {
ac894d30 KB	229	char *name;
ac894d30	230	u_int set, noset;
998e1ac9	231	u_char *ctab;
ac894d30	232	} clist[] = {
998e1ac9 KB	233	"ascii", C_ASCII, C_EBCDIC, e2a_POSIX,
	234	"block", C_BLOCK, C_UNBLOCK, NULL,
	235	"ebcdic", C_EBCDIC, C_ASCII, a2e_POSIX,
	236	"ibm", C_EBCDIC, C_ASCII, a2ibm_POSIX,
	237	"lcase", C_LCASE, C_UCASE, NULL,
	238	"noerror", C_NOERROR, 0, NULL,
	239	"notrunc", C_NOTRUNC, 0, NULL,
	240	"oldascii", C_ASCII, C_EBCDIC, e2a_32V,
	241	"oldebcdic", C_EBCDIC, C_ASCII, a2e_32V,
	242	"oldibm", C_EBCDIC, C_ASCII, a2ibm_32V,
	243	"swab", C_SWAB, 0, NULL,
	244	"sync", C_SYNC, 0, NULL,
	245	"ucase", C_UCASE, C_LCASE, NULL,
	246	"unblock", C_UNBLOCK, C_BLOCK, NULL,
ac894d30 KB	247	};
	248
	249	static void
	250	f_conv(arg)
	251	char *arg;
	252	{
	253	register struct conv *cp;
	254	struct conv tmp;
	255	static int c_conv __P((const void , const void ));
	256
	257	while (arg != NULL) {
	258	tmp.name = strsep(&arg, ",");
	259	if (!(cp = (struct conv *)bsearch(&tmp, clist,
	260	sizeof(clist)/sizeof(struct conv), sizeof(struct conv),
	261	c_conv)))
	262	err("unknown conversion %s", tmp.name);
	263	if (ddflags & cp->noset)
	264	err("%s: illegal conversion combination", tmp.name);
	265	ddflags \|= cp->set;
998e1ac9 KB	266	if (cp->ctab)
998e1ac9 KB	267	ctab = cp->ctab;
ac894d30 KB	268	}
	269	}
	270
	271	static int
	272	c_conv(a, b)
	273	const void a, b;
	274	{
	275	return (strcmp(((struct conv )a)->name, ((struct conv )b)->name));
	276	}
	277
ac894d30 KB	278	/*
	279	* Convert an expression of the following forms to an unsigned long.
	280	* 1) A positive decimal number.
998e1ac9 KB	281	* 2) A positive decimal number followed by a b (mult by 512).
	282	* 3) A positive decimal number followed by a k (mult by 1024).
	283	* 4) A positive decimal number followed by a m (mult by 512).
	284	* 5) A positive decimal number followed by a w (mult by sizeof int)
	285	* 6) Two or more positive decimal numbers (with/without k,b or w).
ac894d30 KB	286	* seperated by x (also * for backwards compatibility), specifying
	287	* the product of the indicated values.
	288	*/
	289	static u_long
	290	get_bsz(val)
	291	char *val;
	292	{
	293	char *expr;
	294	u_long num, t;
	295
	296	num = strtoul(val, &expr, 0);
	297	if (num == ULONG_MAX) /* Overflow. */
	298	err("%s: %s", oper, strerror(errno));
	299	if (expr == val) /* No digits. */
	300	err("%s: illegal numeric value", oper);
	301
	302	switch(*expr) {
	303	case 'b':
	304	t = num;
	305	num *= 512;
	306	if (t > num)
	307	goto erange;
	308	++expr;
	309	break;
	310	case 'k':
	311	t = num;
	312	num *= 1024;
	313	if (t > num)
	314	goto erange;
	315	++expr;
	316	break;
0520b714 KB	317	case 'm':
	318	t = num;
	319	num *= 1048576;
	320	if (t > num)
	321	goto erange;
	322	++expr;
	323	break;
998e1ac9	324	case 'w':
ac894d30 KB	325	t = num;
	326	num *= sizeof(int);
	327	if (t > num)
	328	goto erange;
	329	++expr;
	330	break;
	331	}
	332
	333	switch(*expr) {
	334	case '\0':
	335	break;
0520b714	336	case '': / Backward compatible. */
ac894d30	337	case 'x':
ac894d30 KB	338	t = num;
	339	num *= get_bsz(expr + 1);
	340	if (t > num)
	341	erange: err("%s: %s", oper, strerror(ERANGE));
	342	break;
	343	default:
	344	err("%s: illegal numeric value", oper);
	345	}
	346	return(num);
	347	}