1
0

getopt_long.c 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. #define _GNU_SOURCE
  2. #include <stddef.h>
  3. #include <getopt.h>
  4. #include <stdio.h>
  5. extern int __optpos, __optreset;
  6. static int __getopt_long(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx, int longonly)
  7. {
  8. if (!optind || __optreset) {
  9. __optreset = 0;
  10. __optpos = 0;
  11. optind = 1;
  12. }
  13. if (optind >= argc || !argv[optind] || argv[optind][0] != '-') return -1;
  14. if ((longonly && argv[optind][1]) ||
  15. (argv[optind][1] == '-' && argv[optind][2]))
  16. {
  17. int i;
  18. for (i=0; longopts[i].name; i++) {
  19. const char *name = longopts[i].name;
  20. char *opt = argv[optind]+1;
  21. if (*opt == '-') opt++;
  22. for (; *name && *name == *opt; name++, opt++);
  23. if (*name || (*opt && *opt != '=')) continue;
  24. if (*opt == '=') {
  25. if (!longopts[i].has_arg) continue;
  26. optarg = opt+1;
  27. } else {
  28. if (longopts[i].has_arg == required_argument) {
  29. if (!(optarg = argv[++optind]))
  30. return ':';
  31. } else optarg = NULL;
  32. }
  33. optind++;
  34. if (idx) *idx = i;
  35. if (longopts[i].flag) {
  36. *longopts[i].flag = longopts[i].val;
  37. return 0;
  38. }
  39. return longopts[i].val;
  40. }
  41. if (argv[optind][1] == '-') {
  42. optind++;
  43. return '?';
  44. }
  45. }
  46. return getopt(argc, argv, optstring);
  47. }
  48. int getopt_long(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx)
  49. {
  50. return __getopt_long(argc, argv, optstring, longopts, idx, 0);
  51. }
  52. int getopt_long_only(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx)
  53. {
  54. return __getopt_long(argc, argv, optstring, longopts, idx, 1);
  55. }