Coverage Report

Created: 2025-06-13 06:18

/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
}