| 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 | } | 
|---|