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