tor-browser

7zFile.c (7121B)

---

/* 7zFile.c -- File IO
2017-04-03 : Igor Pavlov : Public domain */

#include "Precomp.h"

#include "7zFile.h"

#ifndef USE_WINDOWS_FILE

#ifndef UNDER_CE
#include <errno.h>
#endif

#else

/*
  ReadFile and WriteFile functions in Windows have BUG:
  If you Read or Write 64MB or more (probably min_failure_size = 64MB - 32KB + 1)
  from/to Network file, it returns ERROR_NO_SYSTEM_RESOURCES
  (Insufficient system resources exist to complete the requested service).
  Probably in some version of Windows there are problems with other sizes:
  for 32 MB (maybe also for 16 MB).
  And message can be "Network connection was lost"
*/

#define kChunkSizeMax (1 << 22)

#endif

void File_Construct(CSzFile *p)
{
 #ifdef USE_WINDOWS_FILE
 p->handle = INVALID_HANDLE_VALUE;
 #else
 p->file = NULL;
 #endif
}

#if !defined(UNDER_CE) || !defined(USE_WINDOWS_FILE)
static WRes File_Open(CSzFile *p, const char *name, int writeMode)
{
 #ifdef USE_WINDOWS_FILE
 p->handle = CreateFileA(name,
     writeMode ? GENERIC_WRITE : GENERIC_READ,
     FILE_SHARE_READ, NULL,
     writeMode ? CREATE_ALWAYS : OPEN_EXISTING,
     FILE_ATTRIBUTE_NORMAL, NULL);
 return (p->handle != INVALID_HANDLE_VALUE) ? 0 : GetLastError();
 #else
 p->file = fopen(name, writeMode ? "wb+" : "rb");
 return (p->file != 0) ? 0 :
   #ifdef UNDER_CE
   2; /* ENOENT */
   #else
   errno;
   #endif
 #endif
}

WRes InFile_Open(CSzFile *p, const char *name) { return File_Open(p, name, 0); }
WRes OutFile_Open(CSzFile *p, const char *name) { return File_Open(p, name, 1); }
#endif

#ifdef USE_WINDOWS_FILE
static WRes File_OpenW(CSzFile *p, const WCHAR *name, int writeMode)
{
 p->handle = CreateFileW(name,
     writeMode ? GENERIC_WRITE : GENERIC_READ,
     FILE_SHARE_READ, NULL,
     writeMode ? CREATE_ALWAYS : OPEN_EXISTING,
     FILE_ATTRIBUTE_NORMAL, NULL);
 return (p->handle != INVALID_HANDLE_VALUE) ? 0 : GetLastError();
}
WRes InFile_OpenW(CSzFile *p, const WCHAR *name) { return File_OpenW(p, name, 0); }
WRes OutFile_OpenW(CSzFile *p, const WCHAR *name) { return File_OpenW(p, name, 1); }
#endif

WRes File_Close(CSzFile *p)
{
 #ifdef USE_WINDOWS_FILE
 if (p->handle != INVALID_HANDLE_VALUE)
 {
   if (!CloseHandle(p->handle))
     return GetLastError();
   p->handle = INVALID_HANDLE_VALUE;
 }
 #else
 if (p->file != NULL)
 {
   int res = fclose(p->file);
   if (res != 0)
     return res;
   p->file = NULL;
 }
 #endif
 return 0;
}

WRes File_Read(CSzFile *p, void *data, size_t *size)
{
 size_t originalSize = *size;
 if (originalSize == 0)
   return 0;

 #ifdef USE_WINDOWS_FILE

 *size = 0;
 do
 {
   DWORD curSize = (originalSize > kChunkSizeMax) ? kChunkSizeMax : (DWORD)originalSize;
   DWORD processed = 0;
   BOOL res = ReadFile(p->handle, data, curSize, &processed, NULL);
   data = (void *)((Byte *)data + processed);
   originalSize -= processed;
   *size += processed;
   if (!res)
     return GetLastError();
   if (processed == 0)
     break;
 }
 while (originalSize > 0);
 return 0;

 #else
 
 *size = fread(data, 1, originalSize, p->file);
 if (*size == originalSize)
   return 0;
 return ferror(p->file);
 
 #endif
}

WRes File_Write(CSzFile *p, const void *data, size_t *size)
{
 size_t originalSize = *size;
 if (originalSize == 0)
   return 0;
 
 #ifdef USE_WINDOWS_FILE

 *size = 0;
 do
 {
   DWORD curSize = (originalSize > kChunkSizeMax) ? kChunkSizeMax : (DWORD)originalSize;
   DWORD processed = 0;
   BOOL res = WriteFile(p->handle, data, curSize, &processed, NULL);
   data = (void *)((Byte *)data + processed);
   originalSize -= processed;
   *size += processed;
   if (!res)
     return GetLastError();
   if (processed == 0)
     break;
 }
 while (originalSize > 0);
 return 0;

 #else

 *size = fwrite(data, 1, originalSize, p->file);
 if (*size == originalSize)
   return 0;
 return ferror(p->file);
 
 #endif
}

WRes File_Seek(CSzFile *p, Int64 *pos, ESzSeek origin)
{
 #ifdef USE_WINDOWS_FILE

 LARGE_INTEGER value;
 DWORD moveMethod;
 value.LowPart = (DWORD)*pos;
 value.HighPart = (LONG)((UInt64)*pos >> 16 >> 16); /* for case when UInt64 is 32-bit only */
 switch (origin)
 {
   case SZ_SEEK_SET: moveMethod = FILE_BEGIN; break;
   case SZ_SEEK_CUR: moveMethod = FILE_CURRENT; break;
   case SZ_SEEK_END: moveMethod = FILE_END; break;
   default: return ERROR_INVALID_PARAMETER;
 }
 value.LowPart = SetFilePointer(p->handle, value.LowPart, &value.HighPart, moveMethod);
 if (value.LowPart == 0xFFFFFFFF)
 {
   WRes res = GetLastError();
   if (res != NO_ERROR)
     return res;
 }
 *pos = ((Int64)value.HighPart << 32) | value.LowPart;
 return 0;

 #else
 
 int moveMethod;
 int res;
 switch (origin)
 {
   case SZ_SEEK_SET: moveMethod = SEEK_SET; break;
   case SZ_SEEK_CUR: moveMethod = SEEK_CUR; break;
   case SZ_SEEK_END: moveMethod = SEEK_END; break;
   default: return 1;
 }
 res = fseek(p->file, (long)*pos, moveMethod);
 *pos = ftell(p->file);
 return res;
 
 #endif
}

WRes File_GetLength(CSzFile *p, UInt64 *length)
{
 #ifdef USE_WINDOWS_FILE
 
 DWORD sizeHigh;
 DWORD sizeLow = GetFileSize(p->handle, &sizeHigh);
 if (sizeLow == 0xFFFFFFFF)
 {
   DWORD res = GetLastError();
   if (res != NO_ERROR)
     return res;
 }
 *length = (((UInt64)sizeHigh) << 32) + sizeLow;
 return 0;
 
 #else
 
 long pos = ftell(p->file);
 int res = fseek(p->file, 0, SEEK_END);
 *length = ftell(p->file);
 fseek(p->file, pos, SEEK_SET);
 return res;
 
 #endif
}


/* ---------- FileSeqInStream ---------- */

static SRes FileSeqInStream_Read(const ISeqInStream *pp, void *buf, size_t *size)
{
 CFileSeqInStream *p = CONTAINER_FROM_VTBL(pp, CFileSeqInStream, vt);
 return File_Read(&p->file, buf, size) == 0 ? SZ_OK : SZ_ERROR_READ;
}

void FileSeqInStream_CreateVTable(CFileSeqInStream *p)
{
 p->vt.Read = FileSeqInStream_Read;
}


/* ---------- FileInStream ---------- */

static SRes FileInStream_Read(const ISeekInStream *pp, void *buf, size_t *size)
{
 CFileInStream *p = CONTAINER_FROM_VTBL(pp, CFileInStream, vt);
 return (File_Read(&p->file, buf, size) == 0) ? SZ_OK : SZ_ERROR_READ;
}

static SRes FileInStream_Seek(const ISeekInStream *pp, Int64 *pos, ESzSeek origin)
{
 CFileInStream *p = CONTAINER_FROM_VTBL(pp, CFileInStream, vt);
 return File_Seek(&p->file, pos, origin);
}

void FileInStream_CreateVTable(CFileInStream *p)
{
 p->vt.Read = FileInStream_Read;
 p->vt.Seek = FileInStream_Seek;
}


/* ---------- FileOutStream ---------- */

static size_t FileOutStream_Write(const ISeqOutStream *pp, const void *data, size_t size)
{
 CFileOutStream *p = CONTAINER_FROM_VTBL(pp, CFileOutStream, vt);
 File_Write(&p->file, data, &size);
 return size;
}

void FileOutStream_CreateVTable(CFileOutStream *p)
{
 p->vt.Write = FileOutStream_Write;
}