• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*---------------------------------------------------------------------------*
2  * <RCS keywords>
3  *
4  * C++ Library
5  *
6  * Copyright 1992-1994, David Gottner
7  *
8  *                    All Rights Reserved
9  *
10  * Permission to use, copy, modify, and distribute this software and its
11  * documentation for any purpose and without fee is hereby granted,
12  * provided that the above copyright notice, this permission notice and
13  * the following disclaimer notice appear unmodified in all copies.
14  *
15  * I DISCLAIM ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL I
17  * BE LIABLE FOR ANY SPECIAL, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY
18  * DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA, OR PROFITS, WHETHER
19  * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
20  * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
21  *
22  * Nevertheless, I would like to know about bugs in this library or
23  * suggestions for improvment.  Send bug reports and feedback to
24  * davegottner@delphi.com.
25  *---------------------------------------------------------------------------*/
26 
27 /* Modified to support --help and --version, as well as /? on Windows
28  * by Georg Brandl. */
29 
30 #include <Python.h>
31 #include <stdio.h>
32 #include <string.h>
33 #include <wchar.h>
34 #include <pygetopt.h>
35 
36 #ifdef __cplusplus
37 extern "C" {
38 #endif
39 
40 int _PyOS_opterr = 1;          /* generate error messages */
41 int _PyOS_optind = 1;          /* index into argv array   */
42 wchar_t *_PyOS_optarg = NULL;     /* optional argument       */
43 
44 static wchar_t *opt_ptr = L"";
45 
_PyOS_ResetGetOpt(void)46 void _PyOS_ResetGetOpt(void)
47 {
48     _PyOS_opterr = 1;
49     _PyOS_optind = 1;
50     _PyOS_optarg = NULL;
51     opt_ptr = L"";
52 }
53 
_PyOS_GetOpt(int argc,wchar_t ** argv,wchar_t * optstring)54 int _PyOS_GetOpt(int argc, wchar_t **argv, wchar_t *optstring)
55 {
56     wchar_t *ptr;
57     wchar_t option;
58 
59     if (*opt_ptr == '\0') {
60 
61         if (_PyOS_optind >= argc)
62             return -1;
63 #ifdef MS_WINDOWS
64         else if (wcscmp(argv[_PyOS_optind], L"/?") == 0) {
65             ++_PyOS_optind;
66             return 'h';
67         }
68 #endif
69 
70         else if (argv[_PyOS_optind][0] != L'-' ||
71                  argv[_PyOS_optind][1] == L'\0' /* lone dash */ )
72             return -1;
73 
74         else if (wcscmp(argv[_PyOS_optind], L"--") == 0) {
75             ++_PyOS_optind;
76             return -1;
77         }
78 
79         else if (wcscmp(argv[_PyOS_optind], L"--help") == 0) {
80             ++_PyOS_optind;
81             return 'h';
82         }
83 
84         else if (wcscmp(argv[_PyOS_optind], L"--version") == 0) {
85             ++_PyOS_optind;
86             return 'V';
87         }
88 
89 
90         opt_ptr = &argv[_PyOS_optind++][1];
91     }
92 
93     if ((option = *opt_ptr++) == L'\0')
94         return -1;
95 
96     if (option == 'J') {
97         if (_PyOS_opterr)
98             fprintf(stderr, "-J is reserved for Jython\n");
99         return '_';
100     }
101 
102     if ((ptr = wcschr(optstring, option)) == NULL) {
103         if (_PyOS_opterr)
104             fprintf(stderr, "Unknown option: -%c\n", (char)option);
105         return '_';
106     }
107 
108     if (*(ptr + 1) == L':') {
109         if (*opt_ptr != L'\0') {
110             _PyOS_optarg  = opt_ptr;
111             opt_ptr = L"";
112         }
113 
114         else {
115             if (_PyOS_optind >= argc) {
116                 if (_PyOS_opterr)
117                     fprintf(stderr,
118                         "Argument expected for the -%c option\n", (char)option);
119                 return '_';
120             }
121 
122             _PyOS_optarg = argv[_PyOS_optind++];
123         }
124     }
125 
126     return option;
127 }
128 
129 #ifdef __cplusplus
130 }
131 #endif
132 
133