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