1 | /* mszip decompression - based on cabextract.c code from
|
---|
2 | * Stuart Caie
|
---|
3 | *
|
---|
4 | * adapted for Samba by Andrew Tridgell and Stefan Metzmacher 2005
|
---|
5 | *
|
---|
6 | * (C) 2000-2001 Stuart Caie <kyzer@4u.net>
|
---|
7 | * reaktivate-specifics by Malte Starostik <malte@kde.org>
|
---|
8 | *
|
---|
9 | * This program is free software; you can redistribute it and/or modify
|
---|
10 | * it under the terms of the GNU General Public License as published by
|
---|
11 | * the Free Software Foundation; either version 3 of the License, or
|
---|
12 | * (at your option) any later version.
|
---|
13 | *
|
---|
14 | * This program is distributed in the hope that it will be useful,
|
---|
15 | * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
16 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
---|
17 | * GNU General Public License for more details.
|
---|
18 | *
|
---|
19 | * You should have received a copy of the GNU General Public License
|
---|
20 | * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
---|
21 | */
|
---|
22 |
|
---|
23 | #include "includes.h"
|
---|
24 | #include "../compression/mszip.h"
|
---|
25 |
|
---|
26 | /*--------------------------------------------------------------------------*/
|
---|
27 | /* our archiver information / state */
|
---|
28 |
|
---|
29 | /* MSZIP stuff */
|
---|
30 | #define ZIPWSIZE 0x8000 /* window size */
|
---|
31 | #define ZIPLBITS 9 /* bits in base literal/length lookup table */
|
---|
32 | #define ZIPDBITS 6 /* bits in base distance lookup table */
|
---|
33 | #define ZIPBMAX 16 /* maximum bit length of any code */
|
---|
34 | #define ZIPN_MAX 288 /* maximum number of codes in any set */
|
---|
35 |
|
---|
36 | struct Ziphuft {
|
---|
37 | uint8_t e; /* number of extra bits or operation */
|
---|
38 | uint8_t b; /* number of bits in this code or subcode */
|
---|
39 | union {
|
---|
40 | uint16_t n; /* literal, length base, or distance base */
|
---|
41 | struct Ziphuft *t; /* pointer to next level of table */
|
---|
42 | } v;
|
---|
43 | };
|
---|
44 |
|
---|
45 | struct ZIPstate {
|
---|
46 | uint32_t window_posn; /* current offset within the window */
|
---|
47 | uint32_t bb; /* bit buffer */
|
---|
48 | uint32_t bk; /* bits in bit buffer */
|
---|
49 | uint32_t ll[288+32]; /* literal/length and distance code lengths */
|
---|
50 | uint32_t c[ZIPBMAX+1]; /* bit length count table */
|
---|
51 | int32_t lx[ZIPBMAX+1]; /* memory for l[-1..ZIPBMAX-1] */
|
---|
52 | struct Ziphuft *u[ZIPBMAX]; /* table stack */
|
---|
53 | uint32_t v[ZIPN_MAX]; /* values in order of bit length */
|
---|
54 | uint32_t x[ZIPBMAX+1]; /* bit offsets, then code stack */
|
---|
55 | uint8_t *inpos;
|
---|
56 | };
|
---|
57 |
|
---|
58 | /* generic stuff */
|
---|
59 | #define CAB(x) (decomp_state->x)
|
---|
60 | #define ZIP(x) (decomp_state->methods.zip.x)
|
---|
61 |
|
---|
62 | /* CAB data blocks are <= 32768 bytes in uncompressed form. Uncompressed
|
---|
63 | * blocks have zero growth. MSZIP guarantees that it won't grow above
|
---|
64 | * uncompressed size by more than 12 bytes. LZX guarantees it won't grow
|
---|
65 | * more than 6144 bytes.
|
---|
66 | */
|
---|
67 | #define CAB_BLOCKMAX (32768)
|
---|
68 | #define CAB_INPUTMAX (CAB_BLOCKMAX+6144)
|
---|
69 |
|
---|
70 | struct decomp_state {
|
---|
71 | struct folder *current; /* current folder we're extracting from */
|
---|
72 | uint32_t offset; /* uncompressed offset within folder */
|
---|
73 | uint8_t *outpos; /* (high level) start of data to use up */
|
---|
74 | uint16_t outlen; /* (high level) amount of data to use up */
|
---|
75 | uint16_t split; /* at which split in current folder? */
|
---|
76 | int (*decompress)(int, int); /* the chosen compression func */
|
---|
77 | uint8_t inbuf[CAB_INPUTMAX+2]; /* +2 for lzx bitbuffer overflows! */
|
---|
78 | uint8_t outbuf[CAB_BLOCKMAX];
|
---|
79 | union {
|
---|
80 | struct ZIPstate zip;
|
---|
81 | } methods;
|
---|
82 | };
|
---|
83 |
|
---|
84 |
|
---|
85 | /* MSZIP decruncher */
|
---|
86 |
|
---|
87 | /* Dirk Stoecker wrote the ZIP decoder, based on the InfoZip deflate code */
|
---|
88 |
|
---|
89 | /* Tables for deflate from PKZIP's appnote.txt. */
|
---|
90 | static const uint8_t Zipborder[] = /* Order of the bit length code lengths */
|
---|
91 | { 16, 17, 18, 0, 8, 7, 9, 6, 10, 5, 11, 4, 12, 3, 13, 2, 14, 1, 15};
|
---|
92 | static const uint16_t Zipcplens[] = /* Copy lengths for literal codes 257..285 */
|
---|
93 | { 3, 4, 5, 6, 7, 8, 9, 10, 11, 13, 15, 17, 19, 23, 27, 31, 35, 43, 51,
|
---|
94 | 59, 67, 83, 99, 115, 131, 163, 195, 227, 258, 0, 0};
|
---|
95 | static const uint16_t Zipcplext[] = /* Extra bits for literal codes 257..285 */
|
---|
96 | { 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4,
|
---|
97 | 4, 5, 5, 5, 5, 0, 99, 99}; /* 99==invalid */
|
---|
98 | static const uint16_t Zipcpdist[] = /* Copy offsets for distance codes 0..29 */
|
---|
99 | { 1, 2, 3, 4, 5, 7, 9, 13, 17, 25, 33, 49, 65, 97, 129, 193, 257, 385,
|
---|
100 | 513, 769, 1025, 1537, 2049, 3073, 4097, 6145, 8193, 12289, 16385, 24577};
|
---|
101 | static const uint16_t Zipcpdext[] = /* Extra bits for distance codes */
|
---|
102 | { 0, 0, 0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10,
|
---|
103 | 10, 11, 11, 12, 12, 13, 13};
|
---|
104 |
|
---|
105 | /* And'ing with Zipmask[n] masks the lower n bits */
|
---|
106 | static const uint16_t Zipmask[17] = {
|
---|
107 | 0x0000, 0x0001, 0x0003, 0x0007, 0x000f, 0x001f, 0x003f, 0x007f, 0x00ff,
|
---|
108 | 0x01ff, 0x03ff, 0x07ff, 0x0fff, 0x1fff, 0x3fff, 0x7fff, 0xffff
|
---|
109 | };
|
---|
110 |
|
---|
111 | #define ZIPNEEDBITS(n) {while(k<(n)){int32_t c=*(ZIP(inpos)++);\
|
---|
112 | b|=((uint32_t)c)<<k;k+=8;}}
|
---|
113 | #define ZIPDUMPBITS(n) {b>>=(n);k-=(n);}
|
---|
114 |
|
---|
115 | static void Ziphuft_free(struct Ziphuft *t)
|
---|
116 | {
|
---|
117 | register struct Ziphuft *p, *q;
|
---|
118 |
|
---|
119 | /* Go through linked list, freeing from the allocated (t[-1]) address. */
|
---|
120 | p = t;
|
---|
121 | while (p != (struct Ziphuft *)NULL)
|
---|
122 | {
|
---|
123 | q = (--p)->v.t;
|
---|
124 | free(p);
|
---|
125 | p = q;
|
---|
126 | }
|
---|
127 | }
|
---|
128 |
|
---|
129 | static int32_t Ziphuft_build(struct decomp_state *decomp_state,
|
---|
130 | uint32_t *b, uint32_t n, uint32_t s, const uint16_t *d, const uint16_t *e,
|
---|
131 | struct Ziphuft **t, int32_t *m)
|
---|
132 | {
|
---|
133 | uint32_t a; /* counter for codes of length k */
|
---|
134 | uint32_t el; /* length of EOB code (value 256) */
|
---|
135 | uint32_t f; /* i repeats in table every f entries */
|
---|
136 | int32_t g; /* maximum code length */
|
---|
137 | int32_t h; /* table level */
|
---|
138 | register uint32_t i; /* counter, current code */
|
---|
139 | register uint32_t j; /* counter */
|
---|
140 | register int32_t k; /* number of bits in current code */
|
---|
141 | int32_t *l; /* stack of bits per table */
|
---|
142 | register uint32_t *p; /* pointer into ZIP(c)[],ZIP(b)[],ZIP(v)[] */
|
---|
143 | register struct Ziphuft *q; /* points to current table */
|
---|
144 | struct Ziphuft r; /* table entry for structure assignment */
|
---|
145 | register int32_t w; /* bits before this table == (l * h) */
|
---|
146 | uint32_t *xp; /* pointer into x */
|
---|
147 | int32_t y; /* number of dummy codes added */
|
---|
148 | uint32_t z; /* number of entries in current table */
|
---|
149 |
|
---|
150 | l = ZIP(lx)+1;
|
---|
151 |
|
---|
152 | /* Generate counts for each bit length */
|
---|
153 | el = n > 256 ? b[256] : ZIPBMAX; /* set length of EOB code, if any */
|
---|
154 |
|
---|
155 | for(i = 0; i < ZIPBMAX+1; ++i)
|
---|
156 | ZIP(c)[i] = 0;
|
---|
157 | p = b; i = n;
|
---|
158 | do
|
---|
159 | {
|
---|
160 | ZIP(c)[*p]++; p++; /* assume all entries <= ZIPBMAX */
|
---|
161 | } while (--i);
|
---|
162 | if (ZIP(c)[0] == n) /* null input--all zero length codes */
|
---|
163 | {
|
---|
164 | *t = (struct Ziphuft *)NULL;
|
---|
165 | *m = 0;
|
---|
166 | return 0;
|
---|
167 | }
|
---|
168 |
|
---|
169 | /* Find minimum and maximum length, bound *m by those */
|
---|
170 | for (j = 1; j <= ZIPBMAX; j++)
|
---|
171 | if (ZIP(c)[j])
|
---|
172 | break;
|
---|
173 | k = j; /* minimum code length */
|
---|
174 | if ((uint32_t)*m < j)
|
---|
175 | *m = j;
|
---|
176 | for (i = ZIPBMAX; i; i--)
|
---|
177 | if (ZIP(c)[i])
|
---|
178 | break;
|
---|
179 | g = i; /* maximum code length */
|
---|
180 | if ((uint32_t)*m > i)
|
---|
181 | *m = i;
|
---|
182 |
|
---|
183 | /* Adjust last length count to fill out codes, if needed */
|
---|
184 | for (y = 1 << j; j < i; j++, y <<= 1)
|
---|
185 | if ((y -= ZIP(c)[j]) < 0)
|
---|
186 | return 2; /* bad input: more codes than bits */
|
---|
187 | if ((y -= ZIP(c)[i]) < 0)
|
---|
188 | return 2;
|
---|
189 | ZIP(c)[i] += y;
|
---|
190 |
|
---|
191 | /* Generate starting offsets int32_to the value table for each length */
|
---|
192 | ZIP(x)[1] = j = 0;
|
---|
193 | p = ZIP(c) + 1; xp = ZIP(x) + 2;
|
---|
194 | while (--i)
|
---|
195 | { /* note that i == g from above */
|
---|
196 | *xp++ = (j += *p++);
|
---|
197 | }
|
---|
198 |
|
---|
199 | /* Make a table of values in order of bit lengths */
|
---|
200 | p = b; i = 0;
|
---|
201 | do{
|
---|
202 | if ((j = *p++) != 0)
|
---|
203 | ZIP(v)[ZIP(x)[j]++] = i;
|
---|
204 | } while (++i < n);
|
---|
205 |
|
---|
206 |
|
---|
207 | /* Generate the Huffman codes and for each, make the table entries */
|
---|
208 | ZIP(x)[0] = i = 0; /* first Huffman code is zero */
|
---|
209 | p = ZIP(v); /* grab values in bit order */
|
---|
210 | h = -1; /* no tables yet--level -1 */
|
---|
211 | w = l[-1] = 0; /* no bits decoded yet */
|
---|
212 | ZIP(u)[0] = (struct Ziphuft *)NULL; /* just to keep compilers happy */
|
---|
213 | q = (struct Ziphuft *)NULL; /* ditto */
|
---|
214 | z = 0; /* ditto */
|
---|
215 |
|
---|
216 | /* go through the bit lengths (k already is bits in shortest code) */
|
---|
217 | for (; k <= g; k++)
|
---|
218 | {
|
---|
219 | a = ZIP(c)[k];
|
---|
220 | while (a--)
|
---|
221 | {
|
---|
222 | /* here i is the Huffman code of length k bits for value *p */
|
---|
223 | /* make tables up to required level */
|
---|
224 | while (k > w + l[h])
|
---|
225 | {
|
---|
226 | w += l[h++]; /* add bits already decoded */
|
---|
227 |
|
---|
228 | /* compute minimum size table less than or equal to *m bits */
|
---|
229 | z = (z = g - w) > (uint32_t)*m ? *m : z; /* upper limit */
|
---|
230 | if ((f = 1 << (j = k - w)) > a + 1) /* try a k-w bit table */
|
---|
231 | { /* too few codes for k-w bit table */
|
---|
232 | f -= a + 1; /* deduct codes from patterns left */
|
---|
233 | xp = ZIP(c) + k;
|
---|
234 | while (++j < z) /* try smaller tables up to z bits */
|
---|
235 | {
|
---|
236 | if ((f <<= 1) <= *++xp)
|
---|
237 | break; /* enough codes to use up j bits */
|
---|
238 | f -= *xp; /* else deduct codes from patterns */
|
---|
239 | }
|
---|
240 | }
|
---|
241 | if ((uint32_t)w + j > el && (uint32_t)w < el)
|
---|
242 | j = el - w; /* make EOB code end at table */
|
---|
243 | z = 1 << j; /* table entries for j-bit table */
|
---|
244 | l[h] = j; /* set table size in stack */
|
---|
245 |
|
---|
246 | /* allocate and link in new table */
|
---|
247 | if (!(q = (struct Ziphuft *)SMB_MALLOC((z + 1)*sizeof(struct Ziphuft))))
|
---|
248 | {
|
---|
249 | if(h)
|
---|
250 | Ziphuft_free(ZIP(u)[0]);
|
---|
251 | return 3; /* not enough memory */
|
---|
252 | }
|
---|
253 | *t = q + 1; /* link to list for Ziphuft_free() */
|
---|
254 | *(t = &(q->v.t)) = (struct Ziphuft *)NULL;
|
---|
255 | ZIP(u)[h] = ++q; /* table starts after link */
|
---|
256 |
|
---|
257 | /* connect to last table, if there is one */
|
---|
258 | if (h)
|
---|
259 | {
|
---|
260 | ZIP(x)[h] = i; /* save pattern for backing up */
|
---|
261 | r.b = (uint8_t)l[h-1]; /* bits to dump before this table */
|
---|
262 | r.e = (uint8_t)(16 + j); /* bits in this table */
|
---|
263 | r.v.t = q; /* pointer to this table */
|
---|
264 | j = (i & ((1 << w) - 1)) >> (w - l[h-1]);
|
---|
265 | ZIP(u)[h-1][j] = r; /* connect to last table */
|
---|
266 | }
|
---|
267 | }
|
---|
268 |
|
---|
269 | /* set up table entry in r */
|
---|
270 | r.b = (uint8_t)(k - w);
|
---|
271 | if (p >= ZIP(v) + n)
|
---|
272 | r.e = 99; /* out of values--invalid code */
|
---|
273 | else if (*p < s)
|
---|
274 | {
|
---|
275 | r.e = (uint8_t)(*p < 256 ? 16 : 15); /* 256 is end-of-block code */
|
---|
276 | r.v.n = *p++; /* simple code is just the value */
|
---|
277 | }
|
---|
278 | else
|
---|
279 | {
|
---|
280 | r.e = (uint8_t)e[*p - s]; /* non-simple--look up in lists */
|
---|
281 | r.v.n = d[*p++ - s];
|
---|
282 | }
|
---|
283 |
|
---|
284 | /* fill code-like entries with r */
|
---|
285 | f = 1 << (k - w);
|
---|
286 | for (j = i >> w; j < z; j += f)
|
---|
287 | q[j] = r;
|
---|
288 |
|
---|
289 | /* backwards increment the k-bit code i */
|
---|
290 | for (j = 1 << (k - 1); i & j; j >>= 1)
|
---|
291 | i ^= j;
|
---|
292 | i ^= j;
|
---|
293 |
|
---|
294 | /* backup over finished tables */
|
---|
295 | while ((i & ((1 << w) - 1)) != ZIP(x)[h])
|
---|
296 | w -= l[--h]; /* don't need to update q */
|
---|
297 | }
|
---|
298 | }
|
---|
299 |
|
---|
300 | /* return actual size of base table */
|
---|
301 | *m = l[0];
|
---|
302 |
|
---|
303 | /* Return true (1) if we were given an incomplete table */
|
---|
304 | return y != 0 && g != 1;
|
---|
305 | }
|
---|
306 |
|
---|
307 | static int32_t Zipinflate_codes(struct decomp_state *decomp_state,
|
---|
308 | struct Ziphuft *tl, struct Ziphuft *td,
|
---|
309 | int32_t bl, int32_t bd)
|
---|
310 | {
|
---|
311 | register uint32_t e; /* table entry flag/number of extra bits */
|
---|
312 | uint32_t n, d; /* length and index for copy */
|
---|
313 | uint32_t w; /* current window position */
|
---|
314 | struct Ziphuft *t; /* pointer to table entry */
|
---|
315 | uint32_t ml, md; /* masks for bl and bd bits */
|
---|
316 | register uint32_t b; /* bit buffer */
|
---|
317 | register uint32_t k; /* number of bits in bit buffer */
|
---|
318 |
|
---|
319 | DEBUG(10,("Zipinflate_codes\n"));
|
---|
320 |
|
---|
321 | /* make local copies of globals */
|
---|
322 | b = ZIP(bb); /* initialize bit buffer */
|
---|
323 | k = ZIP(bk);
|
---|
324 | w = ZIP(window_posn); /* initialize window position */
|
---|
325 |
|
---|
326 | /* inflate the coded data */
|
---|
327 | ml = Zipmask[bl]; /* precompute masks for speed */
|
---|
328 | md = Zipmask[bd];
|
---|
329 |
|
---|
330 | for(;;)
|
---|
331 | {
|
---|
332 | ZIPNEEDBITS((uint32_t)bl)
|
---|
333 | if((e = (t = tl + ((uint32_t)b & ml))->e) > 16)
|
---|
334 | do
|
---|
335 | {
|
---|
336 | if (e == 99)
|
---|
337 | return 1;
|
---|
338 | ZIPDUMPBITS(t->b)
|
---|
339 | e -= 16;
|
---|
340 | ZIPNEEDBITS(e)
|
---|
341 | } while ((e = (t = t->v.t + ((uint32_t)b & Zipmask[e]))->e) > 16);
|
---|
342 | ZIPDUMPBITS(t->b)
|
---|
343 | if (w >= CAB_BLOCKMAX) break;
|
---|
344 | if (e == 16) /* then it's a literal */
|
---|
345 | CAB(outbuf)[w++] = (uint8_t)t->v.n;
|
---|
346 | else /* it's an EOB or a length */
|
---|
347 | {
|
---|
348 | /* exit if end of block */
|
---|
349 | if(e == 15)
|
---|
350 | break;
|
---|
351 |
|
---|
352 | /* get length of block to copy */
|
---|
353 | ZIPNEEDBITS(e)
|
---|
354 | n = t->v.n + ((uint32_t)b & Zipmask[e]);
|
---|
355 | ZIPDUMPBITS(e);
|
---|
356 |
|
---|
357 | /* decode distance of block to copy */
|
---|
358 | ZIPNEEDBITS((uint32_t)bd)
|
---|
359 | if ((e = (t = td + ((uint32_t)b & md))->e) > 16)
|
---|
360 | do {
|
---|
361 | if (e == 99)
|
---|
362 | return 1;
|
---|
363 | ZIPDUMPBITS(t->b)
|
---|
364 | e -= 16;
|
---|
365 | ZIPNEEDBITS(e)
|
---|
366 | } while ((e = (t = t->v.t + ((uint32_t)b & Zipmask[e]))->e) > 16);
|
---|
367 | ZIPDUMPBITS(t->b)
|
---|
368 | ZIPNEEDBITS(e)
|
---|
369 | d = w - t->v.n - ((uint32_t)b & Zipmask[e]);
|
---|
370 | ZIPDUMPBITS(e)
|
---|
371 | do
|
---|
372 | {
|
---|
373 | n -= (e = (e = ZIPWSIZE - ((d &= ZIPWSIZE-1) > w ? d : w)) > n ?n:e);
|
---|
374 | do
|
---|
375 | {
|
---|
376 | CAB(outbuf)[w++] = CAB(outbuf)[d++];
|
---|
377 | } while (--e);
|
---|
378 | } while (n);
|
---|
379 | }
|
---|
380 | }
|
---|
381 |
|
---|
382 | /* restore the globals from the locals */
|
---|
383 | ZIP(window_posn) = w; /* restore global window pointer */
|
---|
384 | ZIP(bb) = b; /* restore global bit buffer */
|
---|
385 | ZIP(bk) = k;
|
---|
386 |
|
---|
387 | /* done */
|
---|
388 | return 0;
|
---|
389 | }
|
---|
390 |
|
---|
391 | /* "decompress" an inflated type 0 (stored) block. */
|
---|
392 | static int32_t Zipinflate_stored(struct decomp_state *decomp_state)
|
---|
393 | {
|
---|
394 | uint32_t n; /* number of bytes in block */
|
---|
395 | uint32_t w; /* current window position */
|
---|
396 | register uint32_t b; /* bit buffer */
|
---|
397 | register uint32_t k; /* number of bits in bit buffer */
|
---|
398 |
|
---|
399 | /* make local copies of globals */
|
---|
400 | b = ZIP(bb); /* initialize bit buffer */
|
---|
401 | k = ZIP(bk);
|
---|
402 | w = ZIP(window_posn); /* initialize window position */
|
---|
403 |
|
---|
404 | /* go to byte boundary */
|
---|
405 | n = k & 7;
|
---|
406 | ZIPDUMPBITS(n);
|
---|
407 |
|
---|
408 | /* get the length and its complement */
|
---|
409 | ZIPNEEDBITS(16)
|
---|
410 | n = ((uint32_t)b & 0xffff);
|
---|
411 | ZIPDUMPBITS(16)
|
---|
412 | ZIPNEEDBITS(16)
|
---|
413 | if (n != (uint32_t)((~b) & 0xffff))
|
---|
414 | return 1; /* error in compressed data */
|
---|
415 | ZIPDUMPBITS(16)
|
---|
416 |
|
---|
417 | /* read and output the compressed data */
|
---|
418 | while(n--)
|
---|
419 | {
|
---|
420 | ZIPNEEDBITS(8)
|
---|
421 | CAB(outbuf)[w++] = (uint8_t)b;
|
---|
422 | ZIPDUMPBITS(8)
|
---|
423 | }
|
---|
424 |
|
---|
425 | /* restore the globals from the locals */
|
---|
426 | ZIP(window_posn) = w; /* restore global window pointer */
|
---|
427 | ZIP(bb) = b; /* restore global bit buffer */
|
---|
428 | ZIP(bk) = k;
|
---|
429 | return 0;
|
---|
430 | }
|
---|
431 |
|
---|
432 | static int32_t Zipinflate_fixed(struct decomp_state *decomp_state)
|
---|
433 | {
|
---|
434 | struct Ziphuft *fixed_tl;
|
---|
435 | struct Ziphuft *fixed_td;
|
---|
436 | int32_t fixed_bl, fixed_bd;
|
---|
437 | int32_t i; /* temporary variable */
|
---|
438 | uint32_t *l;
|
---|
439 |
|
---|
440 | l = ZIP(ll);
|
---|
441 |
|
---|
442 | /* literal table */
|
---|
443 | for(i = 0; i < 144; i++)
|
---|
444 | l[i] = 8;
|
---|
445 | for(; i < 256; i++)
|
---|
446 | l[i] = 9;
|
---|
447 | for(; i < 280; i++)
|
---|
448 | l[i] = 7;
|
---|
449 | for(; i < 288; i++) /* make a complete, but wrong code set */
|
---|
450 | l[i] = 8;
|
---|
451 | fixed_bl = 7;
|
---|
452 | if((i = Ziphuft_build(decomp_state, l, 288, 257, Zipcplens, Zipcplext, &fixed_tl, &fixed_bl)))
|
---|
453 | return i;
|
---|
454 |
|
---|
455 | /* distance table */
|
---|
456 | for(i = 0; i < 30; i++) /* make an incomplete code set */
|
---|
457 | l[i] = 5;
|
---|
458 | fixed_bd = 5;
|
---|
459 | if((i = Ziphuft_build(decomp_state, l, 30, 0, Zipcpdist, Zipcpdext, &fixed_td, &fixed_bd)) > 1)
|
---|
460 | {
|
---|
461 | Ziphuft_free(fixed_tl);
|
---|
462 | return i;
|
---|
463 | }
|
---|
464 |
|
---|
465 | /* decompress until an end-of-block code */
|
---|
466 | i = Zipinflate_codes(decomp_state, fixed_tl, fixed_td, fixed_bl, fixed_bd);
|
---|
467 |
|
---|
468 | Ziphuft_free(fixed_td);
|
---|
469 | Ziphuft_free(fixed_tl);
|
---|
470 | return i;
|
---|
471 | }
|
---|
472 |
|
---|
473 | /* decompress an inflated type 2 (dynamic Huffman codes) block. */
|
---|
474 | static int32_t Zipinflate_dynamic(struct decomp_state *decomp_state)
|
---|
475 | {
|
---|
476 | int32_t i; /* temporary variables */
|
---|
477 | uint32_t j;
|
---|
478 | uint32_t *ll;
|
---|
479 | uint32_t l; /* last length */
|
---|
480 | uint32_t m; /* mask for bit lengths table */
|
---|
481 | uint32_t n; /* number of lengths to get */
|
---|
482 | struct Ziphuft *tl; /* literal/length code table */
|
---|
483 | struct Ziphuft *td; /* distance code table */
|
---|
484 | int32_t bl; /* lookup bits for tl */
|
---|
485 | int32_t bd; /* lookup bits for td */
|
---|
486 | uint32_t nb; /* number of bit length codes */
|
---|
487 | uint32_t nl; /* number of literal/length codes */
|
---|
488 | uint32_t nd; /* number of distance codes */
|
---|
489 | register uint32_t b; /* bit buffer */
|
---|
490 | register uint32_t k; /* number of bits in bit buffer */
|
---|
491 |
|
---|
492 | /* make local bit buffer */
|
---|
493 | b = ZIP(bb);
|
---|
494 | k = ZIP(bk);
|
---|
495 | ll = ZIP(ll);
|
---|
496 |
|
---|
497 | /* read in table lengths */
|
---|
498 | ZIPNEEDBITS(5)
|
---|
499 | nl = 257 + ((uint32_t)b & 0x1f); /* number of literal/length codes */
|
---|
500 | ZIPDUMPBITS(5)
|
---|
501 | ZIPNEEDBITS(5)
|
---|
502 | nd = 1 + ((uint32_t)b & 0x1f); /* number of distance codes */
|
---|
503 | ZIPDUMPBITS(5)
|
---|
504 | ZIPNEEDBITS(4)
|
---|
505 | nb = 4 + ((uint32_t)b & 0xf); /* number of bit length codes */
|
---|
506 | ZIPDUMPBITS(4)
|
---|
507 | if(nl > 288 || nd > 32)
|
---|
508 | return 1; /* bad lengths */
|
---|
509 |
|
---|
510 | /* read in bit-length-code lengths */
|
---|
511 | for(j = 0; j < nb; j++)
|
---|
512 | {
|
---|
513 | ZIPNEEDBITS(3)
|
---|
514 | ll[Zipborder[j]] = (uint32_t)b & 7;
|
---|
515 | ZIPDUMPBITS(3)
|
---|
516 | }
|
---|
517 | for(; j < 19; j++)
|
---|
518 | ll[Zipborder[j]] = 0;
|
---|
519 |
|
---|
520 | /* build decoding table for trees--single level, 7 bit lookup */
|
---|
521 | bl = 7;
|
---|
522 | if((i = Ziphuft_build(decomp_state, ll, 19, 19, NULL, NULL, &tl, &bl)) != 0)
|
---|
523 | {
|
---|
524 | if(i == 1)
|
---|
525 | Ziphuft_free(tl);
|
---|
526 | return i; /* incomplete code set */
|
---|
527 | }
|
---|
528 |
|
---|
529 | /* read in literal and distance code lengths */
|
---|
530 | n = nl + nd;
|
---|
531 | m = Zipmask[bl];
|
---|
532 | i = l = 0;
|
---|
533 | while((uint32_t)i < n)
|
---|
534 | {
|
---|
535 | ZIPNEEDBITS((uint32_t)bl)
|
---|
536 | j = (td = tl + ((uint32_t)b & m))->b;
|
---|
537 | ZIPDUMPBITS(j)
|
---|
538 | j = td->v.n;
|
---|
539 | if (j < 16) /* length of code in bits (0..15) */
|
---|
540 | ll[i++] = l = j; /* save last length in l */
|
---|
541 | else if (j == 16) /* repeat last length 3 to 6 times */
|
---|
542 | {
|
---|
543 | ZIPNEEDBITS(2)
|
---|
544 | j = 3 + ((uint32_t)b & 3);
|
---|
545 | ZIPDUMPBITS(2)
|
---|
546 | if((uint32_t)i + j > n)
|
---|
547 | return 1;
|
---|
548 | while (j--)
|
---|
549 | ll[i++] = l;
|
---|
550 | }
|
---|
551 | else if (j == 17) /* 3 to 10 zero length codes */
|
---|
552 | {
|
---|
553 | ZIPNEEDBITS(3)
|
---|
554 | j = 3 + ((uint32_t)b & 7);
|
---|
555 | ZIPDUMPBITS(3)
|
---|
556 | if ((uint32_t)i + j > n)
|
---|
557 | return 1;
|
---|
558 | while (j--)
|
---|
559 | ll[i++] = 0;
|
---|
560 | l = 0;
|
---|
561 | }
|
---|
562 | else /* j == 18: 11 to 138 zero length codes */
|
---|
563 | {
|
---|
564 | ZIPNEEDBITS(7)
|
---|
565 | j = 11 + ((uint32_t)b & 0x7f);
|
---|
566 | ZIPDUMPBITS(7)
|
---|
567 | if ((uint32_t)i + j > n)
|
---|
568 | return 1;
|
---|
569 | while (j--)
|
---|
570 | ll[i++] = 0;
|
---|
571 | l = 0;
|
---|
572 | }
|
---|
573 | }
|
---|
574 |
|
---|
575 | /* free decoding table for trees */
|
---|
576 | Ziphuft_free(tl);
|
---|
577 |
|
---|
578 | /* restore the global bit buffer */
|
---|
579 | ZIP(bb) = b;
|
---|
580 | ZIP(bk) = k;
|
---|
581 |
|
---|
582 | /* build the decoding tables for literal/length and distance codes */
|
---|
583 | bl = ZIPLBITS;
|
---|
584 | if((i = Ziphuft_build(decomp_state, ll, nl, 257, Zipcplens, Zipcplext, &tl, &bl)) != 0)
|
---|
585 | {
|
---|
586 | if(i == 1)
|
---|
587 | Ziphuft_free(tl);
|
---|
588 | return i; /* incomplete code set */
|
---|
589 | }
|
---|
590 | bd = ZIPDBITS;
|
---|
591 | Ziphuft_build(decomp_state, ll + nl, nd, 0, Zipcpdist, Zipcpdext, &td, &bd);
|
---|
592 |
|
---|
593 | /* decompress until an end-of-block code */
|
---|
594 | if(Zipinflate_codes(decomp_state, tl, td, bl, bd))
|
---|
595 | return 1;
|
---|
596 |
|
---|
597 | /* free the decoding tables, return */
|
---|
598 | Ziphuft_free(tl);
|
---|
599 | Ziphuft_free(td);
|
---|
600 | return 0;
|
---|
601 | }
|
---|
602 |
|
---|
603 | /* e == last block flag */
|
---|
604 | static int32_t Zipinflate_block(struct decomp_state *decomp_state, int32_t *e)
|
---|
605 | { /* decompress an inflated block */
|
---|
606 | uint32_t t; /* block type */
|
---|
607 | register uint32_t b; /* bit buffer */
|
---|
608 | register uint32_t k; /* number of bits in bit buffer */
|
---|
609 |
|
---|
610 | DEBUG(10,("Zipinflate_block\n"));
|
---|
611 |
|
---|
612 | /* make local bit buffer */
|
---|
613 | b = ZIP(bb);
|
---|
614 | k = ZIP(bk);
|
---|
615 |
|
---|
616 | /* read in last block bit */
|
---|
617 | ZIPNEEDBITS(1)
|
---|
618 | *e = (int32_t)b & 1;
|
---|
619 | ZIPDUMPBITS(1)
|
---|
620 |
|
---|
621 | /* read in block type */
|
---|
622 | ZIPNEEDBITS(2)
|
---|
623 | t = (uint32_t)b & 3;
|
---|
624 | ZIPDUMPBITS(2)
|
---|
625 |
|
---|
626 | /* restore the global bit buffer */
|
---|
627 | ZIP(bb) = b;
|
---|
628 | ZIP(bk) = k;
|
---|
629 |
|
---|
630 | DEBUG(10,("inflate type %d\n", t));
|
---|
631 |
|
---|
632 | /* inflate that block type */
|
---|
633 | if(t == 2)
|
---|
634 | return Zipinflate_dynamic(decomp_state);
|
---|
635 | if(t == 0)
|
---|
636 | return Zipinflate_stored(decomp_state);
|
---|
637 | if(t == 1)
|
---|
638 | return Zipinflate_fixed(decomp_state);
|
---|
639 | /* bad block type */
|
---|
640 | return 2;
|
---|
641 | }
|
---|
642 |
|
---|
643 | _PUBLIC_ struct decomp_state *ZIPdecomp_state(TALLOC_CTX *mem_ctx)
|
---|
644 | {
|
---|
645 | return talloc_zero(mem_ctx, struct decomp_state);
|
---|
646 | }
|
---|
647 |
|
---|
648 | int ZIPdecompress(struct decomp_state *decomp_state, DATA_BLOB *inbuf, DATA_BLOB *outbuf)
|
---|
649 | {
|
---|
650 | int32_t e = 0;/* last block flag */
|
---|
651 |
|
---|
652 | ZIP(inpos) = CAB(inbuf);
|
---|
653 | ZIP(bb) = ZIP(bk) = ZIP(window_posn) = 0;
|
---|
654 |
|
---|
655 | if (inbuf->length > sizeof(decomp_state->inbuf)) return DECR_INPUT;
|
---|
656 |
|
---|
657 | if (outbuf->length > sizeof(decomp_state->outbuf)) return DECR_OUTPUT;
|
---|
658 |
|
---|
659 | if (outbuf->length > ZIPWSIZE) return DECR_DATAFORMAT;
|
---|
660 |
|
---|
661 | memcpy(decomp_state->inbuf, inbuf->data, inbuf->length);
|
---|
662 |
|
---|
663 | /* CK = Chris Kirmse, official Microsoft purloiner */
|
---|
664 | if (ZIP(inpos)[0] != 'C' || ZIP(inpos)[1] != 'K') return DECR_ILLEGALDATA;
|
---|
665 | ZIP(inpos) += 2;
|
---|
666 |
|
---|
667 | while (!e) {
|
---|
668 | if (Zipinflate_block(decomp_state, &e)) {
|
---|
669 | return DECR_ILLEGALDATA;
|
---|
670 | }
|
---|
671 | }
|
---|
672 |
|
---|
673 | memcpy(outbuf->data, decomp_state->outbuf, outbuf->length);
|
---|
674 |
|
---|
675 | return DECR_OK;
|
---|
676 | }
|
---|