xref: /dragonfly/games/worms/worms.c (revision 0db87cb7) 
1 /*-
2  * Copyright (c) 1980, 1993
3  *	The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. Neither the name of the University nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  *
29  * @(#) Copyright (c) 1980, 1993 The Regents of the University of California.  All rights reserved.
30  * @(#)worms.c	8.1 (Berkeley) 5/31/93
31  * $FreeBSD: src/games/worms/worms.c,v 1.8.2.1 2000/12/04 10:36:17 alex Exp $
32  * $DragonFly: src/games/worms/worms.c,v 1.5 2007/04/18 18:32:12 swildner Exp $
33  */
34 
35 /*
36  *
37  *	 @@@        @@@    @@@@@@@@@@     @@@@@@@@@@@    @@@@@@@@@@@@
38  *	 @@@        @@@   @@@@@@@@@@@@    @@@@@@@@@@@@   @@@@@@@@@@@@@
39  *	 @@@        @@@  @@@@      @@@@   @@@@           @@@@ @@@  @@@@
40  *	 @@@   @@   @@@  @@@        @@@   @@@            @@@  @@@   @@@
41  *	 @@@  @@@@  @@@  @@@        @@@   @@@            @@@  @@@   @@@
42  *	 @@@@ @@@@ @@@@  @@@        @@@   @@@            @@@  @@@   @@@
43  *	  @@@@@@@@@@@@   @@@@      @@@@   @@@            @@@  @@@   @@@
44  *	   @@@@  @@@@     @@@@@@@@@@@@    @@@            @@@  @@@   @@@
45  *	    @@    @@       @@@@@@@@@@     @@@            @@@  @@@   @@@
46  *
47  *				 Eric P. Scott
48  *			  Caltech High Energy Physics
49  *				 October, 1980
50  *
51  */
52 #include <sys/types.h>
53 #include <curses.h>
54 #include <err.h>
55 #include <signal.h>
56 #include <stdio.h>
57 #include <stdlib.h>
58 #include <unistd.h>
59 
60 static const struct options {
61 	int nopts;
62 	int opts[3];
63 }
64 	normal[8] = {
65 	{ 3, { 7, 0, 1 } },
66 	{ 3, { 0, 1, 2 } },
67 	{ 3, { 1, 2, 3 } },
68 	{ 3, { 2, 3, 4 } },
69 	{ 3, { 3, 4, 5 } },
70 	{ 3, { 4, 5, 6 } },
71 	{ 3, { 5, 6, 7 } },
72 	{ 3, { 6, 7, 0 } }
73 },	upper[8] = {
74 	{ 1, { 1, 0, 0 } },
75 	{ 2, { 1, 2, 0 } },
76 	{ 0, { 0, 0, 0 } },
77 	{ 0, { 0, 0, 0 } },
78 	{ 0, { 0, 0, 0 } },
79 	{ 2, { 4, 5, 0 } },
80 	{ 1, { 5, 0, 0 } },
81 	{ 2, { 1, 5, 0 } }
82 },
83 	left[8] = {
84 	{ 0, { 0, 0, 0 } },
85 	{ 0, { 0, 0, 0 } },
86 	{ 0, { 0, 0, 0 } },
87 	{ 2, { 2, 3, 0 } },
88 	{ 1, { 3, 0, 0 } },
89 	{ 2, { 3, 7, 0 } },
90 	{ 1, { 7, 0, 0 } },
91 	{ 2, { 7, 0, 0 } }
92 },
93 	right[8] = {
94 	{ 1, { 7, 0, 0 } },
95 	{ 2, { 3, 7, 0 } },
96 	{ 1, { 3, 0, 0 } },
97 	{ 2, { 3, 4, 0 } },
98 	{ 0, { 0, 0, 0 } },
99 	{ 0, { 0, 0, 0 } },
100 	{ 0, { 0, 0, 0 } },
101 	{ 2, { 6, 7, 0 } }
102 },
103 	lower[8] = {
104 	{ 0, { 0, 0, 0 } },
105 	{ 2, { 0, 1, 0 } },
106 	{ 1, { 1, 0, 0 } },
107 	{ 2, { 1, 5, 0 } },
108 	{ 1, { 5, 0, 0 } },
109 	{ 2, { 5, 6, 0 } },
110 	{ 0, { 0, 0, 0 } },
111 	{ 0, { 0, 0, 0 } }
112 },
113 	upleft[8] = {
114 	{ 0, { 0, 0, 0 } },
115 	{ 0, { 0, 0, 0 } },
116 	{ 0, { 0, 0, 0 } },
117 	{ 0, { 0, 0, 0 } },
118 	{ 0, { 0, 0, 0 } },
119 	{ 1, { 3, 0, 0 } },
120 	{ 2, { 1, 3, 0 } },
121 	{ 1, { 1, 0, 0 } }
122 },
123 	upright[8] = {
124 	{ 2, { 3, 5, 0 } },
125 	{ 1, { 3, 0, 0 } },
126 	{ 0, { 0, 0, 0 } },
127 	{ 0, { 0, 0, 0 } },
128 	{ 0, { 0, 0, 0 } },
129 	{ 0, { 0, 0, 0 } },
130 	{ 0, { 0, 0, 0 } },
131 	{ 1, { 5, 0, 0 } }
132 },
133 	lowleft[8] = {
134 	{ 3, { 7, 0, 1 } },
135 	{ 0, { 0, 0, 0 } },
136 	{ 0, { 0, 0, 0 } },
137 	{ 1, { 1, 0, 0 } },
138 	{ 2, { 1, 7, 0 } },
139 	{ 1, { 7, 0, 0 } },
140 	{ 0, { 0, 0, 0 } },
141 	{ 0, { 0, 0, 0 } }
142 },
143 	lowright[8] = {
144 	{ 0, { 0, 0, 0 } },
145 	{ 1, { 7, 0, 0 } },
146 	{ 2, { 5, 7, 0 } },
147 	{ 1, { 5, 0, 0 } },
148 	{ 0, { 0, 0, 0 } },
149 	{ 0, { 0, 0, 0 } },
150 	{ 0, { 0, 0, 0 } },
151 	{ 0, { 0, 0, 0 } }
152 };
153 
154 
155 static const char	flavor[] = {
156 	'O', '*', '#', '$', '%', '0', '@', '~'
157 };
158 static const short	xinc[] = {
159 	1,  1,  1,  0, -1, -1, -1,  0
160 }, yinc[] = {
161 	-1,  0,  1,  1,  1,  0, -1, -1
162 };
163 static struct	worm {
164 	int orientation, head;
165 	short *xpos, *ypos;
166 } *worm;
167 
168 volatile sig_atomic_t sig_caught = 0;
169 
170 void nomem(void);
171 void onsig(int);
172 
173 int
174 main(int argc, char **argv)
175 {
176 	int x, y, h, n;
177 	struct worm *w;
178 	const struct options *op;
179 	short *ip;
180 	int CO, LI, last, bottom, ch, length, number, trail;
181 	short **ref;
182 	const char *field;
183 	char *mp;
184 	unsigned int delay = 0;
185 
186 	mp = NULL;
187 	length = 16;
188 	number = 3;
189 	trail = ' ';
190 	field = NULL;
191 	while ((ch = getopt(argc, argv, "d:fl:n:t")) != -1)
192 		switch(ch) {
193 		case 'd':
194 			if ((delay = (unsigned int)strtoul(optarg, NULL, 10)) < 1 || delay > 1000)
195 				errx(1, "invalid delay (1-1000)");
196 			delay *= 1000;  /* ms -> us */
197 			break;
198 		case 'f':
199 			field = "WORM";
200 			break;
201 		case 'l':
202 			if ((length = atoi(optarg)) < 2 || length > 1024) {
203 				errx(1, "invalid length (%d - %d).",
204 				     2, 1024);
205 			}
206 			break;
207 		case 'n':
208 			if ((number = atoi(optarg)) < 1) {
209 				errx(1, "invalid number of worms.");
210 			}
211 			break;
212 		case 't':
213 			trail = '.';
214 			break;
215 		case '?':
216 		default:
217 			(void)fprintf(stderr,
218 			    "usage: worms [-ft] [-d delay] [-l length] [-n number]\n");
219 			exit(1);
220 		}
221 
222 	if (!(worm = malloc((size_t)number *
223 	    sizeof(struct worm))) || !(mp = malloc((size_t)1024)))
224 		nomem();
225 	initscr();
226 	CO = COLS;
227 	LI = LINES;
228 	last = CO - 1;
229 	bottom = LI - 1;
230 	if (!(ip = malloc((size_t)(LI * CO * sizeof(short)))))
231 		nomem();
232 	if (!(ref = malloc((size_t)(LI * sizeof(short *)))))
233 		nomem();
234 	for (n = 0; n < LI; ++n) {
235 		ref[n] = ip;
236 		ip += CO;
237 	}
238 	for (ip = ref[0], n = LI * CO; --n >= 0;)
239 		*ip++ = 0;
240 	for (n = number, w = &worm[0]; --n >= 0; w++) {
241 		w->orientation = w->head = 0;
242 		if (!(ip = malloc((size_t)(length * sizeof(short)))))
243 			nomem();
244 		w->xpos = ip;
245 		for (x = length; --x >= 0;)
246 			*ip++ = -1;
247 		if (!(ip = malloc((size_t)(length * sizeof(short)))))
248 			nomem();
249 		w->ypos = ip;
250 		for (y = length; --y >= 0;)
251 			*ip++ = -1;
252 	}
253 
254 	(void)signal(SIGHUP, onsig);
255 	(void)signal(SIGINT, onsig);
256 	(void)signal(SIGQUIT, onsig);
257 	(void)signal(SIGSTOP, onsig);
258 	(void)signal(SIGTSTP, onsig);
259 	(void)signal(SIGTERM, onsig);
260 
261 	if (field) {
262 		const char *p = field;
263 
264 		for (y = LI; --y >= 0;) {
265 			for (x = CO; --x >= 0;) {
266 				addch(*p++);
267 				if (!*p)
268 					p = field;
269 			}
270 			refresh();
271 		}
272 	}
273 	for (;;) {
274 		refresh();
275 		if (sig_caught) {
276 			endwin();
277 			exit(0);
278 		}
279 		for (n = 0, w = &worm[0]; n < number; n++, w++) {
280 			if ((x = w->xpos[h = w->head]) < 0) {
281 				mvaddch(y = w->ypos[h] = bottom,
282 					x = w->xpos[h] = 0,
283 					flavor[n % sizeof(flavor)]);
284 				ref[y][x]++;
285 			}
286 			else
287 				y = w->ypos[h];
288 			if (++h == length)
289 				h = 0;
290 			if (w->xpos[w->head = h] >= 0) {
291 				int x1, y1;
292 
293 				x1 = w->xpos[h];
294 				y1 = w->ypos[h];
295 				if (--ref[y1][x1] == 0) {
296 					mvaddch(y1, x1, trail);
297 				}
298 			}
299 			op = &(!x ? (!y ? upleft : (y == bottom ? lowleft : left)) : (x == last ? (!y ? upright : (y == bottom ? lowright : right)) : (!y ? upper : (y == bottom ? lower : normal))))[w->orientation];
300 			switch (op->nopts) {
301 			case 0:
302 				refresh();
303 				abort();
304 				return(1);
305 			case 1:
306 				w->orientation = op->opts[0];
307 				break;
308 			default:
309 				w->orientation =
310 				    op->opts[(int)random() % op->nopts];
311 			}
312 			mvaddch(y += yinc[w->orientation],
313 				x += xinc[w->orientation],
314 				flavor[n % sizeof(flavor)]);
315 			ref[w->ypos[h] = y][w->xpos[h] = x]++;
316 		}
317 		if (usleep(delay))
318 			onsig(SIGTERM);
319 	}
320 }
321 
322 void
323 onsig(int signo __unused)
324 {
325 	sig_caught = 1;
326 }
327 
328 void
329 nomem(void)
330 {
331 	errx(1, "not enough memory.");
332 }
333