/src/gdal/port/cpl_minizip_zip.cpp
Line | Count | Source (jump to first uncovered line) |
1 | | /****************************************************************************** |
2 | | * |
3 | | * Project: CPL - Common Portability Library |
4 | | * Author: Frank Warmerdam, warmerdam@pobox.com |
5 | | * Purpose: Adjusted minizip "zip.c" source code for zip services. |
6 | | * |
7 | | * Modified version by Even Rouault. : |
8 | | * - Decoration of symbol names unz* -> cpl_unz* |
9 | | * - Undef EXPORT so that we are sure the symbols are not exported |
10 | | * - Remove old C style function prototypes |
11 | | * - Added CPL* simplified API at bottom. |
12 | | * |
13 | | * Original license available in port/LICENCE_minizip |
14 | | * |
15 | | *****************************************************************************/ |
16 | | |
17 | | /* zip.c -- IO on .zip files using zlib |
18 | | Version 1.1, February 14h, 2010 |
19 | | part of the MiniZip project - ( http://www.winimage.com/zLibDll/minizip.html |
20 | | ) |
21 | | |
22 | | Copyright (C) 1998-2010 Gilles Vollant (minizip) ( |
23 | | http://www.winimage.com/zLibDll/minizip.html ) |
24 | | |
25 | | Modifications for Zip64 support |
26 | | Copyright (C) 2009-2010 Mathias Svensson ( http://result42.com ) |
27 | | |
28 | | For more info read MiniZip_info.txt |
29 | | |
30 | | Changes |
31 | | Oct-2009 - Mathias Svensson - Remove old C style function prototypes |
32 | | Oct-2009 - Mathias Svensson - Added Zip64 Support when creating new file |
33 | | archives Oct-2009 - Mathias Svensson - Did some code cleanup and refactoring |
34 | | to get better overview of some functions. Oct-2009 - Mathias Svensson - Added |
35 | | zipRemoveExtraInfoBlock to strip extra field data from its ZIP64 data It is |
36 | | used when recreating zip archive with RAW when deleting items from a zip. |
37 | | ZIP64 data is automatically added to items that |
38 | | needs it, and existing ZIP64 data need to be removed. Oct-2009 - Mathias |
39 | | Svensson - Added support for BZIP2 as compression mode (bzip2 lib is |
40 | | required) Jan-2010 - back to unzip and minizip 1.0 name scheme, with |
41 | | compatibility layer |
42 | | |
43 | | Copyright (c) 2010-2018, Even Rouault <even dot rouault at spatialys.com> |
44 | | |
45 | | */ |
46 | | |
47 | | #include "cpl_port.h" |
48 | | #include "cpl_minizip_zip.h" |
49 | | |
50 | | #include <algorithm> |
51 | | #include <limits> |
52 | | |
53 | | #include <cassert> |
54 | | #include <cstddef> |
55 | | #include <cstdlib> |
56 | | #include <cstring> |
57 | | #if HAVE_FCNTL_H |
58 | | #include <fcntl.h> |
59 | | #endif |
60 | | #include <time.h> |
61 | | |
62 | | #include "cpl_conv.h" |
63 | | #include "cpl_error.h" |
64 | | #include "cpl_minizip_unzip.h" |
65 | | #include "cpl_string.h" |
66 | | #include "cpl_time.h" |
67 | | #include "cpl_vsi_virtual.h" |
68 | | |
69 | | #ifdef NO_ERRNO_H |
70 | | extern int errno; |
71 | | #else |
72 | | #include <errno.h> |
73 | | #endif |
74 | | |
75 | | #ifndef VERSIONMADEBY |
76 | 0 | #define VERSIONMADEBY (0x0) /* platform dependent */ |
77 | | #endif |
78 | | |
79 | | #ifndef Z_BUFSIZE |
80 | 0 | #define Z_BUFSIZE (16384) |
81 | | #endif |
82 | | |
83 | | #ifndef ALLOC |
84 | 0 | #define ALLOC(size) (malloc(size)) |
85 | | #endif |
86 | | #ifndef TRYFREE |
87 | | #define TRYFREE(p) \ |
88 | 0 | { \ |
89 | 0 | if (p) \ |
90 | 0 | free(p); \ |
91 | 0 | } |
92 | | #endif |
93 | | |
94 | | /* |
95 | | #define SIZECENTRALDIRITEM (0x2e) |
96 | | #define SIZEZIPLOCALHEADER (0x1e) |
97 | | */ |
98 | | |
99 | | /* I've found an old Unix (a SunOS 4.1.3_U1) without all SEEK_* defined... */ |
100 | | |
101 | | #ifndef SEEK_CUR |
102 | | #define SEEK_CUR 1 |
103 | | #endif |
104 | | |
105 | | #ifndef SEEK_END |
106 | | #define SEEK_END 2 |
107 | | #endif |
108 | | |
109 | | #ifndef SEEK_SET |
110 | | #define SEEK_SET 0 |
111 | | #endif |
112 | | |
113 | | #ifndef DEF_MEM_LEVEL |
114 | | #if MAX_MEM_LEVEL >= 8 |
115 | | #define DEF_MEM_LEVEL 8 |
116 | | #else |
117 | | #define DEF_MEM_LEVEL MAX_MEM_LEVEL |
118 | | #endif |
119 | | #endif |
120 | | |
121 | | CPL_UNUSED static const char zip_copyright[] = |
122 | | " zip 1.01 Copyright 1998-2004 Gilles Vollant - " |
123 | | "http://www.winimage.com/zLibDll"; |
124 | | |
125 | 0 | #define SIZEDATA_INDATABLOCK (4096 - (4 * 4)) |
126 | | |
127 | 0 | #define LOCALHEADERMAGIC (0x04034b50) |
128 | 0 | #define CENTRALHEADERMAGIC (0x02014b50) |
129 | 0 | #define ENDHEADERMAGIC (0x06054b50) |
130 | 0 | #define ZIP64ENDHEADERMAGIC (0x6064b50) |
131 | 0 | #define ZIP64ENDLOCHEADERMAGIC (0x7064b50) |
132 | | |
133 | | #define FLAG_LOCALHEADER_OFFSET (0x06) |
134 | | #define CRC_LOCALHEADER_OFFSET (0x0e) |
135 | | |
136 | 0 | #define SIZECENTRALHEADER (0x2e) /* 46 */ |
137 | | |
138 | | typedef struct linkedlist_datablock_internal_s |
139 | | { |
140 | | struct linkedlist_datablock_internal_s *next_datablock; |
141 | | uLong avail_in_this_block; |
142 | | uLong filled_in_this_block; |
143 | | uLong unused; // For future use and alignment. |
144 | | unsigned char data[SIZEDATA_INDATABLOCK]; |
145 | | } linkedlist_datablock_internal; |
146 | | |
147 | | typedef struct linkedlist_data_s |
148 | | { |
149 | | linkedlist_datablock_internal *first_block; |
150 | | linkedlist_datablock_internal *last_block; |
151 | | } linkedlist_data; |
152 | | |
153 | | typedef struct |
154 | | { |
155 | | z_stream stream; /* zLib stream structure for inflate */ |
156 | | int stream_initialised; /* 1 is stream is initialized */ |
157 | | uInt pos_in_buffered_data; /* last written byte in buffered_data */ |
158 | | |
159 | | ZPOS64_T pos_local_header; /* offset of the local header of the file |
160 | | currently writing */ |
161 | | char *local_header; |
162 | | uInt size_local_header; |
163 | | uInt size_local_header_extrafield; |
164 | | |
165 | | char *central_header; /* central header data for the current file */ |
166 | | uLong size_centralExtra; |
167 | | uLong size_centralheader; /* size of the central header for cur file */ |
168 | | uLong size_centralExtraFree; /* Extra bytes allocated to the centralheader |
169 | | but that are not used */ |
170 | | uLong flag; /* flag of the file currently writing */ |
171 | | |
172 | | // TODO: What is "wr"? "to"? |
173 | | int method; /* compression method of file currently wr.*/ |
174 | | int raw; /* 1 for directly writing raw data */ |
175 | | Byte buffered_data[Z_BUFSIZE]; /* buffer contain compressed data to be |
176 | | written. */ |
177 | | uLong dosDate; |
178 | | uLong crc32; |
179 | | int encrypt; |
180 | | ZPOS64_T pos_zip64extrainfo; |
181 | | ZPOS64_T totalCompressedData; |
182 | | ZPOS64_T totalUncompressedData; |
183 | | #ifndef NOCRYPT |
184 | | unsigned long keys[3]; /* keys defining the pseudo-random sequence */ |
185 | | const unsigned long *pcrc_32_tab; |
186 | | int crypt_header_size; |
187 | | #endif |
188 | | } curfile64_info; |
189 | | |
190 | | typedef struct |
191 | | { |
192 | | zlib_filefunc_def z_filefunc; |
193 | | voidpf filestream; /* IO structure of the zipfile */ |
194 | | linkedlist_data central_dir; /* datablock with central dir in construction*/ |
195 | | int in_opened_file_inzip; /* 1 if a file in the zip is currently writ.*/ |
196 | | curfile64_info ci; /* info on the file currently writing */ |
197 | | |
198 | | ZPOS64_T begin_pos; /* position of the beginning of the zipfile */ |
199 | | ZPOS64_T add_position_when_writing_offset; |
200 | | ZPOS64_T number_entry; |
201 | | #ifndef NO_ADDFILEINEXISTINGZIP |
202 | | char *globalcomment; |
203 | | #endif |
204 | | int use_cpl_io; |
205 | | vsi_l_offset vsi_raw_length_before; |
206 | | VSIVirtualHandle *vsi_deflate_handle; |
207 | | size_t nChunkSize; |
208 | | int nThreads; |
209 | | size_t nOffsetSize; |
210 | | std::vector<uint8_t> *sozip_index; |
211 | | } zip64_internal; |
212 | | |
213 | | #ifndef NOCRYPT |
214 | | #define INCLUDECRYPTINGCODE_IFCRYPTALLOWED |
215 | | #include "crypt.h" |
216 | | #endif |
217 | | |
218 | | static linkedlist_datablock_internal *allocate_new_datablock() |
219 | 0 | { |
220 | 0 | linkedlist_datablock_internal *ldi; |
221 | 0 | ldi = static_cast<linkedlist_datablock_internal *>( |
222 | 0 | ALLOC(sizeof(linkedlist_datablock_internal))); |
223 | 0 | if (ldi != nullptr) |
224 | 0 | { |
225 | 0 | ldi->next_datablock = nullptr; |
226 | 0 | ldi->filled_in_this_block = 0; |
227 | 0 | ldi->avail_in_this_block = SIZEDATA_INDATABLOCK; |
228 | 0 | } |
229 | 0 | return ldi; |
230 | 0 | } |
231 | | |
232 | | static void free_datablock(linkedlist_datablock_internal *ldi) |
233 | 0 | { |
234 | 0 | while (ldi != nullptr) |
235 | 0 | { |
236 | 0 | linkedlist_datablock_internal *ldinext = ldi->next_datablock; |
237 | 0 | TRYFREE(ldi); |
238 | 0 | ldi = ldinext; |
239 | 0 | } |
240 | 0 | } |
241 | | |
242 | | static void init_linkedlist(linkedlist_data *ll) |
243 | 0 | { |
244 | 0 | ll->first_block = ll->last_block = nullptr; |
245 | 0 | } |
246 | | |
247 | | static void free_linkedlist(linkedlist_data *ll) |
248 | 0 | { |
249 | 0 | free_datablock(ll->first_block); |
250 | 0 | ll->first_block = ll->last_block = nullptr; |
251 | 0 | } |
252 | | |
253 | | static int add_data_in_datablock(linkedlist_data *ll, const void *buf, |
254 | | uLong len) |
255 | 0 | { |
256 | 0 | linkedlist_datablock_internal *ldi; |
257 | 0 | const unsigned char *from_copy; |
258 | |
|
259 | 0 | if (ll == nullptr) |
260 | 0 | return ZIP_INTERNALERROR; |
261 | | |
262 | 0 | if (ll->last_block == nullptr) |
263 | 0 | { |
264 | 0 | ll->first_block = ll->last_block = allocate_new_datablock(); |
265 | 0 | if (ll->first_block == nullptr) |
266 | 0 | return ZIP_INTERNALERROR; |
267 | 0 | } |
268 | | |
269 | 0 | ldi = ll->last_block; |
270 | 0 | from_copy = reinterpret_cast<const unsigned char *>(buf); |
271 | |
|
272 | 0 | while (len > 0) |
273 | 0 | { |
274 | 0 | uInt copy_this; |
275 | 0 | uInt i; |
276 | 0 | unsigned char *to_copy; |
277 | |
|
278 | 0 | if (ldi->avail_in_this_block == 0) |
279 | 0 | { |
280 | 0 | ldi->next_datablock = allocate_new_datablock(); |
281 | 0 | if (ldi->next_datablock == nullptr) |
282 | 0 | return ZIP_INTERNALERROR; |
283 | 0 | ldi = ldi->next_datablock; |
284 | 0 | ll->last_block = ldi; |
285 | 0 | } |
286 | | |
287 | 0 | if (ldi->avail_in_this_block < len) |
288 | 0 | copy_this = static_cast<uInt>(ldi->avail_in_this_block); |
289 | 0 | else |
290 | 0 | copy_this = static_cast<uInt>(len); |
291 | |
|
292 | 0 | to_copy = &(ldi->data[ldi->filled_in_this_block]); |
293 | |
|
294 | 0 | for (i = 0; i < copy_this; i++) |
295 | 0 | *(to_copy + i) = *(from_copy + i); |
296 | |
|
297 | 0 | ldi->filled_in_this_block += copy_this; |
298 | 0 | ldi->avail_in_this_block -= copy_this; |
299 | 0 | from_copy += copy_this; |
300 | 0 | len -= copy_this; |
301 | 0 | } |
302 | 0 | return ZIP_OK; |
303 | 0 | } |
304 | | |
305 | | /****************************************************************************/ |
306 | | |
307 | | #ifndef NO_ADDFILEINEXISTINGZIP |
308 | | /* =========================================================================== |
309 | | Inputs a long in LSB order to the given file |
310 | | nbByte == 1, 2 ,4 or 8 (byte, short or long, ZPOS64_T) |
311 | | */ |
312 | | |
313 | | static int zip64local_putValue(const zlib_filefunc_def *pzlib_filefunc_def, |
314 | | voidpf filestream, ZPOS64_T x, int nbByte) |
315 | 0 | { |
316 | 0 | unsigned char buf[8]; |
317 | 0 | for (int n = 0; n < nbByte; n++) |
318 | 0 | { |
319 | 0 | buf[n] = static_cast<unsigned char>(x & 0xff); |
320 | 0 | x >>= 8; |
321 | 0 | } |
322 | 0 | if (x != 0) |
323 | 0 | { /* data overflow - hack for ZIP64 (X Roche) */ |
324 | 0 | for (int n = 0; n < nbByte; n++) |
325 | 0 | { |
326 | 0 | buf[n] = 0xff; |
327 | 0 | } |
328 | 0 | } |
329 | |
|
330 | 0 | if (ZWRITE64(*pzlib_filefunc_def, filestream, buf, nbByte) != |
331 | 0 | static_cast<uLong>(nbByte)) |
332 | 0 | return ZIP_ERRNO; |
333 | 0 | else |
334 | 0 | return ZIP_OK; |
335 | 0 | } |
336 | | |
337 | | static void zip64local_putValue_inmemory(void *dest, ZPOS64_T x, int nbByte) |
338 | 0 | { |
339 | 0 | unsigned char *buf = reinterpret_cast<unsigned char *>(dest); |
340 | 0 | for (int n = 0; n < nbByte; n++) |
341 | 0 | { |
342 | 0 | buf[n] = static_cast<unsigned char>(x & 0xff); |
343 | 0 | x >>= 8; |
344 | 0 | } |
345 | |
|
346 | 0 | if (x != 0) |
347 | 0 | { /* data overflow - hack for ZIP64 */ |
348 | 0 | for (int n = 0; n < nbByte; n++) |
349 | 0 | { |
350 | 0 | buf[n] = 0xff; |
351 | 0 | } |
352 | 0 | } |
353 | 0 | } |
354 | | |
355 | | /****************************************************************************/ |
356 | | |
357 | | static uLong zip64local_TmzDateToDosDate(const tm_zip *ptm) |
358 | 0 | { |
359 | 0 | uLong year = static_cast<uLong>(ptm->tm_year); |
360 | 0 | if (year > 1980) |
361 | 0 | year -= 1980; |
362 | 0 | else if (year > 80) |
363 | 0 | year -= 80; |
364 | 0 | return static_cast<uLong>( |
365 | 0 | ((ptm->tm_mday) + (32 * (ptm->tm_mon + 1)) + (512 * year)) |
366 | 0 | << 16) | |
367 | 0 | ((ptm->tm_sec / 2) + (32 * ptm->tm_min) + |
368 | 0 | (2048 * static_cast<uLong>(ptm->tm_hour))); |
369 | 0 | } |
370 | | |
371 | | /****************************************************************************/ |
372 | | |
373 | | static int zip64local_getByte(const zlib_filefunc_def *pzlib_filefunc_def, |
374 | | voidpf filestream, int *pi) |
375 | 0 | { |
376 | 0 | unsigned char c = 0; |
377 | 0 | const int err = |
378 | 0 | static_cast<int>(ZREAD64(*pzlib_filefunc_def, filestream, &c, 1)); |
379 | 0 | if (err == 1) |
380 | 0 | { |
381 | 0 | *pi = static_cast<int>(c); |
382 | 0 | return ZIP_OK; |
383 | 0 | } |
384 | 0 | else |
385 | 0 | { |
386 | 0 | if (ZERROR64(*pzlib_filefunc_def, filestream)) |
387 | 0 | return ZIP_ERRNO; |
388 | 0 | else |
389 | 0 | return ZIP_EOF; |
390 | 0 | } |
391 | 0 | } |
392 | | |
393 | | /* =========================================================================== |
394 | | Reads a long in LSB order from the given gz_stream. Sets |
395 | | */ |
396 | | static int zip64local_getShort(const zlib_filefunc_def *pzlib_filefunc_def, |
397 | | voidpf filestream, uLong *pX) |
398 | 0 | { |
399 | 0 | int i = 0; |
400 | 0 | int err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
401 | 0 | uLong x = static_cast<uLong>(i); |
402 | |
|
403 | 0 | if (err == ZIP_OK) |
404 | 0 | err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
405 | 0 | x += static_cast<uLong>(i) << 8; |
406 | |
|
407 | 0 | if (err == ZIP_OK) |
408 | 0 | *pX = x; |
409 | 0 | else |
410 | 0 | *pX = 0; |
411 | 0 | return err; |
412 | 0 | } |
413 | | |
414 | | static int zip64local_getLong(const zlib_filefunc_def *pzlib_filefunc_def, |
415 | | voidpf filestream, uLong *pX) |
416 | 0 | { |
417 | 0 | int i = 0; |
418 | 0 | int err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
419 | 0 | uLong x = static_cast<uLong>(i); |
420 | |
|
421 | 0 | if (err == ZIP_OK) |
422 | 0 | err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
423 | 0 | x += static_cast<uLong>(i) << 8; |
424 | |
|
425 | 0 | if (err == ZIP_OK) |
426 | 0 | err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
427 | 0 | x += static_cast<uLong>(i) << 16; |
428 | |
|
429 | 0 | if (err == ZIP_OK) |
430 | 0 | err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
431 | 0 | x += static_cast<uLong>(i) << 24; |
432 | |
|
433 | 0 | if (err == ZIP_OK) |
434 | 0 | *pX = x; |
435 | 0 | else |
436 | 0 | *pX = 0; |
437 | 0 | return err; |
438 | 0 | } |
439 | | |
440 | | static int zip64local_getLong64(const zlib_filefunc_def *pzlib_filefunc_def, |
441 | | voidpf filestream, ZPOS64_T *pX) |
442 | 0 | { |
443 | 0 | ZPOS64_T x; |
444 | 0 | int i = 0; |
445 | 0 | int err; |
446 | |
|
447 | 0 | err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
448 | 0 | x = static_cast<ZPOS64_T>(i); |
449 | |
|
450 | 0 | if (err == ZIP_OK) |
451 | 0 | err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
452 | 0 | x += static_cast<ZPOS64_T>(i) << 8; |
453 | |
|
454 | 0 | if (err == ZIP_OK) |
455 | 0 | err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
456 | 0 | x += static_cast<ZPOS64_T>(i) << 16; |
457 | |
|
458 | 0 | if (err == ZIP_OK) |
459 | 0 | err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
460 | 0 | x += static_cast<ZPOS64_T>(i) << 24; |
461 | |
|
462 | 0 | if (err == ZIP_OK) |
463 | 0 | err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
464 | 0 | x += static_cast<ZPOS64_T>(i) << 32; |
465 | |
|
466 | 0 | if (err == ZIP_OK) |
467 | 0 | err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
468 | 0 | x += static_cast<ZPOS64_T>(i) << 40; |
469 | |
|
470 | 0 | if (err == ZIP_OK) |
471 | 0 | err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
472 | 0 | x += static_cast<ZPOS64_T>(i) << 48; |
473 | |
|
474 | 0 | if (err == ZIP_OK) |
475 | 0 | err = zip64local_getByte(pzlib_filefunc_def, filestream, &i); |
476 | 0 | x += static_cast<ZPOS64_T>(i) << 56; |
477 | |
|
478 | 0 | if (err == ZIP_OK) |
479 | 0 | *pX = x; |
480 | 0 | else |
481 | 0 | *pX = 0; |
482 | |
|
483 | 0 | return err; |
484 | 0 | } |
485 | | |
486 | | #ifndef BUFREADCOMMENT |
487 | 0 | #define BUFREADCOMMENT (0x400) |
488 | | #endif |
489 | | /* |
490 | | Locate the Central directory of a zipfile (at the end, just before |
491 | | the global comment) |
492 | | */ |
493 | | static ZPOS64_T |
494 | | zip64local_SearchCentralDir(const zlib_filefunc_def *pzlib_filefunc_def, |
495 | | voidpf filestream) |
496 | 0 | { |
497 | 0 | ZPOS64_T uMaxBack = 0xffff; /* maximum size of global comment */ |
498 | 0 | ZPOS64_T uPosFound = 0; |
499 | |
|
500 | 0 | if (ZSEEK64(*pzlib_filefunc_def, filestream, 0, ZLIB_FILEFUNC_SEEK_END) != |
501 | 0 | 0) |
502 | 0 | return 0; |
503 | | |
504 | 0 | ZPOS64_T uSizeFile = ZTELL64(*pzlib_filefunc_def, filestream); |
505 | |
|
506 | 0 | if (uMaxBack > uSizeFile) |
507 | 0 | uMaxBack = uSizeFile; |
508 | |
|
509 | 0 | unsigned char *buf = |
510 | 0 | static_cast<unsigned char *>(ALLOC(BUFREADCOMMENT + 4)); |
511 | 0 | if (buf == nullptr) |
512 | 0 | return 0; |
513 | | |
514 | 0 | ZPOS64_T uBackRead = 4; |
515 | 0 | while (uBackRead < uMaxBack) |
516 | 0 | { |
517 | 0 | if (uBackRead + BUFREADCOMMENT > uMaxBack) |
518 | 0 | uBackRead = uMaxBack; |
519 | 0 | else |
520 | 0 | uBackRead += BUFREADCOMMENT; |
521 | 0 | ZPOS64_T uReadPos = uSizeFile - uBackRead; |
522 | |
|
523 | 0 | uLong uReadSize = ((BUFREADCOMMENT + 4) < (uSizeFile - uReadPos)) |
524 | 0 | ? (BUFREADCOMMENT + 4) |
525 | 0 | : static_cast<uLong>(uSizeFile - uReadPos); |
526 | 0 | if (ZSEEK64(*pzlib_filefunc_def, filestream, uReadPos, |
527 | 0 | ZLIB_FILEFUNC_SEEK_SET) != 0) |
528 | 0 | break; |
529 | | |
530 | 0 | if (ZREAD64(*pzlib_filefunc_def, filestream, buf, uReadSize) != |
531 | 0 | uReadSize) |
532 | 0 | break; |
533 | | |
534 | 0 | for (int i = static_cast<int>(uReadSize) - 3; (i--) > 0;) |
535 | 0 | if (((*(buf + i)) == 0x50) && ((*(buf + i + 1)) == 0x4b) && |
536 | 0 | ((*(buf + i + 2)) == 0x05) && ((*(buf + i + 3)) == 0x06)) |
537 | 0 | { |
538 | 0 | uPosFound = uReadPos + i; |
539 | 0 | break; |
540 | 0 | } |
541 | |
|
542 | 0 | if (uPosFound != 0) |
543 | 0 | break; |
544 | 0 | } |
545 | 0 | TRYFREE(buf); |
546 | 0 | return uPosFound; |
547 | 0 | } |
548 | | |
549 | | /* |
550 | | Locate the End of Zip64 Central directory locator and from there find the CD of |
551 | | a zipfile (at the end, just before the global comment) |
552 | | */ |
553 | | static ZPOS64_T |
554 | | zip64local_SearchCentralDir64(const zlib_filefunc_def *pzlib_filefunc_def, |
555 | | voidpf filestream) |
556 | 0 | { |
557 | 0 | unsigned char *buf; |
558 | 0 | ZPOS64_T uSizeFile; |
559 | 0 | ZPOS64_T uBackRead; |
560 | 0 | ZPOS64_T uMaxBack = 0xffff; /* maximum size of global comment */ |
561 | 0 | ZPOS64_T uPosFound = 0; |
562 | 0 | uLong uL; |
563 | 0 | ZPOS64_T relativeOffset; |
564 | |
|
565 | 0 | if (ZSEEK64(*pzlib_filefunc_def, filestream, 0, ZLIB_FILEFUNC_SEEK_END) != |
566 | 0 | 0) |
567 | 0 | return 0; |
568 | | |
569 | 0 | uSizeFile = ZTELL64(*pzlib_filefunc_def, filestream); |
570 | |
|
571 | 0 | if (uMaxBack > uSizeFile) |
572 | 0 | uMaxBack = uSizeFile; |
573 | |
|
574 | 0 | buf = static_cast<unsigned char *>(ALLOC(BUFREADCOMMENT + 4)); |
575 | 0 | if (buf == nullptr) |
576 | 0 | return 0; |
577 | | |
578 | 0 | uBackRead = 4; |
579 | 0 | while (uBackRead < uMaxBack) |
580 | 0 | { |
581 | 0 | uLong uReadSize; |
582 | 0 | ZPOS64_T uReadPos; |
583 | 0 | int i; |
584 | 0 | if (uBackRead + BUFREADCOMMENT > uMaxBack) |
585 | 0 | uBackRead = uMaxBack; |
586 | 0 | else |
587 | 0 | uBackRead += BUFREADCOMMENT; |
588 | 0 | uReadPos = uSizeFile - uBackRead; |
589 | |
|
590 | 0 | uReadSize = ((BUFREADCOMMENT + 4) < (uSizeFile - uReadPos)) |
591 | 0 | ? (BUFREADCOMMENT + 4) |
592 | 0 | : static_cast<uLong>(uSizeFile - uReadPos); |
593 | 0 | if (ZSEEK64(*pzlib_filefunc_def, filestream, uReadPos, |
594 | 0 | ZLIB_FILEFUNC_SEEK_SET) != 0) |
595 | 0 | break; |
596 | | |
597 | 0 | if (ZREAD64(*pzlib_filefunc_def, filestream, buf, uReadSize) != |
598 | 0 | uReadSize) |
599 | 0 | break; |
600 | | |
601 | 0 | for (i = static_cast<int>(uReadSize) - 3; (i--) > 0;) |
602 | 0 | { |
603 | | // Signature "0x07064b50" Zip64 end of central directory locater |
604 | 0 | if (((*(buf + i)) == 0x50) && ((*(buf + i + 1)) == 0x4b) && |
605 | 0 | ((*(buf + i + 2)) == 0x06) && ((*(buf + i + 3)) == 0x07)) |
606 | 0 | { |
607 | 0 | uPosFound = uReadPos + i; |
608 | 0 | break; |
609 | 0 | } |
610 | 0 | } |
611 | |
|
612 | 0 | if (uPosFound != 0) |
613 | 0 | break; |
614 | 0 | } |
615 | |
|
616 | 0 | TRYFREE(buf); |
617 | 0 | if (uPosFound == 0) |
618 | 0 | return 0; |
619 | | |
620 | | /* Zip64 end of central directory locator */ |
621 | 0 | if (ZSEEK64(*pzlib_filefunc_def, filestream, uPosFound, |
622 | 0 | ZLIB_FILEFUNC_SEEK_SET) != 0) |
623 | 0 | return 0; |
624 | | |
625 | | /* the signature, already checked */ |
626 | 0 | if (zip64local_getLong(pzlib_filefunc_def, filestream, &uL) != ZIP_OK) |
627 | 0 | return 0; |
628 | | |
629 | | /* number of the disk with the start of the zip64 end of central directory |
630 | | */ |
631 | 0 | if (zip64local_getLong(pzlib_filefunc_def, filestream, &uL) != ZIP_OK) |
632 | 0 | return 0; |
633 | 0 | if (uL != 0) |
634 | 0 | return 0; |
635 | | |
636 | | /* relative offset of the zip64 end of central directory record */ |
637 | 0 | if (zip64local_getLong64(pzlib_filefunc_def, filestream, &relativeOffset) != |
638 | 0 | ZIP_OK) |
639 | 0 | return 0; |
640 | | |
641 | | /* total number of disks */ |
642 | 0 | if (zip64local_getLong(pzlib_filefunc_def, filestream, &uL) != ZIP_OK) |
643 | 0 | return 0; |
644 | | /* Some .zip declare 0 disks, such as in |
645 | | * http://trac.osgeo.org/gdal/ticket/5615 */ |
646 | 0 | if (uL != 0 && uL != 1) |
647 | 0 | return 0; |
648 | | |
649 | | /* Goto Zip64 end of central directory record */ |
650 | 0 | if (ZSEEK64(*pzlib_filefunc_def, filestream, relativeOffset, |
651 | 0 | ZLIB_FILEFUNC_SEEK_SET) != 0) |
652 | 0 | return 0; |
653 | | |
654 | | /* the signature */ |
655 | 0 | if (zip64local_getLong(pzlib_filefunc_def, filestream, &uL) != ZIP_OK) |
656 | 0 | return 0; |
657 | | |
658 | 0 | if (uL != 0x06064b50) // signature of 'Zip64 end of central directory' |
659 | 0 | return 0; |
660 | | |
661 | 0 | return relativeOffset; |
662 | 0 | } |
663 | | |
664 | | static int LoadCentralDirectoryRecord(zip64_internal *pziinit) |
665 | 0 | { |
666 | 0 | int err = ZIP_OK; |
667 | 0 | ZPOS64_T byte_before_the_zipfile; /* byte before the zipfile, (>0 for sfx)*/ |
668 | |
|
669 | 0 | ZPOS64_T size_central_dir; /* size of the central directory */ |
670 | 0 | ZPOS64_T offset_central_dir; /* offset of start of central directory */ |
671 | 0 | ZPOS64_T central_pos; |
672 | 0 | uLong uL; |
673 | |
|
674 | 0 | uLong number_disk; /* number of the current dist, used for |
675 | | spanning ZIP, unsupported, always 0*/ |
676 | 0 | uLong number_disk_with_CD; /* number the disk with central dir, used |
677 | | for spanning ZIP, unsupported, always 0*/ |
678 | 0 | ZPOS64_T number_entry; |
679 | 0 | ZPOS64_T number_entry_CD; /* total number of entries in |
680 | | the central dir |
681 | | (same than number_entry on nospan) */ |
682 | 0 | uLong VersionMadeBy; |
683 | 0 | uLong VersionNeeded; |
684 | 0 | uLong size_comment; |
685 | |
|
686 | 0 | int hasZIP64Record = 0; |
687 | | |
688 | | // check first if we find a ZIP64 record |
689 | 0 | central_pos = zip64local_SearchCentralDir64(&pziinit->z_filefunc, |
690 | 0 | pziinit->filestream); |
691 | 0 | if (central_pos > 0) |
692 | 0 | { |
693 | 0 | hasZIP64Record = 1; |
694 | 0 | } |
695 | 0 | else /* if (central_pos == 0) */ |
696 | 0 | { |
697 | 0 | central_pos = zip64local_SearchCentralDir(&pziinit->z_filefunc, |
698 | 0 | pziinit->filestream); |
699 | 0 | } |
700 | | |
701 | | /* disable to allow appending to empty ZIP archive |
702 | | if (central_pos==0) |
703 | | err=ZIP_ERRNO; |
704 | | */ |
705 | |
|
706 | 0 | if (hasZIP64Record) |
707 | 0 | { |
708 | 0 | ZPOS64_T sizeEndOfCentralDirectory; |
709 | 0 | if (ZSEEK64(pziinit->z_filefunc, pziinit->filestream, central_pos, |
710 | 0 | ZLIB_FILEFUNC_SEEK_SET) != 0) |
711 | 0 | err = ZIP_ERRNO; |
712 | | |
713 | | /* the signature, already checked */ |
714 | 0 | if (zip64local_getLong(&pziinit->z_filefunc, pziinit->filestream, |
715 | 0 | &uL) != ZIP_OK) |
716 | 0 | err = ZIP_ERRNO; |
717 | | |
718 | | /* size of zip64 end of central directory record */ |
719 | 0 | if (zip64local_getLong64(&pziinit->z_filefunc, pziinit->filestream, |
720 | 0 | &sizeEndOfCentralDirectory) != ZIP_OK) |
721 | 0 | err = ZIP_ERRNO; |
722 | | |
723 | | /* version made by */ |
724 | 0 | if (zip64local_getShort(&pziinit->z_filefunc, pziinit->filestream, |
725 | 0 | &VersionMadeBy) != ZIP_OK) |
726 | 0 | err = ZIP_ERRNO; |
727 | | |
728 | | /* version needed to extract */ |
729 | 0 | if (zip64local_getShort(&pziinit->z_filefunc, pziinit->filestream, |
730 | 0 | &VersionNeeded) != ZIP_OK) |
731 | 0 | err = ZIP_ERRNO; |
732 | | |
733 | | /* number of this disk */ |
734 | 0 | if (zip64local_getLong(&pziinit->z_filefunc, pziinit->filestream, |
735 | 0 | &number_disk) != ZIP_OK) |
736 | 0 | err = ZIP_ERRNO; |
737 | | |
738 | | /* number of the disk with the start of the central directory */ |
739 | 0 | if (zip64local_getLong(&pziinit->z_filefunc, pziinit->filestream, |
740 | 0 | &number_disk_with_CD) != ZIP_OK) |
741 | 0 | err = ZIP_ERRNO; |
742 | | |
743 | | /* total number of entries in the central directory on this disk */ |
744 | 0 | if (zip64local_getLong64(&pziinit->z_filefunc, pziinit->filestream, |
745 | 0 | &number_entry) != ZIP_OK) |
746 | 0 | err = ZIP_ERRNO; |
747 | | |
748 | | /* total number of entries in the central directory */ |
749 | 0 | if (zip64local_getLong64(&pziinit->z_filefunc, pziinit->filestream, |
750 | 0 | &number_entry_CD) != ZIP_OK) |
751 | 0 | err = ZIP_ERRNO; |
752 | |
|
753 | 0 | if ((number_entry_CD != number_entry) || (number_disk_with_CD != 0) || |
754 | 0 | (number_disk != 0)) |
755 | 0 | err = ZIP_BADZIPFILE; |
756 | | |
757 | | /* size of the central directory */ |
758 | 0 | if (zip64local_getLong64(&pziinit->z_filefunc, pziinit->filestream, |
759 | 0 | &size_central_dir) != ZIP_OK) |
760 | 0 | err = ZIP_ERRNO; |
761 | | |
762 | | /* offset of start of central directory with respect to the |
763 | | starting disk number */ |
764 | 0 | if (zip64local_getLong64(&pziinit->z_filefunc, pziinit->filestream, |
765 | 0 | &offset_central_dir) != ZIP_OK) |
766 | 0 | err = ZIP_ERRNO; |
767 | | |
768 | | // TODO.. |
769 | | // read the comment from the standard central header. |
770 | 0 | size_comment = 0; |
771 | 0 | } |
772 | 0 | else |
773 | 0 | { |
774 | | // Read End of central Directory info |
775 | 0 | if (ZSEEK64(pziinit->z_filefunc, pziinit->filestream, central_pos, |
776 | 0 | ZLIB_FILEFUNC_SEEK_SET) != 0) |
777 | 0 | err = ZIP_ERRNO; |
778 | | |
779 | | /* the signature, already checked */ |
780 | 0 | if (zip64local_getLong(&pziinit->z_filefunc, pziinit->filestream, |
781 | 0 | &uL) != ZIP_OK) |
782 | 0 | err = ZIP_ERRNO; |
783 | | |
784 | | /* number of this disk */ |
785 | 0 | if (zip64local_getShort(&pziinit->z_filefunc, pziinit->filestream, |
786 | 0 | &number_disk) != ZIP_OK) |
787 | 0 | err = ZIP_ERRNO; |
788 | | |
789 | | /* number of the disk with the start of the central directory */ |
790 | 0 | if (zip64local_getShort(&pziinit->z_filefunc, pziinit->filestream, |
791 | 0 | &number_disk_with_CD) != ZIP_OK) |
792 | 0 | err = ZIP_ERRNO; |
793 | | |
794 | | /* total number of entries in the central dir on this disk */ |
795 | 0 | number_entry = 0; |
796 | 0 | if (zip64local_getShort(&pziinit->z_filefunc, pziinit->filestream, |
797 | 0 | &uL) != ZIP_OK) |
798 | 0 | err = ZIP_ERRNO; |
799 | 0 | else |
800 | 0 | number_entry = uL; |
801 | | |
802 | | /* total number of entries in the central dir */ |
803 | 0 | number_entry_CD = 0; |
804 | 0 | if (zip64local_getShort(&pziinit->z_filefunc, pziinit->filestream, |
805 | 0 | &uL) != ZIP_OK) |
806 | 0 | err = ZIP_ERRNO; |
807 | 0 | else |
808 | 0 | number_entry_CD = uL; |
809 | |
|
810 | 0 | if ((number_entry_CD != number_entry) || (number_disk_with_CD != 0) || |
811 | 0 | (number_disk != 0)) |
812 | 0 | err = ZIP_BADZIPFILE; |
813 | | |
814 | | /* size of the central directory */ |
815 | 0 | size_central_dir = 0; |
816 | 0 | if (zip64local_getLong(&pziinit->z_filefunc, pziinit->filestream, |
817 | 0 | &uL) != ZIP_OK) |
818 | 0 | err = ZIP_ERRNO; |
819 | 0 | else |
820 | 0 | size_central_dir = uL; |
821 | | |
822 | | /* offset of start of central directory with respect to the starting |
823 | | * disk number */ |
824 | 0 | offset_central_dir = 0; |
825 | 0 | if (zip64local_getLong(&pziinit->z_filefunc, pziinit->filestream, |
826 | 0 | &uL) != ZIP_OK) |
827 | 0 | err = ZIP_ERRNO; |
828 | 0 | else |
829 | 0 | offset_central_dir = uL; |
830 | | |
831 | | /* zipfile global comment length */ |
832 | 0 | if (zip64local_getShort(&pziinit->z_filefunc, pziinit->filestream, |
833 | 0 | &size_comment) != ZIP_OK) |
834 | 0 | err = ZIP_ERRNO; |
835 | 0 | } |
836 | |
|
837 | 0 | if ((central_pos < offset_central_dir + size_central_dir) && |
838 | 0 | (err == ZIP_OK)) |
839 | 0 | err = ZIP_BADZIPFILE; |
840 | |
|
841 | 0 | if (err != ZIP_OK) |
842 | 0 | { |
843 | 0 | ZCLOSE64(pziinit->z_filefunc, pziinit->filestream); |
844 | 0 | return ZIP_ERRNO; |
845 | 0 | } |
846 | | |
847 | 0 | if (size_comment > 0) |
848 | 0 | { |
849 | 0 | pziinit->globalcomment = static_cast<char *>(ALLOC(size_comment + 1)); |
850 | 0 | if (pziinit->globalcomment) |
851 | 0 | { |
852 | 0 | size_comment = ZREAD64(pziinit->z_filefunc, pziinit->filestream, |
853 | 0 | pziinit->globalcomment, size_comment); |
854 | 0 | pziinit->globalcomment[size_comment] = 0; |
855 | 0 | } |
856 | 0 | } |
857 | |
|
858 | 0 | byte_before_the_zipfile = |
859 | 0 | central_pos - (offset_central_dir + size_central_dir); |
860 | 0 | pziinit->add_position_when_writing_offset = byte_before_the_zipfile; |
861 | |
|
862 | 0 | { |
863 | 0 | ZPOS64_T size_central_dir_to_read = size_central_dir; |
864 | 0 | size_t buf_size = SIZEDATA_INDATABLOCK; |
865 | 0 | void *buf_read = ALLOC(buf_size); |
866 | 0 | if (ZSEEK64(pziinit->z_filefunc, pziinit->filestream, |
867 | 0 | offset_central_dir + byte_before_the_zipfile, |
868 | 0 | ZLIB_FILEFUNC_SEEK_SET) != 0) |
869 | 0 | err = ZIP_ERRNO; |
870 | |
|
871 | 0 | while ((size_central_dir_to_read > 0) && (err == ZIP_OK)) |
872 | 0 | { |
873 | 0 | ZPOS64_T read_this = SIZEDATA_INDATABLOCK; |
874 | 0 | if (read_this > size_central_dir_to_read) |
875 | 0 | read_this = size_central_dir_to_read; |
876 | |
|
877 | 0 | if (ZREAD64(pziinit->z_filefunc, pziinit->filestream, buf_read, |
878 | 0 | static_cast<uLong>(read_this)) != read_this) |
879 | 0 | err = ZIP_ERRNO; |
880 | |
|
881 | 0 | if (err == ZIP_OK) |
882 | 0 | err = add_data_in_datablock(&pziinit->central_dir, buf_read, |
883 | 0 | static_cast<uLong>(read_this)); |
884 | |
|
885 | 0 | size_central_dir_to_read -= read_this; |
886 | 0 | } |
887 | 0 | TRYFREE(buf_read); |
888 | 0 | } |
889 | 0 | pziinit->begin_pos = byte_before_the_zipfile; |
890 | 0 | pziinit->number_entry = number_entry_CD; |
891 | |
|
892 | 0 | if (ZSEEK64(pziinit->z_filefunc, pziinit->filestream, |
893 | 0 | offset_central_dir + byte_before_the_zipfile, |
894 | 0 | ZLIB_FILEFUNC_SEEK_SET) != 0) |
895 | 0 | err = ZIP_ERRNO; |
896 | |
|
897 | 0 | return err; |
898 | 0 | } |
899 | | |
900 | | #endif /* !NO_ADDFILEINEXISTINGZIP*/ |
901 | | |
902 | | /************************************************************/ |
903 | | extern zipFile ZEXPORT cpl_zipOpen2(const char *pathname, int append, |
904 | | zipcharpc *globalcomment, |
905 | | zlib_filefunc_def *pzlib_filefunc_def) |
906 | 0 | { |
907 | 0 | zip64_internal ziinit; |
908 | 0 | memset(&ziinit, 0, sizeof(ziinit)); |
909 | |
|
910 | 0 | if (pzlib_filefunc_def == nullptr) |
911 | 0 | cpl_fill_fopen_filefunc(&ziinit.z_filefunc); |
912 | 0 | else |
913 | 0 | ziinit.z_filefunc = *pzlib_filefunc_def; |
914 | |
|
915 | 0 | ziinit.filestream = (*(ziinit.z_filefunc.zopen_file))( |
916 | 0 | ziinit.z_filefunc.opaque, pathname, |
917 | 0 | (append == APPEND_STATUS_CREATE) |
918 | 0 | ? (ZLIB_FILEFUNC_MODE_READ | ZLIB_FILEFUNC_MODE_WRITE | |
919 | 0 | ZLIB_FILEFUNC_MODE_CREATE) |
920 | 0 | : (ZLIB_FILEFUNC_MODE_READ | ZLIB_FILEFUNC_MODE_WRITE | |
921 | 0 | ZLIB_FILEFUNC_MODE_EXISTING)); |
922 | |
|
923 | 0 | if (ziinit.filestream == nullptr) |
924 | 0 | return nullptr; |
925 | | |
926 | 0 | if (append == APPEND_STATUS_CREATEAFTER) |
927 | 0 | ZSEEK64(ziinit.z_filefunc, ziinit.filestream, 0, SEEK_END); |
928 | |
|
929 | 0 | ziinit.begin_pos = ZTELL64(ziinit.z_filefunc, ziinit.filestream); |
930 | 0 | ziinit.in_opened_file_inzip = 0; |
931 | 0 | ziinit.ci.stream_initialised = 0; |
932 | 0 | ziinit.number_entry = 0; |
933 | 0 | ziinit.add_position_when_writing_offset = 0; |
934 | 0 | ziinit.use_cpl_io = (pzlib_filefunc_def == nullptr) ? 1 : 0; |
935 | 0 | ziinit.vsi_raw_length_before = 0; |
936 | 0 | ziinit.vsi_deflate_handle = nullptr; |
937 | 0 | ziinit.nChunkSize = 0; |
938 | 0 | ziinit.nThreads = 0; |
939 | 0 | ziinit.nOffsetSize = 0; |
940 | 0 | ziinit.sozip_index = nullptr; |
941 | 0 | init_linkedlist(&(ziinit.central_dir)); |
942 | |
|
943 | 0 | zip64_internal *zi = |
944 | 0 | static_cast<zip64_internal *>(ALLOC(sizeof(zip64_internal))); |
945 | 0 | if (zi == nullptr) |
946 | 0 | { |
947 | 0 | ZCLOSE64(ziinit.z_filefunc, ziinit.filestream); |
948 | 0 | return nullptr; |
949 | 0 | } |
950 | | |
951 | | /* now we add file in a zipfile */ |
952 | 0 | #ifndef NO_ADDFILEINEXISTINGZIP |
953 | 0 | ziinit.globalcomment = nullptr; |
954 | |
|
955 | 0 | int err = ZIP_OK; |
956 | 0 | if (append == APPEND_STATUS_ADDINZIP) |
957 | 0 | { |
958 | | // Read and Cache Central Directory Records |
959 | 0 | err = LoadCentralDirectoryRecord(&ziinit); |
960 | 0 | } |
961 | |
|
962 | 0 | if (globalcomment) |
963 | 0 | { |
964 | 0 | *globalcomment = ziinit.globalcomment; |
965 | 0 | } |
966 | 0 | #endif /* !NO_ADDFILEINEXISTINGZIP*/ |
967 | |
|
968 | 0 | if (err != ZIP_OK) |
969 | 0 | { |
970 | 0 | #ifndef NO_ADDFILEINEXISTINGZIP |
971 | 0 | TRYFREE(ziinit.globalcomment); |
972 | 0 | #endif /* !NO_ADDFILEINEXISTINGZIP*/ |
973 | 0 | TRYFREE(zi); |
974 | 0 | return nullptr; |
975 | 0 | } |
976 | 0 | else |
977 | 0 | { |
978 | 0 | *zi = ziinit; |
979 | 0 | return static_cast<zipFile>(zi); |
980 | 0 | } |
981 | 0 | } |
982 | | |
983 | | extern zipFile ZEXPORT cpl_zipOpen(const char *pathname, int append) |
984 | 0 | { |
985 | 0 | return cpl_zipOpen2(pathname, append, nullptr, nullptr); |
986 | 0 | } |
987 | | |
988 | | static void zip64local_putValue_inmemory_update(char **dest, ZPOS64_T x, |
989 | | int nbByte) |
990 | 0 | { |
991 | 0 | zip64local_putValue_inmemory(*dest, x, nbByte); |
992 | 0 | *dest += nbByte; |
993 | 0 | } |
994 | | |
995 | | static int Write_LocalFileHeader(zip64_internal *zi, const char *filename, |
996 | | uInt size_extrafield_local, |
997 | | const void *extrafield_local, int zip64) |
998 | 0 | { |
999 | | /* write the local header */ |
1000 | 0 | int err = ZIP_OK; |
1001 | 0 | uInt size_filename = static_cast<uInt>(strlen(filename)); |
1002 | 0 | uInt size_extrafield = size_extrafield_local; |
1003 | |
|
1004 | 0 | if (zip64) |
1005 | 0 | { |
1006 | 0 | size_extrafield += 20; |
1007 | 0 | } |
1008 | |
|
1009 | 0 | uInt size_local_header = 30 + size_filename + size_extrafield; |
1010 | 0 | char *local_header = static_cast<char *>(ALLOC(size_local_header)); |
1011 | 0 | char *p = local_header; |
1012 | |
|
1013 | 0 | zip64local_putValue_inmemory_update(&p, LOCALHEADERMAGIC, 4); |
1014 | 0 | if (zip64) |
1015 | 0 | zip64local_putValue_inmemory_update(&p, 45, |
1016 | 0 | 2); /* version needed to extract */ |
1017 | 0 | else |
1018 | 0 | zip64local_putValue_inmemory_update(&p, 20, |
1019 | 0 | 2); /* version needed to extract */ |
1020 | |
|
1021 | 0 | zip64local_putValue_inmemory_update(&p, zi->ci.flag, 2); |
1022 | |
|
1023 | 0 | zip64local_putValue_inmemory_update(&p, zi->ci.method, 2); |
1024 | |
|
1025 | 0 | zip64local_putValue_inmemory_update(&p, zi->ci.dosDate, 4); |
1026 | | |
1027 | | // CRC / Compressed size / Uncompressed size will be filled in later and |
1028 | | // rewritten later |
1029 | 0 | zip64local_putValue_inmemory_update(&p, 0, 4); /* crc 32, unknown */ |
1030 | |
|
1031 | 0 | if (zip64) |
1032 | 0 | zip64local_putValue_inmemory_update(&p, 0xFFFFFFFFU, |
1033 | 0 | 4); /* compressed size, unknown */ |
1034 | 0 | else |
1035 | 0 | zip64local_putValue_inmemory_update(&p, 0, |
1036 | 0 | 4); /* compressed size, unknown */ |
1037 | |
|
1038 | 0 | if (zip64) |
1039 | 0 | zip64local_putValue_inmemory_update(&p, 0xFFFFFFFFU, |
1040 | 0 | 4); /* uncompressed size, unknown */ |
1041 | 0 | else |
1042 | 0 | zip64local_putValue_inmemory_update(&p, 0, |
1043 | 0 | 4); /* uncompressed size, unknown */ |
1044 | |
|
1045 | 0 | zip64local_putValue_inmemory_update(&p, size_filename, 2); |
1046 | |
|
1047 | 0 | zi->ci.size_local_header_extrafield = size_extrafield; |
1048 | |
|
1049 | 0 | zip64local_putValue_inmemory_update(&p, size_extrafield, 2); |
1050 | |
|
1051 | 0 | if (size_filename > 0) |
1052 | 0 | { |
1053 | 0 | memcpy(p, filename, size_filename); |
1054 | 0 | p += size_filename; |
1055 | 0 | } |
1056 | |
|
1057 | 0 | if (size_extrafield_local > 0) |
1058 | 0 | { |
1059 | 0 | memcpy(p, extrafield_local, size_extrafield_local); |
1060 | 0 | p += size_extrafield_local; |
1061 | 0 | } |
1062 | |
|
1063 | 0 | if (zip64) |
1064 | 0 | { |
1065 | | // write the Zip64 extended info |
1066 | 0 | short HeaderID = 1; |
1067 | 0 | short DataSize = 16; |
1068 | 0 | ZPOS64_T CompressedSize = 0; |
1069 | 0 | ZPOS64_T UncompressedSize = 0; |
1070 | | |
1071 | | // Remember position of Zip64 extended info for the local file header. |
1072 | | // (needed when we update size after done with file) |
1073 | 0 | zi->ci.pos_zip64extrainfo = |
1074 | 0 | ZTELL64(zi->z_filefunc, zi->filestream) + p - local_header; |
1075 | |
|
1076 | 0 | zip64local_putValue_inmemory_update(&p, HeaderID, 2); |
1077 | 0 | zip64local_putValue_inmemory_update(&p, DataSize, 2); |
1078 | |
|
1079 | 0 | zip64local_putValue_inmemory_update(&p, UncompressedSize, 8); |
1080 | 0 | zip64local_putValue_inmemory_update(&p, CompressedSize, 8); |
1081 | 0 | } |
1082 | 0 | assert(p == local_header + size_local_header); |
1083 | | |
1084 | 0 | if (ZWRITE64(zi->z_filefunc, zi->filestream, local_header, |
1085 | 0 | size_local_header) != size_local_header) |
1086 | 0 | err = ZIP_ERRNO; |
1087 | |
|
1088 | 0 | zi->ci.local_header = local_header; |
1089 | 0 | zi->ci.size_local_header = size_local_header; |
1090 | |
|
1091 | 0 | return err; |
1092 | 0 | } |
1093 | | |
1094 | | extern int ZEXPORT cpl_zipOpenNewFileInZip3( |
1095 | | zipFile file, const char *filename, const zip_fileinfo *zipfi, |
1096 | | const void *extrafield_local, uInt size_extrafield_local, |
1097 | | const void *extrafield_global, uInt size_extrafield_global, |
1098 | | const char *comment, int method, int level, int raw, int windowBits, |
1099 | | int memLevel, int strategy, const char *password, |
1100 | | #ifdef NOCRYPT |
1101 | | uLong /* crcForCrypting */ |
1102 | | #else |
1103 | | uLong crcForCrypting |
1104 | | #endif |
1105 | | , |
1106 | | bool bZip64, bool bIncludeInCentralDirectory) |
1107 | 0 | { |
1108 | 0 | zip64_internal *zi; |
1109 | 0 | uInt size_filename; |
1110 | 0 | uInt size_comment; |
1111 | 0 | uInt i; |
1112 | 0 | int err = ZIP_OK; |
1113 | 0 | uLong flagBase = 0; |
1114 | |
|
1115 | 0 | #ifdef NOCRYPT |
1116 | 0 | if (password != nullptr) |
1117 | 0 | return ZIP_PARAMERROR; |
1118 | 0 | #endif |
1119 | | |
1120 | 0 | if (file == nullptr) |
1121 | 0 | return ZIP_PARAMERROR; |
1122 | 0 | if ((method != 0) && (method != Z_DEFLATED)) |
1123 | 0 | return ZIP_PARAMERROR; |
1124 | | |
1125 | 0 | zi = reinterpret_cast<zip64_internal *>(file); |
1126 | |
|
1127 | 0 | if (zi->in_opened_file_inzip == 1) |
1128 | 0 | { |
1129 | 0 | err = cpl_zipCloseFileInZip(file); |
1130 | 0 | if (err != ZIP_OK) |
1131 | 0 | return err; |
1132 | 0 | } |
1133 | | |
1134 | 0 | if (filename == nullptr) |
1135 | 0 | filename = "-"; |
1136 | | |
1137 | | // The filename and comment length must fit in 16 bits. |
1138 | 0 | if ((filename != nullptr) && (strlen(filename) > 0xffff)) |
1139 | 0 | return ZIP_PARAMERROR; |
1140 | 0 | if ((comment != nullptr) && (strlen(comment) > 0xffff)) |
1141 | 0 | return ZIP_PARAMERROR; |
1142 | | // The extra field length must fit in 16 bits. If the member also requires |
1143 | | // a Zip64 extra block, that will also need to fit within that 16-bit |
1144 | | // length, but that will be checked for later. |
1145 | 0 | if ((size_extrafield_local > 0xffff) || (size_extrafield_global > 0xffff)) |
1146 | 0 | return ZIP_PARAMERROR; |
1147 | | |
1148 | 0 | if (comment == nullptr) |
1149 | 0 | size_comment = 0; |
1150 | 0 | else |
1151 | 0 | size_comment = static_cast<uInt>(strlen(comment)); |
1152 | |
|
1153 | 0 | size_filename = static_cast<uInt>(strlen(filename)); |
1154 | |
|
1155 | 0 | if (zipfi == nullptr) |
1156 | 0 | zi->ci.dosDate = 0; |
1157 | 0 | else |
1158 | 0 | { |
1159 | 0 | if (zipfi->dosDate != 0) |
1160 | 0 | zi->ci.dosDate = zipfi->dosDate; |
1161 | 0 | else |
1162 | 0 | zi->ci.dosDate = zip64local_TmzDateToDosDate(&zipfi->tmz_date); |
1163 | 0 | } |
1164 | |
|
1165 | 0 | zi->ci.flag = flagBase; |
1166 | 0 | if ((level == 8) || (level == 9)) |
1167 | 0 | zi->ci.flag |= 2; |
1168 | 0 | if (level == 2) |
1169 | 0 | zi->ci.flag |= 4; |
1170 | 0 | if (level == 1) |
1171 | 0 | zi->ci.flag |= 6; |
1172 | | #ifndef NOCRYPT |
1173 | | if (password != nullptr) |
1174 | | zi->ci.flag |= 1; |
1175 | | #endif |
1176 | |
|
1177 | 0 | zi->ci.crc32 = 0; |
1178 | 0 | zi->ci.method = method; |
1179 | 0 | zi->ci.encrypt = 0; |
1180 | 0 | zi->ci.stream_initialised = 0; |
1181 | 0 | zi->ci.pos_in_buffered_data = 0; |
1182 | 0 | zi->ci.raw = raw; |
1183 | 0 | zi->ci.pos_local_header = ZTELL64(zi->z_filefunc, zi->filestream); |
1184 | |
|
1185 | 0 | if (bIncludeInCentralDirectory) |
1186 | 0 | { |
1187 | 0 | zi->ci.size_centralheader = SIZECENTRALHEADER + size_filename + |
1188 | 0 | size_extrafield_global + size_comment; |
1189 | 0 | zi->ci.size_centralExtraFree = |
1190 | 0 | 32; // Extra space we have reserved in case we need to add ZIP64 |
1191 | | // extra info data |
1192 | |
|
1193 | 0 | zi->ci.central_header = static_cast<char *>(ALLOC(static_cast<uInt>( |
1194 | 0 | zi->ci.size_centralheader + zi->ci.size_centralExtraFree))); |
1195 | |
|
1196 | 0 | zi->ci.size_centralExtra = size_extrafield_global; |
1197 | 0 | zip64local_putValue_inmemory(zi->ci.central_header, CENTRALHEADERMAGIC, |
1198 | 0 | 4); |
1199 | | /* version info */ |
1200 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 4, VERSIONMADEBY, |
1201 | 0 | 2); |
1202 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 6, 20, 2); |
1203 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 8, |
1204 | 0 | static_cast<uLong>(zi->ci.flag), 2); |
1205 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 10, |
1206 | 0 | static_cast<uLong>(zi->ci.method), 2); |
1207 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 12, |
1208 | 0 | static_cast<uLong>(zi->ci.dosDate), 4); |
1209 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 16, 0, 4); /*crc*/ |
1210 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 20, 0, |
1211 | 0 | 4); /*compr size*/ |
1212 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 24, 0, |
1213 | 0 | 4); /*uncompr size*/ |
1214 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 28, |
1215 | 0 | static_cast<uLong>(size_filename), 2); |
1216 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 30, |
1217 | 0 | static_cast<uLong>(size_extrafield_global), |
1218 | 0 | 2); |
1219 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 32, |
1220 | 0 | static_cast<uLong>(size_comment), 2); |
1221 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 34, 0, |
1222 | 0 | 2); /*disk nm start*/ |
1223 | |
|
1224 | 0 | if (zipfi == nullptr) |
1225 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 36, 0, 2); |
1226 | 0 | else |
1227 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 36, |
1228 | 0 | static_cast<uLong>(zipfi->internal_fa), |
1229 | 0 | 2); |
1230 | |
|
1231 | 0 | if (zipfi == nullptr) |
1232 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 38, 0, 4); |
1233 | 0 | else |
1234 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 38, |
1235 | 0 | static_cast<uLong>(zipfi->external_fa), |
1236 | 0 | 4); |
1237 | |
|
1238 | 0 | if (zi->ci.pos_local_header >= 0xffffffff) |
1239 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 42, |
1240 | 0 | static_cast<uLong>(0xffffffff), 4); |
1241 | 0 | else |
1242 | 0 | zip64local_putValue_inmemory( |
1243 | 0 | zi->ci.central_header + 42, |
1244 | 0 | static_cast<uLong>(zi->ci.pos_local_header) - |
1245 | 0 | zi->add_position_when_writing_offset, |
1246 | 0 | 4); |
1247 | |
|
1248 | 0 | for (i = 0; i < size_filename; i++) |
1249 | 0 | *(zi->ci.central_header + SIZECENTRALHEADER + i) = *(filename + i); |
1250 | |
|
1251 | 0 | for (i = 0; i < size_extrafield_global; i++) |
1252 | 0 | *(zi->ci.central_header + SIZECENTRALHEADER + size_filename + i) = |
1253 | 0 | *((reinterpret_cast<const char *>(extrafield_global)) + i); |
1254 | |
|
1255 | 0 | for (i = 0; i < size_comment; i++) |
1256 | 0 | *(zi->ci.central_header + SIZECENTRALHEADER + size_filename + |
1257 | 0 | size_extrafield_global + i) = *(comment + i); |
1258 | 0 | if (zi->ci.central_header == nullptr) |
1259 | 0 | return ZIP_INTERNALERROR; |
1260 | 0 | } |
1261 | 0 | else |
1262 | 0 | { |
1263 | 0 | zi->ci.central_header = nullptr; |
1264 | 0 | } |
1265 | | |
1266 | 0 | zi->ci.totalCompressedData = 0; |
1267 | 0 | zi->ci.totalUncompressedData = 0; |
1268 | 0 | zi->ci.pos_zip64extrainfo = 0; |
1269 | | |
1270 | | // For now default is to generate zip64 extra fields |
1271 | 0 | err = Write_LocalFileHeader(zi, filename, size_extrafield_local, |
1272 | 0 | extrafield_local, bZip64 ? 1 : 0); |
1273 | |
|
1274 | 0 | zi->ci.stream.avail_in = 0; |
1275 | 0 | zi->ci.stream.avail_out = Z_BUFSIZE; |
1276 | 0 | zi->ci.stream.next_out = zi->ci.buffered_data; |
1277 | 0 | zi->ci.stream.total_in = 0; |
1278 | 0 | zi->ci.stream.total_out = 0; |
1279 | 0 | zi->ci.stream.data_type = Z_UNKNOWN; |
1280 | |
|
1281 | 0 | if ((err == ZIP_OK) && (zi->ci.method == Z_DEFLATED) && (!zi->ci.raw)) |
1282 | 0 | { |
1283 | 0 | zi->ci.stream.zalloc = nullptr; |
1284 | 0 | zi->ci.stream.zfree = nullptr; |
1285 | 0 | zi->ci.stream.opaque = nullptr; |
1286 | |
|
1287 | 0 | if (windowBits > 0) |
1288 | 0 | windowBits = -windowBits; |
1289 | |
|
1290 | 0 | if (zi->use_cpl_io) |
1291 | 0 | { |
1292 | 0 | auto fpRaw = reinterpret_cast<VSIVirtualHandle *>(zi->filestream); |
1293 | 0 | zi->vsi_raw_length_before = fpRaw->Tell(); |
1294 | 0 | zi->vsi_deflate_handle = VSICreateGZipWritable( |
1295 | 0 | fpRaw, CPL_DEFLATE_TYPE_RAW_DEFLATE, false, zi->nThreads, |
1296 | 0 | zi->nChunkSize, zi->nOffsetSize, zi->sozip_index); |
1297 | 0 | err = Z_OK; |
1298 | 0 | } |
1299 | 0 | else |
1300 | 0 | { |
1301 | 0 | err = deflateInit2(&zi->ci.stream, level, Z_DEFLATED, windowBits, |
1302 | 0 | memLevel, strategy); |
1303 | 0 | } |
1304 | |
|
1305 | 0 | if (err == Z_OK) |
1306 | 0 | zi->ci.stream_initialised = 1; |
1307 | 0 | } |
1308 | | #ifndef NOCRYPT |
1309 | | zi->ci.crypt_header_size = 0; |
1310 | | if ((err == Z_OK) && (password != nullptr)) |
1311 | | { |
1312 | | unsigned char bufHead[RAND_HEAD_LEN]; |
1313 | | unsigned int sizeHead = 0; |
1314 | | zi->ci.encrypt = 1; |
1315 | | zi->ci.pcrc_32_tab = get_crc_table(); |
1316 | | /*init_keys(password,zi->ci.keys,zi->ci.pcrc_32_tab);*/ |
1317 | | |
1318 | | sizeHead = crypthead(password, bufHead, RAND_HEAD_LEN, zi->ci.keys, |
1319 | | zi->ci.pcrc_32_tab, crcForCrypting); |
1320 | | zi->ci.crypt_header_size = sizeHead; |
1321 | | |
1322 | | if (ZWRITE64(zi->z_filefunc, zi->filestream, bufHead, sizeHead) != |
1323 | | sizeHead) |
1324 | | err = ZIP_ERRNO; |
1325 | | } |
1326 | | #endif |
1327 | |
|
1328 | 0 | if (err == Z_OK) |
1329 | 0 | zi->in_opened_file_inzip = 1; |
1330 | 0 | else |
1331 | 0 | { |
1332 | 0 | free(zi->ci.central_header); |
1333 | 0 | zi->ci.central_header = nullptr; |
1334 | 0 | free(zi->ci.local_header); |
1335 | 0 | zi->ci.local_header = nullptr; |
1336 | 0 | } |
1337 | |
|
1338 | 0 | return err; |
1339 | 0 | } |
1340 | | |
1341 | | extern int ZEXPORT cpl_zipOpenNewFileInZip2( |
1342 | | zipFile file, const char *filename, const zip_fileinfo *zipfi, |
1343 | | const void *extrafield_local, uInt size_extrafield_local, |
1344 | | const void *extrafield_global, uInt size_extrafield_global, |
1345 | | const char *comment, int method, int level, int raw) |
1346 | 0 | { |
1347 | 0 | return cpl_zipOpenNewFileInZip3( |
1348 | 0 | file, filename, zipfi, extrafield_local, size_extrafield_local, |
1349 | 0 | extrafield_global, size_extrafield_global, comment, method, level, raw, |
1350 | 0 | -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY, nullptr, 0, true, true); |
1351 | 0 | } |
1352 | | |
1353 | | extern int ZEXPORT cpl_zipOpenNewFileInZip( |
1354 | | zipFile file, const char *filename, const zip_fileinfo *zipfi, |
1355 | | const void *extrafield_local, uInt size_extrafield_local, |
1356 | | const void *extrafield_global, uInt size_extrafield_global, |
1357 | | const char *comment, int method, int level) |
1358 | 0 | { |
1359 | 0 | return cpl_zipOpenNewFileInZip2( |
1360 | 0 | file, filename, zipfi, extrafield_local, size_extrafield_local, |
1361 | 0 | extrafield_global, size_extrafield_global, comment, method, level, 0); |
1362 | 0 | } |
1363 | | |
1364 | | static int zip64FlushWriteBuffer(zip64_internal *zi) |
1365 | 0 | { |
1366 | 0 | int err = ZIP_OK; |
1367 | |
|
1368 | 0 | if (zi->ci.encrypt != 0) |
1369 | 0 | { |
1370 | | #ifndef NOCRYPT |
1371 | | int t = 0; |
1372 | | for (uInt i = 0; i < zi->ci.pos_in_buffered_data; i++) |
1373 | | zi->ci.buffered_data[i] = zencode(zi->ci.keys, zi->ci.pcrc_32_tab, |
1374 | | zi->ci.buffered_data[i], t); |
1375 | | #endif |
1376 | 0 | } |
1377 | 0 | if (ZWRITE64(zi->z_filefunc, zi->filestream, zi->ci.buffered_data, |
1378 | 0 | zi->ci.pos_in_buffered_data) != zi->ci.pos_in_buffered_data) |
1379 | 0 | err = ZIP_ERRNO; |
1380 | |
|
1381 | 0 | zi->ci.totalCompressedData += zi->ci.pos_in_buffered_data; |
1382 | 0 | zi->ci.totalUncompressedData += zi->ci.stream.total_in; |
1383 | 0 | zi->ci.stream.total_in = 0; |
1384 | |
|
1385 | 0 | zi->ci.pos_in_buffered_data = 0; |
1386 | 0 | return err; |
1387 | 0 | } |
1388 | | |
1389 | | extern int ZEXPORT cpl_zipWriteInFileInZip(zipFile file, const void *buf, |
1390 | | unsigned len) |
1391 | 0 | { |
1392 | 0 | if (file == nullptr) |
1393 | 0 | return ZIP_PARAMERROR; |
1394 | | |
1395 | 0 | zip64_internal *zi = reinterpret_cast<zip64_internal *>(file); |
1396 | |
|
1397 | 0 | if (zi->in_opened_file_inzip == 0) |
1398 | 0 | return ZIP_PARAMERROR; |
1399 | | |
1400 | 0 | zi->ci.stream.next_in = reinterpret_cast<Bytef *>(const_cast<void *>(buf)); |
1401 | 0 | zi->ci.stream.avail_in = len; |
1402 | 0 | zi->ci.crc32 = |
1403 | 0 | crc32(zi->ci.crc32, reinterpret_cast<const Bytef *>(buf), len); |
1404 | |
|
1405 | 0 | int err = ZIP_OK; |
1406 | 0 | while ((err == ZIP_OK) && (zi->ci.stream.avail_in > 0)) |
1407 | 0 | { |
1408 | 0 | if (zi->ci.stream.avail_out == 0) |
1409 | 0 | { |
1410 | 0 | if (zip64FlushWriteBuffer(zi) == ZIP_ERRNO) |
1411 | 0 | err = ZIP_ERRNO; |
1412 | 0 | zi->ci.stream.avail_out = Z_BUFSIZE; |
1413 | 0 | zi->ci.stream.next_out = zi->ci.buffered_data; |
1414 | 0 | } |
1415 | |
|
1416 | 0 | if (err != ZIP_OK) |
1417 | 0 | break; |
1418 | | |
1419 | 0 | if ((zi->ci.method == Z_DEFLATED) && (!zi->ci.raw)) |
1420 | 0 | { |
1421 | 0 | if (zi->vsi_deflate_handle) |
1422 | 0 | { |
1423 | 0 | zi->ci.totalUncompressedData += len; |
1424 | 0 | if (zi->vsi_deflate_handle->Write(buf, 1, len) < len) |
1425 | 0 | err = ZIP_INTERNALERROR; |
1426 | 0 | zi->ci.stream.avail_in = 0; |
1427 | 0 | } |
1428 | 0 | else |
1429 | 0 | { |
1430 | 0 | uLong uTotalOutBefore = zi->ci.stream.total_out; |
1431 | 0 | err = deflate(&zi->ci.stream, Z_NO_FLUSH); |
1432 | 0 | zi->ci.pos_in_buffered_data += static_cast<uInt>( |
1433 | 0 | zi->ci.stream.total_out - uTotalOutBefore); |
1434 | 0 | } |
1435 | 0 | } |
1436 | 0 | else |
1437 | 0 | { |
1438 | 0 | uInt copy_this; |
1439 | 0 | if (zi->ci.stream.avail_in < zi->ci.stream.avail_out) |
1440 | 0 | copy_this = zi->ci.stream.avail_in; |
1441 | 0 | else |
1442 | 0 | copy_this = zi->ci.stream.avail_out; |
1443 | 0 | for (uInt i = 0; i < copy_this; i++) |
1444 | 0 | *((reinterpret_cast<char *>(zi->ci.stream.next_out)) + i) = |
1445 | 0 | *((reinterpret_cast<const char *>(zi->ci.stream.next_in)) + |
1446 | 0 | i); |
1447 | 0 | { |
1448 | 0 | zi->ci.stream.avail_in -= copy_this; |
1449 | 0 | zi->ci.stream.avail_out -= copy_this; |
1450 | 0 | zi->ci.stream.next_in += copy_this; |
1451 | 0 | zi->ci.stream.next_out += copy_this; |
1452 | 0 | zi->ci.stream.total_in += copy_this; |
1453 | 0 | zi->ci.stream.total_out += copy_this; |
1454 | 0 | zi->ci.pos_in_buffered_data += copy_this; |
1455 | 0 | } |
1456 | 0 | } |
1457 | 0 | } |
1458 | |
|
1459 | 0 | return err; |
1460 | 0 | } |
1461 | | |
1462 | | extern int ZEXPORT cpl_zipCloseFileInZipRaw(zipFile file, |
1463 | | ZPOS64_T uncompressed_size, |
1464 | | uLong crc32) |
1465 | 0 | { |
1466 | 0 | if (file == nullptr) |
1467 | 0 | return ZIP_PARAMERROR; |
1468 | | |
1469 | 0 | zip64_internal *zi = reinterpret_cast<zip64_internal *>(file); |
1470 | |
|
1471 | 0 | if (zi->in_opened_file_inzip == 0) |
1472 | 0 | return ZIP_PARAMERROR; |
1473 | 0 | zi->ci.stream.avail_in = 0; |
1474 | |
|
1475 | 0 | int err = ZIP_OK; |
1476 | 0 | if ((zi->ci.method == Z_DEFLATED) && (!zi->ci.raw)) |
1477 | 0 | { |
1478 | 0 | if (zi->vsi_deflate_handle) |
1479 | 0 | { |
1480 | 0 | auto fpRaw = reinterpret_cast<VSIVirtualHandle *>(zi->filestream); |
1481 | 0 | delete zi->vsi_deflate_handle; |
1482 | 0 | zi->vsi_deflate_handle = nullptr; |
1483 | 0 | zi->ci.totalCompressedData = |
1484 | 0 | fpRaw->Tell() - zi->vsi_raw_length_before; |
1485 | |
|
1486 | 0 | if (zi->sozip_index) |
1487 | 0 | { |
1488 | 0 | uint64_t nVal = |
1489 | 0 | static_cast<uint64_t>(zi->ci.totalCompressedData); |
1490 | 0 | CPL_LSBPTR64(&nVal); |
1491 | 0 | memcpy(zi->sozip_index->data() + 24, &nVal, sizeof(uint64_t)); |
1492 | 0 | } |
1493 | 0 | } |
1494 | 0 | else |
1495 | 0 | { |
1496 | 0 | while (err == ZIP_OK) |
1497 | 0 | { |
1498 | 0 | if (zi->ci.stream.avail_out == 0) |
1499 | 0 | { |
1500 | 0 | if (zip64FlushWriteBuffer(zi) == ZIP_ERRNO) |
1501 | 0 | { |
1502 | 0 | err = ZIP_ERRNO; |
1503 | 0 | break; |
1504 | 0 | } |
1505 | 0 | zi->ci.stream.avail_out = Z_BUFSIZE; |
1506 | 0 | zi->ci.stream.next_out = zi->ci.buffered_data; |
1507 | 0 | } |
1508 | 0 | uLong uTotalOutBefore = zi->ci.stream.total_out; |
1509 | 0 | err = deflate(&zi->ci.stream, Z_FINISH); |
1510 | 0 | zi->ci.pos_in_buffered_data += static_cast<uInt>( |
1511 | 0 | zi->ci.stream.total_out - uTotalOutBefore); |
1512 | 0 | } |
1513 | 0 | } |
1514 | 0 | } |
1515 | |
|
1516 | 0 | if (err == Z_STREAM_END) |
1517 | 0 | err = ZIP_OK; /* this is normal */ |
1518 | |
|
1519 | 0 | if ((zi->ci.pos_in_buffered_data > 0) && (err == ZIP_OK)) |
1520 | 0 | if (zip64FlushWriteBuffer(zi) == ZIP_ERRNO) |
1521 | 0 | err = ZIP_ERRNO; |
1522 | |
|
1523 | 0 | if (!zi->use_cpl_io && (zi->ci.method == Z_DEFLATED) && (!zi->ci.raw)) |
1524 | 0 | { |
1525 | 0 | err = deflateEnd(&zi->ci.stream); |
1526 | 0 | zi->ci.stream_initialised = 0; |
1527 | 0 | } |
1528 | |
|
1529 | 0 | if (!zi->ci.raw) |
1530 | 0 | { |
1531 | 0 | crc32 = static_cast<uLong>(zi->ci.crc32); |
1532 | 0 | uncompressed_size = zi->ci.totalUncompressedData; |
1533 | 0 | } |
1534 | 0 | ZPOS64_T compressed_size = zi->ci.totalCompressedData; |
1535 | | #ifndef NOCRYPT |
1536 | | compressed_size += zi->ci.crypt_header_size; |
1537 | | #endif |
1538 | |
|
1539 | | #ifdef disabled |
1540 | | // Code finally disabled since it causes compatibility issues with |
1541 | | // libreoffice for .xlsx or .ods files |
1542 | | if (zi->ci.pos_zip64extrainfo && uncompressed_size < 0xffffffff && |
1543 | | compressed_size < 0xffffffff) |
1544 | | { |
1545 | | // Update the LocalFileHeader to be a regular one and not a ZIP64 one |
1546 | | // by removing its trailing 20 bytes, and moving it in the file |
1547 | | // 20 bytes further of its original position. |
1548 | | |
1549 | | ZPOS64_T cur_pos_inzip = ZTELL64(zi->z_filefunc, zi->filestream); |
1550 | | |
1551 | | if (ZSEEK64(zi->z_filefunc, zi->filestream, zi->ci.pos_local_header, |
1552 | | ZLIB_FILEFUNC_SEEK_SET) != 0) |
1553 | | err = ZIP_ERRNO; |
1554 | | |
1555 | | // Insert leading padding |
1556 | | constexpr uInt nZIP64ExtraBytes = 20; |
1557 | | char padding[nZIP64ExtraBytes]; |
1558 | | memset(padding, 0, sizeof(padding)); |
1559 | | if (ZWRITE64(zi->z_filefunc, zi->filestream, padding, |
1560 | | nZIP64ExtraBytes) != nZIP64ExtraBytes) |
1561 | | err = ZIP_ERRNO; |
1562 | | |
1563 | | // Correct version needed to extract |
1564 | | zip64local_putValue_inmemory(zi->ci.local_header + 4, 20, 2); |
1565 | | |
1566 | | // Correct extra field length |
1567 | | zi->ci.size_local_header_extrafield -= nZIP64ExtraBytes; |
1568 | | zip64local_putValue_inmemory(zi->ci.local_header + 28, |
1569 | | zi->ci.size_local_header_extrafield, 2); |
1570 | | |
1571 | | zi->ci.size_local_header -= nZIP64ExtraBytes; |
1572 | | |
1573 | | // Rewrite local header |
1574 | | if (ZWRITE64(zi->z_filefunc, zi->filestream, zi->ci.local_header, |
1575 | | zi->ci.size_local_header) != zi->ci.size_local_header) |
1576 | | err = ZIP_ERRNO; |
1577 | | |
1578 | | if (ZSEEK64(zi->z_filefunc, zi->filestream, cur_pos_inzip, |
1579 | | ZLIB_FILEFUNC_SEEK_SET) != 0) |
1580 | | err = ZIP_ERRNO; |
1581 | | |
1582 | | zi->ci.pos_zip64extrainfo = 0; |
1583 | | |
1584 | | // Correct central header offset to local header |
1585 | | zi->ci.pos_local_header += nZIP64ExtraBytes; |
1586 | | if (zi->ci.central_header) |
1587 | | { |
1588 | | if (zi->ci.pos_local_header >= 0xffffffff) |
1589 | | zip64local_putValue_inmemory(zi->ci.central_header + 42, |
1590 | | static_cast<uLong>(0xffffffff), 4); |
1591 | | else |
1592 | | zip64local_putValue_inmemory( |
1593 | | zi->ci.central_header + 42, |
1594 | | static_cast<uLong>(zi->ci.pos_local_header) - |
1595 | | zi->add_position_when_writing_offset, |
1596 | | 4); |
1597 | | } |
1598 | | } |
1599 | | #endif |
1600 | |
|
1601 | 0 | const bool bInCentralHeader = zi->ci.central_header != nullptr; |
1602 | 0 | if (zi->ci.central_header) |
1603 | 0 | { |
1604 | | // update Current Item crc and sizes, |
1605 | 0 | if (zi->ci.pos_zip64extrainfo || compressed_size >= 0xffffffff || |
1606 | 0 | uncompressed_size >= 0xffffffff || |
1607 | 0 | zi->ci.pos_local_header >= 0xffffffff) |
1608 | 0 | { |
1609 | | /*version Made by*/ |
1610 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 4, 45, 2); |
1611 | | /*version needed*/ |
1612 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 6, 45, 2); |
1613 | 0 | } |
1614 | |
|
1615 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 16, crc32, |
1616 | 0 | 4); /*crc*/ |
1617 | |
|
1618 | 0 | const uLong invalidValue = 0xffffffff; |
1619 | 0 | if (compressed_size >= 0xffffffff) |
1620 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 20, |
1621 | 0 | invalidValue, 4); /*compr size*/ |
1622 | 0 | else |
1623 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 20, |
1624 | 0 | compressed_size, 4); /*compr size*/ |
1625 | | |
1626 | | /// set internal file attributes field |
1627 | 0 | if (zi->ci.stream.data_type == Z_ASCII) |
1628 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 36, Z_ASCII, |
1629 | 0 | 2); |
1630 | |
|
1631 | 0 | if (uncompressed_size >= 0xffffffff) |
1632 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 24, |
1633 | 0 | invalidValue, 4); /*uncompr size*/ |
1634 | 0 | else |
1635 | 0 | zip64local_putValue_inmemory(zi->ci.central_header + 24, |
1636 | 0 | uncompressed_size, 4); /*uncompr size*/ |
1637 | |
|
1638 | 0 | short datasize = 0; |
1639 | | // Add ZIP64 extra info field for uncompressed size |
1640 | 0 | if (uncompressed_size >= 0xffffffff) |
1641 | 0 | datasize += 8; |
1642 | | |
1643 | | // Add ZIP64 extra info field for compressed size |
1644 | 0 | if (compressed_size >= 0xffffffff) |
1645 | 0 | datasize += 8; |
1646 | | |
1647 | | // Add ZIP64 extra info field for relative offset to local file header |
1648 | | // of current file |
1649 | 0 | if (zi->ci.pos_local_header >= 0xffffffff) |
1650 | 0 | datasize += 8; |
1651 | |
|
1652 | 0 | if (datasize > 0) |
1653 | 0 | { |
1654 | 0 | char *p = nullptr; |
1655 | |
|
1656 | 0 | if (static_cast<uLong>(datasize + 4) > zi->ci.size_centralExtraFree) |
1657 | 0 | { |
1658 | | // we can not write more data to the buffer that we have room |
1659 | | // for. |
1660 | 0 | return ZIP_BADZIPFILE; |
1661 | 0 | } |
1662 | | |
1663 | 0 | p = zi->ci.central_header + zi->ci.size_centralheader; |
1664 | | |
1665 | | // Add Extra Information Header for 'ZIP64 information' |
1666 | 0 | zip64local_putValue_inmemory(p, 0x0001, 2); // HeaderID |
1667 | 0 | p += 2; |
1668 | 0 | zip64local_putValue_inmemory(p, datasize, 2); // DataSize |
1669 | 0 | p += 2; |
1670 | |
|
1671 | 0 | if (uncompressed_size >= 0xffffffff) |
1672 | 0 | { |
1673 | 0 | zip64local_putValue_inmemory(p, uncompressed_size, 8); |
1674 | 0 | p += 8; |
1675 | 0 | } |
1676 | |
|
1677 | 0 | if (compressed_size >= 0xffffffff) |
1678 | 0 | { |
1679 | 0 | zip64local_putValue_inmemory(p, compressed_size, 8); |
1680 | 0 | p += 8; |
1681 | 0 | } |
1682 | |
|
1683 | 0 | if (zi->ci.pos_local_header >= 0xffffffff) |
1684 | 0 | { |
1685 | 0 | zip64local_putValue_inmemory(p, zi->ci.pos_local_header, 8); |
1686 | | // p += 8; |
1687 | 0 | } |
1688 | | |
1689 | | // Update how much extra free space we got in the memory buffer |
1690 | | // and increase the centralheader size so the new ZIP64 fields are |
1691 | | // included ( 4 below is the size of HeaderID and DataSize field ) |
1692 | 0 | zi->ci.size_centralExtraFree -= datasize + 4; |
1693 | 0 | zi->ci.size_centralheader += datasize + 4; |
1694 | | |
1695 | | // Update the extra info size field |
1696 | 0 | zi->ci.size_centralExtra += datasize + 4; |
1697 | 0 | zip64local_putValue_inmemory( |
1698 | 0 | zi->ci.central_header + 30, |
1699 | 0 | static_cast<uLong>(zi->ci.size_centralExtra), 2); |
1700 | 0 | } |
1701 | | |
1702 | 0 | if (err == ZIP_OK) |
1703 | 0 | err = add_data_in_datablock( |
1704 | 0 | &zi->central_dir, zi->ci.central_header, |
1705 | 0 | static_cast<uLong>(zi->ci.size_centralheader)); |
1706 | 0 | free(zi->ci.central_header); |
1707 | 0 | zi->ci.central_header = nullptr; |
1708 | 0 | } |
1709 | | |
1710 | 0 | free(zi->ci.local_header); |
1711 | 0 | zi->ci.local_header = nullptr; |
1712 | |
|
1713 | 0 | if (err == ZIP_OK) |
1714 | 0 | { |
1715 | | // Update the LocalFileHeader with the new values. |
1716 | |
|
1717 | 0 | ZPOS64_T cur_pos_inzip = ZTELL64(zi->z_filefunc, zi->filestream); |
1718 | |
|
1719 | 0 | if (ZSEEK64(zi->z_filefunc, zi->filestream, |
1720 | 0 | zi->ci.pos_local_header + 14, ZLIB_FILEFUNC_SEEK_SET) != 0) |
1721 | 0 | err = ZIP_ERRNO; |
1722 | |
|
1723 | 0 | if (err == ZIP_OK) |
1724 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, crc32, |
1725 | 0 | 4); /* crc 32, unknown */ |
1726 | |
|
1727 | 0 | if (uncompressed_size >= 0xffffffff || compressed_size >= 0xffffffff) |
1728 | 0 | { |
1729 | 0 | if (zi->ci.pos_zip64extrainfo > 0) |
1730 | 0 | { |
1731 | | // Update the size in the ZIP64 extended field. |
1732 | 0 | if (ZSEEK64(zi->z_filefunc, zi->filestream, |
1733 | 0 | zi->ci.pos_zip64extrainfo + 4, |
1734 | 0 | ZLIB_FILEFUNC_SEEK_SET) != 0) |
1735 | 0 | err = ZIP_ERRNO; |
1736 | |
|
1737 | 0 | if (err == ZIP_OK) /* compressed size, unknown */ |
1738 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1739 | 0 | uncompressed_size, 8); |
1740 | |
|
1741 | 0 | if (err == ZIP_OK) /* uncompressed size, unknown */ |
1742 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1743 | 0 | compressed_size, 8); |
1744 | 0 | } |
1745 | 0 | else |
1746 | 0 | err = ZIP_BADZIPFILE; // Caller passed zip64 = 0, so no room |
1747 | | // for zip64 info -> fatal |
1748 | 0 | } |
1749 | 0 | else |
1750 | 0 | { |
1751 | 0 | if (err == ZIP_OK) /* compressed size, unknown */ |
1752 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1753 | 0 | compressed_size, 4); |
1754 | |
|
1755 | 0 | if (err == ZIP_OK) /* uncompressed size, unknown */ |
1756 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1757 | 0 | uncompressed_size, 4); |
1758 | 0 | } |
1759 | 0 | if (ZSEEK64(zi->z_filefunc, zi->filestream, cur_pos_inzip, |
1760 | 0 | ZLIB_FILEFUNC_SEEK_SET) != 0) |
1761 | 0 | err = ZIP_ERRNO; |
1762 | 0 | } |
1763 | |
|
1764 | 0 | if (bInCentralHeader) |
1765 | 0 | zi->number_entry++; |
1766 | 0 | zi->in_opened_file_inzip = 0; |
1767 | |
|
1768 | 0 | return err; |
1769 | 0 | } |
1770 | | |
1771 | | extern int ZEXPORT cpl_zipCloseFileInZip(zipFile file) |
1772 | 0 | { |
1773 | 0 | return cpl_zipCloseFileInZipRaw(file, 0, 0); |
1774 | 0 | } |
1775 | | |
1776 | | static int Write_Zip64EndOfCentralDirectoryLocator(zip64_internal *zi, |
1777 | | ZPOS64_T zip64eocd_pos_inzip) |
1778 | 0 | { |
1779 | 0 | int err = ZIP_OK; |
1780 | 0 | ZPOS64_T pos = zip64eocd_pos_inzip - zi->add_position_when_writing_offset; |
1781 | |
|
1782 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1783 | 0 | ZIP64ENDLOCHEADERMAGIC, 4); |
1784 | | |
1785 | | /*num disks*/ |
1786 | 0 | if (err == |
1787 | 0 | ZIP_OK) /* number of the disk with the start of the central directory */ |
1788 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, 0, 4); |
1789 | | |
1790 | | /*relative offset*/ |
1791 | 0 | if (err == ZIP_OK) /* Relative offset to the Zip64EndOfCentralDirectory */ |
1792 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, pos, 8); |
1793 | | |
1794 | | /*total disks*/ /* Do not support spawning of disk so always say 1 here*/ |
1795 | 0 | if (err == |
1796 | 0 | ZIP_OK) /* number of the disk with the start of the central directory */ |
1797 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, 1, 4); |
1798 | |
|
1799 | 0 | return err; |
1800 | 0 | } |
1801 | | |
1802 | | static int Write_Zip64EndOfCentralDirectoryRecord(zip64_internal *zi, |
1803 | | uLong size_centraldir, |
1804 | | ZPOS64_T centraldir_pos_inzip) |
1805 | 0 | { |
1806 | 0 | int err = ZIP_OK; |
1807 | |
|
1808 | 0 | uLong Zip64DataSize = 44; |
1809 | |
|
1810 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1811 | 0 | ZIP64ENDHEADERMAGIC, 4); |
1812 | |
|
1813 | 0 | if (err == ZIP_OK) /* size of this 'zip64 end of central directory' */ |
1814 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1815 | 0 | Zip64DataSize, 8); // why ZPOS64_T of this ? |
1816 | |
|
1817 | 0 | if (err == ZIP_OK) /* version made by */ |
1818 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, 45, 2); |
1819 | |
|
1820 | 0 | if (err == ZIP_OK) /* version needed */ |
1821 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, 45, 2); |
1822 | |
|
1823 | 0 | if (err == ZIP_OK) /* number of this disk */ |
1824 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, 0, 4); |
1825 | |
|
1826 | 0 | if (err == |
1827 | 0 | ZIP_OK) /* number of the disk with the start of the central directory */ |
1828 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, 0, 4); |
1829 | |
|
1830 | 0 | if (err == |
1831 | 0 | ZIP_OK) /* total number of entries in the central dir on this disk */ |
1832 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1833 | 0 | zi->number_entry, 8); |
1834 | |
|
1835 | 0 | if (err == ZIP_OK) /* total number of entries in the central dir */ |
1836 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1837 | 0 | zi->number_entry, 8); |
1838 | |
|
1839 | 0 | if (err == ZIP_OK) /* size of the central directory */ |
1840 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1841 | 0 | size_centraldir, 8); |
1842 | |
|
1843 | 0 | if (err == ZIP_OK) /* offset of start of central directory with respect to |
1844 | | the starting disk number */ |
1845 | 0 | { |
1846 | 0 | ZPOS64_T pos = |
1847 | 0 | centraldir_pos_inzip - zi->add_position_when_writing_offset; |
1848 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, pos, 8); |
1849 | 0 | } |
1850 | 0 | return err; |
1851 | 0 | } |
1852 | | |
1853 | | static int Write_EndOfCentralDirectoryRecord(zip64_internal *zi, |
1854 | | uLong size_centraldir, |
1855 | | ZPOS64_T centraldir_pos_inzip) |
1856 | 0 | { |
1857 | 0 | int err = ZIP_OK; |
1858 | | |
1859 | | /*signature*/ |
1860 | 0 | err = |
1861 | 0 | zip64local_putValue(&zi->z_filefunc, zi->filestream, ENDHEADERMAGIC, 4); |
1862 | |
|
1863 | 0 | if (err == ZIP_OK) /* number of this disk */ |
1864 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, 0, 2); |
1865 | |
|
1866 | 0 | if (err == |
1867 | 0 | ZIP_OK) /* number of the disk with the start of the central directory */ |
1868 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, 0, 2); |
1869 | |
|
1870 | 0 | if (err == |
1871 | 0 | ZIP_OK) /* total number of entries in the central dir on this disk */ |
1872 | 0 | { |
1873 | 0 | { |
1874 | 0 | if (zi->number_entry >= 0xFFFF) |
1875 | 0 | err = |
1876 | 0 | zip64local_putValue(&zi->z_filefunc, zi->filestream, 0xffff, |
1877 | 0 | 2); // use value in ZIP64 record |
1878 | 0 | else |
1879 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1880 | 0 | zi->number_entry, 2); |
1881 | 0 | } |
1882 | 0 | } |
1883 | |
|
1884 | 0 | if (err == ZIP_OK) /* total number of entries in the central dir */ |
1885 | 0 | { |
1886 | 0 | if (zi->number_entry >= 0xFFFF) |
1887 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, 0xffff, |
1888 | 0 | 2); // use value in ZIP64 record |
1889 | 0 | else |
1890 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1891 | 0 | zi->number_entry, 2); |
1892 | 0 | } |
1893 | |
|
1894 | 0 | if (err == ZIP_OK) /* size of the central directory */ |
1895 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1896 | 0 | size_centraldir, 4); |
1897 | |
|
1898 | 0 | if (err == ZIP_OK) /* offset of start of central directory with respect to |
1899 | | the starting disk number */ |
1900 | 0 | { |
1901 | 0 | ZPOS64_T pos = |
1902 | 0 | centraldir_pos_inzip - zi->add_position_when_writing_offset; |
1903 | 0 | if (pos >= 0xffffffff) |
1904 | 0 | { |
1905 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1906 | 0 | 0xffffffff, 4); |
1907 | 0 | } |
1908 | 0 | else |
1909 | 0 | err = zip64local_putValue( |
1910 | 0 | &zi->z_filefunc, zi->filestream, |
1911 | 0 | (centraldir_pos_inzip - zi->add_position_when_writing_offset), |
1912 | 0 | 4); |
1913 | 0 | } |
1914 | |
|
1915 | 0 | return err; |
1916 | 0 | } |
1917 | | |
1918 | | static int Write_GlobalComment(zip64_internal *zi, const char *global_comment) |
1919 | 0 | { |
1920 | 0 | int err = ZIP_OK; |
1921 | 0 | uInt size_global_comment = 0; |
1922 | |
|
1923 | 0 | if (global_comment != nullptr) |
1924 | 0 | size_global_comment = static_cast<uInt>(strlen(global_comment)); |
1925 | |
|
1926 | 0 | err = zip64local_putValue(&zi->z_filefunc, zi->filestream, |
1927 | 0 | size_global_comment, 2); |
1928 | |
|
1929 | 0 | if (err == ZIP_OK && size_global_comment > 0) |
1930 | 0 | { |
1931 | 0 | if (ZWRITE64(zi->z_filefunc, zi->filestream, global_comment, |
1932 | 0 | size_global_comment) != size_global_comment) |
1933 | 0 | err = ZIP_ERRNO; |
1934 | 0 | } |
1935 | 0 | return err; |
1936 | 0 | } |
1937 | | |
1938 | | extern int ZEXPORT cpl_zipClose(zipFile file, const char *global_comment) |
1939 | 0 | { |
1940 | 0 | int err = 0; |
1941 | 0 | uLong size_centraldir = 0; |
1942 | 0 | ZPOS64_T centraldir_pos_inzip; |
1943 | 0 | ZPOS64_T pos; |
1944 | |
|
1945 | 0 | if (file == nullptr) |
1946 | 0 | return ZIP_PARAMERROR; |
1947 | | |
1948 | 0 | zip64_internal *zi = reinterpret_cast<zip64_internal *>(file); |
1949 | |
|
1950 | 0 | if (zi->in_opened_file_inzip == 1) |
1951 | 0 | { |
1952 | 0 | err = cpl_zipCloseFileInZip(file); |
1953 | 0 | } |
1954 | |
|
1955 | 0 | #ifndef NO_ADDFILEINEXISTINGZIP |
1956 | 0 | if (global_comment == nullptr) |
1957 | 0 | global_comment = zi->globalcomment; |
1958 | 0 | #endif |
1959 | |
|
1960 | 0 | centraldir_pos_inzip = ZTELL64(zi->z_filefunc, zi->filestream); |
1961 | 0 | if (err == ZIP_OK) |
1962 | 0 | { |
1963 | 0 | linkedlist_datablock_internal *ldi = zi->central_dir.first_block; |
1964 | 0 | while (ldi != nullptr) |
1965 | 0 | { |
1966 | 0 | if ((err == ZIP_OK) && (ldi->filled_in_this_block > 0)) |
1967 | 0 | if (ZWRITE64(zi->z_filefunc, zi->filestream, ldi->data, |
1968 | 0 | ldi->filled_in_this_block) != |
1969 | 0 | ldi->filled_in_this_block) |
1970 | 0 | err = ZIP_ERRNO; |
1971 | |
|
1972 | 0 | size_centraldir += ldi->filled_in_this_block; |
1973 | 0 | ldi = ldi->next_datablock; |
1974 | 0 | } |
1975 | 0 | } |
1976 | 0 | free_linkedlist(&(zi->central_dir)); |
1977 | |
|
1978 | 0 | pos = centraldir_pos_inzip - zi->add_position_when_writing_offset; |
1979 | 0 | if (pos >= 0xffffffff || zi->number_entry > 0xFFFF) |
1980 | 0 | { |
1981 | 0 | ZPOS64_T Zip64EOCDpos = ZTELL64(zi->z_filefunc, zi->filestream); |
1982 | 0 | Write_Zip64EndOfCentralDirectoryRecord(zi, size_centraldir, |
1983 | 0 | centraldir_pos_inzip); |
1984 | |
|
1985 | 0 | Write_Zip64EndOfCentralDirectoryLocator(zi, Zip64EOCDpos); |
1986 | 0 | } |
1987 | |
|
1988 | 0 | if (err == ZIP_OK) |
1989 | 0 | err = Write_EndOfCentralDirectoryRecord(zi, size_centraldir, |
1990 | 0 | centraldir_pos_inzip); |
1991 | |
|
1992 | 0 | if (err == ZIP_OK) |
1993 | 0 | err = Write_GlobalComment(zi, global_comment); |
1994 | |
|
1995 | 0 | if (ZCLOSE64(zi->z_filefunc, zi->filestream) != 0) |
1996 | 0 | if (err == ZIP_OK) |
1997 | 0 | err = ZIP_ERRNO; |
1998 | |
|
1999 | 0 | #ifndef NO_ADDFILEINEXISTINGZIP |
2000 | 0 | TRYFREE(zi->globalcomment); |
2001 | 0 | #endif |
2002 | 0 | TRYFREE(zi); |
2003 | |
|
2004 | 0 | return err; |
2005 | 0 | } |
2006 | | |
2007 | | /************************************************************************/ |
2008 | | /* ==================================================================== */ |
2009 | | /* The following is a simplified CPL API for creating ZIP files */ |
2010 | | /* exported from cpl_conv.h. */ |
2011 | | /* ==================================================================== */ |
2012 | | /************************************************************************/ |
2013 | | |
2014 | | #include "cpl_minizip_unzip.h" |
2015 | | |
2016 | | typedef struct |
2017 | | { |
2018 | | zipFile hZip; |
2019 | | char **papszFilenames; |
2020 | | } CPLZip; |
2021 | | |
2022 | | /************************************************************************/ |
2023 | | /* CPLCreateZip() */ |
2024 | | /************************************************************************/ |
2025 | | |
2026 | | /** Create ZIP file */ |
2027 | | void *CPLCreateZip(const char *pszZipFilename, char **papszOptions) |
2028 | | |
2029 | 0 | { |
2030 | 0 | const bool bAppend = |
2031 | 0 | CPLTestBool(CSLFetchNameValueDef(papszOptions, "APPEND", "FALSE")); |
2032 | 0 | char **papszFilenames = nullptr; |
2033 | |
|
2034 | 0 | if (bAppend) |
2035 | 0 | { |
2036 | 0 | zipFile unzF = cpl_unzOpen(pszZipFilename); |
2037 | 0 | if (unzF != nullptr) |
2038 | 0 | { |
2039 | 0 | if (cpl_unzGoToFirstFile(unzF) == UNZ_OK) |
2040 | 0 | { |
2041 | 0 | do |
2042 | 0 | { |
2043 | 0 | char fileName[8193]; |
2044 | 0 | unz_file_info file_info; |
2045 | 0 | cpl_unzGetCurrentFileInfo(unzF, &file_info, fileName, |
2046 | 0 | sizeof(fileName) - 1, nullptr, 0, |
2047 | 0 | nullptr, 0); |
2048 | 0 | fileName[sizeof(fileName) - 1] = '\0'; |
2049 | 0 | papszFilenames = CSLAddString(papszFilenames, fileName); |
2050 | 0 | } while (cpl_unzGoToNextFile(unzF) == UNZ_OK); |
2051 | 0 | } |
2052 | 0 | cpl_unzClose(unzF); |
2053 | 0 | } |
2054 | 0 | } |
2055 | |
|
2056 | 0 | zipFile hZip = cpl_zipOpen(pszZipFilename, bAppend ? APPEND_STATUS_ADDINZIP |
2057 | 0 | : APPEND_STATUS_CREATE); |
2058 | 0 | if (hZip == nullptr) |
2059 | 0 | { |
2060 | 0 | CSLDestroy(papszFilenames); |
2061 | 0 | return nullptr; |
2062 | 0 | } |
2063 | | |
2064 | 0 | CPLZip *psZip = static_cast<CPLZip *>(CPLMalloc(sizeof(CPLZip))); |
2065 | 0 | psZip->hZip = hZip; |
2066 | 0 | psZip->papszFilenames = papszFilenames; |
2067 | 0 | return psZip; |
2068 | 0 | } |
2069 | | |
2070 | | /************************************************************************/ |
2071 | | /* CPLCreateFileInZip() */ |
2072 | | /************************************************************************/ |
2073 | | |
2074 | | /** Create a file in a ZIP file */ |
2075 | | CPLErr CPLCreateFileInZip(void *hZip, const char *pszFilename, |
2076 | | char **papszOptions) |
2077 | | |
2078 | 0 | { |
2079 | 0 | if (hZip == nullptr) |
2080 | 0 | return CE_Failure; |
2081 | | |
2082 | 0 | CPLZip *psZip = static_cast<CPLZip *>(hZip); |
2083 | |
|
2084 | 0 | if (CSLFindString(psZip->papszFilenames, pszFilename) >= 0) |
2085 | 0 | { |
2086 | 0 | CPLError(CE_Failure, CPLE_AppDefined, "%s already exists in ZIP file", |
2087 | 0 | pszFilename); |
2088 | 0 | return CE_Failure; |
2089 | 0 | } |
2090 | | |
2091 | 0 | const bool bCompressed = |
2092 | 0 | CPLTestBool(CSLFetchNameValueDef(papszOptions, "COMPRESSED", "TRUE")); |
2093 | |
|
2094 | 0 | char *pszCPFilename = nullptr; |
2095 | 0 | std::vector<GByte> abyExtra; |
2096 | | // If the filename is not ASCII only, we need an extended field |
2097 | 0 | if (!CPLIsASCII(pszFilename, strlen(pszFilename))) |
2098 | 0 | { |
2099 | 0 | const char *pszDestEncoding = CPLGetConfigOption("CPL_ZIP_ENCODING", |
2100 | | #if defined(_WIN32) && !defined(HAVE_ICONV) |
2101 | | "CP_OEMCP" |
2102 | | #else |
2103 | 0 | "CP437" |
2104 | 0 | #endif |
2105 | 0 | ); |
2106 | |
|
2107 | 0 | { |
2108 | 0 | CPLErrorStateBackuper oBackuper(CPLQuietErrorHandler); |
2109 | 0 | pszCPFilename = |
2110 | 0 | CPLRecode(pszFilename, CPL_ENC_UTF8, pszDestEncoding); |
2111 | |
|
2112 | 0 | char *pszBackToUTF8 = |
2113 | 0 | CPLRecode(pszCPFilename, pszDestEncoding, CPL_ENC_UTF8); |
2114 | 0 | if (strcmp(pszBackToUTF8, pszFilename) != 0) |
2115 | 0 | { |
2116 | | // If the UTF-8 name cannot be properly encoded to CPL_ZIP_ENCODING, |
2117 | | // then generate an ASCII name for it where non-ASCII characters |
2118 | | // are replaced by an hexadecimal representation |
2119 | 0 | std::string s; |
2120 | 0 | for (int i = 0; pszFilename[i]; ++i) |
2121 | 0 | { |
2122 | 0 | if (static_cast<unsigned char>(pszFilename[i]) <= 127) |
2123 | 0 | { |
2124 | 0 | s += pszFilename[i]; |
2125 | 0 | } |
2126 | 0 | else |
2127 | 0 | { |
2128 | 0 | s += CPLSPrintf("0x%02X", static_cast<unsigned char>( |
2129 | 0 | pszFilename[i])); |
2130 | 0 | } |
2131 | 0 | } |
2132 | 0 | CPLFree(pszCPFilename); |
2133 | 0 | pszCPFilename = CPLStrdup(s.c_str()); |
2134 | 0 | } |
2135 | 0 | CPLFree(pszBackToUTF8); |
2136 | 0 | } |
2137 | | |
2138 | | /* Create a Info-ZIP Unicode Path Extra Field (0x7075) */ |
2139 | 0 | const size_t nDataLength = |
2140 | 0 | sizeof(GByte) + sizeof(uint32_t) + strlen(pszFilename); |
2141 | 0 | if (abyExtra.size() + 2 * sizeof(uint16_t) + nDataLength > |
2142 | 0 | std::numeric_limits<uint16_t>::max()) |
2143 | 0 | { |
2144 | 0 | CPLError(CE_Warning, CPLE_AppDefined, |
2145 | 0 | "Too much content to fit in ZIP ExtraField"); |
2146 | 0 | } |
2147 | 0 | else |
2148 | 0 | { |
2149 | 0 | const uint16_t nHeaderIdLE = CPL_LSBWORD16(0x7075); |
2150 | 0 | abyExtra.insert(abyExtra.end(), |
2151 | 0 | reinterpret_cast<const GByte *>(&nHeaderIdLE), |
2152 | 0 | reinterpret_cast<const GByte *>(&nHeaderIdLE) + 2); |
2153 | 0 | const uint16_t nDataLengthLE = |
2154 | 0 | CPL_LSBWORD16(static_cast<uint16_t>(nDataLength)); |
2155 | 0 | abyExtra.insert( |
2156 | 0 | abyExtra.end(), reinterpret_cast<const GByte *>(&nDataLengthLE), |
2157 | 0 | reinterpret_cast<const GByte *>(&nDataLengthLE) + 2); |
2158 | 0 | const GByte nVersion = 1; |
2159 | 0 | abyExtra.push_back(nVersion); |
2160 | 0 | const uint32_t nNameCRC32 = static_cast<uint32_t>( |
2161 | 0 | crc32(0, reinterpret_cast<const Bytef *>(pszCPFilename), |
2162 | 0 | static_cast<uInt>(strlen(pszCPFilename)))); |
2163 | 0 | const uint32_t nNameCRC32LE = CPL_LSBWORD32(nNameCRC32); |
2164 | 0 | abyExtra.insert(abyExtra.end(), |
2165 | 0 | reinterpret_cast<const GByte *>(&nNameCRC32LE), |
2166 | 0 | reinterpret_cast<const GByte *>(&nNameCRC32LE) + 4); |
2167 | 0 | abyExtra.insert(abyExtra.end(), |
2168 | 0 | reinterpret_cast<const GByte *>(pszFilename), |
2169 | 0 | reinterpret_cast<const GByte *>(pszFilename) + |
2170 | 0 | strlen(pszFilename)); |
2171 | 0 | } |
2172 | 0 | } |
2173 | 0 | else |
2174 | 0 | { |
2175 | 0 | pszCPFilename = CPLStrdup(pszFilename); |
2176 | 0 | } |
2177 | |
|
2178 | 0 | const char *pszContentType = |
2179 | 0 | CSLFetchNameValue(papszOptions, "CONTENT_TYPE"); |
2180 | 0 | if (pszContentType) |
2181 | 0 | { |
2182 | 0 | const size_t nDataLength = strlen("KeyValuePairs") + sizeof(GByte) + |
2183 | 0 | sizeof(uint16_t) + strlen("Content-Type") + |
2184 | 0 | sizeof(uint16_t) + strlen(pszContentType); |
2185 | 0 | if (abyExtra.size() + 2 * sizeof(uint16_t) + nDataLength > |
2186 | 0 | std::numeric_limits<uint16_t>::max()) |
2187 | 0 | { |
2188 | 0 | CPLError(CE_Warning, CPLE_AppDefined, |
2189 | 0 | "Too much content to fit in ZIP ExtraField"); |
2190 | 0 | } |
2191 | 0 | else |
2192 | 0 | { |
2193 | 0 | abyExtra.push_back(GByte('K')); |
2194 | 0 | abyExtra.push_back(GByte('V')); |
2195 | 0 | const uint16_t nDataLengthLE = |
2196 | 0 | CPL_LSBWORD16(static_cast<uint16_t>(nDataLength)); |
2197 | 0 | abyExtra.insert( |
2198 | 0 | abyExtra.end(), reinterpret_cast<const GByte *>(&nDataLengthLE), |
2199 | 0 | reinterpret_cast<const GByte *>(&nDataLengthLE) + 2); |
2200 | 0 | abyExtra.insert(abyExtra.end(), |
2201 | 0 | reinterpret_cast<const GByte *>("KeyValuePairs"), |
2202 | 0 | reinterpret_cast<const GByte *>("KeyValuePairs") + |
2203 | 0 | strlen("KeyValuePairs")); |
2204 | 0 | abyExtra.push_back(1); // number of key/value pairs |
2205 | 0 | const uint16_t nKeyLen = |
2206 | 0 | CPL_LSBWORD16(static_cast<uint16_t>(strlen("Content-Type"))); |
2207 | 0 | abyExtra.insert(abyExtra.end(), |
2208 | 0 | reinterpret_cast<const GByte *>(&nKeyLen), |
2209 | 0 | reinterpret_cast<const GByte *>(&nKeyLen) + 2); |
2210 | 0 | abyExtra.insert(abyExtra.end(), |
2211 | 0 | reinterpret_cast<const GByte *>("Content-Type"), |
2212 | 0 | reinterpret_cast<const GByte *>("Content-Type") + |
2213 | 0 | strlen("Content-Type")); |
2214 | 0 | const uint16_t nValLen = |
2215 | 0 | CPL_LSBWORD16(static_cast<uint16_t>(strlen(pszContentType))); |
2216 | 0 | abyExtra.insert(abyExtra.end(), |
2217 | 0 | reinterpret_cast<const GByte *>(&nValLen), |
2218 | 0 | reinterpret_cast<const GByte *>(&nValLen) + 2); |
2219 | 0 | abyExtra.insert(abyExtra.end(), |
2220 | 0 | reinterpret_cast<const GByte *>(pszContentType), |
2221 | 0 | reinterpret_cast<const GByte *>( |
2222 | 0 | pszContentType + strlen(pszContentType))); |
2223 | 0 | } |
2224 | 0 | } |
2225 | |
|
2226 | 0 | const bool bIncludeInCentralDirectory = CPLTestBool(CSLFetchNameValueDef( |
2227 | 0 | papszOptions, "INCLUDE_IN_CENTRAL_DIRECTORY", "YES")); |
2228 | 0 | const bool bZip64 = CPLTestBool(CSLFetchNameValueDef( |
2229 | 0 | papszOptions, "ZIP64", CPLGetConfigOption("CPL_CREATE_ZIP64", "ON"))); |
2230 | | |
2231 | | // Set datetime to write |
2232 | 0 | zip_fileinfo fileinfo; |
2233 | 0 | memset(&fileinfo, 0, sizeof(fileinfo)); |
2234 | 0 | const char *pszTimeStamp = |
2235 | 0 | CSLFetchNameValueDef(papszOptions, "TIMESTAMP", "NOW"); |
2236 | 0 | GIntBig unixTime = |
2237 | 0 | EQUAL(pszTimeStamp, "NOW") |
2238 | 0 | ? time(nullptr) |
2239 | 0 | : static_cast<GIntBig>(std::strtoll(pszTimeStamp, nullptr, 10)); |
2240 | 0 | struct tm brokenDown; |
2241 | 0 | CPLUnixTimeToYMDHMS(unixTime, &brokenDown); |
2242 | 0 | fileinfo.tmz_date.tm_year = brokenDown.tm_year; |
2243 | 0 | fileinfo.tmz_date.tm_mon = brokenDown.tm_mon; |
2244 | 0 | fileinfo.tmz_date.tm_mday = brokenDown.tm_mday; |
2245 | 0 | fileinfo.tmz_date.tm_hour = brokenDown.tm_hour; |
2246 | 0 | fileinfo.tmz_date.tm_min = brokenDown.tm_min; |
2247 | 0 | fileinfo.tmz_date.tm_sec = brokenDown.tm_sec; |
2248 | |
|
2249 | 0 | const int nErr = cpl_zipOpenNewFileInZip3( |
2250 | 0 | psZip->hZip, pszCPFilename, &fileinfo, |
2251 | 0 | abyExtra.empty() ? nullptr : abyExtra.data(), |
2252 | 0 | static_cast<uInt>(abyExtra.size()), |
2253 | 0 | abyExtra.empty() ? nullptr : abyExtra.data(), |
2254 | 0 | static_cast<uInt>(abyExtra.size()), "", bCompressed ? Z_DEFLATED : 0, |
2255 | 0 | bCompressed ? Z_DEFAULT_COMPRESSION : 0, |
2256 | 0 | /* raw = */ 0, -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY, |
2257 | 0 | /* password = */ nullptr, |
2258 | 0 | /* crcForCtypting = */ 0, bZip64, bIncludeInCentralDirectory); |
2259 | |
|
2260 | 0 | CPLFree(pszCPFilename); |
2261 | |
|
2262 | 0 | if (nErr != ZIP_OK) |
2263 | 0 | return CE_Failure; |
2264 | | |
2265 | 0 | if (bIncludeInCentralDirectory) |
2266 | 0 | psZip->papszFilenames = |
2267 | 0 | CSLAddString(psZip->papszFilenames, pszFilename); |
2268 | |
|
2269 | 0 | return CE_None; |
2270 | 0 | } |
2271 | | |
2272 | | /************************************************************************/ |
2273 | | /* CPLWriteFileInZip() */ |
2274 | | /************************************************************************/ |
2275 | | |
2276 | | /** Write in current file inside a ZIP file */ |
2277 | | CPLErr CPLWriteFileInZip(void *hZip, const void *pBuffer, int nBufferSize) |
2278 | | |
2279 | 0 | { |
2280 | 0 | if (hZip == nullptr) |
2281 | 0 | return CE_Failure; |
2282 | | |
2283 | 0 | CPLZip *psZip = static_cast<CPLZip *>(hZip); |
2284 | |
|
2285 | 0 | int nErr = cpl_zipWriteInFileInZip(psZip->hZip, pBuffer, |
2286 | 0 | static_cast<unsigned int>(nBufferSize)); |
2287 | |
|
2288 | 0 | if (nErr != ZIP_OK) |
2289 | 0 | return CE_Failure; |
2290 | | |
2291 | 0 | return CE_None; |
2292 | 0 | } |
2293 | | |
2294 | | /************************************************************************/ |
2295 | | /* CPLCloseFileInZip() */ |
2296 | | /************************************************************************/ |
2297 | | |
2298 | | /** Close current file inside ZIP file */ |
2299 | | CPLErr CPLCloseFileInZip(void *hZip) |
2300 | | |
2301 | 0 | { |
2302 | 0 | if (hZip == nullptr) |
2303 | 0 | return CE_Failure; |
2304 | | |
2305 | 0 | CPLZip *psZip = static_cast<CPLZip *>(hZip); |
2306 | |
|
2307 | 0 | int nErr = cpl_zipCloseFileInZip(psZip->hZip); |
2308 | |
|
2309 | 0 | if (nErr != ZIP_OK) |
2310 | 0 | return CE_Failure; |
2311 | | |
2312 | 0 | return CE_None; |
2313 | 0 | } |
2314 | | |
2315 | | /************************************************************************/ |
2316 | | /* CPLAddFileInZip() */ |
2317 | | /************************************************************************/ |
2318 | | |
2319 | | /** Add a file inside a ZIP file opened/created with CPLCreateZip(). |
2320 | | * |
2321 | | * This combines calls to CPLCreateFileInZip(), CPLWriteFileInZip(), |
2322 | | * and CPLCloseFileInZip() in a more convenient and powerful way. |
2323 | | * |
2324 | | * In particular, this enables to add a compressed file using the seek |
2325 | | * optimization extension. |
2326 | | * |
2327 | | * Supported options are: |
2328 | | * <ul> |
2329 | | * <li>SOZIP_ENABLED=AUTO/YES/NO: whether to generate a SOZip index for the |
2330 | | * file. The default can be changed with the CPL_SOZIP_ENABLED configuration |
2331 | | * option.</li> |
2332 | | * <li>SOZIP_CHUNK_SIZE: chunk size to use for SOZip generation. Defaults to |
2333 | | * 32768. |
2334 | | * </li> |
2335 | | * <li>SOZIP_MIN_FILE_SIZE: minimum file size to consider to enable SOZip index |
2336 | | * generation in SOZIP_ENABLED=AUTO mode. Defaults to 1 MB. |
2337 | | * </li> |
2338 | | * <li>NUM_THREADS: number of threads used for SOZip generation. Defaults to |
2339 | | * ALL_CPUS.</li> |
2340 | | * <li>TIMESTAMP=AUTO/NOW/timestamp_as_epoch_since_jan_1_1970: in AUTO mode, |
2341 | | * the timestamp of pszInputFilename will be used (if available), otherwise |
2342 | | * it will fallback to NOW.</li> |
2343 | | * <li>CONTENT_TYPE=string: Content-Type value for the file. This is stored as |
2344 | | * a key-value pair in the extra field extension 'KV' (0x564b) dedicated to |
2345 | | * storing key-value pair metadata.</li> |
2346 | | * </ul> |
2347 | | * |
2348 | | * @param hZip ZIP file handle |
2349 | | * @param pszArchiveFilename Filename (in UTF-8) stored in the archive. |
2350 | | * @param pszInputFilename Filename of the file to add. If NULL, fpInput must |
2351 | | * not be NULL |
2352 | | * @param fpInput File handle opened on the file to add. May be NULL if |
2353 | | * pszInputFilename is provided. |
2354 | | * @param papszOptions Options. |
2355 | | * @param pProgressFunc Progress callback, or NULL. |
2356 | | * @param pProgressData User data of progress callback, or NULL. |
2357 | | * @return CE_None in case of success. |
2358 | | * |
2359 | | * @since GDAL 3.7 |
2360 | | */ |
2361 | | CPLErr CPLAddFileInZip(void *hZip, const char *pszArchiveFilename, |
2362 | | const char *pszInputFilename, VSILFILE *fpInput, |
2363 | | CSLConstList papszOptions, |
2364 | | GDALProgressFunc pProgressFunc, void *pProgressData) |
2365 | 0 | { |
2366 | 0 | if (!hZip || !pszArchiveFilename || (!pszInputFilename && !fpInput)) |
2367 | 0 | return CE_Failure; |
2368 | | |
2369 | 0 | CPLZip *psZip = static_cast<CPLZip *>(hZip); |
2370 | 0 | zip64_internal *zi = reinterpret_cast<zip64_internal *>(psZip->hZip); |
2371 | |
|
2372 | 0 | VSIVirtualHandleUniquePtr poFileHandleAutoClose; |
2373 | 0 | if (!fpInput) |
2374 | 0 | { |
2375 | 0 | fpInput = VSIFOpenL(pszInputFilename, "rb"); |
2376 | 0 | if (!fpInput) |
2377 | 0 | return CE_Failure; |
2378 | 0 | poFileHandleAutoClose.reset(fpInput); |
2379 | 0 | } |
2380 | | |
2381 | 0 | VSIFSeekL(fpInput, 0, SEEK_END); |
2382 | 0 | const auto nUncompressedSize = VSIFTellL(fpInput); |
2383 | 0 | VSIFSeekL(fpInput, 0, SEEK_SET); |
2384 | |
|
2385 | 0 | CPLStringList aosNewsOptions(papszOptions); |
2386 | 0 | bool bSeekOptimized = false; |
2387 | 0 | const char *pszSOZIP = |
2388 | 0 | CSLFetchNameValueDef(papszOptions, "SOZIP_ENABLED", |
2389 | 0 | CPLGetConfigOption("CPL_SOZIP_ENABLED", "AUTO")); |
2390 | |
|
2391 | 0 | const char *pszChunkSize = CSLFetchNameValueDef( |
2392 | 0 | papszOptions, "SOZIP_CHUNK_SIZE", |
2393 | 0 | CPLGetConfigOption("CPL_VSIL_DEFLATE_CHUNK_SIZE", nullptr)); |
2394 | 0 | const bool bChunkSizeSpecified = pszChunkSize != nullptr; |
2395 | 0 | if (!pszChunkSize) |
2396 | 0 | pszChunkSize = "1024K"; |
2397 | 0 | unsigned nChunkSize = static_cast<unsigned>(atoi(pszChunkSize)); |
2398 | 0 | if (strchr(pszChunkSize, 'K')) |
2399 | 0 | nChunkSize *= 1024; |
2400 | 0 | else if (strchr(pszChunkSize, 'M')) |
2401 | 0 | nChunkSize *= 1024 * 1024; |
2402 | 0 | nChunkSize = |
2403 | 0 | std::max(static_cast<unsigned>(1), |
2404 | 0 | std::min(static_cast<unsigned>(UINT_MAX), nChunkSize)); |
2405 | |
|
2406 | 0 | const char *pszMinFileSize = CSLFetchNameValueDef( |
2407 | 0 | papszOptions, "SOZIP_MIN_FILE_SIZE", |
2408 | 0 | CPLGetConfigOption("CPL_SOZIP_MIN_FILE_SIZE", "1M")); |
2409 | 0 | uint64_t nSOZipMinFileSize = std::strtoull(pszMinFileSize, nullptr, 10); |
2410 | 0 | if (strchr(pszMinFileSize, 'K')) |
2411 | 0 | nSOZipMinFileSize *= 1024; |
2412 | 0 | else if (strchr(pszMinFileSize, 'M')) |
2413 | 0 | nSOZipMinFileSize *= 1024 * 1024; |
2414 | 0 | else if (strchr(pszMinFileSize, 'G')) |
2415 | 0 | nSOZipMinFileSize *= 1024 * 1024 * 1024; |
2416 | |
|
2417 | 0 | std::vector<uint8_t> sozip_index; |
2418 | 0 | uint64_t nExpectedIndexSize = 0; |
2419 | 0 | constexpr unsigned nDefaultSOZipChunkSize = 32 * 1024; |
2420 | 0 | constexpr size_t nOffsetSize = 8; |
2421 | 0 | if (((EQUAL(pszSOZIP, "AUTO") && nUncompressedSize > nSOZipMinFileSize) || |
2422 | 0 | (!EQUAL(pszSOZIP, "AUTO") && CPLTestBool(pszSOZIP))) && |
2423 | 0 | ((bChunkSizeSpecified && |
2424 | 0 | nUncompressedSize > static_cast<unsigned>(nChunkSize)) || |
2425 | 0 | (!bChunkSizeSpecified && nUncompressedSize > nDefaultSOZipChunkSize))) |
2426 | 0 | { |
2427 | 0 | if (!bChunkSizeSpecified) |
2428 | 0 | nChunkSize = nDefaultSOZipChunkSize; |
2429 | |
|
2430 | 0 | bSeekOptimized = true; |
2431 | |
|
2432 | 0 | aosNewsOptions.SetNameValue( |
2433 | 0 | "UNCOMPRESSED_SIZE", CPLSPrintf(CPL_FRMT_GUIB, nUncompressedSize)); |
2434 | |
|
2435 | 0 | zi->nOffsetSize = nOffsetSize; |
2436 | 0 | nExpectedIndexSize = |
2437 | 0 | 32 + ((nUncompressedSize - 1) / nChunkSize) * nOffsetSize; |
2438 | 0 | if (nExpectedIndexSize > |
2439 | 0 | static_cast<uint64_t>(std::numeric_limits<int>::max())) |
2440 | 0 | { |
2441 | 0 | CPLError(CE_Failure, CPLE_AppDefined, |
2442 | 0 | "Too big file w.r.t CHUNK_SIZE"); |
2443 | 0 | return CE_Failure; |
2444 | 0 | } |
2445 | 0 | try |
2446 | 0 | { |
2447 | 0 | sozip_index.reserve(static_cast<size_t>(nExpectedIndexSize)); |
2448 | 0 | } |
2449 | 0 | catch (const std::exception &) |
2450 | 0 | { |
2451 | 0 | CPLError(CE_Failure, CPLE_OutOfMemory, |
2452 | 0 | "Cannot allocate memory for SOZip index"); |
2453 | 0 | return CE_Failure; |
2454 | 0 | } |
2455 | 0 | sozip_index.resize(32); |
2456 | 0 | uint32_t nVal32; |
2457 | | // Version |
2458 | 0 | nVal32 = CPL_LSBWORD32(1); |
2459 | 0 | memcpy(sozip_index.data(), &nVal32, sizeof(nVal32)); |
2460 | | // Extra reserved space after 32 bytes of header |
2461 | 0 | nVal32 = CPL_LSBWORD32(0); |
2462 | 0 | memcpy(sozip_index.data() + 4, &nVal32, sizeof(nVal32)); |
2463 | | // Chunksize |
2464 | 0 | nVal32 = CPL_LSBWORD32(nChunkSize); |
2465 | 0 | memcpy(sozip_index.data() + 8, &nVal32, sizeof(nVal32)); |
2466 | | // SOZIPIndexEltSize |
2467 | 0 | nVal32 = CPL_LSBWORD32(static_cast<uint32_t>(nOffsetSize)); |
2468 | 0 | memcpy(sozip_index.data() + 12, &nVal32, sizeof(nVal32)); |
2469 | | // Uncompressed size |
2470 | 0 | uint64_t nVal64 = nUncompressedSize; |
2471 | 0 | CPL_LSBPTR64(&nVal64); |
2472 | 0 | memcpy(sozip_index.data() + 16, &nVal64, sizeof(nVal64)); |
2473 | 0 | zi->sozip_index = &sozip_index; |
2474 | |
|
2475 | 0 | zi->nChunkSize = nChunkSize; |
2476 | |
|
2477 | 0 | const char *pszThreads = CSLFetchNameValue(papszOptions, "NUM_THREADS"); |
2478 | 0 | if (pszThreads == nullptr || EQUAL(pszThreads, "ALL_CPUS")) |
2479 | 0 | zi->nThreads = CPLGetNumCPUs(); |
2480 | 0 | else |
2481 | 0 | zi->nThreads = atoi(pszThreads); |
2482 | 0 | zi->nThreads = std::max(1, std::min(128, zi->nThreads)); |
2483 | 0 | } |
2484 | | |
2485 | 0 | aosNewsOptions.SetNameValue("ZIP64", |
2486 | 0 | nUncompressedSize > 0xFFFFFFFFU ? "YES" : "NO"); |
2487 | |
|
2488 | 0 | if (pszInputFilename != nullptr && |
2489 | 0 | aosNewsOptions.FetchNameValue("TIMESTAMP") == nullptr) |
2490 | 0 | { |
2491 | 0 | VSIStatBufL sStat; |
2492 | 0 | if (VSIStatL(pszInputFilename, &sStat) == 0 && sStat.st_mtime != 0) |
2493 | 0 | { |
2494 | 0 | aosNewsOptions.SetNameValue( |
2495 | 0 | "TIMESTAMP", |
2496 | 0 | CPLSPrintf(CPL_FRMT_GIB, static_cast<GIntBig>(sStat.st_mtime))); |
2497 | 0 | } |
2498 | 0 | } |
2499 | |
|
2500 | 0 | if (CPLCreateFileInZip(hZip, pszArchiveFilename, aosNewsOptions.List()) != |
2501 | 0 | CE_None) |
2502 | 0 | { |
2503 | 0 | zi->sozip_index = nullptr; |
2504 | 0 | zi->nChunkSize = 0; |
2505 | 0 | zi->nThreads = 0; |
2506 | 0 | return CE_Failure; |
2507 | 0 | } |
2508 | 0 | zi->nChunkSize = 0; |
2509 | 0 | zi->nThreads = 0; |
2510 | |
|
2511 | 0 | constexpr int CHUNK_READ_MAX_SIZE = 1024 * 1024; |
2512 | 0 | std::vector<GByte> abyChunk(CHUNK_READ_MAX_SIZE); |
2513 | 0 | vsi_l_offset nOffset = 0; |
2514 | 0 | while (true) |
2515 | 0 | { |
2516 | 0 | const int nRead = static_cast<int>( |
2517 | 0 | VSIFReadL(abyChunk.data(), 1, abyChunk.size(), fpInput)); |
2518 | 0 | if (nRead > 0 && |
2519 | 0 | CPLWriteFileInZip(hZip, abyChunk.data(), nRead) != CE_None) |
2520 | 0 | { |
2521 | 0 | CPLCloseFileInZip(hZip); |
2522 | 0 | zi->sozip_index = nullptr; |
2523 | 0 | return CE_Failure; |
2524 | 0 | } |
2525 | 0 | nOffset += nRead; |
2526 | 0 | if (pProgressFunc && |
2527 | 0 | !pProgressFunc(nUncompressedSize == 0 |
2528 | 0 | ? 1.0 |
2529 | 0 | : double(nOffset) / nUncompressedSize, |
2530 | 0 | nullptr, pProgressData)) |
2531 | 0 | { |
2532 | 0 | CPLCloseFileInZip(hZip); |
2533 | 0 | zi->sozip_index = nullptr; |
2534 | 0 | return CE_Failure; |
2535 | 0 | } |
2536 | 0 | if (nRead < CHUNK_READ_MAX_SIZE) |
2537 | 0 | break; |
2538 | 0 | } |
2539 | | |
2540 | 0 | if (CPLCloseFileInZip(hZip) != CE_None) |
2541 | 0 | { |
2542 | 0 | zi->sozip_index = nullptr; |
2543 | 0 | return CE_Failure; |
2544 | 0 | } |
2545 | | |
2546 | 0 | if (bSeekOptimized && sozip_index.size() != nExpectedIndexSize) |
2547 | 0 | { |
2548 | | // shouldn't happen |
2549 | 0 | CPLError(CE_Failure, CPLE_AppDefined, |
2550 | 0 | "sozip_index.size() (=%u) != nExpectedIndexSize (=%u)", |
2551 | 0 | static_cast<unsigned>(sozip_index.size()), |
2552 | 0 | static_cast<unsigned>(nExpectedIndexSize)); |
2553 | 0 | } |
2554 | 0 | else if (bSeekOptimized) |
2555 | 0 | { |
2556 | 0 | std::string osIdxName; |
2557 | 0 | const char *pszLastSlash = strchr(pszArchiveFilename, '/'); |
2558 | 0 | if (pszLastSlash) |
2559 | 0 | { |
2560 | 0 | osIdxName.assign(pszArchiveFilename, |
2561 | 0 | pszLastSlash - pszArchiveFilename + 1); |
2562 | 0 | osIdxName += '.'; |
2563 | 0 | osIdxName += pszLastSlash + 1; |
2564 | 0 | } |
2565 | 0 | else |
2566 | 0 | { |
2567 | 0 | osIdxName = '.'; |
2568 | 0 | osIdxName += pszArchiveFilename; |
2569 | 0 | } |
2570 | 0 | osIdxName += ".sozip.idx"; |
2571 | |
|
2572 | 0 | CPLStringList aosIndexOptions; |
2573 | 0 | aosIndexOptions.SetNameValue("COMPRESSED", "NO"); |
2574 | 0 | aosIndexOptions.SetNameValue("ZIP64", "NO"); |
2575 | 0 | aosIndexOptions.SetNameValue("INCLUDE_IN_CENTRAL_DIRECTORY", "NO"); |
2576 | 0 | aosIndexOptions.SetNameValue( |
2577 | 0 | "TIMESTAMP", aosNewsOptions.FetchNameValue("TIMESTAMP")); |
2578 | 0 | if (CPLCreateFileInZip(hZip, osIdxName.c_str(), |
2579 | 0 | aosIndexOptions.List()) != CE_None) |
2580 | 0 | { |
2581 | 0 | zi->sozip_index = nullptr; |
2582 | 0 | return CE_Failure; |
2583 | 0 | } |
2584 | | |
2585 | 0 | if (CPLWriteFileInZip(hZip, sozip_index.data(), |
2586 | 0 | static_cast<int>(sozip_index.size())) != CE_None) |
2587 | 0 | { |
2588 | 0 | zi->sozip_index = nullptr; |
2589 | 0 | CPLCloseFileInZip(hZip); |
2590 | 0 | return CE_Failure; |
2591 | 0 | } |
2592 | | |
2593 | 0 | zi->sozip_index = nullptr; |
2594 | 0 | if (CPLCloseFileInZip(hZip) != CE_None) |
2595 | 0 | { |
2596 | 0 | return CE_Failure; |
2597 | 0 | } |
2598 | 0 | } |
2599 | | |
2600 | 0 | zi->sozip_index = nullptr; |
2601 | |
|
2602 | 0 | return CE_None; |
2603 | 0 | } |
2604 | | |
2605 | | /************************************************************************/ |
2606 | | /* CPLCloseZip() */ |
2607 | | /************************************************************************/ |
2608 | | |
2609 | | /** Close ZIP file */ |
2610 | | CPLErr CPLCloseZip(void *hZip) |
2611 | 0 | { |
2612 | 0 | if (hZip == nullptr) |
2613 | 0 | return CE_Failure; |
2614 | | |
2615 | 0 | CPLZip *psZip = static_cast<CPLZip *>(hZip); |
2616 | |
|
2617 | 0 | int nErr = cpl_zipClose(psZip->hZip, nullptr); |
2618 | |
|
2619 | 0 | psZip->hZip = nullptr; |
2620 | 0 | CSLDestroy(psZip->papszFilenames); |
2621 | 0 | psZip->papszFilenames = nullptr; |
2622 | 0 | CPLFree(psZip); |
2623 | |
|
2624 | 0 | if (nErr != ZIP_OK) |
2625 | 0 | return CE_Failure; |
2626 | | |
2627 | 0 | return CE_None; |
2628 | 0 | } |