Blame


1 fb1a36c0 2022-01-09 op /*
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.
10 fb1a36c0 2022-01-09 op *
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.
14 fb1a36c0 2022-01-09 op *
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.
22 fb1a36c0 2022-01-09 op */
23 fb1a36c0 2022-01-09 op
24 fb1a36c0 2022-01-09 op %{
25 fb1a36c0 2022-01-09 op
26 bbcba3ed 2022-01-10 op #include "compat.h"
27 fb1a36c0 2022-01-09 op
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>
40 fb1a36c0 2022-01-09 op
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"
44 fb1a36c0 2022-01-09 op
45 fb1a36c0 2022-01-09 op #include "script.h"
46 fb1a36c0 2022-01-09 op
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;
50 fb1a36c0 2022-01-09 op FILE *stream;
51 fb1a36c0 2022-01-09 op char *name;
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;
56 fb1a36c0 2022-01-09 op int lineno;
57 fb1a36c0 2022-01-09 op int errors;
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);
72 fb1a36c0 2022-01-09 op
73 fb1a36c0 2022-01-09 op static int shouldfail;
74 fb1a36c0 2022-01-09 op
75 fb1a36c0 2022-01-09 op typedef struct {
76 fb1a36c0 2022-01-09 op union {
77 fb1a36c0 2022-01-09 op struct op *op;
78 fb1a36c0 2022-01-09 op struct proc *proc;
79 fb1a36c0 2022-01-09 op char *str;
80 fb1a36c0 2022-01-09 op int64_t num;
81 fb1a36c0 2022-01-09 op } v;
82 fb1a36c0 2022-01-09 op int lineno;
83 fb1a36c0 2022-01-09 op } YYSTYPE;
84 fb1a36c0 2022-01-09 op
85 fb1a36c0 2022-01-09 op %}
86 fb1a36c0 2022-01-09 op
87 fb1a36c0 2022-01-09 op /*
88 fb1a36c0 2022-01-09 op * for bison:
89 fb1a36c0 2022-01-09 op * %define parse.error verbose
90 fb1a36c0 2022-01-09 op */
91 fb1a36c0 2022-01-09 op
92 fb1a36c0 2022-01-09 op %token ASSERT
93 fb1a36c0 2022-01-09 op %token CONST
94 fb1a36c0 2022-01-09 op %token ERROR
95 fb1a36c0 2022-01-09 op %token INCLUDE
96 fb1a36c0 2022-01-09 op %token PROC
97 fb1a36c0 2022-01-09 op %token REPEAT
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
101 fb1a36c0 2022-01-09 op %token VARGS
102 fb1a36c0 2022-01-09 op
103 fb1a36c0 2022-01-09 op %token <v.str> STRING SYMBOL
104 fb1a36c0 2022-01-09 op %token <v.num> NUMBER
105 fb1a36c0 2022-01-09 op
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
108 fb1a36c0 2022-01-09 op
109 fb1a36c0 2022-01-09 op %type <v.proc> procname
110 fb1a36c0 2022-01-09 op
111 fb1a36c0 2022-01-09 op %%
112 fb1a36c0 2022-01-09 op
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'
119 fb1a36c0 2022-01-09 op ;
120 fb1a36c0 2022-01-09 op
121 fb1a36c0 2022-01-09 op optnl : '\n' optnl /* zero or more newlines */
122 fb1a36c0 2022-01-09 op | /*empty*/
123 fb1a36c0 2022-01-09 op ;
124 fb1a36c0 2022-01-09 op
125 fb1a36c0 2022-01-09 op nl : '\n' optnl ;
126 fb1a36c0 2022-01-09 op
127 fb1a36c0 2022-01-09 op include : INCLUDE STRING {
128 fb1a36c0 2022-01-09 op struct file *nfile;
129 fb1a36c0 2022-01-09 op
130 fb1a36c0 2022-01-09 op if ((nfile = pushfile($2)) == NULL) {
131 fb1a36c0 2022-01-09 op yyerror("failed to include file %s", $2);
132 fb1a36c0 2022-01-09 op free($2);
133 fb1a36c0 2022-01-09 op YYERROR;
134 fb1a36c0 2022-01-09 op }
135 fb1a36c0 2022-01-09 op free($2);
136 fb1a36c0 2022-01-09 op
137 fb1a36c0 2022-01-09 op file = nfile;
138 fb1a36c0 2022-01-09 op lungetc('\n');
139 fb1a36c0 2022-01-09 op }
140 fb1a36c0 2022-01-09 op ;
141 fb1a36c0 2022-01-09 op
142 fb1a36c0 2022-01-09 op const : CONST consti
143 fb1a36c0 2022-01-09 op | CONST '(' optnl mconst ')'
144 fb1a36c0 2022-01-09 op ;
145 fb1a36c0 2022-01-09 op
146 fb1a36c0 2022-01-09 op mconst : consti nl | mconst consti nl ;
147 fb1a36c0 2022-01-09 op
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);
151 fb1a36c0 2022-01-09 op free($1);
152 fb1a36c0 2022-01-09 op free_op($3);
153 fb1a36c0 2022-01-09 op YYERROR;
154 fb1a36c0 2022-01-09 op }
155 fb1a36c0 2022-01-09 op }
156 fb1a36c0 2022-01-09 op ;
157 fb1a36c0 2022-01-09 op
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); } ;
162 fb1a36c0 2022-01-09 op
163 fb1a36c0 2022-01-09 op /*
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.
166 fb1a36c0 2022-01-09 op */
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); }
170 fb1a36c0 2022-01-09 op ;
171 fb1a36c0 2022-01-09 op
172 fb1a36c0 2022-01-09 op expr : literal | funcall | varref | check | cast | faccess | vargs ;
173 fb1a36c0 2022-01-09 op
174 fb1a36c0 2022-01-09 op vargs : VARGS { $$ = op_vargs(); } ;
175 fb1a36c0 2022-01-09 op
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); }
180 fb1a36c0 2022-01-09 op ;
181 fb1a36c0 2022-01-09 op
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); }
184 fb1a36c0 2022-01-09 op ;
185 fb1a36c0 2022-01-09 op
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);
189 fb1a36c0 2022-01-09 op free($1);
190 fb1a36c0 2022-01-09 op YYERROR;
191 fb1a36c0 2022-01-09 op }
192 fb1a36c0 2022-01-09 op free($1);
193 fb1a36c0 2022-01-09 op }
194 fb1a36c0 2022-01-09 op ;
195 fb1a36c0 2022-01-09 op
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;
200 fb1a36c0 2022-01-09 op int argc;
201 fb1a36c0 2022-01-09 op
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;
205 fb1a36c0 2022-01-09 op
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 $$ */
210 fb1a36c0 2022-01-09 op YYERROR;
211 fb1a36c0 2022-01-09 op }
212 fb1a36c0 2022-01-09 op
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,
216 fb1a36c0 2022-01-09 op argc);
217 fb1a36c0 2022-01-09 op /* TODO: recursively free $$ */
218 fb1a36c0 2022-01-09 op YYERROR;
219 fb1a36c0 2022-01-09 op }
220 fb1a36c0 2022-01-09 op }
221 fb1a36c0 2022-01-09 op ;
222 fb1a36c0 2022-01-09 op
223 fb1a36c0 2022-01-09 op optcomma: /* empty */ | ',' ;
224 fb1a36c0 2022-01-09 op
225 fb1a36c0 2022-01-09 op dots : '.' '.' '.' ;
226 fb1a36c0 2022-01-09 op
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()); }
232 fb1a36c0 2022-01-09 op ;
233 fb1a36c0 2022-01-09 op
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",
239 fb1a36c0 2022-01-09 op $2);
240 fb1a36c0 2022-01-09 op free($2);
241 fb1a36c0 2022-01-09 op YYERROR;
242 fb1a36c0 2022-01-09 op }
243 fb1a36c0 2022-01-09 op } '{' optnl block '}' {
244 fb1a36c0 2022-01-09 op proc_done($2);
245 fb1a36c0 2022-01-09 op }
246 fb1a36c0 2022-01-09 op ;
247 fb1a36c0 2022-01-09 op
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); }
253 fb1a36c0 2022-01-09 op ;
254 fb1a36c0 2022-01-09 op
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); }
257 fb1a36c0 2022-01-09 op ;
258 fb1a36c0 2022-01-09 op
259 fb1a36c0 2022-01-09 op assert : ASSERT asserti
260 fb1a36c0 2022-01-09 op | ASSERT '(' optnl massert ')'
261 fb1a36c0 2022-01-09 op ;
262 fb1a36c0 2022-01-09 op
263 fb1a36c0 2022-01-09 op massert : asserti nl | massert asserti nl ;
264 fb1a36c0 2022-01-09 op
265 fb1a36c0 2022-01-09 op asserti : check { block_push(op_assert($1)); }
266 fb1a36c0 2022-01-09 op ;
267 fb1a36c0 2022-01-09 op
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;
273 fb1a36c0 2022-01-09 op }
274 fb1a36c0 2022-01-09 op ;
275 fb1a36c0 2022-01-09 op
276 fb1a36c0 2022-01-09 op testopt : /* empty */
277 fb1a36c0 2022-01-09 op | SHOULD_FAIL { shouldfail = 1; }
278 fb1a36c0 2022-01-09 op ;
279 fb1a36c0 2022-01-09 op
280 fb1a36c0 2022-01-09 op %%
281 fb1a36c0 2022-01-09 op
282 fb1a36c0 2022-01-09 op struct keywords {
283 fb1a36c0 2022-01-09 op const char *k_name;
284 fb1a36c0 2022-01-09 op int k_val;
285 fb1a36c0 2022-01-09 op };
286 fb1a36c0 2022-01-09 op
287 fb1a36c0 2022-01-09 op int
288 fb1a36c0 2022-01-09 op yyerror(const char *fmt, ...)
289 fb1a36c0 2022-01-09 op {
290 fb1a36c0 2022-01-09 op va_list ap;
291 fb1a36c0 2022-01-09 op char *msg;
292 fb1a36c0 2022-01-09 op
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");
297 fb1a36c0 2022-01-09 op va_end(ap);
298 fb1a36c0 2022-01-09 op logit(LOG_CRIT, "%s:%d: %s", file->name, yylval.lineno, msg);
299 fb1a36c0 2022-01-09 op free(msg);
300 fb1a36c0 2022-01-09 op return 0;
301 fb1a36c0 2022-01-09 op }
302 fb1a36c0 2022-01-09 op
303 fb1a36c0 2022-01-09 op int
304 fb1a36c0 2022-01-09 op kw_cmp(const void *k, const void *e)
305 fb1a36c0 2022-01-09 op {
306 fb1a36c0 2022-01-09 op return strcmp(k, ((const struct keywords *)e)->k_name);
307 fb1a36c0 2022-01-09 op }
308 fb1a36c0 2022-01-09 op
309 fb1a36c0 2022-01-09 op int
310 fb1a36c0 2022-01-09 op lookup(char *s)
311 fb1a36c0 2022-01-09 op {
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},
324 fb1a36c0 2022-01-09 op {"u8", U8},
325 fb1a36c0 2022-01-09 op {"vargs", VARGS},
326 fb1a36c0 2022-01-09 op };
327 fb1a36c0 2022-01-09 op const struct keywords *p;
328 fb1a36c0 2022-01-09 op
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);
331 fb1a36c0 2022-01-09 op
332 fb1a36c0 2022-01-09 op if (p)
333 fb1a36c0 2022-01-09 op return p->k_val;
334 fb1a36c0 2022-01-09 op else
335 fb1a36c0 2022-01-09 op return SYMBOL;
336 fb1a36c0 2022-01-09 op }
337 fb1a36c0 2022-01-09 op
338 fb1a36c0 2022-01-09 op #define START_EXPAND 1
339 fb1a36c0 2022-01-09 op #define DONE_EXPAND 2
340 fb1a36c0 2022-01-09 op
341 fb1a36c0 2022-01-09 op static int expanding;
342 fb1a36c0 2022-01-09 op
343 fb1a36c0 2022-01-09 op int
344 fb1a36c0 2022-01-09 op igetc(void)
345 fb1a36c0 2022-01-09 op {
346 fb1a36c0 2022-01-09 op int c;
347 fb1a36c0 2022-01-09 op
348 fb1a36c0 2022-01-09 op while (1) {
349 fb1a36c0 2022-01-09 op if (file->ungetpos > 0)
350 fb1a36c0 2022-01-09 op c = file->ungetbuf[--file->ungetpos];
351 fb1a36c0 2022-01-09 op else
352 fb1a36c0 2022-01-09 op c = getc(file->stream);
353 fb1a36c0 2022-01-09 op
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;
358 fb1a36c0 2022-01-09 op else
359 fb1a36c0 2022-01-09 op break;
360 fb1a36c0 2022-01-09 op }
361 fb1a36c0 2022-01-09 op return c;
362 fb1a36c0 2022-01-09 op }
363 fb1a36c0 2022-01-09 op
364 fb1a36c0 2022-01-09 op int
365 fb1a36c0 2022-01-09 op lgetc(int quotec)
366 fb1a36c0 2022-01-09 op {
367 fb1a36c0 2022-01-09 op int c, next;
368 fb1a36c0 2022-01-09 op
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)
374 fb1a36c0 2022-01-09 op return EOF;
375 fb1a36c0 2022-01-09 op return quotec;
376 fb1a36c0 2022-01-09 op }
377 fb1a36c0 2022-01-09 op return c;
378 fb1a36c0 2022-01-09 op }
379 fb1a36c0 2022-01-09 op
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') {
383 fb1a36c0 2022-01-09 op c = next;
384 fb1a36c0 2022-01-09 op break;
385 fb1a36c0 2022-01-09 op }
386 fb1a36c0 2022-01-09 op yylval.lineno = file->lineno;
387 fb1a36c0 2022-01-09 op file->lineno++;
388 fb1a36c0 2022-01-09 op }
389 fb1a36c0 2022-01-09 op
390 fb1a36c0 2022-01-09 op if (c == EOF) {
391 fb1a36c0 2022-01-09 op /*
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.
395 fb1a36c0 2022-01-09 op */
396 fb1a36c0 2022-01-09 op if (file->eof_reached == 0) {
397 fb1a36c0 2022-01-09 op file->eof_reached = 1;
398 fb1a36c0 2022-01-09 op return '\n';
399 fb1a36c0 2022-01-09 op }
400 fb1a36c0 2022-01-09 op while (c == EOF) {
401 fb1a36c0 2022-01-09 op if (file == topfile || popfile() == EOF)
402 fb1a36c0 2022-01-09 op return EOF;
403 fb1a36c0 2022-01-09 op c = igetc();
404 fb1a36c0 2022-01-09 op }
405 fb1a36c0 2022-01-09 op }
406 fb1a36c0 2022-01-09 op return c;
407 fb1a36c0 2022-01-09 op }
408 fb1a36c0 2022-01-09 op
409 fb1a36c0 2022-01-09 op void
410 fb1a36c0 2022-01-09 op lungetc(int c)
411 fb1a36c0 2022-01-09 op {
412 fb1a36c0 2022-01-09 op if (c == EOF)
413 fb1a36c0 2022-01-09 op return;
414 fb1a36c0 2022-01-09 op
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;
421 fb1a36c0 2022-01-09 op }
422 fb1a36c0 2022-01-09 op file->ungetbuf[file->ungetpos++] = c;
423 fb1a36c0 2022-01-09 op }
424 fb1a36c0 2022-01-09 op
425 fb1a36c0 2022-01-09 op int
426 fb1a36c0 2022-01-09 op findeol(void)
427 fb1a36c0 2022-01-09 op {
428 fb1a36c0 2022-01-09 op int c;
429 fb1a36c0 2022-01-09 op
430 fb1a36c0 2022-01-09 op /* Skip to either EOF or the first real EOL. */
431 fb1a36c0 2022-01-09 op while (1) {
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++;
435 fb1a36c0 2022-01-09 op break;
436 fb1a36c0 2022-01-09 op }
437 fb1a36c0 2022-01-09 op if (c == EOF)
438 fb1a36c0 2022-01-09 op break;
439 fb1a36c0 2022-01-09 op }
440 fb1a36c0 2022-01-09 op return ERROR;
441 fb1a36c0 2022-01-09 op }
442 fb1a36c0 2022-01-09 op
443 fb1a36c0 2022-01-09 op
444 fb1a36c0 2022-01-09 op #if 0
445 fb1a36c0 2022-01-09 op int my_yylex(void);
446 fb1a36c0 2022-01-09 op
447 fb1a36c0 2022-01-09 op int
448 fb1a36c0 2022-01-09 op yylex(void)
449 fb1a36c0 2022-01-09 op {
450 fb1a36c0 2022-01-09 op int x;
451 fb1a36c0 2022-01-09 op
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;
464 fb1a36c0 2022-01-09 op
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;
468 fb1a36c0 2022-01-09 op
469 fb1a36c0 2022-01-09 op default:
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");
473 fb1a36c0 2022-01-09 op else
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");
477 fb1a36c0 2022-01-09 op break;
478 fb1a36c0 2022-01-09 op }
479 fb1a36c0 2022-01-09 op
480 fb1a36c0 2022-01-09 op return x;
481 fb1a36c0 2022-01-09 op }
482 fb1a36c0 2022-01-09 op
483 fb1a36c0 2022-01-09 op int
484 fb1a36c0 2022-01-09 op my_yylex(void)
485 fb1a36c0 2022-01-09 op #else
486 fb1a36c0 2022-01-09 op int
487 fb1a36c0 2022-01-09 op yylex(void)
488 fb1a36c0 2022-01-09 op #endif
489 fb1a36c0 2022-01-09 op {
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;
493 fb1a36c0 2022-01-09 op int token;
494 fb1a36c0 2022-01-09 op
495 fb1a36c0 2022-01-09 op p = buf;
496 fb1a36c0 2022-01-09 op while ((c = lgetc(0)) == ' ' || c == '\t' || c == '\f')
497 fb1a36c0 2022-01-09 op ; /* nop */
498 fb1a36c0 2022-01-09 op
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)
502 fb1a36c0 2022-01-09 op ; /* nop */
503 fb1a36c0 2022-01-09 op
504 fb1a36c0 2022-01-09 op switch (c) {
505 fb1a36c0 2022-01-09 op case ':':
506 fb1a36c0 2022-01-09 op return c;
507 fb1a36c0 2022-01-09 op break;
508 fb1a36c0 2022-01-09 op case '\'':
509 fb1a36c0 2022-01-09 op case '\"':
510 fb1a36c0 2022-01-09 op quotec = c;
511 fb1a36c0 2022-01-09 op while (1) {
512 fb1a36c0 2022-01-09 op if ((c = lgetc(quotec)) == EOF)
513 fb1a36c0 2022-01-09 op return 0;
514 fb1a36c0 2022-01-09 op if (c == '\n') {
515 fb1a36c0 2022-01-09 op file->lineno++;
516 fb1a36c0 2022-01-09 op continue;
517 fb1a36c0 2022-01-09 op } else if (c == '\\') {
518 fb1a36c0 2022-01-09 op if ((next = lgetc(quotec)) == EOF)
519 fb1a36c0 2022-01-09 op return 0;
520 fb1a36c0 2022-01-09 op if (next == quotec || next == ' ' ||
521 fb1a36c0 2022-01-09 op next == '\t')
522 fb1a36c0 2022-01-09 op c = next;
523 fb1a36c0 2022-01-09 op else if (next == '\n') {
524 fb1a36c0 2022-01-09 op file->lineno++;
525 fb1a36c0 2022-01-09 op continue;
526 fb1a36c0 2022-01-09 op } else
527 fb1a36c0 2022-01-09 op lungetc(next);
528 fb1a36c0 2022-01-09 op } else if (c == quotec) {
529 fb1a36c0 2022-01-09 op *p = '\0';
530 fb1a36c0 2022-01-09 op break;
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();
534 fb1a36c0 2022-01-09 op }
535 fb1a36c0 2022-01-09 op
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();
539 fb1a36c0 2022-01-09 op }
540 fb1a36c0 2022-01-09 op
541 fb1a36c0 2022-01-09 op *p++ = c;
542 fb1a36c0 2022-01-09 op }
543 fb1a36c0 2022-01-09 op
544 fb1a36c0 2022-01-09 op yylval.v.str = xstrdup(buf);
545 fb1a36c0 2022-01-09 op return STRING;
546 fb1a36c0 2022-01-09 op }
547 fb1a36c0 2022-01-09 op
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 == ':')
551 fb1a36c0 2022-01-09 op
552 fb1a36c0 2022-01-09 op if (c == '-' || isdigit(c)) {
553 fb1a36c0 2022-01-09 op do {
554 fb1a36c0 2022-01-09 op *p++ = 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();
558 fb1a36c0 2022-01-09 op }
559 fb1a36c0 2022-01-09 op } while ((c = lgetc(0)) != EOF && (isdigit(c) || c == 'x'));
560 fb1a36c0 2022-01-09 op lungetc(c);
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)) {
564 fb1a36c0 2022-01-09 op char *ep;
565 fb1a36c0 2022-01-09 op
566 fb1a36c0 2022-01-09 op *p = '\0';
567 fb1a36c0 2022-01-09 op errno = 0;
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",
573 fb1a36c0 2022-01-09 op buf);
574 fb1a36c0 2022-01-09 op return findeol();
575 fb1a36c0 2022-01-09 op }
576 fb1a36c0 2022-01-09 op
577 fb1a36c0 2022-01-09 op return NUMBER;
578 fb1a36c0 2022-01-09 op } else {
579 fb1a36c0 2022-01-09 op nodigits:
580 fb1a36c0 2022-01-09 op while (p > buf + 1)
581 fb1a36c0 2022-01-09 op lungetc(*--p);
582 fb1a36c0 2022-01-09 op c = *--p;
583 fb1a36c0 2022-01-09 op if (c == '-')
584 fb1a36c0 2022-01-09 op return c;
585 fb1a36c0 2022-01-09 op }
586 fb1a36c0 2022-01-09 op }
587 fb1a36c0 2022-01-09 op
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 != ':'))
594 fb1a36c0 2022-01-09 op
595 fb1a36c0 2022-01-09 op if (isalnum(c) || c == ':' || c == '_') {
596 fb1a36c0 2022-01-09 op do {
597 fb1a36c0 2022-01-09 op *p++ = 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();
601 fb1a36c0 2022-01-09 op }
602 fb1a36c0 2022-01-09 op } while ((c = lgetc(0)) != EOF && (allowed_in_symbol(c)));
603 fb1a36c0 2022-01-09 op lungetc(c);
604 fb1a36c0 2022-01-09 op *p = '\0';
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;
608 fb1a36c0 2022-01-09 op }
609 fb1a36c0 2022-01-09 op
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++;
613 fb1a36c0 2022-01-09 op }
614 fb1a36c0 2022-01-09 op if (c == EOF)
615 fb1a36c0 2022-01-09 op return 0;
616 fb1a36c0 2022-01-09 op return c;
617 fb1a36c0 2022-01-09 op }
618 fb1a36c0 2022-01-09 op
619 fb1a36c0 2022-01-09 op struct file *
620 fb1a36c0 2022-01-09 op pushfile(const char *name)
621 fb1a36c0 2022-01-09 op {
622 fb1a36c0 2022-01-09 op struct file *nfile;
623 fb1a36c0 2022-01-09 op
624 fb1a36c0 2022-01-09 op if ((nfile = calloc(1, sizeof(struct file))) == NULL) {
625 fb1a36c0 2022-01-09 op log_warn("calloc");
626 fb1a36c0 2022-01-09 op return NULL;
627 fb1a36c0 2022-01-09 op }
628 fb1a36c0 2022-01-09 op if ((nfile->name = strdup(name)) == NULL) {
629 fb1a36c0 2022-01-09 op log_warn("strdup");
630 fb1a36c0 2022-01-09 op free(nfile);
631 fb1a36c0 2022-01-09 op return NULL;
632 fb1a36c0 2022-01-09 op }
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);
636 fb1a36c0 2022-01-09 op free(nfile);
637 fb1a36c0 2022-01-09 op return NULL;
638 fb1a36c0 2022-01-09 op }
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);
646 fb1a36c0 2022-01-09 op free(nfile);
647 fb1a36c0 2022-01-09 op return NULL;
648 fb1a36c0 2022-01-09 op }
649 fb1a36c0 2022-01-09 op TAILQ_INSERT_TAIL(&files, nfile, entry);
650 fb1a36c0 2022-01-09 op return nfile;
651 fb1a36c0 2022-01-09 op }
652 fb1a36c0 2022-01-09 op
653 fb1a36c0 2022-01-09 op int
654 fb1a36c0 2022-01-09 op popfile(void)
655 fb1a36c0 2022-01-09 op {
656 fb1a36c0 2022-01-09 op struct file *prev;
657 fb1a36c0 2022-01-09 op
658 fb1a36c0 2022-01-09 op if ((prev = TAILQ_PREV(file, files, entry)) != NULL)
659 fb1a36c0 2022-01-09 op prev->errors += file->errors;
660 fb1a36c0 2022-01-09 op
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);
665 fb1a36c0 2022-01-09 op free(file);
666 fb1a36c0 2022-01-09 op file = prev;
667 fb1a36c0 2022-01-09 op return file ? 0 : EOF;
668 fb1a36c0 2022-01-09 op }
669 fb1a36c0 2022-01-09 op
670 fb1a36c0 2022-01-09 op void
671 fb1a36c0 2022-01-09 op loadfile(const char *path)
672 fb1a36c0 2022-01-09 op {
673 aead84be 2022-01-10 op int pwdfd, errors;
674 aead84be 2022-01-10 op char p[PATH_MAX], *dir;
675 fb1a36c0 2022-01-09 op
676 aead84be 2022-01-10 op /*
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
681 aead84be 2022-01-10 op * fchdir.
682 aead84be 2022-01-10 op */
683 aead84be 2022-01-10 op
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);
690 aead84be 2022-01-10 op
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;
697 fb1a36c0 2022-01-09 op
698 fb1a36c0 2022-01-09 op yyparse();
699 fb1a36c0 2022-01-09 op errors = file->errors;
700 fb1a36c0 2022-01-09 op popfile();
701 fb1a36c0 2022-01-09 op
702 aead84be 2022-01-10 op fchdir(pwdfd);
703 aead84be 2022-01-10 op close(pwdfd);
704 aead84be 2022-01-10 op
705 fb1a36c0 2022-01-09 op if (errors)
706 fb1a36c0 2022-01-09 op errx(1, "can't load %s because of errors", path);
707 fb1a36c0 2022-01-09 op }