source4/lib/compression/mszip.c

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