2bbffdc13c1aeafe0224014060b23043eca47fb1
[cubemap] / util.cpp
1 #include <errno.h>
2 #include <stddef.h>
3 #include <stdio.h>
4 #include <stdlib.h>
5 #include <unistd.h>
6 #include <sys/stat.h>
7 #include <sys/types.h>
8
9 #include "log.h"
10 #include "util.h"
11
12 using namespace std;
13
14 int make_tempfile(const std::string &contents)
15 {
16         char filename[] = "/tmp/cubemap.XXXXXX";
17         mode_t old_umask = umask(0600);
18         int fd = mkstemp(filename);
19         umask(old_umask);
20         if (fd == -1) {
21                 log_perror("mkstemp");
22                 return -1;
23         }
24
25         if (unlink(filename) == -1) {
26                 log_perror("unlink");
27                 // Can still continue;
28         }
29
30         const char *ptr = contents.data();
31         size_t to_write = contents.size();
32         while (to_write > 0) {
33                 ssize_t ret = write(fd, ptr, to_write);
34                 if (ret == -1) {
35                         log_perror("write");
36                         safe_close(fd);
37                         return -1;
38                 }
39
40                 ptr += ret;
41                 to_write -= ret;
42         }
43
44         return fd;
45 }
46
47 bool read_tempfile_and_close(int fd, std::string *contents)
48 {
49         bool ok = read_tempfile(fd, contents);
50         safe_close(fd);  // Implicitly deletes the file.
51         return ok;
52 }
53
54 bool read_tempfile(int fd, std::string *contents)
55 {
56         ssize_t ret, has_read;
57
58         off_t len = lseek(fd, 0, SEEK_END);
59         if (len == -1) {
60                 log_perror("lseek");
61                 return false;
62         }
63
64         contents->resize(len);
65
66         if (lseek(fd, 0, SEEK_SET) == -1) {
67                 log_perror("lseek");
68                 return false;
69         }
70
71         has_read = 0;
72         while (has_read < len) {
73                 ret = read(fd, &((*contents)[has_read]), len - has_read);
74                 if (ret == -1) {
75                         log_perror("read");
76                         return false;
77                 }
78                 if (ret == 0) {
79                         log(ERROR, "Unexpected EOF!");
80                         return false;
81                 }
82                 has_read += ret;
83         }
84
85         return true;
86 }
87
88 int safe_close(int fd)
89 {
90         int ret;
91         do {
92                 ret = close(fd);
93         } while (ret == -1 && errno == EINTR);
94
95         if (ret == -1) {
96                 log_perror("close()");
97         }
98
99         return ret;
100 }