2 c734c0e9 2021-08-03 op * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
3 c734c0e9 2021-08-03 op * Copyright (c) 2018 Florian Obser <florian@openbsd.org>
4 c734c0e9 2021-08-03 op * Copyright (c) 2004, 2005 Esben Norby <norby@openbsd.org>
5 c734c0e9 2021-08-03 op * Copyright (c) 2004 Ryan McBride <mcbride@openbsd.org>
6 c734c0e9 2021-08-03 op * Copyright (c) 2002, 2003, 2004 Henning Brauer <henning@openbsd.org>
7 c734c0e9 2021-08-03 op * Copyright (c) 2001 Markus Friedl. All rights reserved.
8 c734c0e9 2021-08-03 op * Copyright (c) 2001 Daniel Hartmeier. All rights reserved.
9 c734c0e9 2021-08-03 op * Copyright (c) 2001 Theo de Raadt. All rights reserved.
11 c734c0e9 2021-08-03 op * Permission to use, copy, modify, and distribute this software for any
12 c734c0e9 2021-08-03 op * purpose with or without fee is hereby granted, provided that the above
13 c734c0e9 2021-08-03 op * copyright notice and this permission notice appear in all copies.
15 c734c0e9 2021-08-03 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
16 c734c0e9 2021-08-03 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
17 c734c0e9 2021-08-03 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
18 c734c0e9 2021-08-03 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
19 c734c0e9 2021-08-03 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
20 c734c0e9 2021-08-03 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
21 c734c0e9 2021-08-03 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
26 c734c0e9 2021-08-03 op #include "compat.h"
28 c734c0e9 2021-08-03 op #include <ctype.h>
29 ef33a69e 2021-08-07 op #include <errno.h>
30 c734c0e9 2021-08-03 op #include <inttypes.h>
31 ef33a69e 2021-08-07 op #include <limits.h>
32 c734c0e9 2021-08-03 op #include <limits.h>
33 c734c0e9 2021-08-03 op #include <stdio.h>
34 c734c0e9 2021-08-03 op #include <stdlib.h>
35 c734c0e9 2021-08-03 op #include <string.h>
36 c734c0e9 2021-08-03 op #include <syslog.h>
38 c734c0e9 2021-08-03 op #include "log.h"
39 c734c0e9 2021-08-03 op #include "kamid.h"
40 c734c0e9 2021-08-03 op #include "utils.h"
42 c734c0e9 2021-08-03 op #include "script.h"
44 c734c0e9 2021-08-03 op TAILQ_HEAD(files, file) files = TAILQ_HEAD_INITIALIZER(files);
45 c734c0e9 2021-08-03 op static struct file {
46 c734c0e9 2021-08-03 op TAILQ_ENTRY(file) entry;
49 c734c0e9 2021-08-03 op size_t ungetpos;
50 c734c0e9 2021-08-03 op size_t ungetsize;
51 c734c0e9 2021-08-03 op u_char *ungetbuf;
52 c734c0e9 2021-08-03 op int eof_reached;
55 c734c0e9 2021-08-03 op } *file, *topfile;
56 c734c0e9 2021-08-03 op struct file *pushfile(const char *);
57 c734c0e9 2021-08-03 op int popfile(void);
58 c734c0e9 2021-08-03 op int yyparse(void);
59 c734c0e9 2021-08-03 op int yylex(void);
60 c734c0e9 2021-08-03 op int yyerror(const char *, ...)
61 c734c0e9 2021-08-03 op __attribute__((__format__ (printf, 1, 2)))
62 c734c0e9 2021-08-03 op __attribute__((__nonnull__ (1)));
63 c734c0e9 2021-08-03 op int kw_cmp(const void *, const void *);
64 c734c0e9 2021-08-03 op int lookup(char *);
65 c734c0e9 2021-08-03 op int igetc(void);
66 c734c0e9 2021-08-03 op int lgetc(int);
67 c734c0e9 2021-08-03 op void lungetc(int);
68 c734c0e9 2021-08-03 op int findeol(void);
70 0b453b63 2021-08-07 op static int shouldfail;
72 c734c0e9 2021-08-03 op typedef struct {
74 c734c0e9 2021-08-03 op struct op *op;
75 c734c0e9 2021-08-03 op struct proc *proc;
86 c734c0e9 2021-08-03 op * %define parse.error verbose
93 c734c0e9 2021-08-03 op %token INCLUDE
96 0b453b63 2021-08-07 op %token SHOULD_FAIL STR
97 c734c0e9 2021-08-03 op %token TESTING
98 c734c0e9 2021-08-03 op %token U8 U16 U32
101 c734c0e9 2021-08-03 op %token <v.str> STRING SYMBOL
102 c734c0e9 2021-08-03 op %token <v.num> NUMBER
104 acf1403a 2021-08-05 op %type <v.op> cast cexpr check expr faccess funcall
105 340ee97d 2021-08-08 op %type <v.op> literal sfail var varref vargs
107 c734c0e9 2021-08-03 op %type <v.proc> procname
111 c734c0e9 2021-08-03 op program : /* empty */
112 c734c0e9 2021-08-03 op | program '\n'
113 c734c0e9 2021-08-03 op | program include '\n'
114 c734c0e9 2021-08-03 op | program const '\n'
115 c734c0e9 2021-08-03 op | program proc '\n'
116 c734c0e9 2021-08-03 op | program test '\n'
119 c734c0e9 2021-08-03 op optnl : '\n' optnl /* zero or more newlines */
123 c01463a3 2021-08-04 op nl : '\n' optnl ;
125 c734c0e9 2021-08-03 op include : INCLUDE STRING {
126 c734c0e9 2021-08-03 op struct file *nfile;
128 c734c0e9 2021-08-03 op if ((nfile = pushfile($2)) == NULL) {
129 c734c0e9 2021-08-03 op yyerror("failed to include file %s", $2);
135 c734c0e9 2021-08-03 op file = nfile;
136 c734c0e9 2021-08-03 op lungetc('\n');
140 c01463a3 2021-08-04 op const : CONST consti
141 e0b4ca44 2021-08-04 op | CONST '(' optnl mconst ')'
144 c01463a3 2021-08-04 op mconst : consti nl | mconst consti nl ;
146 c01463a3 2021-08-04 op consti : SYMBOL '=' expr {
147 c01463a3 2021-08-04 op if (!global_set($1, $3)) {
148 a3097d05 2021-08-04 op yyerror("can't set %s: illegal expression", $1);
156 c734c0e9 2021-08-03 op var : SYMBOL '=' expr { $$ = op_assign($1, $3); } ;
157 c734c0e9 2021-08-03 op varref : SYMBOL { $$ = op_var($1); } ;
158 c734c0e9 2021-08-03 op literal : STRING { $$ = op_lit_str($1); }
159 c734c0e9 2021-08-03 op | NUMBER { $$ = op_lit_num($1); } ;
162 c734c0e9 2021-08-03 op * `expr '=' '=' expr` is ambiguous. furthermore, we're not
163 c734c0e9 2021-08-03 op * interested in checking all the possibilities here.
165 acf1403a 2021-08-05 op cexpr : literal | varref | funcall | faccess ;
166 c734c0e9 2021-08-03 op check : cexpr '=' '=' cexpr { $$ = op_cmp_eq($1, $4); } ;
168 340ee97d 2021-08-08 op expr : literal | funcall | varref | check | cast | faccess | vargs ;
170 340ee97d 2021-08-08 op vargs : VARGS { $$ = op_vargs(); } ;
172 c734c0e9 2021-08-03 op cast : expr ':' U8 { $$ = op_cast($1, V_U8); }
173 c734c0e9 2021-08-03 op | expr ':' U16 { $$ = op_cast($1, V_U16); }
174 c734c0e9 2021-08-03 op | expr ':' U32 { $$ = op_cast($1, V_U32); }
175 c734c0e9 2021-08-03 op | expr ':' STR { $$ = op_cast($1, V_STR); }
178 acf1403a 2021-08-05 op faccess : varref '.' SYMBOL { $$ = op_faccess($1, $3); }
179 acf1403a 2021-08-05 op | faccess '.' SYMBOL { $$ = op_faccess($1, $3); }
182 c734c0e9 2021-08-03 op procname: SYMBOL {
183 c734c0e9 2021-08-03 op if (($$ = proc_by_name($1)) == NULL) {
184 c734c0e9 2021-08-03 op yyerror("unknown proc %s", $1);
192 c734c0e9 2021-08-03 op funcall : procname {
193 d9d02161 2021-08-04 op prepare_funcall();
194 c734c0e9 2021-08-03 op } '(' args optcomma ')' {
195 7d79f46c 2021-08-04 op struct proc *proc;
198 d9d02161 2021-08-04 op $$ = op_funcall($1);
199 7d79f46c 2021-08-04 op proc = $$->v.funcall.proc;
200 7d79f46c 2021-08-04 op argc = $$->v.funcall.argc;
202 7d79f46c 2021-08-04 op if (argc != proc->minargs && !proc->vararg) {
203 7d79f46c 2021-08-04 op yyerror("invalid arity for `%s': want %d arguments "
204 7d79f46c 2021-08-04 op "but %d given.", $1->name, proc->minargs, argc);
205 c092e54b 2021-08-04 op /* TODO: recursively free $$ */
209 7d79f46c 2021-08-04 op if (argc < proc->minargs && proc->vararg) {
210 7d79f46c 2021-08-04 op yyerror("invalid arity for `%s': want at least %d "
211 7d79f46c 2021-08-04 op "arguments but %d given.", $1->name, proc->minargs,
213 7d79f46c 2021-08-04 op /* TODO: recursively free $$ */
219 c734c0e9 2021-08-03 op optcomma: /* empty */ | ',' ;
221 0e62706a 2021-08-04 op dots : '.' '.' '.' ;
223 c734c0e9 2021-08-03 op args : /* empty */
224 c734c0e9 2021-08-03 op | args ',' expr { push_arg($3); }
225 0e62706a 2021-08-04 op | args ',' dots { push_arg(op_rest()); }
226 c734c0e9 2021-08-03 op | expr { push_arg($1); }
227 0e62706a 2021-08-04 op | dots { push_arg(op_rest()); }
230 c734c0e9 2021-08-03 op proc : PROC SYMBOL {
231 d9d02161 2021-08-04 op prepare_proc();
232 c734c0e9 2021-08-03 op } '(' args ')' {
233 d9d02161 2021-08-04 op if (!proc_setup_body()) {
234 d9d02161 2021-08-04 op yyerror("invalid argument in proc `%s' definition",
239 c734c0e9 2021-08-03 op } '{' optnl block '}' {
240 d9d02161 2021-08-04 op proc_done($2);
244 c734c0e9 2021-08-03 op block : /* empty */
245 d7be3bcc 2021-08-04 op | block var nl { block_push($2); }
246 d7be3bcc 2021-08-04 op | block funcall nl { block_push($2); }
247 d7be3bcc 2021-08-04 op | block assert nl
248 8250ab19 2021-08-07 op | block sfail nl { block_push($2); }
251 8250ab19 2021-08-07 op sfail : SHOULD_FAIL expr { $$ = op_sfail($2, NULL); }
252 8250ab19 2021-08-07 op | SHOULD_FAIL expr ':' STRING { $$ = op_sfail($2, $4); }
255 e0b4ca44 2021-08-04 op assert : ASSERT asserti
256 e0b4ca44 2021-08-04 op | ASSERT '(' optnl massert ')'
259 e0b4ca44 2021-08-04 op massert : asserti nl | massert asserti nl ;
261 e0b4ca44 2021-08-04 op asserti : check { block_push(op_assert($1)); }
264 c734c0e9 2021-08-03 op test : TESTING STRING DIR STRING {
265 d9d02161 2021-08-04 op prepare_test();
266 0b453b63 2021-08-07 op } testopt '{' optnl block '}' {
267 0b453b63 2021-08-07 op test_done(shouldfail, $2, $4);
268 0b453b63 2021-08-07 op shouldfail = 0;
272 0b453b63 2021-08-07 op testopt : /* empty */
273 0b453b63 2021-08-07 op | SHOULD_FAIL { shouldfail = 1; }
278 c734c0e9 2021-08-03 op struct keywords {
279 c734c0e9 2021-08-03 op const char *k_name;
284 c734c0e9 2021-08-03 op yyerror(const char *fmt, ...)
289 c734c0e9 2021-08-03 op file->errors++;
290 c734c0e9 2021-08-03 op va_start(ap, fmt);
291 c734c0e9 2021-08-03 op if (vasprintf(&msg, fmt, ap) == -1)
292 c734c0e9 2021-08-03 op fatalx("yyerror vasprintf");
294 c734c0e9 2021-08-03 op logit(LOG_CRIT, "%s:%d: %s", file->name, yylval.lineno, msg);
300 c734c0e9 2021-08-03 op kw_cmp(const void *k, const void *e)
302 c734c0e9 2021-08-03 op return strcmp(k, ((const struct keywords *)e)->k_name);
306 c734c0e9 2021-08-03 op lookup(char *s)
308 c734c0e9 2021-08-03 op /* This has to be sorted always. */
309 c734c0e9 2021-08-03 op static const struct keywords keywords[] = {
310 c734c0e9 2021-08-03 op {"assert", ASSERT},
311 c734c0e9 2021-08-03 op {"const", CONST},
312 c734c0e9 2021-08-03 op {"dir", DIR},
313 c734c0e9 2021-08-03 op {"include", INCLUDE},
314 c734c0e9 2021-08-03 op {"proc", PROC},
315 c734c0e9 2021-08-03 op {"repeat", REPEAT},
316 0b453b63 2021-08-07 op {"should-fail", SHOULD_FAIL},
317 c734c0e9 2021-08-03 op {"str", STR},
318 c734c0e9 2021-08-03 op {"testing", TESTING},
319 c734c0e9 2021-08-03 op {"u16", U16},
320 c734c0e9 2021-08-03 op {"u32", U32},
322 340ee97d 2021-08-08 op {"vargs", VARGS},
324 c734c0e9 2021-08-03 op const struct keywords *p;
326 c734c0e9 2021-08-03 op p = bsearch(s, keywords, sizeof(keywords)/sizeof(keywords[0]),
327 c734c0e9 2021-08-03 op sizeof(keywords[0]), kw_cmp);
330 c734c0e9 2021-08-03 op return p->k_val;
332 c734c0e9 2021-08-03 op return SYMBOL;
335 c734c0e9 2021-08-03 op #define START_EXPAND 1
336 c734c0e9 2021-08-03 op #define DONE_EXPAND 2
338 c734c0e9 2021-08-03 op static int expanding;
346 c734c0e9 2021-08-03 op if (file->ungetpos > 0)
347 c734c0e9 2021-08-03 op c = file->ungetbuf[--file->ungetpos];
349 c734c0e9 2021-08-03 op c = getc(file->stream);
351 c734c0e9 2021-08-03 op if (c == START_EXPAND)
352 c734c0e9 2021-08-03 op expanding = 1;
353 c734c0e9 2021-08-03 op else if (c == DONE_EXPAND)
354 c734c0e9 2021-08-03 op expanding = 0;
362 c734c0e9 2021-08-03 op lgetc(int quotec)
366 c734c0e9 2021-08-03 op if (quotec) {
367 c734c0e9 2021-08-03 op if ((c = igetc()) == EOF) {
368 c734c0e9 2021-08-03 op yyerror("reached end of file while parsing "
369 c734c0e9 2021-08-03 op "quoted string");
370 c734c0e9 2021-08-03 op if (file == topfile || popfile() == EOF)
372 c734c0e9 2021-08-03 op return quotec;
377 c734c0e9 2021-08-03 op while ((c = igetc()) == '\\') {
378 c734c0e9 2021-08-03 op next = igetc();
379 c734c0e9 2021-08-03 op if (next != '\n') {
383 c734c0e9 2021-08-03 op yylval.lineno = file->lineno;
384 c734c0e9 2021-08-03 op file->lineno++;
387 c734c0e9 2021-08-03 op if (c == EOF) {
389 c734c0e9 2021-08-03 op * Fake EOL when hit EOF for the first time. This gets line
390 c734c0e9 2021-08-03 op * count right if last line in included file is syntactically
391 c734c0e9 2021-08-03 op * invalid and has no newline.
393 c734c0e9 2021-08-03 op if (file->eof_reached == 0) {
394 c734c0e9 2021-08-03 op file->eof_reached = 1;
397 c734c0e9 2021-08-03 op while (c == EOF) {
398 c734c0e9 2021-08-03 op if (file == topfile || popfile() == EOF)
407 c734c0e9 2021-08-03 op lungetc(int c)
409 c734c0e9 2021-08-03 op if (c == EOF)
412 c734c0e9 2021-08-03 op if (file->ungetpos >= file->ungetsize) {
413 c734c0e9 2021-08-03 op void *p = reallocarray(file->ungetbuf, file->ungetsize, 2);
414 c734c0e9 2021-08-03 op if (p == NULL)
415 c734c0e9 2021-08-03 op err(1, "lungetc");
416 c734c0e9 2021-08-03 op file->ungetbuf = p;
417 c734c0e9 2021-08-03 op file->ungetsize *= 2;
419 c734c0e9 2021-08-03 op file->ungetbuf[file->ungetpos++] = c;
423 c734c0e9 2021-08-03 op findeol(void)
427 c734c0e9 2021-08-03 op /* Skip to either EOF or the first real EOL. */
429 c734c0e9 2021-08-03 op c = lgetc(0);
430 c734c0e9 2021-08-03 op if (c == '\n') {
431 c734c0e9 2021-08-03 op file->lineno++;
434 c734c0e9 2021-08-03 op if (c == EOF)
437 c734c0e9 2021-08-03 op return ERROR;
442 c734c0e9 2021-08-03 op int my_yylex(void);
449 c734c0e9 2021-08-03 op switch (x = my_yylex()) {
450 c734c0e9 2021-08-03 op case ASSERT: puts("assert"); break;
451 c734c0e9 2021-08-03 op case CONST: puts("const"); break;
452 c734c0e9 2021-08-03 op case DIR: puts("dir"); break;
453 c734c0e9 2021-08-03 op case ERROR: puts("error"); break;
454 c734c0e9 2021-08-03 op case INCLUDE: puts("include"); break;
455 c734c0e9 2021-08-03 op case PROC: puts("proc"); break;
456 c734c0e9 2021-08-03 op case REPEAT: puts("repeat"); break;
457 c734c0e9 2021-08-03 op case STR: puts(":str"); break;
458 c734c0e9 2021-08-03 op case TESTING: puts("testing"); break;
459 c734c0e9 2021-08-03 op case U8: puts(":u8"); break;
460 c734c0e9 2021-08-03 op case U16: puts(":u16"); break;
461 c734c0e9 2021-08-03 op case U32: puts(":u32"); break;
463 c734c0e9 2021-08-03 op case STRING: printf("string \"%s\"\n", yylval.v.str); break;
464 c734c0e9 2021-08-03 op case SYMBOL: printf("symbol %s\n", yylval.v.str); break;
465 c734c0e9 2021-08-03 op case NUMBER: printf("number %"PRIu64"\n", yylval.v.num); break;
468 c734c0e9 2021-08-03 op printf("character ");
469 c734c0e9 2021-08-03 op if (x == '\n')
470 c734c0e9 2021-08-03 op printf("\\n");
472 c734c0e9 2021-08-03 op printf("%c", x);
473 c734c0e9 2021-08-03 op printf(" [0x%x]", x);
474 c734c0e9 2021-08-03 op printf("\n");
482 c734c0e9 2021-08-03 op my_yylex(void)
488 c734c0e9 2021-08-03 op unsigned char buf[8096];
489 c734c0e9 2021-08-03 op unsigned char *p;
490 c734c0e9 2021-08-03 op int quotec, next, c;
494 c734c0e9 2021-08-03 op while ((c = lgetc(0)) == ' ' || c == '\t' || c == '\f')
497 c734c0e9 2021-08-03 op yylval.lineno = file->lineno;
498 c734c0e9 2021-08-03 op if (c == '#')
499 c734c0e9 2021-08-03 op while ((c = lgetc(0)) != '\n' && c != EOF)
510 c734c0e9 2021-08-03 op if ((c = lgetc(quotec)) == EOF)
512 c734c0e9 2021-08-03 op if (c == '\n') {
513 c734c0e9 2021-08-03 op file->lineno++;
515 c734c0e9 2021-08-03 op } else if (c == '\\') {
516 c734c0e9 2021-08-03 op if ((next = lgetc(quotec)) == EOF)
518 c734c0e9 2021-08-03 op if (next == quotec || next == ' ' ||
519 c734c0e9 2021-08-03 op next == '\t')
521 c734c0e9 2021-08-03 op else if (next == '\n') {
522 c734c0e9 2021-08-03 op file->lineno++;
525 c734c0e9 2021-08-03 op lungetc(next);
526 c734c0e9 2021-08-03 op } else if (c == quotec) {
529 c734c0e9 2021-08-03 op } else if (c == '\0') {
530 c734c0e9 2021-08-03 op yyerror("syntax error");
531 c734c0e9 2021-08-03 op return findeol();
534 c734c0e9 2021-08-03 op if (p + 1 >= buf + sizeof(buf) - 1) {
535 c734c0e9 2021-08-03 op yyerror("string too long");
536 c734c0e9 2021-08-03 op return findeol();
542 c734c0e9 2021-08-03 op yylval.v.str = xstrdup(buf);
543 c734c0e9 2021-08-03 op return STRING;
546 c734c0e9 2021-08-03 op #define allowed_to_end_number(x) \
547 c734c0e9 2021-08-03 op (isspace(x) || x == ')' || x == ',' || x == '/' || x == '}' \
548 1433867a 2021-08-04 op || x == '=' || x == ':')
550 c734c0e9 2021-08-03 op if (c == '-' || isdigit(c)) {
553 c734c0e9 2021-08-03 op if ((size_t)(p-buf) >= sizeof(buf)) {
554 c734c0e9 2021-08-03 op yyerror("string too long");
555 c734c0e9 2021-08-03 op return findeol();
557 ef33a69e 2021-08-07 op } while ((c = lgetc(0)) != EOF && (isdigit(c) || c == 'x'));
559 c734c0e9 2021-08-03 op if (p == buf + 1 && buf[0] == '-')
560 c734c0e9 2021-08-03 op goto nodigits;
561 c734c0e9 2021-08-03 op if (c == EOF || allowed_to_end_number(c)) {
566 ef33a69e 2021-08-07 op yylval.v.num = strtoll(buf, &ep, 0);
567 7abcb0db 2021-12-15 op if (*ep != '\0' || (errno == ERANGE &&
568 ef33a69e 2021-08-07 op (yylval.v.num == LONG_MAX ||
569 7abcb0db 2021-12-15 op yylval.v.num == LONG_MIN))) {
570 ef33a69e 2021-08-07 op yyerror("\"%s\" invalid number or out of range",
572 c734c0e9 2021-08-03 op return findeol();
575 c734c0e9 2021-08-03 op return NUMBER;
578 c734c0e9 2021-08-03 op while (p > buf + 1)
579 c734c0e9 2021-08-03 op lungetc(*--p);
581 c734c0e9 2021-08-03 op if (c == '-')
586 c734c0e9 2021-08-03 op #define allowed_in_symbol(x) \
587 c734c0e9 2021-08-03 op (isalnum(x) || (ispunct(x) && x != '(' && x != ')' && \
588 c734c0e9 2021-08-03 op x != '{' && x != '}' && \
589 c734c0e9 2021-08-03 op x != '!' && x != '=' && \
590 e48a50b7 2021-08-07 op x != '#' && x != ',' && \
591 02fc4f51 2021-08-07 op x != '.' && x != ':'))
593 c734c0e9 2021-08-03 op if (isalnum(c) || c == ':' || c == '_') {
596 c734c0e9 2021-08-03 op if ((size_t)(p-buf) >= sizeof(buf)) {
597 c734c0e9 2021-08-03 op yyerror("string too long");
598 c734c0e9 2021-08-03 op return findeol();
600 c734c0e9 2021-08-03 op } while ((c = lgetc(0)) != EOF && (allowed_in_symbol(c)));
603 c734c0e9 2021-08-03 op if ((token = lookup(buf)) == SYMBOL)
604 c734c0e9 2021-08-03 op yylval.v.str = xstrdup(buf);
605 c734c0e9 2021-08-03 op return token;
608 c734c0e9 2021-08-03 op if (c == '\n') {
609 c734c0e9 2021-08-03 op yylval.lineno = file->lineno;
610 c734c0e9 2021-08-03 op file->lineno++;
612 c734c0e9 2021-08-03 op if (c == EOF)
617 c734c0e9 2021-08-03 op struct file *
618 c734c0e9 2021-08-03 op pushfile(const char *name)
620 c734c0e9 2021-08-03 op struct file *nfile;
622 c734c0e9 2021-08-03 op if ((nfile = calloc(1, sizeof(struct file))) == NULL) {
623 c734c0e9 2021-08-03 op log_warn("calloc");
626 c734c0e9 2021-08-03 op if ((nfile->name = strdup(name)) == NULL) {
627 c734c0e9 2021-08-03 op log_warn("strdup");
631 c734c0e9 2021-08-03 op if ((nfile->stream = fopen(nfile->name, "r")) == NULL) {
632 c734c0e9 2021-08-03 op log_warn("%s", nfile->name);
633 c734c0e9 2021-08-03 op free(nfile->name);
637 c734c0e9 2021-08-03 op nfile->lineno = TAILQ_EMPTY(&files) ? 1 : 0;
638 c734c0e9 2021-08-03 op nfile->ungetsize = 16;
639 c734c0e9 2021-08-03 op nfile->ungetbuf = malloc(nfile->ungetsize);
640 c734c0e9 2021-08-03 op if (nfile->ungetbuf == NULL) {
641 c734c0e9 2021-08-03 op log_warn("malloc");
642 c734c0e9 2021-08-03 op fclose(nfile->stream);
643 c734c0e9 2021-08-03 op free(nfile->name);
647 c734c0e9 2021-08-03 op TAILQ_INSERT_TAIL(&files, nfile, entry);
648 c734c0e9 2021-08-03 op return nfile;
652 c734c0e9 2021-08-03 op popfile(void)
654 c734c0e9 2021-08-03 op struct file *prev;
656 c734c0e9 2021-08-03 op if ((prev = TAILQ_PREV(file, files, entry)) != NULL)
657 c734c0e9 2021-08-03 op prev->errors += file->errors;
659 c734c0e9 2021-08-03 op TAILQ_REMOVE(&files, file, entry);
660 c734c0e9 2021-08-03 op fclose(file->stream);
661 c734c0e9 2021-08-03 op free(file->name);
662 c734c0e9 2021-08-03 op free(file->ungetbuf);
665 c734c0e9 2021-08-03 op return file ? 0 : EOF;
669 c734c0e9 2021-08-03 op loadfile(const char *path)
673 c734c0e9 2021-08-03 op file = pushfile(path);
674 c734c0e9 2021-08-03 op if (file == NULL)
675 c734c0e9 2021-08-03 op err(1, "pushfile");
676 c734c0e9 2021-08-03 op topfile = file;
679 c734c0e9 2021-08-03 op errors = file->errors;
683 c734c0e9 2021-08-03 op errx(1, "can't load %s because of errors", path);