xref: /openbsd-src/usr.bin/yacc/main.c (revision 99fd087599a8791921855f21bd7e36130f39aadc)
1 /* $OpenBSD: main.c,v 1.29 2017/05/25 20:11:03 tedu Exp $	 */
2 /* $NetBSD: main.c,v 1.5 1996/03/19 03:21:38 jtc Exp $	 */
3 
4 /*
5  * Copyright (c) 1989 The Regents of the University of California.
6  * All rights reserved.
7  *
8  * This code is derived from software contributed to Berkeley by
9  * Robert Paul Corbett.
10  *
11  * Redistribution and use in source and binary forms, with or without
12  * modification, are permitted provided that the following conditions
13  * are met:
14  * 1. Redistributions of source code must retain the above copyright
15  *    notice, this list of conditions and the following disclaimer.
16  * 2. Redistributions in binary form must reproduce the above copyright
17  *    notice, this list of conditions and the following disclaimer in the
18  *    documentation and/or other materials provided with the distribution.
19  * 3. Neither the name of the University nor the names of its contributors
20  *    may be used to endorse or promote products derived from this software
21  *    without specific prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33  * SUCH DAMAGE.
34  */
35 
36 #include <sys/types.h>
37 #include <fcntl.h>
38 #include <paths.h>
39 #include <signal.h>
40 #include <stdlib.h>
41 #include <unistd.h>
42 #include "defs.h"
43 
44 char dflag;
45 char lflag;
46 char rflag;
47 char tflag;
48 char vflag;
49 
50 char *symbol_prefix;
51 char *file_prefix = "y";
52 
53 int lineno;
54 int outline;
55 
56 int explicit_file_name;
57 
58 char *action_file_name;
59 char *code_file_name;
60 char *defines_file_name;
61 char *input_file_name = "";
62 char *output_file_name;
63 char *text_file_name;
64 char *union_file_name;
65 char *verbose_file_name;
66 
67 FILE *action_file;	/* a temp file, used to save actions associated    */
68 			/* with rules until the parser is written	   */
69 FILE *code_file;	/* y.code.c (used when the -r option is specified) */
70 FILE *defines_file;	/* y.tab.h					   */
71 FILE *input_file;	/* the input file				   */
72 FILE *output_file;	/* y.tab.c					   */
73 FILE *text_file;	/* a temp file, used to save text until all	   */
74 			/* symbols have been defined			   */
75 FILE *union_file;	/* a temp file, used to save the union		   */
76 			/* definition until all symbol have been	   */
77 			/* defined					   */
78 FILE *verbose_file;	/* y.output					   */
79 
80 int nitems;
81 int nrules;
82 int nsyms;
83 int ntokens;
84 int nvars;
85 
86 int start_symbol;
87 char **symbol_name;
88 short *symbol_value;
89 short *symbol_prec;
90 char *symbol_assoc;
91 
92 short *ritem;
93 short *rlhs;
94 short *rrhs;
95 short *rprec;
96 char *rassoc;
97 short **derives;
98 char *nullable;
99 
100 void onintr(int);
101 void set_signals(void);
102 void usage(void);
103 void getargs(int, char *[]);
104 void create_file_names(void);
105 void open_files(void);
106 
107 volatile sig_atomic_t sigdie;
108 
109 void
110 done(int k)
111 {
112 	if (action_file)
113 		unlink(action_file_name);
114 	if (text_file)
115 		unlink(text_file_name);
116 	if (union_file)
117 		unlink(union_file_name);
118 	if (sigdie)
119 		_exit(k);
120 	exit(k);
121 }
122 
123 
124 void
125 onintr(__unused int signo)
126 {
127 	sigdie = 1;
128 	done(1);
129 }
130 
131 
132 void
133 set_signals(void)
134 {
135 #ifdef SIGINT
136 	if (signal(SIGINT, SIG_IGN) != SIG_IGN)
137 		signal(SIGINT, onintr);
138 #endif
139 #ifdef SIGTERM
140 	if (signal(SIGTERM, SIG_IGN) != SIG_IGN)
141 		signal(SIGTERM, onintr);
142 #endif
143 #ifdef SIGHUP
144 	if (signal(SIGHUP, SIG_IGN) != SIG_IGN)
145 		signal(SIGHUP, onintr);
146 #endif
147 }
148 
149 
150 void
151 usage(void)
152 {
153 	fprintf(stderr, "usage: %s [-dlrtv] [-b file_prefix] [-o output_file] [-p symbol_prefix] file\n", __progname);
154 	exit(1);
155 }
156 
157 
158 void
159 getargs(int argc, char *argv[])
160 {
161 	int ch;
162 
163 	while ((ch = getopt(argc, argv, "b:dlo:p:rtv")) != -1) {
164 		switch (ch) {
165 		case 'b':
166 			file_prefix = optarg;
167 			break;
168 
169 		case 'd':
170 			dflag = 1;
171 			break;
172 
173 		case 'l':
174 			lflag = 1;
175 			break;
176 
177 		case 'o':
178 			output_file_name = optarg;
179 			explicit_file_name = 1;
180 			break;
181 
182 		case 'p':
183 			symbol_prefix = optarg;
184 			break;
185 
186 		case 'r':
187 			rflag = 1;
188 			break;
189 
190 		case 't':
191 			tflag = 1;
192 			break;
193 
194 		case 'v':
195 			vflag = 1;
196 			break;
197 
198 		default:
199 			usage();
200 		}
201 	}
202 	argc -= optind;
203 	argv += optind;
204 
205 	if (argc != 1)
206 		usage();
207 	if (strcmp(*argv, "-") == 0)
208 		input_file = stdin;
209 	else
210 		input_file_name = *argv;
211 }
212 
213 
214 void *
215 allocate(size_t n)
216 {
217 	void *v;
218 
219 	v = NULL;
220 	if (n) {
221 		v = calloc(1, n);
222 		if (!v)
223 			no_space();
224 	}
225 	return (v);
226 }
227 
228 #define TEMPNAME(s, c, d, l)	\
229 	(asprintf(&(s), "%.*s/yacc.%xXXXXXXXXXX", (int)(l), (d), (c)))
230 
231 void
232 create_file_names(void)
233 {
234 	size_t len;
235 	char *tmpdir;
236 
237 	tmpdir = _PATH_TMP;
238 
239 	len = strlen(tmpdir);
240 	if (tmpdir[len - 1] == '/')
241 		len--;
242 
243 	if (TEMPNAME(action_file_name, 'a', tmpdir, len) == -1 ||
244 	    TEMPNAME(text_file_name, 'r', tmpdir, len) == -1 ||
245 	    TEMPNAME(union_file_name, 'u', tmpdir, len) == -1)
246 		no_space();
247 
248 	if (output_file_name == NULL) {
249 		if (asprintf(&output_file_name, "%s%s", file_prefix, OUTPUT_SUFFIX)
250 		    == -1)
251 			no_space();
252 	}
253 	if (rflag) {
254 		if (asprintf(&code_file_name, "%s%s", file_prefix, CODE_SUFFIX) == -1)
255 			no_space();
256 	} else
257 		code_file_name = output_file_name;
258 
259 	if (dflag) {
260 		if (explicit_file_name) {
261 			char *suffix;
262 
263 			defines_file_name = strdup(output_file_name);
264 			if (defines_file_name == 0)
265 				no_space();
266 
267 			/* does the output_file_name have a known suffix */
268 			if ((suffix = strrchr(output_file_name, '.')) != 0 &&
269 			    (!strcmp(suffix, ".c") ||	/* good, old-fashioned C */
270 			     !strcmp(suffix, ".C") ||	/* C++, or C on Windows */
271 			     !strcmp(suffix, ".cc") ||	/* C++ */
272 			     !strcmp(suffix, ".cxx") ||	/* C++ */
273 			     !strcmp(suffix, ".cpp"))) {/* C++ (Windows) */
274 				strncpy(defines_file_name, output_file_name,
275 					suffix - output_file_name + 1);
276 				defines_file_name[suffix - output_file_name + 1] = 'h';
277 				defines_file_name[suffix - output_file_name + 2] = '\0';
278 			} else {
279 				fprintf(stderr, "%s: suffix of output file name %s"
280 				 " not recognized, no -d file generated.\n",
281 					__progname, output_file_name);
282 				dflag = 0;
283 				free(defines_file_name);
284 				defines_file_name = 0;
285 			}
286 		} else {
287 			if (asprintf(&defines_file_name, "%s%s", file_prefix,
288 				     DEFINES_SUFFIX) == -1)
289 				no_space();
290 		}
291 	}
292 	if (vflag) {
293 		if (asprintf(&verbose_file_name, "%s%s", file_prefix,
294 			     VERBOSE_SUFFIX) == -1)
295 			no_space();
296 	}
297 }
298 
299 
300 void
301 open_files(void)
302 {
303 	int fd;
304 
305 	create_file_names();
306 
307 	if (input_file == 0) {
308 		input_file = fopen(input_file_name, "r");
309 		if (input_file == 0)
310 			open_error(input_file_name);
311 	}
312 	fd = mkstemp(action_file_name);
313 	if (fd == -1 || (action_file = fdopen(fd, "w")) == NULL)
314 		open_error(action_file_name);
315 
316 	fd = mkstemp(text_file_name);
317 	if (fd == -1 || (text_file = fdopen(fd, "w")) == NULL)
318 		open_error(text_file_name);
319 
320 	if (vflag) {
321 		verbose_file = fopen(verbose_file_name, "w");
322 		if (verbose_file == 0)
323 			open_error(verbose_file_name);
324 	}
325 	if (dflag) {
326 		defines_file = fopen(defines_file_name, "w");
327 		if (defines_file == NULL)
328 			open_write_error(defines_file_name);
329 		fd = mkstemp(union_file_name);
330 		if (fd == -1 || (union_file = fdopen(fd, "w")) == NULL)
331 			open_error(union_file_name);
332 	}
333 	output_file = fopen(output_file_name, "w");
334 	if (output_file == 0)
335 		open_error(output_file_name);
336 
337 	if (rflag) {
338 		code_file = fopen(code_file_name, "w");
339 		if (code_file == 0)
340 			open_error(code_file_name);
341 	} else
342 		code_file = output_file;
343 }
344 
345 
346 int
347 main(int argc, char *argv[])
348 {
349 	if (pledge("stdio rpath wpath cpath", NULL) == -1)
350 		fatal("pledge: invalid arguments");
351 
352 	set_signals();
353 	getargs(argc, argv);
354 	open_files();
355 	reader();
356 	lr0();
357 	lalr();
358 	make_parser();
359 	verbose();
360 	output();
361 	done(0);
362 	/* NOTREACHED */
363 	return (0);
364 }
365