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, see <http://www.gnu.org/licenses/>.
  21  */
  22
  23 #include "includes.h"
  24 #include "lib/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 *) 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 }