1 bff58270 2022-09-19 op /*
  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.
 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 bff58270 2022-09-19 op %token	RENDER TQFOREACH UNSAFE URLESCAPE
 91 bff58270 2022-09-19 op %token	<v.string>	STRING
 92 bff58270 2022-09-19 op %type	<v.string>	string
 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 bff58270 2022-09-19 op raw		: STRING {
141 bff58270 2022-09-19 op 			dbg();
142 14c31a43 2022-11-25 op 			fprintf(fp, "if ((tp_ret = tp->tp_puts(tp, ");
143 bff58270 2022-09-19 op 			printq($1);
144 14c31a43 2022-11-25 op 			fputs(")) == -1) goto err;\n", fp);
145 bff58270 2022-09-19 op 
146 bff58270 2022-09-19 op 			free($1);
147 bff58270 2022-09-19 op 		}
148 bff58270 2022-09-19 op 		;
149 bff58270 2022-09-19 op 
150 bff58270 2022-09-19 op block		: define body end {
151 14c31a43 2022-11-25 op 			fputs("err:\n", fp);
152 14c31a43 2022-11-25 op 			fputs("return tp_ret;\n", fp);
153 14c31a43 2022-11-25 op 			fputs("}\n", fp);
154 bff58270 2022-09-19 op 			in_define = 0;
155 bff58270 2022-09-19 op 		}
156 bff58270 2022-09-19 op 		| define body finally end {
157 14c31a43 2022-11-25 op 			fputs("return tp_ret;\n", fp);
158 14c31a43 2022-11-25 op 			fputs("}\n", fp);
159 bff58270 2022-09-19 op 			in_define = 0;
160 bff58270 2022-09-19 op 		}
161 bff58270 2022-09-19 op 		;
162 bff58270 2022-09-19 op 
163 bff58270 2022-09-19 op define		: '{' DEFINE string '}' {
164 bff58270 2022-09-19 op 			in_define = 1;
165 bff58270 2022-09-19 op 
166 bff58270 2022-09-19 op 			dbg();
167 14c31a43 2022-11-25 op 			fprintf(fp, "int\n%s\n{\n", $3);
168 14c31a43 2022-11-25 op 			fputs("int tp_ret = 0;\n", fp);
169 bff58270 2022-09-19 op 
170 bff58270 2022-09-19 op 			free($3);
171 bff58270 2022-09-19 op 		}
172 bff58270 2022-09-19 op 		;
173 bff58270 2022-09-19 op 
174 bff58270 2022-09-19 op body		: /* empty */
175 bff58270 2022-09-19 op 		| body verbatim
176 bff58270 2022-09-19 op 		| body raw
177 bff58270 2022-09-19 op 		| body special
178 bff58270 2022-09-19 op 		;
179 bff58270 2022-09-19 op 
180 bff58270 2022-09-19 op special		: '{' RENDER string '}' {
181 bff58270 2022-09-19 op 			dbg();
182 14c31a43 2022-11-25 op 			fprintf(fp, "if ((tp_ret = %s) == -1) goto err;\n",
183 14c31a43 2022-11-25 op 			    $3);
184 bff58270 2022-09-19 op 			free($3);
185 bff58270 2022-09-19 op 		}
186 83e0f74d 2022-09-29 op 		| printf
187 14c31a43 2022-11-25 op 		| if body endif			{ fputs("}\n", fp); }
188 bff58270 2022-09-19 op 		| loop
189 bff58270 2022-09-19 op 		| '{' string '|' UNSAFE '}' {
190 bff58270 2022-09-19 op 			dbg();
191 14c31a43 2022-11-25 op 			fprintf(fp,
192 14c31a43 2022-11-25 op 			    "if ((tp_ret = tp->tp_puts(tp, %s)) == -1)\n",
193 83f0f95a 2022-09-29 op 			    $2);
194 14c31a43 2022-11-25 op 			fputs("goto err;\n", fp);
195 bff58270 2022-09-19 op 			free($2);
196 bff58270 2022-09-19 op 		}
197 bff58270 2022-09-19 op 		| '{' string '|' URLESCAPE '}' {
198 bff58270 2022-09-19 op 			dbg();
199 14c31a43 2022-11-25 op 			fprintf(fp,
200 14c31a43 2022-11-25 op 			    "if ((tp_ret = tp_urlescape(tp, %s)) == -1)\n",
201 83f0f95a 2022-09-29 op 			    $2);
202 14c31a43 2022-11-25 op 			fputs("goto err;\n", fp);
203 bff58270 2022-09-19 op 			free($2);
204 bff58270 2022-09-19 op 		}
205 bff58270 2022-09-19 op 		| '{' string '}' {
206 bff58270 2022-09-19 op 			dbg();
207 14c31a43 2022-11-25 op 			fprintf(fp,
208 14c31a43 2022-11-25 op 			    "if ((tp_ret = tp->tp_escape(tp, %s)) == -1)\n",
209 83f0f95a 2022-09-29 op 			    $2);
210 14c31a43 2022-11-25 op 			fputs("goto err;\n", fp);
211 bff58270 2022-09-19 op 			free($2);
212 83e0f74d 2022-09-29 op 		}
213 83e0f74d 2022-09-29 op 		;
214 83e0f74d 2022-09-29 op 
215 83e0f74d 2022-09-29 op printf		: '{' PRINTF {
216 83e0f74d 2022-09-29 op 			dbg();
217 14c31a43 2022-11-25 op 			fprintf(fp, "if (asprintf(&tp->tp_tmp, ");
218 83e0f74d 2022-09-29 op 		} printfargs '}' {
219 14c31a43 2022-11-25 op 			fputs(") == -1)\n", fp);
220 14c31a43 2022-11-25 op 			fputs("goto err;\n", fp);
221 14c31a43 2022-11-25 op 			fputs("if ((tp_ret = tp->tp_escape(tp, tp->tp_tmp)) "
222 14c31a43 2022-11-25 op 			    "== -1)\n", fp);
223 14c31a43 2022-11-25 op 			fputs("goto err;\n", fp);
224 14c31a43 2022-11-25 op 			fputs("free(tp->tp_tmp);\n", fp);
225 14c31a43 2022-11-25 op 			fputs("tp->tp_tmp = NULL;\n", fp);
226 bff58270 2022-09-19 op 		}
227 bff58270 2022-09-19 op 		;
228 83e0f74d 2022-09-29 op 
229 83e0f74d 2022-09-29 op printfargs	: /* empty */
230 83e0f74d 2022-09-29 op 		| printfargs STRING {
231 14c31a43 2022-11-25 op 			fprintf(fp, " %s", $2);
232 83e0f74d 2022-09-29 op 			free($2);
233 83e0f74d 2022-09-29 op 		}
234 6125bd61 2022-09-29 op 		;
235 bff58270 2022-09-19 op 
236 6125bd61 2022-09-29 op if		: '{' IF stringy '}' {
237 bff58270 2022-09-19 op 			dbg();
238 14c31a43 2022-11-25 op 			fprintf(fp, "if (%s) {\n", $3);
239 bff58270 2022-09-19 op 			free($3);
240 bff58270 2022-09-19 op 		}
241 bff58270 2022-09-19 op 		;
242 bff58270 2022-09-19 op 
243 bff58270 2022-09-19 op endif		: end
244 bff58270 2022-09-19 op 		| else body end
245 bff58270 2022-09-19 op 		| elsif body endif
246 bff58270 2022-09-19 op 		;
247 bff58270 2022-09-19 op 
248 6125bd61 2022-09-29 op elsif		: '{' ELSE IF stringy '}' {
249 bff58270 2022-09-19 op 			dbg();
250 14c31a43 2022-11-25 op 			fprintf(fp, "} else if (%s) {\n", $4);
251 bff58270 2022-09-19 op 			free($4);
252 bff58270 2022-09-19 op 		}
253 bff58270 2022-09-19 op 		;
254 bff58270 2022-09-19 op 
255 bff58270 2022-09-19 op else		: '{' ELSE '}' {
256 bff58270 2022-09-19 op 			dbg();
257 14c31a43 2022-11-25 op 			fputs("} else {\n", fp);
258 bff58270 2022-09-19 op 		}
259 bff58270 2022-09-19 op 		;
260 bff58270 2022-09-19 op 
261 9c75d045 2022-10-03 op loop		: '{' FOR stringy '}' {
262 14c31a43 2022-11-25 op 			fprintf(fp, "for (%s) {\n", $3);
263 9c75d045 2022-10-03 op 			free($3);
264 9c75d045 2022-10-03 op 		} body end {
265 14c31a43 2022-11-25 op 			fputs("}\n", fp);
266 9c75d045 2022-10-03 op 		}
267 9c75d045 2022-10-03 op 		| '{' TQFOREACH STRING STRING STRING '}' {
268 14c31a43 2022-11-25 op 			fprintf(fp, "TAILQ_FOREACH(%s, %s, %s) {\n",
269 bff58270 2022-09-19 op 			    $3, $4, $5);
270 bff58270 2022-09-19 op 			free($3);
271 bff58270 2022-09-19 op 			free($4);
272 bff58270 2022-09-19 op 			free($5);
273 bff58270 2022-09-19 op 		} body end {
274 14c31a43 2022-11-25 op 			fputs("}\n", fp);
275 bff58270 2022-09-19 op 		}
276 bff58270 2022-09-19 op 		;
277 bff58270 2022-09-19 op 
278 bff58270 2022-09-19 op end		: '{' END '}'
279 bff58270 2022-09-19 op 		;
280 bff58270 2022-09-19 op 
281 bff58270 2022-09-19 op finally		: '{' FINALLY '}' {
282 bff58270 2022-09-19 op 			dbg();
283 14c31a43 2022-11-25 op 			fputs("err:\n", fp);
284 bff58270 2022-09-19 op 		} verbatims
285 bff58270 2022-09-19 op 		;
286 bff58270 2022-09-19 op 
287 bff58270 2022-09-19 op string		: STRING string {
288 bff58270 2022-09-19 op 			if (asprintf(&$$, "%s %s", $1, $2) == -1)
289 bff58270 2022-09-19 op 				err(1, "asprintf");
290 bff58270 2022-09-19 op 			free($1);
291 bff58270 2022-09-19 op 			free($2);
292 bff58270 2022-09-19 op 		}
293 bff58270 2022-09-19 op 		| STRING
294 6125bd61 2022-09-29 op 		;
295 6125bd61 2022-09-29 op 
296 6125bd61 2022-09-29 op stringy		: STRING
297 6125bd61 2022-09-29 op 		| STRING stringy {
298 6125bd61 2022-09-29 op 			if (asprintf(&$$, "%s %s", $1, $2) == -1)
299 6125bd61 2022-09-29 op 				err(1, "asprintf");
300 6125bd61 2022-09-29 op 			free($1);
301 6125bd61 2022-09-29 op 			free($2);
302 6125bd61 2022-09-29 op 		}
303 6125bd61 2022-09-29 op 		| '|' stringy {
304 6125bd61 2022-09-29 op 			if (asprintf(&$$, "|%s", $2) == -1)
305 6125bd61 2022-09-29 op 				err(1, "asprintf");
306 6125bd61 2022-09-29 op 			free($2);
307 6125bd61 2022-09-29 op 		}
308 bff58270 2022-09-19 op 		;
309 bff58270 2022-09-19 op 
310 bff58270 2022-09-19 op %%
311 bff58270 2022-09-19 op 
312 bff58270 2022-09-19 op struct keywords {
313 bff58270 2022-09-19 op 	const char	*k_name;
314 bff58270 2022-09-19 op 	int		 k_val;
315 bff58270 2022-09-19 op };
316 bff58270 2022-09-19 op 
317 bff58270 2022-09-19 op int
318 bff58270 2022-09-19 op yyerror(const char *fmt, ...)
319 bff58270 2022-09-19 op {
320 bff58270 2022-09-19 op 	va_list	 ap;
321 bff58270 2022-09-19 op 	char	*msg;
322 bff58270 2022-09-19 op 
323 bff58270 2022-09-19 op 	file->errors++;
324 bff58270 2022-09-19 op 	va_start(ap, fmt);
325 bff58270 2022-09-19 op 	if (vasprintf(&msg, fmt, ap) == -1)
326 bff58270 2022-09-19 op 		err(1, "yyerror vasprintf");
327 bff58270 2022-09-19 op 	va_end(ap);
328 bff58270 2022-09-19 op 	fprintf(stderr, "%s:%d: %s\n", file->name, yylval.lineno, msg);
329 bff58270 2022-09-19 op 	free(msg);
330 bff58270 2022-09-19 op 	return (0);
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 kw_cmp(const void *k, const void *e)
335 bff58270 2022-09-19 op {
336 bff58270 2022-09-19 op 	return (strcmp(k, ((const struct keywords *)e)->k_name));
337 bff58270 2022-09-19 op }
338 bff58270 2022-09-19 op 
339 bff58270 2022-09-19 op int
340 bff58270 2022-09-19 op lookup(char *s)
341 bff58270 2022-09-19 op {
342 bff58270 2022-09-19 op 	/* this has to be sorted always */
343 bff58270 2022-09-19 op 	static const struct keywords keywords[] = {
344 bff58270 2022-09-19 op 		{ "define",		DEFINE },
345 bff58270 2022-09-19 op 		{ "else",		ELSE },
346 bff58270 2022-09-19 op 		{ "end",		END },
347 bff58270 2022-09-19 op 		{ "finally",		FINALLY },
348 9c75d045 2022-10-03 op 		{ "for",		FOR },
349 bff58270 2022-09-19 op 		{ "if",			IF },
350 bff58270 2022-09-19 op 		{ "include",		INCLUDE },
351 83e0f74d 2022-09-29 op 		{ "printf",		PRINTF },
352 bff58270 2022-09-19 op 		{ "render",		RENDER },
353 bff58270 2022-09-19 op 		{ "tailq-foreach",	TQFOREACH },
354 bff58270 2022-09-19 op 		{ "unsafe",		UNSAFE },
355 bff58270 2022-09-19 op 		{ "urlescape",		URLESCAPE },
356 bff58270 2022-09-19 op 	};
357 bff58270 2022-09-19 op 	const struct keywords	*p;
358 bff58270 2022-09-19 op 
359 bff58270 2022-09-19 op 	p = bsearch(s, keywords, nitems(keywords), sizeof(keywords[0]),
360 bff58270 2022-09-19 op 	    kw_cmp);
361 bff58270 2022-09-19 op 
362 bff58270 2022-09-19 op 	if (p)
363 bff58270 2022-09-19 op 		return (p->k_val);
364 bff58270 2022-09-19 op 	else
365 bff58270 2022-09-19 op 		return (STRING);
366 bff58270 2022-09-19 op }
367 bff58270 2022-09-19 op 
368 bff58270 2022-09-19 op #define START_EXPAND	1
369 bff58270 2022-09-19 op #define DONE_EXPAND	2
370 bff58270 2022-09-19 op 
371 bff58270 2022-09-19 op static int	expanding;
372 bff58270 2022-09-19 op 
373 bff58270 2022-09-19 op int
374 bff58270 2022-09-19 op igetc(void)
375 bff58270 2022-09-19 op {
376 bff58270 2022-09-19 op 	int	c;
377 bff58270 2022-09-19 op 
378 bff58270 2022-09-19 op 	while (1) {
379 bff58270 2022-09-19 op 		if (file->ungetpos > 0)
380 bff58270 2022-09-19 op 			c = file->ungetbuf[--file->ungetpos];
381 bff58270 2022-09-19 op 		else
382 bff58270 2022-09-19 op 			c = getc(file->stream);
383 bff58270 2022-09-19 op 
384 bff58270 2022-09-19 op 		if (c == START_EXPAND)
385 bff58270 2022-09-19 op 			expanding = 1;
386 bff58270 2022-09-19 op 		else if (c == DONE_EXPAND)
387 bff58270 2022-09-19 op 			expanding = 0;
388 bff58270 2022-09-19 op 		else
389 bff58270 2022-09-19 op 			break;
390 bff58270 2022-09-19 op 	}
391 bff58270 2022-09-19 op 	return (c);
392 bff58270 2022-09-19 op }
393 bff58270 2022-09-19 op 
394 bff58270 2022-09-19 op int
395 bff58270 2022-09-19 op lgetc(int quotec)
396 bff58270 2022-09-19 op {
397 ee34715e 2022-09-29 op 	int		c;
398 bff58270 2022-09-19 op 
399 bff58270 2022-09-19 op 	if (quotec) {
400 bff58270 2022-09-19 op 		if ((c = igetc()) == EOF) {
401 bff58270 2022-09-19 op 			yyerror("reached end of filewhile parsing "
402 bff58270 2022-09-19 op 			    "quoted string");
403 bff58270 2022-09-19 op 			if (file == topfile || popfile() == EOF)
404 bff58270 2022-09-19 op 				return (EOF);
405 bff58270 2022-09-19 op 			return (quotec);
406 bff58270 2022-09-19 op 		}
407 bff58270 2022-09-19 op 		return (c);
408 bff58270 2022-09-19 op 	}
409 bff58270 2022-09-19 op 
410 ee34715e 2022-09-29 op 	c = igetc();
411 bff58270 2022-09-19 op 	if (c == '\t' || c == ' ') {
412 bff58270 2022-09-19 op 		/* Compress blanks to a sigle space. */
413 bff58270 2022-09-19 op 		do {
414 bff58270 2022-09-19 op 			c = getc(file->stream);
415 bff58270 2022-09-19 op 		} while (c == '\t'  || c == ' ');
416 bff58270 2022-09-19 op 		ungetc(c, file->stream);
417 bff58270 2022-09-19 op 		c = ' ';
418 bff58270 2022-09-19 op 	}
419 bff58270 2022-09-19 op 
420 bff58270 2022-09-19 op 	if (c == EOF) {
421 bff58270 2022-09-19 op 		/*
422 bff58270 2022-09-19 op 		 * Fake EOL when hit EOF for the first time. This gets line
423 bff58270 2022-09-19 op 		 * count rigchtif last line included file is syntactically
424 bff58270 2022-09-19 op 		 * invalid and has no newline.
425 bff58270 2022-09-19 op 		 */
426 bff58270 2022-09-19 op 		if (file->eof_reached == 0) {
427 bff58270 2022-09-19 op 			file->eof_reached = 1;
428 bff58270 2022-09-19 op 			return ('\n');
429 bff58270 2022-09-19 op 		}
430 bff58270 2022-09-19 op 		while (c == EOF) {
431 bff58270 2022-09-19 op 			if (file == topfile || popfile() == EOF)
432 bff58270 2022-09-19 op 				return (EOF);
433 bff58270 2022-09-19 op 			c = igetc();
434 bff58270 2022-09-19 op 		}
435 bff58270 2022-09-19 op 	}
436 bff58270 2022-09-19 op 	return (c);
437 bff58270 2022-09-19 op }
438 bff58270 2022-09-19 op 
439 bff58270 2022-09-19 op void
440 bff58270 2022-09-19 op lungetc(int c)
441 bff58270 2022-09-19 op {
442 bff58270 2022-09-19 op 	if (c == EOF)
443 bff58270 2022-09-19 op 		return;
444 bff58270 2022-09-19 op 
445 bff58270 2022-09-19 op 	if (file->ungetpos >= file->ungetsize) {
446 bff58270 2022-09-19 op 		void *p = reallocarray(file->ungetbuf, file->ungetsize, 2);
447 bff58270 2022-09-19 op 		if (p == NULL)
448 bff58270 2022-09-19 op 			err(1, "reallocarray");
449 bff58270 2022-09-19 op 		file->ungetbuf = p;
450 bff58270 2022-09-19 op 		file->ungetsize *= 2;
451 bff58270 2022-09-19 op 	}
452 bff58270 2022-09-19 op 	file->ungetbuf[file->ungetpos++] = c;
453 bff58270 2022-09-19 op }
454 bff58270 2022-09-19 op 
455 bff58270 2022-09-19 op int
456 bff58270 2022-09-19 op findeol(void)
457 bff58270 2022-09-19 op {
458 bff58270 2022-09-19 op 	int	c;
459 bff58270 2022-09-19 op 
460 bff58270 2022-09-19 op 	/* skip to either EOF or the first real EOL */
461 bff58270 2022-09-19 op 	while (1) {
462 bff58270 2022-09-19 op 		c = lgetc(0);
463 bff58270 2022-09-19 op 		if (c == '\n') {
464 bff58270 2022-09-19 op 			file->lineno++;
465 bff58270 2022-09-19 op 			break;
466 bff58270 2022-09-19 op 		}
467 bff58270 2022-09-19 op 		if (c == EOF)
468 bff58270 2022-09-19 op 			break;
469 bff58270 2022-09-19 op 	}
470 bff58270 2022-09-19 op 	return (ERROR);
471 bff58270 2022-09-19 op }
472 bff58270 2022-09-19 op 
473 bff58270 2022-09-19 op int
474 bff58270 2022-09-19 op yylex(void)
475 bff58270 2022-09-19 op {
476 bff58270 2022-09-19 op 	char		 buf[8096];
477 bff58270 2022-09-19 op 	char		*p = buf;
478 bff58270 2022-09-19 op 	int		 c;
479 bff58270 2022-09-19 op 	int		 token;
480 bff58270 2022-09-19 op 	int		 starting = 0;
481 bff58270 2022-09-19 op 	int		 ending = 0;
482 bff58270 2022-09-19 op 
483 bff58270 2022-09-19 op 	if (!in_define && block == 0) {
484 bff58270 2022-09-19 op 		while ((c = lgetc(0)) != '{' && c != EOF) {
485 bff58270 2022-09-19 op 			if (c == '\n')
486 bff58270 2022-09-19 op 				file->lineno++;
487 bff58270 2022-09-19 op 		}
488 bff58270 2022-09-19 op 
489 bff58270 2022-09-19 op 		if (c == EOF)
490 bff58270 2022-09-19 op 			return (0);
491 bff58270 2022-09-19 op 
492 bff58270 2022-09-19 op newblock:
493 bff58270 2022-09-19 op 		c = lgetc(0);
494 bff58270 2022-09-19 op 		if (c == '{' || c == '!') {
495 bff58270 2022-09-19 op 			if (c == '{')
496 bff58270 2022-09-19 op 				block = '}';
497 bff58270 2022-09-19 op 			else
498 bff58270 2022-09-19 op 				block = c;
499 bff58270 2022-09-19 op 			return (c);
500 bff58270 2022-09-19 op 		}
501 bff58270 2022-09-19 op 		if (c == '\n')
502 bff58270 2022-09-19 op 			file->lineno++;
503 bff58270 2022-09-19 op 	}
504 bff58270 2022-09-19 op 
505 bff58270 2022-09-19 op 	while ((c = lgetc(0)) == ' ' || c == '\t' || c == '\n') {
506 bff58270 2022-09-19 op 		if (c == '\n')
507 bff58270 2022-09-19 op 			file->lineno++;
508 bff58270 2022-09-19 op 	}
509 bff58270 2022-09-19 op 
510 bff58270 2022-09-19 op 	if (c == EOF) {
511 bff58270 2022-09-19 op 		yyerror("unterminated block");
512 bff58270 2022-09-19 op 		return (0);
513 bff58270 2022-09-19 op 	}
514 bff58270 2022-09-19 op 
515 bff58270 2022-09-19 op 	yylval.lineno = file->lineno;
516 bff58270 2022-09-19 op 
517 bff58270 2022-09-19 op 	if (block != 0 && c == block) {
518 bff58270 2022-09-19 op 		if ((c = lgetc(0)) == '}') {
519 bff58270 2022-09-19 op 			if (block == '!') {
520 bff58270 2022-09-19 op 				block = 0;
521 bff58270 2022-09-19 op 				return ('!');
522 bff58270 2022-09-19 op 			}
523 bff58270 2022-09-19 op 			block = 0;
524 bff58270 2022-09-19 op 			return ('}');
525 bff58270 2022-09-19 op 		}
526 bff58270 2022-09-19 op 		lungetc(c);
527 bff58270 2022-09-19 op 		c = block;
528 bff58270 2022-09-19 op 	}
529 bff58270 2022-09-19 op 
530 bff58270 2022-09-19 op 	if (in_define && block == 0) {
531 bff58270 2022-09-19 op 		if (c == '{')
532 bff58270 2022-09-19 op 			goto newblock;
533 bff58270 2022-09-19 op 
534 bff58270 2022-09-19 op 		do {
535 bff58270 2022-09-19 op 			if (starting) {
536 bff58270 2022-09-19 op 				if (c == '!' || c == '{') {
537 bff58270 2022-09-19 op 					lungetc(c);
538 c063deb1 2022-10-21 op 					lungetc('{');
539 bff58270 2022-09-19 op 					break;
540 bff58270 2022-09-19 op 				}
541 bff58270 2022-09-19 op 				starting = 0;
542 bff58270 2022-09-19 op 				lungetc(c);
543 bff58270 2022-09-19 op 				c = '{';
544 bff58270 2022-09-19 op 			} else if (c == '{') {
545 bff58270 2022-09-19 op 				starting = 1;
546 bff58270 2022-09-19 op 				continue;
547 bff58270 2022-09-19 op 			}
548 bff58270 2022-09-19 op 
549 bff58270 2022-09-19 op 			*p++ = c;
550 bff58270 2022-09-19 op 			if ((size_t)(p - buf) >= sizeof(buf)) {
551 bff58270 2022-09-19 op 				yyerror("string too long");
552 bff58270 2022-09-19 op 				return (findeol());
553 bff58270 2022-09-19 op 			}
554 bff58270 2022-09-19 op 		} while ((c = lgetc(0)) != EOF && c != '\n');
555 bff58270 2022-09-19 op 		*p = '\0';
556 bff58270 2022-09-19 op 		if (c == EOF) {
557 bff58270 2022-09-19 op 			yyerror("unterminated block");
558 bff58270 2022-09-19 op 			return (0);
559 bff58270 2022-09-19 op 		}
560 bff58270 2022-09-19 op 		if (c == '\n')
561 bff58270 2022-09-19 op 			file->lineno++;
562 bff58270 2022-09-19 op 		if ((yylval.v.string = strdup(buf)) == NULL)
563 bff58270 2022-09-19 op 			err(1, "strdup");
564 bff58270 2022-09-19 op 		return (STRING);
565 bff58270 2022-09-19 op 	}
566 bff58270 2022-09-19 op 
567 bff58270 2022-09-19 op 	if (block == '!') {
568 bff58270 2022-09-19 op 		do {
569 bff58270 2022-09-19 op 			if (ending) {
570 bff58270 2022-09-19 op 				if (c == '}') {
571 bff58270 2022-09-19 op 					lungetc(c);
572 bff58270 2022-09-19 op 					lungetc(block);
573 bff58270 2022-09-19 op 					break;
574 bff58270 2022-09-19 op 				}
575 bff58270 2022-09-19 op 				ending = 0;
576 bff58270 2022-09-19 op 				lungetc(c);
577 bff58270 2022-09-19 op 				c = block;
578 bff58270 2022-09-19 op 			} else if (c == '!') {
579 bff58270 2022-09-19 op 				ending = 1;
580 bff58270 2022-09-19 op 				continue;
581 bff58270 2022-09-19 op 			}
582 bff58270 2022-09-19 op 
583 bff58270 2022-09-19 op 			*p++ = c;
584 bff58270 2022-09-19 op 			if ((size_t)(p - buf) >= sizeof(buf)) {
585 bff58270 2022-09-19 op 				yyerror("line too long");
586 bff58270 2022-09-19 op 				return (findeol());
587 bff58270 2022-09-19 op 			}
588 bff58270 2022-09-19 op 		} while ((c = lgetc(0)) != EOF && c != '\n');
589 bff58270 2022-09-19 op 		*p = '\0';
590 bff58270 2022-09-19 op 
591 bff58270 2022-09-19 op 		if (c == EOF) {
592 bff58270 2022-09-19 op 			yyerror("unterminated block");
593 bff58270 2022-09-19 op 			return (0);
594 bff58270 2022-09-19 op 		}
595 bff58270 2022-09-19 op 		if (c == '\n')
596 bff58270 2022-09-19 op 			file->lineno++;
597 bff58270 2022-09-19 op 
598 bff58270 2022-09-19 op 		if ((yylval.v.string = strdup(buf)) == NULL)
599 bff58270 2022-09-19 op 			err(1, "strdup");
600 bff58270 2022-09-19 op 		return (STRING);
601 bff58270 2022-09-19 op 	}
602 bff58270 2022-09-19 op 
603 bff58270 2022-09-19 op 	if (c == '|')
604 bff58270 2022-09-19 op 		return (c);
605 bff58270 2022-09-19 op 
606 bff58270 2022-09-19 op 	do {
607 bff58270 2022-09-19 op 		if (c == '|') {
608 bff58270 2022-09-19 op 			lungetc(c);
609 bff58270 2022-09-19 op 			break;
610 bff58270 2022-09-19 op 		}
611 bff58270 2022-09-19 op 
612 bff58270 2022-09-19 op 		if (ending) {
613 bff58270 2022-09-19 op 			if (c == '}') {
614 bff58270 2022-09-19 op 				lungetc(c);
615 bff58270 2022-09-19 op 				lungetc('}');
616 bff58270 2022-09-19 op 				break;
617 bff58270 2022-09-19 op 			}
618 bff58270 2022-09-19 op 			ending = 0;
619 bff58270 2022-09-19 op 			lungetc(c);
620 bff58270 2022-09-19 op 			c = block;
621 bff58270 2022-09-19 op 		} else if (c == '}') {
622 bff58270 2022-09-19 op 			ending = 1;
623 bff58270 2022-09-19 op 			continue;
624 bff58270 2022-09-19 op 		}
625 bff58270 2022-09-19 op 
626 bff58270 2022-09-19 op 		*p++ = c;
627 bff58270 2022-09-19 op 		if ((size_t)(p - buf) >= sizeof(buf)) {
628 bff58270 2022-09-19 op 			yyerror("string too long");
629 bff58270 2022-09-19 op 			return (findeol());
630 bff58270 2022-09-19 op 		}
631 bff58270 2022-09-19 op 	} while ((c = lgetc(0)) != EOF && !isspace((unsigned char)c));
632 bff58270 2022-09-19 op 	*p = '\0';
633 bff58270 2022-09-19 op 
634 bff58270 2022-09-19 op 	if (c == EOF) {
635 bff58270 2022-09-19 op 		yyerror("unterminated block");
636 bff58270 2022-09-19 op 		return (0);
637 bff58270 2022-09-19 op 	}
638 bff58270 2022-09-19 op 	if (c ==  '\n')
639 bff58270 2022-09-19 op 		file->lineno++;
640 bff58270 2022-09-19 op 	if ((token = lookup(buf)) == STRING)
641 bff58270 2022-09-19 op 		if ((yylval.v.string = strdup(buf)) == NULL)
642 bff58270 2022-09-19 op 			err(1, "strdup");
643 bff58270 2022-09-19 op 	return (token);
644 bff58270 2022-09-19 op }
645 bff58270 2022-09-19 op 
646 bff58270 2022-09-19 op struct file *
647 bff58270 2022-09-19 op pushfile(const char *name, int secret)
648 bff58270 2022-09-19 op {
649 bff58270 2022-09-19 op 	struct file	*nfile;
650 bff58270 2022-09-19 op 
651 bff58270 2022-09-19 op 	if ((nfile = calloc(1, sizeof(*nfile))) == NULL)
652 bff58270 2022-09-19 op 		err(1, "calloc");
653 bff58270 2022-09-19 op 	if ((nfile->name = strdup(name)) == NULL)
654 bff58270 2022-09-19 op 		err(1, "strdup");
655 bff58270 2022-09-19 op 	if ((nfile->stream = fopen(nfile->name, "r")) == NULL) {
656 bff58270 2022-09-19 op 		warn("can't open %s", nfile->name);
657 bff58270 2022-09-19 op 		free(nfile->name);
658 bff58270 2022-09-19 op 		free(nfile);
659 bff58270 2022-09-19 op 		return (NULL);
660 bff58270 2022-09-19 op 	}
661 bff58270 2022-09-19 op 	nfile->lineno = TAILQ_EMPTY(&files) ? 1 : 0;
662 bff58270 2022-09-19 op 	nfile->ungetsize = 16;
663 bff58270 2022-09-19 op 	nfile->ungetbuf = malloc(nfile->ungetsize);
664 bff58270 2022-09-19 op 	if (nfile->ungetbuf == NULL)
665 bff58270 2022-09-19 op 		err(1, "malloc");
666 bff58270 2022-09-19 op 	TAILQ_INSERT_TAIL(&files, nfile, entry);
667 bff58270 2022-09-19 op 	return (nfile);
668 bff58270 2022-09-19 op }
669 bff58270 2022-09-19 op 
670 bff58270 2022-09-19 op int
671 bff58270 2022-09-19 op popfile(void)
672 bff58270 2022-09-19 op {
673 bff58270 2022-09-19 op 	struct file	*prev;
674 bff58270 2022-09-19 op 
675 bff58270 2022-09-19 op 	if ((prev = TAILQ_PREV(file, files, entry)) != NULL)
676 bff58270 2022-09-19 op 		prev->errors += file->errors;
677 bff58270 2022-09-19 op 
678 bff58270 2022-09-19 op 	TAILQ_REMOVE(&files, file, entry);
679 bff58270 2022-09-19 op 	fclose(file->stream);
680 bff58270 2022-09-19 op 	free(file->name);
681 bff58270 2022-09-19 op 	free(file->ungetbuf);
682 bff58270 2022-09-19 op 	free(file);
683 bff58270 2022-09-19 op 	file = prev;
684 bff58270 2022-09-19 op 	return (file ? 0 : EOF);
685 bff58270 2022-09-19 op }
686 bff58270 2022-09-19 op 
687 bff58270 2022-09-19 op int
688 14c31a43 2022-11-25 op parse(FILE *outfile, const char *filename)
689 bff58270 2022-09-19 op {
690 14c31a43 2022-11-25 op 	fp = outfile;
691 14c31a43 2022-11-25 op 
692 bff58270 2022-09-19 op 	if ((file = pushfile(filename, 0)) == 0)
693 bff58270 2022-09-19 op 		return (-1);
694 bff58270 2022-09-19 op 	topfile = file;
695 bff58270 2022-09-19 op 
696 bff58270 2022-09-19 op 	yyparse();
697 bff58270 2022-09-19 op 	errors = file->errors;
698 bff58270 2022-09-19 op 	popfile();
699 bff58270 2022-09-19 op 
700 bff58270 2022-09-19 op 	return (errors ? -1 : 0);
701 bff58270 2022-09-19 op }
702 bff58270 2022-09-19 op 
703 bff58270 2022-09-19 op void
704 bff58270 2022-09-19 op dbg(void)
705 bff58270 2022-09-19 op {
706 bff58270 2022-09-19 op 	if (nodebug)
707 bff58270 2022-09-19 op 		return;
708 bff58270 2022-09-19 op 
709 9a90a3d1 2022-09-29 op 	if (yylval.lineno == lastline + 1) {
710 9a90a3d1 2022-09-29 op 		lastline = yylval.lineno;
711 9a90a3d1 2022-09-29 op 		return;
712 9a90a3d1 2022-09-29 op 	}
713 9a90a3d1 2022-09-29 op 	lastline = yylval.lineno;
714 9a90a3d1 2022-09-29 op 
715 14c31a43 2022-11-25 op 	fprintf(fp, "#line %d ", yylval.lineno);
716 bff58270 2022-09-19 op 	printq(file->name);
717 14c31a43 2022-11-25 op 	putc('\n', fp);
718 bff58270 2022-09-19 op }
719 bff58270 2022-09-19 op 
720 bff58270 2022-09-19 op void
721 bff58270 2022-09-19 op printq(const char *str)
722 bff58270 2022-09-19 op {
723 14c31a43 2022-11-25 op 	putc('"', fp);
724 bff58270 2022-09-19 op 	for (; *str; ++str) {
725 bff58270 2022-09-19 op 		if (*str == '"')
726 14c31a43 2022-11-25 op 			putc('\\', fp);
727 14c31a43 2022-11-25 op 		putc(*str, fp);
728 bff58270 2022-09-19 op 	}
729 14c31a43 2022-11-25 op 	putc('"', fp);
730 bff58270 2022-09-19 op }