1 /* iowin32.c -- IO base function header for compress/uncompress .zip
2      Version 1.1, February 14h, 2010
3      part of the MiniZip project - ( http://www.winimage.com/zLibDll/minizip.html )
4 
5          Copyright (C) 1998-2010 Gilles Vollant (minizip) ( http://www.winimage.com/zLibDll/minizip.html )
6 
7          Modifications for Zip64 support
8          Copyright (C) 2009-2010 Mathias Svensson ( http://result42.com )
9 
10      For more info read MiniZip_info.txt
11 
12 */
13 
14 #include <stdlib.h>
15 
16 #include "zlib.h"
17 #include "ioapi.h"
18 #include "iowin32.h"
19 
20 #ifndef INVALID_HANDLE_VALUE
21 #define INVALID_HANDLE_VALUE (0xFFFFFFFF)
22 #endif
23 
24 #ifndef INVALID_SET_FILE_POINTER
25 #define INVALID_SET_FILE_POINTER ((DWORD)-1)
26 #endif
27 
28 
29 // see Include/shared/winapifamily.h in the Windows Kit
30 #if defined(WINAPI_FAMILY_PARTITION) && (!(defined(IOWIN32_USING_WINRT_API)))
31 
32 #if !defined(WINAPI_FAMILY_ONE_PARTITION)
33 #define WINAPI_FAMILY_ONE_PARTITION(PartitionSet, Partition) ((WINAPI_FAMILY & PartitionSet) == Partition)
34 #endif
35 
36 #if WINAPI_FAMILY_ONE_PARTITION(WINAPI_FAMILY, WINAPI_PARTITION_APP)
37 #define IOWIN32_USING_WINRT_API 1
38 #endif
39 #endif
40 
41 voidpf  ZCALLBACK win32_open_file_func  OF((voidpf opaque, const char* filename, int mode));
42 uLong   ZCALLBACK win32_read_file_func  OF((voidpf opaque, voidpf stream, void* buf, uLong size));
43 uLong   ZCALLBACK win32_write_file_func OF((voidpf opaque, voidpf stream, const void* buf, uLong size));
44 ZPOS64_T ZCALLBACK win32_tell64_file_func  OF((voidpf opaque, voidpf stream));
45 long    ZCALLBACK win32_seek64_file_func  OF((voidpf opaque, voidpf stream, ZPOS64_T offset, int origin));
46 int     ZCALLBACK win32_close_file_func OF((voidpf opaque, voidpf stream));
47 int     ZCALLBACK win32_error_file_func OF((voidpf opaque, voidpf stream));
48 
49 typedef struct
50 {
51     HANDLE hf;
52     int error;
53 } WIN32FILE_IOWIN;
54 
55 
56 static void win32_translate_open_mode(int mode,
57                                       DWORD* lpdwDesiredAccess,
58                                       DWORD* lpdwCreationDisposition,
59                                       DWORD* lpdwShareMode,
60                                       DWORD* lpdwFlagsAndAttributes)
61 {
62     *lpdwDesiredAccess = *lpdwShareMode = *lpdwFlagsAndAttributes = *lpdwCreationDisposition = 0;
63 
64     if ((mode & ZLIB_FILEFUNC_MODE_READWRITEFILTER)==ZLIB_FILEFUNC_MODE_READ)
65     {
66         *lpdwDesiredAccess = GENERIC_READ;
67         *lpdwCreationDisposition = OPEN_EXISTING;
68         *lpdwShareMode = FILE_SHARE_READ;
69     }
70     else if (mode & ZLIB_FILEFUNC_MODE_EXISTING)
71     {
72         *lpdwDesiredAccess = GENERIC_WRITE | GENERIC_READ;
73         *lpdwCreationDisposition = OPEN_EXISTING;
74     }
75     else if (mode & ZLIB_FILEFUNC_MODE_CREATE)
76     {
77         *lpdwDesiredAccess = GENERIC_WRITE | GENERIC_READ;
78         *lpdwCreationDisposition = CREATE_ALWAYS;
79     }
80 }
81 
82 static voidpf win32_build_iowin(HANDLE hFile)
83 {
84     voidpf ret=NULL;
85 
86     if ((hFile != NULL) && (hFile != INVALID_HANDLE_VALUE))
87     {
88         WIN32FILE_IOWIN w32fiow;
89         w32fiow.hf = hFile;
90         w32fiow.error = 0;
91         ret = malloc(sizeof(WIN32FILE_IOWIN));
92 
93         if (ret==NULL)
94             CloseHandle(hFile);
95         else
96             *((WIN32FILE_IOWIN*)ret) = w32fiow;
97     }
98     return ret;
99 }
100 
101 voidpf ZCALLBACK win32_open64_file_func (voidpf opaque,const void* filename,int mode)
102 {
103     //const char* mode_fopen = NULL;
104     DWORD dwDesiredAccess,dwCreationDisposition,dwShareMode,dwFlagsAndAttributes ;
105     HANDLE hFile = NULL;
106 
107     win32_translate_open_mode(mode,&dwDesiredAccess,&dwCreationDisposition,&dwShareMode,&dwFlagsAndAttributes);
108 
109 #ifdef IOWIN32_USING_WINRT_API
110 #ifdef UNICODE
111     if ((filename!=NULL) && (dwDesiredAccess != 0))
112         hFile = CreateFile2((LPCTSTR)filename, dwDesiredAccess, dwShareMode, dwCreationDisposition, NULL);
113 #else
114     if ((filename!=NULL) && (dwDesiredAccess != 0))
115     {
116         WCHAR filenameW[FILENAME_MAX + 0x200 + 1];
117         MultiByteToWideChar(CP_ACP,0,(const char*)filename,-1,filenameW,FILENAME_MAX + 0x200);
118         hFile = CreateFile2(filenameW, dwDesiredAccess, dwShareMode, dwCreationDisposition, NULL);
119     }
120 #endif
121 #else
122     if ((filename!=NULL) && (dwDesiredAccess != 0))
123         hFile = CreateFile((LPCTSTR)filename, dwDesiredAccess, dwShareMode, NULL, dwCreationDisposition, dwFlagsAndAttributes, NULL);
124 #endif
125 
126     return win32_build_iowin(hFile);
127 }
128 
129 
130 voidpf ZCALLBACK win32_open64_file_funcA (voidpf opaque,const void* filename,int mode)
131 {
132     //const char* mode_fopen = NULL;
133     DWORD dwDesiredAccess,dwCreationDisposition,dwShareMode,dwFlagsAndAttributes ;
134     HANDLE hFile = NULL;
135 
136     win32_translate_open_mode(mode,&dwDesiredAccess,&dwCreationDisposition,&dwShareMode,&dwFlagsAndAttributes);
137 
138 #ifdef IOWIN32_USING_WINRT_API
139     if ((filename!=NULL) && (dwDesiredAccess != 0))
140     {
141         WCHAR filenameW[FILENAME_MAX + 0x200 + 1];
142         MultiByteToWideChar(CP_ACP,0,(const char*)filename,-1,filenameW,FILENAME_MAX + 0x200);
143         hFile = CreateFile2(filenameW, dwDesiredAccess, dwShareMode, dwCreationDisposition, NULL);
144     }
145 #else
146     if ((filename!=NULL) && (dwDesiredAccess != 0))
147         hFile = CreateFileA((LPCSTR)filename, dwDesiredAccess, dwShareMode, NULL, dwCreationDisposition, dwFlagsAndAttributes, NULL);
148 #endif
149 
150     return win32_build_iowin(hFile);
151 }
152 
153 
154 voidpf ZCALLBACK win32_open64_file_funcW (voidpf opaque,const void* filename,int mode)
155 {
156     //const char* mode_fopen = NULL;
157     DWORD dwDesiredAccess,dwCreationDisposition,dwShareMode,dwFlagsAndAttributes ;
158     HANDLE hFile = NULL;
159 
160     win32_translate_open_mode(mode,&dwDesiredAccess,&dwCreationDisposition,&dwShareMode,&dwFlagsAndAttributes);
161 
162 #ifdef IOWIN32_USING_WINRT_API
163     if ((filename!=NULL) && (dwDesiredAccess != 0))
164         hFile = CreateFile2((LPCWSTR)filename, dwDesiredAccess, dwShareMode, dwCreationDisposition,NULL);
165 #else
166     if ((filename!=NULL) && (dwDesiredAccess != 0))
167         hFile = CreateFileW((LPCWSTR)filename, dwDesiredAccess, dwShareMode, NULL, dwCreationDisposition, dwFlagsAndAttributes, NULL);
168 #endif
169 
170     return win32_build_iowin(hFile);
171 }
172 
173 
174 voidpf ZCALLBACK win32_open_file_func (voidpf opaque,const char* filename,int mode)
175 {
176     //const char* mode_fopen = NULL;
177     DWORD dwDesiredAccess,dwCreationDisposition,dwShareMode,dwFlagsAndAttributes ;
178     HANDLE hFile = NULL;
179 
180     win32_translate_open_mode(mode,&dwDesiredAccess,&dwCreationDisposition,&dwShareMode,&dwFlagsAndAttributes);
181 
182 #ifdef IOWIN32_USING_WINRT_API
183 #ifdef UNICODE
184     if ((filename!=NULL) && (dwDesiredAccess != 0))
185         hFile = CreateFile2((LPCTSTR)filename, dwDesiredAccess, dwShareMode, dwCreationDisposition, NULL);
186 #else
187     if ((filename!=NULL) && (dwDesiredAccess != 0))
188     {
189         WCHAR filenameW[FILENAME_MAX + 0x200 + 1];
190         MultiByteToWideChar(CP_ACP,0,(const char*)filename,-1,filenameW,FILENAME_MAX + 0x200);
191         hFile = CreateFile2(filenameW, dwDesiredAccess, dwShareMode, dwCreationDisposition, NULL);
192     }
193 #endif
194 #else
195     if ((filename!=NULL) && (dwDesiredAccess != 0))
196         hFile = CreateFile((LPCTSTR)filename, dwDesiredAccess, dwShareMode, NULL, dwCreationDisposition, dwFlagsAndAttributes, NULL);
197 #endif
198 
199     return win32_build_iowin(hFile);
200 }
201 
202 
203 uLong ZCALLBACK win32_read_file_func (voidpf opaque, voidpf stream, void* buf,uLong size)
204 {
205     uLong ret=0;
206     HANDLE hFile = NULL;
207     if (stream!=NULL)
208         hFile = ((WIN32FILE_IOWIN*)stream) -> hf;
209 
210     if (hFile != NULL)
211     {
212         if (!ReadFile(hFile, buf, size, &ret, NULL))
213         {
214             DWORD dwErr = GetLastError();
215             if (dwErr == ERROR_HANDLE_EOF)
216                 dwErr = 0;
217             ((WIN32FILE_IOWIN*)stream) -> error=(int)dwErr;
218         }
219     }
220 
221     return ret;
222 }
223 
224 
225 uLong ZCALLBACK win32_write_file_func (voidpf opaque,voidpf stream,const void* buf,uLong size)
226 {
227     uLong ret=0;
228     HANDLE hFile = NULL;
229     if (stream!=NULL)
230         hFile = ((WIN32FILE_IOWIN*)stream) -> hf;
231 
232     if (hFile != NULL)
233     {
234         if (!WriteFile(hFile, buf, size, &ret, NULL))
235         {
236             DWORD dwErr = GetLastError();
237             if (dwErr == ERROR_HANDLE_EOF)
238                 dwErr = 0;
239             ((WIN32FILE_IOWIN*)stream) -> error=(int)dwErr;
240         }
241     }
242 
243     return ret;
244 }
245 
246 static BOOL MySetFilePointerEx(HANDLE hFile, LARGE_INTEGER pos, LARGE_INTEGER *newPos,  DWORD dwMoveMethod)
247 {
248 #ifdef IOWIN32_USING_WINRT_API
249     return SetFilePointerEx(hFile, pos, newPos, dwMoveMethod);
250 #else
251     LONG lHigh = pos.HighPart;
252     DWORD dwNewPos = SetFilePointer(hFile, pos.LowPart, &lHigh, dwMoveMethod);
253     BOOL fOk = TRUE;
254     if (dwNewPos == 0xFFFFFFFF)
255         if (GetLastError() != NO_ERROR)
256             fOk = FALSE;
257     if ((newPos != NULL) && (fOk))
258     {
259         newPos->LowPart = dwNewPos;
260         newPos->HighPart = lHigh;
261     }
262     return fOk;
263 #endif
264 }
265 
266 long ZCALLBACK win32_tell_file_func (voidpf opaque,voidpf stream)
267 {
268     long ret=-1;
269     HANDLE hFile = NULL;
270     if (stream!=NULL)
271         hFile = ((WIN32FILE_IOWIN*)stream) -> hf;
272     if (hFile != NULL)
273     {
274         LARGE_INTEGER pos;
275         pos.QuadPart = 0;
276 
277         if (!MySetFilePointerEx(hFile, pos, &pos, FILE_CURRENT))
278         {
279             DWORD dwErr = GetLastError();
280             ((WIN32FILE_IOWIN*)stream) -> error=(int)dwErr;
281             ret = -1;
282         }
283         else
284             ret=(long)pos.LowPart;
285     }
286     return ret;
287 }
288 
289 ZPOS64_T ZCALLBACK win32_tell64_file_func (voidpf opaque, voidpf stream)
290 {
291     ZPOS64_T ret= (ZPOS64_T)-1;
292     HANDLE hFile = NULL;
293     if (stream!=NULL)
294         hFile = ((WIN32FILE_IOWIN*)stream)->hf;
295 
296     if (hFile)
297     {
298         LARGE_INTEGER pos;
299         pos.QuadPart = 0;
300 
301         if (!MySetFilePointerEx(hFile, pos, &pos, FILE_CURRENT))
302         {
303             DWORD dwErr = GetLastError();
304             ((WIN32FILE_IOWIN*)stream) -> error=(int)dwErr;
305             ret = (ZPOS64_T)-1;
306         }
307         else
308             ret=pos.QuadPart;
309     }
310     return ret;
311 }
312 
313 
314 long ZCALLBACK win32_seek_file_func (voidpf opaque,voidpf stream,uLong offset,int origin)
315 {
316     DWORD dwMoveMethod=0xFFFFFFFF;
317     HANDLE hFile = NULL;
318 
319     long ret=-1;
320     if (stream!=NULL)
321         hFile = ((WIN32FILE_IOWIN*)stream) -> hf;
322     switch (origin)
323     {
324     case ZLIB_FILEFUNC_SEEK_CUR :
325         dwMoveMethod = FILE_CURRENT;
326         break;
327     case ZLIB_FILEFUNC_SEEK_END :
328         dwMoveMethod = FILE_END;
329         break;
330     case ZLIB_FILEFUNC_SEEK_SET :
331         dwMoveMethod = FILE_BEGIN;
332         break;
333     default: return -1;
334     }
335 
336     if (hFile != NULL)
337     {
338         LARGE_INTEGER pos;
339         pos.QuadPart = offset;
340         if (!MySetFilePointerEx(hFile, pos, NULL, dwMoveMethod))
341         {
342             DWORD dwErr = GetLastError();
343             ((WIN32FILE_IOWIN*)stream) -> error=(int)dwErr;
344             ret = -1;
345         }
346         else
347             ret=0;
348     }
349     return ret;
350 }
351 
352 long ZCALLBACK win32_seek64_file_func (voidpf opaque, voidpf stream,ZPOS64_T offset,int origin)
353 {
354     DWORD dwMoveMethod=0xFFFFFFFF;
355     HANDLE hFile = NULL;
356     long ret=-1;
357 
358     if (stream!=NULL)
359         hFile = ((WIN32FILE_IOWIN*)stream)->hf;
360 
361     switch (origin)
362     {
363         case ZLIB_FILEFUNC_SEEK_CUR :
364             dwMoveMethod = FILE_CURRENT;
365             break;
366         case ZLIB_FILEFUNC_SEEK_END :
367             dwMoveMethod = FILE_END;
368             break;
369         case ZLIB_FILEFUNC_SEEK_SET :
370             dwMoveMethod = FILE_BEGIN;
371             break;
372         default: return -1;
373     }
374 
375     if (hFile)
376     {
377         LARGE_INTEGER pos;
378         pos.QuadPart = offset;
379         if (!MySetFilePointerEx(hFile, pos, NULL, dwMoveMethod))
380         {
381             DWORD dwErr = GetLastError();
382             ((WIN32FILE_IOWIN*)stream) -> error=(int)dwErr;
383             ret = -1;
384         }
385         else
386             ret=0;
387     }
388     return ret;
389 }
390 
391 int ZCALLBACK win32_close_file_func (voidpf opaque, voidpf stream)
392 {
393     int ret=-1;
394 
395     if (stream!=NULL)
396     {
397         HANDLE hFile;
398         hFile = ((WIN32FILE_IOWIN*)stream) -> hf;
399         if (hFile != NULL)
400         {
401             CloseHandle(hFile);
402             ret=0;
403         }
404         free(stream);
405     }
406     return ret;
407 }
408 
409 int ZCALLBACK win32_error_file_func (voidpf opaque,voidpf stream)
410 {
411     int ret=-1;
412     if (stream!=NULL)
413     {
414         ret = ((WIN32FILE_IOWIN*)stream) -> error;
415     }
416     return ret;
417 }
418 
419 void fill_win32_filefunc (zlib_filefunc_def* pzlib_filefunc_def)
420 {
421     pzlib_filefunc_def->zopen_file = win32_open_file_func;
422     pzlib_filefunc_def->zread_file = win32_read_file_func;
423     pzlib_filefunc_def->zwrite_file = win32_write_file_func;
424     pzlib_filefunc_def->ztell_file = win32_tell_file_func;
425     pzlib_filefunc_def->zseek_file = win32_seek_file_func;
426     pzlib_filefunc_def->zclose_file = win32_close_file_func;
427     pzlib_filefunc_def->zerror_file = win32_error_file_func;
428     pzlib_filefunc_def->opaque = NULL;
429 }
430 
431 void fill_win32_filefunc64(zlib_filefunc64_def* pzlib_filefunc_def)
432 {
433     pzlib_filefunc_def->zopen64_file = win32_open64_file_func;
434     pzlib_filefunc_def->zread_file = win32_read_file_func;
435     pzlib_filefunc_def->zwrite_file = win32_write_file_func;
436     pzlib_filefunc_def->ztell64_file = win32_tell64_file_func;
437     pzlib_filefunc_def->zseek64_file = win32_seek64_file_func;
438     pzlib_filefunc_def->zclose_file = win32_close_file_func;
439     pzlib_filefunc_def->zerror_file = win32_error_file_func;
440     pzlib_filefunc_def->opaque = NULL;
441 }
442 
443 
444 void fill_win32_filefunc64A(zlib_filefunc64_def* pzlib_filefunc_def)
445 {
446     pzlib_filefunc_def->zopen64_file = win32_open64_file_funcA;
447     pzlib_filefunc_def->zread_file = win32_read_file_func;
448     pzlib_filefunc_def->zwrite_file = win32_write_file_func;
449     pzlib_filefunc_def->ztell64_file = win32_tell64_file_func;
450     pzlib_filefunc_def->zseek64_file = win32_seek64_file_func;
451     pzlib_filefunc_def->zclose_file = win32_close_file_func;
452     pzlib_filefunc_def->zerror_file = win32_error_file_func;
453     pzlib_filefunc_def->opaque = NULL;
454 }
455 
456 
457 void fill_win32_filefunc64W(zlib_filefunc64_def* pzlib_filefunc_def)
458 {
459     pzlib_filefunc_def->zopen64_file = win32_open64_file_funcW;
460     pzlib_filefunc_def->zread_file = win32_read_file_func;
461     pzlib_filefunc_def->zwrite_file = win32_write_file_func;
462     pzlib_filefunc_def->ztell64_file = win32_tell64_file_func;
463     pzlib_filefunc_def->zseek64_file = win32_seek64_file_func;
464     pzlib_filefunc_def->zclose_file = win32_close_file_func;
465     pzlib_filefunc_def->zerror_file = win32_error_file_func;
466     pzlib_filefunc_def->opaque = NULL;
467 }
468