compile.c revision 1.38 1 1.38 msaitoh /* $NetBSD: compile.c,v 1.38 2012/12/14 08:16:51 msaitoh Exp $ */
2 1.15 tls
3 1.1 alm /*-
4 1.9 cgd * Copyright (c) 1992, 1993
5 1.9 cgd * The Regents of the University of California. All rights reserved.
6 1.1 alm *
7 1.1 alm * This code is derived from software contributed to Berkeley by
8 1.1 alm * Diomidis Spinellis of Imperial College, University of London.
9 1.1 alm *
10 1.1 alm * Redistribution and use in source and binary forms, with or without
11 1.1 alm * modification, are permitted provided that the following conditions
12 1.1 alm * are met:
13 1.1 alm * 1. Redistributions of source code must retain the above copyright
14 1.1 alm * notice, this list of conditions and the following disclaimer.
15 1.1 alm * 2. Redistributions in binary form must reproduce the above copyright
16 1.1 alm * notice, this list of conditions and the following disclaimer in the
17 1.1 alm * documentation and/or other materials provided with the distribution.
18 1.25 agc * 3. Neither the name of the University nor the names of its contributors
19 1.25 agc * may be used to endorse or promote products derived from this software
20 1.25 agc * without specific prior written permission.
21 1.25 agc *
22 1.25 agc * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23 1.25 agc * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 1.25 agc * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 1.25 agc * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26 1.25 agc * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 1.25 agc * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 1.25 agc * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 1.25 agc * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 1.25 agc * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 1.25 agc * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 1.25 agc * SUCH DAMAGE.
33 1.25 agc */
34 1.25 agc
35 1.25 agc /*-
36 1.25 agc * Copyright (c) 1992 Diomidis Spinellis.
37 1.25 agc *
38 1.25 agc * This code is derived from software contributed to Berkeley by
39 1.25 agc * Diomidis Spinellis of Imperial College, University of London.
40 1.25 agc *
41 1.25 agc * Redistribution and use in source and binary forms, with or without
42 1.25 agc * modification, are permitted provided that the following conditions
43 1.25 agc * are met:
44 1.25 agc * 1. Redistributions of source code must retain the above copyright
45 1.25 agc * notice, this list of conditions and the following disclaimer.
46 1.25 agc * 2. Redistributions in binary form must reproduce the above copyright
47 1.25 agc * notice, this list of conditions and the following disclaimer in the
48 1.25 agc * documentation and/or other materials provided with the distribution.
49 1.1 alm * 3. All advertising materials mentioning features or use of this software
50 1.1 alm * must display the following acknowledgement:
51 1.1 alm * This product includes software developed by the University of
52 1.1 alm * California, Berkeley and its contributors.
53 1.1 alm * 4. Neither the name of the University nor the names of its contributors
54 1.1 alm * may be used to endorse or promote products derived from this software
55 1.1 alm * without specific prior written permission.
56 1.1 alm *
57 1.1 alm * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
58 1.1 alm * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
59 1.1 alm * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
60 1.1 alm * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
61 1.1 alm * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
62 1.1 alm * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
63 1.1 alm * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
64 1.1 alm * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
65 1.1 alm * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
66 1.1 alm * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
67 1.1 alm * SUCH DAMAGE.
68 1.1 alm */
69 1.1 alm
70 1.34 gdamore #if HAVE_NBTOOL_CONFIG_H
71 1.34 gdamore #include "nbtool_config.h"
72 1.34 gdamore #endif
73 1.34 gdamore
74 1.17 lukem #include <sys/cdefs.h>
75 1.1 alm #ifndef lint
76 1.16 mrg #if 0
77 1.16 mrg static char sccsid[] = "@(#)compile.c 8.2 (Berkeley) 4/28/95";
78 1.16 mrg #else
79 1.38 msaitoh __RCSID("$NetBSD: compile.c,v 1.38 2012/12/14 08:16:51 msaitoh Exp $");
80 1.16 mrg #endif
81 1.1 alm #endif /* not lint */
82 1.1 alm
83 1.1 alm #include <sys/types.h>
84 1.1 alm #include <sys/stat.h>
85 1.1 alm
86 1.1 alm #include <ctype.h>
87 1.1 alm #include <errno.h>
88 1.1 alm #include <fcntl.h>
89 1.1 alm #include <limits.h>
90 1.1 alm #include <regex.h>
91 1.1 alm #include <stdio.h>
92 1.1 alm #include <stdlib.h>
93 1.1 alm #include <string.h>
94 1.1 alm
95 1.1 alm #include "defs.h"
96 1.1 alm #include "extern.h"
97 1.1 alm
98 1.35 christos #ifndef _POSIX2_LINE_MAX
99 1.35 christos #define _POSIX2_LINE_MAX (2 * BUFSIZ)
100 1.35 christos #endif
101 1.35 christos
102 1.9 cgd #define LHSZ 128
103 1.9 cgd #define LHMASK (LHSZ - 1)
104 1.9 cgd static struct labhash {
105 1.9 cgd struct labhash *lh_next;
106 1.9 cgd u_int lh_hash;
107 1.9 cgd struct s_command *lh_cmd;
108 1.9 cgd int lh_ref;
109 1.9 cgd } *labels[LHSZ];
110 1.9 cgd
111 1.24 wiz static char *compile_addr(char *, struct s_addr *);
112 1.24 wiz static char *compile_ccl(char **, char *);
113 1.24 wiz static char *compile_delimited(char *, char *);
114 1.24 wiz static char *compile_flags(char *, struct s_subst *);
115 1.24 wiz static char *compile_re(char *, regex_t **);
116 1.24 wiz static char *compile_subst(char *, struct s_subst *);
117 1.24 wiz static char *compile_text(void);
118 1.24 wiz static char *compile_tr(char *, char **);
119 1.1 alm static struct s_command
120 1.24 wiz **compile_stream(struct s_command **);
121 1.36 lukem static char *duptoeol(char *, const char *);
122 1.24 wiz static void enterlabel(struct s_command *);
123 1.1 alm static struct s_command
124 1.24 wiz *findlabel(char *);
125 1.24 wiz static void fixuplabel(struct s_command *, struct s_command *);
126 1.24 wiz static void uselabel(void);
127 1.1 alm
128 1.1 alm /*
129 1.1 alm * Command specification. This is used to drive the command parser.
130 1.1 alm */
131 1.1 alm struct s_format {
132 1.1 alm char code; /* Command code */
133 1.1 alm int naddr; /* Number of address args */
134 1.1 alm enum e_args args; /* Argument type */
135 1.1 alm };
136 1.1 alm
137 1.1 alm static struct s_format cmd_fmts[] = {
138 1.1 alm {'{', 2, GROUP},
139 1.14 mycroft {'}', 0, ENDGROUP},
140 1.1 alm {'a', 1, TEXT},
141 1.1 alm {'b', 2, BRANCH},
142 1.1 alm {'c', 2, TEXT},
143 1.1 alm {'d', 2, EMPTY},
144 1.1 alm {'D', 2, EMPTY},
145 1.1 alm {'g', 2, EMPTY},
146 1.1 alm {'G', 2, EMPTY},
147 1.1 alm {'h', 2, EMPTY},
148 1.1 alm {'H', 2, EMPTY},
149 1.1 alm {'i', 1, TEXT},
150 1.1 alm {'l', 2, EMPTY},
151 1.1 alm {'n', 2, EMPTY},
152 1.1 alm {'N', 2, EMPTY},
153 1.1 alm {'p', 2, EMPTY},
154 1.1 alm {'P', 2, EMPTY},
155 1.1 alm {'q', 1, EMPTY},
156 1.1 alm {'r', 1, RFILE},
157 1.1 alm {'s', 2, SUBST},
158 1.1 alm {'t', 2, BRANCH},
159 1.1 alm {'w', 2, WFILE},
160 1.1 alm {'x', 2, EMPTY},
161 1.1 alm {'y', 2, TR},
162 1.1 alm {'!', 2, NONSEL},
163 1.1 alm {':', 0, LABEL},
164 1.1 alm {'#', 0, COMMENT},
165 1.1 alm {'=', 1, EMPTY},
166 1.1 alm {'\0', 0, COMMENT},
167 1.1 alm };
168 1.1 alm
169 1.1 alm /* The compiled program. */
170 1.1 alm struct s_command *prog;
171 1.1 alm
172 1.1 alm /*
173 1.1 alm * Compile the program into prog.
174 1.1 alm * Initialise appends.
175 1.1 alm */
176 1.1 alm void
177 1.24 wiz compile(void)
178 1.1 alm {
179 1.14 mycroft *compile_stream(&prog) = NULL;
180 1.9 cgd fixuplabel(prog, NULL);
181 1.9 cgd uselabel();
182 1.18 drochner if (appendnum > 0)
183 1.18 drochner appends = xmalloc(sizeof(struct s_appends) * appendnum);
184 1.1 alm match = xmalloc((maxnsub + 1) * sizeof(regmatch_t));
185 1.1 alm }
186 1.1 alm
187 1.32 christos #define EATSPACE() \
188 1.32 christos while (*p && isascii((unsigned char)*p) && \
189 1.32 christos isspace((unsigned char)*p)) \
190 1.32 christos p++ \
191 1.1 alm
192 1.1 alm static struct s_command **
193 1.24 wiz compile_stream(struct s_command **link)
194 1.14 mycroft {
195 1.17 lukem char *p;
196 1.37 tnn static char *lbuf; /* To avoid excessive malloc calls */
197 1.37 tnn static size_t bufsize;
198 1.14 mycroft struct s_command *cmd, *cmd2, *stack;
199 1.1 alm struct s_format *fp;
200 1.1 alm int naddr; /* Number of addresses */
201 1.1 alm
202 1.14 mycroft stack = 0;
203 1.1 alm for (;;) {
204 1.37 tnn if ((p = cu_fgets(&lbuf, &bufsize)) == NULL) {
205 1.14 mycroft if (stack != 0)
206 1.1 alm err(COMPILE, "unexpected EOF (pending }'s)");
207 1.1 alm return (link);
208 1.1 alm }
209 1.1 alm
210 1.1 alm semicolon: EATSPACE();
211 1.32 christos if (*p == '#' || *p == '\0')
212 1.32 christos continue;
213 1.32 christos else if (*p == ';') {
214 1.32 christos p++;
215 1.32 christos goto semicolon;
216 1.20 kleink }
217 1.1 alm *link = cmd = xmalloc(sizeof(struct s_command));
218 1.1 alm link = &cmd->next;
219 1.1 alm cmd->nonsel = cmd->inrange = 0;
220 1.1 alm /* First parse the addresses */
221 1.1 alm naddr = 0;
222 1.1 alm
223 1.1 alm /* Valid characters to start an address */
224 1.1 alm #define addrchar(c) (strchr("0123456789/\\$", (c)))
225 1.1 alm if (addrchar(*p)) {
226 1.1 alm naddr++;
227 1.1 alm cmd->a1 = xmalloc(sizeof(struct s_addr));
228 1.1 alm p = compile_addr(p, cmd->a1);
229 1.1 alm EATSPACE(); /* EXTENSION */
230 1.1 alm if (*p == ',') {
231 1.1 alm p++;
232 1.1 alm EATSPACE(); /* EXTENSION */
233 1.13 mycroft naddr++;
234 1.1 alm cmd->a2 = xmalloc(sizeof(struct s_addr));
235 1.1 alm p = compile_addr(p, cmd->a2);
236 1.13 mycroft EATSPACE();
237 1.12 mycroft } else
238 1.12 mycroft cmd->a2 = 0;
239 1.12 mycroft } else
240 1.13 mycroft cmd->a1 = cmd->a2 = 0;
241 1.1 alm
242 1.1 alm nonsel: /* Now parse the command */
243 1.1 alm if (!*p)
244 1.1 alm err(COMPILE, "command expected");
245 1.1 alm cmd->code = *p;
246 1.1 alm for (fp = cmd_fmts; fp->code; fp++)
247 1.1 alm if (fp->code == *p)
248 1.1 alm break;
249 1.1 alm if (!fp->code)
250 1.1 alm err(COMPILE, "invalid command code %c", *p);
251 1.1 alm if (naddr > fp->naddr)
252 1.1 alm err(COMPILE,
253 1.1 alm "command %c expects up to %d address(es), found %d", *p, fp->naddr, naddr);
254 1.1 alm switch (fp->args) {
255 1.1 alm case NONSEL: /* ! */
256 1.13 mycroft p++;
257 1.13 mycroft EATSPACE();
258 1.1 alm cmd->nonsel = ! cmd->nonsel;
259 1.1 alm goto nonsel;
260 1.1 alm case GROUP: /* { */
261 1.1 alm p++;
262 1.1 alm EATSPACE();
263 1.14 mycroft cmd->next = stack;
264 1.14 mycroft stack = cmd;
265 1.14 mycroft link = &cmd->u.c;
266 1.14 mycroft if (*p)
267 1.14 mycroft goto semicolon;
268 1.14 mycroft break;
269 1.14 mycroft case ENDGROUP:
270 1.12 mycroft /*
271 1.12 mycroft * Short-circuit command processing, since end of
272 1.12 mycroft * group is really just a noop.
273 1.12 mycroft */
274 1.14 mycroft cmd->nonsel = 1;
275 1.14 mycroft if (stack == 0)
276 1.14 mycroft err(COMPILE, "unexpected }");
277 1.14 mycroft cmd2 = stack;
278 1.14 mycroft stack = cmd2->next;
279 1.14 mycroft cmd2->next = cmd;
280 1.14 mycroft /*FALLTHROUGH*/
281 1.1 alm case EMPTY: /* d D g G h H l n N p P q x = \0 */
282 1.1 alm p++;
283 1.1 alm EATSPACE();
284 1.1 alm if (*p == ';') {
285 1.1 alm p++;
286 1.1 alm link = &cmd->next;
287 1.1 alm goto semicolon;
288 1.1 alm }
289 1.1 alm if (*p)
290 1.1 alm err(COMPILE,
291 1.1 alm "extra characters at the end of %c command", cmd->code);
292 1.1 alm break;
293 1.1 alm case TEXT: /* a c i */
294 1.1 alm p++;
295 1.1 alm EATSPACE();
296 1.1 alm if (*p != '\\')
297 1.1 alm err(COMPILE,
298 1.1 alm "command %c expects \\ followed by text", cmd->code);
299 1.1 alm p++;
300 1.1 alm EATSPACE();
301 1.1 alm if (*p)
302 1.1 alm err(COMPILE,
303 1.1 alm "extra characters after \\ at the end of %c command", cmd->code);
304 1.1 alm cmd->t = compile_text();
305 1.1 alm break;
306 1.1 alm case COMMENT: /* \0 # */
307 1.1 alm break;
308 1.1 alm case WFILE: /* w */
309 1.1 alm p++;
310 1.1 alm EATSPACE();
311 1.1 alm if (*p == '\0')
312 1.1 alm err(COMPILE, "filename expected");
313 1.9 cgd cmd->t = duptoeol(p, "w command");
314 1.1 alm if (aflag)
315 1.1 alm cmd->u.fd = -1;
316 1.1 alm else if ((cmd->u.fd = open(p,
317 1.1 alm O_WRONLY|O_APPEND|O_CREAT|O_TRUNC,
318 1.1 alm DEFFILEMODE)) == -1)
319 1.23 itojun err(FATAL, "%s: %s", p, strerror(errno));
320 1.1 alm break;
321 1.1 alm case RFILE: /* r */
322 1.1 alm p++;
323 1.1 alm EATSPACE();
324 1.1 alm if (*p == '\0')
325 1.1 alm err(COMPILE, "filename expected");
326 1.1 alm else
327 1.9 cgd cmd->t = duptoeol(p, "read command");
328 1.1 alm break;
329 1.1 alm case BRANCH: /* b t */
330 1.1 alm p++;
331 1.1 alm EATSPACE();
332 1.1 alm if (*p == '\0')
333 1.1 alm cmd->t = NULL;
334 1.1 alm else
335 1.9 cgd cmd->t = duptoeol(p, "branch");
336 1.1 alm break;
337 1.1 alm case LABEL: /* : */
338 1.1 alm p++;
339 1.1 alm EATSPACE();
340 1.9 cgd cmd->t = duptoeol(p, "label");
341 1.1 alm if (strlen(p) == 0)
342 1.1 alm err(COMPILE, "empty label");
343 1.9 cgd enterlabel(cmd);
344 1.1 alm break;
345 1.1 alm case SUBST: /* s */
346 1.1 alm p++;
347 1.1 alm if (*p == '\0' || *p == '\\')
348 1.1 alm err(COMPILE,
349 1.1 alm "substitute pattern can not be delimited by newline or backslash");
350 1.1 alm cmd->u.s = xmalloc(sizeof(struct s_subst));
351 1.1 alm p = compile_re(p, &cmd->u.s->re);
352 1.1 alm if (p == NULL)
353 1.1 alm err(COMPILE, "unterminated substitute pattern");
354 1.1 alm --p;
355 1.1 alm p = compile_subst(p, cmd->u.s);
356 1.1 alm p = compile_flags(p, cmd->u.s);
357 1.1 alm EATSPACE();
358 1.1 alm if (*p == ';') {
359 1.1 alm p++;
360 1.1 alm link = &cmd->next;
361 1.1 alm goto semicolon;
362 1.1 alm }
363 1.1 alm break;
364 1.1 alm case TR: /* y */
365 1.1 alm p++;
366 1.33 mrg p = compile_tr(p, (char **)(void *)&cmd->u.y);
367 1.1 alm EATSPACE();
368 1.1 alm if (*p == ';') {
369 1.1 alm p++;
370 1.1 alm link = &cmd->next;
371 1.1 alm goto semicolon;
372 1.1 alm }
373 1.1 alm if (*p)
374 1.1 alm err(COMPILE,
375 1.1 alm "extra text at the end of a transform command");
376 1.1 alm break;
377 1.1 alm }
378 1.1 alm }
379 1.1 alm }
380 1.1 alm
381 1.1 alm /*
382 1.22 wiz * Get a delimited string. P points to the delimiter of the string; d points
383 1.1 alm * to a buffer area. Newline and delimiter escapes are processed; other
384 1.1 alm * escapes are ignored.
385 1.1 alm *
386 1.1 alm * Returns a pointer to the first character after the final delimiter or NULL
387 1.1 alm * in the case of a non-terminated string. The character array d is filled
388 1.1 alm * with the processed string.
389 1.1 alm */
390 1.1 alm static char *
391 1.24 wiz compile_delimited(char *p, char *d)
392 1.1 alm {
393 1.1 alm char c;
394 1.1 alm
395 1.1 alm c = *p++;
396 1.1 alm if (c == '\0')
397 1.1 alm return (NULL);
398 1.1 alm else if (c == '\\')
399 1.1 alm err(COMPILE, "\\ can not be used as a string delimiter");
400 1.1 alm else if (c == '\n')
401 1.1 alm err(COMPILE, "newline can not be used as a string delimiter");
402 1.1 alm while (*p) {
403 1.11 alm if (*p == '[') {
404 1.11 alm if ((d = compile_ccl(&p, d)) == NULL)
405 1.11 alm err(COMPILE, "unbalanced brackets ([])");
406 1.11 alm continue;
407 1.11 alm } else if (*p == '\\' && p[1] == '[') {
408 1.11 alm *d++ = *p++;
409 1.11 alm } else if (*p == '\\' && p[1] == c)
410 1.1 alm p++;
411 1.1 alm else if (*p == '\\' && p[1] == 'n') {
412 1.1 alm *d++ = '\n';
413 1.1 alm p += 2;
414 1.1 alm continue;
415 1.1 alm } else if (*p == '\\' && p[1] == '\\')
416 1.1 alm *d++ = *p++;
417 1.1 alm else if (*p == c) {
418 1.1 alm *d = '\0';
419 1.1 alm return (p + 1);
420 1.1 alm }
421 1.1 alm *d++ = *p++;
422 1.1 alm }
423 1.1 alm return (NULL);
424 1.11 alm }
425 1.11 alm
426 1.11 alm
427 1.11 alm /* compile_ccl: expand a POSIX character class */
428 1.11 alm static char *
429 1.24 wiz compile_ccl(char **sp, char *t)
430 1.11 alm {
431 1.11 alm int c, d;
432 1.11 alm char *s = *sp;
433 1.11 alm
434 1.11 alm *t++ = *s++;
435 1.11 alm if (*s == '^')
436 1.11 alm *t++ = *s++;
437 1.11 alm if (*s == ']')
438 1.11 alm *t++ = *s++;
439 1.11 alm for (; *s && (*t = *s) != ']'; s++, t++)
440 1.11 alm if (*s == '[' && ((d = *(s+1)) == '.' || d == ':' || d == '=')) {
441 1.11 alm *++t = *++s, t++, s++;
442 1.11 alm for (c = *s; (*t = *s) != ']' || c != d; s++, t++)
443 1.11 alm if ((c = *s) == '\0')
444 1.11 alm return NULL;
445 1.11 alm } else if (*s == '\\' && s[1] == 'n')
446 1.11 alm *t = '\n', s++;
447 1.11 alm return (*s == ']') ? *sp = ++s, ++t : NULL;
448 1.2 alm }
449 1.2 alm
450 1.1 alm /*
451 1.1 alm * Get a regular expression. P points to the delimiter of the regular
452 1.1 alm * expression; repp points to the address of a regexp pointer. Newline
453 1.1 alm * and delimiter escapes are processed; other escapes are ignored.
454 1.1 alm * Returns a pointer to the first character after the final delimiter
455 1.1 alm * or NULL in the case of a non terminated regular expression. The regexp
456 1.1 alm * pointer is set to the compiled regular expression.
457 1.1 alm * Cflags are passed to regcomp.
458 1.1 alm */
459 1.1 alm static char *
460 1.24 wiz compile_re(char *p, regex_t **repp)
461 1.1 alm {
462 1.1 alm int eval;
463 1.37 tnn char *re;
464 1.1 alm
465 1.37 tnn re = xmalloc(strlen(p) + 1); /* strlen(re) <= strlen(p) */
466 1.1 alm p = compile_delimited(p, re);
467 1.1 alm if (p && strlen(re) == 0) {
468 1.1 alm *repp = NULL;
469 1.37 tnn free(re);
470 1.1 alm return (p);
471 1.1 alm }
472 1.1 alm *repp = xmalloc(sizeof(regex_t));
473 1.21 atatat if (p && (eval = regcomp(*repp, re, ere)) != 0)
474 1.1 alm err(COMPILE, "RE error: %s", strregerror(eval, *repp));
475 1.1 alm if (maxnsub < (*repp)->re_nsub)
476 1.1 alm maxnsub = (*repp)->re_nsub;
477 1.38 msaitoh free(re);
478 1.1 alm return (p);
479 1.1 alm }
480 1.1 alm
481 1.1 alm /*
482 1.1 alm * Compile the substitution string of a regular expression and set res to
483 1.1 alm * point to a saved copy of it. Nsub is the number of parenthesized regular
484 1.1 alm * expressions.
485 1.1 alm */
486 1.1 alm static char *
487 1.24 wiz compile_subst(char *p, struct s_subst *s)
488 1.1 alm {
489 1.37 tnn static char *lbuf;
490 1.37 tnn static size_t bufsize;
491 1.37 tnn int asize, ref, size, len;
492 1.1 alm char c, *text, *op, *sp;
493 1.26 minskim int sawesc = 0;
494 1.1 alm
495 1.1 alm c = *p++; /* Terminator character */
496 1.1 alm if (c == '\0')
497 1.1 alm return (NULL);
498 1.1 alm
499 1.1 alm s->maxbref = 0;
500 1.1 alm s->linenum = linenum;
501 1.37 tnn text = NULL;
502 1.37 tnn asize = size = 0;
503 1.1 alm do {
504 1.37 tnn len = ROUNDLEN(strlen(p) + 1);
505 1.37 tnn if (asize - size < len) {
506 1.37 tnn do {
507 1.37 tnn asize += len;
508 1.37 tnn } while (asize - size < len);
509 1.37 tnn text = xrealloc(text, asize);
510 1.37 tnn }
511 1.1 alm op = sp = text + size;
512 1.1 alm for (; *p; p++) {
513 1.26 minskim if (*p == '\\' || sawesc) {
514 1.26 minskim /*
515 1.26 minskim * If this is a continuation from the last
516 1.26 minskim * buffer, we won't have a character to
517 1.26 minskim * skip over.
518 1.26 minskim */
519 1.26 minskim if (sawesc)
520 1.26 minskim sawesc = 0;
521 1.26 minskim else
522 1.26 minskim p++;
523 1.26 minskim
524 1.26 minskim if (*p == '\0') {
525 1.26 minskim /*
526 1.26 minskim * This escaped character is continued
527 1.26 minskim * in the next part of the line. Note
528 1.26 minskim * this fact, then cause the loop to
529 1.26 minskim * exit w/ normal EOL case and reenter
530 1.26 minskim * above with the new buffer.
531 1.26 minskim */
532 1.26 minskim sawesc = 1;
533 1.26 minskim p--;
534 1.26 minskim continue;
535 1.26 minskim } else if (strchr("123456789", *p) != NULL) {
536 1.1 alm *sp++ = '\\';
537 1.1 alm ref = *p - '0';
538 1.1 alm if (s->re != NULL &&
539 1.36 lukem (size_t)ref > s->re->re_nsub)
540 1.1 alm err(COMPILE,
541 1.1 alm "\\%c not defined in the RE", *p);
542 1.1 alm if (s->maxbref < ref)
543 1.1 alm s->maxbref = ref;
544 1.1 alm } else if (*p == '&' || *p == '\\')
545 1.1 alm *sp++ = '\\';
546 1.1 alm } else if (*p == c) {
547 1.1 alm p++;
548 1.1 alm *sp++ = '\0';
549 1.1 alm size += sp - op;
550 1.1 alm s->new = xrealloc(text, size);
551 1.1 alm return (p);
552 1.1 alm } else if (*p == '\n') {
553 1.1 alm err(COMPILE,
554 1.1 alm "unescaped newline inside substitute pattern");
555 1.1 alm /* NOTREACHED */
556 1.1 alm }
557 1.1 alm *sp++ = *p;
558 1.1 alm }
559 1.1 alm size += sp - op;
560 1.37 tnn } while ((p = cu_fgets(&lbuf, &bufsize)));
561 1.1 alm err(COMPILE, "unterminated substitute in regular expression");
562 1.1 alm /* NOTREACHED */
563 1.17 lukem return (NULL);
564 1.1 alm }
565 1.1 alm
566 1.1 alm /*
567 1.1 alm * Compile the flags of the s command
568 1.1 alm */
569 1.1 alm static char *
570 1.24 wiz compile_flags(char *p, struct s_subst *s)
571 1.1 alm {
572 1.1 alm int gn; /* True if we have seen g or n */
573 1.37 tnn char wfile[PATH_MAX], *q;
574 1.1 alm
575 1.1 alm s->n = 1; /* Default */
576 1.1 alm s->p = 0;
577 1.1 alm s->wfile = NULL;
578 1.1 alm s->wfd = -1;
579 1.1 alm for (gn = 0;;) {
580 1.1 alm EATSPACE(); /* EXTENSION */
581 1.1 alm switch (*p) {
582 1.1 alm case 'g':
583 1.1 alm if (gn)
584 1.1 alm err(COMPILE,
585 1.1 alm "more than one number or 'g' in substitute flags");
586 1.1 alm gn = 1;
587 1.1 alm s->n = 0;
588 1.1 alm break;
589 1.1 alm case '\0':
590 1.1 alm case '\n':
591 1.1 alm case ';':
592 1.1 alm return (p);
593 1.1 alm case 'p':
594 1.1 alm s->p = 1;
595 1.1 alm break;
596 1.1 alm case '1': case '2': case '3':
597 1.1 alm case '4': case '5': case '6':
598 1.1 alm case '7': case '8': case '9':
599 1.1 alm if (gn)
600 1.1 alm err(COMPILE,
601 1.1 alm "more than one number or 'g' in substitute flags");
602 1.1 alm gn = 1;
603 1.1 alm /* XXX Check for overflow */
604 1.1 alm s->n = (int)strtol(p, &p, 10);
605 1.27 grant p--;
606 1.1 alm break;
607 1.1 alm case 'w':
608 1.1 alm p++;
609 1.9 cgd #ifdef HISTORIC_PRACTICE
610 1.1 alm if (*p != ' ') {
611 1.1 alm err(WARNING, "space missing before w wfile");
612 1.1 alm return (p);
613 1.1 alm }
614 1.1 alm #endif
615 1.1 alm EATSPACE();
616 1.1 alm q = wfile;
617 1.1 alm while (*p) {
618 1.1 alm if (*p == '\n')
619 1.1 alm break;
620 1.1 alm *q++ = *p++;
621 1.1 alm }
622 1.1 alm *q = '\0';
623 1.1 alm if (q == wfile)
624 1.1 alm err(COMPILE, "no wfile specified");
625 1.1 alm s->wfile = strdup(wfile);
626 1.1 alm if (!aflag && (s->wfd = open(wfile,
627 1.1 alm O_WRONLY|O_APPEND|O_CREAT|O_TRUNC,
628 1.1 alm DEFFILEMODE)) == -1)
629 1.23 itojun err(FATAL, "%s: %s", wfile, strerror(errno));
630 1.1 alm return (p);
631 1.1 alm default:
632 1.1 alm err(COMPILE,
633 1.1 alm "bad flag in substitute command: '%c'", *p);
634 1.1 alm break;
635 1.1 alm }
636 1.1 alm p++;
637 1.1 alm }
638 1.1 alm }
639 1.1 alm
640 1.1 alm /*
641 1.1 alm * Compile a translation set of strings into a lookup table.
642 1.1 alm */
643 1.1 alm static char *
644 1.24 wiz compile_tr(char *p, char **transtab)
645 1.1 alm {
646 1.1 alm int i;
647 1.1 alm char *lt, *op, *np;
648 1.37 tnn char *old = NULL, *new = NULL;
649 1.1 alm
650 1.1 alm if (*p == '\0' || *p == '\\')
651 1.1 alm err(COMPILE,
652 1.1 alm "transform pattern can not be delimited by newline or backslash");
653 1.37 tnn old = xmalloc(strlen(p) + 1);
654 1.1 alm p = compile_delimited(p, old);
655 1.1 alm if (p == NULL) {
656 1.1 alm err(COMPILE, "unterminated transform source string");
657 1.37 tnn goto bad;
658 1.1 alm }
659 1.37 tnn new = xmalloc(strlen(p) + 1);
660 1.1 alm p = compile_delimited(--p, new);
661 1.1 alm if (p == NULL) {
662 1.1 alm err(COMPILE, "unterminated transform target string");
663 1.37 tnn goto bad;
664 1.1 alm }
665 1.1 alm EATSPACE();
666 1.1 alm if (strlen(new) != strlen(old)) {
667 1.1 alm err(COMPILE, "transform strings are not the same length");
668 1.37 tnn goto bad;
669 1.1 alm }
670 1.1 alm /* We assume characters are 8 bits */
671 1.28 grant lt = xmalloc(UCHAR_MAX+1);
672 1.1 alm for (i = 0; i <= UCHAR_MAX; i++)
673 1.1 alm lt[i] = (char)i;
674 1.1 alm for (op = old, np = new; *op; op++, np++)
675 1.1 alm lt[(u_char)*op] = *np;
676 1.1 alm *transtab = lt;
677 1.37 tnn free(old);
678 1.37 tnn free(new);
679 1.1 alm return (p);
680 1.37 tnn bad:
681 1.37 tnn free(old);
682 1.37 tnn free(new);
683 1.37 tnn return (NULL);
684 1.1 alm }
685 1.1 alm
686 1.1 alm /*
687 1.31 matt * Compile the text following an a, c, or i command.
688 1.1 alm */
689 1.1 alm static char *
690 1.24 wiz compile_text(void)
691 1.1 alm {
692 1.37 tnn int asize, size, len;
693 1.37 tnn char *lbuf, *text, *p, *op, *s;
694 1.37 tnn size_t bufsize;
695 1.37 tnn
696 1.37 tnn lbuf = text = NULL;
697 1.37 tnn asize = size = 0;
698 1.37 tnn while ((p = cu_fgets(&lbuf, &bufsize))) {
699 1.37 tnn len = ROUNDLEN(strlen(p) + 1);
700 1.37 tnn if (asize - size < len) {
701 1.37 tnn do {
702 1.37 tnn asize += len;
703 1.37 tnn } while (asize - size < len);
704 1.37 tnn text = xrealloc(text, asize);
705 1.37 tnn }
706 1.1 alm op = s = text + size;
707 1.1 alm for (; *p; p++) {
708 1.1 alm if (*p == '\\')
709 1.1 alm p++;
710 1.1 alm *s++ = *p;
711 1.1 alm }
712 1.1 alm size += s - op;
713 1.1 alm if (p[-2] != '\\') {
714 1.1 alm *s = '\0';
715 1.1 alm break;
716 1.1 alm }
717 1.1 alm }
718 1.37 tnn free(lbuf);
719 1.1 alm return (xrealloc(text, size + 1));
720 1.1 alm }
721 1.1 alm
722 1.1 alm /*
723 1.1 alm * Get an address and return a pointer to the first character after
724 1.1 alm * it. Fill the structure pointed to according to the address.
725 1.1 alm */
726 1.1 alm static char *
727 1.24 wiz compile_addr(char *p, struct s_addr *a)
728 1.1 alm {
729 1.1 alm char *end;
730 1.1 alm
731 1.1 alm switch (*p) {
732 1.1 alm case '\\': /* Context address */
733 1.1 alm ++p;
734 1.1 alm /* FALLTHROUGH */
735 1.1 alm case '/': /* Context address */
736 1.1 alm p = compile_re(p, &a->u.r);
737 1.1 alm if (p == NULL)
738 1.1 alm err(COMPILE, "unterminated regular expression");
739 1.1 alm a->type = AT_RE;
740 1.1 alm return (p);
741 1.1 alm
742 1.1 alm case '$': /* Last line */
743 1.1 alm a->type = AT_LAST;
744 1.1 alm return (p + 1);
745 1.1 alm /* Line number */
746 1.1 alm case '0': case '1': case '2': case '3': case '4':
747 1.1 alm case '5': case '6': case '7': case '8': case '9':
748 1.1 alm a->type = AT_LINE;
749 1.1 alm a->u.l = strtol(p, &end, 10);
750 1.1 alm return (end);
751 1.1 alm default:
752 1.1 alm err(COMPILE, "expected context address");
753 1.1 alm return (NULL);
754 1.1 alm }
755 1.1 alm }
756 1.1 alm
757 1.1 alm /*
758 1.9 cgd * duptoeol --
759 1.9 cgd * Return a copy of all the characters up to \n or \0.
760 1.1 alm */
761 1.1 alm static char *
762 1.36 lukem duptoeol(char *s, const char *ctype)
763 1.1 alm {
764 1.1 alm size_t len;
765 1.9 cgd int ws;
766 1.1 alm char *start;
767 1.1 alm
768 1.9 cgd ws = 0;
769 1.9 cgd for (start = s; *s != '\0' && *s != '\n'; ++s)
770 1.19 christos ws = isspace((unsigned char)*s);
771 1.1 alm *s = '\0';
772 1.9 cgd if (ws)
773 1.9 cgd err(WARNING, "whitespace after %s", ctype);
774 1.1 alm len = s - start + 1;
775 1.1 alm return (memmove(xmalloc(len), start, len));
776 1.1 alm }
777 1.1 alm
778 1.1 alm /*
779 1.9 cgd * Convert goto label names to addresses, and count a and r commands, in
780 1.9 cgd * the given subset of the script. Free the memory used by labels in b
781 1.9 cgd * and t commands (but not by :).
782 1.9 cgd *
783 1.1 alm * TODO: Remove } nodes
784 1.1 alm */
785 1.1 alm static void
786 1.24 wiz fixuplabel(struct s_command *cp, struct s_command *end)
787 1.1 alm {
788 1.1 alm
789 1.1 alm for (; cp != end; cp = cp->next)
790 1.1 alm switch (cp->code) {
791 1.1 alm case 'a':
792 1.1 alm case 'r':
793 1.1 alm appendnum++;
794 1.1 alm break;
795 1.1 alm case 'b':
796 1.1 alm case 't':
797 1.9 cgd /* Resolve branch target. */
798 1.1 alm if (cp->t == NULL) {
799 1.1 alm cp->u.c = NULL;
800 1.1 alm break;
801 1.1 alm }
802 1.9 cgd if ((cp->u.c = findlabel(cp->t)) == NULL)
803 1.1 alm err(COMPILE2, "undefined label '%s'", cp->t);
804 1.1 alm free(cp->t);
805 1.1 alm break;
806 1.1 alm case '{':
807 1.9 cgd /* Do interior commands. */
808 1.9 cgd fixuplabel(cp->u.c, cp->next);
809 1.1 alm break;
810 1.1 alm }
811 1.9 cgd }
812 1.9 cgd
813 1.9 cgd /*
814 1.9 cgd * Associate the given command label for later lookup.
815 1.9 cgd */
816 1.9 cgd static void
817 1.24 wiz enterlabel(struct s_command *cp)
818 1.9 cgd {
819 1.17 lukem struct labhash **lhp, *lh;
820 1.17 lukem u_char *p;
821 1.17 lukem u_int h, c;
822 1.9 cgd
823 1.9 cgd for (h = 0, p = (u_char *)cp->t; (c = *p) != 0; p++)
824 1.9 cgd h = (h << 5) + h + c;
825 1.9 cgd lhp = &labels[h & LHMASK];
826 1.9 cgd for (lh = *lhp; lh != NULL; lh = lh->lh_next)
827 1.9 cgd if (lh->lh_hash == h && strcmp(cp->t, lh->lh_cmd->t) == 0)
828 1.9 cgd err(COMPILE2, "duplicate label '%s'", cp->t);
829 1.9 cgd lh = xmalloc(sizeof *lh);
830 1.9 cgd lh->lh_next = *lhp;
831 1.9 cgd lh->lh_hash = h;
832 1.9 cgd lh->lh_cmd = cp;
833 1.9 cgd lh->lh_ref = 0;
834 1.9 cgd *lhp = lh;
835 1.9 cgd }
836 1.9 cgd
837 1.9 cgd /*
838 1.9 cgd * Find the label contained in the command l in the command linked
839 1.9 cgd * list cp. L is excluded from the search. Return NULL if not found.
840 1.9 cgd */
841 1.9 cgd static struct s_command *
842 1.24 wiz findlabel(char *name)
843 1.9 cgd {
844 1.17 lukem struct labhash *lh;
845 1.17 lukem u_char *p;
846 1.17 lukem u_int h, c;
847 1.9 cgd
848 1.9 cgd for (h = 0, p = (u_char *)name; (c = *p) != 0; p++)
849 1.9 cgd h = (h << 5) + h + c;
850 1.9 cgd for (lh = labels[h & LHMASK]; lh != NULL; lh = lh->lh_next) {
851 1.9 cgd if (lh->lh_hash == h && strcmp(name, lh->lh_cmd->t) == 0) {
852 1.9 cgd lh->lh_ref = 1;
853 1.9 cgd return (lh->lh_cmd);
854 1.9 cgd }
855 1.9 cgd }
856 1.9 cgd return (NULL);
857 1.9 cgd }
858 1.9 cgd
859 1.9 cgd /*
860 1.9 cgd * Warn about any unused labels. As a side effect, release the label hash
861 1.9 cgd * table space.
862 1.9 cgd */
863 1.9 cgd static void
864 1.24 wiz uselabel(void)
865 1.9 cgd {
866 1.17 lukem struct labhash *lh, *next;
867 1.17 lukem int i;
868 1.9 cgd
869 1.9 cgd for (i = 0; i < LHSZ; i++) {
870 1.9 cgd for (lh = labels[i]; lh != NULL; lh = next) {
871 1.9 cgd next = lh->lh_next;
872 1.9 cgd if (!lh->lh_ref)
873 1.9 cgd err(WARNING, "unused label '%s'",
874 1.9 cgd lh->lh_cmd->t);
875 1.9 cgd free(lh);
876 1.9 cgd }
877 1.9 cgd }
878 1.1 alm }
879