ce188d4d |
1 | /* 7zFile.c -- File IO\r |
2 | 2009-11-24 : Igor Pavlov : Public domain */\r |
3 | \r |
4 | #include "Precomp.h"\r |
5 | \r |
6 | #include "7zFile.h"\r |
7 | \r |
8 | #ifndef USE_WINDOWS_FILE\r |
9 | \r |
10 | #ifndef UNDER_CE\r |
11 | #include <errno.h>\r |
12 | #endif\r |
13 | \r |
14 | #else\r |
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 |
39 | #if !defined(UNDER_CE) || !defined(USE_WINDOWS_FILE)\r |
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 |
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 |
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 |
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 |
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 |
216 | DWORD sizeHigh;\r |
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 |