inflate.c revision 1.1 1 1.1 christos /* $NetBSD: inflate.c,v 1.1 2006/01/14 20:10:31 christos Exp $ */
2 1.1 christos
3 1.1 christos /* inflate.c -- zlib decompression
4 1.1 christos * Copyright (C) 1995-2005 Mark Adler
5 1.1 christos * For conditions of distribution and use, see copyright notice in zlib.h
6 1.1 christos */
7 1.1 christos
8 1.1 christos /*
9 1.1 christos * Change history:
10 1.1 christos *
11 1.1 christos * 1.2.beta0 24 Nov 2002
12 1.1 christos * - First version -- complete rewrite of inflate to simplify code, avoid
13 1.1 christos * creation of window when not needed, minimize use of window when it is
14 1.1 christos * needed, make inffast.c even faster, implement gzip decoding, and to
15 1.1 christos * improve code readability and style over the previous zlib inflate code
16 1.1 christos *
17 1.1 christos * 1.2.beta1 25 Nov 2002
18 1.1 christos * - Use pointers for available input and output checking in inffast.c
19 1.1 christos * - Remove input and output counters in inffast.c
20 1.1 christos * - Change inffast.c entry and loop from avail_in >= 7 to >= 6
21 1.1 christos * - Remove unnecessary second byte pull from length extra in inffast.c
22 1.1 christos * - Unroll direct copy to three copies per loop in inffast.c
23 1.1 christos *
24 1.1 christos * 1.2.beta2 4 Dec 2002
25 1.1 christos * - Change external routine names to reduce potential conflicts
26 1.1 christos * - Correct filename to inffixed.h for fixed tables in inflate.c
27 1.1 christos * - Make hbuf[] unsigned char to match parameter type in inflate.c
28 1.1 christos * - Change strm->next_out[-state->offset] to *(strm->next_out - state->offset)
29 1.1 christos * to avoid negation problem on Alphas (64 bit) in inflate.c
30 1.1 christos *
31 1.1 christos * 1.2.beta3 22 Dec 2002
32 1.1 christos * - Add comments on state->bits assertion in inffast.c
33 1.1 christos * - Add comments on op field in inftrees.h
34 1.1 christos * - Fix bug in reuse of allocated window after inflateReset()
35 1.1 christos * - Remove bit fields--back to byte structure for speed
36 1.1 christos * - Remove distance extra == 0 check in inflate_fast()--only helps for lengths
37 1.1 christos * - Change post-increments to pre-increments in inflate_fast(), PPC biased?
38 1.1 christos * - Add compile time option, POSTINC, to use post-increments instead (Intel?)
39 1.1 christos * - Make MATCH copy in inflate() much faster for when inflate_fast() not used
40 1.1 christos * - Use local copies of stream next and avail values, as well as local bit
41 1.1 christos * buffer and bit count in inflate()--for speed when inflate_fast() not used
42 1.1 christos *
43 1.1 christos * 1.2.beta4 1 Jan 2003
44 1.1 christos * - Split ptr - 257 statements in inflate_table() to avoid compiler warnings
45 1.1 christos * - Move a comment on output buffer sizes from inffast.c to inflate.c
46 1.1 christos * - Add comments in inffast.c to introduce the inflate_fast() routine
47 1.1 christos * - Rearrange window copies in inflate_fast() for speed and simplification
48 1.1 christos * - Unroll last copy for window match in inflate_fast()
49 1.1 christos * - Use local copies of window variables in inflate_fast() for speed
50 1.1 christos * - Pull out common write == 0 case for speed in inflate_fast()
51 1.1 christos * - Make op and len in inflate_fast() unsigned for consistency
52 1.1 christos * - Add FAR to lcode and dcode declarations in inflate_fast()
53 1.1 christos * - Simplified bad distance check in inflate_fast()
54 1.1 christos * - Added inflateBackInit(), inflateBack(), and inflateBackEnd() in new
55 1.1 christos * source file infback.c to provide a call-back interface to inflate for
56 1.1 christos * programs like gzip and unzip -- uses window as output buffer to avoid
57 1.1 christos * window copying
58 1.1 christos *
59 1.1 christos * 1.2.beta5 1 Jan 2003
60 1.1 christos * - Improved inflateBack() interface to allow the caller to provide initial
61 1.1 christos * input in strm.
62 1.1 christos * - Fixed stored blocks bug in inflateBack()
63 1.1 christos *
64 1.1 christos * 1.2.beta6 4 Jan 2003
65 1.1 christos * - Added comments in inffast.c on effectiveness of POSTINC
66 1.1 christos * - Typecasting all around to reduce compiler warnings
67 1.1 christos * - Changed loops from while (1) or do {} while (1) to for (;;), again to
68 1.1 christos * make compilers happy
69 1.1 christos * - Changed type of window in inflateBackInit() to unsigned char *
70 1.1 christos *
71 1.1 christos * 1.2.beta7 27 Jan 2003
72 1.1 christos * - Changed many types to unsigned or unsigned short to avoid warnings
73 1.1 christos * - Added inflateCopy() function
74 1.1 christos *
75 1.1 christos * 1.2.0 9 Mar 2003
76 1.1 christos * - Changed inflateBack() interface to provide separate opaque descriptors
77 1.1 christos * for the in() and out() functions
78 1.1 christos * - Changed inflateBack() argument and in_func typedef to swap the length
79 1.1 christos * and buffer address return values for the input function
80 1.1 christos * - Check next_in and next_out for Z_NULL on entry to inflate()
81 1.1 christos *
82 1.1 christos * The history for versions after 1.2.0 are in ChangeLog in zlib distribution.
83 1.1 christos */
84 1.1 christos
85 1.1 christos #include "zutil.h"
86 1.1 christos #include "inftrees.h"
87 1.1 christos #include "inflate.h"
88 1.1 christos #include "inffast.h"
89 1.1 christos
90 1.1 christos #ifdef MAKEFIXED
91 1.1 christos # ifndef BUILDFIXED
92 1.1 christos # define BUILDFIXED
93 1.1 christos # endif
94 1.1 christos #endif
95 1.1 christos
96 1.1 christos /* function prototypes */
97 1.1 christos local void fixedtables OF((struct inflate_state FAR *state));
98 1.1 christos local int updatewindow OF((z_streamp strm, unsigned out));
99 1.1 christos #ifdef BUILDFIXED
100 1.1 christos void makefixed OF((void));
101 1.1 christos #endif
102 1.1 christos local unsigned syncsearch OF((unsigned FAR *have, unsigned char FAR *buf,
103 1.1 christos unsigned len));
104 1.1 christos
105 1.1 christos int ZEXPORT inflateReset(strm)
106 1.1 christos z_streamp strm;
107 1.1 christos {
108 1.1 christos struct inflate_state FAR *state;
109 1.1 christos
110 1.1 christos if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
111 1.1 christos state = (struct inflate_state FAR *)strm->state;
112 1.1 christos strm->total_in = strm->total_out = state->total = 0;
113 1.1 christos strm->msg = Z_NULL;
114 1.1 christos strm->adler = 1; /* to support ill-conceived Java test suite */
115 1.1 christos state->mode = HEAD;
116 1.1 christos state->last = 0;
117 1.1 christos state->havedict = 0;
118 1.1 christos state->dmax = 32768U;
119 1.1 christos state->head = Z_NULL;
120 1.1 christos state->wsize = 0;
121 1.1 christos state->whave = 0;
122 1.1 christos state->write = 0;
123 1.1 christos state->hold = 0;
124 1.1 christos state->bits = 0;
125 1.1 christos state->lencode = state->distcode = state->next = state->codes;
126 1.1 christos Tracev((stderr, "inflate: reset\n"));
127 1.1 christos return Z_OK;
128 1.1 christos }
129 1.1 christos
130 1.1 christos int ZEXPORT inflatePrime(strm, bits, value)
131 1.1 christos z_streamp strm;
132 1.1 christos int bits;
133 1.1 christos int value;
134 1.1 christos {
135 1.1 christos struct inflate_state FAR *state;
136 1.1 christos
137 1.1 christos if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
138 1.1 christos state = (struct inflate_state FAR *)strm->state;
139 1.1 christos if (bits > 16 || state->bits + bits > 32) return Z_STREAM_ERROR;
140 1.1 christos value &= (1L << bits) - 1;
141 1.1 christos state->hold += value << state->bits;
142 1.1 christos state->bits += bits;
143 1.1 christos return Z_OK;
144 1.1 christos }
145 1.1 christos
146 1.1 christos int ZEXPORT inflateInit2_(strm, windowBits, version, stream_size)
147 1.1 christos z_streamp strm;
148 1.1 christos int windowBits;
149 1.1 christos const char *version;
150 1.1 christos int stream_size;
151 1.1 christos {
152 1.1 christos struct inflate_state FAR *state;
153 1.1 christos
154 1.1 christos if (version == Z_NULL || version[0] != ZLIB_VERSION[0] ||
155 1.1 christos stream_size != (int)(sizeof(z_stream)))
156 1.1 christos return Z_VERSION_ERROR;
157 1.1 christos if (strm == Z_NULL) return Z_STREAM_ERROR;
158 1.1 christos strm->msg = Z_NULL; /* in case we return an error */
159 1.1 christos if (strm->zalloc == (alloc_func)0) {
160 1.1 christos strm->zalloc = zcalloc;
161 1.1 christos strm->opaque = (voidpf)0;
162 1.1 christos }
163 1.1 christos if (strm->zfree == (free_func)0) strm->zfree = zcfree;
164 1.1 christos state = (struct inflate_state FAR *)
165 1.1 christos ZALLOC(strm, 1, sizeof(struct inflate_state));
166 1.1 christos if (state == Z_NULL) return Z_MEM_ERROR;
167 1.1 christos Tracev((stderr, "inflate: allocated\n"));
168 1.1 christos strm->state = (struct internal_state FAR *)state;
169 1.1 christos if (windowBits < 0) {
170 1.1 christos state->wrap = 0;
171 1.1 christos windowBits = -windowBits;
172 1.1 christos }
173 1.1 christos else {
174 1.1 christos state->wrap = (windowBits >> 4) + 1;
175 1.1 christos #ifdef GUNZIP
176 1.1 christos if (windowBits < 48) windowBits &= 15;
177 1.1 christos #endif
178 1.1 christos }
179 1.1 christos if (windowBits < 8 || windowBits > 15) {
180 1.1 christos ZFREE(strm, state);
181 1.1 christos strm->state = Z_NULL;
182 1.1 christos return Z_STREAM_ERROR;
183 1.1 christos }
184 1.1 christos state->wbits = (unsigned)windowBits;
185 1.1 christos state->window = Z_NULL;
186 1.1 christos return inflateReset(strm);
187 1.1 christos }
188 1.1 christos
189 1.1 christos int ZEXPORT inflateInit_(strm, version, stream_size)
190 1.1 christos z_streamp strm;
191 1.1 christos const char *version;
192 1.1 christos int stream_size;
193 1.1 christos {
194 1.1 christos return inflateInit2_(strm, DEF_WBITS, version, stream_size);
195 1.1 christos }
196 1.1 christos
197 1.1 christos /*
198 1.1 christos Return state with length and distance decoding tables and index sizes set to
199 1.1 christos fixed code decoding. Normally this returns fixed tables from inffixed.h.
200 1.1 christos If BUILDFIXED is defined, then instead this routine builds the tables the
201 1.1 christos first time it's called, and returns those tables the first time and
202 1.1 christos thereafter. This reduces the size of the code by about 2K bytes, in
203 1.1 christos exchange for a little execution time. However, BUILDFIXED should not be
204 1.1 christos used for threaded applications, since the rewriting of the tables and virgin
205 1.1 christos may not be thread-safe.
206 1.1 christos */
207 1.1 christos local void fixedtables(state)
208 1.1 christos struct inflate_state FAR *state;
209 1.1 christos {
210 1.1 christos #ifdef BUILDFIXED
211 1.1 christos static int virgin = 1;
212 1.1 christos static code *lenfix, *distfix;
213 1.1 christos static code fixed[544];
214 1.1 christos
215 1.1 christos /* build fixed huffman tables if first call (may not be thread safe) */
216 1.1 christos if (virgin) {
217 1.1 christos unsigned sym, bits;
218 1.1 christos static code *next;
219 1.1 christos
220 1.1 christos /* literal/length table */
221 1.1 christos sym = 0;
222 1.1 christos while (sym < 144) state->lens[sym++] = 8;
223 1.1 christos while (sym < 256) state->lens[sym++] = 9;
224 1.1 christos while (sym < 280) state->lens[sym++] = 7;
225 1.1 christos while (sym < 288) state->lens[sym++] = 8;
226 1.1 christos next = fixed;
227 1.1 christos lenfix = next;
228 1.1 christos bits = 9;
229 1.1 christos inflate_table(LENS, state->lens, 288, &(next), &(bits), state->work);
230 1.1 christos
231 1.1 christos /* distance table */
232 1.1 christos sym = 0;
233 1.1 christos while (sym < 32) state->lens[sym++] = 5;
234 1.1 christos distfix = next;
235 1.1 christos bits = 5;
236 1.1 christos inflate_table(DISTS, state->lens, 32, &(next), &(bits), state->work);
237 1.1 christos
238 1.1 christos /* do this just once */
239 1.1 christos virgin = 0;
240 1.1 christos }
241 1.1 christos #else /* !BUILDFIXED */
242 1.1 christos # include "inffixed.h"
243 1.1 christos #endif /* BUILDFIXED */
244 1.1 christos state->lencode = lenfix;
245 1.1 christos state->lenbits = 9;
246 1.1 christos state->distcode = distfix;
247 1.1 christos state->distbits = 5;
248 1.1 christos }
249 1.1 christos
250 1.1 christos #ifdef MAKEFIXED
251 1.1 christos #include <stdio.h>
252 1.1 christos
253 1.1 christos /*
254 1.1 christos Write out the inffixed.h that is #include'd above. Defining MAKEFIXED also
255 1.1 christos defines BUILDFIXED, so the tables are built on the fly. makefixed() writes
256 1.1 christos those tables to stdout, which would be piped to inffixed.h. A small program
257 1.1 christos can simply call makefixed to do this:
258 1.1 christos
259 1.1 christos void makefixed(void);
260 1.1 christos
261 1.1 christos int main(void)
262 1.1 christos {
263 1.1 christos makefixed();
264 1.1 christos return 0;
265 1.1 christos }
266 1.1 christos
267 1.1 christos Then that can be linked with zlib built with MAKEFIXED defined and run:
268 1.1 christos
269 1.1 christos a.out > inffixed.h
270 1.1 christos */
271 1.1 christos void makefixed()
272 1.1 christos {
273 1.1 christos unsigned low, size;
274 1.1 christos struct inflate_state state;
275 1.1 christos
276 1.1 christos fixedtables(&state);
277 1.1 christos puts(" /* inffixed.h -- table for decoding fixed codes");
278 1.1 christos puts(" * Generated automatically by makefixed().");
279 1.1 christos puts(" */");
280 1.1 christos puts("");
281 1.1 christos puts(" /* WARNING: this file should *not* be used by applications.");
282 1.1 christos puts(" It is part of the implementation of this library and is");
283 1.1 christos puts(" subject to change. Applications should only use zlib.h.");
284 1.1 christos puts(" */");
285 1.1 christos puts("");
286 1.1 christos size = 1U << 9;
287 1.1 christos printf(" static const code lenfix[%u] = {", size);
288 1.1 christos low = 0;
289 1.1 christos for (;;) {
290 1.1 christos if ((low % 7) == 0) printf("\n ");
291 1.1 christos printf("{%u,%u,%d}", state.lencode[low].op, state.lencode[low].bits,
292 1.1 christos state.lencode[low].val);
293 1.1 christos if (++low == size) break;
294 1.1 christos putchar(',');
295 1.1 christos }
296 1.1 christos puts("\n };");
297 1.1 christos size = 1U << 5;
298 1.1 christos printf("\n static const code distfix[%u] = {", size);
299 1.1 christos low = 0;
300 1.1 christos for (;;) {
301 1.1 christos if ((low % 6) == 0) printf("\n ");
302 1.1 christos printf("{%u,%u,%d}", state.distcode[low].op, state.distcode[low].bits,
303 1.1 christos state.distcode[low].val);
304 1.1 christos if (++low == size) break;
305 1.1 christos putchar(',');
306 1.1 christos }
307 1.1 christos puts("\n };");
308 1.1 christos }
309 1.1 christos #endif /* MAKEFIXED */
310 1.1 christos
311 1.1 christos /*
312 1.1 christos Update the window with the last wsize (normally 32K) bytes written before
313 1.1 christos returning. If window does not exist yet, create it. This is only called
314 1.1 christos when a window is already in use, or when output has been written during this
315 1.1 christos inflate call, but the end of the deflate stream has not been reached yet.
316 1.1 christos It is also called to create a window for dictionary data when a dictionary
317 1.1 christos is loaded.
318 1.1 christos
319 1.1 christos Providing output buffers larger than 32K to inflate() should provide a speed
320 1.1 christos advantage, since only the last 32K of output is copied to the sliding window
321 1.1 christos upon return from inflate(), and since all distances after the first 32K of
322 1.1 christos output will fall in the output data, making match copies simpler and faster.
323 1.1 christos The advantage may be dependent on the size of the processor's data caches.
324 1.1 christos */
325 1.1 christos local int updatewindow(strm, out)
326 1.1 christos z_streamp strm;
327 1.1 christos unsigned out;
328 1.1 christos {
329 1.1 christos struct inflate_state FAR *state;
330 1.1 christos unsigned copy, dist;
331 1.1 christos
332 1.1 christos state = (struct inflate_state FAR *)strm->state;
333 1.1 christos
334 1.1 christos /* if it hasn't been done already, allocate space for the window */
335 1.1 christos if (state->window == Z_NULL) {
336 1.1 christos state->window = (unsigned char FAR *)
337 1.1 christos ZALLOC(strm, 1U << state->wbits,
338 1.1 christos sizeof(unsigned char));
339 1.1 christos if (state->window == Z_NULL) return 1;
340 1.1 christos }
341 1.1 christos
342 1.1 christos /* if window not in use yet, initialize */
343 1.1 christos if (state->wsize == 0) {
344 1.1 christos state->wsize = 1U << state->wbits;
345 1.1 christos state->write = 0;
346 1.1 christos state->whave = 0;
347 1.1 christos }
348 1.1 christos
349 1.1 christos /* copy state->wsize or less output bytes into the circular window */
350 1.1 christos copy = out - strm->avail_out;
351 1.1 christos if (copy >= state->wsize) {
352 1.1 christos zmemcpy(state->window, strm->next_out - state->wsize, state->wsize);
353 1.1 christos state->write = 0;
354 1.1 christos state->whave = state->wsize;
355 1.1 christos }
356 1.1 christos else {
357 1.1 christos dist = state->wsize - state->write;
358 1.1 christos if (dist > copy) dist = copy;
359 1.1 christos zmemcpy(state->window + state->write, strm->next_out - copy, dist);
360 1.1 christos copy -= dist;
361 1.1 christos if (copy) {
362 1.1 christos zmemcpy(state->window, strm->next_out - copy, copy);
363 1.1 christos state->write = copy;
364 1.1 christos state->whave = state->wsize;
365 1.1 christos }
366 1.1 christos else {
367 1.1 christos state->write += dist;
368 1.1 christos if (state->write == state->wsize) state->write = 0;
369 1.1 christos if (state->whave < state->wsize) state->whave += dist;
370 1.1 christos }
371 1.1 christos }
372 1.1 christos return 0;
373 1.1 christos }
374 1.1 christos
375 1.1 christos /* Macros for inflate(): */
376 1.1 christos
377 1.1 christos /* check function to use adler32() for zlib or crc32() for gzip */
378 1.1 christos #ifdef GUNZIP
379 1.1 christos # define UPDATE(check, buf, len) \
380 1.1 christos (state->flags ? crc32(check, buf, len) : adler32(check, buf, len))
381 1.1 christos #else
382 1.1 christos # define UPDATE(check, buf, len) adler32(check, buf, len)
383 1.1 christos #endif
384 1.1 christos
385 1.1 christos /* check macros for header crc */
386 1.1 christos #ifdef GUNZIP
387 1.1 christos # define CRC2(check, word) \
388 1.1 christos do { \
389 1.1 christos hbuf[0] = (unsigned char)(word); \
390 1.1 christos hbuf[1] = (unsigned char)((word) >> 8); \
391 1.1 christos check = crc32(check, hbuf, 2); \
392 1.1 christos } while (0)
393 1.1 christos
394 1.1 christos # define CRC4(check, word) \
395 1.1 christos do { \
396 1.1 christos hbuf[0] = (unsigned char)(word); \
397 1.1 christos hbuf[1] = (unsigned char)((word) >> 8); \
398 1.1 christos hbuf[2] = (unsigned char)((word) >> 16); \
399 1.1 christos hbuf[3] = (unsigned char)((word) >> 24); \
400 1.1 christos check = crc32(check, hbuf, 4); \
401 1.1 christos } while (0)
402 1.1 christos #endif
403 1.1 christos
404 1.1 christos /* Load registers with state in inflate() for speed */
405 1.1 christos #define LOAD() \
406 1.1 christos do { \
407 1.1 christos put = strm->next_out; \
408 1.1 christos left = strm->avail_out; \
409 1.1 christos next = strm->next_in; \
410 1.1 christos have = strm->avail_in; \
411 1.1 christos hold = state->hold; \
412 1.1 christos bits = state->bits; \
413 1.1 christos } while (0)
414 1.1 christos
415 1.1 christos /* Restore state from registers in inflate() */
416 1.1 christos #define RESTORE() \
417 1.1 christos do { \
418 1.1 christos strm->next_out = put; \
419 1.1 christos strm->avail_out = left; \
420 1.1 christos strm->next_in = next; \
421 1.1 christos strm->avail_in = have; \
422 1.1 christos state->hold = hold; \
423 1.1 christos state->bits = bits; \
424 1.1 christos } while (0)
425 1.1 christos
426 1.1 christos /* Clear the input bit accumulator */
427 1.1 christos #define INITBITS() \
428 1.1 christos do { \
429 1.1 christos hold = 0; \
430 1.1 christos bits = 0; \
431 1.1 christos } while (0)
432 1.1 christos
433 1.1 christos /* Get a byte of input into the bit accumulator, or return from inflate()
434 1.1 christos if there is no input available. */
435 1.1 christos #define PULLBYTE() \
436 1.1 christos do { \
437 1.1 christos if (have == 0) goto inf_leave; \
438 1.1 christos have--; \
439 1.1 christos hold += (unsigned long)(*next++) << bits; \
440 1.1 christos bits += 8; \
441 1.1 christos } while (0)
442 1.1 christos
443 1.1 christos /* Assure that there are at least n bits in the bit accumulator. If there is
444 1.1 christos not enough available input to do that, then return from inflate(). */
445 1.1 christos #define NEEDBITS(n) \
446 1.1 christos do { \
447 1.1 christos while (bits < (unsigned)(n)) \
448 1.1 christos PULLBYTE(); \
449 1.1 christos } while (0)
450 1.1 christos
451 1.1 christos /* Return the low n bits of the bit accumulator (n < 16) */
452 1.1 christos #define BITS(n) \
453 1.1 christos ((unsigned)hold & ((1U << (n)) - 1))
454 1.1 christos
455 1.1 christos /* Remove n bits from the bit accumulator */
456 1.1 christos #define DROPBITS(n) \
457 1.1 christos do { \
458 1.1 christos hold >>= (n); \
459 1.1 christos bits -= (unsigned)(n); \
460 1.1 christos } while (0)
461 1.1 christos
462 1.1 christos /* Remove zero to seven bits as needed to go to a byte boundary */
463 1.1 christos #define BYTEBITS() \
464 1.1 christos do { \
465 1.1 christos hold >>= bits & 7; \
466 1.1 christos bits -= bits & 7; \
467 1.1 christos } while (0)
468 1.1 christos
469 1.1 christos /* Reverse the bytes in a 32-bit value */
470 1.1 christos #define REVERSE(q) \
471 1.1 christos ((((q) >> 24) & 0xff) + (((q) >> 8) & 0xff00) + \
472 1.1 christos (((q) & 0xff00) << 8) + (((q) & 0xff) << 24))
473 1.1 christos
474 1.1 christos /*
475 1.1 christos inflate() uses a state machine to process as much input data and generate as
476 1.1 christos much output data as possible before returning. The state machine is
477 1.1 christos structured roughly as follows:
478 1.1 christos
479 1.1 christos for (;;) switch (state) {
480 1.1 christos ...
481 1.1 christos case STATEn:
482 1.1 christos if (not enough input data or output space to make progress)
483 1.1 christos return;
484 1.1 christos ... make progress ...
485 1.1 christos state = STATEm;
486 1.1 christos break;
487 1.1 christos ...
488 1.1 christos }
489 1.1 christos
490 1.1 christos so when inflate() is called again, the same case is attempted again, and
491 1.1 christos if the appropriate resources are provided, the machine proceeds to the
492 1.1 christos next state. The NEEDBITS() macro is usually the way the state evaluates
493 1.1 christos whether it can proceed or should return. NEEDBITS() does the return if
494 1.1 christos the requested bits are not available. The typical use of the BITS macros
495 1.1 christos is:
496 1.1 christos
497 1.1 christos NEEDBITS(n);
498 1.1 christos ... do something with BITS(n) ...
499 1.1 christos DROPBITS(n);
500 1.1 christos
501 1.1 christos where NEEDBITS(n) either returns from inflate() if there isn't enough
502 1.1 christos input left to load n bits into the accumulator, or it continues. BITS(n)
503 1.1 christos gives the low n bits in the accumulator. When done, DROPBITS(n) drops
504 1.1 christos the low n bits off the accumulator. INITBITS() clears the accumulator
505 1.1 christos and sets the number of available bits to zero. BYTEBITS() discards just
506 1.1 christos enough bits to put the accumulator on a byte boundary. After BYTEBITS()
507 1.1 christos and a NEEDBITS(8), then BITS(8) would return the next byte in the stream.
508 1.1 christos
509 1.1 christos NEEDBITS(n) uses PULLBYTE() to get an available byte of input, or to return
510 1.1 christos if there is no input available. The decoding of variable length codes uses
511 1.1 christos PULLBYTE() directly in order to pull just enough bytes to decode the next
512 1.1 christos code, and no more.
513 1.1 christos
514 1.1 christos Some states loop until they get enough input, making sure that enough
515 1.1 christos state information is maintained to continue the loop where it left off
516 1.1 christos if NEEDBITS() returns in the loop. For example, want, need, and keep
517 1.1 christos would all have to actually be part of the saved state in case NEEDBITS()
518 1.1 christos returns:
519 1.1 christos
520 1.1 christos case STATEw:
521 1.1 christos while (want < need) {
522 1.1 christos NEEDBITS(n);
523 1.1 christos keep[want++] = BITS(n);
524 1.1 christos DROPBITS(n);
525 1.1 christos }
526 1.1 christos state = STATEx;
527 1.1 christos case STATEx:
528 1.1 christos
529 1.1 christos As shown above, if the next state is also the next case, then the break
530 1.1 christos is omitted.
531 1.1 christos
532 1.1 christos A state may also return if there is not enough output space available to
533 1.1 christos complete that state. Those states are copying stored data, writing a
534 1.1 christos literal byte, and copying a matching string.
535 1.1 christos
536 1.1 christos When returning, a "goto inf_leave" is used to update the total counters,
537 1.1 christos update the check value, and determine whether any progress has been made
538 1.1 christos during that inflate() call in order to return the proper return code.
539 1.1 christos Progress is defined as a change in either strm->avail_in or strm->avail_out.
540 1.1 christos When there is a window, goto inf_leave will update the window with the last
541 1.1 christos output written. If a goto inf_leave occurs in the middle of decompression
542 1.1 christos and there is no window currently, goto inf_leave will create one and copy
543 1.1 christos output to the window for the next call of inflate().
544 1.1 christos
545 1.1 christos In this implementation, the flush parameter of inflate() only affects the
546 1.1 christos return code (per zlib.h). inflate() always writes as much as possible to
547 1.1 christos strm->next_out, given the space available and the provided input--the effect
548 1.1 christos documented in zlib.h of Z_SYNC_FLUSH. Furthermore, inflate() always defers
549 1.1 christos the allocation of and copying into a sliding window until necessary, which
550 1.1 christos provides the effect documented in zlib.h for Z_FINISH when the entire input
551 1.1 christos stream available. So the only thing the flush parameter actually does is:
552 1.1 christos when flush is set to Z_FINISH, inflate() cannot return Z_OK. Instead it
553 1.1 christos will return Z_BUF_ERROR if it has not reached the end of the stream.
554 1.1 christos */
555 1.1 christos
556 1.1 christos int ZEXPORT inflate(strm, flush)
557 1.1 christos z_streamp strm;
558 1.1 christos int flush;
559 1.1 christos {
560 1.1 christos struct inflate_state FAR *state;
561 1.1 christos unsigned char FAR *next; /* next input */
562 1.1 christos unsigned char FAR *put; /* next output */
563 1.1 christos unsigned have, left; /* available input and output */
564 1.1 christos unsigned long hold; /* bit buffer */
565 1.1 christos unsigned bits; /* bits in bit buffer */
566 1.1 christos unsigned in, out; /* save starting available input and output */
567 1.1 christos unsigned copy; /* number of stored or match bytes to copy */
568 1.1 christos unsigned char FAR *from; /* where to copy match bytes from */
569 1.1 christos code this; /* current decoding table entry */
570 1.1 christos code last; /* parent table entry */
571 1.1 christos unsigned len; /* length to copy for repeats, bits to drop */
572 1.1 christos int ret; /* return code */
573 1.1 christos #ifdef GUNZIP
574 1.1 christos unsigned char hbuf[4]; /* buffer for gzip header crc calculation */
575 1.1 christos #endif
576 1.1 christos static const unsigned short order[19] = /* permutation of code lengths */
577 1.1 christos {16, 17, 18, 0, 8, 7, 9, 6, 10, 5, 11, 4, 12, 3, 13, 2, 14, 1, 15};
578 1.1 christos
579 1.1 christos if (strm == Z_NULL || strm->state == Z_NULL || strm->next_out == Z_NULL ||
580 1.1 christos (strm->next_in == Z_NULL && strm->avail_in != 0))
581 1.1 christos return Z_STREAM_ERROR;
582 1.1 christos
583 1.1 christos state = (struct inflate_state FAR *)strm->state;
584 1.1 christos if (state->mode == TYPE) state->mode = TYPEDO; /* skip check */
585 1.1 christos LOAD();
586 1.1 christos in = have;
587 1.1 christos out = left;
588 1.1 christos ret = Z_OK;
589 1.1 christos for (;;)
590 1.1 christos switch (state->mode) {
591 1.1 christos case HEAD:
592 1.1 christos if (state->wrap == 0) {
593 1.1 christos state->mode = TYPEDO;
594 1.1 christos break;
595 1.1 christos }
596 1.1 christos NEEDBITS(16);
597 1.1 christos #ifdef GUNZIP
598 1.1 christos if ((state->wrap & 2) && hold == 0x8b1f) { /* gzip header */
599 1.1 christos state->check = crc32(0L, Z_NULL, 0);
600 1.1 christos CRC2(state->check, hold);
601 1.1 christos INITBITS();
602 1.1 christos state->mode = FLAGS;
603 1.1 christos break;
604 1.1 christos }
605 1.1 christos state->flags = 0; /* expect zlib header */
606 1.1 christos if (state->head != Z_NULL)
607 1.1 christos state->head->done = -1;
608 1.1 christos if (!(state->wrap & 1) || /* check if zlib header allowed */
609 1.1 christos #else
610 1.1 christos if (
611 1.1 christos #endif
612 1.1 christos ((BITS(8) << 8) + (hold >> 8)) % 31) {
613 1.1 christos strm->msg = (char *)"incorrect header check";
614 1.1 christos state->mode = BAD;
615 1.1 christos break;
616 1.1 christos }
617 1.1 christos if (BITS(4) != Z_DEFLATED) {
618 1.1 christos strm->msg = (char *)"unknown compression method";
619 1.1 christos state->mode = BAD;
620 1.1 christos break;
621 1.1 christos }
622 1.1 christos DROPBITS(4);
623 1.1 christos len = BITS(4) + 8;
624 1.1 christos if (len > state->wbits) {
625 1.1 christos strm->msg = (char *)"invalid window size";
626 1.1 christos state->mode = BAD;
627 1.1 christos break;
628 1.1 christos }
629 1.1 christos state->dmax = 1U << len;
630 1.1 christos Tracev((stderr, "inflate: zlib header ok\n"));
631 1.1 christos strm->adler = state->check = adler32(0L, Z_NULL, 0);
632 1.1 christos state->mode = hold & 0x200 ? DICTID : TYPE;
633 1.1 christos INITBITS();
634 1.1 christos break;
635 1.1 christos #ifdef GUNZIP
636 1.1 christos case FLAGS:
637 1.1 christos NEEDBITS(16);
638 1.1 christos state->flags = (int)(hold);
639 1.1 christos if ((state->flags & 0xff) != Z_DEFLATED) {
640 1.1 christos strm->msg = (char *)"unknown compression method";
641 1.1 christos state->mode = BAD;
642 1.1 christos break;
643 1.1 christos }
644 1.1 christos if (state->flags & 0xe000) {
645 1.1 christos strm->msg = (char *)"unknown header flags set";
646 1.1 christos state->mode = BAD;
647 1.1 christos break;
648 1.1 christos }
649 1.1 christos if (state->head != Z_NULL)
650 1.1 christos state->head->text = (int)((hold >> 8) & 1);
651 1.1 christos if (state->flags & 0x0200) CRC2(state->check, hold);
652 1.1 christos INITBITS();
653 1.1 christos state->mode = TIME;
654 1.1 christos case TIME:
655 1.1 christos NEEDBITS(32);
656 1.1 christos if (state->head != Z_NULL)
657 1.1 christos state->head->time = hold;
658 1.1 christos if (state->flags & 0x0200) CRC4(state->check, hold);
659 1.1 christos INITBITS();
660 1.1 christos state->mode = OS;
661 1.1 christos case OS:
662 1.1 christos NEEDBITS(16);
663 1.1 christos if (state->head != Z_NULL) {
664 1.1 christos state->head->xflags = (int)(hold & 0xff);
665 1.1 christos state->head->os = (int)(hold >> 8);
666 1.1 christos }
667 1.1 christos if (state->flags & 0x0200) CRC2(state->check, hold);
668 1.1 christos INITBITS();
669 1.1 christos state->mode = EXLEN;
670 1.1 christos case EXLEN:
671 1.1 christos if (state->flags & 0x0400) {
672 1.1 christos NEEDBITS(16);
673 1.1 christos state->length = (unsigned)(hold);
674 1.1 christos if (state->head != Z_NULL)
675 1.1 christos state->head->extra_len = (unsigned)hold;
676 1.1 christos if (state->flags & 0x0200) CRC2(state->check, hold);
677 1.1 christos INITBITS();
678 1.1 christos }
679 1.1 christos else if (state->head != Z_NULL)
680 1.1 christos state->head->extra = Z_NULL;
681 1.1 christos state->mode = EXTRA;
682 1.1 christos case EXTRA:
683 1.1 christos if (state->flags & 0x0400) {
684 1.1 christos copy = state->length;
685 1.1 christos if (copy > have) copy = have;
686 1.1 christos if (copy) {
687 1.1 christos if (state->head != Z_NULL &&
688 1.1 christos state->head->extra != Z_NULL) {
689 1.1 christos len = state->head->extra_len - state->length;
690 1.1 christos zmemcpy(state->head->extra + len, next,
691 1.1 christos len + copy > state->head->extra_max ?
692 1.1 christos state->head->extra_max - len : copy);
693 1.1 christos }
694 1.1 christos if (state->flags & 0x0200)
695 1.1 christos state->check = crc32(state->check, next, copy);
696 1.1 christos have -= copy;
697 1.1 christos next += copy;
698 1.1 christos state->length -= copy;
699 1.1 christos }
700 1.1 christos if (state->length) goto inf_leave;
701 1.1 christos }
702 1.1 christos state->length = 0;
703 1.1 christos state->mode = NAME;
704 1.1 christos case NAME:
705 1.1 christos if (state->flags & 0x0800) {
706 1.1 christos if (have == 0) goto inf_leave;
707 1.1 christos copy = 0;
708 1.1 christos do {
709 1.1 christos len = (unsigned)(next[copy++]);
710 1.1 christos if (state->head != Z_NULL &&
711 1.1 christos state->head->name != Z_NULL &&
712 1.1 christos state->length < state->head->name_max)
713 1.1 christos state->head->name[state->length++] = len;
714 1.1 christos } while (len && copy < have);
715 1.1 christos if (state->flags & 0x0200)
716 1.1 christos state->check = crc32(state->check, next, copy);
717 1.1 christos have -= copy;
718 1.1 christos next += copy;
719 1.1 christos if (len) goto inf_leave;
720 1.1 christos }
721 1.1 christos else if (state->head != Z_NULL)
722 1.1 christos state->head->name = Z_NULL;
723 1.1 christos state->length = 0;
724 1.1 christos state->mode = COMMENT;
725 1.1 christos case COMMENT:
726 1.1 christos if (state->flags & 0x1000) {
727 1.1 christos if (have == 0) goto inf_leave;
728 1.1 christos copy = 0;
729 1.1 christos do {
730 1.1 christos len = (unsigned)(next[copy++]);
731 1.1 christos if (state->head != Z_NULL &&
732 1.1 christos state->head->comment != Z_NULL &&
733 1.1 christos state->length < state->head->comm_max)
734 1.1 christos state->head->comment[state->length++] = len;
735 1.1 christos } while (len && copy < have);
736 1.1 christos if (state->flags & 0x0200)
737 1.1 christos state->check = crc32(state->check, next, copy);
738 1.1 christos have -= copy;
739 1.1 christos next += copy;
740 1.1 christos if (len) goto inf_leave;
741 1.1 christos }
742 1.1 christos else if (state->head != Z_NULL)
743 1.1 christos state->head->comment = Z_NULL;
744 1.1 christos state->mode = HCRC;
745 1.1 christos case HCRC:
746 1.1 christos if (state->flags & 0x0200) {
747 1.1 christos NEEDBITS(16);
748 1.1 christos if (hold != (state->check & 0xffff)) {
749 1.1 christos strm->msg = (char *)"header crc mismatch";
750 1.1 christos state->mode = BAD;
751 1.1 christos break;
752 1.1 christos }
753 1.1 christos INITBITS();
754 1.1 christos }
755 1.1 christos if (state->head != Z_NULL) {
756 1.1 christos state->head->hcrc = (int)((state->flags >> 9) & 1);
757 1.1 christos state->head->done = 1;
758 1.1 christos }
759 1.1 christos strm->adler = state->check = crc32(0L, Z_NULL, 0);
760 1.1 christos state->mode = TYPE;
761 1.1 christos break;
762 1.1 christos #endif
763 1.1 christos case DICTID:
764 1.1 christos NEEDBITS(32);
765 1.1 christos strm->adler = state->check = REVERSE(hold);
766 1.1 christos INITBITS();
767 1.1 christos state->mode = DICT;
768 1.1 christos case DICT:
769 1.1 christos if (state->havedict == 0) {
770 1.1 christos RESTORE();
771 1.1 christos return Z_NEED_DICT;
772 1.1 christos }
773 1.1 christos strm->adler = state->check = adler32(0L, Z_NULL, 0);
774 1.1 christos state->mode = TYPE;
775 1.1 christos case TYPE:
776 1.1 christos if (flush == Z_BLOCK) goto inf_leave;
777 1.1 christos case TYPEDO:
778 1.1 christos if (state->last) {
779 1.1 christos BYTEBITS();
780 1.1 christos state->mode = CHECK;
781 1.1 christos break;
782 1.1 christos }
783 1.1 christos NEEDBITS(3);
784 1.1 christos state->last = BITS(1);
785 1.1 christos DROPBITS(1);
786 1.1 christos switch (BITS(2)) {
787 1.1 christos case 0: /* stored block */
788 1.1 christos Tracev((stderr, "inflate: stored block%s\n",
789 1.1 christos state->last ? " (last)" : ""));
790 1.1 christos state->mode = STORED;
791 1.1 christos break;
792 1.1 christos case 1: /* fixed block */
793 1.1 christos fixedtables(state);
794 1.1 christos Tracev((stderr, "inflate: fixed codes block%s\n",
795 1.1 christos state->last ? " (last)" : ""));
796 1.1 christos state->mode = LEN; /* decode codes */
797 1.1 christos break;
798 1.1 christos case 2: /* dynamic block */
799 1.1 christos Tracev((stderr, "inflate: dynamic codes block%s\n",
800 1.1 christos state->last ? " (last)" : ""));
801 1.1 christos state->mode = TABLE;
802 1.1 christos break;
803 1.1 christos case 3:
804 1.1 christos strm->msg = (char *)"invalid block type";
805 1.1 christos state->mode = BAD;
806 1.1 christos }
807 1.1 christos DROPBITS(2);
808 1.1 christos break;
809 1.1 christos case STORED:
810 1.1 christos BYTEBITS(); /* go to byte boundary */
811 1.1 christos NEEDBITS(32);
812 1.1 christos if ((hold & 0xffff) != ((hold >> 16) ^ 0xffff)) {
813 1.1 christos strm->msg = (char *)"invalid stored block lengths";
814 1.1 christos state->mode = BAD;
815 1.1 christos break;
816 1.1 christos }
817 1.1 christos state->length = (unsigned)hold & 0xffff;
818 1.1 christos Tracev((stderr, "inflate: stored length %u\n",
819 1.1 christos state->length));
820 1.1 christos INITBITS();
821 1.1 christos state->mode = COPY;
822 1.1 christos case COPY:
823 1.1 christos copy = state->length;
824 1.1 christos if (copy) {
825 1.1 christos if (copy > have) copy = have;
826 1.1 christos if (copy > left) copy = left;
827 1.1 christos if (copy == 0) goto inf_leave;
828 1.1 christos zmemcpy(put, next, copy);
829 1.1 christos have -= copy;
830 1.1 christos next += copy;
831 1.1 christos left -= copy;
832 1.1 christos put += copy;
833 1.1 christos state->length -= copy;
834 1.1 christos break;
835 1.1 christos }
836 1.1 christos Tracev((stderr, "inflate: stored end\n"));
837 1.1 christos state->mode = TYPE;
838 1.1 christos break;
839 1.1 christos case TABLE:
840 1.1 christos NEEDBITS(14);
841 1.1 christos state->nlen = BITS(5) + 257;
842 1.1 christos DROPBITS(5);
843 1.1 christos state->ndist = BITS(5) + 1;
844 1.1 christos DROPBITS(5);
845 1.1 christos state->ncode = BITS(4) + 4;
846 1.1 christos DROPBITS(4);
847 1.1 christos #ifndef PKZIP_BUG_WORKAROUND
848 1.1 christos if (state->nlen > 286 || state->ndist > 30) {
849 1.1 christos strm->msg = (char *)"too many length or distance symbols";
850 1.1 christos state->mode = BAD;
851 1.1 christos break;
852 1.1 christos }
853 1.1 christos #endif
854 1.1 christos Tracev((stderr, "inflate: table sizes ok\n"));
855 1.1 christos state->have = 0;
856 1.1 christos state->mode = LENLENS;
857 1.1 christos case LENLENS:
858 1.1 christos while (state->have < state->ncode) {
859 1.1 christos NEEDBITS(3);
860 1.1 christos state->lens[order[state->have++]] = (unsigned short)BITS(3);
861 1.1 christos DROPBITS(3);
862 1.1 christos }
863 1.1 christos while (state->have < 19)
864 1.1 christos state->lens[order[state->have++]] = 0;
865 1.1 christos state->next = state->codes;
866 1.1 christos state->lencode = (code const FAR *)(state->next);
867 1.1 christos state->lenbits = 7;
868 1.1 christos ret = inflate_table(CODES, state->lens, 19, &(state->next),
869 1.1 christos &(state->lenbits), state->work);
870 1.1 christos if (ret) {
871 1.1 christos strm->msg = (char *)"invalid code lengths set";
872 1.1 christos state->mode = BAD;
873 1.1 christos break;
874 1.1 christos }
875 1.1 christos Tracev((stderr, "inflate: code lengths ok\n"));
876 1.1 christos state->have = 0;
877 1.1 christos state->mode = CODELENS;
878 1.1 christos case CODELENS:
879 1.1 christos while (state->have < state->nlen + state->ndist) {
880 1.1 christos for (;;) {
881 1.1 christos this = state->lencode[BITS(state->lenbits)];
882 1.1 christos if ((unsigned)(this.bits) <= bits) break;
883 1.1 christos PULLBYTE();
884 1.1 christos }
885 1.1 christos if (this.val < 16) {
886 1.1 christos NEEDBITS(this.bits);
887 1.1 christos DROPBITS(this.bits);
888 1.1 christos state->lens[state->have++] = this.val;
889 1.1 christos }
890 1.1 christos else {
891 1.1 christos if (this.val == 16) {
892 1.1 christos NEEDBITS(this.bits + 2);
893 1.1 christos DROPBITS(this.bits);
894 1.1 christos if (state->have == 0) {
895 1.1 christos strm->msg = (char *)"invalid bit length repeat";
896 1.1 christos state->mode = BAD;
897 1.1 christos break;
898 1.1 christos }
899 1.1 christos len = state->lens[state->have - 1];
900 1.1 christos copy = 3 + BITS(2);
901 1.1 christos DROPBITS(2);
902 1.1 christos }
903 1.1 christos else if (this.val == 17) {
904 1.1 christos NEEDBITS(this.bits + 3);
905 1.1 christos DROPBITS(this.bits);
906 1.1 christos len = 0;
907 1.1 christos copy = 3 + BITS(3);
908 1.1 christos DROPBITS(3);
909 1.1 christos }
910 1.1 christos else {
911 1.1 christos NEEDBITS(this.bits + 7);
912 1.1 christos DROPBITS(this.bits);
913 1.1 christos len = 0;
914 1.1 christos copy = 11 + BITS(7);
915 1.1 christos DROPBITS(7);
916 1.1 christos }
917 1.1 christos if (state->have + copy > state->nlen + state->ndist) {
918 1.1 christos strm->msg = (char *)"invalid bit length repeat";
919 1.1 christos state->mode = BAD;
920 1.1 christos break;
921 1.1 christos }
922 1.1 christos while (copy--)
923 1.1 christos state->lens[state->have++] = (unsigned short)len;
924 1.1 christos }
925 1.1 christos }
926 1.1 christos
927 1.1 christos /* handle error breaks in while */
928 1.1 christos if (state->mode == BAD) break;
929 1.1 christos
930 1.1 christos /* build code tables */
931 1.1 christos state->next = state->codes;
932 1.1 christos state->lencode = (code const FAR *)(state->next);
933 1.1 christos state->lenbits = 9;
934 1.1 christos ret = inflate_table(LENS, state->lens, state->nlen, &(state->next),
935 1.1 christos &(state->lenbits), state->work);
936 1.1 christos if (ret) {
937 1.1 christos strm->msg = (char *)"invalid literal/lengths set";
938 1.1 christos state->mode = BAD;
939 1.1 christos break;
940 1.1 christos }
941 1.1 christos state->distcode = (code const FAR *)(state->next);
942 1.1 christos state->distbits = 6;
943 1.1 christos ret = inflate_table(DISTS, state->lens + state->nlen, state->ndist,
944 1.1 christos &(state->next), &(state->distbits), state->work);
945 1.1 christos if (ret) {
946 1.1 christos strm->msg = (char *)"invalid distances set";
947 1.1 christos state->mode = BAD;
948 1.1 christos break;
949 1.1 christos }
950 1.1 christos Tracev((stderr, "inflate: codes ok\n"));
951 1.1 christos state->mode = LEN;
952 1.1 christos case LEN:
953 1.1 christos if (have >= 6 && left >= 258) {
954 1.1 christos RESTORE();
955 1.1 christos inflate_fast(strm, out);
956 1.1 christos LOAD();
957 1.1 christos break;
958 1.1 christos }
959 1.1 christos for (;;) {
960 1.1 christos this = state->lencode[BITS(state->lenbits)];
961 1.1 christos if ((unsigned)(this.bits) <= bits) break;
962 1.1 christos PULLBYTE();
963 1.1 christos }
964 1.1 christos if (this.op && (this.op & 0xf0) == 0) {
965 1.1 christos last = this;
966 1.1 christos for (;;) {
967 1.1 christos this = state->lencode[last.val +
968 1.1 christos (BITS(last.bits + last.op) >> last.bits)];
969 1.1 christos if ((unsigned)(last.bits + this.bits) <= bits) break;
970 1.1 christos PULLBYTE();
971 1.1 christos }
972 1.1 christos DROPBITS(last.bits);
973 1.1 christos }
974 1.1 christos DROPBITS(this.bits);
975 1.1 christos state->length = (unsigned)this.val;
976 1.1 christos if ((int)(this.op) == 0) {
977 1.1 christos Tracevv((stderr, this.val >= 0x20 && this.val < 0x7f ?
978 1.1 christos "inflate: literal '%c'\n" :
979 1.1 christos "inflate: literal 0x%02x\n", this.val));
980 1.1 christos state->mode = LIT;
981 1.1 christos break;
982 1.1 christos }
983 1.1 christos if (this.op & 32) {
984 1.1 christos Tracevv((stderr, "inflate: end of block\n"));
985 1.1 christos state->mode = TYPE;
986 1.1 christos break;
987 1.1 christos }
988 1.1 christos if (this.op & 64) {
989 1.1 christos strm->msg = (char *)"invalid literal/length code";
990 1.1 christos state->mode = BAD;
991 1.1 christos break;
992 1.1 christos }
993 1.1 christos state->extra = (unsigned)(this.op) & 15;
994 1.1 christos state->mode = LENEXT;
995 1.1 christos case LENEXT:
996 1.1 christos if (state->extra) {
997 1.1 christos NEEDBITS(state->extra);
998 1.1 christos state->length += BITS(state->extra);
999 1.1 christos DROPBITS(state->extra);
1000 1.1 christos }
1001 1.1 christos Tracevv((stderr, "inflate: length %u\n", state->length));
1002 1.1 christos state->mode = DIST;
1003 1.1 christos case DIST:
1004 1.1 christos for (;;) {
1005 1.1 christos this = state->distcode[BITS(state->distbits)];
1006 1.1 christos if ((unsigned)(this.bits) <= bits) break;
1007 1.1 christos PULLBYTE();
1008 1.1 christos }
1009 1.1 christos if ((this.op & 0xf0) == 0) {
1010 1.1 christos last = this;
1011 1.1 christos for (;;) {
1012 1.1 christos this = state->distcode[last.val +
1013 1.1 christos (BITS(last.bits + last.op) >> last.bits)];
1014 1.1 christos if ((unsigned)(last.bits + this.bits) <= bits) break;
1015 1.1 christos PULLBYTE();
1016 1.1 christos }
1017 1.1 christos DROPBITS(last.bits);
1018 1.1 christos }
1019 1.1 christos DROPBITS(this.bits);
1020 1.1 christos if (this.op & 64) {
1021 1.1 christos strm->msg = (char *)"invalid distance code";
1022 1.1 christos state->mode = BAD;
1023 1.1 christos break;
1024 1.1 christos }
1025 1.1 christos state->offset = (unsigned)this.val;
1026 1.1 christos state->extra = (unsigned)(this.op) & 15;
1027 1.1 christos state->mode = DISTEXT;
1028 1.1 christos case DISTEXT:
1029 1.1 christos if (state->extra) {
1030 1.1 christos NEEDBITS(state->extra);
1031 1.1 christos state->offset += BITS(state->extra);
1032 1.1 christos DROPBITS(state->extra);
1033 1.1 christos }
1034 1.1 christos #ifdef INFLATE_STRICT
1035 1.1 christos if (state->offset > state->dmax) {
1036 1.1 christos strm->msg = (char *)"invalid distance too far back";
1037 1.1 christos state->mode = BAD;
1038 1.1 christos break;
1039 1.1 christos }
1040 1.1 christos #endif
1041 1.1 christos if (state->offset > state->whave + out - left) {
1042 1.1 christos strm->msg = (char *)"invalid distance too far back";
1043 1.1 christos state->mode = BAD;
1044 1.1 christos break;
1045 1.1 christos }
1046 1.1 christos Tracevv((stderr, "inflate: distance %u\n", state->offset));
1047 1.1 christos state->mode = MATCH;
1048 1.1 christos case MATCH:
1049 1.1 christos if (left == 0) goto inf_leave;
1050 1.1 christos copy = out - left;
1051 1.1 christos if (state->offset > copy) { /* copy from window */
1052 1.1 christos copy = state->offset - copy;
1053 1.1 christos if (copy > state->write) {
1054 1.1 christos copy -= state->write;
1055 1.1 christos from = state->window + (state->wsize - copy);
1056 1.1 christos }
1057 1.1 christos else
1058 1.1 christos from = state->window + (state->write - copy);
1059 1.1 christos if (copy > state->length) copy = state->length;
1060 1.1 christos }
1061 1.1 christos else { /* copy from output */
1062 1.1 christos from = put - state->offset;
1063 1.1 christos copy = state->length;
1064 1.1 christos }
1065 1.1 christos if (copy > left) copy = left;
1066 1.1 christos left -= copy;
1067 1.1 christos state->length -= copy;
1068 1.1 christos do {
1069 1.1 christos *put++ = *from++;
1070 1.1 christos } while (--copy);
1071 1.1 christos if (state->length == 0) state->mode = LEN;
1072 1.1 christos break;
1073 1.1 christos case LIT:
1074 1.1 christos if (left == 0) goto inf_leave;
1075 1.1 christos *put++ = (unsigned char)(state->length);
1076 1.1 christos left--;
1077 1.1 christos state->mode = LEN;
1078 1.1 christos break;
1079 1.1 christos case CHECK:
1080 1.1 christos if (state->wrap) {
1081 1.1 christos NEEDBITS(32);
1082 1.1 christos out -= left;
1083 1.1 christos strm->total_out += out;
1084 1.1 christos state->total += out;
1085 1.1 christos if (out)
1086 1.1 christos strm->adler = state->check =
1087 1.1 christos UPDATE(state->check, put - out, out);
1088 1.1 christos out = left;
1089 1.1 christos if ((
1090 1.1 christos #ifdef GUNZIP
1091 1.1 christos state->flags ? hold :
1092 1.1 christos #endif
1093 1.1 christos REVERSE(hold)) != state->check) {
1094 1.1 christos strm->msg = (char *)"incorrect data check";
1095 1.1 christos state->mode = BAD;
1096 1.1 christos break;
1097 1.1 christos }
1098 1.1 christos INITBITS();
1099 1.1 christos Tracev((stderr, "inflate: check matches trailer\n"));
1100 1.1 christos }
1101 1.1 christos #ifdef GUNZIP
1102 1.1 christos state->mode = LENGTH;
1103 1.1 christos case LENGTH:
1104 1.1 christos if (state->wrap && state->flags) {
1105 1.1 christos NEEDBITS(32);
1106 1.1 christos if (hold != (state->total & 0xffffffffUL)) {
1107 1.1 christos strm->msg = (char *)"incorrect length check";
1108 1.1 christos state->mode = BAD;
1109 1.1 christos break;
1110 1.1 christos }
1111 1.1 christos INITBITS();
1112 1.1 christos Tracev((stderr, "inflate: length matches trailer\n"));
1113 1.1 christos }
1114 1.1 christos #endif
1115 1.1 christos state->mode = DONE;
1116 1.1 christos case DONE:
1117 1.1 christos ret = Z_STREAM_END;
1118 1.1 christos goto inf_leave;
1119 1.1 christos case BAD:
1120 1.1 christos ret = Z_DATA_ERROR;
1121 1.1 christos goto inf_leave;
1122 1.1 christos case MEM:
1123 1.1 christos return Z_MEM_ERROR;
1124 1.1 christos case SYNC:
1125 1.1 christos default:
1126 1.1 christos return Z_STREAM_ERROR;
1127 1.1 christos }
1128 1.1 christos
1129 1.1 christos /*
1130 1.1 christos Return from inflate(), updating the total counts and the check value.
1131 1.1 christos If there was no progress during the inflate() call, return a buffer
1132 1.1 christos error. Call updatewindow() to create and/or update the window state.
1133 1.1 christos Note: a memory error from inflate() is non-recoverable.
1134 1.1 christos */
1135 1.1 christos inf_leave:
1136 1.1 christos RESTORE();
1137 1.1 christos if (state->wsize || (state->mode < CHECK && out != strm->avail_out))
1138 1.1 christos if (updatewindow(strm, out)) {
1139 1.1 christos state->mode = MEM;
1140 1.1 christos return Z_MEM_ERROR;
1141 1.1 christos }
1142 1.1 christos in -= strm->avail_in;
1143 1.1 christos out -= strm->avail_out;
1144 1.1 christos strm->total_in += in;
1145 1.1 christos strm->total_out += out;
1146 1.1 christos state->total += out;
1147 1.1 christos if (state->wrap && out)
1148 1.1 christos strm->adler = state->check =
1149 1.1 christos UPDATE(state->check, strm->next_out - out, out);
1150 1.1 christos strm->data_type = state->bits + (state->last ? 64 : 0) +
1151 1.1 christos (state->mode == TYPE ? 128 : 0);
1152 1.1 christos if (((in == 0 && out == 0) || flush == Z_FINISH) && ret == Z_OK)
1153 1.1 christos ret = Z_BUF_ERROR;
1154 1.1 christos return ret;
1155 1.1 christos }
1156 1.1 christos
1157 1.1 christos int ZEXPORT inflateEnd(strm)
1158 1.1 christos z_streamp strm;
1159 1.1 christos {
1160 1.1 christos struct inflate_state FAR *state;
1161 1.1 christos if (strm == Z_NULL || strm->state == Z_NULL || strm->zfree == (free_func)0)
1162 1.1 christos return Z_STREAM_ERROR;
1163 1.1 christos state = (struct inflate_state FAR *)strm->state;
1164 1.1 christos if (state->window != Z_NULL) ZFREE(strm, state->window);
1165 1.1 christos ZFREE(strm, strm->state);
1166 1.1 christos strm->state = Z_NULL;
1167 1.1 christos Tracev((stderr, "inflate: end\n"));
1168 1.1 christos return Z_OK;
1169 1.1 christos }
1170 1.1 christos
1171 1.1 christos int ZEXPORT inflateSetDictionary(strm, dictionary, dictLength)
1172 1.1 christos z_streamp strm;
1173 1.1 christos const Bytef *dictionary;
1174 1.1 christos uInt dictLength;
1175 1.1 christos {
1176 1.1 christos struct inflate_state FAR *state;
1177 1.1 christos unsigned long id;
1178 1.1 christos
1179 1.1 christos /* check state */
1180 1.1 christos if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
1181 1.1 christos state = (struct inflate_state FAR *)strm->state;
1182 1.1 christos if (state->wrap != 0 && state->mode != DICT)
1183 1.1 christos return Z_STREAM_ERROR;
1184 1.1 christos
1185 1.1 christos /* check for correct dictionary id */
1186 1.1 christos if (state->mode == DICT) {
1187 1.1 christos id = adler32(0L, Z_NULL, 0);
1188 1.1 christos id = adler32(id, dictionary, dictLength);
1189 1.1 christos if (id != state->check)
1190 1.1 christos return Z_DATA_ERROR;
1191 1.1 christos }
1192 1.1 christos
1193 1.1 christos /* copy dictionary to window */
1194 1.1 christos if (updatewindow(strm, strm->avail_out)) {
1195 1.1 christos state->mode = MEM;
1196 1.1 christos return Z_MEM_ERROR;
1197 1.1 christos }
1198 1.1 christos if (dictLength > state->wsize) {
1199 1.1 christos zmemcpy(state->window, dictionary + dictLength - state->wsize,
1200 1.1 christos state->wsize);
1201 1.1 christos state->whave = state->wsize;
1202 1.1 christos }
1203 1.1 christos else {
1204 1.1 christos zmemcpy(state->window + state->wsize - dictLength, dictionary,
1205 1.1 christos dictLength);
1206 1.1 christos state->whave = dictLength;
1207 1.1 christos }
1208 1.1 christos state->havedict = 1;
1209 1.1 christos Tracev((stderr, "inflate: dictionary set\n"));
1210 1.1 christos return Z_OK;
1211 1.1 christos }
1212 1.1 christos
1213 1.1 christos int ZEXPORT inflateGetHeader(strm, head)
1214 1.1 christos z_streamp strm;
1215 1.1 christos gz_headerp head;
1216 1.1 christos {
1217 1.1 christos struct inflate_state FAR *state;
1218 1.1 christos
1219 1.1 christos /* check state */
1220 1.1 christos if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
1221 1.1 christos state = (struct inflate_state FAR *)strm->state;
1222 1.1 christos if ((state->wrap & 2) == 0) return Z_STREAM_ERROR;
1223 1.1 christos
1224 1.1 christos /* save header structure */
1225 1.1 christos state->head = head;
1226 1.1 christos head->done = 0;
1227 1.1 christos return Z_OK;
1228 1.1 christos }
1229 1.1 christos
1230 1.1 christos /*
1231 1.1 christos Search buf[0..len-1] for the pattern: 0, 0, 0xff, 0xff. Return when found
1232 1.1 christos or when out of input. When called, *have is the number of pattern bytes
1233 1.1 christos found in order so far, in 0..3. On return *have is updated to the new
1234 1.1 christos state. If on return *have equals four, then the pattern was found and the
1235 1.1 christos return value is how many bytes were read including the last byte of the
1236 1.1 christos pattern. If *have is less than four, then the pattern has not been found
1237 1.1 christos yet and the return value is len. In the latter case, syncsearch() can be
1238 1.1 christos called again with more data and the *have state. *have is initialized to
1239 1.1 christos zero for the first call.
1240 1.1 christos */
1241 1.1 christos local unsigned syncsearch(have, buf, len)
1242 1.1 christos unsigned FAR *have;
1243 1.1 christos unsigned char FAR *buf;
1244 1.1 christos unsigned len;
1245 1.1 christos {
1246 1.1 christos unsigned got;
1247 1.1 christos unsigned next;
1248 1.1 christos
1249 1.1 christos got = *have;
1250 1.1 christos next = 0;
1251 1.1 christos while (next < len && got < 4) {
1252 1.1 christos if ((int)(buf[next]) == (got < 2 ? 0 : 0xff))
1253 1.1 christos got++;
1254 1.1 christos else if (buf[next])
1255 1.1 christos got = 0;
1256 1.1 christos else
1257 1.1 christos got = 4 - got;
1258 1.1 christos next++;
1259 1.1 christos }
1260 1.1 christos *have = got;
1261 1.1 christos return next;
1262 1.1 christos }
1263 1.1 christos
1264 1.1 christos int ZEXPORT inflateSync(strm)
1265 1.1 christos z_streamp strm;
1266 1.1 christos {
1267 1.1 christos unsigned len; /* number of bytes to look at or looked at */
1268 1.1 christos unsigned long in, out; /* temporary to save total_in and total_out */
1269 1.1 christos unsigned char buf[4]; /* to restore bit buffer to byte string */
1270 1.1 christos struct inflate_state FAR *state;
1271 1.1 christos
1272 1.1 christos /* check parameters */
1273 1.1 christos if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
1274 1.1 christos state = (struct inflate_state FAR *)strm->state;
1275 1.1 christos if (strm->avail_in == 0 && state->bits < 8) return Z_BUF_ERROR;
1276 1.1 christos
1277 1.1 christos /* if first time, start search in bit buffer */
1278 1.1 christos if (state->mode != SYNC) {
1279 1.1 christos state->mode = SYNC;
1280 1.1 christos state->hold <<= state->bits & 7;
1281 1.1 christos state->bits -= state->bits & 7;
1282 1.1 christos len = 0;
1283 1.1 christos while (state->bits >= 8) {
1284 1.1 christos buf[len++] = (unsigned char)(state->hold);
1285 1.1 christos state->hold >>= 8;
1286 1.1 christos state->bits -= 8;
1287 1.1 christos }
1288 1.1 christos state->have = 0;
1289 1.1 christos syncsearch(&(state->have), buf, len);
1290 1.1 christos }
1291 1.1 christos
1292 1.1 christos /* search available input */
1293 1.1 christos len = syncsearch(&(state->have), strm->next_in, strm->avail_in);
1294 1.1 christos strm->avail_in -= len;
1295 1.1 christos strm->next_in += len;
1296 1.1 christos strm->total_in += len;
1297 1.1 christos
1298 1.1 christos /* return no joy or set up to restart inflate() on a new block */
1299 1.1 christos if (state->have != 4) return Z_DATA_ERROR;
1300 1.1 christos in = strm->total_in; out = strm->total_out;
1301 1.1 christos inflateReset(strm);
1302 1.1 christos strm->total_in = in; strm->total_out = out;
1303 1.1 christos state->mode = TYPE;
1304 1.1 christos return Z_OK;
1305 1.1 christos }
1306 1.1 christos
1307 1.1 christos /*
1308 1.1 christos Returns true if inflate is currently at the end of a block generated by
1309 1.1 christos Z_SYNC_FLUSH or Z_FULL_FLUSH. This function is used by one PPP
1310 1.1 christos implementation to provide an additional safety check. PPP uses
1311 1.1 christos Z_SYNC_FLUSH but removes the length bytes of the resulting empty stored
1312 1.1 christos block. When decompressing, PPP checks that at the end of input packet,
1313 1.1 christos inflate is waiting for these length bytes.
1314 1.1 christos */
1315 1.1 christos int ZEXPORT inflateSyncPoint(strm)
1316 1.1 christos z_streamp strm;
1317 1.1 christos {
1318 1.1 christos struct inflate_state FAR *state;
1319 1.1 christos
1320 1.1 christos if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
1321 1.1 christos state = (struct inflate_state FAR *)strm->state;
1322 1.1 christos return state->mode == STORED && state->bits == 0;
1323 1.1 christos }
1324 1.1 christos
1325 1.1 christos int ZEXPORT inflateCopy(dest, source)
1326 1.1 christos z_streamp dest;
1327 1.1 christos z_streamp source;
1328 1.1 christos {
1329 1.1 christos struct inflate_state FAR *state;
1330 1.1 christos struct inflate_state FAR *copy;
1331 1.1 christos unsigned char FAR *window;
1332 1.1 christos unsigned wsize;
1333 1.1 christos
1334 1.1 christos /* check input */
1335 1.1 christos if (dest == Z_NULL || source == Z_NULL || source->state == Z_NULL ||
1336 1.1 christos source->zalloc == (alloc_func)0 || source->zfree == (free_func)0)
1337 1.1 christos return Z_STREAM_ERROR;
1338 1.1 christos state = (struct inflate_state FAR *)source->state;
1339 1.1 christos
1340 1.1 christos /* allocate space */
1341 1.1 christos copy = (struct inflate_state FAR *)
1342 1.1 christos ZALLOC(source, 1, sizeof(struct inflate_state));
1343 1.1 christos if (copy == Z_NULL) return Z_MEM_ERROR;
1344 1.1 christos window = Z_NULL;
1345 1.1 christos if (state->window != Z_NULL) {
1346 1.1 christos window = (unsigned char FAR *)
1347 1.1 christos ZALLOC(source, 1U << state->wbits, sizeof(unsigned char));
1348 1.1 christos if (window == Z_NULL) {
1349 1.1 christos ZFREE(source, copy);
1350 1.1 christos return Z_MEM_ERROR;
1351 1.1 christos }
1352 1.1 christos }
1353 1.1 christos
1354 1.1 christos /* copy state */
1355 1.1 christos zmemcpy(dest, source, sizeof(z_stream));
1356 1.1 christos zmemcpy(copy, state, sizeof(struct inflate_state));
1357 1.1 christos if (state->lencode >= state->codes &&
1358 1.1 christos state->lencode <= state->codes + ENOUGH - 1) {
1359 1.1 christos copy->lencode = copy->codes + (state->lencode - state->codes);
1360 1.1 christos copy->distcode = copy->codes + (state->distcode - state->codes);
1361 1.1 christos }
1362 1.1 christos copy->next = copy->codes + (state->next - state->codes);
1363 1.1 christos if (window != Z_NULL) {
1364 1.1 christos wsize = 1U << state->wbits;
1365 1.1 christos zmemcpy(window, state->window, wsize);
1366 1.1 christos }
1367 1.1 christos copy->window = window;
1368 1.1 christos dest->state = (struct internal_state FAR *)copy;
1369 1.1 christos return Z_OK;
1370 1.1 christos }
1371