1/***********************************************************************
2*                                                                      *
3*               This software is part of the ast package               *
4*          Copyright (c) 1985-2011 AT&T Intellectual Property          *
5*                      and is licensed under the                       *
6*                 Eclipse Public License, Version 1.0                  *
7*                    by AT&T Intellectual Property                     *
8*                                                                      *
9*                A copy of the License is available at                 *
10*          http://www.eclipse.org/org/documents/epl-v10.html           *
11*         (with md5 checksum b35adb5213ca9657e911e9befb180842)         *
12*                                                                      *
13*              Information and Software Systems Research               *
14*                            AT&T Research                             *
15*                           Florham Park NJ                            *
16*                                                                      *
17*                 Glenn Fowler <gsf@research.att.com>                  *
18*                  David Korn <dgk@research.att.com>                   *
19*                   Phong Vo <kpv@research.att.com>                    *
20*                                                                      *
21***********************************************************************/
22#pragma prototyped
23
24#include <ast.h>
25#include <ast_getopt.h>
26
27#undef	_BLD_ast	/* enable ast imports since we're user static */
28
29#include <error.h>
30#include <option.h>
31#include <getopt.h>
32#include <ctype.h>
33
34static const char*		lastoptstring;
35static const struct option*	lastlongopts;
36static char*			usage;
37static Sfio_t*			up;
38
39static int			lastoptind;
40
41static int
42golly(int argc, char* const* argv, const char* optstring, const struct option* longopts, int* longindex, int flags)
43{
44	register char*			s;
45	register const struct option*	o;
46	register int			c;
47	char*				t;
48
49	if (!up || optstring != lastoptstring || longopts != lastlongopts)
50	{
51		if (!up && !(up = sfstropen()) || !(t = strdup(optstring)))
52			return -1;
53		sfprintf(up, "[-1p%d]", flags);
54		for (o = longopts; o->name; o++)
55		{
56			if (o->flag || o->val <= 0 || o->val > UCHAR_MAX || !isalnum(o->val))
57				sfprintf(up, "\n[%d:%s]", UCHAR_MAX + 1 + (o - longopts), o->name);
58			else
59			{
60				sfprintf(up, "\n[%c:%s]", o->val, o->name);
61				if (s = strchr(t, o->val))
62				{
63					*s++ = ' ';
64					if (*s == ':')
65					{
66						*s++ = ' ';
67						if (*s == ':')
68							*s = ' ';
69					}
70				}
71			}
72			if (o->has_arg)
73			{
74				sfputc(up, ':');
75				if (o->has_arg == optional_argument)
76					sfputc(up, '?');
77				sfprintf(up, "[string]");
78			}
79		}
80		s = t;
81		while (c = *s++)
82			if (c != ' ')
83			{
84				sfprintf(up, "\n[%c]", c);
85				if (*s == ':')
86				{
87					sfputc(up, *s);
88					if (*++s == ':')
89					{
90						sfputc(up, '?');
91						s++;
92					}
93					sfputc(up, '[');
94					sfputc(up, ']');
95				}
96			}
97		sfputc(up, '\n');
98		free(t);
99		if (!(usage = sfstruse(up)))
100			return -1;
101		lastoptstring = optstring;
102		lastlongopts = longopts;
103	}
104	opt_info.index = (optind > 1 || optind == lastoptind) ? optind : 0;
105	if (opt_info.index >= argc || !(c = optget((char**)argv, usage)))
106	{
107		sfstrclose(up);
108		up = 0;
109		c = -1;
110	}
111	else
112	{
113		if (c == ':' || c == '?')
114		{
115			if (opterr && (!optstring || *optstring != ':'))
116			{
117				if (!error_info.id)
118					error_info.id = argv[0];
119				errormsg(NiL, c == '?' ? (ERROR_USAGE|4) : 2, "%s", opt_info.arg);
120			}
121			optopt = opt_info.option[1];
122			c = '?';
123		}
124		optarg = opt_info.arg;
125		if (c < 0)
126		{
127			o = longopts - c - UCHAR_MAX - 1;
128			if (o->flag)
129			{
130				*o->flag = o->val;
131				c = 0;
132			}
133			else
134				c = o->val;
135		}
136	}
137	lastoptind = optind = opt_info.index;
138	return c;
139}
140
141extern int
142getopt_long(int argc, char* const* argv, const char* optstring, const struct option* longopts, int* longindex)
143{
144	return golly(argc, argv, optstring, longopts, longindex, 2);
145}
146
147extern int
148getopt_long_only(int argc, char* const* argv, const char* optstring, const struct option* longopts, int* longindex)
149{
150	return golly(argc, argv, optstring, longopts, longindex, 1);
151}
152