forked from osmocom/wireshark
270 lines
7.0 KiB
C
270 lines
7.0 KiB
C
/*
|
|
* musl as a whole is licensed under the following standard MIT license:
|
|
*
|
|
* ----------------------------------------------------------------------
|
|
* Copyright © 2005-2020 Rich Felker, et al.
|
|
*
|
|
* 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.
|
|
* ----------------------------------------------------------------------
|
|
*/
|
|
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <limits.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <wchar.h>
|
|
|
|
#include <wsutil/ws_getopt.h>
|
|
|
|
char *ws_optarg;
|
|
int ws_optind=1, ws_opterr=1, ws_optopt, ws_optpos, ws_optreset=0;
|
|
|
|
static void __getopt_msg(const char *prog, const char *errstr,
|
|
const char *optbuf, size_t optsize)
|
|
{
|
|
FILE *f = stderr;
|
|
if ((fputs(prog, f) < 0) ||
|
|
(fputs(errstr, f) < 0) ||
|
|
(fwrite(optbuf, sizeof(char), optsize, f) != optsize)) {
|
|
return;
|
|
}
|
|
putc('\n', f);
|
|
}
|
|
|
|
static void permute(char *const *argv, int dest, int src)
|
|
{
|
|
char **av = (char **)argv;
|
|
char *tmp = av[src];
|
|
int i;
|
|
for (i=src; i>dest; i--)
|
|
av[i] = av[i-1];
|
|
av[dest] = tmp;
|
|
}
|
|
|
|
int ws_getopt(int argc, char * const argv[], const char *optstring)
|
|
{
|
|
int i;
|
|
wchar_t c, d;
|
|
int k, l;
|
|
char *optchar;
|
|
|
|
if (!ws_optind || ws_optreset) {
|
|
ws_optreset = 0;
|
|
ws_optpos = 0;
|
|
ws_optind = 1;
|
|
}
|
|
|
|
if (ws_optind >= argc || !argv[ws_optind])
|
|
return -1;
|
|
|
|
if (argv[ws_optind][0] != '-') {
|
|
if (optstring[0] == '-') {
|
|
ws_optarg = argv[ws_optind++];
|
|
return 1;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
if (!argv[ws_optind][1])
|
|
return -1;
|
|
|
|
if (argv[ws_optind][1] == '-' && !argv[ws_optind][2]) {
|
|
ws_optind++;
|
|
return -1;
|
|
}
|
|
|
|
if (!ws_optpos) ws_optpos++;
|
|
if ((k = mbtowc(&c, argv[ws_optind]+ws_optpos, MB_LEN_MAX)) < 0) {
|
|
k = 1;
|
|
c = 0xfffd; /* replacement char */
|
|
}
|
|
optchar = argv[ws_optind]+ws_optpos;
|
|
ws_optpos += k;
|
|
|
|
if (!argv[ws_optind][ws_optpos]) {
|
|
ws_optind++;
|
|
ws_optpos = 0;
|
|
}
|
|
|
|
if (optstring[0] == '-' || optstring[0] == '+')
|
|
optstring++;
|
|
|
|
i = 0;
|
|
d = 0;
|
|
do {
|
|
l = mbtowc(&d, optstring+i, MB_LEN_MAX);
|
|
if (l>0) i+=l; else i++;
|
|
} while (l && d != c);
|
|
|
|
if (d != c || c == ':') {
|
|
ws_optopt = c;
|
|
if (optstring[0] != ':' && ws_opterr)
|
|
__getopt_msg(argv[0], ": unrecognized option: ", optchar, k);
|
|
return '?';
|
|
}
|
|
if (optstring[i] == ':') {
|
|
ws_optarg = 0;
|
|
if (optstring[i+1] != ':' || ws_optpos) {
|
|
ws_optarg = argv[ws_optind++] + ws_optpos;
|
|
ws_optpos = 0;
|
|
}
|
|
if (ws_optind > argc) {
|
|
ws_optopt = c;
|
|
if (optstring[0] == ':') return ':';
|
|
if (ws_opterr) __getopt_msg(argv[0],
|
|
": option requires an argument: ",
|
|
optchar, k);
|
|
return '?';
|
|
}
|
|
}
|
|
return c;
|
|
}
|
|
|
|
static int __getopt_long_core(int argc, char *const *argv, const char *optstring, const struct ws_option *longopts, int *idx, int longonly);
|
|
|
|
static int __getopt_long(int argc, char *const *argv, const char *optstring, const struct ws_option *longopts, int *idx, int longonly)
|
|
{
|
|
int ret, skipped, resumed;
|
|
if (!ws_optind || ws_optreset) {
|
|
ws_optreset = 0;
|
|
ws_optpos = 0;
|
|
ws_optind = 1;
|
|
}
|
|
if (ws_optind >= argc || !argv[ws_optind]) return -1;
|
|
skipped = ws_optind;
|
|
if (optstring[0] != '+' && optstring[0] != '-') {
|
|
int i;
|
|
for (i=ws_optind; ; i++) {
|
|
if (i >= argc || !argv[i]) return -1;
|
|
if (argv[i][0] == '-' && argv[i][1]) break;
|
|
}
|
|
ws_optind = i;
|
|
}
|
|
resumed = ws_optind;
|
|
ret = __getopt_long_core(argc, argv, optstring, longopts, idx, longonly);
|
|
if (resumed > skipped) {
|
|
int i, cnt = ws_optind-resumed;
|
|
for (i=0; i<cnt; i++)
|
|
permute(argv, skipped, ws_optind-1);
|
|
ws_optind = skipped + cnt;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static int __getopt_long_core(int argc, char *const *argv, const char *optstring, const struct ws_option *longopts, int *idx, int longonly)
|
|
{
|
|
ws_optarg = 0;
|
|
if (longopts && argv[ws_optind][0] == '-' &&
|
|
((longonly && argv[ws_optind][1] && argv[ws_optind][1] != '-') ||
|
|
(argv[ws_optind][1] == '-' && argv[ws_optind][2])))
|
|
{
|
|
int colon = optstring[optstring[0]=='+'||optstring[0]=='-']==':';
|
|
int i, cnt, match = -1;
|
|
char *arg = NULL, *opt, *start = argv[ws_optind]+1;
|
|
for (cnt=i=0; longopts[i].name; i++) {
|
|
const char *name = longopts[i].name;
|
|
opt = start;
|
|
if (*opt == '-') opt++;
|
|
while (*opt && *opt != '=' && *opt == *name) {
|
|
name++;
|
|
opt++;
|
|
}
|
|
if (*opt && *opt != '=') continue;
|
|
arg = opt;
|
|
match = i;
|
|
if (!*name) {
|
|
cnt = 1;
|
|
break;
|
|
}
|
|
cnt++;
|
|
}
|
|
if (cnt==1 && longonly && arg-start == mblen(start, MB_LEN_MAX)) {
|
|
ptrdiff_t l = arg - start;
|
|
for (i=0; optstring[i]; i++) {
|
|
ptrdiff_t j;
|
|
for (j=0; j<l && start[j]==optstring[i+j]; j++);
|
|
if (j==l) {
|
|
cnt++;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if (cnt==1) {
|
|
i = match;
|
|
opt = arg;
|
|
ws_optind++;
|
|
if (*opt == '=') {
|
|
if (!longopts[i].has_arg) {
|
|
ws_optopt = longopts[i].val;
|
|
if (colon || !ws_opterr)
|
|
return '?';
|
|
__getopt_msg(argv[0],
|
|
": option does not take an argument: ",
|
|
longopts[i].name,
|
|
strlen(longopts[i].name));
|
|
return '?';
|
|
}
|
|
ws_optarg = opt+1;
|
|
} else if (longopts[i].has_arg == ws_required_argument) {
|
|
if (!(ws_optarg = argv[ws_optind])) {
|
|
ws_optopt = longopts[i].val;
|
|
if (colon) return ':';
|
|
if (!ws_opterr) return '?';
|
|
__getopt_msg(argv[0],
|
|
": option requires an argument: ",
|
|
longopts[i].name,
|
|
strlen(longopts[i].name));
|
|
return '?';
|
|
}
|
|
ws_optind++;
|
|
}
|
|
if (idx) *idx = i;
|
|
if (longopts[i].flag) {
|
|
*longopts[i].flag = longopts[i].val;
|
|
return 0;
|
|
}
|
|
return longopts[i].val;
|
|
}
|
|
if (argv[ws_optind][1] == '-') {
|
|
ws_optopt = 0;
|
|
if (!colon && ws_opterr)
|
|
__getopt_msg(argv[0], cnt ?
|
|
": option is ambiguous: " :
|
|
": unrecognized option: ",
|
|
argv[ws_optind]+2,
|
|
strlen(argv[ws_optind]+2));
|
|
ws_optind++;
|
|
return '?';
|
|
}
|
|
}
|
|
return ws_getopt(argc, argv, optstring);
|
|
}
|
|
|
|
int ws_getopt_long(int argc, char *const *argv, const char *optstring, const struct ws_option *longopts, int *idx)
|
|
{
|
|
return __getopt_long(argc, argv, optstring, longopts, idx, 0);
|
|
}
|
|
|
|
int ws_getopt_long_only(int argc, char *const *argv, const char *optstring, const struct ws_option *longopts, int *idx)
|
|
{
|
|
return __getopt_long(argc, argv, optstring, longopts, idx, 1);
|
|
}
|