You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
2.5 KiB
109 lines
2.5 KiB
3 months ago
|
/*
|
||
|
* UCW Library -- Configuration files: getopt wrapper
|
||
|
*
|
||
|
* (c) 2001--2006 Robert Spalek <robert@ucw.cz>
|
||
|
* (c) 2003--2012 Martin Mares <mj@ucw.cz>
|
||
|
*
|
||
|
* This software may be freely distributed and used according to the terms
|
||
|
* of the GNU Lesser General Public License.
|
||
|
*/
|
||
|
|
||
|
#include <ucw/lib.h>
|
||
|
#include <ucw/conf.h>
|
||
|
#include <ucw/conf-internal.h>
|
||
|
#include <ucw/getopt.h>
|
||
|
#include <ucw/fastbuf.h>
|
||
|
|
||
|
#include <stdlib.h>
|
||
|
|
||
|
#ifndef CONFIG_UCW_DEFAULT_CONFIG
|
||
|
#define CONFIG_UCW_DEFAULT_CONFIG NULL
|
||
|
#endif
|
||
|
char *cf_def_file = CONFIG_UCW_DEFAULT_CONFIG;
|
||
|
|
||
|
#ifndef CONFIG_UCW_ENV_VAR_CONFIG
|
||
|
#define CONFIG_UCW_ENV_VAR_CONFIG NULL
|
||
|
#endif
|
||
|
char *cf_env_file = CONFIG_UCW_ENV_VAR_CONFIG;
|
||
|
|
||
|
void
|
||
|
cf_load_default(struct cf_context *cc)
|
||
|
{
|
||
|
if (cc->config_loaded++)
|
||
|
return;
|
||
|
if (cf_def_file)
|
||
|
{
|
||
|
char *env;
|
||
|
if (cf_env_file && (env = getenv(cf_env_file)))
|
||
|
{
|
||
|
if (cf_load(env))
|
||
|
die("Cannot load config file %s", env);
|
||
|
}
|
||
|
else if (cf_load(cf_def_file))
|
||
|
die("Cannot load default config %s", cf_def_file);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// We need to create an empty pool and initialize all configuration items
|
||
|
struct cf_journal_item *oldj = cf_journal_new_transaction(1);
|
||
|
cf_init_stack(cc);
|
||
|
cf_done_stack(cc);
|
||
|
cf_journal_commit_transaction(1, oldj);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
end_of_options(struct cf_context *cc)
|
||
|
{
|
||
|
cf_load_default(cc);
|
||
|
if (cc->postpone_commit && cf_close_group())
|
||
|
die("Loading of configuration failed");
|
||
|
}
|
||
|
|
||
|
int
|
||
|
cf_getopt(int argc, char *const argv[], const char *short_opts, const struct option *long_opts, int *long_index)
|
||
|
{
|
||
|
struct cf_context *cc = cf_get_context();
|
||
|
if (!cc->postpone_commit)
|
||
|
cf_open_group();
|
||
|
|
||
|
while (1)
|
||
|
{
|
||
|
int res = getopt_long(argc, argv, short_opts, long_opts, long_index);
|
||
|
if (res == 'S' || res == 'C' || res == 0x64436667)
|
||
|
{
|
||
|
if (cc->other_options)
|
||
|
die("The -S and -C options must precede all other arguments");
|
||
|
if (res == 'S')
|
||
|
{
|
||
|
cf_load_default(cc);
|
||
|
if (cf_set(optarg))
|
||
|
die("Cannot set %s", optarg);
|
||
|
}
|
||
|
else if (res == 'C')
|
||
|
{
|
||
|
if (cf_load(optarg))
|
||
|
die("Cannot load config file %s", optarg);
|
||
|
}
|
||
|
#ifdef CONFIG_UCW_DEBUG
|
||
|
else
|
||
|
{ /* --dumpconfig */
|
||
|
end_of_options(cc);
|
||
|
struct fastbuf *b = bfdopen(1, 4096);
|
||
|
cf_dump_sections(b);
|
||
|
bclose(b);
|
||
|
exit(0);
|
||
|
}
|
||
|
#endif
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
/* unhandled option or end of options */
|
||
|
if (res != ':' && res != '?')
|
||
|
end_of_options(cc);
|
||
|
cc->other_options++;
|
||
|
return res;
|
||
|
}
|
||
|
}
|
||
|
}
|