blob: 743dd787cb837c433947da465b423adeb445e5c8 [file] [log] [blame]
Antoine Cœur0c2fceb2017-10-05 15:30:37 +08001/* zip.c -- Zip manipulation
Antoine Cœur0c2fceb2017-10-05 15:30:37 +08002 part of the MiniZip project
3
Nathan Moinvazirib39f7a02020-02-12 08:25:33 -08004 Copyright (C) 2010-2020 Nathan Moinvaziri
Antoine Cœur0c2fceb2017-10-05 15:30:37 +08005 https://github.com/nmoinvaz/minizip
Nathan Moinvaziri79cfab02018-08-17 12:21:06 -07006 Copyright (C) 2009-2010 Mathias Svensson
Antoine Cœur0c2fceb2017-10-05 15:30:37 +08007 Modifications for Zip64 support
8 http://result42.com
Nathan Moinvaziri79cfab02018-08-17 12:21:06 -07009 Copyright (C) 2007-2008 Even Rouault
10 Modifications of Unzip for Zip64
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080011 Copyright (C) 1998-2010 Gilles Vollant
Viktor Szakats9dea6f22018-10-17 22:39:01 +000012 https://www.winimage.com/zLibDll/minizip.html
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080013
14 This program is distributed under the terms of the same license as zlib.
15 See the accompanying LICENSE file for the full text of the license.
16*/
17
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080018
Nathan Moinvaziri00cca9f2017-10-16 07:37:11 -070019#include "mz.h"
Nathan Moinvaziri5f091882018-10-24 18:06:08 -070020#include "mz_crypt.h"
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080021#include "mz_strm.h"
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080022#ifdef HAVE_BZIP2
23# include "mz_strm_bzip.h"
24#endif
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -080025#ifdef HAVE_LIBCOMP
Nathan Moinvaziri90d31c72018-11-20 03:17:20 -080026# include "mz_strm_libcomp.h"
27#endif
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080028#ifdef HAVE_LZMA
29# include "mz_strm_lzma.h"
30#endif
Nathan Moinvazirib47b41b2018-07-11 09:44:29 -070031#include "mz_strm_mem.h"
Nathan Moinvaziri0e5c9dc2018-05-23 20:21:48 -070032#ifdef HAVE_PKCRYPT
33# include "mz_strm_pkcrypt.h"
34#endif
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -080035#ifdef HAVE_WZAES
Nathan Moinvaziri21a31022018-10-24 09:50:16 -070036# include "mz_strm_wzaes.h"
37#endif
Nathan Moinvaziri4f6a0e32017-11-10 08:45:14 -080038#ifdef HAVE_ZLIB
39# include "mz_strm_zlib.h"
40#endif
Nathan Moinvaziri1f4758b2020-06-14 21:52:55 -070041#ifdef HAVE_ZSTD
42# include "mz_strm_zstd.h"
43#endif
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080044
45#include "mz_zip.h"
46
Nathan Moinvazirib16ab562018-11-20 16:56:21 -080047#include <ctype.h> /* tolower */
Nathan Moinvazirie8496272018-11-21 15:05:58 -080048#include <stdio.h> /* snprintf */
Nathan Moinvaziri1ee609b2018-11-19 21:34:35 -080049
zedcf076662019-05-15 21:43:56 +030050#if defined(_MSC_VER) || defined(__MINGW32__)
Nathan Moinvaziri2e0a20a2019-04-28 13:03:11 -070051# define localtime_r(t1,t2) (localtime_s(t2,t1) == 0 ? t1 : NULL)
zed65447dd2019-05-16 22:19:08 +030052#endif
53#if defined(_MSC_VER) && (_MSC_VER < 1900)
54# define snprintf _snprintf
Nathan Moinvaziric565fa82018-10-19 08:48:33 -070055#endif
56
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080057/***************************************************************************/
58
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070059#define MZ_ZIP_MAGIC_LOCALHEADER (0x04034b50)
Nathan Moinvaziribe483d82019-10-19 17:30:23 -070060#define MZ_ZIP_MAGIC_LOCALHEADERU8 { 0x50, 0x4b, 0x03, 0x04 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070061#define MZ_ZIP_MAGIC_CENTRALHEADER (0x02014b50)
Nathan Moinvaziri545d7292019-10-21 15:32:18 -070062#define MZ_ZIP_MAGIC_CENTRALHEADERU8 { 0x50, 0x4b, 0x01, 0x02 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070063#define MZ_ZIP_MAGIC_ENDHEADER (0x06054b50)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -080064#define MZ_ZIP_MAGIC_ENDHEADERU8 { 0x50, 0x4b, 0x05, 0x06 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070065#define MZ_ZIP_MAGIC_ENDHEADER64 (0x06064b50)
66#define MZ_ZIP_MAGIC_ENDLOCHEADER64 (0x07064b50)
67#define MZ_ZIP_MAGIC_DATADESCRIPTOR (0x08074b50)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -080068#define MZ_ZIP_MAGIC_DATADESCRIPTORU8 { 0x50, 0x4b, 0x07, 0x08 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070069
Nathan Moinvaziriba1db162018-11-23 10:07:35 -080070#define MZ_ZIP_SIZE_LD_ITEM (30)
Nathan Moinvaziri638f31f2018-08-27 08:17:16 -070071#define MZ_ZIP_SIZE_CD_ITEM (46)
72#define MZ_ZIP_SIZE_CD_LOCATOR64 (20)
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -080073#define MZ_ZIP_SIZE_MAX_DATA_DESCRIPTOR (24)
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080074
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -080075#define MZ_ZIP_OFFSET_CRC_SIZES (14)
76
Nathan Moinvaziric78c7822018-12-01 20:42:36 -080077#ifndef MZ_ZIP_EOCD_MAX_BACK
78#define MZ_ZIP_EOCD_MAX_BACK (1 << 20)
79#endif
80
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080081/***************************************************************************/
82
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -070083typedef struct mz_zip_s {
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080084 mz_zip_file file_info;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070085 mz_zip_file local_file_info;
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080086
Nathan Moinvazirib16ab562018-11-20 16:56:21 -080087 void *stream; /* main stream */
88 void *cd_stream; /* pointer to the stream with the cd */
89 void *cd_mem_stream; /* memory stream for central directory */
90 void *compress_stream; /* compression stream */
91 void *crypt_stream; /* encryption stream */
92 void *file_info_stream; /* memory stream for storing file info */
93 void *local_file_info_stream; /* memory stream for storing local file info */
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080094
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070095 int32_t open_mode;
Nathan Moinvaziridd103f42018-11-13 14:50:32 -080096 uint8_t recover;
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -080097 uint8_t data_descriptor;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070098
Nathan Moinvazirib16ab562018-11-20 16:56:21 -080099 uint32_t disk_number_with_cd; /* number of the disk with the central dir */
100 int64_t disk_offset_shift; /* correction for zips that have wrong offset start of cd */
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700101
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800102 int64_t cd_start_pos; /* pos of the first file in the central dir stream */
103 int64_t cd_current_pos; /* pos of the current file in the central dir */
104 int64_t cd_offset; /* offset of start of central directory */
105 int64_t cd_size; /* size of the central directory */
Nathan Moinvaziri0e1fd892019-09-28 20:55:37 -0700106 uint32_t cd_signature; /* signature of central directory */
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700107
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800108 uint8_t entry_scanned; /* entry header information read ok */
109 uint8_t entry_opened; /* entry is open for read/write */
110 uint8_t entry_raw; /* entry opened with raw mode */
111 uint32_t entry_crc32; /* entry crc32 */
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700112
Nathan Moinvaziri904c4082018-10-08 23:31:21 -0700113 uint64_t number_entry;
Nathan Moinvaziri6e21e862017-10-19 09:57:54 -0700114
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -0700115 uint16_t version_madeby;
Nathan Moinvaziri6e21e862017-10-19 09:57:54 -0700116 char *comment;
Antoine Cœur0c2fceb2017-10-05 15:30:37 +0800117} mz_zip;
118
119/***************************************************************************/
120
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800121#if 0
122# define mz_zip_print printf
123#else
124# define mz_zip_print(fmt,...)
125#endif
126
127/***************************************************************************/
128
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800129/* Locate the end of central directory */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700130static int32_t mz_zip_search_eocd(void *stream, int64_t *central_pos) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -0700131 int64_t file_size = 0;
Nathan Moinvaziric78c7822018-12-01 20:42:36 -0800132 int64_t max_back = MZ_ZIP_EOCD_MAX_BACK;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800133 uint8_t find[4] = MZ_ZIP_MAGIC_ENDHEADERU8;
Nathan Moinvaziri05e03ca2018-10-28 16:15:13 -0700134 int32_t err = MZ_OK;
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700135
Nathan Moinvaziri05e03ca2018-10-28 16:15:13 -0700136 err = mz_stream_seek(stream, 0, MZ_SEEK_END);
Nathan Moinvaziria93a2ad2018-11-13 17:51:42 -0800137 if (err != MZ_OK)
138 return err;
Antoine Cœur0c2fceb2017-10-05 15:30:37 +0800139
140 file_size = mz_stream_tell(stream);
Nathan Moinvaziri286a8172018-11-08 14:56:14 -0800141
Nathan Moinvaziric78c7822018-12-01 20:42:36 -0800142 if (max_back <= 0 || max_back > file_size)
Antoine Cœur0c2fceb2017-10-05 15:30:37 +0800143 max_back = file_size;
144
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800145 return mz_stream_find_reverse(stream, (const void *)find, sizeof(find), max_back, central_pos);
Antoine Cœur0c2fceb2017-10-05 15:30:37 +0800146}
147
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800148/* Locate the end of central directory 64 of a zip file */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700149static int32_t mz_zip_search_zip64_eocd(void *stream, const int64_t end_central_offset, int64_t *central_pos) {
Nathan Moinvaziri904c4082018-10-08 23:31:21 -0700150 int64_t offset = 0;
Antoine Cœur0c2fceb2017-10-05 15:30:37 +0800151 uint32_t value32 = 0;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -0700152 int32_t err = MZ_OK;
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700153
154
155 *central_pos = 0;
Antoine Cœur0c2fceb2017-10-05 15:30:37 +0800156
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800157 /* Zip64 end of central directory locator */
Nathan Moinvaziri79ac3382017-10-23 17:22:36 -0700158 err = mz_stream_seek(stream, end_central_offset - MZ_ZIP_SIZE_CD_LOCATOR64, MZ_SEEK_SET);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800159 /* Read locator signature */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700160 if (err == MZ_OK) {
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700161 err = mz_stream_read_uint32(stream, &value32);
Nathan Moinvaziri00cca9f2017-10-16 07:37:11 -0700162 if (value32 != MZ_ZIP_MAGIC_ENDLOCHEADER64)
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700163 err = MZ_FORMAT_ERROR;
164 }
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800165 /* Number of the disk with the start of the zip64 end of central directory */
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700166 if (err == MZ_OK)
167 err = mz_stream_read_uint32(stream, &value32);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800168 /* Relative offset of the zip64 end of central directory record8 */
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700169 if (err == MZ_OK)
Nathan Moinvaziri904c4082018-10-08 23:31:21 -0700170 err = mz_stream_read_uint64(stream, (uint64_t *)&offset);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800171 /* Total number of disks */
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700172 if (err == MZ_OK)
173 err = mz_stream_read_uint32(stream, &value32);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800174 /* Goto end of central directory record */
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700175 if (err == MZ_OK)
Nathan Moinvaziri904c4082018-10-08 23:31:21 -0700176 err = mz_stream_seek(stream, (int64_t)offset, MZ_SEEK_SET);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800177 /* The signature */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700178 if (err == MZ_OK) {
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700179 err = mz_stream_read_uint32(stream, &value32);
Nathan Moinvaziri00cca9f2017-10-16 07:37:11 -0700180 if (value32 != MZ_ZIP_MAGIC_ENDHEADER64)
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700181 err = MZ_FORMAT_ERROR;
182 }
Antoine Cœur0c2fceb2017-10-05 15:30:37 +0800183
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700184 if (err == MZ_OK)
185 *central_pos = offset;
186
187 return err;
Antoine Cœur0c2fceb2017-10-05 15:30:37 +0800188}
189
Nathan Moinvaziri215a7a82020-12-13 10:58:41 -0800190/* Get PKWARE traditional encryption verifier */
191static uint16_t mz_zip_get_pk_verify(uint32_t dos_date, uint64_t crc, uint16_t flag)
192{
Nathan Moinvaziri0030fa22020-12-21 19:31:07 -0800193 /* Info-ZIP modification to ZipCrypto format: if bit 3 of the general
Nathan Moinvaziri215a7a82020-12-13 10:58:41 -0800194 * purpose bit flag is set, it uses high byte of 16-bit File Time. */
195 if (flag & MZ_ZIP_FLAG_DATA_DESCRIPTOR)
196 return ((dos_date >> 16) & 0xff) << 8 | ((dos_date >> 8) & 0xff);
197 return ((crc >> 16) & 0xff) << 8 | ((crc >> 24) & 0xff);
198}
199
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800200/* Get info about the current file in the zip file */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700201static int32_t mz_zip_entry_read_header(void *stream, uint8_t local, mz_zip_file *file_info, void *file_extra_stream) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700202 uint64_t ntfs_time = 0;
203 uint32_t reserved = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700204 uint32_t magic = 0;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700205 uint32_t dos_date = 0;
Nathan Moinvaziri69e35492018-11-05 15:48:41 -0800206 uint32_t field_pos = 0;
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700207 uint16_t field_type = 0;
208 uint16_t field_length = 0;
209 uint32_t field_length_read = 0;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700210 uint16_t ntfs_attrib_id = 0;
211 uint16_t ntfs_attrib_size = 0;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700212 uint16_t linkname_size;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700213 uint16_t value16 = 0;
214 uint32_t value32 = 0;
Nathan Moinvazirif9e34482018-11-27 09:45:09 -0800215 int64_t extrafield_pos = 0;
216 int64_t comment_pos = 0;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700217 int64_t linkname_pos = 0;
218 int64_t saved_pos = 0;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -0700219 int32_t err = MZ_OK;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700220 char *linkname = NULL;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700221
222
223 memset(file_info, 0, sizeof(mz_zip_file));
224
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800225 /* Check the magic */
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700226 err = mz_stream_read_uint32(stream, &magic);
Nathan Moinvaziricda36002017-10-21 09:37:18 -0700227 if (err == MZ_END_OF_STREAM)
228 err = MZ_END_OF_LIST;
229 else if (magic == MZ_ZIP_MAGIC_ENDHEADER || magic == MZ_ZIP_MAGIC_ENDHEADER64)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700230 err = MZ_END_OF_LIST;
231 else if ((local) && (magic != MZ_ZIP_MAGIC_LOCALHEADER))
232 err = MZ_FORMAT_ERROR;
233 else if ((!local) && (magic != MZ_ZIP_MAGIC_CENTRALHEADER))
234 err = MZ_FORMAT_ERROR;
Viktor Szakats915b82e2018-04-24 10:02:39 +0000235
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800236 /* Read header fields */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700237 if (err == MZ_OK) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700238 if (!local)
239 err = mz_stream_read_uint16(stream, &file_info->version_madeby);
240 if (err == MZ_OK)
241 err = mz_stream_read_uint16(stream, &file_info->version_needed);
242 if (err == MZ_OK)
243 err = mz_stream_read_uint16(stream, &file_info->flag);
244 if (err == MZ_OK)
245 err = mz_stream_read_uint16(stream, &file_info->compression_method);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700246 if (err == MZ_OK) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700247 err = mz_stream_read_uint32(stream, &dos_date);
248 file_info->modified_date = mz_zip_dosdate_to_time_t(dos_date);
249 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700250 if (err == MZ_OK)
251 err = mz_stream_read_uint32(stream, &file_info->crc);
Nathan Moinvaziricf4029d2020-12-02 18:07:01 -0800252#ifdef HAVE_PKCRYPT
Nathan Moinvaziri9ffb08e2020-11-30 21:30:10 -0800253 if (err == MZ_OK && file_info->flag & MZ_ZIP_FLAG_ENCRYPTED) {
Nathan Moinvaziri215a7a82020-12-13 10:58:41 -0800254 /* Use dos_date from header instead of derived from time in zip extensions */
255 file_info->pk_verify = mz_zip_get_pk_verify(dos_date, file_info->crc, file_info->flag);
Nathan Moinvaziri9ffb08e2020-11-30 21:30:10 -0800256 }
Nathan Moinvaziricf4029d2020-12-02 18:07:01 -0800257#endif
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700258 if (err == MZ_OK) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700259 err = mz_stream_read_uint32(stream, &value32);
Nathan Moinvaziri2ecd8a02018-07-13 12:31:39 -0700260 file_info->compressed_size = value32;
261 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700262 if (err == MZ_OK) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700263 err = mz_stream_read_uint32(stream, &value32);
Nathan Moinvaziri2ecd8a02018-07-13 12:31:39 -0700264 file_info->uncompressed_size = value32;
265 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700266 if (err == MZ_OK)
267 err = mz_stream_read_uint16(stream, &file_info->filename_size);
268 if (err == MZ_OK)
269 err = mz_stream_read_uint16(stream, &file_info->extrafield_size);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700270 if (!local) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700271 if (err == MZ_OK)
272 err = mz_stream_read_uint16(stream, &file_info->comment_size);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700273 if (err == MZ_OK) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700274 err = mz_stream_read_uint16(stream, &value16);
Nathan Moinvaziri2ecd8a02018-07-13 12:31:39 -0700275 file_info->disk_number = value16;
276 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700277 if (err == MZ_OK)
278 err = mz_stream_read_uint16(stream, &file_info->internal_fa);
279 if (err == MZ_OK)
280 err = mz_stream_read_uint32(stream, &file_info->external_fa);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700281 if (err == MZ_OK) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700282 err = mz_stream_read_uint32(stream, &value32);
Nathan Moinvaziri2ecd8a02018-07-13 12:31:39 -0700283 file_info->disk_offset = value32;
284 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700285 }
286 }
287
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -0700288 if (err == MZ_OK)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800289 err = mz_stream_seek(file_extra_stream, 0, MZ_SEEK_SET);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700290
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800291 /* Copy variable length data to memory stream for later retrieval */
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700292 if ((err == MZ_OK) && (file_info->filename_size > 0))
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700293 err = mz_stream_copy(file_extra_stream, stream, file_info->filename_size);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800294 mz_stream_write_uint8(file_extra_stream, 0);
Nathan Moinvazirif9e34482018-11-27 09:45:09 -0800295 extrafield_pos = mz_stream_tell(file_extra_stream);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700296
Nathan Moinvaziri69e35492018-11-05 15:48:41 -0800297 if ((err == MZ_OK) && (file_info->extrafield_size > 0))
298 err = mz_stream_copy(file_extra_stream, stream, file_info->extrafield_size);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800299 mz_stream_write_uint8(file_extra_stream, 0);
Nathan Moinvaziri69e35492018-11-05 15:48:41 -0800300
Nathan Moinvazirif9e34482018-11-27 09:45:09 -0800301 comment_pos = mz_stream_tell(file_extra_stream);
Nathan Moinvaziri69e35492018-11-05 15:48:41 -0800302 if ((err == MZ_OK) && (file_info->comment_size > 0))
303 err = mz_stream_copy(file_extra_stream, stream, file_info->comment_size);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800304 mz_stream_write_uint8(file_extra_stream, 0);
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -0700305
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700306 linkname_pos = mz_stream_tell(file_extra_stream);
307 /* Overwrite if we encounter UNIX1 extra block */
308 mz_stream_write_uint8(file_extra_stream, 0);
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -0700309
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700310 if ((err == MZ_OK) && (file_info->extrafield_size > 0)) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800311 /* Seek to and parse the extra field */
Nathan Moinvaziri69e35492018-11-05 15:48:41 -0800312 err = mz_stream_seek(file_extra_stream, extrafield_pos, MZ_SEEK_SET);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700313
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700314 while ((err == MZ_OK) && (field_pos + 4 <= file_info->extrafield_size)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700315 err = mz_zip_extrafield_read(file_extra_stream, &field_type, &field_length);
Nathan Moinvaziriea2bd792018-11-19 17:44:44 -0800316 if (err != MZ_OK)
317 break;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800318 field_pos += 4;
Nathan Moinvazirifeb86c52018-11-05 11:40:21 -0800319
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800320 /* Don't allow field length to exceed size of remaining extrafield */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800321 if (field_length > (file_info->extrafield_size - field_pos))
Nathan Moinvaziri1d6d1832018-11-10 09:03:55 -0800322 field_length = (uint16_t)(file_info->extrafield_size - field_pos);
Nathan Moinvazirifeb86c52018-11-05 11:40:21 -0800323
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800324 /* Read ZIP64 extra field */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700325 if ((field_type == MZ_ZIP_EXTENSION_ZIP64) && (field_length >= 8)) {
326 if ((err == MZ_OK) && (file_info->uncompressed_size == UINT32_MAX)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700327 err = mz_stream_read_int64(file_extra_stream, &file_info->uncompressed_size);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800328 if (file_info->uncompressed_size < 0)
329 err = MZ_FORMAT_ERROR;
330 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700331 if ((err == MZ_OK) && (file_info->compressed_size == UINT32_MAX)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700332 err = mz_stream_read_int64(file_extra_stream, &file_info->compressed_size);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800333 if (file_info->compressed_size < 0)
334 err = MZ_FORMAT_ERROR;
335 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700336 if ((err == MZ_OK) && (file_info->disk_offset == UINT32_MAX)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700337 err = mz_stream_read_int64(file_extra_stream, &file_info->disk_offset);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800338 if (file_info->disk_offset < 0)
339 err = MZ_FORMAT_ERROR;
340 }
juanii4ae79922018-02-11 14:29:36 -0300341 if ((err == MZ_OK) && (file_info->disk_number == UINT16_MAX))
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700342 err = mz_stream_read_uint32(file_extra_stream, &file_info->disk_number);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700343 }
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800344 /* Read NTFS extra field */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700345 else if ((field_type == MZ_ZIP_EXTENSION_NTFS) && (field_length > 4)) {
Nathan Moinvaziri7a3b6982018-05-09 00:45:02 -0700346 if (err == MZ_OK)
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700347 err = mz_stream_read_uint32(file_extra_stream, &reserved);
348 field_length_read = 4;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700349
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700350 while ((err == MZ_OK) && (field_length_read + 4 <= field_length)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700351 err = mz_stream_read_uint16(file_extra_stream, &ntfs_attrib_id);
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700352 if (err == MZ_OK)
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700353 err = mz_stream_read_uint16(file_extra_stream, &ntfs_attrib_size);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800354 field_length_read += 4;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700355
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700356 if ((err == MZ_OK) && (ntfs_attrib_id == 0x01) && (ntfs_attrib_size == 24)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700357 err = mz_stream_read_uint64(file_extra_stream, &ntfs_time);
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700358 mz_zip_ntfs_to_unix_time(ntfs_time, &file_info->modified_date);
359
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700360 if (err == MZ_OK) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700361 err = mz_stream_read_uint64(file_extra_stream, &ntfs_time);
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700362 mz_zip_ntfs_to_unix_time(ntfs_time, &file_info->accessed_date);
363 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700364 if (err == MZ_OK) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700365 err = mz_stream_read_uint64(file_extra_stream, &ntfs_time);
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700366 mz_zip_ntfs_to_unix_time(ntfs_time, &file_info->creation_date);
367 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700368 } else if ((err == MZ_OK) && (field_length_read + ntfs_attrib_size <= field_length)) {
Nathan Moinvaziri60008442018-11-07 06:29:01 -0800369 err = mz_stream_seek(file_extra_stream, ntfs_attrib_size, MZ_SEEK_CUR);
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700370 }
371
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800372 field_length_read += ntfs_attrib_size;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700373 }
374 }
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800375 /* Read UNIX1 extra field */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700376 else if ((field_type == MZ_ZIP_EXTENSION_UNIX1) && (field_length >= 12)) {
377 if (err == MZ_OK) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700378 err = mz_stream_read_uint32(file_extra_stream, &value32);
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700379 if (err == MZ_OK && file_info->accessed_date == 0)
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -0700380 file_info->accessed_date = value32;
381 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700382 if (err == MZ_OK) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700383 err = mz_stream_read_uint32(file_extra_stream, &value32);
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700384 if (err == MZ_OK && file_info->modified_date == 0)
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -0700385 file_info->modified_date = value32;
386 }
387 if (err == MZ_OK)
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800388 err = mz_stream_read_uint16(file_extra_stream, &value16); /* User id */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -0700389 if (err == MZ_OK)
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800390 err = mz_stream_read_uint16(file_extra_stream, &value16); /* Group id */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -0700391
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700392 /* Copy linkname to end of file extra stream so we can return null
393 terminated string */
394 linkname_size = field_length - 12;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700395 if ((err == MZ_OK) && (linkname_size > 0)) {
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700396 linkname = (char *)MZ_ALLOC(linkname_size);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700397 if (linkname != NULL) {
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700398 if (mz_stream_read(file_extra_stream, linkname, linkname_size) != linkname_size)
399 err = MZ_READ_ERROR;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700400 if (err == MZ_OK) {
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700401 saved_pos = mz_stream_tell(file_extra_stream);
402
403 mz_stream_seek(file_extra_stream, linkname_pos, MZ_SEEK_SET);
404 mz_stream_write(file_extra_stream, linkname, linkname_size);
405 mz_stream_write_uint8(file_extra_stream, 0);
406
407 mz_stream_seek(file_extra_stream, saved_pos, MZ_SEEK_SET);
408 }
409 MZ_FREE(linkname);
410 }
411 }
Nathan Moinvaziri413822a2018-10-20 09:45:07 -0700412 }
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -0800413#ifdef HAVE_WZAES
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800414 /* Read AES extra field */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700415 else if ((field_type == MZ_ZIP_EXTENSION_AES) && (field_length == 7)) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700416 uint8_t value8 = 0;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800417 /* Verify version info */
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700418 err = mz_stream_read_uint16(file_extra_stream, &value16);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800419 /* Support AE-1 and AE-2 */
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700420 if (value16 != 1 && value16 != 2)
421 err = MZ_FORMAT_ERROR;
422 file_info->aes_version = value16;
423 if (err == MZ_OK)
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700424 err = mz_stream_read_uint8(file_extra_stream, &value8);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700425 if ((char)value8 != 'A')
426 err = MZ_FORMAT_ERROR;
427 if (err == MZ_OK)
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700428 err = mz_stream_read_uint8(file_extra_stream, &value8);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700429 if ((char)value8 != 'E')
430 err = MZ_FORMAT_ERROR;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800431 /* Get AES encryption strength and actual compression method */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700432 if (err == MZ_OK) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700433 err = mz_stream_read_uint8(file_extra_stream, &value8);
Nathan Moinvaziri2ecd8a02018-07-13 12:31:39 -0700434 file_info->aes_encryption_mode = value8;
435 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700436 if (err == MZ_OK) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700437 err = mz_stream_read_uint16(file_extra_stream, &value16);
Nathan Moinvaziri2ecd8a02018-07-13 12:31:39 -0700438 file_info->compression_method = value16;
439 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700440 }
441#endif
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700442 else if (field_length > 0) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700443 err = mz_stream_seek(file_extra_stream, field_length, MZ_SEEK_CUR);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700444 }
445
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800446 field_pos += field_length;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700447 }
448 }
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -0700449
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800450 /* Get pointers to variable length data */
Nathan Moinvaziri60008442018-11-07 06:29:01 -0800451 mz_stream_mem_get_buffer(file_extra_stream, (const void **)&file_info->filename);
Nathan Moinvazirif9e34482018-11-27 09:45:09 -0800452 mz_stream_mem_get_buffer_at(file_extra_stream, extrafield_pos, (const void **)&file_info->extrafield);
453 mz_stream_mem_get_buffer_at(file_extra_stream, comment_pos, (const void **)&file_info->comment);
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700454 mz_stream_mem_get_buffer_at(file_extra_stream, linkname_pos, (const void **)&file_info->linkname);
Nathan Moinvaziri60008442018-11-07 06:29:01 -0800455
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800456 /* Set to empty string just in-case */
Nathan Moinvaziri60008442018-11-07 06:29:01 -0800457 if (file_info->filename == NULL)
458 file_info->filename = "";
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800459 if (file_info->extrafield == NULL)
460 file_info->extrafield_size = 0;
Nathan Moinvaziri60008442018-11-07 06:29:01 -0800461 if (file_info->comment == NULL)
462 file_info->comment = "";
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700463 if (file_info->linkname == NULL)
464 file_info->linkname = "";
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700465
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700466 if (err == MZ_OK) {
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -0700467 mz_zip_print("Zip - Entry - Read header - %s (local %" PRId8 ")\n",
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800468 file_info->filename, local);
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -0700469 mz_zip_print("Zip - Entry - Read header compress (ucs %" PRId64 " cs %" PRId64 " crc 0x%08" PRIx32 ")\n",
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800470 file_info->uncompressed_size, file_info->compressed_size, file_info->crc);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700471 if (!local) {
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -0700472 mz_zip_print("Zip - Entry - Read header disk (disk %" PRIu32 " offset %" PRId64 ")\n",
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800473 file_info->disk_number, file_info->disk_offset);
Nathan Moinvaziria93a2ad2018-11-13 17:51:42 -0800474 }
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -0700475 mz_zip_print("Zip - Entry - Read header variable (fnl %" PRId32 " efs %" PRId32 " cms %" PRId32 ")\n",
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800476 file_info->filename_size, file_info->extrafield_size, file_info->comment_size);
477 }
478
479 return err;
480}
481
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700482static int32_t mz_zip_entry_read_descriptor(void *stream, uint8_t zip64, uint32_t *crc32, int64_t *compressed_size, int64_t *uncompressed_size) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800483 uint32_t value32 = 0;
484 int64_t value64 = 0;
485 int32_t err = MZ_OK;
486
487
488 err = mz_stream_read_uint32(stream, &value32);
489 if (value32 != MZ_ZIP_MAGIC_DATADESCRIPTOR)
490 err = MZ_FORMAT_ERROR;
491 if (err == MZ_OK)
492 err = mz_stream_read_uint32(stream, &value32);
493 if ((err == MZ_OK) && (crc32 != NULL))
494 *crc32 = value32;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700495 if (err == MZ_OK) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800496 /* If zip 64 extension is enabled then read as 8 byte */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700497 if (!zip64) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800498 err = mz_stream_read_uint32(stream, &value32);
499 value64 = value32;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700500 } else {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800501 err = mz_stream_read_int64(stream, &value64);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800502 if (value64 < 0)
503 err = MZ_FORMAT_ERROR;
504 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800505 if ((err == MZ_OK) && (compressed_size != NULL))
506 *compressed_size = value64;
507 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700508 if (err == MZ_OK) {
509 if (!zip64) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800510 err = mz_stream_read_uint32(stream, &value32);
511 value64 = value32;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700512 } else {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800513 err = mz_stream_read_int64(stream, &value64);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800514 if (value64 < 0)
515 err = MZ_FORMAT_ERROR;
516 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800517 if ((err == MZ_OK) && (uncompressed_size != NULL))
518 *uncompressed_size = value64;
519 }
520
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700521 return err;
522}
523
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800524static int32_t mz_zip_entry_write_crc_sizes(void *stream, uint8_t zip64, uint8_t mask, mz_zip_file *file_info) {
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -0800525 int32_t err = MZ_OK;
526
527 if (mask)
528 err = mz_stream_write_uint32(stream, 0);
529 else
530 err = mz_stream_write_uint32(stream, file_info->crc); /* crc */
531
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800532 /* For backwards-compatibility with older zip applications we set all sizes to UINT32_MAX
533 * when zip64 is needed, instead of only setting sizes larger than UINT32_MAX. */
534
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700535 if (err == MZ_OK) {
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800536 if (zip64) /* compr size */
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -0800537 err = mz_stream_write_uint32(stream, UINT32_MAX);
538 else
539 err = mz_stream_write_uint32(stream, (uint32_t)file_info->compressed_size);
540 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700541 if (err == MZ_OK) {
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800542 if (mask) /* uncompr size */
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -0800543 err = mz_stream_write_uint32(stream, 0);
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800544 else if (zip64)
545 err = mz_stream_write_uint32(stream, UINT32_MAX);
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -0800546 else
547 err = mz_stream_write_uint32(stream, (uint32_t)file_info->uncompressed_size);
548 }
549 return err;
550}
551
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800552static int32_t mz_zip_entry_needs_zip64(mz_zip_file *file_info, uint8_t local, uint8_t *zip64) {
553 uint8_t needs_zip64 = 0;
Vlad Lipskiy4b252192020-12-23 18:35:37 +0300554 uint32_t max_uncompressed_size = UINT32_MAX;
555
556 /* At local header we might not know yet whether compressed_size will overflow unsigned
557 32-bit integer which might happen for high entropy data, so we add a 2MB buffer */
558 if (local)
559 max_uncompressed_size -= (2 * 1024 * 1024);
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800560
561 if (zip64 == NULL)
562 return MZ_PARAM_ERROR;
563
Vlad Lipskiy4b252192020-12-23 18:35:37 +0300564 needs_zip64 = (file_info->uncompressed_size >= max_uncompressed_size) ||
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800565 (file_info->compressed_size >= UINT32_MAX) ||
566 (file_info->disk_offset >= UINT32_MAX);
567
568 if (file_info->zip64 == MZ_ZIP64_AUTO) {
569 /* If uncompressed size is unknown, assume zip64 for 64-bit data descriptors */
570 *zip64 = (local && file_info->uncompressed_size == 0) || (needs_zip64);
571 } else if (file_info->zip64 == MZ_ZIP64_FORCE) {
572 *zip64 = 1;
573 } else if (file_info->zip64 == MZ_ZIP64_DISABLE) {
574 /* Zip64 extension is required to zip file */
575 if (needs_zip64)
576 return MZ_PARAM_ERROR;
577
578 *zip64 = 0;
579 }
580
581 return MZ_OK;
582}
583
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700584static int32_t mz_zip_entry_write_header(void *stream, uint8_t local, mz_zip_file *file_info) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700585 uint64_t ntfs_time = 0;
586 uint32_t reserved = 0;
587 uint32_t dos_date = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700588 uint16_t extrafield_size = 0;
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700589 uint16_t field_type = 0;
590 uint16_t field_length = 0;
591 uint16_t field_length_zip64 = 0;
592 uint16_t field_length_ntfs = 0;
593 uint16_t field_length_aes = 0;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700594 uint16_t field_length_unix1 = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700595 uint16_t filename_size = 0;
Nathan Moinvaziri3ad1a922018-05-02 13:04:41 -0700596 uint16_t filename_length = 0;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700597 uint16_t linkname_size = 0;
Nathan Moinvaziri46f71432018-05-03 17:24:32 -0700598 uint16_t version_needed = 0;
Nathan Moinvaziridd103f42018-11-13 14:50:32 -0800599 int32_t comment_size = 0;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -0700600 int32_t err = MZ_OK;
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700601 int32_t err_mem = MZ_OK;
602 uint8_t zip64 = 0;
603 uint8_t skip_aes = 0;
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700604 uint8_t mask = 0;
605 uint8_t write_end_slash = 0;
606 const char *filename = NULL;
Nathan Moinvazirifa620e42018-10-20 10:37:10 -0700607 char masked_name[64];
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700608 void *file_extra_stream = NULL;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700609
610 if (file_info == NULL)
611 return MZ_PARAM_ERROR;
612
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700613 if ((local) && (file_info->flag & MZ_ZIP_FLAG_MASK_LOCAL_INFO))
614 mask = 1;
615
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800616 /* Determine if zip64 extension is necessary */
617 err = mz_zip_entry_needs_zip64(file_info, local, &zip64);
618 if (err != MZ_OK)
619 return err;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -0700620
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800621 /* Start calculating extra field sizes */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700622 if (zip64) {
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -0800623 field_length_zip64 = 8 + 8;
624 if ((!local) && (file_info->disk_offset >= UINT32_MAX))
625 field_length_zip64 += 8;
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800626
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700627 extrafield_size += 4;
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700628 extrafield_size += field_length_zip64;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700629 }
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700630
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800631 /* Calculate extra field size and check for duplicates */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700632 if (file_info->extrafield_size > 0) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700633 mz_stream_mem_create(&file_extra_stream);
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -0700634 mz_stream_mem_set_buffer(file_extra_stream, (void *)file_info->extrafield,
Nathan Moinvaziri915c5132018-10-26 20:00:52 -0700635 file_info->extrafield_size);
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700636
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700637 do {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700638 err_mem = mz_stream_read_uint16(file_extra_stream, &field_type);
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700639 if (err_mem == MZ_OK)
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700640 err_mem = mz_stream_read_uint16(file_extra_stream, &field_length);
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700641 if (err_mem != MZ_OK)
642 break;
643
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800644 /* Prefer incoming aes extensions over ours */
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700645 if (field_type == MZ_ZIP_EXTENSION_AES)
646 skip_aes = 1;
647
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700648 /* Prefer our zip64, ntfs, unix1 extension over incoming */
649 if (field_type != MZ_ZIP_EXTENSION_ZIP64 && field_type != MZ_ZIP_EXTENSION_NTFS &&
650 field_type != MZ_ZIP_EXTENSION_UNIX1)
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700651 extrafield_size += 4 + field_length;
652
653 if (err_mem == MZ_OK)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800654 err_mem = mz_stream_seek(file_extra_stream, field_length, MZ_SEEK_CUR);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700655 } while (err_mem == MZ_OK);
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700656 }
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -0700657
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -0800658#ifdef HAVE_WZAES
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700659 if (!skip_aes) {
660 if ((file_info->flag & MZ_ZIP_FLAG_ENCRYPTED) && (file_info->aes_version)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700661 field_length_aes = 1 + 1 + 1 + 2 + 2;
662 extrafield_size += 4 + field_length_aes;
Nathan Moinvaziri413822a2018-10-20 09:45:07 -0700663 }
664 }
Nathan Moinvaziria2bb08c2018-10-28 13:45:30 -0700665#else
Nathan Moinvaziri7de94cf2018-11-19 20:51:06 -0800666 MZ_UNUSED(field_length_aes);
Nathan Moinvaziri4ae469a2018-10-28 15:32:04 -0700667 MZ_UNUSED(skip_aes);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700668#endif
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800669 /* NTFS timestamps */
Nathan Moinvazirid9e159a2018-02-13 15:30:30 -0800670 if ((file_info->modified_date != 0) &&
671 (file_info->accessed_date != 0) &&
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700672 (file_info->creation_date != 0) && (!mask)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700673 field_length_ntfs = 8 + 8 + 8 + 4 + 2 + 2;
674 extrafield_size += 4 + field_length_ntfs;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700675 }
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -0700676
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700677 /* Unix1 symbolic links */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700678 if (file_info->linkname != NULL && *file_info->linkname != 0) {
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700679 linkname_size = (uint16_t)strlen(file_info->linkname);
680 field_length_unix1 = 12 + linkname_size;
681 extrafield_size += 4 + field_length_unix1;
682 }
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -0700683
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700684 if (local)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700685 err = mz_stream_write_uint32(stream, MZ_ZIP_MAGIC_LOCALHEADER);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700686 else {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700687 err = mz_stream_write_uint32(stream, MZ_ZIP_MAGIC_CENTRALHEADER);
688 if (err == MZ_OK)
689 err = mz_stream_write_uint16(stream, file_info->version_madeby);
690 }
691
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800692 /* Calculate version needed to extract */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700693 if (err == MZ_OK) {
Nathan Moinvaziri46f71432018-05-03 17:24:32 -0700694 version_needed = file_info->version_needed;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700695 if (version_needed == 0) {
Nathan Moinvaziri46f71432018-05-03 17:24:32 -0700696 version_needed = 20;
697 if (zip64)
698 version_needed = 45;
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -0800699#ifdef HAVE_WZAES
Nathan Moinvaziri46f71432018-05-03 17:24:32 -0700700 if ((file_info->flag & MZ_ZIP_FLAG_ENCRYPTED) && (file_info->aes_version))
701 version_needed = 51;
702#endif
Nathan Moinvaziri1a7e7e42020-10-25 14:50:53 -0700703#if defined(HAVE_LZMA) || defined(HAVE_LIBCOMP)
Nathan Moinvaziric4668ed2020-10-25 11:56:19 -0700704 if ((file_info->compression_method == MZ_COMPRESS_METHOD_LZMA) ||
705 (file_info->compression_method == MZ_COMPRESS_METHOD_XZ))
Nathan Moinvaziri46f71432018-05-03 17:24:32 -0700706 version_needed = 63;
707#endif
708 }
709 err = mz_stream_write_uint16(stream, version_needed);
710 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700711 if (err == MZ_OK)
712 err = mz_stream_write_uint16(stream, file_info->flag);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700713 if (err == MZ_OK) {
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -0800714#ifdef HAVE_WZAES
Nathan Moinvaziri413822a2018-10-20 09:45:07 -0700715 if ((file_info->flag & MZ_ZIP_FLAG_ENCRYPTED) && (file_info->aes_version))
Nathan Moinvaziri6e21e862017-10-19 09:57:54 -0700716 err = mz_stream_write_uint16(stream, MZ_COMPRESS_METHOD_AES);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700717 else
718#endif
719 err = mz_stream_write_uint16(stream, file_info->compression_method);
720 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700721 if (err == MZ_OK) {
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700722 if (file_info->modified_date != 0 && !mask)
Nathan Moinvaziri17cdaec2017-10-26 10:18:41 -0700723 dos_date = mz_zip_time_t_to_dos_date(file_info->modified_date);
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700724 err = mz_stream_write_uint32(stream, dos_date);
725 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700726
727 if (err == MZ_OK)
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800728 err = mz_zip_entry_write_crc_sizes(stream, zip64, mask, file_info);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700729
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700730 if (mask) {
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -0700731 snprintf(masked_name, sizeof(masked_name), "%" PRIx32 "_%" PRIx64,
Nathan Moinvaziri264dc182018-11-13 17:42:13 -0800732 file_info->disk_number, file_info->disk_offset);
Nathan Moinvazirifa620e42018-10-20 10:37:10 -0700733 filename = masked_name;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700734 } else {
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700735 filename = file_info->filename;
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700736 }
737
738 filename_length = (uint16_t)strlen(filename);
739 filename_size += filename_length;
740
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800741 if ((mz_zip_attrib_is_dir(file_info->external_fa, file_info->version_madeby) == MZ_OK) &&
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700742 ((filename[filename_length - 1] != '/') && (filename[filename_length - 1] != '\\'))) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800743 filename_size += 1;
744 write_end_slash = 1;
745 }
746
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700747 if (err == MZ_OK)
748 err = mz_stream_write_uint16(stream, filename_size);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700749 if (err == MZ_OK)
750 err = mz_stream_write_uint16(stream, extrafield_size);
751
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700752 if (!local) {
753 if (file_info->comment != NULL) {
Nathan Moinvaziridd103f42018-11-13 14:50:32 -0800754 comment_size = (int32_t)strlen(file_info->comment);
755 if (comment_size > UINT16_MAX)
756 comment_size = UINT16_MAX;
757 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700758 if (err == MZ_OK)
Nathan Moinvaziridd103f42018-11-13 14:50:32 -0800759 err = mz_stream_write_uint16(stream, (uint16_t)comment_size);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700760 if (err == MZ_OK)
Nathan Moinvazirifd039e32017-10-22 14:40:39 -0700761 err = mz_stream_write_uint16(stream, (uint16_t)file_info->disk_number);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700762 if (err == MZ_OK)
763 err = mz_stream_write_uint16(stream, file_info->internal_fa);
764 if (err == MZ_OK)
765 err = mz_stream_write_uint32(stream, file_info->external_fa);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700766 if (err == MZ_OK) {
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -0800767 if (file_info->disk_offset >= UINT32_MAX)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700768 err = mz_stream_write_uint32(stream, UINT32_MAX);
769 else
770 err = mz_stream_write_uint32(stream, (uint32_t)file_info->disk_offset);
771 }
772 }
773
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700774 if (err == MZ_OK) {
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700775 if (mz_stream_write(stream, filename, filename_length) != filename_length)
Nathan Moinvaziri05e03ca2018-10-28 16:15:13 -0700776 err = MZ_WRITE_ERROR;
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700777
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800778 /* Ensure that directories have a slash appended to them for compatibility */
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700779 if (err == MZ_OK && write_end_slash)
780 err = mz_stream_write_uint8(stream, '/');
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700781 }
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700782
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700783 if (file_info->extrafield_size > 0) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800784 err_mem = mz_stream_mem_seek(file_extra_stream, 0, MZ_SEEK_SET);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700785 while (err == MZ_OK && err_mem == MZ_OK) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700786 err_mem = mz_stream_read_uint16(file_extra_stream, &field_type);
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700787 if (err_mem == MZ_OK)
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700788 err_mem = mz_stream_read_uint16(file_extra_stream, &field_length);
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700789 if (err_mem != MZ_OK)
790 break;
791
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700792 /* Prefer our zip 64, ntfs, unix1 extensions over incoming */
793 if (field_type == MZ_ZIP_EXTENSION_ZIP64 || field_type == MZ_ZIP_EXTENSION_NTFS ||
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700794 field_type == MZ_ZIP_EXTENSION_UNIX1) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800795 err_mem = mz_stream_seek(file_extra_stream, field_length, MZ_SEEK_CUR);
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700796 continue;
797 }
798
799 err = mz_stream_write_uint16(stream, field_type);
800 if (err == MZ_OK)
801 err = mz_stream_write_uint16(stream, field_length);
802 if (err == MZ_OK)
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700803 err = mz_stream_copy(stream, file_extra_stream, field_length);
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700804 }
805
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700806 mz_stream_mem_delete(&file_extra_stream);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700807 }
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700808
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800809 /* Write ZIP64 extra field */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700810 if ((err == MZ_OK) && (zip64)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700811 err = mz_zip_extrafield_write(stream, MZ_ZIP_EXTENSION_ZIP64, field_length_zip64);
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800812 if (err == MZ_OK) {
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700813 if (mask)
814 err = mz_stream_write_int64(stream, 0);
815 else
816 err = mz_stream_write_int64(stream, file_info->uncompressed_size);
817 }
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800818 if (err == MZ_OK)
Nathan Moinvaziri904c4082018-10-08 23:31:21 -0700819 err = mz_stream_write_int64(stream, file_info->compressed_size);
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -0800820 if ((err == MZ_OK) && (!local) && (file_info->disk_offset >= UINT32_MAX))
Nathan Moinvaziri904c4082018-10-08 23:31:21 -0700821 err = mz_stream_write_int64(stream, file_info->disk_offset);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700822 }
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800823 /* Write NTFS extra field */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700824 if ((err == MZ_OK) && (field_length_ntfs > 0)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700825 err = mz_zip_extrafield_write(stream, MZ_ZIP_EXTENSION_NTFS, field_length_ntfs);
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700826 if (err == MZ_OK)
827 err = mz_stream_write_uint32(stream, reserved);
828 if (err == MZ_OK)
829 err = mz_stream_write_uint16(stream, 0x01);
830 if (err == MZ_OK)
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700831 err = mz_stream_write_uint16(stream, field_length_ntfs - 8);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700832 if (err == MZ_OK) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700833 mz_zip_unix_to_ntfs_time(file_info->modified_date, &ntfs_time);
834 err = mz_stream_write_uint64(stream, ntfs_time);
835 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700836 if (err == MZ_OK) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700837 mz_zip_unix_to_ntfs_time(file_info->accessed_date, &ntfs_time);
838 err = mz_stream_write_uint64(stream, ntfs_time);
839 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700840 if (err == MZ_OK) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700841 mz_zip_unix_to_ntfs_time(file_info->creation_date, &ntfs_time);
842 err = mz_stream_write_uint64(stream, ntfs_time);
843 }
844 }
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700845 /* Write UNIX extra block extra field */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700846 if ((err == MZ_OK) && (field_length_unix1 > 0)) {
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700847 err = mz_zip_extrafield_write(stream, MZ_ZIP_EXTENSION_UNIX1, field_length_unix1);
848 if (err == MZ_OK)
Nathan Moinvaziriaeb6cdd2019-05-06 13:40:32 -0700849 err = mz_stream_write_uint32(stream, (uint32_t)file_info->accessed_date);
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700850 if (err == MZ_OK)
Nathan Moinvaziriaeb6cdd2019-05-06 13:40:32 -0700851 err = mz_stream_write_uint32(stream, (uint32_t)file_info->modified_date);
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700852 if (err == MZ_OK) /* User id */
853 err = mz_stream_write_uint16(stream, 0);
854 if (err == MZ_OK) /* Group id */
855 err = mz_stream_write_uint16(stream, 0);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700856 if (err == MZ_OK && linkname_size > 0) {
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700857 if (mz_stream_write(stream, file_info->linkname, linkname_size) != linkname_size)
858 err = MZ_WRITE_ERROR;
859 }
860 }
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -0800861#ifdef HAVE_WZAES
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800862 /* Write AES extra field */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700863 if ((err == MZ_OK) && (!skip_aes) && (file_info->flag & MZ_ZIP_FLAG_ENCRYPTED) && (file_info->aes_version)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700864 err = mz_zip_extrafield_write(stream, MZ_ZIP_EXTENSION_AES, field_length_aes);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700865 if (err == MZ_OK)
866 err = mz_stream_write_uint16(stream, file_info->aes_version);
867 if (err == MZ_OK)
868 err = mz_stream_write_uint8(stream, 'A');
869 if (err == MZ_OK)
870 err = mz_stream_write_uint8(stream, 'E');
871 if (err == MZ_OK)
872 err = mz_stream_write_uint8(stream, file_info->aes_encryption_mode);
873 if (err == MZ_OK)
874 err = mz_stream_write_uint16(stream, file_info->compression_method);
875 }
876#endif
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700877 if ((err == MZ_OK) && (!local) && (file_info->comment != NULL)) {
Nathan Moinvazirica014e62018-08-15 07:31:12 -0700878 if (mz_stream_write(stream, file_info->comment, file_info->comment_size) != file_info->comment_size)
Nathan Moinvaziri05e03ca2018-10-28 16:15:13 -0700879 err = MZ_WRITE_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700880 }
881
882 return err;
883}
884
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700885static int32_t mz_zip_entry_write_descriptor(void *stream, uint8_t zip64, uint32_t crc32, int64_t compressed_size, int64_t uncompressed_size) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800886 int32_t err = MZ_OK;
887
888 err = mz_stream_write_uint32(stream, MZ_ZIP_MAGIC_DATADESCRIPTOR);
889 if (err == MZ_OK)
890 err = mz_stream_write_uint32(stream, crc32);
891
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800892 /* Store data descriptor as 8 bytes if zip 64 extension enabled */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700893 if (err == MZ_OK) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800894 /* Zip 64 extension is enabled when uncompressed size is > UINT32_MAX */
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800895 if (!zip64)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800896 err = mz_stream_write_uint32(stream, (uint32_t)compressed_size);
897 else
898 err = mz_stream_write_int64(stream, compressed_size);
899 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700900 if (err == MZ_OK) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800901 if (!zip64)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800902 err = mz_stream_write_uint32(stream, (uint32_t)uncompressed_size);
903 else
904 err = mz_stream_write_int64(stream, uncompressed_size);
905 }
906
907 return err;
908}
909
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700910static int32_t mz_zip_read_cd(void *handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800911 mz_zip *zip = (mz_zip *)handle;
912 uint64_t number_entry_cd64 = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800913 uint64_t number_entry_cd = 0;
914 int64_t eocd_pos = 0;
915 int64_t eocd_pos64 = 0;
916 int64_t value64i = 0;
917 uint16_t value16 = 0;
918 uint32_t value32 = 0;
919 uint64_t value64 = 0;
920 uint16_t comment_size = 0;
921 int32_t comment_read = 0;
922 int32_t err = MZ_OK;
923
924
925 if (zip == NULL)
926 return MZ_PARAM_ERROR;
927
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800928 /* Read and cache central directory records */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800929 err = mz_zip_search_eocd(zip->stream, &eocd_pos);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700930 if (err == MZ_OK) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800931 /* The signature, already checked */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800932 err = mz_stream_read_uint32(zip->stream, &value32);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800933 /* Number of this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800934 if (err == MZ_OK)
935 err = mz_stream_read_uint16(zip->stream, &value16);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800936 /* Number of the disk with the start of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800937 if (err == MZ_OK)
938 err = mz_stream_read_uint16(zip->stream, &value16);
939 zip->disk_number_with_cd = value16;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800940 /* Total number of entries in the central dir on this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800941 if (err == MZ_OK)
942 err = mz_stream_read_uint16(zip->stream, &value16);
943 zip->number_entry = value16;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800944 /* Total number of entries in the central dir */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800945 if (err == MZ_OK)
946 err = mz_stream_read_uint16(zip->stream, &value16);
947 number_entry_cd = value16;
948 if (number_entry_cd != zip->number_entry)
949 err = MZ_FORMAT_ERROR;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800950 /* Size of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800951 if (err == MZ_OK)
952 err = mz_stream_read_uint32(zip->stream, &value32);
953 if (err == MZ_OK)
954 zip->cd_size = value32;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800955 /* Offset of start of central directory with respect to the starting disk number */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800956 if (err == MZ_OK)
957 err = mz_stream_read_uint32(zip->stream, &value32);
958 if (err == MZ_OK)
959 zip->cd_offset = value32;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800960 /* Zip file global comment length */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800961 if (err == MZ_OK)
962 err = mz_stream_read_uint16(zip->stream, &comment_size);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700963 if ((err == MZ_OK) && (comment_size > 0)) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800964 zip->comment = (char *)MZ_ALLOC(comment_size + 1);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700965 if (zip->comment != NULL) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800966 comment_read = mz_stream_read(zip->stream, zip->comment, comment_size);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800967 /* Don't fail if incorrect comment length read, not critical */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800968 if (comment_read < 0)
969 comment_read = 0;
970 zip->comment[comment_read] = 0;
971 }
972 }
973
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700974 if ((err == MZ_OK) && ((number_entry_cd == UINT16_MAX) || (zip->cd_offset == UINT32_MAX))) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800975 /* Format should be Zip64, as the central directory or file size is too large */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700976 if (mz_zip_search_zip64_eocd(zip->stream, eocd_pos, &eocd_pos64) == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800977 eocd_pos = eocd_pos64;
978
979 err = mz_stream_seek(zip->stream, eocd_pos, MZ_SEEK_SET);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800980 /* The signature, already checked */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800981 if (err == MZ_OK)
982 err = mz_stream_read_uint32(zip->stream, &value32);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800983 /* Size of zip64 end of central directory record */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800984 if (err == MZ_OK)
985 err = mz_stream_read_uint64(zip->stream, &value64);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800986 /* Version made by */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800987 if (err == MZ_OK)
988 err = mz_stream_read_uint16(zip->stream, &zip->version_madeby);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800989 /* Version needed to extract */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800990 if (err == MZ_OK)
991 err = mz_stream_read_uint16(zip->stream, &value16);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800992 /* Number of this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800993 if (err == MZ_OK)
994 err = mz_stream_read_uint32(zip->stream, &value32);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800995 /* Number of the disk with the start of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800996 if (err == MZ_OK)
997 err = mz_stream_read_uint32(zip->stream, &zip->disk_number_with_cd);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800998 /* Total number of entries in the central directory on this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800999 if (err == MZ_OK)
Gabriel Dubois3b3961e2020-03-20 15:48:40 +00001000 err = mz_stream_read_uint64(zip->stream, &zip->number_entry);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001001 /* Total number of entries in the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001002 if (err == MZ_OK)
1003 err = mz_stream_read_uint64(zip->stream, &number_entry_cd64);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001004 if (zip->number_entry != number_entry_cd64)
Gabriel Dubois3b3961e2020-03-20 15:48:40 +00001005 err = MZ_FORMAT_ERROR;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001006 /* Size of the central directory */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001007 if (err == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001008 err = mz_stream_read_int64(zip->stream, &zip->cd_size);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08001009 if (zip->cd_size < 0)
1010 err = MZ_FORMAT_ERROR;
1011 }
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001012 /* Offset of start of central directory with respect to the starting disk number */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001013 if (err == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001014 err = mz_stream_read_int64(zip->stream, &zip->cd_offset);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08001015 if (zip->cd_offset < 0)
1016 err = MZ_FORMAT_ERROR;
1017 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001018 } else if ((zip->number_entry == UINT16_MAX) || (number_entry_cd != zip->number_entry) ||
1019 (zip->cd_size == UINT16_MAX) || (zip->cd_offset == UINT32_MAX)) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001020 err = MZ_FORMAT_ERROR;
1021 }
1022 }
1023 }
1024
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001025 if (err == MZ_OK) {
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07001026 mz_zip_print("Zip - Read cd (disk %" PRId32 " entries %" PRId64 " offset %" PRId64 " size %" PRId64 ")\n",
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001027 zip->disk_number_with_cd, zip->number_entry, zip->cd_offset, zip->cd_size);
1028
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001029 /* Verify central directory signature exists at offset */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001030 err = mz_stream_seek(zip->stream, zip->cd_offset, MZ_SEEK_SET);
1031 if (err == MZ_OK)
Nathan Moinvaziri0e1fd892019-09-28 20:55:37 -07001032 err = mz_stream_read_uint32(zip->stream, &zip->cd_signature);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001033 if ((err == MZ_OK) && (zip->cd_signature != MZ_ZIP_MAGIC_CENTRALHEADER)) {
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001034 /* If cd exists in large file and no zip-64 support, error for recover */
1035 if (eocd_pos > UINT32_MAX && eocd_pos64 == 0)
1036 err = MZ_FORMAT_ERROR;
Nathan Moinvaziri0e1fd892019-09-28 20:55:37 -07001037 /* If cd not found attempt to seek backward to find it */
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001038 if (err == MZ_OK)
1039 err = mz_stream_seek(zip->stream, eocd_pos - zip->cd_size, MZ_SEEK_SET);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001040 if (err == MZ_OK)
Nathan Moinvaziri0e1fd892019-09-28 20:55:37 -07001041 err = mz_stream_read_uint32(zip->stream, &zip->cd_signature);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001042 if ((err == MZ_OK) && (zip->cd_signature == MZ_ZIP_MAGIC_CENTRALHEADER)) {
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001043
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001044 /* If found compensate for incorrect locations */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001045 value64i = zip->cd_offset;
1046 zip->cd_offset = eocd_pos - zip->cd_size;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001047 /* Assume disk has prepended data */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001048 zip->disk_offset_shift = zip->cd_offset - value64i;
1049 }
1050 }
1051 }
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001052
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001053 if (err == MZ_OK) {
1054 if (eocd_pos < zip->cd_offset) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001055 /* End of central dir should always come after central dir */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001056 err = MZ_FORMAT_ERROR;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001057 } else if (eocd_pos < zip->cd_offset + zip->cd_size) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001058 /* Truncate size of cd if incorrect size or offset provided */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001059 zip->cd_size = eocd_pos - zip->cd_offset;
1060 }
1061 }
1062
1063 return err;
1064}
1065
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001066static int32_t mz_zip_write_cd(void *handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001067 mz_zip *zip = (mz_zip *)handle;
1068 int64_t zip64_eocd_pos_inzip = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001069 int64_t disk_number = 0;
1070 int64_t disk_size = 0;
Nathan Moinvaziridd103f42018-11-13 14:50:32 -08001071 int32_t comment_size = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001072 int32_t err = MZ_OK;
1073
1074
1075 if (zip == NULL)
1076 return MZ_PARAM_ERROR;
1077
1078 if (mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, &disk_number) == MZ_OK)
1079 zip->disk_number_with_cd = (uint32_t)disk_number;
1080 if (mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_SIZE, &disk_size) == MZ_OK && disk_size > 0)
1081 zip->disk_number_with_cd += 1;
1082 mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, -1);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001083 if ((zip->disk_number_with_cd > 0) && (zip->open_mode & MZ_OPEN_MODE_APPEND)) {
Nathan Moinvazirif286b832019-11-10 17:09:48 -08001084 // Overwrite existing central directory if using split disks
1085 mz_stream_seek(zip->stream, 0, MZ_SEEK_SET);
1086 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001087
1088 zip->cd_offset = mz_stream_tell(zip->stream);
1089 mz_stream_seek(zip->cd_mem_stream, 0, MZ_SEEK_END);
1090 zip->cd_size = (uint32_t)mz_stream_tell(zip->cd_mem_stream);
1091 mz_stream_seek(zip->cd_mem_stream, 0, MZ_SEEK_SET);
1092
1093 err = mz_stream_copy(zip->stream, zip->cd_mem_stream, (int32_t)zip->cd_size);
1094
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07001095 mz_zip_print("Zip - Write cd (disk %" PRId32 " entries %" PRId64 " offset %" PRId64 " size %" PRId64 ")\n",
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001096 zip->disk_number_with_cd, zip->number_entry, zip->cd_offset, zip->cd_size);
1097
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001098 if (zip->cd_size == 0 && zip->number_entry > 0) {
Nathan Moinvaziri64faa252019-04-21 21:38:48 -07001099 // Zip does not contain central directory, open with recovery option
1100 return MZ_FORMAT_ERROR;
1101 }
1102
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001103 /* Write the ZIP64 central directory header */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001104 if (zip->cd_offset >= UINT32_MAX || zip->number_entry > UINT16_MAX) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001105 zip64_eocd_pos_inzip = mz_stream_tell(zip->stream);
1106
1107 err = mz_stream_write_uint32(zip->stream, MZ_ZIP_MAGIC_ENDHEADER64);
1108
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001109 /* Size of this 'zip64 end of central directory' */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001110 if (err == MZ_OK)
1111 err = mz_stream_write_uint64(zip->stream, (uint64_t)44);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001112 /* Version made by */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001113 if (err == MZ_OK)
1114 err = mz_stream_write_uint16(zip->stream, zip->version_madeby);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001115 /* Version needed */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001116 if (err == MZ_OK)
1117 err = mz_stream_write_uint16(zip->stream, (uint16_t)45);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001118 /* Number of this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001119 if (err == MZ_OK)
1120 err = mz_stream_write_uint32(zip->stream, zip->disk_number_with_cd);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001121 /* Number of the disk with the start of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001122 if (err == MZ_OK)
1123 err = mz_stream_write_uint32(zip->stream, zip->disk_number_with_cd);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001124 /* Total number of entries in the central dir on this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001125 if (err == MZ_OK)
1126 err = mz_stream_write_uint64(zip->stream, zip->number_entry);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001127 /* Total number of entries in the central dir */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001128 if (err == MZ_OK)
1129 err = mz_stream_write_uint64(zip->stream, zip->number_entry);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001130 /* Size of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001131 if (err == MZ_OK)
1132 err = mz_stream_write_int64(zip->stream, zip->cd_size);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001133 /* Offset of start of central directory with respect to the starting disk number */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001134 if (err == MZ_OK)
1135 err = mz_stream_write_int64(zip->stream, zip->cd_offset);
1136 if (err == MZ_OK)
1137 err = mz_stream_write_uint32(zip->stream, MZ_ZIP_MAGIC_ENDLOCHEADER64);
1138
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001139 /* Number of the disk with the start of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001140 if (err == MZ_OK)
1141 err = mz_stream_write_uint32(zip->stream, zip->disk_number_with_cd);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001142 /* Relative offset to the end of zip64 central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001143 if (err == MZ_OK)
1144 err = mz_stream_write_int64(zip->stream, zip64_eocd_pos_inzip);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001145 /* Number of the disk with the start of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001146 if (err == MZ_OK)
1147 err = mz_stream_write_uint32(zip->stream, zip->disk_number_with_cd + 1);
1148 }
1149
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001150 /* Write the central directory header */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001151
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001152 /* Signature */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001153 if (err == MZ_OK)
1154 err = mz_stream_write_uint32(zip->stream, MZ_ZIP_MAGIC_ENDHEADER);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001155 /* Number of this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001156 if (err == MZ_OK)
1157 err = mz_stream_write_uint16(zip->stream, (uint16_t)zip->disk_number_with_cd);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001158 /* Number of the disk with the start of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001159 if (err == MZ_OK)
1160 err = mz_stream_write_uint16(zip->stream, (uint16_t)zip->disk_number_with_cd);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001161 /* Total number of entries in the central dir on this disk */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001162 if (err == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001163 if (zip->number_entry >= UINT16_MAX)
1164 err = mz_stream_write_uint16(zip->stream, UINT16_MAX);
1165 else
1166 err = mz_stream_write_uint16(zip->stream, (uint16_t)zip->number_entry);
1167 }
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001168 /* Total number of entries in the central dir */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001169 if (err == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001170 if (zip->number_entry >= UINT16_MAX)
1171 err = mz_stream_write_uint16(zip->stream, UINT16_MAX);
1172 else
1173 err = mz_stream_write_uint16(zip->stream, (uint16_t)zip->number_entry);
1174 }
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001175 /* Size of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001176 if (err == MZ_OK)
1177 err = mz_stream_write_uint32(zip->stream, (uint32_t)zip->cd_size);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001178 /* Offset of start of central directory with respect to the starting disk number */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001179 if (err == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001180 if (zip->cd_offset >= UINT32_MAX)
1181 err = mz_stream_write_uint32(zip->stream, UINT32_MAX);
1182 else
1183 err = mz_stream_write_uint32(zip->stream, (uint32_t)zip->cd_offset);
1184 }
1185
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001186 /* Write global comment */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001187 if (zip->comment != NULL) {
Nathan Moinvaziridd103f42018-11-13 14:50:32 -08001188 comment_size = (int32_t)strlen(zip->comment);
1189 if (comment_size > UINT16_MAX)
1190 comment_size = UINT16_MAX;
1191 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001192 if (err == MZ_OK)
Nathan Moinvaziridd103f42018-11-13 14:50:32 -08001193 err = mz_stream_write_uint16(zip->stream, (uint16_t)comment_size);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001194 if (err == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001195 if (mz_stream_write(zip->stream, zip->comment, comment_size) != comment_size)
1196 err = MZ_READ_ERROR;
1197 }
1198 return err;
1199}
1200
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001201static int32_t mz_zip_recover_cd(void *handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001202 mz_zip *zip = (mz_zip *)handle;
1203 mz_zip_file local_file_info;
1204 void *local_file_info_stream = NULL;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001205 void *cd_mem_stream = NULL;
1206 uint64_t number_entry = 0;
1207 int64_t descriptor_pos = 0;
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001208 int64_t next_header_pos = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001209 int64_t disk_offset = 0;
1210 int64_t disk_number = 0;
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001211 int64_t compressed_pos = 0;
1212 int64_t compressed_end_pos = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001213 int64_t compressed_size = 0;
1214 int64_t uncompressed_size = 0;
1215 uint8_t descriptor_magic[4] = MZ_ZIP_MAGIC_DATADESCRIPTORU8;
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001216 uint8_t local_header_magic[4] = MZ_ZIP_MAGIC_LOCALHEADERU8;
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001217 uint8_t central_header_magic[4] = MZ_ZIP_MAGIC_CENTRALHEADERU8;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001218 uint32_t crc32 = 0;
1219 int32_t disk_number_with_cd = 0;
1220 int32_t err = MZ_OK;
1221 uint8_t zip64 = 0;
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001222 uint8_t eof = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001223
1224
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001225 mz_zip_print("Zip - Recover - Start\n");
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001226
1227 mz_zip_get_cd_mem_stream(handle, &cd_mem_stream);
1228
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001229 /* Determine if we are on a split disk or not */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001230 mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, 0);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001231 if (mz_stream_tell(zip->stream) < 0) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001232 mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, -1);
1233 mz_stream_seek(zip->stream, 0, MZ_SEEK_SET);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001234 } else
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001235 disk_number_with_cd = 1;
1236
1237 if (mz_stream_is_open(cd_mem_stream) != MZ_OK)
1238 err = mz_stream_mem_open(cd_mem_stream, NULL, MZ_OPEN_MODE_CREATE);
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001239
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001240 mz_stream_mem_create(&local_file_info_stream);
1241 mz_stream_mem_open(local_file_info_stream, NULL, MZ_OPEN_MODE_CREATE);
Nathan Moinvaziri57cbcd22019-10-21 15:37:58 -07001242
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001243 if (err == MZ_OK) {
Nathan Moinvaziri57cbcd22019-10-21 15:37:58 -07001244 err = mz_stream_find(zip->stream, (const void *)local_header_magic, sizeof(local_header_magic),
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001245 INT64_MAX, &next_header_pos);
Nathan Moinvaziri57cbcd22019-10-21 15:37:58 -07001246 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001247
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001248 while (err == MZ_OK && !eof) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001249 /* Get current offset and disk number for central dir record */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001250 disk_offset = mz_stream_tell(zip->stream);
1251 mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, &disk_number);
1252
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001253 /* Read local headers */
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001254 memset(&local_file_info, 0, sizeof(local_file_info));
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001255 err = mz_zip_entry_read_header(zip->stream, 1, &local_file_info, local_file_info_stream);
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001256 if (err != MZ_OK)
1257 break;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001258
1259 local_file_info.disk_offset = disk_offset;
1260 if (disk_number < 0)
1261 disk_number = 0;
1262 local_file_info.disk_number = (uint32_t)disk_number;
Nathan Moinvaziri370ab032019-11-15 13:24:17 -08001263
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001264 compressed_pos = mz_stream_tell(zip->stream);
Nathan Moinvaziri370ab032019-11-15 13:24:17 -08001265
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001266 if ((err == MZ_OK) && (local_file_info.compressed_size > 0)) {
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001267 mz_stream_seek(zip->stream, local_file_info.compressed_size, MZ_SEEK_CUR);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001268 }
1269
Nathan Moinvaziri958226c2020-07-30 11:06:25 -07001270 for (;;) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001271 /* Search for the next local header */
1272 err = mz_stream_find(zip->stream, (const void *)local_header_magic, sizeof(local_header_magic),
1273 INT64_MAX, &next_header_pos);
Nathan Moinvaziri370ab032019-11-15 13:24:17 -08001274
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001275 if (err == MZ_EXIST_ERROR) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001276 mz_stream_seek(zip->stream, compressed_pos, MZ_SEEK_SET);
1277
1278 /* Search for central dir if no local header found */
1279 err = mz_stream_find(zip->stream, (const void *)central_header_magic, sizeof(central_header_magic),
1280 INT64_MAX, &next_header_pos);
1281
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001282 if (err == MZ_EXIST_ERROR) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001283 /* Get end of stream if no central header found */
1284 mz_stream_seek(zip->stream, 0, MZ_SEEK_END);
1285 next_header_pos = mz_stream_tell(zip->stream);
1286 }
1287
1288 eof = 1;
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001289 }
1290
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001291 if (local_file_info.flag & MZ_ZIP_FLAG_DATA_DESCRIPTOR || local_file_info.compressed_size == 0) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001292 /* Search backwards for the descriptor, seeking too far back will be incorrect if compressed size is small */
1293 err = mz_stream_find_reverse(zip->stream, (const void *)descriptor_magic, sizeof(descriptor_magic),
1294 MZ_ZIP_SIZE_MAX_DATA_DESCRIPTOR, &descriptor_pos);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001295 if (err == MZ_OK) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001296 if (mz_zip_extrafield_contains(local_file_info.extrafield,
1297 local_file_info.extrafield_size, MZ_ZIP_EXTENSION_ZIP64, NULL) == MZ_OK)
1298 zip64 = 1;
1299
1300 err = mz_zip_entry_read_descriptor(zip->stream, zip64, &crc32,
1301 &compressed_size, &uncompressed_size);
1302
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001303 if (err == MZ_OK) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001304 if (local_file_info.crc == 0)
1305 local_file_info.crc = crc32;
1306 if (local_file_info.compressed_size == 0)
1307 local_file_info.compressed_size = compressed_size;
1308 if (local_file_info.uncompressed_size == 0)
1309 local_file_info.uncompressed_size = uncompressed_size;
1310 }
1311
1312 compressed_end_pos = descriptor_pos;
Nathan Moinvaziri958226c2020-07-30 11:06:25 -07001313 } else if (eof) {
1314 compressed_end_pos = next_header_pos;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001315 } else if (local_file_info.flag & MZ_ZIP_FLAG_DATA_DESCRIPTOR) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001316 /* Wrong local file entry found, keep searching */
1317 next_header_pos += 1;
1318 mz_stream_seek(zip->stream, next_header_pos, MZ_SEEK_SET);
1319 continue;
1320 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001321 } else {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001322 compressed_end_pos = next_header_pos;
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001323 }
1324
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001325 break;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001326 }
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001327
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001328 compressed_size = compressed_end_pos - compressed_pos;
1329
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001330 if (compressed_size > UINT32_MAX) {
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001331 /* Update sizes if 4GB file is written with no ZIP64 support */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001332 if (local_file_info.uncompressed_size < UINT32_MAX) {
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001333 local_file_info.compressed_size = compressed_size;
1334 local_file_info.uncompressed_size = 0;
1335 }
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001336 }
1337
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001338 mz_zip_print("Zip - Recover - Entry %s (csize %" PRId64 " usize %" PRId64 " flags 0x%" PRIx16 ")\n",
1339 local_file_info.filename, local_file_info.compressed_size, local_file_info.uncompressed_size,
1340 local_file_info.flag);
1341
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001342 /* Rewrite central dir with local headers and offsets */
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001343 err = mz_zip_entry_write_header(cd_mem_stream, 0, &local_file_info);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001344 if (err == MZ_OK)
Nathan Moinvaziri370ab032019-11-15 13:24:17 -08001345 number_entry += 1;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001346
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001347 err = mz_stream_seek(zip->stream, next_header_pos, MZ_SEEK_SET);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001348 }
1349
1350 mz_stream_mem_delete(&local_file_info_stream);
1351
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001352 mz_zip_print("Zip - Recover - Complete (cddisk %" PRId32 " entries %" PRId64 ")\n",
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001353 disk_number_with_cd, number_entry);
1354
1355 if (number_entry == 0)
1356 return err;
Nathan Moinvaziri370ab032019-11-15 13:24:17 -08001357
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001358 /* Set new upper seek boundary for central dir mem stream */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001359 disk_offset = mz_stream_tell(cd_mem_stream);
1360 mz_stream_mem_set_buffer_limit(cd_mem_stream, (int32_t)disk_offset);
1361
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001362 /* Set new central directory info */
Nathan Moinvaziri4d089732018-11-19 20:24:26 -08001363 mz_zip_set_cd_stream(handle, 0, cd_mem_stream);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001364 mz_zip_set_number_entry(handle, number_entry);
1365 mz_zip_set_disk_number_with_cd(handle, disk_number_with_cd);
1366
1367 return MZ_OK;
1368}
1369
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001370void *mz_zip_create(void **handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001371 mz_zip *zip = NULL;
1372
1373 zip = (mz_zip *)MZ_ALLOC(sizeof(mz_zip));
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001374 if (zip != NULL) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001375 memset(zip, 0, sizeof(mz_zip));
Nathan Moinvaziri41fcda92020-01-04 16:55:21 -08001376 zip->data_descriptor = 1;
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08001377 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001378 if (handle != NULL)
1379 *handle = zip;
1380
1381 return zip;
1382}
1383
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001384void mz_zip_delete(void **handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001385 mz_zip *zip = NULL;
1386 if (handle == NULL)
1387 return;
1388 zip = (mz_zip *)*handle;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001389 if (zip != NULL) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001390 MZ_FREE(zip);
1391 }
1392 *handle = NULL;
1393}
1394
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001395int32_t mz_zip_open(void *handle, void *stream, int32_t mode) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001396 mz_zip *zip = (mz_zip *)handle;
1397 int32_t err = MZ_OK;
1398
1399
1400 if (zip == NULL)
1401 return MZ_PARAM_ERROR;
1402
1403 mz_zip_print("Zip - Open\n");
1404
1405 zip->stream = stream;
1406
1407 mz_stream_mem_create(&zip->cd_mem_stream);
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001408
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001409 if (mode & MZ_OPEN_MODE_WRITE) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001410 mz_stream_mem_open(zip->cd_mem_stream, NULL, MZ_OPEN_MODE_CREATE);
1411 zip->cd_stream = zip->cd_mem_stream;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001412 } else {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001413 zip->cd_stream = stream;
1414 }
1415
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001416 if ((mode & MZ_OPEN_MODE_READ) || (mode & MZ_OPEN_MODE_APPEND)) {
1417 if ((mode & MZ_OPEN_MODE_CREATE) == 0) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001418 err = mz_zip_read_cd(zip);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001419 if (err != MZ_OK) {
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001420 mz_zip_print("Zip - Error detected reading cd (%" PRId32 ")\n", err);
Nathan Moinvaziridd103f42018-11-13 14:50:32 -08001421 if (zip->recover && mz_zip_recover_cd(zip) == MZ_OK)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001422 err = MZ_OK;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001423 }
1424 }
1425
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001426 if ((err == MZ_OK) && (mode & MZ_OPEN_MODE_APPEND)) {
1427 if (zip->cd_size > 0) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001428 /* Store central directory in memory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001429 err = mz_stream_seek(zip->stream, zip->cd_offset, MZ_SEEK_SET);
1430 if (err == MZ_OK)
1431 err = mz_stream_copy(zip->cd_mem_stream, zip->stream, (int32_t)zip->cd_size);
1432 if (err == MZ_OK)
1433 err = mz_stream_seek(zip->stream, zip->cd_offset, MZ_SEEK_SET);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001434 } else {
1435 if (zip->cd_signature == MZ_ZIP_MAGIC_ENDHEADER) {
Nathan Moinvaziri0e1fd892019-09-28 20:55:37 -07001436 /* If tiny zip then overwrite end header */
1437 err = mz_stream_seek(zip->stream, zip->cd_offset, MZ_SEEK_SET);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001438 } else {
Nathan Moinvaziri0e1fd892019-09-28 20:55:37 -07001439 /* If no central directory, append new zip to end of file */
1440 err = mz_stream_seek(zip->stream, 0, MZ_SEEK_END);
1441 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001442 }
Nathan Moinvaziri2ce1fe22018-11-23 11:53:11 -08001443
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001444 if (zip->disk_number_with_cd > 0) {
Nathan Moinvaziri2ce1fe22018-11-23 11:53:11 -08001445 /* Move to last disk to begin appending */
1446 mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, zip->disk_number_with_cd - 1);
1447 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001448 } else {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001449 zip->cd_start_pos = zip->cd_offset;
1450 }
1451 }
1452
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001453 if (err != MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001454 mz_zip_close(zip);
1455 return err;
1456 }
1457
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001458 /* Memory streams used to store variable length file info data */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001459 mz_stream_mem_create(&zip->file_info_stream);
1460 mz_stream_mem_open(zip->file_info_stream, NULL, MZ_OPEN_MODE_CREATE);
1461
1462 mz_stream_mem_create(&zip->local_file_info_stream);
1463 mz_stream_mem_open(zip->local_file_info_stream, NULL, MZ_OPEN_MODE_CREATE);
1464
1465 zip->open_mode = mode;
1466
1467 return err;
1468}
1469
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001470int32_t mz_zip_close(void *handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001471 mz_zip *zip = (mz_zip *)handle;
1472 int32_t err = MZ_OK;
1473
1474 if (zip == NULL)
1475 return MZ_PARAM_ERROR;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001476
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001477 mz_zip_print("Zip - Close\n");
1478
1479 if (mz_zip_entry_is_open(handle) == MZ_OK)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001480 err = mz_zip_entry_close(handle);
Nathan Moinvazirieffc1422018-11-08 15:24:17 -08001481
1482 if ((err == MZ_OK) && (zip->open_mode & MZ_OPEN_MODE_WRITE))
1483 err = mz_zip_write_cd(handle);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001484
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001485 if (zip->cd_mem_stream != NULL) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001486 mz_stream_close(zip->cd_mem_stream);
1487 mz_stream_delete(&zip->cd_mem_stream);
1488 }
1489
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001490 if (zip->file_info_stream != NULL) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001491 mz_stream_mem_close(zip->file_info_stream);
1492 mz_stream_mem_delete(&zip->file_info_stream);
1493 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001494 if (zip->local_file_info_stream != NULL) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001495 mz_stream_mem_close(zip->local_file_info_stream);
1496 mz_stream_mem_delete(&zip->local_file_info_stream);
1497 }
1498
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001499 if (zip->comment) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001500 MZ_FREE(zip->comment);
1501 zip->comment = NULL;
1502 }
1503
1504 zip->stream = NULL;
1505 zip->cd_stream = NULL;
1506
1507 return err;
1508}
1509
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001510int32_t mz_zip_get_comment(void *handle, const char **comment) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001511 mz_zip *zip = (mz_zip *)handle;
1512 if (zip == NULL || comment == NULL)
1513 return MZ_PARAM_ERROR;
1514 if (zip->comment == NULL)
1515 return MZ_EXIST_ERROR;
1516 *comment = zip->comment;
1517 return MZ_OK;
1518}
1519
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001520int32_t mz_zip_set_comment(void *handle, const char *comment) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001521 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziridd103f42018-11-13 14:50:32 -08001522 int32_t comment_size = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001523 if (zip == NULL || comment == NULL)
1524 return MZ_PARAM_ERROR;
1525 if (zip->comment != NULL)
1526 MZ_FREE(zip->comment);
Nathan Moinvaziri2df1ecb2018-12-01 08:59:21 -08001527 comment_size = (int32_t)strlen(comment);
1528 if (comment_size > UINT16_MAX)
1529 return MZ_PARAM_ERROR;
Nathan Moinvaziria6ecab52019-01-22 13:14:01 -08001530 zip->comment = (char *)MZ_ALLOC(comment_size+1);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001531 if (zip->comment == NULL)
1532 return MZ_MEM_ERROR;
Nathan Moinvaziria6ecab52019-01-22 13:14:01 -08001533 memset(zip->comment, 0, comment_size+1);
Nathan Moinvaziri2df1ecb2018-12-01 08:59:21 -08001534 strncpy(zip->comment, comment, comment_size);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001535 return MZ_OK;
1536}
1537
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001538int32_t mz_zip_get_version_madeby(void *handle, uint16_t *version_madeby) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001539 mz_zip *zip = (mz_zip *)handle;
1540 if (zip == NULL || version_madeby == NULL)
1541 return MZ_PARAM_ERROR;
1542 *version_madeby = zip->version_madeby;
1543 return MZ_OK;
1544}
1545
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001546int32_t mz_zip_set_version_madeby(void *handle, uint16_t version_madeby) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001547 mz_zip *zip = (mz_zip *)handle;
1548 if (zip == NULL)
1549 return MZ_PARAM_ERROR;
1550 zip->version_madeby = version_madeby;
1551 return MZ_OK;
1552}
1553
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001554int32_t mz_zip_set_recover(void *handle, uint8_t recover) {
Nathan Moinvaziridd103f42018-11-13 14:50:32 -08001555 mz_zip *zip = (mz_zip *)handle;
1556 if (zip == NULL)
1557 return MZ_PARAM_ERROR;
1558 zip->recover = recover;
1559 return MZ_OK;
1560}
1561
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001562int32_t mz_zip_set_data_descriptor(void *handle, uint8_t data_descriptor) {
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08001563 mz_zip *zip = (mz_zip *)handle;
1564 if (zip == NULL)
1565 return MZ_PARAM_ERROR;
1566 zip->data_descriptor = data_descriptor;
1567 return MZ_OK;
1568}
1569
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001570int32_t mz_zip_get_stream(void *handle, void **stream) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001571 mz_zip *zip = (mz_zip *)handle;
1572 if (zip == NULL || stream == NULL)
1573 return MZ_PARAM_ERROR;
1574 *stream = zip->stream;
1575 if (*stream == NULL)
1576 return MZ_EXIST_ERROR;
1577 return MZ_OK;
1578}
1579
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001580int32_t mz_zip_set_cd_stream(void *handle, int64_t cd_start_pos, void *cd_stream) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001581 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri4d089732018-11-19 20:24:26 -08001582 if (zip == NULL || cd_stream == NULL)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001583 return MZ_PARAM_ERROR;
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001584 zip->cd_offset = 0;
Nathan Moinvaziri4d089732018-11-19 20:24:26 -08001585 zip->cd_stream = cd_stream;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001586 zip->cd_start_pos = cd_start_pos;
1587 return MZ_OK;
1588}
1589
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001590int32_t mz_zip_get_cd_mem_stream(void *handle, void **cd_mem_stream) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001591 mz_zip *zip = (mz_zip *)handle;
1592 if (zip == NULL || cd_mem_stream == NULL)
1593 return MZ_PARAM_ERROR;
1594 *cd_mem_stream = zip->cd_mem_stream;
1595 if (*cd_mem_stream == NULL)
1596 return MZ_EXIST_ERROR;
1597 return MZ_OK;
1598}
1599
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001600int32_t mz_zip_set_number_entry(void *handle, uint64_t number_entry) {
Nathan Moinvaziri163352f2020-02-08 21:28:23 -08001601 mz_zip *zip = (mz_zip *)handle;
1602 if (zip == NULL)
1603 return MZ_PARAM_ERROR;
1604 zip->number_entry = number_entry;
1605 return MZ_OK;
1606}
1607
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001608int32_t mz_zip_get_number_entry(void *handle, uint64_t *number_entry) {
Nathan Moinvaziri163352f2020-02-08 21:28:23 -08001609 mz_zip *zip = (mz_zip *)handle;
1610 if (zip == NULL || number_entry == NULL)
1611 return MZ_PARAM_ERROR;
1612 *number_entry = zip->number_entry;
1613 return MZ_OK;
1614}
1615
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001616int32_t mz_zip_set_disk_number_with_cd(void *handle, uint32_t disk_number_with_cd) {
Nathan Moinvaziri163352f2020-02-08 21:28:23 -08001617 mz_zip *zip = (mz_zip *)handle;
1618 if (zip == NULL)
1619 return MZ_PARAM_ERROR;
1620 zip->disk_number_with_cd = disk_number_with_cd;
1621 return MZ_OK;
1622}
1623
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001624int32_t mz_zip_get_disk_number_with_cd(void *handle, uint32_t *disk_number_with_cd) {
Nathan Moinvaziri163352f2020-02-08 21:28:23 -08001625 mz_zip *zip = (mz_zip *)handle;
1626 if (zip == NULL || disk_number_with_cd == NULL)
1627 return MZ_PARAM_ERROR;
1628 *disk_number_with_cd = zip->disk_number_with_cd;
1629 return MZ_OK;
1630}
1631
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001632static int32_t mz_zip_entry_close_int(void *handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001633 mz_zip *zip = (mz_zip *)handle;
1634
1635 if (zip->crypt_stream != NULL)
1636 mz_stream_delete(&zip->crypt_stream);
1637 zip->crypt_stream = NULL;
1638 if (zip->compress_stream != NULL)
1639 mz_stream_delete(&zip->compress_stream);
1640 zip->compress_stream = NULL;
1641
1642 zip->entry_opened = 0;
1643
1644 return MZ_OK;
1645}
1646
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001647static int32_t mz_zip_entry_open_int(void *handle, uint8_t raw, int16_t compress_level, const char *password) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001648 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001649 int64_t max_total_in = 0;
Nathan Moinvazirifa8105b2018-07-08 19:05:00 -07001650 int64_t header_size = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001651 int64_t footer_size = 0;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001652 int32_t err = MZ_OK;
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001653 uint8_t use_crypt = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001654
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001655 if (zip == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001656 return MZ_PARAM_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001657
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001658 switch (zip->file_info.compression_method) {
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001659 case MZ_COMPRESS_METHOD_STORE:
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001660 case MZ_COMPRESS_METHOD_DEFLATE:
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001661#ifdef HAVE_BZIP2
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001662 case MZ_COMPRESS_METHOD_BZIP2:
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001663#endif
Nathan Moinvaziri566dfe02018-10-26 00:36:30 -07001664#ifdef HAVE_LZMA
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001665 case MZ_COMPRESS_METHOD_LZMA:
Nathan Moinvaziri1a7e7e42020-10-25 14:50:53 -07001666#endif
1667#if defined(HAVE_LZMA) || defined(HAVE_LIBCOMP)
Nathan Moinvaziric4668ed2020-10-25 11:56:19 -07001668 case MZ_COMPRESS_METHOD_XZ:
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001669#endif
Nathan Moinvaziri1f4758b2020-06-14 21:52:55 -07001670#ifdef HAVE_ZSTD
1671 case MZ_COMPRESS_METHOD_ZSTD:
Nathan Moinvaziri1f4758b2020-06-14 21:52:55 -07001672#endif
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001673 err = MZ_OK;
1674 break;
1675 default:
Nathan Moinvaziri829ffb52018-08-14 14:00:16 -07001676 return MZ_SUPPORT_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001677 }
1678
Nathan Moinvazirif9e34482018-11-27 09:45:09 -08001679#ifndef HAVE_WZAES
1680 if (zip->file_info.aes_version)
1681 return MZ_SUPPORT_ERROR;
1682#endif
1683
Nathan Moinvaziria64c9192018-08-08 17:47:12 -07001684 zip->entry_raw = raw;
1685
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001686 if ((zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED) && (password != NULL)) {
1687 if (zip->open_mode & MZ_OPEN_MODE_WRITE) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001688 /* Encrypt only when we are not trying to write raw and password is supplied. */
Nathan Moinvaziria64c9192018-08-08 17:47:12 -07001689 if (!zip->entry_raw)
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001690 use_crypt = 1;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001691 } else if (zip->open_mode & MZ_OPEN_MODE_READ) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001692 /* Decrypt only when password is supplied. Don't error when password */
1693 /* is not supplied as we may want to read the raw encrypted data. */
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001694 use_crypt = 1;
1695 }
1696 }
1697
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001698 if ((err == MZ_OK) && (use_crypt)) {
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -08001699#ifdef HAVE_WZAES
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001700 if (zip->file_info.aes_version) {
Nathan Moinvaziri21a31022018-10-24 09:50:16 -07001701 mz_stream_wzaes_create(&zip->crypt_stream);
1702 mz_stream_wzaes_set_password(zip->crypt_stream, password);
1703 mz_stream_wzaes_set_encryption_mode(zip->crypt_stream, zip->file_info.aes_encryption_mode);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001704 } else
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001705#endif
1706 {
Nathan Moinvaziri0e5c9dc2018-05-23 20:21:48 -07001707#ifdef HAVE_PKCRYPT
Nathan Moinvaziri9ffb08e2020-11-30 21:30:10 -08001708 uint8_t verify1 = (uint8_t)((zip->file_info.pk_verify >> 8) & 0xff);
1709 uint8_t verify2 = (uint8_t)((zip->file_info.pk_verify) & 0xff);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001710
Nathan Moinvaziri0e5c9dc2018-05-23 20:21:48 -07001711 mz_stream_pkcrypt_create(&zip->crypt_stream);
1712 mz_stream_pkcrypt_set_password(zip->crypt_stream, password);
1713 mz_stream_pkcrypt_set_verify(zip->crypt_stream, verify1, verify2);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001714#endif
1715 }
1716 }
1717
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001718 if (err == MZ_OK) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001719 if (zip->crypt_stream == NULL)
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001720 mz_stream_raw_create(&zip->crypt_stream);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001721
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001722 mz_stream_set_base(zip->crypt_stream, zip->stream);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001723
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001724 err = mz_stream_open(zip->crypt_stream, NULL, zip->open_mode);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001725 }
1726
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001727 if (err == MZ_OK) {
Nathan Moinvaziria64c9192018-08-08 17:47:12 -07001728 if (zip->entry_raw || zip->file_info.compression_method == MZ_COMPRESS_METHOD_STORE)
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001729 mz_stream_raw_create(&zip->compress_stream);
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -08001730#if defined(HAVE_ZLIB) || defined(HAVE_LIBCOMP)
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001731 else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_DEFLATE)
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001732 mz_stream_zlib_create(&zip->compress_stream);
Andy Maloney3909c232018-05-22 09:02:09 -04001733#endif
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001734#ifdef HAVE_BZIP2
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001735 else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_BZIP2)
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001736 mz_stream_bzip_create(&zip->compress_stream);
1737#endif
Nathan Moinvaziri1a7e7e42020-10-25 14:50:53 -07001738#ifdef HAVE_LIBCOMP
1739 else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_XZ) {
Nathan Moinvaziria7343862020-10-25 15:22:10 -07001740 mz_stream_libcomp_create(&zip->compress_stream);
Nathan Moinvaziri1a7e7e42020-10-25 14:50:53 -07001741 mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_COMPRESS_METHOD,
1742 zip->file_info.compression_method);
1743 }
1744#endif
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001745#ifdef HAVE_LZMA
Nathan Moinvaziric4668ed2020-10-25 11:56:19 -07001746 else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_LZMA ||
1747 zip->file_info.compression_method == MZ_COMPRESS_METHOD_XZ) {
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001748 mz_stream_lzma_create(&zip->compress_stream);
Nathan Moinvaziriff7ef662020-10-25 12:27:29 -07001749 mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_COMPRESS_METHOD,
1750 zip->file_info.compression_method);
1751 }
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001752#endif
Nathan Moinvaziri1f4758b2020-06-14 21:52:55 -07001753#ifdef HAVE_ZSTD
Force Charlie32ccb3e2020-06-24 10:55:18 +08001754 else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_ZSTD)
Nathan Moinvaziri1f4758b2020-06-14 21:52:55 -07001755 mz_stream_zstd_create(&zip->compress_stream);
1756#endif
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001757 else
1758 err = MZ_PARAM_ERROR;
1759 }
1760
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001761 if (err == MZ_OK) {
1762 if (zip->open_mode & MZ_OPEN_MODE_WRITE) {
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001763 mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_COMPRESS_LEVEL, compress_level);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001764 } else {
Nathan Moinvaziri91261772020-06-19 21:16:16 -07001765 int32_t set_end_of_stream = 0;
1766
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -08001767#ifndef HAVE_LIBCOMP
Nathan Moinvaziriff7ef662020-10-25 12:27:29 -07001768 if (zip->entry_raw ||
1769 zip->file_info.compression_method == MZ_COMPRESS_METHOD_STORE ||
1770 zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED)
Nathan Moinvaziri90d31c72018-11-20 03:17:20 -08001771#endif
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001772 {
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001773 max_total_in = zip->file_info.compressed_size;
Nathan Moinvazirifa8105b2018-07-08 19:05:00 -07001774 mz_stream_set_prop_int64(zip->crypt_stream, MZ_STREAM_PROP_TOTAL_IN_MAX, max_total_in);
1775
1776 if (mz_stream_get_prop_int64(zip->crypt_stream, MZ_STREAM_PROP_HEADER_SIZE, &header_size) == MZ_OK)
1777 max_total_in -= header_size;
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001778 if (mz_stream_get_prop_int64(zip->crypt_stream, MZ_STREAM_PROP_FOOTER_SIZE, &footer_size) == MZ_OK)
1779 max_total_in -= footer_size;
Nathan Moinvazirifa8105b2018-07-08 19:05:00 -07001780
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001781 mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_IN_MAX, max_total_in);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001782 }
Nathan Moinvaziri91261772020-06-19 21:16:16 -07001783
1784 switch (zip->file_info.compression_method) {
1785 case MZ_COMPRESS_METHOD_LZMA:
Nathan Moinvaziri5cd28792020-10-25 12:50:12 -07001786 case MZ_COMPRESS_METHOD_XZ:
Nathan Moinvaziri91261772020-06-19 21:16:16 -07001787 set_end_of_stream = (zip->file_info.flag & MZ_ZIP_FLAG_LZMA_EOS_MARKER);
1788 break;
1789 case MZ_COMPRESS_METHOD_ZSTD:
Nathan Moinvaziri91261772020-06-19 21:16:16 -07001790 set_end_of_stream = 1;
1791 break;
juanii55bcdaf2018-02-11 20:55:57 -03001792 }
Nathan Moinvaziri91261772020-06-19 21:16:16 -07001793
1794 if (set_end_of_stream) {
Force.Charlie7cecfa62020-06-15 07:43:25 -07001795 mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_IN_MAX, zip->file_info.compressed_size);
1796 mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_OUT_MAX, zip->file_info.uncompressed_size);
1797 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001798 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001799
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001800 mz_stream_set_base(zip->compress_stream, zip->crypt_stream);
1801
1802 err = mz_stream_open(zip->compress_stream, NULL, zip->open_mode);
1803 }
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07001804
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001805 if (err == MZ_OK) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001806 zip->entry_opened = 1;
Nathan Moinvazirie63d2312018-11-03 19:45:41 -07001807 zip->entry_crc32 = 0;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001808 } else {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001809 mz_zip_entry_close_int(handle);
1810 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001811
1812 return err;
1813}
1814
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001815int32_t mz_zip_entry_is_open(void *handle) {
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07001816 mz_zip *zip = (mz_zip *)handle;
1817 if (zip == NULL)
1818 return MZ_PARAM_ERROR;
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07001819 if (zip->entry_opened == 0)
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07001820 return MZ_EXIST_ERROR;
1821 return MZ_OK;
1822}
1823
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001824int32_t mz_zip_entry_read_open(void *handle, uint8_t raw, const char *password) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001825 mz_zip *zip = (mz_zip *)handle;
1826 int32_t err = MZ_OK;
Nathan Moinvaziriaacf8732018-11-06 09:38:46 -08001827 int32_t err_shift = MZ_OK;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001828
Nathan Moinvaziri40427632018-07-22 11:12:40 -07001829#if defined(MZ_ZIP_NO_ENCRYPTION)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001830 if (password != NULL)
Nathan Moinvazirif9e34482018-11-27 09:45:09 -08001831 return MZ_SUPPORT_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001832#endif
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001833 if (zip == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001834 return MZ_PARAM_ERROR;
Nathan Moinvaziri79ac3382017-10-23 17:22:36 -07001835 if ((zip->open_mode & MZ_OPEN_MODE_READ) == 0)
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001836 return MZ_PARAM_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001837 if (zip->entry_scanned == 0)
1838 return MZ_PARAM_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001839
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07001840 mz_zip_print("Zip - Entry - Read open (raw %" PRId32 ")\n", raw);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001841
Nathan Moinvaziri0030fa22020-12-21 19:31:07 -08001842 err = mz_zip_entry_seek_local_header(handle);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001843 if (err == MZ_OK)
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001844 err = mz_zip_entry_read_header(zip->stream, 1, &zip->local_file_info, zip->local_file_info_stream);
Nathan Moinvaziri903b73d2017-10-20 08:47:19 -07001845
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001846 if (err == MZ_FORMAT_ERROR && zip->disk_offset_shift > 0) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001847 /* Perhaps we didn't compensated correctly for incorrect cd offset */
Nathan Moinvaziriaacf8732018-11-06 09:38:46 -08001848 err_shift = mz_stream_seek(zip->stream, zip->file_info.disk_offset, MZ_SEEK_SET);
1849 if (err_shift == MZ_OK)
1850 err_shift = mz_zip_entry_read_header(zip->stream, 1, &zip->local_file_info, zip->local_file_info_stream);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001851 if (err_shift == MZ_OK) {
Nathan Moinvaziriaacf8732018-11-06 09:38:46 -08001852 zip->disk_offset_shift = 0;
1853 err = err_shift;
1854 }
1855 }
1856
Nathan Moinvaziria6d1f662018-07-22 10:35:49 -07001857#ifdef MZ_ZIP_NO_DECOMPRESSION
Nathan Moinvaziri2eb29072018-11-23 10:32:21 -08001858 if (!raw && zip->file_info.compression_method != MZ_COMPRESS_METHOD_STORE)
Nathan Moinvaziri0a9282d2018-06-19 11:59:07 -07001859 err = MZ_SUPPORT_ERROR;
Viktor Szakatse7215072018-09-04 15:06:53 +00001860#endif
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001861 if (err == MZ_OK)
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001862 err = mz_zip_entry_open_int(handle, raw, 0, password);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001863
1864 return err;
1865}
1866
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001867int32_t mz_zip_entry_write_open(void *handle, const mz_zip_file *file_info, int16_t compress_level, uint8_t raw, const char *password) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001868 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri6b64b922018-11-23 11:41:35 -08001869 int64_t filename_pos = -1;
Nathan Moinvazirif9e34482018-11-27 09:45:09 -08001870 int64_t extrafield_pos = 0;
1871 int64_t comment_pos = 0;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -07001872 int64_t linkname_pos = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001873 int64_t disk_number = 0;
Nathan Moinvazirifa620e42018-10-20 10:37:10 -07001874 uint8_t is_dir = 0;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001875 int32_t err = MZ_OK;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001876
Nathan Moinvaziri40427632018-07-22 11:12:40 -07001877#if defined(MZ_ZIP_NO_ENCRYPTION)
tz-lomb1b25802017-11-10 15:03:02 +03001878 if (password != NULL)
Nathan Moinvazirif9e34482018-11-27 09:45:09 -08001879 return MZ_SUPPORT_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001880#endif
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001881 if (zip == NULL || file_info == NULL || file_info->filename == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001882 return MZ_PARAM_ERROR;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001883
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001884 if (mz_zip_entry_is_open(handle) == MZ_OK) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001885 err = mz_zip_entry_close(handle);
1886 if (err != MZ_OK)
1887 return err;
1888 }
1889
1890 memcpy(&zip->file_info, file_info, sizeof(mz_zip_file));
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001891
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07001892 mz_zip_print("Zip - Entry - Write open - %s (level %" PRId16 " raw %" PRId8 ")\n",
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001893 zip->file_info.filename, compress_level, raw);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001894
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001895 mz_stream_seek(zip->file_info_stream, 0, MZ_SEEK_SET);
Nathan Moinvazirif5cc4c02018-08-14 17:46:36 -07001896 mz_stream_write(zip->file_info_stream, file_info, sizeof(mz_zip_file));
1897
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001898 /* Copy filename, extrafield, and comment internally */
Nathan Moinvaziri6b64b922018-11-23 11:41:35 -08001899 filename_pos = mz_stream_tell(zip->file_info_stream);
Nathan Moinvazirif5cc4c02018-08-14 17:46:36 -07001900 if (file_info->filename != NULL)
Nathan Moinvaziri6b64b922018-11-23 11:41:35 -08001901 mz_stream_write(zip->file_info_stream, file_info->filename, (int32_t)strlen(file_info->filename));
1902 mz_stream_write_uint8(zip->file_info_stream, 0);
1903
1904 extrafield_pos = mz_stream_tell(zip->file_info_stream);
Nathan Moinvazirif5cc4c02018-08-14 17:46:36 -07001905 if (file_info->extrafield != NULL)
Nathan Moinvazirica014e62018-08-15 07:31:12 -07001906 mz_stream_write(zip->file_info_stream, file_info->extrafield, file_info->extrafield_size);
Nathan Moinvaziri6b64b922018-11-23 11:41:35 -08001907 mz_stream_write_uint8(zip->file_info_stream, 0);
1908
1909 comment_pos = mz_stream_tell(zip->file_info_stream);
Nathan Moinvazirif5cc4c02018-08-14 17:46:36 -07001910 if (file_info->comment != NULL)
Nathan Moinvaziri6b64b922018-11-23 11:41:35 -08001911 mz_stream_write(zip->file_info_stream, file_info->comment, file_info->comment_size);
1912 mz_stream_write_uint8(zip->file_info_stream, 0);
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001913
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -07001914 linkname_pos = mz_stream_tell(zip->file_info_stream);
1915 if (file_info->linkname != NULL)
1916 mz_stream_write(zip->file_info_stream, file_info->linkname, (int32_t)strlen(file_info->linkname));
1917 mz_stream_write_uint8(zip->file_info_stream, 0);
Nathan Moinvaziri6b64b922018-11-23 11:41:35 -08001918
Nathan Moinvazirif9e34482018-11-27 09:45:09 -08001919 mz_stream_mem_get_buffer_at(zip->file_info_stream, filename_pos, (const void **)&zip->file_info.filename);
1920 mz_stream_mem_get_buffer_at(zip->file_info_stream, extrafield_pos, (const void **)&zip->file_info.extrafield);
1921 mz_stream_mem_get_buffer_at(zip->file_info_stream, comment_pos, (const void **)&zip->file_info.comment);
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -07001922 mz_stream_mem_get_buffer_at(zip->file_info_stream, linkname_pos, (const void **)&zip->file_info.linkname);
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001923
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001924 if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_DEFLATE) {
Nathan Moinvaziri91a76f62017-11-10 07:39:06 -08001925 if ((compress_level == 8) || (compress_level == 9))
1926 zip->file_info.flag |= MZ_ZIP_FLAG_DEFLATE_MAX;
1927 if (compress_level == 2)
1928 zip->file_info.flag |= MZ_ZIP_FLAG_DEFLATE_FAST;
1929 if (compress_level == 1)
1930 zip->file_info.flag |= MZ_ZIP_FLAG_DEFLATE_SUPER_FAST;
1931 }
Nathan Moinvaziri1a7e7e42020-10-25 14:50:53 -07001932#if defined(HAVE_LZMA) || defined(HAVE_LIBCOMP)
Nathan Moinvaziri5cd28792020-10-25 12:50:12 -07001933 else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_LZMA ||
1934 zip->file_info.compression_method == MZ_COMPRESS_METHOD_XZ)
Nathan Moinvaziri91a76f62017-11-10 07:39:06 -08001935 zip->file_info.flag |= MZ_ZIP_FLAG_LZMA_EOS_MARKER;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001936#endif
Nathan Moinvaziri91a76f62017-11-10 07:39:06 -08001937
Nathan Moinvazirifa620e42018-10-20 10:37:10 -07001938 if (mz_zip_attrib_is_dir(zip->file_info.external_fa, zip->file_info.version_madeby) == MZ_OK)
1939 is_dir = 1;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001940
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001941 if (!is_dir) {
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08001942 if (zip->data_descriptor)
1943 zip->file_info.flag |= MZ_ZIP_FLAG_DATA_DESCRIPTOR;
Nathan Moinvazirifa620e42018-10-20 10:37:10 -07001944 if (password != NULL)
1945 zip->file_info.flag |= MZ_ZIP_FLAG_ENCRYPTED;
1946 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001947
juaniib8887e92018-02-14 00:51:05 -03001948 mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, &disk_number);
1949 zip->file_info.disk_number = (uint32_t)disk_number;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001950 zip->file_info.disk_offset = mz_stream_tell(zip->stream);
Nathan Moinvaziri215a7a82020-12-13 10:58:41 -08001951
1952 if (zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED) {
1953#ifdef HAVE_PKCRYPT
1954 /* Pre-calculated CRC value is required for PKWARE traditional encryption */
1955 uint32_t dos_date = mz_zip_time_t_to_dos_date(zip->file_info.modified_date);
1956 zip->file_info.pk_verify = mz_zip_get_pk_verify(dos_date, zip->file_info.crc, zip->file_info.flag);
1957#endif
1958#ifdef HAVE_WZAES
1959 if (zip->file_info.aes_version && zip->file_info.aes_encryption_mode == 0)
1960 zip->file_info.aes_encryption_mode = MZ_AES_ENCRYPTION_MODE_256;
1961#endif
1962 }
1963
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001964 zip->file_info.crc = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001965 zip->file_info.compressed_size = 0;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001966
Nathan Moinvazirifa620e42018-10-20 10:37:10 -07001967 if ((compress_level == 0) || (is_dir))
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001968 zip->file_info.compression_method = MZ_COMPRESS_METHOD_STORE;
Viktor Szakats2884e672018-04-30 08:12:13 +00001969
Nathan Moinvaziria6d1f662018-07-22 10:35:49 -07001970#ifdef MZ_ZIP_NO_COMPRESSION
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001971 if (zip->file_info.compression_method != MZ_COMPRESS_METHOD_STORE)
Nathan Moinvaziri0a9282d2018-06-19 11:59:07 -07001972 err = MZ_SUPPORT_ERROR;
1973#endif
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001974 if (err == MZ_OK)
Nathan Moinvazirifa620e42018-10-20 10:37:10 -07001975 err = mz_zip_entry_write_header(zip->stream, 1, &zip->file_info);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001976 if (err == MZ_OK)
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001977 err = mz_zip_entry_open_int(handle, raw, compress_level, password);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001978
1979 return err;
1980}
1981
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001982int32_t mz_zip_entry_read(void *handle, void *buf, int32_t len) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001983 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001984 int32_t read = 0;
1985
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07001986 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001987 return MZ_PARAM_ERROR;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001988 if (UINT_MAX == UINT16_MAX && len > UINT16_MAX) /* zlib limitation */
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001989 return MZ_PARAM_ERROR;
Nathan Moinvazirid7814e92018-07-11 14:54:14 -07001990 if (len == 0)
1991 return MZ_PARAM_ERROR;
1992
Nathan Moinvazirieb80cd92018-07-11 16:45:09 -07001993 if (zip->file_info.compressed_size == 0)
1994 return 0;
1995
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001996 /* Read entire entry even if uncompressed_size = 0, otherwise */
1997 /* aes encryption validation will fail if compressed_size > 0 */
Nathan Moinvazirie63d2312018-11-03 19:45:41 -07001998 read = mz_stream_read(zip->compress_stream, buf, len);
1999 if (read > 0)
2000 zip->entry_crc32 = mz_crypt_crc32_update(zip->entry_crc32, buf, read);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08002001
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07002002 mz_zip_print("Zip - Entry - Read - %" PRId32 " (max %" PRId32 ")\n", read, len);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08002003
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002004 return read;
2005}
2006
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002007int32_t mz_zip_entry_write(void *handle, const void *buf, int32_t len) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002008 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri0a9282d2018-06-19 11:59:07 -07002009 int32_t written = 0;
2010
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07002011 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002012 return MZ_PARAM_ERROR;
Nathan Moinvazirie63d2312018-11-03 19:45:41 -07002013 written = mz_stream_write(zip->compress_stream, buf, len);
2014 if (written > 0)
2015 zip->entry_crc32 = mz_crypt_crc32_update(zip->entry_crc32, buf, written);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08002016
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07002017 mz_zip_print("Zip - Entry - Write - %" PRId32 " (max %" PRId32 ")\n", written, len);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08002018
Nathan Moinvaziri0a9282d2018-06-19 11:59:07 -07002019 return written;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002020}
2021
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002022int32_t mz_zip_entry_read_close(void *handle, uint32_t *crc32, int64_t *compressed_size,
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002023 int64_t *uncompressed_size) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002024 mz_zip *zip = (mz_zip *)handle;
2025 int64_t total_in = 0;
2026 int32_t err = MZ_OK;
2027 uint8_t zip64 = 0;
2028
2029 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
2030 return MZ_PARAM_ERROR;
2031
2032 mz_stream_close(zip->compress_stream);
2033
2034 mz_zip_print("Zip - Entry - Read Close\n");
2035
2036 if (crc32 != NULL)
2037 *crc32 = zip->file_info.crc;
2038 if (compressed_size != NULL)
2039 *compressed_size = zip->file_info.compressed_size;
2040 if (uncompressed_size != NULL)
2041 *uncompressed_size = zip->file_info.uncompressed_size;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002042
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002043 mz_stream_get_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_IN, &total_in);
2044
2045 if ((zip->file_info.flag & MZ_ZIP_FLAG_DATA_DESCRIPTOR) &&
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002046 ((zip->file_info.flag & MZ_ZIP_FLAG_MASK_LOCAL_INFO) == 0) &&
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002047 (crc32 != NULL || compressed_size != NULL || uncompressed_size != NULL)) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002048 /* Check to see if data descriptor is zip64 bit format or not */
2049 if (mz_zip_extrafield_contains(zip->local_file_info.extrafield,
2050 zip->local_file_info.extrafield_size, MZ_ZIP_EXTENSION_ZIP64, NULL) == MZ_OK)
2051 zip64 = 1;
2052
Nathan Moinvaziri0030fa22020-12-21 19:31:07 -08002053 err = mz_zip_entry_seek_local_header(handle);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002054
2055 /* Seek to end of compressed stream since we might have over-read during compression */
2056 if (err == MZ_OK)
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002057 err = mz_stream_seek(zip->stream, MZ_ZIP_SIZE_LD_ITEM +
2058 (int64_t)zip->local_file_info.filename_size +
2059 (int64_t)zip->local_file_info.extrafield_size +
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002060 total_in, MZ_SEEK_CUR);
2061
2062 /* Read data descriptor */
2063 if (err == MZ_OK)
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002064 err = mz_zip_entry_read_descriptor(zip->stream, zip64,
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002065 crc32, compressed_size, uncompressed_size);
2066 }
2067
2068 /* If entire entry was not read verification will fail */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002069 if ((err == MZ_OK) && (total_in > 0) && (!zip->entry_raw)) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002070#ifdef HAVE_WZAES
2071 /* AES zip version AE-1 will expect a valid crc as well */
2072 if (zip->file_info.aes_version <= 0x0001)
2073#endif
2074 {
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002075 if (zip->entry_crc32 != zip->file_info.crc) {
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07002076 mz_zip_print("Zip - Entry - Crc failed (actual 0x%08" PRIx32 " expected 0x%08" PRIx32 ")\n",
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002077 zip->entry_crc32, zip->file_info.crc);
2078
2079 err = MZ_CRC_ERROR;
2080 }
2081 }
2082 }
2083
2084 mz_zip_entry_close_int(handle);
2085
2086 return err;
2087}
2088
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002089int32_t mz_zip_entry_write_close(void *handle, uint32_t crc32, int64_t compressed_size,
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002090 int64_t uncompressed_size) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002091 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002092 int64_t end_disk_number = 0;
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002093 int32_t err = MZ_OK;
2094 uint8_t zip64 = 0;
2095
2096 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
2097 return MZ_PARAM_ERROR;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002098
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002099 mz_stream_close(zip->compress_stream);
2100
2101 if (!zip->entry_raw)
2102 crc32 = zip->entry_crc32;
2103
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07002104 mz_zip_print("Zip - Entry - Write Close (crc 0x%08" PRIx32 " cs %" PRId64 " ucs %" PRId64 ")\n",
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002105 crc32, compressed_size, uncompressed_size);
2106
2107 /* If sizes are not set, then read them from the compression stream */
2108 if (compressed_size < 0)
2109 mz_stream_get_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_OUT, &compressed_size);
2110 if (uncompressed_size < 0)
Nathan Moinvazirid9318432018-12-01 07:56:28 -08002111 mz_stream_get_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_IN, &uncompressed_size);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002112
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002113 if (zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002114 mz_stream_set_base(zip->crypt_stream, zip->stream);
2115 err = mz_stream_close(zip->crypt_stream);
2116
2117 mz_stream_get_prop_int64(zip->crypt_stream, MZ_STREAM_PROP_TOTAL_OUT, &compressed_size);
2118 }
2119
Nathan Moinvazirief93d652020-12-22 17:20:04 -08002120 mz_zip_entry_needs_zip64(&zip->file_info, 1, &zip64);
2121
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002122 if ((err == MZ_OK) && (zip->file_info.flag & MZ_ZIP_FLAG_DATA_DESCRIPTOR)) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002123 /* Determine if we need to write data descriptor in zip64 format,
2124 if local extrafield was saved with zip64 extrafield */
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002125
2126 if (zip->file_info.flag & MZ_ZIP_FLAG_MASK_LOCAL_INFO)
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002127 err = mz_zip_entry_write_descriptor(zip->stream,
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002128 zip64, 0, compressed_size, 0);
2129 else
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002130 err = mz_zip_entry_write_descriptor(zip->stream,
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002131 zip64, crc32, compressed_size, uncompressed_size);
2132 }
2133
2134 /* Write file info to central directory */
2135
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07002136 mz_zip_print("Zip - Entry - Write cd (ucs %" PRId64 " cs %" PRId64 " crc 0x%08" PRIx32 ")\n",
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002137 uncompressed_size, compressed_size, crc32);
2138
2139 zip->file_info.crc = crc32;
2140 zip->file_info.compressed_size = compressed_size;
2141 zip->file_info.uncompressed_size = uncompressed_size;
2142
2143 if (err == MZ_OK)
2144 err = mz_zip_entry_write_header(zip->cd_mem_stream, 0, &zip->file_info);
2145
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002146 /* Update local header with crc32 and sizes */
2147 if ((err == MZ_OK) && ((zip->file_info.flag & MZ_ZIP_FLAG_DATA_DESCRIPTOR) == 0) &&
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002148 ((zip->file_info.flag & MZ_ZIP_FLAG_MASK_LOCAL_INFO) == 0)) {
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002149 /* Save the disk number and position we are to seek back after updating local header */
Nathan Moinvaziri58b37d32020-02-08 20:32:58 -08002150 int64_t end_pos = mz_stream_tell(zip->stream);
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002151 mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, &end_disk_number);
2152
Nathan Moinvaziri0030fa22020-12-21 19:31:07 -08002153 err = mz_zip_entry_seek_local_header(handle);
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002154
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002155 if (err == MZ_OK) {
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002156 /* Seek to crc32 and sizes offset in local header */
2157 err = mz_stream_seek(zip->stream, MZ_ZIP_OFFSET_CRC_SIZES, MZ_SEEK_CUR);
2158 }
2159
2160 if (err == MZ_OK)
Nathan Moinvazirief93d652020-12-22 17:20:04 -08002161 err = mz_zip_entry_write_crc_sizes(zip->stream, zip64, 0, &zip->file_info);
2162
2163 /* Seek to and update zip64 extension sizes */
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -08002164 if ((err == MZ_OK) && (zip64)) {
Nathan Moinvazirief93d652020-12-22 17:20:04 -08002165 uint16_t length = 0;
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -08002166 uint16_t filename_size = 0;
2167 uint16_t extrafield_size = 0;
Nathan Moinvazirief93d652020-12-22 17:20:04 -08002168
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -08002169 /* Read extrafield sizes and seek ahead to extrafield */
2170 err = mz_stream_read_uint16(zip->stream, &filename_size);
2171 if (err == MZ_OK)
2172 err = mz_stream_read_uint16(zip->stream, &extrafield_size);
2173 if ((err == MZ_OK) && (extrafield_size > 0)) {
2174 err = mz_stream_seek(zip->stream, filename_size, MZ_SEEK_CUR);
Nathan Moinvazirief93d652020-12-22 17:20:04 -08002175
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -08002176 /* Find zip64 extrafield and update compressed and uncompressed sizes */
2177 if (err == MZ_OK)
Nathan Moinvaziri8f769682020-12-23 11:26:30 -08002178 err = mz_zip_extrafield_find(zip->stream, MZ_ZIP_EXTENSION_ZIP64, extrafield_size, &length);
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -08002179 if (err == MZ_OK) {
2180 if (length >= 8)
2181 err = mz_stream_write_uint64(zip->stream, zip->file_info.compressed_size);
2182 if ((err == MZ_OK) && (length >= 16))
2183 err = mz_stream_write_uint64(zip->stream, zip->file_info.uncompressed_size);
2184 }
Nathan Moinvazirief93d652020-12-22 17:20:04 -08002185 }
2186 }
2187
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002188 mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, end_disk_number);
2189 mz_stream_seek(zip->stream, end_pos, MZ_SEEK_SET);
2190 }
2191
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002192 zip->number_entry += 1;
2193
2194 mz_zip_entry_close_int(handle);
2195
2196 return err;
2197}
2198
Nathan Moinvaziri0030fa22020-12-21 19:31:07 -08002199int32_t mz_zip_entry_seek_local_header(void *handle) {
2200 mz_zip *zip = (mz_zip *)handle;
2201 int64_t disk_size = 0;
2202 uint32_t disk_number = zip->file_info.disk_number;
2203
2204 if (disk_number == zip->disk_number_with_cd) {
2205 mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_SIZE, &disk_size);
2206 if ((disk_size == 0) || ((zip->open_mode & MZ_OPEN_MODE_WRITE) == 0))
2207 disk_number = (uint32_t)-1;
2208 }
2209
2210 mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, disk_number);
2211
2212 mz_zip_print("Zip - Entry - Seek local (disk %" PRId32 " offset %" PRId64 ")\n",
2213 disk_number, zip->file_info.disk_offset);
2214
2215 /* Guard against seek overflows */
2216 if ((zip->disk_offset_shift > 0) &&
2217 (zip->file_info.disk_offset > (INT64_MAX - zip->disk_offset_shift)))
2218 return MZ_FORMAT_ERROR;
2219
2220 return mz_stream_seek(zip->stream, zip->file_info.disk_offset + zip->disk_offset_shift, MZ_SEEK_SET);
2221}
2222
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002223int32_t mz_zip_entry_close(void *handle) {
Nathan Moinvaziri163352f2020-02-08 21:28:23 -08002224 return mz_zip_entry_close_raw(handle, UINT64_MAX, 0);
2225}
2226
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002227int32_t mz_zip_entry_close_raw(void *handle, int64_t uncompressed_size, uint32_t crc32) {
Nathan Moinvaziri163352f2020-02-08 21:28:23 -08002228 mz_zip *zip = (mz_zip *)handle;
2229 int32_t err = MZ_OK;
2230
2231 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
2232 return MZ_PARAM_ERROR;
2233
2234 if (zip->open_mode & MZ_OPEN_MODE_WRITE)
2235 err = mz_zip_entry_write_close(handle, crc32, UINT64_MAX, uncompressed_size);
2236 else
2237 err = mz_zip_entry_read_close(handle, NULL, NULL, NULL);
2238
2239 return err;
2240}
2241
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002242int32_t mz_zip_entry_is_dir(void *handle) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002243 mz_zip *zip = (mz_zip *)handle;
2244 int32_t filename_length = 0;
2245
2246 if (zip == NULL)
2247 return MZ_PARAM_ERROR;
2248 if (zip->entry_scanned == 0)
2249 return MZ_PARAM_ERROR;
2250 if (mz_zip_attrib_is_dir(zip->file_info.external_fa, zip->file_info.version_madeby) == MZ_OK)
2251 return MZ_OK;
2252
2253 filename_length = (int32_t)strlen(zip->file_info.filename);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002254 if (filename_length > 0) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002255 if ((zip->file_info.filename[filename_length - 1] == '/') ||
2256 (zip->file_info.filename[filename_length - 1] == '\\'))
2257 return MZ_OK;
2258 }
2259 return MZ_EXIST_ERROR;
2260}
2261
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002262int32_t mz_zip_entry_is_symlink(void *handle) {
Nathan Moinvaziri3249eac2019-05-02 21:07:39 -07002263 mz_zip *zip = (mz_zip *)handle;
2264
2265 if (zip == NULL)
2266 return MZ_PARAM_ERROR;
2267 if (zip->entry_scanned == 0)
2268 return MZ_PARAM_ERROR;
Nathan Moinvaziri426543d2019-05-07 09:46:25 -07002269 if (mz_zip_attrib_is_symlink(zip->file_info.external_fa, zip->file_info.version_madeby) != MZ_OK)
2270 return MZ_EXIST_ERROR;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -07002271 if (zip->file_info.linkname == NULL || *zip->file_info.linkname == 0)
2272 return MZ_EXIST_ERROR;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002273
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -07002274 return MZ_OK;
Nathan Moinvaziri3249eac2019-05-02 21:07:39 -07002275}
2276
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002277int32_t mz_zip_entry_get_info(void *handle, mz_zip_file **file_info) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002278 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002279
2280 if (zip == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002281 return MZ_PARAM_ERROR;
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002282
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002283 if ((zip->open_mode & MZ_OPEN_MODE_WRITE) == 0) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002284 if (!zip->entry_scanned)
2285 return MZ_PARAM_ERROR;
2286 }
2287
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002288 *file_info = &zip->file_info;
2289 return MZ_OK;
2290}
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002291
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002292int32_t mz_zip_entry_get_local_info(void *handle, mz_zip_file **local_file_info) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002293 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07002294 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002295 return MZ_PARAM_ERROR;
2296 *local_file_info = &zip->local_file_info;
2297 return MZ_OK;
2298}
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002299
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002300int32_t mz_zip_entry_set_extrafield(void *handle, const uint8_t *extrafield, uint16_t extrafield_size) {
Nathan Moinvaziri915c5132018-10-26 20:00:52 -07002301 mz_zip *zip = (mz_zip *)handle;
2302
2303 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
2304 return MZ_PARAM_ERROR;
2305
2306 zip->file_info.extrafield = extrafield;
2307 zip->file_info.extrafield_size = extrafield_size;
2308 return MZ_OK;
2309}
2310
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002311static int32_t mz_zip_goto_next_entry_int(void *handle) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002312 mz_zip *zip = (mz_zip *)handle;
2313 int32_t err = MZ_OK;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002314
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002315 if (zip == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002316 return MZ_PARAM_ERROR;
2317
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002318 zip->entry_scanned = 0;
2319
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002320 mz_stream_set_prop_int64(zip->cd_stream, MZ_STREAM_PROP_DISK_NUMBER, -1);
Viktor Szakats915b82e2018-04-24 10:02:39 +00002321
Nathan Moinvaziri79ac3382017-10-23 17:22:36 -07002322 err = mz_stream_seek(zip->cd_stream, zip->cd_current_pos, MZ_SEEK_SET);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002323 if (err == MZ_OK)
Nathan Moinvaziricda36002017-10-21 09:37:18 -07002324 err = mz_zip_entry_read_header(zip->cd_stream, 0, &zip->file_info, zip->file_info_stream);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002325 if (err == MZ_OK)
2326 zip->entry_scanned = 1;
2327 return err;
2328}
2329
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002330int64_t mz_zip_get_entry(void *handle) {
Viktor Szakatsce26dba2018-04-22 19:30:34 +00002331 mz_zip *zip = (mz_zip *)handle;
2332
2333 if (zip == NULL)
2334 return MZ_PARAM_ERROR;
2335
2336 return zip->cd_current_pos;
2337}
2338
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002339int32_t mz_zip_goto_entry(void *handle, int64_t cd_pos) {
Viktor Szakatsce26dba2018-04-22 19:30:34 +00002340 mz_zip *zip = (mz_zip *)handle;
2341
2342 if (zip == NULL)
2343 return MZ_PARAM_ERROR;
2344
2345 if (cd_pos < zip->cd_start_pos || cd_pos > zip->cd_start_pos + zip->cd_size)
2346 return MZ_PARAM_ERROR;
2347
2348 zip->cd_current_pos = cd_pos;
2349
2350 return mz_zip_goto_next_entry_int(handle);
2351}
2352
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002353int32_t mz_zip_goto_first_entry(void *handle) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002354 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002355
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002356 if (zip == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002357 return MZ_PARAM_ERROR;
2358
Nathan Moinvaziricda36002017-10-21 09:37:18 -07002359 zip->cd_current_pos = zip->cd_start_pos;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002360
2361 return mz_zip_goto_next_entry_int(handle);
2362}
2363
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002364int32_t mz_zip_goto_next_entry(void *handle) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002365 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002366
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002367 if (zip == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002368 return MZ_PARAM_ERROR;
2369
Nathan Moinvaziri904c4082018-10-08 23:31:21 -07002370 zip->cd_current_pos += (int64_t)MZ_ZIP_SIZE_CD_ITEM + zip->file_info.filename_size +
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002371 zip->file_info.extrafield_size + zip->file_info.comment_size;
2372
2373 return mz_zip_goto_next_entry_int(handle);
2374}
2375
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002376int32_t mz_zip_locate_entry(void *handle, const char *filename, uint8_t ignore_case) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002377 mz_zip *zip = (mz_zip *)handle;
2378 int32_t err = MZ_OK;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002379 int32_t result = 0;
2380
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07002381 if (zip == NULL || filename == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002382 return MZ_PARAM_ERROR;
2383
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002384 /* If we are already on the current entry, no need to search */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002385 if ((zip->entry_scanned) && (zip->file_info.filename != NULL)) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002386 result = mz_zip_path_compare(zip->file_info.filename, filename, ignore_case);
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07002387 if (result == 0)
2388 return MZ_OK;
2389 }
2390
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002391 /* Search all entries starting at the first */
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002392 err = mz_zip_goto_first_entry(handle);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002393 while (err == MZ_OK) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002394 result = mz_zip_path_compare(zip->file_info.filename, filename, ignore_case);
2395 if (result == 0)
2396 return MZ_OK;
2397
2398 err = mz_zip_goto_next_entry(handle);
2399 }
2400
2401 return err;
2402}
2403
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002404int32_t mz_zip_locate_first_entry(void *handle, void *userdata, mz_zip_locate_entry_cb cb) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002405 mz_zip *zip = (mz_zip *)handle;
2406 int32_t err = MZ_OK;
2407 int32_t result = 0;
2408
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002409 /* Search first entry looking for match */
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002410 err = mz_zip_goto_first_entry(handle);
2411 if (err != MZ_OK)
2412 return err;
2413
2414 result = cb(handle, userdata, &zip->file_info);
2415 if (result == 0)
2416 return MZ_OK;
2417
2418 return mz_zip_locate_next_entry(handle, userdata, cb);
2419}
2420
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002421int32_t mz_zip_locate_next_entry(void *handle, void *userdata, mz_zip_locate_entry_cb cb) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002422 mz_zip *zip = (mz_zip *)handle;
2423 int32_t err = MZ_OK;
2424 int32_t result = 0;
2425
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002426 /* Search next entries looking for match */
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002427 err = mz_zip_goto_next_entry(handle);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002428 while (err == MZ_OK) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002429 result = cb(handle, userdata, &zip->file_info);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002430 if (result == 0)
2431 return MZ_OK;
2432
2433 err = mz_zip_goto_next_entry(handle);
2434 }
2435
2436 return err;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002437}
2438
2439/***************************************************************************/
2440
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002441int32_t mz_zip_attrib_is_dir(uint32_t attrib, int32_t version_madeby) {
Nathan Moinvaziri2bb21d72018-10-08 21:47:15 -07002442 uint32_t posix_attrib = 0;
Nathan Moinvaziri904c4082018-10-08 23:31:21 -07002443 uint8_t system = MZ_HOST_SYSTEM(version_madeby);
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002444 int32_t err = MZ_OK;
Nathan Moinvaziribb3b75b2018-05-02 10:48:51 -07002445
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002446 err = mz_zip_attrib_convert(system, attrib, MZ_HOST_SYSTEM_UNIX, &posix_attrib);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002447 if (err == MZ_OK) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002448 if ((posix_attrib & 0170000) == 0040000) /* S_ISDIR */
Nathan Moinvaziribb3b75b2018-05-02 10:48:51 -07002449 return MZ_OK;
2450 }
2451
2452 return MZ_EXIST_ERROR;
2453}
2454
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002455int32_t mz_zip_attrib_is_symlink(uint32_t attrib, int32_t version_madeby) {
Nathan Moinvaziri426543d2019-05-07 09:46:25 -07002456 uint32_t posix_attrib = 0;
2457 uint8_t system = MZ_HOST_SYSTEM(version_madeby);
2458 int32_t err = MZ_OK;
2459
2460 err = mz_zip_attrib_convert(system, attrib, MZ_HOST_SYSTEM_UNIX, &posix_attrib);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002461 if (err == MZ_OK) {
Nathan Moinvaziri426543d2019-05-07 09:46:25 -07002462 if ((posix_attrib & 0170000) == 0120000) /* S_ISLNK */
2463 return MZ_OK;
2464 }
2465
2466 return MZ_EXIST_ERROR;
2467}
2468
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002469int32_t mz_zip_attrib_convert(uint8_t src_sys, uint32_t src_attrib, uint8_t target_sys, uint32_t *target_attrib) {
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002470 if (target_attrib == NULL)
2471 return MZ_PARAM_ERROR;
2472
2473 *target_attrib = 0;
2474
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002475 if ((src_sys == MZ_HOST_SYSTEM_MSDOS) || (src_sys == MZ_HOST_SYSTEM_WINDOWS_NTFS)) {
2476 if ((target_sys == MZ_HOST_SYSTEM_MSDOS) || (target_sys == MZ_HOST_SYSTEM_WINDOWS_NTFS)) {
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002477 *target_attrib = src_attrib;
2478 return MZ_OK;
2479 }
Cameron Cawley28f78ff2019-08-06 14:14:14 +01002480 if ((target_sys == MZ_HOST_SYSTEM_UNIX) || (target_sys == MZ_HOST_SYSTEM_OSX_DARWIN) || (target_sys == MZ_HOST_SYSTEM_RISCOS))
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002481 return mz_zip_attrib_win32_to_posix(src_attrib, target_attrib);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002482 } else if ((src_sys == MZ_HOST_SYSTEM_UNIX) || (src_sys == MZ_HOST_SYSTEM_OSX_DARWIN) || (src_sys == MZ_HOST_SYSTEM_RISCOS)) {
2483 if ((target_sys == MZ_HOST_SYSTEM_UNIX) || (target_sys == MZ_HOST_SYSTEM_OSX_DARWIN) || (target_sys == MZ_HOST_SYSTEM_RISCOS)) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002484 /* If high bytes are set, it contains unix specific attributes */
Nathan Moinvaziri2bb21d72018-10-08 21:47:15 -07002485 if ((src_attrib >> 16) != 0)
2486 src_attrib >>= 16;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002487
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002488 *target_attrib = src_attrib;
2489 return MZ_OK;
2490 }
Nathan Moinvaziri2bb21d72018-10-08 21:47:15 -07002491 if ((target_sys == MZ_HOST_SYSTEM_MSDOS) || (target_sys == MZ_HOST_SYSTEM_WINDOWS_NTFS))
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002492 return mz_zip_attrib_posix_to_win32(src_attrib, target_attrib);
2493 }
2494
2495 return MZ_SUPPORT_ERROR;
2496}
2497
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002498int32_t mz_zip_attrib_posix_to_win32(uint32_t posix_attrib, uint32_t *win32_attrib) {
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002499 if (win32_attrib == NULL)
2500 return MZ_PARAM_ERROR;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002501
Nathan Moinvaziri2bb21d72018-10-08 21:47:15 -07002502 *win32_attrib = 0;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002503
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002504 /* S_IWUSR | S_IWGRP | S_IWOTH | S_IXUSR | S_IXGRP | S_IXOTH */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002505 if ((posix_attrib & 0000333) == 0 && (posix_attrib & 0000444) != 0)
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002506 *win32_attrib |= 0x01; /* FILE_ATTRIBUTE_READONLY */
Nathan Moinvaziri426543d2019-05-07 09:46:25 -07002507 /* S_IFLNK */
2508 if ((posix_attrib & 0170000) == 0120000)
2509 *win32_attrib |= 0x400; /* FILE_ATTRIBUTE_REPARSE_POINT */
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002510 /* S_IFDIR */
Nathan Moinvaziri7d94b1f2019-05-08 23:41:47 -07002511 else if ((posix_attrib & 0170000) == 0040000)
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002512 *win32_attrib |= 0x10; /* FILE_ATTRIBUTE_DIRECTORY */
2513 /* S_IFREG */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002514 else
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002515 *win32_attrib |= 0x80; /* FILE_ATTRIBUTE_NORMAL */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002516
2517 return MZ_OK;
2518}
2519
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002520int32_t mz_zip_attrib_win32_to_posix(uint32_t win32_attrib, uint32_t *posix_attrib) {
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002521 if (posix_attrib == NULL)
2522 return MZ_PARAM_ERROR;
2523
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002524 *posix_attrib = 0000444; /* S_IRUSR | S_IRGRP | S_IROTH */
2525 /* FILE_ATTRIBUTE_READONLY */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002526 if ((win32_attrib & 0x01) == 0)
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002527 *posix_attrib |= 0000222; /* S_IWUSR | S_IWGRP | S_IWOTH */
Nathan Moinvaziri426543d2019-05-07 09:46:25 -07002528 /* FILE_ATTRIBUTE_REPARSE_POINT */
2529 if ((win32_attrib & 0x400) == 0x400)
2530 *posix_attrib |= 0120000; /* S_IFLNK */
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002531 /* FILE_ATTRIBUTE_DIRECTORY */
Nathan Moinvaziri7d94b1f2019-05-08 23:41:47 -07002532 else if ((win32_attrib & 0x10) == 0x10)
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002533 *posix_attrib |= 0040111; /* S_IFDIR | S_IXUSR | S_IXGRP | S_IXOTH */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002534 else
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002535 *posix_attrib |= 0100000; /* S_IFREG */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002536
2537 return MZ_OK;
2538}
2539
Nathan Moinvaziribb3b75b2018-05-02 10:48:51 -07002540/***************************************************************************/
2541
Nathan Moinvaziri8f769682020-12-23 11:26:30 -08002542int32_t mz_zip_extrafield_find(void *stream, uint16_t type, int32_t max_seek, uint16_t *length) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002543 int32_t err = MZ_OK;
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002544 uint16_t field_type = 0;
2545 uint16_t field_length = 0;
2546
Nathan Moinvaziri8f769682020-12-23 11:26:30 -08002547
2548 if (max_seek < 4)
2549 return MZ_EXIST_ERROR;
2550
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002551 do {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002552 err = mz_stream_read_uint16(stream, &field_type);
2553 if (err == MZ_OK)
2554 err = mz_stream_read_uint16(stream, &field_length);
2555 if (err != MZ_OK)
2556 break;
2557
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002558 if (type == field_type) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002559 if (length != NULL)
2560 *length = field_length;
2561 return MZ_OK;
2562 }
2563
Nathan Moinvaziri8f769682020-12-23 11:26:30 -08002564 max_seek -= field_length - 4;
2565 if (max_seek < 0)
2566 return MZ_EXIST_ERROR;
2567
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08002568 err = mz_stream_seek(stream, field_length, MZ_SEEK_CUR);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002569 } while (err == MZ_OK);
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002570
2571 return MZ_EXIST_ERROR;
2572}
2573
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002574int32_t mz_zip_extrafield_contains(const uint8_t *extrafield, int32_t extrafield_size,
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002575 uint16_t type, uint16_t *length) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002576 void *file_extra_stream = NULL;
2577 int32_t err = MZ_OK;
2578
2579 if (extrafield == NULL || extrafield_size == 0)
2580 return MZ_PARAM_ERROR;
2581
2582 mz_stream_mem_create(&file_extra_stream);
2583 mz_stream_mem_set_buffer(file_extra_stream, (void *)extrafield, extrafield_size);
2584
Nathan Moinvaziri8f769682020-12-23 11:26:30 -08002585 err = mz_zip_extrafield_find(file_extra_stream, type, extrafield_size, length);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002586
2587 mz_stream_mem_delete(&file_extra_stream);
2588
2589 return err;
2590}
2591
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002592int32_t mz_zip_extrafield_read(void *stream, uint16_t *type, uint16_t *length) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002593 int32_t err = MZ_OK;
2594 if (type == NULL || length == NULL)
2595 return MZ_PARAM_ERROR;
2596 err = mz_stream_read_uint16(stream, type);
2597 if (err == MZ_OK)
2598 err = mz_stream_read_uint16(stream, length);
2599 return err;
2600}
2601
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002602int32_t mz_zip_extrafield_write(void *stream, uint16_t type, uint16_t length) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002603 int32_t err = MZ_OK;
2604 err = mz_stream_write_uint16(stream, type);
2605 if (err == MZ_OK)
2606 err = mz_stream_write_uint16(stream, length);
2607 return err;
2608}
2609
2610/***************************************************************************/
2611
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002612static int32_t mz_zip_invalid_date(const struct tm *ptm) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002613#define datevalue_in_range(min, max, value) ((min) <= (value) && (value) <= (max))
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002614 return (!datevalue_in_range(0, 127 + 80, ptm->tm_year) || /* 1980-based year, allow 80 extra */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002615 !datevalue_in_range(0, 11, ptm->tm_mon) ||
2616 !datevalue_in_range(1, 31, ptm->tm_mday) ||
2617 !datevalue_in_range(0, 23, ptm->tm_hour) ||
2618 !datevalue_in_range(0, 59, ptm->tm_min) ||
2619 !datevalue_in_range(0, 59, ptm->tm_sec));
2620#undef datevalue_in_range
2621}
2622
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002623static void mz_zip_dosdate_to_raw_tm(uint64_t dos_date, struct tm *ptm) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002624 uint64_t date = (uint64_t)(dos_date >> 16);
2625
Nathan Moinvaziri1d6d1832018-11-10 09:03:55 -08002626 ptm->tm_mday = (uint16_t)(date & 0x1f);
2627 ptm->tm_mon = (uint16_t)(((date & 0x1E0) / 0x20) - 1);
2628 ptm->tm_year = (uint16_t)(((date & 0x0FE00) / 0x0200) + 80);
2629 ptm->tm_hour = (uint16_t)((dos_date & 0xF800) / 0x800);
2630 ptm->tm_min = (uint16_t)((dos_date & 0x7E0) / 0x20);
2631 ptm->tm_sec = (uint16_t)(2 * (dos_date & 0x1f));
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002632 ptm->tm_isdst = -1;
2633}
2634
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002635int32_t mz_zip_dosdate_to_tm(uint64_t dos_date, struct tm *ptm) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002636 if (ptm == NULL)
2637 return MZ_PARAM_ERROR;
2638
2639 mz_zip_dosdate_to_raw_tm(dos_date, ptm);
2640
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002641 if (mz_zip_invalid_date(ptm)) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002642 /* Invalid date stored, so don't return it */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002643 memset(ptm, 0, sizeof(struct tm));
2644 return MZ_FORMAT_ERROR;
2645 }
2646 return MZ_OK;
2647}
2648
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002649time_t mz_zip_dosdate_to_time_t(uint64_t dos_date) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002650 struct tm ptm;
2651 mz_zip_dosdate_to_raw_tm(dos_date, &ptm);
2652 return mktime(&ptm);
2653}
2654
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002655int32_t mz_zip_time_t_to_tm(time_t unix_time, struct tm *ptm) {
Nathan Moinvaziricd3e9e12019-04-28 10:22:49 -07002656 struct tm ltm;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002657 if (ptm == NULL)
2658 return MZ_PARAM_ERROR;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002659 if (localtime_r(&unix_time, &ltm) == NULL) { /* Returns a 1900-based year */
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002660 /* Invalid date stored, so don't return it */
Viktor Szakatsce26dba2018-04-22 19:30:34 +00002661 memset(ptm, 0, sizeof(struct tm));
Nathan Moinvaziri17fcdcd2017-10-26 08:13:13 -07002662 return MZ_INTERNAL_ERROR;
Viktor Szakatsce26dba2018-04-22 19:30:34 +00002663 }
Nathan Moinvaziricd3e9e12019-04-28 10:22:49 -07002664 memcpy(ptm, &ltm, sizeof(struct tm));
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002665 return MZ_OK;
2666}
2667
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002668uint32_t mz_zip_time_t_to_dos_date(time_t unix_time) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002669 struct tm ptm;
2670 mz_zip_time_t_to_tm(unix_time, &ptm);
2671 return mz_zip_tm_to_dosdate((const struct tm *)&ptm);
2672}
2673
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002674uint32_t mz_zip_tm_to_dosdate(const struct tm *ptm) {
Nathan Moinvazirie33916b2018-05-01 13:45:08 -07002675 struct tm fixed_tm;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002676
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002677 /* Years supported: */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002678
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002679 /* [00, 79] (assumed to be between 2000 and 2079) */
2680 /* [80, 207] (assumed to be between 1980 and 2107, typical output of old */
2681 /* software that does 'year-1900' to get a double digit year) */
2682 /* [1980, 2107] (due to format limitations, only years 1980-2107 can be stored.) */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002683
2684 memcpy(&fixed_tm, ptm, sizeof(struct tm));
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002685 if (fixed_tm.tm_year >= 1980) /* range [1980, 2107] */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002686 fixed_tm.tm_year -= 1980;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002687 else if (fixed_tm.tm_year >= 80) /* range [80, 207] */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002688 fixed_tm.tm_year -= 80;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002689 else /* range [00, 79] */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002690 fixed_tm.tm_year += 20;
2691
Viktor Szakatsce26dba2018-04-22 19:30:34 +00002692 if (mz_zip_invalid_date(&fixed_tm))
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002693 return 0;
2694
Anand K Mistry57b65f82018-11-09 10:52:19 +11002695 return (((uint32_t)fixed_tm.tm_mday + (32 * ((uint32_t)fixed_tm.tm_mon + 1)) + (512 * (uint32_t)fixed_tm.tm_year)) << 16) |
Nathan Moinvaziri904c4082018-10-08 23:31:21 -07002696 (((uint32_t)fixed_tm.tm_sec / 2) + (32 * (uint32_t)fixed_tm.tm_min) + (2048 * (uint32_t)fixed_tm.tm_hour));
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002697}
2698
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002699int32_t mz_zip_ntfs_to_unix_time(uint64_t ntfs_time, time_t *unix_time) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002700 *unix_time = (time_t)((ntfs_time - 116444736000000000LL) / 10000000);
2701 return MZ_OK;
2702}
2703
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002704int32_t mz_zip_unix_to_ntfs_time(time_t unix_time, uint64_t *ntfs_time) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002705 *ntfs_time = ((uint64_t)unix_time * 10000000) + 116444736000000000LL;
2706 return MZ_OK;
2707}
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002708
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002709/***************************************************************************/
2710
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002711int32_t mz_zip_path_compare(const char *path1, const char *path2, uint8_t ignore_case) {
2712 do {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002713 if ((*path1 == '\\' && *path2 == '/') ||
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002714 (*path2 == '\\' && *path1 == '/')) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002715 /* Ignore comparison of path slashes */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002716 } else if (ignore_case) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002717 if (tolower(*path1) != tolower(*path2))
2718 break;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002719 } else if (*path1 != *path2) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002720 break;
2721 }
2722
2723 path1 += 1;
2724 path2 += 1;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002725 } while (*path1 != 0 && *path2 != 0);
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002726
2727 if (ignore_case)
2728 return (int32_t)(tolower(*path1) - tolower(*path2));
2729
2730 return (int32_t)(*path1 - *path2);
2731}
2732
2733/***************************************************************************/
Nathan Moinvaziric4668ed2020-10-25 11:56:19 -07002734
2735const char* mz_zip_get_compression_method_string(int32_t compression_method)
2736{
2737 const char *method = "?";
Nathan Moinvaziri5cd28792020-10-25 12:50:12 -07002738 switch (compression_method) {
Nathan Moinvaziric4668ed2020-10-25 11:56:19 -07002739 case MZ_COMPRESS_METHOD_STORE:
2740 method = "stored";
2741 break;
2742 case MZ_COMPRESS_METHOD_DEFLATE:
2743 method = "deflate";
2744 break;
2745 case MZ_COMPRESS_METHOD_BZIP2:
2746 method = "bzip2";
2747 break;
2748 case MZ_COMPRESS_METHOD_LZMA:
2749 method = "lzma";
2750 break;
2751 case MZ_COMPRESS_METHOD_XZ:
2752 method = "xz";
2753 break;
2754 case MZ_COMPRESS_METHOD_ZSTD:
2755 method = "zstd";
2756 break;
2757 }
2758 return method;
2759}
2760
2761/***************************************************************************/