blob: 61fa4e222d02e10af38dd433bf7fc6b8bb5d74d4 [file] [log] [blame]
Antoine Cœur0c2fceb2017-10-05 15:30:37 +08001/* zip.c -- Zip manipulation
Nathan Moinvaziridb958942021-01-23 16:18:11 -08002 part of the minizip-ng project
Antoine Cœur0c2fceb2017-10-05 15:30:37 +08003
Nathan Moinvaziridb958942021-01-23 16:18:11 -08004 Copyright (C) 2010-2021 Nathan Moinvaziri
5 https://github.com/zlib-ng/minizip-ng
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
Nathan Moinvaziri00cca9f2017-10-16 07:37:11 -070018#include "mz.h"
Nathan Moinvaziri5f091882018-10-24 18:06:08 -070019#include "mz_crypt.h"
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080020#include "mz_strm.h"
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080021#ifdef HAVE_BZIP2
22# include "mz_strm_bzip.h"
23#endif
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -080024#ifdef HAVE_LIBCOMP
Nathan Moinvaziri90d31c72018-11-20 03:17:20 -080025# include "mz_strm_libcomp.h"
26#endif
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080027#ifdef HAVE_LZMA
28# include "mz_strm_lzma.h"
29#endif
Nathan Moinvazirib47b41b2018-07-11 09:44:29 -070030#include "mz_strm_mem.h"
Nathan Moinvaziri0e5c9dc2018-05-23 20:21:48 -070031#ifdef HAVE_PKCRYPT
32# include "mz_strm_pkcrypt.h"
33#endif
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -080034#ifdef HAVE_WZAES
Nathan Moinvaziri21a31022018-10-24 09:50:16 -070035# include "mz_strm_wzaes.h"
36#endif
Nathan Moinvaziri4f6a0e32017-11-10 08:45:14 -080037#ifdef HAVE_ZLIB
38# include "mz_strm_zlib.h"
39#endif
Nathan Moinvaziri1f4758b2020-06-14 21:52:55 -070040#ifdef HAVE_ZSTD
41# include "mz_strm_zstd.h"
42#endif
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080043
44#include "mz_zip.h"
45
Nathan Moinvazirib16ab562018-11-20 16:56:21 -080046#include <ctype.h> /* tolower */
Nathan Moinvazirie8496272018-11-21 15:05:58 -080047#include <stdio.h> /* snprintf */
Nathan Moinvaziri1ee609b2018-11-19 21:34:35 -080048
zedcf076662019-05-15 21:43:56 +030049#if defined(_MSC_VER) || defined(__MINGW32__)
Nathan Moinvaziri2e0a20a2019-04-28 13:03:11 -070050# define localtime_r(t1,t2) (localtime_s(t2,t1) == 0 ? t1 : NULL)
zed65447dd2019-05-16 22:19:08 +030051#endif
52#if defined(_MSC_VER) && (_MSC_VER < 1900)
53# define snprintf _snprintf
Nathan Moinvaziric565fa82018-10-19 08:48:33 -070054#endif
55
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080056/***************************************************************************/
57
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070058#define MZ_ZIP_MAGIC_LOCALHEADER (0x04034b50)
Nathan Moinvaziribe483d82019-10-19 17:30:23 -070059#define MZ_ZIP_MAGIC_LOCALHEADERU8 { 0x50, 0x4b, 0x03, 0x04 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070060#define MZ_ZIP_MAGIC_CENTRALHEADER (0x02014b50)
Nathan Moinvaziri545d7292019-10-21 15:32:18 -070061#define MZ_ZIP_MAGIC_CENTRALHEADERU8 { 0x50, 0x4b, 0x01, 0x02 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070062#define MZ_ZIP_MAGIC_ENDHEADER (0x06054b50)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -080063#define MZ_ZIP_MAGIC_ENDHEADERU8 { 0x50, 0x4b, 0x05, 0x06 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070064#define MZ_ZIP_MAGIC_ENDHEADER64 (0x06064b50)
65#define MZ_ZIP_MAGIC_ENDLOCHEADER64 (0x07064b50)
66#define MZ_ZIP_MAGIC_DATADESCRIPTOR (0x08074b50)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -080067#define MZ_ZIP_MAGIC_DATADESCRIPTORU8 { 0x50, 0x4b, 0x07, 0x08 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -070068
Nathan Moinvaziriba1db162018-11-23 10:07:35 -080069#define MZ_ZIP_SIZE_LD_ITEM (30)
Nathan Moinvaziri638f31f2018-08-27 08:17:16 -070070#define MZ_ZIP_SIZE_CD_ITEM (46)
71#define MZ_ZIP_SIZE_CD_LOCATOR64 (20)
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -080072#define MZ_ZIP_SIZE_MAX_DATA_DESCRIPTOR (24)
Antoine Cœur0c2fceb2017-10-05 15:30:37 +080073
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -080074#define MZ_ZIP_OFFSET_CRC_SIZES (14)
Nathan Moinvaziri161b2662020-12-23 11:50:24 -080075#define MZ_ZIP_UNCOMPR_SIZE64_CUSHION (2 * 1024 * 1024)
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -080076
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
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700154 *central_pos = 0;
Antoine Cœur0c2fceb2017-10-05 15:30:37 +0800155
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800156 /* Zip64 end of central directory locator */
Nathan Moinvaziri79ac3382017-10-23 17:22:36 -0700157 err = mz_stream_seek(stream, end_central_offset - MZ_ZIP_SIZE_CD_LOCATOR64, MZ_SEEK_SET);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800158 /* Read locator signature */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700159 if (err == MZ_OK) {
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700160 err = mz_stream_read_uint32(stream, &value32);
Nathan Moinvaziri00cca9f2017-10-16 07:37:11 -0700161 if (value32 != MZ_ZIP_MAGIC_ENDLOCHEADER64)
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700162 err = MZ_FORMAT_ERROR;
163 }
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800164 /* Number of the disk with the start of the zip64 end of central directory */
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700165 if (err == MZ_OK)
166 err = mz_stream_read_uint32(stream, &value32);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800167 /* Relative offset of the zip64 end of central directory record8 */
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700168 if (err == MZ_OK)
Nathan Moinvaziri904c4082018-10-08 23:31:21 -0700169 err = mz_stream_read_uint64(stream, (uint64_t *)&offset);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800170 /* Total number of disks */
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700171 if (err == MZ_OK)
172 err = mz_stream_read_uint32(stream, &value32);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800173 /* Goto end of central directory record */
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700174 if (err == MZ_OK)
Nathan Moinvaziri904c4082018-10-08 23:31:21 -0700175 err = mz_stream_seek(stream, (int64_t)offset, MZ_SEEK_SET);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800176 /* The signature */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700177 if (err == MZ_OK) {
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700178 err = mz_stream_read_uint32(stream, &value32);
Nathan Moinvaziri00cca9f2017-10-16 07:37:11 -0700179 if (value32 != MZ_ZIP_MAGIC_ENDHEADER64)
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700180 err = MZ_FORMAT_ERROR;
181 }
Antoine Cœur0c2fceb2017-10-05 15:30:37 +0800182
Nathan Moinvaziri3a0fb552017-10-10 12:47:48 -0700183 if (err == MZ_OK)
184 *central_pos = offset;
185
186 return err;
Antoine Cœur0c2fceb2017-10-05 15:30:37 +0800187}
188
hekkihek15e2bb72022-07-16 16:31:21 +0200189#ifdef HAVE_PKCRYPT
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}
hekkihek15e2bb72022-07-16 16:31:21 +0200199#endif
Nathan Moinvaziri215a7a82020-12-13 10:58:41 -0800200
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800201/* Get info about the current file in the zip file */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700202static 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 -0700203 uint64_t ntfs_time = 0;
204 uint32_t reserved = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700205 uint32_t magic = 0;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700206 uint32_t dos_date = 0;
Nathan Moinvaziri69e35492018-11-05 15:48:41 -0800207 uint32_t field_pos = 0;
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700208 uint16_t field_type = 0;
209 uint16_t field_length = 0;
210 uint32_t field_length_read = 0;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700211 uint16_t ntfs_attrib_id = 0;
212 uint16_t ntfs_attrib_size = 0;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700213 uint16_t linkname_size;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700214 uint16_t value16 = 0;
215 uint32_t value32 = 0;
Nathan Moinvazirif9e34482018-11-27 09:45:09 -0800216 int64_t extrafield_pos = 0;
217 int64_t comment_pos = 0;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700218 int64_t linkname_pos = 0;
219 int64_t saved_pos = 0;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -0700220 int32_t err = MZ_OK;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700221 char *linkname = NULL;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700222
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700223 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
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800487 err = mz_stream_read_uint32(stream, &value32);
488 if (value32 != MZ_ZIP_MAGIC_DATADESCRIPTOR)
489 err = MZ_FORMAT_ERROR;
490 if (err == MZ_OK)
491 err = mz_stream_read_uint32(stream, &value32);
492 if ((err == MZ_OK) && (crc32 != NULL))
493 *crc32 = value32;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700494 if (err == MZ_OK) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800495 /* If zip 64 extension is enabled then read as 8 byte */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700496 if (!zip64) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800497 err = mz_stream_read_uint32(stream, &value32);
498 value64 = value32;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700499 } else {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800500 err = mz_stream_read_int64(stream, &value64);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800501 if (value64 < 0)
502 err = MZ_FORMAT_ERROR;
503 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800504 if ((err == MZ_OK) && (compressed_size != NULL))
505 *compressed_size = value64;
506 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700507 if (err == MZ_OK) {
508 if (!zip64) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800509 err = mz_stream_read_uint32(stream, &value32);
510 value64 = value32;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700511 } else {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800512 err = mz_stream_read_int64(stream, &value64);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800513 if (value64 < 0)
514 err = MZ_FORMAT_ERROR;
515 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800516 if ((err == MZ_OK) && (uncompressed_size != NULL))
517 *uncompressed_size = value64;
518 }
519
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700520 return err;
521}
522
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800523static 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 -0800524 int32_t err = MZ_OK;
525
526 if (mask)
527 err = mz_stream_write_uint32(stream, 0);
528 else
529 err = mz_stream_write_uint32(stream, file_info->crc); /* crc */
530
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800531 /* For backwards-compatibility with older zip applications we set all sizes to UINT32_MAX
532 * when zip64 is needed, instead of only setting sizes larger than UINT32_MAX. */
533
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700534 if (err == MZ_OK) {
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800535 if (zip64) /* compr size */
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -0800536 err = mz_stream_write_uint32(stream, UINT32_MAX);
537 else
538 err = mz_stream_write_uint32(stream, (uint32_t)file_info->compressed_size);
539 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700540 if (err == MZ_OK) {
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800541 if (mask) /* uncompr size */
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -0800542 err = mz_stream_write_uint32(stream, 0);
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800543 else if (zip64)
544 err = mz_stream_write_uint32(stream, UINT32_MAX);
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -0800545 else
546 err = mz_stream_write_uint32(stream, (uint32_t)file_info->uncompressed_size);
547 }
548 return err;
549}
550
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800551static int32_t mz_zip_entry_needs_zip64(mz_zip_file *file_info, uint8_t local, uint8_t *zip64) {
Vlad Lipskiy4b252192020-12-23 18:35:37 +0300552 uint32_t max_uncompressed_size = UINT32_MAX;
Nathan Moinvazirid183a402020-12-23 22:16:36 -0800553 uint8_t needs_zip64 = 0;
Vlad Lipskiy4b252192020-12-23 18:35:37 +0300554
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800555 if (zip64 == NULL)
556 return MZ_PARAM_ERROR;
557
Nathan Moinvazirica270892020-12-27 14:54:57 -0800558 *zip64 = 0;
559
Nathan Moinvaziri4cc3ffd2020-12-24 09:27:29 -0800560 if (local) {
Nathan Moinvaziri4cc3ffd2020-12-24 09:27:29 -0800561 /* At local header we might not know yet whether compressed size will overflow unsigned
562 32-bit integer which might happen for high entropy data so we give it some cushion */
563
Nathan Moinvaziri161b2662020-12-23 11:50:24 -0800564 max_uncompressed_size -= MZ_ZIP_UNCOMPR_SIZE64_CUSHION;
Nathan Moinvaziri4cc3ffd2020-12-24 09:27:29 -0800565 }
Nathan Moinvaziri161b2662020-12-23 11:50:24 -0800566
Vlad Lipskiy4b252192020-12-23 18:35:37 +0300567 needs_zip64 = (file_info->uncompressed_size >= max_uncompressed_size) ||
Nathan Moinvazirica270892020-12-27 14:54:57 -0800568 (file_info->compressed_size >= UINT32_MAX);
569
570 if (!local) {
571 /* Disk offset and number only used in central directory header */
572 needs_zip64 |= (file_info->disk_offset >= UINT32_MAX) ||
573 (file_info->disk_number >= UINT16_MAX);
574 }
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800575
576 if (file_info->zip64 == MZ_ZIP64_AUTO) {
577 /* If uncompressed size is unknown, assume zip64 for 64-bit data descriptors */
Nathan Moinvazirica270892020-12-27 14:54:57 -0800578 if (local && file_info->uncompressed_size == 0) {
579 /* Don't use zip64 for local header directory entries */
580 if (mz_zip_attrib_is_dir(file_info->external_fa, file_info->version_madeby) != MZ_OK) {
581 *zip64 = 1;
582 }
583 }
584 *zip64 |= needs_zip64;
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800585 } else if (file_info->zip64 == MZ_ZIP64_FORCE) {
586 *zip64 = 1;
587 } else if (file_info->zip64 == MZ_ZIP64_DISABLE) {
588 /* Zip64 extension is required to zip file */
589 if (needs_zip64)
590 return MZ_PARAM_ERROR;
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800591 }
592
593 return MZ_OK;
594}
595
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700596static int32_t mz_zip_entry_write_header(void *stream, uint8_t local, mz_zip_file *file_info) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700597 uint64_t ntfs_time = 0;
598 uint32_t reserved = 0;
599 uint32_t dos_date = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700600 uint16_t extrafield_size = 0;
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700601 uint16_t field_type = 0;
602 uint16_t field_length = 0;
603 uint16_t field_length_zip64 = 0;
604 uint16_t field_length_ntfs = 0;
605 uint16_t field_length_aes = 0;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700606 uint16_t field_length_unix1 = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700607 uint16_t filename_size = 0;
Nathan Moinvaziri3ad1a922018-05-02 13:04:41 -0700608 uint16_t filename_length = 0;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700609 uint16_t linkname_size = 0;
Nathan Moinvaziri46f71432018-05-03 17:24:32 -0700610 uint16_t version_needed = 0;
Nathan Moinvaziridd103f42018-11-13 14:50:32 -0800611 int32_t comment_size = 0;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -0700612 int32_t err = MZ_OK;
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700613 int32_t err_mem = MZ_OK;
614 uint8_t zip64 = 0;
615 uint8_t skip_aes = 0;
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700616 uint8_t mask = 0;
617 uint8_t write_end_slash = 0;
618 const char *filename = NULL;
Nathan Moinvazirifa620e42018-10-20 10:37:10 -0700619 char masked_name[64];
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700620 void *file_extra_stream = NULL;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700621
622 if (file_info == NULL)
623 return MZ_PARAM_ERROR;
624
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700625 if ((local) && (file_info->flag & MZ_ZIP_FLAG_MASK_LOCAL_INFO))
626 mask = 1;
627
Nathan Moinvaziri8516c572020-12-23 21:59:35 -0800628 /* Determine if zip64 extra field is necessary */
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800629 err = mz_zip_entry_needs_zip64(file_info, local, &zip64);
630 if (err != MZ_OK)
631 return err;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -0700632
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800633 /* Start calculating extra field sizes */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700634 if (zip64) {
Nathan Moinvaziri3afaad42020-12-23 12:02:57 -0800635 /* Both compressed and uncompressed sizes must be included (at least in local header) */
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -0800636 field_length_zip64 = 8 + 8;
637 if ((!local) && (file_info->disk_offset >= UINT32_MAX))
638 field_length_zip64 += 8;
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800639
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700640 extrafield_size += 4;
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700641 extrafield_size += field_length_zip64;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700642 }
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700643
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800644 /* Calculate extra field size and check for duplicates */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700645 if (file_info->extrafield_size > 0) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700646 mz_stream_mem_create(&file_extra_stream);
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -0700647 mz_stream_mem_set_buffer(file_extra_stream, (void *)file_info->extrafield,
Nathan Moinvaziri915c5132018-10-26 20:00:52 -0700648 file_info->extrafield_size);
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700649
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700650 do {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700651 err_mem = mz_stream_read_uint16(file_extra_stream, &field_type);
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700652 if (err_mem == MZ_OK)
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700653 err_mem = mz_stream_read_uint16(file_extra_stream, &field_length);
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700654 if (err_mem != MZ_OK)
655 break;
656
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800657 /* Prefer incoming aes extensions over ours */
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700658 if (field_type == MZ_ZIP_EXTENSION_AES)
659 skip_aes = 1;
660
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700661 /* Prefer our zip64, ntfs, unix1 extension over incoming */
662 if (field_type != MZ_ZIP_EXTENSION_ZIP64 && field_type != MZ_ZIP_EXTENSION_NTFS &&
663 field_type != MZ_ZIP_EXTENSION_UNIX1)
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700664 extrafield_size += 4 + field_length;
665
666 if (err_mem == MZ_OK)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800667 err_mem = mz_stream_seek(file_extra_stream, field_length, MZ_SEEK_CUR);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700668 } while (err_mem == MZ_OK);
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700669 }
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -0700670
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -0800671#ifdef HAVE_WZAES
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700672 if (!skip_aes) {
673 if ((file_info->flag & MZ_ZIP_FLAG_ENCRYPTED) && (file_info->aes_version)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700674 field_length_aes = 1 + 1 + 1 + 2 + 2;
675 extrafield_size += 4 + field_length_aes;
Nathan Moinvaziri413822a2018-10-20 09:45:07 -0700676 }
677 }
Nathan Moinvaziria2bb08c2018-10-28 13:45:30 -0700678#else
Nathan Moinvaziri7de94cf2018-11-19 20:51:06 -0800679 MZ_UNUSED(field_length_aes);
Nathan Moinvaziri4ae469a2018-10-28 15:32:04 -0700680 MZ_UNUSED(skip_aes);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700681#endif
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800682 /* NTFS timestamps */
Nathan Moinvazirid9e159a2018-02-13 15:30:30 -0800683 if ((file_info->modified_date != 0) &&
684 (file_info->accessed_date != 0) &&
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700685 (file_info->creation_date != 0) && (!mask)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700686 field_length_ntfs = 8 + 8 + 8 + 4 + 2 + 2;
687 extrafield_size += 4 + field_length_ntfs;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700688 }
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -0700689
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700690 /* Unix1 symbolic links */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700691 if (file_info->linkname != NULL && *file_info->linkname != 0) {
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700692 linkname_size = (uint16_t)strlen(file_info->linkname);
693 field_length_unix1 = 12 + linkname_size;
694 extrafield_size += 4 + field_length_unix1;
695 }
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -0700696
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700697 if (local)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700698 err = mz_stream_write_uint32(stream, MZ_ZIP_MAGIC_LOCALHEADER);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700699 else {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700700 err = mz_stream_write_uint32(stream, MZ_ZIP_MAGIC_CENTRALHEADER);
701 if (err == MZ_OK)
702 err = mz_stream_write_uint16(stream, file_info->version_madeby);
703 }
704
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800705 /* Calculate version needed to extract */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700706 if (err == MZ_OK) {
Nathan Moinvaziri46f71432018-05-03 17:24:32 -0700707 version_needed = file_info->version_needed;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700708 if (version_needed == 0) {
Nathan Moinvaziri46f71432018-05-03 17:24:32 -0700709 version_needed = 20;
710 if (zip64)
711 version_needed = 45;
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -0800712#ifdef HAVE_WZAES
Nathan Moinvaziri46f71432018-05-03 17:24:32 -0700713 if ((file_info->flag & MZ_ZIP_FLAG_ENCRYPTED) && (file_info->aes_version))
714 version_needed = 51;
715#endif
Nathan Moinvaziri1a7e7e42020-10-25 14:50:53 -0700716#if defined(HAVE_LZMA) || defined(HAVE_LIBCOMP)
Nathan Moinvaziric4668ed2020-10-25 11:56:19 -0700717 if ((file_info->compression_method == MZ_COMPRESS_METHOD_LZMA) ||
718 (file_info->compression_method == MZ_COMPRESS_METHOD_XZ))
Nathan Moinvaziri46f71432018-05-03 17:24:32 -0700719 version_needed = 63;
720#endif
721 }
722 err = mz_stream_write_uint16(stream, version_needed);
723 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700724 if (err == MZ_OK)
725 err = mz_stream_write_uint16(stream, file_info->flag);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700726 if (err == MZ_OK) {
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -0800727#ifdef HAVE_WZAES
Nathan Moinvaziri413822a2018-10-20 09:45:07 -0700728 if ((file_info->flag & MZ_ZIP_FLAG_ENCRYPTED) && (file_info->aes_version))
Nathan Moinvaziri6e21e862017-10-19 09:57:54 -0700729 err = mz_stream_write_uint16(stream, MZ_COMPRESS_METHOD_AES);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700730 else
731#endif
732 err = mz_stream_write_uint16(stream, file_info->compression_method);
733 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700734 if (err == MZ_OK) {
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700735 if (file_info->modified_date != 0 && !mask)
Nathan Moinvaziri17cdaec2017-10-26 10:18:41 -0700736 dos_date = mz_zip_time_t_to_dos_date(file_info->modified_date);
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700737 err = mz_stream_write_uint32(stream, dos_date);
738 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700739
740 if (err == MZ_OK)
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800741 err = mz_zip_entry_write_crc_sizes(stream, zip64, mask, file_info);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700742
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700743 if (mask) {
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -0700744 snprintf(masked_name, sizeof(masked_name), "%" PRIx32 "_%" PRIx64,
Nathan Moinvaziri264dc182018-11-13 17:42:13 -0800745 file_info->disk_number, file_info->disk_offset);
Nathan Moinvazirifa620e42018-10-20 10:37:10 -0700746 filename = masked_name;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700747 } else {
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700748 filename = file_info->filename;
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700749 }
750
751 filename_length = (uint16_t)strlen(filename);
752 filename_size += filename_length;
753
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800754 if ((mz_zip_attrib_is_dir(file_info->external_fa, file_info->version_madeby) == MZ_OK) &&
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700755 ((filename[filename_length - 1] != '/') && (filename[filename_length - 1] != '\\'))) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800756 filename_size += 1;
757 write_end_slash = 1;
758 }
759
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700760 if (err == MZ_OK)
761 err = mz_stream_write_uint16(stream, filename_size);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700762 if (err == MZ_OK)
763 err = mz_stream_write_uint16(stream, extrafield_size);
764
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700765 if (!local) {
766 if (file_info->comment != NULL) {
Nathan Moinvaziridd103f42018-11-13 14:50:32 -0800767 comment_size = (int32_t)strlen(file_info->comment);
768 if (comment_size > UINT16_MAX)
769 comment_size = UINT16_MAX;
770 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700771 if (err == MZ_OK)
Nathan Moinvaziridd103f42018-11-13 14:50:32 -0800772 err = mz_stream_write_uint16(stream, (uint16_t)comment_size);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700773 if (err == MZ_OK)
Nathan Moinvazirifd039e32017-10-22 14:40:39 -0700774 err = mz_stream_write_uint16(stream, (uint16_t)file_info->disk_number);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700775 if (err == MZ_OK)
776 err = mz_stream_write_uint16(stream, file_info->internal_fa);
777 if (err == MZ_OK)
778 err = mz_stream_write_uint32(stream, file_info->external_fa);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700779 if (err == MZ_OK) {
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -0800780 if (file_info->disk_offset >= UINT32_MAX)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700781 err = mz_stream_write_uint32(stream, UINT32_MAX);
782 else
783 err = mz_stream_write_uint32(stream, (uint32_t)file_info->disk_offset);
784 }
785 }
786
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700787 if (err == MZ_OK) {
Nathan Moinvaziri6005ae72022-08-06 10:13:08 -0700788 const char *backslash = NULL;
789 const char *next = filename;
790 int32_t left = filename_length;
Nathan Moinvaziria4b0c782022-11-29 16:51:20 -0800791
Nathan Moinvaziri6005ae72022-08-06 10:13:08 -0700792 /* Ensure all slashes are written as forward slashes according to 4.4.17.1 */
793 while ((err == MZ_OK) && (backslash = strrchr(next, '\\')) != NULL) {
794 int32_t part_length = (int32_t)(backslash - next);
795
796 if (mz_stream_write(stream, next, part_length) != part_length ||
Nathan Moinvaziria4b0c782022-11-29 16:51:20 -0800797 mz_stream_write(stream, "/", 1) != 1)
Nathan Moinvaziri6005ae72022-08-06 10:13:08 -0700798 err = MZ_WRITE_ERROR;
799
800 left -= part_length + 1;
801 next = backslash + 1;
802 }
803
804 if (err == MZ_OK && left > 0) {
805 if (mz_stream_write(stream, next, left) != left)
806 err = MZ_WRITE_ERROR;
807 }
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700808
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800809 /* Ensure that directories have a slash appended to them for compatibility */
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700810 if (err == MZ_OK && write_end_slash)
811 err = mz_stream_write_uint8(stream, '/');
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700812 }
Nathan Moinvaziri298126a2018-07-31 10:19:48 -0700813
Nathan Moinvaziri8516c572020-12-23 21:59:35 -0800814 /* Write ZIP64 extra field first so we can update sizes later if data descriptor not used */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700815 if ((err == MZ_OK) && (zip64)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700816 err = mz_zip_extrafield_write(stream, MZ_ZIP_EXTENSION_ZIP64, field_length_zip64);
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800817 if (err == MZ_OK) {
Nathan Moinvaziric565fa82018-10-19 08:48:33 -0700818 if (mask)
819 err = mz_stream_write_int64(stream, 0);
820 else
821 err = mz_stream_write_int64(stream, file_info->uncompressed_size);
822 }
Nathan Moinvazirief93d652020-12-22 17:20:04 -0800823 if (err == MZ_OK)
Nathan Moinvaziri904c4082018-10-08 23:31:21 -0700824 err = mz_stream_write_int64(stream, file_info->compressed_size);
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -0800825 if ((err == MZ_OK) && (!local) && (file_info->disk_offset >= UINT32_MAX))
Nathan Moinvaziri904c4082018-10-08 23:31:21 -0700826 err = mz_stream_write_int64(stream, file_info->disk_offset);
Nathan Moinvazirib2a148c2020-12-23 23:55:55 -0800827 if ((err == MZ_OK) && (!local) && (file_info->disk_number >= UINT16_MAX))
828 err = mz_stream_write_uint32(stream, file_info->disk_number);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700829 }
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800830 /* Write NTFS extra field */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700831 if ((err == MZ_OK) && (field_length_ntfs > 0)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700832 err = mz_zip_extrafield_write(stream, MZ_ZIP_EXTENSION_NTFS, field_length_ntfs);
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700833 if (err == MZ_OK)
834 err = mz_stream_write_uint32(stream, reserved);
835 if (err == MZ_OK)
836 err = mz_stream_write_uint16(stream, 0x01);
837 if (err == MZ_OK)
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700838 err = mz_stream_write_uint16(stream, field_length_ntfs - 8);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700839 if (err == MZ_OK) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700840 mz_zip_unix_to_ntfs_time(file_info->modified_date, &ntfs_time);
841 err = mz_stream_write_uint64(stream, ntfs_time);
842 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700843 if (err == MZ_OK) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700844 mz_zip_unix_to_ntfs_time(file_info->accessed_date, &ntfs_time);
845 err = mz_stream_write_uint64(stream, ntfs_time);
846 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700847 if (err == MZ_OK) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -0700848 mz_zip_unix_to_ntfs_time(file_info->creation_date, &ntfs_time);
849 err = mz_stream_write_uint64(stream, ntfs_time);
850 }
851 }
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700852 /* Write UNIX extra block extra field */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700853 if ((err == MZ_OK) && (field_length_unix1 > 0)) {
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700854 err = mz_zip_extrafield_write(stream, MZ_ZIP_EXTENSION_UNIX1, field_length_unix1);
855 if (err == MZ_OK)
Nathan Moinvaziriaeb6cdd2019-05-06 13:40:32 -0700856 err = mz_stream_write_uint32(stream, (uint32_t)file_info->accessed_date);
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700857 if (err == MZ_OK)
Nathan Moinvaziriaeb6cdd2019-05-06 13:40:32 -0700858 err = mz_stream_write_uint32(stream, (uint32_t)file_info->modified_date);
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700859 if (err == MZ_OK) /* User id */
860 err = mz_stream_write_uint16(stream, 0);
861 if (err == MZ_OK) /* Group id */
862 err = mz_stream_write_uint16(stream, 0);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700863 if (err == MZ_OK && linkname_size > 0) {
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -0700864 if (mz_stream_write(stream, file_info->linkname, linkname_size) != linkname_size)
865 err = MZ_WRITE_ERROR;
866 }
867 }
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -0800868#ifdef HAVE_WZAES
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800869 /* Write AES extra field */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700870 if ((err == MZ_OK) && (!skip_aes) && (file_info->flag & MZ_ZIP_FLAG_ENCRYPTED) && (file_info->aes_version)) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -0700871 err = mz_zip_extrafield_write(stream, MZ_ZIP_EXTENSION_AES, field_length_aes);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700872 if (err == MZ_OK)
873 err = mz_stream_write_uint16(stream, file_info->aes_version);
874 if (err == MZ_OK)
875 err = mz_stream_write_uint8(stream, 'A');
876 if (err == MZ_OK)
877 err = mz_stream_write_uint8(stream, 'E');
878 if (err == MZ_OK)
879 err = mz_stream_write_uint8(stream, file_info->aes_encryption_mode);
880 if (err == MZ_OK)
881 err = mz_stream_write_uint16(stream, file_info->compression_method);
882 }
883#endif
Nathan Moinvaziri8516c572020-12-23 21:59:35 -0800884
885 if (file_info->extrafield_size > 0) {
886 err_mem = mz_stream_mem_seek(file_extra_stream, 0, MZ_SEEK_SET);
887 while (err == MZ_OK && err_mem == MZ_OK) {
888 err_mem = mz_stream_read_uint16(file_extra_stream, &field_type);
889 if (err_mem == MZ_OK)
890 err_mem = mz_stream_read_uint16(file_extra_stream, &field_length);
891 if (err_mem != MZ_OK)
892 break;
893
894 /* Prefer our zip 64, ntfs, unix1 extensions over incoming */
895 if (field_type == MZ_ZIP_EXTENSION_ZIP64 || field_type == MZ_ZIP_EXTENSION_NTFS ||
896 field_type == MZ_ZIP_EXTENSION_UNIX1) {
897 err_mem = mz_stream_seek(file_extra_stream, field_length, MZ_SEEK_CUR);
898 continue;
899 }
900
901 err = mz_stream_write_uint16(stream, field_type);
902 if (err == MZ_OK)
903 err = mz_stream_write_uint16(stream, field_length);
904 if (err == MZ_OK)
905 err = mz_stream_copy(stream, file_extra_stream, field_length);
906 }
907
908 mz_stream_mem_delete(&file_extra_stream);
909 }
910
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700911 if ((err == MZ_OK) && (!local) && (file_info->comment != NULL)) {
Nathan Moinvazirica014e62018-08-15 07:31:12 -0700912 if (mz_stream_write(stream, file_info->comment, file_info->comment_size) != file_info->comment_size)
Nathan Moinvaziri05e03ca2018-10-28 16:15:13 -0700913 err = MZ_WRITE_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -0700914 }
915
916 return err;
917}
918
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700919static 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 -0800920 int32_t err = MZ_OK;
921
922 err = mz_stream_write_uint32(stream, MZ_ZIP_MAGIC_DATADESCRIPTOR);
923 if (err == MZ_OK)
924 err = mz_stream_write_uint32(stream, crc32);
925
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800926 /* Store data descriptor as 8 bytes if zip 64 extension enabled */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700927 if (err == MZ_OK) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800928 /* Zip 64 extension is enabled when uncompressed size is > UINT32_MAX */
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800929 if (!zip64)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800930 err = mz_stream_write_uint32(stream, (uint32_t)compressed_size);
931 else
932 err = mz_stream_write_int64(stream, compressed_size);
933 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700934 if (err == MZ_OK) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -0800935 if (!zip64)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800936 err = mz_stream_write_uint32(stream, (uint32_t)uncompressed_size);
937 else
938 err = mz_stream_write_int64(stream, uncompressed_size);
939 }
940
941 return err;
942}
943
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700944static int32_t mz_zip_read_cd(void *handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800945 mz_zip *zip = (mz_zip *)handle;
946 uint64_t number_entry_cd64 = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800947 uint64_t number_entry_cd = 0;
948 int64_t eocd_pos = 0;
949 int64_t eocd_pos64 = 0;
950 int64_t value64i = 0;
951 uint16_t value16 = 0;
952 uint32_t value32 = 0;
953 uint64_t value64 = 0;
954 uint16_t comment_size = 0;
955 int32_t comment_read = 0;
956 int32_t err = MZ_OK;
957
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800958 if (zip == NULL)
959 return MZ_PARAM_ERROR;
960
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800961 /* Read and cache central directory records */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800962 err = mz_zip_search_eocd(zip->stream, &eocd_pos);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700963 if (err == MZ_OK) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800964 /* The signature, already checked */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800965 err = mz_stream_read_uint32(zip->stream, &value32);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800966 /* Number of this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800967 if (err == MZ_OK)
968 err = mz_stream_read_uint16(zip->stream, &value16);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800969 /* Number of the disk with the start of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800970 if (err == MZ_OK)
971 err = mz_stream_read_uint16(zip->stream, &value16);
972 zip->disk_number_with_cd = value16;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800973 /* Total number of entries in the central dir on this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800974 if (err == MZ_OK)
975 err = mz_stream_read_uint16(zip->stream, &value16);
976 zip->number_entry = value16;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800977 /* Total number of entries in the central dir */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800978 if (err == MZ_OK)
979 err = mz_stream_read_uint16(zip->stream, &value16);
980 number_entry_cd = value16;
981 if (number_entry_cd != zip->number_entry)
982 err = MZ_FORMAT_ERROR;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800983 /* Size of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800984 if (err == MZ_OK)
985 err = mz_stream_read_uint32(zip->stream, &value32);
986 if (err == MZ_OK)
987 zip->cd_size = value32;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800988 /* Offset of start of central directory with respect to the starting disk number */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800989 if (err == MZ_OK)
990 err = mz_stream_read_uint32(zip->stream, &value32);
991 if (err == MZ_OK)
992 zip->cd_offset = value32;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -0800993 /* Zip file global comment length */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800994 if (err == MZ_OK)
995 err = mz_stream_read_uint16(zip->stream, &comment_size);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700996 if ((err == MZ_OK) && (comment_size > 0)) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800997 zip->comment = (char *)MZ_ALLOC(comment_size + 1);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -0700998 if (zip->comment != NULL) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -0800999 comment_read = mz_stream_read(zip->stream, zip->comment, comment_size);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001000 /* Don't fail if incorrect comment length read, not critical */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001001 if (comment_read < 0)
1002 comment_read = 0;
1003 zip->comment[comment_read] = 0;
1004 }
1005 }
1006
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001007 if ((err == MZ_OK) && ((number_entry_cd == UINT16_MAX) || (zip->cd_offset == UINT32_MAX))) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001008 /* Format should be Zip64, as the central directory or file size is too large */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001009 if (mz_zip_search_zip64_eocd(zip->stream, eocd_pos, &eocd_pos64) == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001010 eocd_pos = eocd_pos64;
1011
1012 err = mz_stream_seek(zip->stream, eocd_pos, MZ_SEEK_SET);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001013 /* The signature, already checked */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001014 if (err == MZ_OK)
1015 err = mz_stream_read_uint32(zip->stream, &value32);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001016 /* Size of zip64 end of central directory record */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001017 if (err == MZ_OK)
1018 err = mz_stream_read_uint64(zip->stream, &value64);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001019 /* Version made by */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001020 if (err == MZ_OK)
1021 err = mz_stream_read_uint16(zip->stream, &zip->version_madeby);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001022 /* Version needed to extract */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001023 if (err == MZ_OK)
1024 err = mz_stream_read_uint16(zip->stream, &value16);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001025 /* Number of this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001026 if (err == MZ_OK)
1027 err = mz_stream_read_uint32(zip->stream, &value32);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001028 /* Number of the disk with the start of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001029 if (err == MZ_OK)
1030 err = mz_stream_read_uint32(zip->stream, &zip->disk_number_with_cd);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001031 /* Total number of entries in the central directory on this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001032 if (err == MZ_OK)
Gabriel Dubois3b3961e2020-03-20 15:48:40 +00001033 err = mz_stream_read_uint64(zip->stream, &zip->number_entry);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001034 /* Total number of entries in the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001035 if (err == MZ_OK)
1036 err = mz_stream_read_uint64(zip->stream, &number_entry_cd64);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001037 if (zip->number_entry != number_entry_cd64)
Gabriel Dubois3b3961e2020-03-20 15:48:40 +00001038 err = MZ_FORMAT_ERROR;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001039 /* Size of the central directory */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001040 if (err == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001041 err = mz_stream_read_int64(zip->stream, &zip->cd_size);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08001042 if (zip->cd_size < 0)
1043 err = MZ_FORMAT_ERROR;
1044 }
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001045 /* Offset of start of central directory with respect to the starting disk number */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001046 if (err == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001047 err = mz_stream_read_int64(zip->stream, &zip->cd_offset);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08001048 if (zip->cd_offset < 0)
1049 err = MZ_FORMAT_ERROR;
1050 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001051 } else if ((zip->number_entry == UINT16_MAX) || (number_entry_cd != zip->number_entry) ||
1052 (zip->cd_size == UINT16_MAX) || (zip->cd_offset == UINT32_MAX)) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001053 err = MZ_FORMAT_ERROR;
1054 }
1055 }
1056 }
1057
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001058 if (err == MZ_OK) {
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07001059 mz_zip_print("Zip - Read cd (disk %" PRId32 " entries %" PRId64 " offset %" PRId64 " size %" PRId64 ")\n",
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001060 zip->disk_number_with_cd, zip->number_entry, zip->cd_offset, zip->cd_size);
1061
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001062 /* Verify central directory signature exists at offset */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001063 err = mz_stream_seek(zip->stream, zip->cd_offset, MZ_SEEK_SET);
1064 if (err == MZ_OK)
Nathan Moinvaziri0e1fd892019-09-28 20:55:37 -07001065 err = mz_stream_read_uint32(zip->stream, &zip->cd_signature);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001066 if ((err == MZ_OK) && (zip->cd_signature != MZ_ZIP_MAGIC_CENTRALHEADER)) {
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001067 /* If cd exists in large file and no zip-64 support, error for recover */
1068 if (eocd_pos > UINT32_MAX && eocd_pos64 == 0)
1069 err = MZ_FORMAT_ERROR;
Nathan Moinvaziri0e1fd892019-09-28 20:55:37 -07001070 /* If cd not found attempt to seek backward to find it */
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001071 if (err == MZ_OK)
1072 err = mz_stream_seek(zip->stream, eocd_pos - zip->cd_size, MZ_SEEK_SET);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001073 if (err == MZ_OK)
Nathan Moinvaziri0e1fd892019-09-28 20:55:37 -07001074 err = mz_stream_read_uint32(zip->stream, &zip->cd_signature);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001075 if ((err == MZ_OK) && (zip->cd_signature == MZ_ZIP_MAGIC_CENTRALHEADER)) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001076 /* If found compensate for incorrect locations */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001077 value64i = zip->cd_offset;
1078 zip->cd_offset = eocd_pos - zip->cd_size;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001079 /* Assume disk has prepended data */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001080 zip->disk_offset_shift = zip->cd_offset - value64i;
1081 }
1082 }
1083 }
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001084
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001085 if (err == MZ_OK) {
1086 if (eocd_pos < zip->cd_offset) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001087 /* End of central dir should always come after central dir */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001088 err = MZ_FORMAT_ERROR;
Nathan Moinvazirie23732e2021-01-29 18:36:28 -08001089 } else if ((uint64_t)eocd_pos < (uint64_t)zip->cd_offset + zip->cd_size) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001090 /* Truncate size of cd if incorrect size or offset provided */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001091 zip->cd_size = eocd_pos - zip->cd_offset;
1092 }
1093 }
1094
1095 return err;
1096}
1097
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001098static int32_t mz_zip_write_cd(void *handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001099 mz_zip *zip = (mz_zip *)handle;
1100 int64_t zip64_eocd_pos_inzip = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001101 int64_t disk_number = 0;
1102 int64_t disk_size = 0;
Nathan Moinvaziridd103f42018-11-13 14:50:32 -08001103 int32_t comment_size = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001104 int32_t err = MZ_OK;
1105
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001106 if (zip == NULL)
1107 return MZ_PARAM_ERROR;
1108
1109 if (mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, &disk_number) == MZ_OK)
1110 zip->disk_number_with_cd = (uint32_t)disk_number;
1111 if (mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_SIZE, &disk_size) == MZ_OK && disk_size > 0)
1112 zip->disk_number_with_cd += 1;
1113 mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, -1);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001114 if ((zip->disk_number_with_cd > 0) && (zip->open_mode & MZ_OPEN_MODE_APPEND)) {
Nathan Moinvazirif286b832019-11-10 17:09:48 -08001115 // Overwrite existing central directory if using split disks
1116 mz_stream_seek(zip->stream, 0, MZ_SEEK_SET);
1117 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001118
1119 zip->cd_offset = mz_stream_tell(zip->stream);
1120 mz_stream_seek(zip->cd_mem_stream, 0, MZ_SEEK_END);
1121 zip->cd_size = (uint32_t)mz_stream_tell(zip->cd_mem_stream);
1122 mz_stream_seek(zip->cd_mem_stream, 0, MZ_SEEK_SET);
1123
1124 err = mz_stream_copy(zip->stream, zip->cd_mem_stream, (int32_t)zip->cd_size);
1125
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07001126 mz_zip_print("Zip - Write cd (disk %" PRId32 " entries %" PRId64 " offset %" PRId64 " size %" PRId64 ")\n",
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001127 zip->disk_number_with_cd, zip->number_entry, zip->cd_offset, zip->cd_size);
1128
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001129 if (zip->cd_size == 0 && zip->number_entry > 0) {
Nathan Moinvaziri64faa252019-04-21 21:38:48 -07001130 // Zip does not contain central directory, open with recovery option
1131 return MZ_FORMAT_ERROR;
1132 }
1133
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001134 /* Write the ZIP64 central directory header */
Nathan Moinvaziri2631b7b2021-11-12 17:53:23 -08001135 if (zip->cd_offset >= UINT32_MAX || zip->number_entry >= UINT16_MAX) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001136 zip64_eocd_pos_inzip = mz_stream_tell(zip->stream);
1137
1138 err = mz_stream_write_uint32(zip->stream, MZ_ZIP_MAGIC_ENDHEADER64);
1139
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001140 /* Size of this 'zip64 end of central directory' */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001141 if (err == MZ_OK)
1142 err = mz_stream_write_uint64(zip->stream, (uint64_t)44);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001143 /* Version made by */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001144 if (err == MZ_OK)
1145 err = mz_stream_write_uint16(zip->stream, zip->version_madeby);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001146 /* Version needed */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001147 if (err == MZ_OK)
1148 err = mz_stream_write_uint16(zip->stream, (uint16_t)45);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001149 /* Number of this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001150 if (err == MZ_OK)
1151 err = mz_stream_write_uint32(zip->stream, zip->disk_number_with_cd);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001152 /* Number of the disk with the start of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001153 if (err == MZ_OK)
1154 err = mz_stream_write_uint32(zip->stream, zip->disk_number_with_cd);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001155 /* Total number of entries in the central dir on this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001156 if (err == MZ_OK)
1157 err = mz_stream_write_uint64(zip->stream, zip->number_entry);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001158 /* Total number of entries in the central dir */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001159 if (err == MZ_OK)
1160 err = mz_stream_write_uint64(zip->stream, zip->number_entry);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001161 /* Size of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001162 if (err == MZ_OK)
1163 err = mz_stream_write_int64(zip->stream, zip->cd_size);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001164 /* Offset of start of central directory with respect to the starting disk number */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001165 if (err == MZ_OK)
1166 err = mz_stream_write_int64(zip->stream, zip->cd_offset);
1167 if (err == MZ_OK)
1168 err = mz_stream_write_uint32(zip->stream, MZ_ZIP_MAGIC_ENDLOCHEADER64);
1169
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001170 /* Number of the disk with the start of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001171 if (err == MZ_OK)
1172 err = mz_stream_write_uint32(zip->stream, zip->disk_number_with_cd);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001173 /* Relative offset to the end of zip64 central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001174 if (err == MZ_OK)
1175 err = mz_stream_write_int64(zip->stream, zip64_eocd_pos_inzip);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001176 /* Number of the disk with the start of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001177 if (err == MZ_OK)
1178 err = mz_stream_write_uint32(zip->stream, zip->disk_number_with_cd + 1);
1179 }
1180
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001181 /* Write the central directory header */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001182
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001183 /* Signature */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001184 if (err == MZ_OK)
1185 err = mz_stream_write_uint32(zip->stream, MZ_ZIP_MAGIC_ENDHEADER);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001186 /* Number of this disk */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001187 if (err == MZ_OK)
1188 err = mz_stream_write_uint16(zip->stream, (uint16_t)zip->disk_number_with_cd);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001189 /* Number of the disk with the start of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001190 if (err == MZ_OK)
1191 err = mz_stream_write_uint16(zip->stream, (uint16_t)zip->disk_number_with_cd);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001192 /* Total number of entries in the central dir on this disk */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001193 if (err == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001194 if (zip->number_entry >= UINT16_MAX)
1195 err = mz_stream_write_uint16(zip->stream, UINT16_MAX);
1196 else
1197 err = mz_stream_write_uint16(zip->stream, (uint16_t)zip->number_entry);
1198 }
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001199 /* Total number of entries in the central dir */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001200 if (err == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001201 if (zip->number_entry >= UINT16_MAX)
1202 err = mz_stream_write_uint16(zip->stream, UINT16_MAX);
1203 else
1204 err = mz_stream_write_uint16(zip->stream, (uint16_t)zip->number_entry);
1205 }
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001206 /* Size of the central directory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001207 if (err == MZ_OK)
1208 err = mz_stream_write_uint32(zip->stream, (uint32_t)zip->cd_size);
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001209 /* Offset of start of central directory with respect to the starting disk number */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001210 if (err == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001211 if (zip->cd_offset >= UINT32_MAX)
1212 err = mz_stream_write_uint32(zip->stream, UINT32_MAX);
1213 else
1214 err = mz_stream_write_uint32(zip->stream, (uint32_t)zip->cd_offset);
1215 }
1216
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001217 /* Write global comment */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001218 if (zip->comment != NULL) {
Nathan Moinvaziridd103f42018-11-13 14:50:32 -08001219 comment_size = (int32_t)strlen(zip->comment);
1220 if (comment_size > UINT16_MAX)
1221 comment_size = UINT16_MAX;
1222 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001223 if (err == MZ_OK)
Nathan Moinvaziridd103f42018-11-13 14:50:32 -08001224 err = mz_stream_write_uint16(zip->stream, (uint16_t)comment_size);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001225 if (err == MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001226 if (mz_stream_write(zip->stream, zip->comment, comment_size) != comment_size)
1227 err = MZ_READ_ERROR;
1228 }
1229 return err;
1230}
1231
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001232static int32_t mz_zip_recover_cd(void *handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001233 mz_zip *zip = (mz_zip *)handle;
1234 mz_zip_file local_file_info;
1235 void *local_file_info_stream = NULL;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001236 void *cd_mem_stream = NULL;
1237 uint64_t number_entry = 0;
1238 int64_t descriptor_pos = 0;
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001239 int64_t next_header_pos = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001240 int64_t disk_offset = 0;
1241 int64_t disk_number = 0;
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001242 int64_t compressed_pos = 0;
1243 int64_t compressed_end_pos = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001244 int64_t compressed_size = 0;
1245 int64_t uncompressed_size = 0;
1246 uint8_t descriptor_magic[4] = MZ_ZIP_MAGIC_DATADESCRIPTORU8;
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001247 uint8_t local_header_magic[4] = MZ_ZIP_MAGIC_LOCALHEADERU8;
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001248 uint8_t central_header_magic[4] = MZ_ZIP_MAGIC_CENTRALHEADERU8;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001249 uint32_t crc32 = 0;
1250 int32_t disk_number_with_cd = 0;
1251 int32_t err = MZ_OK;
1252 uint8_t zip64 = 0;
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001253 uint8_t eof = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001254
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001255 mz_zip_print("Zip - Recover - Start\n");
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001256
1257 mz_zip_get_cd_mem_stream(handle, &cd_mem_stream);
1258
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001259 /* Determine if we are on a split disk or not */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001260 mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, 0);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001261 if (mz_stream_tell(zip->stream) < 0) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001262 mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, -1);
1263 mz_stream_seek(zip->stream, 0, MZ_SEEK_SET);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001264 } else
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001265 disk_number_with_cd = 1;
1266
1267 if (mz_stream_is_open(cd_mem_stream) != MZ_OK)
1268 err = mz_stream_mem_open(cd_mem_stream, NULL, MZ_OPEN_MODE_CREATE);
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001269
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001270 mz_stream_mem_create(&local_file_info_stream);
1271 mz_stream_mem_open(local_file_info_stream, NULL, MZ_OPEN_MODE_CREATE);
Nathan Moinvaziri57cbcd22019-10-21 15:37:58 -07001272
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001273 if (err == MZ_OK) {
Nathan Moinvaziri57cbcd22019-10-21 15:37:58 -07001274 err = mz_stream_find(zip->stream, (const void *)local_header_magic, sizeof(local_header_magic),
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001275 INT64_MAX, &next_header_pos);
Nathan Moinvaziri57cbcd22019-10-21 15:37:58 -07001276 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001277
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001278 while (err == MZ_OK && !eof) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001279 /* Get current offset and disk number for central dir record */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001280 disk_offset = mz_stream_tell(zip->stream);
1281 mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, &disk_number);
1282
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001283 /* Read local headers */
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001284 memset(&local_file_info, 0, sizeof(local_file_info));
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001285 err = mz_zip_entry_read_header(zip->stream, 1, &local_file_info, local_file_info_stream);
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001286 if (err != MZ_OK)
1287 break;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001288
1289 local_file_info.disk_offset = disk_offset;
1290 if (disk_number < 0)
1291 disk_number = 0;
1292 local_file_info.disk_number = (uint32_t)disk_number;
Nathan Moinvaziri370ab032019-11-15 13:24:17 -08001293
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001294 compressed_pos = mz_stream_tell(zip->stream);
Nathan Moinvaziri370ab032019-11-15 13:24:17 -08001295
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001296 if ((err == MZ_OK) && (local_file_info.compressed_size > 0)) {
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001297 mz_stream_seek(zip->stream, local_file_info.compressed_size, MZ_SEEK_CUR);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001298 }
1299
Nathan Moinvaziri958226c2020-07-30 11:06:25 -07001300 for (;;) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001301 /* Search for the next local header */
1302 err = mz_stream_find(zip->stream, (const void *)local_header_magic, sizeof(local_header_magic),
1303 INT64_MAX, &next_header_pos);
Nathan Moinvaziri370ab032019-11-15 13:24:17 -08001304
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001305 if (err == MZ_EXIST_ERROR) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001306 mz_stream_seek(zip->stream, compressed_pos, MZ_SEEK_SET);
1307
1308 /* Search for central dir if no local header found */
1309 err = mz_stream_find(zip->stream, (const void *)central_header_magic, sizeof(central_header_magic),
1310 INT64_MAX, &next_header_pos);
1311
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001312 if (err == MZ_EXIST_ERROR) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001313 /* Get end of stream if no central header found */
1314 mz_stream_seek(zip->stream, 0, MZ_SEEK_END);
1315 next_header_pos = mz_stream_tell(zip->stream);
1316 }
1317
1318 eof = 1;
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001319 }
1320
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001321 if (local_file_info.flag & MZ_ZIP_FLAG_DATA_DESCRIPTOR || local_file_info.compressed_size == 0) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001322 /* Search backwards for the descriptor, seeking too far back will be incorrect if compressed size is small */
1323 err = mz_stream_find_reverse(zip->stream, (const void *)descriptor_magic, sizeof(descriptor_magic),
1324 MZ_ZIP_SIZE_MAX_DATA_DESCRIPTOR, &descriptor_pos);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001325 if (err == MZ_OK) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001326 if (mz_zip_extrafield_contains(local_file_info.extrafield,
1327 local_file_info.extrafield_size, MZ_ZIP_EXTENSION_ZIP64, NULL) == MZ_OK)
1328 zip64 = 1;
1329
1330 err = mz_zip_entry_read_descriptor(zip->stream, zip64, &crc32,
1331 &compressed_size, &uncompressed_size);
1332
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001333 if (err == MZ_OK) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001334 if (local_file_info.crc == 0)
1335 local_file_info.crc = crc32;
1336 if (local_file_info.compressed_size == 0)
1337 local_file_info.compressed_size = compressed_size;
1338 if (local_file_info.uncompressed_size == 0)
1339 local_file_info.uncompressed_size = uncompressed_size;
1340 }
1341
1342 compressed_end_pos = descriptor_pos;
Nathan Moinvaziri958226c2020-07-30 11:06:25 -07001343 } else if (eof) {
1344 compressed_end_pos = next_header_pos;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001345 } else if (local_file_info.flag & MZ_ZIP_FLAG_DATA_DESCRIPTOR) {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001346 /* Wrong local file entry found, keep searching */
1347 next_header_pos += 1;
1348 mz_stream_seek(zip->stream, next_header_pos, MZ_SEEK_SET);
1349 continue;
1350 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001351 } else {
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001352 compressed_end_pos = next_header_pos;
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001353 }
1354
Nathan Moinvaziri4b5bd872019-11-11 17:04:05 -08001355 break;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001356 }
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001357
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001358 compressed_size = compressed_end_pos - compressed_pos;
1359
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001360 if (compressed_size > UINT32_MAX) {
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001361 /* Update sizes if 4GB file is written with no ZIP64 support */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001362 if (local_file_info.uncompressed_size < UINT32_MAX) {
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001363 local_file_info.compressed_size = compressed_size;
1364 local_file_info.uncompressed_size = 0;
1365 }
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001366 }
1367
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001368 mz_zip_print("Zip - Recover - Entry %s (csize %" PRId64 " usize %" PRId64 " flags 0x%" PRIx16 ")\n",
1369 local_file_info.filename, local_file_info.compressed_size, local_file_info.uncompressed_size,
1370 local_file_info.flag);
1371
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001372 /* Rewrite central dir with local headers and offsets */
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001373 err = mz_zip_entry_write_header(cd_mem_stream, 0, &local_file_info);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001374 if (err == MZ_OK)
Nathan Moinvaziri370ab032019-11-15 13:24:17 -08001375 number_entry += 1;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001376
Nathan Moinvaziri545d7292019-10-21 15:32:18 -07001377 err = mz_stream_seek(zip->stream, next_header_pos, MZ_SEEK_SET);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001378 }
1379
1380 mz_stream_mem_delete(&local_file_info_stream);
1381
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001382 mz_zip_print("Zip - Recover - Complete (cddisk %" PRId32 " entries %" PRId64 ")\n",
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001383 disk_number_with_cd, number_entry);
1384
1385 if (number_entry == 0)
1386 return err;
Nathan Moinvaziri370ab032019-11-15 13:24:17 -08001387
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001388 /* Set new upper seek boundary for central dir mem stream */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001389 disk_offset = mz_stream_tell(cd_mem_stream);
1390 mz_stream_mem_set_buffer_limit(cd_mem_stream, (int32_t)disk_offset);
1391
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001392 /* Set new central directory info */
Nathan Moinvaziri4d089732018-11-19 20:24:26 -08001393 mz_zip_set_cd_stream(handle, 0, cd_mem_stream);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001394 mz_zip_set_number_entry(handle, number_entry);
1395 mz_zip_set_disk_number_with_cd(handle, disk_number_with_cd);
1396
1397 return MZ_OK;
1398}
1399
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001400void *mz_zip_create(void **handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001401 mz_zip *zip = NULL;
1402
1403 zip = (mz_zip *)MZ_ALLOC(sizeof(mz_zip));
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001404 if (zip != NULL) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001405 memset(zip, 0, sizeof(mz_zip));
Nathan Moinvaziri41fcda92020-01-04 16:55:21 -08001406 zip->data_descriptor = 1;
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08001407 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001408 if (handle != NULL)
1409 *handle = zip;
1410
1411 return zip;
1412}
1413
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001414void mz_zip_delete(void **handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001415 mz_zip *zip = NULL;
1416 if (handle == NULL)
1417 return;
1418 zip = (mz_zip *)*handle;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001419 if (zip != NULL) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001420 MZ_FREE(zip);
1421 }
1422 *handle = NULL;
1423}
1424
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001425int32_t mz_zip_open(void *handle, void *stream, int32_t mode) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001426 mz_zip *zip = (mz_zip *)handle;
1427 int32_t err = MZ_OK;
1428
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001429 if (zip == NULL)
1430 return MZ_PARAM_ERROR;
1431
1432 mz_zip_print("Zip - Open\n");
1433
1434 zip->stream = stream;
1435
1436 mz_stream_mem_create(&zip->cd_mem_stream);
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001437
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001438 if (mode & MZ_OPEN_MODE_WRITE) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001439 mz_stream_mem_open(zip->cd_mem_stream, NULL, MZ_OPEN_MODE_CREATE);
1440 zip->cd_stream = zip->cd_mem_stream;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001441 } else {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001442 zip->cd_stream = stream;
1443 }
1444
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001445 if ((mode & MZ_OPEN_MODE_READ) || (mode & MZ_OPEN_MODE_APPEND)) {
1446 if ((mode & MZ_OPEN_MODE_CREATE) == 0) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001447 err = mz_zip_read_cd(zip);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001448 if (err != MZ_OK) {
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001449 mz_zip_print("Zip - Error detected reading cd (%" PRId32 ")\n", err);
Nathan Moinvaziridd103f42018-11-13 14:50:32 -08001450 if (zip->recover && mz_zip_recover_cd(zip) == MZ_OK)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001451 err = MZ_OK;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001452 }
1453 }
1454
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001455 if ((err == MZ_OK) && (mode & MZ_OPEN_MODE_APPEND)) {
1456 if (zip->cd_size > 0) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001457 /* Store central directory in memory */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001458 err = mz_stream_seek(zip->stream, zip->cd_offset, MZ_SEEK_SET);
1459 if (err == MZ_OK)
1460 err = mz_stream_copy(zip->cd_mem_stream, zip->stream, (int32_t)zip->cd_size);
1461 if (err == MZ_OK)
1462 err = mz_stream_seek(zip->stream, zip->cd_offset, MZ_SEEK_SET);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001463 } else {
1464 if (zip->cd_signature == MZ_ZIP_MAGIC_ENDHEADER) {
Nathan Moinvaziri0e1fd892019-09-28 20:55:37 -07001465 /* If tiny zip then overwrite end header */
1466 err = mz_stream_seek(zip->stream, zip->cd_offset, MZ_SEEK_SET);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001467 } else {
Nathan Moinvaziri0e1fd892019-09-28 20:55:37 -07001468 /* If no central directory, append new zip to end of file */
1469 err = mz_stream_seek(zip->stream, 0, MZ_SEEK_END);
1470 }
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001471 }
Nathan Moinvaziri2ce1fe22018-11-23 11:53:11 -08001472
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001473 if (zip->disk_number_with_cd > 0) {
Nathan Moinvaziri2ce1fe22018-11-23 11:53:11 -08001474 /* Move to last disk to begin appending */
1475 mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, zip->disk_number_with_cd - 1);
1476 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001477 } else {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001478 zip->cd_start_pos = zip->cd_offset;
1479 }
1480 }
1481
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001482 if (err != MZ_OK) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001483 mz_zip_close(zip);
1484 return err;
1485 }
1486
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001487 /* Memory streams used to store variable length file info data */
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001488 mz_stream_mem_create(&zip->file_info_stream);
1489 mz_stream_mem_open(zip->file_info_stream, NULL, MZ_OPEN_MODE_CREATE);
1490
1491 mz_stream_mem_create(&zip->local_file_info_stream);
1492 mz_stream_mem_open(zip->local_file_info_stream, NULL, MZ_OPEN_MODE_CREATE);
1493
1494 zip->open_mode = mode;
1495
1496 return err;
1497}
1498
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001499int32_t mz_zip_close(void *handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001500 mz_zip *zip = (mz_zip *)handle;
1501 int32_t err = MZ_OK;
1502
1503 if (zip == NULL)
1504 return MZ_PARAM_ERROR;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001505
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001506 mz_zip_print("Zip - Close\n");
1507
1508 if (mz_zip_entry_is_open(handle) == MZ_OK)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001509 err = mz_zip_entry_close(handle);
Nathan Moinvazirieffc1422018-11-08 15:24:17 -08001510
1511 if ((err == MZ_OK) && (zip->open_mode & MZ_OPEN_MODE_WRITE))
1512 err = mz_zip_write_cd(handle);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001513
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001514 if (zip->cd_mem_stream != NULL) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001515 mz_stream_close(zip->cd_mem_stream);
1516 mz_stream_delete(&zip->cd_mem_stream);
1517 }
1518
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001519 if (zip->file_info_stream != NULL) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001520 mz_stream_mem_close(zip->file_info_stream);
1521 mz_stream_mem_delete(&zip->file_info_stream);
1522 }
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001523 if (zip->local_file_info_stream != NULL) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001524 mz_stream_mem_close(zip->local_file_info_stream);
1525 mz_stream_mem_delete(&zip->local_file_info_stream);
1526 }
1527
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001528 if (zip->comment) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001529 MZ_FREE(zip->comment);
1530 zip->comment = NULL;
1531 }
1532
1533 zip->stream = NULL;
1534 zip->cd_stream = NULL;
1535
1536 return err;
1537}
1538
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001539int32_t mz_zip_get_comment(void *handle, const char **comment) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001540 mz_zip *zip = (mz_zip *)handle;
1541 if (zip == NULL || comment == NULL)
1542 return MZ_PARAM_ERROR;
1543 if (zip->comment == NULL)
1544 return MZ_EXIST_ERROR;
1545 *comment = zip->comment;
1546 return MZ_OK;
1547}
1548
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001549int32_t mz_zip_set_comment(void *handle, const char *comment) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001550 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziridd103f42018-11-13 14:50:32 -08001551 int32_t comment_size = 0;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001552 if (zip == NULL || comment == NULL)
1553 return MZ_PARAM_ERROR;
1554 if (zip->comment != NULL)
1555 MZ_FREE(zip->comment);
Nathan Moinvaziri2df1ecb2018-12-01 08:59:21 -08001556 comment_size = (int32_t)strlen(comment);
1557 if (comment_size > UINT16_MAX)
1558 return MZ_PARAM_ERROR;
Nathan Moinvaziria6ecab52019-01-22 13:14:01 -08001559 zip->comment = (char *)MZ_ALLOC(comment_size+1);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001560 if (zip->comment == NULL)
1561 return MZ_MEM_ERROR;
Nathan Moinvaziria6ecab52019-01-22 13:14:01 -08001562 memset(zip->comment, 0, comment_size+1);
Nathan Moinvaziri2df1ecb2018-12-01 08:59:21 -08001563 strncpy(zip->comment, comment, comment_size);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001564 return MZ_OK;
1565}
1566
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001567int32_t mz_zip_get_version_madeby(void *handle, uint16_t *version_madeby) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001568 mz_zip *zip = (mz_zip *)handle;
1569 if (zip == NULL || version_madeby == NULL)
1570 return MZ_PARAM_ERROR;
1571 *version_madeby = zip->version_madeby;
1572 return MZ_OK;
1573}
1574
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001575int32_t mz_zip_set_version_madeby(void *handle, uint16_t version_madeby) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001576 mz_zip *zip = (mz_zip *)handle;
1577 if (zip == NULL)
1578 return MZ_PARAM_ERROR;
1579 zip->version_madeby = version_madeby;
1580 return MZ_OK;
1581}
1582
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001583int32_t mz_zip_set_recover(void *handle, uint8_t recover) {
Nathan Moinvaziridd103f42018-11-13 14:50:32 -08001584 mz_zip *zip = (mz_zip *)handle;
1585 if (zip == NULL)
1586 return MZ_PARAM_ERROR;
1587 zip->recover = recover;
1588 return MZ_OK;
1589}
1590
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001591int32_t mz_zip_set_data_descriptor(void *handle, uint8_t data_descriptor) {
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08001592 mz_zip *zip = (mz_zip *)handle;
1593 if (zip == NULL)
1594 return MZ_PARAM_ERROR;
1595 zip->data_descriptor = data_descriptor;
1596 return MZ_OK;
1597}
1598
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001599int32_t mz_zip_get_stream(void *handle, void **stream) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001600 mz_zip *zip = (mz_zip *)handle;
1601 if (zip == NULL || stream == NULL)
1602 return MZ_PARAM_ERROR;
1603 *stream = zip->stream;
1604 if (*stream == NULL)
1605 return MZ_EXIST_ERROR;
1606 return MZ_OK;
1607}
1608
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001609int32_t mz_zip_set_cd_stream(void *handle, int64_t cd_start_pos, void *cd_stream) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001610 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri4d089732018-11-19 20:24:26 -08001611 if (zip == NULL || cd_stream == NULL)
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001612 return MZ_PARAM_ERROR;
Nathan Moinvaziribe483d82019-10-19 17:30:23 -07001613 zip->cd_offset = 0;
Nathan Moinvaziri4d089732018-11-19 20:24:26 -08001614 zip->cd_stream = cd_stream;
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001615 zip->cd_start_pos = cd_start_pos;
1616 return MZ_OK;
1617}
1618
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001619int32_t mz_zip_get_cd_mem_stream(void *handle, void **cd_mem_stream) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001620 mz_zip *zip = (mz_zip *)handle;
1621 if (zip == NULL || cd_mem_stream == NULL)
1622 return MZ_PARAM_ERROR;
1623 *cd_mem_stream = zip->cd_mem_stream;
1624 if (*cd_mem_stream == NULL)
1625 return MZ_EXIST_ERROR;
1626 return MZ_OK;
1627}
1628
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001629int32_t mz_zip_set_number_entry(void *handle, uint64_t number_entry) {
Nathan Moinvaziri163352f2020-02-08 21:28:23 -08001630 mz_zip *zip = (mz_zip *)handle;
1631 if (zip == NULL)
1632 return MZ_PARAM_ERROR;
1633 zip->number_entry = number_entry;
1634 return MZ_OK;
1635}
1636
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001637int32_t mz_zip_get_number_entry(void *handle, uint64_t *number_entry) {
Nathan Moinvaziri163352f2020-02-08 21:28:23 -08001638 mz_zip *zip = (mz_zip *)handle;
1639 if (zip == NULL || number_entry == NULL)
1640 return MZ_PARAM_ERROR;
1641 *number_entry = zip->number_entry;
1642 return MZ_OK;
1643}
1644
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001645int32_t mz_zip_set_disk_number_with_cd(void *handle, uint32_t disk_number_with_cd) {
Nathan Moinvaziri163352f2020-02-08 21:28:23 -08001646 mz_zip *zip = (mz_zip *)handle;
1647 if (zip == NULL)
1648 return MZ_PARAM_ERROR;
1649 zip->disk_number_with_cd = disk_number_with_cd;
1650 return MZ_OK;
1651}
1652
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001653int32_t mz_zip_get_disk_number_with_cd(void *handle, uint32_t *disk_number_with_cd) {
Nathan Moinvaziri163352f2020-02-08 21:28:23 -08001654 mz_zip *zip = (mz_zip *)handle;
1655 if (zip == NULL || disk_number_with_cd == NULL)
1656 return MZ_PARAM_ERROR;
1657 *disk_number_with_cd = zip->disk_number_with_cd;
1658 return MZ_OK;
1659}
1660
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001661static int32_t mz_zip_entry_close_int(void *handle) {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001662 mz_zip *zip = (mz_zip *)handle;
1663
1664 if (zip->crypt_stream != NULL)
1665 mz_stream_delete(&zip->crypt_stream);
1666 zip->crypt_stream = NULL;
1667 if (zip->compress_stream != NULL)
1668 mz_stream_delete(&zip->compress_stream);
1669 zip->compress_stream = NULL;
1670
1671 zip->entry_opened = 0;
1672
1673 return MZ_OK;
1674}
1675
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001676static 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 -07001677 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001678 int64_t max_total_in = 0;
Nathan Moinvazirifa8105b2018-07-08 19:05:00 -07001679 int64_t header_size = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001680 int64_t footer_size = 0;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001681 int32_t err = MZ_OK;
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001682 uint8_t use_crypt = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001683
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001684 if (zip == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001685 return MZ_PARAM_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001686
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001687 switch (zip->file_info.compression_method) {
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001688 case MZ_COMPRESS_METHOD_STORE:
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001689 case MZ_COMPRESS_METHOD_DEFLATE:
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001690#ifdef HAVE_BZIP2
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001691 case MZ_COMPRESS_METHOD_BZIP2:
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001692#endif
Nathan Moinvaziri566dfe02018-10-26 00:36:30 -07001693#ifdef HAVE_LZMA
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001694 case MZ_COMPRESS_METHOD_LZMA:
Nathan Moinvaziri1a7e7e42020-10-25 14:50:53 -07001695#endif
1696#if defined(HAVE_LZMA) || defined(HAVE_LIBCOMP)
Nathan Moinvaziric4668ed2020-10-25 11:56:19 -07001697 case MZ_COMPRESS_METHOD_XZ:
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001698#endif
Nathan Moinvaziri1f4758b2020-06-14 21:52:55 -07001699#ifdef HAVE_ZSTD
1700 case MZ_COMPRESS_METHOD_ZSTD:
Nathan Moinvaziri1f4758b2020-06-14 21:52:55 -07001701#endif
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001702 err = MZ_OK;
1703 break;
1704 default:
Nathan Moinvaziri829ffb52018-08-14 14:00:16 -07001705 return MZ_SUPPORT_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001706 }
1707
Nathan Moinvazirif9e34482018-11-27 09:45:09 -08001708#ifndef HAVE_WZAES
1709 if (zip->file_info.aes_version)
1710 return MZ_SUPPORT_ERROR;
1711#endif
1712
Nathan Moinvaziria64c9192018-08-08 17:47:12 -07001713 zip->entry_raw = raw;
1714
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001715 if ((zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED) && (password != NULL)) {
1716 if (zip->open_mode & MZ_OPEN_MODE_WRITE) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001717 /* Encrypt only when we are not trying to write raw and password is supplied. */
Nathan Moinvaziria64c9192018-08-08 17:47:12 -07001718 if (!zip->entry_raw)
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001719 use_crypt = 1;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001720 } else if (zip->open_mode & MZ_OPEN_MODE_READ) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001721 /* Decrypt only when password is supplied. Don't error when password */
1722 /* is not supplied as we may want to read the raw encrypted data. */
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001723 use_crypt = 1;
1724 }
1725 }
1726
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001727 if ((err == MZ_OK) && (use_crypt)) {
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -08001728#ifdef HAVE_WZAES
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001729 if (zip->file_info.aes_version) {
Nathan Moinvaziri21a31022018-10-24 09:50:16 -07001730 mz_stream_wzaes_create(&zip->crypt_stream);
1731 mz_stream_wzaes_set_password(zip->crypt_stream, password);
1732 mz_stream_wzaes_set_encryption_mode(zip->crypt_stream, zip->file_info.aes_encryption_mode);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001733 } else
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001734#endif
1735 {
Nathan Moinvaziri0e5c9dc2018-05-23 20:21:48 -07001736#ifdef HAVE_PKCRYPT
Nathan Moinvaziri9ffb08e2020-11-30 21:30:10 -08001737 uint8_t verify1 = (uint8_t)((zip->file_info.pk_verify >> 8) & 0xff);
1738 uint8_t verify2 = (uint8_t)((zip->file_info.pk_verify) & 0xff);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001739
Nathan Moinvaziri0e5c9dc2018-05-23 20:21:48 -07001740 mz_stream_pkcrypt_create(&zip->crypt_stream);
1741 mz_stream_pkcrypt_set_password(zip->crypt_stream, password);
1742 mz_stream_pkcrypt_set_verify(zip->crypt_stream, verify1, verify2);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001743#endif
1744 }
1745 }
1746
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001747 if (err == MZ_OK) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001748 if (zip->crypt_stream == NULL)
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001749 mz_stream_raw_create(&zip->crypt_stream);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001750
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001751 mz_stream_set_base(zip->crypt_stream, zip->stream);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001752
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001753 err = mz_stream_open(zip->crypt_stream, NULL, zip->open_mode);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001754 }
1755
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001756 if (err == MZ_OK) {
Nathan Moinvaziria64c9192018-08-08 17:47:12 -07001757 if (zip->entry_raw || zip->file_info.compression_method == MZ_COMPRESS_METHOD_STORE)
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001758 mz_stream_raw_create(&zip->compress_stream);
Nathan Moinvaziri052ceb12020-12-27 17:38:43 -08001759#ifdef HAVE_ZLIB
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001760 else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_DEFLATE)
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001761 mz_stream_zlib_create(&zip->compress_stream);
Andy Maloney3909c232018-05-22 09:02:09 -04001762#endif
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001763#ifdef HAVE_BZIP2
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001764 else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_BZIP2)
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001765 mz_stream_bzip_create(&zip->compress_stream);
1766#endif
Nathan Moinvaziri1a7e7e42020-10-25 14:50:53 -07001767#ifdef HAVE_LIBCOMP
Nathan Moinvaziri052ceb12020-12-27 17:38:43 -08001768 else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_DEFLATE ||
1769 zip->file_info.compression_method == MZ_COMPRESS_METHOD_XZ) {
Nathan Moinvaziria7343862020-10-25 15:22:10 -07001770 mz_stream_libcomp_create(&zip->compress_stream);
Nathan Moinvaziri1a7e7e42020-10-25 14:50:53 -07001771 mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_COMPRESS_METHOD,
1772 zip->file_info.compression_method);
1773 }
1774#endif
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001775#ifdef HAVE_LZMA
Nathan Moinvaziric4668ed2020-10-25 11:56:19 -07001776 else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_LZMA ||
1777 zip->file_info.compression_method == MZ_COMPRESS_METHOD_XZ) {
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001778 mz_stream_lzma_create(&zip->compress_stream);
Nathan Moinvaziriff7ef662020-10-25 12:27:29 -07001779 mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_COMPRESS_METHOD,
1780 zip->file_info.compression_method);
1781 }
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001782#endif
Nathan Moinvaziri1f4758b2020-06-14 21:52:55 -07001783#ifdef HAVE_ZSTD
Force Charlie32ccb3e2020-06-24 10:55:18 +08001784 else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_ZSTD)
Nathan Moinvaziri1f4758b2020-06-14 21:52:55 -07001785 mz_stream_zstd_create(&zip->compress_stream);
1786#endif
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001787 else
1788 err = MZ_PARAM_ERROR;
1789 }
1790
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001791 if (err == MZ_OK) {
1792 if (zip->open_mode & MZ_OPEN_MODE_WRITE) {
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001793 mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_COMPRESS_LEVEL, compress_level);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001794 } else {
Nathan Moinvaziri91261772020-06-19 21:16:16 -07001795 int32_t set_end_of_stream = 0;
1796
Nathan Moinvaziri76fb8902018-11-21 15:14:09 -08001797#ifndef HAVE_LIBCOMP
Nathan Moinvaziriff7ef662020-10-25 12:27:29 -07001798 if (zip->entry_raw ||
1799 zip->file_info.compression_method == MZ_COMPRESS_METHOD_STORE ||
1800 zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED)
Nathan Moinvaziri90d31c72018-11-20 03:17:20 -08001801#endif
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001802 {
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001803 max_total_in = zip->file_info.compressed_size;
Nathan Moinvazirifa8105b2018-07-08 19:05:00 -07001804 mz_stream_set_prop_int64(zip->crypt_stream, MZ_STREAM_PROP_TOTAL_IN_MAX, max_total_in);
1805
1806 if (mz_stream_get_prop_int64(zip->crypt_stream, MZ_STREAM_PROP_HEADER_SIZE, &header_size) == MZ_OK)
1807 max_total_in -= header_size;
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001808 if (mz_stream_get_prop_int64(zip->crypt_stream, MZ_STREAM_PROP_FOOTER_SIZE, &footer_size) == MZ_OK)
1809 max_total_in -= footer_size;
Nathan Moinvazirifa8105b2018-07-08 19:05:00 -07001810
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001811 mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_IN_MAX, max_total_in);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001812 }
Nathan Moinvaziri91261772020-06-19 21:16:16 -07001813
1814 switch (zip->file_info.compression_method) {
1815 case MZ_COMPRESS_METHOD_LZMA:
Nathan Moinvaziri5cd28792020-10-25 12:50:12 -07001816 case MZ_COMPRESS_METHOD_XZ:
Nathan Moinvaziri91261772020-06-19 21:16:16 -07001817 set_end_of_stream = (zip->file_info.flag & MZ_ZIP_FLAG_LZMA_EOS_MARKER);
1818 break;
1819 case MZ_COMPRESS_METHOD_ZSTD:
Nathan Moinvaziri91261772020-06-19 21:16:16 -07001820 set_end_of_stream = 1;
1821 break;
juanii55bcdaf2018-02-11 20:55:57 -03001822 }
Nathan Moinvaziri91261772020-06-19 21:16:16 -07001823
1824 if (set_end_of_stream) {
Force.Charlie7cecfa62020-06-15 07:43:25 -07001825 mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_IN_MAX, zip->file_info.compressed_size);
1826 mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_OUT_MAX, zip->file_info.uncompressed_size);
1827 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001828 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001829
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001830 mz_stream_set_base(zip->compress_stream, zip->crypt_stream);
1831
1832 err = mz_stream_open(zip->compress_stream, NULL, zip->open_mode);
1833 }
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07001834
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001835 if (err == MZ_OK) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001836 zip->entry_opened = 1;
Nathan Moinvazirie63d2312018-11-03 19:45:41 -07001837 zip->entry_crc32 = 0;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001838 } else {
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001839 mz_zip_entry_close_int(handle);
1840 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001841
1842 return err;
1843}
1844
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001845int32_t mz_zip_entry_is_open(void *handle) {
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07001846 mz_zip *zip = (mz_zip *)handle;
1847 if (zip == NULL)
1848 return MZ_PARAM_ERROR;
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07001849 if (zip->entry_opened == 0)
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07001850 return MZ_EXIST_ERROR;
1851 return MZ_OK;
1852}
1853
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001854int32_t mz_zip_entry_read_open(void *handle, uint8_t raw, const char *password) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001855 mz_zip *zip = (mz_zip *)handle;
1856 int32_t err = MZ_OK;
Nathan Moinvaziriaacf8732018-11-06 09:38:46 -08001857 int32_t err_shift = MZ_OK;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001858
Nathan Moinvaziri40427632018-07-22 11:12:40 -07001859#if defined(MZ_ZIP_NO_ENCRYPTION)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001860 if (password != NULL)
Nathan Moinvazirif9e34482018-11-27 09:45:09 -08001861 return MZ_SUPPORT_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001862#endif
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001863 if (zip == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001864 return MZ_PARAM_ERROR;
Nathan Moinvaziri79ac3382017-10-23 17:22:36 -07001865 if ((zip->open_mode & MZ_OPEN_MODE_READ) == 0)
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001866 return MZ_PARAM_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001867 if (zip->entry_scanned == 0)
1868 return MZ_PARAM_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001869
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07001870 mz_zip_print("Zip - Entry - Read open (raw %" PRId32 ")\n", raw);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001871
Nathan Moinvaziri0030fa22020-12-21 19:31:07 -08001872 err = mz_zip_entry_seek_local_header(handle);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001873 if (err == MZ_OK)
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001874 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 -07001875
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001876 if (err == MZ_FORMAT_ERROR && zip->disk_offset_shift > 0) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001877 /* Perhaps we didn't compensated correctly for incorrect cd offset */
Nathan Moinvaziriaacf8732018-11-06 09:38:46 -08001878 err_shift = mz_stream_seek(zip->stream, zip->file_info.disk_offset, MZ_SEEK_SET);
1879 if (err_shift == MZ_OK)
1880 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 -07001881 if (err_shift == MZ_OK) {
Nathan Moinvaziriaacf8732018-11-06 09:38:46 -08001882 zip->disk_offset_shift = 0;
1883 err = err_shift;
1884 }
1885 }
1886
Nathan Moinvaziria6d1f662018-07-22 10:35:49 -07001887#ifdef MZ_ZIP_NO_DECOMPRESSION
Nathan Moinvaziri2eb29072018-11-23 10:32:21 -08001888 if (!raw && zip->file_info.compression_method != MZ_COMPRESS_METHOD_STORE)
Nathan Moinvaziri0a9282d2018-06-19 11:59:07 -07001889 err = MZ_SUPPORT_ERROR;
Viktor Szakatse7215072018-09-04 15:06:53 +00001890#endif
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001891 if (err == MZ_OK)
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001892 err = mz_zip_entry_open_int(handle, raw, 0, password);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001893
1894 return err;
1895}
1896
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001897int32_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 -07001898 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri6b64b922018-11-23 11:41:35 -08001899 int64_t filename_pos = -1;
Nathan Moinvazirif9e34482018-11-27 09:45:09 -08001900 int64_t extrafield_pos = 0;
1901 int64_t comment_pos = 0;
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -07001902 int64_t linkname_pos = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001903 int64_t disk_number = 0;
Nathan Moinvazirifa620e42018-10-20 10:37:10 -07001904 uint8_t is_dir = 0;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001905 int32_t err = MZ_OK;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001906
Nathan Moinvaziri40427632018-07-22 11:12:40 -07001907#if defined(MZ_ZIP_NO_ENCRYPTION)
tz-lomb1b25802017-11-10 15:03:02 +03001908 if (password != NULL)
Nathan Moinvazirif9e34482018-11-27 09:45:09 -08001909 return MZ_SUPPORT_ERROR;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001910#endif
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001911 if (zip == NULL || file_info == NULL || file_info->filename == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001912 return MZ_PARAM_ERROR;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001913
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001914 if (mz_zip_entry_is_open(handle) == MZ_OK) {
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001915 err = mz_zip_entry_close(handle);
1916 if (err != MZ_OK)
1917 return err;
1918 }
1919
1920 memcpy(&zip->file_info, file_info, sizeof(mz_zip_file));
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001921
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07001922 mz_zip_print("Zip - Entry - Write open - %s (level %" PRId16 " raw %" PRId8 ")\n",
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001923 zip->file_info.filename, compress_level, raw);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001924
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08001925 mz_stream_seek(zip->file_info_stream, 0, MZ_SEEK_SET);
Nathan Moinvazirif5cc4c02018-08-14 17:46:36 -07001926 mz_stream_write(zip->file_info_stream, file_info, sizeof(mz_zip_file));
1927
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08001928 /* Copy filename, extrafield, and comment internally */
Nathan Moinvaziri6b64b922018-11-23 11:41:35 -08001929 filename_pos = mz_stream_tell(zip->file_info_stream);
Nathan Moinvazirif5cc4c02018-08-14 17:46:36 -07001930 if (file_info->filename != NULL)
Nathan Moinvaziri6b64b922018-11-23 11:41:35 -08001931 mz_stream_write(zip->file_info_stream, file_info->filename, (int32_t)strlen(file_info->filename));
1932 mz_stream_write_uint8(zip->file_info_stream, 0);
1933
1934 extrafield_pos = mz_stream_tell(zip->file_info_stream);
Nathan Moinvazirif5cc4c02018-08-14 17:46:36 -07001935 if (file_info->extrafield != NULL)
Nathan Moinvazirica014e62018-08-15 07:31:12 -07001936 mz_stream_write(zip->file_info_stream, file_info->extrafield, file_info->extrafield_size);
Nathan Moinvaziri6b64b922018-11-23 11:41:35 -08001937 mz_stream_write_uint8(zip->file_info_stream, 0);
1938
1939 comment_pos = mz_stream_tell(zip->file_info_stream);
Nathan Moinvazirif5cc4c02018-08-14 17:46:36 -07001940 if (file_info->comment != NULL)
Nathan Moinvaziri6b64b922018-11-23 11:41:35 -08001941 mz_stream_write(zip->file_info_stream, file_info->comment, file_info->comment_size);
1942 mz_stream_write_uint8(zip->file_info_stream, 0);
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07001943
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -07001944 linkname_pos = mz_stream_tell(zip->file_info_stream);
1945 if (file_info->linkname != NULL)
1946 mz_stream_write(zip->file_info_stream, file_info->linkname, (int32_t)strlen(file_info->linkname));
1947 mz_stream_write_uint8(zip->file_info_stream, 0);
Nathan Moinvaziri6b64b922018-11-23 11:41:35 -08001948
Nathan Moinvazirif9e34482018-11-27 09:45:09 -08001949 mz_stream_mem_get_buffer_at(zip->file_info_stream, filename_pos, (const void **)&zip->file_info.filename);
1950 mz_stream_mem_get_buffer_at(zip->file_info_stream, extrafield_pos, (const void **)&zip->file_info.extrafield);
1951 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 -07001952 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 -07001953
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001954 if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_DEFLATE) {
Nathan Moinvaziri91a76f62017-11-10 07:39:06 -08001955 if ((compress_level == 8) || (compress_level == 9))
1956 zip->file_info.flag |= MZ_ZIP_FLAG_DEFLATE_MAX;
1957 if (compress_level == 2)
1958 zip->file_info.flag |= MZ_ZIP_FLAG_DEFLATE_FAST;
1959 if (compress_level == 1)
1960 zip->file_info.flag |= MZ_ZIP_FLAG_DEFLATE_SUPER_FAST;
1961 }
Nathan Moinvaziri1a7e7e42020-10-25 14:50:53 -07001962#if defined(HAVE_LZMA) || defined(HAVE_LIBCOMP)
Nathan Moinvaziri5cd28792020-10-25 12:50:12 -07001963 else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_LZMA ||
1964 zip->file_info.compression_method == MZ_COMPRESS_METHOD_XZ)
Nathan Moinvaziri91a76f62017-11-10 07:39:06 -08001965 zip->file_info.flag |= MZ_ZIP_FLAG_LZMA_EOS_MARKER;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001966#endif
Nathan Moinvaziri91a76f62017-11-10 07:39:06 -08001967
Nathan Moinvazirifa620e42018-10-20 10:37:10 -07001968 if (mz_zip_attrib_is_dir(zip->file_info.external_fa, zip->file_info.version_madeby) == MZ_OK)
1969 is_dir = 1;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001970
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07001971 if (!is_dir) {
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08001972 if (zip->data_descriptor)
1973 zip->file_info.flag |= MZ_ZIP_FLAG_DATA_DESCRIPTOR;
Nathan Moinvazirifa620e42018-10-20 10:37:10 -07001974 if (password != NULL)
1975 zip->file_info.flag |= MZ_ZIP_FLAG_ENCRYPTED;
1976 }
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001977
juaniib8887e92018-02-14 00:51:05 -03001978 mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, &disk_number);
1979 zip->file_info.disk_number = (uint32_t)disk_number;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001980 zip->file_info.disk_offset = mz_stream_tell(zip->stream);
Nathan Moinvaziri215a7a82020-12-13 10:58:41 -08001981
1982 if (zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED) {
1983#ifdef HAVE_PKCRYPT
1984 /* Pre-calculated CRC value is required for PKWARE traditional encryption */
1985 uint32_t dos_date = mz_zip_time_t_to_dos_date(zip->file_info.modified_date);
1986 zip->file_info.pk_verify = mz_zip_get_pk_verify(dos_date, zip->file_info.crc, zip->file_info.flag);
1987#endif
1988#ifdef HAVE_WZAES
1989 if (zip->file_info.aes_version && zip->file_info.aes_encryption_mode == 0)
1990 zip->file_info.aes_encryption_mode = MZ_AES_ENCRYPTION_MODE_256;
1991#endif
1992 }
1993
Nathan Moinvaziria66cc312017-10-18 16:51:10 -07001994 zip->file_info.crc = 0;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07001995 zip->file_info.compressed_size = 0;
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07001996
Nathan Moinvazirifa620e42018-10-20 10:37:10 -07001997 if ((compress_level == 0) || (is_dir))
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07001998 zip->file_info.compression_method = MZ_COMPRESS_METHOD_STORE;
Viktor Szakats2884e672018-04-30 08:12:13 +00001999
Nathan Moinvaziria6d1f662018-07-22 10:35:49 -07002000#ifdef MZ_ZIP_NO_COMPRESSION
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07002001 if (zip->file_info.compression_method != MZ_COMPRESS_METHOD_STORE)
Nathan Moinvaziri0a9282d2018-06-19 11:59:07 -07002002 err = MZ_SUPPORT_ERROR;
2003#endif
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002004 if (err == MZ_OK)
Nathan Moinvazirifa620e42018-10-20 10:37:10 -07002005 err = mz_zip_entry_write_header(zip->stream, 1, &zip->file_info);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002006 if (err == MZ_OK)
Nathan Moinvazirifa146ad2018-08-07 23:50:31 -07002007 err = mz_zip_entry_open_int(handle, raw, compress_level, password);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002008
2009 return err;
2010}
2011
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002012int32_t mz_zip_entry_read(void *handle, void *buf, int32_t len) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002013 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002014 int32_t read = 0;
2015
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07002016 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002017 return MZ_PARAM_ERROR;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002018 if (UINT_MAX == UINT16_MAX && len > UINT16_MAX) /* zlib limitation */
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002019 return MZ_PARAM_ERROR;
Nathan Moinvazirid7814e92018-07-11 14:54:14 -07002020 if (len == 0)
2021 return MZ_PARAM_ERROR;
2022
Nathan Moinvazirieb80cd92018-07-11 16:45:09 -07002023 if (zip->file_info.compressed_size == 0)
2024 return 0;
2025
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002026 /* Read entire entry even if uncompressed_size = 0, otherwise */
2027 /* aes encryption validation will fail if compressed_size > 0 */
Nathan Moinvazirie63d2312018-11-03 19:45:41 -07002028 read = mz_stream_read(zip->compress_stream, buf, len);
2029 if (read > 0)
2030 zip->entry_crc32 = mz_crypt_crc32_update(zip->entry_crc32, buf, read);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08002031
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07002032 mz_zip_print("Zip - Entry - Read - %" PRId32 " (max %" PRId32 ")\n", read, len);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08002033
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002034 return read;
2035}
2036
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002037int32_t mz_zip_entry_write(void *handle, const void *buf, int32_t len) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002038 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri0a9282d2018-06-19 11:59:07 -07002039 int32_t written = 0;
2040
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07002041 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002042 return MZ_PARAM_ERROR;
Nathan Moinvazirie63d2312018-11-03 19:45:41 -07002043 written = mz_stream_write(zip->compress_stream, buf, len);
2044 if (written > 0)
2045 zip->entry_crc32 = mz_crypt_crc32_update(zip->entry_crc32, buf, written);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08002046
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07002047 mz_zip_print("Zip - Entry - Write - %" PRId32 " (max %" PRId32 ")\n", written, len);
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08002048
Nathan Moinvaziri0a9282d2018-06-19 11:59:07 -07002049 return written;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002050}
2051
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002052int32_t mz_zip_entry_read_close(void *handle, uint32_t *crc32, int64_t *compressed_size,
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002053 int64_t *uncompressed_size) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002054 mz_zip *zip = (mz_zip *)handle;
2055 int64_t total_in = 0;
2056 int32_t err = MZ_OK;
2057 uint8_t zip64 = 0;
2058
2059 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
2060 return MZ_PARAM_ERROR;
2061
2062 mz_stream_close(zip->compress_stream);
2063
2064 mz_zip_print("Zip - Entry - Read Close\n");
2065
2066 if (crc32 != NULL)
2067 *crc32 = zip->file_info.crc;
2068 if (compressed_size != NULL)
2069 *compressed_size = zip->file_info.compressed_size;
2070 if (uncompressed_size != NULL)
2071 *uncompressed_size = zip->file_info.uncompressed_size;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002072
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002073 mz_stream_get_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_IN, &total_in);
2074
2075 if ((zip->file_info.flag & MZ_ZIP_FLAG_DATA_DESCRIPTOR) &&
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002076 ((zip->file_info.flag & MZ_ZIP_FLAG_MASK_LOCAL_INFO) == 0) &&
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002077 (crc32 != NULL || compressed_size != NULL || uncompressed_size != NULL)) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002078 /* Check to see if data descriptor is zip64 bit format or not */
2079 if (mz_zip_extrafield_contains(zip->local_file_info.extrafield,
2080 zip->local_file_info.extrafield_size, MZ_ZIP_EXTENSION_ZIP64, NULL) == MZ_OK)
2081 zip64 = 1;
2082
Nathan Moinvaziri0030fa22020-12-21 19:31:07 -08002083 err = mz_zip_entry_seek_local_header(handle);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002084
2085 /* Seek to end of compressed stream since we might have over-read during compression */
2086 if (err == MZ_OK)
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002087 err = mz_stream_seek(zip->stream, MZ_ZIP_SIZE_LD_ITEM +
2088 (int64_t)zip->local_file_info.filename_size +
2089 (int64_t)zip->local_file_info.extrafield_size +
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002090 total_in, MZ_SEEK_CUR);
2091
2092 /* Read data descriptor */
2093 if (err == MZ_OK)
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002094 err = mz_zip_entry_read_descriptor(zip->stream, zip64,
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002095 crc32, compressed_size, uncompressed_size);
2096 }
2097
2098 /* If entire entry was not read verification will fail */
Markus Alindff817302022-11-01 16:35:20 +01002099 if ((err == MZ_OK) && (total_in == zip->file_info.compressed_size) && (!zip->entry_raw)) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002100#ifdef HAVE_WZAES
2101 /* AES zip version AE-1 will expect a valid crc as well */
2102 if (zip->file_info.aes_version <= 0x0001)
2103#endif
2104 {
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002105 if (zip->entry_crc32 != zip->file_info.crc) {
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07002106 mz_zip_print("Zip - Entry - Crc failed (actual 0x%08" PRIx32 " expected 0x%08" PRIx32 ")\n",
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002107 zip->entry_crc32, zip->file_info.crc);
2108
2109 err = MZ_CRC_ERROR;
2110 }
2111 }
2112 }
2113
2114 mz_zip_entry_close_int(handle);
2115
2116 return err;
2117}
2118
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002119int32_t mz_zip_entry_write_close(void *handle, uint32_t crc32, int64_t compressed_size,
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002120 int64_t uncompressed_size) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002121 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002122 int64_t end_disk_number = 0;
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002123 int32_t err = MZ_OK;
2124 uint8_t zip64 = 0;
2125
2126 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
2127 return MZ_PARAM_ERROR;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002128
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002129 mz_stream_close(zip->compress_stream);
2130
2131 if (!zip->entry_raw)
2132 crc32 = zip->entry_crc32;
2133
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07002134 mz_zip_print("Zip - Entry - Write Close (crc 0x%08" PRIx32 " cs %" PRId64 " ucs %" PRId64 ")\n",
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002135 crc32, compressed_size, uncompressed_size);
2136
2137 /* If sizes are not set, then read them from the compression stream */
2138 if (compressed_size < 0)
2139 mz_stream_get_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_OUT, &compressed_size);
2140 if (uncompressed_size < 0)
Nathan Moinvazirid9318432018-12-01 07:56:28 -08002141 mz_stream_get_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_IN, &uncompressed_size);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002142
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002143 if (zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002144 mz_stream_set_base(zip->crypt_stream, zip->stream);
2145 err = mz_stream_close(zip->crypt_stream);
2146
2147 mz_stream_get_prop_int64(zip->crypt_stream, MZ_STREAM_PROP_TOTAL_OUT, &compressed_size);
2148 }
2149
Nathan Moinvazirief93d652020-12-22 17:20:04 -08002150 mz_zip_entry_needs_zip64(&zip->file_info, 1, &zip64);
2151
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002152 if ((err == MZ_OK) && (zip->file_info.flag & MZ_ZIP_FLAG_DATA_DESCRIPTOR)) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002153 /* Determine if we need to write data descriptor in zip64 format,
2154 if local extrafield was saved with zip64 extrafield */
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002155
2156 if (zip->file_info.flag & MZ_ZIP_FLAG_MASK_LOCAL_INFO)
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002157 err = mz_zip_entry_write_descriptor(zip->stream,
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002158 zip64, 0, compressed_size, 0);
2159 else
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002160 err = mz_zip_entry_write_descriptor(zip->stream,
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002161 zip64, crc32, compressed_size, uncompressed_size);
2162 }
2163
2164 /* Write file info to central directory */
2165
Nathan Moinvazirife33c3d2019-09-16 11:07:43 -07002166 mz_zip_print("Zip - Entry - Write cd (ucs %" PRId64 " cs %" PRId64 " crc 0x%08" PRIx32 ")\n",
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002167 uncompressed_size, compressed_size, crc32);
2168
2169 zip->file_info.crc = crc32;
2170 zip->file_info.compressed_size = compressed_size;
2171 zip->file_info.uncompressed_size = uncompressed_size;
2172
2173 if (err == MZ_OK)
2174 err = mz_zip_entry_write_header(zip->cd_mem_stream, 0, &zip->file_info);
2175
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002176 /* Update local header with crc32 and sizes */
2177 if ((err == MZ_OK) && ((zip->file_info.flag & MZ_ZIP_FLAG_DATA_DESCRIPTOR) == 0) &&
Nathan Moinvazirica270892020-12-27 14:54:57 -08002178 ((zip->file_info.flag & MZ_ZIP_FLAG_MASK_LOCAL_INFO) == 0)) {
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002179 /* Save the disk number and position we are to seek back after updating local header */
Nathan Moinvaziri58b37d32020-02-08 20:32:58 -08002180 int64_t end_pos = mz_stream_tell(zip->stream);
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002181 mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, &end_disk_number);
2182
Nathan Moinvaziri0030fa22020-12-21 19:31:07 -08002183 err = mz_zip_entry_seek_local_header(handle);
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002184
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002185 if (err == MZ_OK) {
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002186 /* Seek to crc32 and sizes offset in local header */
2187 err = mz_stream_seek(zip->stream, MZ_ZIP_OFFSET_CRC_SIZES, MZ_SEEK_CUR);
2188 }
2189
2190 if (err == MZ_OK)
Nathan Moinvazirief93d652020-12-22 17:20:04 -08002191 err = mz_zip_entry_write_crc_sizes(zip->stream, zip64, 0, &zip->file_info);
2192
2193 /* Seek to and update zip64 extension sizes */
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -08002194 if ((err == MZ_OK) && (zip64)) {
Nathan Moinvaziri8516c572020-12-23 21:59:35 -08002195 int64_t filename_size = zip->file_info.filename_size;
Nathan Moinvazirief93d652020-12-22 17:20:04 -08002196
Nathan Moinvaziri8516c572020-12-23 21:59:35 -08002197 if (filename_size == 0)
2198 filename_size = strlen(zip->file_info.filename);
2199
2200 /* Since we write zip64 extension first we know its offset */
2201 err = mz_stream_seek(zip->stream, 2 + 2 + filename_size + 4, MZ_SEEK_CUR);
2202
Nathan Moinvaziri3d83acd2020-12-23 11:14:59 -08002203 if (err == MZ_OK)
Nathan Moinvaziri8516c572020-12-23 21:59:35 -08002204 err = mz_stream_write_uint64(zip->stream, zip->file_info.uncompressed_size);
2205 if (err == MZ_OK)
2206 err = mz_stream_write_uint64(zip->stream, zip->file_info.compressed_size);
Nathan Moinvazirief93d652020-12-22 17:20:04 -08002207 }
2208
Nathan Moinvaziri6c87ff42020-01-03 16:58:16 -08002209 mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, end_disk_number);
2210 mz_stream_seek(zip->stream, end_pos, MZ_SEEK_SET);
2211 }
2212
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002213 zip->number_entry += 1;
2214
2215 mz_zip_entry_close_int(handle);
2216
2217 return err;
2218}
2219
Nathan Moinvaziri0030fa22020-12-21 19:31:07 -08002220int32_t mz_zip_entry_seek_local_header(void *handle) {
2221 mz_zip *zip = (mz_zip *)handle;
2222 int64_t disk_size = 0;
2223 uint32_t disk_number = zip->file_info.disk_number;
2224
2225 if (disk_number == zip->disk_number_with_cd) {
2226 mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_SIZE, &disk_size);
2227 if ((disk_size == 0) || ((zip->open_mode & MZ_OPEN_MODE_WRITE) == 0))
2228 disk_number = (uint32_t)-1;
2229 }
2230
2231 mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, disk_number);
2232
2233 mz_zip_print("Zip - Entry - Seek local (disk %" PRId32 " offset %" PRId64 ")\n",
2234 disk_number, zip->file_info.disk_offset);
2235
2236 /* Guard against seek overflows */
2237 if ((zip->disk_offset_shift > 0) &&
2238 (zip->file_info.disk_offset > (INT64_MAX - zip->disk_offset_shift)))
2239 return MZ_FORMAT_ERROR;
2240
2241 return mz_stream_seek(zip->stream, zip->file_info.disk_offset + zip->disk_offset_shift, MZ_SEEK_SET);
2242}
2243
Nathan Moinvaziri9e754e52022-04-27 10:56:23 -07002244int32_t mz_zip_entry_get_compress_stream(void *handle, void **compress_stream) {
2245 mz_zip *zip = (mz_zip *)handle;
2246 if (zip == NULL || compress_stream == NULL)
2247 return MZ_PARAM_ERROR;
2248 *compress_stream = zip->compress_stream;
2249 if (*compress_stream == NULL)
2250 return MZ_EXIST_ERROR;
2251 return MZ_OK;
2252}
2253
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002254int32_t mz_zip_entry_close(void *handle) {
Nathan Moinvaziri163352f2020-02-08 21:28:23 -08002255 return mz_zip_entry_close_raw(handle, UINT64_MAX, 0);
2256}
2257
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002258int32_t mz_zip_entry_close_raw(void *handle, int64_t uncompressed_size, uint32_t crc32) {
Nathan Moinvaziri163352f2020-02-08 21:28:23 -08002259 mz_zip *zip = (mz_zip *)handle;
2260 int32_t err = MZ_OK;
2261
2262 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
2263 return MZ_PARAM_ERROR;
2264
2265 if (zip->open_mode & MZ_OPEN_MODE_WRITE)
2266 err = mz_zip_entry_write_close(handle, crc32, UINT64_MAX, uncompressed_size);
2267 else
2268 err = mz_zip_entry_read_close(handle, NULL, NULL, NULL);
2269
2270 return err;
2271}
2272
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002273int32_t mz_zip_entry_is_dir(void *handle) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002274 mz_zip *zip = (mz_zip *)handle;
2275 int32_t filename_length = 0;
2276
2277 if (zip == NULL)
2278 return MZ_PARAM_ERROR;
2279 if (zip->entry_scanned == 0)
2280 return MZ_PARAM_ERROR;
2281 if (mz_zip_attrib_is_dir(zip->file_info.external_fa, zip->file_info.version_madeby) == MZ_OK)
2282 return MZ_OK;
2283
2284 filename_length = (int32_t)strlen(zip->file_info.filename);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002285 if (filename_length > 0) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002286 if ((zip->file_info.filename[filename_length - 1] == '/') ||
2287 (zip->file_info.filename[filename_length - 1] == '\\'))
2288 return MZ_OK;
2289 }
2290 return MZ_EXIST_ERROR;
2291}
2292
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002293int32_t mz_zip_entry_is_symlink(void *handle) {
Nathan Moinvaziri3249eac2019-05-02 21:07:39 -07002294 mz_zip *zip = (mz_zip *)handle;
2295
2296 if (zip == NULL)
2297 return MZ_PARAM_ERROR;
2298 if (zip->entry_scanned == 0)
2299 return MZ_PARAM_ERROR;
Nathan Moinvaziri426543d2019-05-07 09:46:25 -07002300 if (mz_zip_attrib_is_symlink(zip->file_info.external_fa, zip->file_info.version_madeby) != MZ_OK)
2301 return MZ_EXIST_ERROR;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002302
Nathan Moinvaziria5a1d5d2019-05-05 20:13:58 -07002303 return MZ_OK;
Nathan Moinvaziri3249eac2019-05-02 21:07:39 -07002304}
2305
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002306int32_t mz_zip_entry_get_info(void *handle, mz_zip_file **file_info) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002307 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002308
2309 if (zip == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002310 return MZ_PARAM_ERROR;
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002311
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002312 if ((zip->open_mode & MZ_OPEN_MODE_WRITE) == 0) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002313 if (!zip->entry_scanned)
2314 return MZ_PARAM_ERROR;
2315 }
2316
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002317 *file_info = &zip->file_info;
2318 return MZ_OK;
2319}
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002320
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002321int32_t mz_zip_entry_get_local_info(void *handle, mz_zip_file **local_file_info) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002322 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07002323 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002324 return MZ_PARAM_ERROR;
2325 *local_file_info = &zip->local_file_info;
2326 return MZ_OK;
2327}
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002328
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002329int32_t mz_zip_entry_set_extrafield(void *handle, const uint8_t *extrafield, uint16_t extrafield_size) {
Nathan Moinvaziri915c5132018-10-26 20:00:52 -07002330 mz_zip *zip = (mz_zip *)handle;
2331
2332 if (zip == NULL || mz_zip_entry_is_open(handle) != MZ_OK)
2333 return MZ_PARAM_ERROR;
2334
2335 zip->file_info.extrafield = extrafield;
2336 zip->file_info.extrafield_size = extrafield_size;
2337 return MZ_OK;
2338}
2339
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002340static int32_t mz_zip_goto_next_entry_int(void *handle) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002341 mz_zip *zip = (mz_zip *)handle;
2342 int32_t err = MZ_OK;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002343
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002344 if (zip == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002345 return MZ_PARAM_ERROR;
2346
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002347 zip->entry_scanned = 0;
2348
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002349 mz_stream_set_prop_int64(zip->cd_stream, MZ_STREAM_PROP_DISK_NUMBER, -1);
Viktor Szakats915b82e2018-04-24 10:02:39 +00002350
Nathan Moinvaziri79ac3382017-10-23 17:22:36 -07002351 err = mz_stream_seek(zip->cd_stream, zip->cd_current_pos, MZ_SEEK_SET);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002352 if (err == MZ_OK)
Nathan Moinvaziricda36002017-10-21 09:37:18 -07002353 err = mz_zip_entry_read_header(zip->cd_stream, 0, &zip->file_info, zip->file_info_stream);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002354 if (err == MZ_OK)
2355 zip->entry_scanned = 1;
2356 return err;
2357}
2358
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002359int64_t mz_zip_get_entry(void *handle) {
Viktor Szakatsce26dba2018-04-22 19:30:34 +00002360 mz_zip *zip = (mz_zip *)handle;
2361
2362 if (zip == NULL)
2363 return MZ_PARAM_ERROR;
2364
2365 return zip->cd_current_pos;
2366}
2367
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002368int32_t mz_zip_goto_entry(void *handle, int64_t cd_pos) {
Viktor Szakatsce26dba2018-04-22 19:30:34 +00002369 mz_zip *zip = (mz_zip *)handle;
2370
2371 if (zip == NULL)
2372 return MZ_PARAM_ERROR;
2373
2374 if (cd_pos < zip->cd_start_pos || cd_pos > zip->cd_start_pos + zip->cd_size)
2375 return MZ_PARAM_ERROR;
2376
2377 zip->cd_current_pos = cd_pos;
2378
2379 return mz_zip_goto_next_entry_int(handle);
2380}
2381
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002382int32_t mz_zip_goto_first_entry(void *handle) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002383 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002384
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002385 if (zip == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002386 return MZ_PARAM_ERROR;
2387
Nathan Moinvaziricda36002017-10-21 09:37:18 -07002388 zip->cd_current_pos = zip->cd_start_pos;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002389
2390 return mz_zip_goto_next_entry_int(handle);
2391}
2392
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002393int32_t mz_zip_goto_next_entry(void *handle) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002394 mz_zip *zip = (mz_zip *)handle;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002395
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002396 if (zip == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002397 return MZ_PARAM_ERROR;
2398
Nathan Moinvaziri904c4082018-10-08 23:31:21 -07002399 zip->cd_current_pos += (int64_t)MZ_ZIP_SIZE_CD_ITEM + zip->file_info.filename_size +
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002400 zip->file_info.extrafield_size + zip->file_info.comment_size;
2401
2402 return mz_zip_goto_next_entry_int(handle);
2403}
2404
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002405int32_t mz_zip_locate_entry(void *handle, const char *filename, uint8_t ignore_case) {
Nathan Moinvaziri5f9e5d32017-10-20 07:59:39 -07002406 mz_zip *zip = (mz_zip *)handle;
2407 int32_t err = MZ_OK;
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002408 int32_t result = 0;
2409
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07002410 if (zip == NULL || filename == NULL)
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002411 return MZ_PARAM_ERROR;
2412
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002413 /* If we are already on the current entry, no need to search */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002414 if ((zip->entry_scanned) && (zip->file_info.filename != NULL)) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002415 result = mz_zip_path_compare(zip->file_info.filename, filename, ignore_case);
Nathan Moinvaziri77687f32018-08-09 16:29:26 -07002416 if (result == 0)
2417 return MZ_OK;
2418 }
2419
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002420 /* Search all entries starting at the first */
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002421 err = mz_zip_goto_first_entry(handle);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002422 while (err == MZ_OK) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002423 result = mz_zip_path_compare(zip->file_info.filename, filename, ignore_case);
2424 if (result == 0)
2425 return MZ_OK;
2426
2427 err = mz_zip_goto_next_entry(handle);
2428 }
2429
2430 return err;
2431}
2432
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002433int32_t mz_zip_locate_first_entry(void *handle, void *userdata, mz_zip_locate_entry_cb cb) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002434 mz_zip *zip = (mz_zip *)handle;
2435 int32_t err = MZ_OK;
2436 int32_t result = 0;
2437
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002438 /* Search first entry looking for match */
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002439 err = mz_zip_goto_first_entry(handle);
2440 if (err != MZ_OK)
2441 return err;
2442
2443 result = cb(handle, userdata, &zip->file_info);
2444 if (result == 0)
2445 return MZ_OK;
2446
2447 return mz_zip_locate_next_entry(handle, userdata, cb);
2448}
2449
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002450int32_t mz_zip_locate_next_entry(void *handle, void *userdata, mz_zip_locate_entry_cb cb) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002451 mz_zip *zip = (mz_zip *)handle;
2452 int32_t err = MZ_OK;
2453 int32_t result = 0;
2454
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002455 /* Search next entries looking for match */
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002456 err = mz_zip_goto_next_entry(handle);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002457 while (err == MZ_OK) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002458 result = cb(handle, userdata, &zip->file_info);
Nathan Moinvaziri8f67ff92017-10-17 23:22:29 -07002459 if (result == 0)
2460 return MZ_OK;
2461
2462 err = mz_zip_goto_next_entry(handle);
2463 }
2464
2465 return err;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002466}
2467
2468/***************************************************************************/
2469
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002470int32_t mz_zip_attrib_is_dir(uint32_t attrib, int32_t version_madeby) {
Nathan Moinvaziri2bb21d72018-10-08 21:47:15 -07002471 uint32_t posix_attrib = 0;
Nathan Moinvaziri904c4082018-10-08 23:31:21 -07002472 uint8_t system = MZ_HOST_SYSTEM(version_madeby);
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002473 int32_t err = MZ_OK;
Nathan Moinvaziribb3b75b2018-05-02 10:48:51 -07002474
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002475 err = mz_zip_attrib_convert(system, attrib, MZ_HOST_SYSTEM_UNIX, &posix_attrib);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002476 if (err == MZ_OK) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002477 if ((posix_attrib & 0170000) == 0040000) /* S_ISDIR */
Nathan Moinvaziribb3b75b2018-05-02 10:48:51 -07002478 return MZ_OK;
2479 }
2480
2481 return MZ_EXIST_ERROR;
2482}
2483
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002484int32_t mz_zip_attrib_is_symlink(uint32_t attrib, int32_t version_madeby) {
Nathan Moinvaziri426543d2019-05-07 09:46:25 -07002485 uint32_t posix_attrib = 0;
2486 uint8_t system = MZ_HOST_SYSTEM(version_madeby);
2487 int32_t err = MZ_OK;
2488
2489 err = mz_zip_attrib_convert(system, attrib, MZ_HOST_SYSTEM_UNIX, &posix_attrib);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002490 if (err == MZ_OK) {
Nathan Moinvaziri426543d2019-05-07 09:46:25 -07002491 if ((posix_attrib & 0170000) == 0120000) /* S_ISLNK */
2492 return MZ_OK;
2493 }
2494
2495 return MZ_EXIST_ERROR;
2496}
2497
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002498int32_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 -07002499 if (target_attrib == NULL)
2500 return MZ_PARAM_ERROR;
2501
2502 *target_attrib = 0;
2503
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002504 if ((src_sys == MZ_HOST_SYSTEM_MSDOS) || (src_sys == MZ_HOST_SYSTEM_WINDOWS_NTFS)) {
2505 if ((target_sys == MZ_HOST_SYSTEM_MSDOS) || (target_sys == MZ_HOST_SYSTEM_WINDOWS_NTFS)) {
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002506 *target_attrib = src_attrib;
2507 return MZ_OK;
2508 }
Cameron Cawley28f78ff2019-08-06 14:14:14 +01002509 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 -07002510 return mz_zip_attrib_win32_to_posix(src_attrib, target_attrib);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002511 } else if ((src_sys == MZ_HOST_SYSTEM_UNIX) || (src_sys == MZ_HOST_SYSTEM_OSX_DARWIN) || (src_sys == MZ_HOST_SYSTEM_RISCOS)) {
Jeremy Bernstein15ebdb72022-10-05 16:36:34 +02002512 /* If high bytes are set, it contains unix specific attributes */
2513 if ((src_attrib >> 16) != 0)
2514 src_attrib >>= 16;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002515
Jeremy Bernstein15ebdb72022-10-05 16:36:34 +02002516 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 -07002517 *target_attrib = src_attrib;
2518 return MZ_OK;
2519 }
Nathan Moinvaziri2bb21d72018-10-08 21:47:15 -07002520 if ((target_sys == MZ_HOST_SYSTEM_MSDOS) || (target_sys == MZ_HOST_SYSTEM_WINDOWS_NTFS))
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002521 return mz_zip_attrib_posix_to_win32(src_attrib, target_attrib);
2522 }
2523
2524 return MZ_SUPPORT_ERROR;
2525}
2526
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002527int32_t mz_zip_attrib_posix_to_win32(uint32_t posix_attrib, uint32_t *win32_attrib) {
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002528 if (win32_attrib == NULL)
2529 return MZ_PARAM_ERROR;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002530
Nathan Moinvaziri2bb21d72018-10-08 21:47:15 -07002531 *win32_attrib = 0;
Nathan Moinvaziri32fd8052019-07-04 10:32:02 -07002532
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002533 /* S_IWUSR | S_IWGRP | S_IWOTH | S_IXUSR | S_IXGRP | S_IXOTH */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002534 if ((posix_attrib & 0000333) == 0 && (posix_attrib & 0000444) != 0)
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002535 *win32_attrib |= 0x01; /* FILE_ATTRIBUTE_READONLY */
Nathan Moinvaziri426543d2019-05-07 09:46:25 -07002536 /* S_IFLNK */
2537 if ((posix_attrib & 0170000) == 0120000)
2538 *win32_attrib |= 0x400; /* FILE_ATTRIBUTE_REPARSE_POINT */
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002539 /* S_IFDIR */
Nathan Moinvaziri7d94b1f2019-05-08 23:41:47 -07002540 else if ((posix_attrib & 0170000) == 0040000)
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002541 *win32_attrib |= 0x10; /* FILE_ATTRIBUTE_DIRECTORY */
2542 /* S_IFREG */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002543 else
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002544 *win32_attrib |= 0x80; /* FILE_ATTRIBUTE_NORMAL */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002545
2546 return MZ_OK;
2547}
2548
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002549int32_t mz_zip_attrib_win32_to_posix(uint32_t win32_attrib, uint32_t *posix_attrib) {
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002550 if (posix_attrib == NULL)
2551 return MZ_PARAM_ERROR;
2552
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002553 *posix_attrib = 0000444; /* S_IRUSR | S_IRGRP | S_IROTH */
2554 /* FILE_ATTRIBUTE_READONLY */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002555 if ((win32_attrib & 0x01) == 0)
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002556 *posix_attrib |= 0000222; /* S_IWUSR | S_IWGRP | S_IWOTH */
Nathan Moinvaziri426543d2019-05-07 09:46:25 -07002557 /* FILE_ATTRIBUTE_REPARSE_POINT */
2558 if ((win32_attrib & 0x400) == 0x400)
2559 *posix_attrib |= 0120000; /* S_IFLNK */
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002560 /* FILE_ATTRIBUTE_DIRECTORY */
Nathan Moinvaziri7d94b1f2019-05-08 23:41:47 -07002561 else if ((win32_attrib & 0x10) == 0x10)
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002562 *posix_attrib |= 0040111; /* S_IFDIR | S_IXUSR | S_IXGRP | S_IXOTH */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002563 else
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002564 *posix_attrib |= 0100000; /* S_IFREG */
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002565
2566 return MZ_OK;
2567}
2568
Nathan Moinvaziribb3b75b2018-05-02 10:48:51 -07002569/***************************************************************************/
2570
Nathan Moinvaziri8f769682020-12-23 11:26:30 -08002571int32_t mz_zip_extrafield_find(void *stream, uint16_t type, int32_t max_seek, uint16_t *length) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002572 int32_t err = MZ_OK;
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002573 uint16_t field_type = 0;
2574 uint16_t field_length = 0;
2575
Nathan Moinvaziri8f769682020-12-23 11:26:30 -08002576 if (max_seek < 4)
2577 return MZ_EXIST_ERROR;
2578
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002579 do {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002580 err = mz_stream_read_uint16(stream, &field_type);
2581 if (err == MZ_OK)
2582 err = mz_stream_read_uint16(stream, &field_length);
2583 if (err != MZ_OK)
2584 break;
2585
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002586 if (type == field_type) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002587 if (length != NULL)
2588 *length = field_length;
2589 return MZ_OK;
2590 }
2591
Nathan Moinvaziri8f769682020-12-23 11:26:30 -08002592 max_seek -= field_length - 4;
2593 if (max_seek < 0)
2594 return MZ_EXIST_ERROR;
2595
Nathan Moinvaziri0dfcdf92018-11-07 20:33:47 -08002596 err = mz_stream_seek(stream, field_length, MZ_SEEK_CUR);
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002597 } while (err == MZ_OK);
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002598
2599 return MZ_EXIST_ERROR;
2600}
2601
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002602int32_t mz_zip_extrafield_contains(const uint8_t *extrafield, int32_t extrafield_size,
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002603 uint16_t type, uint16_t *length) {
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002604 void *file_extra_stream = NULL;
2605 int32_t err = MZ_OK;
2606
2607 if (extrafield == NULL || extrafield_size == 0)
2608 return MZ_PARAM_ERROR;
2609
2610 mz_stream_mem_create(&file_extra_stream);
2611 mz_stream_mem_set_buffer(file_extra_stream, (void *)extrafield, extrafield_size);
2612
Nathan Moinvaziri8f769682020-12-23 11:26:30 -08002613 err = mz_zip_extrafield_find(file_extra_stream, type, extrafield_size, length);
Nathan Moinvaziriba1db162018-11-23 10:07:35 -08002614
2615 mz_stream_mem_delete(&file_extra_stream);
2616
2617 return err;
2618}
2619
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002620int32_t mz_zip_extrafield_read(void *stream, uint16_t *type, uint16_t *length) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002621 int32_t err = MZ_OK;
2622 if (type == NULL || length == NULL)
2623 return MZ_PARAM_ERROR;
2624 err = mz_stream_read_uint16(stream, type);
2625 if (err == MZ_OK)
2626 err = mz_stream_read_uint16(stream, length);
2627 return err;
2628}
2629
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002630int32_t mz_zip_extrafield_write(void *stream, uint16_t type, uint16_t length) {
Nathan Moinvazirie1f68fc2018-10-23 09:04:04 -07002631 int32_t err = MZ_OK;
2632 err = mz_stream_write_uint16(stream, type);
2633 if (err == MZ_OK)
2634 err = mz_stream_write_uint16(stream, length);
2635 return err;
2636}
2637
2638/***************************************************************************/
2639
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002640static int32_t mz_zip_invalid_date(const struct tm *ptm) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002641#define datevalue_in_range(min, max, value) ((min) <= (value) && (value) <= (max))
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002642 return (!datevalue_in_range(0, 127 + 80, ptm->tm_year) || /* 1980-based year, allow 80 extra */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002643 !datevalue_in_range(0, 11, ptm->tm_mon) ||
2644 !datevalue_in_range(1, 31, ptm->tm_mday) ||
2645 !datevalue_in_range(0, 23, ptm->tm_hour) ||
2646 !datevalue_in_range(0, 59, ptm->tm_min) ||
2647 !datevalue_in_range(0, 59, ptm->tm_sec));
2648#undef datevalue_in_range
2649}
2650
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002651static void mz_zip_dosdate_to_raw_tm(uint64_t dos_date, struct tm *ptm) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002652 uint64_t date = (uint64_t)(dos_date >> 16);
2653
Nathan Moinvaziri1d6d1832018-11-10 09:03:55 -08002654 ptm->tm_mday = (uint16_t)(date & 0x1f);
2655 ptm->tm_mon = (uint16_t)(((date & 0x1E0) / 0x20) - 1);
2656 ptm->tm_year = (uint16_t)(((date & 0x0FE00) / 0x0200) + 80);
2657 ptm->tm_hour = (uint16_t)((dos_date & 0xF800) / 0x800);
2658 ptm->tm_min = (uint16_t)((dos_date & 0x7E0) / 0x20);
2659 ptm->tm_sec = (uint16_t)(2 * (dos_date & 0x1f));
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002660 ptm->tm_isdst = -1;
2661}
2662
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002663int32_t mz_zip_dosdate_to_tm(uint64_t dos_date, struct tm *ptm) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002664 if (ptm == NULL)
2665 return MZ_PARAM_ERROR;
2666
2667 mz_zip_dosdate_to_raw_tm(dos_date, ptm);
2668
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002669 if (mz_zip_invalid_date(ptm)) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002670 /* Invalid date stored, so don't return it */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002671 memset(ptm, 0, sizeof(struct tm));
2672 return MZ_FORMAT_ERROR;
2673 }
2674 return MZ_OK;
2675}
2676
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002677time_t mz_zip_dosdate_to_time_t(uint64_t dos_date) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002678 struct tm ptm;
2679 mz_zip_dosdate_to_raw_tm(dos_date, &ptm);
2680 return mktime(&ptm);
2681}
2682
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002683int32_t mz_zip_time_t_to_tm(time_t unix_time, struct tm *ptm) {
Nathan Moinvaziricd3e9e12019-04-28 10:22:49 -07002684 struct tm ltm;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002685 if (ptm == NULL)
2686 return MZ_PARAM_ERROR;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002687 if (localtime_r(&unix_time, &ltm) == NULL) { /* Returns a 1900-based year */
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002688 /* Invalid date stored, so don't return it */
Viktor Szakatsce26dba2018-04-22 19:30:34 +00002689 memset(ptm, 0, sizeof(struct tm));
Nathan Moinvaziri17fcdcd2017-10-26 08:13:13 -07002690 return MZ_INTERNAL_ERROR;
Viktor Szakatsce26dba2018-04-22 19:30:34 +00002691 }
Nathan Moinvaziricd3e9e12019-04-28 10:22:49 -07002692 memcpy(ptm, &ltm, sizeof(struct tm));
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002693 return MZ_OK;
2694}
2695
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002696uint32_t mz_zip_time_t_to_dos_date(time_t unix_time) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002697 struct tm ptm;
2698 mz_zip_time_t_to_tm(unix_time, &ptm);
2699 return mz_zip_tm_to_dosdate((const struct tm *)&ptm);
2700}
2701
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002702uint32_t mz_zip_tm_to_dosdate(const struct tm *ptm) {
Nathan Moinvazirie33916b2018-05-01 13:45:08 -07002703 struct tm fixed_tm;
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002704
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002705 /* Years supported: */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002706
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002707 /* [00, 79] (assumed to be between 2000 and 2079) */
2708 /* [80, 207] (assumed to be between 1980 and 2107, typical output of old */
2709 /* software that does 'year-1900' to get a double digit year) */
2710 /* [1980, 2107] (due to format limitations, only years 1980-2107 can be stored.) */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002711
2712 memcpy(&fixed_tm, ptm, sizeof(struct tm));
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002713 if (fixed_tm.tm_year >= 1980) /* range [1980, 2107] */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002714 fixed_tm.tm_year -= 1980;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002715 else if (fixed_tm.tm_year >= 80) /* range [80, 207] */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002716 fixed_tm.tm_year -= 80;
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002717 else /* range [00, 79] */
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002718 fixed_tm.tm_year += 20;
2719
Viktor Szakatsce26dba2018-04-22 19:30:34 +00002720 if (mz_zip_invalid_date(&fixed_tm))
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002721 return 0;
2722
Anand K Mistry57b65f82018-11-09 10:52:19 +11002723 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 -07002724 (((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 -07002725}
2726
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002727int32_t mz_zip_ntfs_to_unix_time(uint64_t ntfs_time, time_t *unix_time) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002728 *unix_time = (time_t)((ntfs_time - 116444736000000000LL) / 10000000);
2729 return MZ_OK;
2730}
2731
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002732int32_t mz_zip_unix_to_ntfs_time(time_t unix_time, uint64_t *ntfs_time) {
Nathan Moinvaziri9eab88e2017-10-22 14:32:37 -07002733 *ntfs_time = ((uint64_t)unix_time * 10000000) + 116444736000000000LL;
2734 return MZ_OK;
2735}
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002736
Nathan Moinvaziri4b8e4b32018-08-20 09:06:23 -07002737/***************************************************************************/
2738
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002739int32_t mz_zip_path_compare(const char *path1, const char *path2, uint8_t ignore_case) {
2740 do {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002741 if ((*path1 == '\\' && *path2 == '/') ||
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002742 (*path2 == '\\' && *path1 == '/')) {
Nathan Moinvazirib16ab562018-11-20 16:56:21 -08002743 /* Ignore comparison of path slashes */
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002744 } else if (ignore_case) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002745 if (tolower(*path1) != tolower(*path2))
2746 break;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002747 } else if (*path1 != *path2) {
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002748 break;
2749 }
2750
2751 path1 += 1;
2752 path2 += 1;
Nathan Moinvaziridc6962b2020-06-14 15:19:14 -07002753 } while (*path1 != 0 && *path2 != 0);
Nathan Moinvaziri9a170d42018-08-13 23:07:42 -07002754
2755 if (ignore_case)
2756 return (int32_t)(tolower(*path1) - tolower(*path2));
2757
2758 return (int32_t)(*path1 - *path2);
2759}
2760
2761/***************************************************************************/
Nathan Moinvaziric4668ed2020-10-25 11:56:19 -07002762
2763const char* mz_zip_get_compression_method_string(int32_t compression_method)
2764{
2765 const char *method = "?";
Nathan Moinvaziri5cd28792020-10-25 12:50:12 -07002766 switch (compression_method) {
Nathan Moinvaziric4668ed2020-10-25 11:56:19 -07002767 case MZ_COMPRESS_METHOD_STORE:
2768 method = "stored";
2769 break;
2770 case MZ_COMPRESS_METHOD_DEFLATE:
2771 method = "deflate";
2772 break;
2773 case MZ_COMPRESS_METHOD_BZIP2:
2774 method = "bzip2";
2775 break;
2776 case MZ_COMPRESS_METHOD_LZMA:
2777 method = "lzma";
2778 break;
2779 case MZ_COMPRESS_METHOD_XZ:
2780 method = "xz";
2781 break;
2782 case MZ_COMPRESS_METHOD_ZSTD:
2783 method = "zstd";
2784 break;
2785 }
2786 return method;
2787}
2788
2789/***************************************************************************/