usr/dash/arith_yylex.c

/*-
 * Copyright (c) 2002
 *	Herbert Xu.
 * Copyright (c) 1993
 *	The Regents of the University of California.  All rights reserved.
 *
 * This code is derived from software contributed to Berkeley by
 * Kenneth Almquist.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <inttypes.h>
#include <stdlib.h>
#include <string.h>
#include "arith_yacc.h"
#include "expand.h"
#include "error.h"
#include "shell.h"
#include "memalloc.h"
#include "syntax.h"
#include "system.h"

#if ARITH_BOR + 11 != ARITH_BORASS || ARITH_ASS + 11 != ARITH_EQ
#error Arithmetic tokens are out of order.
#endif

extern const char *arith_buf;

int
yylex()
{
	int value;
	const char *buf = arith_buf;
	const char *p;

	for (;;) {
		value = *buf;
		switch (value) {
		case ' ':
		case '\t':
		case '\n':
			buf++;
			continue;
		default:
			return ARITH_BAD;
		case '0':
		case '1':
		case '2':
		case '3':
		case '4':
		case '5':
		case '6':
		case '7':
		case '8':
		case '9':
			yylval.val = strtoimax(buf, (char **)&arith_buf, 0);
			return ARITH_NUM;
		case 'A':
		case 'B':
		case 'C':
		case 'D':
		case 'E':
		case 'F':
		case 'G':
		case 'H':
		case 'I':
		case 'J':
		case 'K':
		case 'L':
		case 'M':
		case 'N':
		case 'O':
		case 'P':
		case 'Q':
		case 'R':
		case 'S':
		case 'T':
		case 'U':
		case 'V':
		case 'W':
		case 'X':
		case 'Y':
		case 'Z':
		case '_':
		case 'a':
		case 'b':
		case 'c':
		case 'd':
		case 'e':
		case 'f':
		case 'g':
		case 'h':
		case 'i':
		case 'j':
		case 'k':
		case 'l':
		case 'm':
		case 'n':
		case 'o':
		case 'p':
		case 'q':
		case 'r':
		case 's':
		case 't':
		case 'u':
		case 'v':
		case 'w':
		case 'x':
		case 'y':
		case 'z':
			p = buf;
			while (buf++, is_in_name(*buf))
				;
			yylval.name = stalloc(buf - p + 1);
			*(char *)mempcpy(yylval.name, p, buf - p) = 0;
			value = ARITH_VAR;
			goto out;
		case '=':
			value += ARITH_ASS - '=';
checkeq:
			buf++;
checkeqcur:
			if (*buf != '=')
				goto out;
			value += 11;
			break;
		case '>':
			switch (*++buf) {
			case '=':
				value += ARITH_GE - '>';
				break;
			case '>':
				value += ARITH_RSHIFT - '>';
				goto checkeq;
			default:
				value += ARITH_GT - '>';
				goto out;
			}
			break;
		case '<':
			switch (*++buf) {
			case '=':
				value += ARITH_LE - '<';
				break;
			case '<':
				value += ARITH_LSHIFT - '<';
				goto checkeq;
			default:
				value += ARITH_LT - '<';
				goto out;
			}
			break;
		case '|':
			if (*++buf != '|') {
				value += ARITH_BOR - '|';
				goto checkeqcur;
			}
			value += ARITH_OR - '|';
			break;
		case '&':
			if (*++buf != '&') {
				value += ARITH_BAND - '&';
				goto checkeqcur;
			}
			value += ARITH_AND - '&';
			break;
		case '!':
			if (*++buf != '=') {
				value += ARITH_NOT - '!';
				goto out;
			}
			value += ARITH_NE - '!';
			break;
		case 0:
			goto out;
		case '(':
			value += ARITH_LPAREN - '(';
			break;
		case ')':
			value += ARITH_RPAREN - ')';
			break;
		case '*':
			value += ARITH_MUL - '*';
			goto checkeq;
		case '/':
			value += ARITH_DIV - '/';
			goto checkeq;
		case '%':
			value += ARITH_REM - '%';
			goto checkeq;
		case '+':
			value += ARITH_ADD - '+';
			goto checkeq;
		case '-':
			value += ARITH_SUB - '-';
			goto checkeq;
		case '~':
			value += ARITH_BNOT - '~';
			break;
		case '^':
			value += ARITH_BXOR - '^';
			goto checkeq;
		case '?':
			value += ARITH_QMARK - '?';
			break;
		case ':':
			value += ARITH_COLON - ':';
			break;
		}
		break;
	}

	buf++;
out:
	arith_buf = buf;
	return value;
}
1	niro	532	/*-
2			* Copyright (c) 2002
3			* Herbert Xu.
4			* Copyright (c) 1993
5			* The Regents of the University of California. All rights reserved.
6			*
7			* This code is derived from software contributed to Berkeley by
8			* Kenneth Almquist.
9			*
10			* Redistribution and use in source and binary forms, with or without
11			* modification, are permitted provided that the following conditions
12			* are met:
13			* 1. Redistributions of source code must retain the above copyright
14			* notice, this list of conditions and the following disclaimer.
15			* 2. Redistributions in binary form must reproduce the above copyright
16			* notice, this list of conditions and the following disclaimer in the
17			* documentation and/or other materials provided with the distribution.
18			* 3. Neither the name of the University nor the names of its contributors
19			* may be used to endorse or promote products derived from this software
20			* without specific prior written permission.
21			*
22			* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23			* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24			* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25			* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26			* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27			* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28			* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29			* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30			* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31			* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32			* SUCH DAMAGE.
33			*/
34
35	niro	1122	#include <inttypes.h>
36	niro	532	#include <stdlib.h>
37	niro	1122	#include <string.h>
38			#include "arith_yacc.h"
39	niro	532	#include "expand.h"
40			#include "error.h"
41	niro	1122	#include "shell.h"
42			#include "memalloc.h"
43			#include "syntax.h"
44			#include "system.h"
45	niro	532
46	niro	1122	#if ARITH_BOR + 11 != ARITH_BORASS \|\| ARITH_ASS + 11 != ARITH_EQ
47			#error Arithmetic tokens are out of order.
48			#endif
49	niro	532
50	niro	1122	extern const char *arith_buf;
51
52	niro	532	int
53			yylex()
54			{
55			int value;
56			const char *buf = arith_buf;
57	niro	1122	const char *p;
58	niro	532
59			for (;;) {
60	niro	1122	value = *buf;
61			switch (value) {
62	niro	532	case ' ':
63			case '\t':
64			case '\n':
65			buf++;
66			continue;
67			default:
68	niro	1122	return ARITH_BAD;
69	niro	532	case '0':
70			case '1':
71			case '2':
72			case '3':
73			case '4':
74			case '5':
75			case '6':
76			case '7':
77			case '8':
78			case '9':
79	niro	1122	yylval.val = strtoimax(buf, (char **)&arith_buf, 0);
80	niro	532	return ARITH_NUM;
81	niro	1122	case 'A':
82			case 'B':
83			case 'C':
84			case 'D':
85			case 'E':
86			case 'F':
87			case 'G':
88			case 'H':
89			case 'I':
90			case 'J':
91			case 'K':
92			case 'L':
93			case 'M':
94			case 'N':
95			case 'O':
96			case 'P':
97			case 'Q':
98			case 'R':
99			case 'S':
100			case 'T':
101			case 'U':
102			case 'V':
103			case 'W':
104			case 'X':
105			case 'Y':
106			case 'Z':
107			case '_':
108			case 'a':
109			case 'b':
110			case 'c':
111			case 'd':
112			case 'e':
113			case 'f':
114			case 'g':
115			case 'h':
116			case 'i':
117			case 'j':
118			case 'k':
119			case 'l':
120			case 'm':
121			case 'n':
122			case 'o':
123			case 'p':
124			case 'q':
125			case 'r':
126			case 's':
127			case 't':
128			case 'u':
129			case 'v':
130			case 'w':
131			case 'x':
132			case 'y':
133			case 'z':
134			p = buf;
135			while (buf++, is_in_name(*buf))
136			;
137			yylval.name = stalloc(buf - p + 1);
138			(char )mempcpy(yylval.name, p, buf - p) = 0;
139			value = ARITH_VAR;
140			goto out;
141	niro	532	case '=':
142	niro	1122	value += ARITH_ASS - '=';
143			checkeq:
144			buf++;
145			checkeqcur:
146			if (*buf != '=')
147			goto out;
148			value += 11;
149	niro	532	break;
150			case '>':
151			switch (*++buf) {
152			case '=':
153	niro	1122	value += ARITH_GE - '>';
154	niro	532	break;
155			case '>':
156	niro	1122	value += ARITH_RSHIFT - '>';
157			goto checkeq;
158	niro	532	default:
159	niro	1122	value += ARITH_GT - '>';
160	niro	532	goto out;
161			}
162			break;
163			case '<':
164			switch (*++buf) {
165			case '=':
166	niro	1122	value += ARITH_LE - '<';
167	niro	532	break;
168			case '<':
169	niro	1122	value += ARITH_LSHIFT - '<';
170			goto checkeq;
171	niro	532	default:
172	niro	1122	value += ARITH_LT - '<';
173	niro	532	goto out;
174			}
175			break;
176			case '\|':
177			if (*++buf != '\|') {
178	niro	1122	value += ARITH_BOR - '\|';
179			goto checkeqcur;
180	niro	532	}
181	niro	1122	value += ARITH_OR - '\|';
182	niro	532	break;
183			case '&':
184			if (*++buf != '&') {
185	niro	1122	value += ARITH_BAND - '&';
186			goto checkeqcur;
187	niro	532	}
188	niro	1122	value += ARITH_AND - '&';
189	niro	532	break;
190			case '!':
191			if (*++buf != '=') {
192	niro	1122	value += ARITH_NOT - '!';
193	niro	532	goto out;
194			}
195	niro	1122	value += ARITH_NE - '!';
196	niro	532	break;
197			case 0:
198			goto out;
199			case '(':
200	niro	1122	value += ARITH_LPAREN - '(';
201	niro	532	break;
202			case ')':
203	niro	1122	value += ARITH_RPAREN - ')';
204	niro	532	break;
205			case '*':
206	niro	1122	value += ARITH_MUL - '*';
207			goto checkeq;
208	niro	532	case '/':
209	niro	1122	value += ARITH_DIV - '/';
210			goto checkeq;
211	niro	532	case '%':
212	niro	1122	value += ARITH_REM - '%';
213			goto checkeq;
214	niro	532	case '+':
215	niro	1122	value += ARITH_ADD - '+';
216			goto checkeq;
217	niro	532	case '-':
218	niro	1122	value += ARITH_SUB - '-';
219			goto checkeq;
220	niro	532	case '~':
221	niro	1122	value += ARITH_BNOT - '~';
222	niro	532	break;
223			case '^':
224	niro	1122	value += ARITH_BXOR - '^';
225			goto checkeq;
226			case '?':
227			value += ARITH_QMARK - '?';
228	niro	532	break;
229	niro	1122	case ':':
230			value += ARITH_COLON - ':';
231			break;
232	niro	532	}
233			break;
234			}
235
236			buf++;
237			out:
238			arith_buf = buf;
239			return value;
240			}