ioutil.c

   1 #include <stdint.h>
   2 #include <string.h>
   3 #include "types.h"
   4 #include "ioutil.h"
   5
   6 #ifndef _WIN32
   7
   8 #include <errno.h>
   9 #include <sys/stat.h>
  10 #include <fcntl.h>
  11 #include <unistd.h>
  12
  13 int writeall(FH fd,const u8 *data,size_t len)
  14 {
  15         ssize_t wrote;
  16         while (len) {
  17                 wrote = write(fd,data,len);
  18                 if (wrote == -1) {
  19                         if (errno == EAGAIN || errno == EWOULDBLOCK || errno == EINTR)
  20                                 continue;
  21                         return -1;
  22                 }
  23                 len -= (size_t) wrote;
  24                 data += wrote;
  25         }
  26         return 0;
  27 }
  28
  29 FH createfile(const char *path,int secret)
  30 {
  31         int fd;
  32         do {
  33                 fd = open(path,O_WRONLY | O_CREAT | O_TRUNC,secret ? 0600 : 0666);
  34                 if (fd == -1) {
  35                         if (errno == EINTR)
  36                                 continue;
  37                         return -1;
  38                 }
  39         } while (0);
  40         return fd;
  41 }
  42
  43 int closefile(FH fd)
  44 {
  45         int cret;
  46         do {
  47                 cret = close(fd);
  48                 if (cret == -1) {
  49                         if (errno == EINTR)
  50                                 continue;
  51                         return -1;
  52                 }
  53         } while (0);
  54         return 0;
  55 }
  56
  57 int createdir(const char *path,int secret)
  58 {
  59         return mkdir(path,secret ? 0700 : 0777);
  60 }
  61
  62 #else
  63
  64 int writeall(FH fd,const u8 *data,size_t len)
  65 {
  66         DWORD wrote;
  67         BOOL success;
  68         while (len) {
  69                 success = WriteFile(fd,data,
  70                         len <= (DWORD)-1 ? (DWORD)len : (DWORD)-1,&wrote,0);
  71                 if (!success)
  72                         return -1;
  73                 data += wrote;
  74                 if (len >= wrote)
  75                         len -= wrote;
  76                 else
  77                         len = 0;
  78         }
  79         return 0;
  80 }
  81
  82 FH createfile(const char *path,int secret)
  83 {
  84         // XXX no support for non-ascii chars
  85         // XXX don't know how to handle secret argument
  86         (void) secret;
  87         return CreateFileA(path,GENERIC_WRITE,0,0,CREATE_ALWAYS,0,0);
  88 }
  89
  90 int closefile(FH fd)
  91 {
  92         return CloseHandle(fd) ? 0 : -1;
  93 }
  94
  95 int createdir(const char *path,int secret)
  96 {
  97         // XXX don't know how to handle secret argument
  98         return CreateDirectoryA(path,0) ? 0 : -1;
  99 }
 100
 101 #endif
 102
 103 int writetofile(const char *path,const u8 *data,size_t len,int secret)
 104 {
 105         FH fd = createfile(path,secret);
 106         int wret = writeall(fd,data,len);
 107         int cret = closefile(fd);
 108         if (cret == -1)
 109                 return -1;
 110         return wret;
 111 }