1 /*	$NetBSD: getopt.c,v 1.27 2005/11/29 03:12:00 christos Exp $	*/
2 
3 /*
4  * Copyright (c) 1987, 1993, 1994
5  *	The Regents of the University of California.  All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. Neither the name of the University nor the names of its contributors
16  *    may be used to endorse or promote products derived from this software
17  *    without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  */
31 
32 #undef LIBC_SCCS
33 
34 #if !defined(_WIN32)
35 # include <sys/cdefs.h>
36 #else
37 # define __UNCONST(x) ((char*)(x))
38 # define getprogname() (nargv[0])
39 # define _DIAGASSERT(x) assert((x))
40 #endif
41 
42 #include <assert.h>
43 #include <errno.h>
44 #include <stdio.h>
45 #include <stdlib.h>
46 #include <string.h>
47 #if !defined(_WIN32)
48 # include <unistd.h>
49 #endif
50 
51 #include <freetds/sysdep_private.h>
52 
53 #ifdef __weak_alias
54 __weak_alias(getopt,_getopt)
55 #endif
56 
57 int	opterr = 1,		/* if error message should be printed */
58 	optind = 1,		/* index into parent argv vector */
59 	optopt,			/* character checked for validity */
60 	optreset;		/* reset getopt */
61 char	*optarg;		/* argument associated with option */
62 
63 #define	BADCH	(int)'?'
64 #define	BADARG	(int)':'
65 #define	EMSG	""
66 
67 /*
68  * getopt --
69  *	Parse argc/argv argument vector.
70  */
71 int
tds_getopt(nargc,nargv,ostr)72 tds_getopt(nargc, nargv, ostr)
73 	int nargc;
74 	char * const nargv[];
75 	const char *ostr;
76 {
77 	static const char *place = EMSG;	/* option letter processing */
78 	char *oli;				/* option letter list index */
79 
80 	_DIAGASSERT(nargv != NULL);
81 	_DIAGASSERT(ostr != NULL);
82 
83 	if (optreset || *place == 0) {		/* update scanning pointer */
84 		optreset = 0;
85 		place = nargv[optind];
86 		if (optind >= nargc || *place++ != '-') {
87 			/* Argument is absent or is not an option */
88 			place = EMSG;
89 			return (-1);
90 		}
91 		optopt = *place++;
92 		if (optopt == '-' && *place == 0) {
93 			/* "--" => end of options */
94 			++optind;
95 			place = EMSG;
96 			return (-1);
97 		}
98 		if (optopt == 0) {
99 			/* Solitary '-', treat as a '-' option
100 			   if the program (eg su) is looking for it. */
101 			place = EMSG;
102 			if (strchr(ostr, '-') == NULL)
103 				return -1;
104 			optopt = '-';
105 		}
106 	} else
107 		optopt = *place++;
108 
109 	/* See if option letter is one the caller wanted... */
110 	if (optopt == ':' || (oli = strchr(ostr, optopt)) == NULL) {
111 		if (*place == 0)
112 			++optind;
113 		if (opterr && *ostr != ':')
114 			(void)fprintf(stderr,
115 			    "%s: unknown option -- %c\n", getprogname(),
116 			    optopt);
117 		return (BADCH);
118 	}
119 
120 	/* Does this option need an argument? */
121 	if (oli[1] != ':') {
122 		/* don't need argument */
123 		optarg = NULL;
124 		if (*place == 0)
125 			++optind;
126 	} else {
127 		/* Option-argument is either the rest of this argument or the
128 		   entire next argument. */
129 		if (*place)
130 			optarg = __UNCONST(place);
131 		else if (nargc > ++optind)
132 			optarg = nargv[optind];
133 		else {
134 			/* option-argument absent */
135 			place = EMSG;
136 			if (*ostr == ':')
137 				return (BADARG);
138 			if (opterr)
139 				(void)fprintf(stderr,
140 				    "%s: option requires an argument -- %c\n",
141 				    getprogname(), optopt);
142 			return (BADCH);
143 		}
144 		place = EMSG;
145 		++optind;
146 	}
147 	return (optopt);			/* return option letter */
148 }
149