src/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. 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 #include <stdio.h>
  35 #include <stdlib.h>
  36 #include <string.h>
  37
  38
  39 int     opterror = 1,                   /* if error message should be printed */
  40         optindex = 1,                        /* index into parent argv vector */
  41         optopt,                             /* character checked for validity */
  42         optreset;                                             /* reset getopt */
  43 char    *optarg;                           /* argument associated with option */
  44
  45 #define BADCH   (int)'?'
  46 #define BADARG  (int)':'
  47 #define EMSG    ""
  48
  49 /*
  50  * getopt --
  51  *      Parse argc/argv argument vector.
  52  */
  53 int my_getopt(int nargc, char * const *nargv, const char *ostr)
  54 {
  55     /*        int nargc;
  56             char * const *nargv;
  57             const char *ostr; */
  58
  59     static char *place = EMSG;                    /* option letter processing */
  60     char *oli;                                    /* option letter list index */
  61
  62     if (optreset || !*place)
  63     {                                              /* update scanning pointer */
  64         optreset = 0;
  65         if (optindex >= nargc || *(place = nargv[optindex]) != '-')
  66         {
  67             place = EMSG;
  68             return (-1);
  69         }
  70         if (place[1] && *++place == '-')
  71         {      /* found "--" */
  72             ++optindex;
  73             place = EMSG;
  74             return (-1);
  75         }
  76     }                                                  /* option letter okay? */
  77     if ((optopt = (int)*place++) == (int)':' ||
  78             !(oli = strchr(ostr, optopt)))
  79     {
  80         /*
  81          * if the user didn't specify '-' as an option,
  82          * assume it means -1.
  83          */
  84         if (optopt == (int)'-')
  85             return (-1);
  86         if (!*place)
  87             ++optindex;
  88         if (opterror && *ostr != ':')
  89             (void)fprintf(stderr,
  90                           "ERROR  : unknown option -- %c\n", optopt);
  91         return (BADCH);
  92     }
  93     if (*++oli != ':')
  94     {                                                  /* don't need argument */
  95         optarg = NULL;
  96         if (!*place)
  97             ++optindex;
  98     }
  99     else
 100     {                                                     /* need an argument */
 101         if (*place)                                         /* no white space */
 102             optarg = place;
 103         else if (nargc <= ++optindex)
 104         {   /* no arg */
 105             place = EMSG;
 106             if (*ostr == ':')
 107                 return (BADARG);
 108             if (opterror)
 109                 (void)fprintf(stderr,
 110                               "ERROR  : option requires an argument -- %c\n",
 111                               optopt);
 112             exit (BADCH);
 113         }
 114         else                                                   /* white space */
 115             optarg = nargv[optindex];
 116         place = EMSG;
 117         ++optindex;
 118     }
 119     return (optopt);                               /* dump back option letter */
 120 }