ucw/conf-getopt.c

   1 /*
   2  *      UCW Library -- Configuration files: getopt wrapper
   3  *
   4  *      (c) 2001--2006 Robert Spalek <robert@ucw.cz>
   5  *      (c) 2003--2012 Martin Mares <mj@ucw.cz>
   6  *
   7  *      This software may be freely distributed and used according to the terms
   8  *      of the GNU Lesser General Public License.
   9  */
  10
  11 #include <ucw/lib.h>
  12 #include <ucw/conf.h>
  13 #include <ucw/conf-internal.h>
  14 #include <ucw/getopt.h>
  15 #include <ucw/fastbuf.h>
  16
  17 #include <stdlib.h>
  18
  19 #ifndef CONFIG_UCW_DEFAULT_CONFIG
  20 #define CONFIG_UCW_DEFAULT_CONFIG NULL
  21 #endif
  22 char *cf_def_file = CONFIG_UCW_DEFAULT_CONFIG;
  23
  24 #ifndef CONFIG_UCW_ENV_VAR_CONFIG
  25 #define CONFIG_UCW_ENV_VAR_CONFIG NULL
  26 #endif
  27 char *cf_env_file = CONFIG_UCW_ENV_VAR_CONFIG;
  28
  29 void
  30 cf_load_default(struct cf_context *cc)
  31 {
  32   if (cc->config_loaded++)
  33     return;
  34   if (cf_def_file)
  35     {
  36       char *env;
  37       if (cf_env_file && (env = getenv(cf_env_file)))
  38         {
  39           if (cf_load(env))
  40             die("Cannot load config file %s", env);
  41         }
  42       else if (cf_load(cf_def_file))
  43         die("Cannot load default config %s", cf_def_file);
  44     }
  45   else
  46     {
  47       // We need to create an empty pool and initialize all configuration items
  48       struct cf_journal_item *oldj = cf_journal_new_transaction(1);
  49       cf_init_stack(cc);
  50       cf_done_stack(cc);
  51       cf_journal_commit_transaction(1, oldj);
  52     }
  53 }
  54
  55 static void
  56 end_of_options(struct cf_context *cc)
  57 {
  58   cf_load_default(cc);
  59   if (cc->postpone_commit && cf_close_group())
  60     die("Loading of configuration failed");
  61 }
  62
  63 int
  64 cf_getopt(int argc, char *const argv[], const char *short_opts, const struct option *long_opts, int *long_index)
  65 {
  66   struct cf_context *cc = cf_get_context();
  67   if (!cc->postpone_commit)
  68     cf_open_group();
  69
  70   while (1)
  71     {
  72       int res = getopt_long(argc, argv, short_opts, long_opts, long_index);
  73       if (res == 'S' || res == 'C' || res == 0x64436667)
  74         {
  75           if (cc->other_options)
  76             die("The -S and -C options must precede all other arguments");
  77           if (res == 'S')
  78             {
  79               cf_load_default(cc);
  80               if (cf_set(optarg))
  81                 die("Cannot set %s", optarg);
  82             }
  83           else if (res == 'C')
  84             {
  85               if (cf_load(optarg))
  86                 die("Cannot load config file %s", optarg);
  87             }
  88 #ifdef CONFIG_UCW_DEBUG
  89           else
  90             {                   /* --dumpconfig */
  91               end_of_options(cc);
  92               struct fastbuf *b = bfdopen(1, 4096);
  93               cf_dump_sections(b);
  94               bclose(b);
  95               exit(0);
  96             }
  97 #endif
  98         }
  99       else
 100         {
 101           /* unhandled option or end of options */
 102           if (res != ':' && res != '?')
 103             end_of_options(cc);
 104           cc->other_options++;
 105           return res;
 106         }
 107     }
 108 }