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