egetopt.c revision 331722 
1/*-
2 * Copyright (c) 1991 Keith Muller.
3 * Copyright (c) 1993
4 *	The Regents of the University of California.  All rights reserved.
5 *
6 * This code is derived from software contributed to Berkeley by
7 * Keith Muller of the University of California, San Diego.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 *    notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 *    notice, this list of conditions and the following disclaimer in the
16 *    documentation and/or other materials provided with the distribution.
17 * 3. All advertising materials mentioning features or use of this software
18 *    must display the following acknowledgement:
19 *	This product includes software developed by the University of
20 *	California, Berkeley and its contributors.
21 * 4. Neither the name of the University nor the names of its contributors
22 *    may be used to endorse or promote products derived from this software
23 *    without specific prior written permission.
24 *
25 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
26 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
28 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
29 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
30 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
31 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
32 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
33 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
34 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
35 * SUCH DAMAGE.
36 */
37
38#if 0
39#ifndef lint
40static char sccsid[] = "@(#)egetopt.c	8.1 (Berkeley) 6/6/93";
41#endif /* not lint */
42#endif
43
44#include <sys/cdefs.h>
45__FBSDID("$FreeBSD: stable/11/usr.bin/pr/egetopt.c 331722 2018-03-29 02:50:57Z eadler $");
46
47#include <ctype.h>
48#include <stdio.h>
49#include <string.h>
50
51#include "extern.h"
52
53/*
54 * egetopt:	get option letter from argument vector (an extended
55 *		version of getopt).
56 *
57 * Non standard additions to the ostr specs are:
58 * 1) '?': immediate value following arg is optional (no white space
59 *    between the arg and the value)
60 * 2) '#': +/- followed by a number (with an optional sign but
61 *    no white space between the arg and the number). The - may be
62 *    combined with other options, but the + cannot.
63 */
64
65int	eopterr = 1;		/* if error message should be printed */
66int	eoptind = 1;		/* index into parent argv vector */
67int	eoptopt;		/* character checked for validity */
68char	*eoptarg;		/* argument associated with option */
69
70#define	BADCH	(int)'?'
71
72static char	emsg[] = "";
73
74int
75egetopt(int nargc, char * const *nargv, const char *ostr)
76{
77	static char *place = emsg;	/* option letter processing */
78	char *oli;			/* option letter list index */
79	static int delim;		/* which option delimiter */
80	char *p;
81	static char savec = '\0';
82
83	if (savec != '\0') {
84		*place = savec;
85		savec = '\0';
86	}
87
88	if (!*place) {
89		/*
90		 * update scanning pointer
91		 */
92		if ((eoptind >= nargc) ||
93		    ((*(place = nargv[eoptind]) != '-') && (*place != '+'))) {
94			place = emsg;
95			return (-1);
96		}
97
98		delim = (int)*place;
99		if (place[1] && *++place == '-' && !place[1]) {
100			/*
101			 * found "--"
102			 */
103			++eoptind;
104			place = emsg;
105			return (-1);
106		}
107	}
108
109	/*
110	 * check option letter
111	 */
112	if ((eoptopt = (int)*place++) == (int)':' || (eoptopt == (int)'?') ||
113	    !(oli = strchr(ostr, eoptopt))) {
114		/*
115		 * if the user didn't specify '-' as an option,
116		 * assume it means -1 when by itself.
117		 */
118		if ((eoptopt == (int)'-') && !*place)
119			return (-1);
120		if (strchr(ostr, '#') && (isdigit(eoptopt) ||
121		    (((eoptopt == (int)'-') || (eoptopt == (int)'+')) &&
122		      isdigit(*place)))) {
123			/*
124			 * # option: +/- with a number is ok
125			 */
126			for (p = place; *p != '\0'; ++p) {
127				if (!isdigit(*p))
128					break;
129			}
130			eoptarg = place-1;
131
132			if (*p == '\0') {
133				place = emsg;
134				++eoptind;
135			} else {
136				place = p;
137				savec = *p;
138				*place = '\0';
139			}
140			return (delim);
141		}
142
143		if (!*place)
144			++eoptind;
145		if (eopterr) {
146			if (!(p = strrchr(*nargv, '/')))
147				p = *nargv;
148			else
149				++p;
150			(void)fprintf(stderr, "%s: illegal option -- %c\n",
151			    p, eoptopt);
152		}
153		return (BADCH);
154	}
155	if (delim == (int)'+') {
156		/*
157		 * '+' is only allowed with numbers
158		 */
159		if (!*place)
160			++eoptind;
161		if (eopterr) {
162			if (!(p = strrchr(*nargv, '/')))
163				p = *nargv;
164			else
165				++p;
166			(void)fprintf(stderr,
167				"%s: illegal '+' delimiter with option -- %c\n",
168				p, eoptopt);
169		}
170		return (BADCH);
171	}
172	++oli;
173	if ((*oli != ':') && (*oli != '?')) {
174		/*
175		 * don't need argument
176		 */
177		eoptarg = NULL;
178		if (!*place)
179			++eoptind;
180		return (eoptopt);
181	}
182
183	if (*place) {
184		/*
185		 * no white space
186		 */
187		eoptarg = place;
188	} else if (*oli == '?') {
189		/*
190		 * no arg, but NOT required
191		 */
192		eoptarg = NULL;
193	} else if (nargc <= ++eoptind) {
194		/*
195		 * no arg, but IS required
196		 */
197		place = emsg;
198		if (eopterr) {
199			if (!(p = strrchr(*nargv, '/')))
200				p = *nargv;
201			else
202				++p;
203			(void)fprintf(stderr,
204			    "%s: option requires an argument -- %c\n", p,
205			    eoptopt);
206		}
207		return (BADCH);
208	} else {
209		/*
210		 * arg has white space
211		 */
212		eoptarg = nargv[eoptind];
213	}
214	place = emsg;
215	++eoptind;
216	return (eoptopt);
217}
218