/
zip.c
678 lines (545 loc) · 18.5 KB
1
2
3
4
5
6
7
8
/*
* ZIP support routines for PhysicsFS.
*
* Please see the file LICENSE in the source's root directory.
*
* This file written by Ryan C. Gordon.
*/
9
10
11
12
13
/*
* !!! FIXME: overall design bugs.
*
* Maybe add a seekToStartOfCurrentFile() in unzip.c if complete seek
* semantics are impossible.
14
15
16
*
* Could be more i/o efficient if we combined unzip.c and this file.
* (and thus lose all the unzGoToNextFile() dummy loops.
17
*/
18
19
20
#include <stdio.h>
#include <stdlib.h>
21
#include <string.h>
22
#include <assert.h>
23
#include "physfs.h"
24
25
#include "unzip.h"
26
27
28
29
30
31
32
33
#define __PHYSICSFS_INTERNAL__
#include "physfs_internal.h"
#if (!defined PHYSFS_SUPPORTS_ZIP)
#error PHYSFS_SUPPORTS_ZIP must be defined.
#endif
34
#define MAXZIPENTRYSIZE 256
35
36
37
typedef struct
{
38
39
40
41
42
43
44
char *name;
unz_file_info info;
char *symlink;
} ZIPentry;
typedef struct
{
45
char *archiveName;
46
47
unz_global_info global;
ZIPentry *entries;
48
49
50
51
} ZIPinfo;
typedef struct
{
52
unzFile handle;
53
54
55
} ZIPfileinfo;
56
57
58
59
/* Number of symlinks to follow before we assume it's a recursive link... */
#define SYMLINK_RECURSE_COUNT 20
60
61
static PHYSFS_sint64 ZIP_read(FileHandle *handle, void *buffer,
PHYSFS_uint32 objSize, PHYSFS_uint32 objCount);
62
static int ZIP_eof(FileHandle *handle);
63
64
65
static PHYSFS_sint64 ZIP_tell(FileHandle *handle);
static int ZIP_seek(FileHandle *handle, PHYSFS_uint64 offset);
static PHYSFS_sint64 ZIP_fileLength(FileHandle *handle);
66
67
static int ZIP_fileClose(FileHandle *handle);
static int ZIP_isArchive(const char *filename, int forWriting);
68
static char *ZIP_realpath(unzFile fh, unz_file_info *info, ZIPentry *entry);
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
static DirHandle *ZIP_openArchive(const char *name, int forWriting);
static LinkedStringList *ZIP_enumerateFiles(DirHandle *h,
const char *dirname,
int omitSymLinks);
static int ZIP_exists(DirHandle *h, const char *name);
static int ZIP_isDirectory(DirHandle *h, const char *name);
static int ZIP_isSymLink(DirHandle *h, const char *name);
static FileHandle *ZIP_openRead(DirHandle *h, const char *filename);
static void ZIP_dirClose(DirHandle *h);
static const FileFunctions __PHYSFS_FileFunctions_ZIP =
{
ZIP_read, /* read() method */
NULL, /* write() method */
ZIP_eof, /* eof() method */
ZIP_tell, /* tell() method */
ZIP_seek, /* seek() method */
ZIP_fileLength, /* fileLength() method */
ZIP_fileClose /* fileClose() method */
};
const DirFunctions __PHYSFS_DirFunctions_ZIP =
{
ZIP_isArchive, /* isArchive() method */
ZIP_openArchive, /* openArchive() method */
ZIP_enumerateFiles, /* enumerateFiles() method */
ZIP_exists, /* exists() method */
ZIP_isDirectory, /* isDirectory() method */
ZIP_isSymLink, /* isSymLink() method */
ZIP_openRead, /* openRead() method */
NULL, /* openWrite() method */
NULL, /* openAppend() method */
NULL, /* remove() method */
NULL, /* mkdir() method */
ZIP_dirClose /* dirClose() method */
};
const PHYSFS_ArchiveInfo __PHYSFS_ArchiveInfo_ZIP =
{
"ZIP",
"PkZip/WinZip/Info-Zip compatible",
112
"Ryan C. Gordon <icculus@clutteredmind.org>",
113
114
115
116
117
"http://www.icculus.org/physfs/",
};
118
119
static PHYSFS_sint64 ZIP_read(FileHandle *handle, void *buffer,
PHYSFS_uint32 objSize, PHYSFS_uint32 objCount)
120
{
121
unzFile fh = ((ZIPfileinfo *) (handle->opaque))->handle;
122
123
int bytes = (int) (objSize * objCount); /* !!! FIXME: overflow? */
PHYSFS_sint32 rc = unzReadCurrentFile(fh, buffer, bytes);
124
125
126
127
128
129
130
131
132
if (rc < bytes)
__PHYSFS_setError(ERR_PAST_EOF);
else if (rc == UNZ_ERRNO)
__PHYSFS_setError(ERR_IO_ERROR);
else if (rc < 0)
__PHYSFS_setError(ERR_COMPRESSION);
return(rc / objSize);
133
134
135
136
137
} /* ZIP_read */
static int ZIP_eof(FileHandle *handle)
{
138
return(unzeof(((ZIPfileinfo *) (handle->opaque))->handle));
139
140
141
} /* ZIP_eof */
142
static PHYSFS_sint64 ZIP_tell(FileHandle *handle)
143
{
144
return(unztell(((ZIPfileinfo *) (handle->opaque))->handle));
145
146
147
} /* ZIP_tell */
148
static int ZIP_seek(FileHandle *handle, PHYSFS_uint64 offset)
149
{
150
/* !!! FIXME : this blows. */
151
unzFile fh = ((ZIPfileinfo *) (handle->opaque))->handle;
152
char *buf = NULL;
153
PHYSFS_uint32 bufsize = 4096 * 2;
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
BAIL_IF_MACRO(unztell(fh) == offset, NULL, 1);
BAIL_IF_MACRO(ZIP_fileLength(handle) <= offset, ERR_PAST_EOF, 0);
/* reset to the start of the zipfile. */
unzCloseCurrentFile(fh);
BAIL_IF_MACRO(unzOpenCurrentFile(fh) != UNZ_OK, ERR_IO_ERROR, 0);
while ((buf == NULL) && (bufsize >= 512))
{
bufsize >>= 1; /* divides by two. */
buf = (char *) malloc(bufsize);
} /* while */
BAIL_IF_MACRO(buf == NULL, ERR_OUT_OF_MEMORY, 0);
while (offset > 0)
{
171
172
173
/* !!! - RYAN, CHECK THIS CAST */
/* !!! This should be okay since offset will be <= bufsize */
PHYSFS_uint32 chunk = (offset > bufsize) ? bufsize : (PHYSFS_uint32)offset;
174
PHYSFS_sint32 rc = unzReadCurrentFile(fh, buf, chunk);
175
176
177
178
179
180
181
182
BAIL_IF_MACRO(rc == 0, ERR_IO_ERROR, 0); /* shouldn't happen. */
BAIL_IF_MACRO(rc == UNZ_ERRNO, ERR_IO_ERROR, 0);
BAIL_IF_MACRO(rc < 0, ERR_COMPRESSION, 0);
offset -= rc;
} /* while */
free(buf);
return(offset == 0);
183
184
185
} /* ZIP_seek */
186
static PHYSFS_sint64 ZIP_fileLength(FileHandle *handle)
187
{
188
189
190
191
192
ZIPfileinfo *finfo = (ZIPfileinfo *) (handle->opaque);
unz_file_info info;
unzGetCurrentFileInfo(finfo->handle, &info, NULL, 0, NULL, 0, NULL, 0);
return(info.uncompressed_size);
193
194
195
} /* ZIP_fileLength */
196
197
static int ZIP_fileClose(FileHandle *handle)
{
198
199
200
201
202
ZIPfileinfo *finfo = (ZIPfileinfo *) (handle->opaque);
unzClose(finfo->handle);
free(finfo);
free(handle);
return(1);
203
204
205
206
207
} /* ZIP_fileClose */
static int ZIP_isArchive(const char *filename, int forWriting)
{
208
int retval = 0;
209
unzFile unz = unzOpen(filename);
210
211
212
213
214
215
216
217
218
219
unz_global_info global;
if (unz != NULL)
{
if (unzGetGlobalInfo(unz, &global) == UNZ_OK)
retval = 1;
unzClose(unz);
} /* if */
return(retval);
220
221
222
} /* ZIP_isArchive */
223
static void freeEntries(ZIPinfo *info, int count, const char *errmsg)
224
{
225
int i;
226
227
228
229
230
231
232
for (i = 0; i < count; i++)
{
free(info->entries[i].name);
if (info->entries[i].symlink != NULL)
free(info->entries[i].symlink);
} /* for */
233
234
free(info->entries);
235
236
237
238
if (errmsg != NULL)
__PHYSFS_setError(errmsg);
} /* freeEntries */
239
240
241
242
243
244
245
246
247
/*
* !!! FIXME: Really implement this.
* !!! FIXME: symlinks in zipfiles can be relative paths, including
* !!! FIXME: "." and ".." entries. These need to be parsed out.
* !!! FIXME: For now, though, we're just copying the relative path. Oh well.
*/
static char *expand_symlink_path(const char *path, ZIPentry *entry)
248
{
249
char *retval = (char *) malloc(strlen(path) + 1);
250
BAIL_IF_MACRO(retval == NULL, ERR_OUT_OF_MEMORY, NULL);
251
strcpy(retval, path);
252
return(retval);
253
} /* expand_symlink_path */
254
255
256
static char *ZIP_realpath(unzFile fh, unz_file_info *info, ZIPentry *entry)
257
{
258
259
260
261
262
263
264
265
266
267
268
269
270
char path[MAXZIPENTRYSIZE];
int size = info->uncompressed_size;
int rc;
BAIL_IF_MACRO(size >= sizeof (path), ERR_IO_ERROR, NULL);
BAIL_IF_MACRO(unzOpenCurrentFile(fh) != UNZ_OK, ERR_IO_ERROR, NULL);
rc = unzReadCurrentFile(fh, path, size);
unzCloseCurrentFile(fh);
BAIL_IF_MACRO(rc != size, ERR_IO_ERROR, NULL);
path[size] = '\0'; /* null terminate it. */
return(expand_symlink_path(path, entry)); /* retval is malloc()'d. */
} /* ZIP_realpath */
271
272
273
274
275
static int version_does_symlinks(uLong version)
{
int retval = 0;
276
277
278
/* !!! - RYAN, CHECK THIS CAST */
/* !!! - You AND the result with 0xFF, so it can't be larger than 0xFF */
unsigned char hosttype = (unsigned char)((version >> 8) & 0xFF);
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
/*
* These are the platforms that can build an archive with symlinks,
* according to the Info-ZIP project.
*/
switch (hosttype)
{
case 3: /* Unix */
case 16: /* BeOS */
case 5: /* Atari */
retval = 1;
break;
} /* switch */
return(retval);
} /* version_does_symlinks */
static int entry_is_symlink(unz_file_info *info)
{
return (
(version_does_symlinks(info->version)) &&
(info->uncompressed_size > 0) &&
(info->external_fa & 0x0120000) /* symlink flag. */
);
} /* entry_is_symlink */
307
static int loadZipEntries(ZIPinfo *info, unzFile unz)
308
{
309
310
311
312
313
int i, max;
BAIL_IF_MACRO(unzGetGlobalInfo(unz, &(info->global)) != UNZ_OK,
ERR_IO_ERROR, 0);
BAIL_IF_MACRO(unzGoToFirstFile(unz) != UNZ_OK, ERR_IO_ERROR, 0);
314
315
316
317
318
319
max = info->global.number_entry;
info->entries = (ZIPentry *) malloc(sizeof (ZIPentry) * max);
BAIL_IF_MACRO(info->entries == NULL, ERR_OUT_OF_MEMORY, 0);
for (i = 0; i < max; i++)
320
{
321
322
unz_file_info *d = &((info->entries[i]).info);
if (unzGetCurrentFileInfo(unz, d, NULL, 0, NULL, 0, NULL, 0) != UNZ_OK)
323
{
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
freeEntries(info, i, ERR_IO_ERROR);
return(0);
} /* if */
(info->entries[i]).name = (char *) malloc(d->size_filename + 1);
if ((info->entries[i]).name == NULL)
{
freeEntries(info, i, ERR_OUT_OF_MEMORY);
return(0);
} /* if */
info->entries[i].symlink = NULL;
if (unzGetCurrentFileInfo(unz, NULL, (info->entries[i]).name,
d->size_filename + 1, NULL, 0,
NULL, 0) != UNZ_OK)
{
freeEntries(info, i + 1, ERR_IO_ERROR);
return(0);
} /* if */
if (entry_is_symlink(d))
{
347
info->entries[i].symlink = ZIP_realpath(unz, d, &info->entries[i]);
348
349
350
351
352
if (info->entries[i].symlink == NULL)
{
freeEntries(info, i + 1, NULL);
return(0);
} /* if */
353
} /* if */
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
if ((unzGoToNextFile(unz) != UNZ_OK) && (i + 1 < max))
{
freeEntries(info, i + 1, ERR_IO_ERROR);
return(0);
} /* if */
} /* for */
return(1);
} /* loadZipEntries */
static DirHandle *ZIP_openArchive(const char *name, int forWriting)
{
unzFile unz = NULL;
DirHandle *retval = NULL;
BAIL_IF_MACRO(forWriting, ERR_ARC_IS_READ_ONLY, NULL);
retval = malloc(sizeof (DirHandle));
BAIL_IF_MACRO(retval == NULL, ERR_OUT_OF_MEMORY, NULL);
unz = unzOpen(name);
if (unz == NULL)
{
free(retval);
BAIL_IF_MACRO(1, ERR_UNSUPPORTED_ARCHIVE, NULL);
} /* if */
retval->opaque = malloc(sizeof (ZIPinfo));
if (retval->opaque == NULL)
{
free(retval);
unzClose(unz);
BAIL_IF_MACRO(1, ERR_OUT_OF_MEMORY, NULL);
} /* if */
((ZIPinfo *) (retval->opaque))->archiveName = malloc(strlen(name) + 1);
if ( (((ZIPinfo *) (retval->opaque))->archiveName == NULL) ||
(!loadZipEntries( (ZIPinfo *) (retval->opaque), unz)) )
{
if (((ZIPinfo *) (retval->opaque))->archiveName != NULL)
free(((ZIPinfo *) (retval->opaque))->archiveName);
free(retval->opaque);
free(retval);
unzClose(unz);
BAIL_IF_MACRO(1, ERR_OUT_OF_MEMORY, NULL);
401
402
} /* if */
403
404
405
406
unzClose(unz);
strcpy(((ZIPinfo *) (retval->opaque))->archiveName, name);
retval->funcs = &__PHYSFS_DirFunctions_ZIP;
407
return(retval);
408
} /* ZIP_openArchive */
409
410
411
/* !!! This is seriously ugly. */
412
413
414
static LinkedStringList *ZIP_enumerateFiles(DirHandle *h,
const char *dirname,
int omitSymLinks)
415
{
416
ZIPinfo *zi = (ZIPinfo *) (h->opaque);
417
unsigned int i;
418
int dlen;
419
420
421
422
LinkedStringList *retval = NULL;
LinkedStringList *l = NULL;
LinkedStringList *prev = NULL;
char *d;
423
424
ZIPentry *entry;
char buf[MAXZIPENTRYSIZE];
425
426
427
428
dlen = strlen(dirname);
d = malloc(dlen + 1);
BAIL_IF_MACRO(d == NULL, ERR_OUT_OF_MEMORY, NULL);
429
strcpy(d, dirname);
430
431
432
433
434
if ((dlen > 0) && (d[dlen - 1] == '/')) /* no trailing slash. */
{
dlen--;
d[dlen] = '\0';
} /* if */
435
436
for (i = 0, entry = zi->entries; i < zi->global.number_entry; i++, entry++)
437
438
{
char *ptr;
439
440
441
char *add_file;
int this_dlen;
442
if ((omitSymLinks) && (entry->symlink != NULL))
443
444
continue;
445
446
447
448
449
450
this_dlen = strlen(entry->name);
if (this_dlen + 1 > MAXZIPENTRYSIZE)
continue; /* ugh. */
strcpy(buf, entry->name);
451
if ((this_dlen > 0) && (buf[this_dlen - 1] == '/')) /* no trailing slash. */
452
{
453
454
455
456
457
458
459
460
461
this_dlen--;
buf[this_dlen] = '\0';
} /* if */
if (this_dlen <= dlen) /* not in this dir. */
continue;
if (*d == '\0')
add_file = buf;
462
463
else
{
464
465
466
467
if (buf[dlen] != '/') /* can't be in same directory? */
continue;
buf[dlen] = '\0';
468
if (__PHYSFS_platformStricmp(d, buf) != 0) /* not same directory? */
469
470
471
continue;
add_file = buf + dlen + 1;
472
473
} /* else */
474
475
476
477
478
479
480
481
/* handle subdirectories... */
ptr = strchr(add_file, '/');
if (ptr != NULL)
{
LinkedStringList *j;
*ptr = '\0';
for (j = retval; j != NULL; j = j->next)
{
482
if (__PHYSFS_platformStricmp(j->str, ptr) == 0)
483
484
485
486
487
488
489
break;
} /* for */
if (j != NULL)
continue;
} /* if */
490
l = (LinkedStringList *) malloc(sizeof (LinkedStringList));
491
if (l == NULL)
492
493
break;
494
l->str = (char *) malloc(strlen(add_file) + 1);
495
496
497
498
499
500
if (l->str == NULL)
{
free(l);
break;
} /* if */
501
strcpy(l->str, add_file);
502
503
504
505
506
507
508
509
510
511
512
513
if (retval == NULL)
retval = l;
else
prev->next = l;
prev = l;
l->next = NULL;
} /* for */
free(d);
return(retval);
514
515
516
} /* ZIP_enumerateFiles */
517
518
519
520
521
522
523
/* !!! This is seriously ugly. */
static int ZIP_exists_symcheck(DirHandle *h, const char *name, int follow)
{
char buf[MAXZIPENTRYSIZE];
ZIPinfo *zi = (ZIPinfo *) (h->opaque);
int dlen;
char *d;
524
unsigned int i;
525
ZIPentry *entry;
526
527
528
dlen = strlen(name);
d = malloc(dlen + 1);
529
BAIL_IF_MACRO(d == NULL, ERR_OUT_OF_MEMORY, -1);
530
531
532
533
534
535
536
strcpy(d, name);
if ((dlen > 0) && (d[dlen - 1] == '/')) /* no trailing slash. */
{
dlen--;
d[dlen] = '\0';
} /* if */
537
for (i = 0, entry = zi->entries; i < zi->global.number_entry; i++, entry++)
538
{
539
540
541
542
543
int this_dlen = strlen(entry->name);
if (this_dlen + 1 > MAXZIPENTRYSIZE)
continue; /* ugh. */
strcpy(buf, entry->name);
544
545
546
547
548
549
550
551
552
if ((this_dlen > 0) && (buf[this_dlen - 1] == '/')) /* no trailing slash. */
{
this_dlen--;
buf[this_dlen] = '\0';
} /* if */
if ( ((buf[dlen] == '/') || (buf[dlen] == '\0')) &&
(strncmp(d, buf, dlen) == 0) )
{
553
int retval = i;
554
555
556
free(d);
if (follow) /* follow symlinks? */
{
557
558
if (entry->symlink != NULL)
retval = ZIP_exists_symcheck(h, entry->symlink, follow-1);
559
560
561
562
563
564
} /* if */
return(retval);
} /* if */
} /* for */
free(d);
565
return(-1);
566
567
568
} /* ZIP_exists_symcheck */
569
static int ZIP_exists(DirHandle *h, const char *name)
570
{
571
572
int retval = ZIP_exists_symcheck(h, name, SYMLINK_RECURSE_COUNT);
int is_sym;
573
574
575
if (retval == -1)
return(0);
576
577
578
579
/* if it's a symlink, then we ran into a possible symlink loop. */
is_sym = ( ((ZIPinfo *)(h->opaque))->entries[retval].symlink != NULL );
BAIL_IF_MACRO(is_sym, ERR_TOO_MANY_SYMLINKS, 0);
580
581
return(1);
582
583
584
} /* ZIP_exists */
585
static int ZIP_isDirectory(DirHandle *h, const char *name)
586
{
587
588
589
int dlen;
int is_sym;
int retval = ZIP_exists_symcheck(h, name, SYMLINK_RECURSE_COUNT);
590
591
592
593
594
595
596
if (retval == -1)
return(0);
/* if it's a symlink, then we ran into a possible symlink loop. */
is_sym = ( ((ZIPinfo *)(h->opaque))->entries[retval].symlink != NULL );
BAIL_IF_MACRO(is_sym, ERR_TOO_MANY_SYMLINKS, 0);
597
598
599
600
dlen = strlen(name);
/* !!! yikes. Better way to check? */
retval = (((ZIPinfo *)(h->opaque))->entries[retval].name[dlen] == '/');
601
return(retval);
602
603
604
} /* ZIP_isDirectory */
605
static int ZIP_isSymLink(DirHandle *h, const char *name)
606
{
607
int retval = ZIP_exists_symcheck(h, name, 0);
608
609
if (retval == -1)
return(0);
610
611
retval = ( ((ZIPinfo *)(h->opaque))->entries[retval].symlink != NULL );
612
return(retval);
613
614
615
} /* ZIP_isSymLink */
616
static FileHandle *ZIP_openRead(DirHandle *h, const char *filename)
617
{
618
FileHandle *retval = NULL;
619
ZIPinfo *zi = ((ZIPinfo *) (h->opaque));
620
ZIPfileinfo *finfo = NULL;
621
int pos = ZIP_exists_symcheck(h, filename, SYMLINK_RECURSE_COUNT);
622
unzFile f;
623
624
BAIL_IF_MACRO(pos == -1, ERR_NO_SUCH_FILE, NULL);
625
626
f = unzOpen(zi->archiveName);
627
628
BAIL_IF_MACRO(f == NULL, ERR_IO_ERROR, NULL);
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
if (unzGoToFirstFile(f) != UNZ_OK)
{
unzClose(f);
BAIL_IF_MACRO(1, ERR_IO_ERROR, NULL);
} /* if */
for (; pos > 0; pos--)
{
if (unzGoToNextFile(f) != UNZ_OK)
{
unzClose(f);
BAIL_IF_MACRO(1, ERR_IO_ERROR, NULL);
} /* if */
} /* for */
if ( (unzOpenCurrentFile(f) != UNZ_OK) ||
645
( (finfo = (ZIPfileinfo *) malloc(sizeof (ZIPfileinfo))) == NULL ) )
646
647
648
649
650
{
unzClose(f);
BAIL_IF_MACRO(1, ERR_IO_ERROR, NULL);
} /* if */
651
652
653
654
655
656
657
658
659
if ( (!(retval = (FileHandle *) malloc(sizeof (FileHandle)))) ||
(!(retval->opaque = (ZIPfileinfo *) malloc(sizeof (ZIPfileinfo)))) )
{
if (retval)
free(retval);
unzClose(f);
BAIL_IF_MACRO(1, ERR_OUT_OF_MEMORY, NULL);
} /* if */
660
finfo->handle = f;
661
retval->opaque = (void *) finfo;
662
retval->funcs = &__PHYSFS_FileFunctions_ZIP;
663
664
retval->dirHandle = h;
return(retval);
665
666
667
} /* ZIP_openRead */
668
static void ZIP_dirClose(DirHandle *h)
669
{
670
671
672
673
ZIPinfo *zi = (ZIPinfo *) (h->opaque);
freeEntries(zi, zi->global.number_entry, NULL);
free(zi->archiveName);
free(zi);
674
free(h);
675
676
} /* ZIP_dirClose */
677
/* end of zip.c ... */