other-licenses/7zstub/src/7zip/Common/FileStreams.cpp

   1 // FileStreams.cpp
   2
   3 #include "StdAfx.h"
   4
   5 #ifndef _WIN32
   6 #include <fcntl.h>
   7 #include <unistd.h>
   8 #include <errno.h>
   9 #endif
  10
  11 #include "FileStreams.h"
  12
  13 static inline HRESULT ConvertBoolToHRESULT(bool result)
  14 {
  15   // return result ? S_OK: E_FAIL;
  16   #ifdef _WIN32
  17   return result ? S_OK: (::GetLastError());
  18   #else
  19   return result ? S_OK: E_FAIL;
  20   #endif
  21 }
  22
  23 bool CInFileStream::Open(LPCTSTR fileName)
  24 {
  25   return File.Open(fileName);
  26 }
  27
  28 #ifdef _WIN32
  29 #ifndef _UNICODE
  30 bool CInFileStream::Open(LPCWSTR fileName)
  31 {
  32   return File.Open(fileName);
  33 }
  34 #endif
  35 #endif
  36
  37 STDMETHODIMP CInFileStream::Read(void *data, UInt32 size, UInt32 *processedSize)
  38 {
  39   #ifdef _WIN32
  40
  41   UInt32 realProcessedSize;
  42   bool result = File.ReadPart(data, size, realProcessedSize);
  43   if(processedSize != NULL)
  44     *processedSize = realProcessedSize;
  45   return ConvertBoolToHRESULT(result);
  46
  47   #else
  48
  49   if(processedSize != NULL)
  50     *processedSize = 0;
  51   ssize_t res = File.Read(data, (size_t)size);
  52   if (res == -1)
  53     return E_FAIL;
  54   if(processedSize != NULL)
  55     *processedSize = (UInt32)res;
  56   return S_OK;
  57
  58   #endif
  59 }
  60
  61 #ifndef _WIN32_WCE
  62 STDMETHODIMP CStdInFileStream::Read(void *data, UInt32 size, UInt32 *processedSize)
  63 {
  64   #ifdef _WIN32
  65   UInt32 realProcessedSize;
  66   BOOL res = ::ReadFile(GetStdHandle(STD_INPUT_HANDLE),
  67       data, size, (DWORD *)&realProcessedSize, NULL);
  68   if(processedSize != NULL)
  69     *processedSize = realProcessedSize;
  70   if (res == FALSE && GetLastError() == ERROR_BROKEN_PIPE)
  71     return S_OK;
  72   return ConvertBoolToHRESULT(res != FALSE);
  73
  74   #else
  75
  76   if(processedSize != NULL)
  77     *processedSize = 0;
  78   ssize_t res;
  79   do
  80   {
  81     res = read(0, data, (size_t)size);
  82   }
  83   while (res < 0 && (errno == EINTR));
  84   if (res == -1)
  85     return E_FAIL;
  86   if(processedSize != NULL)
  87     *processedSize = (UInt32)res;
  88   return S_OK;
  89
  90   #endif
  91 }
  92
  93 #endif
  94
  95 STDMETHODIMP CInFileStream::Seek(Int64 offset, UInt32 seekOrigin,
  96     UInt64 *newPosition)
  97 {
  98   if(seekOrigin >= 3)
  99     return STG_E_INVALIDFUNCTION;
 100
 101   #ifdef _WIN32
 102
 103   UInt64 realNewPosition;
 104   bool result = File.Seek(offset, seekOrigin, realNewPosition);
 105   if(newPosition != NULL)
 106     *newPosition = realNewPosition;
 107   return ConvertBoolToHRESULT(result);
 108
 109   #else
 110
 111   off_t res = File.Seek(offset, seekOrigin);
 112   if (res == -1)
 113     return E_FAIL;
 114   if(newPosition != NULL)
 115     *newPosition = (UInt64)res;
 116   return S_OK;
 117
 118   #endif
 119 }
 120
 121 STDMETHODIMP CInFileStream::GetSize(UInt64 *size)
 122 {
 123   return ConvertBoolToHRESULT(File.GetLength(*size));
 124 }
 125
 126
 127 //////////////////////////
 128 // COutFileStream
 129
 130 bool COutFileStream::Create(LPCTSTR fileName, bool createAlways)
 131 {
 132   return File.Create(fileName, createAlways);
 133 }
 134
 135 #ifdef _WIN32
 136 #ifndef _UNICODE
 137 bool COutFileStream::Create(LPCWSTR fileName, bool createAlways)
 138 {
 139   return File.Create(fileName, createAlways);
 140 }
 141 #endif
 142 #endif
 143
 144 STDMETHODIMP COutFileStream::Write(const void *data, UInt32 size, UInt32 *processedSize)
 145 {
 146   #ifdef _WIN32
 147
 148   UInt32 realProcessedSize;
 149   bool result = File.WritePart(data, size, realProcessedSize);
 150   if(processedSize != NULL)
 151     *processedSize = realProcessedSize;
 152   return ConvertBoolToHRESULT(result);
 153
 154   #else
 155
 156   if(processedSize != NULL)
 157     *processedSize = 0;
 158   ssize_t res = File.Write(data, (size_t)size);
 159   if (res == -1)
 160     return E_FAIL;
 161   if(processedSize != NULL)
 162     *processedSize = (UInt32)res;
 163   return S_OK;
 164
 165   #endif
 166 }
 167
 168 STDMETHODIMP COutFileStream::Seek(Int64 offset, UInt32 seekOrigin,
 169     UInt64 *newPosition)
 170 {
 171   if(seekOrigin >= 3)
 172     return STG_E_INVALIDFUNCTION;
 173   #ifdef _WIN32
 174
 175   UInt64 realNewPosition;
 176   bool result = File.Seek(offset, seekOrigin, realNewPosition);
 177   if(newPosition != NULL)
 178     *newPosition = realNewPosition;
 179   return ConvertBoolToHRESULT(result);
 180
 181   #else
 182
 183   off_t res = File.Seek(offset, seekOrigin);
 184   if (res == -1)
 185     return E_FAIL;
 186   if(newPosition != NULL)
 187     *newPosition = (UInt64)res;
 188   return S_OK;
 189
 190   #endif
 191 }
 192
 193 STDMETHODIMP COutFileStream::SetSize(Int64 newSize)
 194 {
 195   #ifdef _WIN32
 196   UInt64 currentPos;
 197   if(!File.Seek(0, FILE_CURRENT, currentPos))
 198     return E_FAIL;
 199   bool result = File.SetLength(newSize);
 200   UInt64 currentPos2;
 201   result = result && File.Seek(currentPos, currentPos2);
 202   return result ? S_OK : E_FAIL;
 203   #else
 204   return E_FAIL;
 205   #endif
 206 }
 207
 208 #ifndef _WIN32_WCE
 209 STDMETHODIMP CStdOutFileStream::Write(const void *data, UInt32 size, UInt32 *processedSize)
 210 {
 211   if(processedSize != NULL)
 212     *processedSize = 0;
 213
 214   #ifdef _WIN32
 215   UInt32 realProcessedSize;
 216   BOOL res = TRUE;
 217   if (size > 0)
 218   {
 219     // Seems that Windows doesn't like big amounts writing to stdout.
 220     // So we limit portions by 32KB.
 221     UInt32 sizeTemp = (1 << 15);
 222     if (sizeTemp > size)
 223       sizeTemp = size;
 224     res = ::WriteFile(GetStdHandle(STD_OUTPUT_HANDLE),
 225         data, sizeTemp, (DWORD *)&realProcessedSize, NULL);
 226     size -= realProcessedSize;
 227     data = (const void *)((const Byte *)data + realProcessedSize);
 228     if(processedSize != NULL)
 229       *processedSize += realProcessedSize;
 230   }
 231   return ConvertBoolToHRESULT(res != FALSE);
 232
 233   #else
 234
 235   ssize_t res;
 236   do
 237   {
 238     res = write(1, data, (size_t)size);
 239   }
 240   while (res < 0 && (errno == EINTR));
 241   if (res == -1)
 242     return E_FAIL;
 243   if(processedSize != NULL)
 244     *processedSize = (UInt32)res;
 245   return S_OK;
 246
 247   return S_OK;
 248   #endif
 249 }
 250
 251 #endif