b91e2391 |
1 | /* -*-c-*- |
2 | * |
d0da523a |
3 | * $Id: mdwopt.c,v 1.9 2000/10/08 09:57:31 mdw Exp $ |
b91e2391 |
4 | * |
5 | * Options parsing, similar to GNU @getopt_long@ |
6 | * |
9cecacb1 |
7 | * (c) 1996 Straylight/Edgeware |
b91e2391 |
8 | */ |
9 | |
10 | /*----- Licensing notice --------------------------------------------------* |
11 | * |
12 | * This file is part of many programs. |
13 | * |
14 | * `mdwopt' is free software; you can redistribute it and/or modify |
15 | * it under the terms of the GNU Library General Public License as |
16 | * published by the Free Software Foundation; either version 2 of the |
17 | * License, or (at your option) any later version. |
18 | * |
19 | * `mdwopt' is distributed in the hope that it will be useful, |
20 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
21 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
22 | * GNU Library General Public License for more details. |
23 | * |
24 | * You should have received a copy of the GNU Library General Public |
4033d96b |
25 | * License along with `mdwopt'; if not, write to the Free |
26 | * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, |
27 | * MA 02111-1307, USA. |
b91e2391 |
28 | */ |
29 | |
30 | /*----- Revision history --------------------------------------------------* |
31 | * |
32 | * $Log: mdwopt.c,v $ |
d0da523a |
33 | * Revision 1.9 2000/10/08 09:57:31 mdw |
34 | * Use a copy of @str_qword@ for word splitting, to handle quotes |
35 | * properly. If building in mLib, use the real thing directly rather than |
36 | * the copy. |
37 | * |
4ab923c4 |
38 | * Revision 1.8 1999/08/19 18:35:27 mdw |
39 | * Lots of reformatting. Spurious wing comments expunged. |
40 | * |
fcda41cd |
41 | * Revision 1.7 1999/06/18 21:59:46 mdw |
42 | * Fix stupid bug which only read one word from environment variables. |
43 | * |
664ee843 |
44 | * Revision 1.6 1999/05/20 23:00:42 mdw |
45 | * Little formatting things. |
46 | * |
379c3233 |
47 | * Revision 1.5 1999/05/19 20:23:59 mdw |
48 | * Change naming to match newer mLib conventions. |
49 | * |
9cecacb1 |
50 | * Revision 1.4 1999/05/15 10:25:38 mdw |
51 | * Fix copyright information. |
52 | * |
4033d96b |
53 | * Revision 1.3 1999/05/14 18:51:42 mdw |
54 | * Reformat the LGPL notice slightly. |
55 | * |
5b59e227 |
56 | * Revision 1.2 1999/05/13 22:57:23 mdw |
57 | * Change `-ise' to `-ize' throughout. |
58 | * |
59 | * Revision 1.1.1.1 1999/05/05 19:23:47 mdw |
60 | * New import. The old CVS repository was lost in a disk disaster. |
b91e2391 |
61 | * |
62 | * --- Previous lives --- |
63 | * |
64 | * %Log: mdwopt.c,v % |
65 | * Revision 1.7 1997/09/11 09:19:11 mdw |
66 | * (mo__nextWord): Arrrgh. Don't free the environment variable buffer! |
67 | * People are still using it! |
68 | * |
69 | * Revision 1.6 1997/09/11 09:05:54 mdw |
70 | * (mo__nextWord): Fix bug which returns too many words from environment |
71 | * variables. |
72 | * |
73 | * Revision 1.5 1997/08/09 20:27:59 mdw |
74 | * Fix spelling of `Licensing'. |
75 | * |
76 | * Revision 1.4 1997/07/29 21:11:35 mdw |
77 | * Reformatted. Fixed buffer overflow when dealing with environment |
78 | * variables. Included NT in list of daft operating systems with `\' as a |
79 | * path separator. Fixed address of the FSF. |
80 | * |
81 | * Revision 1.3 1997/02/26 00:41:10 mdw |
82 | * Added GPL notice to the top. Slight formatting changes. |
83 | * |
84 | * Revision 1.2 1996/10/28 13:12:13 mdw |
85 | * Fixed calls to ctype.h routines. Arguments are cast to unsigned char |
86 | * to avoid invoking undefined behaviour caused by signedness of chars. |
87 | * |
88 | * Revision 1.1 1996/09/24 18:01:28 mdw |
89 | * Initial revision |
90 | * |
91 | */ |
92 | |
93 | /*----- External dependencies ---------------------------------------------*/ |
94 | |
95 | #include <ctype.h> |
96 | #include <stdio.h> |
97 | #include <stdlib.h> |
98 | #include <string.h> |
99 | |
100 | #include "mdwopt.h" |
101 | |
102 | /*----- Configuration things ----------------------------------------------*/ |
103 | |
104 | #if defined(__riscos) |
105 | # define PATHSEP '.' |
106 | #elif defined(__OS2__) || defined(__MSDOS__) || defined(__WINNT__) |
107 | # define PATHSEP '\\' |
108 | #else /* Assume a sane filing system */ |
109 | # define PATHSEP '/' |
110 | #endif |
111 | |
112 | /*----- Global variables --------------------------------------------------*/ |
113 | |
114 | mdwopt_data mdwopt_global = {0, 0, 0, 1, 0, 0, 0, 0, 0}; |
115 | |
116 | enum { |
379c3233 |
117 | ORD_PERMUTE = 0, /* Permute the options (default) */ |
118 | ORD_RETURN = 1, /* Return non-option things */ |
119 | ORD_POSIX = 2, /* Do POSIX-type hacking */ |
120 | ORD_NEGATE = 4 /* Magic negate-next-thing flag */ |
b91e2391 |
121 | }; |
122 | |
d0da523a |
123 | /*----- Word splitting ----------------------------------------------------*/ |
124 | |
125 | #ifdef BUILDING_MLIB |
126 | # include "str.h" |
127 | # define qword str_qword |
128 | #else |
129 | |
130 | /* --- @qword@ --- * |
131 | * |
132 | * Arguments: @char **pp@ = address of pointer into string |
133 | * @unsigned f@ = various flags |
134 | * |
135 | * Returns: Pointer to the next space-separated possibly-quoted word from |
136 | * the string, or null. |
137 | * |
138 | * Use: Fetches the next word from a string. If the flag |
139 | * @STRF_QUOTE@ is set, the `\' character acts as an escape, and |
140 | * single and double quotes protect whitespace. |
141 | */ |
142 | |
143 | #define STRF_QUOTE 1u |
144 | |
145 | static char *qword(char **pp, unsigned f) |
146 | { |
147 | char *p = *pp, *q, *qq; |
148 | int st = 0, pst = 0; |
149 | |
150 | /* --- Preliminaries --- */ |
151 | |
152 | if (!p) |
153 | return (0); |
154 | while (isspace((unsigned char)*p)) |
155 | p++; |
156 | if (!*p) { |
157 | *pp = 0; |
158 | return (0); |
159 | } |
160 | |
161 | /* --- Main work --- */ |
162 | |
163 | for (q = qq = p; *q; q++) { |
164 | switch (st) { |
165 | case '\\': |
166 | *qq++ = *q; |
167 | st = pst; |
168 | break; |
169 | case '\'': |
170 | case '\"': |
171 | if (*q == st) |
172 | st = pst = 0; |
173 | else if (*q == '\\') |
174 | st = '\\'; |
175 | else |
176 | *qq++ = *q; |
177 | break; |
178 | default: |
179 | if (isspace((unsigned char)*q)) { |
180 | do q++; while (*q && isspace((unsigned char)*q)); |
181 | goto done; |
182 | } else if (!(f & STRF_QUOTE)) |
183 | goto stdchar; |
184 | switch (*q) { |
185 | case '\\': |
186 | st = '\\'; |
187 | break; |
188 | case '\'': |
189 | case '\"': |
190 | st = pst = *q; |
191 | break; |
192 | default: |
193 | stdchar: |
194 | *qq++ = *q; |
195 | break; |
196 | } |
197 | } |
198 | } |
199 | |
200 | /* --- Finished --- */ |
201 | |
202 | done: |
203 | *pp = *q ? q : 0; |
204 | *qq++ = 0; |
205 | return (p); |
206 | } |
207 | |
208 | #endif |
209 | |
b91e2391 |
210 | /*----- Main code ---------------------------------------------------------*/ |
211 | |
d0da523a |
212 | /* --- @nextword@ --- * |
b91e2391 |
213 | * |
214 | * Arguments: @int argc@ = number of command line options |
215 | * @char *argv[]@ = pointer to command line options |
216 | * @mdwopt_data *data@ = pointer to persistent state |
217 | * |
218 | * Returns: Pointer to the next word to handle, or 0 |
219 | * |
220 | * Use: Extracts the next word from the command line or environment |
221 | * variable. |
222 | */ |
223 | |
d0da523a |
224 | static char *nextword(int argc, char *const *argv, mdwopt_data *data) |
b91e2391 |
225 | { |
226 | if (data->ind == -1) { |
d0da523a |
227 | char *p; |
228 | if ((p = qword(&data->env, STRF_QUOTE)) != 0) |
229 | return (p); |
b91e2391 |
230 | data->ind = 1; |
231 | } |
232 | |
233 | if (data->next == argc) |
234 | return (0); |
235 | return (argv[data->next++]); |
236 | } |
237 | |
379c3233 |
238 | /* --- @permute@ --- * |
b91e2391 |
239 | * |
240 | * Arguments: @char *argv[]@ = pointer to command line arguments |
241 | * @mdwopt_data *data@ = pointer to persistent data |
242 | * |
243 | * Returns: -- |
244 | * |
245 | * Use: Moves a command line option into the right place. |
246 | */ |
247 | |
379c3233 |
248 | static void permute(char *const *argv, mdwopt_data *data) |
b91e2391 |
249 | { |
250 | char **v = (char **)argv; |
251 | if (data->ind != -1) { |
252 | int i = data->next - 1; |
253 | char *p = v[i]; |
254 | while (i > data->ind) { |
255 | v[i] = v[i - 1]; |
256 | i--; |
257 | } |
258 | v[i] = p; |
259 | data->ind++; |
260 | } |
261 | } |
262 | |
379c3233 |
263 | /* --- @findOpt@ --- * |
b91e2391 |
264 | * |
265 | * Arguments: @int o@ = which option to search for |
266 | * @const char *shortopt@ = short options string to search |
267 | * @mdwopt_data *data@ = pointer to persistant state |
268 | * |
269 | * Returns: Pointer to rest of short options string (including magic |
270 | * characters) |
271 | * |
272 | * Use: Looks up a short option in the given string. |
273 | */ |
274 | |
379c3233 |
275 | static const char *findOpt(int o, const char *shortopt, |
b91e2391 |
276 | mdwopt_data *data) |
277 | { |
4ab923c4 |
278 | const char *p = shortopt; |
b91e2391 |
279 | for (;;) { |
4ab923c4 |
280 | if (!*p) |
b91e2391 |
281 | return (0); |
282 | |
379c3233 |
283 | if (o != *p || (p[1] != '+' && data->order & ORD_NEGATE)) { |
4ab923c4 |
284 | p++; |
285 | while (*p == '+') |
286 | p++; |
287 | while (*p == ':') |
b91e2391 |
288 | p++; |
b91e2391 |
289 | } |
290 | else |
291 | return (p + 1); |
292 | } |
293 | } |
294 | |
295 | /* --- @mdwopt@ --- * |
296 | * |
297 | * Arguments: @int argc@ = number of command line arguments |
298 | * @char * const *argv@ = pointer to command line arguments |
299 | * @const char *shortopt@ = pointer to short options information |
300 | * @const struct option *longopts@ = pointer to long opts info |
301 | * @int *longind@ = where to store matched longopt |
302 | * @mdwopt_data *data@ = persistent state for the parser |
303 | * @int flags@ = various useful flags |
304 | * |
305 | * Returns: Value of option found next, or an error character, or |
306 | * @EOF@ for the last thing. |
307 | * |
308 | * Use: Reads options. The routine should be more-or-less compatible |
309 | * with standard getopts, although it provides many more |
310 | * features even than the standard GNU implementation. |
311 | * |
312 | * The precise manner of options parsing is determined by |
313 | * various flag settings, which are described below. By setting |
314 | * flag values appropriately, you can achieve behaviour very |
315 | * similar to most other getopt routines. |
316 | * |
317 | * |
318 | * How options parsing appears to users |
319 | * |
320 | * A command line consists of a number of `words' (which may |
321 | * contain spaces, according to various shell quoting |
322 | * conventions). A word may be an option, an argument to an |
323 | * option, or a non-option. An option begins with a special |
324 | * character, usually `%|-|%', although `%|+|%' is also used |
325 | * sometimes. As special exceptions, the word containing only a |
326 | * `%|-|%' is considered to be a non-option, since it usually |
327 | * represents standard input or output as a filename, and the |
328 | * word containing a double-dash `%|--|%' is used to mark all |
329 | * following words as being non-options regardless of their |
330 | * initial character. |
331 | * |
332 | * Traditionally, all words after the first non-option have been |
333 | * considered to be non-options automatically, so that options |
334 | * must be specified before filenames. However, this |
335 | * implementation can extract all the options from the command |
336 | * line regardless of their position. This can usually be |
337 | * disabled by setting one of the environment variables |
338 | * `%|POSIXLY_CORRECT|%' or `%|_POSIX_OPTION_ORDER|%'. |
339 | * |
340 | * There are two different styles of options: `short' and |
341 | * `long'. |
342 | * |
343 | * Short options are the sort which Unix has known for ages: an |
344 | * option is a single letter, preceded by a `%|-|%'. Short |
345 | * options can be joined together to save space (and possibly to |
346 | * make silly words): e.g., instead of giving options |
347 | * `%|-x -y|%', a user could write `%|-xy|%'. Some short |
348 | * options can have arguments, which appear after the option |
349 | * letter, either immediately following, or in the next `word' |
350 | * (so an option with an argument could be written as |
351 | * `%|-o foo|%' or as `%|-ofoo|%'). Note that options with |
352 | * optional arguments must be written in the second style. |
353 | * |
354 | * When a short option controls a flag setting, it is sometimes |
355 | * possible to explicitly turn the flag off, as well as turning |
356 | * it on, (usually to override default options). This is |
357 | * usually done by using a `%|+|%' instead of a `%|-|%' to |
358 | * introduce the option. |
359 | * |
5b59e227 |
360 | * Long options, as popularized by the GNU utilities, are given |
b91e2391 |
361 | * long-ish memorable names, preceded by a double-dash `%|--|%'. |
362 | * Since their names are more than a single character, long |
363 | * options can't be combined in the same way as short options. |
364 | * Arguments to long options may be given either in the same |
365 | * `word', separated from the option name by an equals sign, or |
366 | * in the following `word'. |
367 | * |
368 | * Long option names can be abbreviated if necessary, as long |
369 | * as the abbreviation is unique. This means that options can |
370 | * have sensible and memorable names but still not require much |
371 | * typing from an experienced user. |
372 | * |
373 | * Like short options, long options can control flag settings. |
374 | * The options to manipulate these settings come in pairs: an |
375 | * option of the form `%|--set-flag|%' might set the flag, while |
376 | * an option of the form `%|--no-set-flag|%' might clear it. |
377 | * |
378 | * It is usual for applications to provide both short and long |
379 | * options with identical behaviour. Some applications with |
380 | * lots of options may only provide long options (although they |
381 | * will often be only two or three characters long). In this |
382 | * case, long options can be preceded with a single `%|-|%' |
383 | * character, and negated by a `%|+|%' character. |
384 | * |
385 | * Finally, some (older) programs accept arguments of the form |
386 | * `%%@.{"-"<number>}%%', to set some numerical parameter, |
387 | * typically a line count of some kind. |
388 | * |
389 | * |
390 | * How programs parse options |
391 | * |
392 | * An application parses its options by calling mdwopt |
393 | * repeatedly. Each time it is called, mdwopt returns a value |
394 | * describing the option just read, and stores information about |
395 | * the option in a data block. The value %$-1$% is returned |
396 | * when there are no more options to be read. The `%|?|%' |
397 | * character is returned when an error is encountered. |
398 | * |
399 | * Before starting to parse options, the value @data->ind@ must |
400 | * be set to 0 or 1. The value of @data->err@ can also be set, |
401 | * to choose whether errors are reported by mdwopt. |
402 | * |
403 | * The program's `@argc@' and `@argv@' arguments are passed to |
404 | * the options parser, so that it can read the command line. A |
405 | * flags word is also passed, allowing the program fine control |
406 | * over parsing. The flags are described above. |
407 | * |
408 | * Short options are described by a string, which once upon a |
409 | * time just contained the permitted option characters. Now the |
410 | * options string begins with a collection of flag characters, |
411 | * and various flag characters can be put after options |
412 | * characters to change their properties. |
413 | * |
414 | * If the first character of the short options string is |
415 | * `%|+|%', `%|-|%' or `%|!|%', the order in which options are |
416 | * read is modified, as follows: |
417 | * |
418 | * `%|+|%' forces the POSIX order to be used. As soon as a non- |
419 | * option is found, mdwopt returns %$-1$%. |
420 | * |
421 | * `%|-|%' makes mdwopt treat non-options as being `special' |
422 | * sorts of option. When a non-option word is found, the |
423 | * value 0 is returned, and the actual text of the word |
424 | * is stored as being the option's argument. |
425 | * |
426 | * `%|!|%' forces the default order to be used. The entire |
427 | * command line is scanned for options, which are |
428 | * returned in order. However, during this process, |
429 | * the options are moved in the @argv@ array, so that |
430 | * they appear before the non- options. |
431 | * |
432 | * A `%|:|%' character may be placed after the ordering flag (or |
433 | * at the very beginning if no ordering flag is given) which |
434 | * indicates that the character `%|:|%', rather than `%|?|%', |
435 | * should be returned if a missing argument error is detected. |
436 | * |
437 | * Each option in the string can be followed by a `%|+|%' sign, |
438 | * indicating that it can be negated, a `%|:|%' sign indicating |
439 | * that it requires an argument, or a `%|::|%' string, |
440 | * indicating an optional argument. Both `%|+|%' and `%|:|%' or |
441 | * `%|::|%' may be given, although the `%|+|%' must come first. |
442 | * |
443 | * If an option is found, the option character is returned to |
444 | * the caller. A pointer to an argument is stored in |
445 | * @data->arg@, or @NULL@ is stored if there was no argument. |
446 | * If a negated option was found, the option character is |
379c3233 |
447 | * returned ORred with @OPTF_NEGATED@ (bit 8 set). |
b91e2391 |
448 | * |
449 | * Long options are described in a table. Each entry in the |
450 | * table is of type @struct option@, and the table is terminated |
451 | * by an entry whose @name@ field is null. Each option has |
452 | * a flags word which, due to historical reasons, is called |
453 | * @has_arg@. This describes various properties of the option, |
454 | * such as what sort of argument it takes, and whether it can |
455 | * be negated. |
456 | * |
457 | * When mdwopt finds a long option, it looks the name up in the |
458 | * table. The index of the matching entry is stored in the |
459 | * @longind@ variable, passed to mdwopt (unless @longind@ is 0): |
460 | * a value of %$-1$% indicates that no long option was |
461 | * found. The behaviour is then dependent on the values in the |
462 | * table entry. If @flag@ is nonzero, it points to an integer |
463 | * to be modified by mdwopt. Usually the value in the @val@ |
464 | * field is simply stored in the @flag@ variable. If the flag |
379c3233 |
465 | * @OPTF_SWITCH@ is set, however, the value is combined with |
b91e2391 |
466 | * the existing value of the flags using a bitwise OR. If |
379c3233 |
467 | * @OPTF_NEGATE@ is set, then the flag bit will be cleared if a |
b91e2391 |
468 | * matching negated long option is found. The value 0 is |
469 | * returned. |
470 | * |
471 | * If @flag@ is zero, the value in @val@ is returned by mdwopt, |
472 | * possibly with bit 8 set if the option was negated. |
473 | * |
474 | * Arguments for long options are stored in @data->arg@, as |
475 | * before. |
476 | * |
477 | * Numeric options, if enabled, cause the value `%|#|%' to be |
478 | * returned, and the numeric value to be stored in @data->opt@. |
479 | * |
379c3233 |
480 | * If the flag @OPTF_ENVVAR@ is set on entry, options will be |
b91e2391 |
481 | * extracted from an environment variable whose name is built by |
664ee843 |
482 | * capitalizing all the letters of the program's name. (This |
b91e2391 |
483 | * allows a user to have different default settings for a |
664ee843 |
484 | * program, by calling it through different symbolic links.) |
485 | */ |
b91e2391 |
486 | |
487 | int mdwopt(int argc, char *const *argv, |
488 | const char *shortopt, |
489 | const struct option *longopts, int *longind, |
490 | mdwopt_data *data, int flags) |
491 | { |
492 | /* --- Local variables --- */ |
493 | |
4ab923c4 |
494 | char *p, *q, *r; |
495 | char *prefix; |
496 | int i; |
497 | char noarg = '?'; |
b91e2391 |
498 | |
499 | /* --- Sort out our data --- */ |
500 | |
4ab923c4 |
501 | if (!data) |
502 | data = &mdwopt_global; |
b91e2391 |
503 | |
504 | /* --- See if this is the first time --- */ |
505 | |
379c3233 |
506 | if (data->ind == 0 || (data->ind == 1 && ~flags & OPTF_NOPROGNAME)) { |
b91e2391 |
507 | |
508 | /* --- Sort out default returning order --- */ |
509 | |
4ab923c4 |
510 | if (getenv("_POSIX_OPTION_ORDER") || |
511 | getenv("POSIXLY_CORRECT")) |
512 | data->order = ORD_POSIX; |
b91e2391 |
513 | else |
4ab923c4 |
514 | data->order = ORD_PERMUTE; |
b91e2391 |
515 | |
516 | /* --- Now see what the caller actually wants --- */ |
517 | |
4ab923c4 |
518 | switch (shortopt[0]) { |
519 | case '-': |
379c3233 |
520 | data->order = ORD_RETURN; |
b91e2391 |
521 | break; |
4ab923c4 |
522 | case '+': |
379c3233 |
523 | data->order = ORD_POSIX; |
b91e2391 |
524 | break; |
4ab923c4 |
525 | case '!': |
379c3233 |
526 | data->order = ORD_PERMUTE; |
b91e2391 |
527 | break; |
528 | } |
529 | |
530 | /* --- Now decide on the program's name --- */ |
531 | |
379c3233 |
532 | if (~flags & OPTF_NOPROGNAME) { |
b91e2391 |
533 | p = q = (char *)argv[0]; |
534 | while (*p) { |
535 | if (*p++ == PATHSEP) |
536 | q = p; |
537 | } |
538 | data->prog = q; |
539 | |
540 | data->ind = data->next = 1; |
541 | data->list = 0; |
542 | |
543 | /* --- See about environment variables --- * |
544 | * |
545 | * Be careful. The program may be setuid, and an attacker might have |
546 | * given us a long name in @argv[0]@. If the name is very long, don't |
547 | * support this option. |
548 | */ |
549 | |
379c3233 |
550 | if (flags & OPTF_ENVVAR && strlen(data->prog) < 48) { |
b91e2391 |
551 | |
552 | char buf[64]; |
553 | |
554 | /* --- For RISC OS, support a different format --- * |
555 | * |
556 | * Acorn's RISC OS tends to put settings in variables named |
557 | * `App$Options' rather than `APP'. Under RISC OS, I'll support |
558 | * both methods, just to avoid confuddlement. |
559 | */ |
560 | |
561 | #ifdef __riscos |
562 | sprintf(buf, "%s$Options", data->prog); |
563 | p = getenv(buf); |
564 | if (!p) { |
565 | #endif |
566 | |
4ab923c4 |
567 | p = buf; |
568 | q = data->prog; |
569 | while (*q) |
570 | *p++ = toupper(*q++); |
571 | *p++ = 0; |
572 | p = getenv(buf); |
b91e2391 |
573 | |
574 | #ifdef __riscos |
575 | } |
576 | #endif |
577 | |
578 | /* --- Copy the options string into a buffer --- */ |
579 | |
4ab923c4 |
580 | if (p) { |
581 | q = malloc(strlen(p) + 1); |
582 | if (!q) { |
583 | fprintf(stderr, |
b91e2391 |
584 | "%s: Not enough memory to read settings in " |
585 | "environment variable\n", |
586 | data->prog); |
4ab923c4 |
587 | } else { |
588 | strcpy(q, p); |
589 | data->ind = -1; |
590 | data->env = data->estart = q; |
b91e2391 |
591 | } |
592 | } |
593 | |
594 | } |
595 | } |
596 | else |
597 | data->ind = data->next = 0; |
598 | } |
599 | |
600 | /* --- Do some initial bodgery --- * |
601 | * |
602 | * The @shortopt@ string can have some interesting characters at the |
603 | * beginning. We'll skip past them. |
604 | */ |
605 | |
606 | switch (shortopt[0]) { |
607 | case '+': |
608 | case '-': |
609 | case '!': |
610 | shortopt++; |
611 | break; |
612 | } |
613 | |
614 | if (shortopt[0] == ':') { |
615 | noarg = shortopt[0]; |
616 | shortopt++; |
617 | } |
618 | |
4ab923c4 |
619 | if (longind) |
620 | *longind = -1; |
621 | data->opt = -1; |
622 | data->arg = 0; |
b91e2391 |
623 | |
624 | /* --- Now go off and search for an option --- */ |
625 | |
626 | if (!data->list || !*data->list) { |
627 | data->order &= 3; /* Clear negation flag */ |
628 | |
629 | /* --- Now we need to find the next option --- * |
630 | * |
631 | * Exactly how we do this depends on the settings of the order variable. |
632 | * We identify options as being things starting with `%|-|%', and which |
633 | * aren't equal to `%|-|%' or `%|--|%'. We'll look for options until: |
634 | * |
379c3233 |
635 | * * We find something which isn't an option AND @order == ORD_POSIX@ |
b91e2391 |
636 | * * We find a `%|--|%' |
637 | * * We reach the end of the list |
638 | * |
639 | * There are some added little wrinkles, which we'll meet as we go. |
640 | */ |
641 | |
4ab923c4 |
642 | for (;;) { |
d0da523a |
643 | p = nextword(argc, argv, data); |
4ab923c4 |
644 | if (!p) |
645 | return (EOF); |
b91e2391 |
646 | |
647 | /* --- See if we've found an option --- */ |
648 | |
379c3233 |
649 | if ((p[0] == '-' || (p[0] == '+' && flags & OPTF_NEGATION)) && |
b91e2391 |
650 | p[1] != 0) { |
4ab923c4 |
651 | if (strcmp(p, "--") == 0) { |
652 | permute(argv, data); |
653 | return (EOF); |
b91e2391 |
654 | } |
4ab923c4 |
655 | break; |
b91e2391 |
656 | } |
657 | |
658 | /* --- Figure out how to proceed --- */ |
659 | |
660 | switch (data->order & 3) { |
4ab923c4 |
661 | case ORD_POSIX: |
662 | return (EOF); |
b91e2391 |
663 | break; |
4ab923c4 |
664 | case ORD_PERMUTE: |
b91e2391 |
665 | break; |
4ab923c4 |
666 | case ORD_RETURN: |
667 | permute(argv, data); |
668 | data->arg = p; |
669 | return (0); |
b91e2391 |
670 | } |
671 | } |
672 | |
673 | /* --- We found an option --- */ |
674 | |
4ab923c4 |
675 | permute(argv, data); |
b91e2391 |
676 | |
677 | /* --- Check for a numeric option --- * |
678 | * |
679 | * We only check the first character (or the second if the first is a |
680 | * sign). This ought to be enough. |
681 | */ |
682 | |
379c3233 |
683 | if (flags & OPTF_NUMBERS && (p[0] == '-' || flags & OPTF_NEGNUMBER)) { |
b91e2391 |
684 | if (((p[1] == '+' || p[1] == '-') && isdigit((unsigned char)p[2])) || |
685 | isdigit((unsigned char)p[1])) { |
686 | data->opt = strtol(p + 1, &data->arg, 10); |
687 | while (isspace((unsigned char)data->arg[0])) |
688 | data->arg++; |
689 | if (!data->arg[0]) |
690 | data->arg = 0; |
379c3233 |
691 | return (p[0] == '-' ? '#' : '#' | OPTF_NEGATED); |
b91e2391 |
692 | } |
693 | } |
694 | |
695 | /* --- Check for a long option --- */ |
696 | |
697 | if (p[0] == '+') |
379c3233 |
698 | data->order |= ORD_NEGATE; |
b91e2391 |
699 | |
700 | if (((p[0] == '-' && p[1] == '-') || |
379c3233 |
701 | (flags & OPTF_NOSHORTS && !findOpt(p[1], shortopt, data))) && |
4ab923c4 |
702 | (~flags & OPTF_NOLONGS)) |
b91e2391 |
703 | { |
4ab923c4 |
704 | int match = -1; |
b91e2391 |
705 | |
4ab923c4 |
706 | if (p[0] == '+') { |
707 | data->order |= ORD_NEGATE; |
708 | p++; |
709 | prefix = "+"; |
710 | } else if (p[1] == '-') { |
379c3233 |
711 | if ((flags & OPTF_NEGATION) && strncmp(p + 2, "no-", 3) == 0) { |
4ab923c4 |
712 | p += 5; |
713 | prefix = "--no-"; |
714 | data->order |= ORD_NEGATE; |
b91e2391 |
715 | } else { |
4ab923c4 |
716 | p += 2; |
717 | prefix = "--"; |
b91e2391 |
718 | } |
719 | } else { |
379c3233 |
720 | if ((flags & OPTF_NEGATION) && strncmp(p + 1, "no-", 3) == 0) { |
4ab923c4 |
721 | p += 4; |
722 | prefix = "-no-"; |
723 | data->order |= ORD_NEGATE; |
b91e2391 |
724 | } else { |
4ab923c4 |
725 | p++; |
726 | prefix = "-"; |
b91e2391 |
727 | } |
728 | } |
729 | |
4ab923c4 |
730 | for (i = 0; longopts[i].name; i++) { |
379c3233 |
731 | if ((data->order & ORD_NEGATE) && |
732 | (~longopts[i].has_arg & OPTF_NEGATE)) |
4ab923c4 |
733 | continue; |
734 | |
735 | r = (char *) longopts[i].name; |
736 | q = p; |
737 | for (;;) { |
738 | if (*q == 0 || *q == '=') { |
739 | if (*r == 0) { |
740 | match = i; |
741 | goto botched; |
b91e2391 |
742 | } |
4ab923c4 |
743 | if (match == -1) { |
744 | match = i; |
745 | break; |
b91e2391 |
746 | } else { |
4ab923c4 |
747 | match = -1; |
748 | goto botched; |
b91e2391 |
749 | } |
750 | } |
4ab923c4 |
751 | else if (*q != *r) |
752 | break; |
753 | q++, r++; |
b91e2391 |
754 | } |
755 | } |
756 | |
757 | botched: |
4ab923c4 |
758 | if (match == -1) { |
b91e2391 |
759 | if (data->err) { |
5b59e227 |
760 | fprintf(stderr, "%s: unrecognized option `%s%s'\n", |
b91e2391 |
761 | data->prog, |
762 | prefix, p); |
763 | } |
764 | return ('?'); |
765 | } |
766 | |
4ab923c4 |
767 | if (longind) |
768 | *longind = match; |
b91e2391 |
769 | |
770 | /* --- Handle argument behaviour --- */ |
771 | |
4ab923c4 |
772 | while (*p != 0 && *p != '=') |
b91e2391 |
773 | p++; |
4ab923c4 |
774 | p = (*p ? p + 1 : 0); |
775 | q = (char *) longopts[match].name; |
b91e2391 |
776 | |
4ab923c4 |
777 | switch (longopts[match].has_arg & OPTF_ARG) { |
778 | case OPTF_NOARG: |
b91e2391 |
779 | if (p) { |
780 | if (data->err) { |
781 | fprintf(stderr, |
782 | "%s: option `%s%s' does not accept arguments\n", |
783 | data->prog, |
784 | prefix, q); |
785 | } |
786 | return ('?'); |
787 | } |
788 | break; |
789 | |
4ab923c4 |
790 | case OPTF_ARGREQ: |
791 | if (!p) { |
d0da523a |
792 | p = nextword(argc, argv, data); |
b91e2391 |
793 | |
4ab923c4 |
794 | if (!p) { |
b91e2391 |
795 | if (data->err) { |
796 | fprintf(stderr, "%s: option `%s%s' requires an argument\n", |
797 | data->prog, |
798 | prefix, q); |
799 | } |
800 | return (noarg); |
801 | } |
802 | |
379c3233 |
803 | permute(argv, data); |
b91e2391 |
804 | } |
805 | break; |
806 | |
4ab923c4 |
807 | case OPTF_ARGOPT: |
b91e2391 |
808 | /* Who cares? */ |
809 | break; |
810 | } |
811 | data->arg = p; |
812 | |
813 | /* --- Do correct things now we have a match --- */ |
814 | |
4ab923c4 |
815 | if (longopts[match].flag) { |
379c3233 |
816 | if (longopts[match].has_arg & OPTF_SWITCH) { |
817 | if (data->order & ORD_NEGATE) |
b91e2391 |
818 | *longopts[match].flag &= ~longopts[match].val; |
819 | else |
820 | *longopts[match].flag |= longopts[match].val; |
821 | } else { |
379c3233 |
822 | if (data->order & ORD_NEGATE) |
b91e2391 |
823 | *longopts[match].flag = 0; |
824 | else |
825 | *longopts[match].flag = longopts[match].val; |
826 | } |
4ab923c4 |
827 | return (0); |
b91e2391 |
828 | } else { |
379c3233 |
829 | if (data->order & ORD_NEGATE) |
830 | return (longopts[match].val | OPTF_NEGATED); |
b91e2391 |
831 | else |
832 | return (longopts[match].val); |
833 | } |
834 | } |
835 | |
836 | /* --- Do short options things --- */ |
837 | |
838 | else { |
4ab923c4 |
839 | if (p[0] == '+') |
379c3233 |
840 | data->order |= ORD_NEGATE; |
4ab923c4 |
841 | data->list = p + 1; |
b91e2391 |
842 | } |
843 | } |
844 | |
845 | /* --- Now process the short options --- */ |
846 | |
4ab923c4 |
847 | i = *data->list++; |
848 | data->opt = i; |
b91e2391 |
849 | |
379c3233 |
850 | p = (char *) findOpt(i, shortopt, data); |
4ab923c4 |
851 | if (!p) { |
b91e2391 |
852 | if (data->err) { |
853 | fprintf(stderr, "%s: unknown option `%c%c'\n", |
854 | data->prog, |
379c3233 |
855 | data->order & ORD_NEGATE ? '+' : '-', |
b91e2391 |
856 | i); |
857 | } |
858 | return ('?'); |
859 | } |
860 | |
4ab923c4 |
861 | data->opt = i; |
b91e2391 |
862 | |
863 | /* --- Sort out an argument, if we expect one --- */ |
864 | |
4ab923c4 |
865 | if (p[0] == ':') { |
866 | q = (data->list[0] ? data->list : 0); |
867 | data->list = 0; |
868 | if (p[1] != ':' && !q) { |
b91e2391 |
869 | |
870 | /* --- Same code as before --- */ |
871 | |
d0da523a |
872 | q = nextword(argc, argv, data); |
4ab923c4 |
873 | if (!q) { |
b91e2391 |
874 | if (data->err) { |
875 | fprintf(stderr, "%s: option `%c%c' requires an argument\n", |
876 | data->prog, |
379c3233 |
877 | data->order & ORD_NEGATE ? '+' : '-', |
b91e2391 |
878 | i); |
879 | } |
880 | return (noarg); |
881 | } |
379c3233 |
882 | permute(argv, data); |
b91e2391 |
883 | } |
884 | |
885 | data->arg = q; |
886 | } |
379c3233 |
887 | return ((data->order & ORD_NEGATE) ? i | OPTF_NEGATED : i); |
b91e2391 |
888 | } |
889 | |
890 | /*----- That's all, folks -------------------------------------------------*/ |