Annotation of src/usr.bin/tr/str.c, Revision 1.8
1.8 ! millert 1: /* $OpenBSD: str.c,v 1.7 2002/02/16 21:27:55 millert Exp $ */
1.1 deraadt 2: /* $NetBSD: str.c,v 1.7 1995/08/31 22:13:47 jtc Exp $ */
3:
4: /*-
5: * Copyright (c) 1991, 1993
6: * The Regents of the University of California. All rights reserved.
7: *
8: * Redistribution and use in source and binary forms, with or without
9: * modification, are permitted provided that the following conditions
10: * are met:
11: * 1. Redistributions of source code must retain the above copyright
12: * notice, this list of conditions and the following disclaimer.
13: * 2. Redistributions in binary form must reproduce the above copyright
14: * notice, this list of conditions and the following disclaimer in the
15: * documentation and/or other materials provided with the distribution.
1.8 ! millert 16: * 3. Neither the name of the University nor the names of its contributors
1.1 deraadt 17: * may be used to endorse or promote products derived from this software
18: * without specific prior written permission.
19: *
20: * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
21: * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23: * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
24: * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25: * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26: * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27: * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28: * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29: * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30: * SUCH DAMAGE.
31: */
32:
33: #ifndef lint
34: #if 0
35: static char sccsid[] = "@(#)str.c 8.2 (Berkeley) 4/28/95";
36: #endif
1.8 ! millert 37: static char rcsid[] = "$OpenBSD: str.c,v 1.7 2002/02/16 21:27:55 millert Exp $";
1.1 deraadt 38: #endif /* not lint */
39:
40: #include <sys/cdefs.h>
41: #include <sys/types.h>
42:
43: #include <errno.h>
44: #include <stddef.h>
45: #include <stdio.h>
46: #include <stdlib.h>
47: #include <string.h>
48: #include <ctype.h>
1.4 mickey 49: #include <err.h>
1.1 deraadt 50:
51: #include "extern.h"
52:
1.7 millert 53: static int backslash(STR *);
54: static int bracket(STR *);
55: static int c_class(const void *, const void *);
56: static void genclass(STR *);
57: static void genequiv(STR *);
58: static int genrange(STR *);
59: static void genseq(STR *);
1.1 deraadt 60:
61: int
62: next(s)
1.6 mpech 63: STR *s;
1.1 deraadt 64: {
1.6 mpech 65: int ch;
1.1 deraadt 66:
67: switch (s->state) {
68: case EOS:
69: return (0);
70: case INFINITE:
71: return (1);
72: case NORMAL:
73: switch (ch = *s->str) {
74: case '\0':
75: s->state = EOS;
76: return (0);
77: case '\\':
78: s->lastch = backslash(s);
79: break;
80: case '[':
81: if (bracket(s))
82: return (next(s));
83: /* FALLTHROUGH */
84: default:
85: ++s->str;
86: s->lastch = ch;
87: break;
88: }
89:
90: /* We can start a range at any time. */
91: if (s->str[0] == '-' && genrange(s))
92: return (next(s));
93: return (1);
94: case RANGE:
95: if (s->cnt-- == 0) {
96: s->state = NORMAL;
97: return (next(s));
98: }
99: ++s->lastch;
100: return (1);
101: case SEQUENCE:
102: if (s->cnt-- == 0) {
103: s->state = NORMAL;
104: return (next(s));
105: }
106: return (1);
107: case SET:
108: if ((s->lastch = s->set[s->cnt++]) == OOBCH) {
109: s->state = NORMAL;
110: return (next(s));
111: }
112: return (1);
1.4 mickey 113: default:
114: return 0;
1.1 deraadt 115: }
116: /* NOTREACHED */
117: }
118:
119: static int
120: bracket(s)
1.6 mpech 121: STR *s;
1.1 deraadt 122: {
1.6 mpech 123: char *p;
1.1 deraadt 124:
125: switch (s->str[1]) {
126: case ':': /* "[:class:]" */
127: if ((p = strstr(s->str + 2, ":]")) == NULL)
128: return (0);
129: *p = '\0';
130: s->str += 2;
131: genclass(s);
132: s->str = p + 2;
133: return (1);
134: case '=': /* "[=equiv=]" */
135: if ((p = strstr(s->str + 2, "=]")) == NULL)
136: return (0);
137: s->str += 2;
138: genequiv(s);
139: return (1);
140: default: /* "[\###*n]" or "[#*n]" */
141: if ((p = strpbrk(s->str + 2, "*]")) == NULL)
142: return (0);
1.3 millert 143: if (p[0] != '*' || strchr(p, ']') == NULL)
1.1 deraadt 144: return (0);
145: s->str += 1;
146: genseq(s);
147: return (1);
148: }
149: /* NOTREACHED */
150: }
151:
152: typedef struct {
153: char *name;
1.7 millert 154: int (*func)(int);
1.1 deraadt 155: int *set;
156: } CLASS;
157:
158: static CLASS classes[] = {
159: { "alnum", isalnum, },
160: { "alpha", isalpha, },
161: { "blank", isblank, },
162: { "cntrl", iscntrl, },
163: { "digit", isdigit, },
164: { "graph", isgraph, },
165: { "lower", islower, },
1.5 deraadt 166: { "print", isprint, },
1.1 deraadt 167: { "punct", ispunct, },
168: { "space", isspace, },
169: { "upper", isupper, },
170: { "xdigit", isxdigit, },
171: };
172:
173: static void
174: genclass(s)
175: STR *s;
176: {
1.7 millert 177: int cnt, (*func)(int);
1.1 deraadt 178: CLASS *cp, tmp;
179: int *p;
180:
181: tmp.name = s->str;
182: if ((cp = (CLASS *)bsearch(&tmp, classes, sizeof(classes) /
183: sizeof(CLASS), sizeof(CLASS), c_class)) == NULL)
1.4 mickey 184: errx(1, "unknown class %s", s->str);
1.1 deraadt 185:
186: if ((cp->set = p = malloc((NCHARS + 1) * sizeof(int))) == NULL)
1.4 mickey 187: errx(1, "no memory for a class");
1.1 deraadt 188: bzero(p, (NCHARS + 1) * sizeof(int));
189: for (cnt = 0, func = cp->func; cnt < NCHARS; ++cnt)
190: if ((func)(cnt))
191: *p++ = cnt;
192: *p = OOBCH;
193:
194: s->cnt = 0;
195: s->state = SET;
196: s->set = cp->set;
197: }
198:
199: static int
200: c_class(a, b)
201: const void *a, *b;
202: {
203: return (strcmp(((CLASS *)a)->name, ((CLASS *)b)->name));
204: }
205:
206: /*
207: * English doesn't have any equivalence classes, so for now
208: * we just syntax check and grab the character.
209: */
210: static void
211: genequiv(s)
212: STR *s;
213: {
214: if (*s->str == '\\') {
215: s->equiv[0] = backslash(s);
216: if (*s->str != '=')
1.4 mickey 217: errx(1, "misplaced equivalence equals sign");
1.1 deraadt 218: } else {
219: s->equiv[0] = s->str[0];
220: if (s->str[1] != '=')
1.4 mickey 221: errx(1, "misplaced equivalence equals sign");
1.1 deraadt 222: }
223: s->str += 2;
224: s->cnt = 0;
225: s->state = SET;
226: s->set = s->equiv;
227: }
228:
229: static int
230: genrange(s)
231: STR *s;
232: {
233: int stopval;
234: char *savestart;
235:
236: savestart = s->str;
237: stopval = *++s->str == '\\' ? backslash(s) : *s->str++;
238: if (stopval < (u_char)s->lastch) {
239: s->str = savestart;
240: return (0);
241: }
242: s->cnt = stopval - s->lastch + 1;
243: s->state = RANGE;
244: --s->lastch;
245: return (1);
246: }
247:
248: static void
249: genseq(s)
250: STR *s;
251: {
252: char *ep;
253:
254: if (s->which == STRING1)
1.4 mickey 255: errx(1, "sequences only valid in string2");
1.1 deraadt 256:
257: if (*s->str == '\\')
258: s->lastch = backslash(s);
259: else
260: s->lastch = *s->str++;
261: if (*s->str != '*')
1.4 mickey 262: errx(1, "misplaced sequence asterisk");
1.1 deraadt 263:
264: switch (*++s->str) {
265: case '\\':
266: s->cnt = backslash(s);
267: break;
268: case ']':
269: s->cnt = 0;
270: ++s->str;
271: break;
272: default:
273: if (isdigit(*s->str)) {
274: s->cnt = strtol(s->str, &ep, 0);
275: if (*ep == ']') {
276: s->str = ep + 1;
277: break;
278: }
279: }
1.4 mickey 280: errx(1, "illegal sequence count");
1.1 deraadt 281: /* NOTREACHED */
282: }
283:
284: s->state = s->cnt ? SEQUENCE : INFINITE;
285: }
286:
287: /*
288: * Translate \??? into a character. Up to 3 octal digits, if no digits either
289: * an escape code or a literal character.
290: */
291: static int
292: backslash(s)
1.6 mpech 293: STR *s;
1.1 deraadt 294: {
1.6 mpech 295: int ch, cnt, val;
1.1 deraadt 296:
297: for (cnt = val = 0;;) {
298: ch = *++s->str;
299: if (!isascii(ch) || !isdigit(ch))
300: break;
301: val = val * 8 + ch - '0';
302: if (++cnt == 3) {
303: ++s->str;
304: break;
305: }
306: }
307: if (cnt)
308: return (val);
309: if (ch != '\0')
310: ++s->str;
311: switch (ch) {
312: case 'a': /* escape characters */
313: return ('\7');
314: case 'b':
315: return ('\b');
316: case 'f':
317: return ('\f');
318: case 'n':
319: return ('\n');
320: case 'r':
321: return ('\r');
322: case 't':
323: return ('\t');
324: case 'v':
325: return ('\13');
326: case '\0': /* \" -> \ */
327: s->state = EOS;
328: return ('\\');
329: default: /* \x" -> x */
330: return (ch);
331: }
332: }