302 lines
10 KiB
C
302 lines
10 KiB
C
|
/*
|
||
|
|
* getopt.c - my re-implementation of getopt.
|
||
|
|
* Copyright 1997, 2000, 2001, 2002, 2006, Benjamin Sittler
|
||
|
|
*
|
||
|
|
* Permission is hereby granted, free of charge, to any person
|
||
|
|
* obtaining a copy of this software and associated documentation
|
||
|
|
* files (the "Software"), to deal in the Software without
|
||
|
|
* restriction, including without limitation the rights to use, copy,
|
||
|
|
* modify, merge, publish, distribute, sublicense, and/or sell copies
|
||
|
|
* of the Software, and to permit persons to whom the Software is
|
||
|
|
* furnished to do so, subject to the following conditions:
|
||
|
|
*
|
||
|
|
* The above copyright notice and this permission notice shall be
|
||
|
|
* included in all copies or substantial portions of the Software.
|
||
|
|
*
|
||
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
||
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
||
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
||
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
||
|
|
* HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
||
|
|
* WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
||
|
|
* DEALINGS IN THE SOFTWARE.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#if HAVE_CONFIG_H
|
||
|
|
#include "clamav-config.h"
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include <sys/types.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include "getopt.h"
|
||
|
|
|
||
|
|
int optind = 1, opterr = 1, optopt = 0;
|
||
|
|
char *optarg = 0;
|
||
|
|
|
||
|
|
/* reset argument parser to start-up values */
|
||
|
|
/*
|
||
|
|
int getopt_reset(void)
|
||
|
|
{
|
||
|
|
optind = 1;
|
||
|
|
opterr = 1;
|
||
|
|
optopt = 0;
|
||
|
|
optarg = 0;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
*/
|
||
|
|
|
||
|
|
/* this is the plain old UNIX getopt, with GNU-style extensions. */
|
||
|
|
/* if you're porting some piece of UNIX software, this is all you need. */
|
||
|
|
/* this supports GNU-style permution and optional arguments */
|
||
|
|
|
||
|
|
int my_getopt(int argc, char *argvc[], const char *opts)
|
||
|
|
{
|
||
|
|
char **argv = (char **)argvc;
|
||
|
|
static int charind = 0;
|
||
|
|
const char *s;
|
||
|
|
char mode, colon_mode;
|
||
|
|
int off = 0, opt = -1;
|
||
|
|
|
||
|
|
if (getenv("POSIXLY_CORRECT"))
|
||
|
|
colon_mode = mode = '+';
|
||
|
|
else {
|
||
|
|
if ((colon_mode = *opts) == ':') off++;
|
||
|
|
if (((mode = opts[off]) == '+') || (mode == '-')) {
|
||
|
|
off++;
|
||
|
|
if ((colon_mode != ':') && ((colon_mode = opts[off]) == ':'))
|
||
|
|
off++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
optarg = 0;
|
||
|
|
if (charind) {
|
||
|
|
optopt = argv[optind][charind];
|
||
|
|
for (s = opts + off; *s; s++)
|
||
|
|
if (optopt == *s) {
|
||
|
|
charind++;
|
||
|
|
if ((*(++s) == ':') || ((optopt == 'W') && (*s == ';'))) {
|
||
|
|
if (argv[optind][charind]) {
|
||
|
|
optarg = &(argv[optind++][charind]);
|
||
|
|
charind = 0;
|
||
|
|
} else if (*(++s) != ':') {
|
||
|
|
charind = 0;
|
||
|
|
if (++optind >= argc) {
|
||
|
|
if (opterr) fprintf(stderr,
|
||
|
|
"%s: option requires an argument -- %c\n",
|
||
|
|
argv[0], optopt);
|
||
|
|
opt = (colon_mode == ':') ? ':' : '?';
|
||
|
|
goto getopt_ok;
|
||
|
|
}
|
||
|
|
optarg = argv[optind++];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
opt = optopt;
|
||
|
|
goto getopt_ok;
|
||
|
|
}
|
||
|
|
if (opterr) fprintf(stderr,
|
||
|
|
"%s: illegal option -- %c\n",
|
||
|
|
argv[0], optopt);
|
||
|
|
opt = '?';
|
||
|
|
if (argv[optind][++charind] == '\0') {
|
||
|
|
optind++;
|
||
|
|
charind = 0;
|
||
|
|
}
|
||
|
|
getopt_ok:
|
||
|
|
if (charind && !argv[optind][charind]) {
|
||
|
|
optind++;
|
||
|
|
charind = 0;
|
||
|
|
}
|
||
|
|
} else if ((optind >= argc) ||
|
||
|
|
((argv[optind][0] == '-') &&
|
||
|
|
(argv[optind][1] == '-') &&
|
||
|
|
(argv[optind][2] == '\0'))) {
|
||
|
|
optind++;
|
||
|
|
opt = -1;
|
||
|
|
} else if ((argv[optind][0] != '-') ||
|
||
|
|
(argv[optind][1] == '\0')) {
|
||
|
|
char *tmp;
|
||
|
|
int i, j, k;
|
||
|
|
|
||
|
|
if (mode == '+')
|
||
|
|
opt = -1;
|
||
|
|
else if (mode == '-') {
|
||
|
|
optarg = argv[optind++];
|
||
|
|
charind = 0;
|
||
|
|
opt = 1;
|
||
|
|
} else {
|
||
|
|
for (i = j = optind; i < argc; i++)
|
||
|
|
if ((argv[i][0] == '-') &&
|
||
|
|
(argv[i][1] != '\0')) {
|
||
|
|
optind = i;
|
||
|
|
opt = my_getopt(argc, argv, opts);
|
||
|
|
while (i > j) {
|
||
|
|
tmp = argv[--i];
|
||
|
|
for (k = i; k + 1 < optind; k++) argv[k] = argv[k + 1];
|
||
|
|
argv[--optind] = tmp;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
if (i == argc) opt = -1;
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
charind++;
|
||
|
|
opt = my_getopt(argc, argv, opts);
|
||
|
|
}
|
||
|
|
if (optind > argc) optind = argc;
|
||
|
|
return opt;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* this is the extended getopt_long{,_only}, with some GNU-like
|
||
|
|
* extensions. Implements _getopt_internal in case any programs
|
||
|
|
* expecting GNU libc getopt call it.
|
||
|
|
*/
|
||
|
|
|
||
|
|
static int _getopt_internal(int argc, char *argv[], const char *shortopts,
|
||
|
|
const struct option *longopts, int *longind,
|
||
|
|
int long_only)
|
||
|
|
{
|
||
|
|
char mode, colon_mode;
|
||
|
|
int shortoff = 0, opt = -1;
|
||
|
|
|
||
|
|
if (getenv("POSIXLY_CORRECT")) {
|
||
|
|
colon_mode = mode = '+';
|
||
|
|
} else {
|
||
|
|
if ((colon_mode = *shortopts) == ':') shortoff++;
|
||
|
|
if (((mode = shortopts[shortoff]) == '+') || (mode == '-')) {
|
||
|
|
shortoff++;
|
||
|
|
if ((colon_mode != ':') && ((colon_mode = shortopts[shortoff]) == ':'))
|
||
|
|
shortoff++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
optarg = 0;
|
||
|
|
if ((optind >= argc) ||
|
||
|
|
((argv[optind][0] == '-') &&
|
||
|
|
(argv[optind][1] == '-') &&
|
||
|
|
(argv[optind][2] == '\0'))) {
|
||
|
|
optind++;
|
||
|
|
opt = -1;
|
||
|
|
} else if ((argv[optind][0] != '-') ||
|
||
|
|
(argv[optind][1] == '\0')) {
|
||
|
|
char *tmp;
|
||
|
|
int i, j, k;
|
||
|
|
|
||
|
|
opt = -1;
|
||
|
|
if (mode == '+')
|
||
|
|
return -1;
|
||
|
|
else if (mode == '-') {
|
||
|
|
optarg = argv[optind++];
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
for (i = j = optind; i < argc; i++)
|
||
|
|
if ((argv[i][0] == '-') &&
|
||
|
|
(argv[i][1] != '\0')) {
|
||
|
|
optind = i;
|
||
|
|
opt = _getopt_internal(argc, argv, shortopts,
|
||
|
|
longopts, longind,
|
||
|
|
long_only);
|
||
|
|
while (i > j) {
|
||
|
|
tmp = argv[--i];
|
||
|
|
for (k = i; k + 1 < optind; k++)
|
||
|
|
argv[k] = argv[k + 1];
|
||
|
|
argv[--optind] = tmp;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
} else if ((!long_only) && (argv[optind][1] != '-'))
|
||
|
|
opt = my_getopt(argc, argv, shortopts);
|
||
|
|
else {
|
||
|
|
int charind, offset;
|
||
|
|
int found = 0, ind, hits = 0;
|
||
|
|
|
||
|
|
if (((optopt = argv[optind][1]) != '-') && !argv[optind][2]) {
|
||
|
|
int c;
|
||
|
|
|
||
|
|
ind = shortoff;
|
||
|
|
while ((c = shortopts[ind++])) {
|
||
|
|
if (((shortopts[ind] == ':') ||
|
||
|
|
((c == 'W') && (shortopts[ind] == ';'))) &&
|
||
|
|
(shortopts[++ind] == ':'))
|
||
|
|
ind++;
|
||
|
|
if (optopt == c) return my_getopt(argc, argv, shortopts);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
offset = 2 - (argv[optind][1] != '-');
|
||
|
|
for (charind = offset;
|
||
|
|
(argv[optind][charind] != '\0') &&
|
||
|
|
(argv[optind][charind] != '=');
|
||
|
|
charind++)
|
||
|
|
;
|
||
|
|
for (ind = 0; longopts[ind].name && !hits; ind++)
|
||
|
|
if ((strlen(longopts[ind].name) == (size_t)(charind - offset)) &&
|
||
|
|
(strncmp(longopts[ind].name,
|
||
|
|
argv[optind] + offset, charind - offset) == 0))
|
||
|
|
found = ind, hits++;
|
||
|
|
if (!hits)
|
||
|
|
for (ind = 0; longopts[ind].name; ind++)
|
||
|
|
if (strncmp(longopts[ind].name,
|
||
|
|
argv[optind] + offset, charind - offset) == 0)
|
||
|
|
found = ind, hits++;
|
||
|
|
if (hits == 1) {
|
||
|
|
opt = 0;
|
||
|
|
|
||
|
|
if (argv[optind][charind] == '=') {
|
||
|
|
if (longopts[found].has_arg == 0) {
|
||
|
|
opt = '?';
|
||
|
|
if (opterr) fprintf(stderr,
|
||
|
|
"%s: option `--%s' doesn't allow an argument\n",
|
||
|
|
argv[0], longopts[found].name);
|
||
|
|
} else {
|
||
|
|
optarg = argv[optind] + ++charind;
|
||
|
|
// charind = 0; // Never used again past here
|
||
|
|
}
|
||
|
|
} else if (longopts[found].has_arg == 1) {
|
||
|
|
if (++optind >= argc) {
|
||
|
|
opt = (colon_mode == ':') ? ':' : '?';
|
||
|
|
if (opterr) fprintf(stderr,
|
||
|
|
"%s: option `--%s' requires an argument\n",
|
||
|
|
argv[0], longopts[found].name);
|
||
|
|
} else
|
||
|
|
optarg = argv[optind];
|
||
|
|
}
|
||
|
|
if (!opt) {
|
||
|
|
if (longind) *longind = found;
|
||
|
|
if (!longopts[found].flag)
|
||
|
|
opt = longopts[found].val;
|
||
|
|
else
|
||
|
|
*(longopts[found].flag) = longopts[found].val;
|
||
|
|
}
|
||
|
|
optind++;
|
||
|
|
} else if (!hits) {
|
||
|
|
if (offset == 1)
|
||
|
|
opt = my_getopt(argc, argv, shortopts);
|
||
|
|
else {
|
||
|
|
opt = '?';
|
||
|
|
if (opterr) fprintf(stderr,
|
||
|
|
"%s: unrecognized option `%s'\n",
|
||
|
|
argv[0], argv[optind++]);
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
opt = '?';
|
||
|
|
if (opterr) fprintf(stderr,
|
||
|
|
"%s: option `%s' is ambiguous\n",
|
||
|
|
argv[0], argv[optind++]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (optind > argc) optind = argc;
|
||
|
|
return opt;
|
||
|
|
}
|
||
|
|
|
||
|
|
int my_getopt_long(int argc, char *argv[], const char *shortopts,
|
||
|
|
const struct option *longopts, int *longind)
|
||
|
|
{
|
||
|
|
return _getopt_internal(argc, argv, shortopts, longopts, longind, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
int my_getopt_long_only(int argc, char *argv[], const char *shortopts,
|
||
|
|
const struct option *longopts, int *longind)
|
||
|
|
{
|
||
|
|
return _getopt_internal(argc, argv, shortopts, longopts, longind, 1);
|
||
|
|
}
|