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 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