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