X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=util.cpp;h=d6d39f29ada05dc4437886d61178d190df189d02;hp=41c13c6f2a3ed743a5345b3d0d41c37aefb6e5bc;hb=71fc5575037bead8b6e927a1fffd199e4fc4514b;hpb=d1ad2cf60ca694f742e2061a596bc27c7e58f5c8

diff --git a/util.cpp b/util.cpp
index 41c13c6..d6d39f2 100644
--- a/util.cpp
+++ b/util.cpp
@@ -3,6 +3,8 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <unistd.h>
+#include <sys/stat.h>
+#include <sys/types.h>
 
 #include "log.h"
 #include "util.h"
@@ -12,14 +14,16 @@ using namespace std;
 int make_tempfile(const std::string &contents)
 {
 	char filename[] = "/tmp/cubemap.XXXXXX";
+	mode_t old_umask = umask(0600);
 	int fd = mkstemp(filename);
+	umask(old_umask);
 	if (fd == -1) {
-		perror("mkstemp");
+		log_perror("mkstemp");
 		return -1;
 	}
 
 	if (unlink(filename) == -1) {
-		perror("unlink");
+		log_perror("unlink");
 		// Can still continue;
 	}
 
@@ -28,8 +32,10 @@ int make_tempfile(const std::string &contents)
 	while (to_write > 0) {
 		ssize_t ret = write(fd, ptr, to_write);
 		if (ret == -1) {
-			perror("write");
-			close(fd);
+			log_perror("write");
+			if (close(fd) == -1) {
+				log_perror("close");
+			}
 			return -1;
 		}
 
@@ -40,51 +46,53 @@ int make_tempfile(const std::string &contents)
 	return fd;
 }
 
+bool read_tempfile_and_close(int fd, std::string *contents)
+{
+	bool ok = read_tempfile(fd, contents);
+
+	int ret;
+	do {
+		ret = close(fd);  // Implicitly deletes the file.
+	} while (ret == -1 && errno == EINTR);
+	
+	if (ret == -1) {
+		log_perror("close");
+		// Can still continue.
+	}
+
+	return ok;
+}
+
 bool read_tempfile(int fd, std::string *contents)
 {
-	bool ok = true;
 	ssize_t ret, has_read;
 
 	off_t len = lseek(fd, 0, SEEK_END);
 	if (len == -1) {
-		perror("lseek");
-		ok = false;
-		goto done;
+		log_perror("lseek");
+		return false;
 	}
 
 	contents->resize(len);
 
 	if (lseek(fd, 0, SEEK_SET) == -1) {
-		perror("lseek");
-		ok = false;
-		goto done;
+		log_perror("lseek");
+		return false;
 	}
 
 	has_read = 0;
 	while (has_read < len) {
 		ret = read(fd, &((*contents)[has_read]), len - has_read);
 		if (ret == -1) {
-			perror("read");
-			ok = false;
-			goto done;
+			log_perror("read");
+			return false;
 		}
 		if (ret == 0) {
 			log(ERROR, "Unexpected EOF!");
-			ok = false;
-			goto done;
+			return false;
 		}
 		has_read += ret;
 	}
 
-done:
-	do {
-		ret = close(fd);  // Implicitly deletes the files.
-	} while (ret == -1 && errno == EINTR);
-	
-	if (ret == -1) {
-		perror("close");
-		// Can still continue.
-	}
-
-	return ok;
+	return true;
 }