maint: remove all uses of OF((...)) prototype-hiding macro
[gzip.git] / inflate.c
blobba6077aaaa0ab66b782d62638935d48afd2781e1
1 /* Inflate deflated data
3 Copyright (C) 1997-1999, 2002, 2006, 2009-2011 Free Software Foundation,
4 Inc.
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3, or (at your option)
9 any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, write to the Free Software Foundation,
18 Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. */
20 /* Not copyrighted 1992 by Mark Adler
21 version c10p1, 10 January 1993 */
23 /* You can do whatever you like with this source file, though I would
24 prefer that if you modify it and redistribute it that you include
25 comments to that effect with your name and the date. Thank you.
26 [The history has been moved to the file ChangeLog.]
30 Inflate deflated (PKZIP's method 8 compressed) data. The compression
31 method searches for as much of the current string of bytes (up to a
32 length of 258) in the previous 32K bytes. If it doesn't find any
33 matches (of at least length 3), it codes the next byte. Otherwise, it
34 codes the length of the matched string and its distance backwards from
35 the current position. There is a single Huffman code that codes both
36 single bytes (called "literals") and match lengths. A second Huffman
37 code codes the distance information, which follows a length code. Each
38 length or distance code actually represents a base value and a number
39 of "extra" (sometimes zero) bits to get to add to the base value. At
40 the end of each deflated block is a special end-of-block (EOB) literal/
41 length code. The decoding process is basically: get a literal/length
42 code; if EOB then done; if a literal, emit the decoded byte; if a
43 length then get the distance and emit the referred-to bytes from the
44 sliding window of previously emitted data.
46 There are (currently) three kinds of inflate blocks: stored, fixed, and
47 dynamic. The compressor deals with some chunk of data at a time, and
48 decides which method to use on a chunk-by-chunk basis. A chunk might
49 typically be 32K or 64K. If the chunk is uncompressible, then the
50 "stored" method is used. In this case, the bytes are simply stored as
51 is, eight bits per byte, with none of the above coding. The bytes are
52 preceded by a count, since there is no longer an EOB code.
54 If the data is compressible, then either the fixed or dynamic methods
55 are used. In the dynamic method, the compressed data is preceded by
56 an encoding of the literal/length and distance Huffman codes that are
57 to be used to decode this block. The representation is itself Huffman
58 coded, and so is preceded by a description of that code. These code
59 descriptions take up a little space, and so for small blocks, there is
60 a predefined set of codes, called the fixed codes. The fixed method is
61 used if the block codes up smaller that way (usually for quite small
62 chunks), otherwise the dynamic method is used. In the latter case, the
63 codes are customized to the probabilities in the current block, and so
64 can code it much better than the pre-determined fixed codes.
66 The Huffman codes themselves are decoded using a multi-level table
67 lookup, in order to maximize the speed of decoding plus the speed of
68 building the decoding tables. See the comments below that precede the
69 lbits and dbits tuning parameters.
74 Notes beyond the 1.93a appnote.txt:
76 1. Distance pointers never point before the beginning of the output
77 stream.
78 2. Distance pointers can point back across blocks, up to 32k away.
79 3. There is an implied maximum of 7 bits for the bit length table and
80 15 bits for the actual data.
81 4. If only one code exists, then it is encoded using one bit. (Zero
82 would be more efficient, but perhaps a little confusing.) If two
83 codes exist, they are coded using one bit each (0 and 1).
84 5. There is no way of sending zero distance codes--a dummy must be
85 sent if there are none. (History: a pre 2.0 version of PKZIP would
86 store blocks with no distance codes, but this was discovered to be
87 too harsh a criterion.) Valid only for 1.93a. 2.04c does allow
88 zero distance codes, which is sent as one code of zero bits in
89 length.
90 6. There are up to 286 literal/length codes. Code 256 represents the
91 end-of-block. Note however that the static length tree defines
92 288 codes just to fill out the Huffman codes. Codes 286 and 287
93 cannot be used though, since there is no length base or extra bits
94 defined for them. Similarly, there are up to 30 distance codes.
95 However, static trees define 32 codes (all 5 bits) to fill out the
96 Huffman codes, but the last two had better not show up in the data.
97 7. Unzip can check dynamic Huffman blocks for complete code sets.
98 The exception is that a single code would not be complete (see #4).
99 8. The five bits following the block type is really the number of
100 literal codes sent minus 257.
101 9. Length codes 8,16,16 are interpreted as 13 length codes of 8 bits
102 (1+6+6). Therefore, to output three times the length, you output
103 three codes (1+1+1), whereas to output four times the same length,
104 you only need two codes (1+3). Hmm.
105 10. In the tree reconstruction algorithm, Code = Code + Increment
106 only if BitLength(i) is not zero. (Pretty obvious.)
107 11. Correction: 4 Bits: # of Bit Length codes - 4 (4 - 19)
108 12. Note: length code 284 can represent 227-258, but length code 285
109 really is 258. The last length deserves its own, short code
110 since it gets used a lot in very redundant files. The length
111 258 is special since 258 - 3 (the min match length) is 255.
112 13. The literal/length and distance code bit lengths are read as a
113 single stream of lengths. It is possible (and advantageous) for
114 a repeat code (16, 17, or 18) to go across the boundary between
115 the two sets of lengths.
118 #include <config.h>
119 #include "tailor.h"
121 #include <stdlib.h>
123 #include "gzip.h"
124 #define slide window
126 /* Huffman code lookup table entry--this entry is four bytes for machines
127 that have 16-bit pointers (e.g. PC's in the small or medium model).
128 Valid extra bits are 0..13. e == 15 is EOB (end of block), e == 16
129 means that v is a literal, 16 < e < 32 means that v is a pointer to
130 the next table, which codes e - 16 bits, and lastly e == 99 indicates
131 an unused code. If a code with e == 99 is looked up, this implies an
132 error in the data. */
133 struct huft {
134 uch e; /* number of extra bits or operation */
135 uch b; /* number of bits in this code or subcode */
136 union {
137 ush n; /* literal, length base, or distance base */
138 struct huft *t; /* pointer to next level of table */
139 } v;
143 /* Function prototypes */
144 int huft_build (unsigned *, unsigned, unsigned, ush *, ush *,
145 struct huft **, int *);
146 int huft_free (struct huft *);
147 int inflate_codes (struct huft *, struct huft *, int, int);
148 int inflate_stored (void);
149 int inflate_fixed (void);
150 int inflate_dynamic (void);
151 int inflate_block (int *);
152 int inflate (void);
155 /* The inflate algorithm uses a sliding 32K byte window on the uncompressed
156 stream to find repeated byte strings. This is implemented here as a
157 circular buffer. The index is updated simply by incrementing and then
158 and'ing with 0x7fff (32K-1). */
159 /* It is left to other modules to supply the 32K area. It is assumed
160 to be usable as if it were declared "uch slide[32768];" or as just
161 "uch *slide;" and then malloc'ed in the latter case. The definition
162 must be in unzip.h, included above. */
163 /* unsigned wp; current position in slide */
164 #define wp outcnt
165 #define flush_output(w) (wp=(w),flush_window())
167 /* Tables for deflate from PKZIP's appnote.txt. */
168 static unsigned border[] = { /* Order of the bit length code lengths */
169 16, 17, 18, 0, 8, 7, 9, 6, 10, 5, 11, 4, 12, 3, 13, 2, 14, 1, 15};
170 static ush cplens[] = { /* Copy lengths for literal codes 257..285 */
171 3, 4, 5, 6, 7, 8, 9, 10, 11, 13, 15, 17, 19, 23, 27, 31,
172 35, 43, 51, 59, 67, 83, 99, 115, 131, 163, 195, 227, 258, 0, 0};
173 /* note: see note #13 above about the 258 in this list. */
174 static ush cplext[] = { /* Extra bits for literal codes 257..285 */
175 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2,
176 3, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 0, 99, 99}; /* 99==invalid */
177 static ush cpdist[] = { /* Copy offsets for distance codes 0..29 */
178 1, 2, 3, 4, 5, 7, 9, 13, 17, 25, 33, 49, 65, 97, 129, 193,
179 257, 385, 513, 769, 1025, 1537, 2049, 3073, 4097, 6145,
180 8193, 12289, 16385, 24577};
181 static ush cpdext[] = { /* Extra bits for distance codes */
182 0, 0, 0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6,
183 7, 7, 8, 8, 9, 9, 10, 10, 11, 11,
184 12, 12, 13, 13};
188 /* Macros for inflate() bit peeking and grabbing.
189 The usage is:
191 NEEDBITS(j)
192 x = b & mask_bits[j];
193 DUMPBITS(j)
195 where NEEDBITS makes sure that b has at least j bits in it, and
196 DUMPBITS removes the bits from b. The macros use the variable k
197 for the number of bits in b. Normally, b and k are register
198 variables for speed, and are initialized at the beginning of a
199 routine that uses these macros from a global bit buffer and count.
200 The macros also use the variable w, which is a cached copy of wp.
202 If we assume that EOB will be the longest code, then we will never
203 ask for bits with NEEDBITS that are beyond the end of the stream.
204 So, NEEDBITS should not read any more bytes than are needed to
205 meet the request. Then no bytes need to be "returned" to the buffer
206 at the end of the last block.
208 However, this assumption is not true for fixed blocks--the EOB code
209 is 7 bits, but the other literal/length codes can be 8 or 9 bits.
210 (The EOB code is shorter than other codes because fixed blocks are
211 generally short. So, while a block always has an EOB, many other
212 literal/length codes have a significantly lower probability of
213 showing up at all.) However, by making the first table have a
214 lookup of seven bits, the EOB code will be found in that first
215 lookup, and so will not require that too many bits be pulled from
216 the stream.
219 ulg bb; /* bit buffer */
220 unsigned bk; /* bits in bit buffer */
222 ush mask_bits[] = {
223 0x0000,
224 0x0001, 0x0003, 0x0007, 0x000f, 0x001f, 0x003f, 0x007f, 0x00ff,
225 0x01ff, 0x03ff, 0x07ff, 0x0fff, 0x1fff, 0x3fff, 0x7fff, 0xffff
228 #define GETBYTE() (inptr < insize ? inbuf[inptr++] : (wp = w, fill_inbuf(0)))
230 #ifdef CRYPT
231 uch cc;
232 # define NEXTBYTE() \
233 (decrypt ? (cc = GETBYTE(), zdecode(cc), cc) : GETBYTE())
234 #else
235 # define NEXTBYTE() (uch)GETBYTE()
236 #endif
237 #define NEEDBITS(n) {while(k<(n)){b|=((ulg)NEXTBYTE())<<k;k+=8;}}
238 #define DUMPBITS(n) {b>>=(n);k-=(n);}
242 Huffman code decoding is performed using a multi-level table lookup.
243 The fastest way to decode is to simply build a lookup table whose
244 size is determined by the longest code. However, the time it takes
245 to build this table can also be a factor if the data being decoded
246 is not very long. The most common codes are necessarily the
247 shortest codes, so those codes dominate the decoding time, and hence
248 the speed. The idea is you can have a shorter table that decodes the
249 shorter, more probable codes, and then point to subsidiary tables for
250 the longer codes. The time it costs to decode the longer codes is
251 then traded against the time it takes to make longer tables.
253 This results of this trade are in the variables lbits and dbits
254 below. lbits is the number of bits the first level table for literal/
255 length codes can decode in one step, and dbits is the same thing for
256 the distance codes. Subsequent tables are also less than or equal to
257 those sizes. These values may be adjusted either when all of the
258 codes are shorter than that, in which case the longest code length in
259 bits is used, or when the shortest code is *longer* than the requested
260 table size, in which case the length of the shortest code in bits is
261 used.
263 There are two different values for the two tables, since they code a
264 different number of possibilities each. The literal/length table
265 codes 286 possible values, or in a flat code, a little over eight
266 bits. The distance table codes 30 possible values, or a little less
267 than five bits, flat. The optimum values for speed end up being
268 about one bit more than those, so lbits is 8+1 and dbits is 5+1.
269 The optimum values may differ though from machine to machine, and
270 possibly even between compilers. Your mileage may vary.
274 int lbits = 9; /* bits in base literal/length lookup table */
275 int dbits = 6; /* bits in base distance lookup table */
278 /* If BMAX needs to be larger than 16, then h and x[] should be ulg. */
279 #define BMAX 16 /* maximum bit length of any code (16 for explode) */
280 #define N_MAX 288 /* maximum number of codes in any set */
283 unsigned hufts; /* track memory usage */
286 int huft_build(b, n, s, d, e, t, m)
287 unsigned *b; /* code lengths in bits (all assumed <= BMAX) */
288 unsigned n; /* number of codes (assumed <= N_MAX) */
289 unsigned s; /* number of simple-valued codes (0..s-1) */
290 ush *d; /* list of base values for non-simple codes */
291 ush *e; /* list of extra bits for non-simple codes */
292 struct huft **t; /* result: starting table */
293 int *m; /* maximum lookup bits, returns actual */
294 /* Given a list of code lengths and a maximum table size, make a set of
295 tables to decode that set of codes. Return zero on success, one if
296 the given code set is incomplete (the tables are still built in this
297 case), two if the input is invalid (all zero length codes or an
298 oversubscribed set of lengths), and three if not enough memory. */
300 unsigned a; /* counter for codes of length k */
301 unsigned c[BMAX+1]; /* bit length count table */
302 unsigned f; /* i repeats in table every f entries */
303 int g; /* maximum code length */
304 int h; /* table level */
305 register unsigned i; /* counter, current code */
306 register unsigned j; /* counter */
307 register int k; /* number of bits in current code */
308 int l; /* bits per table (returned in m) */
309 register unsigned *p; /* pointer into c[], b[], or v[] */
310 register struct huft *q; /* points to current table */
311 struct huft r; /* table entry for structure assignment */
312 struct huft *u[BMAX]; /* table stack */
313 unsigned v[N_MAX]; /* values in order of bit length */
314 register int w; /* bits before this table == (l * h) */
315 unsigned x[BMAX+1]; /* bit offsets, then code stack */
316 unsigned *xp; /* pointer into x */
317 int y; /* number of dummy codes added */
318 unsigned z; /* number of entries in current table */
321 /* Generate counts for each bit length */
322 memzero(c, sizeof(c));
323 p = b; i = n;
324 do {
325 Tracecv(*p, (stderr, (n-i >= ' ' && n-i <= '~' ? "%c %d\n" : "0x%x %d\n"),
326 n-i, *p));
327 c[*p]++; /* assume all entries <= BMAX */
328 p++; /* Can't combine with above line (Solaris bug) */
329 } while (--i);
330 if (c[0] == n) /* null input--all zero length codes */
332 q = (struct huft *) malloc (3 * sizeof *q);
333 if (!q)
334 return 3;
335 hufts += 3;
336 q[0].v.t = (struct huft *) NULL;
337 q[1].e = 99; /* invalid code marker */
338 q[1].b = 1;
339 q[2].e = 99; /* invalid code marker */
340 q[2].b = 1;
341 *t = q + 1;
342 *m = 1;
343 return 0;
347 /* Find minimum and maximum length, bound *m by those */
348 l = *m;
349 for (j = 1; j <= BMAX; j++)
350 if (c[j])
351 break;
352 k = j; /* minimum code length */
353 if ((unsigned)l < j)
354 l = j;
355 for (i = BMAX; i; i--)
356 if (c[i])
357 break;
358 g = i; /* maximum code length */
359 if ((unsigned)l > i)
360 l = i;
361 *m = l;
364 /* Adjust last length count to fill out codes, if needed */
365 for (y = 1 << j; j < i; j++, y <<= 1)
366 if ((y -= c[j]) < 0)
367 return 2; /* bad input: more codes than bits */
368 if ((y -= c[i]) < 0)
369 return 2;
370 c[i] += y;
373 /* Generate starting offsets into the value table for each length */
374 x[1] = j = 0;
375 p = c + 1; xp = x + 2;
376 while (--i) { /* note that i == g from above */
377 *xp++ = (j += *p++);
381 /* Make a table of values in order of bit lengths */
382 p = b; i = 0;
383 do {
384 if ((j = *p++) != 0)
385 v[x[j]++] = i;
386 } while (++i < n);
387 n = x[g]; /* set n to length of v */
390 /* Generate the Huffman codes and for each, make the table entries */
391 x[0] = i = 0; /* first Huffman code is zero */
392 p = v; /* grab values in bit order */
393 h = -1; /* no tables yet--level -1 */
394 w = -l; /* bits decoded == (l * h) */
395 u[0] = (struct huft *)NULL; /* just to keep compilers happy */
396 q = (struct huft *)NULL; /* ditto */
397 z = 0; /* ditto */
399 /* go through the bit lengths (k already is bits in shortest code) */
400 for (; k <= g; k++)
402 a = c[k];
403 while (a--)
405 /* here i is the Huffman code of length k bits for value *p */
406 /* make tables up to required level */
407 while (k > w + l)
409 h++;
410 w += l; /* previous table always l bits */
412 /* compute minimum size table less than or equal to l bits */
413 z = (z = g - w) > (unsigned)l ? l : z; /* upper limit on table size */
414 if ((f = 1 << (j = k - w)) > a + 1) /* try a k-w bit table */
415 { /* too few codes for k-w bit table */
416 f -= a + 1; /* deduct codes from patterns left */
417 xp = c + k;
418 if (j < z)
419 while (++j < z) /* try smaller tables up to z bits */
421 if ((f <<= 1) <= *++xp)
422 break; /* enough codes to use up j bits */
423 f -= *xp; /* else deduct codes from patterns */
426 z = 1 << j; /* table entries for j-bit table */
428 /* allocate and link in new table */
429 if ((q = (struct huft *)malloc((z + 1)*sizeof(struct huft))) ==
430 (struct huft *)NULL)
432 if (h)
433 huft_free(u[0]);
434 return 3; /* not enough memory */
436 hufts += z + 1; /* track memory usage */
437 *t = q + 1; /* link to list for huft_free() */
438 *(t = &(q->v.t)) = (struct huft *)NULL;
439 u[h] = ++q; /* table starts after link */
441 /* connect to last table, if there is one */
442 if (h)
444 x[h] = i; /* save pattern for backing up */
445 r.b = (uch)l; /* bits to dump before this table */
446 r.e = (uch)(16 + j); /* bits in this table */
447 r.v.t = q; /* pointer to this table */
448 j = i >> (w - l); /* (get around Turbo C bug) */
449 u[h-1][j] = r; /* connect to last table */
453 /* set up table entry in r */
454 r.b = (uch)(k - w);
455 if (p >= v + n)
456 r.e = 99; /* out of values--invalid code */
457 else if (*p < s)
459 r.e = (uch)(*p < 256 ? 16 : 15); /* 256 is end-of-block code */
460 r.v.n = (ush)(*p); /* simple code is just the value */
461 p++; /* one compiler does not like *p++ */
463 else
465 r.e = (uch)e[*p - s]; /* non-simple--look up in lists */
466 r.v.n = d[*p++ - s];
469 /* fill code-like entries with r */
470 f = 1 << (k - w);
471 for (j = i >> w; j < z; j += f)
472 q[j] = r;
474 /* backwards increment the k-bit code i */
475 for (j = 1 << (k - 1); i & j; j >>= 1)
476 i ^= j;
477 i ^= j;
479 /* backup over finished tables */
480 while ((i & ((1 << w) - 1)) != x[h])
482 h--; /* don't need to update q */
483 w -= l;
489 /* Return true (1) if we were given an incomplete table */
490 return y != 0 && g != 1;
495 int huft_free(t)
496 struct huft *t; /* table to free */
497 /* Free the malloc'ed tables built by huft_build(), which makes a linked
498 list of the tables it made, with the links in a dummy first entry of
499 each table. */
501 register struct huft *p, *q;
504 /* Go through linked list, freeing from the malloced (t[-1]) address. */
505 p = t;
506 while (p != (struct huft *)NULL)
508 q = (--p)->v.t;
509 free(p);
510 p = q;
512 return 0;
516 int inflate_codes(tl, td, bl, bd)
517 struct huft *tl, *td; /* literal/length and distance decoder tables */
518 int bl, bd; /* number of bits decoded by tl[] and td[] */
519 /* inflate (decompress) the codes in a deflated (compressed) block.
520 Return an error code or zero if it all goes ok. */
522 register unsigned e; /* table entry flag/number of extra bits */
523 unsigned n, d; /* length and index for copy */
524 unsigned w; /* current window position */
525 struct huft *t; /* pointer to table entry */
526 unsigned ml, md; /* masks for bl and bd bits */
527 register ulg b; /* bit buffer */
528 register unsigned k; /* number of bits in bit buffer */
531 /* make local copies of globals */
532 b = bb; /* initialize bit buffer */
533 k = bk;
534 w = wp; /* initialize window position */
536 /* inflate the coded data */
537 ml = mask_bits[bl]; /* precompute masks for speed */
538 md = mask_bits[bd];
539 for (;;) /* do until end of block */
541 NEEDBITS((unsigned)bl)
542 if ((e = (t = tl + ((unsigned)b & ml))->e) > 16)
543 do {
544 if (e == 99)
545 return 1;
546 DUMPBITS(t->b)
547 e -= 16;
548 NEEDBITS(e)
549 } while ((e = (t = t->v.t + ((unsigned)b & mask_bits[e]))->e) > 16);
550 DUMPBITS(t->b)
551 if (e == 16) /* then it's a literal */
553 slide[w++] = (uch)t->v.n;
554 Tracevv((stderr, "%c", slide[w-1]));
555 if (w == WSIZE)
557 flush_output(w);
558 w = 0;
561 else /* it's an EOB or a length */
563 /* exit if end of block */
564 if (e == 15)
565 break;
567 /* get length of block to copy */
568 NEEDBITS(e)
569 n = t->v.n + ((unsigned)b & mask_bits[e]);
570 DUMPBITS(e);
572 /* decode distance of block to copy */
573 NEEDBITS((unsigned)bd)
574 if ((e = (t = td + ((unsigned)b & md))->e) > 16)
575 do {
576 if (e == 99)
577 return 1;
578 DUMPBITS(t->b)
579 e -= 16;
580 NEEDBITS(e)
581 } while ((e = (t = t->v.t + ((unsigned)b & mask_bits[e]))->e) > 16);
582 DUMPBITS(t->b)
583 NEEDBITS(e)
584 d = w - t->v.n - ((unsigned)b & mask_bits[e]);
585 DUMPBITS(e)
586 Tracevv((stderr,"\\[%d,%d]", w-d, n));
588 /* do the copy */
589 do {
590 n -= (e = (e = WSIZE - ((d &= WSIZE-1) > w ? d : w)) > n ? n : e);
591 #ifndef DEBUG
592 if (e <= (d < w ? w - d : d - w))
594 memcpy(slide + w, slide + d, e);
595 w += e;
596 d += e;
598 else /* do it slow to avoid memcpy() overlap */
599 #endif
600 do {
601 slide[w++] = slide[d++];
602 Tracevv((stderr, "%c", slide[w-1]));
603 } while (--e);
604 if (w == WSIZE)
606 flush_output(w);
607 w = 0;
609 } while (n);
614 /* restore the globals from the locals */
615 wp = w; /* restore global window pointer */
616 bb = b; /* restore global bit buffer */
617 bk = k;
619 /* done */
620 return 0;
625 int inflate_stored()
626 /* "decompress" an inflated type 0 (stored) block. */
628 unsigned n; /* number of bytes in block */
629 unsigned w; /* current window position */
630 register ulg b; /* bit buffer */
631 register unsigned k; /* number of bits in bit buffer */
634 /* make local copies of globals */
635 b = bb; /* initialize bit buffer */
636 k = bk;
637 w = wp; /* initialize window position */
640 /* go to byte boundary */
641 n = k & 7;
642 DUMPBITS(n);
645 /* get the length and its complement */
646 NEEDBITS(16)
647 n = ((unsigned)b & 0xffff);
648 DUMPBITS(16)
649 NEEDBITS(16)
650 if (n != (unsigned)((~b) & 0xffff))
651 return 1; /* error in compressed data */
652 DUMPBITS(16)
655 /* read and output the compressed data */
656 while (n--)
658 NEEDBITS(8)
659 slide[w++] = (uch)b;
660 if (w == WSIZE)
662 flush_output(w);
663 w = 0;
665 DUMPBITS(8)
669 /* restore the globals from the locals */
670 wp = w; /* restore global window pointer */
671 bb = b; /* restore global bit buffer */
672 bk = k;
673 return 0;
678 int inflate_fixed()
679 /* decompress an inflated type 1 (fixed Huffman codes) block. We should
680 either replace this with a custom decoder, or at least precompute the
681 Huffman tables. */
683 int i; /* temporary variable */
684 struct huft *tl; /* literal/length code table */
685 struct huft *td; /* distance code table */
686 int bl; /* lookup bits for tl */
687 int bd; /* lookup bits for td */
688 unsigned l[288]; /* length list for huft_build */
691 /* set up literal table */
692 for (i = 0; i < 144; i++)
693 l[i] = 8;
694 for (; i < 256; i++)
695 l[i] = 9;
696 for (; i < 280; i++)
697 l[i] = 7;
698 for (; i < 288; i++) /* make a complete, but wrong code set */
699 l[i] = 8;
700 bl = 7;
701 if ((i = huft_build(l, 288, 257, cplens, cplext, &tl, &bl)) != 0)
702 return i;
705 /* set up distance table */
706 for (i = 0; i < 30; i++) /* make an incomplete code set */
707 l[i] = 5;
708 bd = 5;
709 if ((i = huft_build(l, 30, 0, cpdist, cpdext, &td, &bd)) > 1)
711 huft_free(tl);
712 return i;
716 /* decompress until an end-of-block code */
717 if (inflate_codes(tl, td, bl, bd))
718 return 1;
721 /* free the decoding tables, return */
722 huft_free(tl);
723 huft_free(td);
724 return 0;
729 int inflate_dynamic()
730 /* decompress an inflated type 2 (dynamic Huffman codes) block. */
732 int i; /* temporary variables */
733 unsigned j;
734 unsigned l; /* last length */
735 unsigned m; /* mask for bit lengths table */
736 unsigned n; /* number of lengths to get */
737 unsigned w; /* current window position */
738 struct huft *tl; /* literal/length code table */
739 struct huft *td; /* distance code table */
740 int bl; /* lookup bits for tl */
741 int bd; /* lookup bits for td */
742 unsigned nb; /* number of bit length codes */
743 unsigned nl; /* number of literal/length codes */
744 unsigned nd; /* number of distance codes */
745 #ifdef PKZIP_BUG_WORKAROUND
746 unsigned ll[288+32]; /* literal/length and distance code lengths */
747 #else
748 unsigned ll[286+30]; /* literal/length and distance code lengths */
749 #endif
750 register ulg b; /* bit buffer */
751 register unsigned k; /* number of bits in bit buffer */
754 /* make local bit buffer */
755 b = bb;
756 k = bk;
757 w = wp;
760 /* read in table lengths */
761 NEEDBITS(5)
762 nl = 257 + ((unsigned)b & 0x1f); /* number of literal/length codes */
763 DUMPBITS(5)
764 NEEDBITS(5)
765 nd = 1 + ((unsigned)b & 0x1f); /* number of distance codes */
766 DUMPBITS(5)
767 NEEDBITS(4)
768 nb = 4 + ((unsigned)b & 0xf); /* number of bit length codes */
769 DUMPBITS(4)
770 #ifdef PKZIP_BUG_WORKAROUND
771 if (nl > 288 || nd > 32)
772 #else
773 if (nl > 286 || nd > 30)
774 #endif
775 return 1; /* bad lengths */
778 /* read in bit-length-code lengths */
779 for (j = 0; j < nb; j++)
781 NEEDBITS(3)
782 ll[border[j]] = (unsigned)b & 7;
783 DUMPBITS(3)
785 for (; j < 19; j++)
786 ll[border[j]] = 0;
789 /* build decoding table for trees--single level, 7 bit lookup */
790 bl = 7;
791 if ((i = huft_build(ll, 19, 19, NULL, NULL, &tl, &bl)) != 0)
793 if (i == 1)
794 huft_free(tl);
795 return i; /* incomplete code set */
798 if (tl == NULL) /* Grrrhhh */
799 return 2;
801 /* read in literal and distance code lengths */
802 n = nl + nd;
803 m = mask_bits[bl];
804 i = l = 0;
805 while ((unsigned)i < n)
807 NEEDBITS((unsigned)bl)
808 j = (td = tl + ((unsigned)b & m))->b;
809 DUMPBITS(j)
810 j = td->v.n;
811 if (j < 16) /* length of code in bits (0..15) */
812 ll[i++] = l = j; /* save last length in l */
813 else if (j == 16) /* repeat last length 3 to 6 times */
815 NEEDBITS(2)
816 j = 3 + ((unsigned)b & 3);
817 DUMPBITS(2)
818 if ((unsigned)i + j > n)
819 return 1;
820 while (j--)
821 ll[i++] = l;
823 else if (j == 17) /* 3 to 10 zero length codes */
825 NEEDBITS(3)
826 j = 3 + ((unsigned)b & 7);
827 DUMPBITS(3)
828 if ((unsigned)i + j > n)
829 return 1;
830 while (j--)
831 ll[i++] = 0;
832 l = 0;
834 else /* j == 18: 11 to 138 zero length codes */
836 NEEDBITS(7)
837 j = 11 + ((unsigned)b & 0x7f);
838 DUMPBITS(7)
839 if ((unsigned)i + j > n)
840 return 1;
841 while (j--)
842 ll[i++] = 0;
843 l = 0;
848 /* free decoding table for trees */
849 huft_free(tl);
852 /* restore the global bit buffer */
853 bb = b;
854 bk = k;
857 /* build the decoding tables for literal/length and distance codes */
858 bl = lbits;
859 if ((i = huft_build(ll, nl, 257, cplens, cplext, &tl, &bl)) != 0)
861 if (i == 1) {
862 Trace ((stderr, " incomplete literal tree\n"));
863 huft_free(tl);
865 return i; /* incomplete code set */
867 bd = dbits;
868 if ((i = huft_build(ll + nl, nd, 0, cpdist, cpdext, &td, &bd)) != 0)
870 if (i == 1) {
871 Trace ((stderr, " incomplete distance tree\n"));
872 #ifdef PKZIP_BUG_WORKAROUND
873 i = 0;
875 #else
876 huft_free(td);
878 huft_free(tl);
879 return i; /* incomplete code set */
880 #endif
885 /* decompress until an end-of-block code */
886 int err = inflate_codes(tl, td, bl, bd) ? 1 : 0;
888 /* free the decoding tables */
889 huft_free(tl);
890 huft_free(td);
892 return err;
898 int inflate_block(e)
899 int *e; /* last block flag */
900 /* decompress an inflated block */
902 unsigned t; /* block type */
903 unsigned w; /* current window position */
904 register ulg b; /* bit buffer */
905 register unsigned k; /* number of bits in bit buffer */
908 /* make local bit buffer */
909 b = bb;
910 k = bk;
911 w = wp;
914 /* read in last block bit */
915 NEEDBITS(1)
916 *e = (int)b & 1;
917 DUMPBITS(1)
920 /* read in block type */
921 NEEDBITS(2)
922 t = (unsigned)b & 3;
923 DUMPBITS(2)
926 /* restore the global bit buffer */
927 bb = b;
928 bk = k;
931 /* inflate that block type */
932 if (t == 2)
933 return inflate_dynamic();
934 if (t == 0)
935 return inflate_stored();
936 if (t == 1)
937 return inflate_fixed();
940 /* bad block type */
941 return 2;
946 int inflate()
947 /* decompress an inflated entry */
949 int e; /* last block flag */
950 int r; /* result code */
951 unsigned h; /* maximum struct huft's malloc'ed */
954 /* initialize window, bit buffer */
955 wp = 0;
956 bk = 0;
957 bb = 0;
960 /* decompress until the last block */
961 h = 0;
962 do {
963 hufts = 0;
964 if ((r = inflate_block(&e)) != 0)
965 return r;
966 if (hufts > h)
967 h = hufts;
968 } while (!e);
970 /* Undo too much lookahead. The next read will be byte aligned so we
971 * can discard unused bits in the last meaningful byte.
973 while (bk >= 8) {
974 bk -= 8;
975 inptr--;
978 /* flush out slide */
979 flush_output(wp);
982 /* return success */
983 Trace ((stderr, "<%u> ", h));
984 return 0;