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 #define ZLIB_WINAPI
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 voidpf  ZCALLBACK win32_open_file_func  OF((voidpf opaque, const char* filename, int mode));
29 uLong   ZCALLBACK win32_read_file_func  OF((voidpf opaque, voidpf stream, void* buf, uLong size));
30 uLong   ZCALLBACK win32_write_file_func OF((voidpf opaque, voidpf stream, const void* buf, uLong size));
31 ZPOS64_T ZCALLBACK win32_tell64_file_func  OF((voidpf opaque, voidpf stream));
32 long    ZCALLBACK win32_seek64_file_func  OF((voidpf opaque, voidpf stream, ZPOS64_T offset, int origin));
33 int     ZCALLBACK win32_close_file_func OF((voidpf opaque, voidpf stream));
34 int     ZCALLBACK win32_error_file_func OF((voidpf opaque, voidpf stream));
35 
36 typedef struct
37 {
38     HANDLE hf;
39     int error;
40 } WIN32FILE_IOWIN;
41 
42 
win32_translate_open_mode(int mode,DWORD * lpdwDesiredAccess,DWORD * lpdwCreationDisposition,DWORD * lpdwShareMode,DWORD * lpdwFlagsAndAttributes)43 static void win32_translate_open_mode(int mode,
44                                       DWORD* lpdwDesiredAccess,
45                                       DWORD* lpdwCreationDisposition,
46                                       DWORD* lpdwShareMode,
47                                       DWORD* lpdwFlagsAndAttributes)
48 {
49     *lpdwDesiredAccess = *lpdwShareMode = *lpdwFlagsAndAttributes = *lpdwCreationDisposition = 0;
50 
51     if ((mode & ZLIB_FILEFUNC_MODE_READWRITEFILTER)==ZLIB_FILEFUNC_MODE_READ)
52     {
53         *lpdwDesiredAccess = GENERIC_READ;
54         *lpdwCreationDisposition = OPEN_EXISTING;
55         *lpdwShareMode = FILE_SHARE_READ;
56     }
57     else if (mode & ZLIB_FILEFUNC_MODE_EXISTING)
58     {
59         *lpdwDesiredAccess = GENERIC_WRITE | GENERIC_READ;
60         *lpdwCreationDisposition = OPEN_EXISTING;
61     }
62     else if (mode & ZLIB_FILEFUNC_MODE_CREATE)
63     {
64         *lpdwDesiredAccess = GENERIC_WRITE | GENERIC_READ;
65         *lpdwCreationDisposition = CREATE_ALWAYS;
66     }
67 }
68 
win32_build_iowin(HANDLE hFile)69 static voidpf win32_build_iowin(HANDLE hFile)
70 {
71     voidpf ret=NULL;
72 
73     if ((hFile != NULL) && (hFile != INVALID_HANDLE_VALUE))
74     {
75         WIN32FILE_IOWIN w32fiow;
76         w32fiow.hf = hFile;
77         w32fiow.error = 0;
78         ret = malloc(sizeof(WIN32FILE_IOWIN));
79 
80         if (ret==NULL)
81             CloseHandle(hFile);
82         else
83             *((WIN32FILE_IOWIN*)ret) = w32fiow;
84     }
85     return ret;
86 }
87 
win32_open64_file_func(voidpf opaque,const void * filename,int mode)88 voidpf ZCALLBACK win32_open64_file_func (voidpf opaque,const void* filename,int mode)
89 {
90     DWORD dwDesiredAccess,dwCreationDisposition,dwShareMode,dwFlagsAndAttributes ;
91     HANDLE hFile = NULL;
92 
93     win32_translate_open_mode(mode,&dwDesiredAccess,&dwCreationDisposition,&dwShareMode,&dwFlagsAndAttributes);
94 
95     if ((filename!=NULL) && (dwDesiredAccess != 0))
96         hFile = CreateFile((LPCTSTR)filename, dwDesiredAccess, dwShareMode, NULL, dwCreationDisposition, dwFlagsAndAttributes, NULL);
97 
98     return win32_build_iowin(hFile);
99 }
100 
101 
win32_open64_file_funcA(voidpf opaque,const void * filename,int mode)102 voidpf ZCALLBACK win32_open64_file_funcA (voidpf opaque,const void* filename,int mode)
103 {
104     DWORD dwDesiredAccess,dwCreationDisposition,dwShareMode,dwFlagsAndAttributes ;
105     HANDLE hFile = NULL;
106 
107     win32_translate_open_mode(mode,&dwDesiredAccess,&dwCreationDisposition,&dwShareMode,&dwFlagsAndAttributes);
108 
109     if ((filename!=NULL) && (dwDesiredAccess != 0))
110         hFile = CreateFileA((LPCSTR)filename, dwDesiredAccess, dwShareMode, NULL, dwCreationDisposition, dwFlagsAndAttributes, NULL);
111 
112     return win32_build_iowin(hFile);
113 }
114 
115 
win32_open64_file_funcW(voidpf opaque,const void * filename,int mode)116 voidpf ZCALLBACK win32_open64_file_funcW (voidpf opaque,const void* filename,int mode)
117 {
118     DWORD dwDesiredAccess,dwCreationDisposition,dwShareMode,dwFlagsAndAttributes ;
119     HANDLE hFile = NULL;
120 
121     win32_translate_open_mode(mode,&dwDesiredAccess,&dwCreationDisposition,&dwShareMode,&dwFlagsAndAttributes);
122 
123     if ((filename!=NULL) && (dwDesiredAccess != 0))
124         hFile = CreateFileW((LPCWSTR)filename, dwDesiredAccess, dwShareMode, NULL, dwCreationDisposition, dwFlagsAndAttributes, NULL);
125 
126     return win32_build_iowin(hFile);
127 }
128 
129 
win32_open_file_func(voidpf opaque,const char * filename,int mode)130 voidpf ZCALLBACK win32_open_file_func (voidpf opaque,const char* filename,int mode)
131 {
132     DWORD dwDesiredAccess,dwCreationDisposition,dwShareMode,dwFlagsAndAttributes ;
133     HANDLE hFile = NULL;
134 
135     win32_translate_open_mode(mode,&dwDesiredAccess,&dwCreationDisposition,&dwShareMode,&dwFlagsAndAttributes);
136 
137     if ((filename!=NULL) && (dwDesiredAccess != 0))
138         hFile = CreateFile((LPCTSTR)filename, dwDesiredAccess, dwShareMode, NULL, dwCreationDisposition, dwFlagsAndAttributes, NULL);
139 
140     return win32_build_iowin(hFile);
141 }
142 
143 
win32_read_file_func(voidpf opaque,voidpf stream,void * buf,uLong size)144 uLong ZCALLBACK win32_read_file_func (voidpf opaque, voidpf stream, void* buf,uLong size)
145 {
146     uLong ret=0;
147     HANDLE hFile = NULL;
148     if (stream!=NULL)
149         hFile = ((WIN32FILE_IOWIN*)stream) -> hf;
150 
151     if (hFile != NULL)
152     {
153         if (!ReadFile(hFile, buf, size, &ret, NULL))
154         {
155             DWORD dwErr = GetLastError();
156             if (dwErr == ERROR_HANDLE_EOF)
157                 dwErr = 0;
158             ((WIN32FILE_IOWIN*)stream) -> error=(int)dwErr;
159         }
160     }
161 
162     return ret;
163 }
164 
165 
win32_write_file_func(voidpf opaque,voidpf stream,const void * buf,uLong size)166 uLong ZCALLBACK win32_write_file_func (voidpf opaque,voidpf stream,const void* buf,uLong size)
167 {
168     uLong ret=0;
169     HANDLE hFile = NULL;
170     if (stream!=NULL)
171         hFile = ((WIN32FILE_IOWIN*)stream) -> hf;
172 
173     if (hFile != NULL)
174     {
175         if (!WriteFile(hFile, buf, size, &ret, NULL))
176         {
177             DWORD dwErr = GetLastError();
178             if (dwErr == ERROR_HANDLE_EOF)
179                 dwErr = 0;
180             ((WIN32FILE_IOWIN*)stream) -> error=(int)dwErr;
181         }
182     }
183 
184     return ret;
185 }
186 
win32_tell_file_func(voidpf opaque,voidpf stream)187 long ZCALLBACK win32_tell_file_func (voidpf opaque,voidpf stream)
188 {
189     long ret=-1;
190     HANDLE hFile = NULL;
191     if (stream!=NULL)
192         hFile = ((WIN32FILE_IOWIN*)stream) -> hf;
193     if (hFile != NULL)
194     {
195         DWORD dwSet = SetFilePointer(hFile, 0, NULL, FILE_CURRENT);
196         if (dwSet == INVALID_SET_FILE_POINTER)
197         {
198             DWORD dwErr = GetLastError();
199             ((WIN32FILE_IOWIN*)stream) -> error=(int)dwErr;
200             ret = -1;
201         }
202         else
203             ret=(long)dwSet;
204     }
205     return ret;
206 }
207 
win32_tell64_file_func(voidpf opaque,voidpf stream)208 ZPOS64_T ZCALLBACK win32_tell64_file_func (voidpf opaque, voidpf stream)
209 {
210     ZPOS64_T ret= (ZPOS64_T)-1;
211     HANDLE hFile = NULL;
212     if (stream!=NULL)
213         hFile = ((WIN32FILE_IOWIN*)stream)->hf;
214 
215     if (hFile)
216     {
217         LARGE_INTEGER li;
218         li.QuadPart = 0;
219         li.u.LowPart = SetFilePointer(hFile, li.u.LowPart, &li.u.HighPart, FILE_CURRENT);
220         if ( (li.LowPart == 0xFFFFFFFF) && (GetLastError() != NO_ERROR))
221         {
222             DWORD dwErr = GetLastError();
223             ((WIN32FILE_IOWIN*)stream) -> error=(int)dwErr;
224             ret = (ZPOS64_T)-1;
225         }
226         else
227             ret=li.QuadPart;
228     }
229     return ret;
230 }
231 
232 
win32_seek_file_func(voidpf opaque,voidpf stream,uLong offset,int origin)233 long ZCALLBACK win32_seek_file_func (voidpf opaque,voidpf stream,uLong offset,int origin)
234 {
235     DWORD dwMoveMethod=0xFFFFFFFF;
236     HANDLE hFile = NULL;
237 
238     long ret=-1;
239     if (stream!=NULL)
240         hFile = ((WIN32FILE_IOWIN*)stream) -> hf;
241     switch (origin)
242     {
243     case ZLIB_FILEFUNC_SEEK_CUR :
244         dwMoveMethod = FILE_CURRENT;
245         break;
246     case ZLIB_FILEFUNC_SEEK_END :
247         dwMoveMethod = FILE_END;
248         break;
249     case ZLIB_FILEFUNC_SEEK_SET :
250         dwMoveMethod = FILE_BEGIN;
251         break;
252     default: return -1;
253     }
254 
255     if (hFile != NULL)
256     {
257         DWORD dwSet = SetFilePointer(hFile, offset, NULL, dwMoveMethod);
258         if (dwSet == INVALID_SET_FILE_POINTER)
259         {
260             DWORD dwErr = GetLastError();
261             ((WIN32FILE_IOWIN*)stream) -> error=(int)dwErr;
262             ret = -1;
263         }
264         else
265             ret=0;
266     }
267     return ret;
268 }
269 
win32_seek64_file_func(voidpf opaque,voidpf stream,ZPOS64_T offset,int origin)270 long ZCALLBACK win32_seek64_file_func (voidpf opaque, voidpf stream,ZPOS64_T offset,int origin)
271 {
272     DWORD dwMoveMethod=0xFFFFFFFF;
273     HANDLE hFile = NULL;
274     long ret=-1;
275 
276     if (stream!=NULL)
277         hFile = ((WIN32FILE_IOWIN*)stream)->hf;
278 
279     switch (origin)
280     {
281         case ZLIB_FILEFUNC_SEEK_CUR :
282             dwMoveMethod = FILE_CURRENT;
283             break;
284         case ZLIB_FILEFUNC_SEEK_END :
285             dwMoveMethod = FILE_END;
286             break;
287         case ZLIB_FILEFUNC_SEEK_SET :
288             dwMoveMethod = FILE_BEGIN;
289             break;
290         default: return -1;
291     }
292 
293     if (hFile)
294     {
295         LARGE_INTEGER* li = (LARGE_INTEGER*)&offset;
296         DWORD dwSet = SetFilePointer(hFile, li->u.LowPart, &li->u.HighPart, dwMoveMethod);
297         if (dwSet == INVALID_SET_FILE_POINTER)
298         {
299             DWORD dwErr = GetLastError();
300             ((WIN32FILE_IOWIN*)stream) -> error=(int)dwErr;
301             ret = -1;
302         }
303         else
304             ret=0;
305     }
306     return ret;
307 }
308 
win32_close_file_func(voidpf opaque,voidpf stream)309 int ZCALLBACK win32_close_file_func (voidpf opaque, voidpf stream)
310 {
311     int ret=-1;
312 
313     if (stream!=NULL)
314     {
315         HANDLE hFile;
316         hFile = ((WIN32FILE_IOWIN*)stream) -> hf;
317         if (hFile != NULL)
318         {
319             CloseHandle(hFile);
320             ret=0;
321         }
322         free(stream);
323     }
324     return ret;
325 }
326 
win32_error_file_func(voidpf opaque,voidpf stream)327 int ZCALLBACK win32_error_file_func (voidpf opaque,voidpf stream)
328 {
329     int ret=-1;
330     if (stream!=NULL)
331     {
332         ret = ((WIN32FILE_IOWIN*)stream) -> error;
333     }
334     return ret;
335 }
336 
fill_win32_filefunc(zlib_filefunc_def * pzlib_filefunc_def)337 void fill_win32_filefunc (zlib_filefunc_def* pzlib_filefunc_def)
338 {
339     pzlib_filefunc_def->zopen_file = win32_open_file_func;
340     pzlib_filefunc_def->zread_file = win32_read_file_func;
341     pzlib_filefunc_def->zwrite_file = win32_write_file_func;
342     pzlib_filefunc_def->ztell_file = win32_tell_file_func;
343     pzlib_filefunc_def->zseek_file = win32_seek_file_func;
344     pzlib_filefunc_def->zclose_file = win32_close_file_func;
345     pzlib_filefunc_def->zerror_file = win32_error_file_func;
346     pzlib_filefunc_def->opaque = NULL;
347 }
348 
fill_win32_filefunc64(zlib_filefunc64_def * pzlib_filefunc_def)349 void fill_win32_filefunc64(zlib_filefunc64_def* pzlib_filefunc_def)
350 {
351     pzlib_filefunc_def->zopen64_file = win32_open64_file_func;
352     pzlib_filefunc_def->zread_file = win32_read_file_func;
353     pzlib_filefunc_def->zwrite_file = win32_write_file_func;
354     pzlib_filefunc_def->ztell64_file = win32_tell64_file_func;
355     pzlib_filefunc_def->zseek64_file = win32_seek64_file_func;
356     pzlib_filefunc_def->zclose_file = win32_close_file_func;
357     pzlib_filefunc_def->zerror_file = win32_error_file_func;
358     pzlib_filefunc_def->opaque = NULL;
359 }
360 
361 
fill_win32_filefunc64A(zlib_filefunc64_def * pzlib_filefunc_def)362 void fill_win32_filefunc64A(zlib_filefunc64_def* pzlib_filefunc_def)
363 {
364     pzlib_filefunc_def->zopen64_file = win32_open64_file_funcA;
365     pzlib_filefunc_def->zread_file = win32_read_file_func;
366     pzlib_filefunc_def->zwrite_file = win32_write_file_func;
367     pzlib_filefunc_def->ztell64_file = win32_tell64_file_func;
368     pzlib_filefunc_def->zseek64_file = win32_seek64_file_func;
369     pzlib_filefunc_def->zclose_file = win32_close_file_func;
370     pzlib_filefunc_def->zerror_file = win32_error_file_func;
371     pzlib_filefunc_def->opaque = NULL;
372 }
373 
374 
fill_win32_filefunc64W(zlib_filefunc64_def * pzlib_filefunc_def)375 void fill_win32_filefunc64W(zlib_filefunc64_def* pzlib_filefunc_def)
376 {
377     pzlib_filefunc_def->zopen64_file = win32_open64_file_funcW;
378     pzlib_filefunc_def->zread_file = win32_read_file_func;
379     pzlib_filefunc_def->zwrite_file = win32_write_file_func;
380     pzlib_filefunc_def->ztell64_file = win32_tell64_file_func;
381     pzlib_filefunc_def->zseek64_file = win32_seek64_file_func;
382     pzlib_filefunc_def->zclose_file = win32_close_file_func;
383     pzlib_filefunc_def->zerror_file = win32_error_file_func;
384     pzlib_filefunc_def->opaque = NULL;
385 }
386