getopt.cpp 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
  1. /*
  2. * Copyright (c) 1987, 1993, 1994
  3. * The Regents of the University of California. All rights reserved.
  4. *
  5. * Redistribution and use in source and binary forms, with or without
  6. * modification, are permitted provided that the following conditions
  7. * are met:
  8. * 1. Redistributions of source code must retain the above copyright
  9. * notice, this list of conditions and the following disclaimer.
  10. * 2. Redistributions in binary form must reproduce the above copyright
  11. * notice, this list of conditions and the following disclaimer in the
  12. * documentation and/or other materials provided with the distribution.
  13. * 3. All advertising materials mentioning features or use of this software
  14. * must display the following acknowledgement:
  15. * This product includes software developed by the University of
  16. * California, Berkeley and its contributors.
  17. * 4. Neither the name of the University nor the names of its contributors
  18. * may be used to endorse or promote products derived from this software
  19. * without specific prior written permission.
  20. *
  21. * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24. * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27. * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31. * SUCH DAMAGE.
  32. */
  33. #include <getopt.h>
  34. #include <assert.h>
  35. #include <errno.h>
  36. #include <stdio.h>
  37. #include <stdlib.h>
  38. #include <string.h>
  39. #include <unistd.h>
  40. int opterr = 1; /* if error message should be printed */
  41. int optind = 1; /* index into parent argv vector */
  42. int optopt; /* character checked for validity */
  43. int optreset; /* reset getopt */
  44. char *optarg; /* argument associated with option */
  45. #define BADCH (int)'?'
  46. #define BADARG (int)':'
  47. #define EMSG ""
  48. int getopt(int nargc, char* const nargv[], const char* ostr)
  49. {
  50. static const char* place = EMSG; /* option letter processing */
  51. char *oli; /* option letter list index */
  52. ASSERT(nargv != NULL);
  53. ASSERT(ostr != NULL);
  54. if (optreset || !*place) { /* update scanning pointer */
  55. optreset = 0;
  56. if (optind >= nargc || *(place = nargv[optind]) != '-') {
  57. place = EMSG;
  58. return (-1);
  59. }
  60. if (place[1] && *++place == '-' /* found "--" */
  61. && place[1] == '\0') {
  62. ++optind;
  63. place = EMSG;
  64. return -1;
  65. }
  66. } /* option letter okay? */
  67. if ((optopt = (int)*place++) == (int)':' || !(oli = strchr(ostr, optopt))) {
  68. /*
  69. * if the user didn't specify '-' as an option,
  70. * assume it means -1.
  71. */
  72. if (optopt == (int)'-')
  73. return -1;
  74. if (!*place)
  75. ++optind;
  76. if (opterr && *ostr != ':')
  77. fprintf(stderr, "unknown option -- %c\n", optopt);
  78. return BADCH;
  79. }
  80. if (*++oli != ':') { /* don't need argument */
  81. optarg = NULL;
  82. if (!*place)
  83. ++optind;
  84. }
  85. else { /* need an argument */
  86. if (*place) /* no white space */
  87. optarg = const_cast<char*>(place);
  88. else if (nargc <= ++optind) { /* no arg */
  89. place = EMSG;
  90. if (*ostr == ':')
  91. return BADARG;
  92. if (opterr)
  93. fprintf(stderr, "option requires an argument -- %c\n", optopt);
  94. return BADCH;
  95. }
  96. else /* white space */
  97. optarg = nargv[optind];
  98. place = EMSG;
  99. ++optind;
  100. }
  101. return optopt; /* dump back option letter */
  102. }