getopt.c 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  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. #if defined(LIBC_SCCS) && !defined(lint)
  34. static char sccsid[] = "@(#)getopt.c 8.3 (Berkeley) 4/27/95";
  35. #endif /* LIBC_SCCS and not lint */
  36. #include <stdio.h>
  37. #include <stdlib.h>
  38. #include <string.h>
  39. #include "getopt.h"
  40. int opterr = 1, /* if error message should be printed */
  41. optind = 1, /* index into parent argv vector */
  42. optopt, /* character checked for validity */
  43. optreset; /* reset getopt */
  44. char *optarg; /* argument associated with option */
  45. #define BADCH (int)'?'
  46. #define BADARG (int)':'
  47. #define EMSG ""
  48. /*
  49. * getopt --
  50. * Parse argc/argv argument vector.
  51. */
  52. int
  53. getopt(int nargc, char * const *nargv, const char *ostr)
  54. {
  55. char *cp;
  56. static char *__progname;
  57. static char *place = EMSG; /* option letter processing */
  58. char *oli; /* option letter list index */
  59. if (__progname == NULL) {
  60. if ((cp = strrchr(nargv[0], '/')) != NULL)
  61. __progname = cp + 1;
  62. else
  63. __progname = nargv[0];
  64. }
  65. if (optreset || !*place) { /* update scanning pointer */
  66. optreset = 0;
  67. if (optind >= nargc || *(place = nargv[optind]) != '-') {
  68. place = EMSG;
  69. return (-1);
  70. }
  71. if (place[1] && *++place == '-') { /* found "--" */
  72. ++optind;
  73. place = EMSG;
  74. return (-1);
  75. }
  76. } /* option letter okay? */
  77. if ((optopt = (int)*place++) == (int)':' ||
  78. !(oli = strchr(ostr, optopt))) {
  79. /*
  80. * if the user didn't specify '-' as an option,
  81. * assume it means -1.
  82. */
  83. if (optopt == (int)'-')
  84. return (-1);
  85. if (!*place)
  86. ++optind;
  87. if (opterr && *ostr != ':')
  88. (void)fprintf(stderr,
  89. "%s: illegal option -- %c\n", __progname, optopt);
  90. return (BADCH);
  91. }
  92. if (*++oli != ':') { /* don't need argument */
  93. optarg = NULL;
  94. if (!*place)
  95. ++optind;
  96. }
  97. else { /* need an argument */
  98. if (*place) /* no white space */
  99. optarg = place;
  100. else if (nargc <= ++optind) { /* no arg */
  101. place = EMSG;
  102. if (*ostr == ':')
  103. return (BADARG);
  104. if (opterr)
  105. (void)fprintf(stderr,
  106. "%s: option requires an argument -- %c\n",
  107. __progname, optopt);
  108. return (BADCH);
  109. }
  110. else /* white space */
  111. optarg = nargv[optind];
  112. place = EMSG;
  113. ++optind;
  114. }
  115. return (optopt); /* dump back option letter */
  116. }