1#define _GNU_SOURCE 2#include <stddef.h> 3#include <getopt.h> 4#include <stdio.h> 5#include <string.h> 6 7extern int __optpos, __optreset; 8 9static void permute(char *const *argv, int dest, int src) 10{ 11 char **av = (char **)argv; 12 char *tmp = av[src]; 13 int i; 14 for (i=src; i>dest; i--) 15 av[i] = av[i-1]; 16 av[dest] = tmp; 17} 18 19void __getopt_msg(const char *, const char *, const char *, size_t); 20 21static int __getopt_long_core(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx, int longonly); 22 23static int __getopt_long(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx, int longonly) 24{ 25 int ret, skipped, resumed; 26 if (!optind || __optreset) { 27 __optreset = 0; 28 __optpos = 0; 29 optind = 1; 30 } 31 if (optind >= argc || !argv[optind]) return -1; 32 skipped = optind; 33 if (optstring[0] != '+' && optstring[0] != '-') { 34 int i; 35 for (i=optind; ; i++) { 36 if (i >= argc || !argv[i]) return -1; 37 if (argv[i][0] == '-' && argv[i][1]) break; 38 } 39 optind = i; 40 } 41 resumed = optind; 42 ret = __getopt_long_core(argc, argv, optstring, longopts, idx, longonly); 43 if (resumed > skipped) { 44 int i, cnt = optind-resumed; 45 for (i=0; i<cnt; i++) 46 permute(argv, skipped, optind-1); 47 optind = skipped + cnt; 48 } 49 return ret; 50} 51 52static int __getopt_long_core(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx, int longonly) 53{ 54 optarg = 0; 55 if (longopts && argv[optind][0] == '-' && 56 ((longonly && argv[optind][1] && argv[optind][1] != '-') || 57 (argv[optind][1] == '-' && argv[optind][2]))) 58 { 59 int colon = optstring[optstring[0]=='+'||optstring[0]=='-']==':'; 60 int i, cnt, match; 61 char *opt; 62 for (cnt=i=0; longopts[i].name; i++) { 63 const char *name = longopts[i].name; 64 opt = argv[optind]+1; 65 if (*opt == '-') opt++; 66 for (; *name && *name == *opt; name++, opt++); 67 if (*opt && *opt != '=') continue; 68 match = i; 69 if (!*name) { 70 cnt = 1; 71 break; 72 } 73 cnt++; 74 } 75 if (cnt==1) { 76 i = match; 77 optind++; 78 optopt = longopts[i].val; 79 if (*opt == '=') { 80 if (!longopts[i].has_arg) { 81 if (colon || !opterr) 82 return '?'; 83 __getopt_msg(argv[0], 84 ": option does not take an argument: ", 85 longopts[i].name, 86 strlen(longopts[i].name)); 87 return '?'; 88 } 89 optarg = opt+1; 90 } else if (longopts[i].has_arg == required_argument) { 91 if (!(optarg = argv[optind])) { 92 if (colon) return ':'; 93 if (!opterr) return '?'; 94 __getopt_msg(argv[0], 95 ": option requires an argument: ", 96 longopts[i].name, 97 strlen(longopts[i].name)); 98 return '?'; 99 } 100 optind++; 101 } 102 if (idx) *idx = i; 103 if (longopts[i].flag) { 104 *longopts[i].flag = longopts[i].val; 105 return 0; 106 } 107 return longopts[i].val; 108 } 109 if (argv[optind][1] == '-') { 110 if (!colon && opterr) 111 __getopt_msg(argv[0], cnt ? 112 ": option is ambiguous: " : 113 ": unrecognized option: ", 114 argv[optind]+2, 115 strlen(argv[optind]+2)); 116 optind++; 117 return '?'; 118 } 119 } 120 return getopt(argc, argv, optstring); 121} 122 123int getopt_long(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx) 124{ 125 return __getopt_long(argc, argv, optstring, longopts, idx, 0); 126} 127 128int getopt_long_only(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx) 129{ 130 return __getopt_long(argc, argv, optstring, longopts, idx, 1); 131} 132