X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=util.cpp;h=5659bed0802bd1062644f79cc49b80234a9698da;hp=2bbffdc13c1aeafe0224014060b23043eca47fb1;hb=8a2935dc58b00839acc493b4d146042d22a9793c;hpb=0cb56be70f7ca4f4564eea892a99d20032359a1d

diff --git a/util.cpp b/util.cpp
index 2bbffdc..5659bed 100644
--- a/util.cpp
+++ b/util.cpp
@@ -2,29 +2,37 @@
 #include <stddef.h>
 #include <stdio.h>
 #include <stdlib.h>
-#include <unistd.h>
+#include <fcntl.h>
 #include <sys/stat.h>
-#include <sys/types.h>
+#include <unistd.h>
 
 #include "log.h"
 #include "util.h"
 
+#ifndef O_TMPFILE
+#define __O_TMPFILE 020000000
+#define O_TMPFILE (__O_TMPFILE | O_DIRECTORY)
+#endif
+
 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);
+	int fd = open(filename, O_RDWR | O_TMPFILE, 0600);
 	if (fd == -1) {
-		log_perror("mkstemp");
-		return -1;
-	}
+		mode_t old_umask = umask(077);
+		fd = mkstemp(filename);
+		if (fd == -1) {
+			log_perror("mkstemp");
+			return -1;
+		}
+		umask(old_umask);
 
-	if (unlink(filename) == -1) {
-		log_perror("unlink");
-		// Can still continue;
+		if (unlink(filename) == -1) {
+			log_perror("unlink");
+			// Can still continue.
+		}
 	}
 
 	const char *ptr = contents.data();