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