2 a0ec53a2 2022-01-09 op * Copyright (c) 2021, 2022 Omar Polo <op@omarpolo.com>
3 fb1a36c0 2022-01-09 op * Copyright (c) 2018 Florian Obser <florian@openbsd.org>
4 fb1a36c0 2022-01-09 op * Copyright (c) 2004, 2005 Esben Norby <norby@openbsd.org>
5 fb1a36c0 2022-01-09 op * Copyright (c) 2004 Ryan McBride <mcbride@openbsd.org>
6 fb1a36c0 2022-01-09 op * Copyright (c) 2002, 2003, 2004 Henning Brauer <henning@openbsd.org>
7 fb1a36c0 2022-01-09 op * Copyright (c) 2001 Markus Friedl. All rights reserved.
8 fb1a36c0 2022-01-09 op * Copyright (c) 2001 Daniel Hartmeier. All rights reserved.
9 fb1a36c0 2022-01-09 op * Copyright (c) 2001 Theo de Raadt. All rights reserved.
11 fb1a36c0 2022-01-09 op * Permission to use, copy, modify, and distribute this software for any
12 fb1a36c0 2022-01-09 op * purpose with or without fee is hereby granted, provided that the above
13 fb1a36c0 2022-01-09 op * copyright notice and this permission notice appear in all copies.
15 fb1a36c0 2022-01-09 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
16 fb1a36c0 2022-01-09 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
17 fb1a36c0 2022-01-09 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
18 fb1a36c0 2022-01-09 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
19 fb1a36c0 2022-01-09 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
20 fb1a36c0 2022-01-09 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
21 fb1a36c0 2022-01-09 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
26 bbcba3ed 2022-01-10 op #include "compat.h"
28 fb1a36c0 2022-01-09 op #include <ctype.h>
29 fb1a36c0 2022-01-09 op #include <errno.h>
30 aead84be 2022-01-10 op #include <fcntl.h>
31 fb1a36c0 2022-01-09 op #include <inttypes.h>
32 aead84be 2022-01-10 op #include <libgen.h>
33 fb1a36c0 2022-01-09 op #include <limits.h>
34 fb1a36c0 2022-01-09 op #include <limits.h>
35 fb1a36c0 2022-01-09 op #include <stdio.h>
36 fb1a36c0 2022-01-09 op #include <stdlib.h>
37 fb1a36c0 2022-01-09 op #include <string.h>
38 fb1a36c0 2022-01-09 op #include <syslog.h>
39 aead84be 2022-01-10 op #include <unistd.h>
41 fb1a36c0 2022-01-09 op #include "log.h"
42 fb1a36c0 2022-01-09 op #include "kami.h"
43 fb1a36c0 2022-01-09 op #include "utils.h"
45 fb1a36c0 2022-01-09 op #include "script.h"
47 fb1a36c0 2022-01-09 op TAILQ_HEAD(files, file) files = TAILQ_HEAD_INITIALIZER(files);
48 fb1a36c0 2022-01-09 op static struct file {
49 fb1a36c0 2022-01-09 op TAILQ_ENTRY(file) entry;
52 fb1a36c0 2022-01-09 op size_t ungetpos;
53 fb1a36c0 2022-01-09 op size_t ungetsize;
54 fb1a36c0 2022-01-09 op u_char *ungetbuf;
55 fb1a36c0 2022-01-09 op int eof_reached;
58 fb1a36c0 2022-01-09 op } *file, *topfile;
59 fb1a36c0 2022-01-09 op struct file *pushfile(const char *);
60 fb1a36c0 2022-01-09 op int popfile(void);
61 fb1a36c0 2022-01-09 op int yyparse(void);
62 fb1a36c0 2022-01-09 op int yylex(void);
63 fb1a36c0 2022-01-09 op int yyerror(const char *, ...)
64 fb1a36c0 2022-01-09 op __attribute__((__format__ (printf, 1, 2)))
65 fb1a36c0 2022-01-09 op __attribute__((__nonnull__ (1)));
66 fb1a36c0 2022-01-09 op int kw_cmp(const void *, const void *);
67 fb1a36c0 2022-01-09 op int lookup(char *);
68 fb1a36c0 2022-01-09 op int igetc(void);
69 fb1a36c0 2022-01-09 op int lgetc(int);
70 fb1a36c0 2022-01-09 op void lungetc(int);
71 fb1a36c0 2022-01-09 op int findeol(void);
73 fb1a36c0 2022-01-09 op static int shouldfail;
75 fb1a36c0 2022-01-09 op typedef struct {
77 fb1a36c0 2022-01-09 op struct op *op;
78 fb1a36c0 2022-01-09 op struct proc *proc;
89 fb1a36c0 2022-01-09 op * %define parse.error verbose
95 fb1a36c0 2022-01-09 op %token INCLUDE
98 fb1a36c0 2022-01-09 op %token SHOULD_FAIL STR
99 fb1a36c0 2022-01-09 op %token TESTING
100 fb1a36c0 2022-01-09 op %token U8 U16 U32
103 fb1a36c0 2022-01-09 op %token <v.str> STRING SYMBOL
104 fb1a36c0 2022-01-09 op %token <v.num> NUMBER
106 fb1a36c0 2022-01-09 op %type <v.op> cast cexpr check expr faccess funcall
107 fb1a36c0 2022-01-09 op %type <v.op> literal sfail var varref vargs
109 fb1a36c0 2022-01-09 op %type <v.proc> procname
113 fb1a36c0 2022-01-09 op program : /* empty */
114 fb1a36c0 2022-01-09 op | program '\n'
115 fb1a36c0 2022-01-09 op | program include '\n'
116 fb1a36c0 2022-01-09 op | program const '\n'
117 fb1a36c0 2022-01-09 op | program proc '\n'
118 fb1a36c0 2022-01-09 op | program test '\n'
121 fb1a36c0 2022-01-09 op optnl : '\n' optnl /* zero or more newlines */
125 fb1a36c0 2022-01-09 op nl : '\n' optnl ;
127 fb1a36c0 2022-01-09 op include : INCLUDE STRING {
128 fb1a36c0 2022-01-09 op struct file *nfile;
130 fb1a36c0 2022-01-09 op if ((nfile = pushfile($2)) == NULL) {
131 fb1a36c0 2022-01-09 op yyerror("failed to include file %s", $2);
137 fb1a36c0 2022-01-09 op file = nfile;
138 fb1a36c0 2022-01-09 op lungetc('\n');
142 fb1a36c0 2022-01-09 op const : CONST consti
143 fb1a36c0 2022-01-09 op | CONST '(' optnl mconst ')'
146 fb1a36c0 2022-01-09 op mconst : consti nl | mconst consti nl ;
148 fb1a36c0 2022-01-09 op consti : SYMBOL '=' expr {
149 fb1a36c0 2022-01-09 op if (!global_set($1, $3)) {
150 fb1a36c0 2022-01-09 op yyerror("can't set %s: illegal expression", $1);
158 fb1a36c0 2022-01-09 op var : SYMBOL '=' expr { $$ = op_assign($1, $3); } ;
159 fb1a36c0 2022-01-09 op varref : SYMBOL { $$ = op_var($1); } ;
160 fb1a36c0 2022-01-09 op literal : STRING { $$ = op_lit_str($1); }
161 fb1a36c0 2022-01-09 op | NUMBER { $$ = op_lit_num($1); } ;
164 fb1a36c0 2022-01-09 op * `expr '=' '=' expr` is ambiguous. furthermore, we're not
165 fb1a36c0 2022-01-09 op * interested in checking all the possibilities here.
167 fb1a36c0 2022-01-09 op cexpr : literal | varref | funcall | faccess ;
168 fb1a36c0 2022-01-09 op check : cexpr '=' '=' cexpr { $$ = op_cmp_eq($1, $4); }
169 fb1a36c0 2022-01-09 op | cexpr '<' '=' cexpr { $$ = op_cmp_leq($1, $4); }
172 fb1a36c0 2022-01-09 op expr : literal | funcall | varref | check | cast | faccess | vargs ;
174 fb1a36c0 2022-01-09 op vargs : VARGS { $$ = op_vargs(); } ;
176 fb1a36c0 2022-01-09 op cast : expr ':' U8 { $$ = op_cast($1, V_U8); }
177 fb1a36c0 2022-01-09 op | expr ':' U16 { $$ = op_cast($1, V_U16); }
178 fb1a36c0 2022-01-09 op | expr ':' U32 { $$ = op_cast($1, V_U32); }
179 fb1a36c0 2022-01-09 op | expr ':' STR { $$ = op_cast($1, V_STR); }
182 fb1a36c0 2022-01-09 op faccess : varref '.' SYMBOL { $$ = op_faccess($1, $3); }
183 fb1a36c0 2022-01-09 op | faccess '.' SYMBOL { $$ = op_faccess($1, $3); }
186 fb1a36c0 2022-01-09 op procname: SYMBOL {
187 fb1a36c0 2022-01-09 op if (($$ = proc_by_name($1)) == NULL) {
188 fb1a36c0 2022-01-09 op yyerror("unknown proc %s", $1);
196 fb1a36c0 2022-01-09 op funcall : procname {
197 fb1a36c0 2022-01-09 op prepare_funcall();
198 fb1a36c0 2022-01-09 op } '(' args optcomma ')' {
199 fb1a36c0 2022-01-09 op struct proc *proc;
202 fb1a36c0 2022-01-09 op $$ = op_funcall($1);
203 fb1a36c0 2022-01-09 op proc = $$->v.funcall.proc;
204 fb1a36c0 2022-01-09 op argc = $$->v.funcall.argc;
206 fb1a36c0 2022-01-09 op if (argc != proc->minargs && !proc->vararg) {
207 fb1a36c0 2022-01-09 op yyerror("invalid arity for `%s': want %d arguments "
208 fb1a36c0 2022-01-09 op "but %d given.", $1->name, proc->minargs, argc);
209 fb1a36c0 2022-01-09 op /* TODO: recursively free $$ */
213 fb1a36c0 2022-01-09 op if (argc < proc->minargs && proc->vararg) {
214 fb1a36c0 2022-01-09 op yyerror("invalid arity for `%s': want at least %d "
215 fb1a36c0 2022-01-09 op "arguments but %d given.", $1->name, proc->minargs,
217 fb1a36c0 2022-01-09 op /* TODO: recursively free $$ */
223 fb1a36c0 2022-01-09 op optcomma: /* empty */ | ',' ;
225 fb1a36c0 2022-01-09 op dots : '.' '.' '.' ;
227 fb1a36c0 2022-01-09 op args : /* empty */
228 fb1a36c0 2022-01-09 op | args ',' expr { push_arg($3); }
229 fb1a36c0 2022-01-09 op | args ',' dots { push_arg(op_rest()); }
230 fb1a36c0 2022-01-09 op | expr { push_arg($1); }
231 fb1a36c0 2022-01-09 op | dots { push_arg(op_rest()); }
234 fb1a36c0 2022-01-09 op proc : PROC SYMBOL {
235 fb1a36c0 2022-01-09 op prepare_proc();
236 fb1a36c0 2022-01-09 op } '(' args ')' {
237 fb1a36c0 2022-01-09 op if (!proc_setup_body()) {
238 fb1a36c0 2022-01-09 op yyerror("invalid argument in proc `%s' definition",
243 fb1a36c0 2022-01-09 op } '{' optnl block '}' {
244 fb1a36c0 2022-01-09 op proc_done($2);
248 fb1a36c0 2022-01-09 op block : /* empty */
249 fb1a36c0 2022-01-09 op | block var nl { block_push($2); }
250 fb1a36c0 2022-01-09 op | block funcall nl { block_push($2); }
251 fb1a36c0 2022-01-09 op | block assert nl
252 fb1a36c0 2022-01-09 op | block sfail nl { block_push($2); }
255 fb1a36c0 2022-01-09 op sfail : SHOULD_FAIL expr { $$ = op_sfail($2, NULL); }
256 fb1a36c0 2022-01-09 op | SHOULD_FAIL expr ':' STRING { $$ = op_sfail($2, $4); }
259 fb1a36c0 2022-01-09 op assert : ASSERT asserti
260 fb1a36c0 2022-01-09 op | ASSERT '(' optnl massert ')'
263 fb1a36c0 2022-01-09 op massert : asserti nl | massert asserti nl ;
265 fb1a36c0 2022-01-09 op asserti : check { block_push(op_assert($1)); }
268 a0ec53a2 2022-01-09 op test : TESTING STRING {
269 fb1a36c0 2022-01-09 op prepare_test();
270 fb1a36c0 2022-01-09 op } testopt '{' optnl block '}' {
271 a0ec53a2 2022-01-09 op test_done(shouldfail, $2);
272 fb1a36c0 2022-01-09 op shouldfail = 0;
276 fb1a36c0 2022-01-09 op testopt : /* empty */
277 fb1a36c0 2022-01-09 op | SHOULD_FAIL { shouldfail = 1; }
282 fb1a36c0 2022-01-09 op struct keywords {
283 fb1a36c0 2022-01-09 op const char *k_name;
288 fb1a36c0 2022-01-09 op yyerror(const char *fmt, ...)
293 fb1a36c0 2022-01-09 op file->errors++;
294 fb1a36c0 2022-01-09 op va_start(ap, fmt);
295 fb1a36c0 2022-01-09 op if (vasprintf(&msg, fmt, ap) == -1)
296 fb1a36c0 2022-01-09 op fatalx("yyerror vasprintf");
298 fb1a36c0 2022-01-09 op logit(LOG_CRIT, "%s:%d: %s", file->name, yylval.lineno, msg);
304 fb1a36c0 2022-01-09 op kw_cmp(const void *k, const void *e)
306 fb1a36c0 2022-01-09 op return strcmp(k, ((const struct keywords *)e)->k_name);
310 fb1a36c0 2022-01-09 op lookup(char *s)
312 fb1a36c0 2022-01-09 op /* This has to be sorted always. */
313 fb1a36c0 2022-01-09 op static const struct keywords keywords[] = {
314 fb1a36c0 2022-01-09 op {"assert", ASSERT},
315 fb1a36c0 2022-01-09 op {"const", CONST},
316 fb1a36c0 2022-01-09 op {"include", INCLUDE},
317 fb1a36c0 2022-01-09 op {"proc", PROC},
318 fb1a36c0 2022-01-09 op {"repeat", REPEAT},
319 fb1a36c0 2022-01-09 op {"should-fail", SHOULD_FAIL},
320 fb1a36c0 2022-01-09 op {"str", STR},
321 fb1a36c0 2022-01-09 op {"testing", TESTING},
322 fb1a36c0 2022-01-09 op {"u16", U16},
323 fb1a36c0 2022-01-09 op {"u32", U32},
325 fb1a36c0 2022-01-09 op {"vargs", VARGS},
327 fb1a36c0 2022-01-09 op const struct keywords *p;
329 fb1a36c0 2022-01-09 op p = bsearch(s, keywords, sizeof(keywords)/sizeof(keywords[0]),
330 fb1a36c0 2022-01-09 op sizeof(keywords[0]), kw_cmp);
333 fb1a36c0 2022-01-09 op return p->k_val;
335 fb1a36c0 2022-01-09 op return SYMBOL;
338 fb1a36c0 2022-01-09 op #define START_EXPAND 1
339 fb1a36c0 2022-01-09 op #define DONE_EXPAND 2
341 fb1a36c0 2022-01-09 op static int expanding;
349 fb1a36c0 2022-01-09 op if (file->ungetpos > 0)
350 fb1a36c0 2022-01-09 op c = file->ungetbuf[--file->ungetpos];
352 fb1a36c0 2022-01-09 op c = getc(file->stream);
354 fb1a36c0 2022-01-09 op if (c == START_EXPAND)
355 fb1a36c0 2022-01-09 op expanding = 1;
356 fb1a36c0 2022-01-09 op else if (c == DONE_EXPAND)
357 fb1a36c0 2022-01-09 op expanding = 0;
365 fb1a36c0 2022-01-09 op lgetc(int quotec)
369 fb1a36c0 2022-01-09 op if (quotec) {
370 fb1a36c0 2022-01-09 op if ((c = igetc()) == EOF) {
371 fb1a36c0 2022-01-09 op yyerror("reached end of file while parsing "
372 fb1a36c0 2022-01-09 op "quoted string");
373 fb1a36c0 2022-01-09 op if (file == topfile || popfile() == EOF)
375 fb1a36c0 2022-01-09 op return quotec;
380 fb1a36c0 2022-01-09 op while ((c = igetc()) == '\\') {
381 fb1a36c0 2022-01-09 op next = igetc();
382 fb1a36c0 2022-01-09 op if (next != '\n') {
386 fb1a36c0 2022-01-09 op yylval.lineno = file->lineno;
387 fb1a36c0 2022-01-09 op file->lineno++;
390 fb1a36c0 2022-01-09 op if (c == EOF) {
392 fb1a36c0 2022-01-09 op * Fake EOL when hit EOF for the first time. This gets line
393 fb1a36c0 2022-01-09 op * count right if last line in included file is syntactically
394 fb1a36c0 2022-01-09 op * invalid and has no newline.
396 fb1a36c0 2022-01-09 op if (file->eof_reached == 0) {
397 fb1a36c0 2022-01-09 op file->eof_reached = 1;
400 fb1a36c0 2022-01-09 op while (c == EOF) {
401 fb1a36c0 2022-01-09 op if (file == topfile || popfile() == EOF)
410 fb1a36c0 2022-01-09 op lungetc(int c)
412 fb1a36c0 2022-01-09 op if (c == EOF)
415 fb1a36c0 2022-01-09 op if (file->ungetpos >= file->ungetsize) {
416 fb1a36c0 2022-01-09 op void *p = reallocarray(file->ungetbuf, file->ungetsize, 2);
417 fb1a36c0 2022-01-09 op if (p == NULL)
418 fb1a36c0 2022-01-09 op err(1, "lungetc");
419 fb1a36c0 2022-01-09 op file->ungetbuf = p;
420 fb1a36c0 2022-01-09 op file->ungetsize *= 2;
422 fb1a36c0 2022-01-09 op file->ungetbuf[file->ungetpos++] = c;
426 fb1a36c0 2022-01-09 op findeol(void)
430 fb1a36c0 2022-01-09 op /* Skip to either EOF or the first real EOL. */
432 fb1a36c0 2022-01-09 op c = lgetc(0);
433 fb1a36c0 2022-01-09 op if (c == '\n') {
434 fb1a36c0 2022-01-09 op file->lineno++;
437 fb1a36c0 2022-01-09 op if (c == EOF)
440 fb1a36c0 2022-01-09 op return ERROR;
445 fb1a36c0 2022-01-09 op int my_yylex(void);
452 fb1a36c0 2022-01-09 op switch (x = my_yylex()) {
453 fb1a36c0 2022-01-09 op case ASSERT: puts("assert"); break;
454 fb1a36c0 2022-01-09 op case CONST: puts("const"); break;
455 fb1a36c0 2022-01-09 op case ERROR: puts("error"); break;
456 fb1a36c0 2022-01-09 op case INCLUDE: puts("include"); break;
457 fb1a36c0 2022-01-09 op case PROC: puts("proc"); break;
458 fb1a36c0 2022-01-09 op case REPEAT: puts("repeat"); break;
459 fb1a36c0 2022-01-09 op case STR: puts(":str"); break;
460 fb1a36c0 2022-01-09 op case TESTING: puts("testing"); break;
461 fb1a36c0 2022-01-09 op case U8: puts(":u8"); break;
462 fb1a36c0 2022-01-09 op case U16: puts(":u16"); break;
463 fb1a36c0 2022-01-09 op case U32: puts(":u32"); break;
465 fb1a36c0 2022-01-09 op case STRING: printf("string \"%s\"\n", yylval.v.str); break;
466 fb1a36c0 2022-01-09 op case SYMBOL: printf("symbol %s\n", yylval.v.str); break;
467 fb1a36c0 2022-01-09 op case NUMBER: printf("number %"PRIu64"\n", yylval.v.num); break;
470 fb1a36c0 2022-01-09 op printf("character ");
471 fb1a36c0 2022-01-09 op if (x == '\n')
472 fb1a36c0 2022-01-09 op printf("\\n");
474 fb1a36c0 2022-01-09 op printf("%c", x);
475 fb1a36c0 2022-01-09 op printf(" [0x%x]", x);
476 fb1a36c0 2022-01-09 op printf("\n");
484 fb1a36c0 2022-01-09 op my_yylex(void)
490 fb1a36c0 2022-01-09 op unsigned char buf[8096];
491 fb1a36c0 2022-01-09 op unsigned char *p;
492 fb1a36c0 2022-01-09 op int quotec, next, c;
496 fb1a36c0 2022-01-09 op while ((c = lgetc(0)) == ' ' || c == '\t' || c == '\f')
499 fb1a36c0 2022-01-09 op yylval.lineno = file->lineno;
500 fb1a36c0 2022-01-09 op if (c == '#')
501 fb1a36c0 2022-01-09 op while ((c = lgetc(0)) != '\n' && c != EOF)
512 fb1a36c0 2022-01-09 op if ((c = lgetc(quotec)) == EOF)
514 fb1a36c0 2022-01-09 op if (c == '\n') {
515 fb1a36c0 2022-01-09 op file->lineno++;
517 fb1a36c0 2022-01-09 op } else if (c == '\\') {
518 fb1a36c0 2022-01-09 op if ((next = lgetc(quotec)) == EOF)
520 fb1a36c0 2022-01-09 op if (next == quotec || next == ' ' ||
521 fb1a36c0 2022-01-09 op next == '\t')
523 fb1a36c0 2022-01-09 op else if (next == '\n') {
524 fb1a36c0 2022-01-09 op file->lineno++;
527 fb1a36c0 2022-01-09 op lungetc(next);
528 fb1a36c0 2022-01-09 op } else if (c == quotec) {
531 fb1a36c0 2022-01-09 op } else if (c == '\0') {
532 fb1a36c0 2022-01-09 op yyerror("syntax error");
533 fb1a36c0 2022-01-09 op return findeol();
536 fb1a36c0 2022-01-09 op if (p + 1 >= buf + sizeof(buf) - 1) {
537 fb1a36c0 2022-01-09 op yyerror("string too long");
538 fb1a36c0 2022-01-09 op return findeol();
544 fb1a36c0 2022-01-09 op yylval.v.str = xstrdup(buf);
545 fb1a36c0 2022-01-09 op return STRING;
548 fb1a36c0 2022-01-09 op #define allowed_to_end_number(x) \
549 fb1a36c0 2022-01-09 op (isspace(x) || x == ')' || x == ',' || x == '/' || x == '}' \
550 fb1a36c0 2022-01-09 op || x == '=' || x == ':')
552 fb1a36c0 2022-01-09 op if (c == '-' || isdigit(c)) {
555 fb1a36c0 2022-01-09 op if ((size_t)(p-buf) >= sizeof(buf)) {
556 fb1a36c0 2022-01-09 op yyerror("string too long");
557 fb1a36c0 2022-01-09 op return findeol();
559 fb1a36c0 2022-01-09 op } while ((c = lgetc(0)) != EOF && (isdigit(c) || c == 'x'));
561 fb1a36c0 2022-01-09 op if (p == buf + 1 && buf[0] == '-')
562 fb1a36c0 2022-01-09 op goto nodigits;
563 fb1a36c0 2022-01-09 op if (c == EOF || allowed_to_end_number(c)) {
568 fb1a36c0 2022-01-09 op yylval.v.num = strtoll(buf, &ep, 0);
569 fb1a36c0 2022-01-09 op if (*ep != '\0' || (errno == ERANGE &&
570 fb1a36c0 2022-01-09 op (yylval.v.num == LONG_MAX ||
571 fb1a36c0 2022-01-09 op yylval.v.num == LONG_MIN))) {
572 fb1a36c0 2022-01-09 op yyerror("\"%s\" invalid number or out of range",
574 fb1a36c0 2022-01-09 op return findeol();
577 fb1a36c0 2022-01-09 op return NUMBER;
580 fb1a36c0 2022-01-09 op while (p > buf + 1)
581 fb1a36c0 2022-01-09 op lungetc(*--p);
583 fb1a36c0 2022-01-09 op if (c == '-')
588 fb1a36c0 2022-01-09 op #define allowed_in_symbol(x) \
589 fb1a36c0 2022-01-09 op (isalnum(x) || (ispunct(x) && x != '(' && x != ')' && \
590 fb1a36c0 2022-01-09 op x != '{' && x != '}' && \
591 fb1a36c0 2022-01-09 op x != '!' && x != '=' && \
592 fb1a36c0 2022-01-09 op x != '#' && x != ',' && \
593 fb1a36c0 2022-01-09 op x != '.' && x != ':'))
595 fb1a36c0 2022-01-09 op if (isalnum(c) || c == ':' || c == '_') {
598 fb1a36c0 2022-01-09 op if ((size_t)(p-buf) >= sizeof(buf)) {
599 fb1a36c0 2022-01-09 op yyerror("string too long");
600 fb1a36c0 2022-01-09 op return findeol();
602 fb1a36c0 2022-01-09 op } while ((c = lgetc(0)) != EOF && (allowed_in_symbol(c)));
605 fb1a36c0 2022-01-09 op if ((token = lookup(buf)) == SYMBOL)
606 fb1a36c0 2022-01-09 op yylval.v.str = xstrdup(buf);
607 fb1a36c0 2022-01-09 op return token;
610 fb1a36c0 2022-01-09 op if (c == '\n') {
611 fb1a36c0 2022-01-09 op yylval.lineno = file->lineno;
612 fb1a36c0 2022-01-09 op file->lineno++;
614 fb1a36c0 2022-01-09 op if (c == EOF)
619 fb1a36c0 2022-01-09 op struct file *
620 fb1a36c0 2022-01-09 op pushfile(const char *name)
622 fb1a36c0 2022-01-09 op struct file *nfile;
624 fb1a36c0 2022-01-09 op if ((nfile = calloc(1, sizeof(struct file))) == NULL) {
625 fb1a36c0 2022-01-09 op log_warn("calloc");
628 fb1a36c0 2022-01-09 op if ((nfile->name = strdup(name)) == NULL) {
629 fb1a36c0 2022-01-09 op log_warn("strdup");
633 fb1a36c0 2022-01-09 op if ((nfile->stream = fopen(nfile->name, "r")) == NULL) {
634 fb1a36c0 2022-01-09 op log_warn("%s", nfile->name);
635 fb1a36c0 2022-01-09 op free(nfile->name);
639 fb1a36c0 2022-01-09 op nfile->lineno = TAILQ_EMPTY(&files) ? 1 : 0;
640 fb1a36c0 2022-01-09 op nfile->ungetsize = 16;
641 fb1a36c0 2022-01-09 op nfile->ungetbuf = malloc(nfile->ungetsize);
642 fb1a36c0 2022-01-09 op if (nfile->ungetbuf == NULL) {
643 fb1a36c0 2022-01-09 op log_warn("malloc");
644 fb1a36c0 2022-01-09 op fclose(nfile->stream);
645 fb1a36c0 2022-01-09 op free(nfile->name);
649 fb1a36c0 2022-01-09 op TAILQ_INSERT_TAIL(&files, nfile, entry);
650 fb1a36c0 2022-01-09 op return nfile;
654 fb1a36c0 2022-01-09 op popfile(void)
656 fb1a36c0 2022-01-09 op struct file *prev;
658 fb1a36c0 2022-01-09 op if ((prev = TAILQ_PREV(file, files, entry)) != NULL)
659 fb1a36c0 2022-01-09 op prev->errors += file->errors;
661 fb1a36c0 2022-01-09 op TAILQ_REMOVE(&files, file, entry);
662 fb1a36c0 2022-01-09 op fclose(file->stream);
663 fb1a36c0 2022-01-09 op free(file->name);
664 fb1a36c0 2022-01-09 op free(file->ungetbuf);
667 fb1a36c0 2022-01-09 op return file ? 0 : EOF;
671 fb1a36c0 2022-01-09 op loadfile(const char *path)
673 aead84be 2022-01-10 op int pwdfd, errors;
674 aead84be 2022-01-10 op char p[PATH_MAX], *dir;
677 aead84be 2022-01-10 op * Ugly workaround: we really need a smarter `include', one that
678 aead84be 2022-01-10 op * is able to resolve path relatively from the currently processed
679 aead84be 2022-01-10 op * file. The workaround consist to save the current directory,
680 aead84be 2022-01-10 op * chdir(2) to the script dirname and then jump back by mean of
684 aead84be 2022-01-10 op if ((pwdfd = open(".", O_RDONLY|O_DIRECTORY)) == -1)
685 aead84be 2022-01-10 op err(1, "can't open .");
686 aead84be 2022-01-10 op strlcpy(p, path, sizeof(p));
687 aead84be 2022-01-10 op dir = dirname(p);
688 aead84be 2022-01-10 op if (chdir(dir) == -1)
689 aead84be 2022-01-10 op err(1, "chdir %s", dir);
691 aead84be 2022-01-10 op /* XXX: include the *basename* of the file after chdir */
692 aead84be 2022-01-10 op strlcpy(p, path, sizeof(p));
693 aead84be 2022-01-10 op file = pushfile(basename(p));
694 fb1a36c0 2022-01-09 op if (file == NULL)
695 fb1a36c0 2022-01-09 op err(1, "pushfile");
696 fb1a36c0 2022-01-09 op topfile = file;
699 fb1a36c0 2022-01-09 op errors = file->errors;
702 aead84be 2022-01-10 op fchdir(pwdfd);
703 aead84be 2022-01-10 op close(pwdfd);
706 fb1a36c0 2022-01-09 op errx(1, "can't load %s because of errors", path);