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;
77 bff58270 2022-09-19 op typedef struct {
86 bff58270 2022-09-19 op %token DEFINE ELSE END ERROR ESCAPE FINALLY IF INCLUDE
87 bff58270 2022-09-19 op %token RENDER TQFOREACH UNSAFE URLESCAPE
88 bff58270 2022-09-19 op %token <v.string> STRING
89 bff58270 2022-09-19 op %type <v.string> string
93 bff58270 2022-09-19 op grammar : /* empty */
94 bff58270 2022-09-19 op | grammar include
95 bff58270 2022-09-19 op | grammar verbatim
96 bff58270 2022-09-19 op | grammar block
97 bff58270 2022-09-19 op | grammar error { file->errors++; }
100 bff58270 2022-09-19 op include : INCLUDE STRING {
101 bff58270 2022-09-19 op struct file *nfile;
103 bff58270 2022-09-19 op if ((nfile = pushfile($2, 0)) == NULL) {
104 bff58270 2022-09-19 op yyerror("failed to include file %s", $2);
110 bff58270 2022-09-19 op file = nfile;
111 bff58270 2022-09-19 op lungetc('\n');
115 bff58270 2022-09-19 op verbatim : '!' verbatim1 '!' {
116 bff58270 2022-09-19 op if (in_define) {
117 bff58270 2022-09-19 op /* TODO: check template status and exit in case */
122 bff58270 2022-09-19 op verbatim1 : /* empty */
123 bff58270 2022-09-19 op | verbatim1 STRING {
124 bff58270 2022-09-19 op if (*$2 != '\0') {
132 bff58270 2022-09-19 op verbatims : /* empty */
133 bff58270 2022-09-19 op | verbatims verbatim
136 bff58270 2022-09-19 op raw : STRING {
138 bff58270 2022-09-19 op printf("if (tp->tp_puts(tp, ");
140 c423c56e 2022-09-25 op printf(") == -1) goto err;\n");
146 bff58270 2022-09-19 op block : define body end {
147 c423c56e 2022-09-25 op printf("err:\n");
148 bff58270 2022-09-19 op puts("return tp->tp_ret;");
150 bff58270 2022-09-19 op in_define = 0;
152 bff58270 2022-09-19 op | define body finally end {
153 bff58270 2022-09-19 op puts("return tp->tp_ret;");
155 bff58270 2022-09-19 op in_define = 0;
159 bff58270 2022-09-19 op define : '{' DEFINE string '}' {
160 bff58270 2022-09-19 op in_define = 1;
163 bff58270 2022-09-19 op printf("int\n%s\n{\n", $3);
169 bff58270 2022-09-19 op body : /* empty */
170 bff58270 2022-09-19 op | body verbatim
172 bff58270 2022-09-19 op | body special
175 bff58270 2022-09-19 op special : '{' RENDER string '}' {
177 bff58270 2022-09-19 op if (strrchr($3, ')') != NULL)
178 c423c56e 2022-09-25 op printf("if (%s == -1) goto err;\n", $3);
180 bff58270 2022-09-19 op printf("if (%s != NULL && %s(tp) == -1) "
181 c423c56e 2022-09-25 op "goto err;\n", $3, $3);
184 bff58270 2022-09-19 op | if body endif { puts("}"); }
186 bff58270 2022-09-19 op | '{' string '|' ESCAPE '}' {
188 c423c56e 2022-09-25 op printf("if (tp->tp_escape(tp, %s) == -1) goto err;\n",
192 bff58270 2022-09-19 op | '{' string '|' UNSAFE '}' {
194 c423c56e 2022-09-25 op printf("if (tp->tp_puts(tp, %s) == -1) goto err;\n",
198 bff58270 2022-09-19 op | '{' string '|' URLESCAPE '}' {
200 c423c56e 2022-09-25 op printf("if (tp_urlescape(tp, %s) == -1) goto err;\n",
204 bff58270 2022-09-19 op | '{' string '}' {
206 c423c56e 2022-09-25 op printf("if (tp->tp_escape(tp, %s) == -1) goto err;\n",
212 bff58270 2022-09-19 op if : '{' IF string '}' {
214 bff58270 2022-09-19 op printf("if (%s) {\n", $3);
220 bff58270 2022-09-19 op | else body end
221 bff58270 2022-09-19 op | elsif body endif
224 bff58270 2022-09-19 op elsif : '{' ELSE IF string '}' {
226 bff58270 2022-09-19 op printf("} else if (%s) {\n", $4);
231 bff58270 2022-09-19 op else : '{' ELSE '}' {
233 bff58270 2022-09-19 op puts("} else {");
237 bff58270 2022-09-19 op loop : '{' TQFOREACH STRING STRING STRING '}' {
238 bff58270 2022-09-19 op printf("TAILQ_FOREACH(%s, %s, %s) {\n",
248 bff58270 2022-09-19 op end : '{' END '}'
251 bff58270 2022-09-19 op finally : '{' FINALLY '}' {
253 c423c56e 2022-09-25 op puts("err:");
257 bff58270 2022-09-19 op string : STRING string {
258 bff58270 2022-09-19 op if (asprintf(&$$, "%s %s", $1, $2) == -1)
259 bff58270 2022-09-19 op err(1, "asprintf");
268 bff58270 2022-09-19 op struct keywords {
269 bff58270 2022-09-19 op const char *k_name;
274 bff58270 2022-09-19 op yyerror(const char *fmt, ...)
279 bff58270 2022-09-19 op file->errors++;
280 bff58270 2022-09-19 op va_start(ap, fmt);
281 bff58270 2022-09-19 op if (vasprintf(&msg, fmt, ap) == -1)
282 bff58270 2022-09-19 op err(1, "yyerror vasprintf");
284 bff58270 2022-09-19 op fprintf(stderr, "%s:%d: %s\n", file->name, yylval.lineno, msg);
290 bff58270 2022-09-19 op kw_cmp(const void *k, const void *e)
292 bff58270 2022-09-19 op return (strcmp(k, ((const struct keywords *)e)->k_name));
296 bff58270 2022-09-19 op lookup(char *s)
298 bff58270 2022-09-19 op /* this has to be sorted always */
299 bff58270 2022-09-19 op static const struct keywords keywords[] = {
300 bff58270 2022-09-19 op { "define", DEFINE },
301 bff58270 2022-09-19 op { "else", ELSE },
302 bff58270 2022-09-19 op { "end", END },
303 bff58270 2022-09-19 op { "escape", ESCAPE },
304 bff58270 2022-09-19 op { "finally", FINALLY },
305 bff58270 2022-09-19 op { "if", IF },
306 bff58270 2022-09-19 op { "include", INCLUDE },
307 bff58270 2022-09-19 op { "render", RENDER },
308 bff58270 2022-09-19 op { "tailq-foreach", TQFOREACH },
309 bff58270 2022-09-19 op { "unsafe", UNSAFE },
310 bff58270 2022-09-19 op { "urlescape", URLESCAPE },
312 bff58270 2022-09-19 op const struct keywords *p;
314 bff58270 2022-09-19 op p = bsearch(s, keywords, nitems(keywords), sizeof(keywords[0]),
318 bff58270 2022-09-19 op return (p->k_val);
320 bff58270 2022-09-19 op return (STRING);
323 bff58270 2022-09-19 op #define START_EXPAND 1
324 bff58270 2022-09-19 op #define DONE_EXPAND 2
326 bff58270 2022-09-19 op static int expanding;
334 bff58270 2022-09-19 op if (file->ungetpos > 0)
335 bff58270 2022-09-19 op c = file->ungetbuf[--file->ungetpos];
337 bff58270 2022-09-19 op c = getc(file->stream);
339 bff58270 2022-09-19 op if (c == START_EXPAND)
340 bff58270 2022-09-19 op expanding = 1;
341 bff58270 2022-09-19 op else if (c == DONE_EXPAND)
342 bff58270 2022-09-19 op expanding = 0;
350 bff58270 2022-09-19 op lgetc(int quotec)
354 bff58270 2022-09-19 op if (quotec) {
355 bff58270 2022-09-19 op if ((c = igetc()) == EOF) {
356 bff58270 2022-09-19 op yyerror("reached end of filewhile parsing "
357 bff58270 2022-09-19 op "quoted string");
358 bff58270 2022-09-19 op if (file == topfile || popfile() == EOF)
359 bff58270 2022-09-19 op return (EOF);
360 bff58270 2022-09-19 op return (quotec);
365 bff58270 2022-09-19 op while ((c = igetc()) == '\\') {
366 bff58270 2022-09-19 op next = igetc();
367 bff58270 2022-09-19 op if (next != '\n') {
371 bff58270 2022-09-19 op yylval.lineno = file->lineno;
372 bff58270 2022-09-19 op file->lineno++;
374 bff58270 2022-09-19 op if (c == '\t' || c == ' ') {
375 bff58270 2022-09-19 op /* Compress blanks to a sigle space. */
377 bff58270 2022-09-19 op c = getc(file->stream);
378 bff58270 2022-09-19 op } while (c == '\t' || c == ' ');
379 bff58270 2022-09-19 op ungetc(c, file->stream);
383 bff58270 2022-09-19 op if (c == EOF) {
385 bff58270 2022-09-19 op * Fake EOL when hit EOF for the first time. This gets line
386 bff58270 2022-09-19 op * count rigchtif last line included file is syntactically
387 bff58270 2022-09-19 op * invalid and has no newline.
389 bff58270 2022-09-19 op if (file->eof_reached == 0) {
390 bff58270 2022-09-19 op file->eof_reached = 1;
391 bff58270 2022-09-19 op return ('\n');
393 bff58270 2022-09-19 op while (c == EOF) {
394 bff58270 2022-09-19 op if (file == topfile || popfile() == EOF)
395 bff58270 2022-09-19 op return (EOF);
403 bff58270 2022-09-19 op lungetc(int c)
405 bff58270 2022-09-19 op if (c == EOF)
408 bff58270 2022-09-19 op if (file->ungetpos >= file->ungetsize) {
409 bff58270 2022-09-19 op void *p = reallocarray(file->ungetbuf, file->ungetsize, 2);
410 bff58270 2022-09-19 op if (p == NULL)
411 bff58270 2022-09-19 op err(1, "reallocarray");
412 bff58270 2022-09-19 op file->ungetbuf = p;
413 bff58270 2022-09-19 op file->ungetsize *= 2;
415 bff58270 2022-09-19 op file->ungetbuf[file->ungetpos++] = c;
419 bff58270 2022-09-19 op findeol(void)
423 bff58270 2022-09-19 op /* skip to either EOF or the first real EOL */
425 bff58270 2022-09-19 op c = lgetc(0);
426 bff58270 2022-09-19 op if (c == '\n') {
427 bff58270 2022-09-19 op file->lineno++;
430 bff58270 2022-09-19 op if (c == EOF)
433 bff58270 2022-09-19 op return (ERROR);
439 bff58270 2022-09-19 op char buf[8096];
440 bff58270 2022-09-19 op char *p = buf;
443 bff58270 2022-09-19 op int starting = 0;
444 bff58270 2022-09-19 op int ending = 0;
446 bff58270 2022-09-19 op if (!in_define && block == 0) {
447 bff58270 2022-09-19 op while ((c = lgetc(0)) != '{' && c != EOF) {
448 bff58270 2022-09-19 op if (c == '\n')
449 bff58270 2022-09-19 op file->lineno++;
452 bff58270 2022-09-19 op if (c == EOF)
456 bff58270 2022-09-19 op c = lgetc(0);
457 bff58270 2022-09-19 op if (c == '{' || c == '!') {
458 bff58270 2022-09-19 op if (c == '{')
464 bff58270 2022-09-19 op if (c == '\n')
465 bff58270 2022-09-19 op file->lineno++;
468 bff58270 2022-09-19 op while ((c = lgetc(0)) == ' ' || c == '\t' || c == '\n') {
469 bff58270 2022-09-19 op if (c == '\n')
470 bff58270 2022-09-19 op file->lineno++;
473 bff58270 2022-09-19 op if (c == EOF) {
474 bff58270 2022-09-19 op yyerror("unterminated block");
478 bff58270 2022-09-19 op yylval.lineno = file->lineno;
480 bff58270 2022-09-19 op if (block != 0 && c == block) {
481 bff58270 2022-09-19 op if ((c = lgetc(0)) == '}') {
482 bff58270 2022-09-19 op if (block == '!') {
484 bff58270 2022-09-19 op return ('!');
487 bff58270 2022-09-19 op return ('}');
493 bff58270 2022-09-19 op if (in_define && block == 0) {
494 bff58270 2022-09-19 op if (c == '{')
495 bff58270 2022-09-19 op goto newblock;
498 bff58270 2022-09-19 op if (starting) {
499 bff58270 2022-09-19 op if (c == '!' || c == '{') {
500 bff58270 2022-09-19 op lungetc('{');
504 bff58270 2022-09-19 op starting = 0;
507 bff58270 2022-09-19 op } else if (c == '{') {
508 bff58270 2022-09-19 op starting = 1;
513 bff58270 2022-09-19 op if ((size_t)(p - buf) >= sizeof(buf)) {
514 bff58270 2022-09-19 op yyerror("string too long");
515 bff58270 2022-09-19 op return (findeol());
517 bff58270 2022-09-19 op } while ((c = lgetc(0)) != EOF && c != '\n');
519 bff58270 2022-09-19 op if (c == EOF) {
520 bff58270 2022-09-19 op yyerror("unterminated block");
523 bff58270 2022-09-19 op if (c == '\n')
524 bff58270 2022-09-19 op file->lineno++;
525 bff58270 2022-09-19 op if ((yylval.v.string = strdup(buf)) == NULL)
526 bff58270 2022-09-19 op err(1, "strdup");
527 bff58270 2022-09-19 op return (STRING);
530 bff58270 2022-09-19 op if (block == '!') {
532 bff58270 2022-09-19 op if (ending) {
533 bff58270 2022-09-19 op if (c == '}') {
535 bff58270 2022-09-19 op lungetc(block);
541 bff58270 2022-09-19 op } else if (c == '!') {
547 bff58270 2022-09-19 op if ((size_t)(p - buf) >= sizeof(buf)) {
548 bff58270 2022-09-19 op yyerror("line too long");
549 bff58270 2022-09-19 op return (findeol());
551 bff58270 2022-09-19 op } while ((c = lgetc(0)) != EOF && c != '\n');
554 bff58270 2022-09-19 op if (c == EOF) {
555 bff58270 2022-09-19 op yyerror("unterminated block");
558 bff58270 2022-09-19 op if (c == '\n')
559 bff58270 2022-09-19 op file->lineno++;
561 bff58270 2022-09-19 op if ((yylval.v.string = strdup(buf)) == NULL)
562 bff58270 2022-09-19 op err(1, "strdup");
563 bff58270 2022-09-19 op return (STRING);
566 bff58270 2022-09-19 op if (c == '|')
570 bff58270 2022-09-19 op if (c == '|') {
575 bff58270 2022-09-19 op if (ending) {
576 bff58270 2022-09-19 op if (c == '}') {
578 bff58270 2022-09-19 op lungetc('}');
584 bff58270 2022-09-19 op } else if (c == '}') {
590 bff58270 2022-09-19 op if ((size_t)(p - buf) >= sizeof(buf)) {
591 bff58270 2022-09-19 op yyerror("string too long");
592 bff58270 2022-09-19 op return (findeol());
594 bff58270 2022-09-19 op } while ((c = lgetc(0)) != EOF && !isspace((unsigned char)c));
597 bff58270 2022-09-19 op if (c == EOF) {
598 bff58270 2022-09-19 op yyerror("unterminated block");
601 bff58270 2022-09-19 op if (c == '\n')
602 bff58270 2022-09-19 op file->lineno++;
603 bff58270 2022-09-19 op if ((token = lookup(buf)) == STRING)
604 bff58270 2022-09-19 op if ((yylval.v.string = strdup(buf)) == NULL)
605 bff58270 2022-09-19 op err(1, "strdup");
606 bff58270 2022-09-19 op return (token);
609 bff58270 2022-09-19 op struct file *
610 bff58270 2022-09-19 op pushfile(const char *name, int secret)
612 bff58270 2022-09-19 op struct file *nfile;
614 bff58270 2022-09-19 op if ((nfile = calloc(1, sizeof(*nfile))) == NULL)
615 bff58270 2022-09-19 op err(1, "calloc");
616 bff58270 2022-09-19 op if ((nfile->name = strdup(name)) == NULL)
617 bff58270 2022-09-19 op err(1, "strdup");
618 bff58270 2022-09-19 op if ((nfile->stream = fopen(nfile->name, "r")) == NULL) {
619 bff58270 2022-09-19 op warn("can't open %s", nfile->name);
620 bff58270 2022-09-19 op free(nfile->name);
622 bff58270 2022-09-19 op return (NULL);
624 bff58270 2022-09-19 op nfile->lineno = TAILQ_EMPTY(&files) ? 1 : 0;
625 bff58270 2022-09-19 op nfile->ungetsize = 16;
626 bff58270 2022-09-19 op nfile->ungetbuf = malloc(nfile->ungetsize);
627 bff58270 2022-09-19 op if (nfile->ungetbuf == NULL)
628 bff58270 2022-09-19 op err(1, "malloc");
629 bff58270 2022-09-19 op TAILQ_INSERT_TAIL(&files, nfile, entry);
630 bff58270 2022-09-19 op return (nfile);
634 bff58270 2022-09-19 op popfile(void)
636 bff58270 2022-09-19 op struct file *prev;
638 bff58270 2022-09-19 op if ((prev = TAILQ_PREV(file, files, entry)) != NULL)
639 bff58270 2022-09-19 op prev->errors += file->errors;
641 bff58270 2022-09-19 op TAILQ_REMOVE(&files, file, entry);
642 bff58270 2022-09-19 op fclose(file->stream);
643 bff58270 2022-09-19 op free(file->name);
644 bff58270 2022-09-19 op free(file->ungetbuf);
647 bff58270 2022-09-19 op return (file ? 0 : EOF);
651 bff58270 2022-09-19 op parse(const char *filename)
653 bff58270 2022-09-19 op if ((file = pushfile(filename, 0)) == 0)
655 bff58270 2022-09-19 op topfile = file;
658 bff58270 2022-09-19 op errors = file->errors;
661 bff58270 2022-09-19 op return (errors ? -1 : 0);
670 bff58270 2022-09-19 op printf("#line %d ", yylval.lineno);
671 bff58270 2022-09-19 op printq(file->name);
672 bff58270 2022-09-19 op putchar('\n');
676 bff58270 2022-09-19 op printq(const char *str)
678 bff58270 2022-09-19 op putchar('"');
679 bff58270 2022-09-19 op for (; *str; ++str) {
680 bff58270 2022-09-19 op if (*str == '"')
681 bff58270 2022-09-19 op putchar('\\');
682 bff58270 2022-09-19 op putchar(*str);
684 bff58270 2022-09-19 op putchar('"');