usr/src/lib/libast/common/comp/getoptl.c

   1 /***********************************************************************
   2 *                                                                      *
   3 *               This software is part of the ast package               *
   4 *          Copyright (c) 1985-2010 AT&T Intellectual Property          *
   5 *                      and is licensed under the                       *
   6 *                  Common Public License, Version 1.0                  *
   7 *                    by AT&T Intellectual Property                     *
   8 *                                                                      *
   9 *                A copy of the License is available at                 *
  10 *            http://www.opensource.org/licenses/cpl1.0.txt             *
  11 *         (with md5 checksum 059e8cd6165cb4c31e351f2b69388fd9)         *
  12 *                                                                      *
  13 *              Information and Software Systems Research               *
  14 *                            AT&T Research                             *
  15 *                           Florham Park NJ                            *
  16 *                                                                      *
  17 *                 Glenn Fowler <gsf@research.att.com>                  *
  18 *                  David Korn <dgk@research.att.com>                   *
  19 *                   Phong Vo <kpv@research.att.com>                    *
  20 *                                                                      *
  21 ***********************************************************************/
  22 #pragma prototyped
  23
  24 #include <ast.h>
  25 #include <ast_getopt.h>
  26
  27 #undef  _BLD_ast        /* enable ast imports since we're user static */
  28
  29 #include <error.h>
  30 #include <option.h>
  31 #include <getopt.h>
  32 #include <ctype.h>
  33
  34 static const char*              lastoptstring;
  35 static const struct option*     lastlongopts;
  36 static char*                    usage;
  37 static Sfio_t*                  up;
  38
  39 static int                      lastoptind;
  40
  41 static int
  42 golly(int argc, char* const* argv, const char* optstring, const struct option* longopts, int* longindex, int flags)
  43 {
  44         register char*                  s;
  45         register const struct option*   o;
  46         register int                    c;
  47         char*                           t;
  48
  49         if (!up || optstring != lastoptstring || longopts != lastlongopts)
  50         {
  51                 if (!up && !(up = sfstropen()))
  52                         return -1;
  53                 sfprintf(up, "[-1p%d]", flags);
  54                 t = strdup(optstring);
  55                 for (o = longopts; o->name; o++)
  56                 {
  57                         if (o->flag || o->val <= 0 || o->val > UCHAR_MAX || !isalnum(o->val))
  58                                 sfprintf(up, "\n[%d:%s]", UCHAR_MAX + 1 + (o - longopts), o->name);
  59                         else
  60                         {
  61                                 sfprintf(up, "\n[%c:%s]", o->val, o->name);
  62                                 if (s = strchr(t, o->val))
  63                                 {
  64                                         *s++ = ' ';
  65                                         if (*s == ':')
  66                                         {
  67                                                 *s++ = ' ';
  68                                                 if (*s == ':')
  69                                                         *s = ' ';
  70                                         }
  71                                 }
  72                         }
  73                         if (o->has_arg)
  74                         {
  75                                 sfputc(up, ':');
  76                                 if (o->has_arg == optional_argument)
  77                                         sfputc(up, '?');
  78                                 sfprintf(up, "[string]");
  79                         }
  80                 }
  81                 s = t;
  82                 while (c = *s++)
  83                         if (c != ' ')
  84                         {
  85                                 sfprintf(up, "\n[%c]", c);
  86                                 if (*s == ':')
  87                                 {
  88                                         sfputc(up, *s);
  89                                         if (*++s == ':')
  90                                         {
  91                                                 sfputc(up, '?');
  92                                                 s++;
  93                                         }
  94                                         sfputc(up, '[');
  95                                         sfputc(up, ']');
  96                                 }
  97                         }
  98                 sfputc(up, '\n');
  99                 if (!(usage = sfstruse(up)))
 100                         return -1;
 101                 lastoptstring = optstring;
 102                 lastlongopts = longopts;
 103         }
 104         opt_info.index = (optind > 1 || optind == lastoptind) ? optind : 0;
 105         if (opt_info.index >= argc || !(c = optget((char**)argv, usage)))
 106         {
 107                 sfstrclose(up);
 108                 up = 0;
 109                 c = -1;
 110         }
 111         else
 112         {
 113                 if (c == ':' || c == '?')
 114                 {
 115                         if (opterr && (!optstring || *optstring != ':'))
 116                         {
 117                                 if (!error_info.id)
 118                                         error_info.id = argv[0];
 119                                 errormsg(NiL, c == '?' ? (ERROR_USAGE|4) : 2, "%s", opt_info.arg);
 120                         }
 121                         optopt = opt_info.option[1];
 122                         c = '?';
 123                 }
 124                 optarg = opt_info.arg;
 125                 if (c < 0)
 126                 {
 127                         o = longopts - c - UCHAR_MAX - 1;
 128                         if (o->flag)
 129                         {
 130                                 *o->flag = o->val;
 131                                 c = 0;
 132                         }
 133                         else
 134                                 c = o->val;
 135                 }
 136         }
 137         lastoptind = optind = opt_info.index;
 138         return c;
 139 }
 140
 141 extern int
 142 getopt_long(int argc, char* const* argv, const char* optstring, const struct option* longopts, int* longindex)
 143 {
 144         return golly(argc, argv, optstring, longopts, longindex, 2);
 145 }
 146
 147 extern int
 148 getopt_long_only(int argc, char* const* argv, const char* optstring, const struct option* longopts, int* longindex)
 149 {
 150         return golly(argc, argv, optstring, longopts, longindex, 1);
 151 }