← Back to branch summary

~zulcss/samba/server-dailies-3.4

« back to all changes in this revision

Viewing changes to lib/compression/mszip.c

  • Committer: Chuck Short
  • Date: 2010-09-28 20:38:39 UTC
  • Revision ID: zulcss@ubuntu.com-20100928203839-pgjulytsi9ue63x1
Initial version

Show diffs side-by-side

added added

removed removed

Lines of Context:
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 "../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
}