1
0

iowin32.c 14 KB

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