[mirror_edk2.git] / BaseTools / Source / C / LzmaCompress / Sdk / C / 7zFile.c

/* 7zFile.c -- File IO\r
2008-11-22 : Igor Pavlov : Public domain */\r
\r
#include "7zFile.h"\r
\r
#ifndef USE_WINDOWS_FILE\r
\r
#include <errno.h>\r
\r
#endif\r
\r
#ifdef USE_WINDOWS_FILE\r
\r
/*\r
   ReadFile and WriteFile functions in Windows have BUG:\r
   If you Read or Write 64MB or more (probably min_failure_size = 64MB - 32KB + 1)\r
   from/to Network file, it returns ERROR_NO_SYSTEM_RESOURCES\r
   (Insufficient system resources exist to complete the requested service).\r
   Probably in some version of Windows there are problems with other sizes:\r
   for 32 MB (maybe also for 16 MB).\r
   And message can be "Network connection was lost"\r
*/\r
\r
#define kChunkSizeMax (1 << 22)\r
\r
#endif\r
\r
void File_Construct(CSzFile *p)\r
{\r
  #ifdef USE_WINDOWS_FILE\r
  p->handle = INVALID_HANDLE_VALUE;\r
  #else\r
  p->file = NULL;\r
  #endif\r
}\r
\r
static WRes File_Open(CSzFile *p, const char *name, int writeMode)\r
{\r
  #ifdef USE_WINDOWS_FILE\r
  p->handle = CreateFileA(name,\r
      writeMode ? GENERIC_WRITE : GENERIC_READ,\r
      FILE_SHARE_READ, NULL,\r
      writeMode ? CREATE_ALWAYS : OPEN_EXISTING,\r
      FILE_ATTRIBUTE_NORMAL, NULL);\r
  return (p->handle != INVALID_HANDLE_VALUE) ? 0 : GetLastError();\r
  #else\r
  p->file = fopen(name, writeMode ? "wb+" : "rb");\r
  return (p->file != 0) ? 0 : errno;\r
  #endif\r
}\r
\r
WRes InFile_Open(CSzFile *p, const char *name) { return File_Open(p, name, 0); }\r
WRes OutFile_Open(CSzFile *p, const char *name) { return File_Open(p, name, 1); }\r
\r
WRes File_Close(CSzFile *p)\r
{\r
  #ifdef USE_WINDOWS_FILE\r
  if (p->handle != INVALID_HANDLE_VALUE)\r
  {\r
    if (!CloseHandle(p->handle))\r
      return GetLastError();\r
    p->handle = INVALID_HANDLE_VALUE;\r
  }\r
  #else\r
  if (p->file != NULL)\r
  {\r
    int res = fclose(p->file);\r
    if (res != 0)\r
      return res;\r
    p->file = NULL;\r
  }\r
  #endif\r
  return 0;\r
}\r
\r
WRes File_Read(CSzFile *p, void *data, size_t *size)\r
{\r
  size_t originalSize = *size;\r
  if (originalSize == 0)\r
    return 0;\r
\r
  #ifdef USE_WINDOWS_FILE\r
\r
  *size = 0;\r
  do\r
  {\r
    DWORD curSize = (originalSize > kChunkSizeMax) ? kChunkSizeMax : (DWORD)originalSize;\r
    DWORD processed = 0;\r
    BOOL res = ReadFile(p->handle, data, curSize, &processed, NULL);\r
    data = (void *)((Byte *)data + processed);\r
    originalSize -= processed;\r
    *size += processed;\r
    if (!res)\r
      return GetLastError();\r
    if (processed == 0)\r
      break;\r
  }\r
  while (originalSize > 0);\r
  return 0;\r
\r
  #else\r
  \r
  *size = fread(data, 1, originalSize, p->file);\r
  if (*size == originalSize)\r
    return 0;\r
  return ferror(p->file);\r
  \r
  #endif\r
}\r
\r
WRes File_Write(CSzFile *p, const void *data, size_t *size)\r
{\r
  size_t originalSize = *size;\r
  if (originalSize == 0)\r
    return 0;\r
  \r
  #ifdef USE_WINDOWS_FILE\r
\r
  *size = 0;\r
  do\r
  {\r
    DWORD curSize = (originalSize > kChunkSizeMax) ? kChunkSizeMax : (DWORD)originalSize;\r
    DWORD processed = 0;\r
    BOOL res = WriteFile(p->handle, data, curSize, &processed, NULL);\r
    data = (void *)((Byte *)data + processed);\r
    originalSize -= processed;\r
    *size += processed;\r
    if (!res)\r
      return GetLastError();\r
    if (processed == 0)\r
      break;\r
  }\r
  while (originalSize > 0);\r
  return 0;\r
\r
  #else\r
\r
  *size = fwrite(data, 1, originalSize, p->file);\r
  if (*size == originalSize)\r
    return 0;\r
  return ferror(p->file);\r
  \r
  #endif\r
}\r
\r
WRes File_Seek(CSzFile *p, Int64 *pos, ESzSeek origin)\r
{\r
  #ifdef USE_WINDOWS_FILE\r
\r
  LARGE_INTEGER value;\r
  DWORD moveMethod;\r
  value.LowPart = (DWORD)*pos;\r
  value.HighPart = (LONG)((UInt64)*pos >> 16 >> 16); /* for case when UInt64 is 32-bit only */\r
  switch (origin)\r
  {\r
    case SZ_SEEK_SET: moveMethod = FILE_BEGIN; break;\r
    case SZ_SEEK_CUR: moveMethod = FILE_CURRENT; break;\r
    case SZ_SEEK_END: moveMethod = FILE_END; break;\r
    default: return ERROR_INVALID_PARAMETER;\r
  }\r
  value.LowPart = SetFilePointer(p->handle, value.LowPart, &value.HighPart, moveMethod);\r
  if (value.LowPart == 0xFFFFFFFF)\r
  {\r
    WRes res = GetLastError();\r
    if (res != NO_ERROR)\r
      return res;\r
  }\r
  *pos = ((Int64)value.HighPart << 32) | value.LowPart;\r
  return 0;\r
\r
  #else\r
  \r
  int moveMethod;\r
  int res;\r
  switch (origin)\r
  {\r
    case SZ_SEEK_SET: moveMethod = SEEK_SET; break;\r
    case SZ_SEEK_CUR: moveMethod = SEEK_CUR; break;\r
    case SZ_SEEK_END: moveMethod = SEEK_END; break;\r
    default: return 1;\r
  }\r
  res = fseek(p->file, (long)*pos, moveMethod);\r
  *pos = ftell(p->file);\r
  return res;\r
  \r
  #endif\r
}\r
\r
WRes File_GetLength(CSzFile *p, UInt64 *length)\r
{\r
  #ifdef USE_WINDOWS_FILE\r
  \r
  DWORD sizeHigh;\r
  DWORD sizeLow = GetFileSize(p->handle, &sizeHigh);\r
  if (sizeLow == 0xFFFFFFFF)\r
  {\r
    DWORD res = GetLastError();\r
    if (res != NO_ERROR)\r
      return res;\r
  }\r
  *length = (((UInt64)sizeHigh) << 32) + sizeLow;\r
  return 0;\r
  \r
  #else\r
  \r
  long pos = ftell(p->file);\r
  int res = fseek(p->file, 0, SEEK_END);\r
  *length = ftell(p->file);\r
  fseek(p->file, pos, SEEK_SET);\r
  return res;\r
  \r
  #endif\r
}\r
\r
\r
/* ---------- FileSeqInStream ---------- */\r
\r
static SRes FileSeqInStream_Read(void *pp, void *buf, size_t *size)\r
{\r
  CFileSeqInStream *p = (CFileSeqInStream *)pp;\r
  return File_Read(&p->file, buf, size) == 0 ? SZ_OK : SZ_ERROR_READ;\r
}\r
\r
void FileSeqInStream_CreateVTable(CFileSeqInStream *p)\r
{\r
  p->s.Read = FileSeqInStream_Read;\r
}\r
\r
\r
/* ---------- FileInStream ---------- */\r
\r
static SRes FileInStream_Read(void *pp, void *buf, size_t *size)\r
{\r
  CFileInStream *p = (CFileInStream *)pp;\r
  return (File_Read(&p->file, buf, size) == 0) ? SZ_OK : SZ_ERROR_READ;\r
}\r
\r
static SRes FileInStream_Seek(void *pp, Int64 *pos, ESzSeek origin)\r
{\r
  CFileInStream *p = (CFileInStream *)pp;\r
  return File_Seek(&p->file, pos, origin);\r
}\r
\r
void FileInStream_CreateVTable(CFileInStream *p)\r
{\r
  p->s.Read = FileInStream_Read;\r
  p->s.Seek = FileInStream_Seek;\r
}\r
\r
\r
/* ---------- FileOutStream ---------- */\r
\r
static size_t FileOutStream_Write(void *pp, const void *data, size_t size)\r
{\r
  CFileOutStream *p = (CFileOutStream *)pp;\r
  File_Write(&p->file, data, &size);\r
  return size;\r
}\r
\r
void FileOutStream_CreateVTable(CFileOutStream *p)\r
{\r
  p->s.Write = FileOutStream_Write;\r
}\r
Commit	Line	Data
30fdf114 LG	1	/* 7zFile.c -- File IO\r
	2	2008-11-22 : Igor Pavlov : Public domain */\r
	3	\r
	4	#include "7zFile.h"\r
	5	\r
	6	#ifndef USE_WINDOWS_FILE\r
	7	\r
	8	#include <errno.h>\r
	9	\r
	10	#endif\r
	11	\r
	12	#ifdef USE_WINDOWS_FILE\r
	13	\r
	14	/*\r
	15	ReadFile and WriteFile functions in Windows have BUG:\r
	16	If you Read or Write 64MB or more (probably min_failure_size = 64MB - 32KB + 1)\r
	17	from/to Network file, it returns ERROR_NO_SYSTEM_RESOURCES\r
	18	(Insufficient system resources exist to complete the requested service).\r
	19	Probably in some version of Windows there are problems with other sizes:\r
	20	for 32 MB (maybe also for 16 MB).\r
	21	And message can be "Network connection was lost"\r
	22	*/\r
	23	\r
	24	#define kChunkSizeMax (1 << 22)\r
	25	\r
	26	#endif\r
	27	\r
	28	void File_Construct(CSzFile *p)\r
	29	{\r
	30	#ifdef USE_WINDOWS_FILE\r
	31	p->handle = INVALID_HANDLE_VALUE;\r
	32	#else\r
	33	p->file = NULL;\r
	34	#endif\r
	35	}\r
	36	\r
	37	static WRes File_Open(CSzFile p, const char name, int writeMode)\r
	38	{\r
	39	#ifdef USE_WINDOWS_FILE\r
	40	p->handle = CreateFileA(name,\r
	41	writeMode ? GENERIC_WRITE : GENERIC_READ,\r
	42	FILE_SHARE_READ, NULL,\r
	43	writeMode ? CREATE_ALWAYS : OPEN_EXISTING,\r
	44	FILE_ATTRIBUTE_NORMAL, NULL);\r
	45	return (p->handle != INVALID_HANDLE_VALUE) ? 0 : GetLastError();\r
	46	#else\r
	47	p->file = fopen(name, writeMode ? "wb+" : "rb");\r
	48	return (p->file != 0) ? 0 : errno;\r
	49	#endif\r
	50	}\r
	51	\r
	52	WRes InFile_Open(CSzFile p, const char name) { return File_Open(p, name, 0); }\r
	53	WRes OutFile_Open(CSzFile p, const char name) { return File_Open(p, name, 1); }\r
	54	\r
	55	WRes File_Close(CSzFile *p)\r
	56	{\r
	57	#ifdef USE_WINDOWS_FILE\r
	58	if (p->handle != INVALID_HANDLE_VALUE)\r
	59	{\r
	60	if (!CloseHandle(p->handle))\r
	61	return GetLastError();\r
	62	p->handle = INVALID_HANDLE_VALUE;\r
	63	}\r
	64	#else\r
65	if (p->file != NULL)\r
66	{\r
67	int res = fclose(p->file);\r
68	if (res != 0)\r
69	return res;\r
70	p->file = NULL;\r
71	}\r
72	#endif\r
73	return 0;\r
74	}\r
75	\r
76	WRes File_Read(CSzFile p, void data, size_t *size)\r
77	{\r
78	size_t originalSize = *size;\r
79	if (originalSize == 0)\r
80	return 0;\r
81	\r
82	#ifdef USE_WINDOWS_FILE\r
83	\r
84	*size = 0;\r
85	do\r
86	{\r
87	DWORD curSize = (originalSize > kChunkSizeMax) ? kChunkSizeMax : (DWORD)originalSize;\r
88	DWORD processed = 0;\r
89	BOOL res = ReadFile(p->handle, data, curSize, &processed, NULL);\r
90	data = (void )((Byte )data + processed);\r
91	originalSize -= processed;\r
92	*size += processed;\r
93	if (!res)\r
94	return GetLastError();\r
95	if (processed == 0)\r
96	break;\r
97	}\r
98	while (originalSize > 0);\r
99	return 0;\r
100	\r
101	#else\r
102	\r
103	*size = fread(data, 1, originalSize, p->file);\r
104	if (*size == originalSize)\r
105	return 0;\r
106	return ferror(p->file);\r
107	\r
108	#endif\r
109	}\r
110	\r
111	WRes File_Write(CSzFile p, const void data, size_t *size)\r
112	{\r
113	size_t originalSize = *size;\r
114	if (originalSize == 0)\r
115	return 0;\r
116	\r
117	#ifdef USE_WINDOWS_FILE\r
118	\r
119	*size = 0;\r
120	do\r
121	{\r
122	DWORD curSize = (originalSize > kChunkSizeMax) ? kChunkSizeMax : (DWORD)originalSize;\r
123	DWORD processed = 0;\r
124	BOOL res = WriteFile(p->handle, data, curSize, &processed, NULL);\r
125	data = (void )((Byte )data + processed);\r
126	originalSize -= processed;\r
127	*size += processed;\r
128	if (!res)\r
129	return GetLastError();\r
130	if (processed == 0)\r
131	break;\r
132	}\r
133	while (originalSize > 0);\r
134	return 0;\r
135	\r
136	#else\r
137	\r
138	*size = fwrite(data, 1, originalSize, p->file);\r
139	if (*size == originalSize)\r
140	return 0;\r
141	return ferror(p->file);\r
142	\r
143	#endif\r
144	}\r
145	\r
146	WRes File_Seek(CSzFile p, Int64 pos, ESzSeek origin)\r
147	{\r
148	#ifdef USE_WINDOWS_FILE\r
149	\r
150	LARGE_INTEGER value;\r
151	DWORD moveMethod;\r
152	value.LowPart = (DWORD)*pos;\r
153	value.HighPart = (LONG)((UInt64)pos >> 16 >> 16); / for case when UInt64 is 32-bit only */\r
154	switch (origin)\r
155	{\r
156	case SZ_SEEK_SET: moveMethod = FILE_BEGIN; break;\r
157	case SZ_SEEK_CUR: moveMethod = FILE_CURRENT; break;\r
158	case SZ_SEEK_END: moveMethod = FILE_END; break;\r
159	default: return ERROR_INVALID_PARAMETER;\r
160	}\r
161	value.LowPart = SetFilePointer(p->handle, value.LowPart, &value.HighPart, moveMethod);\r
162	if (value.LowPart == 0xFFFFFFFF)\r
163	{\r
164	WRes res = GetLastError();\r
165	if (res != NO_ERROR)\r
166	return res;\r
167	}\r
168	*pos = ((Int64)value.HighPart << 32) \| value.LowPart;\r
169	return 0;\r
170	\r
171	#else\r
172	\r
173	int moveMethod;\r
174	int res;\r
175	switch (origin)\r
176	{\r
177	case SZ_SEEK_SET: moveMethod = SEEK_SET; break;\r
178	case SZ_SEEK_CUR: moveMethod = SEEK_CUR; break;\r
179	case SZ_SEEK_END: moveMethod = SEEK_END; break;\r
180	default: return 1;\r
181	}\r
182	res = fseek(p->file, (long)*pos, moveMethod);\r
183	*pos = ftell(p->file);\r
184	return res;\r
185	\r
186	#endif\r
187	}\r
188	\r
189	WRes File_GetLength(CSzFile p, UInt64 length)\r
190	{\r
191	#ifdef USE_WINDOWS_FILE\r
192	\r
193	DWORD sizeHigh;\r
194	DWORD sizeLow = GetFileSize(p->handle, &sizeHigh);\r
195	if (sizeLow == 0xFFFFFFFF)\r
196	{\r
197	DWORD res = GetLastError();\r
198	if (res != NO_ERROR)\r
199	return res;\r
200	}\r
201	*length = (((UInt64)sizeHigh) << 32) + sizeLow;\r
202	return 0;\r
203	\r
204	#else\r
205	\r
206	long pos = ftell(p->file);\r
207	int res = fseek(p->file, 0, SEEK_END);\r
208	*length = ftell(p->file);\r
209	fseek(p->file, pos, SEEK_SET);\r
210	return res;\r
211	\r
212	#endif\r
213	}\r
214	\r
215	\r
216	/* ---------- FileSeqInStream ---------- */\r
217	\r
218	static SRes FileSeqInStream_Read(void pp, void buf, size_t *size)\r
219	{\r
220	CFileSeqInStream p = (CFileSeqInStream )pp;\r
221	return File_Read(&p->file, buf, size) == 0 ? SZ_OK : SZ_ERROR_READ;\r
222	}\r
223	\r
224	void FileSeqInStream_CreateVTable(CFileSeqInStream *p)\r
225	{\r
226	p->s.Read = FileSeqInStream_Read;\r
227	}\r
228	\r
229	\r
230	/* ---------- FileInStream ---------- */\r
231	\r
232	static SRes FileInStream_Read(void pp, void buf, size_t *size)\r
233	{\r
234	CFileInStream p = (CFileInStream )pp;\r
235	return (File_Read(&p->file, buf, size) == 0) ? SZ_OK : SZ_ERROR_READ;\r
236	}\r
237	\r
238	static SRes FileInStream_Seek(void pp, Int64 pos, ESzSeek origin)\r
239	{\r
240	CFileInStream p = (CFileInStream )pp;\r
241	return File_Seek(&p->file, pos, origin);\r
242	}\r
243	\r
244	void FileInStream_CreateVTable(CFileInStream *p)\r
245	{\r
246	p->s.Read = FileInStream_Read;\r
247	p->s.Seek = FileInStream_Seek;\r
248	}\r
249	\r
250	\r
251	/* ---------- FileOutStream ---------- */\r
252	\r
253	static size_t FileOutStream_Write(void pp, const void data, size_t size)\r
254	{\r
255	CFileOutStream p = (CFileOutStream )pp;\r
256	File_Write(&p->file, data, &size);\r
257	return size;\r
258	}\r
259	\r
260	void FileOutStream_CreateVTable(CFileOutStream *p)\r
261	{\r
262	p->s.Write = FileOutStream_Write;\r
263	}\r