]> git.cameronkatri.com Git - bsdgames-darwin.git/blob - atc/main.c
Input validation, from David A. Holland on #NetBSD-code
[bsdgames-darwin.git] / atc / main.c
1 /* $NetBSD: main.c,v 1.17 2006/06/07 09:35:03 jnemeth Exp $ */
2
3 /*-
4 * Copyright (c) 1990, 1993
5 * The Regents of the University of California. All rights reserved.
6 *
7 * This code is derived from software contributed to Berkeley by
8 * Ed James.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 * 3. Neither the name of the University nor the names of its contributors
19 * may be used to endorse or promote products derived from this software
20 * without specific prior written permission.
21 *
22 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * SUCH DAMAGE.
33 */
34
35 /*
36 * Copyright (c) 1987 by Ed James, UC Berkeley. All rights reserved.
37 *
38 * Copy permission is hereby granted provided that this notice is
39 * retained on all partial or complete copies.
40 *
41 * For more info on this and all of my stuff, mail edjames@berkeley.edu.
42 */
43
44 #include <sys/cdefs.h>
45 #ifndef lint
46 __COPYRIGHT("@(#) Copyright (c) 1990, 1993\n\
47 The Regents of the University of California. All rights reserved.\n");
48 #endif /* not lint */
49
50 #ifndef lint
51 #if 0
52 static char sccsid[] = "@(#)main.c 8.1 (Berkeley) 5/31/93";
53 #else
54 __RCSID("$NetBSD: main.c,v 1.17 2006/06/07 09:35:03 jnemeth Exp $");
55 #endif
56 #endif /* not lint */
57
58 #include "include.h"
59 #include "pathnames.h"
60
61 extern FILE *yyin;
62
63 int
64 main(int argc, char *argv[])
65 {
66 int seed;
67 int f_usage = 0, f_list = 0, f_showscore = 0;
68 int f_printpath = 0;
69 const char *file = NULL;
70 int ch;
71 struct sigaction sa;
72 #ifdef BSD
73 struct itimerval itv;
74 #endif
75
76 /* Open the score file then revoke setgid privileges */
77 open_score_file();
78 (void)setgid(getgid());
79
80 start_time = seed = time(NULL);
81
82 while ((ch = getopt(argc, argv, ":u?lstpg:f:r:")) != -1) {
83 switch (ch) {
84 case '?':
85 case 'u':
86 default:
87 f_usage++;
88 break;
89 case 'l':
90 f_list++;
91 break;
92 case 's':
93 case 't':
94 f_showscore++;
95 break;
96 case 'p':
97 f_printpath++;
98 break;
99 case 'r':
100 seed = atoi(optarg);
101 break;
102 case 'f':
103 case 'g':
104 file = optarg;
105 break;
106 }
107 }
108 if (optind < argc)
109 f_usage++;
110 srandom((unsigned long)seed);
111
112 if (f_usage)
113 (void)fprintf(stderr,
114 "Usage: %s -[u?lstp] [-[gf] game_name] [-r random seed]\n",
115 argv[0]);
116 if (f_showscore)
117 (void)log_score(1);
118 if (f_list)
119 (void)list_games();
120 if (f_printpath) {
121 char buf[100];
122
123 (void)strlcpy(buf, _PATH_GAMES, 100);
124 (void)puts(buf);
125 }
126
127 if (f_usage || f_showscore || f_list || f_printpath)
128 exit(0);
129
130 if (file == NULL)
131 file = default_game();
132 else
133 file = okay_game(file);
134
135 if (file == NULL || read_file(file) < 0)
136 exit(1);
137
138 init_gr();
139 setup_screen(sp);
140
141 (void)addplane();
142
143 (void)signal(SIGINT, quit);
144 (void)signal(SIGQUIT, quit);
145 #ifdef BSD
146 (void)signal(SIGTSTP, SIG_IGN);
147 #endif
148 (void)signal(SIGHUP, log_score_quit);
149 (void)signal(SIGTERM, log_score_quit);
150
151 (void)tcgetattr(fileno(stdin), &tty_start);
152 tty_new = tty_start;
153 tty_new.c_lflag &= ~(ICANON|ECHO);
154 tty_new.c_iflag |= ICRNL;
155 tty_new.c_cc[VMIN] = 1;
156 tty_new.c_cc[VTIME] = 0;
157 (void)tcsetattr(fileno(stdin), TCSADRAIN, &tty_new);
158
159 sa.sa_handler = update;
160 (void)sigemptyset(&sa.sa_mask);
161 (void)sigaddset(&sa.sa_mask, SIGALRM);
162 (void)sigaddset(&sa.sa_mask, SIGINT);
163 sa.sa_flags = 0;
164 (void)sigaction(SIGALRM, &sa, (struct sigaction *)0);
165
166 #ifdef BSD
167 itv.it_value.tv_sec = 0;
168 itv.it_value.tv_usec = 1;
169 itv.it_interval.tv_sec = sp->update_secs;
170 itv.it_interval.tv_usec = 0;
171 (void)setitimer(ITIMER_REAL, &itv, NULL);
172 #endif
173 #ifdef SYSV
174 alarm(sp->update_secs);
175 #endif
176
177 for (;;) {
178 if (getcommand() != 1)
179 planewin();
180 else {
181 #ifdef BSD
182 itv.it_value.tv_sec = 0;
183 itv.it_value.tv_usec = 0;
184 (void)setitimer(ITIMER_REAL, &itv, NULL);
185 #endif
186 #ifdef SYSV
187 alarm(0);
188 #endif
189
190 update(0);
191
192 #ifdef BSD
193 itv.it_value.tv_sec = sp->update_secs;
194 itv.it_value.tv_usec = 0;
195 itv.it_interval.tv_sec = sp->update_secs;
196 itv.it_interval.tv_usec = 0;
197 (void)setitimer(ITIMER_REAL, &itv, NULL);
198 #endif
199 #ifdef SYSV
200 alarm(sp->update_secs);
201 #endif
202 }
203 }
204 }
205
206 int
207 read_file(const char *s)
208 {
209 int retval;
210
211 filename = s;
212 yyin = fopen(s, "r");
213 if (yyin == NULL) {
214 warn("fopen %s", s);
215 return (-1);
216 }
217 retval = yyparse();
218 (void)fclose(yyin);
219
220 if (retval != 0)
221 return (-1);
222 else
223 return (0);
224 }
225
226 const char *
227 default_game(void)
228 {
229 FILE *fp;
230 static char file[256];
231 char line[256], games[256];
232
233 (void)strlcpy(games, _PATH_GAMES, 256);
234 (void)strlcat(games, GAMES, 256);
235
236 if ((fp = fopen(games, "r")) == NULL) {
237 warn("fopen %s", games);
238 return (NULL);
239 }
240 if (fgets(line, sizeof(line), fp) == NULL) {
241 (void)fprintf(stderr, "%s: no default game available\n", games);
242 fclose(fp);
243 return (NULL);
244 }
245 (void)fclose(fp);
246 line[strlen(line) - 1] = '\0';
247 (void)strlcpy(file, _PATH_GAMES, 256);
248 (void)strlcat(file, line, 256);
249 return (file);
250 }
251
252 const char *
253 okay_game(const char *s)
254 {
255 FILE *fp;
256 static char file[256];
257 const char *ret = NULL;
258 char line[256], games[256];
259
260 (void)strlcpy(games, _PATH_GAMES, 256);
261 (void)strlcat(games, GAMES, 256);
262
263 if ((fp = fopen(games, "r")) == NULL) {
264 warn("fopen %s", games);
265 return (NULL);
266 }
267 while (fgets(line, sizeof(line), fp) != NULL) {
268 line[strlen(line) - 1] = '\0';
269 if (strcmp(s, line) == 0) {
270 (void)strlcpy(file, _PATH_GAMES, 256);
271 (void)strlcat(file, line, 256);
272 ret = file;
273 break;
274 }
275 }
276 (void)fclose(fp);
277 if (ret == NULL) {
278 test_mode = 1;
279 ret = s;
280 (void)fprintf(stderr, "%s: %s: game not found\n", games, s);
281 (void)fprintf(stderr, "Your score will not be logged.\n");
282 (void)sleep(2); /* give the guy time to read it */
283 }
284 return (ret);
285 }
286
287 int
288 list_games(void)
289 {
290 FILE *fp;
291 char line[256], games[256];
292 int num_games = 0;
293
294 (void)strlcpy(games, _PATH_GAMES, 256);
295 (void)strlcat(games, GAMES, 256);
296
297 if ((fp = fopen(games, "r")) == NULL) {
298 warn("fopen %s", games);
299 return (-1);
300 }
301 (void)puts("available games:");
302 while (fgets(line, sizeof(line), fp) != NULL) {
303 (void)printf(" %s", line);
304 num_games++;
305 }
306 (void)fclose(fp);
307 if (num_games == 0) {
308 (void)fprintf(stderr, "%s: no games available\n", games);
309 return (-1);
310 }
311 return (0);
312 }