2 bff58270 2022-09-19 op * Copyright (c) 2022 Omar Polo <op@omarpolo.com>
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.
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.
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.
26 bff58270 2022-09-19 op #include <sys/queue.h>
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>
37 bff58270 2022-09-19 op #ifndef nitems
38 bff58270 2022-09-19 op #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
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;
46 bff58270 2022-09-19 op size_t ungetpos;
47 bff58270 2022-09-19 op size_t ungetsize;
48 bff58270 2022-09-19 op u_char *ungetbuf;
49 bff58270 2022-09-19 op int eof_reached;
52 bff58270 2022-09-19 op } *file, *topfile;
53 bff58270 2022-09-19 op int parse(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);
68 bff58270 2022-09-19 op void dbg(void);
69 bff58270 2022-09-19 op void printq(const char *);
71 bff58270 2022-09-19 op extern int nodebug;
73 bff58270 2022-09-19 op static int block;
74 bff58270 2022-09-19 op static int in_define;
75 bff58270 2022-09-19 op static int errors;
76 9a90a3d1 2022-09-29 op static int lastline = -1;
78 bff58270 2022-09-19 op typedef struct {
87 83e0f74d 2022-09-29 op %token DEFINE ELSE END ERROR FINALLY IF INCLUDE PRINTF
88 bff58270 2022-09-19 op %token RENDER TQFOREACH UNSAFE URLESCAPE
89 bff58270 2022-09-19 op %token <v.string> STRING
90 bff58270 2022-09-19 op %type <v.string> string
94 bff58270 2022-09-19 op grammar : /* empty */
95 bff58270 2022-09-19 op | grammar include
96 bff58270 2022-09-19 op | grammar verbatim
97 bff58270 2022-09-19 op | grammar block
98 bff58270 2022-09-19 op | grammar error { file->errors++; }
101 bff58270 2022-09-19 op include : INCLUDE STRING {
102 bff58270 2022-09-19 op struct file *nfile;
104 bff58270 2022-09-19 op if ((nfile = pushfile($2, 0)) == NULL) {
105 bff58270 2022-09-19 op yyerror("failed to include file %s", $2);
111 bff58270 2022-09-19 op file = nfile;
112 bff58270 2022-09-19 op lungetc('\n');
116 bff58270 2022-09-19 op verbatim : '!' verbatim1 '!' {
117 bff58270 2022-09-19 op if (in_define) {
118 bff58270 2022-09-19 op /* TODO: check template status and exit in case */
123 bff58270 2022-09-19 op verbatim1 : /* empty */
124 bff58270 2022-09-19 op | verbatim1 STRING {
125 bff58270 2022-09-19 op if (*$2 != '\0') {
133 bff58270 2022-09-19 op verbatims : /* empty */
134 bff58270 2022-09-19 op | verbatims verbatim
137 bff58270 2022-09-19 op raw : STRING {
139 bff58270 2022-09-19 op printf("if (tp->tp_puts(tp, ");
141 83f0f95a 2022-09-29 op printf(") == -1) goto err;\n");
147 bff58270 2022-09-19 op block : define body end {
148 83f0f95a 2022-09-29 op printf("err:\n");
149 bff58270 2022-09-19 op puts("return tp->tp_ret;");
151 bff58270 2022-09-19 op in_define = 0;
153 bff58270 2022-09-19 op | define body finally end {
154 bff58270 2022-09-19 op puts("return tp->tp_ret;");
156 bff58270 2022-09-19 op in_define = 0;
160 bff58270 2022-09-19 op define : '{' DEFINE string '}' {
161 bff58270 2022-09-19 op in_define = 1;
164 bff58270 2022-09-19 op printf("int\n%s\n{\n", $3);
170 bff58270 2022-09-19 op body : /* empty */
171 bff58270 2022-09-19 op | body verbatim
173 bff58270 2022-09-19 op | body special
176 bff58270 2022-09-19 op special : '{' RENDER string '}' {
178 bff58270 2022-09-19 op if (strrchr($3, ')') != NULL)
179 83f0f95a 2022-09-29 op printf("if (%s == -1) goto err;\n", $3);
181 bff58270 2022-09-19 op printf("if (%s != NULL && %s(tp) == -1) "
182 83f0f95a 2022-09-29 op "goto err;\n", $3, $3);
186 bff58270 2022-09-19 op | if body endif { puts("}"); }
188 bff58270 2022-09-19 op | '{' string '|' UNSAFE '}' {
190 83f0f95a 2022-09-29 op printf("if (tp->tp_puts(tp, %s) == -1) goto err;\n",
194 bff58270 2022-09-19 op | '{' string '|' URLESCAPE '}' {
196 83f0f95a 2022-09-29 op printf("if (tp_urlescape(tp, %s) == -1) goto err;\n",
200 bff58270 2022-09-19 op | '{' string '}' {
202 83f0f95a 2022-09-29 op printf("if (tp->tp_escape(tp, %s) == -1) goto err;\n",
208 83e0f74d 2022-09-29 op printf : '{' PRINTF {
210 83e0f74d 2022-09-29 op printf("if (asprintf(&tp->tp_tmp, ");
211 83e0f74d 2022-09-29 op } printfargs '}' {
212 83e0f74d 2022-09-29 op printf(") == -1)\n goto err;\n");
213 83e0f74d 2022-09-29 op puts("if (tp->tp_escape(tp, tp->tp_tmp) == -1)");
214 83e0f74d 2022-09-29 op puts("goto err;");
215 83e0f74d 2022-09-29 op puts("free(tp->tp_tmp);");
216 83e0f74d 2022-09-29 op puts("tp->tp_tmp = NULL;");
220 83e0f74d 2022-09-29 op printfargs : /* empty */
221 83e0f74d 2022-09-29 op | printfargs STRING {
222 83e0f74d 2022-09-29 op printf(" %s", $2);
226 bff58270 2022-09-19 op if : '{' IF string '}' {
228 bff58270 2022-09-19 op printf("if (%s) {\n", $3);
234 bff58270 2022-09-19 op | else body end
235 bff58270 2022-09-19 op | elsif body endif
238 bff58270 2022-09-19 op elsif : '{' ELSE IF string '}' {
240 bff58270 2022-09-19 op printf("} else if (%s) {\n", $4);
245 bff58270 2022-09-19 op else : '{' ELSE '}' {
247 bff58270 2022-09-19 op puts("} else {");
251 bff58270 2022-09-19 op loop : '{' TQFOREACH STRING STRING STRING '}' {
252 bff58270 2022-09-19 op printf("TAILQ_FOREACH(%s, %s, %s) {\n",
262 bff58270 2022-09-19 op end : '{' END '}'
265 bff58270 2022-09-19 op finally : '{' FINALLY '}' {
267 83f0f95a 2022-09-29 op puts("err:");
271 bff58270 2022-09-19 op string : STRING string {
272 bff58270 2022-09-19 op if (asprintf(&$$, "%s %s", $1, $2) == -1)
273 bff58270 2022-09-19 op err(1, "asprintf");
282 bff58270 2022-09-19 op struct keywords {
283 bff58270 2022-09-19 op const char *k_name;
288 bff58270 2022-09-19 op yyerror(const char *fmt, ...)
293 bff58270 2022-09-19 op file->errors++;
294 bff58270 2022-09-19 op va_start(ap, fmt);
295 bff58270 2022-09-19 op if (vasprintf(&msg, fmt, ap) == -1)
296 bff58270 2022-09-19 op err(1, "yyerror vasprintf");
298 bff58270 2022-09-19 op fprintf(stderr, "%s:%d: %s\n", file->name, yylval.lineno, msg);
304 bff58270 2022-09-19 op kw_cmp(const void *k, const void *e)
306 bff58270 2022-09-19 op return (strcmp(k, ((const struct keywords *)e)->k_name));
310 bff58270 2022-09-19 op lookup(char *s)
312 bff58270 2022-09-19 op /* this has to be sorted always */
313 bff58270 2022-09-19 op static const struct keywords keywords[] = {
314 bff58270 2022-09-19 op { "define", DEFINE },
315 bff58270 2022-09-19 op { "else", ELSE },
316 bff58270 2022-09-19 op { "end", END },
317 bff58270 2022-09-19 op { "finally", FINALLY },
318 bff58270 2022-09-19 op { "if", IF },
319 bff58270 2022-09-19 op { "include", INCLUDE },
320 83e0f74d 2022-09-29 op { "printf", PRINTF },
321 bff58270 2022-09-19 op { "render", RENDER },
322 bff58270 2022-09-19 op { "tailq-foreach", TQFOREACH },
323 bff58270 2022-09-19 op { "unsafe", UNSAFE },
324 bff58270 2022-09-19 op { "urlescape", URLESCAPE },
326 bff58270 2022-09-19 op const struct keywords *p;
328 bff58270 2022-09-19 op p = bsearch(s, keywords, nitems(keywords), sizeof(keywords[0]),
332 bff58270 2022-09-19 op return (p->k_val);
334 bff58270 2022-09-19 op return (STRING);
337 bff58270 2022-09-19 op #define START_EXPAND 1
338 bff58270 2022-09-19 op #define DONE_EXPAND 2
340 bff58270 2022-09-19 op static int expanding;
348 bff58270 2022-09-19 op if (file->ungetpos > 0)
349 bff58270 2022-09-19 op c = file->ungetbuf[--file->ungetpos];
351 bff58270 2022-09-19 op c = getc(file->stream);
353 bff58270 2022-09-19 op if (c == START_EXPAND)
354 bff58270 2022-09-19 op expanding = 1;
355 bff58270 2022-09-19 op else if (c == DONE_EXPAND)
356 bff58270 2022-09-19 op expanding = 0;
364 bff58270 2022-09-19 op lgetc(int quotec)
368 bff58270 2022-09-19 op if (quotec) {
369 bff58270 2022-09-19 op if ((c = igetc()) == EOF) {
370 bff58270 2022-09-19 op yyerror("reached end of filewhile parsing "
371 bff58270 2022-09-19 op "quoted string");
372 bff58270 2022-09-19 op if (file == topfile || popfile() == EOF)
373 bff58270 2022-09-19 op return (EOF);
374 bff58270 2022-09-19 op return (quotec);
380 bff58270 2022-09-19 op if (c == '\t' || c == ' ') {
381 bff58270 2022-09-19 op /* Compress blanks to a sigle space. */
383 bff58270 2022-09-19 op c = getc(file->stream);
384 bff58270 2022-09-19 op } while (c == '\t' || c == ' ');
385 bff58270 2022-09-19 op ungetc(c, file->stream);
389 bff58270 2022-09-19 op if (c == EOF) {
391 bff58270 2022-09-19 op * Fake EOL when hit EOF for the first time. This gets line
392 bff58270 2022-09-19 op * count rigchtif last line included file is syntactically
393 bff58270 2022-09-19 op * invalid and has no newline.
395 bff58270 2022-09-19 op if (file->eof_reached == 0) {
396 bff58270 2022-09-19 op file->eof_reached = 1;
397 bff58270 2022-09-19 op return ('\n');
399 bff58270 2022-09-19 op while (c == EOF) {
400 bff58270 2022-09-19 op if (file == topfile || popfile() == EOF)
401 bff58270 2022-09-19 op return (EOF);
409 bff58270 2022-09-19 op lungetc(int c)
411 bff58270 2022-09-19 op if (c == EOF)
414 bff58270 2022-09-19 op if (file->ungetpos >= file->ungetsize) {
415 bff58270 2022-09-19 op void *p = reallocarray(file->ungetbuf, file->ungetsize, 2);
416 bff58270 2022-09-19 op if (p == NULL)
417 bff58270 2022-09-19 op err(1, "reallocarray");
418 bff58270 2022-09-19 op file->ungetbuf = p;
419 bff58270 2022-09-19 op file->ungetsize *= 2;
421 bff58270 2022-09-19 op file->ungetbuf[file->ungetpos++] = c;
425 bff58270 2022-09-19 op findeol(void)
429 bff58270 2022-09-19 op /* skip to either EOF or the first real EOL */
431 bff58270 2022-09-19 op c = lgetc(0);
432 bff58270 2022-09-19 op if (c == '\n') {
433 bff58270 2022-09-19 op file->lineno++;
436 bff58270 2022-09-19 op if (c == EOF)
439 bff58270 2022-09-19 op return (ERROR);
445 bff58270 2022-09-19 op char buf[8096];
446 bff58270 2022-09-19 op char *p = buf;
449 bff58270 2022-09-19 op int starting = 0;
450 bff58270 2022-09-19 op int ending = 0;
452 bff58270 2022-09-19 op if (!in_define && block == 0) {
453 bff58270 2022-09-19 op while ((c = lgetc(0)) != '{' && c != EOF) {
454 bff58270 2022-09-19 op if (c == '\n')
455 bff58270 2022-09-19 op file->lineno++;
458 bff58270 2022-09-19 op if (c == EOF)
462 bff58270 2022-09-19 op c = lgetc(0);
463 bff58270 2022-09-19 op if (c == '{' || c == '!') {
464 bff58270 2022-09-19 op if (c == '{')
470 bff58270 2022-09-19 op if (c == '\n')
471 bff58270 2022-09-19 op file->lineno++;
474 bff58270 2022-09-19 op while ((c = lgetc(0)) == ' ' || c == '\t' || c == '\n') {
475 bff58270 2022-09-19 op if (c == '\n')
476 bff58270 2022-09-19 op file->lineno++;
479 bff58270 2022-09-19 op if (c == EOF) {
480 bff58270 2022-09-19 op yyerror("unterminated block");
484 bff58270 2022-09-19 op yylval.lineno = file->lineno;
486 bff58270 2022-09-19 op if (block != 0 && c == block) {
487 bff58270 2022-09-19 op if ((c = lgetc(0)) == '}') {
488 bff58270 2022-09-19 op if (block == '!') {
490 bff58270 2022-09-19 op return ('!');
493 bff58270 2022-09-19 op return ('}');
499 bff58270 2022-09-19 op if (in_define && block == 0) {
500 bff58270 2022-09-19 op if (c == '{')
501 bff58270 2022-09-19 op goto newblock;
504 bff58270 2022-09-19 op if (starting) {
505 bff58270 2022-09-19 op if (c == '!' || c == '{') {
506 bff58270 2022-09-19 op lungetc('{');
510 bff58270 2022-09-19 op starting = 0;
513 bff58270 2022-09-19 op } else if (c == '{') {
514 bff58270 2022-09-19 op starting = 1;
519 bff58270 2022-09-19 op if ((size_t)(p - buf) >= sizeof(buf)) {
520 bff58270 2022-09-19 op yyerror("string too long");
521 bff58270 2022-09-19 op return (findeol());
523 bff58270 2022-09-19 op } while ((c = lgetc(0)) != EOF && c != '\n');
525 bff58270 2022-09-19 op if (c == EOF) {
526 bff58270 2022-09-19 op yyerror("unterminated block");
529 bff58270 2022-09-19 op if (c == '\n')
530 bff58270 2022-09-19 op file->lineno++;
531 bff58270 2022-09-19 op if ((yylval.v.string = strdup(buf)) == NULL)
532 bff58270 2022-09-19 op err(1, "strdup");
533 bff58270 2022-09-19 op return (STRING);
536 bff58270 2022-09-19 op if (block == '!') {
538 bff58270 2022-09-19 op if (ending) {
539 bff58270 2022-09-19 op if (c == '}') {
541 bff58270 2022-09-19 op lungetc(block);
547 bff58270 2022-09-19 op } else if (c == '!') {
553 bff58270 2022-09-19 op if ((size_t)(p - buf) >= sizeof(buf)) {
554 bff58270 2022-09-19 op yyerror("line too long");
555 bff58270 2022-09-19 op return (findeol());
557 bff58270 2022-09-19 op } while ((c = lgetc(0)) != EOF && c != '\n');
560 bff58270 2022-09-19 op if (c == EOF) {
561 bff58270 2022-09-19 op yyerror("unterminated block");
564 bff58270 2022-09-19 op if (c == '\n')
565 bff58270 2022-09-19 op file->lineno++;
567 bff58270 2022-09-19 op if ((yylval.v.string = strdup(buf)) == NULL)
568 bff58270 2022-09-19 op err(1, "strdup");
569 bff58270 2022-09-19 op return (STRING);
572 bff58270 2022-09-19 op if (c == '|')
576 bff58270 2022-09-19 op if (c == '|') {
581 bff58270 2022-09-19 op if (ending) {
582 bff58270 2022-09-19 op if (c == '}') {
584 bff58270 2022-09-19 op lungetc('}');
590 bff58270 2022-09-19 op } else if (c == '}') {
596 bff58270 2022-09-19 op if ((size_t)(p - buf) >= sizeof(buf)) {
597 bff58270 2022-09-19 op yyerror("string too long");
598 bff58270 2022-09-19 op return (findeol());
600 bff58270 2022-09-19 op } while ((c = lgetc(0)) != EOF && !isspace((unsigned char)c));
603 bff58270 2022-09-19 op if (c == EOF) {
604 bff58270 2022-09-19 op yyerror("unterminated block");
607 bff58270 2022-09-19 op if (c == '\n')
608 bff58270 2022-09-19 op file->lineno++;
609 bff58270 2022-09-19 op if ((token = lookup(buf)) == STRING)
610 bff58270 2022-09-19 op if ((yylval.v.string = strdup(buf)) == NULL)
611 bff58270 2022-09-19 op err(1, "strdup");
612 bff58270 2022-09-19 op return (token);
615 bff58270 2022-09-19 op struct file *
616 bff58270 2022-09-19 op pushfile(const char *name, int secret)
618 bff58270 2022-09-19 op struct file *nfile;
620 bff58270 2022-09-19 op if ((nfile = calloc(1, sizeof(*nfile))) == NULL)
621 bff58270 2022-09-19 op err(1, "calloc");
622 bff58270 2022-09-19 op if ((nfile->name = strdup(name)) == NULL)
623 bff58270 2022-09-19 op err(1, "strdup");
624 bff58270 2022-09-19 op if ((nfile->stream = fopen(nfile->name, "r")) == NULL) {
625 bff58270 2022-09-19 op warn("can't open %s", nfile->name);
626 bff58270 2022-09-19 op free(nfile->name);
628 bff58270 2022-09-19 op return (NULL);
630 bff58270 2022-09-19 op nfile->lineno = TAILQ_EMPTY(&files) ? 1 : 0;
631 bff58270 2022-09-19 op nfile->ungetsize = 16;
632 bff58270 2022-09-19 op nfile->ungetbuf = malloc(nfile->ungetsize);
633 bff58270 2022-09-19 op if (nfile->ungetbuf == NULL)
634 bff58270 2022-09-19 op err(1, "malloc");
635 bff58270 2022-09-19 op TAILQ_INSERT_TAIL(&files, nfile, entry);
636 bff58270 2022-09-19 op return (nfile);
640 bff58270 2022-09-19 op popfile(void)
642 bff58270 2022-09-19 op struct file *prev;
644 bff58270 2022-09-19 op if ((prev = TAILQ_PREV(file, files, entry)) != NULL)
645 bff58270 2022-09-19 op prev->errors += file->errors;
647 bff58270 2022-09-19 op TAILQ_REMOVE(&files, file, entry);
648 bff58270 2022-09-19 op fclose(file->stream);
649 bff58270 2022-09-19 op free(file->name);
650 bff58270 2022-09-19 op free(file->ungetbuf);
653 bff58270 2022-09-19 op return (file ? 0 : EOF);
657 bff58270 2022-09-19 op parse(const char *filename)
659 bff58270 2022-09-19 op if ((file = pushfile(filename, 0)) == 0)
661 bff58270 2022-09-19 op topfile = file;
664 bff58270 2022-09-19 op errors = file->errors;
667 bff58270 2022-09-19 op return (errors ? -1 : 0);
676 9a90a3d1 2022-09-29 op if (yylval.lineno == lastline + 1) {
677 9a90a3d1 2022-09-29 op lastline = yylval.lineno;
680 9a90a3d1 2022-09-29 op lastline = yylval.lineno;
682 bff58270 2022-09-19 op printf("#line %d ", yylval.lineno);
683 bff58270 2022-09-19 op printq(file->name);
684 bff58270 2022-09-19 op putchar('\n');
688 bff58270 2022-09-19 op printq(const char *str)
690 bff58270 2022-09-19 op putchar('"');
691 bff58270 2022-09-19 op for (; *str; ++str) {
692 bff58270 2022-09-19 op if (*str == '"')
693 bff58270 2022-09-19 op putchar('\\');
694 bff58270 2022-09-19 op putchar(*str);
696 bff58270 2022-09-19 op putchar('"');