]> granicus.if.org Git - postgresql/blob - src/port/getopt.c
Make psql reject attempts to set special variables to invalid values.
[postgresql] / src / port / getopt.c
1 /* src/port/getopt.c */
2
3 /* This is used by psql under Win32 */
4
5 /*
6  * Copyright (c) 1987, 1993, 1994
7  *      The Regents of the University of California.  All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  * 1. Redistributions of source code must retain the above copyright
13  *        notice, this list of conditions and the following disclaimer.
14  * 2. Redistributions in binary form must reproduce the above copyright
15  *        notice, this list of conditions and the following disclaimer in the
16  *        documentation and/or other materials provided with the distribution.
17  * 3. 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  */
32
33 #include "c.h"
34
35 #include "pg_getopt.h"
36
37 #if defined(LIBC_SCCS) && !defined(lint)
38 static char sccsid[] = "@(#)getopt.c    8.3 (Berkeley) 4/27/95";
39 #endif   /* LIBC_SCCS and not lint */
40
41
42 /*
43  * On some versions of Solaris, opterr and friends are defined in core libc
44  * rather than in a separate getopt module.  Define these variables only
45  * if configure found they aren't there by default.  (We assume that testing
46  * opterr is sufficient for all of these.)
47  */
48 #ifndef HAVE_INT_OPTERR
49
50 int                     opterr = 1,                     /* if error message should be printed */
51                         optind = 1,                     /* index into parent argv vector */
52                         optopt;                         /* character checked for validity */
53 char       *optarg;                             /* argument associated with option */
54
55 #endif
56
57 #define BADCH   (int)'?'
58 #define BADARG  (int)':'
59 #define EMSG    ""
60
61 /*
62  * getopt
63  *      Parse argc/argv argument vector.
64  *
65  * This implementation does not use optreset.  Instead, we guarantee that
66  * it can be restarted on a new argv array after a previous call returned -1,
67  * if the caller resets optind to 1 before the first call of the new series.
68  * (Internally, this means we must be sure to reset "place" to EMSG before
69  * returning -1.)
70  */
71 int
72 getopt(int nargc, char *const * nargv, const char *ostr)
73 {
74         static char *place = EMSG;      /* option letter processing */
75         char       *oli;                        /* option letter list index */
76
77         if (!*place)
78         {                                                       /* update scanning pointer */
79                 if (optind >= nargc || *(place = nargv[optind]) != '-')
80                 {
81                         place = EMSG;
82                         return -1;
83                 }
84                 if (place[1] && *++place == '-' && place[1] == '\0')
85                 {                                               /* found "--" */
86                         ++optind;
87                         place = EMSG;
88                         return -1;
89                 }
90         }                                                       /* option letter okay? */
91         if ((optopt = (int) *place++) == (int) ':' ||
92                 !(oli = strchr(ostr, optopt)))
93         {
94                 /*
95                  * if the user didn't specify '-' as an option, assume it means -1.
96                  */
97                 if (optopt == (int) '-')
98                 {
99                         place = EMSG;
100                         return -1;
101                 }
102                 if (!*place)
103                         ++optind;
104                 if (opterr && *ostr != ':')
105                         (void) fprintf(stderr,
106                                                    "illegal option -- %c\n", optopt);
107                 return BADCH;
108         }
109         if (*++oli != ':')
110         {                                                       /* don't need argument */
111                 optarg = NULL;
112                 if (!*place)
113                         ++optind;
114         }
115         else
116         {                                                       /* need an argument */
117                 if (*place)                             /* no white space */
118                         optarg = place;
119                 else if (nargc <= ++optind)
120                 {                                               /* no arg */
121                         place = EMSG;
122                         if (*ostr == ':')
123                                 return BADARG;
124                         if (opterr)
125                                 (void) fprintf(stderr,
126                                                            "option requires an argument -- %c\n",
127                                                            optopt);
128                         return BADCH;
129                 }
130                 else
131                         /* white space */
132                         optarg = nargv[optind];
133                 place = EMSG;
134                 ++optind;
135         }
136         return optopt;                          /* dump back option letter */
137 }