|
1 /* gzlib.c -- zlib functions common to reading and writing gzip files |
|
2 * Copyright (C) 2004, 2010 Mark Adler |
|
3 * For conditions of distribution and use, see copyright notice in zlib.h |
|
4 */ |
|
5 |
|
6 #include "gzguts.h" |
|
7 |
|
8 #if defined(_LARGEFILE64_SOURCE) && _LFS64_LARGEFILE-0 |
|
9 # define LSEEK lseek64 |
|
10 #else |
|
11 # define LSEEK lseek |
|
12 #endif |
|
13 |
|
14 /* Local functions */ |
|
15 local void gz_reset OF((gz_statep)); |
|
16 local gzFile gz_open OF((const char *, int, const char *)); |
|
17 |
|
18 #if defined UNDER_CE |
|
19 |
|
20 /* Map the Windows error number in ERROR to a locale-dependent error message |
|
21 string and return a pointer to it. Typically, the values for ERROR come |
|
22 from GetLastError. |
|
23 |
|
24 The string pointed to shall not be modified by the application, but may be |
|
25 overwritten by a subsequent call to gz_strwinerror |
|
26 |
|
27 The gz_strwinerror function does not change the current setting of |
|
28 GetLastError. */ |
|
29 char ZLIB_INTERNAL *gz_strwinerror (error) |
|
30 DWORD error; |
|
31 { |
|
32 static char buf[1024]; |
|
33 |
|
34 wchar_t *msgbuf; |
|
35 DWORD lasterr = GetLastError(); |
|
36 DWORD chars = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM |
|
37 | FORMAT_MESSAGE_ALLOCATE_BUFFER, |
|
38 NULL, |
|
39 error, |
|
40 0, /* Default language */ |
|
41 (LPVOID)&msgbuf, |
|
42 0, |
|
43 NULL); |
|
44 if (chars != 0) { |
|
45 /* If there is an \r\n appended, zap it. */ |
|
46 if (chars >= 2 |
|
47 && msgbuf[chars - 2] == '\r' && msgbuf[chars - 1] == '\n') { |
|
48 chars -= 2; |
|
49 msgbuf[chars] = 0; |
|
50 } |
|
51 |
|
52 if (chars > sizeof (buf) - 1) { |
|
53 chars = sizeof (buf) - 1; |
|
54 msgbuf[chars] = 0; |
|
55 } |
|
56 |
|
57 wcstombs(buf, msgbuf, chars + 1); |
|
58 LocalFree(msgbuf); |
|
59 } |
|
60 else { |
|
61 sprintf(buf, "unknown win32 error (%ld)", error); |
|
62 } |
|
63 |
|
64 SetLastError(lasterr); |
|
65 return buf; |
|
66 } |
|
67 |
|
68 #endif /* UNDER_CE */ |
|
69 |
|
70 /* Reset gzip file state */ |
|
71 local void gz_reset(state) |
|
72 gz_statep state; |
|
73 { |
|
74 if (state->mode == GZ_READ) { /* for reading ... */ |
|
75 state->have = 0; /* no output data available */ |
|
76 state->eof = 0; /* not at end of file */ |
|
77 state->how = LOOK; /* look for gzip header */ |
|
78 state->direct = 1; /* default for empty file */ |
|
79 } |
|
80 state->seek = 0; /* no seek request pending */ |
|
81 gz_error(state, Z_OK, NULL); /* clear error */ |
|
82 state->pos = 0; /* no uncompressed data yet */ |
|
83 state->strm.avail_in = 0; /* no input data yet */ |
|
84 } |
|
85 |
|
86 /* Open a gzip file either by name or file descriptor. */ |
|
87 local gzFile gz_open(path, fd, mode) |
|
88 const char *path; |
|
89 int fd; |
|
90 const char *mode; |
|
91 { |
|
92 gz_statep state; |
|
93 |
|
94 /* allocate gzFile structure to return */ |
|
95 state = malloc(sizeof(gz_state)); |
|
96 if (state == NULL) |
|
97 return NULL; |
|
98 state->size = 0; /* no buffers allocated yet */ |
|
99 state->want = GZBUFSIZE; /* requested buffer size */ |
|
100 state->msg = NULL; /* no error message yet */ |
|
101 |
|
102 /* interpret mode */ |
|
103 state->mode = GZ_NONE; |
|
104 state->level = Z_DEFAULT_COMPRESSION; |
|
105 state->strategy = Z_DEFAULT_STRATEGY; |
|
106 while (*mode) { |
|
107 if (*mode >= '0' && *mode <= '9') |
|
108 state->level = *mode - '0'; |
|
109 else |
|
110 switch (*mode) { |
|
111 case 'r': |
|
112 state->mode = GZ_READ; |
|
113 break; |
|
114 #ifndef NO_GZCOMPRESS |
|
115 case 'w': |
|
116 state->mode = GZ_WRITE; |
|
117 break; |
|
118 case 'a': |
|
119 state->mode = GZ_APPEND; |
|
120 break; |
|
121 #endif |
|
122 case '+': /* can't read and write at the same time */ |
|
123 free(state); |
|
124 return NULL; |
|
125 case 'b': /* ignore -- will request binary anyway */ |
|
126 break; |
|
127 case 'f': |
|
128 state->strategy = Z_FILTERED; |
|
129 break; |
|
130 case 'h': |
|
131 state->strategy = Z_HUFFMAN_ONLY; |
|
132 break; |
|
133 case 'R': |
|
134 state->strategy = Z_RLE; |
|
135 break; |
|
136 case 'F': |
|
137 state->strategy = Z_FIXED; |
|
138 default: /* could consider as an error, but just ignore */ |
|
139 ; |
|
140 } |
|
141 mode++; |
|
142 } |
|
143 |
|
144 /* must provide an "r", "w", or "a" */ |
|
145 if (state->mode == GZ_NONE) { |
|
146 free(state); |
|
147 return NULL; |
|
148 } |
|
149 |
|
150 /* save the path name for error messages */ |
|
151 state->path = malloc(strlen(path) + 1); |
|
152 if (state->path == NULL) { |
|
153 free(state); |
|
154 return NULL; |
|
155 } |
|
156 strcpy(state->path, path); |
|
157 |
|
158 /* open the file with the appropriate mode (or just use fd) */ |
|
159 state->fd = fd != -1 ? fd : |
|
160 open(path, |
|
161 #ifdef O_LARGEFILE |
|
162 O_LARGEFILE | |
|
163 #endif |
|
164 #ifdef O_BINARY |
|
165 O_BINARY | |
|
166 #endif |
|
167 (state->mode == GZ_READ ? |
|
168 O_RDONLY : |
|
169 (O_WRONLY | O_CREAT | ( |
|
170 state->mode == GZ_WRITE ? |
|
171 O_TRUNC : |
|
172 O_APPEND))), |
|
173 0666); |
|
174 if (state->fd == -1) { |
|
175 free(state->path); |
|
176 free(state); |
|
177 return NULL; |
|
178 } |
|
179 if (state->mode == GZ_APPEND) |
|
180 state->mode = GZ_WRITE; /* simplify later checks */ |
|
181 |
|
182 /* save the current position for rewinding (only if reading) */ |
|
183 if (state->mode == GZ_READ) { |
|
184 state->start = LSEEK(state->fd, 0, SEEK_CUR); |
|
185 if (state->start == -1) state->start = 0; |
|
186 } |
|
187 |
|
188 /* initialize stream */ |
|
189 gz_reset(state); |
|
190 |
|
191 /* return stream */ |
|
192 return (gzFile)state; |
|
193 } |
|
194 |
|
195 /* -- see zlib.h -- */ |
|
196 gzFile ZEXPORT gzopen(path, mode) |
|
197 const char *path; |
|
198 const char *mode; |
|
199 { |
|
200 return gz_open(path, -1, mode); |
|
201 } |
|
202 |
|
203 /* -- see zlib.h -- */ |
|
204 gzFile ZEXPORT gzopen64(path, mode) |
|
205 const char *path; |
|
206 const char *mode; |
|
207 { |
|
208 return gz_open(path, -1, mode); |
|
209 } |
|
210 |
|
211 /* -- see zlib.h -- */ |
|
212 gzFile ZEXPORT gzdopen(fd, mode) |
|
213 int fd; |
|
214 const char *mode; |
|
215 { |
|
216 char *path; /* identifier for error messages */ |
|
217 gzFile gz; |
|
218 |
|
219 if (fd == -1 || (path = malloc(7 + 3 * sizeof(int))) == NULL) |
|
220 return NULL; |
|
221 sprintf(path, "<fd:%d>", fd); /* for debugging */ |
|
222 gz = gz_open(path, fd, mode); |
|
223 free(path); |
|
224 return gz; |
|
225 } |
|
226 |
|
227 /* -- see zlib.h -- */ |
|
228 int ZEXPORT gzbuffer(file, size) |
|
229 gzFile file; |
|
230 unsigned size; |
|
231 { |
|
232 gz_statep state; |
|
233 |
|
234 /* get internal structure and check integrity */ |
|
235 if (file == NULL) |
|
236 return -1; |
|
237 state = (gz_statep)file; |
|
238 if (state->mode != GZ_READ && state->mode != GZ_WRITE) |
|
239 return -1; |
|
240 |
|
241 /* make sure we haven't already allocated memory */ |
|
242 if (state->size != 0) |
|
243 return -1; |
|
244 |
|
245 /* check and set requested size */ |
|
246 if (size == 0) |
|
247 return -1; |
|
248 state->want = size; |
|
249 return 0; |
|
250 } |
|
251 |
|
252 /* -- see zlib.h -- */ |
|
253 int ZEXPORT gzrewind(file) |
|
254 gzFile file; |
|
255 { |
|
256 gz_statep state; |
|
257 |
|
258 /* get internal structure */ |
|
259 if (file == NULL) |
|
260 return -1; |
|
261 state = (gz_statep)file; |
|
262 |
|
263 /* check that we're reading and that there's no error */ |
|
264 if (state->mode != GZ_READ || state->err != Z_OK) |
|
265 return -1; |
|
266 |
|
267 /* back up and start over */ |
|
268 if (LSEEK(state->fd, state->start, SEEK_SET) == -1) |
|
269 return -1; |
|
270 gz_reset(state); |
|
271 return 0; |
|
272 } |
|
273 |
|
274 /* -- see zlib.h -- */ |
|
275 z_off64_t ZEXPORT gzseek64(file, offset, whence) |
|
276 gzFile file; |
|
277 z_off64_t offset; |
|
278 int whence; |
|
279 { |
|
280 unsigned n; |
|
281 z_off64_t ret; |
|
282 gz_statep state; |
|
283 |
|
284 /* get internal structure and check integrity */ |
|
285 if (file == NULL) |
|
286 return -1; |
|
287 state = (gz_statep)file; |
|
288 if (state->mode != GZ_READ && state->mode != GZ_WRITE) |
|
289 return -1; |
|
290 |
|
291 /* check that there's no error */ |
|
292 if (state->err != Z_OK) |
|
293 return -1; |
|
294 |
|
295 /* can only seek from start or relative to current position */ |
|
296 if (whence != SEEK_SET && whence != SEEK_CUR) |
|
297 return -1; |
|
298 |
|
299 /* normalize offset to a SEEK_CUR specification */ |
|
300 if (whence == SEEK_SET) |
|
301 offset -= state->pos; |
|
302 else if (state->seek) |
|
303 offset += state->skip; |
|
304 state->seek = 0; |
|
305 |
|
306 /* if within raw area while reading, just go there */ |
|
307 if (state->mode == GZ_READ && state->how == COPY && |
|
308 state->pos + offset >= state->raw) { |
|
309 ret = LSEEK(state->fd, offset - state->have, SEEK_CUR); |
|
310 if (ret == -1) |
|
311 return -1; |
|
312 state->have = 0; |
|
313 state->eof = 0; |
|
314 state->seek = 0; |
|
315 gz_error(state, Z_OK, NULL); |
|
316 state->strm.avail_in = 0; |
|
317 state->pos += offset; |
|
318 return state->pos; |
|
319 } |
|
320 |
|
321 /* calculate skip amount, rewinding if needed for back seek when reading */ |
|
322 if (offset < 0) { |
|
323 if (state->mode != GZ_READ) /* writing -- can't go backwards */ |
|
324 return -1; |
|
325 offset += state->pos; |
|
326 if (offset < 0) /* before start of file! */ |
|
327 return -1; |
|
328 if (gzrewind(file) == -1) /* rewind, then skip to offset */ |
|
329 return -1; |
|
330 } |
|
331 |
|
332 /* if reading, skip what's in output buffer (one less gzgetc() check) */ |
|
333 if (state->mode == GZ_READ) { |
|
334 n = GT_OFF(state->have) || (z_off64_t)state->have > offset ? |
|
335 (unsigned)offset : state->have; |
|
336 state->have -= n; |
|
337 state->next += n; |
|
338 state->pos += n; |
|
339 offset -= n; |
|
340 } |
|
341 |
|
342 /* request skip (if not zero) */ |
|
343 if (offset) { |
|
344 state->seek = 1; |
|
345 state->skip = offset; |
|
346 } |
|
347 return state->pos + offset; |
|
348 } |
|
349 |
|
350 /* -- see zlib.h -- */ |
|
351 z_off_t ZEXPORT gzseek(file, offset, whence) |
|
352 gzFile file; |
|
353 z_off_t offset; |
|
354 int whence; |
|
355 { |
|
356 z_off64_t ret; |
|
357 |
|
358 ret = gzseek64(file, (z_off64_t)offset, whence); |
|
359 return ret == (z_off_t)ret ? (z_off_t)ret : -1; |
|
360 } |
|
361 |
|
362 /* -- see zlib.h -- */ |
|
363 z_off64_t ZEXPORT gztell64(file) |
|
364 gzFile file; |
|
365 { |
|
366 gz_statep state; |
|
367 |
|
368 /* get internal structure and check integrity */ |
|
369 if (file == NULL) |
|
370 return -1; |
|
371 state = (gz_statep)file; |
|
372 if (state->mode != GZ_READ && state->mode != GZ_WRITE) |
|
373 return -1; |
|
374 |
|
375 /* return position */ |
|
376 return state->pos + (state->seek ? state->skip : 0); |
|
377 } |
|
378 |
|
379 /* -- see zlib.h -- */ |
|
380 z_off_t ZEXPORT gztell(file) |
|
381 gzFile file; |
|
382 { |
|
383 z_off64_t ret; |
|
384 |
|
385 ret = gztell64(file); |
|
386 return ret == (z_off_t)ret ? (z_off_t)ret : -1; |
|
387 } |
|
388 |
|
389 /* -- see zlib.h -- */ |
|
390 z_off64_t ZEXPORT gzoffset64(file) |
|
391 gzFile file; |
|
392 { |
|
393 z_off64_t offset; |
|
394 gz_statep state; |
|
395 |
|
396 /* get internal structure and check integrity */ |
|
397 if (file == NULL) |
|
398 return -1; |
|
399 state = (gz_statep)file; |
|
400 if (state->mode != GZ_READ && state->mode != GZ_WRITE) |
|
401 return -1; |
|
402 |
|
403 /* compute and return effective offset in file */ |
|
404 offset = LSEEK(state->fd, 0, SEEK_CUR); |
|
405 if (offset == -1) |
|
406 return -1; |
|
407 if (state->mode == GZ_READ) /* reading */ |
|
408 offset -= state->strm.avail_in; /* don't count buffered input */ |
|
409 return offset; |
|
410 } |
|
411 |
|
412 /* -- see zlib.h -- */ |
|
413 z_off_t ZEXPORT gzoffset(file) |
|
414 gzFile file; |
|
415 { |
|
416 z_off64_t ret; |
|
417 |
|
418 ret = gzoffset64(file); |
|
419 return ret == (z_off_t)ret ? (z_off_t)ret : -1; |
|
420 } |
|
421 |
|
422 /* -- see zlib.h -- */ |
|
423 int ZEXPORT gzeof(file) |
|
424 gzFile file; |
|
425 { |
|
426 gz_statep state; |
|
427 |
|
428 /* get internal structure and check integrity */ |
|
429 if (file == NULL) |
|
430 return 0; |
|
431 state = (gz_statep)file; |
|
432 if (state->mode != GZ_READ && state->mode != GZ_WRITE) |
|
433 return 0; |
|
434 |
|
435 /* return end-of-file state */ |
|
436 return state->mode == GZ_READ ? |
|
437 (state->eof && state->strm.avail_in == 0 && state->have == 0) : 0; |
|
438 } |
|
439 |
|
440 /* -- see zlib.h -- */ |
|
441 const char * ZEXPORT gzerror(file, errnum) |
|
442 gzFile file; |
|
443 int *errnum; |
|
444 { |
|
445 gz_statep state; |
|
446 |
|
447 /* get internal structure and check integrity */ |
|
448 if (file == NULL) |
|
449 return NULL; |
|
450 state = (gz_statep)file; |
|
451 if (state->mode != GZ_READ && state->mode != GZ_WRITE) |
|
452 return NULL; |
|
453 |
|
454 /* return error information */ |
|
455 if (errnum != NULL) |
|
456 *errnum = state->err; |
|
457 return state->msg == NULL ? "" : state->msg; |
|
458 } |
|
459 |
|
460 /* -- see zlib.h -- */ |
|
461 void ZEXPORT gzclearerr(file) |
|
462 gzFile file; |
|
463 { |
|
464 gz_statep state; |
|
465 |
|
466 /* get internal structure and check integrity */ |
|
467 if (file == NULL) |
|
468 return; |
|
469 state = (gz_statep)file; |
|
470 if (state->mode != GZ_READ && state->mode != GZ_WRITE) |
|
471 return; |
|
472 |
|
473 /* clear error and end-of-file */ |
|
474 if (state->mode == GZ_READ) |
|
475 state->eof = 0; |
|
476 gz_error(state, Z_OK, NULL); |
|
477 } |
|
478 |
|
479 /* Create an error message in allocated memory and set state->err and |
|
480 state->msg accordingly. Free any previous error message already there. Do |
|
481 not try to free or allocate space if the error is Z_MEM_ERROR (out of |
|
482 memory). Simply save the error message as a static string. If there is an |
|
483 allocation failure constructing the error message, then convert the error to |
|
484 out of memory. */ |
|
485 void ZLIB_INTERNAL gz_error(state, err, msg) |
|
486 gz_statep state; |
|
487 int err; |
|
488 const char *msg; |
|
489 { |
|
490 /* free previously allocated message and clear */ |
|
491 if (state->msg != NULL) { |
|
492 if (state->err != Z_MEM_ERROR) |
|
493 free(state->msg); |
|
494 state->msg = NULL; |
|
495 } |
|
496 |
|
497 /* set error code, and if no message, then done */ |
|
498 state->err = err; |
|
499 if (msg == NULL) |
|
500 return; |
|
501 |
|
502 /* for an out of memory error, save as static string */ |
|
503 if (err == Z_MEM_ERROR) { |
|
504 state->msg = (char *)msg; |
|
505 return; |
|
506 } |
|
507 |
|
508 /* construct error message with path */ |
|
509 if ((state->msg = malloc(strlen(state->path) + strlen(msg) + 3)) == NULL) { |
|
510 state->err = Z_MEM_ERROR; |
|
511 state->msg = (char *)"out of memory"; |
|
512 return; |
|
513 } |
|
514 strcpy(state->msg, state->path); |
|
515 strcat(state->msg, ": "); |
|
516 strcat(state->msg, msg); |
|
517 return; |
|
518 } |
|
519 |
|
520 #ifndef INT_MAX |
|
521 /* portably return maximum value for an int (when limits.h presumed not |
|
522 available) -- we need to do this to cover cases where 2's complement not |
|
523 used, since C standard permits 1's complement and sign-bit representations, |
|
524 otherwise we could just use ((unsigned)-1) >> 1 */ |
|
525 unsigned ZLIB_INTERNAL gz_intmax() |
|
526 { |
|
527 unsigned p, q; |
|
528 |
|
529 p = 1; |
|
530 do { |
|
531 q = p; |
|
532 p <<= 1; |
|
533 p++; |
|
534 } while (p > q); |
|
535 return q >> 1; |
|
536 } |
|
537 #endif |