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 95448b62 2022-09-29 op printf("if ((tp_ret = tp->tp_puts(tp, ");
141 95448b62 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 95448b62 2022-09-29 op puts("return tp_ret;");
151 bff58270 2022-09-19 op in_define = 0;
153 bff58270 2022-09-19 op | define body finally end {
154 95448b62 2022-09-29 op puts("return 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);
165 95448b62 2022-09-29 op puts("int tp_ret = 0;");
171 bff58270 2022-09-19 op body : /* empty */
172 bff58270 2022-09-19 op | body verbatim
174 bff58270 2022-09-19 op | body special
177 bff58270 2022-09-19 op special : '{' RENDER string '}' {
179 95448b62 2022-09-29 op printf("if ((tp_ret = %s) == -1) goto err;\n", $3);
183 bff58270 2022-09-19 op | if body endif { puts("}"); }
185 bff58270 2022-09-19 op | '{' string '|' UNSAFE '}' {
187 95448b62 2022-09-29 op printf("if ((tp_ret = tp->tp_puts(tp, %s)) == -1)\n",
189 95448b62 2022-09-29 op puts("goto err;");
192 bff58270 2022-09-19 op | '{' string '|' URLESCAPE '}' {
194 95448b62 2022-09-29 op printf("if ((tp_ret = tp_urlescape(tp, %s)) == -1)\n",
196 95448b62 2022-09-29 op puts("goto err;");
199 bff58270 2022-09-19 op | '{' string '}' {
201 95448b62 2022-09-29 op printf("if ((tp_ret = tp->tp_escape(tp, %s)) == -1)\n",
203 95448b62 2022-09-29 op puts("goto err;");
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 95448b62 2022-09-29 op puts("if ((tp_ret = tp->tp_escape(tp, tp->tp_tmp)) "
215 83e0f74d 2022-09-29 op puts("goto err;");
216 83e0f74d 2022-09-29 op puts("free(tp->tp_tmp);");
217 83e0f74d 2022-09-29 op puts("tp->tp_tmp = NULL;");
221 83e0f74d 2022-09-29 op printfargs : /* empty */
222 83e0f74d 2022-09-29 op | printfargs STRING {
223 83e0f74d 2022-09-29 op printf(" %s", $2);
227 bff58270 2022-09-19 op if : '{' IF string '}' {
229 bff58270 2022-09-19 op printf("if (%s) {\n", $3);
235 bff58270 2022-09-19 op | else body end
236 bff58270 2022-09-19 op | elsif body endif
239 bff58270 2022-09-19 op elsif : '{' ELSE IF string '}' {
241 bff58270 2022-09-19 op printf("} else if (%s) {\n", $4);
246 bff58270 2022-09-19 op else : '{' ELSE '}' {
248 bff58270 2022-09-19 op puts("} else {");
252 bff58270 2022-09-19 op loop : '{' TQFOREACH STRING STRING STRING '}' {
253 bff58270 2022-09-19 op printf("TAILQ_FOREACH(%s, %s, %s) {\n",
263 bff58270 2022-09-19 op end : '{' END '}'
266 bff58270 2022-09-19 op finally : '{' FINALLY '}' {
268 83f0f95a 2022-09-29 op puts("err:");
272 bff58270 2022-09-19 op string : STRING string {
273 bff58270 2022-09-19 op if (asprintf(&$$, "%s %s", $1, $2) == -1)
274 bff58270 2022-09-19 op err(1, "asprintf");
283 bff58270 2022-09-19 op struct keywords {
284 bff58270 2022-09-19 op const char *k_name;
289 bff58270 2022-09-19 op yyerror(const char *fmt, ...)
294 bff58270 2022-09-19 op file->errors++;
295 bff58270 2022-09-19 op va_start(ap, fmt);
296 bff58270 2022-09-19 op if (vasprintf(&msg, fmt, ap) == -1)
297 bff58270 2022-09-19 op err(1, "yyerror vasprintf");
299 bff58270 2022-09-19 op fprintf(stderr, "%s:%d: %s\n", file->name, yylval.lineno, msg);
305 bff58270 2022-09-19 op kw_cmp(const void *k, const void *e)
307 bff58270 2022-09-19 op return (strcmp(k, ((const struct keywords *)e)->k_name));
311 bff58270 2022-09-19 op lookup(char *s)
313 bff58270 2022-09-19 op /* this has to be sorted always */
314 bff58270 2022-09-19 op static const struct keywords keywords[] = {
315 bff58270 2022-09-19 op { "define", DEFINE },
316 bff58270 2022-09-19 op { "else", ELSE },
317 bff58270 2022-09-19 op { "end", END },
318 bff58270 2022-09-19 op { "finally", FINALLY },
319 bff58270 2022-09-19 op { "if", IF },
320 bff58270 2022-09-19 op { "include", INCLUDE },
321 83e0f74d 2022-09-29 op { "printf", PRINTF },
322 bff58270 2022-09-19 op { "render", RENDER },
323 bff58270 2022-09-19 op { "tailq-foreach", TQFOREACH },
324 bff58270 2022-09-19 op { "unsafe", UNSAFE },
325 bff58270 2022-09-19 op { "urlescape", URLESCAPE },
327 bff58270 2022-09-19 op const struct keywords *p;
329 bff58270 2022-09-19 op p = bsearch(s, keywords, nitems(keywords), sizeof(keywords[0]),
333 bff58270 2022-09-19 op return (p->k_val);
335 bff58270 2022-09-19 op return (STRING);
338 bff58270 2022-09-19 op #define START_EXPAND 1
339 bff58270 2022-09-19 op #define DONE_EXPAND 2
341 bff58270 2022-09-19 op static int expanding;
349 bff58270 2022-09-19 op if (file->ungetpos > 0)
350 bff58270 2022-09-19 op c = file->ungetbuf[--file->ungetpos];
352 bff58270 2022-09-19 op c = getc(file->stream);
354 bff58270 2022-09-19 op if (c == START_EXPAND)
355 bff58270 2022-09-19 op expanding = 1;
356 bff58270 2022-09-19 op else if (c == DONE_EXPAND)
357 bff58270 2022-09-19 op expanding = 0;
365 bff58270 2022-09-19 op lgetc(int quotec)
369 bff58270 2022-09-19 op if (quotec) {
370 bff58270 2022-09-19 op if ((c = igetc()) == EOF) {
371 bff58270 2022-09-19 op yyerror("reached end of filewhile parsing "
372 bff58270 2022-09-19 op "quoted string");
373 bff58270 2022-09-19 op if (file == topfile || popfile() == EOF)
374 bff58270 2022-09-19 op return (EOF);
375 bff58270 2022-09-19 op return (quotec);
381 bff58270 2022-09-19 op if (c == '\t' || c == ' ') {
382 bff58270 2022-09-19 op /* Compress blanks to a sigle space. */
384 bff58270 2022-09-19 op c = getc(file->stream);
385 bff58270 2022-09-19 op } while (c == '\t' || c == ' ');
386 bff58270 2022-09-19 op ungetc(c, file->stream);
390 bff58270 2022-09-19 op if (c == EOF) {
392 bff58270 2022-09-19 op * Fake EOL when hit EOF for the first time. This gets line
393 bff58270 2022-09-19 op * count rigchtif last line included file is syntactically
394 bff58270 2022-09-19 op * invalid and has no newline.
396 bff58270 2022-09-19 op if (file->eof_reached == 0) {
397 bff58270 2022-09-19 op file->eof_reached = 1;
398 bff58270 2022-09-19 op return ('\n');
400 bff58270 2022-09-19 op while (c == EOF) {
401 bff58270 2022-09-19 op if (file == topfile || popfile() == EOF)
402 bff58270 2022-09-19 op return (EOF);
410 bff58270 2022-09-19 op lungetc(int c)
412 bff58270 2022-09-19 op if (c == EOF)
415 bff58270 2022-09-19 op if (file->ungetpos >= file->ungetsize) {
416 bff58270 2022-09-19 op void *p = reallocarray(file->ungetbuf, file->ungetsize, 2);
417 bff58270 2022-09-19 op if (p == NULL)
418 bff58270 2022-09-19 op err(1, "reallocarray");
419 bff58270 2022-09-19 op file->ungetbuf = p;
420 bff58270 2022-09-19 op file->ungetsize *= 2;
422 bff58270 2022-09-19 op file->ungetbuf[file->ungetpos++] = c;
426 bff58270 2022-09-19 op findeol(void)
430 bff58270 2022-09-19 op /* skip to either EOF or the first real EOL */
432 bff58270 2022-09-19 op c = lgetc(0);
433 bff58270 2022-09-19 op if (c == '\n') {
434 bff58270 2022-09-19 op file->lineno++;
437 bff58270 2022-09-19 op if (c == EOF)
440 bff58270 2022-09-19 op return (ERROR);
446 bff58270 2022-09-19 op char buf[8096];
447 bff58270 2022-09-19 op char *p = buf;
450 bff58270 2022-09-19 op int starting = 0;
451 bff58270 2022-09-19 op int ending = 0;
453 bff58270 2022-09-19 op if (!in_define && block == 0) {
454 bff58270 2022-09-19 op while ((c = lgetc(0)) != '{' && c != EOF) {
455 bff58270 2022-09-19 op if (c == '\n')
456 bff58270 2022-09-19 op file->lineno++;
459 bff58270 2022-09-19 op if (c == EOF)
463 bff58270 2022-09-19 op c = lgetc(0);
464 bff58270 2022-09-19 op if (c == '{' || c == '!') {
465 bff58270 2022-09-19 op if (c == '{')
471 bff58270 2022-09-19 op if (c == '\n')
472 bff58270 2022-09-19 op file->lineno++;
475 bff58270 2022-09-19 op while ((c = lgetc(0)) == ' ' || c == '\t' || c == '\n') {
476 bff58270 2022-09-19 op if (c == '\n')
477 bff58270 2022-09-19 op file->lineno++;
480 bff58270 2022-09-19 op if (c == EOF) {
481 bff58270 2022-09-19 op yyerror("unterminated block");
485 bff58270 2022-09-19 op yylval.lineno = file->lineno;
487 bff58270 2022-09-19 op if (block != 0 && c == block) {
488 bff58270 2022-09-19 op if ((c = lgetc(0)) == '}') {
489 bff58270 2022-09-19 op if (block == '!') {
491 bff58270 2022-09-19 op return ('!');
494 bff58270 2022-09-19 op return ('}');
500 bff58270 2022-09-19 op if (in_define && block == 0) {
501 bff58270 2022-09-19 op if (c == '{')
502 bff58270 2022-09-19 op goto newblock;
505 bff58270 2022-09-19 op if (starting) {
506 bff58270 2022-09-19 op if (c == '!' || c == '{') {
507 bff58270 2022-09-19 op lungetc('{');
511 bff58270 2022-09-19 op starting = 0;
514 bff58270 2022-09-19 op } else if (c == '{') {
515 bff58270 2022-09-19 op starting = 1;
520 bff58270 2022-09-19 op if ((size_t)(p - buf) >= sizeof(buf)) {
521 bff58270 2022-09-19 op yyerror("string too long");
522 bff58270 2022-09-19 op return (findeol());
524 bff58270 2022-09-19 op } while ((c = lgetc(0)) != EOF && c != '\n');
526 bff58270 2022-09-19 op if (c == EOF) {
527 bff58270 2022-09-19 op yyerror("unterminated block");
530 bff58270 2022-09-19 op if (c == '\n')
531 bff58270 2022-09-19 op file->lineno++;
532 bff58270 2022-09-19 op if ((yylval.v.string = strdup(buf)) == NULL)
533 bff58270 2022-09-19 op err(1, "strdup");
534 bff58270 2022-09-19 op return (STRING);
537 bff58270 2022-09-19 op if (block == '!') {
539 bff58270 2022-09-19 op if (ending) {
540 bff58270 2022-09-19 op if (c == '}') {
542 bff58270 2022-09-19 op lungetc(block);
548 bff58270 2022-09-19 op } else if (c == '!') {
554 bff58270 2022-09-19 op if ((size_t)(p - buf) >= sizeof(buf)) {
555 bff58270 2022-09-19 op yyerror("line too long");
556 bff58270 2022-09-19 op return (findeol());
558 bff58270 2022-09-19 op } while ((c = lgetc(0)) != EOF && c != '\n');
561 bff58270 2022-09-19 op if (c == EOF) {
562 bff58270 2022-09-19 op yyerror("unterminated block");
565 bff58270 2022-09-19 op if (c == '\n')
566 bff58270 2022-09-19 op file->lineno++;
568 bff58270 2022-09-19 op if ((yylval.v.string = strdup(buf)) == NULL)
569 bff58270 2022-09-19 op err(1, "strdup");
570 bff58270 2022-09-19 op return (STRING);
573 bff58270 2022-09-19 op if (c == '|')
577 bff58270 2022-09-19 op if (c == '|') {
582 bff58270 2022-09-19 op if (ending) {
583 bff58270 2022-09-19 op if (c == '}') {
585 bff58270 2022-09-19 op lungetc('}');
591 bff58270 2022-09-19 op } else if (c == '}') {
597 bff58270 2022-09-19 op if ((size_t)(p - buf) >= sizeof(buf)) {
598 bff58270 2022-09-19 op yyerror("string too long");
599 bff58270 2022-09-19 op return (findeol());
601 bff58270 2022-09-19 op } while ((c = lgetc(0)) != EOF && !isspace((unsigned char)c));
604 bff58270 2022-09-19 op if (c == EOF) {
605 bff58270 2022-09-19 op yyerror("unterminated block");
608 bff58270 2022-09-19 op if (c == '\n')
609 bff58270 2022-09-19 op file->lineno++;
610 bff58270 2022-09-19 op if ((token = lookup(buf)) == STRING)
611 bff58270 2022-09-19 op if ((yylval.v.string = strdup(buf)) == NULL)
612 bff58270 2022-09-19 op err(1, "strdup");
613 bff58270 2022-09-19 op return (token);
616 bff58270 2022-09-19 op struct file *
617 bff58270 2022-09-19 op pushfile(const char *name, int secret)
619 bff58270 2022-09-19 op struct file *nfile;
621 bff58270 2022-09-19 op if ((nfile = calloc(1, sizeof(*nfile))) == NULL)
622 bff58270 2022-09-19 op err(1, "calloc");
623 bff58270 2022-09-19 op if ((nfile->name = strdup(name)) == NULL)
624 bff58270 2022-09-19 op err(1, "strdup");
625 bff58270 2022-09-19 op if ((nfile->stream = fopen(nfile->name, "r")) == NULL) {
626 bff58270 2022-09-19 op warn("can't open %s", nfile->name);
627 bff58270 2022-09-19 op free(nfile->name);
629 bff58270 2022-09-19 op return (NULL);
631 bff58270 2022-09-19 op nfile->lineno = TAILQ_EMPTY(&files) ? 1 : 0;
632 bff58270 2022-09-19 op nfile->ungetsize = 16;
633 bff58270 2022-09-19 op nfile->ungetbuf = malloc(nfile->ungetsize);
634 bff58270 2022-09-19 op if (nfile->ungetbuf == NULL)
635 bff58270 2022-09-19 op err(1, "malloc");
636 bff58270 2022-09-19 op TAILQ_INSERT_TAIL(&files, nfile, entry);
637 bff58270 2022-09-19 op return (nfile);
641 bff58270 2022-09-19 op popfile(void)
643 bff58270 2022-09-19 op struct file *prev;
645 bff58270 2022-09-19 op if ((prev = TAILQ_PREV(file, files, entry)) != NULL)
646 bff58270 2022-09-19 op prev->errors += file->errors;
648 bff58270 2022-09-19 op TAILQ_REMOVE(&files, file, entry);
649 bff58270 2022-09-19 op fclose(file->stream);
650 bff58270 2022-09-19 op free(file->name);
651 bff58270 2022-09-19 op free(file->ungetbuf);
654 bff58270 2022-09-19 op return (file ? 0 : EOF);
658 bff58270 2022-09-19 op parse(const char *filename)
660 bff58270 2022-09-19 op if ((file = pushfile(filename, 0)) == 0)
662 bff58270 2022-09-19 op topfile = file;
665 bff58270 2022-09-19 op errors = file->errors;
668 bff58270 2022-09-19 op return (errors ? -1 : 0);
677 9a90a3d1 2022-09-29 op if (yylval.lineno == lastline + 1) {
678 9a90a3d1 2022-09-29 op lastline = yylval.lineno;
681 9a90a3d1 2022-09-29 op lastline = yylval.lineno;
683 bff58270 2022-09-19 op printf("#line %d ", yylval.lineno);
684 bff58270 2022-09-19 op printq(file->name);
685 bff58270 2022-09-19 op putchar('\n');
689 bff58270 2022-09-19 op printq(const char *str)
691 bff58270 2022-09-19 op putchar('"');
692 bff58270 2022-09-19 op for (; *str; ++str) {
693 bff58270 2022-09-19 op if (*str == '"')
694 bff58270 2022-09-19 op putchar('\\');
695 bff58270 2022-09-19 op putchar(*str);
697 bff58270 2022-09-19 op putchar('"');