libbb/getopt_ulflags.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * universal getopt_ulflags implementation for busybox
   4  *
   5  * Copyright (C) 2003  Vladimir Oleynik  <dzo@simtreas.ru>
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15  * General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  20  *
  21  */
  22
  23 #include <getopt.h>
  24 #include <string.h>
  25 #include <assert.h>
  26 #include <stdlib.h>
  27 #include "libbb.h"
  28
  29 /*
  30 You can set bb_opt_complementaly as string with one or more
  31 complementaly or incongruously options.
  32 If sequential founded option haved from this string
  33 then your incongruously pairs unsets and complementaly make add sets.
  34 Format:
  35 one char - option for check,
  36 chars - complementaly option for add sets.
  37 - chars - option triggered for unsets.
  38 ~ chars - option incongruously.
  39 *       - option list, called add_to_list(*ptr_from_usaged, optarg)
  40 :       - separator.
  41 Example: du applet can have options "-s" and "-d size"
  42 If getopt found -s then -d option flag unset or if found -d then -s unset.
  43 For this result you must set bb_opt_complementaly = "s-d:d-s".
  44 Result have last option flag only from called arguments.
  45 Warning! You can check returned flag, pointer to "d:" argument seted
  46 to own optarg always.
  47 Example two: cut applet must only one type of list may be specified,
  48 and -b, -c and -f incongruously option, overwited option is error also.
  49 You must set bb_opt_complementaly = "b~bcf:c~bcf:f~bcf".
  50 If called have more one specified, return value have error flag -
  51 high bite set (0x80000000UL).
  52 Example three: grep applet can have one or more "-e pattern" arguments.
  53 You should use bb_getopt_ulflags() as
  54 llist_t *paterns;
  55 bb_opt_complementaly = "e*";
  56 bb_getopt_ulflags (argc, argv, "e:", &paterns);
  57 */
  58
  59 const char *bb_opt_complementaly;
  60
  61 typedef struct
  62 {
  63         char opt;
  64         char list_flg;
  65         unsigned long switch_on;
  66         unsigned long switch_off;
  67         unsigned long incongruously;
  68         void **optarg;              /* char **optarg or llist_t **optarg */
  69 } t_complementaly;
  70
  71 /* You can set bb_applet_long_options for parse called long options */
  72
  73 static const struct option bb_default_long_options[] = {
  74      /*   { "help", 0, NULL, '?' }, */
  75         { 0, 0, 0, 0 }
  76 };
  77
  78 const struct option *bb_applet_long_options = bb_default_long_options;
  79
  80
  81 unsigned long
  82 bb_getopt_ulflags (int argc, char **argv, const char *applet_opts, ...)
  83 {
  84         unsigned long flags = 0;
  85   int c = 0;
  86         const char *s;
  87   t_complementaly *complementaly;
  88   t_complementaly *on_off;
  89   va_list p;
  90
  91   va_start (p, applet_opts);
  92
  93   /* skip GNU extension */
  94   s = applet_opts;
  95   if(*s == '+' || *s == '-')
  96         s++;
  97
  98   for (; *s; s++) {
  99         c++;
 100         while (s[1] == ':') {
 101             /* check GNU extension "o::" - optional arg */
 102             s++;
 103         }
 104   }
 105   complementaly = xcalloc (c + 1, sizeof (t_complementaly));
 106   c = 0;
 107   /* skip GNU extension */
 108   s = applet_opts;
 109   if(*s == '+' || *s == '-')
 110         s++;
 111
 112   for (; *s; s++) {
 113         complementaly->opt = *s;
 114         complementaly->switch_on |= (1 << c);
 115         c++;
 116         if (s[1] == ':') {
 117           complementaly->optarg = va_arg (p, void **);
 118           do
 119                 s++;
 120           while (s[1] == ':');
 121                 }
 122         complementaly++;
 123         }
 124   complementaly->opt = 0;
 125   complementaly -= c;
 126   c = 0;
 127   for (s = bb_opt_complementaly; s && *s; s++) {
 128     t_complementaly *pair;
 129
 130     if (*s == ':') {
 131           c = 0;
 132           continue;
 133     }
 134     if (c)
 135           continue;
 136     for (on_off = complementaly; on_off->opt; on_off++)
 137           if (on_off->opt == *s)
 138             break;
 139     pair = on_off;
 140     for(s++; *s && *s != ':'; s++) {
 141       if (*s == '-' || *s == '~') {
 142           c = *s;
 143       } else if(*s == '*') {
 144         pair->list_flg++;
 145       } else {
 146           unsigned long *pair_switch = &(pair->switch_on);
 147
 148           if(c)
 149             pair_switch = c == '-' ? &(pair->switch_off) : &(pair->incongruously);
 150           for (on_off = complementaly; on_off->opt; on_off++)
 151             if (on_off->opt == *s) {
 152                   *pair_switch |= on_off->switch_on;
 153                   break;
 154             }
 155       }
 156     }
 157     s--;
 158   }
 159
 160   while ((c = getopt_long (argc, argv, applet_opts,
 161                             bb_applet_long_options, NULL)) > 0) {
 162
 163         for (on_off = complementaly; on_off->opt != c; on_off++) {
 164             if(!on_off->opt)
 165                 bb_show_usage ();
 166         }
 167         if(flags & on_off->incongruously)
 168             flags |= 0x80000000UL;
 169         flags &= ~on_off->switch_off;
 170         flags |= on_off->switch_on;
 171         if(on_off->list_flg) {
 172            *(llist_t **)(on_off->optarg) =
 173                 llist_add_to(*(llist_t **)(on_off->optarg), optarg);
 174         } else if (on_off->optarg) {
 175             *(char **)(on_off->optarg) = optarg;
 176         }
 177   }
 178   free(complementaly);
 179         return flags;
 180 }