compat/getopt.c

   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. Neither the name of the University nor the names of its contributors
  14  *    may be used to endorse or promote products derived from this software
  15  *    without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  */
  29
  30 /* OPENBSD ORIGINAL: lib/libc/stdlib/getopt.c */
  31
  32 #include "compat.h"
  33
  34 #include <stdio.h>
  35 #include <stdlib.h>
  36 #include <string.h>
  37
  38 int     BSDopterr = 1,          /* if error message should be printed */
  39         BSDoptind = 1,          /* index into parent argv vector */
  40         BSDoptopt,              /* character checked for validity */
  41         BSDoptreset;            /* reset getopt */
  42 char    *BSDoptarg;             /* argument associated with option */
  43
  44 #define BADCH   (int)'?'
  45 #define BADARG  (int)':'
  46 #define EMSG    ""
  47
  48 /*
  49  * getopt --
  50  *      Parse argc/argv argument vector.
  51  */
  52 int
  53 BSDgetopt(int nargc, char *const *nargv, const char *ostr)
  54 {
  55         static const char *place = EMSG;        /* option letter processing */
  56         char *oli;                              /* option letter list index */
  57
  58         if (ostr == NULL)
  59                 return (-1);
  60
  61         if (BSDoptreset || !*place) {           /* update scanning pointer */
  62                 BSDoptreset = 0;
  63                 if (BSDoptind >= nargc || *(place = nargv[BSDoptind]) != '-') {
  64                         place = EMSG;
  65                         return (-1);
  66                 }
  67                 if (place[1] && *++place == '-') {      /* found "--" */
  68                         if (place[1])
  69                                 return (BADCH);
  70                         ++BSDoptind;
  71                         place = EMSG;
  72                         return (-1);
  73                 }
  74         }                                       /* option letter okay? */
  75         if ((BSDoptopt = (int)*place++) == (int)':' ||
  76             !(oli = strchr(ostr, BSDoptopt))) {
  77                 /*
  78                  * if the user didn't specify '-' as an option,
  79                  * assume it means -1.
  80                  */
  81                 if (BSDoptopt == (int)'-')
  82                         return (-1);
  83                 if (!*place)
  84                         ++BSDoptind;
  85                 if (BSDopterr && *ostr != ':')
  86                         (void)fprintf(stderr,
  87                             "%s: unknown option -- %c\n", getprogname(),
  88                             BSDoptopt);
  89                 return (BADCH);
  90         }
  91         if (*++oli != ':') {                    /* don't need argument */
  92                 BSDoptarg = NULL;
  93                 if (!*place)
  94                         ++BSDoptind;
  95         }
  96         else {                                  /* need an argument */
  97                 if (*place)                     /* no white space */
  98                         BSDoptarg = (char *)place;
  99                 else if (nargc <= ++BSDoptind) {        /* no arg */
 100                         place = EMSG;
 101                         if (*ostr == ':')
 102                                 return (BADARG);
 103                         if (BSDopterr)
 104                                 (void)fprintf(stderr,
 105                                     "%s: option requires an argument -- %c\n",
 106                                     getprogname(), BSDoptopt);
 107                         return (BADCH);
 108                 }
 109                 else                            /* white space */
 110                         BSDoptarg = nargv[BSDoptind];
 111                 place = EMSG;
 112                 ++BSDoptind;
 113         }
 114         return (BSDoptopt);                     /* dump back option letter */
 115 }