inftrees.c
上传用户:hxb_1234
上传日期:2010-03-30
资源大小:8328k
文件大小:9k
源码类别:

VC书籍

开发平台:

Visual C++

  1. #include "zutil.h"
  2. #include "inftrees.h"
  3. #if !defined(BUILDFIXED) && !defined(STDC)
  4. #  define BUILDFIXED   
  5. #endif
  6. const char inflate_copyright[] =
  7.    " inflate 1.1.4 Copyright 1995-2002 Mark Adler ";
  8. struct internal_state  {int dummy;}; 
  9. #define exop word.what.Exop
  10. #define bits word.what.Bits
  11. local int huft_build OF((
  12.     uIntf *,            
  13.     uInt,              
  14.     uInt,               
  15.     const uIntf *,      
  16.     const uIntf *,      
  17.     inflate_huft * FAR*,
  18.     uIntf *,            
  19.     inflate_huft *,     
  20.     uInt *,            
  21.     uIntf * ));        
  22. local const uInt cplens[31] = { 
  23.         3, 4, 5, 6, 7, 8, 9, 10, 11, 13, 15, 17, 19, 23, 27, 31,
  24.         35, 43, 51, 59, 67, 83, 99, 115, 131, 163, 195, 227, 258, 0, 0};
  25.        
  26. local const uInt cplext[31] = { 
  27.         0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2,
  28.         3, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 0, 112, 112}; 
  29. local const uInt cpdist[30] = { 
  30.         1, 2, 3, 4, 5, 7, 9, 13, 17, 25, 33, 49, 65, 97, 129, 193,
  31.         257, 385, 513, 769, 1025, 1537, 2049, 3073, 4097, 6145,
  32.         8193, 12289, 16385, 24577};
  33. local const uInt cpdext[30] = {
  34.         0, 0, 0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6,
  35.         7, 7, 8, 8, 9, 9, 10, 10, 11, 11,
  36.         12, 12, 13, 13};
  37. #define BMAX 15        
  38. local int huft_build(b, n, s, d, e, t, m, hp, hn, v)
  39. uIntf *b;               
  40. uInt n;                 
  41. uInt s;                 
  42. const uIntf *d;         
  43. const uIntf *e;         
  44. inflate_huft * FAR *t;  
  45. uIntf *m;               
  46. inflate_huft *hp;       
  47. uInt *hn;            
  48. uIntf *v;               
  49. {
  50.   uInt a;                      
  51.   uInt c[BMAX+1];              
  52.   uInt f;                      
  53.   int g;                       
  54.   int h;                       
  55.   register uInt i;            
  56.   register uInt j;             
  57.   register int k;            
  58.   int l;                       
  59.   uInt mask;                   
  60.   register uIntf *p;            
  61.   inflate_huft *q;            
  62.   struct inflate_huft_s r;     
  63.   inflate_huft *u[BMAX];        
  64.   register int w;              
  65.   uInt x[BMAX+1];               
  66.   uIntf *xp;                 
  67.   int y;                        
  68.   uInt z;                     
  69.   p = c;
  70. #define C0 *p++ = 0;
  71. #define C2 C0 C0 C0 C0
  72. #define C4 C2 C2 C2 C2
  73.   C4                            
  74.   p = b;  i = n;
  75.   do {
  76.     c[*p++]++;               
  77.   } while (--i);
  78.   if (c[0] == n)                
  79.   {
  80.     *t = (inflate_huft *)Z_NULL;
  81.     *m = 0;
  82.     return Z_OK;
  83.   }
  84.   l = *m;
  85.   for (j = 1; j <= BMAX; j++)
  86.     if (c[j])
  87.       break;
  88.   k = j;                        
  89.   if ((uInt)l < j)
  90.     l = j;
  91.   for (i = BMAX; i; i--)
  92.     if (c[i])
  93.       break;
  94.   g = i;                       
  95.   if ((uInt)l > i)
  96.     l = i;
  97.   *m = l;
  98.   for (y = 1 << j; j < i; j++, y <<= 1)
  99.     if ((y -= c[j]) < 0)
  100.       return Z_DATA_ERROR;
  101.   if ((y -= c[i]) < 0)
  102.     return Z_DATA_ERROR;
  103.   c[i] += y;
  104.   x[1] = j = 0;
  105.   p = c + 1;  xp = x + 2;
  106.   while (--i) {                
  107.     *xp++ = (j += *p++);
  108.   }
  109.   p = b;  i = 0;
  110.   do {
  111.     if ((j = *p++) != 0)
  112.       v[x[j]++] = i;
  113.   } while (++i < n);
  114.   n = x[g];                    
  115.   x[0] = i = 0;                
  116.   p = v;                        
  117.   h = -1;                  
  118.   w = -l;                       
  119.   u[0] = (inflate_huft *)Z_NULL;       
  120.   q = (inflate_huft *)Z_NULL;  
  121.   z = 0;                        
  122.  
  123.   for (; k <= g; k++)
  124.   {
  125.     a = c[k];
  126.     while (a--)
  127.     {
  128.       
  129.       while (k > w + l)
  130.       {
  131.         h++;
  132.         w += l;                
  133.         z = g - w;
  134.         z = z > (uInt)l ? l : z;        
  135.         if ((f = 1 << (j = k - w)) > a + 1)    
  136.         {                    
  137.           f -= a + 1;           
  138.           xp = c + k;
  139.           if (j < z)
  140.             while (++j < z)     
  141.             {
  142.               if ((f <<= 1) <= *++xp)
  143.                 break;         
  144.               f -= *xp;        
  145.             }
  146.         }
  147.         z = 1 << j;             
  148.         
  149.         if (*hn + z > MANY)    
  150.           return Z_DATA_ERROR;  
  151.         u[h] = q = hp + *hn;
  152.         *hn += z;
  153.         if (h)
  154.         {
  155.           x[h] = i;             
  156.           r.bits = (Byte)l;     
  157.           r.exop = (Byte)j;     
  158.           j = i >> (w - l);
  159.           r.base = (uInt)(q - u[h-1] - j);   
  160.           u[h-1][j] = r;        
  161.         }
  162.         else
  163.           *t = q;              
  164.       }
  165.       r.bits = (Byte)(k - w);
  166.       if (p >= v + n)
  167.         r.exop = 128 + 64;     
  168.       else if (*p < s)
  169.       {
  170.         r.exop = (Byte)(*p < 256 ? 0 : 32 + 64);    
  171.         r.base = *p++;         
  172.       }
  173.       else
  174.       {
  175.         r.exop = (Byte)(e[*p - s] + 16 + 64);
  176.         r.base = d[*p++ - s];
  177.       }
  178.       
  179.       f = 1 << (k - w);
  180.       for (j = i >> w; j < z; j += f)
  181.         q[j] = r;
  182.     
  183.       for (j = 1 << (k - 1); i & j; j >>= 1)
  184.         i ^= j;
  185.       i ^= j;
  186.       mask = (1 << w) - 1;      
  187.       while ((i & mask) != x[h])
  188.       {
  189.         h--;                    
  190.         w -= l;
  191.         mask = (1 << w) - 1;
  192.       }
  193.     }
  194.   }
  195.   return y != 0 && g != 1 ? Z_BUF_ERROR : Z_OK;
  196. }
  197. int inflate_trees_bits(c, bb, tb, hp, z)
  198. uIntf *c;               
  199. uIntf *bb;              
  200. inflate_huft * FAR *tb; 
  201. inflate_huft *hp;       
  202. z_streamp z;            
  203. {
  204.   int r;
  205.   uInt hn = 0;          
  206.   uIntf *v;             
  207.   if ((v = (uIntf*)ZALLOC(z, 19, sizeof(uInt))) == Z_NULL)
  208.     return Z_MEM_ERROR;
  209.   r = huft_build(c, 19, 19, (uIntf*)Z_NULL, (uIntf*)Z_NULL,
  210.                  tb, bb, hp, &hn, v);
  211.   if (r == Z_DATA_ERROR)
  212.     z->msg = (char*)"oversubscribed dynamic bit lengths tree";
  213.   else if (r == Z_BUF_ERROR || *bb == 0)
  214.   {
  215.     z->msg = (char*)"incomplete dynamic bit lengths tree";
  216.     r = Z_DATA_ERROR;
  217.   }
  218.   ZFREE(z, v);
  219.   return r;
  220. }
  221. int inflate_trees_dynamic(nl, nd, c, bl, bd, tl, td, hp, z)
  222. uInt nl;               
  223. uInt nd;               
  224. uIntf *c;               
  225. uIntf *bl;              
  226. uIntf *bd;              
  227. inflate_huft * FAR *tl; 
  228. inflate_huft * FAR *td; 
  229. inflate_huft *hp;       
  230. z_streamp z;            
  231. {
  232.   int r;
  233.   uInt hn = 0;          
  234.   uIntf *v;             
  235.   if ((v = (uIntf*)ZALLOC(z, 288, sizeof(uInt))) == Z_NULL)
  236.     return Z_MEM_ERROR;
  237.   r = huft_build(c, nl, 257, cplens, cplext, tl, bl, hp, &hn, v);
  238.   if (r != Z_OK || *bl == 0)
  239.   {
  240.     if (r == Z_DATA_ERROR)
  241.       z->msg = (char*)"oversubscribed literal/length tree";
  242.     else if (r != Z_MEM_ERROR)
  243.     {
  244.       z->msg = (char*)"incomplete literal/length tree";
  245.       r = Z_DATA_ERROR;
  246.     }
  247.     ZFREE(z, v);
  248.     return r;
  249.   }
  250.   r = huft_build(c + nl, nd, 0, cpdist, cpdext, td, bd, hp, &hn, v);
  251.   if (r != Z_OK || (*bd == 0 && nl > 257))
  252.   {
  253.     if (r == Z_DATA_ERROR)
  254.       z->msg = (char*)"oversubscribed distance tree";
  255.     else if (r == Z_BUF_ERROR) {
  256. #ifdef PKZIP_BUG_WORKAROUND
  257.       r = Z_OK;
  258.     }
  259. #else
  260.       z->msg = (char*)"incomplete distance tree";
  261.       r = Z_DATA_ERROR;
  262.     }
  263.     else if (r != Z_MEM_ERROR)
  264.     {
  265.       z->msg = (char*)"empty distance tree with lengths";
  266.       r = Z_DATA_ERROR;
  267.     }
  268.     ZFREE(z, v);
  269.     return r;
  270. #endif
  271.   }
  272.   ZFREE(z, v);
  273.   return Z_OK;
  274. }
  275. #ifdef BUILDFIXED
  276. local int fixed_built = 0;
  277. #define FIXEDH 544      
  278. local inflate_huft fixed_mem[FIXEDH];
  279. local uInt fixed_bl;
  280. local uInt fixed_bd;
  281. local inflate_huft *fixed_tl;
  282. local inflate_huft *fixed_td;
  283. #else
  284. #include "inffixed.h"
  285. #endif
  286. int inflate_trees_fixed(bl, bd, tl, td, z)
  287. uIntf *bl;               
  288. uIntf *bd;               
  289. inflate_huft * FAR *tl; 
  290. inflate_huft * FAR *td;  
  291. z_streamp z;            
  292. {
  293. #ifdef BUILDFIXED
  294.   if (!fixed_built)
  295.   {
  296.     int k;              
  297.     uInt f = 0;        
  298.     uIntf *c;           
  299.     uIntf *v;         
  300.     
  301.     if ((c = (uIntf*)ZALLOC(z, 288, sizeof(uInt))) == Z_NULL)
  302.       return Z_MEM_ERROR;
  303.     if ((v = (uIntf*)ZALLOC(z, 288, sizeof(uInt))) == Z_NULL)
  304.     {
  305.       ZFREE(z, c);
  306.       return Z_MEM_ERROR;
  307.     }
  308.     for (k = 0; k < 144; k++)
  309.       c[k] = 8;
  310.     for (; k < 256; k++)
  311.       c[k] = 9;
  312.     for (; k < 280; k++)
  313.       c[k] = 7;
  314.     for (; k < 288; k++)
  315.       c[k] = 8;
  316.     fixed_bl = 9;
  317.     huft_build(c, 288, 257, cplens, cplext, &fixed_tl, &fixed_bl,
  318.                fixed_mem, &f, v);
  319.     for (k = 0; k < 30; k++)
  320.       c[k] = 5;
  321.     fixed_bd = 5;
  322.     huft_build(c, 30, 0, cpdist, cpdext, &fixed_td, &fixed_bd,
  323.                fixed_mem, &f, v);
  324.     ZFREE(z, v);
  325.     ZFREE(z, c);
  326.     fixed_built = 1;
  327.   }
  328. #endif
  329.   *bl = fixed_bl;
  330.   *bd = fixed_bd;
  331.   *tl = fixed_tl;
  332.   *td = fixed_td;
  333.   return Z_OK;
  334. }