Use setgid(), not setregid().

[bsdgames-darwin.git] / worm / worm.c

diff --git a/worm/worm.c b/worm/worm.c
index 1d7e187f189b1253050348562e036f959ac3d422..a78f27fb9b055f46e3cf216b1a88cf4ab49cba88 100644 (file)
--- a/worm/worm.c
+++ b/worm/worm.c
@@ -1,4 +1,4 @@
-/*     $NetBSD: worm.c,v 1.10 1999/07/28 23:09:45 hubertf Exp $        */
+/*     $NetBSD: worm.c,v 1.19 2000/05/08 07:56:06 mycroft Exp $        */
 
 /*
  * Copyright (c) 1980, 1993
@@ -43,7 +43,7 @@ __COPYRIGHT("@(#) Copyright (c) 1980, 1993\n\
 #if 0
 static char sccsid[] = "@(#)worm.c     8.1 (Berkeley) 5/31/93";
 #else
-__RCSID("$NetBSD: worm.c,v 1.10 1999/07/28 23:09:45 hubertf Exp $");
+__RCSID("$NetBSD: worm.c,v 1.19 2000/05/08 07:56:06 mycroft Exp $");
 #endif
 #endif /* not lint */
 
@@ -54,6 +54,7 @@ __RCSID("$NetBSD: worm.c,v 1.10 1999/07/28 23:09:45 hubertf Exp $");
 
 #include <ctype.h>
 #include <curses.h>
+#include <err.h>
 #include <signal.h>
 #include <stdlib.h>
 #include <termios.h>
@@ -79,20 +80,19 @@ int running = 0;
 int slow = 0;
 int score = 0;
 int start_len = LENGTH;
-char lastch;
+int lastch;
 char outbuf[BUFSIZ];
 
 void   crash __P((void)) __attribute__((__noreturn__));
-void   display __P((struct body *, char));
+void   display __P((const struct body *, char));
 int    main __P((int, char **));
 void   leave __P((int)) __attribute__((__noreturn__));
 void   life __P((void));
 void   newpos __P((struct body *));
-void   process __P((char));
+void   process __P((int));
 void   prize __P((void));
 int    rnd __P((int));
 void   setup __P((void));
-void   suspend __P((int));
 void   wake __P((int));
 
 int
@@ -100,7 +100,9 @@ main(argc, argv)
        int argc;
        char **argv;
 {
-       char ch;
+
+       /* Revoke setgid privileges */
+       setgid(getgid());
 
        if (argc == 2)
                start_len = atoi(argv[1]);
@@ -111,10 +113,12 @@ main(argc, argv)
        signal(SIGALRM, wake);
        signal(SIGINT, leave);
        signal(SIGQUIT, leave);
-       signal(SIGTSTP, suspend);       /* process control signal */
        initscr();
        crmode();
        noecho();
+#ifdef KEY_LEFT
+       keypad(stdscr, TRUE);
+#endif
        slow = (baudrate() <= 1200);
        clear();
        stw = newwin(1, COLS-1, 0, 0);
@@ -139,8 +143,7 @@ main(argc, argv)
                else
                {
                    fflush(stdout);
-                   if (read(0, &ch, 1) >= 0)
-                       process(ch);
+                   process(getch());
                }
        }
 }
@@ -153,12 +156,16 @@ life()
 
        np = NULL;
        head = newlink();
+       if (head == NULL)
+               err(1, NULL);
        head->x = start_len+2;
        head->y = 12;
        head->next = NULL;
        display(head, HEAD);
        for (i = 0, bp = head; i < start_len; i++, bp = np) {
                np = newlink();
+               if (np == NULL)
+                       err(1, NULL);
                np->next = bp;
                bp->prev = np;
                np->x = bp->x - 1;
@@ -171,7 +178,7 @@ life()
 
 void
 display(pos, chr)
-       struct body *pos;
+       const struct body *pos;
        char chr;
 {
        wmove(tv, pos->y, pos->x);
@@ -183,12 +190,17 @@ leave(dummy)
        int dummy;
 {
        endwin();
+
+       if (dummy == 0){        /* called via crash() */
+               printf("\nWell, you ran into something and the game is over.\n");
+               printf("Your final score was %d\n\n", score);
+       }
        exit(0);
 }
 
 void
 wake(dummy)
-       int dummy;
+       int dummy __attribute__((__unused__));
 {
        signal(SIGALRM, wake);
        fflush(stdout);
@@ -226,7 +238,7 @@ prize()
 
 void
 process(ch)
-       char ch;
+       int ch;
 {
        int x,y;
        struct body *nh;
@@ -236,18 +248,42 @@ process(ch)
        y = head->y;
        switch(ch)
        {
-               case 'h': x--; break;
-               case 'j': y++; break;
-               case 'k': y--; break;
-               case 'l': x++; break;
+#ifdef KEY_LEFT
+               case KEY_LEFT:
+#endif
+               case 'h':
+                       x--; break;
+
+#ifdef KEY_DOWN
+               case KEY_DOWN:
+#endif
+               case 'j':
+                       y++; break;
+
+#ifdef KEY_UP
+               case KEY_UP:
+#endif
+               case 'k':
+                       y--; break;
+
+#ifdef KEY_RIGHT
+               case KEY_RIGHT:
+#endif
+               case 'l':
+                       x++; break;
+
                case 'H': x--; running = RUNLEN; ch = tolower(ch); break;
                case 'J': y++; running = RUNLEN/2; ch = tolower(ch); break;
                case 'K': y--; running = RUNLEN/2; ch = tolower(ch); break;
                case 'L': x++; running = RUNLEN; ch = tolower(ch); break;
                case '\f': setup(); return;
-               case CNTRL('Z'): suspend(0); return;
-               case CNTRL('C'): crash(); return;
-               case CNTRL('D'): crash(); return;
+
+               case ERR:
+               case CNTRL('C'):
+               case CNTRL('D'):
+                       crash();
+                       return;
+
                default: if (! running) alarm(1);
                           return;
        }
@@ -275,6 +311,8 @@ process(ch)
        }
        else if(ch != ' ') crash();
        nh = newlink();
+       if (nh == NULL)
+               err(1, NULL);
        nh->next = NULL;
        nh->prev = head;
        head->next = nh;
@@ -294,30 +332,9 @@ process(ch)
 void
 crash()
 {
-       sleep(2);
-       clear();
-       move(23, 0);
-       refresh();
-       printf("Well, you ran into something and the game is over.\n");
-       printf("Your final score was %d\n", score);
        leave(0);
 }
 
-void
-suspend(dummy)
-       int dummy;
-{
-       move(LINES-1, 0);
-       refresh();
-       endwin();
-       fflush(stdout);
-       kill(getpid(), SIGTSTP);
-       signal(SIGTSTP, suspend);
-       crmode();
-       noecho();
-       setup();
-}
-
 void
 setup()
 {