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

VC书籍

开发平台:

Visual C++

  1. /* inffast.c -- process literals and length/distance pairs fast
  2.  * Copyright (C) 1995-2002 Mark Adler
  3.  * For conditions of distribution and use, see copyright notice in zlib.h 
  4.  */
  5. #include "zutil.h"
  6. #include "inftrees.h"
  7. #include "infblock.h"
  8. #include "infcodes.h"
  9. #include "infutil.h"
  10. #include "inffast.h"
  11. struct inflate_codes_state {int dummy;}; /* for buggy compilers */
  12. /* simplify the use of the inflate_huft type with some defines */
  13. #define exop word.what.Exop
  14. #define bits word.what.Bits
  15. /* macros for bit input with no checking and for returning unused bytes */
  16. #define GRABBITS(j) {while(k<(j)){b|=((uLong)NEXTBYTE)<<k;k+=8;}}
  17. #define UNGRAB {c=z->avail_in-n;c=(k>>3)<c?k>>3:c;n+=c;p-=c;k-=c<<3;}
  18. /* Called with number of bytes left to write in window at least 258
  19.    (the maximum string length) and number of input bytes available
  20.    at least ten.  The ten bytes are six bytes for the longest length/
  21.    distance pair plus four bytes for overloading the bit buffer. */
  22. int inflate_fast(bl, bd, tl, td, s, z)
  23. uInt bl, bd;
  24. inflate_huft *tl;
  25. inflate_huft *td; /* need separate declaration for Borland C++ */
  26. inflate_blocks_statef *s;
  27. z_streamp z;
  28. {
  29.   inflate_huft *t;      /* temporary pointer */
  30.   uInt e;               /* extra bits or operation */
  31.   uLong b;              /* bit buffer */
  32.   uInt k;               /* bits in bit buffer */
  33.   Bytef *p;             /* input data pointer */
  34.   uInt n;               /* bytes available there */
  35.   Bytef *q;             /* output window write pointer */
  36.   uInt m;               /* bytes to end of window or read pointer */
  37.   uInt ml;              /* mask for literal/length tree */
  38.   uInt md;              /* mask for distance tree */
  39.   uInt c;               /* bytes to copy */
  40.   uInt d;               /* distance back to copy from */
  41.   Bytef *r;             /* copy source pointer */
  42.   /* load input, output, bit values */
  43.   LOAD
  44.   /* initialize masks */
  45.   ml = inflate_mask[bl];
  46.   md = inflate_mask[bd];
  47.   /* do until not enough input or output space for fast loop */
  48.   do {                          /* assume called with m >= 258 && n >= 10 */
  49.     /* get literal/length code */
  50.     GRABBITS(20)                /* max bits for literal/length code */
  51.     if ((e = (t = tl + ((uInt)b & ml))->exop) == 0)
  52.     {
  53.       DUMPBITS(t->bits)
  54.       Tracevv((stderr, t->base >= 0x20 && t->base < 0x7f ?
  55.                 "inflate:         * literal '%c'n" :
  56.                 "inflate:         * literal 0x%02xn", t->base));
  57.       *q++ = (Byte)t->base;
  58.       m--;
  59.       continue;
  60.     }
  61.     do {
  62.       DUMPBITS(t->bits)
  63.       if (e & 16)
  64.       {
  65.         /* get extra bits for length */
  66.         e &= 15;
  67.         c = t->base + ((uInt)b & inflate_mask[e]);
  68.         DUMPBITS(e)
  69.         Tracevv((stderr, "inflate:         * length %un", c));
  70.         /* decode distance base of block to copy */
  71.         GRABBITS(15);           /* max bits for distance code */
  72.         e = (t = td + ((uInt)b & md))->exop;
  73.         do {
  74.           DUMPBITS(t->bits)
  75.           if (e & 16)
  76.           {
  77.             /* get extra bits to add to distance base */
  78.             e &= 15;
  79.             GRABBITS(e)         /* get extra bits (up to 13) */
  80.             d = t->base + ((uInt)b & inflate_mask[e]);
  81.             DUMPBITS(e)
  82.             Tracevv((stderr, "inflate:         * distance %un", d));
  83.             /* do the copy */
  84.             m -= c;
  85.             r = q - d;
  86.             if (r < s->window)                  /* wrap if needed */
  87.             {
  88.               do {
  89.                 r += s->end - s->window;        /* force pointer in window */
  90.               } while (r < s->window);          /* covers invalid distances */
  91.               e = s->end - r;
  92.               if (c > e)
  93.               {
  94.                 c -= e;                         /* wrapped copy */
  95.                 do {
  96.                     *q++ = *r++;
  97.                 } while (--e);
  98.                 r = s->window;
  99.                 do {
  100.                     *q++ = *r++;
  101.                 } while (--c);
  102.               }
  103.               else                              /* normal copy */
  104.               {
  105.                 *q++ = *r++;  c--;
  106.                 *q++ = *r++;  c--;
  107.                 do {
  108.                     *q++ = *r++;
  109.                 } while (--c);
  110.               }
  111.             }
  112.             else                                /* normal copy */
  113.             {
  114.               *q++ = *r++;  c--;
  115.               *q++ = *r++;  c--;
  116.               do {
  117.                 *q++ = *r++;
  118.               } while (--c);
  119.             }
  120.             break;
  121.           }
  122.           else if ((e & 64) == 0)
  123.           {
  124.             t += t->base;
  125.             e = (t += ((uInt)b & inflate_mask[e]))->exop;
  126.           }
  127.           else
  128.           {
  129.             z->msg = (char*)"invalid distance code";
  130.             UNGRAB
  131.             UPDATE
  132.             return Z_DATA_ERROR;
  133.           }
  134.         } while (1);
  135.         break;
  136.       }
  137.       if ((e & 64) == 0)
  138.       {
  139.         t += t->base;
  140.         if ((e = (t += ((uInt)b & inflate_mask[e]))->exop) == 0)
  141.         {
  142.           DUMPBITS(t->bits)
  143.           Tracevv((stderr, t->base >= 0x20 && t->base < 0x7f ?
  144.                     "inflate:         * literal '%c'n" :
  145.                     "inflate:         * literal 0x%02xn", t->base));
  146.           *q++ = (Byte)t->base;
  147.           m--;
  148.           break;
  149.         }
  150.       }
  151.       else if (e & 32)
  152.       {
  153.         Tracevv((stderr, "inflate:         * end of blockn"));
  154.         UNGRAB
  155.         UPDATE
  156.         return Z_STREAM_END;
  157.       }
  158.       else
  159.       {
  160.         z->msg = (char*)"invalid literal/length code";
  161.         UNGRAB
  162.         UPDATE
  163.         return Z_DATA_ERROR;
  164.       }
  165.     } while (1);
  166.   } while (m >= 258 && n >= 10);
  167.   /* not enough input or output--restore pointers and return */
  168.   UNGRAB
  169.   UPDATE
  170.   return Z_OK;
  171. }