git.sesse.net Git - cubemap/blob - 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 }