From 3fadb9c28981b6b4fb1c176b6ed61e7fe45645e2 Mon Sep 17 00:00:00 2001
From: ache <ache@FreeBSD.org>
Date: Wed, 24 Sep 1997 09:42:33 +0000
Subject: Revoke privs once at earlier stage and not mess with setegid in save
 Fix save: missing creation modes arg and wrong symlink test Small C cleanup

---
 games/battlestar/save.c | 16 +++-------------
 1 file changed, 3 insertions(+), 13 deletions(-)

(limited to 'games/battlestar/save.c')

diff --git a/games/battlestar/save.c b/games/battlestar/save.c
index 8b72506..df61213 100644
--- a/games/battlestar/save.c
+++ b/games/battlestar/save.c
@@ -39,12 +39,12 @@ static char sccsid[] = "@(#)save.c	8.1 (Berkeley) 5/31/93";
 #include <sys/stat.h>
 #include <sys/param.h>			/* MAXPATHLEN */
 #include <fcntl.h>
+#include <stdlib.h>
 #include "externs.h"
 
 void
 restore()
 {
-	char *getenv();
 	char *home;
 	char home1[MAXPATHLEN];
 	register int n;
@@ -55,13 +55,10 @@ restore()
   	  sprintf(home1, "%.*s/Bstar", MAXPATHLEN - 7, home);
 	else return;
 
-	setegid(egid);
 	if ((fp = fopen(home1, "r")) == 0) {
 		perror(home1);
-		setegid(getgid());
 		return;
 	}
-	setegid(getgid());
 	fread(&WEIGHT, sizeof WEIGHT, 1, fp);
 	fread(&CUMBER, sizeof CUMBER, 1, fp);
 	fread(&gclock, sizeof gclock, 1, fp);
@@ -103,7 +100,6 @@ void
 save()
 {
 	struct stat sbuf;
-	char *getenv();
 	char *home;
 	char home1[MAXPATHLEN];
 	register int n;
@@ -115,36 +111,30 @@ save()
 		return;
 	sprintf(home1, "%.*s/Bstar", MAXPATHLEN - 7, home);
 
-	setegid(egid);
 	/* Try to open the file safely. */
 	if (stat(home1, &sbuf) < 0) {	  	
-		fd = open(home1, O_WRONLY|O_CREAT|O_EXCL);
+		fd = open(home1, O_WRONLY|O_CREAT|O_EXCL, 0600);
 	        if (fd < 0) {
           		fprintf(stderr, "Can't create %s\n", home1);
-				setegid(getgid());
            		return;
 	        }
 	} else {
-		if (sbuf.st_mode & S_IFLNK) {
+		if ((sbuf.st_mode & S_IFLNK) == S_IFLNK) {
 			fprintf(stderr, "No symlinks!\n");
-			setegid(getgid());
 			return;
 		}
 
 		fd = open(home1, O_WRONLY|O_EXCL);
 		if (fd < 0) {
 			fprintf(stderr, "Can't open %s for writing\n", home1);
-			setegid(getgid());
 			return;
 		}
 	}
 
 	if ((fp = fdopen(fd, "w")) == 0) {
 		perror(home1);
-		setegid(getgid());
 		return;
 	}
-	setegid(getgid());
 
 	printf("Saved in %s.\n", home1);
 	fwrite(&WEIGHT, sizeof WEIGHT, 1, fp);
-- 
cgit v1.1