29typedef unsigned char mz_validate_uint16[
sizeof(mz_uint16) == 2 ? 1 : -1];
30typedef unsigned char mz_validate_uint32[
sizeof(mz_uint32) == 4 ? 1 : -1];
31typedef unsigned char mz_validate_uint64[
sizeof(mz_uint64) == 8 ? 1 : -1];
39mz_ulong mz_adler32(mz_ulong adler,
const unsigned char *ptr,
size_t buf_len)
41 mz_uint32 i, s1 = (mz_uint32)(adler & 0xffff), s2 = (mz_uint32)(adler >> 16);
42 size_t block_len = buf_len % 5552;
44 return MZ_ADLER32_INIT;
47 for (i = 0; i + 7 < block_len; i += 8, ptr += 8)
49 s1 += ptr[0], s2 += s1;
50 s1 += ptr[1], s2 += s1;
51 s1 += ptr[2], s2 += s1;
52 s1 += ptr[3], s2 += s1;
53 s1 += ptr[4], s2 += s1;
54 s1 += ptr[5], s2 += s1;
55 s1 += ptr[6], s2 += s1;
56 s1 += ptr[7], s2 += s1;
58 for (; i < block_len; ++i)
59 s1 += *ptr++, s2 += s1;
60 s1 %= 65521U, s2 %= 65521U;
64 return (s2 << 16) + s1;
69 mz_ulong mz_crc32(mz_ulong crc,
const mz_uint8 *ptr,
size_t buf_len)
71 static const mz_uint32 s_crc32[16] = { 0, 0x1db71064, 0x3b6e20c8, 0x26d930ac, 0x76dc4190, 0x6b6b51f4, 0x4db26158, 0x5005713c,
72 0xedb88320, 0xf00f9344, 0xd6d6a3e8, 0xcb61b38c, 0x9b64c2b0, 0x86d3d2d4, 0xa00ae278, 0xbdbdf21c };
73 mz_uint32 crcu32 = (mz_uint32)crc;
80 crcu32 = (crcu32 >> 4) ^ s_crc32[(crcu32 & 0xF) ^ (b & 0xF)];
81 crcu32 = (crcu32 >> 4) ^ s_crc32[(crcu32 & 0xF) ^ (b >> 4)];
88mz_ulong mz_crc32(mz_ulong crc,
const mz_uint8 *ptr,
size_t buf_len)
90 static const mz_uint32 s_crc_table[256] =
92 0x00000000, 0x77073096, 0xEE0E612C, 0x990951BA, 0x076DC419, 0x706AF48F, 0xE963A535,
93 0x9E6495A3, 0x0EDB8832, 0x79DCB8A4, 0xE0D5E91E, 0x97D2D988, 0x09B64C2B, 0x7EB17CBD,
94 0xE7B82D07, 0x90BF1D91, 0x1DB71064, 0x6AB020F2, 0xF3B97148, 0x84BE41DE, 0x1ADAD47D,
95 0x6DDDE4EB, 0xF4D4B551, 0x83D385C7, 0x136C9856, 0x646BA8C0, 0xFD62F97A, 0x8A65C9EC,
96 0x14015C4F, 0x63066CD9, 0xFA0F3D63, 0x8D080DF5, 0x3B6E20C8, 0x4C69105E, 0xD56041E4,
97 0xA2677172, 0x3C03E4D1, 0x4B04D447, 0xD20D85FD, 0xA50AB56B, 0x35B5A8FA, 0x42B2986C,
98 0xDBBBC9D6, 0xACBCF940, 0x32D86CE3, 0x45DF5C75, 0xDCD60DCF, 0xABD13D59, 0x26D930AC,
99 0x51DE003A, 0xC8D75180, 0xBFD06116, 0x21B4F4B5, 0x56B3C423, 0xCFBA9599, 0xB8BDA50F,
100 0x2802B89E, 0x5F058808, 0xC60CD9B2, 0xB10BE924, 0x2F6F7C87, 0x58684C11, 0xC1611DAB,
101 0xB6662D3D, 0x76DC4190, 0x01DB7106, 0x98D220BC, 0xEFD5102A, 0x71B18589, 0x06B6B51F,
102 0x9FBFE4A5, 0xE8B8D433, 0x7807C9A2, 0x0F00F934, 0x9609A88E, 0xE10E9818, 0x7F6A0DBB,
103 0x086D3D2D, 0x91646C97, 0xE6635C01, 0x6B6B51F4, 0x1C6C6162, 0x856530D8, 0xF262004E,
104 0x6C0695ED, 0x1B01A57B, 0x8208F4C1, 0xF50FC457, 0x65B0D9C6, 0x12B7E950, 0x8BBEB8EA,
105 0xFCB9887C, 0x62DD1DDF, 0x15DA2D49, 0x8CD37CF3, 0xFBD44C65, 0x4DB26158, 0x3AB551CE,
106 0xA3BC0074, 0xD4BB30E2, 0x4ADFA541, 0x3DD895D7, 0xA4D1C46D, 0xD3D6F4FB, 0x4369E96A,
107 0x346ED9FC, 0xAD678846, 0xDA60B8D0, 0x44042D73, 0x33031DE5, 0xAA0A4C5F, 0xDD0D7CC9,
108 0x5005713C, 0x270241AA, 0xBE0B1010, 0xC90C2086, 0x5768B525, 0x206F85B3, 0xB966D409,
109 0xCE61E49F, 0x5EDEF90E, 0x29D9C998, 0xB0D09822, 0xC7D7A8B4, 0x59B33D17, 0x2EB40D81,
110 0xB7BD5C3B, 0xC0BA6CAD, 0xEDB88320, 0x9ABFB3B6, 0x03B6E20C, 0x74B1D29A, 0xEAD54739,
111 0x9DD277AF, 0x04DB2615, 0x73DC1683, 0xE3630B12, 0x94643B84, 0x0D6D6A3E, 0x7A6A5AA8,
112 0xE40ECF0B, 0x9309FF9D, 0x0A00AE27, 0x7D079EB1, 0xF00F9344, 0x8708A3D2, 0x1E01F268,
113 0x6906C2FE, 0xF762575D, 0x806567CB, 0x196C3671, 0x6E6B06E7, 0xFED41B76, 0x89D32BE0,
114 0x10DA7A5A, 0x67DD4ACC, 0xF9B9DF6F, 0x8EBEEFF9, 0x17B7BE43, 0x60B08ED5, 0xD6D6A3E8,
115 0xA1D1937E, 0x38D8C2C4, 0x4FDFF252, 0xD1BB67F1, 0xA6BC5767, 0x3FB506DD, 0x48B2364B,
116 0xD80D2BDA, 0xAF0A1B4C, 0x36034AF6, 0x41047A60, 0xDF60EFC3, 0xA867DF55, 0x316E8EEF,
117 0x4669BE79, 0xCB61B38C, 0xBC66831A, 0x256FD2A0, 0x5268E236, 0xCC0C7795, 0xBB0B4703,
118 0x220216B9, 0x5505262F, 0xC5BA3BBE, 0xB2BD0B28, 0x2BB45A92, 0x5CB36A04, 0xC2D7FFA7,
119 0xB5D0CF31, 0x2CD99E8B, 0x5BDEAE1D, 0x9B64C2B0, 0xEC63F226, 0x756AA39C, 0x026D930A,
120 0x9C0906A9, 0xEB0E363F, 0x72076785, 0x05005713, 0x95BF4A82, 0xE2B87A14, 0x7BB12BAE,
121 0x0CB61B38, 0x92D28E9B, 0xE5D5BE0D, 0x7CDCEFB7, 0x0BDBDF21, 0x86D3D2D4, 0xF1D4E242,
122 0x68DDB3F8, 0x1FDA836E, 0x81BE16CD, 0xF6B9265B, 0x6FB077E1, 0x18B74777, 0x88085AE6,
123 0xFF0F6A70, 0x66063BCA, 0x11010B5C, 0x8F659EFF, 0xF862AE69, 0x616BFFD3, 0x166CCF45,
124 0xA00AE278, 0xD70DD2EE, 0x4E048354, 0x3903B3C2, 0xA7672661, 0xD06016F7, 0x4969474D,
125 0x3E6E77DB, 0xAED16A4A, 0xD9D65ADC, 0x40DF0B66, 0x37D83BF0, 0xA9BCAE53, 0xDEBB9EC5,
126 0x47B2CF7F, 0x30B5FFE9, 0xBDBDF21C, 0xCABAC28A, 0x53B39330, 0x24B4A3A6, 0xBAD03605,
127 0xCDD70693, 0x54DE5729, 0x23D967BF, 0xB3667A2E, 0xC4614AB8, 0x5D681B02, 0x2A6F2B94,
128 0xB40BBE37, 0xC30C8EA1, 0x5A05DF1B, 0x2D02EF8D
131 mz_uint32 crc32 = (mz_uint32)crc ^ 0xFFFFFFFF;
132 const mz_uint8 *pByte_buf = (
const mz_uint8 *)ptr;
136 crc32 = (crc32 >> 8) ^ s_crc_table[(crc32 ^ pByte_buf[0]) & 0xFF];
137 crc32 = (crc32 >> 8) ^ s_crc_table[(crc32 ^ pByte_buf[1]) & 0xFF];
138 crc32 = (crc32 >> 8) ^ s_crc_table[(crc32 ^ pByte_buf[2]) & 0xFF];
139 crc32 = (crc32 >> 8) ^ s_crc_table[(crc32 ^ pByte_buf[3]) & 0xFF];
146 crc32 = (crc32 >> 8) ^ s_crc_table[(crc32 ^ pByte_buf[0]) & 0xFF];
160void *miniz_def_alloc_func(
void *opaque,
size_t items,
size_t size)
162 (void)opaque, (
void)items, (void)size;
163 return MZ_MALLOC(items * size);
165void miniz_def_free_func(
void *opaque,
void *address)
167 (void)opaque, (
void)address;
170void *miniz_def_realloc_func(
void *opaque,
void *address,
size_t items,
size_t size)
172 (void)opaque, (
void)address, (void)items, (
void)size;
173 return MZ_REALLOC(address, items * size);
176const char *mz_version(
void)
181#ifndef MINIZ_NO_ZLIB_APIS
183int mz_deflateInit(
mz_streamp pStream,
int level)
185 return mz_deflateInit2(pStream, level, MZ_DEFLATED, MZ_DEFAULT_WINDOW_BITS, 9, MZ_DEFAULT_STRATEGY);
188int mz_deflateInit2(
mz_streamp pStream,
int level,
int method,
int window_bits,
int mem_level,
int strategy)
191 mz_uint comp_flags = TDEFL_COMPUTE_ADLER32 | tdefl_create_comp_flags_from_zip_params(level, window_bits, strategy);
194 return MZ_STREAM_ERROR;
195 if ((method != MZ_DEFLATED) || ((mem_level < 1) || (mem_level > 9)) || ((window_bits != MZ_DEFAULT_WINDOW_BITS) && (-window_bits != MZ_DEFAULT_WINDOW_BITS)))
196 return MZ_PARAM_ERROR;
198 pStream->data_type = 0;
199 pStream->adler = MZ_ADLER32_INIT;
201 pStream->reserved = 0;
202 pStream->total_in = 0;
203 pStream->total_out = 0;
204 if (!pStream->zalloc)
205 pStream->zalloc = miniz_def_alloc_func;
207 pStream->zfree = miniz_def_free_func;
213 pStream->state = (
struct mz_internal_state *)pComp;
215 if (tdefl_init(pComp, NULL, NULL, comp_flags) != TDEFL_STATUS_OKAY)
217 mz_deflateEnd(pStream);
218 return MZ_PARAM_ERROR;
226 if ((!pStream) || (!pStream->state) || (!pStream->zalloc) || (!pStream->zfree))
227 return MZ_STREAM_ERROR;
228 pStream->total_in = pStream->total_out = 0;
235 size_t in_bytes, out_bytes;
236 mz_ulong orig_total_in, orig_total_out;
237 int mz_status = MZ_OK;
239 if ((!pStream) || (!pStream->state) || (flush < 0) || (flush > MZ_FINISH) || (!pStream->next_out))
240 return MZ_STREAM_ERROR;
241 if (!pStream->avail_out)
244 if (flush == MZ_PARTIAL_FLUSH)
245 flush = MZ_SYNC_FLUSH;
247 if (((
tdefl_compressor *)pStream->state)->m_prev_return_status == TDEFL_STATUS_DONE)
248 return (flush == MZ_FINISH) ? MZ_STREAM_END : MZ_BUF_ERROR;
250 orig_total_in = pStream->total_in;
251 orig_total_out = pStream->total_out;
254 tdefl_status defl_status;
255 in_bytes = pStream->avail_in;
256 out_bytes = pStream->avail_out;
258 defl_status = tdefl_compress((
tdefl_compressor *)pStream->state, pStream->next_in, &in_bytes, pStream->next_out, &out_bytes, (tdefl_flush)flush);
259 pStream->next_in += (mz_uint)in_bytes;
260 pStream->avail_in -= (mz_uint)in_bytes;
261 pStream->total_in += (mz_uint)in_bytes;
264 pStream->next_out += (mz_uint)out_bytes;
265 pStream->avail_out -= (mz_uint)out_bytes;
266 pStream->total_out += (mz_uint)out_bytes;
270 mz_status = MZ_STREAM_ERROR;
273 else if (defl_status == TDEFL_STATUS_DONE)
275 mz_status = MZ_STREAM_END;
278 else if (!pStream->avail_out)
280 else if ((!pStream->avail_in) && (
flush != MZ_FINISH))
282 if ((flush) || (pStream->total_in != orig_total_in) || (pStream->total_out != orig_total_out))
294 return MZ_STREAM_ERROR;
297 pStream->zfree(pStream->opaque, pStream->state);
298 pStream->state = NULL;
303mz_ulong mz_deflateBound(
mz_streamp pStream, mz_ulong source_len)
307 return MZ_MAX(128 + (source_len * 110) / 100, 128 + source_len + ((source_len / (31 * 1024)) + 1) * 5);
310int mz_compress2(
unsigned char *pDest, mz_ulong *pDest_len,
const unsigned char *pSource, mz_ulong source_len,
int level)
314 memset(&stream, 0,
sizeof(stream));
317 if ((source_len | *pDest_len) > 0xFFFFFFFFU)
318 return MZ_PARAM_ERROR;
320 stream.next_in = pSource;
321 stream.avail_in = (mz_uint32)source_len;
322 stream.next_out = pDest;
323 stream.avail_out = (mz_uint32)*pDest_len;
325 status = mz_deflateInit(&stream, level);
329 status = mz_deflate(&stream, MZ_FINISH);
330 if (status != MZ_STREAM_END)
332 mz_deflateEnd(&stream);
333 return (status == MZ_OK) ? MZ_BUF_ERROR : status;
336 *pDest_len = stream.total_out;
337 return mz_deflateEnd(&stream);
340int mz_compress(
unsigned char *pDest, mz_ulong *pDest_len,
const unsigned char *pSource, mz_ulong source_len)
342 return mz_compress2(pDest, pDest_len, pSource, source_len, MZ_DEFAULT_COMPRESSION);
345mz_ulong mz_compressBound(mz_ulong source_len)
347 return mz_deflateBound(NULL, source_len);
353 mz_uint m_dict_ofs, m_dict_avail, m_first_call, m_has_flushed;
355 mz_uint8 m_dict[TINFL_LZ_DICT_SIZE];
356 tinfl_status m_last_status;
359int mz_inflateInit2(
mz_streamp pStream,
int window_bits)
363 return MZ_STREAM_ERROR;
364 if ((window_bits != MZ_DEFAULT_WINDOW_BITS) && (-window_bits != MZ_DEFAULT_WINDOW_BITS))
365 return MZ_PARAM_ERROR;
367 pStream->data_type = 0;
370 pStream->total_in = 0;
371 pStream->total_out = 0;
372 pStream->reserved = 0;
373 if (!pStream->zalloc)
374 pStream->zalloc = miniz_def_alloc_func;
376 pStream->zfree = miniz_def_free_func;
382 pStream->state = (
struct mz_internal_state *)pDecomp;
384 tinfl_init(&pDecomp->m_decomp);
385 pDecomp->m_dict_ofs = 0;
386 pDecomp->m_dict_avail = 0;
387 pDecomp->m_last_status = TINFL_STATUS_NEEDS_MORE_INPUT;
388 pDecomp->m_first_call = 1;
389 pDecomp->m_has_flushed = 0;
390 pDecomp->m_window_bits = window_bits;
397 return mz_inflateInit2(pStream, MZ_DEFAULT_WINDOW_BITS);
404 return MZ_STREAM_ERROR;
406 pStream->data_type = 0;
409 pStream->total_in = 0;
410 pStream->total_out = 0;
411 pStream->reserved = 0;
415 tinfl_init(&pDecomp->m_decomp);
416 pDecomp->m_dict_ofs = 0;
417 pDecomp->m_dict_avail = 0;
418 pDecomp->m_last_status = TINFL_STATUS_NEEDS_MORE_INPUT;
419 pDecomp->m_first_call = 1;
420 pDecomp->m_has_flushed = 0;
429 mz_uint n, first_call, decomp_flags = TINFL_FLAG_COMPUTE_ADLER32;
430 size_t in_bytes, out_bytes, orig_avail_in;
433 if ((!pStream) || (!pStream->state))
434 return MZ_STREAM_ERROR;
435 if (flush == MZ_PARTIAL_FLUSH)
436 flush = MZ_SYNC_FLUSH;
437 if ((flush) && (flush != MZ_SYNC_FLUSH) && (flush != MZ_FINISH))
438 return MZ_STREAM_ERROR;
441 if (pState->m_window_bits > 0)
442 decomp_flags |= TINFL_FLAG_PARSE_ZLIB_HEADER;
443 orig_avail_in = pStream->avail_in;
445 first_call = pState->m_first_call;
446 pState->m_first_call = 0;
447 if (pState->m_last_status < 0)
448 return MZ_DATA_ERROR;
450 if (pState->m_has_flushed && (flush != MZ_FINISH))
451 return MZ_STREAM_ERROR;
452 pState->m_has_flushed |= (
flush == MZ_FINISH);
454 if ((flush == MZ_FINISH) && (first_call))
457 decomp_flags |= TINFL_FLAG_USING_NON_WRAPPING_OUTPUT_BUF;
458 in_bytes = pStream->avail_in;
459 out_bytes = pStream->avail_out;
460 status = tinfl_decompress(&pState->m_decomp, pStream->next_in, &in_bytes, pStream->next_out, pStream->next_out, &out_bytes, decomp_flags);
461 pState->m_last_status = status;
462 pStream->next_in += (mz_uint)in_bytes;
463 pStream->avail_in -= (mz_uint)in_bytes;
464 pStream->total_in += (mz_uint)in_bytes;
465 pStream->adler = tinfl_get_adler32(&pState->m_decomp);
466 pStream->next_out += (mz_uint)out_bytes;
467 pStream->avail_out -= (mz_uint)out_bytes;
468 pStream->total_out += (mz_uint)out_bytes;
471 return MZ_DATA_ERROR;
472 else if (status != TINFL_STATUS_DONE)
474 pState->m_last_status = TINFL_STATUS_FAILED;
477 return MZ_STREAM_END;
480 if (flush != MZ_FINISH)
481 decomp_flags |= TINFL_FLAG_HAS_MORE_INPUT;
483 if (pState->m_dict_avail)
485 n = MZ_MIN(pState->m_dict_avail, pStream->avail_out);
486 memcpy(pStream->next_out, pState->m_dict + pState->m_dict_ofs, n);
487 pStream->next_out += n;
488 pStream->avail_out -= n;
489 pStream->total_out += n;
490 pState->m_dict_avail -= n;
491 pState->m_dict_ofs = (pState->m_dict_ofs + n) & (TINFL_LZ_DICT_SIZE - 1);
492 return ((pState->m_last_status == TINFL_STATUS_DONE) && (!pState->m_dict_avail)) ? MZ_STREAM_END : MZ_OK;
497 in_bytes = pStream->avail_in;
498 out_bytes = TINFL_LZ_DICT_SIZE - pState->m_dict_ofs;
500 status = tinfl_decompress(&pState->m_decomp, pStream->next_in, &in_bytes, pState->m_dict, pState->m_dict + pState->m_dict_ofs, &out_bytes, decomp_flags);
501 pState->m_last_status = status;
503 pStream->next_in += (mz_uint)in_bytes;
504 pStream->avail_in -= (mz_uint)in_bytes;
505 pStream->total_in += (mz_uint)in_bytes;
506 pStream->adler = tinfl_get_adler32(&pState->m_decomp);
508 pState->m_dict_avail = (mz_uint)out_bytes;
510 n = MZ_MIN(pState->m_dict_avail, pStream->avail_out);
511 memcpy(pStream->next_out, pState->m_dict + pState->m_dict_ofs, n);
512 pStream->next_out += n;
513 pStream->avail_out -= n;
514 pStream->total_out += n;
515 pState->m_dict_avail -= n;
516 pState->m_dict_ofs = (pState->m_dict_ofs + n) & (TINFL_LZ_DICT_SIZE - 1);
519 return MZ_DATA_ERROR;
520 else if ((status == TINFL_STATUS_NEEDS_MORE_INPUT) && (!orig_avail_in))
522 else if (flush == MZ_FINISH)
525 if (status == TINFL_STATUS_DONE)
526 return pState->m_dict_avail ? MZ_BUF_ERROR : MZ_STREAM_END;
528 else if (!pStream->avail_out)
531 else if ((status == TINFL_STATUS_DONE) || (!pStream->avail_in) || (!pStream->avail_out) || (pState->m_dict_avail))
535 return ((status == TINFL_STATUS_DONE) && (!pState->m_dict_avail)) ? MZ_STREAM_END : MZ_OK;
541 return MZ_STREAM_ERROR;
544 pStream->zfree(pStream->opaque, pStream->state);
545 pStream->state = NULL;
550int mz_uncompress(
unsigned char *pDest, mz_ulong *pDest_len,
const unsigned char *pSource, mz_ulong source_len)
554 memset(&stream, 0,
sizeof(stream));
557 if ((source_len | *pDest_len) > 0xFFFFFFFFU)
558 return MZ_PARAM_ERROR;
560 stream.next_in = pSource;
561 stream.avail_in = (mz_uint32)source_len;
562 stream.next_out = pDest;
563 stream.avail_out = (mz_uint32)*pDest_len;
565 status = mz_inflateInit(&stream);
569 status = mz_inflate(&stream, MZ_FINISH);
570 if (status != MZ_STREAM_END)
572 mz_inflateEnd(&stream);
573 return ((status == MZ_BUF_ERROR) && (!stream.avail_in)) ? MZ_DATA_ERROR : status;
575 *pDest_len = stream.total_out;
577 return mz_inflateEnd(&stream);
580const char *mz_error(
int err)
588 { MZ_OK,
"" }, { MZ_STREAM_END,
"stream end" }, { MZ_NEED_DICT,
"need dictionary" }, { MZ_ERRNO,
"file error" }, { MZ_STREAM_ERROR,
"stream error" }, { MZ_DATA_ERROR,
"data error" }, { MZ_MEM_ERROR,
"out of memory" }, { MZ_BUF_ERROR,
"buf error" }, { MZ_VERSION_ERROR,
"version error" }, { MZ_PARAM_ERROR,
"parameter error" }
591 for (i = 0; i <
sizeof(s_error_descs) /
sizeof(s_error_descs[0]); ++i)
592 if (s_error_descs[i].m_err == err)
593 return s_error_descs[i].m_pDesc;
665static const mz_uint16 s_tdefl_len_sym[256] =
667 257, 258, 259, 260, 261, 262, 263, 264, 265, 265, 266, 266, 267, 267, 268, 268, 269, 269, 269, 269, 270, 270, 270, 270, 271, 271, 271, 271, 272, 272, 272, 272,
668 273, 273, 273, 273, 273, 273, 273, 273, 274, 274, 274, 274, 274, 274, 274, 274, 275, 275, 275, 275, 275, 275, 275, 275, 276, 276, 276, 276, 276, 276, 276, 276,
669 277, 277, 277, 277, 277, 277, 277, 277, 277, 277, 277, 277, 277, 277, 277, 277, 278, 278, 278, 278, 278, 278, 278, 278, 278, 278, 278, 278, 278, 278, 278, 278,
670 279, 279, 279, 279, 279, 279, 279, 279, 279, 279, 279, 279, 279, 279, 279, 279, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280,
671 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281, 281,
672 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282, 282,
673 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283, 283,
674 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 284, 285
677static const mz_uint8 s_tdefl_len_extra[256] =
679 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
680 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
681 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
682 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 0
685static const mz_uint8 s_tdefl_small_dist_sym[512] =
687 0, 1, 2, 3, 4, 4, 5, 5, 6, 6, 6, 6, 7, 7, 7, 7, 8, 8, 8, 8, 8, 8, 8, 8, 9, 9, 9, 9, 9, 9, 9, 9, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 11, 11, 11, 11, 11, 11,
688 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 13,
689 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14,
690 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14,
691 14, 14, 14, 14, 14, 14, 14, 14, 14, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15,
692 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16,
693 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16,
694 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16,
695 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17,
696 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17,
697 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17,
698 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17, 17
701static const mz_uint8 s_tdefl_small_dist_extra[512] =
703 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 5, 5, 5, 5, 5, 5, 5, 5,
704 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
705 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
706 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
707 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
708 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
709 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
710 7, 7, 7, 7, 7, 7, 7, 7
713static const mz_uint8 s_tdefl_large_dist_sym[128] =
715 0, 0, 18, 19, 20, 20, 21, 21, 22, 22, 22, 22, 23, 23, 23, 23, 24, 24, 24, 24, 24, 24, 24, 24, 25, 25, 25, 25, 25, 25, 25, 25, 26, 26, 26, 26, 26, 26, 26, 26, 26, 26, 26, 26,
716 26, 26, 26, 26, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28,
717 28, 28, 28, 28, 28, 28, 28, 28, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29, 29
720static const mz_uint8 s_tdefl_large_dist_extra[128] =
722 0, 0, 8, 8, 9, 9, 9, 9, 10, 10, 10, 10, 10, 10, 10, 10, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12,
723 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13,
724 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13
730 mz_uint16 m_key, m_sym_index;
734 mz_uint32 total_passes = 2, pass_shift, pass, i, hist[256 * 2];
737 for (i = 0; i < num_syms; i++)
739 mz_uint freq = pSyms0[i].m_key;
741 hist[256 + ((freq >> 8) & 0xFF)]++;
743 while ((total_passes > 1) && (num_syms == hist[(total_passes - 1) * 256]))
745 for (pass_shift = 0, pass = 0; pass < total_passes; pass++, pass_shift += 8)
747 const mz_uint32 *pHist = &hist[pass << 8];
748 mz_uint offsets[256], cur_ofs = 0;
749 for (i = 0; i < 256; i++)
751 offsets[i] = cur_ofs;
754 for (i = 0; i < num_syms; i++)
755 pNew_syms[offsets[(pCur_syms[i].m_key >> pass_shift) & 0xFF]++] = pCur_syms[i];
758 pCur_syms = pNew_syms;
766static void tdefl_calculate_minimum_redundancy(
tdefl_sym_freq *A,
int n)
768 int root, leaf, next, avbl, used, dpth;
776 A[0].m_key += A[1].m_key;
779 for (next = 1; next < n - 1; next++)
781 if (leaf >= n || A[root].m_key < A[leaf].m_key)
783 A[next].m_key = A[root].m_key;
784 A[root++].m_key = (mz_uint16)next;
787 A[next].m_key = A[leaf++].m_key;
788 if (leaf >= n || (root < next && A[root].m_key < A[leaf].m_key))
790 A[next].m_key = (mz_uint16)(A[next].m_key + A[root].m_key);
791 A[root++].m_key = (mz_uint16)next;
794 A[next].m_key = (mz_uint16)(A[next].m_key + A[leaf++].m_key);
797 for (next = n - 3; next >= 0; next--)
798 A[next].m_key = A[A[next].m_key].m_key + 1;
805 while (root >= 0 && (
int)A[root].m_key == dpth)
812 A[next--].m_key = (mz_uint16)(dpth);
824 TDEFL_MAX_SUPPORTED_HUFF_CODESIZE = 32
826static void tdefl_huffman_enforce_max_code_size(
int *pNum_codes,
int code_list_len,
int max_code_size)
830 if (code_list_len <= 1)
832 for (i = max_code_size + 1; i <= TDEFL_MAX_SUPPORTED_HUFF_CODESIZE; i++)
833 pNum_codes[max_code_size] += pNum_codes[i];
834 for (i = max_code_size; i > 0; i--)
835 total += (((mz_uint32)pNum_codes[i]) << (max_code_size - i));
836 while (total != (1UL << max_code_size))
838 pNum_codes[max_code_size]--;
839 for (i = max_code_size - 1; i > 0; i--)
843 pNum_codes[i + 1] += 2;
850static void tdefl_optimize_huffman_table(
tdefl_compressor *d,
int table_num,
int table_len,
int code_size_limit,
int static_table)
852 int i, j, l, num_codes[1 + TDEFL_MAX_SUPPORTED_HUFF_CODESIZE];
853 mz_uint next_code[TDEFL_MAX_SUPPORTED_HUFF_CODESIZE + 1];
854 MZ_CLEAR_OBJ(num_codes);
857 for (i = 0; i < table_len; i++)
858 num_codes[d->m_huff_code_sizes[table_num][i]]++;
862 tdefl_sym_freq syms0[TDEFL_MAX_HUFF_SYMBOLS], syms1[TDEFL_MAX_HUFF_SYMBOLS], *pSyms;
863 int num_used_syms = 0;
864 const mz_uint16 *pSym_count = &d->m_huff_count[table_num][0];
865 for (i = 0; i < table_len; i++)
868 syms0[num_used_syms].m_key = (mz_uint16)pSym_count[i];
869 syms0[num_used_syms++].m_sym_index = (mz_uint16)i;
872 pSyms = tdefl_radix_sort_syms(num_used_syms, syms0, syms1);
873 tdefl_calculate_minimum_redundancy(pSyms, num_used_syms);
875 for (i = 0; i < num_used_syms; i++)
876 num_codes[pSyms[i].m_key]++;
878 tdefl_huffman_enforce_max_code_size(num_codes, num_used_syms, code_size_limit);
880 MZ_CLEAR_OBJ(d->m_huff_code_sizes[table_num]);
881 MZ_CLEAR_OBJ(d->m_huff_codes[table_num]);
882 for (i = 1, j = num_used_syms; i <= code_size_limit; i++)
883 for (l = num_codes[i]; l > 0; l--)
884 d->m_huff_code_sizes[table_num][pSyms[--j].m_sym_index] = (mz_uint8)(i);
888 for (j = 0, i = 2; i <= code_size_limit; i++)
889 next_code[i] = j = ((j + num_codes[i - 1]) << 1);
891 for (i = 0; i < table_len; i++)
893 mz_uint rev_code = 0, code, code_size;
894 if ((code_size = d->m_huff_code_sizes[table_num][i]) == 0)
896 code = next_code[code_size]++;
897 for (l = code_size; l > 0; l--, code >>= 1)
898 rev_code = (rev_code << 1) | (code & 1);
899 d->m_huff_codes[table_num][i] = (mz_uint16)rev_code;
903#define TDEFL_PUT_BITS(b, l) \
908 MZ_ASSERT(bits <= ((1U << len) - 1U)); \
909 d->m_bit_buffer |= (bits << d->m_bits_in); \
910 d->m_bits_in += len; \
911 while (d->m_bits_in >= 8) \
913 if (d->m_pOutput_buf < d->m_pOutput_buf_end) \
914 *d->m_pOutput_buf++ = (mz_uint8)(d->m_bit_buffer); \
915 d->m_bit_buffer >>= 8; \
921#define TDEFL_RLE_PREV_CODE_SIZE() \
923 if (rle_repeat_count) \
925 if (rle_repeat_count < 3) \
927 d->m_huff_count[2][prev_code_size] = (mz_uint16)(d->m_huff_count[2][prev_code_size] + rle_repeat_count); \
928 while (rle_repeat_count--) \
929 packed_code_sizes[num_packed_code_sizes++] = prev_code_size; \
933 d->m_huff_count[2][16] = (mz_uint16)(d->m_huff_count[2][16] + 1); \
934 packed_code_sizes[num_packed_code_sizes++] = 16; \
935 packed_code_sizes[num_packed_code_sizes++] = (mz_uint8)(rle_repeat_count - 3); \
937 rle_repeat_count = 0; \
941#define TDEFL_RLE_ZERO_CODE_SIZE() \
945 if (rle_z_count < 3) \
947 d->m_huff_count[2][0] = (mz_uint16)(d->m_huff_count[2][0] + rle_z_count); \
948 while (rle_z_count--) \
949 packed_code_sizes[num_packed_code_sizes++] = 0; \
951 else if (rle_z_count <= 10) \
953 d->m_huff_count[2][17] = (mz_uint16)(d->m_huff_count[2][17] + 1); \
954 packed_code_sizes[num_packed_code_sizes++] = 17; \
955 packed_code_sizes[num_packed_code_sizes++] = (mz_uint8)(rle_z_count - 3); \
959 d->m_huff_count[2][18] = (mz_uint16)(d->m_huff_count[2][18] + 1); \
960 packed_code_sizes[num_packed_code_sizes++] = 18; \
961 packed_code_sizes[num_packed_code_sizes++] = (mz_uint8)(rle_z_count - 11); \
967static mz_uint8 s_tdefl_packed_code_size_syms_swizzle[] = { 16, 17, 18, 0, 8, 7, 9, 6, 10, 5, 11, 4, 12, 3, 13, 2, 14, 1, 15 };
971 int num_lit_codes, num_dist_codes, num_bit_lengths;
972 mz_uint i, total_code_sizes_to_pack, num_packed_code_sizes, rle_z_count, rle_repeat_count, packed_code_sizes_index;
973 mz_uint8 code_sizes_to_pack[TDEFL_MAX_HUFF_SYMBOLS_0 + TDEFL_MAX_HUFF_SYMBOLS_1], packed_code_sizes[TDEFL_MAX_HUFF_SYMBOLS_0 + TDEFL_MAX_HUFF_SYMBOLS_1], prev_code_size = 0xFF;
975 d->m_huff_count[0][256] = 1;
977 tdefl_optimize_huffman_table(d, 0, TDEFL_MAX_HUFF_SYMBOLS_0, 15, MZ_FALSE);
978 tdefl_optimize_huffman_table(d, 1, TDEFL_MAX_HUFF_SYMBOLS_1, 15, MZ_FALSE);
980 for (num_lit_codes = 286; num_lit_codes > 257; num_lit_codes--)
981 if (d->m_huff_code_sizes[0][num_lit_codes - 1])
983 for (num_dist_codes = 30; num_dist_codes > 1; num_dist_codes--)
984 if (d->m_huff_code_sizes[1][num_dist_codes - 1])
987 memcpy(code_sizes_to_pack, &d->m_huff_code_sizes[0][0], num_lit_codes);
988 memcpy(code_sizes_to_pack + num_lit_codes, &d->m_huff_code_sizes[1][0], num_dist_codes);
989 total_code_sizes_to_pack = num_lit_codes + num_dist_codes;
990 num_packed_code_sizes = 0;
992 rle_repeat_count = 0;
994 memset(&d->m_huff_count[2][0], 0,
sizeof(d->m_huff_count[2][0]) * TDEFL_MAX_HUFF_SYMBOLS_2);
995 for (i = 0; i < total_code_sizes_to_pack; i++)
997 mz_uint8 code_size = code_sizes_to_pack[i];
1000 TDEFL_RLE_PREV_CODE_SIZE();
1001 if (++rle_z_count == 138)
1003 TDEFL_RLE_ZERO_CODE_SIZE();
1008 TDEFL_RLE_ZERO_CODE_SIZE();
1009 if (code_size != prev_code_size)
1011 TDEFL_RLE_PREV_CODE_SIZE();
1012 d->m_huff_count[2][code_size] = (mz_uint16)(d->m_huff_count[2][code_size] + 1);
1013 packed_code_sizes[num_packed_code_sizes++] = code_size;
1015 else if (++rle_repeat_count == 6)
1017 TDEFL_RLE_PREV_CODE_SIZE();
1020 prev_code_size = code_size;
1022 if (rle_repeat_count)
1024 TDEFL_RLE_PREV_CODE_SIZE();
1028 TDEFL_RLE_ZERO_CODE_SIZE();
1031 tdefl_optimize_huffman_table(d, 2, TDEFL_MAX_HUFF_SYMBOLS_2, 7, MZ_FALSE);
1033 TDEFL_PUT_BITS(2, 2);
1035 TDEFL_PUT_BITS(num_lit_codes - 257, 5);
1036 TDEFL_PUT_BITS(num_dist_codes - 1, 5);
1038 for (num_bit_lengths = 18; num_bit_lengths >= 0; num_bit_lengths--)
1039 if (d->m_huff_code_sizes[2][s_tdefl_packed_code_size_syms_swizzle[num_bit_lengths]])
1041 num_bit_lengths = MZ_MAX(4, (num_bit_lengths + 1));
1042 TDEFL_PUT_BITS(num_bit_lengths - 4, 4);
1043 for (i = 0; (int)i < num_bit_lengths; i++)
1044 TDEFL_PUT_BITS(d->m_huff_code_sizes[2][s_tdefl_packed_code_size_syms_swizzle[i]], 3);
1046 for (packed_code_sizes_index = 0; packed_code_sizes_index < num_packed_code_sizes;)
1048 mz_uint code = packed_code_sizes[packed_code_sizes_index++];
1049 MZ_ASSERT(code < TDEFL_MAX_HUFF_SYMBOLS_2);
1050 TDEFL_PUT_BITS(d->m_huff_codes[2][code], d->m_huff_code_sizes[2][code]);
1052 TDEFL_PUT_BITS(packed_code_sizes[packed_code_sizes_index++],
"\02\03\07"[code - 16]);
1059 mz_uint8 *p = &d->m_huff_code_sizes[0][0];
1061 for (i = 0; i <= 143; ++i)
1063 for (; i <= 255; ++i)
1065 for (; i <= 279; ++i)
1067 for (; i <= 287; ++i)
1070 memset(d->m_huff_code_sizes[1], 5, 32);
1072 tdefl_optimize_huffman_table(d, 0, 288, 15, MZ_TRUE);
1073 tdefl_optimize_huffman_table(d, 1, 32, 15, MZ_TRUE);
1075 TDEFL_PUT_BITS(1, 2);
1078static const mz_uint mz_bitmasks[17] = { 0x0000, 0x0001, 0x0003, 0x0007, 0x000F, 0x001F, 0x003F, 0x007F, 0x00FF, 0x01FF, 0x03FF, 0x07FF, 0x0FFF, 0x1FFF, 0x3FFF, 0x7FFF, 0xFFFF };
1080#if MINIZ_USE_UNALIGNED_LOADS_AND_STORES && MINIZ_LITTLE_ENDIAN && MINIZ_HAS_64BIT_REGISTERS
1084 mz_uint8 *pLZ_codes;
1085 mz_uint8 *pOutput_buf = d->m_pOutput_buf;
1086 mz_uint8 *pLZ_code_buf_end = d->m_pLZ_code_buf;
1087 mz_uint64 bit_buffer = d->m_bit_buffer;
1088 mz_uint bits_in = d->m_bits_in;
1090#define TDEFL_PUT_BITS_FAST(b, l) \
1092 bit_buffer |= (((mz_uint64)(b)) << bits_in); \
1097 for (pLZ_codes = d->m_lz_code_buf; pLZ_codes < pLZ_code_buf_end; flags >>= 1)
1100 flags = *pLZ_codes++ | 0x100;
1104 mz_uint s0, s1, n0, n1, sym, num_extra_bits;
1105 mz_uint match_len = pLZ_codes[0], match_dist = *(
const mz_uint16 *)(pLZ_codes + 1);
1108 MZ_ASSERT(d->m_huff_code_sizes[0][s_tdefl_len_sym[match_len]]);
1109 TDEFL_PUT_BITS_FAST(d->m_huff_codes[0][s_tdefl_len_sym[match_len]], d->m_huff_code_sizes[0][s_tdefl_len_sym[match_len]]);
1110 TDEFL_PUT_BITS_FAST(match_len & mz_bitmasks[s_tdefl_len_extra[match_len]], s_tdefl_len_extra[match_len]);
1113 s0 = s_tdefl_small_dist_sym[match_dist & 511];
1114 n0 = s_tdefl_small_dist_extra[match_dist & 511];
1115 s1 = s_tdefl_large_dist_sym[match_dist >> 8];
1116 n1 = s_tdefl_large_dist_extra[match_dist >> 8];
1117 sym = (match_dist < 512) ? s0 : s1;
1118 num_extra_bits = (match_dist < 512) ? n0 : n1;
1120 MZ_ASSERT(d->m_huff_code_sizes[1][sym]);
1121 TDEFL_PUT_BITS_FAST(d->m_huff_codes[1][sym], d->m_huff_code_sizes[1][sym]);
1122 TDEFL_PUT_BITS_FAST(match_dist & mz_bitmasks[num_extra_bits], num_extra_bits);
1126 mz_uint lit = *pLZ_codes++;
1127 MZ_ASSERT(d->m_huff_code_sizes[0][lit]);
1128 TDEFL_PUT_BITS_FAST(d->m_huff_codes[0][lit], d->m_huff_code_sizes[0][lit]);
1130 if (((flags & 2) == 0) && (pLZ_codes < pLZ_code_buf_end))
1134 MZ_ASSERT(d->m_huff_code_sizes[0][lit]);
1135 TDEFL_PUT_BITS_FAST(d->m_huff_codes[0][lit], d->m_huff_code_sizes[0][lit]);
1137 if (((flags & 2) == 0) && (pLZ_codes < pLZ_code_buf_end))
1141 MZ_ASSERT(d->m_huff_code_sizes[0][lit]);
1142 TDEFL_PUT_BITS_FAST(d->m_huff_codes[0][lit], d->m_huff_code_sizes[0][lit]);
1147 if (pOutput_buf >= d->m_pOutput_buf_end)
1150 *(mz_uint64 *)pOutput_buf = bit_buffer;
1151 pOutput_buf += (bits_in >> 3);
1152 bit_buffer >>= (bits_in & ~7);
1156#undef TDEFL_PUT_BITS_FAST
1158 d->m_pOutput_buf = pOutput_buf;
1160 d->m_bit_buffer = 0;
1164 mz_uint32 n = MZ_MIN(bits_in, 16);
1165 TDEFL_PUT_BITS((mz_uint)bit_buffer & mz_bitmasks[n], n);
1170 TDEFL_PUT_BITS(d->m_huff_codes[0][256], d->m_huff_code_sizes[0][256]);
1172 return (d->m_pOutput_buf < d->m_pOutput_buf_end);
1178 mz_uint8 *pLZ_codes;
1181 for (pLZ_codes = d->m_lz_code_buf; pLZ_codes < d->m_pLZ_code_buf; flags >>= 1)
1184 flags = *pLZ_codes++ | 0x100;
1187 mz_uint sym, num_extra_bits;
1188 mz_uint match_len = pLZ_codes[0], match_dist = (pLZ_codes[1] | (pLZ_codes[2] << 8));
1191 MZ_ASSERT(d->m_huff_code_sizes[0][s_tdefl_len_sym[match_len]]);
1192 TDEFL_PUT_BITS(d->m_huff_codes[0][s_tdefl_len_sym[match_len]], d->m_huff_code_sizes[0][s_tdefl_len_sym[match_len]]);
1193 TDEFL_PUT_BITS(match_len & mz_bitmasks[s_tdefl_len_extra[match_len]], s_tdefl_len_extra[match_len]);
1195 if (match_dist < 512)
1197 sym = s_tdefl_small_dist_sym[match_dist];
1198 num_extra_bits = s_tdefl_small_dist_extra[match_dist];
1202 sym = s_tdefl_large_dist_sym[match_dist >> 8];
1203 num_extra_bits = s_tdefl_large_dist_extra[match_dist >> 8];
1205 MZ_ASSERT(d->m_huff_code_sizes[1][sym]);
1206 TDEFL_PUT_BITS(d->m_huff_codes[1][sym], d->m_huff_code_sizes[1][sym]);
1207 TDEFL_PUT_BITS(match_dist & mz_bitmasks[num_extra_bits], num_extra_bits);
1211 mz_uint lit = *pLZ_codes++;
1212 MZ_ASSERT(d->m_huff_code_sizes[0][lit]);
1213 TDEFL_PUT_BITS(d->m_huff_codes[0][lit], d->m_huff_code_sizes[0][lit]);
1217 TDEFL_PUT_BITS(d->m_huff_codes[0][256], d->m_huff_code_sizes[0][256]);
1219 return (d->m_pOutput_buf < d->m_pOutput_buf_end);
1223static mz_bool tdefl_compress_block(
tdefl_compressor *d, mz_bool static_block)
1226 tdefl_start_static_block(d);
1228 tdefl_start_dynamic_block(d);
1229 return tdefl_compress_lz_codes(d);
1234 mz_uint saved_bit_buf, saved_bits_in;
1235 mz_uint8 *pSaved_output_buf;
1236 mz_bool comp_block_succeeded = MZ_FALSE;
1237 int n, use_raw_block = ((d->m_flags & TDEFL_FORCE_ALL_RAW_BLOCKS) != 0) && (d->m_lookahead_pos - d->m_lz_code_buf_dict_pos) <= d->m_dict_size;
1238 mz_uint8 *pOutput_buf_start = ((d->m_pPut_buf_func == NULL) && ((*d->m_pOut_buf_size - d->m_out_buf_ofs) >= TDEFL_OUT_BUF_SIZE)) ? ((mz_uint8 *)d->m_pOut_buf + d->m_out_buf_ofs) : d->m_output_buf;
1240 d->m_pOutput_buf = pOutput_buf_start;
1241 d->m_pOutput_buf_end = d->m_pOutput_buf + TDEFL_OUT_BUF_SIZE - 16;
1243 MZ_ASSERT(!d->m_output_flush_remaining);
1244 d->m_output_flush_ofs = 0;
1245 d->m_output_flush_remaining = 0;
1247 *d->m_pLZ_flags = (mz_uint8)(*d->m_pLZ_flags >> d->m_num_flags_left);
1248 d->m_pLZ_code_buf -= (d->m_num_flags_left == 8);
1250 if ((d->m_flags & TDEFL_WRITE_ZLIB_HEADER) && (!d->m_block_index))
1252 TDEFL_PUT_BITS(0x78, 8);
1253 TDEFL_PUT_BITS(0x01, 8);
1256 TDEFL_PUT_BITS(flush == TDEFL_FINISH, 1);
1258 pSaved_output_buf = d->m_pOutput_buf;
1259 saved_bit_buf = d->m_bit_buffer;
1260 saved_bits_in = d->m_bits_in;
1263 comp_block_succeeded = tdefl_compress_block(d, (d->m_flags & TDEFL_FORCE_ALL_STATIC_BLOCKS) || (d->m_total_lz_bytes < 48));
1266 if (((use_raw_block) || ((d->m_total_lz_bytes) && ((d->m_pOutput_buf - pSaved_output_buf + 1U) >= d->m_total_lz_bytes))) &&
1267 ((d->m_lookahead_pos - d->m_lz_code_buf_dict_pos) <= d->m_dict_size))
1270 d->m_pOutput_buf = pSaved_output_buf;
1271 d->m_bit_buffer = saved_bit_buf, d->m_bits_in = saved_bits_in;
1272 TDEFL_PUT_BITS(0, 2);
1275 TDEFL_PUT_BITS(0, 8 - d->m_bits_in);
1277 for (i = 2; i; --i, d->m_total_lz_bytes ^= 0xFFFF)
1279 TDEFL_PUT_BITS(d->m_total_lz_bytes & 0xFFFF, 16);
1281 for (i = 0; i < d->m_total_lz_bytes; ++i)
1283 TDEFL_PUT_BITS(d->m_dict[(d->m_lz_code_buf_dict_pos + i) & TDEFL_LZ_DICT_SIZE_MASK], 8);
1287 else if (!comp_block_succeeded)
1289 d->m_pOutput_buf = pSaved_output_buf;
1290 d->m_bit_buffer = saved_bit_buf, d->m_bits_in = saved_bits_in;
1291 tdefl_compress_block(d, MZ_TRUE);
1296 if (flush == TDEFL_FINISH)
1300 TDEFL_PUT_BITS(0, 8 - d->m_bits_in);
1302 if (d->m_flags & TDEFL_WRITE_ZLIB_HEADER)
1304 mz_uint i, a = d->m_adler32;
1305 for (i = 0; i < 4; i++)
1307 TDEFL_PUT_BITS((a >> 24) & 0xFF, 8);
1315 TDEFL_PUT_BITS(0, 3);
1318 TDEFL_PUT_BITS(0, 8 - d->m_bits_in);
1320 for (i = 2; i; --i, z ^= 0xFFFF)
1322 TDEFL_PUT_BITS(z & 0xFFFF, 16);
1327 MZ_ASSERT(d->m_pOutput_buf < d->m_pOutput_buf_end);
1329 memset(&d->m_huff_count[0][0], 0,
sizeof(d->m_huff_count[0][0]) * TDEFL_MAX_HUFF_SYMBOLS_0);
1330 memset(&d->m_huff_count[1][0], 0,
sizeof(d->m_huff_count[1][0]) * TDEFL_MAX_HUFF_SYMBOLS_1);
1332 d->m_pLZ_code_buf = d->m_lz_code_buf + 1;
1333 d->m_pLZ_flags = d->m_lz_code_buf;
1334 d->m_num_flags_left = 8;
1335 d->m_lz_code_buf_dict_pos += d->m_total_lz_bytes;
1336 d->m_total_lz_bytes = 0;
1339 if ((n = (
int)(d->m_pOutput_buf - pOutput_buf_start)) != 0)
1341 if (d->m_pPut_buf_func)
1343 *d->m_pIn_buf_size = d->m_pSrc - (
const mz_uint8 *)d->m_pIn_buf;
1344 if (!(*d->m_pPut_buf_func)(d->m_output_buf, n, d->m_pPut_buf_user))
1345 return (d->m_prev_return_status = TDEFL_STATUS_PUT_BUF_FAILED);
1347 else if (pOutput_buf_start == d->m_output_buf)
1349 int bytes_to_copy = (int)MZ_MIN((
size_t)n, (size_t)(*d->m_pOut_buf_size - d->m_out_buf_ofs));
1350 memcpy((mz_uint8 *)d->m_pOut_buf + d->m_out_buf_ofs, d->m_output_buf, bytes_to_copy);
1351 d->m_out_buf_ofs += bytes_to_copy;
1352 if ((n -= bytes_to_copy) != 0)
1354 d->m_output_flush_ofs = bytes_to_copy;
1355 d->m_output_flush_remaining = n;
1360 d->m_out_buf_ofs += n;
1364 return d->m_output_flush_remaining;
1367#if MINIZ_USE_UNALIGNED_LOADS_AND_STORES
1368#ifdef MINIZ_UNALIGNED_USE_MEMCPY
1369static mz_uint16 TDEFL_READ_UNALIGNED_WORD(
const mz_uint8* p)
1372 memcpy(&ret, p,
sizeof(mz_uint16));
1375static mz_uint16 TDEFL_READ_UNALIGNED_WORD2(
const mz_uint16* p)
1378 memcpy(&ret, p,
sizeof(mz_uint16));
1382#define TDEFL_READ_UNALIGNED_WORD(p) *(const mz_uint16 *)(p)
1383#define TDEFL_READ_UNALIGNED_WORD2(p) *(const mz_uint16 *)(p)
1385static MZ_FORCEINLINE
void tdefl_find_match(
tdefl_compressor *d, mz_uint lookahead_pos, mz_uint max_dist, mz_uint max_match_len, mz_uint *pMatch_dist, mz_uint *pMatch_len)
1387 mz_uint dist, pos = lookahead_pos & TDEFL_LZ_DICT_SIZE_MASK, match_len = *pMatch_len, probe_pos = pos, next_probe_pos, probe_len;
1388 mz_uint num_probes_left = d->m_max_probes[match_len >= 32];
1389 const mz_uint16 *s = (
const mz_uint16 *)(d->m_dict + pos), *p, *q;
1390 mz_uint16 c01 = TDEFL_READ_UNALIGNED_WORD(&d->m_dict[pos + match_len - 1]), s01 = TDEFL_READ_UNALIGNED_WORD2(s);
1391 MZ_ASSERT(max_match_len <= TDEFL_MAX_MATCH_LEN);
1392 if (max_match_len <= match_len)
1398 if (--num_probes_left == 0)
1400#define TDEFL_PROBE \
1401 next_probe_pos = d->m_next[probe_pos]; \
1402 if ((!next_probe_pos) || ((dist = (mz_uint16)(lookahead_pos - next_probe_pos)) > max_dist)) \
1404 probe_pos = next_probe_pos & TDEFL_LZ_DICT_SIZE_MASK; \
1405 if (TDEFL_READ_UNALIGNED_WORD(&d->m_dict[probe_pos + match_len - 1]) == c01) \
1413 q = (
const mz_uint16 *)(d->m_dict + probe_pos);
1414 if (TDEFL_READ_UNALIGNED_WORD2(q) != s01)
1420 }
while ((TDEFL_READ_UNALIGNED_WORD2(++p) == TDEFL_READ_UNALIGNED_WORD2(++q)) && (TDEFL_READ_UNALIGNED_WORD2(++p) == TDEFL_READ_UNALIGNED_WORD2(++q)) &&
1421 (TDEFL_READ_UNALIGNED_WORD2(++p) == TDEFL_READ_UNALIGNED_WORD2(++q)) && (TDEFL_READ_UNALIGNED_WORD2(++p) == TDEFL_READ_UNALIGNED_WORD2(++q)) && (--probe_len > 0));
1424 *pMatch_dist = dist;
1425 *pMatch_len = MZ_MIN(max_match_len, (mz_uint)TDEFL_MAX_MATCH_LEN);
1428 else if ((probe_len = ((mz_uint)(p - s) * 2) + (mz_uint)(*(
const mz_uint8 *)p == *(
const mz_uint8 *)q)) > match_len)
1430 *pMatch_dist = dist;
1431 if ((*pMatch_len = match_len = MZ_MIN(max_match_len, probe_len)) == max_match_len)
1433 c01 = TDEFL_READ_UNALIGNED_WORD(&d->m_dict[pos + match_len - 1]);
1438static MZ_FORCEINLINE
void tdefl_find_match(
tdefl_compressor *d, mz_uint lookahead_pos, mz_uint max_dist, mz_uint max_match_len, mz_uint *pMatch_dist, mz_uint *pMatch_len)
1440 mz_uint dist, pos = lookahead_pos & TDEFL_LZ_DICT_SIZE_MASK, match_len = *pMatch_len, probe_pos = pos, next_probe_pos, probe_len;
1441 mz_uint num_probes_left = d->m_max_probes[match_len >= 32];
1442 const mz_uint8 *s = d->m_dict + pos, *p, *q;
1443 mz_uint8 c0 = d->m_dict[pos + match_len], c1 = d->m_dict[pos + match_len - 1];
1444 MZ_ASSERT(max_match_len <= TDEFL_MAX_MATCH_LEN);
1445 if (max_match_len <= match_len)
1451 if (--num_probes_left == 0)
1453#define TDEFL_PROBE \
1454 next_probe_pos = d->m_next[probe_pos]; \
1455 if ((!next_probe_pos) || ((dist = (mz_uint16)(lookahead_pos - next_probe_pos)) > max_dist)) \
1457 probe_pos = next_probe_pos & TDEFL_LZ_DICT_SIZE_MASK; \
1458 if ((d->m_dict[probe_pos + match_len] == c0) && (d->m_dict[probe_pos + match_len - 1] == c1)) \
1467 q = d->m_dict + probe_pos;
1468 for (probe_len = 0; probe_len < max_match_len; probe_len++)
1471 if (probe_len > match_len)
1473 *pMatch_dist = dist;
1474 if ((*pMatch_len = match_len = probe_len) == max_match_len)
1476 c0 = d->m_dict[pos + match_len];
1477 c1 = d->m_dict[pos + match_len - 1];
1483#if MINIZ_USE_UNALIGNED_LOADS_AND_STORES && MINIZ_LITTLE_ENDIAN
1484#ifdef MINIZ_UNALIGNED_USE_MEMCPY
1485static mz_uint32 TDEFL_READ_UNALIGNED_WORD32(
const mz_uint8* p)
1488 memcpy(&ret, p,
sizeof(mz_uint32));
1492#define TDEFL_READ_UNALIGNED_WORD32(p) *(const mz_uint32 *)(p)
1497 mz_uint lookahead_pos = d->m_lookahead_pos, lookahead_size = d->m_lookahead_size, dict_size = d->m_dict_size, total_lz_bytes = d->m_total_lz_bytes, num_flags_left = d->m_num_flags_left;
1498 mz_uint8 *pLZ_code_buf = d->m_pLZ_code_buf, *pLZ_flags = d->m_pLZ_flags;
1499 mz_uint cur_pos = lookahead_pos & TDEFL_LZ_DICT_SIZE_MASK;
1501 while ((d->m_src_buf_left) || ((d->m_flush) && (lookahead_size)))
1503 const mz_uint TDEFL_COMP_FAST_LOOKAHEAD_SIZE = 4096;
1504 mz_uint dst_pos = (lookahead_pos + lookahead_size) & TDEFL_LZ_DICT_SIZE_MASK;
1505 mz_uint num_bytes_to_process = (mz_uint)MZ_MIN(d->m_src_buf_left, TDEFL_COMP_FAST_LOOKAHEAD_SIZE - lookahead_size);
1506 d->m_src_buf_left -= num_bytes_to_process;
1507 lookahead_size += num_bytes_to_process;
1509 while (num_bytes_to_process)
1511 mz_uint32 n = MZ_MIN(TDEFL_LZ_DICT_SIZE - dst_pos, num_bytes_to_process);
1512 memcpy(d->m_dict + dst_pos, d->m_pSrc, n);
1513 if (dst_pos < (TDEFL_MAX_MATCH_LEN - 1))
1514 memcpy(d->m_dict + TDEFL_LZ_DICT_SIZE + dst_pos, d->m_pSrc, MZ_MIN(n, (TDEFL_MAX_MATCH_LEN - 1) - dst_pos));
1516 dst_pos = (dst_pos + n) & TDEFL_LZ_DICT_SIZE_MASK;
1517 num_bytes_to_process -= n;
1520 dict_size = MZ_MIN(TDEFL_LZ_DICT_SIZE - lookahead_size, dict_size);
1521 if ((!d->m_flush) && (lookahead_size < TDEFL_COMP_FAST_LOOKAHEAD_SIZE))
1524 while (lookahead_size >= 4)
1526 mz_uint cur_match_dist, cur_match_len = 1;
1527 mz_uint8 *pCur_dict = d->m_dict + cur_pos;
1528 mz_uint first_trigram = TDEFL_READ_UNALIGNED_WORD32(pCur_dict) & 0xFFFFFF;
1529 mz_uint hash = (first_trigram ^ (first_trigram >> (24 - (TDEFL_LZ_HASH_BITS - 8)))) & TDEFL_LEVEL1_HASH_SIZE_MASK;
1530 mz_uint probe_pos = d->m_hash[hash];
1531 d->m_hash[hash] = (mz_uint16)lookahead_pos;
1533 if (((cur_match_dist = (mz_uint16)(lookahead_pos - probe_pos)) <= dict_size) && ((TDEFL_READ_UNALIGNED_WORD32(d->m_dict + (probe_pos &= TDEFL_LZ_DICT_SIZE_MASK)) & 0xFFFFFF) == first_trigram))
1535 const mz_uint16 *p = (
const mz_uint16 *)pCur_dict;
1536 const mz_uint16 *q = (
const mz_uint16 *)(d->m_dict + probe_pos);
1537 mz_uint32 probe_len = 32;
1540 }
while ((TDEFL_READ_UNALIGNED_WORD2(++p) == TDEFL_READ_UNALIGNED_WORD2(++q)) && (TDEFL_READ_UNALIGNED_WORD2(++p) == TDEFL_READ_UNALIGNED_WORD2(++q)) &&
1541 (TDEFL_READ_UNALIGNED_WORD2(++p) == TDEFL_READ_UNALIGNED_WORD2(++q)) && (TDEFL_READ_UNALIGNED_WORD2(++p) == TDEFL_READ_UNALIGNED_WORD2(++q)) && (--probe_len > 0));
1542 cur_match_len = ((mz_uint)(p - (
const mz_uint16 *)pCur_dict) * 2) + (mz_uint)(*(
const mz_uint8 *)p == *(
const mz_uint8 *)q);
1544 cur_match_len = cur_match_dist ? TDEFL_MAX_MATCH_LEN : 0;
1546 if ((cur_match_len < TDEFL_MIN_MATCH_LEN) || ((cur_match_len == TDEFL_MIN_MATCH_LEN) && (cur_match_dist >= 8U * 1024U)))
1549 *pLZ_code_buf++ = (mz_uint8)first_trigram;
1550 *pLZ_flags = (mz_uint8)(*pLZ_flags >> 1);
1551 d->m_huff_count[0][(mz_uint8)first_trigram]++;
1556 cur_match_len = MZ_MIN(cur_match_len, lookahead_size);
1558 MZ_ASSERT((cur_match_len >= TDEFL_MIN_MATCH_LEN) && (cur_match_dist >= 1) && (cur_match_dist <= TDEFL_LZ_DICT_SIZE));
1562 pLZ_code_buf[0] = (mz_uint8)(cur_match_len - TDEFL_MIN_MATCH_LEN);
1563#ifdef MINIZ_UNALIGNED_USE_MEMCPY
1564 memcpy(&pLZ_code_buf[1], &cur_match_dist,
sizeof(cur_match_dist));
1566 *(mz_uint16 *)(&pLZ_code_buf[1]) = (mz_uint16)cur_match_dist;
1569 *pLZ_flags = (mz_uint8)((*pLZ_flags >> 1) | 0x80);
1571 s0 = s_tdefl_small_dist_sym[cur_match_dist & 511];
1572 s1 = s_tdefl_large_dist_sym[cur_match_dist >> 8];
1573 d->m_huff_count[1][(cur_match_dist < 512) ? s0 : s1]++;
1575 d->m_huff_count[0][s_tdefl_len_sym[cur_match_len - TDEFL_MIN_MATCH_LEN]]++;
1580 *pLZ_code_buf++ = (mz_uint8)first_trigram;
1581 *pLZ_flags = (mz_uint8)(*pLZ_flags >> 1);
1582 d->m_huff_count[0][(mz_uint8)first_trigram]++;
1585 if (--num_flags_left == 0)
1588 pLZ_flags = pLZ_code_buf++;
1591 total_lz_bytes += cur_match_len;
1592 lookahead_pos += cur_match_len;
1593 dict_size = MZ_MIN(dict_size + cur_match_len, (mz_uint)TDEFL_LZ_DICT_SIZE);
1594 cur_pos = (cur_pos + cur_match_len) & TDEFL_LZ_DICT_SIZE_MASK;
1595 MZ_ASSERT(lookahead_size >= cur_match_len);
1596 lookahead_size -= cur_match_len;
1598 if (pLZ_code_buf > &d->m_lz_code_buf[TDEFL_LZ_CODE_BUF_SIZE - 8])
1601 d->m_lookahead_pos = lookahead_pos;
1602 d->m_lookahead_size = lookahead_size;
1603 d->m_dict_size = dict_size;
1604 d->m_total_lz_bytes = total_lz_bytes;
1605 d->m_pLZ_code_buf = pLZ_code_buf;
1606 d->m_pLZ_flags = pLZ_flags;
1607 d->m_num_flags_left = num_flags_left;
1608 if ((n = tdefl_flush_block(d, 0)) != 0)
1609 return (n < 0) ? MZ_FALSE : MZ_TRUE;
1610 total_lz_bytes = d->m_total_lz_bytes;
1611 pLZ_code_buf = d->m_pLZ_code_buf;
1612 pLZ_flags = d->m_pLZ_flags;
1613 num_flags_left = d->m_num_flags_left;
1617 while (lookahead_size)
1619 mz_uint8 lit = d->m_dict[cur_pos];
1622 *pLZ_code_buf++ = lit;
1623 *pLZ_flags = (mz_uint8)(*pLZ_flags >> 1);
1624 if (--num_flags_left == 0)
1627 pLZ_flags = pLZ_code_buf++;
1630 d->m_huff_count[0][lit]++;
1633 dict_size = MZ_MIN(dict_size + 1, (mz_uint)TDEFL_LZ_DICT_SIZE);
1634 cur_pos = (cur_pos + 1) & TDEFL_LZ_DICT_SIZE_MASK;
1637 if (pLZ_code_buf > &d->m_lz_code_buf[TDEFL_LZ_CODE_BUF_SIZE - 8])
1640 d->m_lookahead_pos = lookahead_pos;
1641 d->m_lookahead_size = lookahead_size;
1642 d->m_dict_size = dict_size;
1643 d->m_total_lz_bytes = total_lz_bytes;
1644 d->m_pLZ_code_buf = pLZ_code_buf;
1645 d->m_pLZ_flags = pLZ_flags;
1646 d->m_num_flags_left = num_flags_left;
1647 if ((n = tdefl_flush_block(d, 0)) != 0)
1648 return (n < 0) ? MZ_FALSE : MZ_TRUE;
1649 total_lz_bytes = d->m_total_lz_bytes;
1650 pLZ_code_buf = d->m_pLZ_code_buf;
1651 pLZ_flags = d->m_pLZ_flags;
1652 num_flags_left = d->m_num_flags_left;
1657 d->m_lookahead_pos = lookahead_pos;
1658 d->m_lookahead_size = lookahead_size;
1659 d->m_dict_size = dict_size;
1660 d->m_total_lz_bytes = total_lz_bytes;
1661 d->m_pLZ_code_buf = pLZ_code_buf;
1662 d->m_pLZ_flags = pLZ_flags;
1663 d->m_num_flags_left = num_flags_left;
1668static MZ_FORCEINLINE
void tdefl_record_literal(
tdefl_compressor *d, mz_uint8 lit)
1670 d->m_total_lz_bytes++;
1671 *d->m_pLZ_code_buf++ = lit;
1672 *d->m_pLZ_flags = (mz_uint8)(*d->m_pLZ_flags >> 1);
1673 if (--d->m_num_flags_left == 0)
1675 d->m_num_flags_left = 8;
1676 d->m_pLZ_flags = d->m_pLZ_code_buf++;
1678 d->m_huff_count[0][lit]++;
1681static MZ_FORCEINLINE
void tdefl_record_match(
tdefl_compressor *d, mz_uint match_len, mz_uint match_dist)
1685 MZ_ASSERT((match_len >= TDEFL_MIN_MATCH_LEN) && (match_dist >= 1) && (match_dist <= TDEFL_LZ_DICT_SIZE));
1687 d->m_total_lz_bytes += match_len;
1689 d->m_pLZ_code_buf[0] = (mz_uint8)(match_len - TDEFL_MIN_MATCH_LEN);
1692 d->m_pLZ_code_buf[1] = (mz_uint8)(match_dist & 0xFF);
1693 d->m_pLZ_code_buf[2] = (mz_uint8)(match_dist >> 8);
1694 d->m_pLZ_code_buf += 3;
1696 *d->m_pLZ_flags = (mz_uint8)((*d->m_pLZ_flags >> 1) | 0x80);
1697 if (--d->m_num_flags_left == 0)
1699 d->m_num_flags_left = 8;
1700 d->m_pLZ_flags = d->m_pLZ_code_buf++;
1703 s0 = s_tdefl_small_dist_sym[match_dist & 511];
1704 s1 = s_tdefl_large_dist_sym[(match_dist >> 8) & 127];
1705 d->m_huff_count[1][(match_dist < 512) ? s0 : s1]++;
1707 if (match_len >= TDEFL_MIN_MATCH_LEN)
1708 d->m_huff_count[0][s_tdefl_len_sym[match_len - TDEFL_MIN_MATCH_LEN]]++;
1713 const mz_uint8 *pSrc = d->m_pSrc;
1714 size_t src_buf_left = d->m_src_buf_left;
1715 tdefl_flush
flush = d->m_flush;
1717 while ((src_buf_left) || ((flush) && (d->m_lookahead_size)))
1719 mz_uint len_to_move, cur_match_dist, cur_match_len, cur_pos;
1721 if ((d->m_lookahead_size + d->m_dict_size) >= (TDEFL_MIN_MATCH_LEN - 1))
1723 mz_uint dst_pos = (d->m_lookahead_pos + d->m_lookahead_size) & TDEFL_LZ_DICT_SIZE_MASK, ins_pos = d->m_lookahead_pos + d->m_lookahead_size - 2;
1724 mz_uint hash = (d->m_dict[ins_pos & TDEFL_LZ_DICT_SIZE_MASK] << TDEFL_LZ_HASH_SHIFT) ^ d->m_dict[(ins_pos + 1) & TDEFL_LZ_DICT_SIZE_MASK];
1725 mz_uint num_bytes_to_process = (mz_uint)MZ_MIN(src_buf_left, TDEFL_MAX_MATCH_LEN - d->m_lookahead_size);
1726 const mz_uint8 *pSrc_end = pSrc + num_bytes_to_process;
1727 src_buf_left -= num_bytes_to_process;
1728 d->m_lookahead_size += num_bytes_to_process;
1729 while (pSrc != pSrc_end)
1731 mz_uint8 c = *pSrc++;
1732 d->m_dict[dst_pos] = c;
1733 if (dst_pos < (TDEFL_MAX_MATCH_LEN - 1))
1734 d->m_dict[TDEFL_LZ_DICT_SIZE + dst_pos] = c;
1735 hash = ((hash << TDEFL_LZ_HASH_SHIFT) ^ c) & (TDEFL_LZ_HASH_SIZE - 1);
1736 d->m_next[ins_pos & TDEFL_LZ_DICT_SIZE_MASK] = d->m_hash[hash];
1737 d->m_hash[hash] = (mz_uint16)(ins_pos);
1738 dst_pos = (dst_pos + 1) & TDEFL_LZ_DICT_SIZE_MASK;
1744 while ((src_buf_left) && (d->m_lookahead_size < TDEFL_MAX_MATCH_LEN))
1746 mz_uint8 c = *pSrc++;
1747 mz_uint dst_pos = (d->m_lookahead_pos + d->m_lookahead_size) & TDEFL_LZ_DICT_SIZE_MASK;
1749 d->m_dict[dst_pos] = c;
1750 if (dst_pos < (TDEFL_MAX_MATCH_LEN - 1))
1751 d->m_dict[TDEFL_LZ_DICT_SIZE + dst_pos] = c;
1752 if ((++d->m_lookahead_size + d->m_dict_size) >= TDEFL_MIN_MATCH_LEN)
1754 mz_uint ins_pos = d->m_lookahead_pos + (d->m_lookahead_size - 1) - 2;
1755 mz_uint hash = ((d->m_dict[ins_pos & TDEFL_LZ_DICT_SIZE_MASK] << (TDEFL_LZ_HASH_SHIFT * 2)) ^ (d->m_dict[(ins_pos + 1) & TDEFL_LZ_DICT_SIZE_MASK] << TDEFL_LZ_HASH_SHIFT) ^ c) & (TDEFL_LZ_HASH_SIZE - 1);
1756 d->m_next[ins_pos & TDEFL_LZ_DICT_SIZE_MASK] = d->m_hash[hash];
1757 d->m_hash[hash] = (mz_uint16)(ins_pos);
1761 d->m_dict_size = MZ_MIN(TDEFL_LZ_DICT_SIZE - d->m_lookahead_size, d->m_dict_size);
1762 if ((!flush) && (d->m_lookahead_size < TDEFL_MAX_MATCH_LEN))
1768 cur_match_len = d->m_saved_match_len ? d->m_saved_match_len : (TDEFL_MIN_MATCH_LEN - 1);
1769 cur_pos = d->m_lookahead_pos & TDEFL_LZ_DICT_SIZE_MASK;
1770 if (d->m_flags & (TDEFL_RLE_MATCHES | TDEFL_FORCE_ALL_RAW_BLOCKS))
1772 if ((d->m_dict_size) && (!(d->m_flags & TDEFL_FORCE_ALL_RAW_BLOCKS)))
1774 mz_uint8 c = d->m_dict[(cur_pos - 1) & TDEFL_LZ_DICT_SIZE_MASK];
1776 while (cur_match_len < d->m_lookahead_size)
1778 if (d->m_dict[cur_pos + cur_match_len] != c)
1782 if (cur_match_len < TDEFL_MIN_MATCH_LEN)
1790 tdefl_find_match(d, d->m_lookahead_pos, d->m_dict_size, d->m_lookahead_size, &cur_match_dist, &cur_match_len);
1792 if (((cur_match_len == TDEFL_MIN_MATCH_LEN) && (cur_match_dist >= 8U * 1024U)) || (cur_pos == cur_match_dist) || ((d->m_flags & TDEFL_FILTER_MATCHES) && (cur_match_len <= 5)))
1794 cur_match_dist = cur_match_len = 0;
1796 if (d->m_saved_match_len)
1798 if (cur_match_len > d->m_saved_match_len)
1800 tdefl_record_literal(d, (mz_uint8)d->m_saved_lit);
1801 if (cur_match_len >= 128)
1803 tdefl_record_match(d, cur_match_len, cur_match_dist);
1804 d->m_saved_match_len = 0;
1805 len_to_move = cur_match_len;
1809 d->m_saved_lit = d->m_dict[cur_pos];
1810 d->m_saved_match_dist = cur_match_dist;
1811 d->m_saved_match_len = cur_match_len;
1816 tdefl_record_match(d, d->m_saved_match_len, d->m_saved_match_dist);
1817 len_to_move = d->m_saved_match_len - 1;
1818 d->m_saved_match_len = 0;
1821 else if (!cur_match_dist)
1822 tdefl_record_literal(d, d->m_dict[MZ_MIN(cur_pos,
sizeof(d->m_dict) - 1)]);
1823 else if ((d->m_greedy_parsing) || (d->m_flags & TDEFL_RLE_MATCHES) || (cur_match_len >= 128))
1825 tdefl_record_match(d, cur_match_len, cur_match_dist);
1826 len_to_move = cur_match_len;
1830 d->m_saved_lit = d->m_dict[MZ_MIN(cur_pos,
sizeof(d->m_dict) - 1)];
1831 d->m_saved_match_dist = cur_match_dist;
1832 d->m_saved_match_len = cur_match_len;
1835 d->m_lookahead_pos += len_to_move;
1836 MZ_ASSERT(d->m_lookahead_size >= len_to_move);
1837 d->m_lookahead_size -= len_to_move;
1838 d->m_dict_size = MZ_MIN(d->m_dict_size + len_to_move, (mz_uint)TDEFL_LZ_DICT_SIZE);
1840 if ((d->m_pLZ_code_buf > &d->m_lz_code_buf[TDEFL_LZ_CODE_BUF_SIZE - 8]) ||
1841 ((d->m_total_lz_bytes > 31 * 1024) && (((((mz_uint)(d->m_pLZ_code_buf - d->m_lz_code_buf) * 115) >> 7) >= d->m_total_lz_bytes) || (d->m_flags & TDEFL_FORCE_ALL_RAW_BLOCKS))))
1845 d->m_src_buf_left = src_buf_left;
1846 if ((n = tdefl_flush_block(d, 0)) != 0)
1847 return (n < 0) ? MZ_FALSE : MZ_TRUE;
1852 d->m_src_buf_left = src_buf_left;
1858 if (d->m_pIn_buf_size)
1860 *d->m_pIn_buf_size = d->m_pSrc - (
const mz_uint8 *)d->m_pIn_buf;
1863 if (d->m_pOut_buf_size)
1865 size_t n = MZ_MIN(*d->m_pOut_buf_size - d->m_out_buf_ofs, d->m_output_flush_remaining);
1866 memcpy((mz_uint8 *)d->m_pOut_buf + d->m_out_buf_ofs, d->m_output_buf + d->m_output_flush_ofs, n);
1867 d->m_output_flush_ofs += (mz_uint)n;
1868 d->m_output_flush_remaining -= (mz_uint)n;
1869 d->m_out_buf_ofs += n;
1871 *d->m_pOut_buf_size = d->m_out_buf_ofs;
1874 return (d->m_finished && !d->m_output_flush_remaining) ? TDEFL_STATUS_DONE : TDEFL_STATUS_OKAY;
1877tdefl_status tdefl_compress(
tdefl_compressor *d,
const void *pIn_buf,
size_t *pIn_buf_size,
void *pOut_buf,
size_t *pOut_buf_size, tdefl_flush flush)
1885 return TDEFL_STATUS_BAD_PARAM;
1888 d->m_pIn_buf = pIn_buf;
1889 d->m_pIn_buf_size = pIn_buf_size;
1890 d->m_pOut_buf = pOut_buf;
1891 d->m_pOut_buf_size = pOut_buf_size;
1892 d->m_pSrc = (
const mz_uint8 *)(pIn_buf);
1893 d->m_src_buf_left = pIn_buf_size ? *pIn_buf_size : 0;
1894 d->m_out_buf_ofs = 0;
1897 if (((d->m_pPut_buf_func != NULL) == ((pOut_buf != NULL) || (pOut_buf_size != NULL))) || (d->m_prev_return_status != TDEFL_STATUS_OKAY) ||
1898 (d->m_wants_to_finish && (flush != TDEFL_FINISH)) || (pIn_buf_size && *pIn_buf_size && !pIn_buf) || (pOut_buf_size && *pOut_buf_size && !pOut_buf))
1904 return (d->m_prev_return_status = TDEFL_STATUS_BAD_PARAM);
1906 d->m_wants_to_finish |= (
flush == TDEFL_FINISH);
1908 if ((d->m_output_flush_remaining) || (d->m_finished))
1909 return (d->m_prev_return_status = tdefl_flush_output_buffer(d));
1911#if MINIZ_USE_UNALIGNED_LOADS_AND_STORES && MINIZ_LITTLE_ENDIAN
1912 if (((d->m_flags & TDEFL_MAX_PROBES_MASK) == 1) &&
1913 ((d->m_flags & TDEFL_GREEDY_PARSING_FLAG) != 0) &&
1914 ((d->m_flags & (TDEFL_FILTER_MATCHES | TDEFL_FORCE_ALL_RAW_BLOCKS | TDEFL_RLE_MATCHES)) == 0))
1916 if (!tdefl_compress_fast(d))
1917 return d->m_prev_return_status;
1922 if (!tdefl_compress_normal(d))
1923 return d->m_prev_return_status;
1926 if ((d->m_flags & (TDEFL_WRITE_ZLIB_HEADER | TDEFL_COMPUTE_ADLER32)) && (pIn_buf))
1927 d->m_adler32 = (mz_uint32)mz_adler32(d->m_adler32, (
const mz_uint8 *)pIn_buf, d->m_pSrc - (
const mz_uint8 *)pIn_buf);
1929 if ((flush) && (!d->m_lookahead_size) && (!d->m_src_buf_left) && (!d->m_output_flush_remaining))
1931 if (tdefl_flush_block(d, flush) < 0)
1932 return d->m_prev_return_status;
1933 d->m_finished = (
flush == TDEFL_FINISH);
1934 if (flush == TDEFL_FULL_FLUSH)
1936 MZ_CLEAR_OBJ(d->m_hash);
1937 MZ_CLEAR_OBJ(d->m_next);
1942 return (d->m_prev_return_status = tdefl_flush_output_buffer(d));
1945tdefl_status tdefl_compress_buffer(
tdefl_compressor *d,
const void *pIn_buf,
size_t in_buf_size, tdefl_flush flush)
1947 MZ_ASSERT(d->m_pPut_buf_func);
1948 return tdefl_compress(d, pIn_buf, &in_buf_size, NULL, NULL, flush);
1951tdefl_status tdefl_init(
tdefl_compressor *d, tdefl_put_buf_func_ptr pPut_buf_func,
void *pPut_buf_user,
int flags)
1953 d->m_pPut_buf_func = pPut_buf_func;
1954 d->m_pPut_buf_user = pPut_buf_user;
1955 d->m_flags = (mz_uint)(flags);
1956 d->m_max_probes[0] = 1 + ((flags & 0xFFF) + 2) / 3;
1957 d->m_greedy_parsing = (flags & TDEFL_GREEDY_PARSING_FLAG) != 0;
1958 d->m_max_probes[1] = 1 + (((flags & 0xFFF) >> 2) + 2) / 3;
1959 if (!(flags & TDEFL_NONDETERMINISTIC_PARSING_FLAG))
1960 MZ_CLEAR_OBJ(d->m_hash);
1961 d->m_lookahead_pos = d->m_lookahead_size = d->m_dict_size = d->m_total_lz_bytes = d->m_lz_code_buf_dict_pos = d->m_bits_in = 0;
1962 d->m_output_flush_ofs = d->m_output_flush_remaining = d->m_finished = d->m_block_index = d->m_bit_buffer = d->m_wants_to_finish = 0;
1963 d->m_pLZ_code_buf = d->m_lz_code_buf + 1;
1964 d->m_pLZ_flags = d->m_lz_code_buf;
1965 d->m_num_flags_left = 8;
1966 d->m_pOutput_buf = d->m_output_buf;
1967 d->m_pOutput_buf_end = d->m_output_buf;
1968 d->m_prev_return_status = TDEFL_STATUS_OKAY;
1969 d->m_saved_match_dist = d->m_saved_match_len = d->m_saved_lit = 0;
1971 d->m_pIn_buf = NULL;
1972 d->m_pOut_buf = NULL;
1973 d->m_pIn_buf_size = NULL;
1974 d->m_pOut_buf_size = NULL;
1975 d->m_flush = TDEFL_NO_FLUSH;
1977 d->m_src_buf_left = 0;
1978 d->m_out_buf_ofs = 0;
1979 if (!(flags & TDEFL_NONDETERMINISTIC_PARSING_FLAG))
1980 MZ_CLEAR_OBJ(d->m_dict);
1981 memset(&d->m_huff_count[0][0], 0,
sizeof(d->m_huff_count[0][0]) * TDEFL_MAX_HUFF_SYMBOLS_0);
1982 memset(&d->m_huff_count[1][0], 0,
sizeof(d->m_huff_count[1][0]) * TDEFL_MAX_HUFF_SYMBOLS_1);
1983 return TDEFL_STATUS_OKAY;
1988 return d->m_prev_return_status;
1993 return d->m_adler32;
1996mz_bool tdefl_compress_mem_to_output(
const void *pBuf,
size_t buf_len, tdefl_put_buf_func_ptr pPut_buf_func,
void *pPut_buf_user,
int flags)
2000 if (((buf_len) && (!pBuf)) || (!pPut_buf_func))
2005 succeeded = (tdefl_init(pComp, pPut_buf_func, pPut_buf_user, flags) == TDEFL_STATUS_OKAY);
2006 succeeded = succeeded && (tdefl_compress_buffer(pComp, pBuf, buf_len, TDEFL_FINISH) == TDEFL_STATUS_DONE);
2013 size_t m_size, m_capacity;
2015 mz_bool m_expandable;
2018static mz_bool tdefl_output_buffer_putter(
const void *pBuf,
int len,
void *pUser)
2021 size_t new_size = p->m_size + len;
2022 if (new_size > p->m_capacity)
2024 size_t new_capacity = p->m_capacity;
2026 if (!p->m_expandable)
2030 new_capacity = MZ_MAX(128U, new_capacity << 1U);
2031 }
while (new_size > new_capacity);
2032 pNew_buf = (mz_uint8 *)MZ_REALLOC(p->m_pBuf, new_capacity);
2035 p->m_pBuf = pNew_buf;
2036 p->m_capacity = new_capacity;
2038 memcpy((mz_uint8 *)p->m_pBuf + p->m_size, pBuf, len);
2039 p->m_size = new_size;
2043void *tdefl_compress_mem_to_heap(
const void *pSrc_buf,
size_t src_buf_len,
size_t *pOut_len,
int flags)
2046 MZ_CLEAR_OBJ(out_buf);
2051 out_buf.m_expandable = MZ_TRUE;
2052 if (!tdefl_compress_mem_to_output(pSrc_buf, src_buf_len, tdefl_output_buffer_putter, &out_buf, flags))
2054 *pOut_len = out_buf.m_size;
2055 return out_buf.m_pBuf;
2058size_t tdefl_compress_mem_to_mem(
void *pOut_buf,
size_t out_buf_len,
const void *pSrc_buf,
size_t src_buf_len,
int flags)
2061 MZ_CLEAR_OBJ(out_buf);
2064 out_buf.m_pBuf = (mz_uint8 *)pOut_buf;
2065 out_buf.m_capacity = out_buf_len;
2066 if (!tdefl_compress_mem_to_output(pSrc_buf, src_buf_len, tdefl_output_buffer_putter, &out_buf, flags))
2068 return out_buf.m_size;
2071static const mz_uint s_tdefl_num_probes[11] = { 0, 1, 6, 32, 16, 32, 128, 256, 512, 768, 1500 };
2074mz_uint tdefl_create_comp_flags_from_zip_params(
int level,
int window_bits,
int strategy)
2076 mz_uint comp_flags = s_tdefl_num_probes[(level >= 0) ? MZ_MIN(10, level) : MZ_DEFAULT_LEVEL] | ((level <= 3) ? TDEFL_GREEDY_PARSING_FLAG : 0);
2077 if (window_bits > 0)
2078 comp_flags |= TDEFL_WRITE_ZLIB_HEADER;
2081 comp_flags |= TDEFL_FORCE_ALL_RAW_BLOCKS;
2082 else if (strategy == MZ_FILTERED)
2083 comp_flags |= TDEFL_FILTER_MATCHES;
2084 else if (strategy == MZ_HUFFMAN_ONLY)
2085 comp_flags &= ~TDEFL_MAX_PROBES_MASK;
2086 else if (strategy == MZ_FIXED)
2087 comp_flags |= TDEFL_FORCE_ALL_STATIC_BLOCKS;
2088 else if (strategy == MZ_RLE)
2089 comp_flags |= TDEFL_RLE_MATCHES;
2095#pragma warning(push)
2096#pragma warning(disable : 4204)
2102void *tdefl_write_image_to_png_file_in_memory_ex(
const void *pImage,
int w,
int h,
int num_chans,
size_t *pLen_out, mz_uint level, mz_bool flip)
2105 static const mz_uint s_tdefl_png_num_probes[11] = { 0, 1, 6, 32, 16, 32, 128, 256, 512, 768, 1500 };
2108 int i, bpl = w * num_chans, y, z;
2113 MZ_CLEAR_OBJ(out_buf);
2114 out_buf.m_expandable = MZ_TRUE;
2115 out_buf.m_capacity = 57 + MZ_MAX(64, (1 + bpl) * h);
2116 if (NULL == (out_buf.m_pBuf = (mz_uint8 *)MZ_MALLOC(out_buf.m_capacity)))
2122 for (z = 41; z; --z)
2123 tdefl_output_buffer_putter(&z, 1, &out_buf);
2125 tdefl_init(pComp, tdefl_output_buffer_putter, &out_buf, s_tdefl_png_num_probes[MZ_MIN(10, level)] | TDEFL_WRITE_ZLIB_HEADER);
2126 for (y = 0; y < h; ++y)
2128 tdefl_compress_buffer(pComp, &z, 1, TDEFL_NO_FLUSH);
2129 tdefl_compress_buffer(pComp, (mz_uint8 *)pImage + (flip ? (h - 1 - y) : y) * bpl, bpl, TDEFL_NO_FLUSH);
2131 if (tdefl_compress_buffer(pComp, NULL, 0, TDEFL_FINISH) != TDEFL_STATUS_DONE)
2134 MZ_FREE(out_buf.m_pBuf);
2138 *pLen_out = out_buf.m_size - 41;
2140 static const mz_uint8 chans[] = { 0x00, 0x00, 0x04, 0x02, 0x06 };
2141 mz_uint8 pnghdr[41] = { 0x89, 0x50, 0x4e, 0x47, 0x0d,
2142 0x0a, 0x1a, 0x0a, 0x00, 0x00,
2143 0x00, 0x0d, 0x49, 0x48, 0x44,
2144 0x52, 0x00, 0x00, 0x00, 0x00,
2145 0x00, 0x00, 0x00, 0x00, 0x08,
2146 0x00, 0x00, 0x00, 0x00, 0x00,
2147 0x00, 0x00, 0x00, 0x00, 0x00,
2148 0x00, 0x00, 0x49, 0x44, 0x41,
2150 pnghdr[18] = (mz_uint8)(w >> 8);
2151 pnghdr[19] = (mz_uint8)w;
2152 pnghdr[22] = (mz_uint8)(h >> 8);
2153 pnghdr[23] = (mz_uint8)h;
2154 pnghdr[25] = chans[num_chans];
2155 pnghdr[33] = (mz_uint8)(*pLen_out >> 24);
2156 pnghdr[34] = (mz_uint8)(*pLen_out >> 16);
2157 pnghdr[35] = (mz_uint8)(*pLen_out >> 8);
2158 pnghdr[36] = (mz_uint8)*pLen_out;
2159 c = (mz_uint32)mz_crc32(MZ_CRC32_INIT, pnghdr + 12, 17);
2160 for (i = 0; i < 4; ++i, c <<= 8)
2161 ((mz_uint8 *)(pnghdr + 29))[i] = (mz_uint8)(c >> 24);
2162 memcpy(out_buf.m_pBuf, pnghdr, 41);
2165 if (!tdefl_output_buffer_putter(
"\0\0\0\0\0\0\0\0\x49\x45\x4e\x44\xae\x42\x60\x82", 16, &out_buf))
2169 MZ_FREE(out_buf.m_pBuf);
2172 c = (mz_uint32)mz_crc32(MZ_CRC32_INIT, out_buf.m_pBuf + 41 - 4, *pLen_out + 4);
2173 for (i = 0; i < 4; ++i, c <<= 8)
2174 (out_buf.m_pBuf + out_buf.m_size - 16)[i] = (mz_uint8)(c >> 24);
2178 return out_buf.m_pBuf;
2180void *tdefl_write_image_to_png_file_in_memory(
const void *pImage,
int w,
int h,
int num_chans,
size_t *pLen_out)
2183 return tdefl_write_image_to_png_file_in_memory_ex(pImage, w, h, num_chans, pLen_out, 6, MZ_FALSE);
2186#ifndef MINIZ_NO_MALLOC
2242#define TINFL_MEMCPY(d, s, l) memcpy(d, s, l)
2243#define TINFL_MEMSET(p, c, l) memset(p, c, l)
2245#define TINFL_CR_BEGIN \
2246 switch (r->m_state) \
2249#define TINFL_CR_RETURN(state_index, result) \
2253 r->m_state = state_index; \
2255 case state_index:; \
2258#define TINFL_CR_RETURN_FOREVER(state_index, result) \
2263 TINFL_CR_RETURN(state_index, result); \
2267#define TINFL_CR_FINISH }
2269#define TINFL_GET_BYTE(state_index, c) \
2272 while (pIn_buf_cur >= pIn_buf_end) \
2274 TINFL_CR_RETURN(state_index, (decomp_flags & TINFL_FLAG_HAS_MORE_INPUT) ? TINFL_STATUS_NEEDS_MORE_INPUT : TINFL_STATUS_FAILED_CANNOT_MAKE_PROGRESS); \
2276 c = *pIn_buf_cur++; \
2280#define TINFL_NEED_BITS(state_index, n) \
2284 TINFL_GET_BYTE(state_index, c); \
2285 bit_buf |= (((tinfl_bit_buf_t)c) << num_bits); \
2287 } while (num_bits < (mz_uint)(n))
2288#define TINFL_SKIP_BITS(state_index, n) \
2291 if (num_bits < (mz_uint)(n)) \
2293 TINFL_NEED_BITS(state_index, n); \
2299#define TINFL_GET_BITS(state_index, b, n) \
2302 if (num_bits < (mz_uint)(n)) \
2304 TINFL_NEED_BITS(state_index, n); \
2306 b = bit_buf & ((1 << (n)) - 1); \
2316#define TINFL_HUFF_BITBUF_FILL(state_index, pHuff) \
2319 temp = (pHuff)->m_look_up[bit_buf & (TINFL_FAST_LOOKUP_SIZE - 1)]; \
2322 code_len = temp >> 9; \
2323 if ((code_len) && (num_bits >= code_len)) \
2326 else if (num_bits > TINFL_FAST_LOOKUP_BITS) \
2328 code_len = TINFL_FAST_LOOKUP_BITS; \
2331 temp = (pHuff)->m_tree[~temp + ((bit_buf >> code_len++) & 1)]; \
2332 } while ((temp < 0) && (num_bits >= (code_len + 1))); \
2336 TINFL_GET_BYTE(state_index, c); \
2337 bit_buf |= (((tinfl_bit_buf_t)c) << num_bits); \
2339 } while (num_bits < 15);
2347#define TINFL_HUFF_DECODE(state_index, sym, pHuff) \
2351 mz_uint code_len, c; \
2352 if (num_bits < 15) \
2354 if ((pIn_buf_end - pIn_buf_cur) < 2) \
2356 TINFL_HUFF_BITBUF_FILL(state_index, pHuff); \
2360 bit_buf |= (((tinfl_bit_buf_t)pIn_buf_cur[0]) << num_bits) | (((tinfl_bit_buf_t)pIn_buf_cur[1]) << (num_bits + 8)); \
2365 if ((temp = (pHuff)->m_look_up[bit_buf & (TINFL_FAST_LOOKUP_SIZE - 1)]) >= 0) \
2366 code_len = temp >> 9, temp &= 511; \
2369 code_len = TINFL_FAST_LOOKUP_BITS; \
2372 temp = (pHuff)->m_tree[~temp + ((bit_buf >> code_len++) & 1)]; \
2373 } while (temp < 0); \
2376 bit_buf >>= code_len; \
2377 num_bits -= code_len; \
2381tinfl_status tinfl_decompress(
tinfl_decompressor *r,
const mz_uint8 *pIn_buf_next,
size_t *pIn_buf_size, mz_uint8 *pOut_buf_start, mz_uint8 *pOut_buf_next,
size_t *pOut_buf_size,
const mz_uint32 decomp_flags)
2383 static const int s_length_base[31] = { 3, 4, 5, 6, 7, 8, 9, 10, 11, 13, 15, 17, 19, 23, 27, 31, 35, 43, 51, 59, 67, 83, 99, 115, 131, 163, 195, 227, 258, 0, 0 };
2384 static const int s_length_extra[31] = { 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 0, 0, 0 };
2385 static const int s_dist_base[32] = { 1, 2, 3, 4, 5, 7, 9, 13, 17, 25, 33, 49, 65, 97, 129, 193, 257, 385, 513, 769, 1025, 1537, 2049, 3073, 4097, 6145, 8193, 12289, 16385, 24577, 0, 0 };
2386 static const int s_dist_extra[32] = { 0, 0, 0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13 };
2387 static const mz_uint8 s_length_dezigzag[19] = { 16, 17, 18, 0, 8, 7, 9, 6, 10, 5, 11, 4, 12, 3, 13, 2, 14, 1, 15 };
2388 static const int s_min_table_sizes[3] = { 257, 1, 4 };
2390 tinfl_status status = TINFL_STATUS_FAILED;
2391 mz_uint32 num_bits, dist, counter, num_extra;
2392 tinfl_bit_buf_t bit_buf;
2393 const mz_uint8 *pIn_buf_cur = pIn_buf_next, *
const pIn_buf_end = pIn_buf_next + *pIn_buf_size;
2394 mz_uint8 *pOut_buf_cur = pOut_buf_next, *
const pOut_buf_end = pOut_buf_next + *pOut_buf_size;
2395 size_t out_buf_size_mask = (decomp_flags & TINFL_FLAG_USING_NON_WRAPPING_OUTPUT_BUF) ? (
size_t)-1 : ((pOut_buf_next - pOut_buf_start) + *pOut_buf_size) - 1, dist_from_out_buf_start;
2398 if (((out_buf_size_mask + 1) & out_buf_size_mask) || (pOut_buf_next < pOut_buf_start))
2400 *pIn_buf_size = *pOut_buf_size = 0;
2401 return TINFL_STATUS_BAD_PARAM;
2404 num_bits = r->m_num_bits;
2405 bit_buf = r->m_bit_buf;
2407 counter = r->m_counter;
2408 num_extra = r->m_num_extra;
2409 dist_from_out_buf_start = r->m_dist_from_out_buf_start;
2412 bit_buf = num_bits = dist = counter = num_extra = r->m_zhdr0 = r->m_zhdr1 = 0;
2413 r->m_z_adler32 = r->m_check_adler32 = 1;
2414 if (decomp_flags & TINFL_FLAG_PARSE_ZLIB_HEADER)
2416 TINFL_GET_BYTE(1, r->m_zhdr0);
2417 TINFL_GET_BYTE(2, r->m_zhdr1);
2418 counter = (((r->m_zhdr0 * 256 + r->m_zhdr1) % 31 != 0) || (r->m_zhdr1 & 32) || ((r->m_zhdr0 & 15) != 8));
2419 if (!(decomp_flags & TINFL_FLAG_USING_NON_WRAPPING_OUTPUT_BUF))
2420 counter |= (((1U << (8U + (r->m_zhdr0 >> 4))) > 32768U) || ((out_buf_size_mask + 1) < (size_t)(1U << (8U + (r->m_zhdr0 >> 4)))));
2423 TINFL_CR_RETURN_FOREVER(36, TINFL_STATUS_FAILED);
2429 TINFL_GET_BITS(3, r->m_final, 3);
2430 r->m_type = r->m_final >> 1;
2433 TINFL_SKIP_BITS(5, num_bits & 7);
2434 for (counter = 0; counter < 4; ++counter)
2437 TINFL_GET_BITS(6, r->m_raw_header[counter], 8);
2439 TINFL_GET_BYTE(7, r->m_raw_header[counter]);
2441 if ((counter = (r->m_raw_header[0] | (r->m_raw_header[1] << 8))) != (mz_uint)(0xFFFF ^ (r->m_raw_header[2] | (r->m_raw_header[3] << 8))))
2443 TINFL_CR_RETURN_FOREVER(39, TINFL_STATUS_FAILED);
2445 while ((counter) && (num_bits))
2447 TINFL_GET_BITS(51, dist, 8);
2448 while (pOut_buf_cur >= pOut_buf_end)
2450 TINFL_CR_RETURN(52, TINFL_STATUS_HAS_MORE_OUTPUT);
2452 *pOut_buf_cur++ = (mz_uint8)dist;
2458 while (pOut_buf_cur >= pOut_buf_end)
2460 TINFL_CR_RETURN(9, TINFL_STATUS_HAS_MORE_OUTPUT);
2462 while (pIn_buf_cur >= pIn_buf_end)
2464 TINFL_CR_RETURN(38, (decomp_flags & TINFL_FLAG_HAS_MORE_INPUT) ? TINFL_STATUS_NEEDS_MORE_INPUT : TINFL_STATUS_FAILED_CANNOT_MAKE_PROGRESS);
2466 n = MZ_MIN(MZ_MIN((
size_t)(pOut_buf_end - pOut_buf_cur), (
size_t)(pIn_buf_end - pIn_buf_cur)), counter);
2467 TINFL_MEMCPY(pOut_buf_cur, pIn_buf_cur, n);
2470 counter -= (mz_uint)n;
2473 else if (r->m_type == 3)
2475 TINFL_CR_RETURN_FOREVER(10, TINFL_STATUS_FAILED);
2481 mz_uint8 *p = r->m_tables[0].m_code_size;
2483 r->m_table_sizes[0] = 288;
2484 r->m_table_sizes[1] = 32;
2485 TINFL_MEMSET(r->m_tables[1].m_code_size, 5, 32);
2486 for (i = 0; i <= 143; ++i)
2488 for (; i <= 255; ++i)
2490 for (; i <= 279; ++i)
2492 for (; i <= 287; ++i)
2497 for (counter = 0; counter < 3; counter++)
2499 TINFL_GET_BITS(11, r->m_table_sizes[counter],
"\05\05\04"[counter]);
2500 r->m_table_sizes[counter] += s_min_table_sizes[counter];
2502 MZ_CLEAR_OBJ(r->m_tables[2].m_code_size);
2503 for (counter = 0; counter < r->m_table_sizes[2]; counter++)
2506 TINFL_GET_BITS(14, s, 3);
2507 r->m_tables[2].m_code_size[s_length_dezigzag[counter]] = (mz_uint8)s;
2509 r->m_table_sizes[2] = 19;
2511 for (; (int)r->m_type >= 0; r->m_type--)
2513 int tree_next, tree_cur;
2515 mz_uint i, j, used_syms, total, sym_index, next_code[17], total_syms[16];
2516 pTable = &r->m_tables[r->m_type];
2517 MZ_CLEAR_OBJ(total_syms);
2518 MZ_CLEAR_OBJ(pTable->m_look_up);
2519 MZ_CLEAR_OBJ(pTable->m_tree);
2520 for (i = 0; i < r->m_table_sizes[r->m_type]; ++i)
2521 total_syms[pTable->m_code_size[i]]++;
2522 used_syms = 0, total = 0;
2523 next_code[0] = next_code[1] = 0;
2524 for (i = 1; i <= 15; ++i)
2526 used_syms += total_syms[i];
2527 next_code[i + 1] = (total = ((total + total_syms[i]) << 1));
2529 if ((65536 != total) && (used_syms > 1))
2531 TINFL_CR_RETURN_FOREVER(35, TINFL_STATUS_FAILED);
2533 for (tree_next = -1, sym_index = 0; sym_index < r->m_table_sizes[r->m_type]; ++sym_index)
2535 mz_uint rev_code = 0, l, cur_code, code_size = pTable->m_code_size[sym_index];
2538 cur_code = next_code[code_size]++;
2539 for (l = code_size; l > 0; l--, cur_code >>= 1)
2540 rev_code = (rev_code << 1) | (cur_code & 1);
2541 if (code_size <= TINFL_FAST_LOOKUP_BITS)
2543 mz_int16 k = (mz_int16)((code_size << 9) | sym_index);
2544 while (rev_code < TINFL_FAST_LOOKUP_SIZE)
2546 pTable->m_look_up[rev_code] = k;
2547 rev_code += (1 << code_size);
2551 if (0 == (tree_cur = pTable->m_look_up[rev_code & (TINFL_FAST_LOOKUP_SIZE - 1)]))
2553 pTable->m_look_up[rev_code & (TINFL_FAST_LOOKUP_SIZE - 1)] = (mz_int16)tree_next;
2554 tree_cur = tree_next;
2557 rev_code >>= (TINFL_FAST_LOOKUP_BITS - 1);
2558 for (j = code_size; j > (TINFL_FAST_LOOKUP_BITS + 1); j--)
2560 tree_cur -= ((rev_code >>= 1) & 1);
2561 if (!pTable->m_tree[-tree_cur - 1])
2563 pTable->m_tree[-tree_cur - 1] = (mz_int16)tree_next;
2564 tree_cur = tree_next;
2568 tree_cur = pTable->m_tree[-tree_cur - 1];
2570 tree_cur -= ((rev_code >>= 1) & 1);
2571 pTable->m_tree[-tree_cur - 1] = (mz_int16)sym_index;
2575 for (counter = 0; counter < (r->m_table_sizes[0] + r->m_table_sizes[1]);)
2578 TINFL_HUFF_DECODE(16, dist, &r->m_tables[2]);
2581 r->m_len_codes[counter++] = (mz_uint8)dist;
2584 if ((dist == 16) && (!counter))
2586 TINFL_CR_RETURN_FOREVER(17, TINFL_STATUS_FAILED);
2588 num_extra =
"\02\03\07"[dist - 16];
2589 TINFL_GET_BITS(18, s, num_extra);
2590 s +=
"\03\03\013"[dist - 16];
2591 TINFL_MEMSET(r->m_len_codes + counter, (dist == 16) ? r->m_len_codes[counter - 1] : 0, s);
2594 if ((r->m_table_sizes[0] + r->m_table_sizes[1]) != counter)
2596 TINFL_CR_RETURN_FOREVER(21, TINFL_STATUS_FAILED);
2598 TINFL_MEMCPY(r->m_tables[0].m_code_size, r->m_len_codes, r->m_table_sizes[0]);
2599 TINFL_MEMCPY(r->m_tables[1].m_code_size, r->m_len_codes + r->m_table_sizes[0], r->m_table_sizes[1]);
2607 if (((pIn_buf_end - pIn_buf_cur) < 4) || ((pOut_buf_end - pOut_buf_cur) < 2))
2609 TINFL_HUFF_DECODE(23, counter, &r->m_tables[0]);
2612 while (pOut_buf_cur >= pOut_buf_end)
2614 TINFL_CR_RETURN(24, TINFL_STATUS_HAS_MORE_OUTPUT);
2616 *pOut_buf_cur++ = (mz_uint8)counter;
2622#if TINFL_USE_64BIT_BITBUF
2625 bit_buf |= (((tinfl_bit_buf_t)MZ_READ_LE32(pIn_buf_cur)) << num_bits);
2632 bit_buf |= (((tinfl_bit_buf_t)MZ_READ_LE16(pIn_buf_cur)) << num_bits);
2637 if ((sym2 = r->m_tables[0].m_look_up[bit_buf & (TINFL_FAST_LOOKUP_SIZE - 1)]) >= 0)
2638 code_len = sym2 >> 9;
2641 code_len = TINFL_FAST_LOOKUP_BITS;
2644 sym2 = r->m_tables[0].m_tree[~sym2 + ((bit_buf >> code_len++) & 1)];
2648 bit_buf >>= code_len;
2649 num_bits -= code_len;
2653#if !TINFL_USE_64BIT_BITBUF
2656 bit_buf |= (((tinfl_bit_buf_t)MZ_READ_LE16(pIn_buf_cur)) << num_bits);
2661 if ((sym2 = r->m_tables[0].m_look_up[bit_buf & (TINFL_FAST_LOOKUP_SIZE - 1)]) >= 0)
2662 code_len = sym2 >> 9;
2665 code_len = TINFL_FAST_LOOKUP_BITS;
2668 sym2 = r->m_tables[0].m_tree[~sym2 + ((bit_buf >> code_len++) & 1)];
2671 bit_buf >>= code_len;
2672 num_bits -= code_len;
2674 pOut_buf_cur[0] = (mz_uint8)counter;
2681 pOut_buf_cur[1] = (mz_uint8)sym2;
2685 if ((counter &= 511) == 256)
2688 num_extra = s_length_extra[counter - 257];
2689 counter = s_length_base[counter - 257];
2693 TINFL_GET_BITS(25, extra_bits, num_extra);
2694 counter += extra_bits;
2697 TINFL_HUFF_DECODE(26, dist, &r->m_tables[1]);
2698 num_extra = s_dist_extra[dist];
2699 dist = s_dist_base[dist];
2703 TINFL_GET_BITS(27, extra_bits, num_extra);
2707 dist_from_out_buf_start = pOut_buf_cur - pOut_buf_start;
2708 if ((dist > dist_from_out_buf_start) && (decomp_flags & TINFL_FLAG_USING_NON_WRAPPING_OUTPUT_BUF))
2710 TINFL_CR_RETURN_FOREVER(37, TINFL_STATUS_FAILED);
2713 pSrc = pOut_buf_start + ((dist_from_out_buf_start - dist) & out_buf_size_mask);
2715 if ((MZ_MAX(pOut_buf_cur, pSrc) + counter) > pOut_buf_end)
2719 while (pOut_buf_cur >= pOut_buf_end)
2721 TINFL_CR_RETURN(53, TINFL_STATUS_HAS_MORE_OUTPUT);
2723 *pOut_buf_cur++ = pOut_buf_start[(dist_from_out_buf_start++ - dist) & out_buf_size_mask];
2727#if MINIZ_USE_UNALIGNED_LOADS_AND_STORES
2728 else if ((counter >= 9) && (counter <= dist))
2730 const mz_uint8 *pSrc_end = pSrc + (counter & ~7);
2733#ifdef MINIZ_UNALIGNED_USE_MEMCPY
2734 memcpy(pOut_buf_cur, pSrc,
sizeof(mz_uint32)*2);
2736 ((mz_uint32 *)pOut_buf_cur)[0] = ((
const mz_uint32 *)pSrc)[0];
2737 ((mz_uint32 *)pOut_buf_cur)[1] = ((
const mz_uint32 *)pSrc)[1];
2740 }
while ((pSrc += 8) < pSrc_end);
2741 if ((counter &= 7) < 3)
2745 pOut_buf_cur[0] = pSrc[0];
2747 pOut_buf_cur[1] = pSrc[1];
2748 pOut_buf_cur += counter;
2756 pOut_buf_cur[0] = pSrc[0];
2757 pOut_buf_cur[1] = pSrc[1];
2758 pOut_buf_cur[2] = pSrc[2];
2765 pOut_buf_cur[0] = pSrc[0];
2767 pOut_buf_cur[1] = pSrc[1];
2768 pOut_buf_cur += counter;
2772 }
while (!(r->m_final & 1));
2776 TINFL_SKIP_BITS(32, num_bits & 7);
2777 while ((pIn_buf_cur > pIn_buf_next) && (num_bits >= 8))
2782 bit_buf &= (tinfl_bit_buf_t)((((mz_uint64)1) << num_bits) - (mz_uint64)1);
2783 MZ_ASSERT(!num_bits);
2785 if (decomp_flags & TINFL_FLAG_PARSE_ZLIB_HEADER)
2787 for (counter = 0; counter < 4; ++counter)
2791 TINFL_GET_BITS(41, s, 8);
2793 TINFL_GET_BYTE(42, s);
2794 r->m_z_adler32 = (r->m_z_adler32 << 8) | s;
2797 TINFL_CR_RETURN_FOREVER(34, TINFL_STATUS_DONE);
2805 if ((status != TINFL_STATUS_NEEDS_MORE_INPUT) && (status != TINFL_STATUS_FAILED_CANNOT_MAKE_PROGRESS))
2807 while ((pIn_buf_cur > pIn_buf_next) && (num_bits >= 8))
2813 r->m_num_bits = num_bits;
2814 r->m_bit_buf = bit_buf & (tinfl_bit_buf_t)((((mz_uint64)1) << num_bits) - (mz_uint64)1);
2816 r->m_counter = counter;
2817 r->m_num_extra = num_extra;
2818 r->m_dist_from_out_buf_start = dist_from_out_buf_start;
2819 *pIn_buf_size = pIn_buf_cur - pIn_buf_next;
2820 *pOut_buf_size = pOut_buf_cur - pOut_buf_next;
2821 if ((decomp_flags & (TINFL_FLAG_PARSE_ZLIB_HEADER | TINFL_FLAG_COMPUTE_ADLER32)) && (status >= 0))
2823 const mz_uint8 *ptr = pOut_buf_next;
2824 size_t buf_len = *pOut_buf_size;
2825 mz_uint32 i, s1 = r->m_check_adler32 & 0xffff, s2 = r->m_check_adler32 >> 16;
2826 size_t block_len = buf_len % 5552;
2829 for (i = 0; i + 7 < block_len; i += 8, ptr += 8)
2831 s1 += ptr[0], s2 += s1;
2832 s1 += ptr[1], s2 += s1;
2833 s1 += ptr[2], s2 += s1;
2834 s1 += ptr[3], s2 += s1;
2835 s1 += ptr[4], s2 += s1;
2836 s1 += ptr[5], s2 += s1;
2837 s1 += ptr[6], s2 += s1;
2838 s1 += ptr[7], s2 += s1;
2840 for (; i < block_len; ++i)
2841 s1 += *ptr++, s2 += s1;
2842 s1 %= 65521U, s2 %= 65521U;
2843 buf_len -= block_len;
2846 r->m_check_adler32 = (s2 << 16) + s1;
2847 if ((status == TINFL_STATUS_DONE) && (decomp_flags & TINFL_FLAG_PARSE_ZLIB_HEADER) && (r->m_check_adler32 != r->m_z_adler32))
2848 status = TINFL_STATUS_ADLER32_MISMATCH;
2854void *tinfl_decompress_mem_to_heap(
const void *pSrc_buf,
size_t src_buf_len,
size_t *pOut_len,
int flags)
2857 void *pBuf = NULL, *pNew_buf;
2858 size_t src_buf_ofs = 0, out_buf_capacity = 0;
2860 tinfl_init(&decomp);
2863 size_t src_buf_size = src_buf_len - src_buf_ofs, dst_buf_size = out_buf_capacity - *pOut_len, new_out_buf_capacity;
2864 tinfl_status status = tinfl_decompress(&decomp, (
const mz_uint8 *)pSrc_buf + src_buf_ofs, &src_buf_size, (mz_uint8 *)pBuf, pBuf ? (mz_uint8 *)pBuf + *pOut_len : NULL, &dst_buf_size,
2865 (flags & ~TINFL_FLAG_HAS_MORE_INPUT) | TINFL_FLAG_USING_NON_WRAPPING_OUTPUT_BUF);
2866 if ((status < 0) || (status == TINFL_STATUS_NEEDS_MORE_INPUT))
2872 src_buf_ofs += src_buf_size;
2873 *pOut_len += dst_buf_size;
2874 if (status == TINFL_STATUS_DONE)
2876 new_out_buf_capacity = out_buf_capacity * 2;
2877 if (new_out_buf_capacity < 128)
2878 new_out_buf_capacity = 128;
2879 pNew_buf = MZ_REALLOC(pBuf, new_out_buf_capacity);
2887 out_buf_capacity = new_out_buf_capacity;
2892size_t tinfl_decompress_mem_to_mem(
void *pOut_buf,
size_t out_buf_len,
const void *pSrc_buf,
size_t src_buf_len,
int flags)
2895 tinfl_status status;
2896 tinfl_init(&decomp);
2897 status = tinfl_decompress(&decomp, (
const mz_uint8 *)pSrc_buf, &src_buf_len, (mz_uint8 *)pOut_buf, (mz_uint8 *)pOut_buf, &out_buf_len, (flags & ~TINFL_FLAG_HAS_MORE_INPUT) | TINFL_FLAG_USING_NON_WRAPPING_OUTPUT_BUF);
2898 return (status != TINFL_STATUS_DONE) ? TINFL_DECOMPRESS_MEM_TO_MEM_FAILED : out_buf_len;
2901int tinfl_decompress_mem_to_callback(
const void *pIn_buf,
size_t *pIn_buf_size, tinfl_put_buf_func_ptr pPut_buf_func,
void *pPut_buf_user,
int flags)
2905 mz_uint8 *pDict = (mz_uint8 *)MZ_MALLOC(TINFL_LZ_DICT_SIZE);
2906 size_t in_buf_ofs = 0, dict_ofs = 0;
2908 return TINFL_STATUS_FAILED;
2909 tinfl_init(&decomp);
2912 size_t in_buf_size = *pIn_buf_size - in_buf_ofs, dst_buf_size = TINFL_LZ_DICT_SIZE - dict_ofs;
2913 tinfl_status status = tinfl_decompress(&decomp, (
const mz_uint8 *)pIn_buf + in_buf_ofs, &in_buf_size, pDict, pDict + dict_ofs, &dst_buf_size,
2914 (flags & ~(TINFL_FLAG_HAS_MORE_INPUT | TINFL_FLAG_USING_NON_WRAPPING_OUTPUT_BUF)));
2915 in_buf_ofs += in_buf_size;
2916 if ((dst_buf_size) && (!(*pPut_buf_func)(pDict + dict_ofs, (
int)dst_buf_size, pPut_buf_user)))
2918 if (status != TINFL_STATUS_HAS_MORE_OUTPUT)
2920 result = (status == TINFL_STATUS_DONE);
2923 dict_ofs = (dict_ofs + dst_buf_size) & (TINFL_LZ_DICT_SIZE - 1);
2926 *pIn_buf_size = in_buf_ofs;
2930#ifndef MINIZ_NO_MALLOC
2935 tinfl_init(pDecomp);
2976#ifndef MINIZ_NO_ARCHIVE_APIS
2984#ifdef MINIZ_NO_STDIO
2985#define MZ_FILE void *
2987#include <sys/stat.h>
2989#if defined(_MSC_VER) || defined(__MINGW64__) || defined(__MINGW32__)
2991#define WIN32_LEAN_AND_MEAN
2995static std::wstring convert_to_utf16(
const char *s)
2997 std::wstring utf16String;
2998 int inputSize = strlen(s);
2999 int requiredSize = MultiByteToWideChar(CP_UTF8, 0, s, inputSize, 0, 0);
3000 if( requiredSize > 0 )
3002 utf16String.resize(requiredSize);
3003 MultiByteToWideChar(CP_UTF8, 0, s, inputSize, &utf16String[0], requiredSize);
3007static FILE *mz_fopen(
const char *pFilename,
const char *pMode)
3009 std::wstring wideFilename = convert_to_utf16(pFilename);
3010 std::wstring wideMode = convert_to_utf16(pMode);
3013 _wfopen_s(&pFile, wideFilename.c_str(), wideMode.c_str());
3016static FILE *mz_freopen(
const char *pPath,
const char *pMode, FILE *pStream)
3018 std::wstring widePath = convert_to_utf16(pPath);
3019 std::wstring wideMode = convert_to_utf16(pMode);
3022 if (_wfreopen_s(&pFile, widePath.c_str(), wideMode.c_str(), pStream))
3026#ifndef MINIZ_NO_TIME
3027#include <sys/utime.h>
3029#define MZ_FOPEN mz_fopen
3030#define MZ_FCLOSE fclose
3031#define MZ_FREAD fread
3032#define MZ_FWRITE fwrite
3033#define MZ_FTELL64 _ftelli64
3034#define MZ_FSEEK64 _fseeki64
3035#define MZ_FILE_STAT_STRUCT _stat64
3036#define MZ_FILE_STAT _stat64
3037#define MZ_FFLUSH fflush
3038#define MZ_FREOPEN mz_freopen
3039#define MZ_DELETE_FILE remove
3055#elif defined(__TINYC__)
3056#ifndef MINIZ_NO_TIME
3057#include <sys/utime.h>
3059#define MZ_FOPEN(f, m) fopen(f, m)
3060#define MZ_FCLOSE fclose
3061#define MZ_FREAD fread
3062#define MZ_FWRITE fwrite
3063#define MZ_FTELL64 ftell
3064#define MZ_FSEEK64 fseek
3065#define MZ_FILE_STAT_STRUCT stat
3066#define MZ_FILE_STAT stat
3067#define MZ_FFLUSH fflush
3068#define MZ_FREOPEN(f, m, s) freopen(f, m, s)
3069#define MZ_DELETE_FILE remove
3070#elif defined(__GNUC__) && defined(_LARGEFILE64_SOURCE)
3071#ifndef MINIZ_NO_TIME
3074#define MZ_FOPEN(f, m) fopen64(f, m)
3075#define MZ_FCLOSE fclose
3076#define MZ_FREAD fread
3077#define MZ_FWRITE fwrite
3078#define MZ_FTELL64 ftello64
3079#define MZ_FSEEK64 fseeko64
3080#define MZ_FILE_STAT_STRUCT stat64
3081#define MZ_FILE_STAT stat64
3082#define MZ_FFLUSH fflush
3083#define MZ_FREOPEN(p, m, s) freopen64(p, m, s)
3084#define MZ_DELETE_FILE remove
3085#elif defined(__APPLE__)
3086#ifndef MINIZ_NO_TIME
3089#define MZ_FOPEN(f, m) fopen(f, m)
3090#define MZ_FCLOSE fclose
3091#define MZ_FREAD fread
3092#define MZ_FWRITE fwrite
3093#define MZ_FTELL64 ftello
3094#define MZ_FSEEK64 fseeko
3095#define MZ_FILE_STAT_STRUCT stat
3096#define MZ_FILE_STAT stat
3097#define MZ_FFLUSH fflush
3098#define MZ_FREOPEN(p, m, s) freopen(p, m, s)
3099#define MZ_DELETE_FILE remove
3102#pragma message("Using fopen, ftello, fseeko, stat() etc. path for file I/O - this path may not support large files.")
3103#ifndef MINIZ_NO_TIME
3106#define MZ_FOPEN(f, m) fopen(f, m)
3107#define MZ_FCLOSE fclose
3108#define MZ_FREAD fread
3109#define MZ_FWRITE fwrite
3110#ifdef __STRICT_ANSI__
3111#define MZ_FTELL64 ftell
3112#define MZ_FSEEK64 fseek
3114#define MZ_FTELL64 ftello
3115#define MZ_FSEEK64 fseeko
3117#define MZ_FILE_STAT_STRUCT stat
3118#define MZ_FILE_STAT stat
3119#define MZ_FFLUSH fflush
3120#define MZ_FREOPEN(f, m, s) freopen(f, m, s)
3121#define MZ_DELETE_FILE remove
3125#define MZ_TOLOWER(c) ((((c) >= 'A') && ((c) <= 'Z')) ? ((c) - 'A' + 'a') : (c))
3131 MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIG = 0x06054b50,
3132 MZ_ZIP_CENTRAL_DIR_HEADER_SIG = 0x02014b50,
3133 MZ_ZIP_LOCAL_DIR_HEADER_SIG = 0x04034b50,
3134 MZ_ZIP_LOCAL_DIR_HEADER_SIZE = 30,
3135 MZ_ZIP_CENTRAL_DIR_HEADER_SIZE = 46,
3136 MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE = 22,
3139 MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIG = 0x06064b50,
3140 MZ_ZIP64_END_OF_CENTRAL_DIR_LOCATOR_SIG = 0x07064b50,
3141 MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIZE = 56,
3142 MZ_ZIP64_END_OF_CENTRAL_DIR_LOCATOR_SIZE = 20,
3143 MZ_ZIP64_EXTENDED_INFORMATION_FIELD_HEADER_ID = 0x0001,
3144 MZ_ZIP_DATA_DESCRIPTOR_ID = 0x08074b50,
3145 MZ_ZIP_DATA_DESCRIPTER_SIZE64 = 24,
3146 MZ_ZIP_DATA_DESCRIPTER_SIZE32 = 16,
3149 MZ_ZIP_CDH_SIG_OFS = 0,
3150 MZ_ZIP_CDH_VERSION_MADE_BY_OFS = 4,
3151 MZ_ZIP_CDH_VERSION_NEEDED_OFS = 6,
3152 MZ_ZIP_CDH_BIT_FLAG_OFS = 8,
3153 MZ_ZIP_CDH_METHOD_OFS = 10,
3154 MZ_ZIP_CDH_FILE_TIME_OFS = 12,
3155 MZ_ZIP_CDH_FILE_DATE_OFS = 14,
3156 MZ_ZIP_CDH_CRC32_OFS = 16,
3157 MZ_ZIP_CDH_COMPRESSED_SIZE_OFS = 20,
3158 MZ_ZIP_CDH_DECOMPRESSED_SIZE_OFS = 24,
3159 MZ_ZIP_CDH_FILENAME_LEN_OFS = 28,
3160 MZ_ZIP_CDH_EXTRA_LEN_OFS = 30,
3161 MZ_ZIP_CDH_COMMENT_LEN_OFS = 32,
3162 MZ_ZIP_CDH_DISK_START_OFS = 34,
3163 MZ_ZIP_CDH_INTERNAL_ATTR_OFS = 36,
3164 MZ_ZIP_CDH_EXTERNAL_ATTR_OFS = 38,
3165 MZ_ZIP_CDH_LOCAL_HEADER_OFS = 42,
3168 MZ_ZIP_LDH_SIG_OFS = 0,
3169 MZ_ZIP_LDH_VERSION_NEEDED_OFS = 4,
3170 MZ_ZIP_LDH_BIT_FLAG_OFS = 6,
3171 MZ_ZIP_LDH_METHOD_OFS = 8,
3172 MZ_ZIP_LDH_FILE_TIME_OFS = 10,
3173 MZ_ZIP_LDH_FILE_DATE_OFS = 12,
3174 MZ_ZIP_LDH_CRC32_OFS = 14,
3175 MZ_ZIP_LDH_COMPRESSED_SIZE_OFS = 18,
3176 MZ_ZIP_LDH_DECOMPRESSED_SIZE_OFS = 22,
3177 MZ_ZIP_LDH_FILENAME_LEN_OFS = 26,
3178 MZ_ZIP_LDH_EXTRA_LEN_OFS = 28,
3179 MZ_ZIP_LDH_BIT_FLAG_HAS_LOCATOR = 1 << 3,
3182 MZ_ZIP_ECDH_SIG_OFS = 0,
3183 MZ_ZIP_ECDH_NUM_THIS_DISK_OFS = 4,
3184 MZ_ZIP_ECDH_NUM_DISK_CDIR_OFS = 6,
3185 MZ_ZIP_ECDH_CDIR_NUM_ENTRIES_ON_DISK_OFS = 8,
3186 MZ_ZIP_ECDH_CDIR_TOTAL_ENTRIES_OFS = 10,
3187 MZ_ZIP_ECDH_CDIR_SIZE_OFS = 12,
3188 MZ_ZIP_ECDH_CDIR_OFS_OFS = 16,
3189 MZ_ZIP_ECDH_COMMENT_SIZE_OFS = 20,
3192 MZ_ZIP64_ECDL_SIG_OFS = 0,
3193 MZ_ZIP64_ECDL_NUM_DISK_CDIR_OFS = 4,
3194 MZ_ZIP64_ECDL_REL_OFS_TO_ZIP64_ECDR_OFS = 8,
3195 MZ_ZIP64_ECDL_TOTAL_NUMBER_OF_DISKS_OFS = 16,
3198 MZ_ZIP64_ECDH_SIG_OFS = 0,
3199 MZ_ZIP64_ECDH_SIZE_OF_RECORD_OFS = 4,
3200 MZ_ZIP64_ECDH_VERSION_MADE_BY_OFS = 12,
3201 MZ_ZIP64_ECDH_VERSION_NEEDED_OFS = 14,
3202 MZ_ZIP64_ECDH_NUM_THIS_DISK_OFS = 16,
3203 MZ_ZIP64_ECDH_NUM_DISK_CDIR_OFS = 20,
3204 MZ_ZIP64_ECDH_CDIR_NUM_ENTRIES_ON_DISK_OFS = 24,
3205 MZ_ZIP64_ECDH_CDIR_TOTAL_ENTRIES_OFS = 32,
3206 MZ_ZIP64_ECDH_CDIR_SIZE_OFS = 40,
3207 MZ_ZIP64_ECDH_CDIR_OFS_OFS = 48,
3208 MZ_ZIP_VERSION_MADE_BY_DOS_FILESYSTEM_ID = 0,
3209 MZ_ZIP_DOS_DIR_ATTRIBUTE_BITFLAG = 0x10,
3210 MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_IS_ENCRYPTED = 1,
3211 MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_COMPRESSED_PATCH_FLAG = 32,
3212 MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_USES_STRONG_ENCRYPTION = 64,
3213 MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_LOCAL_DIR_IS_MASKED = 8192,
3214 MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_UTF8 = 1 << 11
3220 size_t m_size, m_capacity;
3221 mz_uint m_element_size;
3231 uint32_t m_init_flags;
3237 mz_bool m_zip64_has_extended_info_fields;
3241 mz_uint64 m_file_archive_start_ofs;
3245 size_t m_mem_capacity;
3248#define MZ_ZIP_ARRAY_SET_ELEMENT_SIZE(array_ptr, element_size) (array_ptr)->m_element_size = element_size
3250#if defined(DEBUG) || defined(_DEBUG) || defined(NDEBUG)
3251static MZ_FORCEINLINE mz_uint mz_zip_array_range_check(
const mz_zip_array *pArray, mz_uint index)
3253 MZ_ASSERT(index < pArray->m_size);
3256#define MZ_ZIP_ARRAY_ELEMENT(array_ptr, element_type, index) ((element_type *)((array_ptr)->m_p))[mz_zip_array_range_check(array_ptr, index)]
3258#define MZ_ZIP_ARRAY_ELEMENT(array_ptr, element_type, index) ((element_type *)((array_ptr)->m_p))[index]
3261static MZ_FORCEINLINE
void mz_zip_array_init(
mz_zip_array *pArray, mz_uint32 element_size)
3264 pArray->m_element_size = element_size;
3269 pZip->m_pFree(pZip->m_pAlloc_opaque, pArray->m_p);
3273static mz_bool mz_zip_array_ensure_capacity(
mz_zip_archive *pZip,
mz_zip_array *pArray,
size_t min_new_capacity, mz_uint growing)
3276 size_t new_capacity = min_new_capacity;
3277 MZ_ASSERT(pArray->m_element_size);
3278 if (pArray->m_capacity >= min_new_capacity)
3282 new_capacity = MZ_MAX(1, pArray->m_capacity);
3283 while (new_capacity < min_new_capacity)
3286 if (NULL == (pNew_p = pZip->m_pRealloc(pZip->m_pAlloc_opaque, pArray->m_p, pArray->m_element_size, new_capacity)))
3288 pArray->m_p = pNew_p;
3289 pArray->m_capacity = new_capacity;
3293static MZ_FORCEINLINE mz_bool mz_zip_array_reserve(
mz_zip_archive *pZip,
mz_zip_array *pArray,
size_t new_capacity, mz_uint growing)
3295 if (new_capacity > pArray->m_capacity)
3297 if (!mz_zip_array_ensure_capacity(pZip, pArray, new_capacity, growing))
3303static MZ_FORCEINLINE mz_bool mz_zip_array_resize(
mz_zip_archive *pZip,
mz_zip_array *pArray,
size_t new_size, mz_uint growing)
3305 if (new_size > pArray->m_capacity)
3307 if (!mz_zip_array_ensure_capacity(pZip, pArray, new_size, growing))
3310 pArray->m_size = new_size;
3316 return mz_zip_array_reserve(pZip, pArray, pArray->m_size + n, MZ_TRUE);
3319static MZ_FORCEINLINE mz_bool mz_zip_array_push_back(
mz_zip_archive *pZip,
mz_zip_array *pArray,
const void *pElements,
size_t n)
3321 size_t orig_size = pArray->m_size;
3322 if (!mz_zip_array_resize(pZip, pArray, orig_size + n, MZ_TRUE))
3325 memcpy((mz_uint8 *)pArray->m_p + orig_size * pArray->m_element_size, pElements, n * pArray->m_element_size);
3329#ifndef MINIZ_NO_TIME
3330static MZ_TIME_T mz_zip_dos_to_time_t(
int dos_time,
int dos_date)
3333 memset(&tm, 0,
sizeof(tm));
3335 tm.tm_year = ((dos_date >> 9) & 127) + 1980 - 1900;
3336 tm.tm_mon = ((dos_date >> 5) & 15) - 1;
3337 tm.tm_mday = dos_date & 31;
3338 tm.tm_hour = (dos_time >> 11) & 31;
3339 tm.tm_min = (dos_time >> 5) & 63;
3340 tm.tm_sec = (dos_time << 1) & 62;
3344#ifndef MINIZ_NO_ARCHIVE_WRITING_APIS
3345static void mz_zip_time_t_to_dos_time(MZ_TIME_T time, mz_uint16 *pDOS_time, mz_uint16 *pDOS_date)
3348 struct tm tm_struct;
3349 struct tm *tm = &tm_struct;
3350 errno_t err = localtime_s(tm, &time);
3358 struct tm *tm = localtime(&time);
3361 *pDOS_time = (mz_uint16)(((tm->tm_hour) << 11) + ((tm->tm_min) << 5) + ((tm->tm_sec) >> 1));
3362 *pDOS_date = (mz_uint16)(((tm->tm_year + 1900 - 1980) << 9) + ((tm->tm_mon + 1) << 5) + tm->tm_mday);
3366#ifndef MINIZ_NO_STDIO
3367#ifndef MINIZ_NO_ARCHIVE_WRITING_APIS
3368static mz_bool mz_zip_get_file_modified_time(
const char *pFilename, MZ_TIME_T *pTime)
3370 struct MZ_FILE_STAT_STRUCT file_stat;
3373 if (MZ_FILE_STAT(pFilename, &file_stat) != 0)
3376 *pTime = file_stat.st_mtime;
3382static mz_bool mz_zip_set_file_times(
const char *pFilename, MZ_TIME_T access_time, MZ_TIME_T modified_time)
3386 memset(&t, 0,
sizeof(t));
3387 t.actime = access_time;
3388 t.modtime = modified_time;
3390 return !utime(pFilename, &t);
3395static MZ_FORCEINLINE mz_bool mz_zip_set_error(
mz_zip_archive *pZip, mz_zip_error err_num)
3398 pZip->m_last_error = err_num;
3402static mz_bool mz_zip_reader_init_internal(
mz_zip_archive *pZip, mz_uint flags)
3405 if ((!pZip) || (pZip->m_pState) || (pZip->m_zip_mode != MZ_ZIP_MODE_INVALID))
3406 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
3408 if (!pZip->m_pAlloc)
3409 pZip->m_pAlloc = miniz_def_alloc_func;
3411 pZip->m_pFree = miniz_def_free_func;
3412 if (!pZip->m_pRealloc)
3413 pZip->m_pRealloc = miniz_def_realloc_func;
3415 pZip->m_archive_size = 0;
3416 pZip->m_central_directory_file_ofs = 0;
3417 pZip->m_total_files = 0;
3418 pZip->m_last_error = MZ_ZIP_NO_ERROR;
3421 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
3424 MZ_ZIP_ARRAY_SET_ELEMENT_SIZE(&pZip->m_pState->m_central_dir,
sizeof(mz_uint8));
3425 MZ_ZIP_ARRAY_SET_ELEMENT_SIZE(&pZip->m_pState->m_central_dir_offsets,
sizeof(mz_uint32));
3426 MZ_ZIP_ARRAY_SET_ELEMENT_SIZE(&pZip->m_pState->m_sorted_central_dir_offsets,
sizeof(mz_uint32));
3427 pZip->m_pState->m_init_flags = flags;
3428 pZip->m_pState->m_zip64 = MZ_FALSE;
3429 pZip->m_pState->m_zip64_has_extended_info_fields = MZ_FALSE;
3431 pZip->m_zip_mode = MZ_ZIP_MODE_READING;
3436static MZ_FORCEINLINE mz_bool mz_zip_reader_filename_less(
const mz_zip_array *pCentral_dir_array,
const mz_zip_array *pCentral_dir_offsets, mz_uint l_index, mz_uint r_index)
3438 const mz_uint8 *pL = &MZ_ZIP_ARRAY_ELEMENT(pCentral_dir_array, mz_uint8, MZ_ZIP_ARRAY_ELEMENT(pCentral_dir_offsets, mz_uint32, l_index)), *pE;
3439 const mz_uint8 *pR = &MZ_ZIP_ARRAY_ELEMENT(pCentral_dir_array, mz_uint8, MZ_ZIP_ARRAY_ELEMENT(pCentral_dir_offsets, mz_uint32, r_index));
3440 mz_uint l_len = MZ_READ_LE16(pL + MZ_ZIP_CDH_FILENAME_LEN_OFS), r_len = MZ_READ_LE16(pR + MZ_ZIP_CDH_FILENAME_LEN_OFS);
3441 mz_uint8 l = 0, r = 0;
3442 pL += MZ_ZIP_CENTRAL_DIR_HEADER_SIZE;
3443 pR += MZ_ZIP_CENTRAL_DIR_HEADER_SIZE;
3444 pE = pL + MZ_MIN(l_len, r_len);
3447 if ((l = MZ_TOLOWER(*pL)) != (r = MZ_TOLOWER(*pR)))
3452 return (pL == pE) ? (l_len < r_len) : (l < r);
3455#define MZ_SWAP_UINT32(a, b) \
3465static void mz_zip_reader_sort_central_dir_offsets_by_filename(
mz_zip_archive *pZip)
3468 const mz_zip_array *pCentral_dir_offsets = &pState->m_central_dir_offsets;
3469 const mz_zip_array *pCentral_dir = &pState->m_central_dir;
3470 mz_uint32 *pIndices;
3471 mz_uint32 start, end;
3472 const mz_uint32 size = pZip->m_total_files;
3477 pIndices = &MZ_ZIP_ARRAY_ELEMENT(&pState->m_sorted_central_dir_offsets, mz_uint32, 0);
3479 start = (size - 2U) >> 1U;
3482 mz_uint64 child, root = start;
3485 if ((child = (root << 1U) + 1U) >= size)
3487 child += (((child + 1U) < size) && (mz_zip_reader_filename_less(pCentral_dir, pCentral_dir_offsets, pIndices[child], pIndices[child + 1U])));
3488 if (!mz_zip_reader_filename_less(pCentral_dir, pCentral_dir_offsets, pIndices[root], pIndices[child]))
3490 MZ_SWAP_UINT32(pIndices[root], pIndices[child]);
3501 mz_uint64 child, root = 0;
3502 MZ_SWAP_UINT32(pIndices[end], pIndices[0]);
3505 if ((child = (root << 1U) + 1U) >= end)
3507 child += (((child + 1U) < end) && mz_zip_reader_filename_less(pCentral_dir, pCentral_dir_offsets, pIndices[child], pIndices[child + 1U]));
3508 if (!mz_zip_reader_filename_less(pCentral_dir, pCentral_dir_offsets, pIndices[root], pIndices[child]))
3510 MZ_SWAP_UINT32(pIndices[root], pIndices[child]);
3517static mz_bool mz_zip_reader_locate_header_sig(
mz_zip_archive *pZip, mz_uint32 record_sig, mz_uint32 record_size, mz_int64 *pOfs)
3519 mz_int64 cur_file_ofs;
3520 mz_uint32 buf_u32[4096 /
sizeof(mz_uint32)];
3521 mz_uint8 *pBuf = (mz_uint8 *)buf_u32;
3524 if (pZip->m_archive_size < record_size)
3528 cur_file_ofs = MZ_MAX((mz_int64)pZip->m_archive_size - (mz_int64)
sizeof(buf_u32), 0);
3531 int i, n = (int)MZ_MIN(
sizeof(buf_u32), pZip->m_archive_size - cur_file_ofs);
3533 if (pZip->m_pRead(pZip->m_pIO_opaque, cur_file_ofs, pBuf, n) != (mz_uint)n)
3536 for (i = n - 4; i >= 0; --i)
3538 mz_uint s = MZ_READ_LE32(pBuf + i);
3539 if (s == record_sig)
3541 if ((pZip->m_archive_size - (cur_file_ofs + i)) >= record_size)
3553 if ((!cur_file_ofs) || ((pZip->m_archive_size - cur_file_ofs) >= (MZ_UINT16_MAX + record_size)))
3556 cur_file_ofs = MZ_MAX(cur_file_ofs - (
sizeof(buf_u32) - 3), 0);
3559 *pOfs = cur_file_ofs;
3563static mz_bool mz_zip_reader_read_central_dir(
mz_zip_archive *pZip, mz_uint flags)
3565 mz_uint cdir_size = 0, cdir_entries_on_this_disk = 0, num_this_disk = 0, cdir_disk_index = 0;
3566 mz_uint64 cdir_ofs = 0;
3567 mz_int64 cur_file_ofs = 0;
3570 mz_uint32 buf_u32[4096 /
sizeof(mz_uint32)];
3571 mz_uint8 *pBuf = (mz_uint8 *)buf_u32;
3572 mz_bool sort_central_dir = ((flags & MZ_ZIP_FLAG_DO_NOT_SORT_CENTRAL_DIRECTORY) == 0);
3573 mz_uint32 zip64_end_of_central_dir_locator_u32[(MZ_ZIP64_END_OF_CENTRAL_DIR_LOCATOR_SIZE +
sizeof(mz_uint32) - 1) /
sizeof(mz_uint32)];
3574 mz_uint8 *pZip64_locator = (mz_uint8 *)zip64_end_of_central_dir_locator_u32;
3576 mz_uint32 zip64_end_of_central_dir_header_u32[(MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIZE +
sizeof(mz_uint32) - 1) /
sizeof(mz_uint32)];
3577 mz_uint8 *pZip64_end_of_central_dir = (mz_uint8 *)zip64_end_of_central_dir_header_u32;
3579 mz_uint64 zip64_end_of_central_dir_ofs = 0;
3582 if (pZip->m_archive_size < MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE)
3583 return mz_zip_set_error(pZip, MZ_ZIP_NOT_AN_ARCHIVE);
3585 if (!mz_zip_reader_locate_header_sig(pZip, MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIG, MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE, &cur_file_ofs))
3586 return mz_zip_set_error(pZip, MZ_ZIP_FAILED_FINDING_CENTRAL_DIR);
3589 if (pZip->m_pRead(pZip->m_pIO_opaque, cur_file_ofs, pBuf, MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE) != MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE)
3590 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
3592 if (MZ_READ_LE32(pBuf + MZ_ZIP_ECDH_SIG_OFS) != MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIG)
3593 return mz_zip_set_error(pZip, MZ_ZIP_NOT_AN_ARCHIVE);
3595 if (cur_file_ofs >= (MZ_ZIP64_END_OF_CENTRAL_DIR_LOCATOR_SIZE + MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIZE))
3597 if (pZip->m_pRead(pZip->m_pIO_opaque, cur_file_ofs - MZ_ZIP64_END_OF_CENTRAL_DIR_LOCATOR_SIZE, pZip64_locator, MZ_ZIP64_END_OF_CENTRAL_DIR_LOCATOR_SIZE) == MZ_ZIP64_END_OF_CENTRAL_DIR_LOCATOR_SIZE)
3599 if (MZ_READ_LE32(pZip64_locator + MZ_ZIP64_ECDL_SIG_OFS) == MZ_ZIP64_END_OF_CENTRAL_DIR_LOCATOR_SIG)
3601 zip64_end_of_central_dir_ofs = MZ_READ_LE64(pZip64_locator + MZ_ZIP64_ECDL_REL_OFS_TO_ZIP64_ECDR_OFS);
3602 if (zip64_end_of_central_dir_ofs > (pZip->m_archive_size - MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIZE))
3603 return mz_zip_set_error(pZip, MZ_ZIP_NOT_AN_ARCHIVE);
3605 if (pZip->m_pRead(pZip->m_pIO_opaque, zip64_end_of_central_dir_ofs, pZip64_end_of_central_dir, MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIZE) == MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIZE)
3607 if (MZ_READ_LE32(pZip64_end_of_central_dir + MZ_ZIP64_ECDH_SIG_OFS) == MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIG)
3609 pZip->m_pState->m_zip64 = MZ_TRUE;
3616 pZip->m_total_files = MZ_READ_LE16(pBuf + MZ_ZIP_ECDH_CDIR_TOTAL_ENTRIES_OFS);
3617 cdir_entries_on_this_disk = MZ_READ_LE16(pBuf + MZ_ZIP_ECDH_CDIR_NUM_ENTRIES_ON_DISK_OFS);
3618 num_this_disk = MZ_READ_LE16(pBuf + MZ_ZIP_ECDH_NUM_THIS_DISK_OFS);
3619 cdir_disk_index = MZ_READ_LE16(pBuf + MZ_ZIP_ECDH_NUM_DISK_CDIR_OFS);
3620 cdir_size = MZ_READ_LE32(pBuf + MZ_ZIP_ECDH_CDIR_SIZE_OFS);
3621 cdir_ofs = MZ_READ_LE32(pBuf + MZ_ZIP_ECDH_CDIR_OFS_OFS);
3623 if (pZip->m_pState->m_zip64)
3625 mz_uint32 zip64_total_num_of_disks = MZ_READ_LE32(pZip64_locator + MZ_ZIP64_ECDL_TOTAL_NUMBER_OF_DISKS_OFS);
3626 mz_uint64 zip64_cdir_total_entries = MZ_READ_LE64(pZip64_end_of_central_dir + MZ_ZIP64_ECDH_CDIR_TOTAL_ENTRIES_OFS);
3627 mz_uint64 zip64_cdir_total_entries_on_this_disk = MZ_READ_LE64(pZip64_end_of_central_dir + MZ_ZIP64_ECDH_CDIR_NUM_ENTRIES_ON_DISK_OFS);
3628 mz_uint64 zip64_size_of_end_of_central_dir_record = MZ_READ_LE64(pZip64_end_of_central_dir + MZ_ZIP64_ECDH_SIZE_OF_RECORD_OFS);
3629 mz_uint64 zip64_size_of_central_directory = MZ_READ_LE64(pZip64_end_of_central_dir + MZ_ZIP64_ECDH_CDIR_SIZE_OFS);
3631 if (zip64_size_of_end_of_central_dir_record < (MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIZE - 12))
3632 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
3634 if (zip64_total_num_of_disks != 1U)
3635 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_MULTIDISK);
3638 if (zip64_cdir_total_entries > MZ_UINT32_MAX)
3639 return mz_zip_set_error(pZip, MZ_ZIP_TOO_MANY_FILES);
3641 pZip->m_total_files = (mz_uint32)zip64_cdir_total_entries;
3643 if (zip64_cdir_total_entries_on_this_disk > MZ_UINT32_MAX)
3644 return mz_zip_set_error(pZip, MZ_ZIP_TOO_MANY_FILES);
3646 cdir_entries_on_this_disk = (mz_uint32)zip64_cdir_total_entries_on_this_disk;
3649 if (zip64_size_of_central_directory > MZ_UINT32_MAX)
3650 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_CDIR_SIZE);
3652 cdir_size = (mz_uint32)zip64_size_of_central_directory;
3654 num_this_disk = MZ_READ_LE32(pZip64_end_of_central_dir + MZ_ZIP64_ECDH_NUM_THIS_DISK_OFS);
3656 cdir_disk_index = MZ_READ_LE32(pZip64_end_of_central_dir + MZ_ZIP64_ECDH_NUM_DISK_CDIR_OFS);
3658 cdir_ofs = MZ_READ_LE64(pZip64_end_of_central_dir + MZ_ZIP64_ECDH_CDIR_OFS_OFS);
3661 if (pZip->m_total_files != cdir_entries_on_this_disk)
3662 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_MULTIDISK);
3664 if (((num_this_disk | cdir_disk_index) != 0) && ((num_this_disk != 1) || (cdir_disk_index != 1)))
3665 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_MULTIDISK);
3667 if (cdir_size < pZip->m_total_files * MZ_ZIP_CENTRAL_DIR_HEADER_SIZE)
3668 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
3670 if ((cdir_ofs + (mz_uint64)cdir_size) > pZip->m_archive_size)
3671 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
3673 pZip->m_central_directory_file_ofs = cdir_ofs;
3675 if (pZip->m_total_files)
3679 if ((!mz_zip_array_resize(pZip, &pZip->m_pState->m_central_dir, cdir_size, MZ_FALSE)) ||
3680 (!mz_zip_array_resize(pZip, &pZip->m_pState->m_central_dir_offsets, pZip->m_total_files, MZ_FALSE)))
3681 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
3683 if (sort_central_dir)
3685 if (!mz_zip_array_resize(pZip, &pZip->m_pState->m_sorted_central_dir_offsets, pZip->m_total_files, MZ_FALSE))
3686 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
3689 if (pZip->m_pRead(pZip->m_pIO_opaque, cdir_ofs, pZip->m_pState->m_central_dir.m_p, cdir_size) != cdir_size)
3690 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
3693 p = (
const mz_uint8 *)pZip->m_pState->m_central_dir.m_p;
3694 for (n = cdir_size, i = 0; i < pZip->m_total_files; ++i)
3696 mz_uint total_header_size, disk_index, bit_flags, filename_size, ext_data_size;
3697 mz_uint64 comp_size, decomp_size, local_header_ofs;
3699 if ((n < MZ_ZIP_CENTRAL_DIR_HEADER_SIZE) || (MZ_READ_LE32(p) != MZ_ZIP_CENTRAL_DIR_HEADER_SIG))
3700 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
3702 MZ_ZIP_ARRAY_ELEMENT(&pZip->m_pState->m_central_dir_offsets, mz_uint32, i) = (mz_uint32)(p - (
const mz_uint8 *)pZip->m_pState->m_central_dir.m_p);
3704 if (sort_central_dir)
3705 MZ_ZIP_ARRAY_ELEMENT(&pZip->m_pState->m_sorted_central_dir_offsets, mz_uint32, i) = i;
3707 comp_size = MZ_READ_LE32(p + MZ_ZIP_CDH_COMPRESSED_SIZE_OFS);
3708 decomp_size = MZ_READ_LE32(p + MZ_ZIP_CDH_DECOMPRESSED_SIZE_OFS);
3709 local_header_ofs = MZ_READ_LE32(p + MZ_ZIP_CDH_LOCAL_HEADER_OFS);
3710 filename_size = MZ_READ_LE16(p + MZ_ZIP_CDH_FILENAME_LEN_OFS);
3711 ext_data_size = MZ_READ_LE16(p + MZ_ZIP_CDH_EXTRA_LEN_OFS);
3713 if ((!pZip->m_pState->m_zip64_has_extended_info_fields) &&
3715 (MZ_MAX(MZ_MAX(comp_size, decomp_size), local_header_ofs) == MZ_UINT32_MAX))
3718 mz_uint32 extra_size_remaining = ext_data_size;
3720 if (extra_size_remaining)
3722 const mz_uint8 *pExtra_data;
3725 if (MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + filename_size + ext_data_size > n)
3727 buf = MZ_MALLOC(ext_data_size);
3729 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
3731 if (pZip->m_pRead(pZip->m_pIO_opaque, cdir_ofs + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + filename_size, buf, ext_data_size) != ext_data_size)
3734 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
3737 pExtra_data = (mz_uint8*)buf;
3741 pExtra_data = p + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + filename_size;
3747 mz_uint32 field_data_size;
3749 if (extra_size_remaining < (
sizeof(mz_uint16) * 2))
3752 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
3755 field_id = MZ_READ_LE16(pExtra_data);
3756 field_data_size = MZ_READ_LE16(pExtra_data +
sizeof(mz_uint16));
3758 if ((field_data_size +
sizeof(mz_uint16) * 2) > extra_size_remaining)
3761 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
3764 if (field_id == MZ_ZIP64_EXTENDED_INFORMATION_FIELD_HEADER_ID)
3767 pZip->m_pState->m_zip64 = MZ_TRUE;
3768 pZip->m_pState->m_zip64_has_extended_info_fields = MZ_TRUE;
3772 pExtra_data +=
sizeof(mz_uint16) * 2 + field_data_size;
3773 extra_size_remaining = extra_size_remaining -
sizeof(mz_uint16) * 2 - field_data_size;
3774 }
while (extra_size_remaining);
3781 if ((comp_size != MZ_UINT32_MAX) && (decomp_size != MZ_UINT32_MAX))
3783 if (((!MZ_READ_LE32(p + MZ_ZIP_CDH_METHOD_OFS)) && (decomp_size != comp_size)) || (decomp_size && !comp_size))
3784 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
3787 disk_index = MZ_READ_LE16(p + MZ_ZIP_CDH_DISK_START_OFS);
3788 if ((disk_index == MZ_UINT16_MAX) || ((disk_index != num_this_disk) && (disk_index != 1)))
3789 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_MULTIDISK);
3791 if (comp_size != MZ_UINT32_MAX)
3793 if (((mz_uint64)MZ_READ_LE32(p + MZ_ZIP_CDH_LOCAL_HEADER_OFS) + MZ_ZIP_LOCAL_DIR_HEADER_SIZE + comp_size) > pZip->m_archive_size)
3794 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
3797 bit_flags = MZ_READ_LE16(p + MZ_ZIP_CDH_BIT_FLAG_OFS);
3798 if (bit_flags & MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_LOCAL_DIR_IS_MASKED)
3799 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_ENCRYPTION);
3801 if ((total_header_size = MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + MZ_READ_LE16(p + MZ_ZIP_CDH_FILENAME_LEN_OFS) + MZ_READ_LE16(p + MZ_ZIP_CDH_EXTRA_LEN_OFS) + MZ_READ_LE16(p + MZ_ZIP_CDH_COMMENT_LEN_OFS)) > n)
3802 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
3804 n -= total_header_size;
3805 p += total_header_size;
3809 if (sort_central_dir)
3810 mz_zip_reader_sort_central_dir_offsets_by_filename(pZip);
3818 MZ_CLEAR_OBJ(*pZip);
3821static mz_bool mz_zip_reader_end_internal(
mz_zip_archive *pZip, mz_bool set_last_error)
3823 mz_bool status = MZ_TRUE;
3828 if ((!pZip->m_pState) || (!pZip->m_pAlloc) || (!pZip->m_pFree) || (pZip->m_zip_mode != MZ_ZIP_MODE_READING))
3831 pZip->m_last_error = MZ_ZIP_INVALID_PARAMETER;
3839 pZip->m_pState = NULL;
3841 mz_zip_array_clear(pZip, &pState->m_central_dir);
3842 mz_zip_array_clear(pZip, &pState->m_central_dir_offsets);
3843 mz_zip_array_clear(pZip, &pState->m_sorted_central_dir_offsets);
3845#ifndef MINIZ_NO_STDIO
3846 if (pState->m_pFile)
3848 if (pZip->m_zip_type == MZ_ZIP_TYPE_FILE)
3850 if (MZ_FCLOSE(pState->m_pFile) == EOF)
3853 pZip->m_last_error = MZ_ZIP_FILE_CLOSE_FAILED;
3857 pState->m_pFile = NULL;
3861 pZip->m_pFree(pZip->m_pAlloc_opaque, pState);
3863 pZip->m_zip_mode = MZ_ZIP_MODE_INVALID;
3870 return mz_zip_reader_end_internal(pZip, MZ_TRUE);
3872mz_bool mz_zip_reader_init(
mz_zip_archive *pZip, mz_uint64 size, mz_uint flags)
3874 if ((!pZip) || (!pZip->m_pRead))
3875 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
3877 if (!mz_zip_reader_init_internal(pZip, flags))
3880 pZip->m_zip_type = MZ_ZIP_TYPE_USER;
3881 pZip->m_archive_size = size;
3883 if (!mz_zip_reader_read_central_dir(pZip, flags))
3885 mz_zip_reader_end_internal(pZip, MZ_FALSE);
3892static size_t mz_zip_mem_read_func(
void *pOpaque, mz_uint64 file_ofs,
void *pBuf,
size_t n)
3895 size_t s = (file_ofs >= pZip->m_archive_size) ? 0 : (
size_t)MZ_MIN(pZip->m_archive_size - file_ofs, n);
3896 memcpy(pBuf, (
const mz_uint8 *)pZip->m_pState->m_pMem + file_ofs, s);
3900mz_bool mz_zip_reader_init_mem(
mz_zip_archive *pZip,
const void *pMem,
size_t size, mz_uint flags)
3903 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
3905 if (size < MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE)
3906 return mz_zip_set_error(pZip, MZ_ZIP_NOT_AN_ARCHIVE);
3908 if (!mz_zip_reader_init_internal(pZip, flags))
3911 pZip->m_zip_type = MZ_ZIP_TYPE_MEMORY;
3912 pZip->m_archive_size = size;
3913 pZip->m_pRead = mz_zip_mem_read_func;
3914 pZip->m_pIO_opaque = pZip;
3915 pZip->m_pNeeds_keepalive = NULL;
3918 pZip->m_pState->m_pMem =
const_cast<void *
>(pMem);
3920 pZip->m_pState->m_pMem = (
void *)pMem;
3923 pZip->m_pState->m_mem_size = size;
3925 if (!mz_zip_reader_read_central_dir(pZip, flags))
3927 mz_zip_reader_end_internal(pZip, MZ_FALSE);
3934#ifndef MINIZ_NO_STDIO
3935static size_t mz_zip_file_read_func(
void *pOpaque, mz_uint64 file_ofs,
void *pBuf,
size_t n)
3938 mz_int64 cur_ofs = MZ_FTELL64(pZip->m_pState->m_pFile);
3940 file_ofs += pZip->m_pState->m_file_archive_start_ofs;
3942 if (((mz_int64)file_ofs < 0) || (((cur_ofs != (mz_int64)file_ofs)) && (MZ_FSEEK64(pZip->m_pState->m_pFile, (mz_int64)file_ofs, SEEK_SET))))
3945 return MZ_FREAD(pBuf, 1, n, pZip->m_pState->m_pFile);
3948mz_bool mz_zip_reader_init_file(
mz_zip_archive *pZip,
const char *pFilename, mz_uint32 flags)
3950 return mz_zip_reader_init_file_v2(pZip, pFilename, flags, 0, 0);
3953mz_bool mz_zip_reader_init_file_v2(
mz_zip_archive *pZip,
const char *pFilename, mz_uint flags, mz_uint64 file_start_ofs, mz_uint64 archive_size)
3955 mz_uint64 file_size;
3958 if ((!pZip) || (!pFilename) || ((archive_size) && (archive_size < MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE)))
3959 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
3961 pFile = MZ_FOPEN(pFilename,
"rb");
3963 return mz_zip_set_error(pZip, MZ_ZIP_FILE_OPEN_FAILED);
3965 file_size = archive_size;
3968 if (MZ_FSEEK64(pFile, 0, SEEK_END))
3971 return mz_zip_set_error(pZip, MZ_ZIP_FILE_SEEK_FAILED);
3974 file_size = MZ_FTELL64(pFile);
3979 if (file_size < MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE)
3982 return mz_zip_set_error(pZip, MZ_ZIP_NOT_AN_ARCHIVE);
3985 if (!mz_zip_reader_init_internal(pZip, flags))
3991 pZip->m_zip_type = MZ_ZIP_TYPE_FILE;
3992 pZip->m_pRead = mz_zip_file_read_func;
3993 pZip->m_pIO_opaque = pZip;
3994 pZip->m_pState->m_pFile = pFile;
3995 pZip->m_archive_size = file_size;
3996 pZip->m_pState->m_file_archive_start_ofs = file_start_ofs;
3998 if (!mz_zip_reader_read_central_dir(pZip, flags))
4000 mz_zip_reader_end_internal(pZip, MZ_FALSE);
4007mz_bool mz_zip_reader_init_cfile(
mz_zip_archive *pZip, MZ_FILE *pFile, mz_uint64 archive_size, mz_uint flags)
4009 mz_uint64 cur_file_ofs;
4011 if ((!pZip) || (!pFile))
4012 return mz_zip_set_error(pZip, MZ_ZIP_FILE_OPEN_FAILED);
4014 cur_file_ofs = MZ_FTELL64(pFile);
4018 if (MZ_FSEEK64(pFile, 0, SEEK_END))
4019 return mz_zip_set_error(pZip, MZ_ZIP_FILE_SEEK_FAILED);
4021 archive_size = MZ_FTELL64(pFile) - cur_file_ofs;
4023 if (archive_size < MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE)
4024 return mz_zip_set_error(pZip, MZ_ZIP_NOT_AN_ARCHIVE);
4027 if (!mz_zip_reader_init_internal(pZip, flags))
4030 pZip->m_zip_type = MZ_ZIP_TYPE_CFILE;
4031 pZip->m_pRead = mz_zip_file_read_func;
4033 pZip->m_pIO_opaque = pZip;
4034 pZip->m_pState->m_pFile = pFile;
4035 pZip->m_archive_size = archive_size;
4036 pZip->m_pState->m_file_archive_start_ofs = cur_file_ofs;
4038 if (!mz_zip_reader_read_central_dir(pZip, flags))
4040 mz_zip_reader_end_internal(pZip, MZ_FALSE);
4049static MZ_FORCEINLINE
const mz_uint8 *mz_zip_get_cdh(
mz_zip_archive *pZip, mz_uint file_index)
4051 if ((!pZip) || (!pZip->m_pState) || (file_index >= pZip->m_total_files))
4053 return &MZ_ZIP_ARRAY_ELEMENT(&pZip->m_pState->m_central_dir, mz_uint8, MZ_ZIP_ARRAY_ELEMENT(&pZip->m_pState->m_central_dir_offsets, mz_uint32, file_index));
4056mz_bool mz_zip_reader_is_file_encrypted(
mz_zip_archive *pZip, mz_uint file_index)
4059 const mz_uint8 *p = mz_zip_get_cdh(pZip, file_index);
4062 mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
4066 m_bit_flag = MZ_READ_LE16(p + MZ_ZIP_CDH_BIT_FLAG_OFS);
4067 return (m_bit_flag & (MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_IS_ENCRYPTED | MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_USES_STRONG_ENCRYPTION)) != 0;
4070mz_bool mz_zip_reader_is_file_supported(
mz_zip_archive *pZip, mz_uint file_index)
4075 const mz_uint8 *p = mz_zip_get_cdh(pZip, file_index);
4078 mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
4082 method = MZ_READ_LE16(p + MZ_ZIP_CDH_METHOD_OFS);
4083 bit_flag = MZ_READ_LE16(p + MZ_ZIP_CDH_BIT_FLAG_OFS);
4085 if ((method != 0) && (method != MZ_DEFLATED))
4087 mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_METHOD);
4091 if (bit_flag & (MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_IS_ENCRYPTED | MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_USES_STRONG_ENCRYPTION))
4093 mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_ENCRYPTION);
4097 if (bit_flag & MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_COMPRESSED_PATCH_FLAG)
4099 mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_FEATURE);
4106mz_bool mz_zip_reader_is_file_a_directory(
mz_zip_archive *pZip, mz_uint file_index)
4108 mz_uint filename_len, attribute_mapping_id, external_attr;
4109 const mz_uint8 *p = mz_zip_get_cdh(pZip, file_index);
4112 mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
4116 filename_len = MZ_READ_LE16(p + MZ_ZIP_CDH_FILENAME_LEN_OFS);
4119 if (*(p + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + filename_len - 1) ==
'/')
4126 attribute_mapping_id = MZ_READ_LE16(p + MZ_ZIP_CDH_VERSION_MADE_BY_OFS) >> 8;
4127 (void)attribute_mapping_id;
4129 external_attr = MZ_READ_LE32(p + MZ_ZIP_CDH_EXTERNAL_ATTR_OFS);
4130 if ((external_attr & MZ_ZIP_DOS_DIR_ATTRIBUTE_BITFLAG) != 0)
4138static mz_bool mz_zip_file_stat_internal(
mz_zip_archive *pZip, mz_uint file_index,
const mz_uint8 *pCentral_dir_header,
mz_zip_archive_file_stat *pStat, mz_bool *pFound_zip64_extra_data)
4141 const mz_uint8 *p = pCentral_dir_header;
4143 if (pFound_zip64_extra_data)
4144 *pFound_zip64_extra_data = MZ_FALSE;
4146 if ((!p) || (!pStat))
4147 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
4150 pStat->m_file_index = file_index;
4151 pStat->m_central_dir_ofs = MZ_ZIP_ARRAY_ELEMENT(&pZip->m_pState->m_central_dir_offsets, mz_uint32, file_index);
4152 pStat->m_version_made_by = MZ_READ_LE16(p + MZ_ZIP_CDH_VERSION_MADE_BY_OFS);
4153 pStat->m_version_needed = MZ_READ_LE16(p + MZ_ZIP_CDH_VERSION_NEEDED_OFS);
4154 pStat->m_bit_flag = MZ_READ_LE16(p + MZ_ZIP_CDH_BIT_FLAG_OFS);
4155 pStat->m_method = MZ_READ_LE16(p + MZ_ZIP_CDH_METHOD_OFS);
4156#ifndef MINIZ_NO_TIME
4157 pStat->m_time = mz_zip_dos_to_time_t(MZ_READ_LE16(p + MZ_ZIP_CDH_FILE_TIME_OFS), MZ_READ_LE16(p + MZ_ZIP_CDH_FILE_DATE_OFS));
4159 pStat->m_crc32 = MZ_READ_LE32(p + MZ_ZIP_CDH_CRC32_OFS);
4160 pStat->m_comp_size = MZ_READ_LE32(p + MZ_ZIP_CDH_COMPRESSED_SIZE_OFS);
4161 pStat->m_uncomp_size = MZ_READ_LE32(p + MZ_ZIP_CDH_DECOMPRESSED_SIZE_OFS);
4162 pStat->m_internal_attr = MZ_READ_LE16(p + MZ_ZIP_CDH_INTERNAL_ATTR_OFS);
4163 pStat->m_external_attr = MZ_READ_LE32(p + MZ_ZIP_CDH_EXTERNAL_ATTR_OFS);
4164 pStat->m_local_header_ofs = MZ_READ_LE32(p + MZ_ZIP_CDH_LOCAL_HEADER_OFS);
4167 n = MZ_READ_LE16(p + MZ_ZIP_CDH_FILENAME_LEN_OFS);
4168 n = MZ_MIN(n, MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE - 1);
4169 memcpy(pStat->m_filename, p + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE, n);
4170 pStat->m_filename[n] =
'\0';
4172 n = MZ_READ_LE16(p + MZ_ZIP_CDH_COMMENT_LEN_OFS);
4173 n = MZ_MIN(n, MZ_ZIP_MAX_ARCHIVE_FILE_COMMENT_SIZE - 1);
4174 pStat->m_comment_size = n;
4175 memcpy(pStat->m_comment, p + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + MZ_READ_LE16(p + MZ_ZIP_CDH_FILENAME_LEN_OFS) + MZ_READ_LE16(p + MZ_ZIP_CDH_EXTRA_LEN_OFS), n);
4176 pStat->m_comment[n] =
'\0';
4179 pStat->m_is_directory = mz_zip_reader_is_file_a_directory(pZip, file_index);
4180 pStat->m_is_encrypted = mz_zip_reader_is_file_encrypted(pZip, file_index);
4181 pStat->m_is_supported = mz_zip_reader_is_file_supported(pZip, file_index);
4185 if (MZ_MAX(MZ_MAX(pStat->m_comp_size, pStat->m_uncomp_size), pStat->m_local_header_ofs) == MZ_UINT32_MAX)
4188 mz_uint32 extra_size_remaining = MZ_READ_LE16(p + MZ_ZIP_CDH_EXTRA_LEN_OFS);
4190 if (extra_size_remaining)
4192 const mz_uint8 *pExtra_data = p + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + MZ_READ_LE16(p + MZ_ZIP_CDH_FILENAME_LEN_OFS);
4197 mz_uint32 field_data_size;
4199 if (extra_size_remaining < (
sizeof(mz_uint16) * 2))
4200 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
4202 field_id = MZ_READ_LE16(pExtra_data);
4203 field_data_size = MZ_READ_LE16(pExtra_data +
sizeof(mz_uint16));
4205 if ((field_data_size +
sizeof(mz_uint16) * 2) > extra_size_remaining)
4206 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
4208 if (field_id == MZ_ZIP64_EXTENDED_INFORMATION_FIELD_HEADER_ID)
4210 const mz_uint8 *pField_data = pExtra_data +
sizeof(mz_uint16) * 2;
4211 mz_uint32 field_data_remaining = field_data_size;
4213 if (pFound_zip64_extra_data)
4214 *pFound_zip64_extra_data = MZ_TRUE;
4216 if (pStat->m_uncomp_size == MZ_UINT32_MAX)
4218 if (field_data_remaining <
sizeof(mz_uint64))
4219 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
4221 pStat->m_uncomp_size = MZ_READ_LE64(pField_data);
4222 pField_data +=
sizeof(mz_uint64);
4223 field_data_remaining -=
sizeof(mz_uint64);
4226 if (pStat->m_comp_size == MZ_UINT32_MAX)
4228 if (field_data_remaining <
sizeof(mz_uint64))
4229 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
4231 pStat->m_comp_size = MZ_READ_LE64(pField_data);
4232 pField_data +=
sizeof(mz_uint64);
4233 field_data_remaining -=
sizeof(mz_uint64);
4236 if (pStat->m_local_header_ofs == MZ_UINT32_MAX)
4238 if (field_data_remaining <
sizeof(mz_uint64))
4239 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
4241 pStat->m_local_header_ofs = MZ_READ_LE64(pField_data);
4242 pField_data +=
sizeof(mz_uint64);
4243 field_data_remaining -=
sizeof(mz_uint64);
4249 pExtra_data +=
sizeof(mz_uint16) * 2 + field_data_size;
4250 extra_size_remaining = extra_size_remaining -
sizeof(mz_uint16) * 2 - field_data_size;
4251 }
while (extra_size_remaining);
4258static MZ_FORCEINLINE mz_bool mz_zip_string_equal(
const char *pA,
const char *pB, mz_uint len, mz_uint flags)
4261 if (flags & MZ_ZIP_FLAG_CASE_SENSITIVE)
4262 return 0 == memcmp(pA, pB, len);
4263 for (i = 0; i < len; ++i)
4264 if (MZ_TOLOWER(pA[i]) != MZ_TOLOWER(pB[i]))
4269static MZ_FORCEINLINE
int mz_zip_filename_compare(
const mz_zip_array *pCentral_dir_array,
const mz_zip_array *pCentral_dir_offsets, mz_uint l_index,
const char *pR, mz_uint r_len)
4271 const mz_uint8 *pL = &MZ_ZIP_ARRAY_ELEMENT(pCentral_dir_array, mz_uint8, MZ_ZIP_ARRAY_ELEMENT(pCentral_dir_offsets, mz_uint32, l_index)), *pE;
4272 mz_uint l_len = MZ_READ_LE16(pL + MZ_ZIP_CDH_FILENAME_LEN_OFS);
4273 mz_uint8 l = 0, r = 0;
4274 pL += MZ_ZIP_CENTRAL_DIR_HEADER_SIZE;
4275 pE = pL + MZ_MIN(l_len, r_len);
4278 if ((l = MZ_TOLOWER(*pL)) != (r = MZ_TOLOWER(*pR)))
4283 return (pL == pE) ? (int)(l_len - r_len) : (l - r);
4286static mz_bool mz_zip_locate_file_binary_search(
mz_zip_archive *pZip,
const char *pFilename, mz_uint32 *pIndex)
4289 const mz_zip_array *pCentral_dir_offsets = &pState->m_central_dir_offsets;
4290 const mz_zip_array *pCentral_dir = &pState->m_central_dir;
4291 mz_uint32 *pIndices = &MZ_ZIP_ARRAY_ELEMENT(&pState->m_sorted_central_dir_offsets, mz_uint32, 0);
4292 const uint32_t size = pZip->m_total_files;
4293 const mz_uint filename_len = (mz_uint)strlen(pFilename);
4302 mz_int64 l = 0, h = (mz_int64)size - 1;
4306 mz_int64 m = l + ((h - l) >> 1);
4307 uint32_t file_index = pIndices[(uint32_t)m];
4309 int comp = mz_zip_filename_compare(pCentral_dir, pCentral_dir_offsets, file_index, pFilename, filename_len);
4313 *pIndex = file_index;
4323 return mz_zip_set_error(pZip, MZ_ZIP_FILE_NOT_FOUND);
4326int mz_zip_reader_locate_file(
mz_zip_archive *pZip,
const char *pName,
const char *pComment, mz_uint flags)
4329 if (!mz_zip_reader_locate_file_v2(pZip, pName, pComment, flags, &index))
4335mz_bool mz_zip_reader_locate_file_v2(
mz_zip_archive *pZip,
const char *pName,
const char *pComment, mz_uint flags, mz_uint32 *pIndex)
4338 size_t name_len, comment_len;
4343 if ((!pZip) || (!pZip->m_pState) || (!pName))
4344 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
4347 if (((pZip->m_pState->m_init_flags & MZ_ZIP_FLAG_DO_NOT_SORT_CENTRAL_DIRECTORY) == 0) &&
4348 (pZip->m_zip_mode == MZ_ZIP_MODE_READING) &&
4349 ((flags & (MZ_ZIP_FLAG_IGNORE_PATH | MZ_ZIP_FLAG_CASE_SENSITIVE)) == 0) && (!pComment) && (pZip->m_pState->m_sorted_central_dir_offsets.m_size))
4351 return mz_zip_locate_file_binary_search(pZip, pName, pIndex);
4355 name_len = strlen(pName);
4356 if (name_len > MZ_UINT16_MAX)
4357 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
4359 comment_len = pComment ? strlen(pComment) : 0;
4360 if (comment_len > MZ_UINT16_MAX)
4361 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
4363 for (file_index = 0; file_index < pZip->m_total_files; file_index++)
4365 const mz_uint8 *pHeader = &MZ_ZIP_ARRAY_ELEMENT(&pZip->m_pState->m_central_dir, mz_uint8, MZ_ZIP_ARRAY_ELEMENT(&pZip->m_pState->m_central_dir_offsets, mz_uint32, file_index));
4366 mz_uint filename_len = MZ_READ_LE16(pHeader + MZ_ZIP_CDH_FILENAME_LEN_OFS);
4367 const char *pFilename = (
const char *)pHeader + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE;
4368 if (filename_len < name_len)
4372 mz_uint file_extra_len = MZ_READ_LE16(pHeader + MZ_ZIP_CDH_EXTRA_LEN_OFS), file_comment_len = MZ_READ_LE16(pHeader + MZ_ZIP_CDH_COMMENT_LEN_OFS);
4373 const char *pFile_comment = pFilename + filename_len + file_extra_len;
4374 if ((file_comment_len != comment_len) || (!mz_zip_string_equal(pComment, pFile_comment, file_comment_len, flags)))
4377 if ((flags & MZ_ZIP_FLAG_IGNORE_PATH) && (filename_len))
4379 int ofs = filename_len - 1;
4382 if ((pFilename[ofs] ==
'/') || (pFilename[ofs] ==
'\\') || (pFilename[ofs] ==
':'))
4384 }
while (--ofs >= 0);
4387 filename_len -= ofs;
4389 if ((filename_len == name_len) && (mz_zip_string_equal(pName, pFilename, filename_len, flags)))
4392 *pIndex = file_index;
4397 return mz_zip_set_error(pZip, MZ_ZIP_FILE_NOT_FOUND);
4400mz_bool mz_zip_reader_extract_to_mem_no_alloc(
mz_zip_archive *pZip, mz_uint file_index,
void *pBuf,
size_t buf_size, mz_uint flags,
void *pUser_read_buf,
size_t user_read_buf_size)
4402 int status = TINFL_STATUS_DONE;
4403 mz_uint64 needed_size, cur_file_ofs, comp_remaining, out_buf_ofs = 0, read_buf_size, read_buf_ofs = 0, read_buf_avail;
4406 mz_uint32 local_header_u32[(MZ_ZIP_LOCAL_DIR_HEADER_SIZE +
sizeof(mz_uint32) - 1) /
sizeof(mz_uint32)];
4407 mz_uint8 *pLocal_header = (mz_uint8 *)local_header_u32;
4410 if ((!pZip) || (!pZip->m_pState) || ((buf_size) && (!pBuf)) || ((user_read_buf_size) && (!pUser_read_buf)) || (!pZip->m_pRead))
4411 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
4413 if (!mz_zip_reader_file_stat(pZip, file_index, &file_stat))
4417 if ((file_stat.m_is_directory) || (!file_stat.m_comp_size))
4421 if (file_stat.m_bit_flag & (MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_IS_ENCRYPTED | MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_USES_STRONG_ENCRYPTION | MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_COMPRESSED_PATCH_FLAG))
4422 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_ENCRYPTION);
4425 if ((!(flags & MZ_ZIP_FLAG_COMPRESSED_DATA)) && (file_stat.m_method != 0) && (file_stat.m_method != MZ_DEFLATED))
4426 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_METHOD);
4429 needed_size = (flags & MZ_ZIP_FLAG_COMPRESSED_DATA) ? file_stat.m_comp_size : file_stat.m_uncomp_size;
4430 if (buf_size < needed_size)
4431 return mz_zip_set_error(pZip, MZ_ZIP_BUF_TOO_SMALL);
4434 cur_file_ofs = file_stat.m_local_header_ofs;
4435 if (pZip->m_pRead(pZip->m_pIO_opaque, cur_file_ofs, pLocal_header, MZ_ZIP_LOCAL_DIR_HEADER_SIZE) != MZ_ZIP_LOCAL_DIR_HEADER_SIZE)
4436 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
4438 if (MZ_READ_LE32(pLocal_header) != MZ_ZIP_LOCAL_DIR_HEADER_SIG)
4439 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
4441 cur_file_ofs += MZ_ZIP_LOCAL_DIR_HEADER_SIZE + MZ_READ_LE16(pLocal_header + MZ_ZIP_LDH_FILENAME_LEN_OFS) + MZ_READ_LE16(pLocal_header + MZ_ZIP_LDH_EXTRA_LEN_OFS);
4442 if ((cur_file_ofs + file_stat.m_comp_size) > pZip->m_archive_size)
4443 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
4445 if ((flags & MZ_ZIP_FLAG_COMPRESSED_DATA) || (!file_stat.m_method))
4448 if (pZip->m_pRead(pZip->m_pIO_opaque, cur_file_ofs, pBuf, (
size_t)needed_size) != needed_size)
4449 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
4451#ifndef MINIZ_DISABLE_ZIP_READER_CRC32_CHECKS
4452 if ((flags & MZ_ZIP_FLAG_COMPRESSED_DATA) == 0)
4454 if (mz_crc32(MZ_CRC32_INIT, (
const mz_uint8 *)pBuf, (
size_t)file_stat.m_uncomp_size) != file_stat.m_crc32)
4455 return mz_zip_set_error(pZip, MZ_ZIP_CRC_CHECK_FAILED);
4463 tinfl_init(&inflator);
4465 if (pZip->m_pState->m_pMem)
4468 pRead_buf = (mz_uint8 *)pZip->m_pState->m_pMem + cur_file_ofs;
4469 read_buf_size = read_buf_avail = file_stat.m_comp_size;
4472 else if (pUser_read_buf)
4475 if (!user_read_buf_size)
4477 pRead_buf = (mz_uint8 *)pUser_read_buf;
4478 read_buf_size = user_read_buf_size;
4480 comp_remaining = file_stat.m_comp_size;
4485 read_buf_size = MZ_MIN(file_stat.m_comp_size, (mz_uint64)MZ_ZIP_MAX_IO_BUF_SIZE);
4486 if (((
sizeof(
size_t) ==
sizeof(mz_uint32))) && (read_buf_size > 0x7FFFFFFF))
4487 return mz_zip_set_error(pZip, MZ_ZIP_INTERNAL_ERROR);
4489 if (NULL == (pRead_buf = pZip->m_pAlloc(pZip->m_pAlloc_opaque, 1, (
size_t)read_buf_size)))
4490 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
4493 comp_remaining = file_stat.m_comp_size;
4499 size_t in_buf_size, out_buf_size = (size_t)(file_stat.m_uncomp_size - out_buf_ofs);
4500 if ((!read_buf_avail) && (!pZip->m_pState->m_pMem))
4502 read_buf_avail = MZ_MIN(read_buf_size, comp_remaining);
4503 if (pZip->m_pRead(pZip->m_pIO_opaque, cur_file_ofs, pRead_buf, (
size_t)read_buf_avail) != read_buf_avail)
4505 status = TINFL_STATUS_FAILED;
4506 mz_zip_set_error(pZip, MZ_ZIP_DECOMPRESSION_FAILED);
4509 cur_file_ofs += read_buf_avail;
4510 comp_remaining -= read_buf_avail;
4513 in_buf_size = (size_t)read_buf_avail;
4514 status = tinfl_decompress(&inflator, (mz_uint8 *)pRead_buf + read_buf_ofs, &in_buf_size, (mz_uint8 *)pBuf, (mz_uint8 *)pBuf + out_buf_ofs, &out_buf_size, TINFL_FLAG_USING_NON_WRAPPING_OUTPUT_BUF | (comp_remaining ? TINFL_FLAG_HAS_MORE_INPUT : 0));
4515 read_buf_avail -= in_buf_size;
4516 read_buf_ofs += in_buf_size;
4517 out_buf_ofs += out_buf_size;
4518 }
while (status == TINFL_STATUS_NEEDS_MORE_INPUT);
4520 if (status == TINFL_STATUS_DONE)
4523 if (out_buf_ofs != file_stat.m_uncomp_size)
4525 mz_zip_set_error(pZip, MZ_ZIP_UNEXPECTED_DECOMPRESSED_SIZE);
4526 status = TINFL_STATUS_FAILED;
4528#ifndef MINIZ_DISABLE_ZIP_READER_CRC32_CHECKS
4529 else if (mz_crc32(MZ_CRC32_INIT, (
const mz_uint8 *)pBuf, (
size_t)file_stat.m_uncomp_size) != file_stat.m_crc32)
4531 mz_zip_set_error(pZip, MZ_ZIP_CRC_CHECK_FAILED);
4532 status = TINFL_STATUS_FAILED;
4537 if ((!pZip->m_pState->m_pMem) && (!pUser_read_buf))
4538 pZip->m_pFree(pZip->m_pAlloc_opaque, pRead_buf);
4540 return status == TINFL_STATUS_DONE;
4543mz_bool mz_zip_reader_extract_file_to_mem_no_alloc(
mz_zip_archive *pZip,
const char *pFilename,
void *pBuf,
size_t buf_size, mz_uint flags,
void *pUser_read_buf,
size_t user_read_buf_size)
4545 mz_uint32 file_index;
4546 if (!mz_zip_reader_locate_file_v2(pZip, pFilename, NULL, flags, &file_index))
4548 return mz_zip_reader_extract_to_mem_no_alloc(pZip, file_index, pBuf, buf_size, flags, pUser_read_buf, user_read_buf_size);
4551mz_bool mz_zip_reader_extract_to_mem(
mz_zip_archive *pZip, mz_uint file_index,
void *pBuf,
size_t buf_size, mz_uint flags)
4553 return mz_zip_reader_extract_to_mem_no_alloc(pZip, file_index, pBuf, buf_size, flags, NULL, 0);
4556mz_bool mz_zip_reader_extract_file_to_mem(
mz_zip_archive *pZip,
const char *pFilename,
void *pBuf,
size_t buf_size, mz_uint flags)
4558 return mz_zip_reader_extract_file_to_mem_no_alloc(pZip, pFilename, pBuf, buf_size, flags, NULL, 0);
4561void *mz_zip_reader_extract_to_heap(
mz_zip_archive *pZip, mz_uint file_index,
size_t *pSize, mz_uint flags)
4563 mz_uint64 comp_size, uncomp_size, alloc_size;
4564 const mz_uint8 *p = mz_zip_get_cdh(pZip, file_index);
4572 mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
4576 comp_size = MZ_READ_LE32(p + MZ_ZIP_CDH_COMPRESSED_SIZE_OFS);
4577 uncomp_size = MZ_READ_LE32(p + MZ_ZIP_CDH_DECOMPRESSED_SIZE_OFS);
4579 alloc_size = (flags & MZ_ZIP_FLAG_COMPRESSED_DATA) ? comp_size : uncomp_size;
4580 if (((
sizeof(
size_t) ==
sizeof(mz_uint32))) && (alloc_size > 0x7FFFFFFF))
4582 mz_zip_set_error(pZip, MZ_ZIP_INTERNAL_ERROR);
4586 if (NULL == (pBuf = pZip->m_pAlloc(pZip->m_pAlloc_opaque, 1, (
size_t)alloc_size)))
4588 mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
4592 if (!mz_zip_reader_extract_to_mem(pZip, file_index, pBuf, (
size_t)alloc_size, flags))
4594 pZip->m_pFree(pZip->m_pAlloc_opaque, pBuf);
4599 *pSize = (size_t)alloc_size;
4603void *mz_zip_reader_extract_file_to_heap(
mz_zip_archive *pZip,
const char *pFilename,
size_t *pSize, mz_uint flags)
4605 mz_uint32 file_index;
4606 if (!mz_zip_reader_locate_file_v2(pZip, pFilename, NULL, flags, &file_index))
4612 return mz_zip_reader_extract_to_heap(pZip, file_index, pSize, flags);
4615mz_bool mz_zip_reader_extract_to_callback(
mz_zip_archive *pZip, mz_uint file_index, mz_file_write_func pCallback,
void *pOpaque, mz_uint flags)
4617 int status = TINFL_STATUS_DONE;
4618 mz_uint file_crc32 = MZ_CRC32_INIT;
4619 mz_uint64 read_buf_size, read_buf_ofs = 0, read_buf_avail, comp_remaining, out_buf_ofs = 0, cur_file_ofs;
4621 void *pRead_buf = NULL;
4622 void *pWrite_buf = NULL;
4623 mz_uint32 local_header_u32[(MZ_ZIP_LOCAL_DIR_HEADER_SIZE +
sizeof(mz_uint32) - 1) /
sizeof(mz_uint32)];
4624 mz_uint8 *pLocal_header = (mz_uint8 *)local_header_u32;
4626 if ((!pZip) || (!pZip->m_pState) || (!pCallback) || (!pZip->m_pRead))
4627 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
4629 if (!mz_zip_reader_file_stat(pZip, file_index, &file_stat))
4633 if ((file_stat.m_is_directory) || (!file_stat.m_comp_size))
4637 if (file_stat.m_bit_flag & (MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_IS_ENCRYPTED | MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_USES_STRONG_ENCRYPTION | MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_COMPRESSED_PATCH_FLAG))
4638 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_ENCRYPTION);
4641 if ((!(flags & MZ_ZIP_FLAG_COMPRESSED_DATA)) && (file_stat.m_method != 0) && (file_stat.m_method != MZ_DEFLATED))
4642 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_METHOD);
4645 cur_file_ofs = file_stat.m_local_header_ofs;
4646 if (pZip->m_pRead(pZip->m_pIO_opaque, cur_file_ofs, pLocal_header, MZ_ZIP_LOCAL_DIR_HEADER_SIZE) != MZ_ZIP_LOCAL_DIR_HEADER_SIZE)
4647 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
4649 if (MZ_READ_LE32(pLocal_header) != MZ_ZIP_LOCAL_DIR_HEADER_SIG)
4650 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
4652 cur_file_ofs += MZ_ZIP_LOCAL_DIR_HEADER_SIZE + MZ_READ_LE16(pLocal_header + MZ_ZIP_LDH_FILENAME_LEN_OFS) + MZ_READ_LE16(pLocal_header + MZ_ZIP_LDH_EXTRA_LEN_OFS);
4653 if ((cur_file_ofs + file_stat.m_comp_size) > pZip->m_archive_size)
4654 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
4657 if (pZip->m_pState->m_pMem)
4659 pRead_buf = (mz_uint8 *)pZip->m_pState->m_pMem + cur_file_ofs;
4660 read_buf_size = read_buf_avail = file_stat.m_comp_size;
4665 read_buf_size = MZ_MIN(file_stat.m_comp_size, (mz_uint64)MZ_ZIP_MAX_IO_BUF_SIZE);
4666 if (NULL == (pRead_buf = pZip->m_pAlloc(pZip->m_pAlloc_opaque, 1, (
size_t)read_buf_size)))
4667 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
4670 comp_remaining = file_stat.m_comp_size;
4673 if ((flags & MZ_ZIP_FLAG_COMPRESSED_DATA) || (!file_stat.m_method))
4676 if (pZip->m_pState->m_pMem)
4678 if (((
sizeof(
size_t) ==
sizeof(mz_uint32))) && (file_stat.m_comp_size > MZ_UINT32_MAX))
4679 return mz_zip_set_error(pZip, MZ_ZIP_INTERNAL_ERROR);
4681 if (pCallback(pOpaque, out_buf_ofs, pRead_buf, (
size_t)file_stat.m_comp_size) != file_stat.m_comp_size)
4683 mz_zip_set_error(pZip, MZ_ZIP_WRITE_CALLBACK_FAILED);
4684 status = TINFL_STATUS_FAILED;
4686 else if (!(flags & MZ_ZIP_FLAG_COMPRESSED_DATA))
4688#ifndef MINIZ_DISABLE_ZIP_READER_CRC32_CHECKS
4689 file_crc32 = (mz_uint32)mz_crc32(file_crc32, (
const mz_uint8 *)pRead_buf, (size_t)file_stat.m_comp_size);
4693 cur_file_ofs += file_stat.m_comp_size;
4694 out_buf_ofs += file_stat.m_comp_size;
4699 while (comp_remaining)
4701 read_buf_avail = MZ_MIN(read_buf_size, comp_remaining);
4702 if (pZip->m_pRead(pZip->m_pIO_opaque, cur_file_ofs, pRead_buf, (
size_t)read_buf_avail) != read_buf_avail)
4704 mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
4705 status = TINFL_STATUS_FAILED;
4709#ifndef MINIZ_DISABLE_ZIP_READER_CRC32_CHECKS
4710 if (!(flags & MZ_ZIP_FLAG_COMPRESSED_DATA))
4712 file_crc32 = (mz_uint32)mz_crc32(file_crc32, (
const mz_uint8 *)pRead_buf, (size_t)read_buf_avail);
4716 if (pCallback(pOpaque, out_buf_ofs, pRead_buf, (
size_t)read_buf_avail) != read_buf_avail)
4718 mz_zip_set_error(pZip, MZ_ZIP_WRITE_CALLBACK_FAILED);
4719 status = TINFL_STATUS_FAILED;
4723 cur_file_ofs += read_buf_avail;
4724 out_buf_ofs += read_buf_avail;
4725 comp_remaining -= read_buf_avail;
4732 tinfl_init(&inflator);
4734 if (NULL == (pWrite_buf = pZip->m_pAlloc(pZip->m_pAlloc_opaque, 1, TINFL_LZ_DICT_SIZE)))
4736 mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
4737 status = TINFL_STATUS_FAILED;
4743 mz_uint8 *pWrite_buf_cur = (mz_uint8 *)pWrite_buf + (out_buf_ofs & (TINFL_LZ_DICT_SIZE - 1));
4744 size_t in_buf_size, out_buf_size = TINFL_LZ_DICT_SIZE - (out_buf_ofs & (TINFL_LZ_DICT_SIZE - 1));
4745 if ((!read_buf_avail) && (!pZip->m_pState->m_pMem))
4747 read_buf_avail = MZ_MIN(read_buf_size, comp_remaining);
4748 if (pZip->m_pRead(pZip->m_pIO_opaque, cur_file_ofs, pRead_buf, (
size_t)read_buf_avail) != read_buf_avail)
4750 mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
4751 status = TINFL_STATUS_FAILED;
4754 cur_file_ofs += read_buf_avail;
4755 comp_remaining -= read_buf_avail;
4759 in_buf_size = (size_t)read_buf_avail;
4760 status = tinfl_decompress(&inflator, (
const mz_uint8 *)pRead_buf + read_buf_ofs, &in_buf_size, (mz_uint8 *)pWrite_buf, pWrite_buf_cur, &out_buf_size, comp_remaining ? TINFL_FLAG_HAS_MORE_INPUT : 0);
4761 read_buf_avail -= in_buf_size;
4762 read_buf_ofs += in_buf_size;
4766 if (pCallback(pOpaque, out_buf_ofs, pWrite_buf_cur, out_buf_size) != out_buf_size)
4768 mz_zip_set_error(pZip, MZ_ZIP_WRITE_CALLBACK_FAILED);
4769 status = TINFL_STATUS_FAILED;
4773#ifndef MINIZ_DISABLE_ZIP_READER_CRC32_CHECKS
4774 file_crc32 = (mz_uint32)mz_crc32(file_crc32, pWrite_buf_cur, out_buf_size);
4776 if ((out_buf_ofs += out_buf_size) > file_stat.m_uncomp_size)
4778 mz_zip_set_error(pZip, MZ_ZIP_DECOMPRESSION_FAILED);
4779 status = TINFL_STATUS_FAILED;
4783 }
while ((status == TINFL_STATUS_NEEDS_MORE_INPUT) || (status == TINFL_STATUS_HAS_MORE_OUTPUT));
4787 if ((status == TINFL_STATUS_DONE) && (!(flags & MZ_ZIP_FLAG_COMPRESSED_DATA)))
4790 if (out_buf_ofs != file_stat.m_uncomp_size)
4792 mz_zip_set_error(pZip, MZ_ZIP_UNEXPECTED_DECOMPRESSED_SIZE);
4793 status = TINFL_STATUS_FAILED;
4795#ifndef MINIZ_DISABLE_ZIP_READER_CRC32_CHECKS
4796 else if (file_crc32 != file_stat.m_crc32)
4798 mz_zip_set_error(pZip, MZ_ZIP_DECOMPRESSION_FAILED);
4799 status = TINFL_STATUS_FAILED;
4804 if (!pZip->m_pState->m_pMem)
4805 pZip->m_pFree(pZip->m_pAlloc_opaque, pRead_buf);
4808 pZip->m_pFree(pZip->m_pAlloc_opaque, pWrite_buf);
4810 return status == TINFL_STATUS_DONE;
4813mz_bool mz_zip_reader_extract_file_to_callback(
mz_zip_archive *pZip,
const char *pFilename, mz_file_write_func pCallback,
void *pOpaque, mz_uint flags)
4815 mz_uint32 file_index;
4816 if (!mz_zip_reader_locate_file_v2(pZip, pFilename, NULL, flags, &file_index))
4819 return mz_zip_reader_extract_to_callback(pZip, file_index, pCallback, pOpaque, flags);
4825 mz_uint32 local_header_u32[(MZ_ZIP_LOCAL_DIR_HEADER_SIZE +
sizeof(mz_uint32) - 1) /
sizeof(mz_uint32)];
4826 mz_uint8 *pLocal_header = (mz_uint8 *)local_header_u32;
4829 if ((!pZip) || (!pZip->m_pState))
4836 mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
4841 if (!mz_zip_reader_file_stat(pZip, file_index, &pState->file_stat))
4843 pZip->m_pFree(pZip->m_pAlloc_opaque, pState);
4848 if (pState->file_stat.m_bit_flag & (MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_IS_ENCRYPTED | MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_USES_STRONG_ENCRYPTION | MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_COMPRESSED_PATCH_FLAG))
4850 mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_ENCRYPTION);
4851 pZip->m_pFree(pZip->m_pAlloc_opaque, pState);
4856 if ((!(flags & MZ_ZIP_FLAG_COMPRESSED_DATA)) && (pState->file_stat.m_method != 0) && (pState->file_stat.m_method != MZ_DEFLATED))
4858 mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_METHOD);
4859 pZip->m_pFree(pZip->m_pAlloc_opaque, pState);
4864 pState->pZip = pZip;
4865 pState->flags = flags;
4868 pState->status = TINFL_STATUS_DONE;
4869#ifndef MINIZ_DISABLE_ZIP_READER_CRC32_CHECKS
4870 pState->file_crc32 = MZ_CRC32_INIT;
4872 pState->read_buf_ofs = 0;
4873 pState->out_buf_ofs = 0;
4874 pState->pRead_buf = NULL;
4875 pState->pWrite_buf = NULL;
4876 pState->out_blk_remain = 0;
4879 pState->cur_file_ofs = pState->file_stat.m_local_header_ofs;
4880 if (pZip->m_pRead(pZip->m_pIO_opaque, pState->cur_file_ofs, pLocal_header, MZ_ZIP_LOCAL_DIR_HEADER_SIZE) != MZ_ZIP_LOCAL_DIR_HEADER_SIZE)
4882 mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
4883 pZip->m_pFree(pZip->m_pAlloc_opaque, pState);
4887 if (MZ_READ_LE32(pLocal_header) != MZ_ZIP_LOCAL_DIR_HEADER_SIG)
4889 mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
4890 pZip->m_pFree(pZip->m_pAlloc_opaque, pState);
4894 pState->cur_file_ofs += MZ_ZIP_LOCAL_DIR_HEADER_SIZE + MZ_READ_LE16(pLocal_header + MZ_ZIP_LDH_FILENAME_LEN_OFS) + MZ_READ_LE16(pLocal_header + MZ_ZIP_LDH_EXTRA_LEN_OFS);
4895 if ((pState->cur_file_ofs + pState->file_stat.m_comp_size) > pZip->m_archive_size)
4897 mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
4898 pZip->m_pFree(pZip->m_pAlloc_opaque, pState);
4903 if (pZip->m_pState->m_pMem)
4905 pState->pRead_buf = (mz_uint8 *)pZip->m_pState->m_pMem + pState->cur_file_ofs;
4906 pState->read_buf_size = pState->read_buf_avail = pState->file_stat.m_comp_size;
4907 pState->comp_remaining = pState->file_stat.m_comp_size;
4911 if (!((flags & MZ_ZIP_FLAG_COMPRESSED_DATA) || (!pState->file_stat.m_method)))
4914 pState->read_buf_size = MZ_MIN(pState->file_stat.m_comp_size, MZ_ZIP_MAX_IO_BUF_SIZE);
4915 if (NULL == (pState->pRead_buf = pZip->m_pAlloc(pZip->m_pAlloc_opaque, 1, (
size_t)pState->read_buf_size)))
4917 mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
4918 pZip->m_pFree(pZip->m_pAlloc_opaque, pState);
4925 pState->read_buf_size = 0;
4927 pState->read_buf_avail = 0;
4928 pState->comp_remaining = pState->file_stat.m_comp_size;
4931 if (!((flags & MZ_ZIP_FLAG_COMPRESSED_DATA) || (!pState->file_stat.m_method)))
4934 tinfl_init( &pState->inflator );
4937 if (NULL == (pState->pWrite_buf = pZip->m_pAlloc(pZip->m_pAlloc_opaque, 1, TINFL_LZ_DICT_SIZE)))
4939 mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
4940 if (pState->pRead_buf)
4941 pZip->m_pFree(pZip->m_pAlloc_opaque, pState->pRead_buf);
4942 pZip->m_pFree(pZip->m_pAlloc_opaque, pState);
4952 mz_uint32 file_index;
4955 if (!mz_zip_reader_locate_file_v2(pZip, pFilename, NULL, flags, &file_index))
4959 return mz_zip_reader_extract_iter_new(pZip, file_index, flags);
4964 size_t copied_to_caller = 0;
4967 if ((!pState) || (!pState->pZip) || (!pState->pZip->m_pState) || (!pvBuf))
4970 if ((pState->flags & MZ_ZIP_FLAG_COMPRESSED_DATA) || (!pState->file_stat.m_method))
4973 copied_to_caller = (size_t)MZ_MIN( buf_size, pState->comp_remaining );
4976 if (pState->pZip->m_pState->m_pMem)
4979 memcpy( pvBuf, pState->pRead_buf, copied_to_caller );
4980 pState->pRead_buf = ((mz_uint8*)pState->pRead_buf) + copied_to_caller;
4985 if (pState->pZip->m_pRead(pState->pZip->m_pIO_opaque, pState->cur_file_ofs, pvBuf, copied_to_caller) != copied_to_caller)
4988 mz_zip_set_error(pState->pZip, MZ_ZIP_FILE_READ_FAILED);
4989 pState->status = TINFL_STATUS_FAILED;
4990 copied_to_caller = 0;
4994#ifndef MINIZ_DISABLE_ZIP_READER_CRC32_CHECKS
4996 if (!(pState->flags & MZ_ZIP_FLAG_COMPRESSED_DATA))
4997 pState->file_crc32 = (mz_uint32)mz_crc32(pState->file_crc32, (
const mz_uint8 *)pvBuf, copied_to_caller);
5001 pState->cur_file_ofs += copied_to_caller;
5002 pState->out_buf_ofs += copied_to_caller;
5003 pState->comp_remaining -= copied_to_caller;
5010 mz_uint8 *pWrite_buf_cur = (mz_uint8 *)pState->pWrite_buf + (pState->out_buf_ofs & (TINFL_LZ_DICT_SIZE - 1));
5013 size_t in_buf_size, out_buf_size = TINFL_LZ_DICT_SIZE - (pState->out_buf_ofs & (TINFL_LZ_DICT_SIZE - 1));
5015 if (!pState->out_blk_remain)
5018 if ((!pState->read_buf_avail) && (!pState->pZip->m_pState->m_pMem))
5021 pState->read_buf_avail = MZ_MIN(pState->read_buf_size, pState->comp_remaining);
5022 if (pState->pZip->m_pRead(pState->pZip->m_pIO_opaque, pState->cur_file_ofs, pState->pRead_buf, (
size_t)pState->read_buf_avail) != pState->read_buf_avail)
5024 mz_zip_set_error(pState->pZip, MZ_ZIP_FILE_READ_FAILED);
5025 pState->status = TINFL_STATUS_FAILED;
5030 pState->cur_file_ofs += pState->read_buf_avail;
5031 pState->comp_remaining -= pState->read_buf_avail;
5032 pState->read_buf_ofs = 0;
5036 in_buf_size = (size_t)pState->read_buf_avail;
5037 pState->status = tinfl_decompress(&pState->inflator, (
const mz_uint8 *)pState->pRead_buf + pState->read_buf_ofs, &in_buf_size, (mz_uint8 *)pState->pWrite_buf, pWrite_buf_cur, &out_buf_size, pState->comp_remaining ? TINFL_FLAG_HAS_MORE_INPUT : 0);
5038 pState->read_buf_avail -= in_buf_size;
5039 pState->read_buf_ofs += in_buf_size;
5042 pState->out_blk_remain = out_buf_size;
5045 if (pState->out_blk_remain)
5048 size_t to_copy = MZ_MIN( (buf_size - copied_to_caller), pState->out_blk_remain );
5051 memcpy( (uint8_t*)pvBuf + copied_to_caller, pWrite_buf_cur, to_copy );
5053#ifndef MINIZ_DISABLE_ZIP_READER_CRC32_CHECKS
5055 pState->file_crc32 = (mz_uint32)mz_crc32(pState->file_crc32, pWrite_buf_cur, to_copy);
5059 pState->out_blk_remain -= to_copy;
5062 if ((pState->out_buf_ofs += to_copy) > pState->file_stat.m_uncomp_size)
5064 mz_zip_set_error(pState->pZip, MZ_ZIP_DECOMPRESSION_FAILED);
5065 pState->status = TINFL_STATUS_FAILED;
5070 copied_to_caller += to_copy;
5072 }
while ( (copied_to_caller < buf_size) && ((pState->status == TINFL_STATUS_NEEDS_MORE_INPUT) || (pState->status == TINFL_STATUS_HAS_MORE_OUTPUT)) );
5076 return copied_to_caller;
5084 if ((!pState) || (!pState->pZip) || (!pState->pZip->m_pState))
5088 if ((pState->status == TINFL_STATUS_DONE) && (!(pState->flags & MZ_ZIP_FLAG_COMPRESSED_DATA)))
5091 if (pState->out_buf_ofs != pState->file_stat.m_uncomp_size)
5093 mz_zip_set_error(pState->pZip, MZ_ZIP_UNEXPECTED_DECOMPRESSED_SIZE);
5094 pState->status = TINFL_STATUS_FAILED;
5096#ifndef MINIZ_DISABLE_ZIP_READER_CRC32_CHECKS
5097 else if (pState->file_crc32 != pState->file_stat.m_crc32)
5099 mz_zip_set_error(pState->pZip, MZ_ZIP_DECOMPRESSION_FAILED);
5100 pState->status = TINFL_STATUS_FAILED;
5106 if (!pState->pZip->m_pState->m_pMem)
5107 pState->pZip->m_pFree(pState->pZip->m_pAlloc_opaque, pState->pRead_buf);
5108 if (pState->pWrite_buf)
5109 pState->pZip->m_pFree(pState->pZip->m_pAlloc_opaque, pState->pWrite_buf);
5112 status = pState->status;
5115 pState->pZip->m_pFree(pState->pZip->m_pAlloc_opaque, pState);
5117 return status == TINFL_STATUS_DONE;
5120#ifndef MINIZ_NO_STDIO
5121static size_t mz_zip_file_write_callback(
void *pOpaque, mz_uint64 ofs,
const void *pBuf,
size_t n)
5125 return MZ_FWRITE(pBuf, 1, n, (MZ_FILE *)pOpaque);
5128mz_bool mz_zip_reader_extract_to_file(
mz_zip_archive *pZip, mz_uint file_index,
const char *pDst_filename, mz_uint flags)
5134 if (!mz_zip_reader_file_stat(pZip, file_index, &file_stat))
5137 if ((file_stat.m_is_directory) || (!file_stat.m_is_supported))
5138 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_FEATURE);
5140 pFile = MZ_FOPEN(pDst_filename,
"wb");
5142 return mz_zip_set_error(pZip, MZ_ZIP_FILE_OPEN_FAILED);
5144 status = mz_zip_reader_extract_to_callback(pZip, file_index, mz_zip_file_write_callback, pFile, flags);
5146 if (MZ_FCLOSE(pFile) == EOF)
5149 mz_zip_set_error(pZip, MZ_ZIP_FILE_CLOSE_FAILED);
5154#if !defined(MINIZ_NO_TIME) && !defined(MINIZ_NO_STDIO)
5156 mz_zip_set_file_times(pDst_filename, file_stat.m_time, file_stat.m_time);
5162mz_bool mz_zip_reader_extract_file_to_file(
mz_zip_archive *pZip,
const char *pArchive_filename,
const char *pDst_filename, mz_uint flags)
5164 mz_uint32 file_index;
5165 if (!mz_zip_reader_locate_file_v2(pZip, pArchive_filename, NULL, flags, &file_index))
5168 return mz_zip_reader_extract_to_file(pZip, file_index, pDst_filename, flags);
5171mz_bool mz_zip_reader_extract_to_cfile(
mz_zip_archive *pZip, mz_uint file_index, MZ_FILE *pFile, mz_uint flags)
5175 if (!mz_zip_reader_file_stat(pZip, file_index, &file_stat))
5178 if ((file_stat.m_is_directory) || (!file_stat.m_is_supported))
5179 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_FEATURE);
5181 return mz_zip_reader_extract_to_callback(pZip, file_index, mz_zip_file_write_callback, pFile, flags);
5184mz_bool mz_zip_reader_extract_file_to_cfile(
mz_zip_archive *pZip,
const char *pArchive_filename, MZ_FILE *pFile, mz_uint flags)
5186 mz_uint32 file_index;
5187 if (!mz_zip_reader_locate_file_v2(pZip, pArchive_filename, NULL, flags, &file_index))
5190 return mz_zip_reader_extract_to_cfile(pZip, file_index, pFile, flags);
5194static size_t mz_zip_compute_crc32_callback(
void *pOpaque, mz_uint64 file_ofs,
const void *pBuf,
size_t n)
5196 mz_uint32 *p = (mz_uint32 *)pOpaque;
5198 *p = (mz_uint32)mz_crc32(*p, (
const mz_uint8 *)pBuf, n);
5202mz_bool mz_zip_validate_file(
mz_zip_archive *pZip, mz_uint file_index, mz_uint flags)
5206 const mz_uint8 *pCentral_dir_header;
5207 mz_bool found_zip64_ext_data_in_cdir = MZ_FALSE;
5208 mz_bool found_zip64_ext_data_in_ldir = MZ_FALSE;
5209 mz_uint32 local_header_u32[(MZ_ZIP_LOCAL_DIR_HEADER_SIZE +
sizeof(mz_uint32) - 1) /
sizeof(mz_uint32)];
5210 mz_uint8 *pLocal_header = (mz_uint8 *)local_header_u32;
5211 mz_uint64 local_header_ofs = 0;
5212 mz_uint32 local_header_filename_len, local_header_extra_len, local_header_crc32;
5213 mz_uint64 local_header_comp_size, local_header_uncomp_size;
5214 mz_uint32 uncomp_crc32 = MZ_CRC32_INIT;
5215 mz_bool has_data_descriptor;
5216 mz_uint32 local_header_bit_flags;
5219 mz_zip_array_init(&file_data_array, 1);
5221 if ((!pZip) || (!pZip->m_pState) || (!pZip->m_pAlloc) || (!pZip->m_pFree) || (!pZip->m_pRead))
5222 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5224 if (file_index > pZip->m_total_files)
5225 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5227 pState = pZip->m_pState;
5229 pCentral_dir_header = mz_zip_get_cdh(pZip, file_index);
5231 if (!mz_zip_file_stat_internal(pZip, file_index, pCentral_dir_header, &file_stat, &found_zip64_ext_data_in_cdir))
5235 if ((file_stat.m_is_directory) || (!file_stat.m_uncomp_size))
5239 if (file_stat.m_is_encrypted)
5240 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_ENCRYPTION);
5243 if ((file_stat.m_method != 0) && (file_stat.m_method != MZ_DEFLATED))
5244 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_METHOD);
5246 if (!file_stat.m_is_supported)
5247 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_FEATURE);
5250 local_header_ofs = file_stat.m_local_header_ofs;
5251 if (pZip->m_pRead(pZip->m_pIO_opaque, local_header_ofs, pLocal_header, MZ_ZIP_LOCAL_DIR_HEADER_SIZE) != MZ_ZIP_LOCAL_DIR_HEADER_SIZE)
5252 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
5254 if (MZ_READ_LE32(pLocal_header) != MZ_ZIP_LOCAL_DIR_HEADER_SIG)
5255 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
5257 local_header_filename_len = MZ_READ_LE16(pLocal_header + MZ_ZIP_LDH_FILENAME_LEN_OFS);
5258 local_header_extra_len = MZ_READ_LE16(pLocal_header + MZ_ZIP_LDH_EXTRA_LEN_OFS);
5259 local_header_comp_size = MZ_READ_LE32(pLocal_header + MZ_ZIP_LDH_COMPRESSED_SIZE_OFS);
5260 local_header_uncomp_size = MZ_READ_LE32(pLocal_header + MZ_ZIP_LDH_DECOMPRESSED_SIZE_OFS);
5261 local_header_crc32 = MZ_READ_LE32(pLocal_header + MZ_ZIP_LDH_CRC32_OFS);
5262 local_header_bit_flags = MZ_READ_LE16(pLocal_header + MZ_ZIP_LDH_BIT_FLAG_OFS);
5263 has_data_descriptor = (local_header_bit_flags & 8) != 0;
5265 if (local_header_filename_len != strlen(file_stat.m_filename))
5266 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
5268 if ((local_header_ofs + MZ_ZIP_LOCAL_DIR_HEADER_SIZE + local_header_filename_len + local_header_extra_len + file_stat.m_comp_size) > pZip->m_archive_size)
5269 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
5271 if (!mz_zip_array_resize(pZip, &file_data_array, MZ_MAX(local_header_filename_len, local_header_extra_len), MZ_FALSE))
5272 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
5274 if (local_header_filename_len)
5276 if (pZip->m_pRead(pZip->m_pIO_opaque, local_header_ofs + MZ_ZIP_LOCAL_DIR_HEADER_SIZE, file_data_array.m_p, local_header_filename_len) != local_header_filename_len)
5278 mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
5279 goto handle_failure;
5283 if (memcmp(file_stat.m_filename, file_data_array.m_p, local_header_filename_len) != 0)
5285 mz_zip_set_error(pZip, MZ_ZIP_VALIDATION_FAILED);
5286 goto handle_failure;
5290 if ((local_header_extra_len) && ((local_header_comp_size == MZ_UINT32_MAX) || (local_header_uncomp_size == MZ_UINT32_MAX)))
5292 mz_uint32 extra_size_remaining = local_header_extra_len;
5293 const mz_uint8 *pExtra_data = (
const mz_uint8 *)file_data_array.m_p;
5295 if (pZip->m_pRead(pZip->m_pIO_opaque, local_header_ofs + MZ_ZIP_LOCAL_DIR_HEADER_SIZE + local_header_filename_len, file_data_array.m_p, local_header_extra_len) != local_header_extra_len)
5297 mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
5298 goto handle_failure;
5303 mz_uint32 field_id, field_data_size, field_total_size;
5305 if (extra_size_remaining < (
sizeof(mz_uint16) * 2))
5306 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
5308 field_id = MZ_READ_LE16(pExtra_data);
5309 field_data_size = MZ_READ_LE16(pExtra_data +
sizeof(mz_uint16));
5310 field_total_size = field_data_size +
sizeof(mz_uint16) * 2;
5312 if (field_total_size > extra_size_remaining)
5313 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
5315 if (field_id == MZ_ZIP64_EXTENDED_INFORMATION_FIELD_HEADER_ID)
5317 const mz_uint8 *pSrc_field_data = pExtra_data +
sizeof(mz_uint32);
5319 if (field_data_size <
sizeof(mz_uint64) * 2)
5321 mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
5322 goto handle_failure;
5325 local_header_uncomp_size = MZ_READ_LE64(pSrc_field_data);
5326 local_header_comp_size = MZ_READ_LE64(pSrc_field_data +
sizeof(mz_uint64));
5328 found_zip64_ext_data_in_ldir = MZ_TRUE;
5332 pExtra_data += field_total_size;
5333 extra_size_remaining -= field_total_size;
5334 }
while (extra_size_remaining);
5339 if ((has_data_descriptor) && (!local_header_comp_size) && (!local_header_crc32))
5341 mz_uint8 descriptor_buf[32];
5343 const mz_uint8 *pSrc;
5344 mz_uint32 file_crc32;
5345 mz_uint64 comp_size = 0, uncomp_size = 0;
5347 mz_uint32 num_descriptor_uint32s = ((pState->m_zip64) || (found_zip64_ext_data_in_ldir)) ? 6 : 4;
5349 if (pZip->m_pRead(pZip->m_pIO_opaque, local_header_ofs + MZ_ZIP_LOCAL_DIR_HEADER_SIZE + local_header_filename_len + local_header_extra_len + file_stat.m_comp_size, descriptor_buf,
sizeof(mz_uint32) * num_descriptor_uint32s) != (
sizeof(mz_uint32) * num_descriptor_uint32s))
5351 mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
5352 goto handle_failure;
5355 has_id = (MZ_READ_LE32(descriptor_buf) == MZ_ZIP_DATA_DESCRIPTOR_ID);
5356 pSrc = has_id ? (descriptor_buf +
sizeof(mz_uint32)) : descriptor_buf;
5358 file_crc32 = MZ_READ_LE32(pSrc);
5360 if ((pState->m_zip64) || (found_zip64_ext_data_in_ldir))
5362 comp_size = MZ_READ_LE64(pSrc +
sizeof(mz_uint32));
5363 uncomp_size = MZ_READ_LE64(pSrc +
sizeof(mz_uint32) +
sizeof(mz_uint64));
5367 comp_size = MZ_READ_LE32(pSrc +
sizeof(mz_uint32));
5368 uncomp_size = MZ_READ_LE32(pSrc +
sizeof(mz_uint32) +
sizeof(mz_uint32));
5371 if ((file_crc32 != file_stat.m_crc32) || (comp_size != file_stat.m_comp_size) || (uncomp_size != file_stat.m_uncomp_size))
5373 mz_zip_set_error(pZip, MZ_ZIP_VALIDATION_FAILED);
5374 goto handle_failure;
5379 if ((local_header_crc32 != file_stat.m_crc32) || (local_header_comp_size != file_stat.m_comp_size) || (local_header_uncomp_size != file_stat.m_uncomp_size))
5381 mz_zip_set_error(pZip, MZ_ZIP_VALIDATION_FAILED);
5382 goto handle_failure;
5386 mz_zip_array_clear(pZip, &file_data_array);
5388 if ((flags & MZ_ZIP_FLAG_VALIDATE_HEADERS_ONLY) == 0)
5390 if (!mz_zip_reader_extract_to_callback(pZip, file_index, mz_zip_compute_crc32_callback, &uncomp_crc32, 0))
5394 if (uncomp_crc32 != file_stat.m_crc32)
5396 mz_zip_set_error(pZip, MZ_ZIP_VALIDATION_FAILED);
5404 mz_zip_array_clear(pZip, &file_data_array);
5408mz_bool mz_zip_validate_archive(
mz_zip_archive *pZip, mz_uint flags)
5413 if ((!pZip) || (!pZip->m_pState) || (!pZip->m_pAlloc) || (!pZip->m_pFree) || (!pZip->m_pRead))
5414 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5416 pState = pZip->m_pState;
5419 if (!pState->m_zip64)
5421 if (pZip->m_total_files > MZ_UINT16_MAX)
5422 return mz_zip_set_error(pZip, MZ_ZIP_ARCHIVE_TOO_LARGE);
5424 if (pZip->m_archive_size > MZ_UINT32_MAX)
5425 return mz_zip_set_error(pZip, MZ_ZIP_ARCHIVE_TOO_LARGE);
5429 if (pZip->m_total_files >= MZ_UINT32_MAX)
5430 return mz_zip_set_error(pZip, MZ_ZIP_ARCHIVE_TOO_LARGE);
5432 if (pState->m_central_dir.m_size >= MZ_UINT32_MAX)
5433 return mz_zip_set_error(pZip, MZ_ZIP_ARCHIVE_TOO_LARGE);
5436 for (i = 0; i < pZip->m_total_files; i++)
5438 if (MZ_ZIP_FLAG_VALIDATE_LOCATE_FILE_FLAG & flags)
5440 mz_uint32 found_index;
5443 if (!mz_zip_reader_file_stat(pZip, i, &stat))
5446 if (!mz_zip_reader_locate_file_v2(pZip, stat.m_filename, NULL, 0, &found_index))
5450 if (found_index != i)
5451 return mz_zip_set_error(pZip, MZ_ZIP_VALIDATION_FAILED);
5454 if (!mz_zip_validate_file(pZip, i, flags))
5461mz_bool mz_zip_validate_mem_archive(
const void *pMem,
size_t size, mz_uint flags, mz_zip_error *pErr)
5463 mz_bool success = MZ_TRUE;
5465 mz_zip_error actual_err = MZ_ZIP_NO_ERROR;
5467 if ((!pMem) || (!size))
5470 *pErr = MZ_ZIP_INVALID_PARAMETER;
5474 mz_zip_zero_struct(&zip);
5476 if (!mz_zip_reader_init_mem(&zip, pMem, size, flags))
5479 *pErr = zip.m_last_error;
5483 if (!mz_zip_validate_archive(&zip, flags))
5485 actual_err = zip.m_last_error;
5489 if (!mz_zip_reader_end_internal(&zip, success))
5492 actual_err = zip.m_last_error;
5502#ifndef MINIZ_NO_STDIO
5503mz_bool mz_zip_validate_file_archive(
const char *pFilename, mz_uint flags, mz_zip_error *pErr)
5505 mz_bool success = MZ_TRUE;
5507 mz_zip_error actual_err = MZ_ZIP_NO_ERROR;
5512 *pErr = MZ_ZIP_INVALID_PARAMETER;
5516 mz_zip_zero_struct(&zip);
5518 if (!mz_zip_reader_init_file_v2(&zip, pFilename, flags, 0, 0))
5521 *pErr = zip.m_last_error;
5525 if (!mz_zip_validate_archive(&zip, flags))
5527 actual_err = zip.m_last_error;
5531 if (!mz_zip_reader_end_internal(&zip, success))
5534 actual_err = zip.m_last_error;
5547#ifndef MINIZ_NO_ARCHIVE_WRITING_APIS
5549static MZ_FORCEINLINE
void mz_write_le16(mz_uint8 *p, mz_uint16 v)
5552 p[1] = (mz_uint8)(v >> 8);
5554static MZ_FORCEINLINE
void mz_write_le32(mz_uint8 *p, mz_uint32 v)
5557 p[1] = (mz_uint8)(v >> 8);
5558 p[2] = (mz_uint8)(v >> 16);
5559 p[3] = (mz_uint8)(v >> 24);
5561static MZ_FORCEINLINE
void mz_write_le64(mz_uint8 *p, mz_uint64 v)
5563 mz_write_le32(p, (mz_uint32)v);
5564 mz_write_le32(p +
sizeof(mz_uint32), (mz_uint32)(v >> 32));
5567#define MZ_WRITE_LE16(p, v) mz_write_le16((mz_uint8 *)(p), (mz_uint16)(v))
5568#define MZ_WRITE_LE32(p, v) mz_write_le32((mz_uint8 *)(p), (mz_uint32)(v))
5569#define MZ_WRITE_LE64(p, v) mz_write_le64((mz_uint8 *)(p), (mz_uint64)(v))
5571static size_t mz_zip_heap_write_func(
void *pOpaque, mz_uint64 file_ofs,
const void *pBuf,
size_t n)
5575 mz_uint64 new_size = MZ_MAX(file_ofs + n, pState->m_mem_size);
5581 if ((
sizeof(
size_t) ==
sizeof(mz_uint32)) && (new_size > 0x7FFFFFFF))
5583 mz_zip_set_error(pZip, MZ_ZIP_FILE_TOO_LARGE);
5587 if (new_size > pState->m_mem_capacity)
5590 size_t new_capacity = MZ_MAX(64, pState->m_mem_capacity);
5592 while (new_capacity < new_size)
5595 if (NULL == (pNew_block = pZip->m_pRealloc(pZip->m_pAlloc_opaque, pState->m_pMem, 1, new_capacity)))
5597 mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
5601 pState->m_pMem = pNew_block;
5602 pState->m_mem_capacity = new_capacity;
5604 memcpy((mz_uint8 *)pState->m_pMem + file_ofs, pBuf, n);
5605 pState->m_mem_size = (size_t)new_size;
5609static mz_bool mz_zip_writer_end_internal(
mz_zip_archive *pZip, mz_bool set_last_error)
5612 mz_bool status = MZ_TRUE;
5614 if ((!pZip) || (!pZip->m_pState) || (!pZip->m_pAlloc) || (!pZip->m_pFree) || ((pZip->m_zip_mode != MZ_ZIP_MODE_WRITING) && (pZip->m_zip_mode != MZ_ZIP_MODE_WRITING_HAS_BEEN_FINALIZED)))
5617 mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5621 pState = pZip->m_pState;
5622 pZip->m_pState = NULL;
5623 mz_zip_array_clear(pZip, &pState->m_central_dir);
5624 mz_zip_array_clear(pZip, &pState->m_central_dir_offsets);
5625 mz_zip_array_clear(pZip, &pState->m_sorted_central_dir_offsets);
5627#ifndef MINIZ_NO_STDIO
5628 if (pState->m_pFile)
5630 if (pZip->m_zip_type == MZ_ZIP_TYPE_FILE)
5632 if (MZ_FCLOSE(pState->m_pFile) == EOF)
5635 mz_zip_set_error(pZip, MZ_ZIP_FILE_CLOSE_FAILED);
5640 pState->m_pFile = NULL;
5644 if ((pZip->m_pWrite == mz_zip_heap_write_func) && (pState->m_pMem))
5646 pZip->m_pFree(pZip->m_pAlloc_opaque, pState->m_pMem);
5647 pState->m_pMem = NULL;
5650 pZip->m_pFree(pZip->m_pAlloc_opaque, pState);
5651 pZip->m_zip_mode = MZ_ZIP_MODE_INVALID;
5655mz_bool mz_zip_writer_init_v2(
mz_zip_archive *pZip, mz_uint64 existing_size, mz_uint flags)
5657 mz_bool zip64 = (flags & MZ_ZIP_FLAG_WRITE_ZIP64) != 0;
5659 if ((!pZip) || (pZip->m_pState) || (!pZip->m_pWrite) || (pZip->m_zip_mode != MZ_ZIP_MODE_INVALID))
5660 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5662 if (flags & MZ_ZIP_FLAG_WRITE_ALLOW_READING)
5665 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5668 if (pZip->m_file_offset_alignment)
5671 if (pZip->m_file_offset_alignment & (pZip->m_file_offset_alignment - 1))
5672 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5675 if (!pZip->m_pAlloc)
5676 pZip->m_pAlloc = miniz_def_alloc_func;
5678 pZip->m_pFree = miniz_def_free_func;
5679 if (!pZip->m_pRealloc)
5680 pZip->m_pRealloc = miniz_def_realloc_func;
5682 pZip->m_archive_size = existing_size;
5683 pZip->m_central_directory_file_ofs = 0;
5684 pZip->m_total_files = 0;
5687 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
5691 MZ_ZIP_ARRAY_SET_ELEMENT_SIZE(&pZip->m_pState->m_central_dir,
sizeof(mz_uint8));
5692 MZ_ZIP_ARRAY_SET_ELEMENT_SIZE(&pZip->m_pState->m_central_dir_offsets,
sizeof(mz_uint32));
5693 MZ_ZIP_ARRAY_SET_ELEMENT_SIZE(&pZip->m_pState->m_sorted_central_dir_offsets,
sizeof(mz_uint32));
5695 pZip->m_pState->m_zip64 = zip64;
5696 pZip->m_pState->m_zip64_has_extended_info_fields = zip64;
5698 pZip->m_zip_type = MZ_ZIP_TYPE_USER;
5699 pZip->m_zip_mode = MZ_ZIP_MODE_WRITING;
5704mz_bool mz_zip_writer_init(
mz_zip_archive *pZip, mz_uint64 existing_size)
5706 return mz_zip_writer_init_v2(pZip, existing_size, 0);
5709mz_bool mz_zip_writer_init_heap_v2(
mz_zip_archive *pZip,
size_t size_to_reserve_at_beginning,
size_t initial_allocation_size, mz_uint flags)
5711 pZip->m_pWrite = mz_zip_heap_write_func;
5712 pZip->m_pNeeds_keepalive = NULL;
5714 if (flags & MZ_ZIP_FLAG_WRITE_ALLOW_READING)
5715 pZip->m_pRead = mz_zip_mem_read_func;
5717 pZip->m_pIO_opaque = pZip;
5719 if (!mz_zip_writer_init_v2(pZip, size_to_reserve_at_beginning, flags))
5722 pZip->m_zip_type = MZ_ZIP_TYPE_HEAP;
5724 if (0 != (initial_allocation_size = MZ_MAX(initial_allocation_size, size_to_reserve_at_beginning)))
5726 if (NULL == (pZip->m_pState->m_pMem = pZip->m_pAlloc(pZip->m_pAlloc_opaque, 1, initial_allocation_size)))
5728 mz_zip_writer_end_internal(pZip, MZ_FALSE);
5729 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
5731 pZip->m_pState->m_mem_capacity = initial_allocation_size;
5737mz_bool mz_zip_writer_init_heap(
mz_zip_archive *pZip,
size_t size_to_reserve_at_beginning,
size_t initial_allocation_size)
5739 return mz_zip_writer_init_heap_v2(pZip, size_to_reserve_at_beginning, initial_allocation_size, 0);
5742#ifndef MINIZ_NO_STDIO
5743static size_t mz_zip_file_write_func(
void *pOpaque, mz_uint64 file_ofs,
const void *pBuf,
size_t n)
5746 mz_int64 cur_ofs = MZ_FTELL64(pZip->m_pState->m_pFile);
5748 file_ofs += pZip->m_pState->m_file_archive_start_ofs;
5750 if (((mz_int64)file_ofs < 0) || (((cur_ofs != (mz_int64)file_ofs)) && (MZ_FSEEK64(pZip->m_pState->m_pFile, (mz_int64)file_ofs, SEEK_SET))))
5752 mz_zip_set_error(pZip, MZ_ZIP_FILE_SEEK_FAILED);
5756 return MZ_FWRITE(pBuf, 1, n, pZip->m_pState->m_pFile);
5759mz_bool mz_zip_writer_init_file(
mz_zip_archive *pZip,
const char *pFilename, mz_uint64 size_to_reserve_at_beginning)
5761 return mz_zip_writer_init_file_v2(pZip, pFilename, size_to_reserve_at_beginning, 0);
5764mz_bool mz_zip_writer_init_file_v2(
mz_zip_archive *pZip,
const char *pFilename, mz_uint64 size_to_reserve_at_beginning, mz_uint flags)
5768 pZip->m_pWrite = mz_zip_file_write_func;
5769 pZip->m_pNeeds_keepalive = NULL;
5771 if (flags & MZ_ZIP_FLAG_WRITE_ALLOW_READING)
5772 pZip->m_pRead = mz_zip_file_read_func;
5774 pZip->m_pIO_opaque = pZip;
5776 if (!mz_zip_writer_init_v2(pZip, size_to_reserve_at_beginning, flags))
5779 if (NULL == (pFile = MZ_FOPEN(pFilename, (flags & MZ_ZIP_FLAG_WRITE_ALLOW_READING) ?
"w+b" :
"wb")))
5781 mz_zip_writer_end(pZip);
5782 return mz_zip_set_error(pZip, MZ_ZIP_FILE_OPEN_FAILED);
5785 pZip->m_pState->m_pFile = pFile;
5786 pZip->m_zip_type = MZ_ZIP_TYPE_FILE;
5788 if (size_to_reserve_at_beginning)
5790 mz_uint64 cur_ofs = 0;
5797 size_t n = (size_t)MZ_MIN(
sizeof(buf), size_to_reserve_at_beginning);
5798 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_ofs, buf, n) != n)
5800 mz_zip_writer_end(pZip);
5801 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
5804 size_to_reserve_at_beginning -= n;
5805 }
while (size_to_reserve_at_beginning);
5811mz_bool mz_zip_writer_init_cfile(
mz_zip_archive *pZip, MZ_FILE *pFile, mz_uint flags)
5813 pZip->m_pWrite = mz_zip_file_write_func;
5814 pZip->m_pNeeds_keepalive = NULL;
5816 if (flags & MZ_ZIP_FLAG_WRITE_ALLOW_READING)
5817 pZip->m_pRead = mz_zip_file_read_func;
5819 pZip->m_pIO_opaque = pZip;
5821 if (!mz_zip_writer_init_v2(pZip, 0, flags))
5824 pZip->m_pState->m_pFile = pFile;
5825 pZip->m_pState->m_file_archive_start_ofs = MZ_FTELL64(pZip->m_pState->m_pFile);
5826 pZip->m_zip_type = MZ_ZIP_TYPE_CFILE;
5832mz_bool mz_zip_writer_init_from_reader_v2(
mz_zip_archive *pZip,
const char *pFilename, mz_uint flags)
5836 if ((!pZip) || (!pZip->m_pState) || (pZip->m_zip_mode != MZ_ZIP_MODE_READING))
5837 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5839 if (flags & MZ_ZIP_FLAG_WRITE_ZIP64)
5842 if (!pZip->m_pState->m_zip64)
5843 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5847 if (pZip->m_pState->m_zip64)
5849 if (pZip->m_total_files == MZ_UINT32_MAX)
5850 return mz_zip_set_error(pZip, MZ_ZIP_TOO_MANY_FILES);
5854 if (pZip->m_total_files == MZ_UINT16_MAX)
5855 return mz_zip_set_error(pZip, MZ_ZIP_TOO_MANY_FILES);
5857 if ((pZip->m_archive_size + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + MZ_ZIP_LOCAL_DIR_HEADER_SIZE) > MZ_UINT32_MAX)
5858 return mz_zip_set_error(pZip, MZ_ZIP_FILE_TOO_LARGE);
5861 pState = pZip->m_pState;
5863 if (pState->m_pFile)
5865#ifdef MINIZ_NO_STDIO
5867 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5869 if (pZip->m_pIO_opaque != pZip)
5870 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5872 if (pZip->m_zip_type == MZ_ZIP_TYPE_FILE)
5875 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5878 if (NULL == (pState->m_pFile = MZ_FREOPEN(pFilename,
"r+b", pState->m_pFile)))
5881 mz_zip_reader_end_internal(pZip, MZ_FALSE);
5882 return mz_zip_set_error(pZip, MZ_ZIP_FILE_OPEN_FAILED);
5886 pZip->m_pWrite = mz_zip_file_write_func;
5887 pZip->m_pNeeds_keepalive = NULL;
5890 else if (pState->m_pMem)
5893 if (pZip->m_pIO_opaque != pZip)
5894 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5896 pState->m_mem_capacity = pState->m_mem_size;
5897 pZip->m_pWrite = mz_zip_heap_write_func;
5898 pZip->m_pNeeds_keepalive = NULL;
5901 else if (!pZip->m_pWrite)
5902 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
5906 pZip->m_archive_size = pZip->m_central_directory_file_ofs;
5907 pZip->m_central_directory_file_ofs = 0;
5912 mz_zip_array_clear(pZip, &pZip->m_pState->m_sorted_central_dir_offsets);
5914 pZip->m_zip_mode = MZ_ZIP_MODE_WRITING;
5919mz_bool mz_zip_writer_init_from_reader(
mz_zip_archive *pZip,
const char *pFilename)
5921 return mz_zip_writer_init_from_reader_v2(pZip, pFilename, 0);
5925mz_bool mz_zip_writer_add_mem(
mz_zip_archive *pZip,
const char *pArchive_name,
const void *pBuf,
size_t buf_size, mz_uint level_and_flags)
5927 return mz_zip_writer_add_mem_ex(pZip, pArchive_name, pBuf, buf_size, NULL, 0, level_and_flags, 0, 0);
5933 mz_uint64 m_cur_archive_file_ofs;
5934 mz_uint64 m_comp_size;
5937static mz_bool mz_zip_writer_add_put_buf_callback(
const void *pBuf,
int len,
void *pUser)
5940 if ((
int)pState->m_pZip->m_pWrite(pState->m_pZip->m_pIO_opaque, pState->m_cur_archive_file_ofs, pBuf, len) != len)
5943 pState->m_cur_archive_file_ofs += len;
5944 pState->m_comp_size += len;
5948#define MZ_ZIP64_MAX_LOCAL_EXTRA_FIELD_SIZE (sizeof(mz_uint16) * 2 + sizeof(mz_uint64) * 2)
5949#define MZ_ZIP64_MAX_CENTRAL_EXTRA_FIELD_SIZE (sizeof(mz_uint16) * 2 + sizeof(mz_uint64) * 3)
5950static mz_uint32 mz_zip_writer_create_zip64_extra_data(mz_uint8 *pBuf, mz_uint64 *pUncomp_size, mz_uint64 *pComp_size, mz_uint64 *pLocal_header_ofs)
5952 mz_uint8 *pDst = pBuf;
5953 mz_uint32 field_size = 0;
5955 MZ_WRITE_LE16(pDst + 0, MZ_ZIP64_EXTENDED_INFORMATION_FIELD_HEADER_ID);
5956 MZ_WRITE_LE16(pDst + 2, 0);
5957 pDst +=
sizeof(mz_uint16) * 2;
5961 MZ_WRITE_LE64(pDst, *pUncomp_size);
5962 pDst +=
sizeof(mz_uint64);
5963 field_size +=
sizeof(mz_uint64);
5968 MZ_WRITE_LE64(pDst, *pComp_size);
5969 pDst +=
sizeof(mz_uint64);
5970 field_size +=
sizeof(mz_uint64);
5973 if (pLocal_header_ofs)
5975 MZ_WRITE_LE64(pDst, *pLocal_header_ofs);
5976 pDst +=
sizeof(mz_uint64);
5977 field_size +=
sizeof(mz_uint64);
5980 MZ_WRITE_LE16(pBuf + 2, field_size);
5982 return (mz_uint32)(pDst - pBuf);
5985static mz_bool mz_zip_writer_create_local_dir_header(
mz_zip_archive *pZip, mz_uint8 *pDst, mz_uint16 filename_size, mz_uint16 extra_size, mz_uint64 uncomp_size, mz_uint64 comp_size, mz_uint32 uncomp_crc32, mz_uint16 method, mz_uint16 bit_flags, mz_uint16 dos_time, mz_uint16 dos_date)
5988 memset(pDst, 0, MZ_ZIP_LOCAL_DIR_HEADER_SIZE);
5989 MZ_WRITE_LE32(pDst + MZ_ZIP_LDH_SIG_OFS, MZ_ZIP_LOCAL_DIR_HEADER_SIG);
5990 MZ_WRITE_LE16(pDst + MZ_ZIP_LDH_VERSION_NEEDED_OFS, method ? 20 : 0);
5991 MZ_WRITE_LE16(pDst + MZ_ZIP_LDH_BIT_FLAG_OFS, bit_flags);
5992 MZ_WRITE_LE16(pDst + MZ_ZIP_LDH_METHOD_OFS, method);
5993 MZ_WRITE_LE16(pDst + MZ_ZIP_LDH_FILE_TIME_OFS, dos_time);
5994 MZ_WRITE_LE16(pDst + MZ_ZIP_LDH_FILE_DATE_OFS, dos_date);
5995 MZ_WRITE_LE32(pDst + MZ_ZIP_LDH_CRC32_OFS, uncomp_crc32);
5996 MZ_WRITE_LE32(pDst + MZ_ZIP_LDH_COMPRESSED_SIZE_OFS, MZ_MIN(comp_size, MZ_UINT32_MAX));
5997 MZ_WRITE_LE32(pDst + MZ_ZIP_LDH_DECOMPRESSED_SIZE_OFS, MZ_MIN(uncomp_size, MZ_UINT32_MAX));
5998 MZ_WRITE_LE16(pDst + MZ_ZIP_LDH_FILENAME_LEN_OFS, filename_size);
5999 MZ_WRITE_LE16(pDst + MZ_ZIP_LDH_EXTRA_LEN_OFS, extra_size);
6003static mz_bool mz_zip_writer_create_central_dir_header(
mz_zip_archive *pZip, mz_uint8 *pDst,
6004 mz_uint16 filename_size, mz_uint16 extra_size, mz_uint16 comment_size,
6005 mz_uint64 uncomp_size, mz_uint64 comp_size, mz_uint32 uncomp_crc32,
6006 mz_uint16 method, mz_uint16 bit_flags, mz_uint16 dos_time, mz_uint16 dos_date,
6007 mz_uint64 local_header_ofs, mz_uint32 ext_attributes)
6010 memset(pDst, 0, MZ_ZIP_CENTRAL_DIR_HEADER_SIZE);
6011 MZ_WRITE_LE32(pDst + MZ_ZIP_CDH_SIG_OFS, MZ_ZIP_CENTRAL_DIR_HEADER_SIG);
6012 MZ_WRITE_LE16(pDst + MZ_ZIP_CDH_VERSION_NEEDED_OFS, method ? 20 : 0);
6013 MZ_WRITE_LE16(pDst + MZ_ZIP_CDH_BIT_FLAG_OFS, bit_flags);
6014 MZ_WRITE_LE16(pDst + MZ_ZIP_CDH_METHOD_OFS, method);
6015 MZ_WRITE_LE16(pDst + MZ_ZIP_CDH_FILE_TIME_OFS, dos_time);
6016 MZ_WRITE_LE16(pDst + MZ_ZIP_CDH_FILE_DATE_OFS, dos_date);
6017 MZ_WRITE_LE32(pDst + MZ_ZIP_CDH_CRC32_OFS, uncomp_crc32);
6018 MZ_WRITE_LE32(pDst + MZ_ZIP_CDH_COMPRESSED_SIZE_OFS, MZ_MIN(comp_size, MZ_UINT32_MAX));
6019 MZ_WRITE_LE32(pDst + MZ_ZIP_CDH_DECOMPRESSED_SIZE_OFS, MZ_MIN(uncomp_size, MZ_UINT32_MAX));
6020 MZ_WRITE_LE16(pDst + MZ_ZIP_CDH_FILENAME_LEN_OFS, filename_size);
6021 MZ_WRITE_LE16(pDst + MZ_ZIP_CDH_EXTRA_LEN_OFS, extra_size);
6022 MZ_WRITE_LE16(pDst + MZ_ZIP_CDH_COMMENT_LEN_OFS, comment_size);
6023 MZ_WRITE_LE32(pDst + MZ_ZIP_CDH_EXTERNAL_ATTR_OFS, ext_attributes);
6024 MZ_WRITE_LE32(pDst + MZ_ZIP_CDH_LOCAL_HEADER_OFS, MZ_MIN(local_header_ofs, MZ_UINT32_MAX));
6028static mz_bool mz_zip_writer_add_to_central_dir(
mz_zip_archive *pZip,
const char *pFilename, mz_uint16 filename_size,
6029 const void *pExtra, mz_uint16 extra_size,
const void *pComment, mz_uint16 comment_size,
6030 mz_uint64 uncomp_size, mz_uint64 comp_size, mz_uint32 uncomp_crc32,
6031 mz_uint16 method, mz_uint16 bit_flags, mz_uint16 dos_time, mz_uint16 dos_date,
6032 mz_uint64 local_header_ofs, mz_uint32 ext_attributes,
6033 const char *user_extra_data, mz_uint user_extra_data_len)
6036 mz_uint32 central_dir_ofs = (mz_uint32)pState->m_central_dir.m_size;
6037 size_t orig_central_dir_size = pState->m_central_dir.m_size;
6038 mz_uint8 central_dir_header[MZ_ZIP_CENTRAL_DIR_HEADER_SIZE];
6040 if (!pZip->m_pState->m_zip64)
6042 if (local_header_ofs > 0xFFFFFFFF)
6043 return mz_zip_set_error(pZip, MZ_ZIP_FILE_TOO_LARGE);
6047 if (((mz_uint64)pState->m_central_dir.m_size + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + filename_size + extra_size + user_extra_data_len + comment_size) >= MZ_UINT32_MAX)
6048 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_CDIR_SIZE);
6050 if (!mz_zip_writer_create_central_dir_header(pZip, central_dir_header, filename_size, (mz_uint16)(extra_size + user_extra_data_len), comment_size, uncomp_size, comp_size, uncomp_crc32, method, bit_flags, dos_time, dos_date, local_header_ofs, ext_attributes))
6051 return mz_zip_set_error(pZip, MZ_ZIP_INTERNAL_ERROR);
6053 if ((!mz_zip_array_push_back(pZip, &pState->m_central_dir, central_dir_header, MZ_ZIP_CENTRAL_DIR_HEADER_SIZE)) ||
6054 (!mz_zip_array_push_back(pZip, &pState->m_central_dir, pFilename, filename_size)) ||
6055 (!mz_zip_array_push_back(pZip, &pState->m_central_dir, pExtra, extra_size)) ||
6056 (!mz_zip_array_push_back(pZip, &pState->m_central_dir, user_extra_data, user_extra_data_len)) ||
6057 (!mz_zip_array_push_back(pZip, &pState->m_central_dir, pComment, comment_size)) ||
6058 (!mz_zip_array_push_back(pZip, &pState->m_central_dir_offsets, ¢ral_dir_ofs, 1)))
6061 mz_zip_array_resize(pZip, &pState->m_central_dir, orig_central_dir_size, MZ_FALSE);
6062 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
6068static mz_bool mz_zip_writer_validate_archive_name(
const char *pArchive_name)
6071 if (*pArchive_name ==
'/')
6079static mz_uint mz_zip_writer_compute_padding_needed_for_file_alignment(
mz_zip_archive *pZip)
6082 if (!pZip->m_file_offset_alignment)
6084 n = (mz_uint32)(pZip->m_archive_size & (pZip->m_file_offset_alignment - 1));
6085 return (mz_uint)((pZip->m_file_offset_alignment - n) & (pZip->m_file_offset_alignment - 1));
6088static mz_bool mz_zip_writer_write_zeros(
mz_zip_archive *pZip, mz_uint64 cur_file_ofs, mz_uint32 n)
6091 memset(buf, 0, MZ_MIN(
sizeof(buf), n));
6094 mz_uint32 s = MZ_MIN(
sizeof(buf), n);
6095 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_file_ofs, buf, s) != s)
6096 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6104mz_bool mz_zip_writer_add_mem_ex(
mz_zip_archive *pZip,
const char *pArchive_name,
const void *pBuf,
size_t buf_size,
const void *pComment, mz_uint16 comment_size, mz_uint level_and_flags,
6105 mz_uint64 uncomp_size, mz_uint32 uncomp_crc32)
6107 return mz_zip_writer_add_mem_ex_v2(pZip, pArchive_name, pBuf, buf_size, pComment, comment_size, level_and_flags, uncomp_size, uncomp_crc32, NULL, NULL, 0, NULL, 0);
6110mz_bool mz_zip_writer_add_mem_ex_v2(
mz_zip_archive *pZip,
const char *pArchive_name,
const void *pBuf,
size_t buf_size,
const void *pComment, mz_uint16 comment_size,
6111 mz_uint level_and_flags, mz_uint64 uncomp_size, mz_uint32 uncomp_crc32, MZ_TIME_T *last_modified,
6112 const char *user_extra_data, mz_uint user_extra_data_len,
const char *user_extra_data_central, mz_uint user_extra_data_central_len)
6114 mz_uint16 method = 0, dos_time = 0, dos_date = 0;
6115 mz_uint level, ext_attributes = 0, num_alignment_padding_bytes;
6116 mz_uint64 local_dir_header_ofs = pZip->m_archive_size, cur_archive_file_ofs = pZip->m_archive_size, comp_size = 0;
6117 size_t archive_name_size;
6118 mz_uint8 local_dir_header[MZ_ZIP_LOCAL_DIR_HEADER_SIZE];
6120 mz_bool store_data_uncompressed;
6122 mz_uint8 *pExtra_data = NULL;
6123 mz_uint32 extra_size = 0;
6124 mz_uint8 extra_data[MZ_ZIP64_MAX_CENTRAL_EXTRA_FIELD_SIZE];
6125 mz_uint16 bit_flags = 0;
6127 if ((
int)level_and_flags < 0)
6128 level_and_flags = MZ_DEFAULT_LEVEL;
6130 if (uncomp_size || (buf_size && !(level_and_flags & MZ_ZIP_FLAG_COMPRESSED_DATA)))
6131 bit_flags |= MZ_ZIP_LDH_BIT_FLAG_HAS_LOCATOR;
6133 if (!(level_and_flags & MZ_ZIP_FLAG_ASCII_FILENAME))
6134 bit_flags |= MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_UTF8;
6136 level = level_and_flags & 0xF;
6137 store_data_uncompressed = ((!level) || (level_and_flags & MZ_ZIP_FLAG_COMPRESSED_DATA));
6139 if ((!pZip) || (!pZip->m_pState) || (pZip->m_zip_mode != MZ_ZIP_MODE_WRITING) || ((buf_size) && (!pBuf)) || (!pArchive_name) || ((comment_size) && (!pComment)) || (level > MZ_UBER_COMPRESSION))
6140 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
6142 pState = pZip->m_pState;
6144 if (pState->m_zip64)
6146 if (pZip->m_total_files == MZ_UINT32_MAX)
6147 return mz_zip_set_error(pZip, MZ_ZIP_TOO_MANY_FILES);
6151 if (pZip->m_total_files == MZ_UINT16_MAX)
6153 pState->m_zip64 = MZ_TRUE;
6156 if ((buf_size > 0xFFFFFFFF) || (uncomp_size > 0xFFFFFFFF))
6158 pState->m_zip64 = MZ_TRUE;
6163 if ((!(level_and_flags & MZ_ZIP_FLAG_COMPRESSED_DATA)) && (uncomp_size))
6164 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
6166 if (!mz_zip_writer_validate_archive_name(pArchive_name))
6167 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_FILENAME);
6169#ifndef MINIZ_NO_TIME
6170 if (last_modified != NULL)
6172 mz_zip_time_t_to_dos_time(*last_modified, &dos_time, &dos_date);
6178 mz_zip_time_t_to_dos_time(cur_time, &dos_time, &dos_date);
6182 if (!(level_and_flags & MZ_ZIP_FLAG_COMPRESSED_DATA))
6184 uncomp_crc32 = (mz_uint32)mz_crc32(MZ_CRC32_INIT, (
const mz_uint8 *)pBuf, buf_size);
6185 uncomp_size = buf_size;
6186 if (uncomp_size <= 3)
6189 store_data_uncompressed = MZ_TRUE;
6193 archive_name_size = strlen(pArchive_name);
6194 if (archive_name_size > MZ_UINT16_MAX)
6195 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_FILENAME);
6197 num_alignment_padding_bytes = mz_zip_writer_compute_padding_needed_for_file_alignment(pZip);
6200 if (((mz_uint64)pState->m_central_dir.m_size + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + archive_name_size + MZ_ZIP64_MAX_CENTRAL_EXTRA_FIELD_SIZE + comment_size) >= MZ_UINT32_MAX)
6201 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_CDIR_SIZE);
6203 if (!pState->m_zip64)
6206 if ((pZip->m_archive_size + num_alignment_padding_bytes + MZ_ZIP_LOCAL_DIR_HEADER_SIZE + archive_name_size
6207 + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + archive_name_size + comment_size + user_extra_data_len +
6208 pState->m_central_dir.m_size + MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE + user_extra_data_central_len
6209 + MZ_ZIP_DATA_DESCRIPTER_SIZE32) > 0xFFFFFFFF)
6211 pState->m_zip64 = MZ_TRUE;
6216 if ((archive_name_size) && (pArchive_name[archive_name_size - 1] ==
'/'))
6219 ext_attributes |= MZ_ZIP_DOS_DIR_ATTRIBUTE_BITFLAG;
6222 if ((buf_size) || (uncomp_size))
6223 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
6227 if ((!mz_zip_array_ensure_room(pZip, &pState->m_central_dir, MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + archive_name_size + comment_size + (pState->m_zip64 ? MZ_ZIP64_MAX_CENTRAL_EXTRA_FIELD_SIZE : 0))) || (!mz_zip_array_ensure_room(pZip, &pState->m_central_dir_offsets, 1)))
6228 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
6230 if ((!store_data_uncompressed) && (buf_size))
6233 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
6236 if (!mz_zip_writer_write_zeros(pZip, cur_archive_file_ofs, num_alignment_padding_bytes))
6238 pZip->m_pFree(pZip->m_pAlloc_opaque, pComp);
6242 local_dir_header_ofs += num_alignment_padding_bytes;
6243 if (pZip->m_file_offset_alignment)
6245 MZ_ASSERT((local_dir_header_ofs & (pZip->m_file_offset_alignment - 1)) == 0);
6247 cur_archive_file_ofs += num_alignment_padding_bytes;
6249 MZ_CLEAR_OBJ(local_dir_header);
6251 if (!store_data_uncompressed || (level_and_flags & MZ_ZIP_FLAG_COMPRESSED_DATA))
6253 method = MZ_DEFLATED;
6256 if (pState->m_zip64)
6258 if (uncomp_size >= MZ_UINT32_MAX || local_dir_header_ofs >= MZ_UINT32_MAX)
6260 pExtra_data = extra_data;
6261 extra_size = mz_zip_writer_create_zip64_extra_data(extra_data, (uncomp_size >= MZ_UINT32_MAX) ? &uncomp_size : NULL,
6262 (uncomp_size >= MZ_UINT32_MAX) ? &comp_size : NULL, (local_dir_header_ofs >= MZ_UINT32_MAX) ? &local_dir_header_ofs : NULL);
6265 if (!mz_zip_writer_create_local_dir_header(pZip, local_dir_header, (mz_uint16)archive_name_size, (mz_uint16)(extra_size + user_extra_data_len), 0, 0, 0, method, bit_flags, dos_time, dos_date))
6266 return mz_zip_set_error(pZip, MZ_ZIP_INTERNAL_ERROR);
6268 if (pZip->m_pWrite(pZip->m_pIO_opaque, local_dir_header_ofs, local_dir_header,
sizeof(local_dir_header)) !=
sizeof(local_dir_header))
6269 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6271 cur_archive_file_ofs +=
sizeof(local_dir_header);
6273 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, pArchive_name, archive_name_size) != archive_name_size)
6275 pZip->m_pFree(pZip->m_pAlloc_opaque, pComp);
6276 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6278 cur_archive_file_ofs += archive_name_size;
6280 if (pExtra_data != NULL)
6282 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, extra_data, extra_size) != extra_size)
6283 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6285 cur_archive_file_ofs += extra_size;
6290 if ((comp_size > MZ_UINT32_MAX) || (cur_archive_file_ofs > MZ_UINT32_MAX))
6291 return mz_zip_set_error(pZip, MZ_ZIP_ARCHIVE_TOO_LARGE);
6292 if (!mz_zip_writer_create_local_dir_header(pZip, local_dir_header, (mz_uint16)archive_name_size, (mz_uint16)user_extra_data_len, 0, 0, 0, method, bit_flags, dos_time, dos_date))
6293 return mz_zip_set_error(pZip, MZ_ZIP_INTERNAL_ERROR);
6295 if (pZip->m_pWrite(pZip->m_pIO_opaque, local_dir_header_ofs, local_dir_header,
sizeof(local_dir_header)) !=
sizeof(local_dir_header))
6296 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6298 cur_archive_file_ofs +=
sizeof(local_dir_header);
6300 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, pArchive_name, archive_name_size) != archive_name_size)
6302 pZip->m_pFree(pZip->m_pAlloc_opaque, pComp);
6303 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6305 cur_archive_file_ofs += archive_name_size;
6308 if (user_extra_data_len > 0)
6310 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, user_extra_data, user_extra_data_len) != user_extra_data_len)
6311 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6313 cur_archive_file_ofs += user_extra_data_len;
6316 if (store_data_uncompressed)
6318 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, pBuf, buf_size) != buf_size)
6320 pZip->m_pFree(pZip->m_pAlloc_opaque, pComp);
6321 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6324 cur_archive_file_ofs += buf_size;
6325 comp_size = buf_size;
6331 state.m_pZip = pZip;
6332 state.m_cur_archive_file_ofs = cur_archive_file_ofs;
6333 state.m_comp_size = 0;
6335 if ((tdefl_init(pComp, mz_zip_writer_add_put_buf_callback, &state, tdefl_create_comp_flags_from_zip_params(level, -15, MZ_DEFAULT_STRATEGY)) != TDEFL_STATUS_OKAY) ||
6336 (tdefl_compress_buffer(pComp, pBuf, buf_size, TDEFL_FINISH) != TDEFL_STATUS_DONE))
6338 pZip->m_pFree(pZip->m_pAlloc_opaque, pComp);
6339 return mz_zip_set_error(pZip, MZ_ZIP_COMPRESSION_FAILED);
6342 comp_size = state.m_comp_size;
6343 cur_archive_file_ofs = state.m_cur_archive_file_ofs;
6346 pZip->m_pFree(pZip->m_pAlloc_opaque, pComp);
6351 mz_uint8 local_dir_footer[MZ_ZIP_DATA_DESCRIPTER_SIZE64];
6352 mz_uint32 local_dir_footer_size = MZ_ZIP_DATA_DESCRIPTER_SIZE32;
6354 MZ_ASSERT(bit_flags & MZ_ZIP_LDH_BIT_FLAG_HAS_LOCATOR);
6356 MZ_WRITE_LE32(local_dir_footer + 0, MZ_ZIP_DATA_DESCRIPTOR_ID);
6357 MZ_WRITE_LE32(local_dir_footer + 4, uncomp_crc32);
6358 if (pExtra_data == NULL)
6360 if (comp_size > MZ_UINT32_MAX)
6361 return mz_zip_set_error(pZip, MZ_ZIP_ARCHIVE_TOO_LARGE);
6363 MZ_WRITE_LE32(local_dir_footer + 8, comp_size);
6364 MZ_WRITE_LE32(local_dir_footer + 12, uncomp_size);
6368 MZ_WRITE_LE64(local_dir_footer + 8, comp_size);
6369 MZ_WRITE_LE64(local_dir_footer + 16, uncomp_size);
6370 local_dir_footer_size = MZ_ZIP_DATA_DESCRIPTER_SIZE64;
6373 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, local_dir_footer, local_dir_footer_size) != local_dir_footer_size)
6376 cur_archive_file_ofs += local_dir_footer_size;
6379 if (pExtra_data != NULL)
6381 extra_size = mz_zip_writer_create_zip64_extra_data(extra_data, (uncomp_size >= MZ_UINT32_MAX) ? &uncomp_size : NULL,
6382 (uncomp_size >= MZ_UINT32_MAX) ? &comp_size : NULL, (local_dir_header_ofs >= MZ_UINT32_MAX) ? &local_dir_header_ofs : NULL);
6385 if (!mz_zip_writer_add_to_central_dir(pZip, pArchive_name, (mz_uint16)archive_name_size, pExtra_data, (mz_uint16)extra_size, pComment,
6386 comment_size, uncomp_size, comp_size, uncomp_crc32, method, bit_flags, dos_time, dos_date, local_dir_header_ofs, ext_attributes,
6387 user_extra_data_central, user_extra_data_central_len))
6390 pZip->m_total_files++;
6391 pZip->m_archive_size = cur_archive_file_ofs;
6396mz_bool mz_zip_writer_add_read_buf_callback(
mz_zip_archive *pZip,
const char *pArchive_name, mz_file_read_func read_callback,
void* callback_opaque, mz_uint64 size_to_add,
const MZ_TIME_T *pFile_time,
const void *pComment, mz_uint16 comment_size, mz_uint level_and_flags,
6397 const char *user_extra_data, mz_uint user_extra_data_len,
const char *user_extra_data_central, mz_uint user_extra_data_central_len)
6399 mz_uint16 gen_flags = MZ_ZIP_LDH_BIT_FLAG_HAS_LOCATOR;
6400 mz_uint uncomp_crc32 = MZ_CRC32_INIT, level, num_alignment_padding_bytes;
6401 mz_uint16 method = 0, dos_time = 0, dos_date = 0, ext_attributes = 0;
6402 mz_uint64 local_dir_header_ofs, cur_archive_file_ofs = pZip->m_archive_size, uncomp_size = size_to_add, comp_size = 0;
6403 size_t archive_name_size;
6404 mz_uint8 local_dir_header[MZ_ZIP_LOCAL_DIR_HEADER_SIZE];
6405 mz_uint8 *pExtra_data = NULL;
6406 mz_uint32 extra_size = 0;
6407 mz_uint8 extra_data[MZ_ZIP64_MAX_CENTRAL_EXTRA_FIELD_SIZE];
6409 mz_uint64 file_ofs = 0;
6411 if (!(level_and_flags & MZ_ZIP_FLAG_ASCII_FILENAME))
6412 gen_flags |= MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_UTF8;
6414 if ((
int)level_and_flags < 0)
6415 level_and_flags = MZ_DEFAULT_LEVEL;
6416 level = level_and_flags & 0xF;
6419 if ((!pZip) || (!pZip->m_pState) || (pZip->m_zip_mode != MZ_ZIP_MODE_WRITING) || (!pArchive_name) || ((comment_size) && (!pComment)) || (level > MZ_UBER_COMPRESSION))
6420 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
6422 pState = pZip->m_pState;
6424 if ((!pState->m_zip64) && (uncomp_size > MZ_UINT32_MAX))
6428 pState->m_zip64 = MZ_TRUE;
6432 if (level_and_flags & MZ_ZIP_FLAG_COMPRESSED_DATA)
6433 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
6435 if (!mz_zip_writer_validate_archive_name(pArchive_name))
6436 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_FILENAME);
6438 if (pState->m_zip64)
6440 if (pZip->m_total_files == MZ_UINT32_MAX)
6441 return mz_zip_set_error(pZip, MZ_ZIP_TOO_MANY_FILES);
6445 if (pZip->m_total_files == MZ_UINT16_MAX)
6447 pState->m_zip64 = MZ_TRUE;
6452 archive_name_size = strlen(pArchive_name);
6453 if (archive_name_size > MZ_UINT16_MAX)
6454 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_FILENAME);
6456 num_alignment_padding_bytes = mz_zip_writer_compute_padding_needed_for_file_alignment(pZip);
6459 if (((mz_uint64)pState->m_central_dir.m_size + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + archive_name_size + MZ_ZIP64_MAX_CENTRAL_EXTRA_FIELD_SIZE + comment_size) >= MZ_UINT32_MAX)
6460 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_CDIR_SIZE);
6462 if (!pState->m_zip64)
6465 if ((pZip->m_archive_size + num_alignment_padding_bytes + MZ_ZIP_LOCAL_DIR_HEADER_SIZE + archive_name_size + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE
6466 + archive_name_size + comment_size + user_extra_data_len + pState->m_central_dir.m_size + MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE + 1024
6467 + MZ_ZIP_DATA_DESCRIPTER_SIZE32 + user_extra_data_central_len) > 0xFFFFFFFF)
6469 pState->m_zip64 = MZ_TRUE;
6474#ifndef MINIZ_NO_TIME
6477 mz_zip_time_t_to_dos_time(*pFile_time, &dos_time, &dos_date);
6481 if (uncomp_size <= 3)
6484 if (!mz_zip_writer_write_zeros(pZip, cur_archive_file_ofs, num_alignment_padding_bytes))
6486 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6489 cur_archive_file_ofs += num_alignment_padding_bytes;
6490 local_dir_header_ofs = cur_archive_file_ofs;
6492 if (pZip->m_file_offset_alignment)
6494 MZ_ASSERT((cur_archive_file_ofs & (pZip->m_file_offset_alignment - 1)) == 0);
6497 if (uncomp_size && level)
6499 method = MZ_DEFLATED;
6502 MZ_CLEAR_OBJ(local_dir_header);
6503 if (pState->m_zip64)
6505 if (uncomp_size >= MZ_UINT32_MAX || local_dir_header_ofs >= MZ_UINT32_MAX)
6507 pExtra_data = extra_data;
6508 extra_size = mz_zip_writer_create_zip64_extra_data(extra_data, (uncomp_size >= MZ_UINT32_MAX) ? &uncomp_size : NULL,
6509 (uncomp_size >= MZ_UINT32_MAX) ? &comp_size : NULL, (local_dir_header_ofs >= MZ_UINT32_MAX) ? &local_dir_header_ofs : NULL);
6512 if (!mz_zip_writer_create_local_dir_header(pZip, local_dir_header, (mz_uint16)archive_name_size, (mz_uint16)(extra_size + user_extra_data_len), 0, 0, 0, method, gen_flags, dos_time, dos_date))
6513 return mz_zip_set_error(pZip, MZ_ZIP_INTERNAL_ERROR);
6515 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, local_dir_header,
sizeof(local_dir_header)) !=
sizeof(local_dir_header))
6516 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6518 cur_archive_file_ofs +=
sizeof(local_dir_header);
6520 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, pArchive_name, archive_name_size) != archive_name_size)
6522 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6525 cur_archive_file_ofs += archive_name_size;
6527 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, extra_data, extra_size) != extra_size)
6528 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6530 cur_archive_file_ofs += extra_size;
6534 if ((comp_size > MZ_UINT32_MAX) || (cur_archive_file_ofs > MZ_UINT32_MAX))
6535 return mz_zip_set_error(pZip, MZ_ZIP_ARCHIVE_TOO_LARGE);
6536 if (!mz_zip_writer_create_local_dir_header(pZip, local_dir_header, (mz_uint16)archive_name_size, (mz_uint16)user_extra_data_len, 0, 0, 0, method, gen_flags, dos_time, dos_date))
6537 return mz_zip_set_error(pZip, MZ_ZIP_INTERNAL_ERROR);
6539 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, local_dir_header,
sizeof(local_dir_header)) !=
sizeof(local_dir_header))
6540 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6542 cur_archive_file_ofs +=
sizeof(local_dir_header);
6544 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, pArchive_name, archive_name_size) != archive_name_size)
6546 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6549 cur_archive_file_ofs += archive_name_size;
6552 if (user_extra_data_len > 0)
6554 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, user_extra_data, user_extra_data_len) != user_extra_data_len)
6555 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
6557 cur_archive_file_ofs += user_extra_data_len;
6562 mz_uint64 uncomp_remaining = uncomp_size;
6563 void *pRead_buf = pZip->m_pAlloc(pZip->m_pAlloc_opaque, 1, MZ_ZIP_MAX_IO_BUF_SIZE);
6566 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
6571 while (uncomp_remaining)
6573 mz_uint n = (mz_uint)MZ_MIN((mz_uint64)MZ_ZIP_MAX_IO_BUF_SIZE, uncomp_remaining);
6574 if ((read_callback(callback_opaque, file_ofs, pRead_buf, n) != n) || (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, pRead_buf, n) != n))
6576 pZip->m_pFree(pZip->m_pAlloc_opaque, pRead_buf);
6577 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
6580 uncomp_crc32 = (mz_uint32)mz_crc32(uncomp_crc32, (
const mz_uint8 *)pRead_buf, n);
6581 uncomp_remaining -= n;
6582 cur_archive_file_ofs += n;
6584 comp_size = uncomp_size;
6588 mz_bool result = MZ_FALSE;
6593 pZip->m_pFree(pZip->m_pAlloc_opaque, pRead_buf);
6594 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
6597 state.m_pZip = pZip;
6598 state.m_cur_archive_file_ofs = cur_archive_file_ofs;
6599 state.m_comp_size = 0;
6601 if (tdefl_init(pComp, mz_zip_writer_add_put_buf_callback, &state, tdefl_create_comp_flags_from_zip_params(level, -15, MZ_DEFAULT_STRATEGY)) != TDEFL_STATUS_OKAY)
6603 pZip->m_pFree(pZip->m_pAlloc_opaque, pComp);
6604 pZip->m_pFree(pZip->m_pAlloc_opaque, pRead_buf);
6605 return mz_zip_set_error(pZip, MZ_ZIP_INTERNAL_ERROR);
6610 size_t in_buf_size = (mz_uint32)MZ_MIN(uncomp_remaining, (mz_uint64)MZ_ZIP_MAX_IO_BUF_SIZE);
6611 tdefl_status status;
6612 tdefl_flush
flush = TDEFL_NO_FLUSH;
6614 if (read_callback(callback_opaque, file_ofs, pRead_buf, in_buf_size)!= in_buf_size)
6616 mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
6620 file_ofs += in_buf_size;
6621 uncomp_crc32 = (mz_uint32)mz_crc32(uncomp_crc32, (
const mz_uint8 *)pRead_buf, in_buf_size);
6622 uncomp_remaining -= in_buf_size;
6624 if (pZip->m_pNeeds_keepalive != NULL && pZip->m_pNeeds_keepalive(pZip->m_pIO_opaque))
6625 flush = TDEFL_FULL_FLUSH;
6627 status = tdefl_compress_buffer(pComp, pRead_buf, in_buf_size, uncomp_remaining ? flush : TDEFL_FINISH);
6628 if (status == TDEFL_STATUS_DONE)
6633 else if (status != TDEFL_STATUS_OKAY)
6635 mz_zip_set_error(pZip, MZ_ZIP_COMPRESSION_FAILED);
6640 pZip->m_pFree(pZip->m_pAlloc_opaque, pComp);
6644 pZip->m_pFree(pZip->m_pAlloc_opaque, pRead_buf);
6648 comp_size = state.m_comp_size;
6649 cur_archive_file_ofs = state.m_cur_archive_file_ofs;
6652 pZip->m_pFree(pZip->m_pAlloc_opaque, pRead_buf);
6656 mz_uint8 local_dir_footer[MZ_ZIP_DATA_DESCRIPTER_SIZE64];
6657 mz_uint32 local_dir_footer_size = MZ_ZIP_DATA_DESCRIPTER_SIZE32;
6659 MZ_WRITE_LE32(local_dir_footer + 0, MZ_ZIP_DATA_DESCRIPTOR_ID);
6660 MZ_WRITE_LE32(local_dir_footer + 4, uncomp_crc32);
6661 if (pExtra_data == NULL)
6663 if (comp_size > MZ_UINT32_MAX)
6664 return mz_zip_set_error(pZip, MZ_ZIP_ARCHIVE_TOO_LARGE);
6666 MZ_WRITE_LE32(local_dir_footer + 8, comp_size);
6667 MZ_WRITE_LE32(local_dir_footer + 12, uncomp_size);
6671 MZ_WRITE_LE64(local_dir_footer + 8, comp_size);
6672 MZ_WRITE_LE64(local_dir_footer + 16, uncomp_size);
6673 local_dir_footer_size = MZ_ZIP_DATA_DESCRIPTER_SIZE64;
6676 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_archive_file_ofs, local_dir_footer, local_dir_footer_size) != local_dir_footer_size)
6679 cur_archive_file_ofs += local_dir_footer_size;
6682 if (pExtra_data != NULL)
6684 extra_size = mz_zip_writer_create_zip64_extra_data(extra_data, (uncomp_size >= MZ_UINT32_MAX) ? &uncomp_size : NULL,
6685 (uncomp_size >= MZ_UINT32_MAX) ? &comp_size : NULL, (local_dir_header_ofs >= MZ_UINT32_MAX) ? &local_dir_header_ofs : NULL);
6688 if (!mz_zip_writer_add_to_central_dir(pZip, pArchive_name, (mz_uint16)archive_name_size, pExtra_data, (mz_uint16)extra_size, pComment, comment_size,
6689 uncomp_size, comp_size, uncomp_crc32, method, gen_flags, dos_time, dos_date, local_dir_header_ofs, ext_attributes,
6690 user_extra_data_central, user_extra_data_central_len))
6693 pZip->m_total_files++;
6694 pZip->m_archive_size = cur_archive_file_ofs;
6699#ifndef MINIZ_NO_STDIO
6701static size_t mz_file_read_func_stdio(
void *pOpaque, mz_uint64 file_ofs,
void *pBuf,
size_t n)
6703 MZ_FILE *pSrc_file = (MZ_FILE *)pOpaque;
6704 mz_int64 cur_ofs = MZ_FTELL64(pSrc_file);
6706 if (((mz_int64)file_ofs < 0) || (((cur_ofs != (mz_int64)file_ofs)) && (MZ_FSEEK64(pSrc_file, (mz_int64)file_ofs, SEEK_SET))))
6709 return MZ_FREAD(pBuf, 1, n, pSrc_file);
6712mz_bool mz_zip_writer_add_cfile(
mz_zip_archive *pZip,
const char *pArchive_name, MZ_FILE *pSrc_file, mz_uint64 size_to_add,
const MZ_TIME_T *pFile_time,
const void *pComment, mz_uint16 comment_size, mz_uint level_and_flags,
6713 const char *user_extra_data, mz_uint user_extra_data_len,
const char *user_extra_data_central, mz_uint user_extra_data_central_len)
6715 return mz_zip_writer_add_read_buf_callback(pZip, pArchive_name, mz_file_read_func_stdio, pSrc_file, size_to_add, pFile_time, pComment, comment_size, level_and_flags,
6716 user_extra_data, user_extra_data_len, user_extra_data_central, user_extra_data_central_len);
6719mz_bool mz_zip_writer_add_file(
mz_zip_archive *pZip,
const char *pArchive_name,
const char *pSrc_filename,
const void *pComment, mz_uint16 comment_size, mz_uint level_and_flags)
6721 MZ_FILE *pSrc_file = NULL;
6722 mz_uint64 uncomp_size = 0;
6723 MZ_TIME_T file_modified_time;
6724 MZ_TIME_T *pFile_time = NULL;
6727 memset(&file_modified_time, 0,
sizeof(file_modified_time));
6729#if !defined(MINIZ_NO_TIME) && !defined(MINIZ_NO_STDIO)
6730 pFile_time = &file_modified_time;
6731 if (!mz_zip_get_file_modified_time(pSrc_filename, &file_modified_time))
6732 return mz_zip_set_error(pZip, MZ_ZIP_FILE_STAT_FAILED);
6735 pSrc_file = MZ_FOPEN(pSrc_filename,
"rb");
6737 return mz_zip_set_error(pZip, MZ_ZIP_FILE_OPEN_FAILED);
6739 MZ_FSEEK64(pSrc_file, 0, SEEK_END);
6740 uncomp_size = MZ_FTELL64(pSrc_file);
6741 MZ_FSEEK64(pSrc_file, 0, SEEK_SET);
6743 status = mz_zip_writer_add_cfile(pZip, pArchive_name, pSrc_file, uncomp_size, pFile_time, pComment, comment_size, level_and_flags, NULL, 0, NULL, 0);
6745 MZ_FCLOSE(pSrc_file);
6751static mz_bool mz_zip_writer_update_zip64_extension_block(
mz_zip_array *pNew_ext,
mz_zip_archive *pZip,
const mz_uint8 *pExt, uint32_t ext_len, mz_uint64 *pComp_size, mz_uint64 *pUncomp_size, mz_uint64 *pLocal_header_ofs, mz_uint32 *pDisk_start)
6754 if (!mz_zip_array_reserve(pZip, pNew_ext, ext_len + 64, MZ_FALSE))
6755 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
6757 mz_zip_array_resize(pZip, pNew_ext, 0, MZ_FALSE);
6759 if ((pUncomp_size) || (pComp_size) || (pLocal_header_ofs) || (pDisk_start))
6761 mz_uint8 new_ext_block[64];
6762 mz_uint8 *pDst = new_ext_block;
6763 mz_write_le16(pDst, MZ_ZIP64_EXTENDED_INFORMATION_FIELD_HEADER_ID);
6764 mz_write_le16(pDst +
sizeof(mz_uint16), 0);
6765 pDst +=
sizeof(mz_uint16) * 2;
6769 mz_write_le64(pDst, *pUncomp_size);
6770 pDst +=
sizeof(mz_uint64);
6775 mz_write_le64(pDst, *pComp_size);
6776 pDst +=
sizeof(mz_uint64);
6779 if (pLocal_header_ofs)
6781 mz_write_le64(pDst, *pLocal_header_ofs);
6782 pDst +=
sizeof(mz_uint64);
6787 mz_write_le32(pDst, *pDisk_start);
6788 pDst +=
sizeof(mz_uint32);
6791 mz_write_le16(new_ext_block +
sizeof(mz_uint16), (mz_uint16)((pDst - new_ext_block) -
sizeof(mz_uint16) * 2));
6793 if (!mz_zip_array_push_back(pZip, pNew_ext, new_ext_block, pDst - new_ext_block))
6794 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
6797 if ((pExt) && (ext_len))
6799 mz_uint32 extra_size_remaining = ext_len;
6800 const mz_uint8 *pExtra_data = pExt;
6804 mz_uint32 field_id, field_data_size, field_total_size;
6806 if (extra_size_remaining < (
sizeof(mz_uint16) * 2))
6807 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
6809 field_id = MZ_READ_LE16(pExtra_data);
6810 field_data_size = MZ_READ_LE16(pExtra_data +
sizeof(mz_uint16));
6811 field_total_size = field_data_size +
sizeof(mz_uint16) * 2;
6813 if (field_total_size > extra_size_remaining)
6814 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
6816 if (field_id != MZ_ZIP64_EXTENDED_INFORMATION_FIELD_HEADER_ID)
6818 if (!mz_zip_array_push_back(pZip, pNew_ext, pExtra_data, field_total_size))
6819 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
6822 pExtra_data += field_total_size;
6823 extra_size_remaining -= field_total_size;
6824 }
while (extra_size_remaining);
6833 mz_uint n, bit_flags, num_alignment_padding_bytes, src_central_dir_following_data_size;
6834 mz_uint64 src_archive_bytes_remaining, local_dir_header_ofs;
6835 mz_uint64 cur_src_file_ofs, cur_dst_file_ofs;
6836 mz_uint32 local_header_u32[(MZ_ZIP_LOCAL_DIR_HEADER_SIZE +
sizeof(mz_uint32) - 1) /
sizeof(mz_uint32)];
6837 mz_uint8 *pLocal_header = (mz_uint8 *)local_header_u32;
6838 mz_uint8 new_central_header[MZ_ZIP_CENTRAL_DIR_HEADER_SIZE];
6839 size_t orig_central_dir_size;
6842 const mz_uint8 *pSrc_central_header;
6844 mz_uint32 src_filename_len, src_comment_len, src_ext_len;
6845 mz_uint32 local_header_filename_size, local_header_extra_len;
6846 mz_uint64 local_header_comp_size, local_header_uncomp_size;
6847 mz_bool found_zip64_ext_data_in_ldir = MZ_FALSE;
6850 if ((!pZip) || (!pZip->m_pState) || (pZip->m_zip_mode != MZ_ZIP_MODE_WRITING) || (!pSource_zip->m_pRead))
6851 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
6853 pState = pZip->m_pState;
6856 if ((pSource_zip->m_pState->m_zip64) && (!pZip->m_pState->m_zip64))
6857 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
6860 if (NULL == (pSrc_central_header = mz_zip_get_cdh(pSource_zip, src_file_index)))
6861 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
6863 if (MZ_READ_LE32(pSrc_central_header + MZ_ZIP_CDH_SIG_OFS) != MZ_ZIP_CENTRAL_DIR_HEADER_SIG)
6864 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
6866 src_filename_len = MZ_READ_LE16(pSrc_central_header + MZ_ZIP_CDH_FILENAME_LEN_OFS);
6867 src_comment_len = MZ_READ_LE16(pSrc_central_header + MZ_ZIP_CDH_COMMENT_LEN_OFS);
6868 src_ext_len = MZ_READ_LE16(pSrc_central_header + MZ_ZIP_CDH_EXTRA_LEN_OFS);
6869 src_central_dir_following_data_size = src_filename_len + src_ext_len + src_comment_len;
6872 if ((pState->m_central_dir.m_size + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + src_central_dir_following_data_size + 32) >= MZ_UINT32_MAX)
6873 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_CDIR_SIZE);
6875 num_alignment_padding_bytes = mz_zip_writer_compute_padding_needed_for_file_alignment(pZip);
6877 if (!pState->m_zip64)
6879 if (pZip->m_total_files == MZ_UINT16_MAX)
6880 return mz_zip_set_error(pZip, MZ_ZIP_TOO_MANY_FILES);
6885 if (pZip->m_total_files == MZ_UINT32_MAX)
6886 return mz_zip_set_error(pZip, MZ_ZIP_TOO_MANY_FILES);
6889 if (!mz_zip_file_stat_internal(pSource_zip, src_file_index, pSrc_central_header, &src_file_stat, NULL))
6892 cur_src_file_ofs = src_file_stat.m_local_header_ofs;
6893 cur_dst_file_ofs = pZip->m_archive_size;
6896 if (pSource_zip->m_pRead(pSource_zip->m_pIO_opaque, cur_src_file_ofs, pLocal_header, MZ_ZIP_LOCAL_DIR_HEADER_SIZE) != MZ_ZIP_LOCAL_DIR_HEADER_SIZE)
6897 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
6899 if (MZ_READ_LE32(pLocal_header) != MZ_ZIP_LOCAL_DIR_HEADER_SIG)
6900 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
6902 cur_src_file_ofs += MZ_ZIP_LOCAL_DIR_HEADER_SIZE;
6905 local_header_filename_size = MZ_READ_LE16(pLocal_header + MZ_ZIP_LDH_FILENAME_LEN_OFS);
6906 local_header_extra_len = MZ_READ_LE16(pLocal_header + MZ_ZIP_LDH_EXTRA_LEN_OFS);
6907 local_header_comp_size = MZ_READ_LE32(pLocal_header + MZ_ZIP_LDH_COMPRESSED_SIZE_OFS);
6908 local_header_uncomp_size = MZ_READ_LE32(pLocal_header + MZ_ZIP_LDH_DECOMPRESSED_SIZE_OFS);
6909 src_archive_bytes_remaining = local_header_filename_size + local_header_extra_len + src_file_stat.m_comp_size;
6912 if ((local_header_extra_len) && ((local_header_comp_size == MZ_UINT32_MAX) || (local_header_uncomp_size == MZ_UINT32_MAX)))
6915 const mz_uint8 *pExtra_data;
6916 mz_uint32 extra_size_remaining = local_header_extra_len;
6918 mz_zip_array_init(&file_data_array, 1);
6919 if (!mz_zip_array_resize(pZip, &file_data_array, local_header_extra_len, MZ_FALSE))
6921 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
6924 if (pSource_zip->m_pRead(pSource_zip->m_pIO_opaque, src_file_stat.m_local_header_ofs + MZ_ZIP_LOCAL_DIR_HEADER_SIZE + local_header_filename_size, file_data_array.m_p, local_header_extra_len) != local_header_extra_len)
6926 mz_zip_array_clear(pZip, &file_data_array);
6927 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
6930 pExtra_data = (
const mz_uint8 *)file_data_array.m_p;
6934 mz_uint32 field_id, field_data_size, field_total_size;
6936 if (extra_size_remaining < (
sizeof(mz_uint16) * 2))
6938 mz_zip_array_clear(pZip, &file_data_array);
6939 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
6942 field_id = MZ_READ_LE16(pExtra_data);
6943 field_data_size = MZ_READ_LE16(pExtra_data +
sizeof(mz_uint16));
6944 field_total_size = field_data_size +
sizeof(mz_uint16) * 2;
6946 if (field_total_size > extra_size_remaining)
6948 mz_zip_array_clear(pZip, &file_data_array);
6949 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
6952 if (field_id == MZ_ZIP64_EXTENDED_INFORMATION_FIELD_HEADER_ID)
6954 const mz_uint8 *pSrc_field_data = pExtra_data +
sizeof(mz_uint32);
6956 if (field_data_size <
sizeof(mz_uint64) * 2)
6958 mz_zip_array_clear(pZip, &file_data_array);
6959 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_HEADER_OR_CORRUPTED);
6962 local_header_uncomp_size = MZ_READ_LE64(pSrc_field_data);
6963 local_header_comp_size = MZ_READ_LE64(pSrc_field_data +
sizeof(mz_uint64));
6965 found_zip64_ext_data_in_ldir = MZ_TRUE;
6969 pExtra_data += field_total_size;
6970 extra_size_remaining -= field_total_size;
6971 }
while (extra_size_remaining);
6973 mz_zip_array_clear(pZip, &file_data_array);
6976 if (!pState->m_zip64)
6980 mz_uint64 approx_new_archive_size = cur_dst_file_ofs + num_alignment_padding_bytes + MZ_ZIP_LOCAL_DIR_HEADER_SIZE + src_archive_bytes_remaining + (
sizeof(mz_uint32) * 4) +
6981 pState->m_central_dir.m_size + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + src_central_dir_following_data_size + MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE + 64;
6983 if (approx_new_archive_size >= MZ_UINT32_MAX)
6984 return mz_zip_set_error(pZip, MZ_ZIP_ARCHIVE_TOO_LARGE);
6988 if (!mz_zip_writer_write_zeros(pZip, cur_dst_file_ofs, num_alignment_padding_bytes))
6991 cur_dst_file_ofs += num_alignment_padding_bytes;
6993 local_dir_header_ofs = cur_dst_file_ofs;
6994 if (pZip->m_file_offset_alignment)
6996 MZ_ASSERT((local_dir_header_ofs & (pZip->m_file_offset_alignment - 1)) == 0);
7000 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_dst_file_ofs, pLocal_header, MZ_ZIP_LOCAL_DIR_HEADER_SIZE) != MZ_ZIP_LOCAL_DIR_HEADER_SIZE)
7001 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
7003 cur_dst_file_ofs += MZ_ZIP_LOCAL_DIR_HEADER_SIZE;
7006 if (NULL == (pBuf = pZip->m_pAlloc(pZip->m_pAlloc_opaque, 1, (
size_t)MZ_MAX(32U, MZ_MIN((mz_uint64)MZ_ZIP_MAX_IO_BUF_SIZE, src_archive_bytes_remaining)))))
7007 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
7009 while (src_archive_bytes_remaining)
7011 n = (mz_uint)MZ_MIN((mz_uint64)MZ_ZIP_MAX_IO_BUF_SIZE, src_archive_bytes_remaining);
7012 if (pSource_zip->m_pRead(pSource_zip->m_pIO_opaque, cur_src_file_ofs, pBuf, n) != n)
7014 pZip->m_pFree(pZip->m_pAlloc_opaque, pBuf);
7015 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
7017 cur_src_file_ofs += n;
7019 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_dst_file_ofs, pBuf, n) != n)
7021 pZip->m_pFree(pZip->m_pAlloc_opaque, pBuf);
7022 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
7024 cur_dst_file_ofs += n;
7026 src_archive_bytes_remaining -= n;
7030 bit_flags = MZ_READ_LE16(pLocal_header + MZ_ZIP_LDH_BIT_FLAG_OFS);
7034 if ((pSource_zip->m_pState->m_zip64) || (found_zip64_ext_data_in_ldir))
7043 if (pSource_zip->m_pRead(pSource_zip->m_pIO_opaque, cur_src_file_ofs, pBuf, (
sizeof(mz_uint32) * 6)) != (
sizeof(mz_uint32) * 6))
7045 pZip->m_pFree(pZip->m_pAlloc_opaque, pBuf);
7046 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
7049 n =
sizeof(mz_uint32) * ((MZ_READ_LE32(pBuf) == MZ_ZIP_DATA_DESCRIPTOR_ID) ? 6 : 5);
7056 if (pSource_zip->m_pRead(pSource_zip->m_pIO_opaque, cur_src_file_ofs, pBuf,
sizeof(mz_uint32) * 4) !=
sizeof(mz_uint32) * 4)
7058 pZip->m_pFree(pZip->m_pAlloc_opaque, pBuf);
7059 return mz_zip_set_error(pZip, MZ_ZIP_FILE_READ_FAILED);
7062 has_id = (MZ_READ_LE32(pBuf) == MZ_ZIP_DATA_DESCRIPTOR_ID);
7064 if (pZip->m_pState->m_zip64)
7067 const mz_uint32 *pSrc_descriptor = (
const mz_uint32 *)((
const mz_uint8 *)pBuf + (has_id ?
sizeof(mz_uint32) : 0));
7068 const mz_uint32 src_crc32 = pSrc_descriptor[0];
7069 const mz_uint64 src_comp_size = pSrc_descriptor[1];
7070 const mz_uint64 src_uncomp_size = pSrc_descriptor[2];
7072 mz_write_le32((mz_uint8 *)pBuf, MZ_ZIP_DATA_DESCRIPTOR_ID);
7073 mz_write_le32((mz_uint8 *)pBuf +
sizeof(mz_uint32) * 1, src_crc32);
7074 mz_write_le64((mz_uint8 *)pBuf +
sizeof(mz_uint32) * 2, src_comp_size);
7075 mz_write_le64((mz_uint8 *)pBuf +
sizeof(mz_uint32) * 4, src_uncomp_size);
7077 n =
sizeof(mz_uint32) * 6;
7082 n =
sizeof(mz_uint32) * (has_id ? 4 : 3);
7086 if (pZip->m_pWrite(pZip->m_pIO_opaque, cur_dst_file_ofs, pBuf, n) != n)
7088 pZip->m_pFree(pZip->m_pAlloc_opaque, pBuf);
7089 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
7092 cur_src_file_ofs += n;
7093 cur_dst_file_ofs += n;
7095 pZip->m_pFree(pZip->m_pAlloc_opaque, pBuf);
7098 orig_central_dir_size = pState->m_central_dir.m_size;
7100 memcpy(new_central_header, pSrc_central_header, MZ_ZIP_CENTRAL_DIR_HEADER_SIZE);
7102 if (pState->m_zip64)
7105 const mz_uint8 *pSrc_ext = pSrc_central_header + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + src_filename_len;
7108 mz_zip_array_init(&new_ext_block,
sizeof(mz_uint8));
7110 MZ_WRITE_LE32(new_central_header + MZ_ZIP_CDH_COMPRESSED_SIZE_OFS, MZ_UINT32_MAX);
7111 MZ_WRITE_LE32(new_central_header + MZ_ZIP_CDH_DECOMPRESSED_SIZE_OFS, MZ_UINT32_MAX);
7112 MZ_WRITE_LE32(new_central_header + MZ_ZIP_CDH_LOCAL_HEADER_OFS, MZ_UINT32_MAX);
7114 if (!mz_zip_writer_update_zip64_extension_block(&new_ext_block, pZip, pSrc_ext, src_ext_len, &src_file_stat.m_comp_size, &src_file_stat.m_uncomp_size, &local_dir_header_ofs, NULL))
7116 mz_zip_array_clear(pZip, &new_ext_block);
7120 MZ_WRITE_LE16(new_central_header + MZ_ZIP_CDH_EXTRA_LEN_OFS, new_ext_block.m_size);
7122 if (!mz_zip_array_push_back(pZip, &pState->m_central_dir, new_central_header, MZ_ZIP_CENTRAL_DIR_HEADER_SIZE))
7124 mz_zip_array_clear(pZip, &new_ext_block);
7125 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
7128 if (!mz_zip_array_push_back(pZip, &pState->m_central_dir, pSrc_central_header + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE, src_filename_len))
7130 mz_zip_array_clear(pZip, &new_ext_block);
7131 mz_zip_array_resize(pZip, &pState->m_central_dir, orig_central_dir_size, MZ_FALSE);
7132 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
7135 if (!mz_zip_array_push_back(pZip, &pState->m_central_dir, new_ext_block.m_p, new_ext_block.m_size))
7137 mz_zip_array_clear(pZip, &new_ext_block);
7138 mz_zip_array_resize(pZip, &pState->m_central_dir, orig_central_dir_size, MZ_FALSE);
7139 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
7142 if (!mz_zip_array_push_back(pZip, &pState->m_central_dir, pSrc_central_header + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE + src_filename_len + src_ext_len, src_comment_len))
7144 mz_zip_array_clear(pZip, &new_ext_block);
7145 mz_zip_array_resize(pZip, &pState->m_central_dir, orig_central_dir_size, MZ_FALSE);
7146 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
7149 mz_zip_array_clear(pZip, &new_ext_block);
7154 if (cur_dst_file_ofs > MZ_UINT32_MAX)
7155 return mz_zip_set_error(pZip, MZ_ZIP_ARCHIVE_TOO_LARGE);
7157 if (local_dir_header_ofs >= MZ_UINT32_MAX)
7158 return mz_zip_set_error(pZip, MZ_ZIP_ARCHIVE_TOO_LARGE);
7160 MZ_WRITE_LE32(new_central_header + MZ_ZIP_CDH_LOCAL_HEADER_OFS, local_dir_header_ofs);
7162 if (!mz_zip_array_push_back(pZip, &pState->m_central_dir, new_central_header, MZ_ZIP_CENTRAL_DIR_HEADER_SIZE))
7163 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
7165 if (!mz_zip_array_push_back(pZip, &pState->m_central_dir, pSrc_central_header + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE, src_central_dir_following_data_size))
7167 mz_zip_array_resize(pZip, &pState->m_central_dir, orig_central_dir_size, MZ_FALSE);
7168 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
7173 if (pState->m_central_dir.m_size >= MZ_UINT32_MAX)
7176 mz_zip_array_resize(pZip, &pState->m_central_dir, orig_central_dir_size, MZ_FALSE);
7177 return mz_zip_set_error(pZip, MZ_ZIP_UNSUPPORTED_CDIR_SIZE);
7180 n = (mz_uint32)orig_central_dir_size;
7181 if (!mz_zip_array_push_back(pZip, &pState->m_central_dir_offsets, &n, 1))
7183 mz_zip_array_resize(pZip, &pState->m_central_dir, orig_central_dir_size, MZ_FALSE);
7184 return mz_zip_set_error(pZip, MZ_ZIP_ALLOC_FAILED);
7187 pZip->m_total_files++;
7188 pZip->m_archive_size = cur_dst_file_ofs;
7196 mz_uint64 central_dir_ofs, central_dir_size;
7199 if ((!pZip) || (!pZip->m_pState) || (pZip->m_zip_mode != MZ_ZIP_MODE_WRITING))
7200 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
7202 pState = pZip->m_pState;
7204 if (pState->m_zip64)
7206 if ((pZip->m_total_files > MZ_UINT32_MAX) || (pState->m_central_dir.m_size >= MZ_UINT32_MAX))
7207 return mz_zip_set_error(pZip, MZ_ZIP_TOO_MANY_FILES);
7211 if ((pZip->m_total_files > MZ_UINT16_MAX) || ((pZip->m_archive_size + pState->m_central_dir.m_size + MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE) > MZ_UINT32_MAX))
7212 return mz_zip_set_error(pZip, MZ_ZIP_TOO_MANY_FILES);
7215 central_dir_ofs = 0;
7216 central_dir_size = 0;
7217 if (pZip->m_total_files)
7220 central_dir_ofs = pZip->m_archive_size;
7221 central_dir_size = pState->m_central_dir.m_size;
7222 pZip->m_central_directory_file_ofs = central_dir_ofs;
7223 if (pZip->m_pWrite(pZip->m_pIO_opaque, central_dir_ofs, pState->m_central_dir.m_p, (
size_t)central_dir_size) != central_dir_size)
7224 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
7226 pZip->m_archive_size += central_dir_size;
7229 if (pState->m_zip64)
7232 mz_uint64 rel_ofs_to_zip64_ecdr = pZip->m_archive_size;
7235 MZ_WRITE_LE32(hdr + MZ_ZIP64_ECDH_SIG_OFS, MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIG);
7236 MZ_WRITE_LE64(hdr + MZ_ZIP64_ECDH_SIZE_OF_RECORD_OFS, MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIZE -
sizeof(mz_uint32) -
sizeof(mz_uint64));
7237 MZ_WRITE_LE16(hdr + MZ_ZIP64_ECDH_VERSION_MADE_BY_OFS, 0x031E);
7238 MZ_WRITE_LE16(hdr + MZ_ZIP64_ECDH_VERSION_NEEDED_OFS, 0x002D);
7239 MZ_WRITE_LE64(hdr + MZ_ZIP64_ECDH_CDIR_NUM_ENTRIES_ON_DISK_OFS, pZip->m_total_files);
7240 MZ_WRITE_LE64(hdr + MZ_ZIP64_ECDH_CDIR_TOTAL_ENTRIES_OFS, pZip->m_total_files);
7241 MZ_WRITE_LE64(hdr + MZ_ZIP64_ECDH_CDIR_SIZE_OFS, central_dir_size);
7242 MZ_WRITE_LE64(hdr + MZ_ZIP64_ECDH_CDIR_OFS_OFS, central_dir_ofs);
7243 if (pZip->m_pWrite(pZip->m_pIO_opaque, pZip->m_archive_size, hdr, MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIZE) != MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIZE)
7244 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
7246 pZip->m_archive_size += MZ_ZIP64_END_OF_CENTRAL_DIR_HEADER_SIZE;
7250 MZ_WRITE_LE32(hdr + MZ_ZIP64_ECDL_SIG_OFS, MZ_ZIP64_END_OF_CENTRAL_DIR_LOCATOR_SIG);
7251 MZ_WRITE_LE64(hdr + MZ_ZIP64_ECDL_REL_OFS_TO_ZIP64_ECDR_OFS, rel_ofs_to_zip64_ecdr);
7252 MZ_WRITE_LE32(hdr + MZ_ZIP64_ECDL_TOTAL_NUMBER_OF_DISKS_OFS, 1);
7253 if (pZip->m_pWrite(pZip->m_pIO_opaque, pZip->m_archive_size, hdr, MZ_ZIP64_END_OF_CENTRAL_DIR_LOCATOR_SIZE) != MZ_ZIP64_END_OF_CENTRAL_DIR_LOCATOR_SIZE)
7254 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
7256 pZip->m_archive_size += MZ_ZIP64_END_OF_CENTRAL_DIR_LOCATOR_SIZE;
7261 MZ_WRITE_LE32(hdr + MZ_ZIP_ECDH_SIG_OFS, MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIG);
7262 MZ_WRITE_LE16(hdr + MZ_ZIP_ECDH_CDIR_NUM_ENTRIES_ON_DISK_OFS, MZ_MIN(MZ_UINT16_MAX, pZip->m_total_files));
7263 MZ_WRITE_LE16(hdr + MZ_ZIP_ECDH_CDIR_TOTAL_ENTRIES_OFS, MZ_MIN(MZ_UINT16_MAX, pZip->m_total_files));
7264 MZ_WRITE_LE32(hdr + MZ_ZIP_ECDH_CDIR_SIZE_OFS, MZ_MIN(MZ_UINT32_MAX, central_dir_size));
7265 MZ_WRITE_LE32(hdr + MZ_ZIP_ECDH_CDIR_OFS_OFS, MZ_MIN(MZ_UINT32_MAX, central_dir_ofs));
7267 if (pZip->m_pWrite(pZip->m_pIO_opaque, pZip->m_archive_size, hdr, MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE) != MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE)
7268 return mz_zip_set_error(pZip, MZ_ZIP_FILE_WRITE_FAILED);
7270#ifndef MINIZ_NO_STDIO
7271 if ((pState->m_pFile) && (MZ_FFLUSH(pState->m_pFile) == EOF))
7272 return mz_zip_set_error(pZip, MZ_ZIP_FILE_CLOSE_FAILED);
7275 pZip->m_archive_size += MZ_ZIP_END_OF_CENTRAL_DIR_HEADER_SIZE;
7277 pZip->m_zip_mode = MZ_ZIP_MODE_WRITING_HAS_BEEN_FINALIZED;
7281mz_bool mz_zip_writer_finalize_heap_archive(
mz_zip_archive *pZip,
void **ppBuf,
size_t *pSize)
7283 if ((!ppBuf) || (!pSize))
7284 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
7289 if ((!pZip) || (!pZip->m_pState))
7290 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
7292 if (pZip->m_pWrite != mz_zip_heap_write_func)
7293 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
7295 if (!mz_zip_writer_finalize_archive(pZip))
7298 *ppBuf = pZip->m_pState->m_pMem;
7299 *pSize = pZip->m_pState->m_mem_size;
7300 pZip->m_pState->m_pMem = NULL;
7301 pZip->m_pState->m_mem_size = pZip->m_pState->m_mem_capacity = 0;
7308 return mz_zip_writer_end_internal(pZip, MZ_TRUE);
7311#ifndef MINIZ_NO_STDIO
7312mz_bool mz_zip_add_mem_to_archive_file_in_place(
const char *pZip_filename,
const char *pArchive_name,
const void *pBuf,
size_t buf_size,
const void *pComment, mz_uint16 comment_size, mz_uint level_and_flags)
7314 return mz_zip_add_mem_to_archive_file_in_place_v2(pZip_filename, pArchive_name, pBuf, buf_size, pComment, comment_size, level_and_flags, NULL);
7317mz_bool mz_zip_add_mem_to_archive_file_in_place_v2(
const char *pZip_filename,
const char *pArchive_name,
const void *pBuf,
size_t buf_size,
const void *pComment, mz_uint16 comment_size, mz_uint level_and_flags, mz_zip_error *pErr)
7319 mz_bool status, created_new_archive = MZ_FALSE;
7321 struct MZ_FILE_STAT_STRUCT file_stat;
7322 mz_zip_error actual_err = MZ_ZIP_NO_ERROR;
7324 mz_zip_zero_struct(&zip_archive);
7325 if ((
int)level_and_flags < 0)
7326 level_and_flags = MZ_DEFAULT_LEVEL;
7328 if ((!pZip_filename) || (!pArchive_name) || ((buf_size) && (!pBuf)) || ((comment_size) && (!pComment)) || ((level_and_flags & 0xF) > MZ_UBER_COMPRESSION))
7331 *pErr = MZ_ZIP_INVALID_PARAMETER;
7335 if (!mz_zip_writer_validate_archive_name(pArchive_name))
7338 *pErr = MZ_ZIP_INVALID_FILENAME;
7344 if (MZ_FILE_STAT(pZip_filename, &file_stat) != 0)
7347 if (!mz_zip_writer_init_file_v2(&zip_archive, pZip_filename, 0, level_and_flags))
7350 *pErr = zip_archive.m_last_error;
7354 created_new_archive = MZ_TRUE;
7359 if (!mz_zip_reader_init_file_v2(&zip_archive, pZip_filename, level_and_flags | MZ_ZIP_FLAG_DO_NOT_SORT_CENTRAL_DIRECTORY, 0, 0))
7362 *pErr = zip_archive.m_last_error;
7366 if (!mz_zip_writer_init_from_reader_v2(&zip_archive, pZip_filename, level_and_flags))
7369 *pErr = zip_archive.m_last_error;
7371 mz_zip_reader_end_internal(&zip_archive, MZ_FALSE);
7377 status = mz_zip_writer_add_mem_ex(&zip_archive, pArchive_name, pBuf, buf_size, pComment, comment_size, level_and_flags, 0, 0);
7378 actual_err = zip_archive.m_last_error;
7381 if (!mz_zip_writer_finalize_archive(&zip_archive))
7384 actual_err = zip_archive.m_last_error;
7389 if (!mz_zip_writer_end_internal(&zip_archive, status))
7392 actual_err = zip_archive.m_last_error;
7397 if ((!status) && (created_new_archive))
7400 int ignoredStatus = MZ_DELETE_FILE(pZip_filename);
7401 (void)ignoredStatus;
7410void *mz_zip_extract_archive_file_to_heap_v2(
const char *pZip_filename,
const char *pArchive_name,
const char *pComment,
size_t *pSize, mz_uint flags, mz_zip_error *pErr)
7412 mz_uint32 file_index;
7419 if ((!pZip_filename) || (!pArchive_name))
7422 *pErr = MZ_ZIP_INVALID_PARAMETER;
7427 mz_zip_zero_struct(&zip_archive);
7428 if (!mz_zip_reader_init_file_v2(&zip_archive, pZip_filename, flags | MZ_ZIP_FLAG_DO_NOT_SORT_CENTRAL_DIRECTORY, 0, 0))
7431 *pErr = zip_archive.m_last_error;
7436 if (mz_zip_reader_locate_file_v2(&zip_archive, pArchive_name, pComment, flags, &file_index))
7438 p = mz_zip_reader_extract_to_heap(&zip_archive, file_index, pSize, flags);
7441 mz_zip_reader_end_internal(&zip_archive, p != NULL);
7444 *pErr = zip_archive.m_last_error;
7449void *mz_zip_extract_archive_file_to_heap(
const char *pZip_filename,
const char *pArchive_name,
size_t *pSize, mz_uint flags)
7451 return mz_zip_extract_archive_file_to_heap_v2(pZip_filename, pArchive_name, NULL, pSize, flags, NULL);
7462 return pZip ? pZip->m_zip_mode : MZ_ZIP_MODE_INVALID;
7467 return pZip ? pZip->m_zip_type : MZ_ZIP_TYPE_INVALID;
7470mz_zip_error mz_zip_set_last_error(
mz_zip_archive *pZip, mz_zip_error err_num)
7472 mz_zip_error prev_err;
7475 return MZ_ZIP_INVALID_PARAMETER;
7477 prev_err = pZip->m_last_error;
7479 pZip->m_last_error = err_num;
7486 return MZ_ZIP_INVALID_PARAMETER;
7488 return pZip->m_last_error;
7493 return mz_zip_set_last_error(pZip, MZ_ZIP_NO_ERROR);
7498 mz_zip_error prev_err;
7501 return MZ_ZIP_INVALID_PARAMETER;
7503 prev_err = pZip->m_last_error;
7505 pZip->m_last_error = MZ_ZIP_NO_ERROR;
7509const char *mz_zip_get_error_string(mz_zip_error mz_err)
7513 case MZ_ZIP_NO_ERROR:
7515 case MZ_ZIP_UNDEFINED_ERROR:
7516 return "undefined error";
7517 case MZ_ZIP_TOO_MANY_FILES:
7518 return "too many files";
7519 case MZ_ZIP_FILE_TOO_LARGE:
7520 return "file too large";
7521 case MZ_ZIP_UNSUPPORTED_METHOD:
7522 return "unsupported method";
7523 case MZ_ZIP_UNSUPPORTED_ENCRYPTION:
7524 return "unsupported encryption";
7525 case MZ_ZIP_UNSUPPORTED_FEATURE:
7526 return "unsupported feature";
7527 case MZ_ZIP_FAILED_FINDING_CENTRAL_DIR:
7528 return "failed finding central directory";
7529 case MZ_ZIP_NOT_AN_ARCHIVE:
7530 return "not a ZIP archive";
7531 case MZ_ZIP_INVALID_HEADER_OR_CORRUPTED:
7532 return "invalid header or archive is corrupted";
7533 case MZ_ZIP_UNSUPPORTED_MULTIDISK:
7534 return "unsupported multidisk archive";
7535 case MZ_ZIP_DECOMPRESSION_FAILED:
7536 return "decompression failed or archive is corrupted";
7537 case MZ_ZIP_COMPRESSION_FAILED:
7538 return "compression failed";
7539 case MZ_ZIP_UNEXPECTED_DECOMPRESSED_SIZE:
7540 return "unexpected decompressed size";
7541 case MZ_ZIP_CRC_CHECK_FAILED:
7542 return "CRC-32 check failed";
7543 case MZ_ZIP_UNSUPPORTED_CDIR_SIZE:
7544 return "unsupported central directory size";
7545 case MZ_ZIP_ALLOC_FAILED:
7546 return "allocation failed";
7547 case MZ_ZIP_FILE_OPEN_FAILED:
7548 return "file open failed";
7549 case MZ_ZIP_FILE_CREATE_FAILED:
7550 return "file create failed";
7551 case MZ_ZIP_FILE_WRITE_FAILED:
7552 return "file write failed";
7553 case MZ_ZIP_FILE_READ_FAILED:
7554 return "file read failed";
7555 case MZ_ZIP_FILE_CLOSE_FAILED:
7556 return "file close failed";
7557 case MZ_ZIP_FILE_SEEK_FAILED:
7558 return "file seek failed";
7559 case MZ_ZIP_FILE_STAT_FAILED:
7560 return "file stat failed";
7561 case MZ_ZIP_INVALID_PARAMETER:
7562 return "invalid parameter";
7563 case MZ_ZIP_INVALID_FILENAME:
7564 return "invalid filename";
7565 case MZ_ZIP_BUF_TOO_SMALL:
7566 return "buffer too small";
7567 case MZ_ZIP_INTERNAL_ERROR:
7568 return "internal error";
7569 case MZ_ZIP_FILE_NOT_FOUND:
7570 return "file not found";
7571 case MZ_ZIP_ARCHIVE_TOO_LARGE:
7572 return "archive is too large";
7573 case MZ_ZIP_VALIDATION_FAILED:
7574 return "validation failed";
7575 case MZ_ZIP_WRITE_CALLBACK_FAILED:
7576 return "write calledback failed";
7581 return "unknown error";
7587 if ((!pZip) || (!pZip->m_pState))
7590 return pZip->m_pState->m_zip64;
7595 if ((!pZip) || (!pZip->m_pState))
7598 return pZip->m_pState->m_central_dir.m_size;
7603 return pZip ? pZip->m_total_files : 0;
7610 return pZip->m_archive_size;
7613mz_uint64 mz_zip_get_archive_file_start_offset(
mz_zip_archive *pZip)
7615 if ((!pZip) || (!pZip->m_pState))
7617 return pZip->m_pState->m_file_archive_start_ofs;
7622 if ((!pZip) || (!pZip->m_pState))
7624 return pZip->m_pState->m_pFile;
7627size_t mz_zip_read_archive_data(
mz_zip_archive *pZip, mz_uint64 file_ofs,
void *pBuf,
size_t n)
7629 if ((!pZip) || (!pZip->m_pState) || (!pBuf) || (!pZip->m_pRead))
7630 return mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
7632 return pZip->m_pRead(pZip->m_pIO_opaque, file_ofs, pBuf, n);
7635mz_uint mz_zip_reader_get_filename(
mz_zip_archive *pZip, mz_uint file_index,
char *pFilename, mz_uint filename_buf_size)
7638 const mz_uint8 *p = mz_zip_get_cdh(pZip, file_index);
7641 if (filename_buf_size)
7642 pFilename[0] =
'\0';
7643 mz_zip_set_error(pZip, MZ_ZIP_INVALID_PARAMETER);
7646 n = MZ_READ_LE16(p + MZ_ZIP_CDH_FILENAME_LEN_OFS);
7647 if (filename_buf_size)
7649 n = MZ_MIN(n, filename_buf_size - 1);
7650 memcpy(pFilename, p + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE, n);
7651 pFilename[n] =
'\0';
7658 return mz_zip_file_stat_internal(pZip, file_index, mz_zip_get_cdh(pZip, file_index), pStat, NULL);
7666 if (pZip->m_zip_mode == MZ_ZIP_MODE_READING)
7667 return mz_zip_reader_end(pZip);
7668#ifndef MINIZ_NO_ARCHIVE_WRITING_APIS
7669 else if ((pZip->m_zip_mode == MZ_ZIP_MODE_WRITING) || (pZip->m_zip_mode == MZ_ZIP_MODE_WRITING_HAS_BEEN_FINALIZED))
7670 return mz_zip_writer_end(pZip);
QTextStream & flush(QTextStream &stream)