zlib/contrib/minizip/iowin32.c

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