2 * Copyright (c) 1997, 1998 Kungliga Tekniska Högskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * 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 Kungliga Tekniska
20 * Högskolan and its contributors.
22 * 4. Neither the name of the Institute nor the names of its contributors
23 * may be used to endorse or promote products derived from this software
24 * without specific prior written permission.
26 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
27 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
29 * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
30 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
31 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
32 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
33 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
34 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
35 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
40 RCSID("$Id: getarg.c,v 1.25 1998/11/22 09:45:05 assar Exp $");
43 #include <sys/ttycom.h>
48 #define ISFLAG(X) ((X).type == arg_flag || (X).type == arg_negative_flag)
51 print_arg (char *string, size_t len, int mdoc, int longp, struct getargs *arg)
62 strncat(string, "= Ns", len);
63 strncat(string, " Ar ", len);
66 strncat (string, "=", len);
68 strncat (string, " ", len);
72 else if (arg->type == arg_integer)
74 else if (arg->type == arg_string)
79 strncat(string, s, len);
84 check_column(FILE *f, int col, int len, int columns)
86 if(col + len > columns) {
88 col = fprintf(f, " ");
94 arg_printusage (struct getargs *args,
97 const char *extra_string)
102 int col = 0, columns;
107 col += fprintf (stderr, "Usage: %s", progname);
108 for (i = 0; i < num_args; ++i) {
111 if (args[i].long_name) {
113 strncat(buf, "[--", sizeof(buf));
115 if(args[i].type == arg_negative_flag) {
116 strncat(buf, "no-", sizeof(buf));
119 strncat(buf, args[i].long_name, sizeof(buf));
120 len += strlen(args[i].long_name);
121 len += print_arg(buf + strlen(buf), sizeof(buf) - strlen(buf),
123 strncat(buf, "]", sizeof(buf));
124 if(args[i].type == arg_strings)
125 strncat(buf, "...", sizeof(buf));
126 col = check_column(stderr, col, strlen(buf) + 1, columns);
127 col += fprintf(stderr, " %s", buf);
129 if (args[i].short_name) {
130 snprintf(buf, sizeof(buf), "[-%c", args[i].short_name);
132 len += print_arg(buf + strlen(buf), sizeof(buf) - strlen(buf),
134 strncat(buf, "]", sizeof(buf));
135 if(args[i].type == arg_strings)
136 strncat(buf, "...", sizeof(buf));
137 col = check_column(stderr, col, strlen(buf) + 1, columns);
138 col += fprintf(stderr, " %s", buf);
140 if (args[i].long_name && args[i].short_name)
142 max_len = max(max_len, len);
145 col = check_column(stderr, col, strlen(extra_string) + 1, columns);
146 fprintf (stderr, " %s\n", extra_string);
148 fprintf (stderr, "\n");
149 for (i = 0; i < num_args; ++i) {
153 if (args[i].short_name) {
154 count += fprintf (stderr, "-%c", args[i].short_name);
155 print_arg (buf, sizeof(buf), 0, 0, &args[i]);
156 count += fprintf(stderr, "%s", buf);
158 if (args[i].short_name && args[i].long_name)
159 count += fprintf (stderr, ", ");
160 if (args[i].long_name) {
161 count += fprintf (stderr, "--");
162 if (args[i].type == arg_negative_flag)
163 count += fprintf (stderr, "no-");
164 count += fprintf (stderr, "%s", args[i].long_name);
165 print_arg (buf, sizeof(buf), 0, 1, &args[i]);
166 count += fprintf(stderr, "%s", buf);
168 while(count++ <= max_len)
170 fprintf (stderr, "%s\n", args[i].help);
176 add_string(getarg_strings *s, char *value)
178 s->strings = realloc(s->strings, (s->num_strings + 1) * sizeof(*s->strings));
179 s->strings[s->num_strings] = value;
184 arg_match_long(struct getargs *args, size_t num_args,
190 int partial_match = 0;
191 struct getargs *partial = NULL;
192 struct getargs *current = NULL;
196 argv_len = strlen(argv);
197 p = strchr (argv, '=');
201 for (i = 0; i < num_args; ++i) {
202 if(args[i].long_name) {
203 int len = strlen(args[i].long_name);
205 int p_len = argv_len;
209 if (strncmp (args[i].long_name, p, p_len) == 0) {
217 } else if (ISFLAG(args[i]) && strncmp (p, "no-", 3) == 0) {
229 if (current == NULL) {
230 if (partial_match == 1)
233 return ARG_ERR_NO_MATCH;
236 if(*optarg == '\0' && !ISFLAG(*current))
237 return ARG_ERR_NO_MATCH;
238 switch(current->type){
242 if(sscanf(optarg + 1, "%d", &tmp) != 1)
243 return ARG_ERR_BAD_ARG;
244 *(int*)current->value = tmp;
249 *(char**)current->value = optarg + 1;
254 add_string((getarg_strings*)current->value, optarg + 1);
258 case arg_negative_flag:
260 int *flag = current->value;
261 if(*optarg == '\0' ||
262 strcmp(optarg + 1, "yes") == 0 ||
263 strcmp(optarg + 1, "true") == 0){
266 } else if (*optarg && strcmp(optarg + 1, "maybe") == 0) {
272 return ARG_ERR_BAD_ARG;
280 getarg(struct getargs *args, size_t num_args,
281 int argc, char **argv, int *optind)
288 for(i = *optind; i < argc; i++) {
289 if(argv[i][0] != '-')
291 if(argv[i][1] == '-'){
296 ret = arg_match_long (args, num_args, argv[i] + 2);
300 for(j = 1; argv[i][j]; j++) {
301 for(k = 0; k < num_args; k++) {
303 if(args[k].short_name == 0)
305 if(argv[i][j] == args[k].short_name){
306 if(args[k].type == arg_flag){
307 *(int*)args[k].value = 1;
310 if(args[k].type == arg_negative_flag){
311 *(int*)args[k].value = 0;
315 optarg = &argv[i][j + 1];
321 return ARG_ERR_NO_ARG;
322 if(args[k].type == arg_integer){
324 if(sscanf(optarg, "%d", &tmp) != 1)
325 return ARG_ERR_BAD_ARG;
326 *(int*)args[k].value = tmp;
328 }else if(args[k].type == arg_string){
329 *(char**)args[k].value = optarg;
331 }else if(args[k].type == arg_strings){
332 add_string((getarg_strings*)args[k].value, optarg);
335 return ARG_ERR_BAD_ARG;
340 return ARG_ERR_NO_MATCH;
356 struct getargs args[] = {
357 { NULL, '1', arg_flag, &flag1, "one", NULL },
358 { NULL, '2', arg_flag, &flag2, "two", NULL },
359 { "foo", 'f', arg_negative_flag, &foo_flag, "foo", NULL },
360 { "bar", 'b', arg_integer, &bar_int, "bar", "seconds"},
361 { "baz", 'x', arg_string, &baz_string, "baz", "name" },
364 int main(int argc, char **argv)
367 while(getarg(args, 5, argc, argv, &optind))
368 printf("Bad arg: %s\n", argv[optind]);
369 printf("flag1 = %d\n", flag1);
370 printf("flag2 = %d\n", flag2);
371 printf("foo_flag = %d\n", foo_flag);
372 printf("bar_int = %d\n", bar_int);
373 printf("baz_flag = %s\n", baz_string);
374 arg_printusage (args, 5, argv[0], "nothing here");