1    | /*
2    |  * Copyright (c) 1987, 1993, 1994
3    |  *      The Regents of the University of California.  All rights reserved.
4    |  *
5    |  * Redistribution and use in source and binary forms, with or without
6    |  * modification, are permitted provided that the following conditions
7    |  * are met:
8    |  * 1. Redistributions of source code must retain the above copyright
9    |  *    notice, this list of conditions and the following disclaimer.
10   |  * 2. Redistributions in binary form must reproduce the above copyright
11   |  *    notice, this list of conditions and the following disclaimer in the
12   |  *    documentation and/or other materials provided with the distribution.
13   |  * 3. All advertising materials mentioning features or use of this software
14   |  *    must display the following acknowledgement:
15   |  *      This product includes software developed by the University of
16   |  *      California, Berkeley and its contributors.
17   |  * 4. 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   | 
34   | /* This code has been modified from the original UC version by 
35   |    Marek Bukowy, RIPE NCC
36   |    to be thread-safe
37   | */
38   | 
39   | #include <stdio.h>
40   | #include <stdlib.h>
41   | #include <string.h>
42   | 
43   | #include "getopt.h"
44   | 
45   | getopt_state_t *
46   | mg_new(int optind)
47   | {
48   |   getopt_state_t *state;
49   | 
50   |   if( (state=calloc(1, sizeof(getopt_state_t))) != NULL ) {
51   |     state->place = EMSG;
52   |     state->optind = optind;
53   |     state->optreset = 1;
54   |   }
55   |   return state;
56   | }
57   | 
58   | 
59   | /*
60   |  * getopt --
61   |  *      Parse argc/argv argument vector.
62   |  */
63   | int
64   | mg_getopt( int            nargc,
65   | 	   char * const   *nargv,
66   | 	   const char     *ostr,
67   | 	   getopt_state_t *state)
68   | {
69   |   char *oli;                              /* option letter list index */
70   | #ifdef __linux__
71   |   int optopt;
72   | #endif    
73   | 
74   |   if (state->optreset || !*state->place) {              /* update scanning pointer */
75   |     state->optreset = 0;
76   | 
77   |     /* handle wrongly tokenised space (grrrrr...) - skip empty fields */
78   |     while(state->optind < nargc &&  *(nargv[state->optind]) == '\0') {
79   |       state->optind++;
80   |     }
81   |     if (state->optind >= nargc || *(state->place = nargv[state->optind]) != '-') {
82   |       state->place = EMSG;
83   |       return (-1);
84   |     }
85   |     if (state->place[1] && *++(state->place) == '-') {      /* found "--" */
86   |       ++state->optind;
87   |       state->place = EMSG;
88   |       return (-1);
89   |     }
90   |   }                                       /* option letter okay? */
91   |   if ((state->optopt = (int)*state->place++) == (int)':' ||
92   |       !(oli = strchr(ostr, state->optopt))) {
93   |     /*
94   |      * if the user didn't specify '-' as an option,
95   |      * assume it means -1.
96   |      */
97   |     if (state->optopt == (int)'-')
98   |       return (-1);
99   |     if (!*state->place)
100  |       ++state->optind;
101  |     return (BADCH);
102  |   }
103  |   if (*++oli != ':') {                    /* don't need argument */
104  |     state->optarg = NULL;
105  |     if (!*state->place)
106  |       ++state->optind;
107  |   }
108  |   else {                                  /* need an argument */
109  |     if (*state->place)                     /* no white space */
110  |       state->optarg = state->place;
111  |     else if (nargc <= ++state->optind) {   /* no arg */
112  |       state->place = EMSG;
113  |       if (*ostr == ':')
114  | 	return (BADARG);
115  |       
116  |       return (BADCH);
117  |     }
118  |     else                            /* white space */
119  |       state->optarg = nargv[state->optind];
120  |     state->place = EMSG;
121  |     ++state->optind;
122  |   }
123  |   return (state->optopt);                        /* dump back option letter */
124  | }