1/*	$NetBSD$	*/
2
3
4/*
5 * \file sort.c
6 *
7 * Time-stamp:      "2010-07-17 10:34:15 bkorb"
8 *
9 *  This module implements argument sorting.
10 *
11 *  This file is part of AutoOpts, a companion to AutoGen.
12 *  AutoOpts is free software.
13 *  AutoOpts is Copyright (c) 1992-2011 by Bruce Korb - all rights reserved
14 *
15 *  AutoOpts is available under any one of two licenses.  The license
16 *  in use must be one of these two and the choice is under the control
17 *  of the user of the license.
18 *
19 *   The GNU Lesser General Public License, version 3 or later
20 *      See the files "COPYING.lgplv3" and "COPYING.gplv3"
21 *
22 *   The Modified Berkeley Software Distribution License
23 *      See the file "COPYING.mbsd"
24 *
25 *  These files have the following md5sums:
26 *
27 *  43b91e8ca915626ed3818ffb1b71248b pkg/libopts/COPYING.gplv3
28 *  06a1a2e4760c90ea5e1dad8dfaac4d39 pkg/libopts/COPYING.lgplv3
29 *  66a5cedaf62c4b2637025f049f9b826f pkg/libopts/COPYING.mbsd
30 */
31
32/* = = = START-STATIC-FORWARD = = = */
33static tSuccess
34mustHandleArg(tOptions* pOpts, char* pzArg, tOptState* pOS,
35              char** ppzOpts, int* pOptsIdx);
36
37static tSuccess
38mayHandleArg(tOptions* pOpts, char* pzArg, tOptState* pOS,
39             char** ppzOpts, int* pOptsIdx);
40
41static tSuccess
42checkShortOpts(tOptions* pOpts, char* pzArg, tOptState* pOS,
43               char** ppzOpts, int* pOptsIdx);
44/* = = = END-STATIC-FORWARD = = = */
45
46/*
47 *  "mustHandleArg" and "mayHandleArg" are really similar.  The biggest
48 *  difference is that "may" will consume the next argument only if it
49 *  does not start with a hyphen and "must" will consume it, hyphen or not.
50 */
51static tSuccess
52mustHandleArg(tOptions* pOpts, char* pzArg, tOptState* pOS,
53              char** ppzOpts, int* pOptsIdx)
54{
55    /*
56     *  An option argument is required.  Long options can either have
57     *  a separate command line argument, or an argument attached by
58     *  the '=' character.  Figure out which.
59     */
60    switch (pOS->optType) {
61    case TOPT_SHORT:
62        /*
63         *  See if an arg string follows the flag character.  If not,
64         *  the next arg must be the option argument.
65         */
66        if (*pzArg != NUL)
67            return SUCCESS;
68        break;
69
70    case TOPT_LONG:
71        /*
72         *  See if an arg string has already been assigned (glued on
73         *  with an `=' character).  If not, the next is the opt arg.
74         */
75        if (pOS->pzOptArg != NULL)
76            return SUCCESS;
77        break;
78
79    default:
80        return FAILURE;
81    }
82    if (pOpts->curOptIdx >= pOpts->origArgCt)
83        return FAILURE;
84
85    ppzOpts[ (*pOptsIdx)++ ] = pOpts->origArgVect[ (pOpts->curOptIdx)++ ];
86    return SUCCESS;
87}
88
89static tSuccess
90mayHandleArg(tOptions* pOpts, char* pzArg, tOptState* pOS,
91             char** ppzOpts, int* pOptsIdx)
92{
93    /*
94     *  An option argument is optional.
95     */
96    switch (pOS->optType) {
97    case TOPT_SHORT:
98        /*
99         *  IF nothing is glued on after the current flag character,
100         *  THEN see if there is another argument.  If so and if it
101         *  does *NOT* start with a hyphen, then it is the option arg.
102         */
103        if (*pzArg != NUL)
104            return SUCCESS;
105        break;
106
107    case TOPT_LONG:
108        /*
109         *  Look for an argument if we don't already have one (glued on
110         *  with a `=' character)
111         */
112        if (pOS->pzOptArg != NULL)
113            return SUCCESS;
114        break;
115
116    default:
117        return FAILURE;
118    }
119    if (pOpts->curOptIdx >= pOpts->origArgCt)
120        return PROBLEM;
121
122    pzArg = pOpts->origArgVect[ pOpts->curOptIdx ];
123    if (*pzArg != '-')
124        ppzOpts[ (*pOptsIdx)++ ] = pOpts->origArgVect[ (pOpts->curOptIdx)++ ];
125    return SUCCESS;
126}
127
128/*
129 *  Process a string of short options glued together.  If the last one
130 *  does or may take an argument, the do the argument processing and leave.
131 */
132static tSuccess
133checkShortOpts(tOptions* pOpts, char* pzArg, tOptState* pOS,
134               char** ppzOpts, int* pOptsIdx)
135{
136    while (*pzArg != NUL) {
137        if (FAILED(shortOptionFind(pOpts, (tAoUC)*pzArg, pOS)))
138            return FAILURE;
139
140        /*
141         *  See if we can have an arg.
142         */
143        if (OPTST_GET_ARGTYPE(pOS->pOD->fOptState) == OPARG_TYPE_NONE) {
144            pzArg++;
145
146        } else if (pOS->pOD->fOptState & OPTST_ARG_OPTIONAL) {
147            /*
148             *  Take an argument if it is not attached and it does not
149             *  start with a hyphen.
150             */
151            if (pzArg[1] != NUL)
152                return SUCCESS;
153
154            pzArg = pOpts->origArgVect[ pOpts->curOptIdx ];
155            if (*pzArg != '-')
156                ppzOpts[ (*pOptsIdx)++ ] =
157                    pOpts->origArgVect[ (pOpts->curOptIdx)++ ];
158            return SUCCESS;
159
160        } else {
161            /*
162             *  IF we need another argument, be sure it is there and
163             *  take it.
164             */
165            if (pzArg[1] == NUL) {
166                if (pOpts->curOptIdx >= pOpts->origArgCt)
167                    return FAILURE;
168                ppzOpts[ (*pOptsIdx)++ ] =
169                    pOpts->origArgVect[ (pOpts->curOptIdx)++ ];
170            }
171            return SUCCESS;
172        }
173    }
174    return SUCCESS;
175}
176
177/*
178 *  If the program wants sorted options (separated operands and options),
179 *  then this routine will to the trick.
180 */
181LOCAL void
182optionSort(tOptions* pOpts)
183{
184    char** ppzOpts;
185    char** ppzOpds;
186    int    optsIdx = 0;
187    int    opdsIdx = 0;
188
189    tOptState os = OPTSTATE_INITIALIZER(DEFINED);
190
191    /*
192     *  Disable for POSIX conformance, or if there are no operands.
193     */
194    if (  (getenv("POSIXLY_CORRECT") != NULL)
195       || NAMED_OPTS(pOpts))
196        return;
197
198    /*
199     *  Make sure we can allocate two full-sized arg vectors.
200     */
201    ppzOpts = malloc(pOpts->origArgCt * sizeof(char*));
202    if (ppzOpts == NULL)
203        goto exit_no_mem;
204
205    ppzOpds = malloc(pOpts->origArgCt * sizeof(char*));
206    if (ppzOpds == NULL) {
207        free(ppzOpts);
208        goto exit_no_mem;
209    }
210
211    pOpts->curOptIdx = 1;
212    pOpts->pzCurOpt  = NULL;
213
214    /*
215     *  Now, process all the options from our current position onward.
216     *  (This allows interspersed options and arguments for the few
217     *  non-standard programs that require it.)
218     */
219    for (;;) {
220        char* pzArg;
221        tSuccess res;
222
223        /*
224         *  If we're out of arguments, we're done.  Join the option and
225         *  operand lists into the original argument vector.
226         */
227        if (pOpts->curOptIdx >= pOpts->origArgCt) {
228            errno = 0;
229            goto joinLists;
230        }
231
232        pzArg = pOpts->origArgVect[ pOpts->curOptIdx ];
233        if (*pzArg != '-') {
234            ppzOpds[ opdsIdx++ ] = pOpts->origArgVect[ (pOpts->curOptIdx)++ ];
235            continue;
236        }
237
238        switch (pzArg[1]) {
239        case NUL:
240            /*
241             *  A single hyphen is an operand.
242             */
243            ppzOpds[ opdsIdx++ ] = pOpts->origArgVect[ (pOpts->curOptIdx)++ ];
244            continue;
245
246        case '-':
247            /*
248             *  Two consecutive hypens.  Put them on the options list and then
249             *  _always_ force the remainder of the arguments to be operands.
250             */
251            if (pzArg[2] == NUL) {
252                ppzOpts[ optsIdx++ ] =
253                    pOpts->origArgVect[ (pOpts->curOptIdx)++ ];
254                goto restOperands;
255            }
256            res = longOptionFind(pOpts, pzArg+2, &os);
257            break;
258
259        default:
260            /*
261             *  If short options are not allowed, then do long
262             *  option processing.  Otherwise the character must be a
263             *  short (i.e. single character) option.
264             */
265            if ((pOpts->fOptSet & OPTPROC_SHORTOPT) == 0) {
266                res = longOptionFind(pOpts, pzArg+1, &os);
267            } else {
268                res = shortOptionFind(pOpts, (tAoUC)pzArg[1], &os);
269            }
270            break;
271        }
272        if (FAILED(res)) {
273            errno = EINVAL;
274            goto freeTemps;
275        }
276
277        /*
278         *  We've found an option.  Add the argument to the option list.
279         *  Next, we have to see if we need to pull another argument to be
280         *  used as the option argument.
281         */
282        ppzOpts[ optsIdx++ ] = pOpts->origArgVect[ (pOpts->curOptIdx)++ ];
283
284        if (OPTST_GET_ARGTYPE(os.pOD->fOptState) == OPARG_TYPE_NONE) {
285            /*
286             *  No option argument.  If we have a short option here,
287             *  then scan for short options until we get to the end
288             *  of the argument string.
289             */
290            if (  (os.optType == TOPT_SHORT)
291               && FAILED(checkShortOpts(pOpts, pzArg+2, &os, ppzOpts,
292                                        &optsIdx)) )  {
293                errno = EINVAL;
294                goto freeTemps;
295            }
296
297        } else if (os.pOD->fOptState & OPTST_ARG_OPTIONAL) {
298            switch (mayHandleArg(pOpts, pzArg+2, &os, ppzOpts, &optsIdx)) {
299            case FAILURE: errno = EIO; goto freeTemps;
300            case PROBLEM: errno = 0;   goto joinLists;
301            }
302
303        } else {
304            switch (mustHandleArg(pOpts, pzArg+2, &os, ppzOpts, &optsIdx)) {
305            case PROBLEM:
306            case FAILURE: errno = EIO; goto freeTemps;
307            }
308        }
309    } /* for (;;) */
310
311 restOperands:
312    while (pOpts->curOptIdx < pOpts->origArgCt)
313        ppzOpds[ opdsIdx++ ] = pOpts->origArgVect[ (pOpts->curOptIdx)++ ];
314
315 joinLists:
316    if (optsIdx > 0)
317        memcpy(pOpts->origArgVect + 1, ppzOpts, optsIdx * sizeof(char*));
318    if (opdsIdx > 0)
319        memcpy(pOpts->origArgVect + 1 + optsIdx, ppzOpds,
320               opdsIdx * sizeof(char*));
321
322 freeTemps:
323    free(ppzOpts);
324    free(ppzOpds);
325    return;
326
327 exit_no_mem:
328    errno = ENOMEM;
329    return;
330}
331
332/*
333 * Local Variables:
334 * mode: C
335 * c-file-style: "stroustrup"
336 * indent-tabs-mode: nil
337 * End:
338 * end of autoopts/sort.c */
339