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