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

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