Blame


1 bff58270 2022-09-19 op /*
2 52afe358 2022-12-17 op * Copyright (c) 2022 Omar Polo <op@openbsd.org>
3 bff58270 2022-09-19 op * Copyright (c) 2007-2016 Reyk Floeter <reyk@openbsd.org>
4 bff58270 2022-09-19 op * Copyright (c) 2004, 2005 Esben Norby <norby@openbsd.org>
5 bff58270 2022-09-19 op * Copyright (c) 2004 Ryan McBride <mcbride@openbsd.org>
6 bff58270 2022-09-19 op * Copyright (c) 2002, 2003, 2004 Henning Brauer <henning@openbsd.org>
7 bff58270 2022-09-19 op * Copyright (c) 2001 Markus Friedl. All rights reserved.
8 bff58270 2022-09-19 op * Copyright (c) 2001 Daniel Hartmeier. All rights reserved.
9 bff58270 2022-09-19 op * Copyright (c) 2001 Theo de Raadt. All rights reserved.
10 bff58270 2022-09-19 op *
11 bff58270 2022-09-19 op * Permission to use, copy, modify, and distribute this software for any
12 bff58270 2022-09-19 op * purpose with or without fee is hereby granted, provided that the above
13 bff58270 2022-09-19 op * copyright notice and this permission notice appear in all copies.
14 bff58270 2022-09-19 op *
15 bff58270 2022-09-19 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
16 bff58270 2022-09-19 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
17 bff58270 2022-09-19 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
18 bff58270 2022-09-19 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
19 bff58270 2022-09-19 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
20 bff58270 2022-09-19 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
21 bff58270 2022-09-19 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
22 bff58270 2022-09-19 op */
23 bff58270 2022-09-19 op
24 bff58270 2022-09-19 op %{
25 bff58270 2022-09-19 op
26 bff58270 2022-09-19 op #include <sys/queue.h>
27 bff58270 2022-09-19 op
28 bff58270 2022-09-19 op #include <ctype.h>
29 bff58270 2022-09-19 op #include <err.h>
30 bff58270 2022-09-19 op #include <stdio.h>
31 bff58270 2022-09-19 op #include <stdlib.h>
32 bff58270 2022-09-19 op #include <stdarg.h>
33 bff58270 2022-09-19 op #include <stdint.h>
34 bff58270 2022-09-19 op #include <string.h>
35 bff58270 2022-09-19 op #include <unistd.h>
36 bff58270 2022-09-19 op
37 bff58270 2022-09-19 op #ifndef nitems
38 bff58270 2022-09-19 op #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
39 bff58270 2022-09-19 op #endif
40 bff58270 2022-09-19 op
41 bff58270 2022-09-19 op TAILQ_HEAD(files, file) files = TAILQ_HEAD_INITIALIZER(files);
42 bff58270 2022-09-19 op static struct file {
43 bff58270 2022-09-19 op TAILQ_ENTRY(file) entry;
44 bff58270 2022-09-19 op FILE *stream;
45 bff58270 2022-09-19 op char *name;
46 bff58270 2022-09-19 op size_t ungetpos;
47 bff58270 2022-09-19 op size_t ungetsize;
48 8eb6aff8 2022-09-30 op unsigned char *ungetbuf;
49 bff58270 2022-09-19 op int eof_reached;
50 bff58270 2022-09-19 op int lineno;
51 bff58270 2022-09-19 op int errors;
52 bff58270 2022-09-19 op } *file, *topfile;
53 14c31a43 2022-11-25 op int parse(FILE *, const char *);
54 bff58270 2022-09-19 op struct file *pushfile(const char *, int);
55 bff58270 2022-09-19 op int popfile(void);
56 bff58270 2022-09-19 op int yyparse(void);
57 bff58270 2022-09-19 op int yylex(void);
58 bff58270 2022-09-19 op int yyerror(const char *, ...)
59 bff58270 2022-09-19 op __attribute__((__format__ (printf, 1, 2)))
60 bff58270 2022-09-19 op __attribute__((__nonnull__ (1)));
61 bff58270 2022-09-19 op int kw_cmp(const void *, const void *);
62 bff58270 2022-09-19 op int lookup(char *);
63 bff58270 2022-09-19 op int igetc(void);
64 bff58270 2022-09-19 op int lgetc(int);
65 bff58270 2022-09-19 op void lungetc(int);
66 bff58270 2022-09-19 op int findeol(void);
67 bff58270 2022-09-19 op
68 bff58270 2022-09-19 op void dbg(void);
69 bff58270 2022-09-19 op void printq(const char *);
70 bff58270 2022-09-19 op
71 bff58270 2022-09-19 op extern int nodebug;
72 bff58270 2022-09-19 op
73 14c31a43 2022-11-25 op static FILE *fp;
74 14c31a43 2022-11-25 op
75 bff58270 2022-09-19 op static int block;
76 bff58270 2022-09-19 op static int in_define;
77 bff58270 2022-09-19 op static int errors;
78 9a90a3d1 2022-09-29 op static int lastline = -1;
79 bff58270 2022-09-19 op
80 bff58270 2022-09-19 op typedef struct {
81 bff58270 2022-09-19 op union {
82 bff58270 2022-09-19 op char *string;
83 bff58270 2022-09-19 op } v;
84 bff58270 2022-09-19 op int lineno;
85 bff58270 2022-09-19 op } YYSTYPE;
86 bff58270 2022-09-19 op
87 bff58270 2022-09-19 op %}
88 bff58270 2022-09-19 op
89 9c75d045 2022-10-03 op %token DEFINE ELSE END ERROR FINALLY FOR IF INCLUDE PRINTF
90 3e8b34ba 2023-01-07 op %token RENDER TQFOREACH UNSAFE URLESCAPE WHILE
91 bff58270 2022-09-19 op %token <v.string> STRING
92 279f742d 2023-09-12 op %type <v.string> string nstring
93 6125bd61 2022-09-29 op %type <v.string> stringy
94 bff58270 2022-09-19 op
95 bff58270 2022-09-19 op %%
96 bff58270 2022-09-19 op
97 bff58270 2022-09-19 op grammar : /* empty */
98 bff58270 2022-09-19 op | grammar include
99 bff58270 2022-09-19 op | grammar verbatim
100 bff58270 2022-09-19 op | grammar block
101 bff58270 2022-09-19 op | grammar error { file->errors++; }
102 bff58270 2022-09-19 op ;
103 bff58270 2022-09-19 op
104 bff58270 2022-09-19 op include : INCLUDE STRING {
105 bff58270 2022-09-19 op struct file *nfile;
106 bff58270 2022-09-19 op
107 bff58270 2022-09-19 op if ((nfile = pushfile($2, 0)) == NULL) {
108 bff58270 2022-09-19 op yyerror("failed to include file %s", $2);
109 bff58270 2022-09-19 op free($2);
110 bff58270 2022-09-19 op YYERROR;
111 bff58270 2022-09-19 op }
112 bff58270 2022-09-19 op free($2);
113 bff58270 2022-09-19 op
114 bff58270 2022-09-19 op file = nfile;
115 bff58270 2022-09-19 op lungetc('\n');
116 bff58270 2022-09-19 op }
117 bff58270 2022-09-19 op ;
118 bff58270 2022-09-19 op
119 bff58270 2022-09-19 op verbatim : '!' verbatim1 '!' {
120 bff58270 2022-09-19 op if (in_define) {
121 bff58270 2022-09-19 op /* TODO: check template status and exit in case */
122 bff58270 2022-09-19 op }
123 bff58270 2022-09-19 op }
124 bff58270 2022-09-19 op ;
125 bff58270 2022-09-19 op
126 bff58270 2022-09-19 op verbatim1 : /* empty */
127 bff58270 2022-09-19 op | verbatim1 STRING {
128 bff58270 2022-09-19 op if (*$2 != '\0') {
129 bff58270 2022-09-19 op dbg();
130 14c31a43 2022-11-25 op fprintf(fp, "%s\n", $2);
131 bff58270 2022-09-19 op }
132 bff58270 2022-09-19 op free($2);
133 bff58270 2022-09-19 op }
134 bff58270 2022-09-19 op ;
135 bff58270 2022-09-19 op
136 bff58270 2022-09-19 op verbatims : /* empty */
137 bff58270 2022-09-19 op | verbatims verbatim
138 bff58270 2022-09-19 op ;
139 bff58270 2022-09-19 op
140 279f742d 2023-09-12 op raw : nstring {
141 bff58270 2022-09-19 op dbg();
142 18906bcc 2023-09-13 op fprintf(fp, "if ((tp_ret = tp_write(tp, ");
143 bff58270 2022-09-19 op printq($1);
144 18906bcc 2023-09-13 op fprintf(fp, ", %zu)) == -1) goto err;\n",
145 18906bcc 2023-09-13 op strlen($1));
146 bff58270 2022-09-19 op
147 bff58270 2022-09-19 op free($1);
148 bff58270 2022-09-19 op }
149 bff58270 2022-09-19 op ;
150 bff58270 2022-09-19 op
151 bff58270 2022-09-19 op block : define body end {
152 14c31a43 2022-11-25 op fputs("err:\n", fp);
153 14c31a43 2022-11-25 op fputs("return tp_ret;\n", fp);
154 14c31a43 2022-11-25 op fputs("}\n", fp);
155 bff58270 2022-09-19 op in_define = 0;
156 bff58270 2022-09-19 op }
157 bff58270 2022-09-19 op | define body finally end {
158 14c31a43 2022-11-25 op fputs("return tp_ret;\n", fp);
159 14c31a43 2022-11-25 op fputs("}\n", fp);
160 bff58270 2022-09-19 op in_define = 0;
161 bff58270 2022-09-19 op }
162 bff58270 2022-09-19 op ;
163 bff58270 2022-09-19 op
164 bff58270 2022-09-19 op define : '{' DEFINE string '}' {
165 bff58270 2022-09-19 op in_define = 1;
166 bff58270 2022-09-19 op
167 bff58270 2022-09-19 op dbg();
168 14c31a43 2022-11-25 op fprintf(fp, "int\n%s\n{\n", $3);
169 14c31a43 2022-11-25 op fputs("int tp_ret = 0;\n", fp);
170 bff58270 2022-09-19 op
171 bff58270 2022-09-19 op free($3);
172 bff58270 2022-09-19 op }
173 bff58270 2022-09-19 op ;
174 bff58270 2022-09-19 op
175 bff58270 2022-09-19 op body : /* empty */
176 bff58270 2022-09-19 op | body verbatim
177 bff58270 2022-09-19 op | body raw
178 bff58270 2022-09-19 op | body special
179 bff58270 2022-09-19 op ;
180 bff58270 2022-09-19 op
181 bff58270 2022-09-19 op special : '{' RENDER string '}' {
182 bff58270 2022-09-19 op dbg();
183 14c31a43 2022-11-25 op fprintf(fp, "if ((tp_ret = %s) == -1) goto err;\n",
184 14c31a43 2022-11-25 op $3);
185 bff58270 2022-09-19 op free($3);
186 bff58270 2022-09-19 op }
187 83e0f74d 2022-09-29 op | printf
188 14c31a43 2022-11-25 op | if body endif { fputs("}\n", fp); }
189 bff58270 2022-09-19 op | loop
190 bff58270 2022-09-19 op | '{' string '|' UNSAFE '}' {
191 bff58270 2022-09-19 op dbg();
192 14c31a43 2022-11-25 op fprintf(fp,
193 18906bcc 2023-09-13 op "if ((tp_ret = tp_writes(tp, %s)) == -1)\n",
194 83f0f95a 2022-09-29 op $2);
195 14c31a43 2022-11-25 op fputs("goto err;\n", fp);
196 bff58270 2022-09-19 op free($2);
197 bff58270 2022-09-19 op }
198 bff58270 2022-09-19 op | '{' string '|' URLESCAPE '}' {
199 bff58270 2022-09-19 op dbg();
200 14c31a43 2022-11-25 op fprintf(fp,
201 14c31a43 2022-11-25 op "if ((tp_ret = tp_urlescape(tp, %s)) == -1)\n",
202 83f0f95a 2022-09-29 op $2);
203 14c31a43 2022-11-25 op fputs("goto err;\n", fp);
204 bff58270 2022-09-19 op free($2);
205 bff58270 2022-09-19 op }
206 bff58270 2022-09-19 op | '{' string '}' {
207 bff58270 2022-09-19 op dbg();
208 14c31a43 2022-11-25 op fprintf(fp,
209 18906bcc 2023-09-13 op "if ((tp_ret = tp_htmlescape(tp, %s)) == -1)\n",
210 83f0f95a 2022-09-29 op $2);
211 14c31a43 2022-11-25 op fputs("goto err;\n", fp);
212 bff58270 2022-09-19 op free($2);
213 83e0f74d 2022-09-29 op }
214 83e0f74d 2022-09-29 op ;
215 83e0f74d 2022-09-29 op
216 83e0f74d 2022-09-29 op printf : '{' PRINTF {
217 83e0f74d 2022-09-29 op dbg();
218 14c31a43 2022-11-25 op fprintf(fp, "if (asprintf(&tp->tp_tmp, ");
219 83e0f74d 2022-09-29 op } printfargs '}' {
220 14c31a43 2022-11-25 op fputs(") == -1)\n", fp);
221 14c31a43 2022-11-25 op fputs("goto err;\n", fp);
222 18906bcc 2023-09-13 op fputs("if ((tp_ret = tp_htmlescape(tp, tp->tp_tmp)) "
223 14c31a43 2022-11-25 op "== -1)\n", fp);
224 14c31a43 2022-11-25 op fputs("goto err;\n", fp);
225 14c31a43 2022-11-25 op fputs("free(tp->tp_tmp);\n", fp);
226 14c31a43 2022-11-25 op fputs("tp->tp_tmp = NULL;\n", fp);
227 bff58270 2022-09-19 op }
228 bff58270 2022-09-19 op ;
229 83e0f74d 2022-09-29 op
230 83e0f74d 2022-09-29 op printfargs : /* empty */
231 83e0f74d 2022-09-29 op | printfargs STRING {
232 14c31a43 2022-11-25 op fprintf(fp, " %s", $2);
233 83e0f74d 2022-09-29 op free($2);
234 83e0f74d 2022-09-29 op }
235 6125bd61 2022-09-29 op ;
236 bff58270 2022-09-19 op
237 6125bd61 2022-09-29 op if : '{' IF stringy '}' {
238 bff58270 2022-09-19 op dbg();
239 14c31a43 2022-11-25 op fprintf(fp, "if (%s) {\n", $3);
240 bff58270 2022-09-19 op free($3);
241 bff58270 2022-09-19 op }
242 bff58270 2022-09-19 op ;
243 bff58270 2022-09-19 op
244 bff58270 2022-09-19 op endif : end
245 bff58270 2022-09-19 op | else body end
246 bff58270 2022-09-19 op | elsif body endif
247 bff58270 2022-09-19 op ;
248 bff58270 2022-09-19 op
249 6125bd61 2022-09-29 op elsif : '{' ELSE IF stringy '}' {
250 bff58270 2022-09-19 op dbg();
251 14c31a43 2022-11-25 op fprintf(fp, "} else if (%s) {\n", $4);
252 bff58270 2022-09-19 op free($4);
253 bff58270 2022-09-19 op }
254 bff58270 2022-09-19 op ;
255 bff58270 2022-09-19 op
256 bff58270 2022-09-19 op else : '{' ELSE '}' {
257 bff58270 2022-09-19 op dbg();
258 14c31a43 2022-11-25 op fputs("} else {\n", fp);
259 bff58270 2022-09-19 op }
260 bff58270 2022-09-19 op ;
261 bff58270 2022-09-19 op
262 9c75d045 2022-10-03 op loop : '{' FOR stringy '}' {
263 14c31a43 2022-11-25 op fprintf(fp, "for (%s) {\n", $3);
264 9c75d045 2022-10-03 op free($3);
265 9c75d045 2022-10-03 op } body end {
266 14c31a43 2022-11-25 op fputs("}\n", fp);
267 9c75d045 2022-10-03 op }
268 9c75d045 2022-10-03 op | '{' TQFOREACH STRING STRING STRING '}' {
269 14c31a43 2022-11-25 op fprintf(fp, "TAILQ_FOREACH(%s, %s, %s) {\n",
270 bff58270 2022-09-19 op $3, $4, $5);
271 bff58270 2022-09-19 op free($3);
272 bff58270 2022-09-19 op free($4);
273 bff58270 2022-09-19 op free($5);
274 bff58270 2022-09-19 op } body end {
275 14c31a43 2022-11-25 op fputs("}\n", fp);
276 bff58270 2022-09-19 op }
277 3e8b34ba 2023-01-07 op | '{' WHILE stringy '}' {
278 3e8b34ba 2023-01-07 op fprintf(fp, "while (%s) {\n", $3);
279 3e8b34ba 2023-01-07 op free($3);
280 3e8b34ba 2023-01-07 op } body end {
281 3e8b34ba 2023-01-07 op fputs("}\n", fp);
282 3e8b34ba 2023-01-07 op }
283 bff58270 2022-09-19 op ;
284 bff58270 2022-09-19 op
285 bff58270 2022-09-19 op end : '{' END '}'
286 bff58270 2022-09-19 op ;
287 bff58270 2022-09-19 op
288 bff58270 2022-09-19 op finally : '{' FINALLY '}' {
289 bff58270 2022-09-19 op dbg();
290 14c31a43 2022-11-25 op fputs("err:\n", fp);
291 bff58270 2022-09-19 op } verbatims
292 bff58270 2022-09-19 op ;
293 bff58270 2022-09-19 op
294 279f742d 2023-09-12 op nstring : STRING nstring {
295 279f742d 2023-09-12 op if (asprintf(&$$, "%s%s", $1, $2) == -1)
296 279f742d 2023-09-12 op err(1, "asprintf");
297 279f742d 2023-09-12 op free($1);
298 279f742d 2023-09-12 op free($2);
299 279f742d 2023-09-12 op }
300 279f742d 2023-09-12 op | STRING
301 279f742d 2023-09-12 op ;
302 279f742d 2023-09-12 op
303 bff58270 2022-09-19 op string : STRING string {
304 bff58270 2022-09-19 op if (asprintf(&$$, "%s %s", $1, $2) == -1)
305 bff58270 2022-09-19 op err(1, "asprintf");
306 bff58270 2022-09-19 op free($1);
307 bff58270 2022-09-19 op free($2);
308 bff58270 2022-09-19 op }
309 bff58270 2022-09-19 op | STRING
310 6125bd61 2022-09-29 op ;
311 6125bd61 2022-09-29 op
312 6125bd61 2022-09-29 op stringy : STRING
313 6125bd61 2022-09-29 op | STRING stringy {
314 6125bd61 2022-09-29 op if (asprintf(&$$, "%s %s", $1, $2) == -1)
315 6125bd61 2022-09-29 op err(1, "asprintf");
316 6125bd61 2022-09-29 op free($1);
317 6125bd61 2022-09-29 op free($2);
318 6125bd61 2022-09-29 op }
319 6125bd61 2022-09-29 op | '|' stringy {
320 6125bd61 2022-09-29 op if (asprintf(&$$, "|%s", $2) == -1)
321 6125bd61 2022-09-29 op err(1, "asprintf");
322 6125bd61 2022-09-29 op free($2);
323 6125bd61 2022-09-29 op }
324 bff58270 2022-09-19 op ;
325 bff58270 2022-09-19 op
326 bff58270 2022-09-19 op %%
327 bff58270 2022-09-19 op
328 bff58270 2022-09-19 op struct keywords {
329 bff58270 2022-09-19 op const char *k_name;
330 bff58270 2022-09-19 op int k_val;
331 bff58270 2022-09-19 op };
332 bff58270 2022-09-19 op
333 bff58270 2022-09-19 op int
334 bff58270 2022-09-19 op yyerror(const char *fmt, ...)
335 bff58270 2022-09-19 op {
336 bff58270 2022-09-19 op va_list ap;
337 bff58270 2022-09-19 op char *msg;
338 bff58270 2022-09-19 op
339 bff58270 2022-09-19 op file->errors++;
340 bff58270 2022-09-19 op va_start(ap, fmt);
341 bff58270 2022-09-19 op if (vasprintf(&msg, fmt, ap) == -1)
342 bff58270 2022-09-19 op err(1, "yyerror vasprintf");
343 bff58270 2022-09-19 op va_end(ap);
344 bff58270 2022-09-19 op fprintf(stderr, "%s:%d: %s\n", file->name, yylval.lineno, msg);
345 bff58270 2022-09-19 op free(msg);
346 bff58270 2022-09-19 op return (0);
347 bff58270 2022-09-19 op }
348 bff58270 2022-09-19 op
349 bff58270 2022-09-19 op int
350 bff58270 2022-09-19 op kw_cmp(const void *k, const void *e)
351 bff58270 2022-09-19 op {
352 bff58270 2022-09-19 op return (strcmp(k, ((const struct keywords *)e)->k_name));
353 bff58270 2022-09-19 op }
354 bff58270 2022-09-19 op
355 bff58270 2022-09-19 op int
356 bff58270 2022-09-19 op lookup(char *s)
357 bff58270 2022-09-19 op {
358 bff58270 2022-09-19 op /* this has to be sorted always */
359 bff58270 2022-09-19 op static const struct keywords keywords[] = {
360 bff58270 2022-09-19 op { "define", DEFINE },
361 bff58270 2022-09-19 op { "else", ELSE },
362 bff58270 2022-09-19 op { "end", END },
363 bff58270 2022-09-19 op { "finally", FINALLY },
364 9c75d045 2022-10-03 op { "for", FOR },
365 bff58270 2022-09-19 op { "if", IF },
366 bff58270 2022-09-19 op { "include", INCLUDE },
367 83e0f74d 2022-09-29 op { "printf", PRINTF },
368 bff58270 2022-09-19 op { "render", RENDER },
369 bff58270 2022-09-19 op { "tailq-foreach", TQFOREACH },
370 bff58270 2022-09-19 op { "unsafe", UNSAFE },
371 bff58270 2022-09-19 op { "urlescape", URLESCAPE },
372 3e8b34ba 2023-01-07 op { "while", WHILE },
373 bff58270 2022-09-19 op };
374 bff58270 2022-09-19 op const struct keywords *p;
375 bff58270 2022-09-19 op
376 bff58270 2022-09-19 op p = bsearch(s, keywords, nitems(keywords), sizeof(keywords[0]),
377 bff58270 2022-09-19 op kw_cmp);
378 bff58270 2022-09-19 op
379 bff58270 2022-09-19 op if (p)
380 bff58270 2022-09-19 op return (p->k_val);
381 bff58270 2022-09-19 op else
382 bff58270 2022-09-19 op return (STRING);
383 bff58270 2022-09-19 op }
384 bff58270 2022-09-19 op
385 bff58270 2022-09-19 op #define START_EXPAND 1
386 bff58270 2022-09-19 op #define DONE_EXPAND 2
387 bff58270 2022-09-19 op
388 bff58270 2022-09-19 op static int expanding;
389 bff58270 2022-09-19 op
390 bff58270 2022-09-19 op int
391 bff58270 2022-09-19 op igetc(void)
392 bff58270 2022-09-19 op {
393 bff58270 2022-09-19 op int c;
394 bff58270 2022-09-19 op
395 bff58270 2022-09-19 op while (1) {
396 bff58270 2022-09-19 op if (file->ungetpos > 0)
397 bff58270 2022-09-19 op c = file->ungetbuf[--file->ungetpos];
398 bff58270 2022-09-19 op else
399 bff58270 2022-09-19 op c = getc(file->stream);
400 bff58270 2022-09-19 op
401 bff58270 2022-09-19 op if (c == START_EXPAND)
402 bff58270 2022-09-19 op expanding = 1;
403 bff58270 2022-09-19 op else if (c == DONE_EXPAND)
404 bff58270 2022-09-19 op expanding = 0;
405 bff58270 2022-09-19 op else
406 bff58270 2022-09-19 op break;
407 bff58270 2022-09-19 op }
408 bff58270 2022-09-19 op return (c);
409 bff58270 2022-09-19 op }
410 bff58270 2022-09-19 op
411 bff58270 2022-09-19 op int
412 bff58270 2022-09-19 op lgetc(int quotec)
413 bff58270 2022-09-19 op {
414 ee34715e 2022-09-29 op int c;
415 bff58270 2022-09-19 op
416 bff58270 2022-09-19 op if (quotec) {
417 bff58270 2022-09-19 op if ((c = igetc()) == EOF) {
418 bff58270 2022-09-19 op yyerror("reached end of filewhile parsing "
419 bff58270 2022-09-19 op "quoted string");
420 bff58270 2022-09-19 op if (file == topfile || popfile() == EOF)
421 bff58270 2022-09-19 op return (EOF);
422 bff58270 2022-09-19 op return (quotec);
423 bff58270 2022-09-19 op }
424 bff58270 2022-09-19 op return (c);
425 bff58270 2022-09-19 op }
426 bff58270 2022-09-19 op
427 ee34715e 2022-09-29 op c = igetc();
428 bff58270 2022-09-19 op if (c == '\t' || c == ' ') {
429 bff58270 2022-09-19 op /* Compress blanks to a sigle space. */
430 bff58270 2022-09-19 op do {
431 bff58270 2022-09-19 op c = getc(file->stream);
432 bff58270 2022-09-19 op } while (c == '\t' || c == ' ');
433 bff58270 2022-09-19 op ungetc(c, file->stream);
434 bff58270 2022-09-19 op c = ' ';
435 bff58270 2022-09-19 op }
436 bff58270 2022-09-19 op
437 bff58270 2022-09-19 op if (c == EOF) {
438 bff58270 2022-09-19 op /*
439 bff58270 2022-09-19 op * Fake EOL when hit EOF for the first time. This gets line
440 d023b3ba 2023-03-26 op * count right if last line in included file is syntactically
441 bff58270 2022-09-19 op * invalid and has no newline.
442 bff58270 2022-09-19 op */
443 bff58270 2022-09-19 op if (file->eof_reached == 0) {
444 bff58270 2022-09-19 op file->eof_reached = 1;
445 bff58270 2022-09-19 op return ('\n');
446 bff58270 2022-09-19 op }
447 bff58270 2022-09-19 op while (c == EOF) {
448 bff58270 2022-09-19 op if (file == topfile || popfile() == EOF)
449 bff58270 2022-09-19 op return (EOF);
450 bff58270 2022-09-19 op c = igetc();
451 bff58270 2022-09-19 op }
452 bff58270 2022-09-19 op }
453 bff58270 2022-09-19 op return (c);
454 bff58270 2022-09-19 op }
455 bff58270 2022-09-19 op
456 bff58270 2022-09-19 op void
457 bff58270 2022-09-19 op lungetc(int c)
458 bff58270 2022-09-19 op {
459 bff58270 2022-09-19 op if (c == EOF)
460 bff58270 2022-09-19 op return;
461 bff58270 2022-09-19 op
462 bff58270 2022-09-19 op if (file->ungetpos >= file->ungetsize) {
463 bff58270 2022-09-19 op void *p = reallocarray(file->ungetbuf, file->ungetsize, 2);
464 bff58270 2022-09-19 op if (p == NULL)
465 bff58270 2022-09-19 op err(1, "reallocarray");
466 bff58270 2022-09-19 op file->ungetbuf = p;
467 bff58270 2022-09-19 op file->ungetsize *= 2;
468 bff58270 2022-09-19 op }
469 bff58270 2022-09-19 op file->ungetbuf[file->ungetpos++] = c;
470 bff58270 2022-09-19 op }
471 bff58270 2022-09-19 op
472 bff58270 2022-09-19 op int
473 bff58270 2022-09-19 op findeol(void)
474 bff58270 2022-09-19 op {
475 bff58270 2022-09-19 op int c;
476 bff58270 2022-09-19 op
477 bff58270 2022-09-19 op /* skip to either EOF or the first real EOL */
478 bff58270 2022-09-19 op while (1) {
479 bff58270 2022-09-19 op c = lgetc(0);
480 bff58270 2022-09-19 op if (c == '\n') {
481 bff58270 2022-09-19 op file->lineno++;
482 bff58270 2022-09-19 op break;
483 bff58270 2022-09-19 op }
484 bff58270 2022-09-19 op if (c == EOF)
485 bff58270 2022-09-19 op break;
486 bff58270 2022-09-19 op }
487 bff58270 2022-09-19 op return (ERROR);
488 bff58270 2022-09-19 op }
489 bff58270 2022-09-19 op
490 bff58270 2022-09-19 op int
491 bff58270 2022-09-19 op yylex(void)
492 bff58270 2022-09-19 op {
493 bff58270 2022-09-19 op char buf[8096];
494 bff58270 2022-09-19 op char *p = buf;
495 bff58270 2022-09-19 op int c;
496 bff58270 2022-09-19 op int token;
497 bff58270 2022-09-19 op int starting = 0;
498 bff58270 2022-09-19 op int ending = 0;
499 14d86fd9 2022-12-04 op int quote = 0;
500 bff58270 2022-09-19 op
501 bff58270 2022-09-19 op if (!in_define && block == 0) {
502 bff58270 2022-09-19 op while ((c = lgetc(0)) != '{' && c != EOF) {
503 bff58270 2022-09-19 op if (c == '\n')
504 bff58270 2022-09-19 op file->lineno++;
505 bff58270 2022-09-19 op }
506 bff58270 2022-09-19 op
507 bff58270 2022-09-19 op if (c == EOF)
508 bff58270 2022-09-19 op return (0);
509 bff58270 2022-09-19 op
510 bff58270 2022-09-19 op newblock:
511 bff58270 2022-09-19 op c = lgetc(0);
512 bff58270 2022-09-19 op if (c == '{' || c == '!') {
513 bff58270 2022-09-19 op if (c == '{')
514 bff58270 2022-09-19 op block = '}';
515 bff58270 2022-09-19 op else
516 bff58270 2022-09-19 op block = c;
517 bff58270 2022-09-19 op return (c);
518 bff58270 2022-09-19 op }
519 bff58270 2022-09-19 op if (c == '\n')
520 bff58270 2022-09-19 op file->lineno++;
521 bff58270 2022-09-19 op }
522 bff58270 2022-09-19 op
523 bff58270 2022-09-19 op while ((c = lgetc(0)) == ' ' || c == '\t' || c == '\n') {
524 bff58270 2022-09-19 op if (c == '\n')
525 bff58270 2022-09-19 op file->lineno++;
526 bff58270 2022-09-19 op }
527 bff58270 2022-09-19 op
528 bff58270 2022-09-19 op if (c == EOF) {
529 bff58270 2022-09-19 op yyerror("unterminated block");
530 bff58270 2022-09-19 op return (0);
531 bff58270 2022-09-19 op }
532 bff58270 2022-09-19 op
533 bff58270 2022-09-19 op yylval.lineno = file->lineno;
534 bff58270 2022-09-19 op
535 bff58270 2022-09-19 op if (block != 0 && c == block) {
536 bff58270 2022-09-19 op if ((c = lgetc(0)) == '}') {
537 bff58270 2022-09-19 op if (block == '!') {
538 bff58270 2022-09-19 op block = 0;
539 bff58270 2022-09-19 op return ('!');
540 bff58270 2022-09-19 op }
541 bff58270 2022-09-19 op block = 0;
542 bff58270 2022-09-19 op return ('}');
543 bff58270 2022-09-19 op }
544 bff58270 2022-09-19 op lungetc(c);
545 bff58270 2022-09-19 op c = block;
546 bff58270 2022-09-19 op }
547 bff58270 2022-09-19 op
548 bff58270 2022-09-19 op if (in_define && block == 0) {
549 bff58270 2022-09-19 op if (c == '{')
550 bff58270 2022-09-19 op goto newblock;
551 bff58270 2022-09-19 op
552 bff58270 2022-09-19 op do {
553 bff58270 2022-09-19 op if (starting) {
554 bff58270 2022-09-19 op if (c == '!' || c == '{') {
555 bff58270 2022-09-19 op lungetc(c);
556 c063deb1 2022-10-21 op lungetc('{');
557 bff58270 2022-09-19 op break;
558 bff58270 2022-09-19 op }
559 bff58270 2022-09-19 op starting = 0;
560 bff58270 2022-09-19 op lungetc(c);
561 bff58270 2022-09-19 op c = '{';
562 bff58270 2022-09-19 op } else if (c == '{') {
563 bff58270 2022-09-19 op starting = 1;
564 bff58270 2022-09-19 op continue;
565 8e6adf95 2023-04-13 op } else if (c == '\n')
566 8e6adf95 2023-04-13 op break;
567 bff58270 2022-09-19 op
568 bff58270 2022-09-19 op *p++ = c;
569 bff58270 2022-09-19 op if ((size_t)(p - buf) >= sizeof(buf)) {
570 bff58270 2022-09-19 op yyerror("string too long");
571 bff58270 2022-09-19 op return (findeol());
572 bff58270 2022-09-19 op }
573 8e6adf95 2023-04-13 op } while ((c = lgetc(0)) != EOF);
574 bff58270 2022-09-19 op *p = '\0';
575 bff58270 2022-09-19 op if (c == EOF) {
576 bff58270 2022-09-19 op yyerror("unterminated block");
577 bff58270 2022-09-19 op return (0);
578 bff58270 2022-09-19 op }
579 bff58270 2022-09-19 op if (c == '\n')
580 bff58270 2022-09-19 op file->lineno++;
581 bff58270 2022-09-19 op if ((yylval.v.string = strdup(buf)) == NULL)
582 bff58270 2022-09-19 op err(1, "strdup");
583 bff58270 2022-09-19 op return (STRING);
584 bff58270 2022-09-19 op }
585 bff58270 2022-09-19 op
586 bff58270 2022-09-19 op if (block == '!') {
587 bff58270 2022-09-19 op do {
588 bff58270 2022-09-19 op if (ending) {
589 bff58270 2022-09-19 op if (c == '}') {
590 bff58270 2022-09-19 op lungetc(c);
591 bff58270 2022-09-19 op lungetc(block);
592 bff58270 2022-09-19 op break;
593 bff58270 2022-09-19 op }
594 bff58270 2022-09-19 op ending = 0;
595 bff58270 2022-09-19 op lungetc(c);
596 bff58270 2022-09-19 op c = block;
597 bff58270 2022-09-19 op } else if (c == '!') {
598 bff58270 2022-09-19 op ending = 1;
599 bff58270 2022-09-19 op continue;
600 8e6adf95 2023-04-13 op } else if (c == '\n')
601 8e6adf95 2023-04-13 op break;
602 bff58270 2022-09-19 op
603 bff58270 2022-09-19 op *p++ = c;
604 bff58270 2022-09-19 op if ((size_t)(p - buf) >= sizeof(buf)) {
605 bff58270 2022-09-19 op yyerror("line too long");
606 bff58270 2022-09-19 op return (findeol());
607 bff58270 2022-09-19 op }
608 8e6adf95 2023-04-13 op } while ((c = lgetc(0)) != EOF);
609 bff58270 2022-09-19 op *p = '\0';
610 bff58270 2022-09-19 op
611 bff58270 2022-09-19 op if (c == EOF) {
612 bff58270 2022-09-19 op yyerror("unterminated block");
613 bff58270 2022-09-19 op return (0);
614 bff58270 2022-09-19 op }
615 bff58270 2022-09-19 op if (c == '\n')
616 bff58270 2022-09-19 op file->lineno++;
617 bff58270 2022-09-19 op
618 bff58270 2022-09-19 op if ((yylval.v.string = strdup(buf)) == NULL)
619 bff58270 2022-09-19 op err(1, "strdup");
620 bff58270 2022-09-19 op return (STRING);
621 bff58270 2022-09-19 op }
622 bff58270 2022-09-19 op
623 bff58270 2022-09-19 op if (c == '|')
624 bff58270 2022-09-19 op return (c);
625 bff58270 2022-09-19 op
626 bff58270 2022-09-19 op do {
627 14d86fd9 2022-12-04 op if (!quote && isspace((unsigned char)c))
628 14d86fd9 2022-12-04 op break;
629 14d86fd9 2022-12-04 op
630 14d86fd9 2022-12-04 op if (c == '"')
631 14d86fd9 2022-12-04 op quote = !quote;
632 14d86fd9 2022-12-04 op
633 14d86fd9 2022-12-04 op if (!quote && c == '|') {
634 bff58270 2022-09-19 op lungetc(c);
635 bff58270 2022-09-19 op break;
636 bff58270 2022-09-19 op }
637 bff58270 2022-09-19 op
638 bff58270 2022-09-19 op if (ending) {
639 bff58270 2022-09-19 op if (c == '}') {
640 bff58270 2022-09-19 op lungetc(c);
641 bff58270 2022-09-19 op lungetc('}');
642 bff58270 2022-09-19 op break;
643 bff58270 2022-09-19 op }
644 bff58270 2022-09-19 op ending = 0;
645 bff58270 2022-09-19 op lungetc(c);
646 bff58270 2022-09-19 op c = block;
647 14d86fd9 2022-12-04 op } else if (!quote && c == '}') {
648 bff58270 2022-09-19 op ending = 1;
649 bff58270 2022-09-19 op continue;
650 bff58270 2022-09-19 op }
651 bff58270 2022-09-19 op
652 bff58270 2022-09-19 op *p++ = c;
653 bff58270 2022-09-19 op if ((size_t)(p - buf) >= sizeof(buf)) {
654 bff58270 2022-09-19 op yyerror("string too long");
655 bff58270 2022-09-19 op return (findeol());
656 bff58270 2022-09-19 op }
657 14d86fd9 2022-12-04 op } while ((c = lgetc(0)) != EOF);
658 bff58270 2022-09-19 op *p = '\0';
659 bff58270 2022-09-19 op
660 bff58270 2022-09-19 op if (c == EOF) {
661 14d86fd9 2022-12-04 op yyerror(quote ? "unterminated quote" : "unterminated block");
662 bff58270 2022-09-19 op return (0);
663 bff58270 2022-09-19 op }
664 bff58270 2022-09-19 op if (c == '\n')
665 bff58270 2022-09-19 op file->lineno++;
666 bff58270 2022-09-19 op if ((token = lookup(buf)) == STRING)
667 bff58270 2022-09-19 op if ((yylval.v.string = strdup(buf)) == NULL)
668 bff58270 2022-09-19 op err(1, "strdup");
669 bff58270 2022-09-19 op return (token);
670 bff58270 2022-09-19 op }
671 bff58270 2022-09-19 op
672 bff58270 2022-09-19 op struct file *
673 bff58270 2022-09-19 op pushfile(const char *name, int secret)
674 bff58270 2022-09-19 op {
675 bff58270 2022-09-19 op struct file *nfile;
676 bff58270 2022-09-19 op
677 bff58270 2022-09-19 op if ((nfile = calloc(1, sizeof(*nfile))) == NULL)
678 bff58270 2022-09-19 op err(1, "calloc");
679 bff58270 2022-09-19 op if ((nfile->name = strdup(name)) == NULL)
680 bff58270 2022-09-19 op err(1, "strdup");
681 bff58270 2022-09-19 op if ((nfile->stream = fopen(nfile->name, "r")) == NULL) {
682 bff58270 2022-09-19 op warn("can't open %s", nfile->name);
683 bff58270 2022-09-19 op free(nfile->name);
684 bff58270 2022-09-19 op free(nfile);
685 bff58270 2022-09-19 op return (NULL);
686 bff58270 2022-09-19 op }
687 bff58270 2022-09-19 op nfile->lineno = TAILQ_EMPTY(&files) ? 1 : 0;
688 bff58270 2022-09-19 op nfile->ungetsize = 16;
689 bff58270 2022-09-19 op nfile->ungetbuf = malloc(nfile->ungetsize);
690 bff58270 2022-09-19 op if (nfile->ungetbuf == NULL)
691 bff58270 2022-09-19 op err(1, "malloc");
692 bff58270 2022-09-19 op TAILQ_INSERT_TAIL(&files, nfile, entry);
693 bff58270 2022-09-19 op return (nfile);
694 bff58270 2022-09-19 op }
695 bff58270 2022-09-19 op
696 bff58270 2022-09-19 op int
697 bff58270 2022-09-19 op popfile(void)
698 bff58270 2022-09-19 op {
699 bff58270 2022-09-19 op struct file *prev;
700 bff58270 2022-09-19 op
701 bff58270 2022-09-19 op if ((prev = TAILQ_PREV(file, files, entry)) != NULL)
702 bff58270 2022-09-19 op prev->errors += file->errors;
703 bff58270 2022-09-19 op
704 bff58270 2022-09-19 op TAILQ_REMOVE(&files, file, entry);
705 bff58270 2022-09-19 op fclose(file->stream);
706 bff58270 2022-09-19 op free(file->name);
707 bff58270 2022-09-19 op free(file->ungetbuf);
708 bff58270 2022-09-19 op free(file);
709 bff58270 2022-09-19 op file = prev;
710 bff58270 2022-09-19 op return (file ? 0 : EOF);
711 bff58270 2022-09-19 op }
712 bff58270 2022-09-19 op
713 bff58270 2022-09-19 op int
714 14c31a43 2022-11-25 op parse(FILE *outfile, const char *filename)
715 bff58270 2022-09-19 op {
716 14c31a43 2022-11-25 op fp = outfile;
717 14c31a43 2022-11-25 op
718 bff58270 2022-09-19 op if ((file = pushfile(filename, 0)) == 0)
719 bff58270 2022-09-19 op return (-1);
720 bff58270 2022-09-19 op topfile = file;
721 bff58270 2022-09-19 op
722 bff58270 2022-09-19 op yyparse();
723 bff58270 2022-09-19 op errors = file->errors;
724 bff58270 2022-09-19 op popfile();
725 bff58270 2022-09-19 op
726 bff58270 2022-09-19 op return (errors ? -1 : 0);
727 bff58270 2022-09-19 op }
728 bff58270 2022-09-19 op
729 bff58270 2022-09-19 op void
730 bff58270 2022-09-19 op dbg(void)
731 bff58270 2022-09-19 op {
732 bff58270 2022-09-19 op if (nodebug)
733 bff58270 2022-09-19 op return;
734 bff58270 2022-09-19 op
735 9a90a3d1 2022-09-29 op if (yylval.lineno == lastline + 1) {
736 9a90a3d1 2022-09-29 op lastline = yylval.lineno;
737 9a90a3d1 2022-09-29 op return;
738 9a90a3d1 2022-09-29 op }
739 9a90a3d1 2022-09-29 op lastline = yylval.lineno;
740 9a90a3d1 2022-09-29 op
741 14c31a43 2022-11-25 op fprintf(fp, "#line %d ", yylval.lineno);
742 bff58270 2022-09-19 op printq(file->name);
743 14c31a43 2022-11-25 op putc('\n', fp);
744 bff58270 2022-09-19 op }
745 bff58270 2022-09-19 op
746 bff58270 2022-09-19 op void
747 bff58270 2022-09-19 op printq(const char *str)
748 bff58270 2022-09-19 op {
749 14c31a43 2022-11-25 op putc('"', fp);
750 bff58270 2022-09-19 op for (; *str; ++str) {
751 bff58270 2022-09-19 op if (*str == '"')
752 14c31a43 2022-11-25 op putc('\\', fp);
753 14c31a43 2022-11-25 op putc(*str, fp);
754 bff58270 2022-09-19 op }
755 14c31a43 2022-11-25 op putc('"', fp);
756 bff58270 2022-09-19 op }