/*
Copyright (C) 2010 ABRT team
Copyright (C) 2010 RedHat Inc
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include <getopt.h>
#include "internal_libreport.h"
#define USAGE_OPTS_WIDTH 30
#define USAGE_GAP 2
const char *g_progname;
const char *abrt_init(char **argv)
{
if (!load_global_configuration())
error_msg_and_die("Cannot continue without libreport global configuration.");
char *env_verbose = getenv("ABRT_VERBOSE");
if (env_verbose)
g_verbose = atoi(env_verbose);
g_progname = strrchr(argv[0], '/');
if (g_progname)
g_progname++;
else
g_progname = argv[0];
char *pfx = getenv("ABRT_PROG_PREFIX");
if (pfx && string_to_bool(pfx))
msg_prefix = g_progname;
return g_progname;
}
void export_abrt_envvars(int pfx)
{
putenv(xasprintf("ABRT_VERBOSE=%u", g_verbose));
if (pfx)
{
putenv((char*)"ABRT_PROG_PREFIX=1");
msg_prefix = g_progname;
}
}
void show_usage_and_die(const char *usage, const struct options *opt)
{
INITIALIZE_LIBREPORT();
fputs(_("Usage: "), stderr);
while (*usage)
{
int len = strchrnul(usage, '&') - usage;
if (len > 0)
{
fprintf(stderr, "%.*s", len, usage);
usage += len;
}
if (*usage == '&')
{
/* Only '&' *followed by whitespace* is substituted
* with program name - all other &'s are printed literally.
*/
if (usage[1] == '\0' || isspace(usage[1]))
fputs(g_progname, stderr);
else
fputc('&', stderr);
usage++;
}
}
fputs("\n\n", stderr);
for (; opt->type != OPTION_END; opt++)
{
size_t pos;
int pad;
if (opt->type == OPTION_GROUP)
{
fputc('\n', stderr);
if (*opt->help)
fprintf(stderr, "%s\n", opt->help);
continue;
}
pos = fprintf(stderr, " ");
if (opt->short_name)
pos += fprintf(stderr, "-%c", opt->short_name);
if (opt->short_name && opt->long_name)
pos += fprintf(stderr, ", ");
if (opt->long_name)
pos += fprintf(stderr, "--%s", opt->long_name);
if (opt->argh)
{
const char *fmt = " %s";
if (opt->type == OPTION_OPTSTRING)
fmt = "[%s]";
pos += fprintf(stderr, fmt, opt->argh);
}
if (pos <= USAGE_OPTS_WIDTH)
pad = USAGE_OPTS_WIDTH - pos;
else
{
fputc('\n', stderr);
pad = USAGE_OPTS_WIDTH;
}
fprintf(stderr, "%*s%s\n", pad + USAGE_GAP, "", opt->help);
}
fputc('\n', stderr);
xfunc_die();
}
static int parse_opt_size(const struct options *opt)
{
unsigned size = 0;
for (; opt->type != OPTION_END; opt++)
size++;
return size;
}
unsigned parse_opts(int argc, char **argv, const struct options *opt,
const char *usage)
{
int help = 0;
int size = parse_opt_size(opt);
const int LONGOPT_OFFSET = 256;
struct strbuf *shortopts = strbuf_new();
struct option *longopts = xzalloc(sizeof(longopts[0]) * (size+2));
struct option *curopt = longopts;
int ii;
for (ii = 0; ii < size; ++ii)
{
curopt->name = opt[ii].long_name;
/*curopt->flag = 0; - xzalloc did it */
if (opt[ii].short_name)
curopt->val = opt[ii].short_name;
else
curopt->val = LONGOPT_OFFSET + ii;
switch (opt[ii].type)
{
case OPTION_BOOL:
curopt->has_arg = no_argument;
if (opt[ii].short_name)
strbuf_append_char(shortopts, opt[ii].short_name);
break;
case OPTION_INTEGER:
case OPTION_STRING:
case OPTION_LIST:
curopt->has_arg = required_argument;
if (opt[ii].short_name)
strbuf_append_strf(shortopts, "%c:", opt[ii].short_name);
break;
case OPTION_OPTSTRING:
curopt->has_arg = optional_argument;
if (opt[ii].short_name)
strbuf_append_strf(shortopts, "%c::", opt[ii].short_name);
break;
case OPTION_GROUP:
case OPTION_END:
break;
}
//log_warning("curopt[%d].name:'%s' .has_arg:%d .flag:%p .val:%d", (int)(curopt-longopts),
// curopt->name, curopt->has_arg, curopt->flag, curopt->val);
/*
* getopt_long() thinks that NULL name marks the end of longopts.
* Example:
* [0] name:'verbose' val:'v'
* [1] name:NULL val:'c'
* [2] name:'force' val:'f'
* ... ... ...
* In this case, --force won't be accepted!
* Therefore we can only advance if name is not NULL.
*/
if (curopt->name)
curopt++;
}
curopt->name = "help";
curopt->has_arg = no_argument;
curopt->flag = &help;
curopt->val = 1;
/* xzalloc did it already:
curopt++;
curopt->name = NULL;
curopt->has_arg = 0;
curopt->flag = NULL;
curopt->val = 0;
*/
unsigned retval = 0;
while (1)
{
int c = getopt_long(argc, argv, shortopts->buf, longopts, NULL);
if (c == -1)
break;
if (c == '?' || help)
{
free(longopts);
strbuf_free(shortopts);
xfunc_error_retval = 0; /* this isn't error, exit code = 0 */
show_usage_and_die(usage, opt);
}
for (ii = 0; ii < size; ++ii)
{
if (opt[ii].short_name == c || LONGOPT_OFFSET + ii == c)
{
if (ii < sizeof(retval)*8)
retval |= (1 << ii);
if (opt[ii].value != NULL) switch (opt[ii].type)
{
case OPTION_BOOL:
*(int*)(opt[ii].value) += 1;
break;
case OPTION_INTEGER:
*(int*)(opt[ii].value) = xatoi(optarg);
break;
case OPTION_STRING:
case OPTION_OPTSTRING:
if (optarg)
*(char**)(opt[ii].value) = (char*)optarg;
break;
case OPTION_LIST:
*(GList**)(opt[ii].value) = g_list_append(*(GList**)(opt[ii].value), optarg);
break;
case OPTION_GROUP:
case OPTION_END:
break;
}
}
}
}
free(longopts);
strbuf_free(shortopts);
return retval;
}